| #ifdef _MSC_VER |
| #define _CRT_SECURE_NO_WARNINGS |
| #else |
| #endif |
| #include <string.h> |
| |
| #include "cn-cbor/cn-cbor.h" |
| |
| #ifdef USE_CBOR_CONTEXT |
| #include "context.h" |
| #define CBOR_CONTEXT_PARAM , NULL |
| #else |
| #define CBOR_CONTEXT_PARAM |
| #endif |
| |
| int CFails; |
| |
| #ifdef USE_CBOR_CONTEXT |
| void CreateTests() |
| { |
| cn_cbor_context* context = NULL; |
| |
| |
| context = CreateContext(-1); |
| |
| // Check the simple create/delete for memory leaks. |
| |
| cn_cbor* cbor = cn_cbor_map_create(context, NULL); |
| cn_cbor_free(cbor, context); |
| |
| uint8_t* pb = (uint8_t*)context->calloc_func(10, 10, context); |
| cbor = cn_cbor_data_create2(pb, 10, 0, context, NULL); |
| cn_cbor_free(cbor, context); |
| |
| char* sz = (char*)context->calloc_func(10, 1, context); |
| strcpy(sz, "ABC"); |
| cbor = cn_cbor_string_create2(sz, 0, context, NULL); |
| cn_cbor_free(cbor, context); |
| |
| cbor = cn_cbor_string_create("This is a string", context, NULL); |
| cn_cbor_dont_free_data(cbor); |
| cn_cbor_free(cbor, context); |
| |
| cbor = cn_cbor_int_create(20, context, NULL); |
| cn_cbor_free(cbor, context); |
| |
| #ifndef CBOR_NO_FLOATS |
| cbor = cn_cbor_float_create((float)20.2, context, NULL); |
| cn_cbor_free(cbor, context); |
| |
| cbor = cn_cbor_double_create(203.3, context, NULL); |
| cn_cbor_free(cbor, context); |
| #endif |
| |
| cbor = cn_cbor_array_create(context, NULL); |
| cn_cbor_free(cbor, context); |
| |
| cbor = cn_cbor_bool_create(false, context, NULL); |
| cn_cbor_free(cbor, context); |
| |
| cbor = cn_cbor_null_create(context, NULL); |
| cn_cbor_free(cbor, context); |
| |
| cbor = cn_cbor_simple_create(23, context, NULL); |
| cn_cbor_free(cbor, context); |
| |
| if (IsContextEmpty(context) > 0) { |
| CFails += 1; |
| } |
| |
| // Test more complex stuctures |
| |
| cn_cbor* cbor_map = cn_cbor_map_create(context, NULL); |
| cn_cbor* cbor_array = cn_cbor_array_create(context, NULL); |
| cn_cbor* cbor2 = NULL; |
| |
| for (int i = 0; i < 10; i++) { |
| cbor2 = cn_cbor_int_create(i, context, NULL); |
| cbor = cn_cbor_int_create(i, context, NULL); |
| cn_cbor_map_put(cbor_map, cbor, cbor2, NULL); |
| |
| cbor = cn_cbor_int_create(i, context, NULL); |
| cn_cbor_array_append(cbor_array, cbor, NULL); |
| } |
| |
| cn_cbor_mapput_string(cbor_map, "KEY", cn_cbor_int_create(20, context, NULL), context, NULL); |
| sz = (char*)context->calloc_func(10, 1, context); |
| strcpy(sz, "ABC"); |
| cn_cbor_mapput_string2(cbor_map, sz, cn_cbor_int_create(-20, context, NULL), 0, context, NULL); |
| |
| cn_cbor_mapput_int(cbor_map, -22, cn_cbor_simple_create(99, context, NULL), context, NULL); |
| |
| cn_cbor_array_append(cbor_array, cbor_map, NULL); |
| cbor_array = cn_cbor_tag_create(99, cbor_array, context, NULL); |
| cn_cbor_free(cbor_array, context); |
| |
| if (IsContextEmpty(context) > 0) { |
| CFails += 1; |
| } |
| |
| |
| } |
| |
| void DecoderTests() {} |
| |
| void EncoderTests() |
| { |
| bool finished = false; |
| for (int passNumber = 0; passNumber < 10000 && !finished; passNumber++) { |
| cn_cbor* cborRoot = NULL; |
| cn_cbor* cbor = NULL; |
| cn_cbor* cbor2 = NULL; |
| uint8_t* pb = NULL; |
| char* s = NULL; |
| |
| cn_cbor_context* context = CreateContext(passNumber); |
| if (context == NULL) { |
| CFails += 1; |
| return; |
| } |
| |
| cborRoot = cn_cbor_array_create(context, NULL); |
| if (cborRoot == NULL) { |
| goto errorReturn; |
| } |
| |
| cbor = cn_cbor_array_create(context, NULL); |
| if (cbor == NULL) { |
| goto errorReturn; |
| } |
| cbor->flags |= CN_CBOR_FL_INDEF; |
| |
| cbor2 = cn_cbor_simple_create(22, context, NULL); |
| if (cbor2 == NULL) { |
| goto errorReturn; |
| } |
| cn_cbor_array_append(cbor, cbor2, NULL); |
| cbor2 = NULL; |
| |
| cbor2 = cn_cbor_simple_create(21, context, NULL); |
| if (cbor2 == NULL) { |
| goto errorReturn; |
| } |
| cn_cbor_array_append(cbor, cbor2, NULL); |
| cbor2 = NULL; |
| |
| cn_cbor_array_append(cborRoot, cbor, NULL); |
| cbor = NULL; |
| |
| cbor = cn_cbor_bool_create(true, context, NULL); |
| if (cbor == NULL) { |
| goto errorReturn; |
| } |
| cn_cbor_array_append(cborRoot, cbor, NULL); |
| cbor = NULL; |
| |
| cbor = cn_cbor_map_create(context, NULL); |
| if (cbor == NULL) { |
| goto errorReturn; |
| } |
| |
| cbor2 = cn_cbor_string_create("Text1", context, NULL); |
| if (cbor2 == NULL) { |
| goto errorReturn; |
| } |
| if (!cn_cbor_mapput_int(cbor, 5, cbor2, context, NULL)) { |
| goto errorReturn; |
| }; |
| cbor2 = NULL; |
| |
| cbor2 = cn_cbor_int_create(99, context, NULL); |
| if (cbor2 == NULL) { |
| goto errorReturn; |
| } |
| if (!cn_cbor_mapput_string(cbor, "key", cbor2, context, NULL)) { |
| goto errorReturn; |
| } |
| cbor2 = NULL; |
| |
| cn_cbor_array_append(cborRoot, cbor, NULL); |
| cbor = NULL; |
| |
| cbor = cn_cbor_map_create(context, NULL); |
| if (cbor == NULL) { |
| goto errorReturn; |
| } |
| cbor->flags |= CN_CBOR_FL_INDEF; |
| |
| cbor2 = cn_cbor_string_create("Text1", context, NULL); |
| if (cbor2 == NULL) { |
| goto errorReturn; |
| } |
| if (!cn_cbor_mapput_int(cbor, 5, cbor2, context, NULL)) { |
| goto errorReturn; |
| } |
| cbor2 = NULL; |
| |
| cbor2 = cn_cbor_int_create(99, context, NULL); |
| if (cbor2 == NULL) { |
| goto errorReturn; |
| } |
| |
| if (!cn_cbor_mapput_string(cbor, "key", cbor2, context, NULL)) { |
| goto errorReturn; |
| } |
| cbor2 = NULL; |
| |
| cn_cbor_array_append(cborRoot, cbor, NULL); |
| cbor = NULL; |
| |
| cbor = cn_cbor_chunked_create(CN_CBOR_BYTES, context, NULL); |
| if (cbor == NULL) { |
| goto errorReturn; |
| } |
| pb = context->calloc_func(10, 10, context); |
| cbor2 = cn_cbor_data_create2(pb, 100, 0, context, NULL); |
| if (cbor2 == NULL) { |
| goto errorReturn; |
| } |
| pb = NULL; |
| |
| cn_cbor_chunked_append(cbor, cbor2, NULL); |
| cbor2 = NULL; |
| |
| uint8_t data2[20] = {1, 2, 3, 4, 5, 6, 7}; |
| cbor2 = cn_cbor_data_create(data2, 20, context, NULL); |
| if (cbor2 == NULL) { |
| goto errorReturn; |
| } |
| cn_cbor_chunked_append(cbor, cbor2, NULL); |
| cbor2 = NULL; |
| |
| cn_cbor_array_append(cborRoot, cbor, NULL); |
| cbor = NULL; |
| |
| cbor = cn_cbor_chunked_create(CN_CBOR_TEXT, context, NULL); |
| if (cbor == NULL) { |
| goto errorReturn; |
| } |
| |
| cbor2 = cn_cbor_string_create("This is a string", context, NULL); |
| if (cbor2 == NULL) { |
| goto errorReturn; |
| } |
| cn_cbor_chunked_append(cbor, cbor2, NULL); |
| cbor2 = NULL; |
| |
| s = context->calloc_func(20, 1, context); |
| if (s == NULL) { |
| goto errorReturn; |
| } |
| strcpy(s, "Hi Mom"); |
| cbor2 = cn_cbor_string_create2(s, 0, context, NULL); |
| if (cbor2 == NULL) { |
| goto errorReturn; |
| } |
| s = NULL; |
| |
| cn_cbor_chunked_append(cbor, cbor2, NULL); |
| cbor2 = NULL; |
| |
| cn_cbor_array_append(cborRoot, cbor, NULL); |
| cbor = NULL; |
| |
| cbor = cn_cbor_simple_create(4, context, NULL); |
| if (cbor == NULL) { |
| goto errorReturn; |
| } |
| cbor2 = cn_cbor_tag_create(99, cbor, context, NULL); |
| if (cbor2 == NULL) { |
| goto errorReturn; |
| } |
| cbor = NULL; |
| cn_cbor_array_append(cborRoot, cbor2, NULL); |
| cbor2 = NULL; |
| |
| #ifndef CBOR_NO_FLOATS |
| cbor = cn_cbor_float_create(9, context, NULL); |
| if (cbor == NULL) { |
| goto errorReturn; |
| } |
| cn_cbor_array_append(cborRoot, cbor, NULL); |
| cbor = NULL; |
| |
| cbor = cn_cbor_double_create(33.225932523223, context, NULL); |
| if (cbor == NULL) { |
| goto errorReturn; |
| } |
| cn_cbor_array_append(cborRoot, cbor, NULL); |
| cbor = NULL; |
| |
| cbor = cn_cbor_float_create(9, context, NULL); |
| if (cbor == NULL) { |
| goto errorReturn; |
| } |
| cbor->flags |= CN_CBOR_FL_KEEP_FLOAT_SIZE; |
| cn_cbor_array_append(cborRoot, cbor, NULL); |
| cbor = NULL; |
| |
| cbor = cn_cbor_double_create(9, context, NULL); |
| if (cbor == NULL) { |
| goto errorReturn; |
| } |
| cbor->flags |= CN_CBOR_FL_KEEP_FLOAT_SIZE; |
| cn_cbor_array_append(cborRoot, cbor, NULL); |
| cbor = NULL; |
| #endif |
| |
| ssize_t cb = cn_cbor_encoder_write(NULL, 0, 0, cborRoot); |
| pb = (uint8_t*)context->calloc_func(cb + 2, 1, context); |
| if (pb == NULL) { |
| goto errorReturn; |
| } |
| |
| ssize_t cb2 = cn_cbor_encoder_write(pb, 0, cb - 1, cborRoot); |
| if (cb2 != -1) { |
| CFails += 1; |
| } |
| |
| cb2 = cn_cbor_encoder_write(pb, 0, cb, cborRoot); |
| if (cb2 != cb) { |
| CFails += 1; |
| } |
| |
| cb2 = cn_cbor_encoder_write(pb, 0, cb + 1, cborRoot); |
| if (cb != cb2) { |
| CFails += 1; |
| } |
| |
| cn_cbor_free(cborRoot, context); |
| cborRoot = NULL; |
| |
| cborRoot = cn_cbor_decode(pb, cb2, context, NULL); |
| if (cborRoot == NULL) { |
| goto errorReturn; |
| } |
| |
| finished = true; |
| |
| errorReturn: |
| if (cborRoot != NULL) { |
| cn_cbor_free(cborRoot, context); |
| } |
| if (cbor != NULL) { |
| cn_cbor_free(cbor, context); |
| } |
| if (cbor2 != NULL) { |
| cn_cbor_free(cbor2, context); |
| } |
| if (pb != NULL) { |
| context->free_func(pb, context); |
| } |
| if (s != NULL) { |
| context->free_func(s, context); |
| } |
| |
| if (IsContextEmpty(context) > 0) { |
| CFails += 1; |
| } |
| |
| FreeContext(context); |
| } |
| } |
| #endif |
| |
| int main(void) |
| { |
| #ifdef USE_CBOR_CONTEXT |
| CreateTests(); |
| EncoderTests(); |
| DecoderTests(); |
| #endif |
| return CFails; |
| } |