mirror of
https://github.com/facebook/sapling.git
synced 2024-10-06 14:58:03 +03:00
f2690f21eb
Summary: This makes the `dummyssh` script Python 3 compatible so we can run it with Python 3 in tests Reviewed By: sfilipco Differential Revision: D19587078 fbshipit-source-id: 134b01f4d5e968cd9600c9358c7230a56e11f163
87 lines
2.3 KiB
Python
Executable File
87 lines
2.3 KiB
Python
Executable File
#!/usr/bin/env python
|
|
|
|
from __future__ import absolute_import, print_function
|
|
|
|
import shlex
|
|
import subprocess
|
|
import os
|
|
import sys
|
|
|
|
os.chdir(os.getenv('TESTTMP'))
|
|
|
|
def parse(cmd):
|
|
"""
|
|
matches hg-ssh-wrapper
|
|
"""
|
|
try:
|
|
return shlex.split(cmd)
|
|
except ValueError as e:
|
|
print('Illegal command "%s": %s\n' % (cmd, e), file=sys.stderr)
|
|
sys.exit(255)
|
|
|
|
def parse_repo_path(path):
|
|
"""
|
|
matches hg-ssh-wrapper
|
|
"""
|
|
path = path.split("?")
|
|
if len(path) == 1:
|
|
repo = path[0]
|
|
marker = None
|
|
elif len(path) == 2:
|
|
repo = path[0]
|
|
marker = path[1]
|
|
else:
|
|
print("Illegal repo name: %s\n" % "?".join(path), file=sys.stderr)
|
|
sys.exit(255)
|
|
|
|
return repo, marker
|
|
|
|
# Skipping SSH options
|
|
host_index = 1
|
|
while host_index < len(sys.argv) and sys.argv[host_index].startswith('-'):
|
|
host_index += 1
|
|
|
|
if sys.argv[host_index] != "user@dummy":
|
|
sys.exit(-1)
|
|
|
|
os.environ["SSH_CLIENT"] = "%s 1 2" % os.environ.get('LOCALIP', '127.0.0.1')
|
|
|
|
log = open("dummylog", "ab")
|
|
log.write(b"Got arguments")
|
|
for i, arg in enumerate(sys.argv[1:]):
|
|
log.write(b" %d:%s" % (i + 1, arg.encode("utf-8")))
|
|
log.write(b"\n")
|
|
log.close()
|
|
hgcmd = sys.argv[host_index + 1]
|
|
if os.name == 'nt':
|
|
# hack to make simple unix single quote quoting work on windows
|
|
hgcmd = hgcmd.replace("'", '"')
|
|
|
|
cmdargv = parse(hgcmd)
|
|
if cmdargv[:2] == ["hg", "-R"] and cmdargv[3:] == ["serve", "--stdio"]:
|
|
path, marker = parse_repo_path(cmdargv[2])
|
|
if marker == "read_copy":
|
|
path = path + "_copy"
|
|
cmdargv[2] = path
|
|
hgcmd = subprocess.list2cmdline(cmdargv)
|
|
|
|
if "hgcli" in hgcmd:
|
|
certdir = os.environ.get("HGTEST_CERTDIR") or os.environ.get("TEST_CERTS")
|
|
if certdir is None:
|
|
raise ValueError("No cert dir")
|
|
cert = os.path.join(certdir, "localhost.crt")
|
|
capem = os.path.join(certdir, "root-ca.crt")
|
|
privatekey = os.path.join(certdir, "localhost.key")
|
|
|
|
hgcmd += (
|
|
" --mononoke-path [::1]:"
|
|
+ os.getenv("MONONOKE_SOCKET")
|
|
+ (" --cert %s --ca-pem %s --private-key %s --common-name localhost" % (cert, capem, privatekey))
|
|
)
|
|
|
|
mock_username = os.environ.get("MOCK_USERNAME")
|
|
hgcmd += " --mock-username '{}'".format(mock_username)
|
|
|
|
r = os.system(hgcmd)
|
|
sys.exit(bool(r))
|