de4dot-cex/de4dot.code/AssemblyClient/NewProcessAssemblyServerLoader.cs
2015-10-29 22:45:26 +01:00

71 lines
2.0 KiB
C#

/*
Copyright (C) 2011-2015 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.Diagnostics;
using AssemblyData;
namespace de4dot.code.AssemblyClient {
// Starts the server in a new process
public class NewProcessAssemblyServerLoader : IpcAssemblyServerLoader {
Process process;
public NewProcessAssemblyServerLoader(AssemblyServiceType serviceType)
: base(serviceType) {
}
public NewProcessAssemblyServerLoader(AssemblyServiceType serviceType, ServerClrVersion version)
: base(serviceType, version) {
}
public override void LoadServer(string filename) {
if (process != null)
throw new ApplicationException("Server is already loaded");
var psi = new ProcessStartInfo {
Arguments = string.Format("{0} {1} {2}", (int)serviceType,
Utils.ShellEscape(ipcName), Utils.ShellEscape(ipcUri)),
CreateNoWindow = true,
ErrorDialog = false,
FileName = filename,
LoadUserProfile = false,
UseShellExecute = false,
WorkingDirectory = Utils.GetOurBaseDir(),
};
process = Process.Start(psi);
if (process == null)
throw new ApplicationException("Could not start process");
}
public override void Dispose() {
if (process != null) {
if (!process.WaitForExit(300)) {
try {
process.Kill();
}
catch (InvalidOperationException) {
// Here if process has already exited.
}
}
process = null;
}
}
}
}