MyFavMirrors/minio
1
0
Fork 0
mirror of https://github.com/minio/minio.git synced 2024-12-24 06:05:55 -05:00
Code Issues Packages Projects Releases Wiki Activity

Close stream on panic (#13605)

Browse Source 
Always close streamHTTPResponse on panic on main thread to avoid 
write/flush after response handler has returned.
This commit is contained in:
Klaus Post 2021-11-08 08:41:27 -08:00 committed by GitHub
parent 9afdbe3648
commit 4f3317effe
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 15 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
cmd/metacache-walk.go
View File

@ -19,9 +19,11 @@ package cmd
import ( import (
"context" "context"
"fmt"
"io" "io"
"net/http" "net/http"
"net/url" "net/url"
"runtime/debug"
"sort" "sort"
"strconv" "strconv"
"strings" "strings"
@ -357,6 +359,12 @@ func (s *storageRESTServer) WalkDirHandler(w http.ResponseWriter, r *http.Reques
prefix := r.Form.Get(storageRESTPrefixFilter) prefix := r.Form.Get(storageRESTPrefixFilter)
forward := r.Form.Get(storageRESTForwardFilter) forward := r.Form.Get(storageRESTForwardFilter)
writer := streamHTTPResponse(w) writer := streamHTTPResponse(w)
defer func() {
if r := recover(); r != nil {
debug.PrintStack()
writer.CloseWithError(fmt.Errorf("panic: %v", r))
}
}()
writer.CloseWithError(s.storage.WalkDir(r.Context(), WalkDirOptions{ writer.CloseWithError(s.storage.WalkDir(r.Context(), WalkDirOptions{
Bucket: volume, Bucket: volume,
BaseDir: dirPath, BaseDir: dirPath,

7
cmd/storage-rest-server.go
View File

@ -30,6 +30,7 @@ import (
"net/http" "net/http"
"os/user" "os/user"
"path" "path"
"runtime/debug"
"strconv" "strconv"
"strings" "strings"
"sync" "sync"
@ -177,6 +178,12 @@ func (s *storageRESTServer) NSScannerHandler(w http.ResponseWriter, r *http.Requ
ctx, cancel := context.WithCancel(r.Context()) ctx, cancel := context.WithCancel(r.Context())
defer cancel() defer cancel()
resp := streamHTTPResponse(w) resp := streamHTTPResponse(w)
defer func() {
if r := recover(); r != nil {
debug.PrintStack()
resp.CloseWithError(fmt.Errorf("panic: %v", r))
}
}()
respW := msgp.NewWriter(resp) respW := msgp.NewWriter(resp)
// Collect updates, stream them before the full cache is sent. // Collect updates, stream them before the full cache is sent.