/* The MIT License (MIT) Copyright (c) 2020 Charles Gunyon Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the "Software"), to deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions: The above copyright notice and this permission notice shall be included in all copies or substantial portions of the Software. THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. */ #include "cmp.h" static const uint32_t cmp_version_ = 20; static const uint32_t cmp_mp_version_ = 5; enum { POSITIVE_FIXNUM_MARKER = 0x00, FIXMAP_MARKER = 0x80, FIXARRAY_MARKER = 0x90, FIXSTR_MARKER = 0xA0, NIL_MARKER = 0xC0, FALSE_MARKER = 0xC2, TRUE_MARKER = 0xC3, BIN8_MARKER = 0xC4, BIN16_MARKER = 0xC5, BIN32_MARKER = 0xC6, EXT8_MARKER = 0xC7, EXT16_MARKER = 0xC8, EXT32_MARKER = 0xC9, FLOAT_MARKER = 0xCA, DOUBLE_MARKER = 0xCB, U8_MARKER = 0xCC, U16_MARKER = 0xCD, U32_MARKER = 0xCE, U64_MARKER = 0xCF, S8_MARKER = 0xD0, S16_MARKER = 0xD1, S32_MARKER = 0xD2, S64_MARKER = 0xD3, FIXEXT1_MARKER = 0xD4, FIXEXT2_MARKER = 0xD5, FIXEXT4_MARKER = 0xD6, FIXEXT8_MARKER = 0xD7, FIXEXT16_MARKER = 0xD8, STR8_MARKER = 0xD9, STR16_MARKER = 0xDA, STR32_MARKER = 0xDB, ARRAY16_MARKER = 0xDC, ARRAY32_MARKER = 0xDD, MAP16_MARKER = 0xDE, MAP32_MARKER = 0xDF, NEGATIVE_FIXNUM_MARKER = 0xE0 }; enum { FIXARRAY_SIZE = 0xF, FIXMAP_SIZE = 0xF, FIXSTR_SIZE = 0x1F }; typedef enum cmp_error_e { CMP_ERROR_NONE, CMP_ERROR_STR_DATA_LENGTH_TOO_LONG, CMP_ERROR_BIN_DATA_LENGTH_TOO_LONG, CMP_ERROR_ARRAY_LENGTH_TOO_LONG, CMP_ERROR_MAP_LENGTH_TOO_LONG, CMP_ERROR_INPUT_VALUE_TOO_LARGE, CMP_ERROR_FIXED_VALUE_WRITING, CMP_ERROR_TYPE_MARKER_READING, CMP_ERROR_TYPE_MARKER_WRITING, CMP_ERROR_DATA_READING, CMP_ERROR_DATA_WRITING, CMP_ERROR_EXT_TYPE_READING, CMP_ERROR_EXT_TYPE_WRITING, CMP_ERROR_INVALID_TYPE, CMP_ERROR_LENGTH_READING, CMP_ERROR_LENGTH_WRITING, CMP_ERROR_SKIP_DEPTH_LIMIT_EXCEEDED, CMP_ERROR_INTERNAL, CMP_ERROR_DISABLED_FLOATING_POINT, CMP_ERROR_MAX } cmp_error_t; static const char *cmp_error_message(cmp_error_t error) { switch (error) { case CMP_ERROR_NONE: return "No Error"; case CMP_ERROR_STR_DATA_LENGTH_TOO_LONG: return "Specified string data length is too long (> 0xFFFFFFFF)"; case CMP_ERROR_BIN_DATA_LENGTH_TOO_LONG: return "Specified binary data length is too long (> 0xFFFFFFFF)"; case CMP_ERROR_ARRAY_LENGTH_TOO_LONG: return "Specified array length is too long (> 0xFFFFFFFF)"; case CMP_ERROR_MAP_LENGTH_TOO_LONG: return "Specified map length is too long (> 0xFFFFFFFF)"; case CMP_ERROR_INPUT_VALUE_TOO_LARGE: return "Input value is too large"; case CMP_ERROR_FIXED_VALUE_WRITING: return "Error writing fixed value"; case CMP_ERROR_TYPE_MARKER_READING: return "Error reading type marker"; case CMP_ERROR_TYPE_MARKER_WRITING: return "Error writing type marker"; case CMP_ERROR_DATA_READING: return "Error reading packed data"; case CMP_ERROR_DATA_WRITING: return "Error writing packed data"; case CMP_ERROR_EXT_TYPE_READING: return "Error reading ext type"; case CMP_ERROR_EXT_TYPE_WRITING: return "Error writing ext type"; case CMP_ERROR_INVALID_TYPE: return "Invalid type"; case CMP_ERROR_LENGTH_READING: return "Error reading size"; case CMP_ERROR_LENGTH_WRITING: return "Error writing size"; case CMP_ERROR_SKIP_DEPTH_LIMIT_EXCEEDED: return "Depth limit exceeded while skipping"; case CMP_ERROR_INTERNAL: return "Internal error"; case CMP_ERROR_DISABLED_FLOATING_POINT: return "Floating point operations disabled"; case CMP_ERROR_MAX: return "Max Error"; } return ""; } static bool is_bigendian(void) { #ifdef WORDS_BIGENDIAN return WORDS_BIGENDIAN; #else const int32_t one = 1; const char *one_bytes = (const char *)&one; return *one_bytes == 0; #endif } static uint16_t be16(uint16_t x) { if (!is_bigendian()) return ((x >> 8) & 0x00ff) | ((x << 8) & 0xff00); return x; } static int16_t sbe16(int16_t x) { return (int16_t)be16((uint16_t)x); } static uint32_t be32(uint32_t x) { if (!is_bigendian()) return ((uint32_t)be16((uint16_t)(x >> 16))) | ((uint32_t)be16((uint16_t)(x & 0xffff)) << 16); return x; } static int32_t sbe32(int32_t x) { return (int32_t)be32((uint32_t)x); } static uint64_t be64(uint64_t x) { if (!is_bigendian()) return ((uint64_t)be32((uint32_t)(x >> 32))) | ((uint64_t)be32((uint32_t)(x & 0xffffffff)) << 32); return x; } static int64_t sbe64(int64_t x) { return (int64_t)be64((uint64_t)x); } #ifndef CMP_NO_FLOAT static float decode_befloat(const char *b) { float f = 0.; char *fb = (char *)&f; if (!is_bigendian()) { fb[0] = b[3]; fb[1] = b[2]; fb[2] = b[1]; fb[3] = b[0]; } else { fb[0] = b[0]; fb[1] = b[1]; fb[2] = b[2]; fb[3] = b[3]; } return f; } static double decode_bedouble(const char *b) { double d = 0.; char *db = (char *)&d; if (!is_bigendian()) { db[0] = b[7]; db[1] = b[6]; db[2] = b[5]; db[3] = b[4]; db[4] = b[3]; db[5] = b[2]; db[6] = b[1]; db[7] = b[0]; } else { db[0] = b[0]; db[1] = b[1]; db[2] = b[2]; db[3] = b[3]; db[4] = b[4]; db[5] = b[5]; db[6] = b[6]; db[7] = b[7]; } return d; } #endif /* CMP_NO_FLOAT */ static bool read_byte(cmp_ctx_t *ctx, uint8_t *x) { return ctx->read(ctx, x, sizeof(uint8_t)); } static bool write_byte(cmp_ctx_t *ctx, uint8_t x) { return ctx->write(ctx, &x, sizeof(uint8_t)) == sizeof(uint8_t); } static bool skip_bytes(cmp_ctx_t *ctx, size_t count) { if (ctx->skip) { return ctx->skip(ctx, count); } else { size_t i; for (i = 0; i < count; ++i) { uint8_t floor; if (!ctx->read(ctx, &floor, sizeof(uint8_t))) { return false; } } return true; } } static bool read_type_marker(cmp_ctx_t *ctx, uint8_t *marker) { if (read_byte(ctx, marker)) { return true; } ctx->error = CMP_ERROR_TYPE_MARKER_READING; return false; } static bool write_type_marker(cmp_ctx_t *ctx, uint8_t marker) { if (write_byte(ctx, marker)) return true; ctx->error = CMP_ERROR_TYPE_MARKER_WRITING; return false; } static bool write_fixed_value(cmp_ctx_t *ctx, uint8_t value) { if (write_byte(ctx, value)) return true; ctx->error = CMP_ERROR_FIXED_VALUE_WRITING; return false; } static bool type_marker_to_cmp_type(uint8_t type_marker, uint8_t *cmp_type) { if (type_marker <= 0x7F) { *cmp_type = CMP_TYPE_POSITIVE_FIXNUM; return true; } if (type_marker <= 0x8F) { *cmp_type = CMP_TYPE_FIXMAP; return true; } if (type_marker <= 0x9F) { *cmp_type = CMP_TYPE_FIXARRAY; return true; } if (type_marker <= 0xBF) { *cmp_type = CMP_TYPE_FIXSTR; return true; } if (type_marker >= 0xE0) { *cmp_type = CMP_TYPE_NEGATIVE_FIXNUM; return true; } switch (type_marker) { case NIL_MARKER: *cmp_type = CMP_TYPE_NIL; return true; case FALSE_MARKER: *cmp_type = CMP_TYPE_BOOLEAN; return true; case TRUE_MARKER: *cmp_type = CMP_TYPE_BOOLEAN; return true; case BIN8_MARKER: *cmp_type = CMP_TYPE_BIN8; return true; case BIN16_MARKER: *cmp_type = CMP_TYPE_BIN16; return true; case BIN32_MARKER: *cmp_type = CMP_TYPE_BIN32; return true; case EXT8_MARKER: *cmp_type = CMP_TYPE_EXT8; return true; case EXT16_MARKER: *cmp_type = CMP_TYPE_EXT16; return true; case EXT32_MARKER: *cmp_type = CMP_TYPE_EXT32; return true; case FLOAT_MARKER: *cmp_type = CMP_TYPE_FLOAT; return true; case DOUBLE_MARKER: *cmp_type = CMP_TYPE_DOUBLE; return true; case U8_MARKER: *cmp_type = CMP_TYPE_UINT8; return true; case U16_MARKER: *cmp_type = CMP_TYPE_UINT16; return true; case U32_MARKER: *cmp_type = CMP_TYPE_UINT32; return true; case U64_MARKER: *cmp_type = CMP_TYPE_UINT64; return true; case S8_MARKER: *cmp_type = CMP_TYPE_SINT8; return true; case S16_MARKER: *cmp_type = CMP_TYPE_SINT16; return true; case S32_MARKER: *cmp_type = CMP_TYPE_SINT32; return true; case S64_MARKER: *cmp_type = CMP_TYPE_SINT64; return true; case FIXEXT1_MARKER: *cmp_type = CMP_TYPE_FIXEXT1; return true; case FIXEXT2_MARKER: *cmp_type = CMP_TYPE_FIXEXT2; return true; case FIXEXT4_MARKER: *cmp_type = CMP_TYPE_FIXEXT4; return true; case FIXEXT8_MARKER: *cmp_type = CMP_TYPE_FIXEXT8; return true; case FIXEXT16_MARKER: *cmp_type = CMP_TYPE_FIXEXT16; return true; case STR8_MARKER: *cmp_type = CMP_TYPE_STR8; return true; case STR16_MARKER: *cmp_type = CMP_TYPE_STR16; return true; case STR32_MARKER: *cmp_type = CMP_TYPE_STR32; return true; case ARRAY16_MARKER: *cmp_type = CMP_TYPE_ARRAY16; return true; case ARRAY32_MARKER: *cmp_type = CMP_TYPE_ARRAY32; return true; case MAP16_MARKER: *cmp_type = CMP_TYPE_MAP16; return true; case MAP32_MARKER: *cmp_type = CMP_TYPE_MAP32; return true; default: return false; } } static bool read_type_size(cmp_ctx_t *ctx, uint8_t type_marker, uint8_t cmp_type, uint32_t *size) { uint8_t u8temp = 0; uint16_t u16temp = 0; uint32_t u32temp = 0; switch (cmp_type) { case CMP_TYPE_POSITIVE_FIXNUM: *size = 0; return true; case CMP_TYPE_FIXMAP: *size = (type_marker & FIXMAP_SIZE); return true; case CMP_TYPE_FIXARRAY: *size = (type_marker & FIXARRAY_SIZE); return true; case CMP_TYPE_FIXSTR: *size = (type_marker & FIXSTR_SIZE); return true; case CMP_TYPE_NIL: *size = 0; return true; case CMP_TYPE_BOOLEAN: *size = 0; return true; case CMP_TYPE_BIN8: if (!ctx->read(ctx, &u8temp, sizeof(uint8_t))) { ctx->error = CMP_ERROR_LENGTH_READING; return false; } *size = u8temp; return true; case CMP_TYPE_BIN16: if (!ctx->read(ctx, &u16temp, sizeof(uint16_t))) { ctx->error = CMP_ERROR_LENGTH_READING; return false; } *size = be16(u16temp); return true; case CMP_TYPE_BIN32: if (!ctx->read(ctx, &u32temp, sizeof(uint32_t))) { ctx->error = CMP_ERROR_LENGTH_READING; return false; } *size = be32(u32temp); return true; case CMP_TYPE_EXT8: if (!ctx->read(ctx, &u8temp, sizeof(uint8_t))) { ctx->error = CMP_ERROR_LENGTH_READING; return false; } *size = u8temp; return true; case CMP_TYPE_EXT16: if (!ctx->read(ctx, &u16temp, sizeof(uint16_t))) { ctx->error = CMP_ERROR_LENGTH_READING; return false; } *size = be16(u16temp); return true; case CMP_TYPE_EXT32: if (!ctx->read(ctx, &u32temp, sizeof(uint32_t))) { ctx->error = CMP_ERROR_LENGTH_READING; return false; } *size = be32(u32temp); return true; case CMP_TYPE_FLOAT: *size = 4; return true; case CMP_TYPE_DOUBLE: *size = 8; return true; case CMP_TYPE_UINT8: *size = 1; return true; case CMP_TYPE_UINT16: *size = 2; return true; case CMP_TYPE_UINT32: *size = 4; return true; case CMP_TYPE_UINT64: *size = 8; return true; case CMP_TYPE_SINT8: *size = 1; return true; case CMP_TYPE_SINT16: *size = 2; return true; case CMP_TYPE_SINT32: *size = 4; return true; case CMP_TYPE_SINT64: *size = 8; return true; case CMP_TYPE_FIXEXT1: *size = 1; return true; case CMP_TYPE_FIXEXT2: *size = 2; return true; case CMP_TYPE_FIXEXT4: *size = 4; return true; case CMP_TYPE_FIXEXT8: *size = 8; return true; case CMP_TYPE_FIXEXT16: *size = 16; return true; case CMP_TYPE_STR8: if (!ctx->read(ctx, &u8temp, sizeof(uint8_t))) { ctx->error = CMP_ERROR_DATA_READING; return false; } *size = u8temp; return true; case CMP_TYPE_STR16: if (!ctx->read(ctx, &u16temp, sizeof(uint16_t))) { ctx->error = CMP_ERROR_DATA_READING; return false; } *size = be16(u16temp); return true; case CMP_TYPE_STR32: if (!ctx->read(ctx, &u32temp, sizeof(uint32_t))) { ctx->error = CMP_ERROR_DATA_READING; return false; } *size = be32(u32temp); return true; case CMP_TYPE_ARRAY16: if (!ctx->read(ctx, &u16temp, sizeof(uint16_t))) { ctx->error = CMP_ERROR_DATA_READING; return false; } *size = be16(u16temp); return true; case CMP_TYPE_ARRAY32: if (!ctx->read(ctx, &u32temp, sizeof(uint32_t))) { ctx->error = CMP_ERROR_DATA_READING; return false; } *size = be32(u32temp); return true; case CMP_TYPE_MAP16: if (!ctx->read(ctx, &u16temp, sizeof(uint16_t))) { ctx->error = CMP_ERROR_DATA_READING; return false; } *size = be16(u16temp); return true; case CMP_TYPE_MAP32: if (!ctx->read(ctx, &u32temp, sizeof(uint32_t))) { ctx->error = CMP_ERROR_DATA_READING; return false; } *size = be32(u32temp); return true; case CMP_TYPE_NEGATIVE_FIXNUM: *size = 0; return true; default: ctx->error = CMP_ERROR_INVALID_TYPE; return false; } } static bool read_obj_data(cmp_ctx_t *ctx, uint8_t type_marker, cmp_object_t *obj) { switch (obj->type) { case CMP_TYPE_POSITIVE_FIXNUM: obj->as.u8 = type_marker; return true; case CMP_TYPE_NEGATIVE_FIXNUM: obj->as.s8 = (int8_t)type_marker; return true; case CMP_TYPE_NIL: obj->as.u8 = 0; return true; case CMP_TYPE_BOOLEAN: switch (type_marker) { case TRUE_MARKER: obj->as.boolean = true; return true; case FALSE_MARKER: obj->as.boolean = false; return true; default: break; } ctx->error = CMP_ERROR_INTERNAL; return false; case CMP_TYPE_UINT8: if (!ctx->read(ctx, &obj->as.u8, sizeof(uint8_t))) { ctx->error = CMP_ERROR_DATA_READING; return false; } return true; case CMP_TYPE_UINT16: if (!ctx->read(ctx, &obj->as.u16, sizeof(uint16_t))) { ctx->error = CMP_ERROR_DATA_READING; return false; } obj->as.u16 = be16(obj->as.u16); return true; case CMP_TYPE_UINT32: if (!ctx->read(ctx, &obj->as.u32, sizeof(uint32_t))) { ctx->error = CMP_ERROR_DATA_READING; return false; } obj->as.u32 = be32(obj->as.u32); return true; case CMP_TYPE_UINT64: if (!ctx->read(ctx, &obj->as.u64, sizeof(uint64_t))) { ctx->error = CMP_ERROR_DATA_READING; return false; } obj->as.u64 = be64(obj->as.u64); return true; case CMP_TYPE_SINT8: if (!ctx->read(ctx, &obj->as.s8, sizeof(int8_t))) { ctx->error = CMP_ERROR_DATA_READING; return false; } return true; case CMP_TYPE_SINT16: if (!ctx->read(ctx, &obj->as.s16, sizeof(int16_t))) { ctx->error = CMP_ERROR_DATA_READING; return false; } obj->as.s16 = sbe16(obj->as.s16); return true; case CMP_TYPE_SINT32: if (!ctx->read(ctx, &obj->as.s32, sizeof(int32_t))) { ctx->error = CMP_ERROR_DATA_READING; return false; } obj->as.s32 = sbe32(obj->as.s32); return true; case CMP_TYPE_SINT64: if (!ctx->read(ctx, &obj->as.s64, sizeof(int64_t))) { ctx->error = CMP_ERROR_DATA_READING; return false; } obj->as.s64 = sbe64(obj->as.s64); return true; case CMP_TYPE_FLOAT: { #ifndef CMP_NO_FLOAT char bytes[4]; if (!ctx->read(ctx, bytes, 4)) { ctx->error = CMP_ERROR_DATA_READING; return false; } obj->as.flt = decode_befloat(bytes); return true; #else /* CMP_NO_FLOAT */ ctx->error = CMP_ERROR_DISABLED_FLOATING_POINT; return false; #endif /* CMP_NO_FLOAT */ } case CMP_TYPE_DOUBLE: { #ifndef CMP_NO_FLOAT char bytes[8]; if (!ctx->read(ctx, bytes, 8)) { ctx->error = CMP_ERROR_DATA_READING; return false; } obj->as.dbl = decode_bedouble(bytes); return true; #else /* CMP_NO_FLOAT */ ctx->error = CMP_ERROR_DISABLED_FLOATING_POINT; return false; #endif /* CMP_NO_FLOAT */ } case CMP_TYPE_BIN8: case CMP_TYPE_BIN16: case CMP_TYPE_BIN32: return read_type_size(ctx, type_marker, obj->type, &obj->as.bin_size); case CMP_TYPE_FIXSTR: case CMP_TYPE_STR8: case CMP_TYPE_STR16: case CMP_TYPE_STR32: return read_type_size(ctx, type_marker, obj->type, &obj->as.str_size); case CMP_TYPE_FIXARRAY: case CMP_TYPE_ARRAY16: case CMP_TYPE_ARRAY32: return read_type_size(ctx, type_marker, obj->type, &obj->as.array_size); case CMP_TYPE_FIXMAP: case CMP_TYPE_MAP16: case CMP_TYPE_MAP32: return read_type_size(ctx, type_marker, obj->type, &obj->as.map_size); case CMP_TYPE_FIXEXT1: if (!ctx->read(ctx, &obj->as.ext.type, sizeof(int8_t))) { ctx->error = CMP_ERROR_EXT_TYPE_READING; return false; } obj->as.ext.size = 1; return true; case CMP_TYPE_FIXEXT2: if (!ctx->read(ctx, &obj->as.ext.type, sizeof(int8_t))) { ctx->error = CMP_ERROR_EXT_TYPE_READING; return false; } obj->as.ext.size = 2; return true; case CMP_TYPE_FIXEXT4: if (!ctx->read(ctx, &obj->as.ext.type, sizeof(int8_t))) { ctx->error = CMP_ERROR_EXT_TYPE_READING; return false; } obj->as.ext.size = 4; return true; case CMP_TYPE_FIXEXT8: if (!ctx->read(ctx, &obj->as.ext.type, sizeof(int8_t))) { ctx->error = CMP_ERROR_EXT_TYPE_READING; return false; } obj->as.ext.size = 8; return true; case CMP_TYPE_FIXEXT16: if (!ctx->read(ctx, &obj->as.ext.type, sizeof(int8_t))) { ctx->error = CMP_ERROR_EXT_TYPE_READING; return false; } obj->as.ext.size = 16; return true; case CMP_TYPE_EXT8: if (!read_type_size(ctx, type_marker, obj->type, &obj->as.ext.size)) { return false; } if (!ctx->read(ctx, &obj->as.ext.type, sizeof(int8_t))) { ctx->error = CMP_ERROR_EXT_TYPE_READING; return false; } return true; case CMP_TYPE_EXT16: if (!read_type_size(ctx, type_marker, obj->type, &obj->as.ext.size)) { return false; } if (!ctx->read(ctx, &obj->as.ext.type, sizeof(int8_t))) { ctx->error = CMP_ERROR_EXT_TYPE_READING; return false; } return true; case CMP_TYPE_EXT32: if (!read_type_size(ctx, type_marker, obj->type, &obj->as.ext.size)) { return false; } if (!ctx->read(ctx, &obj->as.ext.type, sizeof(int8_t))) { ctx->error = CMP_ERROR_EXT_TYPE_READING; return false; } return true; default: break; } ctx->error = CMP_ERROR_INVALID_TYPE; return false; } void cmp_init(cmp_ctx_t *ctx, void *buf, cmp_reader read, cmp_skipper skip, cmp_writer write) { ctx->error = CMP_ERROR_NONE; ctx->buf = buf; ctx->read = read; ctx->skip = skip; ctx->write = write; } uint32_t cmp_version(void) { return cmp_version_; } uint32_t cmp_mp_version(void) { return cmp_mp_version_; } const char* cmp_strerror(const cmp_ctx_t *ctx) { if (ctx->error > CMP_ERROR_NONE && ctx->error < CMP_ERROR_MAX) return cmp_error_message((cmp_error_t)ctx->error); return ""; } bool cmp_write_pfix(cmp_ctx_t *ctx, uint8_t c) { if (c <= 0x7F) return write_fixed_value(ctx, c); ctx->error = CMP_ERROR_INPUT_VALUE_TOO_LARGE; return false; } bool cmp_write_nfix(cmp_ctx_t *ctx, int8_t c) { if (c >= -0x20 && c <= -1) return write_fixed_value(ctx, (uint8_t)c); ctx->error = CMP_ERROR_INPUT_VALUE_TOO_LARGE; return false; } bool cmp_write_sfix(cmp_ctx_t *ctx, int8_t c) { if (c >= 0) return cmp_write_pfix(ctx, (uint8_t)c); if (c >= -0x20 && c <= -1) return cmp_write_nfix(ctx, c); ctx->error = CMP_ERROR_INPUT_VALUE_TOO_LARGE; return false; } bool cmp_write_s8(cmp_ctx_t *ctx, int8_t c) { if (!write_type_marker(ctx, S8_MARKER)) return false; return ctx->write(ctx, &c, sizeof(int8_t)) == sizeof(int8_t); } bool cmp_write_s16(cmp_ctx_t *ctx, int16_t s) { if (!write_type_marker(ctx, S16_MARKER)) return false; s = sbe16(s); return ctx->write(ctx, &s, sizeof(int16_t)) == sizeof(int16_t); } bool cmp_write_s32(cmp_ctx_t *ctx, int32_t i) { if (!write_type_marker(ctx, S32_MARKER)) return false; i = sbe32(i); return ctx->write(ctx, &i, sizeof(int32_t)) == sizeof(int32_t); } bool cmp_write_s64(cmp_ctx_t *ctx, int64_t l) { if (!write_type_marker(ctx, S64_MARKER)) return false; l = sbe64(l); return ctx->write(ctx, &l, sizeof(int64_t)) == sizeof(int64_t); } bool cmp_write_integer(cmp_ctx_t *ctx, int64_t d) { if (d >= 0) return cmp_write_uinteger(ctx, (uint64_t)d); if (d >= -0x20) return cmp_write_nfix(ctx, (int8_t)d); if (d >= -0x80) return cmp_write_s8(ctx, (int8_t)d); if (d >= -0x8000) return cmp_write_s16(ctx, (int16_t)d); if (d >= -INT64_C(0x80000000)) return cmp_write_s32(ctx, (int32_t)d); return cmp_write_s64(ctx, d); } bool cmp_write_ufix(cmp_ctx_t *ctx, uint8_t c) { return cmp_write_pfix(ctx, c); } bool cmp_write_u8(cmp_ctx_t *ctx, uint8_t c) { if (!write_type_marker(ctx, U8_MARKER)) return false; return ctx->write(ctx, &c, sizeof(uint8_t)) == sizeof(uint8_t); } bool cmp_write_u16(cmp_ctx_t *ctx, uint16_t s) { if (!write_type_marker(ctx, U16_MARKER)) return false; s = be16(s); return ctx->write(ctx, &s, sizeof(uint16_t)) == sizeof(uint16_t); } bool cmp_write_u32(cmp_ctx_t *ctx, uint32_t i) { if (!write_type_marker(ctx, U32_MARKER)) return false; i = be32(i); return ctx->write(ctx, &i, sizeof(uint32_t)) == sizeof(uint32_t); } bool cmp_write_u64(cmp_ctx_t *ctx, uint64_t l) { if (!write_type_marker(ctx, U64_MARKER)) return false; l = be64(l); return ctx->write(ctx, &l, sizeof(uint64_t)) == sizeof(uint64_t); } bool cmp_write_uinteger(cmp_ctx_t *ctx, uint64_t u) { if (u <= 0x7F) return cmp_write_pfix(ctx, (uint8_t)u); if (u <= 0xFF) return cmp_write_u8(ctx, (uint8_t)u); if (u <= 0xFFFF) return cmp_write_u16(ctx, (uint16_t)u); if (u <= 0xFFFFFFFF) return cmp_write_u32(ctx, (uint32_t)u); return cmp_write_u64(ctx, u); } #ifndef CMP_NO_FLOAT bool cmp_write_float(cmp_ctx_t *ctx, float f) { if (!write_type_marker(ctx, FLOAT_MARKER)) return false; /* * We may need to swap the float's bytes, but we can't just swap them inside * the float because the swapped bytes may not constitute a valid float. * Therefore, we have to create a buffer and swap the bytes there. */ if (!is_bigendian()) { char swapped[sizeof(float)]; char *fbuf = (char *)&f; size_t i; for (i = 0; i < sizeof(float); ++i) swapped[i] = fbuf[sizeof(float) - i - 1]; return ctx->write(ctx, swapped, sizeof(float)) == sizeof(float); } return ctx->write(ctx, &f, sizeof(float)) == sizeof(float); } bool cmp_write_double(cmp_ctx_t *ctx, double d) { if (!write_type_marker(ctx, DOUBLE_MARKER)) return false; /* Same deal for doubles */ if (!is_bigendian()) { char swapped[sizeof(double)]; char *dbuf = (char *)&d; size_t i; for (i = 0; i < sizeof(double); ++i) swapped[i] = dbuf[sizeof(double) - i - 1]; return ctx->write(ctx, swapped, sizeof(double)) == sizeof(double); } return ctx->write(ctx, &d, sizeof(double)) == sizeof(double); } bool cmp_write_decimal(cmp_ctx_t *ctx, double d) { float f = (float)d; double df = (double)f; if (df == d) return cmp_write_float(ctx, f); else return cmp_write_double(ctx, d); } #endif /* CMP_NO_FLOAT */ bool cmp_write_nil(cmp_ctx_t *ctx) { return write_type_marker(ctx, NIL_MARKER); } bool cmp_write_true(cmp_ctx_t *ctx) { return write_type_marker(ctx, TRUE_MARKER); } bool cmp_write_false(cmp_ctx_t *ctx) { return write_type_marker(ctx, FALSE_MARKER); } bool cmp_write_bool(cmp_ctx_t *ctx, bool b) { if (b) return cmp_write_true(ctx); return cmp_write_false(ctx); } bool cmp_write_u8_as_bool(cmp_ctx_t *ctx, uint8_t b) { if (b) return cmp_write_true(ctx); return cmp_write_false(ctx); } bool cmp_write_fixstr_marker(cmp_ctx_t *ctx, uint8_t size) { if (size <= FIXSTR_SIZE) return write_fixed_value(ctx, FIXSTR_MARKER | size); ctx->error = CMP_ERROR_INPUT_VALUE_TOO_LARGE; return false; } bool cmp_write_fixstr(cmp_ctx_t *ctx, const char *data, uint8_t size) { if (!cmp_write_fixstr_marker(ctx, size)) return false; if (size == 0) return true; if (ctx->write(ctx, data, size) == size) return true; ctx->error = CMP_ERROR_DATA_WRITING; return false; } bool cmp_write_str8_marker(cmp_ctx_t *ctx, uint8_t size) { if (!write_type_marker(ctx, STR8_MARKER)) return false; if (ctx->write(ctx, &size, sizeof(uint8_t)) == sizeof(uint8_t)) return true; ctx->error = CMP_ERROR_LENGTH_WRITING; return false; } bool cmp_write_str8(cmp_ctx_t *ctx, const char *data, uint8_t size) { if (!cmp_write_str8_marker(ctx, size)) return false; if (size == 0) return true; if (ctx->write(ctx, data, size) == size) return true; ctx->error = CMP_ERROR_DATA_WRITING; return false; } bool cmp_write_str16_marker(cmp_ctx_t *ctx, uint16_t size) { if (!write_type_marker(ctx, STR16_MARKER)) return false; size = be16(size); if (ctx->write(ctx, &size, sizeof(uint16_t)) == sizeof(uint16_t)) return true; ctx->error = CMP_ERROR_LENGTH_WRITING; return false; } bool cmp_write_str16(cmp_ctx_t *ctx, const char *data, uint16_t size) { if (!cmp_write_str16_marker(ctx, size)) return false; if (size == 0) return true; if (ctx->write(ctx, data, size) == size) return true; ctx->error = CMP_ERROR_DATA_WRITING; return false; } bool cmp_write_str32_marker(cmp_ctx_t *ctx, uint32_t size) { if (!write_type_marker(ctx, STR32_MARKER)) return false; size = be32(size); if (ctx->write(ctx, &size, sizeof(uint32_t)) == sizeof(uint32_t)) return true; ctx->error = CMP_ERROR_LENGTH_WRITING; return false; } bool cmp_write_str32(cmp_ctx_t *ctx, const char *data, uint32_t size) { if (!cmp_write_str32_marker(ctx, size)) return false; if (size == 0) return true; if (ctx->write(ctx, data, size) == size) return true; ctx->error = CMP_ERROR_DATA_WRITING; return false; } bool cmp_write_str_marker(cmp_ctx_t *ctx, uint32_t size) { if (size <= FIXSTR_SIZE) return cmp_write_fixstr_marker(ctx, (uint8_t)size); if (size <= 0xFF) return cmp_write_str8_marker(ctx, (uint8_t)size); if (size <= 0xFFFF) return cmp_write_str16_marker(ctx, (uint16_t)size); return cmp_write_str32_marker(ctx, size); } bool cmp_write_str_marker_v4(cmp_ctx_t *ctx, uint32_t size) { if (size <= FIXSTR_SIZE) return cmp_write_fixstr_marker(ctx, (uint8_t)size); if (size <= 0xFFFF) return cmp_write_str16_marker(ctx, (uint16_t)size); return cmp_write_str32_marker(ctx, size); } bool cmp_write_str(cmp_ctx_t *ctx, const char *data, uint32_t size) { if (size <= FIXSTR_SIZE) return cmp_write_fixstr(ctx, data, (uint8_t)size); if (size <= 0xFF) return cmp_write_str8(ctx, data, (uint8_t)size); if (size <= 0xFFFF) return cmp_write_str16(ctx, data, (uint16_t)size); return cmp_write_str32(ctx, data, size); } bool cmp_write_str_v4(cmp_ctx_t *ctx, const char *data, uint32_t size) { if (size <= FIXSTR_SIZE) return cmp_write_fixstr(ctx, data, (uint8_t)size); if (size <= 0xFFFF) return cmp_write_str16(ctx, data, (uint16_t)size); return cmp_write_str32(ctx, data, size); } bool cmp_write_bin8_marker(cmp_ctx_t *ctx, uint8_t size) { if (!write_type_marker(ctx, BIN8_MARKER)) return false; if (ctx->write(ctx, &size, sizeof(uint8_t)) == sizeof(uint8_t)) return true; ctx->error = CMP_ERROR_LENGTH_WRITING; return false; } bool cmp_write_bin8(cmp_ctx_t *ctx, const void *data, uint8_t size) { if (!cmp_write_bin8_marker(ctx, size)) return false; if (size == 0) return true; if (ctx->write(ctx, data, size) == size) return true; ctx->error = CMP_ERROR_DATA_WRITING; return false; } bool cmp_write_bin16_marker(cmp_ctx_t *ctx, uint16_t size) { if (!write_type_marker(ctx, BIN16_MARKER)) return false; size = be16(size); if (ctx->write(ctx, &size, sizeof(uint16_t)) == sizeof(uint16_t)) return true; ctx->error = CMP_ERROR_LENGTH_WRITING; return false; } bool cmp_write_bin16(cmp_ctx_t *ctx, const void *data, uint16_t size) { if (!cmp_write_bin16_marker(ctx, size)) return false; if (size == 0) return true; if (ctx->write(ctx, data, size) == size) return true; ctx->error = CMP_ERROR_DATA_WRITING; return false; } bool cmp_write_bin32_marker(cmp_ctx_t *ctx, uint32_t size) { if (!write_type_marker(ctx, BIN32_MARKER)) return false; size = be32(size); if (ctx->write(ctx, &size, sizeof(uint32_t)) == sizeof(uint32_t)) return true; ctx->error = CMP_ERROR_LENGTH_WRITING; return false; } bool cmp_write_bin32(cmp_ctx_t *ctx, const void *data, uint32_t size) { if (!cmp_write_bin32_marker(ctx, size)) return false; if (size == 0) return true; if (ctx->write(ctx, data, size) == size) return true; ctx->error = CMP_ERROR_DATA_WRITING; return false; } bool cmp_write_bin_marker(cmp_ctx_t *ctx, uint32_t size) { if (size <= 0xFF) return cmp_write_bin8_marker(ctx, (uint8_t)size); if (size <= 0xFFFF) return cmp_write_bin16_marker(ctx, (uint16_t)size); return cmp_write_bin32_marker(ctx, size); } bool cmp_write_bin(cmp_ctx_t *ctx, const void *data, uint32_t size) { if (size <= 0xFF) return cmp_write_bin8(ctx, data, (uint8_t)size); if (size <= 0xFFFF) return cmp_write_bin16(ctx, data, (uint16_t)size); return cmp_write_bin32(ctx, data, size); } bool cmp_write_fixarray(cmp_ctx_t *ctx, uint8_t size) { if (size <= FIXARRAY_SIZE) return write_fixed_value(ctx, FIXARRAY_MARKER | size); ctx->error = CMP_ERROR_INPUT_VALUE_TOO_LARGE; return false; } bool cmp_write_array16(cmp_ctx_t *ctx, uint16_t size) { if (!write_type_marker(ctx, ARRAY16_MARKER)) return false; size = be16(size); if (ctx->write(ctx, &size, sizeof(uint16_t)) == sizeof(uint16_t)) return true; ctx->error = CMP_ERROR_LENGTH_WRITING; return false; } bool cmp_write_array32(cmp_ctx_t *ctx, uint32_t size) { if (!write_type_marker(ctx, ARRAY32_MARKER)) return false; size = be32(size); if (ctx->write(ctx, &size, sizeof(uint32_t)) == sizeof(uint32_t)) return true; ctx->error = CMP_ERROR_LENGTH_WRITING; return false; } bool cmp_write_array(cmp_ctx_t *ctx, uint32_t size) { if (size <= FIXARRAY_SIZE) return cmp_write_fixarray(ctx, (uint8_t)size); if (size <= 0xFFFF) return cmp_write_array16(ctx, (uint16_t)size); return cmp_write_array32(ctx, size); } bool cmp_write_fixmap(cmp_ctx_t *ctx, uint8_t size) { if (size <= FIXMAP_SIZE) return write_fixed_value(ctx, FIXMAP_MARKER | size); ctx->error = CMP_ERROR_INPUT_VALUE_TOO_LARGE; return false; } bool cmp_write_map16(cmp_ctx_t *ctx, uint16_t size) { if (!write_type_marker(ctx, MAP16_MARKER)) return false; size = be16(size); if (ctx->write(ctx, &size, sizeof(uint16_t)) == sizeof(uint16_t)) return true; ctx->error = CMP_ERROR_LENGTH_WRITING; return false; } bool cmp_write_map32(cmp_ctx_t *ctx, uint32_t size) { if (!write_type_marker(ctx, MAP32_MARKER)) return false; size = be32(size); if (ctx->write(ctx, &size, sizeof(uint32_t)) == sizeof(uint32_t)) return true; ctx->error = CMP_ERROR_LENGTH_WRITING; return false; } bool cmp_write_map(cmp_ctx_t *ctx, uint32_t size) { if (size <= FIXMAP_SIZE) return cmp_write_fixmap(ctx, (uint8_t)size); if (size <= 0xFFFF) return cmp_write_map16(ctx, (uint16_t)size); return cmp_write_map32(ctx, size); } bool cmp_write_fixext1_marker(cmp_ctx_t *ctx, int8_t type) { if (!write_type_marker(ctx, FIXEXT1_MARKER)) return false; if (ctx->write(ctx, &type, sizeof(int8_t)) == sizeof(int8_t)) return true; ctx->error = CMP_ERROR_EXT_TYPE_WRITING; return false; } bool cmp_write_fixext1(cmp_ctx_t *ctx, int8_t type, const void *data) { if (!cmp_write_fixext1_marker(ctx, type)) return false; if (ctx->write(ctx, data, 1) == 1) return true; ctx->error = CMP_ERROR_DATA_WRITING; return false; } bool cmp_write_fixext2_marker(cmp_ctx_t *ctx, int8_t type) { if (!write_type_marker(ctx, FIXEXT2_MARKER)) return false; if (ctx->write(ctx, &type, sizeof(int8_t)) == sizeof(int8_t)) return true; ctx->error = CMP_ERROR_EXT_TYPE_WRITING; return false; } bool cmp_write_fixext2(cmp_ctx_t *ctx, int8_t type, const void *data) { if (!cmp_write_fixext2_marker(ctx, type)) return false; if (ctx->write(ctx, data, 2) == 2) return true; ctx->error = CMP_ERROR_DATA_WRITING; return false; } bool cmp_write_fixext4_marker(cmp_ctx_t *ctx, int8_t type) { if (!write_type_marker(ctx, FIXEXT4_MARKER)) return false; if (ctx->write(ctx, &type, sizeof(int8_t)) == sizeof(int8_t)) return true; ctx->error = CMP_ERROR_EXT_TYPE_WRITING; return false; } bool cmp_write_fixext4(cmp_ctx_t *ctx, int8_t type, const void *data) { if (!cmp_write_fixext4_marker(ctx, type)) return false; if (ctx->write(ctx, data, 4) == 4) return true; ctx->error = CMP_ERROR_DATA_WRITING; return false; } bool cmp_write_fixext8_marker(cmp_ctx_t *ctx, int8_t type) { if (!write_type_marker(ctx, FIXEXT8_MARKER)) return false; if (ctx->write(ctx, &type, sizeof(int8_t)) == sizeof(int8_t)) return true; ctx->error = CMP_ERROR_EXT_TYPE_WRITING; return false; } bool cmp_write_fixext8(cmp_ctx_t *ctx, int8_t type, const void *data) { if (!cmp_write_fixext8_marker(ctx, type)) return false; if (ctx->write(ctx, data, 8) == 8) return true; ctx->error = CMP_ERROR_DATA_WRITING; return false; } bool cmp_write_fixext16_marker(cmp_ctx_t *ctx, int8_t type) { if (!write_type_marker(ctx, FIXEXT16_MARKER)) return false; if (ctx->write(ctx, &type, sizeof(int8_t)) == sizeof(int8_t)) return true; ctx->error = CMP_ERROR_EXT_TYPE_WRITING; return false; } bool cmp_write_fixext16(cmp_ctx_t *ctx, int8_t type, const void *data) { if (!cmp_write_fixext16_marker(ctx, type)) return false; if (ctx->write(ctx, data, 16) == 16) return true; ctx->error = CMP_ERROR_DATA_WRITING; return false; } bool cmp_write_ext8_marker(cmp_ctx_t *ctx, int8_t type, uint8_t size) { if (!write_type_marker(ctx, EXT8_MARKER)) return false; if (ctx->write(ctx, &size, sizeof(uint8_t)) != sizeof(uint8_t)) { ctx->error = CMP_ERROR_LENGTH_WRITING; return false; } if (ctx->write(ctx, &type, sizeof(int8_t)) == sizeof(int8_t)) return true; ctx->error = CMP_ERROR_EXT_TYPE_WRITING; return false; } bool cmp_write_ext8(cmp_ctx_t *ctx, int8_t type, uint8_t size, const void *data) { if (!cmp_write_ext8_marker(ctx, type, size)) return false; if (ctx->write(ctx, data, size) == size) return true; ctx->error = CMP_ERROR_DATA_WRITING; return false; } bool cmp_write_ext16_marker(cmp_ctx_t *ctx, int8_t type, uint16_t size) { if (!write_type_marker(ctx, EXT16_MARKER)) return false; size = be16(size); if (ctx->write(ctx, &size, sizeof(uint16_t)) != sizeof(uint16_t)) { ctx->error = CMP_ERROR_LENGTH_WRITING; return false; } if (ctx->write(ctx, &type, sizeof(int8_t)) == sizeof(int8_t)) return true; ctx->error = CMP_ERROR_EXT_TYPE_WRITING; return false; } bool cmp_write_ext16(cmp_ctx_t *ctx, int8_t type, uint16_t size, const void *data) { if (!cmp_write_ext16_marker(ctx, type, size)) return false; if (ctx->write(ctx, data, size) == size) return true; ctx->error = CMP_ERROR_DATA_WRITING; return false; } bool cmp_write_ext32_marker(cmp_ctx_t *ctx, int8_t type, uint32_t size) { if (!write_type_marker(ctx, EXT32_MARKER)) return false; size = be32(size); if (ctx->write(ctx, &size, sizeof(uint32_t)) != sizeof(uint32_t)) { ctx->error = CMP_ERROR_LENGTH_WRITING; return false; } if (ctx->write(ctx, &type, sizeof(int8_t)) == sizeof(int8_t)) return true; ctx->error = CMP_ERROR_EXT_TYPE_WRITING; return false; } bool cmp_write_ext32(cmp_ctx_t *ctx, int8_t type, uint32_t size, const void *data) { if (!cmp_write_ext32_marker(ctx, type, size)) return false; if (ctx->write(ctx, data, size) == size) return true; ctx->error = CMP_ERROR_DATA_WRITING; return false; } bool cmp_write_ext_marker(cmp_ctx_t *ctx, int8_t type, uint32_t size) { if (size == 1) return cmp_write_fixext1_marker(ctx, type); if (size == 2) return cmp_write_fixext2_marker(ctx, type); if (size == 4) return cmp_write_fixext4_marker(ctx, type); if (size == 8) return cmp_write_fixext8_marker(ctx, type); if (size == 16) return cmp_write_fixext16_marker(ctx, type); if (size <= 0xFF) return cmp_write_ext8_marker(ctx, type, (uint8_t)size); if (size <= 0xFFFF) return cmp_write_ext16_marker(ctx, type, (uint16_t)size); return cmp_write_ext32_marker(ctx, type, size); } bool cmp_write_ext(cmp_ctx_t *ctx, int8_t type, uint32_t size, const void *data) { if (size == 1) return cmp_write_fixext1(ctx, type, data); if (size == 2) return cmp_write_fixext2(ctx, type, data); if (size == 4) return cmp_write_fixext4(ctx, type, data); if (size == 8) return cmp_write_fixext8(ctx, type, data); if (size == 16) return cmp_write_fixext16(ctx, type, data); if (size <= 0xFF) return cmp_write_ext8(ctx, type, (uint8_t)size, data); if (size <= 0xFFFF) return cmp_write_ext16(ctx, type, (uint16_t)size, data); return cmp_write_ext32(ctx, type, size, data); } bool cmp_write_object(cmp_ctx_t *ctx, const cmp_object_t *obj) { switch(obj->type) { case CMP_TYPE_POSITIVE_FIXNUM: return cmp_write_pfix(ctx, obj->as.u8); case CMP_TYPE_FIXMAP: return cmp_write_fixmap(ctx, (uint8_t)obj->as.map_size); case CMP_TYPE_FIXARRAY: return cmp_write_fixarray(ctx, (uint8_t)obj->as.array_size); case CMP_TYPE_FIXSTR: return cmp_write_fixstr_marker(ctx, (uint8_t)obj->as.str_size); case CMP_TYPE_NIL: return cmp_write_nil(ctx); case CMP_TYPE_BOOLEAN: if (obj->as.boolean) return cmp_write_true(ctx); return cmp_write_false(ctx); case CMP_TYPE_BIN8: return cmp_write_bin8_marker(ctx, (uint8_t)obj->as.bin_size); case CMP_TYPE_BIN16: return cmp_write_bin16_marker(ctx, (uint16_t)obj->as.bin_size); case CMP_TYPE_BIN32: return cmp_write_bin32_marker(ctx, obj->as.bin_size); case CMP_TYPE_EXT8: return cmp_write_ext8_marker( ctx, obj->as.ext.type, (uint8_t)obj->as.ext.size ); case CMP_TYPE_EXT16: return cmp_write_ext16_marker( ctx, obj->as.ext.type, (uint16_t)obj->as.ext.size ); case CMP_TYPE_EXT32: return cmp_write_ext32_marker(ctx, obj->as.ext.type, obj->as.ext.size); case CMP_TYPE_FLOAT: #ifndef CMP_NO_FLOAT return cmp_write_float(ctx, obj->as.flt); #else /* CMP_NO_FLOAT */ ctx->error = CMP_ERROR_DISABLED_FLOATING_POINT; return false; #endif /* CMP_NO_FLOAT */ case CMP_TYPE_DOUBLE: #ifndef CMP_NO_FLOAT return cmp_write_double(ctx, obj->as.dbl); #else /* CMP_NO_FLOAT */ ctx->error = CMP_ERROR_DISABLED_FLOATING_POINT; return false; #endif case CMP_TYPE_UINT8: return cmp_write_u8(ctx, obj->as.u8); case CMP_TYPE_UINT16: return cmp_write_u16(ctx, obj->as.u16); case CMP_TYPE_UINT32: return cmp_write_u32(ctx, obj->as.u32); case CMP_TYPE_UINT64: return cmp_write_u64(ctx, obj->as.u64); case CMP_TYPE_SINT8: return cmp_write_s8(ctx, obj->as.s8); case CMP_TYPE_SINT16: return cmp_write_s16(ctx, obj->as.s16); case CMP_TYPE_SINT32: return cmp_write_s32(ctx, obj->as.s32); case CMP_TYPE_SINT64: return cmp_write_s64(ctx, obj->as.s64); case CMP_TYPE_FIXEXT1: return cmp_write_fixext1_marker(ctx, obj->as.ext.type); case CMP_TYPE_FIXEXT2: return cmp_write_fixext2_marker(ctx, obj->as.ext.type); case CMP_TYPE_FIXEXT4: return cmp_write_fixext4_marker(ctx, obj->as.ext.type); case CMP_TYPE_FIXEXT8: return cmp_write_fixext8_marker(ctx, obj->as.ext.type); case CMP_TYPE_FIXEXT16: return cmp_write_fixext16_marker(ctx, obj->as.ext.type); case CMP_TYPE_STR8: return cmp_write_str8_marker(ctx, (uint8_t)obj->as.str_size); case CMP_TYPE_STR16: return cmp_write_str16_marker(ctx, (uint16_t)obj->as.str_size); case CMP_TYPE_STR32: return cmp_write_str32_marker(ctx, obj->as.str_size); case CMP_TYPE_ARRAY16: return cmp_write_array16(ctx, (uint16_t)obj->as.array_size); case CMP_TYPE_ARRAY32: return cmp_write_array32(ctx, obj->as.array_size); case CMP_TYPE_MAP16: return cmp_write_map16(ctx, (uint16_t)obj->as.map_size); case CMP_TYPE_MAP32: return cmp_write_map32(ctx, obj->as.map_size); case CMP_TYPE_NEGATIVE_FIXNUM: return cmp_write_nfix(ctx, obj->as.s8); default: ctx->error = CMP_ERROR_INVALID_TYPE; return false; } } bool cmp_write_object_v4(cmp_ctx_t *ctx, const cmp_object_t *obj) { switch(obj->type) { case CMP_TYPE_POSITIVE_FIXNUM: return cmp_write_pfix(ctx, obj->as.u8); case CMP_TYPE_FIXMAP: return cmp_write_fixmap(ctx, (uint8_t)obj->as.map_size); case CMP_TYPE_FIXARRAY: return cmp_write_fixarray(ctx, (uint8_t)obj->as.array_size); case CMP_TYPE_FIXSTR: return cmp_write_fixstr_marker(ctx, (uint8_t)obj->as.str_size); case CMP_TYPE_NIL: return cmp_write_nil(ctx); case CMP_TYPE_BOOLEAN: if (obj->as.boolean) return cmp_write_true(ctx); return cmp_write_false(ctx); case CMP_TYPE_EXT8: return cmp_write_ext8_marker(ctx, obj->as.ext.type, (uint8_t)obj->as.ext.size); case CMP_TYPE_EXT16: return cmp_write_ext16_marker( ctx, obj->as.ext.type, (uint16_t)obj->as.ext.size ); case CMP_TYPE_EXT32: return cmp_write_ext32_marker(ctx, obj->as.ext.type, obj->as.ext.size); case CMP_TYPE_FLOAT: #ifndef CMP_NO_FLOAT return cmp_write_float(ctx, obj->as.flt); #else /* CMP_NO_FLOAT */ ctx->error = CMP_ERROR_DISABLED_FLOATING_POINT; return false; #endif case CMP_TYPE_DOUBLE: #ifndef CMP_NO_FLOAT return cmp_write_double(ctx, obj->as.dbl); #else ctx->error = CMP_ERROR_DISABLED_FLOATING_POINT; return false; #endif case CMP_TYPE_UINT8: return cmp_write_u8(ctx, obj->as.u8); case CMP_TYPE_UINT16: return cmp_write_u16(ctx, obj->as.u16); case CMP_TYPE_UINT32: return cmp_write_u32(ctx, obj->as.u32); case CMP_TYPE_UINT64: return cmp_write_u64(ctx, obj->as.u64); case CMP_TYPE_SINT8: return cmp_write_s8(ctx, obj->as.s8); case CMP_TYPE_SINT16: return cmp_write_s16(ctx, obj->as.s16); case CMP_TYPE_SINT32: return cmp_write_s32(ctx, obj->as.s32); case CMP_TYPE_SINT64: return cmp_write_s64(ctx, obj->as.s64); case CMP_TYPE_FIXEXT1: return cmp_write_fixext1_marker(ctx, obj->as.ext.type); case CMP_TYPE_FIXEXT2: return cmp_write_fixext2_marker(ctx, obj->as.ext.type); case CMP_TYPE_FIXEXT4: return cmp_write_fixext4_marker(ctx, obj->as.ext.type); case CMP_TYPE_FIXEXT8: return cmp_write_fixext8_marker(ctx, obj->as.ext.type); case CMP_TYPE_FIXEXT16: return cmp_write_fixext16_marker(ctx, obj->as.ext.type); case CMP_TYPE_STR16: return cmp_write_str16_marker(ctx, (uint16_t)obj->as.str_size); case CMP_TYPE_STR32: return cmp_write_str32_marker(ctx, obj->as.str_size); case CMP_TYPE_ARRAY16: return cmp_write_array16(ctx, (uint16_t)obj->as.array_size); case CMP_TYPE_ARRAY32: return cmp_write_array32(ctx, obj->as.array_size); case CMP_TYPE_MAP16: return cmp_write_map16(ctx, (uint16_t)obj->as.map_size); case CMP_TYPE_MAP32: return cmp_write_map32(ctx, obj->as.map_size); case CMP_TYPE_NEGATIVE_FIXNUM: return cmp_write_nfix(ctx, obj->as.s8); default: ctx->error = CMP_ERROR_INVALID_TYPE; return false; } } bool cmp_read_pfix(cmp_ctx_t *ctx, uint8_t *c) { cmp_object_t obj; if (!cmp_read_object(ctx, &obj)) return false; if (obj.type != CMP_TYPE_POSITIVE_FIXNUM) { ctx->error = CMP_ERROR_INVALID_TYPE; return false; } *c = obj.as.u8; return true; } bool cmp_read_nfix(cmp_ctx_t *ctx, int8_t *c) { cmp_object_t obj; if (!cmp_read_object(ctx, &obj)) return false; if (obj.type != CMP_TYPE_NEGATIVE_FIXNUM) { ctx->error = CMP_ERROR_INVALID_TYPE; return false; } *c = obj.as.s8; return true; } bool cmp_read_sfix(cmp_ctx_t *ctx, int8_t *c) { cmp_object_t obj; if (!cmp_read_object(ctx, &obj)) return false; switch (obj.type) { case CMP_TYPE_POSITIVE_FIXNUM: case CMP_TYPE_NEGATIVE_FIXNUM: *c = obj.as.s8; return true; default: ctx->error = CMP_ERROR_INVALID_TYPE; return false; } } bool cmp_read_s8(cmp_ctx_t *ctx, int8_t *c) { cmp_object_t obj; if (!cmp_read_object(ctx, &obj)) return false; if (obj.type != CMP_TYPE_SINT8) { ctx->error = CMP_ERROR_INVALID_TYPE; return false; } *c = obj.as.s8; return true; } bool cmp_read_s16(cmp_ctx_t *ctx, int16_t *s) { cmp_object_t obj; if (!cmp_read_object(ctx, &obj)) return false; if (obj.type != CMP_TYPE_SINT16) { ctx->error = CMP_ERROR_INVALID_TYPE; return false; } *s = obj.as.s16; return true; } bool cmp_read_s32(cmp_ctx_t *ctx, int32_t *i) { cmp_object_t obj; if (!cmp_read_object(ctx, &obj)) return false; if (obj.type != CMP_TYPE_SINT32) { ctx->error = CMP_ERROR_INVALID_TYPE; return false; } *i = obj.as.s32; return true; } bool cmp_read_s64(cmp_ctx_t *ctx, int64_t *l) { cmp_object_t obj; if (!cmp_read_object(ctx, &obj)) return false; if (obj.type != CMP_TYPE_SINT64) { ctx->error = CMP_ERROR_INVALID_TYPE; return false; } *l = obj.as.s64; return true; } bool cmp_read_char(cmp_ctx_t *ctx, int8_t *c) { cmp_object_t obj; if (!cmp_read_object(ctx, &obj)) return false; switch (obj.type) { case CMP_TYPE_POSITIVE_FIXNUM: case CMP_TYPE_NEGATIVE_FIXNUM: case CMP_TYPE_SINT8: *c = obj.as.s8; return true; case CMP_TYPE_UINT8: if (obj.as.u8 <= 0x7F) { *c = (int8_t)obj.as.u8; return true; } break; default: break; } ctx->error = CMP_ERROR_INVALID_TYPE; return false; } bool cmp_read_short(cmp_ctx_t *ctx, int16_t *s) { cmp_object_t obj; if (!cmp_read_object(ctx, &obj)) return false; switch (obj.type) { case CMP_TYPE_POSITIVE_FIXNUM: case CMP_TYPE_NEGATIVE_FIXNUM: case CMP_TYPE_SINT8: *s = obj.as.s8; return true; case CMP_TYPE_UINT8: *s = obj.as.u8; return true; case CMP_TYPE_SINT16: *s = obj.as.s16; return true; case CMP_TYPE_UINT16: if (obj.as.u16 <= 0x7FFF) { *s = (int16_t)obj.as.u16; return true; } break; default: break; } ctx->error = CMP_ERROR_INVALID_TYPE; return false; } bool cmp_read_int(cmp_ctx_t *ctx, int32_t *i) { cmp_object_t obj; if (!cmp_read_object(ctx, &obj)) return false; switch (obj.type) { case CMP_TYPE_POSITIVE_FIXNUM: case CMP_TYPE_NEGATIVE_FIXNUM: case CMP_TYPE_SINT8: *i = obj.as.s8; return true; case CMP_TYPE_UINT8: *i = obj.as.u8; return true; case CMP_TYPE_SINT16: *i = obj.as.s16; return true; case CMP_TYPE_UINT16: *i = obj.as.u16; return true; case CMP_TYPE_SINT32: *i = obj.as.s32; return true; case CMP_TYPE_UINT32: if (obj.as.u32 <= 0x7FFFFFFF) { *i = (int32_t)obj.as.u32; return true; } break; default: break; } ctx->error = CMP_ERROR_INVALID_TYPE; return false; } bool cmp_read_long(cmp_ctx_t *ctx, int64_t *d) { cmp_object_t obj; if (!cmp_read_object(ctx, &obj)) return false; switch (obj.type) { case CMP_TYPE_POSITIVE_FIXNUM: case CMP_TYPE_NEGATIVE_FIXNUM: case CMP_TYPE_SINT8: *d = obj.as.s8; return true; case CMP_TYPE_UINT8: *d = obj.as.u8; return true; case CMP_TYPE_SINT16: *d = obj.as.s16; return true; case CMP_TYPE_UINT16: *d = obj.as.u16; return true; case CMP_TYPE_SINT32: *d = obj.as.s32; return true; case CMP_TYPE_UINT32: *d = obj.as.u32; return true; case CMP_TYPE_SINT64: *d = obj.as.s64; return true; case CMP_TYPE_UINT64: if (obj.as.u64 <= UINT64_C(0x7FFFFFFFFFFFFFFF)) { *d = (int64_t)obj.as.u64; return true; } break; default: break; } ctx->error = CMP_ERROR_INVALID_TYPE; return false; } bool cmp_read_integer(cmp_ctx_t *ctx, int64_t *d) { return cmp_read_long(ctx, d); } bool cmp_read_ufix(cmp_ctx_t *ctx, uint8_t *c) { return cmp_read_pfix(ctx, c); } bool cmp_read_u8(cmp_ctx_t *ctx, uint8_t *c) { cmp_object_t obj; if (!cmp_read_object(ctx, &obj)) return false; if (obj.type != CMP_TYPE_UINT8) { ctx->error = CMP_ERROR_INVALID_TYPE; return false; } *c = obj.as.u8; return true; } bool cmp_read_u16(cmp_ctx_t *ctx, uint16_t *s) { cmp_object_t obj; if (!cmp_read_object(ctx, &obj)) return false; if (obj.type != CMP_TYPE_UINT16) { ctx->error = CMP_ERROR_INVALID_TYPE; return false; } *s = obj.as.u16; return true; } bool cmp_read_u32(cmp_ctx_t *ctx, uint32_t *i) { cmp_object_t obj; if (!cmp_read_object(ctx, &obj)) return false; if (obj.type != CMP_TYPE_UINT32) { ctx->error = CMP_ERROR_INVALID_TYPE; return false; } *i = obj.as.u32; return true; } bool cmp_read_u64(cmp_ctx_t *ctx, uint64_t *l) { cmp_object_t obj; if (!cmp_read_object(ctx, &obj)) return false; if (obj.type != CMP_TYPE_UINT64) { ctx->error = CMP_ERROR_INVALID_TYPE; return false; } *l = obj.as.u64; return true; } bool cmp_read_uchar(cmp_ctx_t *ctx, uint8_t *c) { cmp_object_t obj; if (!cmp_read_object(ctx, &obj)) return false; switch (obj.type) { case CMP_TYPE_POSITIVE_FIXNUM: case CMP_TYPE_UINT8: *c = obj.as.u8; return true; case CMP_TYPE_NEGATIVE_FIXNUM: case CMP_TYPE_SINT8: if (obj.as.s8 >= 0) { *c = (uint8_t)obj.as.s8; return true; } break; default: break; } ctx->error = CMP_ERROR_INVALID_TYPE; return false; } bool cmp_read_ushort(cmp_ctx_t *ctx, uint16_t *s) { cmp_object_t obj; if (!cmp_read_object(ctx, &obj)) return false; switch (obj.type) { case CMP_TYPE_POSITIVE_FIXNUM: case CMP_TYPE_UINT8: *s = obj.as.u8; return true; case CMP_TYPE_UINT16: *s = obj.as.u16; return true; case CMP_TYPE_NEGATIVE_FIXNUM: case CMP_TYPE_SINT8: if (obj.as.s8 >= 0) { *s = (uint8_t)obj.as.s8; return true; } break; case CMP_TYPE_SINT16: if (obj.as.s16 >= 0) { *s = (uint16_t)obj.as.s16; return true; } break; default: break; } ctx->error = CMP_ERROR_INVALID_TYPE; return false; } bool cmp_read_uint(cmp_ctx_t *ctx, uint32_t *i) { cmp_object_t obj; if (!cmp_read_object(ctx, &obj)) return false; switch (obj.type) { case CMP_TYPE_POSITIVE_FIXNUM: case CMP_TYPE_UINT8: *i = obj.as.u8; return true; case CMP_TYPE_UINT16: *i = obj.as.u16; return true; case CMP_TYPE_UINT32: *i = obj.as.u32; return true; case CMP_TYPE_NEGATIVE_FIXNUM: case CMP_TYPE_SINT8: if (obj.as.s8 >= 0) { *i = (uint8_t)obj.as.s8; return true; } break; case CMP_TYPE_SINT16: if (obj.as.s16 >= 0) { *i = (uint16_t)obj.as.s16; return true; } break; case CMP_TYPE_SINT32: if (obj.as.s32 >= 0) { *i = (uint32_t)obj.as.s32; return true; } break; default: break; } ctx->error = CMP_ERROR_INVALID_TYPE; return false; } bool cmp_read_ulong(cmp_ctx_t *ctx, uint64_t *u) { cmp_object_t obj; if (!cmp_read_object(ctx, &obj)) return false; switch (obj.type) { case CMP_TYPE_POSITIVE_FIXNUM: case CMP_TYPE_UINT8: *u = obj.as.u8; return true; case CMP_TYPE_UINT16: *u = obj.as.u16; return true; case CMP_TYPE_UINT32: *u = obj.as.u32; return true; case CMP_TYPE_UINT64: *u = obj.as.u64; return true; case CMP_TYPE_NEGATIVE_FIXNUM: case CMP_TYPE_SINT8: if (obj.as.s8 >= 0) { *u = (uint8_t)obj.as.s8; return true; } break; case CMP_TYPE_SINT16: if (obj.as.s16 >= 0) { *u = (uint16_t)obj.as.s16; return true; } break; case CMP_TYPE_SINT32: if (obj.as.s32 >= 0) { *u = (uint32_t)obj.as.s32; return true; } break; case CMP_TYPE_SINT64: if (obj.as.s64 >= 0) { *u = (uint64_t)obj.as.s64; return true; } break; default: break; } ctx->error = CMP_ERROR_INVALID_TYPE; return false; } bool cmp_read_uinteger(cmp_ctx_t *ctx, uint64_t *u) { return cmp_read_ulong(ctx, u); } #ifndef CMP_NO_FLOAT bool cmp_read_float(cmp_ctx_t *ctx, float *f) { cmp_object_t obj; if (!cmp_read_object(ctx, &obj)) return false; if (obj.type != CMP_TYPE_FLOAT) { ctx->error = CMP_ERROR_INVALID_TYPE; return false; } *f = obj.as.flt; return true; } bool cmp_read_double(cmp_ctx_t *ctx, double *d) { cmp_object_t obj; if (!cmp_read_object(ctx, &obj)) return false; if (obj.type != CMP_TYPE_DOUBLE) { ctx->error = CMP_ERROR_INVALID_TYPE; return false; } *d = obj.as.dbl; return true; } bool cmp_read_decimal(cmp_ctx_t *ctx, double *d) { cmp_object_t obj; if (!cmp_read_object(ctx, &obj)) return false; switch (obj.type) { case CMP_TYPE_FLOAT: *d = (double)obj.as.flt; return true; case CMP_TYPE_DOUBLE: *d = obj.as.dbl; return true; default: ctx->error = CMP_ERROR_INVALID_TYPE; return false; } } #endif /* CMP_NO_FLOAT */ bool cmp_read_nil(cmp_ctx_t *ctx) { cmp_object_t obj; if (!cmp_read_object(ctx, &obj)) return false; if (obj.type == CMP_TYPE_NIL) return true; ctx->error = CMP_ERROR_INVALID_TYPE; return false; } bool cmp_read_bool(cmp_ctx_t *ctx, bool *b) { cmp_object_t obj; if (!cmp_read_object(ctx, &obj)) return false; if (obj.type != CMP_TYPE_BOOLEAN) { ctx->error = CMP_ERROR_INVALID_TYPE; return false; } if (obj.as.boolean) *b = true; else *b = false; return true; } bool cmp_read_bool_as_u8(cmp_ctx_t *ctx, uint8_t *b) { cmp_object_t obj; if (!cmp_read_object(ctx, &obj)) return false; if (obj.type != CMP_TYPE_BOOLEAN) { ctx->error = CMP_ERROR_INVALID_TYPE; return false; } if (obj.as.boolean) *b = 1; else *b = 0; return true; } bool cmp_read_str_size(cmp_ctx_t *ctx, uint32_t *size) { cmp_object_t obj; if (!cmp_read_object(ctx, &obj)) return false; switch (obj.type) { case CMP_TYPE_FIXSTR: case CMP_TYPE_STR8: case CMP_TYPE_STR16: case CMP_TYPE_STR32: *size = obj.as.str_size; return true; default: ctx->error = CMP_ERROR_INVALID_TYPE; return false; } } bool cmp_read_str(cmp_ctx_t *ctx, char *data, uint32_t *size) { uint32_t str_size = 0; if (!cmp_read_str_size(ctx, &str_size)) return false; if (str_size >= *size) { *size = str_size; ctx->error = CMP_ERROR_STR_DATA_LENGTH_TOO_LONG; return false; } if (!ctx->read(ctx, data, str_size)) { ctx->error = CMP_ERROR_DATA_READING; return false; } data[str_size] = 0; *size = str_size; return true; } bool cmp_read_bin_size(cmp_ctx_t *ctx, uint32_t *size) { cmp_object_t obj; if (!cmp_read_object(ctx, &obj)) return false; switch (obj.type) { case CMP_TYPE_BIN8: case CMP_TYPE_BIN16: case CMP_TYPE_BIN32: *size = obj.as.bin_size; return true; default: ctx->error = CMP_ERROR_INVALID_TYPE; return false; } } bool cmp_read_bin(cmp_ctx_t *ctx, void *data, uint32_t *size) { uint32_t bin_size = 0; if (!cmp_read_bin_size(ctx, &bin_size)) return false; if (bin_size > *size) { ctx->error = CMP_ERROR_BIN_DATA_LENGTH_TOO_LONG; return false; } if (!ctx->read(ctx, data, bin_size)) { ctx->error = CMP_ERROR_DATA_READING; return false; } *size = bin_size; return true; } bool cmp_read_array(cmp_ctx_t *ctx, uint32_t *size) { cmp_object_t obj; if (!cmp_read_object(ctx, &obj)) return false; switch (obj.type) { case CMP_TYPE_FIXARRAY: case CMP_TYPE_ARRAY16: case CMP_TYPE_ARRAY32: *size = obj.as.array_size; return true; default: ctx->error = CMP_ERROR_INVALID_TYPE; return false; } } bool cmp_read_map(cmp_ctx_t *ctx, uint32_t *size) { cmp_object_t obj; if (!cmp_read_object(ctx, &obj)) return false; switch (obj.type) { case CMP_TYPE_FIXMAP: case CMP_TYPE_MAP16: case CMP_TYPE_MAP32: *size = obj.as.map_size; return true; default: ctx->error = CMP_ERROR_INVALID_TYPE; return false; } } bool cmp_read_fixext1_marker(cmp_ctx_t *ctx, int8_t *type) { cmp_object_t obj; if (!cmp_read_object(ctx, &obj)) return false; if (obj.type != CMP_TYPE_FIXEXT1) { ctx->error = CMP_ERROR_INVALID_TYPE; return false; } *type = obj.as.ext.type; return true; } bool cmp_read_fixext1(cmp_ctx_t *ctx, int8_t *type, void *data) { if (!cmp_read_fixext1_marker(ctx, type)) return false; if (ctx->read(ctx, data, 1)) return true; ctx->error = CMP_ERROR_DATA_READING; return false; } bool cmp_read_fixext2_marker(cmp_ctx_t *ctx, int8_t *type) { cmp_object_t obj; if (!cmp_read_object(ctx, &obj)) return false; if (obj.type != CMP_TYPE_FIXEXT2) { ctx->error = CMP_ERROR_INVALID_TYPE; return false; } *type = obj.as.ext.type; return true; } bool cmp_read_fixext2(cmp_ctx_t *ctx, int8_t *type, void *data) { if (!cmp_read_fixext2_marker(ctx, type)) return false; if (ctx->read(ctx, data, 2)) return true; ctx->error = CMP_ERROR_DATA_READING; return false; } bool cmp_read_fixext4_marker(cmp_ctx_t *ctx, int8_t *type) { cmp_object_t obj; if (!cmp_read_object(ctx, &obj)) return false; if (obj.type != CMP_TYPE_FIXEXT4) { ctx->error = CMP_ERROR_INVALID_TYPE; return false; } *type = obj.as.ext.type; return true; } bool cmp_read_fixext4(cmp_ctx_t *ctx, int8_t *type, void *data) { if (!cmp_read_fixext4_marker(ctx, type)) return false; if (ctx->read(ctx, data, 4)) return true; ctx->error = CMP_ERROR_DATA_READING; return false; } bool cmp_read_fixext8_marker(cmp_ctx_t *ctx, int8_t *type) { cmp_object_t obj; if (!cmp_read_object(ctx, &obj)) return false; if (obj.type != CMP_TYPE_FIXEXT8) { ctx->error = CMP_ERROR_INVALID_TYPE; return false; } *type = obj.as.ext.type; return true; } bool cmp_read_fixext8(cmp_ctx_t *ctx, int8_t *type, void *data) { if (!cmp_read_fixext8_marker(ctx, type)) return false; if (ctx->read(ctx, data, 8)) return true; ctx->error = CMP_ERROR_DATA_READING; return false; } bool cmp_read_fixext16_marker(cmp_ctx_t *ctx, int8_t *type) { cmp_object_t obj; if (!cmp_read_object(ctx, &obj)) return false; if (obj.type != CMP_TYPE_FIXEXT16) { ctx->error = CMP_ERROR_INVALID_TYPE; return false; } *type = obj.as.ext.type; return true; } bool cmp_read_fixext16(cmp_ctx_t *ctx, int8_t *type, void *data) { if (!cmp_read_fixext16_marker(ctx, type)) return false; if (ctx->read(ctx, data, 16)) return true; ctx->error = CMP_ERROR_DATA_READING; return false; } bool cmp_read_ext8_marker(cmp_ctx_t *ctx, int8_t *type, uint8_t *size) { cmp_object_t obj; if (!cmp_read_object(ctx, &obj)) return false; if (obj.type != CMP_TYPE_EXT8) { ctx->error = CMP_ERROR_INVALID_TYPE; return false; } *type = obj.as.ext.type; *size = (uint8_t)obj.as.ext.size; return true; } bool cmp_read_ext8(cmp_ctx_t *ctx, int8_t *type, uint8_t *size, void *data) { if (!cmp_read_ext8_marker(ctx, type, size)) return false; if (ctx->read(ctx, data, *size)) return true; ctx->error = CMP_ERROR_DATA_READING; return false; } bool cmp_read_ext16_marker(cmp_ctx_t *ctx, int8_t *type, uint16_t *size) { cmp_object_t obj; if (!cmp_read_object(ctx, &obj)) return false; if (obj.type != CMP_TYPE_EXT16) { ctx->error = CMP_ERROR_INVALID_TYPE; return false; } *type = obj.as.ext.type; *size = (uint16_t)obj.as.ext.size; return true; } bool cmp_read_ext16(cmp_ctx_t *ctx, int8_t *type, uint16_t *size, void *data) { if (!cmp_read_ext16_marker(ctx, type, size)) return false; if (ctx->read(ctx, data, *size)) return true; ctx->error = CMP_ERROR_DATA_READING; return false; } bool cmp_read_ext32_marker(cmp_ctx_t *ctx, int8_t *type, uint32_t *size) { cmp_object_t obj; if (!cmp_read_object(ctx, &obj)) return false; if (obj.type != CMP_TYPE_EXT32) { ctx->error = CMP_ERROR_INVALID_TYPE; return false; } *type = obj.as.ext.type; *size = obj.as.ext.size; return true; } bool cmp_read_ext32(cmp_ctx_t *ctx, int8_t *type, uint32_t *size, void *data) { if (!cmp_read_ext32_marker(ctx, type, size)) return false; if (ctx->read(ctx, data, *size)) return true; ctx->error = CMP_ERROR_DATA_READING; return false; } bool cmp_read_ext_marker(cmp_ctx_t *ctx, int8_t *type, uint32_t *size) { cmp_object_t obj; if (!cmp_read_object(ctx, &obj)) return false; switch (obj.type) { case CMP_TYPE_FIXEXT1: case CMP_TYPE_FIXEXT2: case CMP_TYPE_FIXEXT4: case CMP_TYPE_FIXEXT8: case CMP_TYPE_FIXEXT16: case CMP_TYPE_EXT8: case CMP_TYPE_EXT16: case CMP_TYPE_EXT32: *type = obj.as.ext.type; *size = obj.as.ext.size; return true; default: ctx->error = CMP_ERROR_INVALID_TYPE; return false; } } bool cmp_read_ext(cmp_ctx_t *ctx, int8_t *type, uint32_t *size, void *data) { if (!cmp_read_ext_marker(ctx, type, size)) return false; if (ctx->read(ctx, data, *size)) return true; ctx->error = CMP_ERROR_DATA_READING; return false; } bool cmp_read_object(cmp_ctx_t *ctx, cmp_object_t *obj) { uint8_t type_marker = 0; if (!read_type_marker(ctx, &type_marker)) return false; if (!type_marker_to_cmp_type(type_marker, &obj->type)) { ctx->error = CMP_ERROR_INVALID_TYPE; return false; } return read_obj_data(ctx, type_marker, obj); } bool cmp_skip_object(cmp_ctx_t *ctx, cmp_object_t *obj) { uint8_t type_marker = 0; uint8_t cmp_type; uint32_t size = 0; if (!read_type_marker(ctx, &type_marker)) { return false; } if (!type_marker_to_cmp_type(type_marker, &cmp_type)) { ctx->error = CMP_ERROR_INVALID_TYPE; return false; } switch (cmp_type) { case CMP_TYPE_FIXARRAY: case CMP_TYPE_ARRAY16: case CMP_TYPE_ARRAY32: case CMP_TYPE_FIXMAP: case CMP_TYPE_MAP16: case CMP_TYPE_MAP32: obj->type = cmp_type; if (!read_obj_data(ctx, type_marker, obj)) { return false; } ctx->error = CMP_ERROR_SKIP_DEPTH_LIMIT_EXCEEDED; return false; default: if (!read_type_size(ctx, type_marker, cmp_type, &size)) { return false; } if (size) { switch (cmp_type) { case CMP_TYPE_FIXEXT1: case CMP_TYPE_FIXEXT2: case CMP_TYPE_FIXEXT4: case CMP_TYPE_FIXEXT8: case CMP_TYPE_FIXEXT16: case CMP_TYPE_EXT8: case CMP_TYPE_EXT16: case CMP_TYPE_EXT32: ++size; break; default: break; } skip_bytes(ctx, size); } } return true; } bool cmp_skip_object_flat(cmp_ctx_t *ctx, cmp_object_t *obj) { size_t element_count = 1; bool in_container = false; while (element_count) { uint8_t type_marker = 0; uint8_t cmp_type; uint32_t size = 0; if (!read_type_marker(ctx, &type_marker)) { return false; } if (!type_marker_to_cmp_type(type_marker, &cmp_type)) { ctx->error = CMP_ERROR_INVALID_TYPE; return false; } switch (cmp_type) { case CMP_TYPE_FIXARRAY: case CMP_TYPE_ARRAY16: case CMP_TYPE_ARRAY32: case CMP_TYPE_FIXMAP: case CMP_TYPE_MAP16: case CMP_TYPE_MAP32: if (in_container) { obj->type = cmp_type; if (!read_obj_data(ctx, type_marker, obj)) { return false; } ctx->error = CMP_ERROR_SKIP_DEPTH_LIMIT_EXCEEDED; return false; } in_container = true; break; default: if (!read_type_size(ctx, type_marker, cmp_type, &size)) { return false; } if (size) { switch (cmp_type) { case CMP_TYPE_FIXEXT1: case CMP_TYPE_FIXEXT2: case CMP_TYPE_FIXEXT4: case CMP_TYPE_FIXEXT8: case CMP_TYPE_FIXEXT16: case CMP_TYPE_EXT8: case CMP_TYPE_EXT16: case CMP_TYPE_EXT32: ++size; break; default: break; } skip_bytes(ctx, size); } } element_count--; switch (cmp_type) { case CMP_TYPE_FIXARRAY: case CMP_TYPE_ARRAY16: case CMP_TYPE_ARRAY32: if (!read_type_size(ctx, type_marker, cmp_type, &size)) { return false; } element_count += size; break; case CMP_TYPE_FIXMAP: case CMP_TYPE_MAP16: case CMP_TYPE_MAP32: if (!read_type_size(ctx, type_marker, cmp_type, &size)) { return false; } element_count += ((size_t)size) * 2; break; default: break; } } return true; } bool cmp_skip_object_no_limit(cmp_ctx_t *ctx) { size_t element_count = 1; while (element_count) { uint8_t type_marker = 0; uint8_t cmp_type = 0; uint32_t size = 0; if (!read_type_marker(ctx, &type_marker)) { return false; } if (!type_marker_to_cmp_type(type_marker, &cmp_type)) { ctx->error = CMP_ERROR_INVALID_TYPE; return false; } switch (cmp_type) { case CMP_TYPE_FIXARRAY: case CMP_TYPE_ARRAY16: case CMP_TYPE_ARRAY32: case CMP_TYPE_FIXMAP: case CMP_TYPE_MAP16: case CMP_TYPE_MAP32: break; default: if (!read_type_size(ctx, type_marker, cmp_type, &size)) { return false; } if (size) { switch (cmp_type) { case CMP_TYPE_FIXEXT1: case CMP_TYPE_FIXEXT2: case CMP_TYPE_FIXEXT4: case CMP_TYPE_FIXEXT8: case CMP_TYPE_FIXEXT16: case CMP_TYPE_EXT8: case CMP_TYPE_EXT16: case CMP_TYPE_EXT32: ++size; break; default: break; } skip_bytes(ctx, size); } } element_count--; switch (cmp_type) { case CMP_TYPE_FIXARRAY: case CMP_TYPE_ARRAY16: case CMP_TYPE_ARRAY32: if (!read_type_size(ctx, type_marker, cmp_type, &size)) { return false; } element_count += size; break; case CMP_TYPE_FIXMAP: case CMP_TYPE_MAP16: case CMP_TYPE_MAP32: if (!read_type_size(ctx, type_marker, cmp_type, &size)) { return false; } element_count += ((size_t)size) * 2; break; default: break; } } return true; } bool cmp_skip_object_limit(cmp_ctx_t *ctx, cmp_object_t *obj, uint32_t limit) { size_t element_count = 1; uint32_t depth = 0; while (element_count) { uint8_t type_marker = 0; uint8_t cmp_type; uint32_t size = 0; if (!read_type_marker(ctx, &type_marker)) { return false; } if (!type_marker_to_cmp_type(type_marker, &cmp_type)) { ctx->error = CMP_ERROR_INVALID_TYPE; return false; } switch (cmp_type) { case CMP_TYPE_FIXARRAY: case CMP_TYPE_ARRAY16: case CMP_TYPE_ARRAY32: case CMP_TYPE_FIXMAP: case CMP_TYPE_MAP16: case CMP_TYPE_MAP32: ++depth; if (depth > limit) { obj->type = cmp_type; if (!read_obj_data(ctx, type_marker, obj)) { return false; } ctx->error = CMP_ERROR_SKIP_DEPTH_LIMIT_EXCEEDED; return false; } break; default: if (!read_type_size(ctx, type_marker, cmp_type, &size)) { return false; } if (size) { switch (cmp_type) { case CMP_TYPE_FIXEXT1: case CMP_TYPE_FIXEXT2: case CMP_TYPE_FIXEXT4: case CMP_TYPE_FIXEXT8: case CMP_TYPE_FIXEXT16: case CMP_TYPE_EXT8: case CMP_TYPE_EXT16: case CMP_TYPE_EXT32: ++size; break; default: break; } skip_bytes(ctx, size); } } element_count--; switch (cmp_type) { case CMP_TYPE_FIXARRAY: case CMP_TYPE_ARRAY16: case CMP_TYPE_ARRAY32: if (!read_type_size(ctx, type_marker, cmp_type, &size)) { return false; } element_count += size; break; case CMP_TYPE_FIXMAP: case CMP_TYPE_MAP16: case CMP_TYPE_MAP32: if (!read_type_size(ctx, type_marker, cmp_type, &size)) { return false; } element_count += ((size_t)size) * 2; break; default: break; } } return true; } bool cmp_object_is_char(const cmp_object_t *obj) { switch (obj->type) { case CMP_TYPE_NEGATIVE_FIXNUM: case CMP_TYPE_SINT8: return true; default: return false; } } bool cmp_object_is_short(const cmp_object_t *obj) { switch (obj->type) { case CMP_TYPE_NEGATIVE_FIXNUM: case CMP_TYPE_SINT8: case CMP_TYPE_SINT16: return true; default: return false; } } bool cmp_object_is_int(const cmp_object_t *obj) { switch (obj->type) { case CMP_TYPE_NEGATIVE_FIXNUM: case CMP_TYPE_SINT8: case CMP_TYPE_SINT16: case CMP_TYPE_SINT32: return true; default: return false; } } bool cmp_object_is_long(const cmp_object_t *obj) { switch (obj->type) { case CMP_TYPE_NEGATIVE_FIXNUM: case CMP_TYPE_SINT8: case CMP_TYPE_SINT16: case CMP_TYPE_SINT32: case CMP_TYPE_SINT64: return true; default: return false; } } bool cmp_object_is_sinteger(const cmp_object_t *obj) { return cmp_object_is_long(obj); } bool cmp_object_is_uchar(const cmp_object_t *obj) { switch (obj->type) { case CMP_TYPE_POSITIVE_FIXNUM: case CMP_TYPE_UINT8: return true; default: return false; } } bool cmp_object_is_ushort(const cmp_object_t *obj) { switch (obj->type) { case CMP_TYPE_POSITIVE_FIXNUM: case CMP_TYPE_UINT8: return true; case CMP_TYPE_UINT16: return true; default: return false; } } bool cmp_object_is_uint(const cmp_object_t *obj) { switch (obj->type) { case CMP_TYPE_POSITIVE_FIXNUM: case CMP_TYPE_UINT8: case CMP_TYPE_UINT16: case CMP_TYPE_UINT32: return true; default: return false; } } bool cmp_object_is_ulong(const cmp_object_t *obj) { switch (obj->type) { case CMP_TYPE_POSITIVE_FIXNUM: case CMP_TYPE_UINT8: case CMP_TYPE_UINT16: case CMP_TYPE_UINT32: case CMP_TYPE_UINT64: return true; default: return false; } } bool cmp_object_is_uinteger(const cmp_object_t *obj) { return cmp_object_is_ulong(obj); } bool cmp_object_is_float(const cmp_object_t *obj) { if (obj->type == CMP_TYPE_FLOAT) return true; return false; } bool cmp_object_is_double(const cmp_object_t *obj) { if (obj->type == CMP_TYPE_DOUBLE) return true; return false; } bool cmp_object_is_nil(const cmp_object_t *obj) { if (obj->type == CMP_TYPE_NIL) return true; return false; } bool cmp_object_is_bool(const cmp_object_t *obj) { if (obj->type == CMP_TYPE_BOOLEAN) return true; return false; } bool cmp_object_is_str(const cmp_object_t *obj) { switch (obj->type) { case CMP_TYPE_FIXSTR: case CMP_TYPE_STR8: case CMP_TYPE_STR16: case CMP_TYPE_STR32: return true; default: return false; } } bool cmp_object_is_bin(const cmp_object_t *obj) { switch (obj->type) { case CMP_TYPE_BIN8: case CMP_TYPE_BIN16: case CMP_TYPE_BIN32: return true; default: return false; } } bool cmp_object_is_array(const cmp_object_t *obj) { switch (obj->type) { case CMP_TYPE_FIXARRAY: case CMP_TYPE_ARRAY16: case CMP_TYPE_ARRAY32: return true; default: return false; } } bool cmp_object_is_map(const cmp_object_t *obj) { switch (obj->type) { case CMP_TYPE_FIXMAP: case CMP_TYPE_MAP16: case CMP_TYPE_MAP32: return true; default: return false; } } bool cmp_object_is_ext(const cmp_object_t *obj) { switch (obj->type) { case CMP_TYPE_FIXEXT1: case CMP_TYPE_FIXEXT2: case CMP_TYPE_FIXEXT4: case CMP_TYPE_FIXEXT8: case CMP_TYPE_FIXEXT16: case CMP_TYPE_EXT8: case CMP_TYPE_EXT16: case CMP_TYPE_EXT32: return true; default: return false; } } bool cmp_object_as_char(const cmp_object_t *obj, int8_t *c) { switch (obj->type) { case CMP_TYPE_POSITIVE_FIXNUM: case CMP_TYPE_NEGATIVE_FIXNUM: case CMP_TYPE_SINT8: *c = obj->as.s8; return true; case CMP_TYPE_UINT8: if (obj->as.u8 <= 0x7F) { *c = obj->as.s8; return true; } else { return false; } default: return false; } } bool cmp_object_as_short(const cmp_object_t *obj, int16_t *s) { switch (obj->type) { case CMP_TYPE_POSITIVE_FIXNUM: case CMP_TYPE_NEGATIVE_FIXNUM: case CMP_TYPE_SINT8: *s = obj->as.s8; return true; case CMP_TYPE_UINT8: *s = obj->as.u8; return true; case CMP_TYPE_SINT16: *s = obj->as.s16; return true; case CMP_TYPE_UINT16: if (obj->as.u16 <= 0x7FFF) { *s = (int16_t)obj->as.u16; return true; } else { return false; } default: return false; } } bool cmp_object_as_int(const cmp_object_t *obj, int32_t *i) { switch (obj->type) { case CMP_TYPE_POSITIVE_FIXNUM: case CMP_TYPE_NEGATIVE_FIXNUM: case CMP_TYPE_SINT8: *i = obj->as.s8; return true; case CMP_TYPE_UINT8: *i = obj->as.u8; return true; case CMP_TYPE_SINT16: *i = obj->as.s16; return true; case CMP_TYPE_UINT16: *i = obj->as.u16; return true; case CMP_TYPE_SINT32: *i = obj->as.s32; return true; case CMP_TYPE_UINT32: if (obj->as.u32 <= 0x7FFFFFFF) { *i = (int32_t)obj->as.u32; return true; } else { return false; } default: return false; } } bool cmp_object_as_long(const cmp_object_t *obj, int64_t *d) { switch (obj->type) { case CMP_TYPE_POSITIVE_FIXNUM: case CMP_TYPE_NEGATIVE_FIXNUM: case CMP_TYPE_SINT8: *d = obj->as.s8; return true; case CMP_TYPE_UINT8: *d = obj->as.u8; return true; case CMP_TYPE_SINT16: *d = obj->as.s16; return true; case CMP_TYPE_UINT16: *d = obj->as.u16; return true; case CMP_TYPE_SINT32: *d = obj->as.s32; return true; case CMP_TYPE_UINT32: *d = obj->as.u32; return true; case CMP_TYPE_SINT64: *d = obj->as.s64; return true; case CMP_TYPE_UINT64: if (obj->as.u64 <= UINT64_C(0x7FFFFFFFFFFFFFFF)) { *d = (int64_t)obj->as.u64; return true; } else { return false; } default: return false; } } bool cmp_object_as_sinteger(const cmp_object_t *obj, int64_t *d) { return cmp_object_as_long(obj, d); } bool cmp_object_as_uchar(const cmp_object_t *obj, uint8_t *c) { switch (obj->type) { case CMP_TYPE_POSITIVE_FIXNUM: case CMP_TYPE_UINT8: *c = obj->as.u8; return true; default: return false; } } bool cmp_object_as_ushort(const cmp_object_t *obj, uint16_t *s) { switch (obj->type) { case CMP_TYPE_POSITIVE_FIXNUM: case CMP_TYPE_UINT8: *s = obj->as.u8; return true; case CMP_TYPE_UINT16: *s = obj->as.u16; return true; default: return false; } } bool cmp_object_as_uint(const cmp_object_t *obj, uint32_t *i) { switch (obj->type) { case CMP_TYPE_POSITIVE_FIXNUM: case CMP_TYPE_UINT8: *i = obj->as.u8; return true; case CMP_TYPE_UINT16: *i = obj->as.u16; return true; case CMP_TYPE_UINT32: *i = obj->as.u32; return true; default: return false; } } bool cmp_object_as_ulong(const cmp_object_t *obj, uint64_t *u) { switch (obj->type) { case CMP_TYPE_POSITIVE_FIXNUM: case CMP_TYPE_UINT8: *u = obj->as.u8; return true; case CMP_TYPE_UINT16: *u = obj->as.u16; return true; case CMP_TYPE_UINT32: *u = obj->as.u32; return true; case CMP_TYPE_UINT64: *u = obj->as.u64; return true; default: return false; } } bool cmp_object_as_uinteger(const cmp_object_t *obj, uint64_t *u) { return cmp_object_as_ulong(obj, u); } #ifndef CMP_NO_FLOAT bool cmp_object_as_float(const cmp_object_t *obj, float *f) { if (obj->type == CMP_TYPE_FLOAT) { *f = obj->as.flt; return true; } return false; } bool cmp_object_as_double(const cmp_object_t *obj, double *d) { if (obj->type == CMP_TYPE_DOUBLE) { *d = obj->as.dbl; return true; } return false; } #endif /* CMP_NO_FLOAT */ bool cmp_object_as_bool(const cmp_object_t *obj, bool *b) { if (obj->type == CMP_TYPE_BOOLEAN) { if (obj->as.boolean) *b = true; else *b = false; return true; } return false; } bool cmp_object_as_str(const cmp_object_t *obj, uint32_t *size) { switch (obj->type) { case CMP_TYPE_FIXSTR: case CMP_TYPE_STR8: case CMP_TYPE_STR16: case CMP_TYPE_STR32: *size = obj->as.str_size; return true; default: return false; } } bool cmp_object_as_bin(const cmp_object_t *obj, uint32_t *size) { switch (obj->type) { case CMP_TYPE_BIN8: case CMP_TYPE_BIN16: case CMP_TYPE_BIN32: *size = obj->as.bin_size; return true; default: return false; } } bool cmp_object_as_array(const cmp_object_t *obj, uint32_t *size) { switch (obj->type) { case CMP_TYPE_FIXARRAY: case CMP_TYPE_ARRAY16: case CMP_TYPE_ARRAY32: *size = obj->as.array_size; return true; default: return false; } } bool cmp_object_as_map(const cmp_object_t *obj, uint32_t *size) { switch (obj->type) { case CMP_TYPE_FIXMAP: case CMP_TYPE_MAP16: case CMP_TYPE_MAP32: *size = obj->as.map_size; return true; default: return false; } } bool cmp_object_as_ext(const cmp_object_t *obj, int8_t *type, uint32_t *size) { switch (obj->type) { case CMP_TYPE_FIXEXT1: case CMP_TYPE_FIXEXT2: case CMP_TYPE_FIXEXT4: case CMP_TYPE_FIXEXT8: case CMP_TYPE_FIXEXT16: case CMP_TYPE_EXT8: case CMP_TYPE_EXT16: case CMP_TYPE_EXT32: *type = obj->as.ext.type; *size = obj->as.ext.size; return true; default: return false; } } bool cmp_object_to_str(cmp_ctx_t *ctx, const cmp_object_t *obj, char *data, uint32_t buf_size) { uint32_t str_size = 0; switch (obj->type) { case CMP_TYPE_FIXSTR: case CMP_TYPE_STR8: case CMP_TYPE_STR16: case CMP_TYPE_STR32: str_size = obj->as.str_size; if (str_size >= buf_size) { ctx->error = CMP_ERROR_STR_DATA_LENGTH_TOO_LONG; return false; } if (!ctx->read(ctx, data, str_size)) { ctx->error = CMP_ERROR_DATA_READING; return false; } data[str_size] = 0; return true; default: return false; } } bool cmp_object_to_bin(cmp_ctx_t *ctx, const cmp_object_t *obj, void *data, uint32_t buf_size) { uint32_t bin_size = 0; switch (obj->type) { case CMP_TYPE_BIN8: case CMP_TYPE_BIN16: case CMP_TYPE_BIN32: bin_size = obj->as.bin_size; if (bin_size > buf_size) { ctx->error = CMP_ERROR_BIN_DATA_LENGTH_TOO_LONG; return false; } if (!ctx->read(ctx, data, bin_size)) { ctx->error = CMP_ERROR_DATA_READING; return false; } return true; default: return false; } } /* vi: set et ts=2 sw=2: */