Add a generic service
This commit is contained in:
parent
1d6a71221a
commit
9ede6f669e
|
@ -41,10 +41,13 @@
|
||||||
<Compile Include="AssemblyService.cs" />
|
<Compile Include="AssemblyService.cs" />
|
||||||
<Compile Include="DelegateStringDecrypter.cs" />
|
<Compile Include="DelegateStringDecrypter.cs" />
|
||||||
<Compile Include="EmuStringDecrypter.cs" />
|
<Compile Include="EmuStringDecrypter.cs" />
|
||||||
|
<Compile Include="GenericService.cs" />
|
||||||
<Compile Include="IAssemblyService.cs" />
|
<Compile Include="IAssemblyService.cs" />
|
||||||
|
<Compile Include="IGenericService.cs" />
|
||||||
<Compile Include="IMethodDecrypterService.cs" />
|
<Compile Include="IMethodDecrypterService.cs" />
|
||||||
<Compile Include="IStringDecrypter.cs" />
|
<Compile Include="IStringDecrypter.cs" />
|
||||||
<Compile Include="IStringDecrypterService.cs" />
|
<Compile Include="IStringDecrypterService.cs" />
|
||||||
|
<Compile Include="IUserGenericService.cs" />
|
||||||
<Compile Include="MethodDecrypterService.cs" />
|
<Compile Include="MethodDecrypterService.cs" />
|
||||||
<Compile Include="methodsrewriter\AssemblyResolver.cs" />
|
<Compile Include="methodsrewriter\AssemblyResolver.cs" />
|
||||||
<Compile Include="methodsrewriter\CodeGenerator.cs" />
|
<Compile Include="methodsrewriter\CodeGenerator.cs" />
|
||||||
|
|
|
@ -35,6 +35,9 @@ namespace AssemblyData {
|
||||||
case AssemblyServiceType.MethodDecrypter:
|
case AssemblyServiceType.MethodDecrypter:
|
||||||
return new MethodDecrypterService();
|
return new MethodDecrypterService();
|
||||||
|
|
||||||
|
case AssemblyServiceType.Generic:
|
||||||
|
return new GenericService();
|
||||||
|
|
||||||
default:
|
default:
|
||||||
throw new ArgumentException("Invalid assembly service type");
|
throw new ArgumentException("Invalid assembly service type");
|
||||||
}
|
}
|
||||||
|
@ -48,6 +51,9 @@ namespace AssemblyData {
|
||||||
case AssemblyServiceType.MethodDecrypter:
|
case AssemblyServiceType.MethodDecrypter:
|
||||||
return typeof(MethodDecrypterService);
|
return typeof(MethodDecrypterService);
|
||||||
|
|
||||||
|
case AssemblyServiceType.Generic:
|
||||||
|
return typeof(GenericService);
|
||||||
|
|
||||||
default:
|
default:
|
||||||
throw new ArgumentException("Invalid assembly service type");
|
throw new ArgumentException("Invalid assembly service type");
|
||||||
}
|
}
|
||||||
|
@ -56,7 +62,7 @@ namespace AssemblyData {
|
||||||
public void DoNothing() {
|
public void DoNothing() {
|
||||||
}
|
}
|
||||||
|
|
||||||
public void Exit() {
|
public virtual void Exit() {
|
||||||
exitEvent.Set();
|
exitEvent.Set();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
69
AssemblyData/GenericService.cs
Normal file
69
AssemblyData/GenericService.cs
Normal file
|
@ -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);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
|
@ -21,6 +21,7 @@ namespace AssemblyData {
|
||||||
public enum AssemblyServiceType {
|
public enum AssemblyServiceType {
|
||||||
StringDecrypter,
|
StringDecrypter,
|
||||||
MethodDecrypter,
|
MethodDecrypter,
|
||||||
|
Generic,
|
||||||
}
|
}
|
||||||
|
|
||||||
public interface IAssemblyService {
|
public interface IAssemblyService {
|
||||||
|
|
31
AssemblyData/IGenericService.cs
Normal file
31
AssemblyData/IGenericService.cs
Normal file
|
@ -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);
|
||||||
|
}
|
||||||
|
}
|
28
AssemblyData/IUserGenericService.cs
Normal file
28
AssemblyData/IUserGenericService.cs
Normal file
|
@ -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);
|
||||||
|
}
|
||||||
|
}
|
|
@ -43,6 +43,10 @@ namespace de4dot.code.AssemblyClient {
|
||||||
get { return (IMethodDecrypterService)service; }
|
get { return (IMethodDecrypterService)service; }
|
||||||
}
|
}
|
||||||
|
|
||||||
|
public IGenericService GenericService {
|
||||||
|
get { return (IGenericService)service; }
|
||||||
|
}
|
||||||
|
|
||||||
public AssemblyClient(IAssemblyServerLoader loader) {
|
public AssemblyClient(IAssemblyServerLoader loader) {
|
||||||
this.loader = loader;
|
this.loader = loader;
|
||||||
}
|
}
|
||||||
|
|
|
@ -25,6 +25,7 @@ namespace de4dot.code.AssemblyClient {
|
||||||
IAssemblyService Service { get; }
|
IAssemblyService Service { get; }
|
||||||
IStringDecrypterService StringDecrypterService { get; }
|
IStringDecrypterService StringDecrypterService { get; }
|
||||||
IMethodDecrypterService MethodDecrypterService { get; }
|
IMethodDecrypterService MethodDecrypterService { get; }
|
||||||
|
IGenericService GenericService { get; }
|
||||||
void Connect();
|
void Connect();
|
||||||
void WaitConnected();
|
void WaitConnected();
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue
Block a user