Add stream operator for uint16_t

Author: RichLogan

include/quicr/detail/ctrl_message_types.h

@@ -14,6 +14,9 @@ namespace quicr::messages {
     quicr::Bytes& operator<<(quicr::Bytes& buffer, std::uint8_t value);
     quicr::BytesSpan operator>>(quicr::BytesSpan buffer, uint8_t& value);
 
+    Bytes& operator<<(Bytes& buffer, std::uint16_t value);
+    BytesSpan operator>>(BytesSpan buffer, std::uint16_t& value);
+
     quicr::Bytes& operator<<(quicr::Bytes& buffer, const quicr::UintVar& value);
 
     using GroupId = uint64_t;

src/ctrl_message_types.cpp

@@ -29,6 +29,14 @@ namespace quicr::messages {
         return buffer;
     }
 
+    Bytes& operator<<(Bytes& buffer, std::uint16_t value)
+    {
+        const std::uint16_t swapped = SwapBytes(value);
+        buffer.push_back(static_cast<uint8_t>(swapped >> 8 & 0xFF));
+        buffer.push_back(static_cast<uint8_t>(swapped & 0xFF));
+        return buffer;
+    }
+
     Bytes& operator<<(Bytes& buffer, std::uint64_t value)
     {
         UintVar varint = value;
@@ -51,6 +59,18 @@ namespace quicr::messages {
         return buffer.subspan(sizeof(value));
     }
 
+    BytesSpan operator>>(BytesSpan buffer, uint16_t& value)
+    {
+        if (buffer.size() < sizeof(value)) {
+            throw std::invalid_argument("Provider buffer too small");
+        }
+        const std::uint16_t high = buffer[0];
+        const std::uint16_t low = buffer[1];
+        value = high << 8 | low;
+        value = SwapBytes(value);
+        return buffer.subspan(sizeof(std::uint16_t));
+    }
+
     BytesSpan operator>>(BytesSpan buffer, uint64_t& value)
     {
         UintVar value_uv(buffer);

test/moq_ctrl_messages.cpp

@@ -978,4 +978,14 @@ TEST_CASE("Key Value Pair encode/decode")
             CHECK_EQ(out.value, value);
         }
     }
+}
+
+TEST_CASE("UInt16 Encode/decode")
+{
+    std::uint16_t value = 65535;
+    Bytes buffer;
+    buffer << value;
+    std::uint16_t reconstructed_value = 0;
+    buffer >> reconstructed_value;
+    CHECK_EQ(reconstructed_value, value);
 }