#!/usr/bin/perl
# $Id: strip_and_check_elf_files 258134 2009-06-25 17:21:10Z peroyvind $
# Strip files

use strict;
use warnings;
use File::Find;

my $buildroot = $ENV{RPM_BUILD_ROOT};
die "No build root defined" unless $buildroot;
die "Invalid build root" unless -d $buildroot;
# normalize build root
$buildroot =~ s|/$||;

my (@shared_libs, @executables, @static_libs);
find(\&keep_wanted, $buildroot);

strip_files() if !$ENV{DONT_STRIP};
check_missing_or_unused_libs();

sub strip_files {
    my @to_strip = (@shared_libs, @executables);

    if ($ENV{EXCLUDE_FROM_STRIP}) {
	my $exclude_pattern = join('|', split(/\s+/, $ENV{EXCLUDE_FROM_STRIP}));
	my $compiled_pattern = qr/($exclude_pattern)/;
	@to_strip = grep { !/$compiled_pattern/ } @to_strip;
    }

    system(
	"strip",
	"--remove-section=.comment",
	"--remove-section=.note",
	$_) foreach @to_strip;
}

sub check_missing_or_unused_libs {
    foreach my $f (@shared_libs, @executables) {
	my (undef, undef, @l) = `ldd -u -r $f 2>/dev/null`;
	@l or next;
	my $f_ = substr($f, length($buildroot));
	print STDERR "Warning: unused libraries in $f_: ", join(' ', map { basename($_) } @l), "\n";
    }
    foreach my $f (@shared_libs) {
	my @l = `ldd -r $f 2>&1 >/dev/null` or next;
	my $f_ = substr($f, length($buildroot));
	print STDERR "Warning: undefined symbols in $f_: ", join(' ', map { /undefined symbol: (\S+)/ ? $1 : () } @l), "\n";
    }
}

# TODO: we should write a binding for libfile...
sub expensive_test {
    my ($file) = @_;
    my $type = `file -- $file`;
}

# Check if a file is an elf binary, shared library, or static library,
# for use by File::Find. It'll fill the following 3 arrays with anything
# it finds:
sub keep_wanted() {
    # skip everything but files
    return unless -f $_;
    # skip symlinks
    return if -l $_;
    return if $File::Find::dir =~ m!/usr/lib/debug($|/)!;

    # Does its filename look like a shared library?
    if (m/\.so/) {
        # Ok, do the expensive test.
        if (expensive_test($_) =~ m/ELF.*shared/) {
            push @shared_libs, $File::Find::name;
            return;
        }
    }
    
    # Is it executable? -x isn't good enough, so we need to use stat.
    my (undef, undef, $mode, undef) = stat(_);
    if ($mode & 0111) {
        # Ok, expensive test.
        if (expensive_test($_) =~ m/ELF.*executable/) {
            push @executables, $File::Find::name;
            return;
        }
    }
    
    # Is it a static library, and not a debug library?
    if (m/lib.*\.a/ && ! m/_g\.a/) {
        push @static_libs, $File::Find::name;
        return;
    }
}

sub basename { local $_ = shift; s|/*\s*$||; s|.*/||; $_ }
