/*
 * Copyright (C) 2001-2003 FhG Fokus
 *
 * 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
 */

/**
 * @file
 * @brief Kamailio core :: modules loading, structures declarations and utilities
 * @ingroup core
 * Module: \ref core
 */


#ifndef sr_module_h
#define sr_module_h

#include <dlfcn.h>

#include "parser/msg_parser.h" /* for sip_msg */
#include "ver_defs.h"
#include "rpc.h"
#include "route_struct.h"
#include "route.h"
#include "str.h"

/* kamailio compat */
#include "counters.h"
#include "pvar.h"

#define MODULE_INTERFACE_VER 4

/**
 * @brief type used for the mod_register function export
 *
 * mod_register is a function called when loading a module
 * (if present), prior to registering the module exports.
 * @param path path to the module, including file name
 * @param dlflags pointer to the dlflags used when loading the module.
 * If the value is changed to a different and non-zero value, the
 * module will be reloaded with the new flags.
 * @param reserved1 - reserved for future use.
 * @param reserved2 - reserved for future use
 * @return 0 on success, -1 on error, all the other values are reserved
 *                      for future use (<0 meaning error and >0 success)
 */
typedef int (*mod_register_function)(
		char *path, int *dlflags, void *reserved1, void *reserved2);

typedef struct module_exports *(*module_register)(void);

/**
 * @brief main two parameter module function
 *
 * Main two parameter module function, default and oldest version.
 * @param sip_msg SIP message
 * @param param1 first parameter
 * @param param2 second parameter
 * @return positive on success, negative on error, 0 to stop processing (drop message)
 */
typedef int (*cmd_function)(struct sip_msg *, char *param1, char *param2);
typedef int (*cmd_function3)(struct sip_msg *, char *, char *, char *);
typedef int (*cmd_function4)(struct sip_msg *, char *, char *, char *, char *);
typedef int (*cmd_function5)(
		struct sip_msg *, char *, char *, char *, char *, char *);
typedef int (*cmd_function6)(
		struct sip_msg *, char *, char *, char *, char *, char *, char *);
/**
 * @brief variable number of parameter module function
 *
 * Variable number of parameter module function, takes as param the sip_msg,
 * extra parameters number and a pointer to an array of parameters
 * @param sip_msg SIP message
 * @param no extra parameters number
 * @param vals extra parameters
 * @return positive on success, negative on error, 0 to stop processing (drop message)
 */
typedef int (*cmd_function_var)(struct sip_msg *, int no, action_u_t *vals);
typedef int (*fixup_function)(void **param, int param_no);
typedef int (*free_fixup_function)(void **param, int param_no);

/**
 * @brief response module function prototype
 * @param sip_msg SIP message
 * @return positive if ok, 0 to stop processing
 */
typedef int (*response_function)(struct sip_msg *);
typedef void (*onbreak_function)(struct sip_msg *);
typedef void (*destroy_function)(void);

typedef int (*init_function)(void);
typedef int (*child_init_function)(int rank);


#define PARAM_STRING (1U << 0) /**< String (char *) parameter type */
#define PARAM_INT (1U << 1)	   /**< Integer parameter type */
#define PARAM_STR (1U << 2)	   /**< struct str parameter type */
#define PARAM_VAR (1U << 3)	   /**< var parameter type - mdoparamx */
#define PARAM_USE_FUNC (1U << (8 * sizeof(int) - 1))
#define PARAM_TYPE_MASK(_x) ((_x) & (~PARAM_USE_FUNC))

/* temporary, for backward compatibility only until all modules adjust it */
#define STR_PARAM PARAM_STRING
#define INT_PARAM PARAM_INT
#define USE_FUNC_PARAM PARAM_USE_FUNC

typedef unsigned int modparam_t;

typedef int (*param_func_t)(modparam_t type, void *val);

/* magic parameter number values */

#define NO_SCRIPT -1 /**< export not usable from scripts */
#define VAR_PARAM_NO \
	-128 /**< function has variable number of parameters
							   (see cmd_function_var for the prototype) */

