X-Git-Url: https://git.realraum.at/?a=blobdiff_plain;f=door_daemon%2Fdoor_daemon.c;h=cf126688d12a2aa858afc785367ec7f48270de27;hb=98b2f4efdb5ac839baa0db88a75c0bf7fcd988d2;hp=b88aa00f66728fa91df0656dd07a5bd9febe4dbc;hpb=6773f3ff20c2826dc0f32e4d20b73572769f6e40;p=svn42.git

diff --git a/door_daemon/door_daemon.c b/door_daemon/door_daemon.c
index b88aa00..cf12668 100644
--- a/door_daemon/door_daemon.c
+++ b/door_daemon/door_daemon.c
@@ -29,50 +29,221 @@
 
 #include "log.h"
 #include "sig_handler.h"
-//#include "options.h"
+#include "options.h"
+
+#include "command_queue.h"
 
 #include "daemon.h"
 #include "sysexec.h"
 
-int process_cmd(int fd)
+int init_command_socket(const char* path)
 {
-  log_printf(INFO, "processing command from %d", fd);
+  int fd = socket(AF_UNIX, SOCK_STREAM, 0);
+  if(fd < 0) {
+    log_printf(ERROR, "unable to open socket: %s", strerror(errno));
+    return -1;
+  }
 
-  char* buffer[100];
-  int ret = 0;
+  struct sockaddr_un local;
+  local.sun_family = AF_UNIX;
+  strcpy(local.sun_path, path); // TODO: strlen ???
+  unlink(local.sun_path);
+  int len = SUN_LEN(&local);
+  int ret = bind(fd, (struct sockaddr*)&local, len);
+  if(ret) {
+    log_printf(ERROR, "unable to bind to '%s': %s", local.sun_path, strerror(errno));
+    return -1;
+  }
+  
+  ret = listen(fd, 4);
+  if(ret) {
+    log_printf(ERROR, "unable to listen on command socket: %s", local.sun_path, strerror(errno));
+    return -1;
+  }
+
+  log_printf(INFO, "now listening on %s for incoming commands", path);
+  
+  return fd;
+}
+
+int send_command(int door_fd, cmd_t* cmd)
+{
+  if(!cmd)
+    return -1;
+  
+  char c;
+  switch(cmd->cmd) {
+  case OPEN: c = 'o'; break;
+  case CLOSE: c = 'c'; break;
+  case TOGGLE: c = 't'; break;
+  case STATUS: c = 's'; break;
+  case RESET: c = 'r'; break;
+  case LOG: return 0;
+  }
+  
+  int ret;
+  do {
+    ret = write(door_fd, &c, 1);
+  } while(!ret || (ret == -1 && errno == EINTR));
+
+  if(ret > 0) {
+    cmd->sent = 1;
+    return 0;
+  }
+
+  return ret;
+}
+
+int send_response(int fd, const char* response)
+{
+  if(!response)
+    return -1;
+
+  int len = strlen(response);
+  int offset = 0;
+  int ret;
+  for(;;) {
+    ret = write(fd, &response[offset], strlen(response));
+    if(ret < 0) {
+      if(errno != EINTR)
+        return ret;
+
+      ret = 0;
+    }
+
+    offset += ret;
+    if(offset+1 >= len)
+      break;
+  }
   do {
+    ret = write(fd, "\n", 1);
+  } while(!ret || (ret == -1 && errno == EINTR));
+
+  if(ret > 0)
+    return 0;
+
+  return ret;
+}
+
+int handle_command(const char* cmd, int fd, cmd_t** cmd_q)
+{
+  if(!cmd_q || !cmd)
+    return -1;
+  
+  cmd_id_t cmd_id;
+  if(!strncmp(cmd, "open", 4))
+    cmd_id = OPEN;
+  else if(!strncmp(cmd, "close", 5))
+    cmd_id = CLOSE;
+  else if(!strncmp(cmd, "toggle", 6))
+    cmd_id = TOGGLE;
+  else if(!strncmp(cmd, "reset", 5))
+    cmd_id = RESET;
+  else if(!strncmp(cmd, "status", 6))
+    cmd_id = STATUS;
+  else if(!strncmp(cmd, "log", 3))
+    cmd_id = LOG;
+  else {
+    log_printf(WARNING, "unknown command '%s'", cmd);
+    return 1;
+  }
+  char* param = strchr(cmd, ' ');
+  if(param) 
+    param++;
+
+  switch(cmd_id) {
+  case OPEN:
+  case CLOSE:
+  case TOGGLE:
+  case STATUS:
+  case RESET: {
+    int ret = cmd_push(cmd_q, fd, cmd_id, param);
+    if(ret)
+      return ret;
+
+    log_printf(NOTICE, "command: %s", cmd); 
+    break;
+  }
+  case LOG: {
+    if(param && param[0])
+      log_printf(NOTICE, "ext msg: %s", param); 
+    else
+      log_printf(DEBUG, "ignoring empty ext log message");
+    break;
+  }
+  }
+  
+  return 0;
+}
+
+int process_cmd(int fd, cmd_t **cmd_q)
+{
+  static char buffer[100];
+  int ret = 0;
+  do { // TODO: replace this whith a actually working readline
+    memset(buffer, 0, 100);
     ret = recv(fd, buffer, sizeof(buffer), 0);
-    if(!ret) return 1;
+    if(!ret)
+      return 1;
+    char* saveptr;
+    char* tok = strtok_r(buffer, "\n\r", &saveptr);
+    do {
+      ret = handle_command(tok, fd, cmd_q);
+      if(ret < 0)
+        return ret;
+    } while(tok = strtok_r(NULL, "\n\r", &saveptr));
   } while (ret == -1 && errno == EINTR);
     
+  log_printf(DEBUG, "processing command from %d", fd);
+
   return 0;
 }
 
