src/lib/support/BufferWriter.cpp

   1 /*
   2  *
   3  *    Copyright (c) 2021 Project CHIP Authors
   4  *
   5  *    Licensed under the Apache License, Version 2.0 (the "License");
   6  *    you may not use this file except in compliance with the License.
   7  *    You may obtain a copy of the License at
   8  *
   9  *        http://www.apache.org/licenses/LICENSE-2.0
  10  *
  11  *    Unless required by applicable law or agreed to in writing, software
  12  *    distributed under the License is distributed on an "AS IS" BASIS,
  13  *    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14  *    See the License for the specific language governing permissions and
  15  *    limitations under the License.
  16  */
  17
  18 #include "BufferWriter.h"
  19
  20 namespace chip {
  21 namespace Encoding {
  22
  23 BufferWriter & BufferWriter::Put(const char * s)
  24 {
  25     static_assert(CHAR_BIT == 8, "We're assuming char and uint8_t are the same size");
  26     while (*s != 0)
  27     {
  28         Put(static_cast<uint8_t>(*s++));
  29     }
  30     return *this;
  31 }
  32
  33 BufferWriter & BufferWriter::Put(const void * buf, size_t len)
  34 {
  35     size_t available = Available();
  36
  37     if (available > 0)
  38     {
  39         memmove(mBuf + mNeeded, buf, available < len ? available : len);
  40     }
  41
  42     mNeeded += len;
  43     return *this;
  44 }
  45
  46 BufferWriter & BufferWriter::Put(uint8_t c)
  47 {
  48     if (mNeeded < mSize)
  49     {
  50         mBuf[mNeeded] = c;
  51     }
  52     ++mNeeded;
  53     return *this;
  54 }
  55
  56 LittleEndian::BufferWriter & LittleEndian::BufferWriter::EndianPut(uint64_t x, size_t size)
  57 {
  58     while (size > 0)
  59     {
  60         uint8_t c = x & 0xff;
  61         Put(c);
  62         x >>= 8;
  63         size--;
  64     }
  65     return *this;
  66 }
  67
  68 BigEndian::BufferWriter & BigEndian::BufferWriter::EndianPut(uint64_t x, size_t size)
  69 {
  70     while (size-- > 0)
  71     {
  72         uint8_t c = (x >> (size * 8)) & 0xff;
  73         Put(c);
  74     }
  75     return *this;
  76 }
  77
  78 } // namespace Encoding
  79 } // namespace chip