Set TMPDIR for tracee for diagnosticport discovery

Author: mdh1418

src/native/eventpipe/ds-ipc.c

@@ -226,7 +226,12 @@ ipc_log_poll_handles (dn_vector_t *ipc_poll_handles)
 		if (ipc_poll_handle.ipc) {
 			if (!(ds_ipc_to_string (ipc_poll_handle.ipc, buffer, (uint32_t)ARRAY_SIZE (buffer)) > 0))
 				buffer [0] = '\0';
-			DS_LOG_DEBUG_2 ("\tSERVER IpcPollHandle[%d] = %s", connection_id, buffer);
+			ep_char8_t uds_path [DS_IPC_MAX_TO_STRING_LEN];
+			if (ds_ipc_get_uds_path (ipc_poll_handle.ipc, uds_path, (uint32_t)ARRAY_SIZE (uds_path)) > 0) {
+				DS_LOG_DEBUG_1 ("ipc_log_poll_handles ipc uds path '%s'", uds_path);
+			} else {
+				DS_LOG_DEBUG_2 ("\tSERVER IpcPollHandle[%d] = %s", connection_id, buffer);
+			}
 		} else {
 			if (!(ds_ipc_stream_to_string (ipc_poll_handle.stream, buffer, (uint32_t)ARRAY_SIZE (buffer))))
 				buffer [0] = '\0';

src/tests/tracing/eventpipe/userevents/userevents.cs

@@ -107,10 +107,14 @@ public static int TestEntryPoint()
             traceeStartInfo.Environment["DOTNET_LogToFile"] = "1";
             traceeStartInfo.Environment["DOTNET_LogFile"] = logFilePath;
             traceeStartInfo.Environment["DOTNET_LogWithPid"] = "1";
+            // Currently record-trace only searches /tmp/ for diagnostic ports https://github.com/microsoft/one-collect/issues/183
+            string diagnosticPortDir = "/tmp/";
+            traceeStartInfo.Environment["TMPDIR"] = diagnosticPortDir;
 
             Console.WriteLine($"Starting tracee process: {traceeStartInfo.FileName} {traceeStartInfo.Arguments}");
             using Process traceeProcess = Process.Start(traceeStartInfo);
-            Console.WriteLine($"Tracee process started with PID: {traceeProcess.Id}");
+            int traceePid = traceeProcess.Id;
+            Console.WriteLine($"Tracee process started with PID: {traceePid}");
             traceeProcess.OutputDataReceived += (_, args) =>
             {
                 if (!string.IsNullOrEmpty(args.Data))
@@ -136,6 +140,8 @@ public static int TestEntryPoint()
             }
             traceeProcess.WaitForExit(); // flush async output
 
+            CleanupTraceeDiagnosticPorts(diagnosticPortDir, traceePid);
+
             if (!recordTraceProcess.HasExited)
             {
                 // Until record-trace supports duration, the only way to stop it is to send SIGINT (ctrl+c)
@@ -171,6 +177,23 @@ public static int TestEntryPoint()
             return 100;
         }
 
+        private static void CleanupTraceeDiagnosticPorts(string diagnosticPortDir, int traceePid)
+        {
+            try
+            {
+                string[] udsFiles = Directory.GetFiles(diagnosticPortDir, $"dotnet-diagnostic-{traceePid}-*-socket");
+                foreach (string udsFile in udsFiles)
+                {
+                    Console.WriteLine($"Deleting tracee diagnostic port UDS file: {udsFile}");
+                    File.Delete(udsFile);
+                }
+            }
+            catch (Exception ex)
+            {
+                Console.Error.WriteLine($"Failed to cleanup tracee diagnostic ports: {ex}");
+            }
+        }
+
         private static bool ValidateTraceeEvents(string traceFilePath)
         {
             using EventPipeEventSource source = new EventPipeEventSource(traceFilePath);

src/tests/tracing/eventpipe/userevents/usereventstracee.cs

@@ -13,6 +13,7 @@ public class UserEventsTracee
 
         public static void Run()
         {
+            Console.WriteLine($"TMPDIR={Environment.GetEnvironmentVariable("TMPDIR")}");
             long startTimestamp = Stopwatch.GetTimestamp();
             long targetTicks = Stopwatch.Frequency * 10; // 10s