mirror of
https://github.com/Rayzggz/server_torii.git
synced 2025-06-17 04:31:22 +08:00
23 lines
499 B
Go
23 lines
499 B
Go
package check
|
|
|
|
import (
|
|
"net"
|
|
"server_torii/internal/action"
|
|
"server_torii/internal/config"
|
|
"server_torii/internal/dataType"
|
|
)
|
|
|
|
func IPBlockList(reqData dataType.UserRequest, ruleSet *config.RuleSet, decision *action.Decision, sharedMem *dataType.SharedMemory) {
|
|
remoteIP := reqData.RemoteIP
|
|
trie := ruleSet.IPBlockTrie
|
|
ip := net.ParseIP(remoteIP)
|
|
if ip == nil {
|
|
return
|
|
}
|
|
if trie.Search(ip) {
|
|
decision.SetCode(action.Done, []byte("403"))
|
|
} else {
|
|
decision.Set(action.Continue)
|
|
}
|
|
}
|