mirror of
https://github.com/pgsty/minio.git
synced 2026-10-16 22:55:58 +03:00
Feng Ruohang
74 lines
2.4 KiB
Go
74 lines
2.4 KiB
Go
// Copyright (c) 2026 PGSTY
|
|
//
|
|
// This file is part of SILO Object Storage.
|
|
// SPDX-License-Identifier: AGPL-3.0-or-later
|
|
//
|
|
// This program is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU Affero General Public License as published by
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version.
|
|
//
|
|
// This program is distributed in the hope that it will be useful
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU Affero General Public License for more details.
|
|
//
|
|
// You should have received a copy of the GNU Affero General Public License
|
|
// along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
package hash
|
|
|
|
import (
|
|
"bytes"
|
|
"errors"
|
|
"io"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
)
|
|
|
|
func TestChecksumMismatchAlgorithm(t *testing.T) {
|
|
content := []byte("actual content")
|
|
for _, typ := range []ChecksumType{ChecksumCRC32, ChecksumCRC32C, ChecksumSHA1, ChecksumSHA256, ChecksumCRC64NVME} {
|
|
t.Run(typ.String(), func(t *testing.T) {
|
|
wrong := NewChecksumFromData(typ, []byte("wrong content"))
|
|
check := func(t *testing.T, err error) {
|
|
t.Helper()
|
|
var mismatch ChecksumMismatch
|
|
if !errors.As(err, &mismatch) || mismatch.Algorithm != typ.String() {
|
|
t.Fatalf("error=%+v, want %s checksum mismatch", err, typ)
|
|
}
|
|
}
|
|
check(t, wrong.Matches(content, 0))
|
|
parts := NewChecksumFromData(typ, content)
|
|
parts.WantParts = 2
|
|
check(t, parts.Matches(content, 1))
|
|
for _, mode := range []string{"header", "trailer", "malformed-trailer"} {
|
|
t.Run(mode, func(t *testing.T) {
|
|
reader, err := NewReader(t.Context(), bytes.NewReader(content), int64(len(content)), "", "", int64(len(content)))
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
req := httptest.NewRequest(http.MethodPut, "http://localhost/bucket/object", nil)
|
|
if mode == "header" {
|
|
req.Header.Set(typ.Key(), wrong.Encoded)
|
|
} else {
|
|
req.Header.Set("x-amz-trailer", typ.Key())
|
|
req.Trailer = make(http.Header)
|
|
value := wrong.Encoded
|
|
if mode == "malformed-trailer" {
|
|
value = "!invalid-base64!"
|
|
}
|
|
req.Trailer.Set(typ.Key(), value)
|
|
}
|
|
if err = reader.AddChecksum(req, false); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
_, err = io.ReadAll(reader)
|
|
check(t, err)
|
|
})
|
|
}
|
|
})
|
|
}
|
|
}
|