diff options
author | Simon Rettberg | 2014-06-03 16:47:36 +0200 |
---|---|---|
committer | Simon Rettberg | 2014-06-03 16:47:36 +0200 |
commit | 32dc5354e2916387a2c62eadae0a4568023f1151 (patch) | |
tree | 7fd9a0173d6073e86d1d48e545646b1bc8c1a5eb /src/main/java/org/openslx/taskmanager/tasks/ListArchive.java | |
download | tmlite-bwlp-32dc5354e2916387a2c62eadae0a4568023f1151.tar.gz tmlite-bwlp-32dc5354e2916387a2c62eadae0a4568023f1151.tar.xz tmlite-bwlp-32dc5354e2916387a2c62eadae0a4568023f1151.zip |
Initial commit
Diffstat (limited to 'src/main/java/org/openslx/taskmanager/tasks/ListArchive.java')
-rw-r--r-- | src/main/java/org/openslx/taskmanager/tasks/ListArchive.java | 116 |
1 files changed, 116 insertions, 0 deletions
diff --git a/src/main/java/org/openslx/taskmanager/tasks/ListArchive.java b/src/main/java/org/openslx/taskmanager/tasks/ListArchive.java new file mode 100644 index 0000000..33bb83a --- /dev/null +++ b/src/main/java/org/openslx/taskmanager/tasks/ListArchive.java @@ -0,0 +1,116 @@ +package org.openslx.taskmanager.tasks; + +import java.io.FileNotFoundException; +import java.io.IOException; +import java.util.ArrayList; +import java.util.List; + +import org.apache.commons.compress.archivers.ArchiveEntry; +import org.apache.commons.compress.archivers.ArchiveException; +import org.apache.commons.compress.archivers.ArchiveInputStream; +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 ListArchive extends AbstractTask +{ + + @Expose + private String file; + + /* + * Own vars/constants not being deserialized + */ + + protected static final String[] ALLOWED_DIRS = + { "/tmp/", "/opt/openslx/" }; + + private Output status = new Output(); + + /* + * Code + */ + + @Override + protected boolean execute() + { + ArchiveInputStream archive; + try { + archive = Archive.getArchiveInputStream( this.file ); + } catch ( FileNotFoundException e1 ) { + status.error = e1.getMessage(); + status.errorCode = Output.ErrorCode.NOT_FOUND; + return false; + } catch ( IllegalArgumentException e1 ) { + status.error = e1.getMessage(); + status.errorCode = Output.ErrorCode.UNKNOWN_ERROR; + return false; + } catch ( ArchiveException e1 ) { + status.error = e1.getMessage(); + status.errorCode = Output.ErrorCode.UNKNOWN_FORMAT; + return false; + } + + status.entries = new ArrayList<>(); + + ArchiveEntry archiveEntry; + try { + // Iterate over every entry + while ( ( archiveEntry = archive.getNextEntry() ) != null ) { + if ( !archive.canReadEntryData( archiveEntry ) ) + continue; + Output.Entry entry = new Output.Entry(); + entry.name = archiveEntry.getName(); + entry.isdir = archiveEntry.isDirectory(); + entry.size = archiveEntry.getSize(); + status.entries.add( entry ); + } + } catch ( IOException e ) { + return false; + } + return true; + } + + @Override + protected boolean initTask() + { + this.setStatusObject( this.status ); + if ( this.file == null || this.file.length() == 0 || this.file.charAt( 0 ) != '/' ) { + status.error = "Need absolute path."; + status.errorCode = Output.ErrorCode.INVALID_DIRECTORY; + return false; + } + this.file = FilenameUtils.normalize( this.file ); + if ( this.file == null || !Util.startsWith( this.file, ALLOWED_DIRS ) ) { + status.error = "File not in allowed directory"; + status.errorCode = Output.ErrorCode.INVALID_DIRECTORY; + return false; + } + return true; + } + + @SuppressWarnings( "unused" ) + private static class Output + { + protected String error = null; + + protected enum ErrorCode + { + NOT_FOUND, UNKNOWN_FORMAT, UNKNOWN_ERROR, INVALID_DIRECTORY + }; + + protected ErrorCode errorCode = null; + protected List<Entry> entries = null; + + public static class Entry + { + protected String name = null; + protected boolean isdir = false; + protected long size = -1; + } + } + +} |