2020-10-12 21:18:28 +02:00
|
|
|
import { promises as fs, existsSync } from 'fs';
|
|
|
|
import * as ipc from 'node-ipc';
|
|
|
|
import * as path from 'path';
|
|
|
|
import * as util from 'util';
|
2020-10-16 18:51:28 +02:00
|
|
|
import { homedir } from 'os';
|
2020-10-12 21:18:28 +02:00
|
|
|
|
|
|
|
import { LogService } from 'jslib/abstractions/log.service';
|
2020-10-21 16:48:40 +02:00
|
|
|
import { ipcMain } from 'electron';
|
2020-10-12 21:18:28 +02:00
|
|
|
import { WindowMain } from 'jslib/electron/window.main';
|
|
|
|
|
|
|
|
export class NativeMessagingMain {
|
|
|
|
private connected = false;
|
|
|
|
|
|
|
|
constructor(private logService: LogService, private windowMain: WindowMain, private userPath: string, private appPath: string) {}
|
|
|
|
|
|
|
|
listen() {
|
|
|
|
ipc.config.id = 'bitwarden';
|
|
|
|
ipc.config.retry = 1500;
|
|
|
|
|
|
|
|
ipc.serve(() => {
|
|
|
|
ipc.server.on('message', (data: any, socket: any) => {
|
|
|
|
// This is a ugly hack until electron is updated 7.0.0 which supports ipcMain.invoke
|
|
|
|
this.windowMain.win.webContents.send('nativeMessaging', data);
|
|
|
|
ipcMain.once('nativeMessagingReply', (event, msg) => {
|
|
|
|
if (msg != null) {
|
|
|
|
this.send(msg, socket);
|
|
|
|
}
|
|
|
|
})
|
|
|
|
});
|
|
|
|
|
|
|
|
ipc.server.on('connect', () => {
|
|
|
|
this.connected = true;
|
|
|
|
})
|
|
|
|
|
|
|
|
ipc.server.on(
|
|
|
|
'socket.disconnected',
|
|
|
|
(socket: any, destroyedSocketID: any) => {
|
|
|
|
this.connected = false;
|
|
|
|
ipc.log(
|
|
|
|
'client ' + destroyedSocketID + ' has disconnected!'
|
|
|
|
);
|
|
|
|
}
|
|
|
|
);
|
|
|
|
});
|
|
|
|
|
|
|
|
ipc.server.start();
|
|
|
|
}
|
|
|
|
|
|
|
|
stop() {
|
|
|
|
ipc.server.stop();
|
|
|
|
}
|
|
|
|
|
|
|
|
send(message: object, socket: any) {
|
|
|
|
ipc.server.emit(socket, 'message', message);
|
|
|
|
}
|
|
|
|
|
|
|
|
generateManifests() {
|
|
|
|
const baseJson = {
|
|
|
|
'name': 'com.8bit.bitwarden',
|
|
|
|
'description': 'Bitwarden desktop <-> browser bridge',
|
2020-10-21 20:40:24 +02:00
|
|
|
'path': this.binaryPath(),
|
2020-10-12 21:18:28 +02:00
|
|
|
'type': 'stdio',
|
|
|
|
}
|
|
|
|
|
2020-10-12 21:26:26 +02:00
|
|
|
const firefoxJson = {...baseJson, ...{ 'allowed_extensions': ['{446900e4-71c2-419f-a6a7-df9c091e268b}']}}
|
2020-11-25 14:29:53 +01:00
|
|
|
const chromeJson = {...baseJson, ...{ 'allowed_origins': ['chrome-extension://nngceckbapebfimnlniiiahkandclblb/']}}
|
2020-10-12 21:18:28 +02:00
|
|
|
|
|
|
|
switch (process.platform) {
|
|
|
|
case 'win32':
|
2020-10-21 16:48:40 +02:00
|
|
|
const destination = path.join(this.userPath, 'browsers');
|
2020-10-12 21:18:28 +02:00
|
|
|
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':
|
2020-10-16 18:51:28 +02:00
|
|
|
if (existsSync(`${homedir()}/Library/Application\ Support/Mozilla/`)) {
|
2020-10-21 16:48:40 +02:00
|
|
|
this.writeManifest(`${homedir()}/Library/Application\ Support/Mozilla/NativeMessagingHosts/com.8bit.bitwarden.json`, firefoxJson)
|
2020-10-12 21:18:28 +02:00
|
|
|
}
|
|
|
|
|
2020-10-16 18:51:28 +02:00
|
|
|
if (existsSync(`${homedir()}/Library/Application\ Support/Google/Chrome`)) {
|
2020-10-21 16:48:40 +02:00
|
|
|
this.writeManifest(`${homedir()}/Library/Application\ Support/Google/Chrome/NativeMessagingHosts/com.8bit.bitwarden.json`, chromeJson)
|
2020-10-12 21:18:28 +02:00
|
|
|
}
|
|
|
|
break;
|
|
|
|
case 'linux':
|
2020-10-16 18:51:28 +02:00
|
|
|
if (existsSync(`${homedir()}/.mozilla/`)) {
|
2020-10-21 16:48:40 +02:00
|
|
|
this.writeManifest(`${homedir()}/.mozilla/native-messaging-hosts/com.8bit.bitwarden.json`, firefoxJson)
|
2020-10-12 21:18:28 +02:00
|
|
|
}
|
|
|
|
|
2020-10-16 18:51:28 +02:00
|
|
|
if (existsSync(`${homedir()}/.config/google-chrome/`)) {
|
2020-10-21 16:48:40 +02:00
|
|
|
this.writeManifest(`${homedir()}/.config/google-chrome/NativeMessagingHosts/com.8bit.bitwarden.json`, chromeJson)
|
2020-10-12 21:18:28 +02:00
|
|
|
}
|
|
|
|
break;
|
|
|
|
default:
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
removeManifests() {
|
|
|
|
switch (process.platform) {
|
|
|
|
case 'win32':
|
2020-10-21 16:48:40 +02:00
|
|
|
fs.unlink(path.join(this.userPath, 'browsers', 'firefox.json'));
|
|
|
|
fs.unlink(path.join(this.userPath, 'browsers', 'chrome.json'));
|
2020-10-12 21:18:28 +02:00
|
|
|
this.deleteWindowsRegistry('HKCU\\SOFTWARE\\Mozilla\\NativeMessagingHosts\\com.8bit.bitwarden');
|
|
|
|
this.deleteWindowsRegistry('HKCU\\SOFTWARE\\Google\\Chrome\\NativeMessagingHosts\\com.8bit.bitwarden');
|
|
|
|
break;
|
|
|
|
case 'darwin':
|
|
|
|
if (existsSync('~/Library/Application Support/Mozilla/NativeMessagingHosts/com.8bit.bitwarden.json')) {
|
2020-10-21 16:48:40 +02:00
|
|
|
fs.unlink('~/Library/Application Support/Mozilla/NativeMessagingHosts/com.8bit.bitwarden.json');
|
2020-10-12 21:18:28 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
if (existsSync('~/Library/Application Support/Google/Chrome/NativeMessagingHosts/com.8bit.bitwarden.json')) {
|
2020-10-21 16:48:40 +02:00
|
|
|
fs.unlink('~/Library/Application Support/Mozilla/NativeMessagingHosts/com.8bit.bitwarden.json');
|
2020-10-12 21:18:28 +02:00
|
|
|
}
|
|
|
|
break;
|
|
|
|
case 'linux':
|
|
|
|
if (existsSync('~/.mozilla/native-messaging-hosts/com.8bit.bitwarden.json')) {
|
2020-10-21 16:48:40 +02:00
|
|
|
fs.unlink('~/.mozilla/native-messaging-hosts/com.8bit.bitwarden.json');
|
2020-10-12 21:18:28 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
if (existsSync('~/.config/google-chrome/NativeMessagingHosts/com.8bit.bitwarden.json')) {
|
2020-10-21 16:48:40 +02:00
|
|
|
fs.unlink('~/.config/google-chrome/NativeMessagingHosts/com.8bit.bitwarden.json');
|
2020-10-12 21:18:28 +02:00
|
|
|
}
|
|
|
|
break;
|
|
|
|
default:
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
private writeManifest(destination: string, manifest: object) {
|
2020-10-21 16:48:40 +02:00
|
|
|
fs.mkdir(path.dirname(destination));
|
2020-10-12 21:18:28 +02:00
|
|
|
fs.writeFile(destination, JSON.stringify(manifest, null, 2)).catch(this.logService.error);
|
|
|
|
}
|
|
|
|
|
2020-10-21 20:40:24 +02:00
|
|
|
private binaryPath() {
|
2020-11-25 14:29:53 +01:00
|
|
|
const dir = path.join(this.appPath, '..');
|
2020-10-12 21:26:26 +02:00
|
|
|
if (process.platform === 'win32') {
|
2020-11-25 14:29:53 +01:00
|
|
|
return path.join(dir, 'native-messaging.bat');
|
2020-10-21 20:40:24 +02:00
|
|
|
} else if (process.platform === 'darwin') {
|
2020-11-25 14:29:53 +01:00
|
|
|
return path.join(dir, '..', 'MacOS', 'Bitwarden');
|
2020-10-12 21:18:28 +02:00
|
|
|
}
|
2020-10-12 21:26:26 +02:00
|
|
|
|
2020-11-25 14:29:53 +01:00
|
|
|
return path.join(dir, 'proxy');
|
2020-10-12 21:18:28 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
private async createWindowsRegistry(check: string, location: string, jsonFile: string) {
|
|
|
|
const regedit = require('regedit');
|
|
|
|
regedit.setExternalVBSLocation('resources/regedit/vbs');
|
|
|
|
|
|
|
|
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 {
|
|
|
|
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 = require('regedit');
|
|
|
|
|
|
|
|
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 {
|
|
|
|
// Do nothing
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|