// ==UserScript==
// @name APM
// @version 1.0.2
// @author ABC & Ray Adams
// @namespace github.com/ABCxFF
// @description Standard for hooking into the client -> server connection in arras.io
// @match *://arras.io/
// @match *://arras.netlify.app/
// @grant none
// @run-at document-start
// @license GPL-3.0
// ==/UserScript==
/*
* Copyright (C) 2021 ABC & Ray Adams
* Licensed under GNU General Public License v3.0
*/
const arras = (() => {
const u32 = new Uint32Array(1);
const u16 = new Uint16Array(1);
const c16 = new Uint8Array(u16.buffer);
const c32 = new Uint8Array(u32.buffer);
const f32 = new Float32Array(u32.buffer);
Array.prototype.remove = function (index) {
if (index === this.length - 1) return this.pop();
this[index] = this.pop();
};
function encode(message) {
let headers = []
let headerCodes = []
let contentSize = 0
let lastTypeCode = 0b1111
let repeatTypeCount = 0
for (let block of message) {
let typeCode = 0
if (block === 0 || block === false) {
typeCode = 0b0000
} else if (block === 1 || block === true) {
typeCode = 0b0001
} else if (typeof block === 'number') {
if (!Number.isInteger(block) || block < -0x100000000 || block >= 0x100000000) {
typeCode = 0b1000
contentSize += 4
} else if (block >= 0) {
if (block < 0x100) {
typeCode = 0b0010
contentSize++
} else if (block < 0x10000) {
typeCode = 0b0100
contentSize += 2
} else if (block < 0x100000000) {
typeCode = 0b0110
contentSize += 4
}
} else {
if (block >= -0x100) {
typeCode = 0b0011
contentSize++
} else if (block >= -0x10000) {
typeCode = 0b0101
contentSize += 2
} else if (block >= -0x100000000) {
typeCode = 0b0111
contentSize += 4
}
}
} else if (typeof block === 'string') {
let hasUnicode = false
for (let i = 0; i < block.length; i++) {
if (block.charAt(i) > '\xff') {
hasUnicode = true
} else if (block.charAt(i) === '\x00') {
console.error('Null containing string', block)
throw new Error('Null containing string')
}
}
if (!hasUnicode && block.length <= 1) {
typeCode = 0b1001
contentSize++
} else if (hasUnicode) {
typeCode = 0b1011
contentSize += block.length * 2 + 2
} else {
typeCode = 0b1010
contentSize += block.length + 1
}
} else {
console.error('Unencodable data type', block)
throw new Error('Unencodable data type')
}
headers.push(typeCode)
if (typeCode === lastTypeCode) {
repeatTypeCount++
} else {
headerCodes.push(lastTypeCode)
if (repeatTypeCount >= 1) {
while (repeatTypeCount > 19) {
headerCodes.push(0b1110)
headerCodes.push(15)
repeatTypeCount -= 19
}
if (repeatTypeCount === 1)
headerCodes.push(lastTypeCode)
else if (repeatTypeCount === 2)
headerCodes.push(0b1100)
else if (repeatTypeCount === 3)
headerCodes.push(0b1101)
else if (repeatTypeCount < 20) {
headerCodes.push(0b1110)
headerCodes.push(repeatTypeCount - 4)
}
}
repeatTypeCount = 0
lastTypeCode = typeCode
}
}
headerCodes.push(lastTypeCode)
if (repeatTypeCount >= 1) {
while (repeatTypeCount > 19) {
headerCodes.push(0b1110)
headerCodes.push(15)
repeatTypeCount -= 19
}
if (repeatTypeCount === 1)
headerCodes.push(lastTypeCode)
else if (repeatTypeCount === 2)
headerCodes.push(0b1100)
else if (repeatTypeCount === 3)
headerCodes.push(0b1101)
else if (repeatTypeCount < 20) {
headerCodes.push(0b1110)
headerCodes.push(repeatTypeCount - 4)
}
}
headerCodes.push(0b1111)
if (headerCodes.length % 2 === 1)
headerCodes.push(0b1111)
let output = new Uint8Array((headerCodes.length >> 1) + contentSize)
for (let i = 0; i < headerCodes.length; i += 2) {
let upper = headerCodes[i]
let lower = headerCodes[i + 1]
output[i >> 1] = (upper << 4) | lower
}
let index = headerCodes.length >> 1
for (let i = 0; i < headers.length; i++) {
let block = message[i]
switch (headers[i]) {
case 0b0000:
case 0b0001:
break
case 0b0010:
case 0b0011:
output[index++] = block
break
case 0b0100:
case 0b0101:
u16[0] = block
output.set(c16, index)
index += 2
break
case 0b0110:
case 0b0111:
u32[0] = block
output.set(c32, index)
index += 4
break
case 0b1000:
f32[0] = block
output.set(c32, index)
index += 4
break
case 0b1001:
{
let byte = block.length === 0 ? 0 : block.charCodeAt(0)
output[index++] = byte
}
break
case 0b1010:
for (let i = 0; i < block.length; i++) {
output[index++] = block.charCodeAt(i)
}
output[index++] = 0
break
case 0b1011:
for (let i = 0; i < block.length; i++) {
let charCode = block.charCodeAt(i)
output[index++] = charCode & 0xff
output[index++] = charCode >> 8
}
output[index++] = 0
output[index++] = 0
break
}
}
return output
};
function decode(packet) {
let data = new Uint8Array(packet)
if (data[0] >> 4 !== 0b1111)
return null
let headers = []
let lastTypeCode = 0b1111
let index = 0
let consumedHalf = true
while (true) {
if (index >= data.length)
return null
let typeCode = data[index]
if (consumedHalf) {
typeCode &= 0b1111
index++
} else {
typeCode >>= 4
}
consumedHalf = !consumedHalf
if ((typeCode & 0b1100) === 0b1100) {
if (typeCode === 0b1111) {
if (consumedHalf)
index++
break
}
let repeat = typeCode - 10 // 0b1100 - 2
if (typeCode === 0b1110) {
if (index >= data.length)
return null
let repeatCode = data[index]
if (consumedHalf) {
repeatCode &= 0b1111
index++
} else {
repeatCode >>= 4
}
consumedHalf = !consumedHalf
repeat += repeatCode
}
for (let i = 0; i < repeat; i++)
headers.push(lastTypeCode)
} else {
headers.push(typeCode)
lastTypeCode = typeCode
}
}
let output = []
for (let header of headers) {
switch (header) {
case 0b0000:
output.push(0)
break
case 0b0001:
output.push(1)
break
case 0b0010:
output.push(data[index++])
break
case 0b0011:
output.push(data[index++] - 0x100)
break
case 0b0100:
c16[0] = data[index++]
c16[1] = data[index++]
output.push(u16[0])
break
case 0b0101:
c16[0] = data[index++]
c16[1] = data[index++]
output.push(u16[0] - 0x10000)
break
case 0b0110:
c32[0] = data[index++]
c32[1] = data[index++]
c32[2] = data[index++]
c32[3] = data[index++]
output.push(u32[0])
break
case 0b0111:
c32[0] = data[index++]
c32[1] = data[index++]
c32[2] = data[index++]
c32[3] = data[index++]
output.push(u32[0] - 0x100000000)
break
case 0b1000:
c32[0] = data[index++]
c32[1] = data[index++]
c32[2] = data[index++]
c32[3] = data[index++]
output.push(f32[0])
break
case 0b1001:
{
let byte = data[index++]
output.push(byte === 0 ? '' : String.fromCharCode(byte))
}
break
case 0b1010:
{
let string = ''
let byte = 0
while (byte = data[index++]) {
string += String.fromCharCode(byte)
}
output.push(string)
}
break
case 0b1011:
{
let string = ''
let byte = 0
while (byte = data[index++] | (data[index++] << 8)) {
string += String.fromCharCode(byte)
}
output.push(string)
}
break
}
}
return output
};
function rotator(packet) {
return {
i: 0,
arr: packet,
get(index) {
return packet[index];
},
set(index, value) {
return (packet[index] = value);
},
nex() {
if (this.i === this.arr.length) {
console.error(new Error('End reached'), this.arr)
return -1;
}
return packet[this.i++];
}
}
};
class BroadcastParser {
constructor() {
this.leaderboard = [];
this.teamMinimap = [];
this.globalMinimap = [];
}
parse(packet) {
const rot = rotator(packet);
if (rot.nex() !== 'b') throw new TypeError('Invalid packet header; expected packet `b`');
this._array(rot, () => {
const del = rot.nex();
this.globalMinimap.remove(this.globalMinimap.findIndex(({ id }) => id === del));
});
this._array(rot, () => {
const dot = {
id: rot.nex(),
type: rot.nex(),
x: rot.nex(),
y: rot.nex(),
color: rot.nex(),
size: rot.nex()
};
let index = this.globalMinimap.findIndex(({ id }) => id === dot.id);
if (index === -1) index = this.globalMinimap.length;
this.globalMinimap[index] = dot;
});
this._array(rot, () => {
const del = rot.nex();
this.teamMinimap.remove(this.teamMinimap.findIndex(({ id }) => id === del));
});
this._array(rot, () => {
const dot = {
id: rot.nex(),
x: rot.nex(),
y: rot.nex(),
size: rot.nex()
};
let index = this.teamMinimap.findIndex(({ id }) => id === dot.id);
if (index === -1) index = this.teamMinimap.length;
this.teamMinimap[index] = dot;
});
this._array(rot, () => {
const del = rot.nex();
this.leaderboard.remove(this.leaderboard.findIndex(({ id }) => id === del));
});
this._array(rot, () => {
const champ = {
id: rot.nex(),
score: rot.nex(),
index: rot.nex(),
name: rot.nex(),
color: rot.nex(),
barColor: rot.nex()
};
let index = this.leaderboard.findIndex(({ id }) => id === champ.id);
if (index === -1) index = this.leaderboard.length;
this.leaderboard[index] = champ;
});
this.leaderboard.sort((c1, c2) => c2.score - c1.score);
return this;
}
_array(rot, read, length = rot.nex()) {
const out = Array(Math.max(0, length));
for (let i = 0; i < length; ++i) out[i] = read.call(this, i, rot);
return out;
}
};
class RecordParser {
constructor() {
this.score = null;
this.seconds = null;
this.killCount = {
players: null,
assists: null,
bosses: null
};
this.killersLength = null;
this.killers = [];
}
parse(packet) {
if (packet.shift() !== 'F') throw new TypeError('Invalid packet header; expected packet `F`');
this.score = packet.shift();
this.seconds = packet.shift();
this.killCount.players = packet.shift();
this.killCount.assists = packet.shift();
this.killCount.bosses = packet.shift();
this.killersLength = packet.shift();
this.killers = packet.slice(0, this.killersLength);
return this;
}
};
class UpdateParser {
constructor(doEntities = true) {
this.camera = { x: null, y: null, vx: null, vy: null, fov: null };
this.now = 0;
this.player = {
fps: 1,
body: {
type: null,
color: null,
id: null,
},
score: null,
points: null,
upgrades: [],
stats: [],
skills: null,
accel: null,
top: null,
party: null
}
this.entities = doEntities ? [] : false;
}
parse(packet) {
const rot = rotator(packet);
if (rot.nex() !== 'u') throw new TypeError('Invalid packet header; expected packet `u`');
this.now = rot.nex();
const version = this.now === 0 ? 2 : 1;
this.camera.x = rot.nex();
this.camera.y = rot.nex();
this.camera.fov = rot.nex();
this.camera.vx = rot.nex();
this.camera.vy = rot.nex();
const flags = rot.nex();
if (flags & 0x0001) this.player.fps = rot.nex();
if (flags & 0x0002) {
this.player.body.type = rot.nex();
this.player.body.color = rot.nex();
this.player.body.id = rot.nex();
}
if (flags & 0x0004) this.player.score = rot.nex();
if (flags & 0x0008) this.player.points = rot.nex();
if (flags & 0x0010) this.player.upgrades = Array(Math.max(0, rot.nex())).fill(-1).map(() => rot.nex());
if (flags & 0x0020) this.player.stats = Array(30).fill(0).map(() => rot.nex());
if (flags & 0x0040) {
const result = rot.nex();
this.player.skills = [
(result / 0x1000000000 & 15),
(result / 0x0100000000 & 15),
(result / 0x0010000000 & 15),
(result / 0x0001000000 & 15),
(result / 0x0000100000 & 15),
(result / 0x0000010000 & 15),
(result / 0x0000001000 & 15),
(result / 0x0000000100 & 15),
(result / 0x0000000010 & 15),
(result / 0x0000000001 & 15)
]
}
if (flags & 0x0080) this.player.accel = rot.nex();
if (flags & 0x0100) this.player.top = rot.nex();
if (flags & 0x0200) this.player.party = rot.nex();
if (flags & 0x0400) this.player.speed = rot.nex();
if (version === 2 && this.entities !== false) {
this._parseEnts(rot)
} else if (version !== 2 && this.entities !== false) {
this.entities = false;
console.error('Invalid version, expected version 2. Disabling entities');
}
return this;
}
_table(rot, read) {
const out = [];
for (let id = rot.nex(); id !== -1; id = rot.nex()) {
out[out.length] = read.call(this, id, rot)
}
return out
}
_parseEnts(rot) {
if (rot.nex() !== -1) return console.warn('uhhhh-cancelling', rot.arr);
this._table(rot, (id) => {
const index = this.entities.findIndex(ent => ent.id === id);
if (index === -1) {
return console.warn('Possible desync, deletion of non existant entity ' + id, this.entities.findIndex(ent => ent.id2 === id), JSON.stringify(this.entities));
}
this.entities[index] = this.entities[this.entities.length - 1]
--this.entities.length;
});
this._table(rot, (id) => {
let index = this.entities.findIndex(ent => ent.id === id)
if (index === -1) this.entities[index = this.entities.length] = { id };
const ent = this.entities[index];
this._parseEnt(ent, rot)
});
}
_parseEnt(ent, rot) {
const flags = rot.nex();
if (!ent) console.log(this.entities.length, rot.get(rot.i - 1));
if (flags & 0x0001) {
let { x: lastX, y: lastY } = ent;
ent.x = rot.nex() * 0.0625;
ent.y = rot.nex() * 0.0625;
if (typeof lastX !== 'undefined') {
ent.vx = (ent.x - lastX);
ent.vy = (ent.y - lastY);
} else ent.vx = ent.vy = 0;
}
if (flags & 0x0002) ent.facing = rot.nex() * (360 / 256);
if (flags & 0x0004) ent.flags = rot.nex();
if (flags & 0x0008) ent.health = rot.nex() / 255;
if (flags & 0x0010) ent.shield = Math.max(0, rot.nex() / 255);
if (flags & 0x0020) ent.alpha = rot.nex() / 255;
if (flags & 0x0040) ent.size = rot.nex() * 0.0625;
if (flags & 0x0080) ent.score = rot.nex();
if (flags & 0x0100) ent.name = rot.nex();
if (flags & 0x0200) ent.id2 = rot.nex();
if (flags & 0x0400) ent.color = rot.nex();
if (flags & 0x0800) ent.layer = rot.nex();
if (flags & 0x1000) {
if (!ent.guns) ent.guns = []
this._table(rot, (index) => {
const flag = rot.nex();
if (!ent.guns[index]) ent.guns[index] = {};
if (flag & 1) ent.guns[index].time = rot.nex();
if (flag & 2) ent.guns[index].power = Math.sqrt(rot.nex()) / 20;
});
}
if (flags & 0x2000) {
if (!ent.turrets) ent.turrets = [];
ent.turrets = this._table(rot, (index) => {
let i = ent.turrets.findIndex(ent => ent.index === index)
if (i === -1) ent.turrets[i = ent.turrets.length] = { index };
const turret = ent.turrets[i];
return this._parseEnt(turret, rot);
});
}
return ent;
}
};
const coder = { encode, decode };
const hijack = () => {
if (window['%arras']) return window['%arras'];
window['%arras'] = new Promise(r => {
const _send = WebSocket.prototype.send;
window.WebSocket = class ArrasSocket extends WebSocket {
constructor(...args) {
super(...args);
this.isntArras = true;
if (Array.isArray(args[1])) {
this.isntArras = false;
this._hook();
this.onopen = () => r(this);
this.sendHooks = [];
this.msgHooks = [];
}
}
_hook() {
if (this.isntArras) throw 'sus';
let send = this.send;
this.send = function(buf) {
return send.call(this, coder.encode(this.sendHooks.reduce((data, hook) => hook(data) || data, coder.decode(buf))));
}
let adv = this.addEventListener;
this.addEventListener = function(type, cb, pro=false) {
if (pro) return adv.call(this, type, cb, pro);
if (type === 'message') {
adv.call(this, 'message', (event) => {
cb(new MessageEvent('message', {
data: coder.encode(this.msgHooks.reduce((data, hook) => hook(data) || data, coder.decode(new Uint8Array(event.data)))).buffer
}));
});
} else return adv.call(this, type, cb, pro);
}
}
hookSend(...funcs) {
this.sendHooks.push.apply(this.sendHooks, funcs)
}
hookMsg(...funcs) {
this.msgHooks.push.apply(this.msgHooks, funcs)
}
directTalk(...data) {
_send.call(this, coder.encode(data));
}
talk(...data) {
this.send(coder.encode(data));
}
}
});
return window['%arras']
};
return { encode, decode, BroadcastParser, RecordParser, UpdateParser, hijack };
})()
