sapling/eden/scm/tests/dummyssh
Adam Simpkins ab3a7cb21f Move fb-mercurial sources into an eden/scm subdirectory.
Summary:
In preparation for merging fb-mercurial sources to the Eden repository,
move everything from the top-level directory into an `eden/scm`
subdirectory.
2019-11-13 16:04:48 -08:00

87 lines
2.2 KiB
Python
Executable File

#!/usr/bin/env python
from __future__ import absolute_import
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:
eprint('Illegal command "%s": %s\n' % (cmd, e))
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:
eprint("Illegal repo name: %s\n" % "?".join(path))
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("Got arguments")
for i, arg in enumerate(sys.argv[1:]):
log.write(" %d:%s" % (i + 1, arg))
log.write("\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))