config

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

type BeaconTableCfg struct {
	BeaconTable string `yaml:"BeaconTable"`
}

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 `yaml:"Database"`
	SourceIPsTable    string `yaml:"SourceIPsTable"`
	DestIPsTable      string `yaml:"DestIPsTable"`
	HostnamesTable    string `yaml:"HostnamesTable"`
	UrlsTable         string `yaml:"UrlsTable"`
}

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 `yaml:"SourceTable"`
	DestTable   string `yaml:"DestinationTable"`
}

type DNSTableCfg struct {
	ExplodedDNSTable string `yaml:"ExplodedDnsTable"`
	HostnamesTable   string `yaml:"HostnamesTable"`
}

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

type LogTableCfg struct {
	RitaLogTable string `yaml:"RitaLogTable"`
}

type MetaTableCfg struct {
	FilesTable     string `yaml:"FilesTable"`
	DatabasesTable string `yaml:"DatabasesTable"`
}

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 `yaml:"ScanTable"`
}

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 `yaml:"ConnectionTable"`
	HTTPTable       string `yaml:"HttpTable"`
	DNSTable        string `yaml:"DnsTable"`
	UniqueConnTable string `yaml:"UniqueConnectionTable"`
	HostTable       string `yaml:"HostTable"`
	IPv4Table       string `yaml:"IPv4Table"`
	IPv6Table       string `yaml:"IPv6Table"`
}

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

type TableCfg struct {
	Log         LogTableCfg         `yaml:"LogConfig"`
	Blacklisted BlacklistedTableCfg `yaml:"BlackListed"`
	DNS         DNSTableCfg         `yaml:"Dns"`
	Crossref    CrossrefTableCfg    `yaml:"Crossref"`
	Scanning    ScanningTableCfg    `yaml:"Scanning"`
	Structure   StructureTableCfg   `yaml:"Structure"`
	Beacon      BeaconTableCfg      `yaml:"Beacon"`
	Urls        UrlsTableCfg        `yaml:"Urls"`
	UserAgent   UserAgentTableCfg   `yaml:"UserAgent"`
	Meta        MetaTableCfg        `yaml:"MetaTables"`
}

type UrlsTableCfg struct {
	UrlsTable string `yaml:"UrlsTable"`
}

type UserAgentTableCfg struct {
	UserAgentTable string `yaml:"UserAgentTable"`
}