/**
 * Copyright (C) 2011 Daniel-Constantin Mierla (asipto.com)
 *
 * Copyright (C) 2012 Vicente Hernando Ara (System One: www.systemonenoc.com)
 *     - for: redis array reply support
 *
 * Copyright (C) 2017 Carsten Bock (ng-voice GmbH)
 *     - for: Cluster support
 *
 * 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 <stdio.h>
#include <unistd.h>
#include <stdlib.h>
#include <ctype.h>

#include "../../core/sr_module.h"
#include "../../core/mem/mem.h"
#include "../../core/dprint.h"
#include "../../core/mod_fix.h"
#include "../../core/trim.h"
#include "../../core/kemi.h"

#include "redis_client.h"
#include "api.h"

MODULE_VERSION

/** parameters */

int redis_srv_param(modparam_t type, void *val);
int init_without_redis = 0;
int redis_connect_timeout_param = 1000;
int redis_cmd_timeout_param = 1000;
int redis_cluster_param = 0;
int redis_disable_time_param = 0;
int redis_allowed_timeouts_param = -1;
int redis_flush_on_reconnect_param = 0;
int redis_allow_dynamic_nodes_param = 0;
int ndb_redis_debug = L_DBG;
#ifdef WITH_SSL
char *ndb_redis_ca_path = 0;
#endif

static int w_redis_cmd3(
		struct sip_msg *msg, char *ssrv, char *scmd, char *sres);
static int w_redis_cmd4(
		struct sip_msg *msg, char *ssrv, char *scmd, char *sargv1, char *sres);
static int w_redis_cmd5(struct sip_msg *msg, char *ssrv, char *scmd,
		char *sargv1, char *sargv2, char *sres);
static int w_redis_cmd6(struct sip_msg *msg, char *ssrv, char *scmd,
		char *sargv1, char *sargv2, char *sargv3, char *sres);
static int w_redis_pipe_cmd3(
		struct sip_msg *msg, char *ssrv, char *scmd, char *sres);
static int w_redis_pipe_cmd4(
		struct sip_msg *msg, char *ssrv, char *scmd, char *sargv1, char *sres);
static int w_redis_pipe_cmd5(struct sip_msg *msg, char *ssrv, char *scmd,
		char *sargv1, char *sargv2, char *sres);
static int w_redis_pipe_cmd6(struct sip_msg *msg, char *ssrv, char *scmd,
		char *sargv1, char *sargv2, char *sargv3, char *sres);
static int fixup_redis_cmd6(void **param, int param_no);
static int w_redis_execute(struct sip_msg *msg, char *ssrv);

static int w_redis_free_reply(struct sip_msg *msg, char *res);

static int mod_init(void);
static int child_init(int rank);
static void mod_destroy(void);

int bind_ndb_redis(ndb_redis_api_t *api);

static int pv_get_redisc(
		struct sip_msg *msg, pv_param_t *param, pv_value_t *res);
static int pv_parse_redisc_name(pv_spec_p sp, str *in);
static int pv_get_rediscd(
		struct sip_msg *msg, pv_param_t *param, pv_value_t *res);
static int pv_parse_rediscd_name(pv_spec_p sp, str *in);

/* clang-format off */
static pv_export_t mod_pvs[] = {
	{{"redis", sizeof("redis") - 1}, PVT_OTHER, pv_get_redisc, 0,
			pv_parse_redisc_name, 0, 0, 0},
	{{"redisd", sizeof("redisd") - 1}, PVT_OTHER, pv_get_rediscd, 0,
			pv_parse_rediscd_name, 0, 0, 0},
	{{0, 0}, 0, 0, 0, 0, 0, 0, 0}
};


static cmd_export_t cmds[] = {
	{"redis_cmd", (cmd_function)w_redis_cmd3, 3,
			fixup_redis_cmd6, 0, ANY_ROUTE},
	{"redis_cmd", (cmd_function)w_redis_cmd4, 4, fixup_redis_cmd6, 0,
			ANY_ROUTE},
	{"redis_cmd", (cmd_function)w_redis_cmd5, 5, fixup_redis_cmd6, 0,
			ANY_ROUTE},
	{"redis_cmd", (cmd_function)w_redis_cmd6, 6, fixup_redis_cmd6, 0,
			ANY_ROUTE},
	{"redis_pipe_cmd", (cmd_function)w_redis_pipe_cmd3, 3, fixup_redis_cmd6,
			0, ANY_ROUTE},
	{"redis_pipe_cmd", (cmd_function)w_redis_pipe_cmd4, 4, fixup_redis_cmd6,
			0, ANY_ROUTE},
	{"redis_pipe_cmd", (cmd_function)w_redis_pipe_cmd5, 5, fixup_redis_cmd6,
			0, ANY_ROUTE},
	{"redis_pipe_cmd", (cmd_function)w_redis_pipe_cmd6, 6, fixup_redis_cmd6,
			0, ANY_ROUTE},
	{"redis_execute", (cmd_function)w_redis_execute, 1, fixup_redis_cmd6, 0,
			ANY_ROUTE},
	{"redis_free", (cmd_function)w_redis_free_reply, 1, fixup_spve_null, 0,
			ANY_ROUTE},

	{"bind_ndb_redis", (cmd_function)bind_ndb_redis, 0, 0, 0, 0},

	{0, 0, 0, 0, 0, 0}
};

