Lerking/python-appimage
1
0
Fork 0
mirror of https://github.com/niess/python-appimage.git synced 2026-03-14 12:20:14 +01:00
Code Issues Packages Projects Releases Wiki Activity

Secure the shebang patcher against binaries

Browse Source
This commit is contained in:
Valentin Niess
2020-04-05 00:00:06 +02:00
parent 15a34b71ab
commit 27ad66b8af
1 changed files with 10 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
python_appimage/data/sitecustomize.py
View File

@@ -37,14 +37,18 @@ def patch_pip_install():
path = os.path.join(sys.prefix, 'bin', exe)
if (not os.path.isfile(path)) or (not os.access(path, os.X_OK)) or \
exe.startswith('python') or os.path.islink(path):
exe.startswith('python') or os.path.islink(path) or \
exe.endswith('.pyc') or exe.endswith('.pyo'):
continue
with open(path, 'r') as f:
header = f.read(2)
if header != '#!':
continue
content = f.read()
try:
with open(path, 'r') as f:
header = f.read(2)
if header != '#!':
continue
content = f.read()
except:
continue
shebang, body = content.split(os.linesep, 1)
shebang = shebang.split()