kernelshark: Add trace_dialog_register_alt_warning()

Add trace_dialog_register_alt_warning() that will allow the tools to
override the warning messages. To hide warnings from poping up,
or to record them elsewhere, this allows to do so.

Signed-off-by: Steven Rostedt <rostedt@goodmis.org>

2 files changed, 22 insertions, 0 deletions

diff --git a/trace-dialog.c b/trace-dialog.c
index c220a52..3f5e1ea 100644
--- a/trace-dialog.c
+++ b/trace-dialog.c
@@ -38,6 +38,8 @@ static GString *statusstr;
 
 static GtkWidget *parent_window;
 
+static void (*alt_warning)(const char *fmt, va_list ap);
+
 void pr_stat(char *fmt, ...)
 {
         GString *str;
@@ -79,11 +81,30 @@ void trace_dialog_register_window(GtkWidget *window)
         parent_window = window;
 }
 
+/**
+ * trace_dialog_register_alt_warning - register an alternate function for warning()
+ * @alt: the function to be called instead of warning.
+ *
+ * Add an alternate warning function to be called instead of a popup.
+ * To go back to the popup, simply call this again with NULL.
+ */
+void trace_dialog_register_alt_warning(void (*alt)(const char *fmt, va_list ap))
+{
+        alt_warning = alt;
+}
+
 void warning(char *fmt, ...)
 {
         GString *str;
         va_list ap;
 
+        if (alt_warning) {
+                va_start(ap, fmt);
+                alt_warning(fmt, ap);
+                va_end(ap);
+                return;
+        }
+
         if (!parent_window) {
                 va_start(ap, fmt);
                 __vwarning(fmt, ap);
diff --git a/trace-gui.h b/trace-gui.h
index a2d261f..bcf8737 100644
--- a/trace-gui.h
+++ b/trace-gui.h
@@ -32,6 +32,7 @@ enum trace_dialog_type {
 GtkWidget *trace_status_bar_new(void);
 
 void trace_dialog_register_window(GtkWidget *window);
+void trace_dialog_register_alt_warning(void (*alt)(const char *fmt, va_list ap));
 
 void trace_show_help(GtkWidget *window, const gchar *link, GError **error);