annotate pydertron.py @ 15:f30bd92e2216

Added support for running test suite over http.
author Atul Varma <varmaa@toolness.com>
date Thu, 10 Sep 2009 14:36:17 -0700
parents 16fe9c63aedb
children 9426f9fa6dc0
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
0
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
1 import sys
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
2 import threading
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
3 import traceback
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
4 import weakref
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
5 import types
7
2117265e4dfe Fixed some threading issues.
Atul Varma <varmaa@toolness.com>
parents: 6
diff changeset
6 import atexit
0
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
7
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
8 import pydermonkey
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
9
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
10 class ContextWatchdogThread(threading.Thread):
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
11 """
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
12 Watches active JS contexts and triggers their operation callbacks
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
13 at a regular interval.
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
14 """
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
15
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
16 # Default interval, in seconds, that the operation callbacks are
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
17 # triggered at.
8
fb0b161542b1 Improved performance of watchdog thread.
Atul Varma <varmaa@toolness.com>
parents: 7
diff changeset
18 DEFAULT_INTERVAL = 0.25
0
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
19
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
20 def __init__(self, interval=DEFAULT_INTERVAL):
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
21 threading.Thread.__init__(self)
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
22 self._lock = threading.Lock()
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
23 self._stop = threading.Event()
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
24 self._contexts = []
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
25 self.interval = interval
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
26
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
27 def add_context(self, cx):
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
28 self._lock.acquire()
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
29 try:
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
30 self._contexts.append(weakref.ref(cx))
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
31 finally:
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
32 self._lock.release()
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
33
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
34 def join(self):
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
35 self._stop.set()
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
36 threading.Thread.join(self)
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
37
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
38 def run(self):
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
39 while not self._stop.isSet():
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
40 new_list = []
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
41 self._lock.acquire()
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
42 try:
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
43 for weakcx in self._contexts:
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
44 cx = weakcx()
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
45 if cx:
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
46 new_list.append(weakcx)
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
47 cx.trigger_operation_callback()
7
2117265e4dfe Fixed some threading issues.
Atul Varma <varmaa@toolness.com>
parents: 6
diff changeset
48 del cx
0
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
49 self._contexts = new_list
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
50 finally:
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
51 self._lock.release()
8
fb0b161542b1 Improved performance of watchdog thread.
Atul Varma <varmaa@toolness.com>
parents: 7
diff changeset
52 self._stop.wait(self.interval)
0
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
53
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
54 # Create a global watchdog.
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
55 watchdog = ContextWatchdogThread()
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
56 watchdog.start()
7
2117265e4dfe Fixed some threading issues.
Atul Varma <varmaa@toolness.com>
parents: 6
diff changeset
57 atexit.register(watchdog.join)
0
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
58
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
59 class InternalError(BaseException):
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
60 """
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
61 Represents an error in a JS-wrapped Python function that wasn't
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
62 expected to happen; because it's derived from BaseException, it
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
63 unrolls the whole JS/Python stack so that the error can be
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
64 reported to the outermost calling code.
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
65 """
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
66
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
67 def __init__(self):
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
68 BaseException.__init__(self)
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
69 self.exc_info = sys.exc_info()
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
70
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
71 class SafeJsObjectWrapper(object):
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
72 """
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
73 Securely wraps a JS object to behave like any normal Python object.
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
74 """
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
75
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
76 __slots__ = ['_jsobject', '_sandbox', '_this']
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
77
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
78 def __init__(self, sandbox, jsobject, this):
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
79 if not isinstance(jsobject, pydermonkey.Object):
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
80 raise TypeError("Cannot wrap '%s' object" %
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
81 type(jsobject).__name__)
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
82 object.__setattr__(self, '_sandbox', sandbox)
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
83 object.__setattr__(self, '_jsobject', jsobject)
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
84 object.__setattr__(self, '_this', this)
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
85
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
86 @property
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
87 def wrapped_jsobject(self):
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
88 return self._jsobject
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
89
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
90 def _wrap_to_python(self, jsvalue):
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
91 return self._sandbox.wrap_jsobject(jsvalue, self._jsobject)
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
92
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
93 def _wrap_to_js(self, value):
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
94 return self._sandbox.wrap_pyobject(value)
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
95
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
96 def __eq__(self, other):
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
97 if isinstance(other, SafeJsObjectWrapper):
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
98 return self._jsobject == other._jsobject
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
99 else:
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
100 return False
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
101
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
102 def __str__(self):
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
103 return self.toString()
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
104
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
105 def __unicode__(self):
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
106 return self.toString()
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
107
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
108 def __setitem__(self, item, value):
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
109 self.__setattr__(item, value)
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
110
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
111 def __setattr__(self, name, value):
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
112 cx = self._sandbox.cx
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
113 jsobject = self._jsobject
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
114
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
115 cx.define_property(jsobject, name,
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
116 self._wrap_to_js(value))
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
117
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
118 def __getitem__(self, item):
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
119 return self.__getattr__(item)
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
120
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
121 def __getattr__(self, name):
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
122 cx = self._sandbox.cx
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
123 jsobject = self._jsobject
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
124
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
125 return self._wrap_to_python(cx.get_property(jsobject, name))
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
126
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
127 def __contains__(self, item):
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
128 cx = self._sandbox.cx
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
129 jsobject = self._jsobject
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
130
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
131 return cx.has_property(jsobject, item)
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
132
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
133 def __iter__(self):
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
134 cx = self._sandbox.cx
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
135 jsobject = self._jsobject
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
136
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
137 properties = cx.enumerate(jsobject)
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
138 for property in properties:
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
139 yield property
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
140
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
141 class SafeJsFunctionWrapper(SafeJsObjectWrapper):
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
142 """
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
143 Securely wraps a JS function to behave like any normal Python object.
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
144 """
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
145
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
146 def __init__(self, sandbox, jsfunction, this):
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
147 if not isinstance(jsfunction, pydermonkey.Function):
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
148 raise TypeError("Cannot wrap '%s' object" %
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
149 type(jsobject).__name__)
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
150 SafeJsObjectWrapper.__init__(self, sandbox, jsfunction, this)
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
151
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
152 def __call__(self, *args):
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
153 cx = self._sandbox.cx
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
154 jsobject = self._jsobject
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
155 this = self._this
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
156
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
157 arglist = []
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
158 for arg in args:
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
159 arglist.append(self._wrap_to_js(arg))
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
160
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
161 obj = cx.call_function(this, jsobject, tuple(arglist))
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
162 return self._wrap_to_python(obj)
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
163
11
665f69f61716 Some refactorings
Atul Varma <varmaa@toolness.com>
parents: 9
diff changeset
164 def format_stack(js_stack, open=open):
0
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
165 """
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
166 Returns a formatted Python-esque stack traceback of the given
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
167 JS stack.
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
168 """
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
169
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
170 STACK_LINE =" File \"%(filename)s\", line %(lineno)d, in %(name)s"
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
171
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
172 lines = []
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
173 while js_stack:
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
174 script = js_stack['script']
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
175 function = js_stack['function']
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
176 if script:
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
177 frameinfo = dict(filename = script.filename,
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
178 lineno = js_stack['lineno'],
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
179 name = '<module>')
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
180 elif function and not function.is_python:
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
181 frameinfo = dict(filename = function.filename,
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
182 lineno = js_stack['lineno'],
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
183 name = function.name)
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
184 else:
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
185 frameinfo = None
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
186 if frameinfo:
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
187 lines.insert(0, STACK_LINE % frameinfo)
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
188 try:
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
189 filelines = open(frameinfo['filename']).readlines()
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
190 line = filelines[frameinfo['lineno'] - 1].strip()
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
191 lines.insert(1, " %s" % line)
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
192 except Exception:
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
193 pass
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
194 js_stack = js_stack['caller']
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
195 lines.insert(0, "Traceback (most recent call last):")
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
196 return '\n'.join(lines)
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
197
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
198 def jsexposed(name=None, on=None):
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
199 """
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
200 Decorator used to expose the decorated function or method to
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
201 untrusted JS.
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
202
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
203 'name' is an optional alternative name for the function.
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
204
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
205 'on' is an optional SafeJsObjectWrapper that the function can be
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
206 automatically attached as a property to.
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
207 """
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
208
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
209 if callable(name):
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
210 func = name
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
211 func.__jsexposed__ = True
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
212 return func
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
213
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
214 def make_exposed(func):
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
215 if name:
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
216 func.__name__ = name
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
217 func.__jsexposed__ = True
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
218 if on:
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
219 on[func.__name__] = func
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
220 return func
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
221 return make_exposed
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
222
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
223 class JsExposedObject(object):
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
224 """
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
225 Trivial base/mixin class for any Python classes that choose to
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
226 expose themselves to JS code.
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
227 """
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
228
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
229 pass
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
230
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
231 class JsSandbox(object):
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
232 """
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
233 A JS runtime and associated functionality capable of securely
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
234 loading and executing scripts.
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
235 """
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
236
11
665f69f61716 Some refactorings
Atul Varma <varmaa@toolness.com>
parents: 9
diff changeset
237 def __init__(self, fs, watchdog=watchdog):
0
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
238 rt = pydermonkey.Runtime()
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
239 cx = rt.new_context()
2
b6f9d743a2b5 Refined require() implementation.
Atul Varma <varmaa@toolness.com>
parents: 1
diff changeset
240 root_proto = cx.new_object()
b6f9d743a2b5 Refined require() implementation.
Atul Varma <varmaa@toolness.com>
parents: 1
diff changeset
241 cx.init_standard_classes(root_proto)
b6f9d743a2b5 Refined require() implementation.
Atul Varma <varmaa@toolness.com>
parents: 1
diff changeset
242 root = cx.new_object(None, root_proto)
0
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
243
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
244 cx.set_operation_callback(self._opcb)
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
245 cx.set_throw_hook(self._throwhook)
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
246 watchdog.add_context(cx)
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
247
11
665f69f61716 Some refactorings
Atul Varma <varmaa@toolness.com>
parents: 9
diff changeset
248 self.fs = fs
0
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
249 self.rt = rt
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
250 self.cx = cx
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
251 self.curr_exc = None
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
252 self.py_stack = None
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
253 self.js_stack = None
11
665f69f61716 Some refactorings
Atul Varma <varmaa@toolness.com>
parents: 9
diff changeset
254 self.__modules = {}
0
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
255 self.__py_to_js = {}
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
256 self.__type_protos = {}
11
665f69f61716 Some refactorings
Atul Varma <varmaa@toolness.com>
parents: 9
diff changeset
257 self.__globals = {}
665f69f61716 Some refactorings
Atul Varma <varmaa@toolness.com>
parents: 9
diff changeset
258 self.__root_proto = root_proto
0
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
259 self.root = self.wrap_jsobject(root, root)
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
260
11
665f69f61716 Some refactorings
Atul Varma <varmaa@toolness.com>
parents: 9
diff changeset
261 def set_globals(self, **globals):
665f69f61716 Some refactorings
Atul Varma <varmaa@toolness.com>
parents: 9
diff changeset
262 self.__globals.update(globals)
665f69f61716 Some refactorings
Atul Varma <varmaa@toolness.com>
parents: 9
diff changeset
263 self._install_globals(self.root)
665f69f61716 Some refactorings
Atul Varma <varmaa@toolness.com>
parents: 9
diff changeset
264
0
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
265 def finish(self):
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
266 """
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
267 Cleans up all resources used by the sandbox, breaking any reference
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
268 cycles created due to issue #2 in pydermonkey:
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
269
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
270 http://code.google.com/p/pydermonkey/issues/detail?id=2
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
271 """
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
272
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
273 for jsobj in self.__py_to_js.values():
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
274 self.cx.clear_object_private(jsobj)
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
275 del self.__py_to_js
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
276 del self.__type_protos
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
277 del self.curr_exc
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
278 del self.py_stack
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
279 del self.js_stack
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
280 del self.cx
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
281 del self.rt
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
282
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
283 def _opcb(self, cx):
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
284 # Don't do anything; if a keyboard interrupt was triggered,
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
285 # it'll get raised here automatically.
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
286 pass
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
287
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
288 def _throwhook(self, cx):
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
289 curr_exc = cx.get_pending_exception()
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
290 if self.curr_exc != curr_exc:
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
291 self.curr_exc = curr_exc
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
292 self.py_stack = traceback.extract_stack()
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
293 self.js_stack = cx.get_stack()
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
294
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
295 def __wrap_pycallable(self, func, pyproto=None):
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
296 if func in self.__py_to_js:
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
297 return self.__py_to_js[func]
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
298
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
299 if hasattr(func, '__name__'):
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
300 name = func.__name__
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
301 else:
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
302 name = ""
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
303
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
304 if pyproto:
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
305 def wrapper(func_cx, this, args):
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
306 try:
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
307 arglist = []
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
308 for arg in args:
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
309 arglist.append(self.wrap_jsobject(arg))
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
310 instance = func_cx.get_object_private(this)
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
311 if instance is None or not isinstance(instance, pyproto):
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
312 raise pydermonkey.error("Method type mismatch")
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
313
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
314 # TODO: Fill in extra required params with
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
315 # pymonkey.undefined? or automatically throw an
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
316 # exception to calling js code?
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
317 return self.wrap_pyobject(func(instance, *arglist))
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
318 except pydermonkey.error:
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
319 raise
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
320 except Exception:
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
321 raise InternalError()
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
322 else:
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
323 def wrapper(func_cx, this, args):
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
324 try:
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
325 arglist = []
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
326 for arg in args:
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
327 arglist.append(self.wrap_jsobject(arg))
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
328
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
329 # TODO: Fill in extra required params with
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
330 # pymonkey.undefined? or automatically throw an
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
331 # exception to calling js code?
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
332 return self.wrap_pyobject(func(*arglist))
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
333 except pydermonkey.error:
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
334 raise
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
335 except Exception:
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
336 raise InternalError()
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
337 wrapper.wrapped_pyobject = func
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
338 wrapper.__name__ = name
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
339
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
340 jsfunc = self.cx.new_function(wrapper, name)
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
341 self.__py_to_js[func] = jsfunc
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
342
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
343 return jsfunc
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
344
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
345 def __wrap_pyinstance(self, value):
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
346 pyproto = type(value)
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
347 if pyproto not in self.__type_protos:
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
348 jsproto = self.cx.new_object()
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
349 if hasattr(pyproto, '__jsprops__'):
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
350 define_getter = self.cx.get_property(jsproto,
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
351 '__defineGetter__')
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
352 define_setter = self.cx.get_property(jsproto,
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
353 '__defineSetter__')
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
354 for name in pyproto.__jsprops__:
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
355 prop = getattr(pyproto, name)
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
356 if not type(prop) == property:
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
357 raise TypeError("Expected attribute '%s' to "
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
358 "be a property" % name)
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
359 getter = None
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
360 setter = None
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
361 if prop.fget:
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
362 getter = self.__wrap_pycallable(prop.fget,
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
363 pyproto)
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
364 if prop.fset:
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
365 setter = self.__wrap_pycallable(prop.fset,
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
366 pyproto)
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
367 if getter:
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
368 self.cx.call_function(jsproto,
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
369 define_getter,
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
370 (name, getter))
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
371 if setter:
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
372 self.cx.call_function(jsproto,
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
373 define_setter,
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
374 (name, setter,))
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
375 for name in dir(pyproto):
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
376 attr = getattr(pyproto, name)
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
377 if (isinstance(attr, types.UnboundMethodType) and
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
378 hasattr(attr, '__jsexposed__') and
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
379 attr.__jsexposed__):
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
380 jsmethod = self.__wrap_pycallable(attr, pyproto)
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
381 self.cx.define_property(jsproto, name, jsmethod)
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
382 self.__type_protos[pyproto] = jsproto
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
383 return self.cx.new_object(value, self.__type_protos[pyproto])
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
384
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
385 def wrap_pyobject(self, value):
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
386 """
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
387 Wraps the given Python object for export to untrusted JS.
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
388
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
389 If the Python object isn't of a type that can be exposed to JS,
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
390 a TypeError is raised.
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
391 """
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
392
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
393 if (isinstance(value, (int, basestring, float, bool)) or
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
394 value is pydermonkey.undefined or
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
395 value is None):
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
396 return value
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
397 if isinstance(value, SafeJsObjectWrapper):
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
398 # It's already wrapped, just unwrap it.
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
399 return value.wrapped_jsobject
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
400 elif callable(value):
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
401 if not (hasattr(value, '__jsexposed__') and
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
402 value.__jsexposed__):
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
403 raise ValueError("Callable isn't configured for exposure "
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
404 "to untrusted JS code")
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
405 return self.__wrap_pycallable(value)
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
406 elif isinstance(value, JsExposedObject):
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
407 return self.__wrap_pyinstance(value)
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
408 else:
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
409 raise TypeError("Can't expose objects of type '%s' to JS." %
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
410 type(value).__name__)
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
411
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
412 def wrap_jsobject(self, jsvalue, this=None):
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
413 """
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
414 Wraps the given pydermonkey.Object for import to trusted
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
415 Python code. If the type is just a primitive, it's simply
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
416 returned, since no wrapping is needed.
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
417 """
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
418
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
419 if this is None:
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
420 this = self.root.wrapped_jsobject
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
421 if isinstance(jsvalue, pydermonkey.Function):
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
422 if jsvalue.is_python:
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
423 # It's a Python function, just unwrap it.
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
424 return self.cx.get_object_private(jsvalue).wrapped_pyobject
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
425 return SafeJsFunctionWrapper(self, jsvalue, this)
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
426 elif isinstance(jsvalue, pydermonkey.Object):
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
427 # It's a wrapped Python object instance, just unwrap it.
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
428 instance = self.cx.get_object_private(jsvalue)
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
429 if instance:
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
430 if not isinstance(instance, JsExposedObject):
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
431 raise AssertionError("Object private is not of type "
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
432 "JsExposedObject")
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
433 return instance
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
434 else:
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
435 return SafeJsObjectWrapper(self, jsvalue, this)
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
436 else:
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
437 # It's a primitive value.
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
438 return jsvalue
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
439
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
440 def new_array(self, *contents):
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
441 array = self.wrap_jsobject(self.cx.new_array_object())
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
442 for item in contents:
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
443 array.push(item)
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
444 return array
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
445
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
446 def new_object(self, **contents):
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
447 obj = self.wrap_jsobject(self.cx.new_object())
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
448 for name in contents:
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
449 obj[name] = contents[name]
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
450 return obj
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
451
11
665f69f61716 Some refactorings
Atul Varma <varmaa@toolness.com>
parents: 9
diff changeset
452 def get_calling_script(self):
665f69f61716 Some refactorings
Atul Varma <varmaa@toolness.com>
parents: 9
diff changeset
453 frame = self.cx.get_stack()['caller']
665f69f61716 Some refactorings
Atul Varma <varmaa@toolness.com>
parents: 9
diff changeset
454 curr_script = None
665f69f61716 Some refactorings
Atul Varma <varmaa@toolness.com>
parents: 9
diff changeset
455 while frame and curr_script is None:
665f69f61716 Some refactorings
Atul Varma <varmaa@toolness.com>
parents: 9
diff changeset
456 if frame['function'] and frame['function'].filename:
665f69f61716 Some refactorings
Atul Varma <varmaa@toolness.com>
parents: 9
diff changeset
457 curr_script = frame['function'].filename
665f69f61716 Some refactorings
Atul Varma <varmaa@toolness.com>
parents: 9
diff changeset
458 elif frame['script']:
665f69f61716 Some refactorings
Atul Varma <varmaa@toolness.com>
parents: 9
diff changeset
459 curr_script = frame['script'].filename
665f69f61716 Some refactorings
Atul Varma <varmaa@toolness.com>
parents: 9
diff changeset
460 frame = frame['caller']
665f69f61716 Some refactorings
Atul Varma <varmaa@toolness.com>
parents: 9
diff changeset
461
665f69f61716 Some refactorings
Atul Varma <varmaa@toolness.com>
parents: 9
diff changeset
462 if curr_script is None:
665f69f61716 Some refactorings
Atul Varma <varmaa@toolness.com>
parents: 9
diff changeset
463 raise RuntimeError("Can't find calling script")
665f69f61716 Some refactorings
Atul Varma <varmaa@toolness.com>
parents: 9
diff changeset
464 return curr_script
665f69f61716 Some refactorings
Atul Varma <varmaa@toolness.com>
parents: 9
diff changeset
465
665f69f61716 Some refactorings
Atul Varma <varmaa@toolness.com>
parents: 9
diff changeset
466 def _install_globals(self, object):
665f69f61716 Some refactorings
Atul Varma <varmaa@toolness.com>
parents: 9
diff changeset
467 for name in self.__globals:
665f69f61716 Some refactorings
Atul Varma <varmaa@toolness.com>
parents: 9
diff changeset
468 object[name] = self.__globals[name]
665f69f61716 Some refactorings
Atul Varma <varmaa@toolness.com>
parents: 9
diff changeset
469 object['require'] = self._require
665f69f61716 Some refactorings
Atul Varma <varmaa@toolness.com>
parents: 9
diff changeset
470
665f69f61716 Some refactorings
Atul Varma <varmaa@toolness.com>
parents: 9
diff changeset
471 @jsexposed(name='require')
665f69f61716 Some refactorings
Atul Varma <varmaa@toolness.com>
parents: 9
diff changeset
472 def _require(self, path):
665f69f61716 Some refactorings
Atul Varma <varmaa@toolness.com>
parents: 9
diff changeset
473 filename = self.fs.find_module(self.get_calling_script(), path)
665f69f61716 Some refactorings
Atul Varma <varmaa@toolness.com>
parents: 9
diff changeset
474 if not filename:
665f69f61716 Some refactorings
Atul Varma <varmaa@toolness.com>
parents: 9
diff changeset
475 raise pydermonkey.error('Module not found: %s' % path)
665f69f61716 Some refactorings
Atul Varma <varmaa@toolness.com>
parents: 9
diff changeset
476 if not filename in self.__modules:
665f69f61716 Some refactorings
Atul Varma <varmaa@toolness.com>
parents: 9
diff changeset
477 cx = self.cx
665f69f61716 Some refactorings
Atul Varma <varmaa@toolness.com>
parents: 9
diff changeset
478 module = cx.new_object(None, self.__root_proto)
665f69f61716 Some refactorings
Atul Varma <varmaa@toolness.com>
parents: 9
diff changeset
479 cx.init_standard_classes(module)
665f69f61716 Some refactorings
Atul Varma <varmaa@toolness.com>
parents: 9
diff changeset
480 exports = cx.new_object()
665f69f61716 Some refactorings
Atul Varma <varmaa@toolness.com>
parents: 9
diff changeset
481 cx.define_property(module, 'exports', exports)
665f69f61716 Some refactorings
Atul Varma <varmaa@toolness.com>
parents: 9
diff changeset
482 self._install_globals(self.wrap_jsobject(module))
665f69f61716 Some refactorings
Atul Varma <varmaa@toolness.com>
parents: 9
diff changeset
483 self.__modules[filename] = self.wrap_jsobject(exports)
665f69f61716 Some refactorings
Atul Varma <varmaa@toolness.com>
parents: 9
diff changeset
484 contents = self.fs.open(filename).read()
665f69f61716 Some refactorings
Atul Varma <varmaa@toolness.com>
parents: 9
diff changeset
485 cx.evaluate_script(module, contents, filename, 1)
665f69f61716 Some refactorings
Atul Varma <varmaa@toolness.com>
parents: 9
diff changeset
486 return self.__modules[filename]
0
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
487
6
97adec8c8127 Now passing all CommonJS SecurableModule compliance tests.
Atul Varma <varmaa@toolness.com>
parents: 5
diff changeset
488 def run_script(self, contents, filename='<string>', lineno=1,
11
665f69f61716 Some refactorings
Atul Varma <varmaa@toolness.com>
parents: 9
diff changeset
489 callback=None, stderr=sys.stderr):
0
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
490 """
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
491 Runs the given JS script, returning 0 on success, -1 on failure.
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
492 """
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
493
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
494 retval = -1
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
495 cx = self.cx
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
496 try:
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
497 result = cx.evaluate_script(self.root.wrapped_jsobject,
6
97adec8c8127 Now passing all CommonJS SecurableModule compliance tests.
Atul Varma <varmaa@toolness.com>
parents: 5
diff changeset
498 contents, filename, lineno)
0
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
499 if callback:
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
500 callback(self.wrap_jsobject(result))
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
501 retval = 0
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
502 except pydermonkey.error, e:
11
665f69f61716 Some refactorings
Atul Varma <varmaa@toolness.com>
parents: 9
diff changeset
503 params = dict(
665f69f61716 Some refactorings
Atul Varma <varmaa@toolness.com>
parents: 9
diff changeset
504 stack_trace = format_stack(self.js_stack, self.fs.open),
665f69f61716 Some refactorings
Atul Varma <varmaa@toolness.com>
parents: 9
diff changeset
505 error = e.args[1]
665f69f61716 Some refactorings
Atul Varma <varmaa@toolness.com>
parents: 9
diff changeset
506 )
665f69f61716 Some refactorings
Atul Varma <varmaa@toolness.com>
parents: 9
diff changeset
507 stderr.write("%(stack_trace)s\n%(error)s\n" % params)
0
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
508 except InternalError, e:
11
665f69f61716 Some refactorings
Atul Varma <varmaa@toolness.com>
parents: 9
diff changeset
509 stderr.write("An internal error occurred.\n")
665f69f61716 Some refactorings
Atul Varma <varmaa@toolness.com>
parents: 9
diff changeset
510 traceback.print_tb(e.exc_info[2], None, stderr)
665f69f61716 Some refactorings
Atul Varma <varmaa@toolness.com>
parents: 9
diff changeset
511 stderr.write("%s\n" % e.exc_info[1])
0
a5b09b685df4 Origination
Atul Varma <varmaa@toolness.com>
parents:
diff changeset
512 return retval
1
ab09b8a10876 Added trivial half-baked implementation of securable modules.
Atul Varma <varmaa@toolness.com>
parents: 0
diff changeset
513
14
16fe9c63aedb Added HttpFileSystem
Atul Varma <varmaa@toolness.com>
parents: 12
diff changeset
514 class HttpFileSystem(object):
16fe9c63aedb Added HttpFileSystem
Atul Varma <varmaa@toolness.com>
parents: 12
diff changeset
515 def __init__(self, base_url):
16fe9c63aedb Added HttpFileSystem
Atul Varma <varmaa@toolness.com>
parents: 12
diff changeset
516 self.base_url = base_url
16fe9c63aedb Added HttpFileSystem
Atul Varma <varmaa@toolness.com>
parents: 12
diff changeset
517
16fe9c63aedb Added HttpFileSystem
Atul Varma <varmaa@toolness.com>
parents: 12
diff changeset
518 def find_module(self, curr_url, path):
16fe9c63aedb Added HttpFileSystem
Atul Varma <varmaa@toolness.com>
parents: 12
diff changeset
519 import urlparse
16fe9c63aedb Added HttpFileSystem
Atul Varma <varmaa@toolness.com>
parents: 12
diff changeset
520
16fe9c63aedb Added HttpFileSystem
Atul Varma <varmaa@toolness.com>
parents: 12
diff changeset
521 if path.startswith("."):
16fe9c63aedb Added HttpFileSystem
Atul Varma <varmaa@toolness.com>
parents: 12
diff changeset
522 base_url = curr_url
16fe9c63aedb Added HttpFileSystem
Atul Varma <varmaa@toolness.com>
parents: 12
diff changeset
523 else:
16fe9c63aedb Added HttpFileSystem
Atul Varma <varmaa@toolness.com>
parents: 12
diff changeset
524 base_url = self.base_url
16fe9c63aedb Added HttpFileSystem
Atul Varma <varmaa@toolness.com>
parents: 12
diff changeset
525
16fe9c63aedb Added HttpFileSystem
Atul Varma <varmaa@toolness.com>
parents: 12
diff changeset
526 url = "%s.js" % urlparse.urljoin(base_url, path)
16fe9c63aedb Added HttpFileSystem
Atul Varma <varmaa@toolness.com>
parents: 12
diff changeset
527 if not url.startswith(self.base_url):
16fe9c63aedb Added HttpFileSystem
Atul Varma <varmaa@toolness.com>
parents: 12
diff changeset
528 return None
16fe9c63aedb Added HttpFileSystem
Atul Varma <varmaa@toolness.com>
parents: 12
diff changeset
529 return url
16fe9c63aedb Added HttpFileSystem
Atul Varma <varmaa@toolness.com>
parents: 12
diff changeset
530
16fe9c63aedb Added HttpFileSystem
Atul Varma <varmaa@toolness.com>
parents: 12
diff changeset
531 def open(self, url):
16fe9c63aedb Added HttpFileSystem
Atul Varma <varmaa@toolness.com>
parents: 12
diff changeset
532 import urllib
16fe9c63aedb Added HttpFileSystem
Atul Varma <varmaa@toolness.com>
parents: 12
diff changeset
533
16fe9c63aedb Added HttpFileSystem
Atul Varma <varmaa@toolness.com>
parents: 12
diff changeset
534 return urllib.urlopen(url)
16fe9c63aedb Added HttpFileSystem
Atul Varma <varmaa@toolness.com>
parents: 12
diff changeset
535
11
665f69f61716 Some refactorings
Atul Varma <varmaa@toolness.com>
parents: 9
diff changeset
536 class SandboxedFileSystem(object):
665f69f61716 Some refactorings
Atul Varma <varmaa@toolness.com>
parents: 9
diff changeset
537 def __init__(self, root_dir):
3
14d8d73774d7 Refactored securable module loader.
Atul Varma <varmaa@toolness.com>
parents: 2
diff changeset
538 self.root_dir = root_dir
2
b6f9d743a2b5 Refined require() implementation.
Atul Varma <varmaa@toolness.com>
parents: 1
diff changeset
539
11
665f69f61716 Some refactorings
Atul Varma <varmaa@toolness.com>
parents: 9
diff changeset
540 def find_module(self, curr_script, path):
14
16fe9c63aedb Added HttpFileSystem
Atul Varma <varmaa@toolness.com>
parents: 12
diff changeset
541 import os
16fe9c63aedb Added HttpFileSystem
Atul Varma <varmaa@toolness.com>
parents: 12
diff changeset
542
11
665f69f61716 Some refactorings
Atul Varma <varmaa@toolness.com>
parents: 9
diff changeset
543 if path.startswith("."):
665f69f61716 Some refactorings
Atul Varma <varmaa@toolness.com>
parents: 9
diff changeset
544 base_dir = os.path.dirname(curr_script)
665f69f61716 Some refactorings
Atul Varma <varmaa@toolness.com>
parents: 9
diff changeset
545 else:
665f69f61716 Some refactorings
Atul Varma <varmaa@toolness.com>
parents: 9
diff changeset
546 base_dir = self.root_dir
2
b6f9d743a2b5 Refined require() implementation.
Atul Varma <varmaa@toolness.com>
parents: 1
diff changeset
547
11
665f69f61716 Some refactorings
Atul Varma <varmaa@toolness.com>
parents: 9
diff changeset
548 ospath = path.replace('/', os.path.sep)
665f69f61716 Some refactorings
Atul Varma <varmaa@toolness.com>
parents: 9
diff changeset
549 filename = os.path.join(base_dir, "%s.js" % ospath)
665f69f61716 Some refactorings
Atul Varma <varmaa@toolness.com>
parents: 9
diff changeset
550 filename = os.path.normpath(filename)
665f69f61716 Some refactorings
Atul Varma <varmaa@toolness.com>
parents: 9
diff changeset
551 if (filename.startswith(self.root_dir) and
665f69f61716 Some refactorings
Atul Varma <varmaa@toolness.com>
parents: 9
diff changeset
552 (os.path.exists(filename) and
665f69f61716 Some refactorings
Atul Varma <varmaa@toolness.com>
parents: 9
diff changeset
553 not os.path.isdir(filename))):
665f69f61716 Some refactorings
Atul Varma <varmaa@toolness.com>
parents: 9
diff changeset
554 return filename
665f69f61716 Some refactorings
Atul Varma <varmaa@toolness.com>
parents: 9
diff changeset
555 else:
665f69f61716 Some refactorings
Atul Varma <varmaa@toolness.com>
parents: 9
diff changeset
556 return None
3
14d8d73774d7 Refactored securable module loader.
Atul Varma <varmaa@toolness.com>
parents: 2
diff changeset
557
11
665f69f61716 Some refactorings
Atul Varma <varmaa@toolness.com>
parents: 9
diff changeset
558 def open(self, filename):
665f69f61716 Some refactorings
Atul Varma <varmaa@toolness.com>
parents: 9
diff changeset
559 return open(filename, 'r')