#include <sys/debug.h>
#include <sys/types.h>
#include <sys/file.h>
#include <sys/errno.h>
#include <sys/uio.h>
#include <sys/open.h>
#include <sys/cred.h>
#include <sys/kmem.h>
#include <sys/conf.h>
#include <sys/cmn_err.h>
#include <sys/modctl.h>
#include <sys/disp.h>
#include <sys/atomic.h>
#include <sys/filio.h>
#include <sys/stat.h>
#include <sys/kstat.h>
#include <sys/ddi.h>
#include <sys/devops.h>
#include <sys/sunddi.h>
#include <sys/esunddi.h>
#include <sys/priv_names.h>
#include <sys/fssnap.h>
#include <sys/fssnap_if.h>
static dev_info_t *fssnap_dip = NULL;
static struct snapshot_id *snapshot = NULL;
static struct snapshot_id snap_ctl;
static int num_snapshots = 0;
static kmutex_t snapshot_mutex;
static char snapname[] = SNAP_NAME;
static int fssnap_taskq_nthreads = FSSNAP_TASKQ_THREADS;
static uint_t fssnap_max_mem_chunks = FSSNAP_MAX_MEM_CHUNKS;
static int fssnap_taskq_maxtasks = FSSNAP_TASKQ_MAXTASKS;
static int snap_getinfo(dev_info_t *, ddi_info_cmd_t, void *, void **);
static int snap_attach(dev_info_t *dip, ddi_attach_cmd_t cmd);
static int snap_detach(dev_info_t *dip, ddi_detach_cmd_t cmd);
static int snap_open(dev_t *devp, int flag, int otyp, cred_t *cred);
static int snap_close(dev_t dev, int flag, int otyp, cred_t *cred);
static int snap_strategy(struct buf *bp);
static int snap_read(dev_t dev, struct uio *uiop, cred_t *credp);
static int snap_print(dev_t dev, char *str);
static int snap_ioctl(dev_t dev, int cmd, intptr_t arg, int mode,
cred_t *credp, int *rvalp);
static int snap_prop_op(dev_t dev, dev_info_t *dip, ddi_prop_op_t prop_op,
int flags, char *name, caddr_t valuep, int *lengthp);
static int snap_getchunk(struct snapshot_id *sidp, chunknumber_t chunk,
int offset, int len, char *buffer);
static void fssnap_strategy_impl(void *, struct buf *);
static void *fssnap_create_impl(chunknumber_t, uint_t, u_offset_t,
struct vnode *, int, struct vnode **, char *, u_offset_t);
static void fssnap_set_candidate_impl(void *, chunknumber_t);
static int fssnap_is_candidate_impl(void *, u_offset_t);
static int fssnap_create_done_impl(void *);
static int fssnap_delete_impl(void *);
static int fssnap_translate(struct snapshot_id **, struct buf *);
static void fssnap_write_taskq(void *);
static void fssnap_create_kstats(snapshot_id_t *, int, const char *,
const char *);
static int fssnap_update_kstat_num(kstat_t *, int);
static void fssnap_delete_kstats(struct cow_info *);
static cow_map_node_t *transtbl_add(cow_map_t *, chunknumber_t, caddr_t);
static cow_map_node_t *transtbl_get(cow_map_t *, chunknumber_t);
static void transtbl_delete(cow_map_t *, cow_map_node_t *);
static void transtbl_free(cow_map_t *);
static kstat_t *fssnap_highwater_kstat;
static struct cb_ops snap_cb_ops = {
snap_open,
snap_close,
snap_strategy,
snap_print,
nodev,
snap_read,
nodev,
snap_ioctl,
nodev,
nodev,
nodev,
nochpoll,
snap_prop_op,
NULL,
D_64BIT | D_NEW | D_MP,
CB_REV,
nodev,
nodev
};
static struct dev_ops snap_ops = {
DEVO_REV,
0,
snap_getinfo,
nulldev,
nulldev,
snap_attach,
snap_detach,
nodev,
&snap_cb_ops,
(struct bus_ops *)NULL,
nulldev,
ddi_quiesce_not_needed,
};
extern struct mod_ops mod_driverops;
static struct modldrv md = {
&mod_driverops,
"snapshot driver",
&snap_ops,
};
static struct modlinkage ml = {
MODREV_1,
&md,
NULL
};
static void *statep;
int
_init(void)
{
int error;
kstat_t *ksp;
kstat_named_t *ksdata;
error = ddi_soft_state_init(&statep, sizeof (struct snapshot_id *), 1);
if (error) {
cmn_err(CE_WARN, "_init: failed to init ddi_soft_state.");
return (error);
}
error = mod_install(&ml);
if (error) {
cmn_err(CE_WARN, "_init: failed to mod_install.");
ddi_soft_state_fini(&statep);
return (error);
}
snapops.fssnap_create = fssnap_create_impl;
snapops.fssnap_set_candidate = fssnap_set_candidate_impl;
snapops.fssnap_is_candidate = fssnap_is_candidate_impl;
snapops.fssnap_create_done = fssnap_create_done_impl;
snapops.fssnap_delete = fssnap_delete_impl;
snapops.fssnap_strategy = fssnap_strategy_impl;
mutex_init(&snapshot_mutex, NULL, MUTEX_DEFAULT, NULL);
ksp = kstat_create(snapname, 0, FSSNAP_KSTAT_HIGHWATER, "misc",
KSTAT_TYPE_NAMED, 1, 0);
if (ksp != NULL) {
ksdata = (kstat_named_t *)ksp->ks_data;
kstat_named_init(ksdata, FSSNAP_KSTAT_HIGHWATER,
KSTAT_DATA_UINT32);
ksdata->value.ui32 = 0;
kstat_install(ksp);
} else {
cmn_err(CE_WARN, "_init: failed to create highwater kstat.");
}
fssnap_highwater_kstat = ksp;
return (0);
}
int
_info(struct modinfo *modinfop)
{
return (mod_info(&ml, modinfop));
}
int
_fini(void)
{
int error;
error = mod_remove(&ml);
if (error)
return (error);
ddi_soft_state_fini(&statep);
kstat_delete(fssnap_highwater_kstat);
mutex_destroy(&snapshot_mutex);
snapops.fssnap_create = NULL;
snapops.fssnap_set_candidate = NULL;
snapops.fssnap_create_done = NULL;
snapops.fssnap_delete = NULL;
snapops.fssnap_strategy = NULL;
return (0);
}
static int
snap_getinfo(dev_info_t *dip, ddi_info_cmd_t infocmd, void *arg, void **result)
{
switch (infocmd) {
case DDI_INFO_DEVT2DEVINFO:
*result = fssnap_dip;
return (DDI_SUCCESS);
case DDI_INFO_DEVT2INSTANCE:
*result = 0;
return (DDI_SUCCESS);
}
return (DDI_FAILURE);
}
static int
snap_attach(dev_info_t *dip, ddi_attach_cmd_t cmd)
{
int error;
switch (cmd) {
case DDI_ATTACH:
error = ddi_create_priv_minor_node(dip, SNAP_CTL_NODE, S_IFCHR,
SNAP_CTL_MINOR, DDI_PSEUDO, PRIVONLY_DEV,
PRIV_SYS_CONFIG, PRIV_SYS_CONFIG, 0666);
if (error == DDI_FAILURE) {
return (DDI_FAILURE);
}
rw_init(&snap_ctl.sid_rwlock, NULL, RW_DEFAULT, NULL);
rw_enter(&snap_ctl.sid_rwlock, RW_WRITER);
fssnap_dip = dip;
snap_ctl.sid_snapnumber = SNAP_CTL_MINOR;
snap_ctl.sid_next = NULL;
snap_ctl.sid_cowinfo = NULL;
snap_ctl.sid_flags = 0;
rw_exit(&snap_ctl.sid_rwlock);
ddi_report_dev(dip);
return (DDI_SUCCESS);
case DDI_PM_RESUME:
return (DDI_SUCCESS);
case DDI_RESUME:
return (DDI_SUCCESS);
default:
return (DDI_FAILURE);
}
}
static int
snap_detach(dev_info_t *dip, ddi_detach_cmd_t cmd)
{
struct snapshot_id *sidp, *sidnextp;
switch (cmd) {
case DDI_DETACH:
mutex_enter(&snapshot_mutex);
if ((num_snapshots != 0) ||
((snap_ctl.sid_flags & SID_CHAR_BUSY) != 0)) {
mutex_exit(&snapshot_mutex);
return (DDI_FAILURE);
}
for (sidp = snapshot; sidp != NULL; sidp = sidnextp) {
ASSERT(SID_AVAILABLE(sidp) &&
!RW_LOCK_HELD(&sidp->sid_rwlock));
sidnextp = sidp->sid_next;
rw_destroy(&sidp->sid_rwlock);
kmem_free(sidp, sizeof (struct snapshot_id));
}
snapshot = NULL;
ddi_remove_minor_node(dip, SNAP_CTL_NODE);
fssnap_dip = NULL;
ASSERT((snap_ctl.sid_flags & SID_CHAR_BUSY) == 0);
rw_destroy(&snap_ctl.sid_rwlock);
mutex_exit(&snapshot_mutex);
return (DDI_SUCCESS);
default:
return (DDI_FAILURE);
}
}
static int
snap_open(dev_t *devp, int flag, int otyp, cred_t *cred)
{
minor_t minor;
struct snapshot_id **sidpp, *sidp;
if (flag & FWRITE)
return (EROFS);
minor = getminor(*devp);
if (minor == SNAP_CTL_MINOR) {
if (((flag & FEXCL) != FEXCL) || (otyp != OTYP_CHR))
return (EINVAL);
rw_enter(&snap_ctl.sid_rwlock, RW_WRITER);
if ((snap_ctl.sid_flags & SID_CHAR_BUSY) != 0) {
rw_exit(&snap_ctl.sid_rwlock);
return (EBUSY);
}
snap_ctl.sid_flags |= SID_CHAR_BUSY;
rw_exit(&snap_ctl.sid_rwlock);
return (0);
}
sidpp = ddi_get_soft_state(statep, minor);
if (sidpp == NULL || *sidpp == NULL)
return (ENXIO);
sidp = *sidpp;
rw_enter(&sidp->sid_rwlock, RW_WRITER);
if ((flag & FEXCL) && SID_BUSY(sidp)) {
rw_exit(&sidp->sid_rwlock);
return (EAGAIN);
}
ASSERT(sidpp != NULL && sidp != NULL);
if (SID_INACTIVE(sidp)) {
cmn_err(CE_WARN, "snap_open: snapshot %d does not exist.",
minor);
rw_exit(&sidp->sid_rwlock);
return (ENXIO);
}
switch (otyp) {
case OTYP_CHR:
sidp->sid_flags |= SID_CHAR_BUSY;
break;
case OTYP_BLK:
sidp->sid_flags |= SID_BLOCK_BUSY;
break;
default:
rw_exit(&sidp->sid_rwlock);
return (EINVAL);
}
rw_exit(&sidp->sid_rwlock);
return (0);
}
static int
snap_close(dev_t dev, int flag, int otyp, cred_t *cred)
{
struct snapshot_id **sidpp, *sidp;
minor_t minor;
char name[20];
minor = getminor(dev);
if (minor == SNAP_CTL_MINOR) {
rw_enter(&snap_ctl.sid_rwlock, RW_WRITER);
snap_ctl.sid_flags &= ~(SID_CHAR_BUSY);
rw_exit(&snap_ctl.sid_rwlock);
return (0);
}
sidpp = ddi_get_soft_state(statep, minor);
if (sidpp == NULL || *sidpp == NULL) {
cmn_err(CE_WARN, "snap_close: could not find state for "
"snapshot %d.", minor);
return (ENXIO);
}
sidp = *sidpp;
mutex_enter(&snapshot_mutex);
rw_enter(&sidp->sid_rwlock, RW_WRITER);
switch (otyp) {
case OTYP_CHR:
sidp->sid_flags &= ~(SID_CHAR_BUSY);
break;
case OTYP_BLK:
sidp->sid_flags &= ~(SID_BLOCK_BUSY);
break;
default:
mutex_exit(&snapshot_mutex);
rw_exit(&sidp->sid_rwlock);
return (EINVAL);
}
if (SID_AVAILABLE(sidp)) {
ASSERT(fssnap_dip != NULL);
(void) snprintf(name, sizeof (name), "%d",
sidp->sid_snapnumber);
ddi_remove_minor_node(fssnap_dip, name);
(void) snprintf(name, sizeof (name), "%d,raw",
sidp->sid_snapnumber);
ddi_remove_minor_node(fssnap_dip, name);
ddi_soft_state_free(statep, sidp->sid_snapnumber);
num_snapshots--;
}
mutex_exit(&snapshot_mutex);
rw_exit(&sidp->sid_rwlock);
return (0);
}
static int
snap_read(dev_t dev, struct uio *uiop, cred_t *credp)
{
minor_t minor;
struct snapshot_id **sidpp;
minor = getminor(dev);
sidpp = ddi_get_soft_state(statep, minor);
if (sidpp == NULL || *sidpp == NULL) {
cmn_err(CE_WARN,
"snap_read: could not find state for snapshot %d.", minor);
return (ENXIO);
}
return (physio(snap_strategy, NULL, dev, B_READ, minphys, uiop));
}
static int
snap_strategy(struct buf *bp)
{
struct snapshot_id **sidpp, *sidp;
minor_t minor;
chunknumber_t chunk;
int off, len;
u_longlong_t reqptr;
int error = 0;
size_t chunksz;
caddr_t buf;
if (bp->b_flags & B_WRITE) {
bioerror(bp, EROFS);
bp->b_resid = bp->b_bcount;
biodone(bp);
return (0);
}
minor = getminor(bp->b_edev);
sidpp = ddi_get_soft_state(statep, minor);
if (sidpp == NULL || *sidpp == NULL) {
cmn_err(CE_WARN,
"snap_strategy: could not find state for snapshot %d.",
minor);
bioerror(bp, ENXIO);
bp->b_resid = bp->b_bcount;
biodone(bp);
return (0);
}
sidp = *sidpp;
ASSERT(sidp);
rw_enter(&sidp->sid_rwlock, RW_READER);
if (SID_INACTIVE(sidp)) {
bioerror(bp, ENXIO);
bp->b_resid = bp->b_bcount;
biodone(bp);
rw_exit(&sidp->sid_rwlock);
return (0);
}
if (bp->b_flags & (B_PAGEIO|B_PHYS))
bp_mapin(bp);
bp->b_resid = bp->b_bcount;
ASSERT(bp->b_un.b_addr);
buf = bp->b_un.b_addr;
chunksz = sidp->sid_cowinfo->cow_map.cmap_chunksz;
reqptr = bp->b_lblkno;
while (bp->b_resid > 0) {
chunk = dbtocowchunk(&sidp->sid_cowinfo->cow_map, reqptr);
off = (reqptr % (chunksz >> DEV_BSHIFT)) << DEV_BSHIFT;
len = min(chunksz - off, bp->b_resid);
ASSERT((off + len) <= chunksz);
if ((error = snap_getchunk(sidp, chunk, off, len, buf)) != 0) {
if (error != EINVAL) {
cmn_err(CE_WARN, "snap_strategy: error "
"calling snap_getchunk, chunk = %llu, "
"offset = %d, len = %d, resid = %lu, "
"error = %d.",
chunk, off, len, bp->b_resid, error);
}
bioerror(bp, error);
biodone(bp);
rw_exit(&sidp->sid_rwlock);
return (0);
}
bp->b_resid -= len;
reqptr += (len >> DEV_BSHIFT);
buf += len;
}
ASSERT(bp->b_resid == 0);
biodone(bp);
rw_exit(&sidp->sid_rwlock);
return (0);
}
static int
snap_getchunk(struct snapshot_id *sidp, chunknumber_t chunk, int offset,
int len, char *buffer)
{
cow_map_t *cmap = &sidp->sid_cowinfo->cow_map;
cow_map_node_t *cmn;
struct buf *snapbuf;
int error = 0;
char *newbuffer;
int newlen = 0;
int partial = 0;
ASSERT(RW_READ_HELD(&sidp->sid_rwlock));
ASSERT(offset + len <= cmap->cmap_chunksz);
if (chunk >= (cmap->cmap_bmsize * NBBY)) {
return (EINVAL);
}
if (isclr(cmap->cmap_candidate, chunk)) {
bzero(buffer, len);
return (0);
}
rw_enter(&cmap->cmap_rwlock, RW_READER);
if (isclr(cmap->cmap_hastrans, chunk)) {
snapbuf = getrbuf(KM_SLEEP);
partial = len % DEV_BSIZE;
snapbuf->b_bcount = len;
snapbuf->b_lblkno = lbtodb(chunk * cmap->cmap_chunksz + offset);
snapbuf->b_un.b_addr = buffer;
snapbuf->b_iodone = NULL;
snapbuf->b_proc = NULL;
snapbuf->b_flags = B_READ | B_BUSY;
snapbuf->b_edev = sidp->sid_fvp->v_vfsp->vfs_dev;
if (partial) {
newlen = (len - partial) + DEV_BSIZE;
newbuffer = kmem_alloc(newlen, KM_SLEEP);
snapbuf->b_bcount = newlen;
snapbuf->b_un.b_addr = newbuffer;
}
(void) bdev_strategy(snapbuf);
(void) biowait(snapbuf);
error = geterror(snapbuf);
if (partial) {
bcopy(newbuffer, buffer, len);
kmem_free(newbuffer, newlen);
}
freerbuf(snapbuf);
rw_exit(&cmap->cmap_rwlock);
return (error);
}
cmn = transtbl_get(cmap, chunk);
if (cmn != NULL) {
ASSERT(cmn->cmn_buf);
bcopy(cmn->cmn_buf + offset, buffer, len);
rw_exit(&cmap->cmap_rwlock);
} else {
ssize_t resid = len;
int bf_index;
rw_exit(&cmap->cmap_rwlock);
bf_index = chunk / cmap->cmap_chunksperbf;
error = vn_rdwr(UIO_READ,
(sidp->sid_cowinfo->cow_backfile_array)[bf_index],
buffer, len, ((chunk % cmap->cmap_chunksperbf) *
cmap->cmap_chunksz) + offset, UIO_SYSSPACE, 0,
RLIM64_INFINITY, kcred, &resid);
}
return (error);
}
static int
snap_print(dev_t dev, char *str)
{
struct snapshot_id **sidpp;
minor_t minor;
minor = getminor(dev);
sidpp = ddi_get_soft_state(statep, minor);
if (sidpp == NULL || *sidpp == NULL) {
cmn_err(CE_WARN,
"snap_print: could not find state for snapshot %d.", minor);
return (ENXIO);
}
cmn_err(CE_NOTE, "snap_print: snapshot %d: %s", minor, str);
return (0);
}
static int
snap_prop_op(dev_t dev, dev_info_t *dip, ddi_prop_op_t prop_op,
int flags, char *name, caddr_t valuep, int *lengthp)
{
int minor;
struct snapshot_id **sidpp;
dev_t mdev;
dev_info_t *mdip;
int error;
minor = getminor(dev);
if ((minor == SNAP_CTL_MINOR) || (dev == DDI_DEV_T_ANY))
return (ddi_prop_op(dev, dip, prop_op, flags, name,
valuep, lengthp));
sidpp = ddi_get_soft_state(statep, minor);
if (sidpp == NULL || *sidpp == NULL) {
cmn_err(CE_WARN,
"snap_prop_op: could not find state for "
"snapshot %d.", minor);
return (DDI_PROP_NOT_FOUND);
}
if (((*sidpp)->sid_fvp == NULL) || ((*sidpp)->sid_fvp->v_vfsp == NULL))
return (ddi_prop_op(dev, dip, prop_op, flags, name,
valuep, lengthp));
mdev = (*sidpp)->sid_fvp->v_vfsp->vfs_dev;
if (mdip = e_ddi_hold_devi_by_dev(mdev, 0)) {
error = cdev_prop_op(mdev, mdip,
prop_op, flags, name, valuep, lengthp);
ddi_release_devi(mdip);
if (error == DDI_PROP_SUCCESS)
return (error);
}
return (ddi_prop_op(dev, dip, prop_op, flags, name, valuep, lengthp));
}
static int
snap_ioctl(dev_t dev, int cmd, intptr_t arg, int mode, cred_t *credp,
int *rvalp)
{
minor_t minor;
int error = 0;
minor = getminor(dev);
if (minor != SNAP_CTL_MINOR) {
return (EINVAL);
}
switch (cmd) {
case _FIOSNAPSHOTCREATE:
{
struct fiosnapcreate fc;
struct file *fp;
struct vnode *vp;
if (ddi_copyin((void *)arg, &fc, sizeof (fc), mode))
return (EFAULT);
if ((fp = getf(fc.rootfiledesc)) == NULL)
return (EBADF);
ASSERT(fp->f_vnode);
vp = fp->f_vnode;
VN_HOLD(vp);
releasef(fc.rootfiledesc);
error = VOP_IOCTL(vp, cmd, arg, 0, credp, rvalp, NULL);
VN_RELE(vp);
break;
}
case _FIOSNAPSHOTCREATE_MULTI:
{
struct fiosnapcreate_multi fc;
struct file *fp;
struct vnode *vp;
if (ddi_copyin((void *)arg, &fc, sizeof (fc), mode))
return (EFAULT);
if ((fp = getf(fc.rootfiledesc)) == NULL)
return (EBADF);
ASSERT(fp->f_vnode);
vp = fp->f_vnode;
VN_HOLD(vp);
releasef(fc.rootfiledesc);
error = VOP_IOCTL(vp, cmd, arg, 0, credp, rvalp, NULL);
VN_RELE(vp);
break;
}
case _FIOSNAPSHOTDELETE:
{
major_t major;
struct fiosnapdelete fc;
snapshot_id_t *sidp = NULL;
snapshot_id_t *sidnextp = NULL;
struct file *fp = NULL;
struct vnode *vp = NULL;
struct vfs *vfsp = NULL;
vfsops_t *vfsops = EIO_vfsops;
if (ddi_copyin((void *)arg, &fc, sizeof (fc), mode))
return (EFAULT);
if ((fp = getf(fc.rootfiledesc)) == NULL)
return (EBADF);
ASSERT(fp->f_vnode);
vp = fp->f_vnode;
VN_HOLD(vp);
releasef(fc.rootfiledesc);
ASSERT(fssnap_dip != NULL);
major = ddi_driver_major(fssnap_dip);
mutex_enter(&snapshot_mutex);
for (sidp = snapshot; sidp != NULL; sidp = sidnextp) {
rw_enter(&sidp->sid_rwlock, RW_READER);
sidnextp = sidp->sid_next;
if (major == getmajor(vp->v_rdev)) {
minor = getminor(vp->v_rdev);
if (sidp->sid_snapnumber == (uint_t)minor &&
sidp->sid_fvp) {
VN_RELE(vp);
vp = sidp->sid_fvp;
VN_HOLD(vp);
rw_exit(&sidp->sid_rwlock);
break;
}
} else {
if (sidp->sid_fvp == vp) {
minor = sidp->sid_snapnumber;
rw_exit(&sidp->sid_rwlock);
break;
}
}
rw_exit(&sidp->sid_rwlock);
}
mutex_exit(&snapshot_mutex);
if (minor == SNAP_CTL_MINOR) {
VN_RELE(vp);
return (EINVAL);
}
dev = makedevice(major, minor);
vfsp = vfs_alloc(KM_SLEEP);
VFS_INIT(vfsp, vfsops, NULL);
VFS_HOLD(vfsp);
vfs_addmip(dev, vfsp);
if ((vfs_devmounting(dev, vfsp)) ||
(vfs_devismounted(dev))) {
vfs_delmip(vfsp);
VFS_RELE(vfsp);
VN_RELE(vp);
return (EBUSY);
}
error = VOP_IOCTL(vp, cmd, arg, 0, credp, rvalp, NULL);
vfs_delmip(vfsp);
VFS_RELE(vfsp);
VN_RELE(vp);
break;
}
default:
cmn_err(CE_WARN, "snap_ioctl: Invalid ioctl cmd %d, minor %d.",
cmd, minor);
return (EINVAL);
}
return (error);
}
static cow_map_node_t *
transtbl_add(cow_map_t *cmap, chunknumber_t chunk, caddr_t buf)
{
cow_map_node_t *cmnode;
ASSERT(RW_WRITE_HELD(&cmap->cmap_rwlock));
cmnode = kmem_alloc(sizeof (cow_map_node_t), KM_SLEEP);
cmnode->cmn_chunk = chunk;
cmnode->cmn_buf = buf;
cmnode->cmn_prev = NULL;
cmnode->cmn_next = cmap->cmap_table;
if (cmnode->cmn_next)
cmnode->cmn_next->cmn_prev = cmnode;
cmap->cmap_table = cmnode;
return (cmnode);
}
static cow_map_node_t *
transtbl_get(cow_map_t *cmap, chunknumber_t chunk)
{
cow_map_node_t *cmn;
ASSERT(RW_READ_HELD(&cmap->cmap_rwlock));
ASSERT(cmap);
for (cmn = cmap->cmap_table; cmn != NULL; cmn = cmn->cmn_next) {
if (cmn->cmn_chunk == chunk)
return (cmn);
}
return (NULL);
}
static void
transtbl_delete(cow_map_t *cmap, cow_map_node_t *cmn)
{
ASSERT(RW_WRITE_HELD(&cmap->cmap_rwlock));
ASSERT(cmn);
ASSERT(cmap->cmap_table);
if (cmap->cmap_table == cmn) {
ASSERT(cmn->cmn_prev == NULL);
cmap->cmap_table = cmn->cmn_next;
}
if (cmn->cmn_prev != NULL) {
ASSERT(cmn->cmn_prev->cmn_next == cmn);
cmn->cmn_prev->cmn_next = cmn->cmn_next;
}
if (cmn->cmn_next != NULL) {
ASSERT(cmn->cmn_next->cmn_prev == cmn);
cmn->cmn_next->cmn_prev = cmn->cmn_prev;
}
ASSERT(cmn->cmn_buf);
kmem_free(cmn->cmn_buf, cmap->cmap_chunksz);
kmem_free(cmn, sizeof (cow_map_node_t));
}
static void
transtbl_free(cow_map_t *cmap)
{
cow_map_node_t *curnode;
cow_map_node_t *tempnode;
for (curnode = cmap->cmap_table; curnode != NULL; curnode = tempnode) {
tempnode = curnode->cmn_next;
kmem_free(curnode->cmn_buf, cmap->cmap_chunksz);
kmem_free(curnode, sizeof (cow_map_node_t));
}
}
static void
fssnap_strategy_impl(void *snapshot_id, buf_t *bp)
{
struct snapshot_id **sidpp;
struct snapshot_id *sidp;
int error;
if (bp->b_flags & B_READ) {
(void) bdev_strategy(bp);
return;
}
sidpp = (struct snapshot_id **)snapshot_id;
sidp = *sidpp;
if (sidp == NULL) {
(void) bdev_strategy(bp);
return;
}
rw_enter(&sidp->sid_rwlock, RW_READER);
if (sidp->sid_flags & SID_DELETE) {
rw_exit(&sidp->sid_rwlock);
(void) fssnap_delete_impl(sidpp);
(void) bdev_strategy(bp);
return;
}
if (sidp != *sidpp || SID_INACTIVE(sidp)) {
rw_exit(&sidp->sid_rwlock);
(void) bdev_strategy(bp);
return;
}
if ((error = fssnap_translate(sidpp, bp)) != 0) {
if (sidp != *sidpp || SID_INACTIVE(sidp)) {
rw_exit(&sidp->sid_rwlock);
(void) bdev_strategy(bp);
return;
}
bioerror(bp, error);
biodone(bp);
}
rw_exit(&sidp->sid_rwlock);
}
static int
fssnap_translate(struct snapshot_id **sidpp, struct buf *wbp)
{
snapshot_id_t *sidp = *sidpp;
struct buf *oldbp;
struct cow_info *cowp = sidp->sid_cowinfo;
cow_map_t *cmap = &cowp->cow_map;
cow_map_node_t *cmn;
chunknumber_t cowchunk, startchunk, endchunk;
int error;
int throttle_write = 0;
ASSERT(RW_READ_HELD(&sidp->sid_rwlock));
startchunk = dbtocowchunk(cmap, wbp->b_lblkno);
endchunk = dbtocowchunk(cmap, wbp->b_lblkno +
((wbp->b_bcount-1) >> DEV_BSHIFT));
throttle_write = !(taskq_member(cowp->cow_taskq, curthread) ||
tsd_get(bypass_snapshot_throttle_key));
for (cowchunk = startchunk; cowchunk <= endchunk; cowchunk++) {
if (cowchunk >= (cmap->cmap_bmsize * NBBY))
break;
if (isclr(cmap->cmap_candidate, cowchunk)) {
continue;
}
if (isset(cmap->cmap_hastrans, cowchunk)) {
continue;
}
if (throttle_write) {
if (sema_tryp(&cmap->cmap_throttle_sem) == 0) {
rw_exit(&sidp->sid_rwlock);
atomic_inc_32(&cmap->cmap_waiters);
sema_p(&cmap->cmap_throttle_sem);
atomic_dec_32(&cmap->cmap_waiters);
rw_enter(&sidp->sid_rwlock, RW_READER);
if (sidp != *sidpp || SID_INACTIVE(sidp)) {
sema_v(&cmap->cmap_throttle_sem);
return (ENXIO);
}
}
}
rw_enter(&cmap->cmap_rwlock, RW_WRITER);
if (isset(cmap->cmap_hastrans, cowchunk)) {
if (throttle_write)
sema_v(&cmap->cmap_throttle_sem);
rw_exit(&cmap->cmap_rwlock);
continue;
}
oldbp = getrbuf(KM_SLEEP);
oldbp->b_lblkno = cowchunktodb(cmap, cowchunk);
oldbp->b_edev = wbp->b_edev;
oldbp->b_bcount = cmap->cmap_chunksz;
oldbp->b_bufsize = cmap->cmap_chunksz;
oldbp->b_iodone = NULL;
oldbp->b_proc = NULL;
oldbp->b_flags = B_READ;
oldbp->b_un.b_addr = kmem_alloc(cmap->cmap_chunksz, KM_SLEEP);
(void) bdev_strategy(oldbp);
(void) biowait(oldbp);
if ((error = geterror(oldbp)) != 0) {
cmn_err(CE_WARN, "fssnap_translate: error reading "
"old data for snapshot %d, chunk %llu, disk block "
"%lld, size %lu, error %d.", sidp->sid_snapnumber,
cowchunk, oldbp->b_lblkno, oldbp->b_bcount, error);
kmem_free(oldbp->b_un.b_addr, cmap->cmap_chunksz);
freerbuf(oldbp);
rw_exit(&cmap->cmap_rwlock);
if (throttle_write)
sema_v(&cmap->cmap_throttle_sem);
return (error);
}
cmn = transtbl_add(cmap, cowchunk, oldbp->b_un.b_addr);
freerbuf(oldbp);
cmn->cmn_sid = sidp;
cmn->release_sem = throttle_write;
setbit(cmap->cmap_hastrans, cowchunk);
rw_exit(&cmap->cmap_rwlock);
if (cowp->cow_backfile_array != NULL)
(void) taskq_dispatch(cowp->cow_taskq,
fssnap_write_taskq, cmn, TQ_SLEEP);
}
(void) bdev_strategy(wbp);
return (0);
}
static void
fssnap_write_taskq(void *arg)
{
cow_map_node_t *cmn = (cow_map_node_t *)arg;
snapshot_id_t *sidp = cmn->cmn_sid;
cow_info_t *cowp = sidp->sid_cowinfo;
cow_map_t *cmap = &cowp->cow_map;
int error;
int bf_index;
int release_sem = cmn->release_sem;
rw_enter(&sidp->sid_rwlock, RW_READER);
if (SID_INACTIVE(sidp)) {
rw_exit(&sidp->sid_rwlock);
if (release_sem)
sema_v(&cmap->cmap_throttle_sem);
return;
}
rw_exit(&sidp->sid_rwlock);
atomic_inc_64((uint64_t *)&cmap->cmap_nchunks);
if ((cmap->cmap_maxsize != 0) &&
((cmap->cmap_nchunks * cmap->cmap_chunksz) > cmap->cmap_maxsize)) {
cmn_err(CE_WARN, "fssnap_write_taskq: snapshot %d (%s) has "
"reached the maximum backing file size specified (%llu "
"bytes) and will be deleted.", sidp->sid_snapnumber,
(char *)cowp->cow_kstat_mntpt->ks_data,
cmap->cmap_maxsize);
if (release_sem)
sema_v(&cmap->cmap_throttle_sem);
atomic_or_uint(&sidp->sid_flags, SID_DELETE);
return;
}
bf_index = cmn->cmn_chunk / cmap->cmap_chunksperbf;
if (error = vn_rdwr(UIO_WRITE, (cowp->cow_backfile_array)[bf_index],
cmn->cmn_buf, cmap->cmap_chunksz,
(cmn->cmn_chunk % cmap->cmap_chunksperbf) * cmap->cmap_chunksz,
UIO_SYSSPACE, 0, RLIM64_INFINITY, kcred, (ssize_t *)NULL)) {
cmn_err(CE_WARN, "fssnap_write_taskq: error writing to "
"backing file. DELETING SNAPSHOT %d, backing file path "
"%s, offset %llu bytes, error %d.", sidp->sid_snapnumber,
(char *)cowp->cow_kstat_bfname->ks_data,
cmn->cmn_chunk * cmap->cmap_chunksz, error);
if (release_sem)
sema_v(&cmap->cmap_throttle_sem);
atomic_or_uint(&sidp->sid_flags, SID_DELETE);
return;
}
rw_enter(&cmap->cmap_rwlock, RW_WRITER);
transtbl_delete(cmap, cmn);
rw_exit(&cmap->cmap_rwlock);
if (release_sem)
sema_v(&cmap->cmap_throttle_sem);
}
static void *
fssnap_create_impl(chunknumber_t nchunks, uint_t chunksz, u_offset_t maxsize,
struct vnode *fsvp, int backfilecount, struct vnode **bfvpp, char *backpath,
u_offset_t max_backfile_size)
{
refstr_t *mountpoint;
char taskqname[50];
struct cow_info *cowp;
struct cow_map *cmap;
struct snapshot_id *sidp;
int lastsnap;
if ((nchunks == 0) ||
((chunksz % DEV_BSIZE) != 0) ||
(bfvpp == NULL)) {
return (NULL);
}
mutex_enter(&snapshot_mutex);
findagain:
lastsnap = 0;
for (sidp = snapshot; sidp != NULL; sidp = sidp->sid_next) {
if (sidp->sid_snapnumber > lastsnap)
lastsnap = sidp->sid_snapnumber;
rw_enter(&sidp->sid_rwlock, RW_READER);
if (SID_AVAILABLE(sidp))
break;
rw_exit(&sidp->sid_rwlock);
}
if (sidp) {
if (rw_tryupgrade(&sidp->sid_rwlock) == 0) {
rw_exit(&sidp->sid_rwlock);
goto findagain;
}
} else {
sidp = kmem_alloc(sizeof (struct snapshot_id), KM_SLEEP);
rw_init(&sidp->sid_rwlock, NULL, RW_DEFAULT, NULL);
rw_enter(&sidp->sid_rwlock, RW_WRITER);
sidp->sid_snapnumber = (snapshot == NULL) ? 0 : lastsnap + 1;
sidp->sid_cowinfo = NULL;
sidp->sid_flags = 0;
sidp->sid_next = snapshot;
snapshot = sidp;
}
ASSERT(RW_WRITE_HELD(&sidp->sid_rwlock));
ASSERT(sidp->sid_cowinfo == NULL);
ASSERT(sidp->sid_snapnumber <= (lastsnap + 1));
sidp->sid_flags |= SID_CREATING;
VN_HOLD(fsvp);
sidp->sid_fvp = fsvp;
num_snapshots++;
cowp = kmem_zalloc(sizeof (struct cow_info), KM_SLEEP);
cowp->cow_backfile_array = bfvpp;
cowp->cow_backcount = backfilecount;
cowp->cow_backfile_sz = max_backfile_size;
(void) snprintf(taskqname, sizeof (taskqname), "%s_taskq_%d", snapname,
sidp->sid_snapnumber);
cowp->cow_taskq = taskq_create(taskqname, fssnap_taskq_nthreads,
minclsyspri, 1, fssnap_taskq_maxtasks, 0);
taskq_suspend(cowp->cow_taskq);
cmap = &cowp->cow_map;
rw_init(&cmap->cmap_rwlock, NULL, RW_DEFAULT, NULL);
rw_enter(&cmap->cmap_rwlock, RW_WRITER);
sema_init(&cmap->cmap_throttle_sem, fssnap_max_mem_chunks, NULL,
SEMA_DEFAULT, NULL);
cmap->cmap_chunksz = chunksz;
cmap->cmap_maxsize = maxsize;
cmap->cmap_chunksperbf = max_backfile_size / chunksz;
cmap->cmap_bmsize = (nchunks + (NBBY - 1)) / NBBY;
cmap->cmap_hastrans = kmem_zalloc(cmap->cmap_bmsize, KM_SLEEP);
cmap->cmap_candidate = kmem_zalloc(cmap->cmap_bmsize, KM_SLEEP);
sidp->sid_cowinfo = cowp;
mountpoint = vfs_getmntpoint(fsvp->v_vfsp);
fssnap_create_kstats(sidp, sidp->sid_snapnumber,
refstr_value(mountpoint), backpath);
refstr_rele(mountpoint);
mutex_exit(&snapshot_mutex);
return (sidp);
}
static void
fssnap_set_candidate_impl(void *snapshot_id, chunknumber_t chunknumber)
{
struct snapshot_id *sid = snapshot_id;
struct cow_info *cowp = sid->sid_cowinfo;
struct cow_map *cmap = &cowp->cow_map;
ASSERT(chunknumber < (cmap->cmap_bmsize * NBBY));
setbit(cmap->cmap_candidate, chunknumber);
}
static int
fssnap_is_candidate_impl(void *snapshot_id, u_offset_t off)
{
struct snapshot_id *sid = snapshot_id;
struct cow_info *cowp = sid->sid_cowinfo;
struct cow_map *cmap = &cowp->cow_map;
ulong_t chunknumber = off / cmap->cmap_chunksz;
ASSERT(chunknumber < (cmap->cmap_bmsize * NBBY));
return (isset(cmap->cmap_candidate, chunknumber));
}
static int
fssnap_create_done_impl(void *snapshot_id)
{
struct snapshot_id **sidpp, *sidp = snapshot_id;
struct cow_info *cowp;
struct cow_map *cmap;
int snapnumber = -1;
char name[20];
ASSERT(sidp);
ASSERT(RW_WRITE_HELD(&sidp->sid_rwlock));
ASSERT(sidp->sid_cowinfo);
cowp = sidp->sid_cowinfo;
cmap = &cowp->cow_map;
ASSERT(RW_WRITE_HELD(&cmap->cmap_rwlock));
sidp->sid_flags &= ~(SID_CREATING | SID_DISABLED);
snapnumber = sidp->sid_snapnumber;
if (ddi_soft_state_zalloc(statep, snapnumber) != DDI_SUCCESS) {
cmn_err(CE_WARN,
"snap_ioctl: create: could not allocate "
"state for snapshot %d.", snapnumber);
snapnumber = -1;
goto out;
}
sidpp = ddi_get_soft_state(statep, snapnumber);
*sidpp = sidp;
ASSERT(fssnap_dip != NULL);
(void) snprintf(name, sizeof (name), "%d", snapnumber);
if (ddi_create_minor_node(fssnap_dip, name, S_IFBLK,
snapnumber, DDI_PSEUDO, 0) != DDI_SUCCESS) {
cmn_err(CE_WARN, "snap_ioctl: could not create "
"block minor node for snapshot %d.", snapnumber);
snapnumber = -1;
goto out;
}
(void) snprintf(name, sizeof (name), "%d,raw", snapnumber);
if (ddi_create_minor_node(fssnap_dip, name, S_IFCHR,
snapnumber, DDI_PSEUDO, 0) != DDI_SUCCESS) {
cmn_err(CE_WARN, "snap_ioctl: could not create "
"character minor node for snapshot %d.", snapnumber);
snapnumber = -1;
}
out:
rw_exit(&sidp->sid_rwlock);
rw_exit(&cmap->cmap_rwlock);
taskq_resume(cowp->cow_taskq);
return (snapnumber);
}
static int
fssnap_delete_impl(void *snapshot_id)
{
struct snapshot_id **sidpp = (struct snapshot_id **)snapshot_id;
struct snapshot_id *sidp;
struct snapshot_id **statesidpp;
struct cow_info *cowp;
struct cow_map *cmap;
char name[20];
int snapnumber = -1;
vnode_t **vpp;
if (sidpp == NULL) {
return (-1);
}
sidp = *sidpp;
rw_enter(&sidp->sid_rwlock, RW_WRITER);
ASSERT(RW_WRITE_HELD(&sidp->sid_rwlock));
if (*sidpp == NULL) {
rw_exit(&sidp->sid_rwlock);
return (-1);
}
sidp->sid_flags |= SID_DISABLING;
if (sidp->sid_flags & SID_DELETE) {
cmn_err(CE_WARN, "Snapshot %d automatically deleted.",
sidp->sid_snapnumber);
sidp->sid_flags &= ~(SID_DELETE);
}
*sidpp = NULL;
cowp = sidp->sid_cowinfo;
if (cowp == NULL) {
rw_exit(&sidp->sid_rwlock);
return (-1);
}
rw_exit(&sidp->sid_rwlock);
if (cowp->cow_taskq) {
taskq_destroy(cowp->cow_taskq);
cowp->cow_taskq = NULL;
}
if (cowp->cow_backfile_array != NULL) {
for (vpp = cowp->cow_backfile_array; *vpp; vpp++)
VN_RELE(*vpp);
kmem_free(cowp->cow_backfile_array,
(cowp->cow_backcount + 1) * sizeof (vnode_t *));
cowp->cow_backfile_array = NULL;
}
sidp->sid_cowinfo = NULL;
cmap = &cowp->cow_map;
ASSERT(cmap);
if (cmap->cmap_candidate)
kmem_free(cmap->cmap_candidate, cmap->cmap_bmsize);
if (cmap->cmap_hastrans)
kmem_free(cmap->cmap_hastrans, cmap->cmap_bmsize);
if (cmap->cmap_table)
transtbl_free(&cowp->cow_map);
rw_destroy(&cmap->cmap_rwlock);
while (cmap->cmap_waiters) {
sema_p(&cmap->cmap_throttle_sem);
sema_v(&cmap->cmap_throttle_sem);
}
sema_destroy(&cmap->cmap_throttle_sem);
fssnap_delete_kstats(cowp);
kmem_free(cowp, sizeof (struct cow_info));
statesidpp = ddi_get_soft_state(statep, sidp->sid_snapnumber);
if (statesidpp == NULL || *statesidpp == NULL) {
cmn_err(CE_WARN,
"fssnap_delete_impl: could not find state for snapshot %d.",
sidp->sid_snapnumber);
}
ASSERT(*statesidpp == sidp);
mutex_enter(&snapshot_mutex);
rw_enter(&sidp->sid_rwlock, RW_WRITER);
sidp->sid_flags &= ~(SID_DISABLING);
sidp->sid_flags |= SID_DISABLED;
VN_RELE(sidp->sid_fvp);
sidp->sid_fvp = NULL;
snapnumber = sidp->sid_snapnumber;
if (SID_AVAILABLE(sidp)) {
ASSERT(fssnap_dip != NULL);
(void) snprintf(name, sizeof (name), "%d",
sidp->sid_snapnumber);
ddi_remove_minor_node(fssnap_dip, name);
(void) snprintf(name, sizeof (name), "%d,raw",
sidp->sid_snapnumber);
ddi_remove_minor_node(fssnap_dip, name);
ddi_soft_state_free(statep, sidp->sid_snapnumber);
num_snapshots--;
}
mutex_exit(&snapshot_mutex);
rw_exit(&sidp->sid_rwlock);
return (snapnumber);
}
static void
fssnap_create_kstats(snapshot_id_t *sidp, int snapnum,
const char *mountpoint, const char *backfilename)
{
kstat_t *num, *mntpoint, *bfname;
kstat_named_t *hw;
struct cow_info *cowp = sidp->sid_cowinfo;
struct cow_kstat_num *stats;
if (fssnap_highwater_kstat == NULL) {
cmn_err(CE_WARN, "fssnap_create_kstats: failed to lookup "
"high water mark kstat.");
return;
}
hw = (kstat_named_t *)fssnap_highwater_kstat->ks_data;
if (hw->value.ui32 < snapnum)
hw->value.ui32 = snapnum;
kstat_delete_byname(snapname, snapnum, FSSNAP_KSTAT_MNTPT);
if (mountpoint != NULL) {
mntpoint = kstat_create(snapname, snapnum, FSSNAP_KSTAT_MNTPT,
"misc", KSTAT_TYPE_RAW, strlen(mountpoint) + 1, 0);
if (mntpoint == NULL) {
cowp->cow_kstat_mntpt = NULL;
cmn_err(CE_WARN, "fssnap_create_kstats: failed to "
"create mount point kstat");
} else {
(void) strncpy(mntpoint->ks_data, mountpoint,
strlen(mountpoint));
cowp->cow_kstat_mntpt = mntpoint;
kstat_install(mntpoint);
}
} else {
cowp->cow_kstat_mntpt = NULL;
cmn_err(CE_WARN, "fssnap_create_kstats: mount point not "
"specified.");
}
kstat_delete_byname(snapname, snapnum, FSSNAP_KSTAT_BFNAME);
if (backfilename == NULL) {
cowp->cow_kstat_bfname = NULL;
} else {
bfname = kstat_create(snapname, snapnum, FSSNAP_KSTAT_BFNAME,
"misc", KSTAT_TYPE_RAW, strlen(backfilename) + 1, 0);
if (bfname != NULL) {
(void) strncpy(bfname->ks_data, backfilename,
strlen(backfilename));
cowp->cow_kstat_bfname = bfname;
kstat_install(bfname);
} else {
cowp->cow_kstat_bfname = NULL;
cmn_err(CE_WARN, "fssnap_create_kstats: failed to "
"create backing file name kstat");
}
}
kstat_delete_byname(snapname, snapnum, FSSNAP_KSTAT_NUM);
num = kstat_create(snapname, snapnum, FSSNAP_KSTAT_NUM,
"misc", KSTAT_TYPE_NAMED,
sizeof (struct cow_kstat_num) / sizeof (kstat_named_t),
0);
if (num == NULL) {
cmn_err(CE_WARN, "fssnap_create_kstats: failed to create "
"numeric kstats");
cowp->cow_kstat_num = NULL;
return;
}
cowp->cow_kstat_num = num;
stats = num->ks_data;
num->ks_update = fssnap_update_kstat_num;
num->ks_private = sidp;
kstat_named_init(&stats->ckn_state, FSSNAP_KSTAT_NUM_STATE,
KSTAT_DATA_INT32);
kstat_named_init(&stats->ckn_bfsize, FSSNAP_KSTAT_NUM_BFSIZE,
KSTAT_DATA_UINT64);
kstat_named_init(&stats->ckn_maxsize, FSSNAP_KSTAT_NUM_MAXSIZE,
KSTAT_DATA_UINT64);
kstat_named_init(&stats->ckn_createtime, FSSNAP_KSTAT_NUM_CREATETIME,
KSTAT_DATA_LONG);
kstat_named_init(&stats->ckn_chunksize, FSSNAP_KSTAT_NUM_CHUNKSIZE,
KSTAT_DATA_UINT32);
stats->ckn_chunksize.value.ui32 = cowp->cow_map.cmap_chunksz;
stats->ckn_maxsize.value.ui64 = cowp->cow_map.cmap_maxsize;
stats->ckn_createtime.value.l = gethrestime_sec();
kstat_install(num);
}
int
fssnap_update_kstat_num(kstat_t *ksp, int rw)
{
snapshot_id_t *sidp = (snapshot_id_t *)ksp->ks_private;
struct cow_info *cowp = sidp->sid_cowinfo;
struct cow_kstat_num *stats = ksp->ks_data;
if (rw == KSTAT_WRITE)
return (EACCES);
if (sidp->sid_flags & SID_CREATING)
stats->ckn_state.value.i32 = COWSTATE_CREATING;
else if (SID_INACTIVE(sidp))
stats->ckn_state.value.i32 = COWSTATE_DISABLED;
else if (SID_BUSY(sidp))
stats->ckn_state.value.i32 = COWSTATE_ACTIVE;
else
stats->ckn_state.value.i32 = COWSTATE_IDLE;
stats->ckn_bfsize.value.ui64 = cowp->cow_map.cmap_nchunks *
cowp->cow_map.cmap_chunksz;
return (0);
}
void
fssnap_delete_kstats(struct cow_info *cowp)
{
if (cowp->cow_kstat_num != NULL) {
kstat_delete(cowp->cow_kstat_num);
cowp->cow_kstat_num = NULL;
}
if (cowp->cow_kstat_mntpt != NULL) {
kstat_delete(cowp->cow_kstat_mntpt);
cowp->cow_kstat_mntpt = NULL;
}
if (cowp->cow_kstat_bfname != NULL) {
kstat_delete(cowp->cow_kstat_bfname);
cowp->cow_kstat_bfname = NULL;
}
}