blob: 3e32cb9274f7dc6ce5c17ed467fed90a60c73e63 [file] [log] [blame]
/*
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* KIND, either express or implied. See the License for the
* specific language governing permissions and limitations
* under the License.
*/
#ifdef HAVE_CONFIG_H
#include "config.h"
#endif
#include <sys/types.h>
#if defined( WIN32 ) || defined( _WIN64 )
typedef int int32_t;
typedef signed char int8_t;
typedef unsigned char uint8_t;
typedef unsigned short uint16_t;
typedef long long int64_t;
typedef unsigned uint32_t;
typedef short int16_t;
typedef unsigned long long uint64_t;
#else
#include <arpa/inet.h>
#endif
#include <stdexcept>
#ifndef bswap_64
#define bswap_64(x) (((uint64_t)(x) << 56) | \
(((uint64_t)(x) << 40) & 0xff000000000000ULL) | \
(((uint64_t)(x) << 24) & 0xff0000000000ULL) | \
(((uint64_t)(x) << 8) & 0xff00000000ULL) | \
(((uint64_t)(x) >> 8) & 0xff000000ULL) | \
(((uint64_t)(x) >> 24) & 0xff0000ULL) | \
(((uint64_t)(x) >> 40) & 0xff00ULL) | \
((uint64_t)(x) >> 56))
#endif
#if __BYTE_ORDER == __LITTLE_ENDIAN
#define htonll(x) bswap_64(x)
#define ntohll(x) bswap_64(x)
#elif __BYTE_ORDER == __BIG_ENDIAN
#define htonll(x) x
#define ntohll(x) x
#else
#error Unknown __BYTE_ORDER
#endif
enum TType {
T_STOP = 0,
T_VOID = 1,
T_BOOL = 2,
T_BYTE = 3,
T_I08 = 3,
T_I16 = 6,
T_I32 = 8,
T_U64 = 9,
T_I64 = 10,
T_DOUBLE = 4,
T_STRING = 11,
T_UTF7 = 11,
T_STRUCT = 12,
T_MAP = 13,
T_SET = 14,
T_LIST = 15,
T_UTF8 = 16,
T_UTF16 = 17
};
const int32_t VERSION_MASK = 0xffff0000;
const int32_t VERSION_1 = 0x80010000;
const int8_t T_CALL = 1;
const int8_t T_REPLY = 2;
const int8_t T_EXCEPTION = 3;
// tprotocolexception
const int INVALID_DATA = 1;
const int BAD_VERSION = 4;
#include "php.h"
#include "zend_interfaces.h"
#include "zend_exceptions.h"
#include "php_thrift_protocol.h"
static function_entry thrift_protocol_functions[] = {
PHP_FE(thrift_protocol_write_binary, NULL)
PHP_FE(thrift_protocol_read_binary, NULL)
{NULL, NULL, NULL}
} ;
zend_module_entry thrift_protocol_module_entry = {
STANDARD_MODULE_HEADER,
"thrift_protocol",
thrift_protocol_functions,
NULL,
NULL,
NULL,
NULL,
NULL,
"1.0",
STANDARD_MODULE_PROPERTIES
};
#ifdef COMPILE_DL_THRIFT_PROTOCOL
ZEND_GET_MODULE(thrift_protocol)
#endif
class PHPExceptionWrapper : public std::exception {
public:
PHPExceptionWrapper(zval* _ex) throw() : ex(_ex) {
snprintf(_what, 40, "PHP exception zval=%p", ex);
}
const char* what() const throw() { return _what; }
~PHPExceptionWrapper() throw() {}
operator zval*() const throw() { return const_cast<zval*>(ex); } // Zend API doesn't do 'const'...
protected:
zval* ex;
char _what[40];
} ;
class PHPTransport {
public:
zval* protocol() { return p; }
zval* transport() { return t; }
protected:
PHPTransport() {}
void construct_with_zval(zval* _p, size_t _buffer_size) {
buffer = reinterpret_cast<char*>(emalloc(_buffer_size));
buffer_ptr = buffer;
buffer_used = 0;
buffer_size = _buffer_size;
p = _p;
// Get the transport for the passed protocol
zval gettransport;
ZVAL_STRING(&gettransport, "getTransport", 0);
MAKE_STD_ZVAL(t);
ZVAL_NULL(t);
TSRMLS_FETCH();
call_user_function(EG(function_table), &p, &gettransport, t, 0, NULL TSRMLS_CC);
}
~PHPTransport() {
efree(buffer);
zval_ptr_dtor(&t);
}
char* buffer;
char* buffer_ptr;
size_t buffer_used;
size_t buffer_size;
zval* p;
zval* t;
};
class PHPOutputTransport : public PHPTransport {
public:
PHPOutputTransport(zval* _p, size_t _buffer_size = 8192) {
construct_with_zval(_p, _buffer_size);
}
~PHPOutputTransport() {
//flush();
}
void write(const char* data, size_t len) {
if ((len + buffer_used) > buffer_size) {
internalFlush();
}
if (len > buffer_size) {
directWrite(data, len);
} else {
memcpy(buffer_ptr, data, len);
buffer_used += len;
buffer_ptr += len;
}
}
void writeI64(int64_t i) {
i = htonll(i);
write((const char*)&i, 8);
}
void writeU32(uint32_t i) {
i = htonl(i);
write((const char*)&i, 4);
}
void writeI32(int32_t i) {
i = htonl(i);
write((const char*)&i, 4);
}
void writeI16(int16_t i) {
i = htons(i);
write((const char*)&i, 2);
}
void writeI8(int8_t i) {
write((const char*)&i, 1);
}
void writeString(const char* str, size_t len) {
writeU32(len);
write(str, len);
}
void flush() {
internalFlush();
directFlush();
}
protected:
void internalFlush() {
if (buffer_used) {
directWrite(buffer, buffer_used);
buffer_ptr = buffer;
buffer_used = 0;
}
}
void directFlush() {
zval ret;
ZVAL_NULL(&ret);
zval flushfn;
ZVAL_STRING(&flushfn, "flush", 0);
TSRMLS_FETCH();
call_user_function(EG(function_table), &t, &flushfn, &ret, 0, NULL TSRMLS_CC);
zval_dtor(&ret);
}
void directWrite(const char* data, size_t len) {
zval writefn;
ZVAL_STRING(&writefn, "write", 0);
char* newbuf = (char*)emalloc(len + 1);
memcpy(newbuf, data, len);
newbuf[len] = '\0';
zval *args[1];
MAKE_STD_ZVAL(args[0]);
ZVAL_STRINGL(args[0], newbuf, len, 0);
TSRMLS_FETCH();
zval ret;
ZVAL_NULL(&ret);
call_user_function(EG(function_table), &t, &writefn, &ret, 1, args TSRMLS_CC);
zval_ptr_dtor(args);
zval_dtor(&ret);
if (EG(exception)) {
zval* ex = EG(exception);
EG(exception) = NULL;
throw PHPExceptionWrapper(ex);
}
}
};
class PHPInputTransport : public PHPTransport {
public:
PHPInputTransport(zval* _p, size_t _buffer_size = 8192) {
construct_with_zval(_p, _buffer_size);
}
~PHPInputTransport() {
put_back();
}
void put_back() {
if (buffer_used) {
zval putbackfn;
ZVAL_STRING(&putbackfn, "putBack", 0);
char* newbuf = (char*)emalloc(buffer_used + 1);
memcpy(newbuf, buffer_ptr, buffer_used);
newbuf[buffer_used] = '\0';
zval *args[1];
MAKE_STD_ZVAL(args[0]);
ZVAL_STRINGL(args[0], newbuf, buffer_used, 0);
TSRMLS_FETCH();
zval ret;
ZVAL_NULL(&ret);
call_user_function(EG(function_table), &t, &putbackfn, &ret, 1, args TSRMLS_CC);
zval_ptr_dtor(args);
zval_dtor(&ret);
}
buffer_used = 0;
buffer_ptr = buffer;
}
void skip(size_t len) {
while (len) {
size_t chunk_size = MIN(len, buffer_used);
if (chunk_size) {
buffer_ptr = reinterpret_cast<char*>(buffer_ptr) + chunk_size;
buffer_used -= chunk_size;
len -= chunk_size;
}
if (! len) break;
refill();
}
}
void readBytes(void* buf, size_t len) {
while (len) {
size_t chunk_size = MIN(len, buffer_used);
if (chunk_size) {
memcpy(buf, buffer_ptr, chunk_size);
buffer_ptr = reinterpret_cast<char*>(buffer_ptr) + chunk_size;
buffer_used -= chunk_size;
buf = reinterpret_cast<char*>(buf) + chunk_size;
len -= chunk_size;
}
if (! len) break;
refill();
}
}
int8_t readI8() {
int8_t c;
readBytes(&c, 1);
return c;
}
int16_t readI16() {
int16_t c;
readBytes(&c, 2);
return (int16_t)ntohs(c);
}
uint32_t readU32() {
uint32_t c;
readBytes(&c, 4);
return (uint32_t)ntohl(c);
}
int32_t readI32() {
int32_t c;
readBytes(&c, 4);
return (int32_t)ntohl(c);
}
protected:
void refill() {
assert(buffer_used == 0);
zval retval;
ZVAL_NULL(&retval);
zval *args[1];
MAKE_STD_ZVAL(args[0]);
ZVAL_LONG(args[0], buffer_size);
TSRMLS_FETCH();
zval funcname;
ZVAL_STRING(&funcname, "read", 0);
call_user_function(EG(function_table), &t, &funcname, &retval, 1, args TSRMLS_CC);
zval_ptr_dtor(args);
if (EG(exception)) {
zval_dtor(&retval);
zval* ex = EG(exception);
EG(exception) = NULL;
throw PHPExceptionWrapper(ex);
}
buffer_used = Z_STRLEN(retval);
memcpy(buffer, Z_STRVAL(retval), buffer_used);
zval_dtor(&retval);
buffer_ptr = buffer;
}
};
void binary_deserialize_spec(zval* zthis, PHPInputTransport& transport, HashTable* spec);
void binary_serialize_spec(zval* zthis, PHPOutputTransport& transport, HashTable* spec);
void binary_serialize(int8_t thrift_typeID, PHPOutputTransport& transport, zval** value, HashTable* fieldspec);
void skip_element(long thrift_typeID, PHPInputTransport& transport);
void protocol_writeMessageBegin(zval *transport, const char* method_name, int32_t msgtype, int32_t seqID);
// Create a PHP object given a typename and call the ctor, optionally passing up to 2 arguments
void createObject(char* obj_typename, zval* return_value, int nargs = 0, zval* arg1 = NULL, zval* arg2 = NULL) {
TSRMLS_FETCH();
size_t obj_typename_len = strlen(obj_typename);
zend_class_entry* ce = zend_fetch_class(obj_typename, obj_typename_len, ZEND_FETCH_CLASS_DEFAULT TSRMLS_CC);
if (! ce) {
php_error_docref(NULL TSRMLS_CC, E_ERROR, "Class %s does not exist", obj_typename);
RETURN_NULL();
}
object_and_properties_init(return_value, ce, NULL);
zend_function* constructor = zend_std_get_constructor(return_value TSRMLS_CC);
zval* ctor_rv = NULL;
zend_call_method(&return_value, ce, &constructor, NULL, 0, &ctor_rv, nargs, arg1, arg2 TSRMLS_CC);
zval_ptr_dtor(&ctor_rv);
}
void throw_tprotocolexception(char* what, long errorcode) {
TSRMLS_FETCH();
zval *zwhat, *zerrorcode;
MAKE_STD_ZVAL(zwhat);
MAKE_STD_ZVAL(zerrorcode);
ZVAL_STRING(zwhat, what, 1);
ZVAL_LONG(zerrorcode, errorcode);
zval* ex;
MAKE_STD_ZVAL(ex);
createObject("TProtocolException", ex, 2, zwhat, zerrorcode);
zval_ptr_dtor(&zwhat);
zval_ptr_dtor(&zerrorcode);
throw PHPExceptionWrapper(ex);
}
// Sets EG(exception), call this and then RETURN_NULL();
void throw_zend_exception_from_std_exception(const std::exception& ex TSRMLS_DC) {
zend_throw_exception(zend_exception_get_default(TSRMLS_C), const_cast<char*>(ex.what()), 0 TSRMLS_CC);
}
void binary_deserialize(int8_t thrift_typeID, PHPInputTransport& transport, zval* return_value, HashTable* fieldspec) {
zval** val_ptr;
Z_TYPE_P(return_value) = IS_NULL; // just in case
switch (thrift_typeID) {
case T_STOP:
case T_VOID:
RETURN_NULL();
return;
case T_STRUCT: {
if (zend_hash_find(fieldspec, "class", 6, (void**)&val_ptr) != SUCCESS) {
throw_tprotocolexception("no class type in spec", INVALID_DATA);
skip_element(T_STRUCT, transport);
RETURN_NULL();
}
char* structType = Z_STRVAL_PP(val_ptr);
createObject(structType, return_value);
if (Z_TYPE_P(return_value) == IS_NULL) {
// unable to create class entry
skip_element(T_STRUCT, transport);
RETURN_NULL();
}
TSRMLS_FETCH();
zval* spec = zend_read_static_property(zend_get_class_entry(return_value TSRMLS_CC), "_TSPEC", 6, false TSRMLS_CC);
if (Z_TYPE_P(spec) != IS_ARRAY) {
char errbuf[128];
snprintf(errbuf, 128, "spec for %s is wrong type: %d\n", structType, Z_TYPE_P(spec));
throw_tprotocolexception(errbuf, INVALID_DATA);
RETURN_NULL();
}
binary_deserialize_spec(return_value, transport, Z_ARRVAL_P(spec));
return;
} break;
case T_BOOL: {
uint8_t c;
transport.readBytes(&c, 1);
RETURN_BOOL(c != 0);
}
//case T_I08: // same numeric value as T_BYTE
case T_BYTE: {
uint8_t c;
transport.readBytes(&c, 1);
RETURN_LONG((int8_t)c);
}
case T_I16: {
uint16_t c;
transport.readBytes(&c, 2);
RETURN_LONG((int16_t)ntohs(c));
}
case T_I32: {
uint32_t c;
transport.readBytes(&c, 4);
RETURN_LONG((int32_t)ntohl(c));
}
case T_U64:
case T_I64: {
uint64_t c;
transport.readBytes(&c, 8);
RETURN_LONG((int64_t)ntohll(c));
}
case T_DOUBLE: {
union {
uint64_t c;
double d;
} a;
transport.readBytes(&(a.c), 8);
a.c = ntohll(a.c);
RETURN_DOUBLE(a.d);
}
//case T_UTF7: // aliases T_STRING
case T_UTF8:
case T_UTF16:
case T_STRING: {
uint32_t size = transport.readU32();
if (size) {
char* strbuf = (char*) emalloc(size + 1);
transport.readBytes(strbuf, size);
strbuf[size] = '\0';
ZVAL_STRINGL(return_value, strbuf, size, 0);
} else {
ZVAL_EMPTY_STRING(return_value);
}
return;
}
case T_MAP: { // array of key -> value
uint8_t types[2];
transport.readBytes(types, 2);
uint32_t size = transport.readU32();
array_init(return_value);
zend_hash_find(fieldspec, "key", 4, (void**)&val_ptr);
HashTable* keyspec = Z_ARRVAL_PP(val_ptr);
zend_hash_find(fieldspec, "val", 4, (void**)&val_ptr);
HashTable* valspec = Z_ARRVAL_PP(val_ptr);
for (uint32_t s = 0; s < size; ++s) {
zval *value;
MAKE_STD_ZVAL(value);
zval* key;
MAKE_STD_ZVAL(key);
binary_deserialize(types[0], transport, key, keyspec);
binary_deserialize(types[1], transport, value, valspec);
if (Z_TYPE_P(key) == IS_LONG) {
zend_hash_index_update(return_value->value.ht, Z_LVAL_P(key), &value, sizeof(zval *), NULL);
}
else {
if (Z_TYPE_P(key) != IS_STRING) convert_to_string(key);
zend_hash_update(return_value->value.ht, Z_STRVAL_P(key), Z_STRLEN_P(key) + 1, &value, sizeof(zval *), NULL);
}
zval_ptr_dtor(&key);
}
return; // return_value already populated
}
case T_LIST: { // array with autogenerated numeric keys
int8_t type = transport.readI8();
uint32_t size = transport.readU32();
zend_hash_find(fieldspec, "elem", 5, (void**)&val_ptr);
HashTable* elemspec = Z_ARRVAL_PP(val_ptr);
array_init(return_value);
for (uint32_t s = 0; s < size; ++s) {
zval *value;
MAKE_STD_ZVAL(value);
binary_deserialize(type, transport, value, elemspec);
zend_hash_next_index_insert(return_value->value.ht, &value, sizeof(zval *), NULL);
}
return;
}
case T_SET: { // array of key -> TRUE
uint8_t type;
uint32_t size;
transport.readBytes(&type, 1);
transport.readBytes(&size, 4);
size = ntohl(size);
zend_hash_find(fieldspec, "elem", 5, (void**)&val_ptr);
HashTable* elemspec = Z_ARRVAL_PP(val_ptr);
array_init(return_value);
for (uint32_t s = 0; s < size; ++s) {
zval* key;
zval* value;
MAKE_STD_ZVAL(key);
MAKE_STD_ZVAL(value);
ZVAL_TRUE(value);
binary_deserialize(type, transport, key, elemspec);
if (Z_TYPE_P(key) == IS_LONG) {
zend_hash_index_update(return_value->value.ht, Z_LVAL_P(key), &value, sizeof(zval *), NULL);
}
else {
if (Z_TYPE_P(key) != IS_STRING) convert_to_string(key);
zend_hash_update(return_value->value.ht, Z_STRVAL_P(key), Z_STRLEN_P(key) + 1, &value, sizeof(zval *), NULL);
}
zval_ptr_dtor(&key);
}
return;
}
};
char errbuf[128];
sprintf(errbuf, "Unknown thrift typeID %d", thrift_typeID);
throw_tprotocolexception(errbuf, INVALID_DATA);
}
void skip_element(long thrift_typeID, PHPInputTransport& transport) {
switch (thrift_typeID) {
case T_STOP:
case T_VOID:
return;
case T_STRUCT:
while (true) {
int8_t ttype = transport.readI8(); // get field type
if (ttype == T_STOP) break;
transport.skip(2); // skip field number, I16
skip_element(ttype, transport); // skip field payload
}
return;
case T_BOOL:
case T_BYTE:
transport.skip(1);
return;
case T_I16:
transport.skip(2);
return;
case T_I32:
transport.skip(4);
return;
case T_U64:
case T_I64:
case T_DOUBLE:
transport.skip(8);
return;
//case T_UTF7: // aliases T_STRING
case T_UTF8:
case T_UTF16:
case T_STRING: {
uint32_t len = transport.readU32();
transport.skip(len);
} return;
case T_MAP: {
int8_t keytype = transport.readI8();
int8_t valtype = transport.readI8();
uint32_t size = transport.readU32();
for (uint32_t i = 0; i < size; ++i) {
skip_element(keytype, transport);
skip_element(valtype, transport);
}
} return;
case T_LIST:
case T_SET: {
int8_t valtype = transport.readI8();
uint32_t size = transport.readU32();
for (uint32_t i = 0; i < size; ++i) {
skip_element(valtype, transport);
}
} return;
};
char errbuf[128];
sprintf(errbuf, "Unknown thrift typeID %ld", thrift_typeID);
throw_tprotocolexception(errbuf, INVALID_DATA);
}
void protocol_writeMessageBegin(zval* transport, const char* method_name, int32_t msgtype, int32_t seqID) {
TSRMLS_FETCH();
zval *args[3];
MAKE_STD_ZVAL(args[0]);
ZVAL_STRINGL(args[0], (char*)method_name, strlen(method_name), 1);
MAKE_STD_ZVAL(args[1]);
ZVAL_LONG(args[1], msgtype);
MAKE_STD_ZVAL(args[2]);
ZVAL_LONG(args[2], seqID);
zval ret;
ZVAL_NULL(&ret);
zval writeMessagefn;
ZVAL_STRING(&writeMessagefn, "writeMessageBegin", 0);
call_user_function(EG(function_table), &transport, &writeMessagefn, &ret, 3, args TSRMLS_CC);
zval_ptr_dtor(&args[0]);
zval_ptr_dtor(&args[1]);
zval_ptr_dtor(&args[2]);
zval_dtor(&ret);
}
void binary_serialize_hashtable_key(int8_t keytype, PHPOutputTransport& transport, HashTable* ht, HashPosition& ht_pos) {
bool keytype_is_numeric = (!((keytype == T_STRING) || (keytype == T_UTF8) || (keytype == T_UTF16)));
char* key;
uint key_len;
long index = 0;
zval* z;
MAKE_STD_ZVAL(z);
int res = zend_hash_get_current_key_ex(ht, &key, &key_len, (ulong*)&index, 0, &ht_pos);
if (keytype_is_numeric) {
if (res == HASH_KEY_IS_STRING) {
index = strtol(key, NULL, 10);
}
ZVAL_LONG(z, index);
} else {
char buf[64];
if (res == HASH_KEY_IS_STRING) {
key_len -= 1; // skip the null terminator
} else {
sprintf(buf, "%ld", index);
key = buf; key_len = strlen(buf);
}
ZVAL_STRINGL(z, key, key_len, 1);
}
binary_serialize(keytype, transport, &z, NULL);
zval_ptr_dtor(&z);
}
inline bool ttype_is_int(int8_t t) {
return ((t == T_BYTE) || ((t >= T_I16) && (t <= T_I64)));
}
inline bool ttypes_are_compatible(int8_t t1, int8_t t2) {
// Integer types of different widths are considered compatible;
// otherwise the typeID must match.
return ((t1 == t2) || (ttype_is_int(t1) && ttype_is_int(t2)));
}
void binary_deserialize_spec(zval* zthis, PHPInputTransport& transport, HashTable* spec) {
// SET and LIST have 'elem' => array('type', [optional] 'class')
// MAP has 'val' => array('type', [optiona] 'class')
TSRMLS_FETCH();
zend_class_entry* ce = zend_get_class_entry(zthis TSRMLS_CC);
while (true) {
zval** val_ptr = NULL;
int8_t ttype = transport.readI8();
if (ttype == T_STOP) return;
int16_t fieldno = transport.readI16();
if (zend_hash_index_find(spec, fieldno, (void**)&val_ptr) == SUCCESS) {
HashTable* fieldspec = Z_ARRVAL_PP(val_ptr);
// pull the field name
// zend hash tables use the null at the end in the length... so strlen(hash key) + 1.
zend_hash_find(fieldspec, "var", 4, (void**)&val_ptr);
char* varname = Z_STRVAL_PP(val_ptr);
// and the type
zend_hash_find(fieldspec, "type", 5, (void**)&val_ptr);
if (Z_TYPE_PP(val_ptr) != IS_LONG) convert_to_long(*val_ptr);
int8_t expected_ttype = Z_LVAL_PP(val_ptr);
if (ttypes_are_compatible(ttype, expected_ttype)) {
zval* rv = NULL;
MAKE_STD_ZVAL(rv);
binary_deserialize(ttype, transport, rv, fieldspec);
zend_update_property(ce, zthis, varname, strlen(varname), rv TSRMLS_CC);
zval_ptr_dtor(&rv);
} else {
skip_element(ttype, transport);
}
} else {
skip_element(ttype, transport);
}
}
}
void binary_serialize(int8_t thrift_typeID, PHPOutputTransport& transport, zval** value, HashTable* fieldspec) {
// At this point the typeID (and field num, if applicable) should've already been written to the output so all we need to do is write the payload.
switch (thrift_typeID) {
case T_STOP:
case T_VOID:
return;
case T_STRUCT: {
TSRMLS_FETCH();
if (Z_TYPE_PP(value) != IS_OBJECT) {
throw_tprotocolexception("Attempt to send non-object type as a T_STRUCT", INVALID_DATA);
}
zval* spec = zend_read_static_property(zend_get_class_entry(*value TSRMLS_CC), "_TSPEC", 6, false TSRMLS_CC);
if (Z_TYPE_P(spec) != IS_ARRAY) {
throw_tprotocolexception("Attempt to send non-Thrift object as a T_STRUCT", INVALID_DATA);
}
binary_serialize_spec(*value, transport, Z_ARRVAL_P(spec));
} return;
case T_BOOL:
if (Z_TYPE_PP(value) != IS_BOOL) convert_to_boolean(*value);
transport.writeI8(Z_BVAL_PP(value) ? 1 : 0);
return;
case T_BYTE:
if (Z_TYPE_PP(value) != IS_LONG) convert_to_long(*value);
transport.writeI8(Z_LVAL_PP(value));
return;
case T_I16:
if (Z_TYPE_PP(value) != IS_LONG) convert_to_long(*value);
transport.writeI16(Z_LVAL_PP(value));
return;
case T_I32:
if (Z_TYPE_PP(value) != IS_LONG) convert_to_long(*value);
transport.writeI32(Z_LVAL_PP(value));
return;
case T_I64:
case T_U64: {
int64_t l_data;
#if defined(_LP64) || defined(_WIN64)
if (Z_TYPE_PP(value) != IS_LONG) convert_to_long(*value);
l_data = Z_LVAL_PP(value);
#else
if (Z_TYPE_PP(value) != IS_DOUBLE) convert_to_double(*value);
l_data = (int64_t)Z_DVAL_PP(value);
#endif
transport.writeI64(l_data);
} return;
case T_DOUBLE: {
union {
int64_t c;
double d;
} a;
if (Z_TYPE_PP(value) != IS_DOUBLE) convert_to_double(*value);
a.d = Z_DVAL_PP(value);
transport.writeI64(a.c);
} return;
//case T_UTF7:
case T_UTF8:
case T_UTF16:
case T_STRING:
if (Z_TYPE_PP(value) != IS_STRING) convert_to_string(*value);
transport.writeString(Z_STRVAL_PP(value), Z_STRLEN_PP(value));
return;
case T_MAP: {
if (Z_TYPE_PP(value) != IS_ARRAY) convert_to_array(*value);
if (Z_TYPE_PP(value) != IS_ARRAY) {
throw_tprotocolexception("Attempt to send an incompatible type as an array (T_MAP)", INVALID_DATA);
}
HashTable* ht = Z_ARRVAL_PP(value);
zval** val_ptr;
zend_hash_find(fieldspec, "ktype", 6, (void**)&val_ptr);
if (Z_TYPE_PP(val_ptr) != IS_LONG) convert_to_long(*val_ptr);
uint8_t keytype = Z_LVAL_PP(val_ptr);
transport.writeI8(keytype);
zend_hash_find(fieldspec, "vtype", 6, (void**)&val_ptr);
if (Z_TYPE_PP(val_ptr) != IS_LONG) convert_to_long(*val_ptr);
uint8_t valtype = Z_LVAL_PP(val_ptr);
transport.writeI8(valtype);
zend_hash_find(fieldspec, "val", 4, (void**)&val_ptr);
HashTable* valspec = Z_ARRVAL_PP(val_ptr);
transport.writeI32(zend_hash_num_elements(ht));
HashPosition key_ptr;
for (zend_hash_internal_pointer_reset_ex(ht, &key_ptr); zend_hash_get_current_data_ex(ht, (void**)&val_ptr, &key_ptr) == SUCCESS; zend_hash_move_forward_ex(ht, &key_ptr)) {
binary_serialize_hashtable_key(keytype, transport, ht, key_ptr);
binary_serialize(valtype, transport, val_ptr, valspec);
}
} return;
case T_LIST: {
if (Z_TYPE_PP(value) != IS_ARRAY) convert_to_array(*value);
if (Z_TYPE_PP(value) != IS_ARRAY) {
throw_tprotocolexception("Attempt to send an incompatible type as an array (T_LIST)", INVALID_DATA);
}
HashTable* ht = Z_ARRVAL_PP(value);
zval** val_ptr;
zend_hash_find(fieldspec, "etype", 6, (void**)&val_ptr);
if (Z_TYPE_PP(val_ptr) != IS_LONG) convert_to_long(*val_ptr);
uint8_t valtype = Z_LVAL_PP(val_ptr);
transport.writeI8(valtype);
zend_hash_find(fieldspec, "elem", 5, (void**)&val_ptr);
HashTable* valspec = Z_ARRVAL_PP(val_ptr);
transport.writeI32(zend_hash_num_elements(ht));
HashPosition key_ptr;
for (zend_hash_internal_pointer_reset_ex(ht, &key_ptr); zend_hash_get_current_data_ex(ht, (void**)&val_ptr, &key_ptr) == SUCCESS; zend_hash_move_forward_ex(ht, &key_ptr)) {
binary_serialize(valtype, transport, val_ptr, valspec);
}
} return;
case T_SET: {
if (Z_TYPE_PP(value) != IS_ARRAY) convert_to_array(*value);
if (Z_TYPE_PP(value) != IS_ARRAY) {
throw_tprotocolexception("Attempt to send an incompatible type as an array (T_SET)", INVALID_DATA);
}
HashTable* ht = Z_ARRVAL_PP(value);
zval** val_ptr;
zend_hash_find(fieldspec, "etype", 6, (void**)&val_ptr);
if (Z_TYPE_PP(val_ptr) != IS_LONG) convert_to_long(*val_ptr);
uint8_t keytype = Z_LVAL_PP(val_ptr);
transport.writeI8(keytype);
transport.writeI32(zend_hash_num_elements(ht));
HashPosition key_ptr;
for (zend_hash_internal_pointer_reset_ex(ht, &key_ptr); zend_hash_get_current_data_ex(ht, (void**)&val_ptr, &key_ptr) == SUCCESS; zend_hash_move_forward_ex(ht, &key_ptr)) {
binary_serialize_hashtable_key(keytype, transport, ht, key_ptr);
}
} return;
};
char errbuf[128];
sprintf(errbuf, "Unknown thrift typeID %d", thrift_typeID);
throw_tprotocolexception(errbuf, INVALID_DATA);
}
void binary_serialize_spec(zval* zthis, PHPOutputTransport& transport, HashTable* spec) {
HashPosition key_ptr;
zval** val_ptr;
TSRMLS_FETCH();
zend_class_entry* ce = zend_get_class_entry(zthis TSRMLS_CC);
for (zend_hash_internal_pointer_reset_ex(spec, &key_ptr); zend_hash_get_current_data_ex(spec, (void**)&val_ptr, &key_ptr) == SUCCESS; zend_hash_move_forward_ex(spec, &key_ptr)) {
ulong fieldno;
if (zend_hash_get_current_key_ex(spec, NULL, NULL, &fieldno, 0, &key_ptr) != HASH_KEY_IS_LONG) {
throw_tprotocolexception("Bad keytype in TSPEC (expected 'long')", INVALID_DATA);
return;
}
HashTable* fieldspec = Z_ARRVAL_PP(val_ptr);
// field name
zend_hash_find(fieldspec, "var", 4, (void**)&val_ptr);
char* varname = Z_STRVAL_PP(val_ptr);
// thrift type
zend_hash_find(fieldspec, "type", 5, (void**)&val_ptr);
if (Z_TYPE_PP(val_ptr) != IS_LONG) convert_to_long(*val_ptr);
int8_t ttype = Z_LVAL_PP(val_ptr);
zval* prop = zend_read_property(ce, zthis, varname, strlen(varname), false TSRMLS_CC);
if (Z_TYPE_P(prop) != IS_NULL) {
transport.writeI8(ttype);
transport.writeI16(fieldno);
binary_serialize(ttype, transport, &prop, fieldspec);
}
}
transport.writeI8(T_STOP); // struct end
}
// 6 params: $transport $method_name $ttype $request_struct $seqID $strict_write
PHP_FUNCTION(thrift_protocol_write_binary) {
int argc = ZEND_NUM_ARGS();
if (argc < 6) {
WRONG_PARAM_COUNT;
}
zval ***args = (zval***) emalloc(argc * sizeof(zval**));
zend_get_parameters_array_ex(argc, args);
if (Z_TYPE_PP(args[0]) != IS_OBJECT) {
php_error_docref(NULL TSRMLS_CC, E_ERROR, "1st parameter is not an object (transport)");
efree(args);
RETURN_NULL();
}
if (Z_TYPE_PP(args[1]) != IS_STRING) {
php_error_docref(NULL TSRMLS_CC, E_ERROR, "2nd parameter is not a string (method name)");
efree(args);
RETURN_NULL();
}
if (Z_TYPE_PP(args[3]) != IS_OBJECT) {
php_error_docref(NULL TSRMLS_CC, E_ERROR, "4th parameter is not an object (request struct)");
efree(args);
RETURN_NULL();
}
try {
PHPOutputTransport transport(*args[0]);
zval *protocol = *args[0];
const char* method_name = Z_STRVAL_PP(args[1]);
convert_to_long(*args[2]);
int32_t msgtype = Z_LVAL_PP(args[2]);
zval* request_struct = *args[3];
convert_to_long(*args[4]);
int32_t seqID = Z_LVAL_PP(args[4]);
convert_to_boolean(*args[5]);
bool strictWrite = Z_BVAL_PP(args[5]);
efree(args);
args = NULL;
protocol_writeMessageBegin(protocol, method_name, msgtype, seqID);
zval* spec = zend_read_static_property(zend_get_class_entry(request_struct TSRMLS_CC), "_TSPEC", 6, false TSRMLS_CC);
if (Z_TYPE_P(spec) != IS_ARRAY) {
throw_tprotocolexception("Attempt to send non-Thrift object", INVALID_DATA);
}
binary_serialize_spec(request_struct, transport, Z_ARRVAL_P(spec));
transport.flush();
} catch (const PHPExceptionWrapper& ex) {
zend_throw_exception_object(ex TSRMLS_CC);
RETURN_NULL();
} catch (const std::exception& ex) {
throw_zend_exception_from_std_exception(ex TSRMLS_CC);
RETURN_NULL();
}
}
// 3 params: $transport $response_Typename $strict_read
PHP_FUNCTION(thrift_protocol_read_binary) {
int argc = ZEND_NUM_ARGS();
if (argc < 3) {
WRONG_PARAM_COUNT;
}
zval ***args = (zval***) emalloc(argc * sizeof(zval**));
zend_get_parameters_array_ex(argc, args);
if (Z_TYPE_PP(args[0]) != IS_OBJECT) {
php_error_docref(NULL TSRMLS_CC, E_ERROR, "1st parameter is not an object (transport)");
efree(args);
RETURN_NULL();
}
if (Z_TYPE_PP(args[1]) != IS_STRING) {
php_error_docref(NULL TSRMLS_CC, E_ERROR, "2nd parameter is not a string (typename of expected response struct)");
efree(args);
RETURN_NULL();
}
try {
PHPInputTransport transport(*args[0]);
char* obj_typename = Z_STRVAL_PP(args[1]);
convert_to_boolean(*args[2]);
bool strict_read = Z_BVAL_PP(args[2]);
efree(args);
args = NULL;
int8_t messageType = 0;
int32_t sz = transport.readI32();
if (sz < 0) {
// Check for correct version number
int32_t version = sz & VERSION_MASK;
if (version != VERSION_1) {
throw_tprotocolexception("Bad version identifier", BAD_VERSION);
}
messageType = (sz & 0x000000ff);
int32_t namelen = transport.readI32();
// skip the name string and the sequence ID, we don't care about those
transport.skip(namelen + 4);
} else {
if (strict_read) {
throw_tprotocolexception("No version identifier... old protocol client in strict mode?", BAD_VERSION);
} else {
// Handle pre-versioned input
transport.skip(sz); // skip string body
messageType = transport.readI8();
transport.skip(4); // skip sequence number
}
}
if (messageType == T_EXCEPTION) {
zval* ex;
MAKE_STD_ZVAL(ex);
createObject("TApplicationException", ex);
zval* spec = zend_read_static_property(zend_get_class_entry(ex TSRMLS_CC), "_TSPEC", 6, false TSRMLS_CC);
binary_deserialize_spec(ex, transport, Z_ARRVAL_P(spec));
throw PHPExceptionWrapper(ex);
}
createObject(obj_typename, return_value);
zval* spec = zend_read_static_property(zend_get_class_entry(return_value TSRMLS_CC), "_TSPEC", 6, false TSRMLS_CC);
binary_deserialize_spec(return_value, transport, Z_ARRVAL_P(spec));
} catch (const PHPExceptionWrapper& ex) {
zend_throw_exception_object(ex TSRMLS_CC);
RETURN_NULL();
} catch (const std::exception& ex) {
throw_zend_exception_from_std_exception(ex TSRMLS_CC);
RETURN_NULL();
}
}