diff --git a/wafenginecore/checkcc.go b/wafenginecore/checkcc.go index b22ab7e..a162cd1 100644 --- a/wafenginecore/checkcc.go +++ b/wafenginecore/checkcc.go @@ -9,7 +9,7 @@ import ( /* * -检测xss +检测cc */ func (waf *WafEngine) CheckCC(weblogbean *innerbean.WebLog, formValue url.Values) detection.Result { result := detection.Result{ diff --git a/wafenginecore/wafengine.go b/wafenginecore/wafengine.go index e9d3bc3..d88df80 100644 --- a/wafenginecore/wafengine.go +++ b/wafenginecore/wafengine.go @@ -87,7 +87,11 @@ func (waf *WafEngine) ServeHTTP(w http.ResponseWriter, r *http.Request) { //检测如果访问IP和远程IP是同一个IP,且远程端口在本地Server已存在则显示配置错误 if ipAndPort[0] == waf.HostTarget[host].Host.Remote_ip && ok == true { resBytes := []byte("500: 配置有误" + host + " 当前IP和访问远端IP一样,且端口也一样,会造成循环问题") - w.Write(resBytes) + _, err := w.Write(resBytes) + if err != nil { + zlog.Debug("write fail:", zap.Any("", err)) + return + } return } @@ -252,7 +256,11 @@ func (waf *WafEngine) ServeHTTP(w http.ResponseWriter, r *http.Request) { return } else { resBytes := []byte("403: Host forbidden " + host) - w.Write(resBytes) + _, err := w.Write(resBytes) + if err != nil { + zlog.Debug("write fail:", zap.Any("", err)) + return + } // 获取请求报文的内容长度 contentLength := r.ContentLength @@ -322,7 +330,11 @@ func EchoErrorInfo(w http.ResponseWriter, r *http.Request, weblogbean innerbean. resBytes := []byte("