# HG changeset patch
# User Atul Varma <varmaa@toolness.com>
# Date 1251091618 25200
# Node ID 1a982ee564585addd2fa9ad8540192a95c3979bb
# Parent  f83a1603c417101a0a69c207f9de703faac4568c
A script's bytecode is now exposed as a buffer object.

diff -r f83a1603c417 -r 1a982ee56458 src/script.cpp
--- a/src/script.cpp	Sun Aug 23 21:59:39 2009 -0700
+++ b/src/script.cpp	Sun Aug 23 22:26:58 2009 -0700
@@ -37,6 +37,8 @@
 #include "script.h"
 #include "utils.h"
 
+#include "jsscript.h"
+
 static void
 PYM_JSScriptDealloc(PYM_JSScript *self)
 {
@@ -44,6 +46,34 @@
   PYM_JSObjectType.tp_dealloc((PyObject *) self);
 }
 
+Py_ssize_t PYM_readbuffer(PYM_JSScript *self, Py_ssize_t segment,
+                          void **ptrptr)
+{
+  *ptrptr = self->script->code;
+  return self->script->length;
+}
+
+Py_ssize_t PYM_segcount(PYM_JSScript *self, Py_ssize_t *lenp)
+{
+  if (lenp)
+    *lenp = self->script->length;
+  return 1;
+}
+
+Py_ssize_t PYM_charbuffer(PYM_JSScript *self, Py_ssize_t segment,
+                          const char **ptrptr)
+{
+  *ptrptr = (char *) self->script->code;
+  return self->script->length;
+}
+
+static PyBufferProcs PYM_bufferProcs = {
+  (readbufferproc) PYM_readbuffer,
+  NULL,
+  (segcountproc) PYM_segcount,
+  (charbufferproc) PYM_charbuffer
+};
+
 PyTypeObject PYM_JSScriptType = {
   PyObject_HEAD_INIT(NULL)
   0,                           /*ob_size*/
@@ -65,9 +95,9 @@
   0,                           /*tp_str*/
   0,                           /*tp_getattro*/
   0,                           /*tp_setattro*/
-  0,                           /*tp_as_buffer*/
+  &PYM_bufferProcs,            /*tp_as_buffer*/
                                /*tp_flags*/
-  Py_TPFLAGS_DEFAULT,
+  Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GETCHARBUFFER,
                                /* tp_doc */
   "JavaScript Script.",
   0,                           /* tp_traverse */
diff -r f83a1603c417 -r 1a982ee56458 tests/test_pymonkey.py
--- a/tests/test_pymonkey.py	Sun Aug 23 21:59:39 2009 -0700
+++ b/tests/test_pymonkey.py	Sun Aug 23 22:26:58 2009 -0700
@@ -48,6 +48,13 @@
                 u'SyntaxError: missing ; before statement'
                 )
 
+    def testScriptIsExposedAsBuffer(self):
+        rt = pymonkey.Runtime()
+        cx = rt.new_context()
+        obj = cx.new_object()
+        script = cx.compile_script(obj, 'foo', '<string>', 1)
+        self.assertTrue(len(buffer(script)) > 0)
+
     def testCompileScriptWorks(self):
         self.assertEqual(self._execjs('5 + 1'), 6)