Author: avg
Date: Tue Jul 12 11:37:19 2016
New Revision: 302651
URL: https://svnweb.freebsd.org/changeset/base/302651

Log:
  7054 dmu_tx_hold_t should use refcount_t to track space
  
  illumos/illumos-gate@0c779ad424a92a84d1e07d47cab7f8009189202b
  
https://github.com/illumos/illumos-gate/commit/0c779ad424a92a84d1e07d47cab7f8009189202b
  
  https://www.illumos.org/issues/7054
    upstream:
    ee0003de7d3e598499be7ac3fe6b61efcc47cb7f
    DLPX-40399 dmu_tx_hold_t should use refcount_t to track space
  
  Reviewed by: George Wilson <george.wil...@delphix.com>
  Reviewed by: Paul Dagnelie <p...@delphix.com>
  Reviewed by: Igor Kozhukhov <ikozhuk...@gmail.com>
  Approved by: Dan McDonald <dan...@omniti.com>
  Author: Matthew Ahrens <mahr...@delphix.com>

Modified:
  vendor-sys/illumos/dist/uts/common/fs/zfs/dmu_tx.c
  vendor-sys/illumos/dist/uts/common/fs/zfs/sys/dmu_tx.h
  vendor-sys/illumos/dist/uts/common/fs/zfs/sys/zap.h
  vendor-sys/illumos/dist/uts/common/fs/zfs/sys/zap_impl.h
  vendor-sys/illumos/dist/uts/common/fs/zfs/zap.c
  vendor-sys/illumos/dist/uts/common/fs/zfs/zap_micro.c

