Changeset 3628 in MondoRescue for branches/3.3/mondo/src/lib/mr_msg.c

Timestamp:

Dec 21, 2016, 3:18:07 PM (7 years ago)

Author:

Bruno Cornec

Message:

Change mr_msg_init interface definition to include argc and argv, which allows to detect the presence of the -K option and setup the log level accordingly as early as possible

File:

• branches/3.3/mondo/src/lib/mr_msg.c (modified) (3 diffs)

branches/3.3/mondo/src/lib/mr_msg.c

@@ -13 +13 @@
 #define _GNU_SOURCE
 #endif
+
 #include <stdio.h>
 #include <stdarg.h>
 #include <stdlib.h>
+#include <unistd.h>
+#include <string.h>
 
 static int mr_loglevel = 0;
@@ -48 +51 @@
 
 /* Initialization function for messages */
-void mr_msg_init(const char *logfile, int loglevel) {
+void mr_msg_init(const char *logfile, int loglevel, int argc, char *argv[]) {
+
+    int i=1;
+
     if (asprintf(&mr_logfile, "%s", logfile) == -1) {
         fprintf(stderr,"Unable to alloc memory\n");
@@ -67 +73 @@
             fprintf(stderr,"Logging desactivated\n");
             mr_msg_close();
+        }
+    }
+    while (i < argc) {
+        if (strcmp(argv[i],"-K") == 0) {
+            i++;
+            if (argv[i]) {
+                loglevel = atoi(argv[i]);
+            } else {
+                fprintf(stderr,"-K option requires a loglevel\n");
+                mr_msg_close();
+                exit(-1);
+            }
         }
     }