[dotnet-trace] Add collect-linux verb

src/Tools/dotnet-trace/CommandLine/Commands/CollectLinuxCommand.cs

@@ -1,15 +1,20 @@
 // Licensed to the .NET Foundation under one or more agreements.
 // The .NET Foundation licenses this file to you under the MIT license.
 using System;
+using System.Collections.Generic;
 using System.CommandLine;
+using System.Diagnostics;
 using System.IO;
+using System.Runtime.InteropServices;
 using System.Threading.Tasks;
 using Microsoft.Internal.Common.Utils;
 
 namespace Microsoft.Diagnostics.Tools.Trace
 {
-    internal static class CollectLinuxCommandHandler
+    internal static partial class CollectLinuxCommandHandler
     {
+        private static int s_recordStatus;
+
         internal sealed record CollectLinuxArgs(
             string[] Providers,
             string ClrEventLevel,
@@ -30,7 +35,7 @@ private static int CollectLinux(CollectLinuxArgs args)
             if (!OperatingSystem.IsLinux())
             {
                 Console.Error.WriteLine("The collect-linux command is only supported on Linux.");
-                return (int)ReturnCode.PlatformNotSupportedError;
+                return (int)ReturnCode.ArgumentError;
             }
 
             return RunRecordTrace(args);
@@ -76,13 +81,105 @@ public static Command CollectLinuxCommand()
 
         private static int RunRecordTrace(CollectLinuxArgs args)
         {
-            return (int)ReturnCode.Ok;
+            s_recordStatus = 0;
+
+            ConsoleCancelEventHandler handler = (sender, e) =>
+            {
+                e.Cancel = true;
+                s_recordStatus = 1;
+            };
+            Console.CancelKeyPress += handler;
+
+            IEnumerable<string> recordTraceArgList = BuildRecordTraceArgs(args, out string scriptPath);
+
+            string options = string.Join(' ', recordTraceArgList);
+            byte[] command = Encoding.UTF8.GetBytes(options);
+            int rc;
+            try
+            {
+                rc = RecordTrace(command, (UIntPtr)command.Length, OutputHandler);
+            }
+            finally
+            {
+                Console.CancelKeyPress -= handler;
+                if (!string.IsNullOrEmpty(scriptPath))
+                {
+                    try {
+                        if (File.Exists(scriptPath))
+                        {
+                            File.Delete(scriptPath);
+                        }
+                    } catch { }
+                }
+            }
+
+            return rc;
+        }
+
+        private static List<string> BuildRecordTraceArgs(CollectLinuxArgs args, out string scriptPath)
+        {
+            Console.WriteLine($"{args.ProcessId}");
+            List<string> recordTraceArgs = new();
+
+            recordTraceArgs.Add("--on-cpu");
+
+            return recordTraceArgs;
+        }
+
+        private static int OutputHandler(uint type, IntPtr data, UIntPtr dataLen)
+        {
+            OutputType ot = (OutputType)type;
+            if (ot != OutputType.Progress)
+            {
+                int len = checked((int)dataLen);
+                if (len > 0)
+                {
+                    byte[] buffer = new byte[len];
+                    Marshal.Copy(data, buffer, 0, len);
+                    string text = Encoding.UTF8.GetString(buffer);
+                    switch (ot)
+                    {
+                    case OutputType.Normal:
+                    case OutputType.Live:
+                        Console.Out.WriteLine(text);
+                        break;
+                    case OutputType.Error:
+                        Console.Error.WriteLine(text);
+                        break;
+                    default:
+                        Console.Error.WriteLine($"[{ot}] {text}");
+                        break;
+                    }
+                }
+            }
+
+            return s_recordStatus;
         }
 
         private static readonly Option<string> PerfEventsOption =
             new("--perf-events")
             {
                 Description = @"Comma-separated list of kernel perf events (e.g. syscalls:sys_enter_execve,sched:sched_switch)."
             };
+
+        private enum OutputType : uint
+        {
+            Normal = 0,
+            Live = 1,
+            Error = 2,
+            Progress = 3,
+        }
+
+        [UnmanagedFunctionPointer(CallingConvention.Cdecl)]
+        private delegate int recordTraceCallback(
+            [In] uint type,
+            [In] IntPtr data,
+            [In] UIntPtr dataLen);
+
+        [LibraryImport("recordtrace")]
+        private static partial int RecordTrace(
+            byte[] command,
+            UIntPtr commandLen,
+            recordTraceCallback callback);
     }
 }

src/Tools/dotnet-trace/dotnet-trace.csproj

@@ -14,6 +14,7 @@
   <ItemGroup>
     <PackageReference Include="System.CommandLine" Version="$(SystemCommandLineVersion)" />
     <PackageReference Include="Microsoft.Diagnostics.Tracing.TraceEvent" Version="$(MicrosoftDiagnosticsTracingTraceEventVersion)" GeneratePathProperty="true" />
+    <PackageReference Include="Microsoft.OneCollect.RecordTrace" Version="0.1.31522" PrivateAssets="All" GeneratePathProperty="true" />
   </ItemGroup>
 
   <ItemGroup>
@@ -29,4 +30,18 @@
     <InternalsVisibleTo Include="DotnetTrace.UnitTests" />
   </ItemGroup>
 
+  <Target Name="ResolveRecordTraceNative" BeforeTargets="CopyFilesToOutputDirectory">
+    <PropertyGroup>
+      <_RecordTraceFromPackage>$(PkgMicrosoft_OneCollect_RecordTrace)/runtimes/$(RuntimeIdentifier)/native/librecordtrace.so</_RecordTraceFromPackage>
+    </PropertyGroup>
+
+    <ItemGroup>
+      <_RecordTraceResolved Include="$(_RecordTraceLocal)" Condition="'$(RuntimeIdentifier)' != '' AND Exists('$(_RecordTraceLocal)')" />
+      <_RecordTraceResolved Include="$(_RecordTraceFromPackage)" Condition="'@(_RecordTraceResolved)' == '' AND '$(RuntimeIdentifier)' != '' AND Exists('$(_RecordTraceFromPackage)')" />
+    </ItemGroup>
+
+    <ItemGroup>
+      <None Include="@(_RecordTraceResolved)" CopyToOutputDirectory="PreserveNewest" Condition="'@(_RecordTraceResolved)' != ''" />
+    </ItemGroup>
+  </Target>
 </Project>