static param_export_t params[] = {
	{"server", PARAM_STRING | USE_FUNC_PARAM, (void *)redis_srv_param},
	{"init_without_redis", INT_PARAM, &init_without_redis},
	{"connect_timeout", INT_PARAM, &redis_connect_timeout_param},
	{"cmd_timeout", INT_PARAM, &redis_cmd_timeout_param},
	{"cluster", INT_PARAM, &redis_cluster_param},
	{"disable_time", INT_PARAM, &redis_disable_time_param},
	{"allowed_timeouts", INT_PARAM, &redis_allowed_timeouts_param},
	{"flush_on_reconnect", INT_PARAM, &redis_flush_on_reconnect_param},
	{"allow_dynamic_nodes", INT_PARAM, &redis_allow_dynamic_nodes_param},
	{"debug", PARAM_INT, &ndb_redis_debug},
#ifdef WITH_SSL
	{"ca_path", PARAM_STRING, &ndb_redis_ca_path},
#endif
	{0, 0, 0}
};

struct module_exports exports = {
	"ndb_redis",		/* module name */
	DEFAULT_DLFLAGS,	/* dlopen flags */
	cmds,				/* exported functions */
	params,				/* exported parameters */
	0,					/* exported RPC methods */
	mod_pvs,			/* exported pseudo-variables */
	0,					/* response function */
	mod_init,			/* module initialization function */
	child_init,			/* per child init function */
	mod_destroy			/* destroy function */
};
/* clang-format on */

/**
 *
 */
static int mod_init(void)
{
	/*
	 * register the need to be called post-fork of all children
	 * with the special rank PROC_POSTCHILDINIT by main attendant
	 */
	ksr_module_set_flag(KSRMOD_FLAG_POSTCHILDINIT);
	return 0;
}

/* each child get a new connection to the database */
static int child_init(int rank)
{
	/* skip child init for special process ranks */
	if(rank == PROC_INIT || rank == PROC_MAIN)
		return 0;

	if(redisc_init() < 0) {
		LM_ERR("failed to initialize redis connections\n");
		return -1;
	}
	return 0;
}

/**
 *
 */
static void mod_destroy(void)
{
	LM_DBG("cleaning up\n");
	redisc_destroy();
}

/**
 *
 */
static int w_redis_cmd3(struct sip_msg *msg, char *ssrv, char *scmd, char *sres)
{
	str s[3];
	int i;

	if(fixup_get_svalue(msg, (gparam_t *)ssrv, &s[0]) != 0) {
		LM_ERR("no redis server name\n");
		return -1;
	}
	if(fixup_get_svalue(msg, (gparam_t *)scmd, &s[1]) != 0) {
		LM_ERR("no redis command\n");
		return -1;
	}
	for(i = 0; i < s[1].len - 1; i++) {
		if(s[1].s[i] == '%') {
			if(s[1].s[i + 1] == 's' || s[1].s[i + 1] == 'b') {
				LM_ERR("command argument specifier found, but no params\n");
				return -1;
			}
		}
	}
	if(fixup_get_svalue(msg, (gparam_t *)sres, &s[2]) != 0) {
		LM_ERR("no redis reply name\n");
		return -1;
	}

	if(redisc_exec(&s[0], &s[2], &s[1]) < 0)
		return -1;
	return 1;
}

/**
 *
 */
static int w_redis_cmd4(
		struct sip_msg *msg, char *ssrv, char *scmd, char *sargv1, char *sres)
{
	str s[3];
	str arg1;
	char c1;

