encode_decode.c 78 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177217821792180218121822183218421852186218721882189219021912192219321942195219621972198219922002201220222032204220522062207220822092210221122122213221422152216221722182219222022212222222322242225222622272228222922302231223222332234223522362237223822392240224122422243224422452246224722482249225022512252225322542255225622572258225922602261226222632264226522662267226822692270227122722273227422752276227722782279228022812282228322842285228622872288228922902291229222932294229522962297229822992300230123022303230423052306230723082309231023112312231323142315
  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. bool is_wrapper_msg(const upb_msgdef* m) {
  103. switch (upb_msgdef_wellknowntype(m)) {
  104. case UPB_WELLKNOWN_DOUBLEVALUE:
  105. case UPB_WELLKNOWN_FLOATVALUE:
  106. case UPB_WELLKNOWN_INT64VALUE:
  107. case UPB_WELLKNOWN_UINT64VALUE:
  108. case UPB_WELLKNOWN_INT32VALUE:
  109. case UPB_WELLKNOWN_UINT32VALUE:
  110. case UPB_WELLKNOWN_STRINGVALUE:
  111. case UPB_WELLKNOWN_BYTESVALUE:
  112. case UPB_WELLKNOWN_BOOLVALUE:
  113. return true;
  114. default:
  115. return false;
  116. }
  117. }
  118. typedef struct {
  119. void* closure;
  120. void* submsg;
  121. bool is_msg;
  122. } wrapperfields_parseframe_t;
  123. #define DEREF(msg, ofs, type) *(type*)(((uint8_t *)msg) + ofs)
  124. // Creates a handlerdata that simply contains the offset for this field.
  125. static const void* newhandlerdata(upb_handlers* h, uint32_t ofs) {
  126. size_t* hd_ofs = (size_t*)malloc(sizeof(size_t));
  127. PHP_PROTO_ASSERT(hd_ofs != NULL);
  128. *hd_ofs = ofs;
  129. upb_handlers_addcleanup(h, hd_ofs, free);
  130. return hd_ofs;
  131. }
  132. static const void* newhandlerfielddata(
  133. upb_handlers* h, const upb_fielddef* field) {
  134. const void** hd_field = malloc(sizeof(void*));
  135. PHP_PROTO_ASSERT(hd_field != NULL);
  136. *hd_field = field;
  137. upb_handlers_addcleanup(h, hd_field, free);
  138. return hd_field;
  139. }
  140. typedef struct {
  141. void* closure;
  142. stringsink sink;
  143. } stringfields_parseframe_t;
  144. typedef size_t (*encodeunknown_handlerfunc)(void* _sink, const void* hd,
  145. const char* ptr, size_t len,
  146. const upb_bufhandle* handle);
  147. typedef struct {
  148. encodeunknown_handlerfunc handler;
  149. } unknownfields_handlerdata_t;
  150. // Creates a handlerdata for unknown fields.
  151. static const void *newunknownfieldshandlerdata(upb_handlers* h) {
  152. unknownfields_handlerdata_t* hd =
  153. (unknownfields_handlerdata_t*)malloc(sizeof(unknownfields_handlerdata_t));
  154. PHP_PROTO_ASSERT(hd != NULL);
  155. hd->handler = stringsink_string;
  156. upb_handlers_addcleanup(h, hd, free);
  157. return hd;
  158. }
  159. typedef struct {
  160. const upb_fielddef *fd;
  161. size_t ofs;
  162. const upb_msgdef *md;
  163. } submsg_handlerdata_t;
  164. // Creates a handlerdata that contains field and submessage type information.
  165. static const void *newsubmsghandlerdata(upb_handlers* h, uint32_t ofs,
  166. const upb_fielddef* f) {
  167. submsg_handlerdata_t* hd =
  168. (submsg_handlerdata_t*)malloc(sizeof(submsg_handlerdata_t));
  169. PHP_PROTO_ASSERT(hd != NULL);
  170. hd->fd = f;
  171. hd->ofs = ofs;
  172. hd->md = upb_fielddef_msgsubdef(f);
  173. upb_handlers_addcleanup(h, hd, free);
  174. return hd;
  175. }
  176. typedef struct {
  177. size_t ofs; // union data slot
  178. size_t case_ofs; // oneof_case field
  179. int property_ofs; // properties table cache
  180. uint32_t oneof_case_num; // oneof-case number to place in oneof_case field
  181. const upb_msgdef *md; // msgdef, for oneof submessage handler
  182. const upb_msgdef *parent_md; // msgdef, for parent submessage
  183. } oneof_handlerdata_t;
  184. static const void *newoneofhandlerdata(upb_handlers *h,
  185. uint32_t ofs,
  186. uint32_t case_ofs,
  187. int property_ofs,
  188. const upb_msgdef *m,
  189. const upb_fielddef *f) {
  190. oneof_handlerdata_t* hd =
  191. (oneof_handlerdata_t*)malloc(sizeof(oneof_handlerdata_t));
  192. PHP_PROTO_ASSERT(hd != NULL);
  193. hd->ofs = ofs;
  194. hd->case_ofs = case_ofs;
  195. hd->property_ofs = property_ofs;
  196. hd->parent_md = m;
  197. // We reuse the field tag number as a oneof union discriminant tag. Note that
  198. // we don't expose these numbers to the user, so the only requirement is that
  199. // we have some unique ID for each union case/possibility. The field tag
  200. // numbers are already present and are easy to use so there's no reason to
  201. // create a separate ID space. In addition, using the field tag number here
  202. // lets us easily look up the field in the oneof accessor.
  203. hd->oneof_case_num = upb_fielddef_number(f);
  204. if (upb_fielddef_type(f) == UPB_TYPE_MESSAGE) {
  205. hd->md = upb_fielddef_msgsubdef(f);
  206. } else {
  207. hd->md = NULL;
  208. }
  209. upb_handlers_addcleanup(h, hd, free);
  210. return hd;
  211. }
  212. // A handler that starts a repeated field. Gets the Repeated*Field instance for
  213. // this field (such an instance always exists even in an empty message).
  214. static void *startseq_handler(void* closure, const void* hd) {
  215. MessageHeader* msg = closure;
  216. const upb_fielddef** field = (const upb_fielddef**) hd;
  217. CACHED_VALUE* cache = find_zval_property(msg, *field);
  218. TSRMLS_FETCH();
  219. repeated_field_ensure_created(*field, cache PHP_PROTO_TSRMLS_CC);
  220. return CACHED_PTR_TO_ZVAL_PTR(cache);
  221. }
  222. // Handlers that append primitive values to a repeated field.
  223. #define DEFINE_APPEND_HANDLER(type, ctype) \
  224. static bool append##type##_handler(void* closure, const void* hd, \
  225. ctype val) { \
  226. zval* array = (zval*)closure; \
  227. TSRMLS_FETCH(); \
  228. RepeatedField* intern = UNBOX(RepeatedField, array); \
  229. repeated_field_push_native(intern, &val); \
  230. return true; \
  231. }
  232. DEFINE_APPEND_HANDLER(bool, bool)
  233. DEFINE_APPEND_HANDLER(int32, int32_t)
  234. DEFINE_APPEND_HANDLER(uint32, uint32_t)
  235. DEFINE_APPEND_HANDLER(float, float)
  236. DEFINE_APPEND_HANDLER(int64, int64_t)
  237. DEFINE_APPEND_HANDLER(uint64, uint64_t)
  238. DEFINE_APPEND_HANDLER(double, double)
  239. // Appends a string or 'bytes' string to a repeated field.
  240. static void* appendstr_handler(void *closure,
  241. const void *hd,
  242. size_t size_hint) {
  243. PHP_PROTO_UNUSED(hd);
  244. stringfields_parseframe_t* frame =
  245. (stringfields_parseframe_t*)malloc(sizeof(stringfields_parseframe_t));
  246. PHP_PROTO_ASSERT(frame != NULL);
  247. frame->closure = closure;
  248. stringsink_init(&frame->sink);
  249. return frame;
  250. }
  251. static bool appendstr_end_handler(void *closure, const void *hd) {
  252. stringfields_parseframe_t* frame = closure;
  253. zval* array = (zval*)frame->closure;
  254. TSRMLS_FETCH();
  255. RepeatedField* intern = UNBOX(RepeatedField, array);
  256. #if PHP_MAJOR_VERSION < 7
  257. zval* str;
  258. MAKE_STD_ZVAL(str);
  259. PHP_PROTO_ZVAL_STRINGL(str, frame->sink.ptr, frame->sink.len, 1);
  260. repeated_field_push_native(intern, &str);
  261. #else
  262. zend_string* str = zend_string_init(frame->sink.ptr, frame->sink.len, 1);
  263. repeated_field_push_native(intern, &str);
  264. #endif
  265. stringsink_uninit(&frame->sink);
  266. free(frame);
  267. return true;
  268. }
  269. // Handlers that append primitive values to a repeated field.
  270. #define DEFINE_SINGULAR_HANDLER(type, ctype) \
  271. static bool type##_handler(void* closure, const void* hd, \
  272. ctype val) { \
  273. MessageHeader* msg = (MessageHeader*)closure; \
  274. const size_t *ofs = hd; \
  275. DEREF(message_data(msg), *ofs, ctype) = val; \
  276. return true; \
  277. }
  278. DEFINE_SINGULAR_HANDLER(bool, bool)
  279. DEFINE_SINGULAR_HANDLER(int32, int32_t)
  280. DEFINE_SINGULAR_HANDLER(uint32, uint32_t)
  281. DEFINE_SINGULAR_HANDLER(float, float)
  282. DEFINE_SINGULAR_HANDLER(int64, int64_t)
  283. DEFINE_SINGULAR_HANDLER(uint64, uint64_t)
  284. DEFINE_SINGULAR_HANDLER(double, double)
  285. #undef DEFINE_SINGULAR_HANDLER
  286. #if PHP_MAJOR_VERSION < 7
  287. static void *empty_php_string(zval** value_ptr) {
  288. SEPARATE_ZVAL_IF_NOT_REF(value_ptr);
  289. if (Z_TYPE_PP(value_ptr) == IS_STRING &&
  290. !IS_INTERNED(Z_STRVAL_PP(value_ptr))) {
  291. FREE(Z_STRVAL_PP(value_ptr));
  292. }
  293. ZVAL_EMPTY_STRING(*value_ptr);
  294. return (void*)(*value_ptr);
  295. }
  296. #else
  297. static void *empty_php_string(zval* value_ptr) {
  298. if (Z_TYPE_P(value_ptr) == IS_STRING) {
  299. zend_string_release(Z_STR_P(value_ptr));
  300. }
  301. ZVAL_EMPTY_STRING(value_ptr);
  302. return value_ptr;
  303. }
  304. #endif
  305. #if PHP_MAJOR_VERSION < 7
  306. static void new_php_string(zval** value_ptr, const char* str, size_t len) {
  307. SEPARATE_ZVAL_IF_NOT_REF(value_ptr);
  308. if (Z_TYPE_PP(value_ptr) == IS_STRING &&
  309. !IS_INTERNED(Z_STRVAL_PP(value_ptr))) {
  310. FREE(Z_STRVAL_PP(value_ptr));
  311. }
  312. ZVAL_STRINGL(*value_ptr, str, len, 1);
  313. }
  314. #else
  315. static void new_php_string(zval* value_ptr, const char* str, size_t len) {
  316. if (Z_TYPE_P(value_ptr) == IS_STRING) {
  317. zend_string_release(Z_STR_P(value_ptr));
  318. }
  319. ZVAL_NEW_STR(value_ptr, zend_string_init(str, len, 0));
  320. }
  321. #endif
  322. // Sets a non-repeated string/bytes field in a message.
  323. static void* str_handler(void *closure,
  324. const void *hd,
  325. size_t size_hint) {
  326. PHP_PROTO_UNUSED(hd);
  327. stringfields_parseframe_t* frame =
  328. (stringfields_parseframe_t*)malloc(sizeof(stringfields_parseframe_t));
  329. PHP_PROTO_ASSERT(frame != NULL);
  330. frame->closure = closure;
  331. stringsink_init(&frame->sink);
  332. return frame;
  333. }
  334. static bool str_end_handler(void *closure, const void *hd) {
  335. stringfields_parseframe_t* frame = closure;
  336. const upb_fielddef **field = (const upb_fielddef **) hd;
  337. MessageHeader* msg = (MessageHeader*)frame->closure;
  338. CACHED_VALUE* cached = find_zval_property(msg, *field);
  339. new_php_string(cached, frame->sink.ptr, frame->sink.len);
  340. stringsink_uninit(&frame->sink);
  341. free(frame);
  342. return true;
  343. }
  344. static bool map_str_end_handler(void *closure, const void *hd) {
  345. stringfields_parseframe_t* frame = closure;
  346. const size_t *ofs = hd;
  347. MessageHeader* msg = (MessageHeader*)frame->closure;
  348. new_php_string(DEREF(message_data(msg), *ofs, CACHED_VALUE*),
  349. frame->sink.ptr, frame->sink.len);
  350. stringsink_uninit(&frame->sink);
  351. free(frame);
  352. return true;
  353. }
  354. static size_t stringdata_handler(void* closure, const void* hd,
  355. const char* str, size_t len,
  356. const upb_bufhandle* handle) {
  357. stringfields_parseframe_t* frame = closure;
  358. return stringsink_string(&frame->sink, hd, str, len, handle);
  359. }
  360. // Appends a submessage to a repeated field.
  361. static void *appendsubmsg_handler(void *closure, const void *hd) {
  362. zval* array = (zval*)closure;
  363. TSRMLS_FETCH();
  364. RepeatedField* intern = UNBOX(RepeatedField, array);
  365. const submsg_handlerdata_t *submsgdata = hd;
  366. DescriptorInternal* subdesc = get_msgdef_desc(submsgdata->md);
  367. register_class(subdesc, false TSRMLS_CC);
  368. zend_class_entry* subklass = subdesc->klass;
  369. MessageHeader* submsg;
  370. #if PHP_MAJOR_VERSION < 7
  371. zval* val = NULL;
  372. MAKE_STD_ZVAL(val);
  373. ZVAL_OBJ(val, subklass->create_object(subklass TSRMLS_CC));
  374. repeated_field_push_native(intern, &val);
  375. submsg = UNBOX(MessageHeader, val);
  376. #else
  377. zend_object* obj = subklass->create_object(subklass TSRMLS_CC);
  378. repeated_field_push_native(intern, &obj);
  379. submsg = (MessageHeader*)((char*)obj - XtOffsetOf(MessageHeader, std));
  380. #endif
  381. custom_data_init(subklass, submsg PHP_PROTO_TSRMLS_CC);
  382. return submsg;
  383. }
  384. // Appends a wrapper submessage to a repeated field.
  385. static void *appendwrappersubmsg_handler(void *closure, const void *hd) {
  386. zval* array = (zval*)closure;
  387. TSRMLS_FETCH();
  388. RepeatedField* intern = UNBOX(RepeatedField, array);
  389. const submsg_handlerdata_t *submsgdata = hd;
  390. DescriptorInternal* subdesc = get_msgdef_desc(submsgdata->md);
  391. register_class(subdesc, false TSRMLS_CC);
  392. zend_class_entry* subklass = subdesc->klass;
  393. MessageHeader* submsg;
  394. wrapperfields_parseframe_t* frame =
  395. (wrapperfields_parseframe_t*)malloc(sizeof(wrapperfields_parseframe_t));
  396. #if PHP_MAJOR_VERSION < 7
  397. zval* val = NULL;
  398. MAKE_STD_ZVAL(val);
  399. ZVAL_OBJ(val, subklass->create_object(subklass TSRMLS_CC));
  400. repeated_field_push_native(intern, &val);
  401. submsg = UNBOX(MessageHeader, val);
  402. #else
  403. zend_object* obj = subklass->create_object(subklass TSRMLS_CC);
  404. repeated_field_push_native(intern, &obj);
  405. submsg = (MessageHeader*)((char*)obj - XtOffsetOf(MessageHeader, std));
  406. #endif
  407. custom_data_init(subklass, submsg PHP_PROTO_TSRMLS_CC);
  408. frame->closure = closure;
  409. frame->submsg = submsg;
  410. frame->is_msg = true;
  411. return frame;
  412. }
  413. // Sets a non-repeated submessage field in a message.
  414. static void *submsg_handler(void *closure, const void *hd) {
  415. MessageHeader* msg = closure;
  416. const submsg_handlerdata_t* submsgdata = hd;
  417. TSRMLS_FETCH();
  418. DescriptorInternal* subdesc = get_msgdef_desc(submsgdata->md);
  419. register_class(subdesc, false TSRMLS_CC);
  420. zend_class_entry* subklass = subdesc->klass;
  421. zval* submsg_php;
  422. MessageHeader* submsg;
  423. CACHED_VALUE* cached = find_zval_property(msg, submsgdata->fd);
  424. if (Z_TYPE_P(CACHED_PTR_TO_ZVAL_PTR(cached)) == IS_NULL) {
  425. #if PHP_MAJOR_VERSION < 7
  426. zval val;
  427. ZVAL_OBJ(&val, subklass->create_object(subklass TSRMLS_CC));
  428. MessageHeader* intern = UNBOX(MessageHeader, &val);
  429. custom_data_init(subklass, intern PHP_PROTO_TSRMLS_CC);
  430. REPLACE_ZVAL_VALUE(cached, &val, 1);
  431. zval_dtor(&val);
  432. #else
  433. zend_object* obj = subklass->create_object(subklass TSRMLS_CC);
  434. ZVAL_OBJ(cached, obj);
  435. MessageHeader* intern = UNBOX_HASHTABLE_VALUE(MessageHeader, obj);
  436. custom_data_init(subklass, intern PHP_PROTO_TSRMLS_CC);
  437. #endif
  438. }
  439. submsg_php = CACHED_PTR_TO_ZVAL_PTR(cached);
  440. submsg = UNBOX(MessageHeader, submsg_php);
  441. return submsg;
  442. }
  443. static void *map_submsg_handler(void *closure, const void *hd) {
  444. MessageHeader* msg = closure;
  445. const submsg_handlerdata_t* submsgdata = hd;
  446. TSRMLS_FETCH();
  447. DescriptorInternal* subdesc = get_msgdef_desc(submsgdata->md);
  448. register_class(subdesc, false TSRMLS_CC);
  449. zend_class_entry* subklass = subdesc->klass;
  450. zval* submsg_php;
  451. MessageHeader* submsg;
  452. CACHED_VALUE* cached =
  453. DEREF(message_data(msg), submsgdata->ofs, CACHED_VALUE*);
  454. if (Z_TYPE_P(CACHED_PTR_TO_ZVAL_PTR(cached)) == IS_NULL) {
  455. #if PHP_MAJOR_VERSION < 7
  456. zval val;
  457. ZVAL_OBJ(&val, subklass->create_object(subklass TSRMLS_CC));
  458. MessageHeader* intern = UNBOX(MessageHeader, &val);
  459. custom_data_init(subklass, intern PHP_PROTO_TSRMLS_CC);
  460. REPLACE_ZVAL_VALUE(cached, &val, 1);
  461. zval_dtor(&val);
  462. #else
  463. zend_object* obj = subklass->create_object(subklass TSRMLS_CC);
  464. ZVAL_OBJ(cached, obj);
  465. MessageHeader* intern = UNBOX_HASHTABLE_VALUE(MessageHeader, obj);
  466. custom_data_init(subklass, intern PHP_PROTO_TSRMLS_CC);
  467. #endif
  468. }
  469. submsg_php = CACHED_PTR_TO_ZVAL_PTR(cached);
  470. submsg = UNBOX(MessageHeader, submsg_php);
  471. return submsg;
  472. }
  473. static void *map_wrapper_submsg_handler(void *closure, const void *hd) {
  474. MessageHeader* msg = closure;
  475. const submsg_handlerdata_t* submsgdata = hd;
  476. TSRMLS_FETCH();
  477. DescriptorInternal* subdesc = get_msgdef_desc(submsgdata->md);
  478. register_class(subdesc, false TSRMLS_CC);
  479. zend_class_entry* subklass = subdesc->klass;
  480. zval* submsg_php;
  481. MessageHeader* submsg;
  482. wrapperfields_parseframe_t* frame =
  483. (wrapperfields_parseframe_t*)malloc(sizeof(wrapperfields_parseframe_t));
  484. CACHED_VALUE* cached =
  485. DEREF(message_data(msg), submsgdata->ofs, CACHED_VALUE*);
  486. if (Z_TYPE_P(CACHED_PTR_TO_ZVAL_PTR(cached)) == IS_NULL) {
  487. #if PHP_MAJOR_VERSION < 7
  488. zval val;
  489. ZVAL_OBJ(&val, subklass->create_object(subklass TSRMLS_CC));
  490. MessageHeader* intern = UNBOX(MessageHeader, &val);
  491. custom_data_init(subklass, intern PHP_PROTO_TSRMLS_CC);
  492. REPLACE_ZVAL_VALUE(cached, &val, 1);
  493. zval_dtor(&val);
  494. #else
  495. zend_object* obj = subklass->create_object(subklass TSRMLS_CC);
  496. ZVAL_OBJ(cached, obj);
  497. MessageHeader* intern = UNBOX_HASHTABLE_VALUE(MessageHeader, obj);
  498. custom_data_init(subklass, intern PHP_PROTO_TSRMLS_CC);
  499. #endif
  500. }
  501. submsg_php = CACHED_PTR_TO_ZVAL_PTR(cached);
  502. submsg = UNBOX(MessageHeader, submsg_php);
  503. frame->closure = closure;
  504. frame->submsg = submsg;
  505. frame->is_msg = true;
  506. return frame;
  507. }
  508. // Handler data for startmap/endmap handlers.
  509. typedef struct {
  510. const upb_fielddef* fd;
  511. const upb_msgdef* value_md;
  512. upb_fieldtype_t key_field_type;
  513. upb_fieldtype_t value_field_type;
  514. } map_handlerdata_t;
  515. // Temporary frame for map parsing: at the beginning of a map entry message, a
  516. // submsg handler allocates a frame to hold (i) a reference to the Map object
  517. // into which this message will be inserted and (ii) storage slots to
  518. // temporarily hold the key and value for this map entry until the end of the
  519. // submessage. When the submessage ends, another handler is called to insert the
  520. // value into the map.
  521. typedef struct {
  522. char key_storage[NATIVE_SLOT_MAX_SIZE];
  523. char value_storage[NATIVE_SLOT_MAX_SIZE];
  524. } map_parse_frame_data_t;
  525. PHP_PROTO_WRAP_OBJECT_START(map_parse_frame_t)
  526. map_parse_frame_data_t* data; // Place needs to be consistent with
  527. // MessageHeader.
  528. zval* map;
  529. // In php7, we cannot allocate zval dynamically. So we need to add zval here
  530. // to help decoding.
  531. zval key_zval;
  532. zval value_zval;
  533. PHP_PROTO_WRAP_OBJECT_END
  534. typedef struct map_parse_frame_t map_parse_frame_t;
  535. static void map_slot_init(
  536. void* memory, upb_fieldtype_t type, zval* cache,
  537. const upb_msgdef* value_msg PHP_PROTO_TSRMLS_DC) {
  538. switch (type) {
  539. case UPB_TYPE_STRING:
  540. case UPB_TYPE_BYTES: {
  541. #if PHP_MAJOR_VERSION < 7
  542. // Store zval** in memory in order to be consistent with the layout of
  543. // singular fields.
  544. zval** holder = ALLOC(zval*);
  545. *(zval***)memory = holder;
  546. zval* tmp;
  547. MAKE_STD_ZVAL(tmp);
  548. PHP_PROTO_ZVAL_STRINGL(tmp, "", 0, 1);
  549. *holder = tmp;
  550. #else
  551. *(zval**)memory = cache;
  552. PHP_PROTO_ZVAL_STRINGL(*(zval**)memory, "", 0, 1);
  553. #endif
  554. break;
  555. }
  556. case UPB_TYPE_MESSAGE: {
  557. DescriptorInternal* subdesc = get_msgdef_desc(value_msg);
  558. register_class(subdesc, false TSRMLS_CC);
  559. zend_class_entry* subklass = subdesc->klass;
  560. MessageHeader* submsg;
  561. #if PHP_MAJOR_VERSION < 7
  562. zval** holder = ALLOC(zval*);
  563. zval* tmp;
  564. MAKE_STD_ZVAL(tmp);
  565. ZVAL_OBJ(tmp, subklass->create_object(subklass TSRMLS_CC));
  566. submsg = UNBOX(MessageHeader, tmp);
  567. custom_data_init(subklass, submsg PHP_PROTO_TSRMLS_CC);
  568. *holder = tmp;
  569. *(zval***)memory = holder;
  570. #else
  571. *(zval**)memory = cache;
  572. ZVAL_OBJ(*(zval**)memory, subklass->create_object(subklass TSRMLS_CC));
  573. submsg = UNBOX(MessageHeader, cache);
  574. custom_data_init(subklass, submsg PHP_PROTO_TSRMLS_CC);
  575. #endif
  576. break;
  577. }
  578. default:
  579. native_slot_init(type, memory, NULL);
  580. }
  581. }
  582. static void map_slot_uninit(void* memory, upb_fieldtype_t type) {
  583. switch (type) {
  584. case UPB_TYPE_MESSAGE:
  585. case UPB_TYPE_STRING:
  586. case UPB_TYPE_BYTES: {
  587. #if PHP_MAJOR_VERSION < 7
  588. zval** holder = *(zval***)memory;
  589. zval_ptr_dtor(holder);
  590. FREE(holder);
  591. #else
  592. php_proto_zval_ptr_dtor(*(zval**)memory);
  593. #endif
  594. break;
  595. }
  596. default:
  597. break;
  598. }
  599. }
  600. static void map_slot_key(upb_fieldtype_t type, const void* from,
  601. const char** keyval,
  602. size_t* length) {
  603. if (type == UPB_TYPE_STRING) {
  604. #if PHP_MAJOR_VERSION < 7
  605. zval* key_php = **(zval***)from;
  606. #else
  607. zval* key_php = *(zval**)from;
  608. #endif
  609. *keyval = Z_STRVAL_P(key_php);
  610. *length = Z_STRLEN_P(key_php);
  611. } else {
  612. *keyval = from;
  613. *length = native_slot_size(type);
  614. }
  615. }
  616. static void map_slot_value(upb_fieldtype_t type, const void* from,
  617. upb_value* v) {
  618. size_t len;
  619. void* to = upb_value_memory(v);
  620. #ifndef NDEBUG
  621. v->ctype = UPB_CTYPE_UINT64;
  622. #endif
  623. memset(to, 0, native_slot_size(type));
  624. switch (type) {
  625. #if PHP_MAJOR_VERSION < 7
  626. case UPB_TYPE_STRING:
  627. case UPB_TYPE_BYTES:
  628. case UPB_TYPE_MESSAGE: {
  629. *(zval**)to = **(zval***)from;
  630. Z_ADDREF_PP((zval**)to);
  631. break;
  632. }
  633. #else
  634. case UPB_TYPE_STRING:
  635. case UPB_TYPE_BYTES:
  636. *(zend_string**)to = Z_STR_P(*(zval**)from);
  637. zend_string_addref(*(zend_string**)to);
  638. break;
  639. case UPB_TYPE_MESSAGE:
  640. if (!ZVAL_IS_NULL(*(zval**)from)) {
  641. *(zend_object**)to = Z_OBJ_P(*(zval**)from);
  642. GC_ADDREF(*(zend_object**)to);
  643. }
  644. break;
  645. #endif
  646. default:
  647. len = native_slot_size(type);
  648. memcpy(to, from, len);
  649. }
  650. }
  651. // Handler to begin a map entry: allocates a temporary frame. This is the
  652. // 'startsubmsg' handler on the msgdef that contains the map field.
  653. static void *startmapentry_handler(void *closure, const void *hd) {
  654. MessageHeader* msg = closure;
  655. const map_handlerdata_t* mapdata = hd;
  656. CACHED_VALUE* cache = find_zval_property(msg, mapdata->fd);
  657. TSRMLS_FETCH();
  658. map_field_ensure_created(mapdata->fd, cache PHP_PROTO_TSRMLS_CC);
  659. zval* map = CACHED_PTR_TO_ZVAL_PTR(cache);
  660. map_parse_frame_t* frame = ALLOC(map_parse_frame_t);
  661. frame->data = ALLOC(map_parse_frame_data_t);
  662. frame->map = map;
  663. map_slot_init(&frame->data->key_storage, mapdata->key_field_type,
  664. &frame->key_zval, NULL PHP_PROTO_TSRMLS_CC);
  665. map_slot_init(&frame->data->value_storage, mapdata->value_field_type,
  666. &frame->value_zval, mapdata->value_md PHP_PROTO_TSRMLS_CC);
  667. return frame;
  668. }
  669. // Handler to end a map entry: inserts the value defined during the message into
  670. // the map. This is the 'endmsg' handler on the map entry msgdef.
  671. static bool endmap_handler(void* closure, const void* hd, upb_status* s) {
  672. map_parse_frame_t* frame = closure;
  673. const map_handlerdata_t* mapdata = hd;
  674. TSRMLS_FETCH();
  675. Map *map = UNBOX(Map, frame->map);
  676. const char* keyval = NULL;
  677. upb_value v;
  678. size_t length;
  679. map_slot_key(map->key_type, &frame->data->key_storage, &keyval, &length);
  680. map_slot_value(map->value_type, &frame->data->value_storage, &v);
  681. map_index_set(map, keyval, length, v);
  682. map_slot_uninit(&frame->data->key_storage, mapdata->key_field_type);
  683. map_slot_uninit(&frame->data->value_storage, mapdata->value_field_type);
  684. FREE(frame->data);
  685. FREE(frame);
  686. return true;
  687. }
  688. // Allocates a new map_handlerdata_t given the map entry message definition. If
  689. // the offset of the field within the parent message is also given, that is
  690. // added to the handler data as well. Note that this is called *twice* per map
  691. // field: once in the parent message handler setup when setting the startsubmsg
  692. // handler and once in the map entry message handler setup when setting the
  693. // key/value and endmsg handlers. The reason is that there is no easy way to
  694. // pass the handlerdata down to the sub-message handler setup.
  695. static map_handlerdata_t* new_map_handlerdata(
  696. const upb_fielddef* field,
  697. const upb_msgdef* mapentry_def) {
  698. const upb_fielddef* key_field;
  699. const upb_fielddef* value_field;
  700. // TODO(teboring): Use emalloc and efree.
  701. map_handlerdata_t* hd =
  702. (map_handlerdata_t*)malloc(sizeof(map_handlerdata_t));
  703. PHP_PROTO_ASSERT(hd != NULL);
  704. hd->fd = field;
  705. key_field = upb_msgdef_itof(mapentry_def, MAP_KEY_FIELD);
  706. PHP_PROTO_ASSERT(key_field != NULL);
  707. hd->key_field_type = upb_fielddef_type(key_field);
  708. value_field = upb_msgdef_itof(mapentry_def, MAP_VALUE_FIELD);
  709. PHP_PROTO_ASSERT(value_field != NULL);
  710. hd->value_field_type = upb_fielddef_type(value_field);
  711. if (upb_fielddef_type(value_field) == UPB_TYPE_MESSAGE) {
  712. hd->value_md = upb_fielddef_msgsubdef(value_field);
  713. } else {
  714. hd->value_md = NULL;
  715. }
  716. return hd;
  717. }
  718. // Handlers that set primitive values in oneofs.
  719. #define DEFINE_ONEOF_HANDLER(type, ctype) \
  720. static bool oneof##type##_handler(void* closure, const void* hd, \
  721. ctype val) { \
  722. const oneof_handlerdata_t* oneofdata = hd; \
  723. MessageHeader* msg = (MessageHeader*)closure; \
  724. DEREF(message_data(closure), oneofdata->case_ofs, uint32_t) = \
  725. oneofdata->oneof_case_num; \
  726. DEREF(message_data(closure), oneofdata->ofs, ctype) = val; \
  727. return true; \
  728. }
  729. DEFINE_ONEOF_HANDLER(bool, bool)
  730. DEFINE_ONEOF_HANDLER(int32, int32_t)
  731. DEFINE_ONEOF_HANDLER(uint32, uint32_t)
  732. DEFINE_ONEOF_HANDLER(float, float)
  733. DEFINE_ONEOF_HANDLER(int64, int64_t)
  734. DEFINE_ONEOF_HANDLER(uint64, uint64_t)
  735. DEFINE_ONEOF_HANDLER(double, double)
  736. #undef DEFINE_ONEOF_HANDLER
  737. static void oneof_cleanup(MessageHeader* msg,
  738. const oneof_handlerdata_t* oneofdata) {
  739. uint32_t old_case_num =
  740. DEREF(message_data(msg), oneofdata->case_ofs, uint32_t);
  741. if (old_case_num == 0) {
  742. return;
  743. }
  744. const upb_fielddef* old_field =
  745. upb_msgdef_itof(oneofdata->parent_md, old_case_num);
  746. bool need_clean = false;
  747. switch (upb_fielddef_type(old_field)) {
  748. case UPB_TYPE_STRING:
  749. case UPB_TYPE_BYTES:
  750. need_clean = true;
  751. break;
  752. case UPB_TYPE_MESSAGE:
  753. if (oneofdata->oneof_case_num != old_case_num) {
  754. need_clean = true;
  755. }
  756. break;
  757. default:
  758. break;
  759. }
  760. if (need_clean) {
  761. #if PHP_MAJOR_VERSION < 7
  762. SEPARATE_ZVAL_IF_NOT_REF(
  763. DEREF(message_data(msg), oneofdata->ofs, CACHED_VALUE*));
  764. php_proto_zval_ptr_dtor(
  765. *DEREF(message_data(msg), oneofdata->ofs, CACHED_VALUE*));
  766. MAKE_STD_ZVAL(*DEREF(message_data(msg), oneofdata->ofs, CACHED_VALUE*));
  767. ZVAL_NULL(*DEREF(message_data(msg), oneofdata->ofs, CACHED_VALUE*));
  768. #endif
  769. }
  770. }
  771. // Handlers for string/bytes in a oneof.
  772. static void *oneofbytes_handler(void *closure,
  773. const void *hd,
  774. size_t size_hint) {
  775. MessageHeader* msg = closure;
  776. const oneof_handlerdata_t *oneofdata = hd;
  777. oneof_cleanup(msg, oneofdata);
  778. DEREF(message_data(msg), oneofdata->case_ofs, uint32_t) =
  779. oneofdata->oneof_case_num;
  780. DEREF(message_data(msg), oneofdata->ofs, CACHED_VALUE*) =
  781. OBJ_PROP(&msg->std, oneofdata->property_ofs);
  782. return empty_php_string(DEREF(
  783. message_data(msg), oneofdata->ofs, CACHED_VALUE*));
  784. }
  785. static bool oneofstr_end_handler(void *closure, const void *hd) {
  786. stringfields_parseframe_t* frame = closure;
  787. MessageHeader* msg = (MessageHeader*)frame->closure;
  788. const oneof_handlerdata_t *oneofdata = hd;
  789. oneof_cleanup(msg, oneofdata);
  790. DEREF(message_data(msg), oneofdata->case_ofs, uint32_t) =
  791. oneofdata->oneof_case_num;
  792. DEREF(message_data(msg), oneofdata->ofs, CACHED_VALUE*) =
  793. OBJ_PROP(&msg->std, oneofdata->property_ofs);
  794. new_php_string(DEREF(message_data(msg), oneofdata->ofs, CACHED_VALUE*),
  795. frame->sink.ptr, frame->sink.len);
  796. stringsink_uninit(&frame->sink);
  797. free(frame);
  798. return true;
  799. }
  800. static void *oneofstr_handler(void *closure,
  801. const void *hd,
  802. size_t size_hint) {
  803. PHP_PROTO_UNUSED(hd);
  804. stringfields_parseframe_t* frame =
  805. (stringfields_parseframe_t*)malloc(sizeof(stringfields_parseframe_t));
  806. PHP_PROTO_ASSERT(frame != NULL);
  807. frame->closure = closure;
  808. stringsink_init(&frame->sink);
  809. return frame;
  810. }
  811. // Handler for a submessage field in a oneof.
  812. static void* oneofsubmsg_handler(void* closure, const void* hd) {
  813. MessageHeader* msg = closure;
  814. const oneof_handlerdata_t *oneofdata = hd;
  815. uint32_t oldcase = DEREF(message_data(msg), oneofdata->case_ofs, uint32_t);
  816. TSRMLS_FETCH();
  817. DescriptorInternal* subdesc = get_msgdef_desc(oneofdata->md);
  818. register_class(subdesc, false TSRMLS_CC);
  819. zend_class_entry* subklass = subdesc->klass;
  820. zval* submsg_php;
  821. MessageHeader* submsg;
  822. if (oldcase != oneofdata->oneof_case_num) {
  823. oneof_cleanup(msg, oneofdata);
  824. // Create new message.
  825. DEREF(message_data(msg), oneofdata->ofs, CACHED_VALUE*) =
  826. OBJ_PROP(&msg->std, oneofdata->property_ofs);
  827. #if PHP_MAJOR_VERSION < 7
  828. zval val;
  829. ZVAL_OBJ(&val, subklass->create_object(subklass TSRMLS_CC));
  830. REPLACE_ZVAL_VALUE(DEREF(message_data(msg), oneofdata->ofs, zval**),
  831. &val, 1);
  832. zval_dtor(&val);
  833. #else
  834. zend_object* obj = subklass->create_object(subklass TSRMLS_CC);
  835. ZVAL_OBJ(DEREF(message_data(msg), oneofdata->ofs, zval*), obj);
  836. #endif
  837. }
  838. DEREF(message_data(msg), oneofdata->case_ofs, uint32_t) =
  839. oneofdata->oneof_case_num;
  840. submsg_php = CACHED_PTR_TO_ZVAL_PTR(
  841. DEREF(message_data(msg), oneofdata->ofs, CACHED_VALUE*));
  842. submsg = UNBOX(MessageHeader, submsg_php);
  843. custom_data_init(subklass, submsg PHP_PROTO_TSRMLS_CC);
  844. return submsg;
  845. }
  846. // Sets a non-repeated wrapper submessage field in a message.
  847. static void* wrapper_submsg_handler(void* closure, const void* hd) {
  848. MessageHeader* msg = closure;
  849. const submsg_handlerdata_t* submsgdata = hd;
  850. TSRMLS_FETCH();
  851. DescriptorInternal* subdesc = get_msgdef_desc(submsgdata->md);
  852. register_class(subdesc, false TSRMLS_CC);
  853. zend_class_entry* subklass = subdesc->klass;
  854. zval* submsg_php;
  855. MessageHeader* submsg;
  856. wrapperfields_parseframe_t* frame =
  857. (wrapperfields_parseframe_t*)malloc(sizeof(wrapperfields_parseframe_t));
  858. CACHED_VALUE* cached = find_zval_property(msg, submsgdata->fd);
  859. submsg_php = CACHED_PTR_TO_ZVAL_PTR(cached);
  860. frame->closure = closure;
  861. if (Z_TYPE_P(CACHED_PTR_TO_ZVAL_PTR(cached)) == IS_OBJECT) {
  862. submsg = UNBOX(MessageHeader, submsg_php);
  863. frame->submsg = submsg;
  864. frame->is_msg = true;
  865. } else {
  866. // In this case, wrapper message hasn't been created and value will be
  867. // stored in cache directly.
  868. frame->submsg = cached;
  869. frame->is_msg = false;
  870. }
  871. return frame;
  872. }
  873. // Handler for a wrapper submessage field in a oneof.
  874. static void* wrapper_oneofsubmsg_handler(void* closure, const void* hd) {
  875. MessageHeader* msg = closure;
  876. const oneof_handlerdata_t *oneofdata = hd;
  877. uint32_t oldcase = DEREF(message_data(msg), oneofdata->case_ofs, uint32_t);
  878. TSRMLS_FETCH();
  879. DescriptorInternal* subdesc = get_msgdef_desc(oneofdata->md);
  880. register_class(subdesc, false TSRMLS_CC);
  881. zend_class_entry* subklass = subdesc->klass;
  882. wrapperfields_parseframe_t* frame =
  883. (wrapperfields_parseframe_t*)malloc(sizeof(wrapperfields_parseframe_t));
  884. CACHED_VALUE* cached = OBJ_PROP(&msg->std, oneofdata->property_ofs);
  885. MessageHeader* submsg;
  886. if (oldcase != oneofdata->oneof_case_num) {
  887. oneof_cleanup(msg, oneofdata);
  888. frame->submsg = cached;
  889. frame->is_msg = false;
  890. } else if (Z_TYPE_P(CACHED_PTR_TO_ZVAL_PTR(cached)) == IS_OBJECT) {
  891. submsg = UNBOX(MessageHeader, CACHED_PTR_TO_ZVAL_PTR(cached));
  892. frame->submsg = submsg;
  893. frame->is_msg = true;
  894. } else {
  895. // In this case, wrapper message hasn't been created and value will be
  896. // stored in cache directly.
  897. frame->submsg = cached;
  898. frame->is_msg = false;
  899. }
  900. DEREF(message_data(msg), oneofdata->case_ofs, uint32_t) =
  901. oneofdata->oneof_case_num;
  902. return frame;
  903. }
  904. static bool wrapper_submsg_end_handler(void *closure, const void *hd) {
  905. wrapperfields_parseframe_t* frame = closure;
  906. free(frame);
  907. return true;
  908. }
  909. // Set up handlers for a repeated field.
  910. static void add_handlers_for_repeated_field(upb_handlers *h,
  911. const upb_fielddef *f,
  912. size_t offset) {
  913. upb_handlerattr attr = UPB_HANDLERATTR_INIT;
  914. attr.handler_data = newhandlerfielddata(h, f);
  915. upb_handlers_setstartseq(h, f, startseq_handler, &attr);
  916. switch (upb_fielddef_type(f)) {
  917. #define SET_HANDLER(utype, ltype) \
  918. case utype: \
  919. upb_handlers_set##ltype(h, f, append##ltype##_handler, NULL); \
  920. break;
  921. SET_HANDLER(UPB_TYPE_BOOL, bool);
  922. SET_HANDLER(UPB_TYPE_INT32, int32);
  923. SET_HANDLER(UPB_TYPE_UINT32, uint32);
  924. SET_HANDLER(UPB_TYPE_ENUM, int32);
  925. SET_HANDLER(UPB_TYPE_FLOAT, float);
  926. SET_HANDLER(UPB_TYPE_INT64, int64);
  927. SET_HANDLER(UPB_TYPE_UINT64, uint64);
  928. SET_HANDLER(UPB_TYPE_DOUBLE, double);
  929. #undef SET_HANDLER
  930. case UPB_TYPE_STRING:
  931. case UPB_TYPE_BYTES: {
  932. upb_handlers_setstartstr(h, f, appendstr_handler, NULL);
  933. upb_handlers_setstring(h, f, stringdata_handler, NULL);
  934. upb_handlers_setendstr(h, f, appendstr_end_handler, &attr);
  935. break;
  936. }
  937. case UPB_TYPE_MESSAGE: {
  938. upb_handlerattr attr = UPB_HANDLERATTR_INIT;
  939. attr.handler_data = newsubmsghandlerdata(h, 0, f);
  940. if (is_wrapper_msg(upb_fielddef_msgsubdef(f))) {
  941. upb_handlers_setstartsubmsg(h, f, appendwrappersubmsg_handler, &attr);
  942. upb_handlers_setendsubmsg(h, f, wrapper_submsg_end_handler, &attr);
  943. } else {
  944. upb_handlers_setstartsubmsg(h, f, appendsubmsg_handler, &attr);
  945. }
  946. break;
  947. }
  948. }
  949. }
  950. // Set up handlers for a singular field.
  951. static void add_handlers_for_singular_field(upb_handlers *h,
  952. const upb_fielddef *f,
  953. size_t offset, bool is_map) {
  954. switch (upb_fielddef_type(f)) {
  955. #define SET_HANDLER(utype, ltype) \
  956. case utype: { \
  957. upb_handlerattr attr = UPB_HANDLERATTR_INIT; \
  958. attr.handler_data = newhandlerdata(h, offset); \
  959. upb_handlers_set##ltype(h, f, ltype##_handler, &attr); \
  960. break; \
  961. }
  962. SET_HANDLER(UPB_TYPE_BOOL, bool);
  963. SET_HANDLER(UPB_TYPE_INT32, int32);
  964. SET_HANDLER(UPB_TYPE_UINT32, uint32);
  965. SET_HANDLER(UPB_TYPE_ENUM, int32);
  966. SET_HANDLER(UPB_TYPE_FLOAT, float);
  967. SET_HANDLER(UPB_TYPE_INT64, int64);
  968. SET_HANDLER(UPB_TYPE_UINT64, uint64);
  969. SET_HANDLER(UPB_TYPE_DOUBLE, double);
  970. #undef SET_HANDLER
  971. case UPB_TYPE_STRING:
  972. case UPB_TYPE_BYTES: {
  973. upb_handlerattr attr = UPB_HANDLERATTR_INIT;
  974. if (is_map) {
  975. attr.handler_data = newhandlerdata(h, offset);
  976. } else {
  977. attr.handler_data = newhandlerfielddata(h, f);
  978. }
  979. upb_handlers_setstartstr(h, f, str_handler, &attr);
  980. upb_handlers_setstring(h, f, stringdata_handler, &attr);
  981. if (is_map) {
  982. upb_handlers_setendstr(h, f, map_str_end_handler, &attr);
  983. } else {
  984. upb_handlers_setendstr(h, f, str_end_handler, &attr);
  985. }
  986. break;
  987. }
  988. case UPB_TYPE_MESSAGE: {
  989. upb_handlerattr attr = UPB_HANDLERATTR_INIT;
  990. if (is_map) {
  991. attr.handler_data = newsubmsghandlerdata(h, offset, f);
  992. if (is_wrapper_msg(upb_fielddef_msgsubdef(f))) {
  993. upb_handlers_setstartsubmsg(h, f, map_wrapper_submsg_handler, &attr);
  994. upb_handlers_setendsubmsg(h, f, wrapper_submsg_end_handler, &attr);
  995. } else {
  996. upb_handlers_setstartsubmsg(h, f, map_submsg_handler, &attr);
  997. }
  998. } else if (is_wrapper_msg(upb_fielddef_msgsubdef(f))) {
  999. attr.handler_data = newsubmsghandlerdata(h, 0, f);
  1000. upb_handlers_setstartsubmsg(h, f, wrapper_submsg_handler, &attr);
  1001. upb_handlers_setendsubmsg(h, f, wrapper_submsg_end_handler, &attr);
  1002. } else {
  1003. attr.handler_data = newsubmsghandlerdata(h, 0, f);
  1004. upb_handlers_setstartsubmsg(h, f, submsg_handler, &attr);
  1005. }
  1006. break;
  1007. }
  1008. }
  1009. }
  1010. // Adds handlers to a map field.
  1011. static void add_handlers_for_mapfield(upb_handlers* h,
  1012. const upb_fielddef* fielddef,
  1013. size_t offset) {
  1014. const upb_msgdef* map_msgdef = upb_fielddef_msgsubdef(fielddef);
  1015. map_handlerdata_t* hd = new_map_handlerdata(fielddef, map_msgdef);
  1016. upb_handlerattr attr = UPB_HANDLERATTR_INIT;
  1017. upb_handlers_addcleanup(h, hd, free);
  1018. attr.handler_data = hd;
  1019. upb_handlers_setstartsubmsg(h, fielddef, startmapentry_handler, &attr);
  1020. }
  1021. // Adds handlers to a map-entry msgdef.
  1022. static void add_handlers_for_mapentry(
  1023. const upb_msgdef* msgdef, upb_handlers* h) {
  1024. const upb_fielddef* key_field = map_entry_key(msgdef);
  1025. const upb_fielddef* value_field = map_entry_value(msgdef);
  1026. map_handlerdata_t* hd = new_map_handlerdata(0, msgdef);
  1027. upb_handlerattr attr = UPB_HANDLERATTR_INIT;
  1028. upb_handlers_addcleanup(h, hd, free);
  1029. attr.handler_data = hd;
  1030. upb_handlers_setendmsg(h, endmap_handler, &attr);
  1031. add_handlers_for_singular_field(h, key_field,
  1032. offsetof(map_parse_frame_data_t,
  1033. key_storage), true);
  1034. add_handlers_for_singular_field(h, value_field,
  1035. offsetof(map_parse_frame_data_t,
  1036. value_storage), true);
  1037. }
  1038. // Set up handlers for a oneof field.
  1039. static void add_handlers_for_oneof_field(upb_handlers *h,
  1040. const upb_msgdef *m,
  1041. const upb_fielddef *f,
  1042. size_t offset,
  1043. size_t oneof_case_offset,
  1044. int property_cache_offset) {
  1045. upb_handlerattr attr = UPB_HANDLERATTR_INIT;
  1046. attr.handler_data = newoneofhandlerdata(h, offset, oneof_case_offset,
  1047. property_cache_offset, m, f);
  1048. switch (upb_fielddef_type(f)) {
  1049. #define SET_HANDLER(utype, ltype) \
  1050. case utype: \
  1051. upb_handlers_set##ltype(h, f, oneof##ltype##_handler, &attr); \
  1052. break;
  1053. SET_HANDLER(UPB_TYPE_BOOL, bool);
  1054. SET_HANDLER(UPB_TYPE_INT32, int32);
  1055. SET_HANDLER(UPB_TYPE_UINT32, uint32);
  1056. SET_HANDLER(UPB_TYPE_ENUM, int32);
  1057. SET_HANDLER(UPB_TYPE_FLOAT, float);
  1058. SET_HANDLER(UPB_TYPE_INT64, int64);
  1059. SET_HANDLER(UPB_TYPE_UINT64, uint64);
  1060. SET_HANDLER(UPB_TYPE_DOUBLE, double);
  1061. #undef SET_HANDLER
  1062. case UPB_TYPE_STRING:
  1063. case UPB_TYPE_BYTES: {
  1064. upb_handlers_setstartstr(h, f, oneofstr_handler, &attr);
  1065. upb_handlers_setstring(h, f, stringdata_handler, NULL);
  1066. upb_handlers_setendstr(h, f, oneofstr_end_handler, &attr);
  1067. break;
  1068. }
  1069. case UPB_TYPE_MESSAGE: {
  1070. if (is_wrapper_msg(upb_fielddef_msgsubdef(f))) {
  1071. upb_handlers_setstartsubmsg(h, f, wrapper_oneofsubmsg_handler, &attr);
  1072. upb_handlers_setendsubmsg(h, f, wrapper_submsg_end_handler, &attr);
  1073. } else {
  1074. upb_handlers_setstartsubmsg(h, f, oneofsubmsg_handler, &attr);
  1075. }
  1076. break;
  1077. }
  1078. }
  1079. }
  1080. #define DEFINE_WRAPPER_HANDLER(utype, type, ctype) \
  1081. static bool type##wrapper_handler( \
  1082. void* closure, const void* hd, ctype val) { \
  1083. wrapperfields_parseframe_t* frame = closure; \
  1084. if (frame->is_msg) { \
  1085. MessageHeader* msg = frame->submsg; \
  1086. const size_t *ofs = hd; \
  1087. DEREF(message_data(msg), *ofs, ctype) = val; \
  1088. } else { \
  1089. TSRMLS_FETCH(); \
  1090. native_slot_get(utype, &val, frame->submsg TSRMLS_CC); \
  1091. } \
  1092. return true; \
  1093. }
  1094. DEFINE_WRAPPER_HANDLER(UPB_TYPE_BOOL, bool, bool)
  1095. DEFINE_WRAPPER_HANDLER(UPB_TYPE_INT32, int32, int32_t)
  1096. DEFINE_WRAPPER_HANDLER(UPB_TYPE_UINT32, uint32, uint32_t)
  1097. DEFINE_WRAPPER_HANDLER(UPB_TYPE_FLOAT, float, float)
  1098. DEFINE_WRAPPER_HANDLER(UPB_TYPE_INT64, int64, int64_t)
  1099. DEFINE_WRAPPER_HANDLER(UPB_TYPE_UINT64, uint64, uint64_t)
  1100. DEFINE_WRAPPER_HANDLER(UPB_TYPE_DOUBLE, double, double)
  1101. #undef DEFINE_WRAPPER_HANDLER
  1102. static bool strwrapper_end_handler(void *closure, const void *hd) {
  1103. stringfields_parseframe_t* frame = closure;
  1104. const upb_fielddef **field = (const upb_fielddef **) hd;
  1105. wrapperfields_parseframe_t* wrapper_frame = frame->closure;
  1106. MessageHeader* msg;
  1107. CACHED_VALUE* cached;
  1108. if (wrapper_frame->is_msg) {
  1109. msg = wrapper_frame->submsg;
  1110. cached = find_zval_property(msg, *field);
  1111. } else {
  1112. cached = wrapper_frame->submsg;
  1113. }
  1114. new_php_string(cached, frame->sink.ptr, frame->sink.len);
  1115. stringsink_uninit(&frame->sink);
  1116. free(frame);
  1117. return true;
  1118. }
  1119. static void add_handlers_for_wrapper(const upb_msgdef* msgdef,
  1120. upb_handlers* h) {
  1121. const upb_fielddef* f = upb_msgdef_itof(msgdef, 1);
  1122. DescriptorInternal* desc;
  1123. size_t offset;
  1124. TSRMLS_FETCH();
  1125. desc = get_msgdef_desc(msgdef);
  1126. offset = desc->layout->fields[upb_fielddef_index(f)].offset;
  1127. switch (upb_msgdef_wellknowntype(msgdef)) {
  1128. #define SET_HANDLER(utype, ltype) \
  1129. case utype: { \
  1130. upb_handlerattr attr = UPB_HANDLERATTR_INIT; \
  1131. attr.handler_data = newhandlerdata(h, offset); \
  1132. upb_handlers_set##ltype(h, f, ltype##wrapper_handler, &attr); \
  1133. break; \
  1134. }
  1135. SET_HANDLER(UPB_WELLKNOWN_BOOLVALUE, bool);
  1136. SET_HANDLER(UPB_WELLKNOWN_INT32VALUE, int32);
  1137. SET_HANDLER(UPB_WELLKNOWN_UINT32VALUE, uint32);
  1138. SET_HANDLER(UPB_WELLKNOWN_FLOATVALUE, float);
  1139. SET_HANDLER(UPB_WELLKNOWN_INT64VALUE, int64);
  1140. SET_HANDLER(UPB_WELLKNOWN_UINT64VALUE, uint64);
  1141. SET_HANDLER(UPB_WELLKNOWN_DOUBLEVALUE, double);
  1142. #undef SET_HANDLER
  1143. case UPB_WELLKNOWN_STRINGVALUE:
  1144. case UPB_WELLKNOWN_BYTESVALUE: {
  1145. upb_handlerattr attr = UPB_HANDLERATTR_INIT;
  1146. attr.handler_data = newhandlerfielddata(h, f);
  1147. upb_handlers_setstartstr(h, f, str_handler, &attr);
  1148. upb_handlers_setstring(h, f, stringdata_handler, &attr);
  1149. upb_handlers_setendstr(h, f, strwrapper_end_handler, &attr);
  1150. break;
  1151. }
  1152. default:
  1153. // Cannot reach here.
  1154. break;
  1155. }
  1156. }
  1157. static bool add_unknown_handler(void* closure, const void* hd, const char* buf,
  1158. size_t size) {
  1159. encodeunknown_handlerfunc handler =
  1160. ((unknownfields_handlerdata_t*)hd)->handler;
  1161. MessageHeader* msg = (MessageHeader*)closure;
  1162. stringsink* unknown = DEREF(message_data(msg), 0, stringsink*);
  1163. if (unknown == NULL) {
  1164. DEREF(message_data(msg), 0, stringsink*) = ALLOC(stringsink);
  1165. unknown = DEREF(message_data(msg), 0, stringsink*);
  1166. stringsink_init(unknown);
  1167. }
  1168. handler(unknown, NULL, buf, size, NULL);
  1169. return true;
  1170. }
  1171. void add_handlers_for_message(const void* closure, upb_handlers* h) {
  1172. const upb_msgdef* msgdef = upb_handlers_msgdef(h);
  1173. TSRMLS_FETCH();
  1174. DescriptorInternal* desc = get_msgdef_desc(msgdef);
  1175. register_class(desc, false TSRMLS_CC);
  1176. upb_msg_field_iter i;
  1177. // If this is a mapentry message type, set up a special set of handlers and
  1178. // bail out of the normal (user-defined) message type handling.
  1179. if (upb_msgdef_mapentry(msgdef)) {
  1180. add_handlers_for_mapentry(msgdef, h);
  1181. return;
  1182. }
  1183. // If this is a wrapper message type, set up a special set of handlers and
  1184. // bail out of the normal (user-defined) message type handling.
  1185. if (is_wrapper_msg(msgdef)) {
  1186. add_handlers_for_wrapper(msgdef, h);
  1187. return;
  1188. }
  1189. upb_handlerattr attr = UPB_HANDLERATTR_INIT;
  1190. attr.handler_data = newunknownfieldshandlerdata(h);
  1191. upb_handlers_setunknown(h, add_unknown_handler, &attr);
  1192. for (upb_msg_field_begin(&i, desc->msgdef);
  1193. !upb_msg_field_done(&i);
  1194. upb_msg_field_next(&i)) {
  1195. const upb_fielddef *f = upb_msg_iter_field(&i);
  1196. size_t offset = desc->layout->fields[upb_fielddef_index(f)].offset;
  1197. if (upb_fielddef_containingoneof(f)) {
  1198. size_t oneof_case_offset =
  1199. desc->layout->fields[upb_fielddef_index(f)].case_offset;
  1200. int property_cache_index =
  1201. desc->layout->fields[upb_fielddef_index(f)].cache_index;
  1202. add_handlers_for_oneof_field(h, desc->msgdef, f, offset,
  1203. oneof_case_offset, property_cache_index);
  1204. } else if (is_map_field(f)) {
  1205. add_handlers_for_mapfield(h, f, offset);
  1206. } else if (upb_fielddef_isseq(f)) {
  1207. add_handlers_for_repeated_field(h, f, offset);
  1208. } else {
  1209. add_handlers_for_singular_field(h, f, offset, false);
  1210. }
  1211. }
  1212. }
  1213. // Constructs the handlers for filling a message's data into an in-memory
  1214. // object.
  1215. const upb_handlers* get_fill_handlers(DescriptorInternal* desc) {
  1216. return upb_handlercache_get(desc->pool->fill_handler_cache, desc->msgdef);
  1217. }
  1218. static const upb_pbdecodermethod *msgdef_decodermethod(DescriptorInternal* desc) {
  1219. return upb_pbcodecache_get(desc->pool->fill_method_cache, desc->msgdef);
  1220. }
  1221. static const upb_json_parsermethod *msgdef_jsonparsermethod(DescriptorInternal* desc) {
  1222. return upb_json_codecache_get(desc->pool->json_fill_method_cache, desc->msgdef);
  1223. }
  1224. // -----------------------------------------------------------------------------
  1225. // Serializing.
  1226. // -----------------------------------------------------------------------------
  1227. static void putmsg(zval* msg, const DescriptorInternal* desc, upb_sink sink,
  1228. int depth, bool is_json TSRMLS_DC);
  1229. static void putrawmsg(MessageHeader* msg, const DescriptorInternal* desc,
  1230. upb_sink sink, int depth, bool is_json,
  1231. bool open_msg TSRMLS_DC);
  1232. static void putwrappervalue(
  1233. zval* value, const upb_fielddef* f,
  1234. upb_sink sink, int depth, bool is_json TSRMLS_DC);
  1235. static void putjsonany(MessageHeader* msg, const DescriptorInternal* desc,
  1236. upb_sink sink, int depth TSRMLS_DC);
  1237. static void putjsonlistvalue(
  1238. MessageHeader* msg, const DescriptorInternal* desc,
  1239. upb_sink sink, int depth TSRMLS_DC);
  1240. static void putjsonstruct(
  1241. MessageHeader* msg, const DescriptorInternal* desc,
  1242. upb_sink sink, int depth TSRMLS_DC);
  1243. static void putstr(zval* str, const upb_fielddef* f, upb_sink sink,
  1244. bool force_default);
  1245. static void putrawstr(const char* str, int len, const upb_fielddef* f,
  1246. upb_sink sink, bool force_default);
  1247. static void putsubmsg(zval* submsg, const upb_fielddef* f, upb_sink sink,
  1248. int depth, bool is_json TSRMLS_DC);
  1249. static void putrawsubmsg(MessageHeader* submsg, const upb_fielddef* f,
  1250. upb_sink sink, int depth, bool is_json TSRMLS_DC);
  1251. static void putarray(zval* array, const upb_fielddef* f, upb_sink sink,
  1252. int depth, bool is_json TSRMLS_DC);
  1253. static void putmap(zval* map, const upb_fielddef* f, upb_sink sink,
  1254. int depth, bool is_json TSRMLS_DC);
  1255. static upb_selector_t getsel(const upb_fielddef* f, upb_handlertype_t type) {
  1256. upb_selector_t ret;
  1257. bool ok = upb_handlers_getselector(f, type, &ret);
  1258. PHP_PROTO_ASSERT(ok);
  1259. return ret;
  1260. }
  1261. static void put_optional_value(const void* memory, int len,
  1262. const upb_fielddef* f,
  1263. int depth, upb_sink sink,
  1264. bool is_json TSRMLS_DC) {
  1265. PHP_PROTO_ASSERT(upb_fielddef_label(f) == UPB_LABEL_OPTIONAL);
  1266. switch (upb_fielddef_type(f)) {
  1267. #define T(upbtypeconst, upbtype, ctype, default_value) \
  1268. case upbtypeconst: { \
  1269. ctype value = DEREF(memory, 0, ctype); \
  1270. if (is_json || value != default_value) { \
  1271. upb_selector_t sel = getsel(f, upb_handlers_getprimitivehandlertype(f)); \
  1272. upb_sink_put##upbtype(sink, sel, value); \
  1273. } \
  1274. } break;
  1275. T(UPB_TYPE_FLOAT, float, float, 0.0)
  1276. T(UPB_TYPE_DOUBLE, double, double, 0.0)
  1277. T(UPB_TYPE_BOOL, bool, uint8_t, 0)
  1278. T(UPB_TYPE_ENUM, int32, int32_t, 0)
  1279. T(UPB_TYPE_INT32, int32, int32_t, 0)
  1280. T(UPB_TYPE_UINT32, uint32, uint32_t, 0)
  1281. T(UPB_TYPE_INT64, int64, int64_t, 0)
  1282. T(UPB_TYPE_UINT64, uint64, uint64_t, 0)
  1283. #undef T
  1284. case UPB_TYPE_STRING:
  1285. case UPB_TYPE_BYTES:
  1286. putrawstr(memory, len, f, sink, is_json);
  1287. break;
  1288. case UPB_TYPE_MESSAGE: {
  1289. #if PHP_MAJOR_VERSION < 7
  1290. MessageHeader *submsg = UNBOX(MessageHeader, *(zval**)memory);
  1291. #else
  1292. MessageHeader *submsg =
  1293. (MessageHeader*)((char*)(*(zend_object**)memory) -
  1294. XtOffsetOf(MessageHeader, std));
  1295. #endif
  1296. putrawsubmsg(submsg, f, sink, depth, is_json TSRMLS_CC);
  1297. break;
  1298. }
  1299. default:
  1300. PHP_PROTO_ASSERT(false);
  1301. }
  1302. }
  1303. // Only string/bytes fields are stored as zval.
  1304. static const char* raw_value(void* memory, const upb_fielddef* f) {
  1305. switch (upb_fielddef_type(f)) {
  1306. case UPB_TYPE_STRING:
  1307. case UPB_TYPE_BYTES:
  1308. #if PHP_MAJOR_VERSION < 7
  1309. return Z_STRVAL_PP((zval**)memory);
  1310. #else
  1311. return ZSTR_VAL(*(zend_string**)memory);
  1312. #endif
  1313. break;
  1314. default:
  1315. return memory;
  1316. }
  1317. }
  1318. static int raw_value_len(void* memory, int len, const upb_fielddef* f) {
  1319. switch (upb_fielddef_type(f)) {
  1320. case UPB_TYPE_STRING:
  1321. case UPB_TYPE_BYTES:
  1322. #if PHP_MAJOR_VERSION < 7
  1323. return Z_STRLEN_PP((zval**)memory);
  1324. #else
  1325. return ZSTR_LEN(*(zend_string**)memory);
  1326. #endif
  1327. default:
  1328. return len;
  1329. }
  1330. }
  1331. static void putmap(zval* map, const upb_fielddef* f, upb_sink sink,
  1332. int depth, bool is_json TSRMLS_DC) {
  1333. upb_sink subsink;
  1334. const upb_fielddef* key_field;
  1335. const upb_fielddef* value_field;
  1336. MapIter it;
  1337. int len, size;
  1338. PHP_PROTO_ASSERT(map != NULL);
  1339. Map* intern = UNBOX(Map, map);
  1340. size = upb_strtable_count(&intern->table);
  1341. if (size == 0) return;
  1342. upb_sink_startseq(sink, getsel(f, UPB_HANDLER_STARTSEQ), &subsink);
  1343. PHP_PROTO_ASSERT(upb_fielddef_type(f) == UPB_TYPE_MESSAGE);
  1344. key_field = map_field_key(f);
  1345. value_field = map_field_value(f);
  1346. for (map_begin(map, &it TSRMLS_CC); !map_done(&it); map_next(&it)) {
  1347. upb_status status;
  1348. upb_sink entry_sink;
  1349. upb_sink_startsubmsg(subsink, getsel(f, UPB_HANDLER_STARTSUBMSG),
  1350. &entry_sink);
  1351. upb_sink_startmsg(entry_sink);
  1352. // Serialize key.
  1353. const char *key = map_iter_key(&it, &len);
  1354. put_optional_value(key, len, key_field, depth + 1,
  1355. entry_sink, is_json TSRMLS_CC);
  1356. // Serialize value.
  1357. upb_value value = map_iter_value(&it, &len);
  1358. put_optional_value(raw_value(upb_value_memory(&value), value_field),
  1359. raw_value_len(upb_value_memory(&value), len, value_field),
  1360. value_field, depth + 1, entry_sink, is_json TSRMLS_CC);
  1361. upb_sink_endmsg(entry_sink, &status);
  1362. upb_sink_endsubmsg(subsink, entry_sink, getsel(f, UPB_HANDLER_ENDSUBMSG));
  1363. }
  1364. upb_sink_endseq(sink, getsel(f, UPB_HANDLER_ENDSEQ));
  1365. }
  1366. static void putmsg(zval* msg_php, const DescriptorInternal* desc, upb_sink sink,
  1367. int depth, bool is_json TSRMLS_DC) {
  1368. MessageHeader* msg = UNBOX(MessageHeader, msg_php);
  1369. putrawmsg(msg, desc, sink, depth, is_json, true TSRMLS_CC);
  1370. }
  1371. static const upb_handlers* msgdef_json_serialize_handlers(
  1372. DescriptorInternal* desc, bool preserve_proto_fieldnames);
  1373. static void putjsonany(MessageHeader* msg, const DescriptorInternal* desc,
  1374. upb_sink sink, int depth TSRMLS_DC) {
  1375. upb_status status;
  1376. const upb_fielddef* type_field = upb_msgdef_itof(desc->msgdef, UPB_ANY_TYPE);
  1377. const upb_fielddef* value_field = upb_msgdef_itof(desc->msgdef, UPB_ANY_VALUE);
  1378. zval* type_url_php_str;
  1379. const upb_msgdef *payload_type = NULL;
  1380. upb_sink_startmsg(sink);
  1381. /* Handle type url */
  1382. type_url_php_str = CACHED_PTR_TO_ZVAL_PTR(find_zval_property(msg, type_field));
  1383. if (Z_STRLEN_P(type_url_php_str) > 0) {
  1384. putstr(type_url_php_str, type_field, sink, false);
  1385. }
  1386. {
  1387. const char* type_url_str = Z_STRVAL_P(type_url_php_str);
  1388. size_t type_url_len = Z_STRLEN_P(type_url_php_str);
  1389. if (type_url_len <= 20 ||
  1390. strncmp(type_url_str, "type.googleapis.com/", 20) != 0) {
  1391. zend_error(E_ERROR, "Invalid type url: %s", type_url_str);
  1392. }
  1393. /* Resolve type url */
  1394. type_url_str += 20;
  1395. type_url_len -= 20;
  1396. payload_type = upb_symtab_lookupmsg2(
  1397. generated_pool->symtab, type_url_str, type_url_len);
  1398. if (payload_type == NULL) {
  1399. zend_error(E_ERROR, "Unknown type: %s", type_url_str);
  1400. return;
  1401. }
  1402. }
  1403. {
  1404. zval* value_php_str;
  1405. const char* value_str;
  1406. size_t value_len;
  1407. value_php_str = CACHED_PTR_TO_ZVAL_PTR(find_zval_property(msg, value_field));
  1408. value_str = Z_STRVAL_P(value_php_str);
  1409. value_len = Z_STRLEN_P(value_php_str);
  1410. if (value_len > 0) {
  1411. DescriptorInternal* payload_desc = get_msgdef_desc(payload_type);
  1412. register_class(payload_desc, false TSRMLS_CC);
  1413. zend_class_entry* payload_klass = payload_desc->klass;
  1414. zval val;
  1415. upb_sink subsink;
  1416. bool is_wellknown;
  1417. /* Create message of the payload type. */
  1418. ZVAL_OBJ(&val, payload_klass->create_object(payload_klass TSRMLS_CC));
  1419. MessageHeader* intern = UNBOX(MessageHeader, &val);
  1420. custom_data_init(payload_klass, intern PHP_PROTO_TSRMLS_CC);
  1421. merge_from_string(value_str, value_len, payload_desc, intern);
  1422. is_wellknown =
  1423. upb_msgdef_wellknowntype(payload_desc->msgdef) !=
  1424. UPB_WELLKNOWN_UNSPECIFIED;
  1425. if (is_wellknown) {
  1426. upb_sink_startstr(sink, getsel(value_field, UPB_HANDLER_STARTSTR), 0,
  1427. &subsink);
  1428. }
  1429. subsink.handlers =
  1430. msgdef_json_serialize_handlers(payload_desc, true);
  1431. subsink.closure = sink.closure;
  1432. putrawmsg(intern, payload_desc, subsink, depth, true,
  1433. is_wellknown TSRMLS_CC);
  1434. zval_dtor(&val);
  1435. }
  1436. }
  1437. upb_sink_endmsg(sink, &status);
  1438. }
  1439. static void putjsonlistvalue(
  1440. MessageHeader* msg, const DescriptorInternal* desc,
  1441. upb_sink sink, int depth TSRMLS_DC) {
  1442. upb_status status;
  1443. upb_sink subsink;
  1444. const upb_fielddef* f = upb_msgdef_itof(desc->msgdef, 1);
  1445. uint32_t offset = desc->layout->fields[upb_fielddef_index(f)].offset;
  1446. zval* array;
  1447. RepeatedField* intern;
  1448. HashTable *ht;
  1449. int size, i;
  1450. upb_sink_startmsg(sink);
  1451. array = CACHED_PTR_TO_ZVAL_PTR(find_zval_property(msg, f));
  1452. if (ZVAL_IS_NULL(array)) {
  1453. upb_sink_startseq(sink, getsel(f, UPB_HANDLER_STARTSEQ), &subsink);
  1454. upb_sink_endseq(sink, getsel(f, UPB_HANDLER_ENDSEQ));
  1455. } else {
  1456. intern = UNBOX(RepeatedField, array);
  1457. ht = PHP_PROTO_HASH_OF(intern->array);
  1458. size = zend_hash_num_elements(ht);
  1459. if (size == 0) {
  1460. upb_sink_startseq(sink, getsel(f, UPB_HANDLER_STARTSEQ), &subsink);
  1461. upb_sink_endseq(sink, getsel(f, UPB_HANDLER_ENDSEQ));
  1462. } else {
  1463. putarray(array, f, sink, depth, true TSRMLS_CC);
  1464. }
  1465. }
  1466. upb_sink_endmsg(sink, &status);
  1467. }
  1468. static void putjsonstruct(
  1469. MessageHeader* msg, const DescriptorInternal* desc,
  1470. upb_sink sink, int depth TSRMLS_DC) {
  1471. upb_status status;
  1472. upb_sink subsink;
  1473. const upb_fielddef* f = upb_msgdef_itof(desc->msgdef, 1);
  1474. uint32_t offset = desc->layout->fields[upb_fielddef_index(f)].offset;
  1475. zval* map;
  1476. Map* intern;
  1477. int size;
  1478. upb_sink_startmsg(sink);
  1479. map = CACHED_PTR_TO_ZVAL_PTR(find_zval_property(msg, f));
  1480. if (ZVAL_IS_NULL(map)) {
  1481. upb_sink_startseq(sink, getsel(f, UPB_HANDLER_STARTSEQ), &subsink);
  1482. upb_sink_endseq(sink, getsel(f, UPB_HANDLER_ENDSEQ));
  1483. } else {
  1484. intern = UNBOX(Map, map);
  1485. size = upb_strtable_count(&intern->table);
  1486. if (size == 0) {
  1487. upb_sink_startseq(sink, getsel(f, UPB_HANDLER_STARTSEQ), &subsink);
  1488. upb_sink_endseq(sink, getsel(f, UPB_HANDLER_ENDSEQ));
  1489. } else {
  1490. putmap(map, f, sink, depth, true TSRMLS_CC);
  1491. }
  1492. }
  1493. upb_sink_endmsg(sink, &status);
  1494. }
  1495. static void putrawmsg(MessageHeader* msg, const DescriptorInternal* desc,
  1496. upb_sink sink, int depth, bool is_json,
  1497. bool open_msg TSRMLS_DC) {
  1498. upb_msg_field_iter i;
  1499. upb_status status;
  1500. if (is_json &&
  1501. upb_msgdef_wellknowntype(desc->msgdef) == UPB_WELLKNOWN_ANY) {
  1502. putjsonany(msg, desc, sink, depth TSRMLS_CC);
  1503. return;
  1504. }
  1505. if (is_json &&
  1506. upb_msgdef_wellknowntype(desc->msgdef) == UPB_WELLKNOWN_LISTVALUE) {
  1507. putjsonlistvalue(msg, desc, sink, depth TSRMLS_CC);
  1508. return;
  1509. }
  1510. if (is_json &&
  1511. upb_msgdef_wellknowntype(desc->msgdef) == UPB_WELLKNOWN_STRUCT) {
  1512. putjsonstruct(msg, desc, sink, depth TSRMLS_CC);
  1513. return;
  1514. }
  1515. if (open_msg) {
  1516. upb_sink_startmsg(sink);
  1517. }
  1518. // Protect against cycles (possible because users may freely reassign message
  1519. // and repeated fields) by imposing a maximum recursion depth.
  1520. if (depth > ENCODE_MAX_NESTING) {
  1521. zend_error(E_ERROR,
  1522. "Maximum recursion depth exceeded during encoding.");
  1523. }
  1524. for (upb_msg_field_begin(&i, desc->msgdef); !upb_msg_field_done(&i);
  1525. upb_msg_field_next(&i)) {
  1526. upb_fielddef* f = upb_msg_iter_field(&i);
  1527. uint32_t offset = desc->layout->fields[upb_fielddef_index(f)].offset;
  1528. bool containing_oneof = false;
  1529. if (upb_fielddef_containingoneof(f)) {
  1530. uint32_t oneof_case_offset =
  1531. desc->layout->fields[upb_fielddef_index(f)].case_offset;
  1532. // For a oneof, check that this field is actually present -- skip all the
  1533. // below if not.
  1534. if (DEREF(message_data(msg), oneof_case_offset, uint32_t) !=
  1535. upb_fielddef_number(f)) {
  1536. continue;
  1537. }
  1538. // Otherwise, fall through to the appropriate singular-field handler
  1539. // below.
  1540. containing_oneof = true;
  1541. }
  1542. if (is_map_field(f)) {
  1543. zval* map = CACHED_PTR_TO_ZVAL_PTR(find_zval_property(msg, f));
  1544. if (!ZVAL_IS_NULL(map)) {
  1545. putmap(map, f, sink, depth, is_json TSRMLS_CC);
  1546. }
  1547. } else if (upb_fielddef_isseq(f)) {
  1548. zval* array = CACHED_PTR_TO_ZVAL_PTR(find_zval_property(msg, f));
  1549. if (!ZVAL_IS_NULL(array)) {
  1550. putarray(array, f, sink, depth, is_json TSRMLS_CC);
  1551. }
  1552. } else if (upb_fielddef_isstring(f)) {
  1553. zval* str = CACHED_PTR_TO_ZVAL_PTR(find_zval_property(msg, f));
  1554. if (containing_oneof || (is_json && is_wrapper_msg(desc->msgdef)) ||
  1555. Z_STRLEN_P(str) > 0) {
  1556. putstr(str, f, sink, is_json && is_wrapper_msg(desc->msgdef));
  1557. }
  1558. } else if (upb_fielddef_issubmsg(f)) {
  1559. zval* submsg = CACHED_PTR_TO_ZVAL_PTR(find_zval_property(msg, f));
  1560. if (is_wrapper_msg(upb_fielddef_msgsubdef(f)) &&
  1561. Z_TYPE_P(submsg) != IS_NULL && Z_TYPE_P(submsg) != IS_OBJECT) {
  1562. putwrappervalue(submsg, f, sink, depth, is_json TSRMLS_CC);
  1563. } else {
  1564. putsubmsg(submsg, f, sink, depth, is_json TSRMLS_CC);
  1565. }
  1566. } else {
  1567. upb_selector_t sel = getsel(f, upb_handlers_getprimitivehandlertype(f));
  1568. #define T(upbtypeconst, upbtype, ctype, default_value) \
  1569. case upbtypeconst: { \
  1570. ctype value = DEREF(message_data(msg), offset, ctype); \
  1571. if (containing_oneof || \
  1572. (is_json && is_wrapper_msg(desc->msgdef)) || \
  1573. value != default_value) { \
  1574. upb_sink_put##upbtype(sink, sel, value); \
  1575. } \
  1576. } break;
  1577. switch (upb_fielddef_type(f)) {
  1578. T(UPB_TYPE_FLOAT, float, float, 0.0)
  1579. T(UPB_TYPE_DOUBLE, double, double, 0.0)
  1580. T(UPB_TYPE_BOOL, bool, uint8_t, 0)
  1581. case UPB_TYPE_ENUM:
  1582. T(UPB_TYPE_INT32, int32, int32_t, 0)
  1583. T(UPB_TYPE_UINT32, uint32, uint32_t, 0)
  1584. T(UPB_TYPE_INT64, int64, int64_t, 0)
  1585. T(UPB_TYPE_UINT64, uint64, uint64_t, 0)
  1586. case UPB_TYPE_STRING:
  1587. case UPB_TYPE_BYTES:
  1588. case UPB_TYPE_MESSAGE:
  1589. zend_error(E_ERROR, "Internal error.");
  1590. }
  1591. #undef T
  1592. }
  1593. }
  1594. stringsink* unknown = DEREF(message_data(msg), 0, stringsink*);
  1595. if (unknown != NULL) {
  1596. upb_sink_putunknown(sink, unknown->ptr, unknown->len);
  1597. }
  1598. if (open_msg) {
  1599. upb_sink_endmsg(sink, &status);
  1600. }
  1601. }
  1602. static void putstr(zval* str, const upb_fielddef *f,
  1603. upb_sink sink, bool force_default) {
  1604. upb_sink subsink;
  1605. if (ZVAL_IS_NULL(str)) return;
  1606. PHP_PROTO_ASSERT(Z_TYPE_P(str) == IS_STRING);
  1607. upb_sink_startstr(sink, getsel(f, UPB_HANDLER_STARTSTR), Z_STRLEN_P(str),
  1608. &subsink);
  1609. // For oneof string field, we may get here with string length is zero.
  1610. if (Z_STRLEN_P(str) > 0 || force_default) {
  1611. // Ensure that the string has the correct encoding. We also check at
  1612. // field-set time, but the user may have mutated the string object since
  1613. // then.
  1614. if (upb_fielddef_type(f) == UPB_TYPE_STRING &&
  1615. !is_structurally_valid_utf8(Z_STRVAL_P(str), Z_STRLEN_P(str))) {
  1616. zend_error(E_USER_ERROR, "Given string is not UTF8 encoded.");
  1617. return;
  1618. }
  1619. upb_sink_putstring(subsink, getsel(f, UPB_HANDLER_STRING), Z_STRVAL_P(str),
  1620. Z_STRLEN_P(str), NULL);
  1621. }
  1622. upb_sink_endstr(sink, getsel(f, UPB_HANDLER_ENDSTR));
  1623. }
  1624. static void putrawstr(const char* str, int len, const upb_fielddef* f,
  1625. upb_sink sink, bool force_default) {
  1626. upb_sink subsink;
  1627. if (len == 0 && !force_default) return;
  1628. // Ensure that the string has the correct encoding. We also check at field-set
  1629. // time, but the user may have mutated the string object since then.
  1630. if (upb_fielddef_type(f) == UPB_TYPE_STRING &&
  1631. !is_structurally_valid_utf8(str, len)) {
  1632. zend_error(E_USER_ERROR, "Given string is not UTF8 encoded.");
  1633. return;
  1634. }
  1635. upb_sink_startstr(sink, getsel(f, UPB_HANDLER_STARTSTR), len, &subsink);
  1636. upb_sink_putstring(subsink, getsel(f, UPB_HANDLER_STRING), str, len, NULL);
  1637. upb_sink_endstr(sink, getsel(f, UPB_HANDLER_ENDSTR));
  1638. }
  1639. static void putsubmsg(zval* submsg_php, const upb_fielddef* f, upb_sink sink,
  1640. int depth, bool is_json TSRMLS_DC) {
  1641. if (Z_TYPE_P(submsg_php) == IS_NULL) return;
  1642. MessageHeader *submsg = UNBOX(MessageHeader, submsg_php);
  1643. putrawsubmsg(submsg, f, sink, depth, is_json TSRMLS_CC);
  1644. }
  1645. static void putwrappervalue(
  1646. zval* value, const upb_fielddef* f,
  1647. upb_sink sink, int depth, bool is_json TSRMLS_DC) {
  1648. upb_sink subsink;
  1649. const upb_msgdef* msgdef = upb_fielddef_msgsubdef(f);
  1650. const upb_fielddef* value_field = upb_msgdef_itof(msgdef, 1);
  1651. upb_selector_t sel =
  1652. getsel(value_field, upb_handlers_getprimitivehandlertype(value_field));
  1653. upb_sink_startsubmsg(sink, getsel(f, UPB_HANDLER_STARTSUBMSG), &subsink);
  1654. #define T(upbtypeconst, upbtype, ctype, default_value) \
  1655. case upbtypeconst: { \
  1656. ctype value_raw; \
  1657. native_slot_set(upb_fielddef_type(value_field), NULL, \
  1658. &value_raw, value PHP_PROTO_TSRMLS_CC); \
  1659. if ((is_json && is_wrapper_msg(msgdef)) || \
  1660. value_raw != default_value) { \
  1661. upb_sink_put##upbtype(subsink, sel, value_raw); \
  1662. } \
  1663. } break;
  1664. switch (upb_fielddef_type(value_field)) {
  1665. T(UPB_TYPE_FLOAT, float, float, 0.0)
  1666. T(UPB_TYPE_DOUBLE, double, double, 0.0)
  1667. T(UPB_TYPE_BOOL, bool, uint8_t, 0)
  1668. T(UPB_TYPE_INT32, int32, int32_t, 0)
  1669. T(UPB_TYPE_UINT32, uint32, uint32_t, 0)
  1670. T(UPB_TYPE_INT64, int64, int64_t, 0)
  1671. T(UPB_TYPE_UINT64, uint64, uint64_t, 0)
  1672. case UPB_TYPE_STRING:
  1673. case UPB_TYPE_BYTES: {
  1674. if ((is_json && is_wrapper_msg(msgdef)) ||
  1675. Z_STRLEN_P(value) > 0) {
  1676. putstr(value, value_field, subsink, is_json && is_wrapper_msg(msgdef));
  1677. }
  1678. break;
  1679. }
  1680. case UPB_TYPE_ENUM:
  1681. case UPB_TYPE_MESSAGE:
  1682. zend_error(E_ERROR, "Internal error.");
  1683. }
  1684. #undef T
  1685. upb_sink_endsubmsg(sink, subsink, getsel(f, UPB_HANDLER_ENDSUBMSG));
  1686. }
  1687. static void putrawsubmsg(MessageHeader* submsg, const upb_fielddef* f,
  1688. upb_sink sink, int depth, bool is_json TSRMLS_DC) {
  1689. upb_sink subsink;
  1690. const upb_msgdef* m = upb_fielddef_msgsubdef(f);
  1691. DescriptorInternal* subdesc = get_msgdef_desc(m);
  1692. upb_sink_startsubmsg(sink, getsel(f, UPB_HANDLER_STARTSUBMSG), &subsink);
  1693. putrawmsg(submsg, subdesc, subsink, depth + 1, is_json, true TSRMLS_CC);
  1694. upb_sink_endsubmsg(sink, subsink, getsel(f, UPB_HANDLER_ENDSUBMSG));
  1695. }
  1696. static void putarray(zval* array, const upb_fielddef* f, upb_sink sink,
  1697. int depth, bool is_json TSRMLS_DC) {
  1698. upb_sink subsink;
  1699. upb_fieldtype_t type = upb_fielddef_type(f);
  1700. upb_selector_t sel = 0;
  1701. int size, i;
  1702. PHP_PROTO_ASSERT(array != NULL);
  1703. RepeatedField* intern = UNBOX(RepeatedField, array);
  1704. HashTable *ht = PHP_PROTO_HASH_OF(intern->array);
  1705. size = zend_hash_num_elements(ht);
  1706. if (size == 0) return;
  1707. upb_sink_startseq(sink, getsel(f, UPB_HANDLER_STARTSEQ), &subsink);
  1708. if (upb_fielddef_isprimitive(f)) {
  1709. sel = getsel(f, upb_handlers_getprimitivehandlertype(f));
  1710. }
  1711. for (i = 0; i < size; i++) {
  1712. void* memory = repeated_field_index_native(intern, i TSRMLS_CC);
  1713. switch (type) {
  1714. #define T(upbtypeconst, upbtype, ctype) \
  1715. case upbtypeconst: \
  1716. upb_sink_put##upbtype(subsink, sel, *((ctype*)memory)); \
  1717. break;
  1718. T(UPB_TYPE_FLOAT, float, float)
  1719. T(UPB_TYPE_DOUBLE, double, double)
  1720. T(UPB_TYPE_BOOL, bool, int8_t)
  1721. case UPB_TYPE_ENUM:
  1722. T(UPB_TYPE_INT32, int32, int32_t)
  1723. T(UPB_TYPE_UINT32, uint32, uint32_t)
  1724. T(UPB_TYPE_INT64, int64, int64_t)
  1725. T(UPB_TYPE_UINT64, uint64, uint64_t)
  1726. case UPB_TYPE_STRING:
  1727. case UPB_TYPE_BYTES: {
  1728. #if PHP_MAJOR_VERSION < 7
  1729. const char* rawstr = Z_STRVAL_P(*(zval**)memory);
  1730. int len = Z_STRLEN_P(*(zval**)memory);
  1731. #else
  1732. const char* rawstr = ZSTR_VAL(*(zend_string**)memory);
  1733. int len = ZSTR_LEN(*(zend_string**)memory);
  1734. #endif
  1735. putrawstr(rawstr, len, f, subsink,
  1736. is_json && is_wrapper_msg(upb_fielddef_containingtype(f)));
  1737. break;
  1738. }
  1739. case UPB_TYPE_MESSAGE: {
  1740. #if PHP_MAJOR_VERSION < 7
  1741. MessageHeader *submsg = UNBOX(MessageHeader, *(zval**)memory);
  1742. #else
  1743. MessageHeader *submsg =
  1744. (MessageHeader*)((char*)(Z_OBJ_P((zval*)memory)) -
  1745. XtOffsetOf(MessageHeader, std));
  1746. #endif
  1747. putrawsubmsg(submsg, f, subsink, depth, is_json TSRMLS_CC);
  1748. break;
  1749. }
  1750. #undef T
  1751. }
  1752. }
  1753. upb_sink_endseq(sink, getsel(f, UPB_HANDLER_ENDSEQ));
  1754. }
  1755. static const upb_handlers* msgdef_pb_serialize_handlers(DescriptorInternal* desc) {
  1756. return upb_handlercache_get(desc->pool->pb_serialize_handler_cache,
  1757. desc->msgdef);
  1758. }
  1759. static const upb_handlers* msgdef_json_serialize_handlers(
  1760. DescriptorInternal* desc, bool preserve_proto_fieldnames) {
  1761. if (preserve_proto_fieldnames) {
  1762. return upb_handlercache_get(
  1763. desc->pool->json_serialize_handler_preserve_cache, desc->msgdef);
  1764. } else {
  1765. return upb_handlercache_get(desc->pool->json_serialize_handler_cache,
  1766. desc->msgdef);
  1767. }
  1768. }
  1769. // -----------------------------------------------------------------------------
  1770. // PHP encode/decode methods
  1771. // -----------------------------------------------------------------------------
  1772. void serialize_to_string(zval* val, zval* return_value TSRMLS_DC) {
  1773. DescriptorInternal* desc = get_ce_desc(Z_OBJCE_P(val));
  1774. stringsink sink;
  1775. stringsink_init(&sink);
  1776. {
  1777. const upb_handlers* serialize_handlers = msgdef_pb_serialize_handlers(desc);
  1778. stackenv se;
  1779. upb_pb_encoder* encoder;
  1780. stackenv_init(&se, "Error occurred during encoding: %s");
  1781. encoder = upb_pb_encoder_create(se.arena, serialize_handlers, sink.sink);
  1782. putmsg(val, desc, upb_pb_encoder_input(encoder), 0, false TSRMLS_CC);
  1783. PHP_PROTO_RETVAL_STRINGL(sink.ptr, sink.len, 1);
  1784. stackenv_uninit(&se);
  1785. stringsink_uninit(&sink);
  1786. }
  1787. }
  1788. PHP_METHOD(Message, serializeToString) {
  1789. serialize_to_string(getThis(), return_value TSRMLS_CC);
  1790. }
  1791. void merge_from_string(const char* data, int data_len, DescriptorInternal* desc,
  1792. MessageHeader* msg) {
  1793. const upb_pbdecodermethod* method = msgdef_decodermethod(desc);
  1794. const upb_handlers* h = upb_pbdecodermethod_desthandlers(method);
  1795. stackenv se;
  1796. upb_sink sink;
  1797. upb_pbdecoder* decoder;
  1798. void* closure;
  1799. stackenv_init(&se, "Error occurred during parsing: %s");
  1800. if (is_wrapper_msg(desc->msgdef)) {
  1801. wrapperfields_parseframe_t* frame =
  1802. (wrapperfields_parseframe_t*)malloc(
  1803. sizeof(wrapperfields_parseframe_t));
  1804. frame->submsg = msg;
  1805. frame->is_msg = true;
  1806. closure = frame;
  1807. } else {
  1808. closure = msg;
  1809. }
  1810. upb_sink_reset(&sink, h, closure);
  1811. decoder = upb_pbdecoder_create(se.arena, method, sink, &se.status);
  1812. upb_bufsrc_putbuf(data, data_len, upb_pbdecoder_input(decoder));
  1813. if (is_wrapper_msg(desc->msgdef)) {
  1814. free((wrapperfields_parseframe_t*)closure);
  1815. }
  1816. stackenv_uninit(&se);
  1817. }
  1818. PHP_METHOD(Message, mergeFromString) {
  1819. DescriptorInternal* desc = get_ce_desc(Z_OBJCE_P(getThis()));
  1820. MessageHeader* msg = UNBOX(MessageHeader, getThis());
  1821. char *data = NULL;
  1822. PHP_PROTO_SIZE data_len;
  1823. if (zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC, "s", &data, &data_len) ==
  1824. FAILURE) {
  1825. return;
  1826. }
  1827. merge_from_string(data, data_len, desc, msg);
  1828. }
  1829. PHP_METHOD(Message, serializeToJsonString) {
  1830. DescriptorInternal* desc = get_ce_desc(Z_OBJCE_P(getThis()));
  1831. zend_bool preserve_proto_fieldnames = false;
  1832. if (zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC, "|b",
  1833. &preserve_proto_fieldnames) == FAILURE) {
  1834. return;
  1835. }
  1836. stringsink sink;
  1837. stringsink_init(&sink);
  1838. {
  1839. const upb_handlers* serialize_handlers =
  1840. msgdef_json_serialize_handlers(desc, preserve_proto_fieldnames);
  1841. upb_json_printer* printer;
  1842. stackenv se;
  1843. stackenv_init(&se, "Error occurred during encoding: %s");
  1844. printer = upb_json_printer_create(se.arena, serialize_handlers, sink.sink);
  1845. putmsg(getThis(), desc, upb_json_printer_input(printer), 0, true TSRMLS_CC);
  1846. PHP_PROTO_RETVAL_STRINGL(sink.ptr, sink.len, 1);
  1847. stackenv_uninit(&se);
  1848. stringsink_uninit(&sink);
  1849. }
  1850. }
  1851. PHP_METHOD(Message, mergeFromJsonString) {
  1852. DescriptorInternal* desc = get_ce_desc(Z_OBJCE_P(getThis()));
  1853. MessageHeader* msg = UNBOX(MessageHeader, getThis());
  1854. char *data = NULL;
  1855. PHP_PROTO_SIZE data_len;
  1856. zend_bool ignore_json_unknown = false;
  1857. if (zend_parse_parameters(
  1858. ZEND_NUM_ARGS() TSRMLS_CC, "s|b", &data, &data_len,
  1859. &ignore_json_unknown) ==
  1860. FAILURE) {
  1861. return;
  1862. }
  1863. // TODO(teboring): Check and respect string encoding. If not UTF-8, we need to
  1864. // convert, because string handlers pass data directly to message string
  1865. // fields.
  1866. // TODO(teboring): Clear message.
  1867. {
  1868. const upb_json_parsermethod* method = msgdef_jsonparsermethod(desc);
  1869. stackenv se;
  1870. upb_sink sink;
  1871. upb_json_parser* parser;
  1872. void* closure;
  1873. stackenv_init(&se, "Error occurred during parsing: %s");
  1874. if (is_wrapper_msg(desc->msgdef)) {
  1875. wrapperfields_parseframe_t* frame =
  1876. (wrapperfields_parseframe_t*)malloc(
  1877. sizeof(wrapperfields_parseframe_t));
  1878. frame->submsg = msg;
  1879. frame->is_msg = true;
  1880. closure = frame;
  1881. } else {
  1882. closure = msg;
  1883. }
  1884. upb_sink_reset(&sink, get_fill_handlers(desc), closure);
  1885. parser = upb_json_parser_create(se.arena, method, generated_pool->symtab,
  1886. sink, &se.status, ignore_json_unknown);
  1887. upb_bufsrc_putbuf(data, data_len, upb_json_parser_input(parser));
  1888. if (is_wrapper_msg(desc->msgdef)) {
  1889. free((wrapperfields_parseframe_t*)closure);
  1890. }
  1891. stackenv_uninit(&se);
  1892. }
  1893. }
  1894. // TODO(teboring): refactoring with putrawmsg
  1895. static void discard_unknown_fields(MessageHeader* msg) {
  1896. upb_msg_field_iter it;
  1897. stringsink* unknown = DEREF(message_data(msg), 0, stringsink*);
  1898. if (unknown != NULL) {
  1899. stringsink_uninit(unknown);
  1900. FREE(unknown);
  1901. DEREF(message_data(msg), 0, stringsink*) = NULL;
  1902. }
  1903. // Recursively discard unknown fields of submessages.
  1904. DescriptorInternal* desc = msg->descriptor;
  1905. TSRMLS_FETCH();
  1906. for (upb_msg_field_begin(&it, desc->msgdef);
  1907. !upb_msg_field_done(&it);
  1908. upb_msg_field_next(&it)) {
  1909. upb_fielddef* f = upb_msg_iter_field(&it);
  1910. uint32_t offset = desc->layout->fields[upb_fielddef_index(f)].offset;
  1911. bool containing_oneof = false;
  1912. if (upb_fielddef_containingoneof(f)) {
  1913. uint32_t oneof_case_offset =
  1914. desc->layout->fields[upb_fielddef_index(f)].case_offset;
  1915. // For a oneof, check that this field is actually present -- skip all the
  1916. // below if not.
  1917. if (DEREF(message_data(msg), oneof_case_offset, uint32_t) !=
  1918. upb_fielddef_number(f)) {
  1919. continue;
  1920. }
  1921. // Otherwise, fall through to the appropriate singular-field handler
  1922. // below.
  1923. containing_oneof = true;
  1924. }
  1925. if (is_map_field(f)) {
  1926. MapIter map_it;
  1927. int len, size;
  1928. const upb_fielddef* value_field;
  1929. value_field = map_field_value(f);
  1930. if (!upb_fielddef_issubmsg(value_field)) continue;
  1931. zval* map_php = CACHED_PTR_TO_ZVAL_PTR(find_zval_property(msg, f));
  1932. if (ZVAL_IS_NULL(map_php)) continue;
  1933. Map* intern = UNBOX(Map, map_php);
  1934. for (map_begin(map_php, &map_it TSRMLS_CC);
  1935. !map_done(&map_it); map_next(&map_it)) {
  1936. upb_value value = map_iter_value(&map_it, &len);
  1937. const void* memory = raw_value(upb_value_memory(&value), value_field);
  1938. #if PHP_MAJOR_VERSION < 7
  1939. MessageHeader *submsg = UNBOX(MessageHeader, *(zval**)memory);
  1940. #else
  1941. MessageHeader *submsg =
  1942. (MessageHeader*)((char*)(Z_OBJ_P((zval*)memory)) -
  1943. XtOffsetOf(MessageHeader, std));
  1944. #endif
  1945. discard_unknown_fields(submsg);
  1946. }
  1947. } else if (upb_fielddef_isseq(f)) {
  1948. if (!upb_fielddef_issubmsg(f)) continue;
  1949. zval* array_php = CACHED_PTR_TO_ZVAL_PTR(find_zval_property(msg, f));
  1950. if (ZVAL_IS_NULL(array_php)) continue;
  1951. int size, i;
  1952. RepeatedField* intern = UNBOX(RepeatedField, array_php);
  1953. HashTable *ht = PHP_PROTO_HASH_OF(intern->array);
  1954. size = zend_hash_num_elements(ht);
  1955. if (size == 0) continue;
  1956. for (i = 0; i < size; i++) {
  1957. void* memory = repeated_field_index_native(intern, i TSRMLS_CC);
  1958. #if PHP_MAJOR_VERSION < 7
  1959. MessageHeader *submsg = UNBOX(MessageHeader, *(zval**)memory);
  1960. #else
  1961. MessageHeader *submsg =
  1962. (MessageHeader*)((char*)(Z_OBJ_P((zval*)memory)) -
  1963. XtOffsetOf(MessageHeader, std));
  1964. #endif
  1965. discard_unknown_fields(submsg);
  1966. }
  1967. } else if (upb_fielddef_issubmsg(f)) {
  1968. zval* submsg_php = CACHED_PTR_TO_ZVAL_PTR(find_zval_property(msg, f));
  1969. if (Z_TYPE_P(submsg_php) == IS_NULL) continue;
  1970. MessageHeader* submsg = UNBOX(MessageHeader, submsg_php);
  1971. discard_unknown_fields(submsg);
  1972. }
  1973. }
  1974. }
  1975. PHP_METHOD(Message, discardUnknownFields) {
  1976. MessageHeader* msg = UNBOX(MessageHeader, getThis());
  1977. discard_unknown_fields(msg);
  1978. }