Initial commit

.gitignore

@@ -0,0 +1 @@
+*~

.gitmodules

@@ -0,0 +1,3 @@
+[submodule "cecil"]
+	path = cecil
+	url = [email protected]:0xd4d/cecil.git

AssemblyData/AssemblyData.csproj

@@ -0,0 +1,57 @@
+<?xml version="1.0" encoding="utf-8"?>
+<Project ToolsVersion="4.0" DefaultTargets="Build" xmlns="http://schemas.microsoft.com/developer/msbuild/2003">
+  <PropertyGroup>
+    <Configuration Condition=" '$(Configuration)' == '' ">Debug</Configuration>
+    <Platform Condition=" '$(Platform)' == '' ">AnyCPU</Platform>
+    <ProductVersion>8.0.30703</ProductVersion>
+    <SchemaVersion>2.0</SchemaVersion>
+    <ProjectGuid>{FBD84077-9D35-41FE-89DF-8D79EFE0B595}</ProjectGuid>
+    <OutputType>Library</OutputType>
+    <AppDesignerFolder>Properties</AppDesignerFolder>
+    <RootNamespace>AssemblyData</RootNamespace>
+    <AssemblyName>AssemblyData</AssemblyName>
+    <TargetFrameworkVersion>v2.0</TargetFrameworkVersion>
+    <FileAlignment>512</FileAlignment>
+  </PropertyGroup>
+  <PropertyGroup Condition=" '$(Configuration)|$(Platform)' == 'Debug|AnyCPU' ">
+    <DebugSymbols>true</DebugSymbols>
+    <DebugType>full</DebugType>
+    <Optimize>false</Optimize>
+    <OutputPath>..\Debug\</OutputPath>
+    <DefineConstants>DEBUG;TRACE</DefineConstants>
+    <ErrorReport>prompt</ErrorReport>
+    <WarningLevel>4</WarningLevel>
+    <PlatformTarget>AnyCPU</PlatformTarget>
+  </PropertyGroup>
+  <PropertyGroup Condition=" '$(Configuration)|$(Platform)' == 'Release|AnyCPU' ">
+    <DebugType>pdbonly</DebugType>
+    <Optimize>true</Optimize>
+    <OutputPath>..\Release\</OutputPath>
+    <DefineConstants>TRACE</DefineConstants>
+    <ErrorReport>prompt</ErrorReport>
+    <WarningLevel>4</WarningLevel>
+    <TreatWarningsAsErrors>true</TreatWarningsAsErrors>
+    <PlatformTarget>AnyCPU</PlatformTarget>
+  </PropertyGroup>
+  <ItemGroup>
+    <Compile Include="AssemblyServer.cs" />
+    <Compile Include="AssemblyService.cs" />
+    <Compile Include="DelegateStringDecrypter.cs" />
+    <Compile Include="IAssemblyService.cs" />
+    <Compile Include="IStringDecrypter.cs" />
+    <Compile Include="Properties\AssemblyInfo.cs" />
+    <Compile Include="SimpleData.cs" />
+    <Compile Include="Utils.cs" />
+  </ItemGroup>
+  <ItemGroup>
+    <Reference Include="System.Runtime.Remoting" />
+  </ItemGroup>
+  <Import Project="$(MSBuildToolsPath)\Microsoft.CSharp.targets" />
+  <!-- To modify your build process, add your task inside one of the targets below and uncomment it. 
+       Other similar extension points exist, see Microsoft.Common.targets.
+  <Target Name="BeforeBuild">
+  </Target>
+  <Target Name="AfterBuild">
+  </Target>
+  -->
+</Project>

AssemblyData/AssemblyServer.cs

@@ -0,0 +1,45 @@
+/*
+    Copyright (C) 2011 [email protected]
+
+    This file is part of de4dot.
+
+    de4dot is free software: you can redistribute it and/or modify
+    it under the terms of the GNU General Public License as published by
+    the Free Software Foundation, either version 3 of the License, or
+    (at your option) any later version.
+
+    de4dot is distributed in the hope that it will be useful,
+    but WITHOUT ANY WARRANTY; without even the implied warranty of
+    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+    GNU General Public License for more details.
+
+    You should have received a copy of the GNU General Public License
+    along with de4dot.  If not, see <http://www.gnu.org/licenses/>.
+*/
+
+using System;
+using System.Runtime.Remoting;
+using System.Runtime.Remoting.Channels;
+using System.Runtime.Remoting.Channels.Ipc;
+using AssemblyData;
+
+namespace AssemblyServer {
+	public static class Start {
+		public static int main(string[] args) {
+			if (args.Length != 2)
+				Environment.Exit(1);
+			var channelName = args[0];
+			var uri = args[1];
+
+			var service = new AssemblyService();
+			startServer(service, channelName, uri);
+			service.waitExit();
+			return 0;
+		}
+
+		static void startServer(AssemblyService service, string name, string uri) {
+			ChannelServices.RegisterChannel(new IpcServerChannel(name), false);
+			RemotingServices.Marshal(service, uri);
+		}
+	}
+}

