1#!/bin/ksh -p
2#
3# CDDL HEADER START
4#
5# The contents of this file are subject to the terms of the
6# Common Development and Distribution License (the "License").
7# You may not use this file except in compliance with the License.
8#
9# You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
10# or http://www.opensolaris.org/os/licensing.
11# See the License for the specific language governing permissions
12# and limitations under the License.
13#
14# When distributing Covered Code, include this CDDL HEADER in each
15# file and include the License file at usr/src/OPENSOLARIS.LICENSE.
16# If applicable, add the following below this CDDL HEADER, with the
17# fields enclosed by brackets "[]" replaced with your own identifying
18# information: Portions Copyright [yyyy] [name of copyright owner]
19#
20# CDDL HEADER END
21#
22
23#
24# Copyright 2009 Sun Microsystems, Inc.  All rights reserved.
25# Use is subject to license terms.
26#
27
28#
29# Copyright (c) 2016 by Jinshan Xiong. All rights reserved.
30#
31
32. $STF_SUITE/include/libtest.shlib
33. $STF_SUITE/include/math.shlib
34. $STF_SUITE/tests/functional/userquota/userquota_common.kshlib
35
36#
37# DESCRIPTION:
38#       Check the user used object accounting in zfs userspace
39#
40#
41# STRATEGY:
42#       1. create a bunch of files by specific users
43#	2. use zfs userspace to check the used objects
44#	3. change the owner of test files and verify object count
45#	4. delete files and verify object count
46#
47
48function cleanup
49{
50	if datasetexists $snapfs; then
51		log_must zfs destroy $snapfs
52	fi
53
54	log_must rm -f ${QFILE}_*
55	log_must cleanup_quota
56}
57
58function user_object_count
59{
60	typeset fs=$1
61	typeset user=$2
62	typeset -i userspacecnt=$(zfs userspace -oname,objused $fs |
63	    awk /$user/'{print $2}')
64	typeset -i zfsgetcnt=$(zfs get -H -ovalue userobjused@$user $fs)
65
66	# 'zfs userspace' and 'zfs get userobjused@' should be equal
67	verify_eq "$userspacecnt" "$zfsgetcnt" "userobjused@$user"
68
69	echo $userspacecnt
70}
71
72log_onexit cleanup
73
74log_assert "Check the zfs userspace object used"
75
76mkmount_writable $QFS
77log_must zfs set xattr=sa $QFS
78
79((user1_cnt = RANDOM % 100 + 1))
80((user2_cnt = RANDOM % 100 + 1))
81
82log_must user_run $QUSER1 mkfiles ${QFILE}_1 $user1_cnt
83log_must user_run $QUSER2 mkfiles ${QFILE}_2 $user2_cnt
84sync_pool
85
86typeset snapfs=$QFS@snap
87
88log_must zfs snapshot $snapfs
89
90log_must eval "zfs userspace $QFS >/dev/null 2>&1"
91log_must eval "zfs userspace $snapfs >/dev/null 2>&1"
92
93for fs in "$QFS" "$snapfs"; do
94	log_note "check the user used objects in zfs userspace $fs"
95	[[ $(user_object_count $fs $QUSER1) -eq $user1_cnt ]] ||
96		log_fail "expected $user1_cnt"
97	[[ $(user_object_count $fs $QUSER2) -eq $user2_cnt ]] ||
98		log_fail "expected $user2_cnt"
99done
100
101log_note "change the owner of files"
102log_must chown $QUSER2 ${QFILE}_1*
103sync_pool
104
105[[ $(user_object_count $QFS $QUSER1) -eq 0 ]] ||
106	log_fail "expected 0 files for $QUSER1"
107
108[[ $(user_object_count $snapfs $QUSER1) -eq $user1_cnt ]] ||
109	log_fail "expected $user_cnt files for $QUSER1 in snapfs"
110
111[[ $(user_object_count $QFS $QUSER2) -eq $((user1_cnt+user2_cnt)) ]] ||
112	log_fail "expected $((user1_cnt+user2_cnt)) files for $QUSER2"
113
114log_note "file removal"
115log_must rm ${QFILE}_*
116sync_pool
117
118[[ $(user_object_count $QFS $QUSER2) -eq 0 ]] ||
119        log_fail "expected 0 files for $QUSER2"
120
121cleanup
122log_pass "Check the zfs userspace object used"
123