Modified: vendor-sys/illumos/dist/uts/common/fs/zfs/dmu_tx.c
==============================================================================
--- vendor-sys/illumos/dist/uts/common/fs/zfs/dmu_tx.c  Tue Jul 12 11:36:22 
2016        (r302650)
+++ vendor-sys/illumos/dist/uts/common/fs/zfs/dmu_tx.c  Tue Jul 12 11:37:19 
2016        (r302651)
@@ -129,6 +129,12 @@ dmu_tx_hold_object_impl(dmu_tx_t *tx, ob
        txh = kmem_zalloc(sizeof (dmu_tx_hold_t), KM_SLEEP);
        txh->txh_tx = tx;
        txh->txh_dnode = dn;
+       refcount_create(&txh->txh_space_towrite);
+       refcount_create(&txh->txh_space_tofree);
+       refcount_create(&txh->txh_space_tooverwrite);
+       refcount_create(&txh->txh_space_tounref);
+       refcount_create(&txh->txh_memory_tohold);
+       refcount_create(&txh->txh_fudge);
 #ifdef ZFS_DEBUG
        txh->txh_type = type;
        txh->txh_arg1 = arg1;
@@ -201,12 +207,18 @@ dmu_tx_count_twig(dmu_tx_hold_t *txh, dn
        freeable = (bp && (freeable ||
            dsl_dataset_block_freeable(ds, bp, bp->blk_birth)));
 
-       if (freeable)
-               txh->txh_space_tooverwrite += space;
-       else
-               txh->txh_space_towrite += space;
-       if (bp)
-               txh->txh_space_tounref += bp_get_dsize(os->os_spa, bp);
+       if (freeable) {
+               (void) refcount_add_many(&txh->txh_space_tooverwrite,
+                   space, FTAG);
+       } else {
+               (void) refcount_add_many(&txh->txh_space_towrite,
+                   space, FTAG);
+       }
+
+       if (bp) {
+               (void) refcount_add_many(&txh->txh_space_tounref,
+                   bp_get_dsize(os->os_spa, bp), FTAG);
+       }
 
        dmu_tx_count_twig(txh, dn, parent, level + 1,
            blkid >> epbs, freeable, history);
@@ -336,8 +348,11 @@ dmu_tx_count_write(dmu_tx_hold_t *txh, u
                                bits = 64 - min_bs;
                                epbs = min_ibs - SPA_BLKPTRSHIFT;
                                for (bits -= epbs * (nlvls - 1);
-                                   bits >= 0; bits -= epbs)
-                                       txh->txh_fudge += 1ULL << max_ibs;
+                                   bits >= 0; bits -= epbs) {
+                                       (void) refcount_add_many(
+                                           &txh->txh_fudge,
+                                           1ULL << max_ibs, FTAG);
+                               }
                                goto out;
                        }
                        off += delta;
@@ -353,7 +368,8 @@ dmu_tx_count_write(dmu_tx_hold_t *txh, u
         */
        start = P2ALIGN(off, 1ULL << max_bs);
        end = P2ROUNDUP(off + len, 1ULL << max_bs) - 1;
-       txh->txh_space_towrite += end - start + 1;
+       (void) refcount_add_many(&txh->txh_space_towrite,
+           end - start + 1, FTAG);
 
        start >>= min_bs;
        end >>= min_bs;
@@ -368,18 +384,21 @@ dmu_tx_count_write(dmu_tx_hold_t *txh, u
                start >>= epbs;
                end >>= epbs;
                ASSERT3U(end, >=, start);
-               txh->txh_space_towrite += (end - start + 1) << max_ibs;
+               (void) refcount_add_many(&txh->txh_space_towrite,
+                   (end - start + 1) << max_ibs, FTAG);
                if (start != 0) {
                        /*
                         * We also need a new blkid=0 indirect block
                         * to reference any existing file data.
                         */
-                       txh->txh_space_towrite += 1ULL << max_ibs;
+                       (void) refcount_add_many(&txh->txh_space_towrite,
+                           1ULL << max_ibs, FTAG);
                }
        }
 
 out:
-       if (txh->txh_space_towrite + txh->txh_space_tooverwrite >
+       if (refcount_count(&txh->txh_space_towrite) +
+           refcount_count(&txh->txh_space_tooverwrite) >
            2 * DMU_MAX_ACCESS)
                err = SET_ERROR(EFBIG);
 
@@ -398,12 +417,15 @@ dmu_tx_count_dnode(dmu_tx_hold_t *txh)
        if (dn && dn->dn_dbuf->db_blkptr &&
            dsl_dataset_block_freeable(dn->dn_objset->os_dsl_dataset,
            dn->dn_dbuf->db_blkptr, dn->dn_dbuf->db_blkptr->blk_birth)) {
-               txh->txh_space_tooverwrite += space;
-               txh->txh_space_tounref += space;
+               (void) refcount_add_many(&txh->txh_space_tooverwrite,
+                   space, FTAG);
+               (void) refcount_add_many(&txh->txh_space_tounref, space, FTAG);
        } else {
-               txh->txh_space_towrite += space;
-               if (dn && dn->dn_dbuf->db_blkptr)
-                       txh->txh_space_tounref += space;
+               (void) refcount_add_many(&txh->txh_space_towrite, space, FTAG);
+               if (dn && dn->dn_dbuf->db_blkptr) {
+                       (void) refcount_add_many(&txh->txh_space_tounref,
+                           space, FTAG);
+               }
        }
 }
 
@@ -525,7 +547,8 @@ dmu_tx_count_free(dmu_tx_hold_t *txh, ui
                        break;
                }
 
-               txh->txh_memory_tohold += dbuf->db.db_size;
+               (void) refcount_add_many(&txh->txh_memory_tohold,
+                   dbuf->db.db_size, FTAG);
 
                /*
                 * We don't check memory_tohold against DMU_MAX_ACCESS because
@@ -578,20 +601,23 @@ dmu_tx_count_free(dmu_tx_hold_t *txh, ui
                    (dn->dn_indblkshift - SPA_BLKPTRSHIFT);
 
                while (level++ < maxlevel) {
-                       txh->txh_memory_tohold += MAX(MIN(blkcnt, nl1blks), 1)
-                           << dn->dn_indblkshift;
+                       (void) refcount_add_many(&txh->txh_memory_tohold,
+                           MAX(MIN(blkcnt, nl1blks), 1) << dn->dn_indblkshift,
+                           FTAG);
                        blkcnt = 1 + (blkcnt >> epbs);
                }
        }
 
        /* account for new level 1 indirect blocks that might show up */
        if (skipped > 0) {
-               txh->txh_fudge += skipped << dn->dn_indblkshift;
+               (void) refcount_add_many(&txh->txh_fudge,
+                   skipped << dn->dn_indblkshift, FTAG);
                skipped = MIN(skipped, DMU_MAX_DELETEBLKCNT >> epbs);
-               txh->txh_memory_tohold += skipped << dn->dn_indblkshift;
+               (void) refcount_add_many(&txh->txh_memory_tohold,
+                   skipped << dn->dn_indblkshift, FTAG);
        }
-       txh->txh_space_tofree += space;
-       txh->txh_space_tounref += unref;
+       (void) refcount_add_many(&txh->txh_space_tofree, space, FTAG);
+       (void) refcount_add_many(&txh->txh_space_tounref, unref, FTAG);
 }
 
 /*
@@ -617,7 +643,10 @@ dmu_tx_mark_netfree(dmu_tx_t *tx)
         * cause overflows when doing math with these values (e.g. in
         * dmu_tx_try_assign()).
         */
-       txh->txh_space_tofree = txh->txh_space_tounref = 1024 * 1024 * 1024;
+       (void) refcount_add_many(&txh->txh_space_tofree,
+           1024 * 1024 * 1024, FTAG);
+       (void) refcount_add_many(&txh->txh_space_tounref,
+           1024 * 1024 * 1024, FTAG);
 }
 
 void
