/*
 * Copyright (C) 2007 iptelorg GmbH
 *
 * This file is part of Kamailio, a free SIP server.
 *
 * Kamailio 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
 *
 * Kamailio 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 the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
 *
 */

#include <string.h>

#include "../ut.h"
#include "../mem/mem.h"
#include "cfg_struct.h"
#include "cfg_ctx.h"
#include "cfg_script.h"
#include "cfg.h"

/*! \brief declares a new cfg group
 *
 * handler is set to the memory area where the variables are stored
 * \return value is -1 on error
 */
int cfg_declare(char *group_name, cfg_def_t *def, void *values, int def_size,
		void **handle)
{
	int i, num, size, group_name_len;
	cfg_mapping_t *mapping = NULL;
	cfg_group_t *group;
	int types;

	if(def == NULL || def[0].name == NULL)
		return -1;

	/* check the number of the variables */
	for(num = 0; def[num].name; num++)
		;

	mapping = (cfg_mapping_t *)pkg_malloc(sizeof(cfg_mapping_t) * num);
	if(!mapping) {
		PKG_MEM_ERROR;
		goto error;
	}
	memset(mapping, 0, sizeof(cfg_mapping_t) * num);
	types = 0;
	/* calculate the size of the memory block that has to
	be allocated for the cfg variables, and set the content of the
	cfg_mapping array the same time */
	for(i = 0, size = 0; i < num; i++) {
		mapping[i].def = &(def[i]);
		mapping[i].name_len = strlen(def[i].name);
		mapping[i].pos = i;
		/* record all the types for sanity checks */
		types |= 1 << CFG_VAR_MASK(def[i].type);

		/* padding depends on the type of the next variable */
		switch(CFG_VAR_MASK(def[i].type)) {

			case CFG_VAR_INT:
				size = ROUND_INT(size);
				mapping[i].offset = size;
				size += sizeof(int);
				break;

			case CFG_VAR_STRING:
			case CFG_VAR_POINTER:
				size = ROUND_POINTER(size);
				mapping[i].offset = size;
				size += sizeof(char *);
				break;

			case CFG_VAR_STR:
				size = ROUND_POINTER(size);
				mapping[i].offset = size;
				size += sizeof(str);
				break;

			default:
				LM_ERR("%s.%s: unsupported variable type\n", group_name,
						def[i].name);
				goto error;
		}

		/* verify the type of the input */
		if(CFG_INPUT_MASK(def[i].type) == 0) {
			def[i].type |= CFG_VAR_MASK(def[i].type) << CFG_INPUT_SHIFT;
		} else {
			if((CFG_INPUT_MASK(def[i].type)
					   != CFG_VAR_MASK(def[i].type) << CFG_INPUT_SHIFT)
					&& (def[i].on_change_cb == 0)) {
				LM_ERR("%s.%s: variable and input types are "
					   "different, but no callback is defined for conversion\n",
						group_name, def[i].name);
				goto error;
			}
		}

		if(CFG_INPUT_MASK(def[i].type) > CFG_INPUT_STR) {
			LM_ERR("%s.%s: unsupported input type\n", group_name, def[i].name);
			goto error;
		}

		if(def[i].type & CFG_ATOMIC) {
			if(CFG_VAR_MASK(def[i].type) != CFG_VAR_INT) {
				LM_ERR("%s.%s: atomic change is allowed "
					   "only for integer types\n",
						group_name, def[i].name);
				goto error;
			}
			if(def[i].on_set_child_cb) {
				LM_ERR("%s.%s: per-child process callback "
					   "does not work together with atomic change\n",
						group_name, def[i].name);
				goto error;
			}
		}
	}

	/* fix the computed size (char*, str or pointer members will force
	   structure padding to multiple of sizeof(pointer)) */
	if(types
			& ((1 << CFG_VAR_STRING) | (1 << CFG_VAR_STR)
					| (1 << CFG_VAR_POINTER)))
		size = ROUND_POINTER(size);
	/* minor validation */
	if(size != def_size) {
		LM_ERR("the specified size (%i) of the config "
			   "structure does not equal with the calculated size (%i), check "
			   "whether "
			   "the variable types are correctly defined!\n",
				def_size, size);
		goto error;
	}

	/* The cfg variables are ready to use, let us set the handle
	before passing the new definitions to the drivers.
	We make the interface usable for the fixup functions
	at this step
	cfg_set_group() and cfg_new_group() need the handle to be set because
	they save its original value. */
	*handle = values;

	group_name_len = strlen(group_name);
	/* check for duplicates */
	if((group = cfg_lookup_group(group_name, group_name_len))) {
		if(group->dynamic != CFG_GROUP_UNKNOWN) {
			/* conflict with another module/core group, or with a dynamic group */
			LM_ERR("configuration group has been already declared: %s\n",
					group_name);
			goto error;
		}
		/* An empty group is found which does not have any variable yet */
		cfg_set_group(group, num, mapping, values, size, handle);
	} else {
		/* create a new group
		I will allocate memory in shm mem for the variables later in a single block,
		when we know the size of all the registered groups. */
		if(!(group = cfg_new_group(group_name, group_name_len, num, mapping,
					 values, size, handle)))
			goto error;
	}
	group->dynamic = CFG_GROUP_STATIC;

	/* notify the drivers about the new config definition */
	cfg_notify_drivers(group_name, group_name_len, def);

	LM_DBG("new config group has been registered: '%s' (num=%d, size=%d)\n",
			group_name, num, size);

	return 0;

error:
	if(mapping)
		pkg_free(mapping);
	LM_ERR("failed to register the config group: %s\n", group_name);

	return -1;
}

