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