@yume-chan/adb 0.0.18 → 0.0.20

package/src/server/client.ts

@@ -0,0 +1,476 @@
+// cspell:ignore tport
+
+import { PromiseResolver } from "@yume-chan/async";
+import type {
+    AbortSignal,
+    Consumable,
+    ReadableWritablePair,
+    WritableStreamDefaultWriter,
+} from "@yume-chan/stream-extra";
+import {
+    BufferedReadableStream,
+    DuplexStreamFactory,
+    UnwrapConsumableStream,
+    WrapWritableStream,
+} from "@yume-chan/stream-extra";
+import type {
+    AsyncExactReadable,
+    ExactReadable,
+    ValueOrPromise,
+} from "@yume-chan/struct";
+import {
+    BigIntFieldType,
+    SyncPromise,
+    decodeUtf8,
+    encodeUtf8,
+} from "@yume-chan/struct";
+
+import type { AdbIncomingSocketHandler, AdbSocket } from "../adb.js";
+import { AdbBanner } from "../banner.js";
+import type { AdbFeature } from "../features.js";
+import { NOOP, hexToNumber, numberToHex } from "../utils/index.js";
+
+import { AdbServerTransport } from "./transport.js";
+
+export interface AdbServerConnectionOptions {
+    unref?: boolean | undefined;
+    signal?: AbortSignal | undefined;
+}
+
+export interface AdbServerConnection {
+    connect(
+        options?: AdbServerConnectionOptions
+    ): ValueOrPromise<ReadableWritablePair<Uint8Array, Uint8Array>>;
+
+    addReverseTunnel(
+        handler: AdbIncomingSocketHandler,
+        address?: string
+    ): ValueOrPromise<string>;
+
+    removeReverseTunnel(address: string): ValueOrPromise<void>;
+
+    clearReverseTunnels(): ValueOrPromise<void>;
+}
+
+export interface AdbServerSocket extends AdbSocket {
+    transportId: bigint;
+}
+
+export type AdbServerDeviceSelector =
+    | {
+          serial: string;
+      }
+    | { transportId: bigint }
+    | { usb: true }
+    | { tcp: true }
+    | undefined;
+
+export interface AdbServerDevice {
+    serial: string;
+    product?: string | undefined;
+    model?: string | undefined;
+    device?: string | undefined;
+    transportId: bigint;
+}
+
+export class AdbServerClient {
+    public static readonly VERSION = 41;
+
+    public readonly connection: AdbServerConnection;
+
+    public constructor(connection: AdbServerConnection) {
+        this.connection = connection;
+    }
+
+    public static readString(stream: ExactReadable): string;
+    public static readString(stream: AsyncExactReadable): PromiseLike<string>;
+    public static readString(
+        stream: ExactReadable | AsyncExactReadable
+    ): string | PromiseLike<string> {
+        return SyncPromise.try(() => stream.readExactly(4))
+            .then((buffer) => {
+                const length = hexToNumber(buffer);
+                return stream.readExactly(length);
+            })
+            .then((valueBuffer) => {
+                return decodeUtf8(valueBuffer);
+            })
+            .valueOrPromise();
+    }
+
+    public static async writeString(
+        writer: WritableStreamDefaultWriter<Uint8Array>,
+        value: string
+    ): Promise<void> {
+        const valueBuffer = encodeUtf8(value);
+        const buffer = new Uint8Array(4 + valueBuffer.length);
+        buffer.set(numberToHex(valueBuffer.length));
+        buffer.set(valueBuffer, 4);
+        await writer.write(buffer);
+    }
+
+    public static async readOkay(
+        stream: ExactReadable | AsyncExactReadable
+    ): Promise<void> {
+        const response = decodeUtf8(await stream.readExactly(4));
+        if (response === "OKAY") {
+            return;
+        }
+
+        if (response === "FAIL") {
+            const reason = await AdbServerClient.readString(stream);
+            throw new Error(reason);
+        }
+
+        throw new Error(`Unexpected response: ${response}`);
+    }
+
+    public async connect(
+        request: string,
+        options?: AdbServerConnectionOptions
+    ): Promise<ReadableWritablePair<Uint8Array, Uint8Array>> {
+        const connection = await this.connection.connect(options);
+
+        const writer = connection.writable.getWriter();
+        await AdbServerClient.writeString(writer, request);
+
+        const readable = new BufferedReadableStream(connection.readable);
+
+        try {
+            // `raceSignal` throws if the signal is aborted,
+            // so the `catch` block can close the connection.
+            await raceSignal(
+                () => AdbServerClient.readOkay(readable),
+                options?.signal
+            );
+
+            writer.releaseLock();
+            return {
+                readable: readable.release(),
+                writable: connection.writable,
+            };
+        } catch (e) {
+            writer.close().catch(NOOP);
+            readable.cancel().catch(NOOP);
+            throw e;
+        }
+    }
+
+    public async getVersion(): Promise<number> {
+        const connection = await this.connect("host:version");
+        const readable = new BufferedReadableStream(connection.readable);
+        try {
+            const length = hexToNumber(await readable.readExactly(4));
+            const version = hexToNumber(await readable.readExactly(length));
+            return version;
+        } finally {
+            connection.writable.close().catch(NOOP);
+            readable.cancel().catch(NOOP);
+        }
+    }
+
+    public async validateVersion() {
+        const version = await this.getVersion();
+        if (version !== AdbServerClient.VERSION) {
+            throw new Error(
+                `adb server version (${version}) doesn't match this client (${AdbServerClient.VERSION})`
+            );
+        }
+    }
+
+    public async killServer(): Promise<void> {
+        const connection = await this.connect("host:kill");
+        connection.writable.close().catch(NOOP);
+        connection.readable.cancel().catch(NOOP);
+    }
+
+    public async getServerFeatures(): Promise<AdbFeature[]> {
+        const connection = await this.connect("host:host-features");
+        const readable = new BufferedReadableStream(connection.readable);
+        try {
+            const response = await AdbServerClient.readString(readable);
+            return response.split(",") as AdbFeature[];
+        } finally {
+            connection.writable.close().catch(NOOP);
+            readable.cancel().catch(NOOP);
+        }
+    }
+
+    public async getDevices(): Promise<AdbServerDevice[]> {
+        const connection = await this.connect("host:devices-l");
+        const readable = new BufferedReadableStream(connection.readable);
+        try {
+            const devices: AdbServerDevice[] = [];
+            const response = await AdbServerClient.readString(readable);
+            for (const line of response.split("\n")) {
+                if (!line) {
+                    continue;
+                }
+
+                const parts = line.split(" ").filter(Boolean);
+                const serial = parts[0]!;
+                const status = parts[1]!;
+                if (status !== "device") {
+                    continue;
+                }
+
+                let product: string | undefined;
+                let model: string | undefined;
+                let device: string | undefined;
+                let transportId: bigint | undefined;
+                for (let i = 2; i < parts.length; i += 1) {
+                    const [key, value] = parts[i]!.split(":");
+                    switch (key) {
+                        case "product":
+                            product = value;
+                            break;
+                        case "model":
+                            model = value;
+                            break;
+                        case "device":
+                            device = value;
+                            break;
+                        case "transport_id":
+                            transportId = BigInt(value!);
+                            break;
+                    }
+                }
+                if (!transportId) {
+                    throw new Error(`No transport id for device ${serial}`);
+                }
+                devices.push({
+                    serial,
+                    product,
+                    model,
+                    device,
+                    transportId,
+                });
+            }
+            return devices;
+        } finally {
+            connection.writable.close().catch(NOOP);
+            readable.cancel().catch(NOOP);
+        }
+    }
+
+    public formatDeviceService(
+        device: AdbServerDeviceSelector,
+        command: string
+    ) {
+        if (!device) {
+            return `host:${command}`;
+        }
+        if ("transportId" in device) {
+            return `host-transport-id:${device.transportId}:${command}`;
+        }
+        if ("serial" in device) {
+            return `host-serial:${device.serial}:${command}`;
+        }
+        if ("usb" in device) {
+            return `host-usb:${command}`;
+        }
+        if ("tcp" in device) {
+            return `host-local:${command}`;
+        }
+        throw new Error("Invalid device selector");
+    }
+
+    /**
+     * Gets the features supported by the device.
+     * The transport ID of the selected device is also returned,
+     * so the caller can execute other commands against the same device.
+     * @param device The device selector
+     * @returns The transport ID of the selected device, and the features supported by the device.
+     */
+    public async getDeviceFeatures(
+        device: AdbServerDeviceSelector
+    ): Promise<{ transportId: bigint; features: AdbFeature[] }> {
+        // Usually the client sends a device command using `connectDevice`,
+        // so the command got forwarded and handled by ADB daemon.
+        // However, in fact, `connectDevice` only forwards unknown services to device,
+        // if the service is a host command, it will still be handled by ADB server.
+        // Also, if the command is about a device, but didn't specify a selector,
+        // it will be executed against the device selected previously by `connectDevice`.
+        // Using this method, we can get the transport ID and device features in one connection.
+        const socket = await this.connectDevice(device, "host:features");
+        try {
+            const readable = new BufferedReadableStream(socket.readable);
+            const featuresString = await AdbServerClient.readString(readable);
+            const features = featuresString.split(",") as AdbFeature[];
+            return { transportId: socket.transportId, features };
+        } finally {
+            await socket.close();
+        }
+    }
+
+    /**
+     * Creates a connection that will forward the service to device.
+     * @param device The device selector
+     * @param service The service to forward
+     * @returns An `AdbServerSocket` that can be used to communicate with the service
+     */
+    public async connectDevice(
+        device: AdbServerDeviceSelector,
+        service: string
+    ): Promise<AdbServerSocket> {
+        await this.validateVersion();
+
+        let switchService: string;
+        let transportId: bigint | undefined;
+        if (!device) {
+            switchService = `host:tport:any`;
+        } else if ("transportId" in device) {
+            switchService = `host:transport-id:${device.transportId}`;
+            transportId = device.transportId;
+        } else if ("serial" in device) {
+            switchService = `host:tport:serial:${device.serial}`;
+        } else if ("usb" in device) {
+            switchService = `host:tport:usb`;
+        } else if ("tcp" in device) {
+            switchService = `host:tport:local`;
+        } else {
+            throw new Error("Invalid device selector");
+        }
+
+        const connection = await this.connect(switchService);
+        const readable = new BufferedReadableStream(connection.readable);
+        const writer = connection.writable.getWriter();
+        try {
+            if (transportId === undefined) {
+                const array = await readable.readExactly(8);
+                // TODO: switch to a more performant algorithm.
+                const dataView = new DataView(
+                    array.buffer,
+                    array.byteOffset,
+                    array.byteLength
+                );
+                transportId = BigIntFieldType.Uint64.getter(dataView, 0, true);
+            }
+
+            await AdbServerClient.writeString(writer, service);
+            await AdbServerClient.readOkay(readable);
+
+            writer.releaseLock();
+
+            const duplex = new DuplexStreamFactory<
+                Uint8Array,
+                Consumable<Uint8Array>
+            >();
+            const wrapReadable = duplex.wrapReadable(readable.release());
+            const wrapWritable = duplex.createWritable(
+                new WrapWritableStream(connection.writable).bePipedThroughFrom(
+                    new UnwrapConsumableStream()
+                )
+            );
+
+            return {
+                transportId,
+                service,
+                readable: wrapReadable,
+                writable: wrapWritable,
+                close() {
+                    return duplex.close();
+                },
+            };
+        } catch (e) {
+            writer.close().catch(NOOP);
+            readable.cancel().catch(NOOP);
+            throw e;
+        }
+    }
+
+    /**
+     * Wait for a device to be connected or disconnected.
+     * @param device The device selector
+     * @param state The state to wait for
+     * @param options The options
+     * @returns A promise that resolves when the condition is met.
+     */
+    public async waitFor(
+        device: AdbServerDeviceSelector,
+        state: "device" | "disconnect",
+        options?: AdbServerConnectionOptions
+    ): Promise<void> {
+        let type: string;
+        if (!device) {
+            type = "any";
+        } else if ("transportId" in device) {
+            type = "any";
+        } else if ("serial" in device) {
+            type = "any";
+        } else if ("usb" in device) {
+            type = "usb";
+        } else if ("tcp" in device) {
+            type = "local";
+        } else {
+            throw new Error("Invalid device selector");
+        }
+
+        // `waitFor` can't use `connectDevice`, because the device
+        // might not be available yet.
+        const service = this.formatDeviceService(
+            device,
+            `wait-for-${type}-${state}`
+        );
+
+        // `connect` resolves when server writes `OKAY`,
+        // but for this command the server writes `OKAY` after the condition is met.
+        await this.connect(service, options);
+    }
+
+    public async createTransport(
+        device: AdbServerDeviceSelector
+    ): Promise<AdbServerTransport> {
+        const { transportId, features } = await this.getDeviceFeatures(device);
+
+        const devices = await this.getDevices();
+        const info = devices.find(
+            (device) => device.transportId === transportId
+        );
+
+        const banner = new AdbBanner(
+            info?.product,
+            info?.model,
+            info?.device,
+            features
+        );
+
+        return new AdbServerTransport(
+            this,
+            info?.serial ?? "",
+            banner,
+            transportId
+        );
+    }
+}
+
+async function raceSignal<T>(
+    callback: () => Promise<T>,
+    ...signals: (AbortSignal | undefined)[]
+): Promise<T> {
+    const abortPromise = new PromiseResolver<never>();
+    function abort(this: AbortSignal) {
+        abortPromise.reject(this.reason);
+    }
+
+    try {
+        for (const signal of signals) {
+            if (!signal) {
+                continue;
+            }
+            if (signal.aborted) {
+                throw signal.reason;
+            }
+            signal.addEventListener("abort", abort);
+        }
+
+        return await Promise.race([callback(), abortPromise.promise]);
+    } finally {
+        for (const signal of signals) {
+            if (!signal) {
+                continue;
+            }
+            signal.removeEventListener("abort", abort);
+        }
+    }
+}

