126 lines
4.5 KiB
JavaScript
126 lines
4.5 KiB
JavaScript
"use strict";
|
|
var __importDefault = (this && this.__importDefault) || function (mod) {
|
|
return (mod && mod.__esModule) ? mod : { "default": mod };
|
|
};
|
|
Object.defineProperty(exports, "__esModule", { value: true });
|
|
exports.FLOOD_SYMBOL = void 0;
|
|
const events_1 = __importDefault(require("events"));
|
|
const crypto_1 = __importDefault(require("crypto"));
|
|
// @ts-ignore
|
|
const lru_1 = __importDefault(require("lru"));
|
|
const debug_1 = __importDefault(require("debug"));
|
|
// @ts-ignore
|
|
const protomux_1 = __importDefault(require("protomux"));
|
|
const messages_js_1 = require("./messages.js");
|
|
// @ts-ignore
|
|
const compact_encoding_1 = __importDefault(require("compact-encoding"));
|
|
const b4a_1 = __importDefault(require("b4a"));
|
|
const debug = (0, debug_1.default)("dht-flood");
|
|
const LRU_SIZE = 255;
|
|
const TTL = 255;
|
|
const PROTOCOL = "lumeweb.flood";
|
|
exports.FLOOD_SYMBOL = Symbol.for(PROTOCOL);
|
|
class DHTFlood extends events_1.default {
|
|
id;
|
|
ttl;
|
|
messageNumber;
|
|
lru;
|
|
swarm;
|
|
constructor({ lruSize = LRU_SIZE, ttl = TTL, messageNumber = 0, id = crypto_1.default.randomBytes(32), swarm = null, } = {}) {
|
|
super();
|
|
this.id = id;
|
|
this.ttl = ttl;
|
|
this.messageNumber = messageNumber;
|
|
this.lru = new lru_1.default(lruSize);
|
|
if (!swarm) {
|
|
throw new Error("swarm is required");
|
|
}
|
|
this.swarm = swarm;
|
|
this.swarm.on("connection", (peer) => {
|
|
const mux = protomux_1.default.from(peer);
|
|
mux.pair({ protocol: PROTOCOL }, () => this.setupPeer(peer));
|
|
});
|
|
}
|
|
handleMessage({ originId, messageNumber, ttl, data }, messenger) {
|
|
const originIdBuf = b4a_1.default.from(originId);
|
|
// Ignore messages from ourselves
|
|
if (originIdBuf.equals(this.id))
|
|
return debug("Got message from self", originId, messageNumber);
|
|
// Ignore messages we've already seen
|
|
const key = originIdBuf.toString("hex") + messageNumber;
|
|
if (this.lru.get(key))
|
|
return debug("Got message that was already seen", originId, messageNumber);
|
|
this.lru.set(key, true);
|
|
this.emit("message", data, originId, messageNumber);
|
|
if (ttl <= 0) {
|
|
return debug("Got message at end of TTL", originId, messageNumber, ttl);
|
|
}
|
|
messenger.send({
|
|
originId,
|
|
messageNumber,
|
|
data,
|
|
ttl: ttl - 1,
|
|
});
|
|
}
|
|
setupPeer(peer) {
|
|
const mux = protomux_1.default.from(peer);
|
|
let chan;
|
|
const self = this;
|
|
if (!mux.opened({ protocol: PROTOCOL })) {
|
|
chan = mux.createChannel({
|
|
protocol: PROTOCOL,
|
|
async onopen() {
|
|
self.emit("peer-open", peer);
|
|
},
|
|
async onclose() {
|
|
self.emit("peer-remove", peer);
|
|
},
|
|
});
|
|
peer[exports.FLOOD_SYMBOL] = chan;
|
|
}
|
|
chan = peer[exports.FLOOD_SYMBOL];
|
|
if (!chan) {
|
|
throw new Error("could not find channel");
|
|
}
|
|
if (!chan.messages.length) {
|
|
chan.addMessage({
|
|
encoding: {
|
|
preencode: (state, m) => compact_encoding_1.default.raw.preencode(state, messages_js_1.Packet.toBinary(messages_js_1.Packet.create(m))),
|
|
encode: (state, m) => compact_encoding_1.default.raw.encode(state, messages_js_1.Packet.toBinary(messages_js_1.Packet.create(m))),
|
|
decode: (state) => messages_js_1.Packet.fromBinary(compact_encoding_1.default.raw.decode(state)),
|
|
},
|
|
onmessage: (msg) => this.handleMessage(msg, chan.messages[0]),
|
|
});
|
|
}
|
|
if (!chan.opened) {
|
|
chan.open();
|
|
}
|
|
return chan.messages[0];
|
|
}
|
|
broadcast(data, ttl = this.ttl) {
|
|
this.messageNumber++;
|
|
const { id, messageNumber } = this;
|
|
for (const peer of this.swarm.connections.values()) {
|
|
const message = this.setupPeer(peer);
|
|
message.send({
|
|
originId: id,
|
|
messageNumber,
|
|
ttl,
|
|
data: b4a_1.default.from(data),
|
|
});
|
|
}
|
|
}
|
|
send(peer, data, ttl = this.ttl) {
|
|
this.messageNumber++;
|
|
const { id, messageNumber } = this;
|
|
const message = this.setupPeer(peer);
|
|
message.send({
|
|
originId: id,
|
|
messageNumber,
|
|
ttl,
|
|
data: b4a_1.default.from(data),
|
|
});
|
|
}
|
|
}
|
|
exports.default = DHTFlood;
|