feat: HTTP FLOOD Speed Limit

add shared memory
This commit is contained in:
Roi Feng
2025-04-09 20:52:40 -04:00
parent 4dad82eb40
commit b49c3573bb
14 changed files with 63 additions and 20 deletions

View File

@ -10,15 +10,15 @@ import (
)
// StartServer starts the HTTP server
func StartServer(cfg *config.MainConfig, ruleSet *config.RuleSet) error {
func StartServer(cfg *config.MainConfig, ruleSet *config.RuleSet, sharedMem *dataType.SharedMemory) error {
http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
userRequestData := processRequestData(cfg, r)
if strings.HasPrefix(userRequestData.Uri, cfg.WebPath) {
CheckTorii(w, r, userRequestData, ruleSet, cfg)
CheckTorii(w, r, userRequestData, ruleSet, cfg, sharedMem)
} else {
CheckMain(w, userRequestData, ruleSet, cfg)
CheckMain(w, userRequestData, ruleSet, cfg, sharedMem)
}
})