aboutsummaryrefslogtreecommitdiff
path: root/weed/s3api/s3_sse_c_range_test.go
diff options
context:
space:
mode:
Diffstat (limited to 'weed/s3api/s3_sse_c_range_test.go')
-rw-r--r--weed/s3api/s3_sse_c_range_test.go66
1 files changed, 66 insertions, 0 deletions
diff --git a/weed/s3api/s3_sse_c_range_test.go b/weed/s3api/s3_sse_c_range_test.go
new file mode 100644
index 000000000..318771d8c
--- /dev/null
+++ b/weed/s3api/s3_sse_c_range_test.go
@@ -0,0 +1,66 @@
+package s3api
+
+import (
+ "bytes"
+ "crypto/md5"
+ "encoding/base64"
+ "io"
+ "net/http"
+ "net/http/httptest"
+ "testing"
+
+ "github.com/gorilla/mux"
+ "github.com/seaweedfs/seaweedfs/weed/s3api/s3_constants"
+)
+
+// ResponseRecorder that also implements http.Flusher
+type recorderFlusher struct{ *httptest.ResponseRecorder }
+
+func (r recorderFlusher) Flush() {}
+
+// TestSSECRangeRequestsSupported verifies that HTTP Range requests are now supported
+// for SSE-C encrypted objects since the IV is stored in metadata and CTR mode allows seeking
+func TestSSECRangeRequestsSupported(t *testing.T) {
+ // Create a request with Range header and valid SSE-C headers
+ req := httptest.NewRequest(http.MethodGet, "/b/o", nil)
+ req.Header.Set("Range", "bytes=10-20")
+ req.Header.Set(s3_constants.AmzServerSideEncryptionCustomerAlgorithm, "AES256")
+
+ key := make([]byte, 32)
+ for i := range key {
+ key[i] = byte(i)
+ }
+ s := md5.Sum(key)
+ keyMD5 := base64.StdEncoding.EncodeToString(s[:])
+
+ req.Header.Set(s3_constants.AmzServerSideEncryptionCustomerKey, base64.StdEncoding.EncodeToString(key))
+ req.Header.Set(s3_constants.AmzServerSideEncryptionCustomerKeyMD5, keyMD5)
+
+ // Attach mux vars to avoid panic in error writer
+ req = mux.SetURLVars(req, map[string]string{"bucket": "b", "object": "o"})
+
+ // Create a mock HTTP response that simulates SSE-C encrypted object metadata
+ proxyResponse := &http.Response{
+ StatusCode: 200,
+ Header: make(http.Header),
+ Body: io.NopCloser(bytes.NewReader([]byte("mock encrypted data"))),
+ }
+ proxyResponse.Header.Set(s3_constants.AmzServerSideEncryptionCustomerAlgorithm, "AES256")
+ proxyResponse.Header.Set(s3_constants.AmzServerSideEncryptionCustomerKeyMD5, keyMD5)
+
+ // Call the function under test - should no longer reject range requests
+ s3a := &S3ApiServer{
+ option: &S3ApiServerOption{
+ BucketsPath: "/buckets",
+ },
+ }
+ rec := httptest.NewRecorder()
+ w := recorderFlusher{rec}
+ statusCode, _ := s3a.handleSSECResponse(req, proxyResponse, w)
+
+ // Range requests should now be allowed to proceed (will be handled by filer layer)
+ // The exact status code depends on the object existence and filer response
+ if statusCode == http.StatusRequestedRangeNotSatisfiable {
+ t.Fatalf("Range requests should no longer be rejected for SSE-C objects, got status %d", statusCode)
+ }
+}