+++ /dev/null
-/* statement.c - the statement type\r
- *\r
- * Copyright (C) 2005-2010 Gerhard Häring <gh@ghaering.de>\r
- *\r
- * This file is part of pysqlite.\r
- *\r
- * This software is provided 'as-is', without any express or implied\r
- * warranty. In no event will the authors be held liable for any damages\r
- * arising from the use of this software.\r
- *\r
- * Permission is granted to anyone to use this software for any purpose,\r
- * including commercial applications, and to alter it and redistribute it\r
- * freely, subject to the following restrictions:\r
- *\r
- * 1. The origin of this software must not be misrepresented; you must not\r
- * claim that you wrote the original software. If you use this software\r
- * in a product, an acknowledgment in the product documentation would be\r
- * appreciated but is not required.\r
- * 2. Altered source versions must be plainly marked as such, and must not be\r
- * misrepresented as being the original software.\r
- * 3. This notice may not be removed or altered from any source distribution.\r
- */\r
-\r
-#include "statement.h"\r
-#include "cursor.h"\r
-#include "connection.h"\r
-#include "microprotocols.h"\r
-#include "prepare_protocol.h"\r
-#include "sqlitecompat.h"\r
-\r
-/* prototypes */\r
-static int pysqlite_check_remaining_sql(const char* tail);\r
-\r
-typedef enum {\r
- LINECOMMENT_1,\r
- IN_LINECOMMENT,\r
- COMMENTSTART_1,\r
- IN_COMMENT,\r
- COMMENTEND_1,\r
- NORMAL\r
-} parse_remaining_sql_state;\r
-\r
-typedef enum {\r
- TYPE_INT,\r
- TYPE_LONG,\r
- TYPE_FLOAT,\r
- TYPE_STRING,\r
- TYPE_UNICODE,\r
- TYPE_BUFFER,\r
- TYPE_UNKNOWN\r
-} parameter_type;\r
-\r
-int pysqlite_statement_create(pysqlite_Statement* self, pysqlite_Connection* connection, PyObject* sql)\r
-{\r
- const char* tail;\r
- int rc;\r
- PyObject* sql_str;\r
- char* sql_cstr;\r
-\r
- self->st = NULL;\r
- self->in_use = 0;\r
-\r
- if (PyString_Check(sql)) {\r
- sql_str = sql;\r
- Py_INCREF(sql_str);\r
- } else if (PyUnicode_Check(sql)) {\r
- sql_str = PyUnicode_AsUTF8String(sql);\r
- if (!sql_str) {\r
- rc = PYSQLITE_SQL_WRONG_TYPE;\r
- return rc;\r
- }\r
- } else {\r
- rc = PYSQLITE_SQL_WRONG_TYPE;\r
- return rc;\r
- }\r
-\r
- self->in_weakreflist = NULL;\r
- self->sql = sql_str;\r
-\r
- sql_cstr = PyString_AsString(sql_str);\r
-\r
- Py_BEGIN_ALLOW_THREADS\r
- rc = sqlite3_prepare(connection->db,\r
- sql_cstr,\r
- -1,\r
- &self->st,\r
- &tail);\r
- Py_END_ALLOW_THREADS\r
-\r
- self->db = connection->db;\r
-\r
- if (rc == SQLITE_OK && pysqlite_check_remaining_sql(tail)) {\r
- (void)sqlite3_finalize(self->st);\r
- self->st = NULL;\r
- rc = PYSQLITE_TOO_MUCH_SQL;\r
- }\r
-\r
- return rc;\r
-}\r
-\r
-int pysqlite_statement_bind_parameter(pysqlite_Statement* self, int pos, PyObject* parameter, int allow_8bit_chars)\r
-{\r
- int rc = SQLITE_OK;\r
- long longval;\r
- PY_LONG_LONG longlongval;\r
- const char* buffer;\r
- char* string;\r
- Py_ssize_t buflen;\r
- PyObject* stringval;\r
- parameter_type paramtype;\r
- char* c;\r
-\r
- if (parameter == Py_None) {\r
- rc = sqlite3_bind_null(self->st, pos);\r
- goto final;\r
- }\r
-\r
- if (PyInt_CheckExact(parameter)) {\r
- paramtype = TYPE_INT;\r
- } else if (PyLong_CheckExact(parameter)) {\r
- paramtype = TYPE_LONG;\r
- } else if (PyFloat_CheckExact(parameter)) {\r
- paramtype = TYPE_FLOAT;\r
- } else if (PyString_CheckExact(parameter)) {\r
- paramtype = TYPE_STRING;\r
- } else if (PyUnicode_CheckExact(parameter)) {\r
- paramtype = TYPE_UNICODE;\r
- } else if (PyBuffer_Check(parameter)) {\r
- paramtype = TYPE_BUFFER;\r
- } else if (PyInt_Check(parameter)) {\r
- paramtype = TYPE_INT;\r
- } else if (PyLong_Check(parameter)) {\r
- paramtype = TYPE_LONG;\r
- } else if (PyFloat_Check(parameter)) {\r
- paramtype = TYPE_FLOAT;\r
- } else if (PyString_Check(parameter)) {\r
- paramtype = TYPE_STRING;\r
- } else if (PyUnicode_Check(parameter)) {\r
- paramtype = TYPE_UNICODE;\r
- } else {\r
- paramtype = TYPE_UNKNOWN;\r
- }\r
-\r
- if (paramtype == TYPE_STRING && !allow_8bit_chars) {\r
- string = PyString_AS_STRING(parameter);\r
- for (c = string; *c != 0; c++) {\r
- if (*c & 0x80) {\r
- PyErr_SetString(pysqlite_ProgrammingError, "You must not use 8-bit bytestrings unless you use a text_factory that can interpret 8-bit bytestrings (like text_factory = str). It is highly recommended that you instead just switch your application to Unicode strings.");\r
- rc = -1;\r
- goto final;\r
- }\r
- }\r
- }\r
-\r
- switch (paramtype) {\r
- case TYPE_INT:\r
- longval = PyInt_AsLong(parameter);\r
- rc = sqlite3_bind_int64(self->st, pos, (sqlite_int64)longval);\r
- break;\r
- case TYPE_LONG:\r
- longlongval = PyLong_AsLongLong(parameter);\r
- /* in the overflow error case, longlongval is -1, and an exception is set */\r
- rc = sqlite3_bind_int64(self->st, pos, (sqlite_int64)longlongval);\r
- break;\r
- case TYPE_FLOAT:\r
- rc = sqlite3_bind_double(self->st, pos, PyFloat_AsDouble(parameter));\r
- break;\r
- case TYPE_STRING:\r
- string = PyString_AS_STRING(parameter);\r
- rc = sqlite3_bind_text(self->st, pos, string, -1, SQLITE_TRANSIENT);\r
- break;\r
- case TYPE_UNICODE:\r
- stringval = PyUnicode_AsUTF8String(parameter);\r
- string = PyString_AsString(stringval);\r
- rc = sqlite3_bind_text(self->st, pos, string, -1, SQLITE_TRANSIENT);\r
- Py_DECREF(stringval);\r
- break;\r
- case TYPE_BUFFER:\r
- if (PyObject_AsCharBuffer(parameter, &buffer, &buflen) == 0) {\r
- rc = sqlite3_bind_blob(self->st, pos, buffer, buflen, SQLITE_TRANSIENT);\r
- } else {\r
- PyErr_SetString(PyExc_ValueError, "could not convert BLOB to buffer");\r
- rc = -1;\r
- }\r
- break;\r
- case TYPE_UNKNOWN:\r
- rc = -1;\r
- }\r
-\r
-final:\r
- return rc;\r
-}\r
-\r
-/* returns 0 if the object is one of Python's internal ones that don't need to be adapted */\r
-static int _need_adapt(PyObject* obj)\r
-{\r
- if (pysqlite_BaseTypeAdapted) {\r
- return 1;\r
- }\r
-\r
- if (PyInt_CheckExact(obj) || PyLong_CheckExact(obj) \r
- || PyFloat_CheckExact(obj) || PyString_CheckExact(obj)\r
- || PyUnicode_CheckExact(obj) || PyBuffer_Check(obj)) {\r
- return 0;\r
- } else {\r
- return 1;\r
- }\r
-}\r
-\r
-void pysqlite_statement_bind_parameters(pysqlite_Statement* self, PyObject* parameters, int allow_8bit_chars)\r
-{\r
- PyObject* current_param;\r
- PyObject* adapted;\r
- const char* binding_name;\r
- int i;\r
- int rc;\r
- int num_params_needed;\r
- int num_params;\r
-\r
- Py_BEGIN_ALLOW_THREADS\r
- num_params_needed = sqlite3_bind_parameter_count(self->st);\r
- Py_END_ALLOW_THREADS\r
-\r
- if (PyTuple_CheckExact(parameters) || PyList_CheckExact(parameters) || (!PyDict_Check(parameters) && PySequence_Check(parameters))) {\r
- /* parameters passed as sequence */\r
- if (PyTuple_CheckExact(parameters)) {\r
- num_params = PyTuple_GET_SIZE(parameters);\r
- } else if (PyList_CheckExact(parameters)) {\r
- num_params = PyList_GET_SIZE(parameters);\r
- } else {\r
- num_params = PySequence_Size(parameters);\r
- }\r
- if (num_params != num_params_needed) {\r
- PyErr_Format(pysqlite_ProgrammingError, "Incorrect number of bindings supplied. The current statement uses %d, and there are %d supplied.",\r
- num_params_needed, num_params);\r
- return;\r
- }\r
- for (i = 0; i < num_params; i++) {\r
- if (PyTuple_CheckExact(parameters)) {\r
- current_param = PyTuple_GET_ITEM(parameters, i);\r
- Py_XINCREF(current_param);\r
- } else if (PyList_CheckExact(parameters)) {\r
- current_param = PyList_GET_ITEM(parameters, i);\r
- Py_XINCREF(current_param);\r
- } else {\r
- current_param = PySequence_GetItem(parameters, i);\r
- }\r
- if (!current_param) {\r
- return;\r
- }\r
-\r
- if (!_need_adapt(current_param)) {\r
- adapted = current_param;\r
- } else {\r
- adapted = pysqlite_microprotocols_adapt(current_param, (PyObject*)&pysqlite_PrepareProtocolType, NULL);\r
- if (adapted) {\r
- Py_DECREF(current_param);\r
- } else {\r
- PyErr_Clear();\r
- adapted = current_param;\r
- }\r
- }\r
-\r
- rc = pysqlite_statement_bind_parameter(self, i + 1, adapted, allow_8bit_chars);\r
- Py_DECREF(adapted);\r
-\r
- if (rc != SQLITE_OK) {\r
- if (!PyErr_Occurred()) {\r
- PyErr_Format(pysqlite_InterfaceError, "Error binding parameter %d - probably unsupported type.", i);\r
- }\r
- return;\r
- }\r
- }\r
- } else if (PyDict_Check(parameters)) {\r
- /* parameters passed as dictionary */\r
- for (i = 1; i <= num_params_needed; i++) {\r
- Py_BEGIN_ALLOW_THREADS\r
- binding_name = sqlite3_bind_parameter_name(self->st, i);\r
- Py_END_ALLOW_THREADS\r
- if (!binding_name) {\r
- PyErr_Format(pysqlite_ProgrammingError, "Binding %d has no name, but you supplied a dictionary (which has only names).", i);\r
- return;\r
- }\r
-\r
- binding_name++; /* skip first char (the colon) */\r
- if (PyDict_CheckExact(parameters)) {\r
- current_param = PyDict_GetItemString(parameters, binding_name);\r
- Py_XINCREF(current_param);\r
- } else {\r
- current_param = PyMapping_GetItemString(parameters, (char*)binding_name);\r
- }\r
- if (!current_param) {\r
- PyErr_Format(pysqlite_ProgrammingError, "You did not supply a value for binding %d.", i);\r
- return;\r
- }\r
-\r
- if (!_need_adapt(current_param)) {\r
- adapted = current_param;\r
- } else {\r
- adapted = pysqlite_microprotocols_adapt(current_param, (PyObject*)&pysqlite_PrepareProtocolType, NULL);\r
- if (adapted) {\r
- Py_DECREF(current_param);\r
- } else {\r
- PyErr_Clear();\r
- adapted = current_param;\r
- }\r
- }\r
-\r
- rc = pysqlite_statement_bind_parameter(self, i, adapted, allow_8bit_chars);\r
- Py_DECREF(adapted);\r
-\r
- if (rc != SQLITE_OK) {\r
- if (!PyErr_Occurred()) {\r
- PyErr_Format(pysqlite_InterfaceError, "Error binding parameter :%s - probably unsupported type.", binding_name);\r
- }\r
- return;\r
- }\r
- }\r
- } else {\r
- PyErr_SetString(PyExc_ValueError, "parameters are of unsupported type");\r
- }\r
-}\r
-\r
-int pysqlite_statement_recompile(pysqlite_Statement* self, PyObject* params)\r
-{\r
- const char* tail;\r
- int rc;\r
- char* sql_cstr;\r
- sqlite3_stmt* new_st;\r
-\r
- sql_cstr = PyString_AsString(self->sql);\r
-\r
- Py_BEGIN_ALLOW_THREADS\r
- rc = sqlite3_prepare(self->db,\r
- sql_cstr,\r
- -1,\r
- &new_st,\r
- &tail);\r
- Py_END_ALLOW_THREADS\r
-\r
- if (rc == SQLITE_OK) {\r
- /* The efficient sqlite3_transfer_bindings is only available in SQLite\r
- * version 3.2.2 or later. For older SQLite releases, that might not\r
- * even define SQLITE_VERSION_NUMBER, we do it the manual way.\r
- */\r
- #ifdef SQLITE_VERSION_NUMBER\r
- #if SQLITE_VERSION_NUMBER >= 3002002\r
- /* The check for the number of parameters is necessary to not trigger a\r
- * bug in certain SQLite versions (experienced in 3.2.8 and 3.3.4). */\r
- if (sqlite3_bind_parameter_count(self->st) > 0) {\r
- (void)sqlite3_transfer_bindings(self->st, new_st);\r
- }\r
- #endif\r
- #else\r
- statement_bind_parameters(self, params);\r
- #endif\r
-\r
- (void)sqlite3_finalize(self->st);\r
- self->st = new_st;\r
- }\r
-\r
- return rc;\r
-}\r
-\r
-int pysqlite_statement_finalize(pysqlite_Statement* self)\r
-{\r
- int rc;\r
-\r
- rc = SQLITE_OK;\r
- if (self->st) {\r
- Py_BEGIN_ALLOW_THREADS\r
- rc = sqlite3_finalize(self->st);\r
- Py_END_ALLOW_THREADS\r
- self->st = NULL;\r
- }\r
-\r
- self->in_use = 0;\r
-\r
- return rc;\r
-}\r
-\r
-int pysqlite_statement_reset(pysqlite_Statement* self)\r
-{\r
- int rc;\r
-\r
- rc = SQLITE_OK;\r
-\r
- if (self->in_use && self->st) {\r
- Py_BEGIN_ALLOW_THREADS\r
- rc = sqlite3_reset(self->st);\r
- Py_END_ALLOW_THREADS\r
-\r
- if (rc == SQLITE_OK) {\r
- self->in_use = 0;\r
- }\r
- }\r
-\r
- return rc;\r
-}\r
-\r
-void pysqlite_statement_mark_dirty(pysqlite_Statement* self)\r
-{\r
- self->in_use = 1;\r
-}\r
-\r
-void pysqlite_statement_dealloc(pysqlite_Statement* self)\r
-{\r
- int rc;\r
-\r
- if (self->st) {\r
- Py_BEGIN_ALLOW_THREADS\r
- rc = sqlite3_finalize(self->st);\r
- Py_END_ALLOW_THREADS\r
- }\r
-\r
- self->st = NULL;\r
-\r
- Py_XDECREF(self->sql);\r
-\r
- if (self->in_weakreflist != NULL) {\r
- PyObject_ClearWeakRefs((PyObject*)self);\r
- }\r
-\r
- Py_TYPE(self)->tp_free((PyObject*)self);\r
-}\r
-\r
-/*\r
- * Checks if there is anything left in an SQL string after SQLite compiled it.\r
- * This is used to check if somebody tried to execute more than one SQL command\r
- * with one execute()/executemany() command, which the DB-API and we don't\r
- * allow.\r
- *\r
- * Returns 1 if there is more left than should be. 0 if ok.\r
- */\r
-static int pysqlite_check_remaining_sql(const char* tail)\r
-{\r
- const char* pos = tail;\r
-\r
- parse_remaining_sql_state state = NORMAL;\r
-\r
- for (;;) {\r
- switch (*pos) {\r
- case 0:\r
- return 0;\r
- case '-':\r
- if (state == NORMAL) {\r
- state = LINECOMMENT_1;\r
- } else if (state == LINECOMMENT_1) {\r
- state = IN_LINECOMMENT;\r
- }\r
- break;\r
- case ' ':\r
- case '\t':\r
- break;\r
- case '\n':\r
- case 13:\r
- if (state == IN_LINECOMMENT) {\r
- state = NORMAL;\r
- }\r
- break;\r
- case '/':\r
- if (state == NORMAL) {\r
- state = COMMENTSTART_1;\r
- } else if (state == COMMENTEND_1) {\r
- state = NORMAL;\r
- } else if (state == COMMENTSTART_1) {\r
- return 1;\r
- }\r
- break;\r
- case '*':\r
- if (state == NORMAL) {\r
- return 1;\r
- } else if (state == LINECOMMENT_1) {\r
- return 1;\r
- } else if (state == COMMENTSTART_1) {\r
- state = IN_COMMENT;\r
- } else if (state == IN_COMMENT) {\r
- state = COMMENTEND_1;\r
- }\r
- break;\r
- default:\r
- if (state == COMMENTEND_1) {\r
- state = IN_COMMENT;\r
- } else if (state == IN_LINECOMMENT) {\r
- } else if (state == IN_COMMENT) {\r
- } else {\r
- return 1;\r
- }\r
- }\r
-\r
- pos++;\r
- }\r
-\r
- return 0;\r
-}\r
-\r
-PyTypeObject pysqlite_StatementType = {\r
- PyVarObject_HEAD_INIT(NULL, 0)\r
- MODULE_NAME ".Statement", /* tp_name */\r
- sizeof(pysqlite_Statement), /* tp_basicsize */\r
- 0, /* tp_itemsize */\r
- (destructor)pysqlite_statement_dealloc, /* tp_dealloc */\r
- 0, /* tp_print */\r
- 0, /* tp_getattr */\r
- 0, /* tp_setattr */\r
- 0, /* tp_compare */\r
- 0, /* tp_repr */\r
- 0, /* tp_as_number */\r
- 0, /* tp_as_sequence */\r
- 0, /* tp_as_mapping */\r
- 0, /* tp_hash */\r
- 0, /* tp_call */\r
- 0, /* tp_str */\r
- 0, /* tp_getattro */\r
- 0, /* tp_setattro */\r
- 0, /* tp_as_buffer */\r
- Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_WEAKREFS, /* tp_flags */\r
- 0, /* tp_doc */\r
- 0, /* tp_traverse */\r
- 0, /* tp_clear */\r
- 0, /* tp_richcompare */\r
- offsetof(pysqlite_Statement, in_weakreflist), /* tp_weaklistoffset */\r
- 0, /* tp_iter */\r
- 0, /* tp_iternext */\r
- 0, /* tp_methods */\r
- 0, /* tp_members */\r
- 0, /* tp_getset */\r
- 0, /* tp_base */\r
- 0, /* tp_dict */\r
- 0, /* tp_descr_get */\r
- 0, /* tp_descr_set */\r
- 0, /* tp_dictoffset */\r
- (initproc)0, /* tp_init */\r
- 0, /* tp_alloc */\r
- 0, /* tp_new */\r
- 0 /* tp_free */\r
-};\r
-\r
-extern int pysqlite_statement_setup_types(void)\r
-{\r
- pysqlite_StatementType.tp_new = PyType_GenericNew;\r
- return PyType_Ready(&pysqlite_StatementType);\r
-}\r