@@ -716,9 +745,7 @@ dmu_tx_hold_zap(dmu_tx_t *tx, uint64_t o
 {
        dmu_tx_hold_t *txh;
        dnode_t *dn;
-       dsl_dataset_phys_t *ds_phys;
-       uint64_t nblocks;
-       int epbs, err;
+       int err;
 
        ASSERT(tx->tx_txg == 0);
 
@@ -761,12 +788,17 @@ dmu_tx_hold_zap(dmu_tx_t *tx, uint64_t o
                 */
                bp = &dn->dn_phys->dn_blkptr[0];
                if (dsl_dataset_block_freeable(dn->dn_objset->os_dsl_dataset,
-                   bp, bp->blk_birth))
-                       txh->txh_space_tooverwrite += MZAP_MAX_BLKSZ;
-               else
-                       txh->txh_space_towrite += MZAP_MAX_BLKSZ;
-               if (!BP_IS_HOLE(bp))
-                       txh->txh_space_tounref += MZAP_MAX_BLKSZ;
+                   bp, bp->blk_birth)) {
+                       (void) refcount_add_many(&txh->txh_space_tooverwrite,
+                           MZAP_MAX_BLKSZ, FTAG);
+               } else {
+                       (void) refcount_add_many(&txh->txh_space_towrite,
+                           MZAP_MAX_BLKSZ, FTAG);
+               }
+               if (!BP_IS_HOLE(bp)) {
+                       (void) refcount_add_many(&txh->txh_space_tounref,
+                           MZAP_MAX_BLKSZ, FTAG);
+               }
                return;
        }
 
@@ -788,15 +820,29 @@ dmu_tx_hold_zap(dmu_tx_t *tx, uint64_t o
 
        /*
         * If the modified blocks are scattered to the four winds,
-        * we'll have to modify an indirect twig for each.
+        * we'll have to modify an indirect twig for each.  We can make
+        * modifications at up to 3 locations:
+        *  - header block at the beginning of the object
+        *  - target leaf block
+        *  - end of the object, where we might need to write:
+        *      - a new leaf block if the target block needs to be split
+        *      - the new pointer table, if it is growing
+        *      - the new cookie table, if it is growing
         */
-       epbs = dn->dn_indblkshift - SPA_BLKPTRSHIFT;
-       ds_phys = dsl_dataset_phys(dn->dn_objset->os_dsl_dataset);
-       for (nblocks = dn->dn_maxblkid >> epbs; nblocks != 0; nblocks >>= epbs)
-               if (ds_phys->ds_prev_snap_obj)
-                       txh->txh_space_towrite += 3 << dn->dn_indblkshift;
-               else
-                       txh->txh_space_tooverwrite += 3 << dn->dn_indblkshift;
+       int epbs = dn->dn_indblkshift - SPA_BLKPTRSHIFT;
+       dsl_dataset_phys_t *ds_phys =
+           dsl_dataset_phys(dn->dn_objset->os_dsl_dataset);
+       for (int lvl = 1; lvl < dn->dn_nlevels; lvl++) {
+               uint64_t num_indirects = 1 + (dn->dn_maxblkid >> (epbs * lvl));
+               uint64_t spc = MIN(3, num_indirects) << dn->dn_indblkshift;
+               if (ds_phys->ds_prev_snap_obj != 0) {
+                       (void) refcount_add_many(&txh->txh_space_towrite,
+                           spc, FTAG);
+               } else {
+                       (void) refcount_add_many(&txh->txh_space_tooverwrite,
+                           spc, FTAG);
+               }
+       }
 }
 
 void
