Fix: fix upgrade header detect (#2134)

This commit is contained in:
Kr328 2022-05-15 09:12:53 +08:00 committed by GitHub
parent da7ffc0da9
commit b384449717
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 9 additions and 1 deletions

View File

@ -12,7 +12,15 @@ import (
)
func isUpgradeRequest(req *http.Request) bool {
return strings.EqualFold(req.Header.Get("Connection"), "Upgrade")
for _, header := range req.Header["Connection"] {
for _, elm := range strings.Split(header, ",") {
if strings.EqualFold(strings.TrimSpace(elm), "Upgrade") {
return true
}
}
}
return false
}
func handleUpgrade(conn net.Conn, request *http.Request, in chan<- C.ConnContext) {