Force host validation

This commit is contained in:
shamoon 2025-01-27 15:54:41 -08:00
parent ee7be33fa1
commit 075a3d4da6
2 changed files with 29 additions and 5 deletions

View File

@ -3,11 +3,16 @@ import { NextResponse } from "next/server";
export function middleware(req) { export function middleware(req) {
// Check the Host header, if HOMEPAGE_ALLOWED_HOSTS is set // Check the Host header, if HOMEPAGE_ALLOWED_HOSTS is set
const host = req.headers.get("host"); const host = req.headers.get("host");
const allowedHosts = process.env.HOMEPAGE_ALLOWED_HOSTS let allowedHosts = ["localhost:3000"];
? process.env.HOMEPAGE_ALLOWED_HOSTS.split(",").concat(["localhost:3000"]) if (process.env.HOMEPAGE_ALLOWED_HOSTS) {
: []; allowedHosts = allowedHosts.concat(process.env.HOMEPAGE_ALLOWED_HOSTS.split(","));
if (allowedHosts.length && !(host || allowedHosts.includes(host))) { }
return new NextResponse("Invalid Host header", { status: 400 }); if (!host || !allowedHosts.includes(host)) {
// eslint-disable-next-line no-console
console.error(
`Host validation failed for: ${host}. Hint: Set HOMEPAGE_ALLOWED_HOSTS to allow requests from this host.`,
);
return NextResponse.json({ error: "Host validation failed. See logs for more details." }, { status: 400 });
} }
return NextResponse.next(); return NextResponse.next();
} }

View File

@ -86,6 +86,7 @@ function Index({ initialSettings, fallback }) {
const windowFocused = useWindowFocus(); const windowFocused = useWindowFocus();
const [stale, setStale] = useState(false); const [stale, setStale] = useState(false);
const { data: errorsData } = useSWR("/api/validate"); const { data: errorsData } = useSWR("/api/validate");
const { error: validateError } = errorsData || {};
const { data: hashData, mutate: mutateHash } = useSWR("/api/hash"); const { data: hashData, mutate: mutateHash } = useSWR("/api/hash");
useEffect(() => { useEffect(() => {
@ -117,6 +118,24 @@ function Index({ initialSettings, fallback }) {
} }
}, [hashData]); }, [hashData]);
if (validateError) {
return (
<div className="w-full h-screen container m-auto justify-center p-10 pointer-events-none">
<div className="flex flex-col">
<div className="basis-1/2 bg-theme-500 dark:bg-theme-600 text-theme-600 dark:text-theme-300 m-2 rounded-md font-mono shadow-md border-4 border-transparent">
<div className="bg-rose-200 text-rose-800 dark:text-rose-200 dark:bg-rose-800 p-2 rounded-md font-bold">
<BiError className="float-right w-6 h-6" />
Error
</div>
<div className="p-2 text-theme-100 dark:text-theme-200">
<pre className="opacity-50 font-bold pb-2">{validateError}</pre>
</div>
</div>
</div>
</div>
);
}
if (stale) { if (stale) {
return ( return (
<div className="flex items-center justify-center h-screen"> <div className="flex items-center justify-center h-screen">