config

type BlacklistedCfg struct {
	ThreadCount    int    `yaml:"ThreadCount"`
	ChannelSize    int    `yaml:"ChannelSize"`
	BlacklistTable string `yaml:"BlackListTable"`
}

type BroCfg struct {
	LogPath      string            `yaml:"LogPath"`
	DBPrefix     string            `yaml:"DBPrefix"`
	MetaDB       string            `yaml:"MetaDB"`
	BufferSize   int               `yaml:"BufferSize"`
	WriteThreads int               `yaml:"WriteThreads"`
	DirectoryMap map[string]string `yaml:"DirectoryMap"`
	FilesTable   string            `yaml:"FilesTable"`
	UseDates     bool              `yaml:"UseDates"`
}

type DnsCfg struct {
	DnsTable string `yaml:"DnsTable"`
}

type DurationCfg struct {
	DurationTimeScale string `yaml:"DurationTimeScale"`
}

type Resources struct {
	System  SystemConfig
	Session *mgo.Session
	Log     *log.Logger
}

type ScanningCfg struct {
	ScanThreshold int    `yaml:"ScanThreshold"`
	ScanTable     string `yaml:"ScanTable"`
}

type StructureCfg struct {
	ConnTable       string `yaml:"ConnectionTable"`
	HttpTable       string `yaml:"HttpTable"`
	UniqueConnTable string `yaml:"UniqueConnectionTable"`
	HostTable       string `yaml:"HostTable"`
}

type SystemConfig struct {
	LogType                 string         `yaml:"LogType"`
	GNUNetcatPath           string         `yaml:"GNUNetcatPath"`
	BaseInstallDirectory    string         `yaml:"BaseInstallDirectory"`
	BatchSize               int            `yaml:"BatchSize"`
	DB                      string         `yaml:"Database"`
	HostIntelDB             string         `yaml:"HostIntelDB"`
	ExternalHostsCollection string         `yaml:"ExternalHostsCollection"`
	DatabaseHost            string         `yaml:"DatabaseHost"`
	LogLevel                int            `yaml:"LogLevel"`
	Prefetch                float64        `yaml:"Prefetch"`
	Whitelist               []string       `yaml:"WhiteList"`
	BlacklistedConfig       BlacklistedCfg `yaml:"BlackListed"`
	DnsConfig               DnsCfg         `yaml:"Dns"`
	DurationConfig          DurationCfg    `yaml:"Duration"`
	ScanningConfig          ScanningCfg    `yaml:"Scanning"`
	StructureConfig         StructureCfg   `yaml:"Structure"`
	TBDConfig               TBDCfg         `yaml:"TBD"`
	UrlsConfig              UrlsCfg        `yaml:"Urls"`
	UserAgentConfig         UserAgentCfg   `yaml:"UserAgent"`
	BroConfig               BroCfg         `yaml:"Bro"`
}

type TBDCfg struct {
	DefaultBucketSize       float64 `yaml:"DefaultBucketSize"`
	DefaultConnectionThresh int     `yaml:"DefaultConnectionThresh"`
	TBDTable                string  `yaml:"TBDTable"`
}

type UrlsCfg struct {
	UrlsTable      string `yaml:"UrlsTable"`
	HostnamesTable string `yaml:"HostnamesTable"`
}

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