winrm.py
87 lines
| 2.0 KiB
| text/x-python
|
PythonLexer
|
r42191 | # winrm.py - Interact with Windows Remote Management (WinRM) | ||
# | ||||
# Copyright 2019 Gregory Szorc <gregory.szorc@gmail.com> | ||||
# | ||||
# This software may be used and distributed according to the terms of the | ||||
# GNU General Public License version 2 or any later version. | ||||
# no-check-code because Python 3 native. | ||||
import logging | ||||
import pprint | ||||
import time | ||||
|
r43346 | from pypsrp.client import Client | ||
|
r42191 | from pypsrp.powershell import ( | ||
PowerShell, | ||||
PSInvocationState, | ||||
RunspacePool, | ||||
) | ||||
import requests.exceptions | ||||
logger = logging.getLogger(__name__) | ||||
|
r42465 | def wait_for_winrm(host, username, password, timeout=180, ssl=False): | ||
|
r42191 | """Wait for the Windows Remoting (WinRM) service to become available. | ||
Returns a ``psrpclient.Client`` instance. | ||||
""" | ||||
end_time = time.time() + timeout | ||||
while True: | ||||
try: | ||||
|
r43346 | client = Client( | ||
host, | ||||
username=username, | ||||
password=password, | ||||
ssl=ssl, | ||||
connection_timeout=5, | ||||
) | ||||
|
r42885 | client.execute_ps("Write-Host 'Hello, World!'") | ||
|
r42191 | return client | ||
except requests.exceptions.ConnectionError: | ||||
if time.time() >= end_time: | ||||
raise | ||||
time.sleep(1) | ||||
def format_object(o): | ||||
if isinstance(o, str): | ||||
return o | ||||
try: | ||||
o = str(o) | ||||
except TypeError: | ||||
o = pprint.pformat(o.extended_properties) | ||||
return o | ||||
def run_powershell(client, script): | ||||
with RunspacePool(client.wsman) as pool: | ||||
ps = PowerShell(pool) | ||||
ps.add_script(script) | ||||
ps.begin_invoke() | ||||
while ps.state == PSInvocationState.RUNNING: | ||||
ps.poll_invoke() | ||||
for o in ps.output: | ||||
print(format_object(o)) | ||||
ps.output[:] = [] | ||||
ps.end_invoke() | ||||
for o in ps.output: | ||||
print(format_object(o)) | ||||
if ps.state == PSInvocationState.FAILED: | ||||
|
r43346 | raise Exception( | ||
'PowerShell execution failed: %s' | ||||
% ' '.join(map(format_object, ps.streams.error)) | ||||
) | ||||