selfpost/internal/web/handlers/handlers_ratelimit.go

package handlers

import (
	"fmt"
	"net"
	"net/http"
	"strconv"
	"strings"

	"github.com/mixeme/selfpost/internal/store"
)

const defaultRateLimitWindowSeconds = 3600

type rateLimitInput struct {
	clear          bool
	mode           string
	ips            []string
	maxMessages    int
	windowSeconds  int
	autoMultiplier float64
}

func (h *Handlers) l1Messages() int {
	if h.cfg.RateLimitMessagesPerIP > 0 {
		return h.cfg.RateLimitMessagesPerIP
	}
	return 100
}

func (h *Handlers) l1Window() int {
	if h.cfg.RateLimitWindowSeconds > 0 {
		return h.cfg.RateLimitWindowSeconds
	}
	return defaultRateLimitWindowSeconds
}

func parseAutoMultiplier(raw string) (float64, error) {
	raw = strings.TrimSpace(raw)
	if raw == "" {
		return store.DefaultAutoMultiplier, nil
	}
	v, err := strconv.ParseFloat(raw, 64)
	if err != nil {
		return 0, fmt.Errorf("enter a valid multiplier (%.1f–%.1f)", store.MinAutoMultiplier, store.MaxAutoMultiplier)
	}
	if v < store.MinAutoMultiplier || v > store.MaxAutoMultiplier {
		return 0, fmt.Errorf("multiplier must be between %.1f and %.1f", store.MinAutoMultiplier, store.MaxAutoMultiplier)
	}
	return v, nil
}

func parseRateLimitMode(raw string) (string, error) {
	raw = strings.TrimSpace(raw)
	if raw == "" {
		return store.RateLimitModeManual, nil
	}
	if raw != store.RateLimitModeManual && raw != store.RateLimitModeAuto {
		return "", fmt.Errorf("choose manual or auto mode")
	}
	return raw, nil
}

func parseRateLimitForm(r *http.Request, l1Max int) (rateLimitInput, error) {
	if err := r.ParseForm(); err != nil {
		return rateLimitInput{}, fmt.Errorf("invalid form submission")
	}
	if r.PostFormValue("clear") != "" {
		return rateLimitInput{clear: true}, nil
	}
	mode, err := parseRateLimitMode(r.PostFormValue("mode"))
	if err != nil {
		return rateLimitInput{}, err
	}
	if mode == store.RateLimitModeAuto {
		mult, err := parseAutoMultiplier(r.PostFormValue("auto_multiplier"))
		if err != nil {
			return rateLimitInput{}, err
		}
		return rateLimitInput{mode: mode, autoMultiplier: mult}, nil
	}

	rawMax := strings.TrimSpace(r.PostFormValue("max_messages"))
	if rawMax == "" {
		return rateLimitInput{clear: true}, nil
	}
	maxMessages, err := parsePositiveInt(rawMax, 0)
	if err != nil || maxMessages <= 0 {
		return rateLimitInput{}, fmt.Errorf("enter a message limit greater than zero")
	}
	if maxMessages > l1Max {
		return rateLimitInput{}, fmt.Errorf("message limit cannot exceed the level-1 backstop (%d)", l1Max)
	}
	windowSeconds, err := parsePositiveInt(r.PostFormValue("window_seconds"), defaultRateLimitWindowSeconds)
	if err != nil || windowSeconds <= 0 {
		return rateLimitInput{}, fmt.Errorf("enter a time window greater than zero seconds")
	}
	return rateLimitInput{mode: store.RateLimitModeManual, maxMessages: maxMessages, windowSeconds: windowSeconds}, nil
}

func parseAppAuthIPsForm(r *http.Request) (bool, []string, error) {
	if err := r.ParseForm(); err != nil {
		return false, nil, fmt.Errorf("invalid form submission")
	}
	restrict := r.PostFormValue("auth_ip_restrict") != ""
	if !restrict {
		return false, nil, nil
	}
	ips, err := parseIPList(r.PostFormValue("auth_allowed_ips"))
	if err != nil {
		return false, nil, err
	}
	if len(ips) == 0 {
		return false, nil, fmt.Errorf("enter at least one client IP when the allow-list is enabled")
	}
	return true, ips, nil
}

