mirror of https://github.com/gophish/gophish
51 lines
1.3 KiB
Go
51 lines
1.3 KiB
Go
package db
|
|
|
|
import (
|
|
"database/sql"
|
|
"fmt"
|
|
"os"
|
|
|
|
"github.com/jordan-wright/gophish/config"
|
|
_ "github.com/mattn/go-sqlite3"
|
|
)
|
|
|
|
var Conn *sql.DB
|
|
|
|
// Setup initializes the Conn object
|
|
// It also populates the Gophish Config object
|
|
func Setup() error {
|
|
createTablesSQL := []string{
|
|
//Create tables
|
|
`CREATE TABLE Users (id INTEGER PRIMARY KEY AUTOINCREMENT, username TEXT NOT NULL, hash VARCHAR(60) NOT NULL, apikey VARCHAR(32));`,
|
|
`CREATE TABLE Campaigns (id INTEGER PRIMARY KEY AUTOINCREMENT, name TEXT NOT NULL, created_date TEXT NOT NULL, completed_date TEXT, status TEXT NOT NULL);`,
|
|
}
|
|
//If the file already exists, delete it and recreate it
|
|
_, err := os.Stat(config.Conf.DBPath)
|
|
if err == nil {
|
|
os.Remove(config.Conf.DBPath)
|
|
}
|
|
fmt.Println("Creating db at " + config.Conf.DBPath)
|
|
Conn, err = sql.Open("sqlite3", config.Conf.DBPath)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
//Create the tables needed
|
|
for _, stmt := range createTablesSQL {
|
|
_, err = Conn.Exec(stmt)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
//Create the default user
|
|
stmt, err := Conn.Prepare(`INSERT INTO Users (username, hash, apikey) VALUES (?, ?, ?);`)
|
|
defer stmt.Close()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
_, err = stmt.Exec("jordan", "$2a$10$d4OtT.RkEOQn.iruVWIQ5u8CeV/85ZYF41y8wKeUwsAPqPNFvTccW", "12345678901234567890123456789012")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|