165 lines
8.6 KiB
Python
Executable File
165 lines
8.6 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
"""Exercise real Docker, SSH, Git, HTTP authorization, database outage, and restart recovery.
|
|
Uses an isolated Compose project and removes only that project's containers/volumes.
|
|
Run prepare-deploy.py first. Ports 18080, 18443, 12222 must be free.
|
|
"""
|
|
import argparse
|
|
import http.cookiejar
|
|
import json
|
|
import os
|
|
from pathlib import Path
|
|
import secrets
|
|
import subprocess
|
|
import tempfile
|
|
import time
|
|
import urllib.error
|
|
import urllib.request
|
|
|
|
ROOT = Path(__file__).resolve().parents[1]
|
|
ORIGIN = 'http://localhost:18080'
|
|
parser = argparse.ArgumentParser(description=__doc__)
|
|
parser.add_argument('--key-type', choices=['ed25519', 'rsa'], default='ed25519')
|
|
args = parser.parse_args()
|
|
|
|
|
|
def run(args, **kwargs):
|
|
return subprocess.run(args, check=True, text=True, **kwargs)
|
|
|
|
|
|
def wait_for(check, timeout=90):
|
|
end = time.monotonic() + timeout
|
|
last = None
|
|
while time.monotonic() < end:
|
|
try:
|
|
value = check()
|
|
if value:
|
|
return value
|
|
except (OSError, ValueError, urllib.error.HTTPError, AssertionError) as error:
|
|
last = error
|
|
time.sleep(1)
|
|
raise AssertionError(f'Timed out: {last}')
|
|
|
|
|
|
class Client:
|
|
def __init__(self):
|
|
self.opener = urllib.request.build_opener(urllib.request.HTTPCookieProcessor(http.cookiejar.CookieJar()))
|
|
self.csrf = None
|
|
|
|
def call(self, method, path, data=None, expected=200):
|
|
headers = {'Origin': ORIGIN, 'Content-Type': 'application/json'}
|
|
if self.csrf:
|
|
headers['X-CSRF-Token'] = self.csrf
|
|
req = urllib.request.Request(ORIGIN + path, method=method, headers=headers,
|
|
data=json.dumps(data).encode() if data is not None else None)
|
|
try:
|
|
response = self.opener.open(req, timeout=10)
|
|
except urllib.error.HTTPError as error:
|
|
assert error.code == expected, (path, error.code, error.read())
|
|
return None
|
|
assert response.status == expected, (path, response.status)
|
|
body = response.read()
|
|
return json.loads(body) if body else None
|
|
|
|
|
|
with tempfile.TemporaryDirectory(prefix='chublog-smoke-') as temporary:
|
|
tmp = Path(temporary)
|
|
password = secrets.token_urlsafe(24)
|
|
hashed = run([str(ROOT / 'deploy/bin/chublog-api'), 'hash-password'], input=password, capture_output=True).stdout.strip()
|
|
envfile = tmp / '.env'
|
|
envfile.write_text(f"PUBLIC_ORIGIN={ORIGIN}\nSITE_ADDRESS=http://localhost:80\nADMIN_PASSWORD_HASH='{hashed}'\n"
|
|
f"CHUSQL_ROOT_PASSWORD={secrets.token_urlsafe(24)}\nCHUSQL_PASSWORD={secrets.token_urlsafe(24)}\n"
|
|
'DEBIAN_MIRROR=mirrors.ustc.edu.cn\nALLOW_REGISTRATION=true\nHTTP_BIND=127.0.0.1\nHTTP_PORT=18080\nHTTPS_PORT=18443\nSSH_BIND=127.0.0.1\nSSH_PORT=12222\n')
|
|
envfile.chmod(0o600)
|
|
project = 'chublog-test-' + secrets.token_hex(4)
|
|
compose = ['docker', 'compose', '--env-file', str(envfile), '-p', project, '-f', str(ROOT / 'deploy/compose.yaml')]
|
|
try:
|
|
run(compose + ['up', '-d', '--build'])
|
|
admin = Client()
|
|
wait_for(lambda: admin.call('GET', '/api/health'))
|
|
wait_for(lambda: subprocess.run(compose + ['run', '--rm', 'db-init'], capture_output=True).returncode == 0)
|
|
admin.csrf = admin.call('POST', '/api/auth/login', {'username': 'admin', 'password': password})['csrf']
|
|
publisher = Client()
|
|
publisher.call('POST', '/api/auth/register', {'username': 'alice', 'password': password})
|
|
publisher.csrf = publisher.call('POST', '/api/auth/login', {'username': 'alice', 'password': password})['csrf']
|
|
run(['ssh-keygen', '-q', '-t', args.key_type, '-b', '3072', '-N', '', '-f', str(tmp / 'key')])
|
|
publisher.call('PUT', '/api/account/ssh-key', {'public_key': (tmp / 'key.pub').read_text().strip()})
|
|
publisher.call('GET', '/api/admin/users', expected=403)
|
|
env = os.environ.copy()
|
|
env['GIT_SSH_COMMAND'] = f'ssh -i {tmp / "key"} -o IdentitiesOnly=yes -o StrictHostKeyChecking=accept-new -o UserKnownHostsFile={tmp / "known_hosts"} -p 12222'
|
|
local = tmp / 'articles'
|
|
run(['git', 'init', '--initial-branch=main', str(local)], capture_output=True)
|
|
run(['git', '-C', str(local), 'config', 'user.name', 'Alice'])
|
|
run(['git', '-C', str(local), 'config', 'user.email', 'alice@example.invalid'])
|
|
run(['git', '-C', str(local), 'remote', 'add', 'origin', 'git@localhost:content.git'])
|
|
article = local / 'posts/alice-first/index.md'
|
|
article.parent.mkdir(parents=True)
|
|
original = (ROOT / 'content/posts/welcome/index.md').read_text().replace('slug: "welcome"', 'slug: "alice-first"')
|
|
article.write_text(original)
|
|
|
|
def commit():
|
|
run(['git', '-C', str(local), 'add', '.'])
|
|
run(['git', '-C', str(local), 'commit', '-m', 'Update article'], capture_output=True)
|
|
|
|
def push(success=True, *extra):
|
|
result = subprocess.run(['git', '-C', str(local), 'push', 'origin', *extra] if extra else ['git', '-C', str(local), 'push', '-u', 'origin', 'main'], env=env, capture_output=True, text=True)
|
|
assert (result.returncode == 0) == success, result.stderr
|
|
return result
|
|
|
|
commit()
|
|
push(False)
|
|
admin.call('PATCH', '/api/admin/users/alice', {'enabled': True, 'role': 'publisher', 'prefixes': ['alice-']})
|
|
push()
|
|
wait_for(lambda: Client().call('GET', '/api/posts/alice-first')['meta']['slug'] == 'alice-first')
|
|
wait_for(lambda: (lambda s: s['indexed_commit'] == s['commit'])(admin.call('GET', '/api/admin/status')), timeout=120)
|
|
print('PASS: registration, independent grant, real SSH push, publication, real ChuSQL index')
|
|
ssh = ['ssh', '-i', str(tmp / 'key'), '-o', 'IdentitiesOnly=yes', '-o', 'StrictHostKeyChecking=accept-new', '-o', f'UserKnownHostsFile={tmp / "known_hosts"}', '-p', '12222', 'git@localhost']
|
|
result = subprocess.run(ssh + ['id'], capture_output=True)
|
|
assert result.returncode != 0, 'SSH shell command unexpectedly succeeded'
|
|
# Invalid commits must not advance the remote branch. Reset only this disposable local test clone.
|
|
for kind in ['script', 'symlink', 'foreign', 'oversize', 'pack']:
|
|
if kind == 'script':
|
|
article.write_text(original + '\n<script>alert(1)</script>\n')
|
|
elif kind == 'symlink':
|
|
article.unlink()
|
|
article.symlink_to('/etc/passwd')
|
|
elif kind == 'foreign':
|
|
other = local / 'posts/bob-first/index.md'
|
|
other.parent.mkdir(parents=True)
|
|
other.write_text(original.replace('alice-first', 'bob-first'))
|
|
elif kind == 'oversize':
|
|
article.write_text(original + 'x' * (257 * 1024))
|
|
else:
|
|
(local / 'oversized.bin').write_bytes(os.urandom(9 * 1024 * 1024))
|
|
commit()
|
|
push(False)
|
|
run(['git', '-C', str(local), 'reset', '--hard', 'origin/main'], capture_output=True)
|
|
push(False, 'main:refs/heads/other')
|
|
push(False, '--delete', 'main')
|
|
print('PASS: arbitrary command, HTML, symlink, other-author edit, oversized article, extra branch and branch deletion rejected')
|
|
article.write_text(original.replace('status: "published"', 'status: "draft"'))
|
|
commit()
|
|
push()
|
|
wait_for(lambda: Client().call('GET', '/api/stats')['published_posts'] == 0)
|
|
Client().call('GET', '/api/posts/alice-first', expected=404)
|
|
article.write_text(original)
|
|
commit()
|
|
push()
|
|
wait_for(lambda: Client().call('GET', '/api/stats')['published_posts'] == 1)
|
|
run(compose + ['stop', 'db'])
|
|
assert Client().call('GET', '/api/posts/alice-first')['meta']['slug'] == 'alice-first'
|
|
run(compose + ['restart', 'api'])
|
|
wait_for(lambda: Client().call('GET', '/api/posts/alice-first')['meta']['slug'] == 'alice-first')
|
|
admin.csrf = admin.call('POST', '/api/auth/login', {'username': 'admin', 'password': password})['csrf']
|
|
admin.call('PATCH', '/api/admin/users/alice', {'enabled': True, 'role': 'reader', 'prefixes': []})
|
|
article.write_text(original + '\nRevoked publisher\n')
|
|
commit()
|
|
push(False)
|
|
run(compose + ['start', 'db'])
|
|
wait_for(lambda: (lambda s: s['indexed_commit'] == s['commit'])(admin.call('GET', '/api/admin/status')), timeout=120)
|
|
print('PASS: draft privacy, database outage, API restart recovery, revocation, automatic index recovery')
|
|
except Exception:
|
|
run(compose + ['logs', '--tail', '40'])
|
|
raise
|
|
finally:
|
|
run(compose + ['down', '--volumes', '--remove-orphans'])
|