Compare commits

..
Author SHA1 Message Date
go0p 881bbc2c89 fix: choose local yak engine before pairing 2026-09-24 17:28:40 +08:00
go0p b44e82d8e1 fix: keep pairing diagnostics visible 2026-09-24 17:16:10 +08:00
go0p a9009c2058 fix: expose browser pairing diagnostics 2026-09-24 17:06:45 +08:00
go0p d35912c7cf feat: support tab-wide recording and release notes
CI / Test and build (push) Waiting to run
2026-09-24 14:50:33 +08:00
go0p 8c84e16012 chore(release): bump version to 0.2.6
CI / Test and build (push) Canceled after 0s
2026-09-17 17:55:17 +08:00
go0p 601f8acbcf fix(capture): automate safe bidirectional gateways for users and agents 2026-09-17 17:24:54 +08:00
go0p 45f53604e8 chore(release): bump version to 0.2.5
CI / Test and build (push) Canceled after 0s
2026-09-16 15:51:56 +08:00
go0p 4b0df8c706 feat(browser): expose proxy state and browser identity (#9)
* feat(proxy): reflect and release browser proxy control

* feat(browser): report browser product identity
2026-09-16 15:51:33 +08:00
go0p 2000d59054 fix(browser): improve transform mapping and browser analysis capabilities
CI / Test and build (push) Canceled after 0s
2026-09-14 11:42:20 +08:00
go0p e3b1775f8d feat(transform): confirm agent profiles locally
CI / Test and build (push) Canceled after 0s
2026-09-07 15:36:59 +08:00
go0p dd61541c3e refactor(authorization): simplify cross-instance testing
CI / Test and build (push) Canceled after 0s
2026-09-04 16:43:11 +08:00
go0p 71a14eb17a fix(authorization): keep capture across same-origin login 2026-09-04 14:34:46 +08:00
go0p 1f703dfbdc feat(authorization): test across ytray browser instances 2026-09-04 11:27:53 +08:00
go0p 951ca046ab fix(authorization): resolve browser targets on demand 2026-09-03 18:09:29 +08:00
go0p d028bfc75c fix(authorization): show connected browser instances 2026-09-03 18:03:16 +08:00
go0p f028cff33b feat(browser): refine workbench navigation and authorization 2026-09-03 17:23:19 +08:00
go0p 8e84735ecf feat(browser): add local handoff presentation 2026-09-03 17:04:00 +08:00
125 changed files with 5421 additions and 10272 deletions
+1 -1
View File
@@ -126,7 +126,7 @@ The independent `page-recorder-main-world.js` entrypoint temporarily wraps user-
Each click or submit begins a five-second business Trace. Nested and subsequent events share that Trace. Inputs and outputs are reduced to bounded evidence paths, byte lengths, encodings, and a randomly seeded 64-bit correlation fingerprint. The seed remains inside one page document and is regenerated for every document observer, so fingerprints cannot be compared across document boundaries. Matching an earlier output fingerprint to a later input fingerprint creates an exact Pipeline link. This is evidence of value equality inside one document segment, not proof of semantic causality.
Raw previews are disabled by default. Enabling them requires `browser.recording.sensitive.read` and caps each preview at 8 KiB. A user-started recording is a tab/frame-scoped Session: the current document keeps live hooks and handles, while the background merges bounded document segments into extension-only `storage.session`. A full navigation is recorded as a first-class Trace event; the previous segment is sealed, the destination document receives a new observer with the same Session identity and a synchronized global sequence, and recording continues until explicit stop, expiry, clear, or tab close. The single per-target Session is removed by a new recording, explicit clear, tab close, or browser-session end. Previews are never written to persistent storage or included in audit or AI request-analysis payloads. Recording is bounded to 500 aggregate events, 48 evidence items per side, 1,000 links, and 64 live callable handles per document.
Raw previews are disabled by default. Enabling them requires `browser.recording.sensitive.read` and caps each preview at 8 KiB. A user-started local recording covers every accessible frame in the selected tab and automatically attaches newly committed frames; each frame still owns an independent document Session, hook set, evidence salt, and callable registry, while the UI merges their bounded timelines. Agent-owned recordings remain explicitly frame-scoped. A full navigation is recorded as a first-class Trace event; the previous segment is sealed, the destination document receives a new observer with the same Session identity and a synchronized global sequence, and recording continues until explicit stop, expiry, clear, or tab close. Sessions are removed by a new recording, explicit clear, tab close, or browser-session end. Previews are never written to persistent storage or included in audit or AI request-analysis payloads. Recording is bounded to 500 aggregate events, 48 evidence items per side, 1,000 links, and 64 live callable handles per document.
Navigation is both a business event and a strict execution-context boundary. Full document navigation, reload, browser Back/Forward, same-document History changes and fragment changes are distinguished. If Back/Forward restores the original document from BFCache, its recorder, handles and callables are resumed without clearing earlier evidence; if the browser performs a hard reload, the historical evidence remains but the destroyed closure heap is truthfully unavailable. MAIN-world lifecycle and the tab-scoped Session are separate states, so a temporary document transition no longer appears as a completed recording. A grant-owned recording remains document-bound and stops at navigation instead of silently extending an Agent's authority into a new document.
+3 -1
View File
@@ -338,7 +338,9 @@ Chrome Store 构建声明 Chrome 138+。用户需要在扩展详情页开启“
https://aliyun-oss.yaklang.com/chrome-extension/manifest.json
```
manifest 的 `latest` 指向最新版本,`versions[0]` 为完整记录,最多保留 10 个历史版本。每个版本按 `variant`(`chrome-store` / `chrome-enterprise` / `firefox` / `firefox-amo`)匹配 artifact,字段包括 `url`、`filename`、`sha256`、`size` 与 `checksum_url`;manifest 自身的 SHA-256 在同目录的 `manifest.json.sha256.txt`。
manifest 的 `latest` 指向最新版本,`versions[0]` 为完整记录,最多保留 10 个历史版本。每个版本的 `notes` 是面向用户的真实更新说明,按 `variant`(`chrome-store` / `chrome-enterprise` / `firefox` / `firefox-amo`)匹配 artifact,字段包括 `url`、`filename`、`sha256`、`size` 与 `checksum_url`;manifest 自身的 SHA-256 在同目录的 `manifest.json.sha256.txt`。
每次修改 `package.json` 的版本时,必须同步在 `release-notes.json` 中增加该版本的更新说明;缺失或内容为空会让打包和发布直接失败。YTray 等消费方会原样展示这些说明。
推荐的消费流程:
+3 -1
View File
@@ -2,7 +2,7 @@
"name": "yakit-chrome-client",
"description": "Yakit Browser Extension",
"private": true,
"version": "0.2.3",
"version": "0.2.7",
"type": "module",
"packageManager": "[email protected]",
"scripts": {
@@ -27,6 +27,8 @@
"verify:aesrsa": "node scripts/verify-aesrsa-transaction.mjs",
"verify:aesserver": "node scripts/verify-aesserver-transaction.mjs",
"verify:des": "node scripts/verify-des-transaction.mjs",
"verify:capture": "pnpm build:enterprise && node scripts/verify-capture-regressions.mjs && node scripts/verify-login-gateway.mjs",
"verify:agent-gateway": "pnpm build:enterprise && node scripts/verify-agent-gateway.mjs",
"verify:agent-contract:aes": "node scripts/verify-aes-agent-contract.mjs",
"verify:agent-contract:aesrsa": "node scripts/verify-aesrsa-transaction.mjs",
"verify:agent-contract:holdout": "AGENT_CONTRACT_HOLDOUT_ONLY=1 EXTENSION_PATH=.output/chrome-mv3-enterprise node scripts/verify-ui.mjs",
+11
View File
@@ -0,0 +1,11 @@
{
"0.2.7": [
"浏览器录制现在覆盖标签页内所有可访问页面,并自动接续后续加载的登录页面。",
"由 YTray 启动的托管浏览器会自动向 Yakit 发起配对连接。",
"修复代理绕过列表无法正常换行编辑的问题,并改善深色界面的选项显示。",
"插件更新页现在会展示随版本发布的真实更新内容。"
],
"0.2.6": [
"自动为用户和 Agent 建立安全的双向浏览器网关。"
]
}
+14 -2
View File
@@ -34,8 +34,13 @@ function artifactFingerprint(artifacts) {
}
function toVersionEntry(entry) {
if (!Array.isArray(entry.notes) || entry.notes.length === 0
|| entry.notes.some((note) => typeof note !== 'string' || note.trim() === '')) {
throw new Error(`release ${entry.version} must contain non-empty notes`);
}
return {
version: entry.version,
notes: entry.notes,
published_at: entry.built_at,
commit: entry.commit ?? null,
artifacts: entry.artifacts.map((a) => ({
@@ -65,6 +70,11 @@ function validate(manifest) {
if (!Array.isArray(versionEntry.artifacts) || versionEntry.artifacts.length === 0) {
throw new Error(`version ${versionEntry.version} has no artifacts`);
}
if (versionEntry.notes !== undefined
&& (!Array.isArray(versionEntry.notes) || versionEntry.notes.length === 0
|| versionEntry.notes.some((note) => typeof note !== 'string' || note.trim() === ''))) {
throw new Error(`version ${versionEntry.version} has invalid notes`);
}
const variants = new Set();
for (const artifact of versionEntry.artifacts) {
if (variants.has(artifact.variant)) throw new Error(`duplicate variant ${artifact.variant} in version ${versionEntry.version}`);
@@ -103,8 +113,10 @@ if (args['existing-manifest']) {
const newEntry = toVersionEntry(entry);
const idx = versions.findIndex((v) => v.version === entry.version);
if (idx >= 0 && artifactFingerprint(versions[idx].artifacts) === artifactFingerprint(entry.artifacts)) {
// Idempotent rerun: keep the original entry (published_at stays stable).
console.log(`version ${entry.version} already in manifest with identical artifacts; kept as-is`);
// An idempotent rerun may backfill release notes without changing immutable
// artifacts or their original publication metadata.
versions[idx] = { ...versions[idx], notes: newEntry.notes };
console.log(`version ${entry.version} already in manifest with identical artifacts; release notes synchronized`);
} else {
if (idx >= 0) {
versions.splice(idx, 1);
+6 -1
View File
@@ -65,6 +65,11 @@ const distDir = resolve(root, String(args.dist ?? 'dist'));
const pkg = JSON.parse(await readFile(resolve(root, 'package.json'), 'utf8'));
const { version } = pkg;
const notesByVersion = JSON.parse(await readFile(resolve(root, 'release-notes.json'), 'utf8'));
const notes = notesByVersion[version]?.map((note) => String(note).trim()).filter(Boolean);
if (!Array.isArray(notes) || notes.length === 0) {
throw new Error(`release-notes.json must contain at least one note for version ${version}`);
}
let commit = null;
try {
@@ -138,6 +143,6 @@ for (const target of VARIANTS) {
console.log(`packaged ${filename} (${size} bytes, sha256 ${sha256.slice(0, 12)}…)`);
}
const entry = { version, commit, built_at: new Date().toISOString(), artifacts };
const entry = { version, notes, commit, built_at: new Date().toISOString(), artifacts };
await writeFile(resolve(distDir, 'release-entry.json'), `${JSON.stringify(entry, null, 2)}\n`);
console.log(`release entry written: ${resolve(distDir, 'release-entry.json').slice(root.length + 1)} (version ${version})`);
+113
View File
@@ -0,0 +1,113 @@
import assert from 'node:assert/strict'
import { spawn } from 'node:child_process'
import { resolve } from 'node:path'
import { createDecipheriv } from 'node:crypto'
import { extensionRequest, launchBrowserAgentContractHarness } from './browser-agent-contract-harness.mjs'
function body(packet) { return JSON.parse(packet.raw.slice(packet.raw.indexOf('\r\n\r\n') + 4)) }
function decrypt(envelope) {
const decipher = createDecipheriv('aes-128-cbc', Buffer.from(envelope.key, 'hex'), Buffer.from(envelope.iv, 'hex'))
return JSON.parse(Buffer.concat([decipher.update(Buffer.from(envelope.message, 'base64')), decipher.final()]).toString())
}
const targetURL = process.env.LOGIN_TARGET || 'http://localhost:8080/crypto/sqli/aes-ecb/encrypt/login'
const engine = spawn('go', ['test', './common/yakgrpc', '-run', '^TestBrowserAgentLiveGateway$', '-count=1', '-v'], {
cwd: process.env.YAKLANG_ROOT || resolve(import.meta.dirname, '../../../go/yaklang'),
env: { ...process.env, YAK_BROWSER_AGENT_E2E: '1' }, stdio: ['ignore', 'pipe', 'pipe'],
})
let output = ''
let harness, connection
const exited = new Promise(resolve => engine.once('exit', code => resolve(code)))
const ready = new Promise((resolve, reject) => {
const timer = setTimeout(() => reject(new Error(`Go Agent bridge startup timed out: ${output.slice(-4000)}`)), 120_000)
engine.stdout.on('data', chunk => {
output += chunk
const match = output.match(/YAK_AGENT_E2E=(\{[^\n]+\})/)
if (match) { clearTimeout(timer); resolve(JSON.parse(match[1])) }
})
engine.stderr.on('data', chunk => { output += chunk })
engine.once('error', error => { clearTimeout(timer); reject(error) })
engine.once('exit', code => { clearTimeout(timer); reject(new Error(`Go bridge exited ${code}: ${output}`)) })
})
try {
connection = await ready
const tool = async (name, params = {}) => {
const response = await fetch(connection.endpoint, { method: 'POST', headers: { 'X-Test-Token': connection.token }, body: JSON.stringify({ tool: name, params }) })
if (!response.ok) throw new Error(`${name}: ${await response.text()}`)
return response.json()
}
harness = await launchBrowserAgentContractHarness({ profilePrefix: 'yakit-agent-gateway-', targetURL })
const { controlPage, targetPage, tabId } = harness
let completed = 0, blocked = 0
targetPage.on('response', response => { if (response.url() === targetURL && response.request().method() === 'POST') completed++ })
targetPage.on('requestfailed', request => { if (request.url() === targetURL && request.failure()?.errorText.includes('BLOCKED_BY_CLIENT')) blocked++ })
const state = await controlPage.evaluate(async () => (await chrome.runtime.sendMessage({ action: 'state.get' })).data)
await extensionRequest(controlPage, 'bridge.config.save', {
transport: 'websocket', nativeHost: 'com.yaklang.browser_agent', endpoint: connection.bridge,
autoConnect: false, installationId: state.bridge.installationId,
})
await controlPage.evaluate(async () => {
const result = await chrome.runtime.sendMessage({ action: 'bridge.pair' })
if (!result.ok) throw new Error(JSON.stringify(result.error))
})
for (let attempt = 0; ; attempt++) {
const status = await controlPage.evaluate(async () => (await chrome.runtime.sendMessage({ action: 'bridge.status' })).data)
if (status.state === 'connected') break
if (attempt === 100) throw new Error(`Bridge did not connect: ${JSON.stringify(status)}`)
await controlPage.waitForTimeout(100)
}
// Close the extension UI. All workflow calls below go through Go Agent tools.
// A previously released human debugging session must not lock out the Agent.
await extensionRequest(controlPage, 'deep.capture.start', { tabId, frameId: 0, matcher: { kind: 'request', urlPattern: '/not-triggered' } })
await extensionRequest(controlPage, 'deep.capture.detach', { tabId, frameId: 0 })
await controlPage.close()
const call = (method, params = {}) => tool('browser.capability.call', { method, params: { tabId, frameId: 0, ...params } })
const catalog = await tool('browser.capability.catalog', { domain: 'debugger' })
assert(catalog.capabilities.some(item => item.method === 'browser.deep_capture.start'))
const context = await call('browser.context', { includeDom: true })
const nodes = context.document.interactive
const username = nodes.find(node => node.name === 'username' || node.id === 'username')
const password = nodes.find(node => node.name === 'password' || node.id === 'password')
const submit = nodes.find(node => node.tag === 'button' && node.type === 'submit')
assert(username && password && submit, JSON.stringify(nodes))
for (const [node, value] of [[username, 'agent-original'], [password, 'agent-original-wrong']]) {
await call('browser.node.action', { captureId: context.captureId, nodeId: node.nodeId, action: 'setValue', value })
}
const inspection = await tool('browser.crypto.inspect', { tabId, frameId: 0, captureId: context.captureId, nodeId: submit.nodeId })
assert.equal(inspection.gatewayPreparation.state, 'capture-required')
const plaintext = { username: 'agent-new-user', password: 'agent-new-wrong' }
const url = new URL(targetURL)
const request = `POST ${url.pathname} HTTP/1.1\r\nHost: ${url.host}\r\nContent-Type: application/json\r\n\r\n${JSON.stringify(plaintext)}`
const prepareInput = { ...inspection.target, candidate_id: inspection.gatewayPreparation.candidateId, request, is_https: false }
await assert.rejects(tool('browser.transform.prepare', { ...prepareInput, captureId: 'stale-context', nodeId: submit.nodeId }), /快照已经失效/)
assert.equal((await call('browser.deep_capture.status')).state, 'detached', 'failed preparation must release its debugger')
assert.equal(completed, 1)
const prepared = await tool('browser.transform.prepare', prepareInput)
assert(prepared.valid, JSON.stringify(prepared))
assert.deepEqual(prepared.validationDraft.directions, { request: true, response: true })
assert.equal(completed, 1, 'automatic preparation must not submit another real browser login')
assert.equal(blocked, 1)
const preparedAgain = await tool('browser.transform.prepare', prepareInput)
assert(preparedAgain.valid)
assert.equal(blocked, 1, 'repeated preparation must reuse existing captured functions')
const tested = await tool('browser.http.test', { validation_id: preparedAgain.validationDraft.id, request, is_https: false })
assert(tested.responseTransformEnabled, JSON.stringify(tested))
assert(tested.requestTransformed, JSON.stringify(tested))
assert(tested.responseTransformed)
assert.deepEqual(decrypt(body(tested.wireRequest)), plaintext)
assert.deepEqual(body(tested.plaintextResponse), decrypt(body(tested.wireResponse)))
const draft = await call('browser.profile.validation.latest')
const saved = await tool('browser.capability.call', { method: 'browser.transform.profile.save', params: draft.profile })
assert(saved.request.enabled && saved.response.enabled)
await tool('browser.capability.call', { method: 'browser.transform.profile.delete', params: { id: saved.id } })
assert.equal(completed, 1)
assert.equal(await targetPage.locator('#username').inputValue(), 'agent-original')
assert.equal(await targetPage.locator('#password').inputValue(), 'agent-original-wrong')
console.log('Agent gateway passed: signed Go bridge, advanced capability access, no plugin UI, failed-capture cleanup, automatic business capture, bidirectional validation, independently checked request/response crypto, default response decryption and Profile save/delete.')
} finally {
await harness?.close()
if (connection) await fetch(`${connection.endpoint}/finish`, { method: 'POST', headers: { 'X-Test-Token': connection.token } }).catch(() => undefined)
else engine.kill('SIGTERM')
const code = await exited
if (code !== 0 && connection) throw new Error(`Go integration exited ${code}: ${output.slice(-4000)}`)
}
+142
View File
@@ -0,0 +1,142 @@
import assert from 'node:assert/strict'
import { createServer } from 'node:http'
import { readFile } from 'node:fs/promises'
import { extensionRequest, launchBrowserAgentContractHarness, waitFor } from './browser-agent-contract-harness.mjs'
const registryKey = (await readFile(new URL('../src/features/page-callable/constants.ts', import.meta.url), 'utf8')).match(/= '([^']+)'/)[1]
const received = []
const server = createServer(async (request, response) => {
let body = ''
for await (const chunk of request) body += chunk
received.push({ url: request.url, method: request.method, body })
if (request.url === '/key' || request.url === '/bad-key') {
response.setHeader('Content-Type', 'application/json')
response.end(JSON.stringify(request.url === '/key' ? { key: 'fresh-server-key' } : {}))
} else if (request.method === 'POST') {
response.setHeader('Content-Type', 'application/json')
response.end('{}')
} else {
response.setHeader('Content-Type', 'text/html')
response.end('<form id="form"><input name="password" value="old"><button>Submit</button></form><script>globalThis.cachedFetch = fetch.bind(window); globalThis.cachedSubmit = HTMLFormElement.prototype.submit;</script>')
}
})
await new Promise(resolve => server.listen(0, '127.0.0.1', resolve))
const targetURL = `http://127.0.0.1:${server.address().port}/`
let harness
try {
harness = await launchBrowserAgentContractHarness({ profilePrefix: 'yakit-capture-regressions-', targetURL })
const { controlPage, targetPage, tabId } = harness
const target = { tabId, frameId: 0 }
const request = (action, payload = {}) => extensionRequest(controlPage, action, { ...target, ...payload })
await request('recording.start', { captureValues: true, maxEntries: 200, maxValueBytes: 8192 })
await targetPage.evaluate(async () => {
await fetch(new Request(location.origin + '/request-object', { method: 'POST', body: 'encrypted-request-body' }))
})
const recorded = await waitFor(controlPage, 'recording.get', target, snapshot => snapshot.events.some(event =>
event.url?.endsWith('/request-object') && event.inputs.some(value => value.path === '$body' && value.preview === 'encrypted-request-body')))
assert(recorded.events.some(event => event.url?.endsWith('/request-object')))
async function register(mode, transaction = false) {
return targetPage.evaluate(({ key, mode, transaction }) => {
const id = crypto.randomUUID()
const endpoint = location.origin + '/' + mode
const savedState = { password: 'old' }
const prerequisites = mode.startsWith('prerequisite') ? [{
boundary: 'fetch', method: 'GET', url: location.origin + (mode === 'prerequisite-invalid' ? '/bad-key' : '/key'),
requestBodyFormat: 'none', maxRequestBodyBytes: 0,
response: { statusCode: 200, url: location.origin + (mode === 'prerequisite-invalid' ? '/bad-key' : '/key'), bodyFormat: 'json', maxBodyBytes: 4096, requiredPaths: ['body.key'] },
}] : []
if (mode === 'controlled') document.querySelector('input').addEventListener('input', event => { savedState.password = event.target.value })
const metadata = {
id, name: mode, kind: transaction ? 'request-transaction' : 'business-closure', operation: mode,
origin: location.origin, lifecycle: 'document', execution: { resultMode: 'auto', timeoutMs: 1500 },
inputSlots: [{ id: 'body', name: 'body', index: 0, role: 'data', dataType: 'object', required: true, retained: false }],
output: { dataType: transaction ? 'object' : 'string', encoding: transaction ? 'json' : 'utf8', shape: transaction ? 'envelope' : 'value', paths: transaction ? ['body.password'] : [] },
provenance: {}, createdAt: Date.now(),
...(transaction ? { transaction: { version: 2, prerequisites, inputMode: 'auto', request: {
boundary: mode === 'multipart-xhr' ? 'xhr' : mode === 'multipart-beacon' ? 'beacon' : 'fetch',
method: 'POST', url: endpoint, bodyFormat: mode.startsWith('multipart') ? 'form' : 'json', expectedDestinations: ['body.password'],
} } } : {}),
}
const registry = globalThis[key] ||= new Map()
registry.set(id, { metadata, invoke(args) {
if (mode === 'native-form') {
const form = document.querySelector('form'); form.method = 'POST'; form.action = endpoint
Reflect.apply(globalThis.cachedSubmit, form, []); return 'unexpected'
}
if (mode === 'cached') return globalThis.cachedFetch(endpoint, { method: 'POST', body: 'leak' }).then(() => 'bad')
if (mode === 'timer') { setTimeout(() => fetch(endpoint, { method: 'POST', body: 'leak' }), 100); return 'ok' }
const value = mode === 'stale' || mode === 'controlled' ? savedState : args[0]
const send = () => {
if (mode.startsWith('multipart')) {
const form = new FormData(); form.set('password', value.password)
if (mode === 'multipart-xhr') { const xhr = new XMLHttpRequest(); xhr.open('POST', endpoint); return xhr.send(form) }
if (mode === 'multipart-beacon') return navigator.sendBeacon(endpoint, form)
return fetch(endpoint, { method: 'POST', body: form })
}
return fetch(endpoint, { method: 'POST', headers: { 'Content-Type': 'application/json' }, body: JSON.stringify(value) })
}
if (mode.startsWith('concurrent')) return new Promise(resolve => setTimeout(resolve, 40)).then(send)
if (prerequisites.length) return fetch(mode === 'prerequisite-unplanned' ? '/unplanned' : prerequisites[0].url).then(response => response.json()).then(send)
return send()
} })
return id
}, { key: registryKey, mode, transaction })
}
const execute = (id, password = 'new') => request('callable.execute', { callableId: id, args: [{ password }] })
await assert.rejects(execute(await register('cached')), /fetch|网络|阻止|Failed/i)
assert.equal((await execute(await register('timer'))).value, 'ok')
await targetPage.waitForTimeout(160)
assert(!received.some(item => ['/cached', '/timer'].includes(item.url)))
await assert.rejects(execute(await register('stale', true)), /新明文|旧状态/)
for (const mode of ['controlled', 'multipart', 'multipart-xhr', 'multipart-beacon']) {
assert.deepEqual((await execute(await register(mode, true))).value, { password: 'new' })
assert.equal(await targetPage.locator('input').inputValue(), 'old')
}
const concurrent = await Promise.all([register('concurrent-a', true), register('concurrent-b', true)])
const results = await Promise.all(concurrent.map((id, index) => execute(id, `new-${index}`)))
assert.deepEqual(results.map(result => result.value), [{ password: 'new-0' }, { password: 'new-1' }])
assert(!received.some(item => ['/stale', '/controlled', '/multipart', '/multipart-xhr', '/multipart-beacon', '/concurrent-a', '/concurrent-b'].includes(item.url)))
assert.deepEqual((await execute(await register('prerequisite', true))).value, { password: 'new' })
assert.equal(received.filter(item => item.url === '/key').length, 1)
await assert.rejects(execute(await register('prerequisite-invalid', true)), /缺少目标字段/)
await assert.rejects(execute(await register('prerequisite-unplanned', true)), /未授权请求/)
assert(!received.some(item => item.url === '/unplanned' || item.method === 'POST' && item.url.startsWith('/prerequisite')))
assert.equal((await controlPage.evaluate(() => chrome.declarativeNetRequest.getSessionRules())).length, 0)
const nonWritableId = await register('timer')
await targetPage.evaluate(() => {
globalThis.beforeIsolation = { fetch, setTimeout, sendBeacon: navigator.sendBeacon }
Object.defineProperty(navigator, 'sendBeacon', { value: navigator.sendBeacon, writable: false, configurable: true })
})
await assert.rejects(execute(nonWritableId), /不能隔离页面边界/)
assert(await targetPage.evaluate(() => {
const restored = fetch === globalThis.beforeIsolation.fetch && setTimeout === globalThis.beforeIsolation.setTimeout
Object.defineProperty(navigator, 'sendBeacon', { value: globalThis.beforeIsolation.sendBeacon, writable: true, configurable: true })
return restored
}))
assert.equal((await controlPage.evaluate(() => chrome.declarativeNetRequest.getSessionRules())).length, 0)
// A strict arrow listener in a one-line script must resolve by its exact function location.
await targetPage.addScriptTag({ content: `document.querySelector('form').addEventListener('submit', e => { 'use strict'; e.preventDefault(); fetch('/strict', { method: 'POST', body: 'cipher' }) }); document.querySelector('form').addEventListener('submit', e => { 'use strict'; e.preventDefault() });` })
await request('deep.capture.start', { matcher: { kind: 'request', urlPattern: '/strict' } })
const submit = targetPage.locator('form button').click({ noWaitAfter: true, timeout: 20_000 })
const paused = await waitFor(controlPage, 'deep.capture.status', target, value => value.state === 'paused' && !value.pause.collecting)
const resolved = paused.pause.frames.filter(frame => frame.functionInspection?.resolution === 'event-listener')
assert.equal(resolved.length, 1, JSON.stringify(paused.pause))
assert(resolved[0].functionInspection.resolved)
await request('deep.capture.resume')
await submit
const nativeFormId = await register('native-form')
const blockedForm = targetPage.waitForEvent('requestfailed', { predicate: request => request.url().endsWith('/native-form'), timeout: 10_000 })
void blockedForm.catch(() => undefined)
let nativeFormError
await assert.rejects(execute(nativeFormId), error => { nativeFormError = error.message; return true })
const failedForm = await blockedForm.catch(error => { throw new Error(`${error.message}; execution=${nativeFormError}; url=${targetPage.url()}; received=${JSON.stringify(received)}`) })
assert(failedForm.failure().errorText.includes('BLOCKED_BY_CLIENT'))
assert(!received.some(item => item.url === '/native-form'))
console.log('Capture regressions passed: Request body, native network guard, timer cleanup, fresh input, controlled forms, multipart, concurrent callables, strict same-line listeners.')
} finally {
await harness?.close()
await new Promise(resolve => server.close(resolve))
}
+103
View File
@@ -0,0 +1,103 @@
import assert from 'node:assert/strict'
import { createDecipheriv } from 'node:crypto'
import { extensionRequest, launchBrowserAgentContractHarness, transformedFetchOptions, waitFor } from './browser-agent-contract-harness.mjs'
const targetURL = process.env.LOGIN_TARGET || 'http://localhost:8080/crypto/sqli/aes-ecb/encrypt/login'
function decrypt(envelope) {
const decipher = createDecipheriv('aes-128-cbc', Buffer.from(envelope.key, 'hex'), Buffer.from(envelope.iv, 'hex'))
return JSON.parse(Buffer.concat([decipher.update(Buffer.from(envelope.message, 'base64')), decipher.final()]).toString())
}
let harness
try {
harness = await launchBrowserAgentContractHarness({ profilePrefix: 'yakit-login-gateway-', targetURL })
const { targetPage, controlPage, tabId, extensionId } = harness
const target = { tabId, frameId: 0 }
let completedPosts = 0
let blockedPosts = 0
targetPage.on('response', response => { if (response.url() === targetURL && response.request().method() === 'POST') completedPosts++ })
targetPage.on('requestfailed', request => {
if (request.url() === targetURL && request.method() === 'POST' && request.failure()?.errorText.includes('BLOCKED_BY_CLIENT')) blockedPosts++
})
await extensionRequest(controlPage, 'recording.start', { ...target, captureValues: true, maxEntries: 300, maxValueBytes: 8192 })
await targetPage.locator('#username').fill('audit-original')
await targetPage.locator('#password').fill('original-wrong-password')
await targetPage.locator('button[type=submit]').click()
const recorded = await waitFor(controlPage, 'recording.get', target, snapshot =>
snapshot.profileCandidates.some(candidate => candidate.direction === 'response' && candidate.status === 'ready'))
await extensionRequest(controlPage, 'recording.stop', target)
assert.equal(completedPosts, 1)
const responseCandidate = recorded.profileCandidates.find(candidate => candidate.direction === 'response' && candidate.status === 'ready')
const requestCandidate = recorded.profileCandidates.find(candidate => candidate.direction === 'request' && candidate.transactionId === responseCandidate.transactionId)
assert.equal(requestCandidate.status, 'capture-required', JSON.stringify(recorded.profileCandidates))
await controlPage.goto(`chrome-extension://${extensionId}/options.html?tabId=${tabId}#gateway`)
await controlPage.evaluate(() => {
globalThis.captureActions = []
const send = chrome.runtime.sendMessage.bind(chrome.runtime)
chrome.runtime.sendMessage = (...args) => { globalThis.captureActions.push(args[0]); return send(...args) }
})
const traceIndex = recorded.traces.findIndex(trace => trace.id === responseCandidate.traceId)
await controlPage.locator('.recording-traces button').nth(traceIndex).click()
await controlPage.locator(`[data-event-id="${responseCandidate.source.eventId}"]`).click()
await controlPage.getByRole('button', { name: '继续捕获请求方向', exact: true }).click()
await waitFor(controlPage, 'deep.capture.status', target, status => status.state === 'armed')
let submitError
const submit = targetPage.locator('button[type=submit]').click({ noWaitAfter: true, timeout: 45_000 }).catch(error => { submitError = error })
let profiles
try {
profiles = await waitFor(controlPage, 'transform.profile.list', {}, values => values.some(profile => profile.request.enabled && profile.response.enabled), 30_000)
} catch (error) {
const status = await extensionRequest(controlPage, 'deep.capture.status', target)
if (status.pause) status.pause.frames = status.pause.frames.map(({ scopes, ...frame }) => frame)
const actions = await controlPage.evaluate(() => globalThis.captureActions.filter(item => /create|start|save/.test(item.action)))
throw new Error(`${error.message}\n${JSON.stringify(status)}\n${JSON.stringify(actions)}\n${await controlPage.locator('body').innerText()}`)
}
await submit
if (submitError) throw submitError
assert.equal(completedPosts, 1, 'the capture submission must not reach the server')
assert.equal(blockedPosts, 1, 'the browser must confirm cancellation of the capture submission')
const profile = profiles.find(value => value.request.enabled && value.response.enabled)
assert.equal(profiles.length, 1, 'both directions must share one gateway')
for (const plaintext of [
{ username: 'audit-new-user', password: 'new-wrong-password' },
{ username: 'audit-second-user', password: 'second-wrong-password' },
]) {
const packet = { method: 'POST', url: targetURL, headers: [{ name: 'Content-Type', value: 'application/json' }], bodyBase64: Buffer.from(JSON.stringify(plaintext)).toString('base64') }
const encrypted = await extensionRequest(controlPage, 'transform.execute', { profileId: profile.id, direction: 'request', packet })
const envelope = JSON.parse(Buffer.from(encrypted.bodyBase64, 'base64').toString())
assert.deepEqual(decrypt(envelope), plaintext, 'the complete new input must be encrypted')
const response = await fetch(targetURL, transformedFetchOptions(encrypted, packet.headers))
const wireResponse = await response.text()
const decrypted = await extensionRequest(controlPage, 'transform.execute', {
profileId: profile.id, direction: 'response', packet: { ...packet, bodyBase64: Buffer.from(wireResponse).toString('base64') },
})
assert.deepEqual(JSON.parse(Buffer.from(decrypted.bodyBase64, 'base64').toString()), decrypt(JSON.parse(wireResponse)))
}
assert.equal(completedPosts, 1, 'local gateway replay must not send browser requests')
assert.equal(await targetPage.locator('#username').inputValue(), 'audit-original')
assert.equal(await targetPage.locator('#password').inputValue(), 'original-wrong-password')
// Also pause inside the native Fetch boundary, where changing window.fetch is too late.
await extensionRequest(controlPage, 'deep.capture.start', {
...target, matcher: { kind: 'request', urlPattern: targetURL, frameHints: requestCandidate.capturePlan.frameHints },
})
const boundaryBlocked = targetPage.waitForEvent('requestfailed', {
predicate: request => request.url() === targetURL && request.failure()?.errorText.includes('BLOCKED_BY_CLIENT'),
timeout: 30_000,
})
void boundaryBlocked.catch(() => undefined)
const boundarySubmit = targetPage.locator('button[type=submit]').click({ noWaitAfter: true, timeout: 30_000 }).catch(error => { submitError = error })
const paused = await waitFor(controlPage, 'deep.capture.status', target, status => status.state === 'paused' && !status.pause.collecting)
assert.equal(paused.pause.automaticCapture.state, 'ready')
const boundaryCallable = await extensionRequest(controlPage, 'callable.create', {
...target, source: 'deep-capture', strategy: 'request-transaction',
callFrameId: paused.pause.automaticCapture.frameId, candidateId: requestCandidate.id,
})
await boundarySubmit
await boundaryBlocked
if (submitError) throw submitError
assert.equal(completedPosts, 1)
assert.equal(blockedPosts, 2)
await extensionRequest(controlPage, 'callable.delete', { ...target, callableId: boundaryCallable.id })
console.log('Login gateway passed: one recording, automatic request capture, browser-confirmed cancellation, one bidirectional profile, two distinct plaintexts and real server response decryption.')
} finally { await harness?.close() }
+102
View File
@@ -0,0 +1,102 @@
import assert from 'node:assert/strict';
import { mkdtemp, mkdir, rm } from 'node:fs/promises';
import { tmpdir } from 'node:os';
import { resolve, join } from 'node:path';
import { chromium } from 'playwright-core';
import { resolveChromiumPath } from './resolve-chromium.mjs';
// Uses an isolated profile; never changes the user's browser or system proxy.
const profile = await mkdtemp(join(tmpdir(), 'yakit-proxy-control-'));
const extension = resolve('.output/chrome-mv3');
const context = await chromium.launchPersistentContext(profile, {
executablePath: await resolveChromiumPath(), headless: true,
viewport: { width: 390, height: 640 }, reducedMotion: 'reduce',
args: [`--disable-extensions-except=${extension}`, `--load-extension=${extension}`, '--proxy-server=http://127.0.0.1:18083'],
});
try {
const worker = context.serviceWorkers()[0] || await context.waitForEvent('serviceworker');
const id = new URL(worker.url()).host;
const page = await context.newPage();
await page.goto(`chrome-extension://${id}/ytray-bootstrap.html?manager=ytray&instanceId=proxy-test&badge=A&startupProxy=${encodeURIComponent('http://127.0.0.1:18083')}&target=chrome://version`);
await page.waitForURL('chrome://version/');
await page.goto(`chrome-extension://${id}/options.html`);
const call = async (action, payload) => {
const response = await page.evaluate(({ action, payload }) => chrome.runtime.sendMessage({ action, payload }), { action, payload });
assert.equal(response.ok, true, response.error);
return response.data;
};
const launch = await call('proxy.status');
assert.equal((await call('state.get')).startupProxy, 'http://127.0.0.1:18083');
assert.equal(launch.followingStartup, true);
assert.equal(launch.control, 'controllable_by_this_extension');
assert.equal(launch.activeProfileId, undefined);
assert.match(launch.label, /18083/);
await call('proxy.switch', { id: 'direct' });
assert.equal((await call('proxy.status')).activeProfileId, 'direct');
await call('proxy.switch', { id: 'yakit-mitm' });
assert.equal((await call('proxy.status')).activeProfileId, 'yakit-mitm');
await call('proxy.auto.apply');
assert.equal((await call('proxy.status')).activeProfileId, 'auto');
await call('proxy.switch', { id: 'system' });
assert.equal((await call('proxy.status')).activeProfileId, 'system');
await call('proxy.release');
assert.deepEqual(await call('proxy.status'), launch);
await page.goto(`chrome-extension://${id}/popup.html`);
await page.getByRole('button', { name: '代理', exact: true }).click();
await page.getByRole('status').filter({ hasText: '实际代理' }).getByText('http://127.0.0.1:18083', { exact: true }).waitFor();
assert.equal(await page.getByRole('radio', { name: /直接连接/ }).getAttribute('aria-checked'), 'false');
const follow = page.getByRole('radio', { name: /跟随启动配置/ });
assert.equal(await follow.getAttribute('aria-checked'), 'true');
await page.evaluate(() => {
const startup = document.querySelector('.startup-proxy-option');
const ordinary = document.querySelector('.popup-proxy-list > button');
for (const [a, b] of [[startup.querySelector('.startup-proxy-icon'), ordinary.querySelector('.popup-mode-icon')], [startup.querySelector('strong'), ordinary.querySelector('strong')], [startup.querySelector('small'), ordinary.querySelector('small')]]) {
if (Math.abs(a.getBoundingClientRect().x - b.getBoundingClientRect().x) > 1) throw new Error('Proxy mode columns are not aligned');
}
});
await page.emulateMedia({ reducedMotion: 'no-preference' });
await page.getByRole('radio', { name: /直接连接/ }).click();
await page.locator('.popup-global-notice').waitFor();
await page.evaluate(() => {
const notice = document.querySelector('.popup-global-notice');
const animation = notice.getAnimations()[0];
if (!animation) throw new Error('Expected notice entrance animation');
animation.pause();
for (const time of [0, 40, 80, 159, 200]) {
animation.currentTime = time;
const rect = notice.getBoundingClientRect();
const parent = notice.offsetParent.getBoundingClientRect();
if (Math.abs(rect.x + rect.width / 2 - (parent.x + parent.width / 2)) > 1) {
throw new Error(`Notice is not centered at animation time ${time}ms`);
}
}
animation.finish();
});
await page.emulateMedia({ reducedMotion: 'reduce' });
await page.waitForFunction(() => document.querySelector('[aria-label="实际代理状态"]')?.textContent === '实际代理直接连接');
assert.equal(await follow.getAttribute('aria-checked'), 'false');
await follow.click();
await page.waitForFunction(() => document.querySelector('.startup-proxy-option [role="radio"]')?.getAttribute('aria-checked') === 'true');
assert.deepEqual(await call('proxy.status'), launch);
await page.getByRole('button', { name: '解释跟随启动配置' }).focus();
await page.getByRole('tooltip').waitFor();
assert.match(await page.getByRole('tooltip').innerText(), /切换后使用浏览器启动时的网络配置/);
assert.doesNotMatch(await page.getByRole('tooltip').innerText(), /清除|接管/);
await page.getByRole('radio', { name: /跟随启动配置/ }).focus();
await page.mouse.move(4, 4);
await mkdir('.artifacts/proxy', { recursive: true });
await page.screenshot({ path: '.artifacts/proxy/launch-proxy.png' });
// External settings changes must update an already-open view, without storage mutations.
await page.evaluate(() => chrome.proxy.settings.set({ scope: 'regular', value: { mode: 'direct' } }));
await page.getByRole('status').filter({ hasText: '实际代理' }).getByText('直接连接', { exact: true }).waitFor();
assert.equal(await page.getByRole('radio', { name: /直接连接/ }).getAttribute('aria-checked'), 'false');
await page.goto(`chrome-extension://${id}/ytray-bootstrap.html?manager=ytray&instanceId=direct-test&badge=A&startupProxy=direct&target=chrome://version`);
await page.waitForURL('chrome://version/');
await page.goto(`chrome-extension://${id}/popup.html`);
await page.getByRole('button', { name: '代理', exact: true }).click();
assert.equal(await page.getByRole('radio', { name: /跟随启动配置/ }).count(), 0);
console.log('PASS: launch proxy → direct → fixed → PAC → system → release; live status; stale selection not marked active.');
} finally {
await context.close();
await rm(profile, { recursive: true, force: true });
}
+2
View File
@@ -89,6 +89,8 @@ const manifest = JSON.parse(manifestBytes.toString('utf8'));
assert(manifest.latest === entry.version, `manifest.latest ${manifest.latest} != ${entry.version}`);
const versionEntry = manifest.versions.find((v) => v.version === entry.version);
assert(versionEntry, `manifest has no entry for version ${entry.version}`);
assert(JSON.stringify(versionEntry.notes) === JSON.stringify(entry.notes),
`manifest release notes do not match release entry for ${entry.version}`);
assert(versionEntry.artifacts.length === entry.artifacts.length,
`manifest artifacts count ${versionEntry.artifacts.length} != ${entry.artifacts.length}`);
for (const artifact of entry.artifacts) {
+4
View File
@@ -18,6 +18,8 @@ import {
saveProxyRuleSource,
setProxyAuthPassword,
switchProxy,
getProxyStatus,
releaseProxy,
} from '@/features/proxy/service';
import { updateState } from '@/platform/storage/state';
@@ -26,6 +28,8 @@ export const handleProxyRequest: BackgroundRequestHandler = async (request) => {
case 'proxy.save': return ok(await saveProxyProfile(request.payload));
case 'proxy.delete': return ok(await removeProxyProfile(request.payload.id));
case 'proxy.switch': return ok(await switchProxy(request.payload.id));
case 'proxy.status': return ok(await getProxyStatus());
case 'proxy.release': return ok(await releaseProxy());
case 'proxy.rule.save': {
const rule = request.payload;
return ok(await updateState((state) => {
+23 -7
View File
@@ -3,11 +3,16 @@ import { ok } from '../response';
import { requiredDebuggerTarget, requiredRequestTarget } from '../request-context';
import {
browserRecordingStatus,
clearTabBrowserRecording,
clearBrowserRecording,
createRecordedPageCallable,
getBrowserRecording,
getTabBrowserRecording,
startTabBrowserRecording,
startBrowserRecording,
stopTabBrowserRecording,
stopBrowserRecording,
tabBrowserRecordingStatus,
} from '@/features/browser-recording/service';
import {
createCapturedPageCallable,
@@ -35,7 +40,9 @@ export const handleRecordingRequest: BackgroundRequestHandler = async (request,
case 'recording.start': {
const input = request.payload;
const target = await requiredRequestTarget(input, sender);
const snapshot = await startBrowserRecording(target, input);
const snapshot = input.scope === 'tab'
? await startTabBrowserRecording(target.tabId, input)
: await startBrowserRecording(target, input);
void appendAuditEvent({
category: 'capability',
action: 'recording.start',
@@ -45,18 +52,25 @@ export const handleRecordingRequest: BackgroundRequestHandler = async (request,
});
return ok(snapshot);
}
case 'recording.status': return ok(await browserRecordingStatus(
await requiredRequestTarget(request.payload, sender),
));
case 'recording.status': {
const target = await requiredRequestTarget(request.payload, sender);
return ok(request.payload.scope === 'tab'
? await tabBrowserRecordingStatus(target.tabId)
: await browserRecordingStatus(target));
}
case 'recording.get': {
const target = await requiredRequestTarget(request.payload, sender);
const snapshot = await getBrowserRecording(target, request.payload.limit, true);
const snapshot = request.payload.scope === 'tab'
? await getTabBrowserRecording(target.tabId, request.payload.limit, true)
: await getBrowserRecording(target, request.payload.limit, true);
await stageBrowserProfileEvidence(snapshot);
return ok(snapshot);
}
case 'recording.clear': {
const target = await requiredRequestTarget(request.payload, sender);
const snapshot = await clearBrowserRecording(target, true);
const snapshot = request.payload.scope === 'tab'
? await clearTabBrowserRecording(target.tabId, true)
: await clearBrowserRecording(target, true);
void appendAuditEvent({
category: 'capability',
action: 'recording.clear',
@@ -67,7 +81,9 @@ export const handleRecordingRequest: BackgroundRequestHandler = async (request,
}
case 'recording.stop': {
const target = await requiredRequestTarget(request.payload, sender);
const snapshot = await stopBrowserRecording(target, true);
const snapshot = request.payload.scope === 'tab'
? await stopTabBrowserRecording(target.tabId, true)
: await stopBrowserRecording(target, true);
await stageBrowserProfileEvidence(snapshot);
void appendAuditEvent({
category: 'capability',
+23
View File
@@ -14,6 +14,7 @@ import {
validateBrowserTransformRecovery,
} from '@/features/browser-transform/service';
import {
discardBrowserTransformValidation,
latestBrowserTransformValidation,
proposeBrowserTransformProfile,
validateInferredBrowserTransformProfile,
@@ -60,6 +61,28 @@ export const handleTransformRequest: BackgroundRequestHandler = async (request,
await requiredRequestTarget(request.payload, sender),
),
);
case 'analysis.profile.validation.resolve': {
const input = request.payload;
const target = await requiredRequestTarget(input, sender);
const draft = await latestBrowserTransformValidation(target);
if (!draft || draft.id !== input.validationId) {
throw new Error('验证草稿不存在或已经过期,请重新生成并验证');
}
if (input.outcome === 'discard') {
await discardBrowserTransformValidation(target, draft.id);
return ok(null);
}
const profile = await saveBrowserTransformProfile(draft.profile);
await discardBrowserTransformValidation(target, draft.id);
void appendAuditEvent({
category: 'capability',
action: 'analysis.profile.validation.save',
outcome: 'success',
targetTabId: profile.target.tabId,
summary: profile.name,
});
return ok(profile);
}
case 'transform.profile.list': {
const input = request.payload;
const target = input.tabId ? await requiredRequestTarget(input, sender) : undefined;
+40 -26
View File
@@ -64,6 +64,7 @@ import { handleCookieRequest } from './handlers/cookies';
import { handleUserAgentRequest } from './handlers/user-agent';
import { handleRecordingRequest } from './handlers/recording';
import { handleTransformRequest } from './handlers/transform';
import { resolveHandoff } from '@/features/handoff/service';
function originOf(url: string): string {
const parsed = new URL(url);
@@ -163,11 +164,8 @@ async function handleRequest(request: ExtensionRequest, sender: Browser.runtime.
request.payload.timeoutMs,
));
}
case 'authorization.yakit.open':
return ok(await engineBridge.requestEngine(
'yakit.browser_authorization.open',
{ workspaceId: request.payload.workspaceId },
));
case 'authorization.yakit.instances':
return ok(await engineBridge.requestEngine('yakit.browser_authorization.instances', {}));
case 'context.capture': {
const { tabId, frameId, documentId, ...options } = request.payload;
const target = await requiredRequestTarget({ tabId, frameId, documentId }, sender);
@@ -299,21 +297,7 @@ async function handleRequest(request: ExtensionRequest, sender: Browser.runtime.
}
case 'handoff.resolve': {
const input = request.payload;
const state = await updateState((current) => {
if (!current.handoff || current.handoff.id !== input.id || current.handoff.state !== 'waiting_for_user') {
throw new ExtensionError('handoff_not_waiting', '人工接管请求不存在或已经结束');
}
return {
...current,
handoff: { ...current.handoff, state: input.outcome, resolvedAt: Date.now() },
};
});
const handoff = state.handoff!;
await setAgentRuntimeState(
input.outcome === 'completed' ? 'running' : 'paused',
await browserInstanceAccess('browser.tabs.read'),
);
await browser.action.setBadgeText({ text: '', tabId: handoff.target.tabId });
const { state, handoff } = await resolveHandoff(input.id, input.outcome);
engineBridge.emitEvent('browser.handoff.changed', handoff);
void appendAuditEvent({
category: 'handoff', action: `handoff.${input.outcome}`, outcome: input.outcome === 'completed' ? 'success' : 'cancelled',
@@ -444,21 +428,50 @@ async function handleRequest(request: ExtensionRequest, sender: Browser.runtime.
}
const state = await updateState((current) => ({
...current,
bridge: { ...current.bridge, managedInstance: request.payload },
startupProxy: request.payload.startupProxy,
bridge: {
...current.bridge,
browserName: request.payload.browserName,
browserVersion: request.payload.browserVersion,
managedInstance: {
manager: request.payload.manager, instanceId: request.payload.instanceId, badge: request.payload.badge,
},
},
}));
await syncManagedInstanceBadge(state.bridge.managedInstance);
if (state.bridge.autoConnect && state.bridge.pairedEngine) {
engineBridge.disconnect();
await stopPairedBrowserTasks();
await engineBridge.connect(state.bridge);
void stopPairedBrowserTasks()
.then(() => engineBridge.connect(state.bridge))
.catch((error) => console.error('Managed browser reconnect failed', error));
} else if (!state.bridge.pairedEngine) {
void engineBridge.startPairing()
.catch((error) => console.error('Managed browser pairing failed', error));
}
return ok(engineBridge.getStatus());
}
case 'bridge.pair': {
const status = await engineBridge.startPairing();
void appendAuditEvent({ category: 'bridge', action: 'bridge.pair', outcome: 'success' });
return ok(status);
const startedAt = Date.now();
try {
const status = await engineBridge.startPairing();
void appendAuditEvent({
category: 'bridge', action: 'bridge.pair',
outcome: status.state === 'rejected'
? 'denied'
: status.state === 'pending' || status.state === 'approved' ? 'success' : 'error',
durationMs: Date.now() - startedAt, summary: `${status.state}: ${status.message}`,
});
return ok(status);
} catch (error) {
void appendAuditEvent({
category: 'bridge', action: 'bridge.pair', outcome: 'error',
durationMs: Date.now() - startedAt, errorCode: errorCode(error),
summary: (error instanceof Error ? error.message : String(error)).slice(0, 512),
});
throw error;
}
}
case 'bridge.discover': return ok(await engineBridge.discoverLocalEngines());
case 'bridge.pair.cancel': return ok(engineBridge.cancelPairing());
case 'bridge.pair.status': return ok(engineBridge.getPairingStatus());
case 'bridge.unpair': {
@@ -528,6 +541,7 @@ export function runBackground(): void {
) => {
if ([
'bridge.status.changed',
'proxy.status.changed',
'bridge.pairing.status.changed',
'network.capture.changed',
'deep.capture.changed',
+2
View File
@@ -63,6 +63,7 @@ async function send<T>(action: string, payload?: unknown): Promise<T> {
export default defineContentScript({
matches: ['http://*/*', 'https://*/*'],
allFrames: true,
runAt: 'document_start',
async main(ctx) {
@@ -77,6 +78,7 @@ export default defineContentScript({
console.warn('[Yakit Browser Agent] MAIN-world bridge is unavailable; continuing without page Eval/Invoke.', error);
});
}
if (window.top !== window) return;
const host = document.createElement('yakit-browser-agent');
const shadow = host.attachShadow({ mode: 'open' });
+14
View File
@@ -75,6 +75,7 @@ input[type='checkbox'] { width: 15px; height: 15px; flex: 0 0 auto; padding: 0;
.sidebar-brand .product-brand { width: 100%; color: var(--foreground); }
.sidebar nav { min-height: 0; padding: 10px 10px 16px; overflow-y: auto; display: grid; gap: 10px; scrollbar-width: thin; }
.sidebar-group { display: grid; gap: 2px; }
.sidebar-group.is-primary { padding-bottom: 8px; border-bottom: 1px solid var(--border); }
.sidebar-group__label { min-height: 24px; padding: 0 10px; display: flex; align-items: center; gap: 6px; color: var(--muted); font-size: var(--text-xs); font-weight: 650; letter-spacing: .04em; }
.sidebar-group__label svg { color: var(--primary); }
.sidebar nav button { width: 100%; height: 40px; padding: 0 10px; display: grid; grid-template-columns: 20px 1fr 14px; align-items: center; gap: 8px; border: 0; border-radius: var(--radius-md); background: transparent; color: var(--muted-strong); font-size: var(--text-md); font-weight: 500; text-align: left; cursor: pointer; transition: background-color .14s ease, color .14s ease; }
@@ -483,6 +484,8 @@ input[type='checkbox'] { width: 15px; height: 15px; flex: 0 0 auto; padding: 0;
.recording-pipeline-step button > span:nth-child(2) { min-width: 0; }
.recording-pipeline-step small, .recording-pipeline-step strong, .recording-pipeline-step em, .recording-pipeline-step b { display: block; overflow: hidden; white-space: nowrap; text-overflow: ellipsis; }
.recording-pipeline-step small { color: var(--muted); font-size: 10px; font-weight: 650; }
.recording-pipeline-step small.is-request { color: var(--warning); }
.recording-pipeline-step small.is-response { color: var(--primary); }
.recording-pipeline-step strong { margin-top: 2px; font-size: var(--text-sm); font-weight: 650; }
.recording-pipeline-step em { margin-top: 2px; color: var(--muted); font-size: var(--text-xs); font-style: normal; }
.recording-pipeline-step b { margin-top: 4px; color: var(--warning); font-size: 10px; font-weight: 650; }
@@ -516,6 +519,12 @@ input[type='checkbox'] { width: 15px; height: 15px; flex: 0 0 auto; padding: 0;
.recording-evidence { border-top: 1px solid var(--border); border-bottom: 1px solid var(--border); }
.recording-evidence summary { padding: 9px 0; color: var(--muted-strong); font-size: var(--text-sm); font-weight: 600; cursor: pointer; }
.recording-evidence[open] { padding-bottom: 10px; }
.recording-related-transform { padding: 10px; display: flex; align-items: center; justify-content: space-between; gap: 10px; border: 1px solid var(--border); border-radius: var(--radius-sm); background: var(--surface-subtle); }
.recording-related-transform > div { min-width: 0; display: flex; align-items: center; gap: 8px; color: var(--primary); }
.recording-related-transform > div span { min-width: 0; }
.recording-related-transform strong, .recording-related-transform small { display: block; overflow: hidden; white-space: nowrap; text-overflow: ellipsis; }
.recording-related-transform strong { color: var(--foreground); font-size: var(--text-sm); }
.recording-related-transform small { margin-top: 2px; color: var(--muted); font-size: var(--text-xs); }
.profile-inference { padding: 11px 0 0; display: grid; gap: 10px; border-top: 2px solid var(--primary); }
.profile-inference.is-medium { border-top-color: var(--warning); }
.profile-inference.is-low { border-top-color: var(--border-strong); }
@@ -707,6 +716,11 @@ input[type='checkbox'] { width: 15px; height: 15px; flex: 0 0 auto; padding: 0;
.pairing-code span { color: var(--muted); font-size: var(--text-xs); font-weight: 650; letter-spacing: .05em; text-transform: uppercase; }
.pairing-code strong { font-family: var(--font-mono); font-size: 30px; font-weight: 700; letter-spacing: .12em; }
.pairing-code small { color: var(--muted); font-size: var(--text-sm); }
.engine-discovery-list { display: grid; gap: 8px; }
.engine-discovery-row { padding: 10px 12px; display: grid; grid-template-columns: 34px minmax(0, 1fr) auto; gap: 10px; align-items: center; border: 1px solid var(--border); border-radius: var(--radius-md); background: var(--surface-subtle); text-align: left; }
.engine-discovery-icon { width: 34px; height: 34px; display: grid; place-items: center; border-radius: 10px; color: var(--accent-strong); background: var(--accent-soft); }
.engine-discovery-row strong, .engine-discovery-row small { display: block; }
.engine-discovery-row small { margin-top: 2px; color: var(--muted); }
.paired-engine-meta { display: grid; grid-template-columns: 1fr 1fr; gap: 12px; }
.paired-engine-meta > div { min-width: 0; }
.paired-engine-meta span { color: var(--muted); font-size: var(--text-xs); font-weight: 650; letter-spacing: .04em; text-transform: uppercase; }
+73 -30
View File
@@ -2,8 +2,8 @@ import { useCallback, useEffect, useMemo, useState, type ReactNode } from 'react
import { browser, type Browser } from 'wxt/browser';
import {
Activity, AlertTriangle, Bot, Braces, Check, ChevronRight, CircleGauge, CloudDownload, Cookie, Copy,
Database, Download, Eye, Fingerprint, History, KeyRound, MousePointer2, Network, Play, Power, Radio,
RefreshCw, Route, Save, Search, Send, Server, ShieldCheck, Square, Trash2, Upload, UserRoundCog, Wrench, X,
Database, Download, Eye, FileKey2, Fingerprint, History, KeyRound, MousePointer2, Network, Play, Power, Radio,
RefreshCw, Route, Save, Search, Send, Server, ShieldCheck, Square, Trash2, Upload, UserRoundCog, X,
} from 'lucide-react';
import { ProductBrand, YakitMark } from '@/components/brand/Brand';
import { Button } from '@/components/ui/button';
@@ -16,12 +16,14 @@ import {
import { cookieKey, cookieRemovalInput } from '@/features/cookies/presentation';
import { AutoSwitchView } from '@/features/proxy/ui/AutoSwitchView';
import { ProxyProfilesView } from '@/features/proxy/ui/ProxyProfilesView';
import { useProxyStatus } from '@/features/proxy/ui/ProxyStatusBar';
import { RuleSourcesView } from '@/features/proxy/ui/RuleSourcesView';
import { RecordingWorkspace } from '@/features/browser-recording/RecordingWorkspace';
import { AuthorizationTestingWorkspace } from '@/features/authorization-testing/ui/AuthorizationTestingWorkspace';
import { AGENT_RUNTIME_STORAGE_KEY, AUDIT_STORAGE_KEY, isStateStorageChange } from '@/protocol/storage';
import type {
ActiveTabInfo, AgentRuntime, AuditEvent, BridgePairingStatus, BridgeStatus, BrowserCookie, BrowserRequestAnalysisBundle, CookieInput, CookieTransferFormat, EnterprisePolicyStatus, ExtensionState, HumanHandoff,
DiscoveredYakEngine,
NetworkCaptureStatus, NetworkRequestExport, NetworkRequestRecord, PageContext, PageEvalResult,
PageFrameSummary, PageNodeDetails, PageNodeSummary,
UserAgentProfile, UserAgentProfileInput, YakPocGenerateResult,
@@ -30,37 +32,43 @@ import { errorMessage, request } from '@/platform/messaging/runtime';
import { APPEARANCE_STORAGE_KEY, getAppearance, setThemePreference, type ThemePreference } from '@/platform/storage/appearance';
import './App.css';
type Section = 'overview' | 'authorization' | 'proxies' | 'rules' | 'sources' | 'cookies' | 'user-agent' | 'network' | 'context' | 'engine' | 'activity';
type Section = 'overview' | 'authorization' | 'network' | 'gateway' | 'proxies' | 'rules' | 'sources' | 'cookies' | 'user-agent' | 'context' | 'engine' | 'activity';
const FIREFOX_AMO_BUILD = import.meta.env.FIREFOX && import.meta.env.MODE === 'store';
const NAVIGATION: Array<{ label: string; icon?: ReactNode; items: Array<{ id: Section; label: string; icon: ReactNode }> }> = [
const NAVIGATION: Array<{ label?: string; items: Array<{ id: Section; label: string; icon: ReactNode }> }> = [
{
label: '工作区',
items: [{ id: 'overview', label: '概览', icon: <CircleGauge size={17} /> }],
},
{
label: '安全测试',
items: [{ id: 'authorization', label: '越权测试', icon: <Fingerprint size={17} /> }],
},
{
label: '请求与改写',
items: [
{ id: 'overview', label: '运行概览', icon: <CircleGauge size={17} /> },
{ id: 'authorization', label: '授权测试', icon: <Fingerprint size={17} /> },
{ id: 'network', label: '请求捕获', icon: <Activity size={17} /> },
{ id: 'gateway', label: '明文网关', icon: <FileKey2 size={17} /> },
],
},
{
label: '网络与流量',
label: '代理',
items: [
{ id: 'proxies', label: '代理出口', icon: <Network size={17} /> },
{ id: 'rules', label: '自动切换', icon: <Route size={17} /> },
{ id: 'proxies', label: '代理设置', icon: <Network size={17} /> },
{ id: 'rules', label: '分流规则', icon: <Route size={17} /> },
{ id: 'sources', label: '规则订阅', icon: <CloudDownload size={17} /> },
{ id: 'network', label: '网络活动', icon: <Activity size={17} /> },
],
},
{
label: '常用工具', icon: <Wrench size={13} />,
label: '浏览器工具',
items: [
{ id: 'cookies', label: 'Cookie Editor', icon: <Cookie size={17} /> },
{ id: 'user-agent', label: 'UA 快速切换', icon: <UserRoundCog size={17} /> },
{ id: 'context', label: '页面上下文', icon: <KeyRound size={17} /> },
{ id: 'cookies', label: 'Cookie 管理', icon: <Cookie size={17} /> },
{ id: 'user-agent', label: 'User-Agent', icon: <UserRoundCog size={17} /> },
],
},
{
label: 'Agent 与系统',
label: '系统',
items: [
{ id: 'context', label: '登录态工作区', icon: <KeyRound size={17} /> },
{ id: 'engine', label: '引擎连接', icon: <Server size={17} /> },
{ id: 'activity', label: '操作记录', icon: <History size={17} /> },
],
@@ -204,7 +212,7 @@ function App() {
<div className="app-shell">
<aside className="sidebar">
<div className="sidebar-brand"><ProductBrand /></div>
<nav>{NAVIGATION.map((group) => <div className="sidebar-group" key={group.label}><span className="sidebar-group__label">{group.icon}{group.label}</span>{group.items.map((item) => <button key={item.id} className={section === item.id ? 'active' : ''} onClick={() => navigate(item.id)}>{item.icon}<span>{item.label}</span><ChevronRight size={14} /></button>)}</div>)}</nav>
<nav>{NAVIGATION.map((group, index) => <div className={`sidebar-group ${index === 0 ? 'is-primary' : ''}`} key={group.label || 'overview'}>{group.label && <span className="sidebar-group__label">{group.label}</span>}{group.items.map((item) => <button key={item.id} className={section === item.id ? 'active' : ''} onClick={() => navigate(item.id)}>{item.icon}<span>{item.label}</span><ChevronRight size={14} /></button>)}</div>)}</nav>
<div className="sidebar-theme">
<span>外观</span>
<select aria-label="界面主题" value={theme} onChange={(event) => { const next = event.target.value as ThemePreference; setTheme(next); void setThemePreference(next); }}>
@@ -232,13 +240,14 @@ function App() {
<div className="content-area">
{section === 'overview' && <Overview state={state} bridge={bridge} tab={tab} navigate={navigate} run={run} busy={busy} />}
{section === 'authorization' && <AuthorizationTestingWorkspace state={state} setState={setState} tabs={tabs} activeTab={tab} bridge={bridge} refreshTabs={refreshTabs} run={run} busy={busy} />}
{section === 'authorization' && <AuthorizationTestingWorkspace bridge={bridge} run={run} busy={busy} />}
{section === 'proxies' && <ProxyProfilesView state={state} setState={setState} run={run} busy={busy} tab={tab} />}
{section === 'rules' && <AutoSwitchView state={state} setState={setState} tab={tab} run={run} busy={busy} />}
{section === 'sources' && <RuleSourcesView state={state} setState={setState} tab={tab} run={run} busy={busy} />}
{section === 'cookies' && <CookieEditor key={tab?.id || 0} tab={tab} run={run} busy={busy} />}
{section === 'user-agent' && <UserAgents state={state} setState={setState} tab={tab} run={run} busy={busy} />}
{section === 'network' && <NetworkActivity key={tab?.id || 0} tab={tab} bridge={bridge} run={run} busy={busy} />}
{section === 'gateway' && <GatewayWorkspace key={tab?.id || 0} tab={tab} bridge={bridge} run={run} busy={busy} />}
{section === 'context' && <ContextTool key={tab?.id || 0} tab={tab} run={run} busy={busy} />}
{section === 'engine' && <EngineSettings state={state} setState={setState} bridge={bridge} setBridge={setBridge} tabs={tabs} run={run} busy={busy} />}
{section === 'activity' && <ActivityLog run={run} busy={busy} />}
@@ -330,7 +339,7 @@ function ActivityLog({ run, busy }: { run: (task: () => Promise<void>, success?:
}
function Overview({ state, bridge, tab, navigate, run, busy }: { state: ExtensionState; bridge: BridgeStatus; tab?: ActiveTabInfo; navigate: (value: Section) => void; run: (task: () => Promise<void>, success?: string) => Promise<void>; busy: boolean }) {
const activeProxy = state.proxyProfiles.find((profile) => profile.id === state.activeProxyId)?.name || (state.activeProxyId === 'auto' ? '自动切换' : '未知');
const activeProxy = useProxyStatus(state).label;
const [runtime, setRuntime] = useState<AgentRuntime>({ state: 'idle', updatedAt: Date.now(), actions: [] });
const [network, setNetwork] = useState<NetworkCaptureStatus>();
const [loginContext, setLoginContext] = useState<PageContext>();
@@ -537,13 +546,6 @@ function NetworkActivity({
const [captureHeaders, setCaptureHeaders] = useState(false);
const [captureBody, setCaptureBody] = useState(false);
const [query, setQuery] = useState('');
const transformShared = bridge.state === 'connected';
const shareTransform = async () => {
if (!tab) throw new Error('请先选择需要使用的页面');
if (bridge.state !== 'connected') await request('bridge.connect');
};
const load = useCallback(async () => {
if (!tab) return;
try {
@@ -655,11 +657,31 @@ function NetworkActivity({
</aside>
</div>}
</div>;
}
function GatewayWorkspace({
tab,
bridge,
run,
busy,
}: {
tab?: ActiveTabInfo;
bridge: BridgeStatus;
run: (task: () => Promise<void>, success?: string) => Promise<void>;
busy: boolean;
}) {
const shareTransform = async () => {
if (!tab) throw new Error('请先选择需要使用的页面');
if (bridge.state !== 'connected') await request('bridge.connect');
};
return <div className="section-view gateway-view">
<RecordingWorkspace
tab={tab}
busy={busy}
run={run}
gatewayShared={transformShared}
gatewayShared={bridge.state === 'connected'}
onShareGateway={shareTransform}
/>
</div>;
@@ -789,7 +811,8 @@ function ContextTool({ tab, run, busy }: { tab?: ActiveTabInfo; run: (task: () =
function EngineSettings({ state, setState, bridge, setBridge, tabs, run, busy }: { state: ExtensionState; setState: (state: ExtensionState) => void; bridge: BridgeStatus; setBridge: (status: BridgeStatus) => void; tabs: ActiveTabInfo[]; run: (task: () => Promise<void>, success?: string) => Promise<void>; busy: boolean }) {
const [draft, setDraft] = useState(state.bridge);
const [pairing, setPairing] = useState<BridgePairingStatus>({ state: 'idle', message: state.bridge.pairedEngine ? '当前浏览器已配对' : '尚未配对' });
const [pairing, setPairing] = useState<BridgePairingStatus>({ state: 'idle', message: state.bridge.pairedEngine ? '当前浏览器已配对' : '点击下方按钮查找正在运行的 Yakit' });
const [engines, setEngines] = useState<DiscoveredYakEngine[]>([]);
const [panelDraft, setPanelDraft] = useState(state.floatingPanel);
const [policy, setPolicy] = useState<EnterprisePolicyStatus>({ managed: false, policy: {}, warnings: [] });
useEffect(() => {
@@ -817,8 +840,27 @@ function EngineSettings({ state, setState, bridge, setBridge, tabs, run, busy }:
setState(next);
setPanelDraft(next.floatingPanel);
}, '悬浮面板策略已保存');
const pairWithEngine = async (engine: DiscoveredYakEngine) => {
const next = await request('bridge.config.save', { ...state.bridge, transport: 'websocket', endpoint: engine.endpoint });
setState(next);
setDraft(next.bridge);
setEngines([]);
setPairing(await request('bridge.pair'));
};
const findLocalEngines = () => run(async () => {
setEngines([]);
const found = await request('bridge.discover');
if (found.length === 1) {
await pairWithEngine(found[0]);
} else if (found.length > 1) {
setEngines(found);
setPairing({ state: 'idle', message: `发现 ${found.length} 个正在运行的 Yakit,请选择一个连接` });
} else {
setPairing(await request('bridge.pair'));
}
});
return <div className="section-view engine-view">
<div className="page-heading"><div><h1>Yak 引擎连接</h1><p>扩展主动连接本机 Bridge,网页无法直接访问此通道。</p></div><span className={`large-status ${bridge.state}`}><Radio size={16} />{bridge.message}</span></div>
<div className="page-heading"><div><h1>连接 Yakit</h1><p>查找并连接正在本机运行的 Yakit。</p></div><span className={`large-status ${bridge.state}`}><Radio size={16} />{bridge.message}</span></div>
{policy.managed && <div className="managed-policy-banner"><ShieldCheck size={16} /><span><strong>此浏览器由组织策略管理</strong><small>{policy.policy.disableWebSocket ? '必须使用 Native Messaging' : policy.policy.bridgeTransport ? `传输锁定为 ${policy.policy.bridgeTransport}` : '连接与授权限制已应用'}{policy.policy.maxGrantMinutes ? ` · 授权最长 ${policy.policy.maxGrantMinutes} 分钟` : ''}{policy.policy.allowProgramEval === false ? ' · 程序 Eval 已禁用' : ''}</small>{policy.warnings.map((warning) => <i key={warning}>{warning}</i>)}</span></div>}
{bridge.state === 'connected' && <div className="bridge-identity-strip"><div><span>引擎实例</span><code title={bridge.engineInstanceId}>{bridge.engineInstanceId?.slice(0, 18)}</code></div><div><span>连接</span><code title={bridge.connectionId}>{bridge.connectionId?.slice(0, 18)}</code></div><div><span>会话</span><code title={bridge.sessionId}>{bridge.sessionId?.slice(0, 18)}</code></div><div><span>心跳</span><strong>{bridge.latencyMs === undefined ? '等待首个回执' : `${bridge.latencyMs} ms`}</strong></div><div><span>恢复</span><strong>{bridge.resumed ? '已恢复 task session' : '新会话'}</strong></div></div>}
<div className="engine-layout"><div className="settings-form">
@@ -826,8 +868,9 @@ function EngineSettings({ state, setState, bridge, setBridge, tabs, run, busy }:
<div className="pairing-workspace__heading"><span className="pairing-icon"><KeyRound size={19} /></span><div><h2>{state.bridge.pairedEngine ? '浏览器已安全配对' : pairing.state === 'pending' ? '等待 Yakit 确认' : '连接本机 Yakit'}</h2><p>{state.bridge.pairedEngine ? '设备身份已锁定到首次批准的 Yak 引擎。' : pairing.message}</p></div></div>
{pairing.state === 'pending' && <div className="pairing-code" aria-live="polite"><span>配对验证码</span><strong>{pairing.code?.slice(0, 3)} {pairing.code?.slice(3)}</strong><small>{pairing.expiresAt ? `${Math.max(0, Math.ceil((pairing.expiresAt - Date.now()) / 1000))} 秒内有效` : ''}</small></div>}
{state.bridge.pairedEngine && <div className="paired-engine-meta"><div><span>引擎身份</span><code title={state.bridge.pairedEngine.engineIdentityId}>{state.bridge.pairedEngine.engineIdentityId.slice(0, 24)}</code></div><div><span>设备 ID</span><code title={state.bridge.pairedEngine.deviceId}>{state.bridge.pairedEngine.deviceId.slice(0, 24)}</code></div></div>}
{engines.length > 1 && <div className="engine-discovery-list" aria-live="polite">{engines.map((engine, index) => <div className="engine-discovery-row" key={`${engine.engineIdentityId}:${engine.engineInstanceId}`}><span className="engine-discovery-icon"><Server size={17} /></span><div><strong>本机 Yakit {index + 1}</strong><small>端口 {new URL(engine.endpoint).port}</small></div><Button size="sm" variant="primary" disabled={busy} onClick={() => void run(() => pairWithEngine(engine))}>连接</Button></div>)}</div>}
<div className="editor-actions">
{!state.bridge.pairedEngine && pairing.state !== 'pending' && <Button variant="primary" disabled={busy || pairing.state === 'requesting'} onClick={() => void run(async () => setPairing(await request('bridge.pair')))}><Power size={16} />{pairing.state === 'requesting' ? '正在查找' : '查找本机 Yakit'}</Button>}
{!state.bridge.pairedEngine && pairing.state !== 'pending' && <Button variant="primary" disabled={busy || pairing.state === 'requesting'} onClick={() => void findLocalEngines()}><Search size={16} />{busy ? '正在查找' : '查找本机 Yakit'}</Button>}
{!state.bridge.pairedEngine && pairing.state === 'pending' && <Button disabled={busy} onClick={() => void run(async () => setPairing(await request('bridge.pair.cancel')))}><X size={16} />取消申请</Button>}
{state.bridge.pairedEngine && <Button variant="primary" disabled={busy} onClick={() => void run(async () => { if (bridge.state === 'connected') await request('bridge.disconnect'); else await request('bridge.connect'); setBridge(await request('bridge.status')); }, bridge.state === 'connected' ? 'Bridge 已断开' : 'Bridge 正在连接')}><Power size={16} />{bridge.state === 'connected' ? '断开连接' : '连接引擎'}</Button>}
{state.bridge.pairedEngine && <Button variant="ghost" disabled={busy} onClick={() => { if (window.confirm('解除当前 Yak 引擎的本地配对?浏览器安装身份会保留,重新配对时 Yakit 将更新原可信记录。')) void run(async () => { const next = await request('bridge.unpair'); setState(next); setDraft(next.bridge); }, '本地配对凭据已清除'); }}><Trash2 size={16} />解除配对</Button>}
+53 -20
View File
@@ -7,7 +7,7 @@ import {
type PageRecorderBridgeResponse,
} from '@/features/browser-recording/bridge-protocol';
import { PAGE_CALLABLE_REGISTRY_KEY } from '@/features/page-callable/constants';
import { executeRequestTransaction, executeSideEffectFreeCallable } from '@/features/page-callable/request-transaction';
import { executeRequestTransaction, executeSideEffectFreeCallable, observeCallableInput } from '@/features/page-callable/request-transaction';
import { callableExecutionPolicy, settleCallableResult } from '@/features/page-callable/execution';
import {
createCryptoAdapterRuntime,
@@ -48,6 +48,7 @@ import {
type RecordingTraceContext,
type RecordingTraceRuntime,
} from '@/features/browser-recording/main-world/trace';
import { recordingExpiryDelay } from '@/features/browser-recording/expiry';
import { RetainedCallBudget } from '@/features/browser-recording/main-world/retained-call-budget';
import { estimateRetainedCallBytes } from '@/features/browser-recording/main-world/retained-value-size';
import { ExtensionError } from '@/shared/errors';
@@ -199,8 +200,13 @@ interface RecordedCallHandle {
original: Function;
thisArg: unknown;
args: unknown[];
inputIndex: number;
originalInput: unknown;
replayInputs: Array<{
path: string;
name: string;
role: CallArgumentRole;
originalInput: unknown;
apply(args: unknown[], value: unknown): void;
}>;
eventId?: string;
traceId?: string;
recordingId?: string;
@@ -208,7 +214,6 @@ interface RecordedCallHandle {
outputDataType?: string;
outputEncoding?: PageCallableMetadata['output']['encoding'];
resultMode: 'sync' | 'promise';
adaptInput(value: unknown): unknown;
}
interface RecorderController {
@@ -484,6 +489,7 @@ export default defineUnlistedScript(() => {
}
const started = performance.now();
const inputIndex = plan.inputIndex;
if (inputIndex >= 0) observeCallableInput(args[inputIndex]);
const callHandleId = plan.callableKind && inputIndex >= 0 ? registerHandle({
kind: plan.callableKind,
operation: `${plan.crypto.adapterId}.${plan.crypto.operation}`,
@@ -491,11 +497,17 @@ export default defineUnlistedScript(() => {
original,
thisArg,
args: [...args],
inputIndex,
originalInput: args[inputIndex],
replayInputs: plan.replayInputs || [{
path: '$input',
name: 'data',
role: 'data',
originalInput: args[inputIndex],
apply: (nextArgs, value) => {
nextArgs[inputIndex] = (plan.adaptInput || ((input) => defaultAdaptInput(input, args[inputIndex])))(value);
},
}],
outputEncoding: plan.outputEncoding || plan.crypto.outputEncoding,
resultMode: operation.resultMode,
adaptInput: plan.adaptInput || ((value) => defaultAdaptInput(value, args[inputIndex])),
}) : undefined;
const item = observe(() => ({
kind: 'crypto',
@@ -673,7 +685,14 @@ export default defineUnlistedScript(() => {
if (active || !startedAt) return;
active = true;
installObservers();
if (options.expiresAt) expiryTimer = window.setTimeout(stop, Math.max(0, options.expiresAt - Date.now()));
scheduleExpiry();
}
function scheduleExpiry(): void {
const delay = recordingExpiryDelay(options.expiresAt);
if (delay === undefined) return;
if (delay === 0) { stop(); return; }
expiryTimer = window.setTimeout(stop, delay);
}
function snapshot(limit = options.maxEntries): RecorderSnapshot {
@@ -720,8 +739,13 @@ export default defineUnlistedScript(() => {
return value;
}
function createRecordedCallable(handle: RecordedCallHandle, name: string): PageCallableMetadata {
function createRecordedCallable(handle: RecordedCallHandle, name: string, inputPaths?: string[]): PageCallableMetadata {
const id = unique('callable');
const replayInputs = inputPaths?.length
? inputPaths.map((path) => handle.replayInputs.find((item) => item.path === path))
: [handle.replayInputs[0]];
if (replayInputs.some((item) => !item)) throw new Error('页面调用不支持请求的动态输入');
const resolvedInputs = replayInputs as RecordedCallHandle['replayInputs'];
const metadata: PageCallableMetadata = {
id,
name: name.trim().slice(0, 120) || handle.operation,
@@ -732,15 +756,15 @@ export default defineUnlistedScript(() => {
origin: location.origin,
lifecycle: 'document',
execution: callableExecutionPolicy(handle.resultMode),
inputSlots: [{
id: 'data',
name: 'data',
index: 0,
role: 'data',
dataType: dataType(handle.originalInput),
inputSlots: resolvedInputs.map((input, index) => ({
id: input.name,
name: input.name,
index,
role: input.role,
dataType: dataType(input.originalInput),
required: true,
retained: false,
}],
})),
output: {
dataType: handle.outputDataType || 'unknown',
encoding: handle.outputEncoding || 'auto',
@@ -759,9 +783,9 @@ export default defineUnlistedScript(() => {
pageCallableRegistry().set(id, {
metadata,
invoke(values) {
if (!values.length) throw new Error('页面函数缺少 data 参数');
if (values.length < resolvedInputs.length) throw new Error(`页面函数需要 ${resolvedInputs.length} 个动态参数`);
const args = [...handle.args];
args[handle.inputIndex] = handle.adaptInput(values[0]);
resolvedInputs.forEach((input, index) => input.apply(args, values[index]));
return Reflect.apply(handle.original, handle.thisArg, args);
},
});
@@ -778,6 +802,11 @@ export default defineUnlistedScript(() => {
logicalInput: values[0],
invoke: (context) => entry.invoke(values, context),
timeoutMs: entry.metadata.execution.timeoutMs,
observeInputs: () => {
const recording = active || Boolean(deepBreakMatcher);
cryptoAdapterRuntime.start();
return () => { if (!recording) cryptoAdapterRuntime.stop(); };
},
})
: entry.metadata.kind === 'business-closure' || entry.metadata.kind === 'global-function'
? await executeSideEffectFreeCallable(() => entry.invoke(values), entry.metadata.execution)
@@ -863,7 +892,7 @@ export default defineUnlistedScript(() => {
reseedFingerprints();
active = true;
installObservers();
if (options.expiresAt) expiryTimer = window.setTimeout(stop, Math.max(0, options.expiresAt - Date.now()));
scheduleExpiry();
return snapshot();
}
if (command === 'resume') {
@@ -933,7 +962,11 @@ export default defineUnlistedScript(() => {
const callHandleId = String(input.callHandleId || '');
const handle = handles.get(callHandleId);
if (!handle) throw new Error('加解密调用句柄不存在或已经失效');
return createRecordedCallable(handle, String(input.name || handle.operation));
return createRecordedCallable(
handle,
String(input.name || handle.operation),
Array.isArray(input.dynamicInputPaths) ? input.dynamicInputPaths.map(String) : undefined,
);
}
if (command === 'callable.list') return callableMetadata();
if (command === 'callable.execute') {
+3 -1
View File
@@ -206,7 +206,9 @@
.popup-footer { margin-top: auto; padding: 10px 14px 12px; border-top: 1px solid var(--border); background: var(--surface); }
.popup-capture { width: 100%; height: 36px; border-radius: 7px; font-size: var(--text-md); box-shadow: 0 1px 0 color-mix(in srgb, var(--primary-strong) 55%, transparent); }
.popup-notice { display: block; margin-top: 6px; overflow: hidden; color: var(--muted); font-size: var(--text-sm); line-height: 16px; text-align: center; white-space: nowrap; text-overflow: ellipsis; }
.popup-global-notice { position: absolute; z-index: 20; left: 50%; bottom: 54px; max-width: calc(100% - 28px); padding: 7px 11px; overflow: hidden; border: 1px solid var(--border); border-radius: 999px; background: var(--foreground); color: var(--surface); box-shadow: var(--shadow-md); font-size: var(--text-xs); line-height: 16px; text-overflow: ellipsis; white-space: nowrap; pointer-events: none; transform: translateX(-50%); animation: popup-content-in .16s ease-out; }
/* Keep horizontal centering independent of the entrance animation's transform. */
.popup-global-notice { position: absolute; z-index: 20; left: 50%; bottom: 54px; max-width: calc(100% - 28px); padding: 7px 11px; overflow: hidden; border: 1px solid var(--border); border-radius: 999px; background: var(--foreground); color: var(--surface); box-shadow: var(--shadow-md); font-size: var(--text-xs); line-height: 16px; text-overflow: ellipsis; white-space: nowrap; pointer-events: none; translate: -50% 0; animation: popup-content-in .16s ease-out; }
@media (prefers-reduced-motion: reduce) { .popup-global-notice { animation: none; } }
.popup-loading { width: 390px; height: 230px; display: flex; align-items: center; justify-content: center; gap: 9px; color: var(--muted); background: var(--background); font-size: var(--text-md); }
.spin { animation: spin .8s linear infinite; }
-1
View File
@@ -110,7 +110,6 @@ function App() {
const toggleEngine = () => run(async () => {
if (!state!.bridge.pairedEngine) {
await request('bridge.pair');
await openTool('engine');
return;
}
@@ -2,6 +2,7 @@ import { Braces, ChevronRight, Cookie, Network, Radio, ShieldCheck, UserRoundCog
import { Button } from '@/components/ui/button';
import { Switch } from '@/components/ui/switch';
import { request } from '@/platform/messaging/runtime';
import { useProxyStatus } from '@/features/proxy/ui/ProxyStatusBar';
import { READ_CAPABILITY_SCOPES } from '@/protocol/capabilities';
import type { ActiveTabInfo, ExtensionState, UserAgentResolution } from '@/types/models';
@@ -25,9 +26,7 @@ interface OverviewQuickViewProps {
export function OverviewQuickView({
state, tab, grantActive, busy, run, setState, cookieCount, uaResolution, onNavigate, onOpenContext, onCapture,
}: OverviewQuickViewProps) {
const activeProxy = state.activeProxyId === 'auto'
? '自动切换'
: state.proxyProfiles.find((profile) => profile.id === state.activeProxyId)?.name || '未选择';
const activeProxy = useProxyStatus(state).label;
const targetAvailable = Boolean(tab?.url?.startsWith('http'));
return <section className="popup-overview-view">
@@ -2,6 +2,7 @@ import { useEffect, useMemo, useState } from 'react';
import { AlertCircle, Check, ExternalLink, Globe2, LoaderCircle, Network, Route } from 'lucide-react';
import { request } from '@/platform/messaging/runtime';
import type { ActiveTabInfo, ExtensionState, ProxyProfile, ProxyRulePreview } from '@/types/models';
import { ProxyStatusBar, StartupProxyOption, useProxyStatus } from '@/features/proxy/ui/ProxyStatusBar';
type RunTask = (task: () => Promise<void>, success?: string) => Promise<void>;
@@ -43,10 +44,11 @@ function routeKindLabel(preview?: ProxyRulePreview): string {
}
export function ProxyQuickView({ state, setState, busy, run, tab, onOpenFull }: ProxyQuickViewProps) {
const status = useProxyStatus(state);
const [preview, setPreview] = useState<ProxyRulePreview>();
const currentHostname = hostname(tab?.url);
const autoActive = state.activeProxyId === 'auto';
const activeProfile = state.proxyProfiles.find((profile) => profile.id === state.activeProxyId);
const autoActive = status.activeProfileId === 'auto';
const activeProfile = state.proxyProfiles.find((profile) => profile.id === status.activeProfileId);
const routableProfiles = useMemo(
() => state.proxyProfiles.filter((profile) => profile.kind === 'direct' || profile.kind === 'fixed_servers'),
[state.proxyProfiles],
@@ -130,7 +132,7 @@ export function ProxyQuickView({ state, setState, busy, run, tab, onOpenFull }:
};
const effectiveProfile = state.proxyProfiles.find((profile) => profile.id === preview?.effectiveProfileId);
const activeModeName = autoActive ? '自动切换' : activeProfile?.name || '未选择';
const activeModeName = autoActive ? '自动切换' : status.label;
const siteHint = !autoActive
? `当前使用“${activeModeName}”;选择网站出口后将启用自动切换。`
: siteTarget === AUTOMATIC_TARGET
@@ -142,13 +144,14 @@ export function ProxyQuickView({ state, setState, busy, run, tab, onOpenFull }:
const routeKindText = autoActive ? routeKindLabel(preview) : '全局模式';
return <section className="popup-view popup-tool-view popup-proxy-view">
<ProxyStatusBar status={status} />
{currentHostname ? <section className="popup-site-router" aria-label="当前站点路由">
<div className="popup-site-router__heading">
<div><Globe2 size={16} /><span><small>当前站点</small><strong title={currentHostname}>{currentHostname}</strong></span></div>
<i className={routeKind}>{routeKindText}</i>
</div>
<div className="popup-site-decision" title={autoActive ? preview?.matchedCondition : activeModeName}>
<span>{routeLabel}</span><i>→</i><strong>{routeProfile?.name || '—'}</strong>
<span>{routeLabel}</span><i>→</i><strong>{routeProfile?.name || status.label}</strong>
</div>
<div className="popup-site-picker">
<label htmlFor="popup-site-proxy">网站出口 <span>选择后立即生效</span></label>
@@ -168,13 +171,14 @@ export function ProxyQuickView({ state, setState, busy, run, tab, onOpenFull }:
<div className="popup-mode-heading"><span><strong>浏览器模式</strong><small>全局切换,不会创建站点规则</small></span><i>{activeModeName}</i></div>
<div className="popup-proxy-list popup-proxy-list--view" role="radiogroup" aria-label="浏览器代理模式">
<StartupProxyOption state={state} status={status} setState={setState} run={run} busy={busy} />
<button role="radio" aria-checked={autoActive} className={autoActive ? 'is-active' : ''} disabled={busy} onClick={() => void switchAuto()}>
<span className="popup-mode-icon"><Route size={15} /></span>
<span><strong>自动切换</strong><small>{state.proxyRules.filter((rule) => rule.enabled).length} 条手动 · {sourceRuleCount.toLocaleString()} 条订阅</small></span>
{state.proxyRuntime.dirty ? <em>待应用</em> : autoActive ? <Check size={14} /> : null}
</button>
{state.proxyProfiles.map((profile) => {
const active = state.activeProxyId === profile.id;
const active = status.activeProfileId === profile.id;
return <button key={profile.id} role="radio" aria-checked={active} className={active ? 'is-active' : ''} disabled={busy} onClick={() => void run(async () => setState(await request('proxy.switch', { id: profile.id })), `${profile.name} 已作为全局模式启用`)}>
<span className="popup-mode-icon"><Network size={15} /></span>
<span><strong>{profile.name}</strong><small>{proxyDetail(profile)}</small></span>
+3
View File
@@ -25,6 +25,9 @@ async function bootstrap(): Promise<void> {
await request('bridge.managed-instance.bind', {
manager: manager as 'ytray' | 'yakit', instanceId, badge,
browserName: query.get('browserName') || undefined,
browserVersion: query.get('browserVersion') || undefined,
startupProxy: query.get('startupProxy') || undefined,
});
const current = await browser.tabs.getCurrent();
@@ -1,173 +0,0 @@
import { browser } from 'wxt/browser';
import type {
BrowserAuthContextAttestation,
BrowserTarget,
} from '@/types/models';
import { ExtensionError } from '@/shared/errors';
import {
AUTH_CONTEXT_TTL_MS,
captureAuthContextSnapshot,
validateAuthContextBinding,
} from './auth-context';
const MAX_ATTESTATIONS = 32;
const MAX_ATTESTATION_STORAGE_BYTES = 64 * 1_024;
const STORAGE_KEY = 'browser.authorization.auth-attestations.v1';
const attestations = new Map<string, BrowserAuthContextAttestation>();
let loaded = false;
function validStoredAttestation(value: unknown): value is BrowserAuthContextAttestation {
if (!value || typeof value !== 'object' || Array.isArray(value)) return false;
const attestation = value as Partial<BrowserAuthContextAttestation>;
return attestation.version === 1
&& typeof attestation.id === 'string'
&& attestation.id.length > 0
&& attestation.id.length <= 160
&& typeof attestation.deviceId === 'string'
&& attestation.deviceId.length > 0
&& attestation.deviceId.length <= 320
&& typeof attestation.installationId === 'string'
&& attestation.installationId.length > 0
&& attestation.installationId.length <= 320
&& typeof attestation.isolationContextId === 'string'
&& attestation.isolationContextId.length > 0
&& attestation.isolationContextId.length <= 320
&& typeof attestation.cookieStoreId === 'string'
&& attestation.cookieStoreId.length > 0
&& attestation.cookieStoreId.length <= 320
&& typeof attestation.origin === 'string'
&& attestation.origin.length > 0
&& attestation.origin.length <= 8_192
&& typeof attestation.grantId === 'string'
&& attestation.grantId.length > 0
&& attestation.grantId.length <= 160
&& typeof attestation.fingerprint === 'string'
&& /^hmac-sha256:[a-f0-9]{64}$/.test(attestation.fingerprint)
&& Boolean(attestation.target)
&& Number.isSafeInteger(attestation.target?.tabId)
&& Number(attestation.target?.tabId) > 0
&& Number.isSafeInteger(attestation.target?.frameId)
&& Number(attestation.target?.frameId) >= 0
&& typeof attestation.target?.documentId === 'string'
&& attestation.target.documentId.length > 0
&& attestation.target.documentId.length <= 160
&& Boolean(attestation.authentication)
&& ['authenticated', 'unauthenticated', 'unknown'].includes(String(attestation.authentication?.status))
&& Number.isSafeInteger(attestation.authentication?.cookieCount)
&& Number(attestation.authentication?.cookieCount) >= 0
&& Number.isSafeInteger(attestation.authentication?.storageEntryCount)
&& Number(attestation.authentication?.storageEntryCount) >= 0
&& Array.isArray(attestation.authentication?.authCookieNames)
&& attestation.authentication.authCookieNames.length <= 100
&& attestation.authentication.authCookieNames.every(
(name) => typeof name === 'string' && name.length <= 500,
)
&& Array.isArray(attestation.authentication?.authStorageKeys)
&& attestation.authentication.authStorageKeys.length <= 100
&& attestation.authentication.authStorageKeys.every(
(key) => typeof key === 'string' && key.length <= 520,
)
&& typeof attestation.createdAt === 'number'
&& typeof attestation.expiresAt === 'number'
&& attestation.expiresAt > attestation.createdAt
&& attestation.expiresAt - attestation.createdAt <= AUTH_CONTEXT_TTL_MS;
}
function purge(now = Date.now(), reserve = 0): boolean {
let changed = false;
for (const [id, attestation] of attestations) {
if (attestation.expiresAt <= now) {
attestations.delete(id);
changed = true;
}
}
while (attestations.size > MAX_ATTESTATIONS - reserve) {
const oldest = attestations.keys().next().value as string | undefined;
if (!oldest) break;
attestations.delete(oldest);
changed = true;
}
return changed;
}
async function load(): Promise<void> {
if (loaded) return;
loaded = true;
try {
const stored = await browser.storage.session.get(STORAGE_KEY);
const values = stored[STORAGE_KEY];
if (!Array.isArray(values)) return;
for (const value of values.slice(-MAX_ATTESTATIONS)) {
if (validStoredAttestation(value)) attestations.set(value.id, value);
}
purge();
} catch {
// The bounded in-memory registry remains valid for this service-worker lifetime.
}
}
async function save(): Promise<void> {
try {
const retained: BrowserAuthContextAttestation[] = [];
for (const attestation of [...attestations.values()].reverse()) {
const candidate = [attestation, ...retained];
if (new TextEncoder().encode(JSON.stringify(candidate)).byteLength > MAX_ATTESTATION_STORAGE_BYTES) break;
retained.unshift(attestation);
}
attestations.clear();
for (const attestation of retained) attestations.set(attestation.id, attestation);
await browser.storage.session.set({ [STORAGE_KEY]: retained });
} catch {
// The bounded in-memory registry remains available when storage.session cannot persist.
}
}
export async function captureAuthContextAttestation(input: {
target: BrowserTarget;
grantId: string;
grantExpiresAt: number;
}): Promise<BrowserAuthContextAttestation> {
await load();
const now = Date.now();
const snapshot = await captureAuthContextSnapshot(input.target);
const attestation: BrowserAuthContextAttestation = {
version: 1,
id: crypto.randomUUID(),
...snapshot,
grantId: input.grantId,
createdAt: now,
expiresAt: Math.min(now + AUTH_CONTEXT_TTL_MS, input.grantExpiresAt),
};
if (attestation.expiresAt <= now) {
throw new ExtensionError('grant_expired', '浏览器共享会话已经过期');
}
purge(now, 1);
attestations.set(attestation.id, attestation);
await save();
return attestation;
}
export async function getAuthContextAttestation(
id: string,
grantId: string,
): Promise<BrowserAuthContextAttestation> {
await load();
if (purge()) await save();
const attestation = attestations.get(id);
if (!attestation || attestation.grantId !== grantId) {
throw new ExtensionError(
'auth_context_stale',
'认证上下文证明不存在、已过期或不属于当前共享会话',
);
}
try {
await validateAuthContextBinding(attestation);
return attestation;
} catch (error) {
attestations.delete(id);
await save();
if (error instanceof ExtensionError && error.code === 'auth_context_stale') throw error;
const message = error instanceof Error ? error.message : String(error);
throw new ExtensionError('auth_context_stale', `认证上下文证明实时复核失败:${message}`);
}
}
@@ -1,96 +0,0 @@
import { describe, expect, it } from 'vitest';
import type { BrowserCookie, PageContext, PageStorageEntry } from '@/types/models';
import { authenticationFingerprint } from './auth-fingerprint';
import { AUTHORIZATION_WORKSPACE_TTL_MS } from './lifetime';
function cookie(name: string, value: string): BrowserCookie {
return {
name,
value,
domain: 'example.test',
path: '/',
secure: true,
httpOnly: true,
sameSite: 'lax',
session: true,
hostOnly: true,
storeId: 'opaque-store',
};
}
function storageEntry(key: string, value: string): PageStorageEntry {
return {
key,
value,
byteLength: value.length,
authRelated: true,
truncated: false,
};
}
function context(cookies: BrowserCookie[], storage: PageStorageEntry[] = []): PageContext {
return {
cookies,
document: {
url: 'https://example.test/account',
localStorage: {
supported: true,
entries: storage,
totalEntries: storage.length,
approximateBytes: 0,
truncated: false,
},
sessionStorage: {
supported: true,
entries: [],
totalEntries: 0,
approximateBytes: 0,
truncated: false,
},
},
} as unknown as PageContext;
}
describe('authorization context fingerprint', () => {
it('keeps authorization context available for human and Agent review', () => {
expect(AUTHORIZATION_WORKSPACE_TTL_MS).toBe(30 * 60_000);
});
it('keeps raw Cookie and Storage values out of the canonical identity fingerprint', async () => {
const signed: string[] = [];
const signer = async (value: string) => {
signed.push(value);
return 'f'.repeat(64);
};
const fingerprint = await authenticationFingerprint(
context(
[cookie('session_id', 'cookie-secret-value')],
[storageEntry('access_token', 'storage-secret-value')],
),
signer,
);
const canonical = signed.at(-1) || '';
expect(fingerprint).toBe(`hmac-sha256:${'f'.repeat(64)}`);
expect(canonical).toContain('session_id');
expect(canonical).toContain('access_token');
expect(canonical).not.toContain('cookie-secret-value');
expect(canonical).not.toContain('storage-secret-value');
});
it('fails closed instead of fingerprinting a truncated Cookie collection', async () => {
const cookies = Array.from({ length: 501 }, (_, index) => cookie(`cookie-${index}`, 'value'));
await expect(authenticationFingerprint(context(cookies), async () => 'f'.repeat(64)))
.rejects.toThrow('超过 500 个 Cookie');
});
it('fails closed when the shared page-context Storage snapshot is incomplete', async () => {
const pageContext = context([cookie('session_id', 'value')]);
pageContext.document.localStorage!.truncated = true;
await expect(authenticationFingerprint(pageContext, async () => 'f'.repeat(64)))
.rejects.toThrow('localStorage 快照发生截断');
});
});
@@ -1,366 +0,0 @@
import { browser } from 'wxt/browser';
import type {
BrowserAuthContextHandle,
BrowserIsolationContext,
BrowserTarget,
} from '@/types/models';
import { capturePageContext } from '@/features/page-context/service';
import { getState } from '@/platform/storage/state';
import { ExtensionError } from '@/shared/errors';
import {
authenticationFingerprint,
authenticationStorageEntries,
} from './auth-fingerprint';
import {
getBrowserIsolationProof,
inspectBrowserIsolation,
} from './isolation';
import { AUTHORIZATION_WORKSPACE_TTL_MS } from './lifetime';
export const AUTH_CONTEXT_TTL_MS = AUTHORIZATION_WORKSPACE_TTL_MS;
const MAX_AUTH_CONTEXTS = 32;
const MAX_AUTH_CONTEXT_STORAGE_BYTES = 64 * 1_024;
const STORAGE_KEY = 'browser.authorization.auth-contexts.v1';
const HMAC_KEY_STORAGE_KEY = 'browser.authorization.hmac-key.v1';
const handles = new Map<string, BrowserAuthContextHandle>();
let handlesLoaded = false;
let hmacKeyPromise: Promise<CryptoKey> | undefined;
function bytesToBase64(bytes: Uint8Array): string {
let binary = '';
for (let offset = 0; offset < bytes.length; offset += 8_192) {
binary += String.fromCharCode(...bytes.subarray(offset, offset + 8_192));
}
return btoa(binary);
}
function base64ToBytes(value: string): Uint8Array {
const binary = atob(value);
return Uint8Array.from(binary, (character) => character.charCodeAt(0));
}
function bytesToHex(bytes: Uint8Array): string {
return [...bytes].map((byte) => byte.toString(16).padStart(2, '0')).join('');
}
async function sessionHmacKey(): Promise<CryptoKey> {
if (hmacKeyPromise) return hmacKeyPromise;
hmacKeyPromise = (async () => {
let raw: Uint8Array | undefined;
try {
const stored = await browser.storage.session.get(HMAC_KEY_STORAGE_KEY);
const encoded = stored[HMAC_KEY_STORAGE_KEY];
if (typeof encoded === 'string') {
const candidate = base64ToBytes(encoded);
if (candidate.byteLength === 32) raw = candidate;
}
} catch {
// A fresh in-memory session key is sufficient when storage.session is unavailable.
}
if (!raw) {
raw = crypto.getRandomValues(new Uint8Array(32));
try {
await browser.storage.session.set({ [HMAC_KEY_STORAGE_KEY]: bytesToBase64(raw) });
} catch {
// Keep the key in this service worker lifetime as the fallback.
}
}
return crypto.subtle.importKey(
'raw',
Uint8Array.from(raw).buffer,
{ name: 'HMAC', hash: 'SHA-256' },
false,
['sign'],
);
})();
return hmacKeyPromise;
}
async function hmac(value: string): Promise<string> {
const signature = await crypto.subtle.sign(
'HMAC',
await sessionHmacKey(),
new TextEncoder().encode(value),
);
return bytesToHex(new Uint8Array(signature));
}
function authRelated(name: string): boolean {
return /(auth|token|jwt|session|login|csrf|xsrf|sid|credential|bearer)/i.test(name);
}
function validStoredHandle(value: unknown): value is BrowserAuthContextHandle {
if (!value || typeof value !== 'object' || Array.isArray(value)) return false;
const handle = value as Partial<BrowserAuthContextHandle>;
return handle.version === 1
&& typeof handle.id === 'string'
&& handle.id.length > 0
&& handle.id.length <= 160
&& ['left', 'right'].includes(String(handle.slotId))
&& typeof handle.deviceId === 'string'
&& handle.deviceId.length > 0
&& handle.deviceId.length <= 320
&& typeof handle.installationId === 'string'
&& handle.installationId.length > 0
&& handle.installationId.length <= 320
&& typeof handle.isolationContextId === 'string'
&& handle.isolationContextId.length > 0
&& handle.isolationContextId.length <= 320
&& typeof handle.isolationProofId === 'string'
&& handle.isolationProofId.length > 0
&& handle.isolationProofId.length <= 160
&& typeof handle.cookieStoreId === 'string'
&& handle.cookieStoreId.length > 0
&& handle.cookieStoreId.length <= 320
&& typeof handle.origin === 'string'
&& handle.origin.length > 0
&& handle.origin.length <= 8_192
&& typeof handle.grantId === 'string'
&& handle.grantId.length > 0
&& handle.grantId.length <= 160
&& typeof handle.fingerprint === 'string'
&& /^hmac-sha256:[a-f0-9]{64}$/.test(handle.fingerprint)
&& (handle.accountLabel === undefined
|| (typeof handle.accountLabel === 'string' && handle.accountLabel.length <= 80))
&& Boolean(handle.target)
&& Number.isSafeInteger(handle.target?.tabId)
&& Number(handle.target?.tabId) > 0
&& Number.isSafeInteger(handle.target?.frameId)
&& Number(handle.target?.frameId) >= 0
&& typeof handle.target?.documentId === 'string'
&& handle.target.documentId.length > 0
&& handle.target.documentId.length <= 160
&& Boolean(handle.authentication)
&& ['authenticated', 'unauthenticated', 'unknown'].includes(String(handle.authentication?.status))
&& Number.isSafeInteger(handle.authentication?.cookieCount)
&& Number(handle.authentication?.cookieCount) >= 0
&& Number.isSafeInteger(handle.authentication?.storageEntryCount)
&& Number(handle.authentication?.storageEntryCount) >= 0
&& Array.isArray(handle.authentication?.authCookieNames)
&& handle.authentication.authCookieNames.length <= 100
&& handle.authentication.authCookieNames.every((name) => typeof name === 'string' && name.length <= 500)
&& Array.isArray(handle.authentication?.authStorageKeys)
&& handle.authentication.authStorageKeys.length <= 100
&& handle.authentication.authStorageKeys.every((key) => typeof key === 'string' && key.length <= 520)
&& typeof handle.createdAt === 'number'
&& typeof handle.expiresAt === 'number'
&& handle.expiresAt > handle.createdAt
&& handle.expiresAt - handle.createdAt <= AUTH_CONTEXT_TTL_MS;
}
function purgeHandles(now = Date.now(), reserve = 0): boolean {
let changed = false;
for (const [id, handle] of handles) {
if (handle.expiresAt <= now) {
handles.delete(id);
changed = true;
}
}
while (handles.size > MAX_AUTH_CONTEXTS - reserve) {
const oldest = handles.keys().next().value as string | undefined;
if (!oldest) break;
handles.delete(oldest);
changed = true;
}
return changed;
}
async function loadHandles(): Promise<void> {
if (handlesLoaded) return;
handlesLoaded = true;
try {
const stored = await browser.storage.session.get(STORAGE_KEY);
const values = stored[STORAGE_KEY];
if (!Array.isArray(values)) return;
for (const value of values.slice(-MAX_AUTH_CONTEXTS)) {
if (validStoredHandle(value)) handles.set(value.id, value);
}
purgeHandles();
} catch {
// Keep the bounded memory registry on adapters without storage.session.
}
}
async function saveHandles(): Promise<void> {
try {
const retained: BrowserAuthContextHandle[] = [];
for (const handle of [...handles.values()].reverse()) {
const candidate = [handle, ...retained];
if (new TextEncoder().encode(JSON.stringify(candidate)).byteLength > MAX_AUTH_CONTEXT_STORAGE_BYTES) break;
retained.unshift(handle);
}
handles.clear();
for (const handle of retained) handles.set(handle.id, handle);
await browser.storage.session.set({
[STORAGE_KEY]: retained,
});
} catch {
// Keep the bounded memory registry on adapters without storage.session.
}
}
function isolationContext(
contexts: BrowserIsolationContext[],
isolationContextId: string | undefined,
): BrowserIsolationContext | undefined {
return contexts.find((context) => context.contextId === isolationContextId);
}
export interface CapturedAuthContextSnapshot {
deviceId: string;
installationId: string;
isolationContextId: string;
cookieStoreId: string;
origin: string;
target: BrowserTarget & { documentId: string };
fingerprint: string;
authentication: BrowserAuthContextHandle['authentication'];
}
type AuthContextBinding = Pick<
BrowserAuthContextHandle,
| 'deviceId'
| 'installationId'
| 'isolationContextId'
| 'cookieStoreId'
| 'origin'
| 'target'
| 'fingerprint'
>;
export async function captureAuthContextSnapshot(
target: BrowserTarget,
): Promise<CapturedAuthContextSnapshot> {
const inspection = await inspectBrowserIsolation([target.tabId]);
const tab = inspection.tabs[0];
const context = isolationContext(inspection.contexts, tab?.isolationContextId);
if (!tab || !context?.cookieStoreId || context.level === 'none') {
throw new ExtensionError('isolation_unresolved', '目标页面没有可用的隔离上下文,不能创建认证快照');
}
const pageContext = await capturePageContext(
{ includeDom: false, includeStorage: true, includeCookies: true },
target,
);
if (!pageContext.target.documentId) {
throw new ExtensionError('stale_document', '目标页面缺少稳定 document 标识');
}
const state = await getState();
const deviceId = state.bridge.pairedEngine?.deviceId;
if (!deviceId) throw new ExtensionError('bridge_disconnected', '插件尚未与 Yak 引擎配对');
const cookies = pageContext.cookies || [];
const storage = authenticationStorageEntries(pageContext);
return {
deviceId,
installationId: state.bridge.installationId,
isolationContextId: context.contextId,
cookieStoreId: context.cookieStoreId,
origin: new URL(pageContext.document.url).origin,
target: {
tabId: pageContext.target.tabId,
frameId: pageContext.target.frameId,
documentId: pageContext.target.documentId,
},
fingerprint: await authenticationFingerprint(pageContext, hmac),
authentication: {
status: pageContext.authentication.status,
cookieCount: cookies.length,
storageEntryCount: storage.length,
authCookieNames: cookies
.filter((cookie) => authRelated(cookie.name))
.map((cookie) => cookie.name)
.slice(0, 100),
authStorageKeys: storage
.filter((entry) => authRelated(entry.key))
.map((entry) => `${entry.area}:${entry.key}`)
.slice(0, 100),
},
};
}
export async function validateAuthContextBinding(binding: AuthContextBinding): Promise<void> {
const state = await getState();
if (state.bridge.pairedEngine?.deviceId !== binding.deviceId
|| state.bridge.installationId !== binding.installationId) {
throw new ExtensionError('auth_context_stale', '插件安装身份或配对引擎已经变化');
}
const current = await captureAuthContextSnapshot(binding.target);
if (current.isolationContextId !== binding.isolationContextId
|| current.cookieStoreId !== binding.cookieStoreId) {
throw new ExtensionError('auth_context_stale', '目标页面的 Cookie Store 或隔离上下文已经变化');
}
if (current.target.documentId !== binding.target.documentId
|| current.origin !== binding.origin
|| current.fingerprint !== binding.fingerprint) {
throw new ExtensionError('auth_context_stale', '目标文档、来源或认证材料已经变化');
}
}
export async function captureAuthContextHandle(input: {
slotId: 'left' | 'right';
accountLabel?: string;
isolationProofId: string;
target: BrowserTarget;
grantId: string;
grantExpiresAt: number;
}): Promise<BrowserAuthContextHandle> {
await loadHandles();
const proof = await getBrowserIsolationProof(input.isolationProofId);
if (proof.level === 'none') {
throw new ExtensionError('isolation_unresolved', '当前证明没有建立两个身份的隔离关系,不能创建认证句柄');
}
const expectedTabId = input.slotId === 'left' ? proof.leftTabId : proof.rightTabId;
if (input.target.tabId !== expectedTabId) {
throw new ExtensionError('target_denied', '认证上下文目标与隔离证明中的身份槽位不一致');
}
const expectedContextId = input.slotId === 'left'
? proof.leftContextId
: proof.rightContextId;
const snapshot = await captureAuthContextSnapshot(input.target);
if (snapshot.isolationContextId !== expectedContextId) {
throw new ExtensionError('isolation_stale', '目标页面的隔离上下文已经变化,请重新执行预检');
}
const now = Date.now();
const handle: BrowserAuthContextHandle = {
version: 1,
id: crypto.randomUUID(),
slotId: input.slotId,
accountLabel: input.accountLabel?.trim().slice(0, 80) || undefined,
...snapshot,
isolationProofId: proof.id,
grantId: input.grantId,
createdAt: now,
expiresAt: Math.min(now + AUTH_CONTEXT_TTL_MS, proof.expiresAt, input.grantExpiresAt),
};
if (handle.expiresAt <= now) throw new ExtensionError('grant_expired', '共享会话或隔离证明已经过期');
purgeHandles(now, 1);
handles.set(handle.id, handle);
await saveHandles();
return handle;
}
export async function getAuthContextHandle(id: string, grantId: string): Promise<BrowserAuthContextHandle> {
await loadHandles();
if (purgeHandles()) await saveHandles();
const handle = handles.get(id);
if (!handle || handle.grantId !== grantId) {
throw new ExtensionError('auth_context_stale', '认证上下文句柄不存在、已过期或不属于当前共享会话');
}
try {
const proof = await getBrowserIsolationProof(handle.isolationProofId);
if (proof.level === 'none') throw new ExtensionError('auth_context_stale', '身份隔离证明已经失效');
const expectedTabId = handle.slotId === 'left' ? proof.leftTabId : proof.rightTabId;
const expectedContextId = handle.slotId === 'left' ? proof.leftContextId : proof.rightContextId;
if (handle.target.tabId !== expectedTabId || handle.isolationContextId !== expectedContextId) {
throw new ExtensionError('auth_context_stale', '认证句柄与当前隔离证明不一致');
}
await validateAuthContextBinding(handle);
return handle;
} catch (error) {
handles.delete(id);
await saveHandles();
if (error instanceof ExtensionError && error.code === 'auth_context_stale') throw error;
const message = error instanceof Error ? error.message : String(error);
throw new ExtensionError('auth_context_stale', `认证上下文实时复核失败:${message}`);
}
}
@@ -1,392 +0,0 @@
import { describe, expect, it } from 'vitest';
import {
applyAuthorizationTransformExecution,
authorizationRequestToTransformPacket,
compileAuthorizationBaselineRequest,
extractAuthorizationResourceValue,
parseAuthorizationRequestPacket,
replaceAuthorizationResourceValue,
} from './baseline-execution';
import { fingerprintAuthorizationComparisonValue } from './baseline-metadata';
function base64(value: string): string {
return btoa(value);
}
describe('authorization baseline execution primitives', () => {
it('parses a bounded request packet without discarding captured credentials', () => {
const packet = parseAuthorizationRequestPacket(base64([
'GET /api/orders/42 HTTP/1.1',
'Host: example.test',
'Cookie: session=secret',
'Authorization: Bearer secret',
'X-CSRF-Token: csrf-secret',
'Sec-Fetch-Site: same-origin',
'',
'',
].join('\r\n')));
expect(packet.method).toBe('GET');
expect(packet.headers).toEqual([
{ name: 'Host', value: 'example.test' },
{ name: 'Cookie', value: 'session=secret' },
{ name: 'Authorization', value: 'Bearer secret' },
{ name: 'X-CSRF-Token', value: 'csrf-secret' },
{ name: 'Sec-Fetch-Site', value: 'same-origin' },
]);
});
it('extracts and replaces a normalized path resource without changing the origin', () => {
const value = extractAuthorizationResourceValue(
'https://example.test/api/orders/42?view=full',
'',
'baseline-left',
{ location: 'path', path: 'path.segment[2]' },
'workspace-hmac-sha256:a'.padEnd(86, 'a'),
);
const replaced = replaceAuthorizationResourceValue(
'https://example.test/api/orders/42?view=full',
{ location: 'path', path: 'path.segment[2]' },
'84',
);
expect(atob(value.valueBase64)).toBe('42');
expect(replaced).toBe('https://example.test/api/orders/84?view=full');
});
it('addresses repeated query parameters by occurrence', () => {
const url = 'https://example.test/api/orders?id=42&view=full&id=84';
const value = extractAuthorizationResourceValue(
url,
'',
'baseline-right',
{ location: 'query', path: 'query.id[1]' },
'workspace-hmac-sha256:b'.padEnd(86, 'b'),
);
const replaced = replaceAuthorizationResourceValue(
url,
{ location: 'query', path: 'query.id[1]' },
'126',
);
expect(atob(value.valueBase64)).toBe('84');
expect(replaced).toBe('https://example.test/api/orders?id=42&view=full&id=126');
expect(() => extractAuthorizationResourceValue(
url,
'',
'baseline-right',
{ location: 'query', path: 'query.id' },
'workspace-hmac-sha256:b'.padEnd(86, 'b'),
)).toThrow('多个同名值');
});
it('compiles a read-only request while retaining the exact captured header block', async () => {
const comparisonKey = btoa(String.fromCharCode(...new Uint8Array(32).fill(7)))
.replace(/\+/g, '-')
.replace(/\//g, '_')
.replace(/=+$/, '');
const valueFingerprint = await fingerprintAuthorizationComparisonValue(comparisonKey, '84');
const raw = [
'GET /api/orders/42 HTTP/1.1',
'Host: example.test',
'Cookie: session=secret',
'Authorization: Bearer secret',
'',
'',
].join('\r\n');
const compiled = await compileAuthorizationBaselineRequest({
baselineId: 'baseline-left',
rawRequestBase64: base64(raw),
requestUrl: 'https://example.test/api/orders/42',
publicUrl: 'https://example.test/api/orders/:resource',
selector: { source: 'wire', location: 'path', path: 'path.segment[2]' },
replacement: {
version: 1,
baselineId: 'baseline-right',
source: 'wire',
location: 'path',
path: 'path.segment[2]',
valueType: 'string',
byteLength: 2,
valueBase64: base64('84'),
valueFingerprint,
},
comparisonKey,
isHttps: true,
});
const request = atob(compiled.rawRequestBase64);
expect(request).toContain('GET /api/orders/84 HTTP/1.1\r\n');
expect(request).toContain('Cookie: session=secret\r\n');
expect(request).toContain('Authorization: Bearer secret\r\n');
expect(compiled.resourceValueFingerprint).toBe(valueFingerprint);
});
it('replaces an explicit resource Header without copying another identity credential', async () => {
const comparisonKey = btoa(String.fromCharCode(...new Uint8Array(32).fill(11)))
.replace(/\+/g, '-')
.replace(/\//g, '_')
.replace(/=+$/, '');
const valueFingerprint = await fingerprintAuthorizationComparisonValue(comparisonKey, 'tenant-b');
const raw = [
'GET /api/orders HTTP/1.1',
'Host: example.test',
'Cookie: session=identity-a',
'X-Tenant-Id: tenant-a',
'',
'',
].join('\r\n');
const resource = extractAuthorizationResourceValue(
'https://example.test/api/orders',
base64(raw),
'baseline-left',
{ location: 'header', path: 'header.x-tenant-id' },
await fingerprintAuthorizationComparisonValue(comparisonKey, 'tenant-a'),
);
const compiled = await compileAuthorizationBaselineRequest({
baselineId: 'baseline-left',
rawRequestBase64: base64(raw),
requestUrl: 'https://example.test/api/orders',
publicUrl: 'https://example.test/api/orders',
selector: { source: 'wire', location: 'header', path: 'header.x-tenant-id' },
replacement: {
version: 1,
baselineId: 'baseline-right',
source: 'wire',
location: 'header',
path: 'header.x-tenant-id',
valueType: 'string',
byteLength: 8,
valueBase64: base64('tenant-b'),
valueFingerprint,
},
comparisonKey,
isHttps: true,
});
expect(atob(resource.valueBase64)).toBe('tenant-a');
expect(atob(compiled.rawRequestBase64)).toContain('X-Tenant-Id: tenant-b\r\n');
expect(atob(compiled.rawRequestBase64)).toContain('Cookie: session=identity-a\r\n');
expect(atob(compiled.rawRequestBase64)).not.toContain('session=identity-b');
});
it('replaces one GraphQL variable in a reviewed POST without changing the operation or credentials', async () => {
const comparisonKey = btoa(String.fromCharCode(...new Uint8Array(32).fill(13)))
.replace(/\+/g, '-')
.replace(/\//g, '_')
.replace(/=+$/, '');
const valueFingerprint = await fingerprintAuthorizationComparisonValue(
comparisonKey,
'84',
);
const body = JSON.stringify({
operationName: 'Order',
query: 'query Order($orderId: ID!) { order(id: $orderId) { id total } }',
variables: {
orderId: 42,
includeAudit: true,
},
});
const raw = [
'POST /graphql HTTP/1.1',
'Host: example.test',
'Content-Type: application/json',
`Content-Length: ${new TextEncoder().encode(body).byteLength}`,
'Cookie: session=identity-a',
'',
body,
].join('\r\n');
const compiled = await compileAuthorizationBaselineRequest({
baselineId: 'baseline-left',
rawRequestBase64: base64(raw),
requestUrl: 'https://example.test/graphql',
publicUrl: 'https://example.test/graphql',
selector: {
source: 'wire',
location: 'body',
path: 'body.variables.orderId',
},
replacement: {
version: 1,
baselineId: 'baseline-right',
source: 'wire',
location: 'body',
path: 'body.variables.orderId',
valueType: 'number',
byteLength: 2,
valueBase64: base64('84'),
valueFingerprint,
},
comparisonKey,
isHttps: true,
});
const compiledPacket = parseAuthorizationRequestPacket(compiled.rawRequestBase64);
const compiledBody = JSON.parse(new TextDecoder().decode(
compiledPacket.bytes.subarray(compiledPacket.bodyOffset),
));
expect(compiledBody.variables).toEqual({
orderId: 84,
includeAudit: true,
});
expect(compiledBody.query).toBe(
'query Order($orderId: ID!) { order(id: $orderId) { id total } }',
);
expect(atob(compiled.rawRequestBase64)).toContain('Cookie: session=identity-a\r\n');
expect(compiledPacket.headers.find(
(header) => header.name.toLowerCase() === 'content-length',
)?.value).toBe(String(new TextEncoder().encode(JSON.stringify(compiledBody)).byteLength));
});
it('addresses a GraphQL batch variable by its ordered operation index', async () => {
const comparisonKey = btoa(String.fromCharCode(...new Uint8Array(32).fill(17)))
.replace(/\+/g, '-')
.replace(/\//g, '_')
.replace(/=+$/, '');
const valueFingerprint = await fingerprintAuthorizationComparisonValue(
comparisonKey,
'user-b',
);
const body = JSON.stringify([
{
operationName: 'Viewer',
query: 'query Viewer { viewer { id } }',
variables: {},
},
{
operationName: 'User',
query: 'query User($userId: ID!) { user(id: $userId) { id } }',
variables: { userId: 'user-a' },
},
]);
const raw = [
'POST /graphql HTTP/1.1',
'Host: example.test',
'Content-Type: application/json',
`Content-Length: ${new TextEncoder().encode(body).byteLength}`,
'Cookie: session=identity-a',
'',
body,
].join('\r\n');
const compiled = await compileAuthorizationBaselineRequest({
baselineId: 'baseline-left',
rawRequestBase64: base64(raw),
requestUrl: 'https://example.test/graphql',
publicUrl: 'https://example.test/graphql',
selector: {
source: 'wire',
location: 'body',
path: 'body[1].variables.userId',
},
replacement: {
version: 1,
baselineId: 'baseline-right',
source: 'wire',
location: 'body',
path: 'body[1].variables.userId',
valueType: 'string',
byteLength: 6,
valueBase64: base64('user-b'),
valueFingerprint,
},
comparisonKey,
isHttps: true,
});
const compiledPacket = parseAuthorizationRequestPacket(compiled.rawRequestBase64);
const compiledBody = JSON.parse(new TextDecoder().decode(
compiledPacket.bytes.subarray(compiledPacket.bodyOffset),
));
expect(compiledBody.map((operation: { operationName: string }) => operation.operationName))
.toEqual(['Viewer', 'User']);
expect(compiledBody[1].variables.userId).toBe('user-b');
});
it('applies an identity-bound query signature without changing captured credentials', async () => {
const raw = base64([
'GET /api/orders/84?nonce=old&signature=old HTTP/1.1',
'Host: example.test',
'Cookie: session=identity-a',
'Authorization: Bearer identity-a',
'',
'',
].join('\r\n'));
const packet = authorizationRequestToTransformPacket(raw, 'https://example.test');
const compiled = await applyAuthorizationTransformExecution({
compiled: {
version: 1,
baselineId: 'baseline-left',
selector: { source: 'wire', location: 'path', path: 'path.segment[2]' },
method: 'GET',
url: 'https://example.test/api/orders/:resource',
isHttps: true,
rawRequestBase64: raw,
resourceValueFingerprint: 'workspace-hmac-sha256:a'.padEnd(88, 'a'),
packetFingerprint: `sha256:${'a'.repeat(64)}`,
},
execution: {
profileId: 'profile-left',
direction: 'request',
url: 'https://example.test/api/orders/84?nonce=fresh&signature=signed-84',
bodyBase64: packet.bodyBase64,
setHeaders: [],
removeHeaders: [],
logicalInput: {},
logicalOutput: {},
nodeDurations: [],
nodeTrace: [],
fieldChanges: [],
durationMs: 1,
},
origin: 'https://example.test',
allowedDestinations: ['query.nonce', 'query.signature'],
});
const request = atob(compiled.rawRequestBase64);
expect(request).toContain('GET /api/orders/84?nonce=fresh&signature=signed-84 HTTP/1.1');
expect(request).toContain('Cookie: session=identity-a');
expect(request).toContain('Authorization: Bearer identity-a');
});
it('rejects dynamic transforms that touch authentication headers', async () => {
const raw = base64([
'GET /api/orders/84?signature=old HTTP/1.1',
'Host: example.test',
'Cookie: session=identity-a',
'',
'',
].join('\r\n'));
const packet = authorizationRequestToTransformPacket(raw, 'https://example.test');
await expect(applyAuthorizationTransformExecution({
compiled: {
version: 1,
baselineId: 'baseline-left',
selector: { source: 'wire', location: 'path', path: 'path.segment[2]' },
method: 'GET',
url: 'https://example.test/api/orders/:resource',
isHttps: true,
rawRequestBase64: raw,
resourceValueFingerprint: 'workspace-hmac-sha256:a'.padEnd(88, 'a'),
packetFingerprint: `sha256:${'a'.repeat(64)}`,
},
execution: {
profileId: 'profile-left',
direction: 'request',
url: packet.url,
bodyBase64: packet.bodyBase64,
setHeaders: [{ name: 'Cookie', value: 'session=identity-b' }],
removeHeaders: [],
logicalInput: {},
logicalOutput: {},
nodeDurations: [],
nodeTrace: [],
fieldChanges: [],
durationMs: 1,
},
origin: 'https://example.test',
allowedDestinations: ['header.cookie'],
})).rejects.toThrow('认证材料');
});
});
@@ -1,571 +0,0 @@
import type {
BrowserAuthorizationCompiledRequest,
BrowserAuthorizationResourceSelector,
BrowserAuthorizationResourceValue,
BrowserTransformExecution,
BrowserTransformPacket,
} from '@/types/models';
import { ExtensionError } from '@/shared/errors';
import { fingerprintAuthorizationComparisonValue } from './baseline-metadata';
import {
replaceStructuredAuthorizationBodyValue,
} from './structured-body';
const MAX_RESOURCE_VALUE_BYTES = 8 * 1_024;
interface ParsedAuthorizationRequest {
method: string;
requestTarget: string;
protocol: string;
headers: Array<{ name: string; value: string }>;
bytes: Uint8Array;
bodyOffset: number;
}
function base64ToBytes(value: string): Uint8Array {
let binary: string;
try {
binary = atob(value);
} catch {
throw new ExtensionError('authorization_value_invalid', '授权资源值不是有效的 Base64');
}
return Uint8Array.from(binary, (character) => character.charCodeAt(0));
}
function bytesToBase64(bytes: Uint8Array): string {
let binary = '';
const chunkSize = 0x8000;
for (let offset = 0; offset < bytes.length; offset += chunkSize) {
binary += String.fromCharCode(...bytes.subarray(offset, offset + chunkSize));
}
return btoa(binary);
}
function packetBodyOffset(bytes: Uint8Array): number {
for (let index = 0; index <= bytes.length - 4; index += 1) {
if (bytes[index] === 13 && bytes[index + 1] === 10
&& bytes[index + 2] === 13 && bytes[index + 3] === 10) {
return index + 4;
}
}
throw new ExtensionError('authorization_baseline_invalid', '授权基线缺少 HTTP Header 分隔符');
}
export function parseAuthorizationRequestPacket(
rawRequestBase64: string,
): ParsedAuthorizationRequest {
const bytes = base64ToBytes(rawRequestBase64);
const offset = packetBodyOffset(bytes);
let head: string;
try {
head = new TextDecoder('utf-8', { fatal: true }).decode(bytes.subarray(0, offset - 4));
} catch {
throw new ExtensionError('authorization_baseline_invalid', '授权基线请求头不是有效的 UTF-8');
}
const lines = head.split('\r\n');
const requestLine = lines.shift()?.split(/\s+/) || [];
if (requestLine.length !== 3 || !/^[A-Z]{1,16}$/.test(requestLine[0])) {
throw new ExtensionError('authorization_baseline_invalid', '授权基线请求行无效');
}
const headers = lines.slice(0, 256).flatMap((line) => {
const separator = line.indexOf(':');
if (separator <= 0) return [];
const name = line.slice(0, separator).trim().slice(0, 256);
const value = line.slice(separator + 1).trim().slice(0, 16_384);
return name ? [{ name, value }] : [];
});
return {
method: requestLine[0],
requestTarget: requestLine[1],
protocol: requestLine[2],
headers,
bytes,
bodyOffset: offset,
};
}
function parameterSelector(
location: 'header' | 'query',
path: string,
): { name: string; index?: number } {
const prefix = `${location}.`;
if (!path.startsWith(prefix)) {
throw new ExtensionError('authorization_selector_invalid', '授权资源字段路径与位置不匹配');
}
const raw = path.slice(prefix.length);
const indexed = raw.match(/^(.*)\[(\d+)]$/);
const name = indexed ? indexed[1] : raw;
const index = indexed ? Number(indexed[2]) : undefined;
if (!name || (index !== undefined && (!Number.isSafeInteger(index) || index < 0))) {
throw new ExtensionError('authorization_selector_invalid', '授权资源字段路径无效');
}
return { name, index };
}
function pathSegmentSelector(path: string): number {
const matched = path.match(/^path\.segment\[(\d+)]$/);
const index = matched ? Number(matched[1]) : -1;
if (!Number.isSafeInteger(index) || index < 0) {
throw new ExtensionError('authorization_selector_invalid', '授权路径资源字段无效');
}
return index;
}
function valuesForQuery(url: URL, name: string): string[] {
return [...url.searchParams].filter(([key]) => key === name).map(([, value]) => value);
}
export function extractAuthorizationResourceValue(
requestUrl: string,
rawRequestBase64: string,
baselineId: string,
selector: { location: 'header' | 'path' | 'query'; path: string },
valueFingerprint: string,
): BrowserAuthorizationResourceValue {
const url = new URL(requestUrl);
let value: string;
if (selector.location === 'header') {
const selected = parameterSelector('header', selector.path);
const values = parseAuthorizationRequestPacket(rawRequestBase64).headers
.filter((header) => header.name.toLowerCase() === selected.name.toLowerCase())
.map((header) => header.value);
if (selected.index === undefined && values.length !== 1) {
throw new ExtensionError('authorization_selector_ambiguous', '授权 Header 字段存在多个同名值,必须选择带序号的字段');
}
const index = selected.index ?? 0;
if (index >= values.length) {
throw new ExtensionError('authorization_selector_invalid', '授权 Header 资源字段不存在');
}
value = values[index];
} else if (selector.location === 'path') {
const index = pathSegmentSelector(selector.path);
const segments = url.pathname.split('/').filter(Boolean);
if (index >= segments.length) {
throw new ExtensionError('authorization_selector_invalid', '授权路径资源字段不存在');
}
try {
value = decodeURIComponent(segments[index]);
} catch {
value = segments[index];
}
} else {
const selected = parameterSelector('query', selector.path);
const values = valuesForQuery(url, selected.name);
if (selected.index === undefined && values.length !== 1) {
throw new ExtensionError('authorization_selector_ambiguous', '授权查询字段存在多个同名值,必须选择带序号的字段');
}
const index = selected.index ?? 0;
if (index >= values.length) {
throw new ExtensionError('authorization_selector_invalid', '授权查询资源字段不存在');
}
value = values[index];
}
const bytes = new TextEncoder().encode(value);
if (bytes.byteLength > MAX_RESOURCE_VALUE_BYTES) {
throw new ExtensionError('authorization_value_too_large', '授权资源值超过 8 KiB 上限');
}
return {
version: 1,
baselineId,
source: 'wire',
location: selector.location,
path: selector.path,
valueType: 'string',
byteLength: bytes.byteLength,
valueBase64: bytesToBase64(bytes),
valueFingerprint,
};
}
export function replaceAuthorizationResourceValue(
requestUrl: string,
selector: { location: 'path' | 'query'; path: string },
replacement: string,
): string {
const url = new URL(requestUrl);
if (selector.location === 'path') {
const selectedIndex = pathSegmentSelector(selector.path);
let currentIndex = -1;
const segments = url.pathname.split('/');
const next = segments.map((segment) => {
if (!segment) return segment;
currentIndex += 1;
return currentIndex === selectedIndex ? encodeURIComponent(replacement) : segment;
});
if (currentIndex < selectedIndex) {
throw new ExtensionError('authorization_selector_invalid', '授权路径资源字段不存在');
}
url.pathname = next.join('/');
return url.toString();
}
const selected = parameterSelector('query', selector.path);
const entries = [...url.searchParams];
const matchingIndexes = entries.flatMap(([name], index) => name === selected.name ? [index] : []);
if (selected.index === undefined && matchingIndexes.length !== 1) {
throw new ExtensionError('authorization_selector_ambiguous', '授权查询字段存在多个同名值,必须选择带序号的字段');
}
const occurrence = selected.index ?? 0;
if (occurrence >= matchingIndexes.length) {
throw new ExtensionError('authorization_selector_invalid', '授权查询资源字段不存在');
}
entries[matchingIndexes[occurrence]][1] = replacement;
url.search = '';
for (const [name, value] of entries) url.searchParams.append(name, value);
return url.toString();
}
export async function compileAuthorizationBaselineRequest(input: {
baselineId: string;
rawRequestBase64: string;
requestUrl: string;
publicUrl: string;
selector: BrowserAuthorizationResourceSelector & { source: 'wire' };
replacement: BrowserAuthorizationResourceValue;
comparisonKey: string;
isHttps: boolean;
}): Promise<BrowserAuthorizationCompiledRequest> {
const packet = parseAuthorizationRequestPacket(input.rawRequestBase64);
const method = packet.method.toUpperCase();
if (input.replacement.source !== 'wire'
|| input.replacement.location !== input.selector.location
|| input.replacement.path !== input.selector.path
|| !['string', 'number', 'boolean'].includes(input.replacement.valueType)) {
throw new ExtensionError('authorization_value_invalid', '授权资源值与矩阵选择器不匹配');
}
const replacementBytes = base64ToBytes(input.replacement.valueBase64);
if (replacementBytes.byteLength !== input.replacement.byteLength
|| replacementBytes.byteLength > MAX_RESOURCE_VALUE_BYTES) {
throw new ExtensionError('authorization_value_invalid', '授权资源值长度无效');
}
let replacementText: string;
try {
replacementText = new TextDecoder('utf-8', { fatal: true }).decode(replacementBytes);
} catch {
throw new ExtensionError('authorization_value_invalid', '授权资源值不是有效的 UTF-8 字符串');
}
let replacement: string | number | boolean;
if (input.replacement.valueType === 'string') {
replacement = replacementText;
} else if (input.replacement.valueType === 'number') {
try {
const parsed: unknown = JSON.parse(replacementText);
if (
typeof parsed !== 'number'
|| !Number.isFinite(parsed)
|| JSON.stringify(parsed) !== replacementText
) {
throw new Error('not canonical');
}
replacement = parsed;
} catch {
throw new ExtensionError('authorization_value_invalid', '授权数字资源值不是规范 JSON 数字');
}
} else if (replacementText === 'true' || replacementText === 'false') {
replacement = replacementText === 'true';
} else {
throw new ExtensionError('authorization_value_invalid', '授权布尔资源值必须是 true 或 false');
}
const fingerprint = await fingerprintAuthorizationComparisonValue(
input.comparisonKey,
replacementText,
);
if (fingerprint !== input.replacement.valueFingerprint) {
throw new ExtensionError('authorization_value_invalid', '授权资源值指纹校验失败');
}
const selector = input.selector;
const selectorLocation = selector.location;
if (selectorLocation === 'body') {
const origin = new URL(input.requestUrl).origin;
const transformed = replaceStructuredAuthorizationBodyValue({
packet: authorizationRequestToTransformPacket(input.rawRequestBase64, origin),
path: selector.path,
replacement,
});
const rawBytes = base64ToBytes(input.rawRequestBase64);
const compiled: BrowserAuthorizationCompiledRequest = {
version: 1,
baselineId: input.baselineId,
selector,
method: method as BrowserAuthorizationCompiledRequest['method'],
url: input.publicUrl,
isHttps: input.isHttps,
rawRequestBase64: input.rawRequestBase64,
resourceValueFingerprint: input.replacement.valueFingerprint,
packetFingerprint: `sha256:${[...new Uint8Array(await crypto.subtle.digest(
'SHA-256',
Uint8Array.from(rawBytes).buffer,
))].map((byte) => byte.toString(16).padStart(2, '0')).join('')}`,
};
return applyAuthorizationTransformExecution({
compiled,
execution: {
profileId: 'authorization-structured-body',
direction: 'request',
url: transformed.url,
bodyBase64: transformed.bodyBase64,
setHeaders: [],
removeHeaders: [],
logicalInput: undefined,
logicalOutput: undefined,
nodeDurations: [],
nodeTrace: [],
fieldChanges: [],
durationMs: 0,
},
origin,
allowedDestinations: [selector.path],
allowBody: true,
});
}
if (typeof replacement !== 'string') {
throw new ExtensionError(
'authorization_value_invalid',
'Header、Path 与 Query 资源替换只接受字符串',
);
}
if (selectorLocation === 'header' && /[\u0000\r\n]/.test(replacement as string)) {
throw new ExtensionError('authorization_value_invalid', '授权 Header 资源值包含非法控制字符');
}
const requestUrl = selectorLocation === 'header'
? input.requestUrl
: replaceAuthorizationResourceValue(
input.requestUrl,
{ location: selectorLocation, path: selector.path },
replacement as string,
);
const originalOrigin = new URL(input.requestUrl).origin;
if (new URL(requestUrl).origin !== originalOrigin) {
throw new ExtensionError('authorization_origin_changed', '资源替换不能改变请求来源');
}
const url = new URL(requestUrl);
const target = selectorLocation === 'header'
? packet.requestTarget
: `${url.pathname || '/'}${url.search}`;
const requestLine = new TextEncoder().encode(`${method} ${target} ${packet.protocol}\r\n`);
const firstLineEnd = packet.bytes.findIndex(
(byte, index) => byte === 13 && packet.bytes[index + 1] === 10,
);
if (firstLineEnd < 0 || firstLineEnd >= packet.bodyOffset - 4) {
throw new ExtensionError('authorization_baseline_invalid', '授权基线请求行边界无效');
}
let remainder = packet.bytes.subarray(firstLineEnd + 2);
if (selectorLocation === 'header') {
const selected = parameterSelector('header', selector.path);
const headerBytes = packet.bytes.subarray(firstLineEnd + 2, packet.bodyOffset - 4);
const headerLines = new TextDecoder('utf-8', { fatal: true }).decode(headerBytes).split('\r\n');
const matching = headerLines.flatMap((line, index) => {
const separator = line.indexOf(':');
return separator > 0 && line.slice(0, separator).trim().toLowerCase() === selected.name.toLowerCase()
? [index]
: [];
});
if (selected.index === undefined && matching.length !== 1) {
throw new ExtensionError('authorization_selector_ambiguous', '授权 Header 字段存在多个同名值,必须选择带序号的字段');
}
const occurrence = selected.index ?? 0;
if (occurrence >= matching.length) {
throw new ExtensionError('authorization_selector_invalid', '授权 Header 资源字段不存在');
}
const lineIndex = matching[occurrence];
const separator = headerLines[lineIndex].indexOf(':');
headerLines[lineIndex] = `${headerLines[lineIndex].slice(0, separator)}: ${replacement as string}`;
const rewrittenHeaders = new TextEncoder().encode(`${headerLines.join('\r\n')}\r\n\r\n`);
const body = packet.bytes.subarray(packet.bodyOffset);
remainder = new Uint8Array(rewrittenHeaders.byteLength + body.byteLength);
remainder.set(rewrittenHeaders);
remainder.set(body, rewrittenHeaders.byteLength);
}
const compiled = new Uint8Array(requestLine.byteLength + remainder.byteLength);
compiled.set(requestLine);
compiled.set(remainder, requestLine.byteLength);
return {
version: 1,
baselineId: input.baselineId,
selector,
method: method as BrowserAuthorizationCompiledRequest['method'],
url: input.publicUrl,
isHttps: input.isHttps,
rawRequestBase64: bytesToBase64(compiled),
resourceValueFingerprint: input.replacement.valueFingerprint,
packetFingerprint: `sha256:${[...new Uint8Array(await crypto.subtle.digest(
'SHA-256',
Uint8Array.from(compiled).buffer,
))].map((byte) => byte.toString(16).padStart(2, '0')).join('')}`,
};
}
function normalizedTransformDestination(destination: string): string {
const trimmed = destination.trim();
if (trimmed.toLowerCase().startsWith('header.')) {
return `header.${trimmed.slice(7).trim().toLowerCase()}`;
}
return trimmed;
}
function queryValueMap(url: URL): Map<string, string[]> {
const output = new Map<string, string[]>();
for (const [name, value] of url.searchParams) {
output.set(name, [...(output.get(name) || []), value]);
}
return output;
}
function sameStringValues(left: string[] | undefined, right: string[] | undefined): boolean {
return JSON.stringify(left || []) === JSON.stringify(right || []);
}
export function authorizationRequestToTransformPacket(
rawRequestBase64: string,
origin: string,
): BrowserTransformPacket {
const parsed = parseAuthorizationRequestPacket(rawRequestBase64);
let url: URL;
try {
url = new URL(parsed.requestTarget, origin);
} catch {
throw new ExtensionError('authorization_baseline_invalid', '授权基线请求目标无法转换为页面报文');
}
if (url.origin !== origin || url.hash) {
throw new ExtensionError('authorization_origin_changed', '授权基线请求目标超出了认证来源');
}
return {
method: parsed.method,
url: url.toString(),
headers: parsed.headers,
bodyBase64: bytesToBase64(parsed.bytes.subarray(parsed.bodyOffset)),
};
}
export async function applyAuthorizationTransformExecution(input: {
compiled: BrowserAuthorizationCompiledRequest;
execution: BrowserTransformExecution;
origin: string;
allowedDestinations: string[];
allowBody?: boolean;
}): Promise<BrowserAuthorizationCompiledRequest> {
const packet = parseAuthorizationRequestPacket(input.compiled.rawRequestBase64);
const baselinePacket = authorizationRequestToTransformPacket(
input.compiled.rawRequestBase64,
input.origin,
);
const allowed = new Set(input.allowedDestinations.map(normalizedTransformDestination));
const bodyChanged = input.execution.bodyBase64 !== baselinePacket.bodyBase64;
const bodyAllowed = input.allowBody && [...allowed].some(
(destination) => destination === 'body'
|| destination.startsWith('body.')
|| destination.startsWith('body['),
);
if (bodyChanged && !bodyAllowed) {
throw new ExtensionError(
'authorization_transform_unsupported',
'授权动态重算只有在逻辑明文绑定后才能改写 Body',
);
}
let transformedURL: URL;
const originalURL = new URL(baselinePacket.url);
try {
transformedURL = new URL(input.execution.url);
} catch {
throw new ExtensionError('authorization_transform_invalid', 'Transform Profile 返回了无效 URL');
}
if (
transformedURL.origin !== input.origin
|| transformedURL.pathname !== originalURL.pathname
|| transformedURL.hash
) {
throw new ExtensionError(
'authorization_transform_invalid',
'动态重算不能改变请求来源、路径或 fragment',
);
}
const originalQuery = queryValueMap(originalURL);
const transformedQuery = queryValueMap(transformedURL);
const queryNames = new Set([...originalQuery.keys(), ...transformedQuery.keys()]);
for (const name of queryNames) {
if (
!sameStringValues(originalQuery.get(name), transformedQuery.get(name))
&& !allowed.has(`query.${name}`)
) {
throw new ExtensionError(
'authorization_transform_invalid',
`Transform Profile 改写了未声明的查询字段: ${name}`,
);
}
}
const forbiddenHeaders = new Set(['authorization', 'cookie', 'proxy-authorization', 'host']);
const removed = new Set<string>();
for (const name of input.execution.removeHeaders) {
const normalized = name.trim().toLowerCase();
if (
forbiddenHeaders.has(normalized)
|| !allowed.has(`header.${normalized}`)
) {
throw new ExtensionError(
'authorization_transform_invalid',
`Transform Profile 尝试删除认证材料或未声明 Header: ${name}`,
);
}
removed.add(normalized);
}
const replacements = new Map<string, { name: string; value: string }>();
for (const header of input.execution.setHeaders) {
const normalized = header.name.trim().toLowerCase();
if (
!normalized
|| /[\r\n:]/.test(header.name)
|| /[\r\n]/.test(header.value)
|| forbiddenHeaders.has(normalized)
|| !allowed.has(`header.${normalized}`)
) {
throw new ExtensionError(
'authorization_transform_invalid',
`Transform Profile 尝试改写认证材料或未声明 Header: ${header.name}`,
);
}
replacements.set(normalized, { name: header.name.trim(), value: header.value });
removed.delete(normalized);
}
let headers = packet.headers.filter(
(header) => !removed.has(header.name.toLowerCase())
&& !replacements.has(header.name.toLowerCase()),
);
headers.push(...replacements.values());
const host = headers.find((header) => header.name.toLowerCase() === 'host')?.value;
if (!host || host !== transformedURL.host) {
throw new ExtensionError('authorization_transform_invalid', '动态重算后的 Host 与认证来源不一致');
}
const body = bodyChanged
? base64ToBytes(input.execution.bodyBase64)
: packet.bytes.subarray(packet.bodyOffset);
if (body.byteLength > 2 * 1_024 * 1_024) {
throw new ExtensionError('authorization_transform_invalid', '动态重算后的请求 Body 超过 2 MiB 上限');
}
if (bodyChanged) {
headers = headers.filter((header) => {
const name = header.name.toLowerCase();
return name !== 'content-length' && name !== 'transfer-encoding';
});
headers.push({ name: 'Content-Length', value: String(body.byteLength) });
}
const head = [
`${packet.method} ${transformedURL.pathname || '/'}${transformedURL.search} ${packet.protocol}`,
...headers.map((header) => `${header.name}: ${header.value}`),
'',
'',
].join('\r\n');
const headBytes = new TextEncoder().encode(head);
const raw = new Uint8Array(headBytes.byteLength + body.byteLength);
raw.set(headBytes);
raw.set(body, headBytes.byteLength);
return {
...input.compiled,
rawRequestBase64: bytesToBase64(raw),
packetFingerprint: `sha256:${[...new Uint8Array(await crypto.subtle.digest(
'SHA-256',
Uint8Array.from(raw).buffer,
))].map((byte) => byte.toString(16).padStart(2, '0')).join('')}`,
};
}
@@ -1,262 +0,0 @@
import { beforeEach, describe, expect, it, vi } from 'vitest';
const mocks = vi.hoisted(() => ({
session: {} as Record<string, unknown>,
getContext: vi.fn(),
loadLogicalBinding: vi.fn(),
listNetworkRequests: vi.fn(),
exportNetworkRequest: vi.fn(),
}));
vi.mock('wxt/browser', () => ({
browser: {
storage: {
session: {
async get(key: string) {
return key in mocks.session
? { [key]: structuredClone(mocks.session[key]) }
: {};
},
async set(values: Record<string, unknown>) {
Object.assign(mocks.session, structuredClone(values));
},
},
},
},
}));
vi.mock('./auth-context', () => ({
getAuthContextHandle: (...args: unknown[]) => mocks.getContext(...args),
}));
vi.mock('./auth-attestation', () => ({
getAuthContextAttestation: (...args: unknown[]) => mocks.getContext(...args),
}));
vi.mock('@/features/network-capture/service', () => ({
exportNetworkRequest: (...args: unknown[]) => mocks.exportNetworkRequest(...args),
listNetworkRequests: (...args: unknown[]) => mocks.listNetworkRequests(...args),
}));
vi.mock('@/features/browser-transform/service', () => ({
executeBrowserTransform: vi.fn(),
getBrowserTransformProfile: vi.fn(),
}));
vi.mock('@/features/browser-transform/replay-draft', () => ({
browserTransformReplayDraftToPacket: vi.fn(),
getBrowserTransformReplayDraft: vi.fn(),
}));
vi.mock('./logical-binding', () => ({
assertAuthorizationLogicalPacketStructure: vi.fn(),
authorizationPacketFingerprint: vi.fn(),
buildAuthorizationLogicalRequestBinding: vi.fn(),
decodeAndVerifyLogicalReplacement: vi.fn(),
loadAuthorizationLogicalRequestBinding: (...args: unknown[]) => (
mocks.loadLogicalBinding(...args)
),
readAuthorizationLogicalResource: vi.fn(),
replaceAuthorizationLogicalResource: vi.fn(),
}));
const storageKey = 'browser.authorization.baselines.v1';
const expiresAt = 4_102_444_800_000;
const fingerprint = `sha256:${'a'.repeat(64)}`;
function target(documentId = 'document-a') {
return { tabId: 7, frameId: 0, documentId };
}
function context(documentId = 'document-a') {
return {
version: 1,
id: 'context-a',
slotId: 'left',
deviceId: 'device-a',
installationId: 'installation-a',
isolationContextId: 'isolation-a',
isolationProofId: 'proof-a',
cookieStoreId: 'store-a',
origin: 'https://example.test',
grantId: 'grant-a',
target: target(documentId),
fingerprint,
authentication: {
status: 'authenticated',
cookieCount: 1,
storageEntryCount: 0,
authCookieNames: ['session'],
authStorageKeys: [],
},
createdAt: 1,
expiresAt,
};
}
function storedBaseline(withLogicalBinding = false) {
const request = {
method: 'GET',
url: 'https://example.test/account',
path: '/account',
contentType: '',
actionFingerprint: fingerprint,
headerNames: ['cookie'],
fields: [],
};
const snapshot = {
version: 1,
id: 'baseline-a',
deviceId: 'device-a',
installationId: 'installation-a',
isolationContextId: 'isolation-a',
cookieStoreId: 'store-a',
origin: 'https://example.test',
grantId: 'grant-a',
target: target(),
authContextReference: { kind: 'handle', id: 'context-a' },
networkRequestId: 'request-a',
request,
createdAt: 1,
expiresAt,
...(withLogicalBinding ? {
logicalRequest: {
version: 1,
source: 'local-replay-draft',
baselineId: 'baseline-a',
profileId: 'profile-a',
profileName: 'account gateway',
isolationContextId: 'isolation-a',
cookieStoreId: 'store-a',
target: target(),
origin: 'https://example.test',
request,
outputDestinations: ['body.encryptedData'],
validation: {
proofLevel: 'structure',
summary: 'validated',
warnings: [],
},
bindingFingerprint: fingerprint,
profileUpdatedAt: 2,
replayUpdatedAt: 2,
createdAt: 2,
expiresAt,
},
} : {}),
};
return {
snapshot,
rawRequestBase64: btoa('GET /account HTTP/1.1\r\nHost: example.test\r\n\r\n'),
requestUrl: 'https://example.test/account',
isHttps: true,
};
}
async function loadService() {
return import('./baseline');
}
describe('authorization baseline lifecycle recovery', () => {
beforeEach(() => {
vi.resetModules();
for (const key of Object.keys(mocks.session)) delete mocks.session[key];
mocks.getContext.mockReset().mockResolvedValue(context());
mocks.loadLogicalBinding.mockReset().mockResolvedValue({});
mocks.listNetworkRequests.mockReset().mockResolvedValue([]);
mocks.exportNetworkRequest.mockReset();
});
it('invalidates and removes a baseline after its page document changes', async () => {
mocks.session[storageKey] = [storedBaseline()];
mocks.getContext.mockResolvedValue(context('document-b'));
const { getAuthorizationBaseline } = await loadService();
await expect(
getAuthorizationBaseline('baseline-a', 'grant-a'),
).rejects.toMatchObject({ code: 'authorization_baseline_stale' });
expect(mocks.session[storageKey]).toEqual([]);
});
it('invalidates and removes a baseline after its isolation context disappears', async () => {
mocks.session[storageKey] = [storedBaseline()];
mocks.getContext.mockRejectedValue(new Error('context unavailable'));
const { getAuthorizationBaseline } = await loadService();
await expect(
getAuthorizationBaseline('baseline-a', 'grant-a'),
).rejects.toMatchObject({ code: 'authorization_baseline_stale' });
expect(mocks.session[storageKey]).toEqual([]);
});
it('drops only the logical binding when its callable or Profile proof changes', async () => {
mocks.session[storageKey] = [storedBaseline(true)];
mocks.loadLogicalBinding.mockRejectedValue(new Error('binding changed'));
const { getAuthorizationBaseline } = await loadService();
const baseline = await getAuthorizationBaseline('baseline-a', 'grant-a');
expect(baseline.logicalRequest).toBeUndefined();
const retained = mocks.session[storageKey] as Array<{
snapshot: { logicalRequest?: unknown };
}>;
expect(retained).toHaveLength(1);
expect(retained[0].snapshot.logicalRequest).toBeUndefined();
});
it('shows same-site WebSocket handshakes as an explicit fail-closed boundary', async () => {
mocks.listNetworkRequests.mockResolvedValue([{
id: 'socket-a',
requestId: 'request-socket-a',
tabId: 7,
frameId: 0,
documentId: 'document-a',
url: 'wss://example.test/events?tenant=alpha',
method: 'GET',
resourceType: 'websocket',
startedAt: 100,
completedAt: 101,
statusCode: 101,
requestHeadersCaptured: true,
requestBodyCaptured: true,
redirects: [],
}]);
const { listAuthorizationBaselineCandidates } = await loadService();
const candidates = await listAuthorizationBaselineCandidates({
target: target(),
grantId: 'grant-a',
authContextKind: 'handle',
authContextId: 'context-a',
limit: 20,
});
expect(candidates).toHaveLength(1);
expect(candidates[0]).toMatchObject({
id: 'socket-a',
resourceType: 'websocket',
eligible: false,
});
expect(candidates[0].reasons[0]).toContain('不会进入 HTTP 授权矩阵');
});
it('rejects a WebSocket handshake even when called outside candidate selection', async () => {
mocks.exportNetworkRequest.mockResolvedValue({
id: 'socket-a',
url: 'wss://example.test/events',
isHttps: true,
rawRequestBase64: btoa('GET /events HTTP/1.1\r\nHost: example.test\r\n\r\n'),
limitations: [],
});
const { captureAuthorizationBaseline } = await loadService();
await expect(captureAuthorizationBaseline({
target: target(),
grantId: 'grant-a',
authContextKind: 'handle',
authContextId: 'context-a',
networkRequestId: 'socket-a',
comparisonKey: 'A'.repeat(43),
})).rejects.toMatchObject({ code: 'authorization_protocol_unsupported' });
});
});
@@ -1,289 +0,0 @@
import { describe, expect, it } from 'vitest';
import { parseAuthorizationBaselineRequest } from './baseline-metadata';
const comparisonKey = 'A'.repeat(43);
function base64(value: string): string {
const bytes = new TextEncoder().encode(value);
let binary = '';
for (const byte of bytes) binary += String.fromCharCode(byte);
return btoa(binary);
}
function request(orderId: number, token: string): string {
const body = JSON.stringify({
orderId,
profile: { userId: `user-${orderId}` },
password: `password-${orderId}`,
clientSecret: `client-secret-${orderId}`,
note: 'visible-business-value',
});
return [
'POST /api/orders?tenantId=tenant-a HTTP/1.1',
'Host: example.test',
'Content-Type: application/json',
`Authorization: Bearer ${token}`,
`Cookie: session=${token}`,
'X-CSRF-Token: csrf-secret',
`X-Tenant-Id: tenant-${orderId}`,
'',
body,
].join('\r\n');
}
function pathRequest(orderId: number): string {
return [
`GET /api/orders/${orderId} HTTP/1.1`,
'Host: example.test',
'Accept: application/json',
'',
'',
].join('\r\n');
}
function graphqlRequest(input: {
operationName: string;
query: string;
orderId: number;
password?: string;
}): string {
const body = JSON.stringify({
operationName: input.operationName,
query: input.query,
variables: {
orderId: input.orderId,
password: input.password || `password-${input.orderId}`,
},
});
return [
'POST /graphql HTTP/1.1',
'Host: example.test',
'Content-Type: application/json',
'',
body,
].join('\r\n');
}
describe('authorization baseline request metadata', () => {
it('returns structural evidence and comparable fingerprints without raw values', async () => {
const metadata = await parseAuthorizationBaselineRequest(
base64(request(42, 'token-secret')),
'https://example.test/api/orders?tenantId=tenant-a',
comparisonKey,
);
const serialized = JSON.stringify(metadata);
expect(metadata.method).toBe('POST');
expect(metadata.url).toBe('https://example.test/api/orders');
expect(metadata.path).toBe('/api/orders');
expect(serialized).not.toContain('token-secret');
expect(serialized).not.toContain('csrf-secret');
expect(serialized).not.toContain('visible-business-value');
expect(metadata.fields.find((field) => field.path === 'header.authorization')).toMatchObject({
category: 'authentication',
valueType: 'string',
});
expect(metadata.fields.find((field) => field.path === 'header.x-csrf-token')).toMatchObject({
category: 'csrf',
});
expect(metadata.fields.find((field) => field.path === 'body.orderId')).toMatchObject({
category: 'resource',
valueType: 'number',
});
expect(metadata.fields.find((field) => field.path === 'body.password')).toMatchObject({
category: 'authentication',
});
expect(metadata.fields.find((field) => field.path === 'body.clientSecret')).toMatchObject({
category: 'authentication',
});
expect(metadata.fields.find((field) => field.path === 'header.x-tenant-id')).toMatchObject({
category: 'resource',
valueType: 'string',
});
});
it('keeps action shape stable while exposing value changes through a shared workspace HMAC', async () => {
const left = await parseAuthorizationBaselineRequest(
base64(request(42, 'token-left')),
'https://example.test/api/orders?tenantId=tenant-a',
comparisonKey,
);
const right = await parseAuthorizationBaselineRequest(
base64(request(84, 'token-right')),
'https://example.test/api/orders?tenantId=tenant-a',
comparisonKey,
);
const leftOrder = left.fields.find((field) => field.path === 'body.orderId');
const rightOrder = right.fields.find((field) => field.path === 'body.orderId');
const leftTenant = left.fields.find((field) => field.path === 'query.tenantId');
const rightTenant = right.fields.find((field) => field.path === 'query.tenantId');
expect(left.actionFingerprint).toBe(right.actionFingerprint);
expect(leftOrder?.valueFingerprint).not.toBe(rightOrder?.valueFingerprint);
expect(leftTenant?.valueFingerprint).toBe(rightTenant?.valueFingerprint);
});
it('rejects caller-supplied comparison keys with the wrong size', async () => {
await expect(parseAuthorizationBaselineRequest(
base64(request(42, 'token')),
'https://example.test/api/orders',
'A'.repeat(42),
)).rejects.toThrow('32 字节');
});
it('normalizes path identifiers while retaining a comparable resource selector', async () => {
const left = await parseAuthorizationBaselineRequest(
base64(pathRequest(42)),
'https://example.test/api/orders/42',
comparisonKey,
);
const right = await parseAuthorizationBaselineRequest(
base64(pathRequest(84)),
'https://example.test/api/orders/84',
comparisonKey,
);
const leftResource = left.fields.find((field) => field.path === 'path.segment[2]');
const rightResource = right.fields.find((field) => field.path === 'path.segment[2]');
expect(left.path).toBe('/api/orders/:resource');
expect(left.url).toBe('https://example.test/api/orders/:resource');
expect(left.actionFingerprint).toBe(right.actionFingerprint);
expect(leftResource).toMatchObject({ location: 'path', category: 'resource' });
expect(leftResource?.valueFingerprint).not.toBe(rightResource?.valueFingerprint);
});
it('pairs the same GraphQL operation while exposing variables as typed resource fields', async () => {
const query = 'query Order($orderId: ID!) { order(id: $orderId) { id total } }';
const left = await parseAuthorizationBaselineRequest(
base64(graphqlRequest({
operationName: 'Order',
query,
orderId: 42,
})),
'https://example.test/graphql',
comparisonKey,
);
const right = await parseAuthorizationBaselineRequest(
base64(graphqlRequest({
operationName: 'Order',
query,
orderId: 84,
})),
'https://example.test/graphql',
comparisonKey,
);
expect(left).toMatchObject({
protocol: 'graphql',
operationNames: ['Order'],
});
expect(left.operationFingerprint).toBe(right.operationFingerprint);
expect(left.actionFingerprint).toBe(right.actionFingerprint);
expect(left.fields.find((item) => item.path === 'body.variables.orderId')).toMatchObject({
location: 'body',
category: 'resource',
valueType: 'number',
});
expect(left.fields.find((item) => item.path === 'body.variables.password')).toMatchObject({
category: 'authentication',
});
expect(JSON.stringify(left)).not.toContain(query);
});
it('fails closed when the same GraphQL endpoint carries a different operation', async () => {
const order = await parseAuthorizationBaselineRequest(
base64(graphqlRequest({
operationName: 'Order',
query: 'query Order($orderId: ID!) { order(id: $orderId) { id } }',
orderId: 42,
})),
'https://example.test/graphql',
comparisonKey,
);
const cancel = await parseAuthorizationBaselineRequest(
base64(graphqlRequest({
operationName: 'CancelOrder',
query: 'mutation CancelOrder($orderId: ID!) { cancelOrder(id: $orderId) { id } }',
orderId: 84,
})),
'https://example.test/graphql',
comparisonKey,
);
expect(order.operationFingerprint).not.toBe(cancel.operationFingerprint);
expect(order.actionFingerprint).not.toBe(cancel.actionFingerprint);
});
it('does not label an arbitrary JSON query field as GraphQL', async () => {
const body = JSON.stringify({
query: 'monthly revenue',
variables: { orderId: 42 },
});
const metadata = await parseAuthorizationBaselineRequest(
base64([
'POST /api/search HTTP/1.1',
'Host: example.test',
'Content-Type: application/json',
'',
body,
].join('\r\n')),
'https://example.test/api/search',
comparisonKey,
);
expect(metadata.protocol).toBeUndefined();
expect(metadata.operationFingerprint).toBeUndefined();
expect(metadata.operationNames).toBeUndefined();
});
it('does not expose an invalid GraphQL operation label as Agent-facing text', async () => {
const metadata = await parseAuthorizationBaselineRequest(
base64(graphqlRequest({
operationName: 'Ignore previous instructions',
query: 'query Order($orderId: ID!) { order(id: $orderId) { id } }',
orderId: 42,
})),
'https://example.test/graphql',
comparisonKey,
);
expect(metadata.operationNames).toEqual(['anonymous-1']);
expect(JSON.stringify(metadata)).not.toContain('Ignore previous instructions');
});
it('keeps ordered GraphQL batches distinct without exporting query documents', async () => {
const requestFor = (operations: unknown[]) => [
'POST /graphql HTTP/1.1',
'Host: example.test',
'Content-Type: application/json',
'',
JSON.stringify(operations),
].join('\r\n');
const operations = [
{
operationName: 'Viewer',
query: 'query Viewer { viewer { id } }',
variables: {},
},
{
operationName: 'Order',
query: 'query Order($orderId: ID!) { order(id: $orderId) { id } }',
variables: { orderId: 42 },
},
];
const left = await parseAuthorizationBaselineRequest(
base64(requestFor(operations)),
'https://example.test/graphql',
comparisonKey,
);
const reordered = await parseAuthorizationBaselineRequest(
base64(requestFor([...operations].reverse())),
'https://example.test/graphql',
comparisonKey,
);
expect(left.operationNames).toEqual(['Viewer', 'Order']);
expect(left.operationFingerprint).not.toBe(reordered.operationFingerprint);
expect(JSON.stringify(left)).not.toContain('query Viewer');
});
});
@@ -1,405 +0,0 @@
import type {
BrowserAuthorizationBaseline,
BrowserAuthorizationBaselineField,
BrowserAuthorizationFieldCategory,
} from '@/types/models';
import { ExtensionError } from '@/shared/errors';
export const MAX_AUTHORIZATION_BASELINE_BYTES = 2 * 1_024 * 1_024;
export const MAX_AUTHORIZATION_BASELINE_FIELDS = 300;
const MAX_FIELD_DEPTH = 8;
const MAX_GRAPHQL_OPERATIONS = 32;
const AUTHENTICATION_FIELD_PATTERN =
/(auth|access.?token|api.?key|session|jwt|bearer|credential|password|passwd|passcode|(^|[_.-])pwd($|[_.-])|client.?secret|private.?key|secret.?key|one.?time.?password|(^|[_.-])otp($|[_.-])|mfa.?code|verification.?code|(^|[_.-])pin($|[_.-])|captcha)/;
function base64ToBytes(value: string): Uint8Array {
const binary = atob(value);
return Uint8Array.from(binary, (character) => character.charCodeAt(0));
}
function base64UrlToBytes(value: string): Uint8Array {
const normalized = value.replace(/-/g, '+').replace(/_/g, '/');
return base64ToBytes(normalized.padEnd(Math.ceil(normalized.length / 4) * 4, '='));
}
function bytesToHex(bytes: Uint8Array): string {
return [...bytes].map((byte) => byte.toString(16).padStart(2, '0')).join('');
}
async function comparisonSigner(
encodedKey: string,
): Promise<(value: string | Uint8Array) => Promise<string>> {
let keyBytes: Uint8Array;
try {
keyBytes = base64UrlToBytes(encodedKey);
} catch {
throw new ExtensionError('authorization_invalid', '基线比较密钥格式无效');
}
if (keyBytes.byteLength !== 32) {
throw new ExtensionError('authorization_invalid', '基线比较密钥必须为 32 字节');
}
const key = await crypto.subtle.importKey(
'raw',
Uint8Array.from(keyBytes).buffer,
{ name: 'HMAC', hash: 'SHA-256' },
false,
['sign'],
);
return async (value: string | Uint8Array) => {
const bytes = typeof value === 'string'
? new TextEncoder().encode(value)
: Uint8Array.from(value);
const signature = await crypto.subtle.sign(
'HMAC',
key,
bytes.buffer,
);
return `workspace-hmac-sha256:${bytesToHex(new Uint8Array(signature))}`;
};
}
export async function fingerprintAuthorizationComparisonValue(
encodedKey: string,
value: string | Uint8Array,
): Promise<string> {
return (await comparisonSigner(encodedKey))(value);
}
async function sha256(value: string): Promise<string> {
const digest = await crypto.subtle.digest('SHA-256', new TextEncoder().encode(value));
return bytesToHex(new Uint8Array(digest));
}
interface GraphQLProtocolMetadata {
protocol: 'graphql';
operationFingerprint: string;
operationNames: string[];
}
function graphqlPersistedQueryHash(value: Record<string, unknown>): string {
const extensions = value.extensions;
if (!extensions || typeof extensions !== 'object' || Array.isArray(extensions)) return '';
const persisted = (extensions as Record<string, unknown>).persistedQuery;
if (!persisted || typeof persisted !== 'object' || Array.isArray(persisted)) return '';
const hash = (persisted as Record<string, unknown>).sha256Hash;
return typeof hash === 'string' && /^[a-f0-9]{64}$/i.test(hash) ? hash.toLowerCase() : '';
}
function looksLikeGraphQLDocument(value: string): boolean {
const normalized = value
.replace(/^\uFEFF/, '')
.replace(/(?:^|\n)\s*#[^\n]*/g, '\n')
.trimStart();
return /^(?:query|mutation|subscription|fragment)\b/.test(normalized)
|| normalized.startsWith('{');
}
function displayGraphQLOperationName(value: unknown, index: number): string {
if (typeof value !== 'string') return `anonymous-${index + 1}`;
const normalized = value.trim();
return /^[A-Za-z_][A-Za-z0-9_]{0,127}$/.test(normalized)
? normalized
: `anonymous-${index + 1}`;
}
async function graphqlProtocolMetadata(value: unknown): Promise<GraphQLProtocolMetadata | undefined> {
const operations = Array.isArray(value) ? value : [value];
if (!operations.length) return undefined;
if (operations.length > MAX_GRAPHQL_OPERATIONS) {
const allGraphQL = operations.every((operation) => {
if (!operation || typeof operation !== 'object' || Array.isArray(operation)) return false;
const envelope = operation as Record<string, unknown>;
return (
typeof envelope.query === 'string'
&& looksLikeGraphQLDocument(envelope.query)
) || Boolean(graphqlPersistedQueryHash(envelope));
});
if (!allGraphQL) return undefined;
const serialized = JSON.stringify(value);
return {
protocol: 'graphql',
operationFingerprint: `sha256:${await sha256(serialized)}`,
operationNames: [`batch-overflow-${operations.length}`],
};
}
const descriptors: Array<{
operationNameFingerprint: string;
queryFingerprint: string;
persistedQueryFingerprint: string;
}> = [];
const operationNames: string[] = [];
for (const [index, operation] of operations.entries()) {
if (!operation || typeof operation !== 'object' || Array.isArray(operation)) return undefined;
const envelope = operation as Record<string, unknown>;
const query = typeof envelope.query === 'string'
&& looksLikeGraphQLDocument(envelope.query)
? envelope.query
: '';
const persistedQueryHash = graphqlPersistedQueryHash(envelope);
if (!query && !persistedQueryHash) return undefined;
const operationName = typeof envelope.operationName === 'string'
? envelope.operationName
: '';
descriptors.push({
operationNameFingerprint: await sha256(operationName),
queryFingerprint: query ? await sha256(query.replace(/\r\n?/g, '\n').trim()) : '',
persistedQueryFingerprint: persistedQueryHash ? await sha256(persistedQueryHash) : '',
});
operationNames.push(displayGraphQLOperationName(envelope.operationName, index));
}
return {
protocol: 'graphql',
operationFingerprint: `sha256:${await sha256(JSON.stringify({
version: 1,
operations: descriptors,
}))}`,
operationNames: operationNames.slice(0, 16),
};
}
function category(name: string): BrowserAuthorizationFieldCategory {
const normalized = name.toLowerCase();
if (normalized === 'authorization'
|| normalized === 'cookie'
|| AUTHENTICATION_FIELD_PATTERN.test(normalized)) {
return 'authentication';
}
if (/(csrf|xsrf)/.test(normalized)) return 'csrf';
if (/(signature|(^|[_.-])sign(ed)?($|[_.-])|hmac)/.test(normalized)) return 'signature';
if (/(nonce|random|request.?id|trace.?id|correlation.?id|idempotency)/.test(normalized)) return 'nonce';
if (/(timestamp|(^|[_.-])time($|[_.-])|(^|[_.-])date($|[_.-]))/.test(normalized)) return 'timestamp';
if (/(^|[_.\-[\]])(id|uid|user.?id|account.?id|tenant.?id|org(anization)?.?id|workspace.?id|project.?id|team.?id|customer.?id|order.?id|resource.?id|object.?id|record.?id|document.?id|file.?id|invoice.?id)($|[_.\-[\]])/.test(normalized)) {
return 'resource';
}
return 'unknown';
}
function primitiveType(value: unknown): BrowserAuthorizationBaselineField['valueType'] {
if (value === null) return 'null';
if (typeof value === 'number') return 'number';
if (typeof value === 'boolean') return 'boolean';
return 'string';
}
function primitiveText(value: unknown): string {
if (value === null) return 'null';
if (typeof value === 'string') return value;
return JSON.stringify(value);
}
async function field(
location: BrowserAuthorizationBaselineField['location'],
path: string,
value: unknown,
sign: (value: string | Uint8Array) => Promise<string>,
valueType: BrowserAuthorizationBaselineField['valueType'] = primitiveType(value),
categoryOverride?: BrowserAuthorizationFieldCategory,
): Promise<BrowserAuthorizationBaselineField> {
const text = primitiveText(value);
return {
location,
path,
valueType,
byteLength: new TextEncoder().encode(text).byteLength,
valueFingerprint: await sign(text),
category: categoryOverride ?? category(path),
};
}
async function flattenJSON(
value: unknown,
sign: (value: string | Uint8Array) => Promise<string>,
): Promise<BrowserAuthorizationBaselineField[]> {
const pending: Array<{ value: unknown; path: string; depth: number }> = [{
value,
path: 'body',
depth: 0,
}];
const output: BrowserAuthorizationBaselineField[] = [];
while (pending.length && output.length < MAX_AUTHORIZATION_BASELINE_FIELDS) {
const current = pending.shift()!;
if (current.depth > MAX_FIELD_DEPTH) continue;
if (Array.isArray(current.value)) {
current.value.slice(0, 50).forEach((child, index) => {
pending.push({ value: child, path: `${current.path}[${index}]`, depth: current.depth + 1 });
});
continue;
}
if (current.value && typeof current.value === 'object') {
Object.entries(current.value as Record<string, unknown>)
.slice(0, 100)
.forEach(([key, child]) => {
pending.push({ value: child, path: `${current.path}.${key}`, depth: current.depth + 1 });
});
continue;
}
output.push(await field('body', current.path, current.value, sign));
}
return output;
}
function headerValues(lines: string[]): Array<{ name: string; value: string }> {
const output: Array<{ name: string; value: string }> = [];
for (const line of lines) {
const separator = line.indexOf(':');
if (separator <= 0) continue;
output.push({
name: line.slice(0, separator).trim().slice(0, 512),
value: line.slice(separator + 1).trim(),
});
}
return output;
}
function indexedFieldPaths(
entries: Array<[string, string]>,
prefix: 'header' | 'query' | 'body',
): Array<{ path: string; value: string }> {
const totals = new Map<string, number>();
for (const [name] of entries) totals.set(name, (totals.get(name) || 0) + 1);
const indexes = new Map<string, number>();
return entries.map(([name, value]) => {
const index = indexes.get(name) || 0;
indexes.set(name, index + 1);
return {
path: totals.get(name) === 1 ? `${prefix}.${name}` : `${prefix}.${name}[${index}]`,
value,
};
});
}
function decodePathSegment(value: string): string {
try {
return decodeURIComponent(value);
} catch {
return value;
}
}
function dynamicPathSegment(value: string): boolean {
const decoded = decodePathSegment(value);
return /^\d+$/.test(decoded)
|| /^[0-9a-f]{8}-[0-9a-f-]{27,}$/i.test(decoded)
|| /^[0-9a-f]{12,}$/i.test(decoded)
|| /^[A-Za-z0-9_-]{16,}$/.test(decoded);
}
export function normalizeAuthorizationPath(pathname: string): {
normalized: string;
resources: Array<{ path: string; value: string }>;
} {
const segments = pathname.split('/').filter(Boolean);
const resources: Array<{ path: string; value: string }> = [];
const normalized = segments.map((segment, index) => {
if (!dynamicPathSegment(segment)) return segment;
resources.push({
path: `path.segment[${index}]`,
value: decodePathSegment(segment),
});
return ':resource';
});
return {
normalized: `/${normalized.join('/')}`,
resources,
};
}
function bodyOffset(bytes: Uint8Array): number {
for (let index = 0; index <= bytes.length - 4; index += 1) {
if (bytes[index] === 13 && bytes[index + 1] === 10
&& bytes[index + 2] === 13 && bytes[index + 3] === 10) {
return index + 4;
}
}
throw new ExtensionError('authorization_baseline_invalid', '捕获请求缺少 HTTP Header 分隔符');
}
export async function parseAuthorizationBaselineRequest(
rawRequestBase64: string,
requestUrl: string,
encodedComparisonKey: string,
): Promise<BrowserAuthorizationBaseline['request']> {
const bytes = base64ToBytes(rawRequestBase64);
if (!bytes.length || bytes.byteLength > MAX_AUTHORIZATION_BASELINE_BYTES) {
throw new ExtensionError('authorization_baseline_too_large', '授权基线请求必须在 1 字节到 2 MiB 之间');
}
const offset = bodyOffset(bytes);
const head = new TextDecoder('utf-8', { fatal: true }).decode(bytes.subarray(0, offset - 4));
const lines = head.split('\r\n');
const requestLine = lines.shift()?.split(/\s+/) || [];
if (requestLine.length !== 3) {
throw new ExtensionError('authorization_baseline_invalid', '授权基线请求行无效');
}
const method = requestLine[0].toUpperCase().slice(0, 32);
const parsedUrl = new URL(requestUrl);
const shapedPath = normalizeAuthorizationPath(parsedUrl.pathname);
const headers = headerValues(lines);
const contentType = headers.find((header) => header.name.toLowerCase() === 'content-type')?.value || '';
const sign = await comparisonSigner(encodedComparisonKey);
const fields: BrowserAuthorizationBaselineField[] = [];
const indexedHeaders = indexedFieldPaths(
headers.slice(0, 256).map((header) => [header.name.toLowerCase(), header.value]),
'header',
);
for (const header of indexedHeaders) {
fields.push(await field('header', header.path, header.value, sign));
}
for (const resource of shapedPath.resources) {
fields.push(await field(
'path',
resource.path,
resource.value,
sign,
primitiveType(resource.value),
'resource',
));
}
for (const parameter of indexedFieldPaths([...parsedUrl.searchParams], 'query')) {
if (fields.length >= MAX_AUTHORIZATION_BASELINE_FIELDS) break;
fields.push(await field('query', parameter.path, parameter.value, sign));
}
const body = bytes.subarray(offset);
let protocolMetadata: GraphQLProtocolMetadata | undefined;
if (body.byteLength && fields.length < MAX_AUTHORIZATION_BASELINE_FIELDS) {
if (contentType.toLowerCase().includes('json')) {
try {
const decoded = JSON.parse(new TextDecoder('utf-8', { fatal: true }).decode(body));
protocolMetadata = await graphqlProtocolMetadata(decoded);
fields.push(...await flattenJSON(decoded, sign));
} catch {
fields.push(await field('body', 'body', bytesToHex(body), sign, 'binary'));
}
} else if (contentType.toLowerCase().includes('application/x-www-form-urlencoded')) {
const params = indexedFieldPaths([
...new URLSearchParams(new TextDecoder().decode(body)),
], 'body');
for (const parameter of params) {
if (fields.length >= MAX_AUTHORIZATION_BASELINE_FIELDS) break;
fields.push(await field('body', parameter.path, parameter.value, sign));
}
} else {
fields.push(await field('body', 'body', bytesToHex(body), sign, 'binary'));
}
}
const boundedFields = fields.slice(0, MAX_AUTHORIZATION_BASELINE_FIELDS);
const actionShape = JSON.stringify({
version: 2,
method,
origin: parsedUrl.origin,
path: shapedPath.normalized,
contentType: contentType.split(';')[0].trim().toLowerCase(),
protocol: protocolMetadata?.protocol || '',
operationFingerprint: protocolMetadata?.operationFingerprint || '',
fields: boundedFields.map((item) => `${item.location}:${item.path}`).sort(),
});
return {
method,
url: `${parsedUrl.origin}${shapedPath.normalized}`,
path: shapedPath.normalized,
contentType: contentType.slice(0, 512),
...protocolMetadata,
actionFingerprint: `sha256:${await sha256(actionShape)}`,
headerNames: headers.map((header) => header.name).slice(0, 256),
fields: boundedFields,
};
}
@@ -1,117 +0,0 @@
import { describe, expect, it } from 'vitest';
import type {
BrowserAuthorizationBaseline,
BrowserTransformPipelineNode,
BrowserTransformProfile,
} from '@/types/models';
import { authorizationDynamicTransformDestinations } from './baseline-transform';
function baseline(): BrowserAuthorizationBaseline {
return {
version: 1,
id: 'baseline-left',
deviceId: 'device-left',
installationId: 'installation-left',
isolationContextId: 'browser-profile:store-left',
cookieStoreId: 'store-left',
origin: 'https://example.test',
grantId: 'grant-left',
target: { tabId: 11, frameId: 0, documentId: 'document-left' },
authContextReference: { kind: 'handle', id: 'auth-left' },
networkRequestId: 'request-left',
request: {
method: 'GET',
url: 'https://example.test/api/orders/:resource',
path: '/api/orders/:resource',
contentType: '',
actionFingerprint: `sha256:${'a'.repeat(64)}`,
headerNames: ['Host', 'Cookie'],
fields: [
{
location: 'path',
path: 'path.segment[2]',
valueType: 'string',
byteLength: 2,
valueFingerprint: `workspace-hmac-sha256:${'a'.repeat(64)}`,
category: 'resource',
},
{
location: 'query',
path: 'query.nonce',
valueType: 'string',
byteLength: 8,
valueFingerprint: `workspace-hmac-sha256:${'b'.repeat(64)}`,
category: 'nonce',
},
{
location: 'header',
path: 'header.x-signature',
valueType: 'string',
byteLength: 64,
valueFingerprint: `workspace-hmac-sha256:${'c'.repeat(64)}`,
category: 'signature',
},
],
},
createdAt: 1,
expiresAt: Date.now() + 60_000,
};
}
function profile(outputs: string[]): BrowserTransformProfile {
const nodes: BrowserTransformPipelineNode[] = [
{
id: 'literal',
name: '动态值',
kind: 'builtin',
operation: 'value.literal',
inputs: [],
options: { value: 'fresh' },
},
...outputs.map((destination, index): BrowserTransformPipelineNode => ({
id: `output-${index}`,
name: destination,
kind: 'output.write',
destination,
source: { nodeId: 'literal' },
encoding: 'text',
})),
];
return {
id: 'profile-left',
name: '身份 A 动态签名',
enabled: true,
target: { tabId: 11, frameId: 0, documentId: 'document-left' },
isolationContextId: 'browser-profile:store-left',
cookieStoreId: 'store-left',
origin: 'https://example.test',
match: { methods: ['GET'], urlPattern: '*/api/orders/*' },
request: { enabled: true, nodes },
response: { enabled: false, nodes: [] },
failMode: 'closed',
maxConcurrency: 1,
createdAt: 1,
updatedAt: 2,
};
}
describe('authorization identity-bound transform contracts', () => {
it('requires the profile to cover every dynamic Header and Query field', () => {
expect(authorizationDynamicTransformDestinations(
baseline(),
profile(['query.nonce', 'header.X-Signature']),
)).toEqual(['header.x-signature', 'query.nonce']);
expect(() => authorizationDynamicTransformDestinations(
baseline(),
profile(['query.nonce']),
)).toThrow('尚未覆盖动态字段');
});
it('keeps encrypted Body envelopes fail-closed until a logical plaintext binding exists', () => {
expect(() => authorizationDynamicTransformDestinations(
baseline(),
profile(['query.nonce', 'header.X-Signature', 'body.encryptedData']),
)).toThrow('Body 加密 envelope');
});
});
@@ -1,77 +0,0 @@
import type {
BrowserAuthorizationBaseline,
BrowserTransformProfile,
} from '@/types/models';
import { ExtensionError } from '@/shared/errors';
const DYNAMIC_FIELD_CATEGORIES = new Set(['signature', 'nonce', 'timestamp', 'csrf']);
function normalizedTransformDestination(destination: string): string {
const trimmed = destination.trim();
if (trimmed.toLowerCase().startsWith('header.')) {
return `header.${trimmed.slice(7).trim().toLowerCase()}`;
}
return trimmed;
}
export function authorizationDynamicTransformDestinations(
baseline: BrowserAuthorizationBaseline,
profile: BrowserTransformProfile,
): string[] {
if (!profile.enabled || !profile.request.enabled) {
throw new ExtensionError('authorization_transform_unavailable', '所选明文网关未启用请求转换');
}
if (profile.recovery && profile.recovery.state !== 'ready') {
throw new ExtensionError('authorization_transform_stale', '所选明文网关正在等待文档恢复或重新验证');
}
const dynamicFields = new Map(
baseline.request.fields
.filter((field) => DYNAMIC_FIELD_CATEGORIES.has(field.category))
.map((field) => [
normalizedTransformDestination(field.path),
field,
]),
);
const required = [...dynamicFields.keys()].filter((path) => {
const field = dynamicFields.get(path);
return field?.category === 'signature'
|| field?.category === 'nonce'
|| field?.category === 'timestamp';
});
if (!required.length) {
throw new ExtensionError('authorization_transform_unnecessary', '当前授权基线没有需要动态重算的签名、Nonce 或时间字段');
}
const destinations = profile.request.nodes
.filter((node) => node.kind === 'output.write')
.map((node) => normalizedTransformDestination(node.destination));
if (!destinations.length) {
throw new ExtensionError('authorization_transform_invalid', '所选明文网关没有请求输出节点');
}
for (const destination of destinations) {
if (
destination === 'body'
|| destination.startsWith('body.')
|| (!destination.startsWith('header.') && !destination.startsWith('query.'))
) {
throw new ExtensionError(
'authorization_transform_unsupported',
'首批授权动态重算只接受 Header/Query 签名字段;Body 加密 envelope 需要逻辑明文绑定',
);
}
if (!dynamicFields.has(destination)) {
throw new ExtensionError(
'authorization_transform_invalid',
`明文网关输出未对应基线中的动态字段: ${destination}`,
);
}
}
const output = [...new Set(destinations)];
const missing = required.find((path) => !output.includes(path));
if (missing) {
throw new ExtensionError(
'authorization_transform_incomplete',
`明文网关尚未覆盖动态字段: ${missing}`,
);
}
return output.sort();
}
@@ -1,779 +0,0 @@
import { browser } from 'wxt/browser';
import type {
BrowserAuthContextAttestation,
BrowserAuthContextHandle,
BrowserAuthorizationBaseline,
BrowserAuthorizationBaselineCandidate,
BrowserAuthorizationBaselinePacket,
BrowserAuthorizationCompiledRequest,
BrowserAuthorizationLogicalRequestBinding,
BrowserAuthorizationResourceSelector,
BrowserAuthorizationResourceValue,
BrowserAuthorizationTransformBinding,
BrowserTarget,
BrowserTransformProfile,
} from '@/types/models';
import {
exportNetworkRequest,
listNetworkRequests,
} from '@/features/network-capture/service';
import { ExtensionError } from '@/shared/errors';
import { getAuthContextHandle } from './auth-context';
import { getAuthContextAttestation } from './auth-attestation';
import {
MAX_AUTHORIZATION_BASELINE_BYTES,
MAX_AUTHORIZATION_BASELINE_FIELDS,
normalizeAuthorizationPath,
parseAuthorizationBaselineRequest,
} from './baseline-metadata';
import {
applyAuthorizationTransformExecution,
authorizationRequestToTransformPacket,
compileAuthorizationBaselineRequest,
extractAuthorizationResourceValue,
} from './baseline-execution';
import {
executeBrowserTransform,
getBrowserTransformProfile,
} from '@/features/browser-transform/service';
import { assertTransformRoute } from '@/features/browser-transform/mapping';
import { authorizationDynamicTransformDestinations } from './baseline-transform';
import {
assertAuthorizationLogicalPacketStructure,
authorizationPacketFingerprint,
buildAuthorizationLogicalRequestBinding,
decodeAndVerifyLogicalReplacement,
loadAuthorizationLogicalRequestBinding,
readAuthorizationLogicalResource,
replaceAuthorizationLogicalResource,
} from './logical-binding';
import {
browserTransformReplayDraftToPacket,
getBrowserTransformReplayDraft,
} from '@/features/browser-transform/replay-draft';
import {
readStructuredAuthorizationBodyValue,
} from './structured-body';
const MAX_BASELINES = 16;
const MAX_BASELINE_STORAGE_BYTES = 8 * 1_024 * 1_024;
const STORAGE_KEY = 'browser.authorization.baselines.v1';
function authorizationBytesToBase64(bytes: Uint8Array): string {
let binary = '';
const chunkSize = 0x8000;
for (let offset = 0; offset < bytes.length; offset += chunkSize) {
binary += String.fromCharCode(...bytes.subarray(offset, offset + chunkSize));
}
return btoa(binary);
}
interface StoredAuthorizationBaseline {
snapshot: BrowserAuthorizationBaseline;
rawRequestBase64: string;
requestUrl: string;
isHttps: boolean;
}
const baselines = new Map<string, StoredAuthorizationBaseline>();
let loaded = false;
function validAuthorizationRequestProtocol(value: {
protocol?: unknown;
operationFingerprint?: unknown;
operationNames?: unknown;
} | undefined): boolean {
if (!value) return false;
if (value.protocol === undefined) {
return value.operationFingerprint === undefined && value.operationNames === undefined;
}
return value.protocol === 'graphql'
&& /^sha256:[a-f0-9]{64}$/.test(String(value.operationFingerprint))
&& Array.isArray(value.operationNames)
&& value.operationNames.length > 0
&& value.operationNames.length <= 16
&& value.operationNames.every((name) => (
typeof name === 'string'
&& (
/^[A-Za-z_][A-Za-z0-9_]{0,127}$/.test(name)
|| /^(?:anonymous|batch-overflow)-[1-9][0-9]*$/.test(name)
)
));
}
function validLogicalRequestBinding(
value: unknown,
snapshot: Partial<BrowserAuthorizationBaseline>,
): value is BrowserAuthorizationLogicalRequestBinding {
if (!value || typeof value !== 'object' || Array.isArray(value)) return false;
const binding = value as Partial<BrowserAuthorizationLogicalRequestBinding>;
return binding.version === 1
&& binding.source === 'local-replay-draft'
&& binding.baselineId === snapshot.id
&& typeof binding.profileId === 'string'
&& binding.profileId.length > 0
&& typeof binding.profileName === 'string'
&& binding.profileName.length > 0
&& binding.isolationContextId === snapshot.isolationContextId
&& binding.cookieStoreId === snapshot.cookieStoreId
&& binding.origin === snapshot.origin
&& binding.target?.tabId === snapshot.target?.tabId
&& binding.target?.frameId === snapshot.target?.frameId
&& binding.target?.documentId === snapshot.target?.documentId
&& Boolean(binding.request)
&& validAuthorizationRequestProtocol(binding.request)
&& /^sha256:[a-f0-9]{64}$/.test(String(binding.request?.actionFingerprint))
&& Array.isArray(binding.request?.fields)
&& binding.request.fields.length <= MAX_AUTHORIZATION_BASELINE_FIELDS
&& Array.isArray(binding.outputDestinations)
&& binding.outputDestinations.length > 0
&& binding.outputDestinations.length <= 32
&& /^sha256:[a-f0-9]{64}$/.test(String(binding.bindingFingerprint))
&& typeof binding.profileUpdatedAt === 'number'
&& typeof binding.replayUpdatedAt === 'number'
&& binding.expiresAt === snapshot.expiresAt;
}
function validStoredBaseline(value: unknown): value is StoredAuthorizationBaseline {
if (!value || typeof value !== 'object' || Array.isArray(value)) return false;
const entry = value as Partial<StoredAuthorizationBaseline>;
const snapshot = entry.snapshot as Partial<BrowserAuthorizationBaseline> | undefined;
return snapshot?.version === 1
&& typeof snapshot.id === 'string'
&& snapshot.id.length > 0
&& typeof snapshot.deviceId === 'string'
&& typeof snapshot.installationId === 'string'
&& typeof snapshot.isolationContextId === 'string'
&& snapshot.isolationContextId.length > 0
&& typeof snapshot.cookieStoreId === 'string'
&& snapshot.cookieStoreId.length > 0
&& typeof snapshot.origin === 'string'
&& typeof snapshot.grantId === 'string'
&& typeof snapshot.networkRequestId === 'string'
&& Boolean(snapshot.target?.documentId)
&& ['handle', 'attestation'].includes(String(snapshot.authContextReference?.kind))
&& typeof snapshot.authContextReference?.id === 'string'
&& Boolean(snapshot.request)
&& validAuthorizationRequestProtocol(snapshot.request)
&& /^sha256:[a-f0-9]{64}$/.test(String(snapshot.request?.actionFingerprint))
&& Array.isArray(snapshot.request?.fields)
&& snapshot.request.fields.length <= MAX_AUTHORIZATION_BASELINE_FIELDS
&& typeof snapshot.createdAt === 'number'
&& typeof snapshot.expiresAt === 'number'
&& snapshot.expiresAt > snapshot.createdAt
&& typeof entry.rawRequestBase64 === 'string'
&& entry.rawRequestBase64.length <= Math.ceil(MAX_AUTHORIZATION_BASELINE_BYTES / 3) * 4 + 4
&& typeof entry.requestUrl === 'string'
&& entry.requestUrl.length <= 8_192
&& typeof entry.isHttps === 'boolean'
&& (
snapshot.logicalRequest === undefined
|| validLogicalRequestBinding(snapshot.logicalRequest, snapshot)
);
}
function purge(now = Date.now(), reserve = 0): boolean {
let changed = false;
for (const [id, baseline] of baselines) {
if (baseline.snapshot.expiresAt <= now) {
baselines.delete(id);
changed = true;
}
}
while (baselines.size > MAX_BASELINES - reserve) {
const oldest = baselines.keys().next().value as string | undefined;
if (!oldest) break;
baselines.delete(oldest);
changed = true;
}
return changed;
}
async function load(): Promise<void> {
if (loaded) return;
loaded = true;
try {
const stored = await browser.storage.session.get(STORAGE_KEY);
const values = stored[STORAGE_KEY];
if (!Array.isArray(values)) return;
for (const value of values.slice(-MAX_BASELINES)) {
if (validStoredBaseline(value)) baselines.set(value.snapshot.id, value);
}
purge();
} catch {
// The bounded in-memory registry remains available.
}
}
async function save(): Promise<void> {
try {
const retained: StoredAuthorizationBaseline[] = [];
for (const baseline of [...baselines.values()].reverse()) {
const candidate = [baseline, ...retained];
if (new TextEncoder().encode(JSON.stringify(candidate)).byteLength > MAX_BASELINE_STORAGE_BYTES) break;
retained.unshift(baseline);
}
baselines.clear();
for (const baseline of retained) baselines.set(baseline.snapshot.id, baseline);
await browser.storage.session.set({ [STORAGE_KEY]: retained });
} catch {
// The bounded in-memory registry remains available.
}
}
async function authContext(
kind: 'handle' | 'attestation',
id: string,
grantId: string,
): Promise<BrowserAuthContextHandle | BrowserAuthContextAttestation> {
return kind === 'handle'
? getAuthContextHandle(id, grantId)
: getAuthContextAttestation(id, grantId);
}
function sameTarget(
left: BrowserTarget,
right: BrowserTarget,
): boolean {
return left.tabId === right.tabId
&& left.frameId === right.frameId
&& left.documentId === right.documentId;
}
function authorizationDocumentOrigin(url: URL): string {
if (url.protocol === 'ws:') return `http://${url.host}`;
if (url.protocol === 'wss:') return `https://${url.host}`;
return url.origin;
}
export async function captureAuthorizationBaseline(input: {
target: BrowserTarget;
grantId: string;
authContextKind: 'handle' | 'attestation';
authContextId: string;
networkRequestId: string;
comparisonKey: string;
}): Promise<BrowserAuthorizationBaseline> {
await load();
const context = await authContext(input.authContextKind, input.authContextId, input.grantId);
if (!sameTarget(context.target, input.target)) {
throw new ExtensionError('target_denied', '授权基线请求与认证上下文不属于同一页面文档');
}
const exported = await exportNetworkRequest(input.target, input.networkRequestId);
const exportedURL = new URL(exported.url);
if (exportedURL.protocol === 'ws:' || exportedURL.protocol === 'wss:') {
throw new ExtensionError(
'authorization_protocol_unsupported',
'WebSocket 握手不能作为 HTTP 授权基线;请在录制中检查消息帧,当前版本不会把握手误当成可重放业务请求',
);
}
if (authorizationDocumentOrigin(exportedURL) !== context.origin) {
throw new ExtensionError('origin_changed', '授权基线请求与认证上下文来源不一致');
}
if (exported.limitations.length) {
throw new ExtensionError(
'authorization_baseline_incomplete',
`捕获请求不完整:${exported.limitations.join(';')}`,
);
}
const now = Date.now();
const snapshot: BrowserAuthorizationBaseline = {
version: 1,
id: crypto.randomUUID(),
deviceId: context.deviceId,
installationId: context.installationId,
isolationContextId: context.isolationContextId,
cookieStoreId: context.cookieStoreId,
origin: context.origin,
grantId: context.grantId,
target: context.target,
authContextReference: {
kind: input.authContextKind,
id: context.id,
},
networkRequestId: input.networkRequestId,
request: await parseAuthorizationBaselineRequest(
exported.rawRequestBase64,
exported.url,
input.comparisonKey,
),
createdAt: now,
expiresAt: context.expiresAt,
};
if (snapshot.expiresAt <= now) {
throw new ExtensionError('auth_context_stale', '认证上下文已经过期');
}
purge(now, 1);
baselines.set(snapshot.id, {
snapshot,
rawRequestBase64: exported.rawRequestBase64,
requestUrl: exported.url,
isHttps: exported.isHttps,
});
await save();
return snapshot;
}
export async function listAuthorizationBaselineCandidates(input: {
target: BrowserTarget;
grantId: string;
authContextKind: 'handle' | 'attestation';
authContextId: string;
limit: number;
}): Promise<BrowserAuthorizationBaselineCandidate[]> {
const context = await authContext(input.authContextKind, input.authContextId, input.grantId);
if (!sameTarget(context.target, input.target)) {
throw new ExtensionError('target_denied', '网络候选与认证上下文不属于同一页面文档');
}
const records = await listNetworkRequests(input.target, input.limit);
return records.flatMap((record) => {
let parsed: URL;
try {
parsed = new URL(record.url);
} catch {
return [];
}
if (authorizationDocumentOrigin(parsed) !== context.origin) return [];
const shapedPath = normalizeAuthorizationPath(parsed.pathname);
const reasons: string[] = [];
if (record.resourceType === 'websocket' || parsed.protocol === 'ws:' || parsed.protocol === 'wss:') {
reasons.push('WebSocket 当前仅保留握手与消息帧证据,不会进入 HTTP 授权矩阵');
}
if (!record.requestHeadersCaptured) reasons.push('未捕获实际请求头');
if (!['GET', 'HEAD', 'OPTIONS'].includes(record.method.toUpperCase())
&& !record.requestBody) {
reasons.push(record.requestBodyCaptured ? '浏览器未提供请求体' : '未捕获请求体');
}
if (record.requestBody?.truncated) reasons.push('请求体已截断');
if (record.requestBody?.reconstructed) reasons.push('请求体由浏览器字段重建');
if (record.error) reasons.push(`请求失败:${record.error}`);
return [{
id: record.id,
method: record.method,
url: `${parsed.origin}${shapedPath.normalized}`,
path: shapedPath.normalized,
resourceType: record.resourceType,
startedAt: record.startedAt,
completedAt: record.completedAt,
durationMs: record.durationMs,
statusCode: record.statusCode,
error: record.error,
eligible: reasons.length === 0,
reasons,
}];
});
}
async function validatedStoredBaseline(
id: string,
grantId: string,
validateLogicalBinding = true,
): Promise<StoredAuthorizationBaseline> {
await load();
if (purge()) await save();
const baseline = baselines.get(id);
if (!baseline || baseline.snapshot.grantId !== grantId) {
throw new ExtensionError('authorization_baseline_stale', '授权基线不存在、已过期或不属于当前共享会话');
}
try {
const context = await authContext(
baseline.snapshot.authContextReference.kind,
baseline.snapshot.authContextReference.id,
grantId,
);
if (!sameTarget(context.target, baseline.snapshot.target)) {
throw new ExtensionError('authorization_baseline_stale', '授权基线的认证上下文已经变化');
}
} catch (error) {
baselines.delete(id);
await save();
if (error instanceof ExtensionError && error.code === 'authorization_baseline_stale') throw error;
const message = error instanceof Error ? error.message : String(error);
throw new ExtensionError('authorization_baseline_stale', `授权基线实时复核失败:${message}`);
}
if (validateLogicalBinding && baseline.snapshot.logicalRequest) {
try {
await loadAuthorizationLogicalRequestBinding({ baseline: baseline.snapshot });
} catch {
baseline.snapshot = {
...baseline.snapshot,
logicalRequest: undefined,
};
baselines.set(id, baseline);
await save();
}
}
return baseline;
}
export async function getAuthorizationBaseline(
id: string,
grantId: string,
): Promise<BrowserAuthorizationBaseline> {
return (await validatedStoredBaseline(id, grantId)).snapshot;
}
export async function bindAuthorizationBaselineLogicalRequest(input: {
id: string;
grantId: string;
profileId: string;
comparisonKey: string;
}): Promise<BrowserAuthorizationBaseline> {
const baseline = await validatedStoredBaseline(input.id, input.grantId, false);
const profile = await getBrowserTransformProfile(input.profileId);
const draft = await getBrowserTransformReplayDraft(
profile.id,
'request',
baseline.snapshot.origin,
);
if (!draft) {
throw new ExtensionError(
'authorization_logical_missing',
'所选明文网关没有本机请求回放草稿,请先在明文网关中保存并验证回放输入',
);
}
const logicalRequest = await buildAuthorizationLogicalRequestBinding({
baseline: baseline.snapshot,
rawRequestBase64: baseline.rawRequestBase64,
profile,
draft,
comparisonKey: input.comparisonKey,
});
baseline.snapshot = {
...baseline.snapshot,
logicalRequest,
};
baselines.set(baseline.snapshot.id, baseline);
await save();
return baseline.snapshot;
}
function selectedBaselineField(
baseline: BrowserAuthorizationBaseline,
selector: BrowserAuthorizationResourceSelector,
) {
const sourceFields = selector.source === 'logical'
? baseline.logicalRequest?.request.fields
: baseline.request.fields;
const fields = (sourceFields || []).filter(
(field) => field.location === selector.location && field.path === selector.path,
);
if (fields.length !== 1) {
throw new ExtensionError(
fields.length ? 'authorization_selector_ambiguous' : 'authorization_selector_invalid',
fields.length ? '授权资源字段在基线中不唯一' : '授权资源字段不属于该请求基线',
);
}
if (!['string', 'number', 'boolean'].includes(fields[0].valueType)) {
throw new ExtensionError(
'authorization_selector_invalid',
'自动矩阵仅支持字符串、数字或布尔资源值',
);
}
return fields[0];
}
export async function readAuthorizationBaselineResource(input: {
id: string;
grantId: string;
selector: BrowserAuthorizationResourceSelector;
}): Promise<BrowserAuthorizationResourceValue> {
const baseline = await validatedStoredBaseline(input.id, input.grantId);
const selected = selectedBaselineField(baseline.snapshot, input.selector);
if (input.selector.source === 'logical') {
return readAuthorizationLogicalResource({
baseline: baseline.snapshot,
selector: input.selector,
});
}
if (input.selector.location === 'body') {
const value = readStructuredAuthorizationBodyValue(
authorizationRequestToTransformPacket(
baseline.rawRequestBase64,
baseline.snapshot.origin,
),
input.selector.path,
);
const bytes = new TextEncoder().encode(value.text);
if (bytes.byteLength > 8 * 1_024) {
throw new ExtensionError(
'authorization_value_too_large',
'授权 Body 资源值超过 8 KiB 上限',
);
}
return {
version: 1,
baselineId: baseline.snapshot.id,
source: 'wire',
location: 'body',
path: input.selector.path,
valueType: value.valueType,
byteLength: bytes.byteLength,
valueBase64: authorizationBytesToBase64(bytes),
valueFingerprint: selected.valueFingerprint,
};
}
const wireSelector = {
location: input.selector.location,
path: input.selector.path,
};
return extractAuthorizationResourceValue(
baseline.requestUrl,
baseline.rawRequestBase64,
baseline.snapshot.id,
wireSelector,
selected.valueFingerprint,
);
}
export async function compileAuthorizationBaseline(input: {
id: string;
grantId: string;
selector: BrowserAuthorizationResourceSelector;
replacement: BrowserAuthorizationResourceValue;
comparisonKey: string;
}): Promise<BrowserAuthorizationCompiledRequest> {
const baseline = await validatedStoredBaseline(input.id, input.grantId);
if (input.selector.source !== 'wire') {
throw new ExtensionError('authorization_selector_invalid', '直接编译只接受线上报文资源字段');
}
const wireSelector = {
source: 'wire' as const,
location: input.selector.location,
path: input.selector.path,
};
selectedBaselineField(baseline.snapshot, input.selector);
return compileAuthorizationBaselineRequest({
baselineId: baseline.snapshot.id,
rawRequestBase64: baseline.rawRequestBase64,
requestUrl: baseline.requestUrl,
publicUrl: baseline.snapshot.request.url,
selector: wireSelector,
replacement: input.replacement,
comparisonKey: input.comparisonKey,
isHttps: baseline.isHttps,
});
}
export async function compileAuthorizationBaselinePacket(input: {
id: string;
grantId: string;
}): Promise<BrowserAuthorizationBaselinePacket> {
const baseline = await validatedStoredBaseline(input.id, input.grantId);
return {
version: 1,
baselineId: baseline.snapshot.id,
method: baseline.snapshot.request.method,
url: baseline.snapshot.request.url,
isHttps: baseline.isHttps,
rawRequestBase64: baseline.rawRequestBase64,
packetFingerprint: await authorizationPacketFingerprint(baseline.rawRequestBase64),
};
}
async function authorizationTransformFingerprint(input: {
baselineId: string;
profileId: string;
profileUpdatedAt: number;
documentId: string;
isolationContextId: string;
cookieStoreId: string;
dynamicPaths: string[];
logicalBindingFingerprint?: string;
}): Promise<string> {
const digest = await crypto.subtle.digest(
'SHA-256',
new TextEncoder().encode(JSON.stringify(input)),
);
return `sha256:${[...new Uint8Array(digest)]
.map((byte) => byte.toString(16).padStart(2, '0'))
.join('')}`;
}
async function validatedAuthorizationTransform(input: {
id: string;
grantId: string;
profileId: string;
}): Promise<{
baseline: StoredAuthorizationBaseline;
profile: BrowserTransformProfile;
binding: BrowserAuthorizationTransformBinding;
logical?: Awaited<ReturnType<typeof loadAuthorizationLogicalRequestBinding>>;
}> {
const baseline = await validatedStoredBaseline(input.id, input.grantId);
const profile = await getBrowserTransformProfile(input.profileId);
const target = baseline.snapshot.target;
if (
profile.target.tabId !== target.tabId
|| profile.target.frameId !== target.frameId
|| profile.target.documentId !== target.documentId
|| profile.origin !== baseline.snapshot.origin
|| profile.isolationContextId !== baseline.snapshot.isolationContextId
|| profile.cookieStoreId !== baseline.snapshot.cookieStoreId
) {
throw new ExtensionError(
'authorization_transform_target_mismatch',
'明文网关必须绑定授权基线所属的同一身份、Frame 与页面文档',
);
}
const logical = baseline.snapshot.logicalRequest?.profileId === profile.id
? await loadAuthorizationLogicalRequestBinding({
baseline: baseline.snapshot,
profileId: profile.id,
})
: undefined;
const packet = logical
? browserTransformReplayDraftToPacket(logical.draft)
: authorizationRequestToTransformPacket(
baseline.rawRequestBase64,
baseline.snapshot.origin,
);
assertTransformRoute(
profile.match.methods,
profile.match.urlPattern,
packet,
profile.origin,
);
const dynamicPaths = logical
? logical.binding.outputDestinations
: authorizationDynamicTransformDestinations(baseline.snapshot, profile);
const createdAt = Date.now();
const binding: BrowserAuthorizationTransformBinding = {
version: 1,
baselineId: baseline.snapshot.id,
profileId: profile.id,
profileName: profile.name,
isolationContextId: baseline.snapshot.isolationContextId,
cookieStoreId: baseline.snapshot.cookieStoreId,
target,
origin: baseline.snapshot.origin,
dynamicPaths,
bindingFingerprint: await authorizationTransformFingerprint({
baselineId: baseline.snapshot.id,
profileId: profile.id,
profileUpdatedAt: profile.updatedAt,
documentId: target.documentId,
isolationContextId: baseline.snapshot.isolationContextId,
cookieStoreId: baseline.snapshot.cookieStoreId,
dynamicPaths,
logicalBindingFingerprint: logical?.binding.bindingFingerprint,
}),
createdAt,
expiresAt: baseline.snapshot.expiresAt,
};
return { baseline, profile, binding, logical };
}
export async function inspectAuthorizationBaselineTransform(input: {
id: string;
grantId: string;
profileId: string;
}): Promise<BrowserAuthorizationTransformBinding> {
return (await validatedAuthorizationTransform(input)).binding;
}
export async function compileAuthorizationBaselineWithTransform(input: {
id: string;
grantId: string;
selector: BrowserAuthorizationResourceSelector;
replacement: BrowserAuthorizationResourceValue;
comparisonKey: string;
profileId: string;
bindingFingerprint: string;
}): Promise<BrowserAuthorizationCompiledRequest> {
const {
baseline,
profile,
binding,
logical,
} = await validatedAuthorizationTransform(input);
if (binding.bindingFingerprint !== input.bindingFingerprint) {
throw new ExtensionError(
'authorization_transform_changed',
'明文网关或页面文档已变化,请重新编译授权矩阵',
);
}
selectedBaselineField(baseline.snapshot, input.selector);
if (input.selector.source === 'logical') {
if (!logical || input.selector.location !== 'body') {
throw new ExtensionError(
'authorization_logical_missing',
'逻辑资源编译当前要求同一明文网关绑定下的 JSON/Form Body 字段',
);
}
const replacement = await decodeAndVerifyLogicalReplacement({
replacement: input.replacement,
selector: input.selector,
comparisonKey: input.comparisonKey,
});
const logicalPacket = replaceAuthorizationLogicalResource({
packet: browserTransformReplayDraftToPacket(logical.draft),
selector: input.selector,
replacement,
});
const execution = await executeBrowserTransform({
profileId: profile.id,
direction: 'request',
packet: logicalPacket,
});
const compiled: BrowserAuthorizationCompiledRequest = {
version: 1,
baselineId: baseline.snapshot.id,
selector: input.selector,
method: baseline.snapshot.request.method,
url: baseline.snapshot.request.url,
isHttps: baseline.isHttps,
rawRequestBase64: baseline.rawRequestBase64,
resourceValueFingerprint: input.replacement.valueFingerprint,
logicalBindingFingerprint: logical.binding.bindingFingerprint,
packetFingerprint: await authorizationPacketFingerprint(baseline.rawRequestBase64),
};
const compiledWithTransform = await applyAuthorizationTransformExecution({
compiled,
execution,
origin: baseline.snapshot.origin,
allowedDestinations: binding.dynamicPaths,
allowBody: true,
});
assertAuthorizationLogicalPacketStructure(
authorizationRequestToTransformPacket(
compiledWithTransform.rawRequestBase64,
baseline.snapshot.origin,
),
authorizationRequestToTransformPacket(
baseline.rawRequestBase64,
baseline.snapshot.origin,
),
);
return compiledWithTransform;
}
const wireSelector = {
source: 'wire' as const,
location: input.selector.location,
path: input.selector.path,
};
const compiled = await compileAuthorizationBaselineRequest({
baselineId: baseline.snapshot.id,
rawRequestBase64: baseline.rawRequestBase64,
requestUrl: baseline.requestUrl,
publicUrl: baseline.snapshot.request.url,
selector: wireSelector,
replacement: input.replacement,
comparisonKey: input.comparisonKey,
isHttps: baseline.isHttps,
});
const execution = await executeBrowserTransform({
profileId: profile.id,
direction: 'request',
packet: authorizationRequestToTransformPacket(
compiled.rawRequestBase64,
baseline.snapshot.origin,
),
});
return applyAuthorizationTransformExecution({
compiled,
execution,
origin: baseline.snapshot.origin,
allowedDestinations: binding.dynamicPaths,
});
}
@@ -1,35 +0,0 @@
import { describe, expect, it } from 'vitest';
import { ExtensionError } from '@/shared/errors';
import { browserAuthorizationWorkspaceRecovery } from './engine';
describe('browser authorization workspace lifecycle recovery', () => {
it.each([
['expired', '自然过期'],
['evicted', '容量达到上限'],
['engine_instance_changed', '引擎已经重启'],
['not_found', '引擎中不存在'],
['replaced', '新工作区替换'],
] as const)('maps %s to an actionable message', (reason, expected) => {
const error = new ExtensionError(
`authorization_workspace_${reason}`,
'server message',
{
reason,
workspaceId: 'workspace-old',
engineInstanceId: 'engine-current',
replacementWorkspaceId: reason === 'replaced' ? 'workspace-new' : undefined,
},
);
expect(browserAuthorizationWorkspaceRecovery(error)).toMatchObject({
reason,
message: expect.stringContaining(expected),
});
});
it('does not reinterpret unrelated bridge errors', () => {
expect(browserAuthorizationWorkspaceRecovery(
new ExtensionError('bridge_disconnected', 'offline'),
)).toBeUndefined();
});
});
+40 -308
View File
@@ -1,342 +1,74 @@
import { request } from '@/platform/messaging/runtime';
import { ExtensionError } from '@/shared/errors';
import { normalizeBrowserAuthorizationTaskResult } from './protocol';
export type BrowserAuthorizationMode = 'horizontal' | 'vertical';
export type BrowserAuthorizationSide = 'left' | 'right';
export interface BrowserAuthorizationBaselineCandidate {
export interface BrowserAuthorizationTarget {
deviceId: string;
tabId: number;
}
export interface BrowserAuthorizationPair {
left: BrowserAuthorizationTarget;
right: BrowserAuthorizationTarget;
}
export interface BrowserAuthorizationRequest {
id: string;
method: string;
url: string;
path: string;
resourceType: string;
startedAt: number;
completedAt?: number;
durationMs?: number;
statusCode?: number;
error?: string;
eligible: boolean;
reasons: string[];
}
export interface BrowserAuthorizationBaseline {
export interface BrowserAuthorizationSelector {
id: string;
networkRequestId: string;
request: {
method: string;
url: string;
path: string;
contentType: string;
actionFingerprint: string;
};
}
export interface BrowserAuthorizationResourceCandidate {
id: string;
source: 'wire' | 'logical';
location: 'header' | 'path' | 'query' | 'body';
location: 'path' | 'query' | 'form' | 'json';
path: string;
category: string;
confidence: 'high' | 'medium' | 'low';
requiresLogicalBinding: boolean;
reasons: string[];
label: string;
}
export interface BrowserAuthorizationOperationCandidate {
id: string;
export interface BrowserAuthorizationPairInspection {
method: string;
path: string;
eligible: boolean;
route: string;
sideEffect: boolean;
requiresDynamicRebuild: boolean;
authenticationPaths: string[];
dynamicPaths: string[];
reasons: string[];
selectors: BrowserAuthorizationSelector[];
limitations: string[];
blockedReason?: string;
}
export interface BrowserAuthorizationWorkspace {
version: 1;
id: string;
engineInstanceId: string;
mode: BrowserAuthorizationMode;
state: 'ready' | 'conditional' | 'blocked' | 'stale';
left: {
accountLabel?: string;
origin: string;
target: { tabId: number; frameId: number; documentId: string };
authentication: {
status: 'authenticated' | 'unauthenticated' | 'unknown';
cookieCount: number;
storageEntryCount: number;
};
};
right: BrowserAuthorizationWorkspace['left'];
proof: {
level: 'strong' | 'conditional' | 'none';
sameOrigin: boolean;
cookieStoreRelation: 'different' | 'same' | 'unknown';
accountEvidenceRelation: 'different' | 'same' | 'unknown';
requestCredentialRelation: 'different' | 'same' | 'unknown';
refreshCheck: 'passed' | 'failed' | 'not-required';
reasons: string[];
};
baselines: {
left?: BrowserAuthorizationBaseline;
right?: BrowserAuthorizationBaseline;
verification?: BrowserAuthorizationBaseline;
};
baselinePair: {
state: 'waiting' | 'matched' | 'mismatch';
reasons: string[];
resourceCandidates: BrowserAuthorizationResourceCandidate[];
operationCandidates: BrowserAuthorizationOperationCandidate[];
};
plan?: {
id: string;
mode: BrowserAuthorizationMode;
candidateId: string;
state: 'ready' | 'review-required' | 'blocked';
selector: {
source: 'wire' | 'logical' | 'operation';
location: 'header' | 'path' | 'query' | 'body' | 'request';
path: string;
};
cases: Array<{
id: string;
label: string;
authContextSide: 'left' | 'right';
resourceValueSide: 'left' | 'right' | '';
method: string;
path: string;
sideEffect: boolean;
}>;
requestBudget: number;
requiresDynamicRebuild: boolean;
reasons: string[];
};
execution?: {
id: string;
state: 'completed' | 'partial';
verdict: 'confirmed' | 'likely' | 'protected' | 'inconclusive' | 'invalid-controls';
confidence: 'high' | 'medium' | 'low' | 'none';
requestCount: number;
cases: Array<{
id: string;
label: string;
state: 'completed' | 'failed' | 'skipped';
result?: {
method: string;
url: string;
status: number;
statusText: string;
outcome: 'success' | 'denied' | 'redirect' | 'client-error' | 'server-error' | 'opaque';
durationMs: number;
timing: BrowserAuthorizationRequestTiming;
response: {
contentType: string;
contentEncoding?: string;
capturedBytes: number;
analysisBytes?: number;
declaredBytes?: number;
truncated: boolean;
decoded?: boolean;
analysisState?: 'identity' | 'decoded' | 'encoded-unavailable';
analysisRepresentation?: 'json' | 'html' | 'form' | 'text' | 'binary' | 'encoded';
};
};
error?: string;
}>;
evidence: Array<{
direction: string;
path: string;
valueFingerprint: string;
source: string;
}>;
evidenceAvailable: boolean;
reasons: string[];
};
expiresAt: number;
staleReason?: string;
recovery?: {
code: string;
scope: string;
message: string;
automatic: false;
};
}
export type BrowserAuthorizationWorkspaceLifecycleReason =
| 'expired'
| 'evicted'
| 'engine_instance_changed'
| 'not_found'
| 'replaced';
export interface BrowserAuthorizationWorkspaceLifecycleDetails {
reason: BrowserAuthorizationWorkspaceLifecycleReason;
workspaceId: string;
engineInstanceId: string;
expiresAt?: number;
replacementWorkspaceId?: string;
}
function parseWorkspaceLifecycleDetails(input: unknown): BrowserAuthorizationWorkspaceLifecycleDetails | undefined {
if (!input || typeof input !== 'object' || Array.isArray(input)) return undefined;
const value = input as Record<string, unknown>;
if (!['expired', 'evicted', 'engine_instance_changed', 'not_found', 'replaced'].includes(String(value.reason))) return undefined;
if (typeof value.workspaceId !== 'string' || typeof value.engineInstanceId !== 'string') return undefined;
return value as unknown as BrowserAuthorizationWorkspaceLifecycleDetails;
}
export function browserAuthorizationWorkspaceRecovery(error: unknown): {
reason: BrowserAuthorizationWorkspaceLifecycleReason;
message: string;
details?: BrowserAuthorizationWorkspaceLifecycleDetails;
} | undefined {
if (!(error instanceof ExtensionError) || !error.code.startsWith('authorization_workspace_')) return undefined;
const details = parseWorkspaceLifecycleDetails(error.details);
const reason = (details?.reason || error.code.slice('authorization_workspace_'.length)) as BrowserAuthorizationWorkspaceLifecycleReason;
const messages: Record<BrowserAuthorizationWorkspaceLifecycleReason, string> = {
expired: '授权工作区已自然过期。A/B 登录页不会受影响,请点击“新建”重新验证身份。',
evicted: '该工作区因引擎内存容量达到上限而被淘汰。请点击“新建”重新建立,已有页面登录态不会丢失。',
engine_instance_changed: 'Yak 引擎已经重启,旧工作区不能跨进程恢复。请确认引擎在线后点击“新建”。',
not_found: '当前页面缓存的工作区在引擎中不存在。请点击“新建”重新建立身份工作区。',
replaced: details?.replacementWorkspaceId
? '该工作区已被同一组身份的新工作区替换。请刷新页面状态,或点击“新建”重新建立。'
: '该工作区已被更新的身份工作区替换。请点击“新建”重新建立。',
};
if (!(reason in messages)) return undefined;
return { reason, message: messages[reason], details };
}
export interface BrowserAuthorizationRequestTiming {
dnsMs: number;
connectMs: number;
tlsMs: number;
ttfbMs: number;
transferMs: number;
totalMs: number;
}
export interface BrowserAuthorizationEvidenceCase {
export interface BrowserAuthorizationCaseResult {
id: string;
label: string;
authContextSide: 'left' | 'right';
resourceValueSide: 'left' | 'right' | '';
state: 'completed' | 'failed' | 'skipped';
status?: number;
outcome?: string;
timing: BrowserAuthorizationRequestTiming;
requestAvailable: boolean;
responseAvailable: boolean;
response?: {
contentType: string;
contentEncoding?: string;
capturedBytes: number;
analysisBytes?: number;
declaredBytes?: number;
truncated: boolean;
decoded?: boolean;
analysisState?: 'identity' | 'decoded' | 'encoded-unavailable';
analysisRepresentation?: 'json' | 'html' | 'form' | 'text' | 'binary' | 'encoded';
};
status: number;
statusText: string;
outcome: 'success' | 'denied' | 'redirect' | 'client-error' | 'server-error' | 'opaque';
durationMs: number;
contentType?: string;
bodyBytes: number;
matchesTarget?: boolean;
}
export interface BrowserAuthorizationEvidenceComparison {
id: string;
label: string;
leftCaseId: string;
rightCaseId: string;
purpose: 'control' | 'authorization' | 'state-change';
}
export interface BrowserAuthorizationEvidenceBundle {
version: 1;
workspaceId: string;
executionId: string;
mode: BrowserAuthorizationMode;
verdict: NonNullable<BrowserAuthorizationWorkspace['execution']>['verdict'];
confidence: NonNullable<BrowserAuthorizationWorkspace['execution']>['confidence'];
cases: BrowserAuthorizationEvidenceCase[];
comparisons: BrowserAuthorizationEvidenceComparison[];
semantic: NonNullable<BrowserAuthorizationWorkspace['execution']>['evidence'];
representations: string[];
expiresAt: number;
}
export interface BrowserAuthorizationEvidenceDiff {
version: 1;
workspaceId: string;
executionId: string;
leftCaseId: string;
rightCaseId: string;
scope: 'request' | 'response';
view: 'redacted' | 'raw';
representation: 'structured' | 'raw';
equal: boolean;
entries: Array<{
path: string;
kind: 'added' | 'removed' | 'changed';
left?: string;
right?: string;
volatile: boolean;
sensitive: boolean;
semantic: boolean;
}>;
omitted: number;
}
export interface BrowserAuthorizationEvidencePacket {
version: 1;
workspaceId: string;
executionId: string;
caseId: string;
side: 'request' | 'response';
view: 'redacted' | 'raw';
packetBase64: string;
capturedBytes: number;
truncated: boolean;
}
export interface BrowserAuthorizationEvidenceValidation {
version: 1;
workspaceId: string;
executionId: string;
direction: 'a-to-b' | 'b-to-a' | 'low-to-privileged' | 'post-state';
verified: boolean;
evidence: NonNullable<BrowserAuthorizationWorkspace['execution']>['evidence'];
rejectedPaths: string[];
verdict: NonNullable<BrowserAuthorizationWorkspace['execution']>['verdict'];
confidence: NonNullable<BrowserAuthorizationWorkspace['execution']>['confidence'];
verdictChanged: boolean;
reason: string;
export interface BrowserAuthorizationResult {
verdict: 'suspected' | 'possible' | 'protected' | 'inconclusive' | 'invalid-controls';
summary: string;
selector: BrowserAuthorizationSelector;
cases: BrowserAuthorizationCaseResult[];
limitations: string[];
}
export type BrowserAuthorizationTaskSchema =
| 'authorization.workspace.create'
| 'authorization.workspace.inspect'
| 'authorization.baseline.candidates'
| 'authorization.baseline.bind'
| 'authorization.logical.bind'
| 'authorization.plan.create'
| 'authorization.plan.execute'
| 'authorization.evidence.inspect'
| 'authorization.evidence.packet'
| 'authorization.evidence.diff'
| 'authorization.evidence.validate';
| 'authorization.capture.start'
| 'authorization.capture.status'
| 'authorization.capture.stop'
| 'authorization.requests'
| 'authorization.pair.inspect'
| 'authorization.execute';
export async function runBrowserAuthorizationTask<T>(
schema: BrowserAuthorizationTaskSchema,
payload: Record<string, unknown>,
timeoutMs = 30_000,
): Promise<T> {
try {
const result = await request('authorization.engine.task', { schema, payload, timeoutMs });
return normalizeBrowserAuthorizationTaskResult<T>(schema, result);
} catch (error) {
const recovery = browserAuthorizationWorkspaceRecovery(error);
if (!recovery || !(error instanceof ExtensionError)) throw error;
throw new ExtensionError(error.code, recovery.message, recovery.details);
}
return request('authorization.engine.task', { schema, payload, timeoutMs }) as Promise<T>;
}
@@ -1,404 +0,0 @@
import { beforeEach, describe, expect, it, vi } from 'vitest';
import type {
BrowserAuthorizationBaseline,
BrowserTransformExecution,
BrowserTransformProfile,
} from '@/types/models';
import type { BrowserTransformReplayDraft } from '@/features/browser-transform/replay-draft';
import {
assertAuthorizationLogicalProtocol,
assertAuthorizationLogicalPacketStructure,
authorizationTransformOutputDestinations,
buildAuthorizationLogicalRequestBinding,
replaceAuthorizationLogicalResource,
} from './logical-binding';
const executeBrowserTransform = vi.fn();
vi.mock('wxt/browser', () => {
const event = { addListener: vi.fn() };
return {
browser: {
tabs: { onRemoved: event, onCreated: event },
webNavigation: {
onBeforeNavigate: event,
onCommitted: event,
onDOMContentLoaded: event,
onCompleted: event,
onHistoryStateUpdated: event,
onReferenceFragmentUpdated: event,
onErrorOccurred: event,
},
},
};
});
vi.mock('@/features/browser-transform/service', () => ({
executeBrowserTransform: (...args: unknown[]) => executeBrowserTransform(...args),
getBrowserTransformProfile: vi.fn(),
}));
function base64(value: string): string {
const bytes = new TextEncoder().encode(value);
return btoa(String.fromCharCode(...bytes));
}
function comparisonKey(): string {
return btoa(String.fromCharCode(...new Uint8Array(32).fill(23)))
.replace(/\+/g, '-')
.replace(/\//g, '_')
.replace(/=+$/, '');
}
function profile(outputs = ['body.encryptedData', 'header.Content-Type']): BrowserTransformProfile {
return {
id: 'profile-left',
name: '登录请求加密',
enabled: true,
target: { tabId: 11, frameId: 0, documentId: 'document-left' },
isolationContextId: 'browser-profile:store-left',
cookieStoreId: 'store-left',
origin: 'https://example.test',
match: { methods: ['POST'], urlPattern: '*/api/login' },
request: {
enabled: true,
nodes: outputs.map((destination, index) => ({
id: `output-${index}`,
name: destination,
kind: 'output.write' as const,
destination,
source: { nodeId: 'callable' },
encoding: 'text' as const,
})),
},
response: { enabled: false, nodes: [] },
failMode: 'closed',
maxConcurrency: 1,
createdAt: 1,
updatedAt: 2,
};
}
function baseline(): BrowserAuthorizationBaseline {
return {
version: 1,
id: 'baseline-left',
deviceId: 'device-left',
installationId: 'installation-left',
isolationContextId: 'browser-profile:store-left',
cookieStoreId: 'store-left',
origin: 'https://example.test',
grantId: 'grant-left',
target: { tabId: 11, frameId: 0, documentId: 'document-left' },
authContextReference: { kind: 'handle', id: 'auth-left' },
networkRequestId: 'request-left',
request: {
method: 'POST',
url: 'https://example.test/api/login',
path: '/api/login',
contentType: 'application/x-www-form-urlencoded',
actionFingerprint: `sha256:${'a'.repeat(64)}`,
headerNames: ['Host', 'Content-Type', 'Cookie'],
fields: [{
location: 'body',
path: 'body.encryptedData',
valueType: 'string',
byteLength: 32,
valueFingerprint: `workspace-hmac-sha256:${'b'.repeat(64)}`,
category: 'unknown',
}],
},
createdAt: 1,
expiresAt: Date.now() + 60_000,
};
}
function draft(): BrowserTransformReplayDraft {
return {
version: 1,
profileId: 'profile-left',
direction: 'request',
origin: 'https://example.test',
method: 'POST',
url: 'https://example.test/api/login',
headers: '{"Content-Type":"application/json"}',
body: '{"username":"alice","orderId":"order-a"}',
updatedAt: 3,
};
}
describe('authorization logical plaintext binding', () => {
beforeEach(() => {
executeBrowserTransform.mockReset();
});
it('rejects a logical replay that changes the observed GraphQL operation', () => {
const observed = baseline().request;
observed.protocol = 'graphql';
observed.operationFingerprint = `sha256:${'1'.repeat(64)}`;
observed.operationNames = ['Order'];
const logical = {
...observed,
operationFingerprint: `sha256:${'2'.repeat(64)}`,
operationNames: ['CancelOrder'],
};
expect(() => assertAuthorizationLogicalProtocol(observed, logical)).toThrow(
'GraphQL operation 与线上基线不一致',
);
});
it('allows a logical GraphQL envelope when the encrypted wire baseline has no protocol metadata', () => {
const observed = baseline().request;
const logical = {
...observed,
protocol: 'graphql' as const,
operationFingerprint: `sha256:${'1'.repeat(64)}`,
operationNames: ['Order'],
};
expect(() => assertAuthorizationLogicalProtocol(observed, logical)).not.toThrow();
});
it('binds private plaintext field metadata only after the generated wire shape matches', async () => {
executeBrowserTransform.mockResolvedValue({
profileId: 'profile-left',
direction: 'request',
url: 'https://example.test/api/login',
bodyBase64: base64('encryptedData=ciphertext'),
setHeaders: [{ name: 'Content-Type', value: 'application/x-www-form-urlencoded' }],
removeHeaders: [],
logicalInput: {},
logicalOutput: {},
nodeDurations: [],
nodeTrace: [],
fieldChanges: [],
durationMs: 1,
} satisfies BrowserTransformExecution);
const raw = base64([
'POST /api/login HTTP/1.1',
'Host: example.test',
'Content-Type: application/x-www-form-urlencoded',
'Cookie: session=identity-a',
'',
'encryptedData=observed-ciphertext',
].join('\r\n'));
const binding = await buildAuthorizationLogicalRequestBinding({
baseline: baseline(),
rawRequestBase64: raw,
profile: profile(),
draft: draft(),
comparisonKey: comparisonKey(),
});
expect(binding.request.fields).toEqual(expect.arrayContaining([
expect.objectContaining({
location: 'body',
path: 'body.orderId',
valueType: 'string',
category: 'resource',
}),
]));
expect(binding.outputDestinations).toEqual(['body.encryptedData', 'header.content-type']);
expect(binding.bindingFingerprint).toMatch(/^sha256:[a-f0-9]{64}$/);
expect(JSON.stringify(binding)).not.toContain('order-a');
expect(JSON.stringify(binding)).not.toContain('alice');
});
it('keeps a multi-output AES plus RSA envelope tied to one logical business object', async () => {
executeBrowserTransform.mockResolvedValue({
profileId: 'profile-left',
direction: 'request',
url: 'https://example.test/api/login',
bodyBase64: base64([
'encryptedData=aes-ciphertext',
'encryptedKey=rsa-wrapped-key',
'encryptedIv=rsa-wrapped-iv',
].join('&')),
setHeaders: [{ name: 'Content-Type', value: 'application/x-www-form-urlencoded' }],
removeHeaders: [],
logicalInput: {},
logicalOutput: {},
nodeDurations: [],
nodeTrace: [],
fieldChanges: [],
durationMs: 1,
} satisfies BrowserTransformExecution);
const raw = base64([
'POST /api/login HTTP/1.1',
'Host: example.test',
'Content-Type: application/x-www-form-urlencoded',
'Cookie: session=identity-a',
'',
[
'encryptedData=observed-aes-ciphertext',
'encryptedKey=observed-rsa-key',
'encryptedIv=observed-rsa-iv',
].join('&'),
].join('\r\n'));
const binding = await buildAuthorizationLogicalRequestBinding({
baseline: baseline(),
rawRequestBase64: raw,
profile: profile([
'body.encryptedData',
'body.encryptedKey',
'body.encryptedIv',
'header.Content-Type',
]),
draft: draft(),
comparisonKey: comparisonKey(),
});
expect(binding.outputDestinations).toEqual([
'body.encryptedData',
'body.encryptedIv',
'body.encryptedKey',
'header.content-type',
]);
expect(binding.request.fields).toEqual(expect.arrayContaining([
expect.objectContaining({ path: 'body.orderId', category: 'resource' }),
expect.objectContaining({ path: 'body.username' }),
]));
expect(binding.validation.proofLevel).toBe('structure');
});
it('rejects a gateway whose generated serialization does not match the captured request', async () => {
executeBrowserTransform.mockResolvedValue({
profileId: 'profile-left',
direction: 'request',
url: 'https://example.test/api/login',
bodyBase64: base64('{"encryptedData":"ciphertext"}'),
setHeaders: [{ name: 'Content-Type', value: 'application/json' }],
removeHeaders: [],
logicalInput: {},
logicalOutput: {},
nodeDurations: [],
nodeTrace: [],
fieldChanges: [],
durationMs: 1,
} satisfies BrowserTransformExecution);
const raw = base64([
'POST /api/login HTTP/1.1',
'Host: example.test',
'Content-Type: application/x-www-form-urlencoded',
'',
'encryptedData=observed-ciphertext',
].join('\r\n'));
await expect(buildAuthorizationLogicalRequestBinding({
baseline: baseline(),
rawRequestBase64: raw,
profile: profile(),
draft: draft(),
comparisonKey: comparisonKey(),
})).rejects.toThrow('结构不一致');
});
it('rejects compressed request bodies because their logical structure cannot be proven', async () => {
executeBrowserTransform.mockResolvedValue({
profileId: 'profile-left',
direction: 'request',
url: 'https://example.test/api/login',
bodyBase64: base64('encryptedData=ciphertext'),
setHeaders: [{ name: 'Content-Type', value: 'application/x-www-form-urlencoded' }],
removeHeaders: [],
logicalInput: {},
logicalOutput: {},
nodeDurations: [],
nodeTrace: [],
fieldChanges: [],
durationMs: 1,
} satisfies BrowserTransformExecution);
const raw = base64([
'POST /api/login HTTP/1.1',
'Host: example.test',
'Content-Type: application/x-www-form-urlencoded',
'Content-Encoding: gzip',
'',
'encryptedData=observed-ciphertext',
].join('\r\n'));
await expect(buildAuthorizationLogicalRequestBinding({
baseline: baseline(),
rawRequestBase64: raw,
profile: profile(),
draft: draft(),
comparisonKey: comparisonKey(),
})).rejects.toThrow('压缩或编码后的请求 Body');
});
it('rejects a conditionally changed output envelope during later matrix compilation', () => {
const observed = {
method: 'POST',
url: 'https://example.test/api/login',
headers: [{ name: 'Content-Type', value: 'application/x-www-form-urlencoded' }],
bodyBase64: base64('encryptedData=observed-ciphertext'),
};
const generated = {
...observed,
bodyBase64: base64('encryptedData=generated-ciphertext&unexpected=side-channel'),
};
expect(() => assertAuthorizationLogicalPacketStructure(
generated,
observed,
)).toThrow('Body 字段与类型结构');
});
it('replaces one explicit JSON plaintext field without touching its siblings', () => {
const packet = {
method: 'POST',
url: 'https://example.test/api/orders',
headers: [{ name: 'Content-Type', value: 'application/json' }],
bodyBase64: base64('{"orderId":"order-a","note":"keep"}'),
};
const replaced = replaceAuthorizationLogicalResource({
packet,
selector: { source: 'logical', location: 'body', path: 'body.orderId' },
replacement: 'order-b',
});
expect(JSON.parse(new TextDecoder().decode(
Uint8Array.from(atob(replaced.bodyBase64), (character) => character.charCodeAt(0)),
))).toEqual({ orderId: 'order-b', note: 'keep' });
});
it('preserves the primitive type of a numeric logical resource', () => {
const packet = {
method: 'POST',
url: 'https://example.test/graphql',
headers: [{ name: 'Content-Type', value: 'application/json' }],
bodyBase64: base64('{"variables":{"orderId":42},"query":"query Order { order { id } }"}'),
};
const replaced = replaceAuthorizationLogicalResource({
packet,
selector: {
source: 'logical',
location: 'body',
path: 'body.variables.orderId',
},
replacement: 84,
});
expect(JSON.parse(new TextDecoder().decode(
Uint8Array.from(atob(replaced.bodyBase64), (character) => character.charCodeAt(0)),
)).variables.orderId).toBe(84);
expect(() => replaceAuthorizationLogicalResource({
packet,
selector: {
source: 'logical',
location: 'body',
path: 'body.variables.orderId',
},
replacement: '84',
})).toThrow('不能改变字段类型');
});
it('refuses profiles that attempt to synthesize authentication headers', () => {
expect(() => authorizationTransformOutputDestinations(
profile(['header.Authorization']),
)).toThrow('认证 Header');
});
});
@@ -1,621 +0,0 @@
import type {
BrowserAuthorizationBaseline,
BrowserAuthorizationLogicalRequestBinding,
BrowserAuthorizationResourceSelector,
BrowserAuthorizationResourceValue,
BrowserTransformExecution,
BrowserTransformPacket,
BrowserTransformProfile,
} from '@/types/models';
import {
applyTransformExecution,
compareBrowserPackets,
} from '@/features/browser-analysis/service';
import {
browserTransformReplayDraftToPacket,
getBrowserTransformReplayDraft,
type BrowserTransformReplayDraft,
} from '@/features/browser-transform/replay-draft';
import {
executeBrowserTransform,
getBrowserTransformProfile,
} from '@/features/browser-transform/service';
import { ExtensionError } from '@/shared/errors';
import {
fingerprintAuthorizationComparisonValue,
parseAuthorizationBaselineRequest,
} from './baseline-metadata';
import {
authorizationRequestToTransformPacket,
} from './baseline-execution';
import {
readStructuredAuthorizationBodyValue,
replaceStructuredAuthorizationBodyValue,
type StructuredAuthorizationPrimitive,
} from './structured-body';
const MAX_LOGICAL_RESOURCE_BYTES = 8 * 1_024;
const MAX_TRANSFORM_BODY_BYTES = 2 * 1_024 * 1_024;
const FORBIDDEN_OUTPUT_HEADERS = new Set([
'authorization',
'cookie',
'host',
'proxy-authorization',
]);
function bytesToBase64(bytes: Uint8Array): string {
let binary = '';
const chunkSize = 0x8000;
for (let offset = 0; offset < bytes.length; offset += chunkSize) {
binary += String.fromCharCode(...bytes.subarray(offset, offset + chunkSize));
}
return btoa(binary);
}
function base64ToBytes(value: string): Uint8Array {
let binary: string;
try {
binary = atob(value);
} catch {
throw new ExtensionError('authorization_value_invalid', '逻辑请求 Body 不是有效的 Base64');
}
return Uint8Array.from(binary, (character) => character.charCodeAt(0));
}
async function sha256(value: string | Uint8Array): Promise<string> {
const bytes = typeof value === 'string' ? new TextEncoder().encode(value) : value;
const digest = await crypto.subtle.digest('SHA-256', Uint8Array.from(bytes).buffer);
return `sha256:${[...new Uint8Array(digest)]
.map((byte) => byte.toString(16).padStart(2, '0'))
.join('')}`;
}
function normalizedDestination(destination: string): string {
const trimmed = destination.trim();
if (trimmed.toLowerCase().startsWith('header.')) {
return `header.${trimmed.slice(7).trim().toLowerCase()}`;
}
return trimmed;
}
export function authorizationTransformOutputDestinations(
profile: BrowserTransformProfile,
): string[] {
if (!profile.enabled || !profile.request.enabled) {
throw new ExtensionError('authorization_transform_unavailable', '所选明文网关未启用请求转换');
}
if (profile.recovery && profile.recovery.state !== 'ready') {
throw new ExtensionError('authorization_transform_stale', '所选明文网关正在等待文档恢复或重新验证');
}
const destinations = [...new Set(profile.request.nodes.flatMap((node) => {
if (node.kind !== 'output.write') return [];
const destination = normalizedDestination(node.destination);
if (destination.toLowerCase().startsWith('header.')) {
const name = destination.slice(7).toLowerCase();
if (FORBIDDEN_OUTPUT_HEADERS.has(name)) {
throw new ExtensionError(
'authorization_transform_invalid',
`授权明文网关不能生成或覆盖认证 Header: ${name}`,
);
}
}
return [destination];
}))].sort();
if (!destinations.length || destinations.length > 32) {
throw new ExtensionError(
'authorization_transform_invalid',
'授权明文网关必须声明 1 到 32 个确定性请求输出',
);
}
return destinations;
}
export function authorizationTransformPacketToRawRequest(
packet: BrowserTransformPacket,
): string {
const method = packet.method?.trim().toUpperCase() || '';
if (!/^[A-Z]{1,16}$/.test(method)) {
throw new ExtensionError('authorization_logical_invalid', '逻辑请求缺少有效的 HTTP 方法');
}
let url: URL;
try {
url = new URL(packet.url);
} catch {
throw new ExtensionError('authorization_logical_invalid', '逻辑请求 URL 无效');
}
if (!['http:', 'https:'].includes(url.protocol) || url.hash) {
throw new ExtensionError('authorization_logical_invalid', '逻辑请求必须使用无 fragment 的 HTTP(S) URL');
}
const headers = packet.headers.filter((header) => header.name.toLowerCase() !== 'host');
for (const header of headers) {
if (
!header.name
|| !/^[!#$%&'*+\-.^_`|~0-9A-Za-z]+$/.test(header.name)
|| /[\r\n]/.test(header.value)
) {
throw new ExtensionError('authorization_logical_invalid', `逻辑请求包含无效 Header: ${header.name}`);
}
}
const body = base64ToBytes(packet.bodyBase64);
if (body.byteLength > MAX_TRANSFORM_BODY_BYTES) {
throw new ExtensionError('authorization_logical_invalid', '逻辑请求 Body 超过 2 MiB 上限');
}
const head = new TextEncoder().encode([
`${method} ${url.pathname || '/'}${url.search} HTTP/1.1`,
`Host: ${url.host}`,
...headers.map((header) => `${header.name}: ${header.value}`),
'',
'',
].join('\r\n'));
const raw = new Uint8Array(head.byteLength + body.byteLength);
raw.set(head);
raw.set(body, head.byteLength);
return bytesToBase64(raw);
}
function sameTarget(
baseline: BrowserAuthorizationBaseline,
profile: BrowserTransformProfile,
): boolean {
return profile.target.tabId === baseline.target.tabId
&& profile.target.frameId === baseline.target.frameId
&& profile.target.documentId === baseline.target.documentId
&& profile.origin === baseline.origin
&& profile.isolationContextId === baseline.isolationContextId
&& profile.cookieStoreId === baseline.cookieStoreId;
}
function assertLogicalProfileIdentity(
baseline: BrowserAuthorizationBaseline,
profile: BrowserTransformProfile,
): void {
if (!sameTarget(baseline, profile)) {
throw new ExtensionError(
'authorization_transform_target_mismatch',
'逻辑明文必须使用授权基线所属同一身份、Frame 与页面文档的明文网关',
);
}
}
function assertGeneratedRoute(
baseline: BrowserAuthorizationBaseline,
execution: BrowserTransformExecution,
): void {
let generated: URL;
try {
generated = new URL(execution.url);
} catch {
throw new ExtensionError('authorization_transform_invalid', '明文网关生成了无效 URL');
}
// The structural packet comparison below performs the exact route check.
// This early guard blocks obvious origin/fragment escapes before comparison.
if (generated.origin !== baseline.origin || generated.hash) {
throw new ExtensionError('authorization_origin_changed', '明文网关不能改变授权请求来源或 fragment');
}
}
function assertIdentityContentEncoding(
packet: BrowserTransformPacket,
label: string,
): void {
const encodings = packet.headers
.filter((header) => header.name.toLowerCase() === 'content-encoding')
.flatMap((header) => header.value.split(','))
.map((encoding) => encoding.trim().toLowerCase())
.filter(Boolean);
if (encodings.some((encoding) => encoding !== 'identity')) {
throw new ExtensionError(
'authorization_content_encoding_unsupported',
`${label}使用了压缩或编码后的请求 Body,当前不能建立可验证的逻辑明文绑定`,
);
}
}
export function assertAuthorizationLogicalPacketStructure(
generated: BrowserTransformPacket,
observed: BrowserTransformPacket,
): { summary: string; warnings: string[] } {
assertIdentityContentEncoding(generated, '明文网关生成报文');
assertIdentityContentEncoding(observed, '线上基线');
const comparison = compareBrowserPackets(generated, observed, 'structure');
if (!comparison.equivalent) {
const failures = comparison.checks
.filter((check) => check.status === 'fail')
.map((check) => check.label.replace(/一致$/, ''))
.join('、');
throw new ExtensionError(
'authorization_logical_mismatch',
`明文网关生成报文与线上基线结构不一致:${failures || comparison.summary}`,
);
}
return {
summary: comparison.summary,
warnings: comparison.checks
.filter((check) => check.status === 'warning')
.map((check) => check.label),
};
}
export function assertAuthorizationLogicalProtocol(
observed: BrowserAuthorizationBaseline['request'],
logical: BrowserAuthorizationBaseline['request'],
): void {
if (
observed.protocol
&& (
logical.protocol !== observed.protocol
|| logical.operationFingerprint !== observed.operationFingerprint
)
) {
throw new ExtensionError(
'authorization_logical_mismatch',
'明文网关回放的 GraphQL operation 与线上基线不一致',
);
}
}
export async function buildAuthorizationLogicalRequestBinding(input: {
baseline: BrowserAuthorizationBaseline;
rawRequestBase64: string;
profile: BrowserTransformProfile;
draft: BrowserTransformReplayDraft;
comparisonKey: string;
}): Promise<BrowserAuthorizationLogicalRequestBinding> {
assertLogicalProfileIdentity(input.baseline, input.profile);
if (
input.draft.profileId !== input.profile.id
|| input.draft.direction !== 'request'
|| input.draft.origin !== input.baseline.origin
) {
throw new ExtensionError(
'authorization_logical_invalid',
'所选明文网关没有与当前身份来源匹配的本机请求回放草稿',
);
}
const logicalPacket = browserTransformReplayDraftToPacket(input.draft);
const execution = await executeBrowserTransform({
profileId: input.profile.id,
direction: 'request',
packet: logicalPacket,
});
assertGeneratedRoute(input.baseline, execution);
const generated = applyTransformExecution(logicalPacket, execution);
const observed = authorizationRequestToTransformPacket(
input.rawRequestBase64,
input.baseline.origin,
);
const validation = assertAuthorizationLogicalPacketStructure(generated, observed);
const request = await parseAuthorizationBaselineRequest(
authorizationTransformPacketToRawRequest(logicalPacket),
logicalPacket.url,
input.comparisonKey,
);
assertAuthorizationLogicalProtocol(input.baseline.request, request);
const outputDestinations = authorizationTransformOutputDestinations(input.profile);
const createdAt = Date.now();
const bindingFingerprint = await sha256(JSON.stringify({
version: 1,
baselineId: input.baseline.id,
profileId: input.profile.id,
profileUpdatedAt: input.profile.updatedAt,
replayUpdatedAt: input.draft.updatedAt,
isolationContextId: input.baseline.isolationContextId,
cookieStoreId: input.baseline.cookieStoreId,
documentId: input.baseline.target.documentId,
actionFingerprint: request.actionFingerprint,
fields: request.fields.map((field) => ({
location: field.location,
path: field.path,
valueType: field.valueType,
valueFingerprint: field.valueFingerprint,
})),
outputDestinations,
warnings: validation.warnings,
}));
return {
version: 1,
source: 'local-replay-draft',
baselineId: input.baseline.id,
profileId: input.profile.id,
profileName: input.profile.name,
isolationContextId: input.baseline.isolationContextId,
cookieStoreId: input.baseline.cookieStoreId,
target: input.baseline.target,
origin: input.baseline.origin,
request,
outputDestinations,
validation: {
proofLevel: 'structure',
summary: validation.summary,
warnings: validation.warnings,
},
bindingFingerprint,
profileUpdatedAt: input.profile.updatedAt,
replayUpdatedAt: input.draft.updatedAt,
createdAt,
expiresAt: input.baseline.expiresAt,
};
}
export async function loadAuthorizationLogicalRequestBinding(input: {
baseline: BrowserAuthorizationBaseline;
profileId?: string;
}): Promise<{
binding: BrowserAuthorizationLogicalRequestBinding;
profile: BrowserTransformProfile;
draft: BrowserTransformReplayDraft;
}> {
const binding = input.baseline.logicalRequest;
if (!binding || (input.profileId && binding.profileId !== input.profileId)) {
throw new ExtensionError('authorization_logical_missing', '授权基线尚未绑定逻辑明文请求');
}
const profile = await getBrowserTransformProfile(binding.profileId);
assertLogicalProfileIdentity(input.baseline, profile);
const draft = await getBrowserTransformReplayDraft(profile.id, 'request', input.baseline.origin);
if (
!draft
|| profile.updatedAt !== binding.profileUpdatedAt
|| draft.updatedAt !== binding.replayUpdatedAt
|| binding.baselineId !== input.baseline.id
|| binding.bindingFingerprint.length !== 71
) {
throw new ExtensionError(
'authorization_logical_changed',
'明文网关或本机回放草稿已变化,请重新绑定逻辑明文',
);
}
return { binding, profile, draft };
}
function indexedName(path: string, prefix: 'header' | 'query' | 'body'): {
name: string;
index?: number;
} {
if (!path.startsWith(`${prefix}.`)) {
throw new ExtensionError('authorization_selector_invalid', '逻辑资源字段路径与位置不匹配');
}
const raw = path.slice(prefix.length + 1);
const matched = raw.match(/^(.*)\[(\d+)]$/);
const name = matched ? matched[1] : raw;
const index = matched ? Number(matched[2]) : undefined;
if (!name || (index !== undefined && !Number.isSafeInteger(index))) {
throw new ExtensionError('authorization_selector_invalid', '逻辑资源字段路径无效');
}
return { name, index };
}
function selectedOccurrence(
entries: Array<[string, string]>,
name: string,
index?: number,
): { entryIndex: number; value: string } {
const matches = entries.flatMap(([key, value], entryIndex) => (
key === name ? [{ entryIndex, value }] : []
));
if (index === undefined && matches.length !== 1) {
throw new ExtensionError('authorization_selector_ambiguous', '逻辑资源字段存在多个同名值,必须选择带序号的字段');
}
const selected = matches[index ?? 0];
if (!selected) {
throw new ExtensionError('authorization_selector_invalid', '逻辑资源字段不存在');
}
return selected;
}
function logicalResourceText(
packet: BrowserTransformPacket,
selector: BrowserAuthorizationResourceSelector,
): string {
if (selector.source !== 'logical') {
throw new ExtensionError('authorization_selector_invalid', '逻辑资源读取器只接受 logical 选择器');
}
if (selector.location === 'body') {
throw new ExtensionError(
'authorization_selector_invalid',
'逻辑 Body 资源必须通过结构化读取器读取',
);
}
if (selector.location === 'query') {
const selected = indexedName(selector.path, 'query');
return selectedOccurrence(
[...new URL(packet.url).searchParams],
selected.name,
selected.index,
).value;
}
if (selector.location === 'header') {
const selected = indexedName(selector.path, 'header');
return selectedOccurrence(
packet.headers.map((header) => [header.name.toLowerCase(), header.value]),
selected.name.toLowerCase(),
selected.index,
).value;
}
const matched = selector.path.match(/^path\.segment\[(\d+)]$/);
const index = matched ? Number(matched[1]) : -1;
const segment = new URL(packet.url).pathname.split('/').filter(Boolean)[index];
if (segment === undefined) {
throw new ExtensionError('authorization_selector_invalid', '逻辑路径资源字段不存在');
}
try {
return decodeURIComponent(segment);
} catch {
return segment;
}
}
export async function readAuthorizationLogicalResource(input: {
baseline: BrowserAuthorizationBaseline;
selector: BrowserAuthorizationResourceSelector;
}): Promise<BrowserAuthorizationResourceValue> {
const { binding, draft } = await loadAuthorizationLogicalRequestBinding({
baseline: input.baseline,
});
const packet = browserTransformReplayDraftToPacket(draft);
const value = (() => {
if (input.selector.location === 'body') {
return readStructuredAuthorizationBodyValue(packet, input.selector.path);
}
const text = logicalResourceText(packet, input.selector);
return { value: text, valueType: 'string' as const, text };
})();
const bytes = new TextEncoder().encode(value.text);
if (bytes.byteLength > MAX_LOGICAL_RESOURCE_BYTES) {
throw new ExtensionError('authorization_value_too_large', '逻辑授权资源值超过 8 KiB 上限');
}
const field = binding.request.fields.filter((candidate) => (
candidate.location === input.selector.location
&& candidate.path === input.selector.path
));
if (
field.length !== 1
|| !['string', 'number', 'boolean'].includes(field[0].valueType)
|| field[0].valueType !== value.valueType
) {
throw new ExtensionError('authorization_selector_invalid', '逻辑资源字段不属于当前明文绑定');
}
return {
version: 1,
baselineId: input.baseline.id,
source: 'logical',
location: input.selector.location,
path: input.selector.path,
valueType: value.valueType,
byteLength: bytes.byteLength,
valueBase64: bytesToBase64(bytes),
valueFingerprint: field[0].valueFingerprint,
logicalBindingFingerprint: binding.bindingFingerprint,
};
}
export function replaceAuthorizationLogicalResource(input: {
packet: BrowserTransformPacket;
selector: BrowserAuthorizationResourceSelector;
replacement: StructuredAuthorizationPrimitive;
}): BrowserTransformPacket {
const { packet, selector, replacement } = input;
if (selector.source !== 'logical') {
throw new ExtensionError('authorization_selector_invalid', '逻辑资源替换器只接受 logical 选择器');
}
if (selector.location === 'body') {
return replaceStructuredAuthorizationBodyValue({
packet,
path: selector.path,
replacement,
});
}
if (selector.location === 'query') {
if (typeof replacement !== 'string') {
throw new ExtensionError('authorization_selector_invalid', '逻辑 Query 资源替换只接受字符串');
}
const selected = indexedName(selector.path, 'query');
const url = new URL(packet.url);
const entries = [...url.searchParams];
const occurrence = selectedOccurrence(entries, selected.name, selected.index);
entries[occurrence.entryIndex][1] = replacement;
url.search = '';
entries.forEach(([name, value]) => url.searchParams.append(name, value));
return { ...packet, url: url.toString() };
}
if (selector.location === 'header') {
if (typeof replacement !== 'string') {
throw new ExtensionError('authorization_selector_invalid', '逻辑 Header 资源替换只接受字符串');
}
const selected = indexedName(selector.path, 'header');
const matching = packet.headers.flatMap((header, index) => (
header.name.toLowerCase() === selected.name.toLowerCase() ? [index] : []
));
if (selected.index === undefined && matching.length !== 1) {
throw new ExtensionError('authorization_selector_ambiguous', '逻辑 Header 存在多个同名值');
}
const headerIndex = matching[selected.index ?? 0];
if (headerIndex === undefined) {
throw new ExtensionError('authorization_selector_invalid', '逻辑 Header 资源字段不存在');
}
const headers = packet.headers.slice();
headers[headerIndex] = { ...headers[headerIndex], value: replacement };
return { ...packet, headers };
}
const matched = selector.path.match(/^path\.segment\[(\d+)]$/);
if (typeof replacement !== 'string') {
throw new ExtensionError('authorization_selector_invalid', '逻辑 Path 资源替换只接受字符串');
}
const index = matched ? Number(matched[1]) : -1;
const url = new URL(packet.url);
let current = -1;
const segments = url.pathname.split('/').map((segment) => {
if (!segment) return segment;
current += 1;
return current === index ? encodeURIComponent(replacement) : segment;
});
if (current < index || index < 0) {
throw new ExtensionError('authorization_selector_invalid', '逻辑路径资源字段不存在');
}
url.pathname = segments.join('/');
return { ...packet, url: url.toString() };
}
export async function decodeAndVerifyLogicalReplacement(input: {
replacement: BrowserAuthorizationResourceValue;
selector: BrowserAuthorizationResourceSelector;
comparisonKey: string;
}): Promise<StructuredAuthorizationPrimitive> {
if (
input.replacement.source !== 'logical'
|| input.replacement.location !== input.selector.location
|| input.replacement.path !== input.selector.path
|| !['string', 'number', 'boolean'].includes(input.replacement.valueType)
) {
throw new ExtensionError('authorization_value_invalid', '逻辑授权资源值与选择器不匹配');
}
const bytes = base64ToBytes(input.replacement.valueBase64);
if (
bytes.byteLength !== input.replacement.byteLength
|| bytes.byteLength > MAX_LOGICAL_RESOURCE_BYTES
) {
throw new ExtensionError('authorization_value_invalid', '逻辑授权资源值长度无效');
}
let text: string;
try {
text = new TextDecoder('utf-8', { fatal: true }).decode(bytes);
} catch {
throw new ExtensionError('authorization_value_invalid', '逻辑授权资源值不是有效的 UTF-8');
}
let value: StructuredAuthorizationPrimitive;
if (input.replacement.valueType === 'string') {
value = text;
} else if (input.replacement.valueType === 'number') {
try {
const parsed: unknown = JSON.parse(text);
if (
typeof parsed !== 'number'
|| !Number.isFinite(parsed)
|| JSON.stringify(parsed) !== text
) {
throw new Error('not canonical');
}
value = parsed;
} catch {
throw new ExtensionError(
'authorization_value_invalid',
'逻辑授权数字资源值不是规范 JSON 数字',
);
}
} else if (text === 'true' || text === 'false') {
value = text === 'true';
} else {
throw new ExtensionError(
'authorization_value_invalid',
'逻辑授权布尔资源值必须是 true 或 false',
);
}
const fingerprint = await fingerprintAuthorizationComparisonValue(input.comparisonKey, text);
if (fingerprint !== input.replacement.valueFingerprint) {
throw new ExtensionError('authorization_value_invalid', '逻辑授权资源值指纹校验失败');
}
return value;
}
export async function authorizationPacketFingerprint(rawRequestBase64: string): Promise<string> {
return sha256(base64ToBytes(rawRequestBase64));
}
@@ -1,83 +0,0 @@
import { describe, expect, it } from 'vitest';
import { ExtensionError } from '@/shared/errors';
import { normalizeBrowserAuthorizationTaskResult } from './protocol';
function context(side: 'left' | 'right') {
return {
side,
target: {tabId: side === 'left' ? 1 : 2, frameId: 0, documentId: `document-${side}`},
authentication: {
status: 'authenticated',
cookieCount: 1,
storageEntryCount: 0,
authCookieNames: null,
authStorageKeys: null,
},
};
}
function workspace(extra: Record<string, unknown> = {}) {
return {
version: 1,
id: 'workspace-1',
engineInstanceId: 'engine-1',
mode: 'horizontal',
state: 'ready',
left: context('left'),
right: context('right'),
proof: {level: 'strong', reasons: null},
baselines: {},
baselinePair: {state: 'waiting', reasons: null, resourceCandidates: null, operationCandidates: null},
createdAt: Date.now(),
expiresAt: Date.now() + 60_000,
...extra,
};
}
describe('authorization task response protocol', () => {
it('normalizes nullable collections before the workspace reaches React', () => {
const result = normalizeBrowserAuthorizationTaskResult<ReturnType<typeof workspace>>(
'authorization.workspace.inspect',
workspace(),
);
expect(result.baselinePair.resourceCandidates).toEqual([]);
expect(result.proof.reasons).toEqual([]);
expect(result.left.authentication.authCookieNames).toEqual([]);
});
it('normalizes a null candidate list and candidate reasons', () => {
expect(normalizeBrowserAuthorizationTaskResult(
'authorization.baseline.candidates',
null,
)).toEqual([]);
expect(normalizeBrowserAuthorizationTaskResult(
'authorization.baseline.candidates',
[{id: 'candidate-1', reasons: null}],
)).toEqual([{id: 'candidate-1', reasons: []}]);
});
it('rejects old versions, extra fields, and wrong collection types with field paths', () => {
expect(() => normalizeBrowserAuthorizationTaskResult(
'authorization.workspace.inspect',
workspace({version: 0}),
)).toThrow('$.version');
expect(() => normalizeBrowserAuthorizationTaskResult(
'authorization.workspace.inspect',
workspace({legacy: true}),
)).toThrow('$.legacy');
expect(() => normalizeBrowserAuthorizationTaskResult(
'authorization.workspace.inspect',
workspace({baselinePair: {state: 'waiting', resourceCandidates: {}, operationCandidates: []}}),
)).toThrow('$.baselinePair.resourceCandidates');
});
it('uses a stable schema mismatch code', () => {
try {
normalizeBrowserAuthorizationTaskResult('authorization.workspace.inspect', null);
throw new Error('expected failure');
} catch (error) {
expect(error).toBeInstanceOf(ExtensionError);
expect((error as ExtensionError).code).toBe('authorization_protocol_schema_mismatch');
}
});
});
@@ -1,240 +0,0 @@
import { ExtensionError } from '@/shared/errors';
import type { BrowserAuthorizationTaskSchema } from './engine';
type JSONObject = Record<string, unknown>;
function mismatch(schema: string, path: string, expected: string): never {
throw new ExtensionError(
'authorization_protocol_schema_mismatch',
`授权测试协议 v1 / ${schema} 在 ${path} 不匹配:应为${expected}。请确认 Yak 与插件来自同一版本并重新建立工作区。`,
{ schema, path, protocolVersion: 1 },
);
}
function objectValue(value: unknown, schema: string, path: string): JSONObject {
if (!value || typeof value !== 'object' || Array.isArray(value)) mismatch(schema, path, '对象');
return value as JSONObject;
}
function strictKeys(value: JSONObject, allowed: readonly string[], schema: string, path: string): void {
const keys = new Set(allowed);
for (const key of Object.keys(value)) {
if (!keys.has(key)) mismatch(schema, `${path}.${key}`, '协议声明字段');
}
}
function requiredString(value: JSONObject, key: string, schema: string, path: string): string {
const result = value[key];
if (typeof result !== 'string' || !result) mismatch(schema, `${path}.${key}`, '非空字符串');
return result;
}
function requiredNumber(value: JSONObject, key: string, schema: string, path: string): number {
const result = value[key];
if (typeof result !== 'number' || !Number.isFinite(result)) mismatch(schema, `${path}.${key}`, '有限数字');
return result;
}
function requiredBoolean(value: JSONObject, key: string, schema: string, path: string): boolean {
const result = value[key];
if (typeof result !== 'boolean') mismatch(schema, `${path}.${key}`, '布尔值');
return result;
}
function collection(value: JSONObject, key: string, schema: string, path: string): unknown[] {
const result = value[key];
if (result === undefined || result === null) return [];
if (!Array.isArray(result)) mismatch(schema, `${path}.${key}`, '数组或空值');
return result;
}
function strings(value: JSONObject, key: string, schema: string, path: string): string[] {
return collection(value, key, schema, path).map((item, index) => {
if (typeof item !== 'string') mismatch(schema, `${path}.${key}[${index}]`, '字符串');
return item;
});
}
function objects(
value: JSONObject,
key: string,
schema: string,
path: string,
normalize: (item: JSONObject, itemPath: string) => JSONObject,
): JSONObject[] {
return collection(value, key, schema, path).map((item, index) => {
const itemPath = `${path}.${key}[${index}]`;
return normalize(objectValue(item, schema, itemPath), itemPath);
});
}
function normalizeContext(value: JSONObject, schema: string, path: string): JSONObject {
const target = objectValue(value.target, schema, `${path}.target`);
requiredNumber(target, 'tabId', schema, `${path}.target`);
requiredNumber(target, 'frameId', schema, `${path}.target`);
requiredString(target, 'documentId', schema, `${path}.target`);
const authentication = objectValue(value.authentication, schema, `${path}.authentication`);
requiredString(authentication, 'status', schema, `${path}.authentication`);
requiredNumber(authentication, 'cookieCount', schema, `${path}.authentication`);
requiredNumber(authentication, 'storageEntryCount', schema, `${path}.authentication`);
return {
...value,
target,
authentication: {
...authentication,
authCookieNames: strings(authentication, 'authCookieNames', schema, `${path}.authentication`),
authStorageKeys: strings(authentication, 'authStorageKeys', schema, `${path}.authentication`),
},
};
}
function normalizeBaseline(value: unknown, schema: string, path: string): JSONObject | undefined {
if (value === undefined || value === null) return undefined;
const baseline = objectValue(value, schema, path);
const request = objectValue(baseline.request, schema, `${path}.request`);
const logical = baseline.logicalRequest === undefined || baseline.logicalRequest === null
? undefined
: objectValue(baseline.logicalRequest, schema, `${path}.logicalRequest`);
return {
...baseline,
request: {
...request,
operationNames: strings(request, 'operationNames', schema, `${path}.request`),
headerNames: strings(request, 'headerNames', schema, `${path}.request`),
fields: collection(request, 'fields', schema, `${path}.request`),
},
logicalRequest: logical ? {
...logical,
outputDestinations: strings(logical, 'outputDestinations', schema, `${path}.logicalRequest`),
} : undefined,
};
}
function normalizeWorkspace(value: unknown, schema: string): JSONObject {
const workspace = objectValue(value, schema, '$');
strictKeys(workspace, [
'version', 'id', 'engineInstanceId', 'mode', 'state', 'left', 'right', 'proof', 'baselines',
'baselinePair', 'plan', 'execution', 'createdAt', 'expiresAt', 'staleReason', 'recovery',
], schema, '$');
if (requiredNumber(workspace, 'version', schema, '$') !== 1) mismatch(schema, '$.version', '版本 1');
for (const key of ['id', 'engineInstanceId', 'mode', 'state']) requiredString(workspace, key, schema, '$');
requiredNumber(workspace, 'createdAt', schema, '$');
requiredNumber(workspace, 'expiresAt', schema, '$');
const proof = objectValue(workspace.proof, schema, '$.proof');
requiredString(proof, 'level', schema, '$.proof');
const baselines = objectValue(workspace.baselines, schema, '$.baselines');
const pair = objectValue(workspace.baselinePair, schema, '$.baselinePair');
requiredString(pair, 'state', schema, '$.baselinePair');
const resourceCandidates = objects(pair, 'resourceCandidates', schema, '$.baselinePair', (item, path) => {
for (const key of ['id', 'source', 'location', 'path', 'category', 'confidence']) requiredString(item, key, schema, path);
requiredBoolean(item, 'requiresLogicalBinding', schema, path);
return { ...item, reasons: strings(item, 'reasons', schema, path) };
});
const operationCandidates = objects(pair, 'operationCandidates', schema, '$.baselinePair', (item, path) => {
for (const key of ['id', 'method', 'path']) requiredString(item, key, schema, path);
requiredBoolean(item, 'eligible', schema, path);
requiredBoolean(item, 'sideEffect', schema, path);
requiredBoolean(item, 'requiresDynamicRebuild', schema, path);
return {
...item,
authenticationPaths: strings(item, 'authenticationPaths', schema, path),
dynamicPaths: strings(item, 'dynamicPaths', schema, path),
reasons: strings(item, 'reasons', schema, path),
};
});
let plan = workspace.plan;
if (plan !== undefined && plan !== null) {
const input = objectValue(plan, schema, '$.plan');
plan = {
...input,
canaryPaths: strings(input, 'canaryPaths', schema, '$.plan'),
cases: collection(input, 'cases', schema, '$.plan'),
reasons: strings(input, 'reasons', schema, '$.plan'),
};
}
let execution = workspace.execution;
if (execution !== undefined && execution !== null) {
const input = objectValue(execution, schema, '$.execution');
execution = {
...input,
cases: collection(input, 'cases', schema, '$.execution'),
evidence: collection(input, 'evidence', schema, '$.execution'),
reasons: strings(input, 'reasons', schema, '$.execution'),
};
}
return {
...workspace,
left: normalizeContext(objectValue(workspace.left, schema, '$.left'), schema, '$.left'),
right: normalizeContext(objectValue(workspace.right, schema, '$.right'), schema, '$.right'),
proof: { ...proof, reasons: strings(proof, 'reasons', schema, '$.proof') },
baselines: {
...baselines,
left: normalizeBaseline(baselines.left, schema, '$.baselines.left'),
right: normalizeBaseline(baselines.right, schema, '$.baselines.right'),
verification: normalizeBaseline(baselines.verification, schema, '$.baselines.verification'),
},
baselinePair: {
...pair,
reasons: strings(pair, 'reasons', schema, '$.baselinePair'),
resourceCandidates,
operationCandidates,
},
plan,
execution,
};
}
function normalizeEvidence(value: unknown, schema: string): JSONObject {
const result = objectValue(value, schema, '$');
strictKeys(result, [
'version', 'workspaceId', 'executionId', 'mode', 'verdict', 'confidence', 'cases', 'comparisons',
'semantic', 'representations', 'expiresAt', 'leftCaseId', 'rightCaseId', 'scope', 'view',
'representation', 'equal', 'entries', 'omitted', 'caseId', 'side', 'packetBase64', 'capturedBytes',
'truncated', 'direction', 'verified', 'evidence', 'rejectedPaths', 'verdictChanged', 'reason',
], schema, '$');
if (requiredNumber(result, 'version', schema, '$') !== 1) mismatch(schema, '$.version', '版本 1');
requiredString(result, 'workspaceId', schema, '$');
requiredString(result, 'executionId', schema, '$');
if (schema === 'authorization.evidence.inspect') return {
...result,
cases: collection(result, 'cases', schema, '$'),
comparisons: collection(result, 'comparisons', schema, '$'),
semantic: collection(result, 'semantic', schema, '$'),
representations: strings(result, 'representations', schema, '$'),
};
if (schema === 'authorization.evidence.diff') return {
...result,
entries: collection(result, 'entries', schema, '$'),
};
if (schema === 'authorization.evidence.validate') return {
...result,
evidence: collection(result, 'evidence', schema, '$'),
rejectedPaths: strings(result, 'rejectedPaths', schema, '$'),
};
requiredString(result, 'packetBase64', schema, '$');
return result;
}
export function normalizeBrowserAuthorizationTaskResult<T>(
schema: BrowserAuthorizationTaskSchema,
value: unknown,
): T {
if (schema === 'authorization.baseline.candidates') {
if (value === undefined || value === null) return [] as T;
if (!Array.isArray(value)) mismatch(schema, '$', '数组或空值');
return value.map((candidate, index) => {
const item = objectValue(candidate, schema, `$[${index}]`);
requiredString(item, 'id', schema, `$[${index}]`);
return { ...item, reasons: strings(item, 'reasons', schema, `$[${index}]`) };
}) as T;
}
if ([
'authorization.workspace.create',
'authorization.workspace.inspect',
'authorization.baseline.bind',
'authorization.logical.bind',
'authorization.plan.create',
'authorization.plan.execute',
].includes(schema)) return normalizeWorkspace(value, schema) as T;
return normalizeEvidence(value, schema) as T;
}
@@ -1,301 +0,0 @@
import type { BrowserTransformPacket } from '@/types/models';
import { ExtensionError } from '@/shared/errors';
const RESERVED_PATH_SEGMENTS = new Set(['__proto__', 'prototype', 'constructor']);
const MAX_BODY_PATH_DEPTH = 64;
type ValuePathSegment = string | number;
export type StructuredAuthorizationPrimitive = string | number | boolean;
export interface StructuredAuthorizationBodyValue {
value: StructuredAuthorizationPrimitive;
valueType: 'string' | 'number' | 'boolean';
text: string;
}
function structuredPrimitive(value: unknown): StructuredAuthorizationBodyValue {
if (typeof value === 'string') {
return { value, valueType: 'string', text: value };
}
if (typeof value === 'number' && Number.isFinite(value)) {
return { value, valueType: 'number', text: JSON.stringify(value) };
}
if (typeof value === 'boolean') {
return { value, valueType: 'boolean', text: JSON.stringify(value) };
}
throw new ExtensionError(
'authorization_selector_invalid',
'自动矩阵只接受字符串、数字或布尔 Body 资源值',
);
}
function base64ToUTF8(value: string): string {
let binary: string;
try {
binary = atob(value);
} catch {
throw new ExtensionError(
'authorization_value_invalid',
'结构化请求 Body 不是有效的 Base64',
);
}
try {
return new TextDecoder('utf-8', { fatal: true }).decode(
Uint8Array.from(binary, (character) => character.charCodeAt(0)),
);
} catch {
throw new ExtensionError(
'authorization_value_invalid',
'结构化请求 Body 不是有效的 UTF-8',
);
}
}
function utf8ToBase64(value: string): string {
const bytes = new TextEncoder().encode(value);
let binary = '';
const chunkSize = 0x8000;
for (let offset = 0; offset < bytes.length; offset += chunkSize) {
binary += String.fromCharCode(...bytes.subarray(offset, offset + chunkSize));
}
return btoa(binary);
}
function packetContentType(packet: BrowserTransformPacket): string {
return packet.headers.find((header) => header.name.toLowerCase() === 'content-type')
?.value.toLowerCase() || '';
}
function parseBodyPath(path: string): ValuePathSegment[] {
if (!path.startsWith('body.') && !path.startsWith('body[')) {
throw new ExtensionError(
'authorization_selector_invalid',
'结构化 Body 资源路径必须从 body. 或 body[ 开始',
);
}
const input = path.slice(4);
const segments: ValuePathSegment[] = [];
const pattern = /(?:^|\.)([A-Za-z0-9_-]+)|\[(\d+)]/g;
let offset = 0;
for (const match of input.matchAll(pattern)) {
if (match.index !== offset) {
throw new ExtensionError(
'authorization_selector_invalid',
'结构化 Body 资源路径包含不支持的字段',
);
}
const segment = match[1] ?? Number(match[2]);
if (
typeof segment === 'string'
&& RESERVED_PATH_SEGMENTS.has(segment.toLowerCase())
) {
throw new ExtensionError(
'authorization_selector_invalid',
'结构化 Body 资源路径包含保留字段',
);
}
segments.push(segment);
offset = match.index + match[0].length;
}
if (
offset !== input.length
|| !segments.length
|| segments.length > MAX_BODY_PATH_DEPTH
) {
throw new ExtensionError(
'authorization_selector_invalid',
'结构化 Body 资源路径无效或过深',
);
}
return segments;
}
function parseIndexedFormPath(path: string): { name: string; index?: number } {
if (!path.startsWith('body.')) {
throw new ExtensionError(
'authorization_selector_invalid',
'Form Body 资源路径必须从 body. 开始',
);
}
const raw = path.slice(5);
const matched = raw.match(/^(.*)\[(\d+)]$/);
const name = matched ? matched[1] : raw;
const index = matched ? Number(matched[2]) : undefined;
if (
!name
|| RESERVED_PATH_SEGMENTS.has(name.toLowerCase())
|| (index !== undefined && (!Number.isSafeInteger(index) || index < 0))
) {
throw new ExtensionError(
'authorization_selector_invalid',
'Form Body 资源路径无效',
);
}
return { name, index };
}
function selectedFormOccurrence(
entries: Array<[string, string]>,
name: string,
index?: number,
): { entryIndex: number; value: string } {
const matches = entries.flatMap(([key, value], entryIndex) => (
key === name ? [{ entryIndex, value }] : []
));
if (index === undefined && matches.length !== 1) {
throw new ExtensionError(
'authorization_selector_ambiguous',
'Form Body 存在多个同名资源字段,必须选择带序号的字段',
);
}
const selected = matches[index ?? 0];
if (!selected) {
throw new ExtensionError(
'authorization_selector_invalid',
'Form Body 资源字段不存在',
);
}
return selected;
}
function readJSONBodyValue(
packet: BrowserTransformPacket,
path: string,
): StructuredAuthorizationBodyValue {
let value: unknown;
try {
value = JSON.parse(base64ToUTF8(packet.bodyBase64));
} catch (error) {
if (error instanceof ExtensionError) throw error;
throw new ExtensionError(
'authorization_structured_body_invalid',
'请求 JSON Body 无法解析',
);
}
for (const segment of parseBodyPath(path)) {
if (!value || typeof value !== 'object' || !(segment in value)) {
throw new ExtensionError(
'authorization_selector_invalid',
'JSON Body 资源字段不存在',
);
}
value = (value as Record<string | number, unknown>)[segment];
}
return structuredPrimitive(value);
}
function replaceJSONBodyValue(
packet: BrowserTransformPacket,
path: string,
replacement: StructuredAuthorizationPrimitive,
): BrowserTransformPacket {
let root: unknown;
try {
root = JSON.parse(base64ToUTF8(packet.bodyBase64));
} catch (error) {
if (error instanceof ExtensionError) throw error;
throw new ExtensionError(
'authorization_structured_body_invalid',
'请求 JSON Body 无法解析',
);
}
const segments = parseBodyPath(path);
let parent = root;
for (const segment of segments.slice(0, -1)) {
if (!parent || typeof parent !== 'object' || !(segment in parent)) {
throw new ExtensionError(
'authorization_selector_invalid',
'JSON Body 资源字段不存在',
);
}
parent = (parent as Record<string | number, unknown>)[segment];
}
const leaf = segments.at(-1);
if (
leaf === undefined
|| !parent
|| typeof parent !== 'object'
|| !(leaf in parent)
) {
throw new ExtensionError(
'authorization_selector_invalid',
'JSON Body 资源字段不存在',
);
}
const current = structuredPrimitive(
(parent as Record<string | number, unknown>)[leaf],
);
if (current.valueType !== typeof replacement) {
throw new ExtensionError(
'authorization_selector_invalid',
'JSON Body 资源替换不能改变字段类型',
);
}
(parent as Record<string | number, unknown>)[leaf] = replacement;
return {
...packet,
bodyBase64: utf8ToBase64(JSON.stringify(root)),
};
}
export function isStructuredAuthorizationBody(packet: BrowserTransformPacket): boolean {
const contentType = packetContentType(packet);
return contentType.includes('json')
|| contentType.includes('application/x-www-form-urlencoded');
}
export function readStructuredAuthorizationBodyValue(
packet: BrowserTransformPacket,
path: string,
): StructuredAuthorizationBodyValue {
const contentType = packetContentType(packet);
if (contentType.includes('json')) {
return readJSONBodyValue(packet, path);
}
if (contentType.includes('application/x-www-form-urlencoded')) {
const selected = parseIndexedFormPath(path);
const value = selectedFormOccurrence(
[...new URLSearchParams(base64ToUTF8(packet.bodyBase64))],
selected.name,
selected.index,
).value;
return { value, valueType: 'string', text: value };
}
throw new ExtensionError(
'authorization_selector_invalid',
'直接 Body 资源替换仅支持 JSON 或 Form 请求',
);
}
export function replaceStructuredAuthorizationBodyValue(input: {
packet: BrowserTransformPacket;
path: string;
replacement: StructuredAuthorizationPrimitive;
}): BrowserTransformPacket {
const contentType = packetContentType(input.packet);
if (contentType.includes('json')) {
return replaceJSONBodyValue(input.packet, input.path, input.replacement);
}
if (contentType.includes('application/x-www-form-urlencoded')) {
if (typeof input.replacement !== 'string') {
throw new ExtensionError(
'authorization_selector_invalid',
'Form Body 资源替换只接受字符串',
);
}
const selected = parseIndexedFormPath(input.path);
const entries = [...new URLSearchParams(base64ToUTF8(input.packet.bodyBase64))];
const occurrence = selectedFormOccurrence(entries, selected.name, selected.index);
entries[occurrence.entryIndex][1] = input.replacement;
const form = new URLSearchParams();
entries.forEach(([name, value]) => form.append(name, value));
return {
...input.packet,
bodyBase64: utf8ToBase64(form.toString()),
};
}
throw new ExtensionError(
'authorization_selector_invalid',
'直接 Body 资源替换仅支持 JSON 或 Form 请求',
);
}
@@ -1,365 +0,0 @@
import { useEffect, useState } from 'react';
import {
AlertTriangle, ArrowRight, Check, CircleCheck, Code2, FileDiff, FileText, Timer,
} from 'lucide-react';
import { errorMessage } from '@/platform/messaging/runtime';
import {
runBrowserAuthorizationTask,
type BrowserAuthorizationEvidenceBundle,
type BrowserAuthorizationEvidenceDiff,
type BrowserAuthorizationEvidencePacket,
type BrowserAuthorizationEvidenceValidation,
type BrowserAuthorizationWorkspace,
} from '../engine';
function decodeEvidencePacket(packetBase64: string): string {
const binary = atob(packetBase64);
const bytes = Uint8Array.from(binary, (character) => character.charCodeAt(0));
return new TextDecoder().decode(bytes);
}
export function compactDuration(value: number): string {
if (!Number.isFinite(value)) return '—';
if (value < 1) return `${value.toFixed(2)} ms`;
if (value < 100) return `${value.toFixed(1)} ms`;
return `${Math.round(value)} ms`;
}
function formatResponseAnalysis(response?: BrowserAuthorizationEvidenceBundle['cases'][number]['response']): string {
if (!response) return '';
if (response.analysisState === 'encoded-unavailable') return ' · 编码正文不可分析';
if (response.analysisRepresentation === 'binary') return ' · 二进制摘要';
if (response.decoded) {
const encoding = response.contentEncoding || '压缩内容';
const representation = response.analysisRepresentation?.toUpperCase() || '正文';
return ` · ${encoding} → ${representation}`;
}
return '';
}
export function AuthorizationEvidenceWorkbench({
workspace,
onWorkspaceChange,
}: {
workspace: BrowserAuthorizationWorkspace;
onWorkspaceChange: (workspace: BrowserAuthorizationWorkspace) => void;
}) {
const execution = workspace.execution!;
const [bundle, setBundle] = useState<BrowserAuthorizationEvidenceBundle>();
const [comparisonId, setComparisonId] = useState('');
const [diff, setDiff] = useState<BrowserAuthorizationEvidenceDiff>();
const [packet, setPacket] = useState<BrowserAuthorizationEvidencePacket>();
const [packetTitle, setPacketTitle] = useState('');
const [view, setView] = useState<'redacted' | 'raw'>('redacted');
const [showVolatile, setShowVolatile] = useState(false);
const [loading, setLoading] = useState(false);
const [validatingPath, setValidatingPath] = useState('');
const [validationMessage, setValidationMessage] = useState('');
const [error, setError] = useState('');
useEffect(() => {
let disposed = false;
setLoading(true);
setError('');
setBundle(undefined);
setDiff(undefined);
setPacket(undefined);
void runBrowserAuthorizationTask<BrowserAuthorizationEvidenceBundle>(
'authorization.evidence.inspect',
{ workspaceId: workspace.id, executionId: execution.id },
).then((next) => {
if (disposed) return;
setBundle(next);
const preferred = next.comparisons.find((item) => item.purpose === 'authorization')
|| next.comparisons[0];
setComparisonId(preferred?.id || '');
}).catch((cause) => {
if (!disposed) setError(errorMessage(cause));
}).finally(() => {
if (!disposed) setLoading(false);
});
return () => { disposed = true; };
}, [execution.id, workspace.id]);
const comparison = bundle?.comparisons.find((item) => item.id === comparisonId);
const comparisonCases = comparison
? bundle?.cases.filter((item) => item.id === comparison.leftCaseId || item.id === comparison.rightCaseId) || []
: [];
const comparisonTruncated = comparisonCases.some((item) => item.response?.truncated);
const comparisonEncodedUnavailable = comparisonCases.some(
(item) => item.response?.analysisState === 'encoded-unavailable',
);
const rawDiffEntries = diff?.entries;
const diffEntries = Array.isArray(rawDiffEntries) ? rawDiffEntries : [];
const diffRepresentationLabel = diff?.representation === 'structured'
? '结构化字段差异'
: diffEntries.some((entry) => entry.path.includes('.body.binary.'))
? '二进制摘要差异'
: diffEntries.some((entry) => entry.path.includes('.body.encoded.'))
? '编码正文元数据差异'
: '原始文本差异';
const volatileCount = diffEntries.filter((entry) => entry.volatile).length;
const visibleEntries = diffEntries.filter((entry) => showVolatile || !entry.volatile);
const executionEvidence = Array.isArray(execution.evidence) ? execution.evidence : [];
const validationDirections: BrowserAuthorizationEvidenceValidation['direction'][] = comparison?.id === 'controls'
? ['a-to-b', 'b-to-a']
: comparison?.id === 'a-to-b'
? ['a-to-b']
: comparison?.id === 'b-to-a'
? ['b-to-a']
: comparison?.id === 'low-vs-privileged' || comparison?.id === 'probe-vs-privileged'
? ['low-to-privileged']
: comparison?.id === 'post-state'
? ['post-state']
: [];
useEffect(() => {
if (!comparison) return;
let disposed = false;
setLoading(true);
setError('');
setPacket(undefined);
void runBrowserAuthorizationTask<BrowserAuthorizationEvidenceDiff>(
'authorization.evidence.diff',
{
workspaceId: workspace.id,
executionId: execution.id,
leftCaseId: comparison.leftCaseId,
rightCaseId: comparison.rightCaseId,
scope: 'response',
view,
},
).then((next) => {
if (!disposed) setDiff(next);
}).catch((cause) => {
if (!disposed) setError(errorMessage(cause));
}).finally(() => {
if (!disposed) setLoading(false);
});
return () => { disposed = true; };
}, [comparison?.id, execution.id, view, workspace.id]);
const changeView = (next: 'redacted' | 'raw') => {
if (next === 'raw' && !window.confirm(
'原始证据可能包含 Cookie、Authorization 与业务敏感值。仅在当前授权测试确有需要时显示。',
)) return;
setView(next);
setPacket(undefined);
};
const openPacket = async (
caseId: string,
side: 'request' | 'response',
label: string,
) => {
setLoading(true);
setError('');
try {
const next = await runBrowserAuthorizationTask<BrowserAuthorizationEvidencePacket>(
'authorization.evidence.packet',
{
workspaceId: workspace.id,
executionId: execution.id,
caseId,
side,
view,
},
);
setPacket(next);
setPacketTitle(`${label} · ${side === 'request' ? '请求' : '响应'}`);
} catch (cause) {
setError(errorMessage(cause));
} finally {
setLoading(false);
}
};
const validatePath = async (
path: string,
direction: BrowserAuthorizationEvidenceValidation['direction'],
) => {
const validationKey = `${direction}:${path}`;
setValidatingPath(validationKey);
setValidationMessage('');
setError('');
try {
const validation = await runBrowserAuthorizationTask<BrowserAuthorizationEvidenceValidation>(
'authorization.evidence.validate',
{
workspaceId: workspace.id,
executionId: execution.id,
direction,
paths: [path],
},
);
setValidationMessage(validation.reason);
const validationEvidence = Array.isArray(validation.evidence) ? validation.evidence : [];
const additions = validationEvidence.filter((candidate) => !executionEvidence.some((current) => (
current.direction === candidate.direction
&& current.path === candidate.path
&& current.source === candidate.source
)));
onWorkspaceChange({
...workspace,
execution: {
...execution,
verdict: validation.verdict,
confidence: validation.confidence,
evidence: [...executionEvidence, ...additions],
reasons: validation.verdictChanged
? [...execution.reasons, validation.reason]
: execution.reasons,
},
});
} catch (cause) {
setError(errorMessage(cause));
} finally {
setValidatingPath('');
}
};
return <div className="authorization-evidence-workbench">
<div className="authorization-evidence-title">
<div>
<span>短时证据包</span>
<strong>交叉请求与业务归属证据</strong>
<small>
报文仅在当前工作区短时保留;差异默认脱敏,时间戳与请求 ID 会单独降噪。
{bundle ? ` · 保留至 ${new Date(bundle.expiresAt).toLocaleTimeString()}` : ''}
</small>
</div>
<div className="authorization-evidence-view">
<button className={view === 'redacted' ? 'active' : ''} onClick={() => changeView('redacted')}>脱敏</button>
<button className={view === 'raw' ? 'active raw' : ''} onClick={() => changeView('raw')}>原始值</button>
</div>
</div>
{bundle && <div className="authorization-evidence-trace" aria-label="测试请求执行顺序">
{bundle.cases.map((item, index) => <div key={item.id}>
<span>{String(index + 1).padStart(2, '0')}</span>
<strong>{item.label}</strong>
<small>
{item.status || '—'} · {compactDuration(item.timing.totalMs)}
{item.timing.ttfbMs > 0 ? ` · 首字节 ${compactDuration(item.timing.ttfbMs)}` : ''}
{formatResponseAnalysis(item.response)}
</small>
<nav>
<button disabled={!item.requestAvailable || loading} onClick={() => void openPacket(item.id, 'request', item.label)}>
<Code2 size={12} />请求
</button>
<button disabled={!item.responseAvailable || loading} onClick={() => void openPacket(item.id, 'response', item.label)}>
<FileText size={12} />响应
</button>
</nav>
</div>)}
</div>}
<div className="authorization-evidence-body">
<aside>
<span>比较关系</span>
{bundle?.comparisons.map((item) => <button
key={item.id}
className={item.id === comparisonId ? 'active' : ''}
onClick={() => {
setComparisonId(item.id);
setPacket(undefined);
}}
>
<i>{item.purpose === 'authorization' ? '关键' : item.purpose === 'state-change' ? '状态' : '对照'}</i>
<strong>{item.label}</strong>
</button>)}
</aside>
<main>
<header>
<div>
{packet ? <FileText size={16} /> : <FileDiff size={16} />}
<span><strong>{packet ? packetTitle : comparison?.label || '响应差异'}</strong>
<small>{packet
? `${packet.view === 'raw' ? '原始' : '脱敏'}报文${packet.truncated ? ' · 已截断' : ''}`
: diffRepresentationLabel}</small>
</span>
</div>
{packet
? <button onClick={() => setPacket(undefined)}><FileDiff size={13} />返回差异</button>
: volatileCount > 0 && <button onClick={() => setShowVolatile((current) => !current)}>
{showVolatile ? '隐藏' : '显示'}动态噪声 · {volatileCount}
</button>}
</header>
{loading && <div className="authorization-evidence-empty"><Timer size={17} />正在读取证据…</div>}
{!loading && error && <div className="authorization-evidence-empty error"><AlertTriangle size={17} />{error}</div>}
{!loading && !error && packet && <pre>{decodeEvidencePacket(packet.packetBase64)}</pre>}
{!loading && !error && !packet && diff?.equal && <div className="authorization-evidence-empty">
<CircleCheck size={17} />{comparison?.purpose === 'authorization'
? comparisonTruncated
? '两项响应已捕获部分一致,但至少一项已截断,不能据此判断资源归属。'
: comparisonEncodedUnavailable
? '两项线上编码正文指纹一致,但正文未能在预算内解码,不能据此提升授权结论。'
: '交叉响应与目标身份响应完全一致;如结论尚未确认,请切换到“身份 A 自有资源 ↔ 身份 B 自有资源”,选择稳定业务字段验证。'
: comparison?.purpose === 'state-change'
? '操作前后的稳定业务字段没有变化。'
: '双方正常响应完全一致,当前对照没有可用于区分资源归属的字段。'}
</div>}
{!loading && !error && !packet && diff && !diff.equal
&& visibleEntries.length === 0 && volatileCount > 0 && !showVolatile
&& <div className="authorization-evidence-empty">
<Timer size={17} />当前差异只有 {volatileCount} 项动态噪声,已默认折叠。
</div>}
{!packet && validationMessage && <div className="authorization-evidence-validation">
<Check size={13} />{validationMessage}
</div>}
{!loading && !error && !packet && diff && !diff.equal && visibleEntries.length > 0 && <div className="authorization-diff-list">
{visibleEntries.slice(0, 80).map((entry) => {
const pendingDirections = validationDirections.filter((direction) => !executionEvidence.some((item) => (
item.path === entry.path && item.direction === direction
)));
const alreadyVerified = pendingDirections.length < validationDirections.length;
const canValidate = Boolean(
pendingDirections.length
&& diff.scope === 'response'
&& entry.path.startsWith('body.')
&& !entry.volatile
&& !entry.sensitive
);
return <div
key={`${entry.path}-${entry.kind}`}
className={`${entry.semantic || alreadyVerified ? 'semantic' : ''} ${entry.volatile ? 'volatile' : ''}`}
>
<div>
<code>{entry.path}</code>
<span>{alreadyVerified
? pendingDirections.length ? '部分已验证' : '已验证'
: entry.semantic ? '归属候选' : entry.volatile ? '动态噪声' : entry.sensitive ? '敏感字段' : entry.kind}</span>
{canValidate && pendingDirections.map((direction) => {
const validationKey = `${direction}:${entry.path}`;
const label = direction === 'a-to-b'
? '验证 A→B'
: direction === 'b-to-a'
? '验证 B→A'
: direction === 'post-state'
? '验证状态变化'
: '核对低权探测';
return <button
key={direction}
disabled={Boolean(validatingPath)}
onClick={() => void validatePath(entry.path, direction)}
>
{validatingPath === validationKey ? '验证中…' : label}
</button>;
})}
</div>
<section>
<p><b>左</b><span title={entry.left}>{entry.left || '—'}</span></p>
<ArrowRight size={13} />
<p><b>右</b><span title={entry.right}>{entry.right || '—'}</span></p>
</section>
</div>;
})}
{(visibleEntries.length > 80 || diff.omitted > 0) && <small className="authorization-diff-omitted">
当前展示前 80 项,另有 {Math.max(0, visibleEntries.length - 80) + diff.omitted} 项未展开
</small>}
</div>}
</main>
</div>
</div>;
}
File diff suppressed because it is too large Load Diff
@@ -1,4 +1,4 @@
import type { ActiveTabInfo, BrowserIsolationContext } from '@/types/models';
import type { ActiveTabInfo, BrowserAuthorizationInstance } from '@/types/models';
function shortPageAddress(tab: ActiveTabInfo): string {
try {
@@ -9,67 +9,59 @@ function shortPageAddress(tab: ActiveTabInfo): string {
}
}
function contextKindLabel(
context: BrowserIsolationContext | undefined,
selectedTab: ActiveTabInfo | undefined,
): string {
if (!selectedTab) return '等待选择页面';
switch (context?.kind) {
case 'chrome-incognito-store': return '无痕隔离上下文';
case 'firefox-container':
return context.containerName ? `Container · ${context.containerName}` : 'Container 隔离上下文';
case 'managed-ephemeral-profile': return '独立浏览器 Profile';
case 'verified-tab-local': return '标签页局部上下文';
case 'sequential-auth-snapshot': return '顺序身份快照';
default: return selectedTab.incognito ? '无痕浏览上下文' : '普通浏览上下文';
}
}
function windowKindLabel(tab: ActiveTabInfo): string {
return tab.incognito ? '无痕窗口' : '普通窗口';
}
export function IdentitySlot({
side, title, label, setLabel, tabId, setTabId, tabs, context, disabledReason, emptyHint,
side, title, label, setLabel, instance, instances, setInstanceId, tabId, setTabId,
}: {
side: 'A' | 'B';
title: string;
label: string;
setLabel: (value: string) => void;
instance?: BrowserAuthorizationInstance;
instances: BrowserAuthorizationInstance[];
setInstanceId?: (value: string) => void;
tabId?: number;
setTabId: (value: number | undefined) => void;
tabs: ActiveTabInfo[];
context?: BrowserIsolationContext;
disabledReason: (tab: ActiveTabInfo) => string | undefined;
emptyHint: string;
}) {
const selectedTab = tabs.find((item) => item.id === tabId);
const selectedTab = instance?.tabs.find((item) => item.id === tabId);
return <div className={`authorization-identity-slot ${selectedTab ? 'is-selected' : 'is-empty'}`}>
<header><span>{side}</span><div><strong>{title}</strong><small>{contextKindLabel(context, selectedTab)}</small></div></header>
<label><span>账号备注</span><input value={label} maxLength={80} onChange={(event) => setLabel(event.target.value)} placeholder={side === 'A' ? '例如:普通用户' : '例如:另一个用户'} /></label>
<label><span>{side === 'A' ? '当前已登录页面' : '另一个已登录页面'}</span><select
aria-label={`身份 ${side} 的已登录页面`}
value={selectedTab?.id || ''}
onChange={(event) => setTabId(event.target.value ? Number(event.target.value) : undefined)}
>
<option value="">{side === 'A' ? '选择当前登录页面' : '选择页面,或在中间创建隔离身份'}</option>
{tabs.map((item) => {
const reason = disabledReason(item);
return <option value={item.id} key={item.id} disabled={Boolean(reason)}>
{item.title} · {shortPageAddress(item)} · {windowKindLabel(item)}{reason ? ` · ${reason}` : ''}
</option>;
})}
</select></label>
<header>
<span>{instance?.badge || side}</span>
<div>
<strong>{title}</strong>
<small>{instance ? `YTray 浏览器 ${instance.badge} · 在线` : '等待在线浏览器'}</small>
</div>
</header>
<label>
<span>账号备注</span>
<input value={label} maxLength={80} onChange={(event) => setLabel(event.target.value)} placeholder={side === 'A' ? '例如:资源所有者' : '例如:对照账号'} />
</label>
{setInstanceId && <label>
<span>浏览器实例</span>
<select aria-label={`身份 ${side} 的浏览器实例`} value={instance?.deviceId || ''} onChange={(event) => setInstanceId(event.target.value)}>
<option value="">选择另一个在线实例</option>
{instances.filter((item) => !item.current).map((item) => <option value={item.deviceId} key={item.deviceId}>
浏览器 {item.badge} · {item.tabs.length} 个页面
</option>)}
</select>
</label>}
<label>
<span>已登录页面</span>
<select
aria-label={`身份 ${side} 的已登录页面`}
value={selectedTab?.id || ''}
disabled={!instance}
onChange={(event) => setTabId(event.target.value ? Number(event.target.value) : undefined)}
>
<option value="">{instance ? '选择 HTTP(S) 页面' : '先选择浏览器实例'}</option>
{instance?.tabs.map((item) => <option value={item.id} key={item.id}>
{item.title || '未命名页面'} · {shortPageAddress(item)}
</option>)}
</select>
</label>
<div className="authorization-identity-meta">
<span><i className={context?.level || ''} />{selectedTab
? context?.level === 'strong'
? '强隔离上下文'
: context?.level === 'conditional'
? '条件隔离上下文'
: '隔离待验证'
: '尚未选择页面'}</span>
<code title={selectedTab?.url || emptyHint}>
{selectedTab ? `${windowKindLabel(selectedTab)} · ${selectedTab.url}` : emptyHint}
<span><i className={instance ? 'strong' : ''} />{instance ? '独立浏览器 Profile' : '尚未选择实例'}</span>
<code title={selectedTab?.url || instance?.error || ''}>
{selectedTab?.url || instance?.error || '请先在该浏览器打开并登录目标站点'}
</code>
</div>
</div>;
File diff suppressed because it is too large Load Diff
@@ -1,96 +0,0 @@
import { describe, expect, it } from 'vitest';
import {
authorizationIdentityOptionDisabledReason,
normalizeAuthorizationIdentityTabSelection,
} from './identity-selection';
describe('normalizeAuthorizationIdentityTabSelection', () => {
it('moves the only surviving persisted page to identity A', () => {
expect(normalizeAuthorizationIdentityTabSelection({
eligibleTabIds: [22],
activeTabId: 22,
leftTabId: 11,
rightTabId: 22,
})).toEqual({
leftTabId: 22,
rightTabId: undefined,
});
});
it('clears stale selections without visually falling back to another page', () => {
expect(normalizeAuthorizationIdentityTabSelection({
eligibleTabIds: [],
leftTabId: 11,
rightTabId: 22,
})).toEqual({
leftTabId: undefined,
rightTabId: undefined,
});
});
it('keeps two different valid user selections', () => {
expect(normalizeAuthorizationIdentityTabSelection({
eligibleTabIds: [11, 22],
activeTabId: 22,
leftTabId: 11,
rightTabId: 22,
})).toEqual({
leftTabId: 11,
rightTabId: 22,
});
});
it('uses the active page for A while preserving a different B page', () => {
expect(normalizeAuthorizationIdentityTabSelection({
eligibleTabIds: [11, 22],
activeTabId: 11,
leftTabId: 99,
rightTabId: 22,
})).toEqual({
leftTabId: 11,
rightTabId: 22,
});
});
it('does not automatically treat a second ordinary tab as identity B', () => {
expect(normalizeAuthorizationIdentityTabSelection({
eligibleTabIds: [11, 22],
activeTabId: 11,
})).toEqual({
leftTabId: 11,
rightTabId: undefined,
});
});
});
describe('authorizationIdentityOptionDisabledReason', () => {
it('disables the exact page already assigned to the other identity', () => {
expect(authorizationIdentityOptionDisabledReason({
candidateTabId: 11,
candidateIsolationContextId: 'profile:normal',
otherTabId: 11,
otherIsolationContextId: 'profile:normal',
otherLabel: '身份 A',
})).toBe('已用于身份 A');
});
it('disables another page that shares the other identity login context', () => {
expect(authorizationIdentityOptionDisabledReason({
candidateTabId: 22,
candidateIsolationContextId: 'profile:normal',
otherTabId: 11,
otherIsolationContextId: 'profile:normal',
otherLabel: '身份 A',
})).toBe('与身份 A 共享登录态');
});
it('keeps pages from another isolation context selectable', () => {
expect(authorizationIdentityOptionDisabledReason({
candidateTabId: 22,
candidateIsolationContextId: 'profile:incognito',
otherTabId: 11,
otherIsolationContextId: 'profile:normal',
otherLabel: '身份 A',
})).toBeUndefined();
});
});
@@ -1,67 +0,0 @@
export interface AuthorizationIdentityTabSelection {
leftTabId?: number;
rightTabId?: number;
}
export interface NormalizeAuthorizationIdentityTabSelectionInput
extends AuthorizationIdentityTabSelection {
eligibleTabIds: readonly number[];
activeTabId?: number;
}
export interface AuthorizationIdentityOptionConflictInput {
candidateTabId: number;
candidateIsolationContextId?: string;
otherTabId?: number;
otherIsolationContextId?: string;
otherLabel: string;
}
export function authorizationIdentityOptionDisabledReason({
candidateTabId,
candidateIsolationContextId,
otherTabId,
otherIsolationContextId,
otherLabel,
}: AuthorizationIdentityOptionConflictInput): string | undefined {
if (otherTabId !== undefined && candidateTabId === otherTabId) {
return `已用于${otherLabel}`;
}
if (
candidateIsolationContextId
&& otherIsolationContextId
&& candidateIsolationContextId === otherIsolationContextId
) {
return `与${otherLabel} 共享登录态`;
}
return undefined;
}
export function normalizeAuthorizationIdentityTabSelection({
eligibleTabIds,
activeTabId,
leftTabId,
rightTabId,
}: NormalizeAuthorizationIdentityTabSelectionInput): AuthorizationIdentityTabSelection {
const available = new Set(
eligibleTabIds.filter((tabId) => Number.isSafeInteger(tabId) && tabId > 0),
);
const existing = (tabId?: number): number | undefined => (
tabId !== undefined && available.has(tabId) ? tabId : undefined
);
let left = existing(leftTabId);
let right = existing(rightTabId);
if (left !== undefined && left === right) right = undefined;
if (left === undefined) {
left = existing(activeTabId) ?? right ?? eligibleTabIds.find((tabId) => available.has(tabId));
if (left === right) right = undefined;
}
return {
leftTabId: left,
rightTabId: right,
};
}
@@ -1,262 +0,0 @@
import { describe, expect, it } from 'vitest';
import type { BrowserAuthorizationWorkspace } from '../engine';
import {
authorizationWorkspaceUIReducer,
authorizationWorkspaceStage,
INITIAL_AUTHORIZATION_WORKSPACE_UI,
normalizePersistedAuthorizationWorkspaceUI,
persistedAuthorizationWorkspaceUI,
} from './workspace-reducer';
function fixtureWorkspace(): BrowserAuthorizationWorkspace {
return {
version: 1,
id: 'workspace-1',
engineInstanceId: 'engine-1',
mode: 'horizontal',
state: 'ready',
left: {
accountLabel: '账号 A',
origin: 'https://example.test',
target: { tabId: 11, frameId: 0, documentId: 'document-a' },
authentication: { status: 'authenticated', cookieCount: 1, storageEntryCount: 0 },
},
right: {
accountLabel: '账号 B',
origin: 'https://example.test',
target: { tabId: 22, frameId: 0, documentId: 'document-b' },
authentication: { status: 'authenticated', cookieCount: 1, storageEntryCount: 0 },
},
proof: {
level: 'strong',
sameOrigin: true,
cookieStoreRelation: 'different',
accountEvidenceRelation: 'different',
requestCredentialRelation: 'different',
refreshCheck: 'passed',
reasons: ['隔离成立'],
},
baselines: {},
baselinePair: {
state: 'waiting',
reasons: ['等待正常请求'],
resourceCandidates: [],
operationCandidates: [],
},
expiresAt: Date.now() + 60_000,
};
}
describe('authorization workspace UI reducer', () => {
it('initializes a renewed workspace and clears evidence tied to the old document', () => {
const workspace = { id: 'renewed' } as BrowserAuthorizationWorkspace;
const previous = {
...INITIAL_AUTHORIZATION_WORKSPACE_UI,
candidates: { left: [{ id: 'old-left' }], right: [{ id: 'old-right' }] } as never,
selected: { left: 'old-left', right: 'old-right' },
selectedPlanCandidateId: 'old-plan',
};
const next = authorizationWorkspaceUIReducer(previous, {
type: 'workspace.initialize',
workspace,
});
expect(next.workspace).toBe(workspace);
expect(next.candidates).toEqual({ left: [], right: [] });
expect(next.selected).toEqual({ left: '', right: '' });
expect(next.selectedPlanCandidateId).toBe('');
});
it('resets workflow evidence without discarding the selected identities', () => {
const previous = {
...INITIAL_AUTHORIZATION_WORKSPACE_UI,
leftTabId: 11,
rightTabId: 12,
workspace: { id: 'old' } as BrowserAuthorizationWorkspace,
capture: { left: { active: true } } as never,
};
const next = authorizationWorkspaceUIReducer(previous, { type: 'workspace.reset' });
expect(next.leftTabId).toBe(11);
expect(next.rightTabId).toBe(12);
expect(next.workspace).toBeUndefined();
expect(next.capture).toEqual({});
});
it('persists only durable workflow state', () => {
const value = persistedAuthorizationWorkspaceUI({
...INITIAL_AUTHORIZATION_WORKSPACE_UI,
inspection: { version: 1 } as never,
capture: { left: { active: true } } as never,
});
expect(value).not.toHaveProperty('inspection');
expect(value).not.toHaveProperty('capture');
});
it('fails closed when a restarted UI session contains a malformed workspace', () => {
const next = authorizationWorkspaceUIReducer(INITIAL_AUTHORIZATION_WORKSPACE_UI, {
type: 'hydrate',
value: {
mode: 'vertical',
leftTabId: 11,
rightTabId: 'not-a-tab',
leftLabel: '低权限账号',
workspace: { id: 'truncated-before-storage-write' },
candidates: { left: [null], right: { invalid: true } },
selected: null,
},
});
expect(next).toMatchObject({
mode: 'vertical',
leftTabId: 11,
leftLabel: '低权限账号',
workspace: undefined,
candidates: { left: [], right: [] },
selected: { left: '', right: '' },
});
expect(next.rightTabId).toBeUndefined();
});
it('normalizes a valid persisted workflow but drops invalid candidate entries', () => {
const workspace = {
...fixtureWorkspace(),
createdAt: Date.now(),
};
const normalized = normalizePersistedAuthorizationWorkspaceUI({
mode: 'horizontal',
leftTabId: 11,
rightTabId: 22,
leftLabel: '账号 A',
rightLabel: '账号 B',
workspace,
candidates: {
left: [{
id: 'left-request',
method: 'GET',
url: 'https://example.test/api/profile?id=1',
path: '/api/profile',
resourceType: 'xmlhttprequest',
startedAt: Date.now(),
eligible: true,
reasons: [],
}, { id: 'invalid-url', url: 'javascript:alert(1)' }],
right: [],
},
selected: { left: 'left-request', right: '' },
selectedPlanCandidateId: '',
canaryPaths: 'data.owner.id',
});
expect(normalized?.workspace?.id).toBe('workspace-1');
expect(normalized?.candidates?.left).toEqual([
expect.objectContaining({ id: 'left-request' }),
]);
expect(normalized?.selected?.left).toBe('left-request');
});
it('models the complete identity-to-evidence workflow without losing capture state', () => {
let current = INITIAL_AUTHORIZATION_WORKSPACE_UI;
expect(authorizationWorkspaceStage(current)).toBe('identity');
const initial = fixtureWorkspace();
current = authorizationWorkspaceUIReducer(current, {
type: 'workspace.initialize',
workspace: initial,
});
current = authorizationWorkspaceUIReducer(current, {
type: 'capture.replace',
capture: {
left: { active: true, count: 1 } as never,
right: { active: true, count: 1 } as never,
},
});
expect(authorizationWorkspaceStage(current)).toBe('normal-requests');
const baseline = {
id: 'baseline',
networkRequestId: 'request',
request: {
method: 'GET',
url: 'https://example.test/api/profile?id=1',
path: '/api/profile',
contentType: 'application/json',
actionFingerprint: 'fingerprint',
},
};
const bound = {
...initial,
baselines: { left: { ...baseline, id: 'left' }, right: { ...baseline, id: 'right' } },
baselinePair: {
state: 'matched' as const,
reasons: ['同类请求'],
resourceCandidates: [{
id: 'resource-id',
source: 'wire' as const,
location: 'query' as const,
path: 'query.id',
category: 'identifier',
confidence: 'high' as const,
requiresLogicalBinding: false,
reasons: ['A/B 值不同'],
}],
operationCandidates: [],
},
};
current = authorizationWorkspaceUIReducer(current, {
type: 'baselines.loaded',
candidates: {
left: [{ id: 'left-request' }] as never,
right: [{ id: 'right-request' }] as never,
},
selected: { left: 'left-request', right: 'right-request' },
});
current = authorizationWorkspaceUIReducer(current, {
type: 'baselines.bound',
workspace: bound,
selectedPlanCandidateId: 'resource-id',
});
expect(authorizationWorkspaceStage(current)).toBe('plan');
const planned = {
...bound,
plan: {
id: 'plan-1',
mode: 'horizontal' as const,
candidateId: 'resource-id',
state: 'ready' as const,
selector: { source: 'wire' as const, location: 'query' as const, path: 'query.id' },
cases: [],
requestBudget: 4,
requiresDynamicRebuild: false,
reasons: ['固定四项矩阵'],
},
};
current = authorizationWorkspaceUIReducer(current, {
type: 'workspace.updated',
workspace: planned,
});
expect(authorizationWorkspaceStage(current)).toBe('execution');
current = authorizationWorkspaceUIReducer(current, {
type: 'workspace.updated',
workspace: {
...planned,
execution: {
id: 'execution-1',
state: 'completed',
verdict: 'protected',
confidence: 'high',
requestCount: 4,
cases: [],
evidence: [],
evidenceAvailable: true,
reasons: ['交叉访问均被拒绝'],
},
},
});
expect(authorizationWorkspaceStage(current)).toBe('evidence');
expect(current.capture.left?.active).toBe(true);
expect(persistedAuthorizationWorkspaceUI(current)).not.toHaveProperty('capture');
});
});
@@ -1,364 +0,0 @@
import type {
BrowserIsolationInspection,
NetworkCaptureStatus,
} from '@/types/models';
import type {
BrowserAuthorizationBaselineCandidate,
BrowserAuthorizationMode,
BrowserAuthorizationSide,
BrowserAuthorizationWorkspace,
} from '../engine';
import { normalizeBrowserAuthorizationTaskResult } from '../protocol';
export const EMPTY_AUTHORIZATION_CANDIDATES: Record<
BrowserAuthorizationSide,
BrowserAuthorizationBaselineCandidate[]
> = { left: [], right: [] };
const EMPTY_SELECTION: Record<BrowserAuthorizationSide, string> = { left: '', right: '' };
export interface PersistedAuthorizationWorkspaceUI {
mode: BrowserAuthorizationMode;
leftTabId?: number;
rightTabId?: number;
leftLabel: string;
rightLabel: string;
workspace?: BrowserAuthorizationWorkspace;
candidates: Record<BrowserAuthorizationSide, BrowserAuthorizationBaselineCandidate[]>;
selected: Record<BrowserAuthorizationSide, string>;
selectedPlanCandidateId: string;
canaryPaths: string;
}
export interface AuthorizationWorkspaceUIState extends PersistedAuthorizationWorkspaceUI {
inspection?: BrowserIsolationInspection;
capture: Partial<Record<BrowserAuthorizationSide, NetworkCaptureStatus>>;
}
export const INITIAL_AUTHORIZATION_WORKSPACE_UI: AuthorizationWorkspaceUIState = {
mode: 'horizontal',
leftLabel: '账号 A',
rightLabel: '账号 B',
candidates: EMPTY_AUTHORIZATION_CANDIDATES,
selected: EMPTY_SELECTION,
selectedPlanCandidateId: '',
canaryPaths: '',
capture: {},
};
export type AuthorizationWorkspaceUIAction =
| { type: 'hydrate'; value?: unknown }
| { type: 'patch'; value: Partial<AuthorizationWorkspaceUIState> }
| { type: 'workspace.initialize'; workspace: BrowserAuthorizationWorkspace }
| { type: 'workspace.updated'; workspace: BrowserAuthorizationWorkspace }
| { type: 'workspace.reset' }
| {
type: 'baselines.loaded';
candidates: Record<BrowserAuthorizationSide, BrowserAuthorizationBaselineCandidate[]>;
selected: Record<BrowserAuthorizationSide, string>;
}
| {
type: 'baselines.bound';
workspace: BrowserAuthorizationWorkspace;
selectedPlanCandidateId: string;
}
| { type: 'capture.replace'; capture: AuthorizationWorkspaceUIState['capture'] }
| { type: 'capture.update'; side: BrowserAuthorizationSide; status: NetworkCaptureStatus };
export type AuthorizationWorkspaceStage =
| 'identity'
| 'recovery'
| 'normal-requests'
| 'plan'
| 'execution'
| 'evidence';
export function authorizationWorkspaceStage(
state: AuthorizationWorkspaceUIState,
): AuthorizationWorkspaceStage {
const workspace = state.workspace;
if (!workspace) return 'identity';
if (workspace.state === 'stale' || workspace.state === 'blocked') return 'recovery';
if (!workspace.baselines.left || !workspace.baselines.right) return 'normal-requests';
if (!workspace.plan) return 'plan';
if (!workspace.execution) return 'execution';
return 'evidence';
}
function normalizedCandidates(
value: PersistedAuthorizationWorkspaceUI['candidates'] | undefined,
): PersistedAuthorizationWorkspaceUI['candidates'] {
return {
left: Array.isArray(value?.left) ? value.left : [],
right: Array.isArray(value?.right) ? value.right : [],
};
}
function record(value: unknown): Record<string, unknown> | undefined {
return value && typeof value === 'object' && !Array.isArray(value)
? value as Record<string, unknown>
: undefined;
}
function stringArray(value: unknown, max = 100): boolean {
return Array.isArray(value) && value.length <= max && value.every((item) => typeof item === 'string');
}
function safeWorkspaceForUI(input: unknown): BrowserAuthorizationWorkspace | undefined {
let workspace: BrowserAuthorizationWorkspace;
try {
workspace = normalizeBrowserAuthorizationTaskResult<BrowserAuthorizationWorkspace>(
'authorization.workspace.inspect',
input,
);
} catch {
return undefined;
}
const value = workspace as unknown as Record<string, unknown>;
const left = record(value.left);
const right = record(value.right);
const proof = record(value.proof);
const baselines = record(value.baselines);
const pair = record(value.baselinePair);
const validSide = (side: Record<string, unknown> | undefined) => {
const target = record(side?.target);
const authentication = record(side?.authentication);
return Boolean(side && target && authentication
&& Number.isSafeInteger(target.tabId) && Number(target.tabId) > 0
&& Number.isSafeInteger(target.frameId) && Number(target.frameId) >= 0
&& typeof target.documentId === 'string' && target.documentId
&& ['authenticated', 'unauthenticated', 'unknown'].includes(String(authentication.status))
&& Number.isFinite(authentication.cookieCount)
&& Number.isFinite(authentication.storageEntryCount));
};
if (value.version !== 1 || typeof value.id !== 'string' || !value.id
|| typeof value.engineInstanceId !== 'string' || !value.engineInstanceId
|| !['horizontal', 'vertical'].includes(String(value.mode))
|| !['ready', 'conditional', 'blocked', 'stale'].includes(String(value.state))
|| !Number.isFinite(value.expiresAt)
|| !validSide(left) || !validSide(right) || !proof || !baselines || !pair
|| !['strong', 'conditional', 'none'].includes(String(proof.level))
|| typeof proof.sameOrigin !== 'boolean'
|| !['different', 'same', 'unknown'].includes(String(proof.cookieStoreRelation))
|| !['different', 'same', 'unknown'].includes(String(proof.accountEvidenceRelation))
|| !['different', 'same', 'unknown'].includes(String(proof.requestCredentialRelation))
|| !['passed', 'failed', 'not-required'].includes(String(proof.refreshCheck))
|| !stringArray(proof.reasons)
|| !['waiting', 'matched', 'mismatch'].includes(String(pair.state))
|| !stringArray(pair.reasons)
|| !Array.isArray(pair.resourceCandidates) || !Array.isArray(pair.operationCandidates)) return undefined;
const resourceCandidatesValid = pair.resourceCandidates.every((item) => {
const candidate = record(item);
return Boolean(candidate && typeof candidate.id === 'string' && candidate.id
&& ['wire', 'logical'].includes(String(candidate.source))
&& ['header', 'path', 'query', 'body'].includes(String(candidate.location))
&& typeof candidate.path === 'string' && typeof candidate.category === 'string'
&& ['high', 'medium', 'low'].includes(String(candidate.confidence))
&& typeof candidate.requiresLogicalBinding === 'boolean'
&& stringArray(candidate.reasons));
});
const operationCandidatesValid = pair.operationCandidates.every((item) => {
const candidate = record(item);
return Boolean(candidate && typeof candidate.id === 'string' && candidate.id
&& typeof candidate.method === 'string' && typeof candidate.path === 'string'
&& typeof candidate.eligible === 'boolean' && typeof candidate.sideEffect === 'boolean'
&& typeof candidate.requiresDynamicRebuild === 'boolean'
&& stringArray(candidate.authenticationPaths) && stringArray(candidate.dynamicPaths)
&& stringArray(candidate.reasons));
});
if (!resourceCandidatesValid || !operationCandidatesValid) return undefined;
if (value.plan !== undefined) {
const plan = record(value.plan);
const selector = record(plan?.selector);
if (!plan || !selector || typeof plan.id !== 'string' || !plan.id
|| !['horizontal', 'vertical'].includes(String(plan.mode))
|| typeof plan.candidateId !== 'string'
|| !['ready', 'review-required', 'blocked'].includes(String(plan.state))
|| typeof selector.source !== 'string' || typeof selector.location !== 'string'
|| typeof selector.path !== 'string' || !Array.isArray(plan.cases)
|| !Number.isSafeInteger(plan.requestBudget) || Number(plan.requestBudget) < 0
|| typeof plan.requiresDynamicRebuild !== 'boolean' || !stringArray(plan.reasons)
|| !plan.cases.every((item) => {
const testCase = record(item);
return Boolean(testCase && typeof testCase.id === 'string' && typeof testCase.label === 'string'
&& ['left', 'right'].includes(String(testCase.authContextSide))
&& ['left', 'right', ''].includes(String(testCase.resourceValueSide))
&& typeof testCase.method === 'string' && typeof testCase.path === 'string'
&& typeof testCase.sideEffect === 'boolean');
})) return undefined;
}
if (value.execution !== undefined) {
const execution = record(value.execution);
if (!execution || typeof execution.id !== 'string' || !execution.id
|| !['completed', 'partial'].includes(String(execution.state))
|| !['confirmed', 'likely', 'protected', 'inconclusive', 'invalid-controls'].includes(String(execution.verdict))
|| !['high', 'medium', 'low', 'none'].includes(String(execution.confidence))
|| !Number.isSafeInteger(execution.requestCount) || Number(execution.requestCount) < 0
|| typeof execution.evidenceAvailable !== 'boolean'
|| !Array.isArray(execution.cases) || !Array.isArray(execution.evidence)
|| !stringArray(execution.reasons)
|| !execution.cases.every((item) => {
const testCase = record(item);
const result = record(testCase?.result);
return Boolean(testCase && typeof testCase.id === 'string' && typeof testCase.label === 'string'
&& ['completed', 'failed', 'skipped'].includes(String(testCase.state))
&& (!result || (Number.isFinite(result.status) && typeof result.statusText === 'string'
&& typeof result.outcome === 'string' && Number.isFinite(result.durationMs))));
})) return undefined;
}
return workspace;
}
function normalizePersistedCandidate(input: unknown): BrowserAuthorizationBaselineCandidate | undefined {
const candidate = record(input);
if (!candidate || typeof candidate.id !== 'string' || !candidate.id
|| typeof candidate.method !== 'string' || !candidate.method
|| typeof candidate.url !== 'string' || typeof candidate.path !== 'string'
|| typeof candidate.resourceType !== 'string' || !Number.isFinite(candidate.startedAt)
|| typeof candidate.eligible !== 'boolean' || !stringArray(candidate.reasons)) return undefined;
try {
const parsed = new URL(candidate.url);
if (!['http:', 'https:'].includes(parsed.protocol)) return undefined;
} catch {
return undefined;
}
return {
id: candidate.id.slice(0, 240),
method: candidate.method.slice(0, 32),
url: candidate.url.slice(0, 8_192),
path: candidate.path.slice(0, 4_096),
resourceType: candidate.resourceType.slice(0, 120),
startedAt: Number(candidate.startedAt),
completedAt: Number.isFinite(candidate.completedAt) ? Number(candidate.completedAt) : undefined,
durationMs: Number.isFinite(candidate.durationMs) ? Number(candidate.durationMs) : undefined,
statusCode: Number.isSafeInteger(candidate.statusCode) ? Number(candidate.statusCode) : undefined,
error: typeof candidate.error === 'string' ? candidate.error.slice(0, 1_024) : undefined,
eligible: candidate.eligible,
reasons: (candidate.reasons as string[]).slice(0, 20).map((item) => item.slice(0, 1_024)),
};
}
export function normalizePersistedAuthorizationWorkspaceUI(
input: unknown,
): Partial<PersistedAuthorizationWorkspaceUI> | undefined {
const value = record(input);
if (!value) return undefined;
const workspace = value.workspace === undefined ? undefined : safeWorkspaceForUI(value.workspace);
const candidateInput = record(value.candidates);
const candidates = workspace ? {
left: (Array.isArray(candidateInput?.left) ? candidateInput.left : [])
.slice(0, 50).map(normalizePersistedCandidate)
.filter((item): item is BrowserAuthorizationBaselineCandidate => Boolean(item)),
right: (Array.isArray(candidateInput?.right) ? candidateInput.right : [])
.slice(0, 50).map(normalizePersistedCandidate)
.filter((item): item is BrowserAuthorizationBaselineCandidate => Boolean(item)),
} : EMPTY_AUTHORIZATION_CANDIDATES;
const selectedInput = record(value.selected);
const selected = {
left: typeof selectedInput?.left === 'string'
&& candidates.left.some((item) => item.id === selectedInput.left) ? selectedInput.left : '',
right: typeof selectedInput?.right === 'string'
&& candidates.right.some((item) => item.id === selectedInput.right) ? selectedInput.right : '',
};
return {
mode: value.mode === 'vertical' ? 'vertical' : 'horizontal',
leftTabId: Number.isSafeInteger(value.leftTabId) && Number(value.leftTabId) > 0 ? Number(value.leftTabId) : undefined,
rightTabId: Number.isSafeInteger(value.rightTabId) && Number(value.rightTabId) > 0 ? Number(value.rightTabId) : undefined,
leftLabel: typeof value.leftLabel === 'string' ? value.leftLabel.slice(0, 80) : '账号 A',
rightLabel: typeof value.rightLabel === 'string' ? value.rightLabel.slice(0, 80) : '账号 B',
workspace,
candidates,
selected,
selectedPlanCandidateId: workspace && typeof value.selectedPlanCandidateId === 'string'
? value.selectedPlanCandidateId.slice(0, 240)
: '',
canaryPaths: typeof value.canaryPaths === 'string' ? value.canaryPaths.slice(0, 4_096) : '',
};
}
export function authorizationWorkspaceUIReducer(
state: AuthorizationWorkspaceUIState,
action: AuthorizationWorkspaceUIAction,
): AuthorizationWorkspaceUIState {
switch (action.type) {
case 'hydrate': {
const value = normalizePersistedAuthorizationWorkspaceUI(action.value);
if (!value) return state;
return {
...state,
mode: value.mode === 'vertical' ? 'vertical' : 'horizontal',
leftTabId: value.leftTabId,
rightTabId: value.rightTabId,
leftLabel: value.leftLabel || '账号 A',
rightLabel: value.rightLabel || '账号 B',
workspace: value.workspace,
candidates: normalizedCandidates(value.candidates),
selected: {
left: value.selected?.left || '',
right: value.selected?.right || '',
},
selectedPlanCandidateId: value.selectedPlanCandidateId || '',
canaryPaths: value.canaryPaths || '',
};
}
case 'patch': return { ...state, ...action.value };
case 'workspace.initialize':
return {
...state,
workspace: action.workspace,
candidates: EMPTY_AUTHORIZATION_CANDIDATES,
selected: EMPTY_SELECTION,
selectedPlanCandidateId: '',
};
case 'workspace.updated':
return { ...state, workspace: action.workspace };
case 'workspace.reset':
return {
...state,
workspace: undefined,
candidates: EMPTY_AUTHORIZATION_CANDIDATES,
selected: EMPTY_SELECTION,
selectedPlanCandidateId: '',
capture: {},
};
case 'baselines.loaded':
return {
...state,
candidates: action.candidates,
selected: action.selected,
};
case 'baselines.bound':
return {
...state,
workspace: action.workspace,
selectedPlanCandidateId: action.selectedPlanCandidateId,
};
case 'capture.replace':
return { ...state, capture: action.capture };
case 'capture.update':
return {
...state,
capture: { ...state.capture, [action.side]: action.status },
};
}
}
export function persistedAuthorizationWorkspaceUI(
state: AuthorizationWorkspaceUIState,
): PersistedAuthorizationWorkspaceUI {
return {
mode: state.mode,
leftTabId: state.leftTabId,
rightTabId: state.rightTabId,
leftLabel: state.leftLabel,
rightLabel: state.rightLabel,
workspace: state.workspace,
candidates: state.candidates,
selected: state.selected,
selectedPlanCandidateId: state.selectedPlanCandidateId,
canaryPaths: state.canaryPaths,
};
}
@@ -13,6 +13,7 @@ vi.mock('wxt/browser', () => {
const event = { addListener: vi.fn() };
return {
browser: {
storage: {},
tabs: { onRemoved: event, onCreated: event },
webNavigation: {
onBeforeNavigate: event,
@@ -33,6 +34,7 @@ import {
BROWSER_TRANSFORM_VALIDATION_DRAFT_MAX_BYTES,
compareBrowserPackets,
comparePacketWithInferenceCandidate,
discardBrowserTransformValidation,
inspectRecordingEvidence,
listRecordingTraces,
promoteObservedEnvelopeCallable,
@@ -75,6 +77,13 @@ function formCandidate(): BrowserProfileInferenceCandidate {
}
describe('browser analysis deterministic tools', () => {
it('rejects confirmation for a missing or expired validation draft', async () => {
await expect(discardBrowserTransformValidation(
{ tabId: 1, frameId: 0, documentId: 'document-1' },
'validation-missing',
)).rejects.toThrow(/不存在或已经过期/);
});
it('bounds validation drafts before session persistence', () => {
const draft = {
contractVersion: 1,
+256 -14
View File
@@ -1,6 +1,7 @@
import { browser } from 'wxt/browser';
import * as v from 'valibot';
import {
createRecordedPageCallable,
getBrowserRecording,
} from '@/features/browser-recording/service';
import { recordingSnapshotForScope } from '@/features/browser-recording/redaction';
@@ -8,12 +9,20 @@ import {
compileGuidedTransform,
parseGuidedTransform,
} from '@/features/browser-transform/guided';
import { createBrowserTransformProfileInput } from '@/features/browser-transform/profile-draft';
import {
createBrowserTransformProfileInput,
pairedBrowserTransformCandidate,
} from '@/features/browser-transform/profile-draft';
import { validateBrowserTransformProfile } from '@/features/browser-transform/service';
import { executePageCallable, listPageCallables } from '@/features/page-callable/service';
import { browserTransformProfileInputSchema } from '@/protocol/transform';
import { ExtensionError } from '@/shared/errors';
import { createOpaqueId } from '@/shared/id';
import { eventMatcher } from '@/features/deep-capture/matcher';
import { createCapturedPageCallable, deepCaptureStatus, detachDeepCapture, startDeepCapture, type DeepCaptureOwner } from '@/features/deep-capture/service';
import { withPageNetworkGuard } from '@/features/page-callable/network-guard';
import { actOnPageNode, capturePageContext } from '@/features/page-context/service';
import { beginPageDialogCapture, endPageDialogCapture } from '@/features/page-context/dialogs';
import {
BROWSER_TRANSFORM_AGENT_CONTRACT_VERSION,
type ActiveTabInfo,
@@ -34,6 +43,8 @@ import {
type BrowserTransformProfileProposalResult,
type BrowserTransformProfileValidationResult,
type BrowserTransformValidationDraft,
type BrowserTransformDirectionName,
type BrowserDeepCaptureMatcher,
} from '@/types/models';
const MAX_TRACE_EVENTS = 80;
@@ -65,6 +76,8 @@ let callableOutputStorageQueue: Promise<void> = Promise.resolve();
interface BrowserProfileEvidenceReference {
candidate: BrowserProfileInferenceCandidate;
requestEvent?: BrowserRecordingEvent;
matcher?: BrowserDeepCaptureMatcher;
triggerKey?: string;
createdAt: number;
expiresAt: number;
}
@@ -134,7 +147,7 @@ function pruneProfileEvidence(
);
}
export async function stageBrowserProfileEvidence(snapshot: BrowserRecordingSnapshot): Promise<void> {
export async function stageBrowserProfileEvidence(snapshot: BrowserRecordingSnapshot, triggerKey?: string): Promise<void> {
if (!snapshot.profileCandidates.length) return;
const scoped = recordingSnapshotForScope(snapshot, false);
const events = new Map(scoped.events.map((event) => [event.id, event]));
@@ -145,6 +158,8 @@ export async function stageBrowserProfileEvidence(snapshot: BrowserRecordingSnap
stored[candidate.id] = {
candidate,
requestEvent: events.get(candidate.request.eventId),
matcher: eventMatcher(events.get(candidate.capturePlan?.matcherEventId || candidate.source.eventId), candidate),
triggerKey: triggerKey || stored[candidate.id]?.triggerKey,
createdAt,
expiresAt: createdAt + PROFILE_EVIDENCE_TTL_MS,
};
@@ -192,12 +207,14 @@ export async function resolveBrowserProfileCaptureTransaction(
}
function callableAnalysis(candidate: BrowserProfileInferenceCandidate): BrowserTransformCallableAnalysis {
const sources = [candidate.source, ...candidate.sources]
.filter((source, index, items) => items.findIndex((item) => item.eventId === source.eventId) === index);
return {
version: 1,
traceId: candidate.traceId,
confidence: { ...candidate.confidence },
flow: candidate.flow.slice(0, 32),
operations: candidate.sources.slice(0, 16).map((source) => ({
operations: sources.slice(0, 16).map((source) => ({
operation: source.operation,
destination: source.destination,
crypto: source.crypto ? structuredClone(source.crypto) : undefined,
@@ -221,6 +238,13 @@ async function resolveStagedProfileCandidate(
return reference.candidate;
}
async function resolvePairedStagedProfileCandidate(
candidate: BrowserProfileInferenceCandidate,
): Promise<BrowserProfileInferenceCandidate | undefined> {
const references = Object.values(pruneProfileEvidence(await readStoredProfileEvidence()));
return pairedBrowserTransformCandidate(references.map((item) => item.candidate), candidate, true);
}
export async function resolveBrowserProfileCallableAnalysis(
target: BrowserTarget,
candidateId: string,
@@ -478,6 +502,57 @@ export async function latestBrowserTransformValidation(
return draft || memoryValidationDrafts.get(key) || null;
}
export async function browserTransformValidationById(
validationId: string,
): Promise<BrowserTransformValidationDraft> {
const now = Date.now();
const stored = await readStoredValidationDrafts();
const drafts = pruneValidationDrafts(stored, now);
if (Object.keys(drafts).length !== Object.keys(stored).length) {
validationDraftStorageQueue = validationDraftStorageQueue.then(() => writeStoredValidationDrafts(drafts));
await validationDraftStorageQueue;
}
const draft = Object.values(drafts).find((item) => item.id === validationId)
|| [...memoryValidationDrafts.values()].find((item) => item.id === validationId && item.expiresAt > now);
if (!draft) {
throw new ExtensionError('validation_draft_stale', '验证草稿不存在或已经过期,请重新生成并验证');
}
return draft;
}
export async function executeBrowserTransformValidation(
validationId: string,
direction: BrowserTransformDirectionName,
packet: BrowserTransformPacket,
): Promise<BrowserTransformExecution> {
const draft = await browserTransformValidationById(validationId);
const { profile, execution } = await validateBrowserTransformProfile(draft.profile, packet, {
direction,
profileId: `transient-${validationId}`,
});
return { ...execution, explanation: profile.explanation, proofLevel: draft.proofLevel };
}
export async function discardBrowserTransformValidation(
target: BrowserTarget,
validationId: string,
): Promise<void> {
const key = validationDraftKey(target);
let discarded = false;
validationDraftStorageQueue = validationDraftStorageQueue.then(async () => {
const drafts = pruneValidationDrafts(await readStoredValidationDrafts());
if (drafts[key]?.id === validationId) {
delete drafts[key];
discarded = true;
}
await writeStoredValidationDrafts(drafts);
});
await validationDraftStorageQueue;
if (!discarded) {
throw new ExtensionError('validation_draft_stale', '验证草稿不存在或已经过期,请重新生成并验证');
}
}
function formValueType(value: string): string {
if (!value) return 'empty';
const trimmed = value.trim();
@@ -781,7 +856,7 @@ export function comparePacketWithInferenceCandidate(
check(
checks,
'body-shape',
'已关联的线上字段存在且没有二次 JSON 包装',
'线上字段结构一致(不验证加密前的输入内容)',
bodyFieldsPresent,
actualShape.signature,
bodyFields,
@@ -1019,12 +1094,28 @@ function originOf(value: string): string {
}
}
function callableMatchesCandidate(
callable: BrowserPageCallable,
candidate: BrowserProfileInferenceCandidate,
): boolean {
const sources = [candidate.source, ...candidate.sources]
.filter((source, index, items) => items.findIndex((item) => item.eventId === source.eventId) === index);
const sourceEventIds = new Set(sources.map((source) => source.eventId));
if (callable.provenance.eventId) return sourceEventIds.has(callable.provenance.eventId);
const analysis = callable.provenance.analysis;
return analysis?.traceId === candidate.traceId && sources.every((source) => (
analysis.operations.some((operation) => operation.operation === source.operation
&& operation.destination === source.destination)
));
}
export async function proposeBrowserTransformProfile(
target: BrowserTarget,
candidateId: string,
callableId: string,
inputPaths?: string[],
name?: string,
packet?: BrowserTransformPacket,
): Promise<BrowserTransformProfileProposalResult> {
const [snapshot, callables, tab, frame] = await Promise.all([
getBrowserRecording(target, 500, false),
@@ -1034,8 +1125,41 @@ export async function proposeBrowserTransformProfile(
]);
const evidence = await resolveProfileEvidence(snapshot, target, candidateId);
const candidate = evidence.candidate;
const pairedCandidate = pairedBrowserTransformCandidate(snapshot.profileCandidates, candidate, true)
|| await resolvePairedStagedProfileCandidate(candidate);
const recordedCallable = callables.find((item) => item.id === callableId);
if (!recordedCallable) throw new ExtensionError('callable_unavailable', `页面函数已经失效: ${callableId}`);
if (candidate.status !== 'ready' && recordedCallable.kind === 'recorded-call') {
throw new ExtensionError(
'gateway_capture_required',
candidate.missing[0]?.label || '当前候选依赖完整业务流程,不能直接复用单个页面加解密调用',
);
}
const recordedPairedCallable = pairedCandidate
? callables.find((item) => item.id !== recordedCallable.id && callableMatchesCandidate(item, pairedCandidate))
: undefined;
if (pairedCandidate && pairedCandidate.status !== 'ready' && recordedPairedCallable?.kind === 'recorded-call') {
throw new ExtensionError(
'gateway_capture_required',
pairedCandidate.missing[0]?.label || `配对的${pairedCandidate.direction === 'request' ? '请求' : '响应'}方向尚未完成捕获`,
);
}
if (pairedCandidate && !recordedPairedCallable) {
throw new ExtensionError(
'gateway_pair_incomplete',
`已经识别同一事务的${pairedCandidate.direction === 'request' ? '请求' : '响应'}方向,但对应页面函数尚未创建`,
);
}
const pairedObservation = pairedCandidate?.direction === 'request' && recordedPairedCallable
? await callableOutputObservation(recordedPairedCallable)
: undefined;
const pairedCallable = pairedObservation && recordedPairedCallable
? promoteObservedEnvelopeCallable(
recordedPairedCallable,
snapshot.events.find((item) => item.id === pairedCandidate!.request.eventId),
pairedObservation.objectKeys,
)
: recordedPairedCallable;
const observation = candidate.direction === 'request'
? await callableOutputObservation(recordedCallable)
: undefined;
@@ -1045,7 +1169,7 @@ export async function proposeBrowserTransformProfile(
if (!sameEvidenceTarget(candidate, target)) {
throw new ExtensionError('target_denied', '自动推断候选不属于当前共享页面');
}
if (callable.provenance.traceId && callable.provenance.traceId !== candidate.traceId) {
if (!callableMatchesCandidate(callable, candidate)) {
throw new ExtensionError('profile_evidence_mismatch', '页面函数与自动推断候选不属于同一条业务 Trace');
}
const pageUrl = frame?.url || callable.origin;
@@ -1062,7 +1186,14 @@ export async function proposeBrowserTransformProfile(
lastAccessed: tab.lastAccessed,
};
const requestEvent = evidence.requestEvent;
let profile = createBrowserTransformProfileInput(tabInfo, requestEvent, callable, candidate);
let profile = createBrowserTransformProfileInput(
tabInfo,
requestEvent,
callable,
candidate,
inputPaths ? undefined : packet,
pairedCandidate && pairedCallable ? { candidate: pairedCandidate, callable: pairedCallable } : undefined,
);
profile = {
...profile,
name: name || profile.name,
@@ -1109,7 +1240,7 @@ export async function proposeBrowserTransformProfile(
? callable.transaction ? 'captured-request-transaction' : 'validated-callable-envelope'
: 'recording-evidence',
},
next: '调用 profile.validate;验证成功后由用户确认保存,AI 不直接持久化配置',
next: '调用 profile.validate;验证成功后可立即用 validationDraft.id 做一次临时明文 HTTP 测试,只有复用配置才需要用户在插件中确认保存',
};
}
@@ -1139,15 +1270,17 @@ export async function validateBrowserTransformProposal(
comparisonMode: 'structure' | 'exact' = 'structure',
candidateId?: string,
): Promise<BrowserTransformProfileValidationResult> {
const { profile: normalized, execution } = await validateBrowserTransformProfile(profile, packet);
const generated = applyTransformExecution(packet, execution);
const candidate = candidateId
? (await resolveProfileEvidence(
await getBrowserRecording(normalized.target, 500, false),
normalized.target,
await getBrowserRecording(profile.target, 500, false),
profile.target,
candidateId,
)).candidate
: undefined;
const { profile: normalized, execution } = await validateBrowserTransformProfile(profile, packet, {
direction: candidate?.direction,
});
const generated = applyTransformExecution(packet, execution);
if (candidate && !sameEvidenceTarget(candidate, normalized.target)) {
throw new ExtensionError('profile_evidence_mismatch', '验证候选不属于明文网关绑定的页面');
}
@@ -1176,19 +1309,22 @@ export async function validateBrowserTransformProposal(
proofLevel,
normalizedProfile,
generated,
execution,
execution: { ...execution, explanation: normalized.explanation },
comparison,
validationDraft: validationDraft ? {
contractVersion: validationDraft.contractVersion,
id: validationDraft.id,
createdAt: validationDraft.createdAt,
expiresAt: validationDraft.expiresAt,
directions: { request: normalizedProfile.request.enabled, response: normalizedProfile.response.enabled },
} : undefined,
next: comparison
? comparison.equivalent
? '确定性验证通过;Yakit 已收到待用户确认的明文网关草稿'
? comparison.mode === 'exact'
? '样本报文对比通过;尚未发送业务测试请求,使用 validationDraft.id 调用 browser.http.test'
: '仅结构校验通过,不证明明文输入、加密语义或业务成功;使用 validationDraft.id 调用 browser.http.test 验证,请勿原样重复 prepare'
: '数据包对比未通过;检查输入映射或重新选择页面函数'
: 'Pipeline 已真实回放并生成待确认草稿;如需更强证明,请提供一份浏览器线上请求进行结构对比',
: 'Pipeline 已真实回放;可将 validationDraft.id 直接交给 browser.http.test。如需更强证明,请提供一份浏览器线上请求进行结构对比',
};
}
@@ -1208,6 +1344,7 @@ export async function validateInferredBrowserTransformProfile(
callableId,
inputPaths,
name,
packet,
);
return validateBrowserTransformProposal(
proposal.profile,
@@ -1217,3 +1354,108 @@ export async function validateInferredBrowserTransformProfile(
candidateId,
);
}
const preparingTabs = new Set<number>();
interface PreparationCapture {
owner: DeepCaptureOwner;
trigger?: { captureId: string; nodeId: string };
authorize(): void;
}
async function captureMissingProfileCallable(
target: BrowserTarget,
candidate: BrowserProfileInferenceCandidate,
options?: PreparationCapture,
): Promise<BrowserPageCallable> {
if (candidate.status !== 'capture-required' || !options) {
throw new ExtensionError('gateway_capture_required', candidate.missing[0]?.label || '候选缺少可安全捕获的业务边界');
}
options.authorize();
const reference = pruneProfileEvidence(await readStoredProfileEvidence())[candidate.id];
if (!reference?.matcher) throw new ExtensionError('gateway_capture_required', '候选缺少断点证据,请重新执行 browser.crypto.inspect');
let trigger = options.trigger;
if (!trigger && reference.triggerKey) {
const context = await capturePageContext({ includeDom: true }, target);
const nodes = context.document.interactive.filter((node) => node.semanticKey === reference.triggerKey);
if (nodes.length === 1) trigger = { captureId: context.captureId, nodeId: nodes[0].nodeId };
}
if (!trigger) throw new ExtensionError('gateway_trigger_required', '无法唯一定位原操作,请用 browser.context 获取新的触发节点,并向 browser.transform.prepare 传入 trigger');
const current = await deepCaptureStatus(target, options.owner);
if (['attached', 'armed', 'paused'].includes(current.state)) throw new ExtensionError('capture_busy', '当前页面已有深度捕获会话,请先完成或释放它');
const transaction = candidate.direction === 'request' ? (await resolveBrowserProfileCaptureContext(target, candidate.id)).transaction : undefined;
return withPageNetworkGuard(target, transaction?.prerequisites || [], async () => {
const dialogOwned = await beginPageDialogCapture(target);
let started = false;
try {
await startDeepCapture(target, reference.matcher!, options.owner);
started = true;
await actOnPageNode(trigger!.captureId, trigger!.nodeId, 'click', target);
const deadline = Math.min(Date.now() + 15_000, options.owner.kind === 'grant' ? options.owner.expiresAt : Infinity);
while (Date.now() < deadline) {
const status = await deepCaptureStatus(target, options.owner);
if (status.state === 'paused' && status.pause && !status.pause.collecting) {
const automatic = status.pause.automaticCapture;
if (automatic?.state !== 'ready' || !automatic.frameId) {
throw new ExtensionError('gateway_capture_ambiguous', automatic?.reason || '暂停现场没有唯一可复用的业务函数', { automaticCapture: automatic });
}
const analysis = callableAnalysis(candidate);
if (automatic.strategy === 'request-transaction') {
if (!transaction) throw new ExtensionError('gateway_capture_required', '该方向没有请求事务证据');
return await createCapturedPageCallable(target, automatic.frameId, { strategy: 'request-transaction', transaction, analysis }, options.owner);
}
return await createCapturedPageCallable(target, automatic.frameId, { strategy: 'selected-frame', analysis }, options.owner);
}
if (!['armed', 'paused', 'attached'].includes(status.state)) throw new ExtensionError('gateway_capture_failed', status.error || '捕获会话已结束,未获取业务函数');
await new Promise((resolve) => setTimeout(resolve, 50));
}
throw new ExtensionError('gateway_capture_timeout', '重触发操作后未命中预期业务边界,请检查当前页面状态');
} finally {
try { if (started) await detachDeepCapture(target, options.owner); }
finally { await endPageDialogCapture(target, dialogOwned); }
}
}, candidate.direction === 'request' ? candidate.request.url : undefined);
}
export async function prepareCapturedBrowserTransformProfile(
target: BrowserTarget,
candidateId: string,
packet: BrowserTransformPacket,
inputPaths?: string[],
name?: string,
capture?: PreparationCapture,
): Promise<BrowserTransformProfileValidationResult> {
if (preparingTabs.has(target.tabId)) throw new ExtensionError('capture_busy', '当前标签页正在准备网关,请等待该操作完成');
preparingTabs.add(target.tabId);
try {
const candidate = await resolveStagedProfileCandidate(target, candidateId);
const pairedCandidate = await resolvePairedStagedProfileCandidate(candidate);
const existingCallables = await listPageCallables(target);
const ensureCallable = async (item: BrowserProfileInferenceCandidate) => {
const source = [item.source, ...item.sources].find((value) => value.callHandleId) || item.source;
const existing = existingCallables.find((callable) => callableMatchesCandidate(callable, item)
&& (item.status === 'ready'
? callable.inputSlots.filter((slot) => !slot.retained).length === (source.dynamicInputPaths?.length || 1)
: callable.kind !== 'recorded-call'));
if (existing) return existing;
if (item.status !== 'ready') return captureMissingProfileCallable(target, item, capture);
if (!source.callHandleId) throw new ExtensionError('gateway_capture_required', '候选缺少可复用的页面调用句柄,请重新执行 browser.crypto.inspect');
return createRecordedPageCallable(target, {
callHandleId: source.callHandleId,
name: name || item.summary.slice(0, 120) || 'Captured page transform',
dynamicInputPaths: source.dynamicInputPaths,
});
};
// Retain an already-recorded opposite direction before re-triggering the page.
const directions = [candidate, pairedCandidate].filter((item): item is BrowserProfileInferenceCandidate => Boolean(item))
.sort((left, right) => Number(right.status === 'ready') - Number(left.status === 'ready'));
let callable: BrowserPageCallable | undefined;
for (const direction of directions) {
const created = await ensureCallable(direction);
if (direction.id === candidate.id) callable = created;
}
return await validateInferredBrowserTransformProfile(target, candidate.id, callable!.id, packet, inputPaths, name);
} finally {
preparingTabs.delete(target.tabId);
}
}
@@ -91,11 +91,15 @@ describe('page crypto adapters', () => {
const CBC = {};
const Pkcs7 = {};
const parsed: string[] = [];
const hexParsed: string[] = [];
const cryptoJs = {
AES: { encrypt() { return 'cipher'; } },
AES: { encrypt() { return 'cipher'; }, decrypt() { return 'plain'; } },
mode: { CBC },
pad: { Pkcs7 },
enc: { Base64: { parse(value: string) { parsed.push(value); return { wordArray: value }; } } },
enc: {
Base64: { parse(value: string) { parsed.push(value); return { wordArray: value }; } },
Hex: { parse(value: string) { hexParsed.push(value); return { hexWordArray: value }; } },
},
};
const scope = { window: { CryptoJS: cryptoJs } as unknown as Window } satisfies CryptoAdapterScope;
const encrypt = cryptoJsAdapter.discover(scope).find((item) => item.operation === 'AES.encrypt');
@@ -117,6 +121,28 @@ describe('page crypto adapters', () => {
]);
expect(plan?.adaptInput?.(new Uint8Array([4, 5, 6]))).toEqual({ wordArray: 'base64:4,5,6' });
expect(parsed).toEqual(['base64:4,5,6']);
const decrypt = cryptoJsAdapter.discover(scope).find((item) => item.operation === 'AES.decrypt');
const decryptPlan = decrypt?.describe(cryptoJs.AES, [
'cipher',
{ sigBytes: 16 },
{ mode: CBC, padding: Pkcs7, iv: { sigBytes: 16 } },
], toolkit());
expect(decryptPlan?.crypto.outputEncoding).toBe('hex');
expect(decryptPlan?.outputEncoding).toBe('hex');
expect(decryptPlan?.replayInputs?.map((input) => input.path)).toEqual([
'$input', '$input.key', '$input.iv',
]);
const replayArgs: unknown[] = ['old-cipher', { oldKey: true }, { mode: CBC, padding: Pkcs7, iv: { oldIv: true } }];
decryptPlan?.replayInputs?.[0].apply(replayArgs, 'new-cipher');
decryptPlan?.replayInputs?.[1].apply(replayArgs, '00112233');
decryptPlan?.replayInputs?.[2].apply(replayArgs, 'aabbccdd');
expect(replayArgs).toEqual([
'new-cipher',
{ hexWordArray: '00112233' },
{ mode: CBC, padding: Pkcs7, iv: { hexWordArray: 'aabbccdd' } },
]);
expect(hexParsed).toEqual(['00112233', 'aabbccdd']);
});
it('retains only bounded JSEncrypt receiver metadata', () => {
@@ -47,6 +47,13 @@ export interface CryptoAdapterInvocationPlan {
arguments: BrowserRecordingCallArgument[];
callableKind?: CallableOperationKind;
outputEncoding?: BrowserPageCallableValueEncoding;
replayInputs?: Array<{
path: string;
name: string;
role: BrowserRecordingCallArgument['role'];
originalInput: unknown;
apply(args: unknown[], value: unknown): void;
}>;
inputEvidence?(value: unknown): BrowserRecordingValueEvidence[];
outputEvidence?(value: unknown): BrowserRecordingValueEvidence[];
outputError?(value: unknown): string | undefined;
@@ -60,6 +60,8 @@ function describe(
const cryptoJs = (scope.window as unknown as { CryptoJS?: Record<string, unknown> }).CryptoJS || {};
const normalized = path.toLowerCase();
const encrypting = normalized.includes('encrypt');
const decrypting = normalized.includes('decrypt');
const outputEncoding = encrypting ? 'base64' : decrypting ? 'hex' : 'auto';
const options = normalized.includes('encrypt') || normalized.includes('decrypt')
? optionsMetadata(cryptoJs, args[2], toolkit)
: {};
@@ -68,6 +70,33 @@ function describe(
else if (normalized.includes('pbkdf2') || normalized.includes('evpkdf')) roles = ['data', 'salt', 'options'];
else if (normalized.includes('.encrypt') || normalized.includes('.decrypt')) roles = ['data', 'key', 'options'];
const callableKind = callableOperationKind(path);
const adaptData = (value: unknown) => {
const originalInput = args[0];
if (originalInput && typeof originalInput === 'object'
&& typeof (originalInput as { sigBytes?: unknown }).sigBytes === 'number') {
const bytes = toolkit.bytesForInput(value);
const encoder = (cryptoJs as { enc?: { Base64?: { parse?(input: string): unknown } } }).enc?.Base64;
if (bytes && typeof encoder?.parse === 'function') return encoder.parse(toolkit.bytesToBase64(bytes));
}
return toolkit.defaultAdaptInput(value, originalInput);
};
const adaptWordArray = (value: unknown, originalInput: unknown) => {
if (!originalInput || typeof originalInput !== 'object'
|| typeof (originalInput as { sigBytes?: unknown }).sigBytes !== 'number') {
return toolkit.defaultAdaptInput(value, originalInput);
}
const enc = (cryptoJs as {
enc?: {
Hex?: { parse?(input: string): unknown };
Base64?: { parse?(input: string): unknown };
};
}).enc;
if (typeof value === 'string' && /^[0-9a-f]+$/i.test(value) && value.length % 2 === 0
&& typeof enc?.Hex?.parse === 'function') return enc.Hex.parse(value);
const bytes = toolkit.bytesForInput(value);
if (bytes && typeof enc?.Base64?.parse === 'function') return enc.Base64.parse(toolkit.bytesToBase64(bytes));
return toolkit.defaultAdaptInput(value, originalInput);
};
return {
crypto: {
adapterId: cryptoJsManifest.id,
@@ -78,12 +107,31 @@ function describe(
mode: options.mode,
padding: options.padding,
inputEncoding: 'auto',
outputEncoding: encrypting ? 'base64' : 'auto',
outputEncoding,
state: { model: 'stateless', phase: 'one-shot' },
},
inputIndex: 0,
callableKind,
outputEncoding: encrypting ? 'base64' : 'auto',
outputEncoding,
replayInputs: decrypting ? [
{
path: '$input', name: 'data', role: 'data', originalInput: args[0],
apply: (nextArgs, value) => { nextArgs[0] = adaptData(value); },
},
{
path: '$input.key', name: 'key', role: 'key', originalInput: args[1],
apply: (nextArgs, value) => { nextArgs[1] = adaptWordArray(value, args[1]); },
},
{
path: '$input.iv', name: 'iv', role: 'iv', originalInput: ownValue(args[2], 'iv'),
apply: (nextArgs, value) => {
const options = nextArgs[2] && typeof nextArgs[2] === 'object'
? nextArgs[2] as Record<string, unknown>
: {};
nextArgs[2] = { ...options, iv: adaptWordArray(value, ownValue(args[2], 'iv')) };
},
},
] : undefined,
arguments: args.slice(0, 8).map((value, index) => toolkit.argument(
index,
roles[index] || 'unknown',
@@ -121,14 +169,7 @@ function describe(
return output.slice(0, 48);
},
adaptInput(value) {
const originalInput = args[0];
if (originalInput && typeof originalInput === 'object'
&& typeof (originalInput as { sigBytes?: unknown }).sigBytes === 'number') {
const bytes = toolkit.bytesForInput(value);
const encoder = (cryptoJs as { enc?: { Base64?: { parse?(input: string): unknown } } }).enc?.Base64;
if (bytes && typeof encoder?.parse === 'function') return encoder.parse(toolkit.bytesToBase64(bytes));
}
return toolkit.defaultAdaptInput(value, originalInput);
return adaptData(value);
},
};
}
+206
View File
@@ -0,0 +1,206 @@
import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
const fixture = vi.hoisted(() => ({
executeScript: vi.fn(),
startRecording: vi.fn(),
stopRecording: vi.fn(),
startNetwork: vi.fn(),
listNetwork: vi.fn(),
stopNetwork: vi.fn(),
act: vi.fn(),
context: vi.fn(),
stageEvidence: vi.fn(),
}));
vi.mock('wxt/browser', () => ({
browser: { scripting: { executeScript: fixture.executeScript } },
}));
vi.mock('@/features/browser-recording/service', () => ({
startBrowserRecording: fixture.startRecording,
stopBrowserRecording: fixture.stopRecording,
}));
vi.mock('@/features/network-capture/service', () => ({
startNetworkCapture: fixture.startNetwork,
listNetworkRequests: fixture.listNetwork,
stopNetworkCapture: fixture.stopNetwork,
}));
vi.mock('@/features/page-context/service', () => ({
actOnPageNode: fixture.act,
capturePageContext: fixture.context,
}));
vi.mock('@/features/browser-analysis/service', () => ({
listRecordingTraces: vi.fn(() => [{ id: 'trace-1', cryptoCount: 1 }]),
stageBrowserProfileEvidence: fixture.stageEvidence,
}));
vi.mock('@/platform/browser/targets', () => ({
scriptingTarget: vi.fn((target) => ({ tabId: target.tabId, documentIds: [target.documentId] })),
}));
describe('atomic page crypto inspection', () => {
beforeEach(() => {
vi.clearAllMocks();
fixture.executeScript
.mockResolvedValueOnce([{ result: true }])
.mockResolvedValueOnce([{ result: [{ type: 'alert', message: 'done', decision: 'auto_dismissed', timestamp: 2 }] }]);
fixture.startRecording.mockResolvedValue({});
fixture.stopRecording.mockResolvedValue({
status: { target: { tabId: 7, frameId: 0, documentId: 'doc-1' }, active: false, documentAvailable: true, count: 1, droppedCount: 0 },
events: [{
id: 'event-1', sequence: 1, timestamp: 1, recordingId: 'recording-1', traceId: 'trace-1',
kind: 'crypto', operation: 'AES.encrypt',
crypto: { adapterId: 'cryptojs', providerKind: 'library', family: 'symmetric', operation: 'AES.encrypt', mode: 'CBC', padding: 'Pkcs7' },
inputs: [], outputs: [], sensitiveCaptured: true,
}],
traces: [], links: [], callables: [], profileCandidates: [{
id: 'candidate-1', direction: 'request', summary: 'login request',
status: 'ready',
confidence: { score: 0.95, level: 'high' },
source: { eventId: 'event-1', callHandleId: 'handle-1' },
sources: [],
request: {
method: 'POST', url: 'https://example.test/api', bodyFormat: 'json',
mappings: [{ sourceEventId: 'event-1', destination: '$body.username' }],
},
}],
});
fixture.startNetwork.mockResolvedValue({});
fixture.listNetwork.mockResolvedValue([{
id: 'request-1', requestId: 'devtools-1', tabId: 7, frameId: 0,
url: 'https://example.test/api', method: 'POST', resourceType: 'xmlhttprequest',
startedAt: 1, completedAt: 2, statusCode: 200,
requestHeadersCaptured: false, requestBodyCaptured: true,
requestBody: { encoding: 'utf8', data: '{"cipher":"abc"}', byteLength: 16, truncated: false },
redirects: [],
}]);
fixture.stopNetwork.mockResolvedValue({});
fixture.act.mockResolvedValue({ action: 'click', status: 'dispatched', dispatchedAt: 1, node: { nodeId: 'n1' } });
fixture.context.mockResolvedValue({
captureId: 'capture-2',
target: { tabId: 7, frameId: 0, documentId: 'doc-1' },
authentication: { state: 'authenticated' },
document: {
title: 'Crypto lab', url: 'https://example.test/', readyState: 'complete', forms: [],
interactive: [{ nodeId: 'n2', role: 'button', name: 'Next operation', visible: true }],
},
});
});
afterEach(() => vi.useRealTimers());
it('captures one click, crypto evidence, request, and modal dialog in one call', async () => {
const { inspectPageCryptoOperation } = await import('./inspect');
const result = await inspectPageCryptoOperation(
{ tabId: 7, frameId: 0, documentId: 'doc-1' },
{ captureId: 'capture-1', nodeId: 'n1', settleMs: 250 },
{ grantId: 'paired', expiresAt: Date.now() + 60_000 },
);
expect(result).toMatchObject({
state: 'observed',
dialogs: [{ type: 'alert', message: 'done', decision: 'auto_dismissed' }],
dialogHandling: { autoDismissedAlerts: 1, autoAcceptedConfirms: 0, autoSubmittedPrompts: 0, count: 1, navigationInferred: false },
postAction: {
sameDocument: true,
captureId: 'capture-2',
document: { interactive: [{ nodeId: 'n2', name: 'Next operation' }] },
},
recording: { count: 1, events: [{ kind: 'crypto', operation: 'AES.encrypt' }] },
network: { count: 1, requests: [{ method: 'POST', statusCode: 200 }] },
gatewayPreparation: {
state: 'ready', candidateId: 'candidate-1',
request: { method: 'POST', destinations: ['$body.username'] },
},
});
expect(fixture.act).toHaveBeenCalledOnce();
expect(fixture.stopRecording).toHaveBeenCalledOnce();
expect(fixture.stopNetwork).toHaveBeenCalledOnce();
expect(fixture.stageEvidence).toHaveBeenCalledOnce();
expect(fixture.context).toHaveBeenCalledWith({ includeDom: true }, { tabId: 7, frameId: 0 });
});
it('prepares a response-only protocol when no request transform was observed', async () => {
fixture.stopRecording.mockResolvedValueOnce({
status: { target: { tabId: 7, frameId: 0, documentId: 'doc-1' }, active: false, documentAvailable: true, count: 1, droppedCount: 0 },
events: [{
id: 'decrypt-1', sequence: 1, timestamp: 1, recordingId: 'recording-1', traceId: 'trace-1',
kind: 'crypto', operation: 'AES.decrypt', inputs: [], outputs: [], sensitiveCaptured: true,
}],
traces: [], links: [], callables: [], profileCandidates: [{
id: 'candidate-response', recordingId: 'recording-1', traceId: 'trace-1', direction: 'response',
status: 'ready', confidence: { score: 100, level: 'high' },
source: { eventId: 'decrypt-1', callHandleId: 'handle-1' }, sources: [],
request: { method: 'POST', url: 'https://example.test/api', bodyFormat: 'json', mappings: [] },
}],
});
const { inspectPageCryptoOperation } = await import('./inspect');
const result = await inspectPageCryptoOperation(
{ tabId: 7, frameId: 0, documentId: 'doc-1' },
{ captureId: 'capture-1', nodeId: 'n1', settleMs: 250 },
{ grantId: 'paired', expiresAt: Date.now() + 60_000 },
);
expect(result.gatewayPreparation).toMatchObject({
state: 'ready',
direction: 'response',
directions: { request: { status: 'absent' }, response: { candidateId: 'candidate-response', status: 'ready' } },
});
});
it('waits for a delayed request instead of treating an empty capture as idle', async () => {
vi.useFakeTimers();
const startedAt = Date.now();
const delayedRequest = {
id: 'request-delayed', requestId: 'devtools-delayed', tabId: 7, frameId: 2,
url: 'https://example.test/delayed', method: 'POST', resourceType: 'xmlhttprequest',
startedAt: 1, completedAt: 2, statusCode: 200,
requestHeadersCaptured: false, requestBodyCaptured: false, redirects: [],
};
fixture.listNetwork.mockImplementation(async () => (Date.now() - startedAt >= 900 ? [delayedRequest] : []));
const { inspectPageCryptoOperation } = await import('./inspect');
let settled = false;
const pending = inspectPageCryptoOperation(
{ tabId: 7, frameId: 2, documentId: 'doc-frame' },
{ captureId: 'capture-1', nodeId: 'n1', settleMs: 1_500 },
{ grantId: 'paired', expiresAt: Date.now() + 60_000 },
).finally(() => { settled = true; });
await vi.advanceTimersByTimeAsync(800);
expect(settled).toBe(false);
await vi.advanceTimersByTimeAsync(800);
const result = await pending;
expect(result).toMatchObject({ network: { count: 1, requests: [{ id: 'request-delayed' }] } });
expect(fixture.context).toHaveBeenCalledWith({ includeDom: true }, { tabId: 7, frameId: 2 });
});
it('handles alert, confirm, and prompt without blocking the page', async () => {
const originalAlert = globalThis.alert;
const originalConfirm = globalThis.confirm;
const originalPrompt = globalThis.prompt;
const alert = vi.fn();
globalThis.alert = alert;
const confirm = vi.fn(() => false);
const prompt = vi.fn(() => 'typed value');
globalThis.confirm = confirm;
globalThis.prompt = prompt;
try {
const { installPageDialogCapture, restorePageDialogCapture } = await import('./inspect');
expect(installPageDialogCapture()).toBe(true);
globalThis.alert('notice');
expect(globalThis.confirm('continue?')).toBe(true);
expect(globalThis.prompt('name?')).toBe('');
expect(restorePageDialogCapture()).toMatchObject([
{ type: 'alert', decision: 'auto_dismissed' },
{ type: 'confirm', decision: 'auto_accepted' },
{ type: 'prompt', decision: 'auto_submitted' },
]);
expect(alert).not.toHaveBeenCalled();
expect(confirm).not.toHaveBeenCalled();
expect(prompt).not.toHaveBeenCalled();
} finally {
globalThis.alert = originalAlert;
globalThis.confirm = originalConfirm;
globalThis.prompt = originalPrompt;
}
});
});
+288
View File
@@ -0,0 +1,288 @@
import {
startBrowserRecording,
stopBrowserRecording,
} from '@/features/browser-recording/service';
import {
listRecordingTraces,
stageBrowserProfileEvidence,
} from '@/features/browser-analysis/service';
import {
listNetworkRequests,
startNetworkCapture,
stopNetworkCapture,
} from '@/features/network-capture/service';
import { actOnPageNode, capturePageContext } from '@/features/page-context/service';
import {
beginPageDialogCapture,
endPageDialogCapture,
installPageDialogCapture,
restorePageDialogCapture,
} from '@/features/page-context/dialogs';
import { ExtensionError } from '@/shared/errors';
import { pairedBrowserTransformCandidate } from '@/features/browser-transform/profile-draft';
import type {
BrowserRecordingEvent,
BrowserRecordingSnapshot,
BrowserTarget,
NetworkRequestRecord,
PageDialog,
PageNodeActionResult,
} from '@/types/models';
type InspectionOwner = { grantId: string; expiresAt: number };
export { installPageDialogCapture, restorePageDialogCapture };
function clipped(value: string | undefined, max: number): string | undefined {
return value === undefined ? undefined : value.slice(0, max);
}
function compactEvent(event: BrowserRecordingEvent): Record<string, unknown> {
return {
id: event.id,
traceId: event.traceId,
kind: event.kind,
operation: event.operation,
label: event.label,
crypto: event.crypto,
transform: event.transform,
direction: event.direction,
method: event.method,
statusCode: event.statusCode,
url: clipped(event.url, 2_048),
dataType: event.dataType,
byteLength: event.byteLength,
resultByteLength: event.resultByteLength,
scriptUrl: clipped(event.scriptUrl, 2_048),
stack: clipped(event.stack, 512),
callableCapable: event.callableCapable,
callHandleId: event.callHandleId,
arguments: event.arguments?.slice(0, 8),
inputs: event.inputs.slice(0, 12),
outputs: event.outputs.slice(0, 12),
inputPreview: clipped(event.inputPreview, 1_024),
outputPreview: clipped(event.outputPreview, 1_024),
error: event.error,
};
}
function compactRequest(request: NetworkRequestRecord): Record<string, unknown> {
return {
id: request.id,
method: request.method,
url: clipped(request.url, 4_096),
resourceType: request.resourceType,
statusCode: request.statusCode,
durationMs: request.durationMs,
error: request.error,
requestBody: request.requestBody && {
...request.requestBody,
data: clipped(request.requestBody.data, 2_048),
},
responseContentType: request.responseContentType,
responseSize: request.responseSize,
};
}
export function summarizeCryptoInspection(
snapshot: BrowserRecordingSnapshot,
requests: NetworkRequestRecord[],
): Record<string, unknown> {
const events = snapshot.events.slice(-16);
const cryptoCount = events.filter((event) => event.kind === 'crypto').length;
const transformCount = events.filter((event) => event.kind === 'transform').length;
const state = cryptoCount + transformCount > 0
? 'observed'
: events.length + requests.length > 0
? 'boundary_only'
: 'no_evidence';
return {
state,
summary: state === 'observed'
? `已观测到 ${cryptoCount} 次密码调用和 ${transformCount} 次编码/序列化转换`
: state === 'boundary_only'
? '已观测到页面或网络边界,但未命中已知加解密适配器'
: '本次页面操作没有产生可分析证据',
recording: {
count: snapshot.status.count,
droppedCount: snapshot.status.droppedCount,
events: events.map(compactEvent),
traces: listRecordingTraces(snapshot, 6),
},
network: {
count: requests.length,
requests: requests.slice(0, 8).map(compactRequest),
},
};
}
async function waitForInspectionIdle(target: BrowserTarget, maxWaitMs: number): Promise<NetworkRequestRecord[]> {
const startedAt = Date.now();
let lastChangedAt = startedAt;
let previousSignature = '';
let observedActivity = false;
let requests: NetworkRequestRecord[] = [];
while (Date.now() - startedAt < maxWaitMs) {
requests = await listNetworkRequests(target, 20);
const signature = requests.map((item) => `${item.id}:${item.completedAt || ''}:${item.error || ''}`).join('|');
if (signature !== previousSignature) {
previousSignature = signature;
lastChangedAt = Date.now();
}
if (requests.length > 0) observedActivity = true;
const allFinished = requests.every((item) => item.completedAt !== undefined || Boolean(item.error));
if (observedActivity && Date.now() - startedAt >= 500 && allFinished && Date.now() - lastChangedAt >= 350) break;
await new Promise((resolve) => globalThis.setTimeout(resolve, 100));
}
return requests;
}
export async function inspectPageCryptoOperation(
target: BrowserTarget,
input: { captureId: string; nodeId: string; settleMs?: number },
owner: InspectionOwner,
): Promise<Record<string, unknown>> {
const startedAt = Date.now();
const settleMs = Math.max(250, Math.min(input.settleMs || 4_000, 5_000));
const warnings: string[] = [];
let action: PageNodeActionResult | undefined;
let snapshot: BrowserRecordingSnapshot | undefined;
let requests: NetworkRequestRecord[] = [];
let dialogs: PageDialog[] = [];
let postAction: Record<string, unknown> | undefined;
let recordingStarted = false;
let networkStarted = false;
let dialogCaptureOwned = false;
dialogCaptureOwned = await beginPageDialogCapture(target);
try {
await startNetworkCapture(target, {
captureHeaders: false,
captureBody: true,
maxEntries: 40,
maxBodyBytes: 8_192,
}, {
kind: 'grant',
grantId: owner.grantId,
expiresAt: owner.expiresAt,
followSameOriginNavigation: true,
});
networkStarted = true;
await startBrowserRecording(target, {
captureValues: true,
maxEntries: 160,
maxValueBytes: 4_096,
expiresAt: owner.expiresAt,
}, { kind: 'grant', grantId: owner.grantId, expiresAt: owner.expiresAt });
recordingStarted = true;
action = await actOnPageNode(input.captureId, input.nodeId, 'click', target);
requests = await waitForInspectionIdle(target, settleMs);
snapshot = await stopBrowserRecording(target, true);
recordingStarted = false;
} finally {
if (recordingStarted) {
try { snapshot = await stopBrowserRecording(target, true); } catch (error) {
warnings.push(`停止页面录制失败: ${error instanceof Error ? error.message : String(error)}`);
}
}
if (networkStarted) {
try {
if (!requests.length) requests = await listNetworkRequests(target, 20);
await stopNetworkCapture(target);
} catch (error) {
warnings.push(`停止网络观察失败: ${error instanceof Error ? error.message : String(error)}`);
}
}
dialogs = await endPageDialogCapture(target, dialogCaptureOwned);
try {
const context = await capturePageContext({ includeDom: true }, {
tabId: target.tabId,
frameId: target.frameId,
});
postAction = {
sameDocument: Boolean(target.documentId && context.target.documentId === target.documentId),
captureId: context.captureId,
target: context.target,
authentication: context.authentication,
document: {
title: context.document.title,
url: context.document.url,
readyState: context.document.readyState,
forms: context.document.forms,
interactive: context.document.interactive,
},
};
} catch (error) {
warnings.push(`采集操作后页面状态失败: ${error instanceof Error ? error.message : String(error)}`);
}
}
if (!snapshot || !action) {
throw new ExtensionError('crypto_inspection_incomplete', '未能完整执行页面加解密检查');
}
await stageBrowserProfileEvidence(snapshot, action.node.semanticKey);
const preparation = snapshot.profileCandidates
.filter((candidate) => [candidate.source, ...candidate.sources]
.some((source) => Boolean(source.callHandleId)))
.sort((left, right) => Number(right.direction === 'request') - Number(left.direction === 'request')
|| right.confidence.score - left.confidence.score)[0];
const pairedPreparation = preparation
? pairedBrowserTransformCandidate(snapshot.profileCandidates, preparation, true)
: undefined;
const directions = [preparation, pairedPreparation].filter(
(candidate): candidate is NonNullable<typeof candidate> => Boolean(candidate),
);
const requestPreparation = directions.find((candidate) => candidate.direction === 'request');
const responsePreparation = directions.find((candidate) => candidate.direction === 'response');
const preparationReady = Boolean(preparation?.status === 'ready'
&& directions.every((candidate) => candidate.status === 'ready'
&& [candidate.source, ...candidate.sources].some((source) => Boolean(source.callHandleId))));
const evidence = summarizeCryptoInspection(snapshot, requests);
return {
version: 1,
target,
trigger: { captureId: input.captureId, nodeId: input.nodeId, action: 'click' },
action,
startedAt,
completedAt: Date.now(),
dialogs,
dialogHandling: {
strategy: 'nonblocking-local-dialog-defaults',
autoDismissedAlerts: dialogs.filter((dialog) => dialog.type === 'alert').length,
autoAcceptedConfirms: dialogs.filter((dialog) => dialog.type === 'confirm').length,
autoSubmittedPrompts: dialogs.filter((dialog) => dialog.type === 'prompt').length,
count: dialogs.length,
navigationInferred: false,
},
postAction,
gatewayPreparation: preparation ? {
state: preparationReady ? 'ready' : 'capture-required',
candidateId: preparation.id,
direction: preparation.direction,
confidence: preparation.confidence,
directions: {
request: requestPreparation
? { candidateId: requestPreparation.id, status: requestPreparation.status }
: { status: 'absent' },
response: responsePreparation
? { candidateId: responsePreparation.id, status: responsePreparation.status }
: { status: 'absent' },
},
request: {
method: preparation.request.method,
url: preparation.request.url,
bodyFormat: preparation.request.bodyFormat,
destinations: preparation.request.mappings.map((mapping) => mapping.destination).filter(Boolean),
},
next: preparationReady
? '需要明文 HTTP 测试时,直接调用 browser.transform.prepare;同一事务的请求与响应会编译进一个 Profile'
: '调用 browser.transform.prepare,插件将自动重触发本次操作、捕获缺失的业务方向并验证完整网关;不需要打开插件 UI',
} : {
state: 'unavailable',
next: '本次证据可用于分析,但不足以生成明文转换;继续使用当前页面,不要重新打开网站',
},
warnings,
...evidence,
purpose: '仅分析这一次页面操作;未创建、验证或保存明文网关 Profile',
};
}
@@ -6,6 +6,7 @@ import type {
} from '@/types/models';
import { inferBrowserTransformProfiles } from './inference';
import { buildRecordingLinks } from '@/features/browser-recording/timeline';
import { pairedBrowserTransformCandidate } from '@/features/browser-transform/profile-draft';
function event(overrides: Partial<BrowserRecordingEvent> & Pick<BrowserRecordingEvent, 'id' | 'sequence' | 'kind' | 'operation'>): BrowserRecordingEvent {
return {
@@ -606,4 +607,112 @@ describe('browser profile inference', () => {
}));
expect(candidate.aiContext.requiredDecision).toBe('none');
});
it('replays response ciphertext, key, and iv as dynamic CryptoJS decrypt inputs', () => {
const response = event({
id: 'encrypted-response-dynamic', sequence: 1, kind: 'fetch', operation: 'response',
direction: 'receive', method: 'POST', url: 'https://example.test/crypto/login', statusCode: 200,
outputs: [
{ path: '$body:json.message', fingerprint: 'cipher', encoding: 'text', byteLength: 88 },
{ path: '$body:json.key', fingerprint: 'key-hex', encoding: 'hex', byteLength: 32 },
{ path: '$body:json.iv', fingerprint: 'iv-hex', encoding: 'hex', byteLength: 32 },
],
});
const key = event({
id: 'parse-key', sequence: 2, kind: 'transform', operation: 'Hex.parse',
inputs: [{ path: '$input', fingerprint: 'key-hex', encoding: 'hex', byteLength: 32 }],
outputs: [{ path: '$output', fingerprint: 'key-word-array', encoding: 'bytes', byteLength: 16 }],
});
const iv = event({
id: 'parse-iv', sequence: 3, kind: 'transform', operation: 'Hex.parse',
inputs: [{ path: '$input', fingerprint: 'iv-hex', encoding: 'hex', byteLength: 32 }],
outputs: [{ path: '$output', fingerprint: 'iv-word-array', encoding: 'bytes', byteLength: 16 }],
});
const decrypt = event({
id: 'decrypt-dynamic-response', sequence: 4, kind: 'crypto', operation: 'AES.decrypt',
crypto: cryptoJsAESDecrypt,
callHandleId: 'decrypt-dynamic-handle', callableCapable: true, arguments: safeArguments,
inputs: [
{ path: '$input', fingerprint: 'cipher', encoding: 'text', byteLength: 88 },
{ path: '$input.key', fingerprint: 'key-word-array', encoding: 'bytes', byteLength: 16 },
{ path: '$input.iv', fingerprint: 'iv-word-array', encoding: 'bytes', byteLength: 16 },
],
outputs: [{ path: '$output', fingerprint: 'plain', encoding: 'hex', byteLength: 42 }],
});
const events = [response, key, iv, decrypt];
const [candidate] = inferBrowserTransformProfiles({
target: { tabId: 7, frameId: 0, documentId: 'document-1' },
events,
links: buildRecordingLinks(events),
});
expect(candidate).toMatchObject({
direction: 'response',
status: 'ready',
source: { dynamicInputPaths: ['$input', '$input.key', '$input.iv'] },
request: {
mappings: [
{ destination: 'body.message' },
{ destination: 'body.key' },
{ destination: 'body.iv' },
],
},
});
expect(candidate.missing).toEqual([]);
expect(candidate.aiContext.requiredDecision).toBe('none');
const withoutCipherLink = inferBrowserTransformProfiles({
target: { tabId: 7, frameId: 0, documentId: 'document-1' },
events: [
{ ...response, outputs: response.outputs.filter((output) => output.path !== '$body:json.message') },
key,
iv,
{ ...decrypt, inputs: decrypt.inputs.filter((input) => input.path !== '$input') },
],
links: buildRecordingLinks([
{ ...response, outputs: response.outputs.filter((output) => output.path !== '$body:json.message') },
key,
iv,
{ ...decrypt, inputs: decrypt.inputs.filter((input) => input.path !== '$input') },
]),
})[0];
expect(withoutCipherLink).toMatchObject({ direction: 'response', status: 'capture-required' });
});
it('pairs request and response candidates by the browser network transaction', () => {
const encrypt = event({
id: 'encrypt-transaction', sequence: 1, kind: 'crypto', operation: 'AES.encrypt', crypto: cryptoJsAES,
callHandleId: 'encrypt-handle', callableCapable: true, arguments: safeArguments,
inputs: [{ path: '$input', fingerprint: 'plain-request', encoding: 'text', byteLength: 20 }],
outputs: [{ path: '$output:string', fingerprint: 'cipher-request', encoding: 'text', byteLength: 44 }],
});
const request = event({
id: 'request-transaction', sequence: 2, kind: 'fetch', operation: 'request', direction: 'send',
channelId: 'fetch-transaction-1', method: 'POST', url: 'https://example.test/login',
inputs: [{ path: '$body:json.message', fingerprint: 'cipher-request', encoding: 'text', byteLength: 44 }],
});
const response = event({
id: 'response-transaction', sequence: 3, kind: 'fetch', operation: 'response', direction: 'receive',
channelId: 'fetch-transaction-1', method: 'POST', url: 'https://example.test/login', statusCode: 200,
outputs: [{ path: '$body:json.message', fingerprint: 'cipher-response', encoding: 'text', byteLength: 44 }],
});
const decrypt = event({
id: 'decrypt-transaction', sequence: 4, kind: 'crypto', operation: 'AES.decrypt', crypto: cryptoJsAESDecrypt,
callHandleId: 'decrypt-handle', callableCapable: true, arguments: safeArguments,
inputs: [{ path: '$input', fingerprint: 'cipher-response', encoding: 'text', byteLength: 44 }],
outputs: [{ path: '$output', fingerprint: 'plain-response', encoding: 'hex', byteLength: 20 }],
});
const events = [encrypt, request, response, decrypt];
const candidates = inferBrowserTransformProfiles({
target: { tabId: 7, frameId: 0, documentId: 'document-1' },
events,
links: buildRecordingLinks(events),
});
const requestCandidate = candidates.find((candidate) => candidate.direction === 'request')!;
const responseCandidate = candidates.find((candidate) => candidate.direction === 'response')!;
expect(requestCandidate.transactionId).toBe('fetch-transaction-1');
expect(responseCandidate.transactionId).toBe('fetch-transaction-1');
expect(pairedBrowserTransformCandidate(candidates, requestCandidate)?.id).toBe(responseCandidate.id);
});
});
+49 -9
View File
@@ -562,6 +562,7 @@ function buildCandidate(
id: candidateId,
recordingId: request.recordingId,
traceId: request.traceId,
transactionId: request.channelId,
target: { ...target },
direction: 'request',
request: {
@@ -698,6 +699,7 @@ function buildUnknownBoundaryCandidate(
id: candidateId,
recordingId: request.recordingId,
traceId: request.traceId,
transactionId: request.channelId,
target: { ...target },
direction: 'request',
request: {
@@ -812,6 +814,7 @@ function buildRequestGraphCandidate(
id: candidateId,
recordingId: request.recordingId,
traceId: request.traceId,
transactionId: request.channelId,
target: { ...target },
direction: 'request',
request: {
@@ -871,6 +874,7 @@ function buildRequestGraphCandidate(
interface LinkedResponseSource {
event: BrowserRecordingEvent;
links: BrowserRecordingLink[];
inputChains: BrowserRecordingLink[][];
stateLinks: BrowserRecordingLink[];
stateEvents: BrowserRecordingEvent[];
}
@@ -896,12 +900,23 @@ function linkedResponseSources(
const chain = [...current.links, link];
if (isReverseCryptoEvent(consumer)) {
const previous = output.get(consumer.id);
if (!previous || chain.length < previous.links.length) {
if (!previous) {
output.set(consumer.id, {
event: consumer,
links: chain,
inputChains: [chain],
...stateSequence(consumer, eventsById, incoming),
});
} else {
const terminal = chain.at(-1)!;
if (!previous.inputChains.some((item) => item.at(-1)?.toPath === terminal.toPath)) {
previous.inputChains.push(chain);
}
const previousTarget = previous.links.at(-1)?.toPath;
if ((terminal.toPath === '$input' && previousTarget !== '$input')
|| (terminal.toPath === previousTarget && chain.length < previous.links.length)) {
previous.links = chain;
}
}
}
const depth = current.depth + 1;
@@ -920,12 +935,24 @@ function buildResponseCandidate(
response: BrowserRecordingEvent,
source: LinkedResponseSource,
): BrowserProfileInferenceCandidate {
const firstLink = source.links[0];
const dataChain = source.inputChains.find((chain) => chain.at(-1)?.toPath === '$input');
const primaryLinks = dataChain || source.links;
const firstLink = dataChain?.[0];
const { destination: inputPath, serialization } = requestMapping(firstLink?.fromPath);
const bodyFormat = responseBodyFormat(response, [serialization]);
const exact = source.links.length > 0 && source.links.every((link) => link.confidence === 'exact');
const responseLinks = [...new Map(source.inputChains.flat().map((link) => [link.id, link])).values()];
const dynamicInputChains = source.inputChains.filter((chain) => chain.at(-1)?.toPath !== '$input');
const dynamicInputs = dynamicInputChains
.map((chain) => chain.at(-1)?.toPath)
.filter((path): path is string => Boolean(path && path !== '$input'));
const replayInputPaths = ['$input', ...dynamicInputs];
const supportsDynamicInputs = source.event.crypto?.adapterId === 'cryptojs'
&& source.event.crypto.operation.toLowerCase().includes('decrypt')
&& dynamicInputs.every((path) => path === '$input.key' || path === '$input.iv');
const exact = responseLinks.length > 0 && responseLinks.every((link) => link.confidence === 'exact');
const hasCallable = Boolean(source.event.callHandleId && source.event.callableCapable);
const replayReady = exact && source.links.length === 1 && Boolean(inputPath) && hasCallable;
const replayReady = exact && dataChain?.length === 1
&& (dynamicInputs.length === 0 || supportsDynamicInputs) && Boolean(inputPath) && hasCallable;
const argumentRoles = source.event.arguments || [];
const responseName = requestLabel(response);
const sourceName = sourceLabel(source.event);
@@ -938,7 +965,7 @@ function buildResponseCandidate(
eventIds: [response.id],
fromPath: firstLink?.fromPath,
}];
source.links.forEach((link, index) => evidence.push({
responseLinks.forEach((link, index) => evidence.push({
id: `evidence-response-link-${link.id || `${response.id}-${source.event.id}-${index}`}`,
kind: link.confidence === 'exact' ? 'exact-value' : 'message-boundary',
strength: link.confidence === 'exact' ? 'proven' : 'supported',
@@ -974,6 +1001,14 @@ function buildResponseCandidate(
label: '页面仍保留本次解密调用的原函数、receiver 与固定参数模板',
eventIds: [source.event.id],
});
const responseMappings = (dataChain ? [dataChain, ...dynamicInputChains] : source.inputChains).map((chain) => {
const mapping = requestMapping(chain[0]?.fromPath);
return {
sourceEventId: source.event.id,
destination: mapping.destination,
serialization: mapping.serialization,
};
});
let score = 20;
if (exact) score += 40;
@@ -989,8 +1024,10 @@ function buildResponseCandidate(
} else {
missing.push({
kind: 'business-callable',
label: exact && inputPath
? '已定位响应解密链;还需捕获上层业务函数,才能保留解码、解压与多阶段解密关系'
label: dynamicInputs.length && !supportsDynamicInputs
? `响应解密还依赖每次响应中的 ${dynamicInputs.map((path) => path.replace(/^\$input\.?/, '')).join('、')};需捕获上层业务函数以保留动态参数关系`
: exact && inputPath
? '已定位响应解密链;还需捕获上层业务函数,才能保留解码、解压与多阶段解密关系'
: '响应字段与页面解密调用尚未形成可回放的直接值链,请继续捕获当前解密现场',
action: 'capture-business-function',
});
@@ -1000,6 +1037,7 @@ function buildResponseCandidate(
id: candidateId,
recordingId: response.recordingId,
traceId: response.traceId,
transactionId: response.channelId,
target: { ...target },
direction: 'response',
request: {
@@ -1009,7 +1047,7 @@ function buildResponseCandidate(
bodyFormat,
destination: inputPath,
serialization,
mappings: [{ sourceEventId: source.event.id, destination: inputPath, serialization }],
mappings: responseMappings,
},
source: {
eventId: source.event.id,
@@ -1017,6 +1055,7 @@ function buildResponseCandidate(
operation: source.event.operation,
crypto: source.event.crypto,
callHandleId: source.event.callHandleId,
dynamicInputPaths: replayReady ? replayInputPaths : undefined,
arguments: argumentRoles,
destination: inputPath,
serialization,
@@ -1027,6 +1066,7 @@ function buildResponseCandidate(
operation: source.event.operation,
crypto: source.event.crypto,
callHandleId: source.event.callHandleId,
dynamicInputPaths: replayReady ? replayInputPaths : undefined,
arguments: argumentRoles,
destination: inputPath,
serialization,
@@ -1038,7 +1078,7 @@ function buildResponseCandidate(
: `已定位 ${responseName} 到 ${sourceName} 的响应解密链`,
flow: [
inputPath ? `${responseName} · ${inputPath}` : responseName,
...(source.links.length > 1 ? [`${source.links.length - 1} 个响应准备步骤`] : []),
...(primaryLinks.length > 1 ? [`${primaryLinks.length - 1} 个响应准备步骤`] : []),
sourceName,
'明文响应',
],
@@ -9,7 +9,7 @@ import { Switch } from '@/components/ui/switch';
import { errorMessage, request } from '@/platform/messaging/runtime';
import type {
ActiveTabInfo, BrowserPageCallable, BrowserPageCallableExecution, BrowserRecordingEvent,
BrowserProfileInferenceCandidate, BrowserRecordingArgumentRole, BrowserRecordingSnapshot,
BrowserProfileInferenceCandidate, BrowserRecordingArgumentRole, BrowserRecordingSnapshot, BrowserTarget,
} from '@/types/models';
import type { CapturedCallableSample } from '@/features/deep-capture/callable-sample';
import { DeepCaptureWorkspace } from '@/features/deep-capture/DeepCaptureWorkspace';
@@ -19,7 +19,12 @@ import {
BrowserTransformWorkspace,
type BrowserTransformSuggestionSeed,
} from '@/features/browser-transform/BrowserTransformWorkspace';
import { createBrowserTransformProfileInput } from '@/features/browser-transform/profile-draft';
import {
createBrowserTransformProfileInput,
pairedBrowserTransformCandidate,
type BrowserTransformProfileBinding,
} from '@/features/browser-transform/profile-draft';
import { browserGatewayNextStep, recordingEventDirection } from './presentation';
type RunTask = (task: () => Promise<void>, success?: string) => Promise<void>;
const DEEP_CAPTURE_AVAILABLE = !import.meta.env.FIREFOX;
@@ -64,6 +69,14 @@ function confidenceLabel(candidate: BrowserProfileInferenceCandidate): string {
return `${level}置信度 · ${candidate.confidence.score}`;
}
function candidateStatusLabel(candidate?: BrowserProfileInferenceCandidate): string {
if (!candidate) return '未检测到';
if (candidate.status === 'ready') return '可直接生成';
if (candidate.status === 'capture-required') return '需要完整捕获';
if (candidate.status === 'mapping-required') return '需要确认映射';
return '证据不足';
}
function eventIcon(kind: BrowserRecordingEvent['kind']) {
if (kind === 'navigation') return <Navigation size={15} />;
if (kind === 'interaction') return <Radio size={15} />;
@@ -154,13 +167,6 @@ function navigationPhaseLabel(event: BrowserRecordingEvent): string {
return '浏览器文档边界';
}
function emptySnapshot(tabId: number): BrowserRecordingSnapshot {
return {
status: { active: false, target: { tabId, frameId: 0 }, documentAvailable: true, count: 0, droppedCount: 0 },
events: [], traces: [], links: [], callables: [], profileCandidates: [],
};
}
function shortSample(event?: BrowserRecordingEvent): string | undefined {
const value = event?.inputPreview || event?.inputs.find((item) => item.preview)?.preview;
return value?.trim() || undefined;
@@ -168,14 +174,21 @@ function shortSample(event?: BrowserRecordingEvent): string | undefined {
function eventAvailableInDocument(
event: BrowserRecordingEvent | undefined,
currentDocumentId: string | undefined,
target: BrowserTarget | undefined,
documentAvailable: boolean,
): boolean {
return documentAvailable && Boolean(event) && (
!event?.documentId || !currentDocumentId || event.documentId === currentDocumentId
(event?.frameId === undefined || !target || event.frameId === target.frameId)
&& (!event?.documentId || !target?.documentId || event.documentId === target.documentId)
);
}
function recordingEventTarget(tabId: number | undefined, event?: BrowserRecordingEvent): BrowserTarget | undefined {
return tabId !== undefined && event
? { tabId, frameId: event.frameId ?? 0, documentId: event.documentId }
: undefined;
}
export function RecordingWorkspace({
tab,
busy,
@@ -200,6 +213,9 @@ export function RecordingWorkspace({
const [callableArguments, setCallableArguments] = useState('[]');
const [callableResult, setCallableResult] = useState<BrowserPageCallableExecution>();
const [gatewaySuggestion, setGatewaySuggestion] = useState<BrowserTransformSuggestionSeed>();
const [pendingGatewayBinding, setPendingGatewayBinding] = useState<BrowserTransformProfileBinding>();
const [captureCandidate, setCaptureCandidate] = useState<BrowserProfileInferenceCandidate>();
useEffect(() => { setCaptureCandidate(undefined); }, [tab?.id]);
const load = useCallback(async () => {
const tabId = tab?.id;
@@ -208,11 +224,8 @@ export function RecordingWorkspace({
return;
}
try {
const target = { tabId, frameId: 0 };
const status = await request('recording.status', target);
const next = status.startedAt
? await request('recording.get', { ...target, limit: 500 })
: emptySnapshot(tabId);
const target = { tabId, frameId: 0, scope: 'tab' as const };
const next = await request('recording.get', { ...target, limit: 500 });
setSnapshot(next);
if (next.status.options) setCaptureValues(next.status.options.captureValues);
setLoadError('');
@@ -256,8 +269,8 @@ export function RecordingWorkspace({
const selectedEvent = snapshot?.events.find((event) => event.id === selectedEventId);
const selectedCallable = snapshot?.callables.find((callable) => callable.id === selectedCallableId);
const recordingTarget = tab ? { tabId: tab.id, frameId: 0 } : undefined;
const documentAvailable = snapshot?.status.documentAvailable !== false;
const callableTarget = snapshot?.status.startedAt && documentAvailable ? snapshot.status.target : undefined;
const tabRecordingTarget = recordingTarget ? { ...recordingTarget, scope: 'tab' as const } : undefined;
const selectedEventTarget = recordingEventTarget(tab?.id, selectedEvent);
useEffect(() => {
if (!selectedEvent) return;
@@ -276,32 +289,36 @@ export function RecordingWorkspace({
const start = () => run(async () => {
if (!tab) throw new Error('请选择目标标签页');
const next = await request('recording.start', {
tabId: tab.id, captureValues, maxEntries: 500, maxValueBytes: 8_192,
tabId: tab.id, scope: 'tab', captureValues, maxEntries: 500, maxValueBytes: 8_192,
});
setSnapshot(next);
setSelectedTraceId('');
setSelectedEventId('');
setCallableResult(undefined);
}, captureValues ? '录制已开始;短时样本仅保留在本次浏览器会话,页面跳转后会自动接续' : '录制已开始,将跨页面记录业务执行链');
setPendingGatewayBinding(undefined);
setCaptureCandidate(undefined);
}, captureValues ? '录制已开始;已覆盖标签页内所有页面,短时样本仅保留在本次浏览器会话' : '录制已开始;已覆盖标签页内所有页面与后续登录 frame');
const stop = () => run(async () => {
if (!recordingTarget) return;
setSnapshot(await request('recording.stop', recordingTarget));
if (!tabRecordingTarget) return;
setSnapshot(await request('recording.stop', tabRecordingTarget));
}, '录制已停止,可以继续验证页面函数');
const clear = () => run(async () => {
if (!recordingTarget) return;
setSnapshot(await request('recording.clear', recordingTarget));
if (!tabRecordingTarget) return;
setSnapshot(await request('recording.clear', tabRecordingTarget));
setCallableResult(undefined);
setPendingGatewayBinding(undefined);
setCaptureCandidate(undefined);
}, '录制与录制型页面函数已清空');
const createCallable = () => run(async () => {
if (snapshot?.status.active) throw new Error('请先停止录制,再保存页面函数');
if (!selectedEventAvailable || !callableTarget || !selectedEvent?.callHandleId) {
if (!selectedEventAvailable || !selectedEventTarget || !selectedEvent?.callHandleId) {
throw new Error(selectedEvent ? '该调用属于另一个页面文档;返回对应页面现场后才能保存' : '当前事件没有可执行调用句柄');
}
const callable = await request('callable.create', {
...callableTarget, source: 'recording', callHandleId: selectedEvent.callHandleId, name: callableName,
...selectedEventTarget, source: 'recording', callHandleId: selectedEvent.callHandleId, name: callableName,
});
setSnapshot((current) => current ? { ...current, callables: [...current.callables.filter((item) => item.id !== callable.id), callable] } : current);
setSelectedCallableId(callable.id);
@@ -310,17 +327,23 @@ export function RecordingWorkspace({
}, '页面函数已创建');
const executeCallable = () => run(async () => {
if (!callableTarget || !selectedCallable) throw new Error(documentAvailable ? '请选择页面函数' : '页面已经导航,旧文档的页面函数不可再执行');
if (!selectedCallable) throw new Error('请选择页面函数');
let args: unknown;
try { args = JSON.parse(callableArguments); } catch { throw new Error('调用参数必须是有效的 JSON 数组'); }
if (!Array.isArray(args)) throw new Error('调用参数必须是 JSON 数组');
setCallableResult(await request('callable.execute', { ...callableTarget, callableId: selectedCallable.id, args }));
setCallableResult(await request('callable.execute', { ...selectedCallable.target, callableId: selectedCallable.id, args }));
}, '页面函数验证完成');
const deleteCallable = () => run(async () => {
if (!callableTarget || !selectedCallable) return;
const callables = await request('callable.delete', { ...callableTarget, callableId: selectedCallable.id });
setSnapshot((current) => current ? { ...current, callables } : current);
if (!selectedCallable) return;
const callables = await request('callable.delete', { ...selectedCallable.target, callableId: selectedCallable.id });
setSnapshot((current) => current ? {
...current,
callables: [
...current.callables.filter((item) => item.target.frameId !== selectedCallable.target.frameId),
...callables,
],
} : current);
setCallableResult(undefined);
}, '页面函数已删除');
@@ -350,18 +373,35 @@ export function RecordingWorkspace({
? `页面函数句柄 ${(snapshot.status.retainedCallBytes / 1024).toFixed(1)} KiB`
: undefined,
].filter(Boolean).join(' · ');
const currentDocumentId = snapshot?.status.target.documentId;
const selectedEventAvailable = eventAvailableInDocument(selectedEvent, currentDocumentId, documentAvailable);
const documentAvailable = snapshot?.status.documentAvailable !== false;
const outgoingLinks = selectedEvent ? snapshot?.links.filter((link) => link.fromEventId === selectedEvent.id) || [] : [];
const incomingLinks = selectedEvent ? snapshot?.links.filter((link) => link.toEventId === selectedEvent.id) || [] : [];
const traceCandidates = snapshot?.profileCandidates.filter((candidate) => candidate.traceId === selectedTraceId) || [];
const selectedCandidate = traceCandidates.find((candidate) => (
candidate.sources.some((source) => source.eventId === selectedEventId) || candidate.request.eventId === selectedEventId
)) || traceCandidates[0];
const sourceCandidates = traceCandidates.filter((candidate) => candidate.source.eventId === selectedEventId);
const selectedCandidate = sourceCandidates.length === 1 ? sourceCandidates[0] : undefined;
const pairedCandidate = selectedCandidate
? pairedBrowserTransformCandidate(snapshot?.profileCandidates || [], selectedCandidate)
: undefined;
const boundaryCandidates = selectedCandidate ? [] : traceCandidates.filter((candidate) => candidate.request.eventId === selectedEventId);
const relatedCandidate = boundaryCandidates.length === 1 ? boundaryCandidates[0] : undefined;
const selectedEventAvailable = eventAvailableInDocument(
selectedEvent,
selectedCandidate?.target || relatedCandidate?.target || selectedEventTarget,
selectedEvent?.frameId === 0 ? documentAvailable : true,
);
const relatedSourceEvent = relatedCandidate
? snapshot?.events.find((event) => event.id === relatedCandidate.source.eventId)
: undefined;
const selectedEventDirection = selectedEvent
? recordingEventDirection(selectedEvent, traceCandidates)
: undefined;
const gatewayNextStep = selectedCandidate
? browserGatewayNextStep(selectedCandidate, pairedCandidate)
: undefined;
const candidateSourceEvent = selectedCandidate
? snapshot?.events.find((event) => event.id === selectedCandidate.source.eventId)
: undefined;
const candidateAvailable = eventAvailableInDocument(candidateSourceEvent, currentDocumentId, documentAvailable);
const candidateAvailable = eventAvailableInDocument(candidateSourceEvent, selectedCandidate?.target, true);
const canDeepCapture = DEEP_CAPTURE_AVAILABLE && selectedEventAvailable && Boolean(selectedEvent
&& ['crypto', 'fetch', 'xhr', 'form', 'beacon', 'worker', 'message'].includes(selectedEvent.kind)
&& (selectedEvent.url || selectedEvent.wrapperHandleId));
@@ -373,14 +413,16 @@ export function RecordingWorkspace({
return;
}
void run(async () => {
if (!recordingTarget) throw new Error('目标标签页不可用');
setSnapshot(await request('recording.stop', recordingTarget));
if (!tabRecordingTarget) throw new Error('目标标签页不可用');
setSnapshot(await request('recording.stop', tabRecordingTarget));
setCallableEditorOpen(true);
}, '录制已停止,请确认页面函数名称');
};
const continueInference = (candidate: BrowserProfileInferenceCandidate) => {
setCaptureCandidate(candidate);
setRecoveryProfileId('');
setSelectedTraceId(candidate.traceId);
setSelectedEventId(candidate.capturePlan?.matcherEventId
|| (candidate.sources.length > 1 ? candidate.request.eventId : candidate.source.eventId));
setAutoArmRequest((current) => current + 1);
@@ -388,6 +430,7 @@ export function RecordingWorkspace({
};
const openRecovery = (profileId: string) => {
setCaptureCandidate(undefined);
setRecoveryProfileId(profileId);
setAutoRecoveryRequest((current) => current + 1);
setWorkspaceMode('deep');
@@ -405,6 +448,37 @@ export function RecordingWorkspace({
capturedSample?: CapturedCallableSample,
) => {
if (!tab) throw new Error('目标标签页已经关闭');
const binding: BrowserTransformProfileBinding = { candidate, callable };
const pair = pairedBrowserTransformCandidate(snapshot?.profileCandidates || [], candidate, true);
let pairedBinding = pendingGatewayBinding?.candidate.id === pair?.id ? pendingGatewayBinding : undefined;
if (!pairedBinding && pair) {
if (pair.status === 'capture-required') {
setPendingGatewayBinding(binding);
continueInference(pair);
return;
}
if (pair.status !== 'ready') {
throw new Error(`已检测到${pair.direction === 'request' ? '请求' : '响应'}方向,但${candidateStatusLabel(pair)},不能静默保存为单向网关`);
}
const pairEvent = snapshot?.events.find((item) => item.id === pair.source.eventId);
if (!snapshot || !eventAvailableInDocument(pairEvent, pair.target, true)) {
throw new Error('配对方向属于另一个页面文档,请返回对应页面现场后再生成');
}
const pairInputCount = pair.source.dynamicInputPaths?.length || 1;
let pairCallable = snapshot.callables.find((item) => item.provenance.eventId === pair.source.eventId
&& item.inputSlots.filter((slot) => !slot.retained).length === pairInputCount);
if (!pairCallable) {
if (!pair.source.callHandleId) throw new Error('配对方向没有可复用的页面调用句柄');
pairCallable = await request('callable.create', {
...pair.target,
source: 'recording',
callHandleId: pair.source.callHandleId,
name: `${pair.source.crypto?.algorithm || pair.source.crypto?.operation || pair.source.operation} 页面函数`,
dynamicInputPaths: pair.source.dynamicInputPaths,
});
}
pairedBinding = { candidate: pair, callable: pairCallable };
}
const sourceEvent = snapshot?.events.find((item) => item.id === candidate.source.eventId);
const boundaryEvent = snapshot?.events.find((item) => item.id === candidate.request.eventId);
const profile = await request('transform.profile.save', createBrowserTransformProfileInput(
@@ -412,15 +486,21 @@ export function RecordingWorkspace({
sourceEvent,
callable,
candidate,
undefined,
pairedBinding,
));
const callables = [callable, pairedBinding?.callable].filter((item): item is BrowserPageCallable => Boolean(item));
const callableIds = new Set(callables.map((item) => item.id));
setSnapshot((current) => current ? {
...current,
callables: [...current.callables.filter((item) => item.id !== callable.id), callable],
callables: [...current.callables.filter((item) => !callableIds.has(item.id)), ...callables],
} : current);
setPendingGatewayBinding(undefined);
setCaptureCandidate(undefined);
setGatewaySuggestion((current) => ({
revision: (current?.revision || 0) + 1,
candidate,
callable,
callables,
profile,
sampleBody: capturedSample?.body || shortSample(candidate.direction === 'response' ? boundaryEvent : sourceEvent),
sampleLabel: capturedSample?.label || (candidate.direction === 'response' && boundaryEvent
@@ -434,28 +514,31 @@ export function RecordingWorkspace({
if (candidate.sources.length !== 1) {
throw new Error('多调用请求需要先捕获上层业务函数,不能把相互依赖的低层调用拆开回放');
}
if (!candidateAvailable || !recordingTarget || !candidate.source.callHandleId) {
if (!candidateAvailable || !tabRecordingTarget || !candidate.source.callHandleId) {
throw new Error(candidateAvailable ? '推断候选没有可复用的页面调用句柄' : '该函数属于另一个页面文档,请返回对应页面现场后再生成');
}
let currentSnapshot = snapshot;
if (currentSnapshot?.status.active) {
currentSnapshot = await request('recording.stop', recordingTarget);
currentSnapshot = await request('recording.stop', tabRecordingTarget);
setSnapshot(currentSnapshot);
}
if (!currentSnapshot) throw new Error('没有可用的录制现场');
const target = currentSnapshot.status.target;
const target = candidate.target;
if (!target) throw new Error('录制文档已经失效');
let callable = currentSnapshot.callables.find((item) => item.provenance.eventId === candidate.source.eventId);
const inputCount = candidate.source.dynamicInputPaths?.length || 1;
let callable = currentSnapshot.callables.find((item) => item.provenance.eventId === candidate.source.eventId
&& item.inputSlots.filter((slot) => !slot.retained).length === inputCount);
if (!callable) {
callable = await request('callable.create', {
...target,
source: 'recording',
callHandleId: candidate.source.callHandleId,
name: `${candidate.source.crypto?.algorithm || candidate.source.crypto?.operation || candidate.source.operation} 页面函数`,
dynamicInputPaths: candidate.source.dynamicInputPaths,
});
}
await openSuggestedGateway(candidate, callable);
}, '已根据录制证据生成并保存明文网关');
}, '当前方向已完成;存在配对方向时将继续捕获并合并为一个网关');
return <section className="recording-section">
<div className="recording-heading">
@@ -474,7 +557,7 @@ export function RecordingWorkspace({
</div>
<div id="recording-mode-panel" className="recording-mode-panel" role="tabpanel" aria-labelledby="recording-mode-tab" hidden={workspaceMode !== 'recording'}><div className="recording-controls">
<label><Switch checked={captureValues} disabled={active || busy} onCheckedChange={setCaptureValues} /><span><strong>保留短时样本</strong><small>关闭时仅保留本次录制的关联指纹</small></span></label>
<label><Switch checked={captureValues} disabled={active || busy} onCheckedChange={setCaptureValues} /><span><strong>保留短时样本</strong><small>自动覆盖当前标签页内所有 frame;关闭时仅保留关联指纹</small></span></label>
<span className="recording-summary" title={persistenceTitle}>{snapshot?.traces.length || 0} 个 Trace · {snapshot?.links.length || 0} 条值关联 · {snapshot?.callables.length || 0} 个页面函数 · {persistenceLabel}{retentionDrops ? ` · ${retentionDrops} 项按预算丢弃` : ''}</span>
<Button size="icon" variant="ghost" aria-label="刷新录制" title="刷新录制" disabled={!tab} onClick={() => void load()}><RefreshCw size={15} /></Button>
<Button size="icon" variant="ghost" aria-label="清空录制" title="清空录制" disabled={!hasRecording || busy} onClick={() => void clear()}><Trash2 size={15} /></Button>
@@ -518,12 +601,18 @@ export function RecordingWorkspace({
<div className="recording-pipeline__body">
{!traceEvents.length ? <div className="recording-column-empty">当前 Trace 没有事件</div> : traceEvents.map((event, index) => {
const linked = snapshot?.links.some((link) => link.fromEventId === event.id || link.toEventId === event.id);
const callableAvailable = eventAvailableInDocument(event, currentDocumentId, documentAvailable);
const callableAvailable = eventAvailableInDocument(
event,
snapshot?.profileCandidates.find((candidate) => candidate.source.eventId === event.id)?.target
|| recordingEventTarget(tab?.id, event),
event.frameId === 0 ? documentAvailable : true,
);
const flowDirection = recordingEventDirection(event, traceCandidates);
return <div className={`recording-pipeline-step ${event.kind === 'navigation' ? 'is-navigation' : ''}`} key={event.id}>
<span className="recording-step-rail" aria-hidden="true"><i>{String(index + 1).padStart(2, '0')}</i>{index < traceEvents.length - 1 ? <span><ArrowDown size={11} /></span> : null}</span>
<button data-event-id={event.id} className={`${event.id === selectedEventId ? 'is-selected' : ''} ${linked ? 'is-linked' : ''}`} onClick={() => setSelectedEventId(event.id)}>
<span className={`recording-event-icon kind-${event.kind}`}>{eventIcon(event.kind)}</span>
<span><small>{KIND_LABELS[event.kind]}</small><strong>{eventTitle(event)}</strong><em>{eventSubtitle(event)}</em>{event.kind === 'navigation' ? <b>{navigationPhaseLabel(event)}</b> : null}</span>
<span><small className={flowDirection ? `is-${flowDirection}` : ''}>{flowDirection === 'request' ? '↑ 请求' : flowDirection === 'response' ? '↓ 响应' : KIND_LABELS[event.kind]}{flowDirection ? ` · ${KIND_LABELS[event.kind]}` : ''}</small><strong>{eventTitle(event)}</strong><em>{eventSubtitle(event)}</em>{event.kind === 'navigation' ? <b>{navigationPhaseLabel(event)}</b> : null}</span>
<span className="recording-event-meta">{event.callableCapable ? <i className={callableAvailable ? '' : 'is-history'}>{callableAvailable ? '当前可用' : '历史现场'}</i> : null}<time title={new Date(event.timestamp).toLocaleString()}>{relativeTime(event.timestamp, selectedTrace?.startedAt)}</time>{event.durationMs !== undefined ? <small>{event.durationMs.toFixed(1)} ms</small> : null}</span>
</button>
</div>;
@@ -533,7 +622,7 @@ export function RecordingWorkspace({
<aside className="recording-inspector">
{!selectedEvent ? <div className="recording-column-empty">选择一个 Pipeline 步骤</div> : <>
<header><div><span>{KIND_LABELS[selectedEvent.kind]}</span><strong>{eventTitle(selectedEvent)}</strong><small title={selectedEvent.url || selectedEvent.scriptUrl}>{selectedEvent.url || selectedEvent.scriptUrl || '页面主世界'}</small></div>{selectedEvent.error ? <i className="is-error">ERROR</i> : <i>#{selectedEvent.sequence}</i>}</header>
<header><div><span>{selectedEventDirection === 'request' ? '↑ 请求' : selectedEventDirection === 'response' ? '↓ 响应' : KIND_LABELS[selectedEvent.kind]}</span><strong>{eventTitle(selectedEvent)}</strong><small title={selectedEvent.url || selectedEvent.scriptUrl}>{selectedEvent.url || selectedEvent.scriptUrl || '页面主世界'}</small></div>{selectedEvent.error ? <i className="is-error">ERROR</i> : <i>#{selectedEvent.sequence}</i>}</header>
{selectedEvent.kind === 'navigation' && selectedEvent.navigation
? <dl className="recording-navigation-detail">
<div><dt>状态</dt><dd>{navigationPhaseLabel(selectedEvent)}</dd></div>
@@ -543,10 +632,15 @@ export function RecordingWorkspace({
</dl>
: <dl><div><dt>输入</dt><dd>{selectedEvent.byteLength === undefined ? `${selectedEvent.inputs.length} 个值` : `${selectedEvent.byteLength} B`}</dd></div><div><dt>输出</dt><dd>{selectedEvent.resultByteLength === undefined ? `${selectedEvent.outputs.length} 个值` : `${selectedEvent.resultByteLength} B`}</dd></div><div><dt>上游</dt><dd>{incomingLinks.length}</dd></div><div><dt>下游</dt><dd>{outgoingLinks.length}</dd></div></dl>}
{relatedCandidate && relatedSourceEvent && <section className="recording-related-transform">
<div><Link2 size={15} /><span><strong>已关联{relatedCandidate.direction === 'request' ? '请求' : '响应'}转换</strong><small>#{relatedSourceEvent.sequence} · {eventTitle(relatedSourceEvent)}</small></span></div>
<Button variant="ghost" onClick={() => setSelectedEventId(relatedSourceEvent.id)}>查看{relatedCandidate.direction === 'request' ? '请求' : '响应'}转换</Button>
</section>}
{selectedCandidate && <section className={`profile-inference is-${selectedCandidate.confidence.level}`}>
<div className="profile-inference__heading">
<span className="profile-inference__mark"><Sparkles size={15} /></span>
<span><small>自动推断 Profile</small><strong>{selectedCandidate.summary}</strong></span>
<span><small>自动识别 · {selectedCandidate.direction === 'request' ? '请求转换' : '响应转换'}</small><strong>{selectedCandidate.summary}</strong></span>
<i><ShieldCheck size={12} />{confidenceLabel(selectedCandidate)}</i>
</div>
<div className="profile-inference__flow" aria-label="推断的数据流">
@@ -554,6 +648,11 @@ export function RecordingWorkspace({
<code>{item}</code>{index < selectedCandidate.flow.length - 1 ? <ChevronRight size={12} /> : null}
</span>)}
</div>
<dl className="profile-inference__arguments">
<div><dt>请求转换</dt><dd>{candidateStatusLabel(selectedCandidate.direction === 'request' ? selectedCandidate : pairedCandidate?.direction === 'request' ? pairedCandidate : undefined)}</dd></div>
<div><dt>响应转换</dt><dd>{candidateStatusLabel(selectedCandidate.direction === 'response' ? selectedCandidate : pairedCandidate?.direction === 'response' ? pairedCandidate : undefined)}</dd></div>
</dl>
{pendingGatewayBinding && <div className="profile-inference__next is-ready"><span>{pendingGatewayBinding.candidate.direction === 'request' ? '请求转换' : '响应转换'}已经捕获,正在完成配对方向;完成后会保存为一个双向网关。</span></div>}
{selectedCandidate.sources.length > 1 && <div className="profile-inference__sources">
{selectedCandidate.sources.map((source, index) => <div key={source.eventId}>
<span>{String(index + 1).padStart(2, '0')}</span>
@@ -571,27 +670,28 @@ export function RecordingWorkspace({
<summary>{selectedCandidate.evidence.length} 项证据</summary>
<ol>{selectedCandidate.evidence.map((item) => <li key={item.id} data-strength={item.strength}><i />{item.label}</li>)}</ol>
</details>
{selectedCandidate.missing[0] && <div className="profile-inference__next"><span>{selectedCandidate.missing[0].label}</span>
{selectedCandidate.missing[0].action === 'capture-business-function' && DEEP_CAPTURE_AVAILABLE
? <Button variant="primary" onClick={() => continueInference(selectedCandidate)}><Sparkles size={14} />{selectedCandidate.direction === 'response' ? '自动捕获完整解密流程' : '自动捕获完整加密流程'}</Button>
: null}
{gatewayNextStep && <div className={`profile-inference__next ${gatewayNextStep.kind === 'create' ? 'is-ready' : ''}`}><span>{candidateAvailable ? gatewayNextStep.description : '关联证据仍然保留;该页面函数属于另一个文档,返回对应页面现场后可以继续。'}</span>
{gatewayNextStep.kind === 'capture' && DEEP_CAPTURE_AVAILABLE
? <Button variant="primary" disabled={busy || !candidateAvailable} onClick={() => continueInference(gatewayNextStep.candidate)}><Sparkles size={14} />{candidateAvailable ? gatewayNextStep.label : '等待对应页面'}</Button>
: gatewayNextStep.kind === 'create'
? <Button variant="primary" disabled={busy || !candidateAvailable} onClick={() => void createSuggestedGateway(selectedCandidate)}><FileKey2 size={14} />{candidateAvailable ? gatewayNextStep.label : '等待对应页面'}</Button>
: null}
</div>}
{selectedCandidate.status === 'ready' && <div className="profile-inference__next is-ready"><span>{candidateAvailable ? (selectedCandidate.direction === 'response' ? '线上响应字段与页面解密调用已经精确关联,可直接生成响应明文网关。' : '页面调用与线上字段已经精确关联,只需确认明文来源和输出形态。') : '关联证据仍然保留;该页面函数属于另一个文档,返回对应页面现场后可以继续生成。'}</span><Button variant="primary" disabled={busy || !candidateAvailable} onClick={() => void createSuggestedGateway(selectedCandidate)}><FileKey2 size={14} />{candidateAvailable ? '生成明文网关' : '等待对应页面'}</Button></div>}
</section>}
{(selectedEvent.inputPreview || selectedEvent.outputPreview) && <div className="recording-values"><strong>短时样本</strong>{selectedEvent.inputPreview && <pre>{selectedEvent.inputPreview}</pre>}{selectedEvent.outputPreview && <pre>{selectedEvent.outputPreview}</pre>}</div>}
{selectedEvent.kind !== 'navigation' ? <details className="recording-evidence"><summary>调用证据</summary><pre>{selectedEvent.stack || selectedEvent.scriptUrl || '没有可用调用栈'}</pre></details> : null}
{canDeepCapture && !selectedCandidate && <section className="recording-deep-action">
{canDeepCapture && !selectedCandidate && !relatedCandidate && <section className="recording-deep-action">
<div><Bug size={15} /><span><strong>捕获真实业务上下文</strong><small>{selectedEvent.kind === 'crypto'
? '下次命中当前加密调用时暂停'
: selectedEvent.kind === 'worker' || selectedEvent.kind === 'message' || selectedEvent.kind === 'beacon'
? '下次命中当前页面通信边界时暂停'
: '下次发出当前请求时暂停'}</small></span></div>
<Button variant="primary" onClick={() => setWorkspaceMode('deep')}><Bug size={14} />深入当前调用</Button>
<Button variant="primary" onClick={() => { setCaptureCandidate(undefined); setWorkspaceMode('deep'); }}><Bug size={14} />深入当前调用</Button>
</section>}
{selectedEvent.callableCapable && selectedEvent.callHandleId && <section className="recording-recipe-action">
{selectedEvent.callableCapable && selectedEvent.callHandleId && !selectedCandidate && <section className="recording-recipe-action">
<div><KeyRound size={15} /><span><strong>保存为页面函数</strong><small>{!selectedEventAvailable ? '该调用属于另一个页面文档,返回对应页面后可以恢复' : active ? '保存前会先停止录制,避免轮询继续改变调用现场' : '保留原函数、receiver 与固定参数,页面刷新后失效'}</small></span></div>
{!callableEditorOpen ? <Button variant="primary" disabled={busy || !selectedEventAvailable} onClick={prepareCallableEditor}><Save size={14} />{active ? '停止录制并保存' : '保存页面函数'}</Button> : <div className="recording-recipe-editor">
<label><span>名称</span><input value={callableName} onChange={(event) => setCallableName(event.target.value)} /></label>
@@ -611,8 +711,9 @@ export function RecordingWorkspace({
{DEEP_CAPTURE_AVAILABLE && <div id="deep-mode-panel" className="recording-mode-panel" role="tabpanel" aria-labelledby="deep-mode-tab" hidden={workspaceMode !== 'deep'}>
<DeepCaptureWorkspace
tab={tab}
recordingTarget={selectedCandidate?.target || relatedCandidate?.target || selectedEventTarget}
selectedEvent={selectedEvent}
selectedCandidate={selectedCandidate}
selectedCandidate={captureCandidate || selectedCandidate}
autoArmRequest={autoArmRequest}
recoveryProfileId={recoveryProfileId}
autoRecoveryRequest={autoRecoveryRequest}
@@ -631,7 +732,7 @@ export function RecordingWorkspace({
run={run}
gatewayShared={gatewayShared}
onShareGateway={onShareGateway}
onOpenCapture={() => { setRecoveryProfileId(''); setWorkspaceMode(DEEP_CAPTURE_AVAILABLE ? 'deep' : 'recording'); }}
onOpenCapture={() => { setCaptureCandidate(undefined); setRecoveryProfileId(''); setWorkspaceMode(DEEP_CAPTURE_AVAILABLE ? 'deep' : 'recording'); }}
onOpenRecovery={DEEP_CAPTURE_AVAILABLE ? openRecovery : () => setWorkspaceMode('recording')}
deepCaptureAvailable={DEEP_CAPTURE_AVAILABLE}
recoveryRevision={recoveryRevision}
@@ -0,0 +1,10 @@
import { describe, expect, it } from 'vitest';
import { recordingExpiryDelay } from './expiry';
describe('recording expiry timer', () => {
it('does not overflow a permanent paired-browser grant into an immediate timeout', () => {
expect(recordingExpiryDelay(Number.MAX_SAFE_INTEGER, 1_000)).toBeUndefined();
expect(recordingExpiryDelay(11_000, 1_000)).toBe(10_000);
expect(recordingExpiryDelay(999, 1_000)).toBe(0);
});
});
+8
View File
@@ -0,0 +1,8 @@
const MAX_BROWSER_TIMER_MS = 0x7fffffff;
export function recordingExpiryDelay(expiresAt: number | undefined, now = Date.now()): number | undefined {
if (expiresAt === undefined) return undefined;
const delay = expiresAt - now;
if (delay <= 0) return 0;
return delay <= MAX_BROWSER_TIMER_MS ? delay : undefined;
}
@@ -2,6 +2,7 @@ import type {
BrowserRecordingEventKind,
BrowserRecordingValueEvidence,
} from '@/types/models';
import { readRequestBody } from '@/shared/request-body';
type NetworkKind = Extract<BrowserRecordingEventKind, 'fetch' | 'xhr' | 'form' | 'websocket'>;
@@ -333,8 +334,9 @@ export function createNetworkBoundaryRuntime(
const request = RequestConstructor && input instanceof RequestConstructor ? input : undefined;
const url = absoluteRequestUrl(request || input);
const method = (init?.method || request?.method || 'GET').toUpperCase().slice(0, 32);
bestEffort(() => {
const body = init?.body;
let stack: ReturnType<NetworkBoundaryHost['stackInfo']> = {};
bestEffort(() => { stack = host.stackInfo(); });
const emitRequest = (body: unknown) => bestEffort(() => {
host.emit({
kind: 'fetch',
operation: 'request',
@@ -350,9 +352,15 @@ export function createNetworkBoundaryRuntime(
...headerEvidence(init?.headers || request?.headers, '$headers'),
...queryEvidence(request || input),
],
...host.stackInfo(),
...stack,
}, context);
});
if (request && init?.body === undefined) {
void readRequestBody(request, MAX_ASYNC_BINARY_BYTES).then(
(body) => emitRequest(body.value),
() => emitRequest(undefined),
);
} else emitRequest(init?.body);
let result: ReturnType<typeof scope.fetch>;
try {
result = Reflect.apply(original, this, [input, init]);
@@ -41,6 +41,18 @@ function environment() {
}
describe('recording trace runtime', () => {
it('inherits unique value provenance across a later interaction without guessing shared values', () => {
const { runtime, setMaxEntries } = environment();
setMaxEntries(20);
const value = { path: '$body', fingerprint: 'ciphertext', encoding: 'text' as const, byteLength: 32 };
const key = { ...value, fingerprint: 'shared-key' };
runtime.bindContext({ traceId: 'B' });
runtime.record({ kind: 'fetch', operation: 'response', outputs: [value, key] }, { traceId: 'A' });
runtime.record({ kind: 'fetch', operation: 'response', outputs: [key] }, { traceId: 'C' });
expect(runtime.record({ kind: 'crypto', operation: 'decrypt', inputs: [value, key] })?.traceId).toBe('A');
runtime.record({ kind: 'fetch', operation: 'response', outputs: [value] }, { traceId: 'C' });
expect(runtime.record({ kind: 'crypto', operation: 'decrypt', inputs: [value] })?.traceId).toBe('B');
});
it('does not create events or trace state while recording is inactive', () => {
const environmentState = environment();
environmentState.setActive(false);
@@ -91,7 +91,19 @@ export function createRecordingTraceRuntime(
): BrowserRecordingEvent | undefined => {
const recordingId = host.recordingId();
if (!host.active() || !recordingId) return undefined;
const eventContext = explicitContext || context();
// Values carry causality across await and intervening user interactions.
// Only a unique recorded origin can override the current interaction.
const fingerprints = new Set((input.inputs || [])
.filter((value) => value.byteLength >= 8).map((value) => value.fingerprint));
let origins: Map<string, RecordingTraceContext> | undefined;
for (const fingerprint of fingerprints) {
const matches = new Map(events.filter((event) => event.outputs.some((value) => value.fingerprint === fingerprint))
.map((event) => [event.traceId, { traceId: event.traceId, interactionId: event.interactionId }]));
if (!matches.size) continue;
origins = origins ? new Map([...origins].filter(([traceId]) => matches.has(traceId))) : matches;
}
const inherited = origins?.size === 1 ? origins.values().next().value : undefined;
const eventContext = explicitContext || inherited || context();
sequence += 1;
const item: BrowserRecordingEvent = {
id: host.unique('event'),
@@ -0,0 +1,52 @@
import { describe, expect, it } from 'vitest';
import type { BrowserProfileInferenceCandidate, BrowserRecordingEvent } from '@/types/models';
import { browserGatewayNextStep, recordingEventDirection } from './presentation';
const event = (id: string, kind: BrowserRecordingEvent['kind'], operation: string): BrowserRecordingEvent => ({
id, kind, operation, sequence: 1, timestamp: 1, recordingId: 'recording-1', traceId: 'trace-1',
inputs: [], outputs: [], sensitiveCaptured: false,
});
const candidate = (direction: 'request' | 'response', status: BrowserProfileInferenceCandidate['status']): BrowserProfileInferenceCandidate => ({
id: `candidate-${direction}`, recordingId: 'recording-1', traceId: 'trace-1',
target: { tabId: 1, frameId: 0 }, direction,
request: { eventId: `${direction}-boundary`, method: 'POST', url: 'https://example.test/login', bodyFormat: 'json', mappings: [] },
source: { eventId: `${direction}-crypto`, kind: 'crypto', operation: direction === 'request' ? 'AES.encrypt' : 'AES.decrypt', arguments: [] },
sources: [], status, confidence: { score: 100, level: 'high' }, summary: '', flow: [], pipeline: [], evidence: [{
id: `${direction}-evidence`, kind: direction === 'request' ? 'request-boundary' : 'response-boundary', strength: 'proven',
label: '', eventIds: [`${direction}-boundary`, `${direction}-crypto`, `${direction}-transform`],
}], missing: status === 'capture-required' ? [{ kind: 'business-callable', label: 'capture', action: 'capture-business-function' }] : [],
aiContext: {
valuePolicy: 'metadata-only',
request: { eventId: `${direction}-boundary`, method: 'POST', url: 'https://example.test/login' },
source: { eventId: `${direction}-crypto`, kind: 'crypto', operation: direction === 'request' ? 'AES.encrypt' : 'AES.decrypt', arguments: [] },
sources: [], evidenceIds: [], requiredDecision: status === 'ready' ? 'none' : 'capture-business-callable',
},
});
describe('recording presentation', () => {
it('labels network boundaries and linked transform events by direction', () => {
const request = candidate('request', 'ready');
const response = candidate('response', 'ready');
expect(recordingEventDirection(event('request-boundary', 'fetch', 'request'), [request, response])).toBe('request');
expect(recordingEventDirection(event('response-boundary', 'fetch', 'response'), [request, response])).toBe('response');
expect(recordingEventDirection(event('response-transform', 'transform', 'Hex.parse'), [request, response])).toBe('response');
expect(recordingEventDirection(event('unrelated', 'transform', 'JSON.stringify'), [request, response])).toBeUndefined();
});
it('offers the next incomplete direction instead of claiming the gateway is finished', () => {
const request = candidate('request', 'capture-required');
const response = candidate('response', 'ready');
expect(browserGatewayNextStep(response, request)).toMatchObject({
kind: 'capture', candidate: request, label: '继续捕获请求方向',
});
expect(browserGatewayNextStep(response, { ...request, status: 'ready' })).toMatchObject({
kind: 'create', label: '生成双向协议网关',
});
expect(browserGatewayNextStep(response)).toMatchObject({
kind: 'create', label: '生成仅响应网关',
});
});
});
@@ -0,0 +1,63 @@
import type { BrowserProfileInferenceCandidate, BrowserRecordingEvent } from '@/types/models';
export type RecordingEventDirection = 'request' | 'response';
const HTTP_EVENT_KINDS = new Set<BrowserRecordingEvent['kind']>(['fetch', 'xhr', 'form', 'beacon']);
export function recordingEventDirection(
event: BrowserRecordingEvent,
candidates: BrowserProfileInferenceCandidate[],
): RecordingEventDirection | undefined {
if (HTTP_EVENT_KINDS.has(event.kind)) {
return event.operation === 'response' || event.operation.startsWith('response.')
? 'response'
: 'request';
}
const directions = new Set(candidates.filter((candidate) => (
candidate.source.eventId === event.id
|| candidate.sources.some((source) => source.eventId === event.id)
|| candidate.evidence.some((evidence) => evidence.eventIds.includes(event.id))
)).map((candidate) => candidate.direction));
return directions.size === 1 ? [...directions][0] : undefined;
}
export type BrowserGatewayNextStep = {
kind: 'capture' | 'create' | 'blocked';
candidate: BrowserProfileInferenceCandidate;
label?: string;
description: string;
};
const directionName = (direction: BrowserProfileInferenceCandidate['direction']) => (
direction === 'request' ? '请求' : '响应'
);
export function browserGatewayNextStep(
candidate: BrowserProfileInferenceCandidate,
paired?: BrowserProfileInferenceCandidate,
): BrowserGatewayNextStep {
if (candidate.status !== 'ready') {
return candidate.status === 'capture-required'
? {
kind: 'capture', candidate, label: `继续捕获${directionName(candidate.direction)}方向`,
description: `需要先捕获完整的${directionName(candidate.direction)}转换。`,
}
: { kind: 'blocked', candidate, description: candidate.missing[0]?.label || '当前转换证据还不完整。' };
}
if (!paired) return {
kind: 'create', candidate, label: `生成仅${directionName(candidate.direction)}网关`,
description: `当前操作只检测到${directionName(candidate.direction)}转换,将生成单向协议网关。`,
};
if (paired.status === 'capture-required') return {
kind: 'capture', candidate: paired, label: `继续捕获${directionName(paired.direction)}方向`,
description: `${directionName(candidate.direction)}方向已就绪,还需要捕获${directionName(paired.direction)}方向。`,
};
if (paired.status !== 'ready') return {
kind: 'blocked', candidate: paired,
description: paired.missing[0]?.label || `${directionName(paired.direction)}转换证据还不完整。`,
};
return {
kind: 'create', candidate, label: '生成双向协议网关',
description: '请求和响应转换都已就绪,将合并为一个双向协议网关。',
};
}
+107 -10
View File
@@ -20,7 +20,8 @@ interface RawSnapshot {
const fixture = vi.hoisted(() => ({
storage: new Map<string, unknown>(),
storageFailure: undefined as Error | undefined,
pages: new Map<number, RawSnapshot>(),
pages: new Map<number | string, RawSnapshot>(),
frames: new Map<number, Array<{ tabId: number; frameId: number; documentId: string; url: string; accessible: boolean }>>(),
listeners: {} as Record<string, Listener>,
storageSet: vi.fn(),
}));
@@ -48,11 +49,16 @@ vi.mock('wxt/browser', () => {
runtime: { sendMessage: vi.fn(async () => undefined) },
scripting: {
executeScript: vi.fn(async (details: Record<string, any>) => {
if (details.files) return [{ frameId: 0 }];
const tabId = details.target.tabId as number;
const documentId = details.target.documentIds?.[0] as string | undefined;
const frameId = details.target.frameIds?.[0]
?? fixture.frames.get(tabId)?.find((frame) => frame.documentId === documentId)?.frameId
?? 0;
if (details.files) return [{ frameId }];
const command = details.args?.[2] as string;
const input = (details.args?.[3] || {}) as Record<string, unknown>;
const current = fixture.pages.get(tabId) || rawSnapshot(tabId);
const pageKey = frameId === 0 ? tabId : `${tabId}:${frameId}`;
const current = fixture.pages.get(pageKey) || rawSnapshot(tabId);
if (command === 'start') {
current.active = true;
current.recordingId = typeof input.recordingId === 'string' ? input.recordingId : `recording-${tabId}`;
@@ -67,8 +73,8 @@ vi.mock('wxt/browser', () => {
} else if (command === 'clear') {
Object.assign(current, rawSnapshot(tabId));
}
fixture.pages.set(tabId, current);
return [{ frameId: 0, result: clone(current) }];
fixture.pages.set(pageKey, current);
return [{ frameId, result: clone(current) }];
}),
},
tabs: {
@@ -84,13 +90,19 @@ vi.mock('wxt/browser', () => {
onCreated: event('created'),
},
cookies: {
getAllCookieStores: vi.fn(async () => [{ id: 'store-default', tabIds: [...fixture.pages.keys()] }]),
getAllCookieStores: vi.fn(async () => [{
id: 'store-default',
tabIds: [...new Set([...fixture.pages.keys()].map((key) => Number(String(key).split(':')[0])))],
}]),
},
webNavigation: {
getFrame: vi.fn(async ({ tabId }: { tabId: number }) => ({
url: `https://site-${tabId}.example.test/page`,
documentId: `document-${tabId}`,
})),
getFrame: vi.fn(async ({ tabId, frameId = 0 }: { tabId: number; frameId?: number }) => {
const frame = fixture.frames.get(tabId)?.find((item) => item.frameId === frameId);
return frame || {
url: `https://site-${tabId}.example.test/page`,
documentId: `document-${tabId}`,
};
}),
onBeforeNavigate: event('beforeNavigate'),
onCommitted: event('committed'),
onDOMContentLoaded: event('domContentLoaded'),
@@ -103,6 +115,25 @@ vi.mock('wxt/browser', () => {
};
});
vi.mock('@/features/page-context/frames', () => ({
getFrameInventory: vi.fn(async (tabId: number) => fixture.frames.get(tabId) || [{
tabId,
frameId: 0,
documentId: `document-${tabId}`,
parentFrameId: -1,
url: `https://site-${tabId}.example.test/page`,
origin: `https://site-${tabId}.example.test`,
title: 'Main frame',
name: '',
frameType: 'outermost_frame',
documentLifecycle: 'active',
isTop: true,
sameOrigin: true,
accessible: true,
sandbox: [],
}]),
}));
function rawSnapshot(tabId: number, events: Array<Record<string, unknown>> = []): RawSnapshot {
return {
version: 9,
@@ -159,6 +190,7 @@ describe('browser recording storage, snapshot and retained-value budgets', () =>
vi.setSystemTime(4_102_444_800_000);
fixture.storage.clear();
fixture.pages.clear();
fixture.frames.clear();
fixture.storageFailure = undefined;
fixture.storageSet.mockClear();
});
@@ -256,4 +288,69 @@ describe('browser recording storage, snapshot and retained-value budgets', () =>
globalSessionCount: 1,
});
});
it('records and merges requests from cross-origin child frames at tab scope', async () => {
const tabId = 30;
fixture.frames.set(tabId, [
{
tabId, frameId: 0, documentId: 'document-30', url: 'https://www.jd.com/', accessible: true,
},
{
tabId, frameId: 4, documentId: 'document-passport', url: 'https://passport.jd.com/new/login.aspx', accessible: true,
},
]);
fixture.pages.set(tabId, rawSnapshot(tabId, [recordingEvent(1)]));
fixture.pages.set(`${tabId}:4`, rawSnapshot(tabId, [{
...recordingEvent(2),
traceId: 'trace-passport-login',
url: 'https://passport.jd.com/uc/loginService?aksParamsU=redacted',
}]));
const service = await freshService();
const snapshot = await service.startTabBrowserRecording(tabId, { captureValues: true });
expect(snapshot.status).toMatchObject({ active: true, scope: 'tab' });
expect(snapshot.events).toEqual(expect.arrayContaining([
expect.objectContaining({ frameId: 0 }),
expect.objectContaining({
frameId: 4,
url: 'https://passport.jd.com/uc/loginService?aksParamsU=redacted',
}),
]));
expect(snapshot.traces.some((trace) => trace.label === 'POST /uc/loginService')).toBe(true);
});
it('automatically attaches a login frame created after recording starts', async () => {
const tabId = 31;
const top = {
tabId, frameId: 0, documentId: 'document-31', url: 'https://www.jd.com/', accessible: true,
};
fixture.frames.set(tabId, [top]);
fixture.pages.set(tabId, rawSnapshot(tabId));
const service = await freshService();
await service.startTabBrowserRecording(tabId);
fixture.frames.set(tabId, [top, {
tabId, frameId: 7, documentId: 'document-passport-late', url: 'https://passport.jd.com/new/login.aspx', accessible: true,
}]);
fixture.pages.set(`${tabId}:7`, rawSnapshot(tabId, [{
...recordingEvent(3),
traceId: 'trace-late-passport-login',
url: 'https://passport.jd.com/uc/loginService?aksParamsU=redacted',
}]));
fixture.listeners.committed({
tabId,
frameId: 7,
documentId: 'document-passport-late',
url: 'https://passport.jd.com/new/login.aspx',
timeStamp: Date.now(),
});
await vi.advanceTimersByTimeAsync(0);
const snapshot = await service.getTabBrowserRecording(tabId, 500, true);
expect(snapshot.events).toContainEqual(expect.objectContaining({
frameId: 7,
url: 'https://passport.jd.com/uc/loginService?aksParamsU=redacted',
}));
});
});
+187 -5
View File
@@ -9,6 +9,7 @@ import { ExtensionError } from '@/shared/errors';
import { inferBrowserTransformProfiles } from '@/features/browser-inference/inference';
import { normalizeBrowserRecordingCrypto } from '@/features/browser-crypto/model';
import { normalizeCallable } from '@/features/page-callable/service';
import { getFrameInventory } from '@/features/page-context/frames';
import { PAGE_RECORDER_PROTOCOL_VERSION, PAGE_RECORDER_REGISTRY_KEY } from './constants';
import {
buildRecordingLinks,
@@ -543,6 +544,7 @@ function normalizeEvent(value: unknown, allowSensitive: boolean): BrowserRecordi
parentEventId: optionalString(input.parentEventId, 160),
kind: input.kind as BrowserRecordingEvent['kind'],
source: input.source === 'browser' ? 'browser' : 'page',
frameId: Number.isSafeInteger(input.frameId) && Number(input.frameId) >= 0 ? Number(input.frameId) : undefined,
documentId: optionalString(input.documentId, 160),
operation: input.operation.slice(0, 160),
inputs: Array.isArray(input.inputs)
@@ -744,9 +746,11 @@ function snapshotFromEvents(
}
function snapshotFrom(target: BrowserTarget, raw: RawRecorderSnapshot): BrowserRecordingSnapshot {
const events = raw.events.map((event) => event.documentId || !target.documentId
? event
: { ...event, documentId: target.documentId });
const events = raw.events.map((event) => ({
...event,
frameId: target.frameId,
documentId: event.documentId || target.documentId,
}));
const callables = raw.callables
.map((item) => normalizeCallable(item, target))
.filter((item): item is BrowserPageCallable => Boolean(item));
@@ -792,6 +796,7 @@ function mergeSessionSnapshot(
globalSessionCount: previous?.status.globalSessionCount,
persistence: previous?.status.persistence,
persistenceError: previous?.status.persistenceError,
scope: previous?.status.scope || current.status.scope,
} : {}),
...status,
}, events, current.callables);
@@ -875,6 +880,7 @@ function applyNavigation(
parentEventId: existing?.parentEventId,
kind: 'navigation',
source: 'browser',
frameId: target.frameId,
documentId: navigation.previousDocumentId || existing?.documentId,
operation: navigationOperation(navigation),
label: navigationLabel(navigation),
@@ -904,6 +910,7 @@ export async function startBrowserRecording(
target: BrowserTarget,
input?: Partial<BrowserRecordingOptions>,
owner: OwnedRecording['owner'] = { kind: 'local' },
scope: BrowserRecordingStatus['scope'] = 'frame',
): Promise<BrowserRecordingSnapshot> {
if (expiredGrantOwner(owner)) throw new ExtensionError('grant_expired', '浏览器共享会话不存在或已经过期');
const options = normalizeOptions(input);
@@ -918,6 +925,7 @@ export async function startBrowserRecording(
if (!raw.startedAt) throw new ExtensionError('recorder_unavailable', '页面录制器尚未在目标文档就绪');
ownedRecordings.set(targetKey(target), { target, owner });
const snapshot = snapshotFrom(target, raw);
snapshot.status.scope = scope;
snapshot.status.isolationContextId = tab.isolationContextId;
snapshot.status.cookieStoreId = tab.cookieStoreId;
snapshot.status.pageUrl = await currentPageUrl(target);
@@ -980,6 +988,167 @@ export async function getBrowserRecording(target: BrowserTarget, limit = MAX_ENT
return recordingSnapshotForScope(snapshot, allowSensitive);
}
function emptyTabRecording(tabId: number): BrowserRecordingSnapshot {
return {
status: {
active: false,
scope: 'tab',
target: { tabId, frameId: 0 },
documentAvailable: true,
count: 0,
droppedCount: 0,
},
events: [],
traces: [],
links: [],
callables: [],
profileCandidates: [],
};
}
function localTabSnapshots(tabId: number): BrowserRecordingSnapshot[] {
return [...latestSnapshots.entries()].flatMap(([key, snapshot]) => {
if (snapshot.status.target.tabId !== tabId) return [];
const owner = ownedRecordings.get(key)?.owner || sessionOwners.get(key);
return owner?.kind === 'grant' ? [] : [snapshot];
});
}
async function tabFrameTargets(tabId: number): Promise<BrowserTarget[]> {
const frames = await getFrameInventory(tabId);
return frames
.filter((frame) => frame.accessible && /^https?:/i.test(frame.url))
.slice(0, RECORDING_MAX_SESSIONS)
.map(({ frameId, documentId }) => ({ tabId, frameId, documentId }));
}
function summedStatus(
snapshots: BrowserRecordingSnapshot[],
key: 'droppedCount' | 'budgetDroppedCount' | 'previewDroppedCount' | 'retainedBytes'
| 'retainedPreviewBytes' | 'retainedCallCount' | 'retainedCallBytes' | 'retainedCallDroppedCount',
): number {
return snapshots.reduce((total, snapshot) => total + (snapshot.status[key] || 0), 0);
}
function mergedPersistence(snapshots: BrowserRecordingSnapshot[]): BrowserRecordingStatus['persistence'] {
const states = snapshots.map((snapshot) => snapshot.status.persistence);
if (states.includes('degraded')) return 'degraded';
if (states.includes('memory-only')) return 'memory-only';
if (states.includes('pending')) return 'pending';
return states.includes('persisted') ? 'persisted' : undefined;
}
export function mergeTabRecordingSnapshots(
tabId: number,
snapshots: BrowserRecordingSnapshot[],
limit = MAX_ENTRIES,
): BrowserRecordingSnapshot {
if (!snapshots.length) return emptyTabRecording(tabId);
const primary = snapshots.find((snapshot) => snapshot.status.target.frameId === 0) || snapshots[0];
const byEventId = new Map<string, BrowserRecordingEvent>();
for (const snapshot of snapshots) {
for (const event of snapshot.events) {
byEventId.set(event.id, { ...event, frameId: event.frameId ?? snapshot.status.target.frameId });
}
}
const events = [...byEventId.values()]
.sort((left, right) => left.timestamp - right.timestamp || left.sequence - right.sequence || left.id.localeCompare(right.id))
.slice(-Math.max(1, Math.min(Math.floor(limit), MAX_ENTRIES)));
const eventIds = new Set(events.map((event) => event.id));
const links = buildRecordingLinks(events);
const callables = [...new Map(snapshots.flatMap((snapshot) => snapshot.callables).map((item) => [item.id, item])).values()];
const profileCandidates = [...new Map(snapshots.flatMap((snapshot) => snapshot.profileCandidates)
.filter((candidate) => eventIds.has(candidate.request.eventId) && candidate.sources.every((source) => eventIds.has(source.eventId)))
.map((candidate) => [candidate.id, candidate])).values()];
const startedAt = snapshots.reduce<number | undefined>((oldest, snapshot) => {
const next = snapshot.status.startedAt;
return next === undefined ? oldest : oldest === undefined ? next : Math.min(oldest, next);
}, undefined);
return {
status: {
...primary.status,
active: snapshots.some((snapshot) => snapshot.status.active),
scope: 'tab',
target: primary.status.target,
startedAt,
count: events.length,
droppedCount: summedStatus(snapshots, 'droppedCount'),
budgetDroppedCount: summedStatus(snapshots, 'budgetDroppedCount'),
previewDroppedCount: summedStatus(snapshots, 'previewDroppedCount'),
retainedBytes: summedStatus(snapshots, 'retainedBytes'),
retainedPreviewBytes: summedStatus(snapshots, 'retainedPreviewBytes'),
retainedCallCount: summedStatus(snapshots, 'retainedCallCount'),
retainedCallBytes: summedStatus(snapshots, 'retainedCallBytes'),
retainedCallDroppedCount: summedStatus(snapshots, 'retainedCallDroppedCount'),
globalRetainedBytes: Math.max(...snapshots.map((snapshot) => snapshot.status.globalRetainedBytes || 0)),
globalSessionCount: Math.max(...snapshots.map((snapshot) => snapshot.status.globalSessionCount || 0)),
persistence: mergedPersistence(snapshots),
persistenceError: snapshots.find((snapshot) => snapshot.status.persistenceError)?.status.persistenceError,
endedReason: snapshots.some((snapshot) => snapshot.status.active) ? undefined : primary.status.endedReason,
},
events,
links,
traces: buildRecordingTraces(events, links),
callables,
profileCandidates,
};
}
export async function startTabBrowserRecording(
tabId: number,
input?: Partial<BrowserRecordingOptions>,
): Promise<BrowserRecordingSnapshot> {
await ensureSessionsRestored();
await Promise.all(localTabSnapshots(tabId).map((snapshot) => clearBrowserRecording(snapshot.status.target)));
const targets = await tabFrameTargets(tabId);
const top = targets.find((target) => target.frameId === 0);
if (!top) throw new ExtensionError('target_unavailable', '标签页主文档当前不可录制');
const snapshots = [await startBrowserRecording(top, input, { kind: 'local' }, 'tab')];
const children = await Promise.allSettled(targets
.filter((target) => target.frameId !== 0)
.map((target) => startBrowserRecording(target, input, { kind: 'local' }, 'tab')));
snapshots.push(...children.flatMap((result) => result.status === 'fulfilled' ? [result.value] : []));
return mergeTabRecordingSnapshots(tabId, snapshots);
}
export async function getTabBrowserRecording(
tabId: number,
limit = MAX_ENTRIES,
allowSensitive = false,
): Promise<BrowserRecordingSnapshot> {
await ensureSessionsRestored();
const stored = localTabSnapshots(tabId);
const snapshots = await Promise.all(stored.map(async (snapshot) => (
getBrowserRecording(snapshot.status.target, limit, allowSensitive).catch(() => recordingSnapshotForScope(snapshot, allowSensitive))
)));
return mergeTabRecordingSnapshots(tabId, snapshots, limit);
}
export async function tabBrowserRecordingStatus(tabId: number): Promise<BrowserRecordingStatus> {
return (await getTabBrowserRecording(tabId, MAX_ENTRIES, false)).status;
}
export async function stopTabBrowserRecording(
tabId: number,
allowSensitive = false,
): Promise<BrowserRecordingSnapshot> {
await ensureSessionsRestored();
const snapshots = await Promise.all(localTabSnapshots(tabId).map((snapshot) => (
stopBrowserRecording(snapshot.status.target, allowSensitive)
)));
return mergeTabRecordingSnapshots(tabId, snapshots);
}
export async function clearTabBrowserRecording(
tabId: number,
allowSensitive = false,
): Promise<BrowserRecordingSnapshot> {
await ensureSessionsRestored();
const snapshots = localTabSnapshots(tabId);
await Promise.all(snapshots.map((snapshot) => clearBrowserRecording(snapshot.status.target, allowSensitive)));
return emptyTabRecording(tabId);
}
export async function clearBrowserRecording(target: BrowserTarget, allowSensitive = false): Promise<BrowserRecordingSnapshot> {
const raw = normalizeRawSnapshot(await executeCommand(target, 'clear').catch(() => ({
version: PAGE_RECORDER_PROTOCOL_VERSION, active: false, count: 0, droppedCount: 0, events: [], callables: [],
@@ -1010,7 +1179,7 @@ export async function stopBrowserRecording(target: BrowserTarget, allowSensitive
export async function createRecordedPageCallable(
target: BrowserTarget,
input: { callHandleId: string; name: string },
input: { callHandleId: string; name: string; dynamicInputPaths?: string[] },
): Promise<BrowserPageCallable> {
const raw = await executeCommand(target, 'callable.create', input);
const callable = normalizeCallable(raw, target);
@@ -1151,9 +1320,22 @@ async function continueRecordingOnDocument(
documentId: details.documentId,
};
const key = targetKey(target);
const stored = await readSession(target);
let stored = await readSession(target);
if (!stored && details.frameId !== 0 && /^https?:/i.test(details.url)) {
const top = await readSession({ tabId: details.tabId, frameId: 0 });
const topKey = targetKey({ tabId: details.tabId, frameId: 0 });
const owner = ownedRecordings.get(topKey)?.owner || top?.owner;
if (top?.snapshot.status.active && top.snapshot.status.scope === 'tab' && owner?.kind !== 'grant') {
const snapshot = await startBrowserRecording(target, top.snapshot.status.options, { kind: 'local' }, 'tab');
notifyRecordingChanged(details.tabId, 'updated');
stored = { snapshot, owner: { kind: 'local' } };
}
}
if (!stored?.snapshot.status.active || !stored.snapshot.status.recordingId) return;
const previous = stored.snapshot;
if (!previous.status.navigation
&& previous.status.target.documentId === details.documentId
&& previous.status.pageUrl === details.url) return;
const previousNavigation = previous.status.navigation;
const hasTransitionEvidence = Boolean(details.transitionType || details.transitionQualifiers?.length);
const kind = hasTransitionEvidence
@@ -10,7 +10,7 @@ import type {
ActiveTabInfo, BrowserPageCallable, BrowserRecordingEvent, BrowserTransformBuiltinOperation,
BrowserTransformDirection,
BrowserTransformNodeReference, BrowserTransformPipelineNode, BrowserTransformProfile,
BrowserTransformProfileInput, BrowserProfileInferenceCandidate,
BrowserTransformProfileInput, BrowserProfileInferenceCandidate, BrowserTransformValidationDraft,
} from '@/types/models';
import {
callableEnvelopeDescription, compileGuidedTransform, defaultGuidedTransform, guidedOutputDescription, parseGuidedTransform,
@@ -56,7 +56,7 @@ interface BrowserTransformWorkspaceProps {
export interface BrowserTransformSuggestionSeed {
revision: number;
candidate: BrowserProfileInferenceCandidate;
callable: BrowserPageCallable;
callables: BrowserPageCallable[];
profile: BrowserTransformProfile;
sampleBody?: string;
sampleLabel?: string;
@@ -232,6 +232,11 @@ function callableKindLabel(callable: BrowserPageCallable): string {
return '全局函数';
}
function profileDirectionLabel(profile: Pick<BrowserTransformProfileInput, 'request' | 'response'>): string {
if (profile.request.enabled && profile.response.enabled) return '双向协议网关';
return profile.request.enabled ? '仅请求转换' : '仅响应转换';
}
function referencesOf(node: BrowserTransformPipelineNode): BrowserTransformNodeReference[] {
if (node.kind === 'builtin') return node.inputs;
if (node.kind === 'page.call') return node.arguments;
@@ -257,6 +262,7 @@ export function BrowserTransformWorkspace({
INITIAL_TRANSFORM_WORKSPACE_STATE,
);
const [workspaceView, setWorkspaceView] = useState<'flow' | 'configure'>('flow');
const [pendingValidation, setPendingValidation] = useState<BrowserTransformValidationDraft | null>(null);
const {
profiles, callables, selectedProfileId, draft, directionName, loadError,
testMethod, testUrl, testHeaders, testBody, testSample, testResult, testError,
@@ -351,11 +357,28 @@ export function BrowserTransformWorkspace({
}
}, [tab]);
const loadPendingValidation = useCallback(async () => {
if (!tab) {
setPendingValidation(null);
return;
}
try {
setPendingValidation(await request('analysis.profile.validation.latest', { tabId: tab.id, frameId: 0 }));
} catch {
setPendingValidation(null);
}
}, [tab]);
useEffect(() => {
workspaceMounted.current = true;
return () => { workspaceMounted.current = false; };
}, []);
useEffect(() => { void load(); }, [load]);
useEffect(() => {
void loadPendingValidation();
const timer = setInterval(() => void loadPendingValidation(), 2_000);
return () => clearInterval(timer);
}, [loadPendingValidation]);
useEffect(() => {
if (recoveryRevision > 0) void load();
}, [load, recoveryRevision]);
@@ -498,8 +521,8 @@ export function BrowserTransformWorkspace({
update: (current) => ({
...current,
callables: [
...current.callables.filter((item) => item.id !== suggestion.callable.id),
suggestion.callable,
...current.callables.filter((item) => !suggestion.callables.some((callable) => callable.id === item.id)),
...suggestion.callables,
],
profiles: [
suggestion.profile,
@@ -731,6 +754,22 @@ export function BrowserTransformWorkspace({
await load();
}, '已取消本次恢复结果,旧网关继续保持停用');
const resolvePendingValidation = (outcome: 'save' | 'discard') => run(async () => {
if (!tab || !pendingValidation) return;
const profile = await request('analysis.profile.validation.resolve', {
tabId: tab.id,
frameId: 0,
validationId: pendingValidation.id,
outcome,
});
setPendingValidation(null);
if (!profile) return;
setProfiles((current) => [profile, ...current.filter((item) => item.id !== profile.id)]);
setSelectedProfileId(profile.id);
setDraft(toInput(profile));
setWorkspaceView('flow');
}, outcome === 'save' ? '明文网关已保存' : '验证草稿已放弃');
const execute = async () => {
if (!draft?.id || dirty) { setTestError('请先保存当前 Pipeline'); return; }
setTestError('');
@@ -762,6 +801,18 @@ export function BrowserTransformWorkspace({
/>
<main className="transform-editor">
{pendingValidation && <section className="transform-validation-pending" role="status">
<span className="transform-validation-pending__mark"><CheckCircle2 size={16} /></span>
<div>
<small>Agent 已完成本地验证 · {pendingValidation.proofLevel === 'exact' ? '报文一致' : pendingValidation.proofLevel === 'structure' ? '结构一致' : '执行通过'}</small>
<strong>{pendingValidation.profile.name}</strong>
<p>{pendingValidation.profile.origin} · {profileDirectionLabel(pendingValidation.profile)} · {Math.max(1, Math.ceil((pendingValidation.expiresAt - Date.now()) / 60_000))} 分钟后过期</p>
</div>
<div className="transform-validation-pending__actions">
<Button size="sm" variant="ghost" disabled={busy} onClick={() => void resolvePendingValidation('discard')}>放弃</Button>
<Button size="sm" variant="primary" disabled={busy} onClick={() => void resolvePendingValidation('save')}><Save size={13} />确认保存</Button>
</div>
</section>}
{!draft ? <div className="transform-editor-empty"><Link2 size={24} /><strong>建立明文与线上报文的转换链路</strong>{callables.length ? <Button variant="primary" onClick={create}><CirclePlus size={14} />新建 Pipeline</Button> : <Button variant="primary" onClick={onOpenCapture}><Code2 size={14} />{deepCaptureAvailable ? '先捕获页面函数' : '回到录制并保存页面函数'}</Button>}</div> : <>
<header className="transform-editor-head">
<div>{workspaceView === 'flow' && savedProfile
@@ -1,8 +1,9 @@
import type { ComponentType } from 'react';
import { useEffect, useMemo, useState, type ComponentType } from 'react';
import {
ArrowDownToLine,
Braces,
CheckCircle2,
ChevronDown,
Code2,
FileInput,
KeyRound,
@@ -20,6 +21,12 @@ import type {
BrowserTransformValueSummary,
} from '@/types/models';
interface FlowStageItem {
id: string;
stage: BrowserTransformExplanationStage;
members: BrowserTransformExplanationStage[];
}
const OWNER_LABELS: Record<BrowserTransformExplanationOwner, string> = {
webfuzzer: 'Web Fuzzer',
extension: '浏览器扩展',
@@ -69,6 +76,44 @@ function operationLabel(operation: BrowserTransformExplanationStage['operations'
return details.join(' · ');
}
function displayStages(
stages: BrowserTransformExplanationStage[],
direction: BrowserTransformDirectionName,
): FlowStageItem[] {
const items: FlowStageItem[] = [];
for (const stage of stages) {
const assembly = stage.owner === 'extension' && (stage.kind === 'builtin' || stage.kind === 'output');
const previous = items[items.length - 1];
if (assembly && previous?.members.every((item) => (
item.owner === 'extension' && (item.kind === 'builtin' || item.kind === 'output')
))) {
previous.members.push(stage);
continue;
}
items.push({ id: stage.id, stage, members: [stage] });
}
return items.map((item) => {
if (item.members.length === 1) return item;
const first = item.members[0];
const last = item.members[item.members.length - 1];
return {
...item,
id: `${first.id}:assembly`,
stage: {
...first,
id: `${first.id}:assembly`,
title: direction === 'request' ? '浏览器扩展组装线上请求' : '浏览器扩展还原逻辑响应',
summary: `${item.members.length} 个受限步骤,将中间结果写入最终报文`,
nodeIds: item.members.flatMap((member) => member.nodeIds),
inputPaths: first.inputPaths,
outputPaths: last.outputPaths,
operations: item.members.flatMap((member) => member.operations),
evidence: item.members.flatMap((member) => member.evidence),
},
};
});
}
export function TransformDataFlowView({
profile,
direction,
@@ -85,6 +130,13 @@ export function TransformDataFlowView({
const explained = profile.explanation?.directions.find((item) => item.direction === direction);
const currentExecution = execution?.direction === direction ? execution : undefined;
const availableDirections = profile.explanation?.directions.map((item) => item.direction) || [];
const stages = useMemo(() => displayStages(explained?.stages || [], direction), [direction, explained?.stages]);
const defaultOpenStageId = stages.find((item) => item.members.some((member) => member.kind === 'page-call'))?.id || '';
const [openStageId, setOpenStageId] = useState('');
useEffect(() => {
setOpenStageId(defaultOpenStageId);
}, [defaultOpenStageId, direction, profile.id]);
if (!explained) return <div className="transform-flow-empty">
<Code2 size={22} />
@@ -97,7 +149,9 @@ export function TransformDataFlowView({
<div>
<span className="transform-data-flow__eyebrow">明文网关 · {direction === 'request' ? '请求方向' : '响应方向'}</span>
<strong>{direction === 'request' ? '明文如何成为线上请求' : '线上响应如何还原为明文'}</strong>
<p>{explained.summary}</p>
<p>{stages.length === explained.stages.length
? explained.summary
: `${explained.stages.length} 个处理步骤已收拢为 ${stages.length} 个主要阶段`}</p>
</div>
{availableDirections.length > 1 && <div className="transform-flow-directions" role="tablist" aria-label="数据流方向">
{availableDirections.map((item) => <button
@@ -119,17 +173,24 @@ export function TransformDataFlowView({
</div>
<div className="transform-flow-timeline">
{explained.stages.map((stage, index) => {
{stages.map((item, index) => {
const { stage } = item;
const Icon = STAGE_ICONS[stage.kind];
const traces = currentExecution?.nodeTrace.filter((trace) => stage.nodeIds.includes(trace.nodeId)) || [];
const stageDuration = traces.reduce((total, trace) => total + trace.durationMs, 0);
const hasDetails = Boolean(stage.inputPaths.length || stage.outputPaths.length || stage.operations.length
|| stage.evidence.length || stage.network || stage.source || traces.length);
return <details className={`transform-flow-stage is-${stage.owner}`} key={stage.id} open={stage.kind === 'page-call'}>
<summary>
return <details className={`transform-flow-stage is-${stage.owner}`} key={item.id} open={openStageId === item.id}>
<summary
aria-expanded={openStageId === item.id}
onClick={(event) => {
event.preventDefault();
if (hasDetails) setOpenStageId((current) => current === item.id ? '' : item.id);
}}
>
<span className="transform-flow-stage__rail">
<i><Icon size={15} /></i>
{index < explained.stages.length - 1 && <b />}
{index < stages.length - 1 && <b />}
</span>
<span className="transform-flow-stage__main">
<span className="transform-flow-stage__meta"><em>{OWNER_LABELS[stage.owner]}</em><i className={`is-${stage.proof}`}>{proofLabel(stage)}</i></span>
@@ -137,16 +198,25 @@ export function TransformDataFlowView({
<small>{stage.summary}</small>
</span>
<span className="transform-flow-stage__status">
{traces.length ? <><CheckCircle2 size={14} /><time>{stageDuration.toFixed(1)} ms</time></> : hasDetails ? <span>详情</span> : null}
{traces.length ? <><CheckCircle2 size={14} /><time>{stageDuration.toFixed(1)} ms</time></> : null}
{hasDetails && <ChevronDown className="transform-flow-stage__chevron" size={14} />}
</span>
</summary>
{hasDetails && <div className="transform-flow-stage__details">
{item.members.length > 1 && <div className="transform-flow-steps">
<span>阶段内操作</span>
<ol>{item.members.map((member, memberIndex) => <li key={member.id}>
<i>{memberIndex + 1}</i>
<span><strong>{member.title}</strong><small>{member.operations.map(operationLabel).join(' · ') || member.summary}</small></span>
<code>{[member.inputPaths.join('、'), member.outputPaths.join('、')].filter(Boolean).join(' → ')}</code>
</li>)}</ol>
</div>}
{stage.network && <dl className="transform-flow-network">
<div><dt>网络边界</dt><dd><code>{stage.network.method}</code> {stage.network.route}</dd></div>
{stage.network.statusCode && <div><dt>录制响应</dt><dd>{stage.network.statusCode}</dd></div>}
</dl>}
{stage.operations.length > 0 && <div className="transform-flow-facts"><span>处理</span><ul>{stage.operations.map((operation, operationIndex) => <li key={`${operation.operation}:${operationIndex}`}><strong>{operationLabel(operation)}</strong>{operation.destination && <code>→ {operation.destination}</code>}</li>)}</ul></div>}
{(stage.inputPaths.length > 0 || stage.outputPaths.length > 0) && <div className="transform-flow-paths">
{item.members.length === 1 && stage.operations.length > 0 && <div className="transform-flow-facts"><span>处理逻辑</span><ul>{stage.operations.map((operation, operationIndex) => <li key={`${operation.operation}:${operationIndex}`}><strong>{operationLabel(operation)}</strong>{operation.destination && <code>→ {operation.destination}</code>}</li>)}</ul></div>}
{item.members.length === 1 && (stage.inputPaths.length > 0 || stage.outputPaths.length > 0) && <div className="transform-flow-paths">
{stage.inputPaths.length > 0 && <div><span>输入</span><p>{stage.inputPaths.map((path) => <code key={path}>{path}</code>)}</p></div>}
{stage.outputPaths.length > 0 && <div><span>输出</span><p>{stage.outputPaths.map((path) => <code key={path}>{path}</code>)}</p></div>}
</div>}
@@ -54,6 +54,9 @@ export function TransformProfileRail({
<header><div><strong>明文网关</strong><span>{profiles.length}</span></div><Button size="icon" variant="ghost" aria-label="新建 Pipeline" title="新建 Pipeline" disabled={!tab} onClick={onCreate}><Plus size={15} /></Button></header>
<div className="transform-profile-list">
{profiles.map((profile) => {
const directionLabel = profile.request.enabled && profile.response.enabled
? '双向'
: profile.request.enabled ? '仅请求' : '仅响应';
const ready = (!profile.recovery || profile.recovery.state === 'ready')
&& originOf(tab?.url) === profile.origin && [profile.request, profile.response]
.flatMap((item) => item.enabled ? item.nodes : [])
@@ -61,7 +64,7 @@ export function TransformProfileRail({
.every((node) => callableIds.has(node.callableId));
return <button key={profile.id} className={selectedProfileId === profile.id ? 'is-selected' : ''} onClick={() => onSelect(profile)}>
<span className={`transform-profile-mark ${ready ? 'is-ready' : ''}`}><FileKey2 size={14} /></span>
<span><strong>{profile.name}</strong><small>{profile.match.methods.join(' / ') || 'ANY'} · {profile.match.urlPattern}</small></span>
<span><strong>{profile.name}</strong><small>{directionLabel} · {profile.match.methods.join(' / ') || 'ANY'} · {profile.match.urlPattern}</small></span>
<i title={ready ? '页面绑定可用' : '页面函数已失效'}>{ready ? <CheckCircle2 size={13} /> : <Unplug size={13} />}</i>
</button>;
})}
@@ -63,6 +63,16 @@
.transform-callable-confirm > div { display: flex; justify-content: flex-end; gap: 6px; }
.transform-editor { max-height: 820px; overflow: auto; display: grid; align-content: start; border-right: 1px solid var(--border); }
.transform-validation-pending { min-width: 0; min-height: 72px; padding: 11px 14px; display: grid; grid-template-columns: 30px minmax(0, 1fr) auto; align-items: center; gap: 10px; border-bottom: 1px solid color-mix(in srgb, var(--success) 28%, var(--border)); background: color-mix(in srgb, var(--success-soft) 68%, var(--surface)); }
.transform-validation-pending__mark { width: 30px; height: 30px; display: grid; place-items: center; border-radius: 50%; background: var(--success-soft); color: var(--success); }
.transform-validation-pending > div { min-width: 0; }
.transform-validation-pending small,
.transform-validation-pending strong,
.transform-validation-pending p { display: block; margin: 0; overflow: hidden; text-overflow: ellipsis; white-space: nowrap; }
.transform-validation-pending small { color: var(--success); font-size: 10px; font-weight: 650; }
.transform-validation-pending strong { margin-top: 3px; font-size: var(--text-sm); }
.transform-validation-pending p { margin-top: 2px; color: var(--muted); font-size: var(--text-xs); }
.transform-validation-pending__actions { display: flex; align-items: center; gap: 6px; }
.transform-editor-empty { min-height: 520px; display: flex; flex-direction: column; align-items: center; justify-content: center; gap: 11px; color: var(--muted); text-align: center; }
.transform-editor-empty strong { color: var(--foreground); font-size: var(--text-md); }
.transform-editor-head { min-height: 64px; padding: 10px 14px; display: flex; align-items: center; justify-content: space-between; gap: 14px; border-bottom: 1px solid var(--border); }
@@ -255,24 +265,28 @@
.transform-flow-stage.is-extension { --owner-color: var(--primary); }
.transform-flow-stage.is-page { --owner-color: #2563eb; }
.transform-flow-stage.is-yak { --owner-color: #6b7280; }
.transform-flow-stage > summary { min-width: 0; min-height: 92px; display: grid; grid-template-columns: 34px minmax(0, 1fr) auto; gap: 11px; list-style: none; cursor: pointer; }
.transform-flow-stage > summary { min-width: 0; min-height: 78px; display: grid; grid-template-columns: 34px minmax(0, 1fr) auto; gap: 11px; border-radius: var(--radius-md); list-style: none; cursor: pointer; transition: background-color .14s ease, box-shadow .14s ease; }
.transform-flow-stage > summary::-webkit-details-marker { display: none; }
.transform-flow-stage__rail { min-height: 92px; display: grid; grid-template-rows: 30px minmax(0, 1fr); justify-items: center; padding-top: 16px; }
.transform-flow-stage > summary:hover { background: color-mix(in srgb, var(--owner-color) 4%, transparent); }
.transform-flow-stage[open] > summary { background: color-mix(in srgb, var(--owner-color) 7%, var(--surface)); box-shadow: inset 3px 0 0 var(--owner-color); }
.transform-flow-stage__rail { min-height: 78px; display: grid; grid-template-rows: 30px minmax(0, 1fr); justify-items: center; padding-top: 13px; }
.transform-flow-stage__rail > i { width: 28px; height: 28px; display: grid; place-items: center; border: 1px solid color-mix(in srgb, var(--owner-color) 30%, var(--border)); border-radius: 50%; background: color-mix(in srgb, var(--owner-color) 8%, var(--surface)); color: var(--owner-color); font-style: normal; }
.transform-flow-stage__rail > b { width: 1px; min-height: 38px; background: color-mix(in srgb, var(--owner-color) 28%, var(--border)); }
.transform-flow-stage__main { min-width: 0; padding: 15px 0 13px; border-bottom: 1px solid var(--border); }
.transform-flow-stage__main { min-width: 0; padding: 12px 0 11px; border-bottom: 1px solid var(--border); }
.transform-flow-stage__meta { display: flex; align-items: center; gap: 7px; }
.transform-flow-stage__meta > em { color: var(--owner-color); font-size: 10px; font-style: normal; font-weight: 700; }
.transform-flow-stage__meta > i { padding: 1px 5px; border-radius: 999px; background: var(--surface-subtle); color: var(--muted); font-size: 9px; font-style: normal; }
.transform-flow-stage__meta > em { color: var(--owner-color); font-size: var(--text-xs); font-style: normal; font-weight: 700; }
.transform-flow-stage__meta > i { padding: 1px 5px; border-radius: 999px; background: var(--surface-subtle); color: var(--muted); font-size: var(--text-xs); font-style: normal; }
.transform-flow-stage__meta > i.is-observed { background: var(--success-soft); color: var(--success); }
.transform-flow-stage__meta > i.is-supported { background: var(--warning-soft); color: var(--warning); }
.transform-flow-stage__main > strong { display: block; margin-top: 5px; font-size: var(--text-sm); }
.transform-flow-stage__main > small { display: block; margin-top: 3px; color: var(--muted); font-size: var(--text-xs); line-height: 1.45; }
.transform-flow-stage__status { min-width: 54px; padding: 15px 0 0 8px; display: flex; align-items: flex-start; justify-content: flex-end; gap: 4px; color: var(--muted); font-size: 10px; }
.transform-flow-stage__status { min-width: 70px; padding: 13px 10px 0 8px; display: flex; align-items: center; justify-content: flex-end; gap: 5px; color: var(--muted); font-size: var(--text-xs); }
.transform-flow-stage__status > svg { color: var(--success); }
.transform-flow-stage__status time { font-variant-numeric: tabular-nums; }
.transform-flow-stage[open] .transform-flow-stage__status > span { color: var(--foreground); }
.transform-flow-stage__details { margin: -6px 0 10px 45px; padding: 0 0 14px; display: grid; gap: 10px; border-bottom: 1px solid var(--border); }
.transform-flow-stage__status > .transform-flow-stage__chevron { color: var(--muted); transition: transform .16s ease; }
.transform-flow-stage[open] .transform-flow-stage__chevron { transform: rotate(180deg); }
.transform-flow-stage__details { margin: 4px 10px 14px 45px; padding: 13px 14px; display: grid; gap: 11px; border: 1px solid color-mix(in srgb, var(--owner-color) 16%, var(--border)); border-radius: var(--radius-md); background: var(--surface); box-shadow: 0 4px 14px rgb(15 23 42 / 4%); }
.transform-flow-network { margin: 0; padding: 8px 10px; display: grid; gap: 5px; background: var(--surface-subtle); }
.transform-flow-network > div { min-width: 0; display: grid; grid-template-columns: 78px minmax(0, 1fr); gap: 8px; font-size: var(--text-xs); }
.transform-flow-network dt { color: var(--muted); }
@@ -305,6 +319,18 @@
.transform-flow-runtime li strong { overflow: hidden; text-overflow: ellipsis; white-space: nowrap; }
.transform-flow-runtime li code { color: var(--muted); }
.transform-flow-runtime li time { color: var(--success); text-align: right; font-variant-numeric: tabular-nums; }
.transform-flow-steps { min-width: 0; display: grid; grid-template-columns: 78px minmax(0, 1fr); gap: 8px; }
.transform-flow-steps > span { color: var(--muted); font-size: var(--text-xs); font-weight: 650; }
.transform-flow-steps ol { min-width: 0; margin: 0; padding: 0; display: grid; gap: 2px; list-style: none; }
.transform-flow-steps li { min-width: 0; min-height: 42px; padding: 6px 8px; display: grid; grid-template-columns: 20px minmax(130px, .8fr) minmax(160px, 1fr); align-items: center; gap: 8px; border-bottom: 1px solid var(--border); }
.transform-flow-steps li:last-child { border-bottom: 0; }
.transform-flow-steps li > i { width: 20px; height: 20px; display: grid; place-items: center; border-radius: 50%; background: color-mix(in srgb, var(--owner-color) 9%, var(--surface)); color: var(--owner-color); font-size: var(--text-xs); font-style: normal; font-weight: 700; }
.transform-flow-steps li > span { min-width: 0; }
.transform-flow-steps li strong,
.transform-flow-steps li small { display: block; overflow: hidden; text-overflow: ellipsis; white-space: nowrap; }
.transform-flow-steps li strong { font-size: var(--text-xs); }
.transform-flow-steps li small { margin-top: 2px; color: var(--muted); font-size: var(--text-xs); }
.transform-flow-steps li > code { min-width: 0; overflow: hidden; color: var(--muted-strong); font-size: var(--text-xs); text-overflow: ellipsis; white-space: nowrap; }
.transform-flow-changes { margin: 14px 0 0 45px; border-top: 2px solid var(--foreground); }
.transform-flow-changes > header { min-height: 54px; display: flex; align-items: center; justify-content: space-between; gap: 10px; border-bottom: 1px solid var(--border); }
.transform-flow-changes > header strong,
@@ -422,6 +448,8 @@
.transform-route label:last-child { grid-column: 1 / -1; }
.transform-recovery { grid-template-columns: 30px minmax(0, 1fr); }
.transform-recovery__actions { grid-column: 2; justify-content: flex-start; flex-wrap: wrap; }
.transform-validation-pending { grid-template-columns: 30px minmax(0, 1fr); }
.transform-validation-pending__actions { grid-column: 2; justify-content: flex-start; }
.transform-step-fields { grid-template-columns: minmax(0, 1fr); }
.transform-step-fields .transform-step-name { grid-column: auto; }
.transform-output-list > div { grid-template-columns: minmax(0, 1fr) 14px minmax(0, 1fr) 32px; }
@@ -438,6 +466,9 @@
.transform-data-flow { padding-inline: 12px; }
.transform-data-flow__head { flex-direction: column; }
.transform-flow-paths { grid-template-columns: minmax(0, 1fr); }
.transform-flow-steps { grid-template-columns: minmax(0, 1fr); }
.transform-flow-steps li { grid-template-columns: 20px minmax(0, 1fr); }
.transform-flow-steps li > code { grid-column: 2; }
.transform-flow-stage__details,
.transform-flow-changes { margin-left: 34px; }
.transform-flow-empty { grid-template-columns: 30px minmax(0, 1fr); }
@@ -76,6 +76,38 @@ describe('guided browser transform compiler', () => {
}).inputPaths).toEqual(['body', 'body.options']);
});
it('decodes CryptoJS decrypt WordArray hex before writing the plaintext body', async () => {
const decryptCallable: BrowserPageCallable = {
...callable,
id: 'decrypt-aes',
operation: 'cryptojs.AES.decrypt',
crypto: {
adapterId: 'cryptojs', providerKind: 'library', family: 'symmetric',
operation: 'AES.decrypt', algorithm: 'AES.decrypt', outputEncoding: 'hex',
},
output: { dataType: 'Object', encoding: 'hex', shape: 'value', paths: [] },
};
const direction = compileGuidedTransform(defaultGuidedTransform(decryptCallable), decryptCallable);
const result = await executeTransformDirection('profile-decrypt', 'response', direction, {
method: 'POST',
url: 'https://example.test/login',
headers: [],
bodyBase64: bodyBase64('cipher'),
}, async (callableId) => ({
callableId,
type: 'object',
preview: '7b226f6b223a747275657d',
value: '7b226f6b223a747275657d',
durationMs: 1,
}));
expect(decodeBody(result.bodyBase64)).toBe('{"ok":true}');
expect(parseGuidedTransform(direction, [decryptCallable])).toMatchObject({
callableId: decryptCallable.id,
outputKind: 'body',
});
});
it('compiles a form field and its content type without exposing DAG details', async () => {
const guide = {
...defaultGuidedTransform(callable, { outputKind: 'form-field', outputField: 'encryptedData' }),
+22 -3
View File
@@ -106,8 +106,21 @@ export function compileGuidedTransform(guide: GuidedTransformDraft, callable?: B
callableId: guide.callableId,
arguments: inputNodes.map((node) => ({ nodeId: node.id })),
};
const callReference = { nodeId: callId, path: guide.resultPath?.trim() || undefined };
const nodes: BrowserTransformPipelineNode[] = [...inputNodes, callNode];
let callReference = { nodeId: callId, path: guide.resultPath?.trim() || undefined };
if (callable?.crypto?.adapterId === 'cryptojs'
&& callable.crypto.operation.toLowerCase().includes('decrypt')
&& callable.output.encoding === 'hex') {
const decodeId = uid('decode');
nodes.push({
id: decodeId,
name: '还原 CryptoJS 解密字节',
kind: 'builtin',
operation: 'hex.decode',
inputs: [callReference],
});
callReference = { nodeId: decodeId, path: undefined };
}
const bodyFormat = envelopeBodyFormat(callable);
if (bodyFormat) {
@@ -243,8 +256,14 @@ export function parseGuidedTransform(
if (bodyOutputs.length !== 1) return undefined;
const output = bodyOutputs[0];
const resultPath = referenceFromCall(output.source.nodeId, output.source.path, call.id);
if (output.source.nodeId !== call.id) return undefined;
const decode = direction.nodes.find((node): node is Extract<BrowserTransformPipelineNode, { kind: 'builtin' }> => (
node.kind === 'builtin' && node.operation === 'hex.decode'
&& node.inputs.length === 1 && node.id === output.source.nodeId
));
const resultPath = decode
? referenceFromCall(decode.inputs[0].nodeId, decode.inputs[0].path, call.id)
: referenceFromCall(output.source.nodeId, output.source.path, call.id);
if (decode ? decode.inputs[0].nodeId !== call.id : output.source.nodeId !== call.id) return undefined;
if (output.destination === 'body') {
return { callableId: call.callableId, inputPaths, resultPath, outputKind: 'body', outputField: '', setFormContentType: false };
}
@@ -4,7 +4,8 @@ import type {
BrowserPageCallable,
BrowserProfileInferenceCandidate,
} from '@/types/models';
import { createBrowserTransformProfileInput } from './profile-draft';
import { createBrowserTransformProfileInput, pairedBrowserTransformCandidate } from './profile-draft';
import { executeTransformDirection } from './mapping';
const tab: ActiveTabInfo = {
id: 7,
@@ -66,6 +67,26 @@ const responseCandidate = {
} satisfies BrowserProfileInferenceCandidate;
describe('browser transform profile draft', () => {
it('reads only the captured form field for a single string input and rejects ambiguous fields', async () => {
const candidate = { ...responseCandidate, direction: 'request' as const,
request: { ...responseCandidate.request, bodyFormat: 'form' as const, serialization: 'form-field' as const } };
const packet = { method: 'POST', url: candidate.request.url,
headers: [{ name: 'Content-Type', value: 'application/x-www-form-urlencoded; charset=utf-8' }],
bodyBase64: btoa('encryptedData={"username":"admin","password":"admin123"}') };
const profile = createBrowserTransformProfileInput(tab, undefined, callable, candidate, packet);
expect(profile.request.nodes.filter((node) => node.kind === 'context.read')).toMatchObject([{ path: 'body.encryptedData' }]);
let received: unknown[] = [];
await executeTransformDirection('test', 'request', profile.request, packet, async (callableId, args) => {
received = args;
return { callableId, type: 'string', preview: 'cipher', value: 'cipher', durationMs: 1 };
});
expect(received).toEqual(['{"username":"admin","password":"admin123"}']);
for (const body of ['username=admin', 'encryptedData=a&encryptedData=b']) {
expect(() => createBrowserTransformProfileInput(tab, undefined, callable, candidate, { ...packet, bodyBase64: btoa(body) })).toThrow(/input_paths/);
}
const jsonPacket = { ...packet, headers: [{ name: 'Content-Type', value: 'application/json' }], bodyBase64: btoa('{"username":"admin"}') };
expect(createBrowserTransformProfileInput(tab, undefined, callable, candidate, jsonPacket).request.nodes[0]).toMatchObject({ path: 'body' });
});
it('compiles an inferred response decryptor into the response direction', () => {
const profile = createBrowserTransformProfileInput(tab, undefined, callable, responseCandidate);
@@ -77,7 +98,106 @@ describe('browser transform profile draft', () => {
expect.objectContaining({ kind: 'output.write', destination: 'body' }),
]));
expect(profile.match).toEqual({ methods: ['GET'], urlPattern: '*/api/profile' });
expect(profile.name).toContain('响应明文网关');
expect(profile.name).toContain('浏览器协议网关');
});
it('maps response ciphertext, key, and iv into one decrypt callable', async () => {
const dynamicCallable: BrowserPageCallable = {
...callable,
inputSlots: [
callable.inputSlots[0],
{ id: 'key', name: 'key', index: 1, role: 'key', dataType: 'string', required: true, retained: false },
{ id: 'iv', name: 'iv', index: 2, role: 'iv', dataType: 'string', required: true, retained: false },
],
};
const dynamicCandidate: BrowserProfileInferenceCandidate = {
...responseCandidate,
request: {
...responseCandidate.request,
mappings: [
{ sourceEventId: 'decrypt-event', destination: 'body.message', serialization: 'json-field' },
{ sourceEventId: 'decrypt-event', destination: 'body.key', serialization: 'json-field' },
{ sourceEventId: 'decrypt-event', destination: 'body.iv', serialization: 'json-field' },
],
},
};
const profile = createBrowserTransformProfileInput(tab, undefined, dynamicCallable, dynamicCandidate);
const packet = {
method: 'POST', url: dynamicCandidate.request.url,
headers: [{ name: 'Content-Type', value: 'application/json' }],
bodyBase64: btoa(JSON.stringify({ message: 'cipher', key: '0011', iv: 'aabb' })),
};
let received: unknown[] = [];
await executeTransformDirection('test', 'response', profile.response, packet, async (callableId, args) => {
received = args;
return { callableId, type: 'string', preview: 'plain', value: 'plain', durationMs: 1 };
});
expect(received).toEqual(['cipher', '0011', 'aabb']);
});
it('pairs one browser transaction and compiles both directions into one profile', () => {
const requestCandidate: BrowserProfileInferenceCandidate = {
...responseCandidate,
id: 'candidate-request',
transactionId: 'fetch-1',
direction: 'request',
request: { ...responseCandidate.request, eventId: 'request-event', method: 'POST' },
};
const pairedResponse: BrowserProfileInferenceCandidate = {
...responseCandidate,
transactionId: 'fetch-1',
request: { ...responseCandidate.request, method: 'POST' },
};
const encryptCallable: BrowserPageCallable = {
...callable,
id: 'encrypt-callable',
name: '页面 AES 加密',
operation: 'AES.encrypt',
provenance: { eventId: 'encrypt-event' },
};
expect(pairedBrowserTransformCandidate([requestCandidate, pairedResponse], requestCandidate)?.id)
.toBe(pairedResponse.id);
const profile = createBrowserTransformProfileInput(
tab,
undefined,
encryptCallable,
requestCandidate,
undefined,
{ candidate: pairedResponse, callable },
);
expect(profile.request.enabled).toBe(true);
expect(profile.response.enabled).toBe(true);
expect(profile.request.nodes).toContainEqual(expect.objectContaining({ kind: 'page.call', callableId: encryptCallable.id }));
expect(profile.response.nodes).toContainEqual(expect.objectContaining({ kind: 'page.call', callableId: callable.id }));
expect(profile.name).toBe('POST */api/profile 浏览器协议网关');
});
it('does not guess when one trace contains multiple opposite candidates for the same route', () => {
const requestCandidate = { ...responseCandidate, id: 'request', direction: 'request' as const };
const responseA = { ...responseCandidate, id: 'response-a' };
const responseB = { ...responseCandidate, id: 'response-b' };
expect(pairedBrowserTransformCandidate([requestCandidate, responseA, responseB], requestCandidate)).toBeUndefined();
expect(() => pairedBrowserTransformCandidate([requestCandidate, responseA, responseB], requestCandidate, true))
.toThrow('尚未保存单向网关');
});
it('does not pair candidates from different recording sessions', () => {
const requestCandidate = { ...responseCandidate, id: 'request', transactionId: 'fetch-1', direction: 'request' as const };
const staleResponse = { ...responseCandidate, id: 'stale-response', transactionId: 'fetch-1', recordingId: 'recording-old' };
expect(pairedBrowserTransformCandidate([requestCandidate, staleResponse], requestCandidate)).toBeUndefined();
});
it('does not fall back to the route when transaction IDs disagree', () => {
const requestCandidate = { ...responseCandidate, id: 'request', transactionId: 'fetch-1', direction: 'request' as const };
const otherResponse = { ...responseCandidate, id: 'other-response', transactionId: 'fetch-2' };
expect(pairedBrowserTransformCandidate([requestCandidate, otherResponse], requestCandidate)).toBeUndefined();
});
it('serializes request-transaction profiles because they mutate one browser session', () => {
@@ -5,7 +5,9 @@ import type {
BrowserRecordingEvent,
BrowserTransformDirection,
BrowserTransformProfileInput,
BrowserTransformPacket,
} from '@/types/models';
import { ExtensionError } from '@/shared/errors';
import { compileGuidedTransform, defaultGuidedTransform, type GuidedTransformOutputKind } from './guided';
interface RequestRouteSource {
@@ -13,6 +15,12 @@ interface RequestRouteSource {
method?: string;
}
export interface BrowserTransformProfileBinding {
candidate: BrowserProfileInferenceCandidate;
callable: BrowserPageCallable;
packet?: BrowserTransformPacket;
}
function originOf(url?: string): string {
try { return url ? new URL(url).origin : ''; } catch { return ''; }
}
@@ -26,7 +34,33 @@ function emptyDirection(enabled = false): BrowserTransformDirection {
return { enabled, nodes: [] };
}
function candidateGuidance(candidate?: BrowserProfileInferenceCandidate): {
function sameTarget(left: BrowserProfileInferenceCandidate, right: BrowserProfileInferenceCandidate): boolean {
return left.target.tabId === right.target.tabId
&& left.target.frameId === right.target.frameId
&& (!left.target.documentId || !right.target.documentId || left.target.documentId === right.target.documentId);
}
export function pairedBrowserTransformCandidate(
candidates: BrowserProfileInferenceCandidate[],
candidate: BrowserProfileInferenceCandidate,
requireUnambiguous = false,
): BrowserProfileInferenceCandidate | undefined {
const opposite = candidates.filter((item) => item.id !== candidate.id
&& item.recordingId === candidate.recordingId
&& item.direction !== candidate.direction && sameTarget(item, candidate));
const matches = candidate.transactionId
? opposite.filter((item) => item.transactionId === candidate.transactionId)
: opposite.filter((item) => !item.transactionId
&& item.traceId === candidate.traceId
&& item.request.method.toUpperCase() === candidate.request.method.toUpperCase()
&& item.request.url === candidate.request.url);
if (requireUnambiguous && matches.length > 1) {
throw new ExtensionError('profile_evidence_ambiguous', `同一事务存在 ${matches.length} 个反方向候选,无法完整合并;尚未保存单向网关`);
}
return matches.length === 1 ? matches[0] : undefined;
}
function candidateGuidance(candidate?: BrowserProfileInferenceCandidate, callable?: BrowserPageCallable, packet?: BrowserTransformPacket): {
inputPaths?: string[];
outputKind?: GuidedTransformOutputKind;
outputField?: string;
@@ -35,9 +69,30 @@ function candidateGuidance(candidate?: BrowserProfileInferenceCandidate): {
const serialization = candidate?.request.serialization;
if (!destination) return {};
if (candidate?.direction === 'response') {
return { inputPaths: [destination], outputKind: 'body' };
return {
inputPaths: candidate.request.mappings
.map((mapping) => mapping.destination)
.filter((path): path is string => Boolean(path)),
outputKind: 'body',
};
}
if (serialization === 'form-field') {
let inputPaths: string[] | undefined;
const slots = callable?.inputSlots.filter((slot) => !slot.retained);
if (packet && slots?.length === 1 && slots[0].dataType === 'string') {
const contentType = packet.headers.find((header) => header.name.toLowerCase() === 'content-type')?.value.split(';')[0].trim().toLowerCase();
if (contentType === 'application/x-www-form-urlencoded') {
const body = new TextDecoder().decode(Uint8Array.from(atob(packet.bodyBase64), (char) => char.charCodeAt(0)));
const fields = new URLSearchParams(body);
const destinations = new Set(candidate?.request.mappings.map((mapping) => mapping.destination));
if (destinations.size > 1 || fields.getAll(destination.slice(5)).length !== 1) {
throw new ExtensionError('profile_input_mismatch', '无法唯一确定表单明文输入,请显式指定 input_paths;尚未发送请求');
}
inputPaths = [destination];
}
}
return { inputPaths, outputKind: 'form-field', outputField: destination.slice(5) };
}
if (serialization === 'form-field') return { outputKind: 'form-field', outputField: destination.slice(5) };
if (serialization === 'json-field') return { outputKind: 'json-field', outputField: destination.slice(5) };
if (serialization === 'header') return { outputKind: 'header', outputField: destination.slice(7) };
if (serialization === 'query') return { outputKind: 'query', outputField: destination.slice(6) };
@@ -49,17 +104,19 @@ export function createBrowserTransformProfileInput(
event?: BrowserRecordingEvent,
callable?: BrowserPageCallable,
candidate?: BrowserProfileInferenceCandidate,
packet?: BrowserTransformPacket,
paired?: BrowserTransformProfileBinding,
): BrowserTransformProfileInput {
const guide = defaultGuidedTransform(callable, candidateGuidance(candidate));
const guide = defaultGuidedTransform(callable, candidateGuidance(candidate, callable, packet));
const compiled = callable ? compileGuidedTransform(guide, callable) : emptyDirection(true);
const responseDirection = candidate?.direction === 'response';
const routeEvent = candidate ? {
url: candidate.request.url,
method: candidate.request.method,
} : event;
return {
const profile: BrowserTransformProfileInput = {
name: routeEvent?.url
? `${routeEvent.method || 'HTTP'} ${routeOf(routeEvent, tab)} ${responseDirection ? '响应' : '请求'}明文网关`
? `${routeEvent.method || 'HTTP'} ${routeOf(routeEvent, tab)} 浏览器协议网关`
: `${tab.title || '当前页面'} 明文网关`,
enabled: true,
target: { tabId: tab.id, frameId: 0 },
@@ -70,4 +127,16 @@ export function createBrowserTransformProfileInput(
failMode: 'closed',
maxConcurrency: callable?.kind === 'request-transaction' ? 1 : 2,
};
if (!paired) return profile;
if (!candidate
|| pairedBrowserTransformCandidate([candidate, paired.candidate], candidate)?.id !== paired.candidate.id) {
throw new ExtensionError('profile_evidence_mismatch', '请求与响应候选不属于同一个浏览器协议网关');
}
const pairedGuide = defaultGuidedTransform(
paired.callable,
candidateGuidance(paired.candidate, paired.callable, paired.packet),
);
profile[paired.candidate.direction] = compileGuidedTransform(pairedGuide, paired.callable);
if (paired.callable.kind === 'request-transaction') profile.maxConcurrency = 1;
return profile;
}
+11 -6
View File
@@ -855,9 +855,10 @@ export async function executeBrowserTransform(input: BrowserTransformExecuteInpu
direction,
input.packet,
);
return input.direction === 'request'
const result = input.direction === 'request'
? await bindOnlineTransactionSession(profile, execution)
: execution;
return { ...result, explanation: profile.explanation };
} finally {
leave();
}
@@ -866,6 +867,7 @@ export async function executeBrowserTransform(input: BrowserTransformExecuteInpu
export async function validateBrowserTransformProfile(
input: BrowserTransformProfileInput,
packet: BrowserTransformExecuteInput['packet'],
options: { direction?: BrowserTransformDirectionName; profileId?: string } = {},
): Promise<{ profile: BrowserTransformProfile; execution: BrowserTransformExecution }> {
const target = await resolveDocumentTarget(input.target);
const isolation = await currentTransformIsolation(target);
@@ -878,19 +880,22 @@ export async function validateBrowserTransformProfile(
const normalized = withRequestTransactionBinding(
normalizeProfile({
...input,
id: `validation-${crypto.randomUUID()}`,
id: options.profileId || `validation-${crypto.randomUUID()}`,
target,
maxConcurrency: transactionSafeConcurrency(input, callables),
}, isolation),
requestTransaction,
);
const profile = withTransformExplanation(normalized, callables);
const directionName: BrowserTransformDirectionName = profile.request.enabled
const directionName: BrowserTransformDirectionName = options.direction || (profile.request.enabled
? 'request'
: profile.response.enabled ? 'response' : 'request';
: profile.response.enabled ? 'response' : 'request');
const direction = profile[directionName];
if (!profile.enabled || !direction.enabled) {
throw new ExtensionError('transform_direction_disabled', '候选明文网关没有启用任何转换方向');
if (!profile.enabled) {
throw new ExtensionError('transform_profile_disabled', '候选明文网关未启用');
}
if (!direction.enabled) {
throw new ExtensionError('transform_direction_disabled', `候选明文网关未启用 ${directionName} 转换方向`);
}
assertTransformRoute(profile.match.methods, profile.match.urlPattern, packet, profile.origin);
assertRequestTransactionPacket(profile, packet);
@@ -8,16 +8,17 @@ import { errorMessage, request } from '@/platform/messaging/runtime';
import type {
ActiveTabInfo, BrowserDeepCaptureFrame, BrowserDeepCaptureMatcher, BrowserDeepCaptureStatus,
BrowserPageCallable, BrowserPageCallableExecution,
BrowserProfileInferenceCandidate, BrowserRecordingEvent,
BrowserProfileInferenceCandidate, BrowserRecordingEvent, BrowserTarget,
} from '@/types/models';
import './deep-capture-workspace.css';
import { cryptoDeepCaptureMatcher } from '@/features/browser-crypto/model';
import { eventMatcher } from './matcher';
import { capturedCallableSample, type CapturedCallableSample } from './callable-sample';
type RunTask = (task: () => Promise<void>, success?: string) => Promise<void>;
interface DeepCaptureWorkspaceProps {
tab?: ActiveTabInfo;
recordingTarget?: BrowserTarget;
selectedEvent?: BrowserRecordingEvent;
selectedCandidate?: BrowserProfileInferenceCandidate;
autoArmRequest?: number;
@@ -43,32 +44,6 @@ const STATUS_LABELS: Record<BrowserDeepCaptureStatus['state'], string> = {
error: '需要处理',
};
function eventMatcher(
event?: BrowserRecordingEvent,
candidate?: BrowserProfileInferenceCandidate,
): BrowserDeepCaptureMatcher | undefined {
if (!event) return undefined;
const frameHints = candidate?.capturePlan?.matcherEventId === event.id
? candidate.capturePlan.frameHints
: undefined;
const crypto = cryptoDeepCaptureMatcher(event);
if (crypto) return { ...crypto, frameHints };
if (['fetch', 'xhr', 'form'].includes(event.kind) && event.url) {
return { kind: 'request', urlPattern: event.url, frameHints };
}
if (['beacon', 'worker', 'message'].includes(event.kind) && event.wrapperHandleId) {
return {
kind: 'boundary',
eventKind: event.kind as 'beacon' | 'worker' | 'message',
operation: event.operation,
wrapperHandleId: event.wrapperHandleId,
scriptUrl: event.scriptUrl,
frameHints,
};
}
return undefined;
}
function compactUrl(value: string): string {
if (!value) return '内联脚本';
try {
@@ -102,6 +77,7 @@ const FRAME_SOURCE_LABELS: Record<BrowserDeepCaptureFrame['sourceKind'], string>
export function DeepCaptureWorkspace({
tab,
recordingTarget,
selectedEvent,
selectedCandidate,
autoArmRequest = 0,
@@ -141,7 +117,13 @@ export function DeepCaptureWorkspace({
const handledAutoCapturePause = useRef(0);
const automaticFlowRequested = useRef(false);
const target = status?.target || (tab ? { tabId: tab.id, frameId: 0 } : undefined);
const baseTarget = useMemo(() => recordingTarget
? { ...recordingTarget }
: tab ? { tabId: tab.id, frameId: 0 } : undefined,
[recordingTarget?.documentId, recordingTarget?.frameId, recordingTarget?.tabId, tab?.id]);
const target = status && baseTarget
&& status.target.tabId === baseTarget.tabId && status.target.frameId === baseTarget.frameId
? status.target : baseTarget;
const paused = status?.state === 'paused' && Boolean(status.pause);
useEffect(() => { statusRef.current = status; }, [status]);
@@ -177,15 +159,15 @@ export function DeepCaptureWorkspace({
return;
}
try {
const nextStatus = await request('deep.capture.status', { tabId: tab.id, frameId: 0 });
const nextStatus = await request('deep.capture.status', baseTarget!);
setStatus(nextStatus);
const nextCallables = await request('callable.list', { tabId: tab.id, frameId: 0 }).catch(() => []);
const nextCallables = await request('callable.list', baseTarget!).catch(() => []);
setCallables(nextCallables);
setLoadError('');
} catch (error) {
setLoadError(errorMessage(error));
}
}, [tab]);
}, [baseTarget?.documentId, baseTarget?.frameId, baseTarget?.tabId, tab]);
useEffect(() => { void load(); }, [load]);
@@ -209,11 +191,11 @@ export function DeepCaptureWorkspace({
}
void run(async () => {
setExecution(undefined);
const next = await request('deep.capture.start', { tabId: tab.id, frameId: 0, matcher: suggestedMatcher });
const next = await request('deep.capture.start', { ...baseTarget!, matcher: suggestedMatcher });
automaticFlowRequested.current = true;
setStatus(next);
}, '自动分析已武装,请在目标页面重复刚才的操作');
}, [autoArmRequest, busy, run, status, suggestedMatcher, tab]);
}, [autoArmRequest, baseTarget, busy, run, status, suggestedMatcher, tab]);
useEffect(() => {
if (!autoRecoveryRequest || handledAutoRecoveryRequest.current >= autoRecoveryRequest
@@ -234,10 +216,10 @@ export function DeepCaptureWorkspace({
useEffect(() => {
if (!tab || !['armed', 'paused', 'attached'].includes(status?.state || '')) return undefined;
const interval = window.setInterval(() => void request('deep.capture.status', { tabId: tab.id, frameId: 0 })
const interval = window.setInterval(() => void request('deep.capture.status', baseTarget!)
.then(setStatus).catch((error) => setLoadError(errorMessage(error))), status?.state === 'armed' ? 450 : 1_200);
return () => window.clearInterval(interval);
}, [status?.state, tab]);
}, [baseTarget, status?.state, tab]);
useEffect(() => {
if (!paused || !target) return undefined;
@@ -299,7 +281,7 @@ export function DeepCaptureWorkspace({
: { kind: 'request', urlPattern: urlPattern.trim(), frameHints };
setExecution(undefined);
automaticFlowRequested.current = false;
setStatus(await request('deep.capture.start', { tabId: tab.id, frameId: 0, matcher }));
setStatus(await request('deep.capture.start', { ...baseTarget!, matcher }));
}, '深度捕获已武装,请在目标页面重现一次操作');
const resume = () => run(async () => {
@@ -416,8 +398,8 @@ export function DeepCaptureWorkspace({
}, recoveryProfileId
? '新页面函数已捕获,旧网关继续停用;请完成本地回放验证'
: captureStrategy === 'request-transaction'
? '页面请求事务与明文网关已自动保存,真实发送将在回放时被截获'
: '完整业务加密流程与明文网关已自动保存');
? '页面请求事务已捕获;存在响应方向时将继续完成同一个协议网关'
: '完整业务转换流程已捕获;存在配对方向时将继续完成同一个协议网关');
}, [
onRecoveryCaptured,
onUseRecommendedCallable,
@@ -433,20 +415,23 @@ export function DeepCaptureWorkspace({
const useRecordedRecommendation = () => run(async () => {
if (!target || !recordedRecommendation?.source.callHandleId) throw new Error('推荐调用已经失效');
setStatus(await request('deep.capture.resume', target));
let callable = callables.find((item) => item.provenance.eventId === recordedRecommendation.source.eventId);
const inputCount = recordedRecommendation.source.dynamicInputPaths?.length || 1;
let callable = callables.find((item) => item.provenance.eventId === recordedRecommendation.source.eventId
&& item.inputSlots.filter((slot) => !slot.retained).length === inputCount);
if (!callable) {
callable = await request('callable.create', {
...target,
source: 'recording',
callHandleId: recordedRecommendation.source.callHandleId,
name: `${recordedRecommendation.source.crypto?.algorithm || recordedRecommendation.source.crypto?.operation || recordedRecommendation.source.operation} 页面函数`,
dynamicInputPaths: recordedRecommendation.source.dynamicInputPaths,
});
}
const selected = callable;
setCallables((current) => [...current.filter((item) => item.id !== selected.id), selected]);
setSelectedCallableId(selected.id);
await onUseRecommendedCallable?.(recordedRecommendation, selected);
}, '已使用录制调用生成并保存明文网关');
}, '当前转换方向已完成;存在配对方向时将继续合并');
const executeCallable = () => run(async () => {
if (!target || !selectedCallableId) throw new Error('请选择页面函数');
@@ -605,7 +590,7 @@ export function DeepCaptureWorkspace({
<header><Braces size={14} /><strong>函数评估</strong></header>
<div className="deep-frame-summary"><strong>{selectedFrame?.functionName || '未选择调用帧'}</strong><small>{selectedFrame ? `${compactUrl(selectedFrame.url)}:${selectedFrame.lineNumber}:${selectedFrame.columnNumber}` : ''}</small>{selectedFrame?.sourceMapUrl && <small title={selectedFrame.sourceMapUrl}>Source Map 元数据 · {compactUrl(selectedFrame.sourceMapUrl)}</small>}<span>{selectedFrame?.thisPreview || ''}</span></div>
{selectedFrame?.sourceKind === 'extension-hook' ? <div className="deep-function-assessment is-hook"><Bug size={15} /><span><strong>这是插件注入的观测帧</strong><small>它只负责记录或设置断点,不是页面业务代码。请选择调用栈中标记为“页面函数”的下游帧。</small></span></div> : selectedFrame?.functionInspection?.resolved ? <div className={`deep-function-assessment ${selectedFrame.functionInspection.riskFlags.length ? 'has-risk' : 'is-clean'}`}>
{selectedFrame.functionInspection.riskFlags.length ? <><ShieldAlert size={15} /><span><strong>已阻止注册为可回放函数</strong><small>{selectedFrame.functionInspection.riskFlags.map((risk) => RISK_LABELS[risk]).join(' · ')}。直接调用可能改变页面或发送真实请求。</small></span></> : <><Check size={15} /><span><strong>函数对象已自动解析</strong><small>{selectedFrame.functionInspection.parameterCount || 0} 个参数 · {selectedFrame.functionInspection.resolution === 'receiver-method' ? '页面方法' : selectedFrame.functionInspection.resolution === 'scope-binding' ? '闭包绑定' : '当前栈帧'} · 未发现明显副作用</small></span></>}
{selectedFrame.functionInspection.riskFlags.length ? <><ShieldAlert size={15} /><span><strong>已阻止注册为可回放函数</strong><small>{selectedFrame.functionInspection.riskFlags.map((risk) => RISK_LABELS[risk]).join(' · ')}。直接调用可能改变页面或发送真实请求。</small></span></> : <><Check size={15} /><span><strong>函数对象已自动解析</strong><small>{selectedFrame.functionInspection.parameterCount || 0} 个参数 · {selectedFrame.functionInspection.resolution === 'receiver-method' ? '页面方法' : selectedFrame.functionInspection.resolution === 'scope-binding' ? '闭包绑定' : selectedFrame.functionInspection.resolution === 'current-function' ? '当前函数' : '当前栈帧'} · 未发现明显副作用</small></span></>}
</div> : <div className="deep-function-assessment has-risk"><AlertTriangle size={15} /><span><strong>无法唯一解析当前函数</strong><small>{selectedFrame?.functionInspection?.candidateCount ? `发现 ${selectedFrame.functionInspection.candidateCount} 个同分候选;` : ''}请选择其他业务栈帧,或在高级模式中指定闭包变量。</small></span></div>}
<div className="deep-adapter-editor__primary">{recoveryProfileId
? <Button variant="primary" disabled={busy || !recoverySelectionReady} onClick={() => void captureRecovery(recoveryCaptureStrategy)}><RotateCcw size={14} />{recoveryCaptureStrategy === 'request-transaction' ? '按所选函数恢复请求事务' : '用所选函数恢复绑定'}</Button>
@@ -21,7 +21,12 @@ function comparableUrl(value?: string): string {
}
function nameMatches(frameName: string, hintName: string): boolean {
return frameName === hintName || frameName.endsWith(`.${hintName}`) || hintName.endsWith(`.${frameName}`);
const normalize = (value: string) => /^(?:\(anonymous\)|<anonymous>|anonymous)$/i.test(value.trim())
? '(anonymous)'
: value;
const frame = normalize(frameName);
const hint = normalize(hintName);
return frame === hint || frame.endsWith(`.${hint}`) || hint.endsWith(`.${frame}`);
}
function matchingHint(frame: BrowserDeepCaptureFrame, hints: BrowserBusinessFrameHint[]): BrowserBusinessFrameHint | undefined {
@@ -30,10 +35,12 @@ function matchingHint(frame: BrowserDeepCaptureFrame, hints: BrowserBusinessFram
}
function isEventHandler(frame: BrowserDeepCaptureFrame): boolean {
if (frame.functionInspection?.resolution === 'event-listener') return true;
if (EVENT_HANDLER_NAME.test(frame.functionName)) return true;
const parameters = frame.functionInspection?.parameterNames || [];
return parameters.some((name) => /^(?:event|evt)$/i.test(name))
&& /(?:Element|Document|Window)/.test(frame.thisPreview);
return parameters.some((name) => /^(?:e|event|evt)$/i.test(name))
&& (/(?:Element|Document|Window|#[A-Za-z_$][\w$-]*)/.test(frame.thisPreview)
|| frame.scopes.some((scope) => scope.variables.some((variable) => /Event\b/.test(variable.preview))));
}
function hintedFrameOrder(
@@ -121,9 +128,13 @@ export function rankBusinessFrames(
.filter((frame) => frame.sourceKind === 'page' && frame.functionInspection?.resolved && matchingHint(frame, hints))
.sort((left, right) => hintedFrameOrder(left, right, hints));
const closestHinted = resolvedHinted[0];
const closestRisks = closestHinted?.functionInspection?.riskFlags || [];
const transactionRequired = Boolean(closestHinted
&& (isEventHandler(closestHinted) || closestRisks.some((risk) => TRANSACTION_RISKS.has(risk))));
const unhintedEventHandler = hints.length ? undefined : ranked.find((frame) => (
frame.sourceKind === 'page' && frame.functionInspection?.resolved && isEventHandler(frame)
));
const transactionFrame = closestHinted || unhintedEventHandler;
const closestRisks = transactionFrame?.functionInspection?.riskFlags || [];
const transactionRequired = Boolean(transactionFrame
&& (isEventHandler(transactionFrame) || closestRisks.some((risk) => TRANSACTION_RISKS.has(risk))));
const transactionBlocked = Boolean(transactionRequired && closestRisks.includes('storage'));
const eligible = ordered.filter((frame) => frame.functionInspection?.resolved
&& !frame.functionInspection.riskFlags.length && !isEventHandler(frame));
@@ -133,19 +144,19 @@ export function rankBusinessFrames(
const automatic = automaticEligible[0];
const alternative = automaticEligible[1];
let automaticCapture: RankedBusinessFrames['automaticCapture'];
if (transactionRequired && !transactionBlocked && closestHinted) {
if (transactionRequired && !transactionBlocked && transactionFrame) {
automaticCapture = {
state: 'ready',
strategy: 'request-transaction',
frameId: closestHinted.id,
reason: isEventHandler(closestHinted)
frameId: transactionFrame.id,
reason: isEventHandler(transactionFrame)
? '共同业务入口是页面事件处理器,将在隔离事务中截获并取消真实请求'
: '共同业务函数直接读取页面或发送请求,将以隔离事务保留完整动态参数关系',
};
} else if (transactionBlocked && closestHinted) {
} else if (transactionBlocked && transactionFrame) {
automaticCapture = {
state: 'blocked',
frameId: closestHinted.id,
frameId: transactionFrame.id,
reason: '共同业务函数会访问页面存储;当前事务回滚无法证明存储副作用已完全隔离',
};
} else if (automatic && alternative && (automatic.businessScore || 0) - (alternative.businessScore || 0) < 8) {
+15
View File
@@ -0,0 +1,15 @@
import type { BrowserDeepCaptureMatcher, BrowserProfileInferenceCandidate, BrowserRecordingEvent } from '@/types/models';
import { cryptoDeepCaptureMatcher } from '@/features/browser-crypto/model';
export function eventMatcher(event?: BrowserRecordingEvent, candidate?: BrowserProfileInferenceCandidate): BrowserDeepCaptureMatcher | undefined {
if (!event) return undefined;
const frameHints = candidate?.capturePlan?.frameHints;
const crypto = cryptoDeepCaptureMatcher(event);
if (crypto) return { ...crypto, frameHints };
if (['fetch', 'xhr', 'form'].includes(event.kind) && event.url) return { kind: 'request', urlPattern: event.url, frameHints };
if (['beacon', 'worker', 'message'].includes(event.kind) && event.wrapperHandleId) return {
kind: 'boundary', eventKind: event.kind as 'beacon' | 'worker' | 'message', operation: event.operation,
wrapperHandleId: event.wrapperHandleId, scriptUrl: event.scriptUrl, frameHints,
};
return undefined;
}
+76 -2
View File
@@ -9,6 +9,7 @@ const attachedTabs = new Set<number>();
const knownTabs = new Set<number>();
let failResume = false;
let failDetach = false;
let commandHandler: ((method: string, params?: Record<string, unknown>) => unknown) | undefined;
const STORAGE_KEY = 'session.deep-capture.v1';
@@ -27,9 +28,9 @@ const debuggerApi = {
getTargets: vi.fn(async () => [...knownTabs].map((tabId) => ({
attached: attachedTabs.has(tabId), tabId, id: `target-${tabId}`, type: 'page', url: 'https://example.test/',
}))),
sendCommand: vi.fn(async (_target: { tabId?: number; sessionId?: string }, method: string) => {
sendCommand: vi.fn(async (_target: { tabId?: number; sessionId?: string }, method: string, params?: Record<string, unknown>) => {
if (method === 'Debugger.resume' && failResume) throw new Error('fixture resume failed');
return {};
return commandHandler?.(method, params) || {};
}),
onEvent: { addListener: vi.fn((listener: typeof eventListeners[number]) => eventListeners.push(listener)) },
onDetach: { addListener: vi.fn((listener: (source: { tabId?: number; sessionId?: string }, reason: string) => void) => detachListeners.push(listener)) },
@@ -125,6 +126,7 @@ describe('deep capture debugger lifecycle', () => {
knownTabs.clear();
failResume = false;
failDetach = false;
commandHandler = undefined;
vi.clearAllMocks();
});
@@ -162,6 +164,17 @@ describe('deep capture debugger lifecycle', () => {
expect(debuggerApi.attach).not.toHaveBeenCalled();
});
it('allows Agent capture after local release but never takes over an active local session', async () => {
const target = { tabId: 39, frameId: 0 };
const matcher = { kind: 'request' as const, urlPattern: '/login' };
const owner = { kind: 'grant' as const, grantId: 'agent', expiresAt: Date.now() + 60_000 };
await startDeepCapture(target, matcher);
await expect(startDeepCapture(target, matcher, owner)).rejects.toMatchObject({ code: 'permission_denied' });
await resumeDeepCapture(target);
await expect(startDeepCapture(target, matcher, owner)).resolves.toMatchObject({ state: 'armed' });
await resumeDeepCapture(target, 'agent-done', owner);
});
it('restores a paused page when callable capture validation fails', async () => {
const target = { tabId: 20, frameId: 0 };
seedStatus(target.tabId, pausedStatus(target.tabId, {
@@ -322,6 +335,67 @@ describe('deep capture debugger lifecycle', () => {
await resumeDeepCapture(target);
});
it('resolves an anonymous form handler through its paused SubmitEvent', async () => {
const target = { tabId: 30, frameId: 0 };
commandHandler = (method, params) => {
if (method === 'Runtime.getProperties' && params?.objectId === 'local-scope') {
return { result: [{ name: 'e', value: { type: 'object', description: 'SubmitEvent' } }] };
}
if (method === 'Runtime.getProperties' && params?.objectId === 'listener-1') {
return { internalProperties: [{
name: '[[FunctionLocation]]',
value: { type: 'object', value: { scriptId: 'page-script', lineNumber: 160 } },
}] };
}
if (method === 'Debugger.evaluateOnCallFrame') {
const expression = String(params?.expression || '');
return expression.includes('arguments.callee')
? { result: { type: 'function', objectId: 'listener-1' } }
: { result: { type: 'undefined' } };
}
if (method === 'Runtime.callFunctionOn' && params?.objectId === 'listener-1') {
return { result: { value: {
functionName: '', parameterCount: 1, parameterNames: ['e'], riskFlags: ['network', 'dom'],
} } };
}
return {};
};
await startDeepCapture(target, {
kind: 'request',
urlPattern: '/crypto/sqli/aes-ecb/encrypt/login',
frameHints: [{
functionName: '<anonymous>', url: 'https://example.test/login', support: 1, averageDepth: 1,
}],
});
for (const listener of eventListeners) listener({ tabId: target.tabId }, 'Debugger.paused', {
reason: 'XHR',
callFrames: [{
callFrameId: 'hook-frame', functionName: 'recordedFetch',
url: 'chrome-extension://fixture/page-recorder-main-world.js',
location: { scriptId: 'hook-script', lineNumber: 10, columnNumber: 1 },
scopeChain: [], this: { type: 'object', description: 'Window' },
}, {
callFrameId: 'page-frame', functionName: '', url: 'https://example.test/login',
location: { scriptId: 'page-script', lineNumber: 162, columnNumber: 38 },
scopeChain: [{ type: 'local', object: { type: 'object', objectId: 'local-scope' } }],
this: { type: 'object', description: 'HTMLFormElement' },
}],
});
await vi.waitFor(async () => expect((await deepCaptureStatus(target)).pause?.collecting).toBe(false));
const result = await deepCaptureStatus(target);
expect(result.pause?.frames[1]?.functionInspection).toMatchObject({
resolved: true, resolution: 'current-function', referenceExpression: 'arguments.callee',
});
expect(result.pause?.automaticCapture).toMatchObject({
state: 'ready', strategy: 'request-transaction', frameId: 'page-frame',
});
expect(debuggerApi.sendCommand).not.toHaveBeenCalledWith(
expect.anything(), 'Debugger.getFunctionLocation', expect.anything(),
);
await resumeDeepCapture(target);
});
it('observes same-origin service-worker targets without routing their pauses into the page capture', async () => {
const target = { tabId: 29, frameId: 0 };
await startDeepCapture(target, { kind: 'request', urlPattern: '/login' });
+106 -39
View File
@@ -20,6 +20,7 @@ import { normalizeCallable } from '@/features/page-callable/service';
import { PAGE_CALLABLE_REGISTRY_KEY } from '@/features/page-callable/constants';
import { rankBusinessFrames } from './business-frame-ranker';
import { getTab } from '@/platform/browser/targets';
import { serializeTabExecution, withPageNetworkGuard } from '@/features/page-callable/network-guard';
interface Debuggee {
tabId?: number;
@@ -58,6 +59,7 @@ interface InspectedFunctionCandidate {
riskFlags: NonNullable<BrowserDeepCaptureFrame['functionInspection']>['riskFlags'];
scriptId?: string;
lineNumber?: number;
columnNumber?: number;
score: number;
}
@@ -71,6 +73,7 @@ interface CDPCallFrame {
callFrameId?: string;
functionName?: string;
location?: { scriptId?: string; lineNumber?: number; columnNumber?: number };
functionLocation?: { scriptId: string; lineNumber: number; columnNumber: number };
url?: string;
scopeChain?: CDPScope[];
this?: CDPRemoteObject;
@@ -114,6 +117,7 @@ const MAX_VALUE_PREVIEW = 512;
const MAX_VARIABLE_DETAIL = 4_096;
const MAX_SCOPE_DETAIL_BUDGET = 16_384;
const MAX_FUNCTION_CANDIDATES = 24;
const RETAINED_FUNCTIONS_KEY = '__YAKIT_DEEP_FUNCTIONS__';
const MAX_WORKER_TARGETS = 16;
const MAX_WORKER_SCRIPT_COUNT = 256;
const acceptedScopeTypes = new Set<BrowserDeepCaptureScope['type']>([
@@ -288,6 +292,8 @@ function debuggerTarget(tabId: number): Debuggee {
function assertSessionOwner(status: StoredDeepCaptureStatus, owner?: DeepCaptureOwner): void {
if (!owner || owner.kind === 'local') return;
// Released local/Agent history is not a live debugger lock.
if (!status.pause && status.recovery?.page === 'running' && status.recovery.debugger === 'detached') return;
if (status.owner.kind !== 'grant' || status.owner.grantId !== owner.grantId) {
throw new ExtensionError('permission_denied', '该页面的深度捕获由另一个会话控制');
}
@@ -616,7 +622,7 @@ async function inspectFunctionExpression(
if (evaluated?.exceptionDetails || evaluated?.result?.type !== 'function' || !evaluated.result.objectId) return undefined;
const objectId = evaluated.result.objectId;
try {
const [metadata, location] = await Promise.all([
const [metadata, properties] = await Promise.all([
sendCommand<{ result?: CDPRemoteObject }>(target, 'Runtime.callFunctionOn', {
objectId,
functionDeclaration: `function () {
@@ -640,8 +646,11 @@ async function inspectFunctionExpression(
returnByValue: true,
silent: true,
}).catch(() => undefined),
sendCommand<{ location?: { scriptId?: string; lineNumber?: number } }>(target, 'Debugger.getFunctionLocation', {
functionId: objectId,
sendCommand<{ internalProperties?: Array<{ name?: string; value?: CDPRemoteObject }> }>(target, 'Runtime.getProperties', {
objectId,
ownProperties: false,
accessorPropertiesOnly: false,
generatePreview: false,
}).catch(() => undefined),
]);
const value = metadata?.result?.value;
@@ -652,14 +661,19 @@ async function inspectFunctionExpression(
if (Array.isArray(input.riskFlags)) {
riskFlags.push(...input.riskFlags.filter((item): item is typeof riskFlags[number] => typeof item === 'string' && allowed.has(item)));
}
const rawLocation = properties?.internalProperties
?.find((property) => property.name === '[[FunctionLocation]]')?.value?.value;
const location = rawLocation && typeof rawLocation === 'object'
? rawLocation as { scriptId?: string; lineNumber?: number; columnNumber?: number }
: undefined;
const functionName = typeof input.functionName === 'string' ? input.functionName.slice(0, 240) : '';
const sameScript = Boolean(location?.location?.scriptId && location.location.scriptId === frame.scriptId);
const sameScript = Boolean(location?.scriptId && location.scriptId === frame.scriptId);
const nameMatch = functionName === frame.functionName || expression === frame.functionName;
let score = resolution === 'frame-name' ? 36 : resolution === 'receiver-method' ? 30 : 12;
let score = resolution === 'current-function' ? 48 : resolution === 'frame-name' ? 36 : resolution === 'receiver-method' ? 30 : 12;
if (sameScript) score += 42;
if (nameMatch) score += 28;
if (sameScript && Number.isFinite(location?.location?.lineNumber)) {
const distance = Math.max(0, frame.lineNumber - (Number(location?.location?.lineNumber) + 1));
if (sameScript && Number.isFinite(location?.lineNumber)) {
const distance = Math.max(0, frame.lineNumber - (Number(location?.lineNumber) + 1));
score += Math.max(0, 12 - Math.min(12, Math.floor(distance / 20)));
}
return {
@@ -671,8 +685,9 @@ async function inspectFunctionExpression(
? input.parameterNames.filter((item): item is string => typeof item === 'string' && validIdentifier(item)).slice(0, 16)
: [],
riskFlags,
scriptId: location?.location?.scriptId,
lineNumber: Number.isFinite(location?.location?.lineNumber) ? Number(location?.location?.lineNumber) + 1 : undefined,
scriptId: location?.scriptId,
lineNumber: Number.isFinite(location?.lineNumber) ? Number(location?.lineNumber) + 1 : undefined,
columnNumber: Number.isFinite(location?.columnNumber) ? Number(location?.columnNumber) + 1 : undefined,
score,
};
} finally {
@@ -689,8 +704,10 @@ async function inspectFrameFunction(
expressions.push({ expression: frame.functionName, resolution: 'frame-name' });
expressions.push({ expression: receiverFunctionExpression(frame.functionName), resolution: 'receiver-method' });
}
if (!validIdentifier(frame.functionName) || frame.functionName === '(anonymous)') {
{
expressions.push({ expression: 'arguments.callee', resolution: 'current-function' });
for (const variable of frame.scopes.flatMap((scope) => scope.variables)) {
if (expressions.length >= MAX_FUNCTION_CANDIDATES) break;
if (variable.type !== 'function' || !validIdentifier(variable.name)) continue;
expressions.push({ expression: variable.name, resolution: 'scope-binding' });
if (expressions.length >= MAX_FUNCTION_CANDIDATES) break;
@@ -700,10 +717,19 @@ async function inspectFrameFunction(
const inspected = (await Promise.all(unique.map((item) => inspectFunctionExpression(
target, frame, item.expression, item.resolution,
)))).filter((item): item is InspectedFunctionCandidate => Boolean(item))
.filter((item) => item.resolution !== 'scope-binding' || item.scriptId === frame.scriptId);
const candidates = [...new Map(inspected
.sort((left, right) => right.score - left.score || left.expression.localeCompare(right.expression))
.map((item) => [`${item.functionName}\n${item.scriptId || ''}\n${item.lineNumber || ''}\n${item.parameterCount}`, item])).values()]
.filter((item) => item.scriptId === frame.scriptId
&& (!frame.functionLocation || (item.lineNumber === frame.functionLocation.lineNumber + 1
&& item.columnNumber === frame.functionLocation.columnNumber + 1)));
if (!inspected.length && frame.functionLocation) {
const listener = await inspectPausedEventListener(target, frame);
if (listener) inspected.push(listener);
}
const distinct = new Map<string, InspectedFunctionCandidate>();
for (const item of inspected.sort((left, right) => right.score - left.score)) {
const key = `${item.scriptId}:${item.lineNumber}:${item.columnNumber}`;
if (!distinct.has(key)) distinct.set(key, item);
}
const candidates = [...distinct.values()]
.sort((left, right) => right.score - left.score || left.expression.localeCompare(right.expression));
const selected = candidates[0];
const ambiguous = Boolean(selected && candidates[1] && selected.score - candidates[1].score < 8);
@@ -719,6 +745,54 @@ async function inspectFrameFunction(
};
}
async function inspectPausedEventListener(target: Debuggee, frame: BrowserDeepCaptureFrame): Promise<InspectedFunctionCandidate | undefined> {
const matches: CDPRemoteObject[] = [];
// Query the browser's listener registry, including delegated and shadow-tree listeners.
// Unlike arguments.callee this also works for strict functions and arrow functions.
for (const expression of ['document', 'window']) {
const evaluated = await sendCommand<{ result?: CDPRemoteObject }>(target, 'Debugger.evaluateOnCallFrame', {
callFrameId: frame.id, expression, objectGroup: 'yakit-deep-capture', silent: true,
}).catch(() => undefined);
if (!evaluated?.result?.objectId) continue;
try {
const result = await sendCommand<{ listeners?: Array<{
scriptId: string; lineNumber: number; columnNumber: number;
handler?: CDPRemoteObject; originalHandler?: CDPRemoteObject;
}> }>(target, 'DOMDebugger.getEventListeners', {
objectId: evaluated.result.objectId, depth: -1, pierce: true,
});
for (const listener of result.listeners || []) {
if (listener.scriptId !== frame.scriptId
|| listener.lineNumber !== frame.functionLocation!.lineNumber
|| listener.columnNumber !== frame.functionLocation!.columnNumber) continue;
const handler = listener.originalHandler || listener.handler;
if (handler?.objectId && handler.type === 'function') matches.push(handler);
}
} finally { await sendCommand(target, 'Runtime.releaseObject', { objectId: evaluated.result.objectId }).catch(() => undefined); }
}
if (!matches.length) return undefined;
const first = matches[0];
try {
for (const other of matches.slice(1)) {
const equal = await sendCommand<{ result?: CDPRemoteObject }>(target, 'Runtime.callFunctionOn', {
objectId: first.objectId, functionDeclaration: 'function(other) { return this === other; }',
arguments: [{ objectId: other.objectId }], returnByValue: true,
});
if (equal.result?.value !== true) return undefined;
}
const id = crypto.randomUUID();
await sendCommand(target, 'Runtime.callFunctionOn', {
objectId: first.objectId,
functionDeclaration: `function(id) { (globalThis[${JSON.stringify(RETAINED_FUNCTIONS_KEY)}] ||= Object.create(null))[id] = this; }`,
arguments: [{ value: id }],
});
return await inspectFunctionExpression(target, frame,
`globalThis[${JSON.stringify(RETAINED_FUNCTIONS_KEY)}][${JSON.stringify(id)}]`, 'event-listener');
} finally {
await Promise.all(matches.map((handler) => sendCommand(target, 'Runtime.releaseObject', { objectId: handler.objectId }).catch(() => undefined)));
}
}
function pauseSkeleton(
params: Record<string, unknown>,
matcher?: BrowserDeepCaptureMatcher,
@@ -741,6 +815,9 @@ function pauseSkeleton(
sourceMapUrl: script?.sourceMapUrl,
lineNumber: Math.max(1, Number(frame.location?.lineNumber || 0) + 1),
columnNumber: Math.max(1, Number(frame.location?.columnNumber || 0) + 1),
functionLocation: frame.functionLocation
? { lineNumber: frame.functionLocation.lineNumber, columnNumber: frame.functionLocation.columnNumber }
: undefined,
scopes: [],
thisPreview: remotePreview(frame.this),
sourceKind,
@@ -1352,6 +1429,7 @@ async function capturePageCallableWhilePaused(
delete globalThis[${JSON.stringify(retainedCallKey)}];
if (!retainedCall || !Array.isArray(retainedCall.args)) throw new Error("业务函数的暂停现场已经失效");
const candidate = (${functionExpression});
delete globalThis[${JSON.stringify(RETAINED_FUNCTIONS_KEY)}];
if (typeof candidate !== "function") throw new Error("选中的表达式不是函数");
const source = Function.prototype.toString.call(candidate).slice(0, 65536);
const transaction = ${JSON.stringify(requestTransaction || null)};
@@ -1368,31 +1446,6 @@ async function capturePageCallableWhilePaused(
registry = new Map();
Object.defineProperty(globalThis, key, { value: registry, configurable: true, enumerable: false });
}
if (transaction && typeof globalThis.fetch === "function") {
const previousFetch = globalThis.fetch;
let restoreTimer;
const restoreFetch = () => {
if (globalThis.fetch === transactionCaptureFetch) globalThis.fetch = previousFetch;
if (restoreTimer) clearTimeout(restoreTimer);
};
const transactionCaptureFetch = async function(input, init) {
let request;
try { request = new Request(input, init); } catch { return Reflect.apply(previousFetch, this, [input, init]); }
const expectedURL = new URL(transaction.request.url, location.href).toString();
if (transaction.request.boundary !== "fetch"
|| request.method.toUpperCase() !== transaction.request.method.toUpperCase()
|| request.url !== expectedURL) {
return Reflect.apply(previousFetch, this, [input, init]);
}
restoreFetch();
return new Response(JSON.stringify({ success: false, error: "request captured before transaction replay" }), {
status: 200,
headers: { "Content-Type": "application/json" }
});
};
globalThis.fetch = transactionCaptureFetch;
restoreTimer = setTimeout(restoreFetch, 10000);
}
const metadata = {
id: ${JSON.stringify(callableId)}, name: ${JSON.stringify(name)}, kind: ${JSON.stringify(callableKind)},
operation: candidate.name || ${JSON.stringify(functionExpression)}, origin: location.origin,
@@ -1476,6 +1529,20 @@ export async function createCapturedPageCallable(
callFrameId: string,
input: CapturedPageCallableInput,
owner?: DeepCaptureOwner,
): Promise<BrowserPageCallable> {
if (input.strategy === 'request-transaction') {
return serializeTabExecution(target.tabId, () => withPageNetworkGuard(
target, [], () => captureAndResume(target, callFrameId, input, owner), input.transaction.request.url,
));
}
return captureAndResume(target, callFrameId, input, owner);
}
async function captureAndResume(
target: BrowserTarget,
callFrameId: string,
input: CapturedPageCallableInput,
owner?: DeepCaptureOwner,
): Promise<BrowserPageCallable> {
let callable: BrowserPageCallable | undefined;
let captureError: unknown;
@@ -87,9 +87,26 @@ vi.mock('@/platform/storage/state', () => ({
vi.mock('@/protocol/capabilities', () => ({
BRIDGE_CAPABILITIES: [],
capabilityVisibleToAgent: vi.fn((method: string) => ![
'browser.thumbnail',
'browser.handoff.presentation.get',
'browser.handoff.focus',
'browser.handoff.resolve',
].includes(method)),
getBridgeCapabilityCatalog: vi.fn(async () => ({ version: 1, capabilities: [] })),
}));
vi.mock('@/features/grants/capability-context', () => ({
browserInstanceAccess: vi.fn(async () => ({
id: 'paired-browser-instance',
taskId: 'paired-browser-instance',
targets: [],
scopes: ['browser.tabs.read'],
createdAt: 0,
expiresAt: Number.MAX_SAFE_INTEGER,
})),
}));
vi.mock('@/features/grants/service', () => ({
routeCapability: vi.fn(async () => ({ ok: true })),
}));
@@ -128,7 +145,9 @@ vi.stubGlobal('WebSocket', FakeWebSocket);
import {
BRIDGE_HEARTBEAT_TIMEOUT_MS,
EngineBridge,
browserClientIdentity,
} from './service';
import { beginAgentAction } from '@/features/agent-runtime/service';
import {
BRIDGE_CHUNK_TIMEOUT_MS,
BRIDGE_PROTOCOL_VERSION,
@@ -144,6 +163,8 @@ function bridgeConfig(paired = true) {
endpoint: 'ws://127.0.0.1:64333/extension',
autoConnect: false,
installationId: 'installation-1',
browserName: 'Chrome for Testing',
browserVersion: '152.0.7977.82',
pairedEngine: paired ? {
engineIdentityId: 'engine-identity-1',
deviceId: 'device-1',
@@ -214,6 +235,26 @@ describe('Engine Bridge transport lifecycle', () => {
expect(socket.readyState).toBe(FakeWebSocket.CLOSED);
});
it('routes local UI capabilities without entering the paused Agent action gate', async () => {
const bridge = new EngineBridge();
const socket = await connect(bridge);
socket.receive({
type: 'request',
id: 'local-ui-1',
method: 'browser.handoff.presentation.get',
params: { handoffId: 'handoff-1' },
});
await vi.advanceTimersByTimeAsync(0);
expect(socket.sent.map((item) => JSON.parse(item)).find((item) => item.id === 'local-ui-1')).toMatchObject({
type: 'response',
id: 'local-ui-1',
result: { ok: true },
});
expect(beginAgentAction).not.toHaveBeenCalled();
});
it('closes a half-open connection and rejects pending calls after missed heartbeats', async () => {
const bridge = new EngineBridge();
const socket = await connect(bridge);
@@ -308,6 +349,8 @@ describe('Engine Bridge transport lifecycle', () => {
expect(auth).toMatchObject({
type: 'auth',
client: 'Chrome for Testing',
version: '152.0.7977.82',
challenge: 'engine-challenge-0123456789',
resumeSessionId: 'previous-session',
});
@@ -353,6 +396,33 @@ describe('Engine Bridge transport lifecycle', () => {
});
});
it('finds every local Yak engine in the bridge port range', async () => {
fixture.state.bridge = bridgeConfig(false);
const bridge = new EngineBridge();
const pending = bridge.discoverLocalEngines();
await Promise.resolve();
await Promise.resolve();
expect(FakeWebSocket.instances).toHaveLength(16);
for (const index of [0, 2]) {
const socket = FakeWebSocket.instances[index];
socket.open();
socket.receive({
type: 'engine',
protocolVersion: BRIDGE_PROTOCOL_VERSION,
engineIdentityId: `engine-identity-${index}`,
engineInstanceId: `engine-instance-${index}`,
endpoint: `ws://127.0.0.1:${64333 + index}/extension`,
});
}
await vi.advanceTimersByTimeAsync(801);
await expect(pending).resolves.toEqual([
expect.objectContaining({ endpoint: 'ws://127.0.0.1:64333/extension' }),
expect.objectContaining({ endpoint: 'ws://127.0.0.1:64335/extension' }),
]);
});
it('uses the engine pairing deadline locally after pair_pending', async () => {
fixture.state.bridge = bridgeConfig(false);
const bridge = new EngineBridge();
@@ -456,3 +526,21 @@ describe('Engine Bridge transport lifecycle', () => {
});
});
});
describe('browser client identity', () => {
it('distinguishes Edge and lets managed Chrome for Testing metadata win', () => {
const config = { ...bridgeConfig(false), browserName: undefined, browserVersion: undefined };
expect(browserClientIdentity(
config,
'Mozilla/5.0 AppleWebKit/537.36 Chrome/152.0.0.0 Safari/537.36 Edg/152.0.1234.5',
)).toEqual({ client: 'Microsoft Edge', version: '152.0.1234.5' });
expect(browserClientIdentity({
...config,
browserName: 'Chrome for Testing',
browserVersion: '152.0.7977.82',
}, 'Mozilla/5.0 Chrome/152.0.0.0')).toEqual({
client: 'Chrome for Testing',
version: '152.0.7977.82',
});
});
});
+77 -11
View File
@@ -1,11 +1,15 @@
import { browser } from 'wxt/browser';
import type { BridgeEnvelope } from '@/types/messages';
import type { BridgeConfig, BridgePairingStatus, BridgePublicKey, BridgeStatus } from '@/types/models';
import { BRIDGE_CAPABILITIES, getBridgeCapabilityCatalog } from '@/protocol/capabilities';
import type { BridgeConfig, BridgePairingStatus, BridgePublicKey, BridgeStatus, DiscoveredYakEngine } from '@/types/models';
import {
BRIDGE_CAPABILITIES,
capabilityVisibleToAgent,
getBridgeCapabilityCatalog,
} from '@/protocol/capabilities';
import {
BRIDGE_CHUNK_BYTES, BRIDGE_CHUNK_THRESHOLD_BYTES, BRIDGE_CHUNK_TIMEOUT_MS,
BRIDGE_MAX_CHUNK_TRANSFERS, BRIDGE_MAX_MESSAGE_BYTES, BRIDGE_PROTOCOL_VERSION, parseBridgeEnvelope,
parseBridgePairingEnvelope, type BridgePairingEnvelope,
parseBridgeDiscoveryEnvelope, parseBridgePairingEnvelope, type BridgePairingEnvelope,
} from '@/protocol/bridge';
import { getBridgeRuntimeSession, getState, setBridgeRuntimeSession, updateState } from '@/platform/storage/state';
import { routeCapability } from '@/features/grants/service';
@@ -28,6 +32,25 @@ const HANDSHAKE_TIMEOUT = 5_000;
const MAX_CONCURRENT_REQUESTS = 8;
const ENGINE_REQUEST_TIMEOUT = 10_000;
const MAX_OUTGOING_REQUESTS = 4;
const DISCOVERY_PORTS = 16;
const DISCOVERY_TIMEOUT = 800;
export function browserClientIdentity(
config: BridgeConfig,
userAgent = globalThis.navigator?.userAgent || '',
): { client: string; version: string } {
const detected = [
[/\bEdg(?:A|iOS)?\/([\d.]+)/, 'Microsoft Edge'],
[/\b(?:Chrome|CriOS)\/([\d.]+)/, 'Google Chrome'],
[/\bChromium\/([\d.]+)/, 'Chromium'],
[/\bFirefox\/([\d.]+)/, 'Firefox'],
].map(([pattern, name]) => ({ match: userAgent.match(pattern as RegExp), name: name as string }))
.find(({ match }) => match);
return {
client: config.browserName?.trim() || detected?.name || 'Browser',
version: config.browserVersion?.trim() || detected?.match?.[1] || '',
};
}
interface OutgoingRequest {
resolve: (value: unknown) => void;
@@ -115,6 +138,45 @@ export class EngineBridge {
return this.pairingStatus;
}
async discoverLocalEngines(): Promise<DiscoveredYakEngine[]> {
const config = (await getState()).bridge;
if (config.transport !== 'websocket' || !isLoopbackEndpoint(config.endpoint)) return [];
const configured = new URL(config.endpoint);
const firstPort = Number(configured.port) || (configured.protocol === 'wss:' ? 443 : 80);
const candidates = Array.from(
{ length: Math.min(DISCOVERY_PORTS, 65_536 - firstPort) },
(_, offset) => {
const url = new URL(configured);
url.port = String(firstPort + offset);
url.pathname = '/discovery';
url.search = '';
url.hash = '';
return url.toString();
},
);
const results = await Promise.all(candidates.map((url) => new Promise<DiscoveredYakEngine | undefined>((resolve) => {
let socket: WebSocket;
try { socket = new WebSocket(url); } catch { resolve(undefined); return; }
let settled = false;
const finish = (engine?: DiscoveredYakEngine) => {
if (settled) return;
settled = true;
globalThis.clearTimeout(timer);
if (socket.readyState !== WebSocket.CLOSED) socket.close();
resolve(engine);
};
const timer = globalThis.setTimeout(() => finish(), DISCOVERY_TIMEOUT);
socket.addEventListener('message', (event) => {
try { finish(parseBridgeDiscoveryEnvelope(String(event.data))); } catch { finish(); }
});
socket.addEventListener('error', () => finish());
socket.addEventListener('close', () => finish());
})));
const engines = results.filter((engine): engine is DiscoveredYakEngine => Boolean(engine));
return [...new Map(engines.map((engine) => [`${engine.engineIdentityId}:${engine.engineInstanceId}`, engine])).values()]
.sort((left, right) => Number(new URL(left.endpoint).port) - Number(new URL(right.endpoint).port));
}
emitEvent(method: string, params: unknown): void {
if (this.status.state !== 'connected') return;
this.trySend({ type: 'event', method, params });
@@ -293,10 +355,11 @@ export class EngineBridge {
const [state, previousSession] = await Promise.all([getState(), getBridgeRuntimeSession()]);
const identity = await getOrCreateBrowserBridgeIdentity(config.installationId);
const capabilityCatalog = await getBridgeCapabilityCatalog();
const browserIdentity = browserClientIdentity(config);
const auth: BridgeEnvelope = {
type: 'auth',
client: 'yakit-browser-extension',
version: browser.runtime.getManifest().version,
client: browserIdentity.client,
version: browserIdentity.version,
protocolVersion: BRIDGE_PROTOCOL_VERSION,
capabilities: [...BRIDGE_CAPABILITIES],
capabilityCatalog,
@@ -525,11 +588,13 @@ export class EngineBridge {
try {
const grant = await browserInstanceAccess('browser.tabs.read');
taskId = grant.taskId;
actionId = (await beginAgentAction(grant, {
requestId: message.id,
method: message.method,
targetTabId,
})).id;
if (capabilityVisibleToAgent(message.method)) {
actionId = (await beginAgentAction(grant, {
requestId: message.id,
method: message.method,
targetTabId,
})).id;
}
const operation = routeCapability(message.method, message.params, (method, params) => this.requestEngine(method, params));
const result = await Promise.race([operation, cancelled]);
const durationMs = performance.now() - startedAt;
@@ -775,6 +840,7 @@ export class EngineBridge {
if (this.pairingSocket && ['requesting', 'pending'].includes(currentPairing.state)) return currentPairing;
this.cancelPairing(false);
const identity = await getOrCreateBrowserBridgeIdentity(config.installationId);
const browserIdentity = browserClientIdentity(config);
const clientNonce = randomBridgeNonce();
const pairingURL = new URL(config.endpoint);
pairingURL.pathname = '/pairing';
@@ -799,7 +865,7 @@ export class EngineBridge {
type: 'pair_request', protocolVersion: BRIDGE_PROTOCOL_VERSION,
installationId: config.installationId,
managedInstance: config.managedInstance,
client: 'yakit-browser-extension', version: browser.runtime.getManifest().version,
client: browserIdentity.client, version: browserIdentity.version,
nonce: clientNonce, publicKey: identity.publicKey,
} satisfies BridgePairingEnvelope));
} catch (error) {
@@ -11,6 +11,7 @@ import { AGENT_RUNTIME_STORAGE_KEY, isStateStorageChange } from '@/protocol/stor
import type { ActiveTabInfo, AgentRuntime, BridgeStatus, ExtensionState, PageContext } from '@/types/models';
import { errorMessage, request } from '@/platform/messaging/runtime';
import { isFloatingPanelShortcut, mergeFloatingTabUpdate } from './host-controller';
import { ProxyStatusBar, StartupProxyOption, useProxyStatus } from '@/features/proxy/ui/ProxyStatusBar';
interface FloatingPanelProps {
initialState: ExtensionState;
@@ -21,6 +22,7 @@ interface FloatingPanelProps {
export function FloatingPanel({ initialState, initialTab, initialBridge, hostChannel }: FloatingPanelProps) {
const [state, setState] = useState(initialState);
const proxyStatus = useProxyStatus(state);
const [bridge, setBridge] = useState(initialBridge);
const [tab, setTab] = useState(initialTab);
const [busy, setBusy] = useState(false);
@@ -142,15 +144,17 @@ export function FloatingPanel({ initialState, initialTab, initialBridge, hostCha
</TabsList>
<TabsContent value="proxy" className="floating-tab-content">
<ProxyStatusBar status={proxyStatus} />
<div className="floating-section-heading"><span>快速切换</span><Button size="icon" variant="ghost" title="代理设置" onClick={() => openWorkspace('proxies')}><Settings size={15} /></Button></div>
<div className="floating-option-list">
<StartupProxyOption state={state} status={proxyStatus} setState={setState} run={run} busy={busy} />
{state.proxyProfiles.map((profile) => (
<button key={profile.id} className={state.activeProxyId === profile.id ? 'is-active' : ''} disabled={busy} onClick={() => void run(async () => setState(await request('proxy.switch', { id: profile.id })))}>
<button key={profile.id} className={proxyStatus.activeProfileId === profile.id ? 'is-active' : ''} disabled={busy} onClick={() => void run(async () => setState(await request('proxy.switch', { id: profile.id })))}>
<i className="floating-radio" />
<span><strong>{profile.name}</strong><small>{profile.kind === 'fixed_servers' ? `${profile.host}:${profile.port}` : profile.kind}</small></span>
</button>
))}
<button className={state.activeProxyId === 'auto' ? 'is-active' : ''} disabled={busy} onClick={() => void run(async () => setState(await request('proxy.auto.apply')))}><i className="floating-radio" /><span><strong>自动切换</strong><small>{state.proxyRules.filter((rule) => rule.enabled).length} 条手动 · {state.proxyRuleSources.filter((source) => source.enabled).length} 个订阅</small></span></button>
<button className={proxyStatus.activeProfileId === 'auto' ? 'is-active' : ''} disabled={busy} onClick={() => void run(async () => setState(await request('proxy.auto.apply')))}><i className="floating-radio" /><span><strong>自动切换</strong><small>{state.proxyRules.filter((rule) => rule.enabled).length} 条手动 · {state.proxyRuleSources.filter((source) => source.enabled).length} 个订阅</small></span></button>
</div>
</TabsContent>
+2 -8
View File
@@ -1,6 +1,5 @@
export type CapabilityDomainId =
| 'navigation-isolation'
| 'authorization'
| 'handoff'
| 'network'
| 'recording-callable-debugger'
@@ -32,11 +31,6 @@ export const NAVIGATION_CAPABILITY_DOMAIN = exactMethods('navigation-isolation',
'browser.isolation.container.remove',
]);
export const AUTHORIZATION_CAPABILITY_DOMAIN: CapabilityDomainDefinition = {
id: 'authorization',
owns: (method) => method.startsWith('browser.authorization.'),
};
export const HANDOFF_CAPABILITY_DOMAIN: CapabilityDomainDefinition = {
id: 'handoff',
owns: (method) => method.startsWith('browser.handoff.'),
@@ -49,7 +43,8 @@ export const NETWORK_CAPABILITY_DOMAIN: CapabilityDomainDefinition = {
export const RECORDING_CAPABILITY_DOMAIN: CapabilityDomainDefinition = {
id: 'recording-callable-debugger',
owns: (method) => method.startsWith('browser.recording.')
owns: (method) => method === 'browser.crypto.inspect'
|| method.startsWith('browser.recording.')
|| method.startsWith('browser.callable.')
|| method.startsWith('browser.deep_capture.'),
};
@@ -78,7 +73,6 @@ export const PROXY_CAPABILITY_DOMAIN = exactMethods('proxy', [
export const CAPABILITY_DOMAINS: readonly CapabilityDomainDefinition[] = [
NAVIGATION_CAPABILITY_DOMAIN,
AUTHORIZATION_CAPABILITY_DOMAIN,
HANDOFF_CAPABILITY_DOMAIN,
NETWORK_CAPABILITY_DOMAIN,
RECORDING_CAPABILITY_DOMAIN,
@@ -1,153 +0,0 @@
import type { BrowserAuthorizationResourceSelector } from '@/types/models';
import type { CapabilityDomainHandler } from '../capability-context';
import { allowedTarget, requireScope } from '../capability-context';
import {
captureAuthContextHandle,
getAuthContextHandle,
} from '@/features/authorization-testing/auth-context';
import {
captureAuthContextAttestation,
getAuthContextAttestation,
} from '@/features/authorization-testing/auth-attestation';
import {
bindAuthorizationBaselineLogicalRequest,
captureAuthorizationBaseline,
compileAuthorizationBaseline,
compileAuthorizationBaselinePacket,
compileAuthorizationBaselineWithTransform,
getAuthorizationBaseline,
inspectAuthorizationBaselineTransform,
listAuthorizationBaselineCandidates,
readAuthorizationBaselineResource,
} from '@/features/authorization-testing/baseline';
import { AUTHORIZATION_CAPABILITY_DOMAIN } from '../capability-domains';
function requireAuthorizationContextScopes(
grant: Parameters<typeof requireScope>[0],
): void {
requireScope(grant, 'browser.cookies.read');
requireScope(grant, 'browser.storage.read');
}
function requireAuthorizationBaselineScopes(
grant: Parameters<typeof requireScope>[0],
): void {
requireScope(grant, 'browser.isolation.read');
requireAuthorizationContextScopes(grant);
}
export const authorizationCapabilityHandler: CapabilityDomainHandler = {
...AUTHORIZATION_CAPABILITY_DOMAIN,
async handle({ method, input, grant }) {
if (method === 'browser.authorization.context.capture') {
requireAuthorizationContextScopes(grant);
return captureAuthContextHandle({
slotId: input.slotId === 'right' ? 'right' : 'left',
accountLabel: typeof input.accountLabel === 'string' ? input.accountLabel : undefined,
isolationProofId: String(input.isolationProofId || ''),
target: await allowedTarget(grant, input),
grantId: grant.id,
grantExpiresAt: grant.expiresAt,
});
}
if (method === 'browser.authorization.context.get') {
requireAuthorizationContextScopes(grant);
return getAuthContextHandle(String(input.id || ''), grant.id);
}
if (method === 'browser.authorization.context.attest') {
requireAuthorizationContextScopes(grant);
return captureAuthContextAttestation({
target: await allowedTarget(grant, input),
grantId: grant.id,
grantExpiresAt: grant.expiresAt,
});
}
if (method === 'browser.authorization.context.attestation.get') {
requireAuthorizationContextScopes(grant);
return getAuthContextAttestation(String(input.id || ''), grant.id);
}
requireAuthorizationBaselineScopes(grant);
if (method === 'browser.authorization.baseline.capture') {
return captureAuthorizationBaseline({
target: await allowedTarget(grant, input),
grantId: grant.id,
authContextKind: input.authContextKind === 'attestation' ? 'attestation' : 'handle',
authContextId: String(input.authContextId || ''),
networkRequestId: String(input.networkRequestId || ''),
comparisonKey: String(input.comparisonKey || ''),
});
}
if (method === 'browser.authorization.baseline.candidates') {
return listAuthorizationBaselineCandidates({
target: await allowedTarget(grant, input),
grantId: grant.id,
authContextKind: input.authContextKind === 'attestation' ? 'attestation' : 'handle',
authContextId: String(input.authContextId || ''),
limit: typeof input.limit === 'number' ? input.limit : 100,
});
}
if (method === 'browser.authorization.baseline.get') {
return getAuthorizationBaseline(String(input.id || ''), grant.id);
}
if (method === 'browser.authorization.baseline.logical.bind') {
requireScope(grant, 'browser.network.sensitive.read');
requireScope(grant, 'browser.transform.execute');
return bindAuthorizationBaselineLogicalRequest({
id: String(input.id || ''),
grantId: grant.id,
profileId: String(input.profileId || ''),
comparisonKey: String(input.comparisonKey || ''),
});
}
if (method === 'browser.authorization.baseline.resource.get') {
return readAuthorizationBaselineResource({
id: String(input.id || ''),
grantId: grant.id,
selector: input.selector as BrowserAuthorizationResourceSelector,
});
}
if (method === 'browser.authorization.baseline.compile') {
requireScope(grant, 'browser.network.sensitive.read');
return compileAuthorizationBaseline({
id: String(input.id || ''),
grantId: grant.id,
selector: input.selector as BrowserAuthorizationResourceSelector,
replacement: input.replacement as Parameters<typeof compileAuthorizationBaseline>[0]['replacement'],
comparisonKey: String(input.comparisonKey || ''),
});
}
if (method === 'browser.authorization.baseline.packet.compile') {
requireScope(grant, 'browser.network.replay');
requireScope(grant, 'browser.network.sensitive.read');
return compileAuthorizationBaselinePacket({
id: String(input.id || ''),
grantId: grant.id,
});
}
if (method === 'browser.authorization.baseline.transform.inspect') {
requireScope(grant, 'browser.network.sensitive.read');
requireScope(grant, 'browser.transform.read');
return inspectAuthorizationBaselineTransform({
id: String(input.id || ''),
grantId: grant.id,
profileId: String(input.profileId || ''),
});
}
if (method === 'browser.authorization.baseline.transform.compile') {
requireScope(grant, 'browser.network.replay');
requireScope(grant, 'browser.network.sensitive.read');
requireScope(grant, 'browser.transform.execute');
return compileAuthorizationBaselineWithTransform({
id: String(input.id || ''),
grantId: grant.id,
selector: input.selector as BrowserAuthorizationResourceSelector,
replacement: input.replacement as Parameters<typeof compileAuthorizationBaselineWithTransform>[0]['replacement'],
comparisonKey: String(input.comparisonKey || ''),
profileId: String(input.profileId || ''),
bindingFingerprint: String(input.bindingFingerprint || ''),
});
}
throw new Error(`授权能力没有实现: ${method}`);
},
};
@@ -2,16 +2,29 @@ import { browser } from 'wxt/browser';
import type { HandoffReason } from '@/types/models';
import type { CapabilityDomainHandler } from '../capability-context';
import { allowedTarget } from '../capability-context';
import { activateTab } from '@/platform/browser/targets';
import { getTab } from '@/platform/browser/targets';
import { getState, updateState } from '@/platform/storage/state';
import { setAgentRuntimeState } from '@/features/agent-runtime/service';
import { ExtensionError } from '@/shared/errors';
import { HANDOFF_CAPABILITY_DOMAIN } from '../capability-domains';
import { focusHandoff, getHandoffPresentation, resolveHandoff } from '@/features/handoff/service';
export const handoffCapabilityHandler: CapabilityDomainHandler = {
...HANDOFF_CAPABILITY_DOMAIN,
async handle({ method, input, grant }) {
if (method === 'browser.handoff.presentation.get') {
return getHandoffPresentation(String(input.handoffId || ''), grant);
}
if (method === 'browser.handoff.focus') {
return focusHandoff(String(input.handoffId || ''), grant);
}
if (method === 'browser.handoff.resolve') {
return resolveHandoff(
String(input.handoffId || ''),
input.outcome === 'cancelled' ? 'cancelled' : 'completed',
grant,
);
}
if (method === 'browser.handoff.status') {
const handoff = (await getState()).handoff;
return handoff?.taskId === grant.taskId ? handoff : { state: 'idle' };
@@ -50,7 +63,6 @@ export const handoffCapabilityHandler: CapabilityDomainHandler = {
},
};
});
await activateTab(resolvedTarget.tabId);
await browser.action.setBadgeBackgroundColor({ color: '#ee7815' });
await browser.action.setBadgeText({ text: '待确认', tabId: resolvedTarget.tabId });
await setAgentRuntimeState('waiting_for_human', grant);
@@ -3,7 +3,9 @@ import type {
YakPocGenerateResult,
} from '@/types/models';
import type { CapabilityDomainHandler } from '../capability-context';
import { allowedTarget, requireScope } from '../capability-context';
import {
allowedTarget, PAIRED_BROWSER_INSTANCE_ACCESS_ID, requireScope,
} from '../capability-context';
import {
clearNetworkRequests,
exportNetworkRequest,
@@ -30,7 +32,12 @@ export const networkCapabilityHandler: CapabilityDomainHandler = {
captureBody: input.captureBody === true,
maxEntries: typeof input.maxEntries === 'number' ? input.maxEntries : undefined,
maxBodyBytes: typeof input.maxBodyBytes === 'number' ? input.maxBodyBytes : undefined,
}, { kind: 'grant', grantId: grant.id, expiresAt: grant.expiresAt });
}, {
kind: 'grant',
grantId: grant.id,
expiresAt: grant.expiresAt,
followSameOriginNavigation: grant.id === PAIRED_BROWSER_INSTANCE_ACCESS_ID,
});
}
if (method === 'browser.network.status') return networkCaptureStatus(target);
if (method === 'browser.network.list') {
@@ -14,6 +14,7 @@ import { listCookies } from '@/features/cookies/service';
import { resolveTabCookieStoreId } from '@/platform/browser/isolation';
import { ExtensionError } from '@/shared/errors';
import { PAGE_CAPABILITY_DOMAIN } from '../capability-domains';
import { getState } from '@/platform/storage/state';
export const pageCapabilityHandler: CapabilityDomainHandler = {
...PAGE_CAPABILITY_DOMAIN,
@@ -63,7 +64,12 @@ export const pageCapabilityHandler: CapabilityDomainHandler = {
await browser.action.setBadgeBackgroundColor({ color: '#f28c28' });
await browser.action.setBadgeText({ text: '接管', tabId: target.tabId });
globalThis.setTimeout(
() => void browser.action.setBadgeText({ text: '', tabId: target.tabId }),
() => void getState()
.then((state) => browser.action.setBadgeText({
text: state.bridge.managedInstance?.badge || '',
tabId: target.tabId,
}))
.catch(() => undefined),
10_000,
);
return { activated: true, target };
@@ -33,10 +33,28 @@ import {
stageBrowserProfileEvidence,
} from '@/features/browser-analysis/service';
import { RECORDING_CAPABILITY_DOMAIN } from '../capability-domains';
import { inspectPageCryptoOperation } from '@/features/browser-crypto/inspect';
export const recordingCapabilityHandler: CapabilityDomainHandler = {
...RECORDING_CAPABILITY_DOMAIN,
async handle({ method, input, grant }) {
if (method === 'browser.crypto.inspect') {
for (const scope of [
'browser.recording.control',
'browser.recording.sensitive.read',
'browser.network.capture',
'browser.network.sensitive.read',
] as const) requireScope(grant, scope);
return inspectPageCryptoOperation(
await allowedTarget(grant, input),
{
captureId: String(input.captureId || ''),
nodeId: String(input.nodeId || ''),
settleMs: typeof input.settleMs === 'number' ? input.settleMs : undefined,
},
{ grantId: grant.id, expiresAt: grant.expiresAt },
);
}
if (method.startsWith('browser.recording.')) {
const target = await allowedTarget(grant, input);
if (method === 'browser.recording.trace.list') {
@@ -140,6 +158,9 @@ export const recordingCapabilityHandler: CapabilityDomainHandler = {
return createRecordedPageCallable(target, {
callHandleId: String(input.callHandleId || ''),
name: String(input.name || ''),
dynamicInputPaths: Array.isArray(input.dynamicInputPaths)
? input.dynamicInputPaths.map(String)
: undefined,
});
}
if (method === 'browser.callable.execute') {
@@ -1,9 +1,9 @@
import type {
BrowserTransformExecuteInput,
BrowserTransformPacket,
BrowserTransformValidationExecuteInput,
BrowserTransformProfileInput,
} from '@/types/models';
import { browser } from 'wxt/browser';
import type { CapabilityDomainHandler } from '../capability-context';
import { allowedTarget, requireScope } from '../capability-context';
import {
@@ -14,23 +14,58 @@ import {
getBrowserTransformProfile,
getBrowserTransformRecovery,
listBrowserTransformProfiles,
resetBrowserTransformRecovery,
saveBrowserTransformProfile,
resetBrowserTransformRecovery,
startBrowserTransformRecovery,
validateBrowserTransformRecovery,
} from '@/features/browser-transform/service';
import {
compareBrowserPackets,
browserTransformValidationById,
executeBrowserTransformValidation,
latestBrowserTransformValidation,
prepareCapturedBrowserTransformProfile,
proposeBrowserTransformProfile,
validateInferredBrowserTransformProfile,
} from '@/features/browser-analysis/service';
import { ExtensionError } from '@/shared/errors';
import { TRANSFORM_CAPABILITY_DOMAIN } from '../capability-domains';
export const transformCapabilityHandler: CapabilityDomainHandler = {
...TRANSFORM_CAPABILITY_DOMAIN,
async handle({ method, input, grant }) {
if (method === 'browser.transform.profile.save') {
const profile = input as unknown as BrowserTransformProfileInput;
await allowedTarget(grant, profile.target);
return saveBrowserTransformProfile(profile);
}
if (method === 'browser.transform.validation.get') {
const draft = await browserTransformValidationById(String(input.validationId || ''));
await allowedTarget(grant, draft.profile.target);
return {
id: draft.id, expiresAt: draft.expiresAt,
directions: { request: draft.profile.request.enabled, response: draft.profile.response.enabled },
};
}
if (method === 'browser.transform.prepare') {
requireScope(grant, 'browser.recording.read');
requireScope(grant, 'browser.callable.execute');
return prepareCapturedBrowserTransformProfile(
await allowedTarget(grant, input),
String(input.candidateId || ''),
input.packet as BrowserTransformPacket,
Array.isArray(input.inputPaths) ? input.inputPaths.map(String) : undefined,
typeof input.name === 'string' ? input.name : undefined,
{
owner: { kind: 'grant', grantId: grant.id, expiresAt: grant.expiresAt },
trigger: input.trigger as { captureId: string; nodeId: string } | undefined,
authorize: () => {
requireScope(grant, 'browser.debugger.control');
requireScope(grant, 'browser.dom.read');
requireScope(grant, 'browser.dom.write');
},
},
);
}
if (method === 'browser.packet.compare') {
await allowedTarget(grant, input);
return compareBrowserPackets(
@@ -122,20 +157,22 @@ export const transformCapabilityHandler: CapabilityDomainHandler = {
}));
return visible.filter(Boolean);
}
if (method === 'browser.transform.profile.save') {
const profileInput = input as unknown as BrowserTransformProfileInput;
const target = await allowedTarget(grant, profileInput.target);
const frame = await browser.webNavigation.getFrame(target);
if (!frame?.url || profileInput.origin !== new URL(frame.url).origin) {
throw new ExtensionError('target_denied', '转换配置来源与当前页面不一致');
}
return saveBrowserTransformProfile({ ...profileInput, target });
}
if (method === 'browser.transform.profile.delete') {
const profile = await getBrowserTransformProfile(String(input.id || ''));
await allowedTarget(grant, profile.target);
return deleteBrowserTransformProfile(profile.id);
}
if (method === 'browser.transform.validation.execute') {
requireScope(grant, 'browser.transform.execute');
const executeValidationInput = input as unknown as BrowserTransformValidationExecuteInput;
const draft = await browserTransformValidationById(executeValidationInput.validationId);
await allowedTarget(grant, draft.profile.target);
return executeBrowserTransformValidation(
executeValidationInput.validationId,
executeValidationInput.direction,
executeValidationInput.packet,
);
}
const executeInput = input as unknown as BrowserTransformExecuteInput;
const profile = await getBrowserTransformProfile(executeInput.profileId);
await allowedTarget(grant, profile.target);
-2
View File
@@ -3,7 +3,6 @@ import type {
CapabilityRouteContext,
} from './capability-context';
import { navigationCapabilityHandler } from './capability-handlers/navigation';
import { authorizationCapabilityHandler } from './capability-handlers/authorization';
import { handoffCapabilityHandler } from './capability-handlers/handoff';
import { networkCapabilityHandler } from './capability-handlers/network';
import { recordingCapabilityHandler } from './capability-handlers/recording';
@@ -13,7 +12,6 @@ import { proxyCapabilityHandler } from './capability-handlers/proxy';
export const CAPABILITY_HANDLERS: readonly CapabilityDomainHandler[] = [
navigationCapabilityHandler,
authorizationCapabilityHandler,
handoffCapabilityHandler,
networkCapabilityHandler,
recordingCapabilityHandler,
+1 -31
View File
@@ -1,7 +1,7 @@
import { describe, expect, it } from 'vitest';
import { CONTROL_CAPABILITY_SCOPES } from '@/protocol/capabilities';
import type { ActiveTabInfo, ExtensionState } from '@/types/models';
import { authorizationShareGrantInput, gatewayShareActive, gatewayShareGrantInput } from './gateway-share';
import { gatewayShareActive, gatewayShareGrantInput } from './gateway-share';
const NOW = 1_000_000;
@@ -84,34 +84,4 @@ describe('gateway quick share', () => {
expect(gatewayShareActive(grant, { ...tab, url: 'https://elsewhere.example.test/' }, NOW)).toBe(false);
});
it('creates a focused two-tab authorization grant without retaining unrelated targets', () => {
const current = state();
current.activeGrant = {
id: 'grant',
taskId: 'existing-task',
createdAt: NOW - 1_000,
expiresAt: NOW + 45 * 60_000,
scopes: ['browser.tabs.read'],
targets: [{
tabId: 99,
frameId: 0,
documentId: 'unrelated',
isolationContextId: 'unrelated',
origin: 'https://other.example.test',
grantedUrl: 'https://other.example.test',
title: 'Unrelated',
}],
};
const right = { ...tab, id: 8, incognito: true };
const input = authorizationShareGrantInput(current, [tab, right], NOW);
expect(input.targets).toEqual([
{ tabId: 7, frameId: 0 },
{ tabId: 8, frameId: 0 },
]);
expect(input.scopes).toEqual(expect.arrayContaining(CONTROL_CAPABILITY_SCOPES));
expect(input.durationMinutes).toBe(45);
expect(input.taskId).toBe('existing-task');
});
});
-21
View File
@@ -63,24 +63,3 @@ export function gatewayShareGrantInput(
taskId: active?.taskId,
};
}
export function authorizationShareGrantInput(
state: ExtensionState,
tabs: [ActiveTabInfo, ActiveTabInfo],
now = Date.now(),
): GrantCreateInput {
const active = state.activeGrant && state.activeGrant.expiresAt > now
? state.activeGrant
: undefined;
const scopes = new Set<CapabilityScope>(active?.scopes || []);
CONTROL_CAPABILITY_SCOPES.forEach((scope) => scopes.add(scope));
const remainingMinutes = active
? Math.ceil((active.expiresAt - now) / 60_000)
: 0;
return {
targets: tabs.map((item) => ({ tabId: item.id, frameId: 0 })),
scopes: [...scopes],
durationMinutes: Math.max(DEFAULT_GATEWAY_GRANT_MINUTES, remainingMinutes),
taskId: active?.taskId,
};
}
+110
View File
@@ -0,0 +1,110 @@
import { beforeEach, describe, expect, it, vi } from 'vitest';
const fixture = vi.hoisted(() => ({
state: {} as Record<string, unknown>,
activateTab: vi.fn(async () => undefined),
getFrame: vi.fn(),
getTab: vi.fn(),
resolveDocumentTarget: vi.fn(),
executeScript: vi.fn(),
scriptingTarget: vi.fn((target) => target),
}));
vi.mock('wxt/browser', () => ({
browser: {
storage: {},
webNavigation: { getFrame: fixture.getFrame },
scripting: { executeScript: fixture.executeScript },
},
}));
vi.mock('@/platform/storage/state', () => ({
getState: vi.fn(async () => structuredClone(fixture.state)),
updateState: vi.fn(),
}));
vi.mock('@/platform/browser/targets', () => ({
activateTab: fixture.activateTab,
getTab: fixture.getTab,
resolveDocumentTarget: fixture.resolveDocumentTarget,
scriptingTarget: fixture.scriptingTarget,
}));
import { ExtensionError } from '@/shared/errors';
import { focusHandoff, getHandoffPresentation, isSafeHandoffPresentationDataUrl } from './service';
const grant = {
id: 'paired-browser-instance',
taskId: 'paired-browser-instance',
targets: [],
scopes: [],
createdAt: 0,
expiresAt: Number.MAX_SAFE_INTEGER,
};
function waitingHandoff(origin = 'https://passport.example.test') {
return {
handoff: {
id: 'handoff-1',
taskId: 'paired-browser-instance',
state: 'waiting_for_user',
reason: 'qr_code',
target: {
tabId: 7,
frameId: 0,
documentId: 'document-old',
origin,
grantedUrl: `${origin}/login`,
title: 'Sign in',
},
},
};
}
describe('handoff presentation data URL validation', () => {
beforeEach(() => {
vi.clearAllMocks();
fixture.state = {};
});
it('accepts bounded raster data and rejects executable or oversized content', () => {
expect(isSafeHandoffPresentationDataUrl('data:image/png;base64,AAAA')).toBe(true);
expect(isSafeHandoffPresentationDataUrl('data:image/svg+xml,<svg onload="alert(1)"/>')).toBe(false);
expect(isSafeHandoffPresentationDataUrl(`data:image/png;base64,${'AAAA'.repeat(350_000)}`)).toBe(false);
});
it('focuses only the waiting handoff owned by the local paired task', async () => {
fixture.state = {
handoff: {
id: 'handoff-1',
taskId: 'paired-browser-instance',
state: 'waiting_for_user',
target: { tabId: 7 },
},
};
await expect(focusHandoff('handoff-1', grant)).resolves.toEqual({ focused: true, tabId: 7 });
expect(fixture.activateTab).toHaveBeenCalledWith(7);
await expect(focusHandoff('other-handoff', grant)).rejects.toMatchObject({ code: 'handoff_not_waiting' });
});
it('rebinds presentation reads after a same-origin document refresh', async () => {
fixture.state = waitingHandoff();
fixture.resolveDocumentTarget
.mockRejectedValueOnce(new ExtensionError('stale_document', 'stale'))
.mockResolvedValueOnce({ tabId: 7, frameId: 0, documentId: 'document-new' });
fixture.getFrame.mockResolvedValue({ url: 'https://passport.example.test/login?refreshed=1' });
fixture.getTab.mockResolvedValue({ id: 7 });
fixture.executeScript.mockResolvedValue([]);
await expect(getHandoffPresentation('handoff-1', grant)).resolves.toMatchObject({ state: 'not_found' });
expect(fixture.resolveDocumentTarget).toHaveBeenLastCalledWith({ tabId: 7, frameId: 0 });
});
it('reports a changed page instead of leaking a stale-document error', async () => {
fixture.state = waitingHandoff();
fixture.resolveDocumentTarget.mockRejectedValueOnce(new ExtensionError('stale_document', 'stale'));
fixture.getFrame.mockResolvedValue({ url: 'https://www.example.test/' });
await expect(getHandoffPresentation('handoff-1', grant)).resolves.toMatchObject({ state: 'page_changed' });
expect(fixture.resolveDocumentTarget).toHaveBeenCalledTimes(1);
});
});
+299
View File
@@ -0,0 +1,299 @@
import { browser, type Browser } from 'wxt/browser';
import { setAgentRuntimeState } from '@/features/agent-runtime/service';
import { browserInstanceAccess } from '@/features/grants/capability-context';
import { activateTab, getTab, resolveDocumentTarget, scriptingTarget } from '@/platform/browser/targets';
import { getState, updateState } from '@/platform/storage/state';
import { ExtensionError } from '@/shared/errors';
import type { BridgeGrant, HandoffState, HumanHandoff } from '@/types/models';
const MAX_PRESENTATION_BYTES = 1024 * 1024;
const SAFE_RASTER_DATA_URL = /^data:image\/(?:png|jpeg|webp);base64,/i;
interface PageQrCandidate {
dataUrl?: string;
source: 'image' | 'canvas' | 'svg' | 'background' | 'screenshot';
rect: { x: number; y: number; width: number; height: number };
viewport: { width: number; height: number; devicePixelRatio: number };
title: string;
url: string;
}
export interface HandoffPresentation {
handoffId: string;
state: HandoffState | 'not_found' | 'page_changed';
title: string;
url: string;
capturedAt: number;
source?: PageQrCandidate['source'];
dataUrl?: string;
}
async function resolvePresentationTarget(target: HumanHandoff['target']) {
try {
return await resolveDocumentTarget(target);
} catch (error) {
if (!(error instanceof ExtensionError) || !['stale_document', 'target_unavailable'].includes(error.code)) throw error;
}
const frame = await browser.webNavigation.getFrame({
tabId: target.tabId,
frameId: target.frameId,
}).catch(() => null);
if (!frame?.url || !/^https?:/i.test(frame.url) || new URL(frame.url).origin !== target.origin) return undefined;
return resolveDocumentTarget({ tabId: target.tabId, frameId: target.frameId }).catch(() => undefined);
}
function dataUrlBytes(value: string): number {
const comma = value.indexOf(',');
return comma < 0 ? Number.MAX_SAFE_INTEGER : Math.ceil((value.length - comma - 1) * 0.75);
}
export function isSafeHandoffPresentationDataUrl(value: unknown): value is string {
return typeof value === 'string'
&& SAFE_RASTER_DATA_URL.test(value)
&& dataUrlBytes(value) <= MAX_PRESENTATION_BYTES;
}
async function findQrCandidateInPage(): Promise<PageQrCandidate | undefined> {
const resolvePresentationDataUrl = async (source: string, width: number, height: number): Promise<string | undefined> => {
const rasterize = async (url: string): Promise<string | undefined> => {
const image = new Image();
image.decoding = 'async';
await new Promise<void>((resolve, reject) => {
image.onload = () => resolve();
image.onerror = () => reject(new Error('image load failed'));
image.src = url;
});
const canvas = document.createElement('canvas');
canvas.width = Math.min(1024, Math.max(1, image.naturalWidth || Math.round(width)));
canvas.height = Math.min(1024, Math.max(1, image.naturalHeight || Math.round(height)));
canvas.getContext('2d')?.drawImage(image, 0, 0, canvas.width, canvas.height);
return canvas.toDataURL('image/png');
};
try {
if (/^data:image\/(?:png|jpeg|webp);base64,/i.test(source)) return source;
if (/^data:image\/svg\+xml/i.test(source)) return rasterize(source);
if (!/^(?:blob:|https?:)/i.test(source)) return undefined;
const response = await fetch(source);
if (!response.ok) return undefined;
const blob = await response.blob();
if (blob.size > 1024 * 1024) return undefined;
const localUrl = URL.createObjectURL(blob);
try {
if (/^image\/(?:png|jpeg|webp)$/i.test(blob.type)) {
return await new Promise<string>((resolve, reject) => {
const reader = new FileReader();
reader.onload = () => resolve(String(reader.result || ''));
reader.onerror = () => reject(reader.error);
reader.readAsDataURL(blob);
});
}
if (blob.type === 'image/svg+xml') return await rasterize(localUrl);
} finally {
URL.revokeObjectURL(localUrl);
}
} catch {
return undefined;
}
return undefined;
};
const keywords = /(?:^|[^a-z])(qr|qrcode|scan)(?:[^a-z]|$)|二维码|扫码|扫码登录/i;
const selector = 'img,canvas,svg,[role="img"],[class*="qr" i],[id*="qr" i]';
const seen = new Set<Element>();
const candidates: Array<{ element: Element; score: number; rect: DOMRect }> = [];
for (const element of document.querySelectorAll(selector)) {
if (seen.has(element)) continue;
seen.add(element);
const rect = element.getBoundingClientRect();
const style = getComputedStyle(element);
if (
rect.width < 96 || rect.height < 96
|| rect.bottom <= 0 || rect.right <= 0
|| rect.top >= innerHeight || rect.left >= innerWidth
|| style.display === 'none' || style.visibility === 'hidden' || Number(style.opacity) === 0
) continue;
const ratio = rect.width / rect.height;
if (ratio < 0.72 || ratio > 1.38) continue;
const ownText = [
element.id,
element.getAttribute('class'),
element.getAttribute('alt'),
element.getAttribute('aria-label'),
element.getAttribute('title'),
].filter(Boolean).join(' ');
let contextText = '';
let parent: Element | null = element;
for (let depth = 0; parent && depth < 4; depth += 1, parent = parent.parentElement) {
contextText += ` ${parent.textContent || ''}`;
if (contextText.length >= 500) break;
}
const inDialog = Boolean(element.closest('dialog,[role="dialog"],[aria-modal="true"]'));
const score = (keywords.test(ownText) ? 8 : 0)
+ (keywords.test(contextText.slice(0, 500)) ? 5 : 0)
+ (Math.abs(1 - ratio) < 0.12 ? 4 : 2)
+ (inDialog ? 2 : 0)
+ (element instanceof HTMLCanvasElement || element instanceof SVGElement ? 1 : 0);
if (score >= 6) candidates.push({ element, score, rect });
}
candidates.sort((left, right) => right.score - left.score || right.rect.width - left.rect.width);
for (const { element, rect } of candidates.slice(0, 8)) {
let source: PageQrCandidate['source'] = 'screenshot';
let dataUrl: string | undefined;
try {
if (element instanceof HTMLCanvasElement) {
source = 'canvas';
dataUrl = element.toDataURL('image/png');
} else if (element instanceof SVGElement) {
source = 'svg';
const svg = new XMLSerializer().serializeToString(element);
dataUrl = await resolvePresentationDataUrl(
`data:image/svg+xml;charset=utf-8,${encodeURIComponent(svg)}`,
rect.width,
rect.height,
);
} else {
const imageSource = element instanceof HTMLImageElement
? element.currentSrc || element.src
: getComputedStyle(element).backgroundImage.match(/^url\(["']?(.*?)["']?\)$/)?.[1] || '';
source = element instanceof HTMLImageElement ? 'image' : 'background';
dataUrl = await resolvePresentationDataUrl(imageSource, rect.width, rect.height);
}
} catch {
dataUrl = undefined;
}
return {
dataUrl,
source: dataUrl ? source : 'screenshot',
rect: { x: rect.x, y: rect.y, width: rect.width, height: rect.height },
viewport: { width: innerWidth, height: innerHeight, devicePixelRatio: devicePixelRatio || 1 },
title: document.title,
url: location.href,
};
}
return undefined;
}
async function cropVisibleTab(
tab: Awaited<ReturnType<typeof getTab>>,
candidate: PageQrCandidate,
): Promise<string | undefined> {
if (!tab.active || typeof OffscreenCanvas === 'undefined') return undefined;
const screenshot = await browser.tabs.captureVisibleTab(tab.windowId, { format: 'png' });
const bitmap = await createImageBitmap(await (await fetch(screenshot)).blob());
const scaleX = bitmap.width / candidate.viewport.width;
const scaleY = bitmap.height / candidate.viewport.height;
const padding = 12;
const x = Math.max(0, Math.floor((candidate.rect.x - padding) * scaleX));
const y = Math.max(0, Math.floor((candidate.rect.y - padding) * scaleY));
const width = Math.min(bitmap.width - x, Math.ceil((candidate.rect.width + padding * 2) * scaleX));
const height = Math.min(bitmap.height - y, Math.ceil((candidate.rect.height + padding * 2) * scaleY));
if (width < 1 || height < 1) return undefined;
const canvas = new OffscreenCanvas(width, height);
const context = canvas.getContext('2d');
if (!context) return undefined;
context.fillStyle = '#fff';
context.fillRect(0, 0, width, height);
context.drawImage(bitmap, x, y, width, height, 0, 0, width, height);
bitmap.close();
const bytes = new Uint8Array(await (await canvas.convertToBlob({ type: 'image/png' })).arrayBuffer());
if (bytes.byteLength > MAX_PRESENTATION_BYTES) return undefined;
let binary = '';
for (let offset = 0; offset < bytes.length; offset += 0x8000) {
binary += String.fromCharCode(...bytes.subarray(offset, offset + 0x8000));
}
return `data:image/png;base64,${btoa(binary)}`;
}
export async function getHandoffPresentation(handoffId: string, grant: BridgeGrant): Promise<HandoffPresentation> {
const handoff = (await getState()).handoff;
if (!handoff || handoff.id !== handoffId || handoff.taskId !== grant.taskId) {
throw new ExtensionError('handoff_not_found', '人工接管请求不存在');
}
const base = {
handoffId,
state: handoff.state,
title: handoff.target.title || '',
url: handoff.target.grantedUrl || '',
capturedAt: Date.now(),
};
if (handoff.state !== 'waiting_for_user' || handoff.reason !== 'qr_code') return base;
const target = await resolvePresentationTarget(handoff.target);
if (!target) return { ...base, state: 'page_changed' };
const tab = await getTab(target.tabId);
const results = await browser.scripting.executeScript({
target: scriptingTarget(target),
world: 'MAIN',
func: findQrCandidateInPage,
}) as Array<Browser.scripting.InjectionResult<PageQrCandidate | undefined>>;
if (results.length !== 1 || !results[0]?.result) return { ...base, state: 'not_found' };
const candidate = results[0].result;
const directDataUrl = isSafeHandoffPresentationDataUrl(candidate.dataUrl) ? candidate.dataUrl : undefined;
const dataUrl = directDataUrl
? directDataUrl
: target.frameId === 0
? await cropVisibleTab(tab, candidate).catch(() => undefined)
: undefined;
if (!isSafeHandoffPresentationDataUrl(dataUrl)) {
return { ...base, state: 'not_found', title: candidate.title || base.title, url: candidate.url || base.url };
}
return {
...base,
state: 'waiting_for_user',
title: candidate.title || base.title,
url: candidate.url || base.url,
source: directDataUrl ? candidate.source : 'screenshot',
dataUrl,
};
}
export async function focusHandoff(handoffId: string, grant: BridgeGrant): Promise<{ focused: true; tabId: number }> {
const handoff = (await getState()).handoff;
if (
!handoff
|| handoff.id !== handoffId
|| handoff.taskId !== grant.taskId
|| handoff.state !== 'waiting_for_user'
) {
throw new ExtensionError('handoff_not_waiting', '人工接管请求不存在或已经结束');
}
await activateTab(handoff.target.tabId);
return { focused: true, tabId: handoff.target.tabId };
}
export async function resolveHandoff(
handoffId: string,
outcome: Extract<HandoffState, 'completed' | 'cancelled'>,
grant?: BridgeGrant,
): Promise<{ state: Awaited<ReturnType<typeof getState>>; handoff: HumanHandoff }> {
const state = await updateState((current) => {
if (
!current.handoff
|| current.handoff.id !== handoffId
|| current.handoff.state !== 'waiting_for_user'
|| (grant && current.handoff.taskId !== grant.taskId)
) {
throw new ExtensionError('handoff_not_waiting', '人工接管请求不存在或已经结束');
}
return {
...current,
handoff: { ...current.handoff, state: outcome, resolvedAt: Date.now() },
};
});
const handoff = state.handoff!;
await setAgentRuntimeState(
outcome === 'completed' ? 'running' : 'paused',
grant || await browserInstanceAccess('browser.tabs.read'),
);
await browser.action.setBadgeText({
text: state.bridge.managedInstance?.badge || '',
tabId: handoff.target.tabId,
}).catch(() => undefined);
return { state, handoff };
}
@@ -185,6 +185,37 @@ describe('network capture lifecycle, budget and persistence', () => {
expect((await networkCaptureStatus({ tabId: 43, frameId: 0, documentId: 'document-cross-origin' })).active).toBe(false);
});
it('continues a paired-browser capture across a same-origin login navigation', async () => {
setTarget(45, 'http://localhost:8080/logic/user/login', 'document-login');
const before = await start(
45,
{ captureHeaders: true, captureBody: true },
{
kind: 'grant',
grantId: 'paired-browser-instance',
expiresAt: Number.MAX_SAFE_INTEGER,
followSameOriginNavigation: true,
},
);
await committed(45, 'http://localhost:8080/logic/user/profile', 'document-profile');
const after = await networkCaptureStatus({ tabId: 45, frameId: 0, documentId: 'document-profile' });
expect(after).toMatchObject({ active: true, startedAt: before.startedAt });
await committed(45, 'http://other.example/landing', 'document-other');
expect((await networkCaptureStatus({ tabId: 45, frameId: 0, documentId: 'document-other' })).active).toBe(false);
});
it('does not silently extend a regular scoped grant across navigation', async () => {
setTarget(46, 'https://scoped.example.test/start', 'document-start');
await start(46, {}, { kind: 'grant', grantId: 'scoped-grant', expiresAt: NOW + 60_000 });
await committed(46, 'https://scoped.example.test/next', 'document-next');
expect((await networkCaptureStatus({ tabId: 46, frameId: 0, documentId: 'document-next' })).active).toBe(false);
});
it('does not retain a cross-origin navigation request before the commit boundary is processed', async () => {
setTarget(44, 'https://source.example.test/start', 'document-source');
await start(44);
+13 -6
View File
@@ -32,7 +32,9 @@ const CAPTURED_RESOURCE_TYPES = [
] as const;
type CapturePersistence = 'pending' | 'persisted' | 'memory-only' | 'degraded';
type CaptureOwner = { kind: 'local' } | { kind: 'grant'; grantId: string; expiresAt: number };
type CaptureOwner = { kind: 'local' } | {
kind: 'grant'; grantId: string; expiresAt: number; followSameOriginNavigation?: boolean;
};
interface CaptureSession {
target: BrowserTarget;
@@ -270,7 +272,7 @@ function addRestoredSession(value: PersistedCaptureSession): boolean {
options: normalizedOptions(value.options),
records,
owner: value.owner?.kind === 'grant' && typeof value.owner.grantId === 'string' && typeof value.owner.expiresAt === 'number'
? value.owner
? { ...value.owner, followSameOriginNavigation: value.owner.followSameOriginNavigation === true }
: { kind: 'local' },
retainedBytes,
recordBytes,
@@ -283,7 +285,11 @@ function addRestoredSession(value: PersistedCaptureSession): boolean {
totalRecordCount += records.length;
totalRetainedBytes += retainedBytes;
const ownerWasValid = value.owner?.kind === 'local'
|| (value.owner?.kind === 'grant' && typeof value.owner.grantId === 'string' && typeof value.owner.expiresAt === 'number');
|| (value.owner?.kind === 'grant'
&& typeof value.owner.grantId === 'string'
&& typeof value.owner.expiresAt === 'number'
&& (value.owner.followSameOriginNavigation === undefined
|| typeof value.owner.followSameOriginNavigation === 'boolean'));
return records.length === value.records.length && ownerWasValid && !existing;
}
@@ -605,7 +611,7 @@ browser.webRequest.onErrorOccurred.addListener((details) => {
dispatchCaptureEvent(errorRecord, details);
}, { urls: ['<all_urls>'], types: [...CAPTURED_RESOURCE_TYPES] });
async function rebindLocalCaptureAfterNavigation(details: {
async function rebindCaptureAfterNavigation(details: {
tabId: number;
frameId: number;
documentId?: string;
@@ -613,7 +619,8 @@ async function rebindLocalCaptureAfterNavigation(details: {
}): Promise<void> {
await restorePromise;
const session = captureSessions.get(details.tabId);
if (!session || session.owner.kind !== 'local' || session.target.frameId !== details.frameId) return;
if (!session || session.target.frameId !== details.frameId
|| (session.owner.kind === 'grant' && !session.owner.followSameOriginNavigation)) return;
if (details.documentId && session.target.documentId === details.documentId) return;
let isolationBoundary: string | undefined;
try {
@@ -639,7 +646,7 @@ async function rebindLocalCaptureAfterNavigation(details: {
}
browser.webNavigation.onCommitted.addListener((details) => {
void rebindLocalCaptureAfterNavigation(details).catch(() => undefined);
void rebindCaptureAfterNavigation(details).catch(() => undefined);
});
browser.tabs.onRemoved.addListener((tabId) => {
if (!deleteSession(tabId)) return;

Some files were not shown because too many files have changed in this diff Show More