Revert flash cookie guard in router

gaby · gaby · commit 93b4bbe309e7 · 2025-11-04T23:51:03.000-05:00
diff --git a/router.go b/router.go
@@ -5,6 +5,7 @@
 package fiber
 
 import (
+	"bytes"
 	"fmt"
 	"slices"
 	"sync/atomic"
@@ -316,7 +317,8 @@ func (app *App) requestHandler(rctx *fasthttp.RequestCtx) {
 		}
 
 		// Optional: Check flash messages
-		if d.Cookies(FlashCookieName) != "" {
+		rawHeaders := d.Request().Header.RawHeaders()
+		if len(rawHeaders) > 0 && bytes.Contains(rawHeaders, []byte(FlashCookieName)) {
 			d.Redirect().parseAndClearFlashMessages()
 		}
 		_, err = app.next(d)
@@ -328,7 +330,8 @@ func (app *App) requestHandler(rctx *fasthttp.RequestCtx) {
 		}
 
 		// Optional: Check flash messages
-		if ctx.Cookies(FlashCookieName) != "" {
+		rawHeaders := ctx.Request().Header.RawHeaders()
+		if len(rawHeaders) > 0 && bytes.Contains(rawHeaders, []byte(FlashCookieName)) {
 			ctx.Redirect().parseAndClearFlashMessages()
 		}
 		_, err = app.nextCustom(ctx)

Original file line number	Diff line number	Diff line change
`@@ -5,6 +5,7 @@`
`5`	`5`	`package fiber`
`6`	`6`
`7`	`7`	`import (`
	`8`	`+ "bytes"`
`8`	`9`	`"fmt"`
`9`	`10`	`"slices"`
`10`	`11`	`"sync/atomic"`
`@@ -316,7 +317,8 @@ func (app App) requestHandler(rctx fasthttp.RequestCtx) {`
`316`	`317`	`}`
`317`	`318`
`318`	`319`	`// Optional: Check flash messages`
`319`		`- if d.Cookies(FlashCookieName) != "" {`
	`320`	`+ rawHeaders := d.Request().Header.RawHeaders()`
	`321`	`+ if len(rawHeaders) > 0 && bytes.Contains(rawHeaders, []byte(FlashCookieName)) {`
`320`	`322`	`d.Redirect().parseAndClearFlashMessages()`
`321`	`323`	`}`
`322`	`324`	`_, err = app.next(d)`
`@@ -328,7 +330,8 @@ func (app App) requestHandler(rctx fasthttp.RequestCtx) {`
`328`	`330`	`}`
`329`	`331`
`330`	`332`	`// Optional: Check flash messages`
`331`		`- if ctx.Cookies(FlashCookieName) != "" {`
	`333`	`+ rawHeaders := ctx.Request().Header.RawHeaders()`
	`334`	`+ if len(rawHeaders) > 0 && bytes.Contains(rawHeaders, []byte(FlashCookieName)) {`
`332`	`335`	`ctx.Redirect().parseAndClearFlashMessages()`
`333`	`336`	`}`
`334`	`337`	`_, err = app.nextCustom(ctx)`