mirror of
https://gitee.com/acl-dev/acl.git
synced 2024-12-15 17:30:53 +08:00
252 lines
6.3 KiB
C
252 lines
6.3 KiB
C
#include "lib_acl.h"
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <errno.h>
|
|
#include <string.h>
|
|
|
|
#include "app_log.h"
|
|
#include "app_main.h"
|
|
|
|
static int __mempool_limit = 0;
|
|
static int __mempool_use_mutex = 0;
|
|
|
|
static APP_RUN_FN __run_fn = NULL;
|
|
static void *__run_ctx = NULL;
|
|
static APP_INIT_FN __app_init_fn = NULL;
|
|
static void *__app_init_ctx = NULL;
|
|
static APP_EXIT_FN __app_exit_fn = NULL;
|
|
static void *__app_exit_ctx = NULL;
|
|
/*
|
|
static APP_THREAD_INIT_FN __app_thread_init_fn = NULL;
|
|
static APP_THREAD_EXIT_FN __app_thread_exit_fn = NULL;
|
|
*/
|
|
|
|
static char *__default_deny_info = "You are not welcome!\r\n";
|
|
static char *__deny_info;
|
|
|
|
static void __service(ACL_ASTREAM *astream, char *service acl_unused, char **argv acl_unused)
|
|
{
|
|
char myname[] = "__service";
|
|
ACL_VSTREAM *stream = acl_aio_vstream(astream);
|
|
char ip[64];
|
|
|
|
/*
|
|
* Sanity check. This service takes no command-line arguments.
|
|
*/
|
|
if (argv[0])
|
|
acl_msg_fatal("%s, %s(%d): unexpected command-line argument: %s, service=%s",
|
|
__FILE__, myname, __LINE__,
|
|
argv[0], service ? service : "null");
|
|
|
|
acl_watchdog_pat();
|
|
|
|
if (isatty(ACL_VSTREAM_SOCK(acl_aio_vstream(astream)))) {
|
|
(void) __run_fn(astream, __run_ctx);
|
|
return;
|
|
}
|
|
|
|
if (acl_getpeername(ACL_VSTREAM_SOCK(stream), ip, sizeof(ip)) < 0) {
|
|
acl_msg_warn("%s, %s(%d): can't get socket's ip",
|
|
__FILE__, myname, __LINE__);
|
|
acl_aio_iocp_close(astream);
|
|
} else if (!acl_access_permit(ip)) {
|
|
acl_msg_warn("%s, %s(%d): ip(%s) be denied",
|
|
__FILE__, myname, __LINE__, ip);
|
|
acl_aio_writen(astream, __deny_info, strlen(__deny_info));
|
|
acl_aio_iocp_close(astream);
|
|
} else if (__run_fn(astream, __run_ctx) != 0)
|
|
acl_aio_iocp_close(astream);
|
|
}
|
|
|
|
static void __pre_accept(char *name acl_unused, char **argv acl_unused)
|
|
{
|
|
}
|
|
|
|
static void __pre_jail_init(char *name acl_unused, char **argv acl_unused)
|
|
{
|
|
/* ÊÇ·ñ²ÉÓà libcore µÄÈÕÖ¾¼Ç¼ */
|
|
#ifdef HAS_LIB_CORE
|
|
# ifdef USE_LIBCORE_LOG
|
|
app_set_libcore_log();
|
|
# endif
|
|
#endif
|
|
}
|
|
|
|
static void __post_jail_init(char *name acl_unused, char **argv acl_unused)
|
|
{
|
|
|
|
if (acl_var_aio_access_allow != NULL)
|
|
acl_access_add(acl_var_aio_access_allow, ",", ":");
|
|
|
|
if (__app_init_fn != NULL)
|
|
__app_init_fn(__app_init_ctx);
|
|
|
|
if (__mempool_limit > 0) {
|
|
acl_msg_info("use mempool, size limit is %d, %s mutex",
|
|
__mempool_limit, __mempool_use_mutex ? "use" : "no");
|
|
}
|
|
|
|
/* TODO: you can add some init functions here */
|
|
}
|
|
|
|
static void __app_on_exit(char *service acl_unused, char **argv acl_unused)
|
|
{
|
|
if (__app_exit_fn)
|
|
__app_exit_fn(__app_exit_ctx);
|
|
#ifdef HAS_LIB_CORE
|
|
# ifdef USE_LIBCORE_LOG
|
|
app_libcore_log_end();
|
|
# endif
|
|
#endif
|
|
}
|
|
|
|
/*
|
|
static void __thread_init(void *arg)
|
|
{
|
|
if (__app_thread_init_fn)
|
|
__app_thread_init_fn(arg);
|
|
}
|
|
|
|
static void __thread_exit(void *arg)
|
|
{
|
|
if (__app_thread_exit_fn)
|
|
__app_thread_exit_fn(arg);
|
|
}
|
|
*/
|
|
|
|
static void app_main_init(void)
|
|
{
|
|
char *ptr, *pname;
|
|
ACL_ARGV *env_argv;
|
|
int i;
|
|
|
|
ptr = getenv("SERVICE_ENV");
|
|
if (ptr == NULL || *ptr == 0)
|
|
return;
|
|
|
|
env_argv = acl_argv_split(ptr, ",\t ");
|
|
if (env_argv == NULL)
|
|
return;
|
|
if (env_argv->argc == 0) {
|
|
acl_argv_free(env_argv);
|
|
return;
|
|
}
|
|
|
|
for (i = 0; i <env_argv->argc; i++) {
|
|
pname = acl_argv_index(env_argv, i);
|
|
ptr = strchr(pname, ':');
|
|
if (ptr == NULL)
|
|
continue;
|
|
*ptr++ = 0;
|
|
if (ptr == 0)
|
|
continue;
|
|
if (strcasecmp(pname, "mempool_limit") == 0) {
|
|
__mempool_limit = atoi(ptr);
|
|
} else if (strcasecmp(pname, "mempool_use_mutex") == 0) {
|
|
if (strcasecmp(ptr, "true") == 0)
|
|
__mempool_use_mutex = 1;
|
|
}
|
|
}
|
|
|
|
acl_argv_free(env_argv);
|
|
|
|
if (__mempool_limit > 0)
|
|
acl_mempool_open(__mempool_limit, __mempool_use_mutex);
|
|
|
|
__deny_info = __default_deny_info;
|
|
}
|
|
|
|
static ACL_CONFIG_BOOL_TABLE null_conf_bool_tab[] = {
|
|
/* TODO: you can add configure variables of int type here */
|
|
{ 0, 0, 0 },
|
|
};
|
|
|
|
static ACL_CONFIG_INT_TABLE null_conf_int_tab[] = {
|
|
/* TODO: you can add configure variables of int type here */
|
|
{ 0, 0, 0, 0, 0 },
|
|
};
|
|
|
|
static ACL_CONFIG_STR_TABLE null_conf_str_tab[] = {
|
|
/* TODO: you can add configure variables of int type here */
|
|
{ 0, 0, 0 },
|
|
};
|
|
|
|
void app_main(int argc, char *argv[], APP_RUN_FN run_fn, void *run_ctx, int name, ...)
|
|
{
|
|
const char *myname = "app_main";
|
|
va_list ap;
|
|
ACL_CONFIG_BOOL_TABLE *bool_tab = null_conf_bool_tab;
|
|
ACL_CONFIG_INT_TABLE *int_tab = null_conf_int_tab;
|
|
ACL_CONFIG_STR_TABLE *str_tab = null_conf_str_tab;
|
|
|
|
app_main_init();
|
|
/* ÌáÇ°½øÐÐÄ£°å³õʼ»¯£¬ÒÔʹÈÕÖ¾¾¡ÔçµØ´ò¿ª */
|
|
acl_master_log_open(argv[0]);
|
|
|
|
if (run_fn == NULL)
|
|
acl_msg_fatal("%s: run_fn null", myname);
|
|
|
|
__run_fn = run_fn;
|
|
__run_ctx = run_ctx;
|
|
|
|
va_start(ap, name);
|
|
|
|
for (; name != APP_CTL_END; name = va_arg(ap, int)) {
|
|
switch (name) {
|
|
case APP_CTL_INIT_FN:
|
|
__app_init_fn = va_arg(ap, APP_INIT_FN);
|
|
break;
|
|
case APP_CTL_INIT_CTX:
|
|
__app_init_ctx = va_arg(ap, void *);
|
|
break;
|
|
case APP_CTL_EXIT_FN:
|
|
__app_exit_fn = va_arg(ap, APP_EXIT_FN);
|
|
break;
|
|
case APP_CTL_EXIT_CTX:
|
|
__app_exit_ctx = va_arg(ap, void *);
|
|
break;
|
|
/*
|
|
case APP_CTL_THREAD_INIT:
|
|
__app_thread_init_fn = va_arg(ap, APP_THREAD_INIT_FN);
|
|
break;
|
|
case APP_CTL_THREAD_INIT_CTX:
|
|
thread_init_ctx = va_arg(ap, void*);
|
|
break;
|
|
case APP_CTL_THREAD_EXIT:
|
|
__app_thread_exit_fn = va_arg(ap, APP_THREAD_EXIT_FN);
|
|
break;
|
|
case APP_CTL_THREAD_EXIT_CTX:
|
|
thread_exit_ctx = va_arg(ap, void*);
|
|
break;
|
|
*/
|
|
case APP_CTL_CFG_BOOL:
|
|
bool_tab = va_arg(ap, ACL_CONFIG_BOOL_TABLE *);
|
|
break;
|
|
case APP_CTL_CFG_INT:
|
|
int_tab = va_arg(ap, ACL_CONFIG_INT_TABLE *);
|
|
break;
|
|
case APP_CTL_CFG_STR:
|
|
str_tab = va_arg(ap, ACL_CONFIG_STR_TABLE *);
|
|
break;
|
|
case APP_CTL_DENY_INFO:
|
|
__deny_info = acl_mystrdup(va_arg(ap, const char*));
|
|
break;
|
|
default:
|
|
acl_msg_info("%s: bad name(%d)", myname, name);
|
|
}
|
|
}
|
|
|
|
va_end(ap);
|
|
|
|
acl_aio_server_main(argc, argv, __service,
|
|
ACL_MASTER_SERVER_BOOL_TABLE, bool_tab,
|
|
ACL_MASTER_SERVER_INT_TABLE, int_tab,
|
|
ACL_MASTER_SERVER_STR_TABLE, str_tab,
|
|
ACL_MASTER_SERVER_PRE_INIT, __pre_jail_init,
|
|
ACL_MASTER_SERVER_PRE_ACCEPT, __pre_accept,
|
|
ACL_MASTER_SERVER_POST_INIT, __post_jail_init,
|
|
ACL_MASTER_SERVER_EXIT, __app_on_exit,
|
|
0);
|
|
}
|
|
|