File indexing completed on 2025-08-27 09:30:26
0001
0002
0003
0004
0005
0006
0007
0008
0009
0010
0011
0012
0013
0014
0015
0016
0017 #ifndef FLATBUFFERS_FLEXBUFFERS_H_
0018 #define FLATBUFFERS_FLEXBUFFERS_H_
0019
0020 #include <algorithm>
0021 #include <map>
0022
0023 #include "flatbuffers/base.h"
0024
0025 #include "flatbuffers/util.h"
0026
0027 #ifdef _MSC_VER
0028 # include <intrin.h>
0029 #endif
0030
0031 #if defined(_MSC_VER)
0032 # pragma warning(push)
0033 # pragma warning(disable : 4127)
0034 #endif
0035
0036 namespace flexbuffers {
0037
0038 class Reference;
0039 class Map;
0040
0041
0042
0043 enum BitWidth {
0044 BIT_WIDTH_8 = 0,
0045 BIT_WIDTH_16 = 1,
0046 BIT_WIDTH_32 = 2,
0047 BIT_WIDTH_64 = 3,
0048 };
0049
0050
0051
0052 enum Type {
0053 FBT_NULL = 0,
0054 FBT_INT = 1,
0055 FBT_UINT = 2,
0056 FBT_FLOAT = 3,
0057
0058 FBT_KEY = 4,
0059 FBT_STRING = 5,
0060 FBT_INDIRECT_INT = 6,
0061 FBT_INDIRECT_UINT = 7,
0062 FBT_INDIRECT_FLOAT = 8,
0063 FBT_MAP = 9,
0064 FBT_VECTOR = 10,
0065 FBT_VECTOR_INT = 11,
0066 FBT_VECTOR_UINT = 12,
0067 FBT_VECTOR_FLOAT = 13,
0068 FBT_VECTOR_KEY = 14,
0069
0070
0071 FBT_VECTOR_STRING_DEPRECATED = 15,
0072 FBT_VECTOR_INT2 = 16,
0073 FBT_VECTOR_UINT2 = 17,
0074 FBT_VECTOR_FLOAT2 = 18,
0075 FBT_VECTOR_INT3 = 19,
0076 FBT_VECTOR_UINT3 = 20,
0077 FBT_VECTOR_FLOAT3 = 21,
0078 FBT_VECTOR_INT4 = 22,
0079 FBT_VECTOR_UINT4 = 23,
0080 FBT_VECTOR_FLOAT4 = 24,
0081 FBT_BLOB = 25,
0082 FBT_BOOL = 26,
0083 FBT_VECTOR_BOOL =
0084 36,
0085
0086 FBT_MAX_TYPE = 37
0087 };
0088
0089 inline bool IsInline(Type t) { return t <= FBT_FLOAT || t == FBT_BOOL; }
0090
0091 inline bool IsTypedVectorElementType(Type t) {
0092 return (t >= FBT_INT && t <= FBT_STRING) || t == FBT_BOOL;
0093 }
0094
0095 inline bool IsTypedVector(Type t) {
0096 return (t >= FBT_VECTOR_INT && t <= FBT_VECTOR_STRING_DEPRECATED) ||
0097 t == FBT_VECTOR_BOOL;
0098 }
0099
0100 inline bool IsFixedTypedVector(Type t) {
0101 return t >= FBT_VECTOR_INT2 && t <= FBT_VECTOR_FLOAT4;
0102 }
0103
0104 inline Type ToTypedVector(Type t, size_t fixed_len = 0) {
0105 FLATBUFFERS_ASSERT(IsTypedVectorElementType(t));
0106 switch (fixed_len) {
0107 case 0: return static_cast<Type>(t - FBT_INT + FBT_VECTOR_INT);
0108 case 2: return static_cast<Type>(t - FBT_INT + FBT_VECTOR_INT2);
0109 case 3: return static_cast<Type>(t - FBT_INT + FBT_VECTOR_INT3);
0110 case 4: return static_cast<Type>(t - FBT_INT + FBT_VECTOR_INT4);
0111 default: FLATBUFFERS_ASSERT(0); return FBT_NULL;
0112 }
0113 }
0114
0115 inline Type ToTypedVectorElementType(Type t) {
0116 FLATBUFFERS_ASSERT(IsTypedVector(t));
0117 return static_cast<Type>(t - FBT_VECTOR_INT + FBT_INT);
0118 }
0119
0120 inline Type ToFixedTypedVectorElementType(Type t, uint8_t *len) {
0121 FLATBUFFERS_ASSERT(IsFixedTypedVector(t));
0122 auto fixed_type = t - FBT_VECTOR_INT2;
0123 *len = static_cast<uint8_t>(fixed_type / 3 +
0124 2);
0125 return static_cast<Type>(fixed_type % 3 + FBT_INT);
0126 }
0127
0128
0129
0130 typedef int16_t half;
0131 typedef int8_t quarter;
0132
0133
0134
0135
0136
0137
0138 template<typename R, typename T1, typename T2, typename T4, typename T8>
0139 R ReadSizedScalar(const uint8_t *data, uint8_t byte_width) {
0140 return byte_width < 4
0141 ? (byte_width < 2
0142 ? static_cast<R>(flatbuffers::ReadScalar<T1>(data))
0143 : static_cast<R>(flatbuffers::ReadScalar<T2>(data)))
0144 : (byte_width < 8
0145 ? static_cast<R>(flatbuffers::ReadScalar<T4>(data))
0146 : static_cast<R>(flatbuffers::ReadScalar<T8>(data)));
0147 }
0148
0149 inline int64_t ReadInt64(const uint8_t *data, uint8_t byte_width) {
0150 return ReadSizedScalar<int64_t, int8_t, int16_t, int32_t, int64_t>(
0151 data, byte_width);
0152 }
0153
0154 inline uint64_t ReadUInt64(const uint8_t *data, uint8_t byte_width) {
0155
0156
0157
0158
0159
0160
0161
0162 #if defined(_MSC_VER) && defined(_M_X64) && !defined(_M_ARM64EC)
0163
0164 uint64_t u = 0;
0165 __movsb(reinterpret_cast<uint8_t *>(&u),
0166 reinterpret_cast<const uint8_t *>(data), byte_width);
0167 return flatbuffers::EndianScalar(u);
0168 #else
0169 return ReadSizedScalar<uint64_t, uint8_t, uint16_t, uint32_t, uint64_t>(
0170 data, byte_width);
0171 #endif
0172
0173 }
0174
0175 inline double ReadDouble(const uint8_t *data, uint8_t byte_width) {
0176 return ReadSizedScalar<double, quarter, half, float, double>(data,
0177 byte_width);
0178 }
0179
0180 inline const uint8_t *Indirect(const uint8_t *offset, uint8_t byte_width) {
0181 return offset - ReadUInt64(offset, byte_width);
0182 }
0183
0184 template<typename T> const uint8_t *Indirect(const uint8_t *offset) {
0185 return offset - flatbuffers::ReadScalar<T>(offset);
0186 }
0187
0188 inline BitWidth WidthU(uint64_t u) {
0189 #define FLATBUFFERS_GET_FIELD_BIT_WIDTH(value, width) \
0190 { \
0191 if (!((u) & ~((1ULL << (width)) - 1ULL))) return BIT_WIDTH_##width; \
0192 }
0193 FLATBUFFERS_GET_FIELD_BIT_WIDTH(u, 8);
0194 FLATBUFFERS_GET_FIELD_BIT_WIDTH(u, 16);
0195 FLATBUFFERS_GET_FIELD_BIT_WIDTH(u, 32);
0196 #undef FLATBUFFERS_GET_FIELD_BIT_WIDTH
0197 return BIT_WIDTH_64;
0198 }
0199
0200 inline BitWidth WidthI(int64_t i) {
0201 auto u = static_cast<uint64_t>(i) << 1;
0202 return WidthU(i >= 0 ? u : ~u);
0203 }
0204
0205 inline BitWidth WidthF(double f) {
0206 return static_cast<double>(static_cast<float>(f)) == f ? BIT_WIDTH_32
0207 : BIT_WIDTH_64;
0208 }
0209
0210
0211
0212 class Object {
0213 public:
0214 Object(const uint8_t *data, uint8_t byte_width)
0215 : data_(data), byte_width_(byte_width) {}
0216
0217 protected:
0218 const uint8_t *data_;
0219 uint8_t byte_width_;
0220 };
0221
0222
0223 class Sized : public Object {
0224 public:
0225
0226 Sized(const uint8_t *data, uint8_t byte_width)
0227 : Object(data, byte_width), size_(read_size()) {}
0228
0229 Sized(const uint8_t *data, uint8_t byte_width, size_t sz)
0230 : Object(data, byte_width), size_(sz) {}
0231 size_t size() const { return size_; }
0232
0233 size_t read_size() const {
0234 return static_cast<size_t>(ReadUInt64(data_ - byte_width_, byte_width_));
0235 }
0236
0237 protected:
0238 size_t size_;
0239 };
0240
0241 class String : public Sized {
0242 public:
0243
0244 String(const uint8_t *data, uint8_t byte_width) : Sized(data, byte_width) {}
0245
0246 String(const uint8_t *data, uint8_t byte_width, size_t sz)
0247 : Sized(data, byte_width, sz) {}
0248
0249 size_t length() const { return size(); }
0250 const char *c_str() const { return reinterpret_cast<const char *>(data_); }
0251 std::string str() const { return std::string(c_str(), size()); }
0252
0253 static String EmptyString() {
0254 static const char *empty_string = "";
0255 return String(reinterpret_cast<const uint8_t *>(empty_string), 1, 0);
0256 }
0257 bool IsTheEmptyString() const { return data_ == EmptyString().data_; }
0258 };
0259
0260 class Blob : public Sized {
0261 public:
0262 Blob(const uint8_t *data_buf, uint8_t byte_width)
0263 : Sized(data_buf, byte_width) {}
0264
0265 static Blob EmptyBlob() {
0266 static const uint8_t empty_blob[] = { 0 };
0267 return Blob(empty_blob + 1, 1);
0268 }
0269 bool IsTheEmptyBlob() const { return data_ == EmptyBlob().data_; }
0270 const uint8_t *data() const { return data_; }
0271 };
0272
0273 class Vector : public Sized {
0274 public:
0275 Vector(const uint8_t *data, uint8_t byte_width) : Sized(data, byte_width) {}
0276
0277 Reference operator[](size_t i) const;
0278
0279 static Vector EmptyVector() {
0280 static const uint8_t empty_vector[] = { 0 };
0281 return Vector(empty_vector + 1, 1);
0282 }
0283 bool IsTheEmptyVector() const { return data_ == EmptyVector().data_; }
0284 };
0285
0286 class TypedVector : public Sized {
0287 public:
0288 TypedVector(const uint8_t *data, uint8_t byte_width, Type element_type)
0289 : Sized(data, byte_width), type_(element_type) {}
0290
0291 Reference operator[](size_t i) const;
0292
0293 static TypedVector EmptyTypedVector() {
0294 static const uint8_t empty_typed_vector[] = { 0 };
0295 return TypedVector(empty_typed_vector + 1, 1, FBT_INT);
0296 }
0297 bool IsTheEmptyVector() const {
0298 return data_ == TypedVector::EmptyTypedVector().data_;
0299 }
0300
0301 Type ElementType() { return type_; }
0302
0303 friend Reference;
0304
0305 private:
0306 Type type_;
0307
0308 friend Map;
0309 };
0310
0311 class FixedTypedVector : public Object {
0312 public:
0313 FixedTypedVector(const uint8_t *data, uint8_t byte_width, Type element_type,
0314 uint8_t len)
0315 : Object(data, byte_width), type_(element_type), len_(len) {}
0316
0317 Reference operator[](size_t i) const;
0318
0319 static FixedTypedVector EmptyFixedTypedVector() {
0320 static const uint8_t fixed_empty_vector[] = { 0 };
0321 return FixedTypedVector(fixed_empty_vector, 1, FBT_INT, 0);
0322 }
0323 bool IsTheEmptyFixedTypedVector() const {
0324 return data_ == FixedTypedVector::EmptyFixedTypedVector().data_;
0325 }
0326
0327 Type ElementType() const { return type_; }
0328 uint8_t size() const { return len_; }
0329
0330 private:
0331 Type type_;
0332 uint8_t len_;
0333 };
0334
0335 class Map : public Vector {
0336 public:
0337 Map(const uint8_t *data, uint8_t byte_width) : Vector(data, byte_width) {}
0338
0339 Reference operator[](const char *key) const;
0340 Reference operator[](const std::string &key) const;
0341
0342 Vector Values() const { return Vector(data_, byte_width_); }
0343
0344 TypedVector Keys() const {
0345 const size_t num_prefixed_fields = 3;
0346 auto keys_offset = data_ - byte_width_ * num_prefixed_fields;
0347 return TypedVector(Indirect(keys_offset, byte_width_),
0348 static_cast<uint8_t>(
0349 ReadUInt64(keys_offset + byte_width_, byte_width_)),
0350 FBT_KEY);
0351 }
0352
0353 static Map EmptyMap() {
0354 static const uint8_t empty_map[] = {
0355 0 , 0 , 1 , 0
0356 };
0357 return Map(empty_map + 4, 1);
0358 }
0359
0360 bool IsTheEmptyMap() const { return data_ == EmptyMap().data_; }
0361 };
0362
0363 inline void IndentString(std::string &s, int indent,
0364 const char *indent_string) {
0365 for (int i = 0; i < indent; i++) s += indent_string;
0366 }
0367
0368 template<typename T>
0369 void AppendToString(std::string &s, T &&v, bool keys_quoted, bool indented,
0370 int cur_indent, const char *indent_string,
0371 bool natural_utf8) {
0372 s += "[";
0373 s += indented ? "\n" : " ";
0374 for (size_t i = 0; i < v.size(); i++) {
0375 if (i) {
0376 s += ",";
0377 s += indented ? "\n" : " ";
0378 }
0379 if (indented) IndentString(s, cur_indent, indent_string);
0380 v[i].ToString(true, keys_quoted, s, indented, cur_indent,
0381 indent_string, natural_utf8);
0382 }
0383 if (indented) {
0384 s += "\n";
0385 IndentString(s, cur_indent - 1, indent_string);
0386 } else {
0387 s += " ";
0388 }
0389 s += "]";
0390 }
0391
0392 template<typename T>
0393 void AppendToString(std::string &s, T &&v, bool keys_quoted) {
0394 AppendToString(s, v, keys_quoted);
0395 }
0396
0397
0398 class Reference {
0399 public:
0400 Reference()
0401 : data_(nullptr), parent_width_(0), byte_width_(0), type_(FBT_NULL) {}
0402
0403 Reference(const uint8_t *data, uint8_t parent_width, uint8_t byte_width,
0404 Type type)
0405 : data_(data),
0406 parent_width_(parent_width),
0407 byte_width_(byte_width),
0408 type_(type) {}
0409
0410 Reference(const uint8_t *data, uint8_t parent_width, uint8_t packed_type)
0411 : data_(data),
0412 parent_width_(parent_width),
0413 byte_width_(static_cast<uint8_t>(1 << (packed_type & 3))),
0414 type_(static_cast<Type>(packed_type >> 2)) {}
0415
0416 Type GetType() const { return type_; }
0417
0418 bool IsNull() const { return type_ == FBT_NULL; }
0419 bool IsBool() const { return type_ == FBT_BOOL; }
0420 bool IsInt() const { return type_ == FBT_INT || type_ == FBT_INDIRECT_INT; }
0421 bool IsUInt() const {
0422 return type_ == FBT_UINT || type_ == FBT_INDIRECT_UINT;
0423 }
0424 bool IsIntOrUint() const { return IsInt() || IsUInt(); }
0425 bool IsFloat() const {
0426 return type_ == FBT_FLOAT || type_ == FBT_INDIRECT_FLOAT;
0427 }
0428 bool IsNumeric() const { return IsIntOrUint() || IsFloat(); }
0429 bool IsString() const { return type_ == FBT_STRING; }
0430 bool IsKey() const { return type_ == FBT_KEY; }
0431 bool IsVector() const { return type_ == FBT_VECTOR || type_ == FBT_MAP; }
0432 bool IsUntypedVector() const { return type_ == FBT_VECTOR; }
0433 bool IsTypedVector() const { return flexbuffers::IsTypedVector(type_); }
0434 bool IsFixedTypedVector() const {
0435 return flexbuffers::IsFixedTypedVector(type_);
0436 }
0437 bool IsAnyVector() const {
0438 return (IsTypedVector() || IsFixedTypedVector() || IsVector());
0439 }
0440 bool IsMap() const { return type_ == FBT_MAP; }
0441 bool IsBlob() const { return type_ == FBT_BLOB; }
0442 bool AsBool() const {
0443 return (type_ == FBT_BOOL ? ReadUInt64(data_, parent_width_)
0444 : AsUInt64()) != 0;
0445 }
0446
0447
0448
0449
0450 int64_t AsInt64() const {
0451 if (type_ == FBT_INT) {
0452
0453 return ReadInt64(data_, parent_width_);
0454 } else
0455 switch (type_) {
0456 case FBT_INDIRECT_INT: return ReadInt64(Indirect(), byte_width_);
0457 case FBT_UINT: return ReadUInt64(data_, parent_width_);
0458 case FBT_INDIRECT_UINT: return ReadUInt64(Indirect(), byte_width_);
0459 case FBT_FLOAT:
0460 return static_cast<int64_t>(ReadDouble(data_, parent_width_));
0461 case FBT_INDIRECT_FLOAT:
0462 return static_cast<int64_t>(ReadDouble(Indirect(), byte_width_));
0463 case FBT_NULL: return 0;
0464 case FBT_STRING: return flatbuffers::StringToInt(AsString().c_str());
0465 case FBT_VECTOR: return static_cast<int64_t>(AsVector().size());
0466 case FBT_BOOL: return ReadInt64(data_, parent_width_);
0467 default:
0468
0469 return 0;
0470 }
0471 }
0472
0473
0474
0475 int32_t AsInt32() const { return static_cast<int32_t>(AsInt64()); }
0476 int16_t AsInt16() const { return static_cast<int16_t>(AsInt64()); }
0477 int8_t AsInt8() const { return static_cast<int8_t>(AsInt64()); }
0478
0479 uint64_t AsUInt64() const {
0480 if (type_ == FBT_UINT) {
0481
0482 return ReadUInt64(data_, parent_width_);
0483 } else
0484 switch (type_) {
0485 case FBT_INDIRECT_UINT: return ReadUInt64(Indirect(), byte_width_);
0486 case FBT_INT: return ReadInt64(data_, parent_width_);
0487 case FBT_INDIRECT_INT: return ReadInt64(Indirect(), byte_width_);
0488 case FBT_FLOAT:
0489 return static_cast<uint64_t>(ReadDouble(data_, parent_width_));
0490 case FBT_INDIRECT_FLOAT:
0491 return static_cast<uint64_t>(ReadDouble(Indirect(), byte_width_));
0492 case FBT_NULL: return 0;
0493 case FBT_STRING: return flatbuffers::StringToUInt(AsString().c_str());
0494 case FBT_VECTOR: return static_cast<uint64_t>(AsVector().size());
0495 case FBT_BOOL: return ReadUInt64(data_, parent_width_);
0496 default:
0497
0498 return 0;
0499 }
0500 }
0501
0502 uint32_t AsUInt32() const { return static_cast<uint32_t>(AsUInt64()); }
0503 uint16_t AsUInt16() const { return static_cast<uint16_t>(AsUInt64()); }
0504 uint8_t AsUInt8() const { return static_cast<uint8_t>(AsUInt64()); }
0505
0506 double AsDouble() const {
0507 if (type_ == FBT_FLOAT) {
0508
0509 return ReadDouble(data_, parent_width_);
0510 } else
0511 switch (type_) {
0512 case FBT_INDIRECT_FLOAT: return ReadDouble(Indirect(), byte_width_);
0513 case FBT_INT:
0514 return static_cast<double>(ReadInt64(data_, parent_width_));
0515 case FBT_UINT:
0516 return static_cast<double>(ReadUInt64(data_, parent_width_));
0517 case FBT_INDIRECT_INT:
0518 return static_cast<double>(ReadInt64(Indirect(), byte_width_));
0519 case FBT_INDIRECT_UINT:
0520 return static_cast<double>(ReadUInt64(Indirect(), byte_width_));
0521 case FBT_NULL: return 0.0;
0522 case FBT_STRING: {
0523 double d;
0524 flatbuffers::StringToNumber(AsString().c_str(), &d);
0525 return d;
0526 }
0527 case FBT_VECTOR: return static_cast<double>(AsVector().size());
0528 case FBT_BOOL:
0529 return static_cast<double>(ReadUInt64(data_, parent_width_));
0530 default:
0531
0532 return 0;
0533 }
0534 }
0535
0536 float AsFloat() const { return static_cast<float>(AsDouble()); }
0537
0538 const char *AsKey() const {
0539 if (type_ == FBT_KEY || type_ == FBT_STRING) {
0540 return reinterpret_cast<const char *>(Indirect());
0541 } else {
0542 return "";
0543 }
0544 }
0545
0546
0547
0548 String AsString() const {
0549 if (type_ == FBT_STRING) {
0550 return String(Indirect(), byte_width_);
0551 } else if (type_ == FBT_KEY) {
0552 auto key = Indirect();
0553 return String(key, byte_width_,
0554 strlen(reinterpret_cast<const char *>(key)));
0555 } else {
0556 return String::EmptyString();
0557 }
0558 }
0559
0560
0561 std::string ToString() const {
0562 std::string s;
0563 ToString(false, false, s);
0564 return s;
0565 }
0566
0567
0568
0569
0570 void ToString(bool strings_quoted, bool keys_quoted, std::string &s) const {
0571 ToString(strings_quoted, keys_quoted, s, false, 0, "", false);
0572 }
0573
0574
0575 void ToString(bool strings_quoted, bool keys_quoted, std::string &s,
0576 bool indented, int cur_indent, const char *indent_string,
0577 bool natural_utf8 = false) const {
0578 if (type_ == FBT_STRING) {
0579 String str(Indirect(), byte_width_);
0580 if (strings_quoted) {
0581 flatbuffers::EscapeString(str.c_str(), str.length(), &s, true, natural_utf8);
0582 } else {
0583 s.append(str.c_str(), str.length());
0584 }
0585 } else if (IsKey()) {
0586 auto str = AsKey();
0587 if (keys_quoted) {
0588 flatbuffers::EscapeString(str, strlen(str), &s, true, natural_utf8);
0589 } else {
0590 s += str;
0591 }
0592 } else if (IsInt()) {
0593 s += flatbuffers::NumToString(AsInt64());
0594 } else if (IsUInt()) {
0595 s += flatbuffers::NumToString(AsUInt64());
0596 } else if (IsFloat()) {
0597 s += flatbuffers::NumToString(AsDouble());
0598 } else if (IsNull()) {
0599 s += "null";
0600 } else if (IsBool()) {
0601 s += AsBool() ? "true" : "false";
0602 } else if (IsMap()) {
0603 s += "{";
0604 s += indented ? "\n" : " ";
0605 auto m = AsMap();
0606 auto keys = m.Keys();
0607 auto vals = m.Values();
0608 for (size_t i = 0; i < keys.size(); i++) {
0609 bool kq = keys_quoted;
0610 if (!kq) {
0611
0612
0613 const char *p = keys[i].AsKey();
0614 if (!flatbuffers::is_alpha(*p) && *p != '_') {
0615 kq = true;
0616 } else {
0617 while (*++p) {
0618 if (!flatbuffers::is_alnum(*p) && *p != '_') {
0619 kq = true;
0620 break;
0621 }
0622 }
0623 }
0624 }
0625 if (indented) IndentString(s, cur_indent + 1, indent_string);
0626 keys[i].ToString(true, kq, s);
0627 s += ": ";
0628 vals[i].ToString(true, keys_quoted, s, indented, cur_indent + 1, indent_string,
0629 natural_utf8);
0630 if (i < keys.size() - 1) {
0631 s += ",";
0632 if (!indented) s += " ";
0633 }
0634 if (indented) s += "\n";
0635 }
0636 if (!indented) s += " ";
0637 if (indented) IndentString(s, cur_indent, indent_string);
0638 s += "}";
0639 } else if (IsVector()) {
0640 AppendToString<Vector>(s, AsVector(), keys_quoted, indented,
0641 cur_indent + 1, indent_string, natural_utf8);
0642 } else if (IsTypedVector()) {
0643 AppendToString<TypedVector>(s, AsTypedVector(), keys_quoted, indented,
0644 cur_indent + 1, indent_string,
0645 natural_utf8);
0646 } else if (IsFixedTypedVector()) {
0647 AppendToString<FixedTypedVector>(s, AsFixedTypedVector(), keys_quoted,
0648 indented, cur_indent + 1, indent_string,
0649 natural_utf8);
0650 } else if (IsBlob()) {
0651 auto blob = AsBlob();
0652 flatbuffers::EscapeString(reinterpret_cast<const char *>(blob.data()),
0653 blob.size(), &s, true, false);
0654 } else {
0655 s += "(?)";
0656 }
0657 }
0658
0659
0660
0661 Blob AsBlob() const {
0662 if (type_ == FBT_BLOB || type_ == FBT_STRING) {
0663 return Blob(Indirect(), byte_width_);
0664 } else {
0665 return Blob::EmptyBlob();
0666 }
0667 }
0668
0669
0670
0671 Vector AsVector() const {
0672 if (type_ == FBT_VECTOR || type_ == FBT_MAP) {
0673 return Vector(Indirect(), byte_width_);
0674 } else {
0675 return Vector::EmptyVector();
0676 }
0677 }
0678
0679 TypedVector AsTypedVector() const {
0680 if (IsTypedVector()) {
0681 auto tv =
0682 TypedVector(Indirect(), byte_width_, ToTypedVectorElementType(type_));
0683 if (tv.type_ == FBT_STRING) {
0684
0685
0686
0687
0688
0689
0690 tv.type_ = FBT_KEY;
0691 }
0692 return tv;
0693 } else {
0694 return TypedVector::EmptyTypedVector();
0695 }
0696 }
0697
0698 FixedTypedVector AsFixedTypedVector() const {
0699 if (IsFixedTypedVector()) {
0700 uint8_t len = 0;
0701 auto vtype = ToFixedTypedVectorElementType(type_, &len);
0702 return FixedTypedVector(Indirect(), byte_width_, vtype, len);
0703 } else {
0704 return FixedTypedVector::EmptyFixedTypedVector();
0705 }
0706 }
0707
0708 Map AsMap() const {
0709 if (type_ == FBT_MAP) {
0710 return Map(Indirect(), byte_width_);
0711 } else {
0712 return Map::EmptyMap();
0713 }
0714 }
0715
0716 template<typename T> T As() const;
0717
0718
0719
0720
0721
0722
0723
0724 bool MutateInt(int64_t i) {
0725 if (type_ == FBT_INT) {
0726 return Mutate(data_, i, parent_width_, WidthI(i));
0727 } else if (type_ == FBT_INDIRECT_INT) {
0728 return Mutate(Indirect(), i, byte_width_, WidthI(i));
0729 } else if (type_ == FBT_UINT) {
0730 auto u = static_cast<uint64_t>(i);
0731 return Mutate(data_, u, parent_width_, WidthU(u));
0732 } else if (type_ == FBT_INDIRECT_UINT) {
0733 auto u = static_cast<uint64_t>(i);
0734 return Mutate(Indirect(), u, byte_width_, WidthU(u));
0735 } else {
0736 return false;
0737 }
0738 }
0739
0740 bool MutateBool(bool b) {
0741 return type_ == FBT_BOOL && Mutate(data_, b, parent_width_, BIT_WIDTH_8);
0742 }
0743
0744 bool MutateUInt(uint64_t u) {
0745 if (type_ == FBT_UINT) {
0746 return Mutate(data_, u, parent_width_, WidthU(u));
0747 } else if (type_ == FBT_INDIRECT_UINT) {
0748 return Mutate(Indirect(), u, byte_width_, WidthU(u));
0749 } else if (type_ == FBT_INT) {
0750 auto i = static_cast<int64_t>(u);
0751 return Mutate(data_, i, parent_width_, WidthI(i));
0752 } else if (type_ == FBT_INDIRECT_INT) {
0753 auto i = static_cast<int64_t>(u);
0754 return Mutate(Indirect(), i, byte_width_, WidthI(i));
0755 } else {
0756 return false;
0757 }
0758 }
0759
0760 bool MutateFloat(float f) {
0761 if (type_ == FBT_FLOAT) {
0762 return MutateF(data_, f, parent_width_, BIT_WIDTH_32);
0763 } else if (type_ == FBT_INDIRECT_FLOAT) {
0764 return MutateF(Indirect(), f, byte_width_, BIT_WIDTH_32);
0765 } else {
0766 return false;
0767 }
0768 }
0769
0770 bool MutateFloat(double d) {
0771 if (type_ == FBT_FLOAT) {
0772 return MutateF(data_, d, parent_width_, WidthF(d));
0773 } else if (type_ == FBT_INDIRECT_FLOAT) {
0774 return MutateF(Indirect(), d, byte_width_, WidthF(d));
0775 } else {
0776 return false;
0777 }
0778 }
0779
0780 bool MutateString(const char *str, size_t len) {
0781 auto s = AsString();
0782 if (s.IsTheEmptyString()) return false;
0783
0784
0785 if (s.length() != len) return false;
0786 memcpy(const_cast<char *>(s.c_str()), str, len);
0787 return true;
0788 }
0789 bool MutateString(const char *str) { return MutateString(str, strlen(str)); }
0790 bool MutateString(const std::string &str) {
0791 return MutateString(str.data(), str.length());
0792 }
0793
0794 private:
0795 const uint8_t *Indirect() const {
0796 return flexbuffers::Indirect(data_, parent_width_);
0797 }
0798
0799 template<typename T>
0800 bool Mutate(const uint8_t *dest, T t, size_t byte_width,
0801 BitWidth value_width) {
0802 auto fits = static_cast<size_t>(static_cast<size_t>(1U) << value_width) <=
0803 byte_width;
0804 if (fits) {
0805 t = flatbuffers::EndianScalar(t);
0806 memcpy(const_cast<uint8_t *>(dest), &t, byte_width);
0807 }
0808 return fits;
0809 }
0810
0811 template<typename T>
0812 bool MutateF(const uint8_t *dest, T t, size_t byte_width,
0813 BitWidth value_width) {
0814 if (byte_width == sizeof(double))
0815 return Mutate(dest, static_cast<double>(t), byte_width, value_width);
0816 if (byte_width == sizeof(float))
0817 return Mutate(dest, static_cast<float>(t), byte_width, value_width);
0818 FLATBUFFERS_ASSERT(false);
0819 return false;
0820 }
0821
0822 friend class Verifier;
0823
0824 const uint8_t *data_;
0825 uint8_t parent_width_;
0826 uint8_t byte_width_;
0827 Type type_;
0828 };
0829
0830
0831 template<> inline bool Reference::As<bool>() const { return AsBool(); }
0832
0833 template<> inline int8_t Reference::As<int8_t>() const { return AsInt8(); }
0834 template<> inline int16_t Reference::As<int16_t>() const { return AsInt16(); }
0835 template<> inline int32_t Reference::As<int32_t>() const { return AsInt32(); }
0836 template<> inline int64_t Reference::As<int64_t>() const { return AsInt64(); }
0837
0838 template<> inline uint8_t Reference::As<uint8_t>() const { return AsUInt8(); }
0839 template<> inline uint16_t Reference::As<uint16_t>() const {
0840 return AsUInt16();
0841 }
0842 template<> inline uint32_t Reference::As<uint32_t>() const {
0843 return AsUInt32();
0844 }
0845 template<> inline uint64_t Reference::As<uint64_t>() const {
0846 return AsUInt64();
0847 }
0848
0849 template<> inline double Reference::As<double>() const { return AsDouble(); }
0850 template<> inline float Reference::As<float>() const { return AsFloat(); }
0851
0852 template<> inline String Reference::As<String>() const { return AsString(); }
0853 template<> inline std::string Reference::As<std::string>() const {
0854 return AsString().str();
0855 }
0856
0857 template<> inline Blob Reference::As<Blob>() const { return AsBlob(); }
0858 template<> inline Vector Reference::As<Vector>() const { return AsVector(); }
0859 template<> inline TypedVector Reference::As<TypedVector>() const {
0860 return AsTypedVector();
0861 }
0862 template<> inline FixedTypedVector Reference::As<FixedTypedVector>() const {
0863 return AsFixedTypedVector();
0864 }
0865 template<> inline Map Reference::As<Map>() const { return AsMap(); }
0866
0867 inline uint8_t PackedType(BitWidth bit_width, Type type) {
0868 return static_cast<uint8_t>(bit_width | (type << 2));
0869 }
0870
0871 inline uint8_t NullPackedType() { return PackedType(BIT_WIDTH_8, FBT_NULL); }
0872
0873
0874
0875
0876
0877
0878
0879
0880 inline Reference Vector::operator[](size_t i) const {
0881 auto len = size();
0882 if (i >= len) return Reference(nullptr, 1, NullPackedType());
0883 auto packed_type = (data_ + len * byte_width_)[i];
0884 auto elem = data_ + i * byte_width_;
0885 return Reference(elem, byte_width_, packed_type);
0886 }
0887
0888 inline Reference TypedVector::operator[](size_t i) const {
0889 auto len = size();
0890 if (i >= len) return Reference(nullptr, 1, NullPackedType());
0891 auto elem = data_ + i * byte_width_;
0892 return Reference(elem, byte_width_, 1, type_);
0893 }
0894
0895 inline Reference FixedTypedVector::operator[](size_t i) const {
0896 if (i >= len_) return Reference(nullptr, 1, NullPackedType());
0897 auto elem = data_ + i * byte_width_;
0898 return Reference(elem, byte_width_, 1, type_);
0899 }
0900
0901 template<typename T> int KeyCompare(const void *key, const void *elem) {
0902 auto str_elem = reinterpret_cast<const char *>(
0903 Indirect<T>(reinterpret_cast<const uint8_t *>(elem)));
0904 auto skey = reinterpret_cast<const char *>(key);
0905 return strcmp(skey, str_elem);
0906 }
0907
0908 inline Reference Map::operator[](const char *key) const {
0909 auto keys = Keys();
0910
0911
0912 int (*comp)(const void *, const void *) = nullptr;
0913 switch (keys.byte_width_) {
0914 case 1: comp = KeyCompare<uint8_t>; break;
0915 case 2: comp = KeyCompare<uint16_t>; break;
0916 case 4: comp = KeyCompare<uint32_t>; break;
0917 case 8: comp = KeyCompare<uint64_t>; break;
0918 default: FLATBUFFERS_ASSERT(false); return Reference();
0919 }
0920 auto res = std::bsearch(key, keys.data_, keys.size(), keys.byte_width_, comp);
0921 if (!res) return Reference(nullptr, 1, NullPackedType());
0922 auto i = (reinterpret_cast<uint8_t *>(res) - keys.data_) / keys.byte_width_;
0923 return (*static_cast<const Vector *>(this))[i];
0924 }
0925
0926 inline Reference Map::operator[](const std::string &key) const {
0927 return (*this)[key.c_str()];
0928 }
0929
0930 inline Reference GetRoot(const uint8_t *buffer, size_t size) {
0931
0932
0933 auto end = buffer + size;
0934 auto byte_width = *--end;
0935 auto packed_type = *--end;
0936 end -= byte_width;
0937 return Reference(end, byte_width, packed_type);
0938 }
0939
0940 inline Reference GetRoot(const std::vector<uint8_t> &buffer) {
0941 return GetRoot(buffer.data(), buffer.size());
0942 }
0943
0944
0945
0946
0947
0948
0949
0950
0951
0952
0953
0954 enum BuilderFlag {
0955 BUILDER_FLAG_NONE = 0,
0956 BUILDER_FLAG_SHARE_KEYS = 1,
0957 BUILDER_FLAG_SHARE_STRINGS = 2,
0958 BUILDER_FLAG_SHARE_KEYS_AND_STRINGS = 3,
0959 BUILDER_FLAG_SHARE_KEY_VECTORS = 4,
0960 BUILDER_FLAG_SHARE_ALL = 7,
0961 };
0962
0963 class Builder FLATBUFFERS_FINAL_CLASS {
0964 public:
0965 Builder(size_t initial_size = 256,
0966 BuilderFlag flags = BUILDER_FLAG_SHARE_KEYS)
0967 : buf_(initial_size),
0968 finished_(false),
0969 has_duplicate_keys_(false),
0970 flags_(flags),
0971 force_min_bit_width_(BIT_WIDTH_8),
0972 key_pool(KeyOffsetCompare(buf_)),
0973 string_pool(StringOffsetCompare(buf_)) {
0974 buf_.clear();
0975 }
0976
0977 #ifdef FLATBUFFERS_DEFAULT_DECLARATION
0978 Builder(Builder &&) = default;
0979 Builder &operator=(Builder &&) = default;
0980 #endif
0981
0982
0983
0984 const std::vector<uint8_t> &GetBuffer() const {
0985 Finished();
0986 return buf_;
0987 }
0988
0989
0990 size_t GetSize() const { return buf_.size(); }
0991
0992
0993 void Clear() {
0994 buf_.clear();
0995 stack_.clear();
0996 finished_ = false;
0997
0998 force_min_bit_width_ = BIT_WIDTH_8;
0999 key_pool.clear();
1000 string_pool.clear();
1001 }
1002
1003
1004
1005
1006
1007 void Null() { stack_.push_back(Value()); }
1008 void Null(const char *key) {
1009 Key(key);
1010 Null();
1011 }
1012
1013 void Int(int64_t i) { stack_.push_back(Value(i, FBT_INT, WidthI(i))); }
1014 void Int(const char *key, int64_t i) {
1015 Key(key);
1016 Int(i);
1017 }
1018
1019 void UInt(uint64_t u) { stack_.push_back(Value(u, FBT_UINT, WidthU(u))); }
1020 void UInt(const char *key, uint64_t u) {
1021 Key(key);
1022 UInt(u);
1023 }
1024
1025 void Float(float f) { stack_.push_back(Value(f)); }
1026 void Float(const char *key, float f) {
1027 Key(key);
1028 Float(f);
1029 }
1030
1031 void Double(double f) { stack_.push_back(Value(f)); }
1032 void Double(const char *key, double d) {
1033 Key(key);
1034 Double(d);
1035 }
1036
1037 void Bool(bool b) { stack_.push_back(Value(b)); }
1038 void Bool(const char *key, bool b) {
1039 Key(key);
1040 Bool(b);
1041 }
1042
1043 void IndirectInt(int64_t i) { PushIndirect(i, FBT_INDIRECT_INT, WidthI(i)); }
1044 void IndirectInt(const char *key, int64_t i) {
1045 Key(key);
1046 IndirectInt(i);
1047 }
1048
1049 void IndirectUInt(uint64_t u) {
1050 PushIndirect(u, FBT_INDIRECT_UINT, WidthU(u));
1051 }
1052 void IndirectUInt(const char *key, uint64_t u) {
1053 Key(key);
1054 IndirectUInt(u);
1055 }
1056
1057 void IndirectFloat(float f) {
1058 PushIndirect(f, FBT_INDIRECT_FLOAT, BIT_WIDTH_32);
1059 }
1060 void IndirectFloat(const char *key, float f) {
1061 Key(key);
1062 IndirectFloat(f);
1063 }
1064
1065 void IndirectDouble(double f) {
1066 PushIndirect(f, FBT_INDIRECT_FLOAT, WidthF(f));
1067 }
1068 void IndirectDouble(const char *key, double d) {
1069 Key(key);
1070 IndirectDouble(d);
1071 }
1072
1073 size_t Key(const char *str, size_t len) {
1074 auto sloc = buf_.size();
1075 WriteBytes(str, len + 1);
1076 if (flags_ & BUILDER_FLAG_SHARE_KEYS) {
1077 auto it = key_pool.find(sloc);
1078 if (it != key_pool.end()) {
1079
1080
1081 buf_.resize(sloc);
1082 sloc = *it;
1083 } else {
1084 key_pool.insert(sloc);
1085 }
1086 }
1087 stack_.push_back(Value(static_cast<uint64_t>(sloc), FBT_KEY, BIT_WIDTH_8));
1088 return sloc;
1089 }
1090
1091 size_t Key(const char *str) { return Key(str, strlen(str)); }
1092 size_t Key(const std::string &str) { return Key(str.c_str(), str.size()); }
1093
1094 size_t String(const char *str, size_t len) {
1095 auto reset_to = buf_.size();
1096 auto sloc = CreateBlob(str, len, 1, FBT_STRING);
1097 if (flags_ & BUILDER_FLAG_SHARE_STRINGS) {
1098 StringOffset so(sloc, len);
1099 auto it = string_pool.find(so);
1100 if (it != string_pool.end()) {
1101
1102
1103 buf_.resize(reset_to);
1104 sloc = it->first;
1105 stack_.back().u_ = sloc;
1106 } else {
1107 string_pool.insert(so);
1108 }
1109 }
1110 return sloc;
1111 }
1112 size_t String(const char *str) { return String(str, strlen(str)); }
1113 size_t String(const std::string &str) {
1114 return String(str.c_str(), str.size());
1115 }
1116 void String(const flexbuffers::String &str) {
1117 String(str.c_str(), str.length());
1118 }
1119
1120 void String(const char *key, const char *str) {
1121 Key(key);
1122 String(str);
1123 }
1124 void String(const char *key, const std::string &str) {
1125 Key(key);
1126 String(str);
1127 }
1128 void String(const char *key, const flexbuffers::String &str) {
1129 Key(key);
1130 String(str);
1131 }
1132
1133 size_t Blob(const void *data, size_t len) {
1134 return CreateBlob(data, len, 0, FBT_BLOB);
1135 }
1136 size_t Blob(const std::vector<uint8_t> &v) {
1137 return CreateBlob(v.data(), v.size(), 0, FBT_BLOB);
1138 }
1139
1140 void Blob(const char *key, const void *data, size_t len) {
1141 Key(key);
1142 Blob(data, len);
1143 }
1144 void Blob(const char *key, const std::vector<uint8_t> &v) {
1145 Key(key);
1146 Blob(v);
1147 }
1148
1149
1150
1151
1152
1153 size_t StartVector() { return stack_.size(); }
1154 size_t StartVector(const char *key) {
1155 Key(key);
1156 return stack_.size();
1157 }
1158 size_t StartMap() { return stack_.size(); }
1159 size_t StartMap(const char *key) {
1160 Key(key);
1161 return stack_.size();
1162 }
1163
1164
1165
1166 size_t EndVector(size_t start, bool typed, bool fixed) {
1167 auto vec = CreateVector(start, stack_.size() - start, 1, typed, fixed);
1168
1169 stack_.resize(start);
1170 stack_.push_back(vec);
1171 return static_cast<size_t>(vec.u_);
1172 }
1173
1174 size_t EndMap(size_t start) {
1175
1176 auto len = MapElementCount(start);
1177
1178 for (auto key = start; key < stack_.size(); key += 2) {
1179 FLATBUFFERS_ASSERT(stack_[key].type_ == FBT_KEY);
1180 }
1181
1182
1183 struct TwoValue {
1184 Value key;
1185 Value val;
1186 };
1187
1188
1189
1190
1191
1192
1193
1194
1195
1196 auto dict = reinterpret_cast<TwoValue *>(stack_.data() + start);
1197 std::sort(
1198 dict, dict + len, [&](const TwoValue &a, const TwoValue &b) -> bool {
1199 auto as = reinterpret_cast<const char *>(buf_.data() + a.key.u_);
1200 auto bs = reinterpret_cast<const char *>(buf_.data() + b.key.u_);
1201 auto comp = strcmp(as, bs);
1202
1203
1204
1205
1206
1207
1208
1209
1210
1211 if (!comp && &a != &b) has_duplicate_keys_ = true;
1212 return comp < 0;
1213 });
1214
1215
1216
1217 auto keys = CreateVector(start, len, 2, true, false);
1218 auto vec = CreateVector(start + 1, len, 2, false, false, &keys);
1219
1220 stack_.resize(start);
1221 stack_.push_back(vec);
1222 return static_cast<size_t>(vec.u_);
1223 }
1224
1225
1226
1227 bool HasDuplicateKeys() const { return has_duplicate_keys_; }
1228
1229 template<typename F> size_t Vector(F f) {
1230 auto start = StartVector();
1231 f();
1232 return EndVector(start, false, false);
1233 }
1234 template<typename F, typename T> size_t Vector(F f, T &state) {
1235 auto start = StartVector();
1236 f(state);
1237 return EndVector(start, false, false);
1238 }
1239 template<typename F> size_t Vector(const char *key, F f) {
1240 auto start = StartVector(key);
1241 f();
1242 return EndVector(start, false, false);
1243 }
1244 template<typename F, typename T>
1245 size_t Vector(const char *key, F f, T &state) {
1246 auto start = StartVector(key);
1247 f(state);
1248 return EndVector(start, false, false);
1249 }
1250
1251 template<typename T> void Vector(const T *elems, size_t len) {
1252 if (flatbuffers::is_scalar<T>::value) {
1253
1254 ScalarVector(elems, len, false);
1255 } else {
1256 auto start = StartVector();
1257 for (size_t i = 0; i < len; i++) Add(elems[i]);
1258 EndVector(start, false, false);
1259 }
1260 }
1261 template<typename T>
1262 void Vector(const char *key, const T *elems, size_t len) {
1263 Key(key);
1264 Vector(elems, len);
1265 }
1266 template<typename T> void Vector(const std::vector<T> &vec) {
1267 Vector(vec.data(), vec.size());
1268 }
1269
1270 template<typename F> size_t TypedVector(F f) {
1271 auto start = StartVector();
1272 f();
1273 return EndVector(start, true, false);
1274 }
1275 template<typename F, typename T> size_t TypedVector(F f, T &state) {
1276 auto start = StartVector();
1277 f(state);
1278 return EndVector(start, true, false);
1279 }
1280 template<typename F> size_t TypedVector(const char *key, F f) {
1281 auto start = StartVector(key);
1282 f();
1283 return EndVector(start, true, false);
1284 }
1285 template<typename F, typename T>
1286 size_t TypedVector(const char *key, F f, T &state) {
1287 auto start = StartVector(key);
1288 f(state);
1289 return EndVector(start, true, false);
1290 }
1291
1292 template<typename T> size_t FixedTypedVector(const T *elems, size_t len) {
1293
1294
1295 FLATBUFFERS_ASSERT(len >= 2 && len <= 4);
1296
1297 static_assert(flatbuffers::is_scalar<T>::value, "Unrelated types");
1298 return ScalarVector(elems, len, true);
1299 }
1300
1301 template<typename T>
1302 size_t FixedTypedVector(const char *key, const T *elems, size_t len) {
1303 Key(key);
1304 return FixedTypedVector(elems, len);
1305 }
1306
1307 template<typename F> size_t Map(F f) {
1308 auto start = StartMap();
1309 f();
1310 return EndMap(start);
1311 }
1312 template<typename F, typename T> size_t Map(F f, T &state) {
1313 auto start = StartMap();
1314 f(state);
1315 return EndMap(start);
1316 }
1317 template<typename F> size_t Map(const char *key, F f) {
1318 auto start = StartMap(key);
1319 f();
1320 return EndMap(start);
1321 }
1322 template<typename F, typename T> size_t Map(const char *key, F f, T &state) {
1323 auto start = StartMap(key);
1324 f(state);
1325 return EndMap(start);
1326 }
1327 template<typename T> void Map(const std::map<std::string, T> &map) {
1328 auto start = StartMap();
1329 for (auto it = map.begin(); it != map.end(); ++it)
1330 Add(it->first.c_str(), it->second);
1331 EndMap(start);
1332 }
1333
1334 size_t MapElementCount(size_t start) {
1335
1336 auto len = stack_.size() - start;
1337 FLATBUFFERS_ASSERT(!(len & 1));
1338 len /= 2;
1339 return len;
1340 }
1341
1342
1343
1344
1345
1346
1347
1348
1349
1350
1351
1352 struct Value;
1353 Value LastValue() { return stack_.back(); }
1354 void ReuseValue(Value v) { stack_.push_back(v); }
1355 void ReuseValue(const char *key, Value v) {
1356 Key(key);
1357 ReuseValue(v);
1358 }
1359
1360
1361
1362 void Undo() {
1363 stack_.pop_back();
1364 }
1365
1366
1367 void Add(int8_t i) { Int(i); }
1368 void Add(int16_t i) { Int(i); }
1369 void Add(int32_t i) { Int(i); }
1370 void Add(int64_t i) { Int(i); }
1371 void Add(uint8_t u) { UInt(u); }
1372 void Add(uint16_t u) { UInt(u); }
1373 void Add(uint32_t u) { UInt(u); }
1374 void Add(uint64_t u) { UInt(u); }
1375 void Add(float f) { Float(f); }
1376 void Add(double d) { Double(d); }
1377 void Add(bool b) { Bool(b); }
1378 void Add(const char *str) { String(str); }
1379 void Add(const std::string &str) { String(str); }
1380 void Add(const flexbuffers::String &str) { String(str); }
1381
1382 template<typename T> void Add(const std::vector<T> &vec) { Vector(vec); }
1383
1384 template<typename T> void Add(const char *key, const T &t) {
1385 Key(key);
1386 Add(t);
1387 }
1388
1389 template<typename T> void Add(const std::map<std::string, T> &map) {
1390 Map(map);
1391 }
1392
1393 template<typename T> void operator+=(const T &t) { Add(t); }
1394
1395
1396
1397
1398
1399 void ForceMinimumBitWidth(BitWidth bw = BIT_WIDTH_8) {
1400 force_min_bit_width_ = bw;
1401 }
1402
1403 void Finish() {
1404
1405
1406
1407
1408 FLATBUFFERS_ASSERT(stack_.size() == 1);
1409
1410
1411 auto byte_width = Align(stack_[0].ElemWidth(buf_.size(), 0));
1412 WriteAny(stack_[0], byte_width);
1413
1414 Write(stack_[0].StoredPackedType(), 1);
1415
1416 Write(byte_width, 1);
1417
1418 finished_ = true;
1419 }
1420
1421 private:
1422 void Finished() const {
1423
1424
1425
1426 FLATBUFFERS_ASSERT(finished_);
1427 }
1428
1429
1430 uint8_t Align(BitWidth alignment) {
1431 auto byte_width = 1U << alignment;
1432 buf_.insert(buf_.end(), flatbuffers::PaddingBytes(buf_.size(), byte_width),
1433 0);
1434 return static_cast<uint8_t>(byte_width);
1435 }
1436
1437 void WriteBytes(const void *val, size_t size) {
1438 buf_.insert(buf_.end(), reinterpret_cast<const uint8_t *>(val),
1439 reinterpret_cast<const uint8_t *>(val) + size);
1440 }
1441
1442 template<typename T> void Write(T val, size_t byte_width) {
1443 FLATBUFFERS_ASSERT(sizeof(T) >= byte_width);
1444 val = flatbuffers::EndianScalar(val);
1445 WriteBytes(&val, byte_width);
1446 }
1447
1448 void WriteDouble(double f, uint8_t byte_width) {
1449 switch (byte_width) {
1450 case 8: Write(f, byte_width); break;
1451 case 4: Write(static_cast<float>(f), byte_width); break;
1452
1453
1454 default: FLATBUFFERS_ASSERT(0);
1455 }
1456 }
1457
1458 void WriteOffset(uint64_t o, uint8_t byte_width) {
1459 auto reloff = buf_.size() - o;
1460 FLATBUFFERS_ASSERT(byte_width == 8 || reloff < 1ULL << (byte_width * 8));
1461 Write(reloff, byte_width);
1462 }
1463
1464 template<typename T> void PushIndirect(T val, Type type, BitWidth bit_width) {
1465 auto byte_width = Align(bit_width);
1466 auto iloc = buf_.size();
1467 Write(val, byte_width);
1468 stack_.push_back(Value(static_cast<uint64_t>(iloc), type, bit_width));
1469 }
1470
1471 static BitWidth WidthB(size_t byte_width) {
1472 switch (byte_width) {
1473 case 1: return BIT_WIDTH_8;
1474 case 2: return BIT_WIDTH_16;
1475 case 4: return BIT_WIDTH_32;
1476 case 8: return BIT_WIDTH_64;
1477 default: FLATBUFFERS_ASSERT(false); return BIT_WIDTH_64;
1478 }
1479 }
1480
1481 template<typename T> static Type GetScalarType() {
1482 static_assert(flatbuffers::is_scalar<T>::value, "Unrelated types");
1483 return flatbuffers::is_floating_point<T>::value ? FBT_FLOAT
1484 : flatbuffers::is_same<T, bool>::value
1485 ? FBT_BOOL
1486 : (flatbuffers::is_unsigned<T>::value ? FBT_UINT : FBT_INT);
1487 }
1488
1489 public:
1490
1491 struct Value {
1492 union {
1493 int64_t i_;
1494 uint64_t u_;
1495 double f_;
1496 };
1497
1498 Type type_;
1499
1500
1501 BitWidth min_bit_width_;
1502
1503 Value() : i_(0), type_(FBT_NULL), min_bit_width_(BIT_WIDTH_8) {}
1504
1505 Value(bool b)
1506 : u_(static_cast<uint64_t>(b)),
1507 type_(FBT_BOOL),
1508 min_bit_width_(BIT_WIDTH_8) {}
1509
1510 Value(int64_t i, Type t, BitWidth bw)
1511 : i_(i), type_(t), min_bit_width_(bw) {}
1512 Value(uint64_t u, Type t, BitWidth bw)
1513 : u_(u), type_(t), min_bit_width_(bw) {}
1514
1515 Value(float f)
1516 : f_(static_cast<double>(f)),
1517 type_(FBT_FLOAT),
1518 min_bit_width_(BIT_WIDTH_32) {}
1519 Value(double f) : f_(f), type_(FBT_FLOAT), min_bit_width_(WidthF(f)) {}
1520
1521 uint8_t StoredPackedType(BitWidth parent_bit_width_ = BIT_WIDTH_8) const {
1522 return PackedType(StoredWidth(parent_bit_width_), type_);
1523 }
1524
1525 BitWidth ElemWidth(size_t buf_size, size_t elem_index) const {
1526 if (IsInline(type_)) {
1527 return min_bit_width_;
1528 } else {
1529
1530
1531
1532
1533
1534 for (size_t byte_width = 1;
1535 byte_width <= sizeof(flatbuffers::largest_scalar_t);
1536 byte_width *= 2) {
1537
1538 auto offset_loc = buf_size +
1539 flatbuffers::PaddingBytes(buf_size, byte_width) +
1540 elem_index * byte_width;
1541
1542 auto offset = offset_loc - u_;
1543
1544 auto bit_width = WidthU(offset);
1545 if (static_cast<size_t>(static_cast<size_t>(1U) << bit_width) ==
1546 byte_width)
1547 return bit_width;
1548 }
1549 FLATBUFFERS_ASSERT(false);
1550 return BIT_WIDTH_64;
1551 }
1552 }
1553
1554 BitWidth StoredWidth(BitWidth parent_bit_width_ = BIT_WIDTH_8) const {
1555 if (IsInline(type_)) {
1556 return (std::max)(min_bit_width_, parent_bit_width_);
1557 } else {
1558 return min_bit_width_;
1559 }
1560 }
1561 };
1562
1563 private:
1564 void WriteAny(const Value &val, uint8_t byte_width) {
1565 switch (val.type_) {
1566 case FBT_NULL:
1567 case FBT_INT: Write(val.i_, byte_width); break;
1568 case FBT_BOOL:
1569 case FBT_UINT: Write(val.u_, byte_width); break;
1570 case FBT_FLOAT: WriteDouble(val.f_, byte_width); break;
1571 default: WriteOffset(val.u_, byte_width); break;
1572 }
1573 }
1574
1575 size_t CreateBlob(const void *data, size_t len, size_t trailing, Type type) {
1576 auto bit_width = WidthU(len);
1577 auto byte_width = Align(bit_width);
1578 Write<uint64_t>(len, byte_width);
1579 auto sloc = buf_.size();
1580 WriteBytes(data, len + trailing);
1581 stack_.push_back(Value(static_cast<uint64_t>(sloc), type, bit_width));
1582 return sloc;
1583 }
1584
1585 template<typename T>
1586 size_t ScalarVector(const T *elems, size_t len, bool fixed) {
1587 auto vector_type = GetScalarType<T>();
1588 auto byte_width = sizeof(T);
1589 auto bit_width = WidthB(byte_width);
1590
1591
1592
1593
1594
1595 FLATBUFFERS_ASSERT(WidthU(len) <= bit_width);
1596 Align(bit_width);
1597 if (!fixed) Write<uint64_t>(len, byte_width);
1598 auto vloc = buf_.size();
1599 for (size_t i = 0; i < len; i++) Write(elems[i], byte_width);
1600 stack_.push_back(Value(static_cast<uint64_t>(vloc),
1601 ToTypedVector(vector_type, fixed ? len : 0),
1602 bit_width));
1603 return vloc;
1604 }
1605
1606 Value CreateVector(size_t start, size_t vec_len, size_t step, bool typed,
1607 bool fixed, const Value *keys = nullptr) {
1608 FLATBUFFERS_ASSERT(
1609 !fixed ||
1610 typed);
1611
1612 auto bit_width = (std::max)(force_min_bit_width_, WidthU(vec_len));
1613 auto prefix_elems = 1;
1614 if (keys) {
1615
1616
1617 bit_width = (std::max)(bit_width, keys->ElemWidth(buf_.size(), 0));
1618 prefix_elems += 2;
1619 }
1620 Type vector_type = FBT_KEY;
1621
1622 for (size_t i = start; i < stack_.size(); i += step) {
1623 auto elem_width =
1624 stack_[i].ElemWidth(buf_.size(), i - start + prefix_elems);
1625 bit_width = (std::max)(bit_width, elem_width);
1626 if (typed) {
1627 if (i == start) {
1628 vector_type = stack_[i].type_;
1629 } else {
1630
1631
1632 FLATBUFFERS_ASSERT(vector_type == stack_[i].type_);
1633 }
1634 }
1635 }
1636
1637
1638 FLATBUFFERS_ASSERT(!typed || IsTypedVectorElementType(vector_type));
1639 auto byte_width = Align(bit_width);
1640
1641 if (keys) {
1642 WriteOffset(keys->u_, byte_width);
1643 Write<uint64_t>(1ULL << keys->min_bit_width_, byte_width);
1644 }
1645 if (!fixed) Write<uint64_t>(vec_len, byte_width);
1646
1647 auto vloc = buf_.size();
1648 for (size_t i = start; i < stack_.size(); i += step) {
1649 WriteAny(stack_[i], byte_width);
1650 }
1651
1652 if (!typed) {
1653 for (size_t i = start; i < stack_.size(); i += step) {
1654 buf_.push_back(stack_[i].StoredPackedType(bit_width));
1655 }
1656 }
1657 return Value(static_cast<uint64_t>(vloc),
1658 keys ? FBT_MAP
1659 : (typed ? ToTypedVector(vector_type, fixed ? vec_len : 0)
1660 : FBT_VECTOR),
1661 bit_width);
1662 }
1663
1664
1665 Builder(const Builder &);
1666 Builder &operator=(const Builder &);
1667
1668 std::vector<uint8_t> buf_;
1669 std::vector<Value> stack_;
1670
1671 bool finished_;
1672 bool has_duplicate_keys_;
1673
1674 BuilderFlag flags_;
1675
1676 BitWidth force_min_bit_width_;
1677
1678 struct KeyOffsetCompare {
1679 explicit KeyOffsetCompare(const std::vector<uint8_t> &buf) : buf_(&buf) {}
1680 bool operator()(size_t a, size_t b) const {
1681 auto stra = reinterpret_cast<const char *>(buf_->data() + a);
1682 auto strb = reinterpret_cast<const char *>(buf_->data() + b);
1683 return strcmp(stra, strb) < 0;
1684 }
1685 const std::vector<uint8_t> *buf_;
1686 };
1687
1688 typedef std::pair<size_t, size_t> StringOffset;
1689 struct StringOffsetCompare {
1690 explicit StringOffsetCompare(const std::vector<uint8_t> &buf)
1691 : buf_(&buf) {}
1692 bool operator()(const StringOffset &a, const StringOffset &b) const {
1693 auto stra = buf_->data() + a.first;
1694 auto strb = buf_->data() + b.first;
1695 auto cr = memcmp(stra, strb, (std::min)(a.second, b.second) + 1);
1696 return cr < 0 || (cr == 0 && a.second < b.second);
1697 }
1698 const std::vector<uint8_t> *buf_;
1699 };
1700
1701 typedef std::set<size_t, KeyOffsetCompare> KeyOffsetMap;
1702 typedef std::set<StringOffset, StringOffsetCompare> StringOffsetMap;
1703
1704 KeyOffsetMap key_pool;
1705 StringOffsetMap string_pool;
1706
1707 friend class Verifier;
1708 };
1709
1710
1711 class Verifier FLATBUFFERS_FINAL_CLASS {
1712 public:
1713 Verifier(const uint8_t *buf, size_t buf_len,
1714
1715
1716
1717
1718 std::vector<uint8_t> *reuse_tracker = nullptr,
1719 bool _check_alignment = true, size_t max_depth = 64)
1720 : buf_(buf),
1721 size_(buf_len),
1722 depth_(0),
1723 max_depth_(max_depth),
1724 num_vectors_(0),
1725 max_vectors_(buf_len),
1726 check_alignment_(_check_alignment),
1727 reuse_tracker_(reuse_tracker) {
1728 FLATBUFFERS_ASSERT(static_cast<int32_t>(size_) < FLATBUFFERS_MAX_BUFFER_SIZE);
1729 if (reuse_tracker_) {
1730 reuse_tracker_->clear();
1731 reuse_tracker_->resize(size_, PackedType(BIT_WIDTH_8, FBT_NULL));
1732 }
1733 }
1734
1735 private:
1736
1737 bool Check(bool ok) const {
1738
1739 #ifdef FLATBUFFERS_DEBUG_VERIFICATION_FAILURE
1740 FLATBUFFERS_ASSERT(ok);
1741 #endif
1742
1743 return ok;
1744 }
1745
1746
1747 bool VerifyFrom(size_t elem, size_t elem_len) const {
1748 return Check(elem_len < size_ && elem <= size_ - elem_len);
1749 }
1750 bool VerifyBefore(size_t elem, size_t elem_len) const {
1751 return Check(elem_len <= elem);
1752 }
1753
1754 bool VerifyFromPointer(const uint8_t *p, size_t len) {
1755 auto o = static_cast<size_t>(p - buf_);
1756 return VerifyFrom(o, len);
1757 }
1758 bool VerifyBeforePointer(const uint8_t *p, size_t len) {
1759 auto o = static_cast<size_t>(p - buf_);
1760 return VerifyBefore(o, len);
1761 }
1762
1763 bool VerifyByteWidth(size_t width) {
1764 return Check(width == 1 || width == 2 || width == 4 || width == 8);
1765 }
1766
1767 bool VerifyType(int type) { return Check(type >= 0 && type < FBT_MAX_TYPE); }
1768
1769 bool VerifyOffset(uint64_t off, const uint8_t *p) {
1770 return Check(off <= static_cast<uint64_t>(size_)) &&
1771 off <= static_cast<uint64_t>(p - buf_);
1772 }
1773
1774 bool VerifyAlignment(const uint8_t *p, size_t size) const {
1775 auto o = static_cast<size_t>(p - buf_);
1776 return Check((o & (size - 1)) == 0 || !check_alignment_);
1777 }
1778
1779
1780 #define FLEX_CHECK_VERIFIED(P, PACKED_TYPE) \
1781 if (reuse_tracker_) { \
1782 auto packed_type = PACKED_TYPE; \
1783 auto existing = (*reuse_tracker_)[P - buf_]; \
1784 if (existing == packed_type) return true; \
1785 \
1786 if (!Check(existing == 0)) return false; \
1787 (*reuse_tracker_)[P - buf_] = packed_type; \
1788 }
1789
1790 bool VerifyVector(Reference r, const uint8_t *p, Type elem_type) {
1791
1792
1793 depth_++;
1794 num_vectors_++;
1795 if (!Check(depth_ <= max_depth_ && num_vectors_ <= max_vectors_))
1796 return false;
1797 auto size_byte_width = r.byte_width_;
1798 if (!VerifyBeforePointer(p, size_byte_width)) return false;
1799 FLEX_CHECK_VERIFIED(p - size_byte_width,
1800 PackedType(Builder::WidthB(size_byte_width), r.type_));
1801 auto sized = Sized(p, size_byte_width);
1802 auto num_elems = sized.size();
1803 auto elem_byte_width = r.type_ == FBT_STRING || r.type_ == FBT_BLOB
1804 ? uint8_t(1)
1805 : r.byte_width_;
1806 auto max_elems = SIZE_MAX / elem_byte_width;
1807 if (!Check(num_elems < max_elems))
1808 return false;
1809 auto byte_size = num_elems * elem_byte_width;
1810 if (!VerifyFromPointer(p, byte_size)) return false;
1811 if (elem_type == FBT_NULL) {
1812
1813 if (!VerifyFromPointer(p + byte_size, num_elems)) return false;
1814 auto v = Vector(p, size_byte_width);
1815 for (size_t i = 0; i < num_elems; i++)
1816 if (!VerifyRef(v[i])) return false;
1817 } else if (elem_type == FBT_KEY) {
1818 auto v = TypedVector(p, elem_byte_width, FBT_KEY);
1819 for (size_t i = 0; i < num_elems; i++)
1820 if (!VerifyRef(v[i])) return false;
1821 } else {
1822 FLATBUFFERS_ASSERT(IsInline(elem_type));
1823 }
1824 depth_--;
1825 return true;
1826 }
1827
1828 bool VerifyKeys(const uint8_t *p, uint8_t byte_width) {
1829
1830 const size_t num_prefixed_fields = 3;
1831 if (!VerifyBeforePointer(p, byte_width * num_prefixed_fields)) return false;
1832 p -= byte_width * num_prefixed_fields;
1833 auto off = ReadUInt64(p, byte_width);
1834 if (!VerifyOffset(off, p)) return false;
1835 auto key_byte_with =
1836 static_cast<uint8_t>(ReadUInt64(p + byte_width, byte_width));
1837 if (!VerifyByteWidth(key_byte_with)) return false;
1838 return VerifyVector(Reference(p, byte_width, key_byte_with, FBT_VECTOR_KEY),
1839 p - off, FBT_KEY);
1840 }
1841
1842 bool VerifyKey(const uint8_t *p) {
1843 FLEX_CHECK_VERIFIED(p, PackedType(BIT_WIDTH_8, FBT_KEY));
1844 while (p < buf_ + size_)
1845 if (*p++) return true;
1846 return false;
1847 }
1848
1849 #undef FLEX_CHECK_VERIFIED
1850
1851 bool VerifyTerminator(const String &s) {
1852 return VerifyFromPointer(reinterpret_cast<const uint8_t *>(s.c_str()),
1853 s.size() + 1);
1854 }
1855
1856 bool VerifyRef(Reference r) {
1857
1858 if (!VerifyByteWidth(r.byte_width_) || !VerifyType(r.type_)) {
1859 return false;
1860 }
1861 if (IsInline(r.type_)) {
1862
1863 return true;
1864 }
1865
1866 auto off = ReadUInt64(r.data_, r.parent_width_);
1867 if (!VerifyOffset(off, r.data_)) return false;
1868 auto p = r.Indirect();
1869 if (!VerifyAlignment(p, r.byte_width_)) return false;
1870 switch (r.type_) {
1871 case FBT_INDIRECT_INT:
1872 case FBT_INDIRECT_UINT:
1873 case FBT_INDIRECT_FLOAT: return VerifyFromPointer(p, r.byte_width_);
1874 case FBT_KEY: return VerifyKey(p);
1875 case FBT_MAP:
1876 return VerifyVector(r, p, FBT_NULL) && VerifyKeys(p, r.byte_width_);
1877 case FBT_VECTOR: return VerifyVector(r, p, FBT_NULL);
1878 case FBT_VECTOR_INT: return VerifyVector(r, p, FBT_INT);
1879 case FBT_VECTOR_BOOL:
1880 case FBT_VECTOR_UINT: return VerifyVector(r, p, FBT_UINT);
1881 case FBT_VECTOR_FLOAT: return VerifyVector(r, p, FBT_FLOAT);
1882 case FBT_VECTOR_KEY: return VerifyVector(r, p, FBT_KEY);
1883 case FBT_VECTOR_STRING_DEPRECATED:
1884
1885 return VerifyVector(r, p, FBT_KEY);
1886 case FBT_BLOB: return VerifyVector(r, p, FBT_UINT);
1887 case FBT_STRING:
1888 return VerifyVector(r, p, FBT_UINT) &&
1889 VerifyTerminator(String(p, r.byte_width_));
1890 case FBT_VECTOR_INT2:
1891 case FBT_VECTOR_UINT2:
1892 case FBT_VECTOR_FLOAT2:
1893 case FBT_VECTOR_INT3:
1894 case FBT_VECTOR_UINT3:
1895 case FBT_VECTOR_FLOAT3:
1896 case FBT_VECTOR_INT4:
1897 case FBT_VECTOR_UINT4:
1898 case FBT_VECTOR_FLOAT4: {
1899 uint8_t len = 0;
1900 auto vtype = ToFixedTypedVectorElementType(r.type_, &len);
1901 if (!VerifyType(vtype)) return false;
1902 return VerifyFromPointer(p, static_cast<size_t>(r.byte_width_) * len);
1903 }
1904 default: return false;
1905 }
1906 }
1907
1908 public:
1909 bool VerifyBuffer() {
1910 if (!Check(size_ >= 3)) return false;
1911 auto end = buf_ + size_;
1912 auto byte_width = *--end;
1913 auto packed_type = *--end;
1914 return VerifyByteWidth(byte_width) && Check(end - buf_ >= byte_width) &&
1915 VerifyRef(Reference(end - byte_width, byte_width, packed_type));
1916 }
1917
1918 private:
1919 const uint8_t *buf_;
1920 size_t size_;
1921 size_t depth_;
1922 const size_t max_depth_;
1923 size_t num_vectors_;
1924 const size_t max_vectors_;
1925 bool check_alignment_;
1926 std::vector<uint8_t> *reuse_tracker_;
1927 };
1928
1929
1930
1931 inline bool VerifyBuffer(const uint8_t *buf, size_t buf_len,
1932 std::vector<uint8_t> *reuse_tracker = nullptr) {
1933 Verifier verifier(buf, buf_len, reuse_tracker);
1934 return verifier.VerifyBuffer();
1935 }
1936
1937 }
1938
1939 #if defined(_MSC_VER)
1940 # pragma warning(pop)
1941 #endif
1942
1943 #endif