mirror of
https://github.com/Rayzggz/server_torii.git
synced 2025-06-19 13:41:23 +08:00
25 lines
574 B
Go
25 lines
574 B
Go
package check
|
|
|
|
import (
|
|
"net"
|
|
"server_torii/internal/action"
|
|
"server_torii/internal/config"
|
|
"server_torii/internal/dataType"
|
|
"server_torii/internal/utils"
|
|
)
|
|
|
|
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) {
|
|
utils.LogInfo(reqData, "", "IPBlockList")
|
|
decision.SetCode(action.Done, []byte("403"))
|
|
} else {
|
|
decision.Set(action.Continue)
|
|
}
|
|
}
|