@willieee802/zigbee-herdsman 0.48.2 → 0.49.0

package/src/adapter/z-stack/znp/index.ts

@@ -0,0 +1,2 @@
+export {Znp} from "./znp";
+export {ZpiObject} from "./zpiObject";

package/src/adapter/z-stack/znp/parameterType.ts

@@ -0,0 +1,22 @@
+enum ParameterType {
+    UINT8 = 0,
+    UINT16 = 1,
+    UINT32 = 2,
+    IEEEADDR = 3,
+
+    BUFFER = 4,
+    BUFFER8 = 5,
+    BUFFER16 = 6,
+    BUFFER18 = 7,
+    BUFFER32 = 8,
+    BUFFER42 = 9,
+    BUFFER100 = 10,
+
+    LIST_UINT8 = 11,
+    LIST_UINT16 = 12,
+    LIST_NETWORK = 16,
+
+    INT8 = 18,
+}
+
+export default ParameterType;

package/src/adapter/z-stack/znp/tstype.ts

@@ -0,0 +1,44 @@
+import type {ClusterId as ZdoClusterId} from "../../../zspec/zdo";
+import type {Type as CommandType} from "../unpi/constants";
+import type ParameterType from "./parameterType";
+
+export type MtType = number | number[] | string | Buffer | {[s: string]: number | string}[];
+
+export interface MtParameter {
+    name: string;
+    parameterType: ParameterType;
+}
+
+interface MtCmdBase {
+    name: string;
+    ID: number;
+    type: number;
+    request: MtParameter[];
+    response: MtParameter[];
+    zdoClusterId: ZdoClusterId;
+}
+
+interface MtCmdAreq extends Omit<MtCmdBase, "response" | "zdoClusterId"> {
+    type: CommandType.AREQ;
+}
+
+interface MtCmdSreq extends Omit<MtCmdBase, "zdoClusterId"> {
+    type: CommandType.SREQ;
+}
+
+export interface MtCmdAreqZdo extends Omit<MtCmdBase, "request" | "response"> {
+    type: CommandType.AREQ;
+}
+
+export interface MtCmdSreqZdo extends Omit<MtCmdBase, "request" | "response"> {
+    type: CommandType.SREQ;
+}
+
+export type MtCmd = MtCmdAreq | MtCmdSreq | MtCmdAreqZdo | MtCmdSreqZdo;
+// biome-ignore lint/suspicious/noExplicitAny: API
+export type ZpiObjectPayload = {[s: string]: any};
+
+export interface BuffaloZnpOptions {
+    length?: number;
+    startIndex?: number;
+}

package/src/adapter/z-stack/znp/utils.ts

@@ -0,0 +1,10 @@
+import {Type} from "../unpi/constants";
+import type {MtCmd, MtCmdAreqZdo, MtCmdSreqZdo} from "./tstype";
+
+export function isMtCmdAreqZdo(cmd: MtCmd): cmd is MtCmdAreqZdo {
+    return cmd.type === Type.AREQ && "zdoClusterId" in cmd;
+}
+
+export function isMtCmdSreqZdo(cmd: MtCmd): cmd is MtCmdSreqZdo {
+    return cmd.type === Type.SREQ && "zdoClusterId" in cmd;
+}

package/src/adapter/z-stack/znp/znp.ts

