You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
445 lines
14 KiB
445 lines
14 KiB
/*
|
|
* Minio Cloud Storage, (C) 2016 Minio, Inc.
|
|
*
|
|
* 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.
|
|
*/
|
|
|
|
package main
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"io"
|
|
"path"
|
|
"sort"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"github.com/Sirupsen/logrus"
|
|
"github.com/skyrings/skyring-common/tools/uuid"
|
|
)
|
|
|
|
// MultipartPartInfo Info of each part kept in the multipart metadata file after
|
|
// CompleteMultipartUpload() is called.
|
|
type MultipartPartInfo struct {
|
|
PartNumber int
|
|
ETag string
|
|
Size int64
|
|
}
|
|
|
|
// MultipartObjectInfo - contents of the multipart metadata file after
|
|
// CompleteMultipartUpload() is called.
|
|
type MultipartObjectInfo []MultipartPartInfo
|
|
|
|
// GetSize - Return the size of the object.
|
|
func (m MultipartObjectInfo) GetSize() (size int64) {
|
|
for _, part := range m {
|
|
size += part.Size
|
|
}
|
|
return
|
|
}
|
|
|
|
// GetPartNumberOffset - given an offset for the whole object, return the part and offset in that part.
|
|
func (m MultipartObjectInfo) GetPartNumberOffset(offset int64) (partIndex int, partOffset int64, err error) {
|
|
partOffset = offset
|
|
for i, part := range m {
|
|
partIndex = i
|
|
if partOffset < part.Size {
|
|
return
|
|
}
|
|
partOffset -= part.Size
|
|
}
|
|
// Offset beyond the size of the object
|
|
err = errUnexpected
|
|
return
|
|
}
|
|
|
|
func partNumToPartFileName(partNum int) string {
|
|
return fmt.Sprintf("%.5d%s", partNum, multipartSuffix)
|
|
}
|
|
|
|
// listLeafEntries - lists all entries if a given prefixPath is a leaf
|
|
// directory, returns error if any - returns empty list if prefixPath
|
|
// is not a leaf directory.
|
|
func (xl xlObjects) listLeafEntries(prefixPath string) (entries []FileInfo, e error) {
|
|
var markerPath string
|
|
for {
|
|
fileInfos, eof, err := xl.storage.ListFiles(minioMetaBucket, prefixPath, markerPath, false, 1000)
|
|
if err != nil {
|
|
log.WithFields(logrus.Fields{
|
|
"prefixPath": prefixPath,
|
|
"markerPath": markerPath,
|
|
}).Errorf("%s", err)
|
|
return nil, err
|
|
}
|
|
for _, fileInfo := range fileInfos {
|
|
// Set marker for next batch of ListFiles.
|
|
markerPath = fileInfo.Name
|
|
if fileInfo.Mode.IsDir() {
|
|
// If a directory is found, doesn't return anything.
|
|
return nil, nil
|
|
}
|
|
fileName := path.Base(fileInfo.Name)
|
|
if !strings.Contains(fileName, ".") {
|
|
// Skip the entry if it is of the pattern bucket/object/uploadID.partNum.md5sum
|
|
// and retain entries of the pattern bucket/object/uploadID
|
|
entries = append(entries, fileInfo)
|
|
}
|
|
}
|
|
if eof {
|
|
break
|
|
}
|
|
}
|
|
return entries, nil
|
|
}
|
|
|
|
// listMetaBucketFiles - list all files at a given prefix inside minioMetaBucket.
|
|
func (xl xlObjects) listMetaBucketFiles(prefixPath string, markerPath string, recursive bool, maxKeys int) (allFileInfos []FileInfo, eof bool, err error) {
|
|
// newMaxKeys tracks the size of entries which are going to be
|
|
// returned back.
|
|
var newMaxKeys int
|
|
|
|
// Following loop gathers and filters out special files inside
|
|
// minio meta volume.
|
|
for {
|
|
var fileInfos []FileInfo
|
|
// List files up to maxKeys-newMaxKeys, since we are skipping entries for special files.
|
|
fileInfos, eof, err = xl.storage.ListFiles(minioMetaBucket, prefixPath, markerPath, recursive, maxKeys-newMaxKeys)
|
|
if err != nil {
|
|
log.WithFields(logrus.Fields{
|
|
"prefixPath": prefixPath,
|
|
"markerPath": markerPath,
|
|
"recursive": recursive,
|
|
"maxKeys": maxKeys,
|
|
}).Errorf("%s", err)
|
|
return nil, true, err
|
|
}
|
|
// Loop through and validate individual file.
|
|
for _, fi := range fileInfos {
|
|
var entries []FileInfo
|
|
if fi.Mode.IsDir() {
|
|
// List all the entries if fi.Name is a leaf directory, if
|
|
// fi.Name is not a leaf directory then the resulting
|
|
// entries are empty.
|
|
entries, err = xl.listLeafEntries(fi.Name)
|
|
if err != nil {
|
|
log.WithFields(logrus.Fields{
|
|
"prefixPath": fi.Name,
|
|
}).Errorf("%s", err)
|
|
return nil, false, err
|
|
}
|
|
}
|
|
// Set markerPath for next batch of listing.
|
|
markerPath = fi.Name
|
|
if len(entries) > 0 {
|
|
// We reach here for non-recursive case and a leaf entry.
|
|
for _, entry := range entries {
|
|
allFileInfos = append(allFileInfos, entry)
|
|
newMaxKeys++
|
|
// If we have reached the maxKeys, it means we have listed
|
|
// everything that was requested. Return right here.
|
|
if newMaxKeys == maxKeys {
|
|
// Return values:
|
|
// allFileInfos : "maxKeys" number of entries.
|
|
// eof : eof returned by xl.storage.ListFiles()
|
|
// error : nil
|
|
return
|
|
}
|
|
}
|
|
} else {
|
|
// We reach here for a non-recursive case non-leaf entry
|
|
// OR recursive case with fi.Name matching pattern bucket/object/uploadID[.partNum.md5sum]
|
|
if !fi.Mode.IsDir() { // Do not skip non-recursive case directory entries.
|
|
// Skip files matching pattern bucket/object/uploadID.partNum.md5sum
|
|
// and retain files matching pattern bucket/object/uploadID
|
|
specialFile := path.Base(fi.Name)
|
|
if strings.Contains(specialFile, ".") {
|
|
// Contains partnumber and md5sum info, skip this.
|
|
continue
|
|
}
|
|
}
|
|
}
|
|
allFileInfos = append(allFileInfos, fi)
|
|
newMaxKeys++
|
|
// If we have reached the maxKeys, it means we have listed
|
|
// everything that was requested. Return right here.
|
|
if newMaxKeys == maxKeys {
|
|
// Return values:
|
|
// allFileInfos : "maxKeys" number of entries.
|
|
// eof : eof returned by xl.storage.ListFiles()
|
|
// error : nil
|
|
return
|
|
}
|
|
}
|
|
// If we have reached eof then we break out.
|
|
if eof {
|
|
break
|
|
}
|
|
}
|
|
|
|
// Return entries here.
|
|
return allFileInfos, eof, nil
|
|
}
|
|
|
|
// ListMultipartUploads - list multipart uploads.
|
|
func (xl xlObjects) ListMultipartUploads(bucket, prefix, keyMarker, uploadIDMarker, delimiter string, maxUploads int) (ListMultipartsInfo, error) {
|
|
result := ListMultipartsInfo{}
|
|
// Verify if bucket is valid.
|
|
if !IsValidBucketName(bucket) {
|
|
return ListMultipartsInfo{}, (BucketNameInvalid{Bucket: bucket})
|
|
}
|
|
if !IsValidObjectPrefix(prefix) {
|
|
return ListMultipartsInfo{}, (ObjectNameInvalid{Bucket: bucket, Object: prefix})
|
|
}
|
|
// Verify if delimiter is anything other than '/', which we do not support.
|
|
if delimiter != "" && delimiter != slashSeparator {
|
|
return ListMultipartsInfo{}, (UnsupportedDelimiter{
|
|
Delimiter: delimiter,
|
|
})
|
|
}
|
|
// Verify if marker has prefix.
|
|
if keyMarker != "" && !strings.HasPrefix(keyMarker, prefix) {
|
|
return ListMultipartsInfo{}, (InvalidMarkerPrefixCombination{
|
|
Marker: keyMarker,
|
|
Prefix: prefix,
|
|
})
|
|
}
|
|
if uploadIDMarker != "" {
|
|
if strings.HasSuffix(keyMarker, slashSeparator) {
|
|
return result, (InvalidUploadIDKeyCombination{
|
|
UploadIDMarker: uploadIDMarker,
|
|
KeyMarker: keyMarker,
|
|
})
|
|
}
|
|
id, err := uuid.Parse(uploadIDMarker)
|
|
if err != nil {
|
|
return result, err
|
|
}
|
|
if id.IsZero() {
|
|
return result, (MalformedUploadID{
|
|
UploadID: uploadIDMarker,
|
|
})
|
|
}
|
|
}
|
|
|
|
recursive := true
|
|
if delimiter == slashSeparator {
|
|
recursive = false
|
|
}
|
|
|
|
result.IsTruncated = true
|
|
result.MaxUploads = maxUploads
|
|
|
|
// Not using path.Join() as it strips off the trailing '/'.
|
|
// Also bucket should always be followed by '/' even if prefix is empty.
|
|
prefixPath := pathJoin(bucket, prefix)
|
|
keyMarkerPath := ""
|
|
if keyMarker != "" {
|
|
keyMarkerPath = pathJoin(pathJoin(bucket, keyMarker), uploadIDMarker)
|
|
}
|
|
// List all the multipart files at prefixPath, starting with marker keyMarkerPath.
|
|
fileInfos, eof, err := xl.listMetaBucketFiles(prefixPath, keyMarkerPath, recursive, maxUploads)
|
|
if err != nil {
|
|
log.WithFields(logrus.Fields{
|
|
"prefixPath": prefixPath,
|
|
"markerPath": keyMarkerPath,
|
|
"recursive": recursive,
|
|
"maxUploads": maxUploads,
|
|
}).Errorf("listMetaBucketFiles failed with %s", err)
|
|
return ListMultipartsInfo{}, err
|
|
}
|
|
|
|
// Loop through all the received files fill in the multiparts result.
|
|
for _, fi := range fileInfos {
|
|
var objectName string
|
|
var uploadID string
|
|
if fi.Mode.IsDir() {
|
|
// All directory entries are common prefixes.
|
|
uploadID = "" // Upload ids are empty for CommonPrefixes.
|
|
objectName = strings.TrimPrefix(fi.Name, retainSlash(bucket))
|
|
result.CommonPrefixes = append(result.CommonPrefixes, objectName)
|
|
} else {
|
|
uploadID = path.Base(fi.Name)
|
|
objectName = strings.TrimPrefix(path.Dir(fi.Name), retainSlash(bucket))
|
|
result.Uploads = append(result.Uploads, uploadMetadata{
|
|
Object: objectName,
|
|
UploadID: uploadID,
|
|
Initiated: fi.ModTime,
|
|
})
|
|
}
|
|
result.NextKeyMarker = objectName
|
|
result.NextUploadIDMarker = uploadID
|
|
}
|
|
result.IsTruncated = !eof
|
|
if !result.IsTruncated {
|
|
result.NextKeyMarker = ""
|
|
result.NextUploadIDMarker = ""
|
|
}
|
|
return result, nil
|
|
}
|
|
|
|
// NewMultipartUpload - initialize a new multipart upload, returns a unique id.
|
|
func (xl xlObjects) NewMultipartUpload(bucket, object string) (string, error) {
|
|
return newMultipartUploadCommon(xl.storage, bucket, object)
|
|
}
|
|
|
|
// PutObjectPart - writes the multipart upload chunks.
|
|
func (xl xlObjects) PutObjectPart(bucket, object, uploadID string, partID int, size int64, data io.Reader, md5Hex string) (string, error) {
|
|
newMD5Hex, err := putObjectPartCommon(xl.storage, bucket, object, uploadID, partID, size, data, md5Hex)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
partSuffix := fmt.Sprintf("%s.%d", uploadID, partID)
|
|
partSuffixPath := path.Join(bucket, object, partSuffix)
|
|
partSuffixMD5 := fmt.Sprintf("%s.%.5d.%s", uploadID, partID, newMD5Hex)
|
|
partSuffixMD5Path := path.Join(bucket, object, partSuffixMD5)
|
|
err = xl.storage.RenameFile(minioMetaBucket, partSuffixPath, minioMetaBucket, partSuffixMD5Path)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
return newMD5Hex, nil
|
|
}
|
|
|
|
func (xl xlObjects) ListObjectParts(bucket, object, uploadID string, partNumberMarker, maxParts int) (ListPartsInfo, error) {
|
|
// Verify if bucket is valid.
|
|
if !IsValidBucketName(bucket) {
|
|
return ListPartsInfo{}, (BucketNameInvalid{Bucket: bucket})
|
|
}
|
|
if !IsValidObjectName(object) {
|
|
return ListPartsInfo{}, (ObjectNameInvalid{Bucket: bucket, Object: object})
|
|
}
|
|
if status, err := isUploadIDExists(xl.storage, bucket, object, uploadID); err != nil {
|
|
return ListPartsInfo{}, err
|
|
} else if !status {
|
|
return ListPartsInfo{}, (InvalidUploadID{UploadID: uploadID})
|
|
}
|
|
result := ListPartsInfo{}
|
|
var markerPath string
|
|
nextPartNumberMarker := 0
|
|
uploadIDPath := path.Join(bucket, object, uploadID)
|
|
// Figure out the marker for the next subsequent calls, if the
|
|
// partNumberMarker is already set.
|
|
if partNumberMarker > 0 {
|
|
partNumberMarkerPath := uploadIDPath + "." + fmt.Sprintf("%.5d", partNumberMarker) + "."
|
|
fileInfos, _, err := xl.storage.ListFiles(minioMetaBucket, partNumberMarkerPath, "", false, 1)
|
|
if err != nil {
|
|
return result, toObjectErr(err, minioMetaBucket, partNumberMarkerPath)
|
|
}
|
|
if len(fileInfos) == 0 {
|
|
return result, (InvalidPart{})
|
|
}
|
|
markerPath = fileInfos[0].Name
|
|
}
|
|
uploadIDPrefix := uploadIDPath + "."
|
|
fileInfos, eof, err := xl.storage.ListFiles(minioMetaBucket, uploadIDPrefix, markerPath, false, maxParts)
|
|
if err != nil {
|
|
return result, InvalidPart{}
|
|
}
|
|
for _, fileInfo := range fileInfos {
|
|
fileName := path.Base(fileInfo.Name)
|
|
splitResult := strings.Split(fileName, ".")
|
|
partNum, err := strconv.Atoi(splitResult[1])
|
|
if err != nil {
|
|
return result, err
|
|
}
|
|
md5sum := splitResult[2]
|
|
result.Parts = append(result.Parts, partInfo{
|
|
PartNumber: partNum,
|
|
LastModified: fileInfo.ModTime,
|
|
ETag: md5sum,
|
|
Size: fileInfo.Size,
|
|
})
|
|
nextPartNumberMarker = partNum
|
|
}
|
|
result.Bucket = bucket
|
|
result.Object = object
|
|
result.UploadID = uploadID
|
|
result.PartNumberMarker = partNumberMarker
|
|
result.NextPartNumberMarker = nextPartNumberMarker
|
|
result.MaxParts = maxParts
|
|
result.IsTruncated = !eof
|
|
return result, nil
|
|
}
|
|
|
|
func (xl xlObjects) CompleteMultipartUpload(bucket string, object string, uploadID string, parts []completePart) (string, error) {
|
|
// Verify if bucket is valid.
|
|
if !IsValidBucketName(bucket) {
|
|
return "", (BucketNameInvalid{Bucket: bucket})
|
|
}
|
|
if !IsValidObjectName(object) {
|
|
return "", (ObjectNameInvalid{
|
|
Bucket: bucket,
|
|
Object: object,
|
|
})
|
|
}
|
|
if status, err := isUploadIDExists(xl.storage, bucket, object, uploadID); err != nil {
|
|
return "", err
|
|
} else if !status {
|
|
return "", (InvalidUploadID{UploadID: uploadID})
|
|
}
|
|
sort.Sort(completedParts(parts))
|
|
var metadata MultipartObjectInfo
|
|
for _, part := range parts {
|
|
partSuffix := fmt.Sprintf("%s.%.5d.%s", uploadID, part.PartNumber, part.ETag)
|
|
fi, err := xl.storage.StatFile(minioMetaBucket, path.Join(bucket, object, partSuffix))
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
metadata = append(metadata, MultipartPartInfo{part.PartNumber, part.ETag, fi.Size})
|
|
}
|
|
var md5Sums []string
|
|
for _, part := range parts {
|
|
// Construct part suffix.
|
|
partSuffix := fmt.Sprintf("%s.%.5d.%s", uploadID, part.PartNumber, part.ETag)
|
|
err := xl.storage.RenameFile(minioMetaBucket, path.Join(bucket, object, partSuffix), bucket, path.Join(object, partNumToPartFileName(part.PartNumber)))
|
|
// We need a way to roll back if of the renames failed.
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
md5Sums = append(md5Sums, part.ETag)
|
|
}
|
|
|
|
if w, err := xl.storage.CreateFile(bucket, pathJoin(object, multipartMetaFile)); err == nil {
|
|
var b []byte
|
|
b, err = json.Marshal(metadata)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
_, err = w.Write(b)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
// Close the writer.
|
|
if err = w.Close(); err != nil {
|
|
return "", err
|
|
}
|
|
} else {
|
|
return "", toObjectErr(err, bucket, object)
|
|
}
|
|
if err := xl.storage.DeleteFile(minioMetaBucket, path.Join(bucket, object, uploadID)); err != nil {
|
|
return "", toObjectErr(err, bucket, object)
|
|
}
|
|
// Save the s3 md5.
|
|
s3MD5, err := makeS3MD5(md5Sums...)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
// Return md5sum.
|
|
return s3MD5, nil
|
|
}
|
|
|
|
// AbortMultipartUpload - aborts a multipart upload.
|
|
func (xl xlObjects) AbortMultipartUpload(bucket, object, uploadID string) error {
|
|
return abortMultipartUploadCommon(xl.storage, bucket, object, uploadID)
|
|
}
|
|
|