stacktic/totp
1
0
Fork 0
Code Releases Activity

Add configuration file for storing secrets in ~/.config/totp/config.json

Browse Source
This commit is contained in:
Arnaud Ysmal 2021-03-06 22:51:37 +01:00
parent 69b1d09fc5
commit 6739cfc984
1 changed files with 48 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

62
main.go
View File

@ -40,6 +40,7 @@ import (
"flag"
"fmt"
"math"
"os"
"os/user"
"strings"
"time"
@ -111,13 +112,14 @@ func decryptAESCBC(p string, enc string) (rv []byte, err error) {
return
}
func getEntriesFromFile(n string, p string) (*map[string]entry, error) {
func getEntriesFromFile(n string, p string) ([]entry, error) {
var data string
var db *sql.DB
var err error
var entries map[string]entry
var pwddb string
var s []byte
var i int
if db, err = sql.Open("sqlite3", n); err != nil {
return nil, err
@ -145,16 +147,18 @@ func getEntriesFromFile(n string, p string) (*map[string]entry, error) {
delete(entries, "key")
}
for k, v := range entries {
if v.Encrypted {
rv := make([]entry, len(entries))
for _, v := range entries {
if v.Encrypted && pwddb != "" {
if s, err = decryptAESCBC(pwddb, v.Secret); err == nil {
v.Secret = string(s)
v.Encrypted = false
entries[k] = v
}
}
rv[i] = v
i++
}
return &entries, err
return rv, err
}
func computeOTP(secret string) (uint32, error) {
@ -196,31 +200,61 @@ func getFirefoxSyncv2Path() (string, error) {
return u.HomeDir + "/.mozilla/firefox/" + p + "/storage-sync-v2.sqlite", nil
}
func getConfigurationFile() (entries []entry, err error) {
var u *user.User
var data []byte
var fname string
if u, err = user.Current(); err != nil {
return
}
fname = u.HomeDir + "/.config/totp/config.json"
if err = os.Chmod(fname, 0600); err != nil {
return
}
if data, err = os.ReadFile(fname); err != nil {
return
}
if err = json.Unmarshal([]byte(data), &entries); err != nil {
return
}
return
}
func main() {
var sentry string
var dbf string
var passwd string
var err error
var entries *map[string]entry
var entries []entry
var firefox bool
flag.StringVar(&dbf, "d", "", "Database file")
flag.StringVar(&sentry, "e", "", "Select entry")
flag.BoolVar(&firefox, "f", false, "Get tokens from Firefox Authenticator Extension module")
flag.StringVar(&passwd, "p", "", "Database file")
flag.Parse()
if dbf == "" {
dbf, err = getFirefoxSyncv2Path()
if err != nil {
if firefox || dbf != "" {
if dbf == "" {
dbf, err = getFirefoxSyncv2Path()
if err != nil {
fmt.Printf("%s\n", err)
return
}
}
if entries, err = getEntriesFromFile(dbf, passwd); err != nil {
fmt.Printf("%s\n", err)
return
}
} else {
if entries, err = getConfigurationFile(); err != nil {
fmt.Printf("%s\n", err)
return
}
}
if entries, err = getEntriesFromFile(dbf, passwd); err != nil {
fmt.Printf("%s\n", err)
return
}
for _, v := range *entries {
for _, v := range entries {
if v.Encrypted {
continue
}