From e6918187568dbd01842d8d1d2c808ce16a894239 Mon Sep 17 00:00:00 2001 From: Daniel Baumann Date: Sun, 21 Apr 2024 13:54:28 +0200 Subject: Adding upstream version 18.2.2. Signed-off-by: Daniel Baumann --- src/test/libcephfs/quota.cc | 167 ++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 167 insertions(+) create mode 100644 src/test/libcephfs/quota.cc (limited to 'src/test/libcephfs/quota.cc') diff --git a/src/test/libcephfs/quota.cc b/src/test/libcephfs/quota.cc new file mode 100644 index 000000000..00afb3d09 --- /dev/null +++ b/src/test/libcephfs/quota.cc @@ -0,0 +1,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 +#include +#include +#include +#include +#include +#include + +#ifdef __linux__ +#include +#include +#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); +} -- cgit v1.2.3