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 * Copyright (c) 2005, 2010, Oracle and/or its affiliates. All rights reserved.
23 * Copyright (c) 2011 Pawel Jakub Dawidek <pawel@dawidek.net>.
24 * All rights reserved.
25 * Copyright (c) 2012, 2014 by Delphix. All rights reserved.
26 * Copyright (c) 2014 Joyent, Inc. All rights reserved.
27 * Copyright (c) 2014 Spectra Logic Corporation, All rights reserved.
28 * Copyright 2015 Nexenta Systems, Inc. All rights reserved.
29 */
30
31 #include <sys/dmu.h>
32 #include <sys/dmu_objset.h>
33 #include <sys/dmu_tx.h>
34 #include <sys/dsl_dataset.h>
35 #include <sys/dsl_dir.h>
36 #include <sys/dsl_prop.h>
37 #include <sys/dsl_synctask.h>
38 #include <sys/dsl_deleg.h>
39 #include <sys/dmu_impl.h>
40 #include <sys/spa.h>
41 #include <sys/metaslab.h>
42 #include <sys/zap.h>
43 #include <sys/zio.h>
44 #include <sys/arc.h>
45 #include <sys/sunddi.h>
46 #include <sys/zvol.h>
47 #ifdef _KERNEL
48 #include <sys/zfs_vfsops.h>
49 #endif
50 #include <sys/zfeature.h>
51 #include <sys/policy.h>
52 #include <sys/zfs_znode.h>
53 #include "zfs_namecheck.h"
54 #include "zfs_prop.h"
55
56 /*
57 * Filesystem and Snapshot Limits
58 * ------------------------------
59 *
60 * These limits are used to restrict the number of filesystems and/or snapshots
61 * that can be created at a given level in the tree or below. A typical
62 * use-case is with a delegated dataset where the administrator wants to ensure
63 * that a user within the zone is not creating too many additional filesystems
64 * or snapshots, even though they're not exceeding their space quota.
65 *
66 * The filesystem and snapshot counts are stored as extensible properties. This
67 * capability is controlled by a feature flag and must be enabled to be used.
68 * Once enabled, the feature is not active until the first limit is set. At
69 * that point, future operations to create/destroy filesystems or snapshots
70 * will validate and update the counts.
71 *
72 * Because the count properties will not exist before the feature is active,
73 * the counts are updated when a limit is first set on an uninitialized
74 * dsl_dir node in the tree (The filesystem/snapshot count on a node includes
75 * all of the nested filesystems/snapshots. Thus, a new leaf node has a
76 * filesystem count of 0 and a snapshot count of 0. Non-existent filesystem and
77 * snapshot count properties on a node indicate uninitialized counts on that
78 * node.) When first setting a limit on an uninitialized node, the code starts
79 * at the filesystem with the new limit and descends into all sub-filesystems
80 * to add the count properties.
81 *
82 * In practice this is lightweight since a limit is typically set when the
83 * filesystem is created and thus has no children. Once valid, changing the
84 * limit value won't require a re-traversal since the counts are already valid.
85 * When recursively fixing the counts, if a node with a limit is encountered
86 * during the descent, the counts are known to be valid and there is no need to
87 * descend into that filesystem's children. The counts on filesystems above the
88 * one with the new limit will still be uninitialized, unless a limit is
89 * eventually set on one of those filesystems. The counts are always recursively
90 * updated when a limit is set on a dataset, unless there is already a limit.
91 * When a new limit value is set on a filesystem with an existing limit, it is
92 * possible for the new limit to be less than the current count at that level
93 * since a user who can change the limit is also allowed to exceed the limit.
94 *
95 * Once the feature is active, then whenever a filesystem or snapshot is
96 * created, the code recurses up the tree, validating the new count against the
97 * limit at each initialized level. In practice, most levels will not have a
98 * limit set. If there is a limit at any initialized level up the tree, the
99 * check must pass or the creation will fail. Likewise, when a filesystem or
100 * snapshot is destroyed, the counts are recursively adjusted all the way up
101 * the initizized nodes in the tree. Renaming a filesystem into different point
102 * in the tree will first validate, then update the counts on each branch up to
103 * the common ancestor. A receive will also validate the counts and then update
104 * them.
105 *
106 * An exception to the above behavior is that the limit is not enforced if the
107 * user has permission to modify the limit. This is primarily so that
108 * recursive snapshots in the global zone always work. We want to prevent a
109 * denial-of-service in which a lower level delegated dataset could max out its
110 * limit and thus block recursive snapshots from being taken in the global zone.
111 * Because of this, it is possible for the snapshot count to be over the limit
112 * and snapshots taken in the global zone could cause a lower level dataset to
113 * hit or exceed its limit. The administrator taking the global zone recursive
114 * snapshot should be aware of this side-effect and behave accordingly.
115 * For consistency, the filesystem limit is also not enforced if the user can
116 * modify the limit.
117 *
118 * The filesystem and snapshot limits are validated by dsl_fs_ss_limit_check()
119 * and updated by dsl_fs_ss_count_adjust(). A new limit value is setup in
120 * dsl_dir_activate_fs_ss_limit() and the counts are adjusted, if necessary, by
121 * dsl_dir_init_fs_ss_count().
122 *
123 * There is a special case when we receive a filesystem that already exists. In
124 * this case a temporary clone name of %X is created (see dmu_recv_begin). We
125 * never update the filesystem counts for temporary clones.
126 *
127 * Likewise, we do not update the snapshot counts for temporary snapshots,
128 * such as those created by zfs diff.
129 */
130
131 extern inline dsl_dir_phys_t *dsl_dir_phys(dsl_dir_t *dd);
132
133 static uint64_t dsl_dir_space_towrite(dsl_dir_t *dd);
134
135 static void
dsl_dir_evict(void * dbu)136 dsl_dir_evict(void *dbu)
137 {
138 dsl_dir_t *dd = dbu;
139 dsl_pool_t *dp = dd->dd_pool;
140 int t;
141
142 dd->dd_dbuf = NULL;
143
144 for (t = 0; t < TXG_SIZE; t++) {
145 ASSERT(!txg_list_member(&dp->dp_dirty_dirs, dd, t));
146 ASSERT(dd->dd_tempreserved[t] == 0);
147 ASSERT(dd->dd_space_towrite[t] == 0);
148 }
149
150 if (dd->dd_parent)
151 dsl_dir_async_rele(dd->dd_parent, dd);
152
153 spa_async_close(dd->dd_pool->dp_spa, dd);
154
155 dsl_prop_fini(dd);
156 mutex_destroy(&dd->dd_lock);
157 kmem_free(dd, sizeof (dsl_dir_t));
158 }
159
160 int
dsl_dir_hold_obj(dsl_pool_t * dp,uint64_t ddobj,const char * tail,void * tag,dsl_dir_t ** ddp)161 dsl_dir_hold_obj(dsl_pool_t *dp, uint64_t ddobj,
162 const char *tail, void *tag, dsl_dir_t **ddp)
163 {
164 dmu_buf_t *dbuf;
165 dsl_dir_t *dd;
166 int err;
167
168 ASSERT(dsl_pool_config_held(dp));
169
170 err = dmu_bonus_hold(dp->dp_meta_objset, ddobj, tag, &dbuf);
171 if (err != 0)
172 return (err);
173 dd = dmu_buf_get_user(dbuf);
174 #ifdef ZFS_DEBUG
175 {
176 dmu_object_info_t doi;
177 dmu_object_info_from_db(dbuf, &doi);
178 ASSERT3U(doi.doi_bonus_type, ==, DMU_OT_DSL_DIR);
179 ASSERT3U(doi.doi_bonus_size, >=, sizeof (dsl_dir_phys_t));
180 }
181 #endif
182 if (dd == NULL) {
183 dsl_dir_t *winner;
184
185 dd = kmem_zalloc(sizeof (dsl_dir_t), KM_SLEEP);
186 dd->dd_object = ddobj;
187 dd->dd_dbuf = dbuf;
188 dd->dd_pool = dp;
189 mutex_init(&dd->dd_lock, NULL, MUTEX_DEFAULT, NULL);
190 dsl_prop_init(dd);
191
192 dsl_dir_snap_cmtime_update(dd);
193
194 if (dsl_dir_phys(dd)->dd_parent_obj) {
195 err = dsl_dir_hold_obj(dp,
196 dsl_dir_phys(dd)->dd_parent_obj, NULL, dd,
197 &dd->dd_parent);
198 if (err != 0)
199 goto errout;
200 if (tail) {
201 #ifdef ZFS_DEBUG
202 uint64_t foundobj;
203
204 err = zap_lookup(dp->dp_meta_objset,
205 dsl_dir_phys(dd->dd_parent)->
206 dd_child_dir_zapobj, tail,
207 sizeof (foundobj), 1, &foundobj);
208 ASSERT(err || foundobj == ddobj);
209 #endif
210 (void) strcpy(dd->dd_myname, tail);
211 } else {
212 err = zap_value_search(dp->dp_meta_objset,
213 dsl_dir_phys(dd->dd_parent)->
214 dd_child_dir_zapobj,
215 ddobj, 0, dd->dd_myname);
216 }
217 if (err != 0)
218 goto errout;
219 } else {
220 (void) strcpy(dd->dd_myname, spa_name(dp->dp_spa));
221 }
222
223 if (dsl_dir_is_clone(dd)) {
224 dmu_buf_t *origin_bonus;
225 dsl_dataset_phys_t *origin_phys;
226
227 /*
228 * We can't open the origin dataset, because
229 * that would require opening this dsl_dir.
230 * Just look at its phys directly instead.
231 */
232 err = dmu_bonus_hold(dp->dp_meta_objset,
233 dsl_dir_phys(dd)->dd_origin_obj, FTAG,
234 &origin_bonus);
235 if (err != 0)
236 goto errout;
237 origin_phys = origin_bonus->db_data;
238 dd->dd_origin_txg =
239 origin_phys->ds_creation_txg;
240 dmu_buf_rele(origin_bonus, FTAG);
241 }
242
243 dmu_buf_init_user(&dd->dd_dbu, dsl_dir_evict, &dd->dd_dbuf);
244 winner = dmu_buf_set_user_ie(dbuf, &dd->dd_dbu);
245 if (winner != NULL) {
246 if (dd->dd_parent)
247 dsl_dir_rele(dd->dd_parent, dd);
248 dsl_prop_fini(dd);
249 mutex_destroy(&dd->dd_lock);
250 kmem_free(dd, sizeof (dsl_dir_t));
251 dd = winner;
252 } else {
253 spa_open_ref(dp->dp_spa, dd);
254 }
255 }
256
257 /*
258 * The dsl_dir_t has both open-to-close and instantiate-to-evict
259 * holds on the spa. We need the open-to-close holds because
260 * otherwise the spa_refcnt wouldn't change when we open a
261 * dir which the spa also has open, so we could incorrectly
262 * think it was OK to unload/export/destroy the pool. We need
263 * the instantiate-to-evict hold because the dsl_dir_t has a
264 * pointer to the dd_pool, which has a pointer to the spa_t.
265 */
266 spa_open_ref(dp->dp_spa, tag);
267 ASSERT3P(dd->dd_pool, ==, dp);
268 ASSERT3U(dd->dd_object, ==, ddobj);
269 ASSERT3P(dd->dd_dbuf, ==, dbuf);
270 *ddp = dd;
271 return (0);
272
273 errout:
274 if (dd->dd_parent)
275 dsl_dir_rele(dd->dd_parent, dd);
276 dsl_prop_fini(dd);
277 mutex_destroy(&dd->dd_lock);
278 kmem_free(dd, sizeof (dsl_dir_t));
279 dmu_buf_rele(dbuf, tag);
280 return (err);
281 }
282
283 void
dsl_dir_rele(dsl_dir_t * dd,void * tag)284 dsl_dir_rele(dsl_dir_t *dd, void *tag)
285 {
286 dprintf_dd(dd, "%s\n", "");
287 spa_close(dd->dd_pool->dp_spa, tag);
288 dmu_buf_rele(dd->dd_dbuf, tag);
289 }
290
291 /*
292 * Remove a reference to the given dsl dir that is being asynchronously
293 * released. Async releases occur from a taskq performing eviction of
294 * dsl datasets and dirs. This process is identical to a normal release
295 * with the exception of using the async API for releasing the reference on
296 * the spa.
297 */
298 void
dsl_dir_async_rele(dsl_dir_t * dd,void * tag)299 dsl_dir_async_rele(dsl_dir_t *dd, void *tag)
300 {
301 dprintf_dd(dd, "%s\n", "");
302 spa_async_close(dd->dd_pool->dp_spa, tag);
303 dmu_buf_rele(dd->dd_dbuf, tag);
304 }
305
306 /* buf must be at least ZFS_MAX_DATASET_NAME_LEN bytes */
307 void
dsl_dir_name(dsl_dir_t * dd,char * buf)308 dsl_dir_name(dsl_dir_t *dd, char *buf)
309 {
310 if (dd->dd_parent) {
311 dsl_dir_name(dd->dd_parent, buf);
312 VERIFY3U(strlcat(buf, "/", ZFS_MAX_DATASET_NAME_LEN), <,
313 ZFS_MAX_DATASET_NAME_LEN);
314 } else {
315 buf[0] = '\0';
316 }
317 if (!MUTEX_HELD(&dd->dd_lock)) {
318 /*
319 * recursive mutex so that we can use
320 * dprintf_dd() with dd_lock held
321 */
322 mutex_enter(&dd->dd_lock);
323 VERIFY3U(strlcat(buf, dd->dd_myname, ZFS_MAX_DATASET_NAME_LEN),
324 <, ZFS_MAX_DATASET_NAME_LEN);
325 mutex_exit(&dd->dd_lock);
326 } else {
327 VERIFY3U(strlcat(buf, dd->dd_myname, ZFS_MAX_DATASET_NAME_LEN),
328 <, ZFS_MAX_DATASET_NAME_LEN);
329 }
330 }
331
332 /* Calculate name length, avoiding all the strcat calls of dsl_dir_name */
333 int
dsl_dir_namelen(dsl_dir_t * dd)334 dsl_dir_namelen(dsl_dir_t *dd)
335 {
336 int result = 0;
337
338 if (dd->dd_parent) {
339 /* parent's name + 1 for the "/" */
340 result = dsl_dir_namelen(dd->dd_parent) + 1;
341 }
342
343 if (!MUTEX_HELD(&dd->dd_lock)) {
344 /* see dsl_dir_name */
345 mutex_enter(&dd->dd_lock);
346 result += strlen(dd->dd_myname);
347 mutex_exit(&dd->dd_lock);
348 } else {
349 result += strlen(dd->dd_myname);
350 }
351
352 return (result);
353 }
354
355 static int
getcomponent(const char * path,char * component,const char ** nextp)356 getcomponent(const char *path, char *component, const char **nextp)
357 {
358 char *p;
359
360 if ((path == NULL) || (path[0] == '\0'))
361 return (SET_ERROR(ENOENT));
362 /* This would be a good place to reserve some namespace... */
363 p = strpbrk(path, "/@");
364 if (p && (p[1] == '/' || p[1] == '@')) {
365 /* two separators in a row */
366 return (SET_ERROR(EINVAL));
367 }
368 if (p == NULL || p == path) {
369 /*
370 * if the first thing is an @ or /, it had better be an
371 * @ and it had better not have any more ats or slashes,
372 * and it had better have something after the @.
373 */
374 if (p != NULL &&
375 (p[0] != '@' || strpbrk(path+1, "/@") || p[1] == '\0'))
376 return (SET_ERROR(EINVAL));
377 if (strlen(path) >= ZFS_MAX_DATASET_NAME_LEN)
378 return (SET_ERROR(ENAMETOOLONG));
379 (void) strcpy(component, path);
380 p = NULL;
381 } else if (p[0] == '/') {
382 if (p - path >= ZFS_MAX_DATASET_NAME_LEN)
383 return (SET_ERROR(ENAMETOOLONG));
384 (void) strncpy(component, path, p - path);
385 component[p - path] = '\0';
386 p++;
387 } else if (p[0] == '@') {
388 /*
389 * if the next separator is an @, there better not be
390 * any more slashes.
391 */
392 if (strchr(path, '/'))
393 return (SET_ERROR(EINVAL));
394 if (p - path >= ZFS_MAX_DATASET_NAME_LEN)
395 return (SET_ERROR(ENAMETOOLONG));
396 (void) strncpy(component, path, p - path);
397 component[p - path] = '\0';
398 } else {
399 panic("invalid p=%p", (void *)p);
400 }
401 *nextp = p;
402 return (0);
403 }
404
405 /*
406 * Return the dsl_dir_t, and possibly the last component which couldn't
407 * be found in *tail. The name must be in the specified dsl_pool_t. This
408 * thread must hold the dp_config_rwlock for the pool. Returns NULL if the
409 * path is bogus, or if tail==NULL and we couldn't parse the whole name.
410 * (*tail)[0] == '@' means that the last component is a snapshot.
411 */
412 int
dsl_dir_hold(dsl_pool_t * dp,const char * name,void * tag,dsl_dir_t ** ddp,const char ** tailp)413 dsl_dir_hold(dsl_pool_t *dp, const char *name, void *tag,
414 dsl_dir_t **ddp, const char **tailp)
415 {
416 char buf[ZFS_MAX_DATASET_NAME_LEN];
417 const char *spaname, *next, *nextnext = NULL;
418 int err;
419 dsl_dir_t *dd;
420 uint64_t ddobj;
421
422 err = getcomponent(name, buf, &next);
423 if (err != 0)
424 return (err);
425
426 /* Make sure the name is in the specified pool. */
427 spaname = spa_name(dp->dp_spa);
428 if (strcmp(buf, spaname) != 0)
429 return (SET_ERROR(EXDEV));
430
431 ASSERT(dsl_pool_config_held(dp));
432
433 err = dsl_dir_hold_obj(dp, dp->dp_root_dir_obj, NULL, tag, &dd);
434 if (err != 0) {
435 return (err);
436 }
437
438 while (next != NULL) {
439 dsl_dir_t *child_dd;
440 err = getcomponent(next, buf, &nextnext);
441 if (err != 0)
442 break;
443 ASSERT(next[0] != '\0');
444 if (next[0] == '@')
445 break;
446 dprintf("looking up %s in obj%lld\n",
447 buf, dsl_dir_phys(dd)->dd_child_dir_zapobj);
448
449 err = zap_lookup(dp->dp_meta_objset,
450 dsl_dir_phys(dd)->dd_child_dir_zapobj,
451 buf, sizeof (ddobj), 1, &ddobj);
452 if (err != 0) {
453 if (err == ENOENT)
454 err = 0;
455 break;
456 }
457
458 err = dsl_dir_hold_obj(dp, ddobj, buf, tag, &child_dd);
459 if (err != 0)
460 break;
461 dsl_dir_rele(dd, tag);
462 dd = child_dd;
463 next = nextnext;
464 }
465
466 if (err != 0) {
467 dsl_dir_rele(dd, tag);
468 return (err);
469 }
470
471 /*
472 * It's an error if there's more than one component left, or
473 * tailp==NULL and there's any component left.
474 */
475 if (next != NULL &&
476 (tailp == NULL || (nextnext && nextnext[0] != '\0'))) {
477 /* bad path name */
478 dsl_dir_rele(dd, tag);
479 dprintf("next=%p (%s) tail=%p\n", next, next?next:"", tailp);
480 err = SET_ERROR(ENOENT);
481 }
482 if (tailp != NULL)
483 *tailp = next;
484 *ddp = dd;
485 return (err);
486 }
487
488 /*
489 * If the counts are already initialized for this filesystem and its
490 * descendants then do nothing, otherwise initialize the counts.
491 *
492 * The counts on this filesystem, and those below, may be uninitialized due to
493 * either the use of a pre-existing pool which did not support the
494 * filesystem/snapshot limit feature, or one in which the feature had not yet
495 * been enabled.
496 *
497 * Recursively descend the filesystem tree and update the filesystem/snapshot
498 * counts on each filesystem below, then update the cumulative count on the
499 * current filesystem. If the filesystem already has a count set on it,
500 * then we know that its counts, and the counts on the filesystems below it,
501 * are already correct, so we don't have to update this filesystem.
502 */
503 static void
dsl_dir_init_fs_ss_count(dsl_dir_t * dd,dmu_tx_t * tx)504 dsl_dir_init_fs_ss_count(dsl_dir_t *dd, dmu_tx_t *tx)
505 {
506 uint64_t my_fs_cnt = 0;
507 uint64_t my_ss_cnt = 0;
508 dsl_pool_t *dp = dd->dd_pool;
509 objset_t *os = dp->dp_meta_objset;
510 zap_cursor_t *zc;
511 zap_attribute_t *za;
512 dsl_dataset_t *ds;
513
514 ASSERT(spa_feature_is_active(dp->dp_spa, SPA_FEATURE_FS_SS_LIMIT));
515 ASSERT(dsl_pool_config_held(dp));
516 ASSERT(dmu_tx_is_syncing(tx));
517
518 dsl_dir_zapify(dd, tx);
519
520 /*
521 * If the filesystem count has already been initialized then we
522 * don't need to recurse down any further.
523 */
524 if (zap_contains(os, dd->dd_object, DD_FIELD_FILESYSTEM_COUNT) == 0)
525 return;
526
527 zc = kmem_alloc(sizeof (zap_cursor_t), KM_SLEEP);
528 za = kmem_alloc(sizeof (zap_attribute_t), KM_SLEEP);
529
530 /* Iterate my child dirs */
531 for (zap_cursor_init(zc, os, dsl_dir_phys(dd)->dd_child_dir_zapobj);
532 zap_cursor_retrieve(zc, za) == 0; zap_cursor_advance(zc)) {
533 dsl_dir_t *chld_dd;
534 uint64_t count;
535
536 VERIFY0(dsl_dir_hold_obj(dp, za->za_first_integer, NULL, FTAG,
537 &chld_dd));
538
539 /*
540 * Ignore hidden ($FREE, $MOS & $ORIGIN) objsets and
541 * temporary datasets.
542 */
543 if (chld_dd->dd_myname[0] == '$' ||
544 chld_dd->dd_myname[0] == '%') {
545 dsl_dir_rele(chld_dd, FTAG);
546 continue;
547 }
548
549 my_fs_cnt++; /* count this child */
550
551 dsl_dir_init_fs_ss_count(chld_dd, tx);
552
553 VERIFY0(zap_lookup(os, chld_dd->dd_object,
554 DD_FIELD_FILESYSTEM_COUNT, sizeof (count), 1, &count));
555 my_fs_cnt += count;
556 VERIFY0(zap_lookup(os, chld_dd->dd_object,
557 DD_FIELD_SNAPSHOT_COUNT, sizeof (count), 1, &count));
558 my_ss_cnt += count;
559
560 dsl_dir_rele(chld_dd, FTAG);
561 }
562 zap_cursor_fini(zc);
563 /* Count my snapshots (we counted children's snapshots above) */
564 VERIFY0(dsl_dataset_hold_obj(dd->dd_pool,
565 dsl_dir_phys(dd)->dd_head_dataset_obj, FTAG, &ds));
566
567 for (zap_cursor_init(zc, os, dsl_dataset_phys(ds)->ds_snapnames_zapobj);
568 zap_cursor_retrieve(zc, za) == 0;
569 zap_cursor_advance(zc)) {
570 /* Don't count temporary snapshots */
571 if (za->za_name[0] != '%')
572 my_ss_cnt++;
573 }
574 zap_cursor_fini(zc);
575
576 dsl_dataset_rele(ds, FTAG);
577
578 kmem_free(zc, sizeof (zap_cursor_t));
579 kmem_free(za, sizeof (zap_attribute_t));
580
581 /* we're in a sync task, update counts */
582 dmu_buf_will_dirty(dd->dd_dbuf, tx);
583 VERIFY0(zap_add(os, dd->dd_object, DD_FIELD_FILESYSTEM_COUNT,
584 sizeof (my_fs_cnt), 1, &my_fs_cnt, tx));
585 VERIFY0(zap_add(os, dd->dd_object, DD_FIELD_SNAPSHOT_COUNT,
586 sizeof (my_ss_cnt), 1, &my_ss_cnt, tx));
587 }
588
589 static int
dsl_dir_actv_fs_ss_limit_check(void * arg,dmu_tx_t * tx)590 dsl_dir_actv_fs_ss_limit_check(void *arg, dmu_tx_t *tx)
591 {
592 char *ddname = (char *)arg;
593 dsl_pool_t *dp = dmu_tx_pool(tx);
594 dsl_dataset_t *ds;
595 dsl_dir_t *dd;
596 int error;
597
598 error = dsl_dataset_hold(dp, ddname, FTAG, &ds);
599 if (error != 0)
600 return (error);
601
602 if (!spa_feature_is_enabled(dp->dp_spa, SPA_FEATURE_FS_SS_LIMIT)) {
603 dsl_dataset_rele(ds, FTAG);
604 return (SET_ERROR(ENOTSUP));
605 }
606
607 dd = ds->ds_dir;
608 if (spa_feature_is_active(dp->dp_spa, SPA_FEATURE_FS_SS_LIMIT) &&
609 dsl_dir_is_zapified(dd) &&
610 zap_contains(dp->dp_meta_objset, dd->dd_object,
611 DD_FIELD_FILESYSTEM_COUNT) == 0) {
612 dsl_dataset_rele(ds, FTAG);
613 return (SET_ERROR(EALREADY));
614 }
615
616 dsl_dataset_rele(ds, FTAG);
617 return (0);
618 }
619
620 static void
dsl_dir_actv_fs_ss_limit_sync(void * arg,dmu_tx_t * tx)621 dsl_dir_actv_fs_ss_limit_sync(void *arg, dmu_tx_t *tx)
622 {
623 char *ddname = (char *)arg;
624 dsl_pool_t *dp = dmu_tx_pool(tx);
625 dsl_dataset_t *ds;
626 spa_t *spa;
627
628 VERIFY0(dsl_dataset_hold(dp, ddname, FTAG, &ds));
629
630 spa = dsl_dataset_get_spa(ds);
631
632 if (!spa_feature_is_active(spa, SPA_FEATURE_FS_SS_LIMIT)) {
633 /*
634 * Since the feature was not active and we're now setting a
635 * limit, increment the feature-active counter so that the
636 * feature becomes active for the first time.
637 *
638 * We are already in a sync task so we can update the MOS.
639 */
640 spa_feature_incr(spa, SPA_FEATURE_FS_SS_LIMIT, tx);
641 }
642
643 /*
644 * Since we are now setting a non-UINT64_MAX limit on the filesystem,
645 * we need to ensure the counts are correct. Descend down the tree from
646 * this point and update all of the counts to be accurate.
647 */
648 dsl_dir_init_fs_ss_count(ds->ds_dir, tx);
649
650 dsl_dataset_rele(ds, FTAG);
651 }
652
653 /*
654 * Make sure the feature is enabled and activate it if necessary.
655 * Since we're setting a limit, ensure the on-disk counts are valid.
656 * This is only called by the ioctl path when setting a limit value.
657 *
658 * We do not need to validate the new limit, since users who can change the
659 * limit are also allowed to exceed the limit.
660 */
661 int
dsl_dir_activate_fs_ss_limit(const char * ddname)662 dsl_dir_activate_fs_ss_limit(const char *ddname)
663 {
664 int error;
665
666 error = dsl_sync_task(ddname, dsl_dir_actv_fs_ss_limit_check,
667 dsl_dir_actv_fs_ss_limit_sync, (void *)ddname, 0,
668 ZFS_SPACE_CHECK_RESERVED);
669
670 if (error == EALREADY)
671 error = 0;
672
673 return (error);
674 }
675
676 /*
677 * Used to determine if the filesystem_limit or snapshot_limit should be
678 * enforced. We allow the limit to be exceeded if the user has permission to
679 * write the property value. We pass in the creds that we got in the open
680 * context since we will always be the GZ root in syncing context. We also have
681 * to handle the case where we are allowed to change the limit on the current
682 * dataset, but there may be another limit in the tree above.
683 *
684 * We can never modify these two properties within a non-global zone. In
685 * addition, the other checks are modeled on zfs_secpolicy_write_perms. We
686 * can't use that function since we are already holding the dp_config_rwlock.
687 * In addition, we already have the dd and dealing with snapshots is simplified
688 * in this code.
689 */
690
691 typedef enum {
692 ENFORCE_ALWAYS,
693 ENFORCE_NEVER,
694 ENFORCE_ABOVE
695 } enforce_res_t;
696
697 static enforce_res_t
dsl_enforce_ds_ss_limits(dsl_dir_t * dd,zfs_prop_t prop,cred_t * cr)698 dsl_enforce_ds_ss_limits(dsl_dir_t *dd, zfs_prop_t prop, cred_t *cr)
699 {
700 enforce_res_t enforce = ENFORCE_ALWAYS;
701 uint64_t obj;
702 dsl_dataset_t *ds;
703 uint64_t zoned;
704
705 ASSERT(prop == ZFS_PROP_FILESYSTEM_LIMIT ||
706 prop == ZFS_PROP_SNAPSHOT_LIMIT);
707
708 #ifdef _KERNEL
709 #ifdef __FreeBSD__
710 if (jailed(cr))
711 #else
712 if (crgetzoneid(cr) != GLOBAL_ZONEID)
713 #endif
714 return (ENFORCE_ALWAYS);
715
716 if (secpolicy_zfs(cr) == 0)
717 return (ENFORCE_NEVER);
718 #endif
719
720 if ((obj = dsl_dir_phys(dd)->dd_head_dataset_obj) == 0)
721 return (ENFORCE_ALWAYS);
722
723 ASSERT(dsl_pool_config_held(dd->dd_pool));
724
725 if (dsl_dataset_hold_obj(dd->dd_pool, obj, FTAG, &ds) != 0)
726 return (ENFORCE_ALWAYS);
727
728 if (dsl_prop_get_ds(ds, "zoned", 8, 1, &zoned, NULL) || zoned) {
729 /* Only root can access zoned fs's from the GZ */
730 enforce = ENFORCE_ALWAYS;
731 } else {
732 if (dsl_deleg_access_impl(ds, zfs_prop_to_name(prop), cr) == 0)
733 enforce = ENFORCE_ABOVE;
734 }
735
736 dsl_dataset_rele(ds, FTAG);
737 return (enforce);
738 }
739
740 /*
741 * Check if adding additional child filesystem(s) would exceed any filesystem
742 * limits or adding additional snapshot(s) would exceed any snapshot limits.
743 * The prop argument indicates which limit to check.
744 *
745 * Note that all filesystem limits up to the root (or the highest
746 * initialized) filesystem or the given ancestor must be satisfied.
747 */
748 int
dsl_fs_ss_limit_check(dsl_dir_t * dd,uint64_t delta,zfs_prop_t prop,dsl_dir_t * ancestor,cred_t * cr)749 dsl_fs_ss_limit_check(dsl_dir_t *dd, uint64_t delta, zfs_prop_t prop,
750 dsl_dir_t *ancestor, cred_t *cr)
751 {
752 objset_t *os = dd->dd_pool->dp_meta_objset;
753 uint64_t limit, count;
754 char *count_prop;
755 enforce_res_t enforce;
756 int err = 0;
757
758 ASSERT(dsl_pool_config_held(dd->dd_pool));
759 ASSERT(prop == ZFS_PROP_FILESYSTEM_LIMIT ||
760 prop == ZFS_PROP_SNAPSHOT_LIMIT);
761
762 /*
763 * If we're allowed to change the limit, don't enforce the limit
764 * e.g. this can happen if a snapshot is taken by an administrative
765 * user in the global zone (i.e. a recursive snapshot by root).
766 * However, we must handle the case of delegated permissions where we
767 * are allowed to change the limit on the current dataset, but there
768 * is another limit in the tree above.
769 */
770 enforce = dsl_enforce_ds_ss_limits(dd, prop, cr);
771 if (enforce == ENFORCE_NEVER)
772 return (0);
773
774 /*
775 * e.g. if renaming a dataset with no snapshots, count adjustment
776 * is 0.
777 */
778 if (delta == 0)
779 return (0);
780
781 if (prop == ZFS_PROP_SNAPSHOT_LIMIT) {
782 /*
783 * We don't enforce the limit for temporary snapshots. This is
784 * indicated by a NULL cred_t argument.
785 */
786 if (cr == NULL)
787 return (0);
788
789 count_prop = DD_FIELD_SNAPSHOT_COUNT;
790 } else {
791 count_prop = DD_FIELD_FILESYSTEM_COUNT;
792 }
793
794 /*
795 * If an ancestor has been provided, stop checking the limit once we
796 * hit that dir. We need this during rename so that we don't overcount
797 * the check once we recurse up to the common ancestor.
798 */
799 if (ancestor == dd)
800 return (0);
801
802 /*
803 * If we hit an uninitialized node while recursing up the tree, we can
804 * stop since we know there is no limit here (or above). The counts are
805 * not valid on this node and we know we won't touch this node's counts.
806 */
807 if (!dsl_dir_is_zapified(dd) || zap_lookup(os, dd->dd_object,
808 count_prop, sizeof (count), 1, &count) == ENOENT)
809 return (0);
810
811 err = dsl_prop_get_dd(dd, zfs_prop_to_name(prop), 8, 1, &limit, NULL,
812 B_FALSE);
813 if (err != 0)
814 return (err);
815
816 /* Is there a limit which we've hit? */
817 if (enforce == ENFORCE_ALWAYS && (count + delta) > limit)
818 return (SET_ERROR(EDQUOT));
819
820 if (dd->dd_parent != NULL)
821 err = dsl_fs_ss_limit_check(dd->dd_parent, delta, prop,
822 ancestor, cr);
823
824 return (err);
825 }
826
827 /*
828 * Adjust the filesystem or snapshot count for the specified dsl_dir_t and all
829 * parents. When a new filesystem/snapshot is created, increment the count on
830 * all parents, and when a filesystem/snapshot is destroyed, decrement the
831 * count.
832 */
833 void
dsl_fs_ss_count_adjust(dsl_dir_t * dd,int64_t delta,const char * prop,dmu_tx_t * tx)834 dsl_fs_ss_count_adjust(dsl_dir_t *dd, int64_t delta, const char *prop,
835 dmu_tx_t *tx)
836 {
837 int err;
838 objset_t *os = dd->dd_pool->dp_meta_objset;
839 uint64_t count;
840
841 ASSERT(dsl_pool_config_held(dd->dd_pool));
842 ASSERT(dmu_tx_is_syncing(tx));
843 ASSERT(strcmp(prop, DD_FIELD_FILESYSTEM_COUNT) == 0 ||
844 strcmp(prop, DD_FIELD_SNAPSHOT_COUNT) == 0);
845
846 /*
847 * When we receive an incremental stream into a filesystem that already
848 * exists, a temporary clone is created. We don't count this temporary
849 * clone, whose name begins with a '%'. We also ignore hidden ($FREE,
850 * $MOS & $ORIGIN) objsets.
851 */
852 if ((dd->dd_myname[0] == '%' || dd->dd_myname[0] == '$') &&
853 strcmp(prop, DD_FIELD_FILESYSTEM_COUNT) == 0)
854 return;
855
856 /*
857 * e.g. if renaming a dataset with no snapshots, count adjustment is 0
858 */
859 if (delta == 0)
860 return;
861
862 /*
863 * If we hit an uninitialized node while recursing up the tree, we can
864 * stop since we know the counts are not valid on this node and we
865 * know we shouldn't touch this node's counts. An uninitialized count
866 * on the node indicates that either the feature has not yet been
867 * activated or there are no limits on this part of the tree.
868 */
869 if (!dsl_dir_is_zapified(dd) || (err = zap_lookup(os, dd->dd_object,
870 prop, sizeof (count), 1, &count)) == ENOENT)
871 return;
872 VERIFY0(err);
873
874 count += delta;
875 /* Use a signed verify to make sure we're not neg. */
876 VERIFY3S(count, >=, 0);
877
878 VERIFY0(zap_update(os, dd->dd_object, prop, sizeof (count), 1, &count,
879 tx));
880
881 /* Roll up this additional count into our ancestors */
882 if (dd->dd_parent != NULL)
883 dsl_fs_ss_count_adjust(dd->dd_parent, delta, prop, tx);
884 }
885
886 uint64_t
dsl_dir_create_sync(dsl_pool_t * dp,dsl_dir_t * pds,const char * name,dmu_tx_t * tx)887 dsl_dir_create_sync(dsl_pool_t *dp, dsl_dir_t *pds, const char *name,
888 dmu_tx_t *tx)
889 {
890 objset_t *mos = dp->dp_meta_objset;
891 uint64_t ddobj;
892 dsl_dir_phys_t *ddphys;
893 dmu_buf_t *dbuf;
894
895 ddobj = dmu_object_alloc(mos, DMU_OT_DSL_DIR, 0,
896 DMU_OT_DSL_DIR, sizeof (dsl_dir_phys_t), tx);
897 if (pds) {
898 VERIFY(0 == zap_add(mos, dsl_dir_phys(pds)->dd_child_dir_zapobj,
899 name, sizeof (uint64_t), 1, &ddobj, tx));
900 } else {
901 /* it's the root dir */
902 VERIFY(0 == zap_add(mos, DMU_POOL_DIRECTORY_OBJECT,
903 DMU_POOL_ROOT_DATASET, sizeof (uint64_t), 1, &ddobj, tx));
904 }
905 VERIFY(0 == dmu_bonus_hold(mos, ddobj, FTAG, &dbuf));
906 dmu_buf_will_dirty(dbuf, tx);
907 ddphys = dbuf->db_data;
908
909 ddphys->dd_creation_time = gethrestime_sec();
910 if (pds) {
911 ddphys->dd_parent_obj = pds->dd_object;
912
913 /* update the filesystem counts */
914 dsl_fs_ss_count_adjust(pds, 1, DD_FIELD_FILESYSTEM_COUNT, tx);
915 }
916 ddphys->dd_props_zapobj = zap_create(mos,
917 DMU_OT_DSL_PROPS, DMU_OT_NONE, 0, tx);
918 ddphys->dd_child_dir_zapobj = zap_create(mos,
919 DMU_OT_DSL_DIR_CHILD_MAP, DMU_OT_NONE, 0, tx);
920 if (spa_version(dp->dp_spa) >= SPA_VERSION_USED_BREAKDOWN)
921 ddphys->dd_flags |= DD_FLAG_USED_BREAKDOWN;
922 dmu_buf_rele(dbuf, FTAG);
923
924 return (ddobj);
925 }
926
927 boolean_t
dsl_dir_is_clone(dsl_dir_t * dd)928 dsl_dir_is_clone(dsl_dir_t *dd)
929 {
930 return (dsl_dir_phys(dd)->dd_origin_obj &&
931 (dd->dd_pool->dp_origin_snap == NULL ||
932 dsl_dir_phys(dd)->dd_origin_obj !=
933 dd->dd_pool->dp_origin_snap->ds_object));
934 }
935
936 void
dsl_dir_stats(dsl_dir_t * dd,nvlist_t * nv)937 dsl_dir_stats(dsl_dir_t *dd, nvlist_t *nv)
938 {
939 mutex_enter(&dd->dd_lock);
940 dsl_prop_nvlist_add_uint64(nv, ZFS_PROP_USED,
941 dsl_dir_phys(dd)->dd_used_bytes);
942 dsl_prop_nvlist_add_uint64(nv, ZFS_PROP_QUOTA,
943 dsl_dir_phys(dd)->dd_quota);
944 dsl_prop_nvlist_add_uint64(nv, ZFS_PROP_RESERVATION,
945 dsl_dir_phys(dd)->dd_reserved);
946 dsl_prop_nvlist_add_uint64(nv, ZFS_PROP_COMPRESSRATIO,
947 dsl_dir_phys(dd)->dd_compressed_bytes == 0 ? 100 :
948 (dsl_dir_phys(dd)->dd_uncompressed_bytes * 100 /
949 dsl_dir_phys(dd)->dd_compressed_bytes));
950 dsl_prop_nvlist_add_uint64(nv, ZFS_PROP_LOGICALUSED,
951 dsl_dir_phys(dd)->dd_uncompressed_bytes);
952 if (dsl_dir_phys(dd)->dd_flags & DD_FLAG_USED_BREAKDOWN) {
953 dsl_prop_nvlist_add_uint64(nv, ZFS_PROP_USEDSNAP,
954 dsl_dir_phys(dd)->dd_used_breakdown[DD_USED_SNAP]);
955 dsl_prop_nvlist_add_uint64(nv, ZFS_PROP_USEDDS,
956 dsl_dir_phys(dd)->dd_used_breakdown[DD_USED_HEAD]);
957 dsl_prop_nvlist_add_uint64(nv, ZFS_PROP_USEDREFRESERV,
958 dsl_dir_phys(dd)->dd_used_breakdown[DD_USED_REFRSRV]);
959 dsl_prop_nvlist_add_uint64(nv, ZFS_PROP_USEDCHILD,
960 dsl_dir_phys(dd)->dd_used_breakdown[DD_USED_CHILD] +
961 dsl_dir_phys(dd)->dd_used_breakdown[DD_USED_CHILD_RSRV]);
962 }
963 mutex_exit(&dd->dd_lock);
964
965 if (dsl_dir_is_zapified(dd)) {
966 uint64_t count;
967 objset_t *os = dd->dd_pool->dp_meta_objset;
968
969 if (zap_lookup(os, dd->dd_object, DD_FIELD_FILESYSTEM_COUNT,
970 sizeof (count), 1, &count) == 0) {
971 dsl_prop_nvlist_add_uint64(nv,
972 ZFS_PROP_FILESYSTEM_COUNT, count);
973 }
974 if (zap_lookup(os, dd->dd_object, DD_FIELD_SNAPSHOT_COUNT,
975 sizeof (count), 1, &count) == 0) {
976 dsl_prop_nvlist_add_uint64(nv,
977 ZFS_PROP_SNAPSHOT_COUNT, count);
978 }
979 }
980
981 if (dsl_dir_is_clone(dd)) {
982 dsl_dataset_t *ds;
983 char buf[ZFS_MAX_DATASET_NAME_LEN];
984
985 VERIFY0(dsl_dataset_hold_obj(dd->dd_pool,
986 dsl_dir_phys(dd)->dd_origin_obj, FTAG, &ds));
987 dsl_dataset_name(ds, buf);
988 dsl_dataset_rele(ds, FTAG);
989 dsl_prop_nvlist_add_string(nv, ZFS_PROP_ORIGIN, buf);
990 }
991 }
992
993 void
dsl_dir_dirty(dsl_dir_t * dd,dmu_tx_t * tx)994 dsl_dir_dirty(dsl_dir_t *dd, dmu_tx_t *tx)
995 {
996 dsl_pool_t *dp = dd->dd_pool;
997
998 ASSERT(dsl_dir_phys(dd));
999
1000 if (txg_list_add(&dp->dp_dirty_dirs, dd, tx->tx_txg)) {
1001 /* up the hold count until we can be written out */
1002 dmu_buf_add_ref(dd->dd_dbuf, dd);
1003 }
1004 }
1005
1006 static int64_t
parent_delta(dsl_dir_t * dd,uint64_t used,int64_t delta)1007 parent_delta(dsl_dir_t *dd, uint64_t used, int64_t delta)
1008 {
1009 uint64_t old_accounted = MAX(used, dsl_dir_phys(dd)->dd_reserved);
1010 uint64_t new_accounted =
1011 MAX(used + delta, dsl_dir_phys(dd)->dd_reserved);
1012 return (new_accounted - old_accounted);
1013 }
1014
1015 void
dsl_dir_sync(dsl_dir_t * dd,dmu_tx_t * tx)1016 dsl_dir_sync(dsl_dir_t *dd, dmu_tx_t *tx)
1017 {
1018 ASSERT(dmu_tx_is_syncing(tx));
1019
1020 mutex_enter(&dd->dd_lock);
1021 ASSERT0(dd->dd_tempreserved[tx->tx_txg&TXG_MASK]);
1022 dprintf_dd(dd, "txg=%llu towrite=%lluK\n", tx->tx_txg,
1023 dd->dd_space_towrite[tx->tx_txg&TXG_MASK] / 1024);
1024 dd->dd_space_towrite[tx->tx_txg&TXG_MASK] = 0;
1025 mutex_exit(&dd->dd_lock);
1026
1027 /* release the hold from dsl_dir_dirty */
1028 dmu_buf_rele(dd->dd_dbuf, dd);
1029 }
1030
1031 static uint64_t
dsl_dir_space_towrite(dsl_dir_t * dd)1032 dsl_dir_space_towrite(dsl_dir_t *dd)
1033 {
1034 uint64_t space = 0;
1035 int i;
1036
1037 ASSERT(MUTEX_HELD(&dd->dd_lock));
1038
1039 for (i = 0; i < TXG_SIZE; i++) {
1040 space += dd->dd_space_towrite[i&TXG_MASK];
1041 ASSERT3U(dd->dd_space_towrite[i&TXG_MASK], >=, 0);
1042 }
1043 return (space);
1044 }
1045
1046 /*
1047 * How much space would dd have available if ancestor had delta applied
1048 * to it? If ondiskonly is set, we're only interested in what's
1049 * on-disk, not estimated pending changes.
1050 */
1051 uint64_t
dsl_dir_space_available(dsl_dir_t * dd,dsl_dir_t * ancestor,int64_t delta,int ondiskonly)1052 dsl_dir_space_available(dsl_dir_t *dd,
1053 dsl_dir_t *ancestor, int64_t delta, int ondiskonly)
1054 {
1055 uint64_t parentspace, myspace, quota, used;
1056
1057 /*
1058 * If there are no restrictions otherwise, assume we have
1059 * unlimited space available.
1060 */
1061 quota = UINT64_MAX;
1062 parentspace = UINT64_MAX;
1063
1064 if (dd->dd_parent != NULL) {
1065 parentspace = dsl_dir_space_available(dd->dd_parent,
1066 ancestor, delta, ondiskonly);
1067 }
1068
1069 mutex_enter(&dd->dd_lock);
1070 if (dsl_dir_phys(dd)->dd_quota != 0)
1071 quota = dsl_dir_phys(dd)->dd_quota;
1072 used = dsl_dir_phys(dd)->dd_used_bytes;
1073 if (!ondiskonly)
1074 used += dsl_dir_space_towrite(dd);
1075
1076 if (dd->dd_parent == NULL) {
1077 uint64_t poolsize = dsl_pool_adjustedsize(dd->dd_pool, FALSE);
1078 quota = MIN(quota, poolsize);
1079 }
1080
1081 if (dsl_dir_phys(dd)->dd_reserved > used && parentspace != UINT64_MAX) {
1082 /*
1083 * We have some space reserved, in addition to what our
1084 * parent gave us.
1085 */
1086 parentspace += dsl_dir_phys(dd)->dd_reserved - used;
1087 }
1088
1089 if (dd == ancestor) {
1090 ASSERT(delta <= 0);
1091 ASSERT(used >= -delta);
1092 used += delta;
1093 if (parentspace != UINT64_MAX)
1094 parentspace -= delta;
1095 }
1096
1097 if (used > quota) {
1098 /* over quota */
1099 myspace = 0;
1100 } else {
1101 /*
1102 * the lesser of the space provided by our parent and
1103 * the space left in our quota
1104 */
1105 myspace = MIN(parentspace, quota - used);
1106 }
1107
1108 mutex_exit(&dd->dd_lock);
1109
1110 return (myspace);
1111 }
1112
1113 struct tempreserve {
1114 list_node_t tr_node;
1115 dsl_dir_t *tr_ds;
1116 uint64_t tr_size;
1117 };
1118
1119 static int
dsl_dir_tempreserve_impl(dsl_dir_t * dd,uint64_t asize,boolean_t netfree,boolean_t ignorequota,boolean_t checkrefquota,list_t * tr_list,dmu_tx_t * tx,boolean_t first)1120 dsl_dir_tempreserve_impl(dsl_dir_t *dd, uint64_t asize, boolean_t netfree,
1121 boolean_t ignorequota, boolean_t checkrefquota, list_t *tr_list,
1122 dmu_tx_t *tx, boolean_t first)
1123 {
1124 uint64_t txg = tx->tx_txg;
1125 uint64_t est_inflight, used_on_disk, quota, parent_rsrv;
1126 uint64_t deferred = 0;
1127 struct tempreserve *tr;
1128 int retval = EDQUOT;
1129 int txgidx = txg & TXG_MASK;
1130 int i;
1131 uint64_t ref_rsrv = 0;
1132
1133 ASSERT3U(txg, !=, 0);
1134 ASSERT3S(asize, >, 0);
1135
1136 mutex_enter(&dd->dd_lock);
1137
1138 /*
1139 * Check against the dsl_dir's quota. We don't add in the delta
1140 * when checking for over-quota because they get one free hit.
1141 */
1142 est_inflight = dsl_dir_space_towrite(dd);
1143 for (i = 0; i < TXG_SIZE; i++)
1144 est_inflight += dd->dd_tempreserved[i];
1145 used_on_disk = dsl_dir_phys(dd)->dd_used_bytes;
1146
1147 /*
1148 * On the first iteration, fetch the dataset's used-on-disk and
1149 * refreservation values. Also, if checkrefquota is set, test if
1150 * allocating this space would exceed the dataset's refquota.
1151 */
1152 if (first && tx->tx_objset) {
1153 int error;
1154 dsl_dataset_t *ds = tx->tx_objset->os_dsl_dataset;
1155
1156 error = dsl_dataset_check_quota(ds, checkrefquota,
1157 asize, est_inflight, &used_on_disk, &ref_rsrv);
1158 if (error) {
1159 mutex_exit(&dd->dd_lock);
1160 return (error);
1161 }
1162 }
1163
1164 /*
1165 * If this transaction will result in a net free of space,
1166 * we want to let it through.
1167 */
1168 if (ignorequota || netfree || dsl_dir_phys(dd)->dd_quota == 0)
1169 quota = UINT64_MAX;
1170 else
1171 quota = dsl_dir_phys(dd)->dd_quota;
1172
1173 /*
1174 * Adjust the quota against the actual pool size at the root
1175 * minus any outstanding deferred frees.
1176 * To ensure that it's possible to remove files from a full
1177 * pool without inducing transient overcommits, we throttle
1178 * netfree transactions against a quota that is slightly larger,
1179 * but still within the pool's allocation slop. In cases where
1180 * we're very close to full, this will allow a steady trickle of
1181 * removes to get through.
1182 */
1183 if (dd->dd_parent == NULL) {
1184 spa_t *spa = dd->dd_pool->dp_spa;
1185 uint64_t poolsize = dsl_pool_adjustedsize(dd->dd_pool, netfree);
1186 deferred = metaslab_class_get_deferred(spa_normal_class(spa));
1187 if (poolsize - deferred < quota) {
1188 quota = poolsize - deferred;
1189 retval = ENOSPC;
1190 }
1191 }
1192
1193 /*
1194 * If they are requesting more space, and our current estimate
1195 * is over quota, they get to try again unless the actual
1196 * on-disk is over quota and there are no pending changes (which
1197 * may free up space for us).
1198 */
1199 if (used_on_disk + est_inflight >= quota) {
1200 if (est_inflight > 0 || used_on_disk < quota ||
1201 (retval == ENOSPC && used_on_disk < quota + deferred))
1202 retval = ERESTART;
1203 dprintf_dd(dd, "failing: used=%lluK inflight = %lluK "
1204 "quota=%lluK tr=%lluK err=%d\n",
1205 used_on_disk>>10, est_inflight>>10,
1206 quota>>10, asize>>10, retval);
1207 mutex_exit(&dd->dd_lock);
1208 return (SET_ERROR(retval));
1209 }
1210
1211 /* We need to up our estimated delta before dropping dd_lock */
1212 dd->dd_tempreserved[txgidx] += asize;
1213
1214 parent_rsrv = parent_delta(dd, used_on_disk + est_inflight,
1215 asize - ref_rsrv);
1216 mutex_exit(&dd->dd_lock);
1217
1218 tr = kmem_zalloc(sizeof (struct tempreserve), KM_SLEEP);
1219 tr->tr_ds = dd;
1220 tr->tr_size = asize;
1221 list_insert_tail(tr_list, tr);
1222
1223 /* see if it's OK with our parent */
1224 if (dd->dd_parent && parent_rsrv) {
1225 boolean_t ismos = (dsl_dir_phys(dd)->dd_head_dataset_obj == 0);
1226
1227 return (dsl_dir_tempreserve_impl(dd->dd_parent,
1228 parent_rsrv, netfree, ismos, TRUE, tr_list, tx, FALSE));
1229 } else {
1230 return (0);
1231 }
1232 }
1233
1234 /*
1235 * Reserve space in this dsl_dir, to be used in this tx's txg.
1236 * After the space has been dirtied (and dsl_dir_willuse_space()
1237 * has been called), the reservation should be canceled, using
1238 * dsl_dir_tempreserve_clear().
1239 */
1240 int
dsl_dir_tempreserve_space(dsl_dir_t * dd,uint64_t lsize,uint64_t asize,uint64_t fsize,uint64_t usize,void ** tr_cookiep,dmu_tx_t * tx)1241 dsl_dir_tempreserve_space(dsl_dir_t *dd, uint64_t lsize, uint64_t asize,
1242 uint64_t fsize, uint64_t usize, void **tr_cookiep, dmu_tx_t *tx)
1243 {
1244 int err;
1245 list_t *tr_list;
1246
1247 if (asize == 0) {
1248 *tr_cookiep = NULL;
1249 return (0);
1250 }
1251
1252 tr_list = kmem_alloc(sizeof (list_t), KM_SLEEP);
1253 list_create(tr_list, sizeof (struct tempreserve),
1254 offsetof(struct tempreserve, tr_node));
1255 ASSERT3S(asize, >, 0);
1256 ASSERT3S(fsize, >=, 0);
1257
1258 err = arc_tempreserve_space(lsize, tx->tx_txg);
1259 if (err == 0) {
1260 struct tempreserve *tr;
1261
1262 tr = kmem_zalloc(sizeof (struct tempreserve), KM_SLEEP);
1263 tr->tr_size = lsize;
1264 list_insert_tail(tr_list, tr);
1265 } else {
1266 if (err == EAGAIN) {
1267 /*
1268 * If arc_memory_throttle() detected that pageout
1269 * is running and we are low on memory, we delay new
1270 * non-pageout transactions to give pageout an
1271 * advantage.
1272 *
1273 * It is unfortunate to be delaying while the caller's
1274 * locks are held.
1275 */
1276 txg_delay(dd->dd_pool, tx->tx_txg,
1277 MSEC2NSEC(10), MSEC2NSEC(10));
1278 err = SET_ERROR(ERESTART);
1279 }
1280 }
1281
1282 if (err == 0) {
1283 err = dsl_dir_tempreserve_impl(dd, asize, fsize >= asize,
1284 FALSE, asize > usize, tr_list, tx, TRUE);
1285 }
1286
1287 if (err != 0)
1288 dsl_dir_tempreserve_clear(tr_list, tx);
1289 else
1290 *tr_cookiep = tr_list;
1291
1292 return (err);
1293 }
1294
1295 /*
1296 * Clear a temporary reservation that we previously made with
1297 * dsl_dir_tempreserve_space().
1298 */
1299 void
dsl_dir_tempreserve_clear(void * tr_cookie,dmu_tx_t * tx)1300 dsl_dir_tempreserve_clear(void *tr_cookie, dmu_tx_t *tx)
1301 {
1302 int txgidx = tx->tx_txg & TXG_MASK;
1303 list_t *tr_list = tr_cookie;
1304 struct tempreserve *tr;
1305
1306 ASSERT3U(tx->tx_txg, !=, 0);
1307
1308 if (tr_cookie == NULL)
1309 return;
1310
1311 while ((tr = list_head(tr_list)) != NULL) {
1312 if (tr->tr_ds) {
1313 mutex_enter(&tr->tr_ds->dd_lock);
1314 ASSERT3U(tr->tr_ds->dd_tempreserved[txgidx], >=,
1315 tr->tr_size);
1316 tr->tr_ds->dd_tempreserved[txgidx] -= tr->tr_size;
1317 mutex_exit(&tr->tr_ds->dd_lock);
1318 } else {
1319 arc_tempreserve_clear(tr->tr_size);
1320 }
1321 list_remove(tr_list, tr);
1322 kmem_free(tr, sizeof (struct tempreserve));
1323 }
1324
1325 kmem_free(tr_list, sizeof (list_t));
1326 }
1327
1328 /*
1329 * This should be called from open context when we think we're going to write
1330 * or free space, for example when dirtying data. Be conservative; it's okay
1331 * to write less space or free more, but we don't want to write more or free
1332 * less than the amount specified.
1333 */
1334 void
dsl_dir_willuse_space(dsl_dir_t * dd,int64_t space,dmu_tx_t * tx)1335 dsl_dir_willuse_space(dsl_dir_t *dd, int64_t space, dmu_tx_t *tx)
1336 {
1337 int64_t parent_space;
1338 uint64_t est_used;
1339
1340 mutex_enter(&dd->dd_lock);
1341 if (space > 0)
1342 dd->dd_space_towrite[tx->tx_txg & TXG_MASK] += space;
1343
1344 est_used = dsl_dir_space_towrite(dd) + dsl_dir_phys(dd)->dd_used_bytes;
1345 parent_space = parent_delta(dd, est_used, space);
1346 mutex_exit(&dd->dd_lock);
1347
1348 /* Make sure that we clean up dd_space_to* */
1349 dsl_dir_dirty(dd, tx);
1350
1351 /* XXX this is potentially expensive and unnecessary... */
1352 if (parent_space && dd->dd_parent)
1353 dsl_dir_willuse_space(dd->dd_parent, parent_space, tx);
1354 }
1355
1356 /* call from syncing context when we actually write/free space for this dd */
1357 void
dsl_dir_diduse_space(dsl_dir_t * dd,dd_used_t type,int64_t used,int64_t compressed,int64_t uncompressed,dmu_tx_t * tx)1358 dsl_dir_diduse_space(dsl_dir_t *dd, dd_used_t type,
1359 int64_t used, int64_t compressed, int64_t uncompressed, dmu_tx_t *tx)
1360 {
1361 int64_t accounted_delta;
1362
1363 /*
1364 * dsl_dataset_set_refreservation_sync_impl() calls this with
1365 * dd_lock held, so that it can atomically update
1366 * ds->ds_reserved and the dsl_dir accounting, so that
1367 * dsl_dataset_check_quota() can see dataset and dir accounting
1368 * consistently.
1369 */
1370 boolean_t needlock = !MUTEX_HELD(&dd->dd_lock);
1371
1372 ASSERT(dmu_tx_is_syncing(tx));
1373 ASSERT(type < DD_USED_NUM);
1374
1375 dmu_buf_will_dirty(dd->dd_dbuf, tx);
1376
1377 if (needlock)
1378 mutex_enter(&dd->dd_lock);
1379 accounted_delta =
1380 parent_delta(dd, dsl_dir_phys(dd)->dd_used_bytes, used);
1381 ASSERT(used >= 0 || dsl_dir_phys(dd)->dd_used_bytes >= -used);
1382 ASSERT(compressed >= 0 ||
1383 dsl_dir_phys(dd)->dd_compressed_bytes >= -compressed);
1384 ASSERT(uncompressed >= 0 ||
1385 dsl_dir_phys(dd)->dd_uncompressed_bytes >= -uncompressed);
1386 dsl_dir_phys(dd)->dd_used_bytes += used;
1387 dsl_dir_phys(dd)->dd_uncompressed_bytes += uncompressed;
1388 dsl_dir_phys(dd)->dd_compressed_bytes += compressed;
1389
1390 if (dsl_dir_phys(dd)->dd_flags & DD_FLAG_USED_BREAKDOWN) {
1391 ASSERT(used > 0 ||
1392 dsl_dir_phys(dd)->dd_used_breakdown[type] >= -used);
1393 dsl_dir_phys(dd)->dd_used_breakdown[type] += used;
1394 #ifdef DEBUG
1395 dd_used_t t;
1396 uint64_t u = 0;
1397 for (t = 0; t < DD_USED_NUM; t++)
1398 u += dsl_dir_phys(dd)->dd_used_breakdown[t];
1399 ASSERT3U(u, ==, dsl_dir_phys(dd)->dd_used_bytes);
1400 #endif
1401 }
1402 if (needlock)
1403 mutex_exit(&dd->dd_lock);
1404
1405 if (dd->dd_parent != NULL) {
1406 dsl_dir_diduse_space(dd->dd_parent, DD_USED_CHILD,
1407 accounted_delta, compressed, uncompressed, tx);
1408 dsl_dir_transfer_space(dd->dd_parent,
1409 used - accounted_delta,
1410 DD_USED_CHILD_RSRV, DD_USED_CHILD, NULL);
1411 }
1412 }
1413
1414 void
dsl_dir_transfer_space(dsl_dir_t * dd,int64_t delta,dd_used_t oldtype,dd_used_t newtype,dmu_tx_t * tx)1415 dsl_dir_transfer_space(dsl_dir_t *dd, int64_t delta,
1416 dd_used_t oldtype, dd_used_t newtype, dmu_tx_t *tx)
1417 {
1418 ASSERT(tx == NULL || dmu_tx_is_syncing(tx));
1419 ASSERT(oldtype < DD_USED_NUM);
1420 ASSERT(newtype < DD_USED_NUM);
1421
1422 if (delta == 0 ||
1423 !(dsl_dir_phys(dd)->dd_flags & DD_FLAG_USED_BREAKDOWN))
1424 return;
1425
1426 if (tx != NULL)
1427 dmu_buf_will_dirty(dd->dd_dbuf, tx);
1428 mutex_enter(&dd->dd_lock);
1429 ASSERT(delta > 0 ?
1430 dsl_dir_phys(dd)->dd_used_breakdown[oldtype] >= delta :
1431 dsl_dir_phys(dd)->dd_used_breakdown[newtype] >= -delta);
1432 ASSERT(dsl_dir_phys(dd)->dd_used_bytes >= ABS(delta));
1433 dsl_dir_phys(dd)->dd_used_breakdown[oldtype] -= delta;
1434 dsl_dir_phys(dd)->dd_used_breakdown[newtype] += delta;
1435 mutex_exit(&dd->dd_lock);
1436 }
1437
1438 typedef struct dsl_dir_set_qr_arg {
1439 const char *ddsqra_name;
1440 zprop_source_t ddsqra_source;
1441 uint64_t ddsqra_value;
1442 } dsl_dir_set_qr_arg_t;
1443
1444 static int
dsl_dir_set_quota_check(void * arg,dmu_tx_t * tx)1445 dsl_dir_set_quota_check(void *arg, dmu_tx_t *tx)
1446 {
1447 dsl_dir_set_qr_arg_t *ddsqra = arg;
1448 dsl_pool_t *dp = dmu_tx_pool(tx);
1449 dsl_dataset_t *ds;
1450 int error;
1451 uint64_t towrite, newval;
1452
1453 error = dsl_dataset_hold(dp, ddsqra->ddsqra_name, FTAG, &ds);
1454 if (error != 0)
1455 return (error);
1456
1457 error = dsl_prop_predict(ds->ds_dir, "quota",
1458 ddsqra->ddsqra_source, ddsqra->ddsqra_value, &newval);
1459 if (error != 0) {
1460 dsl_dataset_rele(ds, FTAG);
1461 return (error);
1462 }
1463
1464 if (newval == 0) {
1465 dsl_dataset_rele(ds, FTAG);
1466 return (0);
1467 }
1468
1469 mutex_enter(&ds->ds_dir->dd_lock);
1470 /*
1471 * If we are doing the preliminary check in open context, and
1472 * there are pending changes, then don't fail it, since the
1473 * pending changes could under-estimate the amount of space to be
1474 * freed up.
1475 */
1476 towrite = dsl_dir_space_towrite(ds->ds_dir);
1477 if ((dmu_tx_is_syncing(tx) || towrite == 0) &&
1478 (newval < dsl_dir_phys(ds->ds_dir)->dd_reserved ||
1479 newval < dsl_dir_phys(ds->ds_dir)->dd_used_bytes + towrite)) {
1480 error = SET_ERROR(ENOSPC);
1481 }
1482 mutex_exit(&ds->ds_dir->dd_lock);
1483 dsl_dataset_rele(ds, FTAG);
1484 return (error);
1485 }
1486
1487 static void
dsl_dir_set_quota_sync(void * arg,dmu_tx_t * tx)1488 dsl_dir_set_quota_sync(void *arg, dmu_tx_t *tx)
1489 {
1490 dsl_dir_set_qr_arg_t *ddsqra = arg;
1491 dsl_pool_t *dp = dmu_tx_pool(tx);
1492 dsl_dataset_t *ds;
1493 uint64_t newval;
1494
1495 VERIFY0(dsl_dataset_hold(dp, ddsqra->ddsqra_name, FTAG, &ds));
1496
1497 if (spa_version(dp->dp_spa) >= SPA_VERSION_RECVD_PROPS) {
1498 dsl_prop_set_sync_impl(ds, zfs_prop_to_name(ZFS_PROP_QUOTA),
1499 ddsqra->ddsqra_source, sizeof (ddsqra->ddsqra_value), 1,
1500 &ddsqra->ddsqra_value, tx);
1501
1502 VERIFY0(dsl_prop_get_int_ds(ds,
1503 zfs_prop_to_name(ZFS_PROP_QUOTA), &newval));
1504 } else {
1505 newval = ddsqra->ddsqra_value;
1506 spa_history_log_internal_ds(ds, "set", tx, "%s=%lld",
1507 zfs_prop_to_name(ZFS_PROP_QUOTA), (longlong_t)newval);
1508 }
1509
1510 dmu_buf_will_dirty(ds->ds_dir->dd_dbuf, tx);
1511 mutex_enter(&ds->ds_dir->dd_lock);
1512 dsl_dir_phys(ds->ds_dir)->dd_quota = newval;
1513 mutex_exit(&ds->ds_dir->dd_lock);
1514 dsl_dataset_rele(ds, FTAG);
1515 }
1516
1517 int
dsl_dir_set_quota(const char * ddname,zprop_source_t source,uint64_t quota)1518 dsl_dir_set_quota(const char *ddname, zprop_source_t source, uint64_t quota)
1519 {
1520 dsl_dir_set_qr_arg_t ddsqra;
1521
1522 ddsqra.ddsqra_name = ddname;
1523 ddsqra.ddsqra_source = source;
1524 ddsqra.ddsqra_value = quota;
1525
1526 return (dsl_sync_task(ddname, dsl_dir_set_quota_check,
1527 dsl_dir_set_quota_sync, &ddsqra, 0, ZFS_SPACE_CHECK_NONE));
1528 }
1529
1530 int
dsl_dir_set_reservation_check(void * arg,dmu_tx_t * tx)1531 dsl_dir_set_reservation_check(void *arg, dmu_tx_t *tx)
1532 {
1533 dsl_dir_set_qr_arg_t *ddsqra = arg;
1534 dsl_pool_t *dp = dmu_tx_pool(tx);
1535 dsl_dataset_t *ds;
1536 dsl_dir_t *dd;
1537 uint64_t newval, used, avail;
1538 int error;
1539
1540 error = dsl_dataset_hold(dp, ddsqra->ddsqra_name, FTAG, &ds);
1541 if (error != 0)
1542 return (error);
1543 dd = ds->ds_dir;
1544
1545 /*
1546 * If we are doing the preliminary check in open context, the
1547 * space estimates may be inaccurate.
1548 */
1549 if (!dmu_tx_is_syncing(tx)) {
1550 dsl_dataset_rele(ds, FTAG);
1551 return (0);
1552 }
1553
1554 error = dsl_prop_predict(ds->ds_dir,
1555 zfs_prop_to_name(ZFS_PROP_RESERVATION),
1556 ddsqra->ddsqra_source, ddsqra->ddsqra_value, &newval);
1557 if (error != 0) {
1558 dsl_dataset_rele(ds, FTAG);
1559 return (error);
1560 }
1561
1562 mutex_enter(&dd->dd_lock);
1563 used = dsl_dir_phys(dd)->dd_used_bytes;
1564 mutex_exit(&dd->dd_lock);
1565
1566 if (dd->dd_parent) {
1567 avail = dsl_dir_space_available(dd->dd_parent,
1568 NULL, 0, FALSE);
1569 } else {
1570 avail = dsl_pool_adjustedsize(dd->dd_pool, B_FALSE) - used;
1571 }
1572
1573 if (MAX(used, newval) > MAX(used, dsl_dir_phys(dd)->dd_reserved)) {
1574 uint64_t delta = MAX(used, newval) -
1575 MAX(used, dsl_dir_phys(dd)->dd_reserved);
1576
1577 if (delta > avail ||
1578 (dsl_dir_phys(dd)->dd_quota > 0 &&
1579 newval > dsl_dir_phys(dd)->dd_quota))
1580 error = SET_ERROR(ENOSPC);
1581 }
1582
1583 dsl_dataset_rele(ds, FTAG);
1584 return (error);
1585 }
1586
1587 void
dsl_dir_set_reservation_sync_impl(dsl_dir_t * dd,uint64_t value,dmu_tx_t * tx)1588 dsl_dir_set_reservation_sync_impl(dsl_dir_t *dd, uint64_t value, dmu_tx_t *tx)
1589 {
1590 uint64_t used;
1591 int64_t delta;
1592
1593 dmu_buf_will_dirty(dd->dd_dbuf, tx);
1594
1595 mutex_enter(&dd->dd_lock);
1596 used = dsl_dir_phys(dd)->dd_used_bytes;
1597 delta = MAX(used, value) - MAX(used, dsl_dir_phys(dd)->dd_reserved);
1598 dsl_dir_phys(dd)->dd_reserved = value;
1599
1600 if (dd->dd_parent != NULL) {
1601 /* Roll up this additional usage into our ancestors */
1602 dsl_dir_diduse_space(dd->dd_parent, DD_USED_CHILD_RSRV,
1603 delta, 0, 0, tx);
1604 }
1605 mutex_exit(&dd->dd_lock);
1606 }
1607
1608 static void
dsl_dir_set_reservation_sync(void * arg,dmu_tx_t * tx)1609 dsl_dir_set_reservation_sync(void *arg, dmu_tx_t *tx)
1610 {
1611 dsl_dir_set_qr_arg_t *ddsqra = arg;
1612 dsl_pool_t *dp = dmu_tx_pool(tx);
1613 dsl_dataset_t *ds;
1614 uint64_t newval;
1615
1616 VERIFY0(dsl_dataset_hold(dp, ddsqra->ddsqra_name, FTAG, &ds));
1617
1618 if (spa_version(dp->dp_spa) >= SPA_VERSION_RECVD_PROPS) {
1619 dsl_prop_set_sync_impl(ds,
1620 zfs_prop_to_name(ZFS_PROP_RESERVATION),
1621 ddsqra->ddsqra_source, sizeof (ddsqra->ddsqra_value), 1,
1622 &ddsqra->ddsqra_value, tx);
1623
1624 VERIFY0(dsl_prop_get_int_ds(ds,
1625 zfs_prop_to_name(ZFS_PROP_RESERVATION), &newval));
1626 } else {
1627 newval = ddsqra->ddsqra_value;
1628 spa_history_log_internal_ds(ds, "set", tx, "%s=%lld",
1629 zfs_prop_to_name(ZFS_PROP_RESERVATION),
1630 (longlong_t)newval);
1631 }
1632
1633 dsl_dir_set_reservation_sync_impl(ds->ds_dir, newval, tx);
1634 dsl_dataset_rele(ds, FTAG);
1635 }
1636
1637 int
dsl_dir_set_reservation(const char * ddname,zprop_source_t source,uint64_t reservation)1638 dsl_dir_set_reservation(const char *ddname, zprop_source_t source,
1639 uint64_t reservation)
1640 {
1641 dsl_dir_set_qr_arg_t ddsqra;
1642
1643 ddsqra.ddsqra_name = ddname;
1644 ddsqra.ddsqra_source = source;
1645 ddsqra.ddsqra_value = reservation;
1646
1647 return (dsl_sync_task(ddname, dsl_dir_set_reservation_check,
1648 dsl_dir_set_reservation_sync, &ddsqra, 0, ZFS_SPACE_CHECK_NONE));
1649 }
1650
1651 static dsl_dir_t *
closest_common_ancestor(dsl_dir_t * ds1,dsl_dir_t * ds2)1652 closest_common_ancestor(dsl_dir_t *ds1, dsl_dir_t *ds2)
1653 {
1654 for (; ds1; ds1 = ds1->dd_parent) {
1655 dsl_dir_t *dd;
1656 for (dd = ds2; dd; dd = dd->dd_parent) {
1657 if (ds1 == dd)
1658 return (dd);
1659 }
1660 }
1661 return (NULL);
1662 }
1663
1664 /*
1665 * If delta is applied to dd, how much of that delta would be applied to
1666 * ancestor? Syncing context only.
1667 */
1668 static int64_t
would_change(dsl_dir_t * dd,int64_t delta,dsl_dir_t * ancestor)1669 would_change(dsl_dir_t *dd, int64_t delta, dsl_dir_t *ancestor)
1670 {
1671 if (dd == ancestor)
1672 return (delta);
1673
1674 mutex_enter(&dd->dd_lock);
1675 delta = parent_delta(dd, dsl_dir_phys(dd)->dd_used_bytes, delta);
1676 mutex_exit(&dd->dd_lock);
1677 return (would_change(dd->dd_parent, delta, ancestor));
1678 }
1679
1680 typedef struct dsl_dir_rename_arg {
1681 const char *ddra_oldname;
1682 const char *ddra_newname;
1683 cred_t *ddra_cred;
1684 } dsl_dir_rename_arg_t;
1685
1686 /* ARGSUSED */
1687 static int
dsl_valid_rename(dsl_pool_t * dp,dsl_dataset_t * ds,void * arg)1688 dsl_valid_rename(dsl_pool_t *dp, dsl_dataset_t *ds, void *arg)
1689 {
1690 int *deltap = arg;
1691 char namebuf[ZFS_MAX_DATASET_NAME_LEN];
1692
1693 dsl_dataset_name(ds, namebuf);
1694
1695 if (strlen(namebuf) + *deltap >= ZFS_MAX_DATASET_NAME_LEN)
1696 return (SET_ERROR(ENAMETOOLONG));
1697 return (0);
1698 }
1699
1700 static int
dsl_dir_rename_check(void * arg,dmu_tx_t * tx)1701 dsl_dir_rename_check(void *arg, dmu_tx_t *tx)
1702 {
1703 dsl_dir_rename_arg_t *ddra = arg;
1704 dsl_pool_t *dp = dmu_tx_pool(tx);
1705 dsl_dir_t *dd, *newparent;
1706 const char *mynewname;
1707 int error;
1708 int delta = strlen(ddra->ddra_newname) - strlen(ddra->ddra_oldname);
1709
1710 /* target dir should exist */
1711 error = dsl_dir_hold(dp, ddra->ddra_oldname, FTAG, &dd, NULL);
1712 if (error != 0)
1713 return (error);
1714
1715 /* new parent should exist */
1716 error = dsl_dir_hold(dp, ddra->ddra_newname, FTAG,
1717 &newparent, &mynewname);
1718 if (error != 0) {
1719 dsl_dir_rele(dd, FTAG);
1720 return (error);
1721 }
1722
1723 /* can't rename to different pool */
1724 if (dd->dd_pool != newparent->dd_pool) {
1725 dsl_dir_rele(newparent, FTAG);
1726 dsl_dir_rele(dd, FTAG);
1727 return (SET_ERROR(EXDEV));
1728 }
1729
1730 /* new name should not already exist */
1731 if (mynewname == NULL) {
1732 dsl_dir_rele(newparent, FTAG);
1733 dsl_dir_rele(dd, FTAG);
1734 return (SET_ERROR(EEXIST));
1735 }
1736
1737 /* if the name length is growing, validate child name lengths */
1738 if (delta > 0) {
1739 error = dmu_objset_find_dp(dp, dd->dd_object, dsl_valid_rename,
1740 &delta, DS_FIND_CHILDREN | DS_FIND_SNAPSHOTS);
1741 if (error != 0) {
1742 dsl_dir_rele(newparent, FTAG);
1743 dsl_dir_rele(dd, FTAG);
1744 return (error);
1745 }
1746 }
1747
1748 if (dmu_tx_is_syncing(tx)) {
1749 if (spa_feature_is_active(dp->dp_spa,
1750 SPA_FEATURE_FS_SS_LIMIT)) {
1751 /*
1752 * Although this is the check function and we don't
1753 * normally make on-disk changes in check functions,
1754 * we need to do that here.
1755 *
1756 * Ensure this portion of the tree's counts have been
1757 * initialized in case the new parent has limits set.
1758 */
1759 dsl_dir_init_fs_ss_count(dd, tx);
1760 }
1761 }
1762
1763 if (newparent != dd->dd_parent) {
1764 /* is there enough space? */
1765 uint64_t myspace =
1766 MAX(dsl_dir_phys(dd)->dd_used_bytes,
1767 dsl_dir_phys(dd)->dd_reserved);
1768 objset_t *os = dd->dd_pool->dp_meta_objset;
1769 uint64_t fs_cnt = 0;
1770 uint64_t ss_cnt = 0;
1771
1772 if (dsl_dir_is_zapified(dd)) {
1773 int err;
1774
1775 err = zap_lookup(os, dd->dd_object,
1776 DD_FIELD_FILESYSTEM_COUNT, sizeof (fs_cnt), 1,
1777 &fs_cnt);
1778 if (err != ENOENT && err != 0) {
1779 dsl_dir_rele(newparent, FTAG);
1780 dsl_dir_rele(dd, FTAG);
1781 return (err);
1782 }
1783
1784 /*
1785 * have to add 1 for the filesystem itself that we're
1786 * moving
1787 */
1788 fs_cnt++;
1789
1790 err = zap_lookup(os, dd->dd_object,
1791 DD_FIELD_SNAPSHOT_COUNT, sizeof (ss_cnt), 1,
1792 &ss_cnt);
1793 if (err != ENOENT && err != 0) {
1794 dsl_dir_rele(newparent, FTAG);
1795 dsl_dir_rele(dd, FTAG);
1796 return (err);
1797 }
1798 }
1799
1800 /* no rename into our descendant */
1801 if (closest_common_ancestor(dd, newparent) == dd) {
1802 dsl_dir_rele(newparent, FTAG);
1803 dsl_dir_rele(dd, FTAG);
1804 return (SET_ERROR(EINVAL));
1805 }
1806
1807 error = dsl_dir_transfer_possible(dd->dd_parent,
1808 newparent, fs_cnt, ss_cnt, myspace, ddra->ddra_cred);
1809 if (error != 0) {
1810 dsl_dir_rele(newparent, FTAG);
1811 dsl_dir_rele(dd, FTAG);
1812 return (error);
1813 }
1814 }
1815
1816 dsl_dir_rele(newparent, FTAG);
1817 dsl_dir_rele(dd, FTAG);
1818 return (0);
1819 }
1820
1821 static void
dsl_dir_rename_sync(void * arg,dmu_tx_t * tx)1822 dsl_dir_rename_sync(void *arg, dmu_tx_t *tx)
1823 {
1824 dsl_dir_rename_arg_t *ddra = arg;
1825 dsl_pool_t *dp = dmu_tx_pool(tx);
1826 dsl_dir_t *dd, *newparent;
1827 const char *mynewname;
1828 int error;
1829 objset_t *mos = dp->dp_meta_objset;
1830
1831 VERIFY0(dsl_dir_hold(dp, ddra->ddra_oldname, FTAG, &dd, NULL));
1832 VERIFY0(dsl_dir_hold(dp, ddra->ddra_newname, FTAG, &newparent,
1833 &mynewname));
1834
1835 /* Log this before we change the name. */
1836 spa_history_log_internal_dd(dd, "rename", tx,
1837 "-> %s", ddra->ddra_newname);
1838
1839 if (newparent != dd->dd_parent) {
1840 objset_t *os = dd->dd_pool->dp_meta_objset;
1841 uint64_t fs_cnt = 0;
1842 uint64_t ss_cnt = 0;
1843
1844 /*
1845 * We already made sure the dd counts were initialized in the
1846 * check function.
1847 */
1848 if (spa_feature_is_active(dp->dp_spa,
1849 SPA_FEATURE_FS_SS_LIMIT)) {
1850 VERIFY0(zap_lookup(os, dd->dd_object,
1851 DD_FIELD_FILESYSTEM_COUNT, sizeof (fs_cnt), 1,
1852 &fs_cnt));
1853 /* add 1 for the filesystem itself that we're moving */
1854 fs_cnt++;
1855
1856 VERIFY0(zap_lookup(os, dd->dd_object,
1857 DD_FIELD_SNAPSHOT_COUNT, sizeof (ss_cnt), 1,
1858 &ss_cnt));
1859 }
1860
1861 dsl_fs_ss_count_adjust(dd->dd_parent, -fs_cnt,
1862 DD_FIELD_FILESYSTEM_COUNT, tx);
1863 dsl_fs_ss_count_adjust(newparent, fs_cnt,
1864 DD_FIELD_FILESYSTEM_COUNT, tx);
1865
1866 dsl_fs_ss_count_adjust(dd->dd_parent, -ss_cnt,
1867 DD_FIELD_SNAPSHOT_COUNT, tx);
1868 dsl_fs_ss_count_adjust(newparent, ss_cnt,
1869 DD_FIELD_SNAPSHOT_COUNT, tx);
1870
1871 dsl_dir_diduse_space(dd->dd_parent, DD_USED_CHILD,
1872 -dsl_dir_phys(dd)->dd_used_bytes,
1873 -dsl_dir_phys(dd)->dd_compressed_bytes,
1874 -dsl_dir_phys(dd)->dd_uncompressed_bytes, tx);
1875 dsl_dir_diduse_space(newparent, DD_USED_CHILD,
1876 dsl_dir_phys(dd)->dd_used_bytes,
1877 dsl_dir_phys(dd)->dd_compressed_bytes,
1878 dsl_dir_phys(dd)->dd_uncompressed_bytes, tx);
1879
1880 if (dsl_dir_phys(dd)->dd_reserved >
1881 dsl_dir_phys(dd)->dd_used_bytes) {
1882 uint64_t unused_rsrv = dsl_dir_phys(dd)->dd_reserved -
1883 dsl_dir_phys(dd)->dd_used_bytes;
1884
1885 dsl_dir_diduse_space(dd->dd_parent, DD_USED_CHILD_RSRV,
1886 -unused_rsrv, 0, 0, tx);
1887 dsl_dir_diduse_space(newparent, DD_USED_CHILD_RSRV,
1888 unused_rsrv, 0, 0, tx);
1889 }
1890 }
1891
1892 dmu_buf_will_dirty(dd->dd_dbuf, tx);
1893
1894 /* remove from old parent zapobj */
1895 error = zap_remove(mos,
1896 dsl_dir_phys(dd->dd_parent)->dd_child_dir_zapobj,
1897 dd->dd_myname, tx);
1898 ASSERT0(error);
1899
1900 (void) strcpy(dd->dd_myname, mynewname);
1901 dsl_dir_rele(dd->dd_parent, dd);
1902 dsl_dir_phys(dd)->dd_parent_obj = newparent->dd_object;
1903 VERIFY0(dsl_dir_hold_obj(dp,
1904 newparent->dd_object, NULL, dd, &dd->dd_parent));
1905
1906 /* add to new parent zapobj */
1907 VERIFY0(zap_add(mos, dsl_dir_phys(newparent)->dd_child_dir_zapobj,
1908 dd->dd_myname, 8, 1, &dd->dd_object, tx));
1909
1910 #ifdef __FreeBSD__
1911 #ifdef _KERNEL
1912 zfsvfs_update_fromname(ddra->ddra_oldname, ddra->ddra_newname);
1913 zvol_rename_minors(ddra->ddra_oldname, ddra->ddra_newname);
1914 #endif
1915 #endif
1916
1917 dsl_prop_notify_all(dd);
1918
1919 dsl_dir_rele(newparent, FTAG);
1920 dsl_dir_rele(dd, FTAG);
1921 }
1922
1923 int
dsl_dir_rename(const char * oldname,const char * newname)1924 dsl_dir_rename(const char *oldname, const char *newname)
1925 {
1926 dsl_dir_rename_arg_t ddra;
1927
1928 ddra.ddra_oldname = oldname;
1929 ddra.ddra_newname = newname;
1930 ddra.ddra_cred = CRED();
1931
1932 return (dsl_sync_task(oldname,
1933 dsl_dir_rename_check, dsl_dir_rename_sync, &ddra,
1934 3, ZFS_SPACE_CHECK_RESERVED));
1935 }
1936
1937 int
dsl_dir_transfer_possible(dsl_dir_t * sdd,dsl_dir_t * tdd,uint64_t fs_cnt,uint64_t ss_cnt,uint64_t space,cred_t * cr)1938 dsl_dir_transfer_possible(dsl_dir_t *sdd, dsl_dir_t *tdd,
1939 uint64_t fs_cnt, uint64_t ss_cnt, uint64_t space, cred_t *cr)
1940 {
1941 dsl_dir_t *ancestor;
1942 int64_t adelta;
1943 uint64_t avail;
1944 int err;
1945
1946 ancestor = closest_common_ancestor(sdd, tdd);
1947 adelta = would_change(sdd, -space, ancestor);
1948 avail = dsl_dir_space_available(tdd, ancestor, adelta, FALSE);
1949 if (avail < space)
1950 return (SET_ERROR(ENOSPC));
1951
1952 err = dsl_fs_ss_limit_check(tdd, fs_cnt, ZFS_PROP_FILESYSTEM_LIMIT,
1953 ancestor, cr);
1954 if (err != 0)
1955 return (err);
1956 err = dsl_fs_ss_limit_check(tdd, ss_cnt, ZFS_PROP_SNAPSHOT_LIMIT,
1957 ancestor, cr);
1958 if (err != 0)
1959 return (err);
1960
1961 return (0);
1962 }
1963
1964 timestruc_t
dsl_dir_snap_cmtime(dsl_dir_t * dd)1965 dsl_dir_snap_cmtime(dsl_dir_t *dd)
1966 {
1967 timestruc_t t;
1968
1969 mutex_enter(&dd->dd_lock);
1970 t = dd->dd_snap_cmtime;
1971 mutex_exit(&dd->dd_lock);
1972
1973 return (t);
1974 }
1975
1976 void
dsl_dir_snap_cmtime_update(dsl_dir_t * dd)1977 dsl_dir_snap_cmtime_update(dsl_dir_t *dd)
1978 {
1979 timestruc_t t;
1980
1981 gethrestime(&t);
1982 mutex_enter(&dd->dd_lock);
1983 dd->dd_snap_cmtime = t;
1984 mutex_exit(&dd->dd_lock);
1985 }
1986
1987 void
dsl_dir_zapify(dsl_dir_t * dd,dmu_tx_t * tx)1988 dsl_dir_zapify(dsl_dir_t *dd, dmu_tx_t *tx)
1989 {
1990 objset_t *mos = dd->dd_pool->dp_meta_objset;
1991 dmu_object_zapify(mos, dd->dd_object, DMU_OT_DSL_DIR, tx);
1992 }
1993
1994 boolean_t
dsl_dir_is_zapified(dsl_dir_t * dd)1995 dsl_dir_is_zapified(dsl_dir_t *dd)
1996 {
1997 dmu_object_info_t doi;
1998
1999 dmu_object_info_from_db(dd->dd_dbuf, &doi);
2000 return (doi.doi_type == DMU_OTN_ZAP_METADATA);
2001 }
2002