X-Git-Url: https://git.ucc.asn.au/?a=blobdiff_plain;f=Tools%2FDiskTool%2Fsrc%2Fmain.c;h=0758fc2999edc29318415292b346dc277040d9b1;hb=48025ef1ac8dd0445bc6a07fbc6c1ac9cb07991e;hp=fbfe7a0b08dbc0713f384336bacb9c1f934d21d1;hpb=d212a6f1517af8a9c1d550f566a202418f04fffe;p=tpg%2Facess2.git

diff --git a/Tools/DiskTool/src/main.c b/Tools/DiskTool/src/main.c
index fbfe7a0b..0758fc29 100644
--- a/Tools/DiskTool/src/main.c
+++ b/Tools/DiskTool/src/main.c
@@ -13,21 +13,67 @@ 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 ) {
+		if( strcmp("mount", 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");
+				fprintf(stderr, "mount takes 2 arguments (image and mountpoint)\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]);
+			if( DiskTool_MountImage(argv[i+2], argv[i+1]) ) {
+				fprintf(stderr, "Unable to mount '%s' as '%s'\n", argv[i+1], argv[i+2]);
+				break;
+			}
+
+			i += 2;
+			continue ;
+		}
+		
+		if( strcmp("mountlvm", argv[i]) == 0 ) {
+			
+			if( argc - i < 3 ) {
+				fprintf(stderr, "mountlvm takes 2 arguments (iamge and ident)\n");
 				exit(-1);
 			}
 
+			if( DiskTool_RegisterLVM(argv[i+2], argv[i+1]) ) {
+				fprintf(stderr, "Unable to register '%s' as LVM '%s'\n", argv[i+1], argv[i+2]);
+				break;
+			}
+			
 			i += 2;
+			continue ;
 		}
+		
+		if( strcmp("ls", argv[i]) == 0 ) {
+			if( argc - i < 2 ) {
+				fprintf(stderr, "ls 1 argument (path)\n");
+				break;
+			}
+
+			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");
+				break;
+			}
+
+			DiskTool_Copy(argv[i+1], argv[i+2]);			
+
+			i += 2;
+			continue ;
+		}
+	
+		fprintf(stderr, "Unknown command '%s'\n", argv[i]);
 	}
+	
+	DiskTool_Cleanup();
+	
 	return 0;
 }
 
@@ -68,13 +114,10 @@ int strucmp(const char *s1, const char *s2)
 	return strcasecmp(s1, s2);
 }
 
-int64_t DivUp(int64_t value, int64_t divisor)
+uint64_t DivMod64U(uint64_t value, uint64_t divisor, uint64_t *remainder)
 {
-	return (value + divisor - 1) / divisor;
-}
-
-int64_t timestamp(int sec, int min, int hr, int day, int month, int year)
-{
-	return 0;
+	if(remainder)
+		*remainder = value % divisor;
+	return value / divisor;
 }