From e1acf37d77297b63d7d9b7c76b38d96573555834 Mon Sep 17 00:00:00 2001
From: Lionel Sambuc <lionel@minix3.org>
Date: Thu, 23 May 2013 18:02:58 +0200
Subject: [PATCH] usr.sbin/mkfs: Allow -b SIZE with proto file

Previously, if a proto file was given, the -b argument would be ignored.

Also added:
 - if the number of inodes is not given, then an estimation is used.
 - sanity checks where moved from one special case to the general case.

Change-Id: I2abfa52bf34206c9087c64b5bfc26af866eb47cb
---
 usr.sbin/mkfs.mfs/mkfs.c | 62 +++++++++++++++++++++++++---------------
 1 file changed, 39 insertions(+), 23 deletions(-)

diff --git a/usr.sbin/mkfs.mfs/mkfs.c b/usr.sbin/mkfs.mfs/mkfs.c
index ee58ef45f..b934ec43f 100644
--- a/usr.sbin/mkfs.mfs/mkfs.c
+++ b/usr.sbin/mkfs.mfs/mkfs.c
@@ -141,7 +141,7 @@ int
 main(int argc, char *argv[])
 {
   int nread, mode, usrid, grpid, ch, extra_space_percent;
-  block_t blocks, maxblocks;
+  block_t blocks, maxblocks, bblocks;
   ino_t inodes, root_inum;
   char *token[MAX_TOKENS], line[LINE_LEN], *sfx;
   struct fs_size fssize;
@@ -151,6 +151,7 @@ main(int argc, char *argv[])
   /* Process switches. */
   blocks = 0;
   inodes = 0;
+  bblocks = 0;
 #ifndef MFS_STATIC_BLOCK_SIZE
   block_size = 0;
 #endif
@@ -179,7 +180,7 @@ main(int argc, char *argv[])
 		(void)sfx;	/* shut up warnings about unused variable...*/
 #endif
 	    case 'b':
-		blocks = strtoul(optarg, (char **) NULL, 0);
+		blocks = bblocks = strtoul(optarg, (char **) NULL, 0);
 		break;
 	    case 'd':
 		dflag = 1;
@@ -254,7 +255,11 @@ main(int argc, char *argv[])
 
   /* Determine the size of the device if not specified as -b or proto. */
   maxblocks = sizeup(argv[optind]);
-  if (argc - optind == 1 && blocks == 0) {
+  if (bblocks != 0 && bblocks > maxblocks){
+	errx(4, "Given size -b %d exeeds device capacity(%d)\n", bblocks, maxblocks);
+  }
+
+  if (argc - optind == 1 && bblocks == 0) {
   	blocks = maxblocks;
   	/* blocks == 0 is checked later, but leads to a funny way of
   	 * reporting a 0-sized device (displays usage).
@@ -285,7 +290,16 @@ main(int argc, char *argv[])
 
 	/* Read the line with the block and inode counts. */
 	get_line(line, token);
-	blocks = strtol(token[0], (char **) NULL, 10);
+	if (bblocks == 0){
+		blocks = strtol(token[0], (char **) NULL, 10);
+	} else {
+		if(bblocks < strtol(token[0], (char **) NULL, 10)) {
+ 			errx(1, "%s: number of blocks given as parameter(%d)"
+				" is too small for given proto file(%d).", 
+				argv[optind], bblocks, 
+				strtol(token[0], (char **) NULL, 10));
+		};
+	}
 	inodes = strtol(token[1], (char **) NULL, 10);
 
 	/* Process mode line for root directory. */
@@ -303,7 +317,7 @@ main(int argc, char *argv[])
 /* XXX is it OK to write on stdout? Use warn() instead? Also consider using verbose */
 		printf("dynamically sized filesystem: %u blocks, %u inodes\n",
 		    (unsigned int) blocks, (unsigned int) inodes);
-	}		
+	}
   } else {
 	lct = 0;
 	if (optind < argc) {
@@ -311,24 +325,6 @@ main(int argc, char *argv[])
 		blocks = strtoul(optarg, (char **) NULL, 0);
 		if (blocks == 0) errx(2, "Can't open prototype file");
 	}
-	if (inodes == 0) {
-		long long kb = ((unsigned long long)blocks*block_size) / 1024;
-
-		inodes = kb / 2;
-		if (kb >= 100000) inodes = kb / 4;
-		if (kb >= 1000000) inodes = kb / 6;
-		if (kb >= 10000000) inodes = kb / 8;
-		if (kb >= 100000000) inodes = kb / 10;
-		if (kb >= 1000000000) inodes = kb / 12;
-/* XXX check overflow: with very large number of blocks, this results in insanely large number of inodes */
-/* XXX check underflow (if/when ino_t is signed), else the message below will look strange */
-
-		/* round up to fill inode block */
-		inodes += inodes_per_block - 1;
-		inodes = inodes / inodes_per_block * inodes_per_block;
-	}
-	if (blocks < 5) errx(1, "Block count too small");
-	if (inodes < 1) errx(1, "Inode count too small");
 
 	/* Make simple file system of the given size, using defaults. */
 	mode = 040777;
@@ -337,6 +333,26 @@ main(int argc, char *argv[])
 	simple = 1;
   }
 
+  if (inodes == 0) {
+	long long kb = ((unsigned long long)blocks*block_size) / 1024;
+
+	inodes = kb / 2;
+	if (kb >= 100000) inodes = kb / 4;
+	if (kb >= 1000000) inodes = kb / 6;
+	if (kb >= 10000000) inodes = kb / 8;
+	if (kb >= 100000000) inodes = kb / 10;
+	if (kb >= 1000000000) inodes = kb / 12;
+/* XXX check overflow: with very large number of blocks, this results in insanely large number of inodes */
+/* XXX check underflow (if/when ino_t is signed), else the message below will look strange */
+
+	/* round up to fill inode block */
+	inodes += inodes_per_block - 1;
+	inodes = inodes / inodes_per_block * inodes_per_block;
+  }
+
+  if (blocks < 5) errx(1, "Block count too small");
+  if (inodes < 1) errx(1, "Inode count too small");
+
   nrblocks = blocks;
   nrinodes = inodes;