mirror of
https://github.com/seaweedfs/seaweedfs.git
synced 2024-11-27 20:59:42 +08:00
s3: pass through s3 presigned headers
fix https://github.com/chrislusf/seaweedfs/discussions/2502
This commit is contained in:
parent
bf4d7affc0
commit
7210558c7b
@ -52,3 +52,12 @@ func GetBucketAndObject(r *http.Request) (bucket, object string) {
|
||||
|
||||
return
|
||||
}
|
||||
|
||||
var PassThroughHeaders = map[string]string{
|
||||
"response-cache-control": "Cache-Control",
|
||||
"response-content-disposition": "Content-Disposition",
|
||||
"response-content-encoding": "Content-Encoding",
|
||||
"response-content-language": "Content-Language",
|
||||
"response-content-type": "Content-Type",
|
||||
"response-expires": "Expires",
|
||||
}
|
||||
|
@ -306,15 +306,6 @@ func (s3a *S3ApiServer) doDeleteEmptyDirectories(client filer_pb.SeaweedFilerCli
|
||||
return
|
||||
}
|
||||
|
||||
var passThroughHeaders = []string{
|
||||
"response-cache-control",
|
||||
"response-content-disposition",
|
||||
"response-content-encoding",
|
||||
"response-content-language",
|
||||
"response-content-type",
|
||||
"response-expires",
|
||||
}
|
||||
|
||||
func (s3a *S3ApiServer) proxyToFiler(w http.ResponseWriter, r *http.Request, destUrl string, responseFn func(proxyResponse *http.Response, w http.ResponseWriter) (statusCode int)) {
|
||||
|
||||
glog.V(3).Infof("s3 proxying %s to %s", r.Method, destUrl)
|
||||
@ -328,24 +319,13 @@ func (s3a *S3ApiServer) proxyToFiler(w http.ResponseWriter, r *http.Request, des
|
||||
}
|
||||
|
||||
proxyReq.Header.Set("X-Forwarded-For", r.RemoteAddr)
|
||||
|
||||
for k, v := range r.URL.Query() {
|
||||
if _, ok := xhttp.PassThroughHeaders[strings.ToLower(k)]; ok {
|
||||
proxyReq.Header[k] = v
|
||||
}
|
||||
}
|
||||
for header, values := range r.Header {
|
||||
// handle s3 related headers
|
||||
passed := false
|
||||
for _, h := range passThroughHeaders {
|
||||
if strings.ToLower(header) == h && len(values) > 0 {
|
||||
proxyReq.Header.Add(header[len("response-"):], values[0])
|
||||
passed = true
|
||||
break
|
||||
}
|
||||
}
|
||||
if passed {
|
||||
continue
|
||||
}
|
||||
// handle other headers
|
||||
for _, value := range values {
|
||||
proxyReq.Header.Add(header, value)
|
||||
}
|
||||
proxyReq.Header[header] = values
|
||||
}
|
||||
|
||||
resp, postErr := client.Do(proxyReq)
|
||||
|
@ -5,6 +5,7 @@ import (
|
||||
"encoding/json"
|
||||
"errors"
|
||||
"fmt"
|
||||
xhttp "github.com/chrislusf/seaweedfs/weed/s3api/http"
|
||||
"io"
|
||||
"io/fs"
|
||||
"mime/multipart"
|
||||
@ -250,13 +251,16 @@ func handleStaticResources2(r *mux.Router) {
|
||||
r.PathPrefix("/seaweedfsstatic/").Handler(http.StripPrefix("/seaweedfsstatic", http.FileServer(http.FS(StaticFS))))
|
||||
}
|
||||
|
||||
func adjustHeaderContentDisposition(w http.ResponseWriter, r *http.Request, filename string) {
|
||||
responseContentDisposition := r.FormValue("response-content-disposition")
|
||||
if responseContentDisposition != "" {
|
||||
w.Header().Set("Content-Disposition", responseContentDisposition)
|
||||
return
|
||||
func adjustPassthroughHeaders(w http.ResponseWriter, r *http.Request, filename string) {
|
||||
for header, values := range r.Header {
|
||||
if normalizedHeader, ok := xhttp.PassThroughHeaders[strings.ToLower(header)]; ok {
|
||||
w.Header()[normalizedHeader] = values
|
||||
}
|
||||
}
|
||||
if w.Header().Get("Content-Disposition") != "" {
|
||||
adjustHeaderContentDisposition(w, r, filename)
|
||||
}
|
||||
func adjustHeaderContentDisposition(w http.ResponseWriter, r *http.Request, filename string) {
|
||||
if contentDisposition := w.Header().Get("Content-Disposition"); contentDisposition != "" {
|
||||
return
|
||||
}
|
||||
if filename != "" {
|
||||
|
@ -130,7 +130,7 @@ func (fs *FilerServer) GetOrHeadHandler(w http.ResponseWriter, r *http.Request)
|
||||
setEtag(w, etag)
|
||||
|
||||
filename := entry.Name()
|
||||
adjustHeaderContentDisposition(w, r, filename)
|
||||
adjustPassthroughHeaders(w, r, filename)
|
||||
|
||||
totalSize := int64(entry.Size())
|
||||
|
||||
|
@ -301,7 +301,7 @@ func writeResponseContent(filename, mimeType string, rs io.ReadSeeker, w http.Re
|
||||
}
|
||||
w.Header().Set("Accept-Ranges", "bytes")
|
||||
|
||||
adjustHeaderContentDisposition(w, r, filename)
|
||||
adjustPassthroughHeaders(w, r, filename)
|
||||
|
||||
if r.Method == "HEAD" {
|
||||
w.Header().Set("Content-Length", strconv.FormatInt(totalSize, 10))
|
||||
|
Loading…
Reference in New Issue
Block a user