protobuf.h 23 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615
  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. extern VALUE generated_pool;
  173. void Descriptor_mark(void* _self);
  174. void Descriptor_free(void* _self);
  175. VALUE Descriptor_alloc(VALUE klass);
  176. void Descriptor_register(VALUE module);
  177. Descriptor* ruby_to_Descriptor(VALUE value);
  178. VALUE Descriptor_initialize(VALUE _self, VALUE file_descriptor_rb);
  179. VALUE Descriptor_name(VALUE _self);
  180. VALUE Descriptor_name_set(VALUE _self, VALUE str);
  181. VALUE Descriptor_each(VALUE _self);
  182. VALUE Descriptor_lookup(VALUE _self, VALUE name);
  183. VALUE Descriptor_add_field(VALUE _self, VALUE obj);
  184. VALUE Descriptor_add_oneof(VALUE _self, VALUE obj);
  185. VALUE Descriptor_each_oneof(VALUE _self);
  186. VALUE Descriptor_lookup_oneof(VALUE _self, VALUE name);
  187. VALUE Descriptor_msgclass(VALUE _self);
  188. VALUE Descriptor_file_descriptor(VALUE _self);
  189. extern const rb_data_type_t _Descriptor_type;
  190. void FileDescriptor_mark(void* _self);
  191. void FileDescriptor_free(void* _self);
  192. VALUE FileDescriptor_alloc(VALUE klass);
  193. void FileDescriptor_register(VALUE module);
  194. FileDescriptor* ruby_to_FileDescriptor(VALUE value);
  195. VALUE FileDescriptor_initialize(int argc, VALUE* argv, VALUE _self);
  196. VALUE FileDescriptor_name(VALUE _self);
  197. VALUE FileDescriptor_syntax(VALUE _self);
  198. VALUE FileDescriptor_syntax_set(VALUE _self, VALUE syntax);
  199. void FieldDescriptor_mark(void* _self);
  200. void FieldDescriptor_free(void* _self);
  201. VALUE FieldDescriptor_alloc(VALUE klass);
  202. void FieldDescriptor_register(VALUE module);
  203. FieldDescriptor* ruby_to_FieldDescriptor(VALUE value);
  204. VALUE FieldDescriptor_name(VALUE _self);
  205. VALUE FieldDescriptor_name_set(VALUE _self, VALUE str);
  206. VALUE FieldDescriptor_type(VALUE _self);
  207. VALUE FieldDescriptor_type_set(VALUE _self, VALUE type);
  208. VALUE FieldDescriptor_default(VALUE _self);
  209. VALUE FieldDescriptor_default_set(VALUE _self, VALUE default_value);
  210. VALUE FieldDescriptor_label(VALUE _self);
  211. VALUE FieldDescriptor_label_set(VALUE _self, VALUE label);
  212. VALUE FieldDescriptor_number(VALUE _self);
  213. VALUE FieldDescriptor_number_set(VALUE _self, VALUE number);
  214. VALUE FieldDescriptor_submsg_name(VALUE _self);
  215. VALUE FieldDescriptor_submsg_name_set(VALUE _self, VALUE value);
  216. VALUE FieldDescriptor_subtype(VALUE _self);
  217. VALUE FieldDescriptor_has(VALUE _self, VALUE msg_rb);
  218. VALUE FieldDescriptor_clear(VALUE _self, VALUE msg_rb);
  219. VALUE FieldDescriptor_get(VALUE _self, VALUE msg_rb);
  220. VALUE FieldDescriptor_set(VALUE _self, VALUE msg_rb, VALUE value);
  221. upb_fieldtype_t ruby_to_fieldtype(VALUE type);
  222. VALUE fieldtype_to_ruby(upb_fieldtype_t type);
  223. void OneofDescriptor_mark(void* _self);
  224. void OneofDescriptor_free(void* _self);
  225. VALUE OneofDescriptor_alloc(VALUE klass);
  226. void OneofDescriptor_register(VALUE module);
  227. OneofDescriptor* ruby_to_OneofDescriptor(VALUE value);
  228. VALUE OneofDescriptor_name(VALUE _self);
  229. VALUE OneofDescriptor_name_set(VALUE _self, VALUE value);
  230. VALUE OneofDescriptor_add_field(VALUE _self, VALUE field);
  231. VALUE OneofDescriptor_each(VALUE _self, VALUE field);
  232. void EnumDescriptor_mark(void* _self);
  233. void EnumDescriptor_free(void* _self);
  234. VALUE EnumDescriptor_alloc(VALUE klass);
  235. void EnumDescriptor_register(VALUE module);
  236. EnumDescriptor* ruby_to_EnumDescriptor(VALUE value);
  237. VALUE EnumDescriptor_initialize(VALUE _self, VALUE file_descriptor_rb);
  238. VALUE EnumDescriptor_file_descriptor(VALUE _self);
  239. VALUE EnumDescriptor_name(VALUE _self);
  240. VALUE EnumDescriptor_name_set(VALUE _self, VALUE str);
  241. VALUE EnumDescriptor_add_value(VALUE _self, VALUE name, VALUE number);
  242. VALUE EnumDescriptor_lookup_name(VALUE _self, VALUE name);
  243. VALUE EnumDescriptor_lookup_value(VALUE _self, VALUE number);
  244. VALUE EnumDescriptor_each(VALUE _self);
  245. VALUE EnumDescriptor_enummodule(VALUE _self);
  246. extern const rb_data_type_t _EnumDescriptor_type;
  247. void MessageBuilderContext_mark(void* _self);
  248. void MessageBuilderContext_free(void* _self);
  249. VALUE MessageBuilderContext_alloc(VALUE klass);
  250. void MessageBuilderContext_register(VALUE module);
  251. MessageBuilderContext* ruby_to_MessageBuilderContext(VALUE value);
  252. VALUE MessageBuilderContext_initialize(VALUE _self,
  253. VALUE descriptor,
  254. VALUE builder);
  255. VALUE MessageBuilderContext_optional(int argc, VALUE* argv, VALUE _self);
  256. VALUE MessageBuilderContext_required(int argc, VALUE* argv, VALUE _self);
  257. VALUE MessageBuilderContext_repeated(int argc, VALUE* argv, VALUE _self);
  258. VALUE MessageBuilderContext_map(int argc, VALUE* argv, VALUE _self);
  259. VALUE MessageBuilderContext_oneof(VALUE _self, VALUE name);
  260. void OneofBuilderContext_mark(void* _self);
  261. void OneofBuilderContext_free(void* _self);
  262. VALUE OneofBuilderContext_alloc(VALUE klass);
  263. void OneofBuilderContext_register(VALUE module);
  264. OneofBuilderContext* ruby_to_OneofBuilderContext(VALUE value);
  265. VALUE OneofBuilderContext_initialize(VALUE _self,
  266. VALUE descriptor,
  267. VALUE builder);
  268. VALUE OneofBuilderContext_optional(int argc, VALUE* argv, VALUE _self);
  269. void EnumBuilderContext_mark(void* _self);
  270. void EnumBuilderContext_free(void* _self);
  271. VALUE EnumBuilderContext_alloc(VALUE klass);
  272. void EnumBuilderContext_register(VALUE module);
  273. EnumBuilderContext* ruby_to_EnumBuilderContext(VALUE value);
  274. VALUE EnumBuilderContext_initialize(VALUE _self, VALUE enumdesc);
  275. VALUE EnumBuilderContext_value(VALUE _self, VALUE name, VALUE number);
  276. void FileBuilderContext_mark(void* _self);
  277. void FileBuilderContext_free(void* _self);
  278. VALUE FileBuilderContext_alloc(VALUE klass);
  279. void FileBuilderContext_register(VALUE module);
  280. VALUE FileBuilderContext_initialize(VALUE _self, VALUE file_descriptor,
  281. VALUE builder);
  282. VALUE FileBuilderContext_add_message(VALUE _self, VALUE name);
  283. VALUE FileBuilderContext_add_enum(VALUE _self, VALUE name);
  284. VALUE FileBuilderContext_pending_descriptors(VALUE _self);
  285. void Builder_mark(void* _self);
  286. void Builder_free(void* _self);
  287. VALUE Builder_alloc(VALUE klass);
  288. void Builder_register(VALUE module);
  289. Builder* ruby_to_Builder(VALUE value);
  290. VALUE Builder_initialize(VALUE _self);
  291. VALUE Builder_add_file(int argc, VALUE *argv, VALUE _self);
  292. VALUE Builder_add_message(VALUE _self, VALUE name);
  293. VALUE Builder_add_enum(VALUE _self, VALUE name);
  294. VALUE Builder_finalize_to_pool(VALUE _self, VALUE pool_rb);
  295. // -----------------------------------------------------------------------------
  296. // Native slot storage abstraction.
  297. // -----------------------------------------------------------------------------
  298. #define NATIVE_SLOT_MAX_SIZE sizeof(uint64_t)
  299. size_t native_slot_size(upb_fieldtype_t type);
  300. void native_slot_set(const char* name,
  301. upb_fieldtype_t type,
  302. VALUE type_class,
  303. void* memory,
  304. VALUE value);
  305. // Atomically (with respect to Ruby VM calls) either update the value and set a
  306. // oneof case, or do neither. If |case_memory| is null, then no case value is
  307. // set.
  308. void native_slot_set_value_and_case(const char* name,
  309. upb_fieldtype_t type,
  310. VALUE type_class,
  311. void* memory,
  312. VALUE value,
  313. uint32_t* case_memory,
  314. uint32_t case_number);
  315. VALUE native_slot_get(upb_fieldtype_t type,
  316. VALUE type_class,
  317. const void* memory);
  318. void native_slot_init(upb_fieldtype_t type, void* memory);
  319. void native_slot_mark(upb_fieldtype_t type, void* memory);
  320. void native_slot_dup(upb_fieldtype_t type, void* to, void* from);
  321. void native_slot_deep_copy(upb_fieldtype_t type, void* to, void* from);
  322. bool native_slot_eq(upb_fieldtype_t type, void* mem1, void* mem2);
  323. VALUE native_slot_encode_and_freeze_string(upb_fieldtype_t type, VALUE value);
  324. void native_slot_check_int_range_precision(const char* name, upb_fieldtype_t type, VALUE value);
  325. extern rb_encoding* kRubyStringUtf8Encoding;
  326. extern rb_encoding* kRubyStringASCIIEncoding;
  327. extern rb_encoding* kRubyString8bitEncoding;
  328. VALUE field_type_class(const upb_fielddef* field);
  329. #define MAP_KEY_FIELD 1
  330. #define MAP_VALUE_FIELD 2
  331. // Oneof case slot value to indicate that no oneof case is set. The value `0` is
  332. // safe because field numbers are used as case identifiers, and no field can
  333. // have a number of 0.
  334. #define ONEOF_CASE_NONE 0
  335. // These operate on a map field (i.e., a repeated field of submessages whose
  336. // submessage type is a map-entry msgdef).
  337. bool is_map_field(const upb_fielddef* field);
  338. const upb_fielddef* map_field_key(const upb_fielddef* field);
  339. const upb_fielddef* map_field_value(const upb_fielddef* field);
  340. // These operate on a map-entry msgdef.
  341. const upb_fielddef* map_entry_key(const upb_msgdef* msgdef);
  342. const upb_fielddef* map_entry_value(const upb_msgdef* msgdef);
  343. // -----------------------------------------------------------------------------
  344. // Repeated field container type.
  345. // -----------------------------------------------------------------------------
  346. typedef struct {
  347. upb_fieldtype_t field_type;
  348. VALUE field_type_class;
  349. void* elements;
  350. int size;
  351. int capacity;
  352. } RepeatedField;
  353. void RepeatedField_mark(void* self);
  354. void RepeatedField_free(void* self);
  355. VALUE RepeatedField_alloc(VALUE klass);
  356. VALUE RepeatedField_init(int argc, VALUE* argv, VALUE self);
  357. void RepeatedField_register(VALUE module);
  358. extern const rb_data_type_t RepeatedField_type;
  359. extern VALUE cRepeatedField;
  360. RepeatedField* ruby_to_RepeatedField(VALUE value);
  361. VALUE RepeatedField_each(VALUE _self);
  362. VALUE RepeatedField_index(int argc, VALUE* argv, VALUE _self);
  363. void* RepeatedField_index_native(VALUE _self, int index);
  364. int RepeatedField_size(VALUE _self);
  365. VALUE RepeatedField_index_set(VALUE _self, VALUE _index, VALUE val);
  366. void RepeatedField_reserve(RepeatedField* self, int new_size);
  367. VALUE RepeatedField_push(VALUE _self, VALUE val);
  368. void RepeatedField_push_native(VALUE _self, void* data);
  369. VALUE RepeatedField_pop_one(VALUE _self);
  370. VALUE RepeatedField_insert(int argc, VALUE* argv, VALUE _self);
  371. VALUE RepeatedField_replace(VALUE _self, VALUE list);
  372. VALUE RepeatedField_clear(VALUE _self);
  373. VALUE RepeatedField_length(VALUE _self);
  374. VALUE RepeatedField_dup(VALUE _self);
  375. VALUE RepeatedField_deep_copy(VALUE _self);
  376. VALUE RepeatedField_to_ary(VALUE _self);
  377. VALUE RepeatedField_eq(VALUE _self, VALUE _other);
  378. VALUE RepeatedField_hash(VALUE _self);
  379. VALUE RepeatedField_inspect(VALUE _self);
  380. VALUE RepeatedField_plus(VALUE _self, VALUE list);
  381. // Defined in repeated_field.c; also used by Map.
  382. void validate_type_class(upb_fieldtype_t type, VALUE klass);
  383. // -----------------------------------------------------------------------------
  384. // Map container type.
  385. // -----------------------------------------------------------------------------
  386. typedef struct {
  387. upb_fieldtype_t key_type;
  388. upb_fieldtype_t value_type;
  389. VALUE value_type_class;
  390. VALUE parse_frame;
  391. upb_strtable table;
  392. } Map;
  393. void Map_mark(void* self);
  394. void Map_free(void* self);
  395. VALUE Map_alloc(VALUE klass);
  396. VALUE Map_init(int argc, VALUE* argv, VALUE self);
  397. void Map_register(VALUE module);
  398. VALUE Map_set_frame(VALUE self, VALUE val);
  399. extern const rb_data_type_t Map_type;
  400. extern VALUE cMap;
  401. Map* ruby_to_Map(VALUE value);
  402. VALUE Map_each(VALUE _self);
  403. VALUE Map_keys(VALUE _self);
  404. VALUE Map_values(VALUE _self);
  405. VALUE Map_index(VALUE _self, VALUE key);
  406. VALUE Map_index_set(VALUE _self, VALUE key, VALUE value);
  407. VALUE Map_has_key(VALUE _self, VALUE key);
  408. VALUE Map_delete(VALUE _self, VALUE key);
  409. VALUE Map_clear(VALUE _self);
  410. VALUE Map_length(VALUE _self);
  411. VALUE Map_dup(VALUE _self);
  412. VALUE Map_deep_copy(VALUE _self);
  413. VALUE Map_eq(VALUE _self, VALUE _other);
  414. VALUE Map_hash(VALUE _self);
  415. VALUE Map_to_h(VALUE _self);
  416. VALUE Map_inspect(VALUE _self);
  417. VALUE Map_merge(VALUE _self, VALUE hashmap);
  418. VALUE Map_merge_into_self(VALUE _self, VALUE hashmap);
  419. typedef struct {
  420. Map* self;
  421. upb_strtable_iter it;
  422. } Map_iter;
  423. void Map_begin(VALUE _self, Map_iter* iter);
  424. void Map_next(Map_iter* iter);
  425. bool Map_done(Map_iter* iter);
  426. VALUE Map_iter_key(Map_iter* iter);
  427. VALUE Map_iter_value(Map_iter* iter);
  428. // -----------------------------------------------------------------------------
  429. // Message layout / storage.
  430. // -----------------------------------------------------------------------------
  431. #define MESSAGE_FIELD_NO_CASE ((size_t)-1)
  432. #define MESSAGE_FIELD_NO_HASBIT ((size_t)-1)
  433. struct MessageField {
  434. size_t offset;
  435. size_t case_offset; // for oneofs, a uint32. Else, MESSAGE_FIELD_NO_CASE.
  436. size_t hasbit;
  437. };
  438. struct MessageLayout {
  439. const upb_msgdef* msgdef;
  440. MessageField* fields;
  441. size_t size;
  442. };
  443. MessageLayout* create_layout(const upb_msgdef* msgdef);
  444. void free_layout(MessageLayout* layout);
  445. bool field_contains_hasbit(MessageLayout* layout,
  446. const upb_fielddef* field);
  447. VALUE layout_get_default(const upb_fielddef* field);
  448. VALUE layout_get(MessageLayout* layout,
  449. const void* storage,
  450. const upb_fielddef* field);
  451. void layout_set(MessageLayout* layout,
  452. void* storage,
  453. const upb_fielddef* field,
  454. VALUE val);
  455. VALUE layout_has(MessageLayout* layout,
  456. const void* storage,
  457. const upb_fielddef* field);
  458. void layout_clear(MessageLayout* layout,
  459. const void* storage,
  460. const upb_fielddef* field);
  461. void layout_init(MessageLayout* layout, void* storage);
  462. void layout_mark(MessageLayout* layout, void* storage);
  463. void layout_dup(MessageLayout* layout, void* to, void* from);
  464. void layout_deep_copy(MessageLayout* layout, void* to, void* from);
  465. VALUE layout_eq(MessageLayout* layout, void* msg1, void* msg2);
  466. VALUE layout_hash(MessageLayout* layout, void* storage);
  467. VALUE layout_inspect(MessageLayout* layout, void* storage);
  468. // -----------------------------------------------------------------------------
  469. // Message class creation.
  470. // -----------------------------------------------------------------------------
  471. // This should probably be factored into a common upb component.
  472. typedef struct {
  473. upb_byteshandler handler;
  474. upb_bytessink sink;
  475. char *ptr;
  476. size_t len, size;
  477. } stringsink;
  478. void stringsink_uninit(stringsink *sink);
  479. struct MessageHeader {
  480. Descriptor* descriptor; // kept alive by self.class.descriptor reference.
  481. stringsink* unknown_fields; // store unknown fields in decoding.
  482. // Data comes after this.
  483. };
  484. extern rb_data_type_t Message_type;
  485. VALUE build_class_from_descriptor(Descriptor* descriptor);
  486. void* Message_data(void* msg);
  487. void Message_mark(void* self);
  488. void Message_free(void* self);
  489. VALUE Message_alloc(VALUE klass);
  490. VALUE Message_method_missing(int argc, VALUE* argv, VALUE _self);
  491. VALUE Message_initialize(int argc, VALUE* argv, VALUE _self);
  492. VALUE Message_dup(VALUE _self);
  493. VALUE Message_deep_copy(VALUE _self);
  494. VALUE Message_eq(VALUE _self, VALUE _other);
  495. VALUE Message_hash(VALUE _self);
  496. VALUE Message_inspect(VALUE _self);
  497. VALUE Message_to_h(VALUE _self);
  498. VALUE Message_index(VALUE _self, VALUE field_name);
  499. VALUE Message_index_set(VALUE _self, VALUE field_name, VALUE value);
  500. VALUE Message_descriptor(VALUE klass);
  501. VALUE Message_decode(VALUE klass, VALUE data);
  502. VALUE Message_encode(VALUE klass, VALUE msg_rb);
  503. VALUE Message_decode_json(int argc, VALUE* argv, VALUE klass);
  504. VALUE Message_encode_json(int argc, VALUE* argv, VALUE klass);
  505. VALUE Google_Protobuf_discard_unknown(VALUE self, VALUE msg_rb);
  506. VALUE Google_Protobuf_deep_copy(VALUE self, VALUE obj);
  507. VALUE build_module_from_enumdesc(EnumDescriptor* enumdef);
  508. VALUE enum_lookup(VALUE self, VALUE number);
  509. VALUE enum_resolve(VALUE self, VALUE sym);
  510. const upb_pbdecodermethod *new_fillmsg_decodermethod(
  511. Descriptor* descriptor, const void *owner);
  512. // Maximum depth allowed during encoding, to avoid stack overflows due to
  513. // cycles.
  514. #define ENCODE_MAX_NESTING 63
  515. // -----------------------------------------------------------------------------
  516. // Global map from upb {msg,enum}defs to wrapper Descriptor/EnumDescriptor
  517. // instances.
  518. // -----------------------------------------------------------------------------
  519. void add_def_obj(const void* def, VALUE value);
  520. VALUE get_def_obj(const void* def);
  521. // -----------------------------------------------------------------------------
  522. // Utilities.
  523. // -----------------------------------------------------------------------------
  524. void check_upb_status(const upb_status* status, const char* msg);
  525. #define CHECK_UPB(code, msg) do { \
  526. upb_status status = UPB_STATUS_INIT; \
  527. code; \
  528. check_upb_status(&status, msg); \
  529. } while (0)
  530. extern ID descriptor_instancevar_interned;
  531. #endif // __GOOGLE_PROTOBUF_RUBY_PROTOBUF_H__