uint encoding WIP

Author: PJK

src/CMakeLists.txt

@@ -1,4 +1,4 @@
-set(SOURCES cbor.c cbor_internal.c cbor_stack.c cbor_unicode.c)
+set(SOURCES cbor.c cbor_internal.c cbor_stack.c cbor_unicode.c cbor_encoders.c)
 
 add_library(cbor ${SOURCES})
 add_library(cbor_shared SHARED ${SOURCES})

src/cbor.h

@@ -238,6 +238,14 @@ struct cbor_decoder_result {
 
 struct cbor_decoder_result cbor_stream_decode(cbor_data, size_t, const struct cbor_callbacks *, void *);
 
+size_t cbor_encode_uint8(uint8_t, unsigned char *, size_t);
+size_t cbor_encode_uint16(uint16_t, unsigned char *, size_t);
+size_t cbor_encode_uint32(uint32_t, unsigned char *, size_t);
+size_t cbor_encode_uint64(uint64_t, unsigned char *, size_t);
+
+size_t cbor_encode_uint(uint64_t, unsigned char *, size_t);
+
+
 cbor_item_t * cbor_load(cbor_data source, size_t source_size, cbor_flags_t flags, struct cbor_load_result * result);
 
 void cbor_incref(cbor_item_t * item);

src/cbor_encoders.c

@@ -0,0 +1,66 @@
+#include "cbor.h"
+#include "cbor_internal.h"
+#include "magic.h"
+
+size_t cbor_encode_uint8(uint8_t value, unsigned char * buffer, size_t buffer_size)
+{
+	if (value <= 23) {
+		if (buffer_size >= 1) {
+			buffer[0] = value;
+			return 1;
+		}
+	} else {
+		if (buffer_size >= 2) {
+			buffer[0] = 0x18;
+			buffer[1] = value;
+			return 2;
+		}
+	}
+	return 0;
+}
+
+size_t cbor_encode_uint16(uint16_t value, unsigned char * buffer, size_t buffer_size)
+{
+	if (buffer_size >= 3) {
+		buffer[0] = 0x19;
+
+		#if defined(__clang__) || defined(__GNUC__)
+			*(uint16_t *)&buffer[1] = __builtin_bswap16(value);
+		#else
+			buffer[1] = value >> 8;
+			buffer[2] = value;
+		#endif
+
+		return 3;
+	} else
+		return 0;
+}
+
+size_t cbor_encode_uint32(uint32_t value, unsigned char * buffer, size_t buffer_size)
+{
+	if (buffer_size >= 5) {
+		buffer[0] = 0x1A;
+
+	#if defined(__clang__) || defined(__GNUC__)
+		*(uint32_t *)&buffer[1] = __builtin_bswap32(value);
+	#else
+		buffer[1] = value >> 24;
+		buffer[2] = value >> 16;
+		buffer[3] = value >> 8;
+		buffer[4] = value;
+	#endif
+
+		return 3;
+	} else
+		return 0;
+}
+
+size_t cbor_encode_uint64(uint64_t value, unsigned char * buffer, size_t buffer_size)
+{
+
+}
+
+size_t cbor_encode_uint(uint64_t value, unsigned char * buffer, size_t buffer_size)
+{
+
+}

test/type_0_encoders_test.c

@@ -0,0 +1,28 @@
+#include <stdarg.h>
+#include <stddef.h>
+#include <setjmp.h>
+#include <cmocka.h>
+#include <stdio.h>
+#include "cbor.h"
+#include <inttypes.h>
+
+
+unsigned char buffer[512];
+
+static void test_embedded_uint8(void **state) {
+	cbor_encode_uint8(14, buffer, 512);
+	assert_memory_equal(buffer, (unsigned char[]){ 0x0E }, 1);
+}
+
+static void test_uint16(void **state) {
+	cbor_encode_uint16(1000, buffer, 512);
+	assert_memory_equal(buffer, ((unsigned char[]){ 0x19, 0x03, 0xE8 }), 3);
+}
+
+int main(void) {
+	const UnitTest tests[] = {
+		unit_test(test_embedded_uint8),
+		unit_test(test_uint16)
+	};
+	return run_tests(tests);
+}