126 lines
4.9 KiB
JavaScript
126 lines
4.9 KiB
JavaScript
"use strict";
|
|
Object.defineProperty(exports, "__esModule", { value: true });
|
|
const PacketFlags_1 = require("../enums/PacketFlags");
|
|
const Zerocoder_1 = require("./Zerocoder");
|
|
const MessageClasses_1 = require("./MessageClasses");
|
|
const MessageFlags_1 = require("../enums/MessageFlags");
|
|
const MessageClass = require("./MessageClasses");
|
|
class Packet {
|
|
constructor() {
|
|
this.packetFlags = 0;
|
|
this.sequenceNumber = 0;
|
|
this.extraHeader = Buffer.allocUnsafe(0);
|
|
}
|
|
getSize() {
|
|
let idSize = 4;
|
|
if (this.message.messageFlags & MessageFlags_1.MessageFlags.FrequencyHigh) {
|
|
idSize = 1;
|
|
}
|
|
else if (this.message.messageFlags & MessageFlags_1.MessageFlags.FrequencyMedium) {
|
|
idSize = 2;
|
|
}
|
|
return 1 + 4 + 1 + this.extraHeader.length + idSize + this.message.getSize();
|
|
}
|
|
writeToBuffer(buf, pos, options) {
|
|
if (options === undefined) {
|
|
options = 0;
|
|
}
|
|
if (this.message.messageFlags & MessageFlags_1.MessageFlags.Zerocoded && !(options & 1)) {
|
|
this.packetFlags = this.packetFlags | PacketFlags_1.PacketFlags.Zerocoded;
|
|
}
|
|
buf.writeUInt8(this.packetFlags, pos++);
|
|
buf.writeUInt32BE(this.sequenceNumber, pos);
|
|
pos = pos + 4;
|
|
buf.writeUInt8(this.extraHeader.length, pos++);
|
|
if (this.extraHeader.length > 0) {
|
|
this.extraHeader.copy(buf, pos);
|
|
pos += this.extraHeader.length;
|
|
}
|
|
const bodyStart = pos;
|
|
if (this.message.messageFlags & MessageFlags_1.MessageFlags.FrequencyHigh) {
|
|
buf.writeUInt8(this.message.id, pos++);
|
|
}
|
|
else if (this.message.messageFlags & MessageFlags_1.MessageFlags.FrequencyMedium) {
|
|
buf.writeUInt16BE(this.message.id, pos);
|
|
pos += 2;
|
|
}
|
|
else {
|
|
buf.writeUInt32BE(this.message.id, pos);
|
|
pos += 4;
|
|
}
|
|
const expectedLength = this.message.getSize();
|
|
const actualLength = this.message.writeToBuffer(buf, pos);
|
|
if (actualLength !== expectedLength) {
|
|
console.error('WARNING: Bytes written does not match expected message data length');
|
|
}
|
|
pos += actualLength;
|
|
if (pos < buf.length) {
|
|
console.error('WARNING: BUFFER UNDERFLOW: Finished writing but we are not at the end of the buffer (Written: ' + pos + ' bytes, expected ' + buf.length);
|
|
}
|
|
if (this.packetFlags & PacketFlags_1.PacketFlags.Zerocoded) {
|
|
buf = Zerocoder_1.Zerocoder.Encode(buf, bodyStart, pos);
|
|
}
|
|
return buf;
|
|
}
|
|
readFromBuffer(buf, pos, ackReceived, sendAck) {
|
|
this.packetFlags = buf.readUInt8(pos++);
|
|
this.sequenceNumber = buf.readUInt32BE(pos);
|
|
if (this.packetFlags & PacketFlags_1.PacketFlags.Reliable) {
|
|
sendAck(this.sequenceNumber);
|
|
}
|
|
pos = pos + 4;
|
|
const extraBytes = buf.readUInt8(pos++);
|
|
if (extraBytes > 0) {
|
|
this.extraHeader = buf.slice(pos, pos + extraBytes);
|
|
pos += extraBytes;
|
|
}
|
|
else {
|
|
this.extraHeader = Buffer.allocUnsafe(0);
|
|
}
|
|
let appendedAcks = 0;
|
|
if (this.packetFlags & PacketFlags_1.PacketFlags.Ack) {
|
|
appendedAcks = buf.readUInt8(buf.length - 1);
|
|
}
|
|
if (this.packetFlags & PacketFlags_1.PacketFlags.Zerocoded) {
|
|
let tail = 0;
|
|
if (this.packetFlags & PacketFlags_1.PacketFlags.Ack) {
|
|
tail = 1;
|
|
if (appendedAcks > 0) {
|
|
tail += appendedAcks * 4;
|
|
}
|
|
}
|
|
buf = Zerocoder_1.Zerocoder.Decode(buf, pos, buf.length - 1, tail);
|
|
}
|
|
let messageID = buf.readUInt8(pos);
|
|
if (messageID === 0xFF) {
|
|
messageID = buf.readUInt16BE(pos);
|
|
if (messageID === 0xFFFF) {
|
|
messageID = buf.readUInt32BE(pos);
|
|
pos += 4;
|
|
}
|
|
else {
|
|
pos += 2;
|
|
}
|
|
}
|
|
else {
|
|
pos++;
|
|
}
|
|
this.message = new MessageClass[MessageClasses_1.nameFromID(messageID)]();
|
|
const readLength = this.message.readFromBuffer(buf, pos);
|
|
pos += readLength;
|
|
if (this.packetFlags & PacketFlags_1.PacketFlags.Ack) {
|
|
for (let i = 0; i < appendedAcks; i++) {
|
|
const ackID = buf.readUInt32BE(pos);
|
|
ackReceived(ackID);
|
|
pos += 4;
|
|
}
|
|
pos++;
|
|
}
|
|
if (pos < buf.length) {
|
|
console.error('WARNING: Finished reading ' + MessageClasses_1.nameFromID(messageID) + ' but we\'re not at the end of the packet (' + pos + ' < ' + buf.length + ', seq ' + this.sequenceNumber + ')');
|
|
}
|
|
return pos;
|
|
}
|
|
}
|
|
exports.Packet = Packet;
|
|
//# sourceMappingURL=Packet.js.map
|