etc/netatalk/netatalk.c

   1 /*
   2  * Copyright (c) 1990,1993 Regents of The University of Michigan.
   3  * All Rights Reserved.  See COPYRIGHT.
   4  */
   5
   6 #ifdef HAVE_CONFIG_H
   7 #include "config.h"
   8 #endif /* HAVE_CONFIG_H */
   9
  10 #include <stdio.h>
  11 #include <stdlib.h>
  12 #include <string.h>
  13 #include <signal.h>
  14 #include <sys/param.h>
  15 #include <sys/uio.h>
  16 #include <sys/time.h>
  17 #include <sys/socket.h>
  18 #include <sys/poll.h>
  19 #include <errno.h>
  20 #include <sys/wait.h>
  21 #include <sys/resource.h>
  22
  23 #include <atalk/logger.h>
  24 #include <atalk/adouble.h>
  25 #include <atalk/compat.h>
  26 #include <atalk/dsi.h>
  27 #include <atalk/afp.h>
  28 #include <atalk/paths.h>
  29 #include <atalk/util.h>
  30 #include <atalk/server_child.h>
  31 #include <atalk/server_ipc.h>
  32 #include <atalk/errchk.h>
  33 #include <atalk/globals.h>
  34 #include <atalk/netatalk_conf.h>
  35 #include <atalk/bstrlib.h>
  36 #include <atalk/bstradd.h>
  37
  38 #include <event2/event.h>
  39
  40 /* how many seconds we wait to shutdown from SIGTERM before we send SIGKILL */
  41 #define KILL_GRACETIME 5
  42
  43 /* forward declaration */
  44 static pid_t run_process(const char *path, ...);
  45 static void kill_childs(int sig, ...);
  46
  47 /* static variables */
  48 static AFPObj obj;
  49 static sig_atomic_t got_chldsig;
  50 static pid_t afpd_pid = -1,  cnid_metad_pid = -1, dbus_pid = -1;
  51 static uint afpd_restarts, cnid_metad_restarts, dbus_restarts;
  52 static struct event_base *base;
  53 struct event *sigterm_ev, *sigquit_ev, *sigchld_ev, *timer_ev;
  54 static int in_shutdown;
  55 static const char *dbus_path;
  56
  57 /******************************************************************
  58  * Misc stuff
  59  ******************************************************************/
  60
  61 /* Set Tracker Miners to index all our volumes */
  62 static int set_sl_volumes(void)
  63 {
  64     EC_INIT;
  65     const struct vol *volumes, *vol;
  66     struct bstrList *vollist = bstrListCreate();
  67     bstring sep = bfromcstr(", ");
  68     bstring volnamelist = NULL, cmd = NULL;
  69
  70     EC_NULL_LOG( volumes = getvolumes() );
  71
  72     for (vol = volumes; vol; vol = vol->v_next) {
  73         bstring volnamequot = bformat("'%s'", vol->v_path);
  74         bstrListPush(vollist, volnamequot);
  75     }
  76
  77     volnamelist = bjoin(vollist, sep);
  78     cmd = bformat("gsettings set org.freedesktop.Tracker.Miner.Files index-recursive-directories \"[%s]\"", bdata(volnamelist));
  79     LOG(log_debug, logtype_sl, "set_sl_volumes: %s", bdata(cmd));
  80     system(bdata(cmd));
  81
  82 EC_CLEANUP:
  83     if (cmd)
  84         bdestroy(cmd);
  85     if (sep)
  86         bdestroy(sep);
  87     if (vollist)
  88         bstrListDestroy(vollist);
  89     if (volnamelist)
  90         bdestroy(volnamelist);
  91     EC_EXIT;
  92 }
  93
  94 /******************************************************************
  95  * libevent helper functions
  96  ******************************************************************/
  97
  98 /* libevent logging callback */
  99 static void libevent_logmsg_cb(int severity, const char *msg)
 100 {
 101     switch (severity) {
 102     case _EVENT_LOG_DEBUG:
 103         LOG(log_debug, logtype_default, "libevent: %s", msg);
 104         break;
 105     case _EVENT_LOG_MSG:
 106         LOG(log_info, logtype_default, "libevent: %s", msg);
 107         break;
 108     case _EVENT_LOG_WARN:
 109         LOG(log_warning, logtype_default, "libevent: %s", msg);
 110         break;
 111     case _EVENT_LOG_ERR:
 112         LOG(log_error, logtype_default, "libevent: %s", msg);
 113         break;
 114     default:
 115         LOG(log_error, logtype_default, "libevent: %s", msg);
 116         break; /* never reached */
 117     }
 118 }
 119
 120 /******************************************************************
 121  * libevent event callbacks
 122  ******************************************************************/
 123
 124 /* SIGTERM callback */
 125 static void sigterm_cb(evutil_socket_t fd, short what, void *arg)
 126 {
 127     sigset_t sigs;
 128     struct timeval tv;
 129
 130     LOG(log_info, logtype_afpd, "Exiting on SIGTERM");
 131
 132     if (in_shutdown)
 133         return;
 134     in_shutdown = 1;
 135
 136     /* block any signal but SIGCHLD */
 137     sigfillset(&sigs);
 138     sigdelset(&sigs, SIGCHLD);
 139     sigprocmask(SIG_SETMASK, &sigs, NULL);
 140
 141     /* add 10 sec timeout timer, remove all events but SIGCHLD */
 142     tv.tv_sec = KILL_GRACETIME;
 143     tv.tv_usec = 0;
 144     event_base_loopexit(base, &tv);
 145     event_del(sigterm_ev);
 146     event_del(sigquit_ev);
 147     event_del(timer_ev);
 148
 149     system("tracker-control -t");
 150     kill_childs(SIGTERM, &afpd_pid, &cnid_metad_pid, &dbus_pid, NULL);
 151 }
 152
 153 /* SIGQUIT callback */
 154 static void sigquit_cb(evutil_socket_t fd, short what, void *arg)
 155 {
 156     LOG(log_note, logtype_afpd, "Exiting on SIGQUIT");
 157     system("tracker-control -t");
 158     kill_childs(SIGQUIT, &afpd_pid, &cnid_metad_pid, &dbus_pid, NULL);
 159 }
 160
 161 /* SIGCHLD callback */
 162 static void sigchld_cb(evutil_socket_t fd, short what, void *arg)
 163 {
 164     int status, i;
 165     pid_t pid;
 166
 167     while ((pid = waitpid(-1, &status, WNOHANG)) > 0) {
 168         if (WIFEXITED(status)) {
 169             if (WEXITSTATUS(status))
 170                 LOG(log_info, logtype_default, "child[%d]: exited %d", pid, WEXITSTATUS(status));
 171             else
 172                 LOG(log_info, logtype_default, "child[%d]: done", pid);
 173         } else {
 174             if (WIFSIGNALED(status))
 175                 LOG(log_info, logtype_default, "child[%d]: killed by signal %d", pid, WTERMSIG(status));
 176             else
 177                 LOG(log_info, logtype_default, "child[%d]: died", pid);
 178         }
 179
 180         if (pid == afpd_pid)
 181             afpd_pid = -1;
 182         else if (pid == cnid_metad_pid)
 183             cnid_metad_pid = -1;
 184         else if (pid == dbus_pid)
 185             dbus_pid = -1;
 186         else
 187             LOG(log_error, logtype_afpd, "Bad pid: %d", pid);
 188     }
 189
 190     if (in_shutdown && afpd_pid == -1 && cnid_metad_pid == -1 && dbus_pid == -1)
 191         event_base_loopbreak(base);
 192 }
 193
 194 /* timer callback */
 195 static void timer_cb(evutil_socket_t fd, short what, void *arg)
 196 {
 197     static int i = 0;
 198
 199     if (in_shutdown)
 200         return;
 201
 202     if (afpd_pid == -1) {
 203         afpd_restarts++;
 204         LOG(log_note, logtype_afpd, "Restarting 'afpd' (restarts: %u)", afpd_restarts);
 205         if ((afpd_pid = run_process(_PATH_AFPD, "-d", "-F", obj.options.configfile, NULL)) == -1) {
 206             LOG(log_error, logtype_default, "Error starting 'afpd'");
 207         }
 208     }
 209
 210     if (cnid_metad_pid == -1) {
 211         cnid_metad_restarts++;
 212         LOG(log_note, logtype_afpd, "Restarting 'cnid_metad' (restarts: %u)", cnid_metad_restarts);
 213         if ((cnid_metad_pid = run_process(_PATH_CNID_METAD, "-d", "-F", obj.options.configfile, NULL)) == -1) {
 214             LOG(log_error, logtype_default, "Error starting 'cnid_metad'");
 215         }
 216     }
 217
 218     if (dbus_pid == -1) {
 219         dbus_restarts++;
 220         LOG(log_note, logtype_afpd, "Restarting 'dbus' (restarts: %u)", dbus_restarts);
 221         if ((dbus_pid = run_process(dbus_path, "--config-file=" _PATH_CONFDIR "dbus.session.conf", NULL)) == -1) {
 222             LOG(log_error, logtype_default, "Error starting '%s'", dbus_path);
 223         }
 224     }
 225 }
 226
 227 /******************************************************************
 228  * helper functions
 229  ******************************************************************/
 230
 231 /* kill processes passed as varargs of type "pid_t *", terminate list with NULL */
 232 static void kill_childs(int sig, ...)
 233 {
 234     va_list args;
 235     pid_t *pid;
 236
 237     va_start(args, sig);
 238
 239     while ((pid = va_arg(args, pid_t *)) != NULL) {
 240         if (*pid == -1)
 241             continue;
 242         kill(*pid, sig);
 243     }
 244     va_end(args);
 245 }
 246
 247 /* this get called when error conditions are met that require us to exit gracefully */
 248 static void netatalk_exit(int ret)
 249 {
 250     server_unlock(_PATH_NETATALK_LOCK);
 251     exit(ret);
 252 }
 253
 254 /* this forks() and exec() "path" with varags as argc[] */
 255 static pid_t run_process(const char *path, ...)
 256 {
 257     int ret, i = 0;
 258     char *myargv[10];
 259     va_list args;
 260     pid_t pid;
 261
 262     if ((pid = fork()) < 0) {
 263         LOG(log_error, logtype_cnid, "error in fork: %s", strerror(errno));
 264         return -1;
 265     }
 266
 267     if (pid == 0) {
 268         myargv[i++] = (char *)path;
 269         va_start(args, path);
 270         while ((myargv[i++] = va_arg(args, char *)) != NULL)
 271             ;
 272         va_end(args);
 273
 274         ret = execv(path, myargv);
 275
 276         /* Yikes! We're still here, so exec failed... */
 277         LOG(log_error, logtype_cnid, "Fatal error in exec: %s", strerror(errno));
 278         exit(1);
 279     }
 280     return pid;
 281 }
 282
 283 static void usage(void)
 284 {
 285     printf("usage: netatalk [-F configfile] \n");
 286 }
 287
 288 int main(int argc, char **argv)
 289 {
 290     const char *configfile = NULL;
 291     int c, ret, debug = 0;
 292     sigset_t blocksigs;
 293     struct timeval tv;
 294
 295     /* Log SIGBUS/SIGSEGV SBT */
 296     fault_setup(NULL);
 297
 298     while ((c = getopt(argc, argv, ":dF:")) != -1) {
 299         switch(c) {
 300         case 'd':
 301             debug = 1;
 302             break;
 303         case 'F':
 304             obj.cmdlineconfigfile = strdup(optarg);
 305             break;
 306         default:
 307             usage();
 308             exit(EXIT_FAILURE);
 309         }
 310     }
 311
 312     if (check_lockfile("netatalk", _PATH_NETATALK_LOCK) != 0)
 313         exit(EXITERR_SYS);
 314
 315     if (!debug && daemonize(0, 0) != 0)
 316         exit(EXITERR_SYS);
 317
 318     if (create_lockfile("netatalk", _PATH_NETATALK_LOCK) != 0)
 319         exit(EXITERR_SYS);
 320
 321     sigfillset(&blocksigs);
 322     sigprocmask(SIG_SETMASK, &blocksigs, NULL);
 323
 324     if (afp_config_parse(&obj, "netatalk") != 0)
 325         netatalk_exit(EXITERR_CONF);
 326
 327     load_volumes(&obj, NULL);
 328
 329     event_set_log_callback(libevent_logmsg_cb);
 330     event_set_fatal_callback(netatalk_exit);
 331
 332     LOG(log_note, logtype_default, "Netatalk AFP server starting");
 333
 334     if ((afpd_pid = run_process(_PATH_AFPD, "-d", "-F", obj.options.configfile, NULL)) == -1) {
 335         LOG(log_error, logtype_afpd, "Error starting 'cnid_metad'");
 336         netatalk_exit(EXITERR_CONF);
 337     }
 338
 339     if ((cnid_metad_pid = run_process(_PATH_CNID_METAD, "-d", "-F", obj.options.configfile, NULL)) == -1) {
 340         LOG(log_error, logtype_afpd, "Error starting 'cnid_metad'");
 341         netatalk_exit(EXITERR_CONF);
 342     }
 343
 344     if ((base = event_base_new()) == NULL) {
 345         LOG(log_error, logtype_afpd, "Error starting event loop");
 346         netatalk_exit(EXITERR_CONF);
 347     }
 348
 349     sigterm_ev = event_new(base, SIGTERM, EV_SIGNAL, sigterm_cb, NULL);
 350     sigquit_ev = event_new(base, SIGQUIT, EV_SIGNAL | EV_PERSIST, sigquit_cb, NULL);
 351     sigchld_ev = event_new(base, SIGCHLD, EV_SIGNAL | EV_PERSIST, sigchld_cb, NULL);
 352     timer_ev = event_new(base, -1, EV_PERSIST, timer_cb, NULL);
 353
 354     tv.tv_sec = 1;
 355     tv.tv_usec = 0;
 356
 357     event_add(sigterm_ev, NULL);
 358     event_add(sigquit_ev, NULL);
 359     event_add(sigchld_ev, NULL);
 360     event_add(timer_ev, &tv);
 361
 362     sigfillset(&blocksigs);
 363     sigdelset(&blocksigs, SIGTERM);
 364     sigdelset(&blocksigs, SIGQUIT);
 365     sigdelset(&blocksigs, SIGCHLD);
 366     sigprocmask(SIG_SETMASK, &blocksigs, NULL);
 367
 368     dbus_path = iniparser_getstring(obj.iniconfig, INISEC_GLOBAL, "dbus daemon path", "/bin/dbus-daemon");
 369     LOG(log_debug, logtype_default, "DBUS: '%s'", dbus_path);
 370     if ((dbus_pid = run_process(dbus_path, "--config-file=" _PATH_CONFDIR "dbus-session.conf", NULL)) == -1) {
 371         LOG(log_error, logtype_default, "Error starting '%s'", dbus_path);
 372         netatalk_exit(EXITERR_CONF);
 373     }
 374
 375     /* Allow dbus some time to start up */
 376     sleep(1);
 377
 378     setenv("DBUS_SESSION_BUS_ADDRESS", "unix:path=/tmp/spotlight.ipc", 1);
 379     set_sl_volumes();
 380     system(TRACKER_PREFIX "/bin/tracker-control -s");
 381
 382     /* run the event loop */
 383     ret = event_base_dispatch(base);
 384
 385     if (afpd_pid != -1 || cnid_metad_pid != -1 || dbus_pid != -1) {
 386         if (afpd_pid != -1)
 387             LOG(log_error, logtype_afpd, "AFP service did not shutdown, killing it");
 388         if (cnid_metad_pid != -1)
 389             LOG(log_error, logtype_afpd, "CNID database service did not shutdown, killing it");
 390         if (dbus_pid != -1)
 391             LOG(log_error, logtype_afpd, "DBUS session daemon still running, killing it");
 392         kill_childs(SIGKILL, &afpd_pid, &cnid_metad_pid, &dbus_pid, NULL);
 393     }
 394
 395     LOG(log_note, logtype_afpd, "Netatalk AFP server exiting");
 396
 397     netatalk_exit(ret);
 398 }