skynet/lua-protobuf/context.c

#include "pbc.h"
#include "alloc.h"
#include "varint.h"
#include "context.h"

#include <stdlib.h>
#include <string.h>
#include <stdint.h>
#include <stdbool.h>

#define INNER_ATOM ((PBC_CONTEXT_CAP - sizeof(struct context)) / sizeof(struct atom))

static char * 
wiretype_decode(uint8_t *buffer, int cap , struct atom *a , int start)
{
	uint8_t temp[10];
	struct longlong r;
	int len;
	if (cap >= 10) {
		len = _pbcV_decode(buffer, &r);
		if (r.hi !=0)
			return NULL;
	} else {
		memcpy(temp, buffer , cap);
		len = _pbcV_decode(temp, &r);
		if (len > cap || r.hi !=0)
			return NULL;
	}

	int wiretype = r.low & 7;
	a->wire_id = r.low;
	buffer += len;
	start += len;
	cap -=len;
	
	switch (wiretype) {
	case WT_VARINT :
		if (cap >=10) {
			len = _pbcV_decode(buffer, &a->v.i);
		} else {
			memcpy(temp, buffer , cap);
			len = _pbcV_decode(temp, &a->v.i);
			if (cap < len)
				return NULL;
		}
		return (char *)buffer+len;
	case WT_BIT64 :
		if (cap < 8)
			return NULL;
		a->v.i.low = buffer[0] |
			buffer[1] << 8 |
			buffer[2] << 16 |
			buffer[3] << 24;
		a->v.i.hi = buffer[4] |
			buffer[5] << 8 |
			buffer[6] << 16 |
			buffer[7] << 24;
		return (char *)buffer + 8;
	case WT_LEND :
		if (cap >=10) {
			len = _pbcV_decode(buffer, &r);
		} else {
			memcpy(temp, buffer , cap);
			len = _pbcV_decode(temp, &r);
		}
		if (cap < len + r.low || r.hi !=0)
			return NULL;
		a->v.s.start = start + len;
		a->v.s.end = start + len + r.low;
		return (char *)buffer + len + r.low;
	case WT_BIT32 :
		if (cap < 4)
			return NULL;
		a->v.i.low = buffer[0] |
			buffer[1] << 8 |
			buffer[2] << 16 |
			buffer[3] << 24;
		a->v.i.hi = 0;
		return (char *)buffer + 4;
	default:
		return NULL;
	}
}

static inline int
_decode_varint(uint8_t * buffer, int size , struct atom * a) {
	a->wire_id = WT_VARINT;
	if (size < 10) {
		uint8_t temp[10];
		memcpy(temp,buffer,size);
		return _pbcV_decode(temp , &(a->v.i));
	} else {
		return _pbcV_decode(buffer , &(a->v.i));
	}
}

static int
_open_packed_varint(struct context * ctx , uint8_t * buffer, int size) {
	struct atom * a = (struct atom *)(ctx + 1);

	int i;

	for (i=0;i<INNER_ATOM;i++) {
		if (size == 0)
			break;
		int len = _decode_varint(buffer, size, &a[i]);
		buffer += len;
		size -= len;
	}

	if (size == 0) {
		ctx->a = a;
	} else {
		int cap = 64;
		ctx->a = malloc(cap * sizeof(struct atom));
		while (size > 0) {
			if (i >= cap) {
				cap = cap + 64;
				ctx->a = realloc(ctx->a, cap * sizeof(struct atom));
				continue;
			}
			int len = _decode_varint(buffer, size, &a[i]);
			buffer += len;
			size -= len;

			++i;
		}
		memcpy(ctx->a, a , sizeof(struct atom) * INNER_ATOM);
	}
	ctx->number = i;

	return i;
}

int 
_pbcC_open_packed(pbc_ctx _ctx, int ptype, void *buffer, int size) {
	struct context * ctx = (struct context *)_ctx;
	ctx->buffer = buffer;
	ctx->size = size;
	ctx->number = 0;
	ctx->a = NULL;

	if (buffer == NULL || size == 0) {
		return 0;
	}

	int bits = 0;

	switch (ptype) {
	case PTYPE_INT64:
	case PTYPE_UINT64:
	case PTYPE_INT32:
	case PTYPE_BOOL:
	case PTYPE_UINT32:
	case PTYPE_ENUM:
	case PTYPE_SINT32:
	case PTYPE_SINT64:
		return _open_packed_varint(ctx , buffer, size);
	case PTYPE_DOUBLE:
	case PTYPE_FIXED64:
	case PTYPE_SFIXED64:
		ctx->number = size / 8;
		bits = 64;
		break;
	case PTYPE_FLOAT:
	case PTYPE_FIXED32:
	case PTYPE_SFIXED32:
		ctx->number = size / 4;
		bits = 32;
		break;
	default:
		return 0;
	}

	struct atom * a = (struct atom *)(ctx + 1);

	if (ctx->number > INNER_ATOM) {
		ctx->a = malloc(ctx->number * sizeof(struct atom));
		a = ctx->a;
	} else {
		ctx->a = a;
	}

	int i;
	if (bits == 64) {
		uint8_t * data = buffer;
		for (i=0;i<ctx->number;i++) {
			a[i].wire_id = WT_BIT64;
			a[i].v.i.low = data[0] |
				data[1] << 8 |
				data[2] << 16 |
				data[3] << 24;
			a[i].v.i.hi = data[4] |
				data[5] << 8 |
				data[6] << 16 |
				data[7] << 24;
			data += 8;
		}
	} else {
		uint8_t * data = buffer;
		for (i=0;i<ctx->number;i++) {
			a[i].wire_id = WT_BIT32;
			a[i].v.i.low = data[0] |
				data[1] << 8 |
				data[2] << 16 |
				data[3] << 24;
			a[i].v.i.hi = 0;
			data += 4;
		}
	}

	return ctx->number;
}

int 
_pbcC_open(pbc_ctx _ctx , void *buffer, int size) {
	struct context * ctx = (struct context *)_ctx;
	ctx->buffer = buffer;
	ctx->size = size;

	if (buffer == NULL || size == 0) {
		ctx->number = 0;
		ctx->a = NULL;
		return 0;
	}

	struct atom * a = (struct atom *)(ctx + 1);

	int i;
	int start = 0;

	ctx->a = a;

	for (i=0;i<INNER_ATOM;i++) {
		if (size == 0)
			break;
		char * next = wiretype_decode(buffer, size , &a[i] , start);
		if (next == NULL)
			return -i;
		start += next - (char *)buffer;
		size -= next - (char *)buffer;
		buffer = next;
	}

	if (size > 0) {
		int cap = 64;
		ctx->a = malloc(cap * sizeof(struct atom));
		while (size > 0) {
			if (i >= cap) {
				cap = cap + 64;
				ctx->a = realloc(ctx->a, cap * sizeof(struct atom));
				continue;
			}
			char * next = wiretype_decode(buffer, size , &ctx->a[i] , start);
			if (next == NULL) {
				return -i;
			}
			start += next - (char *)buffer;
			size -= next - (char *)buffer;
			buffer = next;
			++i;
		}
		memcpy(ctx->a, a , sizeof(struct atom) * INNER_ATOM);
	}
	ctx->number = i;

	return i;
}


void 
_pbcC_close(pbc_ctx _ctx) {
	struct context * ctx = (struct context *)_ctx;
	if (ctx->a != NULL && (struct atom *)(ctx+1) != ctx->a) {
		free(ctx->a);
		ctx->a = NULL;
	}
}