Ejemplo n.º 1
0
        private static void Retarget(TypeDefinition type, DllImportMap map, DllImports imports)
        {
            foreach (var nested in type.NestedTypes)
            {
                Retarget(nested, map, imports);
            }

            foreach (MethodDefinition md in type.Methods)
            {
                if (!md.HasBody)
                {
                    continue;
                }

                for (int i = 0; i < md.Body.Instructions.Count; i++)
                {
                    Instruction ins = md.Body.Instructions[i];

                    if (ins.OpCode == OpCodes.Call)
                    {
                        MethodDefinition method_operand = ins.Operand as MethodDefinition;
                        if (method_operand == null)
                        {
                            continue;
                        }

                        PInvokeInfo pinfo = method_operand.PInvokeInfo;
                        if (pinfo == null)
                        {
                            continue;
                        }

                        ImportKey key = new ImportKey(pinfo.Module.Name, pinfo.EntryPoint);
                        if (imports.ContainsKey(key))
                        {
                            //Console.WriteLine ("{0} is a pinvoke, {1}/{2}", method_operand, pinfo.EntryPoint, pinfo.Module.Name);
                            if (map.ContainsKey(key))
                            {
                                Console.WriteLine("retargeting reference to method method {0}/{1}", key.module_name, key.entry_point);

                                var il = md.Body.GetILProcessor();
                                MethodDefinition mapped_method = map[key];
                                MethodReference  mapped_ref;

                                mapped_ref = type.Module.Import(mapped_method);

                                Instruction callMethod = il.Create(OpCodes.Call, mapped_ref);

                                il.Replace(ins, callMethod);
                            }
                            else
                            {
                                Console.WriteLine("WARNING: no map entry for method {0}/{1}", key.module_name, key.entry_point);
                            }
                        }
                    }
                }
            }
        }
Ejemplo n.º 2
0
        public override bool Equals(object o)
        {
            ImportKey key = o as ImportKey;

            if (key == null)
            {
                return(false);
            }
            return(key.module_name == module_name && key.entry_point == entry_point);
        }
Ejemplo n.º 3
0
        private static DllImports CollectDllImports(AssemblyDefinition assembly)
        {
            DllImports imports = new DllImports();

            foreach (TypeDefinition t in assembly.MainModule.GetAllTypes())
            {
                if (t.Name == "<Module>")
                {
                    continue;
                }

                if (t.IsSpecialName || t.IsRuntimeSpecialName)
                {
                    continue;
                }

                foreach (MethodDefinition md in t.Methods)
                {
                    if (md.IsSpecialName)
                    {
                        continue;
                    }

                    if (IsFinalizer(md))
                    {
                        continue;
                    }

                    PInvokeInfo pinfo = md.PInvokeInfo;
                    if (pinfo == null)
                    {
                        continue;
                    }

//                  Console.WriteLine ("{0} is a pinvoke, hashcode = {1}", md, md.GetHashCode());

                    ImportKey key = new ImportKey(pinfo.Module.Name, pinfo.EntryPoint);
                    if (imports.ContainsKey(key))
                    {
                        Console.WriteLine("WARNING: pinvoke {0}/{1} shows up more than once in input assembly", key.module_name, key.entry_point);
                    }
                    else
                    {
                        imports.Add(key, md);
                    }
                }
            }

            return(imports);
        }
Ejemplo n.º 4
0
        private static DllImportMap BuildMap(string map_assembly)
        {
            DllImportMap map = new DllImportMap();

            var assembly = AssemblyDefinition.ReadAssembly(map_assembly);

            foreach (TypeDefinition t in assembly.MainModule.GetAllTypes())
            {
                if (t.Name == "<Module>")
                {
                    continue;
                }

                if (t.IsSpecialName || t.IsRuntimeSpecialName)
                {
                    continue;
                }

                foreach (MethodDefinition md in t.Methods)
                {
                    if (md.IsSpecialName)
                    {
                        continue;
                    }

                    if (IsFinalizer(md))
                    {
                        continue;
                    }

                    CustomAttribute attr = GetMapDllImportAttribute(md);
                    if (attr == null)
                    {
                        continue;
                    }

                    ImportKey key = new ImportKey(attr.ConstructorArguments [0].Value.ToString(), md.Name);
                    map.Add(key, md);
                }
            }

            return(map);
        }