\!/ KyuuKazami \!/

#!/usr/local/cpanel/3rdparty/bin/perl
# cpanel - quota_auto_fix                            Copyright 2010 cPanel, Inc.
#                                                           All rights Reserved.
# copyright@cpanel.net                                         http://cpanel.net
# This code is subject to the cpanel license. Unauthorized copying is prohibited
#
# This script auto repairs quotas when they become severely corrupted.

use strict;
use Cpanel::FindBin ();

my $repquota_cmd = Cpanel::FindBin::findbin('repquota');
if ( !-x $repquota_cmd ) {
    print "Unable to locate repquota command.\n";
    exit 1;
}

my $broken = 0;
my $signal = 0;
my $zpid   = open READCHLD, '-|';
if ($zpid) {
    while (<READCHLD>) {
        if (/^quota_auto_fix signal: (\d+)/) {
            $signal = $1;
        }
        elsif (/has wrong format/i) {
            print;
            $broken = 1;
        }
        elsif (/Quota file not found/i) {
            print;
            $broken = 1;
        }
    }
}
else {
    open STDERR, ">&STDOUT";
    system $repquota_cmd, '-a';
    my $rsignal = ( $? & 127 );
    print "quota_auto_fix signal: $rsignal\n";
    exit 0;
}

if ($broken) {
    system '/usr/local/cpanel/scripts/fixquotas';
}
elsif ( $signal == 11 ) {
    print "repquota exited with signal 11 ... rebuilding quota databases.\n";
    system '/usr/local/cpanel/scripts/fixquotas';
}
else {
    print "quota test: check passed\n";
}