Memory fixes

Signed-off-by: Richard Yao <ryao@gentoo.org>

Author: ryao

cmd/zfs/zfs_iter.c

@@ -95,6 +95,8 @@ zfs_callback(zfs_handle_t *zhp, void *data)
 	boolean_t dontclose = B_FALSE;
 	boolean_t include_snaps = zfs_include_snapshots(zhp, cb);
 
+	zhp = zfs_handle_dup(zhp);
+
 	if ((zfs_get_type(zhp) & cb->cb_types) ||
 	    ((zfs_get_type(zhp) == ZFS_TYPE_SNAPSHOT) && include_snaps)) {
 		uu_avl_index_t idx;
@@ -447,6 +449,7 @@ zfs_for_each(int argc, char **argv, int flags, zfs_type_t types,
 				    limit_specified, zfs_callback, &cb);
 			else
 				ret = 1;
+			zfs_close(zhp);
 		}
 	}
 

lib/libzfs/libzfs_iter.c

@@ -145,6 +145,8 @@ zfs_iter_cb(nvlist_t *nvl, void *data)
 
 	ret = (*cb->zicb_func)(nzhp, cb->zicb_data);
 
+	zfs_close(nzhp);
+
 	return (ret);
 }