void worker_main(int _worker_n)
{
worker_n = _worker_n;
attach_on_exit(on_exit_handler);
if(is_daemon == 1 && !getarg("gcore")) {
set_process_user(/*user*/ NULL, /*group*/ NULL);
}
if(!init_ntoa_cache() || !init_access_cache()) {
LOGF(ERR, "Couldn't init rbtree");
exit(1);
}
init_mime_types();
shm_serv_status = _shm_serv_status->p;
if(luaL_loadfile(_L, "core.lua")) {
LOGF(ERR, "Couldn't load file: %s", lua_tostring(_L, -1));
exit(1);
}
int mrkey = luaL_ref(_L, LUA_REGISTRYINDEX);
lua_pushstring(_L, "__main");
lua_rawgeti(_L, LUA_REGISTRYINDEX, mrkey);
lua_rawset(_L, LUA_GLOBALSINDEX);
int thread_count = 1000;
if(getarg("thread")) {
thread_count = atoi(getarg("thread"));
if(thread_count < 100) {
thread_count = 100;
} else if(thread_count > 10000) {
thread_count = 10000;
}
}
init_lua_threads(_L, thread_count);
/// 进入 loop 处理循环
loop_fd = se_create(4096);
set_loop_fd(loop_fd, _worker_n); // for coevent module
se_accept(loop_fd, server_fd, be_accept);
if(ssl_server_fd > 0) {
se_accept(loop_fd, ssl_server_fd, be_ssl_accept);
}
se_loop(loop_fd, 10, other_simple_jobs); // loop
exit(0);
}
void worker_main(int _worker_n)
{
worker_n = _worker_n;
attach_on_exit(on_exit_handler);
set_process_user(/*user*/ NULL, /*group*/ NULL);
/// 进入 loop 处理循环
loop_fd = se_create(4096);
se_accept(loop_fd, server_fd, be_accept);
se_loop(loop_fd, 10, other_simple_jobs); // loop
exit(0);
}
static void worker_main ( int at )
{
attach_on_exit ( on_exit_handler );
setProcessUser ( /*user*/ NULL, /*group*/ NULL );
setProcTitle ( "worker process", 0 );
_process_chdir = process_chdir;
/// 进入 loop 处理循环
network_worker ( worker_process, process_count, at );
exit ( 0 );
}
int lua_f_startloop(lua_State *L)
{
if(_loop_fd == -1) {
_loop_fd = se_create(4096);
attach_on_exit(on_exit_handler);
atexit(on_exit_handler);
}
luaL_argcheck(L, lua_isfunction(L, 1)
&& !lua_iscfunction(L, 1), 1, "Lua function expected");
job_L = lua_newthread(L);
/*
lua_pushvalue(L, LUA_GLOBALSINDEX);
lua_xmove(L, job_L, 1);
lua_replace(job_L, LUA_GLOBALSINDEX);
lua_xmove(LM, L, 1);*/
lua_pushvalue(L, 1); /* copy entry function to top of L*/
lua_xmove(L, job_L, 1); /* move entry function from L to co */
lua_pushvalue(L, 1); /* move function to top */
lua_xmove(L, job_L, 1); /* move function from L to job_L */
if(lua_resume(job_L, 0) == LUA_ERRRUN) {
lua_gc(job_L, LUA_GCCOLLECT, 0);
if(lua_isstring(job_L, -1)) {
luaL_error(L, lua_tostring(job_L, -1));
}
lua_pop(job_L, 1);
}
LM = job_L;
se_loop(_loop_fd, 10, _do_other_jobs);
return 0;
}
int main ( int argc, char *argv[] )
{
attach_on_exit ( on_master_exit_handler );
/// 初始化进程命令行信息
char *cwd = initProcTitle ( argc, argv );
char *msg = NULL;
if ( getarg ( "cache-size" ) ) {
msg = getarg ( "cache-size" );
YAC_CACHE_SIZE = atoi ( msg );
if(msg[strlen(msg)-1] == 'm')YAC_CACHE_SIZE = YAC_CACHE_SIZE*1024*1024;
else if(msg[strlen(msg)-1] == 'k')YAC_CACHE_SIZE = YAC_CACHE_SIZE*1024;
}
if ( YAC_CACHE_SIZE < 1024*1024*2 ) {
YAC_CACHE_SIZE = 1024*1024*2;
}
if ( !yac_storage_startup ( YAC_CACHE_SIZE/16, YAC_CACHE_SIZE-(YAC_CACHE_SIZE/16), &msg ) ) {
printf ( "Shared memory allocator startup failed at '%s': %s", msg,
strerror ( errno ) );
exit ( 1 );
}
if ( getarg ( "log" ) && strlen ( getarg ( "log" ) ) > 0 ) {
LOG_FD = fopen ( getarg ( "log" ), "a+" );
if ( !LOG_FD ) {
printf ( "fopen %s error\n" , getarg ( "log" ) );
return -1;
}
setvbuf ( LOG_FD , LOG_BUF, _IOFBF , 40960 );
}
if ( getarg ( "errorlog" ) && strlen ( getarg ( "errorlog" ) ) > 0 ) {
ERR_FD = fopen ( getarg ( "errorlog" ), "a+" );
if ( !ERR_FD ) {
printf ( "fopen %s error\n" , getarg ( "errorlog" ) );
return -1;
}
setvbuf ( ERR_FD , ERR_BUF, _IOFBF , 4096 );
}
hostname[1023] = '\0';
gethostname ( hostname, 1000 );
lua_State *L = luaL_newstate();
luaL_openlibs ( L );
lua_getglobal ( L, "_VERSION" );
const char *lua_ver = lua_tostring ( L, -1 );
lua_getglobal ( L, "jit" );
if ( lua_istable ( L, -1 ) ) {
lua_getfield ( L, -1, "version" );
if ( lua_isstring ( L, -1 ) ) {
lua_ver = lua_tostring ( L, -1 );
}
}
sprintf ( hostname, "%s/%s", hostname, lua_ver );
lua_close ( L );
if ( getarg ( "help" ) ) {
printf ( "This is the aLiLua/%s Web Server. Usage:\n"
"\n"
" alilua [options]\n"
"\n"
"Options:\n"
"\n"
" --bind=127.0.0.1:80 server bind. or --bind=80 for bind at 0.0.0.0:80\n"
" --daemon process mode\n"
" --process=number workers\n"
" --log=file-path access log\n"
" --errorlog=file-path error log\n"
" --host-route Special route file path\n"
" --code-cache-ttl number of code cache time(sec)\n"
" --cache-size size of YAC shared memory cache (1m or 4096000k)\n"
" \n"
"\n",
version
);
exit ( 0 );
}
/// 把进程放入后台
if ( getarg ( "daemon" ) ) {
is_daemon = 1;
daemonize();
}
/// 创建4个子进程
if ( getarg ( "process" ) ) {
process_count = atoi ( getarg ( "process" ) );
} else if ( process_count > 32 ) {
process_count = 32;
}
if ( !getarg ( "daemon" ) ) {
process_count = 1;
}
char *bind_addr = "0.0.0.0";
if ( getarg ( "bind" ) ) {
bind_addr = getarg ( "bind" );
}
char *_port;
if ( !strstr ( bind_addr, "." ) ) {
bind_addr = "0.0.0.0";
_port = getarg ( "bind" );
} else {
_port = strstr ( bind_addr, ":" );
if ( _port ) {
bind_addr[strlen ( bind_addr ) - strlen ( _port )] = '\0';
_port = _port + 1;
}
}
int port = default_port;
if ( _port ) {
port = atoi ( _port );
}
if ( port < 1 ) {
port = default_port;
}
int i = 0, status = 0;
{
/// init lua state
_L = luaL_newstate();
lua_gc ( _L, LUA_GCSTOP, 0 );
luaL_openlibs ( _L ); /* Load Lua libraries */
lua_gc ( _L, LUA_GCRESTART, 0 );
if ( getarg ( "code-cache-ttl" ) ) { /// default = 60s
lua_pushnumber ( _L, atoi ( getarg ( "code-cache-ttl" ) ) );
lua_setglobal ( _L, "CODE_CACHE_TTL" );
}
if ( getarg ( "host-route" ) ) {
lua_pushstring ( _L, getarg ( "host-route" ) );
lua_setglobal ( _L, "HOST_ROUTE" );
}
lua_register ( _L, "errorlog", lua_errorlog );
lua_register ( _L, "echo", lua_echo );
lua_register ( _L, "sendfile", lua_sendfile );
lua_register ( _L, "header", lua_header );
lua_register ( _L, "clear_header", lua_clear_header );
lua_register ( _L, "die", lua_die );
lua_register ( _L, "get_post_body", lua_get_post_body );
lua_register ( _L, "check_timeout", lua_check_timeout );
lua_register ( _L, "is_websocket", lua_f_is_websocket );
lua_register ( _L, "upgrade_to_websocket", lua_f_upgrade_to_websocket );
lua_register ( _L, "websocket_send", lua_f_websocket_send );
lua_register ( _L, "sleep", lua_f_sleep );
lua_register ( _L, "random_string", lua_f_random_string );
lua_register ( _L, "file_exists", lua_f_file_exists );
lua_register ( _L, "cache_set", lua_f_cache_set );
lua_register ( _L, "cache_get", lua_f_cache_get );
lua_register ( _L, "cache_del", lua_f_cache_del );
luaopen_fastlz ( _L );
luaopen_coevent ( _L );
luaopen_libfs ( _L );
luaopen_string_utils ( _L );
luaopen_crypto ( _L );
lua_pop ( _L, 1 );
sprintf ( tbuf_4096,
"package.path = '%s/lua-libs/?.lua;' .. package.path package.cpath = '%s/lua-libs/?.so;' .. package.cpath",
cwd, cwd );
luaL_dostring ( _L, tbuf_4096 );
/* Load the file containing the script we are going to run */
status = luaL_loadfile ( _L, "script.lua" );
if ( status || lua_resume ( _L, 0 ) ) {
/* If something went wrong, error message is at the top of */
/* the stack */
fprintf ( stderr, "Couldn't load file: %s\n", lua_tostring ( _L, -1 ) );
exit ( 1 );
}
lua_getglobal ( _L, "main" );
main_handle_ref = luaL_ref ( _L, LUA_REGISTRYINDEX );
}
server_fd = network_bind ( bind_addr, port );
for ( i = 0; i < process_count; i++ ) {
if ( getarg ( "daemon" ) ) {
forkProcess ( worker_main );
} else {
active_cpu ( 0 );
new_thread_i ( worker_main, 0 );
}
}
/// 设置进程归属用户
setProcessUser ( /*user*/ NULL, /*group*/ NULL );
/// 进入主进程处理
master_main();
return 0;
}
int main(int argc, const char **argv)
{
bind_port = default_port;
char *cwd = init_process_title(argc, argv);
char *msg = NULL;
update_time();
if(getarg("cache-size")) {
msg = getarg("cache-size");
YAC_CACHE_SIZE = atoi(msg);
if(msg[strlen(msg) - 1] == 'm') {
YAC_CACHE_SIZE = YAC_CACHE_SIZE * 1024 * 1024;
} else if(msg[strlen(msg) - 1] == 'k') {
YAC_CACHE_SIZE = YAC_CACHE_SIZE * 1024;
}
}
if(YAC_CACHE_SIZE < 1024 * 1024 * 2) {
YAC_CACHE_SIZE = 1024 * 1024 * 2;
}
if(!yac_storage_startup(YAC_CACHE_SIZE / 16, YAC_CACHE_SIZE - (YAC_CACHE_SIZE / 16), &msg)) {
LOGF(ERR, "Shared memory allocator startup failed at '%s': %s", msg, strerror(errno));
exit(1);
}
lua_State *L = luaL_newstate();
if(!L) {
LOGF(ERR, "error for luaL_newstate");
exit(1);
}
luaL_openlibs(L);
lua_getglobal(L, "_VERSION");
const char *lua_ver = lua_tostring(L, -1);
lua_getglobal(L, "jit");
if(lua_istable(L, -1)) {
lua_getfield(L, -1, "version");
if(lua_isstring(L, -1)) {
lua_ver = lua_tostring(L, -1);
}
}
sprintf(hostname, "%s", lua_ver);
lua_close(L);
_L = luaL_newstate();
lua_gc(_L, LUA_GCSTOP, 0);
luaL_openlibs(_L); /* Load Lua libraries */
lua_gc(_L, LUA_GCRESTART, 0);
if(getarg("host-route")) {
lua_pushstring(_L, getarg("host-route"));
lua_setglobal(_L, "HOST_ROUTE");
}
if(!update_vhost_routes(getarg("host-route")) && !getarg("app")) {
LOGF(WARN, "no host-route or app arguments! using defalut settings.");
sprintf(tbuf_4096, "%s/host-route.lua", process_chdir);
update_vhost_routes(tbuf_4096);
}
if(getarg("code-cache-ttl")) { /// default = 60s
lua_pushnumber(_L, atoi(getarg("code-cache-ttl")));
lua_setglobal(_L, "CODE_CACHE_TTL");
} else {
lua_pushnumber(_L, code_cache_ttl);
lua_setglobal(_L, "CODE_CACHE_TTL");
}
lua_getglobal(_L, "require");
lua_pushcfunction(_L, lua_f_package_require);
lua_getfenv(_L, -2);
int ret = lua_setfenv(_L, -2);
lua_setglobal(_L, "require");
lua_pop(_L, 1);
lua_register(_L, "echo", lua_echo);
lua_register(_L, "print_error", lua_print_error);
lua_register(_L, "sendfile", lua_sendfile);
lua_register(_L, "header", lua_header);
lua_register(_L, "clear_header", lua_clear_header);
lua_register(_L, "__end", lua_end);
lua_register(_L, "die", lua_die);
lua_register(_L, "flush", lua_flush);
lua_register(_L, "read_request_body", lua_read_request_body);
lua_register(_L, "get_boundary", lua_f_get_boundary);
lua_register(_L, "check_timeout", lua_check_timeout);
lua_register(_L, "is_websocket", lua_f_is_websocket);
lua_register(_L, "upgrade_to_websocket", lua_f_upgrade_to_websocket);
lua_register(_L, "websocket_send", lua_f_websocket_send);
lua_register(_L, "check_websocket_close", lua_f_check_websocket_close);
lua_register(_L, "router", lua_f_router);
lua_register(_L, "random_string", lua_f_random_string);
lua_register(_L, "file_exists", lua_f_file_exists);
lua_register(_L, "readfile", lua_f_readfile);
lua_register(_L, "filemtime", lua_f_filemtime);
lua_register(_L, "cache_set", lua_f_cache_set);
lua_register(_L, "cache_get", lua_f_cache_get);
lua_register(_L, "cache_del", lua_f_cache_del);
luaopen_fastlz(_L);
luaopen_coevent(_L);
luaopen_libfs(_L);
luaopen_string_utils(_L);
luaopen_i18n(_L);
luaopen_crypto(_L);
lua_pop(_L, 1);
sprintf(tbuf_4096,
"package.path = '%slua-libs/?.lua;' .. package.path package.cpath = '%slua-libs/?.so;' .. package.cpath", cwd, cwd);
luaL_dostring(_L, tbuf_4096);
luaL_dostring(_L, ""
"if not CODE_CACHE_TTL then CODE_CACHE_TTL = 60 end " \
"startloop = nil __CodeCache = {{},{}} __CodeCacheC = {false,false} "
);
if(getarg("accesslog")) {
ACCESS_LOG = open_log(getarg("accesslog"), 40960);
if(!ACCESS_LOG) {
LOGF(ERR, "Couldn't open access log file: %s", getarg("accesslog"));
}
}
if(getarg("ssl-bind") && getarg("ssl-cert") && getarg("ssl-key")) {
ssl_ctx = SSL_CTX_new(SSLv23_server_method());
if(!ssl_ctx) {
LOGF(ERR, "SSL_CTX_new Failed");
exit(1);
}
if(SSL_CTX_use_certificate_file(ssl_ctx, getarg("ssl-cert"), SSL_FILETYPE_PEM) != 1) {
SSL_CTX_free(ssl_ctx);
LOGF(ERR, "SSL_CTX_use_certificate_file");
exit(1);
}
if(SSL_CTX_use_PrivateKey_file(ssl_ctx, getarg("ssl-key"), SSL_FILETYPE_PEM) != 1) {
SSL_CTX_free(ssl_ctx);
LOGF(ERR, "SSL_CTX_use_PrivateKey_file");
exit(1);
}
SSL_CTX_set_options(ssl_ctx, SSL_OP_ALL);
if(getarg("ssl-ca")) {
ssl_epd_idx = SSL_get_ex_new_index(0, NULL, NULL, NULL, NULL);
if(ssl_epd_idx == -1) {
LOGF(ERR, "SSL_get_ex_new_index Failed");
exit(1);
}
if(SSL_CTX_load_verify_locations(ssl_ctx, getarg("ssl-ca"), NULL) != 1) {
SSL_CTX_free(ssl_ctx);
LOGF(ERR, "SSL_CTX_load_verify_locations");
exit(1);
} else {
SSL_CTX_set_verify(ssl_ctx, SSL_VERIFY_PEER | SSL_VERIFY_FAIL_IF_NO_PEER_CERT, verify_callback);
SSL_CTX_set_verify_depth(ssl_ctx, 4);
}
}
}
_shm_serv_status = shm_malloc(sizeof(serv_status_t));
bzero(_shm_serv_status->p, sizeof(serv_status_t));
attach_on_exit(on_master_exit_handler);
return merry_start(argc, argv, help, master_main, on_master_exit_handler, worker_main, 0);
}