-int process_ttyusb(int ttyusb_fd)
+int process_door(int door_fd, cmd_t **cmd_q)
 {
-  log_printf(INFO, "processing data from ttyusb (fd=%d)", ttyusb_fd);
-
-  char* buffer[100];
+  static char buffer[100];
   int ret = 0;
-  do {
-    ret = read(ttyusb_fd, buffer, sizeof(buffer));
+  do { // TODO: replace this whith a actually working readline
+    memset(buffer, 0, 100);
+    ret = read(door_fd, buffer, sizeof(buffer));
+    if(!ret) 
+      return 2;
+    char* saveptr;
+    char* tok = strtok_r(buffer, "\n\r", &saveptr);
+    do {
+      if(!cmd_q || !(*cmd_q))
+        break;
+      ret = send_response((*cmd_q)->fd, tok);
+      if(ret < 0)
+        return ret;
+
+      cmd_pop(cmd_q);
+    } while(tok = strtok_r(NULL, "\n\r", &saveptr));
   } while (ret == -1 && errno == EINTR);
 
+  log_printf(DEBUG, "processing data from door (fd=%d)", door_fd);
+
   return 0;
 }
 
 
-int main_loop(int ttyusb_fd, int cmd_listen_fd)
+int main_loop(int door_fd, int cmd_listen_fd)
 {
   log_printf(INFO, "entering main loop");
 
   fd_set readfds, tmpfds;
   FD_ZERO(&readfds);
-  FD_SET(ttyusb_fd, &readfds);
+  FD_SET(door_fd, &readfds);
   FD_SET(cmd_listen_fd, &readfds);
-  int max_fd = ttyusb_fd > cmd_listen_fd ? ttyusb_fd : cmd_listen_fd;
+  int max_fd = door_fd > cmd_listen_fd ? door_fd : cmd_listen_fd;
+  cmd_t* cmd_q = NULL;
+
+  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;
 
-  signal_init();
   int return_value = 0;
   while(!return_value) {
     memcpy(&tmpfds, &readfds, sizeof(tmpfds));
@@ -83,20 +254,21 @@ int main_loop(int ttyusb_fd, int cmd_listen_fd)
       return_value = -1;
       break;
     }
-    if(!ret)
+    if(!ret || ret == -1)
       continue;
 
-    if(signal_exit) {
+    if(signal_handle()) {
       return_value = 1;
       break;
     }
+    FD_CLR(sig_fd, &tmpfds);
 
-    if(FD_ISSET(ttyusb_fd, &tmpfds)) {
-      return_value = process_ttyusb(ttyusb_fd);
+    if(FD_ISSET(door_fd, &tmpfds)) {
+      return_value = process_door(door_fd, &cmd_q);
       if(return_value)
         break;
 
-      FD_CLR(ttyusb_fd, &tmpfds);
+      FD_CLR(door_fd, &tmpfds);
     }
 
     if(FD_ISSET(cmd_listen_fd, &tmpfds)) {
@@ -106,7 +278,7 @@ int main_loop(int ttyusb_fd, int cmd_listen_fd)
         return_value = -1;
         break;
       }  
-      log_printf(INFO, "new command connection (fd=%d)", new_fd);
+      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;
       FD_CLR(cmd_listen_fd, &tmpfds);
@@ -115,8 +287,9 @@ int main_loop(int ttyusb_fd, int cmd_listen_fd)
     int fd;
     for(fd = 0; fd <= max_fd; fd++) {
       if(FD_ISSET(fd, &tmpfds)) {
-        return_value = process_cmd(fd); 
+        return_value = process_cmd(fd, &cmd_q); 
         if(return_value == 1) {
+          log_printf(DEBUG, "removing closed command connection (fd=%d)", fd);
           close(fd);
           FD_CLR(fd, &readfds);
           return_value = 0;
@@ -125,161 +298,134 @@ int main_loop(int ttyusb_fd, int cmd_listen_fd)
           break;
       }
     }
+
+    if(cmd_q && !cmd_q->sent)
+      send_command(door_fd, cmd_q);
   }
 
+  cmd_clear(&cmd_q);
+  signal_stop();
   return return_value;
 }
 
 int main(int argc, char* argv[])
 {
   log_init();
-  int ret = 0;
 
-/*   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, exitting\n"); */
-/*     } */
-/*     if(ret == -3) { */
-/*       fprintf(stderr, "syntax error: -4 and -6 are mutual exclusive\n\n"); */
-/*     } */
-/*     if(ret == -4) { */
-/*       fprintf(stderr, "syntax error: unknown role name\n\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,door_daemon,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_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 != -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,door_daemon,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_add_target("stdout:5");
+        options_clear(&opt);
+        log_close();
+        exit(ret);
+      }
+      tmp = tmp->next_;
+    }
+  }
   log_printf(NOTICE, "just started...");
-/*   options_parse_post(&opt); */
-
-/*   priv_info_t priv; */
-/*   if(opt.username_) */
-/*     if(priv_init(&priv, opt.username_, opt.groupname_)) { */
-/*       options_clear(&opt); */
-/*       log_close(); */
-/*       exit(-1); */
-/*     } */
-
-  int ttyusb_fd = open("/dev/ttyUSB0", O_RDWR);
-  if(ttyusb_fd < 0) {
-    log_printf(ERROR, "unable to open /dev/ttyUSB0: %s", strerror(errno));
-/*     options_clear(&opt); */
-    log_close();
-    exit(-1);
+  options_parse_post(&opt);
+
+  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));
+    }
   }
 
-  int cmd_listen_fd = socket(AF_UNIX, SOCK_STREAM, 0);
-  struct sockaddr_un local;
-  int len;
-  if(cmd_listen_fd < 0) {
-    log_printf(ERROR, "unable to open socket: %s", strerror(errno));
-    close(ttyusb_fd);
-/*     options_clear(&opt); */
-    log_close();
-    exit(-1);
+  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);
   }
