summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorDennis Friis <peavey@inspircd.org>2009-05-27 20:36:18 +0200
committerDennis Friis <peavey@inspircd.org>2009-05-27 20:36:18 +0200
commit21c2dceddb86302b4255f2a581b5c2092ef30806 (patch)
tree728e24999d2e4f693e5fbea513295e1e8ba26a65
parent6404931f2eba6c18066984ac282b01a21c6e3702 (diff)
downloadtmwa-21c2dceddb86302b4255f2a581b5c2092ef30806.tar.gz
tmwa-21c2dceddb86302b4255f2a581b5c2092ef30806.tar.bz2
tmwa-21c2dceddb86302b4255f2a581b5c2092ef30806.tar.xz
tmwa-21c2dceddb86302b4255f2a581b5c2092ef30806.zip
I forgot to add the source for the monitor eons ago when creating this repo.
-rw-r--r--eathena-monitor.c272
1 files changed, 272 insertions, 0 deletions
diff --git a/eathena-monitor.c b/eathena-monitor.c
new file mode 100644
index 0000000..13048d4
--- /dev/null
+++ b/eathena-monitor.c
@@ -0,0 +1,272 @@
+/*
+Name: eAthena processes monitor
+Author: Bartosz Waszak <waszi@evil.org.pl>
+License: GPL
+Compilation:
+gcc -o eathena-monitor eathena-monitor.c
+*/
+
+#include <unistd.h>
+#include <stdio.h>
+#include <string.h>
+#include <stdlib.h>
+#include <sys/types.h>
+#include <sys/vfs.h>
+#include <dirent.h>
+#include <errno.h>
+#include <time.h>
+#include <fcntl.h>
+#include <sys/wait.h>
+#include <signal.h>
+
+#define HOME getenv("HOME")
+#define LOGIN_SERVER "login-server"
+#define MAP_SERVER "map-server"
+#define CHAR_SERVER "char-server"
+#define CONFIG "conf/eathena-monitor.conf"
+#define LOGFILE "log/eathena-monitor.log"
+#define PATH_MAX 4096
+
+#define IS_BLANK(ptr) \
+ (((*(ptr)) == ' ') || ((*(ptr)) == '\b') || \
+ ((*(ptr)) == '\n') || ((*(ptr)) == '\r'))
+
+#define SKIP_BLANK(ptr) \
+ { while (((*(ptr)) == ' ') || ((*(ptr)) == '\b') || \
+ ((*(ptr)) == '\n') || ((*(ptr)) == '\r')) ptr++; }
+
+#define GOTO_EQL(ptr) \
+ { while (((*(ptr)) != '\0') && ((*(ptr)) != '=') && \
+ ((*(ptr)) != '\n') && ((*(ptr)) != '\r')) ptr++; }
+
+#define GOTO_EOL(ptr) \
+ { while (((*(ptr)) != '\0') && \
+ ((*(ptr)) != '\n') && ((*(ptr)) != '\r')) ptr++; }
+
+char *workdir;
+char *login_server;
+char *map_server;
+char *char_server;
+char *config;
+char *logfile;
+unsigned int interval = 5;
+unsigned int pid_login, pid_map, pid_char;
+char use_login = 1;
+
+void parse_option (char *name, char *value) {
+ if (!strcasecmp(name, "login_server")) {
+ login_server = strdup(value);
+ } else if (!strcasecmp(name, "map_server")) {
+ map_server = strdup(value);
+ } else if (!strcasecmp(name, "char_server")) {
+ char_server = strdup(value);
+ } else if (!strcasecmp(name, "workdir")) {
+ workdir = strdup(value);
+ } else if (!strcasecmp(name, "logfile")) {
+ logfile = strdup(value);
+ } else if (!strcasecmp(name, "interval")) {
+ interval = atoi(strdup(value));
+ } else {
+ printf("WARNING: ingnoring invalid options '%s'\n", name);
+ }
+}
+
+int read_config(char *filename) {
+ FILE *input;
+ char *str, *base;
+ char string[1000];
+ char *name;
+ char *value;
+ int errors = 0;
+
+ if (!(input = fopen(filename,"r")) && !(input = fopen (config, "r"))) {
+ fprintf (stderr, "ERROR: Config file doesn't exist (%s and %s), using builtin defaults\n", filename, config);
+ return -1;
+ }
+
+ while (1) {
+ if (fgets (&string[0], sizeof (string) - 1, input) == NULL)
+ break;
+ str = &string[0];
+ SKIP_BLANK(str);
+ string[sizeof (string) - 1] = '\0';
+ if (*str == '#')
+ continue;
+ if (*str == '\0')
+ continue;
+ name = str;
+
+ GOTO_EQL (str);
+
+ if (*str != '=') {
+ continue;
+ }
+
+ *str++ = '\0';
+ SKIP_BLANK(str);
+ value = str;
+ GOTO_EOL(str);
+ *str = '\0';
+ parse_option(name, value);
+ }
+
+ fclose (input);
+ return(0);
+}
+
+void start_process(char *exec) {
+ pid_t pid;
+ pid = fork();
+ if (pid == 0) {
+ if (!fork()) {
+ execl(exec,exec,NULL);
+ exit(0);
+ }
+ exit(0);
+ }
+ wait(0);
+}
+
+void stop_process(int sig) {
+ system("killall map-server");
+ system("killall login-server");
+ system("killall char-server");
+ exit(0);
+}
+
+int main(int argc, char *argv[]) {
+ DIR *procdir;
+ FILE *log;
+
+ int fd;
+
+ char pathbuf[PATH_MAX];
+ char link[PATH_MAX];
+ char timestamp[256];
+
+ struct tm *tmp;
+ struct dirent *procdirp;
+ struct statfs sfs;
+
+ unsigned int proc_login, proc_map, proc_char;
+
+ time_t t;
+ size_t l_size;
+
+ if ( statfs("/proc", &sfs) == -1 ) {
+ fprintf(stderr,"ERROR: /proc filesystem is unaccessible\n");
+ return(255);
+ }
+
+ signal(SIGTERM, stop_process);
+ signal(SIGINT, stop_process);
+
+ workdir = (char *) malloc(sizeof(char) * 1024);
+ login_server = (char *) malloc(sizeof(char) * 1024);
+ char_server = (char *) malloc(sizeof(char) * 1024);
+ map_server = (char *) malloc(sizeof(char) * 1024);
+ logfile = (char *) malloc(sizeof(char) * 1024);
+ config = (char *) malloc(sizeof(char) * 1024);
+
+ sprintf(workdir,"%s/tmwserver",HOME);
+ sprintf(login_server,"%s/%s", workdir, LOGIN_SERVER);
+ sprintf(map_server,"%s/%s", workdir, MAP_SERVER);
+ sprintf(char_server,"%s/%s", workdir, CHAR_SERVER);
+ sprintf(logfile,"%s/%s", workdir,LOGFILE);
+ sprintf(config,"%s/%s", workdir, CONFIG);
+
+ read_config(argv[1]);
+
+ chdir(workdir);
+
+ if (strlen(login_server) == 0) use_login = 0;
+
+ printf ("Starting:\n");
+ printf ("* interval: %d s\n", interval);
+ printf ("* workdir: %s\n", workdir);
+ if (use_login)
+ printf ("* login_server: %s\n", login_server);
+ else
+ printf ("* login_server: (none)\n");
+ printf ("* map_server: %s\n", map_server);
+ printf ("* char_server: %s\n", char_server);
+
+ if (fork()) {
+ exit(0);
+ }
+
+ if ((fd = open("/dev/null", O_RDONLY)) != 0) {
+ dup2(fd, 0);
+ close(fd);
+ }
+
+ if ((fd = open("/dev/null", O_WRONLY)) != 1) {
+ dup2(fd, 1);
+ close(fd);
+ }
+ dup2(1, 2);
+
+ while (1) {
+ if (use_login) proc_login = 0;
+ proc_map = 0;
+ proc_char = 0;
+
+ if ((procdir = opendir("/proc")) == NULL) {
+ fprintf(stderr,"ERROR: Cannot open /proc filesystem\n");
+ return(255);
+ }
+
+ while ((procdirp = readdir(procdir)) != NULL) {
+ if (strtok(procdirp->d_name, "0123456789") == NULL) {
+ sprintf(pathbuf, "%s%s%s", "/proc/", procdirp->d_name, "/exe");
+ l_size = readlink(pathbuf, link, PATH_MAX);
+
+ if (l_size != -1) {
+ link[l_size] = '\0';
+ if (use_login && !strcmp(link, login_server)) {
+ proc_login = 1;
+ pid_login = (unsigned int) procdirp->d_name;
+ }
+
+ if (!strcmp(link, char_server)) {
+ proc_char = 1;
+ pid_char = (unsigned int) procdirp->d_name;
+ }
+
+ if (!strcmp(link, map_server)) {
+ proc_map = 1;
+ pid_map = (unsigned int) procdirp->d_name;
+ }
+ }
+ }
+ }
+ closedir(procdir);
+
+ if (!(log = fopen (logfile,"a"))) {
+ log = fopen("/tmp/monitor.log","a");
+ }
+
+ t = time(NULL);
+ tmp = localtime(&t);
+ strftime(timestamp, sizeof(timestamp), "%F %X", tmp);
+
+ if (use_login && proc_login == 0) {
+ fprintf (log,"[%d][%s] NOTICE: Login server is dead - restarting\n", getpid(), timestamp);
+ start_process(login_server);
+ sleep(2);
+ }
+ if (proc_char == 0) {
+ fprintf (log,"[%d][%s] NOTICE: Char server is dead - restarting\n", getpid(), timestamp);
+ start_process(char_server);
+ sleep(2);
+ }
+ if (proc_map == 0) {
+ fprintf (log,"[%d][%s] NOTICE: Map server is dead - restarting\n", getpid(), timestamp);
+ start_process(map_server);
+ sleep(2);
+ }
+
+ fclose(log);
+ sleep(interval);
+ }
+}