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 <libdevinfo.h>
315367Sahrens #include <libintl.h>
325367Sahrens #include <stdio.h>
335367Sahrens #include <stdlib.h>
345367Sahrens #include <strings.h>
355367Sahrens #include <unistd.h>
365367Sahrens #include <stddef.h>
375367Sahrens #include <fcntl.h>
385367Sahrens #include <sys/mount.h>
395367Sahrens #include <sys/mntent.h>
405367Sahrens #include <sys/mnttab.h>
415367Sahrens #include <sys/avl.h>
425367Sahrens #include <stddef.h>
435367Sahrens 
445367Sahrens #include <libzfs.h>
455367Sahrens 
465367Sahrens #include "zfs_namecheck.h"
475367Sahrens #include "zfs_prop.h"
485367Sahrens #include "libzfs_impl.h"
495367Sahrens 
505367Sahrens #include <fletcher.c> /* XXX */
515367Sahrens 
527366STim.Haley@Sun.COM static int zfs_receive_impl(libzfs_handle_t *, const char *, recvflags_t,
537366STim.Haley@Sun.COM     int, avl_tree_t *, char **);
547366STim.Haley@Sun.COM 
555367Sahrens /*
565367Sahrens  * Routines for dealing with the AVL tree of fs-nvlists
575367Sahrens  */
585367Sahrens typedef struct fsavl_node {
595367Sahrens 	avl_node_t fn_node;
605367Sahrens 	nvlist_t *fn_nvfs;
615367Sahrens 	char *fn_snapname;
625367Sahrens 	uint64_t fn_guid;
635367Sahrens } fsavl_node_t;
645367Sahrens 
655367Sahrens static int
665367Sahrens fsavl_compare(const void *arg1, const void *arg2)
675367Sahrens {
685367Sahrens 	const fsavl_node_t *fn1 = arg1;
695367Sahrens 	const fsavl_node_t *fn2 = arg2;
705367Sahrens 
715367Sahrens 	if (fn1->fn_guid > fn2->fn_guid)
725367Sahrens 		return (+1);
735367Sahrens 	else if (fn1->fn_guid < fn2->fn_guid)
745367Sahrens 		return (-1);
755367Sahrens 	else
765367Sahrens 		return (0);
775367Sahrens }
785367Sahrens 
795367Sahrens /*
805367Sahrens  * Given the GUID of a snapshot, find its containing filesystem and
815367Sahrens  * (optionally) name.
825367Sahrens  */
835367Sahrens static nvlist_t *
845367Sahrens fsavl_find(avl_tree_t *avl, uint64_t snapguid, char **snapname)
855367Sahrens {
865367Sahrens 	fsavl_node_t fn_find;
875367Sahrens 	fsavl_node_t *fn;
885367Sahrens 
895367Sahrens 	fn_find.fn_guid = snapguid;
905367Sahrens 
915367Sahrens 	fn = avl_find(avl, &fn_find, NULL);
925367Sahrens 	if (fn) {
935367Sahrens 		if (snapname)
945367Sahrens 			*snapname = fn->fn_snapname;
955367Sahrens 		return (fn->fn_nvfs);
965367Sahrens 	}
975367Sahrens 	return (NULL);
985367Sahrens }
995367Sahrens 
1005896Sahl static void
1015896Sahl fsavl_destroy(avl_tree_t *avl)
1025896Sahl {
1035896Sahl 	fsavl_node_t *fn;
1045896Sahl 	void *cookie;
1055896Sahl 
1065896Sahl 	if (avl == NULL)
1075896Sahl 		return;
1085896Sahl 
1095896Sahl 	cookie = NULL;
1105896Sahl 	while ((fn = avl_destroy_nodes(avl, &cookie)) != NULL)
1115896Sahl 		free(fn);
1125896Sahl 	avl_destroy(avl);
1135896Sahl 	free(avl);
1145896Sahl }
1155896Sahl 
11610200SLori.Alt@Sun.COM /*
11710200SLori.Alt@Sun.COM  * Given an nvlist, produce an avl tree of snapshots, ordered by guid
11810200SLori.Alt@Sun.COM  */
1195367Sahrens static avl_tree_t *
1205367Sahrens fsavl_create(nvlist_t *fss)
1215367Sahrens {
1225367Sahrens 	avl_tree_t *fsavl;
1235367Sahrens 	nvpair_t *fselem = NULL;
1245367Sahrens 
1255896Sahl 	if ((fsavl = malloc(sizeof (avl_tree_t))) == NULL)
1265896Sahl 		return (NULL);
1275896Sahl 
1285367Sahrens 	avl_create(fsavl, fsavl_compare, sizeof (fsavl_node_t),
1295367Sahrens 	    offsetof(fsavl_node_t, fn_node));
1305367Sahrens 
1315367Sahrens 	while ((fselem = nvlist_next_nvpair(fss, fselem)) != NULL) {
1325367Sahrens 		nvlist_t *nvfs, *snaps;
1335367Sahrens 		nvpair_t *snapelem = NULL;
1345367Sahrens 
1355367Sahrens 		VERIFY(0 == nvpair_value_nvlist(fselem, &nvfs));
1365367Sahrens 		VERIFY(0 == nvlist_lookup_nvlist(nvfs, "snaps", &snaps));
1375367Sahrens 
1385367Sahrens 		while ((snapelem =
1395367Sahrens 		    nvlist_next_nvpair(snaps, snapelem)) != NULL) {
1405367Sahrens 			fsavl_node_t *fn;
1415367Sahrens 			uint64_t guid;
1425367Sahrens 
1435367Sahrens 			VERIFY(0 == nvpair_value_uint64(snapelem, &guid));
1445896Sahl 			if ((fn = malloc(sizeof (fsavl_node_t))) == NULL) {
1455896Sahl 				fsavl_destroy(fsavl);
1465896Sahl 				return (NULL);
1475896Sahl 			}
1485367Sahrens 			fn->fn_nvfs = nvfs;
1495367Sahrens 			fn->fn_snapname = nvpair_name(snapelem);
1505367Sahrens 			fn->fn_guid = guid;
1515367Sahrens 
1525367Sahrens 			/*
1535367Sahrens 			 * Note: if there are multiple snaps with the
1545367Sahrens 			 * same GUID, we ignore all but one.
1555367Sahrens 			 */
1565367Sahrens 			if (avl_find(fsavl, fn, NULL) == NULL)
1575367Sahrens 				avl_add(fsavl, fn);
1585367Sahrens 			else
1595367Sahrens 				free(fn);
1605367Sahrens 		}
1615367Sahrens 	}
1625367Sahrens 
1635367Sahrens 	return (fsavl);
1645367Sahrens }
1655367Sahrens 
1665367Sahrens /*
1675367Sahrens  * Routines for dealing with the giant nvlist of fs-nvlists, etc.
1685367Sahrens  */
1695367Sahrens typedef struct send_data {
1705367Sahrens 	uint64_t parent_fromsnap_guid;
1715367Sahrens 	nvlist_t *parent_snaps;
1725367Sahrens 	nvlist_t *fss;
1737265Sahrens 	nvlist_t *snapprops;
1745367Sahrens 	const char *fromsnap;
1755367Sahrens 	const char *tosnap;
1765367Sahrens 
1775367Sahrens 	/*
1785367Sahrens 	 * The header nvlist is of the following format:
1795367Sahrens 	 * {
1805367Sahrens 	 *   "tosnap" -> string
1815367Sahrens 	 *   "fromsnap" -> string (if incremental)
1825367Sahrens 	 *   "fss" -> {
1835367Sahrens 	 *	id -> {
1845367Sahrens 	 *
1855367Sahrens 	 *	 "name" -> string (full name; for debugging)
1865367Sahrens 	 *	 "parentfromsnap" -> number (guid of fromsnap in parent)
1875367Sahrens 	 *
1885367Sahrens 	 *	 "props" -> { name -> value (only if set here) }
1895367Sahrens 	 *	 "snaps" -> { name (lastname) -> number (guid) }
1907265Sahrens 	 *	 "snapprops" -> { name (lastname) -> { name -> value } }
1915367Sahrens 	 *
1925367Sahrens 	 *	 "origin" -> number (guid) (if clone)
1935367Sahrens 	 *	 "sent" -> boolean (not on-disk)
1945367Sahrens 	 *	}
1955367Sahrens 	 *   }
1965367Sahrens 	 * }
1975367Sahrens 	 *
1985367Sahrens 	 */
1995367Sahrens } send_data_t;
2005367Sahrens 
2017265Sahrens static void send_iterate_prop(zfs_handle_t *zhp, nvlist_t *nv);
2027265Sahrens 
2035367Sahrens static int
2045367Sahrens send_iterate_snap(zfs_handle_t *zhp, void *arg)
2055367Sahrens {
2065367Sahrens 	send_data_t *sd = arg;
2075367Sahrens 	uint64_t guid = zhp->zfs_dmustats.dds_guid;
2085367Sahrens 	char *snapname;
2097265Sahrens 	nvlist_t *nv;
2105367Sahrens 
2115367Sahrens 	snapname = strrchr(zhp->zfs_name, '@')+1;
2125367Sahrens 
2135367Sahrens 	VERIFY(0 == nvlist_add_uint64(sd->parent_snaps, snapname, guid));
2145367Sahrens 	/*
2155367Sahrens 	 * NB: if there is no fromsnap here (it's a newly created fs in
2165367Sahrens 	 * an incremental replication), we will substitute the tosnap.
2175367Sahrens 	 */
2185367Sahrens 	if ((sd->fromsnap && strcmp(snapname, sd->fromsnap) == 0) ||
2195367Sahrens 	    (sd->parent_fromsnap_guid == 0 && sd->tosnap &&
2205367Sahrens 	    strcmp(snapname, sd->tosnap) == 0)) {
2215367Sahrens 		sd->parent_fromsnap_guid = guid;
2225367Sahrens 	}
2235367Sahrens 
2247265Sahrens 	VERIFY(0 == nvlist_alloc(&nv, NV_UNIQUE_NAME, 0));
2257265Sahrens 	send_iterate_prop(zhp, nv);
2267265Sahrens 	VERIFY(0 == nvlist_add_nvlist(sd->snapprops, snapname, nv));
2277265Sahrens 	nvlist_free(nv);
2287265Sahrens 
2295367Sahrens 	zfs_close(zhp);
2305367Sahrens 	return (0);
2315367Sahrens }
2325367Sahrens 
2335367Sahrens static void
2345367Sahrens send_iterate_prop(zfs_handle_t *zhp, nvlist_t *nv)
2355367Sahrens {
2365367Sahrens 	nvpair_t *elem = NULL;
2375367Sahrens 
2385367Sahrens 	while ((elem = nvlist_next_nvpair(zhp->zfs_props, elem)) != NULL) {
2395367Sahrens 		char *propname = nvpair_name(elem);
2405367Sahrens 		zfs_prop_t prop = zfs_name_to_prop(propname);
2415367Sahrens 		nvlist_t *propnv;
2425367Sahrens 
2439396SMatthew.Ahrens@Sun.COM 		assert(zfs_prop_user(propname) || prop != ZPROP_INVAL);
2449396SMatthew.Ahrens@Sun.COM 
2455367Sahrens 		if (!zfs_prop_user(propname) && zfs_prop_readonly(prop))
2465367Sahrens 			continue;
2475367Sahrens 
2485367Sahrens 		verify(nvpair_value_nvlist(elem, &propnv) == 0);
24910195SSanjeev.Bagewadi@Sun.COM 		if (prop == ZFS_PROP_QUOTA || prop == ZFS_PROP_RESERVATION ||
25010195SSanjeev.Bagewadi@Sun.COM 		    prop == ZFS_PROP_REFQUOTA ||
25110195SSanjeev.Bagewadi@Sun.COM 		    prop == ZFS_PROP_REFRESERVATION) {
2525367Sahrens 			/* these guys are modifyable, but have no source */
2535367Sahrens 			uint64_t value;
2545367Sahrens 			verify(nvlist_lookup_uint64(propnv,
2555367Sahrens 			    ZPROP_VALUE, &value) == 0);
2567265Sahrens 			if (zhp->zfs_type == ZFS_TYPE_SNAPSHOT)
2577265Sahrens 				continue;
2585367Sahrens 		} else {
2595367Sahrens 			char *source;
2605367Sahrens 			if (nvlist_lookup_string(propnv,
2615367Sahrens 			    ZPROP_SOURCE, &source) != 0)
2625367Sahrens 				continue;
2635367Sahrens 			if (strcmp(source, zhp->zfs_name) != 0)
2645367Sahrens 				continue;
2655367Sahrens 		}
2665367Sahrens 
2675367Sahrens 		if (zfs_prop_user(propname) ||
2685367Sahrens 		    zfs_prop_get_type(prop) == PROP_TYPE_STRING) {
2695367Sahrens 			char *value;
2705367Sahrens 			verify(nvlist_lookup_string(propnv,
2715367Sahrens 			    ZPROP_VALUE, &value) == 0);
2725367Sahrens 			VERIFY(0 == nvlist_add_string(nv, propname, value));
2735367Sahrens 		} else {
2745367Sahrens 			uint64_t value;
2755367Sahrens 			verify(nvlist_lookup_uint64(propnv,
2765367Sahrens 			    ZPROP_VALUE, &value) == 0);
2775367Sahrens 			VERIFY(0 == nvlist_add_uint64(nv, propname, value));
2785367Sahrens 		}
2795367Sahrens 	}
2805367Sahrens }
2815367Sahrens 
28210200SLori.Alt@Sun.COM /*
28310200SLori.Alt@Sun.COM  * recursively generate nvlists describing datasets.  See comment
28410200SLori.Alt@Sun.COM  * for the data structure send_data_t above for description of contents
28510200SLori.Alt@Sun.COM  * of the nvlist.
28610200SLori.Alt@Sun.COM  */
2875367Sahrens static int
2885367Sahrens send_iterate_fs(zfs_handle_t *zhp, void *arg)
2895367Sahrens {
2905367Sahrens 	send_data_t *sd = arg;
2915367Sahrens 	nvlist_t *nvfs, *nv;
2925367Sahrens 	int rv;
2935367Sahrens 	uint64_t parent_fromsnap_guid_save = sd->parent_fromsnap_guid;
2945367Sahrens 	uint64_t guid = zhp->zfs_dmustats.dds_guid;
2955367Sahrens 	char guidstring[64];
2965367Sahrens 
2975367Sahrens 	VERIFY(0 == nvlist_alloc(&nvfs, NV_UNIQUE_NAME, 0));
2985367Sahrens 	VERIFY(0 == nvlist_add_string(nvfs, "name", zhp->zfs_name));
2995367Sahrens 	VERIFY(0 == nvlist_add_uint64(nvfs, "parentfromsnap",
3005367Sahrens 	    sd->parent_fromsnap_guid));
3015367Sahrens 
3025367Sahrens 	if (zhp->zfs_dmustats.dds_origin[0]) {
3035367Sahrens 		zfs_handle_t *origin = zfs_open(zhp->zfs_hdl,
3045367Sahrens 		    zhp->zfs_dmustats.dds_origin, ZFS_TYPE_SNAPSHOT);
3055367Sahrens 		if (origin == NULL)
3065367Sahrens 			return (-1);
3075367Sahrens 		VERIFY(0 == nvlist_add_uint64(nvfs, "origin",
3085367Sahrens 		    origin->zfs_dmustats.dds_guid));
3095367Sahrens 	}
3105367Sahrens 
3115367Sahrens 	/* iterate over props */
3125367Sahrens 	VERIFY(0 == nvlist_alloc(&nv, NV_UNIQUE_NAME, 0));
3135367Sahrens 	send_iterate_prop(zhp, nv);
3145367Sahrens 	VERIFY(0 == nvlist_add_nvlist(nvfs, "props", nv));
3155367Sahrens 	nvlist_free(nv);
3165367Sahrens 
3175367Sahrens 	/* iterate over snaps, and set sd->parent_fromsnap_guid */
3185367Sahrens 	sd->parent_fromsnap_guid = 0;
3195367Sahrens 	VERIFY(0 == nvlist_alloc(&sd->parent_snaps, NV_UNIQUE_NAME, 0));
3207265Sahrens 	VERIFY(0 == nvlist_alloc(&sd->snapprops, NV_UNIQUE_NAME, 0));
3215367Sahrens 	(void) zfs_iter_snapshots(zhp, send_iterate_snap, sd);
3225367Sahrens 	VERIFY(0 == nvlist_add_nvlist(nvfs, "snaps", sd->parent_snaps));
3237265Sahrens 	VERIFY(0 == nvlist_add_nvlist(nvfs, "snapprops", sd->snapprops));
3245367Sahrens 	nvlist_free(sd->parent_snaps);
3257265Sahrens 	nvlist_free(sd->snapprops);
3265367Sahrens 
3275367Sahrens 	/* add this fs to nvlist */
3285367Sahrens 	(void) snprintf(guidstring, sizeof (guidstring),
3295367Sahrens 	    "0x%llx", (longlong_t)guid);
3305367Sahrens 	VERIFY(0 == nvlist_add_nvlist(sd->fss, guidstring, nvfs));
3315367Sahrens 	nvlist_free(nvfs);
3325367Sahrens 
3335367Sahrens 	/* iterate over children */
3345367Sahrens 	rv = zfs_iter_filesystems(zhp, send_iterate_fs, sd);
3355367Sahrens 
3365367Sahrens 	sd->parent_fromsnap_guid = parent_fromsnap_guid_save;
3375367Sahrens 
3385367Sahrens 	zfs_close(zhp);
3395367Sahrens 	return (rv);
3405367Sahrens }
3415367Sahrens 
3425367Sahrens static int
3435367Sahrens gather_nvlist(libzfs_handle_t *hdl, const char *fsname, const char *fromsnap,
3445367Sahrens     const char *tosnap, nvlist_t **nvlp, avl_tree_t **avlp)
3455367Sahrens {
3465367Sahrens 	zfs_handle_t *zhp;
3475367Sahrens 	send_data_t sd = { 0 };
3485367Sahrens 	int error;
3495367Sahrens 
3505367Sahrens 	zhp = zfs_open(hdl, fsname, ZFS_TYPE_FILESYSTEM | ZFS_TYPE_VOLUME);
3515367Sahrens 	if (zhp == NULL)
3525367Sahrens 		return (EZFS_BADTYPE);
3535367Sahrens 
3545367Sahrens 	VERIFY(0 == nvlist_alloc(&sd.fss, NV_UNIQUE_NAME, 0));
3555367Sahrens 	sd.fromsnap = fromsnap;
3565367Sahrens 	sd.tosnap = tosnap;
3575896Sahl 
3585896Sahl 	if ((error = send_iterate_fs(zhp, &sd)) != 0) {
3595896Sahl 		nvlist_free(sd.fss);
3605896Sahl 		if (avlp != NULL)
3615896Sahl 			*avlp = NULL;
3625896Sahl 		*nvlp = NULL;
3635896Sahl 		return (error);
3645896Sahl 	}
3655896Sahl 
3665896Sahl 	if (avlp != NULL && (*avlp = fsavl_create(sd.fss)) == NULL) {
3675896Sahl 		nvlist_free(sd.fss);
3685896Sahl 		*nvlp = NULL;
3695896Sahl 		return (EZFS_NOMEM);
3705896Sahl 	}
3715367Sahrens 
3725367Sahrens 	*nvlp = sd.fss;
3735896Sahl 	return (0);
3745367Sahrens }
3755367Sahrens 
3765367Sahrens /*
3775367Sahrens  * Routines for dealing with the sorted snapshot functionality
3785367Sahrens  */
3795367Sahrens typedef struct zfs_node {
3805367Sahrens 	zfs_handle_t	*zn_handle;
3815367Sahrens 	avl_node_t	zn_avlnode;
3825367Sahrens } zfs_node_t;
3835367Sahrens 
3845367Sahrens static int
3855367Sahrens zfs_sort_snaps(zfs_handle_t *zhp, void *data)
3865367Sahrens {
3875367Sahrens 	avl_tree_t *avl = data;
3885367Sahrens 	zfs_node_t *node = zfs_alloc(zhp->zfs_hdl, sizeof (zfs_node_t));
3895367Sahrens 
3905367Sahrens 	node->zn_handle = zhp;
3915367Sahrens 	avl_add(avl, node);
3925367Sahrens 	return (0);
3935367Sahrens }
3945367Sahrens 
3955367Sahrens /* ARGSUSED */
3965367Sahrens static int
3975367Sahrens zfs_snapshot_compare(const void *larg, const void *rarg)
3985367Sahrens {
3995367Sahrens 	zfs_handle_t *l = ((zfs_node_t *)larg)->zn_handle;
4005367Sahrens 	zfs_handle_t *r = ((zfs_node_t *)rarg)->zn_handle;
4015367Sahrens 	uint64_t lcreate, rcreate;
4025367Sahrens 
4035367Sahrens 	/*
4045367Sahrens 	 * Sort them according to creation time.  We use the hidden
4055367Sahrens 	 * CREATETXG property to get an absolute ordering of snapshots.
4065367Sahrens 	 */
4075367Sahrens 	lcreate = zfs_prop_get_int(l, ZFS_PROP_CREATETXG);
4085367Sahrens 	rcreate = zfs_prop_get_int(r, ZFS_PROP_CREATETXG);
4095367Sahrens 
4105367Sahrens 	if (lcreate < rcreate)
4115367Sahrens 		return (-1);
4125367Sahrens 	else if (lcreate > rcreate)
4135367Sahrens 		return (+1);
4145367Sahrens 	else
4155367Sahrens 		return (0);
4165367Sahrens }
4175367Sahrens 
4185367Sahrens static int
4195367Sahrens zfs_iter_snapshots_sorted(zfs_handle_t *zhp, zfs_iter_f callback, void *data)
4205367Sahrens {
4215367Sahrens 	int ret = 0;
4225367Sahrens 	zfs_node_t *node;
4235367Sahrens 	avl_tree_t avl;
4245367Sahrens 	void *cookie = NULL;
4255367Sahrens 
4265367Sahrens 	avl_create(&avl, zfs_snapshot_compare,
4275367Sahrens 	    sizeof (zfs_node_t), offsetof(zfs_node_t, zn_avlnode));
4285367Sahrens 
4295367Sahrens 	ret = zfs_iter_snapshots(zhp, zfs_sort_snaps, &avl);
4305367Sahrens 
4315367Sahrens 	for (node = avl_first(&avl); node != NULL; node = AVL_NEXT(&avl, node))
4325367Sahrens 		ret |= callback(node->zn_handle, data);
4335367Sahrens 
4345367Sahrens 	while ((node = avl_destroy_nodes(&avl, &cookie)) != NULL)
4355367Sahrens 		free(node);
4365367Sahrens 
4375367Sahrens 	avl_destroy(&avl);
4385367Sahrens 
4395367Sahrens 	return (ret);
4405367Sahrens }
4415367Sahrens 
4425367Sahrens /*
4435367Sahrens  * Routines specific to "zfs send"
4445367Sahrens  */
4455367Sahrens typedef struct send_dump_data {
4465367Sahrens 	/* these are all just the short snapname (the part after the @) */
4475367Sahrens 	const char *fromsnap;
4485367Sahrens 	const char *tosnap;
4495367Sahrens 	char lastsnap[ZFS_MAXNAMELEN];
4505367Sahrens 	boolean_t seenfrom, seento, replicate, doall, fromorigin;
4515367Sahrens 	boolean_t verbose;
4525367Sahrens 	int outfd;
4535367Sahrens 	boolean_t err;
4545367Sahrens 	nvlist_t *fss;
4555367Sahrens 	avl_tree_t *fsavl;
4565367Sahrens } send_dump_data_t;
4575367Sahrens 
4585367Sahrens /*
4595367Sahrens  * Dumps a backup of the given snapshot (incremental from fromsnap if it's not
4605367Sahrens  * NULL) to the file descriptor specified by outfd.
4615367Sahrens  */
4625367Sahrens static int
4635367Sahrens dump_ioctl(zfs_handle_t *zhp, const char *fromsnap, boolean_t fromorigin,
4645367Sahrens     int outfd)
4655367Sahrens {
4665367Sahrens 	zfs_cmd_t zc = { 0 };
4675367Sahrens 	libzfs_handle_t *hdl = zhp->zfs_hdl;
4685367Sahrens 
4695367Sahrens 	assert(zhp->zfs_type == ZFS_TYPE_SNAPSHOT);
4705367Sahrens 	assert(fromsnap == NULL || fromsnap[0] == '\0' || !fromorigin);
4715367Sahrens 
4725367Sahrens 	(void) strlcpy(zc.zc_name, zhp->zfs_name, sizeof (zc.zc_name));
4735367Sahrens 	if (fromsnap)
4745765Sek110237 		(void) strlcpy(zc.zc_value, fromsnap, sizeof (zc.zc_value));
4755367Sahrens 	zc.zc_cookie = outfd;
4765367Sahrens 	zc.zc_obj = fromorigin;
4775367Sahrens 
4785367Sahrens 	if (ioctl(zhp->zfs_hdl->libzfs_fd, ZFS_IOC_SEND, &zc) != 0) {
4795367Sahrens 		char errbuf[1024];
4805367Sahrens 		(void) snprintf(errbuf, sizeof (errbuf), dgettext(TEXT_DOMAIN,
4815367Sahrens 		    "warning: cannot send '%s'"), zhp->zfs_name);
4825367Sahrens 
4835367Sahrens 		switch (errno) {
4845367Sahrens 
4855367Sahrens 		case EXDEV:
4865367Sahrens 			zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
4875367Sahrens 			    "not an earlier snapshot from the same fs"));
4885367Sahrens 			return (zfs_error(hdl, EZFS_CROSSTARGET, errbuf));
4895367Sahrens 
4905367Sahrens 		case ENOENT:
4915367Sahrens 			if (zfs_dataset_exists(hdl, zc.zc_name,
4925367Sahrens 			    ZFS_TYPE_SNAPSHOT)) {
4935367Sahrens 				zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
4945367Sahrens 				    "incremental source (@%s) does not exist"),
4955367Sahrens 				    zc.zc_value);
4965367Sahrens 			}
4975367Sahrens 			return (zfs_error(hdl, EZFS_NOENT, errbuf));
4985367Sahrens 
4995367Sahrens 		case EDQUOT:
5005367Sahrens 		case EFBIG:
5015367Sahrens 		case EIO:
5025367Sahrens 		case ENOLINK:
5035367Sahrens 		case ENOSPC:
5045367Sahrens 		case ENOSTR:
5055367Sahrens 		case ENXIO:
5065367Sahrens 		case EPIPE:
5075367Sahrens 		case ERANGE:
5085367Sahrens 		case EFAULT:
5095367Sahrens 		case EROFS:
5105367Sahrens 			zfs_error_aux(hdl, strerror(errno));
5115367Sahrens 			return (zfs_error(hdl, EZFS_BADBACKUP, errbuf));
5125367Sahrens 
5135367Sahrens 		default:
5145367Sahrens 			return (zfs_standard_error(hdl, errno, errbuf));
5155367Sahrens 		}
5165367Sahrens 	}
5175367Sahrens 
5185367Sahrens 	return (0);
5195367Sahrens }
5205367Sahrens 
5215367Sahrens static int
5225367Sahrens dump_snapshot(zfs_handle_t *zhp, void *arg)
5235367Sahrens {
5245367Sahrens 	send_dump_data_t *sdd = arg;
5255367Sahrens 	const char *thissnap;
5265367Sahrens 	int err;
5275367Sahrens 
5285367Sahrens 	thissnap = strchr(zhp->zfs_name, '@') + 1;
5295367Sahrens 
5305367Sahrens 	if (sdd->fromsnap && !sdd->seenfrom &&
5315367Sahrens 	    strcmp(sdd->fromsnap, thissnap) == 0) {
5325367Sahrens 		sdd->seenfrom = B_TRUE;
5335367Sahrens 		(void) strcpy(sdd->lastsnap, thissnap);
5345367Sahrens 		zfs_close(zhp);
5355367Sahrens 		return (0);
5365367Sahrens 	}
5375367Sahrens 
5385367Sahrens 	if (sdd->seento || !sdd->seenfrom) {
5395367Sahrens 		zfs_close(zhp);
5405367Sahrens 		return (0);
5415367Sahrens 	}
5425367Sahrens 
5435367Sahrens 	/* send it */
5445367Sahrens 	if (sdd->verbose) {
5455367Sahrens 		(void) fprintf(stderr, "sending from @%s to %s\n",
5465367Sahrens 		    sdd->lastsnap, zhp->zfs_name);
5475367Sahrens 	}
5485367Sahrens 
5495367Sahrens 	err = dump_ioctl(zhp, sdd->lastsnap,
5505367Sahrens 	    sdd->lastsnap[0] == '\0' && (sdd->fromorigin || sdd->replicate),
5515367Sahrens 	    sdd->outfd);
5525367Sahrens 
5535367Sahrens 	if (!sdd->seento && strcmp(sdd->tosnap, thissnap) == 0)
5545367Sahrens 		sdd->seento = B_TRUE;
5555367Sahrens 
5565367Sahrens 	(void) strcpy(sdd->lastsnap, thissnap);
5575367Sahrens 	zfs_close(zhp);
5585367Sahrens 	return (err);
5595367Sahrens }
5605367Sahrens 
5615367Sahrens static int
5625367Sahrens dump_filesystem(zfs_handle_t *zhp, void *arg)
5635367Sahrens {
5645367Sahrens 	int rv = 0;
5655367Sahrens 	send_dump_data_t *sdd = arg;
5665367Sahrens 	boolean_t missingfrom = B_FALSE;
5675367Sahrens 	zfs_cmd_t zc = { 0 };
5685367Sahrens 
5695367Sahrens 	(void) snprintf(zc.zc_name, sizeof (zc.zc_name), "%s@%s",
5705367Sahrens 	    zhp->zfs_name, sdd->tosnap);
5715367Sahrens 	if (ioctl(zhp->zfs_hdl->libzfs_fd, ZFS_IOC_OBJSET_STATS, &zc) != 0) {
5725367Sahrens 		(void) fprintf(stderr, "WARNING: "
5735367Sahrens 		    "could not send %s@%s: does not exist\n",
5745367Sahrens 		    zhp->zfs_name, sdd->tosnap);
5755367Sahrens 		sdd->err = B_TRUE;
5765367Sahrens 		return (0);
5775367Sahrens 	}
5785367Sahrens 
5795367Sahrens 	if (sdd->replicate && sdd->fromsnap) {
5805367Sahrens 		/*
5815367Sahrens 		 * If this fs does not have fromsnap, and we're doing
5825367Sahrens 		 * recursive, we need to send a full stream from the
5835367Sahrens 		 * beginning (or an incremental from the origin if this
5845367Sahrens 		 * is a clone).  If we're doing non-recursive, then let
5855367Sahrens 		 * them get the error.
5865367Sahrens 		 */
5875367Sahrens 		(void) snprintf(zc.zc_name, sizeof (zc.zc_name), "%s@%s",
5885367Sahrens 		    zhp->zfs_name, sdd->fromsnap);
5895367Sahrens 		if (ioctl(zhp->zfs_hdl->libzfs_fd,
5905367Sahrens 		    ZFS_IOC_OBJSET_STATS, &zc) != 0) {
5915367Sahrens 			missingfrom = B_TRUE;
5925367Sahrens 		}
5935367Sahrens 	}
5945367Sahrens 
5955367Sahrens 	if (sdd->doall) {
5965367Sahrens 		sdd->seenfrom = sdd->seento = sdd->lastsnap[0] = 0;
5975367Sahrens 		if (sdd->fromsnap == NULL || missingfrom)
5985367Sahrens 			sdd->seenfrom = B_TRUE;
5995367Sahrens 
6005367Sahrens 		rv = zfs_iter_snapshots_sorted(zhp, dump_snapshot, arg);
6015367Sahrens 		if (!sdd->seenfrom) {
6025367Sahrens 			(void) fprintf(stderr,
6035367Sahrens 			    "WARNING: could not send %s@%s:\n"
6045367Sahrens 			    "incremental source (%s@%s) does not exist\n",
6055367Sahrens 			    zhp->zfs_name, sdd->tosnap,
6065367Sahrens 			    zhp->zfs_name, sdd->fromsnap);
6075367Sahrens 			sdd->err = B_TRUE;
6085367Sahrens 		} else if (!sdd->seento) {
6099396SMatthew.Ahrens@Sun.COM 			if (sdd->fromsnap) {
6109396SMatthew.Ahrens@Sun.COM 				(void) fprintf(stderr,
6119396SMatthew.Ahrens@Sun.COM 				    "WARNING: could not send %s@%s:\n"
6129396SMatthew.Ahrens@Sun.COM 				    "incremental source (%s@%s) "
6139396SMatthew.Ahrens@Sun.COM 				    "is not earlier than it\n",
6149396SMatthew.Ahrens@Sun.COM 				    zhp->zfs_name, sdd->tosnap,
6159396SMatthew.Ahrens@Sun.COM 				    zhp->zfs_name, sdd->fromsnap);
6169396SMatthew.Ahrens@Sun.COM 			} else {
6179396SMatthew.Ahrens@Sun.COM 				(void) fprintf(stderr, "WARNING: "
6189396SMatthew.Ahrens@Sun.COM 				    "could not send %s@%s: does not exist\n",
6199396SMatthew.Ahrens@Sun.COM 				    zhp->zfs_name, sdd->tosnap);
6209396SMatthew.Ahrens@Sun.COM 			}
6215367Sahrens 			sdd->err = B_TRUE;
6225367Sahrens 		}
6235367Sahrens 	} else {
6245367Sahrens 		zfs_handle_t *snapzhp;
6255367Sahrens 		char snapname[ZFS_MAXNAMELEN];
6265367Sahrens 
6275367Sahrens 		(void) snprintf(snapname, sizeof (snapname), "%s@%s",
6285367Sahrens 		    zfs_get_name(zhp), sdd->tosnap);
6295367Sahrens 		snapzhp = zfs_open(zhp->zfs_hdl, snapname, ZFS_TYPE_SNAPSHOT);
6306476Sahl 		if (snapzhp == NULL) {
6316476Sahl 			rv = -1;
6326476Sahl 		} else {
6336476Sahl 			rv = dump_ioctl(snapzhp,
6346476Sahl 			    missingfrom ? NULL : sdd->fromsnap,
6356476Sahl 			    sdd->fromorigin || missingfrom,
6366476Sahl 			    sdd->outfd);
6376476Sahl 			sdd->seento = B_TRUE;
6386476Sahl 			zfs_close(snapzhp);
6396476Sahl 		}
6405367Sahrens 	}
6415367Sahrens 
6425367Sahrens 	return (rv);
6435367Sahrens }
6445367Sahrens 
6455367Sahrens static int
6465367Sahrens dump_filesystems(zfs_handle_t *rzhp, void *arg)
6475367Sahrens {
6485367Sahrens 	send_dump_data_t *sdd = arg;
6495367Sahrens 	nvpair_t *fspair;
6505367Sahrens 	boolean_t needagain, progress;
6515367Sahrens 
6525367Sahrens 	if (!sdd->replicate)
6535367Sahrens 		return (dump_filesystem(rzhp, sdd));
6545367Sahrens 
6555367Sahrens again:
6565367Sahrens 	needagain = progress = B_FALSE;
6575367Sahrens 	for (fspair = nvlist_next_nvpair(sdd->fss, NULL); fspair;
6585367Sahrens 	    fspair = nvlist_next_nvpair(sdd->fss, fspair)) {
6595367Sahrens 		nvlist_t *fslist;
6605367Sahrens 		char *fsname;
6615367Sahrens 		zfs_handle_t *zhp;
6625367Sahrens 		int err;
6635367Sahrens 		uint64_t origin_guid = 0;
6645367Sahrens 		nvlist_t *origin_nv;
6655367Sahrens 
6665367Sahrens 		VERIFY(nvpair_value_nvlist(fspair, &fslist) == 0);
6675367Sahrens 		if (nvlist_lookup_boolean(fslist, "sent") == 0)
6685367Sahrens 			continue;
6695367Sahrens 
6705367Sahrens 		VERIFY(nvlist_lookup_string(fslist, "name", &fsname) == 0);
6715367Sahrens 		(void) nvlist_lookup_uint64(fslist, "origin", &origin_guid);
6725367Sahrens 
6735367Sahrens 		origin_nv = fsavl_find(sdd->fsavl, origin_guid, NULL);
6745367Sahrens 		if (origin_nv &&
6755367Sahrens 		    nvlist_lookup_boolean(origin_nv, "sent") == ENOENT) {
6765367Sahrens 			/*
6775367Sahrens 			 * origin has not been sent yet;
6785367Sahrens 			 * skip this clone.
6795367Sahrens 			 */
6805367Sahrens 			needagain = B_TRUE;
6815367Sahrens 			continue;
6825367Sahrens 		}
6835367Sahrens 
6845367Sahrens 		zhp = zfs_open(rzhp->zfs_hdl, fsname, ZFS_TYPE_DATASET);
6856476Sahl 		if (zhp == NULL)
6866476Sahl 			return (-1);
6875367Sahrens 		err = dump_filesystem(zhp, sdd);
6885367Sahrens 		VERIFY(nvlist_add_boolean(fslist, "sent") == 0);
6895367Sahrens 		progress = B_TRUE;
6905367Sahrens 		zfs_close(zhp);
6915367Sahrens 		if (err)
6925367Sahrens 			return (err);
6935367Sahrens 	}
6945367Sahrens 	if (needagain) {
6955367Sahrens 		assert(progress);
6965367Sahrens 		goto again;
6975367Sahrens 	}
6985367Sahrens 	return (0);
6995367Sahrens }
7005367Sahrens 
7015367Sahrens /*
70210200SLori.Alt@Sun.COM  * Generate a send stream for the dataset identified by the argument zhp.
70310200SLori.Alt@Sun.COM  *
70410200SLori.Alt@Sun.COM  * The content of the send stream is the snapshot identified by
70510200SLori.Alt@Sun.COM  * 'tosnap'.  Incremental streams are requested in two ways:
70610200SLori.Alt@Sun.COM  *     - from the snapshot identified by "fromsnap" (if non-null) or
70710200SLori.Alt@Sun.COM  *     - from the origin of the dataset identified by zhp, which must
70810200SLori.Alt@Sun.COM  *	 be a clone.  In this case, "fromsnap" is null and "fromorigin"
70910200SLori.Alt@Sun.COM  *	 is TRUE.
71010200SLori.Alt@Sun.COM  *
71110200SLori.Alt@Sun.COM  * The send stream is recursive (i.e. dumps a hierarchy of snapshots) and
71210200SLori.Alt@Sun.COM  * uses a special header (with a version field of DMU_BACKUP_HEADER_VERSION)
71310200SLori.Alt@Sun.COM  * if "replicate" is set.  If "doall" is set, dump all the intermediate
71410200SLori.Alt@Sun.COM  * snapshots. The DMU_BACKUP_HEADER_VERSION header is used in the "doall"
71510200SLori.Alt@Sun.COM  * case too.
7165367Sahrens  */
7175367Sahrens int
7185367Sahrens zfs_send(zfs_handle_t *zhp, const char *fromsnap, const char *tosnap,
7195367Sahrens     boolean_t replicate, boolean_t doall, boolean_t fromorigin,
7205367Sahrens     boolean_t verbose, int outfd)
7215367Sahrens {
7225367Sahrens 	char errbuf[1024];
7235367Sahrens 	send_dump_data_t sdd = { 0 };
7245367Sahrens 	int err;
7255367Sahrens 	nvlist_t *fss = NULL;
7265367Sahrens 	avl_tree_t *fsavl = NULL;
7275367Sahrens 
7285367Sahrens 	(void) snprintf(errbuf, sizeof (errbuf), dgettext(TEXT_DOMAIN,
7295367Sahrens 	    "cannot send '%s'"), zhp->zfs_name);
7305367Sahrens 
7315367Sahrens 	if (fromsnap && fromsnap[0] == '\0') {
7325367Sahrens 		zfs_error_aux(zhp->zfs_hdl, dgettext(TEXT_DOMAIN,
7335367Sahrens 		    "zero-length incremental source"));
7345367Sahrens 		return (zfs_error(zhp->zfs_hdl, EZFS_NOENT, errbuf));
7355367Sahrens 	}
7365367Sahrens 
7375367Sahrens 	if (replicate || doall) {
7385367Sahrens 		dmu_replay_record_t drr = { 0 };
7395367Sahrens 		char *packbuf = NULL;
7405367Sahrens 		size_t buflen = 0;
7415367Sahrens 		zio_cksum_t zc = { 0 };
7425367Sahrens 
7435367Sahrens 		assert(fromsnap || doall);
7445367Sahrens 
7455367Sahrens 		if (replicate) {
7465367Sahrens 			nvlist_t *hdrnv;
7475367Sahrens 
7485367Sahrens 			VERIFY(0 == nvlist_alloc(&hdrnv, NV_UNIQUE_NAME, 0));
7495367Sahrens 			if (fromsnap) {
7505367Sahrens 				VERIFY(0 == nvlist_add_string(hdrnv,
7515367Sahrens 				    "fromsnap", fromsnap));
7525367Sahrens 			}
7535367Sahrens 			VERIFY(0 == nvlist_add_string(hdrnv, "tosnap", tosnap));
7545367Sahrens 
7555367Sahrens 			err = gather_nvlist(zhp->zfs_hdl, zhp->zfs_name,
7565367Sahrens 			    fromsnap, tosnap, &fss, &fsavl);
7575367Sahrens 			if (err)
7585367Sahrens 				return (err);
7595367Sahrens 			VERIFY(0 == nvlist_add_nvlist(hdrnv, "fss", fss));
7605367Sahrens 			err = nvlist_pack(hdrnv, &packbuf, &buflen,
7615367Sahrens 			    NV_ENCODE_XDR, 0);
7625367Sahrens 			nvlist_free(hdrnv);
7635367Sahrens 			if (err) {
7645367Sahrens 				fsavl_destroy(fsavl);
7655367Sahrens 				nvlist_free(fss);
7665367Sahrens 				return (zfs_standard_error(zhp->zfs_hdl,
7675367Sahrens 				    err, errbuf));
7685367Sahrens 			}
7695367Sahrens 		}
7705367Sahrens 
7715367Sahrens 		/* write first begin record */
7725367Sahrens 		drr.drr_type = DRR_BEGIN;
7735367Sahrens 		drr.drr_u.drr_begin.drr_magic = DMU_BACKUP_MAGIC;
7745367Sahrens 		drr.drr_u.drr_begin.drr_version = DMU_BACKUP_HEADER_VERSION;
7755367Sahrens 		(void) snprintf(drr.drr_u.drr_begin.drr_toname,
7765367Sahrens 		    sizeof (drr.drr_u.drr_begin.drr_toname),
7775367Sahrens 		    "%s@%s", zhp->zfs_name, tosnap);
7785367Sahrens 		drr.drr_payloadlen = buflen;
7795367Sahrens 		fletcher_4_incremental_native(&drr, sizeof (drr), &zc);
7805367Sahrens 		err = write(outfd, &drr, sizeof (drr));
7815367Sahrens 
7825367Sahrens 		/* write header nvlist */
7835367Sahrens 		if (err != -1) {
7845367Sahrens 			fletcher_4_incremental_native(packbuf, buflen, &zc);
7855367Sahrens 			err = write(outfd, packbuf, buflen);
7865367Sahrens 		}
7875367Sahrens 		free(packbuf);
7885367Sahrens 		if (err == -1) {
7895367Sahrens 			fsavl_destroy(fsavl);
7905367Sahrens 			nvlist_free(fss);
7915367Sahrens 			return (zfs_standard_error(zhp->zfs_hdl,
7925367Sahrens 			    errno, errbuf));
7935367Sahrens 		}
7945367Sahrens 
7955367Sahrens 		/* write end record */
7965367Sahrens 		if (err != -1) {
7975367Sahrens 			bzero(&drr, sizeof (drr));
7985367Sahrens 			drr.drr_type = DRR_END;
7995367Sahrens 			drr.drr_u.drr_end.drr_checksum = zc;
8005367Sahrens 			err = write(outfd, &drr, sizeof (drr));
8015367Sahrens 			if (err == -1) {
8025367Sahrens 				fsavl_destroy(fsavl);
8035367Sahrens 				nvlist_free(fss);
8045367Sahrens 				return (zfs_standard_error(zhp->zfs_hdl,
8055367Sahrens 				    errno, errbuf));
8065367Sahrens 			}
8075367Sahrens 		}
8085367Sahrens 	}
8095367Sahrens 
8105367Sahrens 	/* dump each stream */
8115367Sahrens 	sdd.fromsnap = fromsnap;
8125367Sahrens 	sdd.tosnap = tosnap;
8135367Sahrens 	sdd.outfd = outfd;
8145367Sahrens 	sdd.replicate = replicate;
8155367Sahrens 	sdd.doall = doall;
8165367Sahrens 	sdd.fromorigin = fromorigin;
8175367Sahrens 	sdd.fss = fss;
8185367Sahrens 	sdd.fsavl = fsavl;
8195367Sahrens 	sdd.verbose = verbose;
8205367Sahrens 	err = dump_filesystems(zhp, &sdd);
8215367Sahrens 	fsavl_destroy(fsavl);
8225367Sahrens 	nvlist_free(fss);
8235367Sahrens 
8245367Sahrens 	if (replicate || doall) {
8255367Sahrens 		/*
8265367Sahrens 		 * write final end record.  NB: want to do this even if
8275367Sahrens 		 * there was some error, because it might not be totally
8285367Sahrens 		 * failed.
8295367Sahrens 		 */
8305367Sahrens 		dmu_replay_record_t drr = { 0 };
8315367Sahrens 		drr.drr_type = DRR_END;
8325367Sahrens 		if (write(outfd, &drr, sizeof (drr)) == -1) {
8335367Sahrens 			return (zfs_standard_error(zhp->zfs_hdl,
8345367Sahrens 			    errno, errbuf));
8355367Sahrens 		}
8365367Sahrens 	}
8375367Sahrens 
8385367Sahrens 	return (err || sdd.err);
8395367Sahrens }
8405367Sahrens 
8415367Sahrens /*
8425367Sahrens  * Routines specific to "zfs recv"
8435367Sahrens  */
8445367Sahrens 
8455367Sahrens static int
8465367Sahrens recv_read(libzfs_handle_t *hdl, int fd, void *buf, int ilen,
8475367Sahrens     boolean_t byteswap, zio_cksum_t *zc)
8485367Sahrens {
8495367Sahrens 	char *cp = buf;
8505367Sahrens 	int rv;
8515367Sahrens 	int len = ilen;
8525367Sahrens 
8535367Sahrens 	do {
8545367Sahrens 		rv = read(fd, cp, len);
8555367Sahrens 		cp += rv;
8565367Sahrens 		len -= rv;
8575367Sahrens 	} while (rv > 0);
8585367Sahrens 
8595367Sahrens 	if (rv < 0 || len != 0) {
8605367Sahrens 		zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
8615367Sahrens 		    "failed to read from stream"));
8625367Sahrens 		return (zfs_error(hdl, EZFS_BADSTREAM, dgettext(TEXT_DOMAIN,
8635367Sahrens 		    "cannot receive")));
8645367Sahrens 	}
8655367Sahrens 
8665367Sahrens 	if (zc) {
8675367Sahrens 		if (byteswap)
8685367Sahrens 			fletcher_4_incremental_byteswap(buf, ilen, zc);
8695367Sahrens 		else
8705367Sahrens 			fletcher_4_incremental_native(buf, ilen, zc);
8715367Sahrens 	}
8725367Sahrens 	return (0);
8735367Sahrens }
8745367Sahrens 
8755367Sahrens static int
8765367Sahrens recv_read_nvlist(libzfs_handle_t *hdl, int fd, int len, nvlist_t **nvp,
8775367Sahrens     boolean_t byteswap, zio_cksum_t *zc)
8785367Sahrens {
8795367Sahrens 	char *buf;
8805367Sahrens 	int err;
8815367Sahrens 
8825367Sahrens 	buf = zfs_alloc(hdl, len);
8835367Sahrens 	if (buf == NULL)
8845367Sahrens 		return (ENOMEM);
8855367Sahrens 
8865367Sahrens 	err = recv_read(hdl, fd, buf, len, byteswap, zc);
8875367Sahrens 	if (err != 0) {
8885367Sahrens 		free(buf);
8895367Sahrens 		return (err);
8905367Sahrens 	}
8915367Sahrens 
8925367Sahrens 	err = nvlist_unpack(buf, len, nvp, 0);
8935367Sahrens 	free(buf);
8945367Sahrens 	if (err != 0) {
8955367Sahrens 		zfs_error_aux(hdl, dgettext(TEXT_DOMAIN, "invalid "
8965367Sahrens 		    "stream (malformed nvlist)"));
8975367Sahrens 		return (EINVAL);
8985367Sahrens 	}
8995367Sahrens 	return (0);
9005367Sahrens }
9015367Sahrens 
9025367Sahrens static int
9035367Sahrens recv_rename(libzfs_handle_t *hdl, const char *name, const char *tryname,
9045367Sahrens     int baselen, char *newname, recvflags_t flags)
9055367Sahrens {
9065367Sahrens 	static int seq;
9075367Sahrens 	zfs_cmd_t zc = { 0 };
9085367Sahrens 	int err;
9096047Sahrens 	prop_changelist_t *clp;
9106047Sahrens 	zfs_handle_t *zhp;
9115367Sahrens 
9126047Sahrens 	zhp = zfs_open(hdl, name, ZFS_TYPE_DATASET);
9136047Sahrens 	if (zhp == NULL)
9146047Sahrens 		return (-1);
9157366STim.Haley@Sun.COM 	clp = changelist_gather(zhp, ZFS_PROP_NAME, 0,
9167366STim.Haley@Sun.COM 	    flags.force ? MS_FORCE : 0);
9176047Sahrens 	zfs_close(zhp);
9186047Sahrens 	if (clp == NULL)
9196047Sahrens 		return (-1);
9206047Sahrens 	err = changelist_prefix(clp);
9216047Sahrens 	if (err)
9226047Sahrens 		return (err);
9235367Sahrens 
92410242Schris.kirby@sun.com 	zc.zc_objset_type = DMU_OST_ZFS;
92510242Schris.kirby@sun.com 	(void) strlcpy(zc.zc_name, name, sizeof (zc.zc_name));
92610242Schris.kirby@sun.com 
9275367Sahrens 	if (tryname) {
9285367Sahrens 		(void) strcpy(newname, tryname);
9295367Sahrens 
9305367Sahrens 		(void) strlcpy(zc.zc_value, tryname, sizeof (zc.zc_value));
9315367Sahrens 
9325367Sahrens 		if (flags.verbose) {
9335367Sahrens 			(void) printf("attempting rename %s to %s\n",
9345367Sahrens 			    zc.zc_name, zc.zc_value);
9355367Sahrens 		}
9365367Sahrens 		err = ioctl(hdl->libzfs_fd, ZFS_IOC_RENAME, &zc);
9376047Sahrens 		if (err == 0)
9385367Sahrens 			changelist_rename(clp, name, tryname);
9395367Sahrens 	} else {
9405367Sahrens 		err = ENOENT;
9415367Sahrens 	}
9425367Sahrens 
9435367Sahrens 	if (err != 0 && strncmp(name+baselen, "recv-", 5) != 0) {
9445367Sahrens 		seq++;
9455367Sahrens 
9465367Sahrens 		(void) strncpy(newname, name, baselen);
9475367Sahrens 		(void) snprintf(newname+baselen, ZFS_MAXNAMELEN-baselen,
9485367Sahrens 		    "recv-%u-%u", getpid(), seq);
9495367Sahrens 		(void) strlcpy(zc.zc_value, newname, sizeof (zc.zc_value));
9505367Sahrens 
9515367Sahrens 		if (flags.verbose) {
9525367Sahrens 			(void) printf("failed - trying rename %s to %s\n",
9535367Sahrens 			    zc.zc_name, zc.zc_value);
9545367Sahrens 		}
9555367Sahrens 		err = ioctl(hdl->libzfs_fd, ZFS_IOC_RENAME, &zc);
9566047Sahrens 		if (err == 0)
9575367Sahrens 			changelist_rename(clp, name, newname);
9585367Sahrens 		if (err && flags.verbose) {
9595367Sahrens 			(void) printf("failed (%u) - "
9605367Sahrens 			    "will try again on next pass\n", errno);
9615367Sahrens 		}
9625367Sahrens 		err = EAGAIN;
9635367Sahrens 	} else if (flags.verbose) {
9645367Sahrens 		if (err == 0)
9655367Sahrens 			(void) printf("success\n");
9665367Sahrens 		else
9675367Sahrens 			(void) printf("failed (%u)\n", errno);
9685367Sahrens 	}
9695367Sahrens 
9706047Sahrens 	(void) changelist_postfix(clp);
9716047Sahrens 	changelist_free(clp);
9725367Sahrens 
9735367Sahrens 	return (err);
9745367Sahrens }
9755367Sahrens 
9765367Sahrens static int
9775367Sahrens recv_destroy(libzfs_handle_t *hdl, const char *name, int baselen,
9785367Sahrens     char *newname, recvflags_t flags)
9795367Sahrens {
9805367Sahrens 	zfs_cmd_t zc = { 0 };
9816047Sahrens 	int err = 0;
9826047Sahrens 	prop_changelist_t *clp;
9836047Sahrens 	zfs_handle_t *zhp;
98410242Schris.kirby@sun.com 	boolean_t defer = B_FALSE;
98510242Schris.kirby@sun.com 	int spa_version;
9866047Sahrens 
9876047Sahrens 	zhp = zfs_open(hdl, name, ZFS_TYPE_DATASET);
9886047Sahrens 	if (zhp == NULL)
9896047Sahrens 		return (-1);
9907366STim.Haley@Sun.COM 	clp = changelist_gather(zhp, ZFS_PROP_NAME, 0,
9917366STim.Haley@Sun.COM 	    flags.force ? MS_FORCE : 0);
99210242Schris.kirby@sun.com 	if (zfs_get_type(zhp) == ZFS_TYPE_SNAPSHOT &&
99310242Schris.kirby@sun.com 	    zfs_spa_version(zhp, &spa_version) == 0 &&
99410242Schris.kirby@sun.com 	    spa_version >= SPA_VERSION_USERREFS)
99510242Schris.kirby@sun.com 		defer = B_TRUE;
9966047Sahrens 	zfs_close(zhp);
9976047Sahrens 	if (clp == NULL)
9986047Sahrens 		return (-1);
9996047Sahrens 	err = changelist_prefix(clp);
10006047Sahrens 	if (err)
10016047Sahrens 		return (err);
10025367Sahrens 
10035367Sahrens 	zc.zc_objset_type = DMU_OST_ZFS;
100410242Schris.kirby@sun.com 	zc.zc_defer_destroy = defer;
10055367Sahrens 	(void) strlcpy(zc.zc_name, name, sizeof (zc.zc_name));
10065367Sahrens 
10075367Sahrens 	if (flags.verbose)
10085367Sahrens 		(void) printf("attempting destroy %s\n", zc.zc_name);
10095367Sahrens 	err = ioctl(hdl->libzfs_fd, ZFS_IOC_DESTROY, &zc);
10106047Sahrens 	if (err == 0) {
10116047Sahrens 		if (flags.verbose)
10126047Sahrens 			(void) printf("success\n");
10136047Sahrens 		changelist_remove(clp, zc.zc_name);
10145367Sahrens 	}
10155367Sahrens 
10166047Sahrens 	(void) changelist_postfix(clp);
10176047Sahrens 	changelist_free(clp);
10186047Sahrens 
101910242Schris.kirby@sun.com 	/*
102010242Schris.kirby@sun.com 	 * Deferred destroy should always succeed. Since we can't tell
102110242Schris.kirby@sun.com 	 * if it destroyed the dataset or just marked it for deferred
102210242Schris.kirby@sun.com 	 * destroy, always do the rename just in case.
102310242Schris.kirby@sun.com 	 */
102410242Schris.kirby@sun.com 	if (err != 0 || defer)
10256047Sahrens 		err = recv_rename(hdl, name, NULL, baselen, newname, flags);
10265367Sahrens 
10275367Sahrens 	return (err);
10285367Sahrens }
10295367Sahrens 
10305367Sahrens typedef struct guid_to_name_data {
10315367Sahrens 	uint64_t guid;
10325367Sahrens 	char *name;
10335367Sahrens } guid_to_name_data_t;
10345367Sahrens 
10355367Sahrens static int
10365367Sahrens guid_to_name_cb(zfs_handle_t *zhp, void *arg)
10375367Sahrens {
10385367Sahrens 	guid_to_name_data_t *gtnd = arg;
10395367Sahrens 	int err;
10405367Sahrens 
10415367Sahrens 	if (zhp->zfs_dmustats.dds_guid == gtnd->guid) {
10425367Sahrens 		(void) strcpy(gtnd->name, zhp->zfs_name);
10435367Sahrens 		return (EEXIST);
10445367Sahrens 	}
10455367Sahrens 	err = zfs_iter_children(zhp, guid_to_name_cb, gtnd);
10465367Sahrens 	zfs_close(zhp);
10475367Sahrens 	return (err);
10485367Sahrens }
10495367Sahrens 
10505367Sahrens static int
10515367Sahrens guid_to_name(libzfs_handle_t *hdl, const char *parent, uint64_t guid,
10525367Sahrens     char *name)
10535367Sahrens {
10545367Sahrens 	/* exhaustive search all local snapshots */
10555367Sahrens 	guid_to_name_data_t gtnd;
10565367Sahrens 	int err = 0;
10575367Sahrens 	zfs_handle_t *zhp;
10585367Sahrens 	char *cp;
10595367Sahrens 
10605367Sahrens 	gtnd.guid = guid;
10615367Sahrens 	gtnd.name = name;
10625367Sahrens 
10635367Sahrens 	if (strchr(parent, '@') == NULL) {
10645367Sahrens 		zhp = make_dataset_handle(hdl, parent);
10655367Sahrens 		if (zhp != NULL) {
10665367Sahrens 			err = zfs_iter_children(zhp, guid_to_name_cb, &gtnd);
10675367Sahrens 			zfs_close(zhp);
10685367Sahrens 			if (err == EEXIST)
10695367Sahrens 				return (0);
10705367Sahrens 		}
10715367Sahrens 	}
10725367Sahrens 
10735367Sahrens 	cp = strchr(parent, '/');
10745367Sahrens 	if (cp)
10755367Sahrens 		*cp = '\0';
10765367Sahrens 	zhp = make_dataset_handle(hdl, parent);
10775367Sahrens 	if (cp)
10785367Sahrens 		*cp = '/';
10795367Sahrens 
10805367Sahrens 	if (zhp) {
10815367Sahrens 		err = zfs_iter_children(zhp, guid_to_name_cb, &gtnd);
10825367Sahrens 		zfs_close(zhp);
10835367Sahrens 	}
10845367Sahrens 
10855367Sahrens 	return (err == EEXIST ? 0 : ENOENT);
10865367Sahrens 
10875367Sahrens }
10885367Sahrens 
10895367Sahrens /*
10905367Sahrens  * Return true if dataset guid1 is created before guid2.
10915367Sahrens  */
10926476Sahl static int
10935367Sahrens created_before(libzfs_handle_t *hdl, avl_tree_t *avl,
10945367Sahrens     uint64_t guid1, uint64_t guid2)
10955367Sahrens {
10965367Sahrens 	nvlist_t *nvfs;
10975367Sahrens 	char *fsname, *snapname;
10985367Sahrens 	char buf[ZFS_MAXNAMELEN];
10996476Sahl 	int rv;
11005367Sahrens 	zfs_node_t zn1, zn2;
11015367Sahrens 
11025367Sahrens 	if (guid2 == 0)
11036476Sahl 		return (0);
11045367Sahrens 	if (guid1 == 0)
11056476Sahl 		return (1);
11065367Sahrens 
11075367Sahrens 	nvfs = fsavl_find(avl, guid1, &snapname);
11085367Sahrens 	VERIFY(0 == nvlist_lookup_string(nvfs, "name", &fsname));
11095367Sahrens 	(void) snprintf(buf, sizeof (buf), "%s@%s", fsname, snapname);
11105367Sahrens 	zn1.zn_handle = zfs_open(hdl, buf, ZFS_TYPE_SNAPSHOT);
11116476Sahl 	if (zn1.zn_handle == NULL)
11126476Sahl 		return (-1);
11135367Sahrens 
11145367Sahrens 	nvfs = fsavl_find(avl, guid2, &snapname);
11155367Sahrens 	VERIFY(0 == nvlist_lookup_string(nvfs, "name", &fsname));
11165367Sahrens 	(void) snprintf(buf, sizeof (buf), "%s@%s", fsname, snapname);
11175367Sahrens 	zn2.zn_handle = zfs_open(hdl, buf, ZFS_TYPE_SNAPSHOT);
11186476Sahl 	if (zn2.zn_handle == NULL) {
11196476Sahl 		zfs_close(zn2.zn_handle);
11206476Sahl 		return (-1);
11216476Sahl 	}
11225367Sahrens 
11235367Sahrens 	rv = (zfs_snapshot_compare(&zn1, &zn2) == -1);
11245367Sahrens 
11255367Sahrens 	zfs_close(zn1.zn_handle);
11265367Sahrens 	zfs_close(zn2.zn_handle);
11275367Sahrens 
11285367Sahrens 	return (rv);
11295367Sahrens }
11305367Sahrens 
11315367Sahrens static int
11325367Sahrens recv_incremental_replication(libzfs_handle_t *hdl, const char *tofs,
11335367Sahrens     recvflags_t flags, nvlist_t *stream_nv, avl_tree_t *stream_avl)
11345367Sahrens {
11355367Sahrens 	nvlist_t *local_nv;
11365367Sahrens 	avl_tree_t *local_avl;
11375367Sahrens 	nvpair_t *fselem, *nextfselem;
11385367Sahrens 	char *tosnap, *fromsnap;
11395367Sahrens 	char newname[ZFS_MAXNAMELEN];
11405367Sahrens 	int error;
11415367Sahrens 	boolean_t needagain, progress;
1142*10299SPrabahar.Jeyaram@Sun.COM 	char *s1, *s2;
11435367Sahrens 
11445367Sahrens 	VERIFY(0 == nvlist_lookup_string(stream_nv, "fromsnap", &fromsnap));
11455367Sahrens 	VERIFY(0 == nvlist_lookup_string(stream_nv, "tosnap", &tosnap));
11465367Sahrens 
11475367Sahrens 	if (flags.dryrun)
11485367Sahrens 		return (0);
11495367Sahrens 
11505367Sahrens again:
11515367Sahrens 	needagain = progress = B_FALSE;
11525367Sahrens 
11535367Sahrens 	if ((error = gather_nvlist(hdl, tofs, fromsnap, NULL,
11545367Sahrens 	    &local_nv, &local_avl)) != 0)
11555367Sahrens 		return (error);
11565367Sahrens 
11575367Sahrens 	/*
11585367Sahrens 	 * Process deletes and renames
11595367Sahrens 	 */
11605367Sahrens 	for (fselem = nvlist_next_nvpair(local_nv, NULL);
11615367Sahrens 	    fselem; fselem = nextfselem) {
11625367Sahrens 		nvlist_t *nvfs, *snaps;
11635367Sahrens 		nvlist_t *stream_nvfs = NULL;
11645367Sahrens 		nvpair_t *snapelem, *nextsnapelem;
11655367Sahrens 		uint64_t fromguid = 0;
11665367Sahrens 		uint64_t originguid = 0;
11675367Sahrens 		uint64_t stream_originguid = 0;
11685367Sahrens 		uint64_t parent_fromsnap_guid, stream_parent_fromsnap_guid;
11695367Sahrens 		char *fsname, *stream_fsname;
11705367Sahrens 
11715367Sahrens 		nextfselem = nvlist_next_nvpair(local_nv, fselem);
11725367Sahrens 
11735367Sahrens 		VERIFY(0 == nvpair_value_nvlist(fselem, &nvfs));
11745367Sahrens 		VERIFY(0 == nvlist_lookup_nvlist(nvfs, "snaps", &snaps));
11755367Sahrens 		VERIFY(0 == nvlist_lookup_string(nvfs, "name", &fsname));
11765367Sahrens 		VERIFY(0 == nvlist_lookup_uint64(nvfs, "parentfromsnap",
11775367Sahrens 		    &parent_fromsnap_guid));
11785367Sahrens 		(void) nvlist_lookup_uint64(nvfs, "origin", &originguid);
11795367Sahrens 
11805367Sahrens 		/*
11815367Sahrens 		 * First find the stream's fs, so we can check for
11825367Sahrens 		 * a different origin (due to "zfs promote")
11835367Sahrens 		 */
11845367Sahrens 		for (snapelem = nvlist_next_nvpair(snaps, NULL);
11855367Sahrens 		    snapelem; snapelem = nvlist_next_nvpair(snaps, snapelem)) {
11865367Sahrens 			uint64_t thisguid;
11875367Sahrens 
11885367Sahrens 			VERIFY(0 == nvpair_value_uint64(snapelem, &thisguid));
11895367Sahrens 			stream_nvfs = fsavl_find(stream_avl, thisguid, NULL);
11905367Sahrens 
11915367Sahrens 			if (stream_nvfs != NULL)
11925367Sahrens 				break;
11935367Sahrens 		}
11945367Sahrens 
11955367Sahrens 		/* check for promote */
11965367Sahrens 		(void) nvlist_lookup_uint64(stream_nvfs, "origin",
11975367Sahrens 		    &stream_originguid);
11985367Sahrens 		if (stream_nvfs && originguid != stream_originguid) {
11996476Sahl 			switch (created_before(hdl, local_avl,
12006476Sahl 			    stream_originguid, originguid)) {
12016476Sahl 			case 1: {
12025367Sahrens 				/* promote it! */
12035367Sahrens 				zfs_cmd_t zc = { 0 };
12045367Sahrens 				nvlist_t *origin_nvfs;
12055367Sahrens 				char *origin_fsname;
12065367Sahrens 
12075367Sahrens 				if (flags.verbose)
12085367Sahrens 					(void) printf("promoting %s\n", fsname);
12095367Sahrens 
12105367Sahrens 				origin_nvfs = fsavl_find(local_avl, originguid,
12115367Sahrens 				    NULL);
12125367Sahrens 				VERIFY(0 == nvlist_lookup_string(origin_nvfs,
12135367Sahrens 				    "name", &origin_fsname));
12145367Sahrens 				(void) strlcpy(zc.zc_value, origin_fsname,
12155367Sahrens 				    sizeof (zc.zc_value));
12165367Sahrens 				(void) strlcpy(zc.zc_name, fsname,
12175367Sahrens 				    sizeof (zc.zc_name));
12185367Sahrens 				error = zfs_ioctl(hdl, ZFS_IOC_PROMOTE, &zc);
12195367Sahrens 				if (error == 0)
12205367Sahrens 					progress = B_TRUE;
12216476Sahl 				break;
12226476Sahl 			}
12236476Sahl 			default:
12246476Sahl 				break;
12256476Sahl 			case -1:
12266476Sahl 				fsavl_destroy(local_avl);
12276476Sahl 				nvlist_free(local_nv);
12286476Sahl 				return (-1);
12295367Sahrens 			}
12305367Sahrens 			/*
12315367Sahrens 			 * We had/have the wrong origin, therefore our
12325367Sahrens 			 * list of snapshots is wrong.  Need to handle
12335367Sahrens 			 * them on the next pass.
12345367Sahrens 			 */
12355367Sahrens 			needagain = B_TRUE;
12365367Sahrens 			continue;
12375367Sahrens 		}
12385367Sahrens 
12395367Sahrens 		for (snapelem = nvlist_next_nvpair(snaps, NULL);
12405367Sahrens 		    snapelem; snapelem = nextsnapelem) {
12415367Sahrens 			uint64_t thisguid;
12425367Sahrens 			char *stream_snapname;
12437265Sahrens 			nvlist_t *found, *props;
12445367Sahrens 
12455367Sahrens 			nextsnapelem = nvlist_next_nvpair(snaps, snapelem);
12465367Sahrens 
12475367Sahrens 			VERIFY(0 == nvpair_value_uint64(snapelem, &thisguid));
12485367Sahrens 			found = fsavl_find(stream_avl, thisguid,
12495367Sahrens 			    &stream_snapname);
12505367Sahrens 
12515367Sahrens 			/* check for delete */
12525367Sahrens 			if (found == NULL) {
12535367Sahrens 				char name[ZFS_MAXNAMELEN];
12545367Sahrens 
12555367Sahrens 				if (!flags.force)
12565367Sahrens 					continue;
12575367Sahrens 
12585367Sahrens 				(void) snprintf(name, sizeof (name), "%s@%s",
12595367Sahrens 				    fsname, nvpair_name(snapelem));
12605367Sahrens 
12615367Sahrens 				error = recv_destroy(hdl, name,
12625367Sahrens 				    strlen(fsname)+1, newname, flags);
12635367Sahrens 				if (error)
12645367Sahrens 					needagain = B_TRUE;
12655367Sahrens 				else
12665367Sahrens 					progress = B_TRUE;
12675367Sahrens 				continue;
12685367Sahrens 			}
12695367Sahrens 
12705367Sahrens 			stream_nvfs = found;
12715367Sahrens 
12727265Sahrens 			if (0 == nvlist_lookup_nvlist(stream_nvfs, "snapprops",
12737265Sahrens 			    &props) && 0 == nvlist_lookup_nvlist(props,
12747265Sahrens 			    stream_snapname, &props)) {
12757265Sahrens 				zfs_cmd_t zc = { 0 };
12767265Sahrens 
12777265Sahrens 				zc.zc_cookie = B_TRUE; /* clear current props */
12787273Sahrens 				(void) snprintf(zc.zc_name, sizeof (zc.zc_name),
12797265Sahrens 				    "%s@%s", fsname, nvpair_name(snapelem));
12807265Sahrens 				if (zcmd_write_src_nvlist(hdl, &zc,
12817265Sahrens 				    props) == 0) {
12827265Sahrens 					(void) zfs_ioctl(hdl,
12837265Sahrens 					    ZFS_IOC_SET_PROP, &zc);
12847265Sahrens 					zcmd_free_nvlists(&zc);
12857265Sahrens 				}
12867265Sahrens 			}
12877265Sahrens 
12885367Sahrens 			/* check for different snapname */
12895367Sahrens 			if (strcmp(nvpair_name(snapelem),
12905367Sahrens 			    stream_snapname) != 0) {
12915367Sahrens 				char name[ZFS_MAXNAMELEN];
12925367Sahrens 				char tryname[ZFS_MAXNAMELEN];
12935367Sahrens 
12945367Sahrens 				(void) snprintf(name, sizeof (name), "%s@%s",
12955367Sahrens 				    fsname, nvpair_name(snapelem));
12965367Sahrens 				(void) snprintf(tryname, sizeof (name), "%s@%s",
12975367Sahrens 				    fsname, stream_snapname);
12985367Sahrens 
12995367Sahrens 				error = recv_rename(hdl, name, tryname,
13005367Sahrens 				    strlen(fsname)+1, newname, flags);
13015367Sahrens 				if (error)
13025367Sahrens 					needagain = B_TRUE;
13035367Sahrens 				else
13045367Sahrens 					progress = B_TRUE;
13055367Sahrens 			}
13065367Sahrens 
13075367Sahrens 			if (strcmp(stream_snapname, fromsnap) == 0)
13085367Sahrens 				fromguid = thisguid;
13095367Sahrens 		}
13105367Sahrens 
13115367Sahrens 		/* check for delete */
13125367Sahrens 		if (stream_nvfs == NULL) {
13135367Sahrens 			if (!flags.force)
13145367Sahrens 				continue;
13155367Sahrens 
13165367Sahrens 			error = recv_destroy(hdl, fsname, strlen(tofs)+1,
13175367Sahrens 			    newname, flags);
13185367Sahrens 			if (error)
13195367Sahrens 				needagain = B_TRUE;
13205367Sahrens 			else
13215367Sahrens 				progress = B_TRUE;
13225367Sahrens 			continue;
13235367Sahrens 		}
13245367Sahrens 
13255367Sahrens 		if (fromguid == 0 && flags.verbose) {
13265367Sahrens 			(void) printf("local fs %s does not have fromsnap "
13275367Sahrens 			    "(%s in stream); must have been deleted locally; "
13285367Sahrens 			    "ignoring\n", fsname, fromsnap);
13295367Sahrens 			continue;
13305367Sahrens 		}
13315367Sahrens 
13325367Sahrens 		VERIFY(0 == nvlist_lookup_string(stream_nvfs,
13335367Sahrens 		    "name", &stream_fsname));
13345367Sahrens 		VERIFY(0 == nvlist_lookup_uint64(stream_nvfs,
13355367Sahrens 		    "parentfromsnap", &stream_parent_fromsnap_guid));
13365367Sahrens 
13375367Sahrens 		/* check for rename */
13385367Sahrens 		if ((stream_parent_fromsnap_guid != 0 &&
13395367Sahrens 		    stream_parent_fromsnap_guid != parent_fromsnap_guid) ||
1340*10299SPrabahar.Jeyaram@Sun.COM 		    ((s1 = strrchr(fsname, '/')) &&
1341*10299SPrabahar.Jeyaram@Sun.COM 		    (s2 = strrchr(stream_fsname, '/')) &&
1342*10299SPrabahar.Jeyaram@Sun.COM 		    strcmp(s1, s2) != 0)) {
13435367Sahrens 			nvlist_t *parent;
13445367Sahrens 			char tryname[ZFS_MAXNAMELEN];
13455367Sahrens 
13465367Sahrens 			parent = fsavl_find(local_avl,
13475367Sahrens 			    stream_parent_fromsnap_guid, NULL);
13485367Sahrens 			/*
13495367Sahrens 			 * NB: parent might not be found if we used the
13505367Sahrens 			 * tosnap for stream_parent_fromsnap_guid,
13515367Sahrens 			 * because the parent is a newly-created fs;
13525367Sahrens 			 * we'll be able to rename it after we recv the
13535367Sahrens 			 * new fs.
13545367Sahrens 			 */
13555367Sahrens 			if (parent != NULL) {
13565367Sahrens 				char *pname;
13575367Sahrens 
13585367Sahrens 				VERIFY(0 == nvlist_lookup_string(parent, "name",
13595367Sahrens 				    &pname));
13605367Sahrens 				(void) snprintf(tryname, sizeof (tryname),
13615367Sahrens 				    "%s%s", pname, strrchr(stream_fsname, '/'));
13625367Sahrens 			} else {
13635367Sahrens 				tryname[0] = '\0';
13645367Sahrens 				if (flags.verbose) {
13655367Sahrens 					(void) printf("local fs %s new parent "
13665367Sahrens 					    "not found\n", fsname);
13675367Sahrens 				}
13685367Sahrens 			}
13695367Sahrens 
13705367Sahrens 			error = recv_rename(hdl, fsname, tryname,
13715367Sahrens 			    strlen(tofs)+1, newname, flags);
13725367Sahrens 			if (error)
13735367Sahrens 				needagain = B_TRUE;
13745367Sahrens 			else
13755367Sahrens 				progress = B_TRUE;
13765367Sahrens 		}
13775367Sahrens 	}
13785367Sahrens 
13795367Sahrens 	fsavl_destroy(local_avl);
13805367Sahrens 	nvlist_free(local_nv);
13815367Sahrens 
13825367Sahrens 	if (needagain && progress) {
13835367Sahrens 		/* do another pass to fix up temporary names */
13845367Sahrens 		if (flags.verbose)
13855367Sahrens 			(void) printf("another pass:\n");
13865367Sahrens 		goto again;
13875367Sahrens 	}
13885367Sahrens 
13895367Sahrens 	return (needagain);
13905367Sahrens }
13915367Sahrens 
13925367Sahrens static int
13935367Sahrens zfs_receive_package(libzfs_handle_t *hdl, int fd, const char *destname,
13947366STim.Haley@Sun.COM     recvflags_t flags, dmu_replay_record_t *drr, zio_cksum_t *zc,
13957366STim.Haley@Sun.COM     char **top_zfs)
13965367Sahrens {
13975367Sahrens 	nvlist_t *stream_nv = NULL;
13985367Sahrens 	avl_tree_t *stream_avl = NULL;
13995367Sahrens 	char *fromsnap = NULL;
14005367Sahrens 	char tofs[ZFS_MAXNAMELEN];
14015367Sahrens 	char errbuf[1024];
14025367Sahrens 	dmu_replay_record_t drre;
14035367Sahrens 	int error;
14045367Sahrens 	boolean_t anyerr = B_FALSE;
14056479Sahrens 	boolean_t softerr = B_FALSE;
14065367Sahrens 
14075367Sahrens 	(void) snprintf(errbuf, sizeof (errbuf), dgettext(TEXT_DOMAIN,
14085367Sahrens 	    "cannot receive"));
14095367Sahrens 
14105367Sahrens 	if (strchr(destname, '@')) {
14115367Sahrens 		zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
14125367Sahrens 		    "can not specify snapshot name for multi-snapshot stream"));
14135367Sahrens 		return (zfs_error(hdl, EZFS_BADSTREAM, errbuf));
14145367Sahrens 	}
14155367Sahrens 
14165367Sahrens 	assert(drr->drr_type == DRR_BEGIN);
14175367Sahrens 	assert(drr->drr_u.drr_begin.drr_magic == DMU_BACKUP_MAGIC);
14185367Sahrens 	assert(drr->drr_u.drr_begin.drr_version == DMU_BACKUP_HEADER_VERSION);
14195367Sahrens 
14205367Sahrens 	/*
14215367Sahrens 	 * Read in the nvlist from the stream.
14225367Sahrens 	 */
14235367Sahrens 	if (drr->drr_payloadlen != 0) {
14245367Sahrens 		if (!flags.isprefix) {
14255367Sahrens 			zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
14265367Sahrens 			    "must use -d to receive replication "
14275367Sahrens 			    "(send -R) stream"));
14285367Sahrens 			return (zfs_error(hdl, EZFS_BADSTREAM, errbuf));
14295367Sahrens 		}
14305367Sahrens 
14315367Sahrens 		error = recv_read_nvlist(hdl, fd, drr->drr_payloadlen,
14325367Sahrens 		    &stream_nv, flags.byteswap, zc);
14335367Sahrens 		if (error) {
14345367Sahrens 			error = zfs_error(hdl, EZFS_BADSTREAM, errbuf);
14355367Sahrens 			goto out;
14365367Sahrens 		}
14375367Sahrens 	}
14385367Sahrens 
14395367Sahrens 	/*
14405367Sahrens 	 * Read in the end record and verify checksum.
14415367Sahrens 	 */
14425367Sahrens 	if (0 != (error = recv_read(hdl, fd, &drre, sizeof (drre),
14435367Sahrens 	    flags.byteswap, NULL)))
14445367Sahrens 		goto out;
14455367Sahrens 	if (flags.byteswap) {
14465367Sahrens 		drre.drr_type = BSWAP_32(drre.drr_type);
14475367Sahrens 		drre.drr_u.drr_end.drr_checksum.zc_word[0] =
14485367Sahrens 		    BSWAP_64(drre.drr_u.drr_end.drr_checksum.zc_word[0]);
14495367Sahrens 		drre.drr_u.drr_end.drr_checksum.zc_word[1] =
14505367Sahrens 		    BSWAP_64(drre.drr_u.drr_end.drr_checksum.zc_word[1]);
14515367Sahrens 		drre.drr_u.drr_end.drr_checksum.zc_word[2] =
14525367Sahrens 		    BSWAP_64(drre.drr_u.drr_end.drr_checksum.zc_word[2]);
14535367Sahrens 		drre.drr_u.drr_end.drr_checksum.zc_word[3] =
14545367Sahrens 		    BSWAP_64(drre.drr_u.drr_end.drr_checksum.zc_word[3]);
14555367Sahrens 	}
14565367Sahrens 	if (drre.drr_type != DRR_END) {
14575367Sahrens 		error = zfs_error(hdl, EZFS_BADSTREAM, errbuf);
14585367Sahrens 		goto out;
14595367Sahrens 	}
14605367Sahrens 	if (!ZIO_CHECKSUM_EQUAL(drre.drr_u.drr_end.drr_checksum, *zc)) {
14615367Sahrens 		zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
14625367Sahrens 		    "incorrect header checksum"));
14635367Sahrens 		error = zfs_error(hdl, EZFS_BADSTREAM, errbuf);
14645367Sahrens 		goto out;
14655367Sahrens 	}
14665367Sahrens 
14675367Sahrens 	(void) nvlist_lookup_string(stream_nv, "fromsnap", &fromsnap);
14685367Sahrens 
14695367Sahrens 	if (drr->drr_payloadlen != 0) {
14705367Sahrens 		nvlist_t *stream_fss;
14715367Sahrens 
14725367Sahrens 		VERIFY(0 == nvlist_lookup_nvlist(stream_nv, "fss",
14735367Sahrens 		    &stream_fss));
14745896Sahl 		if ((stream_avl = fsavl_create(stream_fss)) == NULL) {
14755896Sahl 			zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
14765896Sahl 			    "couldn't allocate avl tree"));
14775896Sahl 			error = zfs_error(hdl, EZFS_NOMEM, errbuf);
14785896Sahl 			goto out;
14795896Sahl 		}
14805367Sahrens 
14815367Sahrens 		if (fromsnap != NULL) {
14825367Sahrens 			(void) strlcpy(tofs, destname, ZFS_MAXNAMELEN);
14835367Sahrens 			if (flags.isprefix) {
14845367Sahrens 				int i = strcspn(drr->drr_u.drr_begin.drr_toname,
14855367Sahrens 				    "/@");
14865367Sahrens 				/* zfs_receive_one() will create_parents() */
14875367Sahrens 				(void) strlcat(tofs,
14885367Sahrens 				    &drr->drr_u.drr_begin.drr_toname[i],
14895367Sahrens 				    ZFS_MAXNAMELEN);
14905367Sahrens 				*strchr(tofs, '@') = '\0';
14915367Sahrens 			}
14926479Sahrens 			softerr = recv_incremental_replication(hdl, tofs,
14935367Sahrens 			    flags, stream_nv, stream_avl);
14945367Sahrens 		}
14955367Sahrens 	}
14965367Sahrens 
14975367Sahrens 
14985367Sahrens 	/* Finally, receive each contained stream */
14995367Sahrens 	do {
15005367Sahrens 		/*
15015367Sahrens 		 * we should figure out if it has a recoverable
15025367Sahrens 		 * error, in which case do a recv_skip() and drive on.
15035367Sahrens 		 * Note, if we fail due to already having this guid,
15045367Sahrens 		 * zfs_receive_one() will take care of it (ie,
15055367Sahrens 		 * recv_skip() and return 0).
15065367Sahrens 		 */
15077366STim.Haley@Sun.COM 		error = zfs_receive_impl(hdl, destname, flags, fd,
15087366STim.Haley@Sun.COM 		    stream_avl, top_zfs);
15095367Sahrens 		if (error == ENODATA) {
15105367Sahrens 			error = 0;
15115367Sahrens 			break;
15125367Sahrens 		}
15135367Sahrens 		anyerr |= error;
15145367Sahrens 	} while (error == 0);
15155367Sahrens 
15165367Sahrens 	if (drr->drr_payloadlen != 0 && fromsnap != NULL) {
15175367Sahrens 		/*
15185367Sahrens 		 * Now that we have the fs's they sent us, try the
15195367Sahrens 		 * renames again.
15205367Sahrens 		 */
15216479Sahrens 		softerr = recv_incremental_replication(hdl, tofs, flags,
15225367Sahrens 		    stream_nv, stream_avl);
15235367Sahrens 	}
15245367Sahrens 
15255367Sahrens out:
15265367Sahrens 	fsavl_destroy(stream_avl);
15275367Sahrens 	if (stream_nv)
15285367Sahrens 		nvlist_free(stream_nv);
15296479Sahrens 	if (softerr)
15306479Sahrens 		error = -2;
15315367Sahrens 	if (anyerr)
15325367Sahrens 		error = -1;
15335367Sahrens 	return (error);
15345367Sahrens }
15355367Sahrens 
15365367Sahrens static int
15375367Sahrens recv_skip(libzfs_handle_t *hdl, int fd, boolean_t byteswap)
15385367Sahrens {
15395367Sahrens 	dmu_replay_record_t *drr;
15405367Sahrens 	void *buf = malloc(1<<20);
15415367Sahrens 
15425367Sahrens 	/* XXX would be great to use lseek if possible... */
15435367Sahrens 	drr = buf;
15445367Sahrens 
15455367Sahrens 	while (recv_read(hdl, fd, drr, sizeof (dmu_replay_record_t),
15465367Sahrens 	    byteswap, NULL) == 0) {
15475367Sahrens 		if (byteswap)
15485367Sahrens 			drr->drr_type = BSWAP_32(drr->drr_type);
15495367Sahrens 
15505367Sahrens 		switch (drr->drr_type) {
15515367Sahrens 		case DRR_BEGIN:
15525367Sahrens 			/* NB: not to be used on v2 stream packages */
15535367Sahrens 			assert(drr->drr_payloadlen == 0);
15545367Sahrens 			break;
15555367Sahrens 
15565367Sahrens 		case DRR_END:
15575367Sahrens 			free(buf);
15585367Sahrens 			return (0);
15595367Sahrens 
15605367Sahrens 		case DRR_OBJECT:
15615367Sahrens 			if (byteswap) {
15625367Sahrens 				drr->drr_u.drr_object.drr_bonuslen =
15635367Sahrens 				    BSWAP_32(drr->drr_u.drr_object.
15645367Sahrens 				    drr_bonuslen);
15655367Sahrens 			}
15665367Sahrens 			(void) recv_read(hdl, fd, buf,
15675367Sahrens 			    P2ROUNDUP(drr->drr_u.drr_object.drr_bonuslen, 8),
15685367Sahrens 			    B_FALSE, NULL);
15695367Sahrens 			break;
15705367Sahrens 
15715367Sahrens 		case DRR_WRITE:
15725367Sahrens 			if (byteswap) {
15735367Sahrens 				drr->drr_u.drr_write.drr_length =
15747366STim.Haley@Sun.COM 				    BSWAP_64(drr->drr_u.drr_write.drr_length);
15755367Sahrens 			}
15765367Sahrens 			(void) recv_read(hdl, fd, buf,
15775367Sahrens 			    drr->drr_u.drr_write.drr_length, B_FALSE, NULL);
15785367Sahrens 			break;
15795367Sahrens 
15805367Sahrens 		case DRR_FREEOBJECTS:
15815367Sahrens 		case DRR_FREE:
15825367Sahrens 			break;
15835367Sahrens 
15845367Sahrens 		default:
15855367Sahrens 			assert(!"invalid record type");
15865367Sahrens 		}
15875367Sahrens 	}
15885367Sahrens 
15895367Sahrens 	free(buf);
15905367Sahrens 	return (-1);
15915367Sahrens }
15925367Sahrens 
15935367Sahrens /*
15945367Sahrens  * Restores a backup of tosnap from the file descriptor specified by infd.
15955367Sahrens  */
15965367Sahrens static int
15975367Sahrens zfs_receive_one(libzfs_handle_t *hdl, int infd, const char *tosnap,
15985367Sahrens     recvflags_t flags, dmu_replay_record_t *drr,
15997366STim.Haley@Sun.COM     dmu_replay_record_t *drr_noswap, avl_tree_t *stream_avl,
16007366STim.Haley@Sun.COM     char **top_zfs)
16015367Sahrens {
16025367Sahrens 	zfs_cmd_t zc = { 0 };
16035367Sahrens 	time_t begin_time;
16045367Sahrens 	int ioctl_err, ioctl_errno, err, choplen;
16055367Sahrens 	char *cp;
16065367Sahrens 	struct drr_begin *drrb = &drr->drr_u.drr_begin;
16075367Sahrens 	char errbuf[1024];
16085367Sahrens 	char chopprefix[ZFS_MAXNAMELEN];
16095367Sahrens 	boolean_t newfs = B_FALSE;
16105367Sahrens 	boolean_t stream_wantsnewfs;
16115367Sahrens 	uint64_t parent_snapguid = 0;
16125367Sahrens 	prop_changelist_t *clp = NULL;
16137265Sahrens 	nvlist_t *snapprops_nvlist = NULL;
16145367Sahrens 
16155367Sahrens 	begin_time = time(NULL);
16165367Sahrens 
16175367Sahrens 	(void) snprintf(errbuf, sizeof (errbuf), dgettext(TEXT_DOMAIN,
16185367Sahrens 	    "cannot receive"));
16195367Sahrens 
16205367Sahrens 	if (stream_avl != NULL) {
16217265Sahrens 		char *snapname;
16227265Sahrens 		nvlist_t *fs = fsavl_find(stream_avl, drrb->drr_toguid,
16237265Sahrens 		    &snapname);
16245367Sahrens 		nvlist_t *props;
16255896Sahl 		int ret;
16265367Sahrens 
16275367Sahrens 		(void) nvlist_lookup_uint64(fs, "parentfromsnap",
16285367Sahrens 		    &parent_snapguid);
16295367Sahrens 		err = nvlist_lookup_nvlist(fs, "props", &props);
16305367Sahrens 		if (err)
16315367Sahrens 			VERIFY(0 == nvlist_alloc(&props, NV_UNIQUE_NAME, 0));
16325896Sahl 
16335367Sahrens 		if (flags.canmountoff) {
16345367Sahrens 			VERIFY(0 == nvlist_add_uint64(props,
16355367Sahrens 			    zfs_prop_to_name(ZFS_PROP_CANMOUNT), 0));
16365367Sahrens 		}
16375896Sahl 		ret = zcmd_write_src_nvlist(hdl, &zc, props);
16385367Sahrens 		if (err)
16395367Sahrens 			nvlist_free(props);
16405896Sahl 
16417265Sahrens 		if (0 == nvlist_lookup_nvlist(fs, "snapprops", &props)) {
16427265Sahrens 			VERIFY(0 == nvlist_lookup_nvlist(props,
16437265Sahrens 			    snapname, &snapprops_nvlist));
16447265Sahrens 		}
16457265Sahrens 
16465896Sahl 		if (ret != 0)
16475896Sahl 			return (-1);
16485367Sahrens 	}
16495367Sahrens 
16505367Sahrens 	/*
16515367Sahrens 	 * Determine how much of the snapshot name stored in the stream
16525367Sahrens 	 * we are going to tack on to the name they specified on the
16535367Sahrens 	 * command line, and how much we are going to chop off.
16545367Sahrens 	 *
16555367Sahrens 	 * If they specified a snapshot, chop the entire name stored in
16565367Sahrens 	 * the stream.
16575367Sahrens 	 */
16585367Sahrens 	(void) strcpy(chopprefix, drrb->drr_toname);
16595367Sahrens 	if (flags.isprefix) {
16605367Sahrens 		/*
16615367Sahrens 		 * They specified a fs with -d, we want to tack on
16625367Sahrens 		 * everything but the pool name stored in the stream
16635367Sahrens 		 */
16645367Sahrens 		if (strchr(tosnap, '@')) {
16655367Sahrens 			zfs_error_aux(hdl, dgettext(TEXT_DOMAIN, "invalid "
16665367Sahrens 			    "argument - snapshot not allowed with -d"));
16675367Sahrens 			return (zfs_error(hdl, EZFS_INVALIDNAME, errbuf));
16685367Sahrens 		}
16695367Sahrens 		cp = strchr(chopprefix, '/');
16705367Sahrens 		if (cp == NULL)
16715367Sahrens 			cp = strchr(chopprefix, '@');
16725367Sahrens 		*cp = '\0';
16735367Sahrens 	} else if (strchr(tosnap, '@') == NULL) {
16745367Sahrens 		/*
16755367Sahrens 		 * If they specified a filesystem without -d, we want to
16765367Sahrens 		 * tack on everything after the fs specified in the
16775367Sahrens 		 * first name from the stream.
16785367Sahrens 		 */
16795367Sahrens 		cp = strchr(chopprefix, '@');
16805367Sahrens 		*cp = '\0';
16815367Sahrens 	}
16825367Sahrens 	choplen = strlen(chopprefix);
16835367Sahrens 
16845367Sahrens 	/*
16855367Sahrens 	 * Determine name of destination snapshot, store in zc_value.
16865367Sahrens 	 */
16875367Sahrens 	(void) strcpy(zc.zc_value, tosnap);
16885367Sahrens 	(void) strncat(zc.zc_value, drrb->drr_toname+choplen,
16895367Sahrens 	    sizeof (zc.zc_value));
16905896Sahl 	if (!zfs_name_valid(zc.zc_value, ZFS_TYPE_SNAPSHOT)) {
16915896Sahl 		zcmd_free_nvlists(&zc);
16925367Sahrens 		return (zfs_error(hdl, EZFS_INVALIDNAME, errbuf));
16935896Sahl 	}
16945367Sahrens 
16955367Sahrens 	/*
16965367Sahrens 	 * Determine the name of the origin snapshot, store in zc_string.
16975367Sahrens 	 */
16985367Sahrens 	if (drrb->drr_flags & DRR_FLAG_CLONE) {
16995367Sahrens 		if (guid_to_name(hdl, tosnap,
17005367Sahrens 		    drrb->drr_fromguid, zc.zc_string) != 0) {
17015896Sahl 			zcmd_free_nvlists(&zc);
17025367Sahrens 			zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
17035367Sahrens 			    "local origin for clone %s does not exist"),
17045367Sahrens 			    zc.zc_value);
17055367Sahrens 			return (zfs_error(hdl, EZFS_NOENT, errbuf));
17065367Sahrens 		}
17075367Sahrens 		if (flags.verbose)
17085367Sahrens 			(void) printf("found clone origin %s\n", zc.zc_string);
17095367Sahrens 	}
17105367Sahrens 
17115367Sahrens 	stream_wantsnewfs = (drrb->drr_fromguid == NULL ||
17125367Sahrens 	    (drrb->drr_flags & DRR_FLAG_CLONE));
17135367Sahrens 
17145367Sahrens 	if (stream_wantsnewfs) {
17155367Sahrens 		/*
17165367Sahrens 		 * if the parent fs does not exist, look for it based on
17175367Sahrens 		 * the parent snap GUID
17185367Sahrens 		 */
17195367Sahrens 		(void) snprintf(errbuf, sizeof (errbuf), dgettext(TEXT_DOMAIN,
17205367Sahrens 		    "cannot receive new filesystem stream"));
17215367Sahrens 
17225367Sahrens 		(void) strcpy(zc.zc_name, zc.zc_value);
17235367Sahrens 		cp = strrchr(zc.zc_name, '/');
17245367Sahrens 		if (cp)
17255367Sahrens 			*cp = '\0';
17265367Sahrens 		if (cp &&
17275367Sahrens 		    !zfs_dataset_exists(hdl, zc.zc_name, ZFS_TYPE_DATASET)) {
17285367Sahrens 			char suffix[ZFS_MAXNAMELEN];
17295367Sahrens 			(void) strcpy(suffix, strrchr(zc.zc_value, '/'));
17305367Sahrens 			if (guid_to_name(hdl, tosnap, parent_snapguid,
17315367Sahrens 			    zc.zc_value) == 0) {
17325367Sahrens 				*strchr(zc.zc_value, '@') = '\0';
17335367Sahrens 				(void) strcat(zc.zc_value, suffix);
17345367Sahrens 			}
17355367Sahrens 		}
17365367Sahrens 	} else {
17375367Sahrens 		/*
17385367Sahrens 		 * if the fs does not exist, look for it based on the
17395367Sahrens 		 * fromsnap GUID
17405367Sahrens 		 */
17415367Sahrens 		(void) snprintf(errbuf, sizeof (errbuf), dgettext(TEXT_DOMAIN,
17425367Sahrens 		    "cannot receive incremental stream"));
17435367Sahrens 
17445367Sahrens 		(void) strcpy(zc.zc_name, zc.zc_value);
17455367Sahrens 		*strchr(zc.zc_name, '@') = '\0';
17465367Sahrens 
17475367Sahrens 		if (!zfs_dataset_exists(hdl, zc.zc_name, ZFS_TYPE_DATASET)) {
17485367Sahrens 			char snap[ZFS_MAXNAMELEN];
17495367Sahrens 			(void) strcpy(snap, strchr(zc.zc_value, '@'));
17505367Sahrens 			if (guid_to_name(hdl, tosnap, drrb->drr_fromguid,
17515367Sahrens 			    zc.zc_value) == 0) {
17525367Sahrens 				*strchr(zc.zc_value, '@') = '\0';
17535367Sahrens 				(void) strcat(zc.zc_value, snap);
17545367Sahrens 			}
17555367Sahrens 		}
17565367Sahrens 	}
17575367Sahrens 
17585367Sahrens 	(void) strcpy(zc.zc_name, zc.zc_value);
17595367Sahrens 	*strchr(zc.zc_name, '@') = '\0';
17605367Sahrens 
17615367Sahrens 	if (zfs_dataset_exists(hdl, zc.zc_name, ZFS_TYPE_DATASET)) {
17625367Sahrens 		zfs_handle_t *zhp;
17635367Sahrens 		/*
17645367Sahrens 		 * Destination fs exists.  Therefore this should either
17655367Sahrens 		 * be an incremental, or the stream specifies a new fs
17665367Sahrens 		 * (full stream or clone) and they want us to blow it
17675367Sahrens 		 * away (and have therefore specified -F and removed any
17685367Sahrens 		 * snapshots).
17695367Sahrens 		 */
17705367Sahrens 
17715367Sahrens 		if (stream_wantsnewfs) {
17725367Sahrens 			if (!flags.force) {
17735896Sahl 				zcmd_free_nvlists(&zc);
17745367Sahrens 				zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
17755367Sahrens 				    "destination '%s' exists\n"
17765367Sahrens 				    "must specify -F to overwrite it"),
17775367Sahrens 				    zc.zc_name);
17785367Sahrens 				return (zfs_error(hdl, EZFS_EXISTS, errbuf));
17795367Sahrens 			}
17805367Sahrens 			if (ioctl(hdl->libzfs_fd, ZFS_IOC_SNAPSHOT_LIST_NEXT,
17815367Sahrens 			    &zc) == 0) {
17825896Sahl 				zcmd_free_nvlists(&zc);
17835367Sahrens 				zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
17845367Sahrens 				    "destination has snapshots (eg. %s)\n"
17855367Sahrens 				    "must destroy them to overwrite it"),
17865367Sahrens 				    zc.zc_name);
17875367Sahrens 				return (zfs_error(hdl, EZFS_EXISTS, errbuf));
17885367Sahrens 			}
17895367Sahrens 		}
17905367Sahrens 
17915896Sahl 		if ((zhp = zfs_open(hdl, zc.zc_name,
17925896Sahl 		    ZFS_TYPE_FILESYSTEM | ZFS_TYPE_VOLUME)) == NULL) {
17935896Sahl 			zcmd_free_nvlists(&zc);
17945367Sahrens 			return (-1);
17955896Sahl 		}
17965896Sahl 
17975367Sahrens 		if (stream_wantsnewfs &&
17985367Sahrens 		    zhp->zfs_dmustats.dds_origin[0]) {
17995896Sahl 			zcmd_free_nvlists(&zc);
18005367Sahrens 			zfs_close(zhp);
18015367Sahrens 			zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
18025367Sahrens 			    "destination '%s' is a clone\n"
18035367Sahrens 			    "must destroy it to overwrite it"),
18045367Sahrens 			    zc.zc_name);
18055367Sahrens 			return (zfs_error(hdl, EZFS_EXISTS, errbuf));
18065367Sahrens 		}
18075367Sahrens 
18085367Sahrens 		if (!flags.dryrun && zhp->zfs_type == ZFS_TYPE_FILESYSTEM &&
18095367Sahrens 		    stream_wantsnewfs) {
18105367Sahrens 			/* We can't do online recv in this case */
18117366STim.Haley@Sun.COM 			clp = changelist_gather(zhp, ZFS_PROP_NAME, 0, 0);
18125896Sahl 			if (clp == NULL) {
181310200SLori.Alt@Sun.COM 				zfs_close(zhp);
18145896Sahl 				zcmd_free_nvlists(&zc);
18155367Sahrens 				return (-1);
18165896Sahl 			}
18175367Sahrens 			if (changelist_prefix(clp) != 0) {
18185367Sahrens 				changelist_free(clp);
181910200SLori.Alt@Sun.COM 				zfs_close(zhp);
18205896Sahl 				zcmd_free_nvlists(&zc);
18215367Sahrens 				return (-1);
18225367Sahrens 			}
18235367Sahrens 		}
18245896Sahl 		if (!flags.dryrun && zhp->zfs_type == ZFS_TYPE_VOLUME &&
18255896Sahl 		    zvol_remove_link(hdl, zhp->zfs_name) != 0) {
18265896Sahl 			zfs_close(zhp);
18275896Sahl 			zcmd_free_nvlists(&zc);
18285896Sahl 			return (-1);
18295367Sahrens 		}
18305367Sahrens 		zfs_close(zhp);
18315367Sahrens 	} else {
18325367Sahrens 		/*
18335896Sahl 		 * Destination filesystem does not exist.  Therefore we better
18345896Sahl 		 * be creating a new filesystem (either from a full backup, or
18355896Sahl 		 * a clone).  It would therefore be invalid if the user
18365896Sahl 		 * specified only the pool name (i.e. if the destination name
18375896Sahl 		 * contained no slash character).
18385367Sahrens 		 */
18395896Sahl 		if (!stream_wantsnewfs ||
18405896Sahl 		    (cp = strrchr(zc.zc_name, '/')) == NULL) {
18415896Sahl 			zcmd_free_nvlists(&zc);
18425367Sahrens 			zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
18435367Sahrens 			    "destination '%s' does not exist"), zc.zc_name);
18445367Sahrens 			return (zfs_error(hdl, EZFS_NOENT, errbuf));
18455367Sahrens 		}
18465367Sahrens 
18475896Sahl 		/*
18485896Sahl 		 * Trim off the final dataset component so we perform the
18495896Sahl 		 * recvbackup ioctl to the filesystems's parent.
18505896Sahl 		 */
18515896Sahl 		*cp = '\0';
18525367Sahrens 
18535896Sahl 		if (flags.isprefix && !flags.dryrun &&
18545896Sahl 		    create_parents(hdl, zc.zc_value, strlen(tosnap)) != 0) {
18555896Sahl 			zcmd_free_nvlists(&zc);
18565896Sahl 			return (zfs_error(hdl, EZFS_BADRESTORE, errbuf));
18575367Sahrens 		}
18585367Sahrens 
18595367Sahrens 		newfs = B_TRUE;
18605367Sahrens 	}
18615367Sahrens 
18625367Sahrens 	zc.zc_begin_record = drr_noswap->drr_u.drr_begin;
18635367Sahrens 	zc.zc_cookie = infd;
18645367Sahrens 	zc.zc_guid = flags.force;
18655367Sahrens 	if (flags.verbose) {
18665367Sahrens 		(void) printf("%s %s stream of %s into %s\n",
18675367Sahrens 		    flags.dryrun ? "would receive" : "receiving",
18685367Sahrens 		    drrb->drr_fromguid ? "incremental" : "full",
18695367Sahrens 		    drrb->drr_toname, zc.zc_value);
18705367Sahrens 		(void) fflush(stdout);
18715367Sahrens 	}
18725367Sahrens 
18735896Sahl 	if (flags.dryrun) {
18745896Sahl 		zcmd_free_nvlists(&zc);
18755367Sahrens 		return (recv_skip(hdl, infd, flags.byteswap));
18765896Sahl 	}
18775367Sahrens 
18785367Sahrens 	err = ioctl_err = zfs_ioctl(hdl, ZFS_IOC_RECV, &zc);
18795367Sahrens 	ioctl_errno = errno;
18807265Sahrens 	zcmd_free_nvlists(&zc);
18817265Sahrens 
18827265Sahrens 	if (err == 0 && snapprops_nvlist) {
18837265Sahrens 		zfs_cmd_t zc2 = { 0 };
18847265Sahrens 
18857273Sahrens 		(void) strcpy(zc2.zc_name, zc.zc_value);
18867265Sahrens 		if (zcmd_write_src_nvlist(hdl, &zc2, snapprops_nvlist) == 0) {
18877265Sahrens 			(void) zfs_ioctl(hdl, ZFS_IOC_SET_PROP, &zc2);
18887265Sahrens 			zcmd_free_nvlists(&zc2);
18897265Sahrens 		}
18907265Sahrens 	}
18917265Sahrens 
18925367Sahrens 	if (err && (ioctl_errno == ENOENT || ioctl_errno == ENODEV)) {
18935367Sahrens 		/*
18945367Sahrens 		 * It may be that this snapshot already exists,
18955367Sahrens 		 * in which case we want to consume & ignore it
18965367Sahrens 		 * rather than failing.
18975367Sahrens 		 */
18985367Sahrens 		avl_tree_t *local_avl;
18995367Sahrens 		nvlist_t *local_nv, *fs;
19005367Sahrens 		char *cp = strchr(zc.zc_value, '@');
19015367Sahrens 
19025367Sahrens 		/*
19035367Sahrens 		 * XXX Do this faster by just iterating over snaps in
19045367Sahrens 		 * this fs.  Also if zc_value does not exist, we will
19055367Sahrens 		 * get a strange "does not exist" error message.
19065367Sahrens 		 */
19075367Sahrens 		*cp = '\0';
19085367Sahrens 		if (gather_nvlist(hdl, zc.zc_value, NULL, NULL,
19095367Sahrens 		    &local_nv, &local_avl) == 0) {
19105367Sahrens 			*cp = '@';
19115367Sahrens 			fs = fsavl_find(local_avl, drrb->drr_toguid, NULL);
19125367Sahrens 			fsavl_destroy(local_avl);
19135367Sahrens 			nvlist_free(local_nv);
19145367Sahrens 
19155367Sahrens 			if (fs != NULL) {
19165367Sahrens 				if (flags.verbose) {
19175367Sahrens 					(void) printf("snap %s already exists; "
19185367Sahrens 					    "ignoring\n", zc.zc_value);
19195367Sahrens 				}
19205367Sahrens 				ioctl_err = recv_skip(hdl, infd,
19215367Sahrens 				    flags.byteswap);
19225367Sahrens 			}
19235367Sahrens 		}
19245367Sahrens 		*cp = '@';
19255367Sahrens 	}
19265367Sahrens 
19275896Sahl 
19285367Sahrens 	if (ioctl_err != 0) {
19295367Sahrens 		switch (ioctl_errno) {
19305367Sahrens 		case ENODEV:
19315367Sahrens 			cp = strchr(zc.zc_value, '@');
19325367Sahrens 			*cp = '\0';
19335367Sahrens 			zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
19345367Sahrens 			    "most recent snapshot of %s does not\n"
19355367Sahrens 			    "match incremental source"), zc.zc_value);
19365367Sahrens 			(void) zfs_error(hdl, EZFS_BADRESTORE, errbuf);
19375367Sahrens 			*cp = '@';
19385367Sahrens 			break;
19395367Sahrens 		case ETXTBSY:
19405367Sahrens 			zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
19415367Sahrens 			    "destination %s has been modified\n"
19425367Sahrens 			    "since most recent snapshot"), zc.zc_name);
19435367Sahrens 			(void) zfs_error(hdl, EZFS_BADRESTORE, errbuf);
19445367Sahrens 			break;
19455367Sahrens 		case EEXIST:
19465367Sahrens 			cp = strchr(zc.zc_value, '@');
19475367Sahrens 			if (newfs) {
19485367Sahrens 				/* it's the containing fs that exists */
19495367Sahrens 				*cp = '\0';
19505367Sahrens 			}
19515367Sahrens 			zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
19525367Sahrens 			    "destination already exists"));
19535367Sahrens 			(void) zfs_error_fmt(hdl, EZFS_EXISTS,
19545367Sahrens 			    dgettext(TEXT_DOMAIN, "cannot restore to %s"),
19555367Sahrens 			    zc.zc_value);
19565367Sahrens 			*cp = '@';
19575367Sahrens 			break;
19585367Sahrens 		case EINVAL:
19595367Sahrens 			(void) zfs_error(hdl, EZFS_BADSTREAM, errbuf);
19605367Sahrens 			break;
19615367Sahrens 		case ECKSUM:
19625367Sahrens 			zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
19635367Sahrens 			    "invalid stream (checksum mismatch)"));
19645367Sahrens 			(void) zfs_error(hdl, EZFS_BADSTREAM, errbuf);
19655367Sahrens 			break;
19665367Sahrens 		default:
19675367Sahrens 			(void) zfs_standard_error(hdl, ioctl_errno, errbuf);
19685367Sahrens 		}
19695367Sahrens 	}
19705367Sahrens 
19715367Sahrens 	/*
19725367Sahrens 	 * Mount or recreate the /dev links for the target filesystem
19735367Sahrens 	 * (if created, or if we tore them down to do an incremental
19745367Sahrens 	 * restore), and the /dev links for the new snapshot (if
19755367Sahrens 	 * created). Also mount any children of the target filesystem
197610200SLori.Alt@Sun.COM 	 * if we did a replication receive (indicated by stream_avl
197710200SLori.Alt@Sun.COM 	 * being non-NULL).
19785367Sahrens 	 */
19795367Sahrens 	cp = strchr(zc.zc_value, '@');
19805367Sahrens 	if (cp && (ioctl_err == 0 || !newfs)) {
19815367Sahrens 		zfs_handle_t *h;
19825367Sahrens 
19835367Sahrens 		*cp = '\0';
19845367Sahrens 		h = zfs_open(hdl, zc.zc_value,
19855367Sahrens 		    ZFS_TYPE_FILESYSTEM | ZFS_TYPE_VOLUME);
19866476Sahl 		if (h != NULL) {
19875367Sahrens 			if (h->zfs_type == ZFS_TYPE_VOLUME) {
19887366STim.Haley@Sun.COM 				*cp = '@';
19895367Sahrens 				err = zvol_create_link(hdl, h->zfs_name);
19905367Sahrens 				if (err == 0 && ioctl_err == 0)
19915367Sahrens 					err = zvol_create_link(hdl,
19925367Sahrens 					    zc.zc_value);
199310200SLori.Alt@Sun.COM 			} else if (newfs || stream_avl) {
19947366STim.Haley@Sun.COM 				/*
19957366STim.Haley@Sun.COM 				 * Track the first/top of hierarchy fs,
19967366STim.Haley@Sun.COM 				 * for mounting and sharing later.
19977366STim.Haley@Sun.COM 				 */
19987366STim.Haley@Sun.COM 				if (top_zfs && *top_zfs == NULL)
19997366STim.Haley@Sun.COM 					*top_zfs = zfs_strdup(hdl, zc.zc_value);
20005367Sahrens 			}
20016476Sahl 			zfs_close(h);
20025367Sahrens 		}
20037366STim.Haley@Sun.COM 		*cp = '@';
20045367Sahrens 	}
20055367Sahrens 
20065367Sahrens 	if (clp) {
20075367Sahrens 		err |= changelist_postfix(clp);
20085367Sahrens 		changelist_free(clp);
20095367Sahrens 	}
20105367Sahrens 
20115367Sahrens 	if (err || ioctl_err)
20125367Sahrens 		return (-1);
20135367Sahrens 
20145367Sahrens 	if (flags.verbose) {
20155367Sahrens 		char buf1[64];
20165367Sahrens 		char buf2[64];
20175367Sahrens 		uint64_t bytes = zc.zc_cookie;
20185367Sahrens 		time_t delta = time(NULL) - begin_time;
20195367Sahrens 		if (delta == 0)
20205367Sahrens 			delta = 1;
20215367Sahrens 		zfs_nicenum(bytes, buf1, sizeof (buf1));
20225367Sahrens 		zfs_nicenum(bytes/delta, buf2, sizeof (buf1));
20235367Sahrens 
20245367Sahrens 		(void) printf("received %sB stream in %lu seconds (%sB/sec)\n",
20255367Sahrens 		    buf1, delta, buf2);
20265367Sahrens 	}
20275367Sahrens 
20285367Sahrens 	return (0);
20295367Sahrens }
20305367Sahrens 
20317366STim.Haley@Sun.COM static int
20327366STim.Haley@Sun.COM zfs_receive_impl(libzfs_handle_t *hdl, const char *tosnap, recvflags_t flags,
20337366STim.Haley@Sun.COM     int infd, avl_tree_t *stream_avl, char **top_zfs)
20345367Sahrens {
20355367Sahrens 	int err;
20365367Sahrens 	dmu_replay_record_t drr, drr_noswap;
20375367Sahrens 	struct drr_begin *drrb = &drr.drr_u.drr_begin;
20385367Sahrens 	char errbuf[1024];
20395367Sahrens 	zio_cksum_t zcksum = { 0 };
20405367Sahrens 
20415367Sahrens 	(void) snprintf(errbuf, sizeof (errbuf), dgettext(TEXT_DOMAIN,
20425367Sahrens 	    "cannot receive"));
20435367Sahrens 
20445367Sahrens 	if (flags.isprefix &&
20455367Sahrens 	    !zfs_dataset_exists(hdl, tosnap, ZFS_TYPE_DATASET)) {
20465367Sahrens 		zfs_error_aux(hdl, dgettext(TEXT_DOMAIN, "specified fs "
20475367Sahrens 		    "(%s) does not exist"), tosnap);
20485367Sahrens 		return (zfs_error(hdl, EZFS_NOENT, errbuf));
20495367Sahrens 	}
20505367Sahrens 
20515367Sahrens 	/* read in the BEGIN record */
20525367Sahrens 	if (0 != (err = recv_read(hdl, infd, &drr, sizeof (drr), B_FALSE,
20535367Sahrens 	    &zcksum)))
20545367Sahrens 		return (err);
20555367Sahrens 
20565367Sahrens 	if (drr.drr_type == DRR_END || drr.drr_type == BSWAP_32(DRR_END)) {
20575367Sahrens 		/* It's the double end record at the end of a package */
20585367Sahrens 		return (ENODATA);
20595367Sahrens 	}
20605367Sahrens 
20615367Sahrens 	/* the kernel needs the non-byteswapped begin record */
20625367Sahrens 	drr_noswap = drr;
20635367Sahrens 
20645446Sahrens 	flags.byteswap = B_FALSE;
20655367Sahrens 	if (drrb->drr_magic == BSWAP_64(DMU_BACKUP_MAGIC)) {
20665367Sahrens 		/*
20675367Sahrens 		 * We computed the checksum in the wrong byteorder in
20685367Sahrens 		 * recv_read() above; do it again correctly.
20695367Sahrens 		 */
20705367Sahrens 		bzero(&zcksum, sizeof (zio_cksum_t));
20715367Sahrens 		fletcher_4_incremental_byteswap(&drr, sizeof (drr), &zcksum);
20725367Sahrens 		flags.byteswap = B_TRUE;
20735367Sahrens 
20745367Sahrens 		drr.drr_type = BSWAP_32(drr.drr_type);
20755367Sahrens 		drr.drr_payloadlen = BSWAP_32(drr.drr_payloadlen);
20765367Sahrens 		drrb->drr_magic = BSWAP_64(drrb->drr_magic);
20775367Sahrens 		drrb->drr_version = BSWAP_64(drrb->drr_version);
20785367Sahrens 		drrb->drr_creation_time = BSWAP_64(drrb->drr_creation_time);
20795367Sahrens 		drrb->drr_type = BSWAP_32(drrb->drr_type);
20805367Sahrens 		drrb->drr_flags = BSWAP_32(drrb->drr_flags);
20815367Sahrens 		drrb->drr_toguid = BSWAP_64(drrb->drr_toguid);
20825367Sahrens 		drrb->drr_fromguid = BSWAP_64(drrb->drr_fromguid);
20835367Sahrens 	}
20845367Sahrens 
20855367Sahrens 	if (drrb->drr_magic != DMU_BACKUP_MAGIC || drr.drr_type != DRR_BEGIN) {
20865367Sahrens 		zfs_error_aux(hdl, dgettext(TEXT_DOMAIN, "invalid "
20875367Sahrens 		    "stream (bad magic number)"));
20885367Sahrens 		return (zfs_error(hdl, EZFS_BADSTREAM, errbuf));
20895367Sahrens 	}
20905367Sahrens 
20915367Sahrens 	if (strchr(drrb->drr_toname, '@') == NULL) {
20925367Sahrens 		zfs_error_aux(hdl, dgettext(TEXT_DOMAIN, "invalid "
20935367Sahrens 		    "stream (bad snapshot name)"));
20945367Sahrens 		return (zfs_error(hdl, EZFS_BADSTREAM, errbuf));
20955367Sahrens 	}
20965367Sahrens 
20975367Sahrens 	if (drrb->drr_version == DMU_BACKUP_STREAM_VERSION) {
20985367Sahrens 		return (zfs_receive_one(hdl, infd, tosnap, flags,
20997366STim.Haley@Sun.COM 		    &drr, &drr_noswap, stream_avl, top_zfs));
21005367Sahrens 	} else if (drrb->drr_version == DMU_BACKUP_HEADER_VERSION) {
21015367Sahrens 		return (zfs_receive_package(hdl, infd, tosnap, flags,
21027366STim.Haley@Sun.COM 		    &drr, &zcksum, top_zfs));
21035367Sahrens 	} else {
21045367Sahrens 		zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
21055367Sahrens 		    "stream is unsupported version %llu"),
21065367Sahrens 		    drrb->drr_version);
21075367Sahrens 		return (zfs_error(hdl, EZFS_BADSTREAM, errbuf));
21085367Sahrens 	}
21095367Sahrens }
21107366STim.Haley@Sun.COM 
21117366STim.Haley@Sun.COM /*
21127366STim.Haley@Sun.COM  * Restores a backup of tosnap from the file descriptor specified by infd.
21137366STim.Haley@Sun.COM  * Return 0 on total success, -2 if some things couldn't be
21147366STim.Haley@Sun.COM  * destroyed/renamed/promoted, -1 if some things couldn't be received.
21157366STim.Haley@Sun.COM  * (-1 will override -2).
21167366STim.Haley@Sun.COM  */
21177366STim.Haley@Sun.COM int
21187366STim.Haley@Sun.COM zfs_receive(libzfs_handle_t *hdl, const char *tosnap, recvflags_t flags,
21197366STim.Haley@Sun.COM     int infd, avl_tree_t *stream_avl)
21207366STim.Haley@Sun.COM {
21217366STim.Haley@Sun.COM 	char *top_zfs = NULL;
21227366STim.Haley@Sun.COM 	int err;
21237366STim.Haley@Sun.COM 
21247366STim.Haley@Sun.COM 	err = zfs_receive_impl(hdl, tosnap, flags, infd, stream_avl, &top_zfs);
21257366STim.Haley@Sun.COM 
21268584SLori.Alt@Sun.COM 	if (err == 0 && !flags.nomount && top_zfs) {
21277366STim.Haley@Sun.COM 		zfs_handle_t *zhp;
21287366STim.Haley@Sun.COM 		prop_changelist_t *clp;
21297366STim.Haley@Sun.COM 
21307366STim.Haley@Sun.COM 		zhp = zfs_open(hdl, top_zfs, ZFS_TYPE_FILESYSTEM);
21317366STim.Haley@Sun.COM 		if (zhp != NULL) {
21327366STim.Haley@Sun.COM 			clp = changelist_gather(zhp, ZFS_PROP_MOUNTPOINT,
21337366STim.Haley@Sun.COM 			    CL_GATHER_MOUNT_ALWAYS, 0);
21347366STim.Haley@Sun.COM 			zfs_close(zhp);
21357366STim.Haley@Sun.COM 			if (clp != NULL) {
21367366STim.Haley@Sun.COM 				/* mount and share received datasets */
21377366STim.Haley@Sun.COM 				err = changelist_postfix(clp);
21387366STim.Haley@Sun.COM 				changelist_free(clp);
21397366STim.Haley@Sun.COM 			}
21407366STim.Haley@Sun.COM 		}
21417366STim.Haley@Sun.COM 		if (zhp == NULL || clp == NULL || err)
21427366STim.Haley@Sun.COM 			err = -1;
21437366STim.Haley@Sun.COM 	}
21447366STim.Haley@Sun.COM 	if (top_zfs)
21457366STim.Haley@Sun.COM 		free(top_zfs);
21467366STim.Haley@Sun.COM 
21477366STim.Haley@Sun.COM 	return (err);
21487366STim.Haley@Sun.COM }
2149