browser/apps/desktop/src/main/nativeMessaging.main.ts

import { existsSync, promises as fs } from "fs";
import { Socket } from "net";
import { homedir, userInfo } from "os";
import * as path from "path";
import * as util from "util";

import { ipcMain } from "electron";
import * as ipc from "node-ipc";

import { LogService } from "@bitwarden/common/abstractions/log.service";
import { WindowMain } from "@bitwarden/electron/window.main";

export class NativeMessagingMain {
  private connected: Socket[] = [];
  private socket: any;

  constructor(
    private logService: LogService,
    private windowMain: WindowMain,
    private userPath: string,
    private exePath: string
  ) {}

  async listen() {
    ipc.config.id = "bitwarden";
    ipc.config.retry = 1500;
    if (process.platform === "darwin") {
      if (!existsSync(`${homedir()}/tmp`)) {
        await fs.mkdir(`${homedir()}/tmp`);
      }
      ipc.config.socketRoot = `${homedir()}/tmp/`;
    }

    ipc.serve(() => {
      ipc.server.on("message", (data: any, socket: any) => {
        this.socket = socket;
        this.windowMain.win.webContents.send("nativeMessaging", data);
      });

      ipcMain.on("nativeMessagingReply", (event, msg) => {
        if (this.socket != null && msg != null) {
          this.send(msg, this.socket);
        }
      });

      ipc.server.on("connect", (socket: Socket) => {
        this.connected.push(socket);
      });

      ipc.server.on("socket.disconnected", (socket, destroyedSocketID) => {
        const index = this.connected.indexOf(socket);
        if (index > -1) {
          this.connected.splice(index, 1);
        }

        this.socket = null;
        ipc.log("client " + destroyedSocketID + " has disconnected!");
      });
    });

    ipc.server.start();
  }

  stop() {
    ipc.server.stop();
    // Kill all existing connections
    this.connected.forEach((socket) => {
      if (!socket.destroyed) {
        socket.destroy();
      }
    });
  }

  send(message: object, socket: any) {
    ipc.server.emit(socket, "message", message);
  }

  generateManifests() {
    const baseJson = {
      name: "com.8bit.bitwarden",
      description: "Bitwarden desktop <-> browser bridge",
      path: this.binaryPath(),
      type: "stdio",
    };

    const firefoxJson = {
      ...baseJson,
      ...{ allowed_extensions: ["{446900e4-71c2-419f-a6a7-df9c091e268b}"] },
    };
    const chromeJson = {
      ...baseJson,
      ...{
        allowed_origins: [
          "chrome-extension://nngceckbapebfimnlniiiahkandclblb/",
          "chrome-extension://jbkfoedolllekgbhcbcoahefnbanhhlh/",
          "chrome-extension://ccnckbpmaceehanjmeomladnmlffdjgn/",
        ],
      },
    };

    switch (process.platform) {
      case "win32": {
        const destination = path.join(this.userPath, "browsers");
        this.writeManifest(path.join(destination, "firefox.json"), firefoxJson);
        this.writeManifest(path.join(destination, "chrome.json"), chromeJson);

        this.createWindowsRegistry(
          "HKLM\\SOFTWARE\\Mozilla\\Firefox",
          "HKCU\\SOFTWARE\\Mozilla\\NativeMessagingHosts\\com.8bit.bitwarden",
          path.join(destination, "firefox.json")
        );
        this.createWindowsRegistry(
          "HKCU\\SOFTWARE\\Google\\Chrome",
          "HKCU\\SOFTWARE\\Google\\Chrome\\NativeMessagingHosts\\com.8bit.bitwarden",
          path.join(destination, "chrome.json")
        );
        break;
      }
      case "darwin": {
        const nmhs = this.getDarwinNMHS();
        for (const [key, value] of Object.entries(nmhs)) {
          if (existsSync(value)) {
            const p = path.join(value, "NativeMessagingHosts", "com.8bit.bitwarden.json");

            let manifest: any = chromeJson;
            if (key === "Firefox") {
              manifest = firefoxJson;
            }

            this.writeManifest(p, manifest).catch((e) =>
              this.logService.error(`Error writing manifest for ${key}. ${e}`)
            );
          } else {
            this.logService.warning(`${key} not found skipping.`);
          }
        }
        break;
      }
      case "linux":
        if (existsSync(`${this.homedir()}/.mozilla/`)) {
          this.writeManifest(
            `${this.homedir()}/.mozilla/native-messaging-hosts/com.8bit.bitwarden.json`,
            firefoxJson
          );
        }

        if (existsSync(`${this.homedir()}/.config/google-chrome/`)) {
          this.writeManifest(
            `${this.homedir()}/.config/google-chrome/NativeMessagingHosts/com.8bit.bitwarden.json`,
            chromeJson
          );
        }

        if (existsSync(`${this.homedir()}/.config/microsoft-edge/`)) {
          this.writeManifest(
            `${this.homedir()}/.config/microsoft-edge/NativeMessagingHosts/com.8bit.bitwarden.json`,
            chromeJson
          );
        }
        break;
      default:
        break;
    }
  }

  generateDdgManifests() {
    const manifest = {
      name: "com.8bit.bitwarden",
      description: "Bitwarden desktop <-> DuckDuckGo bridge",
      path: this.binaryPath(),
      type: "stdio",
    };
    switch (process.platform) {
      case "darwin": {
        /* eslint-disable-next-line no-useless-escape */
        const path = `${this.homedir()}/Library/Containers/com.duckduckgo.macos.browser/Data/Library/Application\ Support/NativeMessagingHosts/com.8bit.bitwarden.json`;
        this.writeManifest(path, manifest).catch((e) =>
          this.logService.error(`Error writing manifest for DuckDuckGo. ${e}`)
        );
        break;
      }
      default:
        break;
    }
  }