package/src/server/index.ts

@@ -0,0 +1,2 @@
+export * from "./client.js";
+export * from "./transport.js";

package/src/server/transport.ts

@@ -0,0 +1,77 @@
+import { PromiseResolver } from "@yume-chan/async";
+import { AbortController } from "@yume-chan/stream-extra";
+import type { ValueOrPromise } from "@yume-chan/struct";
+
+import type {
+    AdbIncomingSocketHandler,
+    AdbSocket,
+    AdbTransport,
+} from "../adb.js";
+import type { AdbBanner } from "../banner.js";
+
+import type { AdbServerClient } from "./client.js";
+
+export class AdbServerTransport implements AdbTransport {
+    #client: AdbServerClient;
+
+    public readonly serial: string;
+
+    public readonly transportId: bigint;
+
+    public readonly maxPayloadSize: number = 1 * 1024 * 1024;
+
+    public readonly banner: AdbBanner;
+
+    #closed = new PromiseResolver<void>();
+    #waitAbortController = new AbortController();
+    public readonly disconnected: Promise<void>;
+
+    public constructor(
+        client: AdbServerClient,
+        serial: string,
+        banner: AdbBanner,
+        transportId: bigint
+    ) {
+        this.#client = client;
+        this.serial = serial;
+        this.banner = banner;
+        this.transportId = transportId;
+
+        this.disconnected = Promise.race([
+            this.#closed.promise,
+            client.waitFor({ transportId }, "disconnect", {
+                signal: this.#waitAbortController.signal,
+                unref: true,
+            }),
+        ]);
+    }
+
+    public async connect(service: string): Promise<AdbSocket> {
+        return await this.#client.connectDevice(
+            {
+                transportId: this.transportId,
+            },
+            service
+        );
+    }
+
+    public async addReverseTunnel(
+        handler: AdbIncomingSocketHandler,
+        address?: string
+    ): Promise<string> {
+        return await this.#client.connection.addReverseTunnel(handler, address);
+    }
+
+    public async removeReverseTunnel(address: string): Promise<void> {
+        await this.#client.connection.removeReverseTunnel(address);
+    }
+
+    public async clearReverseTunnels(): Promise<void> {
+        await this.#client.connection.clearReverseTunnels();
+    }
+
+    close(): ValueOrPromise<void> {
+        this.#closed.resolve();
+        this.#waitAbortController.abort();
+    }
+}

