writer.c

#include <string.h>

#include <php.h>

#include "protobuf.h"
#include "writer.h"

#define WRITER_DATA_SIZE_INCREMENT 1024
#define WRITER_VARINT_SPACE 10
#define WRITER_32BIT_SPACE 4
#define WRITER_64BIT_SPACE 8

zend_class_entry *pb_writer_entry;

static inline int writer_ensure_space(writer_t *writer, size_t len);
static inline void writer_write_varint(writer_t *writer, int64_t value);

static inline void write_fixed32(fixed32_t value, uint8_t *out);
static inline void write_fixed64(fixed64_t value, uint8_t *out);

void writer_free_pack(writer_t *writer)
{
	if (writer->data != NULL) {
		efree(writer->data);
		writer->data = NULL;

		writer->size = 0;
		writer->pos = 0;
	}
}

void writer_init(writer_t *writer)
{
	if ((writer->data = (uint8_t *) emalloc(WRITER_DATA_SIZE_INCREMENT)) != NULL)
		writer->size = WRITER_DATA_SIZE_INCREMENT;
	else
		writer->size = 0;

	writer->pos = 0;
}

char *writer_get_pack(writer_t *writer, int *size)
{
	*size = writer->pos;
	writer->data[writer->pos] = '\0';
	return (char *) writer->data;
}

int writer_write_double(writer_t *writer, uint32_t field_number, double value)
{
	int64_t key;
	fixed64_t val;

	if (writer_ensure_space(writer, WRITER_VARINT_SPACE + WRITER_64BIT_SPACE) != 0)
		return -1;

	key = ((uint64_t) field_number << 3 | WIRE_TYPE_64BIT);

	writer_write_varint(writer, key);

	val.d_val = value;
	write_fixed64(val, writer->data + writer->pos);
	writer->pos += WRITER_64BIT_SPACE;

	return 0;
}

int writer_write_fixed32(writer_t *writer, uint32_t field_number, int32_t value)
{
	int64_t key;
	fixed32_t val;

	if (writer_ensure_space(writer, WRITER_VARINT_SPACE + WRITER_32BIT_SPACE) != 0)
		return -1;

	key = ((uint64_t) field_number << 3 | WIRE_TYPE_32BIT);

	writer_write_varint(writer, key);

	val.i_val = value;
	write_fixed32(val, writer->data + writer->pos);
	writer->pos += WRITER_32BIT_SPACE;

	return 0;
}

int writer_write_fixed64(writer_t *writer, uint32_t field_number, int64_t value)
{
	int64_t key;
	fixed64_t val;

	if (writer_ensure_space(writer, WRITER_VARINT_SPACE + WRITER_64BIT_SPACE) != 0)
		return -1;

	key = ((uint64_t) field_number << 3 | WIRE_TYPE_64BIT);

	writer_write_varint(writer, key);

	val.i_val = value;
	write_fixed64(val, writer->data + writer->pos);
	writer->pos += WRITER_64BIT_SPACE;

	return 0;
}

int writer_write_float(writer_t *writer, uint32_t field_number, double value)
{
	int64_t key;
	fixed32_t val;

	if (writer_ensure_space(writer, WRITER_VARINT_SPACE + WRITER_32BIT_SPACE) != 0)
		return -1;

	key = ((uint64_t) field_number << 3 | WIRE_TYPE_32BIT);

	writer_write_varint(writer, key);

	val.f_val = value;
	write_fixed32(val, writer->data + writer->pos);
	writer->pos += WRITER_32BIT_SPACE;

	return 0;
}

int writer_write_int(writer_t *writer, uint32_t field_number, int64_t value)
{
	int64_t key;

	if (writer_ensure_space(writer, 2 * WRITER_VARINT_SPACE) != 0)
		return -1;

	key = ((uint64_t) field_number << 3 | WIRE_TYPE_VARINT);

	writer_write_varint(writer, key);
	writer_write_varint(writer, value);

	return 0;
}

int writer_write_signed_int(writer_t *writer, uint32_t field_number, int64_t value)
{
	int64_t key;

	if (writer_ensure_space(writer, 2 * WRITER_VARINT_SPACE) != 0)
		return -1;

	key = ((uint64_t) field_number << 3 | WIRE_TYPE_VARINT);

	writer_write_varint(writer, key);

	if (value < 0)
		*(uint64_t *) &value = ((uint64_t) (-value) * 2) - 1;
	else
		*(uint64_t *) &value = 2 * value;

	writer_write_varint(writer, value);

	return 0;
}

