starred/godoxy-yusing
1
0
Fork 0
mirror of https://github.com/yusing/godoxy.git synced 2026-04-23 00:38:33 +02:00
Code Issues 18 Packages Projects Releases 10 Wiki Activity

add host filter

Browse Source
This commit is contained in:
yusing
2024-12-18 11:44:38 +08:00
parent 34d5edd6b9
commit 36069cbe6d
2 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
internal/net/http/accesslog/config.go
View File

@@ -7,6 +7,7 @@ type (
Filters struct { Filters struct {
StatusCodes LogFilter[*StatusCodeRange] `json:"status_codes"` StatusCodes LogFilter[*StatusCodeRange] `json:"status_codes"`
Method LogFilter[HTTPMethod] `json:"method"` Method LogFilter[HTTPMethod] `json:"method"`
Host LogFilter[Host] `json:"host"`
Headers LogFilter[*HTTPHeader] `json:"headers"` // header exists or header == value Headers LogFilter[*HTTPHeader] `json:"headers"` // header exists or header == value
CIDR LogFilter[*CIDR] `json:"cidr"` CIDR LogFilter[*CIDR] `json:"cidr"`
} }

5
internal/net/http/accesslog/filter.go
View File

@@ -22,6 +22,7 @@ type (
HTTPHeader struct { HTTPHeader struct {
Key, Value string Key, Value string
} }
Host string
CIDR struct{ types.CIDR } CIDR struct{ types.CIDR }
) )
@@ -79,6 +80,10 @@ func (k *HTTPHeader) Fulfill(req *http.Request, res *http.Response) bool {
return false return false
} }
func (h Host) Fulfill(req *http.Request, res *http.Response) bool {
return req.Host == string(h)
}
func (cidr CIDR) Fulfill(req *http.Request, res *http.Response) bool { func (cidr CIDR) Fulfill(req *http.Request, res *http.Response) bool {
ip, _, err := net.SplitHostPort(req.RemoteAddr) ip, _, err := net.SplitHostPort(req.RemoteAddr)
if err != nil { if err != nil {