diff --git a/hadoop-common-project/hadoop-common/src/main/java/org/apache/hadoop/fs/FileSystemMultipartUploader.java b/hadoop-common-project/hadoop-common/src/main/java/org/apache/hadoop/fs/FileSystemMultipartUploader.java index a700a9fd0b..f13b50bd20 100644 --- a/hadoop-common-project/hadoop-common/src/main/java/org/apache/hadoop/fs/FileSystemMultipartUploader.java +++ b/hadoop-common-project/hadoop-common/src/main/java/org/apache/hadoop/fs/FileSystemMultipartUploader.java @@ -68,6 +68,7 @@ public PartHandle putPart(Path filePath, InputStream inputStream, throws IOException { byte[] uploadIdByteArray = uploadId.toByteArray(); + checkUploadId(uploadIdByteArray); Path collectorPath = new Path(new String(uploadIdByteArray, 0, uploadIdByteArray.length, Charsets.UTF_8)); Path partPath = @@ -101,6 +102,8 @@ public PathHandle complete(Path filePath, List> handles, UploadHandle multipartUploadId) throws IOException { + checkUploadId(multipartUploadId.toByteArray()); + if (handles.isEmpty()) { throw new IOException("Empty upload"); } @@ -133,8 +136,7 @@ public PathHandle complete(Path filePath, @Override public void abort(Path filePath, UploadHandle uploadId) throws IOException { byte[] uploadIdByteArray = uploadId.toByteArray(); - Preconditions.checkArgument(uploadIdByteArray.length != 0, - "UploadId is empty"); + checkUploadId(uploadIdByteArray); Path collectorPath = new Path(new String(uploadIdByteArray, 0, uploadIdByteArray.length, Charsets.UTF_8)); diff --git a/hadoop-common-project/hadoop-common/src/main/java/org/apache/hadoop/fs/MultipartUploader.java b/hadoop-common-project/hadoop-common/src/main/java/org/apache/hadoop/fs/MultipartUploader.java index 47fd9f29b9..76f58d3597 100644 --- a/hadoop-common-project/hadoop-common/src/main/java/org/apache/hadoop/fs/MultipartUploader.java +++ b/hadoop-common-project/hadoop-common/src/main/java/org/apache/hadoop/fs/MultipartUploader.java @@ -21,6 +21,7 @@ import java.io.InputStream; import java.util.List; +import com.google.common.base.Preconditions; import org.slf4j.Logger; import org.slf4j.LoggerFactory; @@ -91,4 +92,14 @@ public abstract PathHandle complete(Path filePath, public abstract void abort(Path filePath, UploadHandle multipartUploadId) throws IOException; + /** + * Utility method to validate uploadIDs + * @param uploadId + * @throws IllegalArgumentException + */ + protected void checkUploadId(byte[] uploadId) + throws IllegalArgumentException { + Preconditions.checkArgument(uploadId.length > 0, + "Empty UploadId is not valid"); + } } diff --git a/hadoop-common-project/hadoop-common/src/test/java/org/apache/hadoop/fs/contract/AbstractContractMultipartUploaderTest.java b/hadoop-common-project/hadoop-common/src/test/java/org/apache/hadoop/fs/contract/AbstractContractMultipartUploaderTest.java index c0e1600d52..85a6861637 100644 --- a/hadoop-common-project/hadoop-common/src/test/java/org/apache/hadoop/fs/contract/AbstractContractMultipartUploaderTest.java +++ b/hadoop-common-project/hadoop-common/src/test/java/org/apache/hadoop/fs/contract/AbstractContractMultipartUploaderTest.java @@ -297,4 +297,47 @@ public void testCompleteEmptyUpload() throws Exception { () -> mpu.complete(dest, new ArrayList<>(), handle)); mpu.abort(dest, handle); } + + /** + * When we pass empty uploadID, putPart throws IllegalArgumentException. + * @throws Exception + */ + @Test + public void testPutPartEmptyUploadID() throws Exception { + describe("Expect IllegalArgumentException when putPart uploadID is empty"); + FileSystem fs = getFileSystem(); + Path dest = path("testCompleteEmptyUpload"); + MultipartUploader mpu = MultipartUploaderFactory.get(fs, null); + mpu.initialize(dest); + UploadHandle emptyHandle = + BBUploadHandle.from(ByteBuffer.wrap(new byte[0])); + byte[] payload = generatePayload(1); + InputStream is = new ByteArrayInputStream(payload); + intercept(IllegalArgumentException.class, + () -> mpu.putPart(dest, is, 1, emptyHandle, payload.length)); + } + + /** + * When we pass empty uploadID, complete throws IllegalArgumentException. + * @throws Exception + */ + @Test + public void testCompleteEmptyUploadID() throws Exception { + describe("Expect IllegalArgumentException when complete uploadID is empty"); + FileSystem fs = getFileSystem(); + Path dest = path("testCompleteEmptyUpload"); + MultipartUploader mpu = MultipartUploaderFactory.get(fs, null); + UploadHandle realHandle = mpu.initialize(dest); + UploadHandle emptyHandle = + BBUploadHandle.from(ByteBuffer.wrap(new byte[0])); + List> partHandles = new ArrayList<>(); + byte[] payload = generatePayload(1); + InputStream is = new ByteArrayInputStream(payload); + PartHandle partHandle = mpu.putPart(dest, is, 1, realHandle, + payload.length); + partHandles.add(Pair.of(1, partHandle)); + + intercept(IllegalArgumentException.class, + () -> mpu.complete(dest, partHandles, emptyHandle)); + } } diff --git a/hadoop-tools/hadoop-aws/src/main/java/org/apache/hadoop/fs/s3a/S3AMultipartUploader.java b/hadoop-tools/hadoop-aws/src/main/java/org/apache/hadoop/fs/s3a/S3AMultipartUploader.java index 6a1df54bd6..4a6cb8c093 100644 --- a/hadoop-tools/hadoop-aws/src/main/java/org/apache/hadoop/fs/s3a/S3AMultipartUploader.java +++ b/hadoop-tools/hadoop-aws/src/main/java/org/apache/hadoop/fs/s3a/S3AMultipartUploader.java @@ -84,9 +84,10 @@ public UploadHandle initialize(Path filePath) throws IOException { public PartHandle putPart(Path filePath, InputStream inputStream, int partNumber, UploadHandle uploadId, long lengthInBytes) throws IOException { - final WriteOperationHelper writeHelper = s3a.getWriteOperationHelper(); - String key = s3a.pathToKey(filePath); byte[] uploadIdBytes = uploadId.toByteArray(); + checkUploadId(uploadIdBytes); + String key = s3a.pathToKey(filePath); + final WriteOperationHelper writeHelper = s3a.getWriteOperationHelper(); String uploadIdString = new String(uploadIdBytes, 0, uploadIdBytes.length, Charsets.UTF_8); UploadPartRequest request = writeHelper.newUploadPartRequest(key, @@ -155,11 +156,6 @@ protected MultipartUploader createMultipartUploader(FileSystem fs, } } - private void checkUploadId(byte[] uploadId) throws IllegalArgumentException { - Preconditions.checkArgument(uploadId.length > 0, - "Empty UploadId is not valid"); - } - /** * Build the payload for marshalling. * @param eTag upload etag