int writer_write_string(writer_t *writer, uint32_t field_number, const char *str, size_t len)
{
	int64_t key;

	if (writer_ensure_space(writer, 2 * WRITER_VARINT_SPACE + len) != 0)
		return -1;

	key = ((uint64_t) field_number << 3 | WIRE_TYPE_LENGTH_DELIMITED);

	writer_write_varint(writer, key);
	writer_write_varint(writer, len);
	memcpy(writer->data + writer->pos, str, len);
	writer->pos += len;

	return 0;
}

static inline int writer_ensure_space(writer_t *writer, size_t space)
{
	uint8_t *data;

	if ((writer->size - writer->pos) + 1 < space) {
		writer->size += space;
		data = (uint8_t *) erealloc(writer->data, writer->size);

		if (data == NULL)
			return -1;

		writer->data = data;
	}

	return 0;
}

static inline void writer_write_varint(writer_t *writer, int64_t value)
{
	int i;
	uint8_t byte;

	if (value == 0) {
		writer->data[writer->pos++] = 0;
	} else if (value < 0) {
		for (i = 0; i < WRITER_VARINT_SPACE - 1; i++) {
			writer->data[writer->pos++] = value | 0x80;
			*(uint64_t *) &value >>= 7;
		}

		writer->data[writer->pos++] = value;
	} else {
		do {
			byte = value;
			value >>= 7;

			if (value != 0)
				byte |= 0x80;
			else
				byte &= 0x7F;

			writer->data[writer->pos++] = byte;
		} while (value != 0);
	}
}

static inline void write_fixed32(fixed32_t value, uint8_t *out)
{
#ifndef WORDS_BIGENDIAN
	memcpy(out, &value.u_val, 4);
#else
	out[0] = value.u_val;
	out[1] = value.u_val >> 8;
	out[2] = value.u_val >> 16;
	out[3] = value.u_val >> 24;
#endif
}

static inline void write_fixed64(fixed64_t value, uint8_t *out)
{
#ifndef WORDS_BIGENDIAN
	memcpy(out, &value.u_val, 8);
#else
	fixed32_t lo;
	lo.u_val = value.u_val;

	fixed32_t hi;
	hi.u_val = value.u_val >> 32;

	write_fixed32(lo, out);
	write_fixed32(hi, out + 4);
#endif
}

#pragma mark -

PHP_METHOD(ProtobufWriter, convertIntToVarint)
{
	long value = -1;
	
	if(zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC, "l", &value) != SUCCESS)
	{
		return;
	}
	
	writer_t writer;
	writer_init(&writer);
	
	if (writer_ensure_space(&writer, WRITER_VARINT_SPACE) != 0)
		return;
	
	writer_write_varint(&writer, value);
	
	char *pack;
	int pack_size;
	
	pack = writer_get_pack(&writer, &pack_size);
	RETURN_STRINGL(pack, pack_size, 0);
	
	writer_free_pack(&writer);
}

PHP_METHOD(ProtobufWriter, convertSignedIntToVarint)
{
	long value = -1;
	
	if(zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC, "l", &value) != SUCCESS)
	{
		return;
	}
	
	writer_t writer;
	writer_init(&writer);
	
	if (writer_ensure_space(&writer, WRITER_VARINT_SPACE) != 0)
		return;
	
	if (value < 0)
		*(uint64_t *) &value = ((uint64_t) (-value) * 2) - 1;
	else
		*(uint64_t *) &value = 2 * value;
	
	writer_write_varint(&writer, value);
	
	char *pack;
	int pack_size;
	
	pack = writer_get_pack(&writer, &pack_size);
	RETURN_STRINGL(pack, pack_size, 0);
	
	writer_free_pack(&writer);
}

ZEND_BEGIN_ARG_INFO_EX(arginfo_convertIntToVarint, 0, 0, 1)
ZEND_ARG_INFO(0, value)
ZEND_END_ARG_INFO()

ZEND_BEGIN_ARG_INFO_EX(arginfo_convertSignedIntToVarint, 0, 0, 1)
ZEND_ARG_INFO(0, value)
ZEND_END_ARG_INFO()

zend_function_entry pb_writer_methods[] = {
	PHP_ME(ProtobufWriter, convertIntToVarint, arginfo_convertIntToVarint, ZEND_ACC_PUBLIC | ZEND_ACC_STATIC)
	PHP_ME(ProtobufWriter, convertSignedIntToVarint, arginfo_convertSignedIntToVarint, ZEND_ACC_PUBLIC | ZEND_ACC_STATIC)
	{NULL, NULL, NULL, 0, 0}
};

void writer_minit()
{
	zend_class_entry ce;
	
	INIT_CLASS_ENTRY(ce, "ProtobufWriter", pb_writer_methods);
	pb_writer_entry = zend_register_internal_class(&ce TSRMLS_CC);
}