blob: c44dbac1f132390faff7d615bc8697a9e635fbba [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;
1337 PyObject *key, *value;
1338 Py_ssize_t pos;
1339 int skipkeys;
1340 Py_ssize_t idx;
1341
1342 if (open_dict == NULL || close_dict == NULL || empty_dict == NULL) {
1343 open_dict = PyUnicode_InternFromString("{");
1344 close_dict = PyUnicode_InternFromString("}");
1345 empty_dict = PyUnicode_InternFromString("{}");
1346 if (open_dict == NULL || close_dict == NULL || empty_dict == NULL)
1347 return -1;
1348 }
1349 if (PyDict_Size(dct) == 0)
1350 return PyList_Append(rval, empty_dict);
1351
1352 if (s->markers != Py_None) {
1353 int has_key;
1354 ident = PyLong_FromVoidPtr(dct);
1355 if (ident == NULL)
1356 goto bail;
1357 has_key = PyDict_Contains(s->markers, ident);
1358 if (has_key) {
1359 if (has_key != -1)
1360 PyErr_SetString(PyExc_ValueError, "Circular reference detected");
1361 goto bail;
1362 }
1363 if (PyDict_SetItem(s->markers, ident, dct)) {
1364 goto bail;
1365 }
1366 }
1367
1368 if (PyList_Append(rval, open_dict))
1369 goto bail;
1370
1371 if (s->indent != Py_None) {
1372 /* TODO: DOES NOT RUN */
1373 indent_level += 1;
1374 /*
1375 newline_indent = '\n' + (' ' * (_indent * _current_indent_level))
1376 separator = _item_separator + newline_indent
1377 buf += newline_indent
1378 */
1379 }
1380
1381 /* TODO: C speedup not implemented for sort_keys */
1382
1383 pos = 0;
1384 skipkeys = PyObject_IsTrue(s->skipkeys);
1385 idx = 0;
1386 while (PyDict_Next(dct, &pos, &key, &value)) {
1387 PyObject *encoded;
1388
1389 if (PyUnicode_Check(key)) {
1390 Py_INCREF(key);
1391 kstr = key;
1392 }
1393 else if (PyFloat_Check(key)) {
1394 kstr = encoder_encode_float(s, key);
1395 if (kstr == NULL)
1396 goto bail;
1397 }
1398 else if (PyLong_Check(key)) {
1399 kstr = PyObject_Str(key);
1400 if (kstr == NULL)
1401 goto bail;
1402 }
1403 else if (key == Py_True || key == Py_False || key == Py_None) {
1404 kstr = _encoded_const(key);
1405 if (kstr == NULL)
1406 goto bail;
1407 }
1408 else if (skipkeys) {
1409 continue;
1410 }
1411 else {
1412 /* TODO: include repr of key */
1413 PyErr_SetString(PyExc_ValueError, "keys must be a string");
1414 goto bail;
1415 }
1416
1417 if (idx) {
1418 if (PyList_Append(rval, s->item_separator))
1419 goto bail;
1420 }
1421
1422 encoded = encoder_encode_string(s, kstr);
1423 Py_CLEAR(kstr);
1424 if (encoded == NULL)
1425 goto bail;
1426 if (PyList_Append(rval, encoded)) {
1427 Py_DECREF(encoded);
1428 goto bail;
1429 }
1430 Py_DECREF(encoded);
1431 if (PyList_Append(rval, s->key_separator))
1432 goto bail;
1433 if (encoder_listencode_obj(s, rval, value, indent_level))
1434 goto bail;
1435 idx += 1;
1436 }
1437 if (ident != NULL) {
1438 if (PyDict_DelItem(s->markers, ident))
1439 goto bail;
1440 Py_CLEAR(ident);
1441 }
1442 if (s->indent != Py_None) {
1443 /* TODO: DOES NOT RUN */
1444 indent_level -= 1;
1445 /*
1446 yield '\n' + (' ' * (_indent * _current_indent_level))
1447 */
1448 }
1449 if (PyList_Append(rval, close_dict))
1450 goto bail;
1451 return 0;
1452
1453bail:
1454 Py_XDECREF(kstr);
1455 Py_XDECREF(ident);
1456 return -1;
1457}
1458
1459
1460static int
1461encoder_listencode_list(PyEncoderObject *s, PyObject *rval, PyObject *seq, Py_ssize_t indent_level)
1462{
1463 /* Encode Python list seq to a JSON term, rval is a PyList */
1464 static PyObject *open_array = NULL;
1465 static PyObject *close_array = NULL;
1466 static PyObject *empty_array = NULL;
1467 PyObject *ident = NULL;
1468 PyObject *s_fast = NULL;
1469 Py_ssize_t num_items;
1470 PyObject **seq_items;
1471 Py_ssize_t i;
1472
1473 if (open_array == NULL || close_array == NULL || empty_array == NULL) {
1474 open_array = PyUnicode_InternFromString("[");
1475 close_array = PyUnicode_InternFromString("]");
1476 empty_array = PyUnicode_InternFromString("[]");
1477 if (open_array == NULL || close_array == NULL || empty_array == NULL)
1478 return -1;
1479 }
1480 ident = NULL;
1481 s_fast = PySequence_Fast(seq, "_iterencode_list needs a sequence");
1482 if (s_fast == NULL)
1483 return -1;
1484 num_items = PySequence_Fast_GET_SIZE(s_fast);
1485 if (num_items == 0) {
1486 Py_DECREF(s_fast);
1487 return PyList_Append(rval, empty_array);
1488 }
1489
1490 if (s->markers != Py_None) {
1491 int has_key;
1492 ident = PyLong_FromVoidPtr(seq);
1493 if (ident == NULL)
1494 goto bail;
1495 has_key = PyDict_Contains(s->markers, ident);
1496 if (has_key) {
1497 if (has_key != -1)
1498 PyErr_SetString(PyExc_ValueError, "Circular reference detected");
1499 goto bail;
1500 }
1501 if (PyDict_SetItem(s->markers, ident, seq)) {
1502 goto bail;
1503 }
1504 }
1505
1506 seq_items = PySequence_Fast_ITEMS(s_fast);
1507 if (PyList_Append(rval, open_array))
1508 goto bail;
1509 if (s->indent != Py_None) {
1510 /* TODO: DOES NOT RUN */
1511 indent_level += 1;
1512 /*
1513 newline_indent = '\n' + (' ' * (_indent * _current_indent_level))
1514 separator = _item_separator + newline_indent
1515 buf += newline_indent
1516 */
1517 }
1518 for (i = 0; i < num_items; i++) {
1519 PyObject *obj = seq_items[i];
1520 if (i) {
1521 if (PyList_Append(rval, s->item_separator))
1522 goto bail;
1523 }
1524 if (encoder_listencode_obj(s, rval, obj, indent_level))
1525 goto bail;
1526 }
1527 if (ident != NULL) {
1528 if (PyDict_DelItem(s->markers, ident))
1529 goto bail;
1530 Py_CLEAR(ident);
1531 }
1532 if (s->indent != Py_None) {
1533 /* TODO: DOES NOT RUN */
1534 indent_level -= 1;
1535 /*
1536 yield '\n' + (' ' * (_indent * _current_indent_level))
1537 */
1538 }
1539 if (PyList_Append(rval, close_array))
1540 goto bail;
1541 Py_DECREF(s_fast);
1542 return 0;
1543
1544bail:
1545 Py_XDECREF(ident);
1546 Py_DECREF(s_fast);
1547 return -1;
1548}
1549
1550static void
1551encoder_dealloc(PyObject *self)
1552{
1553 /* Deallocate Encoder */
1554 encoder_clear(self);
1555 Py_TYPE(self)->tp_free(self);
1556}
1557
1558static int
1559encoder_traverse(PyObject *self, visitproc visit, void *arg)
1560{
1561 PyEncoderObject *s;
1562 assert(PyEncoder_Check(self));
1563 s = (PyEncoderObject *)self;
1564 Py_VISIT(s->markers);
1565 Py_VISIT(s->defaultfn);
1566 Py_VISIT(s->encoder);
1567 Py_VISIT(s->indent);
1568 Py_VISIT(s->key_separator);
1569 Py_VISIT(s->item_separator);
1570 Py_VISIT(s->sort_keys);
1571 Py_VISIT(s->skipkeys);
1572 return 0;
1573}
1574
1575static int
1576encoder_clear(PyObject *self)
1577{
1578 /* Deallocate Encoder */
1579 PyEncoderObject *s;
1580 assert(PyEncoder_Check(self));
1581 s = (PyEncoderObject *)self;
1582 Py_CLEAR(s->markers);
1583 Py_CLEAR(s->defaultfn);
1584 Py_CLEAR(s->encoder);
1585 Py_CLEAR(s->indent);
1586 Py_CLEAR(s->key_separator);
1587 Py_CLEAR(s->item_separator);
1588 Py_CLEAR(s->sort_keys);
1589 Py_CLEAR(s->skipkeys);
1590 return 0;
1591}
1592
1593PyDoc_STRVAR(encoder_doc, "_iterencode(obj, _current_indent_level) -> iterable");
1594
1595static
1596PyTypeObject PyEncoderType = {
1597 PyVarObject_HEAD_INIT(NULL, 0)
1598 "_json.Encoder", /* tp_name */
1599 sizeof(PyEncoderObject), /* tp_basicsize */
1600 0, /* tp_itemsize */
1601 encoder_dealloc, /* tp_dealloc */
1602 0, /* tp_print */
1603 0, /* tp_getattr */
1604 0, /* tp_setattr */
1605 0, /* tp_compare */
1606 0, /* tp_repr */
1607 0, /* tp_as_number */
1608 0, /* tp_as_sequence */
1609 0, /* tp_as_mapping */
1610 0, /* tp_hash */
1611 encoder_call, /* tp_call */
1612 0, /* tp_str */
1613 0, /* tp_getattro */
1614 0, /* tp_setattro */
1615 0, /* tp_as_buffer */
1616 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC, /* tp_flags */
1617 encoder_doc, /* tp_doc */
1618 encoder_traverse, /* tp_traverse */
1619 encoder_clear, /* tp_clear */
1620 0, /* tp_richcompare */
1621 0, /* tp_weaklistoffset */
1622 0, /* tp_iter */
1623 0, /* tp_iternext */
1624 0, /* tp_methods */
1625 encoder_members, /* tp_members */
1626 0, /* tp_getset */
1627 0, /* tp_base */
1628 0, /* tp_dict */
1629 0, /* tp_descr_get */
1630 0, /* tp_descr_set */
1631 0, /* tp_dictoffset */
1632 encoder_init, /* tp_init */
1633 0, /* tp_alloc */
1634 encoder_new, /* tp_new */
1635 0, /* tp_free */
1636};
1637
1638static PyMethodDef speedups_methods[] = {
1639 {"encode_basestring_ascii",
1640 (PyCFunction)py_encode_basestring_ascii,
1641 METH_O,
1642 pydoc_encode_basestring_ascii},
1643 {"scanstring",
1644 (PyCFunction)py_scanstring,
1645 METH_VARARGS,
1646 pydoc_scanstring},
Christian Heimes90540002008-05-08 14:29:10 +00001647 {NULL, NULL, 0, NULL}
1648};
1649
1650PyDoc_STRVAR(module_doc,
1651"json speedups\n");
1652
Martin v. Löwis1a214512008-06-11 05:26:20 +00001653static struct PyModuleDef jsonmodule = {
1654 PyModuleDef_HEAD_INIT,
1655 "_json",
1656 module_doc,
1657 -1,
Benjamin Petersonc6b607d2009-05-02 12:36:44 +00001658 speedups_methods,
Martin v. Löwis1a214512008-06-11 05:26:20 +00001659 NULL,
1660 NULL,
1661 NULL,
1662 NULL
1663};
1664
1665PyObject*
1666PyInit__json(void)
Christian Heimes90540002008-05-08 14:29:10 +00001667{
Benjamin Petersonc6b607d2009-05-02 12:36:44 +00001668 PyObject *m = PyModule_Create(&jsonmodule);
1669 if (!m)
1670 return NULL;
1671 PyScannerType.tp_new = PyType_GenericNew;
1672 if (PyType_Ready(&PyScannerType) < 0)
1673 goto fail;
1674 PyEncoderType.tp_new = PyType_GenericNew;
1675 if (PyType_Ready(&PyEncoderType) < 0)
1676 goto fail;
1677 Py_INCREF((PyObject*)&PyScannerType);
1678 if (PyModule_AddObject(m, "make_scanner", (PyObject*)&PyScannerType) < 0) {
1679 Py_DECREF((PyObject*)&PyScannerType);
1680 goto fail;
1681 }
1682 Py_INCREF((PyObject*)&PyEncoderType);
1683 if (PyModule_AddObject(m, "make_encoder", (PyObject*)&PyEncoderType) < 0) {
1684 Py_DECREF((PyObject*)&PyEncoderType);
1685 goto fail;
1686 }
1687 return m;
1688 fail:
1689 Py_DECREF(m);
1690 return NULL;
Christian Heimes90540002008-05-08 14:29:10 +00001691}