You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
80 lines
1.8 KiB
80 lines
1.8 KiB
package browser
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
|
|
"hack-browser-data/pkg/browser/outputter"
|
|
"hack-browser-data/pkg/log"
|
|
)
|
|
|
|
func TestPickChromium(t *testing.T) {
|
|
browsers := pickChromium("chrome")
|
|
log.InitLog("debug")
|
|
filetype := "json"
|
|
dir := "result"
|
|
output := outputter.NewOutPutter(filetype)
|
|
if err := output.MakeDir("result"); err != nil {
|
|
panic(err)
|
|
}
|
|
for _, b := range browsers {
|
|
fmt.Printf("%+v\n", b)
|
|
if err := b.CopyItemFileToLocal(); err != nil {
|
|
panic(err)
|
|
}
|
|
masterKey, err := b.GetMasterKey()
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
}
|
|
browserName := b.GetName()
|
|
multiData := b.GetBrowsingData()
|
|
for _, data := range multiData {
|
|
if err := data.Parse(masterKey); err != nil {
|
|
fmt.Println(err)
|
|
}
|
|
filename := fmt.Sprintf("%s_%s.%s", browserName, data.Name(), filetype)
|
|
file, err := output.CreateFile(dir, filename)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
if err := output.Write(data, file); err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestPickFirefox(t *testing.T) {
|
|
browsers := pickFirefox("all")
|
|
filetype := "json"
|
|
dir := "result"
|
|
output := outputter.NewOutPutter(filetype)
|
|
if err := output.MakeDir("result"); err != nil {
|
|
panic(err)
|
|
}
|
|
for _, b := range browsers {
|
|
fmt.Printf("%+v\n", b)
|
|
if err := b.CopyItemFileToLocal(); err != nil {
|
|
panic(err)
|
|
}
|
|
masterKey, err := b.GetMasterKey()
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
}
|
|
browserName := b.GetName()
|
|
multiData := b.GetBrowsingData()
|
|
for _, data := range multiData {
|
|
if err := data.Parse(masterKey); err != nil {
|
|
fmt.Println(err)
|
|
}
|
|
filename := fmt.Sprintf("%s_%s.%s", browserName, data.Name(), filetype)
|
|
file, err := output.CreateFile(dir, filename)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
if err := output.Write(data, file); err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|