def.c 40 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236
  1. // Protocol Buffers - Google's data interchange format
  2. // Copyright 2008 Google Inc. All rights reserved.
  3. // https://developers.google.com/protocol-buffers/
  4. //
  5. // Redistribution and use in source and binary forms, with or without
  6. // modification, are permitted provided that the following conditions are
  7. // met:
  8. //
  9. // * Redistributions of source code must retain the above copyright
  10. // notice, this list of conditions and the following disclaimer.
  11. // * Redistributions in binary form must reproduce the above
  12. // copyright notice, this list of conditions and the following disclaimer
  13. // in the documentation and/or other materials provided with the
  14. // distribution.
  15. // * Neither the name of Google Inc. nor the names of its
  16. // contributors may be used to endorse or promote products derived from
  17. // this software without specific prior written permission.
  18. //
  19. // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  20. // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  21. // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  22. // A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
  23. // OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  24. // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  25. // LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  26. // DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  27. // THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  28. // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  29. // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  30. #include <php.h>
  31. #include <Zend/zend_exceptions.h>
  32. #include "protobuf.h"
  33. #include "builtin_descriptors.inc"
  34. // Forward declare.
  35. static void descriptor_init_c_instance(Descriptor* intern TSRMLS_DC);
  36. static void descriptor_free_c(Descriptor* object TSRMLS_DC);
  37. static void field_descriptor_init_c_instance(FieldDescriptor* intern TSRMLS_DC);
  38. static void field_descriptor_free_c(FieldDescriptor* object TSRMLS_DC);
  39. static void enum_descriptor_init_c_instance(EnumDescriptor* intern TSRMLS_DC);
  40. static void enum_descriptor_free_c(EnumDescriptor* object TSRMLS_DC);
  41. static void enum_value_descriptor_init_c_instance(
  42. EnumValueDescriptor *intern TSRMLS_DC);
  43. static void enum_value_descriptor_free_c(EnumValueDescriptor *object TSRMLS_DC);
  44. static void descriptor_pool_free_c(DescriptorPool* object TSRMLS_DC);
  45. static void descriptor_pool_init_c_instance(DescriptorPool* pool TSRMLS_DC);
  46. static void internal_descriptor_pool_free_c(
  47. InternalDescriptorPool *object TSRMLS_DC);
  48. static void internal_descriptor_pool_init_c_instance(
  49. InternalDescriptorPool *pool TSRMLS_DC);
  50. static void oneof_descriptor_free_c(Oneof* object TSRMLS_DC);
  51. static void oneof_descriptor_init_c_instance(Oneof* pool TSRMLS_DC);
  52. // -----------------------------------------------------------------------------
  53. // Common Utilities
  54. // -----------------------------------------------------------------------------
  55. static void check_upb_status(const upb_status* status, const char* msg) {
  56. if (!upb_ok(status)) {
  57. zend_error(E_ERROR, "%s: %s\n", msg, upb_status_errmsg(status));
  58. }
  59. }
  60. // Camel-case the field name and append "Entry" for generated map entry name.
  61. // e.g. map<KeyType, ValueType> foo_map => FooMapEntry
  62. static void append_map_entry_name(char *result, const char *field_name,
  63. int pos) {
  64. bool cap_next = true;
  65. int i;
  66. for (i = 0; i < strlen(field_name); ++i) {
  67. if (field_name[i] == '_') {
  68. cap_next = true;
  69. } else if (cap_next) {
  70. // Note: Do not use ctype.h due to locales.
  71. if ('a' <= field_name[i] && field_name[i] <= 'z') {
  72. result[pos++] = field_name[i] - 'a' + 'A';
  73. } else {
  74. result[pos++] = field_name[i];
  75. }
  76. cap_next = false;
  77. } else {
  78. result[pos++] = field_name[i];
  79. }
  80. }
  81. strcat(result, "Entry");
  82. }
  83. // -----------------------------------------------------------------------------
  84. // GPBType
  85. // -----------------------------------------------------------------------------
  86. zend_class_entry* gpb_type_type;
  87. static zend_function_entry gpb_type_methods[] = {
  88. ZEND_FE_END
  89. };
  90. void gpb_type_init(TSRMLS_D) {
  91. zend_class_entry class_type;
  92. INIT_CLASS_ENTRY(class_type, "Google\\Protobuf\\Internal\\GPBType",
  93. gpb_type_methods);
  94. gpb_type_type = zend_register_internal_class(&class_type TSRMLS_CC);
  95. zend_declare_class_constant_long(gpb_type_type, STR("DOUBLE"), 1 TSRMLS_CC);
  96. zend_declare_class_constant_long(gpb_type_type, STR("FLOAT"), 2 TSRMLS_CC);
  97. zend_declare_class_constant_long(gpb_type_type, STR("INT64"), 3 TSRMLS_CC);
  98. zend_declare_class_constant_long(gpb_type_type, STR("UINT64"), 4 TSRMLS_CC);
  99. zend_declare_class_constant_long(gpb_type_type, STR("INT32"), 5 TSRMLS_CC);
  100. zend_declare_class_constant_long(gpb_type_type, STR("FIXED64"), 6 TSRMLS_CC);
  101. zend_declare_class_constant_long(gpb_type_type, STR("FIXED32"), 7 TSRMLS_CC);
  102. zend_declare_class_constant_long(gpb_type_type, STR("BOOL"), 8 TSRMLS_CC);
  103. zend_declare_class_constant_long(gpb_type_type, STR("STRING"), 9 TSRMLS_CC);
  104. zend_declare_class_constant_long(gpb_type_type, STR("GROUP"), 10 TSRMLS_CC);
  105. zend_declare_class_constant_long(gpb_type_type, STR("MESSAGE"), 11 TSRMLS_CC);
  106. zend_declare_class_constant_long(gpb_type_type, STR("BYTES"), 12 TSRMLS_CC);
  107. zend_declare_class_constant_long(gpb_type_type, STR("UINT32"), 13 TSRMLS_CC);
  108. zend_declare_class_constant_long(gpb_type_type, STR("ENUM"), 14 TSRMLS_CC);
  109. zend_declare_class_constant_long(gpb_type_type, STR("SFIXED32"),
  110. 15 TSRMLS_CC);
  111. zend_declare_class_constant_long(gpb_type_type, STR("SFIXED64"),
  112. 16 TSRMLS_CC);
  113. zend_declare_class_constant_long(gpb_type_type, STR("SINT32"), 17 TSRMLS_CC);
  114. zend_declare_class_constant_long(gpb_type_type, STR("SINT64"), 18 TSRMLS_CC);
  115. }
  116. // -----------------------------------------------------------------------------
  117. // Descriptor
  118. // -----------------------------------------------------------------------------
  119. static zend_function_entry descriptor_methods[] = {
  120. PHP_ME(Descriptor, getClass, NULL, ZEND_ACC_PUBLIC)
  121. PHP_ME(Descriptor, getFullName, NULL, ZEND_ACC_PUBLIC)
  122. PHP_ME(Descriptor, getField, NULL, ZEND_ACC_PUBLIC)
  123. PHP_ME(Descriptor, getFieldCount, NULL, ZEND_ACC_PUBLIC)
  124. PHP_ME(Descriptor, getOneofDecl, NULL, ZEND_ACC_PUBLIC)
  125. PHP_ME(Descriptor, getOneofDeclCount, NULL, ZEND_ACC_PUBLIC)
  126. ZEND_FE_END
  127. };
  128. DEFINE_CLASS(Descriptor, descriptor, "Google\\Protobuf\\Descriptor");
  129. static void descriptor_free_c(Descriptor *self TSRMLS_DC) {
  130. }
  131. static void descriptor_init_c_instance(Descriptor *desc TSRMLS_DC) {
  132. desc->intern = NULL;
  133. }
  134. PHP_METHOD(Descriptor, getClass) {
  135. Descriptor* desc = UNBOX(Descriptor, getThis());
  136. DescriptorInternal* intern = desc->intern;
  137. #if PHP_MAJOR_VERSION < 7
  138. const char* classname = intern->klass->name;
  139. #else
  140. const char* classname = ZSTR_VAL(intern->klass->name);
  141. #endif
  142. PHP_PROTO_RETVAL_STRINGL(classname, strlen(classname), 1);
  143. }
  144. PHP_METHOD(Descriptor, getFullName) {
  145. Descriptor* desc = UNBOX(Descriptor, getThis());
  146. DescriptorInternal* intern = desc->intern;
  147. const char* fullname = upb_msgdef_fullname(intern->msgdef);
  148. PHP_PROTO_RETVAL_STRINGL(fullname, strlen(fullname), 1);
  149. }
  150. PHP_METHOD(Descriptor, getField) {
  151. long index;
  152. if (zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC, "l", &index) ==
  153. FAILURE) {
  154. zend_error(E_USER_ERROR, "Expect integer for index.\n");
  155. return;
  156. }
  157. Descriptor* desc = UNBOX(Descriptor, getThis());
  158. DescriptorInternal* intern = desc->intern;
  159. int field_num = upb_msgdef_numfields(intern->msgdef);
  160. if (index < 0 || index >= field_num) {
  161. zend_error(E_USER_ERROR, "Cannot get element at %ld.\n", index);
  162. return;
  163. }
  164. upb_msg_field_iter iter;
  165. int i;
  166. for(upb_msg_field_begin(&iter, intern->msgdef), i = 0;
  167. !upb_msg_field_done(&iter) && i < index;
  168. upb_msg_field_next(&iter), i++);
  169. const upb_fielddef *field = upb_msg_iter_field(&iter);
  170. PHP_PROTO_HASHTABLE_VALUE field_hashtable_value = get_def_obj(field);
  171. if (field_hashtable_value == NULL) {
  172. #if PHP_MAJOR_VERSION < 7
  173. MAKE_STD_ZVAL(field_hashtable_value);
  174. ZVAL_OBJ(field_hashtable_value, field_descriptor_type->create_object(
  175. field_descriptor_type TSRMLS_CC));
  176. Z_DELREF_P(field_hashtable_value);
  177. #else
  178. field_hashtable_value =
  179. field_descriptor_type->create_object(field_descriptor_type TSRMLS_CC);
  180. GC_DELREF(field_hashtable_value);
  181. #endif
  182. FieldDescriptor *field_php =
  183. UNBOX_HASHTABLE_VALUE(FieldDescriptor, field_hashtable_value);
  184. field_php->fielddef = field;
  185. add_def_obj(field, field_hashtable_value);
  186. }
  187. #if PHP_MAJOR_VERSION < 7
  188. RETURN_ZVAL(field_hashtable_value, 1, 0);
  189. #else
  190. GC_ADDREF(field_hashtable_value);
  191. RETURN_OBJ(field_hashtable_value);
  192. #endif
  193. }
  194. PHP_METHOD(Descriptor, getFieldCount) {
  195. Descriptor* desc = UNBOX(Descriptor, getThis());
  196. DescriptorInternal* intern = desc->intern;
  197. RETURN_LONG(upb_msgdef_numfields(intern->msgdef));
  198. }
  199. PHP_METHOD(Descriptor, getOneofDecl) {
  200. long index;
  201. if (zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC, "l", &index) ==
  202. FAILURE) {
  203. zend_error(E_USER_ERROR, "Expect integer for index.\n");
  204. return;
  205. }
  206. Descriptor* desc = UNBOX(Descriptor, getThis());
  207. DescriptorInternal* intern = desc->intern;
  208. int field_num = upb_msgdef_numoneofs(intern->msgdef);
  209. if (index < 0 || index >= field_num) {
  210. zend_error(E_USER_ERROR, "Cannot get element at %ld.\n", index);
  211. return;
  212. }
  213. upb_msg_oneof_iter iter;
  214. int i;
  215. for(upb_msg_oneof_begin(&iter, intern->msgdef), i = 0;
  216. !upb_msg_oneof_done(&iter) && i < index;
  217. upb_msg_oneof_next(&iter), i++);
  218. const upb_oneofdef *oneof = upb_msg_iter_oneof(&iter);
  219. ZVAL_OBJ(return_value, oneof_descriptor_type->create_object(
  220. oneof_descriptor_type TSRMLS_CC));
  221. Oneof *oneof_php = UNBOX(Oneof, return_value);
  222. oneof_php->oneofdef = oneof;
  223. }
  224. PHP_METHOD(Descriptor, getOneofDeclCount) {
  225. Descriptor* desc = UNBOX(Descriptor, getThis());
  226. DescriptorInternal* intern = desc->intern;
  227. RETURN_LONG(upb_msgdef_numoneofs(intern->msgdef));
  228. }
  229. // -----------------------------------------------------------------------------
  230. // EnumDescriptor
  231. // -----------------------------------------------------------------------------
  232. static zend_function_entry enum_descriptor_methods[] = {
  233. PHP_ME(EnumDescriptor, getValue, NULL, ZEND_ACC_PUBLIC)
  234. PHP_ME(EnumDescriptor, getValueCount, NULL, ZEND_ACC_PUBLIC)
  235. ZEND_FE_END
  236. };
  237. DEFINE_CLASS(EnumDescriptor, enum_descriptor,
  238. "Google\\Protobuf\\EnumDescriptor");
  239. static void enum_descriptor_free_c(EnumDescriptor *self TSRMLS_DC) {
  240. }
  241. static void enum_descriptor_init_c_instance(EnumDescriptor *self TSRMLS_DC) {
  242. self->intern = NULL;
  243. }
  244. PHP_METHOD(EnumDescriptor, getValue) {
  245. long index;
  246. if (zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC, "l", &index) ==
  247. FAILURE) {
  248. zend_error(E_USER_ERROR, "Expect integer for index.\n");
  249. return;
  250. }
  251. EnumDescriptor *desc = UNBOX(EnumDescriptor, getThis());
  252. EnumDescriptorInternal *intern = desc->intern;
  253. int field_num = upb_enumdef_numvals(intern->enumdef);
  254. if (index < 0 || index >= field_num) {
  255. zend_error(E_USER_ERROR, "Cannot get element at %ld.\n", index);
  256. return;
  257. }
  258. upb_enum_iter iter;
  259. int i;
  260. for(upb_enum_begin(&iter, intern->enumdef), i = 0;
  261. !upb_enum_done(&iter) && i < index;
  262. upb_enum_next(&iter), i++);
  263. ZVAL_OBJ(return_value, enum_value_descriptor_type->create_object(
  264. enum_value_descriptor_type TSRMLS_CC));
  265. EnumValueDescriptor *enum_value_php =
  266. UNBOX(EnumValueDescriptor, return_value);
  267. enum_value_php->name = upb_enum_iter_name(&iter);
  268. enum_value_php->number = upb_enum_iter_number(&iter);
  269. }
  270. PHP_METHOD(EnumDescriptor, getValueCount) {
  271. EnumDescriptor *desc = UNBOX(EnumDescriptor, getThis());
  272. EnumDescriptorInternal *intern = desc->intern;
  273. RETURN_LONG(upb_enumdef_numvals(intern->enumdef));
  274. }
  275. // -----------------------------------------------------------------------------
  276. // EnumValueDescriptor
  277. // -----------------------------------------------------------------------------
  278. static zend_function_entry enum_value_descriptor_methods[] = {
  279. PHP_ME(EnumValueDescriptor, getName, NULL, ZEND_ACC_PUBLIC)
  280. PHP_ME(EnumValueDescriptor, getNumber, NULL, ZEND_ACC_PUBLIC)
  281. ZEND_FE_END
  282. };
  283. DEFINE_CLASS(EnumValueDescriptor, enum_value_descriptor,
  284. "Google\\Protobuf\\EnumValueDescriptor");
  285. static void enum_value_descriptor_free_c(EnumValueDescriptor *self TSRMLS_DC) {
  286. }
  287. static void enum_value_descriptor_init_c_instance(EnumValueDescriptor *self TSRMLS_DC) {
  288. self->name = NULL;
  289. self->number = 0;
  290. }
  291. PHP_METHOD(EnumValueDescriptor, getName) {
  292. EnumValueDescriptor *intern = UNBOX(EnumValueDescriptor, getThis());
  293. PHP_PROTO_RETVAL_STRINGL(intern->name, strlen(intern->name), 1);
  294. }
  295. PHP_METHOD(EnumValueDescriptor, getNumber) {
  296. EnumValueDescriptor *intern = UNBOX(EnumValueDescriptor, getThis());
  297. RETURN_LONG(intern->number);
  298. }
  299. // -----------------------------------------------------------------------------
  300. // FieldDescriptor
  301. // -----------------------------------------------------------------------------
  302. static zend_function_entry field_descriptor_methods[] = {
  303. PHP_ME(FieldDescriptor, getName, NULL, ZEND_ACC_PUBLIC)
  304. PHP_ME(FieldDescriptor, getNumber, NULL, ZEND_ACC_PUBLIC)
  305. PHP_ME(FieldDescriptor, getLabel, NULL, ZEND_ACC_PUBLIC)
  306. PHP_ME(FieldDescriptor, getType, NULL, ZEND_ACC_PUBLIC)
  307. PHP_ME(FieldDescriptor, isMap, NULL, ZEND_ACC_PUBLIC)
  308. PHP_ME(FieldDescriptor, getEnumType, NULL, ZEND_ACC_PUBLIC)
  309. PHP_ME(FieldDescriptor, getMessageType, NULL, ZEND_ACC_PUBLIC)
  310. ZEND_FE_END
  311. };
  312. DEFINE_CLASS(FieldDescriptor, field_descriptor,
  313. "Google\\Protobuf\\FieldDescriptor");
  314. static void field_descriptor_free_c(FieldDescriptor *self TSRMLS_DC) {
  315. }
  316. static void field_descriptor_init_c_instance(FieldDescriptor *self TSRMLS_DC) {
  317. self->fielddef = NULL;
  318. }
  319. upb_fieldtype_t to_fieldtype(upb_descriptortype_t type) {
  320. switch (type) {
  321. #define CASE(descriptor_type, type) \
  322. case UPB_DESCRIPTOR_TYPE_##descriptor_type: \
  323. return UPB_TYPE_##type;
  324. CASE(FLOAT, FLOAT);
  325. CASE(DOUBLE, DOUBLE);
  326. CASE(BOOL, BOOL);
  327. CASE(STRING, STRING);
  328. CASE(BYTES, BYTES);
  329. CASE(MESSAGE, MESSAGE);
  330. CASE(GROUP, MESSAGE);
  331. CASE(ENUM, ENUM);
  332. CASE(INT32, INT32);
  333. CASE(INT64, INT64);
  334. CASE(UINT32, UINT32);
  335. CASE(UINT64, UINT64);
  336. CASE(SINT32, INT32);
  337. CASE(SINT64, INT64);
  338. CASE(FIXED32, UINT32);
  339. CASE(FIXED64, UINT64);
  340. CASE(SFIXED32, INT32);
  341. CASE(SFIXED64, INT64);
  342. #undef CONVERT
  343. }
  344. zend_error(E_ERROR, "Unknown field type.");
  345. return 0;
  346. }
  347. PHP_METHOD(FieldDescriptor, getName) {
  348. FieldDescriptor *intern = UNBOX(FieldDescriptor, getThis());
  349. const char* name = upb_fielddef_name(intern->fielddef);
  350. PHP_PROTO_RETVAL_STRINGL(name, strlen(name), 1);
  351. }
  352. PHP_METHOD(FieldDescriptor, getNumber) {
  353. FieldDescriptor *intern = UNBOX(FieldDescriptor, getThis());
  354. RETURN_LONG(upb_fielddef_number(intern->fielddef));
  355. }
  356. PHP_METHOD(FieldDescriptor, getLabel) {
  357. FieldDescriptor *intern = UNBOX(FieldDescriptor, getThis());
  358. RETURN_LONG(upb_fielddef_label(intern->fielddef));
  359. }
  360. PHP_METHOD(FieldDescriptor, getType) {
  361. FieldDescriptor *intern = UNBOX(FieldDescriptor, getThis());
  362. RETURN_LONG(upb_fielddef_descriptortype(intern->fielddef));
  363. }
  364. PHP_METHOD(FieldDescriptor, isMap) {
  365. FieldDescriptor *intern = UNBOX(FieldDescriptor, getThis());
  366. RETURN_BOOL(upb_fielddef_ismap(intern->fielddef));
  367. }
  368. PHP_METHOD(FieldDescriptor, getEnumType) {
  369. FieldDescriptor *intern = UNBOX(FieldDescriptor, getThis());
  370. if (upb_fielddef_type(intern->fielddef) != UPB_TYPE_ENUM) {
  371. zend_throw_exception_ex(NULL, 0 TSRMLS_CC,
  372. "Cannot get enum type for non-enum field '%s'",
  373. upb_fielddef_name(intern->fielddef));
  374. return;
  375. }
  376. const upb_enumdef *enumdef = upb_fielddef_enumsubdef(intern->fielddef);
  377. PHP_PROTO_HASHTABLE_VALUE desc_php = get_def_obj(enumdef);
  378. if (desc_php == NULL) {
  379. EnumDescriptorInternal* intern = get_enumdef_enumdesc(enumdef);
  380. #if PHP_MAJOR_VERSION < 7
  381. MAKE_STD_ZVAL(desc_php);
  382. ZVAL_OBJ(desc_php, enum_descriptor_type->create_object(
  383. enum_descriptor_type TSRMLS_CC));
  384. Z_DELREF_P(desc_php);
  385. #else
  386. desc_php =
  387. enum_descriptor_type->create_object(enum_descriptor_type TSRMLS_CC);
  388. GC_DELREF(desc_php);
  389. #endif
  390. EnumDescriptor* desc = UNBOX_HASHTABLE_VALUE(EnumDescriptor, desc_php);
  391. desc->intern = intern;
  392. add_def_obj(enumdef, desc_php);
  393. add_ce_obj(intern->klass, desc_php);
  394. }
  395. #if PHP_MAJOR_VERSION < 7
  396. RETURN_ZVAL(desc_php, 1, 0);
  397. #else
  398. GC_ADDREF(desc_php);
  399. RETURN_OBJ(desc_php);
  400. #endif
  401. }
  402. PHP_METHOD(FieldDescriptor, getMessageType) {
  403. FieldDescriptor *intern = UNBOX(FieldDescriptor, getThis());
  404. if (upb_fielddef_type(intern->fielddef) != UPB_TYPE_MESSAGE) {
  405. zend_throw_exception_ex(
  406. NULL, 0 TSRMLS_CC, "Cannot get message type for non-message field '%s'",
  407. upb_fielddef_name(intern->fielddef));
  408. return;
  409. }
  410. const upb_msgdef *msgdef = upb_fielddef_msgsubdef(intern->fielddef);
  411. PHP_PROTO_HASHTABLE_VALUE desc_php = get_def_obj(msgdef);
  412. if (desc_php == NULL) {
  413. DescriptorInternal* intern = get_msgdef_desc(msgdef);
  414. #if PHP_MAJOR_VERSION < 7
  415. MAKE_STD_ZVAL(desc_php);
  416. ZVAL_OBJ(desc_php, descriptor_type->create_object(
  417. descriptor_type TSRMLS_CC));
  418. Z_DELREF_P(desc_php);
  419. #else
  420. desc_php =
  421. descriptor_type->create_object(descriptor_type TSRMLS_CC);
  422. GC_DELREF(desc_php);
  423. #endif
  424. Descriptor* desc = UNBOX_HASHTABLE_VALUE(Descriptor, desc_php);
  425. desc->intern = intern;
  426. add_def_obj(msgdef, desc_php);
  427. add_ce_obj(intern->klass, desc_php);
  428. }
  429. #if PHP_MAJOR_VERSION < 7
  430. RETURN_ZVAL(desc_php, 1, 0);
  431. #else
  432. GC_ADDREF(desc_php);
  433. RETURN_OBJ(desc_php);
  434. #endif
  435. }
  436. // -----------------------------------------------------------------------------
  437. // Oneof
  438. // -----------------------------------------------------------------------------
  439. static zend_function_entry oneof_descriptor_methods[] = {
  440. PHP_ME(Oneof, getName, NULL, ZEND_ACC_PUBLIC)
  441. PHP_ME(Oneof, getField, NULL, ZEND_ACC_PUBLIC)
  442. PHP_ME(Oneof, getFieldCount, NULL, ZEND_ACC_PUBLIC)
  443. ZEND_FE_END
  444. };
  445. DEFINE_CLASS(Oneof, oneof_descriptor,
  446. "Google\\Protobuf\\OneofDescriptor");
  447. static void oneof_descriptor_free_c(Oneof *self TSRMLS_DC) {
  448. }
  449. static void oneof_descriptor_init_c_instance(Oneof *self TSRMLS_DC) {
  450. self->oneofdef = NULL;
  451. }
  452. PHP_METHOD(Oneof, getName) {
  453. Oneof *intern = UNBOX(Oneof, getThis());
  454. const char *name = upb_oneofdef_name(intern->oneofdef);
  455. PHP_PROTO_RETVAL_STRINGL(name, strlen(name), 1);
  456. }
  457. PHP_METHOD(Oneof, getField) {
  458. long index;
  459. if (zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC, "l", &index) ==
  460. FAILURE) {
  461. zend_error(E_USER_ERROR, "Expect integer for index.\n");
  462. return;
  463. }
  464. Oneof *intern = UNBOX(Oneof, getThis());
  465. int field_num = upb_oneofdef_numfields(intern->oneofdef);
  466. if (index < 0 || index >= field_num) {
  467. zend_error(E_USER_ERROR, "Cannot get element at %ld.\n", index);
  468. return;
  469. }
  470. upb_oneof_iter iter;
  471. int i;
  472. for(upb_oneof_begin(&iter, intern->oneofdef), i = 0;
  473. !upb_oneof_done(&iter) && i < index;
  474. upb_oneof_next(&iter), i++);
  475. const upb_fielddef *field = upb_oneof_iter_field(&iter);
  476. PHP_PROTO_HASHTABLE_VALUE field_hashtable_value = get_def_obj(field);
  477. if (field_hashtable_value == NULL) {
  478. #if PHP_MAJOR_VERSION < 7
  479. MAKE_STD_ZVAL(field_hashtable_value);
  480. ZVAL_OBJ(field_hashtable_value, field_descriptor_type->create_object(
  481. field_descriptor_type TSRMLS_CC));
  482. #else
  483. field_hashtable_value =
  484. field_descriptor_type->create_object(field_descriptor_type TSRMLS_CC);
  485. #endif
  486. FieldDescriptor *field_php =
  487. UNBOX_HASHTABLE_VALUE(FieldDescriptor, field_hashtable_value);
  488. field_php->fielddef = field;
  489. add_def_obj(field, field_hashtable_value);
  490. }
  491. #if PHP_MAJOR_VERSION < 7
  492. RETURN_ZVAL(field_hashtable_value, 1, 0);
  493. #else
  494. GC_ADDREF(field_hashtable_value);
  495. RETURN_OBJ(field_hashtable_value);
  496. #endif
  497. }
  498. PHP_METHOD(Oneof, getFieldCount) {
  499. Oneof *intern = UNBOX(Oneof, getThis());
  500. RETURN_LONG(upb_oneofdef_numfields(intern->oneofdef));
  501. }
  502. // -----------------------------------------------------------------------------
  503. // DescriptorPool
  504. // -----------------------------------------------------------------------------
  505. static zend_function_entry descriptor_pool_methods[] = {
  506. PHP_ME(DescriptorPool, getGeneratedPool, NULL,
  507. ZEND_ACC_PUBLIC|ZEND_ACC_STATIC)
  508. PHP_ME(DescriptorPool, getDescriptorByClassName, NULL, ZEND_ACC_PUBLIC)
  509. PHP_ME(DescriptorPool, getEnumDescriptorByClassName, NULL, ZEND_ACC_PUBLIC)
  510. ZEND_FE_END
  511. };
  512. static zend_function_entry internal_descriptor_pool_methods[] = {
  513. PHP_ME(InternalDescriptorPool, getGeneratedPool, NULL,
  514. ZEND_ACC_PUBLIC|ZEND_ACC_STATIC)
  515. PHP_ME(InternalDescriptorPool, internalAddGeneratedFile, NULL, ZEND_ACC_PUBLIC)
  516. ZEND_FE_END
  517. };
  518. DEFINE_CLASS(DescriptorPool, descriptor_pool,
  519. "Google\\Protobuf\\DescriptorPool");
  520. DEFINE_CLASS(InternalDescriptorPool, internal_descriptor_pool,
  521. "Google\\Protobuf\\Internal\\DescriptorPool");
  522. // wrapper of generated pool
  523. #if PHP_MAJOR_VERSION < 7
  524. zval* generated_pool_php;
  525. zval* internal_generated_pool_php;
  526. #else
  527. zend_object *generated_pool_php;
  528. zend_object *internal_generated_pool_php;
  529. #endif
  530. InternalDescriptorPoolImpl *generated_pool;
  531. InternalDescriptorPoolImpl generated_pool_impl; // The actual generated pool
  532. void init_generated_pool_once(TSRMLS_D) {
  533. if (generated_pool == NULL) {
  534. #if PHP_MAJOR_VERSION < 7
  535. MAKE_STD_ZVAL(generated_pool_php);
  536. MAKE_STD_ZVAL(internal_generated_pool_php);
  537. ZVAL_OBJ(internal_generated_pool_php,
  538. internal_descriptor_pool_type->create_object(
  539. internal_descriptor_pool_type TSRMLS_CC));
  540. ZVAL_OBJ(generated_pool_php, descriptor_pool_type->create_object(
  541. descriptor_pool_type TSRMLS_CC));
  542. #else
  543. internal_generated_pool_php = internal_descriptor_pool_type->create_object(
  544. internal_descriptor_pool_type TSRMLS_CC);
  545. generated_pool_php =
  546. descriptor_pool_type->create_object(descriptor_pool_type TSRMLS_CC);
  547. #endif
  548. generated_pool = &generated_pool_impl;
  549. }
  550. }
  551. static void internal_descriptor_pool_init_c_instance(
  552. InternalDescriptorPool *pool TSRMLS_DC) {
  553. pool->intern = &generated_pool_impl;
  554. }
  555. static void internal_descriptor_pool_free_c(
  556. InternalDescriptorPool *pool TSRMLS_DC) {
  557. }
  558. void internal_descriptor_pool_impl_init(
  559. InternalDescriptorPoolImpl *pool TSRMLS_DC) {
  560. pool->symtab = upb_symtab_new();
  561. pool->fill_handler_cache =
  562. upb_handlercache_new(add_handlers_for_message, NULL);
  563. pool->pb_serialize_handler_cache = upb_pb_encoder_newcache();
  564. pool->json_serialize_handler_cache = upb_json_printer_newcache(false);
  565. pool->json_serialize_handler_preserve_cache = upb_json_printer_newcache(true);
  566. pool->fill_method_cache = upb_pbcodecache_new(pool->fill_handler_cache);
  567. pool->json_fill_method_cache = upb_json_codecache_new();
  568. }
  569. void internal_descriptor_pool_impl_destroy(
  570. InternalDescriptorPoolImpl *pool TSRMLS_DC) {
  571. upb_symtab_free(pool->symtab);
  572. upb_handlercache_free(pool->fill_handler_cache);
  573. upb_handlercache_free(pool->pb_serialize_handler_cache);
  574. upb_handlercache_free(pool->json_serialize_handler_cache);
  575. upb_handlercache_free(pool->json_serialize_handler_preserve_cache);
  576. upb_pbcodecache_free(pool->fill_method_cache);
  577. upb_json_codecache_free(pool->json_fill_method_cache);
  578. }
  579. static void descriptor_pool_init_c_instance(DescriptorPool *pool TSRMLS_DC) {
  580. assert(generated_pool != NULL);
  581. pool->intern = generated_pool;
  582. }
  583. static void descriptor_pool_free_c(DescriptorPool *pool TSRMLS_DC) {
  584. }
  585. static void validate_enumdef(const upb_enumdef *enumdef) {
  586. // Verify that an entry exists with integer value 0. (This is the default
  587. // value.)
  588. const char *lookup = upb_enumdef_iton(enumdef, 0);
  589. if (lookup == NULL) {
  590. zend_error(E_USER_ERROR,
  591. "Enum definition does not contain a value for '0'.");
  592. }
  593. }
  594. static void validate_msgdef(const upb_msgdef* msgdef) {
  595. // Verify that no required fields exist. proto3 does not support these.
  596. upb_msg_field_iter it;
  597. for (upb_msg_field_begin(&it, msgdef);
  598. !upb_msg_field_done(&it);
  599. upb_msg_field_next(&it)) {
  600. const upb_fielddef* field = upb_msg_iter_field(&it);
  601. if (upb_fielddef_label(field) == UPB_LABEL_REQUIRED) {
  602. zend_error(E_ERROR, "Required fields are unsupported in proto3.");
  603. }
  604. }
  605. }
  606. PHP_METHOD(DescriptorPool, getGeneratedPool) {
  607. init_generated_pool_once(TSRMLS_C);
  608. #if PHP_MAJOR_VERSION < 7
  609. RETURN_ZVAL(generated_pool_php, 1, 0);
  610. #else
  611. GC_ADDREF(generated_pool_php);
  612. RETURN_OBJ(generated_pool_php);
  613. #endif
  614. }
  615. PHP_METHOD(InternalDescriptorPool, getGeneratedPool) {
  616. init_generated_pool_once(TSRMLS_C);
  617. #if PHP_MAJOR_VERSION < 7
  618. RETURN_ZVAL(internal_generated_pool_php, 1, 0);
  619. #else
  620. GC_ADDREF(internal_generated_pool_php);
  621. RETURN_OBJ(internal_generated_pool_php);
  622. #endif
  623. }
  624. static size_t classname_len_max(const char *fullname,
  625. const char *package,
  626. const char *php_namespace,
  627. const char *prefix) {
  628. size_t fullname_len = strlen(fullname);
  629. size_t package_len = 0;
  630. size_t prefix_len = 0;
  631. size_t namespace_len = 0;
  632. size_t length = fullname_len;
  633. int i, segment, classname_start = 0;
  634. if (package != NULL) {
  635. package_len = strlen(package);
  636. }
  637. if (prefix != NULL) {
  638. prefix_len = strlen(prefix);
  639. }
  640. if (php_namespace != NULL) {
  641. namespace_len = strlen(php_namespace);
  642. }
  643. // Process package
  644. if (package_len > 0) {
  645. segment = 1;
  646. for (i = 0; i < package_len; i++) {
  647. if (package[i] == '.') {
  648. segment++;
  649. }
  650. }
  651. // In case of reserved name in package.
  652. length += 3 * segment;
  653. classname_start = package_len + 1;
  654. }
  655. // Process class name
  656. segment = 1;
  657. for (i = classname_start; i < fullname_len; i++) {
  658. if (fullname[i] == '.') {
  659. segment++;
  660. }
  661. }
  662. if (prefix_len == 0) {
  663. length += 3 * segment;
  664. } else {
  665. length += prefix_len * segment;
  666. }
  667. // The additional 2, one is for preceding '.' and the other is for trailing 0.
  668. return length + namespace_len + 2;
  669. }
  670. static bool is_reserved(const char *segment, int length) {
  671. bool result;
  672. char* lower = ALLOC_N(char, length + 1);
  673. memset(lower, 0, length + 1);
  674. memcpy(lower, segment, length);
  675. int i = 0;
  676. while(lower[i]) {
  677. lower[i] = (char)tolower(lower[i]);
  678. i++;
  679. }
  680. lower[length] = 0;
  681. result = is_reserved_name(lower);
  682. FREE(lower);
  683. return result;
  684. }
  685. static void fill_prefix(const char *segment, int length,
  686. const char *prefix_given,
  687. const char *package_name,
  688. stringsink *classname) {
  689. size_t i;
  690. if (prefix_given != NULL && strcmp(prefix_given, "") != 0) {
  691. stringsink_string(classname, NULL, prefix_given,
  692. strlen(prefix_given), NULL);
  693. } else {
  694. if (is_reserved(segment, length)) {
  695. if (package_name != NULL &&
  696. strcmp("google.protobuf", package_name) == 0) {
  697. stringsink_string(classname, NULL, "GPB", 3, NULL);
  698. } else {
  699. stringsink_string(classname, NULL, "PB", 2, NULL);
  700. }
  701. }
  702. }
  703. }
  704. static void fill_segment(const char *segment, int length,
  705. stringsink *classname, bool use_camel) {
  706. if (use_camel && (segment[0] < 'A' || segment[0] > 'Z')) {
  707. char first = segment[0] + ('A' - 'a');
  708. stringsink_string(classname, NULL, &first, 1, NULL);
  709. stringsink_string(classname, NULL, segment + 1, length - 1, NULL);
  710. } else {
  711. stringsink_string(classname, NULL, segment, length, NULL);
  712. }
  713. }
  714. static void fill_namespace(const char *package, const char *php_namespace,
  715. stringsink *classname) {
  716. if (php_namespace != NULL) {
  717. if (strlen(php_namespace) != 0) {
  718. stringsink_string(classname, NULL, php_namespace, strlen(php_namespace),
  719. NULL);
  720. stringsink_string(classname, NULL, "\\", 1, NULL);
  721. }
  722. } else if (package != NULL) {
  723. int i = 0, j, offset = 0;
  724. size_t package_len = strlen(package);
  725. while (i < package_len) {
  726. j = i;
  727. while (j < package_len && package[j] != '.') {
  728. j++;
  729. }
  730. fill_prefix(package + i, j - i, "", package, classname);
  731. fill_segment(package + i, j - i, classname, true);
  732. stringsink_string(classname, NULL, "\\", 1, NULL);
  733. i = j + 1;
  734. }
  735. }
  736. }
  737. static void fill_classname(const char *fullname,
  738. const char *package,
  739. const char *prefix,
  740. stringsink *classname,
  741. bool use_nested_submsg) {
  742. int classname_start = 0;
  743. if (package != NULL) {
  744. size_t package_len = strlen(package);
  745. classname_start = package_len == 0 ? 0 : package_len + 1;
  746. }
  747. size_t fullname_len = strlen(fullname);
  748. bool is_first_segment = true;
  749. int i = classname_start, j;
  750. while (i < fullname_len) {
  751. j = i;
  752. while (j < fullname_len && fullname[j] != '.') {
  753. j++;
  754. }
  755. if (use_nested_submsg || is_first_segment && j == fullname_len) {
  756. fill_prefix(fullname + i, j - i, prefix, package, classname);
  757. }
  758. is_first_segment = false;
  759. fill_segment(fullname + i, j - i, classname, false);
  760. if (j != fullname_len) {
  761. if (use_nested_submsg) {
  762. stringsink_string(classname, NULL, "\\", 1, NULL);
  763. } else {
  764. stringsink_string(classname, NULL, "_", 1, NULL);
  765. }
  766. }
  767. i = j + 1;
  768. }
  769. }
  770. static void fill_classname_for_desc(void *desc, bool is_enum) {
  771. const upb_filedef *file;
  772. const char *fullname;
  773. bool use_nested_submsg;
  774. if (is_enum) {
  775. EnumDescriptorInternal* enumdesc = desc;
  776. file = upb_enumdef_file(enumdesc->enumdef);
  777. fullname = upb_enumdef_fullname(enumdesc->enumdef);
  778. use_nested_submsg = enumdesc->use_nested_submsg;
  779. } else {
  780. DescriptorInternal* msgdesc = desc;
  781. file = upb_msgdef_file(msgdesc->msgdef);
  782. fullname = upb_msgdef_fullname(msgdesc->msgdef);
  783. use_nested_submsg = msgdesc->use_nested_submsg;
  784. }
  785. // Prepend '.' to package name to make it absolute. In the 5 additional
  786. // bytes allocated, one for '.', one for trailing 0, and 3 for 'GPB' if
  787. // given message is google.protobuf.Empty.
  788. const char *package = upb_filedef_package(file);
  789. const char *php_namespace = upb_filedef_phpnamespace(file);
  790. const char *prefix = upb_filedef_phpprefix(file);
  791. size_t classname_len =
  792. classname_len_max(fullname, package, php_namespace, prefix);
  793. char* after_package;
  794. stringsink namesink;
  795. stringsink_init(&namesink);
  796. fill_namespace(package, php_namespace, &namesink);
  797. fill_classname(fullname, package, prefix, &namesink, use_nested_submsg);
  798. stringsink_string(&namesink, NULL, "\0", 1, NULL);
  799. if (is_enum) {
  800. EnumDescriptorInternal* enumdesc = desc;
  801. enumdesc->classname = strdup(namesink.ptr);
  802. } else {
  803. DescriptorInternal* msgdesc = desc;
  804. msgdesc->classname = strdup(namesink.ptr);
  805. }
  806. stringsink_uninit(&namesink);
  807. }
  808. void register_class(void *desc, bool is_enum TSRMLS_DC) {
  809. const char *classname;
  810. const char *fullname;
  811. zend_class_entry* ret;
  812. if (is_enum) {
  813. EnumDescriptorInternal* enumdesc = desc;
  814. if (enumdesc->klass) {
  815. return;
  816. }
  817. classname = enumdesc->classname;
  818. fullname = upb_enumdef_fullname(enumdesc->enumdef);
  819. } else {
  820. DescriptorInternal* msgdesc = desc;
  821. if (msgdesc->klass) {
  822. return;
  823. }
  824. if (!msgdesc->classname) {
  825. return;
  826. }
  827. classname = msgdesc->classname;
  828. fullname = upb_msgdef_fullname(msgdesc->msgdef);
  829. }
  830. PHP_PROTO_CE_DECLARE pce;
  831. if (php_proto_zend_lookup_class(classname, strlen(classname), &pce) ==
  832. FAILURE) {
  833. zend_error(
  834. E_ERROR,
  835. "Generated message class %s hasn't been defined (%s)",
  836. classname);
  837. return;
  838. }
  839. ret = PHP_PROTO_CE_UNREF(pce);
  840. if (is_enum) {
  841. EnumDescriptorInternal* enumdesc = desc;
  842. add_ce_enumdesc(ret, desc);
  843. add_proto_enumdesc(fullname, desc);
  844. enumdesc->klass = ret;
  845. } else {
  846. DescriptorInternal* msgdesc = desc;
  847. add_ce_desc(ret, desc);
  848. msgdesc->klass = ret;
  849. // Map entries don't have existing php class.
  850. if (!upb_msgdef_mapentry(msgdesc->msgdef)) {
  851. if (msgdesc->layout == NULL) {
  852. MessageLayout* layout = create_layout(msgdesc->msgdef);
  853. msgdesc->layout = layout;
  854. }
  855. }
  856. }
  857. }
  858. bool depends_on_descriptor(const google_protobuf_FileDescriptorProto* file) {
  859. const upb_strview *deps;
  860. upb_strview name = upb_strview_makez("google/protobuf/descriptor.proto");
  861. size_t i, n;
  862. deps = google_protobuf_FileDescriptorProto_dependency(file, &n);
  863. for (i = 0; i < n; i++) {
  864. if (upb_strview_eql(deps[i], name)) {
  865. return true;
  866. }
  867. }
  868. return false;
  869. }
  870. const upb_filedef *parse_and_add_descriptor(const char *data,
  871. PHP_PROTO_SIZE data_len,
  872. InternalDescriptorPoolImpl *pool,
  873. upb_arena *arena) {
  874. size_t n;
  875. google_protobuf_FileDescriptorSet *set;
  876. const google_protobuf_FileDescriptorProto* const* files;
  877. const upb_filedef* file;
  878. upb_status status;
  879. set = google_protobuf_FileDescriptorSet_parse(
  880. data, data_len, arena);
  881. if (!set) {
  882. zend_error(E_ERROR, "Failed to parse binary descriptor\n");
  883. return NULL;
  884. }
  885. files = google_protobuf_FileDescriptorSet_file(set, &n);
  886. if (n != 1) {
  887. zend_error(E_ERROR, "Serialized descriptors should have exactly one file");
  888. return NULL;
  889. }
  890. // Check whether file has already been added.
  891. upb_strview name = google_protobuf_FileDescriptorProto_name(files[0]);
  892. // TODO(teboring): Needs another look up method which takes data and length.
  893. file = upb_symtab_lookupfile2(pool->symtab, name.data, name.size);
  894. if (file != NULL) {
  895. return NULL;
  896. }
  897. // The PHP code generator currently special-cases descriptor.proto. It
  898. // doesn't add it as a dependency even if the proto file actually does
  899. // depend on it.
  900. if (depends_on_descriptor(files[0]) &&
  901. upb_symtab_lookupfile(pool->symtab, "google/protobuf/descriptor.proto") ==
  902. NULL) {
  903. if (!parse_and_add_descriptor((char *)descriptor_proto,
  904. descriptor_proto_len, pool, arena)) {
  905. return NULL;
  906. }
  907. }
  908. upb_status_clear(&status);
  909. file = upb_symtab_addfile(pool->symtab, files[0], &status);
  910. check_upb_status(&status, "Unable to load descriptor");
  911. return file;
  912. }
  913. void internal_add_generated_file(const char *data, PHP_PROTO_SIZE data_len,
  914. InternalDescriptorPoolImpl *pool,
  915. bool use_nested_submsg TSRMLS_DC) {
  916. int i;
  917. upb_arena *arena;
  918. const upb_filedef* file;
  919. arena = upb_arena_new();
  920. file = parse_and_add_descriptor(data, data_len, pool, arena);
  921. upb_arena_free(arena);
  922. if (!file) return;
  923. // For each enum/message, we need its PHP class, upb descriptor and its PHP
  924. // wrapper. These information are needed later for encoding, decoding and type
  925. // checking. However, sometimes we just have one of them. In order to find
  926. // them quickly, here, we store the mapping for them.
  927. for (i = 0; i < upb_filedef_msgcount(file); i++) {
  928. const upb_msgdef *msgdef = upb_filedef_msg(file, i);
  929. CREATE_HASHTABLE_VALUE(desc, desc_php, Descriptor, descriptor_type);
  930. desc->intern = SYS_MALLOC(DescriptorInternal);
  931. desc->intern->msgdef = msgdef;
  932. desc->intern->pool = pool;
  933. desc->intern->layout = NULL;
  934. desc->intern->klass = NULL;
  935. desc->intern->use_nested_submsg = use_nested_submsg;
  936. desc->intern->classname = NULL;
  937. add_def_obj(desc->intern->msgdef, desc_php);
  938. add_msgdef_desc(desc->intern->msgdef, desc->intern);
  939. // Unlike other messages, MapEntry is shared by all map fields and doesn't
  940. // have generated PHP class.
  941. if (upb_msgdef_mapentry(msgdef)) {
  942. continue;
  943. }
  944. fill_classname_for_desc(desc->intern, false);
  945. add_class_desc(desc->intern->classname, desc->intern);
  946. add_proto_desc(upb_msgdef_fullname(desc->intern->msgdef), desc->intern);
  947. }
  948. for (i = 0; i < upb_filedef_enumcount(file); i++) {
  949. const upb_enumdef *enumdef = upb_filedef_enum(file, i);
  950. CREATE_HASHTABLE_VALUE(desc, desc_php, EnumDescriptor, enum_descriptor_type);
  951. desc->intern = SYS_MALLOC(EnumDescriptorInternal);
  952. desc->intern->enumdef = enumdef;
  953. desc->intern->klass = NULL;
  954. desc->intern->use_nested_submsg = use_nested_submsg;
  955. desc->intern->classname = NULL;
  956. add_def_obj(desc->intern->enumdef, desc_php);
  957. add_enumdef_enumdesc(desc->intern->enumdef, desc->intern);
  958. fill_classname_for_desc(desc->intern, true);
  959. add_class_enumdesc(desc->intern->classname, desc->intern);
  960. }
  961. }
  962. PHP_METHOD(InternalDescriptorPool, internalAddGeneratedFile) {
  963. char *data = NULL;
  964. PHP_PROTO_SIZE data_len;
  965. upb_filedef **files;
  966. zend_bool use_nested_submsg = false;
  967. size_t i;
  968. if (zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC, "s|b",
  969. &data, &data_len, &use_nested_submsg) ==
  970. FAILURE) {
  971. return;
  972. }
  973. InternalDescriptorPool *pool = UNBOX(InternalDescriptorPool, getThis());
  974. internal_add_generated_file(data, data_len, pool->intern,
  975. use_nested_submsg TSRMLS_CC);
  976. }
  977. PHP_METHOD(DescriptorPool, getDescriptorByClassName) {
  978. char *classname = NULL;
  979. PHP_PROTO_SIZE classname_len;
  980. if (zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC, "s", &classname,
  981. &classname_len) == FAILURE) {
  982. return;
  983. }
  984. PHP_PROTO_CE_DECLARE pce;
  985. if (php_proto_zend_lookup_class(classname, classname_len, &pce) ==
  986. FAILURE) {
  987. RETURN_NULL();
  988. }
  989. PHP_PROTO_HASHTABLE_VALUE desc_php = get_ce_obj(PHP_PROTO_CE_UNREF(pce));
  990. if (desc_php == NULL) {
  991. DescriptorInternal* intern = get_ce_desc(PHP_PROTO_CE_UNREF(pce));
  992. if (intern == NULL) {
  993. RETURN_NULL();
  994. }
  995. #if PHP_MAJOR_VERSION < 7
  996. MAKE_STD_ZVAL(desc_php);
  997. ZVAL_OBJ(desc_php, descriptor_type->create_object(
  998. descriptor_type TSRMLS_CC));
  999. Z_DELREF_P(desc_php);
  1000. #else
  1001. desc_php =
  1002. descriptor_type->create_object(descriptor_type TSRMLS_CC);
  1003. GC_DELREF(desc_php);
  1004. #endif
  1005. Descriptor* desc = UNBOX_HASHTABLE_VALUE(Descriptor, desc_php);
  1006. desc->intern = intern;
  1007. add_def_obj(intern->msgdef, desc_php);
  1008. add_ce_obj(PHP_PROTO_CE_UNREF(pce), desc_php);
  1009. }
  1010. zend_class_entry* instance_ce = HASHTABLE_VALUE_CE(desc_php);
  1011. if (!instanceof_function(instance_ce, descriptor_type TSRMLS_CC)) {
  1012. RETURN_NULL();
  1013. }
  1014. #if PHP_MAJOR_VERSION < 7
  1015. RETURN_ZVAL(desc_php, 1, 0);
  1016. #else
  1017. GC_ADDREF(desc_php);
  1018. RETURN_OBJ(desc_php);
  1019. #endif
  1020. }
  1021. PHP_METHOD(DescriptorPool, getEnumDescriptorByClassName) {
  1022. char *classname = NULL;
  1023. PHP_PROTO_SIZE classname_len;
  1024. if (zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC, "s", &classname,
  1025. &classname_len) == FAILURE) {
  1026. return;
  1027. }
  1028. PHP_PROTO_CE_DECLARE pce;
  1029. if (php_proto_zend_lookup_class(classname, classname_len, &pce) ==
  1030. FAILURE) {
  1031. RETURN_NULL();
  1032. }
  1033. zend_class_entry* ce = PHP_PROTO_CE_UNREF(pce);
  1034. PHP_PROTO_HASHTABLE_VALUE desc_php = get_ce_obj(ce);
  1035. if (desc_php == NULL) {
  1036. #if PHP_MAJOR_VERSION < 7
  1037. EnumDescriptorInternal* intern = get_class_enumdesc(ce->name);
  1038. #else
  1039. EnumDescriptorInternal* intern = get_class_enumdesc(ZSTR_VAL(ce->name));
  1040. #endif
  1041. register_class(intern, true TSRMLS_CC);
  1042. if (intern == NULL) {
  1043. RETURN_NULL();
  1044. }
  1045. #if PHP_MAJOR_VERSION < 7
  1046. MAKE_STD_ZVAL(desc_php);
  1047. ZVAL_OBJ(desc_php, enum_descriptor_type->create_object(
  1048. enum_descriptor_type TSRMLS_CC));
  1049. Z_DELREF_P(desc_php);
  1050. #else
  1051. desc_php =
  1052. enum_descriptor_type->create_object(enum_descriptor_type TSRMLS_CC);
  1053. GC_DELREF(desc_php);
  1054. #endif
  1055. EnumDescriptor* desc = UNBOX_HASHTABLE_VALUE(EnumDescriptor, desc_php);
  1056. desc->intern = intern;
  1057. add_def_obj(intern->enumdef, desc_php);
  1058. add_ce_obj(ce, desc_php);
  1059. }
  1060. zend_class_entry* instance_ce = HASHTABLE_VALUE_CE(desc_php);
  1061. if (!instanceof_function(instance_ce, enum_descriptor_type TSRMLS_CC)) {
  1062. RETURN_NULL();
  1063. }
  1064. #if PHP_MAJOR_VERSION < 7
  1065. RETURN_ZVAL(desc_php, 1, 0);
  1066. #else
  1067. GC_ADDREF(desc_php);
  1068. RETURN_OBJ(desc_php);
  1069. #endif
  1070. }