HackBrowserData/core/browser.go

package core

import (
	"errors"
	"strings"

	"hack-browser-data/core/common"
	"hack-browser-data/log"
	"hack-browser-data/utils"
)

const (
	chromeName    = "Chrome"
	edgeName      = "Microsoft Edge"
	firefoxName   = "Firefox"
	speed360Name  = "360speed"
	qqBrowserName = "qq"
)

type Browser interface {
	InitSecretKey() error
	GetName() string
	GetSecretKey() []byte
	GetAllItems(itemName string) ([]common.Item, error)
}

const (
	cookie   = "cookie"
	history  = "history"
	bookmark = "bookmark"
	password = "password"
)

var (
	errItemNotSupported    = errors.New(`item not supported, default is "all", choose from history|password|bookmark|cookie`)
	errBrowserNotSupported = errors.New("browser not supported")
	errChromeSecretIsEmpty = errors.New("chrome secret is empty")
	errDbusSecretIsEmpty   = errors.New("dbus secret key is empty")
)

var (
	chromiumItems = map[string]struct {
		mainFile string
		newItem  func(mainFile, subFile string) common.Item
	}{
		bookmark: {
			mainFile: common.ChromeBookmarkFile,
			newItem:  common.NewBookmarks,
		},
		cookie: {
			mainFile: common.ChromeCookieFile,
			newItem:  common.NewCookies,
		},
		history: {
			mainFile: common.ChromeHistoryFile,
			newItem:  common.NewHistoryData,
		},
		password: {
			mainFile: common.ChromePasswordFile,
			newItem:  common.NewCPasswords,
		},
	}
	firefoxItems = map[string]struct {
		mainFile string
		subFile  string
		newItem  func(mainFile, subFile string) common.Item
	}{
		bookmark: {
			mainFile: common.FirefoxDataFile,
			newItem:  common.NewBookmarks,
		},
		cookie: {
			mainFile: common.FirefoxCookieFile,
			newItem:  common.NewCookies,
		},
		history: {
			mainFile: common.FirefoxDataFile,
			newItem:  common.NewHistoryData,
		},
		password: {
			mainFile: common.FirefoxKey4File,
			subFile:  common.FirefoxLoginFile,
			newItem:  common.NewFPasswords,
		},
	}
)

type Chromium struct {
	name        string
	profilePath string
	keyPath     string
	secretKey   []byte
}

func NewChromium(profile, key, name string) (Browser, error) {
	return &Chromium{profilePath: profile, keyPath: key, name: name}, nil
}

func (c *Chromium) GetName() string {
	return c.name
}

func (c *Chromium) GetSecretKey() []byte {
	return c.secretKey
}

func (c *Chromium) GetAllItems(itemName string) (Items []common.Item, err error) {
	itemName = strings.ToLower(itemName)
	var items []common.Item
	if itemName == "all" {
		for item, choice := range chromiumItems {
			m, err := utils.GetItemPath(c.profilePath, choice.mainFile)
			if err != nil {
				log.Errorf("%s find %s file failed, ERR:%s", c.name, item, err)
				continue
			}
			i := choice.newItem(m, "")
			log.Debugf("%s find %s File Success", c.name, item)
			items = append(items, i)
		}
	} else if item, ok := chromiumItems[itemName]; ok {
		m, err := utils.GetItemPath(c.profilePath, item.mainFile)
		if err != nil {
			log.Errorf("%s find %s file failed, ERR:%s", c.name, item.mainFile, err)
		}
		i := item.newItem(m, "")
		items = append(items, i)
		return items, nil
	} else {
		return nil, errItemNotSupported
	}
	return items, nil
}

type Firefox struct {
	name        string
	profilePath string
	keyPath     string
}

func NewFirefox(profile, key, name string) (Browser, error) {
	return &Firefox{profilePath: profile, keyPath: key, name: name}, nil
}

func (f *Firefox) GetAllItems(itemName string) ([]common.Item, error) {
	itemName = strings.ToLower(itemName)
	var items []common.Item
	if itemName == "all" {
		for item, choice := range firefoxItems {
			var (
				sub, main string
				err       error
			)
			if choice.subFile != "" {
				sub, err = utils.GetItemPath(f.profilePath, choice.subFile)
				if err != nil {
					log.Errorf("%s find %s file failed, ERR:%s", f.name, item, err)
					continue
				}
			}
			main, err = utils.GetItemPath(f.profilePath, choice.mainFile)
			if err != nil {
				log.Errorf("%s find %s file failed, ERR:%s", f.name, item, err)
				continue
			}
			i := choice.newItem(main, sub)
			log.Debugf("%s find %s file success", f.name, item)
			items = append(items, i)
		}
	} else if item, ok := firefoxItems[itemName]; ok {
		var (
			sub, main string
			err       error
		)
		if item.subFile != "" {
			sub, err = utils.GetItemPath(f.profilePath, item.subFile)
			if err != nil {
				log.Errorf("%s find %s file failed, ERR:%s", f.name, item.subFile, err)
			}
		}
		main, err = utils.GetItemPath(f.profilePath, item.mainFile)
		if err != nil {
			log.Errorf("%s find %s file failed, ERR:%s", f.name, item.mainFile, err)
		}
		i := item.newItem(main, sub)
		log.Debugf("%s find %s file success", f.name, item.mainFile)
		items = append(items, i)
		return items, nil
	} else {
		return nil, errItemNotSupported
	}
	return items, nil
}

func (f *Firefox) GetName() string {
	return f.name
}

func (f *Firefox) GetSecretKey() []byte {
	return nil
}

func (f *Firefox) InitSecretKey() error {
	return nil
}

func PickBrowser(name string) ([]Browser, error) {
	var browsers []Browser
	name = strings.ToLower(name)
	if name == "all" {
		for _, v := range browserList {
			b, err := v.New(v.ProfilePath, v.KeyPath, v.Name)
			if err != nil {
				log.Error(err)
			}
			browsers = append(browsers, b)
		}
		return browsers, nil
	} else if choice, ok := browserList[name]; ok {
		b, err := choice.New(choice.ProfilePath, choice.KeyPath, choice.Name)
		browsers = append(browsers, b)
		return browsers, err
	}
	return nil, errBrowserNotSupported
}

func ListBrowser() []string {
	var l []string
	for k := range browserList {
		l = append(l, k)
	}
	return l
}

func ListItem() []string {
	var l []string
	for k := range chromiumItems {
		l = append(l, k)
	}
	return l
}