From 03175f0c5cd50a3651840e95b7e13d89934e10ae Mon Sep 17 00:00:00 2001
From: Adrien Pensart <crunchengine@gmail.com>
Date: Mon, 4 Sep 2023 12:57:09 +0000
Subject: [PATCH] Add: request_body replace directive
---
modules/caddyhttp/requestbody/caddyfile.go | 6 ++++++
modules/caddyhttp/requestbody/requestbody.go | 13 +++++++++++++
2 files changed, 19 insertions(+)
diff --git a/modules/caddyhttp/requestbody/caddyfile.go b/modules/caddyhttp/requestbody/caddyfile.go
index 8378ad7f4..e2382d546 100644
--- a/modules/caddyhttp/requestbody/caddyfile.go
+++ b/modules/caddyhttp/requestbody/caddyfile.go
@@ -68,6 +68,12 @@ func parseCaddyfile(h httpcaddyfile.Helper) (caddyhttp.MiddlewareHandler, error)
}
rb.WriteTimeout = timeout
+ case "set":
+ var setStr string
+ if !h.AllArgs(&setStr) {
+ return nil, h.ArgErr()
+ }
+ rb.Set = setStr
default:
return nil, h.Errf("unrecognized request_body subdirective '%s'", h.Val())
}
diff --git a/modules/caddyhttp/requestbody/requestbody.go b/modules/caddyhttp/requestbody/requestbody.go
index 830050416..1bf74c4cf 100644
--- a/modules/caddyhttp/requestbody/requestbody.go
+++ b/modules/caddyhttp/requestbody/requestbody.go
@@ -18,6 +18,7 @@ import (
"errors"
"io"
"net/http"
+ "strings"
"time"
"go.uber.org/zap"
@@ -43,6 +44,8 @@ type RequestBody struct {
// EXPERIMENTAL. Subject to change/removal.
WriteTimeout time.Duration `json:"write_timeout,omitempty"`
+ Set string `json:"set,omitempty"`
+
logger *zap.Logger
}
@@ -60,6 +63,16 @@ func (rb *RequestBody) Provision(ctx caddy.Context) error {
}
func (rb RequestBody) ServeHTTP(w http.ResponseWriter, r *http.Request, next caddyhttp.Handler) error {
+ if rb.Set != "" {
+ err := r.Body.Close()
+ if err != nil {
+ return err
+ }
+ repl := r.Context().Value(caddy.ReplacerCtxKey).(*caddy.Replacer)
+ replacedBody := repl.ReplaceAll(rb.Set, "")
+ r.Body = io.NopCloser(strings.NewReader(replacedBody))
+ r.ContentLength = int64(len(rb.Set))
+ }
if r.Body == nil {
return next.ServeHTTP(w, r)
}