PageRenderTime 56ms CodeModel.GetById 22ms RepoModel.GetById 0ms app.codeStats 0ms

/pypy/rpython/llinterp.py

https://bitbucket.org/pwaller/pypy
Python | 1412 lines | 1400 code | 9 blank | 3 comment | 13 complexity | ae1c2bf837d76406046f77cf14ce3219 MD5 | raw file

Large files files are truncated, but you can click here to view the full file

  1. from pypy.objspace.flow.model import FunctionGraph, Constant, Variable, c_last_exception
  2. from pypy.rlib.rarithmetic import intmask, r_uint, ovfcheck, r_longlong
  3. from pypy.rlib.rarithmetic import r_ulonglong, is_valid_int
  4. from pypy.rpython.lltypesystem import lltype, llmemory, lloperation, llheap
  5. from pypy.rpython.lltypesystem import rclass
  6. from pypy.rpython.ootypesystem import ootype
  7. from pypy.rlib.objectmodel import ComputedIntSymbolic, CDefinedIntSymbolic
  8. from pypy.rlib.objectmodel import Symbolic
  9. from pypy.rlib import rstackovf
  10. import sys, os
  11. import math
  12. import py
  13. import traceback, cStringIO
  14. log = py.log.Producer('llinterp')
  15. class LLException(Exception):
  16. def __init__(self, *args):
  17. "NOT_RPYTHON"
  18. Exception.__init__(self, *args)
  19. def __str__(self):
  20. etype = self.args[0]
  21. #evalue = self.args[1]
  22. if len(self.args) > 2:
  23. f = cStringIO.StringIO()
  24. original_type, original_value, original_tb = self.args[2]
  25. traceback.print_exception(original_type, original_value, original_tb,
  26. file=f)
  27. extra = '\n' + f.getvalue().rstrip('\n')
  28. extra = extra.replace('\n', '\n | ') + '\n `------'
  29. else:
  30. extra = ''
  31. return '<LLException %r%s>' % (type_name(etype), extra)
  32. class LLFatalError(Exception):
  33. def __str__(self):
  34. return ': '.join([str(x) for x in self.args])
  35. def type_name(etype):
  36. if isinstance(lltype.typeOf(etype), lltype.Ptr):
  37. return ''.join(etype.name).rstrip('\x00')
  38. else:
  39. # ootype!
  40. return etype._INSTANCE._name.split(".")[-1]
  41. class LLInterpreter(object):
  42. """ low level interpreter working with concrete values. """
  43. current_interpreter = None
  44. def __init__(self, typer, tracing=True, exc_data_ptr=None):
  45. self.bindings = {}
  46. self.typer = typer
  47. # 'heap' is module or object that provides malloc, etc for lltype ops
  48. self.heap = llheap
  49. self.exc_data_ptr = exc_data_ptr
  50. self.frame_stack = []
  51. self.tracer = None
  52. self.frame_class = LLFrame
  53. if tracing:
  54. self.tracer = Tracer()
  55. def eval_graph(self, graph, args=(), recursive=False):
  56. llframe = self.frame_class(graph, args, self)
  57. if self.tracer and not recursive:
  58. global tracer1
  59. tracer1 = self.tracer
  60. self.tracer.start()
  61. retval = None
  62. self.traceback_frames = []
  63. old_frame_stack = self.frame_stack[:]
  64. prev_interpreter = LLInterpreter.current_interpreter
  65. LLInterpreter.current_interpreter = self
  66. try:
  67. try:
  68. retval = llframe.eval()
  69. except LLException, e:
  70. log.error("LLEXCEPTION: %s" % (e, ))
  71. self.print_traceback()
  72. if self.tracer:
  73. self.tracer.dump('LLException: %s\n' % (e,))
  74. raise
  75. except Exception, e:
  76. if getattr(e, '_go_through_llinterp_uncaught_', False):
  77. raise
  78. log.error("AN ERROR OCCURED: %s" % (e, ))
  79. self.print_traceback()
  80. if self.tracer:
  81. line = str(e)
  82. if line:
  83. line = ': ' + line
  84. line = '* %s' % (e.__class__.__name__,) + line
  85. self.tracer.dump(line + '\n')
  86. raise
  87. finally:
  88. LLInterpreter.current_interpreter = prev_interpreter
  89. assert old_frame_stack == self.frame_stack
  90. if self.tracer:
  91. if retval is not None:
  92. self.tracer.dump(' ---> %r\n' % (retval,))
  93. if not recursive:
  94. self.tracer.stop()
  95. return retval
  96. def print_traceback(self):
  97. frames = self.traceback_frames
  98. frames.reverse()
  99. self.traceback_frames = []
  100. lines = []
  101. for frame in frames:
  102. logline = frame.graph.name + "()"
  103. if frame.curr_block is None:
  104. logline += " <not running yet>"
  105. lines.append(logline)
  106. continue
  107. try:
  108. logline += " " + self.typer.annotator.annotated[frame.curr_block].func.__module__
  109. except (KeyError, AttributeError, TypeError):
  110. logline += " <unknown module>"
  111. lines.append(logline)
  112. for i, operation in enumerate(frame.curr_block.operations):
  113. if i == frame.curr_operation_index:
  114. logline = "E %s"
  115. else:
  116. logline = " %s"
  117. lines.append(logline % (operation, ))
  118. if self.tracer:
  119. self.tracer.dump('Traceback\n', bold=True)
  120. for line in lines:
  121. self.tracer.dump(line + '\n')
  122. for line in lines:
  123. log.traceback(line)
  124. def find_roots(self):
  125. """Return a list of the addresses of the roots."""
  126. #log.findroots("starting")
  127. roots = []
  128. for frame in self.frame_stack:
  129. #log.findroots("graph", frame.graph.name)
  130. frame.find_roots(roots)
  131. return roots
  132. def find_exception(self, exc):
  133. assert isinstance(exc, LLException)
  134. klass, inst = exc.args[0], exc.args[1]
  135. exdata = self.typer.getexceptiondata()
  136. frame = self.frame_class(None, [], self)
  137. for cls in enumerate_exceptions_top_down():
  138. evalue = frame.op_direct_call(exdata.fn_pyexcclass2exc,
  139. lltype.pyobjectptr(cls))
  140. etype = frame.op_direct_call(exdata.fn_type_of_exc_inst, evalue)
  141. if etype == klass:
  142. return cls
  143. raise ValueError("couldn't match exception, maybe it"
  144. " has RPython attributes like OSError?")
  145. def get_transformed_exc_data(self, graph):
  146. if hasattr(graph, 'exceptiontransformed'):
  147. return graph.exceptiontransformed
  148. if getattr(graph, 'rgenop', False):
  149. return self.exc_data_ptr
  150. return None
  151. def _store_exception(self, exc):
  152. raise PleaseOverwriteStoreException("You just invoked ll2ctypes callback without overwriting _store_exception on llinterpreter")
  153. class PleaseOverwriteStoreException(Exception):
  154. pass
  155. def checkptr(ptr):
  156. assert isinstance(lltype.typeOf(ptr), lltype.Ptr)
  157. def checkadr(addr):
  158. assert lltype.typeOf(addr) is llmemory.Address
  159. def is_inst(inst):
  160. return isinstance(lltype.typeOf(inst), (ootype.Instance, ootype.BuiltinType, ootype.StaticMethod))
  161. def checkinst(inst):
  162. assert is_inst(inst)
  163. class LLFrame(object):
  164. def __init__(self, graph, args, llinterpreter):
  165. assert not graph or isinstance(graph, FunctionGraph)
  166. self.graph = graph
  167. self.args = args
  168. self.llinterpreter = llinterpreter
  169. self.heap = llinterpreter.heap
  170. self.bindings = {}
  171. self.curr_block = None
  172. self.curr_operation_index = 0
  173. self.alloca_objects = []
  174. def newsubframe(self, graph, args):
  175. return self.__class__(graph, args, self.llinterpreter)
  176. # _______________________________________________________
  177. # variable setters/getters helpers
  178. def clear(self):
  179. self.bindings.clear()
  180. def fillvars(self, block, values):
  181. vars = block.inputargs
  182. assert len(vars) == len(values), (
  183. "block %s received %d args, expected %d" % (
  184. block, len(values), len(vars)))
  185. for var, val in zip(vars, values):
  186. self.setvar(var, val)
  187. def setvar(self, var, val):
  188. if var.concretetype is not lltype.Void:
  189. try:
  190. val = lltype.enforce(var.concretetype, val)
  191. except TypeError:
  192. assert False, "type error: input value of type:\n\n\t%r\n\n===> variable of type:\n\n\t%r\n" % (lltype.typeOf(val), var.concretetype)
  193. assert isinstance(var, Variable)
  194. self.bindings[var] = val
  195. def setifvar(self, var, val):
  196. if isinstance(var, Variable):
  197. self.setvar(var, val)
  198. def getval(self, varorconst):
  199. try:
  200. val = varorconst.value
  201. except AttributeError:
  202. val = self.bindings[varorconst]
  203. if isinstance(val, ComputedIntSymbolic):
  204. val = val.compute_fn()
  205. if varorconst.concretetype is not lltype.Void:
  206. try:
  207. val = lltype.enforce(varorconst.concretetype, val)
  208. except TypeError:
  209. assert False, "type error: %r val from %r var/const" % (lltype.typeOf(val), varorconst.concretetype)
  210. return val
  211. def getval_or_subop(self, varorsubop):
  212. from pypy.translator.oosupport.treebuilder import SubOperation
  213. if isinstance(varorsubop, SubOperation):
  214. self.eval_operation(varorsubop.op)
  215. resultval = self.getval(varorsubop.op.result)
  216. del self.bindings[varorsubop.op.result] # XXX hack
  217. return resultval
  218. else:
  219. return self.getval(varorsubop)
  220. # _______________________________________________________
  221. # other helpers
  222. def getoperationhandler(self, opname):
  223. ophandler = getattr(self, 'op_' + opname, None)
  224. if ophandler is None:
  225. # try to import the operation from opimpl.py
  226. ophandler = lloperation.LL_OPERATIONS[opname].fold
  227. setattr(self.__class__, 'op_' + opname, staticmethod(ophandler))
  228. return ophandler
  229. # _______________________________________________________
  230. # evaling functions
  231. def eval(self):
  232. graph = self.graph
  233. tracer = self.llinterpreter.tracer
  234. if tracer:
  235. tracer.enter(graph)
  236. self.llinterpreter.frame_stack.append(self)
  237. try:
  238. try:
  239. nextblock = graph.startblock
  240. args = self.args
  241. while 1:
  242. self.clear()
  243. self.fillvars(nextblock, args)
  244. nextblock, args = self.eval_block(nextblock)
  245. if nextblock is None:
  246. for obj in self.alloca_objects:
  247. obj._obj._free()
  248. return args
  249. except Exception:
  250. self.llinterpreter.traceback_frames.append(self)
  251. raise
  252. finally:
  253. leavingframe = self.llinterpreter.frame_stack.pop()
  254. assert leavingframe is self
  255. if tracer:
  256. tracer.leave()
  257. def eval_block(self, block):
  258. """ return (nextblock, values) tuple. If nextblock
  259. is None, values is the concrete return value.
  260. """
  261. self.curr_block = block
  262. catch_exception = block.exitswitch == c_last_exception
  263. e = None
  264. try:
  265. for i, op in enumerate(block.operations):
  266. self.curr_operation_index = i
  267. self.eval_operation(op)
  268. except LLException, e:
  269. if not (catch_exception and op is block.operations[-1]):
  270. raise
  271. except RuntimeError, e:
  272. rstackovf.check_stack_overflow()
  273. # xxx fish fish fish for proper etype and evalue to use
  274. rtyper = self.llinterpreter.typer
  275. bk = rtyper.annotator.bookkeeper
  276. classdef = bk.getuniqueclassdef(rstackovf._StackOverflow)
  277. exdata = rtyper.getexceptiondata()
  278. evalue = exdata.get_standard_ll_exc_instance(rtyper, classdef)
  279. etype = exdata.fn_type_of_exc_inst(evalue)
  280. e = LLException(etype, evalue)
  281. if not (catch_exception and op is block.operations[-1]):
  282. raise e
  283. # determine nextblock and/or return value
  284. if len(block.exits) == 0:
  285. # return block
  286. tracer = self.llinterpreter.tracer
  287. if len(block.inputargs) == 2:
  288. # exception
  289. if tracer:
  290. tracer.dump('raise')
  291. etypevar, evaluevar = block.getvariables()
  292. etype = self.getval(etypevar)
  293. evalue = self.getval(evaluevar)
  294. # watch out, these are _ptr's
  295. raise LLException(etype, evalue)
  296. resultvar, = block.getvariables()
  297. result = self.getval(resultvar)
  298. exc_data = self.llinterpreter.get_transformed_exc_data(self.graph)
  299. if exc_data:
  300. # re-raise the exception set by this graph, if any
  301. etype = exc_data.exc_type
  302. if etype:
  303. evalue = exc_data.exc_value
  304. if tracer:
  305. tracer.dump('raise')
  306. exc_data.exc_type = lltype.typeOf(etype )._defl()
  307. exc_data.exc_value = lltype.typeOf(evalue)._defl()
  308. from pypy.translator import exceptiontransform
  309. T = resultvar.concretetype
  310. errvalue = exceptiontransform.error_value(T)
  311. # check that the exc-transformed graph returns the error
  312. # value when it returns with an exception set
  313. assert result == errvalue
  314. raise LLException(etype, evalue)
  315. if tracer:
  316. tracer.dump('return')
  317. return None, result
  318. elif block.exitswitch is None:
  319. # single-exit block
  320. assert len(block.exits) == 1
  321. link = block.exits[0]
  322. elif catch_exception:
  323. link = block.exits[0]
  324. if e:
  325. exdata = self.llinterpreter.typer.getexceptiondata()
  326. cls = e.args[0]
  327. inst = e.args[1]
  328. for link in block.exits[1:]:
  329. assert issubclass(link.exitcase, py.builtin.BaseException)
  330. if self.op_direct_call(exdata.fn_exception_match,
  331. cls, link.llexitcase):
  332. self.setifvar(link.last_exception, cls)
  333. self.setifvar(link.last_exc_value, inst)
  334. break
  335. else:
  336. # no handler found, pass on
  337. raise e
  338. else:
  339. llexitvalue = self.getval(block.exitswitch)
  340. if block.exits[-1].exitcase == "default":
  341. defaultexit = block.exits[-1]
  342. nondefaultexits = block.exits[:-1]
  343. assert defaultexit.llexitcase is None
  344. else:
  345. defaultexit = None
  346. nondefaultexits = block.exits
  347. for link in nondefaultexits:
  348. if link.llexitcase == llexitvalue:
  349. break # found -- the result is in 'link'
  350. else:
  351. if defaultexit is None:
  352. raise ValueError("exit case %r not found in the exit links "
  353. "of %r" % (llexitvalue, block))
  354. else:
  355. link = defaultexit
  356. return link.target, [self.getval(x) for x in link.args]
  357. def eval_operation(self, operation):
  358. tracer = self.llinterpreter.tracer
  359. if tracer:
  360. tracer.dump(str(operation))
  361. ophandler = self.getoperationhandler(operation.opname)
  362. # XXX slighly unnice but an important safety check
  363. if operation.opname == 'direct_call':
  364. assert isinstance(operation.args[0], Constant)
  365. elif operation.opname == 'indirect_call':
  366. assert isinstance(operation.args[0], Variable)
  367. if getattr(ophandler, 'specialform', False):
  368. retval = ophandler(*operation.args)
  369. else:
  370. vals = [self.getval_or_subop(x) for x in operation.args]
  371. if getattr(ophandler, 'need_result_type', False):
  372. vals.insert(0, operation.result.concretetype)
  373. try:
  374. retval = ophandler(*vals)
  375. except LLException, e:
  376. # safety check check that the operation is allowed to raise that
  377. # exception
  378. if operation.opname in lloperation.LL_OPERATIONS:
  379. canraise = lloperation.LL_OPERATIONS[operation.opname].canraise
  380. if Exception not in canraise:
  381. exc = self.llinterpreter.find_exception(e)
  382. for canraiseexc in canraise:
  383. if issubclass(exc, canraiseexc):
  384. break
  385. else:
  386. raise TypeError("the operation %s is not expected to raise %s" % (operation, exc))
  387. # for exception-transformed graphs, store the LLException
  388. # into the exc_data used by this graph
  389. exc_data = self.llinterpreter.get_transformed_exc_data(
  390. self.graph)
  391. if exc_data:
  392. etype = e.args[0]
  393. evalue = e.args[1]
  394. exc_data.exc_type = etype
  395. exc_data.exc_value = evalue
  396. from pypy.translator import exceptiontransform
  397. retval = exceptiontransform.error_value(
  398. operation.result.concretetype)
  399. else:
  400. raise
  401. self.setvar(operation.result, retval)
  402. if tracer:
  403. if retval is None:
  404. tracer.dump('\n')
  405. else:
  406. tracer.dump(' ---> %r\n' % (retval,))
  407. def make_llexception(self, exc=None):
  408. if exc is None:
  409. original = sys.exc_info()
  410. exc = original[1]
  411. # it makes no sense to convert some exception classes that
  412. # just mean something buggy crashed
  413. if isinstance(exc, (AssertionError, AttributeError,
  414. TypeError, NameError,
  415. KeyboardInterrupt, SystemExit,
  416. ImportError, SyntaxError)):
  417. raise original[0], original[1], original[2] # re-raise it
  418. # for testing the JIT (see ContinueRunningNormally) we need
  419. # to let some exceptions introduced by the JIT go through
  420. # the llinterpreter uncaught
  421. if getattr(exc, '_go_through_llinterp_uncaught_', False):
  422. raise original[0], original[1], original[2] # re-raise it
  423. extraargs = (original,)
  424. else:
  425. extraargs = ()
  426. typer = self.llinterpreter.typer
  427. exdata = typer.getexceptiondata()
  428. if isinstance(exc, OSError):
  429. self.op_direct_call(exdata.fn_raise_OSError, exc.errno)
  430. assert False, "op_direct_call above should have raised"
  431. else:
  432. exc_class = exc.__class__
  433. evalue = self.op_direct_call(exdata.fn_pyexcclass2exc,
  434. self.heap.pyobjectptr(exc_class))
  435. etype = self.op_direct_call(exdata.fn_type_of_exc_inst, evalue)
  436. raise LLException(etype, evalue, *extraargs)
  437. def invoke_callable_with_pyexceptions(self, fptr, *args):
  438. obj = self.llinterpreter.typer.type_system.deref(fptr)
  439. try:
  440. return obj._callable(*args)
  441. except LLException, e:
  442. raise
  443. except Exception, e:
  444. if getattr(e, '_go_through_llinterp_uncaught_', False):
  445. raise
  446. if getattr(obj, '_debugexc', False):
  447. log.ERROR('The llinterpreter got an '
  448. 'unexpected exception when calling')
  449. log.ERROR('the external function %r:' % (fptr,))
  450. log.ERROR('%s: %s' % (e.__class__.__name__, e))
  451. if self.llinterpreter.tracer:
  452. self.llinterpreter.tracer.flush()
  453. import sys
  454. from pypy.translator.tool.pdbplus import PdbPlusShow
  455. PdbPlusShow(None).post_mortem(sys.exc_info()[2])
  456. self.make_llexception()
  457. def find_roots(self, roots):
  458. #log.findroots(self.curr_block.inputargs)
  459. vars = []
  460. for v in self.curr_block.inputargs:
  461. if isinstance(v, Variable):
  462. vars.append(v)
  463. for op in self.curr_block.operations[:self.curr_operation_index]:
  464. vars.append(op.result)
  465. for v in vars:
  466. TYPE = getattr(v, 'concretetype', None)
  467. if isinstance(TYPE, lltype.Ptr) and TYPE.TO._gckind == 'gc':
  468. roots.append(_address_of_local_var(self, v))
  469. # __________________________________________________________
  470. # misc LL operation implementations
  471. def op_debug_view(self, *ll_objects):
  472. from pypy.translator.tool.lltracker import track
  473. track(*ll_objects)
  474. def op_debug_assert(self, x, msg):
  475. assert x, msg
  476. def op_debug_fatalerror(self, ll_msg, ll_exc=None):
  477. msg = ''.join(ll_msg.chars)
  478. if ll_exc is None:
  479. raise LLFatalError(msg)
  480. else:
  481. ll_exc_type = lltype.cast_pointer(rclass.OBJECTPTR, ll_exc).typeptr
  482. raise LLFatalError(msg, LLException(ll_exc_type, ll_exc))
  483. def op_debug_llinterpcall(self, pythonfunction, *args_ll):
  484. try:
  485. return pythonfunction(*args_ll)
  486. except:
  487. self.make_llexception()
  488. def op_debug_start_traceback(self, *args):
  489. pass # xxx write debugging code here?
  490. def op_debug_reraise_traceback(self, *args):
  491. pass # xxx write debugging code here?
  492. def op_debug_record_traceback(self, *args):
  493. pass # xxx write debugging code here?
  494. def op_debug_print_traceback(self, *args):
  495. pass # xxx write debugging code here?
  496. def op_debug_catch_exception(self, *args):
  497. pass # xxx write debugging code here?
  498. def op_jit_marker(self, *args):
  499. pass
  500. def op_jit_record_known_class(self, *args):
  501. pass
  502. def op_get_exception_addr(self, *args):
  503. raise NotImplementedError
  504. def op_get_exc_value_addr(self, *args):
  505. raise NotImplementedError
  506. def op_instrument_count(self, ll_tag, ll_label):
  507. pass # xxx for now
  508. def op_keepalive(self, value):
  509. pass
  510. def op_hint(self, x, hints):
  511. return x
  512. def op_decode_arg(self, fname, i, name, vargs, vkwds):
  513. raise NotImplementedError("decode_arg")
  514. def op_decode_arg_def(self, fname, i, name, vargs, vkwds, default):
  515. raise NotImplementedError("decode_arg_def")
  516. def op_check_no_more_arg(self, fname, n, vargs):
  517. raise NotImplementedError("check_no_more_arg")
  518. def op_getslice(self, vargs, start, stop_should_be_None):
  519. raise NotImplementedError("getslice") # only for argument parsing
  520. def op_check_self_nonzero(self, fname, vself):
  521. raise NotImplementedError("check_self_nonzero")
  522. def op_setfield(self, obj, fieldname, fieldvalue):
  523. # obj should be pointer
  524. FIELDTYPE = getattr(lltype.typeOf(obj).TO, fieldname)
  525. if FIELDTYPE is not lltype.Void:
  526. self.heap.setfield(obj, fieldname, fieldvalue)
  527. def op_bare_setfield(self, obj, fieldname, fieldvalue):
  528. # obj should be pointer
  529. FIELDTYPE = getattr(lltype.typeOf(obj).TO, fieldname)
  530. if FIELDTYPE is not lltype.Void:
  531. setattr(obj, fieldname, fieldvalue)
  532. def op_getinteriorfield(self, obj, *offsets):
  533. checkptr(obj)
  534. ob = obj
  535. for o in offsets:
  536. if isinstance(o, str):
  537. ob = getattr(ob, o)
  538. else:
  539. ob = ob[o]
  540. assert not isinstance(ob, lltype._interior_ptr)
  541. return ob
  542. def getinneraddr(self, obj, *offsets):
  543. TYPE = lltype.typeOf(obj).TO
  544. addr = llmemory.cast_ptr_to_adr(obj)
  545. for o in offsets:
  546. if isinstance(o, str):
  547. addr += llmemory.offsetof(TYPE, o)
  548. TYPE = getattr(TYPE, o)
  549. else:
  550. addr += llmemory.itemoffsetof(TYPE, o)
  551. TYPE = TYPE.OF
  552. return addr, TYPE
  553. def op_setinteriorfield(self, obj, *fieldnamesval):
  554. offsets, fieldvalue = fieldnamesval[:-1], fieldnamesval[-1]
  555. inneraddr, FIELD = self.getinneraddr(obj, *offsets)
  556. if FIELD is not lltype.Void:
  557. self.heap.setinterior(obj, inneraddr, FIELD, fieldvalue, offsets)
  558. def op_bare_setinteriorfield(self, obj, *fieldnamesval):
  559. offsets, fieldvalue = fieldnamesval[:-1], fieldnamesval[-1]
  560. inneraddr, FIELD = self.getinneraddr(obj, *offsets)
  561. if FIELD is not lltype.Void:
  562. llheap.setinterior(obj, inneraddr, FIELD, fieldvalue)
  563. def op_getarrayitem(self, array, index):
  564. return array[index]
  565. def op_setarrayitem(self, array, index, item):
  566. # array should be a pointer
  567. ITEMTYPE = lltype.typeOf(array).TO.OF
  568. if ITEMTYPE is not lltype.Void:
  569. self.heap.setarrayitem(array, index, item)
  570. def op_bare_setarrayitem(self, array, index, item):
  571. # array should be a pointer
  572. ITEMTYPE = lltype.typeOf(array).TO.OF
  573. if ITEMTYPE is not lltype.Void:
  574. array[index] = item
  575. def perform_call(self, f, ARGS, args):
  576. fobj = self.llinterpreter.typer.type_system.deref(f)
  577. has_callable = getattr(fobj, '_callable', None) is not None
  578. if hasattr(fobj, 'graph'):
  579. graph = fobj.graph
  580. else:
  581. assert has_callable, "don't know how to execute %r" % f
  582. return self.invoke_callable_with_pyexceptions(f, *args)
  583. args_v = graph.getargs()
  584. if len(ARGS) != len(args_v):
  585. raise TypeError("graph with %d args called with wrong func ptr type: %r" %(len(args_v), ARGS))
  586. for T, v in zip(ARGS, args_v):
  587. if not lltype.isCompatibleType(T, v.concretetype):
  588. raise TypeError("graph with %r args called with wrong func ptr type: %r" %
  589. (tuple([v.concretetype for v in args_v]), ARGS))
  590. frame = self.newsubframe(graph, args)
  591. return frame.eval()
  592. def op_direct_call(self, f, *args):
  593. FTYPE = self.llinterpreter.typer.type_system.derefType(lltype.typeOf(f))
  594. return self.perform_call(f, FTYPE.ARGS, args)
  595. def op_indirect_call(self, f, *args):
  596. graphs = args[-1]
  597. args = args[:-1]
  598. if graphs is not None:
  599. obj = self.llinterpreter.typer.type_system.deref(f)
  600. if hasattr(obj, 'graph'):
  601. assert obj.graph in graphs
  602. else:
  603. pass
  604. #log.warn("op_indirect_call with graphs=None:", f)
  605. return self.op_direct_call(f, *args)
  606. def op_malloc(self, obj, flags):
  607. flavor = flags['flavor']
  608. zero = flags.get('zero', False)
  609. track_allocation = flags.get('track_allocation', True)
  610. if flavor == "stack":
  611. result = self.heap.malloc(obj, zero=zero, flavor='raw')
  612. self.alloca_objects.append(result)
  613. return result
  614. ptr = self.heap.malloc(obj, zero=zero, flavor=flavor,
  615. track_allocation=track_allocation)
  616. return ptr
  617. def op_malloc_varsize(self, obj, flags, size):
  618. flavor = flags['flavor']
  619. zero = flags.get('zero', False)
  620. track_allocation = flags.get('track_allocation', True)
  621. assert flavor in ('gc', 'raw')
  622. try:
  623. ptr = self.heap.malloc(obj, size, zero=zero, flavor=flavor,
  624. track_allocation=track_allocation)
  625. return ptr
  626. except MemoryError:
  627. self.make_llexception()
  628. def op_malloc_nonmovable(self, TYPE, flags):
  629. flavor = flags['flavor']
  630. assert flavor == 'gc'
  631. zero = flags.get('zero', False)
  632. return self.heap.malloc_nonmovable(TYPE, zero=zero)
  633. def op_malloc_nonmovable_varsize(self, TYPE, flags, size):
  634. flavor = flags['flavor']
  635. assert flavor == 'gc'
  636. zero = flags.get('zero', False)
  637. return self.heap.malloc_nonmovable(TYPE, size, zero=zero)
  638. def op_free(self, obj, flags):
  639. assert flags['flavor'] == 'raw'
  640. track_allocation = flags.get('track_allocation', True)
  641. self.heap.free(obj, flavor='raw', track_allocation=track_allocation)
  642. def op_gc_add_memory_pressure(self, size):
  643. self.heap.add_memory_pressure(size)
  644. def op_shrink_array(self, obj, smallersize):
  645. return self.heap.shrink_array(obj, smallersize)
  646. def op_zero_gc_pointers_inside(self, obj):
  647. raise NotImplementedError("zero_gc_pointers_inside")
  648. def op_gc_writebarrier_before_copy(self, source, dest,
  649. source_start, dest_start, length):
  650. if hasattr(self.heap, 'writebarrier_before_copy'):
  651. return self.heap.writebarrier_before_copy(source, dest,
  652. source_start, dest_start,
  653. length)
  654. else:
  655. return True
  656. def op_getfield(self, obj, field):
  657. checkptr(obj)
  658. # check the difference between op_getfield and op_getsubstruct:
  659. assert not isinstance(getattr(lltype.typeOf(obj).TO, field),
  660. lltype.ContainerType)
  661. return getattr(obj, field)
  662. def op_force_cast(self, RESTYPE, obj):
  663. from pypy.rpython.lltypesystem import ll2ctypes
  664. return ll2ctypes.force_cast(RESTYPE, obj)
  665. op_force_cast.need_result_type = True
  666. def op_cast_int_to_ptr(self, RESTYPE, int1):
  667. return lltype.cast_int_to_ptr(RESTYPE, int1)
  668. op_cast_int_to_ptr.need_result_type = True
  669. def op_cast_ptr_to_int(self, ptr1):
  670. checkptr(ptr1)
  671. return lltype.cast_ptr_to_int(ptr1)
  672. def op_cast_opaque_ptr(self, RESTYPE, obj):
  673. checkptr(obj)
  674. return lltype.cast_opaque_ptr(RESTYPE, obj)
  675. op_cast_opaque_ptr.need_result_type = True
  676. def op_cast_ptr_to_adr(self, ptr):
  677. checkptr(ptr)
  678. return llmemory.cast_ptr_to_adr(ptr)
  679. def op_cast_adr_to_int(self, adr, mode):
  680. checkadr(adr)
  681. return llmemory.cast_adr_to_int(adr, mode)
  682. def op_convert_float_bytes_to_longlong(self, f):
  683. from pypy.rlib import longlong2float
  684. return longlong2float.float2longlong(f)
  685. def op_weakref_create(self, v_obj):
  686. def objgetter(): # special support for gcwrapper.py
  687. return self.getval(v_obj)
  688. return self.heap.weakref_create_getlazy(objgetter)
  689. op_weakref_create.specialform = True
  690. def op_weakref_deref(self, PTRTYPE, obj):
  691. return self.heap.weakref_deref(PTRTYPE, obj)
  692. op_weakref_deref.need_result_type = True
  693. def op_cast_ptr_to_weakrefptr(self, obj):
  694. return llmemory.cast_ptr_to_weakrefptr(obj)
  695. def op_cast_weakrefptr_to_ptr(self, PTRTYPE, obj):
  696. return llmemory.cast_weakrefptr_to_ptr(PTRTYPE, obj)
  697. op_cast_weakrefptr_to_ptr.need_result_type = True
  698. def op_gc__collect(self, *gen):
  699. self.heap.collect(*gen)
  700. def op_gc_heap_stats(self):
  701. raise NotImplementedError
  702. def op_gc_obtain_free_space(self, size):
  703. raise NotImplementedError
  704. def op_gc_can_move(self, ptr):
  705. addr = llmemory.cast_ptr_to_adr(ptr)
  706. return self.heap.can_move(addr)
  707. def op_gc_thread_prepare(self):
  708. self.heap.thread_prepare()
  709. def op_gc_thread_run(self):
  710. self.heap.thread_run()
  711. def op_gc_thread_start(self):
  712. self.heap.thread_start()
  713. def op_gc_thread_die(self):
  714. self.heap.thread_die()
  715. def op_gc_thread_before_fork(self):
  716. raise NotImplementedError
  717. def op_gc_thread_after_fork(self):
  718. raise NotImplementedError
  719. def op_gc_free(self, addr):
  720. # what can you do?
  721. pass
  722. #raise NotImplementedError("gc_free")
  723. def op_gc_fetch_exception(self):
  724. raise NotImplementedError("gc_fetch_exception")
  725. def op_gc_restore_exception(self, exc):
  726. raise NotImplementedError("gc_restore_exception")
  727. def op_gc_adr_of_nursery_top(self):
  728. raise NotImplementedError
  729. def op_gc_adr_of_nursery_free(self):
  730. raise NotImplementedError
  731. def op_gc_adr_of_root_stack_base(self):
  732. raise NotImplementedError
  733. def op_gc_adr_of_root_stack_top(self):
  734. raise NotImplementedError
  735. def op_gc_call_rtti_destructor(self, rtti, addr):
  736. if hasattr(rtti._obj, 'destructor_funcptr'):
  737. d = rtti._obj.destructor_funcptr
  738. obptr = addr.ref()
  739. return self.op_direct_call(d, obptr)
  740. def op_gc_deallocate(self, TYPE, addr):
  741. raise NotImplementedError("gc_deallocate")
  742. def op_gc_push_alive_pyobj(self, pyobj):
  743. raise NotImplementedError("gc_push_alive_pyobj")
  744. def op_gc_pop_alive_pyobj(self, pyobj):
  745. raise NotImplementedError("gc_pop_alive_pyobj")
  746. def op_gc_reload_possibly_moved(self, v_newaddr, v_ptr):
  747. assert v_newaddr.concretetype is llmemory.Address
  748. assert isinstance(v_ptr.concretetype, lltype.Ptr)
  749. assert v_ptr.concretetype.TO._gckind == 'gc'
  750. newaddr = self.getval(v_newaddr)
  751. p = llmemory.cast_adr_to_ptr(newaddr, v_ptr.concretetype)
  752. if isinstance(v_ptr, Constant):
  753. assert v_ptr.value == p
  754. else:
  755. self.setvar(v_ptr, p)
  756. op_gc_reload_possibly_moved.specialform = True
  757. def op_gc_identityhash(self, obj):
  758. return lltype.identityhash(obj)
  759. def op_gc_id(self, ptr):
  760. PTR = lltype.typeOf(ptr)
  761. if isinstance(PTR, lltype.Ptr):
  762. return self.heap.gc_id(ptr)
  763. elif isinstance(PTR, ootype.OOType):
  764. return ootype.identityhash(ptr) # XXX imprecise
  765. raise NotImplementedError("gc_id on %r" % (PTR,))
  766. def op_gc_set_max_heap_size(self, maxsize):
  767. raise NotImplementedError("gc_set_max_heap_size")
  768. def op_gc_asmgcroot_static(self, index):
  769. raise NotImplementedError("gc_asmgcroot_static")
  770. def op_gc_stack_bottom(self):
  771. pass # marker for trackgcroot.py
  772. def op_gc_shadowstackref_new(self): # stacklet+shadowstack
  773. raise NotImplementedError("gc_shadowstackref_new")
  774. def op_gc_shadowstackref_context(self):
  775. raise NotImplementedError("gc_shadowstackref_context")
  776. def op_gc_shadowstackref_destroy(self):
  777. raise NotImplementedError("gc_shadowstackref_destroy")
  778. def op_gc_save_current_state_away(self):
  779. raise NotImplementedError("gc_save_current_state_away")
  780. def op_gc_forget_current_state(self):
  781. raise NotImplementedError("gc_forget_current_state")
  782. def op_gc_restore_state_from(self):
  783. raise NotImplementedError("gc_restore_state_from")
  784. def op_gc_start_fresh_new_state(self):
  785. raise NotImplementedError("gc_start_fresh_new_state")
  786. def op_gc_get_type_info_group(self):
  787. raise NotImplementedError("gc_get_type_info_group")
  788. def op_gc_get_rpy_memory_usage(self):
  789. raise NotImplementedError("gc_get_rpy_memory_usage")
  790. def op_gc_get_rpy_roots(self):
  791. raise NotImplementedError("gc_get_rpy_roots")
  792. def op_gc_get_rpy_referents(self):
  793. raise NotImplementedError("gc_get_rpy_referents")
  794. def op_gc_is_rpy_instance(self):
  795. raise NotImplementedError("gc_is_rpy_instance")
  796. def op_gc_get_rpy_type_index(self):
  797. raise NotImplementedError("gc_get_rpy_type_index")
  798. def op_gc_dump_rpy_heap(self):
  799. raise NotImplementedError("gc_dump_rpy_heap")
  800. def op_gc_typeids_z(self):
  801. raise NotImplementedError("gc_typeids_z")
  802. def op_do_malloc_fixedsize_clear(self):
  803. raise NotImplementedError("do_malloc_fixedsize_clear")
  804. def op_do_malloc_varsize_clear(self):
  805. raise NotImplementedError("do_malloc_varsize_clear")
  806. def op_get_write_barrier_failing_case(self):
  807. raise NotImplementedError("get_write_barrier_failing_case")
  808. def op_get_write_barrier_from_array_failing_case(self):
  809. raise NotImplementedError("get_write_barrier_from_array_failing_case")
  810. def op_stack_current(self):
  811. return 0
  812. # operations on pyobjects!
  813. for opname in lloperation.opimpls.keys():
  814. exec py.code.Source("""
  815. def op_%(opname)s(self, *pyobjs):
  816. for pyo in pyobjs:
  817. assert lltype.typeOf(pyo) == lltype.Ptr(lltype.PyObject)
  818. func = lloperation.opimpls[%(opname)r]
  819. try:
  820. pyo = func(*[pyo._obj.value for pyo in pyobjs])
  821. except Exception:
  822. self.make_llexception()
  823. return self.heap.pyobjectptr(pyo)
  824. """ % locals()).compile()
  825. del opname
  826. def op_simple_call(self, f, *args):
  827. assert lltype.typeOf(f) == lltype.Ptr(lltype.PyObject)
  828. for pyo in args:
  829. assert lltype.typeOf(pyo) == lltype.Ptr(lltype.PyObject)
  830. res = f._obj.value(*[pyo._obj.value for pyo in args])
  831. return self.heap.pyobjectptr(res)
  832. # __________________________________________________________
  833. # operations on addresses
  834. def op_raw_malloc(self, size):
  835. assert lltype.typeOf(size) == lltype.Signed
  836. return llmemory.raw_malloc(size)
  837. op_boehm_malloc = op_boehm_malloc_atomic = op_raw_malloc
  838. def op_boehm_register_finalizer(self, p, finalizer):
  839. pass
  840. def op_boehm_disappearing_link(self, link, obj):
  841. pass
  842. def op_raw_malloc_usage(self, size):
  843. assert lltype.typeOf(size) == lltype.Signed
  844. return llmemory.raw_malloc_usage(size)
  845. def op_raw_free(self, addr):
  846. checkadr(addr)
  847. llmemory.raw_free(addr)
  848. def op_raw_memclear(self, addr, size):
  849. checkadr(addr)
  850. llmemory.raw_memclear(addr, size)
  851. def op_raw_memcopy(self, fromaddr, toaddr, size):
  852. checkadr(fromaddr)
  853. checkadr(toaddr)
  854. llmemory.raw_memcopy(fromaddr, toaddr, size)
  855. op_raw_memmove = op_raw_memcopy # this is essentially the same here
  856. def op_raw_load(self, addr, typ, offset):
  857. checkadr(addr)
  858. value = getattr(addr, str(typ).lower())[offset]
  859. assert lltype.typeOf(value) == typ
  860. return value
  861. def op_raw_store(self, addr, typ, offset, value):
  862. checkadr(addr)
  863. assert lltype.typeOf(value) == typ
  864. getattr(addr, str(typ).lower())[offset] = value
  865. def op_stack_malloc(self, size): # mmh
  866. raise NotImplementedError("backend only")
  867. def op_track_alloc_start(self, addr):
  868. # we don't do tracking at this level
  869. checkadr(addr)
  870. def op_track_alloc_stop(self, addr):
  871. checkadr(addr)
  872. # ____________________________________________________________
  873. # Overflow-detecting variants
  874. def op_int_neg_ovf(self, x):
  875. assert is_valid_int(x)
  876. try:
  877. return ovfcheck(-x)
  878. except OverflowError:
  879. self.make_llexception()
  880. def op_int_abs_ovf(self, x):
  881. assert is_valid_int(x)
  882. try:
  883. return ovfcheck(abs(x))
  884. except OverflowError:
  885. self.make_llexception()
  886. def op_int_lshift_ovf(self, x, y):
  887. assert is_valid_int(x)
  888. assert is_valid_int(y)
  889. try:
  890. return ovfcheck(x << y)
  891. except OverflowError:
  892. self.make_llexception()
  893. def _makefunc2(fn, operator, xtype, ytype=None):
  894. import sys
  895. d = sys._getframe(1).f_locals
  896. if ytype is None:
  897. ytype = xtype
  898. if '_ovf' in fn:
  899. checkfn = 'ovfcheck'
  900. elif fn.startswith('op_int_'):
  901. checkfn = 'intmask'
  902. else:
  903. checkfn = ''
  904. if operator == '//':
  905. code = '''r = %(checkfn)s(x // y)
  906. if x^y < 0 and x%%y != 0:
  907. r += 1
  908. return r
  909. '''%locals()
  910. elif operator == '%':
  911. ## overflow check on % does not work with emulated int
  912. code = '''%(checkfn)s(x // y)
  913. r = x %% y
  914. if x^y < 0 and x%%y != 0:
  915. r -= y
  916. return r
  917. '''%locals()
  918. else:
  919. code = 'return %(checkfn)s(x %(operator)s y)'%locals()
  920. exec py.code.Source("""
  921. def %(fn)s(self, x, y):
  922. assert isinstance(x, %(xtype)s)
  923. assert isinstance(y, %(ytype)s)
  924. try:
  925. %(code)s
  926. except (OverflowError, ValueError, ZeroDivisionError):
  927. self.make_llexception()
  928. """ % locals()).compile() in globals(), d
  929. _makefunc2('op_int_add_ovf', '+', '(int, long, llmemory.AddressOffset)')
  930. _makefunc2('op_int_mul_ovf', '*', '(int, long, llmemory.AddressOffset)', '(int, long)')
  931. _makefunc2('op_int_sub_ovf', '-', '(int, long)')
  932. _makefunc2('op_int_floordiv_ovf', '//', '(int, long)') # XXX negative args
  933. _makefunc2('op_int_floordiv_zer', '//', '(int, long)') # can get off-by-one
  934. _makefunc2('op_int_floordiv_ovf_zer', '//', '(int, long)') # (see op_int_floordiv)
  935. _makefunc2('op_int_mod_ovf', '%', '(int, long)')
  936. _makefunc2('op_int_mod_zer', '%', '(int, long)')
  937. _makefunc2('op_int_mod_ovf_zer', '%', '(int, long)')
  938. _makefunc2('op_uint_floordiv_zer', '//', 'r_uint')
  939. _makefunc2('op_uint_mod_zer', '%', 'r_uint')
  940. _makefunc2('op_llong_floordiv_zer', '//', 'r_longlong')
  941. _makefunc2('op_llong_mod_zer', '%', 'r_longlong')
  942. _makefunc2('op_ullong_floordiv_zer', '//', 'r_ulonglong')
  943. _makefunc2('op_ullong_mod_zer', '%', 'r_ulonglong')
  944. def op_int_add_nonneg_ovf(self, x, y):
  945. if isinstance(y, int):
  946. assert y >= 0
  947. return self.op_int_add_ovf(x, y)
  948. def op_int_is_true(self, x):
  949. # special case
  950. if type(x) is CDefinedIntSymbolic:
  951. x = x.default
  952. # if type(x) is a subclass of Symbolic, bool(x) will usually raise
  953. # a TypeError -- unless __nonzero__ has been explicitly overridden.
  954. assert is_valid_int(x) or isinstance(x, Symbolic)
  955. return bool(x)
  956. # hack for jit.codegen.llgraph
  957. def op_check_and_clear_exc(self):
  958. exc_data = self.llinterpreter.get_transformed_exc_data(self.graph)
  959. assert exc_data
  960. etype = exc_data.exc_type
  961. evalue = exc_data.exc_value
  962. exc_data.exc_type = lltype.typeOf(etype )._defl()
  963. exc_data.exc_value = lltype.typeOf(evalue)._defl()
  964. return bool(etype)
  965. #Operation of ootype
  966. def op_new(self, INST):
  967. assert isinstance(INST, (ootype.Instance, ootype.BuiltinType))
  968. return ootype.new(INST)
  969. def op_oonewarray(self, ARRAY, length):
  970. assert isinstance(ARRAY, ootype.Array)
  971. assert is_valid_int(length)
  972. return ootype.oonewarray(ARRAY, length)
  973. def op_runtimenew(self, class_):
  974. return ootype.runtimenew(class_)
  975. def op_oonewcustomdict(self, DICT, eq_func, eq_obj, eq_method_name,
  976. hash_func, hash_obj, hash_method_name):
  977. eq_name, interp_eq = \
  978. wrap_callable(self.llinterpreter, eq_func, eq_obj, eq_method_name)
  979. EQ_FUNC = ootype.StaticMethod([DICT._KEYTYPE, DICT._KEYTYPE], ootype.Bool)
  980. sm_eq = ootype.static_meth(EQ_FUNC, eq_name, _callable=interp_eq)
  981. hash_name, interp_hash = \
  982. wrap_callable(self.llinterpreter, hash_func, hash_obj, hash_method_name)
  983. HASH_FUNC = ootype.StaticMethod([DICT._KEYTYPE], ootype.Signed)
  984. sm_hash = ootype.static_meth(HASH_FUNC, hash_name, _callable=interp_hash)
  985. # XXX: is it fine to have StaticMethod type for bound methods, too?
  986. return ootype.oonewcustomdict(DICT, sm_eq, sm_hash)
  987. def op_oosetfield(self, inst, name, value):
  988. checkinst(inst)
  989. assert isinstance(name, str)
  990. FIELDTYPE = lltype.typeOf(inst)._field_type(name)
  991. if FIELDTYPE is not lltype.Void:
  992. setattr(inst, name, value)
  993. def op_oogetfield(self, inst, name):
  994. checkinst(inst)
  995. assert isinstance(name, str)
  996. return getattr(inst, name)
  997. def op_oosend(self, message, inst, *args):
  998. checkinst(inst)
  999. assert isinstance(message, str)
  1000. bm = getattr(inst, message)
  1001. inst = bm.inst
  1002. m = bm.meth
  1003. args = m._checkargs(args, check_callable=False)
  1004. if getattr(m, 'abstract', False):
  1005. raise RuntimeError("calling abstract method %r" % (m,))
  1006. return self.perform_call(m, (lltype.typeOf(inst),)+lltype.typeOf(m).ARGS, [inst]+args)
  1007. def op_oostring(self, obj, base):
  1008. return ootype.oostring(obj, base)
  1009. def op_oounicode(self, obj, base):
  1010. try:
  1011. return ootype.oounicode(obj, base)
  1012. except UnicodeDecodeError:
  1013. self.make_llexception()
  1014. def op_ooparse_int(self, s, base):
  1015. try:
  1016. return ootype.ooparse_int(s, base)
  1017. except ValueError:
  1018. self.make_llexception()
  1019. def op_ooparse_float(self, s):
  1020. try:
  1021. return ootype.ooparse_float(s)
  1022. except ValueError:
  1023. self.make_llexception()
  1024. def op_oobox_int(self, i):
  1025. return ootype.oobox_int(i)
  1026. def op_oounbox_int(self, x):
  1027. return ootype.oounbox_int(x)
  1028. class Tracer(object):
  1029. Counter = 0
  1030. file = None
  1031. TRACE = int(os.getenv('PYPY_TRACE') or '0')
  1032. HEADER = """<html><head>
  1033. <script language=javascript type='text/javascript'>
  1034. function togglestate(n) {
  1035. var item = document.getElementById('div'+n)
  1036. if (item.style.display == 'none')
  1037. item.style.display = 'block';
  1038. else
  1039. item.style.display = 'none';
  1040. }
  1041. function toggleall(lst) {
  1042. for (var i = 0; i<lst.length; i++) {
  1043. togglestate(lst[i]);
  1044. }
  1045. }
  1046. </script>
  1047. </head>
  1048. <body><pre>
  1049. """
  1050. FOOTER = """</pre>
  1051. <script language=javascript type='text/javascript'>
  1052. toggleall(%r);
  1053. </script>
  1054. </body></html>"""
  1055. ENTER = ('''\n\t<a href="javascript:togglestate(%d)">%s</a>'''
  1056. '''\n<div id="div%d" style="display: %s">\t''')
  1057. LEAVE = '''\n</div>\t'''
  1058. def htmlquote(self, s, text_to_html={}):
  1059. # HTML quoting, lazily initialized
  1060. if not text_to_html:
  1061. import htmlentitydefs
  1062. for key, value in htmlentitydefs.entitydefs.items():
  1063. text_to_html[value] = '&' + key + ';'
  1064. return ''.join([text_to_html.get(c, c) for c in s])
  1065. def start(self):
  1066. # start of a dump file
  1067. if not self.TRACE:
  1068. return
  1069. from pypy.tool.udir import udir
  1070. n = Tracer.Counter
  1071. Tracer.Counter += 1
  1072. filename = 'llinterp_trace_%d.html' % n
  1073. self.file = udir.join(filename).open('w')
  1074. print >> self.file, self.HEADER
  1075. linkname = str(udir.join('llinterp_trace.html'))
  1076. try:
  1077. os.unlink(linkname)
  1078. except OSError:
  1079. pass
  1080. try:
  1081. os.symlink(filename, linkname)
  1082. except (AttributeError, OSError):
  1083. pass
  1084. self.count = 0
  1085. self.indentation = ''
  1086. self.depth = 0
  1087. self.latest_call_chain = []
  1088. def stop(self):
  1089. # end of a dump file
  1090. if self.file:
  1091. print >> self.file, self.FOOTER % (self.latest_call_chain[1:])
  1092. self.file.close()
  1093. self.file = None
  1094. def enter(self, graph):
  1095. # enter evaluation of a graph
  1096. if self.file:
  1097. del self.latest_call_chain[self.depth:]
  1098. self.depth += 1
  1099. self.latest_call_chain.append(self.count)
  1100. s = self.htmlquote(str(graph))
  1101. i = s.rfind(')')
  1102. s = s[:i+1] + '<b>' + s[i+1:] + '</b>'
  1103. if self.count == 0:
  1104. display = 'block'
  1105. else:
  1106. display = 'none'
  1107. text = self.ENTER % (self.count, s, self.count, display)
  1108. self.indentation += ' '
  1109. self.file.write(text.replace('\t', self.indentation))
  1110. self.count += 1
  1111. def leave(self):
  1112. # leave evaluation of a graph
  1113. if self.file:
  1114. self.indentation = self.indentation[:-4]
  1115. self.file.write(self.LEAVE.replace('\t', self.indentation))
  1116. self.depth -= 1
  1117. def dump(self, text, bold=False):
  1118. if self.file:
  1119. text = self.htmlquote(text)
  1120. if bold:
  1121. text = '<b>%s</b>' % (text,)
  1122. self.file.write(text.replace('\n', '\n'+self.indentation))
  1123. def flush(self):
  1124. if self.file:
  1125. self.file.flush()
  1126. def wrap_callable(llinterpreter, fn, obj, method_name):
  1127. if method_name is None:
  1128. # fn is a StaticMethod
  1129. if obj is not None:
  1130. self_arg = [obj]
  1131. else:
  1132. self_arg = []
  1133. func_graph = fn.graph
  1134. else:
  1135. # obj is an instance, we want to call 'method_name' on it
  1136. assert fn is None
  1137. self_arg = [obj]
  1138. func_graph = obj._TYPE._methods[method_name._str].graph
  1139. return wrap_graph(llinterpreter, func_graph, self_arg)
  1140. def wrap_graph(llinterpreter, graph, self_arg):
  1141. """
  1142. Returns a callable that inteprets the given func or method_name when called.
  1143. """
  1144. def interp_func(*args):
  1145. graph_args = self_arg + list(args)
  1146. return llinterpreter.eval_graph(graph, args=graph_args)
  1147. interp_func.graph = graph
  1148. interp_func.self_arg = self_arg
  1149. return graph.name, interp_func
  1150. def enumerate_exceptions_top_down():
  1151. import exceptions
  1152. result = []
  1153. seen = {}
  1154. def addcls(cls):
  1155. if (type(cls) is type(Exception) and
  1156. issubclass(cls, py.builtin.BaseException)):
  1157. if cls in seen:
  1158. return
  1159. for base in cls.__bases__: # bases first
  1160. addcls(base)
  1161. resul

Large files files are truncated, but you can click here to view the full file