src/misc-progs/suricatactrl.c

   1 /* This file is part of the IPFire Firewall.
   2  *
   3  * This program is distributed under the terms of the GNU General Public
   4  * Licence.  See the file COPYING for details.
   5  *
   6  */
   7
   8 #include <stdlib.h>
   9 #include <stdio.h>
  10 #include <string.h>
  11 #include <unistd.h>
  12 #include <sys/types.h>
  13 #include <fcntl.h>
  14 #include "setuid.h"
  15
  16 int main(int argc, char *argv[]) {
  17
  18         if (!(initsetuid()))
  19                 exit(1);
  20
  21         if (argc < 2) {
  22                 fprintf(stderr, "\nNo argument given.\n\nsuricatactrl (start|stop|restart|reload)\n\n");
  23                 exit(1);
  24         }
  25
  26         if (strcmp(argv[1], "start") == 0) {
  27                 safe_system("/etc/rc.d/init.d/suricata start");
  28         } else if (strcmp(argv[1], "stop") == 0) {
  29                 safe_system("/etc/rc.d/init.d/suricata stop");
  30         } else if (strcmp(argv[1], "restart") == 0) {
  31                 safe_system("/etc/rc.d/init.d/suricata restart");
  32         } else if (strcmp(argv[1], "reload") == 0) {
  33                 safe_system("/etc/rc.d/init.d/suricata reload");
  34         } else if (strcmp(argv[1], "fix-rules-dir") == 0) {
  35                 safe_system("chown -R nobody:nobody /var/lib/suricata");
  36         } else if (strcmp(argv[1], "cron") == 0) {
  37                         safe_system("rm /etc/fcron.*/suricata >/dev/null 2>&1");
  38                 if (strcmp(argv[2], "off") == 0) {
  39                         return(1);
  40                 } else if (strcmp(argv[2], "daily") == 0){
  41                         safe_system("ln -s /usr/local/bin/update-ids-ruleset /etc/fcron.daily/suricata");
  42                 } else if (strcmp(argv[2], "weekly") == 0){
  43                         safe_system("ln -s /usr/local/bin/update-ids-ruleset /etc/fcron.weekly/suricata");
  44                 } else{
  45                         printf("invalid parameter(s)\n");
  46                 return(1);
  47                 }
  48         } else {
  49                 fprintf(stderr, "\nBad argument given.\n\nsuricatactrl (start|stop|restart|reload)\n\n");
  50                 exit(1);
  51         }
  52
  53         return 0;
  54 }