protected override async Task <Action <AsyncCodeActivityContext> > ExecuteAsync(AsyncCodeActivityContext context, CancellationToken cancellationToken)
{
// Inputs
var command = Command.Get(context);
var checkExitCode = CheckExitCode.Get(context);
var shellStream = ShellStream.Get(context);
var shellExpectedPrompt = ShellExpectedPrompt.Get(context);
// Set a timeout on the execution
var task = ExecuteWithTimeout(context, cancellationToken);
if (await Task.WhenAny(task, Task.Delay(TimeoutMS.Get(context), cancellationToken)) != task)
{
throw new TimeoutException(Resources.Timeout_Error);
}
if (task.Exception != null)
{
throw task.Exception;
}
// Outputs
return((ctx) => {
Result.Set(ctx, task.Result);
});
}
private async Task ExecuteWithTimeout(NativeActivityContext context, CancellationToken cancellationToken = default)
{
var sshPassword = new NetworkCredential("", Password.Get(context)).Password;
var proxyPassword = new NetworkCredential("", ProxyPassword.Get(context)).Password;
var sshTimeout = TimeSpan.FromMilliseconds(SSHTimeoutMS.Get(context));
ConnectionInfo connectionInfo;
if (!string.IsNullOrEmpty(ProxyHost.Get(context))) // Proxy defined
{
if (!string.IsNullOrEmpty(ProxyUsername.Get(context))) // Proxy authentication
{
connectionInfo = new PasswordConnectionInfo(Host.Get(context), Port.Get(context), Username.Get(context), Encoding.UTF8.GetBytes(sshPassword), ProxyTypes.Http, ProxyHost.Get(context), ProxyPort.Get(context), ProxyUsername.Get(context), proxyPassword);
}
else // No proxy authentication
{
connectionInfo = new PasswordConnectionInfo(Host.Get(context), Port.Get(context), Username.Get(context), sshPassword, ProxyTypes.Http, ProxyHost.Get(context), ProxyPort.Get(context));
}
}
else // No Proxy defined
{
connectionInfo = new PasswordConnectionInfo(Host.Get(context), Port.Get(context), Username.Get(context), sshPassword);
}
connectionInfo.Timeout = sshTimeout;
sshClient = new SshClient(connectionInfo);
sshClient.Connect();
if (ShellExpectedPrompt.Expression != null)
{
var terminalMode = new Dictionary <TerminalModes, uint>
{
{ TerminalModes.ECHO, 53 }
};
expectedPromptRegex = new Regex(ShellExpectedPrompt.Get(context), RegexOptions.Compiled);
currentShellStream = sshClient.CreateShellStream("UiPathTeam.SSHConnector.Shell", 0, 0, 0, 0, 4096, terminalMode);
var welcomeMessage = SSHHelpers.Expect(currentShellStream, expectedPromptRegex, string.Empty, sshTimeout);
ShellWelcomeMessage.Set(context, welcomeMessage);
}
}
private async Task <string> ExecuteWithTimeout(AsyncCodeActivityContext context, CancellationToken cancellationToken = default)
{
ShellStream shellStream;
Regex expectedShellPromptRegex;
IObjectContainer scopeObjectContainer = context.DataContext.GetProperties()[SSHConnectScope.ParentContainerPropertyTag]
.GetValue(context.DataContext) as IObjectContainer;
if (ShellStream.Expression == null)
{
if (scopeObjectContainer == null)
{
throw new ArgumentException("ShellStream was not provided and cannot be retrieved from the container.");
}
shellStream = scopeObjectContainer.Get <ShellStream>();
}
else
{
shellStream = ShellStream.Get(context);
}
if (ShellExpectedPrompt.Expression == null)
{
if (scopeObjectContainer == null)
{
throw new ArgumentException("Shell Expected Prompt was not provided and cannot be retrieved from the container.");
}
expectedShellPromptRegex = scopeObjectContainer.Get <Regex>();
}
else
{
expectedShellPromptRegex = new Regex(ShellExpectedPrompt.Get(context), RegexOptions.Compiled);
}
return(SSHHelpers.Execute(shellStream,
expectedShellPromptRegex,
Command.Get(context),
TimeSpan.FromMilliseconds(SSHTimeoutMS.Get(context)),
CheckExitCode.Get(context)));
}
protected override async Task <Action <NativeActivityContext> > ExecuteAsync(NativeActivityContext context, CancellationToken cancellationToken)
{
// Inputs
var activityTimeout = TimeoutMS.Get(context);
var host = Host.Get(context);
var port = Port.Get(context);
var username = Username.Get(context);
var password = Password.Get(context);
var shellExpectedPrompt = ShellExpectedPrompt.Get(context);
var proxyHost = ProxyHost.Get(context);
var proxyPort = ProxyPort.Get(context);
var proxyUsername = ProxyUsername.Get(context);
var proxyPassword = ProxyPassword.Get(context);
// Set a timeout on the execution
var task = ExecuteWithTimeout(context, cancellationToken);
if (await Task.WhenAny(task, Task.Delay(activityTimeout, cancellationToken)) != task)
{
throw new TimeoutException(Resources.Timeout_Error);
}
if (task.Exception != null)
{
throw task.Exception;
}
_objectContainer.Add(sshClient);
_objectContainer.Add(currentShellStream);
_objectContainer.Add(expectedPromptRegex);
return((ctx) => {
// Schedule child activities
if (Body != null)
{
ctx.ScheduleAction <IObjectContainer>(Body, _objectContainer, OnCompleted, OnFaulted);
}
});
}