From 9cd2f61f1b14645879b6c44fb2906ca50f0c855c Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E1=B4=8D=E1=B4=8F=E1=B4=8F=C9=B4D4=CA=80=E1=B4=8B?= Date: Tue, 12 Apr 2022 18:23:02 +0800 Subject: [PATCH] feat: add cli options --- cmd/hack-browser-data/main.go | 172 +++++++++++-------------------- internal/browingdata/bookmark.go | 6 +- internal/browingdata/cookie.go | 4 +- internal/browingdata/password.go | 4 +- internal/browser/browser.go | 4 +- 5 files changed, 70 insertions(+), 120 deletions(-) diff --git a/cmd/hack-browser-data/main.go b/cmd/hack-browser-data/main.go index 0024368..95377ee 100644 --- a/cmd/hack-browser-data/main.go +++ b/cmd/hack-browser-data/main.go @@ -3,10 +3,14 @@ package main import ( "fmt" "os" + "strings" "hack-browser-data/internal/browser" "hack-browser-data/internal/log" "hack-browser-data/internal/outputter" + "hack-browser-data/internal/utils" + + "github.com/urfave/cli/v2" ) var ( @@ -23,122 +27,62 @@ func main() { } func Execute() { - browsers := browser.PickBrowser("firefox") - log.InitLog("debug") - filetype := "json" - dir := "results" - // dir := "result" - output := outputter.New(filetype) - var file *os.File - fmt.Println("browser number:", len(browsers)) - for _, b := range browsers { - browsingData, err := b.GetBrowsingData() - if err != nil { - panic(err) - } - for _, source := range browsingData.Sources { - filename := fmt.Sprintf("%s_%s.%s", b.Name(), source.Name(), filetype) - file, err = output.CreateFile(dir, filename) - err = output.Write(source, file) + 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 -cc]\n Get all browingdata(password/cookie/history/bookmark) from chrome", + Version: "0.4.0", + 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: &browserName, Value: "all", Usage: "available browsers: all|" + strings.Join(browser.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: "csv", Usage: "format, csv|json|console"}, + &cli.StringFlag{Name: "profile-dir-path", Aliases: []string{"p"}, Destination: &customProfilePath, Value: "", Usage: "custom profile dir path, get with chrome://version"}, + }, + HideHelpCommand: true, + Action: func(c *cli.Context) error { + var ( + browsers []browser.Browser + err error + ) + if verbose { + log.InitLog("debug") + } else { + log.InitLog("error") + } + browsers, err = browser.PickBrowser(browserName) if err != nil { log.Error(err) } - } + output := outputter.New(outputFormat) + + for _, b := range browsers { + data, err := b.GetBrowsingData() + if err != nil { + log.Error(err) + } + var f *os.File + for _, source := range data.Sources { + filename := fmt.Sprintf("%s_%s.%s", b.Name(), source.Name(), outputFormat) + f, err = output.CreateFile(exportDir, filename) + err = output.Write(source, f) + if err != nil { + log.Error(err) + } + } + } + if compress { + err = utils.Compress(exportDir) + if err != nil { + log.Error(err) + } + } + return nil + }, + } + err := app.Run(os.Args) + if err != nil { + log.Error(err) } } - -// 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 -cc]\n Get all browingdata(password/cookie/history/bookmark) from chrome", -// Version: "0.3.7", -// 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: &browserName, 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: "csv", Usage: "format, csv|json|console"}, -// &cli.StringFlag{Name: "profile-dir-path", Aliases: []string{"p"}, Destination: &customProfilePath, Value: "", Usage: "custom profile dir path, get with chrome://version"}, -// }, -// HideHelpCommand: true, -// Action: func(c *cli.Context) error { -// var ( -// browsers []core.Browser -// err error -// ) -// if verbose { -// log.InitLog("debug") -// } else { -// log.InitLog("error") -// } -// if customProfilePath != "" { -// browsers, err = core.PickCustomBrowser(browserName, customProfilePath, customKeyPath) -// if err != nil { -// log.Error(err) -// } -// } else { -// // default select all browsers -// browsers, err = core.PickBrowser(browserName) -// if err != nil { -// log.Error(err) -// } -// } -// err = utils.MakeDir(exportDir) -// if err != nil { -// log.Error(err) -// } -// for _, browser := range browsers { -// err := browser.InitSecretKey() -// if err != nil { -// log.Error(err) -// } -// // default select all items -// // you can get single item with browser.GetItem(itemName) -// items, err := browser.GetAllItems() -// if err != nil { -// log.Error(err) -// } -// name := browser.Name() -// key := browser.GetSecretKey() -// for _, item := range items { -// err := item.CopyDB() -// if err != nil { -// log.Error(err) -// } -// switch browser.(type) { -// case *core.Chromium: -// err := item.ChromeParse(key) -// if err != nil { -// log.Error(err) -// } -// case *core.Firefox: -// err := item.FirefoxParse() -// if err != nil { -// log.Error(err) -// } -// } -// err = item.Release() -// if err != nil { -// log.Error(err) -// } -// err = item.OutPut(outputFormat, name, exportDir) -// if err != nil { -// log.Error(err) -// } -// } -// } -// if compress { -// err = utils.Compress(exportDir) -// if err != nil { -// log.Error(err) -// } -// } -// return nil -// }, -// } -// err := app.Run(os.Args) -// if err != nil { -// log.Error(err) -// } -// } diff --git a/internal/browingdata/bookmark.go b/internal/browingdata/bookmark.go index 94afb7f..35f7925 100644 --- a/internal/browingdata/bookmark.go +++ b/internal/browingdata/bookmark.go @@ -40,7 +40,7 @@ func (c *ChromiumBookmark) Parse(masterKey []byte) error { return true }) } - // TODO: 使用泛型重构 + // TODO: refactor with go generics sort.Slice(*c, func(i, j int) bool { return (*c)[i].DateAdded.After((*c)[j].DateAdded) }) @@ -95,7 +95,9 @@ func (f *FirefoxBookmark) Parse(masterKey []byte) error { defer os.Remove(item.TempFirefoxBookmark) defer keyDB.Close() _, err = keyDB.Exec(closeJournalMode) - + if err != nil { + log.Error(err) + } bookmarkRows, err = keyDB.Query(queryFirefoxBookMark) if err != nil { return err diff --git a/internal/browingdata/cookie.go b/internal/browingdata/cookie.go index 082382f..515b564 100644 --- a/internal/browingdata/cookie.go +++ b/internal/browingdata/cookie.go @@ -58,7 +58,9 @@ func (c *ChromiumCookie) Parse(masterKey []byte) error { } else { value, err = decrypter.ChromePass(masterKey, encryptValue) } - log.Error(err) + if err != nil { + log.Error(err) + } } cookie.Value = string(value) *c = append(*c, cookie) diff --git a/internal/browingdata/password.go b/internal/browingdata/password.go index 79b5577..168dd86 100644 --- a/internal/browingdata/password.go +++ b/internal/browingdata/password.go @@ -54,7 +54,9 @@ func (c *ChromiumPassword) Parse(masterKey []byte) error { } else { password, err = decrypter.ChromePass(masterKey, pwd) } - log.Error(err) + if err != nil { + log.Error(err) + } } if create > time.Now().Unix() { login.CreateDate = utils.TimeEpochFormat(create) diff --git a/internal/browser/browser.go b/internal/browser/browser.go index 9625d7f..635a3f8 100644 --- a/internal/browser/browser.go +++ b/internal/browser/browser.go @@ -18,7 +18,7 @@ type Browser interface { GetBrowsingData() (*browingdata.Data, error) } -func PickBrowser(name string) []Browser { +func PickBrowser(name string) ([]Browser, error) { var browsers []Browser clist := pickChromium(name) for _, b := range clist { @@ -32,7 +32,7 @@ func PickBrowser(name string) []Browser { browsers = append(browsers, b) } } - return browsers + return browsers, nil } func pickChromium(name string) []Browser {