/*

 * CDDL HEADER START

 *

 * The contents of this file are subject to the terms of the

 * Common Development and Distribution License (the "License").

 * You may not use this file except in compliance with the License.

 *

 * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE

 * or http://www.opensolaris.org/os/licensing.

 * See the License for the specific language governing permissions

 * and limitations under the License.

 *

 * When distributing Covered Code, include this CDDL HEADER in each

 * file and include the License file at usr/src/OPENSOLARIS.LICENSE.

 * If applicable, add the following below this CDDL HEADER, with the

 * fields enclosed by brackets "[]" replaced with your own identifying

 * information: Portions Copyright [yyyy] [name of copyright owner]

 *

 * CDDL HEADER END

 */

/*

 * Copyright 2007 Sun Microsystems, Inc.  All rights reserved.

 * Use is subject to license terms.

 */

#pragma ident	"%Z%%M%	%I%	%E% SMI"

#include <sys/dmu.h>

#include <sys/dmu_impl.h>

#include <sys/dmu_tx.h>

#include <sys/dbuf.h>

#include <sys/dnode.h>

#include <sys/zfs_context.h>

#include <sys/dmu_objset.h>

#include <sys/dmu_traverse.h>

#include <sys/dsl_dataset.h>

#include <sys/dsl_dir.h>

#include <sys/dsl_pool.h>

#include <sys/dsl_synctask.h>

#include <sys/zfs_ioctl.h>

#include <sys/zap.h>

#include <sys/zio_checksum.h>

struct backuparg {

	dmu_replay_record_t *drr;

	vnode_t *vp;

	objset_t *os;

	zio_cksum_t zc;

	int err;

};

static int

dump_bytes(struct backuparg *ba, void *buf, int len)

{

	ssize_t resid; /* have to get resid to get detailed errno */

	ASSERT3U(len % 8, ==, 0);

	fletcher_4_incremental_native(buf, len, &ba->zc);

	ba->err = vn_rdwr(UIO_WRITE, ba->vp,

	    (caddr_t)buf, len,

	    0, UIO_SYSSPACE, FAPPEND, RLIM64_INFINITY, CRED(), &resid);

	return (ba->err);

}

static int

dump_free(struct backuparg *ba, uint64_t object, uint64_t offset,

    uint64_t length)

{

	/* write a FREE record */

	bzero(ba->drr, sizeof (dmu_replay_record_t));

	ba->drr->drr_type = DRR_FREE;

	ba->drr->drr_u.drr_free.drr_object = object;

	ba->drr->drr_u.drr_free.drr_offset = offset;

	ba->drr->drr_u.drr_free.drr_length = length;

	if (dump_bytes(ba, ba->drr, sizeof (dmu_replay_record_t)))

		return (EINTR);

	return (0);

}

static int

dump_data(struct backuparg *ba, dmu_object_type_t type,

    uint64_t object, uint64_t offset, int blksz, void *data)

{

	/* write a DATA record */

	bzero(ba->drr, sizeof (dmu_replay_record_t));

	ba->drr->drr_type = DRR_WRITE;

	ba->drr->drr_u.drr_write.drr_object = object;

	ba->drr->drr_u.drr_write.drr_type = type;

	ba->drr->drr_u.drr_write.drr_offset = offset;

	ba->drr->drr_u.drr_write.drr_length = blksz;

	if (dump_bytes(ba, ba->drr, sizeof (dmu_replay_record_t)))

		return (EINTR);

	if (dump_bytes(ba, data, blksz))

		return (EINTR);

	return (0);

}

static int

dump_freeobjects(struct backuparg *ba, uint64_t firstobj, uint64_t numobjs)

{

	/* write a FREEOBJECTS record */

	bzero(ba->drr, sizeof (dmu_replay_record_t));

	ba->drr->drr_type = DRR_FREEOBJECTS;

	ba->drr->drr_u.drr_freeobjects.drr_firstobj = firstobj;

	ba->drr->drr_u.drr_freeobjects.drr_numobjs = numobjs;

	if (dump_bytes(ba, ba->drr, sizeof (dmu_replay_record_t)))

		return (EINTR);

	return (0);

}

static int

dump_dnode(struct backuparg *ba, uint64_t object, dnode_phys_t *dnp)

