feat: remove export item selection, default select all items

pull/83/head
ᴍᴏᴏɴD4ʀᴋ 4 years ago
parent 7c54a282c0
commit 84d02b1833
  1. 13
      cmd/cmd.go
  2. 37
      core/browser.go
  3. 17
      core/common/parse.go

@ -12,7 +12,6 @@ import (
var (
browser string
exportData string
exportDir string
outputFormat string
verbose bool
@ -23,18 +22,16 @@ func Execute() {
app := &cli.App{
Name: "hack-browser-data",
Usage: "Export passwords/cookies/history/bookmarks from browser",
UsageText: "[hack-browser-data -b chrome -f json -dir results -e all -cc]\n Get all data(password/cookie/history/bookmark) from chrome",
Version: "0.2.2",
UsageText: "[hack-browser-data -b chrome -f json -dir results -cc]\n Get all data(password/cookie/history/bookmark) from chrome",
Version: "0.2.3",
Flags: []cli.Flag{
&cli.BoolFlag{Name: "verbose", Aliases: []string{"vv"}, Destination: &verbose, Value: false, Usage: "Verbose"},
&cli.BoolFlag{Name: "compress", Aliases: []string{"cc"}, Destination: &compress, Value: false, Usage: "Compress result to zip"},
&cli.StringFlag{Name: "browser", Aliases: []string{"b"}, Destination: &browser, Value: "all", Usage: "Available browsers: all|" + strings.Join(core.ListBrowser(), "|")},
&cli.StringFlag{Name: "results-dir", Aliases: []string{"dir"}, Destination: &exportDir, Value: "results", Usage: "Export dir"},
&cli.StringFlag{Name: "format", Aliases: []string{"f"}, Destination: &outputFormat, Value: "json", Usage: "Format, csv|json|console"},
&cli.StringFlag{Name: "export-data", Aliases: []string{"e"}, Destination: &exportData, Value: "all", Usage: "all|" + strings.Join(core.ListItem(), "|")},
},
HideHelpCommand: true,
HideVersion: true,
Action: func(c *cli.Context) error {
if verbose {
log.InitLog("debug")
@ -55,14 +52,16 @@ func Execute() {
if err != nil {
log.Error(err)
}
items, err := browser.GetAllItems(exportData)
// default select all items
// also you can get single item with browser.GetItem(itemName)
items, err := browser.GetAllItems()
if err != nil {
log.Error(err)
}
name := browser.GetName()
key := browser.GetSecretKey()
for _, item := range items {
err := item.CopyItem()
err := item.CopyDB()
if err != nil {
log.Error(err)
}

@ -27,8 +27,11 @@ type Browser interface {
// GetSecretKey return browser secret key
GetSecretKey() []byte
// GetAllItems, default return all of items(password|bookmark|cookie|history)
GetAllItems(itemName string) ([]common.Item, error)
// GetAllItems return all of items (password|bookmark|cookie|history)
GetAllItems() ([]common.Item, error)
// GetItem return single one from password|bookmark|cookie|history
GetItem(itemName string) (common.Item, error)
}
const (
@ -111,10 +114,8 @@ func (c *Chromium) GetSecretKey() []byte {
return c.secretKey
}
func (c *Chromium) GetAllItems(itemName string) (Items []common.Item, err error) {
itemName = strings.ToLower(itemName)
func (c *Chromium) GetAllItems() (Items []common.Item, err error) {
var items []common.Item
if itemName == "all" {
for item, choice := range chromiumItems {
m, err := utils.GetItemPath(c.profilePath, choice.mainFile)
if err != nil {
@ -125,18 +126,21 @@ func (c *Chromium) GetAllItems(itemName string) (Items []common.Item, err error)
log.Debugf("%s find %s File Success", c.name, item)
items = append(items, i)
}
} else if item, ok := chromiumItems[itemName]; ok {
return items, nil
}
func (c *Chromium) GetItem(itemName string) (common.Item, error) {
itemName = strings.ToLower(itemName)
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
return i, nil
} else {
return nil, errItemNotSupported
}
return items, nil
}
type Firefox struct {
@ -149,10 +153,8 @@ 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)
func (f *Firefox) GetAllItems() ([]common.Item, error) {
var items []common.Item
if itemName == "all" {
for item, choice := range firefoxItems {
var (
sub, main string
@ -174,7 +176,12 @@ func (f *Firefox) GetAllItems(itemName string) ([]common.Item, error) {
log.Debugf("%s find %s file success", f.name, item)
items = append(items, i)
}
} else if item, ok := firefoxItems[itemName]; ok {
return items, nil
}
func (f *Firefox) GetItem(itemName string) (common.Item, error) {
itemName = strings.ToLower(itemName)
if item, ok := firefoxItems[itemName]; ok {
var (
sub, main string
err error
@ -191,12 +198,10 @@ func (f *Firefox) GetAllItems(itemName string) ([]common.Item, error) {
}
i := item.newItem(main, sub)
log.Debugf("%s find %s file success", f.name, item.mainFile)
items = append(items, i)
return items, nil
return i, nil
} else {
return nil, errItemNotSupported
}
return items, nil
}
func (f *Firefox) GetName() string {

@ -25,13 +25,13 @@ type Item interface {
// FirefoxParse parse firefox items
FirefoxParse() error
// OutPut with json or csv
// OutPut file name and format type
OutPut(format, browser, dir string) error
// Copy item file to local path
CopyItem() error
// CopyDB is copy item db file to current dir
CopyDB() error
// Release item file
// Release is delete item db file
Release() error
}
@ -138,7 +138,7 @@ func (b *bookmarks) FirefoxParse() error {
return nil
}
func (b *bookmarks) CopyItem() error {
func (b *bookmarks) CopyDB() error {
return utils.CopyDB(b.mainPath, filepath.Base(b.mainPath))
}
@ -266,7 +266,7 @@ func (c *cookies) FirefoxParse() error {
return nil
}
func (c *cookies) CopyItem() error {
func (c *cookies) CopyDB() error {
return utils.CopyDB(c.mainPath, filepath.Base(c.mainPath))
}
@ -383,7 +383,7 @@ func (h *historyData) FirefoxParse() error {
return nil
}
func (h *historyData) CopyItem() error {
func (h *historyData) CopyDB() error {
return utils.CopyDB(h.mainPath, filepath.Base(h.mainPath))
}
@ -526,14 +526,13 @@ func (p *passwords) FirefoxParse() error {
Password: string(decrypt.PKCS5UnPadding(pwd)),
CreateDate: v.CreateDate,
})
}
}
}
return nil
}
func (p *passwords) CopyItem() error {
func (p *passwords) CopyDB() error {
err := utils.CopyDB(p.mainPath, filepath.Base(p.mainPath))
if err != nil {
log.Error(err)

Loading…
Cancel
Save