fixing a test case (#147)

Cherry-picking the changes from #137

Signed-off-by: Vishnu Itta <[email protected]>

Author: pawanpraka1

cmd/uzfs_test/uzfs_test_rebuilding.c

@@ -207,29 +207,6 @@ uzfs_test_meta_diff_traverse_cb(off_t offset, size_t len,
 	return (err);
 }
 
-static void
-check_snapshot(zvol_state_t *zv, blk_metadata_t *md, boolean_t err)
-{
-	objset_t *s_obj;
-	char *dataset;
-	int ret = 0;
-
-	dataset = kmem_asprintf("%s@%s%lu", zv->zv_name,
-	    IO_DIFF_SNAPNAME, md->io_num);
-
-	ret = dmu_objset_own(dataset, DMU_OST_ANY, B_TRUE, zv, &s_obj);
-	if ((ret != 0 && err) ||
-	    (!err && ret == 0)) {
-		printf("ret:%d\n", ret);
-		printf("snapshot %s %s\n", dataset,
-		    (err) ? "should not be removed" : "should be removed");
-		exit(1);
-	}
-
-	if (ret == 0)
-		dmu_objset_disown(s_obj, zv);
-}
-
 static void
 fetch_modified_data(void *arg)
 {
@@ -258,9 +235,7 @@ fetch_modified_data(void *arg)
 			break;
 
 		offset += len;
-		if (offset != r_data->zvol->zv_volsize)
-			check_snapshot(repl_data->zvol, &md, B_TRUE);
-		else
+		if (offset == r_data->zvol->zv_volsize)
 			break;
 	}
 
@@ -269,8 +244,6 @@ fetch_modified_data(void *arg)
 		exit(1);
 	}
 
-	check_snapshot(repl_data->zvol, &md, B_FALSE);
-
 	printf("finished fetching modified data\n");
 
 	mutex_enter(&r_data->mtx);

include/uzfs_rebuilding.h

@@ -60,6 +60,10 @@ uzfs_zvol_create_snaprebuild_clone(zvol_state_t *zv,
 int
 uzfs_zvol_destroy_snaprebuild_clone(zvol_state_t *zv,
     zvol_state_t *snap_zv);
+int
+uzfs_zvol_create_internal_snapshot(zvol_state_t *zv, zvol_state_t **snap_zv,
+    uint64_t io_num, char **snap);
+
 #ifdef __cplusplus
 }
 #endif