1 files changed, 67 insertions, 0 deletions
diff --git a/tools/lldb-server/LLDBServerUtilities.cpp b/tools/lldb-server/LLDBServerUtilities.cpp
new file mode 100644
index 0000000..8df4875
--- /dev/null
+++ b/tools/lldb-server/LLDBServerUtilities.cpp
@@ -0,0 +1,67 @@
+//===-- LLDBServerUtilities.cpp ---------------------------------*- C++ -*-===//
+//
+//                     The LLVM Compiler Infrastructure
+//
+// This file is distributed under the University of Illinois Open Source
+// License. See LICENSE.TXT for details.
+//
+//===----------------------------------------------------------------------===//
+
+#include "LLDBServerUtilities.h"
+
+#include "lldb/Core/Log.h"
+#include "lldb/Core/StreamFile.h"
+#include "lldb/Core/StreamString.h"
+#include "lldb/Interpreter/Args.h"
+
+#include "llvm/ADT/SmallVector.h"
+#include "llvm/ADT/StringRef.h"
+
+using namespace lldb;
+using namespace lldb_private::lldb_server;
+using namespace llvm;
+
+bool
+LLDBServerUtilities::SetupLogging(const std::string& log_file,
+	                              const StringRef& log_channels,
+	                              uint32_t log_options)
+{
+    lldb::StreamSP log_stream_sp;
+    if (log_file.empty())
+    {
+        log_stream_sp.reset(new StreamFile(stdout, false));
+    }
+    else
+    {
+        uint32_t options = File::eOpenOptionWrite | File::eOpenOptionCanCreate |
+                           File::eOpenOptionCloseOnExec | File::eOpenOptionAppend;
+        if (!(log_options & LLDB_LOG_OPTION_APPEND))
+            options |= File::eOpenOptionTruncate;
+
+        log_stream_sp.reset(new StreamFile(log_file.c_str(), options));
+    }
+
+    SmallVector<StringRef, 32> channel_array;
+    log_channels.split(channel_array, ":");
+    for (auto channel_with_categories : channel_array)
+    {
+        StreamString error_stream;
+        Args channel_then_categories(channel_with_categories);
+        std::string channel(channel_then_categories.GetArgumentAtIndex(0));
+        channel_then_categories.Shift (); // Shift off the channel
+
+        bool success = Log::EnableLogChannel(log_stream_sp,
+                                             log_options,
+                                             channel.c_str(),
+                                             channel_then_categories.GetConstArgumentVector(),
+                                             error_stream);
+        if (!success)
+        {
+            fprintf(stderr, "Unable to open log file '%s' for channel \"%s\"\n",
+                    log_file.c_str(),
+                    channel_with_categories.str().c_str());
+            return false;
+        }
+    }
+    return true;
+}