0b55f94828
* refactor config * fix saas config * feat(config): scanmodule for each server in config.toml * feat(config): enable to specify containersOnly in config.toml * add new keys of config.toml to discover.go * fix summary output, logging
54 lines
1.2 KiB
Go
54 lines
1.2 KiB
Go
package config
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
)
|
|
|
|
// MetasploitConf is metasploit config
|
|
type MetasploitConf struct {
|
|
// DB type for metasploit dictionary (sqlite3, mysql, postgres or redis)
|
|
Type string
|
|
|
|
// http://metasploit-dictionary.com:1324 or DB connection string
|
|
URL string `json:"-"`
|
|
|
|
// /path/to/metasploit.sqlite3
|
|
SQLite3Path string `json:"-"`
|
|
}
|
|
|
|
func (cnf *MetasploitConf) setDefault() {
|
|
if cnf.Type == "" {
|
|
cnf.Type = "sqlite3"
|
|
}
|
|
if cnf.URL == "" && cnf.SQLite3Path == "" {
|
|
wd, _ := os.Getwd()
|
|
cnf.SQLite3Path = filepath.Join(wd, "go-msfdb.sqlite3")
|
|
}
|
|
}
|
|
|
|
const metasploitDBType = "METASPLOITDB_TYPE"
|
|
const metasploitDBURL = "METASPLOITDB_URL"
|
|
const metasploitDBPATH = "METASPLOITDB_SQLITE3_PATH"
|
|
|
|
// Init set options with the following priority.
|
|
// 1. Environment variable
|
|
// 2. config.toml
|
|
func (cnf *MetasploitConf) Init() {
|
|
if os.Getenv(metasploitDBType) != "" {
|
|
cnf.Type = os.Getenv(metasploitDBType)
|
|
}
|
|
if os.Getenv(metasploitDBURL) != "" {
|
|
cnf.URL = os.Getenv(metasploitDBURL)
|
|
}
|
|
if os.Getenv(metasploitDBPATH) != "" {
|
|
cnf.SQLite3Path = os.Getenv(metasploitDBPATH)
|
|
}
|
|
cnf.setDefault()
|
|
}
|
|
|
|
// IsFetchViaHTTP returns wether fetch via http
|
|
func (cnf *MetasploitConf) IsFetchViaHTTP() bool {
|
|
return Conf.Metasploit.Type == "http"
|
|
}
|