feat: add firefox decrypt

internal/browser/browser.go

@@ -7,6 +7,7 @@ import (
 
 	"hack-browser-data/internal/browingdata"
 	"hack-browser-data/internal/browser/chromium"
+	"hack-browser-data/internal/browser/firefox"
 )
 
 type Browser interface {
@@ -69,15 +70,15 @@ func pickFirefox(name string) []Browser {
 	var browsers []Browser
 	name = strings.ToLower(name)
 	if name == "all" || name == "firefox" {
-		// for _, f := range firefoxList {
-		// multiFirefox, err := firefox(f.browserInfo, f.items)
-		// if err != nil {
-		// 	panic(err)
-		// }
-		// for _, browser := range multiFirefox {
-		// 	browsers = append(browsers, browser)
-		// }
-		// }
+		for _, v := range firefoxList {
+			multiFirefox, err := firefox.New(v.name, v.storage, v.profilePath, v.items)
+			if err != nil {
+				panic(err)
+			}
+			for _, browser := range multiFirefox {
+				browsers = append(browsers, browser)
+			}
+		}
 		return browsers
 	}
 	return nil

internal/browser/firefox/firefox.go

@@ -1,116 +1,125 @@
 package firefox
 
-// type firefox struct {
-// 	name           string
-// 	storage        string
-// 	profilePath    string
-// 	masterKey      []byte
-// 	items          []item.Item
-// 	itemPaths      map[item.Item]string
-// 	multiItemPaths map[string]map[item.Item]string
-// }
-//
-// // New
-// func New(info *browserInfo, items []item.Item) ([]*firefox, error) {
-// 	f := &firefox{
-// 		browserInfo: info,
-// 		items:       items,
-// 	}
-// 	multiItemPaths, err := getFirefoxItemAbsPath(f.browserInfo.profilePath, f.items)
-// 	if err != nil {
-// 		if strings.Contains(err.Error(), "profile path is not exist") {
-// 			fmt.Println(err)
-// 			return nil, nil
-// 		}
-// 		panic(err)
-// 	}
-// 	var firefoxList []*firefox
-// 	for name, value := range multiItemPaths {
-// 		firefoxList = append(firefoxList, &firefox{
-// 			browserInfo: &browserInfo{
-// 				name:      name,
-// 				masterKey: nil,
-// 			},
-// 			items:     items,
-// 			itemPaths: value,
-// 		})
-// 	}
-// 	return firefoxList, nil
-// }
-//
-// func getFirefoxItemAbsPath(profilePath string, items []item.Item) (map[string]map[item.Item]string, error) {
-// 	var multiItemPaths = make(map[string]map[item.Item]string)
-// 	absProfilePath := path.Join(homeDir, filepath.Clean(profilePath))
-// 	// TODO: Handle read file error
-// 	if !isFileExist(absProfilePath) {
-// 		return nil, fmt.Errorf("%s profile path is not exist", absProfilePath)
-// 	}
-// 	err := filepath.Walk(absProfilePath, firefoxWalkFunc(items, multiItemPaths))
-// 	return multiItemPaths, err
-// }
-//
-// func (f *firefox) CopyItemFileToLocal() error {
-// 	for item, sourcePath := range f.itemPaths {
-// 		var dstFilename = item.TempName()
-// 		locals, _ := filepath.Glob("*")
-// 		for _, v := range locals {
-// 			if v == dstFilename {
-// 				err := os.Remove(dstFilename)
-// 				// TODO: Should Continue all iteration error
-// 				if err != nil {
-// 					return err
-// 				}
-// 			}
-// 		}
-//
-// 		// TODO: Handle read file name error
-// 		sourceFile, err := ioutil.ReadFile(sourcePath)
-// 		if err != nil {
-// 			fmt.Println(err.Error())
-// 		}
-// 		err = ioutil.WriteFile(dstFilename, sourceFile, 0777)
-// 		if err != nil {
-// 			return err
-// 		}
-// 	}
-// 	return nil
-// }
-//
-// func firefoxWalkFunc(items []item.Item, multiItemPaths map[string]map[item.Item]string) filepath.WalkFunc {
-// 	return func(path string, info fs.FileInfo, err error) error {
-// 		for _, v := range items {
-// 			if info.Name() == v.FileName() {
-// 				parentDir := getParentDir(path)
-// 				if _, exist := multiItemPaths[parentDir]; exist {
-// 					multiItemPaths[parentDir][v] = path
-// 				} else {
-// 					multiItemPaths[parentDir] = map[item.Item]string{v: path}
-// 				}
-// 			}
-// 		}
-// 		return err
-// 	}
-// }
-//
-// func getParentDir(absPath string) string {
-// 	return filepath.Base(filepath.Dir(absPath))
-// }
-//
-// func (f *firefox) GetMasterKey() ([]byte, error) {
-// 	return f.masterKey, nil
-// }
-//
-// func (f *firefox) Name() string {
-// 	return f.name
-// }
-//
-// func (f *firefox) GetBrowsingData() []browingdata.Source {
-// 	var browsingData []browingdata.Source
-// 	for item := range f.itemPaths {
-// 		d := item.NewBrowsingData()
-// 		if d != nil {
-// 			browsingData = append(browsingData, d)
-// 		}
-// 	}
-// 	return browsingData
-// }
+import (
+	"fmt"
+	"io/fs"
+	"io/ioutil"
+	"path/filepath"
+	"strings"
+
+	"hack-browser-data/internal/browingdata"
+	"hack-browser-data/internal/item"
+	"hack-browser-data/internal/utils/fileutil"
+	"hack-browser-data/internal/utils/typeutil"
+)
+
+type firefox struct {
+	name        string
+	storage     string
+	profilePath string
+	masterKey   []byte
+	items       []item.Item
+	itemPaths   map[item.Item]string
+}
+
+// New returns a new firefox instance.
+func New(name, storage, profilePath string, items []item.Item) ([]*firefox, error) {
+	f := &firefox{
+		name:        name,
+		storage:     storage,
+		profilePath: profilePath,
+		items:       items,
+	}
+	if !fileutil.FolderExists(profilePath) {
+		return nil, fmt.Errorf("%s profile path is not exist: %s", name, profilePath)
+	}
+
+	multiItemPaths, err := f.getMultiItemPath(f.profilePath, f.items)
+	if err != nil {
+		if strings.Contains(err.Error(), "profile path is not exist") {
+			fmt.Println(err)
+			return nil, nil
+		}
+		return nil, err
+	}
+	var firefoxList []*firefox
+	for name, itemPaths := range multiItemPaths {
+		firefoxList = append(firefoxList, &firefox{
+			name:      name,
+			items:     typeutil.Keys(itemPaths),
+			itemPaths: itemPaths,
+		})
+	}
+	return firefoxList, nil
+}
+
+func (f *firefox) getMultiItemPath(profilePath string, items []item.Item) (map[string]map[item.Item]string, error) {
+	var multiItemPaths = make(map[string]map[item.Item]string)
+
+	err := filepath.Walk(profilePath, firefoxWalkFunc(items, multiItemPaths))
+	return multiItemPaths, err
+}
+
+func (f *firefox) copyItemToLocal() error {
+	for i, path := range f.itemPaths {
+		// var dstFilename = item.TempName()
+		var filename = i.String()
+		// TODO: Handle read file error
+		d, err := ioutil.ReadFile(path)
+		if err != nil {
+			fmt.Println(err.Error())
+		}
+		err = ioutil.WriteFile(filename, d, 0777)
+		if err != nil {
+			return err
+		}
+	}
+	return nil
+}
+
+func firefoxWalkFunc(items []item.Item, multiItemPaths map[string]map[item.Item]string) filepath.WalkFunc {
+	return func(path string, info fs.FileInfo, err error) error {
+		for _, v := range items {
+			if info.Name() == v.FileName() {
+				parentDir := getParentDir(path)
+				if _, exist := multiItemPaths[parentDir]; exist {
+					multiItemPaths[parentDir][v] = path
+				} else {
+					multiItemPaths[parentDir] = map[item.Item]string{v: path}
+				}
+			}
+		}
+		return err
+	}
+}
+
+func getParentDir(absPath string) string {
+	return filepath.Base(filepath.Dir(absPath))
+}
+
+func (f *firefox) GetMasterKey() ([]byte, error) {
+	return f.masterKey, nil
+}
+
+func (f *firefox) Name() string {
+	return f.name
+}
+
+func (f *firefox) GetBrowsingData() (*browingdata.Data, error) {
+	b := browingdata.New(f.items)
+
+	if err := f.copyItemToLocal(); err != nil {
+		return nil, err
+	}
+
+	masterKey, err := f.GetMasterKey()
+	if err != nil {
+		return nil, err
+	}
+
+	f.masterKey = masterKey
+	if err := b.Recovery(f.masterKey); err != nil {
+		return nil, err
+	}
+	return b, nil
+}