{

	if (dnp == NULL || dnp->dn_type == DMU_OT_NONE)

		return (dump_freeobjects(ba, object, 1));

	/* write an OBJECT record */

	bzero(ba->drr, sizeof (dmu_replay_record_t));

	ba->drr->drr_type = DRR_OBJECT;

	ba->drr->drr_u.drr_object.drr_object = object;

	ba->drr->drr_u.drr_object.drr_type = dnp->dn_type;

	ba->drr->drr_u.drr_object.drr_bonustype = dnp->dn_bonustype;

	ba->drr->drr_u.drr_object.drr_blksz =

	    dnp->dn_datablkszsec << SPA_MINBLOCKSHIFT;

	ba->drr->drr_u.drr_object.drr_bonuslen = dnp->dn_bonuslen;

	ba->drr->drr_u.drr_object.drr_checksum = dnp->dn_checksum;

	ba->drr->drr_u.drr_object.drr_compress = dnp->dn_compress;

	if (dump_bytes(ba, ba->drr, sizeof (dmu_replay_record_t)))

		return (EINTR);

	if (dump_bytes(ba, DN_BONUS(dnp), P2ROUNDUP(dnp->dn_bonuslen, 8)))

		return (EINTR);

	/* free anything past the end of the file */

	if (dump_free(ba, object, (dnp->dn_maxblkid + 1) *

	    (dnp->dn_datablkszsec << SPA_MINBLOCKSHIFT), -1ULL))

		return (EINTR);

	if (ba->err)

		return (EINTR);

	return (0);

}

#define	BP_SPAN(dnp, level) \

	(((uint64_t)dnp->dn_datablkszsec) << (SPA_MINBLOCKSHIFT + \

	(level) * (dnp->dn_indblkshift - SPA_BLKPTRSHIFT)))

static int

backup_cb(traverse_blk_cache_t *bc, spa_t *spa, void *arg)

{

	struct backuparg *ba = arg;

	uint64_t object = bc->bc_bookmark.zb_object;

	int level = bc->bc_bookmark.zb_level;

	uint64_t blkid = bc->bc_bookmark.zb_blkid;

	blkptr_t *bp = bc->bc_blkptr.blk_birth ? &bc->bc_blkptr : NULL;

	dmu_object_type_t type = bp ? BP_GET_TYPE(bp) : DMU_OT_NONE;

	void *data = bc->bc_data;

	int err = 0;

	if (issig(JUSTLOOKING) && issig(FORREAL))

		return (EINTR);

	ASSERT(data || bp == NULL);

	if (bp == NULL && object == 0) {

		uint64_t span = BP_SPAN(bc->bc_dnode, level);

		uint64_t dnobj = (blkid * span) >> DNODE_SHIFT;

		err = dump_freeobjects(ba, dnobj, span >> DNODE_SHIFT);

	} else if (bp == NULL) {

		uint64_t span = BP_SPAN(bc->bc_dnode, level);

		err = dump_free(ba, object, blkid * span, span);

	} else if (data && level == 0 && type == DMU_OT_DNODE) {

		dnode_phys_t *blk = data;

		int i;

		int blksz = BP_GET_LSIZE(bp);

		for (i = 0; i < blksz >> DNODE_SHIFT; i++) {

			uint64_t dnobj =

			    (blkid << (DNODE_BLOCK_SHIFT - DNODE_SHIFT)) + i;

			err = dump_dnode(ba, dnobj, blk+i);

			if (err)

				break;

		}

	} else if (level == 0 &&

	    type != DMU_OT_DNODE && type != DMU_OT_OBJSET) {

		int blksz = BP_GET_LSIZE(bp);

		if (data == NULL) {

			uint32_t aflags = ARC_WAIT;

			arc_buf_t *abuf;

			zbookmark_t zb;

			zb.zb_objset = ba->os->os->os_dsl_dataset->ds_object;

			zb.zb_object = object;

			zb.zb_level = level;

			zb.zb_blkid = blkid;

			(void) arc_read(NULL, spa, bp,

			    dmu_ot[type].ot_byteswap, arc_getbuf_func, &abuf,

			    ZIO_PRIORITY_ASYNC_READ, ZIO_FLAG_MUSTSUCCEED,

			    &aflags, &zb);

			if (abuf) {

				err = dump_data(ba, type, object, blkid * blksz,

				    blksz, abuf->b_data);

				(void) arc_buf_remove_ref(abuf, &abuf);

			}

		} else {

			err = dump_data(ba, type, object, blkid * blksz,

			    blksz, data);

		}

	}

	ASSERT(err == 0 || err == EINTR);

	return (err);

}