/**
 * special fixup function flags fparam fixup, rve ready,
 * they are kept in the first 2 bits inside the pointer
 */
#define FIXUP_F_FPARAM_RVE (unsigned long)1

#define call_fixup(fixup, param, param_no) \
	((fixup) ? (fixup)(param, param_no) : 0)

/* flags to execute child_init() for special ranks */
#define KSRMOD_FLAG_POSTCHILDINIT (1 << 1)
void ksr_module_set_flag(unsigned int flag);

/* Macros - used as rank in child_init function */
#define PROC_MAIN 0		   /**< Main process */
#define PROC_TIMER -1	   /**< Timer attendant process */
#define PROC_RPC -2		   /**< RPC type process */
#define PROC_FIFO PROC_RPC /**< FIFO attendant process */
#define PROC_TCP_MAIN -4   /**< TCP main process */
#define PROC_UNIXSOCK -5   /**< Unix socket server */
#define PROC_ATTENDANT -10 /**< main "attendant process */
#define PROC_POSTCHILDINIT \
	-126 /**< special rank - main kamailio process after
									after all child_init() are executed */
#define PROC_INIT \
	-127 /**< special rank, the context is the main kamailio
							  process, but this is guaranteed to be executed
							  before any process is forked, so it can be used
							  to setup shared variables that depend on some
							  after mod_init available information (e.g.
							  total number of processes).
							  @warning child_init(PROC_MAIN) is again called
							 in the same process (main), but latter
							 (before tcp), so make sure you don't init things
							 twice, both in PROC_MAIN and PROC_INT */
#define PROC_NOCHLDINIT \
	-128 /**< no child init functions will be called
                                if this rank is used in fork_process() */

#define PROC_SIPINIT \
	1 /**< First (special) SIP worker - some modules do
						special processing in this child, like loading db data */
#define PROC_SIPROUTER \
	2 /**< First (pure) SIP worker - can be used to check
						if just a normal sip router or initialize custom worker
						processes by starting from this value */
#define PROC_RANK127 \
	127 /**< Used to init RPC worker as SIP commands
							   handler. Don't do any special processing in the
							   child init with this rank - just bare child
							   initialization */

#define PROC_SIPRPC PROC_RANK127

#define PROC_XWORKER PROC_RANK127

#define PROC_MIN PROC_NOCHLDINIT /**< Minimum process rank */


#define DEFAULT_DLFLAGS \
	0 /**< value that signals to module loader to
							use default dlopen flags in Kamailio */
#ifndef RTLD_NOW
/* for openbsd */
#define RTLD_NOW DL_LAZY
#endif

#define KAMAILIO_DLFLAGS RTLD_NOW


#define MODULE_VERSION                       \
	char *module_version = SER_FULL_VERSION; \
	char *module_flags = SER_COMPILE_FLAGS;  \
	unsigned int module_interface_ver = MODULE_INTERFACE_VER;


/** kamailio/openser version */
typedef struct cmd_export
{
	char *name;			   /**< null terminated command name */
	cmd_function function; /**< pointer to the corresponding function */
	int param_no;		   /**< number of parameters used by the function */
	fixup_function fixup;  /**< pointer to the function called to "fix" the
							   parameters */
	free_fixup_function free_fixup; /**< function called to free the "fixed"
									   parameters */
	unsigned int flags;				/**< Function flags */
} cmd_export_t;

/** sip-router version */
typedef struct ksr_cmd_export
{
	char *name;			   /**< null terminated command name */
	cmd_function function; /**< pointer to the corresponding function */
	int param_no;		   /**< number of parameters used by the function */
	fixup_function fixup;  /**< pointer to the function called to "fix" the
							   parameters */
	free_fixup_function free_fixup; /**< function called to free the "fixed"
									   parameters */
	unsigned int flags;				/**< Function flags */
	unsigned int fixup_flags;
	void *module_exports; /**< pointer to module structure */
} ksr_cmd_export_t;


