diff options
Diffstat (limited to 'src')
-rw-r--r-- | src/ipcpd/ipcp.c | 39 | ||||
-rw-r--r-- | src/ipcpd/ipcp.h | 2 | ||||
-rw-r--r-- | src/ipcpd/local/main.c | 6 | ||||
-rw-r--r-- | src/ipcpd/shim-eth-llc/main.c | 4 | ||||
-rw-r--r-- | src/ipcpd/shim-udp/main.c | 6 | ||||
-rw-r--r-- | src/irmd/main.c | 67 | ||||
-rw-r--r-- | src/lib/CMakeLists.txt | 1 | ||||
-rw-r--r-- | src/lib/ipcp.c | 12 | ||||
-rw-r--r-- | src/lib/logs.c | 36 |
9 files changed, 158 insertions, 15 deletions
diff --git a/src/ipcpd/ipcp.c b/src/ipcpd/ipcp.c index 579203c2..28004a92 100644 --- a/src/ipcpd/ipcp.c +++ b/src/ipcpd/ipcp.c @@ -22,13 +22,15 @@ #include <ouroboros/config.h> #include <ouroboros/ipcp.h> -#include <sys/socket.h> -#include <stdlib.h> -#include "ipcp.h" #define OUROBOROS_PREFIX "ipcpd/ipcp" #include <ouroboros/logs.h> +#include <string.h> +#include <sys/socket.h> +#include <stdlib.h> +#include "ipcp.h" + struct ipcp * ipcp_instance_create() { struct ipcp * i = malloc(sizeof *i); @@ -45,15 +47,42 @@ struct ipcp * ipcp_instance_create() return i; } -int ipcp_arg_check(int argc, char * argv[]) +int ipcp_parse_arg(int argc, char * argv[]) { - if (argc != 2) + char * log_file; + size_t len = 0; + + if (!(argc == 3 || argc == 2)) return -1; /* argument 1: api of irmd */ if (atoi(argv[1]) == 0) return -1; + if (argv[2] == NULL) + return 0; + + len += strlen(INSTALL_PREFIX); + len += strlen(LOG_DIR); + len += strlen(argv[2]); + + log_file = malloc(len + 1); + if (log_file == NULL) { + LOG_ERR("Failed to malloc"); + return -1; + } + + strcpy(log_file, INSTALL_PREFIX); + strcat(log_file, LOG_DIR); + strcat(log_file, argv[2]); + log_file[len] = '\0'; + + if (set_logfile(log_file)) + LOG_ERR("Cannot open %s, falling back to stdout for logs.", + log_file); + + free(log_file); + return 0; } diff --git a/src/ipcpd/ipcp.h b/src/ipcpd/ipcp.h index 27c3cf8e..bbf1d1f7 100644 --- a/src/ipcpd/ipcp.h +++ b/src/ipcpd/ipcp.h @@ -50,6 +50,6 @@ struct ipcp { struct ipcp * ipcp_instance_create(); void * ipcp_main_loop(void * o); void * ipcp_sdu_loop(void * o); -int ipcp_arg_check(int argc, char * argv[]); +int ipcp_parse_arg(int argc, char * argv[]); #endif diff --git a/src/ipcpd/local/main.c b/src/ipcpd/local/main.c index 17327fce..ffe69dfb 100644 --- a/src/ipcpd/local/main.c +++ b/src/ipcpd/local/main.c @@ -573,7 +573,7 @@ static struct ipcp * ipcp_local_create() #ifndef MAKE_CHECK -int main (int argc, char * argv[]) +int main(int argc, char * argv[]) { struct sigaction sig_act; sigset_t sigset; @@ -583,8 +583,8 @@ int main (int argc, char * argv[]) sigaddset(&sigset, SIGHUP); sigaddset(&sigset, SIGPIPE); - if (ipcp_arg_check(argc, argv)) { - LOG_ERR("Wrong arguments."); + if (ipcp_parse_arg(argc, argv)) { + LOG_ERR("Failed to parse arguments."); exit(1); } diff --git a/src/ipcpd/shim-eth-llc/main.c b/src/ipcpd/shim-eth-llc/main.c index 8f65819f..5e355bd7 100644 --- a/src/ipcpd/shim-eth-llc/main.c +++ b/src/ipcpd/shim-eth-llc/main.c @@ -1245,8 +1245,8 @@ int main(int argc, char * argv[]) sigaddset(&sigset, SIGHUP); sigaddset(&sigset, SIGPIPE); - if (ipcp_arg_check(argc, argv)) { - LOG_ERR("Wrong arguments."); + if (ipcp_parse_arg(argc, argv)) { + LOG_ERR("Failed to parse arguments."); exit(1); } diff --git a/src/ipcpd/shim-udp/main.c b/src/ipcpd/shim-udp/main.c index 874168ef..92204970 100644 --- a/src/ipcpd/shim-udp/main.c +++ b/src/ipcpd/shim-udp/main.c @@ -1557,7 +1557,7 @@ static struct ipcp * ipcp_udp_create() #ifndef MAKE_CHECK -int main (int argc, char * argv[]) +int main(int argc, char * argv[]) { struct sigaction sig_act; sigset_t sigset; @@ -1567,8 +1567,8 @@ int main (int argc, char * argv[]) sigaddset(&sigset, SIGHUP); sigaddset(&sigset, SIGPIPE); - if (ipcp_arg_check(argc, argv)) { - LOG_ERR("Wrong arguments."); + if (ipcp_parse_arg(argc, argv)) { + LOG_ERR("Failed to parse arguments."); exit(1); } diff --git a/src/irmd/main.c b/src/irmd/main.c index 8ac22118..26251858 100644 --- a/src/irmd/main.c +++ b/src/irmd/main.c @@ -50,6 +50,7 @@ #include <limits.h> #include <pthread.h> #include <sys/stat.h> +#include <dirent.h> #include <sys/wait.h> #define IRMD_CLEANUP_TIMER ((IRMD_FLOW_TIMEOUT / 20) * MILLION) /* ns */ @@ -1782,17 +1783,81 @@ static struct irm * irm_create() return instance; } -int main() +static void usage() +{ + LOG_ERR("Usage: irmd \n\n" + " [--stdout (Print to stdout instead of logs)]\n"); +} + +int main(int argc, char ** argv) { struct sigaction sig_act; int t = 0; + char * log_file = INSTALL_PREFIX LOG_DIR "irmd.log"; + DIR * log_dir; + struct dirent * ent; + char * point; + char * log_path; + size_t len = 0; + bool use_stdout = false; + if (geteuid() != 0) { LOG_ERR("IPC Resource Manager must be run as root."); exit(EXIT_FAILURE); } + argc--; + argv++; + while (argc > 0) { + if (strcmp(*argv, "--stdout") == 0) { + use_stdout = true; + argc--; + argv++; + } else { + usage(); + exit(EXIT_FAILURE); + } + } + + + if (!use_stdout && + (log_dir = opendir(INSTALL_PREFIX LOG_DIR)) != NULL) { + while ((ent = readdir(log_dir)) != NULL) { + point = strrchr(ent->d_name,'.'); + if (point == NULL || + strcmp(point, ".log") != 0) + continue; + + len += strlen(INSTALL_PREFIX); + len += strlen(LOG_DIR); + len += strlen(ent->d_name); + + log_path = malloc(len + 1); + if (log_path == NULL) { + LOG_ERR("Failed to malloc"); + exit(EXIT_FAILURE); + } + + strcpy(log_path, INSTALL_PREFIX); + strcat(log_path, LOG_DIR); + strcat(log_path, ent->d_name); + + unlink(log_path); + + free(log_path); + len = 0; + } + closedir(log_dir); + } + + if (!use_stdout) + if (set_logfile(log_file)) + LOG_ERR("Cannot open %s, falling back to " + "stdout for logs.", + log_file); + /* init sig_act */ memset(&sig_act, 0, sizeof sig_act); diff --git a/src/lib/CMakeLists.txt b/src/lib/CMakeLists.txt index 57f44f15..9c45269c 100644 --- a/src/lib/CMakeLists.txt +++ b/src/lib/CMakeLists.txt @@ -31,6 +31,7 @@ set(SOURCE_FILES ipcp.c irm.c list.c + logs.c nsm.c shm_ap_rbuff.c shm_du_map.c diff --git a/src/lib/ipcp.c b/src/lib/ipcp.c index e4a82b60..5d46940d 100644 --- a/src/lib/ipcp.c +++ b/src/lib/ipcp.c @@ -109,6 +109,7 @@ pid_t ipcp_create(enum ipcp_type ipcp_type) char * ipcp_dir = "/sbin/"; char * full_name = NULL; char * exec_name = NULL; + char * log_file = NULL; sprintf(irmd_api, "%u", getpid()); @@ -149,8 +150,19 @@ pid_t ipcp_create(enum ipcp_type ipcp_type) strcat(full_name, exec_name); full_name[len] = '\0'; + if (logfile != NULL) { + log_file = malloc(20); + if (log_file == NULL) { + LOG_ERR("Failed to malloc."); + exit(EXIT_FAILURE); + } + sprintf(log_file, "ipcpd-%u.log", getpid()); + } + + /* log_file to be placed at the end */ char * argv[] = {full_name, irmd_api, + log_file, 0}; char * envp[] = {0}; diff --git a/src/lib/logs.c b/src/lib/logs.c new file mode 100644 index 00000000..d86073e4 --- /dev/null +++ b/src/lib/logs.c @@ -0,0 +1,36 @@ +/* + * Ouroboros - Copyright (C) 2016 + * + * Logging facilities + * + * Sander Vrijders <[email protected]> + * + * This program is free software; you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation; either version 2 of the License, or + * (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program; if not, write to the Free Software + * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA. + */ + +#define OUROBOROS_PREFIX "logging" + +#include <ouroboros/logs.h> + +FILE * logfile = NULL; + +int set_logfile(char * filename) +{ + logfile = fopen(filename, "w"); + if (logfile == NULL) + return -1; + + return 0; +} |