chiark - git - mdw - disorder/blob - server/state.c

   1 /*
   2  * This file is part of DisOrder.
   3  * Copyright (C) 2004, 2005, 2007, 2008 Richard Kettlewell
   4  *
   5  * This program is free software; you can redistribute it and/or modify
   6  * it under the terms of the GNU General Public License as published by
   7  * the Free Software Foundation; either version 2 of the License, or
   8  * (at your option) any later version.
   9  *
  10  * This program is distributed in the hope that it will be useful, but
  11  * WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13  * General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU General Public License
  16  * along with this program; if not, write to the Free Software
  17  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
  18  * USA
  19  */
  20
  21 #include <config.h>
  22
  23 #include <stdlib.h>
  24 #include <string.h>
  25 #include <errno.h>
  26 #include <sys/types.h>
  27 #include <sys/socket.h>
  28 #include <sys/stat.h>
  29 #include <unistd.h>
  30 #include <locale.h>
  31 #include <stdio.h>
  32 #include <pcre.h>
  33 #include <netdb.h>
  34 #include <sys/un.h>
  35 #include <netinet/in.h>
  36
  37 #include "event.h"
  38 #include "play.h"
  39 #include "rights.h"
  40 #include "trackdb.h"
  41 #include "state.h"
  42 #include "configuration.h"
  43 #include "log.h"
  44 #include "queue.h"
  45 #include "server-queue.h"
  46 #include "server.h"
  47 #include "printf.h"
  48 #include "addr.h"
  49
  50 static const char *current_unix;
  51 static int current_unix_fd;
  52
  53 static struct addrinfo *current_listen_addrinfo;
  54 static int current_listen_fd;
  55
  56 void quit(ev_source *ev) {
  57   info("shutting down...");
  58   quitting(ev);
  59   trackdb_close();
  60   trackdb_deinit();
  61   info("exiting");
  62   exit(0);
  63 }
  64
  65 static void reset_socket(ev_source *ev) {
  66   const char *new_unix;
  67   struct addrinfo *res;
  68   struct sockaddr_un sun;
  69   char *name;
  70
  71   static const struct addrinfo pref = {
  72     AI_PASSIVE,
  73     PF_INET,
  74     SOCK_STREAM,
  75     IPPROTO_TCP,
  76     0,
  77     0,
  78     0,
  79     0
  80   };
  81
  82   /* unix first */
  83   new_unix = config_get_file("socket");
  84   if(!current_unix || strcmp(current_unix, new_unix)) {
  85     /* either there was no socket, or there was but a different path */
  86     if(current_unix) {
  87       /* stop the old one and remove it from the filesystem */
  88       server_stop(ev, current_unix_fd);
  89       if(unlink(current_unix) < 0)
  90         fatal(errno, "unlink %s", current_unix);
  91     }
  92     /* start the new one */
  93     if(strlen(new_unix) >= sizeof sun.sun_path)
  94       fatal(0, "socket path %s is too long", new_unix);
  95     memset(&sun, 0, sizeof sun);
  96     sun.sun_family = AF_UNIX;
  97     strcpy(sun.sun_path, new_unix);
  98     if(unlink(new_unix) < 0 && errno != ENOENT)
  99       fatal(errno, "unlink %s", new_unix);
 100     if((current_unix_fd = server_start(ev, PF_UNIX, sizeof sun,
 101                                        (const struct sockaddr *)&sun,
 102                                        new_unix)) >= 0) {
 103       current_unix = new_unix;
 104       if(chmod(new_unix, 0777) < 0)
 105         fatal(errno, "error calling chmod %s", new_unix);
 106     } else
 107       current_unix = 0;
 108   }
 109
 110   /* get the new listen config */
 111   if(config->listen.n)
 112     res = get_address(&config->listen, &pref, &name);
 113   else
 114     res = 0;
 115
 116   if((res && !current_listen_addrinfo)
 117      || (current_listen_addrinfo
 118          && (!res
 119              || addrinfocmp(res, current_listen_addrinfo)))) {
 120     /* something has to change */
 121     if(current_listen_addrinfo) {
 122       /* delete the old listener */
 123       server_stop(ev, current_listen_fd);
 124       freeaddrinfo(current_listen_addrinfo);
 125       current_listen_addrinfo = 0;
 126     }
 127     if(res) {
 128       /* start the new listener */
 129       if((current_listen_fd = server_start(ev, res->ai_family, res->ai_addrlen,
 130                                            res->ai_addr, name)) >= 0) {
 131         current_listen_addrinfo = res;
 132         res = 0;
 133       }
 134     }
 135   }
 136   /* if res is still set it needs freeing */
 137   if(res)
 138     freeaddrinfo(res);
 139 }
 140
 141 int reconfigure(ev_source *ev, int reload) {
 142   int need_another_rescan = 0;
 143   int ret = 0;
 144
 145   D(("reconfigure(%d)", reload));
 146   if(reload) {
 147     need_another_rescan = trackdb_rescan_cancel();
 148     trackdb_close();
 149     if(config_read(1))
 150       ret = -1;
 151     else {
 152       /* Tell the speaker it needs to reload its config too. */
 153       speaker_reload();
 154       info("%s: installed new configuration", configfile);
 155     }
 156     trackdb_open(TRACKDB_NO_UPGRADE);
 157   } else
 158     /* We only allow for upgrade at startup */
 159     trackdb_open(TRACKDB_CAN_UPGRADE);
 160   if(need_another_rescan)
 161     trackdb_rescan(ev, 1/*check*/);
 162   if(!ret) {
 163     queue_read();
 164     recent_read();
 165     reset_socket(ev);
 166   }
 167   return ret;
 168 }
 169
 170 /*
 171 Local Variables:
 172 c-basic-offset:2
 173 comment-column:40
 174 End:
 175 */