typedef struct param_export
{
	char *name;			 /**< null terminated param. name */
	modparam_t type;	 /**< param. type */
	void *param_pointer; /**< pointer to the param. memory location */
} param_export_t;


/*
 * Allowed parameter types, the types _must_ be in "fallback" order,
 * e.g. FPARAM_STR should be the last to allow fallback to it,
 * F_PARAM_PVS should be in front of F_PARAM_AVP (so that
 * for fix_param_types(FPARAM_AVP|FPARAM_PVS|FPARAM_STR, param) and $foo
 * the pvars will be checked first and only if no pvar is found the
 * param will be resolved to an avp)
 */
enum
{
	FPARAM_UNSPEC = 0,
	FPARAM_INT = (1 << 0),
	FPARAM_SELECT = (1 << 1),
	FPARAM_PVS = (1 << 2),
	FPARAM_AVP = (1 << 3),
	FPARAM_STRING = (1 << 4),
	FPARAM_STR = (1 << 5),
	/* special types: no fallback between them possible */
	FPARAM_REGEX = (1 << 6),
	FPARAM_SUBST = (1 << 7),
	FPARAM_PVE = (1 << 8)
};

/**
 * Function parameter
 */
typedef struct fparam
{
	char *orig; /**< The original value */
	int type;	/**< Type of parameter */
	union
	{
		char *asciiz;			  /**< Zero terminated ASCII string */
		struct _str str;		  /**< pointer/len string */
		int i;					  /**< Integer value */
		regex_t *regex;			  /**< Compiled regular expression */
		avp_ident_t avp;		  /**< AVP identifier */
		select_t *select;		  /**< select structure */
		struct subst_expr *subst; /**< Regex substitution */
		pv_spec_t *pvs;			  /**< kamailio pseudo-vars */
		pv_elem_t *pve;			  /**< kamailio pseudo-vars in a string */
	} v;
	void *fixed;
} fparam_t;

/** kamailio module exports version */
typedef struct module_exports
{
	/**< null terminated module name */
	char *name;
	/**< flags for dlopen  */
	unsigned int dlflags;
	/**< null terminated array of the exported commands (config functions)*/
	cmd_export_t *cmds;
	/**< null terminated array of the exported module parameters */
	param_export_t *params;
	/**< null terminated array of exported rpc methods */
	rpc_export_t *rpc_methods;
	/*!< null terminated array of the exported module items (pseudo-variables) */
	pv_export_t *pv_items;
	/**< function used for responses, returns yes or no; can be null */
	response_function response_f;
	/**< Initialization function */
	init_function init_mod_f;
	/**< function called by all processes after the fork */
	child_init_function init_child_f;
	/**< function called when the module is "destroyed" (on server shut down) */
	destroy_function destroy_mod_f;
} module_exports_t;


/** kamailio module exports version coverted for core operations */
typedef struct ksr_module_exports
{
	/**< null terminated module name */
	char *name;
	/**< flags for dlopen  */
	unsigned int dlflags;
	/**< null terminated array of the exported commands (config functions)*/
	ksr_cmd_export_t *cmds;
	/**< null terminated array of the exported module parameters */
	param_export_t *params;
	/**< null terminated array of exported rpc methods */
	rpc_export_t *rpc_methods;
	/*!< null terminated array of the exported module items (pseudo-variables) */
	pv_export_t *pv_items;
	/**< function used for responses, returns yes or no; can be null */
	response_function response_f;
	/**< Initialization function */
	init_function init_mod_f;
	/**< function called by all processes after the fork */
	child_init_function init_child_f;
	/**< function called when the module is "destroyed" (on server shut down) */
	destroy_function destroy_mod_f;
} ksr_module_exports_t;

typedef struct sr_module
{
	char *path;
	void *handle;
	ksr_module_exports_t exports;
	unsigned int modflags;
	struct sr_module *next;
} sr_module_t;


