public void Run(RemoteHooking.IContext hookingContext, bool isDebugging, string hackPath, string installPath)
{
try
{
PluginSystem.HackPath = hackPath;
PluginSystem.IsDebugging = isDebugging;
if (isDebugging)
{
DebuggerApplication.Start(hackPath);
while (!DebuggerApplication.IsReady)
{
Thread.Sleep(1); // Sleep(0) is a nono.
}
}
// Everything traced will be written to "debug.log".
string logPath = Path.Combine(hackPath, "logs\\runtime");
if (!Directory.Exists(logPath))
{
Directory.CreateDirectory(logPath);
}
Trace.Listeners.Add(new TextWriterTraceListener(Path.Combine(logPath, $"{DateTime.Now:yyyy-MM-dd HH.mm.ss}.log")));
Trace.IndentSize = 2;
// We autoflush our trace, so we get everything immediately. This
// makes tracing a bit more expensive, but means we still get a log
// even if there's a fatal crash.
Trace.AutoFlush = true;
Trace.WriteLine("-------------------");
Trace.WriteLine(DateTime.Now);
Trace.WriteLine("-------------------");
AppDomain.CurrentDomain.AssemblyResolve += (object sender, ResolveEventArgs args) =>
{
var path = string.Empty;
// extract the file name
var file = string.Empty;
if (args.Name.IndexOf(',') >= 0)
{
file = args.Name.Substring(0, args.Name.IndexOf(',')) + ".dll";
}
else if (args.Name.IndexOf(".dll") >= 0)
{
file = Path.GetFileName(args.Name);
}
else
{
return(null);
}
// locate the actual file
path = Directory.GetFiles(hackPath, file, SearchOption.AllDirectories).FirstOrDefault();
if (!string.IsNullOrEmpty(path))
{
try
{
return(Assembly.LoadFrom(path));
}
catch
{
return(Assembly.Load(File.ReadAllBytes(path)));
}
}
path = Directory.GetFiles(pluginsFolder, file, SearchOption.AllDirectories).FirstOrDefault();
if (!string.IsNullOrEmpty(path))
{
try
{
return(Assembly.LoadFrom(path));
}
catch
{
return(Assembly.Load(File.ReadAllBytes(path)));
}
}
path = Directory.GetFiles(libraryFolder, file, SearchOption.AllDirectories).FirstOrDefault();
if (!string.IsNullOrEmpty(path))
{
try
{
return(Assembly.LoadFrom(path));
}
catch
{
return(Assembly.Load(File.ReadAllBytes(path)));
}
}
return(null);
};
AppDomain.CurrentDomain.ReflectionOnlyAssemblyResolve += (object sender, ResolveEventArgs args) =>
{
var path = string.Empty;
// extract the file name
var file = string.Empty;
if (args.Name.IndexOf(',') >= 0)
{
try
{
return(Assembly.ReflectionOnlyLoad(args.Name));
}
catch
{
file = args.Name.Substring(0, args.Name.IndexOf(',')) + ".dll";
}
}
else if (args.Name.IndexOf(".dll") >= 0)
{
file = Path.GetFileName(args.Name);
}
else
{
return(null);
}
// locate the actual file
path = Directory.GetFiles(hackPath, file, SearchOption.AllDirectories).FirstOrDefault();
if (!string.IsNullOrEmpty(path))
{
try
{
return(Assembly.ReflectionOnlyLoadFrom(path));
}
catch
{
return(Assembly.ReflectionOnlyLoad(File.ReadAllBytes(path)));
}
}
path = Directory.GetFiles(pluginsFolder, file, SearchOption.AllDirectories).FirstOrDefault();
if (!string.IsNullOrEmpty(path))
{
try
{
return(Assembly.ReflectionOnlyLoadFrom(path));
}
catch
{
return(Assembly.ReflectionOnlyLoad(File.ReadAllBytes(path)));
}
}
path = Directory.GetFiles(libraryFolder, file, SearchOption.AllDirectories).FirstOrDefault();
if (!string.IsNullOrEmpty(path))
{
try
{
return(Assembly.ReflectionOnlyLoadFrom(path));
}
catch
{
return(Assembly.ReflectionOnlyLoad(File.ReadAllBytes(path)));
}
}
return(null);
};
var sw = new Stopwatch();
Trace.WriteLine("Preparing folders . . . ");
Trace.Indent();
sw.Restart();
pluginsFolder = Path.Combine(hackPath, "plugins");
libraryFolder = Path.Combine(hackPath, "lib");
if (!Directory.Exists(pluginsFolder))
{
Directory.CreateDirectory(pluginsFolder);
}
sw.Stop();
Trace.WriteLine($"Install Path: {installPath}");
Trace.WriteLine($"Hack Path: {hackPath}");
if (installPath.Equals(hackPath, StringComparison.OrdinalIgnoreCase))
{
Trace.WriteLine("WARNING: Install Path and Hack Path are the same. This is not supported.");
}
//if (File.Exists(Path.Combine(installPath, "Launcher.exe")))
// Trace.WriteLine("WARNING: Launcher.exe detected in the Warcraft III folder. This is not supported.");
//if (File.Exists(Path.Combine(installPath, "Cirnix.JassNative.Runtime.dll")))
// Trace.WriteLine("WARNING: Cirnix.JassNative.Runtime.dll detected in the Warcraft III folder. This is not supported.");
Trace.WriteLine($"Done! ({sw.Elapsed.TotalMilliseconds:0.00} ms)");
Trace.Unindent();
Trace.WriteLine($"Loading plugins from '{pluginsFolder}' . . .");
Trace.Indent();
sw.Restart();
PluginSystem.InitSystem();
PluginSystem.LoadPlugins(pluginsFolder);
sw.Stop();
Trace.WriteLine($"Done! ({sw.Elapsed.TotalMilliseconds:0.00} ms)");
Trace.Unindent();
// Prepare the OnGameLoad hook.
LoadLibraryA = Memory.InstallHook(LocalHook.GetProcAddress("kernel32.dll", "LoadLibraryA"), new Kernel32.LoadLibraryAPrototype(LoadLibraryAHook), false, true);
// Everyone has had their chance to inject stuff,
// time to wake up the process.
RemoteHooking.WakeUpProcess();
Trace.WriteLine("WakeUpProcess Proceed!");
// Let the thread stay alive, so all hooks stay alive as well.
// This might need to be shutdown properly on exit.
Trace.WriteLine("Sleep Proceed!");
Thread.Sleep(Timeout.Infinite);
}
catch (Exception exception)
{
MessageBox.Show(
"Fatal exception!" + Environment.NewLine +
exception + Environment.NewLine +
"Aborting execution!",
GetType() + ".Run(...)", MessageBoxButton.OK, MessageBoxImage.Error);
Process.GetCurrentProcess().Kill();
}
}
