chore: update project layout

browingdata/cookie/cookie.go

@@ -0,0 +1,148 @@
+package cookie
+
+import (
+	"database/sql"
+	"os"
+	"sort"
+	"time"
+
+	// import sqlite3 driver
+	_ "github.com/mattn/go-sqlite3"
+
+	"github.com/moond4rk/HackBrowserData/crypto"
+	"github.com/moond4rk/HackBrowserData/item"
+	"github.com/moond4rk/HackBrowserData/log"
+	"github.com/moond4rk/HackBrowserData/utils/typeutil"
+)
+
+type ChromiumCookie []cookie
+
+type cookie struct {
+	Host         string
+	Path         string
+	KeyName      string
+	encryptValue []byte
+	Value        string
+	IsSecure     bool
+	IsHTTPOnly   bool
+	HasExpire    bool
+	IsPersistent bool
+	CreateDate   time.Time
+	ExpireDate   time.Time
+}
+
+const (
+	queryChromiumCookie = `SELECT name, encrypted_value, host_key, path, creation_utc, expires_utc, is_secure, is_httponly, has_expires, is_persistent FROM cookies`
+)
+
+func (c *ChromiumCookie) Parse(masterKey []byte) error {
+	cookieDB, err := sql.Open("sqlite3", item.TempChromiumCookie)
+	if err != nil {
+		return err
+	}
+	defer os.Remove(item.TempChromiumCookie)
+	defer cookieDB.Close()
+	rows, err := cookieDB.Query(queryChromiumCookie)
+	if err != nil {
+		return err
+	}
+	defer rows.Close()
+	for rows.Next() {
+		var (
+			key, host, path                               string
+			isSecure, isHTTPOnly, hasExpire, isPersistent int
+			createDate, expireDate                        int64
+			value, encryptValue                           []byte
+		)
+		if err = rows.Scan(&key, &encryptValue, &host, &path, &createDate, &expireDate, &isSecure, &isHTTPOnly, &hasExpire, &isPersistent); err != nil {
+			log.Warn(err)
+		}
+
+		cookie := cookie{
+			KeyName:      key,
+			Host:         host,
+			Path:         path,
+			encryptValue: encryptValue,
+			IsSecure:     typeutil.IntToBool(isSecure),
+			IsHTTPOnly:   typeutil.IntToBool(isHTTPOnly),
+			HasExpire:    typeutil.IntToBool(hasExpire),
+			IsPersistent: typeutil.IntToBool(isPersistent),
+			CreateDate:   typeutil.TimeEpoch(createDate),
+			ExpireDate:   typeutil.TimeEpoch(expireDate),
+		}
+		if len(encryptValue) > 0 {
+			var err error
+			if masterKey == nil {
+				value, err = crypto.DPAPI(encryptValue)
+			} else {
+				value, err = crypto.Chromium(masterKey, encryptValue)
+			}
+			if err != nil {
+				log.Error(err)
+			}
+		}
+		cookie.Value = string(value)
+		*c = append(*c, cookie)
+	}
+	sort.Slice(*c, func(i, j int) bool {
+		return (*c)[i].CreateDate.After((*c)[j].CreateDate)
+	})
+	return nil
+}
+
+func (c *ChromiumCookie) Name() string {
+	return "cookie"
+}
+
+func (c *ChromiumCookie) Length() int {
+	return len(*c)
+}
+
+type FirefoxCookie []cookie
+
+const (
+	queryFirefoxCookie = `SELECT name, value, host, path, creationTime, expiry, isSecure, isHttpOnly FROM moz_cookies`
+)
+
+func (f *FirefoxCookie) Parse(masterKey []byte) error {
+	cookieDB, err := sql.Open("sqlite3", item.TempFirefoxCookie)
+	if err != nil {
+		return err
+	}
+	defer os.Remove(item.TempFirefoxCookie)
+	defer cookieDB.Close()
+	rows, err := cookieDB.Query(queryFirefoxCookie)
+	if err != nil {
+		return err
+	}
+	defer rows.Close()
+	for rows.Next() {
+		var (
+			name, value, host, path string
+			isSecure, isHTTPOnly    int
+			creationTime, expiry    int64
+		)
+		if err = rows.Scan(&name, &value, &host, &path, &creationTime, &expiry, &isSecure, &isHTTPOnly); err != nil {
+			log.Warn(err)
+		}
+		*f = append(*f, cookie{
+			KeyName:    name,
+			Host:       host,
+			Path:       path,
+			IsSecure:   typeutil.IntToBool(isSecure),
+			IsHTTPOnly: typeutil.IntToBool(isHTTPOnly),
+			CreateDate: typeutil.TimeStamp(creationTime / 1000000),
+			ExpireDate: typeutil.TimeStamp(expiry),
+			Value:      value,
+		})
+	}
+	return nil
+}
+
+func (f *FirefoxCookie) Name() string {
+	return "cookie"
+}
+
+func (f *FirefoxCookie) Length() int {
+	return len(*f)
+}