	if(fixup_get_svalue(msg, (gparam_t *)ssrv, &s[0]) != 0) {
		LM_ERR("no redis server name\n");
		return -1;
	}
	if(fixup_get_svalue(msg, (gparam_t *)scmd, &s[1]) != 0) {
		LM_ERR("no redis command\n");
		return -1;
	}
	if(fixup_get_svalue(msg, (gparam_t *)sargv1, &arg1) != 0) {
		LM_ERR("no argument 1\n");
		return -1;
	}
	if(fixup_get_svalue(msg, (gparam_t *)sres, &s[2]) != 0) {
		LM_ERR("no redis reply name\n");
		return -1;
	}

	STR_VTOZ(arg1.s[arg1.len], c1);
	if(redisc_exec(&s[0], &s[2], &s[1], arg1.s) < 0) {
		STR_ZTOV(arg1.s[arg1.len], c1);
		return -1;
	}
	STR_ZTOV(arg1.s[arg1.len], c1);
	return 1;
}

/**
 *
 */
static int w_redis_cmd5(struct sip_msg *msg, char *ssrv, char *scmd,
		char *sargv1, char *sargv2, char *sres)
{
	str s[3];
	str arg1, arg2;
	char c1, c2;

	if(fixup_get_svalue(msg, (gparam_t *)ssrv, &s[0]) != 0) {
		LM_ERR("no redis server name\n");
		return -1;
	}
	if(fixup_get_svalue(msg, (gparam_t *)scmd, &s[1]) != 0) {
		LM_ERR("no redis command\n");
		return -1;
	}
	if(fixup_get_svalue(msg, (gparam_t *)sargv1, &arg1) != 0) {
		LM_ERR("no argument 1\n");
		return -1;
	}
	if(fixup_get_svalue(msg, (gparam_t *)sargv2, &arg2) != 0) {
		LM_ERR("no argument 2\n");
		return -1;
	}
	if(fixup_get_svalue(msg, (gparam_t *)sres, &s[2]) != 0) {
		LM_ERR("no redis reply name\n");
		return -1;
	}

	STR_VTOZ(arg1.s[arg1.len], c1);
	STR_VTOZ(arg2.s[arg2.len], c2);
	if(redisc_exec(&s[0], &s[2], &s[1], arg1.s, arg2.s) < 0) {
		STR_ZTOV(arg1.s[arg1.len], c1);
		STR_ZTOV(arg2.s[arg2.len], c2);
		return -1;
	}
	STR_ZTOV(arg1.s[arg1.len], c1);
	STR_ZTOV(arg2.s[arg2.len], c2);
	return 1;
}

/**
 *
 */
static int w_redis_cmd6(struct sip_msg *msg, char *ssrv, char *scmd,
		char *sargv1, char *sargv2, char *sargv3, char *sres)
{
	str s[3];
	str arg1, arg2, arg3;
	char c1, c2, c3;

	if(fixup_get_svalue(msg, (gparam_t *)ssrv, &s[0]) != 0) {
		LM_ERR("no redis server name\n");
		return -1;
	}
	if(fixup_get_svalue(msg, (gparam_t *)scmd, &s[1]) != 0) {
		LM_ERR("no redis command\n");
		return -1;
	}
	if(fixup_get_svalue(msg, (gparam_t *)sargv1, &arg1) != 0) {
		LM_ERR("no argument 1\n");
		return -1;
	}
	if(fixup_get_svalue(msg, (gparam_t *)sargv2, &arg2) != 0) {
		LM_ERR("no argument 2\n");
		return -1;
	}
	if(fixup_get_svalue(msg, (gparam_t *)sargv3, &arg3) != 0) {
		LM_ERR("no argument 3\n");
		return -1;
	}
	if(fixup_get_svalue(msg, (gparam_t *)sres, &s[2]) != 0) {
		LM_ERR("no redis reply name\n");
		return -1;
	}

	STR_VTOZ(arg1.s[arg1.len], c1);
	STR_VTOZ(arg2.s[arg2.len], c2);
	STR_VTOZ(arg3.s[arg3.len], c3);
	if(redisc_exec(&s[0], &s[2], &s[1], arg1.s, arg2.s, arg3.s) < 0) {
		STR_ZTOV(arg1.s[arg1.len], c1);
		STR_ZTOV(arg2.s[arg2.len], c2);
		STR_ZTOV(arg3.s[arg3.len], c3);
		return -1;
	}
	STR_ZTOV(arg1.s[arg1.len], c1);
	STR_ZTOV(arg2.s[arg2.len], c2);
	STR_ZTOV(arg3.s[arg3.len], c3);
	return 1;
}

/**
 *
 */