int

dmu_sendbackup(objset_t *tosnap, objset_t *fromsnap, vnode_t *vp)

{

	dsl_dataset_t *ds = tosnap->os->os_dsl_dataset;

	dsl_dataset_t *fromds = fromsnap ? fromsnap->os->os_dsl_dataset : NULL;

	dmu_replay_record_t *drr;

	struct backuparg ba;

	int err;

	/* tosnap must be a snapshot */

	if (ds->ds_phys->ds_next_snap_obj == 0)

		return (EINVAL);

	/* fromsnap must be an earlier snapshot from the same fs as tosnap */

	if (fromds && (ds->ds_dir != fromds->ds_dir ||

	    fromds->ds_phys->ds_creation_txg >=

	    ds->ds_phys->ds_creation_txg))

		return (EXDEV);

	drr = kmem_zalloc(sizeof (dmu_replay_record_t), KM_SLEEP);

	drr->drr_type = DRR_BEGIN;

	drr->drr_u.drr_begin.drr_magic = DMU_BACKUP_MAGIC;

	drr->drr_u.drr_begin.drr_version = DMU_BACKUP_VERSION;

	drr->drr_u.drr_begin.drr_creation_time =

	    ds->ds_phys->ds_creation_time;

	drr->drr_u.drr_begin.drr_type = tosnap->os->os_phys->os_type;

	drr->drr_u.drr_begin.drr_toguid = ds->ds_phys->ds_guid;

	if (fromds)

		drr->drr_u.drr_begin.drr_fromguid = fromds->ds_phys->ds_guid;

	dsl_dataset_name(ds, drr->drr_u.drr_begin.drr_toname);

	ba.drr = drr;

	ba.vp = vp;

	ba.os = tosnap;

	ZIO_SET_CHECKSUM(&ba.zc, 0, 0, 0, 0);

	if (dump_bytes(&ba, drr, sizeof (dmu_replay_record_t))) {

		kmem_free(drr, sizeof (dmu_replay_record_t));

		return (ba.err);

	}

	err = traverse_dsl_dataset(ds,

	    fromds ? fromds->ds_phys->ds_creation_txg : 0,

	    ADVANCE_PRE | ADVANCE_HOLES | ADVANCE_DATA | ADVANCE_NOLOCK,

	    backup_cb, &ba);

	if (err) {

		if (err == EINTR && ba.err)

			err = ba.err;

		kmem_free(drr, sizeof (dmu_replay_record_t));

		return (err);

	}

	bzero(drr, sizeof (dmu_replay_record_t));

	drr->drr_type = DRR_END;

	drr->drr_u.drr_end.drr_checksum = ba.zc;

	if (dump_bytes(&ba, drr, sizeof (dmu_replay_record_t))) {

		kmem_free(drr, sizeof (dmu_replay_record_t));

		return (ba.err);

	}

	kmem_free(drr, sizeof (dmu_replay_record_t));

	return (0);

}

struct restorearg {

	int err;

	int byteswap;

	vnode_t *vp;

	char *buf;

	uint64_t voff;

	int buflen; /* number of valid bytes in buf */

	int bufoff; /* next offset to read */

	int bufsize; /* amount of memory allocated for buf */

	zio_cksum_t zc;

};

static int

{

	const char *snapname;

	int err;

	uint64_t val;

	/* must already be a snapshot of this fs */

	if (ds->ds_phys->ds_prev_snap_obj == 0)

		return (ENODEV);

	/* most recent snapshot must match fromguid */

	if (ds->ds_prev->ds_phys->ds_guid != drrb->drr_fromguid)

		return (ENODEV);

	/* new snapshot name must not exist */

	snapname = strrchr(drrb->drr_toname, '@');

	if (snapname == NULL)

		return (EEXIST);

	snapname++;

	err = zap_lookup(ds->ds_dir->dd_pool->dp_meta_objset,

	    ds->ds_phys->ds_snapnames_zapobj, snapname, 8, 1, &val);

	if (err == 0)

	if (err != ENOENT)

	return (0);

}

/* ARGSUSED */

static void

{

	dsl_dataset_t *ds = arg1;

	dmu_buf_will_dirty(ds->ds_dbuf, tx);

	ds->ds_phys->ds_flags |= DS_FLAG_INCONSISTENT;

	spa_history_internal_log(LOG_DS_REPLAY_INC_SYNC,

	    ds->ds_dir->dd_pool->dp_spa, tx, cr, "dataset = %lld",

	    ds->ds_phys->ds_dir_obj);

}

