HackBrowserData/internal/browingdata/history/history.go

131 lines
2.7 KiB
Go
Raw Normal View History

package history
import (
"database/sql"
2022-04-11 11:57:40 +00:00
"os"
"sort"
"time"
2022-04-11 07:53:19 +00:00
"hack-browser-data/internal/item"
2022-04-12 03:29:01 +00:00
"hack-browser-data/internal/log"
"hack-browser-data/internal/utils/typeutil"
2022-08-14 13:22:34 +00:00
// import sqlite3 driver
_ "github.com/mattn/go-sqlite3"
)
type ChromiumHistory []history
type history struct {
Title string
2022-08-14 13:22:34 +00:00
URL string
VisitCount int
LastVisitTime time.Time
}
const (
queryChromiumHistory = `SELECT url, title, visit_count, last_visit_time FROM urls`
)
func (c *ChromiumHistory) Parse(masterKey []byte) error {
2022-04-11 07:53:19 +00:00
historyDB, err := sql.Open("sqlite3", item.TempChromiumHistory)
if err != nil {
return err
}
2022-04-11 11:57:40 +00:00
defer os.Remove(item.TempChromiumHistory)
defer historyDB.Close()
rows, err := historyDB.Query(queryChromiumHistory)
if err != nil {
return err
}
defer rows.Close()
for rows.Next() {
var (
url, title string
visitCount int
lastVisitTime int64
)
if err := rows.Scan(&url, &title, &visitCount, &lastVisitTime); err != nil {
2022-04-12 03:29:01 +00:00
log.Warn(err)
}
data := history{
2022-08-14 13:22:34 +00:00
URL: url,
Title: title,
VisitCount: visitCount,
LastVisitTime: typeutil.TimeEpoch(lastVisitTime),
}
*c = append(*c, data)
}
sort.Slice(*c, func(i, j int) bool {
return (*c)[i].VisitCount > (*c)[j].VisitCount
})
return nil
}
func (c *ChromiumHistory) Name() string {
return "history"
}
2022-01-11 10:19:17 +00:00
func (c *ChromiumHistory) Length() int {
return len(*c)
}
2022-01-11 10:19:17 +00:00
type FirefoxHistory []history
const (
queryFirefoxHistory = `SELECT id, url, COALESCE(last_visit_date, 0), COALESCE(title, ''), visit_count FROM moz_places`
closeJournalMode = `PRAGMA journal_mode=off`
)
2022-01-11 10:19:17 +00:00
func (f *FirefoxHistory) Parse(masterKey []byte) error {
var (
err error
keyDB *sql.DB
historyRows *sql.Rows
)
2022-04-11 07:53:19 +00:00
keyDB, err = sql.Open("sqlite3", item.TempFirefoxHistory)
2022-01-11 10:19:17 +00:00
if err != nil {
return err
}
2022-04-11 12:53:50 +00:00
defer os.Remove(item.TempFirefoxHistory)
defer keyDB.Close()
2022-01-11 10:19:17 +00:00
_, err = keyDB.Exec(closeJournalMode)
if err != nil {
return err
}
defer keyDB.Close()
historyRows, err = keyDB.Query(queryFirefoxHistory)
if err != nil {
return err
}
defer historyRows.Close()
for historyRows.Next() {
var (
id, visitDate int64
url, title string
visitCount int
)
if err = historyRows.Scan(&id, &url, &visitDate, &title, &visitCount); err != nil {
2022-04-12 03:29:01 +00:00
log.Warn(err)
2022-01-11 10:19:17 +00:00
}
*f = append(*f, history{
Title: title,
2022-08-14 13:22:34 +00:00
URL: url,
2022-01-11 10:19:17 +00:00
VisitCount: visitCount,
LastVisitTime: typeutil.TimeStamp(visitDate / 1000000),
2022-01-11 10:19:17 +00:00
})
}
sort.Slice(*f, func(i, j int) bool {
return (*f)[i].VisitCount < (*f)[j].VisitCount
})
return nil
}
func (f *FirefoxHistory) Name() string {
return "history"
}
func (f *FirefoxHistory) Length() int {
return len(*f)
}