const crypto = require('crypto'); const fs = require("fs").promises; class Helpers { /** * Generates a random long integer (8 bytes), which is optionally signed * @returns {number} */ static generateRandomLong(signed) { let buf = Buffer.from(this.generateRandomBytes(8)); // 0x12345678 = 305419896 if (signed) return buf.readInt32BE(0); else return buf.readUInt32LE(0); } /** * Generates a random bytes array * @param count * @returns {Buffer} */ static generateRandomBytes(count) { return crypto.randomBytes(count); } /** * Loads the user settings located under `api/` * @param path * @returns {Promise} */ static async loadSettings(path = "../api/settings") { let settings = {}; let left, right, value_pair; let data = await fs.readFile(path, 'utf-8'); for (let line of data.toString().split('\n')) { value_pair = line.split("="); if (value_pair.length !== 2) { break; } left = value_pair[0].replace(/ \r?\n|\r/g, ''); right = value_pair[1].replace(/ \r?\n|\r/g, ''); if (!isNaN(right)) { settings[left] = Number.parseInt(right); } else { settings[left] = right; } } return settings; } /** * Calculate the key based on Telegram guidelines, specifying whether it's the client or not * @param shared_key * @param msg_key * @param client * @returns {[*, *]} */ static calcKey(shared_key, msg_key, client) { let x = client !== null ? 0 : 8; let iv, key, sha1a, sha1b, sha1c, sha1d; sha1a = this.sha1((msg_key + shared_key.slice(x, (x + 32)))); sha1b = this.sha1(((shared_key.slice((x + 32), (x + 48)) + msg_key) + shared_key.slice((x + 48), (x + 64)))); sha1c = this.sha1((shared_key.slice((x + 64), (x + 96)) + msg_key)); sha1d = this.sha1((msg_key + shared_key.slice((x + 96), (x + 128)))); key = ((sha1a.slice(0, 8) + sha1b.slice(8, 20)) + sha1c.slice(4, 16)); iv = (((sha1a.slice(8, 20) + sha1b.slice(0, 8)) + sha1c.slice(16, 20)) + sha1d.slice(0, 8)); return [key, iv]; } /** * Calculates the message key from the given data * @param data * @returns {Buffer} */ static calcMsgKey(data) { return this.sha1(data).slice(4, 20); } /** * Generates the key data corresponding to the given nonces * @param serverNonce * @param newNonce * @returns {{ivBuffer: Buffer, keyBuffer: Buffer}} */ static generateKeyDataFromNonces(serverNonce, newNonce) { let hash1 = this.sha1(Buffer.concat([newNonce, serverNonce])); let hash2 = this.sha1(Buffer.concat([serverNonce, newNonce])); let hash3 = this.sha1(Buffer.concat([newNonce, newNonce])); let keyBuffer = Buffer.concat([hash1, hash1.slice(0, 12)]); let ivBuffer = Buffer.concat([hash2.slice(12, 20), hash3, newNonce.slice(0, 4)]); return {keyBuffer: keyBuffer, ivBuffer: ivBuffer} } /** * Calculates the SHA1 digest for the given data * @param data * @returns {Buffer} */ static sha1(data) { let shaSum = crypto.createHash('sha1'); shaSum.update(data); return shaSum.digest(); } /** * * @param buffer {Buffer} * @param offset {Number} * @returns {{data: {Buffer}, offset: {Number}}} */ static tgReadByte(buffer, offset) { let firstByte = buffer.readInt8(offset); offset += 1; let padding, length; if (firstByte === 255) { length = buffer.readInt8(offset) | (buffer.readInt8(offset) << 8) | (buffer.readInt8(offset) << 16); offset += 1; padding = length % 4; } else { length = firstByte; padding = (length + 1) % 4; } let data = buffer.readInt8(offset); offset += 1; if (padding > 0) { padding = 4 - padding; offset += padding; } return {data, offset} } static tgWriteBytes(data) { let buffer; let padding; if (data.length < 254) { padding = (data.length + 1) % 4; if (padding !== 0) { padding = 4 - padding; } buffer = Buffer.from([data.length, data]); } else { padding = data.length % 4; if (padding !== 0) { padding = 4 - padding; } buffer = Buffer.concat([Buffer.from([254]), Buffer.from([data.length % 256]), Buffer.from([(data.length >> 8) % 256]), Buffer.from([(data.length >> 16) % 256]), Buffer.from([data]), Buffer.from([padding]) ]); } return buffer; } /** * returns a random int from min (inclusive) and max (inclusive) * @param min * @param max * @returns {number} */ static getRandomInt(min, max) { min = Math.ceil(min); max = Math.floor(max); return Math.floor(Math.random() * (max - min + 1)) + min; } /** * Sleeps a specified amount of time * @param ms time in milliseconds * @returns {Promise} */ static sleep = ms => new Promise(resolve => setTimeout(resolve, ms)); } module.exports = Helpers;