示例#1
0
        private void LoadDependencies(ModuleDefMD module)
        {
            if (seenModules.Contains(module.Location))
            {
                return;
            }

            seenModules.Add(module.Location);
            Modules.Add(module);
            Resolver.AddToCache(module);

            foreach (var assemblyRef in module.GetAssemblyRefs())
            {
                // There are cases, where the Resolver will not be able to resolve the assemblies
                // automatically, even if they are in the same directory.
                // (maybe this has todo with linux / specific mono versions?)
                // So, try to load them manually recursively first.
                var subModuleFile = Path.Combine(Path.GetDirectoryName(module.Location), assemblyRef.Name + ".dll");
                if (File.Exists(subModuleFile))
                {
                    var subModule = ModuleDefMD.Load(subModuleFile, Resolver.DefaultModuleContext);
                    if (subModule != null)
                    {
                        LoadDependencies(subModule);
                    }
                }

                var assembly = Resolver.ResolveThrow(assemblyRef, null);

                foreach (var moduleRef in assembly.Modules)
                {
                    LoadDependencies((ModuleDefMD)moduleRef);
                }
            }
        }
示例#2
0
        public void Patch()
        {
            var resolver = new Resolver(this);
            var context  = new ModuleContext(resolver);

            resolver.Context = context;
            bool modified = false;

            logger.Log($"Loading {AssemblyCSharp.Name}...");
            var assemblyCSharp = ModuleDefMD.Load(File.ReadAllBytes(AssemblyCSharp.FullName), context);

            logger.Log(" -> Loaded: {0}", assemblyCSharp.FullName);
            resolver.AddToCache(assemblyCSharp.Assembly);

            logger.Log($"Loading {AssemblyCSharpFirstpass.Name}...");
            var assemblyCSharpFirstpass = ModuleDefMD.Load(File.ReadAllBytes(AssemblyCSharpFirstpass.FullName), context);

            logger.Log(" -> Loaded: {0}", assemblyCSharpFirstpass.FullName);
            resolver.AddToCache(assemblyCSharpFirstpass.Assembly);

            foreach (var p in patches)
            {
                logger.Log("Executing patch {0}...", p.Name);
                if (p.Patch(assemblyCSharp, assemblyCSharpFirstpass))
                {
                    modified = true;
                }
                else
                {
                    logger.Log("{0} was not applied", p.Name);
                }
            }

            if (modified)
            {
                var suffix = $".{DateTime.Now.Ticks}.bak";
                SaveModule(AssemblyCSharp, assemblyCSharp, suffix);
                SaveModule(AssemblyCSharpFirstpass, assemblyCSharpFirstpass, suffix);
            }
            else
            {
                logger.Error("The assembly was not modified");
            }
        }
        private void LoadDependencies(ModuleDefMD module)
        {
            if (seenModules.Contains(module.Location))
            {
                return;
            }
            seenModules.Add(module.Location);
            Modules.Add(module);
            Resolver.AddToCache(module);

            foreach (var assemblyRef in module.GetAssemblyRefs())
            {
                AssemblyDef assembly = Resolver.ResolveThrow(assemblyRef, null);
                foreach (var moduleRef in assembly.Modules)
                {
                    LoadDependencies((ModuleDefMD)moduleRef);
                }
            }
        }