#!/usr/bin/perl

###############################################################
#This script is used to ensure the permissions of significant
#files are set correctly, and that the correct number of files
#and directories are existent in the proper location.
###############################################################

#Used to detect all directories in a given original directory.
sub getfulldirectorylist {
    my @final = ();
    my $dirtop = @_[0];
    print "Reading data from $dirtop\n";
    my @dirlist = `find $dirtop -type d`;
    my $dirlist = @dirlist;
    for ($i = 0; $i < $dirlist; $i++) {
        my $temp = @dirlist[$i];
        $temp =~ s/( |\$)/\\$1/g;
        if ($temp =~ m/\/store\/ariel/) {
        }
        else {
        @final[$i] = $temp;
        }
    }
    return @final;
}


#Used to detect all files found in a given directory.
sub getdirectorycontents {
    my $dirname = @_[0];
    my @contents = `ls -A $dirname`;
    return @contents;
}

#Used to detect the permissions of a given file.
sub getpermissions {
    my ($filename, $location) = @_;

    my $permission = `cd $location; ls -l | grep $filename | grep nobody`;
    if ($permission) {
            return "nobody";
    }

    my $permission = `cd $location; ls -l | grep $filename | grep root`;
    if ($permission) {
        return "root";
    }
}

#Compares the baseline file with the file being tested.
sub compare {
    my ($baseline, $test) = @_;
    open(FILE, $baseline) or die "Can't open $baseline : $!";
    my @baselinefile = <FILE>;
    close(FILE);
    open(TEST, $test) or die "Can't open $test : $!";
    my @testfile = <TEST>;
    close(TEST);
    my $baselinefile = @baselinefile;
    my $testfile = @testfile;
    print "\nStarting matching process\n";
    my @testresults = ();
    $found = 0;
    $unfound = 0;
    my @unfoundlist = ();
    for ($i = 0; $i < $baselinefile; $i++) {
        if (($i % 10000) == 0) {
            print ".";
        }
        for ($j = 0; $j < $testfile; $j++) {
            if (@baselinefile[$i] eq @testfile[$j]) {
                $found = 1;
                last;
            }
        }
        if ($found == 0) {
            push(@unfoundlist, "@baselinefile[$i]");
            $unfound++;
        }
        $found = 0;
    }
    $result = $baselinefile - $unfound;
    print "\n$result out of $baselinefile files & folders were sucessfully identified\n\n";
    print "Writing unfound results to text file: results.txt\n\n";
    print "Comparison complete.\n";
    
    write_file("results.txt", @unfoundlist);
}

#Used for writing the file to be tested.
sub write_file {
    my ($f, @data ) = @_;
    @data = () unless @data;
    open F, "> $f" or die "Can't open $f : $!";
    print F @data;
    close F;
}

#Will populate a file with a list all the directories and files
sub get_info {
    my $filename = @_[0];
    my @list = ();
    print "\n";
    for my $topdir ("/store", "/opt/qradar") {
        my @dir = getfulldirectorylist("$topdir");
        foreach $dir (@dir) {
            my @content = getdirectorycontents($dir);
            push(@list, "$dir");
            foreach $content (@content) {
                push(@list, "$content");
            }
            push(@list, "\n");
        }
    }
    write_file("$filename", @list);
}

##############################################################
#Usability for simple creation of a baseline file or test file
##############################################################

print "What would you like to do?\n1. Create a baseline - type '1'.\n2. Test a box - type '2'.\n";
$input = <STDIN>;
chomp $input;

if ("$input" eq "1") {
    get_info("baseline.txt");
    print "Baseline written. See baseline.txt for results\n";
}

elsif ("$input" eq "2") {
    get_info("test.txt");
    compare("baseline.txt", "test.txt");
}

#Get info on the file
@statinfo = stat("$dirname/$name");
# Is it a directory
if (-d(_))
{
     #Yes, print the name, size and date
     print ("$name (directory), last modified " .
scalar(localtime($statinfo[9])) . "\n");
     }
   else
     {
     print ("$name, size = " . $statinfo[7] . ", last modified " .
scalar(localtime($statinfo[9])) . "\n");
     }
     }