upstream/mercurial-mirror Files · contrib/automation/hgautomation/winrm.py

util: avoid echoing the password to the console on Windows py3 (issue6446)...

util: avoid echoing the password to the console on Windows py3 (issue6446) The `getpass.getpass()` implementation on Windows first checks if `sys.stdin` and `sys.__stdin__` are the same object. It's not on py3 because the former is replaced in dispatch.py with something that doesn't normalize '\n' to '\r\n'. When they aren't the same object, it simply calls `sys.stdin.readline()` instead of the mscvrt functions that read the input characters before they are echoed. This simply copies the `getpass.win_getpass()` implementation without the stdin check, and byteifies around the edges. I'm not sure if there's a reasonable replacement for the check that we could implement. When echoing input into the hg command, the `ui.interactive()` check causes `ui.getpass()` to bail before getting here. If the proper config switches are used to bypass that and call this, the process stalls until '\n' is input into the console. So there could be a deadlock here when run by another command if the wrong config settings are applied. Differential Revision: https://phab.mercurial-scm.org/D10708

Gregory Szorc - - Load All Authors

File last commit:

r43487:6a350194 default


                r47949:5b351317

stable

Download file

             winrm.py
        
                    87 lines
            
             | 2.0 KiB
            
                | text/x-python
            
             |
                PythonLexer
            
             / contrib / automation / hgautomation / winrm.py
          
                    History
                
                 |
                  Annotation
                 | Raw
                 |Copy content
                 |Copy permalink

      # 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

      from pypsrp.client import Client

      from pypsrp.powershell import (

          PowerShell,

          PSInvocationState,

          RunspacePool,

      )

      import requests.exceptions

      logger = logging.getLogger(__name__)

      def wait_for_winrm(host, username, password, timeout=180, ssl=False):

          """Wait for the Windows Remoting (WinRM) service to become available.

          Returns a ``psrpclient.Client`` instance.

          """

          end_time = time.time() + timeout

          while True:

              try:

                  client = Client(

                      host,

                      username=username,

                      password=password,

                      ssl=ssl,

                      connection_timeout=5,

                  )

                  client.execute_ps("Write-Host 'Hello, World!'")

                  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 (AttributeError, 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:

                  raise Exception(

                      'PowerShell execution failed: %s'

                      % ' '.join(map(format_object, ps.streams.error))

                  )

	Site-wide shortcuts
/	Use quick search box
g h	Goto home page
g g	Goto my private gists page
g G	Goto my public gists page
g 0-9	Goto bookmarked items from 0-9
n r	New repository page
n g	New gist page

	Repositories
g s	Goto summary page
g c	Goto changelog page
g f	Goto files page
g F	Goto files page with file search activated
g p	Goto pull requests page
g o	Goto repository settings
g O	Goto repository access permissions settings
t s	Toggle sidebar on some pages

				# 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

				from pypsrp.client import Client
				from pypsrp.powershell import (
				PowerShell,
				PSInvocationState,
				RunspacePool,
				)
				import requests.exceptions


				logger = logging.getLogger(__name__)


				def wait_for_winrm(host, username, password, timeout=180, ssl=False):
				"""Wait for the Windows Remoting (WinRM) service to become available.

				Returns a ``psrpclient.Client`` instance.
				"""

				end_time = time.time() + timeout

				while True:
				try:
				client = Client(
				host,
				username=username,
				password=password,
				ssl=ssl,
				connection_timeout=5,
				)
				client.execute_ps("Write-Host 'Hello, World!'")
				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 (AttributeError, 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:
				raise Exception(
				'PowerShell execution failed: %s'
				% ' '.join(map(format_object, ps.streams.error))
				)