Martin v. Löwis | 7090ed1 | 2001-09-19 10:37:50 +0000 | [diff] [blame] | 1 | #include "Python.h" |
Fred Drake | 4113b13 | 2001-03-24 19:58:26 +0000 | [diff] [blame] | 2 | #include <ctype.h> |
| 3 | |
Martin v. Löwis | 0078f6c | 2001-01-21 10:18:10 +0000 | [diff] [blame] | 4 | #include "frameobject.h" |
Fred Drake | a77254a | 2000-09-29 19:23:29 +0000 | [diff] [blame] | 5 | #include "expat.h" |
Martin v. Löwis | 0078f6c | 2001-01-21 10:18:10 +0000 | [diff] [blame] | 6 | |
Fredrik Lundh | c334504 | 2005-12-13 19:49:55 +0000 | [diff] [blame] | 7 | #include "pyexpat.h" |
| 8 | |
Martin v. Löwis | c847f40 | 2003-01-21 11:09:21 +0000 | [diff] [blame] | 9 | #define XML_COMBINED_VERSION (10000*XML_MAJOR_VERSION+100*XML_MINOR_VERSION+XML_MICRO_VERSION) |
| 10 | |
Jeremy Hylton | 9263f57 | 2003-06-27 16:13:17 +0000 | [diff] [blame] | 11 | #define FIX_TRACE |
Martin v. Löwis | 339d0f7 | 2001-08-17 18:39:25 +0000 | [diff] [blame] | 12 | |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 13 | enum HandlerTypes { |
| 14 | StartElement, |
| 15 | EndElement, |
| 16 | ProcessingInstruction, |
| 17 | CharacterData, |
| 18 | UnparsedEntityDecl, |
| 19 | NotationDecl, |
| 20 | StartNamespaceDecl, |
| 21 | EndNamespaceDecl, |
| 22 | Comment, |
| 23 | StartCdataSection, |
| 24 | EndCdataSection, |
| 25 | Default, |
| 26 | DefaultHandlerExpand, |
| 27 | NotStandalone, |
Martin v. Löwis | 0078f6c | 2001-01-21 10:18:10 +0000 | [diff] [blame] | 28 | ExternalEntityRef, |
| 29 | StartDoctypeDecl, |
| 30 | EndDoctypeDecl, |
Fred Drake | 85d835f | 2001-02-08 15:39:08 +0000 | [diff] [blame] | 31 | EntityDecl, |
| 32 | XmlDecl, |
| 33 | ElementDecl, |
| 34 | AttlistDecl, |
Martin v. Löwis | c847f40 | 2003-01-21 11:09:21 +0000 | [diff] [blame] | 35 | #if XML_COMBINED_VERSION >= 19504 |
Martin v. Löwis | 069dde2 | 2003-01-21 10:58:18 +0000 | [diff] [blame] | 36 | SkippedEntity, |
Martin v. Löwis | c847f40 | 2003-01-21 11:09:21 +0000 | [diff] [blame] | 37 | #endif |
Fred Drake | 85d835f | 2001-02-08 15:39:08 +0000 | [diff] [blame] | 38 | _DummyDecl |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 39 | }; |
| 40 | |
| 41 | static PyObject *ErrorObject; |
| 42 | |
| 43 | /* ----------------------------------------------------- */ |
| 44 | |
| 45 | /* Declarations for objects of type xmlparser */ |
| 46 | |
| 47 | typedef struct { |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 48 | PyObject_HEAD |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 49 | |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 50 | XML_Parser itself; |
Fred Drake | 85d835f | 2001-02-08 15:39:08 +0000 | [diff] [blame] | 51 | int ordered_attributes; /* Return attributes as a list. */ |
| 52 | int specified_attributes; /* Report only specified attributes. */ |
Fred Drake | bd6101c | 2001-02-14 18:29:45 +0000 | [diff] [blame] | 53 | int in_callback; /* Is a callback active? */ |
Martin v. Löwis | 069dde2 | 2003-01-21 10:58:18 +0000 | [diff] [blame] | 54 | int ns_prefixes; /* Namespace-triplets mode? */ |
Fred Drake | 2a3d7db | 2002-06-28 22:56:48 +0000 | [diff] [blame] | 55 | XML_Char *buffer; /* Buffer used when accumulating characters */ |
| 56 | /* NULL if not enabled */ |
| 57 | int buffer_size; /* Size of buffer, in XML_Char units */ |
| 58 | int buffer_used; /* Buffer units in use */ |
Fred Drake | b91a36b | 2002-06-27 19:40:48 +0000 | [diff] [blame] | 59 | PyObject *intern; /* Dictionary to intern strings */ |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 60 | PyObject **handlers; |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 61 | } xmlparseobject; |
| 62 | |
Fred Drake | 2a3d7db | 2002-06-28 22:56:48 +0000 | [diff] [blame] | 63 | #define CHARACTER_DATA_BUFFER_SIZE 8192 |
| 64 | |
Jeremy Hylton | 938ace6 | 2002-07-17 16:30:39 +0000 | [diff] [blame] | 65 | static PyTypeObject Xmlparsetype; |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 66 | |
Fred Drake | 117ac85 | 2002-09-24 16:24:54 +0000 | [diff] [blame] | 67 | typedef void (*xmlhandlersetter)(XML_Parser self, void *meth); |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 68 | typedef void* xmlhandler; |
| 69 | |
Andrew M. Kuchling | beba056 | 2000-06-27 00:33:30 +0000 | [diff] [blame] | 70 | struct HandlerInfo { |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 71 | const char *name; |
| 72 | xmlhandlersetter setter; |
| 73 | xmlhandler handler; |
Martin v. Löwis | 0078f6c | 2001-01-21 10:18:10 +0000 | [diff] [blame] | 74 | PyCodeObject *tb_code; |
Fred Drake | 71b63ff | 2002-06-28 22:29:01 +0000 | [diff] [blame] | 75 | PyObject *nameobj; |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 76 | }; |
| 77 | |
Jeremy Hylton | 938ace6 | 2002-07-17 16:30:39 +0000 | [diff] [blame] | 78 | static struct HandlerInfo handler_info[64]; |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 79 | |
Fred Drake | bd6101c | 2001-02-14 18:29:45 +0000 | [diff] [blame] | 80 | /* Set an integer attribute on the error object; return true on success, |
| 81 | * false on an exception. |
| 82 | */ |
| 83 | static int |
| 84 | set_error_attr(PyObject *err, char *name, int value) |
| 85 | { |
Christian Heimes | 217cfd1 | 2007-12-02 14:31:20 +0000 | [diff] [blame] | 86 | PyObject *v = PyLong_FromLong(value); |
Fred Drake | 85d835f | 2001-02-08 15:39:08 +0000 | [diff] [blame] | 87 | |
Neal Norwitz | 2f5e990 | 2006-03-08 06:36:45 +0000 | [diff] [blame] | 88 | if (v == NULL || PyObject_SetAttrString(err, name, v) == -1) { |
| 89 | Py_XDECREF(v); |
Fred Drake | bd6101c | 2001-02-14 18:29:45 +0000 | [diff] [blame] | 90 | return 0; |
| 91 | } |
Michael W. Hudson | 0bb8454 | 2004-08-03 11:31:31 +0000 | [diff] [blame] | 92 | Py_DECREF(v); |
Fred Drake | bd6101c | 2001-02-14 18:29:45 +0000 | [diff] [blame] | 93 | return 1; |
| 94 | } |
| 95 | |
| 96 | /* Build and set an Expat exception, including positioning |
| 97 | * information. Always returns NULL. |
| 98 | */ |
Fred Drake | 85d835f | 2001-02-08 15:39:08 +0000 | [diff] [blame] | 99 | static PyObject * |
Martin v. Löwis | 069dde2 | 2003-01-21 10:58:18 +0000 | [diff] [blame] | 100 | set_error(xmlparseobject *self, enum XML_Error code) |
Fred Drake | 85d835f | 2001-02-08 15:39:08 +0000 | [diff] [blame] | 101 | { |
| 102 | PyObject *err; |
| 103 | char buffer[256]; |
| 104 | XML_Parser parser = self->itself; |
Fred Drake | bd6101c | 2001-02-14 18:29:45 +0000 | [diff] [blame] | 105 | int lineno = XML_GetErrorLineNumber(parser); |
| 106 | int column = XML_GetErrorColumnNumber(parser); |
Fred Drake | 85d835f | 2001-02-08 15:39:08 +0000 | [diff] [blame] | 107 | |
Martin v. Löwis | 6b2cf0e | 2002-06-30 06:03:35 +0000 | [diff] [blame] | 108 | /* There is no risk of overflowing this buffer, since |
| 109 | even for 64-bit integers, there is sufficient space. */ |
| 110 | sprintf(buffer, "%.200s: line %i, column %i", |
Fred Drake | bd6101c | 2001-02-14 18:29:45 +0000 | [diff] [blame] | 111 | XML_ErrorString(code), lineno, column); |
Fred Drake | 85d835f | 2001-02-08 15:39:08 +0000 | [diff] [blame] | 112 | err = PyObject_CallFunction(ErrorObject, "s", buffer); |
Fred Drake | bd6101c | 2001-02-14 18:29:45 +0000 | [diff] [blame] | 113 | if ( err != NULL |
| 114 | && set_error_attr(err, "code", code) |
| 115 | && set_error_attr(err, "offset", column) |
| 116 | && set_error_attr(err, "lineno", lineno)) { |
| 117 | PyErr_SetObject(ErrorObject, err); |
Fred Drake | 85d835f | 2001-02-08 15:39:08 +0000 | [diff] [blame] | 118 | } |
Neal Norwitz | 2f5e990 | 2006-03-08 06:36:45 +0000 | [diff] [blame] | 119 | Py_XDECREF(err); |
Fred Drake | 85d835f | 2001-02-08 15:39:08 +0000 | [diff] [blame] | 120 | return NULL; |
| 121 | } |
| 122 | |
Fred Drake | 71b63ff | 2002-06-28 22:29:01 +0000 | [diff] [blame] | 123 | static int |
| 124 | have_handler(xmlparseobject *self, int type) |
| 125 | { |
| 126 | PyObject *handler = self->handlers[type]; |
| 127 | return handler != NULL; |
| 128 | } |
| 129 | |
| 130 | static PyObject * |
| 131 | get_handler_name(struct HandlerInfo *hinfo) |
| 132 | { |
| 133 | PyObject *name = hinfo->nameobj; |
| 134 | if (name == NULL) { |
Neal Norwitz | 392c5be | 2007-08-25 17:20:32 +0000 | [diff] [blame] | 135 | name = PyUnicode_FromString(hinfo->name); |
Fred Drake | 71b63ff | 2002-06-28 22:29:01 +0000 | [diff] [blame] | 136 | hinfo->nameobj = name; |
| 137 | } |
| 138 | Py_XINCREF(name); |
| 139 | return name; |
| 140 | } |
| 141 | |
Fred Drake | 85d835f | 2001-02-08 15:39:08 +0000 | [diff] [blame] | 142 | |
Andrew M. Kuchling | beba056 | 2000-06-27 00:33:30 +0000 | [diff] [blame] | 143 | /* Convert a string of XML_Chars into a Unicode string. |
| 144 | Returns None if str is a null pointer. */ |
| 145 | |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 146 | static PyObject * |
Fred Drake | b91a36b | 2002-06-27 19:40:48 +0000 | [diff] [blame] | 147 | conv_string_to_unicode(const XML_Char *str) |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 148 | { |
Fred Drake | 71b63ff | 2002-06-28 22:29:01 +0000 | [diff] [blame] | 149 | /* XXX currently this code assumes that XML_Char is 8-bit, |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 150 | and hence in UTF-8. */ |
| 151 | /* UTF-8 from Expat, Unicode desired */ |
| 152 | if (str == NULL) { |
| 153 | Py_INCREF(Py_None); |
| 154 | return Py_None; |
| 155 | } |
Fred Drake | 71b63ff | 2002-06-28 22:29:01 +0000 | [diff] [blame] | 156 | return PyUnicode_DecodeUTF8(str, strlen(str), "strict"); |
Andrew M. Kuchling | beba056 | 2000-06-27 00:33:30 +0000 | [diff] [blame] | 157 | } |
| 158 | |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 159 | static PyObject * |
| 160 | conv_string_len_to_unicode(const XML_Char *str, int len) |
| 161 | { |
Fred Drake | 71b63ff | 2002-06-28 22:29:01 +0000 | [diff] [blame] | 162 | /* XXX currently this code assumes that XML_Char is 8-bit, |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 163 | and hence in UTF-8. */ |
| 164 | /* UTF-8 from Expat, Unicode desired */ |
| 165 | if (str == NULL) { |
| 166 | Py_INCREF(Py_None); |
| 167 | return Py_None; |
| 168 | } |
Fred Drake | 6f98762 | 2000-08-25 18:03:30 +0000 | [diff] [blame] | 169 | return PyUnicode_DecodeUTF8((const char *)str, len, "strict"); |
Andrew M. Kuchling | beba056 | 2000-06-27 00:33:30 +0000 | [diff] [blame] | 170 | } |
Andrew M. Kuchling | beba056 | 2000-06-27 00:33:30 +0000 | [diff] [blame] | 171 | |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 172 | /* Callback routines */ |
| 173 | |
Martin v. Löwis | 5b68ce3 | 2001-10-21 08:53:52 +0000 | [diff] [blame] | 174 | static void clear_handlers(xmlparseobject *self, int initial); |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 175 | |
Martin v. Löwis | 069dde2 | 2003-01-21 10:58:18 +0000 | [diff] [blame] | 176 | /* This handler is used when an error has been detected, in the hope |
| 177 | that actual parsing can be terminated early. This will only help |
| 178 | if an external entity reference is encountered. */ |
| 179 | static int |
| 180 | error_external_entity_ref_handler(XML_Parser parser, |
| 181 | const XML_Char *context, |
| 182 | const XML_Char *base, |
| 183 | const XML_Char *systemId, |
| 184 | const XML_Char *publicId) |
| 185 | { |
| 186 | return 0; |
| 187 | } |
| 188 | |
Thomas Wouters | 0e3f591 | 2006-08-11 14:57:12 +0000 | [diff] [blame] | 189 | /* Dummy character data handler used when an error (exception) has |
| 190 | been detected, and the actual parsing can be terminated early. |
| 191 | This is needed since character data handler can't be safely removed |
| 192 | from within the character data handler, but can be replaced. It is |
| 193 | used only from the character data handler trampoline, and must be |
| 194 | used right after `flag_error()` is called. */ |
| 195 | static void |
| 196 | noop_character_data_handler(void *userData, const XML_Char *data, int len) |
| 197 | { |
| 198 | /* Do nothing. */ |
| 199 | } |
| 200 | |
Fred Drake | 6f98762 | 2000-08-25 18:03:30 +0000 | [diff] [blame] | 201 | static void |
| 202 | flag_error(xmlparseobject *self) |
| 203 | { |
Martin v. Löwis | 5b68ce3 | 2001-10-21 08:53:52 +0000 | [diff] [blame] | 204 | clear_handlers(self, 0); |
Martin v. Löwis | 069dde2 | 2003-01-21 10:58:18 +0000 | [diff] [blame] | 205 | XML_SetExternalEntityRefHandler(self->itself, |
| 206 | error_external_entity_ref_handler); |
Martin v. Löwis | 0078f6c | 2001-01-21 10:18:10 +0000 | [diff] [blame] | 207 | } |
| 208 | |
| 209 | static PyCodeObject* |
| 210 | getcode(enum HandlerTypes slot, char* func_name, int lineno) |
| 211 | { |
Fred Drake | bd6101c | 2001-02-14 18:29:45 +0000 | [diff] [blame] | 212 | PyObject *code = NULL; |
| 213 | PyObject *name = NULL; |
| 214 | PyObject *nulltuple = NULL; |
| 215 | PyObject *filename = NULL; |
| 216 | |
| 217 | if (handler_info[slot].tb_code == NULL) { |
Christian Heimes | 72b710a | 2008-05-26 13:28:38 +0000 | [diff] [blame] | 218 | code = PyBytes_FromString(""); |
Fred Drake | bd6101c | 2001-02-14 18:29:45 +0000 | [diff] [blame] | 219 | if (code == NULL) |
| 220 | goto failed; |
Guido van Rossum | 00bc0e0 | 2007-10-15 02:52:41 +0000 | [diff] [blame] | 221 | name = PyUnicode_FromString(func_name); |
Fred Drake | bd6101c | 2001-02-14 18:29:45 +0000 | [diff] [blame] | 222 | if (name == NULL) |
| 223 | goto failed; |
| 224 | nulltuple = PyTuple_New(0); |
| 225 | if (nulltuple == NULL) |
| 226 | goto failed; |
Guido van Rossum | 00bc0e0 | 2007-10-15 02:52:41 +0000 | [diff] [blame] | 227 | filename = PyUnicode_DecodeFSDefault(__FILE__); |
Fred Drake | bd6101c | 2001-02-14 18:29:45 +0000 | [diff] [blame] | 228 | handler_info[slot].tb_code = |
| 229 | PyCode_New(0, /* argcount */ |
Guido van Rossum | 4f72a78 | 2006-10-27 23:31:49 +0000 | [diff] [blame] | 230 | 0, /* kwonlyargcount */ |
Fred Drake | bd6101c | 2001-02-14 18:29:45 +0000 | [diff] [blame] | 231 | 0, /* nlocals */ |
| 232 | 0, /* stacksize */ |
| 233 | 0, /* flags */ |
| 234 | code, /* code */ |
| 235 | nulltuple, /* consts */ |
| 236 | nulltuple, /* names */ |
| 237 | nulltuple, /* varnames */ |
Martin v. Löwis | 76192ee | 2001-02-06 09:34:40 +0000 | [diff] [blame] | 238 | #if PYTHON_API_VERSION >= 1010 |
Fred Drake | bd6101c | 2001-02-14 18:29:45 +0000 | [diff] [blame] | 239 | nulltuple, /* freevars */ |
| 240 | nulltuple, /* cellvars */ |
Martin v. Löwis | 76192ee | 2001-02-06 09:34:40 +0000 | [diff] [blame] | 241 | #endif |
Fred Drake | bd6101c | 2001-02-14 18:29:45 +0000 | [diff] [blame] | 242 | filename, /* filename */ |
| 243 | name, /* name */ |
| 244 | lineno, /* firstlineno */ |
| 245 | code /* lnotab */ |
| 246 | ); |
| 247 | if (handler_info[slot].tb_code == NULL) |
| 248 | goto failed; |
| 249 | Py_DECREF(code); |
| 250 | Py_DECREF(nulltuple); |
| 251 | Py_DECREF(filename); |
| 252 | Py_DECREF(name); |
| 253 | } |
| 254 | return handler_info[slot].tb_code; |
| 255 | failed: |
| 256 | Py_XDECREF(code); |
| 257 | Py_XDECREF(name); |
| 258 | return NULL; |
Martin v. Löwis | 0078f6c | 2001-01-21 10:18:10 +0000 | [diff] [blame] | 259 | } |
| 260 | |
Jeremy Hylton | 9263f57 | 2003-06-27 16:13:17 +0000 | [diff] [blame] | 261 | #ifdef FIX_TRACE |
Martin v. Löwis | 7d6e19d | 2002-08-04 08:24:49 +0000 | [diff] [blame] | 262 | static int |
| 263 | trace_frame(PyThreadState *tstate, PyFrameObject *f, int code, PyObject *val) |
| 264 | { |
| 265 | int result = 0; |
| 266 | if (!tstate->use_tracing || tstate->tracing) |
| 267 | return 0; |
| 268 | if (tstate->c_profilefunc != NULL) { |
| 269 | tstate->tracing++; |
| 270 | result = tstate->c_profilefunc(tstate->c_profileobj, |
| 271 | f, code , val); |
| 272 | tstate->use_tracing = ((tstate->c_tracefunc != NULL) |
| 273 | || (tstate->c_profilefunc != NULL)); |
| 274 | tstate->tracing--; |
| 275 | if (result) |
| 276 | return result; |
| 277 | } |
| 278 | if (tstate->c_tracefunc != NULL) { |
| 279 | tstate->tracing++; |
| 280 | result = tstate->c_tracefunc(tstate->c_traceobj, |
| 281 | f, code , val); |
| 282 | tstate->use_tracing = ((tstate->c_tracefunc != NULL) |
| 283 | || (tstate->c_profilefunc != NULL)); |
| 284 | tstate->tracing--; |
| 285 | } |
| 286 | return result; |
| 287 | } |
Jeremy Hylton | 9263f57 | 2003-06-27 16:13:17 +0000 | [diff] [blame] | 288 | |
| 289 | static int |
| 290 | trace_frame_exc(PyThreadState *tstate, PyFrameObject *f) |
| 291 | { |
| 292 | PyObject *type, *value, *traceback, *arg; |
| 293 | int err; |
| 294 | |
| 295 | if (tstate->c_tracefunc == NULL) |
| 296 | return 0; |
| 297 | |
| 298 | PyErr_Fetch(&type, &value, &traceback); |
| 299 | if (value == NULL) { |
| 300 | value = Py_None; |
| 301 | Py_INCREF(value); |
| 302 | } |
Martin v. Löwis | 9171f02 | 2004-10-13 19:50:11 +0000 | [diff] [blame] | 303 | #if PY_VERSION_HEX < 0x02040000 |
| 304 | arg = Py_BuildValue("(OOO)", type, value, traceback); |
| 305 | #else |
Raymond Hettinger | 8ae4689 | 2003-10-12 19:09:37 +0000 | [diff] [blame] | 306 | arg = PyTuple_Pack(3, type, value, traceback); |
Martin v. Löwis | 9171f02 | 2004-10-13 19:50:11 +0000 | [diff] [blame] | 307 | #endif |
Jeremy Hylton | 9263f57 | 2003-06-27 16:13:17 +0000 | [diff] [blame] | 308 | if (arg == NULL) { |
| 309 | PyErr_Restore(type, value, traceback); |
| 310 | return 0; |
| 311 | } |
| 312 | err = trace_frame(tstate, f, PyTrace_EXCEPTION, arg); |
| 313 | Py_DECREF(arg); |
| 314 | if (err == 0) |
| 315 | PyErr_Restore(type, value, traceback); |
| 316 | else { |
| 317 | Py_XDECREF(type); |
| 318 | Py_XDECREF(value); |
| 319 | Py_XDECREF(traceback); |
| 320 | } |
| 321 | return err; |
| 322 | } |
Martin v. Löwis | 069dde2 | 2003-01-21 10:58:18 +0000 | [diff] [blame] | 323 | #endif |
Martin v. Löwis | 7d6e19d | 2002-08-04 08:24:49 +0000 | [diff] [blame] | 324 | |
Martin v. Löwis | 0078f6c | 2001-01-21 10:18:10 +0000 | [diff] [blame] | 325 | static PyObject* |
Fred Drake | 39689c5 | 2004-08-13 03:12:57 +0000 | [diff] [blame] | 326 | call_with_frame(PyCodeObject *c, PyObject* func, PyObject* args, |
| 327 | xmlparseobject *self) |
Martin v. Löwis | 0078f6c | 2001-01-21 10:18:10 +0000 | [diff] [blame] | 328 | { |
Fred Drake | bd6101c | 2001-02-14 18:29:45 +0000 | [diff] [blame] | 329 | PyThreadState *tstate = PyThreadState_GET(); |
| 330 | PyFrameObject *f; |
| 331 | PyObject *res; |
| 332 | |
| 333 | if (c == NULL) |
| 334 | return NULL; |
Martin v. Löwis | 7d6e19d | 2002-08-04 08:24:49 +0000 | [diff] [blame] | 335 | |
Jeremy Hylton | 9263f57 | 2003-06-27 16:13:17 +0000 | [diff] [blame] | 336 | f = PyFrame_New(tstate, c, PyEval_GetGlobals(), NULL); |
Fred Drake | bd6101c | 2001-02-14 18:29:45 +0000 | [diff] [blame] | 337 | if (f == NULL) |
| 338 | return NULL; |
| 339 | tstate->frame = f; |
Jeremy Hylton | 9263f57 | 2003-06-27 16:13:17 +0000 | [diff] [blame] | 340 | #ifdef FIX_TRACE |
| 341 | if (trace_frame(tstate, f, PyTrace_CALL, Py_None) < 0) { |
Martin v. Löwis | 7d6e19d | 2002-08-04 08:24:49 +0000 | [diff] [blame] | 342 | return NULL; |
| 343 | } |
Martin v. Löwis | 069dde2 | 2003-01-21 10:58:18 +0000 | [diff] [blame] | 344 | #endif |
Fred Drake | bd6101c | 2001-02-14 18:29:45 +0000 | [diff] [blame] | 345 | res = PyEval_CallObject(func, args); |
Jeremy Hylton | 9263f57 | 2003-06-27 16:13:17 +0000 | [diff] [blame] | 346 | if (res == NULL) { |
| 347 | if (tstate->curexc_traceback == NULL) |
| 348 | PyTraceBack_Here(f); |
Fred Drake | 39689c5 | 2004-08-13 03:12:57 +0000 | [diff] [blame] | 349 | XML_StopParser(self->itself, XML_FALSE); |
Jeremy Hylton | 9263f57 | 2003-06-27 16:13:17 +0000 | [diff] [blame] | 350 | #ifdef FIX_TRACE |
| 351 | if (trace_frame_exc(tstate, f) < 0) { |
| 352 | return NULL; |
| 353 | } |
| 354 | } |
Martin v. Löwis | 7d6e19d | 2002-08-04 08:24:49 +0000 | [diff] [blame] | 355 | else { |
Jeremy Hylton | 9263f57 | 2003-06-27 16:13:17 +0000 | [diff] [blame] | 356 | if (trace_frame(tstate, f, PyTrace_RETURN, res) < 0) { |
Martin v. Löwis | 7d6e19d | 2002-08-04 08:24:49 +0000 | [diff] [blame] | 357 | Py_XDECREF(res); |
| 358 | res = NULL; |
| 359 | } |
| 360 | } |
Jeremy Hylton | 9263f57 | 2003-06-27 16:13:17 +0000 | [diff] [blame] | 361 | #else |
| 362 | } |
Martin v. Löwis | 069dde2 | 2003-01-21 10:58:18 +0000 | [diff] [blame] | 363 | #endif |
Fred Drake | bd6101c | 2001-02-14 18:29:45 +0000 | [diff] [blame] | 364 | tstate->frame = f->f_back; |
| 365 | Py_DECREF(f); |
| 366 | return res; |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 367 | } |
| 368 | |
Fred Drake | b91a36b | 2002-06-27 19:40:48 +0000 | [diff] [blame] | 369 | static PyObject* |
| 370 | string_intern(xmlparseobject *self, const char* str) |
| 371 | { |
Guido van Rossum | 4ca9471 | 2007-07-23 17:42:32 +0000 | [diff] [blame] | 372 | PyObject *result = conv_string_to_unicode(str); |
Fred Drake | b91a36b | 2002-06-27 19:40:48 +0000 | [diff] [blame] | 373 | PyObject *value; |
Neal Norwitz | 484d9a4 | 2005-09-30 04:46:49 +0000 | [diff] [blame] | 374 | /* result can be NULL if the unicode conversion failed. */ |
| 375 | if (!result) |
| 376 | return result; |
Fred Drake | b91a36b | 2002-06-27 19:40:48 +0000 | [diff] [blame] | 377 | if (!self->intern) |
| 378 | return result; |
| 379 | value = PyDict_GetItem(self->intern, result); |
| 380 | if (!value) { |
| 381 | if (PyDict_SetItem(self->intern, result, result) == 0) |
| 382 | return result; |
| 383 | else |
| 384 | return NULL; |
| 385 | } |
| 386 | Py_INCREF(value); |
| 387 | Py_DECREF(result); |
| 388 | return value; |
| 389 | } |
| 390 | |
Fred Drake | 2a3d7db | 2002-06-28 22:56:48 +0000 | [diff] [blame] | 391 | /* Return 0 on success, -1 on exception. |
| 392 | * flag_error() will be called before return if needed. |
| 393 | */ |
| 394 | static int |
| 395 | call_character_handler(xmlparseobject *self, const XML_Char *buffer, int len) |
| 396 | { |
| 397 | PyObject *args; |
| 398 | PyObject *temp; |
| 399 | |
| 400 | args = PyTuple_New(1); |
| 401 | if (args == NULL) |
| 402 | return -1; |
Guido van Rossum | 4ca9471 | 2007-07-23 17:42:32 +0000 | [diff] [blame] | 403 | temp = (conv_string_len_to_unicode(buffer, len)); |
Fred Drake | 2a3d7db | 2002-06-28 22:56:48 +0000 | [diff] [blame] | 404 | if (temp == NULL) { |
| 405 | Py_DECREF(args); |
| 406 | flag_error(self); |
Thomas Wouters | 0e3f591 | 2006-08-11 14:57:12 +0000 | [diff] [blame] | 407 | XML_SetCharacterDataHandler(self->itself, |
| 408 | noop_character_data_handler); |
Fred Drake | 2a3d7db | 2002-06-28 22:56:48 +0000 | [diff] [blame] | 409 | return -1; |
| 410 | } |
| 411 | PyTuple_SET_ITEM(args, 0, temp); |
| 412 | /* temp is now a borrowed reference; consider it unused. */ |
| 413 | self->in_callback = 1; |
| 414 | temp = call_with_frame(getcode(CharacterData, "CharacterData", __LINE__), |
Fred Drake | 39689c5 | 2004-08-13 03:12:57 +0000 | [diff] [blame] | 415 | self->handlers[CharacterData], args, self); |
Fred Drake | 2a3d7db | 2002-06-28 22:56:48 +0000 | [diff] [blame] | 416 | /* temp is an owned reference again, or NULL */ |
| 417 | self->in_callback = 0; |
| 418 | Py_DECREF(args); |
| 419 | if (temp == NULL) { |
| 420 | flag_error(self); |
Thomas Wouters | 0e3f591 | 2006-08-11 14:57:12 +0000 | [diff] [blame] | 421 | XML_SetCharacterDataHandler(self->itself, |
| 422 | noop_character_data_handler); |
Fred Drake | 2a3d7db | 2002-06-28 22:56:48 +0000 | [diff] [blame] | 423 | return -1; |
| 424 | } |
| 425 | Py_DECREF(temp); |
| 426 | return 0; |
| 427 | } |
| 428 | |
| 429 | static int |
| 430 | flush_character_buffer(xmlparseobject *self) |
| 431 | { |
| 432 | int rc; |
| 433 | if (self->buffer == NULL || self->buffer_used == 0) |
| 434 | return 0; |
| 435 | rc = call_character_handler(self, self->buffer, self->buffer_used); |
| 436 | self->buffer_used = 0; |
| 437 | return rc; |
| 438 | } |
| 439 | |
| 440 | static void |
| 441 | my_CharacterDataHandler(void *userData, const XML_Char *data, int len) |
| 442 | { |
| 443 | xmlparseobject *self = (xmlparseobject *) userData; |
| 444 | if (self->buffer == NULL) |
| 445 | call_character_handler(self, data, len); |
| 446 | else { |
| 447 | if ((self->buffer_used + len) > self->buffer_size) { |
| 448 | if (flush_character_buffer(self) < 0) |
| 449 | return; |
| 450 | /* handler might have changed; drop the rest on the floor |
| 451 | * if there isn't a handler anymore |
| 452 | */ |
| 453 | if (!have_handler(self, CharacterData)) |
| 454 | return; |
| 455 | } |
| 456 | if (len > self->buffer_size) { |
| 457 | call_character_handler(self, data, len); |
| 458 | self->buffer_used = 0; |
| 459 | } |
| 460 | else { |
| 461 | memcpy(self->buffer + self->buffer_used, |
| 462 | data, len * sizeof(XML_Char)); |
| 463 | self->buffer_used += len; |
| 464 | } |
| 465 | } |
| 466 | } |
| 467 | |
Fred Drake | 85d835f | 2001-02-08 15:39:08 +0000 | [diff] [blame] | 468 | static void |
| 469 | my_StartElementHandler(void *userData, |
Fred Drake | 71b63ff | 2002-06-28 22:29:01 +0000 | [diff] [blame] | 470 | const XML_Char *name, const XML_Char *atts[]) |
Fred Drake | 85d835f | 2001-02-08 15:39:08 +0000 | [diff] [blame] | 471 | { |
| 472 | xmlparseobject *self = (xmlparseobject *)userData; |
| 473 | |
Fred Drake | 71b63ff | 2002-06-28 22:29:01 +0000 | [diff] [blame] | 474 | if (have_handler(self, StartElement)) { |
Fred Drake | 85d835f | 2001-02-08 15:39:08 +0000 | [diff] [blame] | 475 | PyObject *container, *rv, *args; |
| 476 | int i, max; |
| 477 | |
Fred Drake | 2a3d7db | 2002-06-28 22:56:48 +0000 | [diff] [blame] | 478 | if (flush_character_buffer(self) < 0) |
| 479 | return; |
Fred Drake | 85d835f | 2001-02-08 15:39:08 +0000 | [diff] [blame] | 480 | /* Set max to the number of slots filled in atts[]; max/2 is |
| 481 | * the number of attributes we need to process. |
| 482 | */ |
| 483 | if (self->specified_attributes) { |
| 484 | max = XML_GetSpecifiedAttributeCount(self->itself); |
| 485 | } |
| 486 | else { |
| 487 | max = 0; |
| 488 | while (atts[max] != NULL) |
| 489 | max += 2; |
| 490 | } |
| 491 | /* Build the container. */ |
| 492 | if (self->ordered_attributes) |
| 493 | container = PyList_New(max); |
| 494 | else |
| 495 | container = PyDict_New(); |
| 496 | if (container == NULL) { |
| 497 | flag_error(self); |
| 498 | return; |
| 499 | } |
| 500 | for (i = 0; i < max; i += 2) { |
Fred Drake | b91a36b | 2002-06-27 19:40:48 +0000 | [diff] [blame] | 501 | PyObject *n = string_intern(self, (XML_Char *) atts[i]); |
Fred Drake | 85d835f | 2001-02-08 15:39:08 +0000 | [diff] [blame] | 502 | PyObject *v; |
| 503 | if (n == NULL) { |
| 504 | flag_error(self); |
| 505 | Py_DECREF(container); |
| 506 | return; |
| 507 | } |
Guido van Rossum | 4ca9471 | 2007-07-23 17:42:32 +0000 | [diff] [blame] | 508 | v = conv_string_to_unicode((XML_Char *) atts[i+1]); |
Fred Drake | 85d835f | 2001-02-08 15:39:08 +0000 | [diff] [blame] | 509 | if (v == NULL) { |
| 510 | flag_error(self); |
| 511 | Py_DECREF(container); |
| 512 | Py_DECREF(n); |
| 513 | return; |
| 514 | } |
| 515 | if (self->ordered_attributes) { |
| 516 | PyList_SET_ITEM(container, i, n); |
| 517 | PyList_SET_ITEM(container, i+1, v); |
| 518 | } |
| 519 | else if (PyDict_SetItem(container, n, v)) { |
| 520 | flag_error(self); |
| 521 | Py_DECREF(n); |
| 522 | Py_DECREF(v); |
| 523 | return; |
| 524 | } |
| 525 | else { |
| 526 | Py_DECREF(n); |
| 527 | Py_DECREF(v); |
| 528 | } |
| 529 | } |
Neal Norwitz | 484d9a4 | 2005-09-30 04:46:49 +0000 | [diff] [blame] | 530 | args = string_intern(self, name); |
| 531 | if (args != NULL) |
| 532 | args = Py_BuildValue("(NN)", args, container); |
Fred Drake | 85d835f | 2001-02-08 15:39:08 +0000 | [diff] [blame] | 533 | if (args == NULL) { |
| 534 | Py_DECREF(container); |
| 535 | return; |
| 536 | } |
| 537 | /* Container is now a borrowed reference; ignore it. */ |
Fred Drake | bd6101c | 2001-02-14 18:29:45 +0000 | [diff] [blame] | 538 | self->in_callback = 1; |
| 539 | rv = call_with_frame(getcode(StartElement, "StartElement", __LINE__), |
Fred Drake | 39689c5 | 2004-08-13 03:12:57 +0000 | [diff] [blame] | 540 | self->handlers[StartElement], args, self); |
Fred Drake | bd6101c | 2001-02-14 18:29:45 +0000 | [diff] [blame] | 541 | self->in_callback = 0; |
| 542 | Py_DECREF(args); |
Fred Drake | 85d835f | 2001-02-08 15:39:08 +0000 | [diff] [blame] | 543 | if (rv == NULL) { |
| 544 | flag_error(self); |
| 545 | return; |
Fred Drake | bd6101c | 2001-02-14 18:29:45 +0000 | [diff] [blame] | 546 | } |
Fred Drake | 85d835f | 2001-02-08 15:39:08 +0000 | [diff] [blame] | 547 | Py_DECREF(rv); |
| 548 | } |
| 549 | } |
| 550 | |
| 551 | #define RC_HANDLER(RC, NAME, PARAMS, INIT, PARAM_FORMAT, CONVERSION, \ |
| 552 | RETURN, GETUSERDATA) \ |
| 553 | static RC \ |
| 554 | my_##NAME##Handler PARAMS {\ |
| 555 | xmlparseobject *self = GETUSERDATA ; \ |
| 556 | PyObject *args = NULL; \ |
| 557 | PyObject *rv = NULL; \ |
| 558 | INIT \ |
| 559 | \ |
Fred Drake | 71b63ff | 2002-06-28 22:29:01 +0000 | [diff] [blame] | 560 | if (have_handler(self, NAME)) { \ |
Fred Drake | 2a3d7db | 2002-06-28 22:56:48 +0000 | [diff] [blame] | 561 | if (flush_character_buffer(self) < 0) \ |
| 562 | return RETURN; \ |
Fred Drake | 85d835f | 2001-02-08 15:39:08 +0000 | [diff] [blame] | 563 | args = Py_BuildValue PARAM_FORMAT ;\ |
Martin v. Löwis | 1d7c55f | 2001-11-10 13:57:55 +0000 | [diff] [blame] | 564 | if (!args) { flag_error(self); return RETURN;} \ |
Fred Drake | bd6101c | 2001-02-14 18:29:45 +0000 | [diff] [blame] | 565 | self->in_callback = 1; \ |
Fred Drake | 85d835f | 2001-02-08 15:39:08 +0000 | [diff] [blame] | 566 | rv = call_with_frame(getcode(NAME,#NAME,__LINE__), \ |
Fred Drake | 39689c5 | 2004-08-13 03:12:57 +0000 | [diff] [blame] | 567 | self->handlers[NAME], args, self); \ |
Fred Drake | bd6101c | 2001-02-14 18:29:45 +0000 | [diff] [blame] | 568 | self->in_callback = 0; \ |
Fred Drake | 85d835f | 2001-02-08 15:39:08 +0000 | [diff] [blame] | 569 | Py_DECREF(args); \ |
| 570 | if (rv == NULL) { \ |
| 571 | flag_error(self); \ |
| 572 | return RETURN; \ |
| 573 | } \ |
| 574 | CONVERSION \ |
| 575 | Py_DECREF(rv); \ |
| 576 | } \ |
| 577 | return RETURN; \ |
| 578 | } |
| 579 | |
Fred Drake | 6f98762 | 2000-08-25 18:03:30 +0000 | [diff] [blame] | 580 | #define VOID_HANDLER(NAME, PARAMS, PARAM_FORMAT) \ |
| 581 | RC_HANDLER(void, NAME, PARAMS, ;, PARAM_FORMAT, ;, ;,\ |
| 582 | (xmlparseobject *)userData) |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 583 | |
Fred Drake | 6f98762 | 2000-08-25 18:03:30 +0000 | [diff] [blame] | 584 | #define INT_HANDLER(NAME, PARAMS, PARAM_FORMAT)\ |
| 585 | RC_HANDLER(int, NAME, PARAMS, int rc=0;, PARAM_FORMAT, \ |
Christian Heimes | 217cfd1 | 2007-12-02 14:31:20 +0000 | [diff] [blame] | 586 | rc = PyLong_AsLong(rv);, rc, \ |
Fred Drake | 6f98762 | 2000-08-25 18:03:30 +0000 | [diff] [blame] | 587 | (xmlparseobject *)userData) |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 588 | |
Fred Drake | 71b63ff | 2002-06-28 22:29:01 +0000 | [diff] [blame] | 589 | VOID_HANDLER(EndElement, |
| 590 | (void *userData, const XML_Char *name), |
Fred Drake | b91a36b | 2002-06-27 19:40:48 +0000 | [diff] [blame] | 591 | ("(N)", string_intern(self, name))) |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 592 | |
Fred Drake | 6f98762 | 2000-08-25 18:03:30 +0000 | [diff] [blame] | 593 | VOID_HANDLER(ProcessingInstruction, |
Fred Drake | 71b63ff | 2002-06-28 22:29:01 +0000 | [diff] [blame] | 594 | (void *userData, |
| 595 | const XML_Char *target, |
Fred Drake | 85d835f | 2001-02-08 15:39:08 +0000 | [diff] [blame] | 596 | const XML_Char *data), |
Guido van Rossum | 4ca9471 | 2007-07-23 17:42:32 +0000 | [diff] [blame] | 597 | ("(NO&)", string_intern(self, target), conv_string_to_unicode ,data)) |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 598 | |
Fred Drake | 6f98762 | 2000-08-25 18:03:30 +0000 | [diff] [blame] | 599 | VOID_HANDLER(UnparsedEntityDecl, |
Fred Drake | 71b63ff | 2002-06-28 22:29:01 +0000 | [diff] [blame] | 600 | (void *userData, |
Fred Drake | 85d835f | 2001-02-08 15:39:08 +0000 | [diff] [blame] | 601 | const XML_Char *entityName, |
| 602 | const XML_Char *base, |
| 603 | const XML_Char *systemId, |
| 604 | const XML_Char *publicId, |
| 605 | const XML_Char *notationName), |
Fred Drake | b91a36b | 2002-06-27 19:40:48 +0000 | [diff] [blame] | 606 | ("(NNNNN)", |
Fred Drake | 71b63ff | 2002-06-28 22:29:01 +0000 | [diff] [blame] | 607 | string_intern(self, entityName), string_intern(self, base), |
| 608 | string_intern(self, systemId), string_intern(self, publicId), |
Fred Drake | b91a36b | 2002-06-27 19:40:48 +0000 | [diff] [blame] | 609 | string_intern(self, notationName))) |
Fred Drake | 85d835f | 2001-02-08 15:39:08 +0000 | [diff] [blame] | 610 | |
Fred Drake | 85d835f | 2001-02-08 15:39:08 +0000 | [diff] [blame] | 611 | VOID_HANDLER(EntityDecl, |
| 612 | (void *userData, |
| 613 | const XML_Char *entityName, |
| 614 | int is_parameter_entity, |
| 615 | const XML_Char *value, |
| 616 | int value_length, |
| 617 | const XML_Char *base, |
| 618 | const XML_Char *systemId, |
| 619 | const XML_Char *publicId, |
| 620 | const XML_Char *notationName), |
Fred Drake | b91a36b | 2002-06-27 19:40:48 +0000 | [diff] [blame] | 621 | ("NiNNNNN", |
| 622 | string_intern(self, entityName), is_parameter_entity, |
Guido van Rossum | 4ca9471 | 2007-07-23 17:42:32 +0000 | [diff] [blame] | 623 | (conv_string_len_to_unicode(value, value_length)), |
Fred Drake | b91a36b | 2002-06-27 19:40:48 +0000 | [diff] [blame] | 624 | string_intern(self, base), string_intern(self, systemId), |
| 625 | string_intern(self, publicId), |
| 626 | string_intern(self, notationName))) |
Fred Drake | 85d835f | 2001-02-08 15:39:08 +0000 | [diff] [blame] | 627 | |
| 628 | VOID_HANDLER(XmlDecl, |
| 629 | (void *userData, |
| 630 | const XML_Char *version, |
| 631 | const XML_Char *encoding, |
| 632 | int standalone), |
| 633 | ("(O&O&i)", |
Guido van Rossum | 4ca9471 | 2007-07-23 17:42:32 +0000 | [diff] [blame] | 634 | conv_string_to_unicode ,version, conv_string_to_unicode ,encoding, |
Fred Drake | 85d835f | 2001-02-08 15:39:08 +0000 | [diff] [blame] | 635 | standalone)) |
| 636 | |
| 637 | static PyObject * |
| 638 | conv_content_model(XML_Content * const model, |
Fred Drake | b91a36b | 2002-06-27 19:40:48 +0000 | [diff] [blame] | 639 | PyObject *(*conv_string)(const XML_Char *)) |
Fred Drake | 85d835f | 2001-02-08 15:39:08 +0000 | [diff] [blame] | 640 | { |
| 641 | PyObject *result = NULL; |
| 642 | PyObject *children = PyTuple_New(model->numchildren); |
| 643 | int i; |
| 644 | |
| 645 | if (children != NULL) { |
Tim Peters | 9544fc5 | 2001-07-28 09:36:36 +0000 | [diff] [blame] | 646 | assert(model->numchildren < INT_MAX); |
| 647 | for (i = 0; i < (int)model->numchildren; ++i) { |
Fred Drake | 85d835f | 2001-02-08 15:39:08 +0000 | [diff] [blame] | 648 | PyObject *child = conv_content_model(&model->children[i], |
| 649 | conv_string); |
| 650 | if (child == NULL) { |
| 651 | Py_XDECREF(children); |
| 652 | return NULL; |
| 653 | } |
| 654 | PyTuple_SET_ITEM(children, i, child); |
| 655 | } |
| 656 | result = Py_BuildValue("(iiO&N)", |
| 657 | model->type, model->quant, |
| 658 | conv_string,model->name, children); |
| 659 | } |
| 660 | return result; |
| 661 | } |
| 662 | |
Fred Drake | 06dd8cf | 2003-02-02 03:54:17 +0000 | [diff] [blame] | 663 | static void |
| 664 | my_ElementDeclHandler(void *userData, |
| 665 | const XML_Char *name, |
| 666 | XML_Content *model) |
Fred Drake | 85d835f | 2001-02-08 15:39:08 +0000 | [diff] [blame] | 667 | { |
Fred Drake | 06dd8cf | 2003-02-02 03:54:17 +0000 | [diff] [blame] | 668 | xmlparseobject *self = (xmlparseobject *)userData; |
| 669 | PyObject *args = NULL; |
Fred Drake | 85d835f | 2001-02-08 15:39:08 +0000 | [diff] [blame] | 670 | |
Fred Drake | 06dd8cf | 2003-02-02 03:54:17 +0000 | [diff] [blame] | 671 | if (have_handler(self, ElementDecl)) { |
| 672 | PyObject *rv = NULL; |
| 673 | PyObject *modelobj, *nameobj; |
| 674 | |
| 675 | if (flush_character_buffer(self) < 0) |
| 676 | goto finally; |
Guido van Rossum | 4ca9471 | 2007-07-23 17:42:32 +0000 | [diff] [blame] | 677 | modelobj = conv_content_model(model, (conv_string_to_unicode)); |
Fred Drake | 06dd8cf | 2003-02-02 03:54:17 +0000 | [diff] [blame] | 678 | if (modelobj == NULL) { |
| 679 | flag_error(self); |
| 680 | goto finally; |
| 681 | } |
| 682 | nameobj = string_intern(self, name); |
| 683 | if (nameobj == NULL) { |
| 684 | Py_DECREF(modelobj); |
| 685 | flag_error(self); |
| 686 | goto finally; |
| 687 | } |
Michael W. Hudson | 0bb8454 | 2004-08-03 11:31:31 +0000 | [diff] [blame] | 688 | args = Py_BuildValue("NN", nameobj, modelobj); |
Fred Drake | 06dd8cf | 2003-02-02 03:54:17 +0000 | [diff] [blame] | 689 | if (args == NULL) { |
| 690 | Py_DECREF(modelobj); |
| 691 | flag_error(self); |
| 692 | goto finally; |
| 693 | } |
| 694 | self->in_callback = 1; |
| 695 | rv = call_with_frame(getcode(ElementDecl, "ElementDecl", __LINE__), |
Fred Drake | 39689c5 | 2004-08-13 03:12:57 +0000 | [diff] [blame] | 696 | self->handlers[ElementDecl], args, self); |
Fred Drake | 06dd8cf | 2003-02-02 03:54:17 +0000 | [diff] [blame] | 697 | self->in_callback = 0; |
| 698 | if (rv == NULL) { |
| 699 | flag_error(self); |
| 700 | goto finally; |
| 701 | } |
| 702 | Py_DECREF(rv); |
| 703 | } |
| 704 | finally: |
| 705 | Py_XDECREF(args); |
| 706 | XML_FreeContentModel(self->itself, model); |
| 707 | return; |
| 708 | } |
Fred Drake | 85d835f | 2001-02-08 15:39:08 +0000 | [diff] [blame] | 709 | |
| 710 | VOID_HANDLER(AttlistDecl, |
| 711 | (void *userData, |
| 712 | const XML_Char *elname, |
| 713 | const XML_Char *attname, |
| 714 | const XML_Char *att_type, |
| 715 | const XML_Char *dflt, |
| 716 | int isrequired), |
Fred Drake | b91a36b | 2002-06-27 19:40:48 +0000 | [diff] [blame] | 717 | ("(NNO&O&i)", |
| 718 | string_intern(self, elname), string_intern(self, attname), |
Guido van Rossum | 4ca9471 | 2007-07-23 17:42:32 +0000 | [diff] [blame] | 719 | conv_string_to_unicode ,att_type, conv_string_to_unicode ,dflt, |
Fred Drake | 85d835f | 2001-02-08 15:39:08 +0000 | [diff] [blame] | 720 | isrequired)) |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 721 | |
Martin v. Löwis | c847f40 | 2003-01-21 11:09:21 +0000 | [diff] [blame] | 722 | #if XML_COMBINED_VERSION >= 19504 |
Martin v. Löwis | 069dde2 | 2003-01-21 10:58:18 +0000 | [diff] [blame] | 723 | VOID_HANDLER(SkippedEntity, |
| 724 | (void *userData, |
| 725 | const XML_Char *entityName, |
| 726 | int is_parameter_entity), |
| 727 | ("Ni", |
| 728 | string_intern(self, entityName), is_parameter_entity)) |
Martin v. Löwis | c847f40 | 2003-01-21 11:09:21 +0000 | [diff] [blame] | 729 | #endif |
Martin v. Löwis | 069dde2 | 2003-01-21 10:58:18 +0000 | [diff] [blame] | 730 | |
Fred Drake | 71b63ff | 2002-06-28 22:29:01 +0000 | [diff] [blame] | 731 | VOID_HANDLER(NotationDecl, |
Andrew M. Kuchling | beba056 | 2000-06-27 00:33:30 +0000 | [diff] [blame] | 732 | (void *userData, |
| 733 | const XML_Char *notationName, |
| 734 | const XML_Char *base, |
| 735 | const XML_Char *systemId, |
| 736 | const XML_Char *publicId), |
Fred Drake | b91a36b | 2002-06-27 19:40:48 +0000 | [diff] [blame] | 737 | ("(NNNN)", |
Fred Drake | 71b63ff | 2002-06-28 22:29:01 +0000 | [diff] [blame] | 738 | string_intern(self, notationName), string_intern(self, base), |
Fred Drake | b91a36b | 2002-06-27 19:40:48 +0000 | [diff] [blame] | 739 | string_intern(self, systemId), string_intern(self, publicId))) |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 740 | |
Fred Drake | 6f98762 | 2000-08-25 18:03:30 +0000 | [diff] [blame] | 741 | VOID_HANDLER(StartNamespaceDecl, |
Andrew M. Kuchling | beba056 | 2000-06-27 00:33:30 +0000 | [diff] [blame] | 742 | (void *userData, |
| 743 | const XML_Char *prefix, |
| 744 | const XML_Char *uri), |
Fred Drake | b91a36b | 2002-06-27 19:40:48 +0000 | [diff] [blame] | 745 | ("(NN)", |
| 746 | string_intern(self, prefix), string_intern(self, uri))) |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 747 | |
Fred Drake | 6f98762 | 2000-08-25 18:03:30 +0000 | [diff] [blame] | 748 | VOID_HANDLER(EndNamespaceDecl, |
Andrew M. Kuchling | beba056 | 2000-06-27 00:33:30 +0000 | [diff] [blame] | 749 | (void *userData, |
| 750 | const XML_Char *prefix), |
Fred Drake | b91a36b | 2002-06-27 19:40:48 +0000 | [diff] [blame] | 751 | ("(N)", string_intern(self, prefix))) |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 752 | |
Fred Drake | 6f98762 | 2000-08-25 18:03:30 +0000 | [diff] [blame] | 753 | VOID_HANDLER(Comment, |
Fred Drake | b91a36b | 2002-06-27 19:40:48 +0000 | [diff] [blame] | 754 | (void *userData, const XML_Char *data), |
Guido van Rossum | 4ca9471 | 2007-07-23 17:42:32 +0000 | [diff] [blame] | 755 | ("(O&)", conv_string_to_unicode ,data)) |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 756 | |
Fred Drake | 6f98762 | 2000-08-25 18:03:30 +0000 | [diff] [blame] | 757 | VOID_HANDLER(StartCdataSection, |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 758 | (void *userData), |
Fred Drake | 6f98762 | 2000-08-25 18:03:30 +0000 | [diff] [blame] | 759 | ("()")) |
Fred Drake | 71b63ff | 2002-06-28 22:29:01 +0000 | [diff] [blame] | 760 | |
Fred Drake | 6f98762 | 2000-08-25 18:03:30 +0000 | [diff] [blame] | 761 | VOID_HANDLER(EndCdataSection, |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 762 | (void *userData), |
Fred Drake | 6f98762 | 2000-08-25 18:03:30 +0000 | [diff] [blame] | 763 | ("()")) |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 764 | |
Fred Drake | 6f98762 | 2000-08-25 18:03:30 +0000 | [diff] [blame] | 765 | VOID_HANDLER(Default, |
Fred Drake | 71b63ff | 2002-06-28 22:29:01 +0000 | [diff] [blame] | 766 | (void *userData, const XML_Char *s, int len), |
Guido van Rossum | 4ca9471 | 2007-07-23 17:42:32 +0000 | [diff] [blame] | 767 | ("(N)", (conv_string_len_to_unicode(s,len)))) |
Andrew M. Kuchling | beba056 | 2000-06-27 00:33:30 +0000 | [diff] [blame] | 768 | |
Fred Drake | 6f98762 | 2000-08-25 18:03:30 +0000 | [diff] [blame] | 769 | VOID_HANDLER(DefaultHandlerExpand, |
Fred Drake | 71b63ff | 2002-06-28 22:29:01 +0000 | [diff] [blame] | 770 | (void *userData, const XML_Char *s, int len), |
Guido van Rossum | 4ca9471 | 2007-07-23 17:42:32 +0000 | [diff] [blame] | 771 | ("(N)", (conv_string_len_to_unicode(s,len)))) |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 772 | |
Fred Drake | 71b63ff | 2002-06-28 22:29:01 +0000 | [diff] [blame] | 773 | INT_HANDLER(NotStandalone, |
| 774 | (void *userData), |
Andrew M. Kuchling | beba056 | 2000-06-27 00:33:30 +0000 | [diff] [blame] | 775 | ("()")) |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 776 | |
Fred Drake | 6f98762 | 2000-08-25 18:03:30 +0000 | [diff] [blame] | 777 | RC_HANDLER(int, ExternalEntityRef, |
Andrew M. Kuchling | beba056 | 2000-06-27 00:33:30 +0000 | [diff] [blame] | 778 | (XML_Parser parser, |
| 779 | const XML_Char *context, |
| 780 | const XML_Char *base, |
| 781 | const XML_Char *systemId, |
| 782 | const XML_Char *publicId), |
| 783 | int rc=0;, |
Fred Drake | b91a36b | 2002-06-27 19:40:48 +0000 | [diff] [blame] | 784 | ("(O&NNN)", |
Guido van Rossum | 4ca9471 | 2007-07-23 17:42:32 +0000 | [diff] [blame] | 785 | conv_string_to_unicode ,context, string_intern(self, base), |
Fred Drake | b91a36b | 2002-06-27 19:40:48 +0000 | [diff] [blame] | 786 | string_intern(self, systemId), string_intern(self, publicId)), |
Christian Heimes | 217cfd1 | 2007-12-02 14:31:20 +0000 | [diff] [blame] | 787 | rc = PyLong_AsLong(rv);, rc, |
Fred Drake | 6f98762 | 2000-08-25 18:03:30 +0000 | [diff] [blame] | 788 | XML_GetUserData(parser)) |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 789 | |
Martin v. Löwis | 0078f6c | 2001-01-21 10:18:10 +0000 | [diff] [blame] | 790 | /* XXX UnknownEncodingHandler */ |
| 791 | |
Fred Drake | 85d835f | 2001-02-08 15:39:08 +0000 | [diff] [blame] | 792 | VOID_HANDLER(StartDoctypeDecl, |
| 793 | (void *userData, const XML_Char *doctypeName, |
| 794 | const XML_Char *sysid, const XML_Char *pubid, |
| 795 | int has_internal_subset), |
Fred Drake | b91a36b | 2002-06-27 19:40:48 +0000 | [diff] [blame] | 796 | ("(NNNi)", string_intern(self, doctypeName), |
| 797 | string_intern(self, sysid), string_intern(self, pubid), |
Fred Drake | 85d835f | 2001-02-08 15:39:08 +0000 | [diff] [blame] | 798 | has_internal_subset)) |
Martin v. Löwis | 0078f6c | 2001-01-21 10:18:10 +0000 | [diff] [blame] | 799 | |
| 800 | VOID_HANDLER(EndDoctypeDecl, (void *userData), ("()")) |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 801 | |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 802 | /* ---------------------------------------------------------------- */ |
| 803 | |
Fred Drake | 71b63ff | 2002-06-28 22:29:01 +0000 | [diff] [blame] | 804 | static PyObject * |
| 805 | get_parse_result(xmlparseobject *self, int rv) |
| 806 | { |
| 807 | if (PyErr_Occurred()) { |
| 808 | return NULL; |
| 809 | } |
| 810 | if (rv == 0) { |
Martin v. Löwis | 069dde2 | 2003-01-21 10:58:18 +0000 | [diff] [blame] | 811 | return set_error(self, XML_GetErrorCode(self->itself)); |
Fred Drake | 71b63ff | 2002-06-28 22:29:01 +0000 | [diff] [blame] | 812 | } |
Fred Drake | 2a3d7db | 2002-06-28 22:56:48 +0000 | [diff] [blame] | 813 | if (flush_character_buffer(self) < 0) { |
| 814 | return NULL; |
| 815 | } |
Christian Heimes | 217cfd1 | 2007-12-02 14:31:20 +0000 | [diff] [blame] | 816 | return PyLong_FromLong(rv); |
Fred Drake | 71b63ff | 2002-06-28 22:29:01 +0000 | [diff] [blame] | 817 | } |
| 818 | |
Martin v. Löwis | 14f8b4c | 2002-06-13 20:33:02 +0000 | [diff] [blame] | 819 | PyDoc_STRVAR(xmlparse_Parse__doc__, |
Thomas Wouters | 3531730 | 2000-07-22 16:34:15 +0000 | [diff] [blame] | 820 | "Parse(data[, isfinal])\n\ |
Martin v. Löwis | 14f8b4c | 2002-06-13 20:33:02 +0000 | [diff] [blame] | 821 | Parse XML data. `isfinal' should be true at end of input."); |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 822 | |
| 823 | static PyObject * |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 824 | xmlparse_Parse(xmlparseobject *self, PyObject *args) |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 825 | { |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 826 | char *s; |
| 827 | int slen; |
| 828 | int isFinal = 0; |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 829 | |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 830 | if (!PyArg_ParseTuple(args, "s#|i:Parse", &s, &slen, &isFinal)) |
| 831 | return NULL; |
Fred Drake | 71b63ff | 2002-06-28 22:29:01 +0000 | [diff] [blame] | 832 | |
| 833 | return get_parse_result(self, XML_Parse(self->itself, s, slen, isFinal)); |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 834 | } |
| 835 | |
Fred Drake | ca1f426 | 2000-09-21 20:10:23 +0000 | [diff] [blame] | 836 | /* File reading copied from cPickle */ |
| 837 | |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 838 | #define BUF_SIZE 2048 |
| 839 | |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 840 | static int |
| 841 | readinst(char *buf, int buf_size, PyObject *meth) |
| 842 | { |
| 843 | PyObject *arg = NULL; |
| 844 | PyObject *bytes = NULL; |
| 845 | PyObject *str = NULL; |
| 846 | int len = -1; |
Guido van Rossum | 98297ee | 2007-11-06 21:34:58 +0000 | [diff] [blame] | 847 | char *ptr; |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 848 | |
Christian Heimes | 217cfd1 | 2007-12-02 14:31:20 +0000 | [diff] [blame] | 849 | if ((bytes = PyLong_FromLong(buf_size)) == NULL) |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 850 | goto finally; |
Fred Drake | 676940b | 2000-09-22 15:21:31 +0000 | [diff] [blame] | 851 | |
Fred Drake | 7b6caff | 2003-07-21 17:05:56 +0000 | [diff] [blame] | 852 | if ((arg = PyTuple_New(1)) == NULL) { |
| 853 | Py_DECREF(bytes); |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 854 | goto finally; |
Fred Drake | 7b6caff | 2003-07-21 17:05:56 +0000 | [diff] [blame] | 855 | } |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 856 | |
Tim Peters | 954eef7 | 2000-09-22 06:01:11 +0000 | [diff] [blame] | 857 | PyTuple_SET_ITEM(arg, 0, bytes); |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 858 | |
Martin v. Löwis | 9171f02 | 2004-10-13 19:50:11 +0000 | [diff] [blame] | 859 | #if PY_VERSION_HEX < 0x02020000 |
| 860 | str = PyObject_CallObject(meth, arg); |
| 861 | #else |
| 862 | str = PyObject_Call(meth, arg, NULL); |
| 863 | #endif |
| 864 | if (str == NULL) |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 865 | goto finally; |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 866 | |
Christian Heimes | 72b710a | 2008-05-26 13:28:38 +0000 | [diff] [blame] | 867 | if (PyBytes_Check(str)) |
| 868 | ptr = PyBytes_AS_STRING(str); |
Christian Heimes | 9c4756e | 2008-05-26 13:22:05 +0000 | [diff] [blame] | 869 | else if (PyByteArray_Check(str)) |
| 870 | ptr = PyByteArray_AS_STRING(str); |
Guido van Rossum | 98297ee | 2007-11-06 21:34:58 +0000 | [diff] [blame] | 871 | else { |
Fred Drake | 71b63ff | 2002-06-28 22:29:01 +0000 | [diff] [blame] | 872 | PyErr_Format(PyExc_TypeError, |
Guido van Rossum | 4ca9471 | 2007-07-23 17:42:32 +0000 | [diff] [blame] | 873 | "read() did not return a bytes object (type=%.400s)", |
Christian Heimes | 90aa764 | 2007-12-19 02:45:37 +0000 | [diff] [blame] | 874 | Py_TYPE(str)->tp_name); |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 875 | goto finally; |
| 876 | } |
Christian Heimes | 90aa764 | 2007-12-19 02:45:37 +0000 | [diff] [blame] | 877 | len = Py_SIZE(str); |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 878 | if (len > buf_size) { |
| 879 | PyErr_Format(PyExc_ValueError, |
| 880 | "read() returned too much data: " |
| 881 | "%i bytes requested, %i returned", |
| 882 | buf_size, len); |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 883 | goto finally; |
| 884 | } |
Guido van Rossum | 98297ee | 2007-11-06 21:34:58 +0000 | [diff] [blame] | 885 | memcpy(buf, ptr, len); |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 886 | finally: |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 887 | Py_XDECREF(arg); |
Fred Drake | ca1f426 | 2000-09-21 20:10:23 +0000 | [diff] [blame] | 888 | Py_XDECREF(str); |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 889 | return len; |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 890 | } |
| 891 | |
Martin v. Löwis | 14f8b4c | 2002-06-13 20:33:02 +0000 | [diff] [blame] | 892 | PyDoc_STRVAR(xmlparse_ParseFile__doc__, |
Thomas Wouters | 3531730 | 2000-07-22 16:34:15 +0000 | [diff] [blame] | 893 | "ParseFile(file)\n\ |
Martin v. Löwis | 14f8b4c | 2002-06-13 20:33:02 +0000 | [diff] [blame] | 894 | Parse XML data from file-like object."); |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 895 | |
| 896 | static PyObject * |
Thomas Wouters | 4d70c3d | 2006-06-08 14:42:34 +0000 | [diff] [blame] | 897 | xmlparse_ParseFile(xmlparseobject *self, PyObject *f) |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 898 | { |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 899 | int rv = 1; |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 900 | FILE *fp; |
| 901 | PyObject *readmethod = NULL; |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 902 | |
Guido van Rossum | da5b8f2 | 2007-06-12 23:30:11 +0000 | [diff] [blame] | 903 | { |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 904 | fp = NULL; |
Fred Drake | ca1f426 | 2000-09-21 20:10:23 +0000 | [diff] [blame] | 905 | readmethod = PyObject_GetAttrString(f, "read"); |
| 906 | if (readmethod == NULL) { |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 907 | PyErr_Clear(); |
Fred Drake | 71b63ff | 2002-06-28 22:29:01 +0000 | [diff] [blame] | 908 | PyErr_SetString(PyExc_TypeError, |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 909 | "argument must have 'read' attribute"); |
Fred Drake | 814f9fe | 2002-07-19 22:03:03 +0000 | [diff] [blame] | 910 | return NULL; |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 911 | } |
| 912 | } |
| 913 | for (;;) { |
| 914 | int bytes_read; |
| 915 | void *buf = XML_GetBuffer(self->itself, BUF_SIZE); |
Fred Drake | 7b6caff | 2003-07-21 17:05:56 +0000 | [diff] [blame] | 916 | if (buf == NULL) { |
Fred Drake | f239c6d | 2003-07-21 17:22:43 +0000 | [diff] [blame] | 917 | Py_XDECREF(readmethod); |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 918 | return PyErr_NoMemory(); |
Fred Drake | 7b6caff | 2003-07-21 17:05:56 +0000 | [diff] [blame] | 919 | } |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 920 | |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 921 | if (fp) { |
| 922 | bytes_read = fread(buf, sizeof(char), BUF_SIZE, fp); |
| 923 | if (bytes_read < 0) { |
| 924 | PyErr_SetFromErrno(PyExc_IOError); |
| 925 | return NULL; |
| 926 | } |
| 927 | } |
| 928 | else { |
| 929 | bytes_read = readinst(buf, BUF_SIZE, readmethod); |
Fred Drake | 7b6caff | 2003-07-21 17:05:56 +0000 | [diff] [blame] | 930 | if (bytes_read < 0) { |
| 931 | Py_DECREF(readmethod); |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 932 | return NULL; |
Fred Drake | 7b6caff | 2003-07-21 17:05:56 +0000 | [diff] [blame] | 933 | } |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 934 | } |
| 935 | rv = XML_ParseBuffer(self->itself, bytes_read, bytes_read == 0); |
Fred Drake | 7b6caff | 2003-07-21 17:05:56 +0000 | [diff] [blame] | 936 | if (PyErr_Occurred()) { |
| 937 | Py_XDECREF(readmethod); |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 938 | return NULL; |
Fred Drake | 7b6caff | 2003-07-21 17:05:56 +0000 | [diff] [blame] | 939 | } |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 940 | |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 941 | if (!rv || bytes_read == 0) |
| 942 | break; |
| 943 | } |
Fred Drake | 7b6caff | 2003-07-21 17:05:56 +0000 | [diff] [blame] | 944 | Py_XDECREF(readmethod); |
Fred Drake | 71b63ff | 2002-06-28 22:29:01 +0000 | [diff] [blame] | 945 | return get_parse_result(self, rv); |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 946 | } |
| 947 | |
Martin v. Löwis | 14f8b4c | 2002-06-13 20:33:02 +0000 | [diff] [blame] | 948 | PyDoc_STRVAR(xmlparse_SetBase__doc__, |
Thomas Wouters | 3531730 | 2000-07-22 16:34:15 +0000 | [diff] [blame] | 949 | "SetBase(base_url)\n\ |
Martin v. Löwis | 14f8b4c | 2002-06-13 20:33:02 +0000 | [diff] [blame] | 950 | Set the base URL for the parser."); |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 951 | |
| 952 | static PyObject * |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 953 | xmlparse_SetBase(xmlparseobject *self, PyObject *args) |
| 954 | { |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 955 | char *base; |
| 956 | |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 957 | if (!PyArg_ParseTuple(args, "s:SetBase", &base)) |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 958 | return NULL; |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 959 | if (!XML_SetBase(self->itself, base)) { |
| 960 | return PyErr_NoMemory(); |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 961 | } |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 962 | Py_INCREF(Py_None); |
| 963 | return Py_None; |
| 964 | } |
| 965 | |
Martin v. Löwis | 14f8b4c | 2002-06-13 20:33:02 +0000 | [diff] [blame] | 966 | PyDoc_STRVAR(xmlparse_GetBase__doc__, |
Thomas Wouters | 3531730 | 2000-07-22 16:34:15 +0000 | [diff] [blame] | 967 | "GetBase() -> url\n\ |
Martin v. Löwis | 14f8b4c | 2002-06-13 20:33:02 +0000 | [diff] [blame] | 968 | Return base URL string for the parser."); |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 969 | |
| 970 | static PyObject * |
Thomas Wouters | 4d70c3d | 2006-06-08 14:42:34 +0000 | [diff] [blame] | 971 | xmlparse_GetBase(xmlparseobject *self, PyObject *unused) |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 972 | { |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 973 | return Py_BuildValue("z", XML_GetBase(self->itself)); |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 974 | } |
| 975 | |
Martin v. Löwis | 14f8b4c | 2002-06-13 20:33:02 +0000 | [diff] [blame] | 976 | PyDoc_STRVAR(xmlparse_GetInputContext__doc__, |
Fred Drake | bd6101c | 2001-02-14 18:29:45 +0000 | [diff] [blame] | 977 | "GetInputContext() -> string\n\ |
| 978 | Return the untranslated text of the input that caused the current event.\n\ |
| 979 | If the event was generated by a large amount of text (such as a start tag\n\ |
Martin v. Löwis | 14f8b4c | 2002-06-13 20:33:02 +0000 | [diff] [blame] | 980 | for an element with many attributes), not all of the text may be available."); |
Fred Drake | bd6101c | 2001-02-14 18:29:45 +0000 | [diff] [blame] | 981 | |
| 982 | static PyObject * |
Thomas Wouters | 4d70c3d | 2006-06-08 14:42:34 +0000 | [diff] [blame] | 983 | xmlparse_GetInputContext(xmlparseobject *self, PyObject *unused) |
Fred Drake | bd6101c | 2001-02-14 18:29:45 +0000 | [diff] [blame] | 984 | { |
Thomas Wouters | 4d70c3d | 2006-06-08 14:42:34 +0000 | [diff] [blame] | 985 | if (self->in_callback) { |
| 986 | int offset, size; |
| 987 | const char *buffer |
| 988 | = XML_GetInputContext(self->itself, &offset, &size); |
Fred Drake | bd6101c | 2001-02-14 18:29:45 +0000 | [diff] [blame] | 989 | |
Thomas Wouters | 4d70c3d | 2006-06-08 14:42:34 +0000 | [diff] [blame] | 990 | if (buffer != NULL) |
Christian Heimes | 72b710a | 2008-05-26 13:28:38 +0000 | [diff] [blame] | 991 | return PyBytes_FromStringAndSize(buffer + offset, |
Thomas Wouters | 4d70c3d | 2006-06-08 14:42:34 +0000 | [diff] [blame] | 992 | size - offset); |
| 993 | else |
| 994 | Py_RETURN_NONE; |
Fred Drake | bd6101c | 2001-02-14 18:29:45 +0000 | [diff] [blame] | 995 | } |
Thomas Wouters | 4d70c3d | 2006-06-08 14:42:34 +0000 | [diff] [blame] | 996 | else |
| 997 | Py_RETURN_NONE; |
Fred Drake | bd6101c | 2001-02-14 18:29:45 +0000 | [diff] [blame] | 998 | } |
Fred Drake | bd6101c | 2001-02-14 18:29:45 +0000 | [diff] [blame] | 999 | |
Martin v. Löwis | 14f8b4c | 2002-06-13 20:33:02 +0000 | [diff] [blame] | 1000 | PyDoc_STRVAR(xmlparse_ExternalEntityParserCreate__doc__, |
Fred Drake | 2d4ac20 | 2001-01-03 15:36:25 +0000 | [diff] [blame] | 1001 | "ExternalEntityParserCreate(context[, encoding])\n\ |
Tim Peters | 51dc968 | 2000-09-24 22:12:45 +0000 | [diff] [blame] | 1002 | Create a parser for parsing an external entity based on the\n\ |
Martin v. Löwis | 14f8b4c | 2002-06-13 20:33:02 +0000 | [diff] [blame] | 1003 | information passed to the ExternalEntityRefHandler."); |
Lars Gustäbel | 4a30a07 | 2000-09-24 20:50:52 +0000 | [diff] [blame] | 1004 | |
| 1005 | static PyObject * |
| 1006 | xmlparse_ExternalEntityParserCreate(xmlparseobject *self, PyObject *args) |
| 1007 | { |
| 1008 | char *context; |
| 1009 | char *encoding = NULL; |
| 1010 | xmlparseobject *new_parser; |
| 1011 | int i; |
| 1012 | |
Martin v. Löwis | c57428d | 2001-09-19 09:55:09 +0000 | [diff] [blame] | 1013 | if (!PyArg_ParseTuple(args, "z|s:ExternalEntityParserCreate", |
Fred Drake | cde7913 | 2001-04-25 16:01:30 +0000 | [diff] [blame] | 1014 | &context, &encoding)) { |
| 1015 | return NULL; |
Lars Gustäbel | 4a30a07 | 2000-09-24 20:50:52 +0000 | [diff] [blame] | 1016 | } |
| 1017 | |
Martin v. Löwis | 894258c | 2001-09-23 10:20:10 +0000 | [diff] [blame] | 1018 | #ifndef Py_TPFLAGS_HAVE_GC |
Martin v. Löwis | b4fcf4d | 2002-06-30 06:40:55 +0000 | [diff] [blame] | 1019 | /* Python versions 2.0 and 2.1 */ |
Lars Gustäbel | 4a30a07 | 2000-09-24 20:50:52 +0000 | [diff] [blame] | 1020 | new_parser = PyObject_New(xmlparseobject, &Xmlparsetype); |
Martin v. Löwis | 894258c | 2001-09-23 10:20:10 +0000 | [diff] [blame] | 1021 | #else |
| 1022 | /* Python versions 2.2 and later */ |
| 1023 | new_parser = PyObject_GC_New(xmlparseobject, &Xmlparsetype); |
| 1024 | #endif |
Fred Drake | 85d835f | 2001-02-08 15:39:08 +0000 | [diff] [blame] | 1025 | |
| 1026 | if (new_parser == NULL) |
| 1027 | return NULL; |
Fred Drake | 2a3d7db | 2002-06-28 22:56:48 +0000 | [diff] [blame] | 1028 | new_parser->buffer_size = self->buffer_size; |
| 1029 | new_parser->buffer_used = 0; |
| 1030 | if (self->buffer != NULL) { |
| 1031 | new_parser->buffer = malloc(new_parser->buffer_size); |
| 1032 | if (new_parser->buffer == NULL) { |
Fred Drake | b28467b | 2002-07-02 15:44:36 +0000 | [diff] [blame] | 1033 | #ifndef Py_TPFLAGS_HAVE_GC |
| 1034 | /* Code for versions 2.0 and 2.1 */ |
| 1035 | PyObject_Del(new_parser); |
| 1036 | #else |
| 1037 | /* Code for versions 2.2 and later. */ |
Fred Drake | 2a3d7db | 2002-06-28 22:56:48 +0000 | [diff] [blame] | 1038 | PyObject_GC_Del(new_parser); |
Fred Drake | b28467b | 2002-07-02 15:44:36 +0000 | [diff] [blame] | 1039 | #endif |
Fred Drake | 2a3d7db | 2002-06-28 22:56:48 +0000 | [diff] [blame] | 1040 | return PyErr_NoMemory(); |
| 1041 | } |
| 1042 | } |
| 1043 | else |
| 1044 | new_parser->buffer = NULL; |
Fred Drake | 85d835f | 2001-02-08 15:39:08 +0000 | [diff] [blame] | 1045 | new_parser->ordered_attributes = self->ordered_attributes; |
| 1046 | new_parser->specified_attributes = self->specified_attributes; |
Fred Drake | bd6101c | 2001-02-14 18:29:45 +0000 | [diff] [blame] | 1047 | new_parser->in_callback = 0; |
Martin v. Löwis | 069dde2 | 2003-01-21 10:58:18 +0000 | [diff] [blame] | 1048 | new_parser->ns_prefixes = self->ns_prefixes; |
Lars Gustäbel | 4a30a07 | 2000-09-24 20:50:52 +0000 | [diff] [blame] | 1049 | new_parser->itself = XML_ExternalEntityParserCreate(self->itself, context, |
Martin v. Löwis | 0078f6c | 2001-01-21 10:18:10 +0000 | [diff] [blame] | 1050 | encoding); |
| 1051 | new_parser->handlers = 0; |
Fred Drake | b91a36b | 2002-06-27 19:40:48 +0000 | [diff] [blame] | 1052 | new_parser->intern = self->intern; |
| 1053 | Py_XINCREF(new_parser->intern); |
Martin v. Löwis | 894258c | 2001-09-23 10:20:10 +0000 | [diff] [blame] | 1054 | #ifdef Py_TPFLAGS_HAVE_GC |
| 1055 | PyObject_GC_Track(new_parser); |
| 1056 | #else |
Martin v. Löwis | 0078f6c | 2001-01-21 10:18:10 +0000 | [diff] [blame] | 1057 | PyObject_GC_Init(new_parser); |
Martin v. Löwis | 894258c | 2001-09-23 10:20:10 +0000 | [diff] [blame] | 1058 | #endif |
Martin v. Löwis | 0078f6c | 2001-01-21 10:18:10 +0000 | [diff] [blame] | 1059 | |
| 1060 | if (!new_parser->itself) { |
Fred Drake | 85d835f | 2001-02-08 15:39:08 +0000 | [diff] [blame] | 1061 | Py_DECREF(new_parser); |
| 1062 | return PyErr_NoMemory(); |
Lars Gustäbel | 4a30a07 | 2000-09-24 20:50:52 +0000 | [diff] [blame] | 1063 | } |
| 1064 | |
| 1065 | XML_SetUserData(new_parser->itself, (void *)new_parser); |
| 1066 | |
| 1067 | /* allocate and clear handlers first */ |
Fred Drake | 2a3d7db | 2002-06-28 22:56:48 +0000 | [diff] [blame] | 1068 | for (i = 0; handler_info[i].name != NULL; i++) |
Fred Drake | 85d835f | 2001-02-08 15:39:08 +0000 | [diff] [blame] | 1069 | /* do nothing */; |
Lars Gustäbel | 4a30a07 | 2000-09-24 20:50:52 +0000 | [diff] [blame] | 1070 | |
Fred Drake | 2a3d7db | 2002-06-28 22:56:48 +0000 | [diff] [blame] | 1071 | new_parser->handlers = malloc(sizeof(PyObject *) * i); |
Martin v. Löwis | 0078f6c | 2001-01-21 10:18:10 +0000 | [diff] [blame] | 1072 | if (!new_parser->handlers) { |
Fred Drake | 85d835f | 2001-02-08 15:39:08 +0000 | [diff] [blame] | 1073 | Py_DECREF(new_parser); |
| 1074 | return PyErr_NoMemory(); |
Martin v. Löwis | 0078f6c | 2001-01-21 10:18:10 +0000 | [diff] [blame] | 1075 | } |
Martin v. Löwis | 5b68ce3 | 2001-10-21 08:53:52 +0000 | [diff] [blame] | 1076 | clear_handlers(new_parser, 1); |
Lars Gustäbel | 4a30a07 | 2000-09-24 20:50:52 +0000 | [diff] [blame] | 1077 | |
| 1078 | /* then copy handlers from self */ |
| 1079 | for (i = 0; handler_info[i].name != NULL; i++) { |
Fred Drake | 71b63ff | 2002-06-28 22:29:01 +0000 | [diff] [blame] | 1080 | PyObject *handler = self->handlers[i]; |
| 1081 | if (handler != NULL) { |
| 1082 | Py_INCREF(handler); |
| 1083 | new_parser->handlers[i] = handler; |
| 1084 | handler_info[i].setter(new_parser->itself, |
Fred Drake | 85d835f | 2001-02-08 15:39:08 +0000 | [diff] [blame] | 1085 | handler_info[i].handler); |
| 1086 | } |
Lars Gustäbel | 4a30a07 | 2000-09-24 20:50:52 +0000 | [diff] [blame] | 1087 | } |
Fred Drake | 71b63ff | 2002-06-28 22:29:01 +0000 | [diff] [blame] | 1088 | return (PyObject *)new_parser; |
Lars Gustäbel | 4a30a07 | 2000-09-24 20:50:52 +0000 | [diff] [blame] | 1089 | } |
| 1090 | |
Martin v. Löwis | 14f8b4c | 2002-06-13 20:33:02 +0000 | [diff] [blame] | 1091 | PyDoc_STRVAR(xmlparse_SetParamEntityParsing__doc__, |
Martin v. Löwis | 0078f6c | 2001-01-21 10:18:10 +0000 | [diff] [blame] | 1092 | "SetParamEntityParsing(flag) -> success\n\ |
| 1093 | Controls parsing of parameter entities (including the external DTD\n\ |
| 1094 | subset). Possible flag values are XML_PARAM_ENTITY_PARSING_NEVER,\n\ |
| 1095 | XML_PARAM_ENTITY_PARSING_UNLESS_STANDALONE and\n\ |
| 1096 | XML_PARAM_ENTITY_PARSING_ALWAYS. Returns true if setting the flag\n\ |
Martin v. Löwis | 14f8b4c | 2002-06-13 20:33:02 +0000 | [diff] [blame] | 1097 | was successful."); |
Martin v. Löwis | 0078f6c | 2001-01-21 10:18:10 +0000 | [diff] [blame] | 1098 | |
| 1099 | static PyObject* |
Fred Drake | bd6101c | 2001-02-14 18:29:45 +0000 | [diff] [blame] | 1100 | xmlparse_SetParamEntityParsing(xmlparseobject *p, PyObject* args) |
Martin v. Löwis | 0078f6c | 2001-01-21 10:18:10 +0000 | [diff] [blame] | 1101 | { |
Fred Drake | 85d835f | 2001-02-08 15:39:08 +0000 | [diff] [blame] | 1102 | int flag; |
| 1103 | if (!PyArg_ParseTuple(args, "i", &flag)) |
| 1104 | return NULL; |
Fred Drake | bd6101c | 2001-02-14 18:29:45 +0000 | [diff] [blame] | 1105 | flag = XML_SetParamEntityParsing(p->itself, flag); |
Christian Heimes | 217cfd1 | 2007-12-02 14:31:20 +0000 | [diff] [blame] | 1106 | return PyLong_FromLong(flag); |
Martin v. Löwis | 0078f6c | 2001-01-21 10:18:10 +0000 | [diff] [blame] | 1107 | } |
| 1108 | |
Martin v. Löwis | c847f40 | 2003-01-21 11:09:21 +0000 | [diff] [blame] | 1109 | |
| 1110 | #if XML_COMBINED_VERSION >= 19505 |
Martin v. Löwis | 069dde2 | 2003-01-21 10:58:18 +0000 | [diff] [blame] | 1111 | PyDoc_STRVAR(xmlparse_UseForeignDTD__doc__, |
| 1112 | "UseForeignDTD([flag])\n\ |
| 1113 | Allows the application to provide an artificial external subset if one is\n\ |
| 1114 | not specified as part of the document instance. This readily allows the\n\ |
| 1115 | use of a 'default' document type controlled by the application, while still\n\ |
| 1116 | getting the advantage of providing document type information to the parser.\n\ |
| 1117 | 'flag' defaults to True if not provided."); |
| 1118 | |
| 1119 | static PyObject * |
| 1120 | xmlparse_UseForeignDTD(xmlparseobject *self, PyObject *args) |
| 1121 | { |
| 1122 | PyObject *flagobj = NULL; |
| 1123 | XML_Bool flag = XML_TRUE; |
| 1124 | enum XML_Error rc; |
Thomas Wouters | 4d70c3d | 2006-06-08 14:42:34 +0000 | [diff] [blame] | 1125 | if (!PyArg_UnpackTuple(args, "UseForeignDTD", 0, 1, &flagobj)) |
Martin v. Löwis | 069dde2 | 2003-01-21 10:58:18 +0000 | [diff] [blame] | 1126 | return NULL; |
| 1127 | if (flagobj != NULL) |
| 1128 | flag = PyObject_IsTrue(flagobj) ? XML_TRUE : XML_FALSE; |
| 1129 | rc = XML_UseForeignDTD(self->itself, flag); |
| 1130 | if (rc != XML_ERROR_NONE) { |
| 1131 | return set_error(self, rc); |
| 1132 | } |
| 1133 | Py_INCREF(Py_None); |
| 1134 | return Py_None; |
| 1135 | } |
Martin v. Löwis | c847f40 | 2003-01-21 11:09:21 +0000 | [diff] [blame] | 1136 | #endif |
Martin v. Löwis | 069dde2 | 2003-01-21 10:58:18 +0000 | [diff] [blame] | 1137 | |
Amaury Forgeot d'Arc | ba4105c | 2008-07-02 21:41:01 +0000 | [diff] [blame] | 1138 | static PyObject *xmlparse_dir(PyObject *self, PyObject* noargs); |
| 1139 | |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 1140 | static struct PyMethodDef xmlparse_methods[] = { |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 1141 | {"Parse", (PyCFunction)xmlparse_Parse, |
Fred Drake | bd6101c | 2001-02-14 18:29:45 +0000 | [diff] [blame] | 1142 | METH_VARARGS, xmlparse_Parse__doc__}, |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 1143 | {"ParseFile", (PyCFunction)xmlparse_ParseFile, |
Thomas Wouters | 4d70c3d | 2006-06-08 14:42:34 +0000 | [diff] [blame] | 1144 | METH_O, xmlparse_ParseFile__doc__}, |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 1145 | {"SetBase", (PyCFunction)xmlparse_SetBase, |
Martin v. Löwis | 069dde2 | 2003-01-21 10:58:18 +0000 | [diff] [blame] | 1146 | METH_VARARGS, xmlparse_SetBase__doc__}, |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 1147 | {"GetBase", (PyCFunction)xmlparse_GetBase, |
Thomas Wouters | 4d70c3d | 2006-06-08 14:42:34 +0000 | [diff] [blame] | 1148 | METH_NOARGS, xmlparse_GetBase__doc__}, |
Lars Gustäbel | 4a30a07 | 2000-09-24 20:50:52 +0000 | [diff] [blame] | 1149 | {"ExternalEntityParserCreate", (PyCFunction)xmlparse_ExternalEntityParserCreate, |
Martin v. Löwis | 069dde2 | 2003-01-21 10:58:18 +0000 | [diff] [blame] | 1150 | METH_VARARGS, xmlparse_ExternalEntityParserCreate__doc__}, |
Fred Drake | bd6101c | 2001-02-14 18:29:45 +0000 | [diff] [blame] | 1151 | {"SetParamEntityParsing", (PyCFunction)xmlparse_SetParamEntityParsing, |
| 1152 | METH_VARARGS, xmlparse_SetParamEntityParsing__doc__}, |
Fred Drake | bd6101c | 2001-02-14 18:29:45 +0000 | [diff] [blame] | 1153 | {"GetInputContext", (PyCFunction)xmlparse_GetInputContext, |
Thomas Wouters | 4d70c3d | 2006-06-08 14:42:34 +0000 | [diff] [blame] | 1154 | METH_NOARGS, xmlparse_GetInputContext__doc__}, |
Martin v. Löwis | c847f40 | 2003-01-21 11:09:21 +0000 | [diff] [blame] | 1155 | #if XML_COMBINED_VERSION >= 19505 |
Martin v. Löwis | 069dde2 | 2003-01-21 10:58:18 +0000 | [diff] [blame] | 1156 | {"UseForeignDTD", (PyCFunction)xmlparse_UseForeignDTD, |
| 1157 | METH_VARARGS, xmlparse_UseForeignDTD__doc__}, |
Martin v. Löwis | c847f40 | 2003-01-21 11:09:21 +0000 | [diff] [blame] | 1158 | #endif |
Amaury Forgeot d'Arc | ba4105c | 2008-07-02 21:41:01 +0000 | [diff] [blame] | 1159 | {"__dir__", xmlparse_dir, METH_NOARGS}, |
Martin v. Löwis | 069dde2 | 2003-01-21 10:58:18 +0000 | [diff] [blame] | 1160 | {NULL, NULL} /* sentinel */ |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 1161 | }; |
| 1162 | |
| 1163 | /* ---------- */ |
| 1164 | |
| 1165 | |
Martin v. Löwis | 0078f6c | 2001-01-21 10:18:10 +0000 | [diff] [blame] | 1166 | |
Fred Drake | 71b63ff | 2002-06-28 22:29:01 +0000 | [diff] [blame] | 1167 | /* pyexpat international encoding support. |
| 1168 | Make it as simple as possible. |
Martin v. Löwis | 0078f6c | 2001-01-21 10:18:10 +0000 | [diff] [blame] | 1169 | */ |
| 1170 | |
Martin v. Löwis | 3af7cc0 | 2001-01-22 08:19:10 +0000 | [diff] [blame] | 1171 | static char template_buffer[257]; |
Fred Drake | bb66a20 | 2001-03-01 20:48:17 +0000 | [diff] [blame] | 1172 | PyObject *template_string = NULL; |
Martin v. Löwis | 0078f6c | 2001-01-21 10:18:10 +0000 | [diff] [blame] | 1173 | |
Fred Drake | 71b63ff | 2002-06-28 22:29:01 +0000 | [diff] [blame] | 1174 | static void |
Martin v. Löwis | 0078f6c | 2001-01-21 10:18:10 +0000 | [diff] [blame] | 1175 | init_template_buffer(void) |
| 1176 | { |
| 1177 | int i; |
Fred Drake | bb66a20 | 2001-03-01 20:48:17 +0000 | [diff] [blame] | 1178 | for (i = 0; i < 256; i++) { |
| 1179 | template_buffer[i] = i; |
Tim Peters | 63cb99e | 2001-02-17 18:12:50 +0000 | [diff] [blame] | 1180 | } |
Fred Drake | bb66a20 | 2001-03-01 20:48:17 +0000 | [diff] [blame] | 1181 | template_buffer[256] = 0; |
Tim Peters | 63cb99e | 2001-02-17 18:12:50 +0000 | [diff] [blame] | 1182 | } |
Martin v. Löwis | 0078f6c | 2001-01-21 10:18:10 +0000 | [diff] [blame] | 1183 | |
Fred Drake | 71b63ff | 2002-06-28 22:29:01 +0000 | [diff] [blame] | 1184 | static int |
| 1185 | PyUnknownEncodingHandler(void *encodingHandlerData, |
| 1186 | const XML_Char *name, |
| 1187 | XML_Encoding *info) |
Martin v. Löwis | 0078f6c | 2001-01-21 10:18:10 +0000 | [diff] [blame] | 1188 | { |
Fred Drake | bb66a20 | 2001-03-01 20:48:17 +0000 | [diff] [blame] | 1189 | PyUnicodeObject *_u_string = NULL; |
| 1190 | int result = 0; |
Martin v. Löwis | 0078f6c | 2001-01-21 10:18:10 +0000 | [diff] [blame] | 1191 | int i; |
Fred Drake | 71b63ff | 2002-06-28 22:29:01 +0000 | [diff] [blame] | 1192 | |
Fred Drake | bb66a20 | 2001-03-01 20:48:17 +0000 | [diff] [blame] | 1193 | /* Yes, supports only 8bit encodings */ |
| 1194 | _u_string = (PyUnicodeObject *) |
| 1195 | PyUnicode_Decode(template_buffer, 256, name, "replace"); |
Fred Drake | 71b63ff | 2002-06-28 22:29:01 +0000 | [diff] [blame] | 1196 | |
Fred Drake | bb66a20 | 2001-03-01 20:48:17 +0000 | [diff] [blame] | 1197 | if (_u_string == NULL) |
Martin v. Löwis | 0078f6c | 2001-01-21 10:18:10 +0000 | [diff] [blame] | 1198 | return result; |
Fred Drake | 71b63ff | 2002-06-28 22:29:01 +0000 | [diff] [blame] | 1199 | |
Fred Drake | bb66a20 | 2001-03-01 20:48:17 +0000 | [diff] [blame] | 1200 | for (i = 0; i < 256; i++) { |
| 1201 | /* Stupid to access directly, but fast */ |
| 1202 | Py_UNICODE c = _u_string->str[i]; |
| 1203 | if (c == Py_UNICODE_REPLACEMENT_CHARACTER) |
Martin v. Löwis | 0078f6c | 2001-01-21 10:18:10 +0000 | [diff] [blame] | 1204 | info->map[i] = -1; |
Fred Drake | bb66a20 | 2001-03-01 20:48:17 +0000 | [diff] [blame] | 1205 | else |
Martin v. Löwis | 0078f6c | 2001-01-21 10:18:10 +0000 | [diff] [blame] | 1206 | info->map[i] = c; |
Tim Peters | 63cb99e | 2001-02-17 18:12:50 +0000 | [diff] [blame] | 1207 | } |
Martin v. Löwis | 0078f6c | 2001-01-21 10:18:10 +0000 | [diff] [blame] | 1208 | info->data = NULL; |
| 1209 | info->convert = NULL; |
| 1210 | info->release = NULL; |
Fred Drake | 71b63ff | 2002-06-28 22:29:01 +0000 | [diff] [blame] | 1211 | result = 1; |
Martin v. Löwis | 0078f6c | 2001-01-21 10:18:10 +0000 | [diff] [blame] | 1212 | Py_DECREF(_u_string); |
| 1213 | return result; |
| 1214 | } |
| 1215 | |
Martin v. Löwis | 0078f6c | 2001-01-21 10:18:10 +0000 | [diff] [blame] | 1216 | |
| 1217 | static PyObject * |
Fred Drake | b91a36b | 2002-06-27 19:40:48 +0000 | [diff] [blame] | 1218 | newxmlparseobject(char *encoding, char *namespace_separator, PyObject *intern) |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 1219 | { |
| 1220 | int i; |
| 1221 | xmlparseobject *self; |
Fred Drake | 71b63ff | 2002-06-28 22:29:01 +0000 | [diff] [blame] | 1222 | |
Martin v. Löwis | 894258c | 2001-09-23 10:20:10 +0000 | [diff] [blame] | 1223 | #ifdef Py_TPFLAGS_HAVE_GC |
| 1224 | /* Code for versions 2.2 and later */ |
| 1225 | self = PyObject_GC_New(xmlparseobject, &Xmlparsetype); |
| 1226 | #else |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 1227 | self = PyObject_New(xmlparseobject, &Xmlparsetype); |
Martin v. Löwis | 894258c | 2001-09-23 10:20:10 +0000 | [diff] [blame] | 1228 | #endif |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 1229 | if (self == NULL) |
| 1230 | return NULL; |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 1231 | |
Fred Drake | 2a3d7db | 2002-06-28 22:56:48 +0000 | [diff] [blame] | 1232 | self->buffer = NULL; |
| 1233 | self->buffer_size = CHARACTER_DATA_BUFFER_SIZE; |
| 1234 | self->buffer_used = 0; |
Fred Drake | 85d835f | 2001-02-08 15:39:08 +0000 | [diff] [blame] | 1235 | self->ordered_attributes = 0; |
| 1236 | self->specified_attributes = 0; |
Fred Drake | bd6101c | 2001-02-14 18:29:45 +0000 | [diff] [blame] | 1237 | self->in_callback = 0; |
Martin v. Löwis | 069dde2 | 2003-01-21 10:58:18 +0000 | [diff] [blame] | 1238 | self->ns_prefixes = 0; |
Martin v. Löwis | 0078f6c | 2001-01-21 10:18:10 +0000 | [diff] [blame] | 1239 | self->handlers = NULL; |
Fred Drake | cde7913 | 2001-04-25 16:01:30 +0000 | [diff] [blame] | 1240 | if (namespace_separator != NULL) { |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 1241 | self->itself = XML_ParserCreateNS(encoding, *namespace_separator); |
| 1242 | } |
Fred Drake | 85d835f | 2001-02-08 15:39:08 +0000 | [diff] [blame] | 1243 | else { |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 1244 | self->itself = XML_ParserCreate(encoding); |
| 1245 | } |
Fred Drake | b91a36b | 2002-06-27 19:40:48 +0000 | [diff] [blame] | 1246 | self->intern = intern; |
| 1247 | Py_XINCREF(self->intern); |
Martin v. Löwis | 894258c | 2001-09-23 10:20:10 +0000 | [diff] [blame] | 1248 | #ifdef Py_TPFLAGS_HAVE_GC |
| 1249 | PyObject_GC_Track(self); |
| 1250 | #else |
Martin v. Löwis | 0078f6c | 2001-01-21 10:18:10 +0000 | [diff] [blame] | 1251 | PyObject_GC_Init(self); |
Martin v. Löwis | 894258c | 2001-09-23 10:20:10 +0000 | [diff] [blame] | 1252 | #endif |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 1253 | if (self->itself == NULL) { |
Fred Drake | 71b63ff | 2002-06-28 22:29:01 +0000 | [diff] [blame] | 1254 | PyErr_SetString(PyExc_RuntimeError, |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 1255 | "XML_ParserCreate failed"); |
| 1256 | Py_DECREF(self); |
| 1257 | return NULL; |
| 1258 | } |
| 1259 | XML_SetUserData(self->itself, (void *)self); |
Fred Drake | 7c75bf2 | 2002-07-01 14:02:31 +0000 | [diff] [blame] | 1260 | XML_SetUnknownEncodingHandler(self->itself, |
| 1261 | (XML_UnknownEncodingHandler) PyUnknownEncodingHandler, NULL); |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 1262 | |
Fred Drake | 2a3d7db | 2002-06-28 22:56:48 +0000 | [diff] [blame] | 1263 | for (i = 0; handler_info[i].name != NULL; i++) |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 1264 | /* do nothing */; |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 1265 | |
Fred Drake | 7c75bf2 | 2002-07-01 14:02:31 +0000 | [diff] [blame] | 1266 | self->handlers = malloc(sizeof(PyObject *) * i); |
| 1267 | if (!self->handlers) { |
Fred Drake | 71b63ff | 2002-06-28 22:29:01 +0000 | [diff] [blame] | 1268 | Py_DECREF(self); |
| 1269 | return PyErr_NoMemory(); |
Martin v. Löwis | 0078f6c | 2001-01-21 10:18:10 +0000 | [diff] [blame] | 1270 | } |
Martin v. Löwis | 5b68ce3 | 2001-10-21 08:53:52 +0000 | [diff] [blame] | 1271 | clear_handlers(self, 1); |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 1272 | |
Martin v. Löwis | 0078f6c | 2001-01-21 10:18:10 +0000 | [diff] [blame] | 1273 | return (PyObject*)self; |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 1274 | } |
| 1275 | |
| 1276 | |
| 1277 | static void |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 1278 | xmlparse_dealloc(xmlparseobject *self) |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 1279 | { |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 1280 | int i; |
Martin v. Löwis | 894258c | 2001-09-23 10:20:10 +0000 | [diff] [blame] | 1281 | #ifdef Py_TPFLAGS_HAVE_GC |
| 1282 | PyObject_GC_UnTrack(self); |
| 1283 | #else |
Martin v. Löwis | 0078f6c | 2001-01-21 10:18:10 +0000 | [diff] [blame] | 1284 | PyObject_GC_Fini(self); |
Martin v. Löwis | 894258c | 2001-09-23 10:20:10 +0000 | [diff] [blame] | 1285 | #endif |
Fred Drake | 85d835f | 2001-02-08 15:39:08 +0000 | [diff] [blame] | 1286 | if (self->itself != NULL) |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 1287 | XML_ParserFree(self->itself); |
| 1288 | self->itself = NULL; |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 1289 | |
Fred Drake | 85d835f | 2001-02-08 15:39:08 +0000 | [diff] [blame] | 1290 | if (self->handlers != NULL) { |
Fred Drake | cde7913 | 2001-04-25 16:01:30 +0000 | [diff] [blame] | 1291 | PyObject *temp; |
Fred Drake | 85d835f | 2001-02-08 15:39:08 +0000 | [diff] [blame] | 1292 | for (i = 0; handler_info[i].name != NULL; i++) { |
Fred Drake | cde7913 | 2001-04-25 16:01:30 +0000 | [diff] [blame] | 1293 | temp = self->handlers[i]; |
| 1294 | self->handlers[i] = NULL; |
| 1295 | Py_XDECREF(temp); |
Fred Drake | 85d835f | 2001-02-08 15:39:08 +0000 | [diff] [blame] | 1296 | } |
| 1297 | free(self->handlers); |
Fred Drake | 71b63ff | 2002-06-28 22:29:01 +0000 | [diff] [blame] | 1298 | self->handlers = NULL; |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 1299 | } |
Fred Drake | 2a3d7db | 2002-06-28 22:56:48 +0000 | [diff] [blame] | 1300 | if (self->buffer != NULL) { |
| 1301 | free(self->buffer); |
| 1302 | self->buffer = NULL; |
| 1303 | } |
Fred Drake | b91a36b | 2002-06-27 19:40:48 +0000 | [diff] [blame] | 1304 | Py_XDECREF(self->intern); |
Martin v. Löwis | 894258c | 2001-09-23 10:20:10 +0000 | [diff] [blame] | 1305 | #ifndef Py_TPFLAGS_HAVE_GC |
Martin v. Löwis | b4fcf4d | 2002-06-30 06:40:55 +0000 | [diff] [blame] | 1306 | /* Code for versions 2.0 and 2.1 */ |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 1307 | PyObject_Del(self); |
Martin v. Löwis | 894258c | 2001-09-23 10:20:10 +0000 | [diff] [blame] | 1308 | #else |
| 1309 | /* Code for versions 2.2 and later. */ |
| 1310 | PyObject_GC_Del(self); |
| 1311 | #endif |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 1312 | } |
| 1313 | |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 1314 | static int |
| 1315 | handlername2int(const char *name) |
| 1316 | { |
| 1317 | int i; |
Fred Drake | 71b63ff | 2002-06-28 22:29:01 +0000 | [diff] [blame] | 1318 | for (i = 0; handler_info[i].name != NULL; i++) { |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 1319 | if (strcmp(name, handler_info[i].name) == 0) { |
| 1320 | return i; |
| 1321 | } |
| 1322 | } |
| 1323 | return -1; |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 1324 | } |
| 1325 | |
| 1326 | static PyObject * |
Fred Drake | 71b63ff | 2002-06-28 22:29:01 +0000 | [diff] [blame] | 1327 | get_pybool(int istrue) |
| 1328 | { |
| 1329 | PyObject *result = istrue ? Py_True : Py_False; |
| 1330 | Py_INCREF(result); |
| 1331 | return result; |
| 1332 | } |
| 1333 | |
| 1334 | static PyObject * |
Amaury Forgeot d'Arc | ba4105c | 2008-07-02 21:41:01 +0000 | [diff] [blame] | 1335 | xmlparse_getattro(xmlparseobject *self, PyObject *nameobj) |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 1336 | { |
Amaury Forgeot d'Arc | ba4105c | 2008-07-02 21:41:01 +0000 | [diff] [blame] | 1337 | char *name = ""; |
| 1338 | int handlernum = -1; |
| 1339 | |
| 1340 | if (PyUnicode_Check(nameobj)) |
Marc-André Lemburg | 4cc0f24 | 2008-08-07 18:54:33 +0000 | [diff] [blame] | 1341 | name = _PyUnicode_AsString(nameobj); |
Amaury Forgeot d'Arc | ba4105c | 2008-07-02 21:41:01 +0000 | [diff] [blame] | 1342 | |
| 1343 | handlernum = handlername2int(name); |
Fred Drake | 71b63ff | 2002-06-28 22:29:01 +0000 | [diff] [blame] | 1344 | |
| 1345 | if (handlernum != -1) { |
| 1346 | PyObject *result = self->handlers[handlernum]; |
| 1347 | if (result == NULL) |
| 1348 | result = Py_None; |
| 1349 | Py_INCREF(result); |
| 1350 | return result; |
| 1351 | } |
| 1352 | if (name[0] == 'E') { |
| 1353 | if (strcmp(name, "ErrorCode") == 0) |
Christian Heimes | 217cfd1 | 2007-12-02 14:31:20 +0000 | [diff] [blame] | 1354 | return PyLong_FromLong((long) |
Fred Drake | 71b63ff | 2002-06-28 22:29:01 +0000 | [diff] [blame] | 1355 | XML_GetErrorCode(self->itself)); |
| 1356 | if (strcmp(name, "ErrorLineNumber") == 0) |
Christian Heimes | 217cfd1 | 2007-12-02 14:31:20 +0000 | [diff] [blame] | 1357 | return PyLong_FromLong((long) |
Fred Drake | 71b63ff | 2002-06-28 22:29:01 +0000 | [diff] [blame] | 1358 | XML_GetErrorLineNumber(self->itself)); |
| 1359 | if (strcmp(name, "ErrorColumnNumber") == 0) |
Christian Heimes | 217cfd1 | 2007-12-02 14:31:20 +0000 | [diff] [blame] | 1360 | return PyLong_FromLong((long) |
Fred Drake | 71b63ff | 2002-06-28 22:29:01 +0000 | [diff] [blame] | 1361 | XML_GetErrorColumnNumber(self->itself)); |
| 1362 | if (strcmp(name, "ErrorByteIndex") == 0) |
Christian Heimes | 217cfd1 | 2007-12-02 14:31:20 +0000 | [diff] [blame] | 1363 | return PyLong_FromLong((long) |
Fred Drake | 71b63ff | 2002-06-28 22:29:01 +0000 | [diff] [blame] | 1364 | XML_GetErrorByteIndex(self->itself)); |
| 1365 | } |
Dave Cole | 3203efb | 2004-08-26 00:37:31 +0000 | [diff] [blame] | 1366 | if (name[0] == 'C') { |
| 1367 | if (strcmp(name, "CurrentLineNumber") == 0) |
Christian Heimes | 217cfd1 | 2007-12-02 14:31:20 +0000 | [diff] [blame] | 1368 | return PyLong_FromLong((long) |
Dave Cole | 3203efb | 2004-08-26 00:37:31 +0000 | [diff] [blame] | 1369 | XML_GetCurrentLineNumber(self->itself)); |
| 1370 | if (strcmp(name, "CurrentColumnNumber") == 0) |
Christian Heimes | 217cfd1 | 2007-12-02 14:31:20 +0000 | [diff] [blame] | 1371 | return PyLong_FromLong((long) |
Dave Cole | 3203efb | 2004-08-26 00:37:31 +0000 | [diff] [blame] | 1372 | XML_GetCurrentColumnNumber(self->itself)); |
| 1373 | if (strcmp(name, "CurrentByteIndex") == 0) |
Christian Heimes | 217cfd1 | 2007-12-02 14:31:20 +0000 | [diff] [blame] | 1374 | return PyLong_FromLong((long) |
Dave Cole | 3203efb | 2004-08-26 00:37:31 +0000 | [diff] [blame] | 1375 | XML_GetCurrentByteIndex(self->itself)); |
| 1376 | } |
Fred Drake | 2a3d7db | 2002-06-28 22:56:48 +0000 | [diff] [blame] | 1377 | if (name[0] == 'b') { |
| 1378 | if (strcmp(name, "buffer_size") == 0) |
Christian Heimes | 217cfd1 | 2007-12-02 14:31:20 +0000 | [diff] [blame] | 1379 | return PyLong_FromLong((long) self->buffer_size); |
Fred Drake | 2a3d7db | 2002-06-28 22:56:48 +0000 | [diff] [blame] | 1380 | if (strcmp(name, "buffer_text") == 0) |
| 1381 | return get_pybool(self->buffer != NULL); |
| 1382 | if (strcmp(name, "buffer_used") == 0) |
Christian Heimes | 217cfd1 | 2007-12-02 14:31:20 +0000 | [diff] [blame] | 1383 | return PyLong_FromLong((long) self->buffer_used); |
Fred Drake | 2a3d7db | 2002-06-28 22:56:48 +0000 | [diff] [blame] | 1384 | } |
Martin v. Löwis | 069dde2 | 2003-01-21 10:58:18 +0000 | [diff] [blame] | 1385 | if (strcmp(name, "namespace_prefixes") == 0) |
| 1386 | return get_pybool(self->ns_prefixes); |
Fred Drake | 85d835f | 2001-02-08 15:39:08 +0000 | [diff] [blame] | 1387 | if (strcmp(name, "ordered_attributes") == 0) |
Fred Drake | 71b63ff | 2002-06-28 22:29:01 +0000 | [diff] [blame] | 1388 | return get_pybool(self->ordered_attributes); |
Fred Drake | 85d835f | 2001-02-08 15:39:08 +0000 | [diff] [blame] | 1389 | if (strcmp(name, "specified_attributes") == 0) |
Fred Drake | 71b63ff | 2002-06-28 22:29:01 +0000 | [diff] [blame] | 1390 | return get_pybool((long) self->specified_attributes); |
Fred Drake | b91a36b | 2002-06-27 19:40:48 +0000 | [diff] [blame] | 1391 | if (strcmp(name, "intern") == 0) { |
| 1392 | if (self->intern == NULL) { |
| 1393 | Py_INCREF(Py_None); |
| 1394 | return Py_None; |
| 1395 | } |
| 1396 | else { |
| 1397 | Py_INCREF(self->intern); |
| 1398 | return self->intern; |
| 1399 | } |
| 1400 | } |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 1401 | |
Amaury Forgeot d'Arc | ba4105c | 2008-07-02 21:41:01 +0000 | [diff] [blame] | 1402 | return PyObject_GenericGetAttr((PyObject*)self, nameobj); |
Neal Norwitz | 8dfc4a9 | 2007-08-11 06:39:53 +0000 | [diff] [blame] | 1403 | } |
| 1404 | |
| 1405 | static PyObject * |
| 1406 | xmlparse_dir(PyObject *self, PyObject* noargs) |
| 1407 | { |
Neal Norwitz | fa56e2d | 2003-01-19 15:40:09 +0000 | [diff] [blame] | 1408 | #define APPEND(list, str) \ |
Martin v. Löwis | 069dde2 | 2003-01-21 10:58:18 +0000 | [diff] [blame] | 1409 | do { \ |
Neal Norwitz | 392c5be | 2007-08-25 17:20:32 +0000 | [diff] [blame] | 1410 | PyObject *o = PyUnicode_FromString(str); \ |
Martin v. Löwis | 069dde2 | 2003-01-21 10:58:18 +0000 | [diff] [blame] | 1411 | if (o != NULL) \ |
| 1412 | PyList_Append(list, o); \ |
| 1413 | Py_XDECREF(o); \ |
| 1414 | } while (0) |
Neal Norwitz | fa56e2d | 2003-01-19 15:40:09 +0000 | [diff] [blame] | 1415 | |
Neal Norwitz | 8dfc4a9 | 2007-08-11 06:39:53 +0000 | [diff] [blame] | 1416 | int i; |
| 1417 | PyObject *rc = PyList_New(0); |
| 1418 | if (!rc) |
| 1419 | return NULL; |
| 1420 | for (i = 0; handler_info[i].name != NULL; i++) { |
| 1421 | PyObject *o = get_handler_name(&handler_info[i]); |
| 1422 | if (o != NULL) |
| 1423 | PyList_Append(rc, o); |
| 1424 | Py_XDECREF(o); |
| 1425 | } |
| 1426 | APPEND(rc, "ErrorCode"); |
| 1427 | APPEND(rc, "ErrorLineNumber"); |
| 1428 | APPEND(rc, "ErrorColumnNumber"); |
| 1429 | APPEND(rc, "ErrorByteIndex"); |
| 1430 | APPEND(rc, "CurrentLineNumber"); |
| 1431 | APPEND(rc, "CurrentColumnNumber"); |
| 1432 | APPEND(rc, "CurrentByteIndex"); |
| 1433 | APPEND(rc, "buffer_size"); |
| 1434 | APPEND(rc, "buffer_text"); |
| 1435 | APPEND(rc, "buffer_used"); |
| 1436 | APPEND(rc, "namespace_prefixes"); |
| 1437 | APPEND(rc, "ordered_attributes"); |
| 1438 | APPEND(rc, "specified_attributes"); |
| 1439 | APPEND(rc, "intern"); |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 1440 | |
Neal Norwitz | fa56e2d | 2003-01-19 15:40:09 +0000 | [diff] [blame] | 1441 | #undef APPEND |
Neal Norwitz | 8dfc4a9 | 2007-08-11 06:39:53 +0000 | [diff] [blame] | 1442 | |
| 1443 | if (PyErr_Occurred()) { |
| 1444 | Py_DECREF(rc); |
| 1445 | rc = NULL; |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 1446 | } |
Neal Norwitz | 8dfc4a9 | 2007-08-11 06:39:53 +0000 | [diff] [blame] | 1447 | |
| 1448 | return rc; |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 1449 | } |
| 1450 | |
Fred Drake | 6f98762 | 2000-08-25 18:03:30 +0000 | [diff] [blame] | 1451 | static int |
| 1452 | sethandler(xmlparseobject *self, const char *name, PyObject* v) |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 1453 | { |
| 1454 | int handlernum = handlername2int(name); |
Fred Drake | 71b63ff | 2002-06-28 22:29:01 +0000 | [diff] [blame] | 1455 | if (handlernum >= 0) { |
| 1456 | xmlhandler c_handler = NULL; |
| 1457 | PyObject *temp = self->handlers[handlernum]; |
| 1458 | |
Thomas Wouters | 0e3f591 | 2006-08-11 14:57:12 +0000 | [diff] [blame] | 1459 | if (v == Py_None) { |
| 1460 | /* If this is the character data handler, and a character |
| 1461 | data handler is already active, we need to be more |
| 1462 | careful. What we can safely do is replace the existing |
| 1463 | character data handler callback function with a no-op |
| 1464 | function that will refuse to call Python. The downside |
| 1465 | is that this doesn't completely remove the character |
| 1466 | data handler from the C layer if there's any callback |
| 1467 | active, so Expat does a little more work than it |
| 1468 | otherwise would, but that's really an odd case. A more |
| 1469 | elaborate system of handlers and state could remove the |
| 1470 | C handler more effectively. */ |
| 1471 | if (handlernum == CharacterData && self->in_callback) |
| 1472 | c_handler = noop_character_data_handler; |
Fred Drake | 71b63ff | 2002-06-28 22:29:01 +0000 | [diff] [blame] | 1473 | v = NULL; |
Thomas Wouters | 0e3f591 | 2006-08-11 14:57:12 +0000 | [diff] [blame] | 1474 | } |
Fred Drake | 71b63ff | 2002-06-28 22:29:01 +0000 | [diff] [blame] | 1475 | else if (v != NULL) { |
| 1476 | Py_INCREF(v); |
| 1477 | c_handler = handler_info[handlernum].handler; |
| 1478 | } |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 1479 | self->handlers[handlernum] = v; |
Fred Drake | 71b63ff | 2002-06-28 22:29:01 +0000 | [diff] [blame] | 1480 | Py_XDECREF(temp); |
| 1481 | handler_info[handlernum].setter(self->itself, c_handler); |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 1482 | return 1; |
| 1483 | } |
| 1484 | return 0; |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 1485 | } |
| 1486 | |
| 1487 | static int |
Fred Drake | 6f98762 | 2000-08-25 18:03:30 +0000 | [diff] [blame] | 1488 | xmlparse_setattr(xmlparseobject *self, char *name, PyObject *v) |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 1489 | { |
Fred Drake | 6f98762 | 2000-08-25 18:03:30 +0000 | [diff] [blame] | 1490 | /* Set attribute 'name' to value 'v'. v==NULL means delete */ |
Fred Drake | 85d835f | 2001-02-08 15:39:08 +0000 | [diff] [blame] | 1491 | if (v == NULL) { |
Fred Drake | 6f98762 | 2000-08-25 18:03:30 +0000 | [diff] [blame] | 1492 | PyErr_SetString(PyExc_RuntimeError, "Cannot delete attribute"); |
| 1493 | return -1; |
| 1494 | } |
Fred Drake | 2a3d7db | 2002-06-28 22:56:48 +0000 | [diff] [blame] | 1495 | if (strcmp(name, "buffer_text") == 0) { |
| 1496 | if (PyObject_IsTrue(v)) { |
| 1497 | if (self->buffer == NULL) { |
| 1498 | self->buffer = malloc(self->buffer_size); |
| 1499 | if (self->buffer == NULL) { |
| 1500 | PyErr_NoMemory(); |
| 1501 | return -1; |
| 1502 | } |
| 1503 | self->buffer_used = 0; |
| 1504 | } |
| 1505 | } |
| 1506 | else if (self->buffer != NULL) { |
| 1507 | if (flush_character_buffer(self) < 0) |
| 1508 | return -1; |
| 1509 | free(self->buffer); |
| 1510 | self->buffer = NULL; |
| 1511 | } |
| 1512 | return 0; |
| 1513 | } |
Martin v. Löwis | 069dde2 | 2003-01-21 10:58:18 +0000 | [diff] [blame] | 1514 | if (strcmp(name, "namespace_prefixes") == 0) { |
| 1515 | if (PyObject_IsTrue(v)) |
| 1516 | self->ns_prefixes = 1; |
| 1517 | else |
| 1518 | self->ns_prefixes = 0; |
| 1519 | XML_SetReturnNSTriplet(self->itself, self->ns_prefixes); |
| 1520 | return 0; |
| 1521 | } |
Fred Drake | 85d835f | 2001-02-08 15:39:08 +0000 | [diff] [blame] | 1522 | if (strcmp(name, "ordered_attributes") == 0) { |
| 1523 | if (PyObject_IsTrue(v)) |
| 1524 | self->ordered_attributes = 1; |
| 1525 | else |
| 1526 | self->ordered_attributes = 0; |
| 1527 | return 0; |
| 1528 | } |
Fred Drake | 85d835f | 2001-02-08 15:39:08 +0000 | [diff] [blame] | 1529 | if (strcmp(name, "specified_attributes") == 0) { |
| 1530 | if (PyObject_IsTrue(v)) |
| 1531 | self->specified_attributes = 1; |
| 1532 | else |
| 1533 | self->specified_attributes = 0; |
Fred Drake | 6f98762 | 2000-08-25 18:03:30 +0000 | [diff] [blame] | 1534 | return 0; |
| 1535 | } |
Christian Heimes | 2380ac7 | 2008-01-09 00:17:24 +0000 | [diff] [blame] | 1536 | |
| 1537 | if (strcmp(name, "buffer_size") == 0) { |
| 1538 | long new_buffer_size; |
| 1539 | if (!PyLong_Check(v)) { |
| 1540 | PyErr_SetString(PyExc_TypeError, "buffer_size must be an integer"); |
| 1541 | return -1; |
| 1542 | } |
| 1543 | |
| 1544 | new_buffer_size=PyLong_AS_LONG(v); |
| 1545 | /* trivial case -- no change */ |
| 1546 | if (new_buffer_size == self->buffer_size) { |
| 1547 | return 0; |
| 1548 | } |
| 1549 | |
| 1550 | if (new_buffer_size <= 0) { |
| 1551 | PyErr_SetString(PyExc_ValueError, "buffer_size must be greater than zero"); |
| 1552 | return -1; |
| 1553 | } |
| 1554 | |
| 1555 | /* check maximum */ |
| 1556 | if (new_buffer_size > INT_MAX) { |
| 1557 | char errmsg[100]; |
| 1558 | sprintf(errmsg, "buffer_size must not be greater than %i", INT_MAX); |
| 1559 | PyErr_SetString(PyExc_ValueError, errmsg); |
| 1560 | return -1; |
| 1561 | } |
| 1562 | |
| 1563 | if (self->buffer != NULL) { |
| 1564 | /* there is already a buffer */ |
| 1565 | if (self->buffer_used != 0) { |
| 1566 | flush_character_buffer(self); |
| 1567 | } |
| 1568 | /* free existing buffer */ |
| 1569 | free(self->buffer); |
| 1570 | } |
| 1571 | self->buffer = malloc(new_buffer_size); |
| 1572 | if (self->buffer == NULL) { |
| 1573 | PyErr_NoMemory(); |
| 1574 | return -1; |
| 1575 | } |
| 1576 | self->buffer_size = new_buffer_size; |
| 1577 | return 0; |
| 1578 | } |
| 1579 | |
Fred Drake | 2a3d7db | 2002-06-28 22:56:48 +0000 | [diff] [blame] | 1580 | if (strcmp(name, "CharacterDataHandler") == 0) { |
| 1581 | /* If we're changing the character data handler, flush all |
| 1582 | * cached data with the old handler. Not sure there's a |
| 1583 | * "right" thing to do, though, but this probably won't |
| 1584 | * happen. |
| 1585 | */ |
| 1586 | if (flush_character_buffer(self) < 0) |
| 1587 | return -1; |
| 1588 | } |
Fred Drake | 6f98762 | 2000-08-25 18:03:30 +0000 | [diff] [blame] | 1589 | if (sethandler(self, name, v)) { |
| 1590 | return 0; |
| 1591 | } |
| 1592 | PyErr_SetString(PyExc_AttributeError, name); |
| 1593 | return -1; |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 1594 | } |
| 1595 | |
Martin v. Löwis | 0078f6c | 2001-01-21 10:18:10 +0000 | [diff] [blame] | 1596 | static int |
| 1597 | xmlparse_traverse(xmlparseobject *op, visitproc visit, void *arg) |
| 1598 | { |
Thomas Wouters | 49fd7fa | 2006-04-21 10:40:58 +0000 | [diff] [blame] | 1599 | int i; |
| 1600 | for (i = 0; handler_info[i].name != NULL; i++) |
| 1601 | Py_VISIT(op->handlers[i]); |
Fred Drake | cde7913 | 2001-04-25 16:01:30 +0000 | [diff] [blame] | 1602 | return 0; |
Martin v. Löwis | 0078f6c | 2001-01-21 10:18:10 +0000 | [diff] [blame] | 1603 | } |
| 1604 | |
| 1605 | static int |
| 1606 | xmlparse_clear(xmlparseobject *op) |
| 1607 | { |
Martin v. Löwis | 5b68ce3 | 2001-10-21 08:53:52 +0000 | [diff] [blame] | 1608 | clear_handlers(op, 0); |
Thomas Wouters | 49fd7fa | 2006-04-21 10:40:58 +0000 | [diff] [blame] | 1609 | Py_CLEAR(op->intern); |
Fred Drake | cde7913 | 2001-04-25 16:01:30 +0000 | [diff] [blame] | 1610 | return 0; |
Martin v. Löwis | 0078f6c | 2001-01-21 10:18:10 +0000 | [diff] [blame] | 1611 | } |
Martin v. Löwis | 0078f6c | 2001-01-21 10:18:10 +0000 | [diff] [blame] | 1612 | |
Martin v. Löwis | 14f8b4c | 2002-06-13 20:33:02 +0000 | [diff] [blame] | 1613 | PyDoc_STRVAR(Xmlparsetype__doc__, "XML parser"); |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 1614 | |
| 1615 | static PyTypeObject Xmlparsetype = { |
Martin v. Löwis | 9f2e346 | 2007-07-21 17:22:18 +0000 | [diff] [blame] | 1616 | PyVarObject_HEAD_INIT(NULL, 0) |
Guido van Rossum | 1464839 | 2001-12-08 18:02:58 +0000 | [diff] [blame] | 1617 | "pyexpat.xmlparser", /*tp_name*/ |
Martin v. Löwis | 0078f6c | 2001-01-21 10:18:10 +0000 | [diff] [blame] | 1618 | sizeof(xmlparseobject) + PyGC_HEAD_SIZE,/*tp_basicsize*/ |
Andrew M. Kuchling | beba056 | 2000-06-27 00:33:30 +0000 | [diff] [blame] | 1619 | 0, /*tp_itemsize*/ |
| 1620 | /* methods */ |
| 1621 | (destructor)xmlparse_dealloc, /*tp_dealloc*/ |
| 1622 | (printfunc)0, /*tp_print*/ |
Amaury Forgeot d'Arc | ba4105c | 2008-07-02 21:41:01 +0000 | [diff] [blame] | 1623 | 0, /*tp_getattr*/ |
Andrew M. Kuchling | beba056 | 2000-06-27 00:33:30 +0000 | [diff] [blame] | 1624 | (setattrfunc)xmlparse_setattr, /*tp_setattr*/ |
| 1625 | (cmpfunc)0, /*tp_compare*/ |
| 1626 | (reprfunc)0, /*tp_repr*/ |
| 1627 | 0, /*tp_as_number*/ |
| 1628 | 0, /*tp_as_sequence*/ |
| 1629 | 0, /*tp_as_mapping*/ |
| 1630 | (hashfunc)0, /*tp_hash*/ |
| 1631 | (ternaryfunc)0, /*tp_call*/ |
| 1632 | (reprfunc)0, /*tp_str*/ |
Amaury Forgeot d'Arc | ba4105c | 2008-07-02 21:41:01 +0000 | [diff] [blame] | 1633 | (getattrofunc)xmlparse_getattro, /* tp_getattro */ |
Martin v. Löwis | 0078f6c | 2001-01-21 10:18:10 +0000 | [diff] [blame] | 1634 | 0, /* tp_setattro */ |
| 1635 | 0, /* tp_as_buffer */ |
Martin v. Löwis | 894258c | 2001-09-23 10:20:10 +0000 | [diff] [blame] | 1636 | #ifdef Py_TPFLAGS_HAVE_GC |
Fred Drake | 71b63ff | 2002-06-28 22:29:01 +0000 | [diff] [blame] | 1637 | Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC, /*tp_flags*/ |
Martin v. Löwis | 894258c | 2001-09-23 10:20:10 +0000 | [diff] [blame] | 1638 | #else |
Fred Drake | 71b63ff | 2002-06-28 22:29:01 +0000 | [diff] [blame] | 1639 | Py_TPFLAGS_DEFAULT | Py_TPFLAGS_GC, /*tp_flags*/ |
Martin v. Löwis | 894258c | 2001-09-23 10:20:10 +0000 | [diff] [blame] | 1640 | #endif |
Martin v. Löwis | 069dde2 | 2003-01-21 10:58:18 +0000 | [diff] [blame] | 1641 | Xmlparsetype__doc__, /* tp_doc - Documentation string */ |
Martin v. Löwis | 0078f6c | 2001-01-21 10:18:10 +0000 | [diff] [blame] | 1642 | (traverseproc)xmlparse_traverse, /* tp_traverse */ |
Neal Norwitz | 8dfc4a9 | 2007-08-11 06:39:53 +0000 | [diff] [blame] | 1643 | (inquiry)xmlparse_clear, /* tp_clear */ |
| 1644 | 0, /* tp_richcompare */ |
| 1645 | 0, /* tp_weaklistoffset */ |
| 1646 | 0, /* tp_iter */ |
| 1647 | 0, /* tp_iternext */ |
Amaury Forgeot d'Arc | ba4105c | 2008-07-02 21:41:01 +0000 | [diff] [blame] | 1648 | xmlparse_methods, /* tp_methods */ |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 1649 | }; |
| 1650 | |
| 1651 | /* End of code for xmlparser objects */ |
| 1652 | /* -------------------------------------------------------- */ |
| 1653 | |
Martin v. Löwis | 14f8b4c | 2002-06-13 20:33:02 +0000 | [diff] [blame] | 1654 | PyDoc_STRVAR(pyexpat_ParserCreate__doc__, |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 1655 | "ParserCreate([encoding[, namespace_separator]]) -> parser\n\ |
Martin v. Löwis | 14f8b4c | 2002-06-13 20:33:02 +0000 | [diff] [blame] | 1656 | Return a new XML parser object."); |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 1657 | |
| 1658 | static PyObject * |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 1659 | pyexpat_ParserCreate(PyObject *notused, PyObject *args, PyObject *kw) |
| 1660 | { |
Fred Drake | cde7913 | 2001-04-25 16:01:30 +0000 | [diff] [blame] | 1661 | char *encoding = NULL; |
| 1662 | char *namespace_separator = NULL; |
Fred Drake | b91a36b | 2002-06-27 19:40:48 +0000 | [diff] [blame] | 1663 | PyObject *intern = NULL; |
| 1664 | PyObject *result; |
| 1665 | int intern_decref = 0; |
Martin v. Löwis | 15e6274 | 2006-02-27 16:46:16 +0000 | [diff] [blame] | 1666 | static char *kwlist[] = {"encoding", "namespace_separator", |
Jeremy Hylton | af68c87 | 2005-12-10 18:50:16 +0000 | [diff] [blame] | 1667 | "intern", NULL}; |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 1668 | |
Fred Drake | b91a36b | 2002-06-27 19:40:48 +0000 | [diff] [blame] | 1669 | if (!PyArg_ParseTupleAndKeywords(args, kw, "|zzO:ParserCreate", kwlist, |
| 1670 | &encoding, &namespace_separator, &intern)) |
Fred Drake | cde7913 | 2001-04-25 16:01:30 +0000 | [diff] [blame] | 1671 | return NULL; |
| 1672 | if (namespace_separator != NULL |
| 1673 | && strlen(namespace_separator) > 1) { |
| 1674 | PyErr_SetString(PyExc_ValueError, |
| 1675 | "namespace_separator must be at most one" |
| 1676 | " character, omitted, or None"); |
| 1677 | return NULL; |
| 1678 | } |
Fred Drake | b91a36b | 2002-06-27 19:40:48 +0000 | [diff] [blame] | 1679 | /* Explicitly passing None means no interning is desired. |
| 1680 | Not passing anything means that a new dictionary is used. */ |
| 1681 | if (intern == Py_None) |
| 1682 | intern = NULL; |
| 1683 | else if (intern == NULL) { |
| 1684 | intern = PyDict_New(); |
| 1685 | if (!intern) |
| 1686 | return NULL; |
| 1687 | intern_decref = 1; |
Fred Drake | 71b63ff | 2002-06-28 22:29:01 +0000 | [diff] [blame] | 1688 | } |
Fred Drake | b91a36b | 2002-06-27 19:40:48 +0000 | [diff] [blame] | 1689 | else if (!PyDict_Check(intern)) { |
| 1690 | PyErr_SetString(PyExc_TypeError, "intern must be a dictionary"); |
| 1691 | return NULL; |
| 1692 | } |
| 1693 | |
| 1694 | result = newxmlparseobject(encoding, namespace_separator, intern); |
| 1695 | if (intern_decref) { |
| 1696 | Py_DECREF(intern); |
| 1697 | } |
| 1698 | return result; |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 1699 | } |
| 1700 | |
Martin v. Löwis | 14f8b4c | 2002-06-13 20:33:02 +0000 | [diff] [blame] | 1701 | PyDoc_STRVAR(pyexpat_ErrorString__doc__, |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 1702 | "ErrorString(errno) -> string\n\ |
Martin v. Löwis | 14f8b4c | 2002-06-13 20:33:02 +0000 | [diff] [blame] | 1703 | Returns string error for given number."); |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 1704 | |
| 1705 | static PyObject * |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 1706 | pyexpat_ErrorString(PyObject *self, PyObject *args) |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 1707 | { |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 1708 | long code = 0; |
| 1709 | |
| 1710 | if (!PyArg_ParseTuple(args, "l:ErrorString", &code)) |
| 1711 | return NULL; |
| 1712 | return Py_BuildValue("z", XML_ErrorString((int)code)); |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 1713 | } |
| 1714 | |
| 1715 | /* List of methods defined in the module */ |
| 1716 | |
| 1717 | static struct PyMethodDef pyexpat_methods[] = { |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 1718 | {"ParserCreate", (PyCFunction)pyexpat_ParserCreate, |
| 1719 | METH_VARARGS|METH_KEYWORDS, pyexpat_ParserCreate__doc__}, |
| 1720 | {"ErrorString", (PyCFunction)pyexpat_ErrorString, |
| 1721 | METH_VARARGS, pyexpat_ErrorString__doc__}, |
Fred Drake | 71b63ff | 2002-06-28 22:29:01 +0000 | [diff] [blame] | 1722 | |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 1723 | {NULL, (PyCFunction)NULL, 0, NULL} /* sentinel */ |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 1724 | }; |
| 1725 | |
Andrew M. Kuchling | beba056 | 2000-06-27 00:33:30 +0000 | [diff] [blame] | 1726 | /* Module docstring */ |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 1727 | |
Martin v. Löwis | 14f8b4c | 2002-06-13 20:33:02 +0000 | [diff] [blame] | 1728 | PyDoc_STRVAR(pyexpat_module_documentation, |
| 1729 | "Python wrapper for Expat parser."); |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 1730 | |
Fred Drake | 4113b13 | 2001-03-24 19:58:26 +0000 | [diff] [blame] | 1731 | /* Return a Python string that represents the version number without the |
| 1732 | * extra cruft added by revision control, even if the right options were |
| 1733 | * given to the "cvs export" command to make it not include the extra |
| 1734 | * cruft. |
| 1735 | */ |
| 1736 | static PyObject * |
| 1737 | get_version_string(void) |
| 1738 | { |
| 1739 | static char *rcsid = "$Revision$"; |
| 1740 | char *rev = rcsid; |
| 1741 | int i = 0; |
| 1742 | |
Neal Norwitz | 30b5c5d | 2005-12-19 06:05:18 +0000 | [diff] [blame] | 1743 | while (!isdigit(Py_CHARMASK(*rev))) |
Fred Drake | 4113b13 | 2001-03-24 19:58:26 +0000 | [diff] [blame] | 1744 | ++rev; |
| 1745 | while (rev[i] != ' ' && rev[i] != '\0') |
| 1746 | ++i; |
| 1747 | |
Neal Norwitz | 392c5be | 2007-08-25 17:20:32 +0000 | [diff] [blame] | 1748 | return PyUnicode_FromStringAndSize(rev, i); |
Fred Drake | 4113b13 | 2001-03-24 19:58:26 +0000 | [diff] [blame] | 1749 | } |
| 1750 | |
Fred Drake | cde7913 | 2001-04-25 16:01:30 +0000 | [diff] [blame] | 1751 | /* Initialization function for the module */ |
| 1752 | |
| 1753 | #ifndef MODULE_NAME |
| 1754 | #define MODULE_NAME "pyexpat" |
| 1755 | #endif |
| 1756 | |
| 1757 | #ifndef MODULE_INITFUNC |
Martin v. Löwis | 1a21451 | 2008-06-11 05:26:20 +0000 | [diff] [blame] | 1758 | #define MODULE_INITFUNC PyInit_pyexpat |
Fred Drake | cde7913 | 2001-04-25 16:01:30 +0000 | [diff] [blame] | 1759 | #endif |
| 1760 | |
Martin v. Löwis | 069dde2 | 2003-01-21 10:58:18 +0000 | [diff] [blame] | 1761 | #ifndef PyMODINIT_FUNC |
| 1762 | # ifdef MS_WINDOWS |
| 1763 | # define PyMODINIT_FUNC __declspec(dllexport) void |
| 1764 | # else |
| 1765 | # define PyMODINIT_FUNC void |
| 1766 | # endif |
| 1767 | #endif |
| 1768 | |
Mark Hammond | 8235ea1 | 2002-07-19 06:55:41 +0000 | [diff] [blame] | 1769 | PyMODINIT_FUNC MODULE_INITFUNC(void); /* avoid compiler warnings */ |
Fred Drake | cde7913 | 2001-04-25 16:01:30 +0000 | [diff] [blame] | 1770 | |
Martin v. Löwis | 1a21451 | 2008-06-11 05:26:20 +0000 | [diff] [blame] | 1771 | static struct PyModuleDef pyexpatmodule = { |
| 1772 | PyModuleDef_HEAD_INIT, |
| 1773 | MODULE_NAME, |
| 1774 | pyexpat_module_documentation, |
| 1775 | -1, |
| 1776 | pyexpat_methods, |
| 1777 | NULL, |
| 1778 | NULL, |
| 1779 | NULL, |
| 1780 | NULL |
| 1781 | }; |
| 1782 | |
Martin v. Löwis | 069dde2 | 2003-01-21 10:58:18 +0000 | [diff] [blame] | 1783 | PyMODINIT_FUNC |
| 1784 | MODULE_INITFUNC(void) |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 1785 | { |
| 1786 | PyObject *m, *d; |
Neal Norwitz | 392c5be | 2007-08-25 17:20:32 +0000 | [diff] [blame] | 1787 | PyObject *errmod_name = PyUnicode_FromString(MODULE_NAME ".errors"); |
Fred Drake | 85d835f | 2001-02-08 15:39:08 +0000 | [diff] [blame] | 1788 | PyObject *errors_module; |
| 1789 | PyObject *modelmod_name; |
| 1790 | PyObject *model_module; |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 1791 | PyObject *sys_modules; |
Fredrik Lundh | d7a4288 | 2005-12-13 20:43:04 +0000 | [diff] [blame] | 1792 | static struct PyExpat_CAPI capi; |
| 1793 | PyObject* capi_object; |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 1794 | |
Fred Drake | 6f98762 | 2000-08-25 18:03:30 +0000 | [diff] [blame] | 1795 | if (errmod_name == NULL) |
Martin v. Löwis | 1a21451 | 2008-06-11 05:26:20 +0000 | [diff] [blame] | 1796 | return NULL; |
Neal Norwitz | 392c5be | 2007-08-25 17:20:32 +0000 | [diff] [blame] | 1797 | modelmod_name = PyUnicode_FromString(MODULE_NAME ".model"); |
Fred Drake | 85d835f | 2001-02-08 15:39:08 +0000 | [diff] [blame] | 1798 | if (modelmod_name == NULL) |
Martin v. Löwis | 1a21451 | 2008-06-11 05:26:20 +0000 | [diff] [blame] | 1799 | return NULL; |
Fred Drake | 6f98762 | 2000-08-25 18:03:30 +0000 | [diff] [blame] | 1800 | |
Amaury Forgeot d'Arc | ba4105c | 2008-07-02 21:41:01 +0000 | [diff] [blame] | 1801 | if (PyType_Ready(&Xmlparsetype) < 0) |
| 1802 | return NULL; |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 1803 | |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 1804 | /* Create the module and add the functions */ |
Martin v. Löwis | 1a21451 | 2008-06-11 05:26:20 +0000 | [diff] [blame] | 1805 | m = PyModule_Create(&pyexpatmodule); |
Neal Norwitz | 1ac754f | 2006-01-19 06:09:39 +0000 | [diff] [blame] | 1806 | if (m == NULL) |
Martin v. Löwis | 1a21451 | 2008-06-11 05:26:20 +0000 | [diff] [blame] | 1807 | return NULL; |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 1808 | |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 1809 | /* Add some symbolic constants to the module */ |
Fred Drake | bd6101c | 2001-02-14 18:29:45 +0000 | [diff] [blame] | 1810 | if (ErrorObject == NULL) { |
| 1811 | ErrorObject = PyErr_NewException("xml.parsers.expat.ExpatError", |
Fred Drake | 93adb69 | 2000-09-23 04:55:48 +0000 | [diff] [blame] | 1812 | NULL, NULL); |
Fred Drake | bd6101c | 2001-02-14 18:29:45 +0000 | [diff] [blame] | 1813 | if (ErrorObject == NULL) |
Martin v. Löwis | 1a21451 | 2008-06-11 05:26:20 +0000 | [diff] [blame] | 1814 | return NULL; |
Fred Drake | bd6101c | 2001-02-14 18:29:45 +0000 | [diff] [blame] | 1815 | } |
| 1816 | Py_INCREF(ErrorObject); |
Fred Drake | 93adb69 | 2000-09-23 04:55:48 +0000 | [diff] [blame] | 1817 | PyModule_AddObject(m, "error", ErrorObject); |
Fred Drake | bd6101c | 2001-02-14 18:29:45 +0000 | [diff] [blame] | 1818 | Py_INCREF(ErrorObject); |
| 1819 | PyModule_AddObject(m, "ExpatError", ErrorObject); |
Fred Drake | 4ba298c | 2000-10-29 04:57:53 +0000 | [diff] [blame] | 1820 | Py_INCREF(&Xmlparsetype); |
| 1821 | PyModule_AddObject(m, "XMLParserType", (PyObject *) &Xmlparsetype); |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 1822 | |
Fred Drake | 4113b13 | 2001-03-24 19:58:26 +0000 | [diff] [blame] | 1823 | PyModule_AddObject(m, "__version__", get_version_string()); |
Fred Drake | 738293d | 2000-12-21 17:25:07 +0000 | [diff] [blame] | 1824 | PyModule_AddStringConstant(m, "EXPAT_VERSION", |
| 1825 | (char *) XML_ExpatVersion()); |
Fred Drake | 85d835f | 2001-02-08 15:39:08 +0000 | [diff] [blame] | 1826 | { |
| 1827 | XML_Expat_Version info = XML_ExpatVersionInfo(); |
| 1828 | PyModule_AddObject(m, "version_info", |
| 1829 | Py_BuildValue("(iii)", info.major, |
| 1830 | info.minor, info.micro)); |
| 1831 | } |
Martin v. Löwis | 0078f6c | 2001-01-21 10:18:10 +0000 | [diff] [blame] | 1832 | init_template_buffer(); |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 1833 | /* XXX When Expat supports some way of figuring out how it was |
Fred Drake | 71b63ff | 2002-06-28 22:29:01 +0000 | [diff] [blame] | 1834 | compiled, this should check and set native_encoding |
| 1835 | appropriately. |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 1836 | */ |
Fred Drake | 93adb69 | 2000-09-23 04:55:48 +0000 | [diff] [blame] | 1837 | PyModule_AddStringConstant(m, "native_encoding", "UTF-8"); |
Fred Drake | c23b523 | 2000-08-24 21:57:43 +0000 | [diff] [blame] | 1838 | |
Fred Drake | 85d835f | 2001-02-08 15:39:08 +0000 | [diff] [blame] | 1839 | sys_modules = PySys_GetObject("modules"); |
Fred Drake | 93adb69 | 2000-09-23 04:55:48 +0000 | [diff] [blame] | 1840 | d = PyModule_GetDict(m); |
Fred Drake | 6f98762 | 2000-08-25 18:03:30 +0000 | [diff] [blame] | 1841 | errors_module = PyDict_GetItem(d, errmod_name); |
| 1842 | if (errors_module == NULL) { |
Fred Drake | cde7913 | 2001-04-25 16:01:30 +0000 | [diff] [blame] | 1843 | errors_module = PyModule_New(MODULE_NAME ".errors"); |
Fred Drake | 6f98762 | 2000-08-25 18:03:30 +0000 | [diff] [blame] | 1844 | if (errors_module != NULL) { |
Fred Drake | 6f98762 | 2000-08-25 18:03:30 +0000 | [diff] [blame] | 1845 | PyDict_SetItem(sys_modules, errmod_name, errors_module); |
Fred Drake | 93adb69 | 2000-09-23 04:55:48 +0000 | [diff] [blame] | 1846 | /* gives away the reference to errors_module */ |
| 1847 | PyModule_AddObject(m, "errors", errors_module); |
Fred Drake | c23b523 | 2000-08-24 21:57:43 +0000 | [diff] [blame] | 1848 | } |
| 1849 | } |
Fred Drake | 6f98762 | 2000-08-25 18:03:30 +0000 | [diff] [blame] | 1850 | Py_DECREF(errmod_name); |
Fred Drake | 85d835f | 2001-02-08 15:39:08 +0000 | [diff] [blame] | 1851 | model_module = PyDict_GetItem(d, modelmod_name); |
| 1852 | if (model_module == NULL) { |
Fred Drake | cde7913 | 2001-04-25 16:01:30 +0000 | [diff] [blame] | 1853 | model_module = PyModule_New(MODULE_NAME ".model"); |
Fred Drake | 85d835f | 2001-02-08 15:39:08 +0000 | [diff] [blame] | 1854 | if (model_module != NULL) { |
| 1855 | PyDict_SetItem(sys_modules, modelmod_name, model_module); |
| 1856 | /* gives away the reference to model_module */ |
| 1857 | PyModule_AddObject(m, "model", model_module); |
| 1858 | } |
| 1859 | } |
| 1860 | Py_DECREF(modelmod_name); |
| 1861 | if (errors_module == NULL || model_module == NULL) |
| 1862 | /* Don't core dump later! */ |
Martin v. Löwis | 1a21451 | 2008-06-11 05:26:20 +0000 | [diff] [blame] | 1863 | return NULL; |
Martin v. Löwis | 069dde2 | 2003-01-21 10:58:18 +0000 | [diff] [blame] | 1864 | |
Martin v. Löwis | c847f40 | 2003-01-21 11:09:21 +0000 | [diff] [blame] | 1865 | #if XML_COMBINED_VERSION > 19505 |
Martin v. Löwis | 069dde2 | 2003-01-21 10:58:18 +0000 | [diff] [blame] | 1866 | { |
| 1867 | const XML_Feature *features = XML_GetFeatureList(); |
| 1868 | PyObject *list = PyList_New(0); |
| 1869 | if (list == NULL) |
| 1870 | /* just ignore it */ |
| 1871 | PyErr_Clear(); |
| 1872 | else { |
| 1873 | int i = 0; |
| 1874 | for (; features[i].feature != XML_FEATURE_END; ++i) { |
| 1875 | int ok; |
| 1876 | PyObject *item = Py_BuildValue("si", features[i].name, |
| 1877 | features[i].value); |
| 1878 | if (item == NULL) { |
| 1879 | Py_DECREF(list); |
| 1880 | list = NULL; |
| 1881 | break; |
| 1882 | } |
| 1883 | ok = PyList_Append(list, item); |
| 1884 | Py_DECREF(item); |
| 1885 | if (ok < 0) { |
| 1886 | PyErr_Clear(); |
| 1887 | break; |
| 1888 | } |
| 1889 | } |
| 1890 | if (list != NULL) |
| 1891 | PyModule_AddObject(m, "features", list); |
| 1892 | } |
| 1893 | } |
Martin v. Löwis | c847f40 | 2003-01-21 11:09:21 +0000 | [diff] [blame] | 1894 | #endif |
Fred Drake | 6f98762 | 2000-08-25 18:03:30 +0000 | [diff] [blame] | 1895 | |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 1896 | #define MYCONST(name) \ |
Fred Drake | 93adb69 | 2000-09-23 04:55:48 +0000 | [diff] [blame] | 1897 | PyModule_AddStringConstant(errors_module, #name, \ |
| 1898 | (char*)XML_ErrorString(name)) |
Fred Drake | 7bd9f41 | 2000-07-04 23:51:31 +0000 | [diff] [blame] | 1899 | |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 1900 | MYCONST(XML_ERROR_NO_MEMORY); |
| 1901 | MYCONST(XML_ERROR_SYNTAX); |
| 1902 | MYCONST(XML_ERROR_NO_ELEMENTS); |
| 1903 | MYCONST(XML_ERROR_INVALID_TOKEN); |
| 1904 | MYCONST(XML_ERROR_UNCLOSED_TOKEN); |
| 1905 | MYCONST(XML_ERROR_PARTIAL_CHAR); |
| 1906 | MYCONST(XML_ERROR_TAG_MISMATCH); |
| 1907 | MYCONST(XML_ERROR_DUPLICATE_ATTRIBUTE); |
| 1908 | MYCONST(XML_ERROR_JUNK_AFTER_DOC_ELEMENT); |
| 1909 | MYCONST(XML_ERROR_PARAM_ENTITY_REF); |
| 1910 | MYCONST(XML_ERROR_UNDEFINED_ENTITY); |
| 1911 | MYCONST(XML_ERROR_RECURSIVE_ENTITY_REF); |
| 1912 | MYCONST(XML_ERROR_ASYNC_ENTITY); |
| 1913 | MYCONST(XML_ERROR_BAD_CHAR_REF); |
| 1914 | MYCONST(XML_ERROR_BINARY_ENTITY_REF); |
| 1915 | MYCONST(XML_ERROR_ATTRIBUTE_EXTERNAL_ENTITY_REF); |
| 1916 | MYCONST(XML_ERROR_MISPLACED_XML_PI); |
| 1917 | MYCONST(XML_ERROR_UNKNOWN_ENCODING); |
| 1918 | MYCONST(XML_ERROR_INCORRECT_ENCODING); |
Martin v. Löwis | 0078f6c | 2001-01-21 10:18:10 +0000 | [diff] [blame] | 1919 | MYCONST(XML_ERROR_UNCLOSED_CDATA_SECTION); |
| 1920 | MYCONST(XML_ERROR_EXTERNAL_ENTITY_HANDLING); |
| 1921 | MYCONST(XML_ERROR_NOT_STANDALONE); |
Fred Drake | 283b670 | 2004-08-04 22:28:16 +0000 | [diff] [blame] | 1922 | MYCONST(XML_ERROR_UNEXPECTED_STATE); |
| 1923 | MYCONST(XML_ERROR_ENTITY_DECLARED_IN_PE); |
| 1924 | MYCONST(XML_ERROR_FEATURE_REQUIRES_XML_DTD); |
| 1925 | MYCONST(XML_ERROR_CANT_CHANGE_FEATURE_ONCE_PARSING); |
| 1926 | /* Added in Expat 1.95.7. */ |
| 1927 | MYCONST(XML_ERROR_UNBOUND_PREFIX); |
| 1928 | /* Added in Expat 1.95.8. */ |
| 1929 | MYCONST(XML_ERROR_UNDECLARING_PREFIX); |
| 1930 | MYCONST(XML_ERROR_INCOMPLETE_PE); |
| 1931 | MYCONST(XML_ERROR_XML_DECL); |
| 1932 | MYCONST(XML_ERROR_TEXT_DECL); |
| 1933 | MYCONST(XML_ERROR_PUBLICID); |
| 1934 | MYCONST(XML_ERROR_SUSPENDED); |
| 1935 | MYCONST(XML_ERROR_NOT_SUSPENDED); |
| 1936 | MYCONST(XML_ERROR_ABORTED); |
| 1937 | MYCONST(XML_ERROR_FINISHED); |
| 1938 | MYCONST(XML_ERROR_SUSPEND_PE); |
Martin v. Löwis | 0078f6c | 2001-01-21 10:18:10 +0000 | [diff] [blame] | 1939 | |
Fred Drake | 85d835f | 2001-02-08 15:39:08 +0000 | [diff] [blame] | 1940 | PyModule_AddStringConstant(errors_module, "__doc__", |
| 1941 | "Constants used to describe error conditions."); |
| 1942 | |
Fred Drake | 93adb69 | 2000-09-23 04:55:48 +0000 | [diff] [blame] | 1943 | #undef MYCONST |
Martin v. Löwis | 0078f6c | 2001-01-21 10:18:10 +0000 | [diff] [blame] | 1944 | |
Fred Drake | 85d835f | 2001-02-08 15:39:08 +0000 | [diff] [blame] | 1945 | #define MYCONST(c) PyModule_AddIntConstant(m, #c, c) |
Martin v. Löwis | 0078f6c | 2001-01-21 10:18:10 +0000 | [diff] [blame] | 1946 | MYCONST(XML_PARAM_ENTITY_PARSING_NEVER); |
| 1947 | MYCONST(XML_PARAM_ENTITY_PARSING_UNLESS_STANDALONE); |
| 1948 | MYCONST(XML_PARAM_ENTITY_PARSING_ALWAYS); |
Fred Drake | 85d835f | 2001-02-08 15:39:08 +0000 | [diff] [blame] | 1949 | #undef MYCONST |
Martin v. Löwis | 0078f6c | 2001-01-21 10:18:10 +0000 | [diff] [blame] | 1950 | |
Fred Drake | 85d835f | 2001-02-08 15:39:08 +0000 | [diff] [blame] | 1951 | #define MYCONST(c) PyModule_AddIntConstant(model_module, #c, c) |
| 1952 | PyModule_AddStringConstant(model_module, "__doc__", |
| 1953 | "Constants used to interpret content model information."); |
Martin v. Löwis | 0078f6c | 2001-01-21 10:18:10 +0000 | [diff] [blame] | 1954 | |
Fred Drake | 85d835f | 2001-02-08 15:39:08 +0000 | [diff] [blame] | 1955 | MYCONST(XML_CTYPE_EMPTY); |
| 1956 | MYCONST(XML_CTYPE_ANY); |
| 1957 | MYCONST(XML_CTYPE_MIXED); |
| 1958 | MYCONST(XML_CTYPE_NAME); |
| 1959 | MYCONST(XML_CTYPE_CHOICE); |
| 1960 | MYCONST(XML_CTYPE_SEQ); |
| 1961 | |
| 1962 | MYCONST(XML_CQUANT_NONE); |
| 1963 | MYCONST(XML_CQUANT_OPT); |
| 1964 | MYCONST(XML_CQUANT_REP); |
| 1965 | MYCONST(XML_CQUANT_PLUS); |
| 1966 | #undef MYCONST |
Fredrik Lundh | c334504 | 2005-12-13 19:49:55 +0000 | [diff] [blame] | 1967 | |
| 1968 | /* initialize pyexpat dispatch table */ |
Fredrik Lundh | d7a4288 | 2005-12-13 20:43:04 +0000 | [diff] [blame] | 1969 | capi.size = sizeof(capi); |
Fredrik Lundh | cc117db | 2005-12-13 21:55:36 +0000 | [diff] [blame] | 1970 | capi.magic = PyExpat_CAPI_MAGIC; |
Fredrik Lundh | d7a4288 | 2005-12-13 20:43:04 +0000 | [diff] [blame] | 1971 | capi.MAJOR_VERSION = XML_MAJOR_VERSION; |
| 1972 | capi.MINOR_VERSION = XML_MINOR_VERSION; |
| 1973 | capi.MICRO_VERSION = XML_MICRO_VERSION; |
| 1974 | capi.ErrorString = XML_ErrorString; |
Fredrik Lundh | cc117db | 2005-12-13 21:55:36 +0000 | [diff] [blame] | 1975 | capi.GetErrorCode = XML_GetErrorCode; |
| 1976 | capi.GetErrorColumnNumber = XML_GetErrorColumnNumber; |
| 1977 | capi.GetErrorLineNumber = XML_GetErrorLineNumber; |
Fredrik Lundh | d7a4288 | 2005-12-13 20:43:04 +0000 | [diff] [blame] | 1978 | capi.Parse = XML_Parse; |
| 1979 | capi.ParserCreate_MM = XML_ParserCreate_MM; |
| 1980 | capi.ParserFree = XML_ParserFree; |
| 1981 | capi.SetCharacterDataHandler = XML_SetCharacterDataHandler; |
| 1982 | capi.SetCommentHandler = XML_SetCommentHandler; |
| 1983 | capi.SetDefaultHandlerExpand = XML_SetDefaultHandlerExpand; |
| 1984 | capi.SetElementHandler = XML_SetElementHandler; |
| 1985 | capi.SetNamespaceDeclHandler = XML_SetNamespaceDeclHandler; |
| 1986 | capi.SetProcessingInstructionHandler = XML_SetProcessingInstructionHandler; |
| 1987 | capi.SetUnknownEncodingHandler = XML_SetUnknownEncodingHandler; |
| 1988 | capi.SetUserData = XML_SetUserData; |
Fredrik Lundh | c334504 | 2005-12-13 19:49:55 +0000 | [diff] [blame] | 1989 | |
| 1990 | /* export as cobject */ |
Fredrik Lundh | cc117db | 2005-12-13 21:55:36 +0000 | [diff] [blame] | 1991 | capi_object = PyCObject_FromVoidPtr(&capi, NULL); |
Fredrik Lundh | d7a4288 | 2005-12-13 20:43:04 +0000 | [diff] [blame] | 1992 | if (capi_object) |
| 1993 | PyModule_AddObject(m, "expat_CAPI", capi_object); |
Martin v. Löwis | 1a21451 | 2008-06-11 05:26:20 +0000 | [diff] [blame] | 1994 | return m; |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 1995 | } |
| 1996 | |
Fred Drake | 6f98762 | 2000-08-25 18:03:30 +0000 | [diff] [blame] | 1997 | static void |
Martin v. Löwis | 5b68ce3 | 2001-10-21 08:53:52 +0000 | [diff] [blame] | 1998 | clear_handlers(xmlparseobject *self, int initial) |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 1999 | { |
Fred Drake | cde7913 | 2001-04-25 16:01:30 +0000 | [diff] [blame] | 2000 | int i = 0; |
| 2001 | PyObject *temp; |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 2002 | |
Fred Drake | 71b63ff | 2002-06-28 22:29:01 +0000 | [diff] [blame] | 2003 | for (; handler_info[i].name != NULL; i++) { |
Martin v. Löwis | 5b68ce3 | 2001-10-21 08:53:52 +0000 | [diff] [blame] | 2004 | if (initial) |
Fred Drake | 71b63ff | 2002-06-28 22:29:01 +0000 | [diff] [blame] | 2005 | self->handlers[i] = NULL; |
Martin v. Löwis | 5b68ce3 | 2001-10-21 08:53:52 +0000 | [diff] [blame] | 2006 | else { |
Fred Drake | cde7913 | 2001-04-25 16:01:30 +0000 | [diff] [blame] | 2007 | temp = self->handlers[i]; |
| 2008 | self->handlers[i] = NULL; |
| 2009 | Py_XDECREF(temp); |
Martin v. Löwis | 5b68ce3 | 2001-10-21 08:53:52 +0000 | [diff] [blame] | 2010 | handler_info[i].setter(self->itself, NULL); |
Fred Drake | cde7913 | 2001-04-25 16:01:30 +0000 | [diff] [blame] | 2011 | } |
Fred Drake | cde7913 | 2001-04-25 16:01:30 +0000 | [diff] [blame] | 2012 | } |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 2013 | } |
| 2014 | |
Tim Peters | 0c32279 | 2002-07-17 16:49:03 +0000 | [diff] [blame] | 2015 | static struct HandlerInfo handler_info[] = { |
Fred Drake | 71b63ff | 2002-06-28 22:29:01 +0000 | [diff] [blame] | 2016 | {"StartElementHandler", |
| 2017 | (xmlhandlersetter)XML_SetStartElementHandler, |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 2018 | (xmlhandler)my_StartElementHandler}, |
Fred Drake | 71b63ff | 2002-06-28 22:29:01 +0000 | [diff] [blame] | 2019 | {"EndElementHandler", |
| 2020 | (xmlhandlersetter)XML_SetEndElementHandler, |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 2021 | (xmlhandler)my_EndElementHandler}, |
Fred Drake | 71b63ff | 2002-06-28 22:29:01 +0000 | [diff] [blame] | 2022 | {"ProcessingInstructionHandler", |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 2023 | (xmlhandlersetter)XML_SetProcessingInstructionHandler, |
| 2024 | (xmlhandler)my_ProcessingInstructionHandler}, |
Fred Drake | 71b63ff | 2002-06-28 22:29:01 +0000 | [diff] [blame] | 2025 | {"CharacterDataHandler", |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 2026 | (xmlhandlersetter)XML_SetCharacterDataHandler, |
| 2027 | (xmlhandler)my_CharacterDataHandler}, |
Fred Drake | 71b63ff | 2002-06-28 22:29:01 +0000 | [diff] [blame] | 2028 | {"UnparsedEntityDeclHandler", |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 2029 | (xmlhandlersetter)XML_SetUnparsedEntityDeclHandler, |
Fred Drake | 2a3d7db | 2002-06-28 22:56:48 +0000 | [diff] [blame] | 2030 | (xmlhandler)my_UnparsedEntityDeclHandler}, |
Fred Drake | 71b63ff | 2002-06-28 22:29:01 +0000 | [diff] [blame] | 2031 | {"NotationDeclHandler", |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 2032 | (xmlhandlersetter)XML_SetNotationDeclHandler, |
Fred Drake | 2a3d7db | 2002-06-28 22:56:48 +0000 | [diff] [blame] | 2033 | (xmlhandler)my_NotationDeclHandler}, |
Fred Drake | 71b63ff | 2002-06-28 22:29:01 +0000 | [diff] [blame] | 2034 | {"StartNamespaceDeclHandler", |
| 2035 | (xmlhandlersetter)XML_SetStartNamespaceDeclHandler, |
Fred Drake | 2a3d7db | 2002-06-28 22:56:48 +0000 | [diff] [blame] | 2036 | (xmlhandler)my_StartNamespaceDeclHandler}, |
Fred Drake | 71b63ff | 2002-06-28 22:29:01 +0000 | [diff] [blame] | 2037 | {"EndNamespaceDeclHandler", |
| 2038 | (xmlhandlersetter)XML_SetEndNamespaceDeclHandler, |
Fred Drake | 2a3d7db | 2002-06-28 22:56:48 +0000 | [diff] [blame] | 2039 | (xmlhandler)my_EndNamespaceDeclHandler}, |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 2040 | {"CommentHandler", |
| 2041 | (xmlhandlersetter)XML_SetCommentHandler, |
| 2042 | (xmlhandler)my_CommentHandler}, |
| 2043 | {"StartCdataSectionHandler", |
Fred Drake | 71b63ff | 2002-06-28 22:29:01 +0000 | [diff] [blame] | 2044 | (xmlhandlersetter)XML_SetStartCdataSectionHandler, |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 2045 | (xmlhandler)my_StartCdataSectionHandler}, |
| 2046 | {"EndCdataSectionHandler", |
Fred Drake | 71b63ff | 2002-06-28 22:29:01 +0000 | [diff] [blame] | 2047 | (xmlhandlersetter)XML_SetEndCdataSectionHandler, |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 2048 | (xmlhandler)my_EndCdataSectionHandler}, |
| 2049 | {"DefaultHandler", |
| 2050 | (xmlhandlersetter)XML_SetDefaultHandler, |
| 2051 | (xmlhandler)my_DefaultHandler}, |
| 2052 | {"DefaultHandlerExpand", |
| 2053 | (xmlhandlersetter)XML_SetDefaultHandlerExpand, |
| 2054 | (xmlhandler)my_DefaultHandlerExpandHandler}, |
| 2055 | {"NotStandaloneHandler", |
| 2056 | (xmlhandlersetter)XML_SetNotStandaloneHandler, |
| 2057 | (xmlhandler)my_NotStandaloneHandler}, |
| 2058 | {"ExternalEntityRefHandler", |
| 2059 | (xmlhandlersetter)XML_SetExternalEntityRefHandler, |
Fred Drake | 2a3d7db | 2002-06-28 22:56:48 +0000 | [diff] [blame] | 2060 | (xmlhandler)my_ExternalEntityRefHandler}, |
Martin v. Löwis | 0078f6c | 2001-01-21 10:18:10 +0000 | [diff] [blame] | 2061 | {"StartDoctypeDeclHandler", |
Fred Drake | 71b63ff | 2002-06-28 22:29:01 +0000 | [diff] [blame] | 2062 | (xmlhandlersetter)XML_SetStartDoctypeDeclHandler, |
Martin v. Löwis | 0078f6c | 2001-01-21 10:18:10 +0000 | [diff] [blame] | 2063 | (xmlhandler)my_StartDoctypeDeclHandler}, |
| 2064 | {"EndDoctypeDeclHandler", |
Fred Drake | 71b63ff | 2002-06-28 22:29:01 +0000 | [diff] [blame] | 2065 | (xmlhandlersetter)XML_SetEndDoctypeDeclHandler, |
Martin v. Löwis | 0078f6c | 2001-01-21 10:18:10 +0000 | [diff] [blame] | 2066 | (xmlhandler)my_EndDoctypeDeclHandler}, |
Fred Drake | 85d835f | 2001-02-08 15:39:08 +0000 | [diff] [blame] | 2067 | {"EntityDeclHandler", |
| 2068 | (xmlhandlersetter)XML_SetEntityDeclHandler, |
| 2069 | (xmlhandler)my_EntityDeclHandler}, |
| 2070 | {"XmlDeclHandler", |
| 2071 | (xmlhandlersetter)XML_SetXmlDeclHandler, |
| 2072 | (xmlhandler)my_XmlDeclHandler}, |
| 2073 | {"ElementDeclHandler", |
| 2074 | (xmlhandlersetter)XML_SetElementDeclHandler, |
| 2075 | (xmlhandler)my_ElementDeclHandler}, |
| 2076 | {"AttlistDeclHandler", |
| 2077 | (xmlhandlersetter)XML_SetAttlistDeclHandler, |
| 2078 | (xmlhandler)my_AttlistDeclHandler}, |
Martin v. Löwis | c847f40 | 2003-01-21 11:09:21 +0000 | [diff] [blame] | 2079 | #if XML_COMBINED_VERSION >= 19504 |
Martin v. Löwis | 069dde2 | 2003-01-21 10:58:18 +0000 | [diff] [blame] | 2080 | {"SkippedEntityHandler", |
| 2081 | (xmlhandlersetter)XML_SetSkippedEntityHandler, |
| 2082 | (xmlhandler)my_SkippedEntityHandler}, |
Martin v. Löwis | c847f40 | 2003-01-21 11:09:21 +0000 | [diff] [blame] | 2083 | #endif |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 2084 | |
Fred Drake | 0582df9 | 2000-07-12 04:49:00 +0000 | [diff] [blame] | 2085 | {NULL, NULL, NULL} /* sentinel */ |
Andrew M. Kuchling | b7f1053 | 2000-03-31 15:43:31 +0000 | [diff] [blame] | 2086 | }; |