| public EnqueueImageSymbol ( Reko.Core.ImageSymbol sym, bool isEntryPoint ) : void | ||
| sym | Reko.Core.ImageSymbol | |
| isEntryPoint | bool | |
| return | void | |
protected Program RewriteFile(string relativePath, Address addrBase)
{
sc = new ServiceContainer();
var config = new FakeDecompilerConfiguration();
var eventListener = new FakeDecompilerEventListener();
sc.AddService<IConfigurationService>(config);
sc.AddService<DecompilerHost>(new FakeDecompilerHost());
sc.AddService<DecompilerEventListener>(eventListener);
sc.AddService<IFileSystemService>(new FileSystemServiceImpl());
ILoader ldr = new Loader(sc);
var program = ldr.AssembleExecutable(
FileUnitTester.MapTestPath(relativePath),
new X86TextAssembler(sc, new X86ArchitectureReal()),
addrBase);
program.Platform = new DefaultPlatform(sc, program.Architecture);
var ep = new ImageSymbol(program.SegmentMap.BaseAddress);
var project = new Project { Programs = { program } };
var scan = new Scanner(
program,
new ImportResolver(project, program, eventListener),
sc);
scan.EnqueueImageSymbol(ep, true);
scan.ScanImage();
var importResolver = new ImportResolver(project, program, eventListener);
var dfa = new DataFlowAnalysis(program, importResolver, eventListener);
dfa.AnalyzeProgram();
return program;
}
private void BuildTest(Address addrBase, IPlatform platform , Action<X86Assembler> asmProg)
{
var sc = new ServiceContainer();
var eventListener = new FakeDecompilerEventListener();
sc.AddService<DecompilerEventListener>(eventListener);
sc.AddService<DecompilerHost>(new FakeDecompilerHost());
sc.AddService<IFileSystemService>(new FileSystemServiceImpl());
var entryPoints = new List<ImageSymbol>();
var asm = new X86Assembler(sc, platform, addrBase, entryPoints);
asmProg(asm);
program = asm.GetImage();
var project = new Project { Programs = { program } };
scanner = new Scanner(
program,
new ImportResolver(project, program, eventListener),
sc);
scanner.EnqueueImageSymbol(new ImageSymbol(addrBase), true);
scanner.ScanImage();
}
protected void RunHexTest(string hexFile, string outputFile)
{
var svc = new ServiceContainer();
var cfg = new FakeDecompilerConfiguration();
var eventListener = new FakeDecompilerEventListener();
svc.AddService<IConfigurationService>(cfg);
svc.AddService<DecompilerEventListener>(eventListener);
svc.AddService<DecompilerHost>(new FakeDecompilerHost());
ILoader ldr = new Loader(svc);
var imgLoader = new DchexLoader(FileUnitTester.MapTestPath( hexFile), svc, null);
var program = imgLoader.Load(null);
var project = new Project { Programs = { program } };
var ep = new ImageSymbol(program.ImageMap.BaseAddress);
var importResolver = new ImportResolver(project, program, eventListener);
var scan = new Scanner(program, importResolver, svc);
scan.EnqueueImageSymbol(ep, true);
scan.ScanImage();
var dfa = new DataFlowAnalysis(program, null, eventListener);
dfa.AnalyzeProgram();
RunTest(program, outputFile);
}
private void RunFileTestx86_32(string relativePath, string outputFile)
{
Program program;
var sc = new ServiceContainer();
var fsSvc = new FileSystemServiceImpl();
var el = new FakeDecompilerEventListener();
sc.AddService<IFileSystemService>(fsSvc);
sc.AddService<DecompilerEventListener>(el);
var arch = new X86ArchitectureFlat32();
var asm = new X86TextAssembler(sc, arch);
using (var rdr = new StreamReader(FileUnitTester.MapTestPath(relativePath)))
{
var platform = new DefaultPlatform(sc, arch);
asm.Platform = platform;
program = asm.Assemble(Address.Ptr32(0x10000000), rdr);
}
var scanner = new Scanner(program, null, sc);
scanner.EnqueueImageSymbol(new ImageSymbol(program.ImageMap.BaseAddress), true);
scanner.ScanImage();
using (var fut = new FileUnitTester(outputFile))
{
foreach (var proc in program.Procedures.Values)
{
proc.Write(false, fut.TextWriter);
}
fut.AssertFilesEqual();
}
}
private Program RewriteProgram()
{
var eventListener = new FakeDecompilerEventListener();
var importResolver = MockRepository.GenerateStub<IImportResolver>();
importResolver.Replay();
var scan = new Scanner(
program,
importResolver,
sc);
foreach (ImageSymbol ep in program.EntryPoints.Values)
{
scan.EnqueueImageSymbol(ep, true);
}
scan.ScanImage();
var dfa = new DataFlowAnalysis(program, importResolver, eventListener);
dfa.AnalyzeProgram();
return program;
}
private void DoRewriteCore()
{
var cfgSvc = MockRepository.GenerateStub<IConfigurationService>();
var env = MockRepository.GenerateStub<OperatingEnvironment>();
var tlSvc = MockRepository.GenerateStub<ITypeLibraryLoaderService>();
var eventListener = new FakeDecompilerEventListener();
cfgSvc.Stub(c => c.GetEnvironment("ms-dos")).Return(env);
cfgSvc.Replay();
env.Stub(e => e.TypeLibraries).Return(new List<ITypeLibraryElement>());
env.Stub(e => e.CharacteristicsLibraries).Return(new List<ITypeLibraryElement>());
env.Replay();
tlSvc.Replay();
sc.AddService<DecompilerHost>(new FakeDecompilerHost());
sc.AddService<DecompilerEventListener>(eventListener);
sc.AddService<IConfigurationService>(cfgSvc);
sc.AddService<ITypeLibraryLoaderService>(tlSvc);
Project project = LoadProject();
project.Programs.Add(this.program);
scanner = new Scanner(
this.program,
new ImportResolver(project, this.program, eventListener),
sc);
ImageSymbol ep = new ImageSymbol(baseAddress);
scanner.EnqueueImageSymbol(ep, true);
var program = project.Programs[0];
foreach (Procedure_v1 sp in program.User.Procedures.Values)
{
scanner.EnqueueUserProcedure(sp);
}
scanner.ScanImage();
}
private void RunTest(string sourceFile, string outputFile)
{
Program program;
var asm = new X86TextAssembler(services, new X86ArchitectureFlat32());
using (StreamReader rdr = new StreamReader(FileUnitTester.MapTestPath(sourceFile)))
{
program = asm.Assemble(Address.Ptr32(0x10000000), rdr);
program.Platform = win32;
}
foreach (var item in asm.ImportReferences)
{
program.ImportReferences.Add(item.Key, item.Value);
}
var project = new Project { Programs = { program } };
Scanner scan = new Scanner(
program,
new ImportResolver(project, program, new FakeDecompilerEventListener()),
services);
foreach (var ep in asm.EntryPoints)
{
scan.EnqueueImageSymbol(ep, true);
}
scan.ScanImage();
using (FileUnitTester fut = new FileUnitTester(outputFile))
{
foreach (Procedure proc in program.Procedures.Values)
{
proc.Write(true, fut.TextWriter);
fut.TextWriter.WriteLine();
}
fut.AssertFilesEqual();
}
}
private Program AssembleFile(string sourceFile, Address addr)
{
var ldr = new Loader(new ServiceContainer());
var arch = new X86ArchitectureReal();
Program program = ldr.AssembleExecutable(
FileUnitTester.MapTestPath(sourceFile),
new X86TextAssembler(sc, arch),
addr);
var project = new Project { Programs = { program } };
var scan = new Scanner(
program,
new ImportResolver(project, program, null), null);
foreach (ImageSymbol ep in program.EntryPoints.Values)
{
scan.EnqueueImageSymbol(ep, true);
}
scan.ScanImage();
return program;
}
private static void Rewrite(Program program, Assembler asm, string configFile)
{
var fakeDiagnosticsService = new FakeDiagnosticsService();
var fakeConfigService = new FakeDecompilerConfiguration();
var eventListener = new FakeDecompilerEventListener();
var sc = new ServiceContainer();
sc.AddService(typeof(IDiagnosticsService), fakeDiagnosticsService);
sc.AddService(typeof(IConfigurationService), fakeConfigService);
sc.AddService<DecompilerEventListener>(eventListener);
sc.AddService<DecompilerHost>(new FakeDecompilerHost());
var loader = new Loader(sc);
var project = string.IsNullOrEmpty(configFile)
? new Project()
: new ProjectLoader(sc, loader).LoadProject(FileUnitTester.MapTestPath(configFile));
var scan = new Scanner(
program,
new ImportResolver(project, program, eventListener),
sc);
scan.EnqueueImageSymbol(new ImageSymbol(asm.StartAddress), true);
foreach (var f in project.Programs)
{
foreach (var sp in f.User.Procedures.Values)
{
scan.EnqueueUserProcedure(sp);
}
}
scan.ScanImage();
}
public void Scanner_CallGraphTree()
{
program = new Program();
var addr = Address.SegPtr(0xC00, 0);
var m = new X86Assembler(sc, new DefaultPlatform(sc, new X86ArchitectureReal()), addr, new List<ImageSymbol>());
m.i86();
m.Proc("main");
m.Call("baz");
m.Ret();
m.Endp("main");
m.Proc("foo");
m.Ret();
m.Endp("foo");
m.Proc("bar");
m.Ret();
m.Endp("bar");
m.Proc("baz");
m.Call("foo");
m.Call("bar");
m.Jmp("foo");
m.Endp("baz");
program = m.GetImage();
program.Platform = new FakePlatform(null, arch);
Given_Project();
var scan = new Scanner(
program,
new ImportResolver(project, program, eventListener),
sc);
var sym = new ImageSymbol(addr);
scan.EnqueueImageSymbol(sym, true);
scan.ScanImage();
Assert.AreEqual(4, program.Procedures.Count);
}
public void Scanner_AddEntryPoint()
{
Given_Trace(new RtlTrace(0x12314)
{
m => { m.Return(4, 0); }
});
Given_Program(Address.Ptr32(0x12314), new byte[1]);
var project = new Project { Programs = { program } };
var sc = new Scanner(
this.program,
new ImportResolver(project, program, eventListener),
this.sc);
sc.EnqueueImageSymbol(new ImageSymbol(Address.Ptr32(0x12314)), true);
sc.ScanImage();
Assert.AreEqual(1, program.Procedures.Count);
Assert.AreEqual(0x12314, program.Procedures.Keys[0].Offset);
Assert.IsTrue(program.CallGraph.EntryPoints.Contains(program.Procedures.Values[0]));
}
