?
Current Path : /scripts/ |
Linux gator3171.hostgator.com 4.19.286-203.ELK.el7.x86_64 #1 SMP Wed Jun 14 04:33:55 CDT 2023 x86_64 |
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; ## TODO @GOLIVE: the cpanelroundcube user is going away if ( $user =~ /^cpanel(horde|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; }