mirror of
https://codeberg.org/forgejo/forgejo
synced 2024-11-29 05:06:11 +01:00
d1353e1f7c
* update code.gitea.io/sdk/gitea v0.13.1 -> v0.13.2 * update github.com/go-swagger/go-swagger v0.25.0 -> v0.26.0 * update github.com/google/uuid v1.1.2 -> v1.2.0 * update github.com/klauspost/compress v1.11.3 -> v1.11.7 * update github.com/lib/pq 083382b7e6fc -> v1.9.0 * update github.com/markbates/goth v1.65.0 -> v1.66.1 * update github.com/mattn/go-sqlite3 v1.14.4 -> v1.14.6 * update github.com/mgechev/revive 246eac737dc7 -> v1.0.3 * update github.com/minio/minio-go/v7 v7.0.6 -> v7.0.7 * update github.com/niklasfasching/go-org v1.3.2 -> v1.4.0 * update github.com/olivere/elastic/v7 v7.0.21 -> v7.0.22 * update github.com/pquerna/otp v1.2.0 -> v1.3.0 * update github.com/xanzy/go-gitlab v0.39.0 -> v0.42.0 * update github.com/yuin/goldmark v1.2.1 -> v1.3.1
144 lines
3.7 KiB
Go
Vendored
144 lines
3.7 KiB
Go
Vendored
// Copyright 2013 The Gorilla Authors. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package handlers
|
|
|
|
import (
|
|
"compress/flate"
|
|
"compress/gzip"
|
|
"io"
|
|
"net/http"
|
|
"strings"
|
|
|
|
"github.com/felixge/httpsnoop"
|
|
)
|
|
|
|
const acceptEncoding string = "Accept-Encoding"
|
|
|
|
type compressResponseWriter struct {
|
|
compressor io.Writer
|
|
w http.ResponseWriter
|
|
}
|
|
|
|
func (cw *compressResponseWriter) WriteHeader(c int) {
|
|
cw.w.Header().Del("Content-Length")
|
|
cw.w.WriteHeader(c)
|
|
}
|
|
|
|
func (cw *compressResponseWriter) Write(b []byte) (int, error) {
|
|
h := cw.w.Header()
|
|
if h.Get("Content-Type") == "" {
|
|
h.Set("Content-Type", http.DetectContentType(b))
|
|
}
|
|
h.Del("Content-Length")
|
|
|
|
return cw.compressor.Write(b)
|
|
}
|
|
|
|
func (cw *compressResponseWriter) ReadFrom(r io.Reader) (int64, error) {
|
|
return io.Copy(cw.compressor, r)
|
|
}
|
|
|
|
type flusher interface {
|
|
Flush() error
|
|
}
|
|
|
|
func (w *compressResponseWriter) Flush() {
|
|
// Flush compressed data if compressor supports it.
|
|
if f, ok := w.compressor.(flusher); ok {
|
|
f.Flush()
|
|
}
|
|
// Flush HTTP response.
|
|
if f, ok := w.w.(http.Flusher); ok {
|
|
f.Flush()
|
|
}
|
|
}
|
|
|
|
// CompressHandler gzip compresses HTTP responses for clients that support it
|
|
// via the 'Accept-Encoding' header.
|
|
//
|
|
// Compressing TLS traffic may leak the page contents to an attacker if the
|
|
// page contains user input: http://security.stackexchange.com/a/102015/12208
|
|
func CompressHandler(h http.Handler) http.Handler {
|
|
return CompressHandlerLevel(h, gzip.DefaultCompression)
|
|
}
|
|
|
|
// CompressHandlerLevel gzip compresses HTTP responses with specified compression level
|
|
// for clients that support it via the 'Accept-Encoding' header.
|
|
//
|
|
// The compression level should be gzip.DefaultCompression, gzip.NoCompression,
|
|
// or any integer value between gzip.BestSpeed and gzip.BestCompression inclusive.
|
|
// gzip.DefaultCompression is used in case of invalid compression level.
|
|
func CompressHandlerLevel(h http.Handler, level int) http.Handler {
|
|
if level < gzip.DefaultCompression || level > gzip.BestCompression {
|
|
level = gzip.DefaultCompression
|
|
}
|
|
|
|
const (
|
|
gzipEncoding = "gzip"
|
|
flateEncoding = "deflate"
|
|
)
|
|
|
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
// detect what encoding to use
|
|
var encoding string
|
|
for _, curEnc := range strings.Split(r.Header.Get(acceptEncoding), ",") {
|
|
curEnc = strings.TrimSpace(curEnc)
|
|
if curEnc == gzipEncoding || curEnc == flateEncoding {
|
|
encoding = curEnc
|
|
break
|
|
}
|
|
}
|
|
|
|
// always add Accept-Encoding to Vary to prevent intermediate caches corruption
|
|
w.Header().Add("Vary", acceptEncoding)
|
|
|
|
// if we weren't able to identify an encoding we're familiar with, pass on the
|
|
// request to the handler and return
|
|
if encoding == "" {
|
|
h.ServeHTTP(w, r)
|
|
return
|
|
}
|
|
|
|
if r.Header.Get("Upgrade") != "" {
|
|
h.ServeHTTP(w, r)
|
|
return
|
|
}
|
|
|
|
// wrap the ResponseWriter with the writer for the chosen encoding
|
|
var encWriter io.WriteCloser
|
|
if encoding == gzipEncoding {
|
|
encWriter, _ = gzip.NewWriterLevel(w, level)
|
|
} else if encoding == flateEncoding {
|
|
encWriter, _ = flate.NewWriter(w, level)
|
|
}
|
|
defer encWriter.Close()
|
|
|
|
w.Header().Set("Content-Encoding", encoding)
|
|
r.Header.Del(acceptEncoding)
|
|
|
|
cw := &compressResponseWriter{
|
|
w: w,
|
|
compressor: encWriter,
|
|
}
|
|
|
|
w = httpsnoop.Wrap(w, httpsnoop.Hooks{
|
|
Write: func(httpsnoop.WriteFunc) httpsnoop.WriteFunc {
|
|
return cw.Write
|
|
},
|
|
WriteHeader: func(httpsnoop.WriteHeaderFunc) httpsnoop.WriteHeaderFunc {
|
|
return cw.WriteHeader
|
|
},
|
|
Flush: func(httpsnoop.FlushFunc) httpsnoop.FlushFunc {
|
|
return cw.Flush
|
|
},
|
|
ReadFrom: func(rff httpsnoop.ReadFromFunc) httpsnoop.ReadFromFunc {
|
|
return cw.ReadFrom
|
|
},
|
|
})
|
|
|
|
h.ServeHTTP(w, r)
|
|
})
|
|
}
|