func parseIPList(raw string) ([]string, error) {
	fields := strings.FieldsFunc(raw, func(r rune) bool {
		return r == '\n' || r == '\r' || r == ',' || r == ' ' || r == '\t' || r == ';'
	})
	var out []string
	seen := make(map[string]bool)
	for _, f := range fields {
		ip := net.ParseIP(f)
		if ip == nil {
			return nil, fmt.Errorf("%q is not a valid IP address", f)
		}
		c := ip.String()
		if !seen[c] {
			seen[c] = true
			out = append(out, c)
		}
	}
	return out, nil
}

func parsePositiveInt(raw string, def int) (int, error) {
	raw = strings.TrimSpace(raw)
	if raw == "" {
		return def, nil
	}
	return strconv.Atoi(raw)
}

func (h *Handlers) HandleDomainRateLimit(w http.ResponseWriter, r *http.Request) {
	d, ok := h.lookupDomain(w, r)
	if !ok {
		return
	}
	in, err := parseRateLimitForm(r, h.l1Messages())
	if err != nil {
		h.renderDomainSettings(w, r, http.StatusBadRequest, d, settingsView{Err: err.Error()})
		return
	}
	if err := h.applyRateLimit(in, store.RateLimitScopeDomain, d.ID, h.domains); err != nil {
		logf("panel: domain %d: save rate limit: %v", d.ID, err)
		http.Error(w, "internal error", http.StatusInternalServerError)
		return
	}
	http.Redirect(w, r, fmt.Sprintf("/outbound/domains/%d?ratelimit=1", d.ID), http.StatusSeeOther)
}

func (h *Handlers) HandleDomainRateLimitRecalc(w http.ResponseWriter, r *http.Request) {
	d, ok := h.lookupDomain(w, r)
	if !ok {
		return
	}
	if err := h.recalcRateLimit(store.RateLimitScopeDomain, d.ID); err != nil {
		logf("panel: domain %d: recalc rate limit: %v", d.ID, err)
		h.renderDomainSettings(w, r, http.StatusBadRequest, d, settingsView{Err: err.Error()})
		return
	}
	http.Redirect(w, r, fmt.Sprintf("/outbound/domains/%d?recalculated=1", d.ID), http.StatusSeeOther)
}

func (h *Handlers) HandleAppRateLimitRecalc(w http.ResponseWriter, r *http.Request) {
	a, d, ok := h.lookupDomainApplication(w, r)
	if !ok {
		return
	}
	if err := h.recalcRateLimit(store.RateLimitScopeApp, a.ID); err != nil {
		h.renderApplicationForm(w, r, http.StatusBadRequest, d, &a, nil, fmt.Sprintf("%s: %s", a.Login, err.Error()))
		return
	}
	http.Redirect(w, r, fmt.Sprintf("/outbound/domains/%d?recalculated=1", a.DomainID), http.StatusSeeOther)
}

func (h *Handlers) recalcRateLimit(scope string, refID int64) error {
	return h.store.RecalcAutoRateLimit(scope, refID, h.sendLogRetentionDays(), h.l1Messages(), h.l1Window())
}

type rateLimitSaver interface {
	ClearRateLimit(id int64) error
	SaveRateLimit(id int64, rl store.RateLimit) error
}

func (h *Handlers) applyRateLimit(in rateLimitInput, scope string, refID int64, saver rateLimitSaver) error {
	if in.clear {
		return saver.ClearRateLimit(refID)
	}
	rl := store.RateLimit{
		Scope:          scope,
		RefID:          refID,
		Mode:           in.mode,
		MaxMessages:    in.maxMessages,
		WindowSeconds:  in.windowSeconds,
		AutoMultiplier: in.autoMultiplier,
	}
	if in.mode == store.RateLimitModeAuto {
		rl.WindowSeconds = h.l1Window()
		if err := saver.SaveRateLimit(refID, rl); err != nil {
			return err
		}
		return h.recalcRateLimit(scope, refID)
	}
	return saver.SaveRateLimit(refID, rl)
}