119 lines
5.9 KiB
Python
119 lines
5.9 KiB
Python
#!/usr/bin/env python3
|
|
"""v0.322.0 burndown-2 browser check: the Data tab's export key field and the 2nd-copy part. Minimal CDP client: log in to the controller through its PUBLIC address in headless Chrome, open each
|
|
app-page tab, assert computed styles, collect console errors, save screenshots. The password is read
|
|
from a 0600 file and only ever sent inside a CDP message; it is never printed."""
|
|
import base64, json, os, socket, struct, subprocess, sys, time, urllib.request
|
|
|
|
CHROME = os.path.expanduser('~/.cache/ms-playwright/chromium_headless_shell-1243/chrome-headless-shell-linux64/chrome-headless-shell')
|
|
BASE = sys.argv[1] # https://felhom.enkisfelhom.hu
|
|
PWFILE = sys.argv[2] # the controller password (0600 file)
|
|
FBPWFILE = sys.argv[3] # FileBrowser's admin password (0600 file) — for the cold „Fajlok" click
|
|
OUT = sys.argv[4]
|
|
PORT = 9333
|
|
os.makedirs(OUT, exist_ok=True)
|
|
prof = os.path.join(OUT, '.profile')
|
|
proc = subprocess.Popen([CHROME, f'--remote-debugging-port={PORT}', '--no-sandbox', f'--user-data-dir={prof}',
|
|
'--window-size=1280,900', 'about:blank'], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
|
|
try:
|
|
for _ in range(50):
|
|
try:
|
|
tabs = json.load(urllib.request.urlopen(f'http://127.0.0.1:{PORT}/json'))
|
|
break
|
|
except Exception:
|
|
time.sleep(0.2)
|
|
ws_url = [t for t in tabs if t['type'] == 'page'][0]['webSocketDebuggerUrl']
|
|
host, path = ws_url[len('ws://'):].split('/', 1)
|
|
h, p = host.split(':')
|
|
sock = socket.create_connection((h, int(p)))
|
|
key = base64.b64encode(os.urandom(16)).decode()
|
|
sock.send(f'GET /{path} HTTP/1.1\r\nHost: {host}\r\nUpgrade: websocket\r\nConnection: Upgrade\r\nSec-WebSocket-Key: {key}\r\nSec-WebSocket-Version: 13\r\n\r\n'.encode())
|
|
buf = b''
|
|
while b'\r\n\r\n' not in buf:
|
|
buf += sock.recv(4096)
|
|
st = {'pending': buf.split(b'\r\n\r\n', 1)[1]}
|
|
|
|
def send(obj):
|
|
data = json.dumps(obj).encode()
|
|
hdr = bytearray([0x81])
|
|
n = len(data)
|
|
if n < 126: hdr.append(0x80 | n)
|
|
elif n < 65536: hdr.append(0x80 | 126); hdr += struct.pack('>H', n)
|
|
else: hdr.append(0x80 | 127); hdr += struct.pack('>Q', n)
|
|
mask = os.urandom(4); hdr += mask
|
|
sock.sendall(bytes(hdr) + bytes(b ^ mask[i % 4] for i, b in enumerate(data)))
|
|
|
|
def recv_exact(n):
|
|
while len(st['pending']) < n:
|
|
st['pending'] += sock.recv(1 << 20)
|
|
out, st['pending'] = st['pending'][:n], st['pending'][n:]
|
|
return out
|
|
|
|
def recv():
|
|
msg = b''
|
|
while True:
|
|
b0, b1 = recv_exact(2)
|
|
n = b1 & 0x7f
|
|
if n == 126: n = struct.unpack('>H', recv_exact(2))[0]
|
|
elif n == 127: n = struct.unpack('>Q', recv_exact(8))[0]
|
|
msg += recv_exact(n)
|
|
if b0 & 0x80: return json.loads(msg)
|
|
|
|
console = []
|
|
mid = [0]
|
|
def call(method, **params):
|
|
mid[0] += 1; i = mid[0]
|
|
send({'id': i, 'method': method, 'params': params})
|
|
while True:
|
|
m = recv()
|
|
if m.get('method') == 'Runtime.exceptionThrown':
|
|
console.append(('exception', m['params']['exceptionDetails'].get('text', ''), m['params']['exceptionDetails'].get('exception', {}).get('description', '')[:200]))
|
|
elif m.get('method') == 'Runtime.consoleAPICalled' and m['params']['type'] in ('error', 'warning'):
|
|
console.append((m['params']['type'], ' '.join(str(a.get('value', a.get('description', ''))) for a in m['params']['args'])[:200]))
|
|
elif m.get('method') == 'Log.entryAdded' and m['params']['entry']['level'] in ('error', 'warning'):
|
|
e = m['params']['entry']; console.append(('log-' + e['level'], e.get('text', '')[:200], e.get('url', '')))
|
|
if m.get('id') == i:
|
|
if 'error' in m: raise RuntimeError(f'{method}: {m["error"]}')
|
|
return m.get('result', {})
|
|
|
|
def ev(expr):
|
|
r = call('Runtime.evaluate', expression=expr, returnByValue=True, awaitPromise=True)
|
|
return r.get('result', {}).get('value')
|
|
|
|
def goto(url):
|
|
call('Page.navigate', url=url)
|
|
for _ in range(100):
|
|
time.sleep(0.2)
|
|
if ev('document.readyState') == 'complete':
|
|
break
|
|
time.sleep(1.5) # images' onerror, layout
|
|
|
|
for d in ('Page', 'Runtime', 'Log', 'Network'):
|
|
call(d + '.enable')
|
|
pw = open(PWFILE).read().strip()
|
|
goto(BASE + '/login')
|
|
ev("(function(){var f=document.querySelector('input[type=password]'); f.value=%s; f.form.submit(); return 1})()" % json.dumps(pw))
|
|
del pw
|
|
time.sleep(3)
|
|
print('after login url:', ev('location.pathname'))
|
|
console.clear() # the login page is not under test
|
|
|
|
def shot(name):
|
|
s = call('Page.captureScreenshot', format='png', captureBeyondViewport=True)
|
|
open(os.path.join(OUT, name + '.png'), 'wb').write(base64.b64decode(s['data']))
|
|
goto(BASE + '/apps/calibre-web/data')
|
|
time.sleep(2) # give any autofill its moment
|
|
r = ev("""(function(){ var k=document.getElementById('fabEncryptionKey');
|
|
return {path: location.pathname, keyField: k ? {name: k.name, autocomplete: k.getAttribute('autocomplete'), type: k.type, valueLength: k.value.length} : null,
|
|
oldIdPresent: !!document.getElementById('exportPassword'),
|
|
statusRows: document.querySelectorAll('[data-backup-row="second"],[data-backup-row="target"]').length,
|
|
targetLabels: document.querySelectorAll('label[for="target"]').length,
|
|
form: !!document.querySelector('[data-second-backup-form]'),
|
|
warnCallout: !!document.querySelector('#app-backup-section .app-callout-warn'),
|
|
scrollWidth: document.documentElement.scrollWidth}; })()""")
|
|
r['console'] = console[:]
|
|
shot('calibre-web-data')
|
|
print(json.dumps(r, indent=1, ensure_ascii=False))
|
|
finally:
|
|
proc.terminate()
|
|
subprocess.run(['rm', '-rf', prof])
|