quakeforge/libs/ruamoko/rua_string.c

/*
	bi_string.c

	QuakeC string api

	Copyright (C) 2004 Bill Currie

	Author: Bill Currie <bill@taniwha.org>
	Date: 2004/1/6

	This program is free software; you can redistribute it and/or
	modify it under the terms of the GNU General Public License
	as published by the Free Software Foundation; either version 2
	of the License, or (at your option) any later version.

	This program is distributed in the hope that it will be useful,
	but WITHOUT ANY WARRANTY; without even the implied warranty of
	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.

	See the GNU General Public License for more details.

	You should have received a copy of the GNU General Public License
	along with this program; if not, write to:

		Free Software Foundation, Inc.
		59 Temple Place - Suite 330
		Boston, MA  02111-1307, USA

*/
#ifdef HAVE_CONFIG_H
# include "config.h"
#endif

#ifdef HAVE_STRING_H
# include <string.h>
#endif
#ifdef HAVE_STRINGS_H
# include <strings.h>
#endif
#include <stdlib.h>
#include <ctype.h>
#include <stdlib.h>

#include "qfalloca.h"

#if defined(_WIN32) && defined(HAVE_MALLOC_H)
#include <malloc.h>
#endif

#include "QF/dstring.h"
#include "QF/progs.h"

#include "rua_internal.h"

static void
bi_strlen (progs_t *pr, void *data)
{
	const char	*s;

	s = P_GSTRING (pr, 0);
	R_INT (pr) = strlen(s);
}

void
RUA_Sprintf (progs_t *pr, dstring_t *dstr, const char *func, int fmt_arg)
{
	const char *fmt = P_GSTRING (pr, fmt_arg);
	int         count = pr->pr_argc - (fmt_arg + 1);
	pr_type_t **args = pr->pr_params + (fmt_arg + 1);

	if (pr->progs->version == PROG_VERSION) {
		__auto_type va_list = &P_PACKED (pr, pr_va_list_t, (fmt_arg + 1));
		count = va_list->count;
		if (count) {
			args = alloca (count * sizeof (pr_type_t *));
			for (int i = 0; i < count; i++) {
				args[i] = &pr->pr_globals[va_list->list + i * 4];
			}
		} else {
			args = 0;
		}
	}

	PR_Sprintf (pr, dstr, func, fmt, count, args);
}

static void
bi_sprintf (progs_t *pr, void *data)
{
	dstring_t  *dstr;

	dstr = dstring_newstr ();
	RUA_Sprintf (pr, dstr, "sprintf", 0);
	RETURN_STRING (pr, dstr->str);
	dstring_delete (dstr);
}

static void
bi_vsprintf (progs_t *pr, void *data)
{
	const char *fmt = P_GSTRING (pr, 0);
	__auto_type args = &P_PACKED (pr, pr_va_list_t, 1);
	pr_type_t  *list_start = PR_GetPointer (pr, args->list);
	pr_type_t **list = alloca (args->count * sizeof (*list));
	dstring_t  *dstr;

	for (int i = 0; i < args->count; i++) {
		list[i] = list_start + i * pr->pr_param_size;
	}

	dstr = dstring_newstr ();
	PR_Sprintf (pr, dstr, "bi_vsprintf", fmt, args->count, list);
	RETURN_STRING (pr, dstr->str);
	dstring_delete (dstr);
}

static void
bi_str_new (progs_t *pr, void *data)
{
	R_STRING (pr) = PR_NewMutableString (pr);
}

static void
bi_str_unmutable (progs_t *pr, void *data)
{
	RETURN_STRING (pr, P_GSTRING (pr, 0));
}

static void
bi_str_free (progs_t *pr, void *data)
{
	PR_FreeString (pr, P_STRING (pr, 0));
}

static void
bi_str_hold (progs_t *pr, void *data)
{
	pr_string_t str = P_STRING (pr, 0);
	PR_HoldString (pr, str);
	R_STRING (pr) = str;
}

