15367Sahrens /*
25367Sahrens  * CDDL HEADER START
35367Sahrens  *
45367Sahrens  * The contents of this file are subject to the terms of the
55367Sahrens  * Common Development and Distribution License (the "License").
65367Sahrens  * You may not use this file except in compliance with the License.
75367Sahrens  *
85367Sahrens  * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
95367Sahrens  * or http://www.opensolaris.org/os/licensing.
105367Sahrens  * See the License for the specific language governing permissions
115367Sahrens  * and limitations under the License.
125367Sahrens  *
135367Sahrens  * When distributing Covered Code, include this CDDL HEADER in each
145367Sahrens  * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
155367Sahrens  * If applicable, add the following below this CDDL HEADER, with the
165367Sahrens  * fields enclosed by brackets "[]" replaced with your own identifying
175367Sahrens  * information: Portions Copyright [yyyy] [name of copyright owner]
185367Sahrens  *
195367Sahrens  * CDDL HEADER END
205367Sahrens  */
215367Sahrens 
225367Sahrens /*
238584SLori.Alt@Sun.COM  * Copyright 2009 Sun Microsystems, Inc.  All rights reserved.
245367Sahrens  * Use is subject to license terms.
255367Sahrens  */
265367Sahrens 
275367Sahrens #include <assert.h>
285367Sahrens #include <ctype.h>
295367Sahrens #include <errno.h>
305367Sahrens #include <libintl.h>
315367Sahrens #include <stdio.h>
325367Sahrens #include <stdlib.h>
335367Sahrens #include <strings.h>
345367Sahrens #include <unistd.h>
355367Sahrens #include <stddef.h>
365367Sahrens #include <fcntl.h>
375367Sahrens #include <sys/mount.h>
385367Sahrens 
395367Sahrens #include <libzfs.h>
405367Sahrens 
415367Sahrens #include "zfs_namecheck.h"
425367Sahrens #include "zfs_prop.h"
4310602SLori.Alt@Sun.COM #include "zfs_fletcher.h"
445367Sahrens #include "libzfs_impl.h"
455367Sahrens 
467366STim.Haley@Sun.COM static int zfs_receive_impl(libzfs_handle_t *, const char *, recvflags_t,
477366STim.Haley@Sun.COM     int, avl_tree_t *, char **);
487366STim.Haley@Sun.COM 
495367Sahrens /*
505367Sahrens  * Routines for dealing with the AVL tree of fs-nvlists
515367Sahrens  */
525367Sahrens typedef struct fsavl_node {
535367Sahrens 	avl_node_t fn_node;
545367Sahrens 	nvlist_t *fn_nvfs;
555367Sahrens 	char *fn_snapname;
565367Sahrens 	uint64_t fn_guid;
575367Sahrens } fsavl_node_t;
585367Sahrens 
595367Sahrens static int
605367Sahrens fsavl_compare(const void *arg1, const void *arg2)
615367Sahrens {
625367Sahrens 	const fsavl_node_t *fn1 = arg1;
635367Sahrens 	const fsavl_node_t *fn2 = arg2;
645367Sahrens 
655367Sahrens 	if (fn1->fn_guid > fn2->fn_guid)
665367Sahrens 		return (+1);
675367Sahrens 	else if (fn1->fn_guid < fn2->fn_guid)
685367Sahrens 		return (-1);
695367Sahrens 	else
705367Sahrens 		return (0);
715367Sahrens }
725367Sahrens 
735367Sahrens /*
745367Sahrens  * Given the GUID of a snapshot, find its containing filesystem and
755367Sahrens  * (optionally) name.
765367Sahrens  */
775367Sahrens static nvlist_t *
785367Sahrens fsavl_find(avl_tree_t *avl, uint64_t snapguid, char **snapname)
795367Sahrens {
805367Sahrens 	fsavl_node_t fn_find;
815367Sahrens 	fsavl_node_t *fn;
825367Sahrens 
835367Sahrens 	fn_find.fn_guid = snapguid;
845367Sahrens 
855367Sahrens 	fn = avl_find(avl, &fn_find, NULL);
865367Sahrens 	if (fn) {
875367Sahrens 		if (snapname)
885367Sahrens 			*snapname = fn->fn_snapname;
895367Sahrens 		return (fn->fn_nvfs);
905367Sahrens 	}
915367Sahrens 	return (NULL);
925367Sahrens }
935367Sahrens 
945896Sahl static void
955896Sahl fsavl_destroy(avl_tree_t *avl)
965896Sahl {
975896Sahl 	fsavl_node_t *fn;
985896Sahl 	void *cookie;
995896Sahl 
1005896Sahl 	if (avl == NULL)
1015896Sahl 		return;
1025896Sahl 
1035896Sahl 	cookie = NULL;
1045896Sahl 	while ((fn = avl_destroy_nodes(avl, &cookie)) != NULL)
1055896Sahl 		free(fn);
1065896Sahl 	avl_destroy(avl);
1075896Sahl 	free(avl);
1085896Sahl }
1095896Sahl 
11010200SLori.Alt@Sun.COM /*
11110200SLori.Alt@Sun.COM  * Given an nvlist, produce an avl tree of snapshots, ordered by guid
11210200SLori.Alt@Sun.COM  */
1135367Sahrens static avl_tree_t *
1145367Sahrens fsavl_create(nvlist_t *fss)
1155367Sahrens {
1165367Sahrens 	avl_tree_t *fsavl;
1175367Sahrens 	nvpair_t *fselem = NULL;
1185367Sahrens 
1195896Sahl 	if ((fsavl = malloc(sizeof (avl_tree_t))) == NULL)
1205896Sahl 		return (NULL);
1215896Sahl 
1225367Sahrens 	avl_create(fsavl, fsavl_compare, sizeof (fsavl_node_t),
1235367Sahrens 	    offsetof(fsavl_node_t, fn_node));
1245367Sahrens 
1255367Sahrens 	while ((fselem = nvlist_next_nvpair(fss, fselem)) != NULL) {
1265367Sahrens 		nvlist_t *nvfs, *snaps;
1275367Sahrens 		nvpair_t *snapelem = NULL;
1285367Sahrens 
1295367Sahrens 		VERIFY(0 == nvpair_value_nvlist(fselem, &nvfs));
1305367Sahrens 		VERIFY(0 == nvlist_lookup_nvlist(nvfs, "snaps", &snaps));
1315367Sahrens 
1325367Sahrens 		while ((snapelem =
1335367Sahrens 		    nvlist_next_nvpair(snaps, snapelem)) != NULL) {
1345367Sahrens 			fsavl_node_t *fn;
1355367Sahrens 			uint64_t guid;
1365367Sahrens 
1375367Sahrens 			VERIFY(0 == nvpair_value_uint64(snapelem, &guid));
1385896Sahl 			if ((fn = malloc(sizeof (fsavl_node_t))) == NULL) {
1395896Sahl 				fsavl_destroy(fsavl);
1405896Sahl 				return (NULL);
1415896Sahl 			}
1425367Sahrens 			fn->fn_nvfs = nvfs;
1435367Sahrens 			fn->fn_snapname = nvpair_name(snapelem);
1445367Sahrens 			fn->fn_guid = guid;
1455367Sahrens 
1465367Sahrens 			/*
1475367Sahrens 			 * Note: if there are multiple snaps with the
1485367Sahrens 			 * same GUID, we ignore all but one.
1495367Sahrens 			 */
1505367Sahrens 			if (avl_find(fsavl, fn, NULL) == NULL)
1515367Sahrens 				avl_add(fsavl, fn);
1525367Sahrens 			else
1535367Sahrens 				free(fn);
1545367Sahrens 		}
1555367Sahrens 	}
1565367Sahrens 
1575367Sahrens 	return (fsavl);
1585367Sahrens }
1595367Sahrens 
1605367Sahrens /*
1615367Sahrens  * Routines for dealing with the giant nvlist of fs-nvlists, etc.
1625367Sahrens  */
1635367Sahrens typedef struct send_data {
1645367Sahrens 	uint64_t parent_fromsnap_guid;
1655367Sahrens 	nvlist_t *parent_snaps;
1665367Sahrens 	nvlist_t *fss;
1677265Sahrens 	nvlist_t *snapprops;
1685367Sahrens 	const char *fromsnap;
1695367Sahrens 	const char *tosnap;
1705367Sahrens 
1715367Sahrens 	/*
1725367Sahrens 	 * The header nvlist is of the following format:
1735367Sahrens 	 * {
1745367Sahrens 	 *   "tosnap" -> string
1755367Sahrens 	 *   "fromsnap" -> string (if incremental)
1765367Sahrens 	 *   "fss" -> {
1775367Sahrens 	 *	id -> {
1785367Sahrens 	 *
1795367Sahrens 	 *	 "name" -> string (full name; for debugging)
1805367Sahrens 	 *	 "parentfromsnap" -> number (guid of fromsnap in parent)
1815367Sahrens 	 *
1825367Sahrens 	 *	 "props" -> { name -> value (only if set here) }
1835367Sahrens 	 *	 "snaps" -> { name (lastname) -> number (guid) }
1847265Sahrens 	 *	 "snapprops" -> { name (lastname) -> { name -> value } }
1855367Sahrens 	 *
1865367Sahrens 	 *	 "origin" -> number (guid) (if clone)
1875367Sahrens 	 *	 "sent" -> boolean (not on-disk)
1885367Sahrens 	 *	}
1895367Sahrens 	 *   }
1905367Sahrens 	 * }
1915367Sahrens 	 *
1925367Sahrens 	 */
1935367Sahrens } send_data_t;
1945367Sahrens 
1957265Sahrens static void send_iterate_prop(zfs_handle_t *zhp, nvlist_t *nv);
1967265Sahrens 
1975367Sahrens static int
1985367Sahrens send_iterate_snap(zfs_handle_t *zhp, void *arg)
1995367Sahrens {
2005367Sahrens 	send_data_t *sd = arg;
2015367Sahrens 	uint64_t guid = zhp->zfs_dmustats.dds_guid;
2025367Sahrens 	char *snapname;
2037265Sahrens 	nvlist_t *nv;
2045367Sahrens 
2055367Sahrens 	snapname = strrchr(zhp->zfs_name, '@')+1;
2065367Sahrens 
2075367Sahrens 	VERIFY(0 == nvlist_add_uint64(sd->parent_snaps, snapname, guid));
2085367Sahrens 	/*
2095367Sahrens 	 * NB: if there is no fromsnap here (it's a newly created fs in
2105367Sahrens 	 * an incremental replication), we will substitute the tosnap.
2115367Sahrens 	 */
2125367Sahrens 	if ((sd->fromsnap && strcmp(snapname, sd->fromsnap) == 0) ||
2135367Sahrens 	    (sd->parent_fromsnap_guid == 0 && sd->tosnap &&
2145367Sahrens 	    strcmp(snapname, sd->tosnap) == 0)) {
2155367Sahrens 		sd->parent_fromsnap_guid = guid;
2165367Sahrens 	}
2175367Sahrens 
2187265Sahrens 	VERIFY(0 == nvlist_alloc(&nv, NV_UNIQUE_NAME, 0));
2197265Sahrens 	send_iterate_prop(zhp, nv);
2207265Sahrens 	VERIFY(0 == nvlist_add_nvlist(sd->snapprops, snapname, nv));
2217265Sahrens 	nvlist_free(nv);
2227265Sahrens 
2235367Sahrens 	zfs_close(zhp);
2245367Sahrens 	return (0);
2255367Sahrens }
2265367Sahrens 
2275367Sahrens static void
2285367Sahrens send_iterate_prop(zfs_handle_t *zhp, nvlist_t *nv)
2295367Sahrens {
2305367Sahrens 	nvpair_t *elem = NULL;
2315367Sahrens 
2325367Sahrens 	while ((elem = nvlist_next_nvpair(zhp->zfs_props, elem)) != NULL) {
2335367Sahrens 		char *propname = nvpair_name(elem);
2345367Sahrens 		zfs_prop_t prop = zfs_name_to_prop(propname);
2355367Sahrens 		nvlist_t *propnv;
2365367Sahrens 
237*10960SEric.Schrock@Sun.COM 		if (!zfs_prop_user(propname)) {
238*10960SEric.Schrock@Sun.COM 			/*
239*10960SEric.Schrock@Sun.COM 			 * Realistically, this should never happen.  However,
240*10960SEric.Schrock@Sun.COM 			 * we want the ability to add DSL properties without
241*10960SEric.Schrock@Sun.COM 			 * needing to make incompatible version changes.  We
242*10960SEric.Schrock@Sun.COM 			 * need to ignore unknown properties to allow older
243*10960SEric.Schrock@Sun.COM 			 * software to still send datasets containing these
244*10960SEric.Schrock@Sun.COM 			 * properties, with the unknown properties elided.
245*10960SEric.Schrock@Sun.COM 			 */
246*10960SEric.Schrock@Sun.COM 			if (prop == ZPROP_INVAL)
247*10960SEric.Schrock@Sun.COM 				continue;
2489396SMatthew.Ahrens@Sun.COM 
249*10960SEric.Schrock@Sun.COM 			if (zfs_prop_readonly(prop))
250*10960SEric.Schrock@Sun.COM 				continue;
251*10960SEric.Schrock@Sun.COM 		}
2525367Sahrens 
2535367Sahrens 		verify(nvpair_value_nvlist(elem, &propnv) == 0);
25410195SSanjeev.Bagewadi@Sun.COM 		if (prop == ZFS_PROP_QUOTA || prop == ZFS_PROP_RESERVATION ||
25510195SSanjeev.Bagewadi@Sun.COM 		    prop == ZFS_PROP_REFQUOTA ||
25610195SSanjeev.Bagewadi@Sun.COM 		    prop == ZFS_PROP_REFRESERVATION) {
2575367Sahrens 			/* these guys are modifyable, but have no source */
2585367Sahrens 			uint64_t value;
2595367Sahrens 			verify(nvlist_lookup_uint64(propnv,
2605367Sahrens 			    ZPROP_VALUE, &value) == 0);
2617265Sahrens 			if (zhp->zfs_type == ZFS_TYPE_SNAPSHOT)
2627265Sahrens 				continue;
2635367Sahrens 		} else {
2645367Sahrens 			char *source;
2655367Sahrens 			if (nvlist_lookup_string(propnv,
2665367Sahrens 			    ZPROP_SOURCE, &source) != 0)
2675367Sahrens 				continue;
2685367Sahrens 			if (strcmp(source, zhp->zfs_name) != 0)
2695367Sahrens 				continue;
2705367Sahrens 		}
2715367Sahrens 
2725367Sahrens 		if (zfs_prop_user(propname) ||
2735367Sahrens 		    zfs_prop_get_type(prop) == PROP_TYPE_STRING) {
2745367Sahrens 			char *value;
2755367Sahrens 			verify(nvlist_lookup_string(propnv,
2765367Sahrens 			    ZPROP_VALUE, &value) == 0);
2775367Sahrens 			VERIFY(0 == nvlist_add_string(nv, propname, value));
2785367Sahrens 		} else {
2795367Sahrens 			uint64_t value;
2805367Sahrens 			verify(nvlist_lookup_uint64(propnv,
2815367Sahrens 			    ZPROP_VALUE, &value) == 0);
2825367Sahrens 			VERIFY(0 == nvlist_add_uint64(nv, propname, value));
2835367Sahrens 		}
2845367Sahrens 	}
2855367Sahrens }
2865367Sahrens 
28710200SLori.Alt@Sun.COM /*
28810200SLori.Alt@Sun.COM  * recursively generate nvlists describing datasets.  See comment
28910200SLori.Alt@Sun.COM  * for the data structure send_data_t above for description of contents
29010200SLori.Alt@Sun.COM  * of the nvlist.
29110200SLori.Alt@Sun.COM  */
2925367Sahrens static int
2935367Sahrens send_iterate_fs(zfs_handle_t *zhp, void *arg)
2945367Sahrens {
2955367Sahrens 	send_data_t *sd = arg;
2965367Sahrens 	nvlist_t *nvfs, *nv;
2975367Sahrens 	int rv;
2985367Sahrens 	uint64_t parent_fromsnap_guid_save = sd->parent_fromsnap_guid;
2995367Sahrens 	uint64_t guid = zhp->zfs_dmustats.dds_guid;
3005367Sahrens 	char guidstring[64];
3015367Sahrens 
3025367Sahrens 	VERIFY(0 == nvlist_alloc(&nvfs, NV_UNIQUE_NAME, 0));
3035367Sahrens 	VERIFY(0 == nvlist_add_string(nvfs, "name", zhp->zfs_name));
3045367Sahrens 	VERIFY(0 == nvlist_add_uint64(nvfs, "parentfromsnap",
3055367Sahrens 	    sd->parent_fromsnap_guid));
3065367Sahrens 
3075367Sahrens 	if (zhp->zfs_dmustats.dds_origin[0]) {
3085367Sahrens 		zfs_handle_t *origin = zfs_open(zhp->zfs_hdl,
3095367Sahrens 		    zhp->zfs_dmustats.dds_origin, ZFS_TYPE_SNAPSHOT);
3105367Sahrens 		if (origin == NULL)
3115367Sahrens 			return (-1);
3125367Sahrens 		VERIFY(0 == nvlist_add_uint64(nvfs, "origin",
3135367Sahrens 		    origin->zfs_dmustats.dds_guid));
3145367Sahrens 	}
3155367Sahrens 
3165367Sahrens 	/* iterate over props */
3175367Sahrens 	VERIFY(0 == nvlist_alloc(&nv, NV_UNIQUE_NAME, 0));
3185367Sahrens 	send_iterate_prop(zhp, nv);
3195367Sahrens 	VERIFY(0 == nvlist_add_nvlist(nvfs, "props", nv));
3205367Sahrens 	nvlist_free(nv);
3215367Sahrens 
3225367Sahrens 	/* iterate over snaps, and set sd->parent_fromsnap_guid */
3235367Sahrens 	sd->parent_fromsnap_guid = 0;
3245367Sahrens 	VERIFY(0 == nvlist_alloc(&sd->parent_snaps, NV_UNIQUE_NAME, 0));
3257265Sahrens 	VERIFY(0 == nvlist_alloc(&sd->snapprops, NV_UNIQUE_NAME, 0));
3265367Sahrens 	(void) zfs_iter_snapshots(zhp, send_iterate_snap, sd);
3275367Sahrens 	VERIFY(0 == nvlist_add_nvlist(nvfs, "snaps", sd->parent_snaps));
3287265Sahrens 	VERIFY(0 == nvlist_add_nvlist(nvfs, "snapprops", sd->snapprops));
3295367Sahrens 	nvlist_free(sd->parent_snaps);
3307265Sahrens 	nvlist_free(sd->snapprops);
3315367Sahrens 
3325367Sahrens 	/* add this fs to nvlist */
3335367Sahrens 	(void) snprintf(guidstring, sizeof (guidstring),
3345367Sahrens 	    "0x%llx", (longlong_t)guid);
3355367Sahrens 	VERIFY(0 == nvlist_add_nvlist(sd->fss, guidstring, nvfs));
3365367Sahrens 	nvlist_free(nvfs);
3375367Sahrens 
3385367Sahrens 	/* iterate over children */
3395367Sahrens 	rv = zfs_iter_filesystems(zhp, send_iterate_fs, sd);
3405367Sahrens 
3415367Sahrens 	sd->parent_fromsnap_guid = parent_fromsnap_guid_save;
3425367Sahrens 
3435367Sahrens 	zfs_close(zhp);
3445367Sahrens 	return (rv);
3455367Sahrens }
3465367Sahrens 
3475367Sahrens static int
3485367Sahrens gather_nvlist(libzfs_handle_t *hdl, const char *fsname, const char *fromsnap,
3495367Sahrens     const char *tosnap, nvlist_t **nvlp, avl_tree_t **avlp)
3505367Sahrens {
3515367Sahrens 	zfs_handle_t *zhp;
3525367Sahrens 	send_data_t sd = { 0 };
3535367Sahrens 	int error;
3545367Sahrens 
3555367Sahrens 	zhp = zfs_open(hdl, fsname, ZFS_TYPE_FILESYSTEM | ZFS_TYPE_VOLUME);
3565367Sahrens 	if (zhp == NULL)
3575367Sahrens 		return (EZFS_BADTYPE);
3585367Sahrens 
3595367Sahrens 	VERIFY(0 == nvlist_alloc(&sd.fss, NV_UNIQUE_NAME, 0));
3605367Sahrens 	sd.fromsnap = fromsnap;
3615367Sahrens 	sd.tosnap = tosnap;
3625896Sahl 
3635896Sahl 	if ((error = send_iterate_fs(zhp, &sd)) != 0) {
3645896Sahl 		nvlist_free(sd.fss);
3655896Sahl 		if (avlp != NULL)
3665896Sahl 			*avlp = NULL;
3675896Sahl 		*nvlp = NULL;
3685896Sahl 		return (error);
3695896Sahl 	}
3705896Sahl 
3715896Sahl 	if (avlp != NULL && (*avlp = fsavl_create(sd.fss)) == NULL) {
3725896Sahl 		nvlist_free(sd.fss);
3735896Sahl 		*nvlp = NULL;
3745896Sahl 		return (EZFS_NOMEM);
3755896Sahl 	}
3765367Sahrens 
3775367Sahrens 	*nvlp = sd.fss;
3785896Sahl 	return (0);
3795367Sahrens }
3805367Sahrens 
3815367Sahrens /*
3825367Sahrens  * Routines for dealing with the sorted snapshot functionality
3835367Sahrens  */
3845367Sahrens typedef struct zfs_node {
3855367Sahrens 	zfs_handle_t	*zn_handle;
3865367Sahrens 	avl_node_t	zn_avlnode;
3875367Sahrens } zfs_node_t;
3885367Sahrens 
3895367Sahrens static int
3905367Sahrens zfs_sort_snaps(zfs_handle_t *zhp, void *data)
3915367Sahrens {
3925367Sahrens 	avl_tree_t *avl = data;
3935367Sahrens 	zfs_node_t *node = zfs_alloc(zhp->zfs_hdl, sizeof (zfs_node_t));
3945367Sahrens 
3955367Sahrens 	node->zn_handle = zhp;
3965367Sahrens 	avl_add(avl, node);
3975367Sahrens 	return (0);
3985367Sahrens }
3995367Sahrens 
4005367Sahrens /* ARGSUSED */
4015367Sahrens static int
4025367Sahrens zfs_snapshot_compare(const void *larg, const void *rarg)
4035367Sahrens {
4045367Sahrens 	zfs_handle_t *l = ((zfs_node_t *)larg)->zn_handle;
4055367Sahrens 	zfs_handle_t *r = ((zfs_node_t *)rarg)->zn_handle;
4065367Sahrens 	uint64_t lcreate, rcreate;
4075367Sahrens 
4085367Sahrens 	/*
4095367Sahrens 	 * Sort them according to creation time.  We use the hidden
4105367Sahrens 	 * CREATETXG property to get an absolute ordering of snapshots.
4115367Sahrens 	 */
4125367Sahrens 	lcreate = zfs_prop_get_int(l, ZFS_PROP_CREATETXG);
4135367Sahrens 	rcreate = zfs_prop_get_int(r, ZFS_PROP_CREATETXG);
4145367Sahrens 
4155367Sahrens 	if (lcreate < rcreate)
4165367Sahrens 		return (-1);
4175367Sahrens 	else if (lcreate > rcreate)
4185367Sahrens 		return (+1);
4195367Sahrens 	else
4205367Sahrens 		return (0);
4215367Sahrens }
4225367Sahrens 
42310342Schris.kirby@sun.com int
4245367Sahrens zfs_iter_snapshots_sorted(zfs_handle_t *zhp, zfs_iter_f callback, void *data)
4255367Sahrens {
4265367Sahrens 	int ret = 0;
4275367Sahrens 	zfs_node_t *node;
4285367Sahrens 	avl_tree_t avl;
4295367Sahrens 	void *cookie = NULL;
4305367Sahrens 
4315367Sahrens 	avl_create(&avl, zfs_snapshot_compare,
4325367Sahrens 	    sizeof (zfs_node_t), offsetof(zfs_node_t, zn_avlnode));
4335367Sahrens 
4345367Sahrens 	ret = zfs_iter_snapshots(zhp, zfs_sort_snaps, &avl);
4355367Sahrens 
4365367Sahrens 	for (node = avl_first(&avl); node != NULL; node = AVL_NEXT(&avl, node))
4375367Sahrens 		ret |= callback(node->zn_handle, data);
4385367Sahrens 
4395367Sahrens 	while ((node = avl_destroy_nodes(&avl, &cookie)) != NULL)
4405367Sahrens 		free(node);
4415367Sahrens 
4425367Sahrens 	avl_destroy(&avl);
4435367Sahrens 
4445367Sahrens 	return (ret);
4455367Sahrens }
4465367Sahrens 
4475367Sahrens /*
4485367Sahrens  * Routines specific to "zfs send"
4495367Sahrens  */
4505367Sahrens typedef struct send_dump_data {
4515367Sahrens 	/* these are all just the short snapname (the part after the @) */
4525367Sahrens 	const char *fromsnap;
4535367Sahrens 	const char *tosnap;
4545367Sahrens 	char lastsnap[ZFS_MAXNAMELEN];
4555367Sahrens 	boolean_t seenfrom, seento, replicate, doall, fromorigin;
4565367Sahrens 	boolean_t verbose;
4575367Sahrens 	int outfd;
4585367Sahrens 	boolean_t err;
4595367Sahrens 	nvlist_t *fss;
4605367Sahrens 	avl_tree_t *fsavl;
4615367Sahrens } send_dump_data_t;
4625367Sahrens 
4635367Sahrens /*
4645367Sahrens  * Dumps a backup of the given snapshot (incremental from fromsnap if it's not
4655367Sahrens  * NULL) to the file descriptor specified by outfd.
4665367Sahrens  */
4675367Sahrens static int
4685367Sahrens dump_ioctl(zfs_handle_t *zhp, const char *fromsnap, boolean_t fromorigin,
4695367Sahrens     int outfd)
4705367Sahrens {
4715367Sahrens 	zfs_cmd_t zc = { 0 };
4725367Sahrens 	libzfs_handle_t *hdl = zhp->zfs_hdl;
4735367Sahrens 
4745367Sahrens 	assert(zhp->zfs_type == ZFS_TYPE_SNAPSHOT);
4755367Sahrens 	assert(fromsnap == NULL || fromsnap[0] == '\0' || !fromorigin);
4765367Sahrens 
4775367Sahrens 	(void) strlcpy(zc.zc_name, zhp->zfs_name, sizeof (zc.zc_name));
4785367Sahrens 	if (fromsnap)
4795765Sek110237 		(void) strlcpy(zc.zc_value, fromsnap, sizeof (zc.zc_value));
4805367Sahrens 	zc.zc_cookie = outfd;
4815367Sahrens 	zc.zc_obj = fromorigin;
4825367Sahrens 
4835367Sahrens 	if (ioctl(zhp->zfs_hdl->libzfs_fd, ZFS_IOC_SEND, &zc) != 0) {
4845367Sahrens 		char errbuf[1024];
4855367Sahrens 		(void) snprintf(errbuf, sizeof (errbuf), dgettext(TEXT_DOMAIN,
4865367Sahrens 		    "warning: cannot send '%s'"), zhp->zfs_name);
4875367Sahrens 
4885367Sahrens 		switch (errno) {
4895367Sahrens 
4905367Sahrens 		case EXDEV:
4915367Sahrens 			zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
4925367Sahrens 			    "not an earlier snapshot from the same fs"));
4935367Sahrens 			return (zfs_error(hdl, EZFS_CROSSTARGET, errbuf));
4945367Sahrens 
4955367Sahrens 		case ENOENT:
4965367Sahrens 			if (zfs_dataset_exists(hdl, zc.zc_name,
4975367Sahrens 			    ZFS_TYPE_SNAPSHOT)) {
4985367Sahrens 				zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
4995367Sahrens 				    "incremental source (@%s) does not exist"),
5005367Sahrens 				    zc.zc_value);
5015367Sahrens 			}
5025367Sahrens 			return (zfs_error(hdl, EZFS_NOENT, errbuf));
5035367Sahrens 
5045367Sahrens 		case EDQUOT:
5055367Sahrens 		case EFBIG:
5065367Sahrens 		case EIO:
5075367Sahrens 		case ENOLINK:
5085367Sahrens 		case ENOSPC:
5095367Sahrens 		case ENOSTR:
5105367Sahrens 		case ENXIO:
5115367Sahrens 		case EPIPE:
5125367Sahrens 		case ERANGE:
5135367Sahrens 		case EFAULT:
5145367Sahrens 		case EROFS:
5155367Sahrens 			zfs_error_aux(hdl, strerror(errno));
5165367Sahrens 			return (zfs_error(hdl, EZFS_BADBACKUP, errbuf));
5175367Sahrens 
5185367Sahrens 		default:
5195367Sahrens 			return (zfs_standard_error(hdl, errno, errbuf));
5205367Sahrens 		}
5215367Sahrens 	}
5225367Sahrens 
5235367Sahrens 	return (0);
5245367Sahrens }
5255367Sahrens 
5265367Sahrens static int
5275367Sahrens dump_snapshot(zfs_handle_t *zhp, void *arg)
5285367Sahrens {
5295367Sahrens 	send_dump_data_t *sdd = arg;
5305367Sahrens 	const char *thissnap;
5315367Sahrens 	int err;
5325367Sahrens 
5335367Sahrens 	thissnap = strchr(zhp->zfs_name, '@') + 1;
5345367Sahrens 
5355367Sahrens 	if (sdd->fromsnap && !sdd->seenfrom &&
5365367Sahrens 	    strcmp(sdd->fromsnap, thissnap) == 0) {
5375367Sahrens 		sdd->seenfrom = B_TRUE;
5385367Sahrens 		(void) strcpy(sdd->lastsnap, thissnap);
5395367Sahrens 		zfs_close(zhp);
5405367Sahrens 		return (0);
5415367Sahrens 	}
5425367Sahrens 
5435367Sahrens 	if (sdd->seento || !sdd->seenfrom) {
5445367Sahrens 		zfs_close(zhp);
5455367Sahrens 		return (0);
5465367Sahrens 	}
5475367Sahrens 
5485367Sahrens 	/* send it */
5495367Sahrens 	if (sdd->verbose) {
5505367Sahrens 		(void) fprintf(stderr, "sending from @%s to %s\n",
5515367Sahrens 		    sdd->lastsnap, zhp->zfs_name);
5525367Sahrens 	}
5535367Sahrens 
5545367Sahrens 	err = dump_ioctl(zhp, sdd->lastsnap,
5555367Sahrens 	    sdd->lastsnap[0] == '\0' && (sdd->fromorigin || sdd->replicate),
5565367Sahrens 	    sdd->outfd);
5575367Sahrens 
5585367Sahrens 	if (!sdd->seento && strcmp(sdd->tosnap, thissnap) == 0)
5595367Sahrens 		sdd->seento = B_TRUE;
5605367Sahrens 
5615367Sahrens 	(void) strcpy(sdd->lastsnap, thissnap);
5625367Sahrens 	zfs_close(zhp);
5635367Sahrens 	return (err);
5645367Sahrens }
5655367Sahrens 
5665367Sahrens static int
5675367Sahrens dump_filesystem(zfs_handle_t *zhp, void *arg)
5685367Sahrens {
5695367Sahrens 	int rv = 0;
5705367Sahrens 	send_dump_data_t *sdd = arg;
5715367Sahrens 	boolean_t missingfrom = B_FALSE;
5725367Sahrens 	zfs_cmd_t zc = { 0 };
5735367Sahrens 
5745367Sahrens 	(void) snprintf(zc.zc_name, sizeof (zc.zc_name), "%s@%s",
5755367Sahrens 	    zhp->zfs_name, sdd->tosnap);
5765367Sahrens 	if (ioctl(zhp->zfs_hdl->libzfs_fd, ZFS_IOC_OBJSET_STATS, &zc) != 0) {
5775367Sahrens 		(void) fprintf(stderr, "WARNING: "
5785367Sahrens 		    "could not send %s@%s: does not exist\n",
5795367Sahrens 		    zhp->zfs_name, sdd->tosnap);
5805367Sahrens 		sdd->err = B_TRUE;
5815367Sahrens 		return (0);
5825367Sahrens 	}
5835367Sahrens 
5845367Sahrens 	if (sdd->replicate && sdd->fromsnap) {
5855367Sahrens 		/*
5865367Sahrens 		 * If this fs does not have fromsnap, and we're doing
5875367Sahrens 		 * recursive, we need to send a full stream from the
5885367Sahrens 		 * beginning (or an incremental from the origin if this
5895367Sahrens 		 * is a clone).  If we're doing non-recursive, then let
5905367Sahrens 		 * them get the error.
5915367Sahrens 		 */
5925367Sahrens 		(void) snprintf(zc.zc_name, sizeof (zc.zc_name), "%s@%s",
5935367Sahrens 		    zhp->zfs_name, sdd->fromsnap);
5945367Sahrens 		if (ioctl(zhp->zfs_hdl->libzfs_fd,
5955367Sahrens 		    ZFS_IOC_OBJSET_STATS, &zc) != 0) {
5965367Sahrens 			missingfrom = B_TRUE;
5975367Sahrens 		}
5985367Sahrens 	}
5995367Sahrens 
6005367Sahrens 	if (sdd->doall) {
6015367Sahrens 		sdd->seenfrom = sdd->seento = sdd->lastsnap[0] = 0;
6025367Sahrens 		if (sdd->fromsnap == NULL || missingfrom)
6035367Sahrens 			sdd->seenfrom = B_TRUE;
6045367Sahrens 
6055367Sahrens 		rv = zfs_iter_snapshots_sorted(zhp, dump_snapshot, arg);
6065367Sahrens 		if (!sdd->seenfrom) {
6075367Sahrens 			(void) fprintf(stderr,
6085367Sahrens 			    "WARNING: could not send %s@%s:\n"
6095367Sahrens 			    "incremental source (%s@%s) does not exist\n",
6105367Sahrens 			    zhp->zfs_name, sdd->tosnap,
6115367Sahrens 			    zhp->zfs_name, sdd->fromsnap);
6125367Sahrens 			sdd->err = B_TRUE;
6135367Sahrens 		} else if (!sdd->seento) {
6149396SMatthew.Ahrens@Sun.COM 			if (sdd->fromsnap) {
6159396SMatthew.Ahrens@Sun.COM 				(void) fprintf(stderr,
6169396SMatthew.Ahrens@Sun.COM 				    "WARNING: could not send %s@%s:\n"
6179396SMatthew.Ahrens@Sun.COM 				    "incremental source (%s@%s) "
6189396SMatthew.Ahrens@Sun.COM 				    "is not earlier than it\n",
6199396SMatthew.Ahrens@Sun.COM 				    zhp->zfs_name, sdd->tosnap,
6209396SMatthew.Ahrens@Sun.COM 				    zhp->zfs_name, sdd->fromsnap);
6219396SMatthew.Ahrens@Sun.COM 			} else {
6229396SMatthew.Ahrens@Sun.COM 				(void) fprintf(stderr, "WARNING: "
6239396SMatthew.Ahrens@Sun.COM 				    "could not send %s@%s: does not exist\n",
6249396SMatthew.Ahrens@Sun.COM 				    zhp->zfs_name, sdd->tosnap);
6259396SMatthew.Ahrens@Sun.COM 			}
6265367Sahrens 			sdd->err = B_TRUE;
6275367Sahrens 		}
6285367Sahrens 	} else {
6295367Sahrens 		zfs_handle_t *snapzhp;
6305367Sahrens 		char snapname[ZFS_MAXNAMELEN];
6315367Sahrens 
6325367Sahrens 		(void) snprintf(snapname, sizeof (snapname), "%s@%s",
6335367Sahrens 		    zfs_get_name(zhp), sdd->tosnap);
6345367Sahrens 		snapzhp = zfs_open(zhp->zfs_hdl, snapname, ZFS_TYPE_SNAPSHOT);
6356476Sahl 		if (snapzhp == NULL) {
6366476Sahl 			rv = -1;
6376476Sahl 		} else {
6386476Sahl 			rv = dump_ioctl(snapzhp,
6396476Sahl 			    missingfrom ? NULL : sdd->fromsnap,
6406476Sahl 			    sdd->fromorigin || missingfrom,
6416476Sahl 			    sdd->outfd);
6426476Sahl 			sdd->seento = B_TRUE;
6436476Sahl 			zfs_close(snapzhp);
6446476Sahl 		}
6455367Sahrens 	}
6465367Sahrens 
6475367Sahrens 	return (rv);
6485367Sahrens }
6495367Sahrens 
6505367Sahrens static int
6515367Sahrens dump_filesystems(zfs_handle_t *rzhp, void *arg)
6525367Sahrens {
6535367Sahrens 	send_dump_data_t *sdd = arg;
6545367Sahrens 	nvpair_t *fspair;
6555367Sahrens 	boolean_t needagain, progress;
6565367Sahrens 
6575367Sahrens 	if (!sdd->replicate)
6585367Sahrens 		return (dump_filesystem(rzhp, sdd));
6595367Sahrens 
6605367Sahrens again:
6615367Sahrens 	needagain = progress = B_FALSE;
6625367Sahrens 	for (fspair = nvlist_next_nvpair(sdd->fss, NULL); fspair;
6635367Sahrens 	    fspair = nvlist_next_nvpair(sdd->fss, fspair)) {
6645367Sahrens 		nvlist_t *fslist;
6655367Sahrens 		char *fsname;
6665367Sahrens 		zfs_handle_t *zhp;
6675367Sahrens 		int err;
6685367Sahrens 		uint64_t origin_guid = 0;
6695367Sahrens 		nvlist_t *origin_nv;
6705367Sahrens 
6715367Sahrens 		VERIFY(nvpair_value_nvlist(fspair, &fslist) == 0);
6725367Sahrens 		if (nvlist_lookup_boolean(fslist, "sent") == 0)
6735367Sahrens 			continue;
6745367Sahrens 
6755367Sahrens 		VERIFY(nvlist_lookup_string(fslist, "name", &fsname) == 0);
6765367Sahrens 		(void) nvlist_lookup_uint64(fslist, "origin", &origin_guid);
6775367Sahrens 
6785367Sahrens 		origin_nv = fsavl_find(sdd->fsavl, origin_guid, NULL);
6795367Sahrens 		if (origin_nv &&
6805367Sahrens 		    nvlist_lookup_boolean(origin_nv, "sent") == ENOENT) {
6815367Sahrens 			/*
6825367Sahrens 			 * origin has not been sent yet;
6835367Sahrens 			 * skip this clone.
6845367Sahrens 			 */
6855367Sahrens 			needagain = B_TRUE;
6865367Sahrens 			continue;
6875367Sahrens 		}
6885367Sahrens 
6895367Sahrens 		zhp = zfs_open(rzhp->zfs_hdl, fsname, ZFS_TYPE_DATASET);
6906476Sahl 		if (zhp == NULL)
6916476Sahl 			return (-1);
6925367Sahrens 		err = dump_filesystem(zhp, sdd);
6935367Sahrens 		VERIFY(nvlist_add_boolean(fslist, "sent") == 0);
6945367Sahrens 		progress = B_TRUE;
6955367Sahrens 		zfs_close(zhp);
6965367Sahrens 		if (err)
6975367Sahrens 			return (err);
6985367Sahrens 	}
6995367Sahrens 	if (needagain) {
7005367Sahrens 		assert(progress);
7015367Sahrens 		goto again;
7025367Sahrens 	}
7035367Sahrens 	return (0);
7045367Sahrens }
7055367Sahrens 
7065367Sahrens /*
70710200SLori.Alt@Sun.COM  * Generate a send stream for the dataset identified by the argument zhp.
70810200SLori.Alt@Sun.COM  *
70910200SLori.Alt@Sun.COM  * The content of the send stream is the snapshot identified by
71010200SLori.Alt@Sun.COM  * 'tosnap'.  Incremental streams are requested in two ways:
71110200SLori.Alt@Sun.COM  *     - from the snapshot identified by "fromsnap" (if non-null) or
71210200SLori.Alt@Sun.COM  *     - from the origin of the dataset identified by zhp, which must
71310200SLori.Alt@Sun.COM  *	 be a clone.  In this case, "fromsnap" is null and "fromorigin"
71410200SLori.Alt@Sun.COM  *	 is TRUE.
71510200SLori.Alt@Sun.COM  *
71610200SLori.Alt@Sun.COM  * The send stream is recursive (i.e. dumps a hierarchy of snapshots) and
71710200SLori.Alt@Sun.COM  * uses a special header (with a version field of DMU_BACKUP_HEADER_VERSION)
71810200SLori.Alt@Sun.COM  * if "replicate" is set.  If "doall" is set, dump all the intermediate
71910200SLori.Alt@Sun.COM  * snapshots. The DMU_BACKUP_HEADER_VERSION header is used in the "doall"
72010200SLori.Alt@Sun.COM  * case too.
7215367Sahrens  */
7225367Sahrens int
7235367Sahrens zfs_send(zfs_handle_t *zhp, const char *fromsnap, const char *tosnap,
7245367Sahrens     boolean_t replicate, boolean_t doall, boolean_t fromorigin,
7255367Sahrens     boolean_t verbose, int outfd)
7265367Sahrens {
7275367Sahrens 	char errbuf[1024];
7285367Sahrens 	send_dump_data_t sdd = { 0 };
7295367Sahrens 	int err;
7305367Sahrens 	nvlist_t *fss = NULL;
7315367Sahrens 	avl_tree_t *fsavl = NULL;
73210342Schris.kirby@sun.com 	char holdtag[128];
73310342Schris.kirby@sun.com 	static uint64_t holdseq;
73410735Schris.kirby@sun.com 	int spa_version;
73510735Schris.kirby@sun.com 	boolean_t holdsnaps = B_FALSE;
7365367Sahrens 
7375367Sahrens 	(void) snprintf(errbuf, sizeof (errbuf), dgettext(TEXT_DOMAIN,
7385367Sahrens 	    "cannot send '%s'"), zhp->zfs_name);
7395367Sahrens 
7405367Sahrens 	if (fromsnap && fromsnap[0] == '\0') {
7415367Sahrens 		zfs_error_aux(zhp->zfs_hdl, dgettext(TEXT_DOMAIN,
7425367Sahrens 		    "zero-length incremental source"));
7435367Sahrens 		return (zfs_error(zhp->zfs_hdl, EZFS_NOENT, errbuf));
7445367Sahrens 	}
7455367Sahrens 
74610735Schris.kirby@sun.com 	if (zfs_spa_version(zhp, &spa_version) == 0 &&
74710735Schris.kirby@sun.com 	    spa_version >= SPA_VERSION_USERREFS)
74810735Schris.kirby@sun.com 		holdsnaps = B_TRUE;
74910735Schris.kirby@sun.com 
7505367Sahrens 	if (replicate || doall) {
7515367Sahrens 		dmu_replay_record_t drr = { 0 };
7525367Sahrens 		char *packbuf = NULL;
7535367Sahrens 		size_t buflen = 0;
7545367Sahrens 		zio_cksum_t zc = { 0 };
7555367Sahrens 
7565367Sahrens 		assert(fromsnap || doall);
7575367Sahrens 
75810735Schris.kirby@sun.com 		if (holdsnaps) {
75910735Schris.kirby@sun.com 			(void) snprintf(holdtag, sizeof (holdtag),
76010735Schris.kirby@sun.com 			    ".send-%d-%llu", getpid(), (u_longlong_t)holdseq);
76110735Schris.kirby@sun.com 			++holdseq;
76210735Schris.kirby@sun.com 			err = zfs_hold_range(zhp, fromsnap, tosnap,
76310735Schris.kirby@sun.com 			    holdtag, B_TRUE);
76410735Schris.kirby@sun.com 			if (err)
76510735Schris.kirby@sun.com 				return (err);
76610735Schris.kirby@sun.com 		}
7675367Sahrens 		if (replicate) {
7685367Sahrens 			nvlist_t *hdrnv;
7695367Sahrens 
7705367Sahrens 			VERIFY(0 == nvlist_alloc(&hdrnv, NV_UNIQUE_NAME, 0));
7715367Sahrens 			if (fromsnap) {
7725367Sahrens 				VERIFY(0 == nvlist_add_string(hdrnv,
7735367Sahrens 				    "fromsnap", fromsnap));
7745367Sahrens 			}
7755367Sahrens 			VERIFY(0 == nvlist_add_string(hdrnv, "tosnap", tosnap));
7765367Sahrens 
7775367Sahrens 			err = gather_nvlist(zhp->zfs_hdl, zhp->zfs_name,
7785367Sahrens 			    fromsnap, tosnap, &fss, &fsavl);
77910342Schris.kirby@sun.com 			if (err) {
78010735Schris.kirby@sun.com 				if (holdsnaps) {
78110735Schris.kirby@sun.com 					(void) zfs_release_range(zhp, fromsnap,
78210735Schris.kirby@sun.com 					    tosnap, holdtag);
78310735Schris.kirby@sun.com 				}
7845367Sahrens 				return (err);
78510342Schris.kirby@sun.com 			}
7865367Sahrens 			VERIFY(0 == nvlist_add_nvlist(hdrnv, "fss", fss));
7875367Sahrens 			err = nvlist_pack(hdrnv, &packbuf, &buflen,
7885367Sahrens 			    NV_ENCODE_XDR, 0);
7895367Sahrens 			nvlist_free(hdrnv);
7905367Sahrens 			if (err) {
7915367Sahrens 				fsavl_destroy(fsavl);
7925367Sahrens 				nvlist_free(fss);
79310735Schris.kirby@sun.com 				if (holdsnaps) {
79410735Schris.kirby@sun.com 					(void) zfs_release_range(zhp, fromsnap,
79510735Schris.kirby@sun.com 					    tosnap, holdtag);
79610735Schris.kirby@sun.com 				}
7975367Sahrens 				return (zfs_standard_error(zhp->zfs_hdl,
7985367Sahrens 				    err, errbuf));
7995367Sahrens 			}
8005367Sahrens 		}
8015367Sahrens 
8025367Sahrens 		/* write first begin record */
8035367Sahrens 		drr.drr_type = DRR_BEGIN;
8045367Sahrens 		drr.drr_u.drr_begin.drr_magic = DMU_BACKUP_MAGIC;
8055367Sahrens 		drr.drr_u.drr_begin.drr_version = DMU_BACKUP_HEADER_VERSION;
8065367Sahrens 		(void) snprintf(drr.drr_u.drr_begin.drr_toname,
8075367Sahrens 		    sizeof (drr.drr_u.drr_begin.drr_toname),
8085367Sahrens 		    "%s@%s", zhp->zfs_name, tosnap);
8095367Sahrens 		drr.drr_payloadlen = buflen;
8105367Sahrens 		fletcher_4_incremental_native(&drr, sizeof (drr), &zc);
8115367Sahrens 		err = write(outfd, &drr, sizeof (drr));
8125367Sahrens 
8135367Sahrens 		/* write header nvlist */
8145367Sahrens 		if (err != -1) {
8155367Sahrens 			fletcher_4_incremental_native(packbuf, buflen, &zc);
8165367Sahrens 			err = write(outfd, packbuf, buflen);
8175367Sahrens 		}
8185367Sahrens 		free(packbuf);
8195367Sahrens 		if (err == -1) {
8205367Sahrens 			fsavl_destroy(fsavl);
8215367Sahrens 			nvlist_free(fss);
82210735Schris.kirby@sun.com 			if (holdsnaps) {
82310735Schris.kirby@sun.com 				(void) zfs_release_range(zhp, fromsnap, tosnap,
82410735Schris.kirby@sun.com 				    holdtag);
82510735Schris.kirby@sun.com 			}
8265367Sahrens 			return (zfs_standard_error(zhp->zfs_hdl,
8275367Sahrens 			    errno, errbuf));
8285367Sahrens 		}
8295367Sahrens 
8305367Sahrens 		/* write end record */
8315367Sahrens 		if (err != -1) {
8325367Sahrens 			bzero(&drr, sizeof (drr));
8335367Sahrens 			drr.drr_type = DRR_END;
8345367Sahrens 			drr.drr_u.drr_end.drr_checksum = zc;
8355367Sahrens 			err = write(outfd, &drr, sizeof (drr));
8365367Sahrens 			if (err == -1) {
8375367Sahrens 				fsavl_destroy(fsavl);
8385367Sahrens 				nvlist_free(fss);
83910735Schris.kirby@sun.com 				if (holdsnaps) {
84010735Schris.kirby@sun.com 					(void) zfs_release_range(zhp, fromsnap,
84110735Schris.kirby@sun.com 					    tosnap, holdtag);
84210735Schris.kirby@sun.com 				}
8435367Sahrens 				return (zfs_standard_error(zhp->zfs_hdl,
8445367Sahrens 				    errno, errbuf));
8455367Sahrens 			}
8465367Sahrens 		}
8475367Sahrens 	}
8485367Sahrens 
8495367Sahrens 	/* dump each stream */
8505367Sahrens 	sdd.fromsnap = fromsnap;
8515367Sahrens 	sdd.tosnap = tosnap;
8525367Sahrens 	sdd.outfd = outfd;
8535367Sahrens 	sdd.replicate = replicate;
8545367Sahrens 	sdd.doall = doall;
8555367Sahrens 	sdd.fromorigin = fromorigin;
8565367Sahrens 	sdd.fss = fss;
8575367Sahrens 	sdd.fsavl = fsavl;
8585367Sahrens 	sdd.verbose = verbose;
8595367Sahrens 	err = dump_filesystems(zhp, &sdd);
8605367Sahrens 	fsavl_destroy(fsavl);
8615367Sahrens 	nvlist_free(fss);
8625367Sahrens 
8635367Sahrens 	if (replicate || doall) {
8645367Sahrens 		/*
8655367Sahrens 		 * write final end record.  NB: want to do this even if
8665367Sahrens 		 * there was some error, because it might not be totally
8675367Sahrens 		 * failed.
8685367Sahrens 		 */
8695367Sahrens 		dmu_replay_record_t drr = { 0 };
8705367Sahrens 		drr.drr_type = DRR_END;
87110735Schris.kirby@sun.com 		if (holdsnaps) {
87210735Schris.kirby@sun.com 			(void) zfs_release_range(zhp, fromsnap, tosnap,
87310735Schris.kirby@sun.com 			    holdtag);
87410735Schris.kirby@sun.com 		}
8755367Sahrens 		if (write(outfd, &drr, sizeof (drr)) == -1) {
8765367Sahrens 			return (zfs_standard_error(zhp->zfs_hdl,
8775367Sahrens 			    errno, errbuf));
8785367Sahrens 		}
8795367Sahrens 	}
8805367Sahrens 
8815367Sahrens 	return (err || sdd.err);
8825367Sahrens }
8835367Sahrens 
8845367Sahrens /*
8855367Sahrens  * Routines specific to "zfs recv"
8865367Sahrens  */
8875367Sahrens 
8885367Sahrens static int
8895367Sahrens recv_read(libzfs_handle_t *hdl, int fd, void *buf, int ilen,
8905367Sahrens     boolean_t byteswap, zio_cksum_t *zc)
8915367Sahrens {
8925367Sahrens 	char *cp = buf;
8935367Sahrens 	int rv;
8945367Sahrens 	int len = ilen;
8955367Sahrens 
8965367Sahrens 	do {
8975367Sahrens 		rv = read(fd, cp, len);
8985367Sahrens 		cp += rv;
8995367Sahrens 		len -= rv;
9005367Sahrens 	} while (rv > 0);
9015367Sahrens 
9025367Sahrens 	if (rv < 0 || len != 0) {
9035367Sahrens 		zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
9045367Sahrens 		    "failed to read from stream"));
9055367Sahrens 		return (zfs_error(hdl, EZFS_BADSTREAM, dgettext(TEXT_DOMAIN,
9065367Sahrens 		    "cannot receive")));
9075367Sahrens 	}
9085367Sahrens 
9095367Sahrens 	if (zc) {
9105367Sahrens 		if (byteswap)
9115367Sahrens 			fletcher_4_incremental_byteswap(buf, ilen, zc);
9125367Sahrens 		else
9135367Sahrens 			fletcher_4_incremental_native(buf, ilen, zc);
9145367Sahrens 	}
9155367Sahrens 	return (0);
9165367Sahrens }
9175367Sahrens 
9185367Sahrens static int
9195367Sahrens recv_read_nvlist(libzfs_handle_t *hdl, int fd, int len, nvlist_t **nvp,
9205367Sahrens     boolean_t byteswap, zio_cksum_t *zc)
9215367Sahrens {
9225367Sahrens 	char *buf;
9235367Sahrens 	int err;
9245367Sahrens 
9255367Sahrens 	buf = zfs_alloc(hdl, len);
9265367Sahrens 	if (buf == NULL)
9275367Sahrens 		return (ENOMEM);
9285367Sahrens 
9295367Sahrens 	err = recv_read(hdl, fd, buf, len, byteswap, zc);
9305367Sahrens 	if (err != 0) {
9315367Sahrens 		free(buf);
9325367Sahrens 		return (err);
9335367Sahrens 	}
9345367Sahrens 
9355367Sahrens 	err = nvlist_unpack(buf, len, nvp, 0);
9365367Sahrens 	free(buf);
9375367Sahrens 	if (err != 0) {
9385367Sahrens 		zfs_error_aux(hdl, dgettext(TEXT_DOMAIN, "invalid "
9395367Sahrens 		    "stream (malformed nvlist)"));
9405367Sahrens 		return (EINVAL);
9415367Sahrens 	}
9425367Sahrens 	return (0);
9435367Sahrens }
9445367Sahrens 
9455367Sahrens static int
9465367Sahrens recv_rename(libzfs_handle_t *hdl, const char *name, const char *tryname,
9475367Sahrens     int baselen, char *newname, recvflags_t flags)
9485367Sahrens {
9495367Sahrens 	static int seq;
9505367Sahrens 	zfs_cmd_t zc = { 0 };
9515367Sahrens 	int err;
9526047Sahrens 	prop_changelist_t *clp;
9536047Sahrens 	zfs_handle_t *zhp;
9545367Sahrens 
9556047Sahrens 	zhp = zfs_open(hdl, name, ZFS_TYPE_DATASET);
9566047Sahrens 	if (zhp == NULL)
9576047Sahrens 		return (-1);
9587366STim.Haley@Sun.COM 	clp = changelist_gather(zhp, ZFS_PROP_NAME, 0,
9597366STim.Haley@Sun.COM 	    flags.force ? MS_FORCE : 0);
9606047Sahrens 	zfs_close(zhp);
9616047Sahrens 	if (clp == NULL)
9626047Sahrens 		return (-1);
9636047Sahrens 	err = changelist_prefix(clp);
9646047Sahrens 	if (err)
9656047Sahrens 		return (err);
9665367Sahrens 
96710242Schris.kirby@sun.com 	zc.zc_objset_type = DMU_OST_ZFS;
96810242Schris.kirby@sun.com 	(void) strlcpy(zc.zc_name, name, sizeof (zc.zc_name));
96910242Schris.kirby@sun.com 
9705367Sahrens 	if (tryname) {
9715367Sahrens 		(void) strcpy(newname, tryname);
9725367Sahrens 
9735367Sahrens 		(void) strlcpy(zc.zc_value, tryname, sizeof (zc.zc_value));
9745367Sahrens 
9755367Sahrens 		if (flags.verbose) {
9765367Sahrens 			(void) printf("attempting rename %s to %s\n",
9775367Sahrens 			    zc.zc_name, zc.zc_value);
9785367Sahrens 		}
9795367Sahrens 		err = ioctl(hdl->libzfs_fd, ZFS_IOC_RENAME, &zc);
9806047Sahrens 		if (err == 0)
9815367Sahrens 			changelist_rename(clp, name, tryname);
9825367Sahrens 	} else {
9835367Sahrens 		err = ENOENT;
9845367Sahrens 	}
9855367Sahrens 
9865367Sahrens 	if (err != 0 && strncmp(name+baselen, "recv-", 5) != 0) {
9875367Sahrens 		seq++;
9885367Sahrens 
9895367Sahrens 		(void) strncpy(newname, name, baselen);
9905367Sahrens 		(void) snprintf(newname+baselen, ZFS_MAXNAMELEN-baselen,
9915367Sahrens 		    "recv-%u-%u", getpid(), seq);
9925367Sahrens 		(void) strlcpy(zc.zc_value, newname, sizeof (zc.zc_value));
9935367Sahrens 
9945367Sahrens 		if (flags.verbose) {
9955367Sahrens 			(void) printf("failed - trying rename %s to %s\n",
9965367Sahrens 			    zc.zc_name, zc.zc_value);
9975367Sahrens 		}
9985367Sahrens 		err = ioctl(hdl->libzfs_fd, ZFS_IOC_RENAME, &zc);
9996047Sahrens 		if (err == 0)
10005367Sahrens 			changelist_rename(clp, name, newname);
10015367Sahrens 		if (err && flags.verbose) {
10025367Sahrens 			(void) printf("failed (%u) - "
10035367Sahrens 			    "will try again on next pass\n", errno);
10045367Sahrens 		}
10055367Sahrens 		err = EAGAIN;
10065367Sahrens 	} else if (flags.verbose) {
10075367Sahrens 		if (err == 0)
10085367Sahrens 			(void) printf("success\n");
10095367Sahrens 		else
10105367Sahrens 			(void) printf("failed (%u)\n", errno);
10115367Sahrens 	}
10125367Sahrens 
10136047Sahrens 	(void) changelist_postfix(clp);
10146047Sahrens 	changelist_free(clp);
10155367Sahrens 
10165367Sahrens 	return (err);
10175367Sahrens }
10185367Sahrens 
10195367Sahrens static int
10205367Sahrens recv_destroy(libzfs_handle_t *hdl, const char *name, int baselen,
10215367Sahrens     char *newname, recvflags_t flags)
10225367Sahrens {
10235367Sahrens 	zfs_cmd_t zc = { 0 };
10246047Sahrens 	int err = 0;
10256047Sahrens 	prop_changelist_t *clp;
10266047Sahrens 	zfs_handle_t *zhp;
102710242Schris.kirby@sun.com 	boolean_t defer = B_FALSE;
102810242Schris.kirby@sun.com 	int spa_version;
10296047Sahrens 
10306047Sahrens 	zhp = zfs_open(hdl, name, ZFS_TYPE_DATASET);
10316047Sahrens 	if (zhp == NULL)
10326047Sahrens 		return (-1);
10337366STim.Haley@Sun.COM 	clp = changelist_gather(zhp, ZFS_PROP_NAME, 0,
10347366STim.Haley@Sun.COM 	    flags.force ? MS_FORCE : 0);
103510242Schris.kirby@sun.com 	if (zfs_get_type(zhp) == ZFS_TYPE_SNAPSHOT &&
103610242Schris.kirby@sun.com 	    zfs_spa_version(zhp, &spa_version) == 0 &&
103710242Schris.kirby@sun.com 	    spa_version >= SPA_VERSION_USERREFS)
103810242Schris.kirby@sun.com 		defer = B_TRUE;
10396047Sahrens 	zfs_close(zhp);
10406047Sahrens 	if (clp == NULL)
10416047Sahrens 		return (-1);
10426047Sahrens 	err = changelist_prefix(clp);
10436047Sahrens 	if (err)
10446047Sahrens 		return (err);
10455367Sahrens 
10465367Sahrens 	zc.zc_objset_type = DMU_OST_ZFS;
104710242Schris.kirby@sun.com 	zc.zc_defer_destroy = defer;
10485367Sahrens 	(void) strlcpy(zc.zc_name, name, sizeof (zc.zc_name));
10495367Sahrens 
10505367Sahrens 	if (flags.verbose)
10515367Sahrens 		(void) printf("attempting destroy %s\n", zc.zc_name);
10525367Sahrens 	err = ioctl(hdl->libzfs_fd, ZFS_IOC_DESTROY, &zc);
10536047Sahrens 	if (err == 0) {
10546047Sahrens 		if (flags.verbose)
10556047Sahrens 			(void) printf("success\n");
10566047Sahrens 		changelist_remove(clp, zc.zc_name);
10575367Sahrens 	}
10585367Sahrens 
10596047Sahrens 	(void) changelist_postfix(clp);
10606047Sahrens 	changelist_free(clp);
10616047Sahrens 
106210242Schris.kirby@sun.com 	/*
106310242Schris.kirby@sun.com 	 * Deferred destroy should always succeed. Since we can't tell
106410242Schris.kirby@sun.com 	 * if it destroyed the dataset or just marked it for deferred
106510242Schris.kirby@sun.com 	 * destroy, always do the rename just in case.
106610242Schris.kirby@sun.com 	 */
106710242Schris.kirby@sun.com 	if (err != 0 || defer)
10686047Sahrens 		err = recv_rename(hdl, name, NULL, baselen, newname, flags);
10695367Sahrens 
10705367Sahrens 	return (err);
10715367Sahrens }
10725367Sahrens 
10735367Sahrens typedef struct guid_to_name_data {
10745367Sahrens 	uint64_t guid;
10755367Sahrens 	char *name;
10765367Sahrens } guid_to_name_data_t;
10775367Sahrens 
10785367Sahrens static int
10795367Sahrens guid_to_name_cb(zfs_handle_t *zhp, void *arg)
10805367Sahrens {
10815367Sahrens 	guid_to_name_data_t *gtnd = arg;
10825367Sahrens 	int err;
10835367Sahrens 
10845367Sahrens 	if (zhp->zfs_dmustats.dds_guid == gtnd->guid) {
10855367Sahrens 		(void) strcpy(gtnd->name, zhp->zfs_name);
10865367Sahrens 		return (EEXIST);
10875367Sahrens 	}
10885367Sahrens 	err = zfs_iter_children(zhp, guid_to_name_cb, gtnd);
10895367Sahrens 	zfs_close(zhp);
10905367Sahrens 	return (err);
10915367Sahrens }
10925367Sahrens 
10935367Sahrens static int
10945367Sahrens guid_to_name(libzfs_handle_t *hdl, const char *parent, uint64_t guid,
10955367Sahrens     char *name)
10965367Sahrens {
10975367Sahrens 	/* exhaustive search all local snapshots */
10985367Sahrens 	guid_to_name_data_t gtnd;
10995367Sahrens 	int err = 0;
11005367Sahrens 	zfs_handle_t *zhp;
11015367Sahrens 	char *cp;
11025367Sahrens 
11035367Sahrens 	gtnd.guid = guid;
11045367Sahrens 	gtnd.name = name;
11055367Sahrens 
11065367Sahrens 	if (strchr(parent, '@') == NULL) {
11075367Sahrens 		zhp = make_dataset_handle(hdl, parent);
11085367Sahrens 		if (zhp != NULL) {
11095367Sahrens 			err = zfs_iter_children(zhp, guid_to_name_cb, &gtnd);
11105367Sahrens 			zfs_close(zhp);
11115367Sahrens 			if (err == EEXIST)
11125367Sahrens 				return (0);
11135367Sahrens 		}
11145367Sahrens 	}
11155367Sahrens 
11165367Sahrens 	cp = strchr(parent, '/');
11175367Sahrens 	if (cp)
11185367Sahrens 		*cp = '\0';
11195367Sahrens 	zhp = make_dataset_handle(hdl, parent);
11205367Sahrens 	if (cp)
11215367Sahrens 		*cp = '/';
11225367Sahrens 
11235367Sahrens 	if (zhp) {
11245367Sahrens 		err = zfs_iter_children(zhp, guid_to_name_cb, &gtnd);
11255367Sahrens 		zfs_close(zhp);
11265367Sahrens 	}
11275367Sahrens 
11285367Sahrens 	return (err == EEXIST ? 0 : ENOENT);
11295367Sahrens 
11305367Sahrens }
11315367Sahrens 
11325367Sahrens /*
11335367Sahrens  * Return true if dataset guid1 is created before guid2.
11345367Sahrens  */
11356476Sahl static int
11365367Sahrens created_before(libzfs_handle_t *hdl, avl_tree_t *avl,
11375367Sahrens     uint64_t guid1, uint64_t guid2)
11385367Sahrens {
11395367Sahrens 	nvlist_t *nvfs;
11405367Sahrens 	char *fsname, *snapname;
11415367Sahrens 	char buf[ZFS_MAXNAMELEN];
11426476Sahl 	int rv;
11435367Sahrens 	zfs_node_t zn1, zn2;
11445367Sahrens 
11455367Sahrens 	if (guid2 == 0)
11466476Sahl 		return (0);
11475367Sahrens 	if (guid1 == 0)
11486476Sahl 		return (1);
11495367Sahrens 
11505367Sahrens 	nvfs = fsavl_find(avl, guid1, &snapname);
11515367Sahrens 	VERIFY(0 == nvlist_lookup_string(nvfs, "name", &fsname));
11525367Sahrens 	(void) snprintf(buf, sizeof (buf), "%s@%s", fsname, snapname);
11535367Sahrens 	zn1.zn_handle = zfs_open(hdl, buf, ZFS_TYPE_SNAPSHOT);
11546476Sahl 	if (zn1.zn_handle == NULL)
11556476Sahl 		return (-1);
11565367Sahrens 
11575367Sahrens 	nvfs = fsavl_find(avl, guid2, &snapname);
11585367Sahrens 	VERIFY(0 == nvlist_lookup_string(nvfs, "name", &fsname));
11595367Sahrens 	(void) snprintf(buf, sizeof (buf), "%s@%s", fsname, snapname);
11605367Sahrens 	zn2.zn_handle = zfs_open(hdl, buf, ZFS_TYPE_SNAPSHOT);
11616476Sahl 	if (zn2.zn_handle == NULL) {
11626476Sahl 		zfs_close(zn2.zn_handle);
11636476Sahl 		return (-1);
11646476Sahl 	}
11655367Sahrens 
11665367Sahrens 	rv = (zfs_snapshot_compare(&zn1, &zn2) == -1);
11675367Sahrens 
11685367Sahrens 	zfs_close(zn1.zn_handle);
11695367Sahrens 	zfs_close(zn2.zn_handle);
11705367Sahrens 
11715367Sahrens 	return (rv);
11725367Sahrens }
11735367Sahrens 
11745367Sahrens static int
11755367Sahrens recv_incremental_replication(libzfs_handle_t *hdl, const char *tofs,
11765367Sahrens     recvflags_t flags, nvlist_t *stream_nv, avl_tree_t *stream_avl)
11775367Sahrens {
11785367Sahrens 	nvlist_t *local_nv;
11795367Sahrens 	avl_tree_t *local_avl;
11805367Sahrens 	nvpair_t *fselem, *nextfselem;
11815367Sahrens 	char *tosnap, *fromsnap;
11825367Sahrens 	char newname[ZFS_MAXNAMELEN];
11835367Sahrens 	int error;
11845367Sahrens 	boolean_t needagain, progress;
118510299SPrabahar.Jeyaram@Sun.COM 	char *s1, *s2;
11865367Sahrens 
11875367Sahrens 	VERIFY(0 == nvlist_lookup_string(stream_nv, "fromsnap", &fromsnap));
11885367Sahrens 	VERIFY(0 == nvlist_lookup_string(stream_nv, "tosnap", &tosnap));
11895367Sahrens 
11905367Sahrens 	if (flags.dryrun)
11915367Sahrens 		return (0);
11925367Sahrens 
11935367Sahrens again:
11945367Sahrens 	needagain = progress = B_FALSE;
11955367Sahrens 
11965367Sahrens 	if ((error = gather_nvlist(hdl, tofs, fromsnap, NULL,
11975367Sahrens 	    &local_nv, &local_avl)) != 0)
11985367Sahrens 		return (error);
11995367Sahrens 
12005367Sahrens 	/*
12015367Sahrens 	 * Process deletes and renames
12025367Sahrens 	 */
12035367Sahrens 	for (fselem = nvlist_next_nvpair(local_nv, NULL);
12045367Sahrens 	    fselem; fselem = nextfselem) {
12055367Sahrens 		nvlist_t *nvfs, *snaps;
12065367Sahrens 		nvlist_t *stream_nvfs = NULL;
12075367Sahrens 		nvpair_t *snapelem, *nextsnapelem;
12085367Sahrens 		uint64_t fromguid = 0;
12095367Sahrens 		uint64_t originguid = 0;
12105367Sahrens 		uint64_t stream_originguid = 0;
12115367Sahrens 		uint64_t parent_fromsnap_guid, stream_parent_fromsnap_guid;
12125367Sahrens 		char *fsname, *stream_fsname;
12135367Sahrens 
12145367Sahrens 		nextfselem = nvlist_next_nvpair(local_nv, fselem);
12155367Sahrens 
12165367Sahrens 		VERIFY(0 == nvpair_value_nvlist(fselem, &nvfs));
12175367Sahrens 		VERIFY(0 == nvlist_lookup_nvlist(nvfs, "snaps", &snaps));
12185367Sahrens 		VERIFY(0 == nvlist_lookup_string(nvfs, "name", &fsname));
12195367Sahrens 		VERIFY(0 == nvlist_lookup_uint64(nvfs, "parentfromsnap",
12205367Sahrens 		    &parent_fromsnap_guid));
12215367Sahrens 		(void) nvlist_lookup_uint64(nvfs, "origin", &originguid);
12225367Sahrens 
12235367Sahrens 		/*
12245367Sahrens 		 * First find the stream's fs, so we can check for
12255367Sahrens 		 * a different origin (due to "zfs promote")
12265367Sahrens 		 */
12275367Sahrens 		for (snapelem = nvlist_next_nvpair(snaps, NULL);
12285367Sahrens 		    snapelem; snapelem = nvlist_next_nvpair(snaps, snapelem)) {
12295367Sahrens 			uint64_t thisguid;
12305367Sahrens 
12315367Sahrens 			VERIFY(0 == nvpair_value_uint64(snapelem, &thisguid));
12325367Sahrens 			stream_nvfs = fsavl_find(stream_avl, thisguid, NULL);
12335367Sahrens 
12345367Sahrens 			if (stream_nvfs != NULL)
12355367Sahrens 				break;
12365367Sahrens 		}
12375367Sahrens 
12385367Sahrens 		/* check for promote */
12395367Sahrens 		(void) nvlist_lookup_uint64(stream_nvfs, "origin",
12405367Sahrens 		    &stream_originguid);
12415367Sahrens 		if (stream_nvfs && originguid != stream_originguid) {
12426476Sahl 			switch (created_before(hdl, local_avl,
12436476Sahl 			    stream_originguid, originguid)) {
12446476Sahl 			case 1: {
12455367Sahrens 				/* promote it! */
12465367Sahrens 				zfs_cmd_t zc = { 0 };
12475367Sahrens 				nvlist_t *origin_nvfs;
12485367Sahrens 				char *origin_fsname;
12495367Sahrens 
12505367Sahrens 				if (flags.verbose)
12515367Sahrens 					(void) printf("promoting %s\n", fsname);
12525367Sahrens 
12535367Sahrens 				origin_nvfs = fsavl_find(local_avl, originguid,
12545367Sahrens 				    NULL);
12555367Sahrens 				VERIFY(0 == nvlist_lookup_string(origin_nvfs,
12565367Sahrens 				    "name", &origin_fsname));
12575367Sahrens 				(void) strlcpy(zc.zc_value, origin_fsname,
12585367Sahrens 				    sizeof (zc.zc_value));
12595367Sahrens 				(void) strlcpy(zc.zc_name, fsname,
12605367Sahrens 				    sizeof (zc.zc_name));
12615367Sahrens 				error = zfs_ioctl(hdl, ZFS_IOC_PROMOTE, &zc);
12625367Sahrens 				if (error == 0)
12635367Sahrens 					progress = B_TRUE;
12646476Sahl 				break;
12656476Sahl 			}
12666476Sahl 			default:
12676476Sahl 				break;
12686476Sahl 			case -1:
12696476Sahl 				fsavl_destroy(local_avl);
12706476Sahl 				nvlist_free(local_nv);
12716476Sahl 				return (-1);
12725367Sahrens 			}
12735367Sahrens 			/*
12745367Sahrens 			 * We had/have the wrong origin, therefore our
12755367Sahrens 			 * list of snapshots is wrong.  Need to handle
12765367Sahrens 			 * them on the next pass.
12775367Sahrens 			 */
12785367Sahrens 			needagain = B_TRUE;
12795367Sahrens 			continue;
12805367Sahrens 		}
12815367Sahrens 
12825367Sahrens 		for (snapelem = nvlist_next_nvpair(snaps, NULL);
12835367Sahrens 		    snapelem; snapelem = nextsnapelem) {
12845367Sahrens 			uint64_t thisguid;
12855367Sahrens 			char *stream_snapname;
12867265Sahrens 			nvlist_t *found, *props;
12875367Sahrens 
12885367Sahrens 			nextsnapelem = nvlist_next_nvpair(snaps, snapelem);
12895367Sahrens 
12905367Sahrens 			VERIFY(0 == nvpair_value_uint64(snapelem, &thisguid));
12915367Sahrens 			found = fsavl_find(stream_avl, thisguid,
12925367Sahrens 			    &stream_snapname);
12935367Sahrens 
12945367Sahrens 			/* check for delete */
12955367Sahrens 			if (found == NULL) {
12965367Sahrens 				char name[ZFS_MAXNAMELEN];
12975367Sahrens 
12985367Sahrens 				if (!flags.force)
12995367Sahrens 					continue;
13005367Sahrens 
13015367Sahrens 				(void) snprintf(name, sizeof (name), "%s@%s",
13025367Sahrens 				    fsname, nvpair_name(snapelem));
13035367Sahrens 
13045367Sahrens 				error = recv_destroy(hdl, name,
13055367Sahrens 				    strlen(fsname)+1, newname, flags);
13065367Sahrens 				if (error)
13075367Sahrens 					needagain = B_TRUE;
13085367Sahrens 				else
13095367Sahrens 					progress = B_TRUE;
13105367Sahrens 				continue;
13115367Sahrens 			}
13125367Sahrens 
13135367Sahrens 			stream_nvfs = found;
13145367Sahrens 
13157265Sahrens 			if (0 == nvlist_lookup_nvlist(stream_nvfs, "snapprops",
13167265Sahrens 			    &props) && 0 == nvlist_lookup_nvlist(props,
13177265Sahrens 			    stream_snapname, &props)) {
13187265Sahrens 				zfs_cmd_t zc = { 0 };
13197265Sahrens 
13207265Sahrens 				zc.zc_cookie = B_TRUE; /* clear current props */
13217273Sahrens 				(void) snprintf(zc.zc_name, sizeof (zc.zc_name),
13227265Sahrens 				    "%s@%s", fsname, nvpair_name(snapelem));
13237265Sahrens 				if (zcmd_write_src_nvlist(hdl, &zc,
13247265Sahrens 				    props) == 0) {
13257265Sahrens 					(void) zfs_ioctl(hdl,
13267265Sahrens 					    ZFS_IOC_SET_PROP, &zc);
13277265Sahrens 					zcmd_free_nvlists(&zc);
13287265Sahrens 				}
13297265Sahrens 			}
13307265Sahrens 
13315367Sahrens 			/* check for different snapname */
13325367Sahrens 			if (strcmp(nvpair_name(snapelem),
13335367Sahrens 			    stream_snapname) != 0) {
13345367Sahrens 				char name[ZFS_MAXNAMELEN];
13355367Sahrens 				char tryname[ZFS_MAXNAMELEN];
13365367Sahrens 
13375367Sahrens 				(void) snprintf(name, sizeof (name), "%s@%s",
13385367Sahrens 				    fsname, nvpair_name(snapelem));
13395367Sahrens 				(void) snprintf(tryname, sizeof (name), "%s@%s",
13405367Sahrens 				    fsname, stream_snapname);
13415367Sahrens 
13425367Sahrens 				error = recv_rename(hdl, name, tryname,
13435367Sahrens 				    strlen(fsname)+1, newname, flags);
13445367Sahrens 				if (error)
13455367Sahrens 					needagain = B_TRUE;
13465367Sahrens 				else
13475367Sahrens 					progress = B_TRUE;
13485367Sahrens 			}
13495367Sahrens 
13505367Sahrens 			if (strcmp(stream_snapname, fromsnap) == 0)
13515367Sahrens 				fromguid = thisguid;
13525367Sahrens 		}
13535367Sahrens 
13545367Sahrens 		/* check for delete */
13555367Sahrens 		if (stream_nvfs == NULL) {
13565367Sahrens 			if (!flags.force)
13575367Sahrens 				continue;
13585367Sahrens 
13595367Sahrens 			error = recv_destroy(hdl, fsname, strlen(tofs)+1,
13605367Sahrens 			    newname, flags);
13615367Sahrens 			if (error)
13625367Sahrens 				needagain = B_TRUE;
13635367Sahrens 			else
13645367Sahrens 				progress = B_TRUE;
13655367Sahrens 			continue;
13665367Sahrens 		}
13675367Sahrens 
13685367Sahrens 		if (fromguid == 0 && flags.verbose) {
13695367Sahrens 			(void) printf("local fs %s does not have fromsnap "
13705367Sahrens 			    "(%s in stream); must have been deleted locally; "
13715367Sahrens 			    "ignoring\n", fsname, fromsnap);
13725367Sahrens 			continue;
13735367Sahrens 		}
13745367Sahrens 
13755367Sahrens 		VERIFY(0 == nvlist_lookup_string(stream_nvfs,
13765367Sahrens 		    "name", &stream_fsname));
13775367Sahrens 		VERIFY(0 == nvlist_lookup_uint64(stream_nvfs,
13785367Sahrens 		    "parentfromsnap", &stream_parent_fromsnap_guid));
13795367Sahrens 
138010302SPrabahar.Jeyaram@Sun.COM 		s1 = strrchr(fsname, '/');
138110302SPrabahar.Jeyaram@Sun.COM 		s2 = strrchr(stream_fsname, '/');
138210302SPrabahar.Jeyaram@Sun.COM 
13835367Sahrens 		/* check for rename */
13845367Sahrens 		if ((stream_parent_fromsnap_guid != 0 &&
13855367Sahrens 		    stream_parent_fromsnap_guid != parent_fromsnap_guid) ||
138610302SPrabahar.Jeyaram@Sun.COM 		    ((s1 != NULL) && (s2 != NULL) && strcmp(s1, s2) != 0)) {
13875367Sahrens 			nvlist_t *parent;
13885367Sahrens 			char tryname[ZFS_MAXNAMELEN];
13895367Sahrens 
13905367Sahrens 			parent = fsavl_find(local_avl,
13915367Sahrens 			    stream_parent_fromsnap_guid, NULL);
13925367Sahrens 			/*
13935367Sahrens 			 * NB: parent might not be found if we used the
13945367Sahrens 			 * tosnap for stream_parent_fromsnap_guid,
13955367Sahrens 			 * because the parent is a newly-created fs;
13965367Sahrens 			 * we'll be able to rename it after we recv the
13975367Sahrens 			 * new fs.
13985367Sahrens 			 */
13995367Sahrens 			if (parent != NULL) {
14005367Sahrens 				char *pname;
14015367Sahrens 
14025367Sahrens 				VERIFY(0 == nvlist_lookup_string(parent, "name",
14035367Sahrens 				    &pname));
14045367Sahrens 				(void) snprintf(tryname, sizeof (tryname),
14055367Sahrens 				    "%s%s", pname, strrchr(stream_fsname, '/'));
14065367Sahrens 			} else {
14075367Sahrens 				tryname[0] = '\0';
14085367Sahrens 				if (flags.verbose) {
14095367Sahrens 					(void) printf("local fs %s new parent "
14105367Sahrens 					    "not found\n", fsname);
14115367Sahrens 				}
14125367Sahrens 			}
14135367Sahrens 
14145367Sahrens 			error = recv_rename(hdl, fsname, tryname,
14155367Sahrens 			    strlen(tofs)+1, newname, flags);
14165367Sahrens 			if (error)
14175367Sahrens 				needagain = B_TRUE;
14185367Sahrens 			else
14195367Sahrens 				progress = B_TRUE;
14205367Sahrens 		}
14215367Sahrens 	}
14225367Sahrens 
14235367Sahrens 	fsavl_destroy(local_avl);
14245367Sahrens 	nvlist_free(local_nv);
14255367Sahrens 
14265367Sahrens 	if (needagain && progress) {
14275367Sahrens 		/* do another pass to fix up temporary names */
14285367Sahrens 		if (flags.verbose)
14295367Sahrens 			(void) printf("another pass:\n");
14305367Sahrens 		goto again;
14315367Sahrens 	}
14325367Sahrens 
14335367Sahrens 	return (needagain);
14345367Sahrens }
14355367Sahrens 
14365367Sahrens static int
14375367Sahrens zfs_receive_package(libzfs_handle_t *hdl, int fd, const char *destname,
14387366STim.Haley@Sun.COM     recvflags_t flags, dmu_replay_record_t *drr, zio_cksum_t *zc,
14397366STim.Haley@Sun.COM     char **top_zfs)
14405367Sahrens {
14415367Sahrens 	nvlist_t *stream_nv = NULL;
14425367Sahrens 	avl_tree_t *stream_avl = NULL;
14435367Sahrens 	char *fromsnap = NULL;
14445367Sahrens 	char tofs[ZFS_MAXNAMELEN];
14455367Sahrens 	char errbuf[1024];
14465367Sahrens 	dmu_replay_record_t drre;
14475367Sahrens 	int error;
14485367Sahrens 	boolean_t anyerr = B_FALSE;
14496479Sahrens 	boolean_t softerr = B_FALSE;
14505367Sahrens 
14515367Sahrens 	(void) snprintf(errbuf, sizeof (errbuf), dgettext(TEXT_DOMAIN,
14525367Sahrens 	    "cannot receive"));
14535367Sahrens 
14545367Sahrens 	if (strchr(destname, '@')) {
14555367Sahrens 		zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
14565367Sahrens 		    "can not specify snapshot name for multi-snapshot stream"));
14575367Sahrens 		return (zfs_error(hdl, EZFS_BADSTREAM, errbuf));
14585367Sahrens 	}
14595367Sahrens 
14605367Sahrens 	assert(drr->drr_type == DRR_BEGIN);
14615367Sahrens 	assert(drr->drr_u.drr_begin.drr_magic == DMU_BACKUP_MAGIC);
14625367Sahrens 	assert(drr->drr_u.drr_begin.drr_version == DMU_BACKUP_HEADER_VERSION);
14635367Sahrens 
14645367Sahrens 	/*
14655367Sahrens 	 * Read in the nvlist from the stream.
14665367Sahrens 	 */
14675367Sahrens 	if (drr->drr_payloadlen != 0) {
14685367Sahrens 		if (!flags.isprefix) {
14695367Sahrens 			zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
14705367Sahrens 			    "must use -d to receive replication "
14715367Sahrens 			    "(send -R) stream"));
14725367Sahrens 			return (zfs_error(hdl, EZFS_BADSTREAM, errbuf));
14735367Sahrens 		}
14745367Sahrens 
14755367Sahrens 		error = recv_read_nvlist(hdl, fd, drr->drr_payloadlen,
14765367Sahrens 		    &stream_nv, flags.byteswap, zc);
14775367Sahrens 		if (error) {
14785367Sahrens 			error = zfs_error(hdl, EZFS_BADSTREAM, errbuf);
14795367Sahrens 			goto out;
14805367Sahrens 		}
14815367Sahrens 	}
14825367Sahrens 
14835367Sahrens 	/*
14845367Sahrens 	 * Read in the end record and verify checksum.
14855367Sahrens 	 */
14865367Sahrens 	if (0 != (error = recv_read(hdl, fd, &drre, sizeof (drre),
14875367Sahrens 	    flags.byteswap, NULL)))
14885367Sahrens 		goto out;
14895367Sahrens 	if (flags.byteswap) {
14905367Sahrens 		drre.drr_type = BSWAP_32(drre.drr_type);
14915367Sahrens 		drre.drr_u.drr_end.drr_checksum.zc_word[0] =
14925367Sahrens 		    BSWAP_64(drre.drr_u.drr_end.drr_checksum.zc_word[0]);
14935367Sahrens 		drre.drr_u.drr_end.drr_checksum.zc_word[1] =
14945367Sahrens 		    BSWAP_64(drre.drr_u.drr_end.drr_checksum.zc_word[1]);
14955367Sahrens 		drre.drr_u.drr_end.drr_checksum.zc_word[2] =
14965367Sahrens 		    BSWAP_64(drre.drr_u.drr_end.drr_checksum.zc_word[2]);
14975367Sahrens 		drre.drr_u.drr_end.drr_checksum.zc_word[3] =
14985367Sahrens 		    BSWAP_64(drre.drr_u.drr_end.drr_checksum.zc_word[3]);
14995367Sahrens 	}
15005367Sahrens 	if (drre.drr_type != DRR_END) {
15015367Sahrens 		error = zfs_error(hdl, EZFS_BADSTREAM, errbuf);
15025367Sahrens 		goto out;
15035367Sahrens 	}
15045367Sahrens 	if (!ZIO_CHECKSUM_EQUAL(drre.drr_u.drr_end.drr_checksum, *zc)) {
15055367Sahrens 		zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
15065367Sahrens 		    "incorrect header checksum"));
15075367Sahrens 		error = zfs_error(hdl, EZFS_BADSTREAM, errbuf);
15085367Sahrens 		goto out;
15095367Sahrens 	}
15105367Sahrens 
15115367Sahrens 	(void) nvlist_lookup_string(stream_nv, "fromsnap", &fromsnap);
15125367Sahrens 
15135367Sahrens 	if (drr->drr_payloadlen != 0) {
15145367Sahrens 		nvlist_t *stream_fss;
15155367Sahrens 
15165367Sahrens 		VERIFY(0 == nvlist_lookup_nvlist(stream_nv, "fss",
15175367Sahrens 		    &stream_fss));
15185896Sahl 		if ((stream_avl = fsavl_create(stream_fss)) == NULL) {
15195896Sahl 			zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
15205896Sahl 			    "couldn't allocate avl tree"));
15215896Sahl 			error = zfs_error(hdl, EZFS_NOMEM, errbuf);
15225896Sahl 			goto out;
15235896Sahl 		}
15245367Sahrens 
15255367Sahrens 		if (fromsnap != NULL) {
15265367Sahrens 			(void) strlcpy(tofs, destname, ZFS_MAXNAMELEN);
15275367Sahrens 			if (flags.isprefix) {
15285367Sahrens 				int i = strcspn(drr->drr_u.drr_begin.drr_toname,
15295367Sahrens 				    "/@");
15305367Sahrens 				/* zfs_receive_one() will create_parents() */
15315367Sahrens 				(void) strlcat(tofs,
15325367Sahrens 				    &drr->drr_u.drr_begin.drr_toname[i],
15335367Sahrens 				    ZFS_MAXNAMELEN);
15345367Sahrens 				*strchr(tofs, '@') = '\0';
15355367Sahrens 			}
15366479Sahrens 			softerr = recv_incremental_replication(hdl, tofs,
15375367Sahrens 			    flags, stream_nv, stream_avl);
15385367Sahrens 		}
15395367Sahrens 	}
15405367Sahrens 
15415367Sahrens 
15425367Sahrens 	/* Finally, receive each contained stream */
15435367Sahrens 	do {
15445367Sahrens 		/*
15455367Sahrens 		 * we should figure out if it has a recoverable
15465367Sahrens 		 * error, in which case do a recv_skip() and drive on.
15475367Sahrens 		 * Note, if we fail due to already having this guid,
15485367Sahrens 		 * zfs_receive_one() will take care of it (ie,
15495367Sahrens 		 * recv_skip() and return 0).
15505367Sahrens 		 */
15517366STim.Haley@Sun.COM 		error = zfs_receive_impl(hdl, destname, flags, fd,
15527366STim.Haley@Sun.COM 		    stream_avl, top_zfs);
15535367Sahrens 		if (error == ENODATA) {
15545367Sahrens 			error = 0;
15555367Sahrens 			break;
15565367Sahrens 		}
15575367Sahrens 		anyerr |= error;
15585367Sahrens 	} while (error == 0);
15595367Sahrens 
15605367Sahrens 	if (drr->drr_payloadlen != 0 && fromsnap != NULL) {
15615367Sahrens 		/*
15625367Sahrens 		 * Now that we have the fs's they sent us, try the
15635367Sahrens 		 * renames again.
15645367Sahrens 		 */
15656479Sahrens 		softerr = recv_incremental_replication(hdl, tofs, flags,
15665367Sahrens 		    stream_nv, stream_avl);
15675367Sahrens 	}
15685367Sahrens 
15695367Sahrens out:
15705367Sahrens 	fsavl_destroy(stream_avl);
15715367Sahrens 	if (stream_nv)
15725367Sahrens 		nvlist_free(stream_nv);
15736479Sahrens 	if (softerr)
15746479Sahrens 		error = -2;
15755367Sahrens 	if (anyerr)
15765367Sahrens 		error = -1;
15775367Sahrens 	return (error);
15785367Sahrens }
15795367Sahrens 
15805367Sahrens static int
15815367Sahrens recv_skip(libzfs_handle_t *hdl, int fd, boolean_t byteswap)
15825367Sahrens {
15835367Sahrens 	dmu_replay_record_t *drr;
15845367Sahrens 	void *buf = malloc(1<<20);
15855367Sahrens 
15865367Sahrens 	/* XXX would be great to use lseek if possible... */
15875367Sahrens 	drr = buf;
15885367Sahrens 
15895367Sahrens 	while (recv_read(hdl, fd, drr, sizeof (dmu_replay_record_t),
15905367Sahrens 	    byteswap, NULL) == 0) {
15915367Sahrens 		if (byteswap)
15925367Sahrens 			drr->drr_type = BSWAP_32(drr->drr_type);
15935367Sahrens 
15945367Sahrens 		switch (drr->drr_type) {
15955367Sahrens 		case DRR_BEGIN:
15965367Sahrens 			/* NB: not to be used on v2 stream packages */
15975367Sahrens 			assert(drr->drr_payloadlen == 0);
15985367Sahrens 			break;
15995367Sahrens 
16005367Sahrens 		case DRR_END:
16015367Sahrens 			free(buf);
16025367Sahrens 			return (0);
16035367Sahrens 
16045367Sahrens 		case DRR_OBJECT:
16055367Sahrens 			if (byteswap) {
16065367Sahrens 				drr->drr_u.drr_object.drr_bonuslen =
16075367Sahrens 				    BSWAP_32(drr->drr_u.drr_object.
16085367Sahrens 				    drr_bonuslen);
16095367Sahrens 			}
16105367Sahrens 			(void) recv_read(hdl, fd, buf,
16115367Sahrens 			    P2ROUNDUP(drr->drr_u.drr_object.drr_bonuslen, 8),
16125367Sahrens 			    B_FALSE, NULL);
16135367Sahrens 			break;
16145367Sahrens 
16155367Sahrens 		case DRR_WRITE:
16165367Sahrens 			if (byteswap) {
16175367Sahrens 				drr->drr_u.drr_write.drr_length =
16187366STim.Haley@Sun.COM 				    BSWAP_64(drr->drr_u.drr_write.drr_length);
16195367Sahrens 			}
16205367Sahrens 			(void) recv_read(hdl, fd, buf,
16215367Sahrens 			    drr->drr_u.drr_write.drr_length, B_FALSE, NULL);
16225367Sahrens 			break;
16235367Sahrens 
16245367Sahrens 		case DRR_FREEOBJECTS:
16255367Sahrens 		case DRR_FREE:
16265367Sahrens 			break;
16275367Sahrens 
16285367Sahrens 		default:
16295367Sahrens 			assert(!"invalid record type");
16305367Sahrens 		}
16315367Sahrens 	}
16325367Sahrens 
16335367Sahrens 	free(buf);
16345367Sahrens 	return (-1);
16355367Sahrens }
16365367Sahrens 
16375367Sahrens /*
16385367Sahrens  * Restores a backup of tosnap from the file descriptor specified by infd.
16395367Sahrens  */
16405367Sahrens static int
16415367Sahrens zfs_receive_one(libzfs_handle_t *hdl, int infd, const char *tosnap,
16425367Sahrens     recvflags_t flags, dmu_replay_record_t *drr,
16437366STim.Haley@Sun.COM     dmu_replay_record_t *drr_noswap, avl_tree_t *stream_avl,
16447366STim.Haley@Sun.COM     char **top_zfs)
16455367Sahrens {
16465367Sahrens 	zfs_cmd_t zc = { 0 };
16475367Sahrens 	time_t begin_time;
16485367Sahrens 	int ioctl_err, ioctl_errno, err, choplen;
16495367Sahrens 	char *cp;
16505367Sahrens 	struct drr_begin *drrb = &drr->drr_u.drr_begin;
16515367Sahrens 	char errbuf[1024];
16525367Sahrens 	char chopprefix[ZFS_MAXNAMELEN];
16535367Sahrens 	boolean_t newfs = B_FALSE;
16545367Sahrens 	boolean_t stream_wantsnewfs;
16555367Sahrens 	uint64_t parent_snapguid = 0;
16565367Sahrens 	prop_changelist_t *clp = NULL;
16577265Sahrens 	nvlist_t *snapprops_nvlist = NULL;
16585367Sahrens 
16595367Sahrens 	begin_time = time(NULL);
16605367Sahrens 
16615367Sahrens 	(void) snprintf(errbuf, sizeof (errbuf), dgettext(TEXT_DOMAIN,
16625367Sahrens 	    "cannot receive"));
16635367Sahrens 
16645367Sahrens 	if (stream_avl != NULL) {
16657265Sahrens 		char *snapname;
16667265Sahrens 		nvlist_t *fs = fsavl_find(stream_avl, drrb->drr_toguid,
16677265Sahrens 		    &snapname);
16685367Sahrens 		nvlist_t *props;
16695896Sahl 		int ret;
16705367Sahrens 
16715367Sahrens 		(void) nvlist_lookup_uint64(fs, "parentfromsnap",
16725367Sahrens 		    &parent_snapguid);
16735367Sahrens 		err = nvlist_lookup_nvlist(fs, "props", &props);
16745367Sahrens 		if (err)
16755367Sahrens 			VERIFY(0 == nvlist_alloc(&props, NV_UNIQUE_NAME, 0));
16765896Sahl 
16775367Sahrens 		if (flags.canmountoff) {
16785367Sahrens 			VERIFY(0 == nvlist_add_uint64(props,
16795367Sahrens 			    zfs_prop_to_name(ZFS_PROP_CANMOUNT), 0));
16805367Sahrens 		}
16815896Sahl 		ret = zcmd_write_src_nvlist(hdl, &zc, props);
16825367Sahrens 		if (err)
16835367Sahrens 			nvlist_free(props);
16845896Sahl 
16857265Sahrens 		if (0 == nvlist_lookup_nvlist(fs, "snapprops", &props)) {
16867265Sahrens 			VERIFY(0 == nvlist_lookup_nvlist(props,
16877265Sahrens 			    snapname, &snapprops_nvlist));
16887265Sahrens 		}
16897265Sahrens 
16905896Sahl 		if (ret != 0)
16915896Sahl 			return (-1);
16925367Sahrens 	}
16935367Sahrens 
16945367Sahrens 	/*
16955367Sahrens 	 * Determine how much of the snapshot name stored in the stream
16965367Sahrens 	 * we are going to tack on to the name they specified on the
16975367Sahrens 	 * command line, and how much we are going to chop off.
16985367Sahrens 	 *
16995367Sahrens 	 * If they specified a snapshot, chop the entire name stored in
17005367Sahrens 	 * the stream.
17015367Sahrens 	 */
17025367Sahrens 	(void) strcpy(chopprefix, drrb->drr_toname);
17035367Sahrens 	if (flags.isprefix) {
17045367Sahrens 		/*
17055367Sahrens 		 * They specified a fs with -d, we want to tack on
17065367Sahrens 		 * everything but the pool name stored in the stream
17075367Sahrens 		 */
17085367Sahrens 		if (strchr(tosnap, '@')) {
17095367Sahrens 			zfs_error_aux(hdl, dgettext(TEXT_DOMAIN, "invalid "
17105367Sahrens 			    "argument - snapshot not allowed with -d"));
17115367Sahrens 			return (zfs_error(hdl, EZFS_INVALIDNAME, errbuf));
17125367Sahrens 		}
17135367Sahrens 		cp = strchr(chopprefix, '/');
17145367Sahrens 		if (cp == NULL)
17155367Sahrens 			cp = strchr(chopprefix, '@');
17165367Sahrens 		*cp = '\0';
17175367Sahrens 	} else if (strchr(tosnap, '@') == NULL) {
17185367Sahrens 		/*
17195367Sahrens 		 * If they specified a filesystem without -d, we want to
17205367Sahrens 		 * tack on everything after the fs specified in the
17215367Sahrens 		 * first name from the stream.
17225367Sahrens 		 */
17235367Sahrens 		cp = strchr(chopprefix, '@');
17245367Sahrens 		*cp = '\0';
17255367Sahrens 	}
17265367Sahrens 	choplen = strlen(chopprefix);
17275367Sahrens 
17285367Sahrens 	/*
17295367Sahrens 	 * Determine name of destination snapshot, store in zc_value.
17305367Sahrens 	 */
17315367Sahrens 	(void) strcpy(zc.zc_value, tosnap);
17325367Sahrens 	(void) strncat(zc.zc_value, drrb->drr_toname+choplen,
17335367Sahrens 	    sizeof (zc.zc_value));
17345896Sahl 	if (!zfs_name_valid(zc.zc_value, ZFS_TYPE_SNAPSHOT)) {
17355896Sahl 		zcmd_free_nvlists(&zc);
17365367Sahrens 		return (zfs_error(hdl, EZFS_INVALIDNAME, errbuf));
17375896Sahl 	}
17385367Sahrens 
17395367Sahrens 	/*
17405367Sahrens 	 * Determine the name of the origin snapshot, store in zc_string.
17415367Sahrens 	 */
17425367Sahrens 	if (drrb->drr_flags & DRR_FLAG_CLONE) {
17435367Sahrens 		if (guid_to_name(hdl, tosnap,
17445367Sahrens 		    drrb->drr_fromguid, zc.zc_string) != 0) {
17455896Sahl 			zcmd_free_nvlists(&zc);
17465367Sahrens 			zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
17475367Sahrens 			    "local origin for clone %s does not exist"),
17485367Sahrens 			    zc.zc_value);
17495367Sahrens 			return (zfs_error(hdl, EZFS_NOENT, errbuf));
17505367Sahrens 		}
17515367Sahrens 		if (flags.verbose)
17525367Sahrens 			(void) printf("found clone origin %s\n", zc.zc_string);
17535367Sahrens 	}
17545367Sahrens 
17555367Sahrens 	stream_wantsnewfs = (drrb->drr_fromguid == NULL ||
17565367Sahrens 	    (drrb->drr_flags & DRR_FLAG_CLONE));
17575367Sahrens 
17585367Sahrens 	if (stream_wantsnewfs) {
17595367Sahrens 		/*
17605367Sahrens 		 * if the parent fs does not exist, look for it based on
17615367Sahrens 		 * the parent snap GUID
17625367Sahrens 		 */
17635367Sahrens 		(void) snprintf(errbuf, sizeof (errbuf), dgettext(TEXT_DOMAIN,
17645367Sahrens 		    "cannot receive new filesystem stream"));
17655367Sahrens 
17665367Sahrens 		(void) strcpy(zc.zc_name, zc.zc_value);
17675367Sahrens 		cp = strrchr(zc.zc_name, '/');
17685367Sahrens 		if (cp)
17695367Sahrens 			*cp = '\0';
17705367Sahrens 		if (cp &&
17715367Sahrens 		    !zfs_dataset_exists(hdl, zc.zc_name, ZFS_TYPE_DATASET)) {
17725367Sahrens 			char suffix[ZFS_MAXNAMELEN];
17735367Sahrens 			(void) strcpy(suffix, strrchr(zc.zc_value, '/'));
17745367Sahrens 			if (guid_to_name(hdl, tosnap, parent_snapguid,
17755367Sahrens 			    zc.zc_value) == 0) {
17765367Sahrens 				*strchr(zc.zc_value, '@') = '\0';
17775367Sahrens 				(void) strcat(zc.zc_value, suffix);
17785367Sahrens 			}
17795367Sahrens 		}
17805367Sahrens 	} else {
17815367Sahrens 		/*
17825367Sahrens 		 * if the fs does not exist, look for it based on the
17835367Sahrens 		 * fromsnap GUID
17845367Sahrens 		 */
17855367Sahrens 		(void) snprintf(errbuf, sizeof (errbuf), dgettext(TEXT_DOMAIN,
17865367Sahrens 		    "cannot receive incremental stream"));
17875367Sahrens 
17885367Sahrens 		(void) strcpy(zc.zc_name, zc.zc_value);
17895367Sahrens 		*strchr(zc.zc_name, '@') = '\0';
17905367Sahrens 
17915367Sahrens 		if (!zfs_dataset_exists(hdl, zc.zc_name, ZFS_TYPE_DATASET)) {
17925367Sahrens 			char snap[ZFS_MAXNAMELEN];
17935367Sahrens 			(void) strcpy(snap, strchr(zc.zc_value, '@'));
17945367Sahrens 			if (guid_to_name(hdl, tosnap, drrb->drr_fromguid,
17955367Sahrens 			    zc.zc_value) == 0) {
17965367Sahrens 				*strchr(zc.zc_value, '@') = '\0';
17975367Sahrens 				(void) strcat(zc.zc_value, snap);
17985367Sahrens 			}
17995367Sahrens 		}
18005367Sahrens 	}
18015367Sahrens 
18025367Sahrens 	(void) strcpy(zc.zc_name, zc.zc_value);
18035367Sahrens 	*strchr(zc.zc_name, '@') = '\0';
18045367Sahrens 
18055367Sahrens 	if (zfs_dataset_exists(hdl, zc.zc_name, ZFS_TYPE_DATASET)) {
18065367Sahrens 		zfs_handle_t *zhp;
18075367Sahrens 		/*
18085367Sahrens 		 * Destination fs exists.  Therefore this should either
18095367Sahrens 		 * be an incremental, or the stream specifies a new fs
18105367Sahrens 		 * (full stream or clone) and they want us to blow it
18115367Sahrens 		 * away (and have therefore specified -F and removed any
18125367Sahrens 		 * snapshots).
18135367Sahrens 		 */
18145367Sahrens 
18155367Sahrens 		if (stream_wantsnewfs) {
18165367Sahrens 			if (!flags.force) {
18175896Sahl 				zcmd_free_nvlists(&zc);
18185367Sahrens 				zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
18195367Sahrens 				    "destination '%s' exists\n"
18205367Sahrens 				    "must specify -F to overwrite it"),
18215367Sahrens 				    zc.zc_name);
18225367Sahrens 				return (zfs_error(hdl, EZFS_EXISTS, errbuf));
18235367Sahrens 			}
18245367Sahrens 			if (ioctl(hdl->libzfs_fd, ZFS_IOC_SNAPSHOT_LIST_NEXT,
18255367Sahrens 			    &zc) == 0) {
18265896Sahl 				zcmd_free_nvlists(&zc);
18275367Sahrens 				zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
18285367Sahrens 				    "destination has snapshots (eg. %s)\n"
18295367Sahrens 				    "must destroy them to overwrite it"),
18305367Sahrens 				    zc.zc_name);
18315367Sahrens 				return (zfs_error(hdl, EZFS_EXISTS, errbuf));
18325367Sahrens 			}
18335367Sahrens 		}
18345367Sahrens 
18355896Sahl 		if ((zhp = zfs_open(hdl, zc.zc_name,
18365896Sahl 		    ZFS_TYPE_FILESYSTEM | ZFS_TYPE_VOLUME)) == NULL) {
18375896Sahl 			zcmd_free_nvlists(&zc);
18385367Sahrens 			return (-1);
18395896Sahl 		}
18405896Sahl 
18415367Sahrens 		if (stream_wantsnewfs &&
18425367Sahrens 		    zhp->zfs_dmustats.dds_origin[0]) {
18435896Sahl 			zcmd_free_nvlists(&zc);
18445367Sahrens 			zfs_close(zhp);
18455367Sahrens 			zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
18465367Sahrens 			    "destination '%s' is a clone\n"
18475367Sahrens 			    "must destroy it to overwrite it"),
18485367Sahrens 			    zc.zc_name);
18495367Sahrens 			return (zfs_error(hdl, EZFS_EXISTS, errbuf));
18505367Sahrens 		}
18515367Sahrens 
18525367Sahrens 		if (!flags.dryrun && zhp->zfs_type == ZFS_TYPE_FILESYSTEM &&
18535367Sahrens 		    stream_wantsnewfs) {
18545367Sahrens 			/* We can't do online recv in this case */
18557366STim.Haley@Sun.COM 			clp = changelist_gather(zhp, ZFS_PROP_NAME, 0, 0);
18565896Sahl 			if (clp == NULL) {
185710200SLori.Alt@Sun.COM 				zfs_close(zhp);
18585896Sahl 				zcmd_free_nvlists(&zc);
18595367Sahrens 				return (-1);
18605896Sahl 			}
18615367Sahrens 			if (changelist_prefix(clp) != 0) {
18625367Sahrens 				changelist_free(clp);
186310200SLori.Alt@Sun.COM 				zfs_close(zhp);
18645896Sahl 				zcmd_free_nvlists(&zc);
18655367Sahrens 				return (-1);
18665367Sahrens 			}
18675367Sahrens 		}
18685367Sahrens 		zfs_close(zhp);
18695367Sahrens 	} else {
18705367Sahrens 		/*
18715896Sahl 		 * Destination filesystem does not exist.  Therefore we better
18725896Sahl 		 * be creating a new filesystem (either from a full backup, or
18735896Sahl 		 * a clone).  It would therefore be invalid if the user
18745896Sahl 		 * specified only the pool name (i.e. if the destination name
18755896Sahl 		 * contained no slash character).
18765367Sahrens 		 */
18775896Sahl 		if (!stream_wantsnewfs ||
18785896Sahl 		    (cp = strrchr(zc.zc_name, '/')) == NULL) {
18795896Sahl 			zcmd_free_nvlists(&zc);
18805367Sahrens 			zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
18815367Sahrens 			    "destination '%s' does not exist"), zc.zc_name);
18825367Sahrens 			return (zfs_error(hdl, EZFS_NOENT, errbuf));
18835367Sahrens 		}
18845367Sahrens 
18855896Sahl 		/*
18865896Sahl 		 * Trim off the final dataset component so we perform the
18875896Sahl 		 * recvbackup ioctl to the filesystems's parent.
18885896Sahl 		 */
18895896Sahl 		*cp = '\0';
18905367Sahrens 
18915896Sahl 		if (flags.isprefix && !flags.dryrun &&
18925896Sahl 		    create_parents(hdl, zc.zc_value, strlen(tosnap)) != 0) {
18935896Sahl 			zcmd_free_nvlists(&zc);
18945896Sahl 			return (zfs_error(hdl, EZFS_BADRESTORE, errbuf));
18955367Sahrens 		}
18965367Sahrens 
18975367Sahrens 		newfs = B_TRUE;
18985367Sahrens 	}
18995367Sahrens 
19005367Sahrens 	zc.zc_begin_record = drr_noswap->drr_u.drr_begin;
19015367Sahrens 	zc.zc_cookie = infd;
19025367Sahrens 	zc.zc_guid = flags.force;
19035367Sahrens 	if (flags.verbose) {
19045367Sahrens 		(void) printf("%s %s stream of %s into %s\n",
19055367Sahrens 		    flags.dryrun ? "would receive" : "receiving",
19065367Sahrens 		    drrb->drr_fromguid ? "incremental" : "full",
19075367Sahrens 		    drrb->drr_toname, zc.zc_value);
19085367Sahrens 		(void) fflush(stdout);
19095367Sahrens 	}
19105367Sahrens 
19115896Sahl 	if (flags.dryrun) {
19125896Sahl 		zcmd_free_nvlists(&zc);
19135367Sahrens 		return (recv_skip(hdl, infd, flags.byteswap));
19145896Sahl 	}
19155367Sahrens 
19165367Sahrens 	err = ioctl_err = zfs_ioctl(hdl, ZFS_IOC_RECV, &zc);
19175367Sahrens 	ioctl_errno = errno;
19187265Sahrens 	zcmd_free_nvlists(&zc);
19197265Sahrens 
19207265Sahrens 	if (err == 0 && snapprops_nvlist) {
19217265Sahrens 		zfs_cmd_t zc2 = { 0 };
19227265Sahrens 
19237273Sahrens 		(void) strcpy(zc2.zc_name, zc.zc_value);
19247265Sahrens 		if (zcmd_write_src_nvlist(hdl, &zc2, snapprops_nvlist) == 0) {
19257265Sahrens 			(void) zfs_ioctl(hdl, ZFS_IOC_SET_PROP, &zc2);
19267265Sahrens 			zcmd_free_nvlists(&zc2);
19277265Sahrens 		}
19287265Sahrens 	}
19297265Sahrens 
19305367Sahrens 	if (err && (ioctl_errno == ENOENT || ioctl_errno == ENODEV)) {
19315367Sahrens 		/*
19325367Sahrens 		 * It may be that this snapshot already exists,
19335367Sahrens 		 * in which case we want to consume & ignore it
19345367Sahrens 		 * rather than failing.
19355367Sahrens 		 */
19365367Sahrens 		avl_tree_t *local_avl;
19375367Sahrens 		nvlist_t *local_nv, *fs;
19385367Sahrens 		char *cp = strchr(zc.zc_value, '@');
19395367Sahrens 
19405367Sahrens 		/*
19415367Sahrens 		 * XXX Do this faster by just iterating over snaps in
19425367Sahrens 		 * this fs.  Also if zc_value does not exist, we will
19435367Sahrens 		 * get a strange "does not exist" error message.
19445367Sahrens 		 */
19455367Sahrens 		*cp = '\0';
19465367Sahrens 		if (gather_nvlist(hdl, zc.zc_value, NULL, NULL,
19475367Sahrens 		    &local_nv, &local_avl) == 0) {
19485367Sahrens 			*cp = '@';
19495367Sahrens 			fs = fsavl_find(local_avl, drrb->drr_toguid, NULL);
19505367Sahrens 			fsavl_destroy(local_avl);
19515367Sahrens 			nvlist_free(local_nv);
19525367Sahrens 
19535367Sahrens 			if (fs != NULL) {
19545367Sahrens 				if (flags.verbose) {
19555367Sahrens 					(void) printf("snap %s already exists; "
19565367Sahrens 					    "ignoring\n", zc.zc_value);
19575367Sahrens 				}
19585367Sahrens 				ioctl_err = recv_skip(hdl, infd,
19595367Sahrens 				    flags.byteswap);
19605367Sahrens 			}
19615367Sahrens 		}
19625367Sahrens 		*cp = '@';
19635367Sahrens 	}
19645367Sahrens 
19655896Sahl 
19665367Sahrens 	if (ioctl_err != 0) {
19675367Sahrens 		switch (ioctl_errno) {
19685367Sahrens 		case ENODEV:
19695367Sahrens 			cp = strchr(zc.zc_value, '@');
19705367Sahrens 			*cp = '\0';
19715367Sahrens 			zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
19725367Sahrens 			    "most recent snapshot of %s does not\n"
19735367Sahrens 			    "match incremental source"), zc.zc_value);
19745367Sahrens 			(void) zfs_error(hdl, EZFS_BADRESTORE, errbuf);
19755367Sahrens 			*cp = '@';
19765367Sahrens 			break;
19775367Sahrens 		case ETXTBSY:
19785367Sahrens 			zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
19795367Sahrens 			    "destination %s has been modified\n"
19805367Sahrens 			    "since most recent snapshot"), zc.zc_name);
19815367Sahrens 			(void) zfs_error(hdl, EZFS_BADRESTORE, errbuf);
19825367Sahrens 			break;
19835367Sahrens 		case EEXIST:
19845367Sahrens 			cp = strchr(zc.zc_value, '@');
19855367Sahrens 			if (newfs) {
19865367Sahrens 				/* it's the containing fs that exists */
19875367Sahrens 				*cp = '\0';
19885367Sahrens 			}
19895367Sahrens 			zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
19905367Sahrens 			    "destination already exists"));
19915367Sahrens 			(void) zfs_error_fmt(hdl, EZFS_EXISTS,
19925367Sahrens 			    dgettext(TEXT_DOMAIN, "cannot restore to %s"),
19935367Sahrens 			    zc.zc_value);
19945367Sahrens 			*cp = '@';
19955367Sahrens 			break;
19965367Sahrens 		case EINVAL:
19975367Sahrens 			(void) zfs_error(hdl, EZFS_BADSTREAM, errbuf);
19985367Sahrens 			break;
19995367Sahrens 		case ECKSUM:
20005367Sahrens 			zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
20015367Sahrens 			    "invalid stream (checksum mismatch)"));
20025367Sahrens 			(void) zfs_error(hdl, EZFS_BADSTREAM, errbuf);
20035367Sahrens 			break;
20045367Sahrens 		default:
20055367Sahrens 			(void) zfs_standard_error(hdl, ioctl_errno, errbuf);
20065367Sahrens 		}
20075367Sahrens 	}
20085367Sahrens 
20095367Sahrens 	/*
201010588SEric.Taylor@Sun.COM 	 * Mount the target filesystem (if created).  Also mount any
201110588SEric.Taylor@Sun.COM 	 * children of the target filesystem if we did a replication
201210588SEric.Taylor@Sun.COM 	 * receive (indicated by stream_avl being non-NULL).
20135367Sahrens 	 */
20145367Sahrens 	cp = strchr(zc.zc_value, '@');
20155367Sahrens 	if (cp && (ioctl_err == 0 || !newfs)) {
20165367Sahrens 		zfs_handle_t *h;
20175367Sahrens 
20185367Sahrens 		*cp = '\0';
20195367Sahrens 		h = zfs_open(hdl, zc.zc_value,
20205367Sahrens 		    ZFS_TYPE_FILESYSTEM | ZFS_TYPE_VOLUME);
20216476Sahl 		if (h != NULL) {
20225367Sahrens 			if (h->zfs_type == ZFS_TYPE_VOLUME) {
20237366STim.Haley@Sun.COM 				*cp = '@';
202410200SLori.Alt@Sun.COM 			} else if (newfs || stream_avl) {
20257366STim.Haley@Sun.COM 				/*
20267366STim.Haley@Sun.COM 				 * Track the first/top of hierarchy fs,
20277366STim.Haley@Sun.COM 				 * for mounting and sharing later.
20287366STim.Haley@Sun.COM 				 */
20297366STim.Haley@Sun.COM 				if (top_zfs && *top_zfs == NULL)
20307366STim.Haley@Sun.COM 					*top_zfs = zfs_strdup(hdl, zc.zc_value);
20315367Sahrens 			}
20326476Sahl 			zfs_close(h);
20335367Sahrens 		}
20347366STim.Haley@Sun.COM 		*cp = '@';
20355367Sahrens 	}
20365367Sahrens 
20375367Sahrens 	if (clp) {
20385367Sahrens 		err |= changelist_postfix(clp);
20395367Sahrens 		changelist_free(clp);
20405367Sahrens 	}
20415367Sahrens 
20425367Sahrens 	if (err || ioctl_err)
20435367Sahrens 		return (-1);
20445367Sahrens 
20455367Sahrens 	if (flags.verbose) {
20465367Sahrens 		char buf1[64];
20475367Sahrens 		char buf2[64];
20485367Sahrens 		uint64_t bytes = zc.zc_cookie;
20495367Sahrens 		time_t delta = time(NULL) - begin_time;
20505367Sahrens 		if (delta == 0)
20515367Sahrens 			delta = 1;
20525367Sahrens 		zfs_nicenum(bytes, buf1, sizeof (buf1));
20535367Sahrens 		zfs_nicenum(bytes/delta, buf2, sizeof (buf1));
20545367Sahrens 
20555367Sahrens 		(void) printf("received %sB stream in %lu seconds (%sB/sec)\n",
20565367Sahrens 		    buf1, delta, buf2);
20575367Sahrens 	}
20585367Sahrens 
20595367Sahrens 	return (0);
20605367Sahrens }
20615367Sahrens 
20627366STim.Haley@Sun.COM static int
20637366STim.Haley@Sun.COM zfs_receive_impl(libzfs_handle_t *hdl, const char *tosnap, recvflags_t flags,
20647366STim.Haley@Sun.COM     int infd, avl_tree_t *stream_avl, char **top_zfs)
20655367Sahrens {
20665367Sahrens 	int err;
20675367Sahrens 	dmu_replay_record_t drr, drr_noswap;
20685367Sahrens 	struct drr_begin *drrb = &drr.drr_u.drr_begin;
20695367Sahrens 	char errbuf[1024];
20705367Sahrens 	zio_cksum_t zcksum = { 0 };
20715367Sahrens 
20725367Sahrens 	(void) snprintf(errbuf, sizeof (errbuf), dgettext(TEXT_DOMAIN,
20735367Sahrens 	    "cannot receive"));
20745367Sahrens 
20755367Sahrens 	if (flags.isprefix &&
20765367Sahrens 	    !zfs_dataset_exists(hdl, tosnap, ZFS_TYPE_DATASET)) {
20775367Sahrens 		zfs_error_aux(hdl, dgettext(TEXT_DOMAIN, "specified fs "
20785367Sahrens 		    "(%s) does not exist"), tosnap);
20795367Sahrens 		return (zfs_error(hdl, EZFS_NOENT, errbuf));
20805367Sahrens 	}
20815367Sahrens 
20825367Sahrens 	/* read in the BEGIN record */
20835367Sahrens 	if (0 != (err = recv_read(hdl, infd, &drr, sizeof (drr), B_FALSE,
20845367Sahrens 	    &zcksum)))
20855367Sahrens 		return (err);
20865367Sahrens 
20875367Sahrens 	if (drr.drr_type == DRR_END || drr.drr_type == BSWAP_32(DRR_END)) {
20885367Sahrens 		/* It's the double end record at the end of a package */
20895367Sahrens 		return (ENODATA);
20905367Sahrens 	}
20915367Sahrens 
20925367Sahrens 	/* the kernel needs the non-byteswapped begin record */
20935367Sahrens 	drr_noswap = drr;
20945367Sahrens 
20955446Sahrens 	flags.byteswap = B_FALSE;
20965367Sahrens 	if (drrb->drr_magic == BSWAP_64(DMU_BACKUP_MAGIC)) {
20975367Sahrens 		/*
20985367Sahrens 		 * We computed the checksum in the wrong byteorder in
20995367Sahrens 		 * recv_read() above; do it again correctly.
21005367Sahrens 		 */
21015367Sahrens 		bzero(&zcksum, sizeof (zio_cksum_t));
21025367Sahrens 		fletcher_4_incremental_byteswap(&drr, sizeof (drr), &zcksum);
21035367Sahrens 		flags.byteswap = B_TRUE;
21045367Sahrens 
21055367Sahrens 		drr.drr_type = BSWAP_32(drr.drr_type);
21065367Sahrens 		drr.drr_payloadlen = BSWAP_32(drr.drr_payloadlen);
21075367Sahrens 		drrb->drr_magic = BSWAP_64(drrb->drr_magic);
21085367Sahrens 		drrb->drr_version = BSWAP_64(drrb->drr_version);
21095367Sahrens 		drrb->drr_creation_time = BSWAP_64(drrb->drr_creation_time);
21105367Sahrens 		drrb->drr_type = BSWAP_32(drrb->drr_type);
21115367Sahrens 		drrb->drr_flags = BSWAP_32(drrb->drr_flags);
21125367Sahrens 		drrb->drr_toguid = BSWAP_64(drrb->drr_toguid);
21135367Sahrens 		drrb->drr_fromguid = BSWAP_64(drrb->drr_fromguid);
21145367Sahrens 	}
21155367Sahrens 
21165367Sahrens 	if (drrb->drr_magic != DMU_BACKUP_MAGIC || drr.drr_type != DRR_BEGIN) {
21175367Sahrens 		zfs_error_aux(hdl, dgettext(TEXT_DOMAIN, "invalid "
21185367Sahrens 		    "stream (bad magic number)"));
21195367Sahrens 		return (zfs_error(hdl, EZFS_BADSTREAM, errbuf));
21205367Sahrens 	}
21215367Sahrens 
21225367Sahrens 	if (strchr(drrb->drr_toname, '@') == NULL) {
21235367Sahrens 		zfs_error_aux(hdl, dgettext(TEXT_DOMAIN, "invalid "
21245367Sahrens 		    "stream (bad snapshot name)"));
21255367Sahrens 		return (zfs_error(hdl, EZFS_BADSTREAM, errbuf));
21265367Sahrens 	}
21275367Sahrens 
21285367Sahrens 	if (drrb->drr_version == DMU_BACKUP_STREAM_VERSION) {
21295367Sahrens 		return (zfs_receive_one(hdl, infd, tosnap, flags,
21307366STim.Haley@Sun.COM 		    &drr, &drr_noswap, stream_avl, top_zfs));
21315367Sahrens 	} else if (drrb->drr_version == DMU_BACKUP_HEADER_VERSION) {
21325367Sahrens 		return (zfs_receive_package(hdl, infd, tosnap, flags,
21337366STim.Haley@Sun.COM 		    &drr, &zcksum, top_zfs));
21345367Sahrens 	} else {
21355367Sahrens 		zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
21365367Sahrens 		    "stream is unsupported version %llu"),
21375367Sahrens 		    drrb->drr_version);
21385367Sahrens 		return (zfs_error(hdl, EZFS_BADSTREAM, errbuf));
21395367Sahrens 	}
21405367Sahrens }
21417366STim.Haley@Sun.COM 
21427366STim.Haley@Sun.COM /*
21437366STim.Haley@Sun.COM  * Restores a backup of tosnap from the file descriptor specified by infd.
21447366STim.Haley@Sun.COM  * Return 0 on total success, -2 if some things couldn't be
21457366STim.Haley@Sun.COM  * destroyed/renamed/promoted, -1 if some things couldn't be received.
21467366STim.Haley@Sun.COM  * (-1 will override -2).
21477366STim.Haley@Sun.COM  */
21487366STim.Haley@Sun.COM int
21497366STim.Haley@Sun.COM zfs_receive(libzfs_handle_t *hdl, const char *tosnap, recvflags_t flags,
21507366STim.Haley@Sun.COM     int infd, avl_tree_t *stream_avl)
21517366STim.Haley@Sun.COM {
21527366STim.Haley@Sun.COM 	char *top_zfs = NULL;
21537366STim.Haley@Sun.COM 	int err;
21547366STim.Haley@Sun.COM 
21557366STim.Haley@Sun.COM 	err = zfs_receive_impl(hdl, tosnap, flags, infd, stream_avl, &top_zfs);
21567366STim.Haley@Sun.COM 
21578584SLori.Alt@Sun.COM 	if (err == 0 && !flags.nomount && top_zfs) {
21587366STim.Haley@Sun.COM 		zfs_handle_t *zhp;
21597366STim.Haley@Sun.COM 		prop_changelist_t *clp;
21607366STim.Haley@Sun.COM 
21617366STim.Haley@Sun.COM 		zhp = zfs_open(hdl, top_zfs, ZFS_TYPE_FILESYSTEM);
21627366STim.Haley@Sun.COM 		if (zhp != NULL) {
21637366STim.Haley@Sun.COM 			clp = changelist_gather(zhp, ZFS_PROP_MOUNTPOINT,
21647366STim.Haley@Sun.COM 			    CL_GATHER_MOUNT_ALWAYS, 0);
21657366STim.Haley@Sun.COM 			zfs_close(zhp);
21667366STim.Haley@Sun.COM 			if (clp != NULL) {
21677366STim.Haley@Sun.COM 				/* mount and share received datasets */
21687366STim.Haley@Sun.COM 				err = changelist_postfix(clp);
21697366STim.Haley@Sun.COM 				changelist_free(clp);
21707366STim.Haley@Sun.COM 			}
21717366STim.Haley@Sun.COM 		}
21727366STim.Haley@Sun.COM 		if (zhp == NULL || clp == NULL || err)
21737366STim.Haley@Sun.COM 			err = -1;
21747366STim.Haley@Sun.COM 	}
21757366STim.Haley@Sun.COM 	if (top_zfs)
21767366STim.Haley@Sun.COM 		free(top_zfs);
21777366STim.Haley@Sun.COM 
21787366STim.Haley@Sun.COM 	return (err);
21797366STim.Haley@Sun.COM }
2180