Merge pull request #125 from moonD4rk/devpull/128/head
ᴍᴏᴏɴD4ʀᴋ
3 years ago
committed by
GitHub
commit
2de62ac1bb
@ -0,0 +1,146 @@ |
|||||||
|
package localstorage |
||||||
|
|
||||||
|
import ( |
||||||
|
"bytes" |
||||||
|
"database/sql" |
||||||
|
"fmt" |
||||||
|
"os" |
||||||
|
"strings" |
||||||
|
|
||||||
|
"github.com/syndtr/goleveldb/leveldb" |
||||||
|
|
||||||
|
"hack-browser-data/internal/item" |
||||||
|
"hack-browser-data/internal/log" |
||||||
|
"hack-browser-data/internal/utils/typeutil" |
||||||
|
) |
||||||
|
|
||||||
|
type ChromiumLocalStorage []storage |
||||||
|
|
||||||
|
type storage struct { |
||||||
|
IsMeta bool |
||||||
|
URL string |
||||||
|
Key string |
||||||
|
Value string |
||||||
|
} |
||||||
|
|
||||||
|
func (c *ChromiumLocalStorage) Parse(masterKey []byte) error { |
||||||
|
db, err := leveldb.OpenFile(item.TempChromiumLocalStorage, nil) |
||||||
|
if err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
defer os.RemoveAll(item.TempChromiumLocalStorage) |
||||||
|
// log.Info("parsing local storage now")
|
||||||
|
defer db.Close() |
||||||
|
|
||||||
|
iter := db.NewIterator(nil, nil) |
||||||
|
for iter.Next() { |
||||||
|
key := iter.Key() |
||||||
|
value := iter.Value() |
||||||
|
// don't parse value upper than 5kB
|
||||||
|
if len(value) > 1024*5 { |
||||||
|
continue |
||||||
|
} |
||||||
|
var s = new(storage) |
||||||
|
s.fillKey(key) |
||||||
|
s.fillValue(value) |
||||||
|
// don't save meta data
|
||||||
|
if s.IsMeta { |
||||||
|
continue |
||||||
|
} |
||||||
|
*c = append(*c, *s) |
||||||
|
} |
||||||
|
iter.Release() |
||||||
|
err = iter.Error() |
||||||
|
return err |
||||||
|
} |
||||||
|
|
||||||
|
func (c *ChromiumLocalStorage) Name() string { |
||||||
|
return "localStorage" |
||||||
|
} |
||||||
|
|
||||||
|
func (s *storage) fillKey(b []byte) { |
||||||
|
keys := bytes.Split(b, []byte("\x00")) |
||||||
|
if len(keys) == 1 && bytes.HasPrefix(keys[0], []byte("META:")) { |
||||||
|
s.IsMeta = true |
||||||
|
s.fillMetaHeader(keys[0]) |
||||||
|
} |
||||||
|
if len(keys) == 2 && bytes.HasPrefix(keys[0], []byte("_")) { |
||||||
|
s.fillHeader(keys[0], keys[1]) |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
func (s *storage) fillMetaHeader(b []byte) { |
||||||
|
s.URL = string(bytes.Trim(b, "META:")) |
||||||
|
} |
||||||
|
|
||||||
|
func (s *storage) fillHeader(url, key []byte) { |
||||||
|
s.URL = string(bytes.Trim(url, "_")) |
||||||
|
s.Key = string(bytes.Trim(key, "\x01")) |
||||||
|
} |
||||||
|
|
||||||
|
// fillValue fills value of the storage
|
||||||
|
// TODO: support unicode charter
|
||||||
|
func (s *storage) fillValue(b []byte) { |
||||||
|
t := fmt.Sprintf("%c", b) |
||||||
|
m := strings.NewReplacer(" ", "", "\x00", "", "\x01", "").Replace(t) |
||||||
|
s.Value = m |
||||||
|
} |
||||||
|
|
||||||
|
type FirefoxLocalStorage []storage |
||||||
|
|
||||||
|
const ( |
||||||
|
queryFirefoxHistory = `SELECT originKey, key, value FROM webappsstore2` |
||||||
|
closeJournalMode = `PRAGMA journal_mode=off` |
||||||
|
) |
||||||
|
|
||||||
|
func (f *FirefoxLocalStorage) Parse(masterKey []byte) error { |
||||||
|
db, err := sql.Open("sqlite3", item.TempFirefoxLocalStorage) |
||||||
|
if err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
if err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
defer os.Remove(item.TempFirefoxLocalStorage) |
||||||
|
defer db.Close() |
||||||
|
_, err = db.Exec(closeJournalMode) |
||||||
|
if err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
defer db.Close() |
||||||
|
rows, err := db.Query(queryFirefoxHistory) |
||||||
|
if err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
defer rows.Close() |
||||||
|
for rows.Next() { |
||||||
|
var ( |
||||||
|
originKey, key, value string |
||||||
|
) |
||||||
|
if err = rows.Scan(&originKey, &key, &value); err != nil { |
||||||
|
log.Warn(err) |
||||||
|
} |
||||||
|
var s = new(storage) |
||||||
|
s.fillFirefox(originKey, key, value) |
||||||
|
*f = append(*f, *s) |
||||||
|
} |
||||||
|
return nil |
||||||
|
} |
||||||
|
|
||||||
|
func (s *storage) fillFirefox(originKey, key, value string) { |
||||||
|
// originKey = moc.buhtig.:https:443
|
||||||
|
p := strings.Split(originKey, ":") |
||||||
|
h := typeutil.Reverse([]byte(p[0])) |
||||||
|
if bytes.HasPrefix(h, []byte(".")) { |
||||||
|
h = h[1:] |
||||||
|
} |
||||||
|
if len(p) == 3 { |
||||||
|
s.URL = fmt.Sprintf("%s://%s:%s", p[1], string(h), p[2]) |
||||||
|
} |
||||||
|
s.Key = key |
||||||
|
s.Value = value |
||||||
|
} |
||||||
|
|
||||||
|
func (f *FirefoxLocalStorage) Name() string { |
||||||
|
return "localStorage" |
||||||
|
} |
||||||
@ -0,0 +1,24 @@ |
|||||||
|
package typeutil |
||||||
|
|
||||||
|
import ( |
||||||
|
"testing" |
||||||
|
) |
||||||
|
|
||||||
|
var ( |
||||||
|
reverseTestCases = [][]any{ |
||||||
|
[]any{1, 2, 3, 4, 5}, |
||||||
|
[]any{"1", "2", "3", "4", "5"}, |
||||||
|
[]any{"1", 2, "3", "4", 5}, |
||||||
|
} |
||||||
|
) |
||||||
|
|
||||||
|
func TestReverse(t *testing.T) { |
||||||
|
for _, ts := range reverseTestCases { |
||||||
|
h := Reverse(ts) |
||||||
|
for i := 0; i < len(ts); i++ { |
||||||
|
if h[len(h)-i-1] != ts[i] { |
||||||
|
t.Errorf("reverse failed %v != %v", h, ts) |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
Loading…
Reference in new issue