static int w_redis_pipe_cmd3(
		struct sip_msg *msg, char *ssrv, char *scmd, char *sres)
{
	str s[3];

	if(redis_cluster_param) {
		LM_ERR("Pipelining is not supported if cluster parameter is enabled\n");
		return -1;
	}
	if(fixup_get_svalue(msg, (gparam_t *)ssrv, &s[0]) != 0) {
		LM_ERR("no redis server name\n");
		return -1;
	}
	if(fixup_get_svalue(msg, (gparam_t *)scmd, &s[1]) != 0) {
		LM_ERR("no redis command\n");
		return -1;
	}
	if(fixup_get_svalue(msg, (gparam_t *)sres, &s[2]) != 0) {
		LM_ERR("no redis reply name\n");
		return -1;
	}

	if(redisc_append_cmd(&s[0], &s[2], &s[1]) < 0)
		return -1;
	return 1;
}

/**
 *
 */
static int w_redis_pipe_cmd4(
		struct sip_msg *msg, char *ssrv, char *scmd, char *sargv1, char *sres)
{
	str s[3];
	str arg1;
	char c1;

	if(redis_cluster_param) {
		LM_ERR("Pipelining is not supported if cluster parameter is enabled\n");
		return -1;
	}
	if(fixup_get_svalue(msg, (gparam_t *)ssrv, &s[0]) != 0) {
		LM_ERR("no redis server name\n");
		return -1;
	}
	if(fixup_get_svalue(msg, (gparam_t *)scmd, &s[1]) != 0) {
		LM_ERR("no redis command\n");
		return -1;
	}
	if(fixup_get_svalue(msg, (gparam_t *)sargv1, &arg1) != 0) {
		LM_ERR("no argument 1\n");
		return -1;
	}
	if(fixup_get_svalue(msg, (gparam_t *)sres, &s[2]) != 0) {
		LM_ERR("no redis reply name\n");
		return -1;
	}

	STR_VTOZ(arg1.s[arg1.len], c1);
	if(redisc_append_cmd(&s[0], &s[2], &s[1], arg1.s) < 0) {
		STR_ZTOV(arg1.s[arg1.len], c1);
		return -1;
	}
	STR_ZTOV(arg1.s[arg1.len], c1);
	return 1;
}

/**
 *
 */
static int w_redis_pipe_cmd5(struct sip_msg *msg, char *ssrv, char *scmd,
		char *sargv1, char *sargv2, char *sres)
{
	str s[3];
	str arg1, arg2;
	char c1, c2;

	if(redis_cluster_param) {
		LM_ERR("Pipelining is not supported if cluster parameter is enabled\n");
		return -1;
	}
	if(fixup_get_svalue(msg, (gparam_t *)ssrv, &s[0]) != 0) {
		LM_ERR("no redis server name\n");
		return -1;
	}
	if(fixup_get_svalue(msg, (gparam_t *)scmd, &s[1]) != 0) {
		LM_ERR("no redis command\n");
		return -1;
	}
	if(fixup_get_svalue(msg, (gparam_t *)sargv1, &arg1) != 0) {
		LM_ERR("no argument 1\n");
		return -1;
	}
	if(fixup_get_svalue(msg, (gparam_t *)sargv2, &arg2) != 0) {
		LM_ERR("no argument 2\n");
		return -1;
	}
	if(fixup_get_svalue(msg, (gparam_t *)sres, &s[2]) != 0) {
		LM_ERR("no redis reply name\n");
		return -1;
	}

	STR_VTOZ(arg1.s[arg1.len], c1);
	STR_VTOZ(arg2.s[arg2.len], c2);
	if(redisc_append_cmd(&s[0], &s[2], &s[1], arg1.s, arg2.s) < 0) {
		STR_ZTOV(arg1.s[arg1.len], c1);
		STR_ZTOV(arg2.s[arg2.len], c2);
		return -1;
	}
	STR_ZTOV(arg1.s[arg1.len], c1);
	STR_ZTOV(arg2.s[arg2.len], c2);
	return 1;
}

/**
 *
 */