extern sr_module_t *modules;				 /**< global module list*/
extern response_function *mod_response_cbks; /**< response callback array */
extern int mod_response_cbk_no; /**< size of reponse callbacks array */

int register_builtin_modules(void);
int ksr_load_module(char *path, char *opts);
int ksr_load_modulex(char *path, char *opts);
ksr_cmd_export_t *find_export_record(char *name, int param_no, int flags);
cmd_function find_export(char *name, int param_no, int flags);
cmd_function find_mod_export(char *mod, char *name, int param_no, int flags);
rpc_export_t *find_rpc_export(char *name, int flags);
void destroy_modules(void);
int init_child(int rank);
int init_modules(void);
struct sr_module *find_module_by_name(char *mod);
sr_module_t *get_loaded_modules(void);

/**< true if the module with name 'mod_name' is loaded */
#define module_loaded(mod_name) (find_module_by_name(mod_name) != 0)


/**
 * @brief Find a parameter with given type
 * @param mod module
 * @param name parameter name
 * @param type_mask parameter mask
 * @param param_type parameter type
 * @return parameter address in memory, if there is no such parameter, NULL is returned
 */
void *find_param_export(struct sr_module *mod, char *name, modparam_t type_mask,
		modparam_t *param_type);


/** API function to get other parameters from fixup */
action_u_t *fixup_get_param(
		void **cur_param, int cur_param_no, int required_param_no);
int fixup_get_param_count(void **cur_param, int cur_param_no);

int fix_flag(modparam_t type, void *val, char *mod_name, char *param_name,
		int *flag);


/*
 * Common function parameter fixups
 */

/**
 * @brief Generic parameter fixup function
 *
 * Generic parameter fixup function which creates fparam_t structure.
 * Type parameter contains allowed parameter types.
 * @param type parameter type
 * @param param fixed parameter
 * @return 0 on success, -1 on error
 */
int fix_param(int type, void **param);
void fparam_free_contents(fparam_t *fp);

/** fix a param to one of the given types (mask).
  */
int fix_param_types(int types, void **param);

/**
 * @brief Fixup variable string,
 *
 * Fixup variable string, the parameter can be AVP, SELECT, or ordinary
 * string. AVP and select identifiers will be resolved to their values
 * during runtime. The parameter value will be converted to fparam structure.
 * @param param fixed parameter value
 * @param param_no number of parameter
 * @return 0 on success, -1 on an error
 */
int fixup_var_str_12(void **param, int param_no);

/** Same as fixup_var_str_12 but applies to the 1st parameter only */
int fixup_var_str_1(void **param, int param_no);

/** Same as fixup_var_str_12 but applies to the 2nd parameter only */
int fixup_var_str_2(void **param, int param_no);

/** fixup variable-pve-only-string. */
int fixup_var_pve_12(void **param, int param_no);

/** fixup variable-pve-string.
 * The parameter can be a PVAR, AVP, SELECT, PVE (pv based format string)
 * or string.
 */
int fixup_var_pve_str_12(void **param, int param_no);

/** same as fixup_var_pve_str_12 but applies to the 1st parameter only */
int fixup_var_pve_str_1(void **param, int param_no);

/** same as fixup_var_pve_str_12 but applies to the 2nd parameter only */
int fixup_var_pve_str_2(void **param, int param_no);

/**
 * @brief Fixup variable integer
 *
 * Fixup variable integer, the parameter can be AVP, SELECT, or ordinary
 * integer. AVP and select identifiers will be resolved to their values
 * and converted to int if necessary during runtime. The parameter value will
 * be converted to fparam structure
 * @param param fixed parameter value
 * @param param_no number of parameter
 * @return 0 on success, -1 on an error
 */
int fixup_var_int_12(void **param, int param_no);

/** Same as fixup_var_int_12 but applies to the 1st parameter only */
int fixup_var_int_1(void **param, int param_no);

/** Same as fixup_var_int_12 but applies to the 2nd parameter only */
int fixup_var_int_2(void **param, int param_no);

