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 sqlite_int64
_pysqlite_long_as_int64(PyObject * py_val)107 _pysqlite_long_as_int64(PyObject * py_val)
108 {
109 int overflow;
110 long long value = PyLong_AsLongLongAndOverflow(py_val, &overflow);
111 if (value == -1 && PyErr_Occurred())
112 return -1;
113 if (!overflow) {
114 # if SIZEOF_LONG_LONG > 8
115 if (-0x8000000000000000LL <= value && value <= 0x7FFFFFFFFFFFFFFFLL)
116 # endif
117 return value;
118 }
119 else if (sizeof(value) < sizeof(sqlite_int64)) {
120 sqlite_int64 int64val;
121 if (_PyLong_AsByteArray((PyLongObject *)py_val,
122 (unsigned char *)&int64val, sizeof(int64val),
123 IS_LITTLE_ENDIAN, 1 /* signed */) >= 0) {
124 return int64val;
125 }
126 }
127 PyErr_SetString(PyExc_OverflowError,
128 "Python int too large to convert to SQLite INTEGER");
129 return -1;
130 }
131