mirror of
https://github.com/moonD4rk/HackBrowserData.git
synced 2026-05-19 18:58:03 +02:00
feat: sort output by timestamp
This commit is contained in:
ᴍᴏᴏɴD4ʀᴋ
@@ -1,387 +0,0 @@
|
||||
package common
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"database/sql"
|
||||
"encoding/json"
|
||||
"hack-browser-data/log"
|
||||
"hack-browser-data/utils"
|
||||
"os"
|
||||
"time"
|
||||
|
||||
"github.com/gocarina/gocsv"
|
||||
|
||||
_ "github.com/mattn/go-sqlite3"
|
||||
"github.com/tidwall/gjson"
|
||||
)
|
||||
|
||||
const (
|
||||
Chrome = "Chrome"
|
||||
Safari = "Safari"
|
||||
)
|
||||
|
||||
var (
|
||||
FullData = new(BrowserData)
|
||||
bookmarkList []*Bookmarks
|
||||
cookieList []*Cookies
|
||||
historyList []*History
|
||||
loginItemList []*LoginData
|
||||
)
|
||||
|
||||
const (
|
||||
bookmarkID = "id"
|
||||
bookmarkAdded = "date_added"
|
||||
bookmarkUrl = "url"
|
||||
bookmarkName = "name"
|
||||
bookmarkType = "type"
|
||||
bookmarkChildren = "children"
|
||||
)
|
||||
|
||||
type (
|
||||
BrowserData struct {
|
||||
BrowserName string
|
||||
OutPutType string
|
||||
LoginData []*LoginData
|
||||
Bookmarks []*Bookmarks
|
||||
Cookies []*Cookies
|
||||
History []*History
|
||||
}
|
||||
LoginData struct {
|
||||
UserName string
|
||||
EncryptPass []byte
|
||||
Password string
|
||||
LoginUrl string
|
||||
CreateDate time.Time
|
||||
}
|
||||
Bookmarks struct {
|
||||
ID string
|
||||
DateAdded time.Time
|
||||
URL string
|
||||
Name string
|
||||
Type string
|
||||
}
|
||||
Cookies struct {
|
||||
KeyName string
|
||||
encryptValue []byte
|
||||
Value string
|
||||
Host string
|
||||
Path string
|
||||
IsSecure bool
|
||||
IsHTTPOnly bool
|
||||
HasExpire bool
|
||||
IsPersistent bool
|
||||
CreateDate time.Time
|
||||
ExpireDate time.Time
|
||||
}
|
||||
History struct {
|
||||
Url string
|
||||
Title string
|
||||
VisitCount int
|
||||
LastVisitTime time.Time
|
||||
}
|
||||
)
|
||||
|
||||
func (b BrowserData) OutPutCsv(dir, format string) error {
|
||||
switch {
|
||||
case len(b.Bookmarks) != 0:
|
||||
filename := utils.FormatFileName(dir, utils.Bookmarks, format)
|
||||
file, err := os.OpenFile(filename, os.O_RDWR|os.O_CREATE|os.O_TRUNC|os.O_APPEND, 0644)
|
||||
defer file.Close()
|
||||
if err != nil {
|
||||
log.Errorf("create file %s fail", filename)
|
||||
}
|
||||
err = gocsv.MarshalFile(b.Bookmarks, file)
|
||||
if err != nil {
|
||||
log.Error(err)
|
||||
}
|
||||
fallthrough
|
||||
case len(b.LoginData) != 0:
|
||||
filename := utils.FormatFileName(dir, utils.LoginData, format)
|
||||
file, err := os.OpenFile(filename, os.O_RDWR|os.O_CREATE|os.O_TRUNC|os.O_APPEND, 0644)
|
||||
defer file.Close()
|
||||
if err != nil {
|
||||
log.Errorf("create file %s fail", filename)
|
||||
}
|
||||
err = gocsv.MarshalFile(b.LoginData, file)
|
||||
if err != nil {
|
||||
log.Error(err)
|
||||
}
|
||||
fallthrough
|
||||
case len(b.Cookies) != 0:
|
||||
filename := utils.FormatFileName(dir, utils.Cookies, format)
|
||||
file, err := os.OpenFile(filename, os.O_RDWR|os.O_CREATE|os.O_TRUNC|os.O_APPEND, 0644)
|
||||
defer file.Close()
|
||||
if err != nil {
|
||||
log.Errorf("create file %s fail", filename)
|
||||
}
|
||||
err = gocsv.MarshalFile(b.Cookies, file)
|
||||
if err != nil {
|
||||
log.Error(err)
|
||||
}
|
||||
fallthrough
|
||||
case len(b.History) != 0:
|
||||
filename := utils.FormatFileName(dir, utils.History, format)
|
||||
file, err := os.OpenFile(filename, os.O_RDWR|os.O_CREATE|os.O_TRUNC|os.O_APPEND, 0644)
|
||||
defer file.Close()
|
||||
if err != nil {
|
||||
log.Errorf("create file %s fail", filename)
|
||||
}
|
||||
err = gocsv.MarshalFile(b.History, file)
|
||||
if err != nil {
|
||||
log.Error(err)
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (b BrowserData) OutPutJson(dir, format string) error {
|
||||
switch {
|
||||
case len(b.Bookmarks) != 0:
|
||||
filename := utils.FormatFileName(dir, utils.Bookmarks, format)
|
||||
file, err := os.OpenFile(filename, os.O_RDWR|os.O_CREATE|os.O_TRUNC|os.O_APPEND, 0644)
|
||||
defer file.Close()
|
||||
if err != nil {
|
||||
log.Errorf("create file %s fail", filename)
|
||||
}
|
||||
w := new(bytes.Buffer)
|
||||
enc := json.NewEncoder(w)
|
||||
enc.SetEscapeHTML(false)
|
||||
enc.SetIndent("", "\t")
|
||||
enc.Encode(b.BrowserName)
|
||||
file.Write(w.Bytes())
|
||||
fallthrough
|
||||
case len(b.Cookies) != 0:
|
||||
filename := utils.FormatFileName(dir, utils.Cookies, format)
|
||||
file, err := os.OpenFile(filename, os.O_RDWR|os.O_CREATE|os.O_TRUNC|os.O_APPEND, 0644)
|
||||
defer file.Close()
|
||||
if err != nil {
|
||||
log.Errorf("create file %s fail", filename)
|
||||
}
|
||||
w := new(bytes.Buffer)
|
||||
enc := json.NewEncoder(w)
|
||||
enc.SetEscapeHTML(false)
|
||||
enc.SetIndent("", "\t")
|
||||
err = enc.Encode(b.Cookies)
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
}
|
||||
file.Write(w.Bytes())
|
||||
fallthrough
|
||||
case len(b.History) != 0:
|
||||
filename := utils.FormatFileName(dir, utils.History, format)
|
||||
file, err := os.OpenFile(filename, os.O_RDWR|os.O_CREATE|os.O_TRUNC|os.O_APPEND, 0644)
|
||||
defer file.Close()
|
||||
if err != nil {
|
||||
log.Errorf("create file %s fail", filename)
|
||||
}
|
||||
w := new(bytes.Buffer)
|
||||
enc := json.NewEncoder(w)
|
||||
enc.SetEscapeHTML(false)
|
||||
enc.SetIndent("", "\t")
|
||||
err = enc.Encode(b.History)
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
}
|
||||
file.Write(w.Bytes())
|
||||
fallthrough
|
||||
case len(b.LoginData) != 0:
|
||||
filename := utils.FormatFileName(dir, utils.LoginData, format)
|
||||
file, err := os.OpenFile(filename, os.O_RDWR|os.O_CREATE|os.O_TRUNC|os.O_APPEND, 0644)
|
||||
defer file.Close()
|
||||
if err != nil {
|
||||
log.Errorf("create file %s fail", filename)
|
||||
}
|
||||
w := new(bytes.Buffer)
|
||||
enc := json.NewEncoder(w)
|
||||
enc.SetEscapeHTML(false)
|
||||
enc.SetIndent("", "\t")
|
||||
err = enc.Encode(b.LoginData)
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
}
|
||||
file.Write(w.Bytes())
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func ParseDB(dbname string) {
|
||||
switch dbname {
|
||||
case utils.LoginData:
|
||||
parseLogin()
|
||||
case utils.Bookmarks:
|
||||
parseBookmarks()
|
||||
case utils.Cookies:
|
||||
parseCookie()
|
||||
case utils.History:
|
||||
parseHistory()
|
||||
}
|
||||
}
|
||||
|
||||
func parseBookmarks() {
|
||||
bookmarks, err := utils.ReadFile(utils.Bookmarks)
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
}
|
||||
r := gjson.Parse(bookmarks)
|
||||
if r.Exists() {
|
||||
roots := r.Get("roots")
|
||||
roots.ForEach(func(key, value gjson.Result) bool {
|
||||
getBookmarkChildren(value)
|
||||
return true
|
||||
})
|
||||
}
|
||||
FullData.Bookmarks = bookmarkList
|
||||
}
|
||||
|
||||
var queryLogin = `SELECT origin_url, username_value, password_value, date_created FROM logins`
|
||||
|
||||
func parseLogin() {
|
||||
login := &LoginData{}
|
||||
loginDB, err := sql.Open("sqlite3", utils.LoginData)
|
||||
defer func() {
|
||||
if err := loginDB.Close(); err != nil {
|
||||
log.Println(err)
|
||||
}
|
||||
}()
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
}
|
||||
err = loginDB.Ping()
|
||||
rows, err := loginDB.Query(queryLogin)
|
||||
defer func() {
|
||||
if err := rows.Close(); err != nil {
|
||||
log.Println(err)
|
||||
}
|
||||
}()
|
||||
for rows.Next() {
|
||||
var (
|
||||
url, username, password string
|
||||
pwd []byte
|
||||
create int64
|
||||
)
|
||||
err = rows.Scan(&url, &username, &pwd, &create)
|
||||
login = &LoginData{
|
||||
UserName: username,
|
||||
EncryptPass: pwd,
|
||||
LoginUrl: url,
|
||||
CreateDate: utils.TimeEpochFormat(create),
|
||||
}
|
||||
password, err = utils.DecryptChromePass(pwd)
|
||||
login.Password = password
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
}
|
||||
loginItemList = append(loginItemList, login)
|
||||
}
|
||||
FullData.LoginData = loginItemList
|
||||
}
|
||||
|
||||
var queryCookie = `SELECT name, encrypted_value, host_key, path, creation_utc, expires_utc, is_secure, is_httponly, has_expires, is_persistent FROM cookies`
|
||||
|
||||
func parseCookie() {
|
||||
cookies := &Cookies{}
|
||||
cookieDB, err := sql.Open("sqlite3", utils.Cookies)
|
||||
defer func() {
|
||||
if err := cookieDB.Close(); err != nil {
|
||||
log.Println(err)
|
||||
}
|
||||
}()
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
}
|
||||
err = cookieDB.Ping()
|
||||
rows, err := cookieDB.Query(queryCookie)
|
||||
defer func() {
|
||||
if err := rows.Close(); err != nil {
|
||||
log.Println(err)
|
||||
}
|
||||
}()
|
||||
for rows.Next() {
|
||||
var (
|
||||
key, host, path, value string
|
||||
isSecure, isHTTPOnly, hasExpire, isPersistent int
|
||||
createDate, expireDate int64
|
||||
encryptValue []byte
|
||||
)
|
||||
err = rows.Scan(&key, &encryptValue, &host, &path, &createDate, &expireDate, &isSecure, &isHTTPOnly, &hasExpire, &isPersistent)
|
||||
cookies = &Cookies{
|
||||
KeyName: key,
|
||||
Host: host,
|
||||
Path: path,
|
||||
encryptValue: encryptValue,
|
||||
IsSecure: utils.IntToBool(isSecure),
|
||||
IsHTTPOnly: utils.IntToBool(isHTTPOnly),
|
||||
HasExpire: utils.IntToBool(hasExpire),
|
||||
IsPersistent: utils.IntToBool(isPersistent),
|
||||
CreateDate: utils.TimeEpochFormat(createDate),
|
||||
ExpireDate: utils.TimeEpochFormat(expireDate),
|
||||
}
|
||||
// remove prefix 'v10'
|
||||
value, err = utils.DecryptChromePass(encryptValue)
|
||||
cookies.Value = value
|
||||
cookieList = append(cookieList, cookies)
|
||||
}
|
||||
FullData.Cookies = cookieList
|
||||
}
|
||||
|
||||
var queryHistory = `SELECT url, title, visit_count, last_visit_time FROM urls`
|
||||
|
||||
func parseHistory() {
|
||||
history := &History{}
|
||||
historyDB, err := sql.Open("sqlite3", utils.History)
|
||||
defer func() {
|
||||
if err := historyDB.Close(); err != nil {
|
||||
log.Println(err)
|
||||
}
|
||||
}()
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
}
|
||||
err = historyDB.Ping()
|
||||
rows, err := historyDB.Query(queryHistory)
|
||||
defer func() {
|
||||
if err := rows.Close(); err != nil {
|
||||
log.Println(err)
|
||||
}
|
||||
}()
|
||||
for rows.Next() {
|
||||
var (
|
||||
url, title string
|
||||
visitCount int
|
||||
lastVisitTime int64
|
||||
)
|
||||
err := rows.Scan(&url, &title, &visitCount, &lastVisitTime)
|
||||
history = &History{
|
||||
Url: url,
|
||||
Title: title,
|
||||
VisitCount: visitCount,
|
||||
LastVisitTime: utils.TimeEpochFormat(lastVisitTime),
|
||||
}
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
continue
|
||||
}
|
||||
historyList = append(historyList, history)
|
||||
}
|
||||
FullData.History = historyList
|
||||
}
|
||||
|
||||
func getBookmarkChildren(value gjson.Result) (children gjson.Result) {
|
||||
b := new(Bookmarks)
|
||||
b.ID = value.Get(bookmarkID).String()
|
||||
nodeType := value.Get(bookmarkType)
|
||||
b.DateAdded = utils.TimeEpochFormat(value.Get(bookmarkAdded).Int())
|
||||
b.URL = value.Get(bookmarkUrl).String()
|
||||
b.Name = value.Get(bookmarkName).String()
|
||||
children = value.Get(bookmarkChildren)
|
||||
if nodeType.Exists() {
|
||||
b.Type = nodeType.String()
|
||||
bookmarkList = append(bookmarkList, b)
|
||||
if children.Exists() && children.IsArray() {
|
||||
for _, v := range children.Array() {
|
||||
children = getBookmarkChildren(v)
|
||||
}
|
||||
}
|
||||
}
|
||||
return children
|
||||
}
|
||||
Reference in New Issue
Block a user