1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
|
import os
import sys
sys.argv = sys.argv[1:len(sys.argv)]
filename = sys.argv[0]
if os.path.exists(filename) or filename[0] == '/':
os.environ["UFTRACE_PYMAIN"] = filename
if filename[0] == '/':
pathname = filename
else:
pathname = os.getcwd() + '/' + filename
else:
for dir in os.environ["PATH"].split(":"):
pathname = dir + '/' + filename
try:
f = open(pathname)
sys.argv[0] = pathname
os.environ["UFTRACE_PYMAIN"] = pathname
f.close()
break
except OSError:
continue
# UFTRACE_PYMAIN must be set before importing uftrace_python
import uftrace_python
# Symbol and debug files are finally written at uftrace_trace_python_finish()
# when program exits, but os._exit() terminates the program immediately so there
# is no chance to write symbol and debug files at the destructor.
# The os._exit() is hooked here to prevent this problem.
def os_exit(n):
uftrace_python.exit(n)
os._exit = os_exit
new_globals = globals()
new_globals["__file__"] = pathname
sys.path.insert(0, os.path.dirname(pathname))
code = open(sys.argv[0]).read()
sys.setprofile(uftrace_python.trace)
exec(code, new_globals)
sys.setprofile(None)
|