@@ -821,7 +867,7 @@ dmu_tx_hold_space(dmu_tx_t *tx, uint64_t
        txh = dmu_tx_hold_object_impl(tx, tx->tx_objset,
            DMU_NEW_OBJECT, THT_SPACE, space, 0);
 
-       txh->txh_space_towrite += space;
+       (void) refcount_add_many(&txh->txh_space_towrite, space, FTAG);
 }
 
 int
@@ -1169,12 +1215,12 @@ dmu_tx_try_assign(dmu_tx_t *tx, txg_how_
                        (void) refcount_add(&dn->dn_tx_holds, tx);
                        mutex_exit(&dn->dn_mtx);
                }
-               towrite += txh->txh_space_towrite;
-               tofree += txh->txh_space_tofree;
-               tooverwrite += txh->txh_space_tooverwrite;
-               tounref += txh->txh_space_tounref;
-               tohold += txh->txh_memory_tohold;
-               fudge += txh->txh_fudge;
+               towrite += refcount_count(&txh->txh_space_towrite);
+               tofree += refcount_count(&txh->txh_space_tofree);
+               tooverwrite += refcount_count(&txh->txh_space_tooverwrite);
+               tounref += refcount_count(&txh->txh_space_tounref);
+               tohold += refcount_count(&txh->txh_memory_tohold);
+               fudge += refcount_count(&txh->txh_fudge);
        }
 
        /*
@@ -1377,24 +1423,59 @@ dmu_tx_willuse_space(dmu_tx_t *tx, int64
 #endif
 }
 
-void
-dmu_tx_commit(dmu_tx_t *tx)
+static void
+dmu_tx_destroy(dmu_tx_t *tx)
 {
        dmu_tx_hold_t *txh;
 
+       while ((txh = list_head(&tx->tx_holds)) != NULL) {
+               dnode_t *dn = txh->txh_dnode;
+
+               list_remove(&tx->tx_holds, txh);
+               refcount_destroy_many(&txh->txh_space_towrite,
+                   refcount_count(&txh->txh_space_towrite));
+               refcount_destroy_many(&txh->txh_space_tofree,
+                   refcount_count(&txh->txh_space_tofree));
+               refcount_destroy_many(&txh->txh_space_tooverwrite,
+                   refcount_count(&txh->txh_space_tooverwrite));
+               refcount_destroy_many(&txh->txh_space_tounref,
+                   refcount_count(&txh->txh_space_tounref));
+               refcount_destroy_many(&txh->txh_memory_tohold,
+                   refcount_count(&txh->txh_memory_tohold));
+               refcount_destroy_many(&txh->txh_fudge,
+                   refcount_count(&txh->txh_fudge));
+               kmem_free(txh, sizeof (dmu_tx_hold_t));
+               if (dn != NULL)
+                       dnode_rele(dn, tx);
+       }
+
+       list_destroy(&tx->tx_callbacks);
+       list_destroy(&tx->tx_holds);
+#ifdef ZFS_DEBUG
+       refcount_destroy_many(&tx->tx_space_written,
+           refcount_count(&tx->tx_space_written));
+       refcount_destroy_many(&tx->tx_space_freed,
+           refcount_count(&tx->tx_space_freed));
+#endif
+       kmem_free(tx, sizeof (dmu_tx_t));
+}
+
+void
+dmu_tx_commit(dmu_tx_t *tx)
+{
        ASSERT(tx->tx_txg != 0);
 
        /*
         * Go through the transaction's hold list and remove holds on
         * associated dnodes, notifying waiters if no holds remain.
         */
