Get all cookies for current tab via CDP.

packages/extension/src/serviceWorker/attachCDP.ts

@@ -0,0 +1,32 @@
+/*
+ * Copyright 2023 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+/**
+ * This function will attach the debugger to the given target.
+ * @param {{ [key: string]: number | string }} target The target where debugger needs to be attached.
+ */
+export default async function attachCDP(target: {
+  [key: string]: number | string;
+}) {
+  await chrome.debugger.attach(target, '1.3');
+  await chrome.debugger.sendCommand(target, 'Target.setAutoAttach', {
+    autoAttach: false,
+    flatten: true,
+    waitForDebuggerOnStart: true,
+  });
+  await chrome.debugger.sendCommand(target, 'Network.enable');
+  await chrome.debugger.sendCommand(target, 'Audits.enable');
+  await chrome.debugger.sendCommand(target, 'Page.enable');
+}

packages/extension/src/serviceWorker/index.ts

@@ -35,6 +35,8 @@ import { getTab } from '../utils/getTab';
 import getQueryParams from '../utils/getQueryParams';
 import reloadCurrentTab from '../utils/reloadCurrentTab';
 import createCookieFromAuditsIssue from '../utils/createCookieFromAuditsIssue';
+import attachCDP from './attachCDP';
+import { getCurrentTabId } from '../utils/getCurrentTabId';
 
 let cookieDB: CookieDatabase | null = null;
 let syncCookieStore: SynchnorousCookieStore | undefined;
@@ -199,11 +201,7 @@ chrome.tabs.onUpdated.addListener(async (tabId, changeInfo, tab) => {
 
   try {
     if (globalIsUsingCDP) {
-      await chrome.debugger.attach({ tabId }, '1.3');
-      await chrome.debugger.sendCommand({ tabId }, 'Network.enable');
-      await chrome.debugger.sendCommand({ tabId }, 'Audits.enable');
-      await chrome.debugger.sendCommand({ tabId }, 'Page.enable');
-      await chrome.debugger.sendCommand({ tabId }, 'DOM.enable');
+      await attachCDP({ tabId });
     } else {
       await chrome.debugger.detach({ tabId });
     }
@@ -296,15 +294,29 @@ chrome.debugger.onEvent.addListener(async (source, method, params) => {
     return;
   }
 
-  let tabId = '';
+  let tabId = await getCurrentTabId();
+  try {
+    const targets = await chrome.debugger.getTargets();
+    await Promise.all(
+      targets.map(async ({ id }) => {
+        await chrome.debugger.attach({ targetId: id }, '1.3');
+        await chrome.debugger.sendCommand({ targetId: id }, 'Network.enable');
+        await chrome.debugger.sendCommand({ targetId: id }, 'Audits.enable');
+      })
+    );
+  } catch (error) {
+    //Fail silently since it gives only one kind of error. Debugger already attached to tabId.
+  }
 
-  if (!source?.tabId) {
-    return;
+  if (source?.tabId) {
+    tabId = source?.tabId?.toString();
   }
 
-  const url = syncCookieStore?.getTabUrl(source?.tabId);
+  if (!tabId) {
+    return;
+  }
 
-  tabId = source?.tabId?.toString();
+  const url = syncCookieStore?.getTabUrl(Number(tabId));
 
   if (tabMode && tabMode !== 'unlimited' && tabToRead !== tabId) {
     return;
@@ -359,10 +371,7 @@ chrome.debugger.onEvent.addListener(async (source, method, params) => {
       return;
     }
 
-    if (
-      requestIdToCDPURLMapping[tabId][requestParams?.requestId].frameId &&
-      requestIdToCDPURLMapping[tabId][requestParams?.requestId].url
-    ) {
+    if (requestIdToCDPURLMapping[tabId][requestParams?.requestId]) {
       const cookies: CookieData[] = parseRequestWillBeSentExtraInfo(
         requestParams,
         cookieDB ?? {},
@@ -411,7 +420,6 @@ chrome.debugger.onEvent.addListener(async (source, method, params) => {
         cookieDB ?? {},
         requestIdToCDPURLMapping[tabId][request.requestId]?.frameId
       );
-
       syncCookieStore?.update(Number(tabId), cookies);
 
       delete unParsedRequestHeaders[tabId][request.requestId];
@@ -429,6 +437,7 @@ chrome.debugger.onEvent.addListener(async (source, method, params) => {
       if (cookieObjectToUpdate) {
         syncCookieStore?.update(Number(tabId), [cookieObjectToUpdate]);
       }
+
       delete auditsIssueForTab[tabId][request.requestId];
     }
 

packages/extension/src/store/synchnorousCookieStore.ts

@@ -48,6 +48,9 @@ class SynchnorousCookieStore {
       devToolsOpenState: boolean;
       popupOpenState: boolean;
       newUpdates: number;
+      frameIdSet: {
+        [frameUrl: string]: string[];
+      };
     };
   } = {};
 
@@ -182,6 +185,35 @@ class SynchnorousCookieStore {
     return this.tabs[tabId].url;
   }
 
+  /**
+   * Gets the frameIDSet for the given tab id if tab exists.
+   * @param {number} tabId Tab id.
+   * @returns {string | null} The url of the tab if exists else null.
+   */
+  getFrameIDSet(tabId: number): { [frameUrl: string]: string[] } | null {
+    if (!this.tabs[tabId]) {
+      return null;
+    }
+
+    return this.tabs[tabId].frameIdSet;
+  }
+
+  /**
+   * Update FrameId set for a given url for a given tab.
+   * @param {number} tabId The url whose url needs to be update.
+   * @param {string} frameUrl The URL where frameId needs to be added.
+   * @param {string} frameIdToAdd The new frameId to be added.
+   */
+  updateFrameIdSet(tabId: number, frameUrl: string, frameIdToAdd: string) {
+    if (!this.tabs[tabId]) {
+      return;
+    } else {
+      this.tabs[tabId].frameIdSet[frameUrl] = [
+        ...new Set([...this.tabs[tabId].frameIdSet[frameUrl], frameIdToAdd]),
+      ];
+    }
+  }
+
   /**
    * Update tab url for given tab
    * @param {number} tabId The url whose url needs to be update.
@@ -304,6 +336,7 @@ class SynchnorousCookieStore {
       devToolsOpenState: false,
       popupOpenState: false,
       newUpdates: 0,
+      frameIdSet: {},
     };
   }