fix physical iOS bridge reliability

This commit is contained in:
Sinabina
2026-07-16 17:43:58 -07:00
parent bb57306d98
commit 2c487305d6
7 changed files with 259 additions and 25 deletions
+77 -1
View File
@@ -5,7 +5,9 @@
// mutate state, restore state."
import { describe, test, expect } from 'bun:test';
import { classifyRoute } from '../src/proxy';
import { classifyRoute, proxyToDevice } from '../src/proxy';
import { createServer } from 'http';
import type { IncomingMessage } from 'http';
describe('classifyRoute', () => {
test('healthz, screenshot, elements, snapshot are observe-tier', () => {
@@ -45,3 +47,77 @@ describe('classifyRoute', () => {
expect(classifyRoute('GET', '/auth/sessions').allowed).toBe(false); // loopback-only
});
});
describe('proxyToDevice failure bounds and bundle assertions', () => {
test('a suspended/non-responsive app returns a bounded 504', async () => {
const server = createServer(() => {
// Deliberately keep the connection open without headers or a body. This
// is the observable shape of a CoreDevice route to a suspended app.
});
await new Promise<void>((resolve) => server.listen(0, '127.0.0.1', resolve));
const addr = server.address();
const port = typeof addr === 'object' && addr ? addr.port : 0;
const started = Date.now();
try {
const result = await proxyToDevice({
inbound: {
method: 'GET',
url: '/screenshot',
headers: { 'content-type': 'application/json' },
} as IncomingMessage,
body: Buffer.alloc(0),
tunnel: {
udid: 'CORE-1',
ipv6Addr: '127.0.0.1',
port,
bootTokenRotated: 'rotated-token',
},
sessionId: null,
timeoutMs: 40,
});
expect(result.status).toBe(504);
expect(JSON.parse(result.body.toString())).toEqual({ error: 'upstream_timeout' });
expect(Date.now() - started).toBeLessThan(1_000);
} finally {
server.closeAllConnections?.();
await new Promise<void>((resolve) => server.close(() => resolve()));
}
});
test('coordinate actions carry the expected active bundle assertion', async () => {
let expectedBundleHeader: string | undefined;
const server = createServer((req, res) => {
expectedBundleHeader = req.headers['x-gstack-expected-bundle-id'] as string | undefined;
req.resume();
req.on('end', () => {
res.writeHead(200, { 'content-type': 'application/json' });
res.end(JSON.stringify({ ok: true }));
});
});
await new Promise<void>((resolve) => server.listen(0, '127.0.0.1', resolve));
const addr = server.address();
const port = typeof addr === 'object' && addr ? addr.port : 0;
try {
const result = await proxyToDevice({
inbound: {
method: 'POST',
url: '/tap',
headers: { 'content-type': 'application/json' },
} as IncomingMessage,
body: Buffer.from('{"x":10,"y":20}'),
tunnel: {
udid: 'CORE-1',
bundleId: 'com.gstack.fixture',
ipv6Addr: '127.0.0.1',
port,
bootTokenRotated: 'rotated-token',
},
sessionId: 'session-1',
});
expect(result.status).toBe(200);
expect(expectedBundleHeader).toBe('com.gstack.fixture');
} finally {
await new Promise<void>((resolve) => server.close(() => resolve()));
}
});
});
+39 -7
View File
@@ -81,6 +81,37 @@ describe('bootstrapTunnel', () => {
if (!r.ok) expect(r.error).toBe('no_devices');
});
test('does not misclassify a devicectl failure as no_devices', async () => {
const spawn = makeSpawn([
{
argsMatch: /devicectl list devices/,
exitCode: 1,
stderr: 'xcrun: error: unable to find utility devicectl',
},
]);
const r = await bootstrapTunnel({ bundleId: 'com.test', spawnImpl: spawn });
expect(r.ok).toBe(false);
if (!r.ok) {
expect(r.error).toBe('device_discovery_failed');
expect(r.detail).toContain('devicectl');
}
});
test('does not turn malformed devicectl JSON into an empty successful list', async () => {
const spawn = makeSpawn([
{
argsMatch: /devicectl list devices/,
jsonOutput: { result: { unexpected: [] } },
},
]);
const r = await bootstrapTunnel({ bundleId: 'com.test', spawnImpl: spawn });
expect(r.ok).toBe(false);
if (!r.ok) {
expect(r.error).toBe('device_discovery_bad_response');
expect(r.detail).toContain('result.devices');
}
});
test('returns no_paired_device when device is connected but not paired', async () => {
const spawn = makeSpawn([
{
@@ -225,6 +256,7 @@ describe('bootstrapTunnel', () => {
expect(r.tunnel.udid).toBe('TEST-UDID');
expect(r.tunnel.ipv6Addr).toBe('fd99::beef');
expect(r.tunnel.port).toBe(9999);
expect(r.tunnel.bundleId).toBe('com.test');
expect(r.tunnel.bootTokenRotated).toMatch(/^[A-Za-z0-9_-]+$/);
expect(r.tunnel.bootTokenRotated).not.toBe('BOOT-TOKEN-XYZ-123');
expect(r.tunnel.bootTokenRotated.length).toBeGreaterThan(20);
@@ -265,39 +297,39 @@ describe('bootstrapTunnel', () => {
if (!r.ok) expect(r.error).toBe('resolve_failed');
});
test('respects explicit udid when set', async () => {
test('accepts a hardware UDID but uses the matching CoreDevice UUID for commands', async () => {
const spawn = makeSpawn([
{
argsMatch: /devicectl list devices/,
jsonOutput: {
result: { devices: [
{ identifier: 'A', connectionProperties: { tunnelState: 'connected', pairingState: 'paired' }, deviceProperties: { name: 'A' }, hardwareProperties: { productType: 'iPhone18,2' } },
{ identifier: 'B', connectionProperties: { tunnelState: 'connected', pairingState: 'paired' }, deviceProperties: { name: 'B' }, hardwareProperties: { productType: 'iPhone18,2' } },
{ identifier: 'COREDEVICE-B', connectionProperties: { tunnelState: 'connected', pairingState: 'paired' }, deviceProperties: { name: 'B' }, hardwareProperties: { productType: 'iPhone18,2', udid: 'HARDWARE-B' } },
] },
},
},
{
argsMatch: /devicectl device info processes -d B/,
argsMatch: /devicectl device info processes -d COREDEVICE-B/,
jsonOutput: { result: { runningProcesses: [{ executable: 'file:///var/containers/Bundle/Application/X/com.test.app/com.test' }] } },
},
{
argsMatch: /devicectl device info details --device B/,
argsMatch: /devicectl device info details --device COREDEVICE-B/,
jsonOutput: { result: { connectionProperties: { tunnelIPAddress: 'fd00::b' } } },
},
{
argsMatch: /devicectl device copy from --device B/,
argsMatch: /devicectl device copy from --device COREDEVICE-B/,
destOutput: 'TOKEN\n',
},
]);
const r = await bootstrapTunnel({
udid: 'B',
udid: 'HARDWARE-B',
bundleId: 'com.test',
spawnImpl: spawn,
resolveImpl: async () => ['fd00::b'],
fetchImpl: (async () => new Response('{"ok":true}', { status: 200 })) as typeof fetch,
});
expect(r.ok).toBe(true);
if (r.ok) expect(r.tunnel.udid).toBe('B');
if (r.ok) expect(r.tunnel.udid).toBe('COREDEVICE-B');
});
});