| Linux sh00002.bigrock.com 5.14.0-687.29.1.el9_8.x86_64 #1 SMP PREEMPT_DYNAMIC Thu Jul 23 16:18:48 EDT 2026 x86_64 Path : /scripts/ |
| Current File : //scripts/resetquotas |
#!/usr/local/cpanel/3rdparty/bin/perl
# cpanel - scripts/resetquotas Copyright 2022 cPanel, L.L.C.
# All rights reserved.
# copyright@cpanel.net http://cpanel.net
# This code is subject to the cPanel license. Unauthorized copying is prohibited
package scripts::resetquotas;
use strict;
use warnings;
use Cpanel::Config::LoadCpUserFile ();
use Cpanel::Config::Users ();
use Cpanel::Quota::Common ();
use Cpanel::FindBin ();
use Cpanel::SafeRun::Errors ();
use Cpanel::SysQuota::Cache ();
__PACKAGE__->run() unless caller();
my %cmd;
sub run { ## no critic qw(Subroutines::ProhibitExcessComplexity)
if ( @ARGV && $ARGV[0] ne '--useplanquotas' ) {
print "Note: call resetquota with --useplanquotas to reset all quotas to plan defaults\n";
}
my $quotaconf_ref = {};
my @systemusers = Cpanel::Config::Users::getcpusers();
%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 reset quotas.\n";
print 'Missing commands: ', join( ', ', @missing_cmds ), "\n";
exit 1;
}
Cpanel::SafeRun::Errors::saferunnoerror( $cmd{'quotaoff'}, '-a' );
my %planquota;
foreach my $user ( sort @systemusers ) {
# Safe because it's only loading from cpusers list.
my $cpuser = Cpanel::Config::LoadCpUserFile::loadcpuserfile($user);
next if !$cpuser->{'PLAN'};
my $plan = $cpuser->{'PLAN'};
$plan =~ tr/\///d;
next if ( $plan eq '' || $plan =~ m/^\.+$/ );
my $pkgquota;
if ( exists $planquota{$plan} ) {
$pkgquota = $planquota{$plan};
}
elsif ( -r '/var/cpanel/packages/' . $plan ) {
if ( open my $plan_fh, '<', '/var/cpanel/packages/' . $plan ) {
print "Reading package $plan\n";
while (<$plan_fh>) {
if (/^QUOTA=(\S+)/) {
my $plan_quota = $1;
if ( $plan_quota =~ m/unlimited/i ) {
$pkgquota = $planquota{$plan} = 0;
}
elsif ( $plan_quota =~ m/(\d+)/i ) {
$pkgquota = $planquota{$plan} = $1;
}
}
}
close $plan_fh;
}
# Default quota to 'unlimited'
if ( !defined $pkgquota ) {
$pkgquota = $planquota{$plan} = 0;
}
}
if ( exists $cpuser->{'DISK_BLOCK_LIMIT'} ) {
$quotaconf_ref->{$user} = $cpuser->{'DISK_BLOCK_LIMIT'} ? ( int( $cpuser->{'DISK_BLOCK_LIMIT'} / $Cpanel::Quota::Common::MEGABYTES_TO_BLOCKS ) || 1 ) : 0;
}
if ( ( !exists $quotaconf_ref->{$user} || $quotaconf_ref->{$user} eq '' ) && $pkgquota ) { $quotaconf_ref->{$user} = $pkgquota; }
if ( @ARGV && $ARGV[0] eq '--useplanquotas' && $pkgquota ne '' ) {
if ( $quotaconf_ref->{$user} != $pkgquota ) {
print "Adjusting user $user quota to the package's ($plan) value.\n";
}
$quotaconf_ref->{$user} = $pkgquota;
}
next if ( !exists $quotaconf_ref->{$user} || $quotaconf_ref->{$user} eq '' );
if ( my $pid = fork() ) {
print "Resetting quota for $user to $quotaconf_ref->{$user} M\n";
waitpid( $pid, 0 );
}
else {
open( STDOUT, '>', '/dev/null' ) || die;
editquota( $user, $quotaconf_ref->{$user} );
exit;
}
}
# Update quota file mtimes
Cpanel::Quota::Common->new()->update_mtimes_to_clear_cache();
Cpanel::SafeRun::Errors::saferunnoerror( $cmd{'quotaon'}, '-a' );
Cpanel::SysQuota::Cache::purge_cache();
exit 0;
}
sub editquota {
my $user = shift || return;
my $quot = shift || 0;
if ( $user =~ /^cpanel(phpmyadmin|phppgadmin|roundcube)$/ ) {
$quot = 0;
}
my $megquota = $quot;
$quot = int $quot;
$quot = ( $quot * 1024 );
$quot = sprintf( '%.0f', $quot );
$ENV{'NQUOTA'} = $quot;
$ENV{'VISUAL'} = '/usr/local/cpanel/scripts/pedquota';
$ENV{'EDITOR'} = '/usr/local/cpanel/scripts/pedquota';
system $cmd{'edquota'}, '-u', $user;
return;
}