Ensure a stashedFile parameter is retained on a Jenkins restart while in the queue

Author: mawinter69

src/main/java/io/jenkins/plugins/file_parameters/StashedFileParameterValue.java

@@ -26,16 +26,22 @@
 
 import edu.umd.cs.findbugs.annotations.SuppressFBWarnings;
 import hudson.EnvVars;
+import hudson.Extension;
 import hudson.FilePath;
 import hudson.Launcher;
 import hudson.Util;
+import hudson.model.ParametersAction;
+import hudson.model.Queue;
 import hudson.model.Run;
 import hudson.model.TaskListener;
+import hudson.model.queue.QueueListener;
 import java.io.File;
 import java.io.IOException;
 import java.io.InputStream;
-import java.nio.file.Files;
 
+import java.util.List;
+import java.util.logging.Level;
+import java.util.logging.Logger;
 import org.apache.commons.fileupload.FileItem;
 import org.apache.commons.io.FileUtils;
 import org.jenkinsci.plugins.workflow.flow.FlowExecutionOwner;
@@ -44,11 +50,14 @@
 
 public final class StashedFileParameterValue extends AbstractFileParameterValue {
 
+    private static final Logger LOGGER = Logger.getLogger(StashedFileParameterValue.class.getName());
+
     private static final long serialVersionUID = 1L;
 
+    private final String tmpFile;
+
     @SuppressFBWarnings(value = "SE_TRANSIENT_FIELD_NOT_RESTORED", justification = "Doesn't make sense to persist it")
-    private transient File tmp;
-    
+
     @DataBoundConstructor public StashedFileParameterValue(String name, FileItem file) throws IOException {
         this(name, file.getInputStream());
         setFilename(file.getName());
@@ -57,14 +66,15 @@ public final class StashedFileParameterValue extends AbstractFileParameterValue
 
     StashedFileParameterValue(String name, InputStream src) throws IOException {
         super(name);
-        tmp = new File(Util.createTempDir(), name);
-        tmp.deleteOnExit();
+        File tmp = new File(Util.createTempDir(), name);
         FileUtils.copyInputStreamToFile(src, tmp);
+        tmpFile = tmp.getAbsolutePath();
     }
 
     @Override public void buildEnvironment(Run<?, ?> build, EnvVars env) {
         super.buildEnvironment(build, env);
-        if (tmp != null) {
+        File tmp = tmpFile != null ? new File(tmpFile) : null;
+        if (tmp != null && tmp.isFile()) {
             try {
                 FlowExecutionOwner feo = build instanceof FlowExecutionOwner.Executable ? ((FlowExecutionOwner.Executable) build).asFlowExecutionOwner() : null;
                 TaskListener listener = feo != null ? feo.getListener() : TaskListener.NULL;
@@ -75,8 +85,7 @@ public final class StashedFileParameterValue extends AbstractFileParameterValue
                 throw new RuntimeException( x );
             }
             try {
-                Files.deleteIfExists(tmp.toPath());
-                tmp = null;
+                FileUtils.deleteDirectory(tmp.getParentFile());
             } catch (IOException e) {
                 throw new RuntimeException(e);
             }
@@ -88,4 +97,30 @@ public final class StashedFileParameterValue extends AbstractFileParameterValue
         return tempDir.child(name);
     }
 
+    @Extension
+    public static class CancelledQueueListener extends QueueListener {
+
+        @Override
+        public void onLeft(Queue.LeftItem li) {
+            if (li.isCancelled()) {
+                List<ParametersAction> actions = li.getActions(ParametersAction.class);
+                actions.forEach(a -> {
+                    a.getAllParameters().stream()
+                            .filter(p -> p instanceof StashedFileParameterValue)
+                            .map(p -> (StashedFileParameterValue) p)
+                            .forEach(p -> {
+                                if (p.tmpFile != null) {
+                                    File tmp = new File(p.tmpFile);
+                                    try {
+                                        FileUtils.deleteDirectory(tmp.getParentFile());
+                                    } catch (IOException | IllegalArgumentException e) {
+                                        LOGGER.log(Level.WARNING, "Unable to delete temporary file {0} for parameter {1} of task {2}",
+                                                new Object[]{tmp.getAbsolutePath(), p.getName(), li.task.getName()});
+                                    }
+                                }
+                            });
+                });
+            }
+        }
+    }
 }

src/test/java/io/jenkins/plugins/file_parameters/RestartTest.java

@@ -28,6 +28,7 @@
 import hudson.model.Node;
 import hudson.model.ParametersDefinitionProperty;
 import hudson.model.Queue;
+import hudson.model.Result;
 import hudson.model.queue.CauseOfBlockage;
 import hudson.model.queue.QueueTaskDispatcher;
 import jenkins.model.Jenkins;
@@ -92,7 +93,32 @@ void restBase64() throws Throwable {
         });
     }
 
-    @TestExtension("restBase64")
+    @Test
+    void stashedFileIsRetained() throws Throwable {
+        rr.then(r -> {
+            r.jenkins.setSecurityRealm(r.createDummySecurityRealm());
+            r.jenkins.setAuthorizationStrategy(new MockAuthorizationStrategy().grant(Jenkins.ADMINISTER).everywhere().to("admin"));
+            WorkflowJob p = r.createProject(WorkflowJob.class, "p");
+            p.addProperty(new ParametersDefinitionProperty(new StashedFileParameterDefinition("FILE")));
+            p.setDefinition(new CpsFlowDefinition("node { unstash 'FILE' }", true));
+            WebRequest req = new WebRequest(new URL(r.getURL() + "job/p/buildWithParameters"), HttpMethod.POST);
+            File f = File.createTempFile("junit", null, tmp);
+            FileUtils.write(f, "uploaded content here", "UTF-8");
+            req.setEncodingType(FormEncodingType.MULTIPART);
+            req.setRequestParameters(Collections.singletonList(new KeyDataPair("FILE", f, "myfile.txt", "text/plain", "UTF-8")));
+            r.createWebClient().withBasicApiToken("admin").getPage(req);
+        });
+        rr.then(r -> {
+            ExtensionList.lookupSingleton(Block.class).ready = true;
+            WorkflowJob p = r.jenkins.getItemByFullName("p", WorkflowJob.class);
+            r.waitUntilNoActivity();
+            WorkflowRun b = p.getBuildByNumber(1);
+            assertNotNull(b);
+            assert(b.getResult().isBetterOrEqualTo(Result.SUCCESS));
+        });
+    }
+
+    @TestExtension({"restBase64", "stashedFileIsRetained"})
     public static final class Block extends QueueTaskDispatcher {
         private boolean ready;
         @Override