Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
e2924d4
commit 841f816
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,178 +1,47 @@ | ||
| import { getStorageKeyValue } from "@/utils/helper.utils"; | ||
| import { recordBuckets } from "./lib/recorder"; | ||
|
|
||
| const requests = chrome.webRequest.onHeadersReceived; | ||
| const storage = chrome.storage.local; | ||
|
|
||
| // To make async function work, and also the ability to remove listener | ||
| const dummyFunc = (detail: chrome.webRequest.WebResponseHeadersDetails) => { | ||
| recordBuckets(detail); | ||
| }; | ||
|
|
||
| const listener = async (toRecord: boolean) => { | ||
| if (toRecord) | ||
| requests.addListener( | ||
| dummyFunc, | ||
| { | ||
| urls: ["<all_urls>"], | ||
| }, | ||
| ["responseHeaders"], | ||
| ); | ||
| else requests.removeListener(dummyFunc); | ||
| }; | ||
|
|
||
| (async () => { | ||
| const [{ buckets }, { record }] = (await getStorageKeyValue([ | ||
| "buckets", | ||
| "record", | ||
| ])) as Record<string, any>[]; | ||
| if (!buckets || !Object.keys(buckets).length) { | ||
| storage.set({ | ||
| buckets: { good: [], bad: [], error: [] }, | ||
| record: true, | ||
| lastSeen: { good: 0, bad: 0, error: 0 }, | ||
| }); | ||
| } | ||
| listener(record ?? true); | ||
| })(); | ||
|
|
||
| chrome.runtime.onMessage.addListener((toRecord: boolean) => { | ||
| listener(toRecord); | ||
| chrome.action.setBadgeText({ | ||
| text: toRecord ? "" : "!", | ||
| }); | ||
| chrome.action.setBadgeBackgroundColor({ | ||
| color: toRecord ? "green" : "orange", | ||
| }); | ||
| }); | ||
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,104 @@ | ||
| import * as cheerio from "cheerio"; | ||
| import { addNumber } from "@/utils/helper.utils"; | ||
| import { IBucketType } from "@/types"; | ||
|
|
||
| const storage = chrome.storage.local; | ||
| //const preRecorded = new Set(); | ||
|
|
||
| /** | ||
| * Gets the permissions from the offscreen page | ||
| * | ||
| * @param $ the cheerio object | ||
| */ | ||
| const getPerms = ($: cheerio.CheerioAPI, hostname: string) => { | ||
| const permissions = {} as Record<string, string[]>; | ||
| const hasUri = $("URI"); | ||
| const hasCode = $("Code"); | ||
| const hasListBucket = $("ListBucketResult"); | ||
| let type = ""; | ||
| const date = new Date().getTime(); | ||
| console.log("Date", date); | ||
| console.log("Hostname", hostname); | ||
|
|
||
| try { | ||
| if (!hasUri.length && !hasCode.length && hasListBucket) | ||
| throw new Error("No permissions"); | ||
| if (hasUri.length) { | ||
| hasUri.toArray().map((elem) => { | ||
| const title = $(elem).text().split("/").pop()!; | ||
| const perm = $(elem).parent().next().text(); | ||
| permissions[title] = [...(permissions[title] || []), perm]; | ||
| }); | ||
| type = "good"; | ||
| } | ||
| if (hasListBucket.length) { | ||
| permissions["ListBucket"] = ["True"]; | ||
| type = "good"; | ||
| } else if (hasCode.length) { | ||
| const elem = hasCode[0]; | ||
| const title = $(elem).text(); | ||
| const perm = $(elem).next().text(); | ||
| type = "bad"; | ||
| permissions[title] = [...(permissions[title] || []), perm]; | ||
| } | ||
| } catch (e) { | ||
| type = "error"; | ||
| console.log("Error in getPerms", e); | ||
| } | ||
| return { type, info: { permissions, date, hostname } }; | ||
| }; | ||
|
|
||
| /** | ||
| * Gets the bucket info from the offscreen page | ||
| * | ||
| * @param hostname the hostname of the bucket | ||
| */ | ||
| const getBucketInfo = async (hostname: string) => { | ||
| try { | ||
| // Combining results to have a single text to search | ||
| const text = await Promise.all([ | ||
| fetch("http://" + hostname + "/?acl").then((res) => res.text()), | ||
| fetch("http://" + hostname + "/").then((res) => res.text()), | ||
| ]).then(([aclText, listBucketText]) => aclText + listBucketText); | ||
| return getPerms(cheerio.load(text), hostname); | ||
| } catch (e) { | ||
| console.log(e); | ||
| return false; | ||
| } | ||
| }; | ||
|
|
||
| /** | ||
| * Records the buckets from the network requests | ||
| * | ||
| * @param response the response from the network request | ||
| */ | ||
| export const recordBuckets = async ( | ||
| response: chrome.webRequest.WebResponseHeadersDetails, | ||
| ) => { | ||
| const s3 = response.responseHeaders?.some( | ||
| (header) => header.name == "x-amz-request-id", | ||
| ); | ||
| if (!s3) return; | ||
| const { buckets }: { buckets: IBucketType } = await storage.get("buckets"); | ||
| const { hostname, pathname } = new URL(response.url); | ||
| // There are instances that the bucket name is a path | ||
| const bucketHostname = | ||
| hostname === "s3.amazonaws.com" | ||
| ? `${hostname}/${pathname.split("/")[1]}` | ||
| : hostname; | ||
| const isPresent = Object.values(buckets) | ||
| .flat() | ||
| .some((bucket) => bucket.hostname === bucketHostname); | ||
| // Double requests that might be recorded that has favicon.ico | ||
| const hasFavicon = bucketHostname.includes("favicon.ico"); | ||
| if (isPresent || hasFavicon) return; | ||
| const bucketInfo = await getBucketInfo(bucketHostname); | ||
| if (!bucketInfo) return; | ||
| storage.set({ | ||
| buckets: { | ||
| ...buckets, | ||
| [bucketInfo.type]: [bucketInfo.info, ...buckets[bucketInfo.type]], | ||
| }, | ||
| }); | ||
| bucketInfo.type === "good" && addNumber(buckets); | ||
| }; | ||
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1 +1 @@ | ||
| {"root":["./src/App.tsx","./src/background.ts","./src/main.tsx","./src/vite-env.d.ts","./src/components/Bucket.tsx","./src/components/CustomSwitch.tsx","./src/components/Menu.tsx","./src/components/TabBuckets.tsx","./src/components/TabButton.tsx","./src/components/ui/accordion.tsx","./src/components/ui/badge.tsx","./src/components/ui/dropdown-menu.tsx","./src/components/ui/separator.tsx","./src/components/ui/switch.tsx","./src/components/ui/tabs.tsx","./src/lib/recorder.ts","./src/lib/utils.ts","./src/store/useBuckets.store.ts","./src/store/useLastSeen.store.ts","./src/types/index.ts","./src/utils/helper.utils.ts"],"version":"5.6.2"} |