118 lines
2.3 KiB
Go
118 lines
2.3 KiB
Go
package main
|
|
|
|
import (
|
|
"database/sql"
|
|
"errors"
|
|
"fmt"
|
|
_ "github.com/lib/pq"
|
|
"os"
|
|
)
|
|
|
|
func db_open() (*sql.DB, error) {
|
|
|
|
// get connection string from environment variables
|
|
|
|
conn := fmt.Sprintf("host=%s port=%s user=%s password=%s dbname=%s sslmode=disable",
|
|
os.Getenv("DB_HOST"),
|
|
os.Getenv("DB_PORT"),
|
|
os.Getenv("DB_USER"),
|
|
os.Getenv("DB_PASSWORD"),
|
|
os.Getenv("DB_NAME"))
|
|
|
|
db, err := sql.Open("postgres", conn)
|
|
if err != nil {
|
|
return db, err
|
|
}
|
|
|
|
return db, nil
|
|
}
|
|
|
|
func db_get_setting(setting_name string) string {
|
|
|
|
setting_value := ""
|
|
|
|
db, err := db_open()
|
|
if err != nil {
|
|
return ""
|
|
}
|
|
defer db.Close()
|
|
|
|
sqlStatement := `select value from settings where name=$1;`
|
|
|
|
row := db.QueryRow(sqlStatement, setting_name)
|
|
err = row.Scan(&setting_value)
|
|
if err != nil {
|
|
return ""
|
|
}
|
|
|
|
return setting_value
|
|
}
|
|
|
|
func db_get_card_name_count(card_name string) (card_count int, err error) {
|
|
|
|
card_count = 0
|
|
|
|
db, err := db_open()
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
defer db.Close()
|
|
|
|
sqlStatement := `SELECT COUNT(card_id) FROM cards WHERE card_name = $1;`
|
|
|
|
row := db.QueryRow(sqlStatement, card_name)
|
|
err = row.Scan(&card_count)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
|
|
return card_count, nil
|
|
}
|
|
|
|
func db_wipe_card(card_name string) (*card_wipe_info, error) {
|
|
|
|
card_wipe_info := card_wipe_info{}
|
|
|
|
db, err := db_open()
|
|
if err != nil {
|
|
return &card_wipe_info, err
|
|
}
|
|
defer db.Close()
|
|
|
|
// set card as wiped and disabled
|
|
|
|
sqlStatement := `UPDATE cards SET` +
|
|
` lnurlw_enable = 'N', lnurlp_enable = 'N', email_enable = 'N', wiped = 'Y'` +
|
|
` WHERE card_name = $1;`
|
|
res, err := db.Exec(sqlStatement, card_name)
|
|
if err != nil {
|
|
return &card_wipe_info, err
|
|
}
|
|
count, err := res.RowsAffected()
|
|
if err != nil {
|
|
return &card_wipe_info, err
|
|
}
|
|
if count != 1 {
|
|
return &card_wipe_info, errors.New("not one card record updated")
|
|
}
|
|
|
|
// get card keys
|
|
|
|
sqlStatement = `SELECT card_id, uid, k0_auth_key, k2_cmac_key, k3, k4` +
|
|
` FROM cards WHERE card_name = $1;`
|
|
row := db.QueryRow(sqlStatement, card_name)
|
|
err = row.Scan(
|
|
&card_wipe_info.id,
|
|
&card_wipe_info.uid,
|
|
&card_wipe_info.k0,
|
|
&card_wipe_info.k2,
|
|
&card_wipe_info.k3,
|
|
&card_wipe_info.k4)
|
|
if err != nil {
|
|
return &card_wipe_info, err
|
|
}
|
|
|
|
card_wipe_info.k1 = db_get_setting("AES_DECRYPT_KEY")
|
|
|
|
return &card_wipe_info, nil
|
|
}
|