[MakeTarball] Task for creating tar from text/blob data

1 files changed, 83 insertions, 0 deletions

diff --git a/src/main/java/org/openslx/taskmanager/tasks/MakeTarball.java b/src/main/java/org/openslx/taskmanager/tasks/MakeTarball.java
new file mode 100644
index 0000000..c8d8bc1
--- /dev/null
+++ b/src/main/java/org/openslx/taskmanager/tasks/MakeTarball.java
@@ -0,0 +1,83 @@
+package org.openslx.taskmanager.tasks;
+
+import java.io.IOException;
+import java.util.Map;
+import java.util.Map.Entry;
+
+import org.apache.commons.compress.archivers.tar.TarArchiveOutputStream;
+import org.apache.commons.io.FilenameUtils;
+import org.openslx.satserver.util.Archive;
+import org.openslx.satserver.util.Util;
+import org.openslx.taskmanager.api.AbstractTask;
+
+import com.google.gson.annotations.Expose;
+
+public class MakeTarball extends AbstractTask
+{
+
+	@Expose
+	private Map<String, byte[]> files;
+
+	@Expose
+	private String destination;
+
+	private Status status = new Status();
+
+	protected static final String[] ALLOWED_DIRS = { "/tmp/", "/opt/openslx/configs/" };
+
+	@Override
+	protected boolean initTask()
+	{
+		this.setStatusObject( this.status );
+		destination = FilenameUtils.normalize( destination );
+		if ( !Util.startsWith( destination, ALLOWED_DIRS ) ) {
+			status.error = "Illegal target directory " + destination;
+			return false;
+		}
+		if ( files.isEmpty() ) {
+			status.error = "Empty list of files";
+			return false;
+		}
+		for ( Entry<String, byte[]> f : files.entrySet() ) {
+			String fn = f.getKey();
+			if ( fn == null || fn.isEmpty() ) {
+				status.error = "File list contains empty file name";
+				return false;
+			}
+			if ( f.getValue() == null ) {
+				status.error = "File list contains NULL file payload for " + fn;
+				return false;
+			}
+		}
+		return true;
+	}
+
+	@Override
+	protected boolean execute()
+	{
+		TarArchiveOutputStream outArchive = null;
+		try {
+			try {
+				outArchive = Archive.createTarArchive( this.destination );
+			} catch ( IOException e ) {
+				status.error = "Could not create archive at " + this.destination;
+				return false;
+			}
+			for ( Entry<String, byte[]> f : files.entrySet() ) {
+				if ( !Archive.tarCreateFileFromString( outArchive, f.getKey(), f.getValue(), 0644 ) ) {
+					status.error = "Cannot add " + f.getKey() + " to " + this.destination;
+					return false;
+				}
+			}
+		} finally {
+			Util.multiClose( outArchive );
+		}
+		return true;
+	}
+
+	static class Status
+	{
+		private String error;
+	}
+
+}