diff options
Diffstat (limited to 'depfixer.py')
| -rwxr-xr-x | depfixer.py | 33 |
1 files changed, 28 insertions, 5 deletions
diff --git a/depfixer.py b/depfixer.py index 026fc30..c93d58c 100755 --- a/depfixer.py +++ b/depfixer.py @@ -187,6 +187,17 @@ class Elf(): self.bf.seek(strtab.val + soname.val) print(self.read_str()) + def get_rpath_offset(self): + sec = self.find_section(b'.dynstr') + for i in self.dynamic: + if i.d_tag == DT_RPATH: + rpath = i + return sec.sh_offset + rpath.val + + def print_rpath(self): + self.bf.seek(self.get_rpath_offset()) + print(self.read_str()) + def print_deps(self): sec = self.find_section(b'.dynstr') deps = [] @@ -217,14 +228,26 @@ class Elf(): self.bf.seek(offset) self.bf.write(newname) + def fix_rpath(self, new_rpath): + rp_off = self.get_rpath_offset() + self.bf.seek(rp_off) + old_rpath = self.read_str() + if len(old_rpath) < len(new_rpath): + print("New rpath must not be longer than the old one.") + self.bf.seek(rp_off) + self.bf.write(new_rpath) + self.bf.write(b'\0'*(len(old_rpath) - len(new_rpath) + 1)) if __name__ == '__main__': - if len(sys.argv) != 3: - print('This application converts absolute dep paths to relative ones.') + if len(sys.argv) < 2 or len(sys.argv) > 3: + print('This application resets target rpath.') print('Don\'t run this unless you know what you are doing.') print('%s: <binary file> <prefix>' % sys.argv[0]) exit(1) e = Elf(sys.argv[1]) - prefix = sys.argv[2] - #e.print_deps() - e.fix_deps(prefix.encode()) + if len(sys.argv) == 2: + e.print_rpath() + else: + new_rpath = sys.argv[2] + e.fix_rpath(new_rpath.encode('utf8')) + #e.fix_deps(prefix.encode()) |