AssemblyData/AssemblyService.cs

@@ -0,0 +1,122 @@
+/*
+    Copyright (C) 2011 [email protected]
+
+    This file is part of de4dot.
+
+    de4dot is free software: you can redistribute it and/or modify
+    it under the terms of the GNU General Public License as published by
+    the Free Software Foundation, either version 3 of the License, or
+    (at your option) any later version.
+
+    de4dot is distributed in the hope that it will be useful,
+    but WITHOUT ANY WARRANTY; without even the implied warranty of
+    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+    GNU General Public License for more details.
+
+    You should have received a copy of the GNU General Public License
+    along with de4dot.  If not, see <http://www.gnu.org/licenses/>.
+*/
+
+using System;
+using System.Reflection;
+using System.Threading;
+
+namespace AssemblyData {
+	public class AssemblyService : MarshalByRefObject, IAssemblyService {
+		IStringDecrypter stringDecrypter = null;
+		ManualResetEvent exitEvent = new ManualResetEvent(false);
+		Assembly assembly = null;
+
+		public void doNothing() {
+		}
+
+		void checkStringDecrypter() {
+			if (stringDecrypter == null)
+				throw new ApplicationException("setStringDecrypterType() hasn't been called yet.");
+		}
+
+		void checkAssembly() {
+			if (assembly == null)
+				throw new ApplicationException("loadAssembly() hasn't been called yet.");
+		}
+
+		public void loadAssembly(string filename) {
+			if (assembly != null)
+				throw new ApplicationException("Only one assembly can be explicitly loaded");
+			try {
+				assembly = Assembly.LoadFrom(filename);
+			}
+			catch (BadImageFormatException) {
+				throw new ApplicationException(string.Format("Could not load assembly {0}. Maybe it's 32-bit or 64-bit only?", filename));
+			}
+		}
+
+		public void setStringDecrypterType(StringDecrypterType type) {
+			if (stringDecrypter != null)
+				throw new ApplicationException("StringDecrypterType already set");
+
+			switch (type) {
+			case StringDecrypterType.Delegate:
+				stringDecrypter = new DelegateStringDecrypter();
+				break;
+
+			default:
+				throw new ApplicationException(string.Format("Unknown StringDecrypterType {0}", type));
+			}
+		}
+
+		public int defineStringDecrypter(int methodToken, int typeToken) {
+			checkStringDecrypter();
+			var methodInfo = findMethod(methodToken, typeToken);
+			if (methodInfo == null)
+				throw new ApplicationException(string.Format("Could not find method {0:X8}", methodToken));
+			if (methodInfo.ReturnType != typeof(string))
+				throw new ApplicationException(string.Format("Method return type must be string: {0}", methodInfo));
+			return stringDecrypter.defineStringDecrypter(methodInfo);
+		}
+
+		public object[] decryptStrings(int stringDecrypterMethod, object[] args) {
+			checkStringDecrypter();
+			foreach (var arg in args)
+				SimpleData.unpack((object[])arg);
+			return SimpleData.pack(stringDecrypter.decryptStrings(stringDecrypterMethod, args));
+		}
+
+		public void exit() {
+			exitEvent.Set();
+		}
+
+		public void waitExit() {
+			exitEvent.WaitOne();
+		}
+
+		public override object InitializeLifetimeService() {
+			return null;
+		}
+
+		MethodInfo findMethod(int methodToken, int typeToken) {
+			checkAssembly();
+
+			const BindingFlags bindingFlags = BindingFlags.NonPublic | BindingFlags.Public | BindingFlags.Static;
+
+			foreach (var module in assembly.GetModules()) {
+				foreach (var method in module.GetMethods(bindingFlags)) {
+					if (method.MetadataToken == methodToken)
+						return method;
+				}
+			}
+
+			foreach (var type in assembly.GetTypes()) {
+				if (type.MetadataToken == typeToken || typeToken == 0) {
+					var methods = type.GetMethods(bindingFlags);
+					foreach (var method in methods) {
+						if (method.MetadataToken == methodToken)
+							return method;
+					}
+				}
+			}
+
+			return null;
+		}
+	}
+}

AssemblyData/DelegateStringDecrypter.cs

