131 lines
4.4 KiB
PowerShell
131 lines
4.4 KiB
PowerShell
#!powershell
|
|
# This file is part of Ansible
|
|
#
|
|
# Ansible 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.
|
|
#
|
|
# Ansible 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 Ansible. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
# WANT_JSON
|
|
# POWERSHELL_COMMON
|
|
|
|
# TODO: add check mode support
|
|
|
|
Set-StrictMode -Version 2
|
|
$ErrorActionPreference = "Stop"
|
|
|
|
$parsed_args = Parse-Args $args $false
|
|
|
|
$raw_command_line = $(Get-AnsibleParam $parsed_args "_raw_params" -failifempty $true).Trim()
|
|
$chdir = Get-AnsibleParam $parsed_args "chdir"
|
|
$creates = Get-AnsibleParam $parsed_args "creates"
|
|
$removes = Get-AnsibleParam $parsed_args "removes"
|
|
|
|
$result = @{changed=$true; warnings=@(); cmd=$raw_command_line}
|
|
|
|
If($creates -and $(Test-Path $creates)) {
|
|
Exit-Json @{cmd=$raw_command_line; msg="skipped, since $creates exists"; changed=$false; skipped=$true; rc=0}
|
|
}
|
|
|
|
If($removes -and -not $(Test-Path $removes)) {
|
|
Exit-Json @{cmd=$raw_command_line; msg="skipped, since $removes does not exist"; changed=$false; skipped=$true; rc=0}
|
|
}
|
|
|
|
$util_def = @'
|
|
using System;
|
|
using System.ComponentModel;
|
|
using System.Linq;
|
|
using System.Runtime.InteropServices;
|
|
|
|
namespace Ansible.Command
|
|
{
|
|
public static class NativeUtil
|
|
{
|
|
[DllImport("shell32.dll", SetLastError = true)]
|
|
static extern IntPtr CommandLineToArgvW([MarshalAs(UnmanagedType.LPWStr)] string lpCmdLine, out int pNumArgs);
|
|
|
|
public static string[] ParseCommandLine(string cmdline)
|
|
{
|
|
int numArgs;
|
|
IntPtr ret = CommandLineToArgvW(cmdline, out numArgs);
|
|
|
|
if (ret == IntPtr.Zero)
|
|
throw new Exception(String.Format("Error parsing command line: {0}", new Win32Exception(Marshal.GetLastWin32Error()).Message));
|
|
|
|
IntPtr[] strptrs = new IntPtr[numArgs];
|
|
Marshal.Copy(ret, strptrs, 0, numArgs);
|
|
string[] cmdlineParts = strptrs.Select(s=>Marshal.PtrToStringUni(s)).ToArray();
|
|
|
|
Marshal.FreeHGlobal(ret);
|
|
|
|
return cmdlineParts;
|
|
}
|
|
}
|
|
}
|
|
'@
|
|
|
|
$util_type = Add-Type -TypeDefinition $util_def
|
|
|
|
# FUTURE: extract this code to separate module_utils as Windows module API version of run_command
|
|
|
|
$exec_args = $null
|
|
|
|
# Parse the command-line with the Win32 parser to get the application name to run. The Win32 parser
|
|
# will deal with quoting/escaping for us...
|
|
# FUTURE: no longer necessary once we switch to raw Win32 CreateProcess
|
|
$parsed_command_line = [Ansible.Command.NativeUtil]::ParseCommandLine($raw_command_line);
|
|
$exec_application = $parsed_command_line[0]
|
|
If($parsed_command_line.Length -gt 1) {
|
|
# lop the application off, then rejoin the args as a single string
|
|
$exec_args = $parsed_command_line[1..$($parsed_command_line.Length-1)] -join " "
|
|
}
|
|
|
|
$proc = New-Object System.Diagnostics.Process
|
|
$psi = $proc.StartInfo
|
|
$psi.FileName = $exec_application
|
|
$psi.Arguments = $exec_args
|
|
$psi.RedirectStandardOutput = $true
|
|
$psi.RedirectStandardError = $true
|
|
$psi.UseShellExecute = $false
|
|
|
|
If ($chdir) {
|
|
$psi.WorkingDirectory = $chdir
|
|
}
|
|
|
|
$start_datetime = [DateTime]::UtcNow
|
|
|
|
Try {
|
|
$proc.Start() | Out-Null # will always return $true for non shell-exec cases
|
|
}
|
|
Catch [System.ComponentModel.Win32Exception] {
|
|
# fail nicely for "normal" error conditions
|
|
# FUTURE: this probably won't work on Nano Server
|
|
$excep = $_
|
|
Exit-Json @{failed=$true;changed=$false;cmd=$raw_command_line;rc=$excep.Exception.NativeErrorCode;msg=$excep.Exception.Message}
|
|
}
|
|
|
|
# TODO: resolve potential deadlock here if stderr fills buffer (~4k) before stdout is closed,
|
|
# perhaps some async stream pumping with Process Output/ErrorDataReceived events...
|
|
|
|
$result.stdout = $proc.StandardOutput.ReadToEnd()
|
|
$result.stderr = $proc.StandardError.ReadToEnd()
|
|
|
|
$proc.WaitForExit() | Out-Null
|
|
|
|
$result.rc = $proc.ExitCode
|
|
|
|
$end_datetime = [DateTime]::UtcNow
|
|
|
|
$result.start = $start_datetime.ToString("yyyy-MM-dd hh:mm:ss.ffffff")
|
|
$result.end = $end_datetime.ToString("yyyy-MM-dd hh:mm:ss.ffffff")
|
|
$result.delta = $($end_datetime - $start_datetime).ToString("h\:mm\:ss\.ffffff")
|
|
|
|
ConvertTo-Json -Depth 99 $result
|