Index: third_party/protobuf/php/ext/google/protobuf/message.c |
diff --git a/third_party/protobuf/php/ext/google/protobuf/message.c b/third_party/protobuf/php/ext/google/protobuf/message.c |
new file mode 100644 |
index 0000000000000000000000000000000000000000..c062d665ab32670534d7b0d5a3c5faaa3ea92268 |
--- /dev/null |
+++ b/third_party/protobuf/php/ext/google/protobuf/message.c |
@@ -0,0 +1,273 @@ |
+// Protocol Buffers - Google's data interchange format |
+// Copyright 2014 Google Inc. All rights reserved. |
+// https://developers.google.com/protocol-buffers/ |
+// |
+// Redistribution and use in source and binary forms, with or without |
+// modification, are permitted provided that the following conditions are |
+// met: |
+// |
+// * Redistributions of source code must retain the above copyright |
+// notice, this list of conditions and the following disclaimer. |
+// * Redistributions in binary form must reproduce the above |
+// copyright notice, this list of conditions and the following disclaimer |
+// in the documentation and/or other materials provided with the |
+// distribution. |
+// * Neither the name of Google Inc. nor the names of its |
+// contributors may be used to endorse or promote products derived from |
+// this software without specific prior written permission. |
+// |
+// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
+// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
+// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
+// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
+// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
+// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
+// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
+// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
+// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
+// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
+// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
+ |
+#include <php.h> |
+ |
+#include "protobuf.h" |
+ |
+// ----------------------------------------------------------------------------- |
+// Class/module creation from msgdefs and enumdefs, respectively. |
+// ----------------------------------------------------------------------------- |
+ |
+void* message_data(void* msg) { |
+ return ((uint8_t *)msg) + sizeof(MessageHeader); |
+} |
+ |
+void message_set_property(zval* object, zval* field_name, zval* value, |
+ const zend_literal* key TSRMLS_DC) { |
+ const upb_fielddef* field; |
+ |
+ MessageHeader* self = zend_object_store_get_object(object TSRMLS_CC); |
+ |
+ CHECK_TYPE(field_name, IS_STRING); |
+ field = upb_msgdef_ntofz(self->descriptor->msgdef, Z_STRVAL_P(field_name)); |
+ if (field == NULL) { |
+ zend_error(E_ERROR, "Unknown field: %s", Z_STRVAL_P(field_name)); |
+ } |
+ layout_set(self->descriptor->layout, message_data(self), field, value); |
+} |
+ |
+zval* message_get_property(zval* object, zval* member, int type, |
+ const zend_literal* key TSRMLS_DC) { |
+ MessageHeader* self = |
+ (MessageHeader*)zend_object_store_get_object(object TSRMLS_CC); |
+ CHECK_TYPE(member, IS_STRING); |
+ |
+ const upb_fielddef* field; |
+ field = upb_msgdef_ntofz(self->descriptor->msgdef, Z_STRVAL_P(member)); |
+ if (field == NULL) { |
+ return EG(uninitialized_zval_ptr); |
+ } |
+ zval* retval = layout_get(self->descriptor->layout, message_data(self), field TSRMLS_CC); |
+ return retval; |
+} |
+ |
+static zend_function_entry message_methods[] = { |
+ PHP_ME(Message, encode, NULL, ZEND_ACC_PUBLIC) |
+ {NULL, NULL, NULL} |
+}; |
+ |
+/* stringsink *****************************************************************/ |
+ |
+// This should probably be factored into a common upb component. |
+ |
+typedef struct { |
+ upb_byteshandler handler; |
+ upb_bytessink sink; |
+ char *ptr; |
+ size_t len, size; |
+} stringsink; |
+ |
+static void *stringsink_start(void *_sink, const void *hd, size_t size_hint) { |
+ stringsink *sink = _sink; |
+ sink->len = 0; |
+ return sink; |
+} |
+ |
+static size_t stringsink_string(void *_sink, const void *hd, const char *ptr, |
+ size_t len, const upb_bufhandle *handle) { |
+ stringsink *sink = _sink; |
+ size_t new_size = sink->size; |
+ |
+ UPB_UNUSED(hd); |
+ UPB_UNUSED(handle); |
+ |
+ while (sink->len + len > new_size) { |
+ new_size *= 2; |
+ } |
+ |
+ if (new_size != sink->size) { |
+ sink->ptr = realloc(sink->ptr, new_size); |
+ sink->size = new_size; |
+ } |
+ |
+ memcpy(sink->ptr + sink->len, ptr, len); |
+ sink->len += len; |
+ |
+ return len; |
+} |
+ |
+void stringsink_init(stringsink *sink) { |
+ upb_byteshandler_init(&sink->handler); |
+ upb_byteshandler_setstartstr(&sink->handler, stringsink_start, NULL); |
+ upb_byteshandler_setstring(&sink->handler, stringsink_string, NULL); |
+ |
+ upb_bytessink_reset(&sink->sink, &sink->handler, sink); |
+ |
+ sink->size = 32; |
+ sink->ptr = malloc(sink->size); |
+ sink->len = 0; |
+} |
+ |
+void stringsink_uninit(stringsink *sink) { free(sink->ptr); } |
+ |
+// Stack-allocated context during an encode/decode operation. Contains the upb |
+// environment and its stack-based allocator, an initial buffer for allocations |
+// to avoid malloc() when possible, and a template for PHP exception messages |
+// if any error occurs. |
+#define STACK_ENV_STACKBYTES 4096 |
+typedef struct { |
+ upb_env env; |
+ upb_seededalloc alloc; |
+ const char *php_error_template; |
+ char allocbuf[STACK_ENV_STACKBYTES]; |
+} stackenv; |
+ |
+static void stackenv_init(stackenv* se, const char* errmsg); |
+static void stackenv_uninit(stackenv* se); |
+ |
+// Callback invoked by upb if any error occurs during parsing or serialization. |
+static bool env_error_func(void* ud, const upb_status* status) { |
+ stackenv* se = ud; |
+ // Free the env -- rb_raise will longjmp up the stack past the encode/decode |
+ // function so it would not otherwise have been freed. |
+ stackenv_uninit(se); |
+ |
+ // TODO(teboring): have a way to verify that this is actually a parse error, |
+ // instead of just throwing "parse error" unconditionally. |
+ zend_error(E_ERROR, se->php_error_template); |
+ // Never reached. |
+ return false; |
+} |
+ |
+static void stackenv_init(stackenv* se, const char* errmsg) { |
+ se->php_error_template = errmsg; |
+ upb_env_init(&se->env); |
+ upb_seededalloc_init(&se->alloc, &se->allocbuf, STACK_ENV_STACKBYTES); |
+ upb_env_setallocfunc(&se->env, upb_seededalloc_getallocfunc(&se->alloc), |
+ &se->alloc); |
+ upb_env_seterrorfunc(&se->env, env_error_func, se); |
+} |
+ |
+static void stackenv_uninit(stackenv* se) { |
+ upb_env_uninit(&se->env); |
+ upb_seededalloc_uninit(&se->alloc); |
+} |
+ |
+// ----------------------------------------------------------------------------- |
+// Message |
+// ----------------------------------------------------------------------------- |
+ |
+static const upb_handlers* msgdef_pb_serialize_handlers(Descriptor* desc) { |
+ if (desc->pb_serialize_handlers == NULL) { |
+ desc->pb_serialize_handlers = |
+ upb_pb_encoder_newhandlers(desc->msgdef, &desc->pb_serialize_handlers); |
+ } |
+ return desc->pb_serialize_handlers; |
+} |
+ |
+PHP_METHOD(Message, encode) { |
+ Descriptor* desc = (Descriptor*)zend_object_store_get_object( |
+ CE_STATIC_MEMBERS(Z_OBJCE_P(getThis()))[0] TSRMLS_CC); |
+ |
+ stringsink sink; |
+ stringsink_init(&sink); |
+ |
+ { |
+ const upb_handlers* serialize_handlers = msgdef_pb_serialize_handlers(desc); |
+ |
+ stackenv se; |
+ upb_pb_encoder* encoder; |
+ |
+ stackenv_init(&se, "Error occurred during encoding: %s"); |
+ encoder = upb_pb_encoder_create(&se.env, serialize_handlers, &sink.sink); |
+ |
+ putmsg(getThis(), desc, upb_pb_encoder_input(encoder), 0); |
+ |
+ RETVAL_STRINGL(sink.ptr, sink.len, 1); |
+ |
+ stackenv_uninit(&se); |
+ stringsink_uninit(&sink); |
+ } |
+} |
+ |
+void message_free(void * object TSRMLS_DC) { |
+ FREE(object); |
+} |
+ |
+zend_object_value message_create(zend_class_entry* ce TSRMLS_DC) { |
+ zend_object_value return_value; |
+ |
+ zval* php_descriptor = get_def_obj(ce); |
+ |
+ Descriptor* desc = zend_object_store_get_object(php_descriptor TSRMLS_CC); |
+ MessageHeader* msg = (MessageHeader*)ALLOC_N( |
+ uint8_t, sizeof(MessageHeader) + desc->layout->size); |
+ memset(message_data(msg), 0, desc->layout->size); |
+ |
+ // We wrap first so that everything in the message object is GC-rooted in case |
+ // a collection happens during object creation in layout_init(). |
+ msg->descriptor = desc; |
+ |
+ layout_init(desc->layout, message_data(msg)); |
+ zend_object_std_init(&msg->std, ce TSRMLS_CC); |
+ |
+ return_value.handle = zend_objects_store_put( |
+ msg, (zend_objects_store_dtor_t)zend_objects_destroy_object, |
+ message_free, NULL TSRMLS_CC); |
+ |
+ return_value.handlers = PROTOBUF_G(message_handlers); |
+ return return_value; |
+} |
+ |
+const zend_class_entry* build_class_from_descriptor( |
+ zval* php_descriptor TSRMLS_DC) { |
+ Descriptor* desc = php_to_descriptor(php_descriptor); |
+ if (desc->layout == NULL) { |
+ MessageLayout* layout = create_layout(desc->msgdef); |
+ desc->layout = layout; |
+ } |
+ // TODO(teboring): Add it back. |
+ // if (desc->fill_method == NULL) { |
+ // desc->fill_method = new_fillmsg_decodermethod(desc, &desc->fill_method); |
+ // } |
+ |
+ const char* name = upb_msgdef_fullname(desc->msgdef); |
+ if (name == NULL) { |
+ zend_error(E_ERROR, "Descriptor does not have assigned name."); |
+ } |
+ |
+ zend_class_entry class_entry; |
+ INIT_CLASS_ENTRY_EX(class_entry, name, strlen(name), message_methods); |
+ zend_class_entry* registered_ce = |
+ zend_register_internal_class(&class_entry TSRMLS_CC); |
+ |
+ add_def_obj(registered_ce, php_descriptor); |
+ |
+ if (PROTOBUF_G(message_handlers) == NULL) { |
+ PROTOBUF_G(message_handlers) = ALLOC(zend_object_handlers); |
+ memcpy(PROTOBUF_G(message_handlers), zend_get_std_object_handlers(), |
+ sizeof(zend_object_handlers)); |
+ PROTOBUF_G(message_handlers)->write_property = message_set_property; |
+ PROTOBUF_G(message_handlers)->read_property = message_get_property; |
+ } |
+ |
+ registered_ce->create_object = message_create; |
+} |