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
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
|
/*
* upb - a minimalist implementation of protocol buffers.
*
* Copyright (c) 2011 Google Inc. See LICENSE for details.
* Author: Josh Haberman <jhaberman@gmail.com>
*
* The set of upb::*Def classes and upb::SymbolTable allow for defining and
* manipulating schema information (as defined in .proto files).
*/
#ifndef UPB_DEF_HPP
#define UPB_DEF_HPP
#include <algorithm>
#include <string>
#include <vector>
#include "upb/def.h"
#include "upb/upb.hpp"
namespace upb {
class MessageDef;
class FieldDef : public upb_fielddef {
public:
static FieldDef* Cast(upb_fielddef *f) { return (FieldDef*)f; }
static const FieldDef* Cast(const upb_fielddef *f) { return (FieldDef*)f; }
static FieldDef* New() { return Cast(upb_fielddef_new()); }
FieldDef* Dup() { return Cast(upb_fielddef_dup(this)); }
// Read accessors -- may be called at any time.
uint8_t type() const { return upb_fielddef_type(this); }
uint8_t label() const { return upb_fielddef_label(this); }
int32_t number() const { return upb_fielddef_number(this); }
std::string name() const { return std::string(upb_fielddef_name(this)); }
Value default_() const { return upb_fielddef_default(this); }
Value bound_value() const { return upb_fielddef_fval(this); }
MessageDef* message() { return (MessageDef*)upb_fielddef_msgdef(this); }
const MessageDef* message() const { return (MessageDef*)upb_fielddef_msgdef(this); }
//Def* subdef() { return upb_fielddef_subdef(this); }
//const Def* subdef() { return upb_fielddef_subdef(this); }
// Returns true if this FieldDef is finalized
bool IsFinalized() const { return upb_fielddef_finalized(this); }
struct _upb_accessor_vtbl *accessor() const {
return upb_fielddef_accessor(this);
}
std::string type_name() const {
return std::string(upb_fielddef_typename(this));
}
// Write accessors -- may not be called once the FieldDef is finalized.
private:
FieldDef();
~FieldDef();
};
class MessageDef : public upb_msgdef {
public:
// Converting from C types to C++ wrapper types.
static MessageDef* Cast(upb_msgdef *md) { return (MessageDef*)md; }
static const MessageDef* Cast(const upb_msgdef *md) {
return (MessageDef*)md;
}
static MessageDef* New() { return Cast(upb_msgdef_new()); }
MessageDef* Dup() { return Cast(upb_msgdef_dup(this)); }
void Ref() const { upb_msgdef_ref(this); }
void Unref() const { upb_msgdef_unref(this); }
// Read accessors -- may be called at any time.
// The total size of in-memory messages created with this MessageDef.
uint16_t instance_size() const { return upb_msgdef_size(this); }
// The number of "hasbit" bytes in a message instance.
uint8_t hasbit_bytes() const { return upb_msgdef_hasbit_bytes(this); }
uint32_t extension_start() const { return upb_msgdef_extstart(this); }
uint32_t extension_end() const { return upb_msgdef_extend(this); }
// Write accessors. May only be called before the msgdef is in a symtab.
void set_instance_size(uint16_t size) { upb_msgdef_setsize(this, size); }
void set_hasbit_bytes(uint16_t size) { upb_msgdef_setsize(this, size); }
bool SetExtensionRange(uint32_t start, uint32_t end) {
return upb_msgdef_setextrange(this, start, end);
}
// Adds a set of fields (upb_fielddef objects) to a msgdef. Caller retains
// its ref on the fielddef. May only be done before the msgdef is in a
// symtab (requires upb_def_ismutable(m) for the msgdef). The fielddef's
// name and number must be set, and the message may not already contain any
// field with this name or number, and this fielddef may not be part of
// another message, otherwise false is returned and no action is performed.
bool AddFields(FieldDef** f, int n) {
return upb_msgdef_addfields(this, (upb_fielddef**)f, n);
}
bool AddFields(const std::vector<FieldDef*>& fields) {
FieldDef *arr[fields.size()];
std::copy(fields.begin(), fields.end(), arr);
return AddFields(arr, fields.size());
}
// Lookup fields by name or number, returning NULL if no such field exists.
FieldDef* FindFieldByName(const char *name) {
return FieldDef::Cast(upb_msgdef_ntof(this, name));
}
FieldDef* FindFieldByName(const std::string& name) {
return FieldDef::Cast(upb_msgdef_ntof(this, name.c_str()));
}
FieldDef* FindFieldByNumber(uint32_t num) {
return FieldDef::Cast(upb_msgdef_itof(this, num));
}
const FieldDef* FindFieldByName(const char *name) const {
return FindFieldByName(name);
}
const FieldDef* FindFieldByName(const std::string& name) const {
return FindFieldByName(name);
}
const FieldDef* FindFieldByNumber(uint32_t num) const {
return FindFieldByNumber(num);
}
// TODO: iteration over fields.
private:
MessageDef();
~MessageDef();
};
class SymbolTable : public upb_symtab {
public:
// Converting from C types to C++ wrapper types.
static SymbolTable* Cast(upb_symtab *s) { return (SymbolTable*)s; }
static const SymbolTable* Cast(const upb_symtab *s) {
return (SymbolTable*)s;
}
static SymbolTable* New() { return Cast(upb_symtab_new()); }
void Ref() const { upb_symtab_unref(this); }
void Unref() const { upb_symtab_unref(this); }
// If the given name refers to a message in this symbol table, returns a new
// ref to that MessageDef object, otherwise returns NULL.
const MessageDef* LookupMessage(const char *name) const {
return MessageDef::Cast(upb_symtab_lookupmsg(this, name));
}
private:
SymbolTable();
~SymbolTable();
};
} // namespace upb
#endif
|
