2024-05-17 18:30:33 +08:00
|
|
|
// Copyright 2021 GoEdge goedge.cdn@gmail.com. All rights reserved.
|
2021-10-25 19:42:12 +08:00
|
|
|
|
|
|
|
|
package waf
|
|
|
|
|
|
|
|
|
|
import (
|
|
|
|
|
"net/http"
|
2024-07-27 15:42:50 +08:00
|
|
|
|
|
|
|
|
"github.com/TeaOSLab/EdgeNode/internal/waf/requests"
|
2021-10-25 19:42:12 +08:00
|
|
|
)
|
|
|
|
|
|
|
|
|
|
type PageAction struct {
|
|
|
|
|
BaseAction
|
|
|
|
|
|
2024-01-20 16:18:07 +08:00
|
|
|
UseDefault bool `yaml:"useDefault" json:"useDefault"`
|
|
|
|
|
Status int `yaml:"status" json:"status"`
|
|
|
|
|
Body string `yaml:"body" json:"body"`
|
2021-10-25 19:42:12 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func (this *PageAction) Init(waf *WAF) error {
|
2024-01-20 16:18:07 +08:00
|
|
|
if waf.DefaultPageAction != nil {
|
|
|
|
|
if this.Status <= 0 || this.UseDefault {
|
|
|
|
|
this.Status = waf.DefaultPageAction.Status
|
|
|
|
|
}
|
|
|
|
|
if len(this.Body) == 0 || this.UseDefault {
|
|
|
|
|
this.Body = waf.DefaultPageAction.Body
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2023-05-28 17:11:33 +08:00
|
|
|
if this.Status <= 0 {
|
|
|
|
|
this.Status = http.StatusForbidden
|
|
|
|
|
}
|
2021-10-25 19:42:12 +08:00
|
|
|
return nil
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func (this *PageAction) Code() string {
|
|
|
|
|
return ActionPage
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func (this *PageAction) IsAttack() bool {
|
|
|
|
|
return false
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// WillChange determine if the action will change the request
|
|
|
|
|
func (this *PageAction) WillChange() bool {
|
|
|
|
|
return true
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Perform the action
|
2024-01-20 20:54:41 +08:00
|
|
|
func (this *PageAction) Perform(waf *WAF, group *RuleGroup, set *RuleSet, request requests.Request, writer http.ResponseWriter) PerformResult {
|
2023-12-09 17:00:21 +08:00
|
|
|
if writer == nil {
|
2024-01-20 20:54:41 +08:00
|
|
|
return PerformResult{}
|
2023-12-09 17:00:21 +08:00
|
|
|
}
|
|
|
|
|
|
2023-06-11 10:46:20 +08:00
|
|
|
request.ProcessResponseHeaders(writer.Header(), this.Status)
|
2021-10-25 19:42:12 +08:00
|
|
|
writer.Header().Set("Content-Type", "text/html; charset=utf-8")
|
|
|
|
|
writer.WriteHeader(this.Status)
|
2023-12-09 15:55:40 +08:00
|
|
|
|
|
|
|
|
var body = this.Body
|
|
|
|
|
if len(body) == 0 {
|
|
|
|
|
body = `<!DOCTYPE html>
|
|
|
|
|
<html lang="en">
|
2024-01-20 10:14:15 +08:00
|
|
|
<head>
|
|
|
|
|
<title>403 Forbidden</title>
|
2023-12-09 15:55:40 +08:00
|
|
|
<style>
|
|
|
|
|
address { line-height: 1.8; }
|
|
|
|
|
</style>
|
2024-01-20 10:14:15 +08:00
|
|
|
</head>
|
2023-12-09 15:55:40 +08:00
|
|
|
<body>
|
|
|
|
|
<h1>403 Forbidden By WAF</h1>
|
|
|
|
|
<address>Connection: ${remoteAddr} (Client) -> ${serverAddr} (Server)</address>
|
|
|
|
|
<address>Request ID: ${requestId}</address>
|
|
|
|
|
</body>
|
|
|
|
|
</html>`
|
|
|
|
|
}
|
|
|
|
|
_, _ = writer.Write([]byte(request.Format(body)))
|
2021-10-25 19:42:12 +08:00
|
|
|
|
2024-01-20 20:54:41 +08:00
|
|
|
return PerformResult{}
|
2021-10-25 19:42:12 +08:00
|
|
|
}
|
