libpsyc/src/packet.c

#include "lib.h"
#include <psyc/syntax.h>
#include <psyc/packet.h>

#include <math.h>

static inline
size_t psyc_getNumLength(size_t n)
{
	return n < 10 ? 1 : log10(n) + 1;
}

inline
psycListFlag psyc_checkListLength (psycList *list)
{
	psycListFlag flag = PSYC_LIST_NO_LENGTH;
	size_t i, length = 0;

	for (i = 0; i < list->num_elems; i++)
	{
		psycString *elem = &list->elems[i];
		length += 1 + elem->length; // |elem
		if (length > PSYC_MODIFIER_SIZE_THRESHOLD ||
				memchr(elem->ptr, (int)'|', elem->length) ||
				memchr(elem->ptr, (int)'\n', elem->length))
		{
			flag = PSYC_LIST_NEED_LENGTH;
			break;
		}
	}

	return flag;
}

inline
psycListFlag psyc_getListLength (psycList *list)
{
	size_t i, length = 0;

	if (list->flag == PSYC_LIST_NEED_LENGTH)
	{
		for (i = 0; i < list->num_elems; i++)
		{
			if (i > 0)
				length++; // |
			length += psyc_getNumLength(list->elems[i].length) + 1 + list->elems[i].length; // length SP elem
		}
	}
	else
	{
		for (i = 0; i < list->num_elems; i++)
			length += 1 + list->elems[i].length; // |elem
	}

	return length;
}

inline
psycList psyc_newList(psycString *elems, size_t num_elems, psycListFlag flag)
{
	psycList list = {num_elems, elems, 0, flag};

	if (flag == PSYC_LIST_CHECK_LENGTH) // check if list elements need length
		list.flag = psyc_checkListLength(&list);

	list.length = psyc_getListLength(&list);
	return list;
}


inline
size_t psyc_getModifierLength (psycModifier *m)
{
	size_t length = 1 + // oper
		m->name.length + 1 + // name\t
		m->value.length + 1; // value\n

	if (m->flag == PSYC_MODIFIER_NEED_LENGTH) // add length of length if needed
		length += psyc_getNumLength(m->value.length) + 1; // SP length

	return length;
}

inline
psycPacketFlag psyc_checkPacketLength(psycPacket *p)
{
	if (p->data.length == 1 && p->data.ptr[0] == C_GLYPH_PACKET_DELIMITER)
		return PSYC_PACKET_NEED_LENGTH;

	if (p->data.length > PSYC_CONTENT_SIZE_THRESHOLD)
		return PSYC_PACKET_NEED_LENGTH;

	int i;
	// if any entity modifiers need length it is possible they contain
	// a packet terminator, thus the content should have a length as well
	for (i = 0; i < p->entity.lines; i++)
		if (p->entity.modifiers[i].flag == PSYC_MODIFIER_NEED_LENGTH)
			return PSYC_PACKET_NEED_LENGTH;

	if (memmem(p->data.ptr, p->data.length, PSYC_C2ARG(PSYC_PACKET_DELIMITER)))
		return PSYC_PACKET_NEED_LENGTH;

	return PSYC_PACKET_NO_LENGTH;
}

inline
size_t psyc_setPacketLength(psycPacket *p)
{
	size_t i;
	p->routingLength = 0;
	p->contentLength = 0;

	// add routing header length
	for (i = 0; i < p->routing.lines; i++)
		p->routingLength += psyc_getModifierLength(&(p->routing.modifiers[i]));

	if (p->content.length)
		p->contentLength = p->content.length;
	else
	{
		// add entity header length
		for (i = 0; i < p->entity.lines; i++)
			p->contentLength += psyc_getModifierLength(&(p->entity.modifiers[i]));

		// add length of method, data & delimiter
		if (p->method.length)
			p->contentLength += p->method.length + 1; // method\n
		if (p->data.length)
			p->contentLength += p->data.length + 1; // data\n
	}

	// set total length: routing-header content |\n
	p->length = p->routingLength + p->contentLength + 2;

	if (p->contentLength > 0 || p->flag == PSYC_PACKET_NEED_LENGTH)
		p->length++; // add \n at the start of the content part

	if (p->flag == PSYC_PACKET_NEED_LENGTH) // add length of length if needed
		p->length += psyc_getNumLength(p->contentLength);

	return p->length;
}

inline
psycPacket psyc_newPacket (psycHeader *routing, psycHeader *entity,
                           psycString *method, psycString *data,
                           psycPacketFlag flag)
{
	psycPacket p = {*routing, *entity, *method, *data, {0,0}, 0, 0, flag};

	if (flag == PSYC_PACKET_CHECK_LENGTH) // find out if it needs a length
		p.flag = psyc_checkPacketLength(&p);

	psyc_setPacketLength(&p);
	return p;
}

inline
psycPacket psyc_newPacket2 (psycModifier *routing, size_t routinglen,
                            psycModifier *entity, size_t entitylen,
                            const char *method, size_t methodlen,
                            const char *data, size_t datalen,
                            psycPacketFlag flag)
{
	psycHeader r = {routinglen, routing};
	psycHeader e = {entitylen, entity};
	psycString m = {methodlen, method};
	psycString d = {datalen, data};

	return psyc_newPacket(&r, &e, &m, &d, flag);
}

inline
psycPacket psyc_newRawPacket (psycHeader *routing, psycString *content,
                              psycPacketFlag flag)
{
	psycPacket p = {*routing, {0,0}, {0,0}, {0,0}, *content, 0, 0, flag};

	if (flag == PSYC_PACKET_CHECK_LENGTH) // find out if it needs a length
		p.flag = psyc_checkPacketLength(&p);

	psyc_setPacketLength(&p);
	return p;
}

inline
psycPacket psyc_newRawPacket2 (psycModifier *routing, size_t routinglen,
                               const char *content, size_t contentlen,
                               psycPacketFlag flag)
{
	psycHeader r = {routinglen, routing};
	psycString c = {contentlen, content};

	return psyc_newRawPacket(&r, &c, flag);
}