initial checkin

1 files changed, 411 insertions, 0 deletions

diff --git a/rhdropbox.c b/rhdropbox.c
new file mode 100644
index 0000000..f09d96b
--- /dev/null
+++ b/rhdropbox.c
@@ -0,0 +1,411 @@
+/*
+ *  rhdropbox
+ *
+ *  Copyright (C) 2009 Christian Pointner <equinox@helsinki.at>
+ *
+ *  This file is part of rhdropbox.
+ *
+ *  rhdropbox is free software: you can redistribute it and/or modify
+ *  it under the terms of the GNU General Public License as published by
+ *  the Free Software Foundation, either version 3 of the License, or
+ *  any later version.
+ *
+ *  rhdropbox is distributed in the hope that it will be useful,
+ *  but WITHOUT ANY WARRANTY; without even the implied warranty of
+ *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ *  GNU General Public License for more details.
+ *
+ *  You should have received a copy of the GNU General Public License
+ *  along with rhdropbox. If not, see <http://www.gnu.org/licenses/>.
+ */
+
+#include "datatypes.h"
+
+#include <stdlib.h>
+#include <stdio.h>
+#include <string.h>
+#include <errno.h>
+
+#include "log.h"
+#include "sig_handler.h"
+#include "options.h"
+
+#include "command_queue.h"
+#include "client_list.h"
+#include "key_value_storage.h"
+
+#include "daemon.h"
+#include "utils.h"
+
+int send_response(int fd, const char* response)
+{
+  if(!response)
+    return -1;
+
+  if(fd < 0)
+    return 0;
+
+  int ret = send_string(fd, response);
+  do {
+    ret = write(fd, "\n", 1);
+  } while(!ret || (ret == -1 && errno == EINTR));
+
+  if(ret > 0)
+    return 0;
+
+  return ret;
+}
+
+/*     client_t* client; */
+/*     int listener_cnt = 0; */
+/*     for(client = client_lst; client; client = client->next) */
+/*       if(client->status_listener && client->fd != fd) { */
+/*         send_response(client->fd, buf); */
+/*         listener_cnt++; */
+/*       } */
+/*     log_printf(DEBUG, "sent status to %d additional listeners", listener_cnt); */
+
+void process_cmd_listen(const char* param, int fd, client_t* client_lst)
+{
+  client_t* listener = client_find(client_lst, fd);
+  if(listener) {
+    if(!param) {
+      listener->status_listener = 1;
+      listener->request_listener = 1;
+    }
+    else {
+      if(!strncmp(param, "status", 6))
+        listener->status_listener = 1;
+      else if(!strncmp(param, "request", 7))
+        listener->request_listener = 1;
+      else if(!strncmp(param, "none", 4)) {
+        listener->request_listener = 0;
+        listener->status_listener = 0;
+      }
+      else {
+        log_printf(DEBUG, "unkown listener type '%s'", param);
+        send_response(fd, "Error: listen: unkown type");
+        return;
+      }
+    }
+    log_printf(DEBUG, "listener %d requests %s messages", fd, param ? param:"all");
+  }
+  else {
+    log_printf(ERROR, "unable to add listener %d", fd);
+    send_response(fd, "Error: listen: client not found in client list?!");
+  }
+}
+
+int process_cmd(const char* cmd, int fd, cmd_t **cmd_q, client_t* client_lst, options_t* opt)
+{
+  log_printf(DEBUG, "processing command from %d", fd);
+
+  if(!cmd_q || !cmd)
+    return -1;
+  
+  cmd_id_t cmd_id;
+  if(!strncmp(cmd, "add", 3))
+    cmd_id = ADD;
+  else if(!strncmp(cmd, "remove", 6))
+    cmd_id = REMOVE;
+  else if(!strncmp(cmd, "status", 6))
+    cmd_id = STATUS;
+  else if(!strncmp(cmd, "log", 3))
+    cmd_id = LOG;
+  else if(!strncmp(cmd, "listen", 6)) {
+    cmd_id = LISTEN;
+  }
+  else {
+    log_printf(WARNING, "unknown command '%s'", cmd);
+    send_response(fd, "Error: unknown command");
+    return 0;
+  }
+  char* param = strchr(cmd, ' ');
+  if(param) 
+    param++;
+
+  if(cmd_id == ADD || cmd_id == REMOVE) {
+    char* resp;
+    asprintf(&resp, "Request: %s", cmd);
+    if(resp) {
+      char* linefeed = strchr(resp, '\n');
+      if(linefeed) linefeed[0] = 0;
+      client_t* client;
+      int listener_cnt = 0;
+      for(client = client_lst; client; client = client->next)
+        if(client->request_listener && client->fd != fd) {
+          send_response(client->fd, resp);
+          listener_cnt++;
+        }
+      free(resp);
+      log_printf(DEBUG, "sent request to %d additional listeners", listener_cnt);
+    }
+// else silently ignore memory alloc error
+  }
+
+  switch(cmd_id) {
+  case ADD: break;
+  case REMOVE: break;
+  case STATUS: break
+  case LOG: {
+    if(param && param[0])
+      log_printf(NOTICE, "ext msg: %s", param); 
+    else
+      log_printf(DEBUG, "ignoring empty ext log message");
+    break;
+  }
+  case LISTEN: process_cmd_listen(param, fd, client_lst); break;
+  }
+  
+  return 0;
+}
+
+int main_loop(int switch_fd, int cmd_listen_fd, options_t* opt)
+{
+  log_printf(NOTICE, "entering main loop");
+
+  fd_set readfds, tmpfds;
+  FD_ZERO(&readfds);
+  FD_SET(switch_fd, &readfds);
+  FD_SET(cmd_listen_fd, &readfds);
+  int max_fd = switch_fd > cmd_listen_fd ? switch_fd : cmd_listen_fd;
+  cmd_t* cmd_q = NULL;
+  client_t* client_lst = NULL;
+
+  read_buffer_t switch_buffer;
+  switch_buffer.offset = 0;
+
+  int sig_fd = signal_init();
+  if(sig_fd < 0)
+    return -1;
+  FD_SET(sig_fd, &readfds);
+  max_fd = (max_fd < sig_fd) ? sig_fd : max_fd;
+
+  int return_value = 0;
+
+  char* channel;
+  if(opt->mode_ == MODE_MASTER && opt->channel_ == CHAN_MAIN) channel = "master_main";
+  else if(opt->mode_ == MODE_MASTER && opt->channel_ == CHAN_MUSIC) channel = "master_music";
+  else if(opt->mode_ == MODE_STANDBY && opt->channel_ == CHAN_MAIN) channel = "standby_main";
+  else if(opt->mode_ == MODE_STANDBY && opt->channel_ == CHAN_MUSIC) channel = "standby_music";
+  
+  char* cmd_param = strdup("*0M1*0ii1");
+  char* ch_nr = key_value_storage_find(&opt->alias_table_, channel);
+  if(!ch_nr || ch_nr[0] == 0 || ch_nr[1] == 0 || ch_nr[2] != 0) {
+    log_printf(ERROR, "invalid channel name or number: %s", channel);
+    free(cmd_param);
+    return_value = -1;
+  }
+  else {
+    cmd_param[6] = ch_nr[0];
+    cmd_param[7] = ch_nr[1];
+    return_value = cmd_push(&cmd_q, -1, SWITCH, cmd_param);
+    free(cmd_param);
+    send_command(switch_fd, cmd_q);
+  }
+
+  struct timeval timeout;
+  while(!return_value) {
+    memcpy(&tmpfds, &readfds, sizeof(tmpfds));
+
+    timeout.tv_sec = 0;
+    timeout.tv_usec = 200000;
+    int ret = select(max_fd+1, &tmpfds, NULL, NULL, &timeout);
+    if(ret == -1 && errno != EINTR) {
+      log_printf(ERROR, "select returned with error: %s", strerror(errno));
+      return_value = -1;
+      break;
+    }
+    if(ret == -1)
+      continue;
+    if(!ret) {
+      if(cmd_q && cmd_has_expired(*cmd_q)) {
+        log_printf(ERROR, "last command expired");
+        cmd_pop(&cmd_q);
+      }
+      else
+        continue;
+    }
+
+    if(FD_ISSET(sig_fd, &tmpfds)) {
+      if(signal_handle()) {
+        return_value = 1;
+        break;
+      }
+    }
+   
+    if(FD_ISSET(switch_fd, &tmpfds)) {
+      return_value = process_switch(&switch_buffer, switch_fd, &cmd_q, client_lst);
+      if(return_value)
+        break;
+    }
+
+    if(FD_ISSET(cmd_listen_fd, &tmpfds)) {
+      int new_fd = accept(cmd_listen_fd, NULL, NULL);
+      if(new_fd < 0) {
+        log_printf(ERROR, "accept returned with error: %s", strerror(errno));
+        return_value = -1;
+        break;
+      }  
+      log_printf(DEBUG, "new command connection (fd=%d)", new_fd);
+      FD_SET(new_fd, &readfds);
+      max_fd = (max_fd < new_fd) ? new_fd : max_fd;
+      fcntl(new_fd, F_SETFL, O_NONBLOCK);
+      client_add(&client_lst, new_fd);
+    }
+
+    client_t* lst = client_lst;
+    while(lst) {
+      if(FD_ISSET(lst->fd, &tmpfds)) {
+        return_value = nonblock_recvline(&(lst->buffer), lst->fd, &cmd_q, client_lst, opt);
+        if(return_value == 2) {
+          log_printf(DEBUG, "removing closed command connection (fd=%d)", lst->fd);
+          client_t* deletee = lst;
+          lst = lst->next;
+          FD_CLR(deletee->fd, &readfds);
+          client_remove(&client_lst, deletee->fd);
+          return_value = 0;
+          continue;
+        }
+        if(return_value)
+          break;
+
+      }
+      lst = lst->next;
+    }
+
+    if(cmd_q && !cmd_q->sent)
+      send_command(switch_fd, cmd_q);
+  }
+
+  cmd_clear(&cmd_q);
+  client_clear(&client_lst);
+  signal_stop();
+  return return_value;
+}
+
+int main(int argc, char* argv[])
+{
+  log_init();
+
+  options_t opt;
+  int ret = options_parse(&opt, argc, argv);
+  if(ret) {
+    if(ret > 0) {
+      fprintf(stderr, "syntax error near: %s\n\n", argv[ret]);
+    }
+    if(ret == -2) {
+      fprintf(stderr, "memory error on options_parse, exiting\n");
+    }
+    if(ret == -3) {
+      fprintf(stderr, "syntax error: mode name must be either master or standby\n");
+    }
+    if(ret == -4) {
+      fprintf(stderr, "syntax error: channel name must be either main or music\n");
+    }
+    if(ret == -5) {
+      fprintf(stderr, "syntax error: invalid baudrate\n");
+    }
+
+    if(ret != -2)
+      options_print_usage();
+
+    options_clear(&opt);
+    log_close();
+    exit(ret);
+  }
+  string_list_element_t* tmp = opt.log_targets_.first_;
+  if(!tmp) {
+    log_add_target("syslog:3,rhdropbox,daemon");
+  }
+  else {
+    while(tmp) {
+      ret = log_add_target(tmp->string_);
+      if(ret) {
+        switch(ret) {
+        case -2: fprintf(stderr, "memory error on log_add_target, exitting\n"); break;
+        case -3: fprintf(stderr, "unknown log target: '%s', exitting\n", tmp->string_); break;
+        case -4: fprintf(stderr, "this log target is only allowed once: '%s', exitting\n", tmp->string_); break;
+        default: fprintf(stderr, "syntax error near: '%s', exitting\n", tmp->string_); break;
+        }
+        
+        options_clear(&opt);
+        log_close();
+        exit(ret);
+      }
+      tmp = tmp->next_;
+    }
+  }
+  log_printf(NOTICE, "just started...");
+  if(options_parse_post(&opt)) {
+    options_clear(&opt);
+    log_close();
+    exit(-1);
+  }
+
+  priv_info_t priv;
+  if(opt.username_)
+    if(priv_init(&priv, opt.username_, opt.groupname_)) {
+      options_clear(&opt);
+      log_close();
+      exit(-1);
+    }
+
+  FILE* pid_file = NULL;
+  if(opt.pid_file_) {
+    pid_file = fopen(opt.pid_file_, "w");
+    if(!pid_file) {
+      log_printf(WARNING, "unable to open pid file: %s", strerror(errno));
+    }
+  }
+
+  if(opt.chroot_dir_)
+    if(do_chroot(opt.chroot_dir_)) {
+      options_clear(&opt);
+      log_close();
+      exit(-1);
+    }
+  if(opt.username_)
+    if(priv_drop(&priv)) {
+      options_clear(&opt);
+      log_close();
+      exit(-1);
+    }
+
+  if(opt.daemonize_) {
+    pid_t oldpid = getpid();
+    daemonize();
+    log_printf(INFO, "running in background now (old pid: %d)", oldpid);
+  }
+
+  if(pid_file) {
+    pid_t pid = getpid();
+    fprintf(pid_file, "%d", pid);
+    fclose(pid_file);
+  }
+
+  int cmd_listen_fd = init_command_socket(opt.command_sock_);
+  if(cmd_listen_fd < 0) {
+    options_clear(&opt);
+    log_close();
+    exit(-1);
+  }
+  
+  ret = main_loop(switch_fd, cmd_listen_fd, &opt);
+
+  close(cmd_listen_fd);
+  if(switch_fd > 0)
+    close(switch_fd);
+
+  if(!ret)
+    log_printf(NOTICE, "normal shutdown");
+  else if(ret < 0)
+    log_printf(NOTICE, "shutdown after error");
+  else
+    log_printf(NOTICE, "shutdown after signal");
+
+  options_clear(&opt);
+  log_close();
+
+  return ret;
+}