Current Path : /scripts/ |
|
Current File : //scripts/editquota |
#!/usr/local/cpanel/3rdparty/bin/perl
# cpanel - editquota Copyright 2018 cPanel, Inc.
# All rights Reserved.
# copyright@cpanel.net http://cpanel.net
# This code is subject to the cPanel license. Unauthorized copying is prohibited
use strict;
use warnings;
use Cpanel::CachedCommand::Utils ();
use Cpanel::FindBin ();
use Cpanel::Quota::Blocks ();
use Cpanel::Quota::OverCache ();
use Cpanel::Locale 'lh';
$| = 1;
if ( !@ARGV ) {
print "Usage: editquota <user> <?M>\n";
exit 1;
}
my $user = $ARGV[0];
my $quota = $ARGV[1];
my $quotaoff = $ARGV[2] || '';
if ( !defined $user || ( ( $user =~ m/^\d+$/ && !getpwuid($user) ) && !getpwnam($user) ) ) {
print "Invalid user\n";
print "Usage: editquota <user> <?M>\n";
exit 1;
}
if ( !defined $quota || ( $quota !~ m/^\d+(?:\.\d+)?M$/i && $quota !~ m/unlimited/i ) ) {
print "Invalid quota\n";
print "Usage: editquota <user> <?M>\n";
exit 1;
}
my %cmd = (
'quotaon' => undef,
'quotaoff' => undef,
'edquota' => undef,
);
my @missing_cmds;
foreach my $cmd_name ( keys %cmd ) {
$cmd{$cmd_name} = Cpanel::FindBin::findbin($cmd_name);
if ( !-x $cmd{$cmd_name} ) {
push @missing_cmds, $cmd_name;
}
}
if ( scalar @missing_cmds ) {
print "Incomplete quota kit: unable to edit quotas.\n";
print 'Missing commands: ', join( ', ', @missing_cmds ), "\n";
exit 1;
}
if ( $user =~ m/^\d+$/ ) {
$user = ( getpwuid($user) )[0];
}
if ( $user =~ /^cpanel(horde|phpmyadmin|phppgadmin|roundcube)$/ ) {
$quota = 0;
}
if ( $quota !~ m/unlimited/i ) {
$quota =~ s/M$//i;
# Round up or down to the nearest whole MB. This ensures that what gets saved in /etc/quota.conf
# follows the expected integer-only format, while staying as close as possible to what was
# entered by the user in the case of a fractional MB value.
$quota = sprintf( '%.0f', $quota );
}
else {
$quota = 0;
}
########################################################################
my $quota_in_blocks = $quota * 1024;
my $cpanel_quota_blocks;
eval {
$cpanel_quota_blocks = Cpanel::Quota::Blocks->new();
$cpanel_quota_blocks->set_user($user)->set_limits( { soft => $quota_in_blocks, hard => $quota_in_blocks } );
};
if ( my $exception = $@ ) {
if ( eval { $exception->isa('Cpanel::Exception::Quota::NotEnabled') } ) {
print lh()->maketext( "Filesystem quotas are not enabled. The “[_1]” user does not require any changes.", $user ) . "\n";
exit;
}
else {
die $exception;
}
}
# This script used to say "Using Universal Quota Support (quota=.......)" because it
# was using /scripts/pedquota to make the change.
print lh()->maketext( "The quota for the “[_1]” user is now [_2] MB ([_3] blocks).", $user, $quota, $quota_in_blocks ) . "\n";
########################################################################
# Invalidate the repquota.datastore file but leave it in case something
# does not care about the valitity period (ie account transfer getacctlist)
if ( -e '/var/cpanel/repquota.cache' ) {
utime 0, 0, '/var/cpanel/repquota.cache';
}
if ( -e '/var/cpanel/repquota.datastore' ) {
utime 0, 0, '/var/cpanel/repquota.datastore';
}
# Update quota file mtimes
$cpanel_quota_blocks->update_mtimes_to_clear_cache();
# Remove cached quota command result for user
my $cached_command = Cpanel::CachedCommand::Utils::_get_datastore_filename( '/usr/local/cpanel/bin/quota', $user );
unlink $cached_command if -e $cached_command;
Cpanel::Quota::OverCache::unset_user_at_blocks_quota($user);
Copyright 2K16 - 2K18 Indonesian Hacker Rulez