commit 58c58cf7ef922ef019ce78ce1c418a7643c39518
author Paul Lawrence <paullawrence@google.com> Wed Jun 04 13:12:21 2014 -0700
committer Paul Lawrence <paullawrence@google.com> Thu Jun 05 19:44:07 2014 +0000
tree 39c83ce6893dd25caeb8b9193f8e07e70d1bea64
parent 6a74dcaa6e646fea8e00b7c04332fc60fe7e017c

Show correct percentage when encrypting used blocks only

Bug: 12975202
Change-Id: I7e4adfa2c4951055bbb2c11986a7784f465f586f

cryptfs.c

diff --git a/cryptfs.c b/cryptfs.c
index 0648231..bd03c08 100644
--- a/cryptfs.c
+++ b/cryptfs.c
@@ -1900,6 +1900,7 @@
     off64_t numblocks;
     off64_t one_pct, cur_pct, new_pct;
     off64_t blocks_already_done, tot_numblocks;
+    off64_t used_blocks_already_done, tot_used_blocks;
     char* real_blkdev, * crypto_blkdev;
     int count;
     off64_t offset;
@@ -1908,15 +1909,27 @@
     int completed;
 };
 
-static void update_progress(struct encryptGroupsData* data)
+static void update_progress(struct encryptGroupsData* data, int is_used)
 {
     data->blocks_already_done++;
-    data->new_pct = data->blocks_already_done / data->one_pct;
+
+    if (is_used) {
+        data->used_blocks_already_done++;
+    }
+
+    if (data->tot_used_blocks) {
+        data->new_pct = data->used_blocks_already_done / data->one_pct;
+    } else {
+        data->new_pct = data->blocks_already_done / data->one_pct;
+    }
+
     if (data->new_pct > data->cur_pct) {
         char buf[8];
         data->cur_pct = data->new_pct;
         snprintf(buf, sizeof(buf), "%lld", data->cur_pct);
         property_set("vold.encrypt_progress", buf);
+
+        SLOGI("Encrypted %lld percent of drive", data->cur_pct);
     }
 }
 
@@ -1994,8 +2007,9 @@
         data->count = 0;
 
         for (block = 0; block < block_count; block++) {
-            update_progress(data);
-            if (bitmap_get_bit(block_bitmap, block)) {
+            int used = bitmap_get_bit(block_bitmap, block);
+            update_progress(data, used);
+            if (used) {
                 if (data->count == 0) {
                     data->offset = offset;
                 }
@@ -2044,7 +2058,7 @@
                                        off64_t tot_size,
                                        off64_t previously_encrypted_upto)
 {
-    int i;
+    u32 i;
     struct encryptGroupsData data;
     int rc = -1;
 
@@ -2085,7 +2099,12 @@
 
     SLOGI("Encrypting filesystem in place...");
 
-    data.one_pct = data.tot_numblocks / 100;
+    data.tot_used_blocks = data.numblocks;
+    for (i = 0; i < aux_info.groups; ++i) {
+      data.tot_used_blocks -= aux_info.bg_desc[i].bg_free_blocks_count;
+    }
+
+    data.one_pct = data.tot_used_blocks / 100;
     data.cur_pct = 0;
 
     rc = encrypt_groups(&data);