diff options
Diffstat (limited to 'php/ext/google/protobuf/def.c')
-rw-r--r-- | php/ext/google/protobuf/def.c | 381 |
1 files changed, 381 insertions, 0 deletions
diff --git a/php/ext/google/protobuf/def.c b/php/ext/google/protobuf/def.c new file mode 100644 index 00000000..fc188069 --- /dev/null +++ b/php/ext/google/protobuf/def.c @@ -0,0 +1,381 @@ +#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); +} |