package/src/utils/auto-reset-event.ts

@@ -1,41 +1,40 @@
 import { PromiseResolver } from "@yume-chan/async";
-import { type Disposable } from "@yume-chan/event";
+import type { Disposable } from "@yume-chan/event";
 
 export class AutoResetEvent implements Disposable {
-    private readonly list: PromiseResolver<void>[] = [];
-
-    private blocking: boolean;
+    #set: boolean;
+    readonly #queue: PromiseResolver<void>[] = [];
 
     public constructor(initialSet = false) {
-        this.blocking = initialSet;
+        this.#set = initialSet;
     }
 
     public wait(): Promise<void> {
-        if (!this.blocking) {
-            this.blocking = true;
+        if (!this.#set) {
+            this.#set = true;
 
-            if (this.list.length === 0) {
+            if (this.#queue.length === 0) {
                 return Promise.resolve();
             }
         }
 
         const resolver = new PromiseResolver<void>();
-        this.list.push(resolver);
+        this.#queue.push(resolver);
         return resolver.promise;
     }
 
-    public notify() {
-        if (this.list.length !== 0) {
-            this.list.pop()!.resolve();
+    public notifyOne() {
+        if (this.#queue.length !== 0) {
+            this.#queue.pop()!.resolve();
         } else {
-            this.blocking = false;
+            this.#set = false;
         }
     }
 
     public dispose() {
-        for (const item of this.list) {
+        for (const item of this.#queue) {
             item.reject(new Error("The AutoResetEvent has been disposed"));
         }
-        this.list.length = 0;
+        this.#queue.length = 0;
     }
 }

package/src/utils/conditional-variable.ts

@@ -0,0 +1,45 @@
+import { PromiseResolver } from "@yume-chan/async";
+import type { Disposable } from "@yume-chan/event";
+
+interface WaitEntry {
+    condition: () => boolean;
+    resolver: PromiseResolver<void>;
+}
+
+export class ConditionalVariable implements Disposable {
+    #locked = false;
+    readonly #queue: WaitEntry[] = [];
+
+    public wait(condition: () => boolean): Promise<void> {
+        if (!this.#locked) {
+            this.#locked = true;
+            if (this.#queue.length === 0 && condition()) {
+                return Promise.resolve();
+            }
+        }
+
+        const resolver = new PromiseResolver<void>();
+        this.#queue.push({ condition, resolver });
+        return resolver.promise;
+    }
+
+    public notifyOne() {
+        const entry = this.#queue.shift();
+        if (entry) {
+            if (entry.condition()) {
+                entry.resolver.resolve();
+            }
+        } else {
+            this.#locked = false;
+        }
+    }
+
+    public dispose(): void {
+        for (const item of this.#queue) {
+            item.resolver.reject(
+                new Error("The ConditionalVariable has been disposed")
+            );
+        }
+        this.#queue.length = 0;
+    }
+}