X-Git-Url: https://git.ucc.asn.au/?a=blobdiff_plain;f=Tools%2FDiskTool%2Fsrc%2Fmain.c;h=ff6f0ae1ddeb7c881fbea86d6359bef9a65a8b5a;hb=97159caf60a26cff3cc8f52e050a44d2492430f8;hp=bb9e9f246bd5c624abfccd6153c3f6e021ab39f0;hpb=0a3c1408fe7dfd6c50d104eb3ecafc534719d4fe;p=tpg%2Facess2.git

diff --git a/Tools/DiskTool/src/main.c b/Tools/DiskTool/src/main.c
index bb9e9f24..ff6f0ae1 100644
--- a/Tools/DiskTool/src/main.c
+++ b/Tools/DiskTool/src/main.c
@@ -5,11 +5,54 @@
 #include <errno.h>
 #include <stdint.h>
 #include <string.h>
+#include <disktool_common.h>
 
 // === CODE ===
 int main(int argc, char *argv[])
 {
-	
+	// Parse arguments
+	for( int i = 1; i < argc; i ++ )
+	{
+		if( strcmp("--image", argv[i]) == 0 || strcmp("-i", argv[i]) == 0 ) {
+			// Mount an image
+			if( argc - i < 3 ) {
+				fprintf(stderr, "--image/-i takes 2 arguments (ident and path)\n");
+				exit(-1);
+			}
+
+			if( DiskTool_MountImage(argv[i+1], argv[i+2]) ) {
+				fprintf(stderr, "Unable to mount '%s' as '%s'\n", argv[i+2], argv[i+1]);
+				exit(-1);
+			}
+
+			i += 2;
+			continue ;
+		}
+		
+		if( strcmp("ls", argv[i]) == 0 ) {
+			if( argc - i < 2 ) {
+				fprintf(stderr, "ls 1 argument (path)\n");
+				exit(-1);
+			}
+
+			DiskTool_ListDirectory(argv[i+1]);
+			i += 1;
+			continue ;
+		}
+		
+		if( strcmp("cp", argv[i]) == 0 ) {
+			
+			if( argc - i < 3 ) {
+				fprintf(stderr, "cp takes 2 arguments (source and destination)\n");
+				exit(-1);
+			}
+
+			DiskTool_Copy(argv[i+1], argv[i+2]);			
+
+			i += 2;
+			continue ;
+		}
+	}
 	return 0;
 }
 
@@ -44,3 +87,16 @@ int strpos(const char *Str, char Ch)
 	if(!r)	return -1;
 	return r - Str;
 }
+
+int strucmp(const char *s1, const char *s2)
+{
+	return strcasecmp(s1, s2);
+}
+
+uint64_t DivMod64U(uint64_t value, uint64_t divisor, uint64_t *remainder)
+{
+	if(remainder)
+		*remainder = value % divisor;
+	return value / divisor;
+}
+