| #include "Python.h" |
| #if PY_VERSION_HEX < 0x020000B1 |
| #include <assert.h> |
| #endif |
| #include <ctype.h> |
| |
| #include "compile.h" |
| #include "frameobject.h" |
| #include "expat.h" |
| |
| #ifndef PyGC_HEAD_SIZE |
| #define PyGC_HEAD_SIZE 0 |
| #define PyObject_GC_Init(x) |
| #define PyObject_GC_Fini(m) |
| #define Py_TPFLAGS_GC 0 |
| #endif |
| |
| #if (PY_MAJOR_VERSION == 1 && PY_MINOR_VERSION > 5) || (PY_MAJOR_VERSION == 2 && PY_MINOR_VERSION < 2) |
| /* In Python 1.6, 2.0 and 2.1, disabling Unicode was not possible. */ |
| #define Py_USING_UNICODE |
| #endif |
| |
| enum HandlerTypes { |
| StartElement, |
| EndElement, |
| ProcessingInstruction, |
| CharacterData, |
| UnparsedEntityDecl, |
| NotationDecl, |
| StartNamespaceDecl, |
| EndNamespaceDecl, |
| Comment, |
| StartCdataSection, |
| EndCdataSection, |
| Default, |
| DefaultHandlerExpand, |
| NotStandalone, |
| ExternalEntityRef, |
| StartDoctypeDecl, |
| EndDoctypeDecl, |
| EntityDecl, |
| XmlDecl, |
| ElementDecl, |
| AttlistDecl, |
| _DummyDecl |
| }; |
| |
| static PyObject *ErrorObject; |
| |
| /* ----------------------------------------------------- */ |
| |
| /* Declarations for objects of type xmlparser */ |
| |
| typedef struct { |
| PyObject_HEAD |
| |
| XML_Parser itself; |
| int returns_unicode; /* True if Unicode strings are returned; |
| if false, UTF-8 strings are returned */ |
| int ordered_attributes; /* Return attributes as a list. */ |
| int specified_attributes; /* Report only specified attributes. */ |
| int in_callback; /* Is a callback active? */ |
| PyObject **handlers; |
| } xmlparseobject; |
| |
| staticforward PyTypeObject Xmlparsetype; |
| |
| typedef void (*xmlhandlersetter)(XML_Parser *self, void *meth); |
| typedef void* xmlhandler; |
| |
| struct HandlerInfo { |
| const char *name; |
| xmlhandlersetter setter; |
| xmlhandler handler; |
| PyCodeObject *tb_code; |
| }; |
| |
| staticforward struct HandlerInfo handler_info[64]; |
| |
| /* Set an integer attribute on the error object; return true on success, |
| * false on an exception. |
| */ |
| static int |
| set_error_attr(PyObject *err, char *name, int value) |
| { |
| PyObject *v = PyInt_FromLong(value); |
| |
| if (v != NULL && PyObject_SetAttrString(err, name, v) == -1) { |
| Py_DECREF(v); |
| return 0; |
| } |
| return 1; |
| } |
| |
| /* Build and set an Expat exception, including positioning |
| * information. Always returns NULL. |
| */ |
| static PyObject * |
| set_error(xmlparseobject *self) |
| { |
| PyObject *err; |
| char buffer[256]; |
| XML_Parser parser = self->itself; |
| int lineno = XML_GetErrorLineNumber(parser); |
| int column = XML_GetErrorColumnNumber(parser); |
| enum XML_Error code = XML_GetErrorCode(parser); |
| |
| PyOS_snprintf(buffer, sizeof(buffer), "%.200s: line %i, column %i", |
| XML_ErrorString(code), lineno, column); |
| err = PyObject_CallFunction(ErrorObject, "s", buffer); |
| if ( err != NULL |
| && set_error_attr(err, "code", code) |
| && set_error_attr(err, "offset", column) |
| && set_error_attr(err, "lineno", lineno)) { |
| PyErr_SetObject(ErrorObject, err); |
| } |
| return NULL; |
| } |
| |
| |
| #ifdef Py_USING_UNICODE |
| /* Convert a string of XML_Chars into a Unicode string. |
| Returns None if str is a null pointer. */ |
| |
| static PyObject * |
| conv_string_to_unicode(XML_Char *str) |
| { |
| /* XXX currently this code assumes that XML_Char is 8-bit, |
| and hence in UTF-8. */ |
| /* UTF-8 from Expat, Unicode desired */ |
| if (str == NULL) { |
| Py_INCREF(Py_None); |
| return Py_None; |
| } |
| return PyUnicode_DecodeUTF8((const char *)str, |
| strlen((const char *)str), |
| "strict"); |
| } |
| |
| static PyObject * |
| conv_string_len_to_unicode(const XML_Char *str, int len) |
| { |
| /* XXX currently this code assumes that XML_Char is 8-bit, |
| and hence in UTF-8. */ |
| /* UTF-8 from Expat, Unicode desired */ |
| if (str == NULL) { |
| Py_INCREF(Py_None); |
| return Py_None; |
| } |
| return PyUnicode_DecodeUTF8((const char *)str, len, "strict"); |
| } |
| #endif |
| |
| /* Convert a string of XML_Chars into an 8-bit Python string. |
| Returns None if str is a null pointer. */ |
| |
| static PyObject * |
| conv_string_to_utf8(XML_Char *str) |
| { |
| /* XXX currently this code assumes that XML_Char is 8-bit, |
| and hence in UTF-8. */ |
| /* UTF-8 from Expat, UTF-8 desired */ |
| if (str == NULL) { |
| Py_INCREF(Py_None); |
| return Py_None; |
| } |
| return PyString_FromString((const char *)str); |
| } |
| |
| static PyObject * |
| conv_string_len_to_utf8(const XML_Char *str, int len) |
| { |
| /* XXX currently this code assumes that XML_Char is 8-bit, |
| and hence in UTF-8. */ |
| /* UTF-8 from Expat, UTF-8 desired */ |
| if (str == NULL) { |
| Py_INCREF(Py_None); |
| return Py_None; |
| } |
| return PyString_FromStringAndSize((const char *)str, len); |
| } |
| |
| /* Callback routines */ |
| |
| static void clear_handlers(xmlparseobject *self, int initial); |
| |
| static void |
| flag_error(xmlparseobject *self) |
| { |
| clear_handlers(self, 0); |
| } |
| |
| static PyCodeObject* |
| getcode(enum HandlerTypes slot, char* func_name, int lineno) |
| { |
| PyObject *code = NULL; |
| PyObject *name = NULL; |
| PyObject *nulltuple = NULL; |
| PyObject *filename = NULL; |
| |
| if (handler_info[slot].tb_code == NULL) { |
| code = PyString_FromString(""); |
| if (code == NULL) |
| goto failed; |
| name = PyString_FromString(func_name); |
| if (name == NULL) |
| goto failed; |
| nulltuple = PyTuple_New(0); |
| if (nulltuple == NULL) |
| goto failed; |
| filename = PyString_FromString(__FILE__); |
| handler_info[slot].tb_code = |
| PyCode_New(0, /* argcount */ |
| 0, /* nlocals */ |
| 0, /* stacksize */ |
| 0, /* flags */ |
| code, /* code */ |
| nulltuple, /* consts */ |
| nulltuple, /* names */ |
| nulltuple, /* varnames */ |
| #if PYTHON_API_VERSION >= 1010 |
| nulltuple, /* freevars */ |
| nulltuple, /* cellvars */ |
| #endif |
| filename, /* filename */ |
| name, /* name */ |
| lineno, /* firstlineno */ |
| code /* lnotab */ |
| ); |
| if (handler_info[slot].tb_code == NULL) |
| goto failed; |
| Py_DECREF(code); |
| Py_DECREF(nulltuple); |
| Py_DECREF(filename); |
| Py_DECREF(name); |
| } |
| return handler_info[slot].tb_code; |
| failed: |
| Py_XDECREF(code); |
| Py_XDECREF(name); |
| return NULL; |
| } |
| |
| static PyObject* |
| call_with_frame(PyCodeObject *c, PyObject* func, PyObject* args) |
| { |
| PyThreadState *tstate = PyThreadState_GET(); |
| PyFrameObject *f; |
| PyObject *res; |
| |
| if (c == NULL) |
| return NULL; |
| f = PyFrame_New( |
| tstate, /*back*/ |
| c, /*code*/ |
| tstate->frame->f_globals, /*globals*/ |
| NULL /*locals*/ |
| ); |
| if (f == NULL) |
| return NULL; |
| tstate->frame = f; |
| res = PyEval_CallObject(func, args); |
| if (res == NULL && tstate->curexc_traceback == NULL) |
| PyTraceBack_Here(f); |
| tstate->frame = f->f_back; |
| Py_DECREF(f); |
| return res; |
| } |
| |
| #ifndef Py_USING_UNICODE |
| #define STRING_CONV_FUNC conv_string_to_utf8 |
| #else |
| /* Python 1.6 and later versions */ |
| #define STRING_CONV_FUNC (self->returns_unicode \ |
| ? conv_string_to_unicode : conv_string_to_utf8) |
| #endif |
| |
| static void |
| my_StartElementHandler(void *userData, |
| const XML_Char *name, const XML_Char **atts) |
| { |
| xmlparseobject *self = (xmlparseobject *)userData; |
| |
| if (self->handlers[StartElement] |
| && self->handlers[StartElement] != Py_None) { |
| PyObject *container, *rv, *args; |
| int i, max; |
| |
| /* Set max to the number of slots filled in atts[]; max/2 is |
| * the number of attributes we need to process. |
| */ |
| if (self->specified_attributes) { |
| max = XML_GetSpecifiedAttributeCount(self->itself); |
| } |
| else { |
| max = 0; |
| while (atts[max] != NULL) |
| max += 2; |
| } |
| /* Build the container. */ |
| if (self->ordered_attributes) |
| container = PyList_New(max); |
| else |
| container = PyDict_New(); |
| if (container == NULL) { |
| flag_error(self); |
| return; |
| } |
| for (i = 0; i < max; i += 2) { |
| PyObject *n = STRING_CONV_FUNC((XML_Char *) atts[i]); |
| PyObject *v; |
| if (n == NULL) { |
| flag_error(self); |
| Py_DECREF(container); |
| return; |
| } |
| v = STRING_CONV_FUNC((XML_Char *) atts[i+1]); |
| if (v == NULL) { |
| flag_error(self); |
| Py_DECREF(container); |
| Py_DECREF(n); |
| return; |
| } |
| if (self->ordered_attributes) { |
| PyList_SET_ITEM(container, i, n); |
| PyList_SET_ITEM(container, i+1, v); |
| } |
| else if (PyDict_SetItem(container, n, v)) { |
| flag_error(self); |
| Py_DECREF(n); |
| Py_DECREF(v); |
| return; |
| } |
| else { |
| Py_DECREF(n); |
| Py_DECREF(v); |
| } |
| } |
| args = Py_BuildValue("(O&N)", STRING_CONV_FUNC,name, container); |
| if (args == NULL) { |
| Py_DECREF(container); |
| return; |
| } |
| /* Container is now a borrowed reference; ignore it. */ |
| self->in_callback = 1; |
| rv = call_with_frame(getcode(StartElement, "StartElement", __LINE__), |
| self->handlers[StartElement], args); |
| self->in_callback = 0; |
| Py_DECREF(args); |
| if (rv == NULL) { |
| flag_error(self); |
| return; |
| } |
| Py_DECREF(rv); |
| } |
| } |
| |
| #define RC_HANDLER(RC, NAME, PARAMS, INIT, PARAM_FORMAT, CONVERSION, \ |
| RETURN, GETUSERDATA) \ |
| static RC \ |
| my_##NAME##Handler PARAMS {\ |
| xmlparseobject *self = GETUSERDATA ; \ |
| PyObject *args = NULL; \ |
| PyObject *rv = NULL; \ |
| INIT \ |
| \ |
| if (self->handlers[NAME] \ |
| && self->handlers[NAME] != Py_None) { \ |
| args = Py_BuildValue PARAM_FORMAT ;\ |
| if (!args) { flag_error(self); return RETURN;} \ |
| self->in_callback = 1; \ |
| rv = call_with_frame(getcode(NAME,#NAME,__LINE__), \ |
| self->handlers[NAME], args); \ |
| self->in_callback = 0; \ |
| Py_DECREF(args); \ |
| if (rv == NULL) { \ |
| flag_error(self); \ |
| return RETURN; \ |
| } \ |
| CONVERSION \ |
| Py_DECREF(rv); \ |
| } \ |
| return RETURN; \ |
| } |
| |
| #define VOID_HANDLER(NAME, PARAMS, PARAM_FORMAT) \ |
| RC_HANDLER(void, NAME, PARAMS, ;, PARAM_FORMAT, ;, ;,\ |
| (xmlparseobject *)userData) |
| |
| #define INT_HANDLER(NAME, PARAMS, PARAM_FORMAT)\ |
| RC_HANDLER(int, NAME, PARAMS, int rc=0;, PARAM_FORMAT, \ |
| rc = PyInt_AsLong(rv);, rc, \ |
| (xmlparseobject *)userData) |
| |
| VOID_HANDLER(EndElement, |
| (void *userData, const XML_Char *name), |
| ("(O&)", STRING_CONV_FUNC, name)) |
| |
| VOID_HANDLER(ProcessingInstruction, |
| (void *userData, |
| const XML_Char *target, |
| const XML_Char *data), |
| ("(O&O&)",STRING_CONV_FUNC,target, STRING_CONV_FUNC,data)) |
| |
| #ifndef Py_USING_UNICODE |
| VOID_HANDLER(CharacterData, |
| (void *userData, const XML_Char *data, int len), |
| ("(N)", conv_string_len_to_utf8(data,len))) |
| #else |
| VOID_HANDLER(CharacterData, |
| (void *userData, const XML_Char *data, int len), |
| ("(N)", (self->returns_unicode |
| ? conv_string_len_to_unicode(data,len) |
| : conv_string_len_to_utf8(data,len)))) |
| #endif |
| |
| VOID_HANDLER(UnparsedEntityDecl, |
| (void *userData, |
| const XML_Char *entityName, |
| const XML_Char *base, |
| const XML_Char *systemId, |
| const XML_Char *publicId, |
| const XML_Char *notationName), |
| ("(O&O&O&O&O&)", |
| STRING_CONV_FUNC,entityName, STRING_CONV_FUNC,base, |
| STRING_CONV_FUNC,systemId, STRING_CONV_FUNC,publicId, |
| STRING_CONV_FUNC,notationName)) |
| |
| #ifndef Py_USING_UNICODE |
| VOID_HANDLER(EntityDecl, |
| (void *userData, |
| const XML_Char *entityName, |
| int is_parameter_entity, |
| const XML_Char *value, |
| int value_length, |
| const XML_Char *base, |
| const XML_Char *systemId, |
| const XML_Char *publicId, |
| const XML_Char *notationName), |
| ("O&iNO&O&O&O&", |
| STRING_CONV_FUNC,entityName, is_parameter_entity, |
| conv_string_len_to_utf8(value, value_length), |
| STRING_CONV_FUNC,base, STRING_CONV_FUNC,systemId, |
| STRING_CONV_FUNC,publicId, STRING_CONV_FUNC,notationName)) |
| #else |
| VOID_HANDLER(EntityDecl, |
| (void *userData, |
| const XML_Char *entityName, |
| int is_parameter_entity, |
| const XML_Char *value, |
| int value_length, |
| const XML_Char *base, |
| const XML_Char *systemId, |
| const XML_Char *publicId, |
| const XML_Char *notationName), |
| ("O&iNO&O&O&O&", |
| STRING_CONV_FUNC,entityName, is_parameter_entity, |
| (self->returns_unicode |
| ? conv_string_len_to_unicode(value, value_length) |
| : conv_string_len_to_utf8(value, value_length)), |
| STRING_CONV_FUNC,base, STRING_CONV_FUNC,systemId, |
| STRING_CONV_FUNC,publicId, STRING_CONV_FUNC,notationName)) |
| #endif |
| |
| VOID_HANDLER(XmlDecl, |
| (void *userData, |
| const XML_Char *version, |
| const XML_Char *encoding, |
| int standalone), |
| ("(O&O&i)", |
| STRING_CONV_FUNC,version, STRING_CONV_FUNC,encoding, |
| standalone)) |
| |
| static PyObject * |
| conv_content_model(XML_Content * const model, |
| PyObject *(*conv_string)(XML_Char *)) |
| { |
| PyObject *result = NULL; |
| PyObject *children = PyTuple_New(model->numchildren); |
| int i; |
| |
| if (children != NULL) { |
| assert(model->numchildren < INT_MAX); |
| for (i = 0; i < (int)model->numchildren; ++i) { |
| PyObject *child = conv_content_model(&model->children[i], |
| conv_string); |
| if (child == NULL) { |
| Py_XDECREF(children); |
| return NULL; |
| } |
| PyTuple_SET_ITEM(children, i, child); |
| } |
| result = Py_BuildValue("(iiO&N)", |
| model->type, model->quant, |
| conv_string,model->name, children); |
| } |
| return result; |
| } |
| |
| static PyObject * |
| conv_content_model_utf8(XML_Content * const model) |
| { |
| return conv_content_model(model, conv_string_to_utf8); |
| } |
| |
| #ifdef Py_USING_UNICODE |
| static PyObject * |
| conv_content_model_unicode(XML_Content * const model) |
| { |
| return conv_content_model(model, conv_string_to_unicode); |
| } |
| |
| VOID_HANDLER(ElementDecl, |
| (void *userData, |
| const XML_Char *name, |
| XML_Content *model), |
| ("O&O&", |
| STRING_CONV_FUNC,name, |
| (self->returns_unicode ? conv_content_model_unicode |
| : conv_content_model_utf8),model)) |
| #else |
| VOID_HANDLER(ElementDecl, |
| (void *userData, |
| const XML_Char *name, |
| XML_Content *model), |
| ("O&O&", |
| STRING_CONV_FUNC,name, conv_content_model_utf8,model)) |
| #endif |
| |
| VOID_HANDLER(AttlistDecl, |
| (void *userData, |
| const XML_Char *elname, |
| const XML_Char *attname, |
| const XML_Char *att_type, |
| const XML_Char *dflt, |
| int isrequired), |
| ("(O&O&O&O&i)", |
| STRING_CONV_FUNC,elname, STRING_CONV_FUNC,attname, |
| STRING_CONV_FUNC,att_type, STRING_CONV_FUNC,dflt, |
| isrequired)) |
| |
| VOID_HANDLER(NotationDecl, |
| (void *userData, |
| const XML_Char *notationName, |
| const XML_Char *base, |
| const XML_Char *systemId, |
| const XML_Char *publicId), |
| ("(O&O&O&O&)", |
| STRING_CONV_FUNC,notationName, STRING_CONV_FUNC,base, |
| STRING_CONV_FUNC,systemId, STRING_CONV_FUNC,publicId)) |
| |
| VOID_HANDLER(StartNamespaceDecl, |
| (void *userData, |
| const XML_Char *prefix, |
| const XML_Char *uri), |
| ("(O&O&)", STRING_CONV_FUNC,prefix, STRING_CONV_FUNC,uri)) |
| |
| VOID_HANDLER(EndNamespaceDecl, |
| (void *userData, |
| const XML_Char *prefix), |
| ("(O&)", STRING_CONV_FUNC,prefix)) |
| |
| VOID_HANDLER(Comment, |
| (void *userData, const XML_Char *prefix), |
| ("(O&)", STRING_CONV_FUNC,prefix)) |
| |
| VOID_HANDLER(StartCdataSection, |
| (void *userData), |
| ("()")) |
| |
| VOID_HANDLER(EndCdataSection, |
| (void *userData), |
| ("()")) |
| |
| #ifndef Py_USING_UNICODE |
| VOID_HANDLER(Default, |
| (void *userData, const XML_Char *s, int len), |
| ("(N)", conv_string_len_to_utf8(s,len))) |
| |
| VOID_HANDLER(DefaultHandlerExpand, |
| (void *userData, const XML_Char *s, int len), |
| ("(N)", conv_string_len_to_utf8(s,len))) |
| #else |
| VOID_HANDLER(Default, |
| (void *userData, const XML_Char *s, int len), |
| ("(N)", (self->returns_unicode |
| ? conv_string_len_to_unicode(s,len) |
| : conv_string_len_to_utf8(s,len)))) |
| |
| VOID_HANDLER(DefaultHandlerExpand, |
| (void *userData, const XML_Char *s, int len), |
| ("(N)", (self->returns_unicode |
| ? conv_string_len_to_unicode(s,len) |
| : conv_string_len_to_utf8(s,len)))) |
| #endif |
| |
| INT_HANDLER(NotStandalone, |
| (void *userData), |
| ("()")) |
| |
| RC_HANDLER(int, ExternalEntityRef, |
| (XML_Parser parser, |
| const XML_Char *context, |
| const XML_Char *base, |
| const XML_Char *systemId, |
| const XML_Char *publicId), |
| int rc=0;, |
| ("(O&O&O&O&)", |
| STRING_CONV_FUNC,context, STRING_CONV_FUNC,base, |
| STRING_CONV_FUNC,systemId, STRING_CONV_FUNC,publicId), |
| rc = PyInt_AsLong(rv);, rc, |
| XML_GetUserData(parser)) |
| |
| /* XXX UnknownEncodingHandler */ |
| |
| VOID_HANDLER(StartDoctypeDecl, |
| (void *userData, const XML_Char *doctypeName, |
| const XML_Char *sysid, const XML_Char *pubid, |
| int has_internal_subset), |
| ("(O&O&O&i)", STRING_CONV_FUNC,doctypeName, |
| STRING_CONV_FUNC,sysid, STRING_CONV_FUNC,pubid, |
| has_internal_subset)) |
| |
| VOID_HANDLER(EndDoctypeDecl, (void *userData), ("()")) |
| |
| /* ---------------------------------------------------------------- */ |
| |
| static char xmlparse_Parse__doc__[] = |
| "Parse(data[, isfinal])\n\ |
| Parse XML data. `isfinal' should be true at end of input."; |
| |
| static PyObject * |
| xmlparse_Parse(xmlparseobject *self, PyObject *args) |
| { |
| char *s; |
| int slen; |
| int isFinal = 0; |
| int rv; |
| |
| if (!PyArg_ParseTuple(args, "s#|i:Parse", &s, &slen, &isFinal)) |
| return NULL; |
| rv = XML_Parse(self->itself, s, slen, isFinal); |
| if (PyErr_Occurred()) { |
| return NULL; |
| } |
| else if (rv == 0) { |
| return set_error(self); |
| } |
| return PyInt_FromLong(rv); |
| } |
| |
| /* File reading copied from cPickle */ |
| |
| #define BUF_SIZE 2048 |
| |
| static int |
| readinst(char *buf, int buf_size, PyObject *meth) |
| { |
| PyObject *arg = NULL; |
| PyObject *bytes = NULL; |
| PyObject *str = NULL; |
| int len = -1; |
| |
| if ((bytes = PyInt_FromLong(buf_size)) == NULL) |
| goto finally; |
| |
| if ((arg = PyTuple_New(1)) == NULL) |
| goto finally; |
| |
| PyTuple_SET_ITEM(arg, 0, bytes); |
| |
| if ((str = PyObject_CallObject(meth, arg)) == NULL) |
| goto finally; |
| |
| /* XXX what to do if it returns a Unicode string? */ |
| if (!PyString_Check(str)) { |
| PyErr_Format(PyExc_TypeError, |
| "read() did not return a string object (type=%.400s)", |
| str->ob_type->tp_name); |
| goto finally; |
| } |
| len = PyString_GET_SIZE(str); |
| if (len > buf_size) { |
| PyErr_Format(PyExc_ValueError, |
| "read() returned too much data: " |
| "%i bytes requested, %i returned", |
| buf_size, len); |
| Py_DECREF(str); |
| goto finally; |
| } |
| memcpy(buf, PyString_AsString(str), len); |
| finally: |
| Py_XDECREF(arg); |
| Py_XDECREF(str); |
| return len; |
| } |
| |
| static char xmlparse_ParseFile__doc__[] = |
| "ParseFile(file)\n\ |
| Parse XML data from file-like object."; |
| |
| static PyObject * |
| xmlparse_ParseFile(xmlparseobject *self, PyObject *args) |
| { |
| int rv = 1; |
| PyObject *f; |
| FILE *fp; |
| PyObject *readmethod = NULL; |
| |
| if (!PyArg_ParseTuple(args, "O:ParseFile", &f)) |
| return NULL; |
| |
| if (PyFile_Check(f)) { |
| fp = PyFile_AsFile(f); |
| } |
| else{ |
| fp = NULL; |
| readmethod = PyObject_GetAttrString(f, "read"); |
| if (readmethod == NULL) { |
| PyErr_Clear(); |
| PyErr_SetString(PyExc_TypeError, |
| "argument must have 'read' attribute"); |
| return 0; |
| } |
| } |
| for (;;) { |
| int bytes_read; |
| void *buf = XML_GetBuffer(self->itself, BUF_SIZE); |
| if (buf == NULL) |
| return PyErr_NoMemory(); |
| |
| if (fp) { |
| bytes_read = fread(buf, sizeof(char), BUF_SIZE, fp); |
| if (bytes_read < 0) { |
| PyErr_SetFromErrno(PyExc_IOError); |
| return NULL; |
| } |
| } |
| else { |
| bytes_read = readinst(buf, BUF_SIZE, readmethod); |
| if (bytes_read < 0) |
| return NULL; |
| } |
| rv = XML_ParseBuffer(self->itself, bytes_read, bytes_read == 0); |
| if (PyErr_Occurred()) |
| return NULL; |
| |
| if (!rv || bytes_read == 0) |
| break; |
| } |
| if (rv == 0) { |
| return set_error(self); |
| } |
| return Py_BuildValue("i", rv); |
| } |
| |
| static char xmlparse_SetBase__doc__[] = |
| "SetBase(base_url)\n\ |
| Set the base URL for the parser."; |
| |
| static PyObject * |
| xmlparse_SetBase(xmlparseobject *self, PyObject *args) |
| { |
| char *base; |
| |
| if (!PyArg_ParseTuple(args, "s:SetBase", &base)) |
| return NULL; |
| if (!XML_SetBase(self->itself, base)) { |
| return PyErr_NoMemory(); |
| } |
| Py_INCREF(Py_None); |
| return Py_None; |
| } |
| |
| static char xmlparse_GetBase__doc__[] = |
| "GetBase() -> url\n\ |
| Return base URL string for the parser."; |
| |
| static PyObject * |
| xmlparse_GetBase(xmlparseobject *self, PyObject *args) |
| { |
| if (!PyArg_ParseTuple(args, ":GetBase")) |
| return NULL; |
| |
| return Py_BuildValue("z", XML_GetBase(self->itself)); |
| } |
| |
| static char xmlparse_GetInputContext__doc__[] = |
| "GetInputContext() -> string\n\ |
| Return the untranslated text of the input that caused the current event.\n\ |
| If the event was generated by a large amount of text (such as a start tag\n\ |
| for an element with many attributes), not all of the text may be available."; |
| |
| static PyObject * |
| xmlparse_GetInputContext(xmlparseobject *self, PyObject *args) |
| { |
| PyObject *result = NULL; |
| |
| if (PyArg_ParseTuple(args, ":GetInputContext")) { |
| if (self->in_callback) { |
| int offset, size; |
| const char *buffer |
| = XML_GetInputContext(self->itself, &offset, &size); |
| |
| if (buffer != NULL) |
| result = PyString_FromStringAndSize(buffer + offset, size); |
| else { |
| result = Py_None; |
| Py_INCREF(result); |
| } |
| } |
| else { |
| result = Py_None; |
| Py_INCREF(result); |
| } |
| } |
| return result; |
| } |
| |
| static char xmlparse_ExternalEntityParserCreate__doc__[] = |
| "ExternalEntityParserCreate(context[, encoding])\n\ |
| Create a parser for parsing an external entity based on the\n\ |
| information passed to the ExternalEntityRefHandler."; |
| |
| static PyObject * |
| xmlparse_ExternalEntityParserCreate(xmlparseobject *self, PyObject *args) |
| { |
| char *context; |
| char *encoding = NULL; |
| xmlparseobject *new_parser; |
| int i; |
| |
| if (!PyArg_ParseTuple(args, "z|s:ExternalEntityParserCreate", |
| &context, &encoding)) { |
| return NULL; |
| } |
| |
| #if PY_MAJOR_VERSION == 1 && PY_MINOR_VERSION < 6 |
| new_parser = PyObject_NEW(xmlparseobject, &Xmlparsetype); |
| #else |
| #ifndef Py_TPFLAGS_HAVE_GC |
| /* Python versions 1.6 to 2.1 */ |
| new_parser = PyObject_New(xmlparseobject, &Xmlparsetype); |
| #else |
| /* Python versions 2.2 and later */ |
| new_parser = PyObject_GC_New(xmlparseobject, &Xmlparsetype); |
| #endif |
| #endif |
| |
| if (new_parser == NULL) |
| return NULL; |
| new_parser->returns_unicode = self->returns_unicode; |
| new_parser->ordered_attributes = self->ordered_attributes; |
| new_parser->specified_attributes = self->specified_attributes; |
| new_parser->in_callback = 0; |
| new_parser->itself = XML_ExternalEntityParserCreate(self->itself, context, |
| encoding); |
| new_parser->handlers = 0; |
| #ifdef Py_TPFLAGS_HAVE_GC |
| PyObject_GC_Track(new_parser); |
| #else |
| PyObject_GC_Init(new_parser); |
| #endif |
| |
| if (!new_parser->itself) { |
| Py_DECREF(new_parser); |
| return PyErr_NoMemory(); |
| } |
| |
| XML_SetUserData(new_parser->itself, (void *)new_parser); |
| |
| /* allocate and clear handlers first */ |
| for(i = 0; handler_info[i].name != NULL; i++) |
| /* do nothing */; |
| |
| new_parser->handlers = malloc(sizeof(PyObject *)*i); |
| if (!new_parser->handlers) { |
| Py_DECREF(new_parser); |
| return PyErr_NoMemory(); |
| } |
| clear_handlers(new_parser, 1); |
| |
| /* then copy handlers from self */ |
| for (i = 0; handler_info[i].name != NULL; i++) { |
| if (self->handlers[i]) { |
| Py_INCREF(self->handlers[i]); |
| new_parser->handlers[i] = self->handlers[i]; |
| handler_info[i].setter(new_parser->itself, |
| handler_info[i].handler); |
| } |
| } |
| return (PyObject *)new_parser; |
| } |
| |
| static char xmlparse_SetParamEntityParsing__doc__[] = |
| "SetParamEntityParsing(flag) -> success\n\ |
| Controls parsing of parameter entities (including the external DTD\n\ |
| subset). Possible flag values are XML_PARAM_ENTITY_PARSING_NEVER,\n\ |
| XML_PARAM_ENTITY_PARSING_UNLESS_STANDALONE and\n\ |
| XML_PARAM_ENTITY_PARSING_ALWAYS. Returns true if setting the flag\n\ |
| was successful."; |
| |
| static PyObject* |
| xmlparse_SetParamEntityParsing(xmlparseobject *p, PyObject* args) |
| { |
| int flag; |
| if (!PyArg_ParseTuple(args, "i", &flag)) |
| return NULL; |
| flag = XML_SetParamEntityParsing(p->itself, flag); |
| return PyInt_FromLong(flag); |
| } |
| |
| static struct PyMethodDef xmlparse_methods[] = { |
| {"Parse", (PyCFunction)xmlparse_Parse, |
| METH_VARARGS, xmlparse_Parse__doc__}, |
| {"ParseFile", (PyCFunction)xmlparse_ParseFile, |
| METH_VARARGS, xmlparse_ParseFile__doc__}, |
| {"SetBase", (PyCFunction)xmlparse_SetBase, |
| METH_VARARGS, xmlparse_SetBase__doc__}, |
| {"GetBase", (PyCFunction)xmlparse_GetBase, |
| METH_VARARGS, xmlparse_GetBase__doc__}, |
| {"ExternalEntityParserCreate", (PyCFunction)xmlparse_ExternalEntityParserCreate, |
| METH_VARARGS, xmlparse_ExternalEntityParserCreate__doc__}, |
| {"SetParamEntityParsing", (PyCFunction)xmlparse_SetParamEntityParsing, |
| METH_VARARGS, xmlparse_SetParamEntityParsing__doc__}, |
| {"GetInputContext", (PyCFunction)xmlparse_GetInputContext, |
| METH_VARARGS, xmlparse_GetInputContext__doc__}, |
| {NULL, NULL} /* sentinel */ |
| }; |
| |
| /* ---------- */ |
| |
| |
| #ifdef Py_USING_UNICODE |
| |
| /* |
| pyexpat international encoding support. |
| Make it as simple as possible. |
| */ |
| |
| static char template_buffer[257]; |
| PyObject *template_string = NULL; |
| |
| static void |
| init_template_buffer(void) |
| { |
| int i; |
| for (i = 0; i < 256; i++) { |
| template_buffer[i] = i; |
| } |
| template_buffer[256] = 0; |
| } |
| |
| int |
| PyUnknownEncodingHandler(void *encodingHandlerData, |
| const XML_Char *name, |
| XML_Encoding * info) |
| { |
| PyUnicodeObject *_u_string = NULL; |
| int result = 0; |
| int i; |
| |
| /* Yes, supports only 8bit encodings */ |
| _u_string = (PyUnicodeObject *) |
| PyUnicode_Decode(template_buffer, 256, name, "replace"); |
| |
| if (_u_string == NULL) |
| return result; |
| |
| for (i = 0; i < 256; i++) { |
| /* Stupid to access directly, but fast */ |
| Py_UNICODE c = _u_string->str[i]; |
| if (c == Py_UNICODE_REPLACEMENT_CHARACTER) |
| info->map[i] = -1; |
| else |
| info->map[i] = c; |
| } |
| |
| info->data = NULL; |
| info->convert = NULL; |
| info->release = NULL; |
| result=1; |
| |
| Py_DECREF(_u_string); |
| return result; |
| } |
| |
| #endif |
| |
| static PyObject * |
| newxmlparseobject(char *encoding, char *namespace_separator) |
| { |
| int i; |
| xmlparseobject *self; |
| |
| #if PY_MAJOR_VERSION == 1 && PY_MINOR_VERSION < 6 |
| self = PyObject_NEW(xmlparseobject, &Xmlparsetype); |
| if (self == NULL) |
| return NULL; |
| |
| self->returns_unicode = 0; |
| #else |
| /* Code for versions 1.6 and later */ |
| #ifdef Py_TPFLAGS_HAVE_GC |
| /* Code for versions 2.2 and later */ |
| self = PyObject_GC_New(xmlparseobject, &Xmlparsetype); |
| #else |
| self = PyObject_New(xmlparseobject, &Xmlparsetype); |
| #endif |
| if (self == NULL) |
| return NULL; |
| |
| self->returns_unicode = 1; |
| #endif |
| self->ordered_attributes = 0; |
| self->specified_attributes = 0; |
| self->in_callback = 0; |
| self->handlers = NULL; |
| if (namespace_separator != NULL) { |
| self->itself = XML_ParserCreateNS(encoding, *namespace_separator); |
| } |
| else { |
| self->itself = XML_ParserCreate(encoding); |
| } |
| #ifdef Py_TPFLAGS_HAVE_GC |
| PyObject_GC_Track(self); |
| #else |
| PyObject_GC_Init(self); |
| #endif |
| if (self->itself == NULL) { |
| PyErr_SetString(PyExc_RuntimeError, |
| "XML_ParserCreate failed"); |
| Py_DECREF(self); |
| return NULL; |
| } |
| XML_SetUserData(self->itself, (void *)self); |
| #ifdef Py_USING_UNICODE |
| XML_SetUnknownEncodingHandler(self->itself, (XML_UnknownEncodingHandler) PyUnknownEncodingHandler, NULL); |
| #endif |
| |
| for(i = 0; handler_info[i].name != NULL; i++) |
| /* do nothing */; |
| |
| self->handlers = malloc(sizeof(PyObject *)*i); |
| if (!self->handlers){ |
| Py_DECREF(self); |
| return PyErr_NoMemory(); |
| } |
| clear_handlers(self, 1); |
| |
| return (PyObject*)self; |
| } |
| |
| |
| static void |
| xmlparse_dealloc(xmlparseobject *self) |
| { |
| int i; |
| #ifdef Py_TPFLAGS_HAVE_GC |
| PyObject_GC_UnTrack(self); |
| #else |
| PyObject_GC_Fini(self); |
| #endif |
| if (self->itself != NULL) |
| XML_ParserFree(self->itself); |
| self->itself = NULL; |
| |
| if (self->handlers != NULL) { |
| PyObject *temp; |
| for (i = 0; handler_info[i].name != NULL; i++) { |
| temp = self->handlers[i]; |
| self->handlers[i] = NULL; |
| Py_XDECREF(temp); |
| } |
| free(self->handlers); |
| } |
| #if PY_MAJOR_VERSION == 1 && PY_MINOR_VERSION < 6 |
| /* Code for versions before 1.6 */ |
| free(self); |
| #else |
| #ifndef Py_TPFLAGS_HAVE_GC |
| /* Code for versions 1.6 to 2.1 */ |
| PyObject_Del(self); |
| #else |
| /* Code for versions 2.2 and later. */ |
| PyObject_GC_Del(self); |
| #endif |
| #endif |
| } |
| |
| static int |
| handlername2int(const char *name) |
| { |
| int i; |
| for (i=0; handler_info[i].name != NULL; i++) { |
| if (strcmp(name, handler_info[i].name) == 0) { |
| return i; |
| } |
| } |
| return -1; |
| } |
| |
| static PyObject * |
| xmlparse_getattr(xmlparseobject *self, char *name) |
| { |
| int handlernum; |
| if (strcmp(name, "ErrorCode") == 0) |
| return PyInt_FromLong((long) XML_GetErrorCode(self->itself)); |
| if (strcmp(name, "ErrorLineNumber") == 0) |
| return PyInt_FromLong((long) XML_GetErrorLineNumber(self->itself)); |
| if (strcmp(name, "ErrorColumnNumber") == 0) |
| return PyInt_FromLong((long) XML_GetErrorColumnNumber(self->itself)); |
| if (strcmp(name, "ErrorByteIndex") == 0) |
| return PyInt_FromLong((long) XML_GetErrorByteIndex(self->itself)); |
| if (strcmp(name, "ordered_attributes") == 0) |
| return PyInt_FromLong((long) self->ordered_attributes); |
| if (strcmp(name, "returns_unicode") == 0) |
| return PyInt_FromLong((long) self->returns_unicode); |
| if (strcmp(name, "specified_attributes") == 0) |
| return PyInt_FromLong((long) self->specified_attributes); |
| |
| handlernum = handlername2int(name); |
| |
| if (handlernum != -1 && self->handlers[handlernum] != NULL) { |
| Py_INCREF(self->handlers[handlernum]); |
| return self->handlers[handlernum]; |
| } |
| if (strcmp(name, "__members__") == 0) { |
| int i; |
| PyObject *rc = PyList_New(0); |
| for(i = 0; handler_info[i].name != NULL; i++) { |
| PyList_Append(rc, PyString_FromString(handler_info[i].name)); |
| } |
| PyList_Append(rc, PyString_FromString("ErrorCode")); |
| PyList_Append(rc, PyString_FromString("ErrorLineNumber")); |
| PyList_Append(rc, PyString_FromString("ErrorColumnNumber")); |
| PyList_Append(rc, PyString_FromString("ErrorByteIndex")); |
| PyList_Append(rc, PyString_FromString("ordered_attributes")); |
| PyList_Append(rc, PyString_FromString("returns_unicode")); |
| PyList_Append(rc, PyString_FromString("specified_attributes")); |
| |
| return rc; |
| } |
| return Py_FindMethod(xmlparse_methods, (PyObject *)self, name); |
| } |
| |
| static int |
| sethandler(xmlparseobject *self, const char *name, PyObject* v) |
| { |
| int handlernum = handlername2int(name); |
| if (handlernum != -1) { |
| Py_INCREF(v); |
| Py_XDECREF(self->handlers[handlernum]); |
| self->handlers[handlernum] = v; |
| handler_info[handlernum].setter(self->itself, |
| handler_info[handlernum].handler); |
| return 1; |
| } |
| return 0; |
| } |
| |
| static int |
| xmlparse_setattr(xmlparseobject *self, char *name, PyObject *v) |
| { |
| /* Set attribute 'name' to value 'v'. v==NULL means delete */ |
| if (v == NULL) { |
| PyErr_SetString(PyExc_RuntimeError, "Cannot delete attribute"); |
| return -1; |
| } |
| if (strcmp(name, "ordered_attributes") == 0) { |
| if (PyObject_IsTrue(v)) |
| self->ordered_attributes = 1; |
| else |
| self->ordered_attributes = 0; |
| return 0; |
| } |
| if (strcmp(name, "returns_unicode") == 0) { |
| if (PyObject_IsTrue(v)) { |
| #ifndef Py_USING_UNICODE |
| PyErr_SetString(PyExc_ValueError, |
| "Cannot return Unicode strings in Python 1.5"); |
| return -1; |
| #else |
| self->returns_unicode = 1; |
| #endif |
| } |
| else |
| self->returns_unicode = 0; |
| return 0; |
| } |
| if (strcmp(name, "specified_attributes") == 0) { |
| if (PyObject_IsTrue(v)) |
| self->specified_attributes = 1; |
| else |
| self->specified_attributes = 0; |
| return 0; |
| } |
| if (sethandler(self, name, v)) { |
| return 0; |
| } |
| PyErr_SetString(PyExc_AttributeError, name); |
| return -1; |
| } |
| |
| #ifdef WITH_CYCLE_GC |
| static int |
| xmlparse_traverse(xmlparseobject *op, visitproc visit, void *arg) |
| { |
| int i, err; |
| for (i = 0; handler_info[i].name != NULL; i++) { |
| if (!op->handlers[i]) |
| continue; |
| err = visit(op->handlers[i], arg); |
| if (err) |
| return err; |
| } |
| return 0; |
| } |
| |
| static int |
| xmlparse_clear(xmlparseobject *op) |
| { |
| clear_handlers(op, 0); |
| return 0; |
| } |
| #endif |
| |
| static char Xmlparsetype__doc__[] = |
| "XML parser"; |
| |
| static PyTypeObject Xmlparsetype = { |
| PyObject_HEAD_INIT(NULL) |
| 0, /*ob_size*/ |
| "pyexpat.xmlparser", /*tp_name*/ |
| sizeof(xmlparseobject) + PyGC_HEAD_SIZE,/*tp_basicsize*/ |
| 0, /*tp_itemsize*/ |
| /* methods */ |
| (destructor)xmlparse_dealloc, /*tp_dealloc*/ |
| (printfunc)0, /*tp_print*/ |
| (getattrfunc)xmlparse_getattr, /*tp_getattr*/ |
| (setattrfunc)xmlparse_setattr, /*tp_setattr*/ |
| (cmpfunc)0, /*tp_compare*/ |
| (reprfunc)0, /*tp_repr*/ |
| 0, /*tp_as_number*/ |
| 0, /*tp_as_sequence*/ |
| 0, /*tp_as_mapping*/ |
| (hashfunc)0, /*tp_hash*/ |
| (ternaryfunc)0, /*tp_call*/ |
| (reprfunc)0, /*tp_str*/ |
| 0, /* tp_getattro */ |
| 0, /* tp_setattro */ |
| 0, /* tp_as_buffer */ |
| #ifdef Py_TPFLAGS_HAVE_GC |
| Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC, /*tp_flags*/ |
| #else |
| Py_TPFLAGS_DEFAULT | Py_TPFLAGS_GC, /*tp_flags*/ |
| #endif |
| Xmlparsetype__doc__, /* Documentation string */ |
| #ifdef WITH_CYCLE_GC |
| (traverseproc)xmlparse_traverse, /* tp_traverse */ |
| (inquiry)xmlparse_clear /* tp_clear */ |
| #else |
| 0, 0 |
| #endif |
| }; |
| |
| /* End of code for xmlparser objects */ |
| /* -------------------------------------------------------- */ |
| |
| static char pyexpat_ParserCreate__doc__[] = |
| "ParserCreate([encoding[, namespace_separator]]) -> parser\n\ |
| Return a new XML parser object."; |
| |
| static PyObject * |
| pyexpat_ParserCreate(PyObject *notused, PyObject *args, PyObject *kw) |
| { |
| char *encoding = NULL; |
| char *namespace_separator = NULL; |
| static char *kwlist[] = {"encoding", "namespace_separator", NULL}; |
| |
| if (!PyArg_ParseTupleAndKeywords(args, kw, "|zz:ParserCreate", kwlist, |
| &encoding, &namespace_separator)) |
| return NULL; |
| if (namespace_separator != NULL |
| && strlen(namespace_separator) > 1) { |
| PyErr_SetString(PyExc_ValueError, |
| "namespace_separator must be at most one" |
| " character, omitted, or None"); |
| return NULL; |
| } |
| return newxmlparseobject(encoding, namespace_separator); |
| } |
| |
| static char pyexpat_ErrorString__doc__[] = |
| "ErrorString(errno) -> string\n\ |
| Returns string error for given number."; |
| |
| static PyObject * |
| pyexpat_ErrorString(PyObject *self, PyObject *args) |
| { |
| long code = 0; |
| |
| if (!PyArg_ParseTuple(args, "l:ErrorString", &code)) |
| return NULL; |
| return Py_BuildValue("z", XML_ErrorString((int)code)); |
| } |
| |
| /* List of methods defined in the module */ |
| |
| static struct PyMethodDef pyexpat_methods[] = { |
| {"ParserCreate", (PyCFunction)pyexpat_ParserCreate, |
| METH_VARARGS|METH_KEYWORDS, pyexpat_ParserCreate__doc__}, |
| {"ErrorString", (PyCFunction)pyexpat_ErrorString, |
| METH_VARARGS, pyexpat_ErrorString__doc__}, |
| |
| {NULL, (PyCFunction)NULL, 0, NULL} /* sentinel */ |
| }; |
| |
| /* Module docstring */ |
| |
| static char pyexpat_module_documentation[] = |
| "Python wrapper for Expat parser."; |
| |
| #if PY_VERSION_HEX < 0x20000F0 |
| |
| /* 1.5 compatibility: PyModule_AddObject */ |
| static int |
| PyModule_AddObject(PyObject *m, char *name, PyObject *o) |
| { |
| PyObject *dict; |
| if (!PyModule_Check(m) || o == NULL) |
| return -1; |
| dict = PyModule_GetDict(m); |
| if (dict == NULL) |
| return -1; |
| if (PyDict_SetItemString(dict, name, o)) |
| return -1; |
| Py_DECREF(o); |
| return 0; |
| } |
| |
| int |
| PyModule_AddIntConstant(PyObject *m, char *name, long value) |
| { |
| return PyModule_AddObject(m, name, PyInt_FromLong(value)); |
| } |
| |
| static int |
| PyModule_AddStringConstant(PyObject *m, char *name, char *value) |
| { |
| return PyModule_AddObject(m, name, PyString_FromString(value)); |
| } |
| |
| #endif |
| |
| |
| /* Return a Python string that represents the version number without the |
| * extra cruft added by revision control, even if the right options were |
| * given to the "cvs export" command to make it not include the extra |
| * cruft. |
| */ |
| static PyObject * |
| get_version_string(void) |
| { |
| static char *rcsid = "$Revision$"; |
| char *rev = rcsid; |
| int i = 0; |
| |
| while (!isdigit(*rev)) |
| ++rev; |
| while (rev[i] != ' ' && rev[i] != '\0') |
| ++i; |
| |
| return PyString_FromStringAndSize(rev, i); |
| } |
| |
| /* Initialization function for the module */ |
| |
| #ifndef MODULE_NAME |
| #define MODULE_NAME "pyexpat" |
| #endif |
| |
| #ifndef MODULE_INITFUNC |
| #define MODULE_INITFUNC initpyexpat |
| #endif |
| |
| void MODULE_INITFUNC(void); /* avoid compiler warnings */ |
| |
| DL_EXPORT(void) |
| MODULE_INITFUNC(void) |
| { |
| PyObject *m, *d; |
| PyObject *errmod_name = PyString_FromString(MODULE_NAME ".errors"); |
| PyObject *errors_module; |
| PyObject *modelmod_name; |
| PyObject *model_module; |
| PyObject *sys_modules; |
| |
| if (errmod_name == NULL) |
| return; |
| modelmod_name = PyString_FromString(MODULE_NAME ".model"); |
| if (modelmod_name == NULL) |
| return; |
| |
| Xmlparsetype.ob_type = &PyType_Type; |
| |
| /* Create the module and add the functions */ |
| m = Py_InitModule3(MODULE_NAME, pyexpat_methods, |
| pyexpat_module_documentation); |
| |
| /* Add some symbolic constants to the module */ |
| if (ErrorObject == NULL) { |
| ErrorObject = PyErr_NewException("xml.parsers.expat.ExpatError", |
| NULL, NULL); |
| if (ErrorObject == NULL) |
| return; |
| } |
| Py_INCREF(ErrorObject); |
| PyModule_AddObject(m, "error", ErrorObject); |
| Py_INCREF(ErrorObject); |
| PyModule_AddObject(m, "ExpatError", ErrorObject); |
| Py_INCREF(&Xmlparsetype); |
| PyModule_AddObject(m, "XMLParserType", (PyObject *) &Xmlparsetype); |
| |
| PyModule_AddObject(m, "__version__", get_version_string()); |
| PyModule_AddStringConstant(m, "EXPAT_VERSION", |
| (char *) XML_ExpatVersion()); |
| { |
| XML_Expat_Version info = XML_ExpatVersionInfo(); |
| PyModule_AddObject(m, "version_info", |
| Py_BuildValue("(iii)", info.major, |
| info.minor, info.micro)); |
| } |
| #ifdef Py_USING_UNICODE |
| init_template_buffer(); |
| #endif |
| /* XXX When Expat supports some way of figuring out how it was |
| compiled, this should check and set native_encoding |
| appropriately. |
| */ |
| PyModule_AddStringConstant(m, "native_encoding", "UTF-8"); |
| |
| sys_modules = PySys_GetObject("modules"); |
| d = PyModule_GetDict(m); |
| errors_module = PyDict_GetItem(d, errmod_name); |
| if (errors_module == NULL) { |
| errors_module = PyModule_New(MODULE_NAME ".errors"); |
| if (errors_module != NULL) { |
| PyDict_SetItem(sys_modules, errmod_name, errors_module); |
| /* gives away the reference to errors_module */ |
| PyModule_AddObject(m, "errors", errors_module); |
| } |
| } |
| Py_DECREF(errmod_name); |
| model_module = PyDict_GetItem(d, modelmod_name); |
| if (model_module == NULL) { |
| model_module = PyModule_New(MODULE_NAME ".model"); |
| if (model_module != NULL) { |
| PyDict_SetItem(sys_modules, modelmod_name, model_module); |
| /* gives away the reference to model_module */ |
| PyModule_AddObject(m, "model", model_module); |
| } |
| } |
| Py_DECREF(modelmod_name); |
| if (errors_module == NULL || model_module == NULL) |
| /* Don't core dump later! */ |
| return; |
| |
| #define MYCONST(name) \ |
| PyModule_AddStringConstant(errors_module, #name, \ |
| (char*)XML_ErrorString(name)) |
| |
| MYCONST(XML_ERROR_NO_MEMORY); |
| MYCONST(XML_ERROR_SYNTAX); |
| MYCONST(XML_ERROR_NO_ELEMENTS); |
| MYCONST(XML_ERROR_INVALID_TOKEN); |
| MYCONST(XML_ERROR_UNCLOSED_TOKEN); |
| MYCONST(XML_ERROR_PARTIAL_CHAR); |
| MYCONST(XML_ERROR_TAG_MISMATCH); |
| MYCONST(XML_ERROR_DUPLICATE_ATTRIBUTE); |
| MYCONST(XML_ERROR_JUNK_AFTER_DOC_ELEMENT); |
| MYCONST(XML_ERROR_PARAM_ENTITY_REF); |
| MYCONST(XML_ERROR_UNDEFINED_ENTITY); |
| MYCONST(XML_ERROR_RECURSIVE_ENTITY_REF); |
| MYCONST(XML_ERROR_ASYNC_ENTITY); |
| MYCONST(XML_ERROR_BAD_CHAR_REF); |
| MYCONST(XML_ERROR_BINARY_ENTITY_REF); |
| MYCONST(XML_ERROR_ATTRIBUTE_EXTERNAL_ENTITY_REF); |
| MYCONST(XML_ERROR_MISPLACED_XML_PI); |
| MYCONST(XML_ERROR_UNKNOWN_ENCODING); |
| MYCONST(XML_ERROR_INCORRECT_ENCODING); |
| MYCONST(XML_ERROR_UNCLOSED_CDATA_SECTION); |
| MYCONST(XML_ERROR_EXTERNAL_ENTITY_HANDLING); |
| MYCONST(XML_ERROR_NOT_STANDALONE); |
| |
| PyModule_AddStringConstant(errors_module, "__doc__", |
| "Constants used to describe error conditions."); |
| |
| #undef MYCONST |
| |
| #define MYCONST(c) PyModule_AddIntConstant(m, #c, c) |
| MYCONST(XML_PARAM_ENTITY_PARSING_NEVER); |
| MYCONST(XML_PARAM_ENTITY_PARSING_UNLESS_STANDALONE); |
| MYCONST(XML_PARAM_ENTITY_PARSING_ALWAYS); |
| #undef MYCONST |
| |
| #define MYCONST(c) PyModule_AddIntConstant(model_module, #c, c) |
| PyModule_AddStringConstant(model_module, "__doc__", |
| "Constants used to interpret content model information."); |
| |
| MYCONST(XML_CTYPE_EMPTY); |
| MYCONST(XML_CTYPE_ANY); |
| MYCONST(XML_CTYPE_MIXED); |
| MYCONST(XML_CTYPE_NAME); |
| MYCONST(XML_CTYPE_CHOICE); |
| MYCONST(XML_CTYPE_SEQ); |
| |
| MYCONST(XML_CQUANT_NONE); |
| MYCONST(XML_CQUANT_OPT); |
| MYCONST(XML_CQUANT_REP); |
| MYCONST(XML_CQUANT_PLUS); |
| #undef MYCONST |
| } |
| |
| static void |
| clear_handlers(xmlparseobject *self, int initial) |
| { |
| int i = 0; |
| PyObject *temp; |
| |
| for (; handler_info[i].name!=NULL; i++) { |
| if (initial) |
| self->handlers[i]=NULL; |
| else { |
| temp = self->handlers[i]; |
| self->handlers[i] = NULL; |
| Py_XDECREF(temp); |
| handler_info[i].setter(self->itself, NULL); |
| } |
| } |
| } |
| |
| typedef void (*pairsetter)(XML_Parser, void *handler1, void *handler2); |
| |
| static void |
| pyxml_UpdatePairedHandlers(xmlparseobject *self, |
| int startHandler, |
| int endHandler, |
| pairsetter setter) |
| { |
| void *start_handler = NULL; |
| void *end_handler = NULL; |
| |
| if (self->handlers[startHandler] |
| && self->handlers[startHandler] != Py_None) { |
| start_handler = handler_info[startHandler].handler; |
| } |
| if (self->handlers[endHandler] |
| && self->handlers[endHandler] != Py_None) { |
| end_handler = handler_info[endHandler].handler; |
| } |
| setter(self->itself, start_handler, end_handler); |
| } |
| |
| static void |
| pyxml_SetStartElementHandler(XML_Parser *parser, void *junk) |
| { |
| pyxml_UpdatePairedHandlers((xmlparseobject *)XML_GetUserData(parser), |
| StartElement, EndElement, |
| (pairsetter)XML_SetElementHandler); |
| } |
| |
| static void |
| pyxml_SetEndElementHandler(XML_Parser *parser, void *junk) |
| { |
| pyxml_UpdatePairedHandlers((xmlparseobject *)XML_GetUserData(parser), |
| StartElement, EndElement, |
| (pairsetter)XML_SetElementHandler); |
| } |
| |
| static void |
| pyxml_SetStartNamespaceDeclHandler(XML_Parser *parser, void *junk) |
| { |
| pyxml_UpdatePairedHandlers((xmlparseobject *)XML_GetUserData(parser), |
| StartNamespaceDecl, EndNamespaceDecl, |
| (pairsetter)XML_SetNamespaceDeclHandler); |
| } |
| |
| static void |
| pyxml_SetEndNamespaceDeclHandler(XML_Parser *parser, void *junk) |
| { |
| pyxml_UpdatePairedHandlers((xmlparseobject *)XML_GetUserData(parser), |
| StartNamespaceDecl, EndNamespaceDecl, |
| (pairsetter)XML_SetNamespaceDeclHandler); |
| } |
| |
| static void |
| pyxml_SetStartCdataSection(XML_Parser *parser, void *junk) |
| { |
| pyxml_UpdatePairedHandlers((xmlparseobject *)XML_GetUserData(parser), |
| StartCdataSection, EndCdataSection, |
| (pairsetter)XML_SetCdataSectionHandler); |
| } |
| |
| static void |
| pyxml_SetEndCdataSection(XML_Parser *parser, void *junk) |
| { |
| pyxml_UpdatePairedHandlers((xmlparseobject *)XML_GetUserData(parser), |
| StartCdataSection, EndCdataSection, |
| (pairsetter)XML_SetCdataSectionHandler); |
| } |
| |
| static void |
| pyxml_SetStartDoctypeDeclHandler(XML_Parser *parser, void *junk) |
| { |
| pyxml_UpdatePairedHandlers((xmlparseobject *)XML_GetUserData(parser), |
| StartDoctypeDecl, EndDoctypeDecl, |
| (pairsetter)XML_SetDoctypeDeclHandler); |
| } |
| |
| static void |
| pyxml_SetEndDoctypeDeclHandler(XML_Parser *parser, void *junk) |
| { |
| pyxml_UpdatePairedHandlers((xmlparseobject *)XML_GetUserData(parser), |
| StartDoctypeDecl, EndDoctypeDecl, |
| (pairsetter)XML_SetDoctypeDeclHandler); |
| } |
| |
| statichere struct HandlerInfo handler_info[] = { |
| {"StartElementHandler", |
| pyxml_SetStartElementHandler, |
| (xmlhandler)my_StartElementHandler}, |
| {"EndElementHandler", |
| pyxml_SetEndElementHandler, |
| (xmlhandler)my_EndElementHandler}, |
| {"ProcessingInstructionHandler", |
| (xmlhandlersetter)XML_SetProcessingInstructionHandler, |
| (xmlhandler)my_ProcessingInstructionHandler}, |
| {"CharacterDataHandler", |
| (xmlhandlersetter)XML_SetCharacterDataHandler, |
| (xmlhandler)my_CharacterDataHandler}, |
| {"UnparsedEntityDeclHandler", |
| (xmlhandlersetter)XML_SetUnparsedEntityDeclHandler, |
| (xmlhandler)my_UnparsedEntityDeclHandler }, |
| {"NotationDeclHandler", |
| (xmlhandlersetter)XML_SetNotationDeclHandler, |
| (xmlhandler)my_NotationDeclHandler }, |
| {"StartNamespaceDeclHandler", |
| pyxml_SetStartNamespaceDeclHandler, |
| (xmlhandler)my_StartNamespaceDeclHandler }, |
| {"EndNamespaceDeclHandler", |
| pyxml_SetEndNamespaceDeclHandler, |
| (xmlhandler)my_EndNamespaceDeclHandler }, |
| {"CommentHandler", |
| (xmlhandlersetter)XML_SetCommentHandler, |
| (xmlhandler)my_CommentHandler}, |
| {"StartCdataSectionHandler", |
| pyxml_SetStartCdataSection, |
| (xmlhandler)my_StartCdataSectionHandler}, |
| {"EndCdataSectionHandler", |
| pyxml_SetEndCdataSection, |
| (xmlhandler)my_EndCdataSectionHandler}, |
| {"DefaultHandler", |
| (xmlhandlersetter)XML_SetDefaultHandler, |
| (xmlhandler)my_DefaultHandler}, |
| {"DefaultHandlerExpand", |
| (xmlhandlersetter)XML_SetDefaultHandlerExpand, |
| (xmlhandler)my_DefaultHandlerExpandHandler}, |
| {"NotStandaloneHandler", |
| (xmlhandlersetter)XML_SetNotStandaloneHandler, |
| (xmlhandler)my_NotStandaloneHandler}, |
| {"ExternalEntityRefHandler", |
| (xmlhandlersetter)XML_SetExternalEntityRefHandler, |
| (xmlhandler)my_ExternalEntityRefHandler }, |
| {"StartDoctypeDeclHandler", |
| pyxml_SetStartDoctypeDeclHandler, |
| (xmlhandler)my_StartDoctypeDeclHandler}, |
| {"EndDoctypeDeclHandler", |
| pyxml_SetEndDoctypeDeclHandler, |
| (xmlhandler)my_EndDoctypeDeclHandler}, |
| {"EntityDeclHandler", |
| (xmlhandlersetter)XML_SetEntityDeclHandler, |
| (xmlhandler)my_EntityDeclHandler}, |
| {"XmlDeclHandler", |
| (xmlhandlersetter)XML_SetXmlDeclHandler, |
| (xmlhandler)my_XmlDeclHandler}, |
| {"ElementDeclHandler", |
| (xmlhandlersetter)XML_SetElementDeclHandler, |
| (xmlhandler)my_ElementDeclHandler}, |
| {"AttlistDeclHandler", |
| (xmlhandlersetter)XML_SetAttlistDeclHandler, |
| (xmlhandler)my_AttlistDeclHandler}, |
| |
| {NULL, NULL, NULL} /* sentinel */ |
| }; |