#!@PERL@

# fakeldd
#
# Replacement for ldd with usage of objdump
#
# (c) 2003-2010, 2013, 2016 Piotr Roszatycki <dexter@debian.org>, LGPL

use strict;

$ENV{LANG} = $ENV{LC_ALL} = 'C';

my @Libs = ();
my %Libs = ();

my $Status = 0;
my $Dynamic = 0;
my $Format = '';

my $Ldsodir = "/lib";
my @Ld_Library_Path = qw(/usr/lib /lib /usr/lib32 /lib32 /usr/lib64 /lib64);

my $Cwd = `pwd`;
chomp $Cwd;

my $Base = defined $ENV{FAKECHROOT_BASE_ORIG} ? $ENV{FAKECHROOT_BASE_ORIG} : '';

sub ldso {
    my ($lib) = @_;

    return if $Libs{$lib};

    my $path;

    if ($lib =~ m{^/}) {
        $path = $lib;
    }
    else {
        foreach my $dir (@Ld_Library_Path) {
            next unless -f "$dir/$lib";

            my $badformat = 0;
            local *PIPE;
            open PIPE, "objdump -p '$dir/$lib' 2>/dev/null |";
            while (my $line = <PIPE>) {
                if ($line =~ /file format (\S*)$/) {
                    $badformat = 1 unless $1 eq $Format;
                    last;
                }
            }
            close PIPE;

            next if $badformat;

            $path = "$dir/$lib";
            last;
        }
    }

    push @Libs, $lib;
    if (-f $path) {
        $path =~ s{^\Q$Base/\E}{/} if $Base;
        $Libs{$lib} = $path;
        objdump($path);
    }
}


sub objdump {
    my (@files) = @_;

    foreach my $file (@files) {
        $file = $file =~ m{^/} ? "$Base$file" : "$Cwd/$file";

        local *PIPE;
        open PIPE, "objdump -p '$file' 2>/dev/null |";

        while (my $line = <PIPE>) {
            $line =~ s/^\s+//;

            if ($line =~ /file format (\S*)$/) {
                if (not $Format) {
                    $Format = $1;

                    if ($^O eq 'linux') {
                        if ($Format =~ /^elf64-(x86-64|sparc)$/) {
                            $Ldsodir = "/lib64";
                        }
                        elsif ($Format =~ /^elf32-x86-64$/) {
                            $Ldsodir = "/libx32";
                        }

                        if ($Format =~ /^elf64-/) {
                            push @Libs, 'linux-vdso.so.1';
                            $Libs{'linux-vdso.so.1'} = '';
                        }
                        else {
                            push @Libs, 'linux-gate.so.1';
                            $Libs{'linux-gate.so.1'} = '';
                        }
                    }

                    foreach my $lib (split /[:\s]/, $ENV{LD_PRELOAD}||'') {
                        ldso($lib);
                    }
                }
                else {
                    next unless $Format eq $1;
                }
            }
            if (not $Dynamic and $line =~ /^Dynamic Section:/) {
                $Dynamic = 1;
            }

            next unless $line =~ /^ \s* NEEDED \s+ (.*) \s* $/x;

            my $needed = $1;
            if ($needed =~ /^ld(-linux)?(\.|-)/) {
                $needed = "$Ldsodir/$needed";
            }
            ldso($needed);
        }
        close PIPE;
    }
}

# mips64el, ppc64el and s390x do not list the linker itself
# if it's missing, obtain it from the .interp section
#
# mips64el: /lib64/ld.so.1
# ppc64el: /lib64/ld64.so.2
# s390x: /lib/ld64.so.1
sub elfinterp {
    my $file = shift;
    my $res = '';
    local *PIPE;
    open PIPE, "objdump -sj .interp '$file' 2>/dev/null |";
    while (my $line = <PIPE>) {
        if ( $line !~ /^ [a-f0-9]+ ([a-f0-9][a-f0-9][a-f0-9 ]{6} [a-f0-9 ]{8} [a-f0-9 ]{8} [a-f0-9 ]{8})  /) {
            next;
        }
        $line = $1;
        $line =~ s/ //g;
        $line =~ s/(..)/chr(hex($1))/eg;
        $res .= $line;
    }
    close PIPE;

    # remove trailing NUL byte
    $res =~ s/\000$//;

    # only add if it is missing
    if ( $res && !exists $Libs{$res} ) {
        push @Libs, $res;
        $Libs{$res} = '';
    }
}


sub load_ldsoconf {
    my ($file) = @_;
    my @paths;

    local *FH;
    open FH, $file;
    while (my $line = <FH>) {
        chomp $line;
        $line =~ s/#.*//;
        next if $line =~ /^\s*$/;

        if ($line =~ /^include\s+(.*)\s*/) {
            my $include = $1;
            foreach my $incfile (glob $include) {
                load_ldsoconf($incfile);
            }
            next;
        }

        push @paths, $line;
    }
    close FH;

    unshift @Ld_Library_Path, @paths;
}


MAIN: {
    my @args = @ARGV;

    if (not `sh -c 'command -v objdump'`) {
        print STDERR "fakeldd: objdump: command not found: install binutils package\n";
        exit 1;
    }

    load_ldsoconf('/etc/ld.so.conf');
    unshift @Ld_Library_Path, split(/:/, $ENV{LD_LIBRARY_PATH}||'');

    while ($args[0] =~ /^-/) {
        my $arg = $args[0];
        shift @args;
        last if $arg eq "--";
    }

    if (not @args) {
        print STDERR "fakeldd: missing file arguments\n";
        exit 1;
    }

    foreach my $file (@args) {
        %Libs = ();
        $Dynamic = 0;

        if (@args > 1) {
            print "$file:\n";
        }

        my $file_in_chroot = $file =~ m{^/} ? "$Base$file" : "$file";

        if (not -f $file_in_chroot) {
            print STDERR "fakeldd: $file: No such file or directory\n";
            $Status = 1;
            next;
        }

        objdump($file);
        elfinterp($file_in_chroot);

        if ($Dynamic == 0) {
            print "\tnot a dynamic executable\n";
            $Status = 1;
        }
        elsif (scalar %Libs eq "0") {
            print "\tstatically linked\n";
        }

        my $address = '0x' . '0' x ($Format =~ /^elf64-/ ? 16 : 8);

        my %seen;

        foreach my $lib (@Libs) {
            next if $seen{$lib}++;
            if ($lib =~ m{^/} or $lib =~ /^linux-/) {
                printf "\t%s (%s)\n", $lib, $address;
            }
            elsif ($Libs{$lib}) {
                printf "\t%s => %s (%s)\n", $lib, $Libs{$lib}, $address;
            }
            else {
                printf "\t%s => not found\n", $lib;
            }
        }

    }
}

END {
    $? = $Status;
}