suricatactrl: Add "cron" command
[ipfire-2.x.git] / 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 }