/* $NetBSD: utils.c,v 1.21 2012/03/20 18:50:30 matt Exp $ */

/*-
 * Copyright (c) 2002, 2003 The NetBSD Foundation, Inc.
 * All rights reserved.
 *
 * This code is derived from software contributed to The NetBSD Foundation
 * by Roland C. Dowdeswell.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
 * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
 * POSSIBILITY OF SUCH DAMAGE.
 */

#include <sys/cdefs.h>
#ifndef lint
__RCSID("$NetBSD: utils.c,v 1.21 2012/03/20 18:50:30 matt Exp $");
#endif

#include <sys/param.h>

#include <stdlib.h>
#include <string.h>
#include <err.h>
#include <util.h>

/* include the resolver gunk in order that we can use b64 routines */
#include <netinet/in.h>
#include <arpa/nameser.h>
#include <resolv.h>

#include "utils.h"


/* just strsep(3), but skips empty fields. */

static char *
strsep_getnext(char **stringp, const char *delim)
{
	char	*ret;

	ret = strsep(stringp, delim);
	while (ret && index(delim, *ret))
		ret = strsep(stringp, delim);
	return ret;
}

/*
 * this function returns a dynamically sized char ** of the words
 * in the line.  the caller is responsible for both free(3)ing
 * each word and the superstructure by calling words_free().
 */
char **
words(const char *line, int *num)
{
	int	  i = 0;
	int	  nwords = 0;
	char	 *cur;
	char	**ret;
	const char	 *tmp;
	char	 *tmp1, *tmpf;

	*num = 0;
	tmp = line;
	if (tmp[0] == '\0')
		return NULL;
	while (tmp[0]) {
		if ((tmp[1] == ' ' || tmp[1] == '\t' || tmp[1] == '\0') &&
		    (tmp[0] != ' ' && tmp[0] != '\t'))
			nwords++;
		tmp++;
	}
	ret = emalloc((nwords+1) * sizeof(char *));
	tmp1 = tmpf = estrdup(line);
	while ((cur = strsep_getnext(&tmpf, " \t")) != NULL)
		ret[i++] = estrdup(cur);
	ret[i] = NULL;
	free(tmp1);
	*num = nwords;
	return ret;
}

void
words_free(char **w, int num)
{
	int	i;

	for (i=0; i < num; i++)
		free(w[i]);
}

/*
 * this is a simple xor that has the same calling conventions as
 * memcpy(3).
 */

void
memxor(void *res, const void *src, size_t len)
{
	char *r;
	const char *s;
	size_t i;

	r = res;
	s = src;
	for (i = 0; i < len; i++)
		r[i] ^= s[i];
}

/*
 * well, a very simple set of string functions...
 *
 * The goal here is basically to manage length encoded strings,
 * but just for safety we nul terminate them anyway.
 */

/* for now we use a very simple encoding */

struct string {
	char	*text;
	size_t	 length;
};

string_t *
string_zero(void)
{
	string_t *out;

	out = emalloc(sizeof(*out));
	out->length = 0;
	out->text = NULL;
	return out;
}

string_t *
string_new(const char *intext, size_t inlength)
{
	string_t *out;

	out = emalloc(sizeof(*out));
	out->length = inlength;
	out->text = emalloc(out->length + 1);
	(void)memcpy(out->text, intext, out->length);
	out->text[out->length] = '\0';
	return out;
}

string_t *
string_dup(const string_t *in)
{

	return string_new(in->text, in->length);
}

void
string_free(string_t *s)
{

	if (!s)
		return;
	free(s->text);
	free(s);
}

void
string_assign(string_t **lhs, string_t *rhs)
{

	string_free(*lhs);
	*lhs = rhs;
}

string_t *
string_add(const string_t *a1, const string_t *a2)
{
	string_t *sum;

	sum = emalloc(sizeof(*sum));
	sum->length = a1->length + a2->length;
	sum->text = emalloc(sum->length + 1);
	(void)memcpy(sum->text, a1->text, a1->length);
	(void)memcpy(sum->text + a1->length, a2->text, a2->length);
	sum->text[sum->length] = '\0';
	return sum;
}

string_t *
string_add_d(string_t *a1, string_t *a2)
{
	string_t *sum;

	sum = string_add(a1, a2);
	string_free(a1);
	string_free(a2);
	return sum;
}

string_t *
string_fromcharstar(const char *in)
{

	return string_new(in, strlen(in));
}

const char *
string_tocharstar(const string_t *in)
{

	return in->text;
}

string_t *
string_fromint(int in)
{
	string_t *ret;

	ret = emalloc(sizeof(*ret));
	ret->length = asprintf(&ret->text, "%d", in);
	if (ret->text == NULL)
		err(1, NULL);
	return ret;
}

void
string_fprint(FILE *f, const string_t *s)
{
	(void)fwrite(s->text, s->length, 1, f);
}

struct bits {
	size_t	 length;
	char	*text;
};

