feat(blobstore/allocated): retrieve blob allocated cluster bitmap

Signed-off-by: Tiago Castro <[email protected]>

include/spdk/blob.h

@@ -1180,6 +1180,17 @@ bool spdk_blob_is_degraded(const struct spdk_blob *blob);
  */
 void spdk_blob_reset_used_clusters_cache(struct spdk_blob *blob);
 
+/**
+ * Get the allocated cluster bitmap of the given blob.
+ *
+ * \param blob The blob to retrieve allocated bitmap from.
+ *
+ * \return NULL if no memory available, otherwise returns a bitmap where each bit corresponds
+ * to a cluster. A set bit indicates the cluster is owned by the blob.
+ * Must be freed using spdk_bit_array_free.
+ */
+struct spdk_bit_array *spdk_blob_get_cluster_bitmap(struct spdk_blob *blob);
+
 #ifdef __cplusplus
 }
 #endif

lib/blob/blobstore.c

@@ -5985,6 +5985,28 @@ spdk_blob_reset_used_clusters_cache(struct spdk_blob *blob)
 		spdk_spin_unlock(&blob->bs->used_lock);
 	}
 }
+
+struct spdk_bit_array *
+spdk_blob_get_cluster_bitmap(struct spdk_blob *blob)
+{
+	struct spdk_bit_array *allocated_bitmap;
+	uint64_t i;
+
+	allocated_bitmap = spdk_bit_array_create(blob->active.num_clusters);
+	if (allocated_bitmap == NULL) {
+		SPDK_ERRLOG("Failed to allocate bit array memory for blob #%ld", blob->id);
+		return NULL;
+	}
+
+	for (i = 0; i < blob->active.num_clusters; i++) {
+		if (blob->active.clusters[i]) {
+			spdk_bit_array_set(allocated_bitmap, i);
+		}
+	}
+
+	return allocated_bitmap;
+}
+
 /* START spdk_bs_create_blob */
 
 static void