| #include "protobuf.h" |
| |
| // ----------------------------------------------------------------------------- |
| // Common Utilities |
| // ----------------------------------------------------------------------------- |
| |
| void check_upb_status(const upb_status* status, const char* msg) { |
| if (!upb_ok(status)) { |
| zend_error("%s: %s\n", msg, upb_status_errmsg(status)); |
| } |
| } |
| |
| |
| static upb_def *check_notfrozen(const upb_def *def) { |
| if (upb_def_isfrozen(def)) { |
| zend_error(E_ERROR, |
| "Attempt to modify a frozen descriptor. Once descriptors are " |
| "added to the descriptor pool, they may not be modified."); |
| } |
| return (upb_def *)def; |
| } |
| |
| static upb_msgdef *check_msgdef_notfrozen(const upb_msgdef *def) { |
| return upb_downcast_msgdef_mutable(check_notfrozen((const upb_def *)def)); |
| } |
| |
| static upb_fielddef *check_fielddef_notfrozen(const upb_fielddef *def) { |
| return upb_downcast_fielddef_mutable(check_notfrozen((const upb_def *)def)); |
| } |
| |
| #define PROTOBUF_WRAP_INTERN(wrapper, intern, intern_dtor) \ |
| Z_TYPE_P(wrapper) = IS_OBJECT; \ |
| Z_OBJVAL_P(wrapper) \ |
| .handle = zend_objects_store_put( \ |
| intern, (zend_objects_store_dtor_t)zend_objects_destroy_object, \ |
| intern_dtor, NULL TSRMLS_CC); \ |
| Z_OBJVAL_P(wrapper).handlers = zend_get_std_object_handlers(); |
| |
| #define PROTOBUF_SETUP_ZEND_WRAPPER(class_name, class_name_lower, wrapper, \ |
| intern) \ |
| Z_TYPE_P(wrapper) = IS_OBJECT; \ |
| class_name *intern = ALLOC(class_name); \ |
| memset(intern, 0, sizeof(class_name)); \ |
| class_name_lower##_init_c_instance(intern TSRMLS_CC); \ |
| Z_OBJVAL_P(wrapper) \ |
| .handle = zend_objects_store_put(intern, NULL, class_name_lower##_free, \ |
| NULL TSRMLS_CC); \ |
| Z_OBJVAL_P(wrapper).handlers = zend_get_std_object_handlers(); |
| |
| #define PROTOBUF_CREATE_ZEND_WRAPPER(class_name, class_name_lower, wrapper, \ |
| intern) \ |
| MAKE_STD_ZVAL(wrapper); \ |
| PROTOBUF_SETUP_ZEND_WRAPPER(class_name, class_name_lower, wrapper, intern); |
| |
| #define DEFINE_CLASS(name, name_lower, string_name) \ |
| zend_class_entry *name_lower##_type; \ |
| void name_lower##_init(TSRMLS_D) { \ |
| zend_class_entry class_type; \ |
| INIT_CLASS_ENTRY(class_type, string_name, name_lower##_methods); \ |
| name_lower##_type = zend_register_internal_class(&class_type TSRMLS_CC); \ |
| name_lower##_type->create_object = name_lower##_create; \ |
| } \ |
| name *php_to_##name_lower(zval *val TSRMLS_DC) { \ |
| return (name *)zend_object_store_get_object(val TSRMLS_CC); \ |
| } \ |
| void name_lower##_free(void *object TSRMLS_DC) { \ |
| name *intern = (name *)object; \ |
| name_lower##_free_c(intern TSRMLS_CC); \ |
| efree(object); \ |
| } \ |
| zend_object_value name_lower##_create(zend_class_entry *ce TSRMLS_DC) { \ |
| zend_object_value return_value; \ |
| name *intern = (name *)emalloc(sizeof(name)); \ |
| memset(intern, 0, sizeof(name)); \ |
| name_lower##_init_c_instance(intern TSRMLS_CC); \ |
| return_value.handle = zend_objects_store_put( \ |
| intern, (zend_objects_store_dtor_t)zend_objects_destroy_object, \ |
| name_lower##_free, NULL TSRMLS_CC); \ |
| return_value.handlers = zend_get_std_object_handlers(); \ |
| return return_value; \ |
| } |
| |
| // ----------------------------------------------------------------------------- |
| // DescriptorPool |
| // ----------------------------------------------------------------------------- |
| |
| static zend_function_entry descriptor_pool_methods[] = { |
| PHP_ME(DescriptorPool, addMessage, NULL, ZEND_ACC_PUBLIC) |
| PHP_ME(DescriptorPool, finalize, NULL, ZEND_ACC_PUBLIC) |
| ZEND_FE_END |
| }; |
| |
| DEFINE_CLASS(DescriptorPool, descriptor_pool, |
| "Google\\Protobuf\\DescriptorPool"); |
| |
| DescriptorPool *generated_pool; // The actual generated pool |
| |
| ZEND_FUNCTION(get_generated_pool) { |
| if (PROTOBUF_G(generated_pool) == NULL) { |
| MAKE_STD_ZVAL(PROTOBUF_G(generated_pool)); |
| Z_TYPE_P(PROTOBUF_G(generated_pool)) = IS_OBJECT; |
| generated_pool = ALLOC(DescriptorPool); |
| descriptor_pool_init_c_instance(generated_pool TSRMLS_CC); |
| Z_OBJ_HANDLE_P(PROTOBUF_G(generated_pool)) = zend_objects_store_put( |
| generated_pool, NULL, |
| (zend_objects_free_object_storage_t)descriptor_pool_free, NULL TSRMLS_CC); |
| Z_OBJ_HT_P(PROTOBUF_G(generated_pool)) = zend_get_std_object_handlers(); |
| } |
| RETURN_ZVAL(PROTOBUF_G(generated_pool), 1, 0); |
| } |
| |
| void descriptor_pool_init_c_instance(DescriptorPool* pool TSRMLS_DC) { |
| zend_object_std_init(&pool->std, descriptor_pool_type TSRMLS_CC); |
| pool->symtab = upb_symtab_new(&pool->symtab); |
| |
| ALLOC_HASHTABLE(pool->pending_list); |
| zend_hash_init(pool->pending_list, 1, NULL, ZVAL_PTR_DTOR, 0); |
| } |
| |
| void descriptor_pool_free_c(DescriptorPool *pool TSRMLS_DC) { |
| upb_symtab_unref(pool->symtab, &pool->symtab); |
| zend_hash_destroy(pool->pending_list); |
| FREE_HASHTABLE(pool->pending_list); |
| } |
| |
| PHP_METHOD(DescriptorPool, addMessage) { |
| char *name = NULL; |
| int str_len; |
| if (zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC, "s", &name, &str_len) == |
| FAILURE) { |
| return; |
| } |
| |
| zval* retval = NULL; |
| PROTOBUF_CREATE_ZEND_WRAPPER(MessageBuilderContext, message_builder_context, |
| retval, context); |
| |
| MAKE_STD_ZVAL(context->pool); |
| ZVAL_ZVAL(context->pool, getThis(), 1, 0); |
| |
| Descriptor *desc = php_to_descriptor(context->descriptor TSRMLS_CC); |
| Descriptor_name_set(desc, name); |
| |
| RETURN_ZVAL(retval, 0, 1); |
| } |
| |
| static void validate_msgdef(const upb_msgdef* msgdef) { |
| // Verify that no required fields exist. proto3 does not support these. |
| upb_msg_field_iter it; |
| for (upb_msg_field_begin(&it, msgdef); |
| !upb_msg_field_done(&it); |
| upb_msg_field_next(&it)) { |
| const upb_fielddef* field = upb_msg_iter_field(&it); |
| if (upb_fielddef_label(field) == UPB_LABEL_REQUIRED) { |
| zend_error(E_ERROR, "Required fields are unsupported in proto3."); |
| } |
| } |
| } |
| |
| PHP_METHOD(DescriptorPool, finalize) { |
| DescriptorPool *self = php_to_descriptor_pool(getThis() TSRMLS_CC); |
| Bucket *temp; |
| int i, num; |
| |
| num = zend_hash_num_elements(self->pending_list); |
| upb_def **defs = emalloc(sizeof(upb_def *) * num); |
| |
| for (i = 0, temp = self->pending_list->pListHead; temp != NULL; |
| temp = temp->pListNext) { |
| zval *def_php = *(zval **)temp->pData; |
| Descriptor* desc = php_to_descriptor(def_php TSRMLS_CC); |
| defs[i] = (upb_def *)desc->msgdef; |
| validate_msgdef((const upb_msgdef *)defs[i++]); |
| } |
| |
| CHECK_UPB(upb_symtab_add(self->symtab, (upb_def **)defs, num, NULL, &status), |
| "Unable to add defs to DescriptorPool"); |
| |
| for (temp = self->pending_list->pListHead; temp != NULL; |
| temp = temp->pListNext) { |
| // zval *def_php = *(zval **)temp->pData; |
| // Descriptor* desc = php_to_descriptor(def_php TSRMLS_CC); |
| build_class_from_descriptor((zval *)temp->pDataPtr TSRMLS_CC); |
| } |
| |
| FREE(defs); |
| zend_hash_destroy(self->pending_list); |
| zend_hash_init(self->pending_list, 1, NULL, ZVAL_PTR_DTOR, 0); |
| } |
| |
| // ----------------------------------------------------------------------------- |
| // Descriptor |
| // ----------------------------------------------------------------------------- |
| |
| static zend_function_entry descriptor_methods[] = { |
| ZEND_FE_END |
| }; |
| |
| DEFINE_CLASS(Descriptor, descriptor, "Google\\Protobuf\\Descriptor"); |
| |
| void descriptor_free_c(Descriptor *self TSRMLS_DC) { |
| upb_msg_field_iter iter; |
| upb_msg_field_begin(&iter, self->msgdef); |
| while (!upb_msg_field_done(&iter)) { |
| upb_fielddef *fielddef = upb_msg_iter_field(&iter); |
| upb_fielddef_unref(fielddef, &fielddef); |
| upb_msg_field_next(&iter); |
| } |
| upb_msgdef_unref(self->msgdef, &self->msgdef); |
| if (self->layout) { |
| free_layout(self->layout); |
| } |
| } |
| |
| static void descriptor_add_field(Descriptor *desc, |
| const upb_fielddef *fielddef) { |
| upb_msgdef *mut_def = check_msgdef_notfrozen(desc->msgdef); |
| upb_fielddef *mut_field_def = check_fielddef_notfrozen(fielddef); |
| CHECK_UPB(upb_msgdef_addfield(mut_def, mut_field_def, NULL, &status), |
| "Adding field to Descriptor failed"); |
| // add_def_obj(fielddef, obj); |
| } |
| |
| void descriptor_init_c_instance(Descriptor* desc TSRMLS_DC) { |
| zend_object_std_init(&desc->std, descriptor_type TSRMLS_CC); |
| desc->msgdef = upb_msgdef_new(&desc->msgdef); |
| desc->layout = NULL; |
| // MAKE_STD_ZVAL(intern->klass); |
| // ZVAL_NULL(intern->klass); |
| desc->pb_serialize_handlers = NULL; |
| } |
| |
| void Descriptor_name_set(Descriptor *desc, const char *name) { |
| upb_msgdef *mut_def = check_msgdef_notfrozen(desc->msgdef); |
| CHECK_UPB(upb_msgdef_setfullname(mut_def, name, &status), |
| "Error setting Descriptor name"); |
| } |
| |
| // ----------------------------------------------------------------------------- |
| // FieldDescriptor |
| // ----------------------------------------------------------------------------- |
| |
| static void field_descriptor_name_set(const upb_fielddef* fielddef, |
| const char *name) { |
| upb_fielddef *mut_def = check_fielddef_notfrozen(fielddef); |
| CHECK_UPB(upb_fielddef_setname(mut_def, name, &status), |
| "Error setting FieldDescriptor name"); |
| } |
| |
| static void field_descriptor_label_set(const upb_fielddef* fielddef, |
| upb_label_t upb_label) { |
| upb_fielddef *mut_def = check_fielddef_notfrozen(fielddef); |
| upb_fielddef_setlabel(mut_def, upb_label); |
| } |
| |
| upb_fieldtype_t string_to_descriptortype(const char *type) { |
| #define CONVERT(upb, str) \ |
| if (!strcmp(type, str)) { \ |
| return UPB_DESCRIPTOR_TYPE_##upb; \ |
| } |
| |
| CONVERT(FLOAT, "float"); |
| CONVERT(DOUBLE, "double"); |
| CONVERT(BOOL, "bool"); |
| CONVERT(STRING, "string"); |
| CONVERT(BYTES, "bytes"); |
| CONVERT(MESSAGE, "message"); |
| CONVERT(GROUP, "group"); |
| CONVERT(ENUM, "enum"); |
| CONVERT(INT32, "int32"); |
| CONVERT(INT64, "int64"); |
| CONVERT(UINT32, "uint32"); |
| CONVERT(UINT64, "uint64"); |
| CONVERT(SINT32, "sint32"); |
| CONVERT(SINT64, "sint64"); |
| CONVERT(FIXED32, "fixed32"); |
| CONVERT(FIXED64, "fixed64"); |
| CONVERT(SFIXED32, "sfixed32"); |
| CONVERT(SFIXED64, "sfixed64"); |
| |
| #undef CONVERT |
| |
| zend_error(E_ERROR, "Unknown field type."); |
| return 0; |
| } |
| |
| static void field_descriptor_type_set(const upb_fielddef* fielddef, |
| const char *type) { |
| upb_fielddef *mut_def = check_fielddef_notfrozen(fielddef); |
| upb_fielddef_setdescriptortype(mut_def, string_to_descriptortype(type)); |
| } |
| |
| static void field_descriptor_number_set(const upb_fielddef* fielddef, |
| int number) { |
| upb_fielddef *mut_def = check_fielddef_notfrozen(fielddef); |
| CHECK_UPB(upb_fielddef_setnumber(mut_def, number, &status), |
| "Error setting field number"); |
| } |
| |
| // ----------------------------------------------------------------------------- |
| // MessageBuilderContext |
| // ----------------------------------------------------------------------------- |
| |
| static zend_function_entry message_builder_context_methods[] = { |
| PHP_ME(MessageBuilderContext, finalizeToPool, NULL, ZEND_ACC_PUBLIC) |
| PHP_ME(MessageBuilderContext, optional, NULL, ZEND_ACC_PUBLIC) |
| {NULL, NULL, NULL} |
| }; |
| |
| DEFINE_CLASS(MessageBuilderContext, message_builder_context, |
| "Google\\Protobuf\\Internal\\MessageBuilderContext"); |
| |
| void message_builder_context_free_c(MessageBuilderContext *context TSRMLS_DC) { |
| zval_ptr_dtor(&context->descriptor); |
| zval_ptr_dtor(&context->pool); |
| } |
| |
| void message_builder_context_init_c_instance( |
| MessageBuilderContext *context TSRMLS_DC) { |
| zend_object_std_init(&context->std, message_builder_context_type TSRMLS_CC); |
| PROTOBUF_CREATE_ZEND_WRAPPER(Descriptor, descriptor, context->descriptor, |
| desc); |
| } |
| |
| static void msgdef_add_field(Descriptor *desc, upb_label_t upb_label, |
| const char *name, const char *type, int number, |
| const char *type_class) { |
| upb_fielddef *fielddef = upb_fielddef_new(&fielddef); |
| upb_fielddef_setpacked(fielddef, false); |
| |
| field_descriptor_label_set(fielddef, upb_label); |
| field_descriptor_name_set(fielddef, name); |
| field_descriptor_type_set(fielddef, type); |
| field_descriptor_number_set(fielddef, number); |
| |
| // // if (type_class != Qnil) { |
| // // if (TYPE(type_class) != T_STRING) { |
| // // rb_raise(rb_eArgError, "Expected string for type class"); |
| // // } |
| // // // Make it an absolute type name by prepending a dot. |
| // // type_class = rb_str_append(rb_str_new2("."), type_class); |
| // // rb_funcall(fielddef, rb_intern("submsg_name="), 1, type_class); |
| // // } |
| descriptor_add_field(desc, fielddef); |
| } |
| |
| PHP_METHOD(MessageBuilderContext, optional) { |
| MessageBuilderContext *self = php_to_message_builder_context(getThis() TSRMLS_CC); |
| Descriptor *desc = php_to_descriptor(self->descriptor TSRMLS_CC); |
| // VALUE name, type, number, type_class; |
| const char *name, *type, *type_class; |
| int number, name_str_len, type_str_len, type_class_str_len; |
| if (ZEND_NUM_ARGS() == 3) { |
| if (zend_parse_parameters(3 TSRMLS_CC, "ssl", &name, |
| &name_str_len, &type, &type_str_len, &number) == FAILURE) { |
| return; |
| } |
| } else { |
| if (zend_parse_parameters(4 TSRMLS_CC, "ssls", &name, |
| &name_str_len, &type, &type_str_len, &number, &type_class, |
| &type_class_str_len) == FAILURE) { |
| return; |
| } |
| } |
| |
| msgdef_add_field(desc, UPB_LABEL_OPTIONAL, name, type, number, type_class); |
| |
| zval_copy_ctor(getThis()); |
| RETURN_ZVAL(getThis(), 1, 0); |
| } |
| |
| PHP_METHOD(MessageBuilderContext, finalizeToPool) { |
| MessageBuilderContext *self = php_to_message_builder_context(getThis() TSRMLS_CC); |
| DescriptorPool *pool = php_to_descriptor_pool(self->pool TSRMLS_CC); |
| Descriptor* desc = php_to_descriptor(self->descriptor TSRMLS_CC); |
| |
| Z_ADDREF_P(self->descriptor); |
| zend_hash_next_index_insert(pool->pending_list, &self->descriptor, |
| sizeof(zval *), NULL); |
| RETURN_ZVAL(self->pool, 1, 0); |
| } |