From 535f454a9acce1219ef1c5a50f04a61266e5a715 Mon Sep 17 00:00:00 2001
From: Simon Rettberg
Date: Wed, 9 Oct 2019 17:39:31 +0200
Subject: [DeleteDirectory] New task

---
 .../openslx/taskmanager/tasks/DeleteDirectory.java | 61 ++++++++++++++++++++++
 1 file changed, 61 insertions(+)
 create mode 100644 src/main/java/org/openslx/taskmanager/tasks/DeleteDirectory.java

(limited to 'src/main/java/org/openslx/taskmanager/tasks/DeleteDirectory.java')

diff --git a/src/main/java/org/openslx/taskmanager/tasks/DeleteDirectory.java b/src/main/java/org/openslx/taskmanager/tasks/DeleteDirectory.java
new file mode 100644
index 0000000..1a0de05
--- /dev/null
+++ b/src/main/java/org/openslx/taskmanager/tasks/DeleteDirectory.java
@@ -0,0 +1,61 @@
+package org.openslx.taskmanager.tasks;
+
+import java.io.File;
+import java.io.IOException;
+import java.util.Arrays;
+
+import org.apache.commons.io.FileUtils;
+import org.apache.commons.io.FilenameUtils;
+import org.openslx.satserver.util.Util;
+import org.openslx.taskmanager.api.AbstractTask;
+
+import com.google.gson.annotations.Expose;
+
+public class DeleteDirectory extends AbstractTask
+{
+
+	protected static final String[] ALLOWED_DIRS =
+	{ "/tmp/", "/srv/openslx/www/boot/" };
+
+	@Expose
+	private String path = null;
+	@Expose
+	private boolean recursive = false;
+
+	private Output status = new Output();
+
+	@Override
+	protected boolean initTask()
+	{
+		this.setStatusObject( status );
+		this.path = FilenameUtils.normalize( this.path );
+		if ( this.path == null || !Util.startsWith( this.path, ALLOWED_DIRS )
+				|| Arrays.asList( ALLOWED_DIRS ).contains( this.path ) ) {
+			status.error = "Directory not in allowed directory";
+			return false;
+		}
+		return true;
+	}
+
+	@Override
+	protected boolean execute()
+	{
+		try {
+			FileUtils.deleteDirectory( new File( this.path ) );
+		} catch ( Exception e1 ) {
+			status.error = e1.toString();
+			return false;
+		}
+		return true;
+	}
+
+	/**
+	 * Output - contains additional status data of this task
+	 */
+	@SuppressWarnings( "unused" )
+	private static class Output
+	{
+		protected String error = null;
+	}
+
+}
-- 
cgit v1.2.3-55-g7522