2016-02-19 19:04:29 -05:00
|
|
|
/*
|
2019-05-08 21:35:40 -04:00
|
|
|
* MinIO Cloud Storage, (C) 2015-2019 MinIO, Inc.
|
2016-02-19 19:04:29 -05:00
|
|
|
*
|
|
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
|
|
* you may not use this file except in compliance with the License.
|
|
|
|
* You may obtain a copy of the License at
|
|
|
|
*
|
|
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
*
|
|
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
|
|
* See the License for the specific language governing permissions and
|
|
|
|
* limitations under the License.
|
|
|
|
*/
|
|
|
|
|
2016-08-18 19:23:42 -04:00
|
|
|
package cmd
|
2016-02-19 19:04:29 -05:00
|
|
|
|
|
|
|
import (
|
2018-11-14 20:36:41 -05:00
|
|
|
"bytes"
|
2018-04-05 18:04:40 -04:00
|
|
|
"context"
|
2016-04-29 17:24:10 -04:00
|
|
|
"encoding/hex"
|
|
|
|
"fmt"
|
2018-09-20 22:22:09 -04:00
|
|
|
"io"
|
2018-05-15 21:20:22 -04:00
|
|
|
"math/rand"
|
2018-12-19 08:13:47 -05:00
|
|
|
"net"
|
2018-09-20 22:22:09 -04:00
|
|
|
"net/http"
|
2016-05-08 13:15:34 -04:00
|
|
|
"path"
|
2017-02-04 02:27:50 -05:00
|
|
|
"runtime"
|
2018-09-27 23:36:17 -04:00
|
|
|
"strconv"
|
2016-04-13 14:32:47 -04:00
|
|
|
"strings"
|
2018-09-20 22:22:09 -04:00
|
|
|
"sync"
|
2018-05-15 21:20:22 -04:00
|
|
|
"time"
|
2016-02-19 19:04:29 -05:00
|
|
|
"unicode/utf8"
|
2016-04-29 17:24:10 -04:00
|
|
|
|
2019-09-26 02:08:24 -04:00
|
|
|
"github.com/klauspost/compress/s2"
|
|
|
|
"github.com/klauspost/readahead"
|
2019-05-29 19:35:12 -04:00
|
|
|
"github.com/minio/minio-go/v6/pkg/s3utils"
|
2018-09-20 22:22:09 -04:00
|
|
|
"github.com/minio/minio/cmd/crypto"
|
2019-07-03 01:34:32 -04:00
|
|
|
xhttp "github.com/minio/minio/cmd/http"
|
2018-04-05 18:04:40 -04:00
|
|
|
"github.com/minio/minio/cmd/logger"
|
2018-05-11 15:02:30 -04:00
|
|
|
"github.com/minio/minio/pkg/dns"
|
2018-11-14 20:36:41 -05:00
|
|
|
"github.com/minio/minio/pkg/hash"
|
2018-09-20 22:22:09 -04:00
|
|
|
"github.com/minio/minio/pkg/ioutil"
|
2018-09-27 23:36:17 -04:00
|
|
|
"github.com/minio/minio/pkg/wildcard"
|
2016-05-20 23:48:47 -04:00
|
|
|
"github.com/skyrings/skyring-common/tools/uuid"
|
2016-04-29 17:24:10 -04:00
|
|
|
)
|
|
|
|
|
|
|
|
const (
|
2019-04-09 14:39:42 -04:00
|
|
|
// MinIO meta bucket.
|
2016-07-12 18:21:29 -04:00
|
|
|
minioMetaBucket = ".minio.sys"
|
2016-05-03 19:10:24 -04:00
|
|
|
// Multipart meta prefix.
|
|
|
|
mpartMetaPrefix = "multipart"
|
2019-04-09 14:39:42 -04:00
|
|
|
// MinIO Multipart meta prefix.
|
2019-08-06 15:08:58 -04:00
|
|
|
minioMetaMultipartBucket = minioMetaBucket + SlashSeparator + mpartMetaPrefix
|
2019-04-09 14:39:42 -04:00
|
|
|
// MinIO Tmp meta prefix.
|
2016-11-20 17:25:43 -05:00
|
|
|
minioMetaTmpBucket = minioMetaBucket + "/tmp"
|
2017-03-03 13:23:41 -05:00
|
|
|
// DNS separator (period), used for bucket name validation.
|
|
|
|
dnsDelimiter = "."
|
2019-09-26 02:08:24 -04:00
|
|
|
// On compressed files bigger than this;
|
|
|
|
compReadAheadSize = 100 << 20
|
|
|
|
// Read this many buffers ahead.
|
|
|
|
compReadAheadBuffers = 5
|
|
|
|
// Size of each buffer.
|
|
|
|
compReadAheadBufSize = 1 << 20
|
2016-02-19 19:04:29 -05:00
|
|
|
)
|
|
|
|
|
2019-04-09 14:39:42 -04:00
|
|
|
// isMinioBucket returns true if given bucket is a MinIO internal
|
2017-01-17 13:02:58 -05:00
|
|
|
// bucket and false otherwise.
|
|
|
|
func isMinioMetaBucketName(bucket string) bool {
|
|
|
|
return bucket == minioMetaBucket ||
|
|
|
|
bucket == minioMetaMultipartBucket ||
|
|
|
|
bucket == minioMetaTmpBucket
|
|
|
|
}
|
|
|
|
|
2017-03-03 13:23:41 -05:00
|
|
|
// IsValidBucketName verifies that a bucket name is in accordance with
|
|
|
|
// Amazon's requirements (i.e. DNS naming conventions). It must be 3-63
|
|
|
|
// characters long, and it must be a sequence of one or more labels
|
|
|
|
// separated by periods. Each label can contain lowercase ascii
|
|
|
|
// letters, decimal digits and hyphens, but must not begin or end with
|
|
|
|
// a hyphen. See:
|
|
|
|
// http://docs.aws.amazon.com/AmazonS3/latest/dev/BucketRestrictions.html
|
2016-02-19 19:04:29 -05:00
|
|
|
func IsValidBucketName(bucket string) bool {
|
2017-01-17 13:02:58 -05:00
|
|
|
// Special case when bucket is equal to one of the meta buckets.
|
|
|
|
if isMinioMetaBucketName(bucket) {
|
2016-07-24 01:51:12 -04:00
|
|
|
return true
|
|
|
|
}
|
2016-02-19 19:04:29 -05:00
|
|
|
if len(bucket) < 3 || len(bucket) > 63 {
|
|
|
|
return false
|
|
|
|
}
|
2017-03-03 13:23:41 -05:00
|
|
|
|
|
|
|
// Split on dot and check each piece conforms to rules.
|
|
|
|
allNumbers := true
|
|
|
|
pieces := strings.Split(bucket, dnsDelimiter)
|
|
|
|
for _, piece := range pieces {
|
|
|
|
if len(piece) == 0 || piece[0] == '-' ||
|
|
|
|
piece[len(piece)-1] == '-' {
|
|
|
|
// Current piece has 0-length or starts or
|
|
|
|
// ends with a hyphen.
|
|
|
|
return false
|
|
|
|
}
|
|
|
|
// Now only need to check if each piece is a valid
|
|
|
|
// 'label' in AWS terminology and if the bucket looks
|
|
|
|
// like an IP address.
|
|
|
|
isNotNumber := false
|
|
|
|
for i := 0; i < len(piece); i++ {
|
|
|
|
switch {
|
|
|
|
case (piece[i] >= 'a' && piece[i] <= 'z' ||
|
|
|
|
piece[i] == '-'):
|
|
|
|
// Found a non-digit character, so
|
|
|
|
// this piece is not a number.
|
|
|
|
isNotNumber = true
|
|
|
|
case piece[i] >= '0' && piece[i] <= '9':
|
|
|
|
// Nothing to do.
|
|
|
|
default:
|
|
|
|
// Found invalid character.
|
|
|
|
return false
|
|
|
|
}
|
|
|
|
}
|
|
|
|
allNumbers = allNumbers && !isNotNumber
|
2016-02-19 19:04:29 -05:00
|
|
|
}
|
2017-03-03 13:23:41 -05:00
|
|
|
// Does the bucket name look like an IP address?
|
|
|
|
return !(len(pieces) == 4 && allNumbers)
|
2016-02-19 19:04:29 -05:00
|
|
|
}
|
|
|
|
|
2016-03-07 22:30:30 -05:00
|
|
|
// IsValidObjectName verifies an object name in accordance with Amazon's
|
|
|
|
// requirements. It cannot exceed 1024 characters and must be a valid UTF8
|
|
|
|
// string.
|
2016-04-13 14:32:47 -04:00
|
|
|
//
|
|
|
|
// See:
|
|
|
|
// http://docs.aws.amazon.com/AmazonS3/latest/dev/UsingMetadata.html
|
|
|
|
//
|
|
|
|
// You should avoid the following characters in a key name because of
|
|
|
|
// significant special handling for consistency across all
|
|
|
|
// applications.
|
|
|
|
//
|
|
|
|
// Rejects strings with following characters.
|
|
|
|
//
|
|
|
|
// - Backslash ("\")
|
2016-05-13 14:43:06 -04:00
|
|
|
//
|
2019-08-06 15:08:58 -04:00
|
|
|
// additionally minio does not support object names with trailing SlashSeparator.
|
2016-02-19 19:04:29 -05:00
|
|
|
func IsValidObjectName(object string) bool {
|
2016-05-13 14:43:06 -04:00
|
|
|
if len(object) == 0 {
|
2016-02-19 19:04:29 -05:00
|
|
|
return false
|
|
|
|
}
|
2019-08-06 15:08:58 -04:00
|
|
|
if hasSuffix(object, SlashSeparator) {
|
2016-05-13 14:43:06 -04:00
|
|
|
return false
|
|
|
|
}
|
|
|
|
return IsValidObjectPrefix(object)
|
2016-02-19 19:04:29 -05:00
|
|
|
}
|
2016-03-22 19:03:08 -04:00
|
|
|
|
|
|
|
// IsValidObjectPrefix verifies whether the prefix is a valid object name.
|
|
|
|
// Its valid to have a empty prefix.
|
|
|
|
func IsValidObjectPrefix(object string) bool {
|
2017-04-24 21:13:46 -04:00
|
|
|
if hasBadPathComponent(object) {
|
|
|
|
return false
|
|
|
|
}
|
2016-05-13 14:43:06 -04:00
|
|
|
if !utf8.ValidString(object) {
|
|
|
|
return false
|
|
|
|
}
|
|
|
|
// Reject unsupported characters in object name.
|
2016-09-30 19:56:36 -04:00
|
|
|
if strings.ContainsAny(object, "\\") {
|
2016-05-13 14:43:06 -04:00
|
|
|
return false
|
|
|
|
}
|
|
|
|
return true
|
2016-03-22 19:03:08 -04:00
|
|
|
}
|
2016-04-16 14:43:03 -04:00
|
|
|
|
2019-07-12 00:38:12 -04:00
|
|
|
// checkObjectNameForLengthAndSlash -check for the validity of object name length and prefis as slash
|
|
|
|
func checkObjectNameForLengthAndSlash(bucket, object string) error {
|
|
|
|
// Check for the length of object name
|
|
|
|
if len(object) > 1024 {
|
|
|
|
return ObjectNameTooLong{
|
|
|
|
Bucket: bucket,
|
|
|
|
Object: object,
|
|
|
|
}
|
|
|
|
}
|
|
|
|
// Check for slash as prefix in object name
|
2019-08-06 15:08:58 -04:00
|
|
|
if hasPrefix(object, SlashSeparator) {
|
2019-07-12 00:38:12 -04:00
|
|
|
return ObjectNamePrefixAsSlash{
|
|
|
|
Bucket: bucket,
|
|
|
|
Object: object,
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2019-08-06 15:08:58 -04:00
|
|
|
// SlashSeparator - slash separator.
|
|
|
|
const SlashSeparator = "/"
|
2016-04-21 21:05:26 -04:00
|
|
|
|
|
|
|
// retainSlash - retains slash from a path.
|
|
|
|
func retainSlash(s string) string {
|
2019-08-06 15:08:58 -04:00
|
|
|
return strings.TrimSuffix(s, SlashSeparator) + SlashSeparator
|
2016-04-21 21:05:26 -04:00
|
|
|
}
|
|
|
|
|
2019-08-06 15:08:58 -04:00
|
|
|
// pathJoin - like path.Join() but retains trailing SlashSeparator of the last element
|
2016-05-09 03:46:54 -04:00
|
|
|
func pathJoin(elem ...string) string {
|
2016-05-08 13:15:34 -04:00
|
|
|
trailingSlash := ""
|
2016-05-09 03:46:54 -04:00
|
|
|
if len(elem) > 0 {
|
2019-08-06 15:08:58 -04:00
|
|
|
if hasSuffix(elem[len(elem)-1], SlashSeparator) {
|
|
|
|
trailingSlash = SlashSeparator
|
2016-05-09 03:46:54 -04:00
|
|
|
}
|
2016-05-08 13:15:34 -04:00
|
|
|
}
|
2016-05-09 03:46:54 -04:00
|
|
|
return path.Join(elem...) + trailingSlash
|
2016-04-16 14:43:03 -04:00
|
|
|
}
|
2016-04-29 17:24:10 -04:00
|
|
|
|
2016-11-22 19:52:37 -05:00
|
|
|
// mustGetUUID - get a random UUID.
|
|
|
|
func mustGetUUID() string {
|
|
|
|
uuid, err := uuid.New()
|
|
|
|
if err != nil {
|
2018-04-19 20:24:43 -04:00
|
|
|
logger.CriticalIf(context.Background(), err)
|
2016-05-20 23:48:47 -04:00
|
|
|
}
|
2016-11-22 19:52:37 -05:00
|
|
|
|
|
|
|
return uuid.String()
|
2016-05-20 23:48:47 -04:00
|
|
|
}
|
|
|
|
|
2016-04-29 17:24:10 -04:00
|
|
|
// Create an s3 compatible MD5sum for complete multipart transaction.
|
2019-05-08 21:35:40 -04:00
|
|
|
func getCompleteMultipartMD5(parts []CompletePart) string {
|
2016-04-29 17:24:10 -04:00
|
|
|
var finalMD5Bytes []byte
|
2016-05-08 05:38:35 -04:00
|
|
|
for _, part := range parts {
|
2019-04-01 15:19:52 -04:00
|
|
|
md5Bytes, err := hex.DecodeString(canonicalizeETag(part.ETag))
|
2016-04-29 17:24:10 -04:00
|
|
|
if err != nil {
|
2019-05-08 21:35:40 -04:00
|
|
|
finalMD5Bytes = append(finalMD5Bytes, []byte(part.ETag)...)
|
|
|
|
} else {
|
|
|
|
finalMD5Bytes = append(finalMD5Bytes, md5Bytes...)
|
2016-04-29 17:24:10 -04:00
|
|
|
}
|
|
|
|
}
|
2016-11-21 16:51:05 -05:00
|
|
|
s3MD5 := fmt.Sprintf("%s-%d", getMD5Hash(finalMD5Bytes), len(parts))
|
2019-05-08 21:35:40 -04:00
|
|
|
return s3MD5
|
2016-04-29 17:24:10 -04:00
|
|
|
}
|
|
|
|
|
2018-01-04 01:14:45 -05:00
|
|
|
// Clean unwanted fields from metadata
|
|
|
|
func cleanMetadata(metadata map[string]string) map[string]string {
|
|
|
|
// Remove STANDARD StorageClass
|
|
|
|
metadata = removeStandardStorageClass(metadata)
|
2019-02-28 14:01:25 -05:00
|
|
|
// Clean meta etag keys 'md5Sum', 'etag', "expires".
|
|
|
|
return cleanMetadataKeys(metadata, "md5Sum", "etag", "expires")
|
2017-05-14 15:05:51 -04:00
|
|
|
}
|
|
|
|
|
2018-01-04 01:14:45 -05:00
|
|
|
// Filter X-Amz-Storage-Class field only if it is set to STANDARD.
|
|
|
|
// This is done since AWS S3 doesn't return STANDARD Storage class as response header.
|
|
|
|
func removeStandardStorageClass(metadata map[string]string) map[string]string {
|
|
|
|
if metadata[amzStorageClass] == standardStorageClass {
|
|
|
|
delete(metadata, amzStorageClass)
|
|
|
|
}
|
|
|
|
return metadata
|
|
|
|
}
|
|
|
|
|
|
|
|
// cleanMetadataKeys takes keyNames to be filtered
|
|
|
|
// and returns a new map with all the entries with keyNames removed.
|
|
|
|
func cleanMetadataKeys(metadata map[string]string, keyNames ...string) map[string]string {
|
2017-05-14 15:05:51 -04:00
|
|
|
var newMeta = make(map[string]string)
|
|
|
|
for k, v := range metadata {
|
|
|
|
if contains(keyNames, k) {
|
|
|
|
continue
|
|
|
|
}
|
|
|
|
newMeta[k] = v
|
|
|
|
}
|
|
|
|
return newMeta
|
|
|
|
}
|
|
|
|
|
|
|
|
// Extracts etag value from the metadata.
|
|
|
|
func extractETag(metadata map[string]string) string {
|
|
|
|
// md5Sum tag is kept for backward compatibility.
|
|
|
|
etag, ok := metadata["md5Sum"]
|
|
|
|
if !ok {
|
|
|
|
etag = metadata["etag"]
|
|
|
|
}
|
|
|
|
// Success.
|
|
|
|
return etag
|
|
|
|
}
|
|
|
|
|
2017-02-04 02:27:50 -05:00
|
|
|
// Prefix matcher string matches prefix in a platform specific way.
|
|
|
|
// For example on windows since its case insensitive we are supposed
|
|
|
|
// to do case insensitive checks.
|
|
|
|
func hasPrefix(s string, prefix string) bool {
|
2017-02-18 16:41:59 -05:00
|
|
|
if runtime.GOOS == globalWindowsOSName {
|
2017-02-04 02:27:50 -05:00
|
|
|
return strings.HasPrefix(strings.ToLower(s), strings.ToLower(prefix))
|
|
|
|
}
|
|
|
|
return strings.HasPrefix(s, prefix)
|
|
|
|
}
|
|
|
|
|
|
|
|
// Suffix matcher string matches suffix in a platform specific way.
|
|
|
|
// For example on windows since its case insensitive we are supposed
|
|
|
|
// to do case insensitive checks.
|
|
|
|
func hasSuffix(s string, suffix string) bool {
|
2017-02-18 16:41:59 -05:00
|
|
|
if runtime.GOOS == globalWindowsOSName {
|
2017-02-04 02:27:50 -05:00
|
|
|
return strings.HasSuffix(strings.ToLower(s), strings.ToLower(suffix))
|
|
|
|
}
|
|
|
|
return strings.HasSuffix(s, suffix)
|
|
|
|
}
|
|
|
|
|
2017-02-18 16:41:59 -05:00
|
|
|
// Validates if two strings are equal.
|
|
|
|
func isStringEqual(s1 string, s2 string) bool {
|
|
|
|
if runtime.GOOS == globalWindowsOSName {
|
|
|
|
return strings.EqualFold(s1, s2)
|
|
|
|
}
|
|
|
|
return s1 == s2
|
|
|
|
}
|
|
|
|
|
2017-02-16 17:52:14 -05:00
|
|
|
// Ignores all reserved bucket names or invalid bucket names.
|
2019-03-05 13:42:32 -05:00
|
|
|
func isReservedOrInvalidBucket(bucketEntry string, strict bool) bool {
|
2019-08-06 15:08:58 -04:00
|
|
|
bucketEntry = strings.TrimSuffix(bucketEntry, SlashSeparator)
|
2019-03-05 13:42:32 -05:00
|
|
|
if strict {
|
|
|
|
if err := s3utils.CheckValidBucketNameStrict(bucketEntry); err != nil {
|
|
|
|
return true
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
if err := s3utils.CheckValidBucketName(bucketEntry); err != nil {
|
|
|
|
return true
|
|
|
|
}
|
2017-02-16 17:52:14 -05:00
|
|
|
}
|
2017-03-03 06:01:42 -05:00
|
|
|
return isMinioMetaBucket(bucketEntry) || isMinioReservedBucket(bucketEntry)
|
|
|
|
}
|
|
|
|
|
|
|
|
// Returns true if input bucket is a reserved minio meta bucket '.minio.sys'.
|
|
|
|
func isMinioMetaBucket(bucketName string) bool {
|
|
|
|
return bucketName == minioMetaBucket
|
|
|
|
}
|
|
|
|
|
|
|
|
// Returns true if input bucket is a reserved minio bucket 'minio'.
|
|
|
|
func isMinioReservedBucket(bucketName string) bool {
|
|
|
|
return bucketName == minioReservedBucket
|
2017-02-16 17:52:14 -05:00
|
|
|
}
|
|
|
|
|
2018-05-11 15:02:30 -04:00
|
|
|
// returns a slice of hosts by reading a slice of DNS records
|
|
|
|
func getHostsSlice(records []dns.SrvRecord) []string {
|
|
|
|
var hosts []string
|
|
|
|
for _, r := range records {
|
2019-08-02 15:40:51 -04:00
|
|
|
hosts = append(hosts, net.JoinHostPort(r.Host, fmt.Sprintf("%d", r.Port)))
|
2018-05-11 15:02:30 -04:00
|
|
|
}
|
|
|
|
return hosts
|
|
|
|
}
|
|
|
|
|
2018-12-19 08:13:47 -05:00
|
|
|
// returns a host (and corresponding port) from a slice of DNS records
|
|
|
|
func getHostFromSrv(records []dns.SrvRecord) string {
|
2018-05-15 21:20:22 -04:00
|
|
|
rand.Seed(time.Now().Unix())
|
|
|
|
srvRecord := records[rand.Intn(len(records))]
|
2018-12-19 08:13:47 -05:00
|
|
|
return net.JoinHostPort(srvRecord.Host, fmt.Sprintf("%d", srvRecord.Port))
|
2018-05-15 21:20:22 -04:00
|
|
|
}
|
|
|
|
|
2018-09-27 23:36:17 -04:00
|
|
|
// IsCompressed returns true if the object is marked as compressed.
|
|
|
|
func (o ObjectInfo) IsCompressed() bool {
|
|
|
|
_, ok := o.UserDefined[ReservedMetadataPrefix+"compression"]
|
|
|
|
return ok
|
|
|
|
}
|
|
|
|
|
2019-09-26 02:08:24 -04:00
|
|
|
// IsCompressedOK returns whether the object is compressed and can be decompressed.
|
|
|
|
func (o ObjectInfo) IsCompressedOK() (bool, error) {
|
|
|
|
scheme, ok := o.UserDefined[ReservedMetadataPrefix+"compression"]
|
|
|
|
if !ok {
|
|
|
|
return false, nil
|
|
|
|
}
|
|
|
|
if crypto.IsEncrypted(o.UserDefined) {
|
|
|
|
return true, fmt.Errorf("compression %q and encryption enabled on same object", scheme)
|
|
|
|
}
|
|
|
|
switch scheme {
|
|
|
|
case compressionAlgorithmV1, compressionAlgorithmV2:
|
|
|
|
return true, nil
|
|
|
|
}
|
|
|
|
return true, fmt.Errorf("unknown compression scheme: %s", scheme)
|
|
|
|
}
|
|
|
|
|
2018-09-27 23:36:17 -04:00
|
|
|
// GetActualSize - read the decompressed size from the meta json.
|
|
|
|
func (o ObjectInfo) GetActualSize() int64 {
|
|
|
|
metadata := o.UserDefined
|
|
|
|
sizeStr, ok := metadata[ReservedMetadataPrefix+"actual-size"]
|
|
|
|
if ok {
|
|
|
|
size, err := strconv.ParseInt(sizeStr, 10, 64)
|
|
|
|
if err == nil {
|
|
|
|
return size
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return -1
|
|
|
|
}
|
|
|
|
|
|
|
|
// Disabling compression for encrypted enabled requests.
|
|
|
|
// Using compression and encryption together enables room for side channel attacks.
|
|
|
|
// Eliminate non-compressible objects by extensions/content-types.
|
|
|
|
func isCompressible(header http.Header, object string) bool {
|
2019-09-20 17:56:12 -04:00
|
|
|
if crypto.IsRequested(header) || excludeForCompression(header, object) {
|
2018-09-27 23:36:17 -04:00
|
|
|
return false
|
|
|
|
}
|
|
|
|
return true
|
|
|
|
}
|
|
|
|
|
|
|
|
// Eliminate the non-compressible objects.
|
|
|
|
func excludeForCompression(header http.Header, object string) bool {
|
|
|
|
objStr := object
|
2019-07-03 01:34:32 -04:00
|
|
|
contentType := header.Get(xhttp.ContentType)
|
2019-09-26 02:08:24 -04:00
|
|
|
if !globalIsCompressionEnabled {
|
|
|
|
return true
|
|
|
|
}
|
|
|
|
|
|
|
|
// We strictly disable compression for standard extensions/content-types (`compressed`).
|
|
|
|
if hasStringSuffixInSlice(objStr, standardExcludeCompressExtensions) || hasPattern(standardExcludeCompressContentTypes, contentType) {
|
|
|
|
return true
|
|
|
|
}
|
|
|
|
|
|
|
|
// Filter compression includes.
|
|
|
|
if len(globalCompressExtensions) == 0 || len(globalCompressMimeTypes) == 0 {
|
|
|
|
return false
|
|
|
|
}
|
|
|
|
|
|
|
|
extensions := globalCompressExtensions
|
|
|
|
mimeTypes := globalCompressMimeTypes
|
|
|
|
if hasStringSuffixInSlice(objStr, extensions) || hasPattern(mimeTypes, contentType) {
|
2018-09-27 23:36:17 -04:00
|
|
|
return false
|
|
|
|
}
|
|
|
|
return true
|
|
|
|
}
|
|
|
|
|
|
|
|
// Utility which returns if a string is present in the list.
|
2019-09-26 02:08:24 -04:00
|
|
|
// Comparison is case insensitive.
|
2018-09-27 23:36:17 -04:00
|
|
|
func hasStringSuffixInSlice(str string, list []string) bool {
|
2019-09-26 02:08:24 -04:00
|
|
|
str = strings.ToLower(str)
|
2018-09-27 23:36:17 -04:00
|
|
|
for _, v := range list {
|
2019-09-26 02:08:24 -04:00
|
|
|
if strings.HasSuffix(str, strings.ToLower(v)) {
|
2018-09-27 23:36:17 -04:00
|
|
|
return true
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return false
|
|
|
|
}
|
|
|
|
|
|
|
|
// Returns true if any of the given wildcard patterns match the matchStr.
|
|
|
|
func hasPattern(patterns []string, matchStr string) bool {
|
|
|
|
for _, pattern := range patterns {
|
|
|
|
if ok := wildcard.MatchSimple(pattern, matchStr); ok {
|
|
|
|
return true
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return false
|
|
|
|
}
|
|
|
|
|
|
|
|
// Returns the part file name which matches the partNumber and etag.
|
|
|
|
func getPartFile(entries []string, partNumber int, etag string) string {
|
|
|
|
for _, entry := range entries {
|
|
|
|
if strings.HasPrefix(entry, fmt.Sprintf("%.5d.%s.", partNumber, etag)) {
|
|
|
|
return entry
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return ""
|
|
|
|
}
|
|
|
|
|
2019-09-26 02:08:24 -04:00
|
|
|
// Returns the compressed offset which should be skipped.
|
2018-09-27 23:36:17 -04:00
|
|
|
func getCompressedOffsets(objectInfo ObjectInfo, offset int64) (int64, int64) {
|
|
|
|
var compressedOffset int64
|
|
|
|
var skipLength int64
|
|
|
|
var cumulativeActualSize int64
|
|
|
|
if len(objectInfo.Parts) > 0 {
|
|
|
|
for _, part := range objectInfo.Parts {
|
|
|
|
cumulativeActualSize += part.ActualSize
|
|
|
|
if cumulativeActualSize <= offset {
|
|
|
|
compressedOffset += part.Size
|
|
|
|
} else {
|
|
|
|
skipLength = cumulativeActualSize - part.ActualSize
|
|
|
|
break
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return compressedOffset, offset - skipLength
|
|
|
|
}
|
|
|
|
|
2016-04-29 17:24:10 -04:00
|
|
|
// byBucketName is a collection satisfying sort.Interface.
|
|
|
|
type byBucketName []BucketInfo
|
|
|
|
|
|
|
|
func (d byBucketName) Len() int { return len(d) }
|
|
|
|
func (d byBucketName) Swap(i, j int) { d[i], d[j] = d[j], d[i] }
|
|
|
|
func (d byBucketName) Less(i, j int) bool { return d[i].Name < d[j].Name }
|
2018-09-20 22:22:09 -04:00
|
|
|
|
|
|
|
// GetObjectReader is a type that wraps a reader with a lock to
|
|
|
|
// provide a ReadCloser interface that unlocks on Close()
|
|
|
|
type GetObjectReader struct {
|
|
|
|
ObjInfo ObjectInfo
|
|
|
|
pReader io.Reader
|
|
|
|
|
|
|
|
cleanUpFns []func()
|
2019-03-06 15:38:41 -05:00
|
|
|
precondFn func(ObjectInfo, string) bool
|
2018-09-20 22:22:09 -04:00
|
|
|
once sync.Once
|
|
|
|
}
|
|
|
|
|
|
|
|
// NewGetObjectReaderFromReader sets up a GetObjectReader with a given
|
|
|
|
// reader. This ignores any object properties.
|
2019-03-06 15:38:41 -05:00
|
|
|
func NewGetObjectReaderFromReader(r io.Reader, oi ObjectInfo, pcfn CheckCopyPreconditionFn, cleanupFns ...func()) (*GetObjectReader, error) {
|
|
|
|
if pcfn != nil {
|
|
|
|
if ok := pcfn(oi, ""); ok {
|
|
|
|
// Call the cleanup funcs
|
|
|
|
for i := len(cleanupFns) - 1; i >= 0; i-- {
|
|
|
|
cleanupFns[i]()
|
|
|
|
}
|
|
|
|
return nil, PreConditionFailed{}
|
|
|
|
}
|
|
|
|
}
|
2018-09-20 22:22:09 -04:00
|
|
|
return &GetObjectReader{
|
|
|
|
ObjInfo: oi,
|
|
|
|
pReader: r,
|
|
|
|
cleanUpFns: cleanupFns,
|
2019-03-06 15:38:41 -05:00
|
|
|
precondFn: pcfn,
|
|
|
|
}, nil
|
2018-09-20 22:22:09 -04:00
|
|
|
}
|
|
|
|
|
|
|
|
// ObjReaderFn is a function type that takes a reader and returns
|
|
|
|
// GetObjectReader and an error. Request headers are passed to provide
|
|
|
|
// encryption parameters. cleanupFns allow cleanup funcs to be
|
|
|
|
// registered for calling after usage of the reader.
|
2019-03-06 15:38:41 -05:00
|
|
|
type ObjReaderFn func(inputReader io.Reader, h http.Header, pcfn CheckCopyPreconditionFn, cleanupFns ...func()) (r *GetObjectReader, err error)
|
2018-09-20 22:22:09 -04:00
|
|
|
|
|
|
|
// NewGetObjectReader creates a new GetObjectReader. The cleanUpFns
|
|
|
|
// are called on Close() in reverse order as passed here. NOTE: It is
|
|
|
|
// assumed that clean up functions do not panic (otherwise, they may
|
|
|
|
// not all run!).
|
2019-03-06 15:38:41 -05:00
|
|
|
func NewGetObjectReader(rs *HTTPRangeSpec, oi ObjectInfo, pcfn CheckCopyPreconditionFn, cleanUpFns ...func()) (
|
2018-09-20 22:22:09 -04:00
|
|
|
fn ObjReaderFn, off, length int64, err error) {
|
|
|
|
|
|
|
|
// Call the clean-up functions immediately in case of exit
|
|
|
|
// with error
|
|
|
|
defer func() {
|
|
|
|
if err != nil {
|
|
|
|
for i := len(cleanUpFns) - 1; i >= 0; i-- {
|
|
|
|
cleanUpFns[i]()
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}()
|
|
|
|
|
|
|
|
isEncrypted := crypto.IsEncrypted(oi.UserDefined)
|
2019-09-26 02:08:24 -04:00
|
|
|
isCompressed, err := oi.IsCompressedOK()
|
|
|
|
if err != nil {
|
|
|
|
return nil, 0, 0, err
|
|
|
|
}
|
2018-09-20 22:22:09 -04:00
|
|
|
var skipLen int64
|
|
|
|
// Calculate range to read (different for
|
|
|
|
// e.g. encrypted/compressed objects)
|
|
|
|
switch {
|
|
|
|
case isEncrypted:
|
|
|
|
var seqNumber uint32
|
|
|
|
var partStart int
|
|
|
|
off, length, skipLen, seqNumber, partStart, err = oi.GetDecryptedRange(rs)
|
|
|
|
if err != nil {
|
|
|
|
return nil, 0, 0, err
|
|
|
|
}
|
|
|
|
var decSize int64
|
|
|
|
decSize, err = oi.DecryptedSize()
|
|
|
|
if err != nil {
|
|
|
|
return nil, 0, 0, err
|
|
|
|
}
|
|
|
|
var decRangeLength int64
|
|
|
|
decRangeLength, err = rs.GetLength(decSize)
|
|
|
|
if err != nil {
|
|
|
|
return nil, 0, 0, err
|
|
|
|
}
|
|
|
|
|
|
|
|
// We define a closure that performs decryption given
|
|
|
|
// a reader that returns the desired range of
|
|
|
|
// encrypted bytes. The header parameter is used to
|
|
|
|
// provide encryption parameters.
|
2019-03-06 15:38:41 -05:00
|
|
|
fn = func(inputReader io.Reader, h http.Header, pcfn CheckCopyPreconditionFn, cFns ...func()) (r *GetObjectReader, err error) {
|
2018-09-25 15:39:46 -04:00
|
|
|
copySource := h.Get(crypto.SSECopyAlgorithm) != ""
|
|
|
|
|
2018-09-20 22:22:09 -04:00
|
|
|
cFns = append(cleanUpFns, cFns...)
|
|
|
|
// Attach decrypter on inputReader
|
|
|
|
var decReader io.Reader
|
|
|
|
decReader, err = DecryptBlocksRequestR(inputReader, h,
|
2018-09-25 15:39:46 -04:00
|
|
|
off, length, seqNumber, partStart, oi, copySource)
|
2018-09-20 22:22:09 -04:00
|
|
|
if err != nil {
|
|
|
|
// Call the cleanup funcs
|
|
|
|
for i := len(cFns) - 1; i >= 0; i-- {
|
|
|
|
cFns[i]()
|
|
|
|
}
|
|
|
|
return nil, err
|
|
|
|
}
|
2019-03-06 15:38:41 -05:00
|
|
|
encETag := oi.ETag
|
|
|
|
oi.ETag = getDecryptedETag(h, oi, copySource) // Decrypt the ETag before top layer consumes this value.
|
|
|
|
|
|
|
|
if pcfn != nil {
|
|
|
|
if ok := pcfn(oi, encETag); ok {
|
|
|
|
// Call the cleanup funcs
|
|
|
|
for i := len(cFns) - 1; i >= 0; i-- {
|
|
|
|
cFns[i]()
|
|
|
|
}
|
|
|
|
return nil, PreConditionFailed{}
|
|
|
|
}
|
|
|
|
}
|
2018-12-19 17:12:53 -05:00
|
|
|
|
2018-09-20 22:22:09 -04:00
|
|
|
// Apply the skipLen and limit on the
|
|
|
|
// decrypted stream
|
|
|
|
decReader = io.LimitReader(ioutil.NewSkipReader(decReader, skipLen), decRangeLength)
|
|
|
|
|
|
|
|
// Assemble the GetObjectReader
|
|
|
|
r = &GetObjectReader{
|
|
|
|
ObjInfo: oi,
|
|
|
|
pReader: decReader,
|
|
|
|
cleanUpFns: cFns,
|
2019-03-06 15:38:41 -05:00
|
|
|
precondFn: pcfn,
|
2018-09-20 22:22:09 -04:00
|
|
|
}
|
|
|
|
return r, nil
|
|
|
|
}
|
2018-09-27 23:36:17 -04:00
|
|
|
case isCompressed:
|
|
|
|
// Read the decompressed size from the meta.json.
|
|
|
|
actualSize := oi.GetActualSize()
|
|
|
|
if actualSize < 0 {
|
|
|
|
return nil, 0, 0, errInvalidDecompressedSize
|
|
|
|
}
|
|
|
|
off, length = int64(0), oi.Size
|
|
|
|
decOff, decLength := int64(0), actualSize
|
|
|
|
if rs != nil {
|
|
|
|
off, length, err = rs.GetOffsetLength(actualSize)
|
|
|
|
if err != nil {
|
|
|
|
return nil, 0, 0, err
|
|
|
|
}
|
2019-09-26 02:08:24 -04:00
|
|
|
// In case of range based queries on multiparts, the offset and length are reduced.
|
2018-09-27 23:36:17 -04:00
|
|
|
off, decOff = getCompressedOffsets(oi, off)
|
|
|
|
decLength = length
|
|
|
|
length = oi.Size - off
|
|
|
|
|
|
|
|
// For negative length we read everything.
|
|
|
|
if decLength < 0 {
|
|
|
|
decLength = actualSize - decOff
|
|
|
|
}
|
|
|
|
|
|
|
|
// Reply back invalid range if the input offset and length fall out of range.
|
|
|
|
if decOff > actualSize || decOff+decLength > actualSize {
|
|
|
|
return nil, 0, 0, errInvalidRange
|
|
|
|
}
|
|
|
|
}
|
2019-03-06 15:38:41 -05:00
|
|
|
fn = func(inputReader io.Reader, _ http.Header, pcfn CheckCopyPreconditionFn, cFns ...func()) (r *GetObjectReader, err error) {
|
|
|
|
cFns = append(cleanUpFns, cFns...)
|
|
|
|
if pcfn != nil {
|
|
|
|
if ok := pcfn(oi, ""); ok {
|
|
|
|
// Call the cleanup funcs
|
|
|
|
for i := len(cFns) - 1; i >= 0; i-- {
|
|
|
|
cFns[i]()
|
|
|
|
}
|
|
|
|
return nil, PreConditionFailed{}
|
|
|
|
}
|
|
|
|
}
|
2018-09-27 23:36:17 -04:00
|
|
|
// Decompression reader.
|
2019-09-26 02:08:24 -04:00
|
|
|
s2Reader := s2.NewReader(inputReader)
|
|
|
|
// Apply the skipLen and limit on the decompressed stream.
|
|
|
|
err = s2Reader.Skip(decOff)
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
|
|
|
|
decReader := io.LimitReader(s2Reader, decLength)
|
|
|
|
if decLength > compReadAheadSize {
|
|
|
|
rah, err := readahead.NewReaderSize(decReader, compReadAheadBuffers, compReadAheadBufSize)
|
|
|
|
if err == nil {
|
|
|
|
decReader = rah
|
|
|
|
cFns = append(cFns, func() {
|
|
|
|
rah.Close()
|
|
|
|
})
|
|
|
|
}
|
|
|
|
}
|
2018-09-27 23:36:17 -04:00
|
|
|
oi.Size = decLength
|
|
|
|
|
|
|
|
// Assemble the GetObjectReader
|
|
|
|
r = &GetObjectReader{
|
|
|
|
ObjInfo: oi,
|
|
|
|
pReader: decReader,
|
2019-03-06 15:38:41 -05:00
|
|
|
cleanUpFns: cFns,
|
|
|
|
precondFn: pcfn,
|
2018-09-27 23:36:17 -04:00
|
|
|
}
|
|
|
|
return r, nil
|
|
|
|
}
|
2018-09-20 22:22:09 -04:00
|
|
|
|
|
|
|
default:
|
|
|
|
off, length, err = rs.GetOffsetLength(oi.Size)
|
|
|
|
if err != nil {
|
|
|
|
return nil, 0, 0, err
|
|
|
|
}
|
2019-03-06 15:38:41 -05:00
|
|
|
fn = func(inputReader io.Reader, _ http.Header, pcfn CheckCopyPreconditionFn, cFns ...func()) (r *GetObjectReader, err error) {
|
|
|
|
cFns = append(cleanUpFns, cFns...)
|
|
|
|
if pcfn != nil {
|
|
|
|
if ok := pcfn(oi, ""); ok {
|
|
|
|
// Call the cleanup funcs
|
|
|
|
for i := len(cFns) - 1; i >= 0; i-- {
|
|
|
|
cFns[i]()
|
|
|
|
}
|
|
|
|
return nil, PreConditionFailed{}
|
|
|
|
}
|
|
|
|
}
|
2018-09-20 22:22:09 -04:00
|
|
|
r = &GetObjectReader{
|
|
|
|
ObjInfo: oi,
|
|
|
|
pReader: inputReader,
|
2019-03-06 15:38:41 -05:00
|
|
|
cleanUpFns: cFns,
|
|
|
|
precondFn: pcfn,
|
2018-09-20 22:22:09 -04:00
|
|
|
}
|
|
|
|
return r, nil
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return fn, off, length, nil
|
|
|
|
}
|
|
|
|
|
|
|
|
// Close - calls the cleanup actions in reverse order
|
|
|
|
func (g *GetObjectReader) Close() error {
|
|
|
|
// sync.Once is used here to ensure that Close() is
|
|
|
|
// idempotent.
|
|
|
|
g.once.Do(func() {
|
|
|
|
for i := len(g.cleanUpFns) - 1; i >= 0; i-- {
|
|
|
|
g.cleanUpFns[i]()
|
|
|
|
}
|
|
|
|
})
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
// Read - to implement Reader interface.
|
|
|
|
func (g *GetObjectReader) Read(p []byte) (n int, err error) {
|
|
|
|
n, err = g.pReader.Read(p)
|
|
|
|
if err != nil {
|
|
|
|
// Calling code may not Close() in case of error, so
|
|
|
|
// we ensure it.
|
|
|
|
g.Close()
|
|
|
|
}
|
|
|
|
return
|
|
|
|
}
|
2018-11-14 20:36:41 -05:00
|
|
|
|
|
|
|
//SealMD5CurrFn seals md5sum with object encryption key and returns sealed
|
|
|
|
// md5sum
|
|
|
|
type SealMD5CurrFn func([]byte) []byte
|
|
|
|
|
|
|
|
// PutObjReader is a type that wraps sio.EncryptReader and
|
|
|
|
// underlying hash.Reader in a struct
|
|
|
|
type PutObjReader struct {
|
|
|
|
*hash.Reader // actual data stream
|
|
|
|
rawReader *hash.Reader // original data stream
|
|
|
|
sealMD5Fn SealMD5CurrFn
|
|
|
|
}
|
|
|
|
|
|
|
|
// Size returns the absolute number of bytes the Reader
|
|
|
|
// will return during reading. It returns -1 for unlimited
|
|
|
|
// data.
|
|
|
|
func (p *PutObjReader) Size() int64 {
|
|
|
|
return p.Reader.Size()
|
|
|
|
}
|
|
|
|
|
|
|
|
// MD5CurrentHexString returns the current MD5Sum or encrypted MD5Sum
|
|
|
|
// as a hex encoded string
|
|
|
|
func (p *PutObjReader) MD5CurrentHexString() string {
|
|
|
|
md5sumCurr := p.rawReader.MD5Current()
|
2019-05-08 21:35:40 -04:00
|
|
|
var appendHyphen bool
|
|
|
|
// md5sumcurr is not empty in two scenarios
|
|
|
|
// - server is running in strict compatibility mode
|
|
|
|
// - client set Content-Md5 during PUT operation
|
|
|
|
if len(md5sumCurr) == 0 {
|
|
|
|
// md5sumCurr is only empty when we are running
|
|
|
|
// in non-compatibility mode.
|
|
|
|
md5sumCurr = make([]byte, 16)
|
|
|
|
rand.Read(md5sumCurr)
|
|
|
|
appendHyphen = true
|
|
|
|
}
|
2018-11-14 20:36:41 -05:00
|
|
|
if p.sealMD5Fn != nil {
|
|
|
|
md5sumCurr = p.sealMD5Fn(md5sumCurr)
|
|
|
|
}
|
2019-05-08 21:35:40 -04:00
|
|
|
if appendHyphen {
|
|
|
|
// Make sure to return etag string upto 32 length, for SSE
|
|
|
|
// requests ETag might be longer and the code decrypting the
|
|
|
|
// ETag ignores ETag in multipart ETag form i.e <hex>-N
|
|
|
|
return hex.EncodeToString(md5sumCurr)[:32] + "-1"
|
|
|
|
}
|
2018-11-14 20:36:41 -05:00
|
|
|
return hex.EncodeToString(md5sumCurr)
|
|
|
|
}
|
|
|
|
|
|
|
|
// NewPutObjReader returns a new PutObjReader and holds
|
|
|
|
// reference to underlying data stream from client and the encrypted
|
|
|
|
// data reader
|
|
|
|
func NewPutObjReader(rawReader *hash.Reader, encReader *hash.Reader, encKey []byte) *PutObjReader {
|
|
|
|
p := PutObjReader{Reader: rawReader, rawReader: rawReader}
|
|
|
|
|
2018-11-27 16:23:32 -05:00
|
|
|
if len(encKey) != 0 && encReader != nil {
|
|
|
|
var objKey crypto.ObjectKey
|
|
|
|
copy(objKey[:], encKey)
|
|
|
|
p.sealMD5Fn = sealETagFn(objKey)
|
2018-11-14 20:36:41 -05:00
|
|
|
p.Reader = encReader
|
|
|
|
}
|
2018-11-27 16:23:32 -05:00
|
|
|
|
2018-11-14 20:36:41 -05:00
|
|
|
return &p
|
|
|
|
}
|
|
|
|
|
|
|
|
func sealETag(encKey crypto.ObjectKey, md5CurrSum []byte) []byte {
|
|
|
|
var emptyKey [32]byte
|
|
|
|
if bytes.Equal(encKey[:], emptyKey[:]) {
|
|
|
|
return md5CurrSum
|
|
|
|
}
|
|
|
|
return encKey.SealETag(md5CurrSum)
|
|
|
|
}
|
2018-11-27 16:23:32 -05:00
|
|
|
|
2018-11-14 20:36:41 -05:00
|
|
|
func sealETagFn(key crypto.ObjectKey) SealMD5CurrFn {
|
2018-11-27 16:23:32 -05:00
|
|
|
fn := func(md5sumcurr []byte) []byte {
|
2018-11-14 20:36:41 -05:00
|
|
|
return sealETag(key, md5sumcurr)
|
|
|
|
}
|
2018-11-27 16:23:32 -05:00
|
|
|
return fn
|
2018-11-14 20:36:41 -05:00
|
|
|
}
|
2019-01-05 17:16:43 -05:00
|
|
|
|
|
|
|
// CleanMinioInternalMetadataKeys removes X-Amz-Meta- prefix from minio internal
|
|
|
|
// encryption metadata that was sent by minio gateway
|
|
|
|
func CleanMinioInternalMetadataKeys(metadata map[string]string) map[string]string {
|
|
|
|
var newMeta = make(map[string]string, len(metadata))
|
|
|
|
for k, v := range metadata {
|
|
|
|
if strings.HasPrefix(k, "X-Amz-Meta-X-Minio-Internal-") {
|
|
|
|
newMeta[strings.TrimPrefix(k, "X-Amz-Meta-")] = v
|
|
|
|
} else {
|
|
|
|
newMeta[k] = v
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return newMeta
|
|
|
|
}
|
2019-03-05 11:35:37 -05:00
|
|
|
|
2019-09-26 02:08:24 -04:00
|
|
|
// newS2CompressReader will read data from r, compress it and return the compressed data as a Reader.
|
|
|
|
// Use Close to ensure resources are released on incomplete streams.
|
|
|
|
func newS2CompressReader(r io.Reader) io.ReadCloser {
|
|
|
|
pr, pw := io.Pipe()
|
|
|
|
comp := s2.NewWriter(pw)
|
|
|
|
// Copy input to compressor
|
|
|
|
go func() {
|
|
|
|
_, err := io.Copy(comp, r)
|
2019-03-05 11:35:37 -05:00
|
|
|
if err != nil {
|
2019-09-26 02:08:24 -04:00
|
|
|
comp.Close()
|
|
|
|
pw.CloseWithError(err)
|
|
|
|
return
|
2019-03-05 11:35:37 -05:00
|
|
|
}
|
2019-09-26 02:08:24 -04:00
|
|
|
// Close the stream.
|
|
|
|
err = comp.Close()
|
|
|
|
if err != nil {
|
|
|
|
pw.CloseWithError(err)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
// Everything ok, do regular close.
|
|
|
|
pw.Close()
|
|
|
|
}()
|
|
|
|
return pr
|
2019-03-05 11:35:37 -05:00
|
|
|
}
|
2019-06-29 01:09:17 -04:00
|
|
|
|
|
|
|
// Returns error if the cancelCh has been closed (indicating that S3 client has disconnected)
|
|
|
|
type detectDisconnect struct {
|
|
|
|
io.ReadCloser
|
|
|
|
cancelCh <-chan struct{}
|
|
|
|
}
|
|
|
|
|
|
|
|
func (d *detectDisconnect) Read(p []byte) (int, error) {
|
|
|
|
select {
|
|
|
|
case <-d.cancelCh:
|
|
|
|
return 0, io.ErrUnexpectedEOF
|
|
|
|
default:
|
|
|
|
return d.ReadCloser.Read(p)
|
|
|
|
}
|
|
|
|
}
|