feat: add parse Chromium download history

core/browser.go

@@ -46,6 +46,7 @@ const (
 	cookie     = "cookie"
 	history    = "history"
 	bookmark   = "bookmark"
+	downloadHistory = "downloadHistory"
 	password   = "password"
 	creditcard = "creditcard"
 )
@@ -74,6 +75,10 @@ var (
 			mainFile: data.ChromeHistoryFile,
 			newItem:  data.NewHistoryData,
 		},
+		downloadHistory: {
+			mainFile: data.ChromeHistoryFile,
+			newItem:  data.NewDownloadHistoryData,
+		},
 		password: {
 			mainFile: data.ChromePasswordFile,
 			newItem:  data.NewCPasswords,

core/data/output.go

@@ -42,6 +42,16 @@ func (h *historyData) outPutJson(browser, dir string) error {
 	return nil
 }
 
+func (d *downloadHistoryData) outPutJson(browser, dir string) error {
+	filename := utils.FormatFileName(dir, browser, "download_history", "json")
+	err := writeToJson(filename, d.downloadHistory)
+	if err != nil {
+		return err
+	}
+	fmt.Printf("%s Get %d history, filename is %s \n", utils.Prefix, len(d.downloadHistory), filename)
+	return nil
+}
+
 func (p *passwords) outPutJson(browser, dir string) error {
 	filename := utils.FormatFileName(dir, browser, "password", "json")
 	err := writeToJson(filename, p.logins)
@@ -111,6 +121,15 @@ func (h *historyData) outPutCsv(browser, dir string) error {
 	return nil
 }
 
+func (d *downloadHistoryData) outPutCsv(browser, dir string) error {
+	filename := utils.FormatFileName(dir, browser, "download_history", "csv")
+	if err := writeToCsv(filename, d.downloadHistory); err != nil {
+		return err
+	}
+	fmt.Printf("%s Get %d download history, filename is %s \n", utils.Prefix, len(d.downloadHistory), filename)
+	return nil
+}
+
 func (p *passwords) outPutCsv(browser, dir string) error {
 	filename := utils.FormatFileName(dir, browser, "password", "csv")
 	if err := writeToCsv(filename, p.logins); err != nil {
@@ -186,6 +205,12 @@ func (h *historyData) outPutConsole() {
 	}
 }
 
+func (d *downloadHistoryData) outPutConsole() {
+	for _, v := range d.downloadHistory {
+		fmt.Printf("%+v\n", v)
+	}
+}
+
 func (p *passwords) outPutConsole() {
 	for _, v := range p.logins {
 		fmt.Printf("%+v\n", v)

core/data/parse.go

@@ -51,6 +51,7 @@ var (
 	queryChromiumCredit   			= `SELECT guid, name_on_card, expiration_month, expiration_year, card_number_encrypted FROM credit_cards`
 	queryChromiumLogin    			= `SELECT origin_url, username_value, password_value, date_created FROM logins`
 	queryChromiumHistory  			= `SELECT url, title, visit_count, last_visit_time FROM urls`
+	queryChromiumDownloadHistory	= `SELECT target_path, tab_url, total_bytes, start_time, end_time FROM downloads`
 	queryChromiumCookie   			= `SELECT name, encrypted_value, host_key, path, creation_utc, expires_utc, is_secure, is_httponly, has_expires, is_persistent FROM cookies`
 	queryFirefoxHistory   			= `SELECT id, url, last_visit_date, title, visit_count FROM moz_places`
 	queryFirefoxBookMarks 			= `SELECT id, fk, type, dateAdded, title FROM moz_bookmarks`
@@ -447,6 +448,81 @@ func (h *historyData) OutPut(format, browser, dir string) error {
 	}
 }
 
+type downloadHistoryData struct {
+	mainPath string
+	downloadHistory  []downloadHistory
+}
+
+func NewDownloadHistoryData(main, sub string) Item {
+	return &downloadHistoryData{mainPath: main}
+}
+
+func (d *downloadHistoryData) ChromeParse(key []byte) error {
+	historyDB, err := sql.Open("sqlite3", ChromeHistoryFile)
+	if err != nil {
+		return err
+	}
+	defer func() {
+		if err := historyDB.Close(); err != nil {
+			log.Error(err)
+		}
+	}()
+	rows, err := historyDB.Query(queryChromiumDownloadHistory)
+	if err != nil {
+		return err
+	}
+	defer func() {
+		if err := rows.Close(); err != nil {
+			log.Debug(err)
+		}
+	}()
+	for rows.Next() {
+		var (
+			target_path, tab_url    string
+			total_bytes, start_time, end_time int64
+		)
+		err := rows.Scan(&target_path, &tab_url, &total_bytes, &start_time, &end_time)
+		data := downloadHistory{
+			TargetPath:	target_path,
+			Url:		tab_url,
+			TotalBytes:	total_bytes,
+			StartTime:	utils.TimeEpochFormat(start_time),
+			EndTime: 	utils.TimeEpochFormat(end_time),
+		}
+		if err != nil {
+			log.Error(err)
+		}
+		d.downloadHistory = append(d.downloadHistory, data)
+	}
+	return nil
+}
+
+func (d *downloadHistoryData) FirefoxParse() error {
+	return nil
+}
+
+func (d *downloadHistoryData) CopyDB() error {
+	return copyToLocalPath(d.mainPath, filepath.Base(d.mainPath))
+}
+
+func (d *downloadHistoryData) Release() error {
+	return os.Remove(filepath.Base(d.mainPath))
+}
+
+func (d *downloadHistoryData) OutPut(format, browser, dir string) error {
+	switch format {
+	case "csv":
+		err := d.outPutCsv(browser, dir)
+		return err
+	case "console":
+		d.outPutConsole()
+		return nil
+	default:
+		err := d.outPutJson(browser, dir)
+		return err
+	}
+}
+
 type passwords struct {
 	mainPath string
 	subPath  string
@@ -597,7 +673,6 @@ func (p *passwords) Release() error {
 }
 
 func (p *passwords) OutPut(format, browser, dir string) error {
-	sort.Sort(p)
 	switch format {
 	case "csv":
 		err := p.outPutCsv(browser, dir)
@@ -806,6 +881,13 @@ type (
 		VisitCount    int
 		LastVisitTime time.Time
 	}
+	downloadHistory struct {
+		TargetPath	string
+		Url			string
+		TotalBytes	int64
+		StartTime	time.Time
+		EndTime 	time.Time
+	}
 	card struct {
 		GUID            string
 		Name            string