  removeManifests() {
    switch (process.platform) {
      case "win32":
        fs.unlink(path.join(this.userPath, "browsers", "firefox.json"));
        fs.unlink(path.join(this.userPath, "browsers", "chrome.json"));
        this.deleteWindowsRegistry(
          "HKCU\\SOFTWARE\\Mozilla\\NativeMessagingHosts\\com.8bit.bitwarden"
        );
        this.deleteWindowsRegistry(
          "HKCU\\SOFTWARE\\Google\\Chrome\\NativeMessagingHosts\\com.8bit.bitwarden"
        );
        break;
      case "darwin": {
        const nmhs = this.getDarwinNMHS();
        for (const [, value] of Object.entries(nmhs)) {
          const p = path.join(value, "NativeMessagingHosts", "com.8bit.bitwarden.json");
          if (existsSync(p)) {
            fs.unlink(p);
          }
        }
        break;
      }
      case "linux":
        if (
          existsSync(`${this.homedir()}/.mozilla/native-messaging-hosts/com.8bit.bitwarden.json`)
        ) {
          fs.unlink(`${this.homedir()}/.mozilla/native-messaging-hosts/com.8bit.bitwarden.json`);
        }

        if (
          existsSync(
            `${this.homedir()}/.config/google-chrome/NativeMessagingHosts/com.8bit.bitwarden.json`
          )
        ) {
          fs.unlink(
            `${this.homedir()}/.config/google-chrome/NativeMessagingHosts/com.8bit.bitwarden.json`
          );
        }

        if (
          existsSync(
            `${this.homedir()}/.config/microsoft-edge/NativeMessagingHosts/com.8bit.bitwarden.json`
          )
        ) {
          fs.unlink(
            `${this.homedir()}/.config/microsoft-edge/NativeMessagingHosts/com.8bit.bitwarden.json`
          );
        }
        break;
      default:
        break;
    }
  }

  removeDdgManifests() {
    switch (process.platform) {
      case "darwin": {
        /* eslint-disable-next-line no-useless-escape */
        const path = `${this.homedir()}/Library/Containers/com.duckduckgo.macos.browser/Data/Library/Application\ Support/NativeMessagingHosts/com.8bit.bitwarden.json`;
        if (existsSync(path)) {
          fs.unlink(path);
        }
        break;
      }
      default:
        break;
    }
  }

  private getDarwinNMHS() {
    /* eslint-disable no-useless-escape */
    return {
      Firefox: `${this.homedir()}/Library/Application\ Support/Mozilla/`,
      Chrome: `${this.homedir()}/Library/Application\ Support/Google/Chrome/`,
      "Chrome Beta": `${this.homedir()}/Library/Application\ Support/Google/Chrome\ Beta/`,
      "Chrome Dev": `${this.homedir()}/Library/Application\ Support/Google/Chrome\ Dev/`,
      "Chrome Canary": `${this.homedir()}/Library/Application\ Support/Google/Chrome\ Canary/`,
      Chromium: `${this.homedir()}/Library/Application\ Support/Chromium/`,
      "Microsoft Edge": `${this.homedir()}/Library/Application\ Support/Microsoft\ Edge/`,
      "Microsoft Edge Beta": `${this.homedir()}/Library/Application\ Support/Microsoft\ Edge\ Beta/`,
      "Microsoft Edge Dev": `${this.homedir()}/Library/Application\ Support/Microsoft\ Edge\ Dev/`,
      "Microsoft Edge Canary": `${this.homedir()}/Library/Application\ Support/Microsoft\ Edge\ Canary/`,
      Vivaldi: `${this.homedir()}/Library/Application\ Support/Vivaldi/`,
    };
    /* eslint-enable no-useless-escape */
  }

  private async writeManifest(destination: string, manifest: object) {
    if (!existsSync(path.dirname(destination))) {
      await fs.mkdir(path.dirname(destination));
    }
    fs.writeFile(destination, JSON.stringify(manifest, null, 2)).catch(this.logService.error);
  }

  private binaryPath() {
    if (process.platform === "win32") {
      return path.join(path.dirname(this.exePath), "resources", "native-messaging.bat");
    }

    return this.exePath;
  }

  private getRegeditInstance() {
    // eslint-disable-next-line
    const regedit = require("regedit");
    regedit.setExternalVBSLocation(path.join(path.dirname(this.exePath), "resources/regedit/vbs"));

    return regedit;
  }

  private async createWindowsRegistry(check: string, location: string, jsonFile: string) {
    const regedit = this.getRegeditInstance();

    const list = util.promisify(regedit.list);
    const createKey = util.promisify(regedit.createKey);
    const putValue = util.promisify(regedit.putValue);

    this.logService.debug(`Adding registry: ${location}`);

    // Check installed
    try {
      await list(check);
    } catch {
      this.logService.warning(`Not finding registry ${check} skipping.`);
      return;
    }

    try {
      await createKey(location);

      // Insert path to manifest
      const obj: any = {};
      obj[location] = {
        default: {
          value: jsonFile,
          type: "REG_DEFAULT",
        },
      };

      return putValue(obj);
    } catch (error) {
      this.logService.error(error);
    }
  }

  private async deleteWindowsRegistry(key: string) {
    const regedit = this.getRegeditInstance();

    const list = util.promisify(regedit.list);
    const deleteKey = util.promisify(regedit.deleteKey);

    this.logService.debug(`Removing registry: ${key}`);

    try {
      await list(key);
      await deleteKey(key);
    } catch {
      this.logService.error(`Unable to delete registry key: ${key}`);
    }
  }

  private homedir() {
    if (process.platform === "darwin") {
      return userInfo().homedir;
    } else {
      return homedir();
    }
  }
}