#include "Python.h"

#define VERSION "1.0.6"

#define USE_EXPAT

#define STATIC_CHILDREN 4

#if 0

static int memory = 0;

#define ALLOC(size, comment)\

do { memory += size; printf("%8d - %s\n", memory, comment); } while (0)

#define RELEASE(size, comment)\

do { memory -= size; printf("%8d - %s\n", memory, comment); } while (0)

#define ALLOC(size, comment)

#define RELEASE(size, comment)

#if defined(_MSC_VER)

#define LOCAL(type) static __inline type __fastcall

#define LOCAL(type) static type

#define JOIN_GET(p) ((Py_uintptr_t) (p) & 1)

#define JOIN_SET(p, flag) ((void*) ((Py_uintptr_t) (JOIN_OBJ(p)) | (flag)))

#define JOIN_OBJ(p) ((PyObject*) ((Py_uintptr_t) (p) & ~1))

static PyObject* elementtree_copyelement_obj;

static PyObject* elementtree_deepcopy_obj;

static PyObject* elementtree_getiterator_obj;

static PyObject* elementpath_obj;

LOCAL(PyObject*)

deepcopy(PyObject* object, PyObject* memo)

{

/* do a deep copy of the given object */

PyObject* args;

PyObject* result;

if (!elementtree_deepcopy_obj) {

PyErr_SetString(

PyExc_RuntimeError,

"deepcopy helper not found"

);

return NULL;

}

args = PyTuple_New(2);

if (!args)

return NULL;

Py_INCREF(object); PyTuple_SET_ITEM(args, 0, (PyObject*) object);

Py_INCREF(memo); PyTuple_SET_ITEM(args, 1, (PyObject*) memo);

result = PyObject_CallObject(elementtree_deepcopy_obj, args);

Py_DECREF(args);

return result;

}

LOCAL(PyObject*)

list_join(PyObject* list)

{

/* join list elements (destroying the list in the process) */

PyObject* joiner;

PyObject* function;

PyObject* args;

PyObject* result;

switch (PyList_GET_SIZE(list)) {

case 0:

Py_DECREF(list);

return PyBytes_FromString("");

case 1:

result = PyList_GET_ITEM(list, 0);

Py_INCREF(result);

Py_DECREF(list);

return result;

}

/* two or more elements: slice out a suitable separator from the

joiner = PySequence_GetSlice(PyList_GET_ITEM(list, 0), 0, 0);

if (!joiner)

return NULL;

function = PyObject_GetAttrString(joiner, "join");

if (!function) {

Py_DECREF(joiner);

return NULL;

}

args = PyTuple_New(1);

if (!args)

return NULL;

PyTuple_SET_ITEM(args, 0, list);

result = PyObject_CallObject(function, args);

Py_DECREF(args); /* also removes list */

Py_DECREF(function);

Py_DECREF(joiner);

return result;

}

#if (PY_VERSION_HEX < 0x02020000)

LOCAL(int)

PyDict_Update(PyObject* dict, PyObject* other)

{

/* PyDict_Update emulation for 2.1 and earlier */

PyObject* res;

res = PyObject_CallMethod(dict, "update", "O", other);

if (!res)

return -1;

Py_DECREF(res);

return 0;

}

typedef struct {

/* attributes (a dictionary object), or None if no attributes */

PyObject* attrib;

/* child elements */

int length; /* actual number of items */

int allocated; /* allocated items */

/* this either points to _children or to a malloced buffer */

PyObject* *children;

PyObject* _children[STATIC_CHILDREN];

} ElementObjectExtra;

