#include "sdtxt.h"
#include <stdlib.h>
#include <string.h>

static size_t _sd2txt_len(const char *key, char *val)
{
	size_t ret = strlen(key);

	if (!*val)
		return ret;

	ret += strlen(val);
	ret++;

	return ret;
}

static void _sd2txt_count(xht_t *h, char *key, void *val, void *arg)
{
	int *count = (int *)arg;

	*count += (int)_sd2txt_len(key, (char *)val) + 1;
}

static void _sd2txt_write(xht_t *h, char *key, void *val, void *arg)
{
	unsigned char **txtp = (unsigned char **)arg;
	char *cval = (char *)val;

	/* Copy in lengths, then strings */
	**txtp = (unsigned char)_sd2txt_len(key, (char *)val);
	(*txtp)++;
	memcpy(*txtp, key, strlen(key));
	*txtp += strlen(key);
	if (!*cval)
		return;

	**txtp = '=';
	(*txtp)++;
	memcpy(*txtp, cval, strlen(cval));
	*txtp += strlen(cval);
}

unsigned char *sd2txt(xht_t *h, int *len)
{
	unsigned char *buf, *raw;

	*len = 0;

	xht_walk(h, _sd2txt_count, (void *)len);
	if (!*len) {
		*len = 1;
		buf = (unsigned char *)MDNSD_malloc(1);
		*buf = 0;
		return buf;
	}

	raw = buf = (unsigned char *)MDNSD_malloc((size_t)(*len));
	xht_walk(h, _sd2txt_write, &buf);

	return raw;
}

xht_t *txt2sd(const unsigned char *txt, int len)
{
	char key[256];
	xht_t *h = NULL;

	if (txt == 0 || len == 0 || *txt == 0)
		return NULL;

	h = xht_new(23);

	/* Loop through data breaking out each block, storing into hashtable */
	for (; len > 0 && *txt <= len; len -= *txt, txt += *txt + 1) {
		char* val;
		if (*txt == 0)
			break;

		memcpy(key, txt + 1, *txt);
		key[*txt] = 0;
		if ((val = strchr(key, '=')) != 0) {
			*val = 0;
			val++;
		}
		if (val != NULL)
			xht_store(h, key, (int)strlen(key), val, (int)strlen(val));
		if (*txt +1 > len)
		    break;
	}

	return h;
}