module/zfs/sa.c

   1 /*
   2  * CDDL HEADER START
   3  *
   4  * The contents of this file are subject to the terms of the
   5  * Common Development and Distribution License (the "License").
   6  * You may not use this file except in compliance with the License.
   7  *
   8  * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
   9  * or http://www.opensolaris.org/os/licensing.
  10  * See the License for the specific language governing permissions
  11  * and limitations under the License.
  12  *
  13  * When distributing Covered Code, include this CDDL HEADER in each
  14  * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
  15  * If applicable, add the following below this CDDL HEADER, with the
  16  * fields enclosed by brackets "[]" replaced with your own identifying
  17  * information: Portions Copyright [yyyy] [name of copyright owner]
  18  *
  19  * CDDL HEADER END
  20  */
  21
  22 /*
  23  * Copyright (c) 2010, Oracle and/or its affiliates. All rights reserved.
  24  * Copyright (c) 2013 by Delphix. All rights reserved.
  25  */
  26
  27 #include <sys/zfs_context.h>
  28 #include <sys/types.h>
  29 #include <sys/param.h>
  30 #include <sys/systm.h>
  31 #include <sys/sysmacros.h>
  32 #include <sys/dmu.h>
  33 #include <sys/dmu_impl.h>
  34 #include <sys/dmu_objset.h>
  35 #include <sys/dbuf.h>
  36 #include <sys/dnode.h>
  37 #include <sys/zap.h>
  38 #include <sys/sa.h>
  39 #include <sys/sunddi.h>
  40 #include <sys/sa_impl.h>
  41 #include <sys/dnode.h>
  42 #include <sys/errno.h>
  43 #include <sys/zfs_context.h>
  44
  45 /*
  46  * ZFS System attributes:
  47  *
  48  * A generic mechanism to allow for arbitrary attributes
  49  * to be stored in a dnode.  The data will be stored in the bonus buffer of
  50  * the dnode and if necessary a special "spill" block will be used to handle
  51  * overflow situations.  The spill block will be sized to fit the data
  52  * from 512 - 128K.  When a spill block is used the BP (blkptr_t) for the
  53  * spill block is stored at the end of the current bonus buffer.  Any
  54  * attributes that would be in the way of the blkptr_t will be relocated
  55  * into the spill block.
  56  *
  57  * Attribute registration:
  58  *
  59  * Stored persistently on a per dataset basis
  60  * a mapping between attribute "string" names and their actual attribute
  61  * numeric values, length, and byteswap function.  The names are only used
  62  * during registration.  All  attributes are known by their unique attribute
  63  * id value.  If an attribute can have a variable size then the value
  64  * 0 will be used to indicate this.
  65  *
  66  * Attribute Layout:
  67  *
  68  * Attribute layouts are a way to compactly store multiple attributes, but
  69  * without taking the overhead associated with managing each attribute
  70  * individually.  Since you will typically have the same set of attributes
  71  * stored in the same order a single table will be used to represent that
  72  * layout.  The ZPL for example will usually have only about 10 different
  73  * layouts (regular files, device files, symlinks,
  74  * regular files + scanstamp, files/dir with extended attributes, and then
  75  * you have the possibility of all of those minus ACL, because it would
  76  * be kicked out into the spill block)
  77  *
  78  * Layouts are simply an array of the attributes and their
  79  * ordering i.e. [0, 1, 4, 5, 2]
  80  *
  81  * Each distinct layout is given a unique layout number and that is whats
  82  * stored in the header at the beginning of the SA data buffer.
  83  *
  84  * A layout only covers a single dbuf (bonus or spill).  If a set of
  85  * attributes is split up between the bonus buffer and a spill buffer then
  86  * two different layouts will be used.  This allows us to byteswap the
  87  * spill without looking at the bonus buffer and keeps the on disk format of
  88  * the bonus and spill buffer the same.
  89  *
  90  * Adding a single attribute will cause the entire set of attributes to
  91  * be rewritten and could result in a new layout number being constructed
  92  * as part of the rewrite if no such layout exists for the new set of
  93  * attribues.  The new attribute will be appended to the end of the already
  94  * existing attributes.
  95  *
  96  * Both the attribute registration and attribute layout information are
  97  * stored in normal ZAP attributes.  Their should be a small number of
  98  * known layouts and the set of attributes is assumed to typically be quite
  99  * small.
 100  *
 101  * The registered attributes and layout "table" information is maintained
 102  * in core and a special "sa_os_t" is attached to the objset_t.
 103  *
 104  * A special interface is provided to allow for quickly applying
 105  * a large set of attributes at once.  sa_replace_all_by_template() is
 106  * used to set an array of attributes.  This is used by the ZPL when
 107  * creating a brand new file.  The template that is passed into the function
 108  * specifies the attribute, size for variable length attributes, location of
 109  * data and special "data locator" function if the data isn't in a contiguous
 110  * location.
 111  *
 112  * Byteswap implications:
 113  *
 114  * Since the SA attributes are not entirely self describing we can't do
 115  * the normal byteswap processing.  The special ZAP layout attribute and
 116  * attribute registration attributes define the byteswap function and the
 117  * size of the attributes, unless it is variable sized.
 118  * The normal ZFS byteswapping infrastructure assumes you don't need
 119  * to read any objects in order to do the necessary byteswapping.  Whereas
 120  * SA attributes can only be properly byteswapped if the dataset is opened
 121  * and the layout/attribute ZAP attributes are available.  Because of this
 122  * the SA attributes will be byteswapped when they are first accessed by
 123  * the SA code that will read the SA data.
 124  */
 125
 126 typedef void (sa_iterfunc_t)(void *hdr, void *addr, sa_attr_type_t,
 127     uint16_t length, int length_idx, boolean_t, void *userp);
 128
 129 static int sa_build_index(sa_handle_t *hdl, sa_buf_type_t buftype);
 130 static void sa_idx_tab_hold(objset_t *os, sa_idx_tab_t *idx_tab);
 131 static void *sa_find_idx_tab(objset_t *os, dmu_object_type_t bonustype,
 132     void *data);
 133 static void sa_idx_tab_rele(objset_t *os, void *arg);
 134 static void sa_copy_data(sa_data_locator_t *func, void *start, void *target,
 135     int buflen);
 136 static int sa_modify_attrs(sa_handle_t *hdl, sa_attr_type_t newattr,
 137     sa_data_op_t action, sa_data_locator_t *locator, void *datastart,
 138     uint16_t buflen, dmu_tx_t *tx);
 139
 140 arc_byteswap_func_t sa_bswap_table[] = {
 141         byteswap_uint64_array,
 142         byteswap_uint32_array,
 143         byteswap_uint16_array,
 144         byteswap_uint8_array,
 145         zfs_acl_byteswap,
 146 };
 147
 148 #define SA_COPY_DATA(f, s, t, l) \
 149         { \
 150                 if (f == NULL) { \
 151                         if (l == 8) { \
 152                                 *(uint64_t *)t = *(uint64_t *)s; \
 153                         } else if (l == 16) { \
 154                                 *(uint64_t *)t = *(uint64_t *)s; \
 155                                 *(uint64_t *)((uintptr_t)t + 8) = \
 156                                     *(uint64_t *)((uintptr_t)s + 8); \
 157                         } else { \
 158                                 bcopy(s, t, l); \
 159                         } \
 160                 } else \
 161                         sa_copy_data(f, s, t, l); \
 162         }
 163
 164 /*
 165  * This table is fixed and cannot be changed.  Its purpose is to
 166  * allow the SA code to work with both old/new ZPL file systems.
 167  * It contains the list of legacy attributes.  These attributes aren't
 168  * stored in the "attribute" registry zap objects, since older ZPL file systems
 169  * won't have the registry.  Only objsets of type ZFS_TYPE_FILESYSTEM will
 170  * use this static table.
 171  */
 172 sa_attr_reg_t sa_legacy_attrs[] = {
 173         {"ZPL_ATIME", sizeof (uint64_t) * 2, SA_UINT64_ARRAY, 0},
 174         {"ZPL_MTIME", sizeof (uint64_t) * 2, SA_UINT64_ARRAY, 1},
 175         {"ZPL_CTIME", sizeof (uint64_t) * 2, SA_UINT64_ARRAY, 2},
 176         {"ZPL_CRTIME", sizeof (uint64_t) * 2, SA_UINT64_ARRAY, 3},
 177         {"ZPL_GEN", sizeof (uint64_t), SA_UINT64_ARRAY, 4},
 178         {"ZPL_MODE", sizeof (uint64_t), SA_UINT64_ARRAY, 5},
 179         {"ZPL_SIZE", sizeof (uint64_t), SA_UINT64_ARRAY, 6},
 180         {"ZPL_PARENT", sizeof (uint64_t), SA_UINT64_ARRAY, 7},
 181         {"ZPL_LINKS", sizeof (uint64_t), SA_UINT64_ARRAY, 8},
 182         {"ZPL_XATTR", sizeof (uint64_t), SA_UINT64_ARRAY, 9},
 183         {"ZPL_RDEV", sizeof (uint64_t), SA_UINT64_ARRAY, 10},
 184         {"ZPL_FLAGS", sizeof (uint64_t), SA_UINT64_ARRAY, 11},
 185         {"ZPL_UID", sizeof (uint64_t), SA_UINT64_ARRAY, 12},
 186         {"ZPL_GID", sizeof (uint64_t), SA_UINT64_ARRAY, 13},
 187         {"ZPL_PAD", sizeof (uint64_t) * 4, SA_UINT64_ARRAY, 14},
 188         {"ZPL_ZNODE_ACL", 88, SA_UINT8_ARRAY, 15},
 189 };
 190
 191 /*
 192  * This is only used for objects of type DMU_OT_ZNODE
 193  */
 194 sa_attr_type_t sa_legacy_zpl_layout[] = {
 195     0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15
 196 };
 197
 198 /*
 199  * Special dummy layout used for buffers with no attributes.
 200  */
 201 sa_attr_type_t sa_dummy_zpl_layout[] = { 0 };
 202
 203 static int sa_legacy_attr_count = 16;
 204 static kmem_cache_t *sa_cache = NULL;
 205 static kmem_cache_t *spill_cache = NULL;
 206
 207 /*ARGSUSED*/
 208 static int
 209 sa_cache_constructor(void *buf, void *unused, int kmflag)
 210 {
 211         sa_handle_t *hdl = buf;
 212
 213         hdl->sa_bonus_tab = NULL;
 214         hdl->sa_spill_tab = NULL;
 215         hdl->sa_os = NULL;
 216         hdl->sa_userp = NULL;
 217         hdl->sa_bonus = NULL;
 218         hdl->sa_spill = NULL;
 219         mutex_init(&hdl->sa_lock, NULL, MUTEX_DEFAULT, NULL);
 220         return (0);
 221 }
 222
 223 /*ARGSUSED*/
 224 static void
 225 sa_cache_destructor(void *buf, void *unused)
 226 {
 227         sa_handle_t *hdl = buf;
 228         mutex_destroy(&hdl->sa_lock);
 229 }
 230
 231 void
 232 sa_cache_init(void)
 233 {
 234         sa_cache = kmem_cache_create("sa_cache",
 235             sizeof (sa_handle_t), 0, sa_cache_constructor,
 236             sa_cache_destructor, NULL, NULL, NULL, 0);
 237         spill_cache = kmem_cache_create("spill_cache",
 238             SPA_MAXBLOCKSIZE, 0, NULL, NULL, NULL, NULL, NULL, 0);
 239 }
 240
 241 void
 242 sa_cache_fini(void)
 243 {
 244         if (sa_cache)
 245                 kmem_cache_destroy(sa_cache);
 246
 247         if (spill_cache)
 248                 kmem_cache_destroy(spill_cache);
 249 }
 250
 251 void *
 252 sa_spill_alloc(int flags)
 253 {
 254         return (kmem_cache_alloc(spill_cache, flags));
 255 }
 256
 257 void
 258 sa_spill_free(void *obj)
 259 {
 260         kmem_cache_free(spill_cache, obj);
 261 }
 262
 263 static int
 264 layout_num_compare(const void *arg1, const void *arg2)
 265 {
 266         const sa_lot_t *node1 = arg1;
 267         const sa_lot_t *node2 = arg2;
 268
 269         if (node1->lot_num > node2->lot_num)
 270                 return (1);
 271         else if (node1->lot_num < node2->lot_num)
 272                 return (-1);
 273         return (0);
 274 }
 275
 276 static int
 277 layout_hash_compare(const void *arg1, const void *arg2)
 278 {
 279         const sa_lot_t *node1 = arg1;
 280         const sa_lot_t *node2 = arg2;
 281
 282         if (node1->lot_hash > node2->lot_hash)
 283                 return (1);
 284         if (node1->lot_hash < node2->lot_hash)
 285                 return (-1);
 286         if (node1->lot_instance > node2->lot_instance)
 287                 return (1);
 288         if (node1->lot_instance < node2->lot_instance)
 289                 return (-1);
 290         return (0);
 291 }
 292
 293 boolean_t
 294 sa_layout_equal(sa_lot_t *tbf, sa_attr_type_t *attrs, int count)
 295 {
 296         int i;
 297
 298         if (count != tbf->lot_attr_count)
 299                 return (1);
 300
 301         for (i = 0; i != count; i++) {
 302                 if (attrs[i] != tbf->lot_attrs[i])
 303                         return (1);
 304         }
 305         return (0);
 306 }
 307
 308 #define SA_ATTR_HASH(attr) (zfs_crc64_table[(-1ULL ^ attr) & 0xFF])
 309
 310 static uint64_t
 311 sa_layout_info_hash(sa_attr_type_t *attrs, int attr_count)
 312 {
 313         int i;
 314         uint64_t crc = -1ULL;
 315
 316         for (i = 0; i != attr_count; i++)
 317                 crc ^= SA_ATTR_HASH(attrs[i]);
 318
 319         return (crc);
 320 }
 321
 322 static int
 323 sa_get_spill(sa_handle_t *hdl)
 324 {
 325         int rc;
 326         if (hdl->sa_spill == NULL) {
 327                 if ((rc = dmu_spill_hold_existing(hdl->sa_bonus, NULL,
 328                     &hdl->sa_spill)) == 0)
 329                         VERIFY(0 == sa_build_index(hdl, SA_SPILL));
 330         } else {
 331                 rc = 0;
 332         }
 333
 334         return (rc);
 335 }
 336
 337 /*
 338  * Main attribute lookup/update function
 339  * returns 0 for success or non zero for failures
 340  *
 341  * Operates on bulk array, first failure will abort further processing
 342  */
 343 int
 344 sa_attr_op(sa_handle_t *hdl, sa_bulk_attr_t *bulk, int count,
 345     sa_data_op_t data_op, dmu_tx_t *tx)
 346 {
 347         sa_os_t *sa = hdl->sa_os->os_sa;
 348         int i;
 349         int error = 0;
 350         sa_buf_type_t buftypes;
 351
 352         buftypes = 0;
 353
 354         ASSERT(count > 0);
 355         for (i = 0; i != count; i++) {
 356                 ASSERT(bulk[i].sa_attr <= hdl->sa_os->os_sa->sa_num_attrs);
 357
 358                 bulk[i].sa_addr = NULL;
 359                 /* First check the bonus buffer */
 360
 361                 if (hdl->sa_bonus_tab && TOC_ATTR_PRESENT(
 362                     hdl->sa_bonus_tab->sa_idx_tab[bulk[i].sa_attr])) {
 363                         SA_ATTR_INFO(sa, hdl->sa_bonus_tab,
 364                             SA_GET_HDR(hdl, SA_BONUS),
 365                             bulk[i].sa_attr, bulk[i], SA_BONUS, hdl);
 366                         if (tx && !(buftypes & SA_BONUS)) {
 367                                 dmu_buf_will_dirty(hdl->sa_bonus, tx);
 368                                 buftypes |= SA_BONUS;
 369                         }
 370                 }
 371                 if (bulk[i].sa_addr == NULL &&
 372                     ((error = sa_get_spill(hdl)) == 0)) {
 373                         if (TOC_ATTR_PRESENT(
 374                             hdl->sa_spill_tab->sa_idx_tab[bulk[i].sa_attr])) {
 375                                 SA_ATTR_INFO(sa, hdl->sa_spill_tab,
 376                                     SA_GET_HDR(hdl, SA_SPILL),
 377                                     bulk[i].sa_attr, bulk[i], SA_SPILL, hdl);
 378                                 if (tx && !(buftypes & SA_SPILL) &&
 379                                     bulk[i].sa_size == bulk[i].sa_length) {
 380                                         dmu_buf_will_dirty(hdl->sa_spill, tx);
 381                                         buftypes |= SA_SPILL;
 382                                 }
 383                         }
 384                 }
 385                 if (error && error != ENOENT) {
 386                         return ((error == ECKSUM) ? EIO : error);
 387                 }
 388
 389                 switch (data_op) {
 390                 case SA_LOOKUP:
 391                         if (bulk[i].sa_addr == NULL)
 392                                 return (SET_ERROR(ENOENT));
 393                         if (bulk[i].sa_data) {
 394                                 SA_COPY_DATA(bulk[i].sa_data_func,
 395                                     bulk[i].sa_addr, bulk[i].sa_data,
 396                                     bulk[i].sa_size);
 397                         }
 398                         continue;
 399
 400                 case SA_UPDATE:
 401                         /* existing rewrite of attr */
 402                         if (bulk[i].sa_addr &&
 403                             bulk[i].sa_size == bulk[i].sa_length) {
 404                                 SA_COPY_DATA(bulk[i].sa_data_func,
 405                                     bulk[i].sa_data, bulk[i].sa_addr,
 406                                     bulk[i].sa_length);
 407                                 continue;
 408                         } else if (bulk[i].sa_addr) { /* attr size change */
 409                                 error = sa_modify_attrs(hdl, bulk[i].sa_attr,
 410                                     SA_REPLACE, bulk[i].sa_data_func,
 411                                     bulk[i].sa_data, bulk[i].sa_length, tx);
 412                         } else { /* adding new attribute */
 413                                 error = sa_modify_attrs(hdl, bulk[i].sa_attr,
 414                                     SA_ADD, bulk[i].sa_data_func,
 415                                     bulk[i].sa_data, bulk[i].sa_length, tx);
 416                         }
 417                         if (error)
 418                                 return (error);
 419                         break;
 420                 default:
 421                         break;
 422                 }
 423         }
 424         return (error);
 425 }
 426
 427 static sa_lot_t *
 428 sa_add_layout_entry(objset_t *os, sa_attr_type_t *attrs, int attr_count,
 429     uint64_t lot_num, uint64_t hash, boolean_t zapadd, dmu_tx_t *tx)
 430 {
 431         sa_os_t *sa = os->os_sa;
 432         sa_lot_t *tb, *findtb;
 433         int i;
 434         avl_index_t loc;
 435
 436         ASSERT(MUTEX_HELD(&sa->sa_lock));
 437         tb = kmem_zalloc(sizeof (sa_lot_t), KM_PUSHPAGE);
 438         tb->lot_attr_count = attr_count;
 439         tb->lot_attrs = kmem_alloc(sizeof (sa_attr_type_t) * attr_count,
 440             KM_PUSHPAGE);
 441         bcopy(attrs, tb->lot_attrs, sizeof (sa_attr_type_t) * attr_count);
 442         tb->lot_num = lot_num;
 443         tb->lot_hash = hash;
 444         tb->lot_instance = 0;
 445
 446         if (zapadd) {
 447                 char attr_name[8];
 448
 449                 if (sa->sa_layout_attr_obj == 0) {
 450                         sa->sa_layout_attr_obj = zap_create_link(os,
 451                             DMU_OT_SA_ATTR_LAYOUTS,
 452                             sa->sa_master_obj, SA_LAYOUTS, tx);
 453                 }
 454
 455                 (void) snprintf(attr_name, sizeof (attr_name),
 456                     "%d", (int)lot_num);
 457                 VERIFY(0 == zap_update(os, os->os_sa->sa_layout_attr_obj,
 458                     attr_name, 2, attr_count, attrs, tx));
 459         }
 460
 461         list_create(&tb->lot_idx_tab, sizeof (sa_idx_tab_t),
 462             offsetof(sa_idx_tab_t, sa_next));
 463
 464         for (i = 0; i != attr_count; i++) {
 465                 if (sa->sa_attr_table[tb->lot_attrs[i]].sa_length == 0)
 466                         tb->lot_var_sizes++;
 467         }
 468
 469         avl_add(&sa->sa_layout_num_tree, tb);
 470
 471         /* verify we don't have a hash collision */
 472         if ((findtb = avl_find(&sa->sa_layout_hash_tree, tb, &loc)) != NULL) {
 473                 for (; findtb && findtb->lot_hash == hash;
 474                     findtb = AVL_NEXT(&sa->sa_layout_hash_tree, findtb)) {
 475                         if (findtb->lot_instance != tb->lot_instance)
 476                                 break;
 477                         tb->lot_instance++;
 478                 }
 479         }
 480         avl_add(&sa->sa_layout_hash_tree, tb);
 481         return (tb);
 482 }
 483
 484 static void
 485 sa_find_layout(objset_t *os, uint64_t hash, sa_attr_type_t *attrs,
 486     int count, dmu_tx_t *tx, sa_lot_t **lot)
 487 {
 488         sa_lot_t *tb, tbsearch;
 489         avl_index_t loc;
 490         sa_os_t *sa = os->os_sa;
 491         boolean_t found = B_FALSE;
 492
 493         mutex_enter(&sa->sa_lock);
 494         tbsearch.lot_hash = hash;
 495         tbsearch.lot_instance = 0;
 496         tb = avl_find(&sa->sa_layout_hash_tree, &tbsearch, &loc);
 497         if (tb) {
 498                 for (; tb && tb->lot_hash == hash;
 499                     tb = AVL_NEXT(&sa->sa_layout_hash_tree, tb)) {
 500                         if (sa_layout_equal(tb, attrs, count) == 0) {
 501                                 found = B_TRUE;
 502                                 break;
 503                         }
 504                 }
 505         }
 506         if (!found) {
 507                 tb = sa_add_layout_entry(os, attrs, count,
 508                     avl_numnodes(&sa->sa_layout_num_tree), hash, B_TRUE, tx);
 509         }
 510         mutex_exit(&sa->sa_lock);
 511         *lot = tb;
 512 }
 513
 514 static int
 515 sa_resize_spill(sa_handle_t *hdl, uint32_t size, dmu_tx_t *tx)
 516 {
 517         int error;
 518         uint32_t blocksize;
 519
 520         if (size == 0) {
 521                 blocksize = SPA_MINBLOCKSIZE;
 522         } else if (size > SPA_MAXBLOCKSIZE) {
 523                 ASSERT(0);
 524                 return (SET_ERROR(EFBIG));
 525         } else {
 526                 blocksize = P2ROUNDUP_TYPED(size, SPA_MINBLOCKSIZE, uint32_t);
 527         }
 528
 529         error = dbuf_spill_set_blksz(hdl->sa_spill, blocksize, tx);
 530         ASSERT(error == 0);
 531         return (error);
 532 }
 533
 534 static void
 535 sa_copy_data(sa_data_locator_t *func, void *datastart, void *target, int buflen)
 536 {
 537         if (func == NULL) {
 538                 bcopy(datastart, target, buflen);
 539         } else {
 540                 boolean_t start;
 541                 int bytes;
 542                 void *dataptr;
 543                 void *saptr = target;
 544                 uint32_t length;
 545
 546                 start = B_TRUE;
 547                 bytes = 0;
 548                 while (bytes < buflen) {
 549                         func(&dataptr, &length, buflen, start, datastart);
 550                         bcopy(dataptr, saptr, length);
 551                         saptr = (void *)((caddr_t)saptr + length);
 552                         bytes += length;
 553                         start = B_FALSE;
 554                 }
 555         }
 556 }
 557
 558 /*
 559  * Determine several different sizes
 560  * first the sa header size
 561  * the number of bytes to be stored
 562  * if spill would occur the index in the attribute array is returned
 563  *
 564  * the boolean will_spill will be set when spilling is necessary.  It
 565  * is only set when the buftype is SA_BONUS
 566  */
 567 static int
 568 sa_find_sizes(sa_os_t *sa, sa_bulk_attr_t *attr_desc, int attr_count,
 569     dmu_buf_t *db, sa_buf_type_t buftype, int *index, int *total,
 570     boolean_t *will_spill)
 571 {
 572         int var_size = 0;
 573         int i;
 574         int full_space;
 575         int hdrsize;
 576         int extra_hdrsize;
 577
 578         if (buftype == SA_BONUS && sa->sa_force_spill) {
 579                 *total = 0;
 580                 *index = 0;
 581                 *will_spill = B_TRUE;
 582                 return (0);
 583         }
 584
 585         *index = -1;
 586         *total = 0;
 587         *will_spill = B_FALSE;
 588
 589         extra_hdrsize = 0;
 590         hdrsize = (SA_BONUSTYPE_FROM_DB(db) == DMU_OT_ZNODE) ? 0 :
 591             sizeof (sa_hdr_phys_t);
 592
 593         full_space = (buftype == SA_BONUS) ? DN_MAX_BONUSLEN : db->db_size;
 594         ASSERT(IS_P2ALIGNED(full_space, 8));
 595
 596         for (i = 0; i != attr_count; i++) {
 597                 boolean_t is_var_sz, might_spill_here;
 598
 599                 *total = P2ROUNDUP(*total, 8);
 600                 *total += attr_desc[i].sa_length;
 601                 if (*will_spill)
 602                         continue;
 603
 604                 is_var_sz = (SA_REGISTERED_LEN(sa, attr_desc[i].sa_attr) == 0);
 605                 if (is_var_sz) {
 606                         var_size++;
 607                 }
 608
 609                 might_spill_here =
 610                     buftype == SA_BONUS && *index == -1 &&
 611                     (*total + P2ROUNDUP(hdrsize, 8)) >
 612                     (full_space - sizeof (blkptr_t));
 613
 614                 if (is_var_sz && var_size > 1) {
 615                         /*
 616                          * Don't worry that the spill block might overflow.
 617                          * It will be resized if needed in sa_build_layouts().
 618                          */
 619                         if (buftype == SA_SPILL ||
 620                             P2ROUNDUP(hdrsize + sizeof (uint16_t), 8) +
 621                             *total < full_space) {
 622                                 /*
 623                                  * Account for header space used by array of
 624                                  * optional sizes of variable-length attributes.
 625                                  * Record the extra header size in case this
 626                                  * increase needs to be reversed due to
 627                                  * spill-over.
 628                                  */
 629                                 hdrsize += sizeof (uint16_t);
 630                                 if (*index != -1 || might_spill_here)
 631                                         extra_hdrsize += sizeof (uint16_t);
 632                         } else {
 633                                 ASSERT(buftype == SA_BONUS);
 634                                 if (*index == -1)
 635                                         *index = i;
 636                                 *will_spill = B_TRUE;
 637                                 continue;
 638                         }
 639                 }
 640
 641                 /*
 642                  * find index of where spill *could* occur.
 643                  * Then continue to count of remainder attribute
 644                  * space.  The sum is used later for sizing bonus
 645                  * and spill buffer.
 646                  */
 647                 if (might_spill_here)
 648                         *index = i;
 649
 650                 if ((*total + P2ROUNDUP(hdrsize, 8)) > full_space &&
 651                     buftype == SA_BONUS)
 652                         *will_spill = B_TRUE;
 653         }
 654
 655         if (*will_spill)
 656                 hdrsize -= extra_hdrsize;
 657
 658         hdrsize = P2ROUNDUP(hdrsize, 8);
 659         return (hdrsize);
 660 }
 661
 662 #define BUF_SPACE_NEEDED(total, header) (total + header)
 663
 664 /*
 665  * Find layout that corresponds to ordering of attributes
 666  * If not found a new layout number is created and added to
 667  * persistent layout tables.
 668  */
 669 static int
 670 sa_build_layouts(sa_handle_t *hdl, sa_bulk_attr_t *attr_desc, int attr_count,
 671     dmu_tx_t *tx)
 672 {
 673         sa_os_t *sa = hdl->sa_os->os_sa;
 674         uint64_t hash;
 675         sa_buf_type_t buftype;
 676         sa_hdr_phys_t *sahdr;
 677         void *data_start;
 678         int buf_space;
 679         sa_attr_type_t *attrs, *attrs_start;
 680         int i, lot_count;
 681         int hdrsize;
 682         int spillhdrsize = 0;
 683         int used;
 684         dmu_object_type_t bonustype;
 685         sa_lot_t *lot;
 686         int len_idx;
 687         int spill_used;
 688         boolean_t spilling;
 689
 690         dmu_buf_will_dirty(hdl->sa_bonus, tx);
 691         bonustype = SA_BONUSTYPE_FROM_DB(hdl->sa_bonus);
 692
 693         /* first determine bonus header size and sum of all attributes */
 694         hdrsize = sa_find_sizes(sa, attr_desc, attr_count, hdl->sa_bonus,
 695             SA_BONUS, &i, &used, &spilling);
 696
 697         if (used > SPA_MAXBLOCKSIZE)
 698                 return (SET_ERROR(EFBIG));
 699
 700         VERIFY(0 == dmu_set_bonus(hdl->sa_bonus, spilling ?
 701             MIN(DN_MAX_BONUSLEN - sizeof (blkptr_t), used + hdrsize) :
 702             used + hdrsize, tx));
 703
 704         ASSERT((bonustype == DMU_OT_ZNODE && spilling == 0) ||
 705             bonustype == DMU_OT_SA);
 706
 707         /* setup and size spill buffer when needed */
 708         if (spilling) {
 709                 boolean_t dummy;
 710
 711                 if (hdl->sa_spill == NULL) {
 712                         VERIFY(dmu_spill_hold_by_bonus(hdl->sa_bonus, NULL,
 713                             &hdl->sa_spill) == 0);
 714                 }
 715                 dmu_buf_will_dirty(hdl->sa_spill, tx);
 716
 717                 spillhdrsize = sa_find_sizes(sa, &attr_desc[i],
 718                     attr_count - i, hdl->sa_spill, SA_SPILL, &i,
 719                     &spill_used, &dummy);
 720
 721                 if (spill_used > SPA_MAXBLOCKSIZE)
 722                         return (SET_ERROR(EFBIG));
 723
 724                 buf_space = hdl->sa_spill->db_size - spillhdrsize;
 725                 if (BUF_SPACE_NEEDED(spill_used, spillhdrsize) >
 726                     hdl->sa_spill->db_size)
 727                         VERIFY(0 == sa_resize_spill(hdl,
 728                             BUF_SPACE_NEEDED(spill_used, spillhdrsize), tx));
 729         }
 730
 731         /* setup starting pointers to lay down data */
 732         data_start = (void *)((uintptr_t)hdl->sa_bonus->db_data + hdrsize);
 733         sahdr = (sa_hdr_phys_t *)hdl->sa_bonus->db_data;
 734         buftype = SA_BONUS;
 735
 736         if (spilling)
 737                 buf_space = (sa->sa_force_spill) ?
 738                     0 : SA_BLKPTR_SPACE - hdrsize;
 739         else
 740                 buf_space = hdl->sa_bonus->db_size - hdrsize;
 741
 742         attrs_start = attrs = kmem_alloc(sizeof (sa_attr_type_t) * attr_count,
 743             KM_PUSHPAGE);
 744         lot_count = 0;
 745
 746         for (i = 0, len_idx = 0, hash = -1ULL; i != attr_count; i++) {
 747                 uint16_t length;
 748
 749                 ASSERT(IS_P2ALIGNED(data_start, 8));
 750                 ASSERT(IS_P2ALIGNED(buf_space, 8));
 751                 attrs[i] = attr_desc[i].sa_attr;
 752                 length = SA_REGISTERED_LEN(sa, attrs[i]);
 753                 if (length == 0)
 754                         length = attr_desc[i].sa_length;
 755
 756                 if (buf_space < length) {  /* switch to spill buffer */
 757                         VERIFY(spilling);
 758                         VERIFY(bonustype == DMU_OT_SA);
 759                         if (buftype == SA_BONUS && !sa->sa_force_spill) {
 760                                 sa_find_layout(hdl->sa_os, hash, attrs_start,
 761                                     lot_count, tx, &lot);
 762                                 SA_SET_HDR(sahdr, lot->lot_num, hdrsize);
 763                         }
 764
 765                         buftype = SA_SPILL;
 766                         hash = -1ULL;
 767                         len_idx = 0;
 768
 769                         sahdr = (sa_hdr_phys_t *)hdl->sa_spill->db_data;
 770                         sahdr->sa_magic = SA_MAGIC;
 771                         data_start = (void *)((uintptr_t)sahdr +
 772                             spillhdrsize);
 773                         attrs_start = &attrs[i];
 774                         buf_space = hdl->sa_spill->db_size - spillhdrsize;
 775                         lot_count = 0;
 776                 }
 777                 hash ^= SA_ATTR_HASH(attrs[i]);
 778                 attr_desc[i].sa_addr = data_start;
 779                 attr_desc[i].sa_size = length;
 780                 SA_COPY_DATA(attr_desc[i].sa_data_func, attr_desc[i].sa_data,
 781                     data_start, length);
 782                 if (sa->sa_attr_table[attrs[i]].sa_length == 0) {
 783                         sahdr->sa_lengths[len_idx++] = length;
 784                 }
 785                 data_start = (void *)P2ROUNDUP(((uintptr_t)data_start +
 786                     length), 8);
 787                 buf_space -= P2ROUNDUP(length, 8);
 788                 lot_count++;
 789         }
 790
 791         sa_find_layout(hdl->sa_os, hash, attrs_start, lot_count, tx, &lot);
 792
 793         /*
 794          * Verify that old znodes always have layout number 0.
 795          * Must be DMU_OT_SA for arbitrary layouts
 796          */
 797         VERIFY((bonustype == DMU_OT_ZNODE && lot->lot_num == 0) ||
 798             (bonustype == DMU_OT_SA && lot->lot_num > 1));
 799
 800         if (bonustype == DMU_OT_SA) {
 801                 SA_SET_HDR(sahdr, lot->lot_num,
 802                     buftype == SA_BONUS ? hdrsize : spillhdrsize);
 803         }
 804
 805         kmem_free(attrs, sizeof (sa_attr_type_t) * attr_count);
 806         if (hdl->sa_bonus_tab) {
 807                 sa_idx_tab_rele(hdl->sa_os, hdl->sa_bonus_tab);
 808                 hdl->sa_bonus_tab = NULL;
 809         }
 810         if (!sa->sa_force_spill)
 811                 VERIFY(0 == sa_build_index(hdl, SA_BONUS));
 812         if (hdl->sa_spill) {
 813                 sa_idx_tab_rele(hdl->sa_os, hdl->sa_spill_tab);
 814                 if (!spilling) {
 815                         /*
 816                          * remove spill block that is no longer needed.
 817                          */
 818                         dmu_buf_rele(hdl->sa_spill, NULL);
 819                         hdl->sa_spill = NULL;
 820                         hdl->sa_spill_tab = NULL;
 821                         VERIFY(0 == dmu_rm_spill(hdl->sa_os,
 822                             sa_handle_object(hdl), tx));
 823                 } else {
 824                         VERIFY(0 == sa_build_index(hdl, SA_SPILL));
 825                 }
 826         }
 827
 828         return (0);
 829 }
 830
 831 static void
 832 sa_free_attr_table(sa_os_t *sa)
 833 {
 834         int i;
 835
 836         if (sa->sa_attr_table == NULL)
 837                 return;
 838
 839         for (i = 0; i != sa->sa_num_attrs; i++) {
 840                 if (sa->sa_attr_table[i].sa_name)
 841                         kmem_free(sa->sa_attr_table[i].sa_name,
 842                             strlen(sa->sa_attr_table[i].sa_name) + 1);
 843         }
 844
 845         kmem_free(sa->sa_attr_table,
 846             sizeof (sa_attr_table_t) * sa->sa_num_attrs);
 847
 848         sa->sa_attr_table = NULL;
 849 }
 850
 851 static int
 852 sa_attr_table_setup(objset_t *os, sa_attr_reg_t *reg_attrs, int count)
 853 {
 854         sa_os_t *sa = os->os_sa;
 855         uint64_t sa_attr_count = 0;
 856         uint64_t sa_reg_count = 0;
 857         int error = 0;
 858         uint64_t attr_value;
 859         sa_attr_table_t *tb;
 860         zap_cursor_t zc;
 861         zap_attribute_t za;
 862         int registered_count = 0;
 863         int i;
 864         dmu_objset_type_t ostype = dmu_objset_type(os);
 865
 866         sa->sa_user_table =
 867             kmem_zalloc(count * sizeof (sa_attr_type_t), KM_PUSHPAGE);
 868         sa->sa_user_table_sz = count * sizeof (sa_attr_type_t);
 869
 870         if (sa->sa_reg_attr_obj != 0) {
 871                 error = zap_count(os, sa->sa_reg_attr_obj,
 872                     &sa_attr_count);
 873
 874                 /*
 875                  * Make sure we retrieved a count and that it isn't zero
 876                  */
 877                 if (error || (error == 0 && sa_attr_count == 0)) {
 878                         if (error == 0)
 879                                 error = SET_ERROR(EINVAL);
 880                         goto bail;
 881                 }
 882                 sa_reg_count = sa_attr_count;
 883         }
 884
 885         if (ostype == DMU_OST_ZFS && sa_attr_count == 0)
 886                 sa_attr_count += sa_legacy_attr_count;
 887
 888         /* Allocate attribute numbers for attributes that aren't registered */
 889         for (i = 0; i != count; i++) {
 890                 boolean_t found = B_FALSE;
 891                 int j;
 892
 893                 if (ostype == DMU_OST_ZFS) {
 894                         for (j = 0; j != sa_legacy_attr_count; j++) {
 895                                 if (strcmp(reg_attrs[i].sa_name,
 896                                     sa_legacy_attrs[j].sa_name) == 0) {
 897                                         sa->sa_user_table[i] =
 898                                             sa_legacy_attrs[j].sa_attr;
 899                                         found = B_TRUE;
 900                                 }
 901                         }
 902                 }
 903                 if (found)
 904                         continue;
 905
 906                 if (sa->sa_reg_attr_obj)
 907                         error = zap_lookup(os, sa->sa_reg_attr_obj,
 908                             reg_attrs[i].sa_name, 8, 1, &attr_value);
 909                 else
 910                         error = SET_ERROR(ENOENT);
 911                 switch (error) {
 912                 case ENOENT:
 913                         sa->sa_user_table[i] = (sa_attr_type_t)sa_attr_count;
 914                         sa_attr_count++;
 915                         break;
 916                 case 0:
 917                         sa->sa_user_table[i] = ATTR_NUM(attr_value);
 918                         break;
 919                 default:
 920                         goto bail;
 921                 }
 922         }
 923
 924         sa->sa_num_attrs = sa_attr_count;
 925         tb = sa->sa_attr_table =
 926             kmem_zalloc(sizeof (sa_attr_table_t) * sa_attr_count, KM_PUSHPAGE);
 927
 928         /*
 929          * Attribute table is constructed from requested attribute list,
 930          * previously foreign registered attributes, and also the legacy
 931          * ZPL set of attributes.
 932          */
 933
 934         if (sa->sa_reg_attr_obj) {
 935                 for (zap_cursor_init(&zc, os, sa->sa_reg_attr_obj);
 936                     (error = zap_cursor_retrieve(&zc, &za)) == 0;
 937                     zap_cursor_advance(&zc)) {
 938                         uint64_t value;
 939                         value  = za.za_first_integer;
 940
 941                         registered_count++;
 942                         tb[ATTR_NUM(value)].sa_attr = ATTR_NUM(value);
 943                         tb[ATTR_NUM(value)].sa_length = ATTR_LENGTH(value);
 944                         tb[ATTR_NUM(value)].sa_byteswap = ATTR_BSWAP(value);
 945                         tb[ATTR_NUM(value)].sa_registered = B_TRUE;
 946
 947                         if (tb[ATTR_NUM(value)].sa_name) {
 948                                 continue;
 949                         }
 950                         tb[ATTR_NUM(value)].sa_name =
 951                             kmem_zalloc(strlen(za.za_name) +1, KM_PUSHPAGE);
 952                         (void) strlcpy(tb[ATTR_NUM(value)].sa_name, za.za_name,
 953                             strlen(za.za_name) +1);
 954                 }
 955                 zap_cursor_fini(&zc);
 956                 /*
 957                  * Make sure we processed the correct number of registered
 958                  * attributes
 959                  */
 960                 if (registered_count != sa_reg_count) {
 961                         ASSERT(error != 0);
 962                         goto bail;
 963                 }
 964
 965         }
 966
 967         if (ostype == DMU_OST_ZFS) {
 968                 for (i = 0; i != sa_legacy_attr_count; i++) {
 969                         if (tb[i].sa_name)
 970                                 continue;
 971                         tb[i].sa_attr = sa_legacy_attrs[i].sa_attr;
 972                         tb[i].sa_length = sa_legacy_attrs[i].sa_length;
 973                         tb[i].sa_byteswap = sa_legacy_attrs[i].sa_byteswap;
 974                         tb[i].sa_registered = B_FALSE;
 975                         tb[i].sa_name =
 976                             kmem_zalloc(strlen(sa_legacy_attrs[i].sa_name) +1,
 977                             KM_PUSHPAGE);
 978                         (void) strlcpy(tb[i].sa_name,
 979                             sa_legacy_attrs[i].sa_name,
 980                             strlen(sa_legacy_attrs[i].sa_name) + 1);
 981                 }
 982         }
 983
 984         for (i = 0; i != count; i++) {
 985                 sa_attr_type_t attr_id;
 986
 987                 attr_id = sa->sa_user_table[i];
 988                 if (tb[attr_id].sa_name)
 989                         continue;
 990
 991                 tb[attr_id].sa_length = reg_attrs[i].sa_length;
 992                 tb[attr_id].sa_byteswap = reg_attrs[i].sa_byteswap;
 993                 tb[attr_id].sa_attr = attr_id;
 994                 tb[attr_id].sa_name =
 995                     kmem_zalloc(strlen(reg_attrs[i].sa_name) + 1, KM_PUSHPAGE);
 996                 (void) strlcpy(tb[attr_id].sa_name, reg_attrs[i].sa_name,
 997                     strlen(reg_attrs[i].sa_name) + 1);
 998         }
 999
1000         sa->sa_need_attr_registration =
1001             (sa_attr_count != registered_count);
1002
1003         return (0);
1004 bail:
1005         kmem_free(sa->sa_user_table, count * sizeof (sa_attr_type_t));
1006         sa->sa_user_table = NULL;
1007         sa_free_attr_table(sa);
1008         return ((error != 0) ? error : EINVAL);
1009 }
1010
1011 int
1012 sa_setup(objset_t *os, uint64_t sa_obj, sa_attr_reg_t *reg_attrs, int count,
1013     sa_attr_type_t **user_table)
1014 {
1015         zap_cursor_t zc;
1016         zap_attribute_t za;
1017         sa_os_t *sa;
1018         dmu_objset_type_t ostype = dmu_objset_type(os);
1019         sa_attr_type_t *tb;
1020         int error;
1021
1022         mutex_enter(&os->os_user_ptr_lock);
1023         if (os->os_sa) {
1024                 mutex_enter(&os->os_sa->sa_lock);
1025                 mutex_exit(&os->os_user_ptr_lock);
1026                 tb = os->os_sa->sa_user_table;
1027                 mutex_exit(&os->os_sa->sa_lock);
1028                 *user_table = tb;
1029                 return (0);
1030         }
1031
1032         sa = kmem_zalloc(sizeof (sa_os_t), KM_PUSHPAGE);
1033         mutex_init(&sa->sa_lock, NULL, MUTEX_DEFAULT, NULL);
1034         sa->sa_master_obj = sa_obj;
1035
1036         os->os_sa = sa;
1037         mutex_enter(&sa->sa_lock);
1038         mutex_exit(&os->os_user_ptr_lock);
1039         avl_create(&sa->sa_layout_num_tree, layout_num_compare,
1040             sizeof (sa_lot_t), offsetof(sa_lot_t, lot_num_node));
1041         avl_create(&sa->sa_layout_hash_tree, layout_hash_compare,
1042             sizeof (sa_lot_t), offsetof(sa_lot_t, lot_hash_node));
1043
1044         if (sa_obj) {
1045                 error = zap_lookup(os, sa_obj, SA_LAYOUTS,
1046                     8, 1, &sa->sa_layout_attr_obj);
1047                 if (error != 0 && error != ENOENT)
1048                         goto fail;
1049                 error = zap_lookup(os, sa_obj, SA_REGISTRY,
1050                     8, 1, &sa->sa_reg_attr_obj);
1051                 if (error != 0 && error != ENOENT)
1052                         goto fail;
1053         }
1054
1055         if ((error = sa_attr_table_setup(os, reg_attrs, count)) != 0)
1056                 goto fail;
1057
1058         if (sa->sa_layout_attr_obj != 0) {
1059                 uint64_t layout_count;
1060
1061                 error = zap_count(os, sa->sa_layout_attr_obj,
1062                     &layout_count);
1063
1064                 /*
1065                  * Layout number count should be > 0
1066                  */
1067                 if (error || (error == 0 && layout_count == 0)) {
1068                         if (error == 0)
1069                                 error = SET_ERROR(EINVAL);
1070                         goto fail;
1071                 }
1072
1073                 for (zap_cursor_init(&zc, os, sa->sa_layout_attr_obj);
1074                     (error = zap_cursor_retrieve(&zc, &za)) == 0;
1075                     zap_cursor_advance(&zc)) {
1076                         sa_attr_type_t *lot_attrs;
1077                         uint64_t lot_num;
1078
1079                         lot_attrs = kmem_zalloc(sizeof (sa_attr_type_t) *
1080                             za.za_num_integers, KM_PUSHPAGE);
1081
1082                         if ((error = (zap_lookup(os, sa->sa_layout_attr_obj,
1083                             za.za_name, 2, za.za_num_integers,
1084                             lot_attrs))) != 0) {
1085                                 kmem_free(lot_attrs, sizeof (sa_attr_type_t) *
1086                                     za.za_num_integers);
1087                                 break;
1088                         }
1089                         VERIFY(ddi_strtoull(za.za_name, NULL, 10,
1090                             (unsigned long long *)&lot_num) == 0);
1091
1092                         (void) sa_add_layout_entry(os, lot_attrs,
1093                             za.za_num_integers, lot_num,
1094                             sa_layout_info_hash(lot_attrs,
1095                             za.za_num_integers), B_FALSE, NULL);
1096                         kmem_free(lot_attrs, sizeof (sa_attr_type_t) *
1097                             za.za_num_integers);
1098                 }
1099                 zap_cursor_fini(&zc);
1100
1101                 /*
1102                  * Make sure layout count matches number of entries added
1103                  * to AVL tree
1104                  */
1105                 if (avl_numnodes(&sa->sa_layout_num_tree) != layout_count) {
1106                         ASSERT(error != 0);
1107                         goto fail;
1108                 }
1109         }
1110
1111         /* Add special layout number for old ZNODES */
1112         if (ostype == DMU_OST_ZFS) {
1113                 (void) sa_add_layout_entry(os, sa_legacy_zpl_layout,
1114                     sa_legacy_attr_count, 0,
1115                     sa_layout_info_hash(sa_legacy_zpl_layout,
1116                     sa_legacy_attr_count), B_FALSE, NULL);
1117
1118                 (void) sa_add_layout_entry(os, sa_dummy_zpl_layout, 0, 1,
1119                     0, B_FALSE, NULL);
1120         }
1121         *user_table = os->os_sa->sa_user_table;
1122         mutex_exit(&sa->sa_lock);
1123         return (0);
1124 fail:
1125         os->os_sa = NULL;
1126         sa_free_attr_table(sa);
1127         if (sa->sa_user_table)
1128                 kmem_free(sa->sa_user_table, sa->sa_user_table_sz);
1129         mutex_exit(&sa->sa_lock);
1130         kmem_free(sa, sizeof (sa_os_t));
1131         return ((error == ECKSUM) ? EIO : error);
1132 }
1133
1134 void
1135 sa_tear_down(objset_t *os)
1136 {
1137         sa_os_t *sa = os->os_sa;
1138         sa_lot_t *layout;
1139         void *cookie;
1140
1141         kmem_free(sa->sa_user_table, sa->sa_user_table_sz);
1142
1143         /* Free up attr table */
1144
1145         sa_free_attr_table(sa);
1146
1147         cookie = NULL;
1148         while ((layout =
1149             avl_destroy_nodes(&sa->sa_layout_hash_tree, &cookie))) {
1150                 sa_idx_tab_t *tab;
1151                 while ((tab = list_head(&layout->lot_idx_tab))) {
1152                         ASSERT(refcount_count(&tab->sa_refcount));
1153                         sa_idx_tab_rele(os, tab);
1154                 }
1155         }
1156
1157         cookie = NULL;
1158         while ((layout = avl_destroy_nodes(&sa->sa_layout_num_tree, &cookie))) {
1159                 kmem_free(layout->lot_attrs,
1160                     sizeof (sa_attr_type_t) * layout->lot_attr_count);
1161                 kmem_free(layout, sizeof (sa_lot_t));
1162         }
1163
1164         avl_destroy(&sa->sa_layout_hash_tree);
1165         avl_destroy(&sa->sa_layout_num_tree);
1166
1167         kmem_free(sa, sizeof (sa_os_t));
1168         os->os_sa = NULL;
1169 }
1170
1171 void
1172 sa_build_idx_tab(void *hdr, void *attr_addr, sa_attr_type_t attr,
1173     uint16_t length, int length_idx, boolean_t var_length, void *userp)
1174 {
1175         sa_idx_tab_t *idx_tab = userp;
1176
1177         if (var_length) {
1178                 ASSERT(idx_tab->sa_variable_lengths);
1179                 idx_tab->sa_variable_lengths[length_idx] = length;
1180         }
1181         TOC_ATTR_ENCODE(idx_tab->sa_idx_tab[attr], length_idx,
1182             (uint32_t)((uintptr_t)attr_addr - (uintptr_t)hdr));
1183 }
1184
1185 static void
1186 sa_attr_iter(objset_t *os, sa_hdr_phys_t *hdr, dmu_object_type_t type,
1187     sa_iterfunc_t func, sa_lot_t *tab, void *userp)
1188 {
1189         void *data_start;
1190         sa_lot_t *tb = tab;
1191         sa_lot_t search;
1192         avl_index_t loc;
1193         sa_os_t *sa = os->os_sa;
1194         int i;
1195         uint16_t *length_start = NULL;
1196         uint8_t length_idx = 0;
1197
1198         if (tab == NULL) {
1199                 search.lot_num = SA_LAYOUT_NUM(hdr, type);
1200                 tb = avl_find(&sa->sa_layout_num_tree, &search, &loc);
1201                 ASSERT(tb);
1202         }
1203
1204         if (IS_SA_BONUSTYPE(type)) {
1205                 data_start = (void *)P2ROUNDUP(((uintptr_t)hdr +
1206                     offsetof(sa_hdr_phys_t, sa_lengths) +
1207                     (sizeof (uint16_t) * tb->lot_var_sizes)), 8);
1208                 length_start = hdr->sa_lengths;
1209         } else {
1210                 data_start = hdr;
1211         }
1212
1213         for (i = 0; i != tb->lot_attr_count; i++) {
1214                 int attr_length, reg_length;
1215                 uint8_t idx_len;
1216
1217                 reg_length = sa->sa_attr_table[tb->lot_attrs[i]].sa_length;
1218                 if (reg_length) {
1219                         attr_length = reg_length;
1220                         idx_len = 0;
1221                 } else {
1222                         attr_length = length_start[length_idx];
1223                         idx_len = length_idx++;
1224                 }
1225
1226                 func(hdr, data_start, tb->lot_attrs[i], attr_length,
1227                     idx_len, reg_length == 0 ? B_TRUE : B_FALSE, userp);
1228
1229                 data_start = (void *)P2ROUNDUP(((uintptr_t)data_start +
1230                     attr_length), 8);
1231         }
1232 }
1233
1234 /*ARGSUSED*/
1235 void
1236 sa_byteswap_cb(void *hdr, void *attr_addr, sa_attr_type_t attr,
1237     uint16_t length, int length_idx, boolean_t variable_length, void *userp)
1238 {
1239         sa_handle_t *hdl = userp;
1240         sa_os_t *sa = hdl->sa_os->os_sa;
1241
1242         sa_bswap_table[sa->sa_attr_table[attr].sa_byteswap](attr_addr, length);
1243 }
1244
1245 void
1246 sa_byteswap(sa_handle_t *hdl, sa_buf_type_t buftype)
1247 {
1248         sa_hdr_phys_t *sa_hdr_phys = SA_GET_HDR(hdl, buftype);
1249         dmu_buf_impl_t *db;
1250         int num_lengths = 1;
1251         int i;
1252         ASSERTV(sa_os_t *sa = hdl->sa_os->os_sa);
1253
1254         ASSERT(MUTEX_HELD(&sa->sa_lock));
1255         if (sa_hdr_phys->sa_magic == SA_MAGIC)
1256                 return;
1257
1258         db = SA_GET_DB(hdl, buftype);
1259
1260         if (buftype == SA_SPILL) {
1261                 arc_release(db->db_buf, NULL);
1262                 arc_buf_thaw(db->db_buf);
1263         }
1264
1265         sa_hdr_phys->sa_magic = BSWAP_32(sa_hdr_phys->sa_magic);
1266         sa_hdr_phys->sa_layout_info = BSWAP_16(sa_hdr_phys->sa_layout_info);
1267
1268         /*
1269          * Determine number of variable lenghts in header
1270          * The standard 8 byte header has one for free and a
1271          * 16 byte header would have 4 + 1;
1272          */
1273         if (SA_HDR_SIZE(sa_hdr_phys) > 8)
1274                 num_lengths += (SA_HDR_SIZE(sa_hdr_phys) - 8) >> 1;
1275         for (i = 0; i != num_lengths; i++)
1276                 sa_hdr_phys->sa_lengths[i] =
1277                     BSWAP_16(sa_hdr_phys->sa_lengths[i]);
1278
1279         sa_attr_iter(hdl->sa_os, sa_hdr_phys, DMU_OT_SA,
1280             sa_byteswap_cb, NULL, hdl);
1281
1282         if (buftype == SA_SPILL)
1283                 arc_buf_freeze(((dmu_buf_impl_t *)hdl->sa_spill)->db_buf);
1284 }
1285
1286 static int
1287 sa_build_index(sa_handle_t *hdl, sa_buf_type_t buftype)
1288 {
1289         sa_hdr_phys_t *sa_hdr_phys;
1290         dmu_buf_impl_t *db = SA_GET_DB(hdl, buftype);
1291         dmu_object_type_t bonustype = SA_BONUSTYPE_FROM_DB(db);
1292         sa_os_t *sa = hdl->sa_os->os_sa;
1293         sa_idx_tab_t *idx_tab;
1294
1295         sa_hdr_phys = SA_GET_HDR(hdl, buftype);
1296
1297         mutex_enter(&sa->sa_lock);
1298
1299         /* Do we need to byteswap? */
1300
1301         /* only check if not old znode */
1302         if (IS_SA_BONUSTYPE(bonustype) && sa_hdr_phys->sa_magic != SA_MAGIC &&
1303             sa_hdr_phys->sa_magic != 0) {
1304                 VERIFY(BSWAP_32(sa_hdr_phys->sa_magic) == SA_MAGIC);
1305                 sa_byteswap(hdl, buftype);
1306         }
1307
1308         idx_tab = sa_find_idx_tab(hdl->sa_os, bonustype, sa_hdr_phys);
1309
1310         if (buftype == SA_BONUS)
1311                 hdl->sa_bonus_tab = idx_tab;
1312         else
1313                 hdl->sa_spill_tab = idx_tab;
1314
1315         mutex_exit(&sa->sa_lock);
1316         return (0);
1317 }
1318
1319 /*ARGSUSED*/
1320 void
1321 sa_evict(dmu_buf_t *db, void *sap)
1322 {
1323         panic("evicting sa dbuf %p\n", (void *)db);
1324 }
1325
1326 static void
1327 sa_idx_tab_rele(objset_t *os, void *arg)
1328 {
1329         sa_os_t *sa = os->os_sa;
1330         sa_idx_tab_t *idx_tab = arg;
1331
1332         if (idx_tab == NULL)
1333                 return;
1334
1335         mutex_enter(&sa->sa_lock);
1336         if (refcount_remove(&idx_tab->sa_refcount, NULL) == 0) {
1337                 list_remove(&idx_tab->sa_layout->lot_idx_tab, idx_tab);
1338                 if (idx_tab->sa_variable_lengths)
1339                         kmem_free(idx_tab->sa_variable_lengths,
1340                             sizeof (uint16_t) *
1341                             idx_tab->sa_layout->lot_var_sizes);
1342                 refcount_destroy(&idx_tab->sa_refcount);
1343                 kmem_free(idx_tab->sa_idx_tab,
1344                     sizeof (uint32_t) * sa->sa_num_attrs);
1345                 kmem_free(idx_tab, sizeof (sa_idx_tab_t));
1346         }
1347         mutex_exit(&sa->sa_lock);
1348 }
1349
1350 static void
1351 sa_idx_tab_hold(objset_t *os, sa_idx_tab_t *idx_tab)
1352 {
1353         ASSERTV(sa_os_t *sa = os->os_sa);
1354
1355         ASSERT(MUTEX_HELD(&sa->sa_lock));
1356         (void) refcount_add(&idx_tab->sa_refcount, NULL);
1357 }
1358
1359 void
1360 sa_spill_rele(sa_handle_t *hdl)
1361 {
1362         mutex_enter(&hdl->sa_lock);
1363         if (hdl->sa_spill) {
1364                 sa_idx_tab_rele(hdl->sa_os, hdl->sa_spill_tab);
1365                 dmu_buf_rele(hdl->sa_spill, NULL);
1366                 hdl->sa_spill = NULL;
1367                 hdl->sa_spill_tab = NULL;
1368         }
1369         mutex_exit(&hdl->sa_lock);
1370 }
1371
1372 void
1373 sa_handle_destroy(sa_handle_t *hdl)
1374 {
1375         mutex_enter(&hdl->sa_lock);
1376         (void) dmu_buf_update_user((dmu_buf_t *)hdl->sa_bonus, hdl,
1377             NULL, NULL, NULL);
1378
1379         if (hdl->sa_bonus_tab) {
1380                 sa_idx_tab_rele(hdl->sa_os, hdl->sa_bonus_tab);
1381                 hdl->sa_bonus_tab = NULL;
1382         }
1383         if (hdl->sa_spill_tab) {
1384                 sa_idx_tab_rele(hdl->sa_os, hdl->sa_spill_tab);
1385                 hdl->sa_spill_tab = NULL;
1386         }
1387
1388         dmu_buf_rele(hdl->sa_bonus, NULL);
1389
1390         if (hdl->sa_spill)
1391                 dmu_buf_rele((dmu_buf_t *)hdl->sa_spill, NULL);
1392         mutex_exit(&hdl->sa_lock);
1393
1394         kmem_cache_free(sa_cache, hdl);
1395 }
1396
1397 int
1398 sa_handle_get_from_db(objset_t *os, dmu_buf_t *db, void *userp,
1399     sa_handle_type_t hdl_type, sa_handle_t **handlepp)
1400 {
1401         int error = 0;
1402         sa_handle_t *handle;
1403 #ifdef ZFS_DEBUG
1404         dmu_object_info_t doi;
1405
1406         dmu_object_info_from_db(db, &doi);
1407         ASSERT(doi.doi_bonus_type == DMU_OT_SA ||
1408             doi.doi_bonus_type == DMU_OT_ZNODE);
1409 #endif
1410         /* find handle, if it exists */
1411         /* if one doesn't exist then create a new one, and initialize it */
1412
1413         handle = (hdl_type == SA_HDL_SHARED) ? dmu_buf_get_user(db) : NULL;
1414         if (handle == NULL) {
1415                 sa_handle_t *newhandle;
1416                 handle = kmem_cache_alloc(sa_cache, KM_SLEEP);
1417                 handle->sa_userp = userp;
1418                 handle->sa_bonus = db;
1419                 handle->sa_os = os;
1420                 handle->sa_spill = NULL;
1421
1422                 error = sa_build_index(handle, SA_BONUS);
1423                 newhandle = (hdl_type == SA_HDL_SHARED) ?
1424                     dmu_buf_set_user_ie(db, handle,
1425                     NULL, sa_evict) : NULL;
1426
1427                 if (newhandle != NULL) {
1428                         kmem_cache_free(sa_cache, handle);
1429                         handle = newhandle;
1430                 }
1431         }
1432         *handlepp = handle;
1433
1434         return (error);
1435 }
1436
1437 int
1438 sa_handle_get(objset_t *objset, uint64_t objid, void *userp,
1439     sa_handle_type_t hdl_type, sa_handle_t **handlepp)
1440 {
1441         dmu_buf_t *db;
1442         int error;
1443
1444         if ((error = dmu_bonus_hold(objset, objid, NULL, &db)))
1445                 return (error);
1446
1447         return (sa_handle_get_from_db(objset, db, userp, hdl_type,
1448             handlepp));
1449 }
1450
1451 int
1452 sa_buf_hold(objset_t *objset, uint64_t obj_num, void *tag, dmu_buf_t **db)
1453 {
1454         return (dmu_bonus_hold(objset, obj_num, tag, db));
1455 }
1456
1457 void
1458 sa_buf_rele(dmu_buf_t *db, void *tag)
1459 {
1460         dmu_buf_rele(db, tag);
1461 }
1462
1463 int
1464 sa_lookup_impl(sa_handle_t *hdl, sa_bulk_attr_t *bulk, int count)
1465 {
1466         ASSERT(hdl);
1467         ASSERT(MUTEX_HELD(&hdl->sa_lock));
1468         return (sa_attr_op(hdl, bulk, count, SA_LOOKUP, NULL));
1469 }
1470
1471 int
1472 sa_lookup(sa_handle_t *hdl, sa_attr_type_t attr, void *buf, uint32_t buflen)
1473 {
1474         int error;
1475         sa_bulk_attr_t bulk;
1476
1477         bulk.sa_attr = attr;
1478         bulk.sa_data = buf;
1479         bulk.sa_length = buflen;
1480         bulk.sa_data_func = NULL;
1481
1482         ASSERT(hdl);
1483         mutex_enter(&hdl->sa_lock);
1484         error = sa_lookup_impl(hdl, &bulk, 1);
1485         mutex_exit(&hdl->sa_lock);
1486         return (error);
1487 }
1488
1489 #ifdef _KERNEL
1490 int
1491 sa_lookup_uio(sa_handle_t *hdl, sa_attr_type_t attr, uio_t *uio)
1492 {
1493         int error;
1494         sa_bulk_attr_t bulk;
1495
1496         bulk.sa_data = NULL;
1497         bulk.sa_attr = attr;
1498         bulk.sa_data_func = NULL;
1499
1500         ASSERT(hdl);
1501
1502         mutex_enter(&hdl->sa_lock);
1503         if ((error = sa_attr_op(hdl, &bulk, 1, SA_LOOKUP, NULL)) == 0) {
1504                 error = uiomove((void *)bulk.sa_addr, MIN(bulk.sa_size,
1505                     uio->uio_resid), UIO_READ, uio);
1506         }
1507         mutex_exit(&hdl->sa_lock);
1508         return (error);
1509 }
1510 #endif
1511
1512 void *
1513 sa_find_idx_tab(objset_t *os, dmu_object_type_t bonustype, void *data)
1514 {
1515         sa_idx_tab_t *idx_tab;
1516         sa_hdr_phys_t *hdr = (sa_hdr_phys_t *)data;
1517         sa_os_t *sa = os->os_sa;
1518         sa_lot_t *tb, search;
1519         avl_index_t loc;
1520
1521         /*
1522          * Deterimine layout number.  If SA node and header == 0 then
1523          * force the index table to the dummy "1" empty layout.
1524          *
1525          * The layout number would only be zero for a newly created file
1526          * that has not added any attributes yet, or with crypto enabled which
1527          * doesn't write any attributes to the bonus buffer.
1528          */
1529
1530         search.lot_num = SA_LAYOUT_NUM(hdr, bonustype);
1531
1532         tb = avl_find(&sa->sa_layout_num_tree, &search, &loc);
1533
1534         /* Verify header size is consistent with layout information */
1535         ASSERT(tb);
1536         ASSERT((IS_SA_BONUSTYPE(bonustype) &&
1537             SA_HDR_SIZE_MATCH_LAYOUT(hdr, tb)) || !IS_SA_BONUSTYPE(bonustype) ||
1538             (IS_SA_BONUSTYPE(bonustype) && hdr->sa_layout_info == 0));
1539
1540         /*
1541          * See if any of the already existing TOC entries can be reused?
1542          */
1543
1544         for (idx_tab = list_head(&tb->lot_idx_tab); idx_tab;
1545             idx_tab = list_next(&tb->lot_idx_tab, idx_tab)) {
1546                 boolean_t valid_idx = B_TRUE;
1547                 int i;
1548
1549                 if (tb->lot_var_sizes != 0 &&
1550                     idx_tab->sa_variable_lengths != NULL) {
1551                         for (i = 0; i != tb->lot_var_sizes; i++) {
1552                                 if (hdr->sa_lengths[i] !=
1553                                     idx_tab->sa_variable_lengths[i]) {
1554                                         valid_idx = B_FALSE;
1555                                         break;
1556                                 }
1557                         }
1558                 }
1559                 if (valid_idx) {
1560                         sa_idx_tab_hold(os, idx_tab);
1561                         return (idx_tab);
1562                 }
1563         }
1564
1565         /* No such luck, create a new entry */
1566         idx_tab = kmem_zalloc(sizeof (sa_idx_tab_t), KM_PUSHPAGE);
1567         idx_tab->sa_idx_tab =
1568             kmem_zalloc(sizeof (uint32_t) * sa->sa_num_attrs, KM_PUSHPAGE);
1569         idx_tab->sa_layout = tb;
1570         refcount_create(&idx_tab->sa_refcount);
1571         if (tb->lot_var_sizes)
1572                 idx_tab->sa_variable_lengths = kmem_alloc(sizeof (uint16_t) *
1573                     tb->lot_var_sizes, KM_PUSHPAGE);
1574
1575         sa_attr_iter(os, hdr, bonustype, sa_build_idx_tab,
1576             tb, idx_tab);
1577         sa_idx_tab_hold(os, idx_tab);   /* one hold for consumer */
1578         sa_idx_tab_hold(os, idx_tab);   /* one for layout */
1579         list_insert_tail(&tb->lot_idx_tab, idx_tab);
1580         return (idx_tab);
1581 }
1582
1583 void
1584 sa_default_locator(void **dataptr, uint32_t *len, uint32_t total_len,
1585     boolean_t start, void *userdata)
1586 {
1587         ASSERT(start);
1588
1589         *dataptr = userdata;
1590         *len = total_len;
1591 }
1592
1593 static void
1594 sa_attr_register_sync(sa_handle_t *hdl, dmu_tx_t *tx)
1595 {
1596         uint64_t attr_value = 0;
1597         sa_os_t *sa = hdl->sa_os->os_sa;
1598         sa_attr_table_t *tb = sa->sa_attr_table;
1599         int i;
1600
1601         mutex_enter(&sa->sa_lock);
1602
1603         if (!sa->sa_need_attr_registration || sa->sa_master_obj == 0) {
1604                 mutex_exit(&sa->sa_lock);
1605                 return;
1606         }
1607
1608         if (sa->sa_reg_attr_obj == 0) {
1609                 sa->sa_reg_attr_obj = zap_create_link(hdl->sa_os,
1610                     DMU_OT_SA_ATTR_REGISTRATION,
1611                     sa->sa_master_obj, SA_REGISTRY, tx);
1612         }
1613         for (i = 0; i != sa->sa_num_attrs; i++) {
1614                 if (sa->sa_attr_table[i].sa_registered)
1615                         continue;
1616                 ATTR_ENCODE(attr_value, tb[i].sa_attr, tb[i].sa_length,
1617                     tb[i].sa_byteswap);
1618                 VERIFY(0 == zap_update(hdl->sa_os, sa->sa_reg_attr_obj,
1619                     tb[i].sa_name, 8, 1, &attr_value, tx));
1620                 tb[i].sa_registered = B_TRUE;
1621         }
1622         sa->sa_need_attr_registration = B_FALSE;
1623         mutex_exit(&sa->sa_lock);
1624 }
1625
1626 /*
1627  * Replace all attributes with attributes specified in template.
1628  * If dnode had a spill buffer then those attributes will be
1629  * also be replaced, possibly with just an empty spill block
1630  *
1631  * This interface is intended to only be used for bulk adding of
1632  * attributes for a new file.  It will also be used by the ZPL
1633  * when converting and old formatted znode to native SA support.
1634  */
1635 int
1636 sa_replace_all_by_template_locked(sa_handle_t *hdl, sa_bulk_attr_t *attr_desc,
1637     int attr_count, dmu_tx_t *tx)
1638 {
1639         sa_os_t *sa = hdl->sa_os->os_sa;
1640
1641         if (sa->sa_need_attr_registration)
1642                 sa_attr_register_sync(hdl, tx);
1643         return (sa_build_layouts(hdl, attr_desc, attr_count, tx));
1644 }
1645
1646 int
1647 sa_replace_all_by_template(sa_handle_t *hdl, sa_bulk_attr_t *attr_desc,
1648     int attr_count, dmu_tx_t *tx)
1649 {
1650         int error;
1651
1652         mutex_enter(&hdl->sa_lock);
1653         error = sa_replace_all_by_template_locked(hdl, attr_desc,
1654             attr_count, tx);
1655         mutex_exit(&hdl->sa_lock);
1656         return (error);
1657 }
1658
1659 /*
1660  * add/remove/replace a single attribute and then rewrite the entire set
1661  * of attributes.
1662  */
1663 static int
1664 sa_modify_attrs(sa_handle_t *hdl, sa_attr_type_t newattr,
1665     sa_data_op_t action, sa_data_locator_t *locator, void *datastart,
1666     uint16_t buflen, dmu_tx_t *tx)
1667 {
1668         sa_os_t *sa = hdl->sa_os->os_sa;
1669         dmu_buf_impl_t *db = (dmu_buf_impl_t *)hdl->sa_bonus;
1670         dnode_t *dn;
1671         sa_bulk_attr_t *attr_desc;
1672         void *old_data[2];
1673         int bonus_attr_count = 0;
1674         int bonus_data_size = 0;
1675         int spill_attr_count = 0;
1676         int error;
1677         uint16_t length;
1678         int i, j, k, length_idx;
1679         sa_hdr_phys_t *hdr;
1680         sa_idx_tab_t *idx_tab;
1681         int attr_count;
1682         int count;
1683
1684         ASSERT(MUTEX_HELD(&hdl->sa_lock));
1685
1686         /* First make of copy of the old data */
1687
1688         DB_DNODE_ENTER(db);
1689         dn = DB_DNODE(db);
1690         if (dn->dn_bonuslen != 0) {
1691                 bonus_data_size = hdl->sa_bonus->db_size;
1692                 old_data[0] = kmem_alloc(bonus_data_size, KM_SLEEP);
1693                 bcopy(hdl->sa_bonus->db_data, old_data[0],
1694                     hdl->sa_bonus->db_size);
1695                 bonus_attr_count = hdl->sa_bonus_tab->sa_layout->lot_attr_count;
1696         } else {
1697                 old_data[0] = NULL;
1698         }
1699         DB_DNODE_EXIT(db);
1700
1701         /* Bring spill buffer online if it isn't currently */
1702
1703         if ((error = sa_get_spill(hdl)) == 0) {
1704                 ASSERT3U(hdl->sa_spill->db_size, <=, SPA_MAXBLOCKSIZE);
1705                 old_data[1] = sa_spill_alloc(KM_SLEEP);
1706                 bcopy(hdl->sa_spill->db_data, old_data[1],
1707                     hdl->sa_spill->db_size);
1708                 spill_attr_count =
1709                     hdl->sa_spill_tab->sa_layout->lot_attr_count;
1710         } else if (error && error != ENOENT) {
1711                 if (old_data[0])
1712                         kmem_free(old_data[0], bonus_data_size);
1713                 return (error);
1714         } else {
1715                 old_data[1] = NULL;
1716         }
1717
1718         /* build descriptor of all attributes */
1719
1720         attr_count = bonus_attr_count + spill_attr_count;
1721         if (action == SA_ADD)
1722                 attr_count++;
1723         else if (action == SA_REMOVE)
1724                 attr_count--;
1725
1726         attr_desc = kmem_zalloc(sizeof (sa_bulk_attr_t) * attr_count, KM_SLEEP);
1727
1728         /*
1729          * loop through bonus and spill buffer if it exists, and
1730          * build up new attr_descriptor to reset the attributes
1731          */
1732         k = j = 0;
1733         count = bonus_attr_count;
1734         hdr = SA_GET_HDR(hdl, SA_BONUS);
1735         idx_tab = SA_IDX_TAB_GET(hdl, SA_BONUS);
1736         for (; k != 2; k++) {
1737                 /*
1738                  * Iterate over each attribute in layout.  Fetch the
1739                  * size of variable-length attributes needing rewrite
1740                  * from sa_lengths[].
1741                  */
1742                 for (i = 0, length_idx = 0; i != count; i++) {
1743                         sa_attr_type_t attr;
1744
1745                         attr = idx_tab->sa_layout->lot_attrs[i];
1746                         length = SA_REGISTERED_LEN(sa, attr);
1747                         if (attr == newattr) {
1748                                 if (length == 0)
1749                                         ++length_idx;
1750                                 if (action == SA_REMOVE) {
1751                                         j++;
1752                                         continue;
1753                                 }
1754                                 ASSERT(length == 0);
1755                                 ASSERT(action == SA_REPLACE);
1756                                 SA_ADD_BULK_ATTR(attr_desc, j, attr,
1757                                     locator, datastart, buflen);
1758                         } else {
1759                                 if (length == 0)
1760                                         length = hdr->sa_lengths[length_idx++];
1761
1762                                 SA_ADD_BULK_ATTR(attr_desc, j, attr,
1763                                     NULL, (void *)
1764                                     (TOC_OFF(idx_tab->sa_idx_tab[attr]) +
1765                                     (uintptr_t)old_data[k]), length);
1766                         }
1767                 }
1768                 if (k == 0 && hdl->sa_spill) {
1769                         hdr = SA_GET_HDR(hdl, SA_SPILL);
1770                         idx_tab = SA_IDX_TAB_GET(hdl, SA_SPILL);
1771                         count = spill_attr_count;
1772                 } else {
1773                         break;
1774                 }
1775         }
1776         if (action == SA_ADD) {
1777                 length = SA_REGISTERED_LEN(sa, newattr);
1778                 if (length == 0) {
1779                         length = buflen;
1780                 }
1781                 SA_ADD_BULK_ATTR(attr_desc, j, newattr, locator,
1782                     datastart, length);
1783         }
1784
1785         error = sa_build_layouts(hdl, attr_desc, attr_count, tx);
1786
1787         if (old_data[0])
1788                 kmem_free(old_data[0], bonus_data_size);
1789         if (old_data[1])
1790                 sa_spill_free(old_data[1]);
1791         kmem_free(attr_desc, sizeof (sa_bulk_attr_t) * attr_count);
1792
1793         return (error);
1794 }
1795
1796 static int
1797 sa_bulk_update_impl(sa_handle_t *hdl, sa_bulk_attr_t *bulk, int count,
1798     dmu_tx_t *tx)
1799 {
1800         int error;
1801         sa_os_t *sa = hdl->sa_os->os_sa;
1802         dmu_object_type_t bonustype;
1803         dmu_buf_t *saved_spill;
1804
1805         ASSERT(hdl);
1806         ASSERT(MUTEX_HELD(&hdl->sa_lock));
1807
1808         bonustype = SA_BONUSTYPE_FROM_DB(SA_GET_DB(hdl, SA_BONUS));
1809         saved_spill = hdl->sa_spill;
1810
1811         /* sync out registration table if necessary */
1812         if (sa->sa_need_attr_registration)
1813                 sa_attr_register_sync(hdl, tx);
1814
1815         error = sa_attr_op(hdl, bulk, count, SA_UPDATE, tx);
1816         if (error == 0 && !IS_SA_BONUSTYPE(bonustype) && sa->sa_update_cb)
1817                 sa->sa_update_cb(hdl, tx);
1818
1819         /*
1820          * If saved_spill is NULL and current sa_spill is not NULL that
1821          * means we increased the refcount of the spill buffer through
1822          * sa_get_spill() or dmu_spill_hold_by_dnode().  Therefore we
1823          * must release the hold before calling dmu_tx_commit() to avoid
1824          * making a copy of this buffer in dbuf_sync_leaf() due to the
1825          * reference count now being greater than 1.
1826          */
1827         if (!saved_spill && hdl->sa_spill) {
1828                 if (hdl->sa_spill_tab) {
1829                         sa_idx_tab_rele(hdl->sa_os, hdl->sa_spill_tab);
1830                         hdl->sa_spill_tab = NULL;
1831                 }
1832
1833                 dmu_buf_rele((dmu_buf_t *)hdl->sa_spill, NULL);
1834                 hdl->sa_spill = NULL;
1835         }
1836
1837         return (error);
1838 }
1839
1840 /*
1841  * update or add new attribute
1842  */
1843 int
1844 sa_update(sa_handle_t *hdl, sa_attr_type_t type,
1845     void *buf, uint32_t buflen, dmu_tx_t *tx)
1846 {
1847         int error;
1848         sa_bulk_attr_t bulk;
1849
1850         bulk.sa_attr = type;
1851         bulk.sa_data_func = NULL;
1852         bulk.sa_length = buflen;
1853         bulk.sa_data = buf;
1854
1855         mutex_enter(&hdl->sa_lock);
1856         error = sa_bulk_update_impl(hdl, &bulk, 1, tx);
1857         mutex_exit(&hdl->sa_lock);
1858         return (error);
1859 }
1860
1861 int
1862 sa_update_from_cb(sa_handle_t *hdl, sa_attr_type_t attr,
1863     uint32_t buflen, sa_data_locator_t *locator, void *userdata, dmu_tx_t *tx)
1864 {
1865         int error;
1866         sa_bulk_attr_t bulk;
1867
1868         bulk.sa_attr = attr;
1869         bulk.sa_data = userdata;
1870         bulk.sa_data_func = locator;
1871         bulk.sa_length = buflen;
1872
1873         mutex_enter(&hdl->sa_lock);
1874         error = sa_bulk_update_impl(hdl, &bulk, 1, tx);
1875         mutex_exit(&hdl->sa_lock);
1876         return (error);
1877 }
1878
1879 /*
1880  * Return size of an attribute
1881  */
1882
1883 int
1884 sa_size(sa_handle_t *hdl, sa_attr_type_t attr, int *size)
1885 {
1886         sa_bulk_attr_t bulk;
1887         int error;
1888
1889         bulk.sa_data = NULL;
1890         bulk.sa_attr = attr;
1891         bulk.sa_data_func = NULL;
1892
1893         ASSERT(hdl);
1894         mutex_enter(&hdl->sa_lock);
1895         if ((error = sa_attr_op(hdl, &bulk, 1, SA_LOOKUP, NULL)) != 0) {
1896                 mutex_exit(&hdl->sa_lock);
1897                 return (error);
1898         }
1899         *size = bulk.sa_size;
1900
1901         mutex_exit(&hdl->sa_lock);
1902         return (0);
1903 }
1904
1905 int
1906 sa_bulk_lookup_locked(sa_handle_t *hdl, sa_bulk_attr_t *attrs, int count)
1907 {
1908         ASSERT(hdl);
1909         ASSERT(MUTEX_HELD(&hdl->sa_lock));
1910         return (sa_lookup_impl(hdl, attrs, count));
1911 }
1912
1913 int
1914 sa_bulk_lookup(sa_handle_t *hdl, sa_bulk_attr_t *attrs, int count)
1915 {
1916         int error;
1917
1918         ASSERT(hdl);
1919         mutex_enter(&hdl->sa_lock);
1920         error = sa_bulk_lookup_locked(hdl, attrs, count);
1921         mutex_exit(&hdl->sa_lock);
1922         return (error);
1923 }
1924
1925 int
1926 sa_bulk_update(sa_handle_t *hdl, sa_bulk_attr_t *attrs, int count, dmu_tx_t *tx)
1927 {
1928         int error;
1929
1930         ASSERT(hdl);
1931         mutex_enter(&hdl->sa_lock);
1932         error = sa_bulk_update_impl(hdl, attrs, count, tx);
1933         mutex_exit(&hdl->sa_lock);
1934         return (error);
1935 }
1936
1937 int
1938 sa_remove(sa_handle_t *hdl, sa_attr_type_t attr, dmu_tx_t *tx)
1939 {
1940         int error;
1941
1942         mutex_enter(&hdl->sa_lock);
1943         error = sa_modify_attrs(hdl, attr, SA_REMOVE, NULL,
1944             NULL, 0, tx);
1945         mutex_exit(&hdl->sa_lock);
1946         return (error);
1947 }
1948
1949 void
1950 sa_object_info(sa_handle_t *hdl, dmu_object_info_t *doi)
1951 {
1952         dmu_object_info_from_db((dmu_buf_t *)hdl->sa_bonus, doi);
1953 }
1954
1955 void
1956 sa_object_size(sa_handle_t *hdl, uint32_t *blksize, u_longlong_t *nblocks)
1957 {
1958         dmu_object_size_from_db((dmu_buf_t *)hdl->sa_bonus,
1959             blksize, nblocks);
1960 }
1961
1962 void
1963 sa_update_user(sa_handle_t *newhdl, sa_handle_t *oldhdl)
1964 {
1965         (void) dmu_buf_update_user((dmu_buf_t *)newhdl->sa_bonus,
1966             oldhdl, newhdl, NULL, sa_evict);
1967         oldhdl->sa_bonus = NULL;
1968 }
1969
1970 void
1971 sa_set_userp(sa_handle_t *hdl, void *ptr)
1972 {
1973         hdl->sa_userp = ptr;
1974 }
1975
1976 dmu_buf_t *
1977 sa_get_db(sa_handle_t *hdl)
1978 {
1979         return ((dmu_buf_t *)hdl->sa_bonus);
1980 }
1981
1982 void *
1983 sa_get_userdata(sa_handle_t *hdl)
1984 {
1985         return (hdl->sa_userp);
1986 }
1987
1988 void
1989 sa_register_update_callback_locked(objset_t *os, sa_update_cb_t *func)
1990 {
1991         ASSERT(MUTEX_HELD(&os->os_sa->sa_lock));
1992         os->os_sa->sa_update_cb = func;
1993 }
1994
1995 void
1996 sa_register_update_callback(objset_t *os, sa_update_cb_t *func)
1997 {
1998
1999         mutex_enter(&os->os_sa->sa_lock);
2000         sa_register_update_callback_locked(os, func);
2001         mutex_exit(&os->os_sa->sa_lock);
2002 }
2003
2004 uint64_t
2005 sa_handle_object(sa_handle_t *hdl)
2006 {
2007         return (hdl->sa_bonus->db_object);
2008 }
2009
2010 boolean_t
2011 sa_enabled(objset_t *os)
2012 {
2013         return (os->os_sa == NULL);
2014 }
2015
2016 int
2017 sa_set_sa_object(objset_t *os, uint64_t sa_object)
2018 {
2019         sa_os_t *sa = os->os_sa;
2020
2021         if (sa->sa_master_obj)
2022                 return (1);
2023
2024         sa->sa_master_obj = sa_object;
2025
2026         return (0);
2027 }
2028
2029 int
2030 sa_hdrsize(void *arg)
2031 {
2032         sa_hdr_phys_t *hdr = arg;
2033
2034         return (SA_HDR_SIZE(hdr));
2035 }
2036
2037 void
2038 sa_handle_lock(sa_handle_t *hdl)
2039 {
2040         ASSERT(hdl);
2041         mutex_enter(&hdl->sa_lock);
2042 }
2043
2044 void
2045 sa_handle_unlock(sa_handle_t *hdl)
2046 {
2047         ASSERT(hdl);
2048         mutex_exit(&hdl->sa_lock);
2049 }
2050
2051 #ifdef _KERNEL
2052 EXPORT_SYMBOL(sa_handle_get);
2053 EXPORT_SYMBOL(sa_handle_get_from_db);
2054 EXPORT_SYMBOL(sa_handle_destroy);
2055 EXPORT_SYMBOL(sa_buf_hold);
2056 EXPORT_SYMBOL(sa_buf_rele);
2057 EXPORT_SYMBOL(sa_spill_rele);
2058 EXPORT_SYMBOL(sa_lookup);
2059 EXPORT_SYMBOL(sa_update);
2060 EXPORT_SYMBOL(sa_remove);
2061 EXPORT_SYMBOL(sa_bulk_lookup);
2062 EXPORT_SYMBOL(sa_bulk_lookup_locked);
2063 EXPORT_SYMBOL(sa_bulk_update);
2064 EXPORT_SYMBOL(sa_size);
2065 EXPORT_SYMBOL(sa_update_from_cb);
2066 EXPORT_SYMBOL(sa_object_info);
2067 EXPORT_SYMBOL(sa_object_size);
2068 EXPORT_SYMBOL(sa_update_user);
2069 EXPORT_SYMBOL(sa_get_userdata);
2070 EXPORT_SYMBOL(sa_set_userp);
2071 EXPORT_SYMBOL(sa_get_db);
2072 EXPORT_SYMBOL(sa_handle_object);
2073 EXPORT_SYMBOL(sa_register_update_callback);
2074 EXPORT_SYMBOL(sa_setup);
2075 EXPORT_SYMBOL(sa_replace_all_by_template);
2076 EXPORT_SYMBOL(sa_replace_all_by_template_locked);
2077 EXPORT_SYMBOL(sa_enabled);
2078 EXPORT_SYMBOL(sa_cache_init);
2079 EXPORT_SYMBOL(sa_cache_fini);
2080 EXPORT_SYMBOL(sa_spill_alloc);
2081 EXPORT_SYMBOL(sa_spill_free);
2082 EXPORT_SYMBOL(sa_set_sa_object);
2083 EXPORT_SYMBOL(sa_hdrsize);
2084 EXPORT_SYMBOL(sa_handle_lock);
2085 EXPORT_SYMBOL(sa_handle_unlock);
2086 EXPORT_SYMBOL(sa_lookup_uio);
2087 #endif /* _KERNEL */