public void StartHost(string profilePath, string logFolder, ILogger outputLogger, LogVerbosity verbosity)
{
if (_hostEnd != null)
{
throw new InvalidOperationException("Host process is already running");
}
var useridentity = User as WindowsIdentity;
// In remote broker User Identity type is always WindowsIdentity
string suppressUI = (useridentity == null) ? string.Empty : "--rhost-suppress-ui ";
string isRepl = _isInteractive ? "--rhost-interactive " : string.Empty;
string brokerPath = Path.GetDirectoryName(typeof(Program).Assembly.GetAssemblyPath());
string rhostExePath = Path.Combine(brokerPath, RHostExe);
string logFolderParam = string.IsNullOrEmpty(logFolder) ? string.Empty : Invariant($"--rhost-log-dir \"{logFolder}\"");
string commandLine = Invariant($"\"{rhostExePath}\" {suppressUI}{isRepl}--rhost-name \"{Id}\" {logFolderParam} --rhost-log-verbosity {(int)verbosity} {CommandLineArguments}");
var usernameBldr = new StringBuilder(NativeMethods.CREDUI_MAX_USERNAME_LENGTH + 1);
var domainBldr = new StringBuilder(NativeMethods.CREDUI_MAX_DOMAIN_LENGTH + 1);
// Get R_HOME value
var shortHome = new StringBuilder(NativeMethods.MAX_PATH);
NativeMethods.GetShortPathName(Interpreter.Info.Path, shortHome, shortHome.Capacity);
Stream stdout, stdin, stderror;
bool loggedOnUser = useridentity != null && WindowsIdentity.GetCurrent().User != useridentity.User;
// build user environment block
Win32EnvironmentBlock eb;
if (loggedOnUser)
{
uint error = NativeMethods.CredUIParseUserName(User.Name, usernameBldr, usernameBldr.Capacity, domainBldr, domainBldr.Capacity);
if (error != 0)
{
_sessionLogger.LogError(Resources.Error_UserNameParse, User.Name, error);
throw new ArgumentException(Resources.Error_UserNameParse.FormatInvariant(User.Name, error));
}
string username = usernameBldr.ToString();
string domain = domainBldr.ToString();
eb = CreateEnvironmentBlockForUser(useridentity, username, profilePath);
}
else
{
eb = Win32EnvironmentBlock.Create((useridentity ?? WindowsIdentity.GetCurrent()).Token);
}
// add additional variables to the environment block
eb["R_HOME"] = shortHome.ToString();
_sessionLogger.LogTrace(Resources.Trace_EnvironmentVariable, "R_HOME", eb["R_HOME"]);
eb["PATH"] = Invariant($"{Interpreter.Info.BinPath};{Environment.GetEnvironmentVariable("PATH")}");
_sessionLogger.LogTrace(Resources.Trace_EnvironmentVariable, "PATH", eb["PATH"]);
_sessionLogger.LogInformation(Resources.Info_StartingRHost, Id, User.Name, commandLine);
using (Win32NativeEnvironmentBlock nativeEnv = eb.GetNativeEnvironmentBlock()) {
if (loggedOnUser)
{
_process = Win32Process.StartProcessAsUser(useridentity, rhostExePath, commandLine, Path.GetDirectoryName(rhostExePath), nativeEnv, out stdin, out stdout, out stderror);
}
else
{
_process = Win32Process.StartProcessAsUser(null, rhostExePath, commandLine, Path.GetDirectoryName(rhostExePath), nativeEnv, out stdin, out stdout, out stderror);
}
}
_process.Exited += delegate(object o, Win32ProcessExitEventArgs exitState){
_hostEnd?.Dispose();
_hostEnd = null;
State = SessionState.Terminated;
if (exitState.HasError())
{
_sessionLogger.LogInformation(Resources.Error_ExitRHost, exitState.ExitCode);
}
};
_process.WaitForExit(250);
if (_process.HasExited && _process.ExitCode < 0)
{
var message = ErrorCodeConverter.MessageFromErrorCode(_process.ExitCode);
if (!string.IsNullOrEmpty(message))
{
throw new Win32Exception(message);
}
throw new Win32Exception(_process.ExitCode);
}
_sessionLogger.LogInformation(Resources.Info_StartedRHost, Id, User.Name);
var hostEnd = _pipe.ConnectHost(_process.ProcessId);
_hostEnd = hostEnd;
ClientToHostWorker(stdin, hostEnd).DoNotWait();
HostToClientWorker(stdout, hostEnd).DoNotWait();
HostToClientErrorWorker(stderror, _process.ProcessId, (int processid, string errdata) => {
outputLogger?.LogTrace(Resources.Trace_ErrorDataReceived, processid, errdata);
}).DoNotWait();
}
public void StartHost(SecureString password, string profilePath, ILogger outputLogger, LogVerbosity verbosity)
{
if (_hostEnd != null)
{
throw new InvalidOperationException("Host process is already running");
}
string brokerPath = Path.GetDirectoryName(typeof(Program).Assembly.GetAssemblyPath());
string rhostExePath = Path.Combine(brokerPath, RHostExe);
string arguments = Invariant($"--rhost-name \"{Id}\" --rhost-log-verbosity {(int)verbosity} {CommandLineArguments}");
var username = new StringBuilder(NativeMethods.CREDUI_MAX_USERNAME_LENGTH + 1);
var domain = new StringBuilder(NativeMethods.CREDUI_MAX_PASSWORD_LENGTH + 1);
ProcessStartInfo psi = new ProcessStartInfo(rhostExePath)
{
UseShellExecute = false,
CreateNoWindow = false,
Arguments = arguments,
RedirectStandardInput = true,
RedirectStandardOutput = true,
RedirectStandardError = true,
LoadUserProfile = true
};
var useridentity = User as WindowsIdentity;
if (useridentity != null && WindowsIdentity.GetCurrent().User != useridentity.User && password != null)
{
uint error = NativeMethods.CredUIParseUserName(User.Name, username, username.Capacity, domain, domain.Capacity);
if (error != 0)
{
_sessionLogger.LogError(Resources.Error_UserNameParse, User.Name, error);
throw new ArgumentException(Resources.Error_UserNameParse.FormatInvariant(User.Name, error));
}
psi.Domain = domain.ToString();
psi.UserName = username.ToString();
psi.Password = password;
_sessionLogger.LogTrace(Resources.Trace_EnvironmentVariableCreationBegin, User.Name, profilePath);
// if broker and rhost are run as different users recreate user environment variables.
psi.EnvironmentVariables["USERNAME"] = username.ToString();
_sessionLogger.LogTrace(Resources.Trace_EnvironmentVariable, "USERNAME", psi.EnvironmentVariables["USERNAME"]);
psi.EnvironmentVariables["HOMEDRIVE"] = profilePath.Substring(0, 2);
_sessionLogger.LogTrace(Resources.Trace_EnvironmentVariable, "HOMEDRIVE", psi.EnvironmentVariables["HOMEDRIVE"]);
psi.EnvironmentVariables["HOMEPATH"] = profilePath.Substring(2);
_sessionLogger.LogTrace(Resources.Trace_EnvironmentVariable, "HOMEPATH", psi.EnvironmentVariables["HOMEPATH"]);
psi.EnvironmentVariables["USERPROFILE"] = Invariant($"{psi.EnvironmentVariables["HOMEDRIVE"]}{psi.EnvironmentVariables["HOMEPATH"]}");
_sessionLogger.LogTrace(Resources.Trace_EnvironmentVariable, "USERPROFILE", psi.EnvironmentVariables["USERPROFILE"]);
psi.EnvironmentVariables["APPDATA"] = Invariant($"{psi.EnvironmentVariables["USERPROFILE"]}\\AppData\\Roaming");
_sessionLogger.LogTrace(Resources.Trace_EnvironmentVariable, "APPDATA", psi.EnvironmentVariables["APPDATA"]);
psi.EnvironmentVariables["LOCALAPPDATA"] = Invariant($"{psi.EnvironmentVariables["USERPROFILE"]}\\AppData\\Local");
_sessionLogger.LogTrace(Resources.Trace_EnvironmentVariable, "LOCALAPPDATA", psi.EnvironmentVariables["LOCALAPPDATA"]);
psi.EnvironmentVariables["TEMP"] = Invariant($"{psi.EnvironmentVariables["LOCALAPPDATA"]}\\Temp");
_sessionLogger.LogTrace(Resources.Trace_EnvironmentVariable, "TEMP", psi.EnvironmentVariables["TEMP"]);
psi.EnvironmentVariables["TMP"] = Invariant($"{psi.EnvironmentVariables["LOCALAPPDATA"]}\\Temp");
_sessionLogger.LogTrace(Resources.Trace_EnvironmentVariable, "TMP", psi.EnvironmentVariables["TMP"]);
}
var shortHome = new StringBuilder(NativeMethods.MAX_PATH);
NativeMethods.GetShortPathName(Interpreter.Info.Path, shortHome, shortHome.Capacity);
psi.EnvironmentVariables["R_HOME"] = shortHome.ToString();
psi.EnvironmentVariables["PATH"] = Interpreter.Info.BinPath + ";" + Environment.GetEnvironmentVariable("PATH");
psi.WorkingDirectory = Path.GetDirectoryName(rhostExePath);
_process = new Process {
StartInfo = psi,
EnableRaisingEvents = true,
};
_process.ErrorDataReceived += (sender, e) => {
var process = (Process)sender;
outputLogger?.LogTrace(Resources.Trace_ErrorDataReceived, process.Id, e.Data);
};
_process.Exited += delegate {
_hostEnd?.Dispose();
_hostEnd = null;
State = SessionState.Terminated;
};
_sessionLogger.LogInformation(Resources.Info_StartingRHost, Id, User.Name, rhostExePath, arguments);
try {
StartSession();
} catch (Exception ex) {
_sessionLogger.LogError(Resources.Error_RHostFailedToStart, ex.Message);
throw;
}
_sessionLogger.LogInformation(Resources.Info_StartedRHost, Id, User.Name);
_process.BeginErrorReadLine();
var hostEnd = _pipe.ConnectHost(_process.Id);
_hostEnd = hostEnd;
ClientToHostWorker(_process.StandardInput.BaseStream, hostEnd).DoNotWait();
HostToClientWorker(_process.StandardOutput.BaseStream, hostEnd).DoNotWait();
}