typedef struct {

PyObject_HEAD

/* element tag (a string). */

PyObject* tag;

/* text before first child. note that this is a tagged pointer;

PyObject* text;

/* text after this element, in parent. note that this is a tagged

PyObject* tail;

ElementObjectExtra* extra;

} ElementObject;

static PyTypeObject Element_Type;

#define Element_CheckExact(op) (Py_TYPE(op) == &Element_Type)

LOCAL(int)

element_new_extra(ElementObject* self, PyObject* attrib)

{

self->extra = PyObject_Malloc(sizeof(ElementObjectExtra));

if (!self->extra)

return -1;

if (!attrib)

attrib = Py_None;

Py_INCREF(attrib);

self->extra->attrib = attrib;

self->extra->length = 0;

self->extra->allocated = STATIC_CHILDREN;

self->extra->children = self->extra->_children;

return 0;

}

LOCAL(void)

element_dealloc_extra(ElementObject* self)

{

int i;

Py_DECREF(self->extra->attrib);

for (i = 0; i < self->extra->length; i++)

Py_DECREF(self->extra->children[i]);

if (self->extra->children != self->extra->_children)

PyObject_Free(self->extra->children);

PyObject_Free(self->extra);

}

LOCAL(PyObject*)

element_new(PyObject* tag, PyObject* attrib)

{

ElementObject* self;

self = PyObject_New(ElementObject, &Element_Type);

if (self == NULL)

return NULL;

/* use None for empty dictionaries */

if (PyDict_CheckExact(attrib) && !PyDict_Size(attrib))

attrib = Py_None;

self->extra = NULL;

if (attrib != Py_None) {

if (element_new_extra(self, attrib) < 0) {

PyObject_Del(self);

return NULL;

}

self->extra->length = 0;

self->extra->allocated = STATIC_CHILDREN;

self->extra->children = self->extra->_children;

}

Py_INCREF(tag);

self->tag = tag;

Py_INCREF(Py_None);

self->text = Py_None;

Py_INCREF(Py_None);

self->tail = Py_None;

ALLOC(sizeof(ElementObject), "create element");

return (PyObject*) self;

}

LOCAL(int)

element_resize(ElementObject* self, int extra)

{

int size;

PyObject* *children;

/* make sure self->children can hold the given number of extra

if (!self->extra)

element_new_extra(self, NULL);

size = self->extra->length + extra;

if (size > self->extra->allocated) {

/* use Python 2.4's list growth strategy */

size = (size >> 3) + (size < 9 ? 3 : 6) + size;

/* Coverity CID #182 size_error: Allocating 1 bytes to pointer "children"

size = size ? size : 1;

if (self->extra->children != self->extra->_children) {

/* Coverity CID #182 size_error: Allocating 1 bytes to pointer

children = PyObject_Realloc(self->extra->children,

size * sizeof(PyObject*));

if (!children)

goto nomemory;

} else {

children = PyObject_Malloc(size * sizeof(PyObject*));

if (!children)

goto nomemory;

/* copy existing children from static area to malloc buffer */

memcpy(children, self->extra->children,

self->extra->length * sizeof(PyObject*));

}

self->extra->children = children;

self->extra->allocated = size;

}

return 0;

nomemory:

PyErr_NoMemory();

return -1;

}

LOCAL(int)

element_add_subelement(ElementObject* self, PyObject* element)

{

/* add a child element to a parent */

if (element_resize(self, 1) < 0)

return -1;

Py_INCREF(element);

self->extra->children[self->extra->length] = element;

self->extra->length++;

return 0;

}

LOCAL(PyObject*)

element_get_attrib(ElementObject* self)

{

/* return borrowed reference to attrib dictionary */

/* note: this function assumes that the extra section exists */

PyObject* res = self->extra->attrib;

if (res == Py_None) {

/* create missing dictionary */

res = PyDict_New();

if (!res)

return NULL;

self->extra->attrib = res;

}

return res;

}

LOCAL(PyObject*)

element_get_text(ElementObject* self)

{

/* return borrowed reference to text attribute */

PyObject* res = self->text;

if (JOIN_GET(res)) {

res = JOIN_OBJ(res);

if (PyList_CheckExact(res)) {

res = list_join(res);

if (!res)

return NULL;

self->text = res;

}

}

return res;

}

LOCAL(PyObject*)

element_get_tail(ElementObject* self)

{

/* return borrowed reference to text attribute */

PyObject* res = self->tail;

if (JOIN_GET(res)) {

res = JOIN_OBJ(res);

if (PyList_CheckExact(res)) {

res = list_join(res);

if (!res)

return NULL;

self->tail = res;

}

}

return res;

}

static PyObject*

