Diff for /qemu/qemu-img.c between versions 1.1.1.10 and 1.1.1.13

version 1.1.1.10, 2018/04/24 18:16:38 version 1.1.1.13, 2018/04/24 18:56:18
Line 23 Line 23
  */   */
 #include "qemu-common.h"  #include "qemu-common.h"
 #include "qemu-option.h"  #include "qemu-option.h"
   #include "qemu-error.h"
 #include "osdep.h"  #include "osdep.h"
   #include "sysemu.h"
 #include "block_int.h"  #include "block_int.h"
 #include <stdio.h>  #include <stdio.h>
   
Line 37  typedef struct img_cmd_t { Line 39  typedef struct img_cmd_t {
 } img_cmd_t;  } img_cmd_t;
   
 /* Default to cache=writeback as data integrity is not important for qemu-tcg. */  /* Default to cache=writeback as data integrity is not important for qemu-tcg. */
 #define BRDV_O_FLAGS BDRV_O_CACHE_WB  #define BDRV_O_FLAGS BDRV_O_CACHE_WB
   #define BDRV_DEFAULT_CACHE "writeback"
 static void QEMU_NORETURN error(const char *fmt, ...)  
 {  
     va_list ap;  
     va_start(ap, fmt);  
     fprintf(stderr, "qemu-img: ");  
     vfprintf(stderr, fmt, ap);  
     fprintf(stderr, "\n");  
     exit(1);  
     va_end(ap);  
 }  
   
 static void format_print(void *opaque, const char *name)  static void format_print(void *opaque, const char *name)
 {  {
Line 58  static void format_print(void *opaque, c Line 50  static void format_print(void *opaque, c
 /* Please keep in synch with qemu-img.texi */  /* Please keep in synch with qemu-img.texi */
 static void help(void)  static void help(void)
 {  {
     printf("qemu-img version " QEMU_VERSION ", Copyright (c) 2004-2008 Fabrice Bellard\n"      const char *help_msg =
              "qemu-img version " QEMU_VERSION ", Copyright (c) 2004-2008 Fabrice Bellard\n"
            "usage: qemu-img command [command options]\n"             "usage: qemu-img command [command options]\n"
            "QEMU disk image utility\n"             "QEMU disk image utility\n"
            "\n"             "\n"
Line 72  static void help(void) Line 65  static void help(void)
            "Command parameters:\n"             "Command parameters:\n"
            "  'filename' is a disk image filename\n"             "  'filename' is a disk image filename\n"
            "  'fmt' is the disk image format. It is guessed automatically in most cases\n"             "  'fmt' is the disk image format. It is guessed automatically in most cases\n"
              "  'cache' is the cache mode used to write the output disk image, the valid\n"
              "    options are: 'none', 'writeback' (default), 'writethrough' and 'unsafe'\n"
            "  'size' is the disk image size in bytes. Optional suffixes\n"             "  'size' is the disk image size in bytes. Optional suffixes\n"
            "    'k' or 'K' (kilobyte, 1024), 'M' (megabyte, 1024k), 'G' (gigabyte, 1024M)\n"             "    'k' or 'K' (kilobyte, 1024), 'M' (megabyte, 1024k), 'G' (gigabyte, 1024M)\n"
            "    and T (terabyte, 1024G) are supported. 'b' is ignored.\n"             "    and T (terabyte, 1024G) are supported. 'b' is ignored.\n"
Line 81  static void help(void) Line 76  static void help(void)
            "    name=value format. Use -o ? for an overview of the options supported by the\n"             "    name=value format. Use -o ? for an overview of the options supported by the\n"
            "    used format\n"             "    used format\n"
            "  '-c' indicates that target image must be compressed (qcow format only)\n"             "  '-c' indicates that target image must be compressed (qcow format only)\n"
              "  '-u' enables unsafe rebasing. It is assumed that old and new backing file\n"
              "       match exactly. The image doesn't need a working backing file before\n"
              "       rebasing in this case (useful for renaming the backing file)\n"
            "  '-h' with or without a command shows this help and lists the supported formats\n"             "  '-h' with or without a command shows this help and lists the supported formats\n"
              "  '-p' show progress of command (only certain commands)\n"
            "\n"             "\n"
            "Parameters to snapshot subcommand:\n"             "Parameters to snapshot subcommand:\n"
            "  'snapshot' is the name of the snapshot to create, apply or delete\n"             "  'snapshot' is the name of the snapshot to create, apply or delete\n"
            "  '-a' applies a snapshot (revert disk to saved state)\n"             "  '-a' applies a snapshot (revert disk to saved state)\n"
            "  '-c' creates a snapshot\n"             "  '-c' creates a snapshot\n"
            "  '-d' deletes a snapshot\n"             "  '-d' deletes a snapshot\n"
            "  '-l' lists all snapshots in the given image\n"             "  '-l' lists all snapshots in the given image\n";
            );  
     printf("\nSupported formats:");      printf("%s\nSupported formats:", help_msg);
     bdrv_iterate_format(format_print, NULL);      bdrv_iterate_format(format_print, NULL);
     printf("\n");      printf("\n");
     exit(1);      exit(1);
Line 184  static int read_password(char *buf, int  Line 183  static int read_password(char *buf, int 
 }  }
 #endif  #endif
   
   static int set_cache_flag(const char *mode, int *flags)
   {
       *flags &= ~BDRV_O_CACHE_MASK;
   
       if (!strcmp(mode, "none") || !strcmp(mode, "off")) {
           *flags |= BDRV_O_CACHE_WB;
           *flags |= BDRV_O_NOCACHE;
       } else if (!strcmp(mode, "writeback")) {
           *flags |= BDRV_O_CACHE_WB;
       } else if (!strcmp(mode, "unsafe")) {
           *flags |= BDRV_O_CACHE_WB;
           *flags |= BDRV_O_NO_FLUSH;
       } else if (!strcmp(mode, "writethrough")) {
           /* this is the default */
       } else {
           return -1;
       }
   
       return 0;
   }
   
   static int print_block_option_help(const char *filename, const char *fmt)
   {
       BlockDriver *drv, *proto_drv;
       QEMUOptionParameter *create_options = NULL;
   
       /* Find driver and parse its options */
       drv = bdrv_find_format(fmt);
       if (!drv) {
           error_report("Unknown file format '%s'", fmt);
           return 1;
       }
   
       proto_drv = bdrv_find_protocol(filename);
       if (!proto_drv) {
           error_report("Unknown protocol '%s'", filename);
           return 1;
       }
   
       create_options = append_option_parameters(create_options,
                                                 drv->create_options);
       create_options = append_option_parameters(create_options,
                                                 proto_drv->create_options);
       print_option_help(create_options);
       free_option_parameters(create_options);
       return 0;
   }
   
 static BlockDriverState *bdrv_new_open(const char *filename,  static BlockDriverState *bdrv_new_open(const char *filename,
                                        const char *fmt)                                         const char *fmt,
                                          int flags)
 {  {
     BlockDriverState *bs;      BlockDriverState *bs;
     BlockDriver *drv;      BlockDriver *drv;
     char password[256];      char password[256];
       int ret;
   
       bs = bdrv_new("image");
   
     bs = bdrv_new("");  
     if (!bs)  
         error("Not enough memory");  
     if (fmt) {      if (fmt) {
         drv = bdrv_find_format(fmt);          drv = bdrv_find_format(fmt);
         if (!drv)          if (!drv) {
             error("Unknown file format '%s'", fmt);              error_report("Unknown file format '%s'", fmt);
               goto fail;
           }
     } else {      } else {
         drv = NULL;          drv = NULL;
     }      }
     if (bdrv_open2(bs, filename, BRDV_O_FLAGS, drv) < 0) {  
         error("Could not open '%s'", filename);      ret = bdrv_open(bs, filename, flags, drv);
       if (ret < 0) {
           error_report("Could not open '%s': %s", filename, strerror(-ret));
           goto fail;
     }      }
   
     if (bdrv_is_encrypted(bs)) {      if (bdrv_is_encrypted(bs)) {
         printf("Disk image '%s' is encrypted.\n", filename);          printf("Disk image '%s' is encrypted.\n", filename);
         if (read_password(password, sizeof(password)) < 0)          if (read_password(password, sizeof(password)) < 0) {
             error("No password given");              error_report("No password given");
         if (bdrv_set_key(bs, password) < 0)              goto fail;
             error("invalid password");          }
           if (bdrv_set_key(bs, password) < 0) {
               error_report("invalid password");
               goto fail;
           }
     }      }
     return bs;      return bs;
   fail:
       if (bs) {
           bdrv_delete(bs);
       }
       return NULL;
 }  }
   
 static void add_old_style_options(const char *fmt, QEMUOptionParameter *list,  static int add_old_style_options(const char *fmt, QEMUOptionParameter *list,
     int flags, const char *base_filename, const char *base_fmt)                                   const char *base_filename,
                                    const char *base_fmt)
 {  {
     if (flags & BLOCK_FLAG_ENCRYPT) {  
         if (set_option_parameter(list, BLOCK_OPT_ENCRYPT, "on")) {  
             error("Encryption not supported for file format '%s'", fmt);  
         }  
     }  
     if (flags & BLOCK_FLAG_COMPAT6) {  
         if (set_option_parameter(list, BLOCK_OPT_COMPAT6, "on")) {  
             error("VMDK version 6 not supported for file format '%s'", fmt);  
         }  
     }  
   
     if (base_filename) {      if (base_filename) {
         if (set_option_parameter(list, BLOCK_OPT_BACKING_FILE, base_filename)) {          if (set_option_parameter(list, BLOCK_OPT_BACKING_FILE, base_filename)) {
             error("Backing file not supported for file format '%s'", fmt);              error_report("Backing file not supported for file format '%s'",
                            fmt);
               return -1;
         }          }
     }      }
     if (base_fmt) {      if (base_fmt) {
         if (set_option_parameter(list, BLOCK_OPT_BACKING_FMT, base_fmt)) {          if (set_option_parameter(list, BLOCK_OPT_BACKING_FMT, base_fmt)) {
             error("Backing file format not supported for file format '%s'", fmt);              error_report("Backing file format not supported for file "
                            "format '%s'", fmt);
               return -1;
         }          }
     }      }
       return 0;
 }  }
   
 static int img_create(int argc, char **argv)  static int img_create(int argc, char **argv)
 {  {
     int c, ret, flags;      int c, ret = 0;
       uint64_t img_size = -1;
     const char *fmt = "raw";      const char *fmt = "raw";
     const char *base_fmt = NULL;      const char *base_fmt = NULL;
     const char *filename;      const char *filename;
     const char *base_filename = NULL;      const char *base_filename = NULL;
     BlockDriver *drv;  
     QEMUOptionParameter *param = NULL;  
     char *options = NULL;      char *options = NULL;
   
     flags = 0;  
     for(;;) {      for(;;) {
         c = getopt(argc, argv, "F:b:f:he6o:");          c = getopt(argc, argv, "F:b:f:he6o:");
         if (c == -1)          if (c == -1) {
             break;              break;
           }
         switch(c) {          switch(c) {
           case '?':
         case 'h':          case 'h':
             help();              help();
             break;              break;
Line 270  static int img_create(int argc, char **a Line 328  static int img_create(int argc, char **a
             fmt = optarg;              fmt = optarg;
             break;              break;
         case 'e':          case 'e':
             flags |= BLOCK_FLAG_ENCRYPT;              error_report("option -e is deprecated, please use \'-o "
             break;                    "encryption\' instead!");
               return 1;
         case '6':          case '6':
             flags |= BLOCK_FLAG_COMPAT6;              error_report("option -6 is deprecated, please use \'-o "
             break;                    "compat6\' instead!");
               return 1;
         case 'o':          case 'o':
             options = optarg;              options = optarg;
             break;              break;
         }          }
     }      }
   
     /* Find driver and parse its options */  
     drv = bdrv_find_format(fmt);  
     if (!drv)  
         error("Unknown file format '%s'", fmt);  
   
     if (options && !strcmp(options, "?")) {  
         print_option_help(drv->create_options);  
         return 0;  
     }  
   
     /* Create parameter list with default values */  
     param = parse_option_parameters("", drv->create_options, param);  
     set_option_parameter_int(param, BLOCK_OPT_SIZE, -1);  
   
     /* Parse -o options */  
     if (options) {  
         param = parse_option_parameters(options, drv->create_options, param);  
         if (param == NULL) {  
             error("Invalid options for file format '%s'.", fmt);  
         }  
     }  
   
     /* Get the filename */      /* Get the filename */
     if (optind >= argc)      if (optind >= argc) {
         help();          help();
       }
     filename = argv[optind++];      filename = argv[optind++];
   
     /* Add size to parameters */      /* Get image size, if specified */
     if (optind < argc) {      if (optind < argc) {
         set_option_parameter(param, BLOCK_OPT_SIZE, argv[optind++]);          int64_t sval;
     }          sval = strtosz_suffix(argv[optind++], NULL, STRTOSZ_DEFSUFFIX_B);
           if (sval < 0) {
     /* Add old-style options to parameters */              error_report("Invalid image size specified! You may use k, M, G or "
     add_old_style_options(fmt, param, flags, base_filename, base_fmt);                    "T suffixes for ");
               error_report("kilobytes, megabytes, gigabytes and terabytes.");
     // The size for the image must always be specified, with one exception:              ret = -1;
     // If we are using a backing file, we can obtain the size from there              goto out;
     if (get_option_parameter(param, BLOCK_OPT_SIZE)->value.n == -1) {  
   
         QEMUOptionParameter *backing_file =  
             get_option_parameter(param, BLOCK_OPT_BACKING_FILE);  
         QEMUOptionParameter *backing_fmt =  
             get_option_parameter(param, BLOCK_OPT_BACKING_FMT);  
   
         if (backing_file && backing_file->value.s) {  
             BlockDriverState *bs;  
             uint64_t size;  
             const char *fmt = NULL;  
             char buf[32];  
   
             if (backing_fmt && backing_fmt->value.s) {  
                  if (bdrv_find_format(backing_fmt->value.s)) {  
                      fmt = backing_fmt->value.s;  
                 } else {  
                      error("Unknown backing file format '%s'",  
                         backing_fmt->value.s);  
                 }  
             }  
   
             bs = bdrv_new_open(backing_file->value.s, fmt);  
             bdrv_get_geometry(bs, &size);  
             size *= 512;  
             bdrv_delete(bs);  
   
             snprintf(buf, sizeof(buf), "%" PRId64, size);  
             set_option_parameter(param, BLOCK_OPT_SIZE, buf);  
         } else {  
             error("Image creation needs a size parameter");  
         }          }
           img_size = (uint64_t)sval;
     }      }
   
     printf("Formatting '%s', fmt=%s ", filename, fmt);      if (options && !strcmp(options, "?")) {
     print_option_parameters(param);          ret = print_block_option_help(filename, fmt);
     puts("");          goto out;
       }
     ret = bdrv_create(drv, filename, param);  
     free_option_parameters(param);  
   
     if (ret < 0) {      ret = bdrv_img_create(filename, fmt, base_filename, base_fmt,
         if (ret == -ENOTSUP) {                            options, img_size, BDRV_O_FLAGS);
             error("Formatting or formatting option not supported for file format '%s'", fmt);  out:
         } else if (ret == -EFBIG) {      if (ret) {
             error("The image size is too large for file format '%s'", fmt);          return 1;
         } else {  
             error("Error while formatting");  
         }  
     }      }
     return 0;      return 0;
 }  }
   
   /*
    * Checks an image for consistency. Exit codes:
    *
    * 0 - Check completed, image is good
    * 1 - Check not completed because of internal errors
    * 2 - Check completed, image is corrupted
    * 3 - Check completed, image has leaked clusters, but is good otherwise
    */
 static int img_check(int argc, char **argv)  static int img_check(int argc, char **argv)
 {  {
     int c, ret;      int c, ret;
     const char *filename, *fmt;      const char *filename, *fmt;
     BlockDriver *drv;  
     BlockDriverState *bs;      BlockDriverState *bs;
       BdrvCheckResult result;
   
     fmt = NULL;      fmt = NULL;
     for(;;) {      for(;;) {
         c = getopt(argc, argv, "f:h");          c = getopt(argc, argv, "f:h");
         if (c == -1)          if (c == -1) {
             break;              break;
           }
         switch(c) {          switch(c) {
           case '?':
         case 'h':          case 'h':
             help();              help();
             break;              break;
Line 392  static int img_check(int argc, char **ar Line 406  static int img_check(int argc, char **ar
             break;              break;
         }          }
     }      }
     if (optind >= argc)      if (optind >= argc) {
         help();          help();
       }
     filename = argv[optind++];      filename = argv[optind++];
   
     bs = bdrv_new("");      bs = bdrv_new_open(filename, fmt, BDRV_O_FLAGS);
     if (!bs)      if (!bs) {
         error("Not enough memory");          return 1;
     if (fmt) {  
         drv = bdrv_find_format(fmt);  
         if (!drv)  
             error("Unknown file format '%s'", fmt);  
     } else {  
         drv = NULL;  
     }      }
     if (bdrv_open2(bs, filename, BRDV_O_FLAGS, drv) < 0) {      ret = bdrv_check(bs, &result);
         error("Could not open '%s'", filename);  
       if (ret == -ENOTSUP) {
           error_report("This image format does not support checks");
           bdrv_delete(bs);
           return 1;
     }      }
     ret = bdrv_check(bs);  
     switch(ret) {      if (!(result.corruptions || result.leaks || result.check_errors)) {
     case 0:  
         printf("No errors were found on the image.\n");          printf("No errors were found on the image.\n");
         break;      } else {
     case -ENOTSUP:          if (result.corruptions) {
         error("This image format does not support checks");              printf("\n%d errors were found on the image.\n"
         break;                  "Data may be corrupted, or further writes to the image "
     default:                  "may corrupt it.\n",
         if (ret < 0) {                  result.corruptions);
             error("An error occurred during the check");          }
         } else {  
             printf("%d errors were found on the image.\n", ret);          if (result.leaks) {
               printf("\n%d leaked clusters were found on the image.\n"
                   "This means waste of disk space, but no harm to data.\n",
                   result.leaks);
           }
   
           if (result.check_errors) {
               printf("\n%d internal errors have occurred during the check.\n",
                   result.check_errors);
         }          }
         break;  
     }      }
   
     bdrv_delete(bs);      bdrv_delete(bs);
     return 0;  
       if (ret < 0 || result.check_errors) {
           printf("\nAn error has occurred during the check: %s\n"
               "The check is not complete and may have missed error.\n",
               strerror(-ret));
           return 1;
       }
   
       if (result.corruptions) {
           return 2;
       } else if (result.leaks) {
           return 3;
       } else {
           return 0;
       }
 }  }
   
 static int img_commit(int argc, char **argv)  static int img_commit(int argc, char **argv)
 {  {
     int c, ret;      int c, ret, flags;
     const char *filename, *fmt;      const char *filename, *fmt, *cache;
     BlockDriver *drv;  
     BlockDriverState *bs;      BlockDriverState *bs;
   
     fmt = NULL;      fmt = NULL;
       cache = BDRV_DEFAULT_CACHE;
     for(;;) {      for(;;) {
         c = getopt(argc, argv, "f:h");          c = getopt(argc, argv, "f:ht:");
         if (c == -1)          if (c == -1) {
             break;              break;
           }
         switch(c) {          switch(c) {
           case '?':
         case 'h':          case 'h':
             help();              help();
             break;              break;
         case 'f':          case 'f':
             fmt = optarg;              fmt = optarg;
             break;              break;
           case 't':
               cache = optarg;
               break;
         }          }
     }      }
     if (optind >= argc)      if (optind >= argc) {
         help();          help();
       }
     filename = argv[optind++];      filename = argv[optind++];
   
     bs = bdrv_new("");      flags = BDRV_O_RDWR;
     if (!bs)      ret = set_cache_flag(cache, &flags);
         error("Not enough memory");      if (ret < 0) {
     if (fmt) {          error_report("Invalid cache option: %s", cache);
         drv = bdrv_find_format(fmt);          return -1;
         if (!drv)  
             error("Unknown file format '%s'", fmt);  
     } else {  
         drv = NULL;  
     }      }
     if (bdrv_open2(bs, filename, BRDV_O_FLAGS, drv) < 0) {  
         error("Could not open '%s'", filename);      bs = bdrv_new_open(filename, fmt, flags);
       if (!bs) {
           return 1;
     }      }
     ret = bdrv_commit(bs);      ret = bdrv_commit(bs);
     switch(ret) {      switch(ret) {
Line 474  static int img_commit(int argc, char **a Line 511  static int img_commit(int argc, char **a
         printf("Image committed.\n");          printf("Image committed.\n");
         break;          break;
     case -ENOENT:      case -ENOENT:
         error("No disk inserted");          error_report("No disk inserted");
         break;          break;
     case -EACCES:      case -EACCES:
         error("Image is read-only");          error_report("Image is read-only");
         break;          break;
     case -ENOTSUP:      case -ENOTSUP:
         error("Image is already committed");          error_report("Image is already committed");
         break;          break;
     default:      default:
         error("Error while committing image");          error_report("Error while committing image");
         break;          break;
     }      }
   
     bdrv_delete(bs);      bdrv_delete(bs);
       if (ret) {
           return 1;
       }
     return 0;      return 0;
 }  }
   
   /*
    * Checks whether the sector is not a zero sector.
    *
    * Attention! The len must be a multiple of 4 * sizeof(long) due to
    * restriction of optimizations in this function.
    */
 static int is_not_zero(const uint8_t *sector, int len)  static int is_not_zero(const uint8_t *sector, int len)
 {  {
       /*
        * Use long as the biggest available internal data type that fits into the
        * CPU register and unroll the loop to smooth out the effect of memory
        * latency.
        */
   
     int i;      int i;
     len >>= 2;      long d0, d1, d2, d3;
     for(i = 0;i < len; i++) {      const long * const data = (const long *) sector;
         if (((uint32_t *)sector)[i] != 0)  
       len /= sizeof(long);
   
       for(i = 0; i < len; i += 4) {
           d0 = data[i + 0];
           d1 = data[i + 1];
           d2 = data[i + 2];
           d3 = data[i + 3];
   
           if (d0 || d1 || d2 || d3) {
             return 1;              return 1;
           }
     }      }
   
     return 0;      return 0;
 }  }
   
Line 527  static int is_allocated_sectors(const ui Line 590  static int is_allocated_sectors(const ui
     return v;      return v;
 }  }
   
   /*
    * Compares two buffers sector by sector. Returns 0 if the first sector of both
    * buffers matches, non-zero otherwise.
    *
    * pnum is set to the number of sectors (including and immediately following
    * the first one) that are known to have the same comparison result
    */
   static int compare_sectors(const uint8_t *buf1, const uint8_t *buf2, int n,
       int *pnum)
   {
       int res, i;
   
       if (n <= 0) {
           *pnum = 0;
           return 0;
       }
   
       res = !!memcmp(buf1, buf2, 512);
       for(i = 1; i < n; i++) {
           buf1 += 512;
           buf2 += 512;
   
           if (!!memcmp(buf1, buf2, 512) != res) {
               break;
           }
       }
   
       *pnum = i;
       return res;
   }
   
 #define IO_BUF_SIZE (2 * 1024 * 1024)  #define IO_BUF_SIZE (2 * 1024 * 1024)
   
 static int img_convert(int argc, char **argv)  static int img_convert(int argc, char **argv)
 {  {
     int c, ret, n, n1, bs_n, bs_i, flags, cluster_size, cluster_sectors;      int c, ret = 0, n, n1, bs_n, bs_i, compress, cluster_size, cluster_sectors;
     const char *fmt, *out_fmt, *out_baseimg, *out_filename;      int progress = 0, flags;
     BlockDriver *drv;      const char *fmt, *out_fmt, *cache, *out_baseimg, *out_filename;
     BlockDriverState **bs, *out_bs;      BlockDriver *drv, *proto_drv;
       BlockDriverState **bs = NULL, *out_bs = NULL;
     int64_t total_sectors, nb_sectors, sector_num, bs_offset;      int64_t total_sectors, nb_sectors, sector_num, bs_offset;
     uint64_t bs_sectors;      uint64_t bs_sectors;
     uint8_t * buf;      uint8_t * buf = NULL;
     const uint8_t *buf1;      const uint8_t *buf1;
     BlockDriverInfo bdi;      BlockDriverInfo bdi;
     QEMUOptionParameter *param = NULL;      QEMUOptionParameter *param = NULL, *create_options = NULL;
       QEMUOptionParameter *out_baseimg_param;
     char *options = NULL;      char *options = NULL;
       const char *snapshot_name = NULL;
       float local_progress;
   
     fmt = NULL;      fmt = NULL;
     out_fmt = "raw";      out_fmt = "raw";
       cache = "unsafe";
     out_baseimg = NULL;      out_baseimg = NULL;
     flags = 0;      compress = 0;
     for(;;) {      for(;;) {
         c = getopt(argc, argv, "f:O:B:hce6o:");          c = getopt(argc, argv, "f:O:B:s:hce6o:pt:");
         if (c == -1)          if (c == -1) {
             break;              break;
           }
         switch(c) {          switch(c) {
           case '?':
         case 'h':          case 'h':
             help();              help();
             break;              break;
Line 565  static int img_convert(int argc, char ** Line 666  static int img_convert(int argc, char **
             out_baseimg = optarg;              out_baseimg = optarg;
             break;              break;
         case 'c':          case 'c':
             flags |= BLOCK_FLAG_COMPRESS;              compress = 1;
             break;              break;
         case 'e':          case 'e':
             flags |= BLOCK_FLAG_ENCRYPT;              error_report("option -e is deprecated, please use \'-o "
             break;                    "encryption\' instead!");
               return 1;
         case '6':          case '6':
             flags |= BLOCK_FLAG_COMPAT6;              error_report("option -6 is deprecated, please use \'-o "
             break;                    "compat6\' instead!");
               return 1;
         case 'o':          case 'o':
             options = optarg;              options = optarg;
             break;              break;
           case 's':
               snapshot_name = optarg;
               break;
           case 'p':
               progress = 1;
               break;
           case 't':
               cache = optarg;
               break;
         }          }
     }      }
   
     bs_n = argc - optind - 1;      bs_n = argc - optind - 1;
     if (bs_n < 1) help();      if (bs_n < 1) {
           help();
       }
   
     out_filename = argv[argc - 1];      out_filename = argv[argc - 1];
   
     if (bs_n > 1 && out_baseimg)      if (options && !strcmp(options, "?")) {
         error("-B makes no sense when concatenating multiple input images");          ret = print_block_option_help(out_filename, out_fmt);
           goto out;
       }
   
       if (bs_n > 1 && out_baseimg) {
           error_report("-B makes no sense when concatenating multiple input "
                        "images");
           ret = -1;
           goto out;
       }
                   
     bs = calloc(bs_n, sizeof(BlockDriverState *));      qemu_progress_init(progress, 2.0);
     if (!bs)      qemu_progress_print(0, 100);
         error("Out of memory");  
       bs = qemu_mallocz(bs_n * sizeof(BlockDriverState *));
   
     total_sectors = 0;      total_sectors = 0;
     for (bs_i = 0; bs_i < bs_n; bs_i++) {      for (bs_i = 0; bs_i < bs_n; bs_i++) {
         bs[bs_i] = bdrv_new_open(argv[optind + bs_i], fmt);          bs[bs_i] = bdrv_new_open(argv[optind + bs_i], fmt, BDRV_O_FLAGS);
         if (!bs[bs_i])          if (!bs[bs_i]) {
             error("Could not open '%s'", argv[optind + bs_i]);              error_report("Could not open '%s'", argv[optind + bs_i]);
               ret = -1;
               goto out;
           }
         bdrv_get_geometry(bs[bs_i], &bs_sectors);          bdrv_get_geometry(bs[bs_i], &bs_sectors);
         total_sectors += bs_sectors;          total_sectors += bs_sectors;
     }      }
   
       if (snapshot_name != NULL) {
           if (bs_n > 1) {
               error_report("No support for concatenating multiple snapshot");
               ret = -1;
               goto out;
           }
           if (bdrv_snapshot_load_tmp(bs[0], snapshot_name) < 0) {
               error_report("Failed to load snapshot");
               ret = -1;
               goto out;
           }
       }
   
     /* Find driver and parse its options */      /* Find driver and parse its options */
     drv = bdrv_find_format(out_fmt);      drv = bdrv_find_format(out_fmt);
     if (!drv)      if (!drv) {
         error("Unknown file format '%s'", out_fmt);          error_report("Unknown file format '%s'", out_fmt);
           ret = -1;
     if (options && !strcmp(options, "?")) {          goto out;
         print_option_help(drv->create_options);      }
         free(bs);  
         return 0;      proto_drv = bdrv_find_protocol(out_filename);
     }      if (!proto_drv) {
           error_report("Unknown protocol '%s'", out_filename);
           ret = -1;
           goto out;
       }
   
       create_options = append_option_parameters(create_options,
                                                 drv->create_options);
       create_options = append_option_parameters(create_options,
                                                 proto_drv->create_options);
   
     if (options) {      if (options) {
         param = parse_option_parameters(options, drv->create_options, param);          param = parse_option_parameters(options, create_options, param);
         if (param == NULL) {          if (param == NULL) {
             error("Invalid options for file format '%s'.", out_fmt);              error_report("Invalid options for file format '%s'.", out_fmt);
               ret = -1;
               goto out;
         }          }
     } else {      } else {
         param = parse_option_parameters("", drv->create_options, param);          param = parse_option_parameters("", create_options, param);
     }      }
   
     set_option_parameter_int(param, BLOCK_OPT_SIZE, total_sectors * 512);      set_option_parameter_int(param, BLOCK_OPT_SIZE, total_sectors * 512);
     add_old_style_options(out_fmt, param, flags, out_baseimg, NULL);      ret = add_old_style_options(out_fmt, param, out_baseimg, NULL);
       if (ret < 0) {
           goto out;
       }
   
       /* Get backing file name if -o backing_file was used */
       out_baseimg_param = get_option_parameter(param, BLOCK_OPT_BACKING_FILE);
       if (out_baseimg_param) {
           out_baseimg = out_baseimg_param->value.s;
       }
   
     /* Check if compression is supported */      /* Check if compression is supported */
     if (flags & BLOCK_FLAG_COMPRESS) {      if (compress) {
         QEMUOptionParameter *encryption =          QEMUOptionParameter *encryption =
             get_option_parameter(param, BLOCK_OPT_ENCRYPT);              get_option_parameter(param, BLOCK_OPT_ENCRYPT);
   
         if (!drv->bdrv_write_compressed) {          if (!drv->bdrv_write_compressed) {
             error("Compression not supported for this file format");              error_report("Compression not supported for this file format");
               ret = -1;
               goto out;
         }          }
   
         if (encryption && encryption->value.n) {          if (encryption && encryption->value.n) {
             error("Compression and encryption not supported at the same time");              error_report("Compression and encryption not supported at "
                            "the same time");
               ret = -1;
               goto out;
         }          }
     }      }
   
     /* Create the new image */      /* Create the new image */
     ret = bdrv_create(drv, out_filename, param);      ret = bdrv_create(drv, out_filename, param);
     free_option_parameters(param);  
   
     if (ret < 0) {      if (ret < 0) {
         if (ret == -ENOTSUP) {          if (ret == -ENOTSUP) {
             error("Formatting not supported for file format '%s'", out_fmt);              error_report("Formatting not supported for file format '%s'",
                            out_fmt);
         } else if (ret == -EFBIG) {          } else if (ret == -EFBIG) {
             error("The image size is too large for file format '%s'", out_fmt);              error_report("The image size is too large for file format '%s'",
                            out_fmt);
         } else {          } else {
             error("Error while formatting '%s'", out_filename);              error_report("%s: error while converting %s: %s",
                            out_filename, out_fmt, strerror(-ret));
         }          }
           goto out;
     }      }
   
     out_bs = bdrv_new_open(out_filename, out_fmt);      flags = BDRV_O_RDWR;
       ret = set_cache_flag(cache, &flags);
       if (ret < 0) {
           error_report("Invalid cache option: %s", cache);
           return -1;
       }
   
       out_bs = bdrv_new_open(out_filename, out_fmt, flags);
       if (!out_bs) {
           ret = -1;
           goto out;
       }
   
     bs_i = 0;      bs_i = 0;
     bs_offset = 0;      bs_offset = 0;
     bdrv_get_geometry(bs[0], &bs_sectors);      bdrv_get_geometry(bs[0], &bs_sectors);
     buf = qemu_malloc(IO_BUF_SIZE);      buf = qemu_malloc(IO_BUF_SIZE);
   
     if (flags & BLOCK_FLAG_COMPRESS) {      if (compress) {
         if (bdrv_get_info(out_bs, &bdi) < 0)          ret = bdrv_get_info(out_bs, &bdi);
             error("could not get block driver info");          if (ret < 0) {
               error_report("could not get block driver info");
               goto out;
           }
         cluster_size = bdi.cluster_size;          cluster_size = bdi.cluster_size;
         if (cluster_size <= 0 || cluster_size > IO_BUF_SIZE)          if (cluster_size <= 0 || cluster_size > IO_BUF_SIZE) {
             error("invalid cluster size");              error_report("invalid cluster size");
               ret = -1;
               goto out;
           }
         cluster_sectors = cluster_size >> 9;          cluster_sectors = cluster_size >> 9;
         sector_num = 0;          sector_num = 0;
   
           nb_sectors = total_sectors;
           local_progress = (float)100 /
               (nb_sectors / MIN(nb_sectors, cluster_sectors));
   
         for(;;) {          for(;;) {
             int64_t bs_num;              int64_t bs_num;
             int remainder;              int remainder;
Line 691  static int img_convert(int argc, char ** Line 880  static int img_convert(int argc, char **
                     bs_offset += bs_sectors;                      bs_offset += bs_sectors;
                     bdrv_get_geometry(bs[bs_i], &bs_sectors);                      bdrv_get_geometry(bs[bs_i], &bs_sectors);
                     bs_num = 0;                      bs_num = 0;
                     /* printf("changing part: sector_num=%lld, "                      /* printf("changing part: sector_num=%" PRId64 ", "
                        "bs_i=%d, bs_offset=%lld, bs_sectors=%lld\n",                         "bs_i=%d, bs_offset=%" PRId64 ", bs_sectors=%" PRId64
                        sector_num, bs_i, bs_offset, bs_sectors); */                         "\n", sector_num, bs_i, bs_offset, bs_sectors); */
                 }                  }
                 assert (bs_num < bs_sectors);                  assert (bs_num < bs_sectors);
   
                 nlow = (remainder > bs_sectors - bs_num) ? bs_sectors - bs_num : remainder;                  nlow = (remainder > bs_sectors - bs_num) ? bs_sectors - bs_num : remainder;
   
                 if (bdrv_read(bs[bs_i], bs_num, buf2, nlow) < 0)                   ret = bdrv_read(bs[bs_i], bs_num, buf2, nlow);
                     error("error while reading");                  if (ret < 0) {
                       error_report("error while reading");
                       goto out;
                   }
   
                 buf2 += nlow * 512;                  buf2 += nlow * 512;
                 bs_num += nlow;                  bs_num += nlow;
Line 709  static int img_convert(int argc, char ** Line 901  static int img_convert(int argc, char **
             }              }
             assert (remainder == 0);              assert (remainder == 0);
   
             if (n < cluster_sectors)              if (n < cluster_sectors) {
                 memset(buf + n * 512, 0, cluster_size - n * 512);                  memset(buf + n * 512, 0, cluster_size - n * 512);
               }
             if (is_not_zero(buf, cluster_size)) {              if (is_not_zero(buf, cluster_size)) {
                 if (bdrv_write_compressed(out_bs, sector_num, buf,                  ret = bdrv_write_compressed(out_bs, sector_num, buf,
                                           cluster_sectors) != 0)                                              cluster_sectors);
                     error("error while compressing sector %" PRId64,                  if (ret != 0) {
                       error_report("error while compressing sector %" PRId64,
                           sector_num);                            sector_num);
                       goto out;
                   }
             }              }
             sector_num += n;              sector_num += n;
               qemu_progress_print(local_progress, 100);
         }          }
         /* signal EOF to align */          /* signal EOF to align */
         bdrv_write_compressed(out_bs, 0, NULL, 0);          bdrv_write_compressed(out_bs, 0, NULL, 0);
     } else {      } else {
           int has_zero_init = bdrv_has_zero_init(out_bs);
   
         sector_num = 0; // total number of sectors converted so far          sector_num = 0; // total number of sectors converted so far
           nb_sectors = total_sectors - sector_num;
           local_progress = (float)100 /
               (nb_sectors / MIN(nb_sectors, IO_BUF_SIZE / 512));
   
         for(;;) {          for(;;) {
             nb_sectors = total_sectors - sector_num;              nb_sectors = total_sectors - sector_num;
             if (nb_sectors <= 0)              if (nb_sectors <= 0) {
                 break;                  break;
             if (nb_sectors >= (IO_BUF_SIZE / 512))              }
               if (nb_sectors >= (IO_BUF_SIZE / 512)) {
                 n = (IO_BUF_SIZE / 512);                  n = (IO_BUF_SIZE / 512);
             else              } else {
                 n = nb_sectors;                  n = nb_sectors;
               }
   
             while (sector_num - bs_offset >= bs_sectors) {              while (sector_num - bs_offset >= bs_sectors) {
                 bs_i ++;                  bs_i ++;
                 assert (bs_i < bs_n);                  assert (bs_i < bs_n);
                 bs_offset += bs_sectors;                  bs_offset += bs_sectors;
                 bdrv_get_geometry(bs[bs_i], &bs_sectors);                  bdrv_get_geometry(bs[bs_i], &bs_sectors);
                 /* printf("changing part: sector_num=%lld, bs_i=%d, "                  /* printf("changing part: sector_num=%" PRId64 ", bs_i=%d, "
                   "bs_offset=%lld, bs_sectors=%lld\n",                    "bs_offset=%" PRId64 ", bs_sectors=%" PRId64 "\n",
                    sector_num, bs_i, bs_offset, bs_sectors); */                     sector_num, bs_i, bs_offset, bs_sectors); */
             }              }
   
             if (n > bs_offset + bs_sectors - sector_num)              if (n > bs_offset + bs_sectors - sector_num) {
                 n = bs_offset + bs_sectors - sector_num;                  n = bs_offset + bs_sectors - sector_num;
               }
   
             if (!drv->no_zero_init) {              if (has_zero_init) {
                 /* If the output image is being created as a copy on write image,                  /* If the output image is being created as a copy on write image,
                    assume that sectors which are unallocated in the input image                     assume that sectors which are unallocated in the input image
                    are present in both the output's and input's base images (no                     are present in both the output's and input's base images (no
Line 764  static int img_convert(int argc, char ** Line 970  static int img_convert(int argc, char **
                 n1 = n;                  n1 = n;
             }              }
   
             if (bdrv_read(bs[bs_i], sector_num - bs_offset, buf, n) < 0)               ret = bdrv_read(bs[bs_i], sector_num - bs_offset, buf, n);
                 error("error while reading");              if (ret < 0) {
                   error_report("error while reading");
                   goto out;
               }
             /* NOTE: at the same time we convert, we do not write zero              /* NOTE: at the same time we convert, we do not write zero
                sectors to have a chance to compress the image. Ideally, we                 sectors to have a chance to compress the image. Ideally, we
                should add a specific call to have the info to go faster */                 should add a specific call to have the info to go faster */
Line 778  static int img_convert(int argc, char ** Line 987  static int img_convert(int argc, char **
                    If the output is to a host device, we also write out                     If the output is to a host device, we also write out
                    sectors that are entirely 0, since whatever data was                     sectors that are entirely 0, since whatever data was
                    already there is garbage, not 0s. */                     already there is garbage, not 0s. */
                 if (drv->no_zero_init || out_baseimg ||                  if (!has_zero_init || out_baseimg ||
                     is_allocated_sectors(buf1, n, &n1)) {                      is_allocated_sectors(buf1, n, &n1)) {
                     if (bdrv_write(out_bs, sector_num, buf1, n1) < 0)                      ret = bdrv_write(out_bs, sector_num, buf1, n1);
                         error("error while writing");                      if (ret < 0) {
                           error_report("error while writing");
                           goto out;
                       }
                 }                  }
                 sector_num += n1;                  sector_num += n1;
                 n -= n1;                  n -= n1;
                 buf1 += n1 * 512;                  buf1 += n1 * 512;
             }              }
               qemu_progress_print(local_progress, 100);
         }          }
     }      }
   out:
       qemu_progress_end();
       free_option_parameters(create_options);
       free_option_parameters(param);
     qemu_free(buf);      qemu_free(buf);
     bdrv_delete(out_bs);      if (out_bs) {
     for (bs_i = 0; bs_i < bs_n; bs_i++)          bdrv_delete(out_bs);
         bdrv_delete(bs[bs_i]);      }
     free(bs);      if (bs) {
           for (bs_i = 0; bs_i < bs_n; bs_i++) {
               if (bs[bs_i]) {
                   bdrv_delete(bs[bs_i]);
               }
           }
           qemu_free(bs);
       }
       if (ret) {
           return 1;
       }
     return 0;      return 0;
 }  }
   
 #ifdef _WIN32  
 static int64_t get_allocated_file_size(const char *filename)  
 {  
     typedef DWORD (WINAPI * get_compressed_t)(const char *filename, DWORD *high);  
     get_compressed_t get_compressed;  
     struct _stati64 st;  
   
     /* WinNT support GetCompressedFileSize to determine allocate size */  
     get_compressed = (get_compressed_t) GetProcAddress(GetModuleHandle("kernel32"), "GetCompressedFileSizeA");  
     if (get_compressed) {  
         DWORD high, low;  
         low = get_compressed(filename, &high);  
         if (low != 0xFFFFFFFFlu || GetLastError() == NO_ERROR)  
             return (((int64_t) high) << 32) + low;  
     }  
   
     if (_stati64(filename, &st) < 0)  
         return -1;  
     return st.st_size;  
 }  
 #else  
 static int64_t get_allocated_file_size(const char *filename)  
 {  
     struct stat st;  
     if (stat(filename, &st) < 0)  
         return -1;  
     return (int64_t)st.st_blocks * 512;  
 }  
 #endif  
   
 static void dump_snapshots(BlockDriverState *bs)  static void dump_snapshots(BlockDriverState *bs)
 {  {
Line 849  static int img_info(int argc, char **arg Line 1047  static int img_info(int argc, char **arg
 {  {
     int c;      int c;
     const char *filename, *fmt;      const char *filename, *fmt;
     BlockDriver *drv;  
     BlockDriverState *bs;      BlockDriverState *bs;
     char fmt_name[128], size_buf[128], dsize_buf[128];      char fmt_name[128], size_buf[128], dsize_buf[128];
     uint64_t total_sectors;      uint64_t total_sectors;
Line 861  static int img_info(int argc, char **arg Line 1058  static int img_info(int argc, char **arg
     fmt = NULL;      fmt = NULL;
     for(;;) {      for(;;) {
         c = getopt(argc, argv, "f:h");          c = getopt(argc, argv, "f:h");
         if (c == -1)          if (c == -1) {
             break;              break;
           }
         switch(c) {          switch(c) {
           case '?':
         case 'h':          case 'h':
             help();              help();
             break;              break;
Line 872  static int img_info(int argc, char **arg Line 1071  static int img_info(int argc, char **arg
             break;              break;
         }          }
     }      }
     if (optind >= argc)      if (optind >= argc) {
         help();          help();
       }
     filename = argv[optind++];      filename = argv[optind++];
   
     bs = bdrv_new("");      bs = bdrv_new_open(filename, fmt, BDRV_O_FLAGS | BDRV_O_NO_BACKING);
     if (!bs)      if (!bs) {
         error("Not enough memory");          return 1;
     if (fmt) {  
         drv = bdrv_find_format(fmt);  
         if (!drv)  
             error("Unknown file format '%s'", fmt);  
     } else {  
         drv = NULL;  
     }  
     if (bdrv_open2(bs, filename, BRDV_O_FLAGS, drv) < 0) {  
         error("Could not open '%s'", filename);  
     }      }
     bdrv_get_format(bs, fmt_name, sizeof(fmt_name));      bdrv_get_format(bs, fmt_name, sizeof(fmt_name));
     bdrv_get_geometry(bs, &total_sectors);      bdrv_get_geometry(bs, &total_sectors);
     get_human_readable_size(size_buf, sizeof(size_buf), total_sectors * 512);      get_human_readable_size(size_buf, sizeof(size_buf), total_sectors * 512);
     allocated_size = get_allocated_file_size(filename);      allocated_size = bdrv_get_allocated_file_size(bs);
     if (allocated_size < 0)      if (allocated_size < 0) {
         snprintf(dsize_buf, sizeof(dsize_buf), "unavailable");          snprintf(dsize_buf, sizeof(dsize_buf), "unavailable");
     else      } else {
         get_human_readable_size(dsize_buf, sizeof(dsize_buf),          get_human_readable_size(dsize_buf, sizeof(dsize_buf),
                                 allocated_size);                                  allocated_size);
       }
     printf("image: %s\n"      printf("image: %s\n"
            "file format: %s\n"             "file format: %s\n"
            "virtual size: %s (%" PRId64 " bytes)\n"             "virtual size: %s (%" PRId64 " bytes)\n"
Line 905  static int img_info(int argc, char **arg Line 1097  static int img_info(int argc, char **arg
            filename, fmt_name, size_buf,             filename, fmt_name, size_buf,
            (total_sectors * 512),             (total_sectors * 512),
            dsize_buf);             dsize_buf);
     if (bdrv_is_encrypted(bs))      if (bdrv_is_encrypted(bs)) {
         printf("encrypted: yes\n");          printf("encrypted: yes\n");
       }
     if (bdrv_get_info(bs, &bdi) >= 0) {      if (bdrv_get_info(bs, &bdi) >= 0) {
         if (bdi.cluster_size != 0)          if (bdi.cluster_size != 0) {
             printf("cluster_size: %d\n", bdi.cluster_size);              printf("cluster_size: %d\n", bdi.cluster_size);
           }
     }      }
     bdrv_get_backing_filename(bs, backing_filename, sizeof(backing_filename));      bdrv_get_backing_filename(bs, backing_filename, sizeof(backing_filename));
     if (backing_filename[0] != '\0') {      if (backing_filename[0] != '\0') {
Line 934  static int img_snapshot(int argc, char * Line 1128  static int img_snapshot(int argc, char *
     BlockDriverState *bs;      BlockDriverState *bs;
     QEMUSnapshotInfo sn;      QEMUSnapshotInfo sn;
     char *filename, *snapshot_name = NULL;      char *filename, *snapshot_name = NULL;
     int c, ret;      int c, ret = 0, bdrv_oflags;
     int action = 0;      int action = 0;
     qemu_timeval tv;      qemu_timeval tv;
   
       bdrv_oflags = BDRV_O_FLAGS | BDRV_O_RDWR;
     /* Parse commandline parameters */      /* Parse commandline parameters */
     for(;;) {      for(;;) {
         c = getopt(argc, argv, "la:c:d:h");          c = getopt(argc, argv, "la:c:d:h");
         if (c == -1)          if (c == -1) {
             break;              break;
           }
         switch(c) {          switch(c) {
           case '?':
         case 'h':          case 'h':
             help();              help();
             return 0;              return 0;
Line 953  static int img_snapshot(int argc, char * Line 1150  static int img_snapshot(int argc, char *
                 return 0;                  return 0;
             }              }
             action = SNAPSHOT_LIST;              action = SNAPSHOT_LIST;
               bdrv_oflags &= ~BDRV_O_RDWR; /* no need for RW */
             break;              break;
         case 'a':          case 'a':
             if (action) {              if (action) {
Line 981  static int img_snapshot(int argc, char * Line 1179  static int img_snapshot(int argc, char *
         }          }
     }      }
   
     if (optind >= argc)      if (optind >= argc) {
         help();          help();
       }
     filename = argv[optind++];      filename = argv[optind++];
   
     /* Open the image */      /* Open the image */
     bs = bdrv_new("");      bs = bdrv_new_open(filename, NULL, bdrv_oflags);
     if (!bs)      if (!bs) {
         error("Not enough memory");          return 1;
   
     if (bdrv_open2(bs, filename, 0, NULL) < 0) {  
         error("Could not open '%s'", filename);  
     }      }
   
     /* Perform the requested action */      /* Perform the requested action */
Line 1009  static int img_snapshot(int argc, char * Line 1205  static int img_snapshot(int argc, char *
         sn.date_nsec = tv.tv_usec * 1000;          sn.date_nsec = tv.tv_usec * 1000;
   
         ret = bdrv_snapshot_create(bs, &sn);          ret = bdrv_snapshot_create(bs, &sn);
         if (ret)          if (ret) {
             error("Could not create snapshot '%s': %d (%s)",              error_report("Could not create snapshot '%s': %d (%s)",
                 snapshot_name, ret, strerror(-ret));                  snapshot_name, ret, strerror(-ret));
           }
         break;          break;
   
     case SNAPSHOT_APPLY:      case SNAPSHOT_APPLY:
         ret = bdrv_snapshot_goto(bs, snapshot_name);          ret = bdrv_snapshot_goto(bs, snapshot_name);
         if (ret)          if (ret) {
             error("Could not apply snapshot '%s': %d (%s)",              error_report("Could not apply snapshot '%s': %d (%s)",
                 snapshot_name, ret, strerror(-ret));                  snapshot_name, ret, strerror(-ret));
           }
         break;          break;
   
     case SNAPSHOT_DELETE:      case SNAPSHOT_DELETE:
         ret = bdrv_snapshot_delete(bs, snapshot_name);          ret = bdrv_snapshot_delete(bs, snapshot_name);
         if (ret)          if (ret) {
             error("Could not delete snapshot '%s': %d (%s)",              error_report("Could not delete snapshot '%s': %d (%s)",
                 snapshot_name, ret, strerror(-ret));                  snapshot_name, ret, strerror(-ret));
           }
         break;          break;
     }      }
   
     /* Cleanup */      /* Cleanup */
     bdrv_delete(bs);      bdrv_delete(bs);
       if (ret) {
           return 1;
       }
       return 0;
   }
   
   static int img_rebase(int argc, char **argv)
   {
       BlockDriverState *bs, *bs_old_backing = NULL, *bs_new_backing = NULL;
       BlockDriver *old_backing_drv, *new_backing_drv;
       char *filename;
       const char *fmt, *cache, *out_basefmt, *out_baseimg;
       int c, flags, ret;
       int unsafe = 0;
       int progress = 0;
   
       /* Parse commandline parameters */
       fmt = NULL;
       cache = BDRV_DEFAULT_CACHE;
       out_baseimg = NULL;
       out_basefmt = NULL;
       for(;;) {
           c = getopt(argc, argv, "uhf:F:b:pt:");
           if (c == -1) {
               break;
           }
           switch(c) {
           case '?':
           case 'h':
               help();
               return 0;
           case 'f':
               fmt = optarg;
               break;
           case 'F':
               out_basefmt = optarg;
               break;
           case 'b':
               out_baseimg = optarg;
               break;
           case 'u':
               unsafe = 1;
               break;
           case 'p':
               progress = 1;
               break;
           case 't':
               cache = optarg;
               break;
           }
       }
   
       if ((optind >= argc) || (!unsafe && !out_baseimg)) {
           help();
       }
       filename = argv[optind++];
   
       qemu_progress_init(progress, 2.0);
       qemu_progress_print(0, 100);
   
       flags = BDRV_O_RDWR | (unsafe ? BDRV_O_NO_BACKING : 0);
       ret = set_cache_flag(cache, &flags);
       if (ret < 0) {
           error_report("Invalid cache option: %s", cache);
           return -1;
       }
   
       /*
        * Open the images.
        *
        * Ignore the old backing file for unsafe rebase in case we want to correct
        * the reference to a renamed or moved backing file.
        */
       bs = bdrv_new_open(filename, fmt, flags);
       if (!bs) {
           return 1;
       }
   
       /* Find the right drivers for the backing files */
       old_backing_drv = NULL;
       new_backing_drv = NULL;
   
       if (!unsafe && bs->backing_format[0] != '\0') {
           old_backing_drv = bdrv_find_format(bs->backing_format);
           if (old_backing_drv == NULL) {
               error_report("Invalid format name: '%s'", bs->backing_format);
               ret = -1;
               goto out;
           }
       }
   
       if (out_basefmt != NULL) {
           new_backing_drv = bdrv_find_format(out_basefmt);
           if (new_backing_drv == NULL) {
               error_report("Invalid format name: '%s'", out_basefmt);
               ret = -1;
               goto out;
           }
       }
   
       /* For safe rebasing we need to compare old and new backing file */
       if (unsafe) {
           /* Make the compiler happy */
           bs_old_backing = NULL;
           bs_new_backing = NULL;
       } else {
           char backing_name[1024];
   
           bs_old_backing = bdrv_new("old_backing");
           bdrv_get_backing_filename(bs, backing_name, sizeof(backing_name));
           ret = bdrv_open(bs_old_backing, backing_name, BDRV_O_FLAGS,
                           old_backing_drv);
           if (ret) {
               error_report("Could not open old backing file '%s'", backing_name);
               goto out;
           }
   
           bs_new_backing = bdrv_new("new_backing");
           ret = bdrv_open(bs_new_backing, out_baseimg, BDRV_O_FLAGS,
                           new_backing_drv);
           if (ret) {
               error_report("Could not open new backing file '%s'", out_baseimg);
               goto out;
           }
       }
   
       /*
        * Check each unallocated cluster in the COW file. If it is unallocated,
        * accesses go to the backing file. We must therefore compare this cluster
        * in the old and new backing file, and if they differ we need to copy it
        * from the old backing file into the COW file.
        *
        * If qemu-img crashes during this step, no harm is done. The content of
        * the image is the same as the original one at any time.
        */
       if (!unsafe) {
           uint64_t num_sectors;
           uint64_t sector;
           int n;
           uint8_t * buf_old;
           uint8_t * buf_new;
           float local_progress;
   
           buf_old = qemu_malloc(IO_BUF_SIZE);
           buf_new = qemu_malloc(IO_BUF_SIZE);
   
           bdrv_get_geometry(bs, &num_sectors);
   
           local_progress = (float)100 /
               (num_sectors / MIN(num_sectors, IO_BUF_SIZE / 512));
           for (sector = 0; sector < num_sectors; sector += n) {
   
               /* How many sectors can we handle with the next read? */
               if (sector + (IO_BUF_SIZE / 512) <= num_sectors) {
                   n = (IO_BUF_SIZE / 512);
               } else {
                   n = num_sectors - sector;
               }
   
               /* If the cluster is allocated, we don't need to take action */
               ret = bdrv_is_allocated(bs, sector, n, &n);
               if (ret) {
                   continue;
               }
   
               /* Read old and new backing file */
               ret = bdrv_read(bs_old_backing, sector, buf_old, n);
               if (ret < 0) {
                   error_report("error while reading from old backing file");
                   goto out;
               }
               ret = bdrv_read(bs_new_backing, sector, buf_new, n);
               if (ret < 0) {
                   error_report("error while reading from new backing file");
                   goto out;
               }
   
               /* If they differ, we need to write to the COW file */
               uint64_t written = 0;
   
               while (written < n) {
                   int pnum;
   
                   if (compare_sectors(buf_old + written * 512,
                       buf_new + written * 512, n - written, &pnum))
                   {
                       ret = bdrv_write(bs, sector + written,
                           buf_old + written * 512, pnum);
                       if (ret < 0) {
                           error_report("Error while writing to COW image: %s",
                               strerror(-ret));
                           goto out;
                       }
                   }
   
                   written += pnum;
               }
               qemu_progress_print(local_progress, 100);
           }
   
           qemu_free(buf_old);
           qemu_free(buf_new);
       }
   
       /*
        * Change the backing file. All clusters that are different from the old
        * backing file are overwritten in the COW file now, so the visible content
        * doesn't change when we switch the backing file.
        */
       ret = bdrv_change_backing_file(bs, out_baseimg, out_basefmt);
       if (ret == -ENOSPC) {
           error_report("Could not change the backing file to '%s': No "
                        "space left in the file header", out_baseimg);
       } else if (ret < 0) {
           error_report("Could not change the backing file to '%s': %s",
               out_baseimg, strerror(-ret));
       }
   
       qemu_progress_print(100, 0);
       /*
        * TODO At this point it is possible to check if any clusters that are
        * allocated in the COW file are the same in the backing file. If so, they
        * could be dropped from the COW file. Don't do this before switching the
        * backing file, in case of a crash this would lead to corruption.
        */
   out:
       qemu_progress_end();
       /* Cleanup */
       if (!unsafe) {
           if (bs_old_backing != NULL) {
               bdrv_delete(bs_old_backing);
           }
           if (bs_new_backing != NULL) {
               bdrv_delete(bs_new_backing);
           }
       }
   
       bdrv_delete(bs);
       if (ret) {
           return 1;
       }
       return 0;
   }
   
   static int img_resize(int argc, char **argv)
   {
       int c, ret, relative;
       const char *filename, *fmt, *size;
       int64_t n, total_size;
       BlockDriverState *bs = NULL;
       QEMUOptionParameter *param;
       QEMUOptionParameter resize_options[] = {
           {
               .name = BLOCK_OPT_SIZE,
               .type = OPT_SIZE,
               .help = "Virtual disk size"
           },
           { NULL }
       };
   
       /* Remove size from argv manually so that negative numbers are not treated
        * as options by getopt. */
       if (argc < 3) {
           help();
           return 1;
       }
   
       size = argv[--argc];
   
       /* Parse getopt arguments */
       fmt = NULL;
       for(;;) {
           c = getopt(argc, argv, "f:h");
           if (c == -1) {
               break;
           }
           switch(c) {
           case '?':
           case 'h':
               help();
               break;
           case 'f':
               fmt = optarg;
               break;
           }
       }
       if (optind >= argc) {
           help();
       }
       filename = argv[optind++];
   
       /* Choose grow, shrink, or absolute resize mode */
       switch (size[0]) {
       case '+':
           relative = 1;
           size++;
           break;
       case '-':
           relative = -1;
           size++;
           break;
       default:
           relative = 0;
           break;
       }
   
       /* Parse size */
       param = parse_option_parameters("", resize_options, NULL);
       if (set_option_parameter(param, BLOCK_OPT_SIZE, size)) {
           /* Error message already printed when size parsing fails */
           ret = -1;
           goto out;
       }
       n = get_option_parameter(param, BLOCK_OPT_SIZE)->value.n;
       free_option_parameters(param);
   
       bs = bdrv_new_open(filename, fmt, BDRV_O_FLAGS | BDRV_O_RDWR);
       if (!bs) {
           ret = -1;
           goto out;
       }
   
       if (relative) {
           total_size = bdrv_getlength(bs) + n * relative;
       } else {
           total_size = n;
       }
       if (total_size <= 0) {
           error_report("New image size must be positive");
           ret = -1;
           goto out;
       }
   
       ret = bdrv_truncate(bs, total_size);
       switch (ret) {
       case 0:
           printf("Image resized.\n");
           break;
       case -ENOTSUP:
           error_report("This image format does not support resize");
           break;
       case -EACCES:
           error_report("Image is read-only");
           break;
       default:
           error_report("Error resizing image (%d)", -ret);
           break;
       }
   out:
       if (bs) {
           bdrv_delete(bs);
       }
       if (ret) {
           return 1;
       }
     return 0;      return 0;
 }  }
   
Line 1049  int main(int argc, char **argv) Line 1602  int main(int argc, char **argv)
     const img_cmd_t *cmd;      const img_cmd_t *cmd;
     const char *cmdname;      const char *cmdname;
   
       error_set_progname(argv[0]);
   
     bdrv_init();      bdrv_init();
     if (argc < 2)      if (argc < 2)
         help();          help();

Removed from v.1.1.1.10  
changed lines
  Added in v.1.1.1.13


unix.superglobalmegacorp.com