blob: a01256068a3ebe60d22d01bf60cefc1d3aea4083 [file] [log] [blame]
Christian Heimes90540002008-05-08 14:29:10 +00001#include "Python.h"
Benjamin Petersonc6b607d2009-05-02 12:36:44 +00002#include "structmember.h"
3#if PY_VERSION_HEX < 0x02060000 && !defined(Py_TYPE)
4#define Py_TYPE(ob) (((PyObject*)(ob))->ob_type)
5#endif
6#if PY_VERSION_HEX < 0x02050000 && !defined(PY_SSIZE_T_MIN)
7typedef int Py_ssize_t;
8#define PY_SSIZE_T_MAX INT_MAX
9#define PY_SSIZE_T_MIN INT_MIN
10#define PyInt_FromSsize_t PyInt_FromLong
11#define PyInt_AsSsize_t PyInt_AsLong
12#endif
13#ifndef Py_IS_FINITE
14#define Py_IS_FINITE(X) (!Py_IS_INFINITY(X) && !Py_IS_NAN(X))
15#endif
Christian Heimes90540002008-05-08 14:29:10 +000016
Benjamin Petersonc6b607d2009-05-02 12:36:44 +000017#ifdef __GNUC__
18#define UNUSED __attribute__((__unused__))
19#else
20#define UNUSED
21#endif
22
23#define PyScanner_Check(op) PyObject_TypeCheck(op, &PyScannerType)
24#define PyScanner_CheckExact(op) (Py_TYPE(op) == &PyScannerType)
25#define PyEncoder_Check(op) PyObject_TypeCheck(op, &PyEncoderType)
26#define PyEncoder_CheckExact(op) (Py_TYPE(op) == &PyEncoderType)
27
28static PyTypeObject PyScannerType;
29static PyTypeObject PyEncoderType;
30
31typedef struct _PyScannerObject {
32 PyObject_HEAD
33 PyObject *strict;
34 PyObject *object_hook;
35 PyObject *object_pairs_hook;
36 PyObject *parse_float;
37 PyObject *parse_int;
38 PyObject *parse_constant;
39} PyScannerObject;
40
41static PyMemberDef scanner_members[] = {
42 {"strict", T_OBJECT, offsetof(PyScannerObject, strict), READONLY, "strict"},
43 {"object_hook", T_OBJECT, offsetof(PyScannerObject, object_hook), READONLY, "object_hook"},
44 {"object_pairs_hook", T_OBJECT, offsetof(PyScannerObject, object_pairs_hook), READONLY},
45 {"parse_float", T_OBJECT, offsetof(PyScannerObject, parse_float), READONLY, "parse_float"},
46 {"parse_int", T_OBJECT, offsetof(PyScannerObject, parse_int), READONLY, "parse_int"},
47 {"parse_constant", T_OBJECT, offsetof(PyScannerObject, parse_constant), READONLY, "parse_constant"},
48 {NULL}
49};
50
51typedef struct _PyEncoderObject {
52 PyObject_HEAD
53 PyObject *markers;
54 PyObject *defaultfn;
55 PyObject *encoder;
56 PyObject *indent;
57 PyObject *key_separator;
58 PyObject *item_separator;
59 PyObject *sort_keys;
60 PyObject *skipkeys;
61 int fast_encode;
62 int allow_nan;
63} PyEncoderObject;
64
65static PyMemberDef encoder_members[] = {
66 {"markers", T_OBJECT, offsetof(PyEncoderObject, markers), READONLY, "markers"},
67 {"default", T_OBJECT, offsetof(PyEncoderObject, defaultfn), READONLY, "default"},
68 {"encoder", T_OBJECT, offsetof(PyEncoderObject, encoder), READONLY, "encoder"},
69 {"indent", T_OBJECT, offsetof(PyEncoderObject, indent), READONLY, "indent"},
70 {"key_separator", T_OBJECT, offsetof(PyEncoderObject, key_separator), READONLY, "key_separator"},
71 {"item_separator", T_OBJECT, offsetof(PyEncoderObject, item_separator), READONLY, "item_separator"},
72 {"sort_keys", T_OBJECT, offsetof(PyEncoderObject, sort_keys), READONLY, "sort_keys"},
73 {"skipkeys", T_OBJECT, offsetof(PyEncoderObject, skipkeys), READONLY, "skipkeys"},
74 {NULL}
75};
76
77static PyObject *
78ascii_escape_unicode(PyObject *pystr);
79static PyObject *
80py_encode_basestring_ascii(PyObject* self UNUSED, PyObject *pystr);
81void init_json(void);
82static PyObject *
83scan_once_unicode(PyScannerObject *s, PyObject *pystr, Py_ssize_t idx, Py_ssize_t *next_idx_ptr);
84static PyObject *
85_build_rval_index_tuple(PyObject *rval, Py_ssize_t idx);
86static PyObject *
87scanner_new(PyTypeObject *type, PyObject *args, PyObject *kwds);
88static int
89scanner_init(PyObject *self, PyObject *args, PyObject *kwds);
90static void
91scanner_dealloc(PyObject *self);
92static int
93scanner_clear(PyObject *self);
94static PyObject *
95encoder_new(PyTypeObject *type, PyObject *args, PyObject *kwds);
96static int
97encoder_init(PyObject *self, PyObject *args, PyObject *kwds);
98static void
99encoder_dealloc(PyObject *self);
100static int
101encoder_clear(PyObject *self);
102static int
103encoder_listencode_list(PyEncoderObject *s, PyObject *rval, PyObject *seq, Py_ssize_t indent_level);
104static int
105encoder_listencode_obj(PyEncoderObject *s, PyObject *rval, PyObject *obj, Py_ssize_t indent_level);
106static int
107encoder_listencode_dict(PyEncoderObject *s, PyObject *rval, PyObject *dct, Py_ssize_t indent_level);
108static PyObject *
Hirokazu Yamamotofecf5d12009-05-02 15:55:19 +0000109_encoded_const(PyObject *obj);
Benjamin Petersonc6b607d2009-05-02 12:36:44 +0000110static void
111raise_errmsg(char *msg, PyObject *s, Py_ssize_t end);
112static PyObject *
113encoder_encode_string(PyEncoderObject *s, PyObject *obj);
114static int
115_convertPyInt_AsSsize_t(PyObject *o, Py_ssize_t *size_ptr);
116static PyObject *
117_convertPyInt_FromSsize_t(Py_ssize_t *size_ptr);
118static PyObject *
119encoder_encode_float(PyEncoderObject *s, PyObject *obj);
120
Christian Heimes90540002008-05-08 14:29:10 +0000121#define S_CHAR(c) (c >= ' ' && c <= '~' && c != '\\' && c != '"')
Benjamin Petersonc6b607d2009-05-02 12:36:44 +0000122#define IS_WHITESPACE(c) (((c) == ' ') || ((c) == '\t') || ((c) == '\n') || ((c) == '\r'))
Christian Heimes90540002008-05-08 14:29:10 +0000123
Benjamin Petersonc6b607d2009-05-02 12:36:44 +0000124#define MIN_EXPANSION 6
Christian Heimes90540002008-05-08 14:29:10 +0000125#ifdef Py_UNICODE_WIDE
126#define MAX_EXPANSION (2 * MIN_EXPANSION)
127#else
128#define MAX_EXPANSION MIN_EXPANSION
129#endif
130
Benjamin Petersonc6b607d2009-05-02 12:36:44 +0000131static int
132_convertPyInt_AsSsize_t(PyObject *o, Py_ssize_t *size_ptr)
Christian Heimes90540002008-05-08 14:29:10 +0000133{
Benjamin Petersonc6b607d2009-05-02 12:36:44 +0000134 /* PyObject to Py_ssize_t converter */
135 *size_ptr = PyLong_AsSsize_t(o);
Georg Brandl59682052009-05-05 07:52:05 +0000136 if (*size_ptr == -1 && PyErr_Occurred())
137 return 0;
138 return 1;
Benjamin Petersonc6b607d2009-05-02 12:36:44 +0000139}
140
141static PyObject *
142_convertPyInt_FromSsize_t(Py_ssize_t *size_ptr)
143{
144 /* Py_ssize_t to PyObject converter */
145 return PyLong_FromSsize_t(*size_ptr);
146}
147
148static Py_ssize_t
149ascii_escape_unichar(Py_UNICODE c, Py_UNICODE *output, Py_ssize_t chars)
150{
151 /* Escape unicode code point c to ASCII escape sequences
152 in char *output. output must have at least 12 bytes unused to
153 accommodate an escaped surrogate pair "\uXXXX\uXXXX" */
Christian Heimes90540002008-05-08 14:29:10 +0000154 output[chars++] = '\\';
155 switch (c) {
Benjamin Petersonc6b607d2009-05-02 12:36:44 +0000156 case '\\': output[chars++] = c; break;
157 case '"': output[chars++] = c; break;
Christian Heimes90540002008-05-08 14:29:10 +0000158 case '\b': output[chars++] = 'b'; break;
159 case '\f': output[chars++] = 'f'; break;
160 case '\n': output[chars++] = 'n'; break;
161 case '\r': output[chars++] = 'r'; break;
162 case '\t': output[chars++] = 't'; break;
163 default:
164#ifdef Py_UNICODE_WIDE
165 if (c >= 0x10000) {
166 /* UTF-16 surrogate pair */
167 Py_UNICODE v = c - 0x10000;
168 c = 0xd800 | ((v >> 10) & 0x3ff);
169 output[chars++] = 'u';
Benjamin Petersonc6b607d2009-05-02 12:36:44 +0000170 output[chars++] = "0123456789abcdef"[(c >> 12) & 0xf];
171 output[chars++] = "0123456789abcdef"[(c >> 8) & 0xf];
172 output[chars++] = "0123456789abcdef"[(c >> 4) & 0xf];
173 output[chars++] = "0123456789abcdef"[(c ) & 0xf];
Christian Heimes90540002008-05-08 14:29:10 +0000174 c = 0xdc00 | (v & 0x3ff);
175 output[chars++] = '\\';
176 }
177#endif
178 output[chars++] = 'u';
Benjamin Petersonc6b607d2009-05-02 12:36:44 +0000179 output[chars++] = "0123456789abcdef"[(c >> 12) & 0xf];
180 output[chars++] = "0123456789abcdef"[(c >> 8) & 0xf];
181 output[chars++] = "0123456789abcdef"[(c >> 4) & 0xf];
182 output[chars++] = "0123456789abcdef"[(c ) & 0xf];
Christian Heimes90540002008-05-08 14:29:10 +0000183 }
184 return chars;
185}
186
187static PyObject *
188ascii_escape_unicode(PyObject *pystr)
189{
Benjamin Petersonc6b607d2009-05-02 12:36:44 +0000190 /* Take a PyUnicode pystr and return a new ASCII-only escaped PyUnicode */
Christian Heimes90540002008-05-08 14:29:10 +0000191 Py_ssize_t i;
192 Py_ssize_t input_chars;
193 Py_ssize_t output_size;
Benjamin Petersonc6b607d2009-05-02 12:36:44 +0000194 Py_ssize_t max_output_size;
Christian Heimes90540002008-05-08 14:29:10 +0000195 Py_ssize_t chars;
196 PyObject *rval;
Benjamin Petersonc6b607d2009-05-02 12:36:44 +0000197 Py_UNICODE *output;
Christian Heimes90540002008-05-08 14:29:10 +0000198 Py_UNICODE *input_unicode;
199
200 input_chars = PyUnicode_GET_SIZE(pystr);
201 input_unicode = PyUnicode_AS_UNICODE(pystr);
Benjamin Petersonc6b607d2009-05-02 12:36:44 +0000202
Christian Heimes90540002008-05-08 14:29:10 +0000203 /* One char input can be up to 6 chars output, estimate 4 of these */
204 output_size = 2 + (MIN_EXPANSION * 4) + input_chars;
Benjamin Petersonc6b607d2009-05-02 12:36:44 +0000205 max_output_size = 2 + (input_chars * MAX_EXPANSION);
206 rval = PyUnicode_FromStringAndSize(NULL, output_size);
Christian Heimes90540002008-05-08 14:29:10 +0000207 if (rval == NULL) {
208 return NULL;
209 }
Benjamin Petersonc6b607d2009-05-02 12:36:44 +0000210 output = PyUnicode_AS_UNICODE(rval);
Christian Heimes90540002008-05-08 14:29:10 +0000211 chars = 0;
212 output[chars++] = '"';
213 for (i = 0; i < input_chars; i++) {
214 Py_UNICODE c = input_unicode[i];
215 if (S_CHAR(c)) {
Benjamin Petersonc6b607d2009-05-02 12:36:44 +0000216 output[chars++] = c;
Christian Heimes90540002008-05-08 14:29:10 +0000217 }
Benjamin Petersonc6b607d2009-05-02 12:36:44 +0000218 else {
219 chars = ascii_escape_unichar(c, output, chars);
Christian Heimes90540002008-05-08 14:29:10 +0000220 }
221 if (output_size - chars < (1 + MAX_EXPANSION)) {
222 /* There's more than four, so let's resize by a lot */
Benjamin Petersonc6b607d2009-05-02 12:36:44 +0000223 Py_ssize_t new_output_size = output_size * 2;
Christian Heimes90540002008-05-08 14:29:10 +0000224 /* This is an upper bound */
Benjamin Petersonc6b607d2009-05-02 12:36:44 +0000225 if (new_output_size > max_output_size) {
226 new_output_size = max_output_size;
Christian Heimes90540002008-05-08 14:29:10 +0000227 }
Benjamin Petersonc6b607d2009-05-02 12:36:44 +0000228 /* Make sure that the output size changed before resizing */
229 if (new_output_size != output_size) {
230 output_size = new_output_size;
231 if (PyUnicode_Resize(&rval, output_size) == -1) {
232 return NULL;
233 }
234 output = PyUnicode_AS_UNICODE(rval);
Christian Heimes90540002008-05-08 14:29:10 +0000235 }
Christian Heimes90540002008-05-08 14:29:10 +0000236 }
237 }
238 output[chars++] = '"';
Benjamin Petersonc6b607d2009-05-02 12:36:44 +0000239 if (PyUnicode_Resize(&rval, chars) == -1) {
Christian Heimes90540002008-05-08 14:29:10 +0000240 return NULL;
241 }
242 return rval;
243}
244
Benjamin Petersonc6b607d2009-05-02 12:36:44 +0000245static void
Christian Heimes90540002008-05-08 14:29:10 +0000246raise_errmsg(char *msg, PyObject *s, Py_ssize_t end)
247{
Benjamin Petersonc6b607d2009-05-02 12:36:44 +0000248 /* Use the Python function json.decoder.errmsg to raise a nice
249 looking ValueError exception */
Christian Heimes90540002008-05-08 14:29:10 +0000250 static PyObject *errmsg_fn = NULL;
251 PyObject *pymsg;
252 if (errmsg_fn == NULL) {
253 PyObject *decoder = PyImport_ImportModule("json.decoder");
254 if (decoder == NULL)
255 return;
256 errmsg_fn = PyObject_GetAttrString(decoder, "errmsg");
Benjamin Petersonc6b607d2009-05-02 12:36:44 +0000257 Py_DECREF(decoder);
Christian Heimes90540002008-05-08 14:29:10 +0000258 if (errmsg_fn == NULL)
259 return;
Christian Heimes90540002008-05-08 14:29:10 +0000260 }
Benjamin Petersonc6b607d2009-05-02 12:36:44 +0000261 pymsg = PyObject_CallFunction(errmsg_fn, "(zOO&)", msg, s, _convertPyInt_FromSsize_t, &end);
Benjamin Petersona13d4752008-10-16 21:17:24 +0000262 if (pymsg) {
263 PyErr_SetObject(PyExc_ValueError, pymsg);
264 Py_DECREF(pymsg);
265 }
Christian Heimes90540002008-05-08 14:29:10 +0000266}
267
268static PyObject *
269join_list_unicode(PyObject *lst)
270{
Benjamin Petersonc6b607d2009-05-02 12:36:44 +0000271 /* return u''.join(lst) */
272 static PyObject *sep = NULL;
273 if (sep == NULL) {
274 sep = PyUnicode_FromStringAndSize("", 0);
275 if (sep == NULL)
276 return NULL;
Christian Heimes90540002008-05-08 14:29:10 +0000277 }
Benjamin Petersonc6b607d2009-05-02 12:36:44 +0000278 return PyUnicode_Join(sep, lst);
279}
280
281static PyObject *
282_build_rval_index_tuple(PyObject *rval, Py_ssize_t idx) {
283 /* return (rval, idx) tuple, stealing reference to rval */
284 PyObject *tpl;
285 PyObject *pyidx;
286 /*
287 steal a reference to rval, returns (rval, idx)
288 */
289 if (rval == NULL) {
Christian Heimes90540002008-05-08 14:29:10 +0000290 return NULL;
291 }
Benjamin Petersonc6b607d2009-05-02 12:36:44 +0000292 pyidx = PyLong_FromSsize_t(idx);
293 if (pyidx == NULL) {
294 Py_DECREF(rval);
295 return NULL;
296 }
297 tpl = PyTuple_New(2);
298 if (tpl == NULL) {
299 Py_DECREF(pyidx);
300 Py_DECREF(rval);
301 return NULL;
302 }
303 PyTuple_SET_ITEM(tpl, 0, rval);
304 PyTuple_SET_ITEM(tpl, 1, pyidx);
305 return tpl;
Christian Heimes90540002008-05-08 14:29:10 +0000306}
307
308static PyObject *
Benjamin Petersonc6b607d2009-05-02 12:36:44 +0000309scanstring_unicode(PyObject *pystr, Py_ssize_t end, int strict, Py_ssize_t *next_end_ptr)
Christian Heimes90540002008-05-08 14:29:10 +0000310{
Benjamin Petersonc6b607d2009-05-02 12:36:44 +0000311 /* Read the JSON string from PyUnicode pystr.
312 end is the index of the first character after the quote.
313 if strict is zero then literal control characters are allowed
314 *next_end_ptr is a return-by-reference index of the character
315 after the end quote
Christian Heimes90540002008-05-08 14:29:10 +0000316
Benjamin Petersonc6b607d2009-05-02 12:36:44 +0000317 Return value is a new PyUnicode
318 */
Christian Heimes90540002008-05-08 14:29:10 +0000319 PyObject *rval;
320 Py_ssize_t len = PyUnicode_GET_SIZE(pystr);
321 Py_ssize_t begin = end - 1;
322 Py_ssize_t next = begin;
323 const Py_UNICODE *buf = PyUnicode_AS_UNICODE(pystr);
324 PyObject *chunks = PyList_New(0);
325 if (chunks == NULL) {
326 goto bail;
327 }
Benjamin Peterson7af6eec2008-07-19 22:26:35 +0000328 if (end < 0 || len <= end) {
329 PyErr_SetString(PyExc_ValueError, "end is out of bounds");
330 goto bail;
331 }
Christian Heimes90540002008-05-08 14:29:10 +0000332 while (1) {
333 /* Find the end of the string or the next escape */
334 Py_UNICODE c = 0;
335 PyObject *chunk = NULL;
336 for (next = end; next < len; next++) {
337 c = buf[next];
338 if (c == '"' || c == '\\') {
339 break;
340 }
341 else if (strict && c <= 0x1f) {
Benjamin Peterson7af6eec2008-07-19 22:26:35 +0000342 raise_errmsg("Invalid control character at", pystr, next);
Christian Heimes90540002008-05-08 14:29:10 +0000343 goto bail;
344 }
345 }
346 if (!(c == '"' || c == '\\')) {
347 raise_errmsg("Unterminated string starting at", pystr, begin);
348 goto bail;
349 }
350 /* Pick up this chunk if it's not zero length */
351 if (next != end) {
352 chunk = PyUnicode_FromUnicode(&buf[end], next - end);
353 if (chunk == NULL) {
354 goto bail;
355 }
356 if (PyList_Append(chunks, chunk)) {
Benjamin Peterson8e8c2152008-10-16 21:56:24 +0000357 Py_DECREF(chunk);
Christian Heimes90540002008-05-08 14:29:10 +0000358 goto bail;
359 }
360 Py_DECREF(chunk);
361 }
362 next++;
363 if (c == '"') {
364 end = next;
365 break;
366 }
367 if (next == len) {
368 raise_errmsg("Unterminated string starting at", pystr, begin);
369 goto bail;
370 }
371 c = buf[next];
372 if (c != 'u') {
373 /* Non-unicode backslash escapes */
374 end = next + 1;
375 switch (c) {
376 case '"': break;
377 case '\\': break;
378 case '/': break;
379 case 'b': c = '\b'; break;
380 case 'f': c = '\f'; break;
381 case 'n': c = '\n'; break;
382 case 'r': c = '\r'; break;
383 case 't': c = '\t'; break;
384 default: c = 0;
385 }
386 if (c == 0) {
387 raise_errmsg("Invalid \\escape", pystr, end - 2);
388 goto bail;
389 }
390 }
391 else {
392 c = 0;
393 next++;
394 end = next + 4;
395 if (end >= len) {
396 raise_errmsg("Invalid \\uXXXX escape", pystr, next - 1);
397 goto bail;
398 }
399 /* Decode 4 hex digits */
400 for (; next < end; next++) {
Christian Heimes90540002008-05-08 14:29:10 +0000401 Py_UNICODE digit = buf[next];
Benjamin Petersonc6b607d2009-05-02 12:36:44 +0000402 c <<= 4;
Christian Heimes90540002008-05-08 14:29:10 +0000403 switch (digit) {
404 case '0': case '1': case '2': case '3': case '4':
405 case '5': case '6': case '7': case '8': case '9':
Benjamin Petersonc6b607d2009-05-02 12:36:44 +0000406 c |= (digit - '0'); break;
Christian Heimes90540002008-05-08 14:29:10 +0000407 case 'a': case 'b': case 'c': case 'd': case 'e':
408 case 'f':
Benjamin Petersonc6b607d2009-05-02 12:36:44 +0000409 c |= (digit - 'a' + 10); break;
Christian Heimes90540002008-05-08 14:29:10 +0000410 case 'A': case 'B': case 'C': case 'D': case 'E':
411 case 'F':
Benjamin Petersonc6b607d2009-05-02 12:36:44 +0000412 c |= (digit - 'A' + 10); break;
Christian Heimes90540002008-05-08 14:29:10 +0000413 default:
414 raise_errmsg("Invalid \\uXXXX escape", pystr, end - 5);
415 goto bail;
416 }
417 }
418#ifdef Py_UNICODE_WIDE
419 /* Surrogate pair */
Benjamin Petersonc6b607d2009-05-02 12:36:44 +0000420 if ((c & 0xfc00) == 0xd800) {
Christian Heimes90540002008-05-08 14:29:10 +0000421 Py_UNICODE c2 = 0;
422 if (end + 6 >= len) {
Benjamin Petersonc6b607d2009-05-02 12:36:44 +0000423 raise_errmsg("Unpaired high surrogate", pystr, end - 5);
424 goto bail;
Christian Heimes90540002008-05-08 14:29:10 +0000425 }
426 if (buf[next++] != '\\' || buf[next++] != 'u') {
Benjamin Petersonc6b607d2009-05-02 12:36:44 +0000427 raise_errmsg("Unpaired high surrogate", pystr, end - 5);
428 goto bail;
Christian Heimes90540002008-05-08 14:29:10 +0000429 }
430 end += 6;
431 /* Decode 4 hex digits */
432 for (; next < end; next++) {
Benjamin Petersonc6b607d2009-05-02 12:36:44 +0000433 c2 <<= 4;
Christian Heimes90540002008-05-08 14:29:10 +0000434 Py_UNICODE digit = buf[next];
435 switch (digit) {
436 case '0': case '1': case '2': case '3': case '4':
437 case '5': case '6': case '7': case '8': case '9':
Benjamin Petersonc6b607d2009-05-02 12:36:44 +0000438 c2 |= (digit - '0'); break;
Christian Heimes90540002008-05-08 14:29:10 +0000439 case 'a': case 'b': case 'c': case 'd': case 'e':
440 case 'f':
Benjamin Petersonc6b607d2009-05-02 12:36:44 +0000441 c2 |= (digit - 'a' + 10); break;
Christian Heimes90540002008-05-08 14:29:10 +0000442 case 'A': case 'B': case 'C': case 'D': case 'E':
443 case 'F':
Benjamin Petersonc6b607d2009-05-02 12:36:44 +0000444 c2 |= (digit - 'A' + 10); break;
Christian Heimes90540002008-05-08 14:29:10 +0000445 default:
446 raise_errmsg("Invalid \\uXXXX escape", pystr, end - 5);
447 goto bail;
448 }
449 }
Benjamin Petersonc6b607d2009-05-02 12:36:44 +0000450 if ((c2 & 0xfc00) != 0xdc00) {
451 raise_errmsg("Unpaired high surrogate", pystr, end - 5);
452 goto bail;
453 }
Christian Heimes90540002008-05-08 14:29:10 +0000454 c = 0x10000 + (((c - 0xd800) << 10) | (c2 - 0xdc00));
455 }
Benjamin Petersonc6b607d2009-05-02 12:36:44 +0000456 else if ((c & 0xfc00) == 0xdc00) {
457 raise_errmsg("Unpaired low surrogate", pystr, end - 5);
458 goto bail;
459 }
Christian Heimes90540002008-05-08 14:29:10 +0000460#endif
461 }
462 chunk = PyUnicode_FromUnicode(&c, 1);
463 if (chunk == NULL) {
464 goto bail;
465 }
466 if (PyList_Append(chunks, chunk)) {
Benjamin Peterson8e8c2152008-10-16 21:56:24 +0000467 Py_DECREF(chunk);
Christian Heimes90540002008-05-08 14:29:10 +0000468 goto bail;
469 }
470 Py_DECREF(chunk);
471 }
472
473 rval = join_list_unicode(chunks);
474 if (rval == NULL) {
475 goto bail;
476 }
Benjamin Petersonc6b607d2009-05-02 12:36:44 +0000477 Py_DECREF(chunks);
478 *next_end_ptr = end;
479 return rval;
Christian Heimes90540002008-05-08 14:29:10 +0000480bail:
Benjamin Petersonc6b607d2009-05-02 12:36:44 +0000481 *next_end_ptr = -1;
Christian Heimes90540002008-05-08 14:29:10 +0000482 Py_XDECREF(chunks);
483 return NULL;
484}
485
486PyDoc_STRVAR(pydoc_scanstring,
Benjamin Petersonc6b607d2009-05-02 12:36:44 +0000487 "scanstring(basestring, end, strict=True) -> (bytes, end)\n"
488 "\n"
489 "Scan the string s for a JSON string. End is the index of the\n"
490 "character in s after the quote that started the JSON string.\n"
491 "Unescapes all valid JSON string escape sequences and raises ValueError\n"
492 "on attempt to decode an invalid string. If strict is False then literal\n"
493 "control characters are allowed in the string.\n"
494 "\n"
495 "Returns a tuple of the decoded string and the index of the character in s\n"
496 "after the end quote."
497);
Christian Heimes90540002008-05-08 14:29:10 +0000498
499static PyObject *
Benjamin Petersonc6b607d2009-05-02 12:36:44 +0000500py_scanstring(PyObject* self UNUSED, PyObject *args)
Christian Heimes90540002008-05-08 14:29:10 +0000501{
502 PyObject *pystr;
Benjamin Petersonc6b607d2009-05-02 12:36:44 +0000503 PyObject *rval;
Christian Heimes90540002008-05-08 14:29:10 +0000504 Py_ssize_t end;
Benjamin Petersonc6b607d2009-05-02 12:36:44 +0000505 Py_ssize_t next_end = -1;
506 int strict = 1;
507 if (!PyArg_ParseTuple(args, "OO&|i:scanstring", &pystr, _convertPyInt_AsSsize_t, &end, &strict)) {
Christian Heimes90540002008-05-08 14:29:10 +0000508 return NULL;
509 }
Benjamin Petersonc6b607d2009-05-02 12:36:44 +0000510 if (PyUnicode_Check(pystr)) {
511 rval = scanstring_unicode(pystr, end, strict, &next_end);
Christian Heimes90540002008-05-08 14:29:10 +0000512 }
513 else {
514 PyErr_Format(PyExc_TypeError,
515 "first argument must be a string or bytes, not %.80s",
516 Py_TYPE(pystr)->tp_name);
517 return NULL;
518 }
Benjamin Petersonc6b607d2009-05-02 12:36:44 +0000519 return _build_rval_index_tuple(rval, next_end);
Christian Heimes90540002008-05-08 14:29:10 +0000520}
521
522PyDoc_STRVAR(pydoc_encode_basestring_ascii,
Benjamin Petersonc6b607d2009-05-02 12:36:44 +0000523 "encode_basestring_ascii(basestring) -> bytes\n"
524 "\n"
525 "Return an ASCII-only JSON representation of a Python string"
526);
Christian Heimes90540002008-05-08 14:29:10 +0000527
528static PyObject *
Benjamin Petersonc6b607d2009-05-02 12:36:44 +0000529py_encode_basestring_ascii(PyObject* self UNUSED, PyObject *pystr)
Christian Heimes90540002008-05-08 14:29:10 +0000530{
531 PyObject *rval;
Benjamin Petersonc6b607d2009-05-02 12:36:44 +0000532 /* Return an ASCII-only JSON representation of a Python string */
Christian Heimes90540002008-05-08 14:29:10 +0000533 /* METH_O */
Benjamin Petersonc6b607d2009-05-02 12:36:44 +0000534 if (PyUnicode_Check(pystr)) {
Christian Heimes90540002008-05-08 14:29:10 +0000535 rval = ascii_escape_unicode(pystr);
536 }
537 else {
Benjamin Petersonc6b607d2009-05-02 12:36:44 +0000538 PyErr_Format(PyExc_TypeError,
539 "first argument must be a string, not %.80s",
Christian Heimes90540002008-05-08 14:29:10 +0000540 Py_TYPE(pystr)->tp_name);
541 return NULL;
542 }
Christian Heimes90540002008-05-08 14:29:10 +0000543 return rval;
544}
545
Benjamin Petersonc6b607d2009-05-02 12:36:44 +0000546static void
547scanner_dealloc(PyObject *self)
548{
549 /* Deallocate scanner object */
550 scanner_clear(self);
551 Py_TYPE(self)->tp_free(self);
552}
553
554static int
555scanner_traverse(PyObject *self, visitproc visit, void *arg)
556{
557 PyScannerObject *s;
558 assert(PyScanner_Check(self));
559 s = (PyScannerObject *)self;
560 Py_VISIT(s->strict);
561 Py_VISIT(s->object_hook);
562 Py_VISIT(s->object_pairs_hook);
563 Py_VISIT(s->parse_float);
564 Py_VISIT(s->parse_int);
565 Py_VISIT(s->parse_constant);
566 return 0;
567}
568
569static int
570scanner_clear(PyObject *self)
571{
572 PyScannerObject *s;
573 assert(PyScanner_Check(self));
574 s = (PyScannerObject *)self;
575 Py_CLEAR(s->strict);
576 Py_CLEAR(s->object_hook);
577 Py_CLEAR(s->object_pairs_hook);
578 Py_CLEAR(s->parse_float);
579 Py_CLEAR(s->parse_int);
580 Py_CLEAR(s->parse_constant);
581 return 0;
582}
583
584static PyObject *
585_parse_object_unicode(PyScannerObject *s, PyObject *pystr, Py_ssize_t idx, Py_ssize_t *next_idx_ptr) {
586 /* Read a JSON object from PyUnicode pystr.
587 idx is the index of the first character after the opening curly brace.
588 *next_idx_ptr is a return-by-reference index to the first character after
589 the closing curly brace.
590
591 Returns a new PyObject (usually a dict, but object_hook can change that)
592 */
593 Py_UNICODE *str = PyUnicode_AS_UNICODE(pystr);
594 Py_ssize_t end_idx = PyUnicode_GET_SIZE(pystr) - 1;
595 PyObject *val = NULL;
596 PyObject *rval = PyList_New(0);
597 PyObject *key = NULL;
598 int strict = PyObject_IsTrue(s->strict);
599 Py_ssize_t next_idx;
600 if (rval == NULL)
601 return NULL;
602
603 /* skip whitespace after { */
604 while (idx <= end_idx && IS_WHITESPACE(str[idx])) idx++;
605
606 /* only loop if the object is non-empty */
607 if (idx <= end_idx && str[idx] != '}') {
608 while (idx <= end_idx) {
609 /* read key */
610 if (str[idx] != '"') {
611 raise_errmsg("Expecting property name", pystr, idx);
612 goto bail;
613 }
614 key = scanstring_unicode(pystr, idx + 1, strict, &next_idx);
615 if (key == NULL)
616 goto bail;
617 idx = next_idx;
618
619 /* skip whitespace between key and : delimiter, read :, skip whitespace */
620 while (idx <= end_idx && IS_WHITESPACE(str[idx])) idx++;
621 if (idx > end_idx || str[idx] != ':') {
622 raise_errmsg("Expecting : delimiter", pystr, idx);
623 goto bail;
624 }
625 idx++;
626 while (idx <= end_idx && IS_WHITESPACE(str[idx])) idx++;
627
628 /* read any JSON term */
629 val = scan_once_unicode(s, pystr, idx, &next_idx);
630 if (val == NULL)
631 goto bail;
632
633 {
634 PyObject *tuple = PyTuple_Pack(2, key, val);
635 if (tuple == NULL)
636 goto bail;
637 if (PyList_Append(rval, tuple) == -1) {
638 Py_DECREF(tuple);
639 goto bail;
640 }
641 Py_DECREF(tuple);
642 }
643
644 Py_CLEAR(key);
645 Py_CLEAR(val);
646 idx = next_idx;
647
648 /* skip whitespace before } or , */
649 while (idx <= end_idx && IS_WHITESPACE(str[idx])) idx++;
650
651 /* bail if the object is closed or we didn't get the , delimiter */
652 if (idx > end_idx) break;
653 if (str[idx] == '}') {
654 break;
655 }
656 else if (str[idx] != ',') {
657 raise_errmsg("Expecting , delimiter", pystr, idx);
658 goto bail;
659 }
660 idx++;
661
662 /* skip whitespace after , delimiter */
663 while (idx <= end_idx && IS_WHITESPACE(str[idx])) idx++;
664 }
665 }
666
667 /* verify that idx < end_idx, str[idx] should be '}' */
668 if (idx > end_idx || str[idx] != '}') {
669 raise_errmsg("Expecting object", pystr, end_idx);
670 goto bail;
671 }
672
673 *next_idx_ptr = idx + 1;
674
675 if (s->object_pairs_hook != Py_None) {
676 val = PyObject_CallFunctionObjArgs(s->object_pairs_hook, rval, NULL);
677 if (val == NULL)
678 goto bail;
679 Py_DECREF(rval);
680 return val;
681 }
682
683 val = PyDict_New();
684 if (val == NULL)
685 goto bail;
686 if (PyDict_MergeFromSeq2(val, rval, 1) == -1)
687 goto bail;
688 Py_DECREF(rval);
689 rval = val;
690
691 /* if object_hook is not None: rval = object_hook(rval) */
692 if (s->object_hook != Py_None) {
693 val = PyObject_CallFunctionObjArgs(s->object_hook, rval, NULL);
694 if (val == NULL)
695 goto bail;
696 Py_DECREF(rval);
697 rval = val;
698 val = NULL;
699 }
700 return rval;
701bail:
702 Py_XDECREF(key);
703 Py_XDECREF(val);
704 Py_DECREF(rval);
705 return NULL;
706}
707
708static PyObject *
709_parse_array_unicode(PyScannerObject *s, PyObject *pystr, Py_ssize_t idx, Py_ssize_t *next_idx_ptr) {
710 /* Read a JSON array from PyString pystr.
711 idx is the index of the first character after the opening brace.
712 *next_idx_ptr is a return-by-reference index to the first character after
713 the closing brace.
714
715 Returns a new PyList
716 */
717 Py_UNICODE *str = PyUnicode_AS_UNICODE(pystr);
718 Py_ssize_t end_idx = PyUnicode_GET_SIZE(pystr) - 1;
719 PyObject *val = NULL;
720 PyObject *rval = PyList_New(0);
721 Py_ssize_t next_idx;
722 if (rval == NULL)
723 return NULL;
724
725 /* skip whitespace after [ */
726 while (idx <= end_idx && IS_WHITESPACE(str[idx])) idx++;
727
728 /* only loop if the array is non-empty */
729 if (idx <= end_idx && str[idx] != ']') {
730 while (idx <= end_idx) {
731
732 /* read any JSON term */
733 val = scan_once_unicode(s, pystr, idx, &next_idx);
734 if (val == NULL)
735 goto bail;
736
737 if (PyList_Append(rval, val) == -1)
738 goto bail;
739
740 Py_CLEAR(val);
741 idx = next_idx;
742
743 /* skip whitespace between term and , */
744 while (idx <= end_idx && IS_WHITESPACE(str[idx])) idx++;
745
746 /* bail if the array is closed or we didn't get the , delimiter */
747 if (idx > end_idx) break;
748 if (str[idx] == ']') {
749 break;
750 }
751 else if (str[idx] != ',') {
752 raise_errmsg("Expecting , delimiter", pystr, idx);
753 goto bail;
754 }
755 idx++;
756
757 /* skip whitespace after , */
758 while (idx <= end_idx && IS_WHITESPACE(str[idx])) idx++;
759 }
760 }
761
762 /* verify that idx < end_idx, str[idx] should be ']' */
763 if (idx > end_idx || str[idx] != ']') {
764 raise_errmsg("Expecting object", pystr, end_idx);
765 goto bail;
766 }
767 *next_idx_ptr = idx + 1;
768 return rval;
769bail:
770 Py_XDECREF(val);
771 Py_DECREF(rval);
772 return NULL;
773}
774
775static PyObject *
776_parse_constant(PyScannerObject *s, char *constant, Py_ssize_t idx, Py_ssize_t *next_idx_ptr) {
777 /* Read a JSON constant from PyString pystr.
778 constant is the constant string that was found
779 ("NaN", "Infinity", "-Infinity").
780 idx is the index of the first character of the constant
781 *next_idx_ptr is a return-by-reference index to the first character after
782 the constant.
783
784 Returns the result of parse_constant
785 */
786 PyObject *cstr;
787 PyObject *rval;
788 /* constant is "NaN", "Infinity", or "-Infinity" */
789 cstr = PyUnicode_InternFromString(constant);
790 if (cstr == NULL)
791 return NULL;
792
793 /* rval = parse_constant(constant) */
794 rval = PyObject_CallFunctionObjArgs(s->parse_constant, cstr, NULL);
795 idx += PyUnicode_GET_SIZE(cstr);
796 Py_DECREF(cstr);
797 *next_idx_ptr = idx;
798 return rval;
799}
800
801static PyObject *
802_match_number_unicode(PyScannerObject *s, PyObject *pystr, Py_ssize_t start, Py_ssize_t *next_idx_ptr) {
803 /* Read a JSON number from PyUnicode pystr.
804 idx is the index of the first character of the number
805 *next_idx_ptr is a return-by-reference index to the first character after
806 the number.
807
808 Returns a new PyObject representation of that number:
809 PyInt, PyLong, or PyFloat.
810 May return other types if parse_int or parse_float are set
811 */
812 Py_UNICODE *str = PyUnicode_AS_UNICODE(pystr);
813 Py_ssize_t end_idx = PyUnicode_GET_SIZE(pystr) - 1;
814 Py_ssize_t idx = start;
815 int is_float = 0;
816 PyObject *rval;
817 PyObject *numstr;
818
819 /* read a sign if it's there, make sure it's not the end of the string */
820 if (str[idx] == '-') {
821 idx++;
822 if (idx > end_idx) {
823 PyErr_SetNone(PyExc_StopIteration);
824 return NULL;
825 }
826 }
827
828 /* read as many integer digits as we find as long as it doesn't start with 0 */
829 if (str[idx] >= '1' && str[idx] <= '9') {
830 idx++;
831 while (idx <= end_idx && str[idx] >= '0' && str[idx] <= '9') idx++;
832 }
833 /* if it starts with 0 we only expect one integer digit */
834 else if (str[idx] == '0') {
835 idx++;
836 }
837 /* no integer digits, error */
838 else {
839 PyErr_SetNone(PyExc_StopIteration);
840 return NULL;
841 }
842
843 /* if the next char is '.' followed by a digit then read all float digits */
844 if (idx < end_idx && str[idx] == '.' && str[idx + 1] >= '0' && str[idx + 1] <= '9') {
845 is_float = 1;
846 idx += 2;
847 while (idx <= end_idx && str[idx] >= '0' && str[idx] <= '9') idx++;
848 }
849
850 /* if the next char is 'e' or 'E' then maybe read the exponent (or backtrack) */
851 if (idx < end_idx && (str[idx] == 'e' || str[idx] == 'E')) {
852 Py_ssize_t e_start = idx;
853 idx++;
854
855 /* read an exponent sign if present */
856 if (idx < end_idx && (str[idx] == '-' || str[idx] == '+')) idx++;
857
858 /* read all digits */
859 while (idx <= end_idx && str[idx] >= '0' && str[idx] <= '9') idx++;
860
861 /* if we got a digit, then parse as float. if not, backtrack */
862 if (str[idx - 1] >= '0' && str[idx - 1] <= '9') {
863 is_float = 1;
864 }
865 else {
866 idx = e_start;
867 }
868 }
869
870 /* copy the section we determined to be a number */
871 numstr = PyUnicode_FromUnicode(&str[start], idx - start);
872 if (numstr == NULL)
873 return NULL;
874 if (is_float) {
875 /* parse as a float using a fast path if available, otherwise call user defined method */
876 if (s->parse_float != (PyObject *)&PyFloat_Type) {
877 rval = PyObject_CallFunctionObjArgs(s->parse_float, numstr, NULL);
878 }
879 else {
880 rval = PyFloat_FromString(numstr);
881 }
882 }
883 else {
884 /* no fast path for unicode -> int, just call */
885 rval = PyObject_CallFunctionObjArgs(s->parse_int, numstr, NULL);
886 }
887 Py_DECREF(numstr);
888 *next_idx_ptr = idx;
889 return rval;
890}
891
892static PyObject *
893scan_once_unicode(PyScannerObject *s, PyObject *pystr, Py_ssize_t idx, Py_ssize_t *next_idx_ptr)
894{
895 /* Read one JSON term (of any kind) from PyUnicode pystr.
896 idx is the index of the first character of the term
897 *next_idx_ptr is a return-by-reference index to the first character after
898 the number.
899
900 Returns a new PyObject representation of the term.
901 */
902 Py_UNICODE *str = PyUnicode_AS_UNICODE(pystr);
903 Py_ssize_t length = PyUnicode_GET_SIZE(pystr);
904 if (idx >= length) {
905 PyErr_SetNone(PyExc_StopIteration);
906 return NULL;
907 }
908 switch (str[idx]) {
909 case '"':
910 /* string */
911 return scanstring_unicode(pystr, idx + 1,
912 PyObject_IsTrue(s->strict),
913 next_idx_ptr);
914 case '{':
915 /* object */
916 return _parse_object_unicode(s, pystr, idx + 1, next_idx_ptr);
917 case '[':
918 /* array */
919 return _parse_array_unicode(s, pystr, idx + 1, next_idx_ptr);
920 case 'n':
921 /* null */
922 if ((idx + 3 < length) && str[idx + 1] == 'u' && str[idx + 2] == 'l' && str[idx + 3] == 'l') {
923 Py_INCREF(Py_None);
924 *next_idx_ptr = idx + 4;
925 return Py_None;
926 }
927 break;
928 case 't':
929 /* true */
930 if ((idx + 3 < length) && str[idx + 1] == 'r' && str[idx + 2] == 'u' && str[idx + 3] == 'e') {
931 Py_INCREF(Py_True);
932 *next_idx_ptr = idx + 4;
933 return Py_True;
934 }
935 break;
936 case 'f':
937 /* false */
938 if ((idx + 4 < length) && str[idx + 1] == 'a' && str[idx + 2] == 'l' && str[idx + 3] == 's' && str[idx + 4] == 'e') {
939 Py_INCREF(Py_False);
940 *next_idx_ptr = idx + 5;
941 return Py_False;
942 }
943 break;
944 case 'N':
945 /* NaN */
946 if ((idx + 2 < length) && str[idx + 1] == 'a' && str[idx + 2] == 'N') {
947 return _parse_constant(s, "NaN", idx, next_idx_ptr);
948 }
949 break;
950 case 'I':
951 /* Infinity */
952 if ((idx + 7 < length) && str[idx + 1] == 'n' && str[idx + 2] == 'f' && str[idx + 3] == 'i' && str[idx + 4] == 'n' && str[idx + 5] == 'i' && str[idx + 6] == 't' && str[idx + 7] == 'y') {
953 return _parse_constant(s, "Infinity", idx, next_idx_ptr);
954 }
955 break;
956 case '-':
957 /* -Infinity */
958 if ((idx + 8 < length) && str[idx + 1] == 'I' && str[idx + 2] == 'n' && str[idx + 3] == 'f' && str[idx + 4] == 'i' && str[idx + 5] == 'n' && str[idx + 6] == 'i' && str[idx + 7] == 't' && str[idx + 8] == 'y') {
959 return _parse_constant(s, "-Infinity", idx, next_idx_ptr);
960 }
961 break;
962 }
963 /* Didn't find a string, object, array, or named constant. Look for a number. */
964 return _match_number_unicode(s, pystr, idx, next_idx_ptr);
965}
966
967static PyObject *
968scanner_call(PyObject *self, PyObject *args, PyObject *kwds)
969{
970 /* Python callable interface to scan_once_{str,unicode} */
971 PyObject *pystr;
972 PyObject *rval;
973 Py_ssize_t idx;
974 Py_ssize_t next_idx = -1;
975 static char *kwlist[] = {"string", "idx", NULL};
976 PyScannerObject *s;
977 assert(PyScanner_Check(self));
978 s = (PyScannerObject *)self;
979 if (!PyArg_ParseTupleAndKeywords(args, kwds, "OO&:scan_once", kwlist, &pystr, _convertPyInt_AsSsize_t, &idx))
980 return NULL;
981
982 if (PyUnicode_Check(pystr)) {
983 rval = scan_once_unicode(s, pystr, idx, &next_idx);
984 }
985 else {
986 PyErr_Format(PyExc_TypeError,
987 "first argument must be a string, not %.80s",
988 Py_TYPE(pystr)->tp_name);
989 return NULL;
990 }
991 return _build_rval_index_tuple(rval, next_idx);
992}
993
994static PyObject *
995scanner_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
996{
997 PyScannerObject *s;
998 s = (PyScannerObject *)type->tp_alloc(type, 0);
999 if (s != NULL) {
1000 s->strict = NULL;
1001 s->object_hook = NULL;
1002 s->object_pairs_hook = NULL;
1003 s->parse_float = NULL;
1004 s->parse_int = NULL;
1005 s->parse_constant = NULL;
1006 }
1007 return (PyObject *)s;
1008}
1009
1010static int
1011scanner_init(PyObject *self, PyObject *args, PyObject *kwds)
1012{
1013 /* Initialize Scanner object */
1014 PyObject *ctx;
1015 static char *kwlist[] = {"context", NULL};
1016 PyScannerObject *s;
1017
1018 assert(PyScanner_Check(self));
1019 s = (PyScannerObject *)self;
1020
1021 if (!PyArg_ParseTupleAndKeywords(args, kwds, "O:make_scanner", kwlist, &ctx))
1022 return -1;
1023
1024 /* All of these will fail "gracefully" so we don't need to verify them */
1025 s->strict = PyObject_GetAttrString(ctx, "strict");
1026 if (s->strict == NULL)
1027 goto bail;
1028 s->object_hook = PyObject_GetAttrString(ctx, "object_hook");
1029 if (s->object_hook == NULL)
1030 goto bail;
1031 s->object_pairs_hook = PyObject_GetAttrString(ctx, "object_pairs_hook");
1032 if (s->object_pairs_hook == NULL)
1033 goto bail;
1034 s->parse_float = PyObject_GetAttrString(ctx, "parse_float");
1035 if (s->parse_float == NULL)
1036 goto bail;
1037 s->parse_int = PyObject_GetAttrString(ctx, "parse_int");
1038 if (s->parse_int == NULL)
1039 goto bail;
1040 s->parse_constant = PyObject_GetAttrString(ctx, "parse_constant");
1041 if (s->parse_constant == NULL)
1042 goto bail;
1043
1044 return 0;
1045
1046bail:
1047 Py_CLEAR(s->strict);
1048 Py_CLEAR(s->object_hook);
1049 Py_CLEAR(s->object_pairs_hook);
1050 Py_CLEAR(s->parse_float);
1051 Py_CLEAR(s->parse_int);
1052 Py_CLEAR(s->parse_constant);
1053 return -1;
1054}
1055
1056PyDoc_STRVAR(scanner_doc, "JSON scanner object");
1057
1058static
1059PyTypeObject PyScannerType = {
1060 PyVarObject_HEAD_INIT(NULL, 0)
1061 "_json.Scanner", /* tp_name */
1062 sizeof(PyScannerObject), /* tp_basicsize */
1063 0, /* tp_itemsize */
1064 scanner_dealloc, /* tp_dealloc */
1065 0, /* tp_print */
1066 0, /* tp_getattr */
1067 0, /* tp_setattr */
1068 0, /* tp_compare */
1069 0, /* tp_repr */
1070 0, /* tp_as_number */
1071 0, /* tp_as_sequence */
1072 0, /* tp_as_mapping */
1073 0, /* tp_hash */
1074 scanner_call, /* tp_call */
1075 0, /* tp_str */
1076 0,/* PyObject_GenericGetAttr, */ /* tp_getattro */
1077 0,/* PyObject_GenericSetAttr, */ /* tp_setattro */
1078 0, /* tp_as_buffer */
1079 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC, /* tp_flags */
1080 scanner_doc, /* tp_doc */
1081 scanner_traverse, /* tp_traverse */
1082 scanner_clear, /* tp_clear */
1083 0, /* tp_richcompare */
1084 0, /* tp_weaklistoffset */
1085 0, /* tp_iter */
1086 0, /* tp_iternext */
1087 0, /* tp_methods */
1088 scanner_members, /* tp_members */
1089 0, /* tp_getset */
1090 0, /* tp_base */
1091 0, /* tp_dict */
1092 0, /* tp_descr_get */
1093 0, /* tp_descr_set */
1094 0, /* tp_dictoffset */
1095 scanner_init, /* tp_init */
1096 0,/* PyType_GenericAlloc, */ /* tp_alloc */
1097 scanner_new, /* tp_new */
1098 0,/* PyObject_GC_Del, */ /* tp_free */
1099};
1100
1101static PyObject *
1102encoder_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
1103{
1104 PyEncoderObject *s;
1105 s = (PyEncoderObject *)type->tp_alloc(type, 0);
1106 if (s != NULL) {
1107 s->markers = NULL;
1108 s->defaultfn = NULL;
1109 s->encoder = NULL;
1110 s->indent = NULL;
1111 s->key_separator = NULL;
1112 s->item_separator = NULL;
1113 s->sort_keys = NULL;
1114 s->skipkeys = NULL;
1115 }
1116 return (PyObject *)s;
1117}
1118
1119static int
1120encoder_init(PyObject *self, PyObject *args, PyObject *kwds)
1121{
1122 /* initialize Encoder object */
1123 static char *kwlist[] = {"markers", "default", "encoder", "indent", "key_separator", "item_separator", "sort_keys", "skipkeys", "allow_nan", NULL};
1124
1125 PyEncoderObject *s;
1126 PyObject *allow_nan;
1127
1128 assert(PyEncoder_Check(self));
1129 s = (PyEncoderObject *)self;
1130
1131 if (!PyArg_ParseTupleAndKeywords(args, kwds, "OOOOOOOOO:make_encoder", kwlist,
1132 &s->markers, &s->defaultfn, &s->encoder, &s->indent, &s->key_separator, &s->item_separator, &s->sort_keys, &s->skipkeys, &allow_nan))
1133 return -1;
1134
1135 Py_INCREF(s->markers);
1136 Py_INCREF(s->defaultfn);
1137 Py_INCREF(s->encoder);
1138 Py_INCREF(s->indent);
1139 Py_INCREF(s->key_separator);
1140 Py_INCREF(s->item_separator);
1141 Py_INCREF(s->sort_keys);
1142 Py_INCREF(s->skipkeys);
1143 s->fast_encode = (PyCFunction_Check(s->encoder) && PyCFunction_GetFunction(s->encoder) == (PyCFunction)py_encode_basestring_ascii);
1144 s->allow_nan = PyObject_IsTrue(allow_nan);
1145 return 0;
1146}
1147
1148static PyObject *
1149encoder_call(PyObject *self, PyObject *args, PyObject *kwds)
1150{
1151 /* Python callable interface to encode_listencode_obj */
1152 static char *kwlist[] = {"obj", "_current_indent_level", NULL};
1153 PyObject *obj;
1154 PyObject *rval;
1155 Py_ssize_t indent_level;
1156 PyEncoderObject *s;
1157 assert(PyEncoder_Check(self));
1158 s = (PyEncoderObject *)self;
1159 if (!PyArg_ParseTupleAndKeywords(args, kwds, "OO&:_iterencode", kwlist,
1160 &obj, _convertPyInt_AsSsize_t, &indent_level))
1161 return NULL;
1162 rval = PyList_New(0);
1163 if (rval == NULL)
1164 return NULL;
1165 if (encoder_listencode_obj(s, rval, obj, indent_level)) {
1166 Py_DECREF(rval);
1167 return NULL;
1168 }
1169 return rval;
1170}
1171
1172static PyObject *
1173_encoded_const(PyObject *obj)
1174{
1175 /* Return the JSON string representation of None, True, False */
1176 if (obj == Py_None) {
1177 static PyObject *s_null = NULL;
1178 if (s_null == NULL) {
1179 s_null = PyUnicode_InternFromString("null");
1180 }
1181 Py_INCREF(s_null);
1182 return s_null;
1183 }
1184 else if (obj == Py_True) {
1185 static PyObject *s_true = NULL;
1186 if (s_true == NULL) {
1187 s_true = PyUnicode_InternFromString("true");
1188 }
1189 Py_INCREF(s_true);
1190 return s_true;
1191 }
1192 else if (obj == Py_False) {
1193 static PyObject *s_false = NULL;
1194 if (s_false == NULL) {
1195 s_false = PyUnicode_InternFromString("false");
1196 }
1197 Py_INCREF(s_false);
1198 return s_false;
1199 }
1200 else {
1201 PyErr_SetString(PyExc_ValueError, "not a const");
1202 return NULL;
1203 }
1204}
1205
1206static PyObject *
1207encoder_encode_float(PyEncoderObject *s, PyObject *obj)
1208{
1209 /* Return the JSON representation of a PyFloat */
1210 double i = PyFloat_AS_DOUBLE(obj);
1211 if (!Py_IS_FINITE(i)) {
1212 if (!s->allow_nan) {
1213 PyErr_SetString(PyExc_ValueError, "Out of range float values are not JSON compliant");
1214 return NULL;
1215 }
1216 if (i > 0) {
1217 return PyUnicode_FromString("Infinity");
1218 }
1219 else if (i < 0) {
1220 return PyUnicode_FromString("-Infinity");
1221 }
1222 else {
1223 return PyUnicode_FromString("NaN");
1224 }
1225 }
1226 /* Use a better float format here? */
1227 return PyObject_Repr(obj);
1228}
1229
1230static PyObject *
1231encoder_encode_string(PyEncoderObject *s, PyObject *obj)
1232{
1233 /* Return the JSON representation of a string */
1234 if (s->fast_encode)
1235 return py_encode_basestring_ascii(NULL, obj);
1236 else
1237 return PyObject_CallFunctionObjArgs(s->encoder, obj, NULL);
1238}
1239
1240static int
1241_steal_list_append(PyObject *lst, PyObject *stolen)
1242{
1243 /* Append stolen and then decrement its reference count */
1244 int rval = PyList_Append(lst, stolen);
1245 Py_DECREF(stolen);
1246 return rval;
1247}
1248
1249static int
1250encoder_listencode_obj(PyEncoderObject *s, PyObject *rval, PyObject *obj, Py_ssize_t indent_level)
1251{
1252 /* Encode Python object obj to a JSON term, rval is a PyList */
1253 PyObject *newobj;
1254 int rv;
1255
1256 if (obj == Py_None || obj == Py_True || obj == Py_False) {
1257 PyObject *cstr = _encoded_const(obj);
1258 if (cstr == NULL)
1259 return -1;
1260 return _steal_list_append(rval, cstr);
1261 }
1262 else if (PyUnicode_Check(obj))
1263 {
1264 PyObject *encoded = encoder_encode_string(s, obj);
1265 if (encoded == NULL)
1266 return -1;
1267 return _steal_list_append(rval, encoded);
1268 }
1269 else if (PyLong_Check(obj)) {
1270 PyObject *encoded = PyObject_Str(obj);
1271 if (encoded == NULL)
1272 return -1;
1273 return _steal_list_append(rval, encoded);
1274 }
1275 else if (PyFloat_Check(obj)) {
1276 PyObject *encoded = encoder_encode_float(s, obj);
1277 if (encoded == NULL)
1278 return -1;
1279 return _steal_list_append(rval, encoded);
1280 }
1281 else if (PyList_Check(obj) || PyTuple_Check(obj)) {
1282 return encoder_listencode_list(s, rval, obj, indent_level);
1283 }
1284 else if (PyDict_Check(obj)) {
1285 return encoder_listencode_dict(s, rval, obj, indent_level);
1286 }
1287 else {
1288 PyObject *ident = NULL;
1289 if (s->markers != Py_None) {
1290 int has_key;
1291 ident = PyLong_FromVoidPtr(obj);
1292 if (ident == NULL)
1293 return -1;
1294 has_key = PyDict_Contains(s->markers, ident);
1295 if (has_key) {
1296 if (has_key != -1)
1297 PyErr_SetString(PyExc_ValueError, "Circular reference detected");
1298 Py_DECREF(ident);
1299 return -1;
1300 }
1301 if (PyDict_SetItem(s->markers, ident, obj)) {
1302 Py_DECREF(ident);
1303 return -1;
1304 }
1305 }
1306 newobj = PyObject_CallFunctionObjArgs(s->defaultfn, obj, NULL);
1307 if (newobj == NULL) {
1308 Py_XDECREF(ident);
1309 return -1;
1310 }
1311 rv = encoder_listencode_obj(s, rval, newobj, indent_level);
1312 Py_DECREF(newobj);
1313 if (rv) {
1314 Py_XDECREF(ident);
1315 return -1;
1316 }
1317 if (ident != NULL) {
1318 if (PyDict_DelItem(s->markers, ident)) {
1319 Py_XDECREF(ident);
1320 return -1;
1321 }
1322 Py_XDECREF(ident);
1323 }
1324 return rv;
1325 }
1326}
1327
1328static int
1329encoder_listencode_dict(PyEncoderObject *s, PyObject *rval, PyObject *dct, Py_ssize_t indent_level)
1330{
1331 /* Encode Python dict dct a JSON term, rval is a PyList */
1332 static PyObject *open_dict = NULL;
1333 static PyObject *close_dict = NULL;
1334 static PyObject *empty_dict = NULL;
1335 PyObject *kstr = NULL;
1336 PyObject *ident = NULL;
Raymond Hettingerc8d952d2009-05-27 06:50:31 +00001337 PyObject *it = NULL;
Raymond Hettingerbcf6f922009-05-27 09:58:34 +00001338 PyObject *items;
1339 PyObject *item = NULL;
Benjamin Petersonc6b607d2009-05-02 12:36:44 +00001340 int skipkeys;
1341 Py_ssize_t idx;
Raymond Hettinger491a4cb2009-05-27 11:19:02 +00001342 PyObject *mapping;
1343 static PyObject *code = NULL;
Benjamin Petersonc6b607d2009-05-02 12:36:44 +00001344
1345 if (open_dict == NULL || close_dict == NULL || empty_dict == NULL) {
1346 open_dict = PyUnicode_InternFromString("{");
1347 close_dict = PyUnicode_InternFromString("}");
1348 empty_dict = PyUnicode_InternFromString("{}");
1349 if (open_dict == NULL || close_dict == NULL || empty_dict == NULL)
1350 return -1;
1351 }
Raymond Hettingerc8d952d2009-05-27 06:50:31 +00001352 if (Py_SIZE(dct) == 0)
Benjamin Petersonc6b607d2009-05-02 12:36:44 +00001353 return PyList_Append(rval, empty_dict);
1354
1355 if (s->markers != Py_None) {
1356 int has_key;
1357 ident = PyLong_FromVoidPtr(dct);
1358 if (ident == NULL)
1359 goto bail;
1360 has_key = PyDict_Contains(s->markers, ident);
1361 if (has_key) {
1362 if (has_key != -1)
1363 PyErr_SetString(PyExc_ValueError, "Circular reference detected");
1364 goto bail;
1365 }
1366 if (PyDict_SetItem(s->markers, ident, dct)) {
1367 goto bail;
1368 }
1369 }
1370
1371 if (PyList_Append(rval, open_dict))
1372 goto bail;
1373
1374 if (s->indent != Py_None) {
1375 /* TODO: DOES NOT RUN */
1376 indent_level += 1;
1377 /*
1378 newline_indent = '\n' + (' ' * (_indent * _current_indent_level))
1379 separator = _item_separator + newline_indent
1380 buf += newline_indent
1381 */
1382 }
1383
Raymond Hettingerbcf6f922009-05-27 09:58:34 +00001384 if (PyObject_IsTrue(s->sort_keys)) {
Raymond Hettinger491a4cb2009-05-27 11:19:02 +00001385 if (code == NULL) {
1386 code = Py_CompileString("sorted(d.items(), key=lambda kv: kv[0])",
1387 "_json.c", Py_eval_input);
1388 if (code == NULL)
1389 goto bail;
1390 }
Benjamin Petersonc6b607d2009-05-02 12:36:44 +00001391
Raymond Hettinger491a4cb2009-05-27 11:19:02 +00001392 mapping = PyDict_New();
1393 if (mapping == NULL)
1394 goto bail;
1395 if (PyDict_SetItemString(mapping, "d", dct) == -1) {
1396 Py_DECREF(mapping);
Raymond Hettingerbcf6f922009-05-27 09:58:34 +00001397 goto bail;
1398 }
Raymond Hettinger491a4cb2009-05-27 11:19:02 +00001399 items = PyEval_EvalCode((PyCodeObject *)code, PyEval_GetGlobals(), mapping);
1400 Py_DECREF(mapping);
1401 } else {
1402 items = PyMapping_Items(dct);
1403 }
1404 if (items == NULL)
1405 goto bail;
Raymond Hettingerbcf6f922009-05-27 09:58:34 +00001406 it = PyObject_GetIter(items);
1407 Py_DECREF(items);
1408 if (it == NULL)
Raymond Hettingerc8d952d2009-05-27 06:50:31 +00001409 goto bail;
Benjamin Petersonc6b607d2009-05-02 12:36:44 +00001410 skipkeys = PyObject_IsTrue(s->skipkeys);
1411 idx = 0;
Raymond Hettingerbcf6f922009-05-27 09:58:34 +00001412 while ((item = PyIter_Next(it)) != NULL) {
1413 PyObject *encoded, *key, *value;
1414 if (!PyTuple_Check(item) || Py_SIZE(item) != 2) {
1415 PyErr_SetString(PyExc_ValueError, "items must return 2-tuples");
1416 goto bail;
1417 }
1418 key = PyTuple_GET_ITEM(item, 0);
Benjamin Petersonc6b607d2009-05-02 12:36:44 +00001419 if (PyUnicode_Check(key)) {
1420 Py_INCREF(key);
1421 kstr = key;
1422 }
1423 else if (PyFloat_Check(key)) {
1424 kstr = encoder_encode_float(s, key);
1425 if (kstr == NULL)
1426 goto bail;
1427 }
Raymond Hettingerbcf6f922009-05-27 09:58:34 +00001428 else if (key == Py_True || key == Py_False || key == Py_None) {
1429 /* This must come before the PyLong_Check because
1430 True and False are also 1 and 0.*/
1431 kstr = _encoded_const(key);
1432 if (kstr == NULL)
1433 goto bail;
1434 }
Benjamin Petersonc6b607d2009-05-02 12:36:44 +00001435 else if (PyLong_Check(key)) {
1436 kstr = PyObject_Str(key);
1437 if (kstr == NULL)
1438 goto bail;
1439 }
Benjamin Petersonc6b607d2009-05-02 12:36:44 +00001440 else if (skipkeys) {
Raymond Hettingerbcf6f922009-05-27 09:58:34 +00001441 Py_DECREF(item);
Benjamin Petersonc6b607d2009-05-02 12:36:44 +00001442 continue;
1443 }
1444 else {
1445 /* TODO: include repr of key */
1446 PyErr_SetString(PyExc_ValueError, "keys must be a string");
1447 goto bail;
1448 }
1449
1450 if (idx) {
1451 if (PyList_Append(rval, s->item_separator))
1452 goto bail;
1453 }
1454
1455 encoded = encoder_encode_string(s, kstr);
1456 Py_CLEAR(kstr);
1457 if (encoded == NULL)
1458 goto bail;
1459 if (PyList_Append(rval, encoded)) {
1460 Py_DECREF(encoded);
1461 goto bail;
1462 }
1463 Py_DECREF(encoded);
1464 if (PyList_Append(rval, s->key_separator))
1465 goto bail;
Raymond Hettingerc8d952d2009-05-27 06:50:31 +00001466
Raymond Hettingerbcf6f922009-05-27 09:58:34 +00001467 value = PyTuple_GET_ITEM(item, 1);
Benjamin Petersonc6b607d2009-05-02 12:36:44 +00001468 if (encoder_listencode_obj(s, rval, value, indent_level))
1469 goto bail;
1470 idx += 1;
Raymond Hettingerbcf6f922009-05-27 09:58:34 +00001471 Py_DECREF(item);
Benjamin Petersonc6b607d2009-05-02 12:36:44 +00001472 }
Raymond Hettingerc8d952d2009-05-27 06:50:31 +00001473 if (PyErr_Occurred())
1474 goto bail;
1475 Py_CLEAR(it);
1476
Benjamin Petersonc6b607d2009-05-02 12:36:44 +00001477 if (ident != NULL) {
1478 if (PyDict_DelItem(s->markers, ident))
1479 goto bail;
1480 Py_CLEAR(ident);
1481 }
1482 if (s->indent != Py_None) {
1483 /* TODO: DOES NOT RUN */
1484 indent_level -= 1;
1485 /*
1486 yield '\n' + (' ' * (_indent * _current_indent_level))
1487 */
1488 }
1489 if (PyList_Append(rval, close_dict))
1490 goto bail;
1491 return 0;
1492
1493bail:
Raymond Hettingerc8d952d2009-05-27 06:50:31 +00001494 Py_XDECREF(it);
Raymond Hettingerbcf6f922009-05-27 09:58:34 +00001495 Py_XDECREF(item);
Benjamin Petersonc6b607d2009-05-02 12:36:44 +00001496 Py_XDECREF(kstr);
1497 Py_XDECREF(ident);
1498 return -1;
1499}
1500
1501
1502static int
1503encoder_listencode_list(PyEncoderObject *s, PyObject *rval, PyObject *seq, Py_ssize_t indent_level)
1504{
1505 /* Encode Python list seq to a JSON term, rval is a PyList */
1506 static PyObject *open_array = NULL;
1507 static PyObject *close_array = NULL;
1508 static PyObject *empty_array = NULL;
1509 PyObject *ident = NULL;
1510 PyObject *s_fast = NULL;
1511 Py_ssize_t num_items;
1512 PyObject **seq_items;
1513 Py_ssize_t i;
1514
1515 if (open_array == NULL || close_array == NULL || empty_array == NULL) {
1516 open_array = PyUnicode_InternFromString("[");
1517 close_array = PyUnicode_InternFromString("]");
1518 empty_array = PyUnicode_InternFromString("[]");
1519 if (open_array == NULL || close_array == NULL || empty_array == NULL)
1520 return -1;
1521 }
1522 ident = NULL;
1523 s_fast = PySequence_Fast(seq, "_iterencode_list needs a sequence");
1524 if (s_fast == NULL)
1525 return -1;
1526 num_items = PySequence_Fast_GET_SIZE(s_fast);
1527 if (num_items == 0) {
1528 Py_DECREF(s_fast);
1529 return PyList_Append(rval, empty_array);
1530 }
1531
1532 if (s->markers != Py_None) {
1533 int has_key;
1534 ident = PyLong_FromVoidPtr(seq);
1535 if (ident == NULL)
1536 goto bail;
1537 has_key = PyDict_Contains(s->markers, ident);
1538 if (has_key) {
1539 if (has_key != -1)
1540 PyErr_SetString(PyExc_ValueError, "Circular reference detected");
1541 goto bail;
1542 }
1543 if (PyDict_SetItem(s->markers, ident, seq)) {
1544 goto bail;
1545 }
1546 }
1547
1548 seq_items = PySequence_Fast_ITEMS(s_fast);
1549 if (PyList_Append(rval, open_array))
1550 goto bail;
1551 if (s->indent != Py_None) {
1552 /* TODO: DOES NOT RUN */
1553 indent_level += 1;
1554 /*
1555 newline_indent = '\n' + (' ' * (_indent * _current_indent_level))
1556 separator = _item_separator + newline_indent
1557 buf += newline_indent
1558 */
1559 }
1560 for (i = 0; i < num_items; i++) {
1561 PyObject *obj = seq_items[i];
1562 if (i) {
1563 if (PyList_Append(rval, s->item_separator))
1564 goto bail;
1565 }
1566 if (encoder_listencode_obj(s, rval, obj, indent_level))
1567 goto bail;
1568 }
1569 if (ident != NULL) {
1570 if (PyDict_DelItem(s->markers, ident))
1571 goto bail;
1572 Py_CLEAR(ident);
1573 }
1574 if (s->indent != Py_None) {
1575 /* TODO: DOES NOT RUN */
1576 indent_level -= 1;
1577 /*
1578 yield '\n' + (' ' * (_indent * _current_indent_level))
1579 */
1580 }
1581 if (PyList_Append(rval, close_array))
1582 goto bail;
1583 Py_DECREF(s_fast);
1584 return 0;
1585
1586bail:
1587 Py_XDECREF(ident);
1588 Py_DECREF(s_fast);
1589 return -1;
1590}
1591
1592static void
1593encoder_dealloc(PyObject *self)
1594{
1595 /* Deallocate Encoder */
1596 encoder_clear(self);
1597 Py_TYPE(self)->tp_free(self);
1598}
1599
1600static int
1601encoder_traverse(PyObject *self, visitproc visit, void *arg)
1602{
1603 PyEncoderObject *s;
1604 assert(PyEncoder_Check(self));
1605 s = (PyEncoderObject *)self;
1606 Py_VISIT(s->markers);
1607 Py_VISIT(s->defaultfn);
1608 Py_VISIT(s->encoder);
1609 Py_VISIT(s->indent);
1610 Py_VISIT(s->key_separator);
1611 Py_VISIT(s->item_separator);
1612 Py_VISIT(s->sort_keys);
1613 Py_VISIT(s->skipkeys);
1614 return 0;
1615}
1616
1617static int
1618encoder_clear(PyObject *self)
1619{
1620 /* Deallocate Encoder */
1621 PyEncoderObject *s;
1622 assert(PyEncoder_Check(self));
1623 s = (PyEncoderObject *)self;
1624 Py_CLEAR(s->markers);
1625 Py_CLEAR(s->defaultfn);
1626 Py_CLEAR(s->encoder);
1627 Py_CLEAR(s->indent);
1628 Py_CLEAR(s->key_separator);
1629 Py_CLEAR(s->item_separator);
1630 Py_CLEAR(s->sort_keys);
1631 Py_CLEAR(s->skipkeys);
1632 return 0;
1633}
1634
1635PyDoc_STRVAR(encoder_doc, "_iterencode(obj, _current_indent_level) -> iterable");
1636
1637static
1638PyTypeObject PyEncoderType = {
1639 PyVarObject_HEAD_INIT(NULL, 0)
1640 "_json.Encoder", /* tp_name */
1641 sizeof(PyEncoderObject), /* tp_basicsize */
1642 0, /* tp_itemsize */
1643 encoder_dealloc, /* tp_dealloc */
1644 0, /* tp_print */
1645 0, /* tp_getattr */
1646 0, /* tp_setattr */
1647 0, /* tp_compare */
1648 0, /* tp_repr */
1649 0, /* tp_as_number */
1650 0, /* tp_as_sequence */
1651 0, /* tp_as_mapping */
1652 0, /* tp_hash */
1653 encoder_call, /* tp_call */
1654 0, /* tp_str */
1655 0, /* tp_getattro */
1656 0, /* tp_setattro */
1657 0, /* tp_as_buffer */
1658 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC, /* tp_flags */
1659 encoder_doc, /* tp_doc */
1660 encoder_traverse, /* tp_traverse */
1661 encoder_clear, /* tp_clear */
1662 0, /* tp_richcompare */
1663 0, /* tp_weaklistoffset */
1664 0, /* tp_iter */
1665 0, /* tp_iternext */
1666 0, /* tp_methods */
1667 encoder_members, /* tp_members */
1668 0, /* tp_getset */
1669 0, /* tp_base */
1670 0, /* tp_dict */
1671 0, /* tp_descr_get */
1672 0, /* tp_descr_set */
1673 0, /* tp_dictoffset */
1674 encoder_init, /* tp_init */
1675 0, /* tp_alloc */
1676 encoder_new, /* tp_new */
1677 0, /* tp_free */
1678};
1679
1680static PyMethodDef speedups_methods[] = {
1681 {"encode_basestring_ascii",
1682 (PyCFunction)py_encode_basestring_ascii,
1683 METH_O,
1684 pydoc_encode_basestring_ascii},
1685 {"scanstring",
1686 (PyCFunction)py_scanstring,
1687 METH_VARARGS,
1688 pydoc_scanstring},
Christian Heimes90540002008-05-08 14:29:10 +00001689 {NULL, NULL, 0, NULL}
1690};
1691
1692PyDoc_STRVAR(module_doc,
1693"json speedups\n");
1694
Martin v. Löwis1a214512008-06-11 05:26:20 +00001695static struct PyModuleDef jsonmodule = {
1696 PyModuleDef_HEAD_INIT,
1697 "_json",
1698 module_doc,
1699 -1,
Benjamin Petersonc6b607d2009-05-02 12:36:44 +00001700 speedups_methods,
Martin v. Löwis1a214512008-06-11 05:26:20 +00001701 NULL,
1702 NULL,
1703 NULL,
1704 NULL
1705};
1706
1707PyObject*
1708PyInit__json(void)
Christian Heimes90540002008-05-08 14:29:10 +00001709{
Benjamin Petersonc6b607d2009-05-02 12:36:44 +00001710 PyObject *m = PyModule_Create(&jsonmodule);
1711 if (!m)
1712 return NULL;
1713 PyScannerType.tp_new = PyType_GenericNew;
1714 if (PyType_Ready(&PyScannerType) < 0)
1715 goto fail;
1716 PyEncoderType.tp_new = PyType_GenericNew;
1717 if (PyType_Ready(&PyEncoderType) < 0)
1718 goto fail;
1719 Py_INCREF((PyObject*)&PyScannerType);
1720 if (PyModule_AddObject(m, "make_scanner", (PyObject*)&PyScannerType) < 0) {
1721 Py_DECREF((PyObject*)&PyScannerType);
1722 goto fail;
1723 }
1724 Py_INCREF((PyObject*)&PyEncoderType);
1725 if (PyModule_AddObject(m, "make_encoder", (PyObject*)&PyEncoderType) < 0) {
1726 Py_DECREF((PyObject*)&PyEncoderType);
1727 goto fail;
1728 }
1729 return m;
1730 fail:
1731 Py_DECREF(m);
1732 return NULL;
Christian Heimes90540002008-05-08 14:29:10 +00001733}