element(PyObject* self, PyObject* args, PyObject* kw)

{

PyObject* elem;

PyObject* tag;

PyObject* attrib = NULL;

if (!PyArg_ParseTuple(args, "O|O!:Element", &tag,

&PyDict_Type, &attrib))

return NULL;

if (attrib || kw) {

attrib = (attrib) ? PyDict_Copy(attrib) : PyDict_New();

if (!attrib)

return NULL;

if (kw)

PyDict_Update(attrib, kw);

} else {

Py_INCREF(Py_None);

attrib = Py_None;

}

elem = element_new(tag, attrib);

Py_DECREF(attrib);

return elem;

}

static PyObject*

subelement(PyObject* self, PyObject* args, PyObject* kw)

{

PyObject* elem;

ElementObject* parent;

PyObject* tag;

PyObject* attrib = NULL;

if (!PyArg_ParseTuple(args, "O!O|O!:SubElement",

&Element_Type, &parent, &tag,

&PyDict_Type, &attrib))

return NULL;

if (attrib || kw) {

attrib = (attrib) ? PyDict_Copy(attrib) : PyDict_New();

if (!attrib)

return NULL;

if (kw)

PyDict_Update(attrib, kw);

} else {

Py_INCREF(Py_None);

attrib = Py_None;

}

elem = element_new(tag, attrib);

Py_DECREF(attrib);

if (element_add_subelement(parent, elem) < 0) {

Py_DECREF(elem);

return NULL;

}

return elem;

}

static void

element_dealloc(ElementObject* self)

{

if (self->extra)

element_dealloc_extra(self);

/* discard attributes */

Py_DECREF(self->tag);

Py_DECREF(JOIN_OBJ(self->text));

Py_DECREF(JOIN_OBJ(self->tail));

RELEASE(sizeof(ElementObject), "destroy element");

PyObject_Del(self);

}

static PyObject*

element_append(ElementObject* self, PyObject* args)

{

PyObject* element;

if (!PyArg_ParseTuple(args, "O!:append", &Element_Type, &element))

return NULL;

if (element_add_subelement(self, element) < 0)

return NULL;

Py_RETURN_NONE;

}

static PyObject*

element_clear(ElementObject* self, PyObject* args)

{

if (!PyArg_ParseTuple(args, ":clear"))

return NULL;

if (self->extra) {

element_dealloc_extra(self);

self->extra = NULL;

}

Py_INCREF(Py_None);

Py_DECREF(JOIN_OBJ(self->text));

self->text = Py_None;

Py_INCREF(Py_None);

Py_DECREF(JOIN_OBJ(self->tail));

self->tail = Py_None;

Py_RETURN_NONE;

}

static PyObject*

element_copy(ElementObject* self, PyObject* args)

{

int i;

ElementObject* element;

if (!PyArg_ParseTuple(args, ":__copy__"))

return NULL;

element = (ElementObject*) element_new(

self->tag, (self->extra) ? self->extra->attrib : Py_None

);

if (!element)

return NULL;

Py_DECREF(JOIN_OBJ(element->text));

element->text = self->text;

Py_INCREF(JOIN_OBJ(element->text));

Py_DECREF(JOIN_OBJ(element->tail));

element->tail = self->tail;

Py_INCREF(JOIN_OBJ(element->tail));

if (self->extra) {

if (element_resize(element, self->extra->length) < 0) {

Py_DECREF(element);

return NULL;

}

for (i = 0; i < self->extra->length; i++) {

Py_INCREF(self->extra->children[i]);

element->extra->children[i] = self->extra->children[i];

}

element->extra->length = self->extra->length;

}

return (PyObject*) element;

}

static PyObject*

element_deepcopy(ElementObject* self, PyObject* args)

