Index: third_party/markupsafe/_speedups.c |
diff --git a/third_party/markupsafe/_speedups.c b/third_party/markupsafe/_speedups.c |
new file mode 100644 |
index 0000000000000000000000000000000000000000..f349febf22d59ec7dfe440b65faf5838c1234b90 |
--- /dev/null |
+++ b/third_party/markupsafe/_speedups.c |
@@ -0,0 +1,239 @@ |
+/** |
+ * markupsafe._speedups |
+ * ~~~~~~~~~~~~~~~~~~~~ |
+ * |
+ * This module implements functions for automatic escaping in C for better |
+ * performance. |
+ * |
+ * :copyright: (c) 2010 by Armin Ronacher. |
+ * :license: BSD. |
+ */ |
+ |
+#include <Python.h> |
+ |
+#define ESCAPED_CHARS_TABLE_SIZE 63 |
+#define UNICHR(x) (PyUnicode_AS_UNICODE((PyUnicodeObject*)PyUnicode_DecodeASCII(x, strlen(x), NULL))); |
+ |
+#if PY_VERSION_HEX < 0x02050000 && !defined(PY_SSIZE_T_MIN) |
+typedef int Py_ssize_t; |
+#define PY_SSIZE_T_MAX INT_MAX |
+#define PY_SSIZE_T_MIN INT_MIN |
+#endif |
+ |
+ |
+static PyObject* markup; |
+static Py_ssize_t escaped_chars_delta_len[ESCAPED_CHARS_TABLE_SIZE]; |
+static Py_UNICODE *escaped_chars_repl[ESCAPED_CHARS_TABLE_SIZE]; |
+ |
+static int |
+init_constants(void) |
+{ |
+ PyObject *module; |
+ /* happing of characters to replace */ |
+ escaped_chars_repl['"'] = UNICHR("""); |
+ escaped_chars_repl['\''] = UNICHR("'"); |
+ escaped_chars_repl['&'] = UNICHR("&"); |
+ escaped_chars_repl['<'] = UNICHR("<"); |
+ escaped_chars_repl['>'] = UNICHR(">"); |
+ |
+ /* lengths of those characters when replaced - 1 */ |
+ memset(escaped_chars_delta_len, 0, sizeof (escaped_chars_delta_len)); |
+ escaped_chars_delta_len['"'] = escaped_chars_delta_len['\''] = \ |
+ escaped_chars_delta_len['&'] = 4; |
+ escaped_chars_delta_len['<'] = escaped_chars_delta_len['>'] = 3; |
+ |
+ /* import markup type so that we can mark the return value */ |
+ module = PyImport_ImportModule("markupsafe"); |
+ if (!module) |
+ return 0; |
+ markup = PyObject_GetAttrString(module, "Markup"); |
+ Py_DECREF(module); |
+ |
+ return 1; |
+} |
+ |
+static PyObject* |
+escape_unicode(PyUnicodeObject *in) |
+{ |
+ PyUnicodeObject *out; |
+ Py_UNICODE *inp = PyUnicode_AS_UNICODE(in); |
+ const Py_UNICODE *inp_end = PyUnicode_AS_UNICODE(in) + PyUnicode_GET_SIZE(in); |
+ Py_UNICODE *next_escp; |
+ Py_UNICODE *outp; |
+ Py_ssize_t delta=0, erepl=0, delta_len=0; |
+ |
+ /* First we need to figure out how long the escaped string will be */ |
+ while (*(inp) || inp < inp_end) { |
+ if (*inp < ESCAPED_CHARS_TABLE_SIZE) { |
+ delta += escaped_chars_delta_len[*inp]; |
+ erepl += !!escaped_chars_delta_len[*inp]; |
+ } |
+ ++inp; |
+ } |
+ |
+ /* Do we need to escape anything at all? */ |
+ if (!erepl) { |
+ Py_INCREF(in); |
+ return (PyObject*)in; |
+ } |
+ |
+ out = (PyUnicodeObject*)PyUnicode_FromUnicode(NULL, PyUnicode_GET_SIZE(in) + delta); |
+ if (!out) |
+ return NULL; |
+ |
+ outp = PyUnicode_AS_UNICODE(out); |
+ inp = PyUnicode_AS_UNICODE(in); |
+ while (erepl-- > 0) { |
+ /* look for the next substitution */ |
+ next_escp = inp; |
+ while (next_escp < inp_end) { |
+ if (*next_escp < ESCAPED_CHARS_TABLE_SIZE && |
+ (delta_len = escaped_chars_delta_len[*next_escp])) { |
+ ++delta_len; |
+ break; |
+ } |
+ ++next_escp; |
+ } |
+ |
+ if (next_escp > inp) { |
+ /* copy unescaped chars between inp and next_escp */ |
+ Py_UNICODE_COPY(outp, inp, next_escp-inp); |
+ outp += next_escp - inp; |
+ } |
+ |
+ /* escape 'next_escp' */ |
+ Py_UNICODE_COPY(outp, escaped_chars_repl[*next_escp], delta_len); |
+ outp += delta_len; |
+ |
+ inp = next_escp + 1; |
+ } |
+ if (inp < inp_end) |
+ Py_UNICODE_COPY(outp, inp, PyUnicode_GET_SIZE(in) - (inp - PyUnicode_AS_UNICODE(in))); |
+ |
+ return (PyObject*)out; |
+} |
+ |
+ |
+static PyObject* |
+escape(PyObject *self, PyObject *text) |
+{ |
+ PyObject *s = NULL, *rv = NULL, *html; |
+ |
+ /* we don't have to escape integers, bools or floats */ |
+ if (PyLong_CheckExact(text) || |
+#if PY_MAJOR_VERSION < 3 |
+ PyInt_CheckExact(text) || |
+#endif |
+ PyFloat_CheckExact(text) || PyBool_Check(text) || |
+ text == Py_None) |
+ return PyObject_CallFunctionObjArgs(markup, text, NULL); |
+ |
+ /* if the object has an __html__ method that performs the escaping */ |
+ html = PyObject_GetAttrString(text, "__html__"); |
+ if (html) { |
+ rv = PyObject_CallObject(html, NULL); |
+ Py_DECREF(html); |
+ return rv; |
+ } |
+ |
+ /* otherwise make the object unicode if it isn't, then escape */ |
+ PyErr_Clear(); |
+ if (!PyUnicode_Check(text)) { |
+#if PY_MAJOR_VERSION < 3 |
+ PyObject *unicode = PyObject_Unicode(text); |
+#else |
+ PyObject *unicode = PyObject_Str(text); |
+#endif |
+ if (!unicode) |
+ return NULL; |
+ s = escape_unicode((PyUnicodeObject*)unicode); |
+ Py_DECREF(unicode); |
+ } |
+ else |
+ s = escape_unicode((PyUnicodeObject*)text); |
+ |
+ /* convert the unicode string into a markup object. */ |
+ rv = PyObject_CallFunctionObjArgs(markup, (PyObject*)s, NULL); |
+ Py_DECREF(s); |
+ return rv; |
+} |
+ |
+ |
+static PyObject* |
+escape_silent(PyObject *self, PyObject *text) |
+{ |
+ if (text != Py_None) |
+ return escape(self, text); |
+ return PyObject_CallFunctionObjArgs(markup, NULL); |
+} |
+ |
+ |
+static PyObject* |
+soft_unicode(PyObject *self, PyObject *s) |
+{ |
+ if (!PyUnicode_Check(s)) |
+#if PY_MAJOR_VERSION < 3 |
+ return PyObject_Unicode(s); |
+#else |
+ return PyObject_Str(s); |
+#endif |
+ Py_INCREF(s); |
+ return s; |
+} |
+ |
+ |
+static PyMethodDef module_methods[] = { |
+ {"escape", (PyCFunction)escape, METH_O, |
+ "escape(s) -> markup\n\n" |
+ "Convert the characters &, <, >, ', and \" in string s to HTML-safe\n" |
+ "sequences. Use this if you need to display text that might contain\n" |
+ "such characters in HTML. Marks return value as markup string."}, |
+ {"escape_silent", (PyCFunction)escape_silent, METH_O, |
+ "escape_silent(s) -> markup\n\n" |
+ "Like escape but converts None to an empty string."}, |
+ {"soft_unicode", (PyCFunction)soft_unicode, METH_O, |
+ "soft_unicode(object) -> string\n\n" |
+ "Make a string unicode if it isn't already. That way a markup\n" |
+ "string is not converted back to unicode."}, |
+ {NULL, NULL, 0, NULL} /* Sentinel */ |
+}; |
+ |
+ |
+#if PY_MAJOR_VERSION < 3 |
+ |
+#ifndef PyMODINIT_FUNC /* declarations for DLL import/export */ |
+#define PyMODINIT_FUNC void |
+#endif |
+PyMODINIT_FUNC |
+init_speedups(void) |
+{ |
+ if (!init_constants()) |
+ return; |
+ |
+ Py_InitModule3("markupsafe._speedups", module_methods, ""); |
+} |
+ |
+#else /* Python 3.x module initialization */ |
+ |
+static struct PyModuleDef module_definition = { |
+ PyModuleDef_HEAD_INIT, |
+ "markupsafe._speedups", |
+ NULL, |
+ -1, |
+ module_methods, |
+ NULL, |
+ NULL, |
+ NULL, |
+ NULL |
+}; |
+ |
+PyMODINIT_FUNC |
+PyInit__speedups(void) |
+{ |
+ if (!init_constants()) |
+ return NULL; |
+ |
+ return PyModule_Create(&module_definition); |
+} |
+ |
+#endif |