@@ -0,0 +1,345 @@
+import assert from "node:assert";
+import events from "node:events";
+import {Socket} from "node:net";
+import {Waitress, wait} from "../../../utils";
+import {AsyncMutex} from "../../../utils/async-mutex";
+import {logger} from "../../../utils/logger";
+import {ClusterId as ZdoClusterId} from "../../../zspec/zdo";
+import {SerialPort} from "../../serialPort";
+import {isTcpPath, parseTcpPath} from "../../utils";
+import * as Constants from "../constants";
+import {Frame as UnpiFrame, Parser as UnpiParser, Writer as UnpiWriter} from "../unpi";
+import {Subsystem, Type} from "../unpi/constants";
+import Definition from "./definition";
+import type {ZpiObjectPayload} from "./tstype";
+import {isMtCmdSreqZdo} from "./utils";
+import {ZpiObject} from "./zpiObject";
+import {KonnextConfig} from "../../../controller/model/konnextConfig";
+
+const {
+    COMMON: {ZnpCommandStatus},
+    Utils: {statusDescription},
+} = Constants;
+
+const timeouts = {
+    SREQ: 6000,
+    reset: 30000,
+    default: 10000,
+};
+
+const NS = "zh:zstack:znp";
+
+interface WaitressMatcher {
+    type: Type;
+    subsystem: Subsystem;
+    command: string;
+    target?: number | string;
+    transid?: number;
+    state?: number;
+}
+
+export class Znp extends events.EventEmitter {
+    private path: string;
+    private baudRate: number;
+    private rtscts: boolean;
+
+    private serialPort?: SerialPort;
+    private socketPort?: Socket;
+    private unpiWriter: UnpiWriter;
+    private unpiParser: UnpiParser;
+    private initialized: boolean;
+    private queue: AsyncMutex;
+    private waitress: Waitress<ZpiObject, WaitressMatcher>;
+    private konnextConfig: KonnextConfig;
+
+    public constructor(path: string, baudRate: number, rtscts: boolean, konnextConfig: KonnextConfig) {
+        super();
+
+        this.path = path;
+        this.baudRate = typeof baudRate === "number" ? baudRate : 115200;
+        this.rtscts = typeof rtscts === "boolean" ? rtscts : false;
+
+        this.initialized = false;
+
+        this.queue = new AsyncMutex();
+        this.waitress = new Waitress<ZpiObject, WaitressMatcher>(this.waitressValidator, this.waitressTimeoutFormatter);
+        this.unpiWriter = new UnpiWriter();
+        this.unpiParser = new UnpiParser(konnextConfig);
+        this.konnextConfig = konnextConfig;
+    }
+
+    private onUnpiParsed(frame: UnpiFrame): void {
+        try {
+            const object = ZpiObject.fromUnpiFrame(frame);
+            logger.debug(() => `<-- ${object.toString(object.subsystem !== Subsystem.ZDO)}`, NS);
+            this.waitress.resolve(object);
+            this.emit("received", object);
+        } catch (error) {
+            logger.error(`Error while parsing to ZpiObject '${error}'`, NS);
+        }
+    }
+
+    public isInitialized(): boolean {
+        return this.initialized;
+    }
+
+    private onPortError(error: Error): void {
+        logger.error(`Port error: ${error}`, NS);
+    }
+
+    private onPortClose(): void {
+        logger.info("Port closed", NS);
+        this.initialized = false;
+        this.emit("close");
+    }
+
+    public async open(): Promise<void> {
+        return isTcpPath(this.path) ? await this.openSocketPort() : await this.openSerialPort();
+    }
+
+    private async openSerialPort(): Promise<void> {
+        const options = {path: this.path, baudRate: this.baudRate, rtscts: this.rtscts, autoOpen: false};
+
+        logger.info(`Opening SerialPort with ${JSON.stringify(options)}`, NS);
+        this.serialPort = new SerialPort(options);
+
+        this.unpiWriter.pipe(this.serialPort);
+        this.serialPort.pipe(this.unpiParser);
+        this.unpiParser.on("parsed", this.onUnpiParsed.bind(this));
+
+        try {
+            await this.serialPort.asyncOpen();
+            logger.info("Serialport opened", NS);
+
+            this.serialPort.once("close", this.onPortClose.bind(this));
+            this.serialPort.once("error", this.onPortError.bind(this));
+
+            this.initialized = true;
+
+            await this.skipBootloader();
+        } catch (error) {
+            this.initialized = false;
+
+            if (this.serialPort.isOpen) {
+                this.serialPort.close();
+            }
+
+            throw error;
+        }
+    }
+
+    private async openSocketPort(): Promise<void> {
+        const info = parseTcpPath(this.path);
+        logger.info(`Opening TCP socket with ${info.host}:${info.port}`, NS);
+
+        this.socketPort = new Socket();
+
+        this.socketPort.setNoDelay(true);
+        this.socketPort.setKeepAlive(true, 15000);
+        this.unpiWriter.pipe(this.socketPort);
+        this.socketPort.pipe(this.unpiParser);
+        this.unpiParser.on("parsed", this.onUnpiParsed.bind(this));
+
+        return await new Promise((resolve, reject): void => {
+            // biome-ignore lint/style/noNonNullAssertion: ignored using `--suppress`
+            this.socketPort!.on("connect", () => {
+                logger.info("Socket connected", NS);
+            });
+            const self = this;
+            // biome-ignore lint/style/noNonNullAssertion: ignored using `--suppress`
+            this.socketPort!.on("ready", async () => {
+                logger.info("Socket ready", NS);
+                await self.skipBootloader();
+                self.initialized = true;
+                resolve();
+            });
+
+            // biome-ignore lint/style/noNonNullAssertion: ignored using `--suppress`
+            this.socketPort!.once("close", this.onPortClose.bind(this));
+
+            // biome-ignore lint/style/noNonNullAssertion: ignored using `--suppress`
+            this.socketPort!.on("error", (error) => {
+                logger.error(`Socket error ${error}`, NS);
+                reject(new Error("Error while opening socket"));
+                self.initialized = false;
+            });
+
+            // biome-ignore lint/style/noNonNullAssertion: ignored using `--suppress`
+            this.socketPort!.connect(info.port, info.host);
+        });
+    }
+
+    private async skipBootloader(): Promise<void> {
+        try {
+            await this.request(Subsystem.SYS, "ping", {capabilities: 1}, undefined, 250);
+        } catch {
+            // Skip bootloader on CC2530/CC2531
+            // Send magic byte: https://github.com/Koenkk/zigbee2mqtt/issues/1343 to bootloader
+            // and give ZNP 1 second to start.
+            try {
+                logger.info("Writing CC2530/CC2531 skip bootloader payload", NS);
+                this.unpiWriter.writeBuffer(Buffer.from([0xef]), this.konnextConfig);
+                await wait(1000);
+                await this.request(Subsystem.SYS, "ping", {capabilities: 1}, undefined, 250 /* v8 ignore next */);
+            } catch {
+                // Skip bootloader on some CC2652 devices (e.g. zzh-p)
+                logger.info("Skip bootloader for CC2652/CC1352", NS);
+                if (this.serialPort) {
+                    await this.serialPort.asyncSet({dtr: false, rts: false});
+                    await wait(150);
+                    await this.serialPort.asyncSet({dtr: false, rts: true});
+                    await wait(150);
+                    await this.serialPort.asyncSet({dtr: false, rts: false});
+                    await wait(150);
+                }
+            }
+        }
+    }
+
+    public async close(): Promise<void> {
+        logger.info("closing", NS);
+        this.queue.clear();
+
+        if (this.initialized) {
+            this.initialized = false;
+
+            if (this.serialPort) {
+                try {
+                    await this.serialPort.asyncFlushAndClose();
+                } catch (error) {
+                    this.emit("close");
+
+                    throw error;
+                }
+            } else {
+                // biome-ignore lint/style/noNonNullAssertion: ignored using `--suppress`
+                this.socketPort!.destroy();
+            }
+        }
+
+        this.emit("close");
+    }
+
+    public async requestWithReply(
+        subsystem: Subsystem,
+        command: string,
+        payload: ZpiObjectPayload,
+        waiterID?: number,
+        timeout?: number,
+        expectedStatuses: Constants.COMMON.ZnpCommandStatus[] = [ZnpCommandStatus.SUCCESS],
+    ): Promise<ZpiObject> {
+        const reply = await this.request(subsystem, command, payload, waiterID, timeout, expectedStatuses);
+        if (reply === undefined) {
+            throw new Error(`Command ${command} has no reply`);
+        }
+        return reply;
+    }
+
+    public request(
+        subsystem: Subsystem,
+        command: string,
+        payload: ZpiObjectPayload,
+        waiterID?: number,
+        timeout?: number,
+        expectedStatuses: Constants.COMMON.ZnpCommandStatus[] = [ZnpCommandStatus.SUCCESS],
+    ): Promise<ZpiObject | undefined> {
+        if (!this.initialized) {
+            throw new Error("Cannot request when znp has not been initialized yet");
+        }
+
+        const object = ZpiObject.createRequest(subsystem, command, payload);
+
+        return this.queue.run<ZpiObject | undefined>(async () => {
+            logger.debug(() => `--> ${object}`, NS);
+
+            if (object.type === Type.SREQ) {
+                const t = object.command.name === "bdbStartCommissioning" || object.command.name === "startupFromApp" ? 40000 : timeouts.SREQ;
+                const waiter = this.waitress.waitFor({type: Type.SRSP, subsystem: object.subsystem, command: object.command.name}, timeout || t);
+                this.unpiWriter.writeFrame(object.unpiFrame, this.konnextConfig);
+                const result = await waiter.start().promise;
+                if (result?.payload.status !== undefined && !expectedStatuses.includes(result.payload.status)) {
+                    if (typeof waiterID === "number") {
+                        this.waitress.remove(waiterID);
+                    }
+
+                    throw new Error(
+                        `--> '${object}' failed with status '${statusDescription(
+                            result.payload.status,
+                        )}' (expected '${expectedStatuses.map(statusDescription)}')`,
+                    );
+                }
+
+                return result;
+            }
+
+            if (object.type === Type.AREQ && object.isResetCommand()) {
+                const waiter = this.waitress.waitFor({type: Type.AREQ, subsystem: Subsystem.SYS, command: "resetInd"}, timeout || timeouts.reset);
+                this.queue.clear();
+                this.unpiWriter.writeFrame(object.unpiFrame, this.konnextConfig);
+                return await waiter.start().promise;
+            }
+
+            if (object.type === Type.AREQ) {
+                this.unpiWriter.writeFrame(object.unpiFrame, this.konnextConfig);
+                /* v8 ignore start */
+            } else {
+                throw new Error(`Unknown type '${object.type}'`);
+            }
+            /* v8 ignore stop */
+        });
+    }
+
+    public requestZdo(clusterId: ZdoClusterId, payload: Buffer, waiterID?: number): Promise<void> {
+        return this.queue.run(async () => {
+            const cmd = Definition[Subsystem.ZDO].find((c) => isMtCmdSreqZdo(c) && c.zdoClusterId === clusterId);
+            assert(cmd, `Command for ZDO cluster ID '${clusterId}' not supported.`);
+
+            const unpiFrame = new UnpiFrame(Type.SREQ, Subsystem.ZDO, cmd.ID, payload);
+            const waiter = this.waitress.waitFor({type: Type.SRSP, subsystem: Subsystem.ZDO, command: cmd.name}, timeouts.SREQ);
+
+            this.unpiWriter.writeFrame(unpiFrame, this.konnextConfig);
+
+            const result = await waiter.start().promise;
+
+            if (result?.payload.status !== undefined && result.payload.status !== ZnpCommandStatus.SUCCESS) {
+                if (waiterID !== undefined) {
+                    this.waitress.remove(waiterID);
+                }
+
+                throw new Error(
+                    `--> 'SREQ: ZDO - ${ZdoClusterId[clusterId]} - ${payload.toString("hex")}' failed with status '${statusDescription(result.payload.status)}'`,
+                );
+            }
+        });
+    }
+
+    private waitressTimeoutFormatter(matcher: WaitressMatcher, timeout: number): string {
+        return `${Type[matcher.type]} - ${Subsystem[matcher.subsystem]} - ${matcher.command} after ${timeout}ms`;
+    }
+
+    public waitFor(
+        type: Type,
+        subsystem: Subsystem,
+        command: string,
+        target: number | string | undefined,
+        transid: number | undefined,
+        state: number | undefined,
+        timeout: number = timeouts.default,
+    ): {start: () => {promise: Promise<ZpiObject>; ID: number}; ID: number} {
+        return this.waitress.waitFor({type, subsystem, command, target, transid, state}, timeout);
+    }
+
+    private waitressValidator(zpiObject: ZpiObject, matcher: WaitressMatcher): boolean {
+        return (
+            matcher.type === zpiObject.type &&
+            matcher.subsystem === zpiObject.subsystem &&
+            matcher.command === zpiObject.command.name &&
+            (matcher.target === undefined ||
+                (typeof matcher.target === "number"
+                    ? matcher.target === zpiObject.payload.srcaddr
+                    : matcher.target === zpiObject.payload.zdo?.[1]?.eui64)) &&
+            (matcher.transid === undefined || matcher.transid === zpiObject.payload.transid) &&
+            (matcher.state === undefined || matcher.state === zpiObject.payload.state)
+        );
+    }
+}