static int w_redis_pipe_cmd6(struct sip_msg *msg, char *ssrv, char *scmd,
		char *sargv1, char *sargv2, char *sargv3, char *sres)
{
	str s[3];
	str arg1, arg2, arg3;
	char c1, c2, c3;

	if(redis_cluster_param) {
		LM_ERR("Pipelining is not supported if cluster parameter is enabled\n");
		return -1;
	}
	if(fixup_get_svalue(msg, (gparam_t *)ssrv, &s[0]) != 0) {
		LM_ERR("no redis server name\n");
		return -1;
	}
	if(fixup_get_svalue(msg, (gparam_t *)scmd, &s[1]) != 0) {
		LM_ERR("no redis command\n");
		return -1;
	}
	if(fixup_get_svalue(msg, (gparam_t *)sargv1, &arg1) != 0) {
		LM_ERR("no argument 1\n");
		return -1;
	}
	if(fixup_get_svalue(msg, (gparam_t *)sargv2, &arg2) != 0) {
		LM_ERR("no argument 2\n");
		return -1;
	}
	if(fixup_get_svalue(msg, (gparam_t *)sargv3, &arg3) != 0) {
		LM_ERR("no argument 3\n");
		return -1;
	}
	if(fixup_get_svalue(msg, (gparam_t *)sres, &s[2]) != 0) {
		LM_ERR("no redis reply name\n");
		return -1;
	}

	STR_VTOZ(arg1.s[arg1.len], c1);
	STR_VTOZ(arg2.s[arg2.len], c2);
	STR_VTOZ(arg3.s[arg3.len], c3);
	if(redisc_append_cmd(&s[0], &s[2], &s[1], arg1.s, arg2.s, arg3.s) < 0) {
		STR_ZTOV(arg1.s[arg1.len], c1);
		STR_ZTOV(arg2.s[arg2.len], c2);
		STR_ZTOV(arg3.s[arg3.len], c3);
		return -1;
	}
	STR_ZTOV(arg1.s[arg1.len], c1);
	STR_ZTOV(arg2.s[arg2.len], c2);
	STR_ZTOV(arg3.s[arg3.len], c3);
	return 1;
}

/**
 *
 */
static int w_redis_execute(struct sip_msg *msg, char *ssrv)
{
	str s;
	int rv;

	if(redis_cluster_param) {
		LM_ERR("Pipelining is not supported if cluster parameter is enabled\n");
		return -1;
	}
	if(fixup_get_svalue(msg, (gparam_t *)ssrv, &s) != 0) {
		LM_ERR("no redis server name\n");
		return -1;
	}
	rv = redisc_exec_pipelined_cmd(&s);
	if(rv)
		return rv;
	return 1;
}

/**
 *
 */
static int fixup_redis_cmd6(void **param, int param_no)
{
	return fixup_spve_null(param, 1);
}


/**
 *
 */
static int w_redis_free_reply(struct sip_msg *msg, char *res)
{
	str name;

	if(fixup_get_svalue(msg, (gparam_t *)res, &name) != 0) {
		LM_ERR("no redis reply name\n");
		return -1;
	}

	if(redisc_free_reply(&name) < 0)
		return -1;

	return 1;
}

/**
 *
 */
int redis_srv_param(modparam_t type, void *val)
{
	return redisc_add_server((char *)val);
}

/**
 *
 */
int redis_parse_index(str *in, gparam_t *gp)
{
	if(in->s[0] == PV_MARKER) {
		gp->type = GPARAM_TYPE_PVS;
		gp->v.pvs = (pv_spec_t *)pkg_malloc(sizeof(pv_spec_t));
		if(gp->v.pvs == NULL) {
			LM_ERR("no pkg memory left for pv_spec_t\n");
			return -1;
		}

		if(pv_parse_spec(in, gp->v.pvs) == NULL) {
			LM_ERR("invalid PV identifier\n");
			pkg_free(gp->v.pvs);
			return -1;
		}
	} else {
		gp->type = GPARAM_TYPE_INT;
		if(str2sint(in, &gp->v.i) != 0) {
			LM_ERR("bad number <%.*s>\n", in->len, in->s);
			return -1;
		}
	}
	return 0;
}


/**
 *
 */
int redis_parse_token(str *in, gparam_t *gp, int *i)
{
	str tok;

	while(*i < in->len && isspace(in->s[*i]))
		(*i)++;

	if(*i >= in->len - 2)
		return -1;

	if(in->s[(*i)++] != '[')
		return -1;

	while(*i < in->len - 1 && isspace(in->s[*i]))
		(*i)++;
	if(*i == in->len - 1 || in->s[*i] == ']')
		return -1;
	tok.s = &(in->s[*i]);

	while(*i < in->len && !isspace(in->s[*i]) && in->s[*i] != ']')
		(*i)++;
	if(*i == in->len)
		return -1;
	tok.len = &(in->s[*i]) - tok.s;
	if(redis_parse_index(&tok, gp) != 0)
		return -1;

	while(*i < in->len && isspace(in->s[*i]))
		(*i)++;
	if(*i == in->len || in->s[*i] != ']')
		return -1;

	return 0;
}