{

int i;

ElementObject* element;

PyObject* tag;

PyObject* attrib;

PyObject* text;

PyObject* tail;

PyObject* id;

PyObject* memo;

if (!PyArg_ParseTuple(args, "O:__deepcopy__", &memo))

return NULL;

tag = deepcopy(self->tag, memo);

if (!tag)

return NULL;

if (self->extra) {

attrib = deepcopy(self->extra->attrib, memo);

if (!attrib) {

Py_DECREF(tag);

return NULL;

}

} else {

Py_INCREF(Py_None);

attrib = Py_None;

}

element = (ElementObject*) element_new(tag, attrib);

Py_DECREF(tag);

Py_DECREF(attrib);

if (!element)

return NULL;

text = deepcopy(JOIN_OBJ(self->text), memo);

if (!text)

goto error;

Py_DECREF(element->text);

element->text = JOIN_SET(text, JOIN_GET(self->text));

tail = deepcopy(JOIN_OBJ(self->tail), memo);

if (!tail)

goto error;

Py_DECREF(element->tail);

element->tail = JOIN_SET(tail, JOIN_GET(self->tail));

if (self->extra) {

if (element_resize(element, self->extra->length) < 0)

goto error;

for (i = 0; i < self->extra->length; i++) {

PyObject* child = deepcopy(self->extra->children[i], memo);

if (!child) {

element->extra->length = i;

goto error;

}

element->extra->children[i] = child;

}

element->extra->length = self->extra->length;

}

/* add object to memo dictionary (so deepcopy won't visit it again) */

id = PyLong_FromLong((Py_uintptr_t) self);

i = PyDict_SetItem(memo, id, (PyObject*) element);

Py_DECREF(id);

if (i < 0)

goto error;

return (PyObject*) element;

error:

Py_DECREF(element);

return NULL;

}

LOCAL(int)

checkpath(PyObject* tag)

{

Py_ssize_t i;

int check = 1;

/* check if a tag contains an xpath character */

#define PATHCHAR(ch) (ch == '/' || ch == '*' || ch == '[' || ch == '@')

if (PyUnicode_Check(tag)) {

Py_UNICODE *p = PyUnicode_AS_UNICODE(tag);

for (i = 0; i < PyUnicode_GET_SIZE(tag); i++) {

if (p[i] == '{')

check = 0;

else if (p[i] == '}')

check = 1;

else if (check && PATHCHAR(p[i]))

return 1;

}

return 0;

}

if (PyBytes_Check(tag)) {

char *p = PyBytes_AS_STRING(tag);

for (i = 0; i < PyBytes_GET_SIZE(tag); i++) {

if (p[i] == '{')

check = 0;

else if (p[i] == '}')

check = 1;

else if (check && PATHCHAR(p[i]))

return 1;

}

return 0;

}

return 1; /* unknown type; might be path expression */

}

static PyObject*

element_find(ElementObject* self, PyObject* args)

{

int i;

PyObject* tag;

if (!PyArg_ParseTuple(args, "O:find", &tag))

return NULL;

if (checkpath(tag))

return PyObject_CallMethod(

elementpath_obj, "find", "OO", self, tag

);

if (!self->extra)

Py_RETURN_NONE;

for (i = 0; i < self->extra->length; i++) {

PyObject* item = self->extra->children[i];

if (Element_CheckExact(item) &&

PyObject_RichCompareBool(((ElementObject*)item)->tag, tag, Py_EQ) == 1) {

Py_INCREF(item);

return item;

}

}

Py_RETURN_NONE;

}

static PyObject*

element_findtext(ElementObject* self, PyObject* args)

{

int i;

PyObject* tag;

PyObject* default_value = Py_None;

if (!PyArg_ParseTuple(args, "O|O:findtext", &tag, &default_value))

return NULL;

if (checkpath(tag))

return PyObject_CallMethod(

elementpath_obj, "findtext", "OOO", self, tag, default_value

);

if (!self->extra) {

Py_INCREF(default_value);

return default_value;

}

for (i = 0; i < self->extra->length; i++) {

ElementObject* item = (ElementObject*) self->extra->children[i];

if (Element_CheckExact(item) && (PyObject_RichCompareBool(item->tag, tag, Py_EQ) == 1)) {

PyObject* text = element_get_text(item);

if (text == Py_None)

return PyBytes_FromString("");

Py_XINCREF(text);

return text;

}

}

Py_INCREF(default_value);

return default_value;

}

static PyObject*

element_findall(ElementObject* self, PyObject* args)

