rust_scheduler::rust_scheduler(rust_kernel *kernel,
rust_srv *srv,
int id) :
ref_count(1),
interrupt_flag(0),
_log(srv, this),
log_lvl(log_debug),
srv(srv),
// TODO: calculate a per scheduler name.
name("main"),
newborn_tasks(this, "newborn"),
running_tasks(this, "running"),
blocked_tasks(this, "blocked"),
dead_tasks(this, "dead"),
cache(this),
kernel(kernel),
id(id),
min_stack_size(kernel->env->min_stack_size),
env(kernel->env)
{
LOGPTR(this, "new dom", (uintptr_t)this);
isaac_init(this, &rctx);
#ifndef __WIN32__
pthread_attr_init(&attr);
pthread_attr_setstacksize(&attr, 1024 * 1024);
pthread_attr_setdetachstate(&attr, true);
#endif
if (!tls_initialized)
init_tls();
}
rust_sched_loop::rust_sched_loop(rust_scheduler *sched, int id, bool killed) :
_log(this),
id(id),
should_exit(false),
cached_c_stack(NULL),
extra_c_stack(NULL),
cached_big_stack(NULL),
extra_big_stack(NULL),
dead_task(NULL),
killed(killed),
pump_signal(NULL),
kernel(sched->kernel),
sched(sched),
log_lvl(log_debug),
min_stack_size(kernel->env->min_stack_size),
local_region(false, kernel->env->detailed_leaks, kernel->env->poison_on_free),
// FIXME #2891: calculate a per-scheduler name.
name("main")
{
LOGPTR(this, "new dom", (uintptr_t)this);
rng_init(&rng, kernel->env->rust_seed, NULL, 0);
if (!tls_initialized)
init_tls();
}
void ATTR_EXPORT
test_compression(void)
{
VERBOSE(SYSTEM, "--- testing compression\n");
init_tls();
static const unsigned char buffer[] = "qwerqwerqwerqwerqwerqwerqwerqwerqwerqwer";
static unsigned char buffer2[sizeof(buffer) + 10];
static unsigned char buffer3[sizeof(buffer) + 10];
uint8_t * out_buf = NULL;
unsigned int compressed_sz = 0;
struct tls_compress * pcomp = &(get_tpd()->logger.compress);
compress(pcomp, buffer, sizeof(buffer), &out_buf, &compressed_sz);
VERBOSE(SYSTEM, "compressed data size: %d\n", compressed_sz);
memcpy(buffer2, out_buf, compressed_sz);
clear_tls();
int uncompressed_sz = sizeof(buffer);
decompress(buffer2, compressed_sz, buffer3,
&uncompressed_sz);
VERBOSE(SYSTEM, "uncompressed data: %s || sz=%d\n", buffer3,
uncompressed_sz);
assert(uncompressed_sz == sizeof(buffer));
VERBOSE(SYSTEM, "--- testing compression2\n");
init_tls();
pcomp = &(get_tpd()->logger.compress);
compressed_sz = sizeof(buffer) + 10;
compress2(pcomp, buffer, sizeof(buffer), buffer2,
&compressed_sz);
clear_tls();
VERBOSE(SYSTEM, "compressed data size: %u\n", compressed_sz);
uncompressed_sz = sizeof(buffer);
decompress(buffer2, compressed_sz, buffer3, &uncompressed_sz);
VERBOSE(SYSTEM, "uncompressed data: %s || sz=%d\n", buffer3,
uncompressed_sz);
assert(uncompressed_sz == sizeof(buffer));
}
void ATTR_EXPORT
test_tls(void)
{
VERBOSE(SYSTEM, "--- test thread local storage\n");
init_tls();
struct thread_private_data * tpd = get_tpd();
VERBOSE(SYSTEM, "tpd reside on %p\n", tpd);
VERBOSE(SYSTEM, "pid=%d, tid=%d, tnr=%d\n", tpd->pid, tpd->tid, tpd->tnr);
clear_tls();
}
static void klib_init()
{
init_tls();
init_kthread();
init_halloc();
init_salloc();
init_malloc();
//test_malloc();
}
static int tls_upgrade(rsconn_t *c) {
SSL *ssl;
SSL_CTX *ctx;
if (first_tls)
init_tls();
ctx = SSL_CTX_new(SSLv23_client_method());
SSL_CTX_set_mode(ctx, SSL_MODE_AUTO_RETRY);
c->tls = ssl = SSL_new(ctx);
c->send = tls_send;
c->recv = tls_recv;
SSL_set_fd(ssl, c->s);
/* SSL_CTX_free(ctx) // check whether this is safe - it should be since ssl has the reference ... */
return SSL_connect(ssl);
}
static int thread_entry(void* arg, size_t ep)
{
if (init_tls())
return -ENOMEM;
//vma_dump();
entry_point_t call_ep = (entry_point_t) ep;
call_ep(arg);
/* After finishing the task, we will return here and call the
cleanup function, which calls the scheduler */
leave_kernel_task();
return 0;
}
/**
* Main routine, start of the program execution.
* \param argc the number of arguments
* \param argv pointer to the arguments array
* \return don't return on sucess, -1 on error
* \see main_loop
*/
int main(int argc, char** argv)
{
/* configure by default logging to syslog */
int cfg_log_stderr = 0;
FILE* cfg_stream;
int c,r;
char *tmp;
int tmp_len;
int port;
int proto;
char *options;
int ret;
unsigned int seed;
int rfd;
/*init*/
ret=-1;
my_argc=argc; my_argv=argv;
/*init pkg mallocs (before parsing cfg or cmd line !)*/
if (init_pkg_mallocs()==-1)
goto error00;
init_route_lists();
/* process command line (get port no, cfg. file path etc) */
opterr=0;
options="f:cCm:b:l:n:N:rRvdDETSVhw:t:u:g:P:G:W:o:";
while((c=getopt(argc,argv,options))!=-1){
switch(c){
case 'f':
cfg_file=optarg;
break;
case 'C':
config_check |= 2;
case 'c':
if (config_check==3)
break;
config_check |= 1;
cfg_log_stderr=1; /* force stderr logging */
break;
case 'm':
shm_mem_size=strtol(optarg, &tmp, 10) * 1024 * 1024;
if (tmp &&(*tmp)){
LM_ERR("bad shmem size number: -m %s\n", optarg);
goto error00;
};
break;
case 'b':
maxbuffer=strtol(optarg, &tmp, 10);
if (tmp &&(*tmp)){
LM_ERR("bad max buffer size number: -b %s\n", optarg);
goto error00;
}
break;
case 'l':
if (parse_phostport(optarg, strlen(optarg), &tmp, &tmp_len,
&port, &proto)<0){
LM_ERR("bad -l address specifier: %s\n", optarg);
goto error00;
}
tmp[tmp_len]=0; /* null terminate the host */
/* add a new addr. to our address list */
if (add_listen_iface(tmp, port, proto, 0, 0, 0)!=0){
LM_ERR("failed to add new listen address\n");
goto error00;
}
break;
case 'n':
children_no=strtol(optarg, &tmp, 10);
if ((tmp==0) ||(*tmp)){
LM_ERR("bad process number: -n %s\n", optarg);
goto error00;
}
break;
case 'v':
check_via=1;
break;
case 'r':
received_dns|=DO_DNS;
break;
case 'R':
received_dns|=DO_REV_DNS;
case 'd':
#ifdef CHANGEABLE_DEBUG_LEVEL
(*debug)++;
#else
debug++;
#endif
break;
case 'D':
dont_fork=1;
break;
case 'E':
cfg_log_stderr=1;
break;
case 'T':
#ifdef USE_TCP
tcp_disable=1;
#else
LM_WARN("tcp support not compiled in\n");
#endif
break;
case 'S':
#ifdef USE_SCTP
sctp_disable=1;
#else
LM_WARN("sctp support not compiled in\n");
#endif
break;
case 'N':
#ifdef USE_TCP
tcp_children_no=strtol(optarg, &tmp, 10);
if ((tmp==0) ||(*tmp)){
LM_ERR("bad process number: -N %s\n", optarg);
goto error00;
}
#else
LM_WARN("tcp support not compiled in\n");
#endif
break;
case 'W':
#ifdef USE_TCP
tcp_poll_method=get_poll_type(optarg);
if (tcp_poll_method==POLL_NONE){
LM_ERR("bad poll method name: -W %s\ntry "
"one of %s.\n", optarg, poll_support);
goto error00;
}
#else
LM_WARN("tcp support not compiled in\n");
#endif
break;
case 'V':
printf("version: %s\n", version);
printf("flags: %s\n", flags );
print_ct_constants();
printf("%s\n",id);
printf("%s compiled on %s with %s\n", __FILE__,
compiled, COMPILER );
exit(0);
break;
case 'h':
printf("version: %s\n", version);
printf("%s",help_msg);
exit(0);
break;
case 'w':
working_dir=optarg;
break;
case 't':
chroot_dir=optarg;
break;
case 'u':
user=optarg;
break;
case 'g':
group=optarg;
break;
case 'P':
pid_file=optarg;
break;
case 'G':
pgid_file=optarg;
break;
case 'o':
if (add_arg_var(optarg) < 0)
LM_ERR("cannot add option %s\n", optarg);
break;
case '?':
if (isprint(optopt))
LM_ERR("Unknown option `-%c`.\n", optopt);
else
LM_ERR("Unknown option character `\\x%x`.\n", optopt);
goto error00;
case ':':
LM_ERR("Option `-%c` requires an argument.\n", optopt);
goto error00;
default:
abort();
}
}
log_stderr = cfg_log_stderr;
/* fill missing arguments with the default values*/
if (cfg_file==0) cfg_file=CFG_FILE;
/* load config file or die */
cfg_stream=fopen (cfg_file, "r");
if (cfg_stream==0){
LM_ERR("loading config file(%s): %s\n", cfg_file,
strerror(errno));
goto error00;
}
/* seed the prng, try to use /dev/urandom if possible */
/* no debugging information is logged, because the standard
log level prior the config file parsing is L_NOTICE */
seed=0;
if ((rfd=open("/dev/urandom", O_RDONLY))!=-1){
try_again:
if (read(rfd, (void*)&seed, sizeof(seed))==-1){
if (errno==EINTR) goto try_again; /* interrupted by signal */
LM_WARN("could not read from /dev/urandom (%d)\n", errno);
}
LM_DBG("initialize the pseudo random generator from "
"/dev/urandom\n");
LM_DBG("read %u from /dev/urandom\n", seed);
close(rfd);
}else{
LM_WARN("could not open /dev/urandom (%d)\n", errno);
LM_WARN("using a unsafe seed for the pseudo random number generator");
}
seed+=getpid()+time(0);
LM_DBG("seeding PRNG with %u\n", seed);
srand(seed);
LM_DBG("test random number %u\n", rand());
/*register builtin modules*/
register_builtin_modules();
#ifdef USE_TLS
/* initialize default TLS domains,
must be done before reading the config */
if (pre_init_tls()<0){
LM_CRIT("could not pre_init_tls, exiting...\n");
goto error00;
}
#endif /* USE_TLS */
if (preinit_black_lists()!=0) {
LM_CRIT("failed to alloc black list's anchor\n");
goto error00;
}
/* parse the config file, prior to this only default values
e.g. for debugging settings will be used */
yyin=cfg_stream;
if ((yyparse()!=0)||(cfg_errors)){
LM_ERR("bad config file (%d errors)\n", cfg_errors);
goto error00;
}
if (config_check>1 && check_rls()!=0) {
LM_ERR("bad function call in config file\n");
return ret;
}
#ifdef EXTRA_DEBUG
print_rl();
#endif
/* init the resolver, before fixing the config */
resolv_init();
/* fix parameters */
if (port_no<=0) port_no=SIP_PORT;
#ifdef USE_TLS
if (tls_port_no<=0) tls_port_no=SIPS_PORT;
#endif
if (children_no<=0) children_no=CHILD_NO;
#ifdef USE_TCP
if (!tcp_disable){
if (tcp_children_no<=0) tcp_children_no=children_no;
}
#endif
if (working_dir==0) working_dir="/";
/* get uid/gid */
if (user){
if (user2uid(&uid, &gid, user)<0){
LM_ERR("bad user name/uid number: -u %s\n", user);
goto error00;
}
}
if (group){
if (group2gid(&gid, group)<0){
LM_ERR("bad group name/gid number: -u %s\n", group);
goto error00;
}
}
if (fix_all_socket_lists()!=0){
LM_ERR("failed to initialize list addresses\n");
goto error00;
}
/* print all the listen addresses */
printf("Listening on \n");
print_all_socket_lists();
printf("Aliases: \n");
/*print_aliases();*/
print_aliases();
printf("\n");
if (dont_fork){
LM_WARN("no fork mode %s\n",
(udp_listen)?(
(udp_listen->next)?" and more than one listen address found"
"(will use only the first one)":""
):"and no udp listen address found" );
}
if (config_check){
LM_NOTICE("config file ok, exiting...\n");
return 0;
}
time(&startup_time);
/*init shm mallocs
* this must be here
* -to allow setting shm mem size from the command line
* => if shm_mem should be settable from the cfg file move
* everything after
* -it must be also before init_timer and init_tcp
* -it must be after we know uid (so that in the SYSV sems case,
* the sems will have the correct euid)
* --andrei */
if (init_shm_mallocs()==-1)
goto error;
/*init timer, before parsing the cfg!*/
if (init_timer()<0){
LM_CRIT("could not initialize timer, exiting...\n");
goto error;
}
#ifdef USE_TCP
if (!tcp_disable){
/*init tcp*/
if (init_tcp()<0){
LM_CRIT("could not initialize tcp, exiting...\n");
goto error;
}
}
#ifdef USE_TLS
if (!tls_disable){
/* init tls*/
if (init_tls()<0){
LM_CRIT("could not initialize tls, exiting...\n");
goto error;
}
}
#endif /* USE_TLS */
#endif /* USE_TCP */
/* init_daemon? */
if (!dont_fork){
if ( daemonize((log_name==0)?argv[0]:log_name, &own_pgid) <0 )
goto error;
}
/* install signal handlers */
if (install_sigs() != 0){
LM_ERR("could not install the signal handlers\n");
goto error;
}
#ifdef CHANGEABLE_DEBUG_LEVEL
#ifdef SHM_MEM
debug=shm_malloc(sizeof(int));
if (debug==0) {
LM_ERR("ERROR: out of memory\n");
goto error;
}
*debug = debug_init;
#else
LM_WARN("no shm mem support compiled -> changeable debug "
"level turned off\n");
#endif
#endif
if (disable_core_dump) set_core_dump(0, 0);
else set_core_dump(1, shm_mem_size+PKG_MEM_POOL_SIZE+4*1024*1024);
if (open_files_limit>0){
if(increase_open_fds(open_files_limit)<0){
LM_ERR("ERROR: error could not increase file limits\n");
goto error;
}
}
/* print OpenSIPS version to log for history tracking */
LM_NOTICE("version: %s\n", version);
/* print some data about the configuration */
#ifdef SHM_MEM
LM_INFO("using %ld Mb shared memory\n", ((shm_mem_size/1024)/1024));
#endif
LM_INFO("using %i Mb private memory per process\n", ((PKG_MEM_POOL_SIZE/1024)/1024));
/* init serial forking engine */
if (init_serialization()!=0) {
LM_ERR("failed to initialize serialization\n");
goto error;
}
/* Init statistics */
if (init_stats_collector()<0) {
LM_ERR("failed to initialize statistics\n");
goto error;
}
/* Init MI */
if (init_mi_core()<0) {
LM_ERR("failed to initialize MI core\n");
goto error;
}
/* init black list engine */
if (init_black_lists()!=0) {
LM_CRIT("failed to init black lists\n");
goto error;
}
/* init resolver's blacklist */
if (resolv_blacklist_init()!=0) {
LM_CRIT("failed to create DNS blacklist\n");
goto error;
}
/* init modules */
if (init_modules() != 0) {
LM_ERR("error while initializing modules\n");
goto error;
}
/* register route timers */
if(register_route_timers() < 0) {
LM_ERR("Failed to register timer\n");
goto error;
}
/* check pv context list */
if(pv_contextlist_check() != 0) {
LM_ERR("used pv context that was not defined\n");
goto error;
}
/* init multi processes support */
if (init_multi_proc_support()!=0) {
LM_ERR("failed to init multi-proc support\n");
goto error;
}
#ifdef PKG_MALLOC
/* init stats support for pkg mem */
if (init_pkg_stats(counted_processes)!=0) {
LM_ERR("failed to init stats for pkg\n");
goto error;
}
#endif
/* fix routing lists */
if ( (r=fix_rls())!=0){
LM_ERR("failed to fix configuration with err code %d\n", r);
goto error;
};
ret=main_loop();
error:
/*kill everything*/
kill_all_children(SIGTERM);
/*clean-up*/
cleanup(0);
error00:
return ret;
}
int main(int argc, char** argv)
{
FILE* cfg_stream;
int c,r;
char *tmp;
int tmp_len;
int port;
int proto;
char *options;
int ret;
unsigned int seed;
int rfd;
/*init*/
ret=-1;
my_argc=argc; my_argv=argv;
/*init pkg mallocs (before parsing cfg or cmd line !)*/
if (init_pkg_mallocs()==-1)
goto error;
#ifdef DBG_MSG_QA
fprintf(stderr, "WARNING: ser startup: "
"DBG_MSG_QA enabled, ser may exit abruptly\n");
#endif
/* process command line (get port no, cfg. file path etc) */
opterr=0;
options=
#ifdef STATS
"s:"
#endif
"f:cm:b:l:n:N:rRvdDETVhw:t:u:g:P:G:i:x:";
while((c=getopt(argc,argv,options))!=-1){
switch(c){
case 'f':
cfg_file=optarg;
break;
case 'c':
config_check=1;
log_stderr=1; /* force stderr logging */
break;
case 's':
#ifdef STATS
stat_file=optarg;
#endif
break;
case 'm':
shm_mem_size=strtol(optarg, &tmp, 10) * 1024 * 1024;
if (tmp &&(*tmp)){
fprintf(stderr, "bad shmem size number: -m %s\n",
optarg);
goto error;
};
LOG(L_INFO, "ser: shared memory: %ld bytes\n",
shm_mem_size );
break;
case 'b':
maxbuffer=strtol(optarg, &tmp, 10);
if (tmp &&(*tmp)){
fprintf(stderr, "bad max buffer size number: -p %s\n",
optarg);
goto error;
}
break;
case 'l':
if (parse_phostport(optarg, &tmp, &tmp_len,
&port, &proto)<0){
fprintf(stderr, "bad -l address specifier: %s\n",
optarg);
goto error;
}
tmp[tmp_len]=0; /* null terminate the host */
/* add a new addr. to our address list */
if (add_listen_iface(tmp, port, proto, 0)!=0){
fprintf(stderr, "failed to add new listen address\n");
goto error;
}
break;
case 'n':
children_no=strtol(optarg, &tmp, 10);
if ((tmp==0) ||(*tmp)){
fprintf(stderr, "bad process number: -n %s\n",
optarg);
goto error;
}
break;
case 'v':
check_via=1;
break;
case 'r':
received_dns|=DO_DNS;
break;
case 'R':
received_dns|=DO_REV_DNS;
case 'd':
debug++;
break;
case 'D':
dont_fork=1;
break;
case 'E':
log_stderr=1;
break;
case 'T':
#ifdef USE_TCP
tcp_disable=1;
#else
fprintf(stderr,"WARNING: tcp support not compiled in\n");
#endif
break;
case 'N':
#ifdef USE_TCP
tcp_children_no=strtol(optarg, &tmp, 10);
if ((tmp==0) ||(*tmp)){
fprintf(stderr, "bad process number: -N %s\n",
optarg);
goto error;
}
#else
fprintf(stderr,"WARNING: tcp support not compiled in\n");
#endif
break;
case 'V':
printf("version: %s\n", version);
printf("flags: %s\n", flags );
print_ct_constants();
printf("%s\n",id);
printf("%s compiled on %s with %s\n", __FILE__,
compiled, COMPILER );
exit(0);
break;
case 'h':
printf("version: %s\n", version);
printf("%s",help_msg);
exit(0);
break;
case 'w':
working_dir=optarg;
break;
case 't':
chroot_dir=optarg;
break;
case 'u':
user=optarg;
break;
case 'g':
group=optarg;
break;
case 'P':
pid_file=optarg;
break;
case 'G':
pgid_file=optarg;
break;
case 'i':
fifo=optarg;
break;
case 'x':
unixsock_name=optarg;
break;
case '?':
if (isprint(optopt))
fprintf(stderr, "Unknown option `-%c´.\n", optopt);
else
fprintf(stderr,
"Unknown option character `\\x%x´.\n",
optopt);
goto error;
case ':':
fprintf(stderr,
"Option `-%c´ requires an argument.\n",
optopt);
goto error;
default:
abort();
}
}
/* fill missing arguments with the default values*/
if (cfg_file==0) cfg_file=CFG_FILE;
/* load config file or die */
cfg_stream=fopen (cfg_file, "r");
if (cfg_stream==0){
fprintf(stderr, "ERROR: loading config file(%s): %s\n", cfg_file,
strerror(errno));
goto error;
}
/* seed the prng */
/* try to use /dev/urandom if possible */
seed=0;
if ((rfd=open("/dev/urandom", O_RDONLY))!=-1){
try_again:
if (read(rfd, (void*)&seed, sizeof(seed))==-1){
if (errno==EINTR) goto try_again; /* interrupted by signal */
LOG(L_WARN, "WARNING: could not read from /dev/urandom (%d)\n",
errno);
}
DBG("read %u from /dev/urandom\n", seed);
close(rfd);
}else{
LOG(L_WARN, "WARNING: could not open /dev/urandom (%d)\n", errno);
}
seed+=getpid()+time(0);
DBG("seeding PRNG with %u\n", seed);
srand(seed);
DBG("test random number %u\n", rand());
/* register a diagnostic FIFO command - moved to fifo server - bogdan
if (register_core_fifo()<0) {
LOG(L_CRIT, "unable to register core FIFO commands\n");
goto error;
}*/
/*register builtin modules*/
register_builtin_modules();
yyin=cfg_stream;
if ((yyparse()!=0)||(cfg_errors)){
fprintf(stderr, "ERROR: bad config file (%d errors)\n", cfg_errors);
goto error;
}
print_rl();
/* fix parameters */
if (port_no<=0) port_no=SIP_PORT;
#ifdef USE_TLS
if (tls_port_no<=0) tls_port_no=SIPS_PORT;
#endif
if (children_no<=0) children_no=CHILD_NO;
#ifdef USE_TCP
if (!tcp_disable){
if (tcp_children_no<=0) tcp_children_no=children_no;
}
#endif
if (working_dir==0) working_dir="/";
/* get uid/gid */
if (user){
if (user2uid(&uid, &gid, user)<0){
fprintf(stderr, "bad user name/uid number: -u %s\n", user);
goto error;
}
}
if (group){
if (group2gid(&gid, group)<0){
fprintf(stderr, "bad group name/gid number: -u %s\n", group);
goto error;
}
}
/* fix sock/fifo uid/gid */
if (sock_user){
if (user2uid(&sock_uid, 0, sock_user)<0){
fprintf(stderr, "bad socket user name/uid number %s\n", user);
goto error;
}
}
if (sock_group){
if (group2gid(&sock_gid, sock_group)<0){
fprintf(stderr, "bad group name/gid number: -u %s\n", group);
goto error;
}
}
if (fix_all_socket_lists()!=0){
fprintf(stderr, "failed to initialize list addresses\n");
goto error;
}
/* print all the listen addresses */
printf("Listening on \n");
print_all_socket_lists();
printf("Aliases: \n");
/*print_aliases();*/
print_aliases();
printf("\n");
if (dont_fork){
fprintf(stderr, "WARNING: no fork mode %s\n",
(udp_listen)?(
(udp_listen->next)?" and more than one listen address found"
"(will use only the the first one)":""
):"and no udp listen address found" );
}
if (config_check){
fprintf(stderr, "config file ok, exiting...\n");
goto error;
}
/*init shm mallocs
* this must be here
* -to allow setting shm mem size from the command line
* => if shm_mem should be settable from the cfg file move
* everything after
* -it must be also before init_timer and init_tcp
* -it must be after we know uid (so that in the SYSV sems case,
* the sems will have the correct euid)
* --andrei */
if (init_shm_mallocs()==-1)
goto error;
/*init timer, before parsing the cfg!*/
if (init_timer()<0){
LOG(L_CRIT, "could not initialize timer, exiting...\n");
goto error;
}
#ifdef USE_TCP
if (!tcp_disable){
/*init tcp*/
if (init_tcp()<0){
LOG(L_CRIT, "could not initialize tcp, exiting...\n");
goto error;
}
}
#ifdef USE_TLS
if (!tls_disable){
/* init tls*/
if (init_tls()<0){
LOG(L_CRIT, "could not initialize tls, exiting...\n");
goto error;
}
}
#endif /* USE_TLS */
#endif /* USE_TCP */
/* init_daemon? */
if (!dont_fork){
if ( daemonize(argv[0]) <0 ) goto error;
}
if (install_sigs() != 0){
fprintf(stderr, "ERROR: could not install the signal handlers\n");
goto error;
}
/*alloc pids*/
#ifdef SHM_MEM
pt=shm_malloc(sizeof(struct process_table)*process_count());
#else
pt=pkg_malloc(sizeof(struct process_table)*process_count());
#endif
if (pt==0){
fprintf(stderr, "ERROR: out of memory\n");
goto error;
}
memset(pt, 0, sizeof(struct process_table)*process_count());
if (disable_core_dump) set_core_dump(0, 0);
else set_core_dump(1, shm_mem_size+PKG_MEM_POOL_SIZE+4*1024*1024);
if (open_files_limit>0){
if(increase_open_fds(open_files_limit)<0){
fprintf(stderr, "ERROR: error could not increase file limits\n");
goto error;
}
}
if (init_modules() != 0) {
fprintf(stderr, "ERROR: error while initializing modules\n");
goto error;
}
/* fix routing lists */
if ( (r=fix_rls())!=0){
fprintf(stderr, "ERROR: error %d while trying to fix configuration\n",
r);
goto error;
};
#ifdef STATS
if (init_stats( dont_fork ? 1 : children_no )==-1) goto error;
#endif
ret=main_loop();
/*kill everything*/
kill_all_children(SIGTERM);
/*clean-up*/
cleanup(0);
return ret;
error:
/*kill everything*/
kill_all_children(SIGTERM);
/*clean-up*/
cleanup(0);
return -1;
}
int child_function(int id)
{
struct socket_info *b,*bp,*a,*ap;
int sleep_time;
is_main = 0;
b = bind_address;
bp = bind_address_port;
a = alternate_address;
ap = alternate_address_port;
//each of the childs has the main address different from the other
if (id == 1)
{
bind_address = b;
bind_address_port = bp;
alternate_address = a;
alternate_address_port = ap;
}
if (id == 2)
{
bind_address = bp;
bind_address_port = b;
alternate_address = ap;
alternate_address_port = a;
}
if (id == 3)
{
bind_address = a;
bind_address_port = ap;
alternate_address = b;
alternate_address_port = bp;
}
if (id == 4)
{
bind_address = ap;
bind_address_port = a;
alternate_address = bp;
alternate_address_port = b;
}
if (id <= 4)
{
is_udp = 1;
LOG("Child %d [%d] started ...\n",id,getpid());
LOG("Listening on:\n");
LOG("\tb [%d] %.*s:%d\n",id,bind_address->name.len,bind_address->name.s,bind_address->port_no);
LOG("\tbp[%d] %.*s:%d\n",id,bind_address_port->name.len,bind_address_port->name.s,bind_address_port->port_no);
LOG("\ta [%d] %.*s:%d\n",id,alternate_address->name.len,alternate_address->name.s,alternate_address->port_no);
LOG("\tap[%d] %.*s:%d\n",id,alternate_address_port->name.len,alternate_address_port->name.s,alternate_address_port->port_no);
/*
for(i=1;i<THREADS_PER_SOCKET;i++)
{
fork();
}
*/
return udp_rcv_loop();
}
#ifdef USE_TLS
//start the TLS server
if (id == 5)
{
//tls server
is_tls = 1;
init_tls();
tls_bind_address = duplicate_sock_info(bind_address);
if (tls_init(tls_bind_address) != 1)
{
LOG("TLS bind failed:trying after 25 seconds\n");
sleep(10);
if (tls_init(tls_bind_address) != 1)
{
LOG("Second TLS bind failed.Givving Up\n");
for(;;) pause();
}
else LOG("TLS succeeded from second attempt\n");
}
return tls_rcv_loop();
}
#endif
#ifdef USE_TLS
//start the timer
if (id == 6)
{
//timer server scope is to detect expired credentials
is_timer = 1;
LOG("[%d] timer activated\n",getpid());
sleep_time = -1;
while(1)
{
if (sleep_time == -1) sleep(EXPIRE-TIMER_RESOLUTION);//EXPIRE-TIMER_RESOLUTION is better
else sleep(sleep_time+10);//to be sure it expired
remove_old_entries(&sleep_time);
LOG("[%d] timer activates now sleep_time=%d\n",getpid(),sleep_time);
}
}
#endif
return 1;
}