/* declares a single variable with integer type */
int cfg_declare_int(char *group_name, char *var_name, int val, int min, int max,
		char *descr)
{
	cfg_script_var_t *var;

	if((var = new_cfg_script_var(group_name, var_name, CFG_VAR_INT, descr))
			== NULL)
		return -1;

	var->val.i = val;
	var->min = min;
	var->max = max;

	return 0;
}

/* declares a single variable with str type */
int cfg_declare_str(char *group_name, char *var_name, char *val, char *descr)
{
	cfg_script_var_t *var;
	int len;

	if((var = new_cfg_script_var(group_name, var_name, CFG_VAR_STR, descr))
			== NULL)
		return -1;

	if(val) {
		len = strlen(val);
		var->val.s.s = (char *)pkg_malloc(sizeof(char) * (len + 1));
		if(!var->val.s.s) {
			PKG_MEM_ERROR;
			return -1;
		}
		memcpy(var->val.s.s, val, len + 1);
		var->val.s.len = len;
	} else {
		var->val.s.s = NULL;
		var->val.s.len = 0;
	}

	return 0;
}

/* Add a variable to a group instance with integer type.
 * The group instance is created if it does not exist.
 * wrapper function for new_add_var()
 */
int cfg_ginst_var_int(
		char *group_name, unsigned int group_id, char *var_name, int val)
{
	str gname, vname;

	gname.s = group_name;
	gname.len = strlen(group_name);
	vname.s = var_name;
	vname.len = strlen(var_name);

	return new_add_var(
			&gname, group_id, &vname, (void *)(long)val, CFG_VAR_INT);
}

/* Add a variable to a group instance with string type.
 * The group instance is created if it does not exist.
 * wrapper function for new_add_var()
 */
int cfg_ginst_var_string(
		char *group_name, unsigned int group_id, char *var_name, char *val)
{
	str gname, vname;

	gname.s = group_name;
	gname.len = strlen(group_name);
	vname.s = var_name;
	vname.len = strlen(var_name);

	return new_add_var(&gname, group_id, &vname, (void *)val, CFG_VAR_STRING);
}

/* Create a new group instance.
 * wrapper function for new_add_var()
 */
int cfg_new_ginst(char *group_name, unsigned int group_id)
{
	str gname;

	gname.s = group_name;
	gname.len = strlen(group_name);

	return new_add_var(
			&gname, group_id, NULL /* var */, NULL /* val */, 0 /* type */);
}

/* returns the handle of a cfg group */
void **cfg_get_handle(char *gname)
{
	cfg_group_t *group;

	group = cfg_lookup_group(gname, strlen(gname));
	if(!group || (group->dynamic != CFG_GROUP_STATIC))
		return NULL;

	return group->handle;
}


/* Set the group_id pointer based on the group string.
 * The string is either "group_name", or "group_name[group_id]"
 * *group_id is set to null in the former case.
 * Warning: changes the group string
 */
int cfg_get_group_id(str *group, unsigned int **group_id)
{
	static unsigned int id;
	str s;

	if(!group->s || (group->s[group->len - 1] != ']')) {
		*group_id = NULL;
		return 0;
	}

	s.s = group->s + group->len - 2;
	s.len = 0;
	while((s.s > group->s) && (*s.s != '[')) {
		s.s--;
		s.len++;
	}
	if(s.s == group->s) /* '[' not found */
		return -1;
	group->len = s.s - group->s;
	s.s++;
	if(!group->len || !s.len)
		return -1;
	if(str2int(&s, &id))
		return -1;

	*group_id = &id;
	return 0;
}
