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