Adding upstream version 2.52.6.

Signed-off-by: Daniel Baumann <daniel@debian.org>

middleware/etag/etag.go

@@ -0,0 +1,116 @@
+package etag
+
+import (
+	"bytes"
+	"hash/crc32"
+
+	"github.com/gofiber/fiber/v2"
+
+	"github.com/valyala/bytebufferpool"
+)
+
+// New creates a new middleware handler
+func New(config ...Config) fiber.Handler {
+	// Set default config
+	cfg := configDefault(config...)
+
+	var (
+		normalizedHeaderETag = []byte("Etag")
+		weakPrefix           = []byte("W/")
+	)
+
+	const crcPol = 0xD5828281
+	crc32q := crc32.MakeTable(crcPol)
+
+	// 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()
+		}
+
+		// Return err if next handler returns one
+		if err := c.Next(); err != nil {
+			return err
+		}
+
+		// Don't generate ETags for invalid responses
+		if c.Response().StatusCode() != fiber.StatusOK {
+			return nil
+		}
+		body := c.Response().Body()
+		// Skips ETag if no response body is present
+		if len(body) == 0 {
+			return nil
+		}
+		// Skip ETag if header is already present
+		if c.Response().Header.PeekBytes(normalizedHeaderETag) != nil {
+			return nil
+		}
+
+		// Generate ETag for response
+		bb := bytebufferpool.Get()
+		defer bytebufferpool.Put(bb)
+
+		// Enable weak tag
+		if cfg.Weak {
+			_, _ = bb.Write(weakPrefix) //nolint:errcheck // This will never fail
+		}
+
+		_ = bb.WriteByte('"') //nolint:errcheck // This will never fail
+		bb.B = appendUint(bb.Bytes(), uint32(len(body)))
+		_ = bb.WriteByte('-') //nolint:errcheck // This will never fail
+		bb.B = appendUint(bb.Bytes(), crc32.Checksum(body, crc32q))
+		_ = bb.WriteByte('"') //nolint:errcheck // This will never fail
+
+		etag := bb.Bytes()
+
+		// Get ETag header from request
+		clientEtag := c.Request().Header.Peek(fiber.HeaderIfNoneMatch)
+
+		// Check if client's ETag is weak
+		if bytes.HasPrefix(clientEtag, weakPrefix) {
+			// Check if server's ETag is weak
+			if bytes.Equal(clientEtag[2:], etag) || bytes.Equal(clientEtag[2:], etag[2:]) {
+				// W/1 == 1 || W/1 == W/1
+				c.Context().ResetBody()
+
+				return c.SendStatus(fiber.StatusNotModified)
+			}
+			// W/1 != W/2 || W/1 != 2
+			c.Response().Header.SetCanonical(normalizedHeaderETag, etag)
+
+			return nil
+		}
+
+		if bytes.Contains(clientEtag, etag) {
+			// 1 == 1
+			c.Context().ResetBody()
+
+			return c.SendStatus(fiber.StatusNotModified)
+		}
+		// 1 != 2
+		c.Response().Header.SetCanonical(normalizedHeaderETag, etag)
+
+		return nil
+	}
+}
+
+// appendUint appends n to dst and returns the extended dst.
+func appendUint(dst []byte, n uint32) []byte {
+	var b [20]byte
+	buf := b[:]
+	i := len(buf)
+	var q uint32
+	for n >= 10 {
+		i--
+		q = n / 10
+		buf[i] = '0' + byte(n-q*10)
+		n = q
+	}
+	i--
+	buf[i] = '0' + byte(n)
+
+	dst = append(dst, buf[i:]...)
+	return dst
+}