From 4ddbbf5467518d55130f5e5d1839f42efef0783c Mon Sep 17 00:00:00 2001
From: Sage Weil <sage@newdream.net>
Date: Thu, 19 Apr 2012 16:18:55 -0700
Subject: [PATCH] librbd: allow image resize to non-block boundaries

The caller is still invalidating the entire cache, so we don't need to
deal with discard at this level.  That might be worth cleaning up
later, though.

Fixes: #2296
Signed-off-by: Sage Weil <sage@newdream.net>
---
 src/librbd.cc | 22 +++++++++++++++++-----
 1 file changed, 17 insertions(+), 5 deletions(-)

diff --git a/src/librbd.cc b/src/librbd.cc
index c9bfb5efff5a9..029f043db58d3 100644
--- a/src/librbd.cc
+++ b/src/librbd.cc
@@ -570,11 +570,23 @@ void trim_image(IoCtx& io_ctx, const rbd_obj_header_ondisk &header, uint64_t new
   uint64_t bsize = get_block_size(header);
   uint64_t numseg = get_max_block(header);
   uint64_t start = get_block_num(header, newsize);
-  ldout(cct, 2) << "trimming image data from " << numseg << " to " << start << " objects..." << dendl;
-  for (uint64_t i=start; i<numseg; i++) {
-    string oid = get_block_oid(header, i);
-    io_ctx.remove(oid);
-    prog_ctx.update_progress(i * bsize, (numseg - start) * bsize);
+
+  uint64_t block_ofs = get_block_ofs(header, newsize);
+  if (block_ofs) {
+    ldout(cct, 2) << "trim_image object " << numseg << " truncate to " << block_ofs << dendl;
+    string oid = get_block_oid(header, start);
+    librados::ObjectWriteOperation write_op;
+    write_op.truncate(block_ofs);
+    io_ctx.operate(oid, &write_op);
+    start++;
+  }
+  if (start < numseg) {
+    ldout(cct, 2) << "trim_image objects " << start << " to " << (numseg-1) << dendl;
+    for (uint64_t i=start; i<numseg; i++) {
+      string oid = get_block_oid(header, i);
+      io_ctx.remove(oid);
+      prog_ctx.update_progress(i * bsize, (numseg - start) * bsize);
+    }
   }
 }
 
-- 
2.39.5