encode_decode.c 63 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887
  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. //
  31. #include <php.h>
  32. #include <Zend/zend_exceptions.h>
  33. #include "protobuf.h"
  34. #include "utf8.h"
  35. /* stringsink *****************************************************************/
  36. static void *stringsink_start(void *_sink, const void *hd, size_t size_hint) {
  37. stringsink *sink = _sink;
  38. sink->len = 0;
  39. return sink;
  40. }
  41. size_t stringsink_string(void *_sink, const void *hd, const char *ptr,
  42. size_t len, const upb_bufhandle *handle) {
  43. stringsink *sink = _sink;
  44. size_t new_size = sink->size;
  45. PHP_PROTO_UNUSED(hd);
  46. PHP_PROTO_UNUSED(handle);
  47. while (sink->len + len > new_size) {
  48. new_size *= 2;
  49. }
  50. if (new_size != sink->size) {
  51. sink->ptr = realloc(sink->ptr, new_size);
  52. sink->size = new_size;
  53. }
  54. memcpy(sink->ptr + sink->len, ptr, len);
  55. sink->len += len;
  56. return len;
  57. }
  58. void stringsink_init(stringsink *sink) {
  59. upb_byteshandler_init(&sink->handler);
  60. upb_byteshandler_setstartstr(&sink->handler, stringsink_start, NULL);
  61. upb_byteshandler_setstring(&sink->handler, stringsink_string, NULL);
  62. upb_bytessink_reset(&sink->sink, &sink->handler, sink);
  63. sink->size = 32;
  64. sink->ptr = malloc(sink->size);
  65. PHP_PROTO_ASSERT(sink->ptr != NULL);
  66. sink->len = 0;
  67. }
  68. void stringsink_uninit(stringsink *sink) { free(sink->ptr); }
  69. void stringsink_uninit_opaque(void *sink) { stringsink_uninit(sink); }
  70. /* stackenv *****************************************************************/
  71. // Stack-allocated context during an encode/decode operation. Contains the upb
  72. // environment and its stack-based allocator, an initial buffer for allocations
  73. // to avoid malloc() when possible, and a template for PHP exception messages
  74. // if any error occurs.
  75. #define STACK_ENV_STACKBYTES 4096
  76. typedef struct {
  77. upb_arena *arena;
  78. upb_status status;
  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. static void stackenv_init(stackenv* se, const char* errmsg) {
  85. se->php_error_template = errmsg;
  86. se->arena = upb_arena_new();
  87. upb_status_clear(&se->status);
  88. }
  89. static void stackenv_uninit(stackenv* se) {
  90. upb_arena_free(se->arena);
  91. if (!upb_ok(&se->status)) {
  92. // TODO(teboring): have a way to verify that this is actually a parse error,
  93. // instead of just throwing "parse error" unconditionally.
  94. TSRMLS_FETCH();
  95. zend_throw_exception_ex(NULL, 0 TSRMLS_CC, se->php_error_template,
  96. upb_status_errmsg(&se->status));
  97. }
  98. }
  99. // -----------------------------------------------------------------------------
  100. // Parsing.
  101. // -----------------------------------------------------------------------------
  102. // TODO(teboring): This shoud be a bit in upb_msgdef
  103. static bool is_wrapper_msg(const upb_msgdef *msg) {
  104. return !strcmp(upb_filedef_name(upb_msgdef_file(msg)),
  105. "google/protobuf/wrappers.proto");
  106. }
  107. #define DEREF(msg, ofs, type) *(type*)(((uint8_t *)msg) + ofs)
  108. // Creates a handlerdata that simply contains the offset for this field.
  109. static const void* newhandlerdata(upb_handlers* h, uint32_t ofs) {
  110. size_t* hd_ofs = (size_t*)malloc(sizeof(size_t));
  111. PHP_PROTO_ASSERT(hd_ofs != NULL);
  112. *hd_ofs = ofs;
  113. upb_handlers_addcleanup(h, hd_ofs, free);
  114. return hd_ofs;
  115. }
  116. typedef struct {
  117. void* closure;
  118. stringsink sink;
  119. } stringfields_parseframe_t;
  120. typedef size_t (*encodeunknown_handlerfunc)(void* _sink, const void* hd,
  121. const char* ptr, size_t len,
  122. const upb_bufhandle* handle);
  123. typedef struct {
  124. encodeunknown_handlerfunc handler;
  125. } unknownfields_handlerdata_t;
  126. // Creates a handlerdata for unknown fields.
  127. static const void *newunknownfieldshandlerdata(upb_handlers* h) {
  128. unknownfields_handlerdata_t* hd =
  129. (unknownfields_handlerdata_t*)malloc(sizeof(unknownfields_handlerdata_t));
  130. PHP_PROTO_ASSERT(hd != NULL);
  131. hd->handler = stringsink_string;
  132. upb_handlers_addcleanup(h, hd, free);
  133. return hd;
  134. }
  135. typedef struct {
  136. size_t ofs;
  137. const upb_msgdef *md;
  138. } submsg_handlerdata_t;
  139. // Creates a handlerdata that contains offset and submessage type information.
  140. static const void *newsubmsghandlerdata(upb_handlers* h, uint32_t ofs,
  141. const upb_fielddef* f) {
  142. submsg_handlerdata_t* hd =
  143. (submsg_handlerdata_t*)malloc(sizeof(submsg_handlerdata_t));
  144. PHP_PROTO_ASSERT(hd != NULL);
  145. hd->ofs = ofs;
  146. hd->md = upb_fielddef_msgsubdef(f);
  147. upb_handlers_addcleanup(h, hd, free);
  148. return hd;
  149. }
  150. typedef struct {
  151. size_t ofs; // union data slot
  152. size_t case_ofs; // oneof_case field
  153. int property_ofs; // properties table cache
  154. uint32_t oneof_case_num; // oneof-case number to place in oneof_case field
  155. const upb_msgdef *md; // msgdef, for oneof submessage handler
  156. const upb_msgdef *parent_md; // msgdef, for parent submessage
  157. } oneof_handlerdata_t;
  158. static const void *newoneofhandlerdata(upb_handlers *h,
  159. uint32_t ofs,
  160. uint32_t case_ofs,
  161. int property_ofs,
  162. const upb_msgdef *m,
  163. const upb_fielddef *f) {
  164. oneof_handlerdata_t* hd =
  165. (oneof_handlerdata_t*)malloc(sizeof(oneof_handlerdata_t));
  166. PHP_PROTO_ASSERT(hd != NULL);
  167. hd->ofs = ofs;
  168. hd->case_ofs = case_ofs;
  169. hd->property_ofs = property_ofs;
  170. hd->parent_md = m;
  171. // We reuse the field tag number as a oneof union discriminant tag. Note that
  172. // we don't expose these numbers to the user, so the only requirement is that
  173. // we have some unique ID for each union case/possibility. The field tag
  174. // numbers are already present and are easy to use so there's no reason to
  175. // create a separate ID space. In addition, using the field tag number here
  176. // lets us easily look up the field in the oneof accessor.
  177. hd->oneof_case_num = upb_fielddef_number(f);
  178. if (upb_fielddef_type(f) == UPB_TYPE_MESSAGE) {
  179. hd->md = upb_fielddef_msgsubdef(f);
  180. } else {
  181. hd->md = NULL;
  182. }
  183. upb_handlers_addcleanup(h, hd, free);
  184. return hd;
  185. }
  186. // A handler that starts a repeated field. Gets the Repeated*Field instance for
  187. // this field (such an instance always exists even in an empty message).
  188. static void *startseq_handler(void* closure, const void* hd) {
  189. MessageHeader* msg = closure;
  190. const size_t *ofs = hd;
  191. return CACHED_PTR_TO_ZVAL_PTR(DEREF(message_data(msg), *ofs, CACHED_VALUE*));
  192. }
  193. // Handlers that append primitive values to a repeated field.
  194. #define DEFINE_APPEND_HANDLER(type, ctype) \
  195. static bool append##type##_handler(void* closure, const void* hd, \
  196. ctype val) { \
  197. zval* array = (zval*)closure; \
  198. TSRMLS_FETCH(); \
  199. RepeatedField* intern = UNBOX(RepeatedField, array); \
  200. repeated_field_push_native(intern, &val); \
  201. return true; \
  202. }
  203. DEFINE_APPEND_HANDLER(bool, bool)
  204. DEFINE_APPEND_HANDLER(int32, int32_t)
  205. DEFINE_APPEND_HANDLER(uint32, uint32_t)
  206. DEFINE_APPEND_HANDLER(float, float)
  207. DEFINE_APPEND_HANDLER(int64, int64_t)
  208. DEFINE_APPEND_HANDLER(uint64, uint64_t)
  209. DEFINE_APPEND_HANDLER(double, double)
  210. // Appends a string or 'bytes' string to a repeated field.
  211. static void* appendstr_handler(void *closure,
  212. const void *hd,
  213. size_t size_hint) {
  214. PHP_PROTO_UNUSED(hd);
  215. stringfields_parseframe_t* frame =
  216. (stringfields_parseframe_t*)malloc(sizeof(stringfields_parseframe_t));
  217. PHP_PROTO_ASSERT(frame != NULL);
  218. frame->closure = closure;
  219. stringsink_init(&frame->sink);
  220. return frame;
  221. }
  222. static bool appendstr_end_handler(void *closure, const void *hd) {
  223. stringfields_parseframe_t* frame = closure;
  224. zval* array = (zval*)frame->closure;
  225. TSRMLS_FETCH();
  226. RepeatedField* intern = UNBOX(RepeatedField, array);
  227. #if PHP_MAJOR_VERSION < 7
  228. zval* str;
  229. MAKE_STD_ZVAL(str);
  230. PHP_PROTO_ZVAL_STRINGL(str, frame->sink.ptr, frame->sink.len, 1);
  231. repeated_field_push_native(intern, &str);
  232. #else
  233. zend_string* str = zend_string_init(frame->sink.ptr, frame->sink.len, 1);
  234. repeated_field_push_native(intern, &str);
  235. #endif
  236. stringsink_uninit(&frame->sink);
  237. free(frame);
  238. return true;
  239. }
  240. // Handlers that append primitive values to a repeated field.
  241. #define DEFINE_SINGULAR_HANDLER(type, ctype) \
  242. static bool type##_handler(void* closure, const void* hd, \
  243. ctype val) { \
  244. MessageHeader* msg = (MessageHeader*)closure; \
  245. const size_t *ofs = hd; \
  246. DEREF(message_data(msg), *ofs, ctype) = val; \
  247. return true; \
  248. }
  249. DEFINE_SINGULAR_HANDLER(bool, bool)
  250. DEFINE_SINGULAR_HANDLER(int32, int32_t)
  251. DEFINE_SINGULAR_HANDLER(uint32, uint32_t)
  252. DEFINE_SINGULAR_HANDLER(float, float)
  253. DEFINE_SINGULAR_HANDLER(int64, int64_t)
  254. DEFINE_SINGULAR_HANDLER(uint64, uint64_t)
  255. DEFINE_SINGULAR_HANDLER(double, double)
  256. #undef DEFINE_SINGULAR_HANDLER
  257. #if PHP_MAJOR_VERSION < 7
  258. static void *empty_php_string(zval** value_ptr) {
  259. SEPARATE_ZVAL_IF_NOT_REF(value_ptr);
  260. if (Z_TYPE_PP(value_ptr) == IS_STRING &&
  261. !IS_INTERNED(Z_STRVAL_PP(value_ptr))) {
  262. FREE(Z_STRVAL_PP(value_ptr));
  263. }
  264. ZVAL_EMPTY_STRING(*value_ptr);
  265. return (void*)(*value_ptr);
  266. }
  267. #else
  268. static void *empty_php_string(zval* value_ptr) {
  269. if (Z_TYPE_P(value_ptr) == IS_STRING) {
  270. zend_string_release(Z_STR_P(value_ptr));
  271. }
  272. ZVAL_EMPTY_STRING(value_ptr);
  273. return value_ptr;
  274. }
  275. #endif
  276. #if PHP_MAJOR_VERSION < 7
  277. static void *empty_php_string2(zval** value_ptr) {
  278. SEPARATE_ZVAL_IF_NOT_REF(value_ptr);
  279. if (Z_TYPE_PP(value_ptr) == IS_STRING &&
  280. !IS_INTERNED(Z_STRVAL_PP(value_ptr))) {
  281. FREE(Z_STRVAL_PP(value_ptr));
  282. }
  283. ZVAL_EMPTY_STRING(*value_ptr);
  284. return (void*)(*value_ptr);
  285. }
  286. static void new_php_string(zval** value_ptr, const char* str, size_t len) {
  287. SEPARATE_ZVAL_IF_NOT_REF(value_ptr);
  288. if (Z_TYPE_PP(value_ptr) == IS_STRING &&
  289. !IS_INTERNED(Z_STRVAL_PP(value_ptr))) {
  290. FREE(Z_STRVAL_PP(value_ptr));
  291. }
  292. ZVAL_EMPTY_STRING(*value_ptr);
  293. ZVAL_STRINGL(*value_ptr, str, len, 1);
  294. }
  295. #else
  296. static void *empty_php_string2(zval* value_ptr) {
  297. if (Z_TYPE_P(value_ptr) == IS_STRING) {
  298. zend_string_release(Z_STR_P(value_ptr));
  299. }
  300. ZVAL_EMPTY_STRING(value_ptr);
  301. return value_ptr;
  302. }
  303. static void new_php_string(zval* value_ptr, const char* str, size_t len) {
  304. if (Z_TYPE_P(value_ptr) == IS_STRING) {
  305. zend_string_release(Z_STR_P(value_ptr));
  306. }
  307. ZVAL_NEW_STR(value_ptr, zend_string_init(str, len, 0));
  308. }
  309. #endif
  310. // Sets a non-repeated string/bytes field in a message.
  311. static void* str_handler(void *closure,
  312. const void *hd,
  313. size_t size_hint) {
  314. PHP_PROTO_UNUSED(hd);
  315. stringfields_parseframe_t* frame =
  316. (stringfields_parseframe_t*)malloc(sizeof(stringfields_parseframe_t));
  317. PHP_PROTO_ASSERT(frame != NULL);
  318. frame->closure = closure;
  319. stringsink_init(&frame->sink);
  320. return frame;
  321. }
  322. static bool str_end_handler(void *closure, const void *hd) {
  323. stringfields_parseframe_t* frame = closure;
  324. const size_t *ofs = hd;
  325. MessageHeader* msg = (MessageHeader*)frame->closure;
  326. new_php_string(DEREF(message_data(msg), *ofs, CACHED_VALUE*),
  327. frame->sink.ptr, frame->sink.len);
  328. stringsink_uninit(&frame->sink);
  329. free(frame);
  330. return true;
  331. }
  332. static size_t stringdata_handler(void* closure, const void* hd,
  333. const char* str, size_t len,
  334. const upb_bufhandle* handle) {
  335. stringfields_parseframe_t* frame = closure;
  336. return stringsink_string(&frame->sink, hd, str, len, handle);
  337. }
  338. // Appends a submessage to a repeated field.
  339. static void *appendsubmsg_handler(void *closure, const void *hd) {
  340. zval* array = (zval*)closure;
  341. TSRMLS_FETCH();
  342. RepeatedField* intern = UNBOX(RepeatedField, array);
  343. const submsg_handlerdata_t *submsgdata = hd;
  344. Descriptor* subdesc =
  345. UNBOX_HASHTABLE_VALUE(Descriptor, get_def_obj((void*)submsgdata->md));
  346. zend_class_entry* subklass = subdesc->klass;
  347. MessageHeader* submsg;
  348. #if PHP_MAJOR_VERSION < 7
  349. zval* val = NULL;
  350. MAKE_STD_ZVAL(val);
  351. ZVAL_OBJ(val, subklass->create_object(subklass TSRMLS_CC));
  352. repeated_field_push_native(intern, &val);
  353. submsg = UNBOX(MessageHeader, val);
  354. #else
  355. zend_object* obj = subklass->create_object(subklass TSRMLS_CC);
  356. repeated_field_push_native(intern, &obj);
  357. submsg = (MessageHeader*)((char*)obj - XtOffsetOf(MessageHeader, std));
  358. #endif
  359. custom_data_init(subklass, submsg PHP_PROTO_TSRMLS_CC);
  360. return submsg;
  361. }
  362. // Sets a non-repeated submessage field in a message.
  363. static void *submsg_handler(void *closure, const void *hd) {
  364. MessageHeader* msg = closure;
  365. const submsg_handlerdata_t* submsgdata = hd;
  366. TSRMLS_FETCH();
  367. Descriptor* subdesc =
  368. UNBOX_HASHTABLE_VALUE(Descriptor, get_def_obj((void*)submsgdata->md));
  369. zend_class_entry* subklass = subdesc->klass;
  370. zval* submsg_php;
  371. MessageHeader* submsg;
  372. if (Z_TYPE_P(CACHED_PTR_TO_ZVAL_PTR(DEREF(message_data(msg), submsgdata->ofs,
  373. CACHED_VALUE*))) == IS_NULL) {
  374. #if PHP_MAJOR_VERSION < 7
  375. zval val;
  376. ZVAL_OBJ(&val, subklass->create_object(subklass TSRMLS_CC));
  377. MessageHeader* intern = UNBOX(MessageHeader, &val);
  378. custom_data_init(subklass, intern PHP_PROTO_TSRMLS_CC);
  379. REPLACE_ZVAL_VALUE(DEREF(message_data(msg), submsgdata->ofs, zval**),
  380. &val, 1);
  381. zval_dtor(&val);
  382. #else
  383. zend_object* obj = subklass->create_object(subklass TSRMLS_CC);
  384. ZVAL_OBJ(DEREF(message_data(msg), submsgdata->ofs, zval*), obj);
  385. MessageHeader* intern = UNBOX_HASHTABLE_VALUE(MessageHeader, obj);
  386. custom_data_init(subklass, intern PHP_PROTO_TSRMLS_CC);
  387. #endif
  388. }
  389. submsg_php = CACHED_PTR_TO_ZVAL_PTR(
  390. DEREF(message_data(msg), submsgdata->ofs, CACHED_VALUE*));
  391. submsg = UNBOX(MessageHeader, submsg_php);
  392. return submsg;
  393. }
  394. // Handler data for startmap/endmap handlers.
  395. typedef struct {
  396. size_t ofs;
  397. upb_fieldtype_t key_field_type;
  398. upb_fieldtype_t value_field_type;
  399. } map_handlerdata_t;
  400. // Temporary frame for map parsing: at the beginning of a map entry message, a
  401. // submsg handler allocates a frame to hold (i) a reference to the Map object
  402. // into which this message will be inserted and (ii) storage slots to
  403. // temporarily hold the key and value for this map entry until the end of the
  404. // submessage. When the submessage ends, another handler is called to insert the
  405. // value into the map.
  406. typedef struct {
  407. char key_storage[NATIVE_SLOT_MAX_SIZE];
  408. char value_storage[NATIVE_SLOT_MAX_SIZE];
  409. } map_parse_frame_data_t;
  410. PHP_PROTO_WRAP_OBJECT_START(map_parse_frame_t)
  411. map_parse_frame_data_t* data; // Place needs to be consistent with
  412. // MessageHeader.
  413. zval* map;
  414. // In php7, we cannot allocate zval dynamically. So we need to add zval here
  415. // to help decoding.
  416. zval key_zval;
  417. zval value_zval;
  418. PHP_PROTO_WRAP_OBJECT_END
  419. typedef struct map_parse_frame_t map_parse_frame_t;
  420. static void map_slot_init(void* memory, upb_fieldtype_t type, zval* cache) {
  421. switch (type) {
  422. case UPB_TYPE_STRING:
  423. case UPB_TYPE_BYTES: {
  424. #if PHP_MAJOR_VERSION < 7
  425. // Store zval** in memory in order to be consistent with the layout of
  426. // singular fields.
  427. zval** holder = ALLOC(zval*);
  428. *(zval***)memory = holder;
  429. zval* tmp;
  430. MAKE_STD_ZVAL(tmp);
  431. PHP_PROTO_ZVAL_STRINGL(tmp, "", 0, 1);
  432. *holder = tmp;
  433. #else
  434. *(zval**)memory = cache;
  435. PHP_PROTO_ZVAL_STRINGL(*(zval**)memory, "", 0, 1);
  436. #endif
  437. break;
  438. }
  439. case UPB_TYPE_MESSAGE: {
  440. #if PHP_MAJOR_VERSION < 7
  441. zval** holder = ALLOC(zval*);
  442. zval* tmp;
  443. MAKE_STD_ZVAL(tmp);
  444. ZVAL_NULL(tmp);
  445. *holder = tmp;
  446. *(zval***)memory = holder;
  447. #else
  448. *(zval**)memory = cache;
  449. ZVAL_NULL(*(zval**)memory);
  450. #endif
  451. break;
  452. }
  453. default:
  454. native_slot_init(type, memory, NULL);
  455. }
  456. }
  457. static void map_slot_uninit(void* memory, upb_fieldtype_t type) {
  458. switch (type) {
  459. case UPB_TYPE_MESSAGE:
  460. case UPB_TYPE_STRING:
  461. case UPB_TYPE_BYTES: {
  462. #if PHP_MAJOR_VERSION < 7
  463. zval** holder = *(zval***)memory;
  464. zval_ptr_dtor(holder);
  465. FREE(holder);
  466. #else
  467. php_proto_zval_ptr_dtor(*(zval**)memory);
  468. #endif
  469. break;
  470. }
  471. default:
  472. break;
  473. }
  474. }
  475. static void map_slot_key(upb_fieldtype_t type, const void* from,
  476. const char** keyval,
  477. size_t* length) {
  478. if (type == UPB_TYPE_STRING) {
  479. #if PHP_MAJOR_VERSION < 7
  480. zval* key_php = **(zval***)from;
  481. #else
  482. zval* key_php = *(zval**)from;
  483. #endif
  484. *keyval = Z_STRVAL_P(key_php);
  485. *length = Z_STRLEN_P(key_php);
  486. } else {
  487. *keyval = from;
  488. *length = native_slot_size(type);
  489. }
  490. }
  491. static void map_slot_value(upb_fieldtype_t type, const void* from,
  492. upb_value* v) {
  493. size_t len;
  494. void* to = upb_value_memory(v);
  495. #ifndef NDEBUG
  496. v->ctype = UPB_CTYPE_UINT64;
  497. #endif
  498. memset(to, 0, native_slot_size(type));
  499. switch (type) {
  500. #if PHP_MAJOR_VERSION < 7
  501. case UPB_TYPE_STRING:
  502. case UPB_TYPE_BYTES:
  503. case UPB_TYPE_MESSAGE: {
  504. *(zval**)to = **(zval***)from;
  505. Z_ADDREF_PP((zval**)to);
  506. break;
  507. }
  508. #else
  509. case UPB_TYPE_STRING:
  510. case UPB_TYPE_BYTES:
  511. *(zend_string**)to = Z_STR_P(*(zval**)from);
  512. zend_string_addref(*(zend_string**)to);
  513. break;
  514. case UPB_TYPE_MESSAGE:
  515. *(zend_object**)to = Z_OBJ_P(*(zval**)from);
  516. GC_ADDREF(*(zend_object**)to);
  517. break;
  518. #endif
  519. default:
  520. len = native_slot_size(type);
  521. memcpy(to, from, len);
  522. }
  523. }
  524. // Handler to begin a map entry: allocates a temporary frame. This is the
  525. // 'startsubmsg' handler on the msgdef that contains the map field.
  526. static void *startmapentry_handler(void *closure, const void *hd) {
  527. MessageHeader* msg = closure;
  528. const map_handlerdata_t* mapdata = hd;
  529. zval* map = CACHED_PTR_TO_ZVAL_PTR(
  530. DEREF(message_data(msg), mapdata->ofs, CACHED_VALUE*));
  531. map_parse_frame_t* frame = ALLOC(map_parse_frame_t);
  532. frame->data = ALLOC(map_parse_frame_data_t);
  533. frame->map = map;
  534. map_slot_init(&frame->data->key_storage, mapdata->key_field_type,
  535. &frame->key_zval);
  536. map_slot_init(&frame->data->value_storage, mapdata->value_field_type,
  537. &frame->value_zval);
  538. return frame;
  539. }
  540. // Handler to end a map entry: inserts the value defined during the message into
  541. // the map. This is the 'endmsg' handler on the map entry msgdef.
  542. static bool endmap_handler(void* closure, const void* hd, upb_status* s) {
  543. map_parse_frame_t* frame = closure;
  544. const map_handlerdata_t* mapdata = hd;
  545. TSRMLS_FETCH();
  546. Map *map = UNBOX(Map, frame->map);
  547. const char* keyval = NULL;
  548. upb_value v;
  549. size_t length;
  550. map_slot_key(map->key_type, &frame->data->key_storage, &keyval, &length);
  551. map_slot_value(map->value_type, &frame->data->value_storage, &v);
  552. map_index_set(map, keyval, length, v);
  553. map_slot_uninit(&frame->data->key_storage, mapdata->key_field_type);
  554. map_slot_uninit(&frame->data->value_storage, mapdata->value_field_type);
  555. FREE(frame->data);
  556. FREE(frame);
  557. return true;
  558. }
  559. // Allocates a new map_handlerdata_t given the map entry message definition. If
  560. // the offset of the field within the parent message is also given, that is
  561. // added to the handler data as well. Note that this is called *twice* per map
  562. // field: once in the parent message handler setup when setting the startsubmsg
  563. // handler and once in the map entry message handler setup when setting the
  564. // key/value and endmsg handlers. The reason is that there is no easy way to
  565. // pass the handlerdata down to the sub-message handler setup.
  566. static map_handlerdata_t* new_map_handlerdata(
  567. size_t ofs,
  568. const upb_msgdef* mapentry_def,
  569. Descriptor* desc) {
  570. const upb_fielddef* key_field;
  571. const upb_fielddef* value_field;
  572. // TODO(teboring): Use emalloc and efree.
  573. map_handlerdata_t* hd =
  574. (map_handlerdata_t*)malloc(sizeof(map_handlerdata_t));
  575. PHP_PROTO_ASSERT(hd != NULL);
  576. hd->ofs = ofs;
  577. key_field = upb_msgdef_itof(mapentry_def, MAP_KEY_FIELD);
  578. PHP_PROTO_ASSERT(key_field != NULL);
  579. hd->key_field_type = upb_fielddef_type(key_field);
  580. value_field = upb_msgdef_itof(mapentry_def, MAP_VALUE_FIELD);
  581. PHP_PROTO_ASSERT(value_field != NULL);
  582. hd->value_field_type = upb_fielddef_type(value_field);
  583. return hd;
  584. }
  585. // Handlers that set primitive values in oneofs.
  586. #define DEFINE_ONEOF_HANDLER(type, ctype) \
  587. static bool oneof##type##_handler(void* closure, const void* hd, \
  588. ctype val) { \
  589. const oneof_handlerdata_t* oneofdata = hd; \
  590. MessageHeader* msg = (MessageHeader*)closure; \
  591. DEREF(message_data(closure), oneofdata->case_ofs, uint32_t) = \
  592. oneofdata->oneof_case_num; \
  593. DEREF(message_data(closure), oneofdata->ofs, ctype) = val; \
  594. return true; \
  595. }
  596. DEFINE_ONEOF_HANDLER(bool, bool)
  597. DEFINE_ONEOF_HANDLER(int32, int32_t)
  598. DEFINE_ONEOF_HANDLER(uint32, uint32_t)
  599. DEFINE_ONEOF_HANDLER(float, float)
  600. DEFINE_ONEOF_HANDLER(int64, int64_t)
  601. DEFINE_ONEOF_HANDLER(uint64, uint64_t)
  602. DEFINE_ONEOF_HANDLER(double, double)
  603. #undef DEFINE_ONEOF_HANDLER
  604. static void oneof_cleanup(MessageHeader* msg,
  605. const oneof_handlerdata_t* oneofdata) {
  606. uint32_t old_case_num =
  607. DEREF(message_data(msg), oneofdata->case_ofs, uint32_t);
  608. if (old_case_num == 0) {
  609. return;
  610. }
  611. const upb_fielddef* old_field =
  612. upb_msgdef_itof(oneofdata->parent_md, old_case_num);
  613. bool need_clean = false;
  614. switch (upb_fielddef_type(old_field)) {
  615. case UPB_TYPE_STRING:
  616. case UPB_TYPE_BYTES:
  617. need_clean = true;
  618. break;
  619. case UPB_TYPE_MESSAGE:
  620. if (oneofdata->oneof_case_num != old_case_num) {
  621. need_clean = true;
  622. }
  623. break;
  624. default:
  625. break;
  626. }
  627. if (need_clean) {
  628. #if PHP_MAJOR_VERSION < 7
  629. SEPARATE_ZVAL_IF_NOT_REF(
  630. DEREF(message_data(msg), oneofdata->ofs, CACHED_VALUE*));
  631. php_proto_zval_ptr_dtor(
  632. *DEREF(message_data(msg), oneofdata->ofs, CACHED_VALUE*));
  633. MAKE_STD_ZVAL(*DEREF(message_data(msg), oneofdata->ofs, CACHED_VALUE*));
  634. ZVAL_NULL(*DEREF(message_data(msg), oneofdata->ofs, CACHED_VALUE*));
  635. #endif
  636. }
  637. }
  638. // Handlers for string/bytes in a oneof.
  639. static void *oneofbytes_handler(void *closure,
  640. const void *hd,
  641. size_t size_hint) {
  642. MessageHeader* msg = closure;
  643. const oneof_handlerdata_t *oneofdata = hd;
  644. oneof_cleanup(msg, oneofdata);
  645. DEREF(message_data(msg), oneofdata->case_ofs, uint32_t) =
  646. oneofdata->oneof_case_num;
  647. DEREF(message_data(msg), oneofdata->ofs, CACHED_VALUE*) =
  648. OBJ_PROP(&msg->std, oneofdata->property_ofs);
  649. return empty_php_string(DEREF(
  650. message_data(msg), oneofdata->ofs, CACHED_VALUE*));
  651. }
  652. static bool oneofstr_end_handler(void *closure, const void *hd) {
  653. stringfields_parseframe_t* frame = closure;
  654. MessageHeader* msg = (MessageHeader*)frame->closure;
  655. const oneof_handlerdata_t *oneofdata = hd;
  656. oneof_cleanup(msg, oneofdata);
  657. DEREF(message_data(msg), oneofdata->case_ofs, uint32_t) =
  658. oneofdata->oneof_case_num;
  659. DEREF(message_data(msg), oneofdata->ofs, CACHED_VALUE*) =
  660. OBJ_PROP(&msg->std, oneofdata->property_ofs);
  661. new_php_string(DEREF(message_data(msg), oneofdata->ofs, CACHED_VALUE*),
  662. frame->sink.ptr, frame->sink.len);
  663. stringsink_uninit(&frame->sink);
  664. free(frame);
  665. return true;
  666. }
  667. static void *oneofstr_handler(void *closure,
  668. const void *hd,
  669. size_t size_hint) {
  670. PHP_PROTO_UNUSED(hd);
  671. stringfields_parseframe_t* frame =
  672. (stringfields_parseframe_t*)malloc(sizeof(stringfields_parseframe_t));
  673. PHP_PROTO_ASSERT(frame != NULL);
  674. frame->closure = closure;
  675. stringsink_init(&frame->sink);
  676. return frame;
  677. }
  678. // Handler for a submessage field in a oneof.
  679. static void* oneofsubmsg_handler(void* closure, const void* hd) {
  680. MessageHeader* msg = closure;
  681. const oneof_handlerdata_t *oneofdata = hd;
  682. uint32_t oldcase = DEREF(message_data(msg), oneofdata->case_ofs, uint32_t);
  683. TSRMLS_FETCH();
  684. Descriptor* subdesc =
  685. UNBOX_HASHTABLE_VALUE(Descriptor, get_def_obj((void*)oneofdata->md));
  686. zend_class_entry* subklass = subdesc->klass;
  687. zval* submsg_php;
  688. MessageHeader* submsg;
  689. if (oldcase != oneofdata->oneof_case_num) {
  690. oneof_cleanup(msg, oneofdata);
  691. // Create new message.
  692. DEREF(message_data(msg), oneofdata->ofs, CACHED_VALUE*) =
  693. OBJ_PROP(&msg->std, oneofdata->property_ofs);
  694. #if PHP_MAJOR_VERSION < 7
  695. zval val;
  696. ZVAL_OBJ(&val, subklass->create_object(subklass TSRMLS_CC));
  697. REPLACE_ZVAL_VALUE(DEREF(message_data(msg), oneofdata->ofs, zval**),
  698. &val, 1);
  699. zval_dtor(&val);
  700. #else
  701. zend_object* obj = subklass->create_object(subklass TSRMLS_CC);
  702. ZVAL_OBJ(DEREF(message_data(msg), oneofdata->ofs, zval*), obj);
  703. #endif
  704. }
  705. DEREF(message_data(msg), oneofdata->case_ofs, uint32_t) =
  706. oneofdata->oneof_case_num;
  707. submsg_php = CACHED_PTR_TO_ZVAL_PTR(
  708. DEREF(message_data(msg), oneofdata->ofs, CACHED_VALUE*));
  709. submsg = UNBOX(MessageHeader, submsg_php);
  710. custom_data_init(subklass, submsg PHP_PROTO_TSRMLS_CC);
  711. return submsg;
  712. }
  713. // Set up handlers for a repeated field.
  714. static void add_handlers_for_repeated_field(upb_handlers *h,
  715. const upb_fielddef *f,
  716. size_t offset) {
  717. upb_handlerattr attr = UPB_HANDLERATTR_INIT;
  718. attr.handler_data = newhandlerdata(h, offset);
  719. upb_handlers_setstartseq(h, f, startseq_handler, &attr);
  720. switch (upb_fielddef_type(f)) {
  721. #define SET_HANDLER(utype, ltype) \
  722. case utype: \
  723. upb_handlers_set##ltype(h, f, append##ltype##_handler, NULL); \
  724. break;
  725. SET_HANDLER(UPB_TYPE_BOOL, bool);
  726. SET_HANDLER(UPB_TYPE_INT32, int32);
  727. SET_HANDLER(UPB_TYPE_UINT32, uint32);
  728. SET_HANDLER(UPB_TYPE_ENUM, int32);
  729. SET_HANDLER(UPB_TYPE_FLOAT, float);
  730. SET_HANDLER(UPB_TYPE_INT64, int64);
  731. SET_HANDLER(UPB_TYPE_UINT64, uint64);
  732. SET_HANDLER(UPB_TYPE_DOUBLE, double);
  733. #undef SET_HANDLER
  734. case UPB_TYPE_STRING:
  735. case UPB_TYPE_BYTES: {
  736. upb_handlers_setstartstr(h, f, appendstr_handler, NULL);
  737. upb_handlers_setstring(h, f, stringdata_handler, NULL);
  738. upb_handlers_setendstr(h, f, appendstr_end_handler, &attr);
  739. break;
  740. }
  741. case UPB_TYPE_MESSAGE: {
  742. upb_handlerattr attr = UPB_HANDLERATTR_INIT;
  743. attr.handler_data = newsubmsghandlerdata(h, 0, f);
  744. upb_handlers_setstartsubmsg(h, f, appendsubmsg_handler, &attr);
  745. break;
  746. }
  747. }
  748. }
  749. // Set up handlers for a singular field.
  750. static void add_handlers_for_singular_field(upb_handlers *h,
  751. const upb_fielddef *f,
  752. size_t offset) {
  753. switch (upb_fielddef_type(f)) {
  754. #define SET_HANDLER(utype, ltype) \
  755. case utype: { \
  756. upb_handlerattr attr = UPB_HANDLERATTR_INIT; \
  757. attr.handler_data = newhandlerdata(h, offset); \
  758. upb_handlers_set##ltype(h, f, ltype##_handler, &attr); \
  759. break; \
  760. }
  761. SET_HANDLER(UPB_TYPE_BOOL, bool);
  762. SET_HANDLER(UPB_TYPE_INT32, int32);
  763. SET_HANDLER(UPB_TYPE_UINT32, uint32);
  764. SET_HANDLER(UPB_TYPE_ENUM, int32);
  765. SET_HANDLER(UPB_TYPE_FLOAT, float);
  766. SET_HANDLER(UPB_TYPE_INT64, int64);
  767. SET_HANDLER(UPB_TYPE_UINT64, uint64);
  768. SET_HANDLER(UPB_TYPE_DOUBLE, double);
  769. #undef SET_HANDLER
  770. case UPB_TYPE_STRING:
  771. case UPB_TYPE_BYTES: {
  772. upb_handlerattr attr = UPB_HANDLERATTR_INIT;
  773. attr.handler_data = newhandlerdata(h, offset);
  774. upb_handlers_setstartstr(h, f, str_handler, &attr);
  775. upb_handlers_setstring(h, f, stringdata_handler, &attr);
  776. upb_handlers_setendstr(h, f, str_end_handler, &attr);
  777. break;
  778. }
  779. case UPB_TYPE_MESSAGE: {
  780. upb_handlerattr attr = UPB_HANDLERATTR_INIT;
  781. attr.handler_data = newsubmsghandlerdata(h, offset, f);
  782. upb_handlers_setstartsubmsg(h, f, submsg_handler, &attr);
  783. break;
  784. }
  785. }
  786. }
  787. // Adds handlers to a map field.
  788. static void add_handlers_for_mapfield(upb_handlers* h,
  789. const upb_fielddef* fielddef,
  790. size_t offset,
  791. Descriptor* desc) {
  792. const upb_msgdef* map_msgdef = upb_fielddef_msgsubdef(fielddef);
  793. map_handlerdata_t* hd = new_map_handlerdata(offset, map_msgdef, desc);
  794. upb_handlerattr attr = UPB_HANDLERATTR_INIT;
  795. upb_handlers_addcleanup(h, hd, free);
  796. attr.handler_data = hd;
  797. upb_handlers_setstartsubmsg(h, fielddef, startmapentry_handler, &attr);
  798. }
  799. // Adds handlers to a map-entry msgdef.
  800. static void add_handlers_for_mapentry(const upb_msgdef* msgdef, upb_handlers* h,
  801. Descriptor* desc) {
  802. const upb_fielddef* key_field = map_entry_key(msgdef);
  803. const upb_fielddef* value_field = map_entry_value(msgdef);
  804. map_handlerdata_t* hd = new_map_handlerdata(0, msgdef, desc);
  805. upb_handlerattr attr = UPB_HANDLERATTR_INIT;
  806. upb_handlers_addcleanup(h, hd, free);
  807. attr.handler_data = hd;
  808. upb_handlers_setendmsg(h, endmap_handler, &attr);
  809. add_handlers_for_singular_field(h, key_field,
  810. offsetof(map_parse_frame_data_t,
  811. key_storage));
  812. add_handlers_for_singular_field(h, value_field,
  813. offsetof(map_parse_frame_data_t,
  814. value_storage));
  815. }
  816. // Set up handlers for a oneof field.
  817. static void add_handlers_for_oneof_field(upb_handlers *h,
  818. const upb_msgdef *m,
  819. const upb_fielddef *f,
  820. size_t offset,
  821. size_t oneof_case_offset,
  822. int property_cache_offset) {
  823. upb_handlerattr attr = UPB_HANDLERATTR_INIT;
  824. attr.handler_data = newoneofhandlerdata(h, offset, oneof_case_offset,
  825. property_cache_offset, m, f);
  826. switch (upb_fielddef_type(f)) {
  827. #define SET_HANDLER(utype, ltype) \
  828. case utype: \
  829. upb_handlers_set##ltype(h, f, oneof##ltype##_handler, &attr); \
  830. break;
  831. SET_HANDLER(UPB_TYPE_BOOL, bool);
  832. SET_HANDLER(UPB_TYPE_INT32, int32);
  833. SET_HANDLER(UPB_TYPE_UINT32, uint32);
  834. SET_HANDLER(UPB_TYPE_ENUM, int32);
  835. SET_HANDLER(UPB_TYPE_FLOAT, float);
  836. SET_HANDLER(UPB_TYPE_INT64, int64);
  837. SET_HANDLER(UPB_TYPE_UINT64, uint64);
  838. SET_HANDLER(UPB_TYPE_DOUBLE, double);
  839. #undef SET_HANDLER
  840. case UPB_TYPE_STRING:
  841. case UPB_TYPE_BYTES: {
  842. upb_handlers_setstartstr(h, f, oneofstr_handler, &attr);
  843. upb_handlers_setstring(h, f, stringdata_handler, NULL);
  844. upb_handlers_setendstr(h, f, oneofstr_end_handler, &attr);
  845. break;
  846. }
  847. case UPB_TYPE_MESSAGE: {
  848. upb_handlers_setstartsubmsg(h, f, oneofsubmsg_handler, &attr);
  849. break;
  850. }
  851. }
  852. }
  853. static bool add_unknown_handler(void* closure, const void* hd, const char* buf,
  854. size_t size) {
  855. encodeunknown_handlerfunc handler =
  856. ((unknownfields_handlerdata_t*)hd)->handler;
  857. MessageHeader* msg = (MessageHeader*)closure;
  858. stringsink* unknown = DEREF(message_data(msg), 0, stringsink*);
  859. if (unknown == NULL) {
  860. DEREF(message_data(msg), 0, stringsink*) = ALLOC(stringsink);
  861. unknown = DEREF(message_data(msg), 0, stringsink*);
  862. stringsink_init(unknown);
  863. }
  864. handler(unknown, NULL, buf, size, NULL);
  865. return true;
  866. }
  867. void add_handlers_for_message(const void* closure, upb_handlers* h) {
  868. const upb_msgdef* msgdef = upb_handlers_msgdef(h);
  869. TSRMLS_FETCH();
  870. Descriptor* desc =
  871. UNBOX_HASHTABLE_VALUE(Descriptor, get_def_obj((void*)msgdef));
  872. upb_msg_field_iter i;
  873. // If this is a mapentry message type, set up a special set of handlers and
  874. // bail out of the normal (user-defined) message type handling.
  875. if (upb_msgdef_mapentry(msgdef)) {
  876. add_handlers_for_mapentry(msgdef, h, desc);
  877. return;
  878. }
  879. // Ensure layout exists. We may be invoked to create handlers for a given
  880. // message if we are included as a submsg of another message type before our
  881. // class is actually built, so to work around this, we just create the layout
  882. // (and handlers, in the class-building function) on-demand.
  883. if (desc->layout == NULL) {
  884. desc->layout = create_layout(desc->msgdef);
  885. }
  886. upb_handlerattr attr = UPB_HANDLERATTR_INIT;
  887. attr.handler_data = newunknownfieldshandlerdata(h);
  888. upb_handlers_setunknown(h, add_unknown_handler, &attr);
  889. for (upb_msg_field_begin(&i, desc->msgdef);
  890. !upb_msg_field_done(&i);
  891. upb_msg_field_next(&i)) {
  892. const upb_fielddef *f = upb_msg_iter_field(&i);
  893. size_t offset = desc->layout->fields[upb_fielddef_index(f)].offset;
  894. if (upb_fielddef_containingoneof(f)) {
  895. size_t oneof_case_offset =
  896. desc->layout->fields[upb_fielddef_index(f)].case_offset;
  897. int property_cache_index =
  898. desc->layout->fields[upb_fielddef_index(f)].cache_index;
  899. add_handlers_for_oneof_field(h, desc->msgdef, f, offset,
  900. oneof_case_offset, property_cache_index);
  901. } else if (is_map_field(f)) {
  902. add_handlers_for_mapfield(h, f, offset, desc);
  903. } else if (upb_fielddef_isseq(f)) {
  904. add_handlers_for_repeated_field(h, f, offset);
  905. } else {
  906. add_handlers_for_singular_field(h, f, offset);
  907. }
  908. }
  909. }
  910. // Constructs the handlers for filling a message's data into an in-memory
  911. // object.
  912. const upb_handlers* get_fill_handlers(Descriptor* desc) {
  913. return upb_handlercache_get(desc->pool->fill_handler_cache, desc->msgdef);
  914. }
  915. static const upb_pbdecodermethod *msgdef_decodermethod(Descriptor* desc) {
  916. return upb_pbcodecache_get(desc->pool->fill_method_cache, desc->msgdef);
  917. }
  918. static const upb_json_parsermethod *msgdef_jsonparsermethod(Descriptor* desc) {
  919. return upb_json_codecache_get(desc->pool->json_fill_method_cache, desc->msgdef);
  920. }
  921. // -----------------------------------------------------------------------------
  922. // Serializing.
  923. // -----------------------------------------------------------------------------
  924. static void putmsg(zval* msg, const Descriptor* desc, upb_sink sink,
  925. int depth, bool is_json TSRMLS_DC);
  926. static void putrawmsg(MessageHeader* msg, const Descriptor* desc,
  927. upb_sink sink, int depth, bool is_json,
  928. bool open_msg TSRMLS_DC);
  929. static void putjsonany(MessageHeader* msg, const Descriptor* desc,
  930. upb_sink sink, int depth TSRMLS_DC);
  931. static void putjsonlistvalue(
  932. MessageHeader* msg, const Descriptor* desc,
  933. upb_sink sink, int depth TSRMLS_DC);
  934. static void putjsonstruct(
  935. MessageHeader* msg, const Descriptor* desc,
  936. upb_sink sink, int depth TSRMLS_DC);
  937. static void putstr(zval* str, const upb_fielddef* f, upb_sink sink,
  938. bool force_default);
  939. static void putrawstr(const char* str, int len, const upb_fielddef* f,
  940. upb_sink sink, bool force_default);
  941. static void putsubmsg(zval* submsg, const upb_fielddef* f, upb_sink sink,
  942. int depth, bool is_json TSRMLS_DC);
  943. static void putrawsubmsg(MessageHeader* submsg, const upb_fielddef* f,
  944. upb_sink sink, int depth, bool is_json TSRMLS_DC);
  945. static void putarray(zval* array, const upb_fielddef* f, upb_sink sink,
  946. int depth, bool is_json TSRMLS_DC);
  947. static void putmap(zval* map, const upb_fielddef* f, upb_sink sink,
  948. int depth, bool is_json TSRMLS_DC);
  949. static upb_selector_t getsel(const upb_fielddef* f, upb_handlertype_t type) {
  950. upb_selector_t ret;
  951. bool ok = upb_handlers_getselector(f, type, &ret);
  952. PHP_PROTO_ASSERT(ok);
  953. return ret;
  954. }
  955. static void put_optional_value(const void* memory, int len,
  956. const upb_fielddef* f,
  957. int depth, upb_sink sink,
  958. bool is_json TSRMLS_DC) {
  959. PHP_PROTO_ASSERT(upb_fielddef_label(f) == UPB_LABEL_OPTIONAL);
  960. switch (upb_fielddef_type(f)) {
  961. #define T(upbtypeconst, upbtype, ctype, default_value) \
  962. case upbtypeconst: { \
  963. ctype value = DEREF(memory, 0, ctype); \
  964. if (is_json || value != default_value) { \
  965. upb_selector_t sel = getsel(f, upb_handlers_getprimitivehandlertype(f)); \
  966. upb_sink_put##upbtype(sink, sel, value); \
  967. } \
  968. } break;
  969. T(UPB_TYPE_FLOAT, float, float, 0.0)
  970. T(UPB_TYPE_DOUBLE, double, double, 0.0)
  971. T(UPB_TYPE_BOOL, bool, uint8_t, 0)
  972. T(UPB_TYPE_ENUM, int32, int32_t, 0)
  973. T(UPB_TYPE_INT32, int32, int32_t, 0)
  974. T(UPB_TYPE_UINT32, uint32, uint32_t, 0)
  975. T(UPB_TYPE_INT64, int64, int64_t, 0)
  976. T(UPB_TYPE_UINT64, uint64, uint64_t, 0)
  977. #undef T
  978. case UPB_TYPE_STRING:
  979. case UPB_TYPE_BYTES:
  980. putrawstr(memory, len, f, sink, is_json);
  981. break;
  982. case UPB_TYPE_MESSAGE: {
  983. #if PHP_MAJOR_VERSION < 7
  984. MessageHeader *submsg = UNBOX(MessageHeader, *(zval**)memory);
  985. #else
  986. MessageHeader *submsg =
  987. (MessageHeader*)((char*)(*(zend_object**)memory) -
  988. XtOffsetOf(MessageHeader, std));
  989. #endif
  990. putrawsubmsg(submsg, f, sink, depth, is_json TSRMLS_CC);
  991. break;
  992. }
  993. default:
  994. PHP_PROTO_ASSERT(false);
  995. }
  996. }
  997. // Only string/bytes fields are stored as zval.
  998. static const char* raw_value(void* memory, const upb_fielddef* f) {
  999. switch (upb_fielddef_type(f)) {
  1000. case UPB_TYPE_STRING:
  1001. case UPB_TYPE_BYTES:
  1002. #if PHP_MAJOR_VERSION < 7
  1003. return Z_STRVAL_PP((zval**)memory);
  1004. #else
  1005. return ZSTR_VAL(*(zend_string**)memory);
  1006. #endif
  1007. break;
  1008. default:
  1009. return memory;
  1010. }
  1011. }
  1012. static int raw_value_len(void* memory, int len, const upb_fielddef* f) {
  1013. switch (upb_fielddef_type(f)) {
  1014. case UPB_TYPE_STRING:
  1015. case UPB_TYPE_BYTES:
  1016. #if PHP_MAJOR_VERSION < 7
  1017. return Z_STRLEN_PP((zval**)memory);
  1018. #else
  1019. return ZSTR_LEN(*(zend_string**)memory);
  1020. #endif
  1021. default:
  1022. return len;
  1023. }
  1024. }
  1025. static void putmap(zval* map, const upb_fielddef* f, upb_sink sink,
  1026. int depth, bool is_json TSRMLS_DC) {
  1027. upb_sink subsink;
  1028. const upb_fielddef* key_field;
  1029. const upb_fielddef* value_field;
  1030. MapIter it;
  1031. int len, size;
  1032. PHP_PROTO_ASSERT(map != NULL);
  1033. Map* intern = UNBOX(Map, map);
  1034. size = upb_strtable_count(&intern->table);
  1035. if (size == 0) return;
  1036. upb_sink_startseq(sink, getsel(f, UPB_HANDLER_STARTSEQ), &subsink);
  1037. PHP_PROTO_ASSERT(upb_fielddef_type(f) == UPB_TYPE_MESSAGE);
  1038. key_field = map_field_key(f);
  1039. value_field = map_field_value(f);
  1040. for (map_begin(map, &it TSRMLS_CC); !map_done(&it); map_next(&it)) {
  1041. upb_status status;
  1042. upb_sink entry_sink;
  1043. upb_sink_startsubmsg(subsink, getsel(f, UPB_HANDLER_STARTSUBMSG),
  1044. &entry_sink);
  1045. upb_sink_startmsg(entry_sink);
  1046. // Serialize key.
  1047. const char *key = map_iter_key(&it, &len);
  1048. put_optional_value(key, len, key_field, depth + 1,
  1049. entry_sink, is_json TSRMLS_CC);
  1050. // Serialize value.
  1051. upb_value value = map_iter_value(&it, &len);
  1052. put_optional_value(raw_value(upb_value_memory(&value), value_field),
  1053. raw_value_len(upb_value_memory(&value), len, value_field),
  1054. value_field, depth + 1, entry_sink, is_json TSRMLS_CC);
  1055. upb_sink_endmsg(entry_sink, &status);
  1056. upb_sink_endsubmsg(subsink, getsel(f, UPB_HANDLER_ENDSUBMSG));
  1057. }
  1058. upb_sink_endseq(sink, getsel(f, UPB_HANDLER_ENDSEQ));
  1059. }
  1060. static void putmsg(zval* msg_php, const Descriptor* desc, upb_sink sink,
  1061. int depth, bool is_json TSRMLS_DC) {
  1062. MessageHeader* msg = UNBOX(MessageHeader, msg_php);
  1063. putrawmsg(msg, desc, sink, depth, is_json, true TSRMLS_CC);
  1064. }
  1065. static const upb_handlers* msgdef_json_serialize_handlers(
  1066. Descriptor* desc, bool preserve_proto_fieldnames);
  1067. static void putjsonany(MessageHeader* msg, const Descriptor* desc,
  1068. upb_sink sink, int depth TSRMLS_DC) {
  1069. upb_status status;
  1070. const upb_fielddef* type_field = upb_msgdef_itof(desc->msgdef, UPB_ANY_TYPE);
  1071. const upb_fielddef* value_field = upb_msgdef_itof(desc->msgdef, UPB_ANY_VALUE);
  1072. uint32_t type_url_offset;
  1073. zval* type_url_php_str;
  1074. const upb_msgdef *payload_type = NULL;
  1075. upb_sink_startmsg(sink);
  1076. /* Handle type url */
  1077. type_url_offset = desc->layout->fields[upb_fielddef_index(type_field)].offset;
  1078. type_url_php_str = CACHED_PTR_TO_ZVAL_PTR(
  1079. DEREF(message_data(msg), type_url_offset, CACHED_VALUE*));
  1080. if (Z_STRLEN_P(type_url_php_str) > 0) {
  1081. putstr(type_url_php_str, type_field, sink, false);
  1082. }
  1083. {
  1084. const char* type_url_str = Z_STRVAL_P(type_url_php_str);
  1085. size_t type_url_len = Z_STRLEN_P(type_url_php_str);
  1086. if (type_url_len <= 20 ||
  1087. strncmp(type_url_str, "type.googleapis.com/", 20) != 0) {
  1088. zend_error(E_ERROR, "Invalid type url: %s", type_url_str);
  1089. }
  1090. /* Resolve type url */
  1091. type_url_str += 20;
  1092. type_url_len -= 20;
  1093. payload_type = upb_symtab_lookupmsg2(
  1094. generated_pool->symtab, type_url_str, type_url_len);
  1095. if (payload_type == NULL) {
  1096. zend_error(E_ERROR, "Unknown type: %s", type_url_str);
  1097. return;
  1098. }
  1099. }
  1100. {
  1101. uint32_t value_offset;
  1102. zval* value_php_str;
  1103. const char* value_str;
  1104. size_t value_len;
  1105. value_offset = desc->layout->fields[upb_fielddef_index(value_field)].offset;
  1106. value_php_str = CACHED_PTR_TO_ZVAL_PTR(
  1107. DEREF(message_data(msg), value_offset, CACHED_VALUE*));
  1108. value_str = Z_STRVAL_P(value_php_str);
  1109. value_len = Z_STRLEN_P(value_php_str);
  1110. if (value_len > 0) {
  1111. Descriptor* payload_desc =
  1112. UNBOX_HASHTABLE_VALUE(Descriptor, get_def_obj((void*)payload_type));
  1113. zend_class_entry* payload_klass = payload_desc->klass;
  1114. zval val;
  1115. upb_sink subsink;
  1116. bool is_wellknown;
  1117. /* Create message of the payload type. */
  1118. ZVAL_OBJ(&val, payload_klass->create_object(payload_klass TSRMLS_CC));
  1119. MessageHeader* intern = UNBOX(MessageHeader, &val);
  1120. custom_data_init(payload_klass, intern PHP_PROTO_TSRMLS_CC);
  1121. merge_from_string(value_str, value_len, payload_desc, intern);
  1122. is_wellknown =
  1123. upb_msgdef_wellknowntype(payload_desc->msgdef) !=
  1124. UPB_WELLKNOWN_UNSPECIFIED;
  1125. if (is_wellknown) {
  1126. upb_sink_startstr(sink, getsel(value_field, UPB_HANDLER_STARTSTR), 0,
  1127. &subsink);
  1128. }
  1129. subsink.handlers =
  1130. msgdef_json_serialize_handlers(payload_desc, true);
  1131. subsink.closure = sink.closure;
  1132. putrawmsg(intern, payload_desc, subsink, depth, true,
  1133. is_wellknown TSRMLS_CC);
  1134. zval_dtor(&val);
  1135. }
  1136. }
  1137. upb_sink_endmsg(sink, &status);
  1138. }
  1139. static void putjsonlistvalue(
  1140. MessageHeader* msg, const Descriptor* desc,
  1141. upb_sink sink, int depth TSRMLS_DC) {
  1142. upb_status status;
  1143. upb_sink subsink;
  1144. const upb_fielddef* f = upb_msgdef_itof(desc->msgdef, 1);
  1145. uint32_t offset = desc->layout->fields[upb_fielddef_index(f)].offset;
  1146. zval* array;
  1147. RepeatedField* intern;
  1148. HashTable *ht;
  1149. int size, i;
  1150. upb_sink_startmsg(sink);
  1151. array = CACHED_PTR_TO_ZVAL_PTR(
  1152. DEREF(message_data(msg), offset, CACHED_VALUE*));
  1153. intern = UNBOX(RepeatedField, array);
  1154. ht = PHP_PROTO_HASH_OF(intern->array);
  1155. size = zend_hash_num_elements(ht);
  1156. if (size == 0) {
  1157. upb_sink_startseq(sink, getsel(f, UPB_HANDLER_STARTSEQ), &subsink);
  1158. upb_sink_endseq(sink, getsel(f, UPB_HANDLER_ENDSEQ));
  1159. } else {
  1160. putarray(array, f, sink, depth, true TSRMLS_CC);
  1161. }
  1162. upb_sink_endmsg(sink, &status);
  1163. }
  1164. static void putjsonstruct(
  1165. MessageHeader* msg, const Descriptor* desc,
  1166. upb_sink sink, int depth TSRMLS_DC) {
  1167. upb_status status;
  1168. upb_sink subsink;
  1169. const upb_fielddef* f = upb_msgdef_itof(desc->msgdef, 1);
  1170. uint32_t offset = desc->layout->fields[upb_fielddef_index(f)].offset;
  1171. zval* map;
  1172. Map* intern;
  1173. int size;
  1174. upb_sink_startmsg(sink);
  1175. map = CACHED_PTR_TO_ZVAL_PTR(
  1176. DEREF(message_data(msg), offset, CACHED_VALUE*));
  1177. intern = UNBOX(Map, map);
  1178. size = upb_strtable_count(&intern->table);
  1179. if (size == 0) {
  1180. upb_sink_startseq(sink, getsel(f, UPB_HANDLER_STARTSEQ), &subsink);
  1181. upb_sink_endseq(sink, getsel(f, UPB_HANDLER_ENDSEQ));
  1182. } else {
  1183. putmap(map, f, sink, depth, true TSRMLS_CC);
  1184. }
  1185. upb_sink_endmsg(sink, &status);
  1186. }
  1187. static void putrawmsg(MessageHeader* msg, const Descriptor* desc,
  1188. upb_sink sink, int depth, bool is_json,
  1189. bool open_msg TSRMLS_DC) {
  1190. upb_msg_field_iter i;
  1191. upb_status status;
  1192. if (is_json &&
  1193. upb_msgdef_wellknowntype(desc->msgdef) == UPB_WELLKNOWN_ANY) {
  1194. putjsonany(msg, desc, sink, depth TSRMLS_CC);
  1195. return;
  1196. }
  1197. if (is_json &&
  1198. upb_msgdef_wellknowntype(desc->msgdef) == UPB_WELLKNOWN_LISTVALUE) {
  1199. putjsonlistvalue(msg, desc, sink, depth TSRMLS_CC);
  1200. return;
  1201. }
  1202. if (is_json &&
  1203. upb_msgdef_wellknowntype(desc->msgdef) == UPB_WELLKNOWN_STRUCT) {
  1204. putjsonstruct(msg, desc, sink, depth TSRMLS_CC);
  1205. return;
  1206. }
  1207. if (open_msg) {
  1208. upb_sink_startmsg(sink);
  1209. }
  1210. // Protect against cycles (possible because users may freely reassign message
  1211. // and repeated fields) by imposing a maximum recursion depth.
  1212. if (depth > ENCODE_MAX_NESTING) {
  1213. zend_error(E_ERROR,
  1214. "Maximum recursion depth exceeded during encoding.");
  1215. }
  1216. for (upb_msg_field_begin(&i, desc->msgdef); !upb_msg_field_done(&i);
  1217. upb_msg_field_next(&i)) {
  1218. upb_fielddef* f = upb_msg_iter_field(&i);
  1219. uint32_t offset = desc->layout->fields[upb_fielddef_index(f)].offset;
  1220. bool containing_oneof = false;
  1221. if (upb_fielddef_containingoneof(f)) {
  1222. uint32_t oneof_case_offset =
  1223. desc->layout->fields[upb_fielddef_index(f)].case_offset;
  1224. // For a oneof, check that this field is actually present -- skip all the
  1225. // below if not.
  1226. if (DEREF(message_data(msg), oneof_case_offset, uint32_t) !=
  1227. upb_fielddef_number(f)) {
  1228. continue;
  1229. }
  1230. // Otherwise, fall through to the appropriate singular-field handler
  1231. // below.
  1232. containing_oneof = true;
  1233. }
  1234. if (is_map_field(f)) {
  1235. zval* map = CACHED_PTR_TO_ZVAL_PTR(
  1236. DEREF(message_data(msg), offset, CACHED_VALUE*));
  1237. if (map != NULL) {
  1238. putmap(map, f, sink, depth, is_json TSRMLS_CC);
  1239. }
  1240. } else if (upb_fielddef_isseq(f)) {
  1241. zval* array = CACHED_PTR_TO_ZVAL_PTR(
  1242. DEREF(message_data(msg), offset, CACHED_VALUE*));
  1243. if (array != NULL) {
  1244. putarray(array, f, sink, depth, is_json TSRMLS_CC);
  1245. }
  1246. } else if (upb_fielddef_isstring(f)) {
  1247. zval* str = CACHED_PTR_TO_ZVAL_PTR(
  1248. DEREF(message_data(msg), offset, CACHED_VALUE*));
  1249. if (containing_oneof || (is_json && is_wrapper_msg(desc->msgdef)) ||
  1250. Z_STRLEN_P(str) > 0) {
  1251. putstr(str, f, sink, is_json && is_wrapper_msg(desc->msgdef));
  1252. }
  1253. } else if (upb_fielddef_issubmsg(f)) {
  1254. putsubmsg(CACHED_PTR_TO_ZVAL_PTR(
  1255. DEREF(message_data(msg), offset, CACHED_VALUE*)),
  1256. f, sink, depth, is_json TSRMLS_CC);
  1257. } else {
  1258. upb_selector_t sel = getsel(f, upb_handlers_getprimitivehandlertype(f));
  1259. #define T(upbtypeconst, upbtype, ctype, default_value) \
  1260. case upbtypeconst: { \
  1261. ctype value = DEREF(message_data(msg), offset, ctype); \
  1262. if (containing_oneof || \
  1263. (is_json && is_wrapper_msg(desc->msgdef)) || \
  1264. value != default_value) { \
  1265. upb_sink_put##upbtype(sink, sel, value); \
  1266. } \
  1267. } break;
  1268. switch (upb_fielddef_type(f)) {
  1269. T(UPB_TYPE_FLOAT, float, float, 0.0)
  1270. T(UPB_TYPE_DOUBLE, double, double, 0.0)
  1271. T(UPB_TYPE_BOOL, bool, uint8_t, 0)
  1272. case UPB_TYPE_ENUM:
  1273. T(UPB_TYPE_INT32, int32, int32_t, 0)
  1274. T(UPB_TYPE_UINT32, uint32, uint32_t, 0)
  1275. T(UPB_TYPE_INT64, int64, int64_t, 0)
  1276. T(UPB_TYPE_UINT64, uint64, uint64_t, 0)
  1277. case UPB_TYPE_STRING:
  1278. case UPB_TYPE_BYTES:
  1279. case UPB_TYPE_MESSAGE:
  1280. zend_error(E_ERROR, "Internal error.");
  1281. }
  1282. #undef T
  1283. }
  1284. }
  1285. stringsink* unknown = DEREF(message_data(msg), 0, stringsink*);
  1286. if (unknown != NULL) {
  1287. upb_sink_putunknown(sink, unknown->ptr, unknown->len);
  1288. }
  1289. if (open_msg) {
  1290. upb_sink_endmsg(sink, &status);
  1291. }
  1292. }
  1293. static void putstr(zval* str, const upb_fielddef *f,
  1294. upb_sink sink, bool force_default) {
  1295. upb_sink subsink;
  1296. if (ZVAL_IS_NULL(str)) return;
  1297. PHP_PROTO_ASSERT(Z_TYPE_P(str) == IS_STRING);
  1298. upb_sink_startstr(sink, getsel(f, UPB_HANDLER_STARTSTR), Z_STRLEN_P(str),
  1299. &subsink);
  1300. // For oneof string field, we may get here with string length is zero.
  1301. if (Z_STRLEN_P(str) > 0 || force_default) {
  1302. // Ensure that the string has the correct encoding. We also check at
  1303. // field-set time, but the user may have mutated the string object since
  1304. // then.
  1305. if (upb_fielddef_type(f) == UPB_TYPE_STRING &&
  1306. !is_structurally_valid_utf8(Z_STRVAL_P(str), Z_STRLEN_P(str))) {
  1307. zend_error(E_USER_ERROR, "Given string is not UTF8 encoded.");
  1308. return;
  1309. }
  1310. upb_sink_putstring(subsink, getsel(f, UPB_HANDLER_STRING), Z_STRVAL_P(str),
  1311. Z_STRLEN_P(str), NULL);
  1312. }
  1313. upb_sink_endstr(sink, getsel(f, UPB_HANDLER_ENDSTR));
  1314. }
  1315. static void putrawstr(const char* str, int len, const upb_fielddef* f,
  1316. upb_sink sink, bool force_default) {
  1317. upb_sink subsink;
  1318. if (len == 0 && !force_default) return;
  1319. // Ensure that the string has the correct encoding. We also check at field-set
  1320. // time, but the user may have mutated the string object since then.
  1321. if (upb_fielddef_type(f) == UPB_TYPE_STRING &&
  1322. !is_structurally_valid_utf8(str, len)) {
  1323. zend_error(E_USER_ERROR, "Given string is not UTF8 encoded.");
  1324. return;
  1325. }
  1326. upb_sink_startstr(sink, getsel(f, UPB_HANDLER_STARTSTR), len, &subsink);
  1327. upb_sink_putstring(subsink, getsel(f, UPB_HANDLER_STRING), str, len, NULL);
  1328. upb_sink_endstr(sink, getsel(f, UPB_HANDLER_ENDSTR));
  1329. }
  1330. static void putsubmsg(zval* submsg_php, const upb_fielddef* f, upb_sink sink,
  1331. int depth, bool is_json TSRMLS_DC) {
  1332. if (Z_TYPE_P(submsg_php) == IS_NULL) return;
  1333. MessageHeader *submsg = UNBOX(MessageHeader, submsg_php);
  1334. putrawsubmsg(submsg, f, sink, depth, is_json TSRMLS_CC);
  1335. }
  1336. static void putrawsubmsg(MessageHeader* submsg, const upb_fielddef* f,
  1337. upb_sink sink, int depth, bool is_json TSRMLS_DC) {
  1338. upb_sink subsink;
  1339. Descriptor* subdesc =
  1340. UNBOX_HASHTABLE_VALUE(Descriptor, get_def_obj(upb_fielddef_msgsubdef(f)));
  1341. upb_sink_startsubmsg(sink, getsel(f, UPB_HANDLER_STARTSUBMSG), &subsink);
  1342. putrawmsg(submsg, subdesc, subsink, depth + 1, is_json, true TSRMLS_CC);
  1343. upb_sink_endsubmsg(sink, getsel(f, UPB_HANDLER_ENDSUBMSG));
  1344. }
  1345. static void putarray(zval* array, const upb_fielddef* f, upb_sink sink,
  1346. int depth, bool is_json TSRMLS_DC) {
  1347. upb_sink subsink;
  1348. upb_fieldtype_t type = upb_fielddef_type(f);
  1349. upb_selector_t sel = 0;
  1350. int size, i;
  1351. PHP_PROTO_ASSERT(array != NULL);
  1352. RepeatedField* intern = UNBOX(RepeatedField, array);
  1353. HashTable *ht = PHP_PROTO_HASH_OF(intern->array);
  1354. size = zend_hash_num_elements(ht);
  1355. if (size == 0) return;
  1356. upb_sink_startseq(sink, getsel(f, UPB_HANDLER_STARTSEQ), &subsink);
  1357. if (upb_fielddef_isprimitive(f)) {
  1358. sel = getsel(f, upb_handlers_getprimitivehandlertype(f));
  1359. }
  1360. for (i = 0; i < size; i++) {
  1361. void* memory = repeated_field_index_native(intern, i TSRMLS_CC);
  1362. switch (type) {
  1363. #define T(upbtypeconst, upbtype, ctype) \
  1364. case upbtypeconst: \
  1365. upb_sink_put##upbtype(subsink, sel, *((ctype*)memory)); \
  1366. break;
  1367. T(UPB_TYPE_FLOAT, float, float)
  1368. T(UPB_TYPE_DOUBLE, double, double)
  1369. T(UPB_TYPE_BOOL, bool, int8_t)
  1370. case UPB_TYPE_ENUM:
  1371. T(UPB_TYPE_INT32, int32, int32_t)
  1372. T(UPB_TYPE_UINT32, uint32, uint32_t)
  1373. T(UPB_TYPE_INT64, int64, int64_t)
  1374. T(UPB_TYPE_UINT64, uint64, uint64_t)
  1375. case UPB_TYPE_STRING:
  1376. case UPB_TYPE_BYTES: {
  1377. #if PHP_MAJOR_VERSION < 7
  1378. const char* rawstr = Z_STRVAL_P(*(zval**)memory);
  1379. int len = Z_STRLEN_P(*(zval**)memory);
  1380. #else
  1381. const char* rawstr = ZSTR_VAL(*(zend_string**)memory);
  1382. int len = ZSTR_LEN(*(zend_string**)memory);
  1383. #endif
  1384. putrawstr(rawstr, len, f, subsink,
  1385. is_json && is_wrapper_msg(upb_fielddef_containingtype(f)));
  1386. break;
  1387. }
  1388. case UPB_TYPE_MESSAGE: {
  1389. #if PHP_MAJOR_VERSION < 7
  1390. MessageHeader *submsg = UNBOX(MessageHeader, *(zval**)memory);
  1391. #else
  1392. MessageHeader *submsg =
  1393. (MessageHeader*)((char*)(Z_OBJ_P((zval*)memory)) -
  1394. XtOffsetOf(MessageHeader, std));
  1395. #endif
  1396. putrawsubmsg(submsg, f, subsink, depth, is_json TSRMLS_CC);
  1397. break;
  1398. }
  1399. #undef T
  1400. }
  1401. }
  1402. upb_sink_endseq(sink, getsel(f, UPB_HANDLER_ENDSEQ));
  1403. }
  1404. static const upb_handlers* msgdef_pb_serialize_handlers(Descriptor* desc) {
  1405. return upb_handlercache_get(desc->pool->pb_serialize_handler_cache,
  1406. desc->msgdef);
  1407. }
  1408. static const upb_handlers* msgdef_json_serialize_handlers(
  1409. Descriptor* desc, bool preserve_proto_fieldnames) {
  1410. if (preserve_proto_fieldnames) {
  1411. return upb_handlercache_get(
  1412. desc->pool->json_serialize_handler_preserve_cache, desc->msgdef);
  1413. } else {
  1414. return upb_handlercache_get(desc->pool->json_serialize_handler_cache,
  1415. desc->msgdef);
  1416. }
  1417. }
  1418. // -----------------------------------------------------------------------------
  1419. // PHP encode/decode methods
  1420. // -----------------------------------------------------------------------------
  1421. void serialize_to_string(zval* val, zval* return_value TSRMLS_DC) {
  1422. Descriptor* desc =
  1423. UNBOX_HASHTABLE_VALUE(Descriptor, get_ce_obj(Z_OBJCE_P(val)));
  1424. stringsink sink;
  1425. stringsink_init(&sink);
  1426. {
  1427. const upb_handlers* serialize_handlers = msgdef_pb_serialize_handlers(desc);
  1428. stackenv se;
  1429. upb_pb_encoder* encoder;
  1430. stackenv_init(&se, "Error occurred during encoding: %s");
  1431. encoder = upb_pb_encoder_create(se.arena, serialize_handlers, sink.sink);
  1432. putmsg(val, desc, upb_pb_encoder_input(encoder), 0, false TSRMLS_CC);
  1433. PHP_PROTO_RETVAL_STRINGL(sink.ptr, sink.len, 1);
  1434. stackenv_uninit(&se);
  1435. stringsink_uninit(&sink);
  1436. }
  1437. }
  1438. PHP_METHOD(Message, serializeToString) {
  1439. serialize_to_string(getThis(), return_value TSRMLS_CC);
  1440. }
  1441. void merge_from_string(const char* data, int data_len, Descriptor* desc,
  1442. MessageHeader* msg) {
  1443. const upb_pbdecodermethod* method = msgdef_decodermethod(desc);
  1444. const upb_handlers* h = upb_pbdecodermethod_desthandlers(method);
  1445. stackenv se;
  1446. upb_sink sink;
  1447. upb_pbdecoder* decoder;
  1448. stackenv_init(&se, "Error occurred during parsing: %s");
  1449. upb_sink_reset(&sink, h, msg);
  1450. decoder = upb_pbdecoder_create(se.arena, method, sink, &se.status);
  1451. upb_bufsrc_putbuf(data, data_len, upb_pbdecoder_input(decoder));
  1452. stackenv_uninit(&se);
  1453. }
  1454. PHP_METHOD(Message, mergeFromString) {
  1455. Descriptor* desc =
  1456. UNBOX_HASHTABLE_VALUE(Descriptor, get_ce_obj(Z_OBJCE_P(getThis())));
  1457. MessageHeader* msg = UNBOX(MessageHeader, getThis());
  1458. char *data = NULL;
  1459. PHP_PROTO_SIZE data_len;
  1460. if (zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC, "s", &data, &data_len) ==
  1461. FAILURE) {
  1462. return;
  1463. }
  1464. merge_from_string(data, data_len, desc, msg);
  1465. }
  1466. PHP_METHOD(Message, serializeToJsonString) {
  1467. Descriptor* desc =
  1468. UNBOX_HASHTABLE_VALUE(Descriptor, get_ce_obj(Z_OBJCE_P(getThis())));
  1469. zend_bool preserve_proto_fieldnames = false;
  1470. if (zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC, "|b",
  1471. &preserve_proto_fieldnames) == FAILURE) {
  1472. return;
  1473. }
  1474. stringsink sink;
  1475. stringsink_init(&sink);
  1476. {
  1477. const upb_handlers* serialize_handlers =
  1478. msgdef_json_serialize_handlers(desc, preserve_proto_fieldnames);
  1479. upb_json_printer* printer;
  1480. stackenv se;
  1481. stackenv_init(&se, "Error occurred during encoding: %s");
  1482. printer = upb_json_printer_create(se.arena, serialize_handlers, sink.sink);
  1483. putmsg(getThis(), desc, upb_json_printer_input(printer), 0, true TSRMLS_CC);
  1484. PHP_PROTO_RETVAL_STRINGL(sink.ptr, sink.len, 1);
  1485. stackenv_uninit(&se);
  1486. stringsink_uninit(&sink);
  1487. }
  1488. }
  1489. PHP_METHOD(Message, mergeFromJsonString) {
  1490. Descriptor* desc =
  1491. UNBOX_HASHTABLE_VALUE(Descriptor, get_ce_obj(Z_OBJCE_P(getThis())));
  1492. MessageHeader* msg = UNBOX(MessageHeader, getThis());
  1493. char *data = NULL;
  1494. PHP_PROTO_SIZE data_len;
  1495. zend_bool ignore_json_unknown = false;
  1496. if (zend_parse_parameters(
  1497. ZEND_NUM_ARGS() TSRMLS_CC, "s|b", &data, &data_len,
  1498. &ignore_json_unknown) ==
  1499. FAILURE) {
  1500. return;
  1501. }
  1502. // TODO(teboring): Check and respect string encoding. If not UTF-8, we need to
  1503. // convert, because string handlers pass data directly to message string
  1504. // fields.
  1505. // TODO(teboring): Clear message.
  1506. {
  1507. const upb_json_parsermethod* method = msgdef_jsonparsermethod(desc);
  1508. stackenv se;
  1509. upb_sink sink;
  1510. upb_json_parser* parser;
  1511. stackenv_init(&se, "Error occurred during parsing: %s");
  1512. upb_sink_reset(&sink, get_fill_handlers(desc), msg);
  1513. parser = upb_json_parser_create(se.arena, method, generated_pool->symtab,
  1514. sink, &se.status, ignore_json_unknown);
  1515. upb_bufsrc_putbuf(data, data_len, upb_json_parser_input(parser));
  1516. stackenv_uninit(&se);
  1517. }
  1518. }
  1519. // TODO(teboring): refactoring with putrawmsg
  1520. static void discard_unknown_fields(MessageHeader* msg) {
  1521. upb_msg_field_iter it;
  1522. stringsink* unknown = DEREF(message_data(msg), 0, stringsink*);
  1523. if (unknown != NULL) {
  1524. stringsink_uninit(unknown);
  1525. FREE(unknown);
  1526. DEREF(message_data(msg), 0, stringsink*) = NULL;
  1527. }
  1528. // Recursively discard unknown fields of submessages.
  1529. Descriptor* desc = msg->descriptor;
  1530. TSRMLS_FETCH();
  1531. for (upb_msg_field_begin(&it, desc->msgdef);
  1532. !upb_msg_field_done(&it);
  1533. upb_msg_field_next(&it)) {
  1534. upb_fielddef* f = upb_msg_iter_field(&it);
  1535. uint32_t offset = desc->layout->fields[upb_fielddef_index(f)].offset;
  1536. bool containing_oneof = false;
  1537. if (upb_fielddef_containingoneof(f)) {
  1538. uint32_t oneof_case_offset =
  1539. desc->layout->fields[upb_fielddef_index(f)].case_offset;
  1540. // For a oneof, check that this field is actually present -- skip all the
  1541. // below if not.
  1542. if (DEREF(message_data(msg), oneof_case_offset, uint32_t) !=
  1543. upb_fielddef_number(f)) {
  1544. continue;
  1545. }
  1546. // Otherwise, fall through to the appropriate singular-field handler
  1547. // below.
  1548. containing_oneof = true;
  1549. }
  1550. if (is_map_field(f)) {
  1551. MapIter map_it;
  1552. int len, size;
  1553. const upb_fielddef* value_field;
  1554. value_field = map_field_value(f);
  1555. if (!upb_fielddef_issubmsg(value_field)) continue;
  1556. zval* map_php = CACHED_PTR_TO_ZVAL_PTR(
  1557. DEREF(message_data(msg), offset, CACHED_VALUE*));
  1558. if (map_php == NULL) continue;
  1559. Map* intern = UNBOX(Map, map_php);
  1560. for (map_begin(map_php, &map_it TSRMLS_CC);
  1561. !map_done(&map_it); map_next(&map_it)) {
  1562. upb_value value = map_iter_value(&map_it, &len);
  1563. const void* memory = raw_value(upb_value_memory(&value), value_field);
  1564. #if PHP_MAJOR_VERSION < 7
  1565. MessageHeader *submsg = UNBOX(MessageHeader, *(zval**)memory);
  1566. #else
  1567. MessageHeader *submsg =
  1568. (MessageHeader*)((char*)(Z_OBJ_P((zval*)memory)) -
  1569. XtOffsetOf(MessageHeader, std));
  1570. #endif
  1571. discard_unknown_fields(submsg);
  1572. }
  1573. } else if (upb_fielddef_isseq(f)) {
  1574. if (!upb_fielddef_issubmsg(f)) continue;
  1575. zval* array_php = CACHED_PTR_TO_ZVAL_PTR(
  1576. DEREF(message_data(msg), offset, CACHED_VALUE*));
  1577. if (array_php == NULL) continue;
  1578. int size, i;
  1579. RepeatedField* intern = UNBOX(RepeatedField, array_php);
  1580. HashTable *ht = PHP_PROTO_HASH_OF(intern->array);
  1581. size = zend_hash_num_elements(ht);
  1582. if (size == 0) continue;
  1583. for (i = 0; i < size; i++) {
  1584. void* memory = repeated_field_index_native(intern, i TSRMLS_CC);
  1585. #if PHP_MAJOR_VERSION < 7
  1586. MessageHeader *submsg = UNBOX(MessageHeader, *(zval**)memory);
  1587. #else
  1588. MessageHeader *submsg =
  1589. (MessageHeader*)((char*)(Z_OBJ_P((zval*)memory)) -
  1590. XtOffsetOf(MessageHeader, std));
  1591. #endif
  1592. discard_unknown_fields(submsg);
  1593. }
  1594. } else if (upb_fielddef_issubmsg(f)) {
  1595. zval* submsg_php = CACHED_PTR_TO_ZVAL_PTR(
  1596. DEREF(message_data(msg), offset, CACHED_VALUE*));
  1597. if (Z_TYPE_P(submsg_php) == IS_NULL) continue;
  1598. MessageHeader* submsg = UNBOX(MessageHeader, submsg_php);
  1599. discard_unknown_fields(submsg);
  1600. }
  1601. }
  1602. }
  1603. PHP_METHOD(Message, discardUnknownFields) {
  1604. MessageHeader* msg = UNBOX(MessageHeader, getThis());
  1605. discard_unknown_fields(msg);
  1606. }