forked from semios/ports
@@ -0,0 +1,61 @@
|
||||
import os
|
||||
import stat
|
||||
import subprocess
|
||||
import tempfile
|
||||
import shutil
|
||||
|
||||
from lib import common
|
||||
|
||||
strip = os.environ.get("STRIP", "strip")
|
||||
|
||||
|
||||
def _is_elf(filepath):
|
||||
with open(filepath, "rb") as f:
|
||||
return f.read(4) == b"\x7fELF"
|
||||
|
||||
|
||||
def autostrip(composer):
|
||||
processed_inodes = {}
|
||||
|
||||
for ent in common.treedir(composer.workdir):
|
||||
if not ent.is_file(follow_symlinks=False):
|
||||
continue
|
||||
if not (os.access(ent.path, os.X_OK) or ".so" in ent.name):
|
||||
continue
|
||||
if not _is_elf(ent.path):
|
||||
continue
|
||||
|
||||
stat_info = os.stat(ent.path)
|
||||
inode_key = (stat_info.st_dev, stat_info.st_ino)
|
||||
if inode_key in processed_inodes:
|
||||
continue
|
||||
processed_inodes[inode_key] = True
|
||||
|
||||
oldperm = stat.S_IMODE(stat_info.st_mode)
|
||||
|
||||
try:
|
||||
with tempfile.NamedTemporaryFile(mode='wb', delete=False) as tmp_file:
|
||||
tmp_path = tmp_file.name
|
||||
shutil.copy2(ent.path, tmp_path)
|
||||
os.chmod(tmp_path, 0o777)
|
||||
|
||||
subprocess.run([strip, tmp_path], check=True)
|
||||
|
||||
with open(tmp_path, 'rb') as tmp_file:
|
||||
stripped_data = tmp_file.read()
|
||||
|
||||
os.chmod(ent.path, 0o777)
|
||||
|
||||
with open(ent.path, 'r+b') as original_file:
|
||||
original_file.truncate(0)
|
||||
original_file.write(stripped_data)
|
||||
original_file.flush()
|
||||
os.fsync(original_file.fileno())
|
||||
|
||||
os.chmod(ent.path, oldperm)
|
||||
os.unlink(tmp_path)
|
||||
|
||||
except Exception as e:
|
||||
if 'tmp_path' in locals() and os.path.exists(tmp_path):
|
||||
os.unlink(tmp_path)
|
||||
raise RuntimeError(f"Failed to strip {ent.path}: {e}") from e
|
||||
Reference in New Issue
Block a user