1 /* util.c - various utility functions
2 *
3 * Copyright (C) 2005-2010 Gerhard H�ring <gh@ghaering.de>
4 *
5 * This file is part of pysqlite.
6 *
7 * This software is provided 'as-is', without any express or implied
8 * warranty. In no event will the authors be held liable for any damages
9 * arising from the use of this software.
10 *
11 * Permission is granted to anyone to use this software for any purpose,
12 * including commercial applications, and to alter it and redistribute it
13 * freely, subject to the following restrictions:
14 *
15 * 1. The origin of this software must not be misrepresented; you must not
16 * claim that you wrote the original software. If you use this software
17 * in a product, an acknowledgment in the product documentation would be
18 * appreciated but is not required.
19 * 2. Altered source versions must be plainly marked as such, and must not be
20 * misrepresented as being the original software.
21 * 3. This notice may not be removed or altered from any source distribution.
22 */
23
24 #include "module.h"
25 #include "connection.h"
26
pysqlite_step(sqlite3_stmt * statement,pysqlite_Connection * connection)27 int pysqlite_step(sqlite3_stmt* statement, pysqlite_Connection* connection)
28 {
29 int rc;
30
31 if (statement == NULL) {
32 /* this is a workaround for SQLite 3.5 and later. it now apparently
33 * returns NULL for "no-operation" statements */
34 rc = SQLITE_OK;
35 } else {
36 Py_BEGIN_ALLOW_THREADS
37 rc = sqlite3_step(statement);
38 Py_END_ALLOW_THREADS
39 }
40
41 return rc;
42 }
43
44 /**
45 * Checks the SQLite error code and sets the appropriate DB-API exception.
46 * Returns the error code (0 means no error occurred).
47 */
_pysqlite_seterror(sqlite3 * db,sqlite3_stmt * st)48 int _pysqlite_seterror(sqlite3* db, sqlite3_stmt* st)
49 {
50 int errorcode;
51
52 /* SQLite often doesn't report anything useful, unless you reset the statement first */
53 if (st != NULL) {
54 (void)sqlite3_reset(st);
55 }
56
57 errorcode = sqlite3_errcode(db);
58
59 switch (errorcode)
60 {
61 case SQLITE_OK:
62 PyErr_Clear();
63 break;
64 case SQLITE_INTERNAL:
65 case SQLITE_NOTFOUND:
66 PyErr_SetString(pysqlite_InternalError, sqlite3_errmsg(db));
67 break;
68 case SQLITE_NOMEM:
69 (void)PyErr_NoMemory();
70 break;
71 case SQLITE_ERROR:
72 case SQLITE_PERM:
73 case SQLITE_ABORT:
74 case SQLITE_BUSY:
75 case SQLITE_LOCKED:
76 case SQLITE_READONLY:
77 case SQLITE_INTERRUPT:
78 case SQLITE_IOERR:
79 case SQLITE_FULL:
80 case SQLITE_CANTOPEN:
81 case SQLITE_PROTOCOL:
82 case SQLITE_EMPTY:
83 case SQLITE_SCHEMA:
84 PyErr_SetString(pysqlite_OperationalError, sqlite3_errmsg(db));
85 break;
86 case SQLITE_CORRUPT:
87 PyErr_SetString(pysqlite_DatabaseError, sqlite3_errmsg(db));
88 break;
89 case SQLITE_TOOBIG:
90 PyErr_SetString(pysqlite_DataError, sqlite3_errmsg(db));
91 break;
92 case SQLITE_CONSTRAINT:
93 case SQLITE_MISMATCH:
94 PyErr_SetString(pysqlite_IntegrityError, sqlite3_errmsg(db));
95 break;
96 case SQLITE_MISUSE:
97 PyErr_SetString(pysqlite_ProgrammingError, sqlite3_errmsg(db));
98 break;
99 default:
100 PyErr_SetString(pysqlite_DatabaseError, sqlite3_errmsg(db));
101 break;
102 }
103
104 return errorcode;
105 }
106
107 #ifdef WORDS_BIGENDIAN
108 # define IS_LITTLE_ENDIAN 0
109 #else
110 # define IS_LITTLE_ENDIAN 1
111 #endif
112
113 PyObject *
_pysqlite_long_from_int64(sqlite_int64 value)114 _pysqlite_long_from_int64(sqlite_int64 value)
115 {
116 #ifdef HAVE_LONG_LONG
117 # if SIZEOF_LONG_LONG < 8
118 if (value > PY_LLONG_MAX || value < PY_LLONG_MIN) {
119 return _PyLong_FromByteArray(&value, sizeof(value),
120 IS_LITTLE_ENDIAN, 1 /* signed */);
121 }
122 # endif
123 # if SIZEOF_LONG < SIZEOF_LONG_LONG
124 if (value > LONG_MAX || value < LONG_MIN)
125 return PyLong_FromLongLong(value);
126 # endif
127 #else
128 # if SIZEOF_LONG < 8
129 if (value > LONG_MAX || value < LONG_MIN) {
130 return _PyLong_FromByteArray(&value, sizeof(value),
131 IS_LITTLE_ENDIAN, 1 /* signed */);
132 }
133 # endif
134 #endif
135 return PyInt_FromLong(Py_SAFE_DOWNCAST(value, sqlite_int64, long));
136 }
137
138 sqlite_int64
_pysqlite_long_as_int64(PyObject * py_val)139 _pysqlite_long_as_int64(PyObject * py_val)
140 {
141 int overflow;
142 #ifdef HAVE_LONG_LONG
143 PY_LONG_LONG value = PyLong_AsLongLongAndOverflow(py_val, &overflow);
144 #else
145 long value = PyLong_AsLongAndOverflow(py_val, &overflow);
146 #endif
147 if (value == -1 && PyErr_Occurred())
148 return -1;
149 if (!overflow) {
150 #ifdef HAVE_LONG_LONG
151 # if SIZEOF_LONG_LONG > 8
152 if (-0x8000000000000000LL <= value && value <= 0x7FFFFFFFFFFFFFFFLL)
153 # endif
154 #else
155 # if SIZEOF_LONG > 8
156 if (-0x8000000000000000L <= value && value <= 0x7FFFFFFFFFFFFFFFL)
157 # endif
158 #endif
159 return value;
160 }
161 else if (sizeof(value) < sizeof(sqlite_int64)) {
162 sqlite_int64 int64val;
163 if (_PyLong_AsByteArray((PyLongObject *)py_val,
164 (unsigned char *)&int64val, sizeof(int64val),
165 IS_LITTLE_ENDIAN, 1 /* signed */) >= 0) {
166 return int64val;
167 }
168 }
169 PyErr_SetString(PyExc_OverflowError,
170 "Python int too large to convert to SQLite INTEGER");
171 return -1;
172 }
173