X-Git-Url: https://arthur.barton.de/cgi-bin/gitweb.cgi?a=blobdiff_plain;f=etc%2Fafpd%2Fmain.c;h=f49daf6f140bc65e428d7ed444e3b23f66540723;hb=7db833b1021f3daa8e30966e6f025b2a894348dc;hp=c549fa09a49349aaee6f4c5e24c6d6ca015b7bbd;hpb=7be2a68aa70fa5641e944ad0aed6f699736d3de9;p=netatalk.git diff --git a/etc/afpd/main.c b/etc/afpd/main.c index c549fa09..f49daf6f 100644 --- a/etc/afpd/main.c +++ b/etc/afpd/main.c @@ -38,22 +38,12 @@ #include "uam_auth.h" #include "afp_zeroconf.h" -#ifdef TRU64 -#include -#include -#include - -static int argc = 0; -static char **argv = NULL; -#endif /* TRU64 */ - #define AFP_LISTENERS 32 #define FDSET_SAFETY 5 -unsigned char nologin = 0; -struct afp_options default_options; +unsigned char nologin = 0; -static AFPConfig *configs; +static AFPObj obj; static server_child *server_children; static sig_atomic_t reloadconfig = 0; static sig_atomic_t gotsigchld = 0; @@ -65,45 +55,37 @@ static int fdset_size; /* current allocated size */ static int fdset_used; /* number of used elements */ static int disasociated_ipc_fd; /* disasociated sessions uses this fd for IPC */ -#ifdef TRU64 -void afp_get_cmdline( int *ac, char ***av) -{ - *ac = argc; - *av = argv; -} -#endif /* TRU64 */ +static afp_child_t *dsi_start(AFPObj *obj, DSI *dsi, server_child *server_children); /* This is registered with atexit() */ static void afp_exit(void) { if (parent_or_child == 0) /* Only do this in the parent */ - server_unlock(default_options.pidfile); + server_unlock(_PATH_AFPDLOCK); } /* ------------------ initialize fd set we are waiting for. */ -static void fd_set_listening_sockets(void) +static void fd_set_listening_sockets(const AFPObj *config) { - AFPConfig *config; + DSI *dsi; - for (config = configs; config; config = config->next) { - if (config->fd < 0) /* for proxies */ - continue; - fdset_add_fd(default_options.connections + AFP_LISTENERS + FDSET_SAFETY, + for (dsi = config->dsi; dsi; dsi = dsi->next) { + fdset_add_fd(config->options.connections + AFP_LISTENERS + FDSET_SAFETY, &fdset, &polldata, &fdset_used, &fdset_size, - config->fd, + dsi->serversock, LISTEN_FD, - config); + dsi); } - if (default_options.flags & OPTION_KEEPSESSIONS) - fdset_add_fd(default_options.connections + AFP_LISTENERS + FDSET_SAFETY, + if (config->options.flags & OPTION_KEEPSESSIONS) + fdset_add_fd(config->options.connections + AFP_LISTENERS + FDSET_SAFETY, &fdset, &polldata, &fdset_used, @@ -113,25 +95,21 @@ static void fd_set_listening_sockets(void) NULL); } -static void fd_reset_listening_sockets(void) +static void fd_reset_listening_sockets(const AFPObj *config) { - AFPConfig *config; + const DSI *dsi; - for (config = configs; config; config = config->next) { - if (config->fd < 0) /* for proxies */ - continue; - fdset_del_fd(&fdset, &polldata, &fdset_used, &fdset_size, config->fd); + for (dsi = config->dsi; dsi; dsi = dsi->next) { + fdset_del_fd(&fdset, &polldata, &fdset_used, &fdset_size, dsi->serversock); } - if (default_options.flags & OPTION_KEEPSESSIONS) + if (config->options.flags & OPTION_KEEPSESSIONS) fdset_del_fd(&fdset, &polldata, &fdset_used, &fdset_size, disasociated_ipc_fd); } /* ------------------ */ static void afp_goaway(int sig) { - AFPConfig *config; - switch( sig ) { case SIGTERM: @@ -141,7 +119,7 @@ static void afp_goaway(int sig) LOG(log_note, logtype_afpd, "AFP Server shutting down on SIGTERM"); break; case SIGQUIT: - if (default_options.flags & OPTION_KEEPSESSIONS) { + if (obj.options.flags & OPTION_KEEPSESSIONS) { LOG(log_note, logtype_afpd, "AFP Server shutting down on SIGQUIT, NOT disconnecting clients"); } else { LOG(log_note, logtype_afpd, "AFP Server shutting down on SIGQUIT"); @@ -152,11 +130,8 @@ static void afp_goaway(int sig) if (server_children) server_child_kill(server_children, CHILD_DSIFORK, sig); - for (config = configs; config; config = config->next) - if (config->server_cleanup) - config->server_cleanup(config); - server_unlock(default_options.pidfile); - exit(0); + server_unlock(_PATH_AFPDLOCK); + _exit(0); break; case SIGUSR1 : @@ -238,50 +213,38 @@ static int setlimits(void) int main(int ac, char **av) { - AFPConfig *config; fd_set rfds; void *ipc; struct sigaction sv; sigset_t sigs; int ret; -#ifdef TRU64 - argc = ac; - argv = av; - set_auth_parameters( ac, av ); -#endif /* TRU64 */ - /* Parse argv args and initialize default options */ - afp_options_init(&default_options); - if (!afp_options_parse(ac, av, &default_options)) - exit(EXITERR_CONF); + afp_options_parse_cmdline(&obj, ac, av); - if (check_lockfile("afpd", default_options.pidfile) != 0) + if (check_lockfile("afpd", _PATH_AFPDLOCK) != 0) exit(EXITERR_SYS); - if (!(default_options.flags & OPTION_DEBUG) && (daemonize(0, 0) != 0)) + if (!(obj.options.flags & OPTION_DEBUG) && (daemonize(0, 0) != 0)) exit(EXITERR_SYS); - if (create_lockfile("afpd", default_options.pidfile) != 0) + if (create_lockfile("afpd", _PATH_AFPDLOCK) != 0) exit(EXITERR_SYS); /* Log SIGBUS/SIGSEGV SBT */ fault_setup(NULL); + atexit(afp_exit); - /* Default log setup: log to syslog */ - set_processname("afpd"); - setuplog("default log_note"); + if (afp_config_parse(&obj) != 0) + exit(EXITERR_CONF); /* Save the user's current umask */ - default_options.save_mask = umask( default_options.umask ); - - atexit(afp_exit); + obj.options.save_mask = umask(obj.options.umask); /* install child handler for asp and dsi. we do this before afp_goaway * as afp_goaway references stuff from here. * XXX: this should really be setup after the initial connections. */ - if (!(server_children = server_child_alloc(default_options.connections, - CHILD_NFORKS))) { + if (!(server_children = server_child_alloc(obj.options.connections, CHILD_NFORKS))) { LOG(log_error, logtype_afpd, "main: server_child alloc: %s", strerror(errno) ); exit(EXITERR_SYS); } @@ -336,7 +299,6 @@ int main(int ac, char **av) exit(EXITERR_SYS); } - sigemptyset( &sv.sa_mask ); sigaddset(&sv.sa_mask, SIGALRM); sigaddset(&sv.sa_mask, SIGHUP); @@ -380,7 +342,7 @@ int main(int ac, char **av) sigaddset(&sigs, SIGCHLD); pthread_sigmask(SIG_BLOCK, &sigs, NULL); - if (!(configs = configinit(&default_options))) { + if (configinit(&obj) != 0) { LOG(log_error, logtype_afpd, "main: no servers configured"); exit(EXITERR_CONF); } @@ -391,12 +353,12 @@ int main(int ac, char **av) /* watch atp, dsi sockets and ipc parent/child file descriptor. */ - if (default_options.flags & OPTION_KEEPSESSIONS) { + if (obj.options.flags & OPTION_KEEPSESSIONS) { LOG(log_note, logtype_afpd, "Activating continous service"); disasociated_ipc_fd = ipc_server_uds(_PATH_AFP_IPC); } - fd_set_listening_sockets(); + fd_set_listening_sockets(&obj); /* set limits */ (void)setlimits(); @@ -428,22 +390,17 @@ int main(int ac, char **av) if (reloadconfig) { nologin++; auth_unload(); - fd_reset_listening_sockets(); + fd_reset_listening_sockets(&obj); LOG(log_info, logtype_afpd, "re-reading configuration file"); - for (config = configs; config; config = config->next) - if (config->server_cleanup) - config->server_cleanup(config); - - /* configfree close atp socket used for DDP tickle, there's an issue - * with atp tid. */ - configfree(configs, NULL); - if (!(configs = configinit(&default_options))) { + + configfree(&obj, NULL); + if (configinit(&obj) != 0) { LOG(log_error, logtype_afpd, "config re-read: no servers configured"); exit(EXITERR_CONF); } - fd_set_listening_sockets(); + fd_set_listening_sockets(&obj); nologin = 0; reloadconfig = 0; @@ -466,11 +423,9 @@ int main(int ac, char **av) switch (polldata[i].fdtype) { case LISTEN_FD: - config = (AFPConfig *)polldata[i].data; - /* config->server_start is afp_config.c:dsi_start() for DSI */ - if (child = config->server_start(config, configs, server_children)) { + if (child = dsi_start(&obj, (DSI *)polldata[i].data, server_children)) { /* Add IPC fd to select fd set */ - fdset_add_fd(default_options.connections + AFP_LISTENERS + FDSET_SAFETY, + fdset_add_fd(obj.options.connections + AFP_LISTENERS + FDSET_SAFETY, &fdset, &polldata, &fdset_used, @@ -489,7 +444,7 @@ int main(int ac, char **av) fdset_del_fd(&fdset, &polldata, &fdset_used, &fdset_size, child->ipc_fds[0]); close(child->ipc_fds[0]); child->ipc_fds[0] = -1; - if ((default_options.flags & OPTION_KEEPSESSIONS) && child->disasociated) { + if ((obj.options.flags & OPTION_KEEPSESSIONS) && child->disasociated) { LOG(log_note, logtype_afpd, "main: removing reattached child[%u]", child->pid); server_child_remove(server_children, CHILD_DSIFORK, child->pid); } @@ -514,7 +469,7 @@ int main(int ac, char **av) break; } child->disasociated = 1; - fdset_add_fd(default_options.connections + AFP_LISTENERS + FDSET_SAFETY, + fdset_add_fd(obj.options.connections + AFP_LISTENERS + FDSET_SAFETY, &fdset, &polldata, &fdset_used, @@ -534,3 +489,27 @@ int main(int ac, char **av) return 0; } + +static afp_child_t *dsi_start(AFPObj *obj, DSI *dsi, server_child *server_children) +{ + afp_child_t *child = NULL; + + if (!(child = dsi_getsession(dsi, + server_children, + obj->options.tickleval))) { + LOG(log_error, logtype_afpd, "dsi_start: session error: %s", strerror(errno)); + return NULL; + } + + /* we've forked. */ + if (parent_or_child == 1) { + configfree(obj, dsi); + obj->ipc_fd = child->ipc_fds[1]; + close(child->ipc_fds[0]); /* Close parent IPC fd */ + free(child); + afp_over_dsi(obj); /* start a session */ + exit (0); + } + + return child; +}