Merge pull request #11 from aeby/fix/auth-header

fix: Don't send empty X-Amz-Security-Token header field
sony · May 6, 2021 · 5be256b · 5be256b
2 parents 9b2af41 + 9670282
commit 5be256b
Showing 1 changed file with 15 additions and 9 deletions.
diff --git a/pure_websocket_subscriber.go b/pure_websocket_subscriber.go
@@ -181,15 +181,21 @@ func signRequest(signer *v4.Signer, url, region string, data []byte) (map[string
 		return nil, err
 	}
 
-	return map[string]string{
-		"accept":               req.Header.Get("accept"),
-		"content-encoding":     req.Header.Get("content-encoding"),
-		"content-type":         req.Header.Get("content-type"),
-		"host":                 req.Host,
-		"x-amz-date":           req.Header.Get("x-amz-date"),
-		"X-Amz-Security-Token": req.Header.Get("X-Amz-Security-Token"),
-		"Authorization":        req.Header.Get("Authorization"),
-	}, nil
+	headers := map[string]string{
+		"accept":           req.Header.Get("accept"),
+		"content-encoding": req.Header.Get("content-encoding"),
+		"content-type":     req.Header.Get("content-type"),
+		"host":             req.Host,
+		"x-amz-date":       req.Header.Get("x-amz-date"),
+		"Authorization":    req.Header.Get("Authorization"),
+	}
+
+	token := req.Header.Get("X-Amz-Security-Token")
+	if token != "" {
+		headers["X-Amz-Security-Token"] = token
+	}
+
+	return headers, nil
 }
 
 // Stop ends the subscription.