murm/mrmBot-Matrix
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Actually made screenshot owner-only, made image API interaction more secure

Browse source
This commit is contained in:
TheEssem 2020-12-14 12:39:18 -06:00
parent 78ae47dbbb
commit 69d8100f23
3 changed files with 39 additions and 32 deletions
Download patch file Download diff file Expand all files Collapse all files

62
utils/image.js
View file

@ -64,39 +64,45 @@ exports.run = (object, fromAPI = false) => {
const currentServer = servers[Math.floor(Math.random() * servers.length)];
const socket = dgram.createSocket("udp4");
const data = Buffer.concat([Buffer.from([0x1]), Buffer.from(JSON.stringify(object))]);
//let jobID;
let timeout = setTimeout(() => {
reject("Timed out");
}, 25000);
let jobID;
socket.on("message", (msg) => {
clearTimeout(timeout);
const opcode = msg.readUint8(0);
const req = msg.slice(1, msg.length);
const req = msg.slice(37, msg.length);
const uuid = msg.slice(1, 36).toString();
if (opcode === 0x0) {
//jobID = req;
//console.log(`Our job UUID is: ${jobID}`);
jobID = uuid;
timeout = setTimeout(() => {
reject("Timed out");
}, 300000);
} else if (opcode === 0x1) {
//console.log(`Job ${jobID} is finished!`);
const client = net.createConnection(req.toString(), currentServer);
const array = [];
client.on("data", (rawData) => {
array.push(rawData);
/*if (rawData.length < 32 * 1024) {
client.end();
}*/
});
client.once("end", () => {
const data = Buffer.concat(array);
const format = getFormat(data, "\n");
const payload = {
buffer: data.slice(format.dataStart + 1),
type: format.buffer.toString().split("/")[1]
};
socket.close();
resolve(payload);
});
client.on("error", (err) => {
throw err;
});
if (jobID === uuid) {
const client = net.createConnection(req.toString(), currentServer);
const array = [];
client.on("data", (rawData) => {
array.push(rawData);
});
client.once("end", () => {
const data = Buffer.concat(array);
const format = getFormat(data, "\n");
const payload = {
buffer: data.slice(format.dataStart + 1),
type: format.buffer.toString().split("/")[1]
};
socket.close();
resolve(payload);
});
client.on("error", (err) => {
throw err;
});
}
} else if (opcode === 0x2) {
reject(req);
if (jobID === uuid) reject(req);
}
});