encode_decode.c 52 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521
  1. // Protocol Buffers - Google's data interchange format
  2. // Copyright 2008 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. #include "protobuf.h"
  31. #include "utf8.h"
  32. /* stringsink *****************************************************************/
  33. typedef struct {
  34. upb_byteshandler handler;
  35. upb_bytessink sink;
  36. char *ptr;
  37. size_t len, size;
  38. } stringsink;
  39. static void *stringsink_start(void *_sink, const void *hd, size_t size_hint) {
  40. stringsink *sink = _sink;
  41. sink->len = 0;
  42. return sink;
  43. }
  44. static size_t stringsink_string(void *_sink, const void *hd, const char *ptr,
  45. size_t len, const upb_bufhandle *handle) {
  46. stringsink *sink = _sink;
  47. size_t new_size = sink->size;
  48. UPB_UNUSED(hd);
  49. UPB_UNUSED(handle);
  50. while (sink->len + len > new_size) {
  51. new_size *= 2;
  52. }
  53. if (new_size != sink->size) {
  54. sink->ptr = realloc(sink->ptr, new_size);
  55. sink->size = new_size;
  56. }
  57. memcpy(sink->ptr + sink->len, ptr, len);
  58. sink->len += len;
  59. return len;
  60. }
  61. void stringsink_init(stringsink *sink) {
  62. upb_byteshandler_init(&sink->handler);
  63. upb_byteshandler_setstartstr(&sink->handler, stringsink_start, NULL);
  64. upb_byteshandler_setstring(&sink->handler, stringsink_string, NULL);
  65. upb_bytessink_reset(&sink->sink, &sink->handler, sink);
  66. sink->size = 32;
  67. sink->ptr = malloc(sink->size);
  68. sink->len = 0;
  69. }
  70. void stringsink_uninit(stringsink *sink) { free(sink->ptr); }
  71. /* stackenv *****************************************************************/
  72. // Stack-allocated context during an encode/decode operation. Contains the upb
  73. // environment and its stack-based allocator, an initial buffer for allocations
  74. // to avoid malloc() when possible, and a template for PHP exception messages
  75. // if any error occurs.
  76. #define STACK_ENV_STACKBYTES 4096
  77. typedef struct {
  78. upb_env env;
  79. const char *php_error_template;
  80. char allocbuf[STACK_ENV_STACKBYTES];
  81. } stackenv;
  82. static void stackenv_init(stackenv* se, const char* errmsg);
  83. static void stackenv_uninit(stackenv* se);
  84. // Callback invoked by upb if any error occurs during parsing or serialization.
  85. static bool env_error_func(void* ud, const upb_status* status) {
  86. char err_msg[100] = "";
  87. stackenv* se = ud;
  88. // Free the env -- zend_error will longjmp up the stack past the
  89. // encode/decode function so it would not otherwise have been freed.
  90. stackenv_uninit(se);
  91. // TODO(teboring): have a way to verify that this is actually a parse error,
  92. // instead of just throwing "parse error" unconditionally.
  93. sprintf(err_msg, se->php_error_template, upb_status_errmsg(status));
  94. TSRMLS_FETCH();
  95. zend_throw_exception(NULL, err_msg, 0 TSRMLS_CC);
  96. // Never reached.
  97. return false;
  98. }
  99. static void stackenv_init(stackenv* se, const char* errmsg) {
  100. se->php_error_template = errmsg;
  101. upb_env_init2(&se->env, se->allocbuf, sizeof(se->allocbuf), NULL);
  102. upb_env_seterrorfunc(&se->env, env_error_func, se);
  103. }
  104. static void stackenv_uninit(stackenv* se) {
  105. upb_env_uninit(&se->env);
  106. }
  107. // -----------------------------------------------------------------------------
  108. // Parsing.
  109. // -----------------------------------------------------------------------------
  110. #define DEREF(msg, ofs, type) *(type*)(((uint8_t *)msg) + ofs)
  111. // Creates a handlerdata that simply contains the offset for this field.
  112. static const void* newhandlerdata(upb_handlers* h, uint32_t ofs) {
  113. size_t* hd_ofs = (size_t*)malloc(sizeof(size_t));
  114. *hd_ofs = ofs;
  115. upb_handlers_addcleanup(h, hd_ofs, free);
  116. return hd_ofs;
  117. }
  118. typedef struct {
  119. size_t ofs;
  120. const upb_msgdef *md;
  121. } submsg_handlerdata_t;
  122. // Creates a handlerdata that contains offset and submessage type information.
  123. static const void *newsubmsghandlerdata(upb_handlers* h, uint32_t ofs,
  124. const upb_fielddef* f) {
  125. submsg_handlerdata_t* hd =
  126. (submsg_handlerdata_t*)malloc(sizeof(submsg_handlerdata_t));
  127. hd->ofs = ofs;
  128. hd->md = upb_fielddef_msgsubdef(f);
  129. upb_handlers_addcleanup(h, hd, free);
  130. return hd;
  131. }
  132. typedef struct {
  133. size_t ofs; // union data slot
  134. size_t case_ofs; // oneof_case field
  135. int property_ofs; // properties table cache
  136. uint32_t oneof_case_num; // oneof-case number to place in oneof_case field
  137. const upb_msgdef *md; // msgdef, for oneof submessage handler
  138. } oneof_handlerdata_t;
  139. static const void *newoneofhandlerdata(upb_handlers *h,
  140. uint32_t ofs,
  141. uint32_t case_ofs,
  142. int property_ofs,
  143. const upb_fielddef *f) {
  144. oneof_handlerdata_t* hd =
  145. (oneof_handlerdata_t*)malloc(sizeof(oneof_handlerdata_t));
  146. hd->ofs = ofs;
  147. hd->case_ofs = case_ofs;
  148. hd->property_ofs = property_ofs;
  149. // We reuse the field tag number as a oneof union discriminant tag. Note that
  150. // we don't expose these numbers to the user, so the only requirement is that
  151. // we have some unique ID for each union case/possibility. The field tag
  152. // numbers are already present and are easy to use so there's no reason to
  153. // create a separate ID space. In addition, using the field tag number here
  154. // lets us easily look up the field in the oneof accessor.
  155. hd->oneof_case_num = upb_fielddef_number(f);
  156. if (upb_fielddef_type(f) == UPB_TYPE_MESSAGE) {
  157. hd->md = upb_fielddef_msgsubdef(f);
  158. } else {
  159. hd->md = NULL;
  160. }
  161. upb_handlers_addcleanup(h, hd, free);
  162. return hd;
  163. }
  164. // A handler that starts a repeated field. Gets the Repeated*Field instance for
  165. // this field (such an instance always exists even in an empty message).
  166. static void *startseq_handler(void* closure, const void* hd) {
  167. MessageHeader* msg = closure;
  168. const size_t *ofs = hd;
  169. return CACHED_PTR_TO_ZVAL_PTR(DEREF(message_data(msg), *ofs, CACHED_VALUE*));
  170. }
  171. // Handlers that append primitive values to a repeated field.
  172. #define DEFINE_APPEND_HANDLER(type, ctype) \
  173. static bool append##type##_handler(void* closure, const void* hd, \
  174. ctype val) { \
  175. zval* array = (zval*)closure; \
  176. TSRMLS_FETCH(); \
  177. RepeatedField* intern = UNBOX(RepeatedField, array); \
  178. repeated_field_push_native(intern, &val); \
  179. return true; \
  180. }
  181. DEFINE_APPEND_HANDLER(bool, bool)
  182. DEFINE_APPEND_HANDLER(int32, int32_t)
  183. DEFINE_APPEND_HANDLER(uint32, uint32_t)
  184. DEFINE_APPEND_HANDLER(float, float)
  185. DEFINE_APPEND_HANDLER(int64, int64_t)
  186. DEFINE_APPEND_HANDLER(uint64, uint64_t)
  187. DEFINE_APPEND_HANDLER(double, double)
  188. // Appends a string to a repeated field.
  189. static void* appendstr_handler(void *closure,
  190. const void *hd,
  191. size_t size_hint) {
  192. zval* array = (zval*)closure;
  193. TSRMLS_FETCH();
  194. RepeatedField* intern = UNBOX(RepeatedField, array);
  195. #if PHP_MAJOR_VERSION < 7
  196. zval* str;
  197. MAKE_STD_ZVAL(str);
  198. PHP_PROTO_ZVAL_STRING(str, "", 1);
  199. repeated_field_push_native(intern, &str);
  200. return (void*)str;
  201. #else
  202. zend_string* str = zend_string_init("", 0, 1);
  203. repeated_field_push_native(intern, &str);
  204. return intern;
  205. #endif
  206. }
  207. // Appends a 'bytes' string to a repeated field.
  208. static void* appendbytes_handler(void *closure,
  209. const void *hd,
  210. size_t size_hint) {
  211. zval* array = (zval*)closure;
  212. TSRMLS_FETCH();
  213. RepeatedField* intern = UNBOX(RepeatedField, array);
  214. #if PHP_MAJOR_VERSION < 7
  215. zval* str;
  216. MAKE_STD_ZVAL(str);
  217. PHP_PROTO_ZVAL_STRING(str, "", 1);
  218. repeated_field_push_native(intern, &str);
  219. return (void*)str;
  220. #else
  221. zend_string* str = zend_string_init("", 0, 1);
  222. repeated_field_push_native(intern, &str);
  223. return intern;
  224. #endif
  225. }
  226. static bool int32_handler(void* closure, const void* hd,
  227. int32_t val) {
  228. MessageHeader* msg = (MessageHeader*)closure;
  229. const size_t *ofs = hd;
  230. DEREF(message_data(msg), *ofs, int32_t) = val;
  231. return true;
  232. }
  233. // Handlers that append primitive values to a repeated field.
  234. #define DEFINE_SINGULAR_HANDLER(type, ctype) \
  235. static bool type##_handler(void* closure, const void* hd, \
  236. ctype val) { \
  237. MessageHeader* msg = (MessageHeader*)closure; \
  238. const size_t *ofs = hd; \
  239. DEREF(message_data(msg), *ofs, ctype) = val; \
  240. return true; \
  241. }
  242. DEFINE_SINGULAR_HANDLER(bool, bool)
  243. // DEFINE_SINGULAR_HANDLER(int32, int32_t)
  244. DEFINE_SINGULAR_HANDLER(uint32, uint32_t)
  245. DEFINE_SINGULAR_HANDLER(float, float)
  246. DEFINE_SINGULAR_HANDLER(int64, int64_t)
  247. DEFINE_SINGULAR_HANDLER(uint64, uint64_t)
  248. DEFINE_SINGULAR_HANDLER(double, double)
  249. #undef DEFINE_SINGULAR_HANDLER
  250. #if PHP_MAJOR_VERSION < 7
  251. static void *empty_php_string(zval** value_ptr) {
  252. SEPARATE_ZVAL_IF_NOT_REF(value_ptr);
  253. return (void*)(*value_ptr);
  254. }
  255. #else
  256. static void *empty_php_string(zval* value_ptr) {
  257. return value_ptr;
  258. }
  259. #endif
  260. // Sets a non-repeated string field in a message.
  261. static void* str_handler(void *closure,
  262. const void *hd,
  263. size_t size_hint) {
  264. MessageHeader* msg = closure;
  265. const size_t *ofs = hd;
  266. return empty_php_string(DEREF(message_data(msg), *ofs, CACHED_VALUE*));
  267. }
  268. // Sets a non-repeated 'bytes' field in a message.
  269. static void* bytes_handler(void *closure,
  270. const void *hd,
  271. size_t size_hint) {
  272. MessageHeader* msg = closure;
  273. const size_t *ofs = hd;
  274. return empty_php_string(DEREF(message_data(msg), *ofs, CACHED_VALUE*));
  275. }
  276. static size_t stringdata_handler(void* closure, const void* hd,
  277. const char* str, size_t len,
  278. const upb_bufhandle* handle) {
  279. zval* php_str = (zval*)closure;
  280. #if PHP_MAJOR_VERSION < 7
  281. // Oneof string/bytes fields may have NULL initial value, which doesn't need
  282. // to be freed.
  283. if (Z_TYPE_P(php_str) == IS_STRING && !IS_INTERNED(Z_STRVAL_P(php_str))) {
  284. FREE(Z_STRVAL_P(php_str));
  285. }
  286. ZVAL_STRINGL(php_str, str, len, 1);
  287. #else
  288. if (Z_TYPE_P(php_str) == IS_STRING) {
  289. zend_string_release(Z_STR_P(php_str));
  290. }
  291. ZVAL_NEW_STR(php_str, zend_string_init(str, len, 0));
  292. #endif
  293. return len;
  294. }
  295. #if PHP_MAJOR_VERSION >= 7
  296. static size_t zendstringdata_handler(void* closure, const void* hd,
  297. const char* str, size_t len,
  298. const upb_bufhandle* handle) {
  299. RepeatedField* intern = (RepeatedField*)closure;
  300. unsigned char memory[NATIVE_SLOT_MAX_SIZE];
  301. memset(memory, 0, NATIVE_SLOT_MAX_SIZE);
  302. *(zend_string**)memory = zend_string_init(str, len, 0);
  303. HashTable *ht = PHP_PROTO_HASH_OF(intern->array);
  304. int index = zend_hash_num_elements(ht) - 1;
  305. php_proto_zend_hash_index_update(
  306. ht, index, memory, sizeof(zend_string*), NULL);
  307. return len;
  308. }
  309. #endif
  310. // Appends a submessage to a repeated field.
  311. static void *appendsubmsg_handler(void *closure, const void *hd) {
  312. zval* array = (zval*)closure;
  313. TSRMLS_FETCH();
  314. RepeatedField* intern = UNBOX(RepeatedField, array);
  315. const submsg_handlerdata_t *submsgdata = hd;
  316. Descriptor* subdesc =
  317. UNBOX_HASHTABLE_VALUE(Descriptor, get_def_obj((void*)submsgdata->md));
  318. zend_class_entry* subklass = subdesc->klass;
  319. MessageHeader* submsg;
  320. #if PHP_MAJOR_VERSION < 7
  321. zval* val = NULL;
  322. MAKE_STD_ZVAL(val);
  323. ZVAL_OBJ(val, subklass->create_object(subklass TSRMLS_CC));
  324. repeated_field_push_native(intern, &val);
  325. submsg = UNBOX(MessageHeader, val);
  326. #else
  327. zend_object* obj = subklass->create_object(subklass TSRMLS_CC);
  328. repeated_field_push_native(intern, &obj);
  329. submsg = (MessageHeader*)((char*)obj - XtOffsetOf(MessageHeader, std));
  330. #endif
  331. custom_data_init(subklass, submsg PHP_PROTO_TSRMLS_CC);
  332. return submsg;
  333. }
  334. // Sets a non-repeated submessage field in a message.
  335. static void *submsg_handler(void *closure, const void *hd) {
  336. MessageHeader* msg = closure;
  337. const submsg_handlerdata_t* submsgdata = hd;
  338. TSRMLS_FETCH();
  339. Descriptor* subdesc =
  340. UNBOX_HASHTABLE_VALUE(Descriptor, get_def_obj((void*)submsgdata->md));
  341. zend_class_entry* subklass = subdesc->klass;
  342. zval* submsg_php;
  343. MessageHeader* submsg;
  344. if (Z_TYPE_P(CACHED_PTR_TO_ZVAL_PTR(DEREF(message_data(msg), submsgdata->ofs,
  345. CACHED_VALUE*))) == IS_NULL) {
  346. #if PHP_MAJOR_VERSION < 7
  347. zval* val = NULL;
  348. MAKE_STD_ZVAL(val);
  349. ZVAL_OBJ(val, subklass->create_object(subklass TSRMLS_CC));
  350. MessageHeader* intern = UNBOX(MessageHeader, val);
  351. custom_data_init(subklass, intern PHP_PROTO_TSRMLS_CC);
  352. php_proto_zval_ptr_dtor(*DEREF(message_data(msg), submsgdata->ofs, zval**));
  353. *DEREF(message_data(msg), submsgdata->ofs, zval**) = val;
  354. #else
  355. zend_object* obj = subklass->create_object(subklass TSRMLS_CC);
  356. ZVAL_OBJ(DEREF(message_data(msg), submsgdata->ofs, zval*), obj);
  357. MessageHeader* intern = UNBOX_HASHTABLE_VALUE(MessageHeader, obj);
  358. custom_data_init(subklass, intern PHP_PROTO_TSRMLS_CC);
  359. #endif
  360. }
  361. submsg_php = CACHED_PTR_TO_ZVAL_PTR(
  362. DEREF(message_data(msg), submsgdata->ofs, CACHED_VALUE*));
  363. submsg = UNBOX(MessageHeader, submsg_php);
  364. return submsg;
  365. }
  366. // Handler data for startmap/endmap handlers.
  367. typedef struct {
  368. size_t ofs;
  369. upb_fieldtype_t key_field_type;
  370. upb_fieldtype_t value_field_type;
  371. // We know that we can hold this reference because the handlerdata has the
  372. // same lifetime as the upb_handlers struct, and the upb_handlers struct holds
  373. // a reference to the upb_msgdef, which in turn has references to its subdefs.
  374. const upb_def* value_field_subdef;
  375. } map_handlerdata_t;
  376. // Temporary frame for map parsing: at the beginning of a map entry message, a
  377. // submsg handler allocates a frame to hold (i) a reference to the Map object
  378. // into which this message will be inserted and (ii) storage slots to
  379. // temporarily hold the key and value for this map entry until the end of the
  380. // submessage. When the submessage ends, another handler is called to insert the
  381. // value into the map.
  382. typedef struct {
  383. char key_storage[NATIVE_SLOT_MAX_SIZE];
  384. char value_storage[NATIVE_SLOT_MAX_SIZE];
  385. } map_parse_frame_data_t;
  386. PHP_PROTO_WRAP_OBJECT_START(map_parse_frame_t)
  387. map_parse_frame_data_t* data; // Place needs to be consistent with
  388. // MessageHeader.
  389. zval* map;
  390. // In php7, we cannot allocate zval dynamically. So we need to add zval here
  391. // to help decoding.
  392. zval key_zval;
  393. zval value_zval;
  394. PHP_PROTO_WRAP_OBJECT_END
  395. typedef struct map_parse_frame_t map_parse_frame_t;
  396. static void map_slot_init(void* memory, upb_fieldtype_t type, zval* cache) {
  397. switch (type) {
  398. case UPB_TYPE_STRING:
  399. case UPB_TYPE_BYTES: {
  400. #if PHP_MAJOR_VERSION < 7
  401. // Store zval** in memory in order to be consistent with the layout of
  402. // singular fields.
  403. zval** holder = ALLOC(zval*);
  404. zval* tmp;
  405. MAKE_STD_ZVAL(tmp);
  406. PHP_PROTO_ZVAL_STRINGL(tmp, "", 0, 1);
  407. *holder = tmp;
  408. *(zval***)memory = holder;
  409. #else
  410. *(zval**)memory = cache;
  411. PHP_PROTO_ZVAL_STRINGL(*(zval**)memory, "", 0, 1);
  412. #endif
  413. break;
  414. }
  415. case UPB_TYPE_MESSAGE: {
  416. #if PHP_MAJOR_VERSION < 7
  417. zval** holder = ALLOC(zval*);
  418. zval* tmp;
  419. MAKE_STD_ZVAL(tmp);
  420. ZVAL_NULL(tmp);
  421. *holder = tmp;
  422. *(zval***)memory = holder;
  423. #else
  424. *(zval**)memory = cache;
  425. ZVAL_NULL(*(zval**)memory);
  426. #endif
  427. break;
  428. }
  429. default:
  430. native_slot_init(type, memory, NULL);
  431. }
  432. }
  433. static void map_slot_uninit(void* memory, upb_fieldtype_t type) {
  434. switch (type) {
  435. case UPB_TYPE_MESSAGE:
  436. case UPB_TYPE_STRING:
  437. case UPB_TYPE_BYTES: {
  438. #if PHP_MAJOR_VERSION < 7
  439. zval** holder = *(zval***)memory;
  440. php_proto_zval_ptr_dtor(*holder);
  441. FREE(holder);
  442. #else
  443. php_proto_zval_ptr_dtor(*(zval**)memory);
  444. #endif
  445. break;
  446. }
  447. default:
  448. break;
  449. }
  450. }
  451. static void map_slot_key(upb_fieldtype_t type, const void* from,
  452. const char** keyval,
  453. size_t* length) {
  454. if (type == UPB_TYPE_STRING) {
  455. #if PHP_MAJOR_VERSION < 7
  456. zval* key_php = **(zval***)from;
  457. #else
  458. zval* key_php = *(zval**)from;
  459. #endif
  460. *keyval = Z_STRVAL_P(key_php);
  461. *length = Z_STRLEN_P(key_php);
  462. } else {
  463. *keyval = from;
  464. *length = native_slot_size(type);
  465. }
  466. }
  467. static void map_slot_value(upb_fieldtype_t type, const void* from,
  468. upb_value* v) {
  469. size_t len;
  470. void* to = upb_value_memory(v);
  471. #ifndef NDEBUG
  472. v->ctype = UPB_CTYPE_UINT64;
  473. #endif
  474. memset(to, 0, native_slot_size(type));
  475. switch (type) {
  476. #if PHP_MAJOR_VERSION < 7
  477. case UPB_TYPE_STRING:
  478. case UPB_TYPE_BYTES:
  479. case UPB_TYPE_MESSAGE: {
  480. *(zval**)to = **(zval***)from;
  481. Z_ADDREF_PP((zval**)to);
  482. break;
  483. }
  484. #else
  485. case UPB_TYPE_STRING:
  486. case UPB_TYPE_BYTES:
  487. *(zend_string**)to = Z_STR_P(*(zval**)from);
  488. zend_string_addref(*(zend_string**)to);
  489. break;
  490. case UPB_TYPE_MESSAGE:
  491. *(zend_object**)to = Z_OBJ_P(*(zval**)from);
  492. ++GC_REFCOUNT(*(zend_object**)to);
  493. break;
  494. #endif
  495. default:
  496. len = native_slot_size(type);
  497. memcpy(to, from, len);
  498. }
  499. }
  500. // Handler to begin a map entry: allocates a temporary frame. This is the
  501. // 'startsubmsg' handler on the msgdef that contains the map field.
  502. static void *startmapentry_handler(void *closure, const void *hd) {
  503. MessageHeader* msg = closure;
  504. const map_handlerdata_t* mapdata = hd;
  505. zval* map = CACHED_PTR_TO_ZVAL_PTR(
  506. DEREF(message_data(msg), mapdata->ofs, CACHED_VALUE*));
  507. map_parse_frame_t* frame = ALLOC(map_parse_frame_t);
  508. frame->data = ALLOC(map_parse_frame_data_t);
  509. frame->map = map;
  510. map_slot_init(&frame->data->key_storage, mapdata->key_field_type,
  511. &frame->key_zval);
  512. map_slot_init(&frame->data->value_storage, mapdata->value_field_type,
  513. &frame->value_zval);
  514. return frame;
  515. }
  516. // Handler to end a map entry: inserts the value defined during the message into
  517. // the map. This is the 'endmsg' handler on the map entry msgdef.
  518. static bool endmap_handler(void* closure, const void* hd, upb_status* s) {
  519. map_parse_frame_t* frame = closure;
  520. const map_handlerdata_t* mapdata = hd;
  521. TSRMLS_FETCH();
  522. Map *map = UNBOX(Map, frame->map);
  523. const char* keyval = NULL;
  524. upb_value v;
  525. size_t length;
  526. map_slot_key(map->key_type, &frame->data->key_storage, &keyval, &length);
  527. map_slot_value(map->value_type, &frame->data->value_storage, &v);
  528. map_index_set(map, keyval, length, v);
  529. map_slot_uninit(&frame->data->key_storage, mapdata->key_field_type);
  530. map_slot_uninit(&frame->data->value_storage, mapdata->value_field_type);
  531. FREE(frame->data);
  532. FREE(frame);
  533. return true;
  534. }
  535. // Allocates a new map_handlerdata_t given the map entry message definition. If
  536. // the offset of the field within the parent message is also given, that is
  537. // added to the handler data as well. Note that this is called *twice* per map
  538. // field: once in the parent message handler setup when setting the startsubmsg
  539. // handler and once in the map entry message handler setup when setting the
  540. // key/value and endmsg handlers. The reason is that there is no easy way to
  541. // pass the handlerdata down to the sub-message handler setup.
  542. static map_handlerdata_t* new_map_handlerdata(
  543. size_t ofs,
  544. const upb_msgdef* mapentry_def,
  545. Descriptor* desc) {
  546. const upb_fielddef* key_field;
  547. const upb_fielddef* value_field;
  548. // TODO(teboring): Use emalloc and efree.
  549. map_handlerdata_t* hd =
  550. (map_handlerdata_t*)malloc(sizeof(map_handlerdata_t));
  551. hd->ofs = ofs;
  552. key_field = upb_msgdef_itof(mapentry_def, MAP_KEY_FIELD);
  553. assert(key_field != NULL);
  554. hd->key_field_type = upb_fielddef_type(key_field);
  555. value_field = upb_msgdef_itof(mapentry_def, MAP_VALUE_FIELD);
  556. assert(value_field != NULL);
  557. hd->value_field_type = upb_fielddef_type(value_field);
  558. hd->value_field_subdef = upb_fielddef_subdef(value_field);
  559. return hd;
  560. }
  561. // Handlers that set primitive values in oneofs.
  562. #define DEFINE_ONEOF_HANDLER(type, ctype) \
  563. static bool oneof##type##_handler(void* closure, const void* hd, \
  564. ctype val) { \
  565. const oneof_handlerdata_t* oneofdata = hd; \
  566. MessageHeader* msg = (MessageHeader*)closure; \
  567. DEREF(message_data(closure), oneofdata->case_ofs, uint32_t) = \
  568. oneofdata->oneof_case_num; \
  569. DEREF(message_data(closure), oneofdata->ofs, ctype) = val; \
  570. return true; \
  571. }
  572. DEFINE_ONEOF_HANDLER(bool, bool)
  573. DEFINE_ONEOF_HANDLER(int32, int32_t)
  574. DEFINE_ONEOF_HANDLER(uint32, uint32_t)
  575. DEFINE_ONEOF_HANDLER(float, float)
  576. DEFINE_ONEOF_HANDLER(int64, int64_t)
  577. DEFINE_ONEOF_HANDLER(uint64, uint64_t)
  578. DEFINE_ONEOF_HANDLER(double, double)
  579. #undef DEFINE_ONEOF_HANDLER
  580. // Handlers for string/bytes in a oneof.
  581. static void *oneofbytes_handler(void *closure,
  582. const void *hd,
  583. size_t size_hint) {
  584. MessageHeader* msg = closure;
  585. const oneof_handlerdata_t *oneofdata = hd;
  586. DEREF(message_data(msg), oneofdata->case_ofs, uint32_t) =
  587. oneofdata->oneof_case_num;
  588. DEREF(message_data(msg), oneofdata->ofs, CACHED_VALUE*) =
  589. &(msg->std.properties_table)[oneofdata->property_ofs];
  590. return empty_php_string(DEREF(
  591. message_data(msg), oneofdata->ofs, CACHED_VALUE*));
  592. }
  593. static void *oneofstr_handler(void *closure,
  594. const void *hd,
  595. size_t size_hint) {
  596. // TODO(teboring): Add it back.
  597. // rb_enc_associate(str, kRubyString8bitEncoding);
  598. return oneofbytes_handler(closure, hd, size_hint);
  599. }
  600. // Handler for a submessage field in a oneof.
  601. static void* oneofsubmsg_handler(void* closure, const void* hd) {
  602. MessageHeader* msg = closure;
  603. const oneof_handlerdata_t *oneofdata = hd;
  604. uint32_t oldcase = DEREF(message_data(msg), oneofdata->case_ofs, uint32_t);
  605. TSRMLS_FETCH();
  606. Descriptor* subdesc =
  607. UNBOX_HASHTABLE_VALUE(Descriptor, get_def_obj((void*)oneofdata->md));
  608. zend_class_entry* subklass = subdesc->klass;
  609. zval* submsg_php;
  610. MessageHeader* submsg;
  611. if (oldcase != oneofdata->oneof_case_num) {
  612. // Ideally, we should clean up the old data. However, we don't even know the
  613. // type of the old data. So, we will defer the desctruction of the old data
  614. // to the time that containing message's destroyed or the same oneof field
  615. // is accessed again and find that the old data hasn't been cleaned.
  616. DEREF(message_data(msg), oneofdata->ofs, CACHED_VALUE*) =
  617. &(msg->std.properties_table)[oneofdata->property_ofs];
  618. // Old data was't cleaned when the oneof was accessed from another field.
  619. if (Z_TYPE_P(CACHED_PTR_TO_ZVAL_PTR(DEREF(
  620. message_data(msg), oneofdata->ofs, CACHED_VALUE*))) != IS_NULL) {
  621. php_proto_zval_ptr_dtor(
  622. CACHED_PTR_TO_ZVAL_PTR(
  623. DEREF(message_data(msg), oneofdata->ofs, CACHED_VALUE*)));
  624. }
  625. // Create new message.
  626. ZVAL_OBJ(CACHED_PTR_TO_ZVAL_PTR(
  627. DEREF(message_data(msg), oneofdata->ofs, CACHED_VALUE*)),
  628. subklass->create_object(subklass TSRMLS_CC));
  629. }
  630. DEREF(message_data(msg), oneofdata->case_ofs, uint32_t) =
  631. oneofdata->oneof_case_num;
  632. submsg_php = CACHED_PTR_TO_ZVAL_PTR(
  633. DEREF(message_data(msg), oneofdata->ofs, CACHED_VALUE*));
  634. submsg = UNBOX(MessageHeader, submsg_php);
  635. custom_data_init(subklass, submsg PHP_PROTO_TSRMLS_CC);
  636. return submsg;
  637. }
  638. // Set up handlers for a repeated field.
  639. static void add_handlers_for_repeated_field(upb_handlers *h,
  640. const upb_fielddef *f,
  641. size_t offset) {
  642. upb_handlerattr attr = UPB_HANDLERATTR_INITIALIZER;
  643. upb_handlerattr_sethandlerdata(&attr, newhandlerdata(h, offset));
  644. upb_handlers_setstartseq(h, f, startseq_handler, &attr);
  645. upb_handlerattr_uninit(&attr);
  646. switch (upb_fielddef_type(f)) {
  647. #define SET_HANDLER(utype, ltype) \
  648. case utype: \
  649. upb_handlers_set##ltype(h, f, append##ltype##_handler, NULL); \
  650. break;
  651. SET_HANDLER(UPB_TYPE_BOOL, bool);
  652. SET_HANDLER(UPB_TYPE_INT32, int32);
  653. SET_HANDLER(UPB_TYPE_UINT32, uint32);
  654. SET_HANDLER(UPB_TYPE_ENUM, int32);
  655. SET_HANDLER(UPB_TYPE_FLOAT, float);
  656. SET_HANDLER(UPB_TYPE_INT64, int64);
  657. SET_HANDLER(UPB_TYPE_UINT64, uint64);
  658. SET_HANDLER(UPB_TYPE_DOUBLE, double);
  659. #undef SET_HANDLER
  660. case UPB_TYPE_STRING:
  661. case UPB_TYPE_BYTES: {
  662. bool is_bytes = upb_fielddef_type(f) == UPB_TYPE_BYTES;
  663. upb_handlers_setstartstr(h, f, is_bytes ?
  664. appendbytes_handler : appendstr_handler,
  665. NULL);
  666. #if PHP_MAJOR_VERSION < 7
  667. upb_handlers_setstring(h, f, stringdata_handler, NULL);
  668. #else
  669. upb_handlers_setstring(h, f, zendstringdata_handler, NULL);
  670. #endif
  671. break;
  672. }
  673. case UPB_TYPE_MESSAGE: {
  674. upb_handlerattr attr = UPB_HANDLERATTR_INITIALIZER;
  675. upb_handlerattr_sethandlerdata(&attr, newsubmsghandlerdata(h, 0, f));
  676. upb_handlers_setstartsubmsg(h, f, appendsubmsg_handler, &attr);
  677. upb_handlerattr_uninit(&attr);
  678. break;
  679. }
  680. }
  681. }
  682. // Set up handlers for a singular field.
  683. static void add_handlers_for_singular_field(upb_handlers *h,
  684. const upb_fielddef *f,
  685. size_t offset) {
  686. switch (upb_fielddef_type(f)) {
  687. #define SET_HANDLER(utype, ltype) \
  688. case utype: { \
  689. upb_handlerattr attr = UPB_HANDLERATTR_INITIALIZER; \
  690. upb_handlerattr_sethandlerdata(&attr, newhandlerdata(h, offset)); \
  691. upb_handlers_set##ltype(h, f, ltype##_handler, &attr); \
  692. break; \
  693. }
  694. SET_HANDLER(UPB_TYPE_BOOL, bool);
  695. SET_HANDLER(UPB_TYPE_INT32, int32);
  696. SET_HANDLER(UPB_TYPE_UINT32, uint32);
  697. SET_HANDLER(UPB_TYPE_ENUM, int32);
  698. SET_HANDLER(UPB_TYPE_FLOAT, float);
  699. SET_HANDLER(UPB_TYPE_INT64, int64);
  700. SET_HANDLER(UPB_TYPE_UINT64, uint64);
  701. SET_HANDLER(UPB_TYPE_DOUBLE, double);
  702. #undef SET_HANDLER
  703. case UPB_TYPE_STRING:
  704. case UPB_TYPE_BYTES: {
  705. bool is_bytes = upb_fielddef_type(f) == UPB_TYPE_BYTES;
  706. upb_handlerattr attr = UPB_HANDLERATTR_INITIALIZER;
  707. upb_handlerattr_sethandlerdata(&attr, newhandlerdata(h, offset));
  708. upb_handlers_setstartstr(h, f,
  709. is_bytes ? bytes_handler : str_handler,
  710. &attr);
  711. upb_handlers_setstring(h, f, stringdata_handler, &attr);
  712. upb_handlerattr_uninit(&attr);
  713. break;
  714. }
  715. case UPB_TYPE_MESSAGE: {
  716. upb_handlerattr attr = UPB_HANDLERATTR_INITIALIZER;
  717. upb_handlerattr_sethandlerdata(&attr, newsubmsghandlerdata(h, offset, f));
  718. upb_handlers_setstartsubmsg(h, f, submsg_handler, &attr);
  719. upb_handlerattr_uninit(&attr);
  720. break;
  721. }
  722. }
  723. }
  724. // Adds handlers to a map field.
  725. static void add_handlers_for_mapfield(upb_handlers* h,
  726. const upb_fielddef* fielddef,
  727. size_t offset,
  728. Descriptor* desc) {
  729. const upb_msgdef* map_msgdef = upb_fielddef_msgsubdef(fielddef);
  730. map_handlerdata_t* hd = new_map_handlerdata(offset, map_msgdef, desc);
  731. upb_handlerattr attr = UPB_HANDLERATTR_INITIALIZER;
  732. upb_handlers_addcleanup(h, hd, free);
  733. upb_handlerattr_sethandlerdata(&attr, hd);
  734. upb_handlers_setstartsubmsg(h, fielddef, startmapentry_handler, &attr);
  735. upb_handlerattr_uninit(&attr);
  736. }
  737. // Adds handlers to a map-entry msgdef.
  738. static void add_handlers_for_mapentry(const upb_msgdef* msgdef, upb_handlers* h,
  739. Descriptor* desc) {
  740. const upb_fielddef* key_field = map_entry_key(msgdef);
  741. const upb_fielddef* value_field = map_entry_value(msgdef);
  742. map_handlerdata_t* hd = new_map_handlerdata(0, msgdef, desc);
  743. upb_handlerattr attr = UPB_HANDLERATTR_INITIALIZER;
  744. upb_handlers_addcleanup(h, hd, free);
  745. upb_handlerattr_sethandlerdata(&attr, hd);
  746. upb_handlers_setendmsg(h, endmap_handler, &attr);
  747. add_handlers_for_singular_field(h, key_field,
  748. offsetof(map_parse_frame_data_t,
  749. key_storage));
  750. add_handlers_for_singular_field(h, value_field,
  751. offsetof(map_parse_frame_data_t,
  752. value_storage));
  753. }
  754. // Set up handlers for a oneof field.
  755. static void add_handlers_for_oneof_field(upb_handlers *h,
  756. const upb_fielddef *f,
  757. size_t offset,
  758. size_t oneof_case_offset,
  759. int property_cache_offset) {
  760. upb_handlerattr attr = UPB_HANDLERATTR_INITIALIZER;
  761. upb_handlerattr_sethandlerdata(
  762. &attr, newoneofhandlerdata(h, offset, oneof_case_offset,
  763. property_cache_offset, f));
  764. switch (upb_fielddef_type(f)) {
  765. #define SET_HANDLER(utype, ltype) \
  766. case utype: \
  767. upb_handlers_set##ltype(h, f, oneof##ltype##_handler, &attr); \
  768. break;
  769. SET_HANDLER(UPB_TYPE_BOOL, bool);
  770. SET_HANDLER(UPB_TYPE_INT32, int32);
  771. SET_HANDLER(UPB_TYPE_UINT32, uint32);
  772. SET_HANDLER(UPB_TYPE_ENUM, int32);
  773. SET_HANDLER(UPB_TYPE_FLOAT, float);
  774. SET_HANDLER(UPB_TYPE_INT64, int64);
  775. SET_HANDLER(UPB_TYPE_UINT64, uint64);
  776. SET_HANDLER(UPB_TYPE_DOUBLE, double);
  777. #undef SET_HANDLER
  778. case UPB_TYPE_STRING:
  779. case UPB_TYPE_BYTES: {
  780. bool is_bytes = upb_fielddef_type(f) == UPB_TYPE_BYTES;
  781. upb_handlers_setstartstr(h, f, is_bytes ?
  782. oneofbytes_handler : oneofstr_handler,
  783. &attr);
  784. upb_handlers_setstring(h, f, stringdata_handler, NULL);
  785. break;
  786. }
  787. case UPB_TYPE_MESSAGE: {
  788. upb_handlers_setstartsubmsg(h, f, oneofsubmsg_handler, &attr);
  789. break;
  790. }
  791. }
  792. upb_handlerattr_uninit(&attr);
  793. }
  794. static void add_handlers_for_message(const void* closure,
  795. upb_handlers* h) {
  796. const upb_msgdef* msgdef = upb_handlers_msgdef(h);
  797. TSRMLS_FETCH();
  798. Descriptor* desc =
  799. UNBOX_HASHTABLE_VALUE(Descriptor, get_def_obj((void*)msgdef));
  800. upb_msg_field_iter i;
  801. // If this is a mapentry message type, set up a special set of handlers and
  802. // bail out of the normal (user-defined) message type handling.
  803. if (upb_msgdef_mapentry(msgdef)) {
  804. add_handlers_for_mapentry(msgdef, h, desc);
  805. return;
  806. }
  807. // Ensure layout exists. We may be invoked to create handlers for a given
  808. // message if we are included as a submsg of another message type before our
  809. // class is actually built, so to work around this, we just create the layout
  810. // (and handlers, in the class-building function) on-demand.
  811. if (desc->layout == NULL) {
  812. desc->layout = create_layout(desc->msgdef);
  813. }
  814. for (upb_msg_field_begin(&i, desc->msgdef);
  815. !upb_msg_field_done(&i);
  816. upb_msg_field_next(&i)) {
  817. const upb_fielddef *f = upb_msg_iter_field(&i);
  818. size_t offset = desc->layout->fields[upb_fielddef_index(f)].offset;
  819. if (upb_fielddef_containingoneof(f)) {
  820. size_t oneof_case_offset =
  821. desc->layout->fields[upb_fielddef_index(f)].case_offset;
  822. int property_cache_index =
  823. desc->layout->fields[upb_fielddef_index(f)].cache_index;
  824. add_handlers_for_oneof_field(h, f, offset, oneof_case_offset,
  825. property_cache_index);
  826. } else if (is_map_field(f)) {
  827. add_handlers_for_mapfield(h, f, offset, desc);
  828. } else if (upb_fielddef_isseq(f)) {
  829. add_handlers_for_repeated_field(h, f, offset);
  830. } else {
  831. add_handlers_for_singular_field(h, f, offset);
  832. }
  833. }
  834. }
  835. // Creates upb handlers for populating a message.
  836. static const upb_handlers *new_fill_handlers(Descriptor* desc,
  837. const void* owner) {
  838. // TODO(cfallin, haberman): once upb gets a caching/memoization layer for
  839. // handlers, reuse subdef handlers so that e.g. if we already parse
  840. // B-with-field-of-type-C, we don't have to rebuild the whole hierarchy to
  841. // parse A-with-field-of-type-B-with-field-of-type-C.
  842. return upb_handlers_newfrozen(desc->msgdef, owner,
  843. add_handlers_for_message, NULL);
  844. }
  845. // Constructs the handlers for filling a message's data into an in-memory
  846. // object.
  847. const upb_handlers* get_fill_handlers(Descriptor* desc) {
  848. if (!desc->fill_handlers) {
  849. desc->fill_handlers =
  850. new_fill_handlers(desc, &desc->fill_handlers);
  851. }
  852. return desc->fill_handlers;
  853. }
  854. const upb_pbdecodermethod *new_fillmsg_decodermethod(Descriptor* desc,
  855. const void* owner) {
  856. const upb_handlers* handlers = get_fill_handlers(desc);
  857. upb_pbdecodermethodopts opts;
  858. upb_pbdecodermethodopts_init(&opts, handlers);
  859. return upb_pbdecodermethod_new(&opts, owner);
  860. }
  861. static const upb_pbdecodermethod *msgdef_decodermethod(Descriptor* desc) {
  862. if (desc->fill_method == NULL) {
  863. desc->fill_method = new_fillmsg_decodermethod(
  864. desc, &desc->fill_method);
  865. }
  866. return desc->fill_method;
  867. }
  868. static const upb_json_parsermethod *msgdef_jsonparsermethod(Descriptor* desc) {
  869. if (desc->json_fill_method == NULL) {
  870. desc->json_fill_method =
  871. upb_json_parsermethod_new(desc->msgdef, &desc->json_fill_method);
  872. }
  873. return desc->json_fill_method;
  874. }
  875. // -----------------------------------------------------------------------------
  876. // Serializing.
  877. // -----------------------------------------------------------------------------
  878. static void putmsg(zval* msg, const Descriptor* desc, upb_sink* sink,
  879. int depth TSRMLS_DC);
  880. static void putrawmsg(MessageHeader* msg, const Descriptor* desc,
  881. upb_sink* sink, int depth TSRMLS_DC);
  882. static void putstr(zval* str, const upb_fielddef* f, upb_sink* sink);
  883. static void putrawstr(const char* str, int len, const upb_fielddef* f,
  884. upb_sink* sink);
  885. static void putsubmsg(zval* submsg, const upb_fielddef* f, upb_sink* sink,
  886. int depth TSRMLS_DC);
  887. static void putrawsubmsg(MessageHeader* submsg, const upb_fielddef* f,
  888. upb_sink* sink, int depth TSRMLS_DC);
  889. static void putarray(zval* array, const upb_fielddef* f, upb_sink* sink,
  890. int depth TSRMLS_DC);
  891. static void putmap(zval* map, const upb_fielddef* f, upb_sink* sink,
  892. int depth TSRMLS_DC);
  893. static upb_selector_t getsel(const upb_fielddef* f, upb_handlertype_t type) {
  894. upb_selector_t ret;
  895. bool ok = upb_handlers_getselector(f, type, &ret);
  896. UPB_ASSERT(ok);
  897. return ret;
  898. }
  899. static void put_optional_value(const void* memory, int len, const upb_fielddef* f,
  900. int depth, upb_sink* sink TSRMLS_DC) {
  901. assert(upb_fielddef_label(f) == UPB_LABEL_OPTIONAL);
  902. switch (upb_fielddef_type(f)) {
  903. #define T(upbtypeconst, upbtype, ctype, default_value) \
  904. case upbtypeconst: { \
  905. ctype value = DEREF(memory, 0, ctype); \
  906. if (value != default_value) { \
  907. upb_selector_t sel = getsel(f, upb_handlers_getprimitivehandlertype(f)); \
  908. upb_sink_put##upbtype(sink, sel, value); \
  909. } \
  910. } break;
  911. T(UPB_TYPE_FLOAT, float, float, 0.0)
  912. T(UPB_TYPE_DOUBLE, double, double, 0.0)
  913. T(UPB_TYPE_BOOL, bool, uint8_t, 0)
  914. T(UPB_TYPE_ENUM, int32, int32_t, 0)
  915. T(UPB_TYPE_INT32, int32, int32_t, 0)
  916. T(UPB_TYPE_UINT32, uint32, uint32_t, 0)
  917. T(UPB_TYPE_INT64, int64, int64_t, 0)
  918. T(UPB_TYPE_UINT64, uint64, uint64_t, 0)
  919. #undef T
  920. case UPB_TYPE_STRING:
  921. case UPB_TYPE_BYTES:
  922. putrawstr(memory, len, f, sink);
  923. break;
  924. case UPB_TYPE_MESSAGE: {
  925. #if PHP_MAJOR_VERSION < 7
  926. MessageHeader *submsg = UNBOX(MessageHeader, *(zval**)memory);
  927. #else
  928. MessageHeader *submsg =
  929. (MessageHeader*)((char*)(*(zend_object**)memory) -
  930. XtOffsetOf(MessageHeader, std));
  931. #endif
  932. putrawsubmsg(submsg, f, sink, depth TSRMLS_CC);
  933. break;
  934. }
  935. default:
  936. assert(false);
  937. }
  938. }
  939. // Only string/bytes fields are stored as zval.
  940. static const char* raw_value(void* memory, const upb_fielddef* f) {
  941. switch (upb_fielddef_type(f)) {
  942. case UPB_TYPE_STRING:
  943. case UPB_TYPE_BYTES:
  944. #if PHP_MAJOR_VERSION < 7
  945. return Z_STRVAL_PP((zval**)memory);
  946. #else
  947. return ZSTR_VAL(*(zend_string**)memory);
  948. #endif
  949. break;
  950. default:
  951. return memory;
  952. }
  953. }
  954. static int raw_value_len(void* memory, int len, const upb_fielddef* f) {
  955. switch (upb_fielddef_type(f)) {
  956. case UPB_TYPE_STRING:
  957. case UPB_TYPE_BYTES:
  958. #if PHP_MAJOR_VERSION < 7
  959. return Z_STRLEN_PP((zval**)memory);
  960. #else
  961. return ZSTR_LEN(*(zend_string**)memory);
  962. #endif
  963. default:
  964. return len;
  965. }
  966. }
  967. static void putmap(zval* map, const upb_fielddef* f, upb_sink* sink,
  968. int depth TSRMLS_DC) {
  969. upb_sink subsink;
  970. const upb_fielddef* key_field;
  971. const upb_fielddef* value_field;
  972. MapIter it;
  973. int len, size;
  974. assert(map != NULL);
  975. Map* intern = UNBOX(Map, map);
  976. size = upb_strtable_count(&intern->table);
  977. if (size == 0) return;
  978. upb_sink_startseq(sink, getsel(f, UPB_HANDLER_STARTSEQ), &subsink);
  979. assert(upb_fielddef_type(f) == UPB_TYPE_MESSAGE);
  980. key_field = map_field_key(f);
  981. value_field = map_field_value(f);
  982. for (map_begin(map, &it TSRMLS_CC); !map_done(&it); map_next(&it)) {
  983. upb_status status;
  984. upb_sink entry_sink;
  985. upb_sink_startsubmsg(&subsink, getsel(f, UPB_HANDLER_STARTSUBMSG),
  986. &entry_sink);
  987. upb_sink_startmsg(&entry_sink);
  988. // Serialize key.
  989. const char *key = map_iter_key(&it, &len);
  990. put_optional_value(key, len, key_field, depth + 1, &entry_sink TSRMLS_CC);
  991. // Serialize value.
  992. upb_value value = map_iter_value(&it, &len);
  993. put_optional_value(raw_value(upb_value_memory(&value), value_field),
  994. raw_value_len(upb_value_memory(&value), len, value_field),
  995. value_field, depth + 1, &entry_sink TSRMLS_CC);
  996. upb_sink_endmsg(&entry_sink, &status);
  997. upb_sink_endsubmsg(&subsink, getsel(f, UPB_HANDLER_ENDSUBMSG));
  998. }
  999. upb_sink_endseq(sink, getsel(f, UPB_HANDLER_ENDSEQ));
  1000. }
  1001. static void putmsg(zval* msg_php, const Descriptor* desc, upb_sink* sink,
  1002. int depth TSRMLS_DC) {
  1003. MessageHeader* msg = UNBOX(MessageHeader, msg_php);
  1004. putrawmsg(msg, desc, sink, depth TSRMLS_CC);
  1005. }
  1006. static void putrawmsg(MessageHeader* msg, const Descriptor* desc,
  1007. upb_sink* sink, int depth TSRMLS_DC) {
  1008. upb_msg_field_iter i;
  1009. upb_status status;
  1010. upb_sink_startmsg(sink);
  1011. // Protect against cycles (possible because users may freely reassign message
  1012. // and repeated fields) by imposing a maximum recursion depth.
  1013. if (depth > ENCODE_MAX_NESTING) {
  1014. zend_error(E_ERROR,
  1015. "Maximum recursion depth exceeded during encoding.");
  1016. }
  1017. for (upb_msg_field_begin(&i, desc->msgdef); !upb_msg_field_done(&i);
  1018. upb_msg_field_next(&i)) {
  1019. upb_fielddef* f = upb_msg_iter_field(&i);
  1020. uint32_t offset = desc->layout->fields[upb_fielddef_index(f)].offset;
  1021. if (upb_fielddef_containingoneof(f)) {
  1022. uint32_t oneof_case_offset =
  1023. desc->layout->fields[upb_fielddef_index(f)].case_offset;
  1024. // For a oneof, check that this field is actually present -- skip all the
  1025. // below if not.
  1026. if (DEREF(message_data(msg), oneof_case_offset, uint32_t) !=
  1027. upb_fielddef_number(f)) {
  1028. continue;
  1029. }
  1030. // Otherwise, fall through to the appropriate singular-field handler
  1031. // below.
  1032. }
  1033. if (is_map_field(f)) {
  1034. zval* map = CACHED_PTR_TO_ZVAL_PTR(
  1035. DEREF(message_data(msg), offset, CACHED_VALUE*));
  1036. if (map != NULL) {
  1037. putmap(map, f, sink, depth TSRMLS_CC);
  1038. }
  1039. } else if (upb_fielddef_isseq(f)) {
  1040. zval* array = CACHED_PTR_TO_ZVAL_PTR(
  1041. DEREF(message_data(msg), offset, CACHED_VALUE*));
  1042. if (array != NULL) {
  1043. putarray(array, f, sink, depth TSRMLS_CC);
  1044. }
  1045. } else if (upb_fielddef_isstring(f)) {
  1046. zval* str = CACHED_PTR_TO_ZVAL_PTR(
  1047. DEREF(message_data(msg), offset, CACHED_VALUE*));
  1048. if (Z_STRLEN_P(str) > 0) {
  1049. putstr(str, f, sink);
  1050. }
  1051. } else if (upb_fielddef_issubmsg(f)) {
  1052. putsubmsg(CACHED_PTR_TO_ZVAL_PTR(
  1053. DEREF(message_data(msg), offset, CACHED_VALUE*)),
  1054. f, sink, depth TSRMLS_CC);
  1055. } else {
  1056. upb_selector_t sel = getsel(f, upb_handlers_getprimitivehandlertype(f));
  1057. #define T(upbtypeconst, upbtype, ctype, default_value) \
  1058. case upbtypeconst: { \
  1059. ctype value = DEREF(message_data(msg), offset, ctype); \
  1060. if (value != default_value) { \
  1061. upb_sink_put##upbtype(sink, sel, value); \
  1062. } \
  1063. } break;
  1064. switch (upb_fielddef_type(f)) {
  1065. T(UPB_TYPE_FLOAT, float, float, 0.0)
  1066. T(UPB_TYPE_DOUBLE, double, double, 0.0)
  1067. T(UPB_TYPE_BOOL, bool, uint8_t, 0)
  1068. case UPB_TYPE_ENUM:
  1069. T(UPB_TYPE_INT32, int32, int32_t, 0)
  1070. T(UPB_TYPE_UINT32, uint32, uint32_t, 0)
  1071. T(UPB_TYPE_INT64, int64, int64_t, 0)
  1072. T(UPB_TYPE_UINT64, uint64, uint64_t, 0)
  1073. case UPB_TYPE_STRING:
  1074. case UPB_TYPE_BYTES:
  1075. case UPB_TYPE_MESSAGE:
  1076. zend_error(E_ERROR, "Internal error.");
  1077. }
  1078. #undef T
  1079. }
  1080. }
  1081. upb_sink_endmsg(sink, &status);
  1082. }
  1083. static void putstr(zval* str, const upb_fielddef *f, upb_sink *sink) {
  1084. upb_sink subsink;
  1085. if (ZVAL_IS_NULL(str)) return;
  1086. assert(Z_TYPE_P(str) == IS_STRING);
  1087. // Ensure that the string has the correct encoding. We also check at field-set
  1088. // time, but the user may have mutated the string object since then.
  1089. if (upb_fielddef_type(f) == UPB_TYPE_STRING &&
  1090. !is_structurally_valid_utf8(Z_STRVAL_P(str), Z_STRLEN_P(str))) {
  1091. zend_error(E_USER_ERROR, "Given string is not UTF8 encoded.");
  1092. return;
  1093. }
  1094. upb_sink_startstr(sink, getsel(f, UPB_HANDLER_STARTSTR), Z_STRLEN_P(str),
  1095. &subsink);
  1096. upb_sink_putstring(&subsink, getsel(f, UPB_HANDLER_STRING), Z_STRVAL_P(str),
  1097. Z_STRLEN_P(str), NULL);
  1098. upb_sink_endstr(sink, getsel(f, UPB_HANDLER_ENDSTR));
  1099. }
  1100. static void putrawstr(const char* str, int len, const upb_fielddef* f,
  1101. upb_sink* sink) {
  1102. upb_sink subsink;
  1103. if (len == 0) return;
  1104. // Ensure that the string has the correct encoding. We also check at field-set
  1105. // time, but the user may have mutated the string object since then.
  1106. if (upb_fielddef_type(f) == UPB_TYPE_STRING &&
  1107. !is_structurally_valid_utf8(str, len)) {
  1108. zend_error(E_USER_ERROR, "Given string is not UTF8 encoded.");
  1109. return;
  1110. }
  1111. upb_sink_startstr(sink, getsel(f, UPB_HANDLER_STARTSTR), len, &subsink);
  1112. upb_sink_putstring(&subsink, getsel(f, UPB_HANDLER_STRING), str, len, NULL);
  1113. upb_sink_endstr(sink, getsel(f, UPB_HANDLER_ENDSTR));
  1114. }
  1115. static void putsubmsg(zval* submsg_php, const upb_fielddef* f, upb_sink* sink,
  1116. int depth TSRMLS_DC) {
  1117. if (Z_TYPE_P(submsg_php) == IS_NULL) return;
  1118. MessageHeader *submsg = UNBOX(MessageHeader, submsg_php);
  1119. putrawsubmsg(submsg, f, sink, depth TSRMLS_CC);
  1120. }
  1121. static void putrawsubmsg(MessageHeader* submsg, const upb_fielddef* f,
  1122. upb_sink* sink, int depth TSRMLS_DC) {
  1123. upb_sink subsink;
  1124. Descriptor* subdesc =
  1125. UNBOX_HASHTABLE_VALUE(Descriptor, get_def_obj(upb_fielddef_msgsubdef(f)));
  1126. upb_sink_startsubmsg(sink, getsel(f, UPB_HANDLER_STARTSUBMSG), &subsink);
  1127. putrawmsg(submsg, subdesc, &subsink, depth + 1 TSRMLS_CC);
  1128. upb_sink_endsubmsg(sink, getsel(f, UPB_HANDLER_ENDSUBMSG));
  1129. }
  1130. static void putarray(zval* array, const upb_fielddef* f, upb_sink* sink,
  1131. int depth TSRMLS_DC) {
  1132. upb_sink subsink;
  1133. upb_fieldtype_t type = upb_fielddef_type(f);
  1134. upb_selector_t sel = 0;
  1135. int size, i;
  1136. assert(array != NULL);
  1137. RepeatedField* intern = UNBOX(RepeatedField, array);
  1138. HashTable *ht = PHP_PROTO_HASH_OF(intern->array);
  1139. size = zend_hash_num_elements(ht);
  1140. // size = zend_hash_num_elements(PHP_PROTO_HASH_OF(intern->array));
  1141. if (size == 0) return;
  1142. upb_sink_startseq(sink, getsel(f, UPB_HANDLER_STARTSEQ), &subsink);
  1143. if (upb_fielddef_isprimitive(f)) {
  1144. sel = getsel(f, upb_handlers_getprimitivehandlertype(f));
  1145. }
  1146. for (i = 0; i < size; i++) {
  1147. void* memory = repeated_field_index_native(intern, i TSRMLS_CC);
  1148. switch (type) {
  1149. #define T(upbtypeconst, upbtype, ctype) \
  1150. case upbtypeconst: \
  1151. upb_sink_put##upbtype(&subsink, sel, *((ctype*)memory)); \
  1152. break;
  1153. T(UPB_TYPE_FLOAT, float, float)
  1154. T(UPB_TYPE_DOUBLE, double, double)
  1155. T(UPB_TYPE_BOOL, bool, int8_t)
  1156. case UPB_TYPE_ENUM:
  1157. T(UPB_TYPE_INT32, int32, int32_t)
  1158. T(UPB_TYPE_UINT32, uint32, uint32_t)
  1159. T(UPB_TYPE_INT64, int64, int64_t)
  1160. T(UPB_TYPE_UINT64, uint64, uint64_t)
  1161. case UPB_TYPE_STRING:
  1162. case UPB_TYPE_BYTES: {
  1163. #if PHP_MAJOR_VERSION < 7
  1164. const char* rawstr = Z_STRVAL_P(*(zval**)memory);
  1165. int len = Z_STRLEN_P(*(zval**)memory);
  1166. #else
  1167. const char* rawstr = ZSTR_VAL(*(zend_string**)memory);
  1168. int len = ZSTR_LEN(*(zend_string**)memory);
  1169. #endif
  1170. putrawstr(rawstr, len, f, &subsink);
  1171. break;
  1172. }
  1173. case UPB_TYPE_MESSAGE: {
  1174. #if PHP_MAJOR_VERSION < 7
  1175. MessageHeader *submsg = UNBOX(MessageHeader, *(zval**)memory);
  1176. #else
  1177. MessageHeader *submsg =
  1178. (MessageHeader*)((char*)(*(zend_object**)memory) -
  1179. XtOffsetOf(MessageHeader, std));
  1180. #endif
  1181. putrawsubmsg(submsg, f, &subsink, depth TSRMLS_CC);
  1182. break;
  1183. }
  1184. #undef T
  1185. }
  1186. }
  1187. upb_sink_endseq(sink, getsel(f, UPB_HANDLER_ENDSEQ));
  1188. }
  1189. static const upb_handlers* msgdef_pb_serialize_handlers(Descriptor* desc) {
  1190. if (desc->pb_serialize_handlers == NULL) {
  1191. desc->pb_serialize_handlers =
  1192. upb_pb_encoder_newhandlers(desc->msgdef, &desc->pb_serialize_handlers);
  1193. }
  1194. return desc->pb_serialize_handlers;
  1195. }
  1196. static const upb_handlers* msgdef_json_serialize_handlers(
  1197. Descriptor* desc, bool preserve_proto_fieldnames) {
  1198. if (preserve_proto_fieldnames) {
  1199. if (desc->json_serialize_handlers == NULL) {
  1200. desc->json_serialize_handlers =
  1201. upb_json_printer_newhandlers(
  1202. desc->msgdef, true, &desc->json_serialize_handlers);
  1203. }
  1204. return desc->json_serialize_handlers;
  1205. } else {
  1206. if (desc->json_serialize_handlers_preserve == NULL) {
  1207. desc->json_serialize_handlers_preserve =
  1208. upb_json_printer_newhandlers(
  1209. desc->msgdef, false, &desc->json_serialize_handlers_preserve);
  1210. }
  1211. return desc->json_serialize_handlers_preserve;
  1212. }
  1213. }
  1214. // -----------------------------------------------------------------------------
  1215. // PHP encode/decode methods
  1216. // -----------------------------------------------------------------------------
  1217. PHP_METHOD(Message, serializeToString) {
  1218. Descriptor* desc =
  1219. UNBOX_HASHTABLE_VALUE(Descriptor, get_ce_obj(Z_OBJCE_P(getThis())));
  1220. stringsink sink;
  1221. stringsink_init(&sink);
  1222. {
  1223. const upb_handlers* serialize_handlers = msgdef_pb_serialize_handlers(desc);
  1224. stackenv se;
  1225. upb_pb_encoder* encoder;
  1226. stackenv_init(&se, "Error occurred during encoding: %s");
  1227. encoder = upb_pb_encoder_create(&se.env, serialize_handlers, &sink.sink);
  1228. putmsg(getThis(), desc, upb_pb_encoder_input(encoder), 0 TSRMLS_CC);
  1229. PHP_PROTO_RETVAL_STRINGL(sink.ptr, sink.len, 1);
  1230. stackenv_uninit(&se);
  1231. stringsink_uninit(&sink);
  1232. }
  1233. }
  1234. PHP_METHOD(Message, mergeFromString) {
  1235. Descriptor* desc =
  1236. UNBOX_HASHTABLE_VALUE(Descriptor, get_ce_obj(Z_OBJCE_P(getThis())));
  1237. MessageHeader* msg = UNBOX(MessageHeader, getThis());
  1238. char *data = NULL;
  1239. PHP_PROTO_SIZE data_len;
  1240. if (zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC, "s", &data, &data_len) ==
  1241. FAILURE) {
  1242. return;
  1243. }
  1244. {
  1245. const upb_pbdecodermethod* method = msgdef_decodermethod(desc);
  1246. const upb_handlers* h = upb_pbdecodermethod_desthandlers(method);
  1247. stackenv se;
  1248. upb_sink sink;
  1249. upb_pbdecoder* decoder;
  1250. stackenv_init(&se, "Error occurred during parsing: %s");
  1251. upb_sink_reset(&sink, h, msg);
  1252. decoder = upb_pbdecoder_create(&se.env, method, &sink);
  1253. upb_bufsrc_putbuf(data, data_len, upb_pbdecoder_input(decoder));
  1254. stackenv_uninit(&se);
  1255. }
  1256. }
  1257. PHP_METHOD(Message, jsonEncode) {
  1258. Descriptor* desc =
  1259. UNBOX_HASHTABLE_VALUE(Descriptor, get_ce_obj(Z_OBJCE_P(getThis())));
  1260. zend_bool preserve_proto_fieldnames = false;
  1261. if (zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC, "|b",
  1262. &preserve_proto_fieldnames) == FAILURE) {
  1263. return;
  1264. }
  1265. stringsink sink;
  1266. stringsink_init(&sink);
  1267. {
  1268. const upb_handlers* serialize_handlers =
  1269. msgdef_json_serialize_handlers(desc, preserve_proto_fieldnames);
  1270. upb_json_printer* printer;
  1271. stackenv se;
  1272. stackenv_init(&se, "Error occurred during encoding: %s");
  1273. printer = upb_json_printer_create(&se.env, serialize_handlers, &sink.sink);
  1274. putmsg(getThis(), desc, upb_json_printer_input(printer), 0 TSRMLS_CC);
  1275. PHP_PROTO_RETVAL_STRINGL(sink.ptr, sink.len, 1);
  1276. stackenv_uninit(&se);
  1277. stringsink_uninit(&sink);
  1278. }
  1279. }
  1280. PHP_METHOD(Message, jsonDecode) {
  1281. Descriptor* desc =
  1282. UNBOX_HASHTABLE_VALUE(Descriptor, get_ce_obj(Z_OBJCE_P(getThis())));
  1283. MessageHeader* msg = UNBOX(MessageHeader, getThis());
  1284. char *data = NULL;
  1285. int data_len;
  1286. if (zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC, "s", &data, &data_len) ==
  1287. FAILURE) {
  1288. return;
  1289. }
  1290. // TODO(teboring): Check and respect string encoding. If not UTF-8, we need to
  1291. // convert, because string handlers pass data directly to message string
  1292. // fields.
  1293. // TODO(teboring): Clear message.
  1294. {
  1295. const upb_json_parsermethod* method = msgdef_jsonparsermethod(desc);
  1296. stackenv se;
  1297. upb_sink sink;
  1298. upb_json_parser* parser;
  1299. stackenv_init(&se, "Error occurred during parsing: %s");
  1300. upb_sink_reset(&sink, get_fill_handlers(desc), msg);
  1301. parser = upb_json_parser_create(&se.env, method, &sink);
  1302. upb_bufsrc_putbuf(data, data_len, upb_json_parser_input(parser));
  1303. stackenv_uninit(&se);
  1304. }
  1305. }