From f2e68440d6ebd7e8b0bc0e0bd9562625f295da37 Mon Sep 17 00:00:00 2001
From: Martin Braun <martin.braun@ettus.com>
Date: Thu, 19 Apr 2018 11:53:29 -0700
Subject: log: Allow disabling of fastpath msgs at runtime

- Fixes an issue with compile time disabling as well
- An UHD_LOG_FASTPATH_DISABLE=1 env var will make it that O/U/S/D won't
  be printed
---
 host/lib/utils/log.cpp | 67 ++++++++++++++++++++++++++++++++++++++++----------
 1 file changed, 54 insertions(+), 13 deletions(-)

(limited to 'host/lib')

diff --git a/host/lib/utils/log.cpp b/host/lib/utils/log.cpp
index a29c275a8..e975bccc0 100644
--- a/host/lib/utils/log.cpp
+++ b/host/lib/utils/log.cpp
@@ -184,7 +184,7 @@ public:
                 this->global_level
         );
 #endif
-       //allow override from environment variables
+        //allow override from environment variables
         const char * log_level_env = std::getenv("UHD_LOG_LEVEL");
         if (log_level_env != NULL && log_level_env[0] != '\0') {
             this->global_level =
@@ -211,9 +211,35 @@ public:
         _pop_task = std::make_shared<std::thread>(
             std::thread([this](){this->pop_task();})
         );
-        _pop_fastpath_task = std::make_shared<std::thread>(
-            std::thread([this](){this->pop_fastpath_task();})
-        );
+
+        // Fastpath message consumer
+#ifndef UHD_LOG_FASTPATH_DISABLE
+        //allow override from environment variables
+        const bool enable_fastpath = [](){
+            const char* disable_fastpath_env =
+                std::getenv("UHD_LOG_FASTPATH_DISABLE");
+            if (disable_fastpath_env != NULL
+                    && disable_fastpath_env[0] != '\0') {
+                return false;
+            }
+            return true;
+        }();
+
+        if (enable_fastpath) {
+            _pop_fastpath_task = std::make_shared<std::thread>(
+                std::thread([this](){this->pop_fastpath_task();})
+            );
+        } else {
+            _pop_fastpath_task = std::make_shared<std::thread>(
+                std::thread([this](){this->pop_fastpath_dummy_task();})
+            );
+            _publish_log_msg("Fastpath logging disabled at runtime.");
+        }
+#else
+        {
+            _publish_log_msg("Fastpath logging disabled at compile time.");
+        }
+#endif
     }
 
     ~log_resource(void){
@@ -252,14 +278,14 @@ public:
         _log_queue.push_with_timed_wait(log_info, PUSH_TIMEOUT);
     }
 
+#ifndef UHD_LOG_FASTPATH_DISABLE
     void push_fastpath(const std::string &message)
     {
         // Never wait. If the buffer is full, we just don't see the message.
         // Too bad.
-#ifndef UHD_LOG_FASTPATH_DISABLE
         _fastpath_queue.push_with_haste(message);
-#endif
     }
+#endif
 
     void _handle_log_info(const uhd::log::logging_info& log_info)
     {
@@ -298,22 +324,32 @@ public:
     void pop_fastpath_task()
     {
 #ifndef UHD_LOG_FASTPATH_DISABLE
+        std::string msg;
         while (!_exit) {
-            std::string msg;
             _fastpath_queue.pop_with_wait(msg);
-            {
-                std::cerr << msg << std::flush;
-            }
+            std::cerr << msg << std::flush;
         }
 
         // Exit procedure: Clear the queue
-        std::string msg;
         while (_fastpath_queue.pop_with_haste(msg)) {
             std::cerr << msg << std::flush;
         }
 #endif
     }
 
+    void pop_fastpath_dummy_task()
+    {
+#ifndef UHD_LOG_FASTPATH_DISABLE
+        std::string msg;
+        while (!_exit) {
+            _fastpath_queue.pop_with_wait(msg);
+        }
+
+        // Exit procedure: Clear the queue
+        while (_fastpath_queue.pop_with_haste(msg));
+#endif
+    }
+
     void add_logger(
         const std::string &key,
         uhd::log::log_fn_t logger_fn
@@ -473,12 +509,17 @@ uhd::_log::log::~log(void)
     }
 }
 
+#ifndef UHD_LOG_FASTPATH_DISABLE
 void uhd::_log::log_fastpath(const std::string &msg)
 {
-#ifndef UHD_LOG_FASTPATH_DISABLE
     log_rs().push_fastpath(msg);
-#endif
 }
+#else
+void uhd::_log::log_fastpath(const std::string &)
+{
+    // nop
+}
+#endif
 
 /***********************************************************************
  * Public API calls
-- 
cgit v1.2.3