@ -3,6 +3,7 @@
$process_util = @ "
$process_util = @ "
using System ;
using System ;
using System . Collections ;
using System . IO ;
using System . IO ;
using System . Linq ;
using System . Linq ;
using System . Runtime . InteropServices ;
using System . Runtime . InteropServices ;
@ -210,7 +211,14 @@ namespace Ansible
return sbOut . ToString ( ) ;
return sbOut . ToString ( ) ;
}
}
public static Tuple < string , string , uint > RunCommand ( string lpApplicationName , string lpCommandLine , string lpCurrentDirectory , string stdinInput , string environmentBlock )
public class CommandResult
{
public string StandardOut { get ; internal set ; }
public string StandardError { get ; internal set ; }
public uint ExitCode { get ; internal set ; }
}
public static CommandResult RunCommand ( string lpApplicationName , string lpCommandLine , string lpCurrentDirectory , string stdinInput , IDictionary environment )
{
{
UInt32 startup_flags = CREATE_UNICODE_ENVIRONMENT | CREATE_NEW_CONSOLE | EXTENDED_STARTUPINFO_PRESENT ;
UInt32 startup_flags = CREATE_UNICODE_ENVIRONMENT | CREATE_NEW_CONSOLE | EXTENDED_STARTUPINFO_PRESENT ;
STARTUPINFOEX si = new STARTUPINFOEX ( ) ;
STARTUPINFOEX si = new STARTUPINFOEX ( ) ;
@ -278,10 +286,20 @@ namespace Ansible
if ( lpCurrentDirectory = = " " )
if ( lpCurrentDirectory = = " " )
lpCurrentDirectory = null ;
lpCurrentDirectory = null ;
StringBuilder environmentString = null ;
if ( environment ! = null & & environment . Count > 0 )
{
environmentString = new StringBuilder ( ) ;
foreach ( DictionaryEntry kv in environment )
environmentString . AppendFormat ( " {0}={1}\0 " , kv . Key , kv . Value ) ;
environmentString . Append ( '\0' ) ;
}
/ / Create the environment block if set
/ / Create the environment block if set
IntPtr lpEnvironment = IntPtr . Zero ;
IntPtr lpEnvironment = IntPtr . Zero ;
if ( environmentBlock ! = " " )
if ( environment String ! = null )
lpEnvironment = Marshal . StringToHGlobalUni ( environmentBlock ) ;
lpEnvironment = Marshal . StringToHGlobalUni ( environment String. ToString ( ) ) ;
/ / Create new process and run
/ / Create new process and run
StringBuilder argument_string = new StringBuilder ( lpCommandLine ) ;
StringBuilder argument_string = new StringBuilder ( lpCommandLine ) ;
@ -316,7 +334,12 @@ namespace Ansible
GetProcessOutput ( stdout , stderr , out stdout_str , out stderr_str ) ;
GetProcessOutput ( stdout , stderr , out stdout_str , out stderr_str ) ;
uint rc = GetProcessExitCode ( pi . hProcess ) ;
uint rc = GetProcessExitCode ( pi . hProcess ) ;
return Tuple . Create ( stdout_str , stderr_str , rc ) ;
return new CommandResult
{
StandardOut = stdout_str ,
StandardError = stderr_str ,
ExitCode = rc
} ;
}
}
private static void GetProcessOutput ( StreamReader stdoutStream , StreamReader stderrStream , out string stdout , out string stderr )
private static void GetProcessOutput ( StreamReader stdoutStream , StreamReader stderrStream , out string stdout , out string stderr )
@ -361,7 +384,7 @@ Function Load-CommandUtils {
# [Ansible.CommandUtil]::RunCommand(string lpApplicationName, string lpCommandLine, string lpCurrentDirectory, string stdinInput, string environmentBlock)
# [Ansible.CommandUtil]::RunCommand(string lpApplicationName, string lpCommandLine, string lpCurrentDirectory, string stdinInput, string environmentBlock)
#
#
# there are also numerous P/Invoke methods that can be called if you are feeling adventurous
# there are also numerous P/Invoke methods that can be called if you are feeling adventurous
Add-Type -TypeDefinition $process_util -IgnoreWarnings
Add-Type -TypeDefinition $process_util -IgnoreWarnings -Debug: $false
}
}
Function Get-ExecutablePath($executable , $directory ) {
Function Get-ExecutablePath($executable , $directory ) {
@ -412,29 +435,14 @@ Function Run-Command {
$arguments = [ Ansible.CommandUtil ] :: ParseCommandLine ( $command )
$arguments = [ Ansible.CommandUtil ] :: ParseCommandLine ( $command )
$executable = Get-ExecutablePath -executable $arguments [ 0 ] -directory $working_directory
$executable = Get-ExecutablePath -executable $arguments [ 0 ] -directory $working_directory
# set the extra environment variables
$environment_string = $null
if ( $environment . Count -gt 0 ) {
$environment_string = " "
}
foreach ( $environment_entry in $environment . GetEnumerator ( ) ) {
$environment_key = $environment_entry . Name
$environment_value = $environment_entry . Value
$environment_string + = " $environment_key = $environment_value `0 "
}
if ( $environment_string ) {
$environment_string + = " `0 "
}
# run the command and get the results
# run the command and get the results
$command_result = [ Ansible.CommandUtil ] :: RunCommand ( $executable , $command , $working_directory , $stdin , $environment _string )
$command_result = [ Ansible.CommandUtil ] :: RunCommand ( $executable , $command , $working_directory , $stdin , $environment )
# RunCommand returns a tuple, we will convert to a hashtable
return , @ {
return , @ {
executable = $executable
executable = $executable
stdout = $command_result . Item1
stdout = $command_result . StandardOut
stderr = $command_result . Item2
stderr = $command_result . StandardError
rc = $command_result . Item3
rc = $command_result . ExitCode
}
}
}
}