From 6fcbb97c0b7f082c4934d7765c6b63222f317ef2 Mon Sep 17 00:00:00 2001
From: Micael Karlberg <bmk@erlang.org>
Date: Mon, 5 Nov 2018 14:37:39 +0100
Subject: [socket-nif] Make it possible to send (nif) debug to file

Make it possible to open a file and send debug printouts to
(instead of stdout) for debug printouts from the nif-code.

OTP-14831
---
 erts/emulator/nifs/common/socket_dbg.c | 24 +++++++++++++++++++++++-
 erts/emulator/nifs/common/socket_dbg.h |  8 ++++++--
 2 files changed, 29 insertions(+), 3 deletions(-)

(limited to 'erts')

diff --git a/erts/emulator/nifs/common/socket_dbg.c b/erts/emulator/nifs/common/socket_dbg.c
index dd11fbca9b..fe9135e5a0 100644
--- a/erts/emulator/nifs/common/socket_dbg.c
+++ b/erts/emulator/nifs/common/socket_dbg.c
@@ -36,10 +36,32 @@
 #define TNAME(__T__)       enif_thread_name( __T__ )
 #define TSNAME()           TNAME(TSELF())
 
+static FILE* dbgout = NULL;
+
 static int realtime(struct timespec* tsP);
 static int timespec2str(char *buf, unsigned int len, struct timespec *ts);
 
 
+extern
+void esock_dbg_init(char* filename)
+{
+  if (filename != NULL) {
+    if (strcmp(filename, ESOCK_DBGOUT_DEFAULT) == 0) {
+      dbgout = stdout;
+    } else if (strcmp(filename, ESOCK_DBGOUT_UNIQUE) == 0) {
+      char template[] = "/tmp/esock-dbg-XXXXXX";
+      dbgout = fdopen(mkstemp(template), "w+");
+    } else {
+      dbgout = fopen(filename, "w+");
+    }
+  } else {
+    char template[] = "/tmp/esock-dbg-XXXXXX";
+    dbgout = fdopen(mkstemp(template), "w+");
+  }
+}
+
+
+
 /*
  * Print a debug format string *with* both a timestamp and the
  * the name of the *current* thread.
@@ -70,7 +92,7 @@ void esock_dbg_printf( const char* prefix, const char* format, ... )
 
     if (res > 0) {
       va_start (args, format);
-      enif_vfprintf (stdout, f, args);
+      enif_vfprintf (dbgout, f, args);
       va_end (args);
       fflush(stdout);
     }
diff --git a/erts/emulator/nifs/common/socket_dbg.h b/erts/emulator/nifs/common/socket_dbg.h
index ad0fcdada9..47739b46da 100644
--- a/erts/emulator/nifs/common/socket_dbg.h
+++ b/erts/emulator/nifs/common/socket_dbg.h
@@ -27,6 +27,10 @@
 #ifndef SOCKET_DBG_H__
 #define SOCKET_DBG_H__
 
+/* Used when calling the init function */
+#define ESOCK_DBGOUT_DEFAULT "stdout"
+#define ESOCK_DBGOUT_UNIQUE  "unique"
+
 
 /* Used in debug printouts */
 #ifdef __WIN32__
@@ -45,7 +49,7 @@ typedef unsigned long long llu_t;
     }
 
 
-extern
-void esock_dbg_printf( const char* prefix, const char* format, ... );
+extern void esock_dbg_init(char* filename);
+extern void esock_dbg_printf( const char* prefix, const char* format, ... );
 
 #endif // SOCKET_DBG_H__
-- 
cgit v1.2.3