/* ARGSUSED */

static int

replay_full_check(void *arg1, void *arg2, dmu_tx_t *tx)

{

	dsl_dir_t *dd = arg1;

	struct drr_begin *drrb = arg2;

	objset_t *mos = dd->dd_pool->dp_meta_objset;

	char *cp;

	uint64_t val;

	int err;

	cp = strchr(drrb->drr_toname, '@');

	*cp = '\0';

	err = zap_lookup(mos, dd->dd_phys->dd_child_dir_zapobj,

	    strrchr(drrb->drr_toname, '/') + 1,

	    sizeof (uint64_t), 1, &val);

	*cp = '@';

	if (err != ENOENT)

		return (err ? err : EEXIST);

	return (0);

}

static void

replay_full_sync(void *arg1, void *arg2, cred_t *cr, dmu_tx_t *tx)

{

	dsl_dir_t *dd = arg1;

	struct drr_begin *drrb = arg2;

	char *cp;

	dsl_dataset_t *ds;

	uint64_t dsobj;

	cp = strchr(drrb->drr_toname, '@');

	*cp = '\0';

	dsobj = dsl_dataset_create_sync(dd, strrchr(drrb->drr_toname, '/') + 1,

	    NULL, tx);

	VERIFY(0 == dsl_dataset_open_obj(dd->dd_pool, dsobj, NULL,

	    DS_MODE_EXCLUSIVE, FTAG, &ds));

	(void) dmu_objset_create_impl(dsl_dataset_get_spa(ds),

	    ds, &ds->ds_phys->ds_bp, drrb->drr_type, tx);

	dmu_buf_will_dirty(ds->ds_dbuf, tx);

	ds->ds_phys->ds_flags |= DS_FLAG_INCONSISTENT;

	spa_history_internal_log(LOG_DS_REPLAY_FULL_SYNC,

	    ds->ds_dir->dd_pool->dp_spa, tx, cr, "dataset = %lld",

	    ds->ds_phys->ds_dir_obj);

	*cp = '@';

	dsl_dataset_close(ds, DS_MODE_EXCLUSIVE, FTAG);

}

static int

replay_end_check(void *arg1, void *arg2, dmu_tx_t *tx)

{

	objset_t *os = arg1;

	struct drr_begin *drrb = arg2;

	char *snapname;

	/* XXX verify that drr_toname is in dd */

	snapname = strchr(drrb->drr_toname, '@');

	if (snapname == NULL)

		return (EINVAL);

	snapname++;

	return (dsl_dataset_snapshot_check(os, snapname, tx));

}

static void

replay_end_sync(void *arg1, void *arg2, cred_t *cr, dmu_tx_t *tx)

{

	objset_t *os = arg1;

	struct drr_begin *drrb = arg2;

	char *snapname;

	dsl_dataset_t *ds, *hds;

	snapname = strchr(drrb->drr_toname, '@') + 1;

	dsl_dataset_snapshot_sync(os, snapname, cr, tx);

	/* set snapshot's creation time and guid */

	hds = os->os->os_dsl_dataset;

	VERIFY(0 == dsl_dataset_open_obj(hds->ds_dir->dd_pool,

	    hds->ds_phys->ds_prev_snap_obj, NULL,

	    DS_MODE_PRIMARY | DS_MODE_READONLY | DS_MODE_INCONSISTENT,

	    FTAG, &ds));

	dmu_buf_will_dirty(ds->ds_dbuf, tx);

	ds->ds_phys->ds_creation_time = drrb->drr_creation_time;

	ds->ds_phys->ds_guid = drrb->drr_toguid;

	ds->ds_phys->ds_flags &= ~DS_FLAG_INCONSISTENT;

	/* log the end of the receive */

	spa_history_internal_log(LOG_DS_RECEIVE, ds->ds_dir->dd_pool->dp_spa,

	    tx, cr, "dataset = %llu", ds->ds_phys->ds_dir_obj);

	dsl_dataset_close(ds, DS_MODE_PRIMARY, FTAG);

	dmu_buf_will_dirty(hds->ds_dbuf, tx);

	hds->ds_phys->ds_flags &= ~DS_FLAG_INCONSISTENT;

}

static void *

restore_read(struct restorearg *ra, int len)

{

	void *rv;

	/* some things will require 8-byte alignment, so everything must */