inspect: Refuse to download software hive if it is huge.

[libguestfs.git] / daemon / mkfs.c

diff --git a/daemon/mkfs.c b/daemon/mkfs.c
index 73fd860..ad3014c 100644 (file)
--- a/daemon/mkfs.c
+++ b/daemon/mkfs.c
@@ -29,15 +29,61 @@
 #include "daemon.h"
 #include "actions.h"
 
-int
-do_mkfs (const char *fstype, const char *device)
+#define MAX_ARGS 16
+
+static int
+mkfs (const char *fstype, const char *device,
+      const char **extra, size_t nr_extra)
 {
-  char *err;
+  const char *argv[MAX_ARGS];
+  size_t i = 0, j;
   int r;
+  char *err;
+
+  argv[i++] = "mkfs";
+  argv[i++] = "-t";
+  argv[i++] = fstype;
+
+  /* mkfs.ntfs requires the -Q argument otherwise it writes zeroes
+   * to every block and does bad block detection, neither of which
+   * are useful behaviour for virtual devices.
+   */
+  if (STREQ (fstype, "ntfs"))
+    argv[i++] = "-Q";
+
+  /* mkfs.reiserfs produces annoying interactive prompts unless you
+   * tell it to be quiet.
+   */
+  if (STREQ (fstype, "reiserfs"))
+    argv[i++] = "-f";
 
-  r = command (NULL, &err, "/sbin/mkfs", "-t", fstype, device, NULL);
+  /* Same for JFS. */
+  if (STREQ (fstype, "jfs"))
+    argv[i++] = "-f";
+
+  /* For GFS, GFS2, assume a single node. */
+  if (STREQ (fstype, "gfs") || STREQ (fstype, "gfs2")) {
+    argv[i++] = "-p";
+    argv[i++] = "lock_nolock";
+    /* The man page says this is default, but it doesn't seem to be: */
+    argv[i++] = "-j";
+    argv[i++] = "1";
+    /* Don't ask questions: */
+    argv[i++] = "-O";
+  }
+
+  for (j = 0; j < nr_extra; ++j)
+    argv[i++] = extra[j];
+
+  argv[i++] = device;
+  argv[i++] = NULL;
+
+  if (i > MAX_ARGS)
+    abort ();
+
+  r = commandv (NULL, &err, argv);
   if (r == -1) {
-    reply_with_error ("mkfs: %s", err);
+    reply_with_error ("%s: %s: %s", fstype, device, err);
     free (err);
     return -1;
   }
@@ -47,22 +93,55 @@ do_mkfs (const char *fstype, const char *device)
 }
 
 int
-do_mkfs_b (const char *fstype, int blocksize, const char *device)
+do_mkfs (const char *fstype, const char *device)
 {
-  char *err;
-  int r;
+  return mkfs (fstype, device, NULL, 0);
+}
 
-  char blocksize_s[32];
-  snprintf (blocksize_s, sizeof blocksize_s, "%d", blocksize);
+int
+do_mkfs_b (const char *fstype, int blocksize, const char *device)
+{
+  const char *extra[2];
+  char n[32];
 
-  r = command (NULL, &err,
-              "/sbin/mkfs", "-t", fstype, "-b", blocksize_s, device, NULL);
-  if (r == -1) {
-    reply_with_error ("mkfs_b: %s", err);
-    free (err);
+  if (blocksize <= 0 || !is_power_of_2 (blocksize)) {
+    reply_with_error ("block size must be > 0 and a power of 2");
     return -1;
   }
 
-  free (err);
-  return 0;
+  if (STREQ (fstype, "vfat") ||
+      STREQ (fstype, "msdos")) {
+    /* For VFAT map the blocksize into a cluster size.  However we
+     * have to determine the block device sector size in order to do
+     * this.
+     */
+    int sectorsize = do_blockdev_getss (device);
+    if (sectorsize == -1)
+      return -1;
+
+    int sectors_per_cluster = blocksize / sectorsize;
+    if (sectors_per_cluster < 1 || sectors_per_cluster > 128) {
+      reply_with_error ("unsupported cluster size for %s filesystem (requested cluster size = %d, sector size = %d, trying sectors per cluster = %d)",
+                        fstype, blocksize, sectorsize, sectors_per_cluster);
+      return -1;
+    }
+
+    snprintf (n, sizeof n, "%d", sectors_per_cluster);
+    extra[0] = "-s";
+    extra[1] = n;
+  }
+  else if (STREQ (fstype, "ntfs")) {
+    /* For NTFS map the blocksize into a cluster size. */
+    snprintf (n, sizeof n, "%d", blocksize);
+    extra[0] = "-c";
+    extra[1] = n;
+  }
+  else {
+    /* For all other filesystem types, try the -b option. */
+    snprintf (n, sizeof n, "%d", blocksize);
+    extra[0] = "-b";
+    extra[1] = n;
+  }
+
+  return mkfs (fstype, device, extra, 2);
 }