#!/usr/bin/python

# Usage: $0 file1 file2 ....

#

# Given binary files, remove all rpath entries from them containing

# $ORIGIN. Other rpath entries are not modified

import subprocess;

import sys;

def print_usage(name):

    print '''Usage: %s file1 file2 ....

    Given binary files, remove all rpath entries from them containing

    $ORIGIN. Other rpath entries are not modified''' % (name,)

def call(args):

    pop = subprocess.Popen(args, bufsize=-1, stdout=subprocess.PIPE, stderr=subprocess.PIPE)

    stdout, stderr = pop.communicate()

    if (len(stderr.strip())) != 0:

        print stderr

    if pop.returncode != 0:

        raise OSError('Error calling %s' % (str(args),))

    return stdout

def check_chrpath_present():

    try:

        print(call(['/usr/bin/chrpath', '-v']))

        return True

    except OSError:

        return False

def main(args):

    binaries = args[1:]

    if len(binaries) == 0:

        print_usage(args[0])

        return 1

    if not check_chrpath_present():

        print('Could not execute "chrpath". Is it installed?')

        return 1

    for binary in binaries:

        try:

            rpath_output = call(['chrpath', '-l', binary])

        except OSError:

            continue

        #print '"' + rpath_output + '"'

        # the output is "file: RUNPATH=path1:path2:path3\n"

        rpath_output = rpath_output.strip()

        start = rpath_output.find('RPATH=')

        rpath = rpath_output[start+len('RPATH='):]

        rpath_parts = rpath.split(':')

        modified_rpath = ''

        for part in rpath_parts:

            if not '$ORIGIN' in part:

                modified_rpath = modified_rpath + ':' + part

        # shave off the last ':'

        modified_rpath = modified_rpath[1:]

        #print '"' + modified_rpath + '"'

        if len(modified_rpath) == 0:

            call(['chrpath', '-d', binary])

            print '%s: Deleted RPATH' % (binary,)

        else:

            call(['chrpath', '-r', modified_rpath, binary])

            print '%s: RPATH=%s' % (binary,modified_rpath)

    return 0

if __name__ == '__main__':

    exit(main(sys.argv))