Mercurial > scratch
annotate pydershell/pydershell.py @ 18:69e5523ebdc6
Separated out SafeJsObjectWrapper into itself and a new subclass, SafeJsFunctionWrapper, so that callable() works properly on instances.
author | Atul Varma <varmaa@toolness.com> |
---|---|
date | Mon, 07 Sep 2009 04:38:44 -0700 |
parents | 1d62177c5c27 |
children | 057260102960 |
rev | line source |
---|---|
11 | 1 #! /usr/bin/env python |
2 | |
10 | 3 import sys |
4 import time | |
5 import threading | |
6 import traceback | |
12
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
7 import weakref |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
8 |
10 | 9 import pydermonkey |
10 | |
12
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
11 class ContextWatchdogThread(threading.Thread): |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
12 """ |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
13 Watches active JS contexts and triggers their operation callbacks |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
14 at a regular interval. |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
15 """ |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
16 |
13 | 17 DEFAULT_INTERVAL = 0.25 |
18 | |
19 def __init__(self, interval=DEFAULT_INTERVAL): | |
12
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
20 threading.Thread.__init__(self) |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
21 self._lock = threading.Lock() |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
22 self._stop = threading.Event() |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
23 self._contexts = [] |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
24 self.interval = interval |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
25 self.setDaemon(True) |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
26 |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
27 def add_context(self, cx): |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
28 self._lock.acquire() |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
29 try: |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
30 self._contexts.append(weakref.ref(cx)) |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
31 finally: |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
32 self._lock.release() |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
33 |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
34 def join(self): |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
35 self._stop.set() |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
36 threading.Thread.join(self) |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
37 |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
38 def run(self): |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
39 while not self._stop.isSet(): |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
40 time.sleep(self.interval) |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
41 new_list = [] |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
42 self._lock.acquire() |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
43 try: |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
44 for weakcx in self._contexts: |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
45 cx = weakcx() |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
46 if cx: |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
47 new_list.append(weakcx) |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
48 cx.trigger_operation_callback() |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
49 self._contexts = new_list |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
50 finally: |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
51 self._lock.release() |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
52 |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
53 # Create a global watchdog. |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
54 watchdog = ContextWatchdogThread() |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
55 watchdog.start() |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
56 |
11 | 57 class InternalError(BaseException): |
12
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
58 """ |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
59 Represents an error in a JS-wrapped Python function that wasn't |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
60 expected to happen; because it's derived from BaseException, it |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
61 unrolls the whole JS/Python stack so that the error can be |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
62 reported to the outermost calling code. |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
63 """ |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
64 |
11 | 65 def __init__(self): |
66 BaseException.__init__(self) | |
67 self.exc_info = sys.exc_info() | |
68 | |
17
1d62177c5c27
Added a safe js object wrapper.
Atul Varma <varmaa@toolness.com>
parents:
16
diff
changeset
|
69 class SafeJsObjectWrapper(object): |
1d62177c5c27
Added a safe js object wrapper.
Atul Varma <varmaa@toolness.com>
parents:
16
diff
changeset
|
70 """ |
1d62177c5c27
Added a safe js object wrapper.
Atul Varma <varmaa@toolness.com>
parents:
16
diff
changeset
|
71 Securely wraps a JS object to behave like any normal Python object. |
1d62177c5c27
Added a safe js object wrapper.
Atul Varma <varmaa@toolness.com>
parents:
16
diff
changeset
|
72 """ |
1d62177c5c27
Added a safe js object wrapper.
Atul Varma <varmaa@toolness.com>
parents:
16
diff
changeset
|
73 |
1d62177c5c27
Added a safe js object wrapper.
Atul Varma <varmaa@toolness.com>
parents:
16
diff
changeset
|
74 __slots__ = ['_jsobject', '_cx', '_this'] |
1d62177c5c27
Added a safe js object wrapper.
Atul Varma <varmaa@toolness.com>
parents:
16
diff
changeset
|
75 |
1d62177c5c27
Added a safe js object wrapper.
Atul Varma <varmaa@toolness.com>
parents:
16
diff
changeset
|
76 def __init__(self, cx, jsobject, this): |
1d62177c5c27
Added a safe js object wrapper.
Atul Varma <varmaa@toolness.com>
parents:
16
diff
changeset
|
77 if not isinstance(jsobject, pydermonkey.Object): |
1d62177c5c27
Added a safe js object wrapper.
Atul Varma <varmaa@toolness.com>
parents:
16
diff
changeset
|
78 raise TypeError("Cannot wrap '%s' object" % |
1d62177c5c27
Added a safe js object wrapper.
Atul Varma <varmaa@toolness.com>
parents:
16
diff
changeset
|
79 type(jsobject).__name__) |
1d62177c5c27
Added a safe js object wrapper.
Atul Varma <varmaa@toolness.com>
parents:
16
diff
changeset
|
80 object.__setattr__(self, '_jsobject', jsobject) |
1d62177c5c27
Added a safe js object wrapper.
Atul Varma <varmaa@toolness.com>
parents:
16
diff
changeset
|
81 object.__setattr__(self, '_cx', cx) |
1d62177c5c27
Added a safe js object wrapper.
Atul Varma <varmaa@toolness.com>
parents:
16
diff
changeset
|
82 object.__setattr__(self, '_this', this) |
1d62177c5c27
Added a safe js object wrapper.
Atul Varma <varmaa@toolness.com>
parents:
16
diff
changeset
|
83 |
18
69e5523ebdc6
Separated out SafeJsObjectWrapper into itself and a new subclass, SafeJsFunctionWrapper, so that callable() works properly on instances.
Atul Varma <varmaa@toolness.com>
parents:
17
diff
changeset
|
84 @staticmethod |
69e5523ebdc6
Separated out SafeJsObjectWrapper into itself and a new subclass, SafeJsFunctionWrapper, so that callable() works properly on instances.
Atul Varma <varmaa@toolness.com>
parents:
17
diff
changeset
|
85 def wrap(cx, jsvalue, this): |
69e5523ebdc6
Separated out SafeJsObjectWrapper into itself and a new subclass, SafeJsFunctionWrapper, so that callable() works properly on instances.
Atul Varma <varmaa@toolness.com>
parents:
17
diff
changeset
|
86 if isinstance(jsvalue, pydermonkey.Function): |
69e5523ebdc6
Separated out SafeJsObjectWrapper into itself and a new subclass, SafeJsFunctionWrapper, so that callable() works properly on instances.
Atul Varma <varmaa@toolness.com>
parents:
17
diff
changeset
|
87 return SafeJsFunctionWrapper(cx, jsvalue, this) |
69e5523ebdc6
Separated out SafeJsObjectWrapper into itself and a new subclass, SafeJsFunctionWrapper, so that callable() works properly on instances.
Atul Varma <varmaa@toolness.com>
parents:
17
diff
changeset
|
88 elif isinstance(jsvalue, pydermonkey.Object): |
69e5523ebdc6
Separated out SafeJsObjectWrapper into itself and a new subclass, SafeJsFunctionWrapper, so that callable() works properly on instances.
Atul Varma <varmaa@toolness.com>
parents:
17
diff
changeset
|
89 return SafeJsObjectWrapper(cx, jsvalue, this) |
17
1d62177c5c27
Added a safe js object wrapper.
Atul Varma <varmaa@toolness.com>
parents:
16
diff
changeset
|
90 else: |
18
69e5523ebdc6
Separated out SafeJsObjectWrapper into itself and a new subclass, SafeJsFunctionWrapper, so that callable() works properly on instances.
Atul Varma <varmaa@toolness.com>
parents:
17
diff
changeset
|
91 # It's a primitive value. |
69e5523ebdc6
Separated out SafeJsObjectWrapper into itself and a new subclass, SafeJsFunctionWrapper, so that callable() works properly on instances.
Atul Varma <varmaa@toolness.com>
parents:
17
diff
changeset
|
92 return jsvalue |
17
1d62177c5c27
Added a safe js object wrapper.
Atul Varma <varmaa@toolness.com>
parents:
16
diff
changeset
|
93 |
18
69e5523ebdc6
Separated out SafeJsObjectWrapper into itself and a new subclass, SafeJsFunctionWrapper, so that callable() works properly on instances.
Atul Varma <varmaa@toolness.com>
parents:
17
diff
changeset
|
94 def _wrap_to_python(self, jsvalue): |
69e5523ebdc6
Separated out SafeJsObjectWrapper into itself and a new subclass, SafeJsFunctionWrapper, so that callable() works properly on instances.
Atul Varma <varmaa@toolness.com>
parents:
17
diff
changeset
|
95 return self.wrap(self._cx, jsvalue, self._jsobject) |
69e5523ebdc6
Separated out SafeJsObjectWrapper into itself and a new subclass, SafeJsFunctionWrapper, so that callable() works properly on instances.
Atul Varma <varmaa@toolness.com>
parents:
17
diff
changeset
|
96 |
69e5523ebdc6
Separated out SafeJsObjectWrapper into itself and a new subclass, SafeJsFunctionWrapper, so that callable() works properly on instances.
Atul Varma <varmaa@toolness.com>
parents:
17
diff
changeset
|
97 def _wrap_to_js(self, value): |
17
1d62177c5c27
Added a safe js object wrapper.
Atul Varma <varmaa@toolness.com>
parents:
16
diff
changeset
|
98 # TODO: Add support for wrapping non-primitive python objects. |
1d62177c5c27
Added a safe js object wrapper.
Atul Varma <varmaa@toolness.com>
parents:
16
diff
changeset
|
99 return value |
1d62177c5c27
Added a safe js object wrapper.
Atul Varma <varmaa@toolness.com>
parents:
16
diff
changeset
|
100 |
1d62177c5c27
Added a safe js object wrapper.
Atul Varma <varmaa@toolness.com>
parents:
16
diff
changeset
|
101 def __eq__(self, other): |
1d62177c5c27
Added a safe js object wrapper.
Atul Varma <varmaa@toolness.com>
parents:
16
diff
changeset
|
102 if isinstance(other, SafeJsObjectWrapper): |
1d62177c5c27
Added a safe js object wrapper.
Atul Varma <varmaa@toolness.com>
parents:
16
diff
changeset
|
103 return self._jsobject == other._jsobject |
1d62177c5c27
Added a safe js object wrapper.
Atul Varma <varmaa@toolness.com>
parents:
16
diff
changeset
|
104 else: |
1d62177c5c27
Added a safe js object wrapper.
Atul Varma <varmaa@toolness.com>
parents:
16
diff
changeset
|
105 return False |
1d62177c5c27
Added a safe js object wrapper.
Atul Varma <varmaa@toolness.com>
parents:
16
diff
changeset
|
106 |
1d62177c5c27
Added a safe js object wrapper.
Atul Varma <varmaa@toolness.com>
parents:
16
diff
changeset
|
107 def __str__(self): |
1d62177c5c27
Added a safe js object wrapper.
Atul Varma <varmaa@toolness.com>
parents:
16
diff
changeset
|
108 return self.toString() |
1d62177c5c27
Added a safe js object wrapper.
Atul Varma <varmaa@toolness.com>
parents:
16
diff
changeset
|
109 |
1d62177c5c27
Added a safe js object wrapper.
Atul Varma <varmaa@toolness.com>
parents:
16
diff
changeset
|
110 def __unicode__(self): |
1d62177c5c27
Added a safe js object wrapper.
Atul Varma <varmaa@toolness.com>
parents:
16
diff
changeset
|
111 return self.toString() |
1d62177c5c27
Added a safe js object wrapper.
Atul Varma <varmaa@toolness.com>
parents:
16
diff
changeset
|
112 |
1d62177c5c27
Added a safe js object wrapper.
Atul Varma <varmaa@toolness.com>
parents:
16
diff
changeset
|
113 def __setitem__(self, item, value): |
1d62177c5c27
Added a safe js object wrapper.
Atul Varma <varmaa@toolness.com>
parents:
16
diff
changeset
|
114 self.__setattr__(item, value) |
1d62177c5c27
Added a safe js object wrapper.
Atul Varma <varmaa@toolness.com>
parents:
16
diff
changeset
|
115 |
1d62177c5c27
Added a safe js object wrapper.
Atul Varma <varmaa@toolness.com>
parents:
16
diff
changeset
|
116 def __setattr__(self, name, value): |
1d62177c5c27
Added a safe js object wrapper.
Atul Varma <varmaa@toolness.com>
parents:
16
diff
changeset
|
117 cx = self._cx |
1d62177c5c27
Added a safe js object wrapper.
Atul Varma <varmaa@toolness.com>
parents:
16
diff
changeset
|
118 jsobject = self._jsobject |
1d62177c5c27
Added a safe js object wrapper.
Atul Varma <varmaa@toolness.com>
parents:
16
diff
changeset
|
119 |
1d62177c5c27
Added a safe js object wrapper.
Atul Varma <varmaa@toolness.com>
parents:
16
diff
changeset
|
120 cx.define_property(jsobject, name, |
18
69e5523ebdc6
Separated out SafeJsObjectWrapper into itself and a new subclass, SafeJsFunctionWrapper, so that callable() works properly on instances.
Atul Varma <varmaa@toolness.com>
parents:
17
diff
changeset
|
121 self._wrap_to_js(value)) |
17
1d62177c5c27
Added a safe js object wrapper.
Atul Varma <varmaa@toolness.com>
parents:
16
diff
changeset
|
122 |
1d62177c5c27
Added a safe js object wrapper.
Atul Varma <varmaa@toolness.com>
parents:
16
diff
changeset
|
123 def __getitem__(self, item): |
1d62177c5c27
Added a safe js object wrapper.
Atul Varma <varmaa@toolness.com>
parents:
16
diff
changeset
|
124 return self.__getattr__(item) |
1d62177c5c27
Added a safe js object wrapper.
Atul Varma <varmaa@toolness.com>
parents:
16
diff
changeset
|
125 |
1d62177c5c27
Added a safe js object wrapper.
Atul Varma <varmaa@toolness.com>
parents:
16
diff
changeset
|
126 def __getattr__(self, name): |
1d62177c5c27
Added a safe js object wrapper.
Atul Varma <varmaa@toolness.com>
parents:
16
diff
changeset
|
127 cx = self._cx |
1d62177c5c27
Added a safe js object wrapper.
Atul Varma <varmaa@toolness.com>
parents:
16
diff
changeset
|
128 jsobject = self._jsobject |
1d62177c5c27
Added a safe js object wrapper.
Atul Varma <varmaa@toolness.com>
parents:
16
diff
changeset
|
129 |
18
69e5523ebdc6
Separated out SafeJsObjectWrapper into itself and a new subclass, SafeJsFunctionWrapper, so that callable() works properly on instances.
Atul Varma <varmaa@toolness.com>
parents:
17
diff
changeset
|
130 return self._wrap_to_python(cx.get_property(jsobject, name)) |
17
1d62177c5c27
Added a safe js object wrapper.
Atul Varma <varmaa@toolness.com>
parents:
16
diff
changeset
|
131 |
1d62177c5c27
Added a safe js object wrapper.
Atul Varma <varmaa@toolness.com>
parents:
16
diff
changeset
|
132 def __contains__(self, item): |
1d62177c5c27
Added a safe js object wrapper.
Atul Varma <varmaa@toolness.com>
parents:
16
diff
changeset
|
133 cx = self._cx |
1d62177c5c27
Added a safe js object wrapper.
Atul Varma <varmaa@toolness.com>
parents:
16
diff
changeset
|
134 jsobject = self._jsobject |
1d62177c5c27
Added a safe js object wrapper.
Atul Varma <varmaa@toolness.com>
parents:
16
diff
changeset
|
135 |
1d62177c5c27
Added a safe js object wrapper.
Atul Varma <varmaa@toolness.com>
parents:
16
diff
changeset
|
136 return cx.has_property(jsobject, item) |
1d62177c5c27
Added a safe js object wrapper.
Atul Varma <varmaa@toolness.com>
parents:
16
diff
changeset
|
137 |
1d62177c5c27
Added a safe js object wrapper.
Atul Varma <varmaa@toolness.com>
parents:
16
diff
changeset
|
138 def __iter__(self): |
1d62177c5c27
Added a safe js object wrapper.
Atul Varma <varmaa@toolness.com>
parents:
16
diff
changeset
|
139 cx = self._cx |
1d62177c5c27
Added a safe js object wrapper.
Atul Varma <varmaa@toolness.com>
parents:
16
diff
changeset
|
140 jsobject = self._jsobject |
1d62177c5c27
Added a safe js object wrapper.
Atul Varma <varmaa@toolness.com>
parents:
16
diff
changeset
|
141 |
1d62177c5c27
Added a safe js object wrapper.
Atul Varma <varmaa@toolness.com>
parents:
16
diff
changeset
|
142 properties = cx.enumerate(jsobject) |
1d62177c5c27
Added a safe js object wrapper.
Atul Varma <varmaa@toolness.com>
parents:
16
diff
changeset
|
143 for property in properties: |
1d62177c5c27
Added a safe js object wrapper.
Atul Varma <varmaa@toolness.com>
parents:
16
diff
changeset
|
144 yield property |
1d62177c5c27
Added a safe js object wrapper.
Atul Varma <varmaa@toolness.com>
parents:
16
diff
changeset
|
145 |
18
69e5523ebdc6
Separated out SafeJsObjectWrapper into itself and a new subclass, SafeJsFunctionWrapper, so that callable() works properly on instances.
Atul Varma <varmaa@toolness.com>
parents:
17
diff
changeset
|
146 class SafeJsFunctionWrapper(SafeJsObjectWrapper): |
69e5523ebdc6
Separated out SafeJsObjectWrapper into itself and a new subclass, SafeJsFunctionWrapper, so that callable() works properly on instances.
Atul Varma <varmaa@toolness.com>
parents:
17
diff
changeset
|
147 """ |
69e5523ebdc6
Separated out SafeJsObjectWrapper into itself and a new subclass, SafeJsFunctionWrapper, so that callable() works properly on instances.
Atul Varma <varmaa@toolness.com>
parents:
17
diff
changeset
|
148 Securely wraps a JS function to behave like any normal Python object. |
69e5523ebdc6
Separated out SafeJsObjectWrapper into itself and a new subclass, SafeJsFunctionWrapper, so that callable() works properly on instances.
Atul Varma <varmaa@toolness.com>
parents:
17
diff
changeset
|
149 """ |
69e5523ebdc6
Separated out SafeJsObjectWrapper into itself and a new subclass, SafeJsFunctionWrapper, so that callable() works properly on instances.
Atul Varma <varmaa@toolness.com>
parents:
17
diff
changeset
|
150 |
69e5523ebdc6
Separated out SafeJsObjectWrapper into itself and a new subclass, SafeJsFunctionWrapper, so that callable() works properly on instances.
Atul Varma <varmaa@toolness.com>
parents:
17
diff
changeset
|
151 def __init__(self, cx, jsfunction, this): |
69e5523ebdc6
Separated out SafeJsObjectWrapper into itself and a new subclass, SafeJsFunctionWrapper, so that callable() works properly on instances.
Atul Varma <varmaa@toolness.com>
parents:
17
diff
changeset
|
152 if not isinstance(jsfunction, pydermonkey.Function): |
69e5523ebdc6
Separated out SafeJsObjectWrapper into itself and a new subclass, SafeJsFunctionWrapper, so that callable() works properly on instances.
Atul Varma <varmaa@toolness.com>
parents:
17
diff
changeset
|
153 raise TypeError("Cannot wrap '%s' object" % |
69e5523ebdc6
Separated out SafeJsObjectWrapper into itself and a new subclass, SafeJsFunctionWrapper, so that callable() works properly on instances.
Atul Varma <varmaa@toolness.com>
parents:
17
diff
changeset
|
154 type(jsobject).__name__) |
69e5523ebdc6
Separated out SafeJsObjectWrapper into itself and a new subclass, SafeJsFunctionWrapper, so that callable() works properly on instances.
Atul Varma <varmaa@toolness.com>
parents:
17
diff
changeset
|
155 SafeJsObjectWrapper.__init__(self, cx, jsfunction, this) |
69e5523ebdc6
Separated out SafeJsObjectWrapper into itself and a new subclass, SafeJsFunctionWrapper, so that callable() works properly on instances.
Atul Varma <varmaa@toolness.com>
parents:
17
diff
changeset
|
156 |
69e5523ebdc6
Separated out SafeJsObjectWrapper into itself and a new subclass, SafeJsFunctionWrapper, so that callable() works properly on instances.
Atul Varma <varmaa@toolness.com>
parents:
17
diff
changeset
|
157 def __call__(self, *args): |
69e5523ebdc6
Separated out SafeJsObjectWrapper into itself and a new subclass, SafeJsFunctionWrapper, so that callable() works properly on instances.
Atul Varma <varmaa@toolness.com>
parents:
17
diff
changeset
|
158 cx = self._cx |
69e5523ebdc6
Separated out SafeJsObjectWrapper into itself and a new subclass, SafeJsFunctionWrapper, so that callable() works properly on instances.
Atul Varma <varmaa@toolness.com>
parents:
17
diff
changeset
|
159 jsobject = self._jsobject |
69e5523ebdc6
Separated out SafeJsObjectWrapper into itself and a new subclass, SafeJsFunctionWrapper, so that callable() works properly on instances.
Atul Varma <varmaa@toolness.com>
parents:
17
diff
changeset
|
160 this = self._this |
69e5523ebdc6
Separated out SafeJsObjectWrapper into itself and a new subclass, SafeJsFunctionWrapper, so that callable() works properly on instances.
Atul Varma <varmaa@toolness.com>
parents:
17
diff
changeset
|
161 |
69e5523ebdc6
Separated out SafeJsObjectWrapper into itself and a new subclass, SafeJsFunctionWrapper, so that callable() works properly on instances.
Atul Varma <varmaa@toolness.com>
parents:
17
diff
changeset
|
162 obj = cx.call_function(this, jsobject, |
69e5523ebdc6
Separated out SafeJsObjectWrapper into itself and a new subclass, SafeJsFunctionWrapper, so that callable() works properly on instances.
Atul Varma <varmaa@toolness.com>
parents:
17
diff
changeset
|
163 self._wrap_to_js(args)) |
69e5523ebdc6
Separated out SafeJsObjectWrapper into itself and a new subclass, SafeJsFunctionWrapper, so that callable() works properly on instances.
Atul Varma <varmaa@toolness.com>
parents:
17
diff
changeset
|
164 return self._wrap_to_python(obj) |
69e5523ebdc6
Separated out SafeJsObjectWrapper into itself and a new subclass, SafeJsFunctionWrapper, so that callable() works properly on instances.
Atul Varma <varmaa@toolness.com>
parents:
17
diff
changeset
|
165 |
12
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
166 def safejsfunc(cx, on_obj, name=None): |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
167 """ |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
168 Exposes the decorated Python function on the given JS object. |
10 | 169 |
12
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
170 Any unexpected exceptions raised by the function will be |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
171 re-raised as InternalError exceptions. |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
172 """ |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
173 |
11 | 174 def make_wrapper(func): |
175 if name is None: | |
176 func_name = func.__name__ | |
177 else: | |
178 func_name = name | |
179 def wrapper(func_cx, this, args): | |
180 try: | |
181 return func(func_cx, this, args) | |
182 except pydermonkey.error: | |
183 raise | |
184 except Exception: | |
185 raise InternalError() | |
186 cx.define_property( | |
187 on_obj, | |
188 func_name, | |
189 cx.new_function(wrapper, func_name) | |
190 ) | |
191 return func | |
192 return make_wrapper | |
193 | |
12
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
194 def format_stack(js_stack): |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
195 """ |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
196 Returns a formatted Python-esque stack traceback of the given |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
197 JS stack. |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
198 """ |
10 | 199 |
12
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
200 STACK_LINE =" File \"%(filename)s\", line %(lineno)d, in %(name)s" |
10 | 201 |
202 lines = [] | |
203 while js_stack: | |
12
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
204 script = js_stack['script'] |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
205 function = js_stack['function'] |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
206 if script: |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
207 frameinfo = dict(filename = script.filename, |
16
a78570a423ea
Show code excerpts in js stack tracebacks
Atul Varma <varmaa@toolness.com>
parents:
15
diff
changeset
|
208 lineno = js_stack['lineno'], |
a78570a423ea
Show code excerpts in js stack tracebacks
Atul Varma <varmaa@toolness.com>
parents:
15
diff
changeset
|
209 name = '<module>') |
12
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
210 elif function and not function.is_python: |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
211 frameinfo = dict(filename = function.filename, |
16
a78570a423ea
Show code excerpts in js stack tracebacks
Atul Varma <varmaa@toolness.com>
parents:
15
diff
changeset
|
212 lineno = js_stack['lineno'], |
a78570a423ea
Show code excerpts in js stack tracebacks
Atul Varma <varmaa@toolness.com>
parents:
15
diff
changeset
|
213 name = function.name) |
10 | 214 else: |
12
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
215 frameinfo = None |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
216 if frameinfo: |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
217 lines.insert(0, STACK_LINE % frameinfo) |
16
a78570a423ea
Show code excerpts in js stack tracebacks
Atul Varma <varmaa@toolness.com>
parents:
15
diff
changeset
|
218 try: |
a78570a423ea
Show code excerpts in js stack tracebacks
Atul Varma <varmaa@toolness.com>
parents:
15
diff
changeset
|
219 filelines = open(frameinfo['filename']).readlines() |
a78570a423ea
Show code excerpts in js stack tracebacks
Atul Varma <varmaa@toolness.com>
parents:
15
diff
changeset
|
220 line = filelines[frameinfo['lineno'] - 1].strip() |
a78570a423ea
Show code excerpts in js stack tracebacks
Atul Varma <varmaa@toolness.com>
parents:
15
diff
changeset
|
221 lines.insert(1, " %s" % line) |
a78570a423ea
Show code excerpts in js stack tracebacks
Atul Varma <varmaa@toolness.com>
parents:
15
diff
changeset
|
222 except Exception: |
a78570a423ea
Show code excerpts in js stack tracebacks
Atul Varma <varmaa@toolness.com>
parents:
15
diff
changeset
|
223 pass |
10 | 224 js_stack = js_stack['caller'] |
225 lines.insert(0, "Traceback (most recent call last):") | |
226 return '\n'.join(lines) | |
227 | |
12
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
228 class JSRuntime(object): |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
229 """ |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
230 A JS runtime capable of loading and executing scripts. |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
231 """ |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
232 |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
233 def __init__(self, watchdog=watchdog): |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
234 rt = pydermonkey.Runtime() |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
235 cx = rt.new_context() |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
236 globalobj = cx.new_object() |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
237 cx.init_standard_classes(globalobj) |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
238 |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
239 @safejsfunc(cx, globalobj) |
17
1d62177c5c27
Added a safe js object wrapper.
Atul Varma <varmaa@toolness.com>
parents:
16
diff
changeset
|
240 def bar(cx, this, args): |
18
69e5523ebdc6
Separated out SafeJsObjectWrapper into itself and a new subclass, SafeJsFunctionWrapper, so that callable() works properly on instances.
Atul Varma <varmaa@toolness.com>
parents:
17
diff
changeset
|
241 obj = SafeJsObjectWrapper.wrap(cx, args[0], this) |
17
1d62177c5c27
Added a safe js object wrapper.
Atul Varma <varmaa@toolness.com>
parents:
16
diff
changeset
|
242 print obj.bar() |
1d62177c5c27
Added a safe js object wrapper.
Atul Varma <varmaa@toolness.com>
parents:
16
diff
changeset
|
243 |
1d62177c5c27
Added a safe js object wrapper.
Atul Varma <varmaa@toolness.com>
parents:
16
diff
changeset
|
244 @safejsfunc(cx, globalobj) |
12
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
245 def foo(cx, this, args): |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
246 return cx.call_function(this, args[0], ()) |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
247 |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
248 @safejsfunc(cx, globalobj, 'print') |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
249 def jsprint(cx, this, args): |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
250 if len(args) > 0: |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
251 print args[0] |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
252 |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
253 cx.set_operation_callback(self._opcb) |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
254 cx.set_throw_hook(self._throwhook) |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
255 watchdog.add_context(cx) |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
256 |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
257 self.rt = rt |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
258 self.cx = cx |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
259 self.globalobj = globalobj |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
260 self.curr_exc = None |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
261 self.py_stack = None |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
262 self.js_stack = None |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
263 |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
264 def _opcb(self, cx): |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
265 # Don't do anything; if a keyboard interrupt was triggered, |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
266 # it'll get raised here automatically. |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
267 pass |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
268 |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
269 def _throwhook(self, cx): |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
270 curr_exc = cx.get_pending_exception() |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
271 if self.curr_exc != curr_exc: |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
272 self.curr_exc = curr_exc |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
273 self.py_stack = traceback.extract_stack() |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
274 self.js_stack = cx.get_stack() |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
275 |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
276 def run_script(self, filename): |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
277 contents = open(filename).read() |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
278 cx = self.cx |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
279 try: |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
280 cx.evaluate_script(self.globalobj, contents, |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
281 filename, 1) |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
282 except pydermonkey.error, e: |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
283 print format_stack(self.js_stack) |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
284 print e.args[1] |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
285 except InternalError, e: |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
286 print "An internal error occurred." |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
287 traceback.print_tb(e.exc_info[2]) |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
288 print e.exc_info[1] |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
289 |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
290 if __name__ == '__main__': |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
291 runtime = JSRuntime() |
f024e41d0fb9
More refactoring, a bit more documentation.
Atul Varma <varmaa@toolness.com>
parents:
11
diff
changeset
|
292 runtime.run_script('test.js') |
14
9bcbf780581f
Added a del statement that makes pydermonkey crash.
Atul Varma <varmaa@toolness.com>
parents:
13
diff
changeset
|
293 del runtime |
15
351525e95a45
Added memory leak detection.
Atul Varma <varmaa@toolness.com>
parents:
14
diff
changeset
|
294 |
351525e95a45
Added memory leak detection.
Atul Varma <varmaa@toolness.com>
parents:
14
diff
changeset
|
295 import gc |
351525e95a45
Added memory leak detection.
Atul Varma <varmaa@toolness.com>
parents:
14
diff
changeset
|
296 gc.collect() |
351525e95a45
Added memory leak detection.
Atul Varma <varmaa@toolness.com>
parents:
14
diff
changeset
|
297 if pydermonkey.get_debug_info()['runtime_count']: |
351525e95a45
Added memory leak detection.
Atul Varma <varmaa@toolness.com>
parents:
14
diff
changeset
|
298 print "WARNING: JS runtime was not destroyed." |