config

type BeaconStaticCfg struct {
	DefaultConnectionThresh int `yaml:"DefaultConnectionThresh"`
}

type BeaconTableCfg struct {
	BeaconTable string
}

type BlacklistedStaticCfg struct {
	UseIPms            bool                  `yaml:"myIP.ms"`
	UseDNSBH           bool                  `yaml:"MalwareDomains.com"`
	UseMDL             bool                  `yaml:"MalwareDomainList.com"`
	SafeBrowsing       SafeBrowsingStaticCfg `yaml:"SafeBrowsing"`
	IPBlacklists       []string              `yaml:"CustomIPBlacklists"`
	HostnameBlacklists []string              `yaml:"CustomHostnameBlacklists"`
	URLBlacklists      []string              `yaml:"CustomURLBlacklists"`
}

type BlacklistedTableCfg struct {
	BlacklistDatabase string
	SourceIPsTable    string
	DestIPsTable      string
	HostnamesTable    string
	UrlsTable         string
}

type BroStaticCfg struct {
	ImportDirectory string `yaml:"ImportDirectory"`
	DBRoot          string `yaml:"DBRoot"`
	MetaDB          string `yaml:"MetaDB"`
	ImportBuffer    int    `yaml:"ImportBuffer"`
}

type Config struct {
	R RunningCfg
	S StaticCfg
	T TableCfg
}

type CrossrefStaticCfg struct {
	BeaconThreshold float64 `yaml:"BeaconThreshold"`
}

type CrossrefTableCfg struct {
	SourceTable string
	DestTable   string
}

type DNSTableCfg struct {
	ExplodedDNSTable string
	HostnamesTable   string
}

type LogStaticCfg struct {
	LogLevel    int    `yaml:"LogLevel"`
	RitaLogPath string `yaml:"RitaLogPath"`
	LogToFile   bool   `yaml:"LogToFile"`
	LogToDB     bool   `yaml:"LogToDB"`
}

type LogTableCfg struct {
	RitaLogTable string
}

type MetaTableCfg struct {
	FilesTable     string
	DatabasesTable string
}

type MongoDBRunningCfg struct {
	AuthMechanismParsed mgosec.AuthMechanism
	TLS                 struct {
		TLSConfig *tls.Config
	}
}

type MongoDBStaticCfg struct {
	ConnectionString string        `yaml:"ConnectionString"`
	AuthMechanism    string        `yaml:"AuthenticationMechanism"`
	SocketTimeout    time.Duration `yaml:"SocketTimeout"`
	TLS              TLSStaticCfg  `yaml:"TLS"`
}

type RunningCfg struct {
	MongoDB MongoDBRunningCfg
	Version semver.Version
}

type SafeBrowsingStaticCfg struct {
	APIKey   string `yaml:"APIKey"`
	Database string `yaml:"Database"`
}

type ScanningStaticCfg struct {
	ScanThreshold int `yaml:"ScanThreshold"`
}

type ScanningTableCfg struct {
	ScanTable string
}

type StaticCfg struct {
	MongoDB      MongoDBStaticCfg     `yaml:"MongoDB"`
	Log          LogStaticCfg         `yaml:"LogConfig"`
	Blacklisted  BlacklistedStaticCfg `yaml:"BlackListed"`
	Crossref     CrossrefStaticCfg    `yaml:"Crossref"`
	Scanning     ScanningStaticCfg    `yaml:"Scanning"`
	Beacon       BeaconStaticCfg      `yaml:"Beacon"`
	Bro          BroStaticCfg         `yaml:"Bro"`
	Version      string
	ExactVersion string
}

type StructureTableCfg struct {
	ConnTable       string
	HTTPTable       string
	DNSTable        string
	UniqueConnTable string
	HostTable       string
	IPv4Table       string
	IPv6Table       string
}

type TLSStaticCfg struct {
	Enabled           bool   `yaml:"Enable"`
	VerifyCertificate bool   `yaml:"VerifyCertificate"`
	CAFile            string `yaml:"CAFile"`
}

type TableCfg struct {
	Log         LogTableCfg
	Blacklisted BlacklistedTableCfg
	DNS         DNSTableCfg
	Crossref    CrossrefTableCfg
	Scanning    ScanningTableCfg
	Structure   StructureTableCfg
	Beacon      BeaconTableCfg
	Urls        UrlsTableCfg
	UserAgent   UserAgentTableCfg
	Meta        MetaTableCfg
}

type UrlsTableCfg struct {
	UrlsTable string
}

type UserAgentTableCfg struct {
	UserAgentTable string
}