mirror of
https://github.com/vincentmli/bpfire.git
synced 2026-04-14 13:02:58 +02:00
81 lines
2.4 KiB
C
81 lines
2.4 KiB
C
/* This file is part of the IPFire Firewall.
|
|
*
|
|
* This program is distributed under the terms of the GNU General Public
|
|
* Licence. See the file COPYING for details.
|
|
*
|
|
*/
|
|
|
|
#include <stdlib.h>
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
#include <unistd.h>
|
|
#include <sys/types.h>
|
|
#include <fcntl.h>
|
|
#include "setuid.h"
|
|
|
|
#define BUFFER_SIZE 1024
|
|
|
|
int main(int argc, char *argv[]) {
|
|
char command[BUFFER_SIZE];
|
|
|
|
if (!(initsetuid()))
|
|
exit(1);
|
|
|
|
if (argc < 3) {
|
|
fprintf(stderr, "\nMissing arguments.\n\naddonctrl addon (start|stop|restart|reload|enable|disable)\n\n");
|
|
exit(1);
|
|
}
|
|
|
|
const char* name = argv[1];
|
|
|
|
if (strlen(name) > 32) {
|
|
fprintf(stderr, "\nString to large.\n\naddonctrl addon (start|stop|restart|reload|enable|disable)\n\n");
|
|
exit(1);
|
|
}
|
|
|
|
// Check if the input argument is valid
|
|
if (!is_valid_argument_alnum(name)) {
|
|
fprintf(stderr, "Invalid add-on name: %s\n", name);
|
|
exit(2);
|
|
}
|
|
|
|
sprintf(command, "/opt/pakfire/db/installed/meta-%s", name);
|
|
FILE *fp = fopen(command,"r");
|
|
if ( fp ) {
|
|
fclose(fp);
|
|
} else {
|
|
fprintf(stderr, "\nAddon '%s' not found.\n\naddonctrl addon (start|stop|restart|reload|status|enable|disable)\n\n", name);
|
|
exit(1);
|
|
}
|
|
|
|
if (strcmp(argv[2], "start") == 0) {
|
|
snprintf(command, BUFFER_SIZE - 1, "/etc/rc.d/init.d/%s start", name);
|
|
safe_system(command);
|
|
} else if (strcmp(argv[2], "stop") == 0) {
|
|
snprintf(command, BUFFER_SIZE - 1, "/etc/rc.d/init.d/%s stop", name);
|
|
safe_system(command);
|
|
} else if (strcmp(argv[2], "restart") == 0) {
|
|
snprintf(command, BUFFER_SIZE - 1, "/etc/rc.d/init.d/%s restart", name);
|
|
safe_system(command);
|
|
} else if (strcmp(argv[2], "reload") == 0) {
|
|
snprintf(command, BUFFER_SIZE - 1, "/etc/rc.d/init.d/%s reload", name);
|
|
safe_system(command);
|
|
} else if (strcmp(argv[2], "status") == 0) {
|
|
snprintf(command, BUFFER_SIZE - 1, "/etc/rc.d/init.d/%s status", name);
|
|
safe_system(command);
|
|
} else if (strcmp(argv[2], "enable") == 0) {
|
|
snprintf(command, BUFFER_SIZE - 1, "mv -f /etc/rc.d/rc3.d/off/S??%s /etc/rc.d/rc3.d" , name);
|
|
safe_system(command);
|
|
} else if (strcmp(argv[2], "disable") == 0) {
|
|
snprintf(command, BUFFER_SIZE - 1, "mkdir -p /etc/rc.d/rc3.d/off");
|
|
safe_system(command);
|
|
snprintf(command, BUFFER_SIZE - 1, "mv -f /etc/rc.d/rc3.d/S??%s /etc/rc.d/rc3.d/off" , name);
|
|
safe_system(command);
|
|
} else {
|
|
fprintf(stderr, "\nBad argument given.\n\naddonctrl addon (start|stop|restart|reload|enable|disable)\n\n");
|
|
exit(1);
|
|
}
|
|
|
|
return 0;
|
|
}
|