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