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
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
|
#! /usr/bin/perl
=head1 NAME
s3_user_quota.pl - Script to test the rgw user quota functionality using s3 interface.
=head1 SYNOPSIS
Use:
perl s3_user_quota.pl [--help]
Examples:
perl s3_user_quota.pl
or
perl s3_user_quota.pl --help
=head1 DESCRIPTION
This script intends to test the rgw user quota funcionality using s3 interface
and reports the test results
=head1 ARGUMENTS
s3_user_quota.pl takes the following arguments:
--help
(optional) Displays the usage message.
=cut
use Amazon::S3;
use Data::Dumper;
use IO::File;
use Getopt::Long;
use Digest::MD5;
use Pod::Usage();
use FindBin;
use lib $FindBin::Bin;
use s3_utilities;
use Net::Domain qw(hostfqdn);
my $help;
Getopt::Long::GetOptions(
'help' => \$help
);
Pod::Usage::pod2usage(-verbose => 1) && exit if ($help);
#== local variables ===
our $mytestfilename;
my $mytestfilename1;
my $logmsg;
my $kruft;
my $s3;
my $hostdom = $ENV{RGW_FQDN}||hostfqdn();
my $port = $ENV{RGW_PORT}||7280;
our $hostname = "$hostdom:$port";
our $testfileloc;
our $cnt;
sub quota_set_max_size_per_user {
my ($maxsize, $size1,$rgw_user) = @_;
run_s3($rgw_user);
my $set_quota = `$radosgw_admin quota set --uid=$rgw_user --quota-scope=user --max-size=$maxsize`;
if (($set_quota !~ /./)&&($maxsize == 0)){
my $ret = test_max_objs($size1, $rgw_user);
if ($ret == 1){
pass("quota set for user: $rgw_user with max_size=$maxsize passed" );
}else {
fail("quota set for user: $rgw_user with max_size=$maxsize failed" );
}
} elsif (($set_quota !~ /./) && ($maxsize != 0)) {
my $ret = test_max_objs($size1, $rgw_user);
if ($ret == 0){
pass("quota set for user: $rgw_user with max_size=$maxsize passed" );
}else {
fail("quota set for user: $rgw_user with max_size=$maxsize failed" );
}
}
delete_keys($mytestfilename);
purge_data($rgw_user);
return 0;
}
sub max_size_per_user {
my ($maxsize, $size1,$rgw_user) = @_;
run_s3($rgw_user);
my $set_quota = `$radosgw_admin quota set --uid=$rgw_user --quota-scope=user --max-size=$maxsize`;
if (($set_quota !~ /./) && ($maxsize != 0)) {
my $ret = test_max_objs($size1, $rgw_user);
if ($ret == 0){
$cnt++;
}
}
return $cnt;
}
sub quota_set_max_obj_per_user {
# set max objects
my ($maxobjs, $size1, $rgw_user) = @_;
run_s3($rgw_user);
my $set_quota = `$radosgw_admin quota set --uid=$rgw_user --quota-scope=user --max-objects=$maxobjs`;
if (($set_quota !~ /./) && ($maxobjs == 0)){
my $ret = test_max_objs($size1, $rgw_user);
if ($ret == 1){
pass("quota set for user: $rgw_user with max_objects=$maxobjs passed" );
}else {
fail("quota set for user: $rgw_user with max_objects=$maxobjs failed" );
}
} elsif (($set_quota !~ /./) && ($maxobjs == 1)) {
my $ret = test_max_objs($size1, $rgw_user);
if ($ret == 0){
pass("quota set for user: $rgw_user with max_objects=$maxobjs passed" );
}else {
fail("quota set for user: $rgw_user with max_objects=$maxobjs failed" );
}
}
delete_keys($mytestfilename);
purge_data($rgw_user);
}
sub quota_enable_user {
my ($rgw_user) = @_;
my $en_quota = `$radosgw_admin quota enable --uid=$rgw_user --quota-scope=user`;
if ($en_quota !~ /./){
print "quota enabled for the user $rgw_user \n";
} else {
print "quota enable failed for the user $rgw_user \n";
exit 1;
}
return 0;
}
sub quota_disable_user {
my $dis_quota = `$radosgw_admin quota disable --uid=$rgw_user --quota-scope=user`;
if ($dis_quota !~ /./){
print "quota disabled for the user $rgw_user \n";
} else {
print "quota disable failed for the user $rgw_user \n";
exit 1;
}
return 0;
}
# upload a file to the bucket
sub upload_file {
print "adding file to bucket $bucketname: $mytestfilename\n";
($bucket->add_key_filename( $mytestfilename, $testfileloc,
{ content_type => 'text/plain', },
) and (print "upload file successful\n" ) and return 0 ) or (return 1);
}
# delete the bucket
sub delete_bucket {
($bucket->delete_bucket) and (print "bucket delete succeeded \n") or die $s3->err . "delete bucket failed\n" . $s3->errstr;
}
#Function to upload the given file size to bucket and verify
sub test_max_objs {
my ($size, $rgw_user) = @_;
create_file($size);
quota_enable_user($rgw_user);
my $ret_value = upload_file();
return $ret_value;
}
# set user quota and ensure it is validated
sub test_user_quota_max_size{
my ($max_buckets,$size, $fsize) = @_;
my $usr = rand();
foreach my $i (1..$max_buckets){
my $ret_value = max_size_per_user($size, $fsize, $usr );
}
if ($ret_value == $max_buckets){
fail( "user quota max size for $usr failed on $max_buckets buckets" );
} else {
pass( "user quota max size for $usr passed on $max_buckets buckets" );
}
delete_keys($mytestfilename);
purge_data($usr);
}
#== Main starts here===
ceph_os_info();
check();
quota_set_max_obj_per_user('0', '10Mb', 'usr1');
quota_set_max_obj_per_user('1', '10Mb', 'usr2');
quota_set_max_size_per_user(0, '10Mb', 'usr1');
quota_set_max_size_per_user(1048576000, '1Gb', 'usr2');
test_user_quota_max_size(3,1048576000,'100Mb');
test_user_quota_max_size(2,1048576000, '1Gb');
print "OK";
|