-       while (txh = list_head(&tx->tx_holds)) {
+       for (dmu_tx_hold_t *txh = list_head(&tx->tx_holds); txh != NULL;
+           txh = list_next(&tx->tx_holds, txh)) {
                dnode_t *dn = txh->txh_dnode;
 
-               list_remove(&tx->tx_holds, txh);
-               kmem_free(txh, sizeof (dmu_tx_hold_t));
                if (dn == NULL)
                        continue;
+
                mutex_enter(&dn->dn_mtx);
                ASSERT3U(dn->dn_assigned_txg, ==, tx->tx_txg);
 
@@ -1403,7 +1484,6 @@ dmu_tx_commit(dmu_tx_t *tx)
                        cv_broadcast(&dn->dn_notxholds);
                }
                mutex_exit(&dn->dn_mtx);
-               dnode_rele(dn, tx);
        }
 
        if (tx->tx_tempreserve_cookie)
@@ -1415,51 +1495,26 @@ dmu_tx_commit(dmu_tx_t *tx)
        if (tx->tx_anyobj == FALSE)
                txg_rele_to_sync(&tx->tx_txgh);
 
-       list_destroy(&tx->tx_callbacks);
-       list_destroy(&tx->tx_holds);
 #ifdef ZFS_DEBUG
        dprintf("towrite=%llu written=%llu tofree=%llu freed=%llu\n",
            tx->tx_space_towrite, refcount_count(&tx->tx_space_written),
            tx->tx_space_tofree, refcount_count(&tx->tx_space_freed));
-       refcount_destroy_many(&tx->tx_space_written,
-           refcount_count(&tx->tx_space_written));
-       refcount_destroy_many(&tx->tx_space_freed,
-           refcount_count(&tx->tx_space_freed));
 #endif
-       kmem_free(tx, sizeof (dmu_tx_t));
+       dmu_tx_destroy(tx);
 }
 
 void
 dmu_tx_abort(dmu_tx_t *tx)
 {
-       dmu_tx_hold_t *txh;
-
        ASSERT(tx->tx_txg == 0);
 
-       while (txh = list_head(&tx->tx_holds)) {
-               dnode_t *dn = txh->txh_dnode;
-
-               list_remove(&tx->tx_holds, txh);
-               kmem_free(txh, sizeof (dmu_tx_hold_t));
-               if (dn != NULL)
-                       dnode_rele(dn, tx);
-       }
-
        /*
         * Call any registered callbacks with an error code.
         */
        if (!list_is_empty(&tx->tx_callbacks))
                dmu_tx_do_callbacks(&tx->tx_callbacks, ECANCELED);
 
-       list_destroy(&tx->tx_callbacks);
-       list_destroy(&tx->tx_holds);
-#ifdef ZFS_DEBUG
-       refcount_destroy_many(&tx->tx_space_written,
-           refcount_count(&tx->tx_space_written));
-       refcount_destroy_many(&tx->tx_space_freed,
-           refcount_count(&tx->tx_space_freed));
-#endif
-       kmem_free(tx, sizeof (dmu_tx_t));
+       dmu_tx_destroy(tx);
 }
 
 uint64_t