/**
 *
 */
static int pv_parse_redisc_name(pv_spec_p sp, str *in)
{
	redisc_pv_t *rpv = NULL;
	str pvs;
	int i;
	int key_idx;
	int last_key;

	if(in->s == NULL || in->len <= 0)
		return -1;

	rpv = (redisc_pv_t *)pkg_malloc(sizeof(redisc_pv_t));
	if(rpv == NULL)
		return -1;

	memset(rpv, 0, sizeof(redisc_pv_t));

	pvs = *in;
	trim(&pvs);

	rpv->rname.s = pvs.s;
	for(i = 0; i < pvs.len - 2; i++) {
		if(isspace(pvs.s[i]) || pvs.s[i] == '=') {
			rpv->rname.len = i;
			break;
		}
	}
	rpv->rname.len = i;

	if(rpv->rname.len == 0)
		goto error_var;

	while(i < pvs.len - 2 && isspace(pvs.s[i]))
		i++;

	key_idx = 0;
	do {

		if(pvs.s[i] != '=')
			goto error_var;

		if(pvs.s[i + 1] != '>')
			goto error_var;

		i += 2;
		while(i < pvs.len && isspace(pvs.s[i]))
			i++;

		if(i >= pvs.len)
			goto error_key;

		rpv->rkey.s = pvs.s + i;
		rpv->rkey.len = pvs.len - i;

		/* Default pos param initialization. */
		rpv->pos[key_idx].type = GPARAM_TYPE_INT;
		rpv->pos[key_idx].v.i = -1;

		last_key = 1;
		if(rpv->rkey.len >= 5 && strncmp(rpv->rkey.s, "value", 5) == 0) {
			rpv->rkeyid = 1;
			if(rpv->rkey.len > 5) {
				i += 5;
				if(redis_parse_token(&pvs, &(rpv->pos[key_idx]), &i) != 0)
					goto error_key;
				i++;
				if(i != pvs.len)
					last_key = 0;
			}
		} else if(rpv->rkey.len >= 4 && strncmp(rpv->rkey.s, "type", 4) == 0) {
			rpv->rkeyid = 0;
			if(rpv->rkey.len > 4) {
				i += 4;
				if(redis_parse_token(&pvs, &(rpv->pos[key_idx]), &i) != 0)
					goto error_key;
			}
		} else if(rpv->rkey.len == 4 && strncmp(rpv->rkey.s, "info", 4) == 0) {
			rpv->rkeyid = 2;
		} else if(rpv->rkey.len == 4 && strncmp(rpv->rkey.s, "size", 4) == 0) {
			rpv->rkeyid = 3;
		} else {
			goto error_key;
		}

		key_idx++;
		if(key_idx > MAXIMUM_NESTED_KEYS) {
			LM_ERR("Too many nested fields");
			goto error_key;
		}

	} while(!last_key);

	rpv->rkeynum = key_idx;
	sp->pvp.pvn.u.dname = (void *)rpv;
	sp->pvp.pvn.type = PV_NAME_OTHER;
	return 0;

error_var:
	LM_ERR("invalid var spec [%.*s]\n", in->len, in->s);
	pkg_free(rpv);
	return -1;

error_key:
	LM_ERR("invalid key spec in [%.*s]\n", in->len, in->s);
	pkg_free(rpv);
	return -1;
}

/**
 *
 */
static int pv_get_redisc(
		struct sip_msg *msg, pv_param_t *param, pv_value_t *res)
{
	redisc_pv_t *rpv;
	str s;
	int pos;
	int key_idx = 0;
	redisReply *reply;

	rpv = (redisc_pv_t *)param->pvn.u.dname;
	if(rpv->reply == NULL) {
		rpv->reply = redisc_get_reply(&rpv->rname);
		if(rpv->reply == NULL)
			return pv_get_null(msg, param, res);
	}

	if(rpv->reply->rplRedis == NULL)
		return pv_get_null(msg, param, res);

	reply = rpv->reply->rplRedis;
	/* if we have arrays within arrays go to the element before last */
	if(rpv->rkeynum > 1)
		for(key_idx = 0; key_idx < rpv->rkeynum - 1; key_idx++) {
			if(fixup_get_ivalue(msg, &rpv->pos[key_idx], &pos) != 0)
				return pv_get_null(msg, param, res);
			if(pos < 0 || pos >= (int)reply->elements)
				return pv_get_null(msg, param, res);
			reply = reply->element[pos];
			if(reply == NULL) {
				LM_ERR("The reply is corrupted");
				return pv_get_null(msg, param, res);
			}
		}

