Report file checksum when encountering artifact transform failures (#…

…87522) (#87531)
elastic · Jun 8, 2022 · 9905bfb · 9905bfb
1 parent 1f2a906
commit 9905bfb
Showing 1 changed file with 30 additions and 5 deletions.
diff --git a/build-tools/src/main/java/org/elasticsearch/gradle/transform/UnpackTransform.java b/build-tools/src/main/java/org/elasticsearch/gradle/transform/UnpackTransform.java
@@ -13,6 +13,7 @@
 import org.gradle.api.artifacts.transform.TransformOutputs;
 import org.gradle.api.artifacts.transform.TransformParameters;
 import org.gradle.api.file.FileSystemLocation;
+import org.gradle.api.logging.Logger;
 import org.gradle.api.logging.Logging;
 import org.gradle.api.provider.Property;
 import org.gradle.api.provider.Provider;
@@ -21,17 +22,25 @@
 import org.gradle.api.tasks.Optional;
 import org.gradle.api.tasks.PathSensitive;
 import org.gradle.api.tasks.PathSensitivity;
-import org.gradle.internal.UncheckedException;
 
 import java.io.File;
+import java.io.FileInputStream;
 import java.io.IOException;
+import java.io.InputStream;
+import java.io.UncheckedIOException;
 import java.nio.file.Path;
 import java.nio.file.Paths;
+import java.security.DigestInputStream;
+import java.security.MessageDigest;
+import java.security.NoSuchAlgorithmException;
+import java.util.HexFormat;
 import java.util.List;
 import java.util.function.Function;
 
 public interface UnpackTransform extends TransformAction<UnpackTransform.Parameters> {
 
+    Logger LOGGER = Logging.getLogger(UnpackTransform.class);
+
     interface Parameters extends TransformParameters {
         @Input
         @Optional
@@ -78,11 +87,16 @@ default void transform(TransformOutputs outputs) {
         }
 
         try {
-            Logging.getLogger(UnpackTransform.class)
-                .info("Unpacking " + archiveFile.getName() + " using " + getClass().getSimpleName() + ".");
+            LOGGER.info("Unpacking {} using {}.", archiveFile.getName(), getClass().getSimpleName());
             unpack(archiveFile, extractedDir, outputs, getParameters().getAsFiletreeOutput());
-        } catch (IOException e) {
-            throw UncheckedException.throwAsUncheckedException(e);
+        } catch (IOException e1) {
+            String hash = "[unknown]";
+            try {
+                hash = getSha1(archiveFile);
+            } catch (Exception e2) {
+                LOGGER.warn("Unable to calculate hash for file " + archiveFile.getPath(), e2);
+            }
+            throw new UncheckedIOException("Failed to unpack " + archiveFile.getName() + " with checksum " + hash, e1);
         }
     }
 
@@ -137,4 +151,15 @@ static Path trimArchiveExtractPath(List<String> keepPatterns, String ignoredPatt
         // finally remove the top-level directories from the output path
         return entryName.subpath(index + 1, entryName.getNameCount());
     }
+
+    static String getSha1(File file) throws IOException, NoSuchAlgorithmException {
+        MessageDigest digest = MessageDigest.getInstance("SHA-1");
+        try (InputStream is = new DigestInputStream(new FileInputStream(file), digest)) {
+            byte[] buffer = new byte[4096];
+            while (is.read(buffer) > 0) {
+                // loop
+            }
+        }
+        return HexFormat.of().formatHex(digest.digest());
+    }
 }