コード例 #1
0
ファイル: Engine.cs プロジェクト: WeavR/WeavR
        public static void Process(ILoggerContext logger, ProjectDetails assemblyDetails, string tempFolder = null)
        {
            if (!assemblyDetails.Verify(logger))
                return;

            tempFolder = tempFolder ?? Path.GetTempPath();

            string pdbFile = Path.ChangeExtension(assemblyDetails.AssemblyPath, ".pdb");

            var newAssemblyPath = Path.Combine(tempFolder, String.Format("{0}.weavr{1}", Path.GetFileNameWithoutExtension(assemblyDetails.AssemblyPath), Path.GetExtension(assemblyDetails.AssemblyPath)));
            var newPdbPath = File.Exists(pdbFile) ? Path.ChangeExtension(newAssemblyPath, ".pdb") : null;

            using (var host = new PeReader.DefaultHost())
            {
                var targetAssembly = (IAssembly)host.LoadUnitFrom(assemblyDetails.AssemblyPath);

                using (var pdbStream = newPdbPath != null ? File.OpenRead(pdbFile) : null)
                using (var pdbReader = newPdbPath != null ? new PdbReader(pdbStream, host) : null)
                {
                    var decompiled = Decompiler.GetCodeModelFromMetadataModel(host, targetAssembly, pdbReader);
                    decompiled = new CodeDeepCopier(host, pdbReader).Copy(decompiled);

                    var engine = new Engine(logger, host);
                    engine.Process(assemblyDetails, decompiled);

                    using (var peStream = File.Create(newAssemblyPath))
                    {
                        if (pdbReader == null)
                        {
                            PeWriter.WritePeToStream(decompiled, host, peStream);
                        }
                        else
                        {
                            using (var pdbWriter = new PdbWriter(newPdbPath, pdbReader))
                            {
                                PeWriter.WritePeToStream(decompiled, host, peStream, pdbReader, pdbReader, pdbWriter);
                            }
                        }
                    }
                }
            }

            File.Delete(assemblyDetails.AssemblyPath);
            File.Move(newAssemblyPath, assemblyDetails.AssemblyPath);

            if (!string.IsNullOrEmpty(newPdbPath))
            {
                File.Delete(pdbFile);
                File.Move(newPdbPath, pdbFile);
            }
        }
コード例 #2
0
ファイル: Engine.cs プロジェクト: WeavR/WeavR
        private void Process(ProjectDetails assemblyDetails, Assembly assembly)
        {
            logger.LogInfo("WeavR (version {0}) Executing", typeof(Engine).Assembly.GetName().Version);

            var stopwatch = Stopwatch.StartNew();

            try
            {
                if (assembly.AllTypes.Any(t => t.Name.Value == "ProcessedByWeavR"))
                {
                    logger.LogWarning("Assembly already processed by WeavR.");
                    return;
                }

                var weaverFinder = new WeaverFinder(logger, assemblyDetails);

                var weaversConfigs = weaverFinder.FindWeaverConfigs()
                    .Select(s => XElement.Load(s))
                    .SelectMany(root => root.Descendants())
                    .ToList();

                var weavers = weaverFinder.FindWeavers();

                var weaverMutators = weaversConfigs
                    .Select(config => new { config, weaver = weavers.FirstOrDefault(w => Path.GetFileNameWithoutExtension(w) == config.Name.LocalName + ".WeavR") })
                    .WhereWithActions(a => a.weaver != null, null, a => logger.LogWarning(@"Could not find a weaver named '{0}'.", a.config.Name.LocalName)) //Error?
                    .Select(a => new WeaverMutator(logger, a.weaver))
                    .ToList();

                if (logger.HasLoggedError)
                    return;

                var mutators = new List<Mutator>();
                mutators.AddRange(weaverMutators);
                mutators.Add(new ProcessedFlagMutator());

                foreach (var mutator in mutators)
                {
                    mutator.Host = host;
                    mutator.Logger = logger.CreateSubContext(mutator.Name);

                    var weaverMutator = mutator as WeaverMutator;
                    if (weaverMutator != null)
                    {
                        var config = weaversConfigs.FirstOrDefault(c => c.Name == weaverMutator.Name);
                        if (config != null)
                            weaverMutator.Configure(config);
                    }

                    mutator.Mutate(assembly);
                }
            }
            catch (Exception ex)
            {
                logger.LogException(ex);
            }
            finally
            {
                logger.LogInfo("WeavR Finished {0}ms.", stopwatch.ElapsedMilliseconds);
            }
        }
コード例 #3
0
ファイル: WeaverFinder.cs プロジェクト: WeavR/WeavR
 public WeaverFinder(ILoggerContext logger, ProjectDetails project)
 {
     this.logger = logger;
     this.project = project;
 }