blob: 621b33451f74a14d427a1cc5acafa597d6a04433 [file] [log] [blame]
#ifndef CN_PRINT_C
#define CN_PRINT_C
#define CN_INCLUDE_DUMPER
#ifdef CN_INCLUDE_DUMPER
#define _CRT_SECURE_NO_WARNINGS 1
#include <stdio.h>
#include <stdio.h>
#ifdef __cplusplus
extern "C" {
#endif
#ifdef EMACS_INDENTATION_HELPER
} /* Duh. */
#endif
#include <stdio.h>
#ifdef MSV_CRT
#include <winsock2.h>
#else
#define _snprintf snprintf
#endif
#include <string.h>
#include <stdbool.h>
#include <assert.h>
#include "cn-cbor/cn-cbor.h"
#include "cbor.h"
typedef struct _write_state {
char *rgbOutput;
ssize_t ib;
size_t cbLeft;
uint8_t *rgFlags;
const char *szIndentWith;
const char *szEndOfLine;
} cn_write_state;
typedef void (*cn_visit_func)(const cn_cbor *cb, int depth, void *context);
extern void _visit(const cn_cbor *cb, cn_visit_func visitor, cn_visit_func breaker, void *context);
const char RgchHex[] = {'0', '1', '2', '3', '4', '5', '6', '7', '8', '9', 'a', 'b', 'c', 'd', 'e', 'f'};
bool _isWritable(cn_write_state *ws, size_t cb)
{
if (ws->rgbOutput == NULL)
return true;
if ((ws->ib < 0) || (ws->ib + cb > ws->cbLeft)) {
ws->ib = -1;
return false;
}
return true;
}
void write_data(cn_write_state *ws, const char *sz, size_t cb)
{
if (_isWritable(ws, cb)) {
if (ws->rgbOutput != NULL)
memcpy(ws->rgbOutput + ws->ib, sz, cb);
ws->ib += cb;
}
}
void _doIndent(cn_write_state *ws, int depth)
{
int i;
char *sz = ws->rgbOutput + ws->ib;
size_t cbIndentWith = strlen(ws->szIndentWith);
int cbIndent = depth * cbIndentWith;
if (ws->rgbOutput == NULL) {
ws->ib += cbIndent;
return;
}
if (_isWritable(ws, cbIndent)) {
for (i = 0; i < depth; i++) {
memcpy(sz, ws->szIndentWith, cbIndentWith);
sz += cbIndentWith;
}
}
ws->ib += cbIndent;
return;
}
void _print_encoder(const cn_cbor *cb, int depth, void *context)
{
int i;
char rgchT[256];
int cch;
cn_write_state *ws = (cn_write_state *)context;
uint8_t flags = ws->rgFlags[depth];
if (flags & 1) {
write_data(ws, ", ", 2);
ws->rgFlags[depth] &= 0xfe;
if (ws->szIndentWith) {
write_data(ws, ws->szEndOfLine, strlen(ws->szEndOfLine));
_doIndent(ws, depth);
}
}
if (flags & 2) {
write_data(ws, ": ", 2);
ws->rgFlags[depth] &= 0xfd;
}
switch (cb->type) {
case CN_CBOR_BYTES_CHUNKED:
case CN_CBOR_TEXT_CHUNKED:
break;
case CN_CBOR_ARRAY:
write_data(ws, "[", 1);
ws->rgFlags[depth] |= 4;
if (ws->szIndentWith) {
write_data(ws, ws->szEndOfLine, strlen(ws->szEndOfLine));
_doIndent(ws, depth + 1);
}
break;
case CN_CBOR_MAP:
write_data(ws, "{", 1);
ws->rgFlags[depth] |= 8;
if (ws->szIndentWith) {
write_data(ws, ws->szEndOfLine, strlen(ws->szEndOfLine));
_doIndent(ws, depth + 1);
}
break;
case CN_CBOR_TAG:
case CN_CBOR_UINT:
case CN_CBOR_SIMPLE:
cch = _snprintf(rgchT, sizeof(rgchT), "%u", (unsigned int)cb->v.uint);
write_data(ws, rgchT, cch);
break;
case CN_CBOR_FALSE:
write_data(ws, "false", 5);
break;
case CN_CBOR_TRUE:
write_data(ws, "true", 4);
break;
case CN_CBOR_NULL:
write_data(ws, "null", 4);
break;
case CN_CBOR_UNDEF:
write_data(ws, "undef", 5);
break;
case CN_CBOR_INT:
cch = _snprintf(rgchT, sizeof(rgchT), "%d", (unsigned int)cb->v.sint);
write_data(ws, rgchT, cch);
break;
#ifndef CBOR_NO_FLOAT
case CN_CBOR_FLOAT:
cch = _snprintf(rgchT, sizeof(rgchT), "%f", cb->v.f);
write_data(ws, rgchT, cch);
break;
case CN_CBOR_DOUBLE:
cch = _snprintf(rgchT, sizeof(rgchT), "%f", cb->v.dbl);
write_data(ws, rgchT, cch);
break;
#endif
case CN_CBOR_INVALID:
write_data(ws, "invalid", 7);
break;
case CN_CBOR_TEXT:
write_data(ws, "\"", 1);
write_data(ws, cb->v.str, cb->length);
write_data(ws, "\"", 1);
break;
case CN_CBOR_BYTES:
write_data(ws, "h'", 2);
for (i = 0; i < cb->length; i++) {
write_data(ws, &RgchHex[(cb->v.str[i] / 16) & 0xf], 1);
write_data(ws, &RgchHex[cb->v.str[i] & 0xf], 1);
}
write_data(ws, "\'", 1);
break;
}
if (depth > 0) {
if (ws->rgFlags[depth - 1] & 4)
ws->rgFlags[depth] |= 1;
else if (ws->rgFlags[depth - 1] & 8) {
if (flags & 2)
ws->rgFlags[depth] |= 1;
else
ws->rgFlags[depth] |= 2;
}
}
}
void _print_breaker(const cn_cbor *cb, int depth, void *context)
{
cn_write_state *ws = (cn_write_state *)context;
switch (cb->type) {
case CN_CBOR_ARRAY:
if (ws->szIndentWith) {
write_data(ws, ws->szEndOfLine, strlen(ws->szEndOfLine));
_doIndent(ws, depth);
}
write_data(ws, "]", 1);
ws->rgFlags[depth + 1] = 0;
break;
case CN_CBOR_MAP:
if (ws->szIndentWith) {
write_data(ws, ws->szEndOfLine, strlen(ws->szEndOfLine));
_doIndent(ws, depth);
}
write_data(ws, "}", 1);
ws->rgFlags[depth + 1] = 0;
break;
default:
break;
}
}
ssize_t cn_cbor_printer_write(char *rgbBuffer,
size_t cbBuffer,
const cn_cbor *cb,
const char *szIndentWith,
const char *szEndOfLine)
{
uint8_t flags[128] = {0};
char rgchZero[1] = {0};
cn_write_state ws = {rgbBuffer, 0, cbBuffer, flags, szIndentWith, szEndOfLine};
_visit(cb, _print_encoder, _print_breaker, &ws);
write_data(&ws, rgchZero, 1);
return ws.ib;
}
#ifdef EMACS_INDENTATION_HELPER
{ /* Duh. */
#endif
#ifdef _cplusplus
} /* extern "C" */
#endif
#endif // CN_INCLUDE_DUMPER
#endif // CN_PRINT_C