/**
 * The parameter must be a regular expression which must compile, the
 * parameter will be converted to compiled regex
 */
int fixup_regex_12(void **param, int param_no);

/** Same as fixup_regex_12 but applies to the 1st parameter only */
int fixup_regex_1(void **param, int param_no);

/** Same as fixup_regex_12 but applies to the 2nd parameter only */
int fixup_regex_2(void **param, int param_no);

/**
 * The string parameter will be converted to integer
 */
int fixup_int_12(void **param, int param_no);

/** Same as fixup_int_12 but applies to the 1st parameter only */
int fixup_int_1(void **param, int param_no);

/** Same as fixup_int_12 but applies to the 2nd parameter only */
int fixup_int_2(void **param, int param_no);

/**
 * Parse the parameter as static string, do not resolve
 * AVPs or selects, convert the parameter to str structure
 */
int fixup_str_12(void **param, int param_no);

/** Same as fixup_str_12 but applies to the 1st parameter only */
int fixup_str_1(void **param, int param_no);

/** Same as fixup_str_12 but applies to the 2nd parameter only */
int fixup_str_2(void **param, int param_no);

/**
 * @brief Get the function parameter value as string
 * @param dst string destination
 * @param msg SIP message
 * @param param function parameters
 * @return 0 on success, 1 on error, e.g. cannot get value
 */
int get_str_fparam(str *dst, struct sip_msg *msg, fparam_t *param);

int get_strbuf_fparam(str *dst, struct sip_msg *msg, fparam_t *param);

/**
 * @brief Get the function parameter value as integer
 * @param dst string destination
 * @param msg SIP message
 * @param param function parameters
 * @return 0 on success, 1 on error, e.g. cannot get value
 */
int get_int_fparam(int *dst, struct sip_msg *msg, fparam_t *param);

/**
 * @brief Get the function parameter value as integer/string
 * @param i_dst int destination
 * @param s_dst string destination
 * @param msg SIP message
 * @param param function parameters
 * @param flags flags to indicate destinations
 * @return 0 on success, 1 on error, e.g. cannot get value
 */
int get_is_fparam(int *i_dst, str *s_dst, struct sip_msg *msg, fparam_t *param,
		unsigned int *flags);

/**
 * @brief Get the function parameter value as compiled regular expression
 * @param dst string destination
 * @param msg SIP message
 * @param param function parameters
 * @return 0 on success, 1 on error, e.g. cannot get value
 */
int get_regex_fparam(regex_t *dst, struct sip_msg *msg, fparam_t *param);


int is_fparam_rve_fixup(fixup_function f);

/**
 * @brief Generic free fixup type function for a fixed fparam
 *
 * Generic free fixup type function for a fixed fparam. It will free whatever
 * was allocated during the initial fparam fixup and restore the original param
 * value.
 * @param param freed parameters
 */
void fparam_free_restore(void **param);
int fixup_free_fparam_all(void **param, int param_no);
int fixup_free_fparam_1(void **param, int param_no);
int fixup_free_fparam_2(void **param, int param_no);

/**
 * @brief returns the corresponding fixup_free* for various known fixup types
 *
 * Returns the corresponding fixup_free* for various known fixup types.
 * Used to automatically fill in free_fixup* functions.
 * @param f fixup function pointer
 * @return free fixup function pointer on success, 0 on failure (unknown
 * fixup or no free fixup function).
 */
free_fixup_function get_fixup_free(fixup_function f);

void set_child_sip_rpc_mode(void);
void set_child_rpc_sip_mode(void);
int is_sip_worker(int rank);
int is_rpc_worker(int rank);

unsigned int set_modinit_delay(unsigned int v);
int destroy_modules_phase(void);
int ksr_shutdown_phase_init(void);
int ksr_shutdown_phase(void);

int ksr_locate_module(char *mod_path, char **new_path);

int ksr_version_control(void *handle, char *path);

#endif /* sr_module_h */
