mirror of
https://github.com/moonD4rk/HackBrowserData
synced 2024-12-19 12:34:36 +00:00
536f2082f9
* refactor: Refactor package names and imports for better code organization. * refactor: Package imports and variable types for consistency * chore: Disable unused-parameter rule in revive. * refactor: Refactor and organize data extraction and browserdata parse. * fix: rename wrong error message info
69 lines
1.8 KiB
Go
69 lines
1.8 KiB
Go
package browserdata
|
|
|
|
import (
|
|
"log/slog"
|
|
|
|
"github.com/moond4rk/hackbrowserdata/extractor"
|
|
"github.com/moond4rk/hackbrowserdata/types"
|
|
"github.com/moond4rk/hackbrowserdata/utils/fileutil"
|
|
)
|
|
|
|
type BrowserData struct {
|
|
extractors map[types.DataType]extractor.Extractor
|
|
}
|
|
|
|
func New(items []types.DataType) *BrowserData {
|
|
bd := &BrowserData{
|
|
extractors: make(map[types.DataType]extractor.Extractor),
|
|
}
|
|
bd.addExtractors(items)
|
|
return bd
|
|
}
|
|
|
|
func (d *BrowserData) Recovery(masterKey []byte) error {
|
|
for _, source := range d.extractors {
|
|
if err := source.Extract(masterKey); err != nil {
|
|
slog.Error("parse error", "source_data", source.Name(), "err", err.Error())
|
|
continue
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (d *BrowserData) Output(dir, browserName, flag string) {
|
|
output := newOutPutter(flag)
|
|
|
|
for _, source := range d.extractors {
|
|
if source.Len() == 0 {
|
|
// if the length of the export data is 0, then it is not necessary to output
|
|
continue
|
|
}
|
|
filename := fileutil.Filename(browserName, source.Name(), output.Ext())
|
|
|
|
f, err := output.CreateFile(dir, filename)
|
|
if err != nil {
|
|
slog.Error("create file error", "filename", filename, "err", err.Error())
|
|
continue
|
|
}
|
|
if err := output.Write(source, f); err != nil {
|
|
slog.Error("write to file error", "filename", filename, "err", err.Error())
|
|
continue
|
|
}
|
|
if err := f.Close(); err != nil {
|
|
slog.Error("close file error", "filename", filename, "err", err.Error())
|
|
continue
|
|
}
|
|
slog.Warn("export success", "filename", filename)
|
|
}
|
|
}
|
|
|
|
func (d *BrowserData) addExtractors(items []types.DataType) {
|
|
for _, itemType := range items {
|
|
if source := extractor.CreateExtractor(itemType); source != nil {
|
|
d.extractors[itemType] = source
|
|
} else {
|
|
slog.Debug("source not found", "source", itemType)
|
|
}
|
|
}
|
|
}
|