-
Notifications
You must be signed in to change notification settings - Fork 29
/
test_request.go
86 lines (68 loc) · 1.59 KB
/
test_request.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
// Copyright 2021 Liuxiangchao iwind.liu@gmail.com. All rights reserved.
package requests
import (
"bytes"
"io"
"net"
"net/http"
)
type TestRequest struct {
req *http.Request
BodyData []byte
}
func NewTestRequest(raw *http.Request) *TestRequest {
return &TestRequest{
req: raw,
}
}
func (this *TestRequest) WAFSetCacheBody(bodyData []byte) {
this.BodyData = bodyData
}
func (this *TestRequest) WAFGetCacheBody() []byte {
return this.BodyData
}
func (this *TestRequest) WAFRaw() *http.Request {
return this.req
}
func (this *TestRequest) WAFRemoteAddr() string {
return this.req.RemoteAddr
}
func (this *TestRequest) WAFRemoteIP() string {
host, _, err := net.SplitHostPort(this.req.RemoteAddr)
if err != nil {
return this.req.RemoteAddr
} else {
return host
}
}
func (this *TestRequest) WAFReadBody(max int64) (data []byte, err error) {
if this.req.ContentLength > 0 {
data, err = io.ReadAll(io.LimitReader(this.req.Body, max))
}
return
}
func (this *TestRequest) WAFRestoreBody(data []byte) {
if len(data) > 0 {
rawReader := bytes.NewBuffer(data)
buf := make([]byte, 1024)
_, _ = io.CopyBuffer(rawReader, this.req.Body, buf)
this.req.Body = io.NopCloser(rawReader)
}
}
func (this *TestRequest) WAFServerId() int64 {
return 0
}
// WAFClose 关闭当前请求所在的连接
func (this *TestRequest) WAFClose() {
}
func (this *TestRequest) Format(s string) string {
return s
}
func (this *TestRequest) WAFOnAction(action interface{}) bool {
return true
}
func (this *TestRequest) WAFFingerprint() []byte {
return nil
}
func (this *TestRequest) DisableAccessLog() {
}