@@ -1498,7 +1553,7 @@ dmu_tx_do_callbacks(list_t *cb_list, int
 {
        dmu_tx_callback_t *dcb;
 
-       while (dcb = list_head(cb_list)) {
+       while ((dcb = list_head(cb_list)) != NULL) {
                list_remove(cb_list, dcb);
                dcb->dcb_func(dcb->dcb_data, error);
                kmem_free(dcb, sizeof (dmu_tx_callback_t));
@@ -1556,18 +1611,24 @@ dmu_tx_hold_spill(dmu_tx_t *tx, uint64_t
 
        /* If blkptr doesn't exist then add space to towrite */
        if (!(dn->dn_phys->dn_flags & DNODE_FLAG_SPILL_BLKPTR)) {
-               txh->txh_space_towrite += SPA_OLD_MAXBLOCKSIZE;
+               (void) refcount_add_many(&txh->txh_space_towrite,
+                   SPA_OLD_MAXBLOCKSIZE, FTAG);
        } else {
                blkptr_t *bp;
 
                bp = &dn->dn_phys->dn_spill;
                if (dsl_dataset_block_freeable(dn->dn_objset->os_dsl_dataset,
-                   bp, bp->blk_birth))
-                       txh->txh_space_tooverwrite += SPA_OLD_MAXBLOCKSIZE;
-               else
-                       txh->txh_space_towrite += SPA_OLD_MAXBLOCKSIZE;
-               if (!BP_IS_HOLE(bp))
-                       txh->txh_space_tounref += SPA_OLD_MAXBLOCKSIZE;
+                   bp, bp->blk_birth)) {
+                       (void) refcount_add_many(&txh->txh_space_tooverwrite,
+                           SPA_OLD_MAXBLOCKSIZE, FTAG);
+               } else {
+                       (void) refcount_add_many(&txh->txh_space_towrite,
+                           SPA_OLD_MAXBLOCKSIZE, FTAG);
+               }
+               if (!BP_IS_HOLE(bp)) {
+                       (void) refcount_add_many(&txh->txh_space_tounref,
+                           SPA_OLD_MAXBLOCKSIZE, FTAG);
+               }
        }
 }
 

Modified: vendor-sys/illumos/dist/uts/common/fs/zfs/sys/dmu_tx.h
==============================================================================
--- vendor-sys/illumos/dist/uts/common/fs/zfs/sys/dmu_tx.h      Tue Jul 12 
11:36:22 2016        (r302650)
+++ vendor-sys/illumos/dist/uts/common/fs/zfs/sys/dmu_tx.h      Tue Jul 12 
11:37:19 2016        (r302651)
@@ -23,7 +23,7 @@
  * Use is subject to license terms.
  */
 /*
- * Copyright (c) 2013 by Delphix. All rights reserved.
+ * Copyright (c) 2012, 2015 by Delphix. All rights reserved.
  */
 
 #ifndef        _SYS_DMU_TX_H
@@ -102,12 +102,12 @@ typedef struct dmu_tx_hold {
        dmu_tx_t *txh_tx;
        list_node_t txh_node;
        struct dnode *txh_dnode;
-       uint64_t txh_space_towrite;
-       uint64_t txh_space_tofree;
-       uint64_t txh_space_tooverwrite;
-       uint64_t txh_space_tounref;
-       uint64_t txh_memory_tohold;
-       uint64_t txh_fudge;
+       refcount_t txh_space_towrite;
+       refcount_t txh_space_tofree;
+       refcount_t txh_space_tooverwrite;
+       refcount_t txh_space_tounref;
+       refcount_t txh_memory_tohold;
+       refcount_t txh_fudge;
 #ifdef ZFS_DEBUG
        enum dmu_tx_hold_type txh_type;
        uint64_t txh_arg1;

Modified: vendor-sys/illumos/dist/uts/common/fs/zfs/sys/zap.h
==============================================================================
--- vendor-sys/illumos/dist/uts/common/fs/zfs/sys/zap.h Tue Jul 12 11:36:22 
2016        (r302650)
+++ vendor-sys/illumos/dist/uts/common/fs/zfs/sys/zap.h Tue Jul 12 11:37:19 
2016        (r302651)
@@ -20,7 +20,7 @@
  */
 /*
  * Copyright (c) 2005, 2010, Oracle and/or its affiliates. All rights reserved.
- * Copyright (c) 2013 by Delphix. All rights reserved.
+ * Copyright (c) 2012, 2015 by Delphix. All rights reserved.
  */
 
 #ifndef        _SYS_ZAP_H
@@ -80,6 +80,7 @@
  */
 
 #include <sys/dmu.h>
+#include <sys/refcount.h>
 
 #ifdef __cplusplus
 extern "C" {
@@ -217,7 +218,7 @@ int zap_prefetch_uint64(objset_t *os, ui
     int key_numints);
 
 int zap_count_write(objset_t *os, uint64_t zapobj, const char *name,
-    int add, uint64_t *towrite, uint64_t *tooverwrite);
+    int add, refcount_t *towrite, refcount_t *tooverwrite);
 
 /*
  * Create an attribute with the given name and value.

Modified: vendor-sys/illumos/dist/uts/common/fs/zfs/sys/zap_impl.h
==============================================================================
--- vendor-sys/illumos/dist/uts/common/fs/zfs/sys/zap_impl.h    Tue Jul 12 
11:36:22 2016        (r302650)
+++ vendor-sys/illumos/dist/uts/common/fs/zfs/sys/zap_impl.h    Tue Jul 12 
11:37:19 2016        (r302651)
@@ -20,7 +20,7 @@
  */
 /*
  * Copyright (c) 2005, 2010, Oracle and/or its affiliates. All rights reserved.
- * Copyright (c) 2013 by Delphix. All rights reserved.
+ * Copyright (c) 2013, 2015 by Delphix. All rights reserved.
  * Copyright (c) 2014 Spectra Logic Corporation, All rights reserved.
  * Copyright (c) 2014 Integros [integros.com]
  */
@@ -214,8 +214,8 @@ int fzap_lookup(zap_name_t *zn,
     uint64_t integer_size, uint64_t num_integers, void *buf,
     char *realname, int rn_len, boolean_t *normalization_conflictp);
 void fzap_prefetch(zap_name_t *zn);
-int fzap_count_write(zap_name_t *zn, int add, uint64_t *towrite,
-    uint64_t *tooverwrite);
+int fzap_count_write(zap_name_t *zn, int add, refcount_t *towrite,
+    refcount_t *tooverwrite);
 int fzap_add(zap_name_t *zn, uint64_t integer_size, uint64_t num_integers,
     const void *val, dmu_tx_t *tx);
 int fzap_update(zap_name_t *zn,

Modified: vendor-sys/illumos/dist/uts/common/fs/zfs/zap.c
==============================================================================
--- vendor-sys/illumos/dist/uts/common/fs/zfs/zap.c     Tue Jul 12 11:36:22 
2016        (r302650)
+++ vendor-sys/illumos/dist/uts/common/fs/zfs/zap.c     Tue Jul 12 11:37:19 
2016        (r302651)
@@ -1317,8 +1317,8 @@ fzap_get_stats(zap_t *zap, zap_stats_t *
 }
 
 int
-fzap_count_write(zap_name_t *zn, int add, uint64_t *towrite,
-    uint64_t *tooverwrite)
+fzap_count_write(zap_name_t *zn, int add, refcount_t *towrite,
+    refcount_t *tooverwrite)
 {
        zap_t *zap = zn->zn_zap;
        zap_leaf_t *l;
@@ -1328,9 +1328,11 @@ fzap_count_write(zap_name_t *zn, int add
         * Account for the header block of the fatzap.
         */
        if (!add && dmu_buf_freeable(zap->zap_dbuf)) {
-               *tooverwrite += zap->zap_dbuf->db_size;
+               (void) refcount_add_many(tooverwrite,
+                   zap->zap_dbuf->db_size, FTAG);
        } else {
-               *towrite += zap->zap_dbuf->db_size;
+               (void) refcount_add_many(towrite,
+                   zap->zap_dbuf->db_size, FTAG);
        }
 
        /*
@@ -1342,10 +1344,13 @@ fzap_count_write(zap_name_t *zn, int add
         *   could extend the table.
         */
        if (add) {
-               if (zap_f_phys(zap)->zap_ptrtbl.zt_blk == 0)
-                       *towrite += zap->zap_dbuf->db_size;
-               else
-                       *towrite += (zap->zap_dbuf->db_size * 3);
+               if (zap_f_phys(zap)->zap_ptrtbl.zt_blk == 0) {
+                       (void) refcount_add_many(towrite,
+                           zap->zap_dbuf->db_size, FTAG);
+               } else {
+                       (void) refcount_add_many(towrite,
+                           zap->zap_dbuf->db_size * 3, FTAG);
+               }
        }
 
        /*
@@ -1358,13 +1363,14 @@ fzap_count_write(zap_name_t *zn, int add
        }
 
        if (!add && dmu_buf_freeable(l->l_dbuf)) {
-               *tooverwrite += l->l_dbuf->db_size;
+               (void) refcount_add_many(tooverwrite, l->l_dbuf->db_size, FTAG);
        } else {
                /*
                 * If this an add operation, the leaf block could split.
                 * Hence, we need to account for an additional leaf block.
                 */
-               *towrite += (add ? 2 : 1) * l->l_dbuf->db_size;
+               (void) refcount_add_many(towrite,
+                   (add ? 2 : 1) * l->l_dbuf->db_size, FTAG);
        }
 
        zap_put_leaf(l);

Modified: vendor-sys/illumos/dist/uts/common/fs/zfs/zap_micro.c
==============================================================================
--- vendor-sys/illumos/dist/uts/common/fs/zfs/zap_micro.c       Tue Jul 12 
11:36:22 2016        (r302650)
+++ vendor-sys/illumos/dist/uts/common/fs/zfs/zap_micro.c       Tue Jul 12 
11:37:19 2016        (r302651)
@@ -20,7 +20,7 @@
  */
 /*
  * Copyright (c) 2005, 2010, Oracle and/or its affiliates. All rights reserved.
- * Copyright (c) 2011, 2014 by Delphix. All rights reserved.
+ * Copyright (c) 2011, 2015 by Delphix. All rights reserved.
  * Copyright (c) 2014 Spectra Logic Corporation, All rights reserved.
  * Copyright (c) 2014 Integros [integros.com]
  */
@@ -1363,7 +1363,7 @@ zap_get_stats(objset_t *os, uint64_t zap
 
 int
 zap_count_write(objset_t *os, uint64_t zapobj, const char *name, int add,
-    uint64_t *towrite, uint64_t *tooverwrite)
+    refcount_t *towrite, refcount_t *tooverwrite)
 {
        zap_t *zap;
        int err = 0;
@@ -1373,14 +1373,15 @@ zap_count_write(objset_t *os, uint64_t z
         * be affected in this operation. So, account for the worst case :
         * - 3 blocks overwritten: target leaf, ptrtbl block, header block
         * - 4 new blocks written if adding:
-        *      - 2 blocks for possibly split leaves,
-        *      - 2 grown ptrtbl blocks
+        *    - 2 blocks for possibly split leaves,
+        *    - 2 grown ptrtbl blocks
         *
         * This also accomodates the case where an add operation to a fairly
         * large microzap results in a promotion to fatzap.
         */
        if (name == NULL) {
-               *towrite += (3 + (add ? 4 : 0)) * SPA_OLD_MAXBLOCKSIZE;
+               (void) refcount_add_many(towrite,
+                   (3 + (add ? 4 : 0)) * SPA_OLD_MAXBLOCKSIZE, FTAG);
                return (err);
        }
 
@@ -1404,7 +1405,8 @@ zap_count_write(objset_t *os, uint64_t z
                        /*
                         * We treat this case as similar to (name == NULL)
                         */
-                       *towrite += (3 + (add ? 4 : 0)) * SPA_OLD_MAXBLOCKSIZE;
+                       (void) refcount_add_many(towrite,
+                           (3 + (add ? 4 : 0)) * SPA_OLD_MAXBLOCKSIZE, FTAG);
                }
        } else {
                /*
@@ -1422,13 +1424,17 @@ zap_count_write(objset_t *os, uint64_t z
                 * 4 new blocks written : 2 new split leaf, 2 grown
                 *                      ptrtbl blocks
                 */
-               if (dmu_buf_freeable(zap->zap_dbuf))
-                       *tooverwrite += MZAP_MAX_BLKSZ;
-               else
-                       *towrite += MZAP_MAX_BLKSZ;
+               if (dmu_buf_freeable(zap->zap_dbuf)) {
+                       (void) refcount_add_many(tooverwrite,
+                           MZAP_MAX_BLKSZ, FTAG);
+               } else {
+                       (void) refcount_add_many(towrite,
+                           MZAP_MAX_BLKSZ, FTAG);
+               }
 
                if (add) {
-                       *towrite += 4 * MZAP_MAX_BLKSZ;
+                       (void) refcount_add_many(towrite,
+                           4 * MZAP_MAX_BLKSZ, FTAG);
                }
        }
 
_______________________________________________
svn-src-all@freebsd.org mailing list
https://lists.freebsd.org/mailman/listinfo/svn-src-all
To unsubscribe, send any mail to "svn-src-all-unsubscr...@freebsd.org"

Reply via email to