	if(fixup_get_ivalue(msg, &rpv->pos[key_idx], &pos) != 0)
		return pv_get_null(msg, param, res);

	switch(rpv->rkeyid) {
		case 1:
			/* value */
			switch(reply->type) {
				case REDIS_REPLY_STRING:
					if(pos != -1)
						return pv_get_null(msg, param, res);
					s.len = reply->len;
					s.s = reply->str;
					return pv_get_strval(msg, param, res, &s);
				case REDIS_REPLY_INTEGER:
					if(pos != -1)
						return pv_get_null(msg, param, res);
					return pv_get_sintval(msg, param, res, (int)reply->integer);
				case REDIS_REPLY_ARRAY:
					if(pos < 0 || pos >= (int)reply->elements)
						return pv_get_null(msg, param, res);
					if(reply->element[pos] == NULL)
						return pv_get_null(msg, param, res);
					switch(reply->element[pos]->type) {
						case REDIS_REPLY_STRING:
							s.len = reply->element[pos]->len;
							s.s = reply->element[pos]->str;
							return pv_get_strval(msg, param, res, &s);
						case REDIS_REPLY_INTEGER:
							return pv_get_sintval(msg, param, res,
									(int)reply->element[pos]->integer);
						default:
							return pv_get_null(msg, param, res);
					}
				default:
					return pv_get_null(msg, param, res);
			}
		case 2:
			/* info */
			if(reply->str == NULL)
				return pv_get_null(msg, param, res);
			s.len = reply->len;
			s.s = reply->str;
			return pv_get_strval(msg, param, res, &s);
		case 3:
			/* size */
			if(reply->type == REDIS_REPLY_ARRAY) {
				return pv_get_uintval(
						msg, param, res, (unsigned int)reply->elements);
			} else {
				return pv_get_null(msg, param, res);
			}
		case 0:
			/* type */
			if(pos == -1)
				return pv_get_sintval(msg, param, res, reply->type);
			if(reply->type != REDIS_REPLY_ARRAY)
				return pv_get_null(msg, param, res);
			if(pos < 0 || pos >= (int)reply->elements)
				return pv_get_null(msg, param, res);
			if(reply->element[pos] == NULL)
				return pv_get_null(msg, param, res);
			return pv_get_sintval(msg, param, res, reply->element[pos]->type);
		default:
			/* We do nothing. */
			return pv_get_null(msg, param, res);
	}
}

/**
 *
 */
int pv_parse_rediscd_name(pv_spec_p sp, str *in)
{
	if(sp == NULL || in == NULL || in->len <= 0)
		return -1;

	switch(in->len) {
		case 7:
			if(strncmp(in->s, "rpl_str", 7) == 0)
				sp->pvp.pvn.u.isname.name.n = 0;
			else if(strncmp(in->s, "rpl_arr", 7) == 0)
				sp->pvp.pvn.u.isname.name.n = 1;
			else if(strncmp(in->s, "rpl_int", 7) == 0)
				sp->pvp.pvn.u.isname.name.n = 2;
			else if(strncmp(in->s, "rpl_nil", 7) == 0)
				sp->pvp.pvn.u.isname.name.n = 3;
			else if(strncmp(in->s, "rpl_sts", 7) == 0)
				sp->pvp.pvn.u.isname.name.n = 4;
			else if(strncmp(in->s, "rpl_err", 7) == 0)
				sp->pvp.pvn.u.isname.name.n = 5;
			else
				goto error;
			break;
		default:
			goto error;
	}
	sp->pvp.pvn.type = PV_NAME_INTSTR;
	sp->pvp.pvn.u.isname.type = 0;

	return 0;

error:
	LM_ERR("unknown redisd name %.*s\n", in->len, in->s);
	return -1;
}

/**
 *
 */
int pv_get_rediscd(struct sip_msg *msg, pv_param_t *param, pv_value_t *res)
{
	if(param == NULL)
		return -1;
	switch(param->pvn.u.isname.name.n) {
		case 0:
			return pv_get_uintval(msg, param, res, REDIS_REPLY_STRING);
		case 1:
			return pv_get_uintval(msg, param, res, REDIS_REPLY_ARRAY);
		case 2:
			return pv_get_uintval(msg, param, res, REDIS_REPLY_INTEGER);
		case 3:
			return pv_get_uintval(msg, param, res, REDIS_REPLY_NIL);
		case 4:
			return pv_get_uintval(msg, param, res, REDIS_REPLY_STATUS);
		case 5:
			return pv_get_uintval(msg, param, res, REDIS_REPLY_ERROR);
		default:
			return pv_get_uintval(msg, param, res, 0);
	}
}

