/* * Vencord, a modification for Discord's desktop app * Copyright (c) 2022 Vendicated and contributors * * This program is free software: you can redistribute it and/or modify * it under the terms of the GNU General Public License as published by * the Free Software Foundation, either version 3 of the License, or * (at your option) any later version. * * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU General Public License for more details. * * You should have received a copy of the GNU General Public License * along with this program. If not, see . */ // eslint-disable-next-line spaced-comment /// // eslint-disable-next-line spaced-comment /// import { readFileSync } from "fs"; import pup, { JSHandle } from "puppeteer-core"; for (const variable of ["DISCORD_TOKEN", "CHROMIUM_BIN"]) { if (!process.env[variable]) { console.error(`Missing environment variable ${variable}`); process.exit(1); } } const CANARY = process.env.USE_CANARY === "true"; const browser = await pup.launch({ headless: "new", executablePath: process.env.CHROMIUM_BIN }); const page = await browser.newPage(); await page.setUserAgent("Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/108.0.0.0 Safari/537.36"); async function maybeGetError(handle: JSHandle) { return (handle as JSHandle).getProperty("message") .then(m => m?.jsonValue() ?? "Unknown Error"); } const report = { badPatches: [] as { plugin: string; type: string; id: string; match: string; error?: string; }[], badStarts: [] as { plugin: string; error: string; }[], otherErrors: [] as string[], badWebpackFinds: [] as string[] }; const IGNORED_DISCORD_ERRORS = [ "KeybindStore: Looking for callback action", "Unable to process domain list delta: Client revision number is null", "Downloading the full bad domains file", /\[GatewaySocket\].{0,110}Cannot access '/, "search for 'name' in undefined", "Attempting to set fast connect zstd when unsupported" ] as Array; function toCodeBlock(s: string) { s = s.replace(/```/g, "`\u200B`\u200B`"); return "```" + s + " ```"; } async function printReport() { console.log(); console.log("# Vencord Report" + (CANARY ? " (Canary)" : "")); console.log(); console.log("## Bad Patches"); report.badPatches.forEach(p => { console.log(`- ${p.plugin} (${p.type})`); console.log(` - ID: \`${p.id}\``); console.log(` - Match: ${toCodeBlock(p.match)}`); if (p.error) console.log(` - Error: ${toCodeBlock(p.error)}`); }); console.log(); console.log("## Bad Webpack Finds"); report.badWebpackFinds.forEach(p => console.log("- " + p)); console.log(); console.log("## Bad Starts"); report.badStarts.forEach(p => { console.log(`- ${p.plugin}`); console.log(` - Error: ${toCodeBlock(p.error)}`); }); console.log(); const ignoredErrors = [] as string[]; report.otherErrors = report.otherErrors.filter(e => { if (IGNORED_DISCORD_ERRORS.some(regex => e.match(regex))) { ignoredErrors.push(e); return false; } return true; }); console.log("## Discord Errors"); report.otherErrors.forEach(e => { console.log(`- ${toCodeBlock(e)}`); }); console.log(); console.log("## Ignored Discord Errors"); ignoredErrors.forEach(e => { console.log(`- ${toCodeBlock(e)}`); }); console.log(); if (process.env.DISCORD_WEBHOOK) { await fetch(process.env.DISCORD_WEBHOOK, { method: "POST", headers: { "Content-Type": "application/json" }, body: JSON.stringify({ description: "Here's the latest Vencord Report!", username: "Vencord Reporter" + (CANARY ? " (Canary)" : ""), avatar_url: "https://cdn.discordapp.com/avatars/1017176847865352332/c312b6b44179ae6817de7e4b09e9c6af.webp?size=512", embeds: [ { title: "Bad Patches", description: report.badPatches.map(p => { const lines = [ `**__${p.plugin} (${p.type}):__**`, `ID: \`${p.id}\``, `Match: ${toCodeBlock(p.match)}` ]; if (p.error) lines.push(`Error: ${toCodeBlock(p.error)}`); return lines.join("\n"); }).join("\n\n") || "None", color: report.badPatches.length ? 0xff0000 : 0x00ff00 }, { title: "Bad Webpack Finds", description: report.badWebpackFinds.map(toCodeBlock).join("\n") || "None", color: report.badWebpackFinds.length ? 0xff0000 : 0x00ff00 }, { title: "Bad Starts", description: report.badStarts.map(p => { const lines = [ `**__${p.plugin}:__**`, toCodeBlock(p.error) ]; return lines.join("\n"); } ).join("\n\n") || "None", color: report.badStarts.length ? 0xff0000 : 0x00ff00 }, { title: "Discord Errors", description: report.otherErrors.length ? toCodeBlock(report.otherErrors.join("\n")) : "None", color: report.otherErrors.length ? 0xff0000 : 0x00ff00 } ] }) }).then(res => { if (!res.ok) console.error(`Webhook failed with status ${res.status}`); else console.error("Posted to Discord Webhook successfully"); }); } } page.on("console", async e => { const level = e.type(); const rawArgs = e.args(); const firstArg = await rawArgs[0]?.jsonValue(); if (firstArg === "[PUPPETEER_TEST_DONE_SIGNAL]") { await browser.close(); await printReport(); process.exit(); } const isVencord = firstArg === "[Vencord]"; const isDebug = firstArg === "[PUP_DEBUG]"; const isWebpackFindFail = firstArg === "[PUP_WEBPACK_FIND_FAIL]"; if (isWebpackFindFail) { process.exitCode = 1; report.badWebpackFinds.push(await rawArgs[1].jsonValue() as string); } if (isVencord) { const args = await Promise.all(e.args().map(a => a.jsonValue())); const [, tag, message] = args as Array; const cause = await maybeGetError(e.args()[3]); switch (tag) { case "WebpackInterceptor:": const patchFailMatch = message.match(/Patch by (.+?) (had no effect|errored|found no module) \(Module id is (.+?)\): (.+)/)!; if (!patchFailMatch) break; process.exitCode = 1; const [, plugin, type, id, regex] = patchFailMatch; report.badPatches.push({ plugin, type, id, match: regex.replace(/\[A-Za-z_\$\]\[\\w\$\]\*/g, "\\i"), error: cause }); break; case "PluginManager:": const failedToStartMatch = message.match(/Failed to start (.+)/); if (!failedToStartMatch) break; process.exitCode = 1; const [, name] = failedToStartMatch; report.badStarts.push({ plugin: name, error: cause }); break; } } async function getText() { try { return await Promise.all( e.args().map(async a => { return await maybeGetError(a) || await a.jsonValue(); }) ).then(a => a.join(" ").trim()); } catch { return e.text(); } } if (isDebug) { const text = await getText(); console.error(text); if (text.includes("A fatal error occurred:")) { process.exit(1); } } else if (level === "error") { const text = await getText(); if (text.length && !text.startsWith("Failed to load resource: the server responded with a status of") && !text.includes("Webpack")) { console.error("[Unexpected Error]", text); report.otherErrors.push(text); } } }); page.on("error", e => console.error("[Error]", e)); page.on("pageerror", e => console.error("[Page Error]", e)); await page.setBypassCSP(true); async function runtime(token: string) { console.log("[PUP_DEBUG]", "Starting test..."); try { // Spoof languages to not be suspicious Object.defineProperty(navigator, "languages", { get: function () { return ["en-US", "en"]; } }); // Monkey patch Logger to not log with custom css // @ts-ignore const originalLog = Vencord.Util.Logger.prototype._log; // @ts-ignore Vencord.Util.Logger.prototype._log = function (level, levelColor, args) { if (level === "warn" || level === "error") return console[level]("[Vencord]", this.name + ":", ...args); return originalLog.call(this, level, levelColor, args); }; // Force enable all plugins and patches Vencord.Plugins.patches.length = 0; Object.values(Vencord.Plugins.plugins).forEach(p => { // Needs native server to run if (p.name === "WebRichPresence (arRPC)") return; Vencord.Settings.plugins[p.name].enabled = true; p.patches?.forEach(patch => { patch.plugin = p.name; delete patch.predicate; delete patch.group; Vencord.Util.canonicalizeFind(patch); if (!Array.isArray(patch.replacement)) { patch.replacement = [patch.replacement]; } patch.replacement.forEach(r => { delete r.predicate; }); Vencord.Plugins.patches.push(patch); }); }); // Enable eagerPatches to make all patches apply regardless of the module being required Vencord.Settings.eagerPatches = true; // The main patch for starting the reporter chunk loading Vencord.Plugins.patches.push({ plugin: "Vencord Reporter", find: '"Could not find app-mount"', replacement: [{ match: /(?<="use strict";)/, replace: "Vencord.Webpack._initReporter();" }] }); Vencord.Webpack.waitFor( "loginToken", m => { console.log("[PUP_DEBUG]", "Logging in with token..."); m.loginToken(token); } ); // @ts-ignore Vencord.Webpack._initReporter = function () { // initReporter is called in the patched entry point of Discord // setImmediate to only start searching for lazy chunks after Discord initialized the app setTimeout(() => { console.log("[PUP_DEBUG]", "Loading all chunks..."); Vencord.Webpack.factoryListeners.add(factory => { // setImmediate to avoid blocking the factory patching execution while checking for lazy chunks setTimeout(() => { let isResolved = false; searchAndLoadLazyChunks(String(factory)) .then(() => isResolved = true) .catch(() => isResolved = true); chunksSearchPromises.push(() => isResolved); }, 0); }); for (const factoryId in wreq.m) { let isResolved = false; searchAndLoadLazyChunks(String(wreq.m[factoryId])) .then(() => isResolved = true) .catch(() => isResolved = true); chunksSearchPromises.push(() => isResolved); } }, 0); }; const wreq = Vencord.Util.proxyLazy(() => Vencord.Webpack.wreq); const { canonicalizeMatch, Logger } = Vencord.Util; const validChunks = new Set(); const invalidChunks = new Set(); const deferredRequires = new Set(); let chunksSearchingResolve: (value: void | PromiseLike) => void; const chunksSearchingDone = new Promise(r => chunksSearchingResolve = r); // True if resolved, false otherwise const chunksSearchPromises = [] as Array<() => boolean>; const LazyChunkRegex = canonicalizeMatch(/(?:Promise\.all\(\[(\i\.\i\("[^)]+?"\)[^\]]+?)\]\)|(\i\.\i\("[^)]+?"\)))\.then\(\i\.bind\(\i,"([^)]+?)"\)\)/g); async function searchAndLoadLazyChunks(factoryCode: string) { const lazyChunks = factoryCode.matchAll(LazyChunkRegex); const validChunkGroups = new Set<[chunkIds: string[], entryPoint: string]>(); // Workaround for a chunk that depends on the ChannelMessage component but may be be force loaded before // the chunk containing the component const shouldForceDefer = factoryCode.includes(".Messages.GUILD_FEED_UNFEATURE_BUTTON_TEXT"); await Promise.all(Array.from(lazyChunks).map(async ([, rawChunkIdsArray, rawChunkIdsSingle, entryPoint]) => { const rawChunkIds = rawChunkIdsArray ?? rawChunkIdsSingle; const chunkIds = rawChunkIds ? Array.from(rawChunkIds.matchAll(Vencord.Webpack.ChunkIdsRegex)).map(m => m[1]) : []; if (chunkIds.length === 0) { return; } let invalidChunkGroup = false; for (const id of chunkIds) { if (wreq.u(id) == null || wreq.u(id) === "undefined.js") continue; const isWasm = await fetch(wreq.p + wreq.u(id)) .then(r => r.text()) .then(t => t.includes(".module.wasm") || !t.includes("(this.webpackChunkdiscord_app=this.webpackChunkdiscord_app||[]).push")); if (isWasm) { invalidChunks.add(id); invalidChunkGroup = true; continue; } validChunks.add(id); } if (!invalidChunkGroup) { validChunkGroups.add([chunkIds, entryPoint]); } })); // Loads all found valid chunk groups await Promise.all( Array.from(validChunkGroups) .map(([chunkIds]) => Promise.all(chunkIds.map(id => wreq.e(id))) ) ); // Requires the entry points for all valid chunk groups for (const [, entryPoint] of validChunkGroups) { try { if (shouldForceDefer) { deferredRequires.add(entryPoint); continue; } if (wreq.m[entryPoint]) wreq(entryPoint); } catch (err) { console.error(err); } } // setImmediate to only check if all chunks were loaded after this function resolves // We check if all chunks were loaded every time a factory is loaded // If we are still looking for chunks in the other factories, the array will have that factory's chunk search promise not resolved // But, if all chunk search promises are resolved, this means we found every lazy chunk loaded by Discord code and manually loaded them setTimeout(() => { let allResolved = true; for (let i = 0; i < chunksSearchPromises.length; i++) { const isResolved = chunksSearchPromises[i](); if (isResolved) { // Remove finished promises to avoid having to iterate through a huge array everytime chunksSearchPromises.splice(i--, 1); } else { allResolved = false; } } if (allResolved) chunksSearchingResolve(); }, 0); } await chunksSearchingDone; // Require deferred entry points for (const deferredRequire of deferredRequires) { wreq(deferredRequire); } // All chunks Discord has mapped to asset files, even if they are not used anymore const allChunks = [] as string[]; // Matches "id" or id: for (const currentMatch of String(wreq.u).matchAll(/(?:"(\d+?)")|(?:(\d+?):)/g)) { const id = currentMatch[1] ?? currentMatch[2]; if (id == null) continue; allChunks.push(id); } if (allChunks.length === 0) throw new Error("Failed to get all chunks"); // Chunks that are not loaded (not used) by Discord code anymore const chunksLeft = allChunks.filter(id => { return !(validChunks.has(id) || invalidChunks.has(id)); }); await Promise.all(chunksLeft.map(async id => { const isWasm = await fetch(wreq.p + wreq.u(id)) .then(r => r.text()) .then(t => t.includes(".module.wasm") || !t.includes("(this.webpackChunkdiscord_app=this.webpackChunkdiscord_app||[]).push")); // Loads and requires a chunk if (!isWasm) { await wreq.e(id); if (wreq.m[id]) wreq(id); } })); console.log("[PUP_DEBUG]", "Finished loading all chunks!"); for (const patch of Vencord.Plugins.patches) { if (!patch.all) { new Logger("WebpackInterceptor").warn(`Patch by ${patch.plugin} found no module (Module id is -): ${patch.find}`); } } for (const [searchType, args] of Vencord.Webpack.lazyWebpackSearchHistory) { let method = searchType; if (searchType === "findComponent") method = "find"; if (searchType === "findExportedComponent") method = "findByProps"; if (searchType === "waitFor" || searchType === "waitForComponent") { if (typeof args[0] === "string") method = "findByProps"; else method = "find"; } if (searchType === "waitForStore") method = "findStore"; try { let result: any; if (method === "proxyLazyWebpack" || method === "LazyComponentWebpack") { const [factory] = args; result = factory(); } else if (method === "extractAndLoadChunks") { const [code, matcher] = args; const module = Vencord.Webpack.findModuleFactory(...code); if (module) result = String(module).match(canonicalizeMatch(matcher)); } else { // @ts-ignore result = Vencord.Webpack[method](...args); } if (result == null || ("$$vencordInternal" in result && result.$$vencordInternal() == null)) throw "a rock at ben shapiro"; } catch (e) { let logMessage = searchType; if (method === "find" || method === "proxyLazyWebpack" || method === "LazyComponentWebpack") logMessage += `(${String(args[0]).slice(0, 147)}...)`; else if (method === "extractAndLoadChunks") logMessage += `([${args[0].map(arg => `"${arg}"`).join(", ")}], ${String(args[1])})`; else logMessage += `(${args.map(arg => `"${arg}"`).join(", ")})`; console.log("[PUP_WEBPACK_FIND_FAIL]", logMessage); } } setTimeout(() => console.log("[PUPPETEER_TEST_DONE_SIGNAL]"), 1000); } catch (e) { console.log("[PUP_DEBUG]", "A fatal error occurred:", e); } } await page.evaluateOnNewDocument(` ${readFileSync("./dist/browser.js", "utf-8")} ;(${runtime.toString()})(${JSON.stringify(process.env.DISCORD_TOKEN)}); `); await page.goto(CANARY ? "https://canary.discord.com/login" : "https://discord.com/login");