diff --git a/handlers.ts b/handlers.ts index 14efa7e..197a2f7 100644 --- a/handlers.ts +++ b/handlers.ts @@ -148,21 +148,21 @@ const deal_with_data = (session: Session, dv: DataView) => { } else { const data = dv.add(8).readByteArray(pkt_len - 4); if (is_new_image) { - if (session.curImage != null && !session.frame_is_bad) { + if (session.curImage.length > 0 && !session.frame_is_bad) { session.eventEmitter.emit("frame"); } session.frame_is_bad = false; - session.curImage = Buffer.from(data.buffer); + session.curImage = [Buffer.from(data.buffer)]; session.rcvSeqId = pkt_id; } else { if (pkt_id <= session.rcvSeqId) { // retransmit return; } - if (session.frame_is_bad) { - return; - } + + let b = Buffer.from(data.buffer); + if (pkt_id > session.rcvSeqId + 1) { // missed some packets -- filling with zeroes still produces a broken // image @@ -172,7 +172,7 @@ const deal_with_data = (session: Session, dv: DataView) => { } session.rcvSeqId = pkt_id; if (session.curImage != null) { - session.curImage = Buffer.concat([session.curImage, Buffer.from(data.buffer)]); + session.curImage.push(b); } } } diff --git a/http_server.ts b/http_server.ts index 8a3da53..b2980e2 100644 --- a/http_server.ts +++ b/http_server.ts @@ -63,9 +63,11 @@ devEv.on("discover", (rinfo: RemoteInfo, dev: DevSerial) => { const header = Buffer.from(`--${BOUNDARY}\r\nContent-Type: image/jpeg\r\n\r\n`); s.eventEmitter.on("frame", () => { + const assembled = Buffer.concat(s.curImage); + responses[dev.devId].forEach((res) => { res.write(header); - res.write(s.curImage); + res.write(assembled); }); }); diff --git a/session.ts b/session.ts index b323144..34d301f 100644 --- a/session.ts +++ b/session.ts @@ -17,9 +17,10 @@ export type Session = { connected: boolean; devName: string; timers: ReturnType[]; - curImage: Buffer | null; + curImage: Buffer[]; rcvSeqId: number; frame_is_bad: boolean; + options: opt; }; export type PacketHandler = (session: Session, dv: DataView, rinfo: RemoteInfo) => void; @@ -29,16 +30,15 @@ type msgCb = ( handlers: Record, msg: Buffer, rinfo: RemoteInfo, - options: opt, ) => void; -const handleIncoming: msgCb = (session, handlers, msg, rinfo, options) => { +const handleIncoming: msgCb = (session, handlers, msg, rinfo) => { const ab = new Uint8Array(msg).buffer; const dv = new DataView(ab); const cmd = CommandsByValue[dv.readU16()]; - if (options.debug) { + if (session.options.debug) { console.log(`<< ${cmd}`); - console.log(hexdump(msg.buffer, { ansi: options.ansi, ansiColor: 1 })); + console.log(hexdump(msg.buffer, { ansi: session.options.ansi, ansiColor: 1 })); } handlers[cmd](session, dv, rinfo); session.lastReceivedPacket = Date.now(); @@ -58,7 +58,7 @@ export const makeSession = ( sock.close(); }); - sock.on("message", (msg, rinfo) => handleIncoming(session, handlers, msg, rinfo, options)); + sock.on("message", (msg, rinfo) => handleIncoming(session, handlers, msg, rinfo)); sock.on("listening", () => { const buf = makePunchPkt(dev); @@ -98,9 +98,10 @@ export const makeSession = ( sock.send(new Uint8Array(msg.buffer), SEND_PORT, session.dst_ip); }, dst_ip: ra.address, - curImage: null, + curImage: [], rcvSeqId: 0, frame_is_bad: false, + options: options, }; session.eventEmitter.on("disconnect", () => {