Merge branch 'main' of https://github.com/remorses/playwriter

playwriter/src/cdp-relay.ts

@@ -1,4 +1,5 @@
 import { Hono } from 'hono'
+import { cors } from 'hono/cors'
 import { serve } from '@hono/node-server'
 import { getConnInfo } from '@hono/node-server/conninfo'
 import { createNodeWebSocket } from '@hono/node-ws'
@@ -418,6 +419,22 @@ export async function startPlayWriterCDPRelayServer({ port = 19988, host = '127.
   }
 
   const app = new Hono()
+  // CORS middleware for HTTP endpoints - only allows our specific extension IDs.
+  // This prevents other extensions from reading responses via fetch/XHR.
+  // WebSocket connections have their own separate origin validation.
+  app.use('*', cors({
+    origin: (origin) => {
+      if (!origin.startsWith('chrome-extension://')) {
+        return null
+      }
+      const extensionId = origin.replace('chrome-extension://', '')
+      if (!OUR_EXTENSION_IDS.includes(extensionId)) {
+        return null
+      }
+      return origin
+    },
+    allowMethods: ['GET', 'POST', 'HEAD', 'OPTIONS'],
+  }))
   const { injectWebSocket, upgradeWebSocket } = createNodeWebSocket({ app })
 
   const getCdpWsUrl = (c: { req: { header: (name: string) => string | undefined } }) => {
@@ -962,8 +979,8 @@ export async function startPlayWriterCDPRelayServer({ port = 19988, host = '127.
         // If this is an old connection closing after we've already established a new one,
         // don't clear the global state
         if (extensionWs && extensionWs !== ws) {
-           logger?.log('Old extension connection closed, keeping new one active')
+          logger?.log('Old extension connection closed, keeping new one active')
-           return
+          return
         }
 
         for (const pending of extensionPendingRequests.values()) {