LumeWeb
/
dht-flood
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

*Update to track symbol and sockets per instance

This commit is contained in:
Derrick Hammer 2022-12-30 00:02:47 -05:00
parent ecee330493
commit ef79936384
Signed by: pcfreak30
GPG Key ID: C997C339BE476FF2
1 changed files with 13 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
src/index.ts
View File

@ -5,7 +5,7 @@ import LRU from "lru";
import debug0 from "debug"; import debug0 from "debug";
// @ts-ignore // @ts-ignore
import Protomux from "protomux"; import Protomux from "protomux";
import { Packet, PacketType } from "./messages.js"; import { Packet } from "./messages.js";
// @ts-ignore // @ts-ignore
import c from "compact-encoding"; import c from "compact-encoding";
import b4a from "b4a"; import b4a from "b4a";
@ -16,10 +16,6 @@ const LRU_SIZE = 255;
const TTL = 255; const TTL = 255;
const PROTOCOL = "lumeweb.flood"; const PROTOCOL = "lumeweb.flood";
export const FLOOD_SYMBOL = Symbol.for(PROTOCOL);
const closedMap = new Set();
export default class DHTFlood extends EventEmitter { export default class DHTFlood extends EventEmitter {
private id: Buffer; private id: Buffer;
private ttl: number; private ttl: number;
@ -27,6 +23,8 @@ export default class DHTFlood extends EventEmitter {
private lru: LRU; private lru: LRU;
private swarm: any; private swarm: any;
private protocol: string; private protocol: string;
private symbol: Symbol;
private socketMap: Set<Function> = new Set<Function>();
constructor({ constructor({
lruSize = LRU_SIZE, lruSize = LRU_SIZE,
@ -52,10 +50,12 @@ export default class DHTFlood extends EventEmitter {
const mux = Protomux.from(peer); const mux = Protomux.from(peer);
mux.pair({ protocol: this.protocol }, () => this.setupPeer(peer)); mux.pair({ protocol: this.protocol }, () => this.setupPeer(peer));
}); });
this.symbol = Symbol.for(this.protocol);
} }
private handleMessage( private handleMessage(
{ originId, messageNumber, ttl, data }: PacketType, { originId, messageNumber, ttl, data }: Packet,
messenger: any messenger: any
) { ) {
const originIdBuf = b4a.from(originId) as Buffer; const originIdBuf = b4a.from(originId) as Buffer;
@ -101,22 +101,24 @@ export default class DHTFlood extends EventEmitter {
}, },
}); });
if (chan) { if (chan) {
peer[FLOOD_SYMBOL] = chan; // @ts-ignore
peer[this.symbol] = chan;
} }
} }
if (!closedMap.has(peer)) { if (!this.socketMap.has(peer)) {
const close = () => { const close = () => {
self.emit("peer-remove", peer); self.emit("peer-remove", peer);
peer.removeListener("close", close); peer.removeListener("close", close);
closedMap.delete(peer); this.socketMap.delete(peer);
}; };
peer.on("close", close); peer.on("close", close);
closedMap.add(peer); this.socketMap.add(peer);
} }
chan = peer[FLOOD_SYMBOL]; // @ts-ignore
chan = peer[this.symbol];
if (!chan) { if (!chan) {
throw new Error("could not find channel"); throw new Error("could not find channel");