bits_t *
bits_new(const void *buf, size_t len)
{
	bits_t	*b;

	b = emalloc(sizeof(*b));
	b->length = len;
	b->text = emalloc(BITS2BYTES(b->length));
	(void)memcpy(b->text, buf, BITS2BYTES(b->length));
	return b;
}

bits_t *
bits_dup(const bits_t *in)
{

	return bits_new(in->text, in->length);
}

void
bits_free(bits_t *b)
{

	if (!b)
		return;
	free(b->text);
	free(b);
}

void
bits_assign(bits_t **lhs, bits_t *rhs)
{

	bits_free(*lhs);
	*lhs = rhs;
}

const void *
bits_getbuf(bits_t *in)
{

	return in->text;
}

size_t
bits_len(bits_t *in)
{

	return in->length;
}

int
bits_match(const bits_t *b1, const bits_t *b2)
{
	size_t i;

	if (b1->length != b2->length)
		return 0;

	for (i = 0; i < BITS2BYTES(b1->length); i++)
		if (b1->text[i] != b2->text[i])
			return 0;

	return 1;
}

bits_t *
bits_xor(const bits_t *x1, const bits_t *x2)
{
	bits_t	*b;
	size_t	 i;

	b = emalloc(sizeof(*b));
	b->length = MAX(x1->length, x2->length);
	b->text = ecalloc(1, BITS2BYTES(b->length));
	for (i=0; i < BITS2BYTES(MIN(x1->length, x2->length)); i++)
		b->text[i] = x1->text[i] ^ x2->text[i];
	return b;
}

bits_t *
bits_xor_d(bits_t *x1, bits_t *x2)
{
	bits_t	*ret;

	ret = bits_xor(x1, x2);
	bits_free(x1);
	bits_free(x2);
	return ret;
}

/*
 * bits_decode() reads an encoded base64 stream.  We interpret
 * the first 32 bits as an unsigned integer in network byte order
 * specifying the number of bits in the stream to give a little
 * resilience.
 */

bits_t *
bits_decode(const string_t *in)
{
	bits_t	*ret;
	size_t	 len;
	size_t	 nbits;
	u_int32_t	*tmp;

	len = in->length;
	tmp = emalloc(len);

	len = __b64_pton(in->text, (void *)tmp, len);

	if (len == (size_t)-1) {
		warnx("bits_decode: mangled base64 stream");
		warnx("  %s", in->text);
		free(tmp);
		return NULL;
	}

	nbits = ntohl(*tmp);
	if (nbits > (len - sizeof(*tmp)) * NBBY) {
		warnx("bits_decode: encoded bits claim to be "
		    "longer than they are (nbits=%zu, stream len=%zu bytes)",
		    nbits, len);
		free(tmp);
		return NULL;
	}

	ret = bits_new(tmp + 1, nbits);
	free(tmp);
	return ret;
}

bits_t *
bits_decode_d(string_t *in)
{
	bits_t *ret;

	ret = bits_decode(in);
	string_free(in);
	return ret;
}

string_t *
bits_encode(const bits_t *in)
{
	string_t *ret;
	size_t	 len;
	char	*out;
	u_int32_t *tmp;

	if (!in)
		return NULL;

	/* compute the total size of the input stream */
	len = BITS2BYTES(in->length) + sizeof(*tmp);

	tmp = emalloc(len);
	out = emalloc(len * 2);
	/* stuff the length up front */
	*tmp = htonl(in->length);
	(void)memcpy(tmp + 1, in->text, len - sizeof(*tmp));

	if ((len = __b64_ntop((void *)tmp, len, out, len * 2)) == (size_t)-1) {
		free(out);
		free(tmp);
		return NULL;
	}
	ret = string_new(out, len);
	free(tmp);
	free(out);
	return ret;
}

string_t *
bits_encode_d(bits_t *in)
{
	string_t *ret;

	ret = bits_encode(in);
	bits_free(in);
	return ret;
}

bits_t *
bits_fget(FILE *f, size_t len)
{
	bits_t	*bits;
	int	 ret;

	bits = emalloc(sizeof(*bits));
	bits->length = len;
	bits->text = emalloc(BITS2BYTES(bits->length));
	ret = fread(bits->text, BITS2BYTES(bits->length), 1, f);
	if (ret != 1) {
		bits_free(bits);
		return NULL;
	}
	return bits;
}

bits_t *
bits_cget(const char *fn, size_t len)
{
	bits_t	*bits;
	FILE	*f;

	f = fopen(fn, "r");
	if (!f)
		return NULL;

	bits = bits_fget(f, len);
	(void)fclose(f);
	return bits;
}

bits_t *
bits_getrandombits(size_t len, int hard)
{

	return bits_cget((hard ? "/dev/random" : "/dev/urandom"), len);
}

void
bits_fprint(FILE *f, const bits_t *bits)
{
	string_t *s;

	s = bits_encode(bits);
	string_fprint(f, s);
	free(s);
}