131 lines
3.4 KiB
Go
131 lines
3.4 KiB
Go
package browserbiometrics
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
|
|
"github.com/quexten/goldwarden/agent/config"
|
|
"github.com/quexten/goldwarden/browserbiometrics/logging"
|
|
)
|
|
|
|
var chromiumPaths = []string{
|
|
"~/.config/google-chrome/",
|
|
"~/.config/google-chrome-beta/",
|
|
"~/.config/google-chrome-unstable/",
|
|
"~/.config/chromium/",
|
|
"~/.config/BraveSoftware/Brave-Browser/",
|
|
"~/.config/thorium/",
|
|
"~/.config/microsoft-edge-beta/",
|
|
"~/.config/microsoft-edge-dev/",
|
|
}
|
|
var mozillaPaths = []string{"~/.mozilla/", "~/.librewolf/", "~/.waterfox/"}
|
|
|
|
const appID = "com.quexten.bw-bio-handler"
|
|
|
|
var transportKey []byte
|
|
|
|
func Main(rtCfg *config.RuntimeConfig) {
|
|
logging.Debugf("Starting browserbiometrics")
|
|
transportKey = generateTransportKey()
|
|
logging.Debugf("Generated transport key")
|
|
|
|
setupCommunication()
|
|
readLoop(rtCfg)
|
|
}
|
|
|
|
func DetectAndInstallBrowsers() error {
|
|
var err error
|
|
|
|
// first, ensure the native messaging hosts dirs exist
|
|
for _, path := range chromiumPaths {
|
|
path = strings.ReplaceAll(path, "~", os.Getenv("HOME"))
|
|
_, err = os.Stat(path)
|
|
if err != nil {
|
|
continue
|
|
}
|
|
|
|
_, err = os.Stat(path + "NativeMessagingHosts/")
|
|
if err == nil {
|
|
fmt.Println("Native messaging host directory already exists: " + path + "NativeMessagingHosts/")
|
|
continue
|
|
}
|
|
err = os.MkdirAll(path+"NativeMessagingHosts/", 0755)
|
|
if err != nil {
|
|
fmt.Println("Error creating native messaging host directory: " + err.Error())
|
|
} else {
|
|
fmt.Println("Created native messaging host directory: " + path + "NativeMessagingHosts/")
|
|
}
|
|
}
|
|
for _, path := range mozillaPaths {
|
|
path = strings.ReplaceAll(path, "~", os.Getenv("HOME"))
|
|
_, err = os.Stat(path)
|
|
if err != nil {
|
|
continue
|
|
}
|
|
|
|
_, err = os.Stat(path + "native-messaging-hosts/")
|
|
if err == nil {
|
|
fmt.Println("Native messaging host directory already exists: " + path + "native-messaging-hosts/")
|
|
continue
|
|
}
|
|
err = os.MkdirAll(path+"native-messaging-hosts/", 0755)
|
|
if err != nil {
|
|
fmt.Println("Error creating native messaging host directory: " + err.Error())
|
|
} else {
|
|
fmt.Println("Created native messaging host directory: " + path + "native-messaging-hosts/")
|
|
}
|
|
}
|
|
|
|
err = detectAndInstallBrowsers(".config")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
for _, path := range mozillaPaths {
|
|
path = strings.ReplaceAll(path, "~/", "")
|
|
err = detectAndInstallBrowsers(path)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func detectAndInstallBrowsers(startPath string) error {
|
|
home := os.Getenv("HOME")
|
|
err := filepath.Walk(home+"/"+startPath, func(path string, info os.FileInfo, err error) error {
|
|
if err != nil {
|
|
return nil
|
|
}
|
|
var tempPath string
|
|
if !strings.HasPrefix(path, home) {
|
|
return nil
|
|
} else {
|
|
tempPath = strings.TrimPrefix(path, home)
|
|
}
|
|
if strings.Count(tempPath, "/") > 3 {
|
|
return nil
|
|
}
|
|
|
|
binPath, err := os.Executable()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if info.IsDir() && info.Name() == "native-messaging-hosts" {
|
|
fmt.Printf("Found mozilla-like browser: %s\n", path)
|
|
manifest := strings.Replace(templateMozilla, "PATH", binPath, 1)
|
|
err = os.WriteFile(path+"/com.8bit.bitwarden.json", []byte(manifest), 0644)
|
|
} else if info.IsDir() && info.Name() == "NativeMessagingHosts" {
|
|
fmt.Printf("Found chrome-like browser: %s\n", path)
|
|
manifest := strings.Replace(templateChrome, "PATH", binPath, 1)
|
|
err = os.WriteFile(path+"/com.8bit.bitwarden.json", []byte(manifest), 0644)
|
|
}
|
|
|
|
return err
|
|
})
|
|
|
|
return err
|
|
}
|