fix: Only allow URLs used by the website when forwarding to onshape API

aklinker1 · Sep 2, 2024 · 64fc6c6 · 64fc6c6
1 parent 55b5afd
commit 64fc6c6
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/web/server/api/onshape/[...path].ts b/web/server/api/onshape/[...path].ts
@@ -1,10 +1,24 @@
 import { FetchError } from 'ofetch';
 import useExtendedNitroApp from '~/server/composables/useExtendedNitroApp';
 
+const ALLOWED_URL_REGEX = [
+  /assemblies\/d\/[0-9a-z]+\/w\/[0-9a-z]+\/e\/[0-9a-z]+\/bom\?indented=false/,
+  /documents\/[0-9a-z]+/,
+  /parts\/d\/[0-9a-z]+\/[vw]\/[0-9a-z]+\/e\/[0-9a-z]+\/partid\/[a-zA-Z]{3}\/boundingboxes\?configuration=.*/,
+];
+
 export default defineEventHandler(async (event) => {
   const { onshape } = useExtendedNitroApp();
   const url = event.node.req.originalUrl!.replace('/api/onshape/', '');
 
+  if (!ALLOWED_URL_REGEX.find((regex) => regex.test(url))) {
+    setResponseStatus(event, 400);
+    return {
+      message: 'URL is not allowed',
+      url,
+    };
+  }
+
   try {
     return await onshape.fetch(url);
   } catch (err) {