DXR is a code search and navigation tool aimed at making sense of large projects. It supports full-text and regex searches as well as structural queries.

Line Code
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75
/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

#ifndef BYTE_WRITER_H_
#define BYTE_WRITER_H_

#include "mozilla/EndianUtils.h"
#include "nsTArray.h"

namespace mozilla {

class ByteWriter
{
public:
  explicit ByteWriter(nsTArray<uint8_t>& aData)
    : mPtr(aData)
  {
  }
  ~ByteWriter()
  {
  }

  MOZ_MUST_USE bool WriteU8(uint8_t aByte)
  {
    return Write(&aByte, 1);
  }

  MOZ_MUST_USE bool WriteU16(uint16_t aShort)
  {
    uint8_t c[2];
    mozilla::BigEndian::writeUint16(&c[0], aShort);
    return Write(&c[0], 2);
  }

  MOZ_MUST_USE bool WriteU32(uint32_t aLong)
  {
    uint8_t c[4];
    mozilla::BigEndian::writeUint32(&c[0], aLong);
    return Write(&c[0], 4);
  }

  MOZ_MUST_USE bool Write32(int32_t aLong)
  {
    uint8_t c[4];
    mozilla::BigEndian::writeInt32(&c[0], aLong);
    return Write(&c[0], 4);
  }

  MOZ_MUST_USE bool WriteU64(uint64_t aLongLong)
  {
    uint8_t c[8];
    mozilla::BigEndian::writeUint64(&c[0], aLongLong);
    return Write(&c[0], 8);
  }

  MOZ_MUST_USE bool Write64(int64_t aLongLong)
  {
    uint8_t c[8];
    mozilla::BigEndian::writeInt64(&c[0], aLongLong);
    return Write(&c[0], 8);
  }

  MOZ_MUST_USE bool Write(const uint8_t* aSrc, size_t aCount)
  {
    return mPtr.AppendElements(aSrc, aCount, mozilla::fallible);
  }

private:
  nsTArray<uint8_t>& mPtr;
};

} // namespace mozilla

#endif