static void
bi_str_valid (progs_t *pr, void *data)
{
	R_INT (pr) = PR_StringValid (pr, P_STRING (pr, 0));
}

static void
bi_str_mutable (progs_t *pr, void *data)
{
	R_INT (pr) = PR_StringMutable (pr, P_STRING (pr, 0));
}

static void
bi_str_copy (progs_t *pr, void *data)
{
	dstring_t  *dst = P_DSTRING (pr, 0);
	const char *src = P_GSTRING (pr, 1);

	dstring_copystr (dst, src);
	R_STRING (pr) = P_STRING (pr, 0);
}

static void
bi_str_cat (progs_t *pr, void *data)
{
	dstring_t  *dst = P_DSTRING (pr, 0);
	const char *src = P_GSTRING (pr, 1);

	dstring_appendstr (dst, src);
	R_STRING (pr) = P_STRING (pr, 0);
}

static void
bi_str_clear (progs_t *pr, void *data)
{
	dstring_t  *str = P_DSTRING (pr, 0);

	dstring_clearstr (str);
	R_STRING (pr) = P_STRING (pr, 0);
}

static void
str_mid (progs_t *pr, const char *str, int pos, int end, int size)
{
	char       *temp;

	R_STRING (pr) = 0;
	if (pos < 0)
		pos += size;
	if (end < 0)
		end += size;
	if (end > size)
		end = size;
	if (pos < 0 || pos >= size || end <= pos)
		return;
	if (end == size) {
		R_STRING (pr) = str + pos - pr->pr_strings;
		return;
	}
	temp = alloca (end - pos + 1);
	strncpy (temp, str + pos, end - pos);
	temp[end - pos] = 0;
	RETURN_STRING (pr, temp);
}

static void
bi_str_mid_2 (progs_t *pr, void *data)
{
	const char *str = P_GSTRING (pr, 0);
	int         pos = P_INT (pr, 1);
	int         size = strlen (str);

	str_mid (pr, str, pos, size, size);
}

static void
bi_str_mid_3 (progs_t *pr, void *data)
{
	const char *str = P_GSTRING (pr, 0);
	int         pos = P_INT (pr, 1);
	int         end = P_INT (pr, 2);
	int         size = strlen (str);

	str_mid (pr, str, pos, end, size);
}

static void
bi_str_str (progs_t *pr, void *data)
{
	const char *haystack = P_GSTRING (pr, 0);
	const char *needle = P_GSTRING (pr, 1);
	char       *res = strstr (haystack, needle);

	R_INT (pr) = -1;
	if (res) {
		R_INT (pr) = res - haystack;
	}
}

static void
bi_str_char (progs_t *pr, void *data)
{
	const char *str = P_GSTRING (pr, 0);
	int         ind = P_INT (pr, 1);

	if (ind < 0) {
		PR_RunError (pr, "negative index to str_char");
	}
	R_INT (pr) = str[ind];
}

static void
bi_str_quote (progs_t *pr, void *data)
{
	const char *str = P_GSTRING (pr, 0);
	// can have up to 4 chars per char (a -> \x61)
	char       *quote = alloca (strlen (str) * 4 + 1);
	char       *q = quote;
	char        c;
	int         h;

	while ((c = *str++)) {
		if (c >= ' ' && c < 127 && c != '\"') {
			*q++ = c;
		} else {
			*q++ = '\\';
			switch (c) {
				case '\a': c = 'a'; break;
				case '\b': c = 'b'; break;
				case '\f': c = 'f'; break;
				case '\n': c = 'n'; break;
				case '\r': c = 'r'; break;
				case '\t': c = 't'; break;
				case '\"': c = '\"'; break;
				default:
					*q++ = 'x';
					h = (c & 0xf0) >> 4;
					*q++ = h > 9 ? h + 'a' - 10 : h + '0';
					h = (c & 0x0f);
					c = h > 9 ? h + 'a' - 10 : h + '0';
					break;
			}
			*q++ = c;
		}
	}
	*q++ = 0;

	RETURN_STRING (pr, quote);
}