/**
 * @brief bind functions to NDB_REDIS API structure
 */
int bind_ndb_redis(ndb_redis_api_t *api)
{
	if(!api) {
		ERR("Invalid parameter value\n");
		return -1;
	}
	memset(api, 0, sizeof(ndb_redis_api_t));
	api->get_server = redisc_get_server;
	api->exec = redisc_exec;
	api->exec_argv = redisc_exec_argv;
	api->get_reply = redisc_get_reply;
	api->free_reply = redisc_free_reply;

	return 0;
}

/**
 *
 */
static int ki_redis_cmd(sip_msg_t *msg, str *srv, str *rcmd, str *sres)
{
	int i;
	if(rcmd == NULL || rcmd->s == NULL) {
		LM_ERR("invalid command\n");
		return -1;
	}
	for(i = 0; i < rcmd->len - 1; i++) {
		if(rcmd->s[i] == '%') {
			if(rcmd->s[i + 1] == 's' || rcmd->s[i + 1] == 'b') {
				LM_ERR("command argument specifier found, but no params\n");
				return -1;
			}
		}
	}
	return redisc_exec(srv, sres, rcmd);
}

/**
 *
 */
static int ki_redis_cmd_p1(
		sip_msg_t *msg, str *srv, str *rcmd, str *p1, str *sres)
{
	return redisc_exec(srv, sres, rcmd, p1->s);
}

/**
 *
 */
static int ki_redis_cmd_p2(
		sip_msg_t *msg, str *srv, str *rcmd, str *p1, str *p2, str *sres)
{
	return redisc_exec(srv, sres, rcmd, p1->s, p2->s);
}

/**
 *
 */
static int ki_redis_cmd_p3(sip_msg_t *msg, str *srv, str *rcmd, str *p1,
		str *p2, str *p3, str *sres)
{
	return redisc_exec(srv, sres, rcmd, p1->s, p2->s, p3->s);
}

/**
 *
 */
static int ki_redis_free_reply(sip_msg_t *msg, str *name)
{
	if(redisc_free_reply(name) < 0)
		return -1;

	return 1;
}
/**
 *
 */
/* clang-format off */
static sr_kemi_t sr_kemi_ndb_redis_exports[] = {
	{ str_init("ndb_redis"), str_init("redis_cmd"),
		SR_KEMIP_INT, ki_redis_cmd,
		{ SR_KEMIP_STR, SR_KEMIP_STR, SR_KEMIP_STR,
			SR_KEMIP_NONE, SR_KEMIP_NONE, SR_KEMIP_NONE }
	},
	{ str_init("ndb_redis"), str_init("redis_cmd_p1"),
		SR_KEMIP_INT, ki_redis_cmd_p1,
		{ SR_KEMIP_STR, SR_KEMIP_STR, SR_KEMIP_STR,
			SR_KEMIP_STR, SR_KEMIP_NONE, SR_KEMIP_NONE }
	},
	{ str_init("ndb_redis"), str_init("redis_cmd_p2"),
		SR_KEMIP_INT, ki_redis_cmd_p2,
		{ SR_KEMIP_STR, SR_KEMIP_STR, SR_KEMIP_STR,
			SR_KEMIP_STR, SR_KEMIP_STR, SR_KEMIP_NONE }
	},
	{ str_init("ndb_redis"), str_init("redis_cmd_p3"),
		SR_KEMIP_INT, ki_redis_cmd_p3,
		{ SR_KEMIP_STR, SR_KEMIP_STR, SR_KEMIP_STR,
			SR_KEMIP_STR, SR_KEMIP_STR, SR_KEMIP_STR }
	},
	{ str_init("ndb_redis"), str_init("redis_free"),
		SR_KEMIP_INT, ki_redis_free_reply,
		{ SR_KEMIP_STR, SR_KEMIP_NONE, SR_KEMIP_NONE,
			SR_KEMIP_NONE, SR_KEMIP_NONE, SR_KEMIP_NONE }
	},

	{ {0, 0}, {0, 0}, 0, NULL, { 0, 0, 0, 0, 0, 0 } }
};
/* clang-format on */

int mod_register(char *path, int *dlflags, void *p1, void *p2)
{
	sr_kemi_modules_add(sr_kemi_ndb_redis_exports);
	return 0;
}
