discover/grub2: Allow to separate the --id argument using a space char

[petitboot] / lib / log / log.c

diff --git a/lib/log/log.c b/lib/log/log.c
index e006fd0b1495c34867648ae58ff047113de751a6..7f14232443fd1538dc8454711b7a69f50b19c950 100644 (file)
--- a/lib/log/log.c
+++ b/lib/log/log.c
@@ -1,23 +1,112 @@
 
+#include <assert.h>
 #include <stdarg.h>
+#include <stdlib.h>
+#include <time.h>
 
 #include "log.h"
 
 static FILE *logf;
+static bool debug;
+
+static void __log_timestamp(void)
+{
+       char hms[20] = {'\0'};
+       time_t t;
+
+       if (!logf)
+               return;
+
+       t = time(NULL);
+       strftime(hms, sizeof(hms), "%T", localtime(&t));
+       fprintf(logf, "[%s] ", hms);
+}
+
+static void __log(const char *fmt, va_list ap)
+{
+       if (!logf)
+               return;
+
+       vfprintf(logf, fmt, ap);
+       fflush(logf);
+}
 
 void pb_log(const char *fmt, ...)
 {
        va_list ap;
-       FILE *stream;
+       va_start(ap, fmt);
+       __log_timestamp();
+       __log(fmt, ap);
+       va_end(ap);
+}
+
+void _pb_log_fn(const char *func, const char *fmt, ...)
+{
+       va_list ap;
+       pb_log("%s: ", func);
+       va_start(ap, fmt);
+       __log(fmt, ap);
+       va_end(ap);
+}
+
+void pb_debug(const char *fmt, ...)
+{
+       va_list ap;
+       if (!debug)
+               return;
+       va_start(ap, fmt);
+       __log_timestamp();
+       __log(fmt, ap);
+       va_end(ap);
+}
 
-       stream = logf ? logf : stdout;
+void _pb_debug_fn(const char *func, const char *fmt, ...)
+{
+       va_list ap;
+       if (!debug)
+               return;
+       pb_log("%s: ", func);
+       va_start(ap, fmt);
+       __log(fmt, ap);
+       va_end(ap);
+}
 
+void _pb_debug_fl(const char *func, int line, const char *fmt, ...)
+{
+       va_list ap;
+       if (!debug)
+               return;
+       pb_log("%s:%d: ", func, line);
        va_start(ap, fmt);
-       vfprintf(stream, fmt, ap);
+       __log(fmt, ap);
        va_end(ap);
 }
 
-void pb_log_set_stream(FILE *stream)
+void __pb_log_init(FILE *fp, bool _debug)
+{
+       if (logf)
+               fflush(logf);
+       logf = fp;
+       debug = _debug;
+}
+
+void pb_log_set_debug(bool _debug)
+{
+       debug = _debug;
+}
+
+bool pb_log_get_debug(void)
+{
+       return debug;
+}
+
+FILE *pb_log_get_stream(void)
 {
-       logf = stream;
+       static FILE *null_stream;
+       if (!logf) {
+               if (!null_stream)
+                       null_stream = fopen("/dev/null", "a");
+               return null_stream;
+       }
+       return logf;
 }