mirror of https://github.com/gofiber/fiber.git
53 lines
1.2 KiB
Go
53 lines
1.2 KiB
Go
package earlydata
|
|
|
|
import (
|
|
"github.com/gofiber/fiber/v3"
|
|
)
|
|
|
|
// The contextKey type is unexported to prevent collisions with context keys defined in
|
|
// other packages.
|
|
type contextKey int
|
|
|
|
const (
|
|
localsKeyAllowed contextKey = 0 // earlydata_allowed
|
|
)
|
|
|
|
// IsEarly returns true if the request is an early-data request
|
|
func IsEarly(c fiber.Ctx) bool {
|
|
return c.Locals(localsKeyAllowed) != nil
|
|
}
|
|
|
|
// New creates a new middleware handler
|
|
// https://datatracker.ietf.org/doc/html/rfc8470#section-5.1
|
|
func New(config ...Config) fiber.Handler {
|
|
// Set default config
|
|
cfg := configDefault(config...)
|
|
|
|
// Return new handler
|
|
return func(c fiber.Ctx) error {
|
|
// Don't execute middleware if Next returns true
|
|
if cfg.Next != nil && cfg.Next(c) {
|
|
return c.Next()
|
|
}
|
|
|
|
// Abort if we can't trust the early-data header
|
|
if !c.IsProxyTrusted() {
|
|
return cfg.Error
|
|
}
|
|
|
|
// Continue stack if request is not an early-data request
|
|
if !cfg.IsEarlyData(c) {
|
|
return c.Next()
|
|
}
|
|
|
|
// Continue stack if we allow early-data for this request
|
|
if cfg.AllowEarlyData(c) {
|
|
_ = c.Locals(localsKeyAllowed, true)
|
|
return c.Next()
|
|
}
|
|
|
|
// Else return our error
|
|
return cfg.Error
|
|
}
|
|
}
|