1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
|
// -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*-
// vim: ts=8 sw=2 smarttab
/*
* Ceph - scalable distributed file system
*
* Copyright (C) 2011 New Dream Network
* Copyright (C) 2022 Red Hat, Inc.
*
* This is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License version 2.1, as published by the Free Software
* Foundation. See file COPYING.
*
*/
#include "include/compat.h"
#include "gtest/gtest.h"
#include "include/cephfs/libcephfs.h"
#include "mds/mdstypes.h"
#include "include/stat.h"
#include <errno.h>
#include <fcntl.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <sys/time.h>
#include <sys/resource.h>
#ifdef __linux__
#include <limits.h>
#include <sys/xattr.h>
#endif
TEST(LibCephFS, SnapQuota) {
struct ceph_mount_info *cmount;
ASSERT_EQ(ceph_create(&cmount, NULL), 0);
ASSERT_EQ(ceph_conf_read_file(cmount, NULL), 0);
ASSERT_EQ(0, ceph_conf_parse_env(cmount, NULL));
ASSERT_EQ(ceph_mount(cmount, NULL), 0);
char test_snap_dir_quota_xattr[256];
char test_snap_subdir_quota_xattr[256];
char test_snap_subdir_noquota_xattr[256];
char xattrk[128];
char xattrv[128];
char c_temp[PATH_MAX];
char gxattrv[128];
int xbuflen = sizeof(gxattrv);
pid_t mypid = getpid();
// create dir and set quota
sprintf(test_snap_dir_quota_xattr, "test_snap_dir_quota_xattr_%d", mypid);
ASSERT_EQ(0, ceph_mkdir(cmount, test_snap_dir_quota_xattr, 0777));
sprintf(xattrk, "ceph.quota.max_bytes");
sprintf(xattrv, "65536");
ASSERT_EQ(0, ceph_setxattr(cmount, test_snap_dir_quota_xattr, xattrk, (void *)xattrv, 5, XATTR_CREATE));
// create subdir and set quota
sprintf(test_snap_subdir_quota_xattr, "test_snap_dir_quota_xattr_%d/subdir_quota", mypid);
ASSERT_EQ(0, ceph_mkdirs(cmount, test_snap_subdir_quota_xattr, 0777));
sprintf(xattrk, "ceph.quota.max_bytes");
sprintf(xattrv, "32768");
ASSERT_EQ(0, ceph_setxattr(cmount, test_snap_subdir_quota_xattr, xattrk, (void *)xattrv, 5, XATTR_CREATE));
// create subdir with no quota
sprintf(test_snap_subdir_noquota_xattr, "test_snap_dir_quota_xattr_%d/subdir_noquota", mypid);
ASSERT_EQ(0, ceph_mkdirs(cmount, test_snap_subdir_noquota_xattr, 0777));
// snapshot dir
sprintf(c_temp, "/.snap/test_snap_dir_quota_xattr_snap_%d", mypid);
ASSERT_EQ(0, ceph_mkdirs(cmount, c_temp, 0777));
// check dir quota under snap
sprintf(c_temp, "/.snap/test_snap_dir_quota_xattr_snap_%d/test_snap_dir_quota_xattr_%d", mypid, mypid);
int alen = ceph_getxattr(cmount, c_temp, "ceph.quota.max_bytes", (void *)gxattrv, xbuflen);
ASSERT_LT(0, alen);
ASSERT_LT(alen, xbuflen);
gxattrv[alen] = '\0';
ASSERT_STREQ(gxattrv, "65536");
// check subdir quota under snap
sprintf(c_temp, "/.snap/test_snap_dir_quota_xattr_snap_%d/test_snap_dir_quota_xattr_%d/subdir_quota", mypid, mypid);
alen = ceph_getxattr(cmount, c_temp, "ceph.quota.max_bytes", (void *)gxattrv, xbuflen);
ASSERT_LT(0, alen);
ASSERT_LT(alen, xbuflen);
gxattrv[alen] = '\0';
ASSERT_STREQ(gxattrv, "32768");
// ensure subdir noquota xattr under snap
sprintf(c_temp, "/.snap/test_snap_dir_quota_xattr_snap_%d/test_snap_dir_quota_xattr_%d/subdir_noquota", mypid, mypid);
EXPECT_EQ(-CEPHFS_ENODATA, ceph_getxattr(cmount, c_temp, "ceph.quota.max_bytes", (void *)gxattrv, xbuflen));
// listxattr() shouldn't return ceph.quota.max_bytes vxattr
sprintf(c_temp, "/.snap/test_snap_dir_quota_xattr_snap_%d/test_snap_dir_quota_xattr_%d", mypid, mypid);
char xattrlist[512];
int len = ceph_listxattr(cmount, c_temp, xattrlist, sizeof(xattrlist));
ASSERT_GE(sizeof(xattrlist), (size_t)len);
char *p = xattrlist;
int found = 0;
while (len > 0) {
if (strcmp(p, "ceph.quota.max_bytes") == 0)
found++;
len -= strlen(p) + 1;
p += strlen(p) + 1;
}
ASSERT_EQ(found, 0);
ceph_shutdown(cmount);
}
void statfs_quota_size_check(struct ceph_mount_info *cmount, const char *path,
int blocks, int bsize)
{
struct statvfs stvfs;
ASSERT_EQ(0, ceph_statfs(cmount, path, &stvfs));
ASSERT_EQ(blocks, stvfs.f_blocks);
ASSERT_EQ(bsize, stvfs.f_bsize);
ASSERT_EQ(bsize, stvfs.f_frsize);
}
TEST(LibCephFS, QuotaRealm) {
struct ceph_mount_info *cmount, *pmount1, *pmount2;
char test_quota_realm_pdir[128];
char test_quota_realm_cdir[256];
char xattrk[32];
char xattrv[16];
ASSERT_EQ(ceph_create(&cmount, NULL), 0);
ASSERT_EQ(ceph_conf_read_file(cmount, NULL), 0);
ASSERT_EQ(0, ceph_conf_parse_env(cmount, NULL));
ASSERT_EQ(ceph_mount(cmount, NULL), 0);
pid_t mypid = getpid();
// create parent directory and set quota size
sprintf(test_quota_realm_pdir, "/test_quota_realm_pdir_%d", mypid);
ASSERT_EQ(0, ceph_mkdir(cmount, test_quota_realm_pdir, 0777));
sprintf(xattrk, "ceph.quota.max_bytes");
sprintf(xattrv, "8388608"); // 8MB
ASSERT_EQ(0, ceph_setxattr(cmount, test_quota_realm_pdir, xattrk, (void *)xattrv, 7, XATTR_CREATE));
// create child directory and set quota file
sprintf(test_quota_realm_cdir, "%s/test_quota_realm_cdir", test_quota_realm_pdir);
ASSERT_EQ(0, ceph_mkdir(cmount, test_quota_realm_cdir, 0777));
sprintf(xattrk, "ceph.quota.max_files");
sprintf(xattrv, "1024"); // 1K files
ASSERT_EQ(0, ceph_setxattr(cmount, test_quota_realm_cdir, xattrk, (void *)xattrv, 4, XATTR_CREATE));
ASSERT_EQ(ceph_create(&pmount1, NULL), 0);
ASSERT_EQ(ceph_conf_read_file(pmount1, NULL), 0);
ASSERT_EQ(0, ceph_conf_parse_env(pmount1, NULL));
ASSERT_EQ(ceph_mount(pmount1, test_quota_realm_pdir), 0);
statfs_quota_size_check(pmount1, "/", 2, 4194304); // 8MB
ASSERT_EQ(ceph_create(&pmount2, NULL), 0);
ASSERT_EQ(ceph_conf_read_file(pmount2, NULL), 0);
ASSERT_EQ(0, ceph_conf_parse_env(pmount2, NULL));
ASSERT_EQ(ceph_mount(pmount2, test_quota_realm_cdir), 0);
statfs_quota_size_check(pmount2, "/", 2, 4194304); // 8MB
ceph_shutdown(pmount1);
ceph_shutdown(pmount2);
ceph_shutdown(cmount);
}
|
