X-Git-Url: https://ruderich.org/simon/gitweb/?a=blobdiff_plain;ds=inline;f=bin%2Fblhc;h=813936890b9369de332a6a9648bff06c6e9a05ba;hb=de65359d7659f6fbadeb128abf39ff881e05c458;hp=c4aefd29ad15d50efe165112eb0d72359baa88a6;hpb=cf83d58fd04f141f4552050b285e2fbfc8189bcb;p=blhc%2Fblhc.git
diff --git a/bin/blhc b/bin/blhc
index c4aefd2..8139368 100755
--- a/bin/blhc
+++ b/bin/blhc
@@ -22,16 +22,194 @@ use strict;
use warnings;
use Getopt::Long ();
+use Term::ANSIColor ();
our $VERSION = '0.01';
+# FUNCTIONS
+
+sub error_flags {
+ my ($message, $missing_flags_ref, $flag_renames_ref, $line) = @_;
+
+ # Rename flags if requested.
+ my @missing_flags = map {
+ (exists $flag_renames_ref->{$_})
+ ? $flag_renames_ref->{$_}
+ : $_
+ } @{$missing_flags_ref};
+
+ my $flags = join ' ', @missing_flags;
+ printf "%s (%s)%s %s",
+ error_color($message, 'red'), $flags, error_color(':', 'yellow'),
+ $line;
+}
+sub error_non_verbose_build {
+ my ($line) = @_;
+
+ printf "%s%s %s",
+ error_color('NONVERBOSE BUILD', 'red'),
+ error_color(':', 'yellow'),
+ $line;
+}
+sub error_color {
+ my ($message, $color) = @_;
+
+ # Use colors when writing to a terminal.
+ if (-t STDOUT) {
+ return Term::ANSIColor::colored($message, $color);
+ } else {
+ return $message;
+ }
+}
+
+sub any_flags_used {
+ my ($line, @flags) = @_;
+
+ foreach my $flag (@flags) {
+ return 1 if $line =~ /\s$flag(\s|\\|$)/;
+ }
+
+ return 0;
+}
+sub all_flags_used {
+ my ($line, $missing_flags_ref, @flags) = @_;
+
+ my @missing_flags = ();
+ foreach my $flag (@flags) {
+ if ($line !~ /\s$flag(\s|\\|$)/) {
+ push @missing_flags, $flag;
+ }
+ }
+
+ if (scalar @missing_flags == 0) {
+ return 1;
+ }
+
+ @{$missing_flags_ref} = @missing_flags;
+ return 0;
+}
+
+# Modifies $missing_flags_ref array.
+sub pic_pie_conflict {
+ my ($line, $pie, $missing_flags_ref, @flags_pie) = @_;
+
+ return 0 if not $pie;
+ return 0 if not any_flags_used($line, ('-fPIC', '-fpic'));
+
+ my %flags = map { $_ => 1 } @flags_pie;
+
+ # Remove all PIE flags from @missing_flags as they are not required with
+ # -fPIC.
+ my @result = grep {
+ not exists $flags{$_}
+ } @{$missing_flags_ref};
+ @{$missing_flags_ref} = @result;
+
+ # We got a conflict when no flags are left, thus only PIE flags were
+ # missing. If other flags were missing abort because the conflict is not
+ # the problem.
+ return scalar @result == 0;
+}
+
+sub is_non_verbose_build {
+ my ($line, $next_line, $cc_regex, $skip_ref) = @_;
+
+ my $cmake_non_verbose = qr/^\s*\[[\d ]+%\] Building (C|CXX) object (.+?)$/;
+ if (not ($line =~ /^checking if you want to see long compiling messages\.\.\. no/
+ or $line =~ /^\s*(CC|CCLD)\s+/
+ or $line =~ /^\s*(C|c)ompiling\s+/
+ or $line =~ /$cmake_non_verbose/)) {
+ return 0;
+ }
+
+ # On the first pass we only check if this line is verbose or not.
+ return 1 if not defined $next_line;
+
+ # Second pass, we have access to the next line.
+ ${$skip_ref} = 0;
+
+ # CMake prints the non-verbose messages also when building verbose. If a
+ # compiler and the file name occurs in the next line, treat it as verbose
+ # build.
+ if ($line =~ /$cmake_non_verbose/) {
+ # Get filename, we can't use the complete path as only parts of it are
+ # used in the real compiler command ...
+ $2 =~ m{/([a-zA-Z0-9._-]+)$};
+ my $file = $1;
+
+ if ($next_line =~ /\Q$file\E/ and $next_line =~ /$cc_regex/) {
+ # We still have to skip the current line as it doesn't contain any
+ # compiler commands.
+ ${$skip_ref} = 1;
+ return 0;
+ }
+ }
+
+ return 1;
+}
+
+
+# CONSTANTS/VARIABLES
+
+# Regex to catch compiler commands.
+my $cc_regex = qr/((? '-Wl,-z,relro',
+ '-Wl,(-z,)?now' => '-Wl,-z,now',
+);
+
+
+# MAIN
+
+# Additional hardening options.
+my $pie = 0;
+my $bindnow = 0;
+
# Parse command line arguments.
+my $option_all = 0;
my $option_help = 0;
my $option_version = 0;
if (not Getopt::Long::GetOptions(
'help|h|?' => \$option_help,
'version' => \$option_version,
+ 'pie' => \$pie,
+ 'bindnow' => \$bindnow,
+ 'all' => \$option_all,
)) {
require Pod::Usage;
Pod::Usage::pod2usage(2);
@@ -59,6 +237,148 @@ along with this program. If not, see .
exit 0;
}
+if ($option_all) {
+ $pie = 1;
+ $bindnow = 1;
+}
+
+# Final exit code.
+my $exit = 0;
+
+# Input lines, contain only the lines with compiler commands.
+my @input = ();
+
+my $continuation = 0;
+while (my $line = <>) {
+ # Ignore compiler warnings for now.
+ next if $line =~ /$warning_regex/;
+
+ # Check if this line indicates a non verbose build.
+ my $non_verbose = is_non_verbose_build($line);
+
+ # One line may contain multiple commands (";"). Treat each one as single
+ # line.
+ my @line = split /(? [-h -? --help]
+B [--pie] [--bindnow] [--all]
+
--help available options
--version version number and license
+ --pie force +pie check
+ --bindnow force +bindbow check
+ --all force +all (+pie, +bindnow) check
=head1 DESCRIPTION
@@ -90,6 +415,51 @@ Print available options.
Print version number and license.
+=item B<--pie>
+
+Force check for all +pie hardening flags. By default it's auto detected.
+
+=item B<--bindnow>
+
+Force check for all +bindnow hardening flags. By default it's auto detected.
+
+=item B<--all>
+
+Force check for all +all (+pie, +bindnow) hardening flags. By default it's
+auto detected.
+
+=back
+
+Auto detection only works if at least one command uses the required hardening
+flag (e.g. -fPIE). Then it's required for all other commands as well.
+
+=head1 EXIT STATUS
+
+The exit status is a "bit mask", each listed status is ORed when the error
+condition occurs to get the result.
+
+=over 8
+
+=item B<0>
+
+Success.
+
+=item B<1>
+
+No compiler commands were found.
+
+=item B<2>
+
+Invalid arguments/options given to blhc.
+
+=item B<4>
+
+Non verbose build.
+
+=item B<8>
+
+Missing hardening flags.
+
=back
=head1 AUTHOR