xref: /onnv-gate/usr/src/lib/libzfs/common/libzfs_sendrecv.c (revision 12786:f6c8601080b4)
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 /*
2312070SMark.Shellenbaum@Sun.COM  * Copyright (c) 2005, 2010, Oracle and/or its affiliates. All rights reserved.
245367Sahrens  */
255367Sahrens 
265367Sahrens #include <assert.h>
275367Sahrens #include <ctype.h>
285367Sahrens #include <errno.h>
295367Sahrens #include <libintl.h>
305367Sahrens #include <stdio.h>
315367Sahrens #include <stdlib.h>
325367Sahrens #include <strings.h>
335367Sahrens #include <unistd.h>
345367Sahrens #include <stddef.h>
355367Sahrens #include <fcntl.h>
365367Sahrens #include <sys/mount.h>
3711007SLori.Alt@Sun.COM #include <pthread.h>
3811007SLori.Alt@Sun.COM #include <umem.h>
395367Sahrens 
405367Sahrens #include <libzfs.h>
415367Sahrens 
425367Sahrens #include "zfs_namecheck.h"
435367Sahrens #include "zfs_prop.h"
4410602SLori.Alt@Sun.COM #include "zfs_fletcher.h"
455367Sahrens #include "libzfs_impl.h"
4611007SLori.Alt@Sun.COM #include <sha2.h>
4711381SLori.Alt@Sun.COM #include <sys/zio_checksum.h>
4811381SLori.Alt@Sun.COM #include <sys/ddt.h>
495367Sahrens 
5011022STom.Erickson@Sun.COM /* in libzfs_dataset.c */
5111022STom.Erickson@Sun.COM extern void zfs_setprop_error(libzfs_handle_t *, zfs_prop_t, int, char *);
5211022STom.Erickson@Sun.COM 
537366STim.Haley@Sun.COM static int zfs_receive_impl(libzfs_handle_t *, const char *, recvflags_t,
5412527SChris.Kirby@oracle.com     int, const char *, nvlist_t *, avl_tree_t *, char **, int, uint64_t *);
557366STim.Haley@Sun.COM 
5611007SLori.Alt@Sun.COM static const zio_cksum_t zero_cksum = { 0 };
5711007SLori.Alt@Sun.COM 
5811007SLori.Alt@Sun.COM typedef struct dedup_arg {
5911007SLori.Alt@Sun.COM 	int	inputfd;
6011007SLori.Alt@Sun.COM 	int	outputfd;
6111007SLori.Alt@Sun.COM 	libzfs_handle_t  *dedup_hdl;
6211007SLori.Alt@Sun.COM } dedup_arg_t;
6311007SLori.Alt@Sun.COM 
6411007SLori.Alt@Sun.COM typedef struct dataref {
6511007SLori.Alt@Sun.COM 	uint64_t ref_guid;
6611007SLori.Alt@Sun.COM 	uint64_t ref_object;
6711007SLori.Alt@Sun.COM 	uint64_t ref_offset;
6811007SLori.Alt@Sun.COM } dataref_t;
6911007SLori.Alt@Sun.COM 
7011007SLori.Alt@Sun.COM typedef struct dedup_entry {
7111007SLori.Alt@Sun.COM 	struct dedup_entry	*dde_next;
7211007SLori.Alt@Sun.COM 	zio_cksum_t dde_chksum;
7311381SLori.Alt@Sun.COM 	uint64_t dde_prop;
7411007SLori.Alt@Sun.COM 	dataref_t dde_ref;
7511007SLori.Alt@Sun.COM } dedup_entry_t;
7611007SLori.Alt@Sun.COM 
7711007SLori.Alt@Sun.COM #define	MAX_DDT_PHYSMEM_PERCENT		20
7811007SLori.Alt@Sun.COM #define	SMALLEST_POSSIBLE_MAX_DDT_MB		128
7911007SLori.Alt@Sun.COM 
8011007SLori.Alt@Sun.COM typedef struct dedup_table {
8111007SLori.Alt@Sun.COM 	dedup_entry_t	**dedup_hash_array;
8211007SLori.Alt@Sun.COM 	umem_cache_t	*ddecache;
8311007SLori.Alt@Sun.COM 	uint64_t	max_ddt_size;  /* max dedup table size in bytes */
8411007SLori.Alt@Sun.COM 	uint64_t	cur_ddt_size;  /* current dedup table size in bytes */
8511007SLori.Alt@Sun.COM 	uint64_t	ddt_count;
8611007SLori.Alt@Sun.COM 	int		numhashbits;
8711007SLori.Alt@Sun.COM 	boolean_t	ddt_full;
8811007SLori.Alt@Sun.COM } dedup_table_t;
8911007SLori.Alt@Sun.COM 
9011007SLori.Alt@Sun.COM static int
high_order_bit(uint64_t n)9111007SLori.Alt@Sun.COM high_order_bit(uint64_t n)
9211007SLori.Alt@Sun.COM {
9311007SLori.Alt@Sun.COM 	int count;
9411007SLori.Alt@Sun.COM 
9511007SLori.Alt@Sun.COM 	for (count = 0; n != 0; count++)
9611007SLori.Alt@Sun.COM 		n >>= 1;
9711007SLori.Alt@Sun.COM 	return (count);
9811007SLori.Alt@Sun.COM }
9911007SLori.Alt@Sun.COM 
10011007SLori.Alt@Sun.COM static size_t
ssread(void * buf,size_t len,FILE * stream)10111007SLori.Alt@Sun.COM ssread(void *buf, size_t len, FILE *stream)
10211007SLori.Alt@Sun.COM {
10311007SLori.Alt@Sun.COM 	size_t outlen;
10411007SLori.Alt@Sun.COM 
10511007SLori.Alt@Sun.COM 	if ((outlen = fread(buf, len, 1, stream)) == 0)
10611007SLori.Alt@Sun.COM 		return (0);
10711007SLori.Alt@Sun.COM 
10811007SLori.Alt@Sun.COM 	return (outlen);
10911007SLori.Alt@Sun.COM }
11011007SLori.Alt@Sun.COM 
11111007SLori.Alt@Sun.COM static void
ddt_hash_append(libzfs_handle_t * hdl,dedup_table_t * ddt,dedup_entry_t ** ddepp,zio_cksum_t * cs,uint64_t prop,dataref_t * dr)11211007SLori.Alt@Sun.COM ddt_hash_append(libzfs_handle_t *hdl, dedup_table_t *ddt, dedup_entry_t **ddepp,
11311381SLori.Alt@Sun.COM     zio_cksum_t *cs, uint64_t prop, dataref_t *dr)
11411007SLori.Alt@Sun.COM {
11511007SLori.Alt@Sun.COM 	dedup_entry_t	*dde;
11611007SLori.Alt@Sun.COM 
11711007SLori.Alt@Sun.COM 	if (ddt->cur_ddt_size >= ddt->max_ddt_size) {
11811007SLori.Alt@Sun.COM 		if (ddt->ddt_full == B_FALSE) {
11911007SLori.Alt@Sun.COM 			zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
12011007SLori.Alt@Sun.COM 			    "Dedup table full.  Deduplication will continue "
12111007SLori.Alt@Sun.COM 			    "with existing table entries"));
12211007SLori.Alt@Sun.COM 			ddt->ddt_full = B_TRUE;
12311007SLori.Alt@Sun.COM 		}
12411007SLori.Alt@Sun.COM 		return;
12511007SLori.Alt@Sun.COM 	}
12611007SLori.Alt@Sun.COM 
12711007SLori.Alt@Sun.COM 	if ((dde = umem_cache_alloc(ddt->ddecache, UMEM_DEFAULT))
12811007SLori.Alt@Sun.COM 	    != NULL) {
12911007SLori.Alt@Sun.COM 		assert(*ddepp == NULL);
13011007SLori.Alt@Sun.COM 		dde->dde_next = NULL;
13111007SLori.Alt@Sun.COM 		dde->dde_chksum = *cs;
13211381SLori.Alt@Sun.COM 		dde->dde_prop = prop;
13311007SLori.Alt@Sun.COM 		dde->dde_ref = *dr;
13411007SLori.Alt@Sun.COM 		*ddepp = dde;
13511007SLori.Alt@Sun.COM 		ddt->cur_ddt_size += sizeof (dedup_entry_t);
13611007SLori.Alt@Sun.COM 		ddt->ddt_count++;
13711007SLori.Alt@Sun.COM 	}
13811007SLori.Alt@Sun.COM }
13911007SLori.Alt@Sun.COM 
14011007SLori.Alt@Sun.COM /*
14111007SLori.Alt@Sun.COM  * Using the specified dedup table, do a lookup for an entry with
14211007SLori.Alt@Sun.COM  * the checksum cs.  If found, return the block's reference info
14311007SLori.Alt@Sun.COM  * in *dr. Otherwise, insert a new entry in the dedup table, using
14411007SLori.Alt@Sun.COM  * the reference information specified by *dr.
14511007SLori.Alt@Sun.COM  *
14611007SLori.Alt@Sun.COM  * return value:  true - entry was found
14711007SLori.Alt@Sun.COM  *		  false - entry was not found
14811007SLori.Alt@Sun.COM  */
14911007SLori.Alt@Sun.COM static boolean_t
ddt_update(libzfs_handle_t * hdl,dedup_table_t * ddt,zio_cksum_t * cs,uint64_t prop,dataref_t * dr)15011007SLori.Alt@Sun.COM ddt_update(libzfs_handle_t *hdl, dedup_table_t *ddt, zio_cksum_t *cs,
15111381SLori.Alt@Sun.COM     uint64_t prop, dataref_t *dr)
15211007SLori.Alt@Sun.COM {
15311007SLori.Alt@Sun.COM 	uint32_t hashcode;
15411007SLori.Alt@Sun.COM 	dedup_entry_t **ddepp;
15511007SLori.Alt@Sun.COM 
15611007SLori.Alt@Sun.COM 	hashcode = BF64_GET(cs->zc_word[0], 0, ddt->numhashbits);
15711007SLori.Alt@Sun.COM 
15811007SLori.Alt@Sun.COM 	for (ddepp = &(ddt->dedup_hash_array[hashcode]); *ddepp != NULL;
15911007SLori.Alt@Sun.COM 	    ddepp = &((*ddepp)->dde_next)) {
16011381SLori.Alt@Sun.COM 		if (ZIO_CHECKSUM_EQUAL(((*ddepp)->dde_chksum), *cs) &&
16111381SLori.Alt@Sun.COM 		    (*ddepp)->dde_prop == prop) {
16211007SLori.Alt@Sun.COM 			*dr = (*ddepp)->dde_ref;
16311007SLori.Alt@Sun.COM 			return (B_TRUE);
16411007SLori.Alt@Sun.COM 		}
16511007SLori.Alt@Sun.COM 	}
16611381SLori.Alt@Sun.COM 	ddt_hash_append(hdl, ddt, ddepp, cs, prop, dr);
16711007SLori.Alt@Sun.COM 	return (B_FALSE);
16811007SLori.Alt@Sun.COM }
16911007SLori.Alt@Sun.COM 
17011007SLori.Alt@Sun.COM static int
cksum_and_write(const void * buf,uint64_t len,zio_cksum_t * zc,int outfd)17111007SLori.Alt@Sun.COM cksum_and_write(const void *buf, uint64_t len, zio_cksum_t *zc, int outfd)
17211007SLori.Alt@Sun.COM {
17311007SLori.Alt@Sun.COM 	fletcher_4_incremental_native(buf, len, zc);
17411007SLori.Alt@Sun.COM 	return (write(outfd, buf, len));
17511007SLori.Alt@Sun.COM }
17611007SLori.Alt@Sun.COM 
17711007SLori.Alt@Sun.COM /*
17811007SLori.Alt@Sun.COM  * This function is started in a separate thread when the dedup option
17911007SLori.Alt@Sun.COM  * has been requested.  The main send thread determines the list of
18011007SLori.Alt@Sun.COM  * snapshots to be included in the send stream and makes the ioctl calls
18111007SLori.Alt@Sun.COM  * for each one.  But instead of having the ioctl send the output to the
18211007SLori.Alt@Sun.COM  * the output fd specified by the caller of zfs_send()), the
18311007SLori.Alt@Sun.COM  * ioctl is told to direct the output to a pipe, which is read by the
18411007SLori.Alt@Sun.COM  * alternate thread running THIS function.  This function does the
18511007SLori.Alt@Sun.COM  * dedup'ing by:
18611007SLori.Alt@Sun.COM  *  1. building a dedup table (the DDT)
18711007SLori.Alt@Sun.COM  *  2. doing checksums on each data block and inserting a record in the DDT
18811007SLori.Alt@Sun.COM  *  3. looking for matching checksums, and
18911007SLori.Alt@Sun.COM  *  4.  sending a DRR_WRITE_BYREF record instead of a write record whenever
19011007SLori.Alt@Sun.COM  *      a duplicate block is found.
19111007SLori.Alt@Sun.COM  * The output of this function then goes to the output fd requested
19211007SLori.Alt@Sun.COM  * by the caller of zfs_send().
19311007SLori.Alt@Sun.COM  */
19411007SLori.Alt@Sun.COM static void *
cksummer(void * arg)19511007SLori.Alt@Sun.COM cksummer(void *arg)
19611007SLori.Alt@Sun.COM {
19711007SLori.Alt@Sun.COM 	dedup_arg_t *dda = arg;
19811007SLori.Alt@Sun.COM 	char *buf = malloc(1<<20);
19911007SLori.Alt@Sun.COM 	dmu_replay_record_t thedrr;
20011007SLori.Alt@Sun.COM 	dmu_replay_record_t *drr = &thedrr;
20111007SLori.Alt@Sun.COM 	struct drr_begin *drrb = &thedrr.drr_u.drr_begin;
20211007SLori.Alt@Sun.COM 	struct drr_end *drre = &thedrr.drr_u.drr_end;
20311007SLori.Alt@Sun.COM 	struct drr_object *drro = &thedrr.drr_u.drr_object;
20411007SLori.Alt@Sun.COM 	struct drr_write *drrw = &thedrr.drr_u.drr_write;
20511935SMark.Shellenbaum@Sun.COM 	struct drr_spill *drrs = &thedrr.drr_u.drr_spill;
20611007SLori.Alt@Sun.COM 	FILE *ofp;
20711007SLori.Alt@Sun.COM 	int outfd;
20811381SLori.Alt@Sun.COM 	dmu_replay_record_t wbr_drr = {0};
20911007SLori.Alt@Sun.COM 	struct drr_write_byref *wbr_drrr = &wbr_drr.drr_u.drr_write_byref;
21011007SLori.Alt@Sun.COM 	dedup_table_t ddt;
21111007SLori.Alt@Sun.COM 	zio_cksum_t stream_cksum;
21211007SLori.Alt@Sun.COM 	uint64_t physmem = sysconf(_SC_PHYS_PAGES) * sysconf(_SC_PAGESIZE);
21311007SLori.Alt@Sun.COM 	uint64_t numbuckets;
21411007SLori.Alt@Sun.COM 
21511007SLori.Alt@Sun.COM 	ddt.max_ddt_size =
21611007SLori.Alt@Sun.COM 	    MAX((physmem * MAX_DDT_PHYSMEM_PERCENT)/100,
21711007SLori.Alt@Sun.COM 	    SMALLEST_POSSIBLE_MAX_DDT_MB<<20);
21811007SLori.Alt@Sun.COM 
21911007SLori.Alt@Sun.COM 	numbuckets = ddt.max_ddt_size/(sizeof (dedup_entry_t));
22011007SLori.Alt@Sun.COM 
22111007SLori.Alt@Sun.COM 	/*
22211007SLori.Alt@Sun.COM 	 * numbuckets must be a power of 2.  Increase number to
22311007SLori.Alt@Sun.COM 	 * a power of 2 if necessary.
22411007SLori.Alt@Sun.COM 	 */
22511007SLori.Alt@Sun.COM 	if (!ISP2(numbuckets))
22611007SLori.Alt@Sun.COM 		numbuckets = 1 << high_order_bit(numbuckets);
22711007SLori.Alt@Sun.COM 
22811007SLori.Alt@Sun.COM 	ddt.dedup_hash_array = calloc(numbuckets, sizeof (dedup_entry_t *));
22911007SLori.Alt@Sun.COM 	ddt.ddecache = umem_cache_create("dde", sizeof (dedup_entry_t), 0,
23011007SLori.Alt@Sun.COM 	    NULL, NULL, NULL, NULL, NULL, 0);
23111007SLori.Alt@Sun.COM 	ddt.cur_ddt_size = numbuckets * sizeof (dedup_entry_t *);
23211007SLori.Alt@Sun.COM 	ddt.numhashbits = high_order_bit(numbuckets) - 1;
23311007SLori.Alt@Sun.COM 	ddt.ddt_full = B_FALSE;
23411007SLori.Alt@Sun.COM 
23511007SLori.Alt@Sun.COM 	/* Initialize the write-by-reference block. */
23611007SLori.Alt@Sun.COM 	wbr_drr.drr_type = DRR_WRITE_BYREF;
23711007SLori.Alt@Sun.COM 	wbr_drr.drr_payloadlen = 0;
23811007SLori.Alt@Sun.COM 
23911007SLori.Alt@Sun.COM 	outfd = dda->outputfd;
24011007SLori.Alt@Sun.COM 	ofp = fdopen(dda->inputfd, "r");
24111007SLori.Alt@Sun.COM 	while (ssread(drr, sizeof (dmu_replay_record_t), ofp) != 0) {
24211007SLori.Alt@Sun.COM 
24311007SLori.Alt@Sun.COM 		switch (drr->drr_type) {
24411007SLori.Alt@Sun.COM 		case DRR_BEGIN:
24511007SLori.Alt@Sun.COM 		{
24611007SLori.Alt@Sun.COM 			int	fflags;
24711007SLori.Alt@Sun.COM 			ZIO_SET_CHECKSUM(&stream_cksum, 0, 0, 0, 0);
24811007SLori.Alt@Sun.COM 
24911007SLori.Alt@Sun.COM 			/* set the DEDUP feature flag for this stream */
25011007SLori.Alt@Sun.COM 			fflags = DMU_GET_FEATUREFLAGS(drrb->drr_versioninfo);
25111381SLori.Alt@Sun.COM 			fflags |= (DMU_BACKUP_FEATURE_DEDUP |
25211381SLori.Alt@Sun.COM 			    DMU_BACKUP_FEATURE_DEDUPPROPS);
25311007SLori.Alt@Sun.COM 			DMU_SET_FEATUREFLAGS(drrb->drr_versioninfo, fflags);
25411007SLori.Alt@Sun.COM 
25511007SLori.Alt@Sun.COM 			if (cksum_and_write(drr, sizeof (dmu_replay_record_t),
25611007SLori.Alt@Sun.COM 			    &stream_cksum, outfd) == -1)
25711007SLori.Alt@Sun.COM 				goto out;
25811007SLori.Alt@Sun.COM 			if (DMU_GET_STREAM_HDRTYPE(drrb->drr_versioninfo) ==
25911007SLori.Alt@Sun.COM 			    DMU_COMPOUNDSTREAM && drr->drr_payloadlen != 0) {
26011007SLori.Alt@Sun.COM 				int sz = drr->drr_payloadlen;
26111007SLori.Alt@Sun.COM 
26211007SLori.Alt@Sun.COM 				if (sz > 1<<20) {
26311007SLori.Alt@Sun.COM 					free(buf);
26411007SLori.Alt@Sun.COM 					buf = malloc(sz);
26511007SLori.Alt@Sun.COM 				}
26611007SLori.Alt@Sun.COM 				(void) ssread(buf, sz, ofp);
26711007SLori.Alt@Sun.COM 				if (ferror(stdin))
26811007SLori.Alt@Sun.COM 					perror("fread");
26911007SLori.Alt@Sun.COM 				if (cksum_and_write(buf, sz, &stream_cksum,
27011007SLori.Alt@Sun.COM 				    outfd) == -1)
27111007SLori.Alt@Sun.COM 					goto out;
27211007SLori.Alt@Sun.COM 			}
27311007SLori.Alt@Sun.COM 			break;
27411007SLori.Alt@Sun.COM 		}
27511007SLori.Alt@Sun.COM 
27611007SLori.Alt@Sun.COM 		case DRR_END:
27711007SLori.Alt@Sun.COM 		{
27811007SLori.Alt@Sun.COM 			/* use the recalculated checksum */
27911007SLori.Alt@Sun.COM 			ZIO_SET_CHECKSUM(&drre->drr_checksum,
28011007SLori.Alt@Sun.COM 			    stream_cksum.zc_word[0], stream_cksum.zc_word[1],
28111007SLori.Alt@Sun.COM 			    stream_cksum.zc_word[2], stream_cksum.zc_word[3]);
28211007SLori.Alt@Sun.COM 			if ((write(outfd, drr,
28311007SLori.Alt@Sun.COM 			    sizeof (dmu_replay_record_t))) == -1)
28411007SLori.Alt@Sun.COM 				goto out;
28511007SLori.Alt@Sun.COM 			break;
28611007SLori.Alt@Sun.COM 		}
28711007SLori.Alt@Sun.COM 
28811007SLori.Alt@Sun.COM 		case DRR_OBJECT:
28911007SLori.Alt@Sun.COM 		{
29011007SLori.Alt@Sun.COM 			if (cksum_and_write(drr, sizeof (dmu_replay_record_t),
29111007SLori.Alt@Sun.COM 			    &stream_cksum, outfd) == -1)
29211007SLori.Alt@Sun.COM 				goto out;
29311007SLori.Alt@Sun.COM 			if (drro->drr_bonuslen > 0) {
29411007SLori.Alt@Sun.COM 				(void) ssread(buf,
29511007SLori.Alt@Sun.COM 				    P2ROUNDUP((uint64_t)drro->drr_bonuslen, 8),
29611007SLori.Alt@Sun.COM 				    ofp);
29711007SLori.Alt@Sun.COM 				if (cksum_and_write(buf,
29811007SLori.Alt@Sun.COM 				    P2ROUNDUP((uint64_t)drro->drr_bonuslen, 8),
29911007SLori.Alt@Sun.COM 				    &stream_cksum, outfd) == -1)
30011007SLori.Alt@Sun.COM 					goto out;
30111007SLori.Alt@Sun.COM 			}
30211007SLori.Alt@Sun.COM 			break;
30311007SLori.Alt@Sun.COM 		}
30411007SLori.Alt@Sun.COM 
30511935SMark.Shellenbaum@Sun.COM 		case DRR_SPILL:
30611935SMark.Shellenbaum@Sun.COM 		{
30711935SMark.Shellenbaum@Sun.COM 			if (cksum_and_write(drr, sizeof (dmu_replay_record_t),
30811935SMark.Shellenbaum@Sun.COM 			    &stream_cksum, outfd) == -1)
30911935SMark.Shellenbaum@Sun.COM 				goto out;
31011935SMark.Shellenbaum@Sun.COM 			(void) ssread(buf, drrs->drr_length, ofp);
31111935SMark.Shellenbaum@Sun.COM 			if (cksum_and_write(buf, drrs->drr_length,
31211935SMark.Shellenbaum@Sun.COM 			    &stream_cksum, outfd) == -1)
31311935SMark.Shellenbaum@Sun.COM 				goto out;
31411935SMark.Shellenbaum@Sun.COM 			break;
31511935SMark.Shellenbaum@Sun.COM 		}
31611935SMark.Shellenbaum@Sun.COM 
31711007SLori.Alt@Sun.COM 		case DRR_FREEOBJECTS:
31811007SLori.Alt@Sun.COM 		{
31911007SLori.Alt@Sun.COM 			if (cksum_and_write(drr, sizeof (dmu_replay_record_t),
32011007SLori.Alt@Sun.COM 			    &stream_cksum, outfd) == -1)
32111007SLori.Alt@Sun.COM 				goto out;
32211007SLori.Alt@Sun.COM 			break;
32311007SLori.Alt@Sun.COM 		}
32411007SLori.Alt@Sun.COM 
32511007SLori.Alt@Sun.COM 		case DRR_WRITE:
32611007SLori.Alt@Sun.COM 		{
32711007SLori.Alt@Sun.COM 			dataref_t	dataref;
32811007SLori.Alt@Sun.COM 
32911007SLori.Alt@Sun.COM 			(void) ssread(buf, drrw->drr_length, ofp);
33011381SLori.Alt@Sun.COM 
33111007SLori.Alt@Sun.COM 			/*
33211381SLori.Alt@Sun.COM 			 * Use the existing checksum if it's dedup-capable,
33311381SLori.Alt@Sun.COM 			 * else calculate a SHA256 checksum for it.
33411007SLori.Alt@Sun.COM 			 */
33511381SLori.Alt@Sun.COM 
33611381SLori.Alt@Sun.COM 			if (ZIO_CHECKSUM_EQUAL(drrw->drr_key.ddk_cksum,
33711381SLori.Alt@Sun.COM 			    zero_cksum) ||
33811381SLori.Alt@Sun.COM 			    !DRR_IS_DEDUP_CAPABLE(drrw->drr_checksumflags)) {
33911007SLori.Alt@Sun.COM 				SHA256_CTX	ctx;
34011007SLori.Alt@Sun.COM 				zio_cksum_t	tmpsha256;
34111007SLori.Alt@Sun.COM 
34211007SLori.Alt@Sun.COM 				SHA256Init(&ctx);
34311007SLori.Alt@Sun.COM 				SHA256Update(&ctx, buf, drrw->drr_length);
34411007SLori.Alt@Sun.COM 				SHA256Final(&tmpsha256, &ctx);
34511381SLori.Alt@Sun.COM 				drrw->drr_key.ddk_cksum.zc_word[0] =
34611007SLori.Alt@Sun.COM 				    BE_64(tmpsha256.zc_word[0]);
34711381SLori.Alt@Sun.COM 				drrw->drr_key.ddk_cksum.zc_word[1] =
34811007SLori.Alt@Sun.COM 				    BE_64(tmpsha256.zc_word[1]);
34911381SLori.Alt@Sun.COM 				drrw->drr_key.ddk_cksum.zc_word[2] =
35011007SLori.Alt@Sun.COM 				    BE_64(tmpsha256.zc_word[2]);
35111381SLori.Alt@Sun.COM 				drrw->drr_key.ddk_cksum.zc_word[3] =
35211007SLori.Alt@Sun.COM 				    BE_64(tmpsha256.zc_word[3]);
35311381SLori.Alt@Sun.COM 				drrw->drr_checksumtype = ZIO_CHECKSUM_SHA256;
35411381SLori.Alt@Sun.COM 				drrw->drr_checksumflags = DRR_CHECKSUM_DEDUP;
35511007SLori.Alt@Sun.COM 			}
35611007SLori.Alt@Sun.COM 
35711007SLori.Alt@Sun.COM 			dataref.ref_guid = drrw->drr_toguid;
35811007SLori.Alt@Sun.COM 			dataref.ref_object = drrw->drr_object;
35911007SLori.Alt@Sun.COM 			dataref.ref_offset = drrw->drr_offset;
36011007SLori.Alt@Sun.COM 
36111007SLori.Alt@Sun.COM 			if (ddt_update(dda->dedup_hdl, &ddt,
36211381SLori.Alt@Sun.COM 			    &drrw->drr_key.ddk_cksum, drrw->drr_key.ddk_prop,
36311381SLori.Alt@Sun.COM 			    &dataref)) {
36411007SLori.Alt@Sun.COM 				/* block already present in stream */
36511007SLori.Alt@Sun.COM 				wbr_drrr->drr_object = drrw->drr_object;
36611007SLori.Alt@Sun.COM 				wbr_drrr->drr_offset = drrw->drr_offset;
36711007SLori.Alt@Sun.COM 				wbr_drrr->drr_length = drrw->drr_length;
36811007SLori.Alt@Sun.COM 				wbr_drrr->drr_toguid = drrw->drr_toguid;
36911007SLori.Alt@Sun.COM 				wbr_drrr->drr_refguid = dataref.ref_guid;
37011007SLori.Alt@Sun.COM 				wbr_drrr->drr_refobject =
37111007SLori.Alt@Sun.COM 				    dataref.ref_object;
37211007SLori.Alt@Sun.COM 				wbr_drrr->drr_refoffset =
37311007SLori.Alt@Sun.COM 				    dataref.ref_offset;
37411007SLori.Alt@Sun.COM 
37511381SLori.Alt@Sun.COM 				wbr_drrr->drr_checksumtype =
37611381SLori.Alt@Sun.COM 				    drrw->drr_checksumtype;
37711381SLori.Alt@Sun.COM 				wbr_drrr->drr_checksumflags =
37811381SLori.Alt@Sun.COM 				    drrw->drr_checksumtype;
37911381SLori.Alt@Sun.COM 				wbr_drrr->drr_key.ddk_cksum =
38011381SLori.Alt@Sun.COM 				    drrw->drr_key.ddk_cksum;
38111381SLori.Alt@Sun.COM 				wbr_drrr->drr_key.ddk_prop =
38211381SLori.Alt@Sun.COM 				    drrw->drr_key.ddk_prop;
38311007SLori.Alt@Sun.COM 
38411007SLori.Alt@Sun.COM 				if (cksum_and_write(&wbr_drr,
38511007SLori.Alt@Sun.COM 				    sizeof (dmu_replay_record_t), &stream_cksum,
38611007SLori.Alt@Sun.COM 				    outfd) == -1)
38711007SLori.Alt@Sun.COM 					goto out;
38811007SLori.Alt@Sun.COM 			} else {
38911007SLori.Alt@Sun.COM 				/* block not previously seen */
39011007SLori.Alt@Sun.COM 				if (cksum_and_write(drr,
39111007SLori.Alt@Sun.COM 				    sizeof (dmu_replay_record_t), &stream_cksum,
39211007SLori.Alt@Sun.COM 				    outfd) == -1)
39311007SLori.Alt@Sun.COM 					goto out;
39411007SLori.Alt@Sun.COM 				if (cksum_and_write(buf,
39511007SLori.Alt@Sun.COM 				    drrw->drr_length,
39611007SLori.Alt@Sun.COM 				    &stream_cksum, outfd) == -1)
39711007SLori.Alt@Sun.COM 					goto out;
39811007SLori.Alt@Sun.COM 			}
39911007SLori.Alt@Sun.COM 			break;
40011007SLori.Alt@Sun.COM 		}
40111007SLori.Alt@Sun.COM 
40211007SLori.Alt@Sun.COM 		case DRR_FREE:
40311007SLori.Alt@Sun.COM 		{
40411007SLori.Alt@Sun.COM 			if (cksum_and_write(drr, sizeof (dmu_replay_record_t),
40511007SLori.Alt@Sun.COM 			    &stream_cksum, outfd) == -1)
40611007SLori.Alt@Sun.COM 				goto out;
40711007SLori.Alt@Sun.COM 			break;
40811007SLori.Alt@Sun.COM 		}
40911007SLori.Alt@Sun.COM 
41011007SLori.Alt@Sun.COM 		default:
41111007SLori.Alt@Sun.COM 			(void) printf("INVALID record type 0x%x\n",
41211007SLori.Alt@Sun.COM 			    drr->drr_type);
41311007SLori.Alt@Sun.COM 			/* should never happen, so assert */
41411007SLori.Alt@Sun.COM 			assert(B_FALSE);
41511007SLori.Alt@Sun.COM 		}
41611007SLori.Alt@Sun.COM 	}
41711007SLori.Alt@Sun.COM out:
41811007SLori.Alt@Sun.COM 	umem_cache_destroy(ddt.ddecache);
41911007SLori.Alt@Sun.COM 	free(ddt.dedup_hash_array);
42011007SLori.Alt@Sun.COM 	free(buf);
42111007SLori.Alt@Sun.COM 	(void) fclose(ofp);
42211007SLori.Alt@Sun.COM 
42311007SLori.Alt@Sun.COM 	return (NULL);
42411007SLori.Alt@Sun.COM }
42511007SLori.Alt@Sun.COM 
4265367Sahrens /*
4275367Sahrens  * Routines for dealing with the AVL tree of fs-nvlists
4285367Sahrens  */
4295367Sahrens typedef struct fsavl_node {
4305367Sahrens 	avl_node_t fn_node;
4315367Sahrens 	nvlist_t *fn_nvfs;
4325367Sahrens 	char *fn_snapname;
4335367Sahrens 	uint64_t fn_guid;
4345367Sahrens } fsavl_node_t;
4355367Sahrens 
4365367Sahrens static int
fsavl_compare(const void * arg1,const void * arg2)4375367Sahrens fsavl_compare(const void *arg1, const void *arg2)
4385367Sahrens {
4395367Sahrens 	const fsavl_node_t *fn1 = arg1;
4405367Sahrens 	const fsavl_node_t *fn2 = arg2;
4415367Sahrens 
4425367Sahrens 	if (fn1->fn_guid > fn2->fn_guid)
4435367Sahrens 		return (+1);
4445367Sahrens 	else if (fn1->fn_guid < fn2->fn_guid)
4455367Sahrens 		return (-1);
4465367Sahrens 	else
4475367Sahrens 		return (0);
4485367Sahrens }
4495367Sahrens 
4505367Sahrens /*
4515367Sahrens  * Given the GUID of a snapshot, find its containing filesystem and
4525367Sahrens  * (optionally) name.
4535367Sahrens  */
4545367Sahrens static nvlist_t *
fsavl_find(avl_tree_t * avl,uint64_t snapguid,char ** snapname)4555367Sahrens fsavl_find(avl_tree_t *avl, uint64_t snapguid, char **snapname)
4565367Sahrens {
4575367Sahrens 	fsavl_node_t fn_find;
4585367Sahrens 	fsavl_node_t *fn;
4595367Sahrens 
4605367Sahrens 	fn_find.fn_guid = snapguid;
4615367Sahrens 
4625367Sahrens 	fn = avl_find(avl, &fn_find, NULL);
4635367Sahrens 	if (fn) {
4645367Sahrens 		if (snapname)
4655367Sahrens 			*snapname = fn->fn_snapname;
4665367Sahrens 		return (fn->fn_nvfs);
4675367Sahrens 	}
4685367Sahrens 	return (NULL);
4695367Sahrens }
4705367Sahrens 
4715896Sahl static void
fsavl_destroy(avl_tree_t * avl)4725896Sahl fsavl_destroy(avl_tree_t *avl)
4735896Sahl {
4745896Sahl 	fsavl_node_t *fn;
4755896Sahl 	void *cookie;
4765896Sahl 
4775896Sahl 	if (avl == NULL)
4785896Sahl 		return;
4795896Sahl 
4805896Sahl 	cookie = NULL;
4815896Sahl 	while ((fn = avl_destroy_nodes(avl, &cookie)) != NULL)
4825896Sahl 		free(fn);
4835896Sahl 	avl_destroy(avl);
4845896Sahl 	free(avl);
4855896Sahl }
4865896Sahl 
48710200SLori.Alt@Sun.COM /*
48810200SLori.Alt@Sun.COM  * Given an nvlist, produce an avl tree of snapshots, ordered by guid
48910200SLori.Alt@Sun.COM  */
4905367Sahrens static avl_tree_t *
fsavl_create(nvlist_t * fss)4915367Sahrens fsavl_create(nvlist_t *fss)
4925367Sahrens {
4935367Sahrens 	avl_tree_t *fsavl;
4945367Sahrens 	nvpair_t *fselem = NULL;
4955367Sahrens 
4965896Sahl 	if ((fsavl = malloc(sizeof (avl_tree_t))) == NULL)
4975896Sahl 		return (NULL);
4985896Sahl 
4995367Sahrens 	avl_create(fsavl, fsavl_compare, sizeof (fsavl_node_t),
5005367Sahrens 	    offsetof(fsavl_node_t, fn_node));
5015367Sahrens 
5025367Sahrens 	while ((fselem = nvlist_next_nvpair(fss, fselem)) != NULL) {
5035367Sahrens 		nvlist_t *nvfs, *snaps;
5045367Sahrens 		nvpair_t *snapelem = NULL;
5055367Sahrens 
5065367Sahrens 		VERIFY(0 == nvpair_value_nvlist(fselem, &nvfs));
5075367Sahrens 		VERIFY(0 == nvlist_lookup_nvlist(nvfs, "snaps", &snaps));
5085367Sahrens 
5095367Sahrens 		while ((snapelem =
5105367Sahrens 		    nvlist_next_nvpair(snaps, snapelem)) != NULL) {
5115367Sahrens 			fsavl_node_t *fn;
5125367Sahrens 			uint64_t guid;
5135367Sahrens 
5145367Sahrens 			VERIFY(0 == nvpair_value_uint64(snapelem, &guid));
5155896Sahl 			if ((fn = malloc(sizeof (fsavl_node_t))) == NULL) {
5165896Sahl 				fsavl_destroy(fsavl);
5175896Sahl 				return (NULL);
5185896Sahl 			}
5195367Sahrens 			fn->fn_nvfs = nvfs;
5205367Sahrens 			fn->fn_snapname = nvpair_name(snapelem);
5215367Sahrens 			fn->fn_guid = guid;
5225367Sahrens 
5235367Sahrens 			/*
5245367Sahrens 			 * Note: if there are multiple snaps with the
5255367Sahrens 			 * same GUID, we ignore all but one.
5265367Sahrens 			 */
5275367Sahrens 			if (avl_find(fsavl, fn, NULL) == NULL)
5285367Sahrens 				avl_add(fsavl, fn);
5295367Sahrens 			else
5305367Sahrens 				free(fn);
5315367Sahrens 		}
5325367Sahrens 	}
5335367Sahrens 
5345367Sahrens 	return (fsavl);
5355367Sahrens }
5365367Sahrens 
5375367Sahrens /*
5385367Sahrens  * Routines for dealing with the giant nvlist of fs-nvlists, etc.
5395367Sahrens  */
5405367Sahrens typedef struct send_data {
5415367Sahrens 	uint64_t parent_fromsnap_guid;
5425367Sahrens 	nvlist_t *parent_snaps;
5435367Sahrens 	nvlist_t *fss;
5447265Sahrens 	nvlist_t *snapprops;
5455367Sahrens 	const char *fromsnap;
5465367Sahrens 	const char *tosnap;
54711022STom.Erickson@Sun.COM 	boolean_t recursive;
5485367Sahrens 
5495367Sahrens 	/*
5505367Sahrens 	 * The header nvlist is of the following format:
5515367Sahrens 	 * {
5525367Sahrens 	 *   "tosnap" -> string
5535367Sahrens 	 *   "fromsnap" -> string (if incremental)
5545367Sahrens 	 *   "fss" -> {
5555367Sahrens 	 *	id -> {
5565367Sahrens 	 *
5575367Sahrens 	 *	 "name" -> string (full name; for debugging)
5585367Sahrens 	 *	 "parentfromsnap" -> number (guid of fromsnap in parent)
5595367Sahrens 	 *
5605367Sahrens 	 *	 "props" -> { name -> value (only if set here) }
5615367Sahrens 	 *	 "snaps" -> { name (lastname) -> number (guid) }
5627265Sahrens 	 *	 "snapprops" -> { name (lastname) -> { name -> value } }
5635367Sahrens 	 *
5645367Sahrens 	 *	 "origin" -> number (guid) (if clone)
5655367Sahrens 	 *	 "sent" -> boolean (not on-disk)
5665367Sahrens 	 *	}
5675367Sahrens 	 *   }
5685367Sahrens 	 * }
5695367Sahrens 	 *
5705367Sahrens 	 */
5715367Sahrens } send_data_t;
5725367Sahrens 
5737265Sahrens static void send_iterate_prop(zfs_handle_t *zhp, nvlist_t *nv);
5747265Sahrens 
5755367Sahrens static int
send_iterate_snap(zfs_handle_t * zhp,void * arg)5765367Sahrens send_iterate_snap(zfs_handle_t *zhp, void *arg)
5775367Sahrens {
5785367Sahrens 	send_data_t *sd = arg;
5795367Sahrens 	uint64_t guid = zhp->zfs_dmustats.dds_guid;
5805367Sahrens 	char *snapname;
5817265Sahrens 	nvlist_t *nv;
5825367Sahrens 
5835367Sahrens 	snapname = strrchr(zhp->zfs_name, '@')+1;
5845367Sahrens 
5855367Sahrens 	VERIFY(0 == nvlist_add_uint64(sd->parent_snaps, snapname, guid));
5865367Sahrens 	/*
5875367Sahrens 	 * NB: if there is no fromsnap here (it's a newly created fs in
5885367Sahrens 	 * an incremental replication), we will substitute the tosnap.
5895367Sahrens 	 */
5905367Sahrens 	if ((sd->fromsnap && strcmp(snapname, sd->fromsnap) == 0) ||
5915367Sahrens 	    (sd->parent_fromsnap_guid == 0 && sd->tosnap &&
5925367Sahrens 	    strcmp(snapname, sd->tosnap) == 0)) {
5935367Sahrens 		sd->parent_fromsnap_guid = guid;
5945367Sahrens 	}
5955367Sahrens 
5967265Sahrens 	VERIFY(0 == nvlist_alloc(&nv, NV_UNIQUE_NAME, 0));
5977265Sahrens 	send_iterate_prop(zhp, nv);
5987265Sahrens 	VERIFY(0 == nvlist_add_nvlist(sd->snapprops, snapname, nv));
5997265Sahrens 	nvlist_free(nv);
6007265Sahrens 
6015367Sahrens 	zfs_close(zhp);
6025367Sahrens 	return (0);
6035367Sahrens }
6045367Sahrens 
6055367Sahrens static void
send_iterate_prop(zfs_handle_t * zhp,nvlist_t * nv)6065367Sahrens send_iterate_prop(zfs_handle_t *zhp, nvlist_t *nv)
6075367Sahrens {
6085367Sahrens 	nvpair_t *elem = NULL;
6095367Sahrens 
6105367Sahrens 	while ((elem = nvlist_next_nvpair(zhp->zfs_props, elem)) != NULL) {
6115367Sahrens 		char *propname = nvpair_name(elem);
6125367Sahrens 		zfs_prop_t prop = zfs_name_to_prop(propname);
6135367Sahrens 		nvlist_t *propnv;
6145367Sahrens 
61510960SEric.Schrock@Sun.COM 		if (!zfs_prop_user(propname)) {
61610960SEric.Schrock@Sun.COM 			/*
61710960SEric.Schrock@Sun.COM 			 * Realistically, this should never happen.  However,
61810960SEric.Schrock@Sun.COM 			 * we want the ability to add DSL properties without
61910960SEric.Schrock@Sun.COM 			 * needing to make incompatible version changes.  We
62010960SEric.Schrock@Sun.COM 			 * need to ignore unknown properties to allow older
62110960SEric.Schrock@Sun.COM 			 * software to still send datasets containing these
62210960SEric.Schrock@Sun.COM 			 * properties, with the unknown properties elided.
62310960SEric.Schrock@Sun.COM 			 */
62410960SEric.Schrock@Sun.COM 			if (prop == ZPROP_INVAL)
62510960SEric.Schrock@Sun.COM 				continue;
6269396SMatthew.Ahrens@Sun.COM 
62710960SEric.Schrock@Sun.COM 			if (zfs_prop_readonly(prop))
62810960SEric.Schrock@Sun.COM 				continue;
62910960SEric.Schrock@Sun.COM 		}
6305367Sahrens 
6315367Sahrens 		verify(nvpair_value_nvlist(elem, &propnv) == 0);
63210195SSanjeev.Bagewadi@Sun.COM 		if (prop == ZFS_PROP_QUOTA || prop == ZFS_PROP_RESERVATION ||
63310195SSanjeev.Bagewadi@Sun.COM 		    prop == ZFS_PROP_REFQUOTA ||
63410195SSanjeev.Bagewadi@Sun.COM 		    prop == ZFS_PROP_REFRESERVATION) {
63511022STom.Erickson@Sun.COM 			char *source;
6365367Sahrens 			uint64_t value;
6375367Sahrens 			verify(nvlist_lookup_uint64(propnv,
6385367Sahrens 			    ZPROP_VALUE, &value) == 0);
6397265Sahrens 			if (zhp->zfs_type == ZFS_TYPE_SNAPSHOT)
6407265Sahrens 				continue;
64111022STom.Erickson@Sun.COM 			/*
64211022STom.Erickson@Sun.COM 			 * May have no source before SPA_VERSION_RECVD_PROPS,
64311022STom.Erickson@Sun.COM 			 * but is still modifiable.
64411022STom.Erickson@Sun.COM 			 */
64511022STom.Erickson@Sun.COM 			if (nvlist_lookup_string(propnv,
64611022STom.Erickson@Sun.COM 			    ZPROP_SOURCE, &source) == 0) {
64711022STom.Erickson@Sun.COM 				if ((strcmp(source, zhp->zfs_name) != 0) &&
64811022STom.Erickson@Sun.COM 				    (strcmp(source,
64911022STom.Erickson@Sun.COM 				    ZPROP_SOURCE_VAL_RECVD) != 0))
65011022STom.Erickson@Sun.COM 					continue;
65111022STom.Erickson@Sun.COM 			}
6525367Sahrens 		} else {
6535367Sahrens 			char *source;
6545367Sahrens 			if (nvlist_lookup_string(propnv,
6555367Sahrens 			    ZPROP_SOURCE, &source) != 0)
6565367Sahrens 				continue;
65711022STom.Erickson@Sun.COM 			if ((strcmp(source, zhp->zfs_name) != 0) &&
65811022STom.Erickson@Sun.COM 			    (strcmp(source, ZPROP_SOURCE_VAL_RECVD) != 0))
6595367Sahrens 				continue;
6605367Sahrens 		}
6615367Sahrens 
6625367Sahrens 		if (zfs_prop_user(propname) ||
6635367Sahrens 		    zfs_prop_get_type(prop) == PROP_TYPE_STRING) {
6645367Sahrens 			char *value;
6655367Sahrens 			verify(nvlist_lookup_string(propnv,
6665367Sahrens 			    ZPROP_VALUE, &value) == 0);
6675367Sahrens 			VERIFY(0 == nvlist_add_string(nv, propname, value));
6685367Sahrens 		} else {
6695367Sahrens 			uint64_t value;
6705367Sahrens 			verify(nvlist_lookup_uint64(propnv,
6715367Sahrens 			    ZPROP_VALUE, &value) == 0);
6725367Sahrens 			VERIFY(0 == nvlist_add_uint64(nv, propname, value));
6735367Sahrens 		}
6745367Sahrens 	}
6755367Sahrens }
6765367Sahrens 
67710200SLori.Alt@Sun.COM /*
67810200SLori.Alt@Sun.COM  * recursively generate nvlists describing datasets.  See comment
67910200SLori.Alt@Sun.COM  * for the data structure send_data_t above for description of contents
68010200SLori.Alt@Sun.COM  * of the nvlist.
68110200SLori.Alt@Sun.COM  */
6825367Sahrens static int
send_iterate_fs(zfs_handle_t * zhp,void * arg)6835367Sahrens send_iterate_fs(zfs_handle_t *zhp, void *arg)
6845367Sahrens {
6855367Sahrens 	send_data_t *sd = arg;
6865367Sahrens 	nvlist_t *nvfs, *nv;
68711022STom.Erickson@Sun.COM 	int rv = 0;
6885367Sahrens 	uint64_t parent_fromsnap_guid_save = sd->parent_fromsnap_guid;
6895367Sahrens 	uint64_t guid = zhp->zfs_dmustats.dds_guid;
6905367Sahrens 	char guidstring[64];
6915367Sahrens 
6925367Sahrens 	VERIFY(0 == nvlist_alloc(&nvfs, NV_UNIQUE_NAME, 0));
6935367Sahrens 	VERIFY(0 == nvlist_add_string(nvfs, "name", zhp->zfs_name));
6945367Sahrens 	VERIFY(0 == nvlist_add_uint64(nvfs, "parentfromsnap",
6955367Sahrens 	    sd->parent_fromsnap_guid));
6965367Sahrens 
6975367Sahrens 	if (zhp->zfs_dmustats.dds_origin[0]) {
6985367Sahrens 		zfs_handle_t *origin = zfs_open(zhp->zfs_hdl,
6995367Sahrens 		    zhp->zfs_dmustats.dds_origin, ZFS_TYPE_SNAPSHOT);
7005367Sahrens 		if (origin == NULL)
7015367Sahrens 			return (-1);
7025367Sahrens 		VERIFY(0 == nvlist_add_uint64(nvfs, "origin",
7035367Sahrens 		    origin->zfs_dmustats.dds_guid));
7045367Sahrens 	}
7055367Sahrens 
7065367Sahrens 	/* iterate over props */
7075367Sahrens 	VERIFY(0 == nvlist_alloc(&nv, NV_UNIQUE_NAME, 0));
7085367Sahrens 	send_iterate_prop(zhp, nv);
7095367Sahrens 	VERIFY(0 == nvlist_add_nvlist(nvfs, "props", nv));
7105367Sahrens 	nvlist_free(nv);
7115367Sahrens 
7125367Sahrens 	/* iterate over snaps, and set sd->parent_fromsnap_guid */
7135367Sahrens 	sd->parent_fromsnap_guid = 0;
7145367Sahrens 	VERIFY(0 == nvlist_alloc(&sd->parent_snaps, NV_UNIQUE_NAME, 0));
7157265Sahrens 	VERIFY(0 == nvlist_alloc(&sd->snapprops, NV_UNIQUE_NAME, 0));
7165367Sahrens 	(void) zfs_iter_snapshots(zhp, send_iterate_snap, sd);
7175367Sahrens 	VERIFY(0 == nvlist_add_nvlist(nvfs, "snaps", sd->parent_snaps));
7187265Sahrens 	VERIFY(0 == nvlist_add_nvlist(nvfs, "snapprops", sd->snapprops));
7195367Sahrens 	nvlist_free(sd->parent_snaps);
7207265Sahrens 	nvlist_free(sd->snapprops);
7215367Sahrens 
7225367Sahrens 	/* add this fs to nvlist */
7235367Sahrens 	(void) snprintf(guidstring, sizeof (guidstring),
7245367Sahrens 	    "0x%llx", (longlong_t)guid);
7255367Sahrens 	VERIFY(0 == nvlist_add_nvlist(sd->fss, guidstring, nvfs));
7265367Sahrens 	nvlist_free(nvfs);
7275367Sahrens 
7285367Sahrens 	/* iterate over children */
72911022STom.Erickson@Sun.COM 	if (sd->recursive)
73011022STom.Erickson@Sun.COM 		rv = zfs_iter_filesystems(zhp, send_iterate_fs, sd);
7315367Sahrens 
7325367Sahrens 	sd->parent_fromsnap_guid = parent_fromsnap_guid_save;
7335367Sahrens 
7345367Sahrens 	zfs_close(zhp);
7355367Sahrens 	return (rv);
7365367Sahrens }
7375367Sahrens 
7385367Sahrens static int
gather_nvlist(libzfs_handle_t * hdl,const char * fsname,const char * fromsnap,const char * tosnap,boolean_t recursive,nvlist_t ** nvlp,avl_tree_t ** avlp)7395367Sahrens gather_nvlist(libzfs_handle_t *hdl, const char *fsname, const char *fromsnap,
74011022STom.Erickson@Sun.COM     const char *tosnap, boolean_t recursive, nvlist_t **nvlp, avl_tree_t **avlp)
7415367Sahrens {
7425367Sahrens 	zfs_handle_t *zhp;
7435367Sahrens 	send_data_t sd = { 0 };
7445367Sahrens 	int error;
7455367Sahrens 
7465367Sahrens 	zhp = zfs_open(hdl, fsname, ZFS_TYPE_FILESYSTEM | ZFS_TYPE_VOLUME);
7475367Sahrens 	if (zhp == NULL)
7485367Sahrens 		return (EZFS_BADTYPE);
7495367Sahrens 
7505367Sahrens 	VERIFY(0 == nvlist_alloc(&sd.fss, NV_UNIQUE_NAME, 0));
7515367Sahrens 	sd.fromsnap = fromsnap;
7525367Sahrens 	sd.tosnap = tosnap;
75311022STom.Erickson@Sun.COM 	sd.recursive = recursive;
7545896Sahl 
7555896Sahl 	if ((error = send_iterate_fs(zhp, &sd)) != 0) {
7565896Sahl 		nvlist_free(sd.fss);
7575896Sahl 		if (avlp != NULL)
7585896Sahl 			*avlp = NULL;
7595896Sahl 		*nvlp = NULL;
7605896Sahl 		return (error);
7615896Sahl 	}
7625896Sahl 
7635896Sahl 	if (avlp != NULL && (*avlp = fsavl_create(sd.fss)) == NULL) {
7645896Sahl 		nvlist_free(sd.fss);
7655896Sahl 		*nvlp = NULL;
7665896Sahl 		return (EZFS_NOMEM);
7675896Sahl 	}
7685367Sahrens 
7695367Sahrens 	*nvlp = sd.fss;
7705896Sahl 	return (0);
7715367Sahrens }
7725367Sahrens 
7735367Sahrens /*
7745367Sahrens  * Routines for dealing with the sorted snapshot functionality
7755367Sahrens  */
7765367Sahrens typedef struct zfs_node {
7775367Sahrens 	zfs_handle_t	*zn_handle;
7785367Sahrens 	avl_node_t	zn_avlnode;
7795367Sahrens } zfs_node_t;
7805367Sahrens 
7815367Sahrens static int
zfs_sort_snaps(zfs_handle_t * zhp,void * data)7825367Sahrens zfs_sort_snaps(zfs_handle_t *zhp, void *data)
7835367Sahrens {
7845367Sahrens 	avl_tree_t *avl = data;
785*12786SChris.Kirby@oracle.com 	zfs_node_t *node;
786*12786SChris.Kirby@oracle.com 	zfs_node_t search;
787*12786SChris.Kirby@oracle.com 
788*12786SChris.Kirby@oracle.com 	search.zn_handle = zhp;
789*12786SChris.Kirby@oracle.com 	node = avl_find(avl, &search, NULL);
790*12786SChris.Kirby@oracle.com 	if (node) {
791*12786SChris.Kirby@oracle.com 		/*
792*12786SChris.Kirby@oracle.com 		 * If this snapshot was renamed while we were creating the
793*12786SChris.Kirby@oracle.com 		 * AVL tree, it's possible that we already inserted it under
794*12786SChris.Kirby@oracle.com 		 * its old name. Remove the old handle before adding the new
795*12786SChris.Kirby@oracle.com 		 * one.
796*12786SChris.Kirby@oracle.com 		 */
797*12786SChris.Kirby@oracle.com 		zfs_close(node->zn_handle);
798*12786SChris.Kirby@oracle.com 		avl_remove(avl, node);
799*12786SChris.Kirby@oracle.com 		free(node);
800*12786SChris.Kirby@oracle.com 	}
801*12786SChris.Kirby@oracle.com 
802*12786SChris.Kirby@oracle.com 	node = zfs_alloc(zhp->zfs_hdl, sizeof (zfs_node_t));
8035367Sahrens 	node->zn_handle = zhp;
8045367Sahrens 	avl_add(avl, node);
805*12786SChris.Kirby@oracle.com 
8065367Sahrens 	return (0);
8075367Sahrens }
8085367Sahrens 
8095367Sahrens static int
zfs_snapshot_compare(const void * larg,const void * rarg)8105367Sahrens zfs_snapshot_compare(const void *larg, const void *rarg)
8115367Sahrens {
8125367Sahrens 	zfs_handle_t *l = ((zfs_node_t *)larg)->zn_handle;
8135367Sahrens 	zfs_handle_t *r = ((zfs_node_t *)rarg)->zn_handle;
8145367Sahrens 	uint64_t lcreate, rcreate;
8155367Sahrens 
8165367Sahrens 	/*
8175367Sahrens 	 * Sort them according to creation time.  We use the hidden
8185367Sahrens 	 * CREATETXG property to get an absolute ordering of snapshots.
8195367Sahrens 	 */
8205367Sahrens 	lcreate = zfs_prop_get_int(l, ZFS_PROP_CREATETXG);
8215367Sahrens 	rcreate = zfs_prop_get_int(r, ZFS_PROP_CREATETXG);
8225367Sahrens 
8235367Sahrens 	if (lcreate < rcreate)
8245367Sahrens 		return (-1);
8255367Sahrens 	else if (lcreate > rcreate)
8265367Sahrens 		return (+1);
8275367Sahrens 	else
8285367Sahrens 		return (0);
8295367Sahrens }
8305367Sahrens 
83110342Schris.kirby@sun.com int
zfs_iter_snapshots_sorted(zfs_handle_t * zhp,zfs_iter_f callback,void * data)8325367Sahrens zfs_iter_snapshots_sorted(zfs_handle_t *zhp, zfs_iter_f callback, void *data)
8335367Sahrens {
8345367Sahrens 	int ret = 0;
8355367Sahrens 	zfs_node_t *node;
8365367Sahrens 	avl_tree_t avl;
8375367Sahrens 	void *cookie = NULL;
8385367Sahrens 
8395367Sahrens 	avl_create(&avl, zfs_snapshot_compare,
8405367Sahrens 	    sizeof (zfs_node_t), offsetof(zfs_node_t, zn_avlnode));
8415367Sahrens 
8425367Sahrens 	ret = zfs_iter_snapshots(zhp, zfs_sort_snaps, &avl);
8435367Sahrens 
8445367Sahrens 	for (node = avl_first(&avl); node != NULL; node = AVL_NEXT(&avl, node))
8455367Sahrens 		ret |= callback(node->zn_handle, data);
8465367Sahrens 
8475367Sahrens 	while ((node = avl_destroy_nodes(&avl, &cookie)) != NULL)
8485367Sahrens 		free(node);
8495367Sahrens 
8505367Sahrens 	avl_destroy(&avl);
8515367Sahrens 
8525367Sahrens 	return (ret);
8535367Sahrens }
8545367Sahrens 
8555367Sahrens /*
8565367Sahrens  * Routines specific to "zfs send"
8575367Sahrens  */
8585367Sahrens typedef struct send_dump_data {
8595367Sahrens 	/* these are all just the short snapname (the part after the @) */
8605367Sahrens 	const char *fromsnap;
8615367Sahrens 	const char *tosnap;
86211007SLori.Alt@Sun.COM 	char prevsnap[ZFS_MAXNAMELEN];
863*12786SChris.Kirby@oracle.com 	uint64_t prevsnap_obj;
8645367Sahrens 	boolean_t seenfrom, seento, replicate, doall, fromorigin;
8655367Sahrens 	boolean_t verbose;
8665367Sahrens 	int outfd;
8675367Sahrens 	boolean_t err;
8685367Sahrens 	nvlist_t *fss;
8695367Sahrens 	avl_tree_t *fsavl;
87011007SLori.Alt@Sun.COM 	snapfilter_cb_t *filter_cb;
87111007SLori.Alt@Sun.COM 	void *filter_cb_arg;
87212296SLin.Ling@Sun.COM 	nvlist_t *debugnv;
873*12786SChris.Kirby@oracle.com 	char holdtag[ZFS_MAXNAMELEN];
874*12786SChris.Kirby@oracle.com 	int cleanup_fd;
8755367Sahrens } send_dump_data_t;
8765367Sahrens 
8775367Sahrens /*
8785367Sahrens  * Dumps a backup of the given snapshot (incremental from fromsnap if it's not
8795367Sahrens  * NULL) to the file descriptor specified by outfd.
8805367Sahrens  */
8815367Sahrens static int
dump_ioctl(zfs_handle_t * zhp,const char * fromsnap,uint64_t fromsnap_obj,boolean_t fromorigin,int outfd,nvlist_t * debugnv)882*12786SChris.Kirby@oracle.com dump_ioctl(zfs_handle_t *zhp, const char *fromsnap, uint64_t fromsnap_obj,
883*12786SChris.Kirby@oracle.com     boolean_t fromorigin, int outfd, nvlist_t *debugnv)
8845367Sahrens {
8855367Sahrens 	zfs_cmd_t zc = { 0 };
8865367Sahrens 	libzfs_handle_t *hdl = zhp->zfs_hdl;
88712296SLin.Ling@Sun.COM 	nvlist_t *thisdbg;
8885367Sahrens 
8895367Sahrens 	assert(zhp->zfs_type == ZFS_TYPE_SNAPSHOT);
890*12786SChris.Kirby@oracle.com 	assert(fromsnap_obj == 0 || !fromorigin);
8915367Sahrens 
8925367Sahrens 	(void) strlcpy(zc.zc_name, zhp->zfs_name, sizeof (zc.zc_name));
8935367Sahrens 	zc.zc_cookie = outfd;
8945367Sahrens 	zc.zc_obj = fromorigin;
895*12786SChris.Kirby@oracle.com 	zc.zc_sendobj = zfs_prop_get_int(zhp, ZFS_PROP_OBJSETID);
896*12786SChris.Kirby@oracle.com 	zc.zc_fromobj = fromsnap_obj;
89711668SChris.Kirby@sun.com 
89812296SLin.Ling@Sun.COM 	VERIFY(0 == nvlist_alloc(&thisdbg, NV_UNIQUE_NAME, 0));
89912296SLin.Ling@Sun.COM 	if (fromsnap && fromsnap[0] != '\0') {
90012296SLin.Ling@Sun.COM 		VERIFY(0 == nvlist_add_string(thisdbg,
90112296SLin.Ling@Sun.COM 		    "fromsnap", fromsnap));
90212296SLin.Ling@Sun.COM 	}
90312296SLin.Ling@Sun.COM 
9045367Sahrens 	if (ioctl(zhp->zfs_hdl->libzfs_fd, ZFS_IOC_SEND, &zc) != 0) {
9055367Sahrens 		char errbuf[1024];
9065367Sahrens 		(void) snprintf(errbuf, sizeof (errbuf), dgettext(TEXT_DOMAIN,
9075367Sahrens 		    "warning: cannot send '%s'"), zhp->zfs_name);
9085367Sahrens 
90912296SLin.Ling@Sun.COM 		VERIFY(0 == nvlist_add_uint64(thisdbg, "error", errno));
91012296SLin.Ling@Sun.COM 		if (debugnv) {
91112296SLin.Ling@Sun.COM 			VERIFY(0 == nvlist_add_nvlist(debugnv,
91212296SLin.Ling@Sun.COM 			    zhp->zfs_name, thisdbg));
91312296SLin.Ling@Sun.COM 		}
91412296SLin.Ling@Sun.COM 		nvlist_free(thisdbg);
91512296SLin.Ling@Sun.COM 
9165367Sahrens 		switch (errno) {
9175367Sahrens 
9185367Sahrens 		case EXDEV:
9195367Sahrens 			zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
9205367Sahrens 			    "not an earlier snapshot from the same fs"));
9215367Sahrens 			return (zfs_error(hdl, EZFS_CROSSTARGET, errbuf));
9225367Sahrens 
9235367Sahrens 		case ENOENT:
9245367Sahrens 			if (zfs_dataset_exists(hdl, zc.zc_name,
9255367Sahrens 			    ZFS_TYPE_SNAPSHOT)) {
9265367Sahrens 				zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
9275367Sahrens 				    "incremental source (@%s) does not exist"),
9285367Sahrens 				    zc.zc_value);
9295367Sahrens 			}
9305367Sahrens 			return (zfs_error(hdl, EZFS_NOENT, errbuf));
9315367Sahrens 
9325367Sahrens 		case EDQUOT:
9335367Sahrens 		case EFBIG:
9345367Sahrens 		case EIO:
9355367Sahrens 		case ENOLINK:
9365367Sahrens 		case ENOSPC:
9375367Sahrens 		case ENOSTR:
9385367Sahrens 		case ENXIO:
9395367Sahrens 		case EPIPE:
9405367Sahrens 		case ERANGE:
9415367Sahrens 		case EFAULT:
9425367Sahrens 		case EROFS:
9435367Sahrens 			zfs_error_aux(hdl, strerror(errno));
9445367Sahrens 			return (zfs_error(hdl, EZFS_BADBACKUP, errbuf));
9455367Sahrens 
9465367Sahrens 		default:
9475367Sahrens 			return (zfs_standard_error(hdl, errno, errbuf));
9485367Sahrens 		}
9495367Sahrens 	}
9505367Sahrens 
95112296SLin.Ling@Sun.COM 	if (debugnv)
95212296SLin.Ling@Sun.COM 		VERIFY(0 == nvlist_add_nvlist(debugnv, zhp->zfs_name, thisdbg));
95312296SLin.Ling@Sun.COM 	nvlist_free(thisdbg);
95412296SLin.Ling@Sun.COM 
9555367Sahrens 	return (0);
9565367Sahrens }
9575367Sahrens 
9585367Sahrens static int
hold_for_send(zfs_handle_t * zhp,send_dump_data_t * sdd)959*12786SChris.Kirby@oracle.com hold_for_send(zfs_handle_t *zhp, send_dump_data_t *sdd)
960*12786SChris.Kirby@oracle.com {
961*12786SChris.Kirby@oracle.com 	zfs_handle_t *pzhp;
962*12786SChris.Kirby@oracle.com 	int error = 0;
963*12786SChris.Kirby@oracle.com 	char *thissnap;
964*12786SChris.Kirby@oracle.com 
965*12786SChris.Kirby@oracle.com 	assert(zhp->zfs_type == ZFS_TYPE_SNAPSHOT);
966*12786SChris.Kirby@oracle.com 
967*12786SChris.Kirby@oracle.com 	/*
968*12786SChris.Kirby@oracle.com 	 * zfs_send() only opens a cleanup_fd for sends that need it,
969*12786SChris.Kirby@oracle.com 	 * e.g. replication and doall.
970*12786SChris.Kirby@oracle.com 	 */
971*12786SChris.Kirby@oracle.com 	if (sdd->cleanup_fd == -1)
972*12786SChris.Kirby@oracle.com 		return (0);
973*12786SChris.Kirby@oracle.com 
974*12786SChris.Kirby@oracle.com 	thissnap = strchr(zhp->zfs_name, '@') + 1;
975*12786SChris.Kirby@oracle.com 	*(thissnap - 1) = '\0';
976*12786SChris.Kirby@oracle.com 	pzhp = zfs_open(zhp->zfs_hdl, zhp->zfs_name, ZFS_TYPE_DATASET);
977*12786SChris.Kirby@oracle.com 	*(thissnap - 1) = '@';
978*12786SChris.Kirby@oracle.com 
979*12786SChris.Kirby@oracle.com 	/*
980*12786SChris.Kirby@oracle.com 	 * It's OK if the parent no longer exists.  The send code will
981*12786SChris.Kirby@oracle.com 	 * handle that error.
982*12786SChris.Kirby@oracle.com 	 */
983*12786SChris.Kirby@oracle.com 	if (pzhp) {
984*12786SChris.Kirby@oracle.com 		error = zfs_hold(pzhp, thissnap, sdd->holdtag,
985*12786SChris.Kirby@oracle.com 		    B_FALSE, B_TRUE, B_TRUE, sdd->cleanup_fd,
986*12786SChris.Kirby@oracle.com 		    zfs_prop_get_int(zhp, ZFS_PROP_OBJSETID),
987*12786SChris.Kirby@oracle.com 		    zfs_prop_get_int(zhp, ZFS_PROP_CREATETXG));
988*12786SChris.Kirby@oracle.com 		zfs_close(pzhp);
989*12786SChris.Kirby@oracle.com 	}
990*12786SChris.Kirby@oracle.com 
991*12786SChris.Kirby@oracle.com 	return (error);
992*12786SChris.Kirby@oracle.com }
993*12786SChris.Kirby@oracle.com 
994*12786SChris.Kirby@oracle.com static int
dump_snapshot(zfs_handle_t * zhp,void * arg)9955367Sahrens dump_snapshot(zfs_handle_t *zhp, void *arg)
9965367Sahrens {
9975367Sahrens 	send_dump_data_t *sdd = arg;
998*12786SChris.Kirby@oracle.com 	char *thissnap;
9995367Sahrens 	int err;
100012282STom.Erickson@Sun.COM 	boolean_t isfromsnap, istosnap;
100112282STom.Erickson@Sun.COM 	boolean_t exclude = B_FALSE;
10025367Sahrens 
10035367Sahrens 	thissnap = strchr(zhp->zfs_name, '@') + 1;
100412282STom.Erickson@Sun.COM 	isfromsnap = (sdd->fromsnap != NULL &&
100512282STom.Erickson@Sun.COM 	    strcmp(sdd->fromsnap, thissnap) == 0);
100612282STom.Erickson@Sun.COM 
100712282STom.Erickson@Sun.COM 	if (!sdd->seenfrom && isfromsnap) {
1008*12786SChris.Kirby@oracle.com 		err = hold_for_send(zhp, sdd);
1009*12786SChris.Kirby@oracle.com 		if (err == 0) {
1010*12786SChris.Kirby@oracle.com 			sdd->seenfrom = B_TRUE;
1011*12786SChris.Kirby@oracle.com 			(void) strcpy(sdd->prevsnap, thissnap);
1012*12786SChris.Kirby@oracle.com 			sdd->prevsnap_obj = zfs_prop_get_int(zhp,
1013*12786SChris.Kirby@oracle.com 			    ZFS_PROP_OBJSETID);
1014*12786SChris.Kirby@oracle.com 		} else if (err == ENOENT) {
1015*12786SChris.Kirby@oracle.com 			err = 0;
1016*12786SChris.Kirby@oracle.com 		}
10175367Sahrens 		zfs_close(zhp);
1018*12786SChris.Kirby@oracle.com 		return (err);
10195367Sahrens 	}
10205367Sahrens 
10215367Sahrens 	if (sdd->seento || !sdd->seenfrom) {
10225367Sahrens 		zfs_close(zhp);
10235367Sahrens 		return (0);
10245367Sahrens 	}
10255367Sahrens 
102612282STom.Erickson@Sun.COM 	istosnap = (strcmp(sdd->tosnap, thissnap) == 0);
102712282STom.Erickson@Sun.COM 	if (istosnap)
102811007SLori.Alt@Sun.COM 		sdd->seento = B_TRUE;
102911007SLori.Alt@Sun.COM 
103012282STom.Erickson@Sun.COM 	if (!sdd->doall && !isfromsnap && !istosnap) {
103112282STom.Erickson@Sun.COM 		if (sdd->replicate) {
103212282STom.Erickson@Sun.COM 			char *snapname;
103312282STom.Erickson@Sun.COM 			nvlist_t *snapprops;
103412282STom.Erickson@Sun.COM 			/*
103512282STom.Erickson@Sun.COM 			 * Filter out all intermediate snapshots except origin
103612282STom.Erickson@Sun.COM 			 * snapshots needed to replicate clones.
103712282STom.Erickson@Sun.COM 			 */
103812282STom.Erickson@Sun.COM 			nvlist_t *nvfs = fsavl_find(sdd->fsavl,
103912282STom.Erickson@Sun.COM 			    zhp->zfs_dmustats.dds_guid, &snapname);
104012282STom.Erickson@Sun.COM 
104112282STom.Erickson@Sun.COM 			VERIFY(0 == nvlist_lookup_nvlist(nvfs,
104212282STom.Erickson@Sun.COM 			    "snapprops", &snapprops));
104312282STom.Erickson@Sun.COM 			VERIFY(0 == nvlist_lookup_nvlist(snapprops,
104412282STom.Erickson@Sun.COM 			    thissnap, &snapprops));
104512282STom.Erickson@Sun.COM 			exclude = !nvlist_exists(snapprops, "is_clone_origin");
104612282STom.Erickson@Sun.COM 		} else {
104712282STom.Erickson@Sun.COM 			exclude = B_TRUE;
104812282STom.Erickson@Sun.COM 		}
104912282STom.Erickson@Sun.COM 	}
105012282STom.Erickson@Sun.COM 
105111007SLori.Alt@Sun.COM 	/*
105211007SLori.Alt@Sun.COM 	 * If a filter function exists, call it to determine whether
105311007SLori.Alt@Sun.COM 	 * this snapshot will be sent.
105411007SLori.Alt@Sun.COM 	 */
105512282STom.Erickson@Sun.COM 	if (exclude || (sdd->filter_cb != NULL &&
105612282STom.Erickson@Sun.COM 	    sdd->filter_cb(zhp, sdd->filter_cb_arg) == B_FALSE)) {
105711007SLori.Alt@Sun.COM 		/*
105811007SLori.Alt@Sun.COM 		 * This snapshot is filtered out.  Don't send it, and don't
1059*12786SChris.Kirby@oracle.com 		 * set prevsnap_obj, so it will be as if this snapshot didn't
106011007SLori.Alt@Sun.COM 		 * exist, and the next accepted snapshot will be sent as
106111007SLori.Alt@Sun.COM 		 * an incremental from the last accepted one, or as the
106211007SLori.Alt@Sun.COM 		 * first (and full) snapshot in the case of a replication,
106311007SLori.Alt@Sun.COM 		 * non-incremental send.
106411007SLori.Alt@Sun.COM 		 */
106511007SLori.Alt@Sun.COM 		zfs_close(zhp);
106611007SLori.Alt@Sun.COM 		return (0);
106711007SLori.Alt@Sun.COM 	}
106811007SLori.Alt@Sun.COM 
1069*12786SChris.Kirby@oracle.com 	err = hold_for_send(zhp, sdd);
1070*12786SChris.Kirby@oracle.com 	if (err) {
1071*12786SChris.Kirby@oracle.com 		if (err == ENOENT)
1072*12786SChris.Kirby@oracle.com 			err = 0;
1073*12786SChris.Kirby@oracle.com 		zfs_close(zhp);
1074*12786SChris.Kirby@oracle.com 		return (err);
1075*12786SChris.Kirby@oracle.com 	}
1076*12786SChris.Kirby@oracle.com 
10775367Sahrens 	/* send it */
10785367Sahrens 	if (sdd->verbose) {
10795367Sahrens 		(void) fprintf(stderr, "sending from @%s to %s\n",
108011007SLori.Alt@Sun.COM 		    sdd->prevsnap, zhp->zfs_name);
10815367Sahrens 	}
10825367Sahrens 
1083*12786SChris.Kirby@oracle.com 	err = dump_ioctl(zhp, sdd->prevsnap, sdd->prevsnap_obj,
108411007SLori.Alt@Sun.COM 	    sdd->prevsnap[0] == '\0' && (sdd->fromorigin || sdd->replicate),
1085*12786SChris.Kirby@oracle.com 	    sdd->outfd, sdd->debugnv);
1086*12786SChris.Kirby@oracle.com 
1087*12786SChris.Kirby@oracle.com 	(void) strcpy(sdd->prevsnap, thissnap);
1088*12786SChris.Kirby@oracle.com 	sdd->prevsnap_obj = zfs_prop_get_int(zhp, ZFS_PROP_OBJSETID);
10895367Sahrens 	zfs_close(zhp);
10905367Sahrens 	return (err);
10915367Sahrens }
10925367Sahrens 
10935367Sahrens static int
dump_filesystem(zfs_handle_t * zhp,void * arg)10945367Sahrens dump_filesystem(zfs_handle_t *zhp, void *arg)
10955367Sahrens {
10965367Sahrens 	int rv = 0;
10975367Sahrens 	send_dump_data_t *sdd = arg;
10985367Sahrens 	boolean_t missingfrom = B_FALSE;
10995367Sahrens 	zfs_cmd_t zc = { 0 };
11005367Sahrens 
11015367Sahrens 	(void) snprintf(zc.zc_name, sizeof (zc.zc_name), "%s@%s",
11025367Sahrens 	    zhp->zfs_name, sdd->tosnap);
11035367Sahrens 	if (ioctl(zhp->zfs_hdl->libzfs_fd, ZFS_IOC_OBJSET_STATS, &zc) != 0) {
11045367Sahrens 		(void) fprintf(stderr, "WARNING: "
11055367Sahrens 		    "could not send %s@%s: does not exist\n",
11065367Sahrens 		    zhp->zfs_name, sdd->tosnap);
11075367Sahrens 		sdd->err = B_TRUE;
11085367Sahrens 		return (0);
11095367Sahrens 	}
11105367Sahrens 
11115367Sahrens 	if (sdd->replicate && sdd->fromsnap) {
11125367Sahrens 		/*
11135367Sahrens 		 * If this fs does not have fromsnap, and we're doing
11145367Sahrens 		 * recursive, we need to send a full stream from the
11155367Sahrens 		 * beginning (or an incremental from the origin if this
11165367Sahrens 		 * is a clone).  If we're doing non-recursive, then let
11175367Sahrens 		 * them get the error.
11185367Sahrens 		 */
11195367Sahrens 		(void) snprintf(zc.zc_name, sizeof (zc.zc_name), "%s@%s",
11205367Sahrens 		    zhp->zfs_name, sdd->fromsnap);
11215367Sahrens 		if (ioctl(zhp->zfs_hdl->libzfs_fd,
11225367Sahrens 		    ZFS_IOC_OBJSET_STATS, &zc) != 0) {
11235367Sahrens 			missingfrom = B_TRUE;
11245367Sahrens 		}
11255367Sahrens 	}
11265367Sahrens 
112712282STom.Erickson@Sun.COM 	sdd->seenfrom = sdd->seento = sdd->prevsnap[0] = 0;
1128*12786SChris.Kirby@oracle.com 	sdd->prevsnap_obj = 0;
112912282STom.Erickson@Sun.COM 	if (sdd->fromsnap == NULL || missingfrom)
113012282STom.Erickson@Sun.COM 		sdd->seenfrom = B_TRUE;
113112282STom.Erickson@Sun.COM 
113212282STom.Erickson@Sun.COM 	rv = zfs_iter_snapshots_sorted(zhp, dump_snapshot, arg);
113312282STom.Erickson@Sun.COM 	if (!sdd->seenfrom) {
113412282STom.Erickson@Sun.COM 		(void) fprintf(stderr,
113512282STom.Erickson@Sun.COM 		    "WARNING: could not send %s@%s:\n"
113612282STom.Erickson@Sun.COM 		    "incremental source (%s@%s) does not exist\n",
113712282STom.Erickson@Sun.COM 		    zhp->zfs_name, sdd->tosnap,
113812282STom.Erickson@Sun.COM 		    zhp->zfs_name, sdd->fromsnap);
113912282STom.Erickson@Sun.COM 		sdd->err = B_TRUE;
114012282STom.Erickson@Sun.COM 	} else if (!sdd->seento) {
114112282STom.Erickson@Sun.COM 		if (sdd->fromsnap) {
11425367Sahrens 			(void) fprintf(stderr,
11435367Sahrens 			    "WARNING: could not send %s@%s:\n"
114412282STom.Erickson@Sun.COM 			    "incremental source (%s@%s) "
114512282STom.Erickson@Sun.COM 			    "is not earlier than it\n",
11465367Sahrens 			    zhp->zfs_name, sdd->tosnap,
11475367Sahrens 			    zhp->zfs_name, sdd->fromsnap);
114812282STom.Erickson@Sun.COM 		} else {
114912282STom.Erickson@Sun.COM 			(void) fprintf(stderr, "WARNING: "
115012282STom.Erickson@Sun.COM 			    "could not send %s@%s: does not exist\n",
115112282STom.Erickson@Sun.COM 			    zhp->zfs_name, sdd->tosnap);
11525367Sahrens 		}
115312282STom.Erickson@Sun.COM 		sdd->err = B_TRUE;
11545367Sahrens 	}
11555367Sahrens 
11565367Sahrens 	return (rv);
11575367Sahrens }
11585367Sahrens 
11595367Sahrens static int
dump_filesystems(zfs_handle_t * rzhp,void * arg)11605367Sahrens dump_filesystems(zfs_handle_t *rzhp, void *arg)
11615367Sahrens {
11625367Sahrens 	send_dump_data_t *sdd = arg;
11635367Sahrens 	nvpair_t *fspair;
11645367Sahrens 	boolean_t needagain, progress;
11655367Sahrens 
11665367Sahrens 	if (!sdd->replicate)
11675367Sahrens 		return (dump_filesystem(rzhp, sdd));
11685367Sahrens 
116912282STom.Erickson@Sun.COM 	/* Mark the clone origin snapshots. */
117012282STom.Erickson@Sun.COM 	for (fspair = nvlist_next_nvpair(sdd->fss, NULL); fspair;
117112282STom.Erickson@Sun.COM 	    fspair = nvlist_next_nvpair(sdd->fss, fspair)) {
117212282STom.Erickson@Sun.COM 		nvlist_t *nvfs;
117312282STom.Erickson@Sun.COM 		uint64_t origin_guid = 0;
117412282STom.Erickson@Sun.COM 
117512282STom.Erickson@Sun.COM 		VERIFY(0 == nvpair_value_nvlist(fspair, &nvfs));
117612282STom.Erickson@Sun.COM 		(void) nvlist_lookup_uint64(nvfs, "origin", &origin_guid);
117712282STom.Erickson@Sun.COM 		if (origin_guid != 0) {
117812282STom.Erickson@Sun.COM 			char *snapname;
117912282STom.Erickson@Sun.COM 			nvlist_t *origin_nv = fsavl_find(sdd->fsavl,
118012282STom.Erickson@Sun.COM 			    origin_guid, &snapname);
118112282STom.Erickson@Sun.COM 			if (origin_nv != NULL) {
118212282STom.Erickson@Sun.COM 				nvlist_t *snapprops;
118312282STom.Erickson@Sun.COM 				VERIFY(0 == nvlist_lookup_nvlist(origin_nv,
118412282STom.Erickson@Sun.COM 				    "snapprops", &snapprops));
118512282STom.Erickson@Sun.COM 				VERIFY(0 == nvlist_lookup_nvlist(snapprops,
118612282STom.Erickson@Sun.COM 				    snapname, &snapprops));
118712282STom.Erickson@Sun.COM 				VERIFY(0 == nvlist_add_boolean(
118812282STom.Erickson@Sun.COM 				    snapprops, "is_clone_origin"));
118912282STom.Erickson@Sun.COM 			}
119012282STom.Erickson@Sun.COM 		}
119112282STom.Erickson@Sun.COM 	}
11925367Sahrens again:
11935367Sahrens 	needagain = progress = B_FALSE;
11945367Sahrens 	for (fspair = nvlist_next_nvpair(sdd->fss, NULL); fspair;
11955367Sahrens 	    fspair = nvlist_next_nvpair(sdd->fss, fspair)) {
11965367Sahrens 		nvlist_t *fslist;
11975367Sahrens 		char *fsname;
11985367Sahrens 		zfs_handle_t *zhp;
11995367Sahrens 		int err;
12005367Sahrens 		uint64_t origin_guid = 0;
12015367Sahrens 
12025367Sahrens 		VERIFY(nvpair_value_nvlist(fspair, &fslist) == 0);
12035367Sahrens 		if (nvlist_lookup_boolean(fslist, "sent") == 0)
12045367Sahrens 			continue;
12055367Sahrens 
12065367Sahrens 		VERIFY(nvlist_lookup_string(fslist, "name", &fsname) == 0);
12075367Sahrens 		(void) nvlist_lookup_uint64(fslist, "origin", &origin_guid);
12085367Sahrens 
120912282STom.Erickson@Sun.COM 		if (origin_guid != 0) {
121012282STom.Erickson@Sun.COM 			nvlist_t *origin_nv = fsavl_find(sdd->fsavl,
121112282STom.Erickson@Sun.COM 			    origin_guid, NULL);
121212282STom.Erickson@Sun.COM 			if (origin_nv != NULL &&
121312282STom.Erickson@Sun.COM 			    nvlist_lookup_boolean(origin_nv,
121412282STom.Erickson@Sun.COM 			    "sent") == ENOENT) {
121512282STom.Erickson@Sun.COM 				/*
121612282STom.Erickson@Sun.COM 				 * origin has not been sent yet;
121712282STom.Erickson@Sun.COM 				 * skip this clone.
121812282STom.Erickson@Sun.COM 				 */
121912282STom.Erickson@Sun.COM 				needagain = B_TRUE;
122012282STom.Erickson@Sun.COM 				continue;
122112282STom.Erickson@Sun.COM 			}
12225367Sahrens 		}
12235367Sahrens 
12245367Sahrens 		zhp = zfs_open(rzhp->zfs_hdl, fsname, ZFS_TYPE_DATASET);
12256476Sahl 		if (zhp == NULL)
12266476Sahl 			return (-1);
12275367Sahrens 		err = dump_filesystem(zhp, sdd);
12285367Sahrens 		VERIFY(nvlist_add_boolean(fslist, "sent") == 0);
12295367Sahrens 		progress = B_TRUE;
12305367Sahrens 		zfs_close(zhp);
12315367Sahrens 		if (err)
12325367Sahrens 			return (err);
12335367Sahrens 	}
12345367Sahrens 	if (needagain) {
12355367Sahrens 		assert(progress);
12365367Sahrens 		goto again;
12375367Sahrens 	}
12385367Sahrens 	return (0);
12395367Sahrens }
12405367Sahrens 
12415367Sahrens /*
124210200SLori.Alt@Sun.COM  * Generate a send stream for the dataset identified by the argument zhp.
124310200SLori.Alt@Sun.COM  *
124410200SLori.Alt@Sun.COM  * The content of the send stream is the snapshot identified by
124510200SLori.Alt@Sun.COM  * 'tosnap'.  Incremental streams are requested in two ways:
124610200SLori.Alt@Sun.COM  *     - from the snapshot identified by "fromsnap" (if non-null) or
124710200SLori.Alt@Sun.COM  *     - from the origin of the dataset identified by zhp, which must
124810200SLori.Alt@Sun.COM  *	 be a clone.  In this case, "fromsnap" is null and "fromorigin"
124910200SLori.Alt@Sun.COM  *	 is TRUE.
125010200SLori.Alt@Sun.COM  *
125110200SLori.Alt@Sun.COM  * The send stream is recursive (i.e. dumps a hierarchy of snapshots) and
125211007SLori.Alt@Sun.COM  * uses a special header (with a hdrtype field of DMU_COMPOUNDSTREAM)
125310200SLori.Alt@Sun.COM  * if "replicate" is set.  If "doall" is set, dump all the intermediate
125411007SLori.Alt@Sun.COM  * snapshots. The DMU_COMPOUNDSTREAM header is used in the "doall"
125511022STom.Erickson@Sun.COM  * case too. If "props" is set, send properties.
12565367Sahrens  */
12575367Sahrens int
zfs_send(zfs_handle_t * zhp,const char * fromsnap,const char * tosnap,sendflags_t flags,int outfd,snapfilter_cb_t filter_func,void * cb_arg,nvlist_t ** debugnvp)12585367Sahrens zfs_send(zfs_handle_t *zhp, const char *fromsnap, const char *tosnap,
125911007SLori.Alt@Sun.COM     sendflags_t flags, int outfd, snapfilter_cb_t filter_func,
126012296SLin.Ling@Sun.COM     void *cb_arg, nvlist_t **debugnvp)
12615367Sahrens {
12625367Sahrens 	char errbuf[1024];
12635367Sahrens 	send_dump_data_t sdd = { 0 };
12645367Sahrens 	int err;
12655367Sahrens 	nvlist_t *fss = NULL;
12665367Sahrens 	avl_tree_t *fsavl = NULL;
126710342Schris.kirby@sun.com 	static uint64_t holdseq;
126810735Schris.kirby@sun.com 	int spa_version;
126910735Schris.kirby@sun.com 	boolean_t holdsnaps = B_FALSE;
127011007SLori.Alt@Sun.COM 	pthread_t tid;
127111007SLori.Alt@Sun.COM 	int pipefd[2];
127211007SLori.Alt@Sun.COM 	dedup_arg_t dda = { 0 };
127311007SLori.Alt@Sun.COM 	int featureflags = 0;
127411935SMark.Shellenbaum@Sun.COM 
12755367Sahrens 	(void) snprintf(errbuf, sizeof (errbuf), dgettext(TEXT_DOMAIN,
12765367Sahrens 	    "cannot send '%s'"), zhp->zfs_name);
12775367Sahrens 
12785367Sahrens 	if (fromsnap && fromsnap[0] == '\0') {
12795367Sahrens 		zfs_error_aux(zhp->zfs_hdl, dgettext(TEXT_DOMAIN,
12805367Sahrens 		    "zero-length incremental source"));
12815367Sahrens 		return (zfs_error(zhp->zfs_hdl, EZFS_NOENT, errbuf));
12825367Sahrens 	}
12835367Sahrens 
1284*12786SChris.Kirby@oracle.com 	if (zhp->zfs_type == ZFS_TYPE_FILESYSTEM) {
1285*12786SChris.Kirby@oracle.com 		uint64_t version;
1286*12786SChris.Kirby@oracle.com 		version = zfs_prop_get_int(zhp, ZFS_PROP_VERSION);
1287*12786SChris.Kirby@oracle.com 		if (version >= ZPL_VERSION_SA) {
1288*12786SChris.Kirby@oracle.com 			featureflags |= DMU_BACKUP_FEATURE_SA_SPILL;
1289*12786SChris.Kirby@oracle.com 		}
1290*12786SChris.Kirby@oracle.com 	}
1291*12786SChris.Kirby@oracle.com 
129210735Schris.kirby@sun.com 	if (zfs_spa_version(zhp, &spa_version) == 0 &&
1293*12786SChris.Kirby@oracle.com 	    spa_version >= SPA_VERSION_USERREFS &&
1294*12786SChris.Kirby@oracle.com 	    (flags.doall || flags.replicate))
129510735Schris.kirby@sun.com 		holdsnaps = B_TRUE;
129610735Schris.kirby@sun.com 
129711007SLori.Alt@Sun.COM 	if (flags.dedup) {
129811381SLori.Alt@Sun.COM 		featureflags |= (DMU_BACKUP_FEATURE_DEDUP |
129911381SLori.Alt@Sun.COM 		    DMU_BACKUP_FEATURE_DEDUPPROPS);
130011007SLori.Alt@Sun.COM 		if (err = pipe(pipefd)) {
130111007SLori.Alt@Sun.COM 			zfs_error_aux(zhp->zfs_hdl, strerror(errno));
130211007SLori.Alt@Sun.COM 			return (zfs_error(zhp->zfs_hdl, EZFS_PIPEFAILED,
130311007SLori.Alt@Sun.COM 			    errbuf));
130411007SLori.Alt@Sun.COM 		}
130511007SLori.Alt@Sun.COM 		dda.outputfd = outfd;
130611007SLori.Alt@Sun.COM 		dda.inputfd = pipefd[1];
130711007SLori.Alt@Sun.COM 		dda.dedup_hdl = zhp->zfs_hdl;
130811007SLori.Alt@Sun.COM 		if (err = pthread_create(&tid, NULL, cksummer, &dda)) {
130911007SLori.Alt@Sun.COM 			(void) close(pipefd[0]);
131011007SLori.Alt@Sun.COM 			(void) close(pipefd[1]);
131111007SLori.Alt@Sun.COM 			zfs_error_aux(zhp->zfs_hdl, strerror(errno));
131211007SLori.Alt@Sun.COM 			return (zfs_error(zhp->zfs_hdl,
131311007SLori.Alt@Sun.COM 			    EZFS_THREADCREATEFAILED, errbuf));
131411007SLori.Alt@Sun.COM 		}
131511007SLori.Alt@Sun.COM 	}
131611007SLori.Alt@Sun.COM 
131711022STom.Erickson@Sun.COM 	if (flags.replicate || flags.doall || flags.props) {
13185367Sahrens 		dmu_replay_record_t drr = { 0 };
13195367Sahrens 		char *packbuf = NULL;
13205367Sahrens 		size_t buflen = 0;
13215367Sahrens 		zio_cksum_t zc = { 0 };
13225367Sahrens 
132311022STom.Erickson@Sun.COM 		if (flags.replicate || flags.props) {
13245367Sahrens 			nvlist_t *hdrnv;
13255367Sahrens 
13265367Sahrens 			VERIFY(0 == nvlist_alloc(&hdrnv, NV_UNIQUE_NAME, 0));
13275367Sahrens 			if (fromsnap) {
13285367Sahrens 				VERIFY(0 == nvlist_add_string(hdrnv,
13295367Sahrens 				    "fromsnap", fromsnap));
13305367Sahrens 			}
13315367Sahrens 			VERIFY(0 == nvlist_add_string(hdrnv, "tosnap", tosnap));
133211022STom.Erickson@Sun.COM 			if (!flags.replicate) {
133311022STom.Erickson@Sun.COM 				VERIFY(0 == nvlist_add_boolean(hdrnv,
133411022STom.Erickson@Sun.COM 				    "not_recursive"));
133511022STom.Erickson@Sun.COM 			}
13365367Sahrens 
13375367Sahrens 			err = gather_nvlist(zhp->zfs_hdl, zhp->zfs_name,
133811022STom.Erickson@Sun.COM 			    fromsnap, tosnap, flags.replicate, &fss, &fsavl);
133912527SChris.Kirby@oracle.com 			if (err)
134011007SLori.Alt@Sun.COM 				goto err_out;
13415367Sahrens 			VERIFY(0 == nvlist_add_nvlist(hdrnv, "fss", fss));
13425367Sahrens 			err = nvlist_pack(hdrnv, &packbuf, &buflen,
13435367Sahrens 			    NV_ENCODE_XDR, 0);
134412296SLin.Ling@Sun.COM 			if (debugnvp)
134512296SLin.Ling@Sun.COM 				*debugnvp = hdrnv;
134612296SLin.Ling@Sun.COM 			else
134712296SLin.Ling@Sun.COM 				nvlist_free(hdrnv);
13485367Sahrens 			if (err) {
13495367Sahrens 				fsavl_destroy(fsavl);
13505367Sahrens 				nvlist_free(fss);
135111007SLori.Alt@Sun.COM 				goto stderr_out;
13525367Sahrens 			}
13535367Sahrens 		}
13545367Sahrens 
13555367Sahrens 		/* write first begin record */
13565367Sahrens 		drr.drr_type = DRR_BEGIN;
13575367Sahrens 		drr.drr_u.drr_begin.drr_magic = DMU_BACKUP_MAGIC;
135811007SLori.Alt@Sun.COM 		DMU_SET_STREAM_HDRTYPE(drr.drr_u.drr_begin.drr_versioninfo,
135911007SLori.Alt@Sun.COM 		    DMU_COMPOUNDSTREAM);
136011007SLori.Alt@Sun.COM 		DMU_SET_FEATUREFLAGS(drr.drr_u.drr_begin.drr_versioninfo,
136111007SLori.Alt@Sun.COM 		    featureflags);
13625367Sahrens 		(void) snprintf(drr.drr_u.drr_begin.drr_toname,
13635367Sahrens 		    sizeof (drr.drr_u.drr_begin.drr_toname),
13645367Sahrens 		    "%s@%s", zhp->zfs_name, tosnap);
13655367Sahrens 		drr.drr_payloadlen = buflen;
136611007SLori.Alt@Sun.COM 		err = cksum_and_write(&drr, sizeof (drr), &zc, outfd);
13675367Sahrens 
13685367Sahrens 		/* write header nvlist */
136911022STom.Erickson@Sun.COM 		if (err != -1 && packbuf != NULL) {
137011007SLori.Alt@Sun.COM 			err = cksum_and_write(packbuf, buflen, &zc, outfd);
13715367Sahrens 		}
13725367Sahrens 		free(packbuf);
13735367Sahrens 		if (err == -1) {
13745367Sahrens 			fsavl_destroy(fsavl);
13755367Sahrens 			nvlist_free(fss);
137611007SLori.Alt@Sun.COM 			err = errno;
137711007SLori.Alt@Sun.COM 			goto stderr_out;
13785367Sahrens 		}
13795367Sahrens 
13805367Sahrens 		/* write end record */
13815367Sahrens 		if (err != -1) {
13825367Sahrens 			bzero(&drr, sizeof (drr));
13835367Sahrens 			drr.drr_type = DRR_END;
13845367Sahrens 			drr.drr_u.drr_end.drr_checksum = zc;
13855367Sahrens 			err = write(outfd, &drr, sizeof (drr));
13865367Sahrens 			if (err == -1) {
13875367Sahrens 				fsavl_destroy(fsavl);
13885367Sahrens 				nvlist_free(fss);
138911668SChris.Kirby@sun.com 				err = errno;
139011007SLori.Alt@Sun.COM 				goto stderr_out;
13915367Sahrens 			}
13925367Sahrens 		}
13935367Sahrens 	}
13945367Sahrens 
13955367Sahrens 	/* dump each stream */
13965367Sahrens 	sdd.fromsnap = fromsnap;
13975367Sahrens 	sdd.tosnap = tosnap;
139811007SLori.Alt@Sun.COM 	if (flags.dedup)
139911007SLori.Alt@Sun.COM 		sdd.outfd = pipefd[0];
140011007SLori.Alt@Sun.COM 	else
140111007SLori.Alt@Sun.COM 		sdd.outfd = outfd;
140211007SLori.Alt@Sun.COM 	sdd.replicate = flags.replicate;
140311007SLori.Alt@Sun.COM 	sdd.doall = flags.doall;
140411007SLori.Alt@Sun.COM 	sdd.fromorigin = flags.fromorigin;
14055367Sahrens 	sdd.fss = fss;
14065367Sahrens 	sdd.fsavl = fsavl;
140711007SLori.Alt@Sun.COM 	sdd.verbose = flags.verbose;
140811007SLori.Alt@Sun.COM 	sdd.filter_cb = filter_func;
140911007SLori.Alt@Sun.COM 	sdd.filter_cb_arg = cb_arg;
141012296SLin.Ling@Sun.COM 	if (debugnvp)
141112296SLin.Ling@Sun.COM 		sdd.debugnv = *debugnvp;
1412*12786SChris.Kirby@oracle.com 	if (holdsnaps) {
1413*12786SChris.Kirby@oracle.com 		++holdseq;
1414*12786SChris.Kirby@oracle.com 		(void) snprintf(sdd.holdtag, sizeof (sdd.holdtag),
1415*12786SChris.Kirby@oracle.com 		    ".send-%d-%llu", getpid(), (u_longlong_t)holdseq);
1416*12786SChris.Kirby@oracle.com 		sdd.cleanup_fd = open(ZFS_DEV, O_RDWR|O_EXCL);
1417*12786SChris.Kirby@oracle.com 		if (sdd.cleanup_fd < 0) {
1418*12786SChris.Kirby@oracle.com 			err = errno;
1419*12786SChris.Kirby@oracle.com 			goto stderr_out;
1420*12786SChris.Kirby@oracle.com 		}
1421*12786SChris.Kirby@oracle.com 	} else {
1422*12786SChris.Kirby@oracle.com 		sdd.cleanup_fd = -1;
1423*12786SChris.Kirby@oracle.com 	}
14245367Sahrens 	err = dump_filesystems(zhp, &sdd);
14255367Sahrens 	fsavl_destroy(fsavl);
14265367Sahrens 	nvlist_free(fss);
14275367Sahrens 
142811007SLori.Alt@Sun.COM 	if (flags.dedup) {
142911007SLori.Alt@Sun.COM 		(void) close(pipefd[0]);
143011007SLori.Alt@Sun.COM 		(void) pthread_join(tid, NULL);
143111007SLori.Alt@Sun.COM 	}
143211022STom.Erickson@Sun.COM 
1433*12786SChris.Kirby@oracle.com 	if (sdd.cleanup_fd != -1) {
1434*12786SChris.Kirby@oracle.com 		VERIFY(0 == close(sdd.cleanup_fd));
1435*12786SChris.Kirby@oracle.com 		sdd.cleanup_fd = -1;
143612527SChris.Kirby@oracle.com 	}
143712527SChris.Kirby@oracle.com 
143811022STom.Erickson@Sun.COM 	if (flags.replicate || flags.doall || flags.props) {
14395367Sahrens 		/*
14405367Sahrens 		 * write final end record.  NB: want to do this even if
14415367Sahrens 		 * there was some error, because it might not be totally
14425367Sahrens 		 * failed.
14435367Sahrens 		 */
14445367Sahrens 		dmu_replay_record_t drr = { 0 };
14455367Sahrens 		drr.drr_type = DRR_END;
14465367Sahrens 		if (write(outfd, &drr, sizeof (drr)) == -1) {
14475367Sahrens 			return (zfs_standard_error(zhp->zfs_hdl,
14485367Sahrens 			    errno, errbuf));
14495367Sahrens 		}
14505367Sahrens 	}
14515367Sahrens 
14525367Sahrens 	return (err || sdd.err);
145311007SLori.Alt@Sun.COM 
145411007SLori.Alt@Sun.COM stderr_out:
145511007SLori.Alt@Sun.COM 	err = zfs_standard_error(zhp->zfs_hdl, err, errbuf);
145611007SLori.Alt@Sun.COM err_out:
1457*12786SChris.Kirby@oracle.com 	if (sdd.cleanup_fd != -1)
1458*12786SChris.Kirby@oracle.com 		VERIFY(0 == close(sdd.cleanup_fd));
145911007SLori.Alt@Sun.COM 	if (flags.dedup) {
146011007SLori.Alt@Sun.COM 		(void) pthread_cancel(tid);
146111007SLori.Alt@Sun.COM 		(void) pthread_join(tid, NULL);
146211007SLori.Alt@Sun.COM 		(void) close(pipefd[0]);
146311007SLori.Alt@Sun.COM 	}
146411007SLori.Alt@Sun.COM 	return (err);
14655367Sahrens }
14665367Sahrens 
14675367Sahrens /*
14685367Sahrens  * Routines specific to "zfs recv"
14695367Sahrens  */
14705367Sahrens 
14715367Sahrens static int
recv_read(libzfs_handle_t * hdl,int fd,void * buf,int ilen,boolean_t byteswap,zio_cksum_t * zc)14725367Sahrens recv_read(libzfs_handle_t *hdl, int fd, void *buf, int ilen,
14735367Sahrens     boolean_t byteswap, zio_cksum_t *zc)
14745367Sahrens {
14755367Sahrens 	char *cp = buf;
14765367Sahrens 	int rv;
14775367Sahrens 	int len = ilen;
14785367Sahrens 
14795367Sahrens 	do {
14805367Sahrens 		rv = read(fd, cp, len);
14815367Sahrens 		cp += rv;
14825367Sahrens 		len -= rv;
14835367Sahrens 	} while (rv > 0);
14845367Sahrens 
14855367Sahrens 	if (rv < 0 || len != 0) {
14865367Sahrens 		zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
14875367Sahrens 		    "failed to read from stream"));
14885367Sahrens 		return (zfs_error(hdl, EZFS_BADSTREAM, dgettext(TEXT_DOMAIN,
14895367Sahrens 		    "cannot receive")));
14905367Sahrens 	}
14915367Sahrens 
14925367Sahrens 	if (zc) {
14935367Sahrens 		if (byteswap)
14945367Sahrens 			fletcher_4_incremental_byteswap(buf, ilen, zc);
14955367Sahrens 		else
14965367Sahrens 			fletcher_4_incremental_native(buf, ilen, zc);
14975367Sahrens 	}
14985367Sahrens 	return (0);
14995367Sahrens }
15005367Sahrens 
15015367Sahrens static int
recv_read_nvlist(libzfs_handle_t * hdl,int fd,int len,nvlist_t ** nvp,boolean_t byteswap,zio_cksum_t * zc)15025367Sahrens recv_read_nvlist(libzfs_handle_t *hdl, int fd, int len, nvlist_t **nvp,
15035367Sahrens     boolean_t byteswap, zio_cksum_t *zc)
15045367Sahrens {
15055367Sahrens 	char *buf;
15065367Sahrens 	int err;
15075367Sahrens 
15085367Sahrens 	buf = zfs_alloc(hdl, len);
15095367Sahrens 	if (buf == NULL)
15105367Sahrens 		return (ENOMEM);
15115367Sahrens 
15125367Sahrens 	err = recv_read(hdl, fd, buf, len, byteswap, zc);
15135367Sahrens 	if (err != 0) {
15145367Sahrens 		free(buf);
15155367Sahrens 		return (err);
15165367Sahrens 	}
15175367Sahrens 
15185367Sahrens 	err = nvlist_unpack(buf, len, nvp, 0);
15195367Sahrens 	free(buf);
15205367Sahrens 	if (err != 0) {
15215367Sahrens 		zfs_error_aux(hdl, dgettext(TEXT_DOMAIN, "invalid "
15225367Sahrens 		    "stream (malformed nvlist)"));
15235367Sahrens 		return (EINVAL);
15245367Sahrens 	}
15255367Sahrens 	return (0);
15265367Sahrens }
15275367Sahrens 
15285367Sahrens static int
recv_rename(libzfs_handle_t * hdl,const char * name,const char * tryname,int baselen,char * newname,recvflags_t flags)15295367Sahrens recv_rename(libzfs_handle_t *hdl, const char *name, const char *tryname,
15305367Sahrens     int baselen, char *newname, recvflags_t flags)
15315367Sahrens {
15325367Sahrens 	static int seq;
15335367Sahrens 	zfs_cmd_t zc = { 0 };
15345367Sahrens 	int err;
15356047Sahrens 	prop_changelist_t *clp;
15366047Sahrens 	zfs_handle_t *zhp;
15375367Sahrens 
15386047Sahrens 	zhp = zfs_open(hdl, name, ZFS_TYPE_DATASET);
15396047Sahrens 	if (zhp == NULL)
15406047Sahrens 		return (-1);
15417366STim.Haley@Sun.COM 	clp = changelist_gather(zhp, ZFS_PROP_NAME, 0,
15427366STim.Haley@Sun.COM 	    flags.force ? MS_FORCE : 0);
15436047Sahrens 	zfs_close(zhp);
15446047Sahrens 	if (clp == NULL)
15456047Sahrens 		return (-1);
15466047Sahrens 	err = changelist_prefix(clp);
15476047Sahrens 	if (err)
15486047Sahrens 		return (err);
15495367Sahrens 
155010242Schris.kirby@sun.com 	zc.zc_objset_type = DMU_OST_ZFS;
155110242Schris.kirby@sun.com 	(void) strlcpy(zc.zc_name, name, sizeof (zc.zc_name));
155210242Schris.kirby@sun.com 
15535367Sahrens 	if (tryname) {
15545367Sahrens 		(void) strcpy(newname, tryname);
15555367Sahrens 
15565367Sahrens 		(void) strlcpy(zc.zc_value, tryname, sizeof (zc.zc_value));
15575367Sahrens 
15585367Sahrens 		if (flags.verbose) {
15595367Sahrens 			(void) printf("attempting rename %s to %s\n",
15605367Sahrens 			    zc.zc_name, zc.zc_value);
15615367Sahrens 		}
15625367Sahrens 		err = ioctl(hdl->libzfs_fd, ZFS_IOC_RENAME, &zc);
15636047Sahrens 		if (err == 0)
15645367Sahrens 			changelist_rename(clp, name, tryname);
15655367Sahrens 	} else {
15665367Sahrens 		err = ENOENT;
15675367Sahrens 	}
15685367Sahrens 
15695367Sahrens 	if (err != 0 && strncmp(name+baselen, "recv-", 5) != 0) {
15705367Sahrens 		seq++;
15715367Sahrens 
15725367Sahrens 		(void) strncpy(newname, name, baselen);
15735367Sahrens 		(void) snprintf(newname+baselen, ZFS_MAXNAMELEN-baselen,
15745367Sahrens 		    "recv-%u-%u", getpid(), seq);
15755367Sahrens 		(void) strlcpy(zc.zc_value, newname, sizeof (zc.zc_value));
15765367Sahrens 
15775367Sahrens 		if (flags.verbose) {
15785367Sahrens 			(void) printf("failed - trying rename %s to %s\n",
15795367Sahrens 			    zc.zc_name, zc.zc_value);
15805367Sahrens 		}
15815367Sahrens 		err = ioctl(hdl->libzfs_fd, ZFS_IOC_RENAME, &zc);
15826047Sahrens 		if (err == 0)
15835367Sahrens 			changelist_rename(clp, name, newname);
15845367Sahrens 		if (err && flags.verbose) {
15855367Sahrens 			(void) printf("failed (%u) - "
15865367Sahrens 			    "will try again on next pass\n", errno);
15875367Sahrens 		}
15885367Sahrens 		err = EAGAIN;
15895367Sahrens 	} else if (flags.verbose) {
15905367Sahrens 		if (err == 0)
15915367Sahrens 			(void) printf("success\n");
15925367Sahrens 		else
15935367Sahrens 			(void) printf("failed (%u)\n", errno);
15945367Sahrens 	}
15955367Sahrens 
15966047Sahrens 	(void) changelist_postfix(clp);
15976047Sahrens 	changelist_free(clp);
15985367Sahrens 
15995367Sahrens 	return (err);
16005367Sahrens }
16015367Sahrens 
16025367Sahrens static int
recv_destroy(libzfs_handle_t * hdl,const char * name,int baselen,char * newname,recvflags_t flags)16035367Sahrens recv_destroy(libzfs_handle_t *hdl, const char *name, int baselen,
16045367Sahrens     char *newname, recvflags_t flags)
16055367Sahrens {
16065367Sahrens 	zfs_cmd_t zc = { 0 };
16076047Sahrens 	int err = 0;
16086047Sahrens 	prop_changelist_t *clp;
16096047Sahrens 	zfs_handle_t *zhp;
161010242Schris.kirby@sun.com 	boolean_t defer = B_FALSE;
161110242Schris.kirby@sun.com 	int spa_version;
16126047Sahrens 
16136047Sahrens 	zhp = zfs_open(hdl, name, ZFS_TYPE_DATASET);
16146047Sahrens 	if (zhp == NULL)
16156047Sahrens 		return (-1);
16167366STim.Haley@Sun.COM 	clp = changelist_gather(zhp, ZFS_PROP_NAME, 0,
16177366STim.Haley@Sun.COM 	    flags.force ? MS_FORCE : 0);
161810242Schris.kirby@sun.com 	if (zfs_get_type(zhp) == ZFS_TYPE_SNAPSHOT &&
161910242Schris.kirby@sun.com 	    zfs_spa_version(zhp, &spa_version) == 0 &&
162010242Schris.kirby@sun.com 	    spa_version >= SPA_VERSION_USERREFS)
162110242Schris.kirby@sun.com 		defer = B_TRUE;
16226047Sahrens 	zfs_close(zhp);
16236047Sahrens 	if (clp == NULL)
16246047Sahrens 		return (-1);
16256047Sahrens 	err = changelist_prefix(clp);
16266047Sahrens 	if (err)
16276047Sahrens 		return (err);
16285367Sahrens 
16295367Sahrens 	zc.zc_objset_type = DMU_OST_ZFS;
163010242Schris.kirby@sun.com 	zc.zc_defer_destroy = defer;
16315367Sahrens 	(void) strlcpy(zc.zc_name, name, sizeof (zc.zc_name));
16325367Sahrens 
16335367Sahrens 	if (flags.verbose)
16345367Sahrens 		(void) printf("attempting destroy %s\n", zc.zc_name);
16355367Sahrens 	err = ioctl(hdl->libzfs_fd, ZFS_IOC_DESTROY, &zc);
16366047Sahrens 	if (err == 0) {
16376047Sahrens 		if (flags.verbose)
16386047Sahrens 			(void) printf("success\n");
16396047Sahrens 		changelist_remove(clp, zc.zc_name);
16405367Sahrens 	}
16415367Sahrens 
16426047Sahrens 	(void) changelist_postfix(clp);
16436047Sahrens 	changelist_free(clp);
16446047Sahrens 
164510242Schris.kirby@sun.com 	/*
164611022STom.Erickson@Sun.COM 	 * Deferred destroy might destroy the snapshot or only mark it to be
164711022STom.Erickson@Sun.COM 	 * destroyed later, and it returns success in either case.
164810242Schris.kirby@sun.com 	 */
164911022STom.Erickson@Sun.COM 	if (err != 0 || (defer && zfs_dataset_exists(hdl, name,
165011022STom.Erickson@Sun.COM 	    ZFS_TYPE_SNAPSHOT))) {
16516047Sahrens 		err = recv_rename(hdl, name, NULL, baselen, newname, flags);
165211022STom.Erickson@Sun.COM 	}
16535367Sahrens 
16545367Sahrens 	return (err);
16555367Sahrens }
16565367Sahrens 
16575367Sahrens typedef struct guid_to_name_data {
16585367Sahrens 	uint64_t guid;
16595367Sahrens 	char *name;
16605367Sahrens } guid_to_name_data_t;
16615367Sahrens 
16625367Sahrens static int
guid_to_name_cb(zfs_handle_t * zhp,void * arg)16635367Sahrens guid_to_name_cb(zfs_handle_t *zhp, void *arg)
16645367Sahrens {
16655367Sahrens 	guid_to_name_data_t *gtnd = arg;
16665367Sahrens 	int err;
16675367Sahrens 
16685367Sahrens 	if (zhp->zfs_dmustats.dds_guid == gtnd->guid) {
16695367Sahrens 		(void) strcpy(gtnd->name, zhp->zfs_name);
167011515STom.Erickson@Sun.COM 		zfs_close(zhp);
16715367Sahrens 		return (EEXIST);
16725367Sahrens 	}
16735367Sahrens 	err = zfs_iter_children(zhp, guid_to_name_cb, gtnd);
16745367Sahrens 	zfs_close(zhp);
16755367Sahrens 	return (err);
16765367Sahrens }
16775367Sahrens 
16785367Sahrens static int
guid_to_name(libzfs_handle_t * hdl,const char * parent,uint64_t guid,char * name)16795367Sahrens guid_to_name(libzfs_handle_t *hdl, const char *parent, uint64_t guid,
16805367Sahrens     char *name)
16815367Sahrens {
16825367Sahrens 	/* exhaustive search all local snapshots */
16835367Sahrens 	guid_to_name_data_t gtnd;
16845367Sahrens 	int err = 0;
16855367Sahrens 	zfs_handle_t *zhp;
16865367Sahrens 	char *cp;
16875367Sahrens 
16885367Sahrens 	gtnd.guid = guid;
16895367Sahrens 	gtnd.name = name;
16905367Sahrens 
16915367Sahrens 	if (strchr(parent, '@') == NULL) {
16925367Sahrens 		zhp = make_dataset_handle(hdl, parent);
16935367Sahrens 		if (zhp != NULL) {
16945367Sahrens 			err = zfs_iter_children(zhp, guid_to_name_cb, &gtnd);
16955367Sahrens 			zfs_close(zhp);
16965367Sahrens 			if (err == EEXIST)
16975367Sahrens 				return (0);
16985367Sahrens 		}
16995367Sahrens 	}
17005367Sahrens 
17015367Sahrens 	cp = strchr(parent, '/');
17025367Sahrens 	if (cp)
17035367Sahrens 		*cp = '\0';
17045367Sahrens 	zhp = make_dataset_handle(hdl, parent);
17055367Sahrens 	if (cp)
17065367Sahrens 		*cp = '/';
17075367Sahrens 
17085367Sahrens 	if (zhp) {
17095367Sahrens 		err = zfs_iter_children(zhp, guid_to_name_cb, &gtnd);
17105367Sahrens 		zfs_close(zhp);
17115367Sahrens 	}
17125367Sahrens 
17135367Sahrens 	return (err == EEXIST ? 0 : ENOENT);
17145367Sahrens 
17155367Sahrens }
17165367Sahrens 
17175367Sahrens /*
17185367Sahrens  * Return true if dataset guid1 is created before guid2.
17195367Sahrens  */
17206476Sahl static int
created_before(libzfs_handle_t * hdl,avl_tree_t * avl,uint64_t guid1,uint64_t guid2)17215367Sahrens created_before(libzfs_handle_t *hdl, avl_tree_t *avl,
17225367Sahrens     uint64_t guid1, uint64_t guid2)
17235367Sahrens {
17245367Sahrens 	nvlist_t *nvfs;
17255367Sahrens 	char *fsname, *snapname;
17265367Sahrens 	char buf[ZFS_MAXNAMELEN];
17276476Sahl 	int rv;
17285367Sahrens 	zfs_node_t zn1, zn2;
17295367Sahrens 
17305367Sahrens 	if (guid2 == 0)
17316476Sahl 		return (0);
17325367Sahrens 	if (guid1 == 0)
17336476Sahl 		return (1);
17345367Sahrens 
17355367Sahrens 	nvfs = fsavl_find(avl, guid1, &snapname);
17365367Sahrens 	VERIFY(0 == nvlist_lookup_string(nvfs, "name", &fsname));
17375367Sahrens 	(void) snprintf(buf, sizeof (buf), "%s@%s", fsname, snapname);
17385367Sahrens 	zn1.zn_handle = zfs_open(hdl, buf, ZFS_TYPE_SNAPSHOT);
17396476Sahl 	if (zn1.zn_handle == NULL)
17406476Sahl 		return (-1);
17415367Sahrens 
17425367Sahrens 	nvfs = fsavl_find(avl, guid2, &snapname);
17435367Sahrens 	VERIFY(0 == nvlist_lookup_string(nvfs, "name", &fsname));
17445367Sahrens 	(void) snprintf(buf, sizeof (buf), "%s@%s", fsname, snapname);
17455367Sahrens 	zn2.zn_handle = zfs_open(hdl, buf, ZFS_TYPE_SNAPSHOT);
17466476Sahl 	if (zn2.zn_handle == NULL) {
17476476Sahl 		zfs_close(zn2.zn_handle);
17486476Sahl 		return (-1);
17496476Sahl 	}
17505367Sahrens 
17515367Sahrens 	rv = (zfs_snapshot_compare(&zn1, &zn2) == -1);
17525367Sahrens 
17535367Sahrens 	zfs_close(zn1.zn_handle);
17545367Sahrens 	zfs_close(zn2.zn_handle);
17555367Sahrens 
17565367Sahrens 	return (rv);
17575367Sahrens }
17585367Sahrens 
17595367Sahrens static int
recv_incremental_replication(libzfs_handle_t * hdl,const char * tofs,recvflags_t flags,nvlist_t * stream_nv,avl_tree_t * stream_avl,nvlist_t * renamed)17605367Sahrens recv_incremental_replication(libzfs_handle_t *hdl, const char *tofs,
176111834STom.Erickson@Sun.COM     recvflags_t flags, nvlist_t *stream_nv, avl_tree_t *stream_avl,
176211834STom.Erickson@Sun.COM     nvlist_t *renamed)
17635367Sahrens {
17645367Sahrens 	nvlist_t *local_nv;
17655367Sahrens 	avl_tree_t *local_avl;
17665367Sahrens 	nvpair_t *fselem, *nextfselem;
176711834STom.Erickson@Sun.COM 	char *fromsnap;
17685367Sahrens 	char newname[ZFS_MAXNAMELEN];
17695367Sahrens 	int error;
177011022STom.Erickson@Sun.COM 	boolean_t needagain, progress, recursive;
177110299SPrabahar.Jeyaram@Sun.COM 	char *s1, *s2;
17725367Sahrens 
17735367Sahrens 	VERIFY(0 == nvlist_lookup_string(stream_nv, "fromsnap", &fromsnap));
17745367Sahrens 
177511022STom.Erickson@Sun.COM 	recursive = (nvlist_lookup_boolean(stream_nv, "not_recursive") ==
177611022STom.Erickson@Sun.COM 	    ENOENT);
177711022STom.Erickson@Sun.COM 
17785367Sahrens 	if (flags.dryrun)
17795367Sahrens 		return (0);
17805367Sahrens 
17815367Sahrens again:
17825367Sahrens 	needagain = progress = B_FALSE;
17835367Sahrens 
17845367Sahrens 	if ((error = gather_nvlist(hdl, tofs, fromsnap, NULL,
178511022STom.Erickson@Sun.COM 	    recursive, &local_nv, &local_avl)) != 0)
17865367Sahrens 		return (error);
17875367Sahrens 
17885367Sahrens 	/*
17895367Sahrens 	 * Process deletes and renames
17905367Sahrens 	 */
17915367Sahrens 	for (fselem = nvlist_next_nvpair(local_nv, NULL);
17925367Sahrens 	    fselem; fselem = nextfselem) {
17935367Sahrens 		nvlist_t *nvfs, *snaps;
17945367Sahrens 		nvlist_t *stream_nvfs = NULL;
17955367Sahrens 		nvpair_t *snapelem, *nextsnapelem;
17965367Sahrens 		uint64_t fromguid = 0;
17975367Sahrens 		uint64_t originguid = 0;
17985367Sahrens 		uint64_t stream_originguid = 0;
17995367Sahrens 		uint64_t parent_fromsnap_guid, stream_parent_fromsnap_guid;
18005367Sahrens 		char *fsname, *stream_fsname;
18015367Sahrens 
18025367Sahrens 		nextfselem = nvlist_next_nvpair(local_nv, fselem);
18035367Sahrens 
18045367Sahrens 		VERIFY(0 == nvpair_value_nvlist(fselem, &nvfs));
18055367Sahrens 		VERIFY(0 == nvlist_lookup_nvlist(nvfs, "snaps", &snaps));
18065367Sahrens 		VERIFY(0 == nvlist_lookup_string(nvfs, "name", &fsname));
18075367Sahrens 		VERIFY(0 == nvlist_lookup_uint64(nvfs, "parentfromsnap",
18085367Sahrens 		    &parent_fromsnap_guid));
18095367Sahrens 		(void) nvlist_lookup_uint64(nvfs, "origin", &originguid);
18105367Sahrens 
18115367Sahrens 		/*
18125367Sahrens 		 * First find the stream's fs, so we can check for
18135367Sahrens 		 * a different origin (due to "zfs promote")
18145367Sahrens 		 */
18155367Sahrens 		for (snapelem = nvlist_next_nvpair(snaps, NULL);
18165367Sahrens 		    snapelem; snapelem = nvlist_next_nvpair(snaps, snapelem)) {
18175367Sahrens 			uint64_t thisguid;
18185367Sahrens 
18195367Sahrens 			VERIFY(0 == nvpair_value_uint64(snapelem, &thisguid));
18205367Sahrens 			stream_nvfs = fsavl_find(stream_avl, thisguid, NULL);
18215367Sahrens 
18225367Sahrens 			if (stream_nvfs != NULL)
18235367Sahrens 				break;
18245367Sahrens 		}
18255367Sahrens 
18265367Sahrens 		/* check for promote */
18275367Sahrens 		(void) nvlist_lookup_uint64(stream_nvfs, "origin",
18285367Sahrens 		    &stream_originguid);
18295367Sahrens 		if (stream_nvfs && originguid != stream_originguid) {
18306476Sahl 			switch (created_before(hdl, local_avl,
18316476Sahl 			    stream_originguid, originguid)) {
18326476Sahl 			case 1: {
18335367Sahrens 				/* promote it! */
18345367Sahrens 				zfs_cmd_t zc = { 0 };
18355367Sahrens 				nvlist_t *origin_nvfs;
18365367Sahrens 				char *origin_fsname;
18375367Sahrens 
18385367Sahrens 				if (flags.verbose)
18395367Sahrens 					(void) printf("promoting %s\n", fsname);
18405367Sahrens 
18415367Sahrens 				origin_nvfs = fsavl_find(local_avl, originguid,
18425367Sahrens 				    NULL);
18435367Sahrens 				VERIFY(0 == nvlist_lookup_string(origin_nvfs,
18445367Sahrens 				    "name", &origin_fsname));
18455367Sahrens 				(void) strlcpy(zc.zc_value, origin_fsname,
18465367Sahrens 				    sizeof (zc.zc_value));
18475367Sahrens 				(void) strlcpy(zc.zc_name, fsname,
18485367Sahrens 				    sizeof (zc.zc_name));
18495367Sahrens 				error = zfs_ioctl(hdl, ZFS_IOC_PROMOTE, &zc);
18505367Sahrens 				if (error == 0)
18515367Sahrens 					progress = B_TRUE;
18526476Sahl 				break;
18536476Sahl 			}
18546476Sahl 			default:
18556476Sahl 				break;
18566476Sahl 			case -1:
18576476Sahl 				fsavl_destroy(local_avl);
18586476Sahl 				nvlist_free(local_nv);
18596476Sahl 				return (-1);
18605367Sahrens 			}
18615367Sahrens 			/*
18625367Sahrens 			 * We had/have the wrong origin, therefore our
18635367Sahrens 			 * list of snapshots is wrong.  Need to handle
18645367Sahrens 			 * them on the next pass.
18655367Sahrens 			 */
18665367Sahrens 			needagain = B_TRUE;
18675367Sahrens 			continue;
18685367Sahrens 		}
18695367Sahrens 
18705367Sahrens 		for (snapelem = nvlist_next_nvpair(snaps, NULL);
18715367Sahrens 		    snapelem; snapelem = nextsnapelem) {
18725367Sahrens 			uint64_t thisguid;
18735367Sahrens 			char *stream_snapname;
18747265Sahrens 			nvlist_t *found, *props;
18755367Sahrens 
18765367Sahrens 			nextsnapelem = nvlist_next_nvpair(snaps, snapelem);
18775367Sahrens 
18785367Sahrens 			VERIFY(0 == nvpair_value_uint64(snapelem, &thisguid));
18795367Sahrens 			found = fsavl_find(stream_avl, thisguid,
18805367Sahrens 			    &stream_snapname);
18815367Sahrens 
18825367Sahrens 			/* check for delete */
18835367Sahrens 			if (found == NULL) {
18845367Sahrens 				char name[ZFS_MAXNAMELEN];
18855367Sahrens 
18865367Sahrens 				if (!flags.force)
18875367Sahrens 					continue;
18885367Sahrens 
18895367Sahrens 				(void) snprintf(name, sizeof (name), "%s@%s",
18905367Sahrens 				    fsname, nvpair_name(snapelem));
18915367Sahrens 
18925367Sahrens 				error = recv_destroy(hdl, name,
18935367Sahrens 				    strlen(fsname)+1, newname, flags);
18945367Sahrens 				if (error)
18955367Sahrens 					needagain = B_TRUE;
18965367Sahrens 				else
18975367Sahrens 					progress = B_TRUE;
18985367Sahrens 				continue;
18995367Sahrens 			}
19005367Sahrens 
19015367Sahrens 			stream_nvfs = found;
19025367Sahrens 
19037265Sahrens 			if (0 == nvlist_lookup_nvlist(stream_nvfs, "snapprops",
19047265Sahrens 			    &props) && 0 == nvlist_lookup_nvlist(props,
19057265Sahrens 			    stream_snapname, &props)) {
19067265Sahrens 				zfs_cmd_t zc = { 0 };
19077265Sahrens 
190811022STom.Erickson@Sun.COM 				zc.zc_cookie = B_TRUE; /* received */
19097273Sahrens 				(void) snprintf(zc.zc_name, sizeof (zc.zc_name),
19107265Sahrens 				    "%s@%s", fsname, nvpair_name(snapelem));
19117265Sahrens 				if (zcmd_write_src_nvlist(hdl, &zc,
19127265Sahrens 				    props) == 0) {
19137265Sahrens 					(void) zfs_ioctl(hdl,
19147265Sahrens 					    ZFS_IOC_SET_PROP, &zc);
19157265Sahrens 					zcmd_free_nvlists(&zc);
19167265Sahrens 				}
19177265Sahrens 			}
19187265Sahrens 
19195367Sahrens 			/* check for different snapname */
19205367Sahrens 			if (strcmp(nvpair_name(snapelem),
19215367Sahrens 			    stream_snapname) != 0) {
19225367Sahrens 				char name[ZFS_MAXNAMELEN];
19235367Sahrens 				char tryname[ZFS_MAXNAMELEN];
19245367Sahrens 
19255367Sahrens 				(void) snprintf(name, sizeof (name), "%s@%s",
19265367Sahrens 				    fsname, nvpair_name(snapelem));
19275367Sahrens 				(void) snprintf(tryname, sizeof (name), "%s@%s",
19285367Sahrens 				    fsname, stream_snapname);
19295367Sahrens 
19305367Sahrens 				error = recv_rename(hdl, name, tryname,
19315367Sahrens 				    strlen(fsname)+1, newname, flags);
19325367Sahrens 				if (error)
19335367Sahrens 					needagain = B_TRUE;
19345367Sahrens 				else
19355367Sahrens 					progress = B_TRUE;
19365367Sahrens 			}
19375367Sahrens 
19385367Sahrens 			if (strcmp(stream_snapname, fromsnap) == 0)
19395367Sahrens 				fromguid = thisguid;
19405367Sahrens 		}
19415367Sahrens 
19425367Sahrens 		/* check for delete */
19435367Sahrens 		if (stream_nvfs == NULL) {
19445367Sahrens 			if (!flags.force)
19455367Sahrens 				continue;
19465367Sahrens 
19475367Sahrens 			error = recv_destroy(hdl, fsname, strlen(tofs)+1,
19485367Sahrens 			    newname, flags);
19495367Sahrens 			if (error)
19505367Sahrens 				needagain = B_TRUE;
19515367Sahrens 			else
19525367Sahrens 				progress = B_TRUE;
19535367Sahrens 			continue;
19545367Sahrens 		}
19555367Sahrens 
195611834STom.Erickson@Sun.COM 		if (fromguid == 0) {
195711834STom.Erickson@Sun.COM 			if (flags.verbose) {
195811834STom.Erickson@Sun.COM 				(void) printf("local fs %s does not have "
195911834STom.Erickson@Sun.COM 				    "fromsnap (%s in stream); must have "
196011834STom.Erickson@Sun.COM 				    "been deleted locally; ignoring\n",
196111834STom.Erickson@Sun.COM 				    fsname, fromsnap);
196211834STom.Erickson@Sun.COM 			}
19635367Sahrens 			continue;
19645367Sahrens 		}
19655367Sahrens 
19665367Sahrens 		VERIFY(0 == nvlist_lookup_string(stream_nvfs,
19675367Sahrens 		    "name", &stream_fsname));
19685367Sahrens 		VERIFY(0 == nvlist_lookup_uint64(stream_nvfs,
19695367Sahrens 		    "parentfromsnap", &stream_parent_fromsnap_guid));
19705367Sahrens 
197110302SPrabahar.Jeyaram@Sun.COM 		s1 = strrchr(fsname, '/');
197210302SPrabahar.Jeyaram@Sun.COM 		s2 = strrchr(stream_fsname, '/');
197310302SPrabahar.Jeyaram@Sun.COM 
197411834STom.Erickson@Sun.COM 		/*
197511834STom.Erickson@Sun.COM 		 * Check for rename. If the exact receive path is specified, it
197611834STom.Erickson@Sun.COM 		 * does not count as a rename, but we still need to check the
197711834STom.Erickson@Sun.COM 		 * datasets beneath it.
197811834STom.Erickson@Sun.COM 		 */
19795367Sahrens 		if ((stream_parent_fromsnap_guid != 0 &&
198011834STom.Erickson@Sun.COM 		    parent_fromsnap_guid != 0 &&
19815367Sahrens 		    stream_parent_fromsnap_guid != parent_fromsnap_guid) ||
198211834STom.Erickson@Sun.COM 		    ((flags.isprefix || strcmp(tofs, fsname) != 0) &&
198311834STom.Erickson@Sun.COM 		    (s1 != NULL) && (s2 != NULL) && strcmp(s1, s2) != 0)) {
19845367Sahrens 			nvlist_t *parent;
19855367Sahrens 			char tryname[ZFS_MAXNAMELEN];
19865367Sahrens 
19875367Sahrens 			parent = fsavl_find(local_avl,
19885367Sahrens 			    stream_parent_fromsnap_guid, NULL);
19895367Sahrens 			/*
19905367Sahrens 			 * NB: parent might not be found if we used the
19915367Sahrens 			 * tosnap for stream_parent_fromsnap_guid,
19925367Sahrens 			 * because the parent is a newly-created fs;
19935367Sahrens 			 * we'll be able to rename it after we recv the
19945367Sahrens 			 * new fs.
19955367Sahrens 			 */
19965367Sahrens 			if (parent != NULL) {
19975367Sahrens 				char *pname;
19985367Sahrens 
19995367Sahrens 				VERIFY(0 == nvlist_lookup_string(parent, "name",
20005367Sahrens 				    &pname));
20015367Sahrens 				(void) snprintf(tryname, sizeof (tryname),
20025367Sahrens 				    "%s%s", pname, strrchr(stream_fsname, '/'));
20035367Sahrens 			} else {
20045367Sahrens 				tryname[0] = '\0';
20055367Sahrens 				if (flags.verbose) {
20065367Sahrens 					(void) printf("local fs %s new parent "
20075367Sahrens 					    "not found\n", fsname);
20085367Sahrens 				}
20095367Sahrens 			}
20105367Sahrens 
201111834STom.Erickson@Sun.COM 			newname[0] = '\0';
201211834STom.Erickson@Sun.COM 
20135367Sahrens 			error = recv_rename(hdl, fsname, tryname,
20145367Sahrens 			    strlen(tofs)+1, newname, flags);
201511834STom.Erickson@Sun.COM 
201611834STom.Erickson@Sun.COM 			if (renamed != NULL && newname[0] != '\0') {
201711834STom.Erickson@Sun.COM 				VERIFY(0 == nvlist_add_boolean(renamed,
201811834STom.Erickson@Sun.COM 				    newname));
201911834STom.Erickson@Sun.COM 			}
202011834STom.Erickson@Sun.COM 
20215367Sahrens 			if (error)
20225367Sahrens 				needagain = B_TRUE;
20235367Sahrens 			else
20245367Sahrens 				progress = B_TRUE;
20255367Sahrens 		}
20265367Sahrens 	}
20275367Sahrens 
20285367Sahrens 	fsavl_destroy(local_avl);
20295367Sahrens 	nvlist_free(local_nv);
20305367Sahrens 
20315367Sahrens 	if (needagain && progress) {
20325367Sahrens 		/* do another pass to fix up temporary names */
20335367Sahrens 		if (flags.verbose)
20345367Sahrens 			(void) printf("another pass:\n");
20355367Sahrens 		goto again;
20365367Sahrens 	}
20375367Sahrens 
20385367Sahrens 	return (needagain);
20395367Sahrens }
20405367Sahrens 
20415367Sahrens static int
zfs_receive_package(libzfs_handle_t * hdl,int fd,const char * destname,recvflags_t flags,dmu_replay_record_t * drr,zio_cksum_t * zc,char ** top_zfs,int cleanup_fd,uint64_t * action_handlep)20425367Sahrens zfs_receive_package(libzfs_handle_t *hdl, int fd, const char *destname,
20437366STim.Haley@Sun.COM     recvflags_t flags, dmu_replay_record_t *drr, zio_cksum_t *zc,
204412527SChris.Kirby@oracle.com     char **top_zfs, int cleanup_fd, uint64_t *action_handlep)
20455367Sahrens {
20465367Sahrens 	nvlist_t *stream_nv = NULL;
20475367Sahrens 	avl_tree_t *stream_avl = NULL;
20485367Sahrens 	char *fromsnap = NULL;
204911834STom.Erickson@Sun.COM 	char *cp;
20505367Sahrens 	char tofs[ZFS_MAXNAMELEN];
205111834STom.Erickson@Sun.COM 	char sendfs[ZFS_MAXNAMELEN];
20525367Sahrens 	char errbuf[1024];
20535367Sahrens 	dmu_replay_record_t drre;
20545367Sahrens 	int error;
20555367Sahrens 	boolean_t anyerr = B_FALSE;
20566479Sahrens 	boolean_t softerr = B_FALSE;
205711834STom.Erickson@Sun.COM 	boolean_t recursive;
20585367Sahrens 
20595367Sahrens 	(void) snprintf(errbuf, sizeof (errbuf), dgettext(TEXT_DOMAIN,
20605367Sahrens 	    "cannot receive"));
20615367Sahrens 
20625367Sahrens 	assert(drr->drr_type == DRR_BEGIN);
20635367Sahrens 	assert(drr->drr_u.drr_begin.drr_magic == DMU_BACKUP_MAGIC);
206411007SLori.Alt@Sun.COM 	assert(DMU_GET_STREAM_HDRTYPE(drr->drr_u.drr_begin.drr_versioninfo) ==
206511007SLori.Alt@Sun.COM 	    DMU_COMPOUNDSTREAM);
20665367Sahrens 
20675367Sahrens 	/*
20685367Sahrens 	 * Read in the nvlist from the stream.
20695367Sahrens 	 */
20705367Sahrens 	if (drr->drr_payloadlen != 0) {
20715367Sahrens 		error = recv_read_nvlist(hdl, fd, drr->drr_payloadlen,
20725367Sahrens 		    &stream_nv, flags.byteswap, zc);
20735367Sahrens 		if (error) {
20745367Sahrens 			error = zfs_error(hdl, EZFS_BADSTREAM, errbuf);
20755367Sahrens 			goto out;
20765367Sahrens 		}
20775367Sahrens 	}
20785367Sahrens 
207911834STom.Erickson@Sun.COM 	recursive = (nvlist_lookup_boolean(stream_nv, "not_recursive") ==
208011834STom.Erickson@Sun.COM 	    ENOENT);
208111834STom.Erickson@Sun.COM 
208211834STom.Erickson@Sun.COM 	if (recursive && strchr(destname, '@')) {
208311834STom.Erickson@Sun.COM 		zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
208411834STom.Erickson@Sun.COM 		    "cannot specify snapshot name for multi-snapshot stream"));
208511834STom.Erickson@Sun.COM 		error = zfs_error(hdl, EZFS_BADSTREAM, errbuf);
208611834STom.Erickson@Sun.COM 		goto out;
208711834STom.Erickson@Sun.COM 	}
208811834STom.Erickson@Sun.COM 
20895367Sahrens 	/*
20905367Sahrens 	 * Read in the end record and verify checksum.
20915367Sahrens 	 */
20925367Sahrens 	if (0 != (error = recv_read(hdl, fd, &drre, sizeof (drre),
20935367Sahrens 	    flags.byteswap, NULL)))
20945367Sahrens 		goto out;
20955367Sahrens 	if (flags.byteswap) {
20965367Sahrens 		drre.drr_type = BSWAP_32(drre.drr_type);
20975367Sahrens 		drre.drr_u.drr_end.drr_checksum.zc_word[0] =
20985367Sahrens 		    BSWAP_64(drre.drr_u.drr_end.drr_checksum.zc_word[0]);
20995367Sahrens 		drre.drr_u.drr_end.drr_checksum.zc_word[1] =
21005367Sahrens 		    BSWAP_64(drre.drr_u.drr_end.drr_checksum.zc_word[1]);
21015367Sahrens 		drre.drr_u.drr_end.drr_checksum.zc_word[2] =
21025367Sahrens 		    BSWAP_64(drre.drr_u.drr_end.drr_checksum.zc_word[2]);
21035367Sahrens 		drre.drr_u.drr_end.drr_checksum.zc_word[3] =
21045367Sahrens 		    BSWAP_64(drre.drr_u.drr_end.drr_checksum.zc_word[3]);
21055367Sahrens 	}
21065367Sahrens 	if (drre.drr_type != DRR_END) {
21075367Sahrens 		error = zfs_error(hdl, EZFS_BADSTREAM, errbuf);
21085367Sahrens 		goto out;
21095367Sahrens 	}
21105367Sahrens 	if (!ZIO_CHECKSUM_EQUAL(drre.drr_u.drr_end.drr_checksum, *zc)) {
21115367Sahrens 		zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
21125367Sahrens 		    "incorrect header checksum"));
21135367Sahrens 		error = zfs_error(hdl, EZFS_BADSTREAM, errbuf);
21145367Sahrens 		goto out;
21155367Sahrens 	}
21165367Sahrens 
21175367Sahrens 	(void) nvlist_lookup_string(stream_nv, "fromsnap", &fromsnap);
21185367Sahrens 
21195367Sahrens 	if (drr->drr_payloadlen != 0) {
21205367Sahrens 		nvlist_t *stream_fss;
21215367Sahrens 
21225367Sahrens 		VERIFY(0 == nvlist_lookup_nvlist(stream_nv, "fss",
21235367Sahrens 		    &stream_fss));
21245896Sahl 		if ((stream_avl = fsavl_create(stream_fss)) == NULL) {
21255896Sahl 			zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
21265896Sahl 			    "couldn't allocate avl tree"));
21275896Sahl 			error = zfs_error(hdl, EZFS_NOMEM, errbuf);
21285896Sahl 			goto out;
21295896Sahl 		}
21305367Sahrens 
21315367Sahrens 		if (fromsnap != NULL) {
213211834STom.Erickson@Sun.COM 			nvlist_t *renamed = NULL;
213311834STom.Erickson@Sun.COM 			nvpair_t *pair = NULL;
213411834STom.Erickson@Sun.COM 
21355367Sahrens 			(void) strlcpy(tofs, destname, ZFS_MAXNAMELEN);
21365367Sahrens 			if (flags.isprefix) {
213711834STom.Erickson@Sun.COM 				struct drr_begin *drrb = &drr->drr_u.drr_begin;
213811834STom.Erickson@Sun.COM 				int i;
213911834STom.Erickson@Sun.COM 
214011834STom.Erickson@Sun.COM 				if (flags.istail) {
214111834STom.Erickson@Sun.COM 					cp = strrchr(drrb->drr_toname, '/');
214211834STom.Erickson@Sun.COM 					if (cp == NULL) {
214311834STom.Erickson@Sun.COM 						(void) strlcat(tofs, "/",
214411834STom.Erickson@Sun.COM 						    ZFS_MAXNAMELEN);
214511834STom.Erickson@Sun.COM 						i = 0;
214611834STom.Erickson@Sun.COM 					} else {
214711834STom.Erickson@Sun.COM 						i = (cp - drrb->drr_toname);
214811834STom.Erickson@Sun.COM 					}
214911834STom.Erickson@Sun.COM 				} else {
215011834STom.Erickson@Sun.COM 					i = strcspn(drrb->drr_toname, "/@");
215111834STom.Erickson@Sun.COM 				}
21525367Sahrens 				/* zfs_receive_one() will create_parents() */
215311834STom.Erickson@Sun.COM 				(void) strlcat(tofs, &drrb->drr_toname[i],
21545367Sahrens 				    ZFS_MAXNAMELEN);
21555367Sahrens 				*strchr(tofs, '@') = '\0';
21565367Sahrens 			}
215711834STom.Erickson@Sun.COM 
215811834STom.Erickson@Sun.COM 			if (recursive && !flags.dryrun && !flags.nomount) {
215911834STom.Erickson@Sun.COM 				VERIFY(0 == nvlist_alloc(&renamed,
216011834STom.Erickson@Sun.COM 				    NV_UNIQUE_NAME, 0));
216111834STom.Erickson@Sun.COM 			}
216211834STom.Erickson@Sun.COM 
216311834STom.Erickson@Sun.COM 			softerr = recv_incremental_replication(hdl, tofs, flags,
216411834STom.Erickson@Sun.COM 			    stream_nv, stream_avl, renamed);
216511834STom.Erickson@Sun.COM 
216611834STom.Erickson@Sun.COM 			/* Unmount renamed filesystems before receiving. */
216711834STom.Erickson@Sun.COM 			while ((pair = nvlist_next_nvpair(renamed,
216811834STom.Erickson@Sun.COM 			    pair)) != NULL) {
216911834STom.Erickson@Sun.COM 				zfs_handle_t *zhp;
217011834STom.Erickson@Sun.COM 				prop_changelist_t *clp = NULL;
217111834STom.Erickson@Sun.COM 
217211834STom.Erickson@Sun.COM 				zhp = zfs_open(hdl, nvpair_name(pair),
217311834STom.Erickson@Sun.COM 				    ZFS_TYPE_FILESYSTEM);
217411834STom.Erickson@Sun.COM 				if (zhp != NULL) {
217511834STom.Erickson@Sun.COM 					clp = changelist_gather(zhp,
217611834STom.Erickson@Sun.COM 					    ZFS_PROP_MOUNTPOINT, 0, 0);
217711834STom.Erickson@Sun.COM 					zfs_close(zhp);
217811834STom.Erickson@Sun.COM 					if (clp != NULL) {
217911834STom.Erickson@Sun.COM 						softerr |=
218011834STom.Erickson@Sun.COM 						    changelist_prefix(clp);
218111834STom.Erickson@Sun.COM 						changelist_free(clp);
218211834STom.Erickson@Sun.COM 					}
218311834STom.Erickson@Sun.COM 				}
218411834STom.Erickson@Sun.COM 			}
218511834STom.Erickson@Sun.COM 
218611834STom.Erickson@Sun.COM 			nvlist_free(renamed);
21875367Sahrens 		}
21885367Sahrens 	}
21895367Sahrens 
219011834STom.Erickson@Sun.COM 	/*
219111834STom.Erickson@Sun.COM 	 * Get the fs specified by the first path in the stream (the top level
219211834STom.Erickson@Sun.COM 	 * specified by 'zfs send') and pass it to each invocation of
219311834STom.Erickson@Sun.COM 	 * zfs_receive_one().
219411834STom.Erickson@Sun.COM 	 */
219511834STom.Erickson@Sun.COM 	(void) strlcpy(sendfs, drr->drr_u.drr_begin.drr_toname,
219611834STom.Erickson@Sun.COM 	    ZFS_MAXNAMELEN);
219711834STom.Erickson@Sun.COM 	if ((cp = strchr(sendfs, '@')) != NULL)
219811834STom.Erickson@Sun.COM 		*cp = '\0';
21995367Sahrens 
22005367Sahrens 	/* Finally, receive each contained stream */
22015367Sahrens 	do {
22025367Sahrens 		/*
22035367Sahrens 		 * we should figure out if it has a recoverable
22045367Sahrens 		 * error, in which case do a recv_skip() and drive on.
22055367Sahrens 		 * Note, if we fail due to already having this guid,
22065367Sahrens 		 * zfs_receive_one() will take care of it (ie,
22075367Sahrens 		 * recv_skip() and return 0).
22085367Sahrens 		 */
22097366STim.Haley@Sun.COM 		error = zfs_receive_impl(hdl, destname, flags, fd,
221012527SChris.Kirby@oracle.com 		    sendfs, stream_nv, stream_avl, top_zfs, cleanup_fd,
221112527SChris.Kirby@oracle.com 		    action_handlep);
22125367Sahrens 		if (error == ENODATA) {
22135367Sahrens 			error = 0;
22145367Sahrens 			break;
22155367Sahrens 		}
22165367Sahrens 		anyerr |= error;
22175367Sahrens 	} while (error == 0);
22185367Sahrens 
22195367Sahrens 	if (drr->drr_payloadlen != 0 && fromsnap != NULL) {
22205367Sahrens 		/*
22215367Sahrens 		 * Now that we have the fs's they sent us, try the
22225367Sahrens 		 * renames again.
22235367Sahrens 		 */
22246479Sahrens 		softerr = recv_incremental_replication(hdl, tofs, flags,
222511834STom.Erickson@Sun.COM 		    stream_nv, stream_avl, NULL);
22265367Sahrens 	}
22275367Sahrens 
22285367Sahrens out:
22295367Sahrens 	fsavl_destroy(stream_avl);
22305367Sahrens 	if (stream_nv)
22315367Sahrens 		nvlist_free(stream_nv);
22326479Sahrens 	if (softerr)
22336479Sahrens 		error = -2;
22345367Sahrens 	if (anyerr)
22355367Sahrens 		error = -1;
22365367Sahrens 	return (error);
22375367Sahrens }
22385367Sahrens 
223911022STom.Erickson@Sun.COM static void
trunc_prop_errs(int truncated)224011022STom.Erickson@Sun.COM trunc_prop_errs(int truncated)
224111022STom.Erickson@Sun.COM {
224211022STom.Erickson@Sun.COM 	ASSERT(truncated != 0);
224311022STom.Erickson@Sun.COM 
224411022STom.Erickson@Sun.COM 	if (truncated == 1)
224511022STom.Erickson@Sun.COM 		(void) fprintf(stderr, dgettext(TEXT_DOMAIN,
224611022STom.Erickson@Sun.COM 		    "1 more property could not be set\n"));
224711022STom.Erickson@Sun.COM 	else
224811022STom.Erickson@Sun.COM 		(void) fprintf(stderr, dgettext(TEXT_DOMAIN,
224911022STom.Erickson@Sun.COM 		    "%d more properties could not be set\n"), truncated);
225011022STom.Erickson@Sun.COM }
225111022STom.Erickson@Sun.COM 
22525367Sahrens static int
recv_skip(libzfs_handle_t * hdl,int fd,boolean_t byteswap)22535367Sahrens recv_skip(libzfs_handle_t *hdl, int fd, boolean_t byteswap)
22545367Sahrens {
22555367Sahrens 	dmu_replay_record_t *drr;
22565367Sahrens 	void *buf = malloc(1<<20);
225711007SLori.Alt@Sun.COM 	char errbuf[1024];
225811007SLori.Alt@Sun.COM 
225911007SLori.Alt@Sun.COM 	(void) snprintf(errbuf, sizeof (errbuf), dgettext(TEXT_DOMAIN,
226011007SLori.Alt@Sun.COM 	    "cannot receive:"));
22615367Sahrens 
22625367Sahrens 	/* XXX would be great to use lseek if possible... */
22635367Sahrens 	drr = buf;
22645367Sahrens 
22655367Sahrens 	while (recv_read(hdl, fd, drr, sizeof (dmu_replay_record_t),
22665367Sahrens 	    byteswap, NULL) == 0) {
22675367Sahrens 		if (byteswap)
22685367Sahrens 			drr->drr_type = BSWAP_32(drr->drr_type);
22695367Sahrens 
22705367Sahrens 		switch (drr->drr_type) {
22715367Sahrens 		case DRR_BEGIN:
22725367Sahrens 			/* NB: not to be used on v2 stream packages */
227311007SLori.Alt@Sun.COM 			if (drr->drr_payloadlen != 0) {
227411007SLori.Alt@Sun.COM 				zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
227511007SLori.Alt@Sun.COM 				    "invalid substream header"));
227611007SLori.Alt@Sun.COM 				return (zfs_error(hdl, EZFS_BADSTREAM, errbuf));
227711007SLori.Alt@Sun.COM 			}
22785367Sahrens 			break;
22795367Sahrens 
22805367Sahrens 		case DRR_END:
22815367Sahrens 			free(buf);
22825367Sahrens 			return (0);
22835367Sahrens 
22845367Sahrens 		case DRR_OBJECT:
22855367Sahrens 			if (byteswap) {
22865367Sahrens 				drr->drr_u.drr_object.drr_bonuslen =
22875367Sahrens 				    BSWAP_32(drr->drr_u.drr_object.
22885367Sahrens 				    drr_bonuslen);
22895367Sahrens 			}
22905367Sahrens 			(void) recv_read(hdl, fd, buf,
22915367Sahrens 			    P2ROUNDUP(drr->drr_u.drr_object.drr_bonuslen, 8),
22925367Sahrens 			    B_FALSE, NULL);
22935367Sahrens 			break;
22945367Sahrens 
22955367Sahrens 		case DRR_WRITE:
22965367Sahrens 			if (byteswap) {
22975367Sahrens 				drr->drr_u.drr_write.drr_length =
22987366STim.Haley@Sun.COM 				    BSWAP_64(drr->drr_u.drr_write.drr_length);
22995367Sahrens 			}
23005367Sahrens 			(void) recv_read(hdl, fd, buf,
23015367Sahrens 			    drr->drr_u.drr_write.drr_length, B_FALSE, NULL);
23025367Sahrens 			break;
230311935SMark.Shellenbaum@Sun.COM 		case DRR_SPILL:
230411935SMark.Shellenbaum@Sun.COM 			if (byteswap) {
230511935SMark.Shellenbaum@Sun.COM 				drr->drr_u.drr_write.drr_length =
230611935SMark.Shellenbaum@Sun.COM 				    BSWAP_64(drr->drr_u.drr_spill.drr_length);
230711935SMark.Shellenbaum@Sun.COM 			}
230811935SMark.Shellenbaum@Sun.COM 			(void) recv_read(hdl, fd, buf,
230911935SMark.Shellenbaum@Sun.COM 			    drr->drr_u.drr_spill.drr_length, B_FALSE, NULL);
231011935SMark.Shellenbaum@Sun.COM 			break;
231111007SLori.Alt@Sun.COM 		case DRR_WRITE_BYREF:
23125367Sahrens 		case DRR_FREEOBJECTS:
23135367Sahrens 		case DRR_FREE:
23145367Sahrens 			break;
23155367Sahrens 
23165367Sahrens 		default:
231711007SLori.Alt@Sun.COM 			zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
231811007SLori.Alt@Sun.COM 			    "invalid record type"));
231911007SLori.Alt@Sun.COM 			return (zfs_error(hdl, EZFS_BADSTREAM, errbuf));
23205367Sahrens 		}
23215367Sahrens 	}
23225367Sahrens 
23235367Sahrens 	free(buf);
23245367Sahrens 	return (-1);
23255367Sahrens }
23265367Sahrens 
23275367Sahrens /*
23285367Sahrens  * Restores a backup of tosnap from the file descriptor specified by infd.
23295367Sahrens  */
23305367Sahrens static int
zfs_receive_one(libzfs_handle_t * hdl,int infd,const char * tosnap,recvflags_t flags,dmu_replay_record_t * drr,dmu_replay_record_t * drr_noswap,const char * sendfs,nvlist_t * stream_nv,avl_tree_t * stream_avl,char ** top_zfs,int cleanup_fd,uint64_t * action_handlep)23315367Sahrens zfs_receive_one(libzfs_handle_t *hdl, int infd, const char *tosnap,
23325367Sahrens     recvflags_t flags, dmu_replay_record_t *drr,
233311834STom.Erickson@Sun.COM     dmu_replay_record_t *drr_noswap, const char *sendfs,
233412527SChris.Kirby@oracle.com     nvlist_t *stream_nv, avl_tree_t *stream_avl, char **top_zfs, int cleanup_fd,
233512527SChris.Kirby@oracle.com     uint64_t *action_handlep)
23365367Sahrens {
23375367Sahrens 	zfs_cmd_t zc = { 0 };
23385367Sahrens 	time_t begin_time;
233911834STom.Erickson@Sun.COM 	int ioctl_err, ioctl_errno, err;
23405367Sahrens 	char *cp;
23415367Sahrens 	struct drr_begin *drrb = &drr->drr_u.drr_begin;
23425367Sahrens 	char errbuf[1024];
234311022STom.Erickson@Sun.COM 	char prop_errbuf[1024];
234411834STom.Erickson@Sun.COM 	const char *chopprefix;
23455367Sahrens 	boolean_t newfs = B_FALSE;
23465367Sahrens 	boolean_t stream_wantsnewfs;
23475367Sahrens 	uint64_t parent_snapguid = 0;
23485367Sahrens 	prop_changelist_t *clp = NULL;
23497265Sahrens 	nvlist_t *snapprops_nvlist = NULL;
235011022STom.Erickson@Sun.COM 	zprop_errflags_t prop_errflags;
235111834STom.Erickson@Sun.COM 	boolean_t recursive;
23525367Sahrens 
23535367Sahrens 	begin_time = time(NULL);
23545367Sahrens 
23555367Sahrens 	(void) snprintf(errbuf, sizeof (errbuf), dgettext(TEXT_DOMAIN,
23565367Sahrens 	    "cannot receive"));
23575367Sahrens 
235811834STom.Erickson@Sun.COM 	recursive = (nvlist_lookup_boolean(stream_nv, "not_recursive") ==
235911834STom.Erickson@Sun.COM 	    ENOENT);
236011834STom.Erickson@Sun.COM 
23615367Sahrens 	if (stream_avl != NULL) {
23627265Sahrens 		char *snapname;
23637265Sahrens 		nvlist_t *fs = fsavl_find(stream_avl, drrb->drr_toguid,
23647265Sahrens 		    &snapname);
23655367Sahrens 		nvlist_t *props;
23665896Sahl 		int ret;
23675367Sahrens 
23685367Sahrens 		(void) nvlist_lookup_uint64(fs, "parentfromsnap",
23695367Sahrens 		    &parent_snapguid);
23705367Sahrens 		err = nvlist_lookup_nvlist(fs, "props", &props);
23715367Sahrens 		if (err)
23725367Sahrens 			VERIFY(0 == nvlist_alloc(&props, NV_UNIQUE_NAME, 0));
23735896Sahl 
23745367Sahrens 		if (flags.canmountoff) {
23755367Sahrens 			VERIFY(0 == nvlist_add_uint64(props,
23765367Sahrens 			    zfs_prop_to_name(ZFS_PROP_CANMOUNT), 0));
23775367Sahrens 		}
23785896Sahl 		ret = zcmd_write_src_nvlist(hdl, &zc, props);
23795367Sahrens 		if (err)
23805367Sahrens 			nvlist_free(props);
23815896Sahl 
23827265Sahrens 		if (0 == nvlist_lookup_nvlist(fs, "snapprops", &props)) {
23837265Sahrens 			VERIFY(0 == nvlist_lookup_nvlist(props,
23847265Sahrens 			    snapname, &snapprops_nvlist));
23857265Sahrens 		}
23867265Sahrens 
23875896Sahl 		if (ret != 0)
23885896Sahl 			return (-1);
23895367Sahrens 	}
23905367Sahrens 
239111834STom.Erickson@Sun.COM 	cp = NULL;
239211834STom.Erickson@Sun.COM 
23935367Sahrens 	/*
23945367Sahrens 	 * Determine how much of the snapshot name stored in the stream
23955367Sahrens 	 * we are going to tack on to the name they specified on the
23965367Sahrens 	 * command line, and how much we are going to chop off.
23975367Sahrens 	 *
23985367Sahrens 	 * If they specified a snapshot, chop the entire name stored in
23995367Sahrens 	 * the stream.
24005367Sahrens 	 */
240111834STom.Erickson@Sun.COM 	if (flags.istail) {
24025367Sahrens 		/*
240311834STom.Erickson@Sun.COM 		 * A filesystem was specified with -e. We want to tack on only
240411834STom.Erickson@Sun.COM 		 * the tail of the sent snapshot path.
24055367Sahrens 		 */
24065367Sahrens 		if (strchr(tosnap, '@')) {
24075367Sahrens 			zfs_error_aux(hdl, dgettext(TEXT_DOMAIN, "invalid "
240811834STom.Erickson@Sun.COM 			    "argument - snapshot not allowed with -e"));
24095367Sahrens 			return (zfs_error(hdl, EZFS_INVALIDNAME, errbuf));
24105367Sahrens 		}
241111834STom.Erickson@Sun.COM 
241211834STom.Erickson@Sun.COM 		chopprefix = strrchr(sendfs, '/');
241311834STom.Erickson@Sun.COM 
241411834STom.Erickson@Sun.COM 		if (chopprefix == NULL) {
241511834STom.Erickson@Sun.COM 			/*
241611834STom.Erickson@Sun.COM 			 * The tail is the poolname, so we need to
241711834STom.Erickson@Sun.COM 			 * prepend a path separator.
241811834STom.Erickson@Sun.COM 			 */
241911834STom.Erickson@Sun.COM 			int len = strlen(drrb->drr_toname);
242011834STom.Erickson@Sun.COM 			cp = malloc(len + 2);
242111834STom.Erickson@Sun.COM 			cp[0] = '/';
242211834STom.Erickson@Sun.COM 			(void) strcpy(&cp[1], drrb->drr_toname);
242311834STom.Erickson@Sun.COM 			chopprefix = cp;
242411834STom.Erickson@Sun.COM 		} else {
242511834STom.Erickson@Sun.COM 			chopprefix = drrb->drr_toname + (chopprefix - sendfs);
242611834STom.Erickson@Sun.COM 		}
242711834STom.Erickson@Sun.COM 	} else if (flags.isprefix) {
242811834STom.Erickson@Sun.COM 		/*
242911834STom.Erickson@Sun.COM 		 * A filesystem was specified with -d. We want to tack on
243011834STom.Erickson@Sun.COM 		 * everything but the first element of the sent snapshot path
243111834STom.Erickson@Sun.COM 		 * (all but the pool name).
243211834STom.Erickson@Sun.COM 		 */
243311834STom.Erickson@Sun.COM 		if (strchr(tosnap, '@')) {
243411834STom.Erickson@Sun.COM 			zfs_error_aux(hdl, dgettext(TEXT_DOMAIN, "invalid "
243511834STom.Erickson@Sun.COM 			    "argument - snapshot not allowed with -d"));
243611834STom.Erickson@Sun.COM 			return (zfs_error(hdl, EZFS_INVALIDNAME, errbuf));
243711834STom.Erickson@Sun.COM 		}
243811834STom.Erickson@Sun.COM 
243911834STom.Erickson@Sun.COM 		chopprefix = strchr(drrb->drr_toname, '/');
244011834STom.Erickson@Sun.COM 		if (chopprefix == NULL)
244111834STom.Erickson@Sun.COM 			chopprefix = strchr(drrb->drr_toname, '@');
24425367Sahrens 	} else if (strchr(tosnap, '@') == NULL) {
24435367Sahrens 		/*
244411834STom.Erickson@Sun.COM 		 * If a filesystem was specified without -d or -e, we want to
244511834STom.Erickson@Sun.COM 		 * tack on everything after the fs specified by 'zfs send'.
24465367Sahrens 		 */
244711834STom.Erickson@Sun.COM 		chopprefix = drrb->drr_toname + strlen(sendfs);
244811834STom.Erickson@Sun.COM 	} else {
244911834STom.Erickson@Sun.COM 		/* A snapshot was specified as an exact path (no -d or -e). */
245011834STom.Erickson@Sun.COM 		if (recursive) {
245111834STom.Erickson@Sun.COM 			zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
245211834STom.Erickson@Sun.COM 			    "cannot specify snapshot name for multi-snapshot "
245311834STom.Erickson@Sun.COM 			    "stream"));
245411834STom.Erickson@Sun.COM 			return (zfs_error(hdl, EZFS_BADSTREAM, errbuf));
245511834STom.Erickson@Sun.COM 		}
245611834STom.Erickson@Sun.COM 		chopprefix = drrb->drr_toname + strlen(drrb->drr_toname);
24575367Sahrens 	}
245811834STom.Erickson@Sun.COM 
245911834STom.Erickson@Sun.COM 	ASSERT(strstr(drrb->drr_toname, sendfs) == drrb->drr_toname);
246011834STom.Erickson@Sun.COM 	ASSERT(chopprefix > drrb->drr_toname);
246111834STom.Erickson@Sun.COM 	ASSERT(chopprefix <= drrb->drr_toname + strlen(drrb->drr_toname));
246211834STom.Erickson@Sun.COM 	ASSERT(chopprefix[0] == '/' || chopprefix[0] == '@' ||
246311834STom.Erickson@Sun.COM 	    chopprefix[0] == '\0');
24645367Sahrens 
24655367Sahrens 	/*
24665367Sahrens 	 * Determine name of destination snapshot, store in zc_value.
24675367Sahrens 	 */
246811007SLori.Alt@Sun.COM 	(void) strcpy(zc.zc_top_ds, tosnap);
24695367Sahrens 	(void) strcpy(zc.zc_value, tosnap);
247011834STom.Erickson@Sun.COM 	(void) strncat(zc.zc_value, chopprefix, sizeof (zc.zc_value));
247111834STom.Erickson@Sun.COM 	free(cp);
24725896Sahl 	if (!zfs_name_valid(zc.zc_value, ZFS_TYPE_SNAPSHOT)) {
24735896Sahl 		zcmd_free_nvlists(&zc);
24745367Sahrens 		return (zfs_error(hdl, EZFS_INVALIDNAME, errbuf));
24755896Sahl 	}
24765367Sahrens 
24775367Sahrens 	/*
24785367Sahrens 	 * Determine the name of the origin snapshot, store in zc_string.
24795367Sahrens 	 */
24805367Sahrens 	if (drrb->drr_flags & DRR_FLAG_CLONE) {
24815367Sahrens 		if (guid_to_name(hdl, tosnap,
24825367Sahrens 		    drrb->drr_fromguid, zc.zc_string) != 0) {
24835896Sahl 			zcmd_free_nvlists(&zc);
24845367Sahrens 			zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
24855367Sahrens 			    "local origin for clone %s does not exist"),
24865367Sahrens 			    zc.zc_value);
24875367Sahrens 			return (zfs_error(hdl, EZFS_NOENT, errbuf));
24885367Sahrens 		}
24895367Sahrens 		if (flags.verbose)
24905367Sahrens 			(void) printf("found clone origin %s\n", zc.zc_string);
24915367Sahrens 	}
24925367Sahrens 
24935367Sahrens 	stream_wantsnewfs = (drrb->drr_fromguid == NULL ||
24945367Sahrens 	    (drrb->drr_flags & DRR_FLAG_CLONE));
24955367Sahrens 
24965367Sahrens 	if (stream_wantsnewfs) {
24975367Sahrens 		/*
24985367Sahrens 		 * if the parent fs does not exist, look for it based on
24995367Sahrens 		 * the parent snap GUID
25005367Sahrens 		 */
25015367Sahrens 		(void) snprintf(errbuf, sizeof (errbuf), dgettext(TEXT_DOMAIN,
25025367Sahrens 		    "cannot receive new filesystem stream"));
25035367Sahrens 
25045367Sahrens 		(void) strcpy(zc.zc_name, zc.zc_value);
25055367Sahrens 		cp = strrchr(zc.zc_name, '/');
25065367Sahrens 		if (cp)
25075367Sahrens 			*cp = '\0';
25085367Sahrens 		if (cp &&
25095367Sahrens 		    !zfs_dataset_exists(hdl, zc.zc_name, ZFS_TYPE_DATASET)) {
25105367Sahrens 			char suffix[ZFS_MAXNAMELEN];
25115367Sahrens 			(void) strcpy(suffix, strrchr(zc.zc_value, '/'));
25125367Sahrens 			if (guid_to_name(hdl, tosnap, parent_snapguid,
25135367Sahrens 			    zc.zc_value) == 0) {
25145367Sahrens 				*strchr(zc.zc_value, '@') = '\0';
25155367Sahrens 				(void) strcat(zc.zc_value, suffix);
25165367Sahrens 			}
25175367Sahrens 		}
25185367Sahrens 	} else {
25195367Sahrens 		/*
25205367Sahrens 		 * if the fs does not exist, look for it based on the
25215367Sahrens 		 * fromsnap GUID
25225367Sahrens 		 */
25235367Sahrens 		(void) snprintf(errbuf, sizeof (errbuf), dgettext(TEXT_DOMAIN,
25245367Sahrens 		    "cannot receive incremental stream"));
25255367Sahrens 
25265367Sahrens 		(void) strcpy(zc.zc_name, zc.zc_value);
25275367Sahrens 		*strchr(zc.zc_name, '@') = '\0';
25285367Sahrens 
252911976STom.Erickson@Sun.COM 		/*
253011976STom.Erickson@Sun.COM 		 * If the exact receive path was specified and this is the
253111976STom.Erickson@Sun.COM 		 * topmost path in the stream, then if the fs does not exist we
253211976STom.Erickson@Sun.COM 		 * should look no further.
253311976STom.Erickson@Sun.COM 		 */
253411976STom.Erickson@Sun.COM 		if ((flags.isprefix || (*(chopprefix = drrb->drr_toname +
253511976STom.Erickson@Sun.COM 		    strlen(sendfs)) != '\0' && *chopprefix != '@')) &&
253611976STom.Erickson@Sun.COM 		    !zfs_dataset_exists(hdl, zc.zc_name, ZFS_TYPE_DATASET)) {
25375367Sahrens 			char snap[ZFS_MAXNAMELEN];
25385367Sahrens 			(void) strcpy(snap, strchr(zc.zc_value, '@'));
25395367Sahrens 			if (guid_to_name(hdl, tosnap, drrb->drr_fromguid,
25405367Sahrens 			    zc.zc_value) == 0) {
25415367Sahrens 				*strchr(zc.zc_value, '@') = '\0';
25425367Sahrens 				(void) strcat(zc.zc_value, snap);
25435367Sahrens 			}
25445367Sahrens 		}
25455367Sahrens 	}
25465367Sahrens 
25475367Sahrens 	(void) strcpy(zc.zc_name, zc.zc_value);
25485367Sahrens 	*strchr(zc.zc_name, '@') = '\0';
25495367Sahrens 
25505367Sahrens 	if (zfs_dataset_exists(hdl, zc.zc_name, ZFS_TYPE_DATASET)) {
25515367Sahrens 		zfs_handle_t *zhp;
255211834STom.Erickson@Sun.COM 
25535367Sahrens 		/*
25545367Sahrens 		 * Destination fs exists.  Therefore this should either
25555367Sahrens 		 * be an incremental, or the stream specifies a new fs
25565367Sahrens 		 * (full stream or clone) and they want us to blow it
25575367Sahrens 		 * away (and have therefore specified -F and removed any
25585367Sahrens 		 * snapshots).
25595367Sahrens 		 */
25605367Sahrens 		if (stream_wantsnewfs) {
25615367Sahrens 			if (!flags.force) {
25625896Sahl 				zcmd_free_nvlists(&zc);
25635367Sahrens 				zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
25645367Sahrens 				    "destination '%s' exists\n"
25655367Sahrens 				    "must specify -F to overwrite it"),
25665367Sahrens 				    zc.zc_name);
25675367Sahrens 				return (zfs_error(hdl, EZFS_EXISTS, errbuf));
25685367Sahrens 			}
25695367Sahrens 			if (ioctl(hdl->libzfs_fd, ZFS_IOC_SNAPSHOT_LIST_NEXT,
25705367Sahrens 			    &zc) == 0) {
25715896Sahl 				zcmd_free_nvlists(&zc);
25725367Sahrens 				zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
25735367Sahrens 				    "destination has snapshots (eg. %s)\n"
25745367Sahrens 				    "must destroy them to overwrite it"),
25755367Sahrens 				    zc.zc_name);
25765367Sahrens 				return (zfs_error(hdl, EZFS_EXISTS, errbuf));
25775367Sahrens 			}
25785367Sahrens 		}
25795367Sahrens 
25805896Sahl 		if ((zhp = zfs_open(hdl, zc.zc_name,
25815896Sahl 		    ZFS_TYPE_FILESYSTEM | ZFS_TYPE_VOLUME)) == NULL) {
25825896Sahl 			zcmd_free_nvlists(&zc);
25835367Sahrens 			return (-1);
25845896Sahl 		}
25855896Sahl 
25865367Sahrens 		if (stream_wantsnewfs &&
25875367Sahrens 		    zhp->zfs_dmustats.dds_origin[0]) {
25885896Sahl 			zcmd_free_nvlists(&zc);
25895367Sahrens 			zfs_close(zhp);
25905367Sahrens 			zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
25915367Sahrens 			    "destination '%s' is a clone\n"
25925367Sahrens 			    "must destroy it to overwrite it"),
25935367Sahrens 			    zc.zc_name);
25945367Sahrens 			return (zfs_error(hdl, EZFS_EXISTS, errbuf));
25955367Sahrens 		}
25965367Sahrens 
25975367Sahrens 		if (!flags.dryrun && zhp->zfs_type == ZFS_TYPE_FILESYSTEM &&
25985367Sahrens 		    stream_wantsnewfs) {
25995367Sahrens 			/* We can't do online recv in this case */
26007366STim.Haley@Sun.COM 			clp = changelist_gather(zhp, ZFS_PROP_NAME, 0, 0);
26015896Sahl 			if (clp == NULL) {
260210200SLori.Alt@Sun.COM 				zfs_close(zhp);
26035896Sahl 				zcmd_free_nvlists(&zc);
26045367Sahrens 				return (-1);
26055896Sahl 			}
26065367Sahrens 			if (changelist_prefix(clp) != 0) {
26075367Sahrens 				changelist_free(clp);
260810200SLori.Alt@Sun.COM 				zfs_close(zhp);
26095896Sahl 				zcmd_free_nvlists(&zc);
26105367Sahrens 				return (-1);
26115367Sahrens 			}
26125367Sahrens 		}
26135367Sahrens 		zfs_close(zhp);
26145367Sahrens 	} else {
26155367Sahrens 		/*
26165896Sahl 		 * Destination filesystem does not exist.  Therefore we better
26175896Sahl 		 * be creating a new filesystem (either from a full backup, or
26185896Sahl 		 * a clone).  It would therefore be invalid if the user
26195896Sahl 		 * specified only the pool name (i.e. if the destination name
26205896Sahl 		 * contained no slash character).
26215367Sahrens 		 */
26225896Sahl 		if (!stream_wantsnewfs ||
26235896Sahl 		    (cp = strrchr(zc.zc_name, '/')) == NULL) {
26245896Sahl 			zcmd_free_nvlists(&zc);
26255367Sahrens 			zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
26265367Sahrens 			    "destination '%s' does not exist"), zc.zc_name);
26275367Sahrens 			return (zfs_error(hdl, EZFS_NOENT, errbuf));
26285367Sahrens 		}
26295367Sahrens 
26305896Sahl 		/*
26315896Sahl 		 * Trim off the final dataset component so we perform the
26325896Sahl 		 * recvbackup ioctl to the filesystems's parent.
26335896Sahl 		 */
26345896Sahl 		*cp = '\0';
26355367Sahrens 
263611834STom.Erickson@Sun.COM 		if (flags.isprefix && !flags.istail && !flags.dryrun &&
26375896Sahl 		    create_parents(hdl, zc.zc_value, strlen(tosnap)) != 0) {
26385896Sahl 			zcmd_free_nvlists(&zc);
26395896Sahl 			return (zfs_error(hdl, EZFS_BADRESTORE, errbuf));
26405367Sahrens 		}
26415367Sahrens 
26425367Sahrens 		newfs = B_TRUE;
26435367Sahrens 	}
26445367Sahrens 
26455367Sahrens 	zc.zc_begin_record = drr_noswap->drr_u.drr_begin;
26465367Sahrens 	zc.zc_cookie = infd;
26475367Sahrens 	zc.zc_guid = flags.force;
26485367Sahrens 	if (flags.verbose) {
26495367Sahrens 		(void) printf("%s %s stream of %s into %s\n",
26505367Sahrens 		    flags.dryrun ? "would receive" : "receiving",
26515367Sahrens 		    drrb->drr_fromguid ? "incremental" : "full",
26525367Sahrens 		    drrb->drr_toname, zc.zc_value);
26535367Sahrens 		(void) fflush(stdout);
26545367Sahrens 	}
26555367Sahrens 
26565896Sahl 	if (flags.dryrun) {
26575896Sahl 		zcmd_free_nvlists(&zc);
26585367Sahrens 		return (recv_skip(hdl, infd, flags.byteswap));
26595896Sahl 	}
26605367Sahrens 
266111022STom.Erickson@Sun.COM 	zc.zc_nvlist_dst = (uint64_t)(uintptr_t)prop_errbuf;
266211022STom.Erickson@Sun.COM 	zc.zc_nvlist_dst_size = sizeof (prop_errbuf);
266312527SChris.Kirby@oracle.com 	zc.zc_cleanup_fd = cleanup_fd;
266412527SChris.Kirby@oracle.com 	zc.zc_action_handle = *action_handlep;
266511022STom.Erickson@Sun.COM 
26665367Sahrens 	err = ioctl_err = zfs_ioctl(hdl, ZFS_IOC_RECV, &zc);
26675367Sahrens 	ioctl_errno = errno;
266811022STom.Erickson@Sun.COM 	prop_errflags = (zprop_errflags_t)zc.zc_obj;
266911022STom.Erickson@Sun.COM 
267011022STom.Erickson@Sun.COM 	if (err == 0) {
267111022STom.Erickson@Sun.COM 		nvlist_t *prop_errors;
267211022STom.Erickson@Sun.COM 		VERIFY(0 == nvlist_unpack((void *)(uintptr_t)zc.zc_nvlist_dst,
267311022STom.Erickson@Sun.COM 		    zc.zc_nvlist_dst_size, &prop_errors, 0));
267411022STom.Erickson@Sun.COM 
267511022STom.Erickson@Sun.COM 		nvpair_t *prop_err = NULL;
267611022STom.Erickson@Sun.COM 
267711022STom.Erickson@Sun.COM 		while ((prop_err = nvlist_next_nvpair(prop_errors,
267811022STom.Erickson@Sun.COM 		    prop_err)) != NULL) {
267911022STom.Erickson@Sun.COM 			char tbuf[1024];
268011022STom.Erickson@Sun.COM 			zfs_prop_t prop;
268111022STom.Erickson@Sun.COM 			int intval;
268211022STom.Erickson@Sun.COM 
268311022STom.Erickson@Sun.COM 			prop = zfs_name_to_prop(nvpair_name(prop_err));
268411022STom.Erickson@Sun.COM 			(void) nvpair_value_int32(prop_err, &intval);
268511022STom.Erickson@Sun.COM 			if (strcmp(nvpair_name(prop_err),
268611022STom.Erickson@Sun.COM 			    ZPROP_N_MORE_ERRORS) == 0) {
268711022STom.Erickson@Sun.COM 				trunc_prop_errs(intval);
268811022STom.Erickson@Sun.COM 				break;
268911022STom.Erickson@Sun.COM 			} else {
269011022STom.Erickson@Sun.COM 				(void) snprintf(tbuf, sizeof (tbuf),
269111022STom.Erickson@Sun.COM 				    dgettext(TEXT_DOMAIN,
269211022STom.Erickson@Sun.COM 				    "cannot receive %s property on %s"),
269311022STom.Erickson@Sun.COM 				    nvpair_name(prop_err), zc.zc_name);
269411022STom.Erickson@Sun.COM 				zfs_setprop_error(hdl, prop, intval, tbuf);
269511022STom.Erickson@Sun.COM 			}
269611022STom.Erickson@Sun.COM 		}
269711022STom.Erickson@Sun.COM 		nvlist_free(prop_errors);
269811022STom.Erickson@Sun.COM 	}
269911022STom.Erickson@Sun.COM 
270011022STom.Erickson@Sun.COM 	zc.zc_nvlist_dst = 0;
270111022STom.Erickson@Sun.COM 	zc.zc_nvlist_dst_size = 0;
27027265Sahrens 	zcmd_free_nvlists(&zc);
27037265Sahrens 
27047265Sahrens 	if (err == 0 && snapprops_nvlist) {
27057265Sahrens 		zfs_cmd_t zc2 = { 0 };
27067265Sahrens 
27077273Sahrens 		(void) strcpy(zc2.zc_name, zc.zc_value);
270811022STom.Erickson@Sun.COM 		zc2.zc_cookie = B_TRUE; /* received */
27097265Sahrens 		if (zcmd_write_src_nvlist(hdl, &zc2, snapprops_nvlist) == 0) {
27107265Sahrens 			(void) zfs_ioctl(hdl, ZFS_IOC_SET_PROP, &zc2);
27117265Sahrens 			zcmd_free_nvlists(&zc2);
27127265Sahrens 		}
27137265Sahrens 	}
27147265Sahrens 
271511834STom.Erickson@Sun.COM 	if (err && (ioctl_errno == ENOENT || ioctl_errno == EEXIST)) {
27165367Sahrens 		/*
27175367Sahrens 		 * It may be that this snapshot already exists,
27185367Sahrens 		 * in which case we want to consume & ignore it
27195367Sahrens 		 * rather than failing.
27205367Sahrens 		 */
27215367Sahrens 		avl_tree_t *local_avl;
27225367Sahrens 		nvlist_t *local_nv, *fs;
272311834STom.Erickson@Sun.COM 		cp = strchr(zc.zc_value, '@');
27245367Sahrens 
27255367Sahrens 		/*
27265367Sahrens 		 * XXX Do this faster by just iterating over snaps in
27275367Sahrens 		 * this fs.  Also if zc_value does not exist, we will
27285367Sahrens 		 * get a strange "does not exist" error message.
27295367Sahrens 		 */
27305367Sahrens 		*cp = '\0';
273111022STom.Erickson@Sun.COM 		if (gather_nvlist(hdl, zc.zc_value, NULL, NULL, B_FALSE,
27325367Sahrens 		    &local_nv, &local_avl) == 0) {
27335367Sahrens 			*cp = '@';
27345367Sahrens 			fs = fsavl_find(local_avl, drrb->drr_toguid, NULL);
27355367Sahrens 			fsavl_destroy(local_avl);
27365367Sahrens 			nvlist_free(local_nv);
27375367Sahrens 
27385367Sahrens 			if (fs != NULL) {
27395367Sahrens 				if (flags.verbose) {
27405367Sahrens 					(void) printf("snap %s already exists; "
27415367Sahrens 					    "ignoring\n", zc.zc_value);
27425367Sahrens 				}
274311022STom.Erickson@Sun.COM 				err = ioctl_err = recv_skip(hdl, infd,
27445367Sahrens 				    flags.byteswap);
27455367Sahrens 			}
27465367Sahrens 		}
27475367Sahrens 		*cp = '@';
27485367Sahrens 	}
27495367Sahrens 
27505367Sahrens 	if (ioctl_err != 0) {
27515367Sahrens 		switch (ioctl_errno) {
27525367Sahrens 		case ENODEV:
27535367Sahrens 			cp = strchr(zc.zc_value, '@');
27545367Sahrens 			*cp = '\0';
27555367Sahrens 			zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
27565367Sahrens 			    "most recent snapshot of %s does not\n"
27575367Sahrens 			    "match incremental source"), zc.zc_value);
27585367Sahrens 			(void) zfs_error(hdl, EZFS_BADRESTORE, errbuf);
27595367Sahrens 			*cp = '@';
27605367Sahrens 			break;
27615367Sahrens 		case ETXTBSY:
27625367Sahrens 			zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
27635367Sahrens 			    "destination %s has been modified\n"
27645367Sahrens 			    "since most recent snapshot"), zc.zc_name);
27655367Sahrens 			(void) zfs_error(hdl, EZFS_BADRESTORE, errbuf);
27665367Sahrens 			break;
27675367Sahrens 		case EEXIST:
27685367Sahrens 			cp = strchr(zc.zc_value, '@');
27695367Sahrens 			if (newfs) {
27705367Sahrens 				/* it's the containing fs that exists */
27715367Sahrens 				*cp = '\0';
27725367Sahrens 			}
27735367Sahrens 			zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
27745367Sahrens 			    "destination already exists"));
27755367Sahrens 			(void) zfs_error_fmt(hdl, EZFS_EXISTS,
27765367Sahrens 			    dgettext(TEXT_DOMAIN, "cannot restore to %s"),
27775367Sahrens 			    zc.zc_value);
27785367Sahrens 			*cp = '@';
27795367Sahrens 			break;
27805367Sahrens 		case EINVAL:
27815367Sahrens 			(void) zfs_error(hdl, EZFS_BADSTREAM, errbuf);
27825367Sahrens 			break;
27835367Sahrens 		case ECKSUM:
27845367Sahrens 			zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
27855367Sahrens 			    "invalid stream (checksum mismatch)"));
27865367Sahrens 			(void) zfs_error(hdl, EZFS_BADSTREAM, errbuf);
27875367Sahrens 			break;
278812070SMark.Shellenbaum@Sun.COM 		case ENOTSUP:
278912070SMark.Shellenbaum@Sun.COM 			zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
279012070SMark.Shellenbaum@Sun.COM 			    "pool must be upgraded to receive this stream."));
279112070SMark.Shellenbaum@Sun.COM 			(void) zfs_error(hdl, EZFS_BADVERSION, errbuf);
279212070SMark.Shellenbaum@Sun.COM 			break;
279312141SJohn.Harres@Sun.COM 		case EDQUOT:
279412141SJohn.Harres@Sun.COM 			zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
279512141SJohn.Harres@Sun.COM 			    "destination %s space quota exceeded"), zc.zc_name);
279612141SJohn.Harres@Sun.COM 			(void) zfs_error(hdl, EZFS_BADRESTORE, errbuf);
279712141SJohn.Harres@Sun.COM 			break;
27985367Sahrens 		default:
27995367Sahrens 			(void) zfs_standard_error(hdl, ioctl_errno, errbuf);
28005367Sahrens 		}
28015367Sahrens 	}
28025367Sahrens 
28035367Sahrens 	/*
280410588SEric.Taylor@Sun.COM 	 * Mount the target filesystem (if created).  Also mount any
280510588SEric.Taylor@Sun.COM 	 * children of the target filesystem if we did a replication
280610588SEric.Taylor@Sun.COM 	 * receive (indicated by stream_avl being non-NULL).
28075367Sahrens 	 */
28085367Sahrens 	cp = strchr(zc.zc_value, '@');
28095367Sahrens 	if (cp && (ioctl_err == 0 || !newfs)) {
28105367Sahrens 		zfs_handle_t *h;
28115367Sahrens 
28125367Sahrens 		*cp = '\0';
28135367Sahrens 		h = zfs_open(hdl, zc.zc_value,
28145367Sahrens 		    ZFS_TYPE_FILESYSTEM | ZFS_TYPE_VOLUME);
28156476Sahl 		if (h != NULL) {
28165367Sahrens 			if (h->zfs_type == ZFS_TYPE_VOLUME) {
28177366STim.Haley@Sun.COM 				*cp = '@';
281810200SLori.Alt@Sun.COM 			} else if (newfs || stream_avl) {
28197366STim.Haley@Sun.COM 				/*
28207366STim.Haley@Sun.COM 				 * Track the first/top of hierarchy fs,
28217366STim.Haley@Sun.COM 				 * for mounting and sharing later.
28227366STim.Haley@Sun.COM 				 */
28237366STim.Haley@Sun.COM 				if (top_zfs && *top_zfs == NULL)
28247366STim.Haley@Sun.COM 					*top_zfs = zfs_strdup(hdl, zc.zc_value);
28255367Sahrens 			}
28266476Sahl 			zfs_close(h);
28275367Sahrens 		}
28287366STim.Haley@Sun.COM 		*cp = '@';
28295367Sahrens 	}
28305367Sahrens 
28315367Sahrens 	if (clp) {
28325367Sahrens 		err |= changelist_postfix(clp);
28335367Sahrens 		changelist_free(clp);
28345367Sahrens 	}
28355367Sahrens 
283611022STom.Erickson@Sun.COM 	if (prop_errflags & ZPROP_ERR_NOCLEAR) {
283711022STom.Erickson@Sun.COM 		(void) fprintf(stderr, dgettext(TEXT_DOMAIN, "Warning: "
283811022STom.Erickson@Sun.COM 		    "failed to clear unreceived properties on %s"),
283911022STom.Erickson@Sun.COM 		    zc.zc_name);
284011022STom.Erickson@Sun.COM 		(void) fprintf(stderr, "\n");
284111022STom.Erickson@Sun.COM 	}
284211022STom.Erickson@Sun.COM 	if (prop_errflags & ZPROP_ERR_NORESTORE) {
284311022STom.Erickson@Sun.COM 		(void) fprintf(stderr, dgettext(TEXT_DOMAIN, "Warning: "
284411022STom.Erickson@Sun.COM 		    "failed to restore original properties on %s"),
284511022STom.Erickson@Sun.COM 		    zc.zc_name);
284611022STom.Erickson@Sun.COM 		(void) fprintf(stderr, "\n");
284711022STom.Erickson@Sun.COM 	}
284811022STom.Erickson@Sun.COM 
28495367Sahrens 	if (err || ioctl_err)
28505367Sahrens 		return (-1);
28515367Sahrens 
285212527SChris.Kirby@oracle.com 	*action_handlep = zc.zc_action_handle;
285312527SChris.Kirby@oracle.com 
28545367Sahrens 	if (flags.verbose) {
28555367Sahrens 		char buf1[64];
28565367Sahrens 		char buf2[64];
28575367Sahrens 		uint64_t bytes = zc.zc_cookie;
28585367Sahrens 		time_t delta = time(NULL) - begin_time;
28595367Sahrens 		if (delta == 0)
28605367Sahrens 			delta = 1;
28615367Sahrens 		zfs_nicenum(bytes, buf1, sizeof (buf1));
28625367Sahrens 		zfs_nicenum(bytes/delta, buf2, sizeof (buf1));
28635367Sahrens 
28645367Sahrens 		(void) printf("received %sB stream in %lu seconds (%sB/sec)\n",
28655367Sahrens 		    buf1, delta, buf2);
28665367Sahrens 	}
28675367Sahrens 
28685367Sahrens 	return (0);
28695367Sahrens }
28705367Sahrens 
28717366STim.Haley@Sun.COM static int
zfs_receive_impl(libzfs_handle_t * hdl,const char * tosnap,recvflags_t flags,int infd,const char * sendfs,nvlist_t * stream_nv,avl_tree_t * stream_avl,char ** top_zfs,int cleanup_fd,uint64_t * action_handlep)28727366STim.Haley@Sun.COM zfs_receive_impl(libzfs_handle_t *hdl, const char *tosnap, recvflags_t flags,
287311834STom.Erickson@Sun.COM     int infd, const char *sendfs, nvlist_t *stream_nv, avl_tree_t *stream_avl,
287412527SChris.Kirby@oracle.com     char **top_zfs, int cleanup_fd, uint64_t *action_handlep)
28755367Sahrens {
28765367Sahrens 	int err;
28775367Sahrens 	dmu_replay_record_t drr, drr_noswap;
28785367Sahrens 	struct drr_begin *drrb = &drr.drr_u.drr_begin;
28795367Sahrens 	char errbuf[1024];
28805367Sahrens 	zio_cksum_t zcksum = { 0 };
288111007SLori.Alt@Sun.COM 	uint64_t featureflags;
288211007SLori.Alt@Sun.COM 	int hdrtype;
28835367Sahrens 
28845367Sahrens 	(void) snprintf(errbuf, sizeof (errbuf), dgettext(TEXT_DOMAIN,
28855367Sahrens 	    "cannot receive"));
28865367Sahrens 
28875367Sahrens 	if (flags.isprefix &&
28885367Sahrens 	    !zfs_dataset_exists(hdl, tosnap, ZFS_TYPE_DATASET)) {
28895367Sahrens 		zfs_error_aux(hdl, dgettext(TEXT_DOMAIN, "specified fs "
28905367Sahrens 		    "(%s) does not exist"), tosnap);
28915367Sahrens 		return (zfs_error(hdl, EZFS_NOENT, errbuf));
28925367Sahrens 	}
28935367Sahrens 
28945367Sahrens 	/* read in the BEGIN record */
28955367Sahrens 	if (0 != (err = recv_read(hdl, infd, &drr, sizeof (drr), B_FALSE,
28965367Sahrens 	    &zcksum)))
28975367Sahrens 		return (err);
28985367Sahrens 
28995367Sahrens 	if (drr.drr_type == DRR_END || drr.drr_type == BSWAP_32(DRR_END)) {
29005367Sahrens 		/* It's the double end record at the end of a package */
29015367Sahrens 		return (ENODATA);
29025367Sahrens 	}
29035367Sahrens 
29045367Sahrens 	/* the kernel needs the non-byteswapped begin record */
29055367Sahrens 	drr_noswap = drr;
29065367Sahrens 
29075446Sahrens 	flags.byteswap = B_FALSE;
29085367Sahrens 	if (drrb->drr_magic == BSWAP_64(DMU_BACKUP_MAGIC)) {
29095367Sahrens 		/*
29105367Sahrens 		 * We computed the checksum in the wrong byteorder in
29115367Sahrens 		 * recv_read() above; do it again correctly.
29125367Sahrens 		 */
29135367Sahrens 		bzero(&zcksum, sizeof (zio_cksum_t));
29145367Sahrens 		fletcher_4_incremental_byteswap(&drr, sizeof (drr), &zcksum);
29155367Sahrens 		flags.byteswap = B_TRUE;
29165367Sahrens 
29175367Sahrens 		drr.drr_type = BSWAP_32(drr.drr_type);
29185367Sahrens 		drr.drr_payloadlen = BSWAP_32(drr.drr_payloadlen);
29195367Sahrens 		drrb->drr_magic = BSWAP_64(drrb->drr_magic);
292011007SLori.Alt@Sun.COM 		drrb->drr_versioninfo = BSWAP_64(drrb->drr_versioninfo);
29215367Sahrens 		drrb->drr_creation_time = BSWAP_64(drrb->drr_creation_time);
29225367Sahrens 		drrb->drr_type = BSWAP_32(drrb->drr_type);
29235367Sahrens 		drrb->drr_flags = BSWAP_32(drrb->drr_flags);
29245367Sahrens 		drrb->drr_toguid = BSWAP_64(drrb->drr_toguid);
29255367Sahrens 		drrb->drr_fromguid = BSWAP_64(drrb->drr_fromguid);
29265367Sahrens 	}
29275367Sahrens 
29285367Sahrens 	if (drrb->drr_magic != DMU_BACKUP_MAGIC || drr.drr_type != DRR_BEGIN) {
29295367Sahrens 		zfs_error_aux(hdl, dgettext(TEXT_DOMAIN, "invalid "
29305367Sahrens 		    "stream (bad magic number)"));
29315367Sahrens 		return (zfs_error(hdl, EZFS_BADSTREAM, errbuf));
29325367Sahrens 	}
29335367Sahrens 
293411007SLori.Alt@Sun.COM 	featureflags = DMU_GET_FEATUREFLAGS(drrb->drr_versioninfo);
293511007SLori.Alt@Sun.COM 	hdrtype = DMU_GET_STREAM_HDRTYPE(drrb->drr_versioninfo);
293611007SLori.Alt@Sun.COM 
293711007SLori.Alt@Sun.COM 	if (!DMU_STREAM_SUPPORTED(featureflags) ||
293811007SLori.Alt@Sun.COM 	    (hdrtype != DMU_SUBSTREAM && hdrtype != DMU_COMPOUNDSTREAM)) {
293911007SLori.Alt@Sun.COM 		zfs_error_aux(hdl, dgettext(TEXT_DOMAIN,
294011007SLori.Alt@Sun.COM 		    "stream has unsupported feature, feature flags = %lx"),
294111007SLori.Alt@Sun.COM 		    featureflags);
294211007SLori.Alt@Sun.COM 		return (zfs_error(hdl, EZFS_BADSTREAM, errbuf));
294311007SLori.Alt@Sun.COM 	}
294411007SLori.Alt@Sun.COM 
29455367Sahrens 	if (strchr(drrb->drr_toname, '@') == NULL) {
29465367Sahrens 		zfs_error_aux(hdl, dgettext(TEXT_DOMAIN, "invalid "
29475367Sahrens 		    "stream (bad snapshot name)"));
29485367Sahrens 		return (zfs_error(hdl, EZFS_BADSTREAM, errbuf));
29495367Sahrens 	}
29505367Sahrens 
295111007SLori.Alt@Sun.COM 	if (DMU_GET_STREAM_HDRTYPE(drrb->drr_versioninfo) == DMU_SUBSTREAM) {
295211834STom.Erickson@Sun.COM 		char nonpackage_sendfs[ZFS_MAXNAMELEN];
295311834STom.Erickson@Sun.COM 		if (sendfs == NULL) {
295411834STom.Erickson@Sun.COM 			/*
295511834STom.Erickson@Sun.COM 			 * We were not called from zfs_receive_package(). Get
295611834STom.Erickson@Sun.COM 			 * the fs specified by 'zfs send'.
295711834STom.Erickson@Sun.COM 			 */
295811834STom.Erickson@Sun.COM 			char *cp;
295911834STom.Erickson@Sun.COM 			(void) strlcpy(nonpackage_sendfs,
296011834STom.Erickson@Sun.COM 			    drr.drr_u.drr_begin.drr_toname, ZFS_MAXNAMELEN);
296111834STom.Erickson@Sun.COM 			if ((cp = strchr(nonpackage_sendfs, '@')) != NULL)
296211834STom.Erickson@Sun.COM 				*cp = '\0';
296311834STom.Erickson@Sun.COM 			sendfs = nonpackage_sendfs;
296411834STom.Erickson@Sun.COM 		}
29655367Sahrens 		return (zfs_receive_one(hdl, infd, tosnap, flags,
296611834STom.Erickson@Sun.COM 		    &drr, &drr_noswap, sendfs, stream_nv, stream_avl,
296712527SChris.Kirby@oracle.com 		    top_zfs, cleanup_fd, action_handlep));
296811976STom.Erickson@Sun.COM 	} else {
296911007SLori.Alt@Sun.COM 		assert(DMU_GET_STREAM_HDRTYPE(drrb->drr_versioninfo) ==
297011007SLori.Alt@Sun.COM 		    DMU_COMPOUNDSTREAM);
29715367Sahrens 		return (zfs_receive_package(hdl, infd, tosnap, flags,
297212527SChris.Kirby@oracle.com 		    &drr, &zcksum, top_zfs, cleanup_fd, action_handlep));
29735367Sahrens 	}
29745367Sahrens }
29757366STim.Haley@Sun.COM 
29767366STim.Haley@Sun.COM /*
29777366STim.Haley@Sun.COM  * Restores a backup of tosnap from the file descriptor specified by infd.
29787366STim.Haley@Sun.COM  * Return 0 on total success, -2 if some things couldn't be
29797366STim.Haley@Sun.COM  * destroyed/renamed/promoted, -1 if some things couldn't be received.
29807366STim.Haley@Sun.COM  * (-1 will override -2).
29817366STim.Haley@Sun.COM  */
29827366STim.Haley@Sun.COM int
zfs_receive(libzfs_handle_t * hdl,const char * tosnap,recvflags_t flags,int infd,avl_tree_t * stream_avl)29837366STim.Haley@Sun.COM zfs_receive(libzfs_handle_t *hdl, const char *tosnap, recvflags_t flags,
29847366STim.Haley@Sun.COM     int infd, avl_tree_t *stream_avl)
29857366STim.Haley@Sun.COM {
29867366STim.Haley@Sun.COM 	char *top_zfs = NULL;
29877366STim.Haley@Sun.COM 	int err;
298812527SChris.Kirby@oracle.com 	int cleanup_fd;
298912527SChris.Kirby@oracle.com 	uint64_t action_handle = 0;
299012527SChris.Kirby@oracle.com 
299112527SChris.Kirby@oracle.com 	cleanup_fd = open(ZFS_DEV, O_RDWR|O_EXCL);
299212527SChris.Kirby@oracle.com 	VERIFY(cleanup_fd >= 0);
29937366STim.Haley@Sun.COM 
299411834STom.Erickson@Sun.COM 	err = zfs_receive_impl(hdl, tosnap, flags, infd, NULL, NULL,
299512527SChris.Kirby@oracle.com 	    stream_avl, &top_zfs, cleanup_fd, &action_handle);
299612527SChris.Kirby@oracle.com 
299712527SChris.Kirby@oracle.com 	VERIFY(0 == close(cleanup_fd));
29987366STim.Haley@Sun.COM 
29998584SLori.Alt@Sun.COM 	if (err == 0 && !flags.nomount && top_zfs) {
30007366STim.Haley@Sun.COM 		zfs_handle_t *zhp;
30017366STim.Haley@Sun.COM 		prop_changelist_t *clp;
30027366STim.Haley@Sun.COM 
30037366STim.Haley@Sun.COM 		zhp = zfs_open(hdl, top_zfs, ZFS_TYPE_FILESYSTEM);
30047366STim.Haley@Sun.COM 		if (zhp != NULL) {
30057366STim.Haley@Sun.COM 			clp = changelist_gather(zhp, ZFS_PROP_MOUNTPOINT,
30067366STim.Haley@Sun.COM 			    CL_GATHER_MOUNT_ALWAYS, 0);
30077366STim.Haley@Sun.COM 			zfs_close(zhp);
30087366STim.Haley@Sun.COM 			if (clp != NULL) {
30097366STim.Haley@Sun.COM 				/* mount and share received datasets */
30107366STim.Haley@Sun.COM 				err = changelist_postfix(clp);
30117366STim.Haley@Sun.COM 				changelist_free(clp);
30127366STim.Haley@Sun.COM 			}
30137366STim.Haley@Sun.COM 		}
30147366STim.Haley@Sun.COM 		if (zhp == NULL || clp == NULL || err)
30157366STim.Haley@Sun.COM 			err = -1;
30167366STim.Haley@Sun.COM 	}
30177366STim.Haley@Sun.COM 	if (top_zfs)
30187366STim.Haley@Sun.COM 		free(top_zfs);
30197366STim.Haley@Sun.COM 
30207366STim.Haley@Sun.COM 	return (err);
30217366STim.Haley@Sun.COM }
3022