Security fixes pass #2

dist/commands/start/index.js

@@ -1,6 +1,9 @@
 import { Command } from "commander";
 import path from "path";
 import writeErrorFile from "../../functions/write-error-file";
+let retries = 0;
+let timeout;
+const MAX_RETRIES = 5;
 export default function () {
     return new Command("start")
         .description("Start production server")
@@ -9,6 +12,11 @@ export default function () {
     });
 }
 async function start() {
+    clearTimeout(timeout);
+    if (retries >= MAX_RETRIES) {
+        console.error(`Production server crashed ${MAX_RETRIES} times. Exiting.`);
+        process.exit(1);
+    }
     const dev_spawn_file = path.resolve(__dirname, "prod-spawn.ts");
     const spawn_options = {
         cmd: ["bun", dev_spawn_file],
@@ -22,6 +30,10 @@ async function start() {
         },
     };
     let dev_process = Bun.spawn(spawn_options);
+    retries++;
+    timeout = setTimeout(() => {
+        retries = 0;
+    }, 10000);
     const exited = await dev_process.exited;
     if (exited) {
         return await start();

dist/functions/bundler/build-on-start-error-handler.js

@@ -4,7 +4,7 @@ export default async function buildOnstartErrorHandler(params) {
     global.BUNDLER_CTX_DISPOSED = true;
     global.RECOMPILING = false;
     global.IS_SERVER_COMPONENT = false;
-    Promise.all([
+    await Promise.all([
         global.SSR_BUNDLER_CTX?.dispose(),
         global.BUNDLER_CTX?.dispose(),
     ]);

dist/functions/bunext-init.d.ts

@@ -1,6 +1,6 @@
 import type { BundlerCTXMap, BunextConfig, GlobalHMRControllerObject, PageFiles } from "../types";
 import type { FileSystemRouter, Server } from "bun";
-import grabDirNames from "../utils/grab-dir-names";
+import { type DirNames } from "../utils/grab-dir-names";
 import { type FSWatcher } from "fs";
 import type { BuildContext } from "esbuild";
 import grabConstants from "../utils/grab-constants";
@@ -31,7 +31,7 @@ declare global {
     var SKIPPED_BROWSER_MODULES: Set<string>;
     var BUNDLER_CTX: BuildContext | undefined;
     var SSR_BUNDLER_CTX: BuildContext | undefined;
-    var DIR_NAMES: ReturnType<typeof grabDirNames>;
+    var DIR_NAMES: DirNames;
     var REACT_IMPORTS_MAP: {
         imports: Record<string, string>;
     };

dist/functions/bunext-init.js

@@ -1,4 +1,4 @@
-import grabDirNames from "../utils/grab-dir-names";
+import grabDirNames, {} from "../utils/grab-dir-names";
 import {} from "fs";
 import init from "./init";
 import isDevelopment from "../utils/is-development";

dist/functions/cache/trim-all-cache.js

@@ -13,6 +13,10 @@ export default async function trimAllCache() {
             const trim_key = await trimCacheKey({
                 key: cache_key,
             });
+            if (trim_key.success) {
+                cached_items.splice(i, 1);
+                i--;
+            }
         }
     }
     catch (error) {

dist/functions/server/bunext-req-handler.js

@@ -8,6 +8,8 @@ import handleBunextPublicAssets from "./handle-bunext-public-assets";
 import checkExcludedPatterns from "../../utils/check-excluded-patterns";
 import { AppData } from "../../data/app-data";
 import fullRebuild from "./full-rebuild";
+const HMR_RETRY_COOLDOWN_MS = 5000;
+let lastHmrRetryTime = 0;
 export default async function bunextRequestHandler({ req: initial_req, server, }) {
     const is_dev = isDevelopment();
     let req = initial_req.clone();
@@ -30,6 +32,11 @@ export default async function bunextRequestHandler({ req: initial_req, server, }
             }
         }
         if (is_dev && url.pathname == AppData["BunextHMRRetryRoute"]) {
+            const now = Date.now();
+            if (now - lastHmrRetryTime < HMR_RETRY_COOLDOWN_MS) {
+                return new Response("Too Many Requests", { status: 429 });
+            }
+            lastHmrRetryTime = now;
             await fullRebuild({ msg: `HMR Retry Rebuild ...` });
             return new Response("Modules Rebuilt");
         }
@@ -60,8 +67,12 @@ export default async function bunextRequestHandler({ req: initial_req, server, }
         return response;
     }
     catch (error) {
+        if (is_dev) {
             return new Response(`Server Error: ${error.message}`, {
                 status: 500,
             });
         }
+        console.error(`Server Error: ${error.message}`, error);
+        return new Response("Internal Server Error", { status: 500 });
+    }
 }

dist/functions/server/handle-bunext-public-assets.js

@@ -2,13 +2,14 @@ import grabDirNames from "../../utils/grab-dir-names";
 import path from "path";
 import isDevelopment from "../../utils/is-development";
 import { readFileResponse } from "./handle-public";
+import isSafePath from "../../utils/is-safe-path";
 const { BUNEXT_PUBLIC_DIR } = grabDirNames();
 export default async function ({ req }) {
     try {
         const is_dev = isDevelopment();
         const url = new URL(req.url);
         const file_path = path.join(BUNEXT_PUBLIC_DIR, url.pathname.replace(/\/\.bunext\/public\//, ""));
-        if (!file_path.startsWith(BUNEXT_PUBLIC_DIR + path.sep)) {
+        if (!isSafePath({ filePath: file_path, allowedDir: BUNEXT_PUBLIC_DIR })) {
             return new Response("Forbidden", { status: 403 });
         }
         return readFileResponse({

dist/functions/server/handle-files.js

@@ -2,13 +2,14 @@ import grabDirNames from "../../utils/grab-dir-names";
 import path from "path";
 import isDevelopment from "../../utils/is-development";
 import { existsSync } from "fs";
+import isSafePath from "../../utils/is-safe-path";
 const { PUBLIC_DIR } = grabDirNames();
 export default async function ({ req }) {
     try {
         const is_dev = isDevelopment();
         const url = new URL(req.url);
         const file_path = path.join(PUBLIC_DIR, url.pathname);
-        if (!file_path.startsWith(PUBLIC_DIR + path.sep)) {
+        if (!isSafePath({ filePath: file_path, allowedDir: PUBLIC_DIR })) {
             return new Response("Forbidden", { status: 403 });
         }
         if (!existsSync(file_path)) {
@@ -17,7 +18,11 @@ export default async function ({ req }) {
             });
         }
         const file = Bun.file(file_path);
-        return new Response(file);
+        const headers = new Headers();
+        if (!is_dev) {
+            headers.set("Cache-Control", "public, max-age=3600");
+        }
+        return new Response(file, { headers });
     }
     catch (error) {
         return new Response(`File Not Found`, {

dist/functions/server/handle-hmr.js

@@ -1,5 +1,21 @@
+function removeController(controller) {
+    const idx = global.HMR_CONTROLLERS.findIndex((c) => c.controller == controller);
+    if (typeof idx == "number" && idx >= 0) {
+        global.HMR_CONTROLLERS.splice(idx, 1);
+    }
+}
 export default async function ({ req }) {
-    const referer_url = new URL(req.headers.get("referer") || "");
+    const referer = req.headers.get("referer");
+    if (!referer) {
+        return new Response("Missing Referer Header", { status: 400 });
+    }
+    let referer_url;
+    try {
+        referer_url = new URL(referer);
+    }
+    catch {
+        return new Response("Invalid Referer Header", { status: 400 });
+    }
     const match = global.ROUTER.match(referer_url.pathname);
     const target_map = match?.filePath
         ? global.BUNDLER_CTX_MAP?.[match.filePath]
@@ -20,16 +36,13 @@ export default async function ({ req }) {
                 }
                 catch {
                     clearInterval(heartbeat);
+                    removeController(controller);
                 }
             }, 5000);
         },
         cancel() {
             clearInterval(heartbeat);
-            const targetControllerIndex = global.HMR_CONTROLLERS.findIndex((c) => c.controller == controller);
+            removeController(controller);
-            if (typeof targetControllerIndex == "number" &&
-                targetControllerIndex >= 0) {
-                global.HMR_CONTROLLERS.splice(targetControllerIndex, 1);
-            }
         },
     });
     return new Response(stream, {

dist/functions/server/handle-public.js

@@ -2,13 +2,14 @@ import grabDirNames from "../../utils/grab-dir-names";
 import path from "path";
 import isDevelopment from "../../utils/is-development";
 import { existsSync } from "fs";
+import isSafePath from "../../utils/is-safe-path";
 const { PUBLIC_DIR } = grabDirNames();
 export default async function ({ req }) {
     try {
         const is_dev = isDevelopment();
         const url = new URL(req.url);
         const file_path = path.join(PUBLIC_DIR, url.pathname.replace(/^\/public/, ""));
-        if (!file_path.startsWith(PUBLIC_DIR + path.sep)) {
+        if (!isSafePath({ filePath: file_path, allowedDir: PUBLIC_DIR })) {
             return new Response("Forbidden", { status: 403 });
         }
         return readFileResponse({ file_path });
@@ -33,6 +34,9 @@ export function readFileResponse({ file_path, cache }) {
     else if (cache?.duration) {
         headers.set("Cache-Control", `public, max-age=${cache.duration}`);
     }
+    else if (!isDevelopment()) {
+        headers.set("Cache-Control", "public, max-age=3600");
+    }
     return new Response(file, {
         headers,
     });

dist/functions/server/handle-routes.js

@@ -41,12 +41,13 @@ export default async function ({ req }) {
         module = await import(import_path);
     }
     const config = module.config;
+    const maxBodyBytes = config?.max_request_body_mb
+        ? config.max_request_body_mb * MBInBytes
+        : ServerDefaultRequestBodyLimitBytes;
     const contentLength = req.headers.get("content-length");
     if (contentLength) {
         const size = parseInt(contentLength, 10);
-        if ((config?.max_request_body_mb &&
+        if (size > maxBodyBytes) {
-            size > config.max_request_body_mb * MBInBytes) ||
-            size > ServerDefaultRequestBodyLimitBytes) {
             return Response.json({
                 success: false,
                 msg: "Request Body Too Large!",
@@ -58,6 +59,21 @@ export default async function ({ req }) {
             });
         }
     }
+    else if (req.method !== "GET" && req.method !== "HEAD") {
+        const body = await req.arrayBuffer();
+        if (body.byteLength > maxBodyBytes) {
+            return Response.json({
+                success: false,
+                msg: "Request Body Too Large!",
+            }, {
+                status: 413,
+                headers: {
+                    "Content-Type": "application/json",
+                },
+            });
+        }
+        routeParams.body = JSON.parse(new TextDecoder().decode(body) || "{}");
+    }
     const target_module = (module["default"] ||
         module["handler"]);
     const res = await target_module?.({

dist/functions/server/server-post-build-fn.js

@@ -30,8 +30,8 @@ export default async function serverPostBuildFn(params) {
             controller.controller.enqueue(reload_enqueue);
             continue;
         }
-        const mock_req = target_artifact.req
+        const mock_req = target_artifact.req_url
-            ? target_artifact.req.clone()
+            ? new Request(target_artifact.req_url)
             : new Request(controller.page_url);
         const page_component = global.IS_SERVER_COMPONENT
             ? await grabPageComponent({

dist/functions/server/web-pages/generate-web-html.js

@@ -76,6 +76,8 @@ export default async function genWebHTML({ component: Main, pageProps, bundledMa
     console.error = () => { };
     console.info = () => { };
     console.debug = () => { };
+    let htmlBody;
+    try {
         const stream = await renderToReadableStream(final_component, {
             onError(error) {
                 if (error.message.includes('unique "key" prop'))
@@ -83,8 +85,11 @@ export default async function genWebHTML({ component: Main, pageProps, bundledMa
                 originalConsole.error(error);
             },
         });
-    const htmlBody = await new Response(stream).text();
+        htmlBody = await new Response(stream).text();
+    }
+    finally {
         Object.assign(console, originalConsole);
+    }
     html += htmlBody;
     return html;
 }

dist/functions/server/web-pages/grab-page-combined-server-res.js

@@ -33,7 +33,7 @@ export default async function grabPageCombinedServerRes({ file_path, debug, url,
     const page_server_ctx = global.SSR_BUNDLER_CTX_MAP[server_file_path || ""];
     const final_page_server_path = page_server_ctx?.local_path
         ? path.join(ROOT_DIR, page_server_ctx.path)
-        : root_server_file_path;
+        : server_file_path;
     const server_module = final_page_server_path
         ? await import(`${final_page_server_path}?t=${now}`)
         : undefined;

dist/functions/server/web-pages/grab-page-component.js

@@ -72,7 +72,7 @@ export default async function grabPageComponent(params) {
             }
         }
         if (req && !is_hydration) {
-            global.BUNDLER_CTX_MAP[file_path].req = req;
+            global.BUNDLER_CTX_MAP[file_path].req_url = req.url;
         }
         if (debug) {
             log.info(`bundledMap:`, bundledMap);

dist/types/index.d.ts

@@ -306,7 +306,7 @@ export type BundlerCTXMap = {
     url_path: string;
     file_name: string;
     css_path?: string;
-    req?: Request;
+    req_url?: string;
 };
 export type GlobalHMRControllerObject = {
     controller: ReadableStreamDefaultController<string>;

dist/utils/deserialize-query.d.ts

@@ -1,6 +1,3 @@
-/**
- * # Convert Serialized Query back to object
- */
 export default function deserializeQuery(query: string | {
     [s: string]: any;
 }): {

dist/utils/deserialize-query.js

@@ -1,18 +1,33 @@
 import EJSON from "./ejson";
-/**
+const DANGEROUS_KEYS = new Set(["__proto__", "constructor", "prototype"]);
- * # Convert Serialized Query back to object
+function sanitize(value) {
- */
+    if (value === null || typeof value !== "object")
+        return value;
+    if (Array.isArray(value))
+        return value.map(sanitize);
+    const clean = Object.create(null);
+    for (const key of Object.keys(value)) {
+        if (DANGEROUS_KEYS.has(key))
+            continue;
+        clean[key] = sanitize(value[key]);
+    }
+    return clean;
+}
 export default function deserializeQuery(query) {
     let queryObject = typeof query == "object" ? query : Object(EJSON.parse(query));
     const keys = Object.keys(queryObject);
     for (let i = 0; i < keys.length; i++) {
         const key = keys[i];
         const value = queryObject[key];
+        if (DANGEROUS_KEYS.has(key)) {
+            delete queryObject[key];
+            continue;
+        }
         if (typeof value == "string") {
             if (value.match(/^\{|^\[/)) {
-                queryObject[key] = EJSON.parse(value);
+                queryObject[key] = sanitize(EJSON.parse(value));
             }
         }
     }
-    return queryObject;
+    return sanitize(queryObject);
 }

dist/utils/grab-dir-names.d.ts

@@ -1,4 +1,4 @@
-export default function grabDirNames(): {
+export type DirNames = {
     ROOT_DIR: string;
     SRC_DIR: string;
     PAGES_DIR: string;
@@ -27,3 +27,4 @@ export default function grabDirNames(): {
     BUNX_ERROR_LOGS_DIR: string;
     BUNX_LOGS_DIR: string;
 };
+export default function grabDirNames(): DirNames;

dist/utils/grab-dir-names.js

@@ -1,5 +1,7 @@
 import path from "path";
 export default function grabDirNames() {
+    if (global.DIR_NAMES)
+        return global.DIR_NAMES;
     const ROOT_DIR = process.cwd();
     const SRC_DIR = path.join(ROOT_DIR, "src");
     const PAGES_DIR = path.join(SRC_DIR, "pages");

dist/utils/is-development.js

@@ -1,10 +1,6 @@
 export default function isDevelopment() {
-    const config = global.CONFIG;
+    if (process.env.NODE_ENV === "production") {
-    if (process.env.NODE_ENV == "production") {
         return false;
     }
-    if (config.development) {
+    return Boolean(global.CONFIG?.development);
-        return true;
-    }
-    return false;
 }

dist/utils/is-safe-path.d.ts

@@ -0,0 +1,4 @@
+export default function isSafePath({ filePath, allowedDir, }: {
+    filePath: string;
+    allowedDir: string;
+}): boolean;

dist/utils/is-safe-path.js

@@ -0,0 +1,15 @@
+import { realpathSync } from "fs";
+import path from "path";
+export default function isSafePath({ filePath, allowedDir, }) {
+    const resolved = path.resolve(filePath);
+    if (!resolved.startsWith(allowedDir + path.sep) && resolved !== allowedDir) {
+        return false;
+    }
+    try {
+        const real = realpathSync(resolved);
+        return (real.startsWith(allowedDir + path.sep) || real === allowedDir);
+    }
+    catch {
+        return false;
+    }
+}

src/commands/start/index.ts

@@ -3,6 +3,10 @@ import path from "path";
 import type { BunSpawnOptions } from "../../types";
 import writeErrorFile from "../../functions/write-error-file";
 
+let retries = 0;
+let timeout: any;
+const MAX_RETRIES = 5;
+
 export default function () {
     return new Command("start")
         .description("Start production server")
@@ -12,6 +16,13 @@ export default function () {
 }
 
 async function start() {
+    clearTimeout(timeout);
+
+    if (retries >= MAX_RETRIES) {
+        console.error(`Production server crashed ${MAX_RETRIES} times. Exiting.`);
+        process.exit(1);
+    }
+
     const dev_spawn_file = path.resolve(__dirname, "prod-spawn.ts");
 
     const spawn_options: BunSpawnOptions = {
@@ -28,6 +39,12 @@ async function start() {
 
     let dev_process = Bun.spawn(spawn_options);
 
+    retries++;
+
+    timeout = setTimeout(() => {
+        retries = 0;
+    }, 10000);
+
     const exited = await dev_process.exited;
 
     if (exited) {

src/functions/bundler/build-on-start-error-handler.ts

@@ -9,7 +9,7 @@ export default async function buildOnstartErrorHandler(params?: Params) {
     global.RECOMPILING = false;
     global.IS_SERVER_COMPONENT = false;
 
-    Promise.all([
+    await Promise.all([
         global.SSR_BUNDLER_CTX?.dispose(),
         global.BUNDLER_CTX?.dispose(),
     ]);

src/functions/bunext-init.ts

@@ -5,7 +5,7 @@ import type {
     PageFiles,
 } from "../types";
 import type { FileSystemRouter, Server } from "bun";
-import grabDirNames from "../utils/grab-dir-names";
+import grabDirNames, { type DirNames } from "../utils/grab-dir-names";
 import { type FSWatcher } from "fs";
 import init from "./init";
 import isDevelopment from "../utils/is-development";
@@ -43,7 +43,7 @@ declare global {
     var BUNDLER_CTX: BuildContext | undefined;
     var SSR_BUNDLER_CTX: BuildContext | undefined;
     // var API_ROUTES_BUNDLER_CTX: BuildContext | undefined;
-    var DIR_NAMES: ReturnType<typeof grabDirNames>;
+    var DIR_NAMES: DirNames;
     var REACT_IMPORTS_MAP: { imports: Record<string, string> };
     var REACT_DOM_SERVER: any;
     var REACT_DOM_MODULE_CACHE: Map<string, { main: any; css: string }>;

src/functions/cache/trim-all-cache.ts

@@ -17,6 +17,11 @@ export default async function trimAllCache() {
             const trim_key = await trimCacheKey({
                 key: cache_key,
             });
+
+            if (trim_key.success) {
+                cached_items.splice(i, 1);
+                i--;
+            }
         }
     } catch (error) {
         return undefined;

src/functions/server/bunext-req-handler.ts

@@ -8,6 +8,10 @@ import handleBunextPublicAssets from "./handle-bunext-public-assets";
 import checkExcludedPatterns from "../../utils/check-excluded-patterns";
 import { AppData } from "../../data/app-data";
 import fullRebuild from "./full-rebuild";
+
+const HMR_RETRY_COOLDOWN_MS = 5000;
+let lastHmrRetryTime = 0;
+
 type Params = {
     req: Request;
     server: Bun.Server<any>;
@@ -45,6 +49,11 @@ export default async function bunextRequestHandler({
         }
 
         if (is_dev && url.pathname == AppData["BunextHMRRetryRoute"]) {
+            const now = Date.now();
+            if (now - lastHmrRetryTime < HMR_RETRY_COOLDOWN_MS) {
+                return new Response("Too Many Requests", { status: 429 });
+            }
+            lastHmrRetryTime = now;
             await fullRebuild({ msg: `HMR Retry Rebuild ...` });
             return new Response("Modules Rebuilt");
         }
@@ -76,8 +85,12 @@ export default async function bunextRequestHandler({
 
         return response;
     } catch (error: any) {
+        if (is_dev) {
             return new Response(`Server Error: ${error.message}`, {
                 status: 500,
             });
         }
+        console.error(`Server Error: ${error.message}`, error);
+        return new Response("Internal Server Error", { status: 500 });
+    }
 }

src/functions/server/handle-bunext-public-assets.ts

@@ -2,6 +2,7 @@ import grabDirNames from "../../utils/grab-dir-names";
 import path from "path";
 import isDevelopment from "../../utils/is-development";
 import { readFileResponse } from "./handle-public";
+import isSafePath from "../../utils/is-safe-path";
 
 const { BUNEXT_PUBLIC_DIR } = grabDirNames();
 
@@ -19,7 +20,7 @@ export default async function ({ req }: Params): Promise<Response> {
             url.pathname.replace(/\/\.bunext\/public\//, ""),
         );
 
-        if (!file_path.startsWith(BUNEXT_PUBLIC_DIR + path.sep)) {
+        if (!isSafePath({ filePath: file_path, allowedDir: BUNEXT_PUBLIC_DIR })) {
             return new Response("Forbidden", { status: 403 });
         }
 

src/functions/server/handle-files.ts

@@ -2,6 +2,7 @@ import grabDirNames from "../../utils/grab-dir-names";
 import path from "path";
 import isDevelopment from "../../utils/is-development";
 import { existsSync } from "fs";
+import isSafePath from "../../utils/is-safe-path";
 
 const { PUBLIC_DIR } = grabDirNames();
 
@@ -15,7 +16,7 @@ export default async function ({ req }: Params): Promise<Response> {
         const url = new URL(req.url);
         const file_path = path.join(PUBLIC_DIR, url.pathname);
 
-        if (!file_path.startsWith(PUBLIC_DIR + path.sep)) {
+        if (!isSafePath({ filePath: file_path, allowedDir: PUBLIC_DIR })) {
             return new Response("Forbidden", { status: 403 });
         }
 
@@ -26,7 +27,13 @@ export default async function ({ req }: Params): Promise<Response> {
         }
 
         const file = Bun.file(file_path);
-        return new Response(file);
+        const headers = new Headers();
+
+        if (!is_dev) {
+            headers.set("Cache-Control", "public, max-age=3600");
+        }
+
+        return new Response(file, { headers });
     } catch (error) {
         return new Response(`File Not Found`, {
             status: 404,

src/functions/server/handle-hmr.ts

@@ -2,8 +2,28 @@ type Params = {
     req: Request;
 };
 
+function removeController(controller: ReadableStreamDefaultController<string>) {
+    const idx = global.HMR_CONTROLLERS.findIndex(
+        (c) => c.controller == controller,
+    );
+    if (typeof idx == "number" && idx >= 0) {
+        global.HMR_CONTROLLERS.splice(idx, 1);
+    }
+}
+
 export default async function ({ req }: Params): Promise<Response> {
-    const referer_url = new URL(req.headers.get("referer") || "");
+    const referer = req.headers.get("referer");
+    if (!referer) {
+        return new Response("Missing Referer Header", { status: 400 });
+    }
+
+    let referer_url: URL;
+    try {
+        referer_url = new URL(referer);
+    } catch {
+        return new Response("Invalid Referer Header", { status: 400 });
+    }
+
     const match = global.ROUTER.match(referer_url.pathname);
 
     const target_map = match?.filePath
@@ -25,21 +45,13 @@ export default async function ({ req }: Params): Promise<Response> {
                     c.enqueue(": keep-alive\n\n");
                 } catch {
                     clearInterval(heartbeat);
+                    removeController(controller);
                 }
             }, 5000);
         },
         cancel() {
             clearInterval(heartbeat);
-            const targetControllerIndex = global.HMR_CONTROLLERS.findIndex(
+            removeController(controller);
-                (c) => c.controller == controller,
-            );
-
-            if (
-                typeof targetControllerIndex == "number" &&
-                targetControllerIndex >= 0
-            ) {
-                global.HMR_CONTROLLERS.splice(targetControllerIndex, 1);
-            }
         },
     });
 

src/functions/server/handle-public.ts

@@ -2,6 +2,7 @@ import grabDirNames from "../../utils/grab-dir-names";
 import path from "path";
 import isDevelopment from "../../utils/is-development";
 import { existsSync } from "fs";
+import isSafePath from "../../utils/is-safe-path";
 
 const { PUBLIC_DIR } = grabDirNames();
 
@@ -19,7 +20,7 @@ export default async function ({ req }: Params): Promise<Response> {
             url.pathname.replace(/^\/public/, ""),
         );
 
-        if (!file_path.startsWith(PUBLIC_DIR + path.sep)) {
+        if (!isSafePath({ filePath: file_path, allowedDir: PUBLIC_DIR })) {
             return new Response("Forbidden", { status: 403 });
         }
 
@@ -53,6 +54,8 @@ export function readFileResponse({ file_path, cache }: FileResponse) {
         headers.set("Cache-Control", "public, max-age=31536000, immutable");
     } else if (cache?.duration) {
         headers.set("Cache-Control", `public, max-age=${cache.duration}`);
+    } else if (!isDevelopment()) {
+        headers.set("Cache-Control", "public, max-age=3600");
     }
 
     return new Response(file, {

src/functions/server/server-post-build-fn.ts

@@ -45,8 +45,8 @@ export default async function serverPostBuildFn(params?: Params) {
             continue;
         }
 
-        const mock_req = target_artifact.req
+        const mock_req = target_artifact.req_url
-            ? target_artifact.req.clone()
+            ? new Request(target_artifact.req_url)
             : new Request(controller.page_url);
 
         const page_component = global.IS_SERVER_COMPONENT

src/functions/server/web-pages/generate-web-html.tsx

@@ -178,6 +178,8 @@ export default async function genWebHTML({
     console.info = () => {};
     console.debug = () => {};
 
+    let htmlBody: string;
+    try {
         const stream = await renderToReadableStream(final_component, {
             onError(error: any) {
                 if (error.message.includes('unique "key" prop')) return;
@@ -185,9 +187,10 @@ export default async function genWebHTML({
             },
         });
 
-    const htmlBody = await new Response(stream).text();
+        htmlBody = await new Response(stream).text();
-
+    } finally {
         Object.assign(console, originalConsole);
+    }
 
     html += htmlBody;
 

src/functions/server/web-pages/grab-page-combined-server-res.ts

@@ -63,7 +63,7 @@ export default async function grabPageCombinedServerRes({
     const page_server_ctx = global.SSR_BUNDLER_CTX_MAP[server_file_path || ""];
     const final_page_server_path = page_server_ctx?.local_path
         ? path.join(ROOT_DIR, page_server_ctx.path)
-        : root_server_file_path;
+        : server_file_path;
 
     const server_module: BunextPageServerModule = final_page_server_path
         ? await import(`${final_page_server_path}?t=${now}`)

src/functions/server/web-pages/grab-page-component.tsx

@@ -117,7 +117,7 @@ export default async function grabPageComponent(
         }
 
         if (req && !is_hydration) {
-            global.BUNDLER_CTX_MAP[file_path].req = req;
+            global.BUNDLER_CTX_MAP[file_path].req_url = req.url;
         }
 
         if (debug) {

src/types/index.ts

@@ -344,7 +344,7 @@ export type BundlerCTXMap = {
     url_path: string;
     file_name: string;
     css_path?: string;
-    req?: Request;
+    req_url?: string;
 };
 
 export type GlobalHMRControllerObject = {

src/utils/grab-dir-names.ts

@@ -1,6 +1,38 @@
 import path from "path";
 
-export default function grabDirNames() {
+export type DirNames = {
+    ROOT_DIR: string;
+    SRC_DIR: string;
+    PAGES_DIR: string;
+    API_DIR: string;
+    PUBLIC_DIR: string;
+    HYDRATION_DST_DIR: string;
+    BUNX_CWD_DIR: string;
+    BUNX_ROOT_DIR: string;
+    CONFIG_FILE: string;
+    BUNX_TMP_DIR: string;
+    BUNX_HYDRATION_SRC_DIR: string;
+    BUNX_ROOT_SRC_DIR: string;
+    BUNX_ROOT_PRESETS_DIR: string;
+    BUNX_ROOT_500_PRESET_COMPONENT: string;
+    BUNX_ROOT_500_FILE_NAME: string;
+    BUNX_ROOT_404_PRESET_COMPONENT: string;
+    BUNX_ROOT_404_FILE_NAME: string;
+    HYDRATION_DST_DIR_MAP_JSON_FILE: string;
+    BUNEXT_CACHE_DIR: string;
+    BUNX_CWD_MODULE_CACHE_DIR: string;
+    BUNX_CWD_PAGES_REWRITE_DIR: string;
+    HYDRATION_DST_DIR_MAP_JSON_FILE_NAME: string;
+    BUNEXT_VENDOR_DIR: string;
+    BUNEXT_PUBLIC_DIR: string;
+    BUNX_BUNDLER_ERROR_EXIT_FILE: string;
+    BUNX_ERROR_LOGS_DIR: string;
+    BUNX_LOGS_DIR: string;
+};
+
+export default function grabDirNames(): DirNames {
+    if (global.DIR_NAMES) return global.DIR_NAMES;
+
     const ROOT_DIR = process.cwd();
     const SRC_DIR = path.join(ROOT_DIR, "src");
     const PAGES_DIR = path.join(SRC_DIR, "pages");

src/utils/is-development.ts

@@ -1,13 +1,7 @@
 export default function isDevelopment() {
-    const config = global.CONFIG;
+    if (process.env.NODE_ENV === "production") {
-
-    if (process.env.NODE_ENV == "production") {
         return false;
     }
 
-    if (config.development) {
+    return Boolean(global.CONFIG?.development);
-        return true;
-    }
-
-    return false;
 }

src/utils/is-safe-path.ts

@@ -0,0 +1,24 @@
+import { realpathSync } from "fs";
+import path from "path";
+
+export default function isSafePath({
+    filePath,
+    allowedDir,
+}: {
+    filePath: string;
+    allowedDir: string;
+}): boolean {
+    const resolved = path.resolve(filePath);
+    if (!resolved.startsWith(allowedDir + path.sep) && resolved !== allowedDir) {
+        return false;
+    }
+
+    try {
+        const real = realpathSync(resolved);
+        return (
+            real.startsWith(allowedDir + path.sep) || real === allowedDir
+        );
+    } catch {
+        return false;
+    }
+}