267 lines
8.9 KiB
Python
267 lines
8.9 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
#
|
||
|
|
# Copyright (c) 2012 Intel Corporation
|
||
|
|
#
|
||
|
|
# This program is free software; you can redistribute it and/or modify
|
||
|
|
# it under the terms of the GNU General Public License version 2 as
|
||
|
|
# published by the Free Software Foundation.
|
||
|
|
#
|
||
|
|
# This program is distributed in the hope that it will be useful,
|
||
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
|
||
|
|
# See the GNU General Public License for more details.
|
||
|
|
#
|
||
|
|
# You should have received a copy of the GNU General Public License
|
||
|
|
# along with this program; if not, write to the Free Software
|
||
|
|
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
||
|
|
#
|
||
|
|
# DESCRIPTION
|
||
|
|
# This script is called by the SDK installer script. It replaces the dynamic
|
||
|
|
# loader path in all binaries and also fixes the SYSDIR paths/lengths and the
|
||
|
|
# location of ld.so.cache in the dynamic loader binary
|
||
|
|
#
|
||
|
|
# AUTHORS
|
||
|
|
# Laurentiu Palcu <laurentiu.palcu@intel.com>
|
||
|
|
#
|
||
|
|
|
||
|
|
import struct
|
||
|
|
import sys
|
||
|
|
import stat
|
||
|
|
import os
|
||
|
|
import re
|
||
|
|
import errno
|
||
|
|
|
||
|
|
if sys.version < '3':
|
||
|
|
def b(x):
|
||
|
|
return x
|
||
|
|
else:
|
||
|
|
def b(x):
|
||
|
|
return x.encode(sys.getfilesystemencoding())
|
||
|
|
|
||
|
|
old_prefix = re.compile(b("##DEFAULT_INSTALL_DIR##"))
|
||
|
|
|
||
|
|
def get_arch():
|
||
|
|
f.seek(0)
|
||
|
|
e_ident =f.read(16)
|
||
|
|
ei_mag0,ei_mag1_3,ei_class = struct.unpack("<B3sB11x", e_ident)
|
||
|
|
|
||
|
|
if (ei_mag0 != 0x7f and ei_mag1_3 != "ELF") or ei_class == 0:
|
||
|
|
return 0
|
||
|
|
|
||
|
|
if ei_class == 1:
|
||
|
|
return 32
|
||
|
|
elif ei_class == 2:
|
||
|
|
return 64
|
||
|
|
|
||
|
|
def parse_elf_header():
|
||
|
|
global e_type, e_machine, e_version, e_entry, e_phoff, e_shoff, e_flags,\
|
||
|
|
e_ehsize, e_phentsize, e_phnum, e_shentsize, e_shnum, e_shstrndx
|
||
|
|
|
||
|
|
f.seek(0)
|
||
|
|
elf_header = f.read(64)
|
||
|
|
|
||
|
|
if arch == 32:
|
||
|
|
# 32bit
|
||
|
|
hdr_fmt = "<HHILLLIHHHHHH"
|
||
|
|
hdr_size = 52
|
||
|
|
else:
|
||
|
|
# 64bit
|
||
|
|
hdr_fmt = "<HHIQQQIHHHHHH"
|
||
|
|
hdr_size = 64
|
||
|
|
|
||
|
|
e_type, e_machine, e_version, e_entry, e_phoff, e_shoff, e_flags,\
|
||
|
|
e_ehsize, e_phentsize, e_phnum, e_shentsize, e_shnum, e_shstrndx =\
|
||
|
|
struct.unpack(hdr_fmt, elf_header[16:hdr_size])
|
||
|
|
|
||
|
|
def change_interpreter(elf_file_name):
|
||
|
|
if arch == 32:
|
||
|
|
ph_fmt = "<IIIIIIII"
|
||
|
|
else:
|
||
|
|
ph_fmt = "<IIQQQQQQ"
|
||
|
|
|
||
|
|
""" look for PT_INTERP section """
|
||
|
|
for i in range(0,e_phnum):
|
||
|
|
f.seek(e_phoff + i * e_phentsize)
|
||
|
|
ph_hdr = f.read(e_phentsize)
|
||
|
|
if arch == 32:
|
||
|
|
# 32bit
|
||
|
|
p_type, p_offset, p_vaddr, p_paddr, p_filesz,\
|
||
|
|
p_memsz, p_flags, p_align = struct.unpack(ph_fmt, ph_hdr)
|
||
|
|
else:
|
||
|
|
# 64bit
|
||
|
|
p_type, p_flags, p_offset, p_vaddr, p_paddr, \
|
||
|
|
p_filesz, p_memsz, p_align = struct.unpack(ph_fmt, ph_hdr)
|
||
|
|
|
||
|
|
""" change interpreter """
|
||
|
|
if p_type == 3:
|
||
|
|
# PT_INTERP section
|
||
|
|
f.seek(p_offset)
|
||
|
|
# External SDKs with mixed pre-compiled binaries should not get
|
||
|
|
# relocated so look for some variant of /lib
|
||
|
|
fname = f.read(11)
|
||
|
|
if fname.startswith(b("/lib/")) or fname.startswith(b("/lib64/")) or \
|
||
|
|
fname.startswith(b("/lib32/")) or fname.startswith(b("/usr/lib32/")) or \
|
||
|
|
fname.startswith(b("/usr/lib32/")) or fname.startswith(b("/usr/lib64/")):
|
||
|
|
break
|
||
|
|
if p_filesz == 0:
|
||
|
|
break
|
||
|
|
if (len(new_dl_path) >= p_filesz):
|
||
|
|
print("ERROR: could not relocate %s, interp size = %i and %i is needed." \
|
||
|
|
% (elf_file_name, p_memsz, len(new_dl_path) + 1))
|
||
|
|
break
|
||
|
|
dl_path = new_dl_path + b("\0") * (p_filesz - len(new_dl_path))
|
||
|
|
f.seek(p_offset)
|
||
|
|
f.write(dl_path)
|
||
|
|
break
|
||
|
|
|
||
|
|
def change_dl_sysdirs(elf_file_name):
|
||
|
|
if arch == 32:
|
||
|
|
sh_fmt = "<IIIIIIIIII"
|
||
|
|
else:
|
||
|
|
sh_fmt = "<IIQQQQIIQQ"
|
||
|
|
|
||
|
|
""" read section string table """
|
||
|
|
f.seek(e_shoff + e_shstrndx * e_shentsize)
|
||
|
|
sh_hdr = f.read(e_shentsize)
|
||
|
|
if arch == 32:
|
||
|
|
sh_offset, sh_size = struct.unpack("<16xII16x", sh_hdr)
|
||
|
|
else:
|
||
|
|
sh_offset, sh_size = struct.unpack("<24xQQ24x", sh_hdr)
|
||
|
|
|
||
|
|
f.seek(sh_offset)
|
||
|
|
sh_strtab = f.read(sh_size)
|
||
|
|
|
||
|
|
sysdirs = sysdirs_len = ""
|
||
|
|
|
||
|
|
""" change ld.so.cache path and default libs path for dynamic loader """
|
||
|
|
for i in range(0,e_shnum):
|
||
|
|
f.seek(e_shoff + i * e_shentsize)
|
||
|
|
sh_hdr = f.read(e_shentsize)
|
||
|
|
|
||
|
|
sh_name, sh_type, sh_flags, sh_addr, sh_offset, sh_size, sh_link,\
|
||
|
|
sh_info, sh_addralign, sh_entsize = struct.unpack(sh_fmt, sh_hdr)
|
||
|
|
|
||
|
|
name = sh_strtab[sh_name:sh_strtab.find(b("\0"), sh_name)]
|
||
|
|
|
||
|
|
""" look only into SHT_PROGBITS sections """
|
||
|
|
if sh_type == 1:
|
||
|
|
f.seek(sh_offset)
|
||
|
|
""" default library paths cannot be changed on the fly because """
|
||
|
|
""" the string lengths have to be changed too. """
|
||
|
|
if name == b(".sysdirs"):
|
||
|
|
sysdirs = f.read(sh_size)
|
||
|
|
sysdirs_off = sh_offset
|
||
|
|
sysdirs_sect_size = sh_size
|
||
|
|
elif name == b(".sysdirslen"):
|
||
|
|
sysdirslen = f.read(sh_size)
|
||
|
|
sysdirslen_off = sh_offset
|
||
|
|
elif name == b(".ldsocache"):
|
||
|
|
ldsocache_path = f.read(sh_size)
|
||
|
|
new_ldsocache_path = old_prefix.sub(new_prefix, ldsocache_path)
|
||
|
|
new_ldsocache_path = new_ldsocache_path.rstrip(b("\0"))
|
||
|
|
if (len(new_ldsocache_path) >= sh_size):
|
||
|
|
print("ERROR: could not relocate %s, .ldsocache section size = %i and %i is needed." \
|
||
|
|
% (elf_file_name, sh_size, len(new_ldsocache_path)))
|
||
|
|
sys.exit(-1)
|
||
|
|
# pad with zeros
|
||
|
|
new_ldsocache_path += b("\0") * (sh_size - len(new_ldsocache_path))
|
||
|
|
# write it back
|
||
|
|
f.seek(sh_offset)
|
||
|
|
f.write(new_ldsocache_path)
|
||
|
|
elif name == b(".gccrelocprefix"):
|
||
|
|
offset = 0
|
||
|
|
while (offset + 4096) <= sh_size:
|
||
|
|
path = f.read(4096)
|
||
|
|
new_path = old_prefix.sub(new_prefix, path)
|
||
|
|
new_path = new_path.rstrip(b("\0"))
|
||
|
|
if (len(new_path) >= 4096):
|
||
|
|
print("ERROR: could not relocate %s, max path size = 4096 and %i is needed." \
|
||
|
|
% (elf_file_name, len(new_path)))
|
||
|
|
sys.exit(-1)
|
||
|
|
# pad with zeros
|
||
|
|
new_path += b("\0") * (4096 - len(new_path))
|
||
|
|
#print "Changing %s to %s at %s" % (str(path), str(new_path), str(offset))
|
||
|
|
# write it back
|
||
|
|
f.seek(sh_offset + offset)
|
||
|
|
f.write(new_path)
|
||
|
|
offset = offset + 4096
|
||
|
|
if sysdirs != "" and sysdirslen != "":
|
||
|
|
paths = sysdirs.split(b("\0"))
|
||
|
|
sysdirs = b("")
|
||
|
|
sysdirslen = b("")
|
||
|
|
for path in paths:
|
||
|
|
""" exit the loop when we encounter first empty string """
|
||
|
|
if path == b(""):
|
||
|
|
break
|
||
|
|
|
||
|
|
new_path = old_prefix.sub(new_prefix, path)
|
||
|
|
sysdirs += new_path + b("\0")
|
||
|
|
|
||
|
|
if arch == 32:
|
||
|
|
sysdirslen += struct.pack("<L", len(new_path))
|
||
|
|
else:
|
||
|
|
sysdirslen += struct.pack("<Q", len(new_path))
|
||
|
|
|
||
|
|
""" pad with zeros """
|
||
|
|
sysdirs += b("\0") * (sysdirs_sect_size - len(sysdirs))
|
||
|
|
|
||
|
|
""" write the sections back """
|
||
|
|
f.seek(sysdirs_off)
|
||
|
|
f.write(sysdirs)
|
||
|
|
f.seek(sysdirslen_off)
|
||
|
|
f.write(sysdirslen)
|
||
|
|
|
||
|
|
# MAIN
|
||
|
|
if len(sys.argv) < 4:
|
||
|
|
sys.exit(-1)
|
||
|
|
|
||
|
|
# In python > 3, strings may also contain Unicode characters. So, convert
|
||
|
|
# them to bytes
|
||
|
|
if sys.version_info < (3,):
|
||
|
|
new_prefix = sys.argv[1]
|
||
|
|
new_dl_path = sys.argv[2]
|
||
|
|
else:
|
||
|
|
new_prefix = sys.argv[1].encode()
|
||
|
|
new_dl_path = sys.argv[2].encode()
|
||
|
|
|
||
|
|
executables_list = sys.argv[3:]
|
||
|
|
|
||
|
|
for e in executables_list:
|
||
|
|
perms = os.stat(e)[stat.ST_MODE]
|
||
|
|
if os.access(e, os.W_OK|os.R_OK):
|
||
|
|
perms = None
|
||
|
|
else:
|
||
|
|
os.chmod(e, perms|stat.S_IRWXU)
|
||
|
|
|
||
|
|
try:
|
||
|
|
f = open(e, "r+b")
|
||
|
|
except IOError:
|
||
|
|
exctype, ioex = sys.exc_info()[:2]
|
||
|
|
if ioex.errno == errno.ETXTBSY:
|
||
|
|
print("Could not open %s. File used by another process.\nPlease "\
|
||
|
|
"make sure you exit all processes that might use any SDK "\
|
||
|
|
"binaries." % e)
|
||
|
|
else:
|
||
|
|
print("Could not open %s: %s(%d)" % (e, ioex.strerror, ioex.errno))
|
||
|
|
sys.exit(-1)
|
||
|
|
|
||
|
|
# Save old size and do a size check at the end. Just a safety measure.
|
||
|
|
old_size = os.path.getsize(e)
|
||
|
|
if old_size >= 64:
|
||
|
|
arch = get_arch()
|
||
|
|
if arch:
|
||
|
|
parse_elf_header()
|
||
|
|
change_interpreter(e)
|
||
|
|
change_dl_sysdirs(e)
|
||
|
|
|
||
|
|
""" change permissions back """
|
||
|
|
if perms:
|
||
|
|
os.chmod(e, perms)
|
||
|
|
|
||
|
|
f.close()
|
||
|
|
|
||
|
|
if old_size != os.path.getsize(e):
|
||
|
|
print("New file size for %s is different. Looks like a relocation error!", e)
|
||
|
|
sys.exit(-1)
|
||
|
|
|