break: remove arRPC
This commit is contained in:
Elysia
parent
a0c7752ed2
commit
54dd333237
@ -1,229 +0,0 @@
|
|||||||
'use strict';
|
|
||||||
|
|
||||||
const { EventEmitter } = require('node:events');
|
|
||||||
const ProcessServer = require('./process/index.js');
|
|
||||||
const IPCServer = require('./transports/ipc.js');
|
|
||||||
const WSServer = require('./transports/websocket.js');
|
|
||||||
const { RichPresence } = require('../../structures/RichPresence.js');
|
|
||||||
const { NitroType } = require('../Constants.js');
|
|
||||||
|
|
||||||
// eslint-disable-next-line no-useless-escape
|
|
||||||
const checkUrl = url => /^(?:http(s)?:\/\/)?[\w.-]+(?:\.[\w\.-]+)+[\w\-\._~:/?#[\]@!\$&'\(\)\*\+,;=.]+$/.test(url);
|
|
||||||
|
|
||||||
let socketId = 0;
|
|
||||||
module.exports = class RPCServer extends EventEmitter {
|
|
||||||
constructor(client, debug = false) {
|
|
||||||
super();
|
|
||||||
Object.defineProperty(this, 'client', { value: client });
|
|
||||||
return (async () => {
|
|
||||||
this.debug = debug;
|
|
||||||
this.onConnection = this.onConnection.bind(this);
|
|
||||||
this.onMessage = this.onMessage.bind(this);
|
|
||||||
this.onClose = this.onClose.bind(this);
|
|
||||||
|
|
||||||
const handlers = {
|
|
||||||
connection: this.onConnection,
|
|
||||||
message: this.onMessage,
|
|
||||||
close: this.onClose,
|
|
||||||
};
|
|
||||||
|
|
||||||
this.ipc = await new IPCServer(handlers, this.debug);
|
|
||||||
this.ws = await new WSServer(handlers, this.debug);
|
|
||||||
this.process = await new ProcessServer(handlers, this.debug);
|
|
||||||
|
|
||||||
return this;
|
|
||||||
})();
|
|
||||||
}
|
|
||||||
|
|
||||||
onConnection(socket) {
|
|
||||||
socket.send({
|
|
||||||
cmd: 'DISPATCH',
|
|
||||||
evt: 'READY',
|
|
||||||
|
|
||||||
data: {
|
|
||||||
v: 1,
|
|
||||||
// Needed otherwise some stuff errors out parsing json strictly
|
|
||||||
user: {
|
|
||||||
// Mock user data using arRPC app/bot
|
|
||||||
id: this.client?.user?.id ?? '1045800378228281345',
|
|
||||||
username: this.client?.user?.username ?? 'arRPC',
|
|
||||||
discriminator: this.client?.user?.discriminator ?? '0000',
|
|
||||||
avatar: this.client?.user?.avatar,
|
|
||||||
flags: this.client?.user?.flags?.bitfield ?? 0,
|
|
||||||
premium_type: this.client?.user?.nitroType ? NitroType[this.client?.user?.nitroType] : 0,
|
|
||||||
},
|
|
||||||
config: {
|
|
||||||
api_endpoint: '//discord.com/api',
|
|
||||||
cdn_host: 'cdn.discordapp.com',
|
|
||||||
environment: 'production',
|
|
||||||
},
|
|
||||||
},
|
|
||||||
});
|
|
||||||
|
|
||||||
socket.socketId = socketId++;
|
|
||||||
|
|
||||||
this.emit('connection', socket);
|
|
||||||
}
|
|
||||||
|
|
||||||
onClose(socket) {
|
|
||||||
this.emit('activity', {
|
|
||||||
activity: null,
|
|
||||||
pid: socket.lastPid,
|
|
||||||
socketId: socket.socketId.toString(),
|
|
||||||
});
|
|
||||||
|
|
||||||
this.emit('close', socket);
|
|
||||||
}
|
|
||||||
|
|
||||||
async onMessage(socket, { cmd, args, nonce }) {
|
|
||||||
this.emit('message', { socket, cmd, args, nonce });
|
|
||||||
|
|
||||||
switch (cmd) {
|
|
||||||
case 'SET_ACTIVITY':
|
|
||||||
if (!socket.clientInfo || !socket.clientAssets) {
|
|
||||||
// https://discord.com/api/v9/oauth2/applications/:id/rpc
|
|
||||||
socket.clientInfo = await this.client.api.oauth2.applications(socket.clientId).rpc.get();
|
|
||||||
socket.clientAssets = await this.client.api.oauth2.applications(socket.clientId).assets.get();
|
|
||||||
}
|
|
||||||
// eslint-disable-next-line no-case-declarations
|
|
||||||
const { activity, pid } = args; // Translate given parameters into what discord dispatch expects
|
|
||||||
|
|
||||||
if (!activity) {
|
|
||||||
return this.emit('activity', {
|
|
||||||
activity: null,
|
|
||||||
pid,
|
|
||||||
socketId: socket.socketId.toString(),
|
|
||||||
});
|
|
||||||
}
|
|
||||||
// eslint-disable-next-line no-case-declarations
|
|
||||||
const { buttons, timestamps, instance, assets } = activity;
|
|
||||||
|
|
||||||
socket.lastPid = pid ?? socket.lastPid;
|
|
||||||
|
|
||||||
// eslint-disable-next-line no-case-declarations
|
|
||||||
const metadata = {};
|
|
||||||
// eslint-disable-next-line no-case-declarations
|
|
||||||
const extra = {};
|
|
||||||
if (buttons) {
|
|
||||||
// Map buttons into expected metadata
|
|
||||||
metadata.button_urls = buttons.map(x => x.url);
|
|
||||||
extra.buttons = buttons.map(x => x.label);
|
|
||||||
}
|
|
||||||
|
|
||||||
if (assets?.large_image) {
|
|
||||||
if (checkUrl(assets.large_image)) {
|
|
||||||
assets.large_image = assets.large_image
|
|
||||||
.replace('https://cdn.discordapp.com/', 'mp:')
|
|
||||||
.replace('http://cdn.discordapp.com/', 'mp:')
|
|
||||||
.replace('https://media.discordapp.net/', 'mp:')
|
|
||||||
.replace('http://media.discordapp.net/', 'mp:');
|
|
||||||
if (!assets.large_image.startsWith('mp:')) {
|
|
||||||
// Fetch
|
|
||||||
const data = await RichPresence.getExternal(this.client, socket.clientId, assets.large_image);
|
|
||||||
assets.large_image = data[0].external_asset_path;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if (/^[0-9]{17,19}$/.test(assets.large_image)) {
|
|
||||||
// ID Assets
|
|
||||||
}
|
|
||||||
if (
|
|
||||||
assets.large_image.startsWith('mp:') ||
|
|
||||||
assets.large_image.startsWith('youtube:') ||
|
|
||||||
assets.large_image.startsWith('spotify:')
|
|
||||||
) {
|
|
||||||
// Image
|
|
||||||
}
|
|
||||||
if (assets.large_image.startsWith('external/')) {
|
|
||||||
assets.large_image = `mp:${assets.large_image}`;
|
|
||||||
} else {
|
|
||||||
const l = socket.clientAssets.find(o => o.name == assets.large_image);
|
|
||||||
if (l) assets.large_image = l.id;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if (assets?.small_image) {
|
|
||||||
if (checkUrl(assets.small_image)) {
|
|
||||||
assets.small_image = assets.small_image
|
|
||||||
.replace('https://cdn.discordapp.com/', 'mp:')
|
|
||||||
.replace('http://cdn.discordapp.com/', 'mp:')
|
|
||||||
.replace('https://media.discordapp.net/', 'mp:')
|
|
||||||
.replace('http://media.discordapp.net/', 'mp:');
|
|
||||||
if (!assets.small_image.startsWith('mp:')) {
|
|
||||||
// Fetch
|
|
||||||
const data = await RichPresence.getExternal(this.client, socket.clientId, assets.small_image);
|
|
||||||
assets.small_image = data[0].external_asset_path;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if (/^[0-9]{17,19}$/.test(assets.small_image)) {
|
|
||||||
// ID Assets
|
|
||||||
}
|
|
||||||
if (
|
|
||||||
assets.small_image.startsWith('mp:') ||
|
|
||||||
assets.small_image.startsWith('youtube:') ||
|
|
||||||
assets.small_image.startsWith('spotify:')
|
|
||||||
) {
|
|
||||||
// Image
|
|
||||||
}
|
|
||||||
if (assets.small_image.startsWith('external/')) {
|
|
||||||
assets.small_image = `mp:${assets.small_image}`;
|
|
||||||
} else {
|
|
||||||
const l = socket.clientAssets.find(o => o.name == assets.small_image);
|
|
||||||
if (l) assets.small_image = l.id;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if (timestamps) {
|
|
||||||
for (const x in timestamps) {
|
|
||||||
// Translate s -> ms timestamps
|
|
||||||
if (Date.now().toString().length - timestamps[x].toString().length > 2) {
|
|
||||||
timestamps[x] = Math.floor(1000 * timestamps[x]);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
this.emit('activity', {
|
|
||||||
activity: {
|
|
||||||
application_id: socket.clientId,
|
|
||||||
type: 0,
|
|
||||||
name: socket.clientInfo.name,
|
|
||||||
metadata,
|
|
||||||
assets,
|
|
||||||
flags: instance ? 1 << 0 : 0,
|
|
||||||
...activity,
|
|
||||||
...extra,
|
|
||||||
},
|
|
||||||
pid,
|
|
||||||
socketId: socket.socketId.toString(),
|
|
||||||
});
|
|
||||||
|
|
||||||
socket.send?.({
|
|
||||||
cmd,
|
|
||||||
data: null,
|
|
||||||
evt: null,
|
|
||||||
nonce,
|
|
||||||
});
|
|
||||||
|
|
||||||
break;
|
|
||||||
|
|
||||||
case 'GUILD_TEMPLATE_BROWSER':
|
|
||||||
case 'INVITE_BROWSER':
|
|
||||||
// eslint-disable-next-line no-case-declarations
|
|
||||||
const { code } = args;
|
|
||||||
socket.send({
|
|
||||||
cmd,
|
|
||||||
data: {
|
|
||||||
code,
|
|
||||||
},
|
|
||||||
nonce,
|
|
||||||
});
|
|
||||||
|
|
||||||
this.emit(cmd === 'INVITE_BROWSER' ? 'invite' : 'guild-template', code);
|
|
||||||
break;
|
|
||||||
|
|
||||||
case 'DEEP_LINK':
|
|
||||||
this.emit('link', args.params);
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
};
|
|
||||||
File diff suppressed because one or more lines are too long
@ -1,102 +0,0 @@
|
|||||||
'use strict';
|
|
||||||
|
|
||||||
const rgb = (r, g, b, msg) => `\x1b[38;2;${r};${g};${b}m${msg}\x1b[0m`;
|
|
||||||
const log = (...args) => console.log(`[${rgb(88, 101, 242, 'arRPC')} > ${rgb(237, 66, 69, 'process')}]`, ...args);
|
|
||||||
const { setInterval } = require('node:timers');
|
|
||||||
const process = require('process');
|
|
||||||
const DetectableDB = require('./detectable.json');
|
|
||||||
const Natives = require('./native/index.js');
|
|
||||||
|
|
||||||
const Native = Natives[process.platform];
|
|
||||||
|
|
||||||
const timestamps = {},
|
|
||||||
names = {},
|
|
||||||
pids = {};
|
|
||||||
module.exports = class ProcessServer {
|
|
||||||
constructor(handlers, debug = false) {
|
|
||||||
this.debug = debug;
|
|
||||||
if (!Native) return; // Log('unsupported platform:', process.platform);
|
|
||||||
|
|
||||||
this.handlers = handlers;
|
|
||||||
|
|
||||||
this.scan = this.scan.bind(this);
|
|
||||||
|
|
||||||
this.scan();
|
|
||||||
setInterval(this.scan, 5000).unref();
|
|
||||||
|
|
||||||
if (this.debug) log('started');
|
|
||||||
}
|
|
||||||
|
|
||||||
async scan() {
|
|
||||||
const processes = await Native.getProcesses();
|
|
||||||
const ids = [];
|
|
||||||
|
|
||||||
for (const [pid, _path] of processes) {
|
|
||||||
const path = _path.toLowerCase().replaceAll('\\', '/');
|
|
||||||
const toCompare = [path.split('/').pop(), path.split('/').slice(-2).join('/')];
|
|
||||||
|
|
||||||
for (const p of toCompare.slice()) {
|
|
||||||
// Add more possible tweaked paths for less false negatives
|
|
||||||
toCompare.push(p.replace('64', '')); // Remove 64bit identifiers-ish
|
|
||||||
toCompare.push(p.replace('.x64', ''));
|
|
||||||
toCompare.push(p.replace('x64', ''));
|
|
||||||
}
|
|
||||||
|
|
||||||
for (const { executables, id, name } of DetectableDB) {
|
|
||||||
if (executables?.some(x => !x.isLauncher && toCompare.some(y => x.name === y))) {
|
|
||||||
names[id] = name;
|
|
||||||
pids[id] = pid;
|
|
||||||
|
|
||||||
ids.push(id);
|
|
||||||
if (!timestamps[id]) {
|
|
||||||
// eslint-disable-next-line max-depth
|
|
||||||
if (this.debug) log('detected game!', name);
|
|
||||||
timestamps[id] = Date.now();
|
|
||||||
|
|
||||||
this.handlers.message(
|
|
||||||
{
|
|
||||||
socketId: id,
|
|
||||||
},
|
|
||||||
{
|
|
||||||
cmd: 'SET_ACTIVITY',
|
|
||||||
args: {
|
|
||||||
activity: {
|
|
||||||
application_id: id,
|
|
||||||
name,
|
|
||||||
timestamps: {
|
|
||||||
start: timestamps[id],
|
|
||||||
},
|
|
||||||
},
|
|
||||||
pid,
|
|
||||||
},
|
|
||||||
},
|
|
||||||
);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
for (const id in timestamps) {
|
|
||||||
if (!ids.includes(id)) {
|
|
||||||
if (this.debug) log('lost game!', names[id]);
|
|
||||||
delete timestamps[id];
|
|
||||||
|
|
||||||
this.handlers.message(
|
|
||||||
{
|
|
||||||
socketId: id,
|
|
||||||
},
|
|
||||||
{
|
|
||||||
cmd: 'SET_ACTIVITY',
|
|
||||||
args: {
|
|
||||||
activity: null,
|
|
||||||
pid: pids[id],
|
|
||||||
},
|
|
||||||
},
|
|
||||||
);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// If (this.debug) log(`finished scan in ${(performance.now() - startTime).toFixed(2)}ms`);
|
|
||||||
// process.stdout.write(`\r${' '.repeat(100)}\r[${rgb(88, 101, 242, 'arRPC')} > ${rgb(237, 66, 69, 'process')}] scanned (took ${(performance.now() - startTime).toFixed(2)}ms)`);
|
|
||||||
}
|
|
||||||
};
|
|
||||||
@ -1,5 +0,0 @@
|
|||||||
'use strict';
|
|
||||||
|
|
||||||
const linux = require('./linux.js');
|
|
||||||
const win32 = require('./win32.js');
|
|
||||||
module.exports = { win32, linux };
|
|
||||||
@ -1,37 +0,0 @@
|
|||||||
'use strict';
|
|
||||||
|
|
||||||
const { exec } = require('child_process');
|
|
||||||
const { readlink } = require('fs/promises');
|
|
||||||
|
|
||||||
const getProcesses = () =>
|
|
||||||
new Promise(res =>
|
|
||||||
exec(`ps a -o "%p;%c;%a"`, async (e, out) => {
|
|
||||||
res(
|
|
||||||
(
|
|
||||||
await Promise.all(
|
|
||||||
out
|
|
||||||
.toString()
|
|
||||||
.split('\n')
|
|
||||||
.slice(1, -1)
|
|
||||||
|
|
||||||
.map(async x => {
|
|
||||||
const split = x.trim().split(';');
|
|
||||||
// If (split.length === 1) return;
|
|
||||||
|
|
||||||
const pid = parseInt(split[0].trim());
|
|
||||||
/* Unused
|
|
||||||
const cmd = split[1].trim();
|
|
||||||
const argv = split.slice(2).join(';').trim();
|
|
||||||
*/
|
|
||||||
|
|
||||||
const path = await readlink(`/proc/${pid}/exe`).catch(() => {}); // Read path from /proc/{pid}/exe symlink
|
|
||||||
|
|
||||||
return [pid, path];
|
|
||||||
}),
|
|
||||||
)
|
|
||||||
).filter(x => x && x[1]),
|
|
||||||
);
|
|
||||||
}),
|
|
||||||
);
|
|
||||||
|
|
||||||
module.exports = { getProcesses };
|
|
||||||
@ -1,25 +0,0 @@
|
|||||||
'use strict';
|
|
||||||
|
|
||||||
const { exec } = require('child_process');
|
|
||||||
|
|
||||||
const getProcesses = () =>
|
|
||||||
new Promise(res =>
|
|
||||||
exec(`wmic process get ProcessID,ExecutablePath /format:csv`, (e, out) => {
|
|
||||||
res(
|
|
||||||
out
|
|
||||||
.toString()
|
|
||||||
.split('\r\n')
|
|
||||||
.slice(2)
|
|
||||||
|
|
||||||
.map(x => {
|
|
||||||
// eslint-disable-next-line newline-per-chained-call
|
|
||||||
const parsed = x.trim().split(',').slice(1).reverse();
|
|
||||||
parsed[0] = parseInt(parsed[0]) || parsed[0]; // Pid to int
|
|
||||||
return parsed;
|
|
||||||
})
|
|
||||||
.filter(x => x[1]),
|
|
||||||
);
|
|
||||||
}),
|
|
||||||
);
|
|
||||||
|
|
||||||
module.exports = { getProcesses };
|
|
||||||
@ -1,281 +0,0 @@
|
|||||||
'use strict';
|
|
||||||
|
|
||||||
const rgb = (r, g, b, msg) => `\x1b[38;2;${r};${g};${b}m${msg}\x1b[0m`;
|
|
||||||
const log = (...args) => console.log(`[${rgb(88, 101, 242, 'arRPC')} > ${rgb(254, 231, 92, 'ipc')}]`, ...args);
|
|
||||||
|
|
||||||
const { Buffer } = require('buffer');
|
|
||||||
const { unlinkSync } = require('fs');
|
|
||||||
const { createServer, createConnection } = require('net');
|
|
||||||
const { setTimeout } = require('node:timers');
|
|
||||||
const { join } = require('path');
|
|
||||||
const { platform, env } = require('process');
|
|
||||||
|
|
||||||
const SOCKET_PATH =
|
|
||||||
platform === 'win32'
|
|
||||||
? '\\\\?\\pipe\\discord-ipc'
|
|
||||||
: join(env.XDG_RUNTIME_DIR || env.TMPDIR || env.TMP || env.TEMP || '/tmp', 'discord-ipc');
|
|
||||||
|
|
||||||
// Enums for various constants
|
|
||||||
const Types = {
|
|
||||||
// Types of packets
|
|
||||||
HANDSHAKE: 0,
|
|
||||||
FRAME: 1,
|
|
||||||
CLOSE: 2,
|
|
||||||
PING: 3,
|
|
||||||
PONG: 4,
|
|
||||||
};
|
|
||||||
|
|
||||||
const CloseCodes = {
|
|
||||||
// Codes for closures
|
|
||||||
CLOSE_NORMAL: 1000,
|
|
||||||
CLOSE_UNSUPPORTED: 1003,
|
|
||||||
CLOSE_ABNORMAL: 1006,
|
|
||||||
};
|
|
||||||
|
|
||||||
const ErrorCodes = {
|
|
||||||
// Codes for errors
|
|
||||||
INVALID_CLIENTID: 4000,
|
|
||||||
INVALID_ORIGIN: 4001,
|
|
||||||
RATELIMITED: 4002,
|
|
||||||
TOKEN_REVOKED: 4003,
|
|
||||||
INVALID_VERSION: 4004,
|
|
||||||
INVALID_ENCODING: 4005,
|
|
||||||
};
|
|
||||||
|
|
||||||
let uniqueId = 0;
|
|
||||||
|
|
||||||
const encode = (type, data) => {
|
|
||||||
data = JSON.stringify(data);
|
|
||||||
const dataSize = Buffer.byteLength(data);
|
|
||||||
|
|
||||||
const buf = Buffer.alloc(dataSize + 8);
|
|
||||||
buf.writeInt32LE(type, 0); // Type
|
|
||||||
buf.writeInt32LE(dataSize, 4); // Data size
|
|
||||||
buf.write(data, 8, dataSize); // Data
|
|
||||||
|
|
||||||
return buf;
|
|
||||||
};
|
|
||||||
|
|
||||||
const read = socket => {
|
|
||||||
let resp = socket.read(8);
|
|
||||||
if (!resp) return;
|
|
||||||
|
|
||||||
resp = Buffer.from(resp);
|
|
||||||
const type = resp.readInt32LE(0);
|
|
||||||
const dataSize = resp.readInt32LE(4);
|
|
||||||
|
|
||||||
if (type < 0 || type >= Object.keys(Types).length) throw new Error('invalid type');
|
|
||||||
|
|
||||||
let data = socket.read(dataSize);
|
|
||||||
if (!data) throw new Error('failed reading data');
|
|
||||||
|
|
||||||
data = JSON.parse(Buffer.from(data).toString());
|
|
||||||
|
|
||||||
switch (type) {
|
|
||||||
case Types.PING:
|
|
||||||
socket.emit('ping', data);
|
|
||||||
socket.write(encode(Types.PONG, data));
|
|
||||||
break;
|
|
||||||
|
|
||||||
case Types.PONG:
|
|
||||||
socket.emit('pong', data);
|
|
||||||
break;
|
|
||||||
|
|
||||||
case Types.HANDSHAKE:
|
|
||||||
if (socket._handshook) throw new Error('already handshook');
|
|
||||||
|
|
||||||
socket._handshook = true;
|
|
||||||
socket.emit('handshake', data);
|
|
||||||
break;
|
|
||||||
|
|
||||||
case Types.FRAME:
|
|
||||||
if (!socket._handshook) throw new Error('need to handshake first');
|
|
||||||
|
|
||||||
socket.emit('request', data);
|
|
||||||
break;
|
|
||||||
|
|
||||||
case Types.CLOSE:
|
|
||||||
socket.end();
|
|
||||||
socket.destroy();
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
|
|
||||||
read(socket);
|
|
||||||
};
|
|
||||||
|
|
||||||
const socketIsAvailable = async socket => {
|
|
||||||
socket.pause();
|
|
||||||
socket.on('readable', () => {
|
|
||||||
try {
|
|
||||||
read(socket);
|
|
||||||
} catch (e) {
|
|
||||||
// Debug: log('error whilst reading', e);
|
|
||||||
socket.end(
|
|
||||||
encode(Types.CLOSE, {
|
|
||||||
code: CloseCodes.CLOSE_UNSUPPORTED,
|
|
||||||
message: e.message,
|
|
||||||
}),
|
|
||||||
);
|
|
||||||
socket.destroy();
|
|
||||||
}
|
|
||||||
});
|
|
||||||
|
|
||||||
const stop = () => {
|
|
||||||
try {
|
|
||||||
socket.end();
|
|
||||||
socket.destroy();
|
|
||||||
} catch {
|
|
||||||
// Debug
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
const possibleOutcomes = Promise.race([
|
|
||||||
new Promise(res => socket.on('error', res)), // Errored
|
|
||||||
// eslint-disable-next-line prefer-promise-reject-errors
|
|
||||||
new Promise((res, rej) => socket.on('pong', () => rej('socket ponged'))), // Ponged
|
|
||||||
// eslint-disable-next-line prefer-promise-reject-errors
|
|
||||||
new Promise((res, rej) => setTimeout(() => rej('timed out'), 1000).unref()), // Timed out
|
|
||||||
]).then(
|
|
||||||
() => true,
|
|
||||||
e => e,
|
|
||||||
);
|
|
||||||
|
|
||||||
socket.write(encode(Types.PING, ++uniqueId));
|
|
||||||
|
|
||||||
const outcome = await possibleOutcomes;
|
|
||||||
stop();
|
|
||||||
// Debug: log('checked if socket is available:', outcome === true, outcome === true ? '' : `- reason: ${outcome}`);
|
|
||||||
|
|
||||||
return outcome === true;
|
|
||||||
};
|
|
||||||
|
|
||||||
const getAvailableSocket = async (tries = 0) => {
|
|
||||||
if (tries > 9) {
|
|
||||||
throw new Error('ran out of tries to find socket', tries);
|
|
||||||
}
|
|
||||||
|
|
||||||
const path = `${SOCKET_PATH}-${tries}`;
|
|
||||||
const socket = createConnection(path);
|
|
||||||
|
|
||||||
// Debug: log('checking', path);
|
|
||||||
|
|
||||||
if (await socketIsAvailable(socket)) {
|
|
||||||
if (platform !== 'win32') {
|
|
||||||
try {
|
|
||||||
unlinkSync(path);
|
|
||||||
} catch {
|
|
||||||
// Debug
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return path;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Debug: log(`not available, trying again (attempt ${tries + 1})`);
|
|
||||||
return getAvailableSocket(tries + 1);
|
|
||||||
};
|
|
||||||
|
|
||||||
module.exports = class IPCServer {
|
|
||||||
constructor(handers, debug = false) {
|
|
||||||
// eslint-disable-next-line no-async-promise-executor
|
|
||||||
return new Promise(async res => {
|
|
||||||
this.debug = debug;
|
|
||||||
this.handlers = handers;
|
|
||||||
|
|
||||||
this.onConnection = this.onConnection.bind(this);
|
|
||||||
this.onMessage = this.onMessage.bind(this);
|
|
||||||
|
|
||||||
const server = createServer(this.onConnection);
|
|
||||||
server.on('error', e => {
|
|
||||||
if (this.debug) log('server error', e);
|
|
||||||
});
|
|
||||||
|
|
||||||
const socketPath = await getAvailableSocket();
|
|
||||||
server.listen(socketPath, () => {
|
|
||||||
if (this.debug) log('listening at', socketPath);
|
|
||||||
this.server = server;
|
|
||||||
|
|
||||||
res(this);
|
|
||||||
});
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
onConnection(socket) {
|
|
||||||
if (this.debug) log('new connection!');
|
|
||||||
|
|
||||||
socket.pause();
|
|
||||||
socket.on('readable', () => {
|
|
||||||
try {
|
|
||||||
read(socket);
|
|
||||||
} catch (e) {
|
|
||||||
if (this.debug) log('error whilst reading', e);
|
|
||||||
|
|
||||||
socket.end(
|
|
||||||
encode(Types.CLOSE, {
|
|
||||||
code: CloseCodes.CLOSE_UNSUPPORTED,
|
|
||||||
message: e.message,
|
|
||||||
}),
|
|
||||||
);
|
|
||||||
socket.destroy();
|
|
||||||
}
|
|
||||||
});
|
|
||||||
|
|
||||||
socket.once('handshake', params => {
|
|
||||||
if (this.debug) log('handshake:', params);
|
|
||||||
|
|
||||||
const ver = parseInt(params.v ?? 1);
|
|
||||||
const clientId = params.client_id ?? '';
|
|
||||||
// Encoding is always json for ipc
|
|
||||||
|
|
||||||
socket.close = (code = CloseCodes.CLOSE_NORMAL, message = '') => {
|
|
||||||
socket.end(
|
|
||||||
encode(Types.CLOSE, {
|
|
||||||
code,
|
|
||||||
message,
|
|
||||||
}),
|
|
||||||
);
|
|
||||||
socket.destroy();
|
|
||||||
};
|
|
||||||
|
|
||||||
if (ver !== 1) {
|
|
||||||
if (this.debug) log('unsupported version requested', ver);
|
|
||||||
|
|
||||||
socket.close(ErrorCodes.INVALID_VERSION);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (clientId === '') {
|
|
||||||
if (this.debug) log('client id required');
|
|
||||||
|
|
||||||
socket.close(ErrorCodes.INVALID_CLIENTID);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
socket.on('error', e => {
|
|
||||||
if (this.debug) log('socket error', e);
|
|
||||||
});
|
|
||||||
|
|
||||||
socket.on('close', e => {
|
|
||||||
if (this.debug) log('socket closed', e);
|
|
||||||
this.handlers.close(socket);
|
|
||||||
});
|
|
||||||
|
|
||||||
socket.on('request', this.onMessage.bind(this, socket));
|
|
||||||
|
|
||||||
socket._send = socket.send;
|
|
||||||
socket.send = msg => {
|
|
||||||
if (this.debug) log('sending', msg);
|
|
||||||
socket.write(encode(Types.FRAME, msg));
|
|
||||||
};
|
|
||||||
|
|
||||||
socket.clientId = clientId;
|
|
||||||
|
|
||||||
this.handlers.connection(socket);
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
onMessage(socket, msg) {
|
|
||||||
if (this.debug) log('message', msg);
|
|
||||||
this.handlers.message(socket, msg);
|
|
||||||
}
|
|
||||||
};
|
|
||||||
@ -1,128 +0,0 @@
|
|||||||
'use strict';
|
|
||||||
|
|
||||||
const rgb = (r, g, b, msg) => `\x1b[38;2;${r};${g};${b}m${msg}\x1b[0m`;
|
|
||||||
const log = (...args) => console.log(`[${rgb(88, 101, 242, 'arRPC')} > ${rgb(235, 69, 158, 'websocket')}]`, ...args);
|
|
||||||
|
|
||||||
const { createServer } = require('http');
|
|
||||||
const { parse } = require('querystring');
|
|
||||||
const { WebSocketServer } = require('ws');
|
|
||||||
|
|
||||||
const portRange = [6463, 6472]; // Ports available/possible: 6463-6472
|
|
||||||
|
|
||||||
module.exports = class WSServer {
|
|
||||||
constructor(handlers, debug = false) {
|
|
||||||
return (async () => {
|
|
||||||
this.debug = debug;
|
|
||||||
|
|
||||||
this.handlers = handlers;
|
|
||||||
|
|
||||||
this.onConnection = this.onConnection.bind(this);
|
|
||||||
this.onMessage = this.onMessage.bind(this);
|
|
||||||
|
|
||||||
let port = portRange[0];
|
|
||||||
|
|
||||||
let http, wss;
|
|
||||||
while (port <= portRange[1]) {
|
|
||||||
if (this.debug) log('trying port', port);
|
|
||||||
|
|
||||||
if (
|
|
||||||
await new Promise(res => {
|
|
||||||
http = createServer();
|
|
||||||
http.on('error', e => {
|
|
||||||
// Log('http error', e);
|
|
||||||
|
|
||||||
if (e.code === 'EADDRINUSE') {
|
|
||||||
if (this.debug) log(port, 'in use!');
|
|
||||||
res(false);
|
|
||||||
}
|
|
||||||
});
|
|
||||||
|
|
||||||
wss = new WebSocketServer({ server: http });
|
|
||||||
// eslint-disable-next-line no-unused-vars
|
|
||||||
wss.on('error', e => {
|
|
||||||
// Debug: Log('wss error', e);
|
|
||||||
});
|
|
||||||
|
|
||||||
wss.on('connection', this.onConnection);
|
|
||||||
|
|
||||||
http.listen(port, '127.0.0.1', () => {
|
|
||||||
if (this.debug) log('listening on', port);
|
|
||||||
|
|
||||||
this.http = http;
|
|
||||||
this.wss = wss;
|
|
||||||
|
|
||||||
res(true);
|
|
||||||
});
|
|
||||||
})
|
|
||||||
) {
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
port++;
|
|
||||||
}
|
|
||||||
|
|
||||||
return this;
|
|
||||||
})();
|
|
||||||
}
|
|
||||||
|
|
||||||
onConnection(socket, req) {
|
|
||||||
const params = parse(req.url.split('?')[1]);
|
|
||||||
const ver = parseInt(params.v ?? 1);
|
|
||||||
const encoding = params.encoding ?? 'json'; // Json | etf (erlpack)
|
|
||||||
const clientId = params.client_id ?? '';
|
|
||||||
|
|
||||||
const origin = req.headers.origin ?? '';
|
|
||||||
|
|
||||||
if (this.debug) log(`new connection! origin:`, origin, JSON.parse(JSON.stringify(params)));
|
|
||||||
|
|
||||||
if (
|
|
||||||
origin !== '' &&
|
|
||||||
!['https://discord.com', 'https://ptb.discord.com', 'https://canary.discord.com/'].includes(origin)
|
|
||||||
) {
|
|
||||||
if (this.debug) log('disallowed origin', origin);
|
|
||||||
|
|
||||||
socket.close();
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (encoding !== 'json') {
|
|
||||||
if (this.debug) log('unsupported encoding requested', encoding);
|
|
||||||
|
|
||||||
socket.close();
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (ver !== 1) {
|
|
||||||
if (this.debug) log('unsupported version requested', ver);
|
|
||||||
|
|
||||||
socket.close();
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
socket.clientId = clientId;
|
|
||||||
socket.encoding = encoding;
|
|
||||||
|
|
||||||
socket.on('error', e => {
|
|
||||||
if (this.debug) log('socket error', e);
|
|
||||||
});
|
|
||||||
|
|
||||||
socket.on('close', (e, r) => {
|
|
||||||
if (this.debug) log('socket closed', e, r);
|
|
||||||
this.handlers.close(socket);
|
|
||||||
});
|
|
||||||
|
|
||||||
socket.on('message', this.onMessage.bind(this, socket));
|
|
||||||
|
|
||||||
socket._send = socket.send;
|
|
||||||
socket.send = msg => {
|
|
||||||
if (this.debug) log('sending', msg);
|
|
||||||
socket._send(JSON.stringify(msg));
|
|
||||||
};
|
|
||||||
|
|
||||||
this.handlers.connection(socket);
|
|
||||||
}
|
|
||||||
|
|
||||||
onMessage(socket, msg) {
|
|
||||||
if (this.debug) log('message', JSON.parse(msg));
|
|
||||||
this.handlers.message(socket, JSON.parse(msg));
|
|
||||||
}
|
|
||||||
};
|
|
||||||
Loading…
Reference in New Issue
Block a user