mirrors/cpython
0
0
Fork 0
mirror of https://github.com/python/cpython.git synced 2024-11-23 09:54:58 +08:00
Code Issues Packages Projects Releases Wiki Activity

gh-103935: Use io.open_code() when executing code in trace and profile modules (GH-103947)

Browse Source
This commit is contained in:
Tian Gao 2023-04-27 13:29:35 -07:00 committed by GitHub
parent bf0b8a9f8d
commit d50f01ad0a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
Lib/cProfile.py
View File

@ -8,6 +8,7 @@ __all__ = ["run", "runctx", "Profile"]
import _lsprof
import importlib.machinery
import io
import profile as _pyprofile
# ____________________________________________________________
@ -168,7 +169,7 @@ def main():
else:
progname = args[0]
sys.path.insert(0, os.path.dirname(progname))
with open(progname, 'rb') as fp:
with io.open_code(progname) as fp:
code = compile(fp.read(), progname, 'exec')
spec = importlib.machinery.ModuleSpec(name='__main__', loader=None,
origin=progname)

3
Lib/profile.py
View File

@ -25,6 +25,7 @@
import importlib.machinery
import io
import sys
import time
import marshal
@ -588,7 +589,7 @@ def main():
else:
progname = args[0]
sys.path.insert(0, os.path.dirname(progname))
with open(progname, 'rb') as fp:
with io.open_code(progname) as fp:
code = compile(fp.read(), progname, 'exec')
spec = importlib.machinery.ModuleSpec(name='__main__', loader=None,
origin=progname)

3
Lib/trace.py
View File

@ -49,6 +49,7 @@ Sample use, programmatically
"""
__all__ = ['Trace', 'CoverageResults']
import io
import linecache
import os
import sys
@ -716,7 +717,7 @@ def main():
sys.argv = [opts.progname, *opts.arguments]
sys.path[0] = os.path.dirname(opts.progname)
with open(opts.progname, 'rb') as fp:
with io.open_code(opts.progname) as fp:
code = compile(fp.read(), opts.progname, 'exec')
# try to emulate __main__ namespace as much as possible
globs = {

1
Misc/NEWS.d/next/Library/2023-04-27-20-03-08.gh-issue-103935.Uaf2M0.rst Normal file
View File

@ -0,0 +1 @@
Use :func:`io.open_code` for files to be executed instead of raw :func:`open`