1 // SPDX-License-Identifier: GPL-2.0
3 * quota.c - CephFS quota
5 * Copyright (C) 2017-2018 SUSE
7 * This program is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU General Public License
9 * as published by the Free Software Foundation; either version 2
10 * of the License, or (at your option) any later version.
12 * This program is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
15 * GNU General Public License for more details.
17 * You should have received a copy of the GNU General Public License
18 * along with this program; if not, see <http://www.gnu.org/licenses/>.
22 #include "mds_client.h"
24 static inline bool ceph_has_quota(struct ceph_inode_info
*ci
)
26 return (ci
&& (ci
->i_max_files
|| ci
->i_max_bytes
));
29 void ceph_handle_quota(struct ceph_mds_client
*mdsc
,
30 struct ceph_mds_session
*session
,
33 struct super_block
*sb
= mdsc
->fsc
->sb
;
34 struct ceph_mds_quota
*h
= msg
->front
.iov_base
;
35 struct ceph_vino vino
;
37 struct ceph_inode_info
*ci
;
39 if (msg
->front
.iov_len
!= sizeof(*h
)) {
40 pr_err("%s corrupt message mds%d len %d\n", __func__
,
41 session
->s_mds
, (int)msg
->front
.iov_len
);
46 /* increment msg sequence number */
47 mutex_lock(&session
->s_mutex
);
49 mutex_unlock(&session
->s_mutex
);
52 vino
.ino
= le64_to_cpu(h
->ino
);
53 vino
.snap
= CEPH_NOSNAP
;
54 inode
= ceph_find_inode(sb
, vino
);
56 pr_warn("Failed to find inode %llu\n", vino
.ino
);
59 ci
= ceph_inode(inode
);
61 spin_lock(&ci
->i_ceph_lock
);
62 ci
->i_rbytes
= le64_to_cpu(h
->rbytes
);
63 ci
->i_rfiles
= le64_to_cpu(h
->rfiles
);
64 ci
->i_rsubdirs
= le64_to_cpu(h
->rsubdirs
);
65 ci
->i_max_bytes
= le64_to_cpu(h
->max_bytes
);
66 ci
->i_max_files
= le64_to_cpu(h
->max_files
);
67 spin_unlock(&ci
->i_ceph_lock
);
73 * This function walks through the snaprealm for an inode and returns the
74 * ceph_snap_realm for the first snaprealm that has quotas set (either max_files
75 * or max_bytes). If the root is reached, return the root ceph_snap_realm
78 * Note that the caller is responsible for calling ceph_put_snap_realm() on the
81 static struct ceph_snap_realm
*get_quota_realm(struct ceph_mds_client
*mdsc
,
84 struct ceph_inode_info
*ci
= NULL
;
85 struct ceph_snap_realm
*realm
, *next
;
86 struct ceph_vino vino
;
90 realm
= ceph_inode(inode
)->i_snap_realm
;
91 ceph_get_snap_realm(mdsc
, realm
);
93 vino
.ino
= realm
->ino
;
94 vino
.snap
= CEPH_NOSNAP
;
95 in
= ceph_find_inode(inode
->i_sb
, vino
);
97 pr_warn("Failed to find inode for %llu\n", vino
.ino
);
101 has_quota
= ceph_has_quota(ci
);
104 next
= realm
->parent
;
105 if (has_quota
|| !next
)
108 ceph_get_snap_realm(mdsc
, next
);
109 ceph_put_snap_realm(mdsc
, realm
);
113 ceph_put_snap_realm(mdsc
, realm
);
118 bool ceph_quota_is_same_realm(struct inode
*old
, struct inode
*new)
120 struct ceph_mds_client
*mdsc
= ceph_inode_to_client(old
)->mdsc
;
121 struct ceph_snap_realm
*old_realm
, *new_realm
;
124 down_read(&mdsc
->snap_rwsem
);
125 old_realm
= get_quota_realm(mdsc
, old
);
126 new_realm
= get_quota_realm(mdsc
, new);
127 is_same
= (old_realm
== new_realm
);
128 up_read(&mdsc
->snap_rwsem
);
131 ceph_put_snap_realm(mdsc
, old_realm
);
133 ceph_put_snap_realm(mdsc
, new_realm
);
138 enum quota_check_op
{
139 QUOTA_CHECK_MAX_FILES_OP
, /* check quota max_files limit */
140 QUOTA_CHECK_MAX_BYTES_OP
, /* check quota max_files limit */
141 QUOTA_CHECK_MAX_BYTES_APPROACHING_OP
/* check if quota max_files
142 limit is approaching */
146 * check_quota_exceeded() will walk up the snaprealm hierarchy and, for each
147 * realm, it will execute quota check operation defined by the 'op' parameter.
148 * The snaprealm walk is interrupted if the quota check detects that the quota
149 * is exceeded or if the root inode is reached.
151 static bool check_quota_exceeded(struct inode
*inode
, enum quota_check_op op
,
154 struct ceph_mds_client
*mdsc
= ceph_inode_to_client(inode
)->mdsc
;
155 struct ceph_inode_info
*ci
;
156 struct ceph_snap_realm
*realm
, *next
;
157 struct ceph_vino vino
;
160 bool exceeded
= false;
162 down_read(&mdsc
->snap_rwsem
);
163 realm
= ceph_inode(inode
)->i_snap_realm
;
164 ceph_get_snap_realm(mdsc
, realm
);
166 vino
.ino
= realm
->ino
;
167 vino
.snap
= CEPH_NOSNAP
;
168 in
= ceph_find_inode(inode
->i_sb
, vino
);
170 pr_warn("Failed to find inode for %llu\n", vino
.ino
);
174 spin_lock(&ci
->i_ceph_lock
);
175 if (op
== QUOTA_CHECK_MAX_FILES_OP
) {
176 max
= ci
->i_max_files
;
177 rvalue
= ci
->i_rfiles
+ ci
->i_rsubdirs
;
179 max
= ci
->i_max_bytes
;
180 rvalue
= ci
->i_rbytes
;
182 spin_unlock(&ci
->i_ceph_lock
);
184 case QUOTA_CHECK_MAX_FILES_OP
:
185 exceeded
= (max
&& (rvalue
>= max
));
187 case QUOTA_CHECK_MAX_BYTES_OP
:
188 exceeded
= (max
&& (rvalue
+ delta
> max
));
190 case QUOTA_CHECK_MAX_BYTES_APPROACHING_OP
:
196 * when we're writing more that 1/16th
197 * of the available space
200 (((max
- rvalue
) >> 4) < delta
);
205 /* Shouldn't happen */
206 pr_warn("Invalid quota check op (%d)\n", op
);
207 exceeded
= true; /* Just break the loop */
211 next
= realm
->parent
;
212 if (exceeded
|| !next
)
214 ceph_get_snap_realm(mdsc
, next
);
215 ceph_put_snap_realm(mdsc
, realm
);
218 ceph_put_snap_realm(mdsc
, realm
);
219 up_read(&mdsc
->snap_rwsem
);
225 * ceph_quota_is_max_files_exceeded - check if we can create a new file
226 * @inode: directory where a new file is being created
228 * This functions returns true is max_files quota allows a new file to be
229 * created. It is necessary to walk through the snaprealm hierarchy (until the
230 * FS root) to check all realms with quotas set.
232 bool ceph_quota_is_max_files_exceeded(struct inode
*inode
)
234 WARN_ON(!S_ISDIR(inode
->i_mode
));
236 return check_quota_exceeded(inode
, QUOTA_CHECK_MAX_FILES_OP
, 0);
240 * ceph_quota_is_max_bytes_exceeded - check if we can write to a file
241 * @inode: inode being written
242 * @newsize: new size if write succeeds
244 * This functions returns true is max_bytes quota allows a file size to reach
245 * @newsize; it returns false otherwise.
247 bool ceph_quota_is_max_bytes_exceeded(struct inode
*inode
, loff_t newsize
)
249 loff_t size
= i_size_read(inode
);
251 /* return immediately if we're decreasing file size */
255 return check_quota_exceeded(inode
, QUOTA_CHECK_MAX_BYTES_OP
, (newsize
- size
));
259 * ceph_quota_is_max_bytes_approaching - check if we're reaching max_bytes
260 * @inode: inode being written
261 * @newsize: new size if write succeeds
263 * This function returns true if the new file size @newsize will be consuming
264 * more than 1/16th of the available quota space; it returns false otherwise.
266 bool ceph_quota_is_max_bytes_approaching(struct inode
*inode
, loff_t newsize
)
268 loff_t size
= ceph_inode(inode
)->i_reported_size
;
270 /* return immediately if we're decreasing file size */
274 return check_quota_exceeded(inode
, QUOTA_CHECK_MAX_BYTES_APPROACHING_OP
,