static void
bi_str_lower (progs_t *pr, void *data)
{
	const char *str = P_GSTRING (pr, 0);
	char       *lower = alloca (strlen (str) + 1);
	char       *l = lower;
	byte        c;

	while ((c = *str++)) {
		*l++ = tolower (c);
	}
	*l++ = 0;

	RETURN_STRING (pr, lower);
}

static void
bi_str_upper (progs_t *pr, void *data)
{
	const char *str = P_GSTRING (pr, 0);
	char       *upper = alloca (strlen (str) + 1);
	char       *l = upper;
	byte        c;

	while ((c = *str++)) {
		*l++ = toupper (c);
	}
	*l++ = 0;

	RETURN_STRING (pr, upper);
}

static void
bi_strtod (progs_t *pr, void *data)
{
	const char *str = P_GSTRING (pr, 0);
	pr_type_t  *end = P_GPOINTER (pr, 1);
	char       *end_ptr;
	R_DOUBLE (pr) = strtod (str, &end_ptr);
	if (end) {
		end->value = end_ptr - str;
	}
}

static void
bi_strtof (progs_t *pr, void *data)
{
	const char *str = P_GSTRING (pr, 0);
	pr_type_t  *end = P_GPOINTER (pr, 1);
	char       *end_ptr;
	R_FLOAT (pr) = strtof (str, &end_ptr);
	if (end) {
		end->value = end_ptr - str;
	}
}

static void
bi_strtol (progs_t *pr, void *data)
{
	const char *str = P_GSTRING (pr, 0);
	pr_type_t  *end = P_GPOINTER (pr, 1);
	char       *end_ptr;
	R_LONG (pr) = strtol (str, &end_ptr, P_INT (pr, 2));
	if (end) {
		end->value = end_ptr - str;
	}
}

static void
bi_strtoul (progs_t *pr, void *data)
{
	const char *str = P_GSTRING (pr, 0);
	pr_type_t  *end = P_GPOINTER (pr, 1);
	char       *end_ptr;
	R_ULONG (pr) = strtoul (str, &end_ptr, P_INT (pr, 2));
	if (end) {
		end->value = end_ptr - str;
	}
}

#define bi(x,np,params...) {#x, bi_##x, -1, np, {params}}
#define p(type) PR_PARAM(type)
#define P(a, s) { .size = (s), .alignment = BITOP_LOG2 (a), }
static builtin_t builtins[] = {
	bi(strlen,      1, p(string)),
	bi(sprintf,     -2, p(string)),
	bi(vsprintf,    2, p(string), P(1, 2)),
	bi(str_new,     0),
	bi(str_unmutable,   1, p(string)),
	bi(str_free,    1, p(string)),
	bi(str_hold,    1, p(string)),
	bi(str_valid,   1, p(string)),
	bi(str_mutable, 1, p(string)),
	bi(str_copy,    2, p(string), p(string)),
	bi(str_cat,     2, p(string), p(string)),
	bi(str_clear,   1, p(string)),
	{"str_mid|*i",	bi_str_mid_2,		-1, 2, {p(string), p(int)}},
	{"str_mid|*ii",	bi_str_mid_3,		-1, 3, {p(string), p(int), p(int)}},
	bi(str_str,     2, p(string), p(string)),
	bi(str_char,    2, p(string), p(int)),
	bi(str_quote,   1, p(string)),
	bi(str_lower,   1, p(string)),
	bi(str_upper,   1, p(string)),
	bi(strtod,      1, p(string)),
	bi(strtof,      1, p(string)),
	bi(strtol,      1, p(string)),
	bi(strtoul,     1, p(string)),
	{0}
};

void
RUA_String_Init (progs_t *pr, int secure)
{
	PR_RegisterBuiltins (pr, builtins, 0);
}