package/src/adapter/z-stack/znp/zpiObject.ts

@@ -0,0 +1,148 @@
+import assert from "node:assert";
+
+import {ClusterId as ZdoClusterId} from "../../../zspec/zdo";
+import {BuffaloZdo} from "../../../zspec/zdo/buffaloZdo";
+import {Frame as UnpiFrame} from "../unpi";
+import {MaxDataSize, Subsystem, Type} from "../unpi/constants";
+import BuffaloZnp from "./buffaloZnp";
+import Definition from "./definition";
+import ParameterType from "./parameterType";
+import type {BuffaloZnpOptions, MtCmd, MtParameter, MtType, ZpiObjectPayload} from "./tstype";
+import {isMtCmdAreqZdo, isMtCmdSreqZdo} from "./utils";
+
+const BufferAndListTypes = [
+    ParameterType.BUFFER,
+    ParameterType.BUFFER8,
+    ParameterType.BUFFER16,
+    ParameterType.BUFFER18,
+    ParameterType.BUFFER32,
+    ParameterType.BUFFER42,
+    ParameterType.BUFFER100,
+    ParameterType.LIST_UINT16,
+    ParameterType.LIST_NETWORK,
+    ParameterType.LIST_UINT8,
+];
+
+type ZpiObjectType = "Request" | "Response";
+
+export class ZpiObject<T extends ZpiObjectType = "Response"> {
+    public readonly type: Type;
+    public readonly subsystem: Subsystem;
+    public readonly command: MtCmd;
+    public readonly payload: ZpiObjectPayload;
+    public readonly unpiFrame: T extends "Request" ? UnpiFrame : undefined;
+
+    private constructor(
+        type: Type,
+        subsystem: Subsystem,
+        command: MtCmd,
+        payload: ZpiObjectPayload,
+        unpiFrame: T extends "Request" ? UnpiFrame : undefined,
+    ) {
+        this.type = type;
+        this.subsystem = subsystem;
+        this.command = command;
+        this.payload = payload;
+        this.unpiFrame = unpiFrame;
+    }
+
+    public static createRequest(subsystem: Subsystem, command: string, payload: ZpiObjectPayload): ZpiObject<"Request"> {
+        if (!Definition[subsystem]) {
+            throw new Error(`Subsystem '${subsystem}' does not exist`);
+        }
+
+        const cmd = Definition[subsystem].find((c) => c.name === command);
+
+        if (cmd === undefined || isMtCmdAreqZdo(cmd) || isMtCmdSreqZdo(cmd) || cmd.request === undefined) {
+            throw new Error(`Command request '${command}' from subsystem '${subsystem}' not found`);
+        }
+
+        // Create the UnpiFrame
+        const buffalo = new BuffaloZnp(Buffer.alloc(MaxDataSize));
+
+        for (const parameter of cmd.request) {
+            const value = payload[parameter.name];
+            buffalo.write(parameter.parameterType, value, {});
+        }
+
+        const buffer = buffalo.getWritten();
+        const unpiFrame = new UnpiFrame(cmd.type, subsystem, cmd.ID, buffer);
+
+        return new ZpiObject(cmd.type, subsystem, cmd, payload, unpiFrame);
+    }
+
+    public static fromUnpiFrame(frame: UnpiFrame): ZpiObject<"Response"> {
+        const cmd = Definition[frame.subsystem].find((c) => c.ID === frame.commandID);
+
+        if (!cmd) {
+            throw new Error(`CommandID '${frame.commandID}' from subsystem '${frame.subsystem}' not found`);
+        }
+
+        let payload: ZpiObjectPayload = {};
+
+        // hotpath AREQ & SREQ ZDO since payload is identical (no need to instantiate BuffaloZnp or parse generically)
+        if (isMtCmdAreqZdo(cmd)) {
+            if (cmd.zdoClusterId === ZdoClusterId.NETWORK_ADDRESS_RESPONSE || cmd.zdoClusterId === ZdoClusterId.IEEE_ADDRESS_RESPONSE) {
+                // ZStack swaps the `startindex` and `numassocdev` compared to ZDO swap them back before handing to ZDO
+                const startIndex = frame.data[11];
+                const assocDevCount = frame.data[12];
+                frame.data[11] = assocDevCount;
+                frame.data[12] = startIndex;
+                payload.zdo = BuffaloZdo.readResponse(false, cmd.zdoClusterId, frame.data);
+            } else {
+                payload.srcaddr = frame.data.readUInt16LE(0);
+                payload.zdo = BuffaloZdo.readResponse(false, cmd.zdoClusterId, frame.data.subarray(2));
+            }
+        } else if (isMtCmdSreqZdo(cmd)) {
+            payload.status = frame.data.readUInt8(0);
+        } else {
+            const parameters = frame.type === Type.SRSP && cmd.type !== Type.AREQ ? cmd.response : cmd.request;
+            assert(
+                parameters,
+                `CommandID '${frame.commandID}' from subsystem '${frame.subsystem}' cannot be a ` +
+                    `${frame.type === Type.SRSP ? "response" : "request"}`,
+            );
+            payload = ZpiObject.readParameters(frame.data, parameters);
+        }
+
+        // GC UnpiFrame as early as possible, no longer needed
+        return new ZpiObject(frame.type, frame.subsystem, cmd, payload, undefined);
+    }
+
+    private static readParameters(buffer: Buffer, parameters: MtParameter[]): ZpiObjectPayload {
+        const buffalo = new BuffaloZnp(buffer);
+        const result: ZpiObjectPayload = {};
+
+        for (const parameter of parameters) {
+            const options: BuffaloZnpOptions = {};
+
+            if (BufferAndListTypes.includes(parameter.parameterType)) {
+                // When reading a buffer, assume that the previous parsed parameter contains
+                // the length of the buffer
+                const lengthParameter = parameters[parameters.indexOf(parameter) - 1];
+                const length: MtType = result[lengthParameter.name];
+
+                if (typeof length === "number") {
+                    options.length = length;
+                }
+            }
+
+            result[parameter.name] = buffalo.read(parameter.parameterType, options);
+        }
+
+        return result;
+    }
+
+    public isResetCommand(): boolean {
+        return (
+            (this.command.name === "resetReq" && this.subsystem === Subsystem.SYS) ||
+            /* v8 ignore next */
+            (this.command.name === "systemReset" && this.subsystem === Subsystem.SAPI)
+        );
+    }
+
+    public toString(includePayload = true): string {
+        const baseStr = `${Type[this.type]}: ${Subsystem[this.subsystem]} - ${this.command.name}`;
+        return includePayload ? `${baseStr} - ${JSON.stringify(this.payload)}` : baseStr;
+    }
+}