{

int i;

PyObject* out;

PyObject* tag;

if (!PyArg_ParseTuple(args, "O:findall", &tag))

return NULL;

if (checkpath(tag))

return PyObject_CallMethod(

elementpath_obj, "findall", "OO", self, tag

);

out = PyList_New(0);

if (!out)

return NULL;

if (!self->extra)

return out;

for (i = 0; i < self->extra->length; i++) {

PyObject* item = self->extra->children[i];

if (Element_CheckExact(item) &&

PyObject_RichCompareBool(((ElementObject*)item)->tag, tag, Py_EQ) == 1) {

if (PyList_Append(out, item) < 0) {

Py_DECREF(out);

return NULL;

}

}

}

return out;

}

static PyObject*

element_get(ElementObject* self, PyObject* args)

{

PyObject* value;

PyObject* key;

PyObject* default_value = Py_None;

if (!PyArg_ParseTuple(args, "O|O:get", &key, &default_value))

return NULL;

if (!self->extra || self->extra->attrib == Py_None)

value = default_value;

else {

value = PyDict_GetItem(self->extra->attrib, key);

if (!value)

value = default_value;

}

Py_INCREF(value);

return value;

}

static PyObject*

element_getchildren(ElementObject* self, PyObject* args)

{

int i;

PyObject* list;

if (!PyArg_ParseTuple(args, ":getchildren"))

return NULL;

if (!self->extra)

return PyList_New(0);

list = PyList_New(self->extra->length);

if (!list)

return NULL;

for (i = 0; i < self->extra->length; i++) {

PyObject* item = self->extra->children[i];

Py_INCREF(item);

PyList_SET_ITEM(list, i, item);

}

return list;

}

static PyObject*

element_getiterator(ElementObject* self, PyObject* args)

{

PyObject* result;

PyObject* tag = Py_None;

if (!PyArg_ParseTuple(args, "|O:getiterator", &tag))

return NULL;

if (!elementtree_getiterator_obj) {

PyErr_SetString(

PyExc_RuntimeError,

"getiterator helper not found"

);

return NULL;

}

args = PyTuple_New(2);

if (!args)

return NULL;

Py_INCREF(self); PyTuple_SET_ITEM(args, 0, (PyObject*) self);

Py_INCREF(tag); PyTuple_SET_ITEM(args, 1, (PyObject*) tag);

result = PyObject_CallObject(elementtree_getiterator_obj, args);

Py_DECREF(args);

return result;

}

static PyObject*

element_getitem(PyObject* self_, Py_ssize_t index)

{

ElementObject* self = (ElementObject*) self_;

if (!self->extra || index < 0 || index >= self->extra->length) {

PyErr_SetString(

PyExc_IndexError,

"child index out of range"

);

return NULL;

}

Py_INCREF(self->extra->children[index]);

return self->extra->children[index];

}

static PyObject*

element_getslice(PyObject* self_, Py_ssize_t start, Py_ssize_t end)

{

ElementObject* self = (ElementObject*) self_;

Py_ssize_t i;

PyObject* list;

if (!self->extra)

return PyList_New(0);

/* standard clamping */

if (start < 0)

start = 0;

if (end < 0)

end = 0;

if (end > self->extra->length)

end = self->extra->length;

if (start > end)

start = end;

list = PyList_New(end - start);

if (!list)

return NULL;

for (i = start; i < end; i++) {

PyObject* item = self->extra->children[i];

Py_INCREF(item);

PyList_SET_ITEM(list, i - start, item);

}

return list;

}

static PyObject*

element_insert(ElementObject* self, PyObject* args)

{

int i;

int index;

PyObject* element;

if (!PyArg_ParseTuple(args, "iO!:insert", &index,

&Element_Type, &element))

return NULL;

if (!self->extra)

element_new_extra(self, NULL);

if (index < 0)

index = 0;

if (index > self->extra->length)

index = self->extra->length;

if (element_resize(self, 1) < 0)

return NULL;

for (i = self->extra->length; i > index; i--)

self->extra->children[i] = self->extra->children[i-1];

Py_INCREF(element);

self->extra->children[index] = element;