@@ -0,0 +1,52 @@
+/*
+    Copyright (C) 2011 [email protected]
+
+    This file is part of de4dot.
+
+    de4dot is free software: you can redistribute it and/or modify
+    it under the terms of the GNU General Public License as published by
+    the Free Software Foundation, either version 3 of the License, or
+    (at your option) any later version.
+
+    de4dot is distributed in the hope that it will be useful,
+    but WITHOUT ANY WARRANTY; without even the implied warranty of
+    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+    GNU General Public License for more details.
+
+    You should have received a copy of the GNU General Public License
+    along with de4dot.  If not, see <http://www.gnu.org/licenses/>.
+*/
+
+using System;
+using System.Reflection;
+using System.Reflection.Emit;
+using System.Collections.Generic;
+
+namespace AssemblyData {
+	class DelegateStringDecrypter : IStringDecrypter {
+		delegate string DecryptString(object[] args);
+		List<DecryptString> stringDecryptMethods = new List<DecryptString>();
+
+		public int defineStringDecrypter(MethodInfo method) {
+			stringDecryptMethods.Add(buildDynamicMethod(method));
+			return stringDecryptMethods.Count - 1;
+		}
+
+		public object[] decryptStrings(int stringDecrypterMethod, object[] args) {
+			if (stringDecrypterMethod > stringDecryptMethods.Count)
+				throw new ApplicationException("Invalid string decrypter method");
+
+			var rv = new object[args.Length];
+			var stringDecrypter = stringDecryptMethods[stringDecrypterMethod];
+			for (int i = 0; i < args.Length; i++)
+				rv[i] = stringDecrypter((object[])args[i]);
+			return rv;
+		}
+
+		DecryptString buildDynamicMethod(MethodInfo method) {
+			var dm = new DynamicMethod("", typeof(string), new Type[] { typeof(object[]) }, typeof(DelegateStringDecrypter), true);
+			Utils.addCallStringDecrypterMethodInstructions(method, dm.GetILGenerator());
+			return (DecryptString)dm.CreateDelegate(typeof(DecryptString));
+		}
+	}
+}

AssemblyData/IAssemblyService.cs

@@ -0,0 +1,33 @@
+/*
+    Copyright (C) 2011 [email protected]
+
+    This file is part of de4dot.
+
+    de4dot is free software: you can redistribute it and/or modify
+    it under the terms of the GNU General Public License as published by
+    the Free Software Foundation, either version 3 of the License, or
+    (at your option) any later version.
+
+    de4dot is distributed in the hope that it will be useful,
+    but WITHOUT ANY WARRANTY; without even the implied warranty of
+    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+    GNU General Public License for more details.
+
+    You should have received a copy of the GNU General Public License
+    along with de4dot.  If not, see <http://www.gnu.org/licenses/>.
+*/
+
+namespace AssemblyData {
+	public enum StringDecrypterType {
+		Delegate,
+	}
+
+	public interface IAssemblyService {
+		void doNothing();
+		void loadAssembly(string filename);
+		void setStringDecrypterType(StringDecrypterType type);
+		int defineStringDecrypter(int methodToken, int typeToken = 0);
+		object[] decryptStrings(int stringDecrypterMethod, object[] args);
+		void exit();
+	}
+}

AssemblyData/IStringDecrypter.cs

@@ -0,0 +1,27 @@
+/*
+    Copyright (C) 2011 [email protected]
+
+    This file is part of de4dot.
+
+    de4dot is free software: you can redistribute it and/or modify
+    it under the terms of the GNU General Public License as published by
+    the Free Software Foundation, either version 3 of the License, or
+    (at your option) any later version.
+
+    de4dot is distributed in the hope that it will be useful,
+    but WITHOUT ANY WARRANTY; without even the implied warranty of
+    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+    GNU General Public License for more details.
+
+    You should have received a copy of the GNU General Public License
+    along with de4dot.  If not, see <http://www.gnu.org/licenses/>.
+*/
+
+using System.Reflection;
+
+namespace AssemblyData {
+	interface IStringDecrypter {
+		int defineStringDecrypter(MethodInfo method);
+		object[] decryptStrings(int stringDecrypterMethod, object[] args);
+	}
+}

AssemblyData/Properties/AssemblyInfo.cs

@@ -0,0 +1,34 @@
+/*
+    Copyright (C) 2011 [email protected]
+
+    This file is part of de4dot.
+
+    de4dot is free software: you can redistribute it and/or modify
+    it under the terms of the GNU General Public License as published by
+    the Free Software Foundation, either version 3 of the License, or
+    (at your option) any later version.
+
+    de4dot is distributed in the hope that it will be useful,
+    but WITHOUT ANY WARRANTY; without even the implied warranty of
+    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+    GNU General Public License for more details.
+
+    You should have received a copy of the GNU General Public License
+    along with de4dot.  If not, see <http://www.gnu.org/licenses/>.
+*/
+
+using System.Reflection;
+using System.Runtime.CompilerServices;
+using System.Runtime.InteropServices;
+
+[assembly: AssemblyTitle("AssemblyData")]
+[assembly: AssemblyDescription(".NET Assembly data for use by client and server")]
+[assembly: AssemblyConfiguration("")]
+[assembly: AssemblyCompany("")]
+[assembly: AssemblyProduct("AssemblyData")]
+[assembly: AssemblyCopyright("Copyright (C) 2011 [email protected]")]
+[assembly: AssemblyTrademark("")]
+[assembly: AssemblyCulture("")]
+[assembly: ComVisible(false)]
+[assembly: AssemblyVersion("1.0.0.3405")]
+[assembly: AssemblyFileVersion("1.0.0.3405")]