|
2 | 2 | using System.Collections.Generic; |
3 | 3 | using System.IO; |
4 | 4 | using System.Linq; |
| 5 | +using System.Reflection.Metadata; |
5 | 6 | using System.Reflection.PortableExecutable; |
6 | 7 | using System.Runtime.CompilerServices; |
7 | 8 | using System.Text; |
@@ -46,18 +47,38 @@ public void LambdaCapturing() |
46 | 47 | TestGeneratePdb(); |
47 | 48 | } |
48 | 49 |
|
| 50 | + [Test] |
| 51 | + public void CustomPdbId() |
| 52 | + { |
| 53 | + // Generate a PDB for an assembly using a randomly-generated ID, then validate that the PDB uses the specified ID |
| 54 | + (string peFileName, string pdbFileName) = CompileTestCase(nameof(CustomPdbId)); |
| 55 | + |
| 56 | + var moduleDefinition = new PEFile(peFileName); |
| 57 | + var resolver = new UniversalAssemblyResolver(peFileName, false, moduleDefinition.Metadata.DetectTargetFrameworkId(), null, PEStreamOptions.PrefetchEntireImage); |
| 58 | + var decompiler = new CSharpDecompiler(moduleDefinition, resolver, new DecompilerSettings()); |
| 59 | + var expectedPdbId = new BlobContentId(Guid.NewGuid(), (uint)Random.Shared.Next()); |
| 60 | + |
| 61 | + using (FileStream pdbStream = File.Open(Path.Combine(TestCasePath, nameof(CustomPdbId) + ".pdb"), FileMode.OpenOrCreate, FileAccess.ReadWrite)) |
| 62 | + { |
| 63 | + pdbStream.SetLength(0); |
| 64 | + PortablePdbWriter.WritePdb(moduleDefinition, decompiler, new DecompilerSettings(), pdbStream, noLogo: true, pdbId: expectedPdbId); |
| 65 | + |
| 66 | + pdbStream.Position = 0; |
| 67 | + var metadataReader = MetadataReaderProvider.FromPortablePdbStream(pdbStream).GetMetadataReader(); |
| 68 | + var generatedPdbId = new BlobContentId(metadataReader.DebugMetadataHeader.Id); |
| 69 | + |
| 70 | + Assert.AreEqual(expectedPdbId.Guid, generatedPdbId.Guid); |
| 71 | + Assert.AreEqual(expectedPdbId.Stamp, generatedPdbId.Stamp); |
| 72 | + } |
| 73 | + } |
| 74 | + |
49 | 75 | private void TestGeneratePdb([CallerMemberName] string testName = null) |
50 | 76 | { |
51 | 77 | const PdbToXmlOptions options = PdbToXmlOptions.IncludeEmbeddedSources | PdbToXmlOptions.ThrowOnError | PdbToXmlOptions.IncludeTokens | PdbToXmlOptions.ResolveTokens | PdbToXmlOptions.IncludeMethodSpans; |
52 | 78 |
|
53 | 79 | string xmlFile = Path.Combine(TestCasePath, testName + ".xml"); |
54 | | - string xmlContent = File.ReadAllText(xmlFile); |
55 | | - XDocument document = XDocument.Parse(xmlContent); |
56 | | - var files = document.Descendants("file").ToDictionary(f => f.Attribute("name").Value, f => f.Value); |
57 | | - Tester.CompileCSharpWithPdb(Path.Combine(TestCasePath, testName + ".expected"), files); |
| 80 | + (string peFileName, string pdbFileName) = CompileTestCase(testName); |
58 | 81 |
|
59 | | - string peFileName = Path.Combine(TestCasePath, testName + ".expected.dll"); |
60 | | - string pdbFileName = Path.Combine(TestCasePath, testName + ".expected.pdb"); |
61 | 82 | var moduleDefinition = new PEFile(peFileName); |
62 | 83 | var resolver = new UniversalAssemblyResolver(peFileName, false, moduleDefinition.Metadata.DetectTargetFrameworkId(), null, PEStreamOptions.PrefetchEntireImage); |
63 | 84 | var decompiler = new CSharpDecompiler(moduleDefinition, resolver, new DecompilerSettings()); |
@@ -87,6 +108,20 @@ private void TestGeneratePdb([CallerMemberName] string testName = null) |
87 | 108 | Assert.AreEqual(Normalize(expectedFileName), Normalize(generatedFileName)); |
88 | 109 | } |
89 | 110 |
|
| 111 | + private (string peFileName, string pdbFileName) CompileTestCase(string testName) |
| 112 | + { |
| 113 | + string xmlFile = Path.Combine(TestCasePath, testName + ".xml"); |
| 114 | + string xmlContent = File.ReadAllText(xmlFile); |
| 115 | + XDocument document = XDocument.Parse(xmlContent); |
| 116 | + var files = document.Descendants("file").ToDictionary(f => f.Attribute("name").Value, f => f.Value); |
| 117 | + Tester.CompileCSharpWithPdb(Path.Combine(TestCasePath, testName + ".expected"), files); |
| 118 | + |
| 119 | + string peFileName = Path.Combine(TestCasePath, testName + ".expected.dll"); |
| 120 | + string pdbFileName = Path.Combine(TestCasePath, testName + ".expected.pdb"); |
| 121 | + |
| 122 | + return (peFileName, pdbFileName); |
| 123 | + } |
| 124 | + |
90 | 125 | private void ProcessXmlFile(string fileName) |
91 | 126 | { |
92 | 127 | var document = XDocument.Load(fileName); |
|
0 commit comments