kilyabin
85 lines
2.0 KiB
Go
85 lines
2.0 KiB
Go
package config
|
|
|
|
import (
|
|
"os"
|
|
|
|
"gopkg.in/yaml.v3"
|
|
)
|
|
|
|
type Config struct {
|
|
Server ServerConfig `yaml:"server"`
|
|
Database DatabaseConfig `yaml:"database"`
|
|
Capture CaptureConfig `yaml:"capture"`
|
|
Detectors DetectorsConfig `yaml:"detectors"`
|
|
}
|
|
|
|
type ServerConfig struct {
|
|
Host string `yaml:"host"`
|
|
Port int `yaml:"port"`
|
|
}
|
|
|
|
type DatabaseConfig struct {
|
|
Path string `yaml:"path"`
|
|
}
|
|
|
|
type CaptureConfig struct {
|
|
DefaultInterface string `yaml:"default_interface"`
|
|
Snaplen int32 `yaml:"snaplen"`
|
|
Promiscuous bool `yaml:"promiscuous"`
|
|
}
|
|
|
|
type DetectorsConfig struct {
|
|
PortScan PortScanConfig `yaml:"port_scan"`
|
|
BruteForce BruteForceConfig `yaml:"brute_force"`
|
|
DoS DoSConfig `yaml:"dos"`
|
|
ARPSpoof ARPSpoofConfig `yaml:"arp_spoof"`
|
|
DNSAnomaly DNSAnomalyConfig `yaml:"dns_anomaly"`
|
|
}
|
|
|
|
type PortScanConfig struct {
|
|
Enabled bool `yaml:"enabled"`
|
|
Threshold int `yaml:"threshold"`
|
|
WindowSeconds int `yaml:"window_seconds"`
|
|
Severity string `yaml:"severity"`
|
|
}
|
|
|
|
type BruteForceConfig struct {
|
|
Enabled bool `yaml:"enabled"`
|
|
Threshold int `yaml:"threshold"`
|
|
WindowSeconds int `yaml:"window_seconds"`
|
|
Severity string `yaml:"severity"`
|
|
Ports []int32 `yaml:"ports"`
|
|
}
|
|
|
|
type DoSConfig struct {
|
|
Enabled bool `yaml:"enabled"`
|
|
PPSThreshold int `yaml:"pps_threshold"`
|
|
WindowSeconds int `yaml:"window_seconds"`
|
|
Severity string `yaml:"severity"`
|
|
}
|
|
|
|
type ARPSpoofConfig struct {
|
|
Enabled bool `yaml:"enabled"`
|
|
Severity string `yaml:"severity"`
|
|
}
|
|
|
|
type DNSAnomalyConfig struct {
|
|
Enabled bool `yaml:"enabled"`
|
|
MaxLabelLength int `yaml:"max_label_length"`
|
|
QueryThreshold int `yaml:"query_threshold"`
|
|
WindowSeconds int `yaml:"window_seconds"`
|
|
Severity string `yaml:"severity"`
|
|
}
|
|
|
|
func Load(path string) (*Config, error) {
|
|
data, err := os.ReadFile(path)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
var cfg Config
|
|
if err := yaml.Unmarshal(data, &cfg); err != nil {
|
|
return nil, err
|
|
}
|
|
return &cfg, nil
|
|
}
|