[PATCH v2] qemu-img: Explicit number replaced by a constant

2020-08-18 Thread Yi Li
Signed-off-by: Yi Li 
---
 qemu-img.c | 10 +-
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/qemu-img.c b/qemu-img.c
index 5308773811..aa2e31c8ae 100644
--- a/qemu-img.c
+++ b/qemu-img.c
@@ -1200,10 +1200,10 @@ static int is_allocated_sectors(const uint8_t *buf, int 
n, int *pnum,
 *pnum = 0;
 return 0;
 }
-is_zero = buffer_is_zero(buf, 512);
+is_zero = buffer_is_zero(buf, BDRV_SECTOR_SIZE);
 for(i = 1; i < n; i++) {
-buf += 512;
-if (is_zero != buffer_is_zero(buf, 512)) {
+buf += BDRV_SECTOR_SIZE;
+if (is_zero != buffer_is_zero(buf, BDRV_SECTOR_SIZE)) {
 break;
 }
 }
@@ -2489,8 +2489,8 @@ static int img_convert(int argc, char **argv)
 }
 }
 
-qemu_opt_set_number(opts, BLOCK_OPT_SIZE, s.total_sectors * 512,
-&error_abort);
+qemu_opt_set_number(opts, BLOCK_OPT_SIZE,
+s.total_sectors * BDRV_SECTOR_SIZE, &error_abort);
 ret = add_old_style_options(out_fmt, opts, out_baseimg, NULL);
 if (ret < 0) {
 goto out;
-- 
2.25.3






[PATCH] qemu-img: Explicit number replaced by a constant

2020-08-17 Thread Yi Li
Signed-off-by: Yi Li 
---
 qemu-img.c | 12 ++--
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/qemu-img.c b/qemu-img.c
index 5308773811..a0fbc2757c 100644
--- a/qemu-img.c
+++ b/qemu-img.c
@@ -1181,7 +1181,7 @@ static int64_t find_nonzero(const uint8_t *buf, int64_t n)
 }
 
 /*
- * Returns true iff the first sector pointed to by 'buf' contains at least
+ * Returns true if the first sector pointed to by 'buf' contains at least
  * a non-NUL byte.
  *
  * 'pnum' is set to the number of sectors (including and immediately following
@@ -1200,10 +1200,10 @@ static int is_allocated_sectors(const uint8_t *buf, int 
n, int *pnum,
 *pnum = 0;
 return 0;
 }
-is_zero = buffer_is_zero(buf, 512);
+is_zero = buffer_is_zero(buf, BDRV_SECTOR_SIZE);
 for(i = 1; i < n; i++) {
-buf += 512;
-if (is_zero != buffer_is_zero(buf, 512)) {
+buf += BDRV_SECTOR_SIZE;
+if (is_zero != buffer_is_zero(buf, BDRV_SECTOR_SIZE)) {
 break;
 }
 }
@@ -2489,8 +2489,8 @@ static int img_convert(int argc, char **argv)
 }
 }
 
-qemu_opt_set_number(opts, BLOCK_OPT_SIZE, s.total_sectors * 512,
-&error_abort);
+qemu_opt_set_number(opts, BLOCK_OPT_SIZE,
+s.total_sectors * BDRV_SECTOR_SIZE, &error_abort);
 ret = add_old_style_options(out_fmt, opts, out_baseimg, NULL);
 if (ret < 0) {
 goto out;
-- 
2.25.3






Re: [PATCH v2] rbd: Use RBD fast-diff for querying actual allocation

2020-06-17 Thread Yi Li
ping ?

On 6/11/20, Yi Li  wrote:
> Since Ceph version Infernalis (9.2.0) the new fast-diff mechanism
> of RBD allows for querying actual rbd image usage.
>
> Prior to this version there was no easy and fast way to query how
> much allocation a RBD image had inside a Ceph cluster.
>
> To use the fast-diff feature it needs to be enabled per RBD image
> and is only supported by Ceph cluster running version Infernalis
> (9.2.0) or newer.
>
> The fast-diff feature disabled or fast-diff map is marked as invalid,
> qemu-img will report an allocation identical to the image capacity.
>
> 'qemu-img info rbd:cepharm/liyi-rbd' might output for example:
>
>   image: json:{"driver": "raw", "file": {"pool": "cepharm",
>   "image": "liyi-rbd", "driver": "rbd"}}
>   file format: raw
>   virtual size: 20 GiB (21474836480 bytes)
>   disk size: 0 B
>   cluster_size: 4194304
>
> Newly created rbds will have the fast-diff feature enabled.
>
> Signed-off-by: Yi Li 
> ---
>  block/rbd.c | 103 
>  1 file changed, 103 insertions(+)
>
> diff --git a/block/rbd.c b/block/rbd.c
> index 617553b022..c1e68ff7e9 100644
> --- a/block/rbd.c
> +++ b/block/rbd.c
> @@ -1107,6 +1107,108 @@ static int64_t qemu_rbd_getlength(BlockDriverState
> *bs)
>  return info.size;
>  }
>
> +#if LIBRBD_VERSION_CODE > 265
> +static int disk_usage_callback(uint64_t offset, size_t len, int exists,
> +   void *arg)
> +{
> +  uint64_t *used_size = (uint64_t *)(arg);
> +  if (exists) {
> +(*used_size) += len;
> +  }
> +  return 0;
> +}
> +
> +static int qemu_rbd_getflags(rbd_image_t image, uint64_t *flags)
> +{
> +int r;
> +
> +r = rbd_get_flags(image, flags);
> +if (r < 0) {
> +return r;
> +}
> +return 0;
> +}
> +
> +static bool qemu_rbd_use_fastdiff(uint64_t features, uint64_t flags)
> +{
> +return (((features & RBD_FEATURE_FAST_DIFF) != 0ULL) &&
> +((flags & RBD_FLAG_FAST_DIFF_INVALID) == 0ULL));
> +}
> +
> +static int qemu_rbd_set_allocation(rbd_image_t image,
> +   rbd_image_info_t *info,
> +   uint64_t *used_size)
> +{
> +int r;
> +/*
> + * RBD image fast-diff feature enabled
> + * Querying for actual allocation.
> + */
> +r = rbd_diff_iterate2(image, NULL, 0, info->size, 0, 1,
> +  &disk_usage_callback,
> +  used_size);
> +if (r < 0) {
> +return r;
> +}
> +return 0;
> +}
> +
> +#else
> +static int qemu_rbd_getflags(rbd_image_t image G_GNUC_UNUSED, uint64_t
> *flags)
> +{
> +*flags = 0;
> +return 0;
> +}
> +
> +static bool qemu_rbd_use_fastdiff(uint64_t features G_GNUC_UNUSED,
> +  uint64_t feature_flags G_GNUC_UNUSED)
> +{
> +return false;
> +}
> +
> +static int qemu_rbd_set_allocation(rbd_image_t image G_GNUC_UNUSED,
> +   rbd_image_info_t *info _GNUC_UNUSED,
> +   uint64_t *used_size _GNUC_UNUSED)
> +{
> +return 0;
> +}
> +#endif
> +
> +static int64_t qemu_rbd_allocated_file_size(BlockDriverState *bs)
> +{
> +BDRVRBDState *s = bs->opaque;
> +rbd_image_info_t info;
> +int r;
> +uint64_t used_size = 0;
> +uint64_t features = 0;
> +uint64_t flags = 0;
> +
> +r = rbd_stat(s->image, &info, sizeof(info));
> +if (r < 0) {
> +return r;
> +}
> +
> +r = rbd_get_features(s->image, &features);
> +if (r < 0) {
> +return r;
> +}
> +
> +r = qemu_rbd_getflags(s->image, &flags);
> +if (r < 0) {
> +return r;
> +}
> +
> +if (qemu_rbd_use_fastdiff(features, flags)) {
> +r = qemu_rbd_set_allocation(s->image, &info, &used_size);
> +if (r < 0) {
> +return r;
> +}
> +} else {
> +used_size = info.size;
> +}
> +return used_size;
> +}
> +
>  static int coroutine_fn qemu_rbd_co_truncate(BlockDriverState *bs,
>   int64_t offset,
>   bool exact,
> @@ -1316,6 +1418,7 @@ static BlockDriver bdrv_rbd = {
>  .bdrv_get_info  = qemu_rbd_getinfo,
>  .create_opts= &qemu_rbd_create_opts,
>  .bdrv_getlength = qemu_rbd_getlength,
> +.bdrv_get_allocated_file_size = qemu_rbd_allocated_file_size,
>  .bdrv_co_truncate   = qemu_rbd_co_truncate,
>  .protocol_name  = "rbd",
>
> --
> 2.25.3
>
>
>
>



[PATCH v2] rbd: Use RBD fast-diff for querying actual allocation

2020-06-09 Thread Yi Li
Since Ceph version Infernalis (9.2.0) the new fast-diff mechanism
of RBD allows for querying actual rbd image usage.

Prior to this version there was no easy and fast way to query how
much allocation a RBD image had inside a Ceph cluster.

To use the fast-diff feature it needs to be enabled per RBD image
and is only supported by Ceph cluster running version Infernalis
(9.2.0) or newer.

The fast-diff feature disabled or fast-diff map is marked as invalid,
qemu-img will report an allocation identical to the image capacity.

'qemu-img info rbd:cepharm/liyi-rbd' might output for example:

  image: json:{"driver": "raw", "file": {"pool": "cepharm",
  "image": "liyi-rbd", "driver": "rbd"}}
  file format: raw
  virtual size: 20 GiB (21474836480 bytes)
  disk size: 0 B
  cluster_size: 4194304

Newly created rbds will have the fast-diff feature enabled.

Signed-off-by: Yi Li 
---
 block/rbd.c | 103 
 1 file changed, 103 insertions(+)

diff --git a/block/rbd.c b/block/rbd.c
index 617553b022..c1e68ff7e9 100644
--- a/block/rbd.c
+++ b/block/rbd.c
@@ -1107,6 +1107,108 @@ static int64_t qemu_rbd_getlength(BlockDriverState *bs)
 return info.size;
 }
 
+#if LIBRBD_VERSION_CODE > 265
+static int disk_usage_callback(uint64_t offset, size_t len, int exists,
+   void *arg)
+{
+  uint64_t *used_size = (uint64_t *)(arg);
+  if (exists) {
+(*used_size) += len;
+  }
+  return 0;
+}
+
+static int qemu_rbd_getflags(rbd_image_t image, uint64_t *flags)
+{
+int r;
+
+r = rbd_get_flags(image, flags);
+if (r < 0) {
+return r;
+}
+return 0;
+}
+
+static bool qemu_rbd_use_fastdiff(uint64_t features, uint64_t flags)
+{
+return (((features & RBD_FEATURE_FAST_DIFF) != 0ULL) &&
+((flags & RBD_FLAG_FAST_DIFF_INVALID) == 0ULL));
+}
+
+static int qemu_rbd_set_allocation(rbd_image_t image,
+   rbd_image_info_t *info,
+   uint64_t *used_size)
+{
+int r;
+/*
+ * RBD image fast-diff feature enabled
+ * Querying for actual allocation.
+ */
+r = rbd_diff_iterate2(image, NULL, 0, info->size, 0, 1,
+  &disk_usage_callback,
+  used_size);
+if (r < 0) {
+return r;
+}
+return 0;
+}
+
+#else
+static int qemu_rbd_getflags(rbd_image_t image G_GNUC_UNUSED, uint64_t *flags)
+{
+*flags = 0;
+return 0;
+}
+
+static bool qemu_rbd_use_fastdiff(uint64_t features G_GNUC_UNUSED,
+  uint64_t feature_flags G_GNUC_UNUSED)
+{
+return false;
+}
+
+static int qemu_rbd_set_allocation(rbd_image_t image G_GNUC_UNUSED,
+   rbd_image_info_t *info _GNUC_UNUSED,
+   uint64_t *used_size _GNUC_UNUSED)
+{
+return 0;
+}
+#endif
+
+static int64_t qemu_rbd_allocated_file_size(BlockDriverState *bs)
+{
+BDRVRBDState *s = bs->opaque;
+rbd_image_info_t info;
+int r;
+uint64_t used_size = 0;
+uint64_t features = 0;
+uint64_t flags = 0;
+
+r = rbd_stat(s->image, &info, sizeof(info));
+if (r < 0) {
+return r;
+}
+
+r = rbd_get_features(s->image, &features);
+if (r < 0) {
+return r;
+}
+
+r = qemu_rbd_getflags(s->image, &flags);
+if (r < 0) {
+return r;
+}
+
+if (qemu_rbd_use_fastdiff(features, flags)) {
+r = qemu_rbd_set_allocation(s->image, &info, &used_size);
+if (r < 0) {
+return r;
+}
+} else {
+used_size = info.size;
+}
+return used_size;
+}
+
 static int coroutine_fn qemu_rbd_co_truncate(BlockDriverState *bs,
  int64_t offset,
  bool exact,
@@ -1316,6 +1418,7 @@ static BlockDriver bdrv_rbd = {
 .bdrv_get_info  = qemu_rbd_getinfo,
 .create_opts= &qemu_rbd_create_opts,
 .bdrv_getlength = qemu_rbd_getlength,
+.bdrv_get_allocated_file_size = qemu_rbd_allocated_file_size,
 .bdrv_co_truncate   = qemu_rbd_co_truncate,
 .protocol_name  = "rbd",
 
-- 
2.25.3