blob: 2e395255f0a2765b795354f99806886069bcaaf3 [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,
Georg Brandl4009c9e2010-10-06 08:26:09 +0000487 "scanstring(string, end, strict=True) -> (string, end)\n"
Benjamin Petersonc6b607d2009-05-02 12:36:44 +0000488 "\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 {
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +0000514 PyErr_Format(PyExc_TypeError,
Georg Brandl4009c9e2010-10-06 08:26:09 +0000515 "first argument must be a string, not %.80s",
Christian Heimes90540002008-05-08 14:29:10 +0000516 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,
Georg Brandl4009c9e2010-10-06 08:26:09 +0000523 "encode_basestring_ascii(string) -> string\n"
Benjamin Petersonc6b607d2009-05-02 12:36:44 +0000524 "\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;
Antoine Pitrou4fad6bd2009-12-08 16:00:03 +00001126 PyObject *markers, *defaultfn, *encoder, *indent, *key_separator;
1127 PyObject *item_separator, *sort_keys, *skipkeys, *allow_nan;
Benjamin Petersonc6b607d2009-05-02 12:36:44 +00001128
1129 assert(PyEncoder_Check(self));
1130 s = (PyEncoderObject *)self;
1131
1132 if (!PyArg_ParseTupleAndKeywords(args, kwds, "OOOOOOOOO:make_encoder", kwlist,
Antoine Pitrou4fad6bd2009-12-08 16:00:03 +00001133 &markers, &defaultfn, &encoder, &indent, &key_separator, &item_separator,
1134 &sort_keys, &skipkeys, &allow_nan))
Benjamin Petersonc6b607d2009-05-02 12:36:44 +00001135 return -1;
1136
Antoine Pitrou4fad6bd2009-12-08 16:00:03 +00001137 s->markers = markers;
1138 s->defaultfn = defaultfn;
1139 s->encoder = encoder;
1140 s->indent = indent;
1141 s->key_separator = key_separator;
1142 s->item_separator = item_separator;
1143 s->sort_keys = sort_keys;
1144 s->skipkeys = skipkeys;
1145 s->fast_encode = (PyCFunction_Check(s->encoder) && PyCFunction_GetFunction(s->encoder) == (PyCFunction)py_encode_basestring_ascii);
1146 s->allow_nan = PyObject_IsTrue(allow_nan);
1147
Benjamin Petersonc6b607d2009-05-02 12:36:44 +00001148 Py_INCREF(s->markers);
1149 Py_INCREF(s->defaultfn);
1150 Py_INCREF(s->encoder);
1151 Py_INCREF(s->indent);
1152 Py_INCREF(s->key_separator);
1153 Py_INCREF(s->item_separator);
1154 Py_INCREF(s->sort_keys);
1155 Py_INCREF(s->skipkeys);
Benjamin Petersonc6b607d2009-05-02 12:36:44 +00001156 return 0;
1157}
1158
1159static PyObject *
1160encoder_call(PyObject *self, PyObject *args, PyObject *kwds)
1161{
1162 /* Python callable interface to encode_listencode_obj */
1163 static char *kwlist[] = {"obj", "_current_indent_level", NULL};
1164 PyObject *obj;
1165 PyObject *rval;
1166 Py_ssize_t indent_level;
1167 PyEncoderObject *s;
1168 assert(PyEncoder_Check(self));
1169 s = (PyEncoderObject *)self;
1170 if (!PyArg_ParseTupleAndKeywords(args, kwds, "OO&:_iterencode", kwlist,
1171 &obj, _convertPyInt_AsSsize_t, &indent_level))
1172 return NULL;
1173 rval = PyList_New(0);
1174 if (rval == NULL)
1175 return NULL;
1176 if (encoder_listencode_obj(s, rval, obj, indent_level)) {
1177 Py_DECREF(rval);
1178 return NULL;
1179 }
1180 return rval;
1181}
1182
1183static PyObject *
1184_encoded_const(PyObject *obj)
1185{
1186 /* Return the JSON string representation of None, True, False */
1187 if (obj == Py_None) {
1188 static PyObject *s_null = NULL;
1189 if (s_null == NULL) {
1190 s_null = PyUnicode_InternFromString("null");
1191 }
1192 Py_INCREF(s_null);
1193 return s_null;
1194 }
1195 else if (obj == Py_True) {
1196 static PyObject *s_true = NULL;
1197 if (s_true == NULL) {
1198 s_true = PyUnicode_InternFromString("true");
1199 }
1200 Py_INCREF(s_true);
1201 return s_true;
1202 }
1203 else if (obj == Py_False) {
1204 static PyObject *s_false = NULL;
1205 if (s_false == NULL) {
1206 s_false = PyUnicode_InternFromString("false");
1207 }
1208 Py_INCREF(s_false);
1209 return s_false;
1210 }
1211 else {
1212 PyErr_SetString(PyExc_ValueError, "not a const");
1213 return NULL;
1214 }
1215}
1216
1217static PyObject *
1218encoder_encode_float(PyEncoderObject *s, PyObject *obj)
1219{
1220 /* Return the JSON representation of a PyFloat */
1221 double i = PyFloat_AS_DOUBLE(obj);
1222 if (!Py_IS_FINITE(i)) {
1223 if (!s->allow_nan) {
1224 PyErr_SetString(PyExc_ValueError, "Out of range float values are not JSON compliant");
1225 return NULL;
1226 }
1227 if (i > 0) {
1228 return PyUnicode_FromString("Infinity");
1229 }
1230 else if (i < 0) {
1231 return PyUnicode_FromString("-Infinity");
1232 }
1233 else {
1234 return PyUnicode_FromString("NaN");
1235 }
1236 }
1237 /* Use a better float format here? */
1238 return PyObject_Repr(obj);
1239}
1240
1241static PyObject *
1242encoder_encode_string(PyEncoderObject *s, PyObject *obj)
1243{
1244 /* Return the JSON representation of a string */
1245 if (s->fast_encode)
1246 return py_encode_basestring_ascii(NULL, obj);
1247 else
1248 return PyObject_CallFunctionObjArgs(s->encoder, obj, NULL);
1249}
1250
1251static int
1252_steal_list_append(PyObject *lst, PyObject *stolen)
1253{
1254 /* Append stolen and then decrement its reference count */
1255 int rval = PyList_Append(lst, stolen);
1256 Py_DECREF(stolen);
1257 return rval;
1258}
1259
1260static int
1261encoder_listencode_obj(PyEncoderObject *s, PyObject *rval, PyObject *obj, Py_ssize_t indent_level)
1262{
1263 /* Encode Python object obj to a JSON term, rval is a PyList */
1264 PyObject *newobj;
1265 int rv;
1266
1267 if (obj == Py_None || obj == Py_True || obj == Py_False) {
1268 PyObject *cstr = _encoded_const(obj);
1269 if (cstr == NULL)
1270 return -1;
1271 return _steal_list_append(rval, cstr);
1272 }
1273 else if (PyUnicode_Check(obj))
1274 {
1275 PyObject *encoded = encoder_encode_string(s, obj);
1276 if (encoded == NULL)
1277 return -1;
1278 return _steal_list_append(rval, encoded);
1279 }
1280 else if (PyLong_Check(obj)) {
1281 PyObject *encoded = PyObject_Str(obj);
1282 if (encoded == NULL)
1283 return -1;
1284 return _steal_list_append(rval, encoded);
1285 }
1286 else if (PyFloat_Check(obj)) {
1287 PyObject *encoded = encoder_encode_float(s, obj);
1288 if (encoded == NULL)
1289 return -1;
1290 return _steal_list_append(rval, encoded);
1291 }
1292 else if (PyList_Check(obj) || PyTuple_Check(obj)) {
1293 return encoder_listencode_list(s, rval, obj, indent_level);
1294 }
1295 else if (PyDict_Check(obj)) {
1296 return encoder_listencode_dict(s, rval, obj, indent_level);
1297 }
1298 else {
1299 PyObject *ident = NULL;
1300 if (s->markers != Py_None) {
1301 int has_key;
1302 ident = PyLong_FromVoidPtr(obj);
1303 if (ident == NULL)
1304 return -1;
1305 has_key = PyDict_Contains(s->markers, ident);
1306 if (has_key) {
1307 if (has_key != -1)
1308 PyErr_SetString(PyExc_ValueError, "Circular reference detected");
1309 Py_DECREF(ident);
1310 return -1;
1311 }
1312 if (PyDict_SetItem(s->markers, ident, obj)) {
1313 Py_DECREF(ident);
1314 return -1;
1315 }
1316 }
1317 newobj = PyObject_CallFunctionObjArgs(s->defaultfn, obj, NULL);
1318 if (newobj == NULL) {
1319 Py_XDECREF(ident);
1320 return -1;
1321 }
1322 rv = encoder_listencode_obj(s, rval, newobj, indent_level);
1323 Py_DECREF(newobj);
1324 if (rv) {
1325 Py_XDECREF(ident);
1326 return -1;
1327 }
1328 if (ident != NULL) {
1329 if (PyDict_DelItem(s->markers, ident)) {
1330 Py_XDECREF(ident);
1331 return -1;
1332 }
1333 Py_XDECREF(ident);
1334 }
1335 return rv;
1336 }
1337}
1338
1339static int
1340encoder_listencode_dict(PyEncoderObject *s, PyObject *rval, PyObject *dct, Py_ssize_t indent_level)
1341{
1342 /* Encode Python dict dct a JSON term, rval is a PyList */
1343 static PyObject *open_dict = NULL;
1344 static PyObject *close_dict = NULL;
1345 static PyObject *empty_dict = NULL;
1346 PyObject *kstr = NULL;
1347 PyObject *ident = NULL;
Raymond Hettingerc8d952d2009-05-27 06:50:31 +00001348 PyObject *it = NULL;
Raymond Hettingerbcf6f922009-05-27 09:58:34 +00001349 PyObject *items;
1350 PyObject *item = NULL;
Benjamin Petersonc6b607d2009-05-02 12:36:44 +00001351 int skipkeys;
1352 Py_ssize_t idx;
Raymond Hettinger491a4cb2009-05-27 11:19:02 +00001353 PyObject *mapping;
1354 static PyObject *code = NULL;
Benjamin Petersonc6b607d2009-05-02 12:36:44 +00001355
1356 if (open_dict == NULL || close_dict == NULL || empty_dict == NULL) {
1357 open_dict = PyUnicode_InternFromString("{");
1358 close_dict = PyUnicode_InternFromString("}");
1359 empty_dict = PyUnicode_InternFromString("{}");
1360 if (open_dict == NULL || close_dict == NULL || empty_dict == NULL)
1361 return -1;
1362 }
Raymond Hettingerc8d952d2009-05-27 06:50:31 +00001363 if (Py_SIZE(dct) == 0)
Benjamin Petersonc6b607d2009-05-02 12:36:44 +00001364 return PyList_Append(rval, empty_dict);
1365
1366 if (s->markers != Py_None) {
1367 int has_key;
1368 ident = PyLong_FromVoidPtr(dct);
1369 if (ident == NULL)
1370 goto bail;
1371 has_key = PyDict_Contains(s->markers, ident);
1372 if (has_key) {
1373 if (has_key != -1)
1374 PyErr_SetString(PyExc_ValueError, "Circular reference detected");
1375 goto bail;
1376 }
1377 if (PyDict_SetItem(s->markers, ident, dct)) {
1378 goto bail;
1379 }
1380 }
1381
1382 if (PyList_Append(rval, open_dict))
1383 goto bail;
1384
1385 if (s->indent != Py_None) {
1386 /* TODO: DOES NOT RUN */
1387 indent_level += 1;
1388 /*
1389 newline_indent = '\n' + (' ' * (_indent * _current_indent_level))
1390 separator = _item_separator + newline_indent
1391 buf += newline_indent
1392 */
1393 }
1394
Raymond Hettingerbcf6f922009-05-27 09:58:34 +00001395 if (PyObject_IsTrue(s->sort_keys)) {
Raymond Hettinger491a4cb2009-05-27 11:19:02 +00001396 if (code == NULL) {
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +00001397 code = Py_CompileString("sorted(d.items(), key=lambda kv: kv[0])",
Raymond Hettinger491a4cb2009-05-27 11:19:02 +00001398 "_json.c", Py_eval_input);
1399 if (code == NULL)
1400 goto bail;
1401 }
Benjamin Petersonc6b607d2009-05-02 12:36:44 +00001402
Raymond Hettinger491a4cb2009-05-27 11:19:02 +00001403 mapping = PyDict_New();
1404 if (mapping == NULL)
1405 goto bail;
1406 if (PyDict_SetItemString(mapping, "d", dct) == -1) {
1407 Py_DECREF(mapping);
Raymond Hettingerbcf6f922009-05-27 09:58:34 +00001408 goto bail;
1409 }
Raymond Hettinger491a4cb2009-05-27 11:19:02 +00001410 items = PyEval_EvalCode((PyCodeObject *)code, PyEval_GetGlobals(), mapping);
1411 Py_DECREF(mapping);
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +00001412 } else {
Raymond Hettinger491a4cb2009-05-27 11:19:02 +00001413 items = PyMapping_Items(dct);
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +00001414 }
1415 if (items == NULL)
Raymond Hettinger491a4cb2009-05-27 11:19:02 +00001416 goto bail;
Raymond Hettingerbcf6f922009-05-27 09:58:34 +00001417 it = PyObject_GetIter(items);
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +00001418 Py_DECREF(items);
1419 if (it == NULL)
Raymond Hettingerc8d952d2009-05-27 06:50:31 +00001420 goto bail;
Benjamin Petersonc6b607d2009-05-02 12:36:44 +00001421 skipkeys = PyObject_IsTrue(s->skipkeys);
1422 idx = 0;
Raymond Hettingerbcf6f922009-05-27 09:58:34 +00001423 while ((item = PyIter_Next(it)) != NULL) {
1424 PyObject *encoded, *key, *value;
1425 if (!PyTuple_Check(item) || Py_SIZE(item) != 2) {
1426 PyErr_SetString(PyExc_ValueError, "items must return 2-tuples");
1427 goto bail;
1428 }
1429 key = PyTuple_GET_ITEM(item, 0);
Benjamin Petersonc6b607d2009-05-02 12:36:44 +00001430 if (PyUnicode_Check(key)) {
1431 Py_INCREF(key);
1432 kstr = key;
1433 }
1434 else if (PyFloat_Check(key)) {
1435 kstr = encoder_encode_float(s, key);
1436 if (kstr == NULL)
1437 goto bail;
1438 }
Raymond Hettingerbcf6f922009-05-27 09:58:34 +00001439 else if (key == Py_True || key == Py_False || key == Py_None) {
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +00001440 /* This must come before the PyLong_Check because
1441 True and False are also 1 and 0.*/
Raymond Hettingerbcf6f922009-05-27 09:58:34 +00001442 kstr = _encoded_const(key);
1443 if (kstr == NULL)
1444 goto bail;
1445 }
Benjamin Petersonc6b607d2009-05-02 12:36:44 +00001446 else if (PyLong_Check(key)) {
1447 kstr = PyObject_Str(key);
1448 if (kstr == NULL)
1449 goto bail;
1450 }
Benjamin Petersonc6b607d2009-05-02 12:36:44 +00001451 else if (skipkeys) {
Raymond Hettingerbcf6f922009-05-27 09:58:34 +00001452 Py_DECREF(item);
Benjamin Petersonc6b607d2009-05-02 12:36:44 +00001453 continue;
1454 }
1455 else {
1456 /* TODO: include repr of key */
Doug Hellmann76e57942010-07-21 12:35:38 +00001457 PyErr_SetString(PyExc_TypeError, "keys must be a string");
Benjamin Petersonc6b607d2009-05-02 12:36:44 +00001458 goto bail;
1459 }
1460
1461 if (idx) {
1462 if (PyList_Append(rval, s->item_separator))
1463 goto bail;
1464 }
1465
1466 encoded = encoder_encode_string(s, kstr);
1467 Py_CLEAR(kstr);
1468 if (encoded == NULL)
1469 goto bail;
1470 if (PyList_Append(rval, encoded)) {
1471 Py_DECREF(encoded);
1472 goto bail;
1473 }
1474 Py_DECREF(encoded);
1475 if (PyList_Append(rval, s->key_separator))
1476 goto bail;
Raymond Hettingerc8d952d2009-05-27 06:50:31 +00001477
Raymond Hettingerbcf6f922009-05-27 09:58:34 +00001478 value = PyTuple_GET_ITEM(item, 1);
Benjamin Petersonc6b607d2009-05-02 12:36:44 +00001479 if (encoder_listencode_obj(s, rval, value, indent_level))
1480 goto bail;
1481 idx += 1;
Raymond Hettingerbcf6f922009-05-27 09:58:34 +00001482 Py_DECREF(item);
Benjamin Petersonc6b607d2009-05-02 12:36:44 +00001483 }
Raymond Hettingerc8d952d2009-05-27 06:50:31 +00001484 if (PyErr_Occurred())
1485 goto bail;
1486 Py_CLEAR(it);
1487
Benjamin Petersonc6b607d2009-05-02 12:36:44 +00001488 if (ident != NULL) {
1489 if (PyDict_DelItem(s->markers, ident))
1490 goto bail;
1491 Py_CLEAR(ident);
1492 }
1493 if (s->indent != Py_None) {
1494 /* TODO: DOES NOT RUN */
1495 indent_level -= 1;
1496 /*
1497 yield '\n' + (' ' * (_indent * _current_indent_level))
1498 */
1499 }
1500 if (PyList_Append(rval, close_dict))
1501 goto bail;
1502 return 0;
1503
1504bail:
Raymond Hettingerc8d952d2009-05-27 06:50:31 +00001505 Py_XDECREF(it);
Raymond Hettingerbcf6f922009-05-27 09:58:34 +00001506 Py_XDECREF(item);
Benjamin Petersonc6b607d2009-05-02 12:36:44 +00001507 Py_XDECREF(kstr);
1508 Py_XDECREF(ident);
1509 return -1;
1510}
1511
1512
1513static int
1514encoder_listencode_list(PyEncoderObject *s, PyObject *rval, PyObject *seq, Py_ssize_t indent_level)
1515{
1516 /* Encode Python list seq to a JSON term, rval is a PyList */
1517 static PyObject *open_array = NULL;
1518 static PyObject *close_array = NULL;
1519 static PyObject *empty_array = NULL;
1520 PyObject *ident = NULL;
1521 PyObject *s_fast = NULL;
1522 Py_ssize_t num_items;
1523 PyObject **seq_items;
1524 Py_ssize_t i;
1525
1526 if (open_array == NULL || close_array == NULL || empty_array == NULL) {
1527 open_array = PyUnicode_InternFromString("[");
1528 close_array = PyUnicode_InternFromString("]");
1529 empty_array = PyUnicode_InternFromString("[]");
1530 if (open_array == NULL || close_array == NULL || empty_array == NULL)
1531 return -1;
1532 }
1533 ident = NULL;
1534 s_fast = PySequence_Fast(seq, "_iterencode_list needs a sequence");
1535 if (s_fast == NULL)
1536 return -1;
1537 num_items = PySequence_Fast_GET_SIZE(s_fast);
1538 if (num_items == 0) {
1539 Py_DECREF(s_fast);
1540 return PyList_Append(rval, empty_array);
1541 }
1542
1543 if (s->markers != Py_None) {
1544 int has_key;
1545 ident = PyLong_FromVoidPtr(seq);
1546 if (ident == NULL)
1547 goto bail;
1548 has_key = PyDict_Contains(s->markers, ident);
1549 if (has_key) {
1550 if (has_key != -1)
1551 PyErr_SetString(PyExc_ValueError, "Circular reference detected");
1552 goto bail;
1553 }
1554 if (PyDict_SetItem(s->markers, ident, seq)) {
1555 goto bail;
1556 }
1557 }
1558
1559 seq_items = PySequence_Fast_ITEMS(s_fast);
1560 if (PyList_Append(rval, open_array))
1561 goto bail;
1562 if (s->indent != Py_None) {
1563 /* TODO: DOES NOT RUN */
1564 indent_level += 1;
1565 /*
1566 newline_indent = '\n' + (' ' * (_indent * _current_indent_level))
1567 separator = _item_separator + newline_indent
1568 buf += newline_indent
1569 */
1570 }
1571 for (i = 0; i < num_items; i++) {
1572 PyObject *obj = seq_items[i];
1573 if (i) {
1574 if (PyList_Append(rval, s->item_separator))
1575 goto bail;
1576 }
1577 if (encoder_listencode_obj(s, rval, obj, indent_level))
1578 goto bail;
1579 }
1580 if (ident != NULL) {
1581 if (PyDict_DelItem(s->markers, ident))
1582 goto bail;
1583 Py_CLEAR(ident);
1584 }
1585 if (s->indent != Py_None) {
1586 /* TODO: DOES NOT RUN */
1587 indent_level -= 1;
1588 /*
1589 yield '\n' + (' ' * (_indent * _current_indent_level))
1590 */
1591 }
1592 if (PyList_Append(rval, close_array))
1593 goto bail;
1594 Py_DECREF(s_fast);
1595 return 0;
1596
1597bail:
1598 Py_XDECREF(ident);
1599 Py_DECREF(s_fast);
1600 return -1;
1601}
1602
1603static void
1604encoder_dealloc(PyObject *self)
1605{
1606 /* Deallocate Encoder */
1607 encoder_clear(self);
1608 Py_TYPE(self)->tp_free(self);
1609}
1610
1611static int
1612encoder_traverse(PyObject *self, visitproc visit, void *arg)
1613{
1614 PyEncoderObject *s;
1615 assert(PyEncoder_Check(self));
1616 s = (PyEncoderObject *)self;
1617 Py_VISIT(s->markers);
1618 Py_VISIT(s->defaultfn);
1619 Py_VISIT(s->encoder);
1620 Py_VISIT(s->indent);
1621 Py_VISIT(s->key_separator);
1622 Py_VISIT(s->item_separator);
1623 Py_VISIT(s->sort_keys);
1624 Py_VISIT(s->skipkeys);
1625 return 0;
1626}
1627
1628static int
1629encoder_clear(PyObject *self)
1630{
1631 /* Deallocate Encoder */
1632 PyEncoderObject *s;
1633 assert(PyEncoder_Check(self));
1634 s = (PyEncoderObject *)self;
1635 Py_CLEAR(s->markers);
1636 Py_CLEAR(s->defaultfn);
1637 Py_CLEAR(s->encoder);
1638 Py_CLEAR(s->indent);
1639 Py_CLEAR(s->key_separator);
1640 Py_CLEAR(s->item_separator);
1641 Py_CLEAR(s->sort_keys);
1642 Py_CLEAR(s->skipkeys);
1643 return 0;
1644}
1645
1646PyDoc_STRVAR(encoder_doc, "_iterencode(obj, _current_indent_level) -> iterable");
1647
1648static
1649PyTypeObject PyEncoderType = {
1650 PyVarObject_HEAD_INIT(NULL, 0)
1651 "_json.Encoder", /* tp_name */
1652 sizeof(PyEncoderObject), /* tp_basicsize */
1653 0, /* tp_itemsize */
1654 encoder_dealloc, /* tp_dealloc */
1655 0, /* tp_print */
1656 0, /* tp_getattr */
1657 0, /* tp_setattr */
1658 0, /* tp_compare */
1659 0, /* tp_repr */
1660 0, /* tp_as_number */
1661 0, /* tp_as_sequence */
1662 0, /* tp_as_mapping */
1663 0, /* tp_hash */
1664 encoder_call, /* tp_call */
1665 0, /* tp_str */
1666 0, /* tp_getattro */
1667 0, /* tp_setattro */
1668 0, /* tp_as_buffer */
1669 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC, /* tp_flags */
1670 encoder_doc, /* tp_doc */
1671 encoder_traverse, /* tp_traverse */
1672 encoder_clear, /* tp_clear */
1673 0, /* tp_richcompare */
1674 0, /* tp_weaklistoffset */
1675 0, /* tp_iter */
1676 0, /* tp_iternext */
1677 0, /* tp_methods */
1678 encoder_members, /* tp_members */
1679 0, /* tp_getset */
1680 0, /* tp_base */
1681 0, /* tp_dict */
1682 0, /* tp_descr_get */
1683 0, /* tp_descr_set */
1684 0, /* tp_dictoffset */
1685 encoder_init, /* tp_init */
1686 0, /* tp_alloc */
1687 encoder_new, /* tp_new */
1688 0, /* tp_free */
1689};
1690
1691static PyMethodDef speedups_methods[] = {
1692 {"encode_basestring_ascii",
1693 (PyCFunction)py_encode_basestring_ascii,
1694 METH_O,
1695 pydoc_encode_basestring_ascii},
1696 {"scanstring",
1697 (PyCFunction)py_scanstring,
1698 METH_VARARGS,
1699 pydoc_scanstring},
Christian Heimes90540002008-05-08 14:29:10 +00001700 {NULL, NULL, 0, NULL}
1701};
1702
1703PyDoc_STRVAR(module_doc,
1704"json speedups\n");
1705
Martin v. Löwis1a214512008-06-11 05:26:20 +00001706static struct PyModuleDef jsonmodule = {
Antoine Pitrou7f14f0d2010-05-09 16:14:21 +00001707 PyModuleDef_HEAD_INIT,
1708 "_json",
1709 module_doc,
1710 -1,
1711 speedups_methods,
1712 NULL,
1713 NULL,
1714 NULL,
1715 NULL
Martin v. Löwis1a214512008-06-11 05:26:20 +00001716};
1717
1718PyObject*
1719PyInit__json(void)
Christian Heimes90540002008-05-08 14:29:10 +00001720{
Benjamin Petersonc6b607d2009-05-02 12:36:44 +00001721 PyObject *m = PyModule_Create(&jsonmodule);
1722 if (!m)
1723 return NULL;
1724 PyScannerType.tp_new = PyType_GenericNew;
1725 if (PyType_Ready(&PyScannerType) < 0)
1726 goto fail;
1727 PyEncoderType.tp_new = PyType_GenericNew;
1728 if (PyType_Ready(&PyEncoderType) < 0)
1729 goto fail;
1730 Py_INCREF((PyObject*)&PyScannerType);
1731 if (PyModule_AddObject(m, "make_scanner", (PyObject*)&PyScannerType) < 0) {
1732 Py_DECREF((PyObject*)&PyScannerType);
1733 goto fail;
1734 }
1735 Py_INCREF((PyObject*)&PyEncoderType);
1736 if (PyModule_AddObject(m, "make_encoder", (PyObject*)&PyEncoderType) < 0) {
1737 Py_DECREF((PyObject*)&PyEncoderType);
1738 goto fail;
1739 }
1740 return m;
1741 fail:
1742 Py_DECREF(m);
1743 return NULL;
Christian Heimes90540002008-05-08 14:29:10 +00001744}