Add a generic service

AssemblyData/AssemblyData.csproj

@@ -41,10 +41,13 @@
     <Compile Include="AssemblyService.cs" />
     <Compile Include="DelegateStringDecrypter.cs" />
     <Compile Include="EmuStringDecrypter.cs" />
+    <Compile Include="GenericService.cs" />
     <Compile Include="IAssemblyService.cs" />
+    <Compile Include="IGenericService.cs" />
     <Compile Include="IMethodDecrypterService.cs" />
     <Compile Include="IStringDecrypter.cs" />
     <Compile Include="IStringDecrypterService.cs" />
+    <Compile Include="IUserGenericService.cs" />
     <Compile Include="MethodDecrypterService.cs" />
     <Compile Include="methodsrewriter\AssemblyResolver.cs" />
     <Compile Include="methodsrewriter\CodeGenerator.cs" />

AssemblyData/AssemblyService.cs

@@ -35,6 +35,9 @@ namespace AssemblyData {
 			case AssemblyServiceType.MethodDecrypter:
 				return new MethodDecrypterService();
 
+			case AssemblyServiceType.Generic:
+				return new GenericService();
+
 			default:
 				throw new ArgumentException("Invalid assembly service type");
 			}
@@ -48,6 +51,9 @@ namespace AssemblyData {
 			case AssemblyServiceType.MethodDecrypter:
 				return typeof(MethodDecrypterService);
 
+			case AssemblyServiceType.Generic:
+				return typeof(GenericService);
+
 			default:
 				throw new ArgumentException("Invalid assembly service type");
 			}
@@ -56,7 +62,7 @@ namespace AssemblyData {
 		public void DoNothing() {
 		}
 
-		public void Exit() {
+		public virtual void Exit() {
 			exitEvent.Set();
 		}
 

AssemblyData/GenericService.cs

@@ -0,0 +1,69 @@
+/*
+    Copyright (C) 2011-2013 de4dot@gmail.com
+
+    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;
+
+namespace AssemblyData {
+	class GenericService : AssemblyService, IGenericService {
+		Assembly userServiceAssembly;
+		IUserGenericService userGenericService;
+
+		public override void Exit() {
+			if (userGenericService != null)
+				userGenericService.Dispose();
+			userGenericService = null;
+			base.Exit();
+		}
+
+		public void LoadUserService(string asmPath, Type createServiceType, object createMethodArgs) {
+			userServiceAssembly = Assembly.LoadFile(asmPath);
+			var createServiceMethod = GetCreateUserServiceMethod(createServiceType);
+			userGenericService = createServiceMethod.Invoke(null, null) as IUserGenericService;
+			if (userGenericService == null)
+				throw new ApplicationException("create-service-method failed to create user service");
+		}
+
+		MethodInfo GetCreateUserServiceMethod(Type createServiceType) {
+			if (createServiceType == null)
+				throw new ApplicationException(string.Format("Create-service-type is null"));
+			foreach (var method in createServiceType.GetMethods(BindingFlags.DeclaredOnly | BindingFlags.Static | BindingFlags.Public | BindingFlags.NonPublic)) {
+				if (method.GetCustomAttributes(typeof(CreateUserGenericServiceAttribute), false).Length > 0)
+					return method;
+			}
+			throw new ApplicationException(string.Format("Failed to find create-service-method. Type token: Type: {0}", createServiceType));
+		}
+
+		void CheckUserService() {
+			if (userGenericService == null)
+				throw new ApplicationException("LoadUserService() hasn't been called yet.");
+		}
+
+		public void LoadAssembly(string filename) {
+			CheckUserService();
+			LoadAssemblyInternal(filename);
+			userGenericService.AssemblyLoaded(assembly);
+		}
+
+		public object SendMessage(int msg, object[] args) {
+			CheckUserService();
+			return userGenericService.HandleMessage(msg, args);
+		}
+	}
+}

AssemblyData/IAssemblyService.cs

@@ -21,6 +21,7 @@ namespace AssemblyData {
 	public enum AssemblyServiceType {
 		StringDecrypter,
 		MethodDecrypter,
+		Generic,
 	}
 
 	public interface IAssemblyService {

AssemblyData/IGenericService.cs

@@ -0,0 +1,31 @@
+/*
+    Copyright (C) 2011-2013 de4dot@gmail.com
+
+    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;
+
+namespace AssemblyData {
+	public class CreateUserGenericServiceAttribute : Attribute {
+	}
+
+	public interface IGenericService : IAssemblyService {
+		void LoadUserService(string asmPath, Type createServiceType, object createMethodArgs);
+		void LoadAssembly(string filename);
+		object SendMessage(int msg, object[] args);
+	}
+}

AssemblyData/IUserGenericService.cs

@@ -0,0 +1,28 @@
+/*
+    Copyright (C) 2011-2013 de4dot@gmail.com
+
+    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;
+
+namespace AssemblyData {
+	public interface IUserGenericService : IDisposable {
+		void AssemblyLoaded(Assembly assembly);
+		object HandleMessage(int msg, object[] args);
+	}
+}

de4dot.code/AssemblyClient/AssemblyClient.cs

@@ -43,6 +43,10 @@ namespace de4dot.code.AssemblyClient {
 			get { return (IMethodDecrypterService)service; }
 		}
 
+		public IGenericService GenericService {
+			get { return (IGenericService)service; }
+		}
+
 		public AssemblyClient(IAssemblyServerLoader loader) {
 			this.loader = loader;
 		}

de4dot.code/AssemblyClient/IAssemblyClient.cs

@@ -25,6 +25,7 @@ namespace de4dot.code.AssemblyClient {
 		IAssemblyService Service { get; }
 		IStringDecrypterService StringDecrypterService { get; }
 		IMethodDecrypterService MethodDecrypterService { get; }
+		IGenericService GenericService { get; }
 		void Connect();
 		void WaitConnected();
 	}