-  local.sun_family = AF_UNIX;
-  strcpy(local.sun_path, "sock");
-  unlink(local.sun_path);
-  len = SUN_LEN(&local);
-  ret = bind(cmd_listen_fd, (struct sockaddr*)&local, len);
-  if(ret) {
-    log_printf(ERROR, "unable to bind to '%s': %s", local.sun_path, strerror(errno));
-    close(cmd_listen_fd);
-    close(ttyusb_fd);
-/*     options_clear(&opt); */
+
+  if(pid_file) {
+    pid_t pid = getpid();
+    fprintf(pid_file, "%d", pid);
+    fclose(pid_file);
+  }
+
+  int door_fd = open(opt.door_dev_, O_RDWR);
+  if(door_fd < 0) {
+    log_printf(ERROR, "unable to open %s: %s", opt.door_dev_, strerror(errno));
+    options_clear(&opt);
     log_close();
     exit(-1);
   }
 
-  ret = listen(cmd_listen_fd, 4);
-  if(ret) {
-    log_printf(ERROR, "unable to listen on command socket: %s", local.sun_path, strerror(errno));
-    close(cmd_listen_fd);
-    close(ttyusb_fd);
-/*     options_clear(&opt); */
+  int cmd_listen_fd = init_command_socket(opt.command_sock_);
+  if(cmd_listen_fd < 0) {
+    close(door_fd);
+    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); */
-/*   } */
-
-  ret = main_loop(ttyusb_fd, cmd_listen_fd);
+
+  ret = main_loop(door_fd, cmd_listen_fd);
 
   close(cmd_listen_fd);
-  close(ttyusb_fd);
-/*   options_clear(&opt); */
+  close(door_fd);
 
   if(!ret)
     log_printf(NOTICE, "normal shutdown");
   else if(ret < 0)
     log_printf(NOTICE, "shutdown after error");
+  else if(ret == 2)
+    log_printf(ERROR, "shutdown after %s read error", opt.door_dev_);
   else
     log_printf(NOTICE, "shutdown after signal");
 
+  options_clear(&opt);
   log_close();
 
   return ret;