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 = sqlite3_errcode(db);
51
52 switch (errorcode)
53 {
54 case SQLITE_OK:
55 PyErr_Clear();
56 break;
57 case SQLITE_INTERNAL:
58 case SQLITE_NOTFOUND:
59 PyErr_SetString(pysqlite_InternalError, sqlite3_errmsg(db));
60 break;
61 case SQLITE_NOMEM:
62 (void)PyErr_NoMemory();
63 break;
64 case SQLITE_ERROR:
65 case SQLITE_PERM:
66 case SQLITE_ABORT:
67 case SQLITE_BUSY:
68 case SQLITE_LOCKED:
69 case SQLITE_READONLY:
70 case SQLITE_INTERRUPT:
71 case SQLITE_IOERR:
72 case SQLITE_FULL:
73 case SQLITE_CANTOPEN:
74 case SQLITE_PROTOCOL:
75 case SQLITE_EMPTY:
76 case SQLITE_SCHEMA:
77 PyErr_SetString(pysqlite_OperationalError, sqlite3_errmsg(db));
78 break;
79 case SQLITE_CORRUPT:
80 PyErr_SetString(pysqlite_DatabaseError, sqlite3_errmsg(db));
81 break;
82 case SQLITE_TOOBIG:
83 PyErr_SetString(pysqlite_DataError, sqlite3_errmsg(db));
84 break;
85 case SQLITE_CONSTRAINT:
86 case SQLITE_MISMATCH:
87 PyErr_SetString(pysqlite_IntegrityError, sqlite3_errmsg(db));
88 break;
89 case SQLITE_MISUSE:
90 PyErr_SetString(pysqlite_ProgrammingError, sqlite3_errmsg(db));
91 break;
92 default:
93 PyErr_SetString(pysqlite_DatabaseError, sqlite3_errmsg(db));
94 break;
95 }
96
97 return errorcode;
98 }
99
100 #ifdef WORDS_BIGENDIAN
101 # define IS_LITTLE_ENDIAN 0
102 #else
103 # define IS_LITTLE_ENDIAN 1
104 #endif
105
106 PyObject *
_pysqlite_long_from_int64(sqlite_int64 value)107 _pysqlite_long_from_int64(sqlite_int64 value)
108 {
109 # if SIZEOF_LONG_LONG < 8
110 if (value > PY_LLONG_MAX || value < PY_LLONG_MIN) {
111 return _PyLong_FromByteArray(&value, sizeof(value),
112 IS_LITTLE_ENDIAN, 1 /* signed */);
113 }
114 # endif
115 # if SIZEOF_LONG < SIZEOF_LONG_LONG
116 if (value > LONG_MAX || value < LONG_MIN)
117 return PyLong_FromLongLong(value);
118 # endif
119 return PyLong_FromLong(Py_SAFE_DOWNCAST(value, sqlite_int64, long));
120 }
121
122 sqlite_int64
_pysqlite_long_as_int64(PyObject * py_val)123 _pysqlite_long_as_int64(PyObject * py_val)
124 {
125 int overflow;
126 long long value = PyLong_AsLongLongAndOverflow(py_val, &overflow);
127 if (value == -1 && PyErr_Occurred())
128 return -1;
129 if (!overflow) {
130 # if SIZEOF_LONG_LONG > 8
131 if (-0x8000000000000000LL <= value && value <= 0x7FFFFFFFFFFFFFFFLL)
132 # endif
133 return value;
134 }
135 else if (sizeof(value) < sizeof(sqlite_int64)) {
136 sqlite_int64 int64val;
137 if (_PyLong_AsByteArray((PyLongObject *)py_val,
138 (unsigned char *)&int64val, sizeof(int64val),
139 IS_LITTLE_ENDIAN, 1 /* signed */) >= 0) {
140 return int64val;
141 }
142 }
143 PyErr_SetString(PyExc_OverflowError,
144 "Python int too large to convert to SQLite INTEGER");
145 return -1;
146 }
147