1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
|
/*
* upb - a minimalist implementation of protocol buffers.
*
* Copyright (c) 2009 Joshua Haberman. See LICENSE for details.
*/
#include "upb_textprinter.h"
#include <inttypes.h>
#include <malloc.h>
#include "upb_def.h"
#include "upb_string.h"
struct _upb_textprinter {
upb_sink sink;
upb_bytesink *bytesink;
upb_string *str;
int indent_depth;
bool single_line;
upb_fielddef *f;
};
static void upb_textprinter_endfield(upb_textprinter *p)
{
if(p->single_line)
upb_bytesink_put(p->bytesink, UPB_STRLIT(' '));
else
upb_bytesink_put(p->bytesink, UPB_STRLIT('\n'));
}
static bool upb_textprinter_putval(upb_textprinter *p, upb_value val) {
p->str = upb_string_tryrecycle(p->str);
#define CASE(fmtstr, member) upb_string_printf(p->str, fmtstr, val.member); break;
switch(p->f->type) {
case UPB_TYPE(DOUBLE):
CASE("%0.f", _double);
case UPB_TYPE(FLOAT):
CASE("%0.f", _float)
case UPB_TYPE(INT64):
case UPB_TYPE(SFIXED64):
case UPB_TYPE(SINT64):
CASE("%" PRId64, int64)
case UPB_TYPE(UINT64):
case UPB_TYPE(FIXED64):
CASE("%" PRIu64, uint64)
case UPB_TYPE(INT32):
case UPB_TYPE(SFIXED32):
case UPB_TYPE(SINT32):
CASE("%" PRId32, int32)
case UPB_TYPE(UINT32):
case UPB_TYPE(FIXED32):
case UPB_TYPE(ENUM):
CASE("%" PRIu32, uint32);
case UPB_TYPE(BOOL):
CASE("%hhu", _bool);
}
upb_bytesink_put(p->bytesink, p->str);
upb_textprinter_endfield(p);
return upb_ok(upb_bytesink_status(p->bytesink));
}
static bool upb_textprinter_putstr(upb_textprinter *p, upb_string *str) {
upb_bytesink_put(p->bytesink, UPB_STRLIT("\""));
// TODO: escaping.
upb_bytesink_put(p->bytesink, str);
upb_bytesink_put(p->bytesink, UPB_STRLIT("\""));
upb_textprinter_endfield(p);
return upb_ok(upb_bytesink_status(p->bytesink));
}
static void upb_textprinter_indent(upb_textprinter *p)
{
if(!p->single_line)
for(int i = 0; i < p->indent_depth; i++)
upb_bytesink_put(p->bytesink, UPB_STRLIT(" "));
}
static bool upb_textprinter_putdef(upb_textprinter *p, upb_fielddef *f)
{
upb_textprinter_indent(p);
upb_bytesink_put(p->bytesink, f->name);
upb_bytesink_put(p->bytesink, UPB_STRLIT(":"));
p->f = f;
return upb_ok(upb_bytesink_status(p->bytesink));
}
static bool upb_textprinter_startmsg(upb_textprinter *p)
{
upb_textprinter_indent(p);
upb_bytesink_put(p->bytesink, p->f->def->fqname);
upb_bytesink_put(p->bytesink, UPB_STRLIT(" {"));
if(!p->single_line) upb_bytesink_put(p->bytesink, UPB_STRLIT('\n'));
p->indent_depth++;
return upb_ok(upb_bytesink_status(p->bytesink));
}
static bool upb_textprinter_endmsg(upb_textprinter *p)
{
p->indent_depth--;
upb_textprinter_indent(p);
upb_bytesink_put(p->bytesink, UPB_STRLIT("}"));
upb_textprinter_endfield(p);
return upb_ok(upb_bytesink_status(p->bytesink));
}
upb_sink_vtable upb_textprinter_vtbl = {
(upb_sink_putdef_fptr)upb_textprinter_putdef,
(upb_sink_putval_fptr)upb_textprinter_putval,
(upb_sink_putstr_fptr)upb_textprinter_putstr,
(upb_sink_startmsg_fptr)upb_textprinter_startmsg,
(upb_sink_endmsg_fptr)upb_textprinter_endmsg,
};
upb_textprinter *upb_textprinter_new() {
upb_textprinter *p = malloc(sizeof(*p));
upb_sink_init(&p->sink, &upb_textprinter_vtbl);
return p;
}
void upb_textprinter_free(upb_textprinter *p) {
free(p);
}
void upb_textprinter_reset(upb_textprinter *p, upb_bytesink *sink,
bool single_line) {
p->bytesink = sink;
p->single_line = single_line;
p->indent_depth = 0;
}
upb_sink *upb_textprinter_sink(upb_textprinter *p) { return &p->sink; }
|
