borekon/lynis-report-converter
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Added to PDF option

Browse Source
This commit is contained in:
dataking 2016-07-28 00:33:59 -07:00
parent fe610b705e
commit 25f0c28449
1 changed files with 22 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
lynis_report.pl
View File

@ -9,13 +9,16 @@ use Getopt::Long qw( :config no_ignore_case bundling );
use Data::Dumper; use Data::Dumper;
#use Spreadsheet::WriteExcel; #use Spreadsheet::WriteExcel;
use Excel::Writer::XLSX; use Excel::Writer::XLSX;
#use File::Basename;
use HTML::HTMLDoc;
my ($help,$verbose,$excel,$output); my ($help,$verbose,$excel,$output,$pdf);
GetOptions( GetOptions(
'h|help' => \$help, 'h|help' => \$help,
'v|verbose+' => \$verbose, 'v|verbose+' => \$verbose,
'E|excel' => \$excel, 'E|excel' => \$excel,
'o|output=s' => \$output, 'o|output=s' => \$output,
'p|pdf' => \$pdf,
); );
if ($help) { &usage; } if ($help) { &usage; }
@ -23,10 +26,16 @@ if ($help) { &usage; }
my %to_bool = ( 0 => 'false', 1 => 'true' ); my %to_bool = ( 0 => 'false', 1 => 'true' );
my %to_long_severity = ( 'C' => 'Critical', 'S' => 'Severe', 'H' => 'High', 'M' => 'Medium', 'L' => 'Low', 'I' => 'Informational' ); my %to_long_severity = ( 'C' => 'Critical', 'S' => 'Severe', 'H' => 'High', 'M' => 'Medium', 'L' => 'Low', 'I' => 'Informational' );
my ($basename, $path, $suffix, $htmldoc);
if ($excel) { if ($excel) {
$output = 'report.xlsx' unless ((defined($output)) and ($output ne "")); $output = 'report.xlsx' unless ((defined($output)) and ($output ne ""));
} elsif ($pdf) {
$output = 'report.pdf' unless ((defined($output)) and ($output ne ''));
$htmldoc = "$$.html";
} else { } else {
$output = "report.html" unless ((defined($output)) and ($output ne "")); $output = "report.html" unless ((defined($output)) and ($output ne ""));
$htmldoc = $output
} }
my $lynis_log = '/var/log/lynis.log'; my $lynis_log = '/var/log/lynis.log';
@ -50,7 +59,8 @@ if (($audit_run) and ($audit_run >= 1)) {
} }
print "Outputting report to $output, in "; print "Outputting report to $output, in ";
if ($excel) { print "Excel " } if ($excel) { print "Excel "; }
elsif ($pdf) { print "PDF "; }
else { print "HTML "; } else { print "HTML "; }
print "format. \n"; print "format. \n";
@ -158,7 +168,7 @@ if ($excel) {
$summary_ws->add_table("A8:D$last_row_number", \%params); $summary_ws->add_table("A8:D$last_row_number", \%params);
} else { } else {
open OUT, ">$output" or die colored("There was a problem opening the output file ($output): $! \n", "bold red"); open OUT, ">$htmldoc" or die colored("There was a problem opening the output file ($htmldoc): $! \n", "bold red");
print OUT <<END; print OUT <<END;
<html > <html >
@ -678,6 +688,15 @@ END
foreach my $idx ( sort @indexes ) { foreach my $idx ( sort @indexes ) {
delete($lynis_report_data{$idx}); delete($lynis_report_data{$idx});
} }
if ($pdf) {
my $htmlobj = new HTML::HTMLDoc();
$htmlobj->set_input_file($htmldoc);
my $pdfdoc = $htmlobj->generate_pdf();
$pdfdoc->to_file($output);
my $errs = system("rm -f $htmldoc");
if ($verbose) { print "Clean up return code: $errs \n"; }
}
} }
#print Dumper(\%lynis_report_data); #print Dumper(\%lynis_report_data);