loginmon

type Action struct {
	Type     string        // Action type: "allow", "alert_only", "block_short", "block_long", "block_permanent"
	IP       string        // IP to act on
	Duration time.Duration // Ban duration (0 = permanent)
	Reason   string        // Why this action was taken
	Risk     float64       // Risk score that triggered this action
	Event    *LoginEvent   // The login event that caused this action
}

type Config struct {
	// Module control
	Enabled bool // Whether the module is enabled (default: true)

	// Ban thresholds - score needed to trigger ban (scale: 0-100)
	// Example: ThresholdTempBan=45 means ban when score >= 45
	ThresholdTempBan   int32 // Score to trigger temp ban (default: 45)
	ThresholdEscalate  int32 // Score to escalate ban (default: 65)
	ThresholdPermanent int32 // Score for permanent ban (default: 100)

	// Ban durations
	TempBanDuration   time.Duration   // Initial temp ban (default: 15m)
	EscalateDurations []time.Duration // Escalation ladder: 2h, 4h, 12h, 24h

	// Score decay (reduces score over time if no new failures)
	ScoreDecayInterval time.Duration // How often to decay scores (default: 5m)
	ScoreDecayAmount   int32         // Points to decay per interval (default: 5)

	// IP retention (how long to remember an IP after last activity)
	IPRetentionDuration time.Duration // How long to keep IP data (default: 24h)

	// SSH scores
	SSHFailedPassword int16 // Score for "Failed password" (default: 10)
	SSHInvalidUser    int16 // Score for invalid user attempt (default: 15)
	SSHPreauth        int16 // Score for preauth disconnect (default: 20)
	SSHTooMany        int16 // Score for "too many failures" (default: 30)
	SSHRootAttempt    int16 // Bonus score for root attempts (default: 10)

	// Mail scores
	DovecotAuthFail int16 // Score for Dovecot auth failure (default: 15)
	PostfixSASL     int16 // Score for Postfix SASL failure (default: 15)
	EximAuthFail    int16 // Score for Exim auth failure (default: 15)

	// FTP scores
	FTPAuthFail int16 // Score for FTP auth failure (default: 15)

	// Panel scores
	DirectAdminLogin int16 // Score for DirectAdmin login failure (default: 20)
	CPanelLogin      int16 // Score for cPanel login failure (default: 20)
	PleskLogin       int16 // Score for Plesk login failure (default: 20)

	// WordPress scores
	WordPressXMLRPC  int16 // Score for WordPress xmlrpc attack (default: 25)
	WordPressWPLogin int16 // Score for wp-login brute force (default: 20)

	// Risk thresholds
	BlockHighRisk   float64 // Block if risk >= this (default: 0.8)
	BlockMediumRisk float64 // Block if risk >= this (default: 0.6)
	AlertRisk       float64 // Alert if risk >= this (default: 0.4)

	// Ban durations (risk-based)
	HighRiskDuration   time.Duration // Ban duration for high risk (default: 24h)
	MediumRiskDuration time.Duration // Ban duration for medium risk (default: 1h)
	LowRiskDuration    time.Duration // Ban duration for low risk (default: 10m)

	// Behavioral analysis weights
	NewIPWeight         float64 // Risk weight for new IP (default: 0.3)
	NewCountryWeight    float64 // Risk weight for new country (default: 0.3)
	BadReputationWeight float64 // Risk weight for bad reputation (default: 0.4)
	MultipleUsersWeight float64 // Risk weight for multiple users from same IP (default: 0.2)

	// Time windows
	FailureWindow    time.Duration // Window to count failed attempts (default: 10m)
	ProfileRetention time.Duration // How long to keep user profiles (default: 30d)

	// Thresholds
	MaxFailedAttempts int // Max failed attempts before blocking (default: 5)
	MaxUsersPerIP     int // Max different users from same IP before suspicious (default: 3)
}

type EVEEvent struct {
	Timestamp string `json:"timestamp"`
	EventType string `json:"event_type"`
	SrcIP     string `json:"src_ip"`
	DestIP    string `json:"dest_ip"`
	SrcPort   int    `json:"src_port"`
	DestPort  int    `json:"dest_port"`
	Alert     *struct {
		Action      string `json:"action"`
		Category    string `json:"category"`
		Signature   string `json:"signature"`
		SignatureID int    `json:"signature_id"`
	} `json:"alert,omitempty"`
	SSH *struct {
		Client string `json:"client"`
		Server string `json:"server"`
	} `json:"ssh,omitempty"`
}

type GeoInfo struct {
	Country     string // ISO country code (e.g., "US", "CN", "RU")
	CountryName string // Human-readable country name
	City        string // City name (optional)
	ASN         string // Autonomous System Number (ISP identifier)
	ASNOrg      string // Organization name for the ASN
	IsProxy     bool   // Known proxy/VPN/Tor exit node
	IsTor       bool   // Known Tor exit node
	IsHosting   bool   // Hosting provider (often used for bots)
}

type IPProfile struct {
	IP           string               // IP address
	Users        map[string]time.Time // All users that logged in from this IP (user -> last seen)
	FailedLogins int                  // Count of recent failed attempts from this IP
	TotalLogins  int                  // Total successful logins from this IP
	LastCountry  string               // Last known country for this IP
	LastASN      string               // Last known ASN for this IP
	FirstSeen    time.Time            // When we first saw this IP
	LastActivity time.Time            // Last login attempt (success or fail)
}

type LoginEvent struct {
	User      string            // Username that logged in
	IP        string            // Source IP address
	Time      time.Time         // When the login occurred
	Service   string            // Service name (ssh, directadmin, webmail, panel, etc.)
	Success   bool              // true = successful login, false = failed attempt
	Method    string            // Auth method (password, publickey, session, etc.)
	Extra     map[string]string // Additional metadata (port, protocol, etc.)
	Timestamp int64             // Unix timestamp for correlation
}

type LoginProfile struct {
	User          string               // Username
	LastIP        string               // Last IP this user logged in from
	LastCountry   string               // Last country (from GeoIP)
	LastASN       string               // Last ASN (ISP/provider)
	LastLoginTime time.Time            // Last successful login timestamp
	KnownIPs      map[string]time.Time // All IPs this user has logged in from (IP -> last seen)
	FailedLogins  int                  // Count of recent failed attempts
	TotalLogins   int                  // Total successful logins
	FirstSeen     time.Time            // When we first saw this user
}

type LoginState struct {
	Users map[string]*LoginProfile // User -> profile
	IPs   map[string]*IPProfile    // IP -> profile
	// contains filtered or unexported fields
}

type Mode string

type Module struct {
	// contains filtered or unexported fields
}

type ReputationInfo struct {
	BadFeeds      int       // Number of threat feeds containing this IP
	SuricataHits  int       // Number of Suricata alerts for this IP
	NFTBanHistory int       // Number of times previously banned by NFTBan
	LastBanTime   time.Time // When this IP was last banned
	BanReason     string    // Reason for last ban (ssh_brute, web_exploit, etc.)
	IsWhitelisted bool      // Is this IP in the whitelist?
}

type RiskScore struct {
	Score       float64            // Overall risk score (0.0 = safe, 1.0 = maximum risk)
	Factors     map[string]float64 // Individual risk factors and their contributions
	Reason      string             // Human-readable explanation
	Recommended string             // Recommended action (allow, alert, block_short, block_long, block_permanent)
}