Fix improper memory reuse in NewFastHTTPHandler

fasthttpadaptor/adaptor_test.go

@@ -139,6 +139,53 @@ func TestNewFastHTTPHandler(t *testing.T) {
  }
 }
 
+func TestNewFastHTTPHandlerWithCookies(t *testing.T) {
+ expectedMethod := fasthttp.MethodPost
+ expectedRequestURI := "/foo/bar?baz=123"
+ expectedHost := "foobar.com"
+ expectedRemoteAddr := "1.2.3.4:6789"
+
+ var ctx fasthttp.RequestCtx
+ var req fasthttp.Request
+
+ req.Header.SetMethod(expectedMethod)
+ req.SetRequestURI(expectedRequestURI)
+ req.Header.SetHost(expectedHost)
+ req.Header.SetCookie("cookieOne", "valueCookieOne")
+ req.Header.SetCookie("cookieTwo", "valueCookieTwo")
+
+ remoteAddr, err := net.ResolveTCPAddr("tcp", expectedRemoteAddr)
+ if err != nil {
+ t.Fatalf("unexpected error: %v", err)
+ }
+ ctx.Init(&req, remoteAddr, nil)
+
+ nethttpH := func(w http.ResponseWriter, r *http.Request) {
+ // real handler warped by middleware, in this example do nothing
+ }
+ fasthttpH := NewFastHTTPHandler(http.HandlerFunc(nethttpH))
+
+ netMiddleware := func(_ http.ResponseWriter, r *http.Request) {
+ // assume middleware do some change on r, such as reset header's host
+ r.Header.Set("Host", "example.com")
+ // convert ctx again in case request may modify by middleware
+ ctx.Request.Header.Set("Host", r.Header.Get("Host"))
+ // since cookies of r are not changed, expect "cookieOne=valueCookieOne"
+ cookie, _ := r.Cookie("cookieOne")
+ if err != nil {
+ // will error, but if line 172 is commented, then no error will happen
+ t.Errorf("should not error")
+ }
+ if cookie.Value != "valueCookieOne" {
+ t.Errorf("cookie error, expect %s, find %s", "valueCookieOne", cookie.Value)
+ }
+ // instead of using responseWriter and r, use ctx again, like what have done in fiber
+ fasthttpH(&ctx)
+ }
+ fastMiddleware := NewFastHTTPHandler(http.HandlerFunc(netMiddleware))
+ fastMiddleware(&ctx)
+}
+
 func setContextValueMiddleware(next fasthttp.RequestHandler, key string, value any) fasthttp.RequestHandler {
  return func(ctx *fasthttp.RequestCtx) {
  ctx.SetUserValue(key, value)

fasthttpadaptor/request.go

@@ -5,6 +5,7 @@ import (
  "io"
  "net/http"
  "net/url"
+ "strings"
 
  "github.com/valyala/fasthttp"
 )
@@ -58,6 +59,9 @@ func ConvertRequest(ctx *fasthttp.RequestCtx, r *http.Request, forServer bool) e
  case "Transfer-Encoding":
  r.TransferEncoding = append(r.TransferEncoding, sv)
  default:
+ if sk == fasthttp.HeaderCookie {
+ sv = strings.Clone(sv)
+ }
  r.Header.Set(sk, sv)
  }
  })