nnduc/movim-rs
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 1 Wiki Activity

omemo keyring

Browse Source
This commit is contained in:
nnduc
2026-10-27 23:49:59 +07:00
parent 9f44778f21
commit 1ce93d883b
16 changed files with 1327 additions and 481 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/e2e/__pycache__/browser_edit.cpython-313.pyc
BIN
View File
Binary file not shown.
tests/e2e/browser_edit.py
+2 -1
View File
@@ -87,4 +87,5 @@ async def main():
print(e)
await browser.close()
asyncio.run(main())
if __name__ == "__main__":
asyncio.run(main())
tests/e2e/browser_omemo_history.py
+68
View File
@@ -0,0 +1,68 @@
"""Own OMEMO messages coming back from the archive (MAM) are shown decrypted, like when edited:
their local id is m_…, the browser stored the plaintext under the id it sent them with."""
import asyncio
import os
import sqlite3
import subprocess
from playwright.async_api import async_playwright
from browser import login, BASE, OUT
from browser_edit import send, edit_last, bubbles, open_chat
D = os.path.dirname(os.path.abspath(__file__))
async def main():
async with async_playwright() as p:
browser = await p.chromium.launch()
actx, alice, aerr = await login(browser, "alice@localhost", "alicepass", "historyalice")
await alice.wait_for_timeout(4000)
await open_chat(alice, "bob@localhost")
await alice.evaluate("ChatOmemo.enableContactState('bob@localhost', false)")
await alice.wait_for_timeout(5000)
await send(alice, "Lịch sử: bản gốc")
await edit_last(alice, "Lịch sử: đã sửa")
await send(alice, "Lịch sử: tin thứ hai")
bctx, bob, berr = await login(browser, "bob@localhost", "bobpass", "historybob")
await bob.wait_for_timeout(4000)
await open_chat(bob, "alice@localhost")
await bob.evaluate("ChatOmemo.enableContactState('alice@localhost', false)")
await bob.wait_for_timeout(5000)
await send(bob, "Lịch sử: trả lời của bob")
await alice.wait_for_timeout(3000)
print("before, alice:", await bubbles(alice, 3))
await bctx.close()
# The server forgets the discussion, the history comes back from MAM
subprocess.run(["kill", open(f"{D}/server.pid").read().strip()])
await alice.wait_for_timeout(1000)
db = sqlite3.connect(f"{D}/data/movim.sqlite")
for table in ("messages", "mam_earliests"):
db.execute(f"DELETE FROM {table} WHERE user_id = 'alice@localhost'")
db.commit()
db.close()
subprocess.run([f"{D}/run.sh"])
# Same browser context: the decrypted messages of IndexedDB are kept
page = await actx.new_page()
await page.goto(BASE + "/login")
await page.wait_for_timeout(3000)
if "/login" in page.url:
await page.wait_for_selector("#login_widget input[type=submit]:not([disabled])", timeout=15000)
await page.fill("input#username", "alice@localhost")
await page.fill("input#password", "alicepass")
await page.click("#login_widget input[type=submit]")
await page.wait_for_url("**/chat**", timeout=30000)
await page.wait_for_timeout(4000)
await open_chat(page, "bob@localhost")
await page.wait_for_timeout(4000)
print("ids:", await page.evaluate(
"[...document.querySelectorAll('#chat_widget .bubble > div.message')].slice(-3).map(m => m.id)"))
print("from history, alice:", await bubbles(page, 3))
await page.screenshot(path=f"{OUT}/history_alice.png")
for e in aerr:
print(e)
await browser.close()
asyncio.run(main())
tests/e2e/browser_omemo_restore.py
+58
View File
@@ -0,0 +1,58 @@
"""OMEMO key backup restored in another browser: the decrypted messages come with it,
including our own (edited) ones that are never encrypted for the sending device."""
import asyncio
from playwright.async_api import async_playwright
from browser import login, BASE, OUT
from browser_edit import send, edit_last, bubbles, open_chat
async def main():
async with async_playwright() as p:
browser = await p.chromium.launch()
actx, alice, aerr = await login(browser, "alice@localhost", "alicepass", "restorealice")
bctx, bob, berr = await login(browser, "bob@localhost", "bobpass", "restorebob")
await alice.wait_for_timeout(4000)
await open_chat(alice, "bob@localhost")
await open_chat(bob, "alice@localhost")
await alice.evaluate("ChatOmemo.enableContactState('bob@localhost', false)")
await alice.wait_for_timeout(5000)
await send(alice, "Khôi phục: bản gốc")
await edit_last(alice, "Khôi phục: đã sửa")
await send(bob, "Khôi phục: trả lời của bob")
print("before, alice:", await bubbles(alice, 2))
# Backup
await alice.goto(BASE + "/configuration")
await alice.wait_for_timeout(3000)
await alice.evaluate("Account_ajaxOmemoBackup()")
await alice.wait_for_selector("form[name=omemo_backup] input[name=password]", timeout=10000)
await alice.fill("form[name=omemo_backup] input[name=password]", "secret")
await alice.fill("form[name=omemo_backup] input[name=password_confirmation]", "secret")
async with alice.expect_download() as dl:
await alice.evaluate("Account.omemoBackup()")
path = f"{OUT}/restore_backup.json"
await (await dl.value).save_as(path)
# Restore in a fresh browser
ctx2, alice2, a2err = await login(browser, "alice@localhost", "alicepass", "restorealice2")
await alice2.wait_for_timeout(4000)
await alice2.goto(BASE + "/configuration")
await alice2.wait_for_timeout(3000)
await alice2.evaluate("Account_ajaxOmemoRestore()")
await alice2.wait_for_selector("form[name=omemo_restore] input[name=file]", timeout=10000)
await alice2.set_input_files("form[name=omemo_restore] input[name=file]", path)
await alice2.fill("form[name=omemo_restore] input[name=password]", "secret")
async with alice2.expect_navigation():
await alice2.evaluate("Account.omemoRestore()")
await alice2.wait_for_timeout(4000)
await open_chat(alice2, "bob@localhost")
print("restored, alice:", await bubbles(alice2, 2))
await alice2.screenshot(path=f"{OUT}/restore_alice2.png")
for e in aerr + berr + a2err:
print(e)
await browser.close()
asyncio.run(main())