tor push

packages/flutter_mozilla_components/javascript/container_proxy/src/background/BackgroundMain.ts

@@ -0,0 +1,114 @@
+import { Store } from '../store/Store'
+import { HttpProxySettings, HttpsProxySettings, ProxySettings } from '../domain/ProxySettings'
+import { ProxyInfo, Socks5ProxyInfo } from '../domain/ProxyInfo'
+import { ProxyType } from '../domain/ProxyType'
+import BlockingResponse = browser.webRequest.BlockingResponse
+import _OnAuthRequiredDetails = browser.webRequest._OnAuthRequiredDetails
+import _OnRequestDetails = browser.proxy._OnRequestDetails
+
+const localhosts = new Set(['localhost', '127.0.0.1', '[::1]'])
+
+const containerIdentifier = 'firefox-container-'
+
+type DoNotProxy = never[]
+export const doNotProxy: DoNotProxy = []
+
+const emergencyBreak: Socks5ProxyInfo = {
+  type: ProxyType.Socks5,
+  host: 'emergency-break-proxy.localhost',
+  port: 1,
+  failoverTimeout: 1,
+  username: 'nonexistent user',
+  password: 'dummy password',
+  proxyDNS: true
+}
+
+export default class BackgroundMain {
+  store: Store
+
+  constructor({ store }: { store: Store }) {
+    this.store = store
+  }
+
+  initializeAuthListener(cookieStoreId: string, proxy: HttpProxySettings | HttpsProxySettings): void {
+    const listener: (details: _OnAuthRequiredDetails) => BlockingResponse = (details) => {
+      if (!details.isProxy) return {}
+
+      if (details.cookieStoreId !== cookieStoreId) return {}
+
+      // TODO: Fix in @types/firefox-webext-browser
+      // @ts-expect-error
+      const info = details.proxyInfo
+      if (info.host !== proxy.host || info.port !== proxy.port || info.type !== proxy.type) return {}
+
+      const result = { authCredentials: { username: proxy.username, password: proxy.password } }
+
+      browser.webRequest.onAuthRequired.removeListener(listener)
+
+      return result
+    }
+
+    browser.webRequest.onAuthRequired.addListener(
+      listener,
+      { urls: ['<all_urls>'] },
+      ['blocking']
+    )
+  }
+
+  // TODO: Fix in @types/firefox-webext-browser
+  async onRequest(requestDetails: Pick<_OnRequestDetails, 'cookieStoreId' | 'url' | 'tabId'>): Promise<DoNotProxy | ProxyInfo[]> {
+    const tab = (await browser.tabs.get(requestDetails.tabId))
+
+    if (tab.cookieStoreId?.startsWith(containerIdentifier) === true) {
+      try {
+        const cookieStoreId = tab.cookieStoreId.substring(containerIdentifier.length)
+
+        const proxies = await this.store.getProxiesForContainer(cookieStoreId)
+
+        if (proxies.length > 0) {
+          proxies.forEach(p => {
+            if (p.type === ProxyType.Http || p.type === ProxyType.Https) {
+              this.initializeAuthListener(cookieStoreId, p)
+            }
+          })
+
+          const result: ProxyInfo[] = proxies.filter((p: ProxySettings) => {
+            try {
+              const documentUrl = new URL(requestDetails.url)
+              const isLocalhost = localhosts.has(documentUrl.hostname)
+              if (isLocalhost && p.doNotProxyLocal) {
+                return false
+              }
+            } catch (e) {
+              console.error(e)
+            }
+
+            return true
+          }).map(p => p.asProxyInfo())
+
+          if (result.length === 0) {
+            return [emergencyBreak]
+          }
+          return result
+        }
+
+        return [emergencyBreak]
+      } catch (e: unknown) {
+        console.error(`Error in onRequest listener: ${e as string}`)
+        return [emergencyBreak]
+      }
+    }
+
+    return doNotProxy
+  }
+
+  run(browser: { proxy: any }): void {
+    const filter = { urls: ['<all_urls>'] }
+
+    browser.proxy.onRequest.addListener(this.onRequest.bind(this), filter)
+
+    browser.proxy.onError.addListener((e: Error) => {
+      console.error('Proxy error', e)
+    })
+  }
+}

packages/flutter_mozilla_components/javascript/container_proxy/src/background/index.ts

@@ -0,0 +1,44 @@
+import { Socks5ProxySettings } from 'src/domain/ProxySettings';
+import { Store } from '../store/Store'
+import BackgroundMain from './BackgroundMain'
+
+console.log('Background script started')
+
+const store = new Store()
+
+interface Message {
+    action: 'setProxyPort' | 'addContainerProxy' | 'removeContainerProxy';
+    args: any;
+}
+
+const port = browser.runtime.connectNative("containerProxy");
+port.onMessage.addListener((raw: unknown): void => {
+    const message = raw as Message;
+    switch (message.action) {
+        case "setProxyPort":
+            store.putProxy(new Socks5ProxySettings({
+                id: 'tor',
+                type: 'socks',
+                host: '127.0.0.1',
+                port: message.args,
+                doNotProxyLocal: true,
+                title: 'Tor',
+                proxyDNS: true,
+            }))
+            console.log('put tor port ' + message.args)
+
+            break
+        case "addContainerProxy":
+            store.setContainerProxyRelation(message.args, "tor")
+            console.log('added container relation ' + message.args)
+            break
+        case "removeContainerProxy":
+            store.removeContainerProxyRelation(message.args, "tor")
+            break
+
+
+    }
+});
+
+const backgroundListener = new BackgroundMain({ store })
+backgroundListener.run(browser)