Add new 'symbolicate' commands to dotnet-stack

src/Tools/dotnet-stack/Program.cs

@@ -16,6 +16,7 @@ public static Task<int> Main(string[] args)
             var parser = new CommandLineBuilder()
                 .AddCommand(ReportCommandHandler.ReportCommand())
                 .AddCommand(ProcessStatusCommandHandler.ProcessStatusCommand("Lists the dotnet processes that traces can be collected"))
+                .AddCommand(SymbolicateHandler.SymbolicateCommand())
                 .UseDefaults()
                 .Build();
 

src/Tools/dotnet-stack/Symbolicate.cs

@@ -0,0 +1,334 @@
+// Licensed to the .NET Foundation under one or more agreements.
+// The .NET Foundation licenses this file to you under the MIT license.
+// See the LICENSE file in the project root for more information.
+
+using Microsoft.Tools.Common;
+using System;
+using System.Collections.Generic;
+using System.CommandLine;
+using System.CommandLine.Binding;
+using System.CommandLine.IO;
+using System.IO;
+using System.Linq;
+using System.Reflection.Metadata;
+using System.Reflection.Metadata.Ecma335;
+using System.Reflection.PortableExecutable;
+using System.Text.RegularExpressions;
+
+namespace Microsoft.Diagnostics.Tools.Stack
+{
+    internal static class SymbolicateHandler
+    {
+        private static readonly Regex s_regex = new Regex(@" at (?<type>[\w+\.?]+)\.(?<method>\w+)\((?<params>.*)\) in (?<filename>[\w+\.?]+):token (?<token>0x\d+)\+(?<offset>0x\d+)", RegexOptions.Compiled);
+        private static readonly Dictionary<string, string> s_assemblyFilePathDictionary = new Dictionary<string, string>();
+        private static readonly Dictionary<string, MetadataReader> s_metadataReaderDictionary = new Dictionary<string, MetadataReader>();
+
+        delegate void SymbolicateDelegate(IConsole console, FileInfo inputPath, DirectoryInfo[] searchDir, FileInfo output, bool stdout);
+
+        /// <summary>
+        /// Get the line number from the Method Token and IL Offset in a stacktrace
+        /// </summary>
+        /// <param name="console"></param>
+        /// <param name="inputPath">Path to the stacktrace text file</param>
+        /// <param name="searchDir">Path of multiple directories with assembly and pdb where the exception occurred</param>
+        /// <param name="output">Output directly to a file</param>
+        /// <param name="stdout">Output directly to a console</param>
+        /// <returns></returns>
+        private static void Symbolicate(IConsole console, FileInfo inputPath, DirectoryInfo[] searchDir, FileInfo output, bool stdout)
+        {
+            try
+            {
+                if (output == null)
+                {
+                    output = new FileInfo(inputPath.FullName + ".symbolicated");
+                }
+
+                SetAssemblyFilePathDictionary(console, searchDir);
+
+                CreateSymbolicateFile(console, inputPath.FullName, output.FullName, stdout);
+            }
+            catch (Exception e)
+            {
+                console.Error.WriteLine(e.Message);
+            }
+        }
+
+        private static void SetAssemblyFilePathDictionary(IConsole console, DirectoryInfo[] searchDir)
+        {
+            try
+            {
+                List<string> searchPaths = new List<string>
+                {
+                    Directory.GetCurrentDirectory()
+                };
+                foreach (var path in searchDir)
+                {
+                    searchPaths.Add(path.FullName);
+                }
+
+                List<string> peFiles = GrabFiles(searchPaths, "*.dll");
+                if (peFiles.Count == 0)
+                {
+                    throw new FileNotFoundException("Assembly file not found\n");
+                }
+                peFiles = peFiles.Distinct().ToList();
+                peFiles.Sort();
+
+                List<string> pdbFiles = GrabFiles(searchPaths, "*.pdb");
+                if (pdbFiles.Count == 0)
+                {
+                    throw new FileNotFoundException("PDB file not found\n");
+                }
+                pdbFiles = pdbFiles.Distinct().ToList();
+                pdbFiles.Sort();
+
+                int pdbCnt = 0;
+                for (int peCnt = 0; peCnt < peFiles.Count; peCnt++)
+                {
+                    if (peFiles[peCnt].Contains(".ni.dll"))
+                    {
+                        continue;
+                    }
+                    int compare = string.Compare(Path.GetFileNameWithoutExtension(peFiles[peCnt]), Path.GetFileNameWithoutExtension(pdbFiles[pdbCnt]), StringComparison.OrdinalIgnoreCase);
+                    if (compare == 0)
+                    {
+                        s_assemblyFilePathDictionary.Add(Path.GetFileNameWithoutExtension(peFiles[peCnt]), peFiles[peCnt]);
+                    }
+                    else if (compare > 0)
+                    {
+                        pdbCnt++;
+                        peCnt--;
+                    }
+                    if (pdbCnt == pdbFiles.Count) break;
+                }
+            }
+            catch (Exception e)
+            {
+                console.Error.WriteLine(e.Message);
+            }
+        }
+
+        private static List<string> GrabFiles(List<string> paths, string searchPattern)
+        {
+            try
+            {
+                List<string> files = new List<string>();
+                foreach (var assemDir in paths)
+                {
+                    if (Directory.Exists(assemDir))
+                    {
+                        files.AddRange(Directory.GetFiles(assemDir, searchPattern, SearchOption.AllDirectories));
+                    }
+                }
+                return files;
+            }
+            catch
+            {
+                return new List<string>();
+            }
+        }
+
+        private static void CreateSymbolicateFile(IConsole console, string inputPath, string outputPath, bool isStdout)
+        {
+            try
+            {
+                using StreamWriter fileStreamWriter = new StreamWriter(new FileStream(outputPath, FileMode.Create, FileAccess.Write));
+                using StreamReader fileStreamReader = new StreamReader(new FileStream(inputPath, FileMode.Open, FileAccess.Read));
+                while (!fileStreamReader.EndOfStream)
+                {
+                    string ret = TrySymbolicateLine(fileStreamReader.ReadLine());
+                    fileStreamWriter?.WriteLine(ret);
+                    if (isStdout) console.Out.WriteLine(ret);
+                }
+                console.Out.WriteLine($"\nOutput: {outputPath}\n");
+            }
+            catch (Exception e)
+            {
+                console.Error.WriteLine(e.Message);
+            }
+        }
+
+        internal sealed class StackTraceInfo
+        {
+            public string Type;
+            public string Method;
+            public string Param;
+            public string Filename;
+            public string Assembly;
+            public string Pdb;
+            public string Token;
+            public string Offset;
+        }
+
+        private static string TrySymbolicateLine(string line)
+        {
+            Match match = s_regex.Match(line);
+            if (!match.Success)
+            {
+                return line;
+            }
+
+            StackTraceInfo stInfo = new StackTraceInfo()
+            {
+                Type = match.Groups["type"].Value,
+                Method = match.Groups["method"].Value,
+                Param = match.Groups["params"].Value,
+                Assembly = match.Groups["filename"].Value,
+                Token = match.Groups["token"].Value,
+                Offset = match.Groups["offset"].Value
+            };
+            if (stInfo.Assembly.Contains(".ni.dll"))
+            {
+                stInfo.Filename = stInfo.Assembly.Replace(".ni.dll", "");
+            }
+            else
+            {
+                stInfo.Filename = stInfo.Assembly.Replace(".dll", "");
+            }
+            stInfo.Pdb = stInfo.Filename + ".pdb";
+
+            return GetLineFromMetadata(TryGetMetadataReader(stInfo.Filename), line, stInfo);
+        }
+
+        private static MetadataReader TryGetMetadataReader(string assemblyName)
+        {
+            MetadataReader reader = null;
+            try
+            {
+                if (s_assemblyFilePathDictionary.TryGetValue(assemblyName, out string filePath))
+                {
+                    if (s_metadataReaderDictionary.TryGetValue(filePath, out reader))
+                    {
+                        return reader;
+                    }
+                    s_metadataReaderDictionary.Add(filePath, SetMetadataReader(filePath));
+                    return s_metadataReaderDictionary[filePath];
+                }
+                return reader;
+            }
+            catch
+            {
+                return reader;
+            }
+        }
+
+        private static MetadataReader SetMetadataReader(string filePath)
+        {
+            MetadataReader reader = null;
+            try
+            {
+                MetadataReaderProvider provider = null;
+                static Stream streamProvider(string sp) => new FileStream(sp, FileMode.Open, FileAccess.Read);
+                using Stream stream = new FileStream(filePath, FileMode.Open, FileAccess.Read);
+                if (stream != null)
+                {
+                    if (filePath.Contains(".dll"))
+                    {
+                        using PEReader peReader = new PEReader(stream);
+                        if (!peReader.TryOpenAssociatedPortablePdb(filePath, streamProvider, out provider, out string pdbPath))
+                        {
+                            return reader;
+                        }
+                    }
+                    /*else if (filePath.Contains(".pdb"))
+                    {
+                        provider = MetadataReaderProvider.FromPortablePdbStream(stream);
+                    }*/
+                    else
+                    {
+                        return reader;
+                    }
+                }
+                return provider?.GetMetadataReader();
+            }
+            catch
+            {
+                return reader;
+            }
+        }
+
+        private static string GetLineFromMetadata(MetadataReader reader, string line, StackTraceInfo stInfo)
+        {
+            try
+            {
+                if (reader != null)
+                {
+                    Handle handle = MetadataTokens.Handle(Convert.ToInt32(stInfo.Token, 16));
+                    if (handle.Kind == HandleKind.MethodDefinition)
+                    {
+                        MethodDebugInformationHandle methodDebugHandle = ((MethodDefinitionHandle)handle).ToDebugInformationHandle();
+                        MethodDebugInformation methodInfo = reader.GetMethodDebugInformation(methodDebugHandle);
+                        if (!methodInfo.SequencePointsBlob.IsNil)
+                        {
+                            SequencePointCollection sequencePoints = methodInfo.GetSequencePoints();
+                            SequencePoint? bestPointSoFar = null;
+                            foreach (SequencePoint point in sequencePoints)
+                            {
+                                if (point.Offset > Convert.ToInt64(stInfo.Offset, 16))
+                                    break;
+
+                                if (point.StartLine != SequencePoint.HiddenLine)
+                                    bestPointSoFar = point;
+                            }
+
+                            if (bestPointSoFar.HasValue)
+                            {
+                                string sourceFile = reader.GetString(reader.GetDocument(bestPointSoFar.Value.Document).Name);
+                                int sourceLine = bestPointSoFar.Value.StartLine;
+                                string pattern = stInfo.Assembly + @":token " + stInfo.Token + @"\+" + stInfo.Offset;
+                                string replacement = sourceFile + @":line " + sourceLine;
+                                return Regex.Replace(line, pattern, replacement);
+                            }
+                        }
+                    }
+                }
+                return line;
+            }
+            catch
+            {
+                return line;
+            }
+        }
+
+        public static Command SymbolicateCommand() =>
+            new Command(
+                name: "symbolicate", description: "Get the line number from the Method Token and IL Offset in a stacktrace")
+            {
+                // Handler
+                HandlerDescriptor.FromDelegate((SymbolicateDelegate)Symbolicate).GetCommandHandler(),
+                // Arguments and Options
+                InputFileArgument(),
+                SearchDirectoryOption(),
+                OutputFileOption(),
+                StandardOutOption()
+            };
+
+        public static Argument<FileInfo> InputFileArgument() =>
+            new Argument<FileInfo>(name: "input-path")
+            {
+                Description = "Path to the stacktrace text file",
+                Arity = ArgumentArity.ExactlyOne
+            }.ExistingOnly();
+
+        public static Option<DirectoryInfo[]> SearchDirectoryOption() =>
+            new Option<DirectoryInfo[]>(new[] { "-d", "--search-dir" }, "Path of multiple directories with assembly and pdb")
+            {
+                Argument = new Argument<DirectoryInfo[]>(name: "directory1 directory2 ...", getDefaultValue: () => new DirectoryInfo(Directory.GetCurrentDirectory()).GetDirectories())
+                {
+                    Arity = ArgumentArity.ZeroOrMore
+                }.ExistingOnly()
+            };
+
+        public static Option<FileInfo> OutputFileOption() =>
+            new Option<FileInfo>(new[] { "-o", "--output" }, "Output directly to a file (Default: <input-path>.symbolicated)")
+            {
+                Argument = new Argument<FileInfo>(name: "output-path")
+                {
+                    Arity = ArgumentArity.ZeroOrOne
+                }
+            };
+
+        public static Option<bool> StandardOutOption() =>
+            new Option<bool>(new[] { "-c", "--stdout" }, getDefaultValue: () => false, "Output directly to a console");
+    }
+}