xref: /freebsd/sys/contrib/openzfs/tests/zfs-tests/tests/functional/userquota/userquota_common.kshlib (revision 271171e0d97b88ba2a7c3bf750c9672b484c1c13)
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 https://opensource.org/licenses/CDDL-1.0.
10# See the License for the specific language governing permissions
11# and limitations under the License.
12#
13# When distributing Covered Code, include this CDDL HEADER in each
14# file and include the License file at usr/src/OPENSOLARIS.LICENSE.
15# If applicable, add the following below this CDDL HEADER, with the
16# fields enclosed by brackets "[]" replaced with your own identifying
17# information: Portions Copyright [yyyy] [name of copyright owner]
18#
19# CDDL HEADER END
20#
21
22#
23# Copyright 2009 Sun Microsystems, Inc.  All rights reserved.
24# Use is subject to license terms.
25#
26
27#
28# Copyright (c) 2013, 2016 by Delphix. All rights reserved.
29#
30
31. $STF_SUITE/include/libtest.shlib
32. $STF_SUITE/tests/functional/userquota/userquota.cfg
33
34#
35# reset the userquota and groupquota and delete temporary files
36#
37function cleanup_quota
38{
39	if datasetexists $QFS; then
40		log_must zfs set userquota@$QUSER1=none $QFS
41		log_must zfs set userobjquota@$QUSER1=none $QFS
42		log_must zfs set userquota@$QUSER2=none $QFS
43		log_must zfs set userobjquota@$QUSER2=none $QFS
44		log_must zfs set groupquota@$QGROUP=none $QFS
45		log_must zfs set groupobjquota@$QGROUP=none $QFS
46		recovery_writable $QFS
47	fi
48
49	[[ -f $QFILE ]] && log_must rm -f $QFILE
50	[[ -f $OFILE ]] && log_must rm -f $OFILE
51	sync_all_pools
52
53	return 0
54}
55
56#
57# delete user and group that created during the test
58#
59function clean_user_group
60{
61	for usr in $QUSER1 $QUSER2; do
62		log_must del_user $usr
63	done
64
65	log_must del_group $QGROUP
66
67	return 0
68}
69
70#
71#  make the $QFS's mountpoint writable for all users
72#
73function mkmount_writable
74{
75	typeset fs=$1
76	typeset mntp=$(get_prop mountpoint $fs)
77	log_must chmod 0777 $mntp
78}
79
80#
81# recovery the directory permission for $QFS
82#
83function recovery_writable
84{
85	typeset fs=$1
86	typeset mntp=$(get_prop mountpoint $fs)
87	log_must chmod 0755 $mntp
88}
89
90#
91# check the quota value of a specific FS
92#
93function check_quota
94{
95	typeset fs=$2
96	typeset prop=$1
97	typeset expected=$3
98	typeset value=$(get_prop $prop $fs)
99
100	if (($value != $expected)); then
101		return 1
102	fi
103}
104