protobuf.h 23 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611
  1. // Protocol Buffers - Google's data interchange format
  2. // Copyright 2014 Google Inc. All rights reserved.
  3. // https://developers.google.com/protocol-buffers/
  4. //
  5. // Redistribution and use in source and binary forms, with or without
  6. // modification, are permitted provided that the following conditions are
  7. // met:
  8. //
  9. // * Redistributions of source code must retain the above copyright
  10. // notice, this list of conditions and the following disclaimer.
  11. // * Redistributions in binary form must reproduce the above
  12. // copyright notice, this list of conditions and the following disclaimer
  13. // in the documentation and/or other materials provided with the
  14. // distribution.
  15. // * Neither the name of Google Inc. nor the names of its
  16. // contributors may be used to endorse or promote products derived from
  17. // this software without specific prior written permission.
  18. //
  19. // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  20. // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  21. // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  22. // A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
  23. // OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  24. // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  25. // LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  26. // DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  27. // THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  28. // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  29. // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  30. #ifndef __GOOGLE_PROTOBUF_RUBY_PROTOBUF_H__
  31. #define __GOOGLE_PROTOBUF_RUBY_PROTOBUF_H__
  32. #include <ruby/ruby.h>
  33. #include <ruby/vm.h>
  34. #include <ruby/encoding.h>
  35. #include "upb.h"
  36. // Forward decls.
  37. struct DescriptorPool;
  38. struct Descriptor;
  39. struct FileDescriptor;
  40. struct FieldDescriptor;
  41. struct EnumDescriptor;
  42. struct MessageLayout;
  43. struct MessageField;
  44. struct MessageHeader;
  45. struct MessageBuilderContext;
  46. struct EnumBuilderContext;
  47. struct FileBuilderContext;
  48. struct Builder;
  49. typedef struct DescriptorPool DescriptorPool;
  50. typedef struct Descriptor Descriptor;
  51. typedef struct FileDescriptor FileDescriptor;
  52. typedef struct FieldDescriptor FieldDescriptor;
  53. typedef struct OneofDescriptor OneofDescriptor;
  54. typedef struct EnumDescriptor EnumDescriptor;
  55. typedef struct MessageLayout MessageLayout;
  56. typedef struct MessageField MessageField;
  57. typedef struct MessageHeader MessageHeader;
  58. typedef struct MessageBuilderContext MessageBuilderContext;
  59. typedef struct OneofBuilderContext OneofBuilderContext;
  60. typedef struct EnumBuilderContext EnumBuilderContext;
  61. typedef struct FileBuilderContext FileBuilderContext;
  62. typedef struct Builder Builder;
  63. /*
  64. It can be a bit confusing how the C structs defined below and the Ruby
  65. objects interact and hold references to each other. First, a few principles:
  66. - Ruby's "TypedData" abstraction lets a Ruby VALUE hold a pointer to a C
  67. struct (or arbitrary memory chunk), own it, and free it when collected.
  68. Thus, each struct below will have a corresponding Ruby object
  69. wrapping/owning it.
  70. - To get back from an underlying upb {msg,enum}def to the Ruby object, we
  71. keep a global hashmap, accessed by get_def_obj/add_def_obj below.
  72. The in-memory structure is then something like:
  73. Ruby | upb
  74. |
  75. DescriptorPool ------------|-----------> upb_symtab____________________
  76. | | (message types) \
  77. | v \
  78. Descriptor ---------------|-----------> upb_msgdef (enum types)|
  79. |--> msgclass | | ^ |
  80. | (dynamically built) | | | (submsg fields) |
  81. |--> MessageLayout | | | /
  82. |--------------------------|> decoder method| | /
  83. \--------------------------|> serialize | | /
  84. | handlers v | /
  85. FieldDescriptor -----------|-----------> upb_fielddef /
  86. | | /
  87. | v (enum fields) /
  88. EnumDescriptor ------------|-----------> upb_enumdef <----------'
  89. |
  90. |
  91. ^ | \___/
  92. `---------------|-----------------' (get_def_obj map)
  93. */
  94. // -----------------------------------------------------------------------------
  95. // Ruby class structure definitions.
  96. // -----------------------------------------------------------------------------
  97. struct DescriptorPool {
  98. upb_symtab* symtab;
  99. };
  100. struct Descriptor {
  101. const upb_msgdef* msgdef;
  102. MessageLayout* layout;
  103. VALUE klass; // begins as nil
  104. const upb_handlers* fill_handlers;
  105. const upb_pbdecodermethod* fill_method;
  106. const upb_json_parsermethod* json_fill_method;
  107. const upb_handlers* pb_serialize_handlers;
  108. const upb_handlers* json_serialize_handlers;
  109. const upb_handlers* json_serialize_handlers_preserve;
  110. };
  111. struct FileDescriptor {
  112. const upb_filedef* filedef;
  113. };
  114. struct FieldDescriptor {
  115. const upb_fielddef* fielddef;
  116. };
  117. struct OneofDescriptor {
  118. const upb_oneofdef* oneofdef;
  119. };
  120. struct EnumDescriptor {
  121. const upb_enumdef* enumdef;
  122. VALUE module; // begins as nil
  123. };
  124. struct MessageBuilderContext {
  125. VALUE descriptor;
  126. VALUE builder;
  127. };
  128. struct OneofBuilderContext {
  129. VALUE descriptor;
  130. VALUE builder;
  131. };
  132. struct EnumBuilderContext {
  133. VALUE enumdesc;
  134. };
  135. struct FileBuilderContext {
  136. VALUE pending_list;
  137. VALUE file_descriptor;
  138. VALUE builder;
  139. };
  140. struct Builder {
  141. VALUE pending_list;
  142. VALUE default_file_descriptor;
  143. upb_def** defs; // used only while finalizing
  144. };
  145. extern VALUE cDescriptorPool;
  146. extern VALUE cDescriptor;
  147. extern VALUE cFileDescriptor;
  148. extern VALUE cFieldDescriptor;
  149. extern VALUE cEnumDescriptor;
  150. extern VALUE cMessageBuilderContext;
  151. extern VALUE cOneofBuilderContext;
  152. extern VALUE cEnumBuilderContext;
  153. extern VALUE cFileBuilderContext;
  154. extern VALUE cBuilder;
  155. extern VALUE cError;
  156. extern VALUE cParseError;
  157. extern VALUE cTypeError;
  158. // We forward-declare all of the Ruby method implementations here because we
  159. // sometimes call the methods directly across .c files, rather than going
  160. // through Ruby's method dispatching (e.g. during message parse). It's cleaner
  161. // to keep the list of object methods together than to split them between
  162. // static-in-file definitions and header declarations.
  163. void DescriptorPool_mark(void* _self);
  164. void DescriptorPool_free(void* _self);
  165. VALUE DescriptorPool_alloc(VALUE klass);
  166. void DescriptorPool_register(VALUE module);
  167. DescriptorPool* ruby_to_DescriptorPool(VALUE value);
  168. VALUE DescriptorPool_add(VALUE _self, VALUE def);
  169. VALUE DescriptorPool_build(int argc, VALUE* argv, VALUE _self);
  170. VALUE DescriptorPool_lookup(VALUE _self, VALUE name);
  171. VALUE DescriptorPool_generated_pool(VALUE _self);
  172. void Descriptor_mark(void* _self);
  173. void Descriptor_free(void* _self);
  174. VALUE Descriptor_alloc(VALUE klass);
  175. void Descriptor_register(VALUE module);
  176. Descriptor* ruby_to_Descriptor(VALUE value);
  177. VALUE Descriptor_initialize(VALUE _self, VALUE file_descriptor_rb);
  178. VALUE Descriptor_name(VALUE _self);
  179. VALUE Descriptor_name_set(VALUE _self, VALUE str);
  180. VALUE Descriptor_each(VALUE _self);
  181. VALUE Descriptor_lookup(VALUE _self, VALUE name);
  182. VALUE Descriptor_add_field(VALUE _self, VALUE obj);
  183. VALUE Descriptor_add_oneof(VALUE _self, VALUE obj);
  184. VALUE Descriptor_each_oneof(VALUE _self);
  185. VALUE Descriptor_lookup_oneof(VALUE _self, VALUE name);
  186. VALUE Descriptor_msgclass(VALUE _self);
  187. VALUE Descriptor_file_descriptor(VALUE _self);
  188. extern const rb_data_type_t _Descriptor_type;
  189. void FileDescriptor_mark(void* _self);
  190. void FileDescriptor_free(void* _self);
  191. VALUE FileDescriptor_alloc(VALUE klass);
  192. void FileDescriptor_register(VALUE module);
  193. FileDescriptor* ruby_to_FileDescriptor(VALUE value);
  194. VALUE FileDescriptor_initialize(int argc, VALUE* argv, VALUE _self);
  195. VALUE FileDescriptor_name(VALUE _self);
  196. VALUE FileDescriptor_syntax(VALUE _self);
  197. VALUE FileDescriptor_syntax_set(VALUE _self, VALUE syntax);
  198. void FieldDescriptor_mark(void* _self);
  199. void FieldDescriptor_free(void* _self);
  200. VALUE FieldDescriptor_alloc(VALUE klass);
  201. void FieldDescriptor_register(VALUE module);
  202. FieldDescriptor* ruby_to_FieldDescriptor(VALUE value);
  203. VALUE FieldDescriptor_name(VALUE _self);
  204. VALUE FieldDescriptor_name_set(VALUE _self, VALUE str);
  205. VALUE FieldDescriptor_type(VALUE _self);
  206. VALUE FieldDescriptor_type_set(VALUE _self, VALUE type);
  207. VALUE FieldDescriptor_default(VALUE _self);
  208. VALUE FieldDescriptor_default_set(VALUE _self, VALUE default_value);
  209. VALUE FieldDescriptor_label(VALUE _self);
  210. VALUE FieldDescriptor_label_set(VALUE _self, VALUE label);
  211. VALUE FieldDescriptor_number(VALUE _self);
  212. VALUE FieldDescriptor_number_set(VALUE _self, VALUE number);
  213. VALUE FieldDescriptor_submsg_name(VALUE _self);
  214. VALUE FieldDescriptor_submsg_name_set(VALUE _self, VALUE value);
  215. VALUE FieldDescriptor_subtype(VALUE _self);
  216. VALUE FieldDescriptor_has(VALUE _self, VALUE msg_rb);
  217. VALUE FieldDescriptor_clear(VALUE _self, VALUE msg_rb);
  218. VALUE FieldDescriptor_get(VALUE _self, VALUE msg_rb);
  219. VALUE FieldDescriptor_set(VALUE _self, VALUE msg_rb, VALUE value);
  220. upb_fieldtype_t ruby_to_fieldtype(VALUE type);
  221. VALUE fieldtype_to_ruby(upb_fieldtype_t type);
  222. void OneofDescriptor_mark(void* _self);
  223. void OneofDescriptor_free(void* _self);
  224. VALUE OneofDescriptor_alloc(VALUE klass);
  225. void OneofDescriptor_register(VALUE module);
  226. OneofDescriptor* ruby_to_OneofDescriptor(VALUE value);
  227. VALUE OneofDescriptor_name(VALUE _self);
  228. VALUE OneofDescriptor_name_set(VALUE _self, VALUE value);
  229. VALUE OneofDescriptor_add_field(VALUE _self, VALUE field);
  230. VALUE OneofDescriptor_each(VALUE _self, VALUE field);
  231. void EnumDescriptor_mark(void* _self);
  232. void EnumDescriptor_free(void* _self);
  233. VALUE EnumDescriptor_alloc(VALUE klass);
  234. void EnumDescriptor_register(VALUE module);
  235. EnumDescriptor* ruby_to_EnumDescriptor(VALUE value);
  236. VALUE EnumDescriptor_initialize(VALUE _self, VALUE file_descriptor_rb);
  237. VALUE EnumDescriptor_file_descriptor(VALUE _self);
  238. VALUE EnumDescriptor_name(VALUE _self);
  239. VALUE EnumDescriptor_name_set(VALUE _self, VALUE str);
  240. VALUE EnumDescriptor_add_value(VALUE _self, VALUE name, VALUE number);
  241. VALUE EnumDescriptor_lookup_name(VALUE _self, VALUE name);
  242. VALUE EnumDescriptor_lookup_value(VALUE _self, VALUE number);
  243. VALUE EnumDescriptor_each(VALUE _self);
  244. VALUE EnumDescriptor_enummodule(VALUE _self);
  245. extern const rb_data_type_t _EnumDescriptor_type;
  246. void MessageBuilderContext_mark(void* _self);
  247. void MessageBuilderContext_free(void* _self);
  248. VALUE MessageBuilderContext_alloc(VALUE klass);
  249. void MessageBuilderContext_register(VALUE module);
  250. MessageBuilderContext* ruby_to_MessageBuilderContext(VALUE value);
  251. VALUE MessageBuilderContext_initialize(VALUE _self,
  252. VALUE descriptor,
  253. VALUE builder);
  254. VALUE MessageBuilderContext_optional(int argc, VALUE* argv, VALUE _self);
  255. VALUE MessageBuilderContext_required(int argc, VALUE* argv, VALUE _self);
  256. VALUE MessageBuilderContext_repeated(int argc, VALUE* argv, VALUE _self);
  257. VALUE MessageBuilderContext_map(int argc, VALUE* argv, VALUE _self);
  258. VALUE MessageBuilderContext_oneof(VALUE _self, VALUE name);
  259. void OneofBuilderContext_mark(void* _self);
  260. void OneofBuilderContext_free(void* _self);
  261. VALUE OneofBuilderContext_alloc(VALUE klass);
  262. void OneofBuilderContext_register(VALUE module);
  263. OneofBuilderContext* ruby_to_OneofBuilderContext(VALUE value);
  264. VALUE OneofBuilderContext_initialize(VALUE _self,
  265. VALUE descriptor,
  266. VALUE builder);
  267. VALUE OneofBuilderContext_optional(int argc, VALUE* argv, VALUE _self);
  268. void EnumBuilderContext_mark(void* _self);
  269. void EnumBuilderContext_free(void* _self);
  270. VALUE EnumBuilderContext_alloc(VALUE klass);
  271. void EnumBuilderContext_register(VALUE module);
  272. EnumBuilderContext* ruby_to_EnumBuilderContext(VALUE value);
  273. VALUE EnumBuilderContext_initialize(VALUE _self, VALUE enumdesc);
  274. VALUE EnumBuilderContext_value(VALUE _self, VALUE name, VALUE number);
  275. void FileBuilderContext_mark(void* _self);
  276. void FileBuilderContext_free(void* _self);
  277. VALUE FileBuilderContext_alloc(VALUE klass);
  278. void FileBuilderContext_register(VALUE module);
  279. VALUE FileBuilderContext_initialize(VALUE _self, VALUE file_descriptor,
  280. VALUE builder);
  281. VALUE FileBuilderContext_add_message(VALUE _self, VALUE name);
  282. VALUE FileBuilderContext_add_enum(VALUE _self, VALUE name);
  283. VALUE FileBuilderContext_pending_descriptors(VALUE _self);
  284. void Builder_mark(void* _self);
  285. void Builder_free(void* _self);
  286. VALUE Builder_alloc(VALUE klass);
  287. void Builder_register(VALUE module);
  288. Builder* ruby_to_Builder(VALUE value);
  289. VALUE Builder_initialize(VALUE _self);
  290. VALUE Builder_add_file(int argc, VALUE *argv, VALUE _self);
  291. VALUE Builder_add_message(VALUE _self, VALUE name);
  292. VALUE Builder_add_enum(VALUE _self, VALUE name);
  293. VALUE Builder_finalize_to_pool(VALUE _self, VALUE pool_rb);
  294. // -----------------------------------------------------------------------------
  295. // Native slot storage abstraction.
  296. // -----------------------------------------------------------------------------
  297. #define NATIVE_SLOT_MAX_SIZE sizeof(uint64_t)
  298. size_t native_slot_size(upb_fieldtype_t type);
  299. void native_slot_set(upb_fieldtype_t type,
  300. VALUE type_class,
  301. void* memory,
  302. VALUE value);
  303. // Atomically (with respect to Ruby VM calls) either update the value and set a
  304. // oneof case, or do neither. If |case_memory| is null, then no case value is
  305. // set.
  306. void native_slot_set_value_and_case(upb_fieldtype_t type,
  307. VALUE type_class,
  308. void* memory,
  309. VALUE value,
  310. uint32_t* case_memory,
  311. uint32_t case_number);
  312. VALUE native_slot_get(upb_fieldtype_t type,
  313. VALUE type_class,
  314. const void* memory);
  315. void native_slot_init(upb_fieldtype_t type, void* memory);
  316. void native_slot_mark(upb_fieldtype_t type, void* memory);
  317. void native_slot_dup(upb_fieldtype_t type, void* to, void* from);
  318. void native_slot_deep_copy(upb_fieldtype_t type, void* to, void* from);
  319. bool native_slot_eq(upb_fieldtype_t type, void* mem1, void* mem2);
  320. VALUE native_slot_encode_and_freeze_string(upb_fieldtype_t type, VALUE value);
  321. void native_slot_check_int_range_precision(upb_fieldtype_t type, VALUE value);
  322. extern rb_encoding* kRubyStringUtf8Encoding;
  323. extern rb_encoding* kRubyStringASCIIEncoding;
  324. extern rb_encoding* kRubyString8bitEncoding;
  325. VALUE field_type_class(const upb_fielddef* field);
  326. #define MAP_KEY_FIELD 1
  327. #define MAP_VALUE_FIELD 2
  328. // Oneof case slot value to indicate that no oneof case is set. The value `0` is
  329. // safe because field numbers are used as case identifiers, and no field can
  330. // have a number of 0.
  331. #define ONEOF_CASE_NONE 0
  332. // These operate on a map field (i.e., a repeated field of submessages whose
  333. // submessage type is a map-entry msgdef).
  334. bool is_map_field(const upb_fielddef* field);
  335. const upb_fielddef* map_field_key(const upb_fielddef* field);
  336. const upb_fielddef* map_field_value(const upb_fielddef* field);
  337. // These operate on a map-entry msgdef.
  338. const upb_fielddef* map_entry_key(const upb_msgdef* msgdef);
  339. const upb_fielddef* map_entry_value(const upb_msgdef* msgdef);
  340. // -----------------------------------------------------------------------------
  341. // Repeated field container type.
  342. // -----------------------------------------------------------------------------
  343. typedef struct {
  344. upb_fieldtype_t field_type;
  345. VALUE field_type_class;
  346. void* elements;
  347. int size;
  348. int capacity;
  349. } RepeatedField;
  350. void RepeatedField_mark(void* self);
  351. void RepeatedField_free(void* self);
  352. VALUE RepeatedField_alloc(VALUE klass);
  353. VALUE RepeatedField_init(int argc, VALUE* argv, VALUE self);
  354. void RepeatedField_register(VALUE module);
  355. extern const rb_data_type_t RepeatedField_type;
  356. extern VALUE cRepeatedField;
  357. RepeatedField* ruby_to_RepeatedField(VALUE value);
  358. VALUE RepeatedField_each(VALUE _self);
  359. VALUE RepeatedField_index(int argc, VALUE* argv, VALUE _self);
  360. void* RepeatedField_index_native(VALUE _self, int index);
  361. int RepeatedField_size(VALUE _self);
  362. VALUE RepeatedField_index_set(VALUE _self, VALUE _index, VALUE val);
  363. void RepeatedField_reserve(RepeatedField* self, int new_size);
  364. VALUE RepeatedField_push(VALUE _self, VALUE val);
  365. void RepeatedField_push_native(VALUE _self, void* data);
  366. VALUE RepeatedField_pop_one(VALUE _self);
  367. VALUE RepeatedField_insert(int argc, VALUE* argv, VALUE _self);
  368. VALUE RepeatedField_replace(VALUE _self, VALUE list);
  369. VALUE RepeatedField_clear(VALUE _self);
  370. VALUE RepeatedField_length(VALUE _self);
  371. VALUE RepeatedField_dup(VALUE _self);
  372. VALUE RepeatedField_deep_copy(VALUE _self);
  373. VALUE RepeatedField_to_ary(VALUE _self);
  374. VALUE RepeatedField_eq(VALUE _self, VALUE _other);
  375. VALUE RepeatedField_hash(VALUE _self);
  376. VALUE RepeatedField_inspect(VALUE _self);
  377. VALUE RepeatedField_plus(VALUE _self, VALUE list);
  378. // Defined in repeated_field.c; also used by Map.
  379. void validate_type_class(upb_fieldtype_t type, VALUE klass);
  380. // -----------------------------------------------------------------------------
  381. // Map container type.
  382. // -----------------------------------------------------------------------------
  383. typedef struct {
  384. upb_fieldtype_t key_type;
  385. upb_fieldtype_t value_type;
  386. VALUE value_type_class;
  387. VALUE parse_frame;
  388. upb_strtable table;
  389. } Map;
  390. void Map_mark(void* self);
  391. void Map_free(void* self);
  392. VALUE Map_alloc(VALUE klass);
  393. VALUE Map_init(int argc, VALUE* argv, VALUE self);
  394. void Map_register(VALUE module);
  395. VALUE Map_set_frame(VALUE self, VALUE val);
  396. extern const rb_data_type_t Map_type;
  397. extern VALUE cMap;
  398. Map* ruby_to_Map(VALUE value);
  399. VALUE Map_each(VALUE _self);
  400. VALUE Map_keys(VALUE _self);
  401. VALUE Map_values(VALUE _self);
  402. VALUE Map_index(VALUE _self, VALUE key);
  403. VALUE Map_index_set(VALUE _self, VALUE key, VALUE value);
  404. VALUE Map_has_key(VALUE _self, VALUE key);
  405. VALUE Map_delete(VALUE _self, VALUE key);
  406. VALUE Map_clear(VALUE _self);
  407. VALUE Map_length(VALUE _self);
  408. VALUE Map_dup(VALUE _self);
  409. VALUE Map_deep_copy(VALUE _self);
  410. VALUE Map_eq(VALUE _self, VALUE _other);
  411. VALUE Map_hash(VALUE _self);
  412. VALUE Map_to_h(VALUE _self);
  413. VALUE Map_inspect(VALUE _self);
  414. VALUE Map_merge(VALUE _self, VALUE hashmap);
  415. VALUE Map_merge_into_self(VALUE _self, VALUE hashmap);
  416. typedef struct {
  417. Map* self;
  418. upb_strtable_iter it;
  419. } Map_iter;
  420. void Map_begin(VALUE _self, Map_iter* iter);
  421. void Map_next(Map_iter* iter);
  422. bool Map_done(Map_iter* iter);
  423. VALUE Map_iter_key(Map_iter* iter);
  424. VALUE Map_iter_value(Map_iter* iter);
  425. // -----------------------------------------------------------------------------
  426. // Message layout / storage.
  427. // -----------------------------------------------------------------------------
  428. #define MESSAGE_FIELD_NO_CASE ((size_t)-1)
  429. #define MESSAGE_FIELD_NO_HASBIT ((size_t)-1)
  430. struct MessageField {
  431. size_t offset;
  432. size_t case_offset; // for oneofs, a uint32. Else, MESSAGE_FIELD_NO_CASE.
  433. size_t hasbit;
  434. };
  435. struct MessageLayout {
  436. const upb_msgdef* msgdef;
  437. MessageField* fields;
  438. size_t size;
  439. };
  440. MessageLayout* create_layout(const upb_msgdef* msgdef);
  441. void free_layout(MessageLayout* layout);
  442. bool field_contains_hasbit(MessageLayout* layout,
  443. const upb_fielddef* field);
  444. VALUE layout_get_default(const upb_fielddef* field);
  445. VALUE layout_get(MessageLayout* layout,
  446. const void* storage,
  447. const upb_fielddef* field);
  448. void layout_set(MessageLayout* layout,
  449. void* storage,
  450. const upb_fielddef* field,
  451. VALUE val);
  452. VALUE layout_has(MessageLayout* layout,
  453. const void* storage,
  454. const upb_fielddef* field);
  455. void layout_clear(MessageLayout* layout,
  456. const void* storage,
  457. const upb_fielddef* field);
  458. void layout_init(MessageLayout* layout, void* storage);
  459. void layout_mark(MessageLayout* layout, void* storage);
  460. void layout_dup(MessageLayout* layout, void* to, void* from);
  461. void layout_deep_copy(MessageLayout* layout, void* to, void* from);
  462. VALUE layout_eq(MessageLayout* layout, void* msg1, void* msg2);
  463. VALUE layout_hash(MessageLayout* layout, void* storage);
  464. VALUE layout_inspect(MessageLayout* layout, void* storage);
  465. // -----------------------------------------------------------------------------
  466. // Message class creation.
  467. // -----------------------------------------------------------------------------
  468. // This should probably be factored into a common upb component.
  469. typedef struct {
  470. upb_byteshandler handler;
  471. upb_bytessink sink;
  472. char *ptr;
  473. size_t len, size;
  474. } stringsink;
  475. void stringsink_uninit(stringsink *sink);
  476. struct MessageHeader {
  477. Descriptor* descriptor; // kept alive by self.class.descriptor reference.
  478. stringsink* unknown_fields; // store unknown fields in decoding.
  479. // Data comes after this.
  480. };
  481. extern rb_data_type_t Message_type;
  482. VALUE build_class_from_descriptor(Descriptor* descriptor);
  483. void* Message_data(void* msg);
  484. void Message_mark(void* self);
  485. void Message_free(void* self);
  486. VALUE Message_alloc(VALUE klass);
  487. VALUE Message_method_missing(int argc, VALUE* argv, VALUE _self);
  488. VALUE Message_initialize(int argc, VALUE* argv, VALUE _self);
  489. VALUE Message_dup(VALUE _self);
  490. VALUE Message_deep_copy(VALUE _self);
  491. VALUE Message_eq(VALUE _self, VALUE _other);
  492. VALUE Message_hash(VALUE _self);
  493. VALUE Message_inspect(VALUE _self);
  494. VALUE Message_to_h(VALUE _self);
  495. VALUE Message_index(VALUE _self, VALUE field_name);
  496. VALUE Message_index_set(VALUE _self, VALUE field_name, VALUE value);
  497. VALUE Message_descriptor(VALUE klass);
  498. VALUE Message_decode(VALUE klass, VALUE data);
  499. VALUE Message_encode(VALUE klass, VALUE msg_rb);
  500. VALUE Message_decode_json(VALUE klass, VALUE data);
  501. VALUE Message_encode_json(int argc, VALUE* argv, VALUE klass);
  502. VALUE Google_Protobuf_discard_unknown(VALUE self, VALUE msg_rb);
  503. VALUE Google_Protobuf_deep_copy(VALUE self, VALUE obj);
  504. VALUE build_module_from_enumdesc(EnumDescriptor* enumdef);
  505. VALUE enum_lookup(VALUE self, VALUE number);
  506. VALUE enum_resolve(VALUE self, VALUE sym);
  507. const upb_pbdecodermethod *new_fillmsg_decodermethod(
  508. Descriptor* descriptor, const void *owner);
  509. // Maximum depth allowed during encoding, to avoid stack overflows due to
  510. // cycles.
  511. #define ENCODE_MAX_NESTING 63
  512. // -----------------------------------------------------------------------------
  513. // Global map from upb {msg,enum}defs to wrapper Descriptor/EnumDescriptor
  514. // instances.
  515. // -----------------------------------------------------------------------------
  516. void add_def_obj(const void* def, VALUE value);
  517. VALUE get_def_obj(const void* def);
  518. // -----------------------------------------------------------------------------
  519. // Utilities.
  520. // -----------------------------------------------------------------------------
  521. void check_upb_status(const upb_status* status, const char* msg);
  522. #define CHECK_UPB(code, msg) do { \
  523. upb_status status = UPB_STATUS_INIT; \
  524. code; \
  525. check_upb_status(&status, msg); \
  526. } while (0)
  527. extern ID descriptor_instancevar_interned;
  528. #endif // __GOOGLE_PROTOBUF_RUBY_PROTOBUF_H__