blob: c5d50bd50ba53e748a3f71815e13e54ae24dc7a0 [file] [log] [blame]
drh75897232000-05-29 14:26:00 +00001/*
drhb19a2bc2001-09-16 00:13:26 +00002** 2001 September 15
drh75897232000-05-29 14:26:00 +00003**
drhb19a2bc2001-09-16 00:13:26 +00004** The author disclaims copyright to this source code. In place of
5** a legal notice, here is a blessing:
drh75897232000-05-29 14:26:00 +00006**
drhb19a2bc2001-09-16 00:13:26 +00007** May you do good and not evil.
8** May you find forgiveness for yourself and forgive others.
9** May you share freely, never taking more than you give.
drh75897232000-05-29 14:26:00 +000010**
11*************************************************************************
12** Main file for the SQLite library. The routines in this file
13** implement the programmer interface to the library. Routines in
14** other files are for internal use by SQLite and should not be
15** accessed by users of the library.
16**
drh28dd4792006-06-26 21:35:44 +000017** $Id: main.c,v 1.349 2006/06/26 21:35:45 drh Exp $
drh75897232000-05-29 14:26:00 +000018*/
19#include "sqliteInt.h"
drh8cfbf082001-09-19 13:22:39 +000020#include "os.h"
drhce9079c2002-05-15 14:17:44 +000021#include <ctype.h>
drh75897232000-05-29 14:26:00 +000022
23/*
drh9c054832004-05-31 18:51:57 +000024** The following constant value is used by the SQLITE_BIGENDIAN and
25** SQLITE_LITTLEENDIAN macros.
drhbbd42a62004-05-22 17:41:58 +000026*/
27const int sqlite3one = 1;
28
drhc2311722002-07-19 17:46:38 +000029/*
drhb217a572000-08-22 13:40:18 +000030** The version of the library
31*/
danielk197724b03fd2004-05-10 10:34:34 +000032const char sqlite3_version[] = SQLITE_VERSION;
drh4aec8b62004-08-28 16:19:00 +000033const char *sqlite3_libversion(void){ return sqlite3_version; }
danielk197799ba19e2005-02-05 07:33:34 +000034int sqlite3_libversion_number(void){ return SQLITE_VERSION_NUMBER; }
drhb217a572000-08-22 13:40:18 +000035
36/*
drhd3d39e92004-05-20 22:16:29 +000037** This is the default collating function named "BINARY" which is always
38** available.
39*/
danielk19772c336542005-01-13 02:14:23 +000040static int binCollFunc(
drhd3d39e92004-05-20 22:16:29 +000041 void *NotUsed,
42 int nKey1, const void *pKey1,
43 int nKey2, const void *pKey2
44){
45 int rc, n;
46 n = nKey1<nKey2 ? nKey1 : nKey2;
47 rc = memcmp(pKey1, pKey2, n);
48 if( rc==0 ){
49 rc = nKey1 - nKey2;
50 }
51 return rc;
52}
53
54/*
danielk1977dc1bdc42004-06-11 10:51:27 +000055** Another built-in collating sequence: NOCASE.
56**
57** This collating sequence is intended to be used for "case independant
58** comparison". SQLite's knowledge of upper and lower case equivalents
59** extends only to the 26 characters used in the English language.
60**
61** At the moment there is only a UTF-8 implementation.
danielk19770202b292004-06-09 09:55:16 +000062*/
63static int nocaseCollatingFunc(
64 void *NotUsed,
65 int nKey1, const void *pKey1,
66 int nKey2, const void *pKey2
67){
68 int r = sqlite3StrNICmp(
drh208f80a2004-08-29 20:08:58 +000069 (const char *)pKey1, (const char *)pKey2, (nKey1<nKey2)?nKey1:nKey2);
danielk19770202b292004-06-09 09:55:16 +000070 if( 0==r ){
71 r = nKey1-nKey2;
72 }
73 return r;
74}
75
76/*
drhaf9ff332002-01-16 21:00:27 +000077** Return the ROWID of the most recent insert
78*/
drh9bb575f2004-09-06 17:24:11 +000079sqlite_int64 sqlite3_last_insert_rowid(sqlite3 *db){
drhaf9ff332002-01-16 21:00:27 +000080 return db->lastRowid;
81}
82
83/*
danielk197724b03fd2004-05-10 10:34:34 +000084** Return the number of changes in the most recent call to sqlite3_exec().
drhc8d30ac2002-04-12 10:08:59 +000085*/
drh9bb575f2004-09-06 17:24:11 +000086int sqlite3_changes(sqlite3 *db){
drhc8d30ac2002-04-12 10:08:59 +000087 return db->nChange;
88}
89
rdcf146a772004-02-25 22:51:06 +000090/*
danielk1977b28af712004-06-21 06:50:26 +000091** Return the number of changes since the database handle was opened.
rdcf146a772004-02-25 22:51:06 +000092*/
danielk1977b28af712004-06-21 06:50:26 +000093int sqlite3_total_changes(sqlite3 *db){
94 return db->nTotalChange;
rdcb0c374f2004-02-20 22:53:38 +000095}
96
drhc8d30ac2002-04-12 10:08:59 +000097/*
drh50e5dad2001-09-15 00:57:28 +000098** Close an existing SQLite database
99*/
drh9bb575f2004-09-06 17:24:11 +0000100int sqlite3_close(sqlite3 *db){
drh8e0a2f92002-02-23 23:45:45 +0000101 HashElem *i;
drh001bbcb2003-03-19 03:14:00 +0000102 int j;
danielk19775c4c7782004-06-16 10:39:23 +0000103
104 if( !db ){
danielk197796d81f92004-06-19 03:33:57 +0000105 return SQLITE_OK;
danielk19775c4c7782004-06-16 10:39:23 +0000106 }
drhc60d0442004-09-30 13:43:13 +0000107 if( sqlite3SafetyCheck(db) ){
danielk1977e35ee192004-06-26 09:50:11 +0000108 return SQLITE_MISUSE;
109 }
110
danielk19771f723bd2005-05-26 12:37:29 +0000111#ifdef SQLITE_SSE
drh37527852006-03-16 16:19:56 +0000112 {
113 extern void sqlite3SseCleanup(sqlite3*);
114 sqlite3SseCleanup(db);
115 }
danielk19771f723bd2005-05-26 12:37:29 +0000116#endif
117
danielk197796d81f92004-06-19 03:33:57 +0000118 /* If there are any outstanding VMs, return SQLITE_BUSY. */
119 if( db->pVdbe ){
120 sqlite3Error(db, SQLITE_BUSY,
121 "Unable to close due to unfinalised statements");
122 return SQLITE_BUSY;
123 }
124 assert( !sqlite3SafetyCheck(db) );
danielk1977e0048402004-06-15 16:51:01 +0000125
126 /* FIX ME: db->magic may be set to SQLITE_MAGIC_CLOSED if the database
127 ** cannot be opened for some reason. So this routine needs to run in
128 ** that case. But maybe there should be an extra magic value for the
129 ** "failed to open" state.
130 */
danielk197796d81f92004-06-19 03:33:57 +0000131 if( db->magic!=SQLITE_MAGIC_CLOSED && sqlite3SafetyOn(db) ){
drh94e92032003-02-16 22:21:32 +0000132 /* printf("DID NOT CLOSE\n"); fflush(stdout); */
danielk197796d81f92004-06-19 03:33:57 +0000133 return SQLITE_ERROR;
drh94e92032003-02-16 22:21:32 +0000134 }
danielk1977e0048402004-06-15 16:51:01 +0000135
danielk1977b7a2f2e2006-06-23 11:34:54 +0000136 sqlite3VtabRollback(db);
137
drh001bbcb2003-03-19 03:14:00 +0000138 for(j=0; j<db->nDb; j++){
drh4d189ca2004-02-12 18:46:38 +0000139 struct Db *pDb = &db->aDb[j];
140 if( pDb->pBt ){
danielk19774adee202004-05-08 08:23:19 +0000141 sqlite3BtreeClose(pDb->pBt);
drh4d189ca2004-02-12 18:46:38 +0000142 pDb->pBt = 0;
danielk1977311019b2006-01-10 07:14:23 +0000143 if( j!=1 ){
144 pDb->pSchema = 0;
145 }
drh113088e2003-03-20 01:16:58 +0000146 }
drhf57b3392001-10-08 13:22:32 +0000147 }
danielk19774adee202004-05-08 08:23:19 +0000148 sqlite3ResetInternalSchema(db, 0);
drh1c2d8412003-03-31 00:30:47 +0000149 assert( db->nDb<=2 );
150 assert( db->aDb==db->aDbStatic );
drh0bce8352002-02-28 00:41:10 +0000151 for(i=sqliteHashFirst(&db->aFunc); i; i=sqliteHashNext(i)){
152 FuncDef *pFunc, *pNext;
153 for(pFunc = (FuncDef*)sqliteHashData(i); pFunc; pFunc=pNext){
drh8e0a2f92002-02-23 23:45:45 +0000154 pNext = pFunc->pNext;
155 sqliteFree(pFunc);
156 }
157 }
danielk1977466be562004-06-10 02:16:01 +0000158
danielk1977d8123362004-06-12 09:25:12 +0000159 for(i=sqliteHashFirst(&db->aCollSeq); i; i=sqliteHashNext(i)){
danielk1977466be562004-06-10 02:16:01 +0000160 CollSeq *pColl = (CollSeq *)sqliteHashData(i);
danielk1977d8123362004-06-12 09:25:12 +0000161 sqliteFree(pColl);
danielk1977466be562004-06-10 02:16:01 +0000162 }
danielk1977d8123362004-06-12 09:25:12 +0000163 sqlite3HashClear(&db->aCollSeq);
drhb9bb7c12006-06-11 23:41:55 +0000164#ifndef SQLITE_OMIT_VIRTUALTABLE
danielk1977d1ab1ba2006-06-15 04:28:13 +0000165 for(i=sqliteHashFirst(&db->aModule); i; i=sqliteHashNext(i)){
166 Module *pMod = (Module *)sqliteHashData(i);
167 sqliteFree(pMod);
168 }
drhb9bb7c12006-06-11 23:41:55 +0000169 sqlite3HashClear(&db->aModule);
170#endif
danielk1977466be562004-06-10 02:16:01 +0000171
danielk19774adee202004-05-08 08:23:19 +0000172 sqlite3HashClear(&db->aFunc);
danielk19776622cce2004-05-20 11:00:52 +0000173 sqlite3Error(db, SQLITE_OK, 0); /* Deallocates any cached error strings. */
danielk1977bfd6cce2004-06-18 04:24:54 +0000174 if( db->pErr ){
175 sqlite3ValueFree(db->pErr);
176 }
drhf1952c52006-06-08 15:48:00 +0000177 sqlite3CloseExtensions(db);
danielk197796d81f92004-06-19 03:33:57 +0000178
179 db->magic = SQLITE_MAGIC_ERROR;
danielk1977311019b2006-01-10 07:14:23 +0000180
181 /* The temp-database schema is allocated differently from the other schema
182 ** objects (using sqliteMalloc() directly, instead of sqlite3BtreeSchema()).
183 ** So it needs to be freed here. Todo: Why not roll the temp schema into
184 ** the same sqliteMalloc() as the one that allocates the database
185 ** structure?
186 */
danielk1977da184232006-01-05 11:34:32 +0000187 sqliteFree(db->aDb[1].pSchema);
drh75897232000-05-29 14:26:00 +0000188 sqliteFree(db);
drh6f7adc82006-01-11 21:41:20 +0000189 sqlite3ReleaseThreadData();
danielk197796d81f92004-06-19 03:33:57 +0000190 return SQLITE_OK;
drh75897232000-05-29 14:26:00 +0000191}
192
193/*
drh001bbcb2003-03-19 03:14:00 +0000194** Rollback all database files.
195*/
drh9bb575f2004-09-06 17:24:11 +0000196void sqlite3RollbackAll(sqlite3 *db){
drh001bbcb2003-03-19 03:14:00 +0000197 int i;
danielk1977f3f06bb2005-12-16 15:24:28 +0000198 int inTrans = 0;
drh001bbcb2003-03-19 03:14:00 +0000199 for(i=0; i<db->nDb; i++){
200 if( db->aDb[i].pBt ){
danielk1977f3f06bb2005-12-16 15:24:28 +0000201 if( sqlite3BtreeIsInTrans(db->aDb[i].pBt) ){
202 inTrans = 1;
203 }
danielk19774adee202004-05-08 08:23:19 +0000204 sqlite3BtreeRollback(db->aDb[i].pBt);
drh001bbcb2003-03-19 03:14:00 +0000205 db->aDb[i].inTrans = 0;
206 }
207 }
danielk1977a298e902006-06-22 09:53:48 +0000208 sqlite3VtabRollback(db);
danielk197771fd80b2005-12-16 06:54:01 +0000209 if( db->flags&SQLITE_InternChanges ){
210 sqlite3ResetInternalSchema(db, 0);
211 }
212
213 /* If one has been configured, invoke the rollback-hook callback */
danielk1977f3f06bb2005-12-16 15:24:28 +0000214 if( db->xRollbackCallback && (inTrans || !db->autoCommit) ){
danielk197771fd80b2005-12-16 06:54:01 +0000215 db->xRollbackCallback(db->pRollbackArg);
216 }
drh001bbcb2003-03-19 03:14:00 +0000217}
218
219/*
drhc22bd472002-05-10 13:14:07 +0000220** Return a static string that describes the kind of error specified in the
221** argument.
drh247be432002-05-10 05:44:55 +0000222*/
danielk1977f20b21c2004-05-31 23:56:42 +0000223const char *sqlite3ErrStr(int rc){
drhc22bd472002-05-10 13:14:07 +0000224 const char *z;
225 switch( rc ){
drhc60d0442004-09-30 13:43:13 +0000226 case SQLITE_ROW:
227 case SQLITE_DONE:
drhc22bd472002-05-10 13:14:07 +0000228 case SQLITE_OK: z = "not an error"; break;
229 case SQLITE_ERROR: z = "SQL logic error or missing database"; break;
drhc22bd472002-05-10 13:14:07 +0000230 case SQLITE_PERM: z = "access permission denied"; break;
231 case SQLITE_ABORT: z = "callback requested query abort"; break;
232 case SQLITE_BUSY: z = "database is locked"; break;
233 case SQLITE_LOCKED: z = "database table is locked"; break;
234 case SQLITE_NOMEM: z = "out of memory"; break;
235 case SQLITE_READONLY: z = "attempt to write a readonly database"; break;
236 case SQLITE_INTERRUPT: z = "interrupted"; break;
237 case SQLITE_IOERR: z = "disk I/O error"; break;
238 case SQLITE_CORRUPT: z = "database disk image is malformed"; break;
drh2db0bbc2005-08-11 02:10:18 +0000239 case SQLITE_FULL: z = "database or disk is full"; break;
drhc22bd472002-05-10 13:14:07 +0000240 case SQLITE_CANTOPEN: z = "unable to open database file"; break;
241 case SQLITE_PROTOCOL: z = "database locking protocol failure"; break;
242 case SQLITE_EMPTY: z = "table contains no data"; break;
243 case SQLITE_SCHEMA: z = "database schema has changed"; break;
drhc22bd472002-05-10 13:14:07 +0000244 case SQLITE_CONSTRAINT: z = "constraint failed"; break;
245 case SQLITE_MISMATCH: z = "datatype mismatch"; break;
246 case SQLITE_MISUSE: z = "library routine called out of sequence";break;
drh8766c342002-11-09 00:33:15 +0000247 case SQLITE_NOLFS: z = "kernel lacks large file support"; break;
drhed6c8672003-01-12 18:02:16 +0000248 case SQLITE_AUTH: z = "authorization denied"; break;
jplyon892f6712003-06-12 08:59:00 +0000249 case SQLITE_FORMAT: z = "auxiliary database format error"; break;
drhb08153d2004-11-20 20:18:55 +0000250 case SQLITE_RANGE: z = "bind or column index out of range"; break;
drhc602f9a2004-02-12 19:01:04 +0000251 case SQLITE_NOTADB: z = "file is encrypted or is not a database";break;
drhc22bd472002-05-10 13:14:07 +0000252 default: z = "unknown error"; break;
drh247be432002-05-10 05:44:55 +0000253 }
drhc22bd472002-05-10 13:14:07 +0000254 return z;
drh247be432002-05-10 05:44:55 +0000255}
256
257/*
drh2dfbbca2000-07-28 14:32:48 +0000258** This routine implements a busy callback that sleeps and tries
259** again until a timeout value is reached. The timeout value is
260** an integer number of milliseconds passed in as the first
261** argument.
262*/
drhdaffd0e2001-04-11 14:28:42 +0000263static int sqliteDefaultBusyCallback(
drh97903fe2005-05-24 20:19:57 +0000264 void *ptr, /* Database connection */
drh2dfbbca2000-07-28 14:32:48 +0000265 int count /* Number of times table has been busy */
266){
drh91c839b2006-03-03 20:54:41 +0000267#if OS_WIN || (defined(HAVE_USLEEP) && HAVE_USLEEP)
drhee570fa2005-04-28 12:06:05 +0000268 static const u8 delays[] =
269 { 1, 2, 5, 10, 15, 20, 25, 25, 25, 50, 50, 100 };
270 static const u8 totals[] =
271 { 0, 1, 3, 8, 18, 33, 53, 78, 103, 128, 178, 228 };
drhd1bec472004-01-15 13:29:31 +0000272# define NDELAY (sizeof(delays)/sizeof(delays[0]))
drh97903fe2005-05-24 20:19:57 +0000273 int timeout = ((sqlite3 *)ptr)->busyTimeout;
274 int delay, prior;
drh2dfbbca2000-07-28 14:32:48 +0000275
drhee570fa2005-04-28 12:06:05 +0000276 assert( count>=0 );
277 if( count < NDELAY ){
278 delay = delays[count];
279 prior = totals[count];
drhd1bec472004-01-15 13:29:31 +0000280 }else{
281 delay = delays[NDELAY-1];
drh68cb6192005-05-06 22:05:56 +0000282 prior = totals[NDELAY-1] + delay*(count-(NDELAY-1));
drh2dfbbca2000-07-28 14:32:48 +0000283 }
drhd1bec472004-01-15 13:29:31 +0000284 if( prior + delay > timeout ){
285 delay = timeout - prior;
drh2dfbbca2000-07-28 14:32:48 +0000286 if( delay<=0 ) return 0;
287 }
drh66560ad2006-01-06 14:32:19 +0000288 sqlite3OsSleep(delay);
drh2dfbbca2000-07-28 14:32:48 +0000289 return 1;
290#else
drh3f737082005-06-14 02:24:31 +0000291 int timeout = ((sqlite3 *)ptr)->busyTimeout;
drh2dfbbca2000-07-28 14:32:48 +0000292 if( (count+1)*1000 > timeout ){
293 return 0;
294 }
drh66560ad2006-01-06 14:32:19 +0000295 sqlite3OsSleep(1000);
drh2dfbbca2000-07-28 14:32:48 +0000296 return 1;
297#endif
298}
299
300/*
drha4afb652005-07-09 02:16:02 +0000301** Invoke the given busy handler.
302**
303** This routine is called when an operation failed with a lock.
304** If this routine returns non-zero, the lock is retried. If it
305** returns 0, the operation aborts with an SQLITE_BUSY error.
306*/
307int sqlite3InvokeBusyHandler(BusyHandler *p){
308 int rc;
309 if( p==0 || p->xFunc==0 || p->nBusy<0 ) return 0;
310 rc = p->xFunc(p->pArg, p->nBusy);
311 if( rc==0 ){
312 p->nBusy = -1;
313 }else{
314 p->nBusy++;
315 }
316 return rc;
317}
318
319/*
drh2dfbbca2000-07-28 14:32:48 +0000320** This routine sets the busy callback for an Sqlite database to the
321** given callback function with the given argument.
322*/
danielk1977f9d64d22004-06-19 08:18:07 +0000323int sqlite3_busy_handler(
324 sqlite3 *db,
danielk19772a764eb2004-06-12 01:43:26 +0000325 int (*xBusy)(void*,int),
drh2dfbbca2000-07-28 14:32:48 +0000326 void *pArg
327){
drhc60d0442004-09-30 13:43:13 +0000328 if( sqlite3SafetyCheck(db) ){
329 return SQLITE_MISUSE;
330 }
danielk197724162fe2004-06-04 06:22:00 +0000331 db->busyHandler.xFunc = xBusy;
332 db->busyHandler.pArg = pArg;
drha4afb652005-07-09 02:16:02 +0000333 db->busyHandler.nBusy = 0;
danielk1977f9d64d22004-06-19 08:18:07 +0000334 return SQLITE_OK;
drh2dfbbca2000-07-28 14:32:48 +0000335}
336
danielk1977348bb5d2003-10-18 09:37:26 +0000337#ifndef SQLITE_OMIT_PROGRESS_CALLBACK
338/*
339** This routine sets the progress callback for an Sqlite database to the
340** given callback function with the given argument. The progress callback will
341** be invoked every nOps opcodes.
342*/
danielk197724b03fd2004-05-10 10:34:34 +0000343void sqlite3_progress_handler(
drh9bb575f2004-09-06 17:24:11 +0000344 sqlite3 *db,
danielk1977348bb5d2003-10-18 09:37:26 +0000345 int nOps,
346 int (*xProgress)(void*),
347 void *pArg
348){
drhc60d0442004-09-30 13:43:13 +0000349 if( !sqlite3SafetyCheck(db) ){
350 if( nOps>0 ){
351 db->xProgress = xProgress;
352 db->nProgressOps = nOps;
353 db->pProgressArg = pArg;
354 }else{
355 db->xProgress = 0;
356 db->nProgressOps = 0;
357 db->pProgressArg = 0;
358 }
danielk1977348bb5d2003-10-18 09:37:26 +0000359 }
360}
361#endif
362
363
drh2dfbbca2000-07-28 14:32:48 +0000364/*
365** This routine installs a default busy handler that waits for the
366** specified number of milliseconds before returning 0.
367*/
danielk1977f9d64d22004-06-19 08:18:07 +0000368int sqlite3_busy_timeout(sqlite3 *db, int ms){
drh2dfbbca2000-07-28 14:32:48 +0000369 if( ms>0 ){
drh97903fe2005-05-24 20:19:57 +0000370 db->busyTimeout = ms;
371 sqlite3_busy_handler(db, sqliteDefaultBusyCallback, (void*)db);
drh2dfbbca2000-07-28 14:32:48 +0000372 }else{
danielk197724b03fd2004-05-10 10:34:34 +0000373 sqlite3_busy_handler(db, 0, 0);
drh2dfbbca2000-07-28 14:32:48 +0000374 }
danielk1977f9d64d22004-06-19 08:18:07 +0000375 return SQLITE_OK;
drh2dfbbca2000-07-28 14:32:48 +0000376}
drh4c504392000-10-16 22:06:40 +0000377
378/*
379** Cause any pending operation to stop at its earliest opportunity.
380*/
drh9bb575f2004-09-06 17:24:11 +0000381void sqlite3_interrupt(sqlite3 *db){
drhc60d0442004-09-30 13:43:13 +0000382 if( !sqlite3SafetyCheck(db) ){
383 db->flags |= SQLITE_Interrupt;
384 }
drh4c504392000-10-16 22:06:40 +0000385}
drhfa86c412002-02-02 15:01:15 +0000386
387/*
drh28dd4792006-06-26 21:35:44 +0000388** Memory allocation routines that use SQLites internal memory
389** memory allocator. Depending on how SQLite is compiled, the
390** internal memory allocator might be just an alias for the
391** system default malloc/realloc/free. Or the built-in allocator
392** might do extra stuff like put sentinals around buffers to
393** check for overruns or look for memory leaks.
drhfa86c412002-02-02 15:01:15 +0000394**
drh28dd4792006-06-26 21:35:44 +0000395** Use sqlite3_free() to free memory returned by sqlite3_mprintf().
drhfa86c412002-02-02 15:01:15 +0000396*/
drh28dd4792006-06-26 21:35:44 +0000397void sqlite3_free(void *p){ if( p ) sqlite3OsFree(p); }
398void *sqlite3_malloc(int nByte){ return nByte>0 ? sqlite3OsMalloc(nByte) : 0; }
399void *sqlite3_realloc(void *pOld, int nByte){
400 if( pOld ){
401 if( nByte>0 ){
402 return sqlite3OsRealloc(pOld, nByte);
403 }else{
404 sqlite3OsFree(pOld);
405 return 0;
406 }
407 }else{
408 return sqlite3_malloc(nByte);
409 }
410}
drhfa86c412002-02-02 15:01:15 +0000411
412/*
danielk1977771151b2006-01-17 13:21:40 +0000413** This function is exactly the same as sqlite3_create_function(), except
414** that it is designed to be called by internal code. The difference is
415** that if a malloc() fails in sqlite3_create_function(), an error code
416** is returned and the mallocFailed flag cleared.
drhfa86c412002-02-02 15:01:15 +0000417*/
danielk1977771151b2006-01-17 13:21:40 +0000418int sqlite3CreateFunc(
danielk197765904932004-05-26 06:18:37 +0000419 sqlite3 *db,
420 const char *zFunctionName,
421 int nArg,
danielk1977d8123362004-06-12 09:25:12 +0000422 int enc,
danielk197765904932004-05-26 06:18:37 +0000423 void *pUserData,
424 void (*xFunc)(sqlite3_context*,int,sqlite3_value **),
425 void (*xStep)(sqlite3_context*,int,sqlite3_value **),
426 void (*xFinal)(sqlite3_context*)
drh8e0a2f92002-02-23 23:45:45 +0000427){
drh0bce8352002-02-28 00:41:10 +0000428 FuncDef *p;
drh4b59ab52002-08-24 18:24:51 +0000429 int nName;
danielk197765904932004-05-26 06:18:37 +0000430
drhc60d0442004-09-30 13:43:13 +0000431 if( sqlite3SafetyCheck(db) ){
432 return SQLITE_MISUSE;
433 }
434 if( zFunctionName==0 ||
danielk1977398eae72004-05-26 06:58:43 +0000435 (xFunc && (xFinal || xStep)) ||
436 (!xFunc && (xFinal && !xStep)) ||
437 (!xFunc && (!xFinal && xStep)) ||
438 (nArg<-1 || nArg>127) ||
439 (255<(nName = strlen(zFunctionName))) ){
drhbadf7a72006-06-14 15:35:36 +0000440 sqlite3Error(db, SQLITE_ERROR, "bad parameters");
danielk197765904932004-05-26 06:18:37 +0000441 return SQLITE_ERROR;
442 }
danielk1977d8123362004-06-12 09:25:12 +0000443
danielk197793758c82005-01-21 08:13:14 +0000444#ifndef SQLITE_OMIT_UTF16
danielk1977d8123362004-06-12 09:25:12 +0000445 /* If SQLITE_UTF16 is specified as the encoding type, transform this
446 ** to one of SQLITE_UTF16LE or SQLITE_UTF16BE using the
447 ** SQLITE_UTF16NATIVE macro. SQLITE_UTF16 is not used internally.
448 **
449 ** If SQLITE_ANY is specified, add three versions of the function
450 ** to the hash table.
451 */
452 if( enc==SQLITE_UTF16 ){
453 enc = SQLITE_UTF16NATIVE;
454 }else if( enc==SQLITE_ANY ){
455 int rc;
danielk1977771151b2006-01-17 13:21:40 +0000456 rc = sqlite3CreateFunc(db, zFunctionName, nArg, SQLITE_UTF8,
danielk1977f9d64d22004-06-19 08:18:07 +0000457 pUserData, xFunc, xStep, xFinal);
danielk1977d8123362004-06-12 09:25:12 +0000458 if( rc!=SQLITE_OK ) return rc;
danielk1977771151b2006-01-17 13:21:40 +0000459 rc = sqlite3CreateFunc(db, zFunctionName, nArg, SQLITE_UTF16LE,
danielk1977f9d64d22004-06-19 08:18:07 +0000460 pUserData, xFunc, xStep, xFinal);
danielk1977d8123362004-06-12 09:25:12 +0000461 if( rc!=SQLITE_OK ) return rc;
462 enc = SQLITE_UTF16BE;
463 }
danielk197793758c82005-01-21 08:13:14 +0000464#else
465 enc = SQLITE_UTF8;
466#endif
danielk19779636c4e2005-01-25 04:27:54 +0000467
468 /* Check if an existing function is being overridden or deleted. If so,
469 ** and there are active VMs, then return SQLITE_BUSY. If a function
470 ** is being overridden/deleted but there are no active VMs, allow the
471 ** operation to continue but invalidate all precompiled statements.
472 */
473 p = sqlite3FindFunction(db, zFunctionName, nName, nArg, enc, 0);
474 if( p && p->iPrefEnc==enc && p->nArg==nArg ){
475 if( db->activeVdbeCnt ){
476 sqlite3Error(db, SQLITE_BUSY,
477 "Unable to delete/modify user-function due to active statements");
danielk19779e128002006-01-18 16:51:35 +0000478 assert( !sqlite3MallocFailed() );
danielk19779636c4e2005-01-25 04:27:54 +0000479 return SQLITE_BUSY;
480 }else{
481 sqlite3ExpirePreparedStatements(db);
482 }
483 }
danielk197765904932004-05-26 06:18:37 +0000484
danielk1977d8123362004-06-12 09:25:12 +0000485 p = sqlite3FindFunction(db, zFunctionName, nName, nArg, enc, 1);
danielk1977771151b2006-01-17 13:21:40 +0000486 if( p ){
487 p->flags = 0;
488 p->xFunc = xFunc;
489 p->xStep = xStep;
490 p->xFinalize = xFinal;
491 p->pUserData = pUserData;
danielk197765fd59f2006-06-24 11:51:33 +0000492 p->nArg = nArg;
danielk1977771151b2006-01-17 13:21:40 +0000493 }
danielk197765904932004-05-26 06:18:37 +0000494 return SQLITE_OK;
495}
danielk1977771151b2006-01-17 13:21:40 +0000496
497/*
498** Create new user functions.
499*/
500int sqlite3_create_function(
501 sqlite3 *db,
502 const char *zFunctionName,
503 int nArg,
504 int enc,
505 void *p,
506 void (*xFunc)(sqlite3_context*,int,sqlite3_value **),
507 void (*xStep)(sqlite3_context*,int,sqlite3_value **),
508 void (*xFinal)(sqlite3_context*)
509){
510 int rc;
danielk19779e128002006-01-18 16:51:35 +0000511 assert( !sqlite3MallocFailed() );
danielk1977771151b2006-01-17 13:21:40 +0000512 rc = sqlite3CreateFunc(db, zFunctionName, nArg, enc, p, xFunc, xStep, xFinal);
513
danielk197754f01982006-01-18 15:25:17 +0000514 return sqlite3ApiExit(db, rc);
danielk1977771151b2006-01-17 13:21:40 +0000515}
516
danielk197793758c82005-01-21 08:13:14 +0000517#ifndef SQLITE_OMIT_UTF16
danielk197765904932004-05-26 06:18:37 +0000518int sqlite3_create_function16(
519 sqlite3 *db,
520 const void *zFunctionName,
521 int nArg,
522 int eTextRep,
danielk1977771151b2006-01-17 13:21:40 +0000523 void *p,
danielk197765904932004-05-26 06:18:37 +0000524 void (*xFunc)(sqlite3_context*,int,sqlite3_value**),
525 void (*xStep)(sqlite3_context*,int,sqlite3_value**),
526 void (*xFinal)(sqlite3_context*)
527){
528 int rc;
drhaf9a7c22005-12-15 03:04:10 +0000529 char *zFunc8;
danielk19779e128002006-01-18 16:51:35 +0000530 assert( !sqlite3MallocFailed() );
danielk1977bfd6cce2004-06-18 04:24:54 +0000531
drhaf9a7c22005-12-15 03:04:10 +0000532 zFunc8 = sqlite3utf16to8(zFunctionName, -1);
danielk1977771151b2006-01-17 13:21:40 +0000533 rc = sqlite3CreateFunc(db, zFunc8, nArg, eTextRep, p, xFunc, xStep, xFinal);
drhaf9a7c22005-12-15 03:04:10 +0000534 sqliteFree(zFunc8);
danielk1977771151b2006-01-17 13:21:40 +0000535
danielk197754f01982006-01-18 15:25:17 +0000536 return sqlite3ApiExit(db, rc);
drh8e0a2f92002-02-23 23:45:45 +0000537}
danielk197793758c82005-01-21 08:13:14 +0000538#endif
drhc9b84a12002-06-20 11:36:48 +0000539
drh19e2d372005-08-29 23:00:03 +0000540#ifndef SQLITE_OMIT_TRACE
drhc9b84a12002-06-20 11:36:48 +0000541/*
drh18de4822003-01-16 16:28:53 +0000542** Register a trace function. The pArg from the previously registered trace
543** is returned.
544**
545** A NULL trace function means that no tracing is executes. A non-NULL
546** trace is a pointer to a function that is invoked at the start of each
drh19e2d372005-08-29 23:00:03 +0000547** SQL statement.
drh18de4822003-01-16 16:28:53 +0000548*/
drh9bb575f2004-09-06 17:24:11 +0000549void *sqlite3_trace(sqlite3 *db, void (*xTrace)(void*,const char*), void *pArg){
drh18de4822003-01-16 16:28:53 +0000550 void *pOld = db->pTraceArg;
551 db->xTrace = xTrace;
552 db->pTraceArg = pArg;
553 return pOld;
drh0d1a6432003-04-03 15:46:04 +0000554}
drh19e2d372005-08-29 23:00:03 +0000555/*
556** Register a profile function. The pArg from the previously registered
557** profile function is returned.
558**
559** A NULL profile function means that no profiling is executes. A non-NULL
560** profile is a pointer to a function that is invoked at the conclusion of
561** each SQL statement that is run.
562*/
563void *sqlite3_profile(
564 sqlite3 *db,
565 void (*xProfile)(void*,const char*,sqlite_uint64),
566 void *pArg
567){
568 void *pOld = db->pProfileArg;
569 db->xProfile = xProfile;
570 db->pProfileArg = pArg;
571 return pOld;
572}
573#endif /* SQLITE_OMIT_TRACE */
paulb0208cc2003-04-13 18:26:49 +0000574
drhaa940ea2004-01-15 02:44:03 +0000575/*** EXPERIMENTAL ***
576**
577** Register a function to be invoked when a transaction comments.
danielk197771fd80b2005-12-16 06:54:01 +0000578** If the invoked function returns non-zero, then the commit becomes a
drhaa940ea2004-01-15 02:44:03 +0000579** rollback.
580*/
danielk197724b03fd2004-05-10 10:34:34 +0000581void *sqlite3_commit_hook(
drh9bb575f2004-09-06 17:24:11 +0000582 sqlite3 *db, /* Attach the hook to this database */
drhaa940ea2004-01-15 02:44:03 +0000583 int (*xCallback)(void*), /* Function to invoke on each commit */
584 void *pArg /* Argument to the function */
585){
586 void *pOld = db->pCommitArg;
587 db->xCommitCallback = xCallback;
588 db->pCommitArg = pArg;
589 return pOld;
590}
591
danielk197794eb6a12005-12-15 15:22:08 +0000592/*
593** Register a callback to be invoked each time a row is updated,
594** inserted or deleted using this database connection.
595*/
danielk197771fd80b2005-12-16 06:54:01 +0000596void *sqlite3_update_hook(
danielk197794eb6a12005-12-15 15:22:08 +0000597 sqlite3 *db, /* Attach the hook to this database */
598 void (*xCallback)(void*,int,char const *,char const *,sqlite_int64),
599 void *pArg /* Argument to the function */
600){
danielk197771fd80b2005-12-16 06:54:01 +0000601 void *pRet = db->pUpdateArg;
danielk197794eb6a12005-12-15 15:22:08 +0000602 db->xUpdateCallback = xCallback;
603 db->pUpdateArg = pArg;
danielk197771fd80b2005-12-16 06:54:01 +0000604 return pRet;
danielk197794eb6a12005-12-15 15:22:08 +0000605}
606
danielk197771fd80b2005-12-16 06:54:01 +0000607/*
608** Register a callback to be invoked each time a transaction is rolled
609** back by this database connection.
610*/
611void *sqlite3_rollback_hook(
612 sqlite3 *db, /* Attach the hook to this database */
613 void (*xCallback)(void*), /* Callback function */
614 void *pArg /* Argument to the function */
615){
616 void *pRet = db->pRollbackArg;
617 db->xRollbackCallback = xCallback;
618 db->pRollbackArg = pArg;
619 return pRet;
620}
drhaa940ea2004-01-15 02:44:03 +0000621
paulb0208cc2003-04-13 18:26:49 +0000622/*
drh13bff812003-04-15 01:19:47 +0000623** This routine is called to create a connection to a database BTree
624** driver. If zFilename is the name of a file, then that file is
625** opened and used. If zFilename is the magic name ":memory:" then
626** the database is stored in memory (and is thus forgotten as soon as
627** the connection is closed.) If zFilename is NULL then the database
drh84bfda42005-07-15 13:05:21 +0000628** is a "virtual" database for transient use only and is deleted as
629** soon as the connection is closed.
drh90f5ecb2004-07-22 01:19:35 +0000630**
drh84bfda42005-07-15 13:05:21 +0000631** A virtual database can be either a disk file (that is automatically
632** deleted when the file is closed) or it an be held entirely in memory,
drh90f5ecb2004-07-22 01:19:35 +0000633** depending on the values of the TEMP_STORE compile-time macro and the
634** db->temp_store variable, according to the following chart:
635**
636** TEMP_STORE db->temp_store Location of temporary database
637** ---------- -------------- ------------------------------
638** 0 any file
639** 1 1 file
640** 1 2 memory
641** 1 0 file
642** 2 1 file
643** 2 2 memory
644** 2 0 memory
645** 3 any memory
paulb0208cc2003-04-13 18:26:49 +0000646*/
danielk19774adee202004-05-08 08:23:19 +0000647int sqlite3BtreeFactory(
drh9bb575f2004-09-06 17:24:11 +0000648 const sqlite3 *db, /* Main database when opening aux otherwise 0 */
paulb0208cc2003-04-13 18:26:49 +0000649 const char *zFilename, /* Name of the file containing the BTree database */
650 int omitJournal, /* if TRUE then do not journal this file */
651 int nCache, /* How many pages in the page cache */
danielk19774adee202004-05-08 08:23:19 +0000652 Btree **ppBtree /* Pointer to new Btree object written here */
653){
danielk19774adee202004-05-08 08:23:19 +0000654 int btree_flags = 0;
drh90f5ecb2004-07-22 01:19:35 +0000655 int rc;
danielk19774adee202004-05-08 08:23:19 +0000656
drheec983e2004-05-08 10:11:36 +0000657 assert( ppBtree != 0);
danielk19774adee202004-05-08 08:23:19 +0000658 if( omitJournal ){
659 btree_flags |= BTREE_OMIT_JOURNAL;
paulb0208cc2003-04-13 18:26:49 +0000660 }
drh7bec5052005-02-06 02:45:41 +0000661 if( db->flags & SQLITE_NoReadlock ){
662 btree_flags |= BTREE_NO_READLOCK;
663 }
drh90f5ecb2004-07-22 01:19:35 +0000664 if( zFilename==0 ){
drh90f5ecb2004-07-22 01:19:35 +0000665#if TEMP_STORE==0
drh22ac46d2004-08-14 18:34:54 +0000666 /* Do nothing */
drh90f5ecb2004-07-22 01:19:35 +0000667#endif
danielk197703aded42004-11-22 05:26:27 +0000668#ifndef SQLITE_OMIT_MEMORYDB
drh90f5ecb2004-07-22 01:19:35 +0000669#if TEMP_STORE==1
drh22ac46d2004-08-14 18:34:54 +0000670 if( db->temp_store==2 ) zFilename = ":memory:";
drh90f5ecb2004-07-22 01:19:35 +0000671#endif
672#if TEMP_STORE==2
drh22ac46d2004-08-14 18:34:54 +0000673 if( db->temp_store!=1 ) zFilename = ":memory:";
drh90f5ecb2004-07-22 01:19:35 +0000674#endif
675#if TEMP_STORE==3
drh22ac46d2004-08-14 18:34:54 +0000676 zFilename = ":memory:";
drh90f5ecb2004-07-22 01:19:35 +0000677#endif
danielk197703aded42004-11-22 05:26:27 +0000678#endif /* SQLITE_OMIT_MEMORYDB */
drh90f5ecb2004-07-22 01:19:35 +0000679 }
danielk19774adee202004-05-08 08:23:19 +0000680
danielk1977da184232006-01-05 11:34:32 +0000681 rc = sqlite3BtreeOpen(zFilename, (sqlite3 *)db, ppBtree, btree_flags);
drh90f5ecb2004-07-22 01:19:35 +0000682 if( rc==SQLITE_OK ){
683 sqlite3BtreeSetBusyHandler(*ppBtree, (void*)&db->busyHandler);
684 sqlite3BtreeSetCacheSize(*ppBtree, nCache);
685 }
686 return rc;
paulb0208cc2003-04-13 18:26:49 +0000687}
danielk19774adee202004-05-08 08:23:19 +0000688
danielk19774ad17132004-05-21 01:47:26 +0000689/*
690** Return UTF-8 encoded English language explanation of the most recent
691** error.
692*/
danielk19776622cce2004-05-20 11:00:52 +0000693const char *sqlite3_errmsg(sqlite3 *db){
drhc60d0442004-09-30 13:43:13 +0000694 const char *z;
danielk19779e128002006-01-18 16:51:35 +0000695 if( !db || sqlite3MallocFailed() ){
danielk1977f20b21c2004-05-31 23:56:42 +0000696 return sqlite3ErrStr(SQLITE_NOMEM);
danielk19774ad17132004-05-21 01:47:26 +0000697 }
drhc60d0442004-09-30 13:43:13 +0000698 if( sqlite3SafetyCheck(db) || db->errCode==SQLITE_MISUSE ){
danielk1977e35ee192004-06-26 09:50:11 +0000699 return sqlite3ErrStr(SQLITE_MISUSE);
700 }
drh2646da72005-12-09 20:02:05 +0000701 z = (char*)sqlite3_value_text(db->pErr);
drhc60d0442004-09-30 13:43:13 +0000702 if( z==0 ){
703 z = sqlite3ErrStr(db->errCode);
danielk19776622cce2004-05-20 11:00:52 +0000704 }
drhc60d0442004-09-30 13:43:13 +0000705 return z;
danielk19776622cce2004-05-20 11:00:52 +0000706}
707
drh6c626082004-11-14 21:56:29 +0000708#ifndef SQLITE_OMIT_UTF16
danielk19774ad17132004-05-21 01:47:26 +0000709/*
710** Return UTF-16 encoded English language explanation of the most recent
711** error.
712*/
danielk19776622cce2004-05-20 11:00:52 +0000713const void *sqlite3_errmsg16(sqlite3 *db){
danielk1977bfd6cce2004-06-18 04:24:54 +0000714 /* Because all the characters in the string are in the unicode
715 ** range 0x00-0xFF, if we pad the big-endian string with a
716 ** zero byte, we can obtain the little-endian string with
717 ** &big_endian[1].
718 */
drh57196282004-10-06 15:41:16 +0000719 static const char outOfMemBe[] = {
danielk1977bfd6cce2004-06-18 04:24:54 +0000720 0, 'o', 0, 'u', 0, 't', 0, ' ',
721 0, 'o', 0, 'f', 0, ' ',
722 0, 'm', 0, 'e', 0, 'm', 0, 'o', 0, 'r', 0, 'y', 0, 0, 0
723 };
drh57196282004-10-06 15:41:16 +0000724 static const char misuseBe [] = {
danielk1977e35ee192004-06-26 09:50:11 +0000725 0, 'l', 0, 'i', 0, 'b', 0, 'r', 0, 'a', 0, 'r', 0, 'y', 0, ' ',
726 0, 'r', 0, 'o', 0, 'u', 0, 't', 0, 'i', 0, 'n', 0, 'e', 0, ' ',
727 0, 'c', 0, 'a', 0, 'l', 0, 'l', 0, 'e', 0, 'd', 0, ' ',
728 0, 'o', 0, 'u', 0, 't', 0, ' ',
729 0, 'o', 0, 'f', 0, ' ',
730 0, 's', 0, 'e', 0, 'q', 0, 'u', 0, 'e', 0, 'n', 0, 'c', 0, 'e', 0, 0, 0
731 };
danielk19774ad17132004-05-21 01:47:26 +0000732
drhc60d0442004-09-30 13:43:13 +0000733 const void *z;
danielk19779e128002006-01-18 16:51:35 +0000734 if( sqlite3MallocFailed() ){
drhc60d0442004-09-30 13:43:13 +0000735 return (void *)(&outOfMemBe[SQLITE_UTF16NATIVE==SQLITE_UTF16LE?1:0]);
736 }
737 if( sqlite3SafetyCheck(db) || db->errCode==SQLITE_MISUSE ){
738 return (void *)(&misuseBe[SQLITE_UTF16NATIVE==SQLITE_UTF16LE?1:0]);
739 }
740 z = sqlite3_value_text16(db->pErr);
741 if( z==0 ){
742 sqlite3ValueSetStr(db->pErr, -1, sqlite3ErrStr(db->errCode),
743 SQLITE_UTF8, SQLITE_STATIC);
744 z = sqlite3_value_text16(db->pErr);
745 }
danielk197754f01982006-01-18 15:25:17 +0000746 sqlite3ApiExit(0, 0);
drhc60d0442004-09-30 13:43:13 +0000747 return z;
danielk19776622cce2004-05-20 11:00:52 +0000748}
drh6c626082004-11-14 21:56:29 +0000749#endif /* SQLITE_OMIT_UTF16 */
danielk19776622cce2004-05-20 11:00:52 +0000750
drhc60d0442004-09-30 13:43:13 +0000751/*
danielk19777ddad962005-12-12 06:53:03 +0000752** Return the most recent error code generated by an SQLite routine. If NULL is
753** passed to this function, we assume a malloc() failed during sqlite3_open().
drhc60d0442004-09-30 13:43:13 +0000754*/
danielk19776622cce2004-05-20 11:00:52 +0000755int sqlite3_errcode(sqlite3 *db){
danielk19779e128002006-01-18 16:51:35 +0000756 if( !db || sqlite3MallocFailed() ){
drhc60d0442004-09-30 13:43:13 +0000757 return SQLITE_NOMEM;
758 }
759 if( sqlite3SafetyCheck(db) ){
760 return SQLITE_MISUSE;
761 }
danielk19776622cce2004-05-20 11:00:52 +0000762 return db->errCode;
763}
764
drh0e819f92006-02-01 13:50:41 +0000765/*
766** Create a new collating function for database "db". The name is zName
767** and the encoding is enc.
768*/
danielk19779a30cf62006-01-18 04:26:07 +0000769static int createCollation(
770 sqlite3* db,
771 const char *zName,
772 int enc,
773 void* pCtx,
774 int(*xCompare)(void*,int,const void*,int,const void*)
775){
776 CollSeq *pColl;
drh7d9bd4e2006-02-16 18:16:36 +0000777 int enc2;
danielk19779a30cf62006-01-18 04:26:07 +0000778
779 if( sqlite3SafetyCheck(db) ){
780 return SQLITE_MISUSE;
781 }
782
783 /* If SQLITE_UTF16 is specified as the encoding type, transform this
784 ** to one of SQLITE_UTF16LE or SQLITE_UTF16BE using the
785 ** SQLITE_UTF16NATIVE macro. SQLITE_UTF16 is not used internally.
786 */
drh7d9bd4e2006-02-16 18:16:36 +0000787 enc2 = enc & ~SQLITE_UTF16_ALIGNED;
788 if( enc2==SQLITE_UTF16 ){
789 enc2 = SQLITE_UTF16NATIVE;
danielk19779a30cf62006-01-18 04:26:07 +0000790 }
791
drh7d9bd4e2006-02-16 18:16:36 +0000792 if( (enc2&~3)!=0 ){
793 sqlite3Error(db, SQLITE_ERROR, "unknown encoding");
danielk19779a30cf62006-01-18 04:26:07 +0000794 return SQLITE_ERROR;
795 }
796
797 /* Check if this call is removing or replacing an existing collation
798 ** sequence. If so, and there are active VMs, return busy. If there
799 ** are no active VMs, invalidate any pre-compiled statements.
800 */
drh7d9bd4e2006-02-16 18:16:36 +0000801 pColl = sqlite3FindCollSeq(db, (u8)enc2, zName, strlen(zName), 0);
danielk19779a30cf62006-01-18 04:26:07 +0000802 if( pColl && pColl->xCmp ){
803 if( db->activeVdbeCnt ){
804 sqlite3Error(db, SQLITE_BUSY,
805 "Unable to delete/modify collation sequence due to active statements");
806 return SQLITE_BUSY;
807 }
808 sqlite3ExpirePreparedStatements(db);
809 }
810
drh7d9bd4e2006-02-16 18:16:36 +0000811 pColl = sqlite3FindCollSeq(db, (u8)enc2, zName, strlen(zName), 1);
danielk19779a30cf62006-01-18 04:26:07 +0000812 if( pColl ){
813 pColl->xCmp = xCompare;
814 pColl->pUser = pCtx;
drh7d9bd4e2006-02-16 18:16:36 +0000815 pColl->enc = enc2 | (enc & SQLITE_UTF16_ALIGNED);
danielk19779a30cf62006-01-18 04:26:07 +0000816 }
817 sqlite3Error(db, SQLITE_OK, 0);
818 return SQLITE_OK;
819}
820
821
danielk19776622cce2004-05-20 11:00:52 +0000822/*
danielk19774ad17132004-05-21 01:47:26 +0000823** This routine does the work of opening a database on behalf of
824** sqlite3_open() and sqlite3_open16(). The database filename "zFilename"
drhee570fa2005-04-28 12:06:05 +0000825** is UTF-8 encoded.
danielk19774ad17132004-05-21 01:47:26 +0000826*/
827static int openDatabase(
828 const char *zFilename, /* Database filename UTF-8 encoded */
danielk19778e227872004-06-07 07:52:17 +0000829 sqlite3 **ppDb /* OUT: Returned database handle */
danielk19774ad17132004-05-21 01:47:26 +0000830){
831 sqlite3 *db;
danielk1977da184232006-01-05 11:34:32 +0000832 int rc;
drhd64fe2f2005-08-28 17:00:23 +0000833 CollSeq *pColl;
danielk19774ad17132004-05-21 01:47:26 +0000834
danielk19779e128002006-01-18 16:51:35 +0000835 assert( !sqlite3MallocFailed() );
danielk19777ddad962005-12-12 06:53:03 +0000836
danielk19774ad17132004-05-21 01:47:26 +0000837 /* Allocate the sqlite data structure */
drh9bb575f2004-09-06 17:24:11 +0000838 db = sqliteMalloc( sizeof(sqlite3) );
danielk19774ad17132004-05-21 01:47:26 +0000839 if( db==0 ) goto opendb_out;
danielk19774ad17132004-05-21 01:47:26 +0000840 db->priorNewRowid = 0;
841 db->magic = SQLITE_MAGIC_BUSY;
842 db->nDb = 2;
843 db->aDb = db->aDbStatic;
danielk19771d850a72004-05-31 08:26:49 +0000844 db->autoCommit = 1;
drh47a6db22005-01-18 16:02:40 +0000845 db->flags |= SQLITE_ShortColNames;
drhf9b596e2004-05-26 16:54:42 +0000846 sqlite3HashInit(&db->aFunc, SQLITE_HASH_STRING, 0);
danielk19774ad17132004-05-21 01:47:26 +0000847 sqlite3HashInit(&db->aCollSeq, SQLITE_HASH_STRING, 0);
drhb9bb7c12006-06-11 23:41:55 +0000848#ifndef SQLITE_OMIT_VIRTUALTABLE
849 sqlite3HashInit(&db->aModule, SQLITE_HASH_STRING, 0);
850#endif
danielk1977da184232006-01-05 11:34:32 +0000851
danielk19770202b292004-06-09 09:55:16 +0000852 /* Add the default collation sequence BINARY. BINARY works for both UTF-8
853 ** and UTF-16, so add a version for each to avoid any unnecessary
854 ** conversions. The only error that can occur here is a malloc() failure.
855 */
drh0e819f92006-02-01 13:50:41 +0000856 if( createCollation(db, "BINARY", SQLITE_UTF8, 0, binCollFunc) ||
857 createCollation(db, "BINARY", SQLITE_UTF16BE, 0, binCollFunc) ||
858 createCollation(db, "BINARY", SQLITE_UTF16LE, 0, binCollFunc) ||
danielk197714db2662006-01-09 16:12:04 +0000859 (db->pDfltColl = sqlite3FindCollSeq(db, SQLITE_UTF8, "BINARY", 6, 0))==0
danielk19777ddad962005-12-12 06:53:03 +0000860 ){
danielk19779e128002006-01-18 16:51:35 +0000861 assert( sqlite3MallocFailed() );
danielk19770202b292004-06-09 09:55:16 +0000862 db->magic = SQLITE_MAGIC_CLOSED;
863 goto opendb_out;
864 }
865
866 /* Also add a UTF-8 case-insensitive collation sequence. */
danielk19779a30cf62006-01-18 04:26:07 +0000867 createCollation(db, "NOCASE", SQLITE_UTF8, 0, nocaseCollatingFunc);
danielk19770202b292004-06-09 09:55:16 +0000868
drhd64fe2f2005-08-28 17:00:23 +0000869 /* Set flags on the built-in collating sequences */
870 db->pDfltColl->type = SQLITE_COLL_BINARY;
871 pColl = sqlite3FindCollSeq(db, SQLITE_UTF8, "NOCASE", 6, 0);
872 if( pColl ){
873 pColl->type = SQLITE_COLL_NOCASE;
874 }
875
danielk19774ad17132004-05-21 01:47:26 +0000876 /* Open the backend database driver */
danielk19774ad17132004-05-21 01:47:26 +0000877 rc = sqlite3BtreeFactory(db, zFilename, 0, MAX_PAGES, &db->aDb[0].pBt);
878 if( rc!=SQLITE_OK ){
danielk19774ad17132004-05-21 01:47:26 +0000879 sqlite3Error(db, rc, 0);
880 db->magic = SQLITE_MAGIC_CLOSED;
881 goto opendb_out;
882 }
danielk1977da184232006-01-05 11:34:32 +0000883 db->aDb[0].pSchema = sqlite3SchemaGet(db->aDb[0].pBt);
884 db->aDb[1].pSchema = sqlite3SchemaGet(0);
danielk197714db2662006-01-09 16:12:04 +0000885
drh03b808a2006-03-13 15:06:05 +0000886
danielk197753c0f742005-03-29 03:10:59 +0000887 /* The default safety_level for the main database is 'full'; for the temp
888 ** database it is 'NONE'. This matches the pager layer defaults.
889 */
890 db->aDb[0].zName = "main";
danielk197791cf71b2004-06-26 06:37:06 +0000891 db->aDb[0].safety_level = 3;
danielk197753c0f742005-03-29 03:10:59 +0000892#ifndef SQLITE_OMIT_TEMPDB
893 db->aDb[1].zName = "temp";
danielk197791cf71b2004-06-26 06:37:06 +0000894 db->aDb[1].safety_level = 1;
danielk197753c0f742005-03-29 03:10:59 +0000895#endif
896
danielk19778e227872004-06-07 07:52:17 +0000897 /* Register all built-in functions, but do not attempt to read the
898 ** database schema yet. This is delayed until the first time the database
899 ** is accessed.
900 */
danielk19779e128002006-01-18 16:51:35 +0000901 if( !sqlite3MallocFailed() ){
danielk1977771151b2006-01-17 13:21:40 +0000902 sqlite3RegisterBuiltinFunctions(db);
903 sqlite3Error(db, SQLITE_OK, 0);
904 }
danielk19774397de52005-01-12 12:44:03 +0000905 db->magic = SQLITE_MAGIC_OPEN;
danielk19774ad17132004-05-21 01:47:26 +0000906
907opendb_out:
danielk19777ddad962005-12-12 06:53:03 +0000908 if( SQLITE_NOMEM==(rc = sqlite3_errcode(db)) ){
909 sqlite3_close(db);
910 db = 0;
danielk197713073932004-06-30 11:54:06 +0000911 }
danielk19774ad17132004-05-21 01:47:26 +0000912 *ppDb = db;
danielk197754f01982006-01-18 15:25:17 +0000913 return sqlite3ApiExit(0, rc);
danielk19774ad17132004-05-21 01:47:26 +0000914}
915
916/*
917** Open a new database handle.
918*/
danielk197780290862004-05-22 09:21:21 +0000919int sqlite3_open(
danielk19774ad17132004-05-21 01:47:26 +0000920 const char *zFilename,
danielk19774f057f92004-06-08 00:02:33 +0000921 sqlite3 **ppDb
danielk19774ad17132004-05-21 01:47:26 +0000922){
danielk19778e227872004-06-07 07:52:17 +0000923 return openDatabase(zFilename, ppDb);
danielk197783ab5a82004-05-21 11:39:05 +0000924}
925
drh6c626082004-11-14 21:56:29 +0000926#ifndef SQLITE_OMIT_UTF16
danielk19774ad17132004-05-21 01:47:26 +0000927/*
928** Open a new database handle.
929*/
930int sqlite3_open16(
931 const void *zFilename,
danielk19774f057f92004-06-08 00:02:33 +0000932 sqlite3 **ppDb
danielk19774ad17132004-05-21 01:47:26 +0000933){
danielk1977bfd6cce2004-06-18 04:24:54 +0000934 char const *zFilename8; /* zFilename encoded in UTF-8 instead of UTF-16 */
danielk197754f01982006-01-18 15:25:17 +0000935 int rc = SQLITE_OK;
danielk1977bfd6cce2004-06-18 04:24:54 +0000936 sqlite3_value *pVal;
danielk19774ad17132004-05-21 01:47:26 +0000937
danielk19777ddad962005-12-12 06:53:03 +0000938 assert( zFilename );
danielk19774ad17132004-05-21 01:47:26 +0000939 assert( ppDb );
danielk1977bfd6cce2004-06-18 04:24:54 +0000940 *ppDb = 0;
941 pVal = sqlite3ValueNew();
942 sqlite3ValueSetStr(pVal, -1, zFilename, SQLITE_UTF16NATIVE, SQLITE_STATIC);
943 zFilename8 = sqlite3ValueText(pVal, SQLITE_UTF8);
944 if( zFilename8 ){
945 rc = openDatabase(zFilename8, ppDb);
946 if( rc==SQLITE_OK && *ppDb ){
drh7dd90a42005-12-06 13:19:07 +0000947 rc = sqlite3_exec(*ppDb, "PRAGMA encoding = 'UTF-16'", 0, 0, 0);
danielk1977950f0542006-01-18 05:51:57 +0000948 if( rc!=SQLITE_OK ){
949 sqlite3_close(*ppDb);
950 *ppDb = 0;
951 }
danielk1977bfd6cce2004-06-18 04:24:54 +0000952 }
danielk19774ad17132004-05-21 01:47:26 +0000953 }
danielk19777ddad962005-12-12 06:53:03 +0000954 sqlite3ValueFree(pVal);
danielk19778e227872004-06-07 07:52:17 +0000955
danielk197754f01982006-01-18 15:25:17 +0000956 return sqlite3ApiExit(0, rc);
danielk19774ad17132004-05-21 01:47:26 +0000957}
drh6c626082004-11-14 21:56:29 +0000958#endif /* SQLITE_OMIT_UTF16 */
danielk19774ad17132004-05-21 01:47:26 +0000959
danielk1977106bb232004-05-21 10:08:53 +0000960/*
961** The following routine destroys a virtual machine that is created by
962** the sqlite3_compile() routine. The integer returned is an SQLITE_
963** success/failure code that describes the result of executing the virtual
964** machine.
965**
966** This routine sets the error code and string returned by
967** sqlite3_errcode(), sqlite3_errmsg() and sqlite3_errmsg16().
968*/
danielk1977fc57d7b2004-05-26 02:04:57 +0000969int sqlite3_finalize(sqlite3_stmt *pStmt){
drh1bcdb0c2004-08-28 14:49:46 +0000970 int rc;
971 if( pStmt==0 ){
972 rc = SQLITE_OK;
973 }else{
974 rc = sqlite3VdbeFinalize((Vdbe*)pStmt);
975 }
976 return rc;
danielk1977106bb232004-05-21 10:08:53 +0000977}
978
979/*
980** Terminate the current execution of an SQL statement and reset it
981** back to its starting state so that it can be reused. A success code from
982** the prior execution is returned.
983**
984** This routine sets the error code and string returned by
985** sqlite3_errcode(), sqlite3_errmsg() and sqlite3_errmsg16().
986*/
danielk1977fc57d7b2004-05-26 02:04:57 +0000987int sqlite3_reset(sqlite3_stmt *pStmt){
drh1bcdb0c2004-08-28 14:49:46 +0000988 int rc;
989 if( pStmt==0 ){
990 rc = SQLITE_OK;
991 }else{
992 rc = sqlite3VdbeReset((Vdbe*)pStmt);
drh13449892005-09-07 21:22:45 +0000993 sqlite3VdbeMakeReady((Vdbe*)pStmt, -1, 0, 0, 0);
drh1bcdb0c2004-08-28 14:49:46 +0000994 }
danielk1977106bb232004-05-21 10:08:53 +0000995 return rc;
996}
danielk19770202b292004-06-09 09:55:16 +0000997
danielk1977d8123362004-06-12 09:25:12 +0000998/*
999** Register a new collation sequence with the database handle db.
1000*/
danielk19770202b292004-06-09 09:55:16 +00001001int sqlite3_create_collation(
1002 sqlite3* db,
1003 const char *zName,
danielk1977466be562004-06-10 02:16:01 +00001004 int enc,
danielk19770202b292004-06-09 09:55:16 +00001005 void* pCtx,
1006 int(*xCompare)(void*,int,const void*,int,const void*)
1007){
danielk19779a30cf62006-01-18 04:26:07 +00001008 int rc;
danielk19779e128002006-01-18 16:51:35 +00001009 assert( !sqlite3MallocFailed() );
danielk19779a30cf62006-01-18 04:26:07 +00001010 rc = createCollation(db, zName, enc, pCtx, xCompare);
danielk197754f01982006-01-18 15:25:17 +00001011 return sqlite3ApiExit(db, rc);
danielk19770202b292004-06-09 09:55:16 +00001012}
1013
drh6c626082004-11-14 21:56:29 +00001014#ifndef SQLITE_OMIT_UTF16
danielk1977d8123362004-06-12 09:25:12 +00001015/*
1016** Register a new collation sequence with the database handle db.
1017*/
danielk19770202b292004-06-09 09:55:16 +00001018int sqlite3_create_collation16(
1019 sqlite3* db,
1020 const char *zName,
danielk1977466be562004-06-10 02:16:01 +00001021 int enc,
danielk19770202b292004-06-09 09:55:16 +00001022 void* pCtx,
1023 int(*xCompare)(void*,int,const void*,int,const void*)
1024){
danielk19779a30cf62006-01-18 04:26:07 +00001025 int rc = SQLITE_OK;
1026 char *zName8;
danielk19779e128002006-01-18 16:51:35 +00001027 assert( !sqlite3MallocFailed() );
drhaf9a7c22005-12-15 03:04:10 +00001028 zName8 = sqlite3utf16to8(zName, -1);
danielk19779a30cf62006-01-18 04:26:07 +00001029 if( zName8 ){
1030 rc = createCollation(db, zName8, enc, pCtx, xCompare);
1031 sqliteFree(zName8);
1032 }
danielk197754f01982006-01-18 15:25:17 +00001033 return sqlite3ApiExit(db, rc);
danielk19770202b292004-06-09 09:55:16 +00001034}
drh6c626082004-11-14 21:56:29 +00001035#endif /* SQLITE_OMIT_UTF16 */
danielk19777cedc8d2004-06-10 10:50:08 +00001036
danielk1977d8123362004-06-12 09:25:12 +00001037/*
1038** Register a collation sequence factory callback with the database handle
1039** db. Replace any previously installed collation sequence factory.
1040*/
danielk19777cedc8d2004-06-10 10:50:08 +00001041int sqlite3_collation_needed(
1042 sqlite3 *db,
1043 void *pCollNeededArg,
1044 void(*xCollNeeded)(void*,sqlite3*,int eTextRep,const char*)
1045){
drhc60d0442004-09-30 13:43:13 +00001046 if( sqlite3SafetyCheck(db) ){
1047 return SQLITE_MISUSE;
1048 }
danielk19777cedc8d2004-06-10 10:50:08 +00001049 db->xCollNeeded = xCollNeeded;
1050 db->xCollNeeded16 = 0;
1051 db->pCollNeededArg = pCollNeededArg;
1052 return SQLITE_OK;
1053}
danielk1977d8123362004-06-12 09:25:12 +00001054
drh6c626082004-11-14 21:56:29 +00001055#ifndef SQLITE_OMIT_UTF16
danielk1977d8123362004-06-12 09:25:12 +00001056/*
1057** Register a collation sequence factory callback with the database handle
1058** db. Replace any previously installed collation sequence factory.
1059*/
danielk19777cedc8d2004-06-10 10:50:08 +00001060int sqlite3_collation_needed16(
1061 sqlite3 *db,
1062 void *pCollNeededArg,
1063 void(*xCollNeeded16)(void*,sqlite3*,int eTextRep,const void*)
1064){
drhc60d0442004-09-30 13:43:13 +00001065 if( sqlite3SafetyCheck(db) ){
1066 return SQLITE_MISUSE;
1067 }
danielk19777cedc8d2004-06-10 10:50:08 +00001068 db->xCollNeeded = 0;
1069 db->xCollNeeded16 = xCollNeeded16;
1070 db->pCollNeededArg = pCollNeededArg;
1071 return SQLITE_OK;
1072}
drh6c626082004-11-14 21:56:29 +00001073#endif /* SQLITE_OMIT_UTF16 */
danielk19776b456a22005-03-21 04:04:02 +00001074
1075#ifndef SQLITE_OMIT_GLOBALRECOVER
1076/*
danielk19777ddad962005-12-12 06:53:03 +00001077** This function is now an anachronism. It used to be used to recover from a
1078** malloc() failure, but SQLite now does this automatically.
danielk19776b456a22005-03-21 04:04:02 +00001079*/
1080int sqlite3_global_recover(){
danielk1977261919c2005-12-06 12:52:59 +00001081 return SQLITE_OK;
danielk19776b456a22005-03-21 04:04:02 +00001082}
1083#endif
drh3e1d8e62005-05-26 16:23:34 +00001084
1085/*
1086** Test to see whether or not the database connection is in autocommit
1087** mode. Return TRUE if it is and FALSE if not. Autocommit mode is on
1088** by default. Autocommit is disabled by a BEGIN statement and reenabled
1089** by the next COMMIT or ROLLBACK.
1090**
1091******* THIS IS AN EXPERIMENTAL API AND IS SUBJECT TO CHANGE ******
1092*/
1093int sqlite3_get_autocommit(sqlite3 *db){
1094 return db->autoCommit;
1095}
drh49285702005-09-17 15:20:26 +00001096
1097#ifdef SQLITE_DEBUG
1098/*
1099** The following routine is subtituted for constant SQLITE_CORRUPT in
1100** debugging builds. This provides a way to set a breakpoint for when
1101** corruption is first detected.
1102*/
1103int sqlite3Corrupt(void){
1104 return SQLITE_CORRUPT;
1105}
1106#endif
drh7c1817e2006-01-10 13:58:48 +00001107
1108
1109#ifndef SQLITE_OMIT_SHARED_CACHE
1110/*
1111** Enable or disable the shared pager and schema features for the
1112** current thread.
1113**
1114** This routine should only be called when there are no open
1115** database connections.
1116*/
1117int sqlite3_enable_shared_cache(int enable){
1118 ThreadData *pTd = sqlite3ThreadData();
danielk1977f2fa8312006-01-24 13:09:33 +00001119 if( pTd ){
1120 /* It is only legal to call sqlite3_enable_shared_cache() when there
1121 ** are no currently open b-trees that were opened by the calling thread.
1122 ** This condition is only easy to detect if the shared-cache were
1123 ** previously enabled (and is being disabled).
1124 */
1125 if( pTd->pBtree && !enable ){
1126 assert( pTd->useSharedData );
1127 return SQLITE_MISUSE;
1128 }
drh7c1817e2006-01-10 13:58:48 +00001129
danielk1977f2fa8312006-01-24 13:09:33 +00001130 pTd->useSharedData = enable;
1131 sqlite3ReleaseThreadData();
1132 }
1133 return sqlite3ApiExit(0, SQLITE_OK);
drh7c1817e2006-01-10 13:58:48 +00001134}
1135#endif
drh6f7adc82006-01-11 21:41:20 +00001136
1137/*
1138** This is a convenience routine that makes sure that all thread-specific
1139** data for this thread has been deallocated.
1140*/
1141void sqlite3_thread_cleanup(void){
drhf98477b2006-01-19 17:42:50 +00001142 ThreadData *pTd = sqlite3OsThreadSpecificData(0);
danielk197754f01982006-01-18 15:25:17 +00001143 if( pTd ){
1144 memset(pTd, 0, sizeof(*pTd));
drhf98477b2006-01-19 17:42:50 +00001145 sqlite3OsThreadSpecificData(-1);
danielk197754f01982006-01-18 15:25:17 +00001146 }
drh6f7adc82006-01-11 21:41:20 +00001147}
danielk1977deb802c2006-02-09 13:43:28 +00001148
1149/*
1150** Return meta information about a specific column of a database table.
1151** See comment in sqlite3.h (sqlite.h.in) for details.
1152*/
1153#ifdef SQLITE_ENABLE_COLUMN_METADATA
1154int sqlite3_table_column_metadata(
1155 sqlite3 *db, /* Connection handle */
1156 const char *zDbName, /* Database name or NULL */
1157 const char *zTableName, /* Table name */
1158 const char *zColumnName, /* Column name */
1159 char const **pzDataType, /* OUTPUT: Declared data type */
1160 char const **pzCollSeq, /* OUTPUT: Collation sequence name */
1161 int *pNotNull, /* OUTPUT: True if NOT NULL constraint exists */
1162 int *pPrimaryKey, /* OUTPUT: True if column part of PK */
1163 int *pAutoinc /* OUTPUT: True if colums is auto-increment */
1164){
1165 int rc;
1166 char *zErrMsg = 0;
1167 Table *pTab = 0;
1168 Column *pCol = 0;
1169 int iCol;
1170
1171 char const *zDataType = 0;
1172 char const *zCollSeq = 0;
1173 int notnull = 0;
1174 int primarykey = 0;
1175 int autoinc = 0;
1176
1177 /* Ensure the database schema has been loaded */
1178 if( sqlite3SafetyOn(db) ){
1179 return SQLITE_MISUSE;
1180 }
1181 rc = sqlite3Init(db, &zErrMsg);
1182 if( SQLITE_OK!=rc ){
1183 goto error_out;
1184 }
1185
1186 /* Locate the table in question */
1187 pTab = sqlite3FindTable(db, zTableName, zDbName);
1188 if( !pTab || pTab->pSelect ){
1189 pTab = 0;
1190 goto error_out;
1191 }
1192
1193 /* Find the column for which info is requested */
1194 if( sqlite3IsRowid(zColumnName) ){
1195 iCol = pTab->iPKey;
1196 if( iCol>=0 ){
1197 pCol = &pTab->aCol[iCol];
1198 }
1199 }else{
1200 for(iCol=0; iCol<pTab->nCol; iCol++){
1201 pCol = &pTab->aCol[iCol];
1202 if( 0==sqlite3StrICmp(pCol->zName, zColumnName) ){
1203 break;
1204 }
1205 }
1206 if( iCol==pTab->nCol ){
1207 pTab = 0;
1208 goto error_out;
1209 }
1210 }
1211
1212 /* The following block stores the meta information that will be returned
1213 ** to the caller in local variables zDataType, zCollSeq, notnull, primarykey
1214 ** and autoinc. At this point there are two possibilities:
1215 **
1216 ** 1. The specified column name was rowid", "oid" or "_rowid_"
1217 ** and there is no explicitly declared IPK column.
1218 **
1219 ** 2. The table is not a view and the column name identified an
1220 ** explicitly declared column. Copy meta information from *pCol.
1221 */
1222 if( pCol ){
1223 zDataType = pCol->zType;
1224 zCollSeq = pCol->zColl;
1225 notnull = (pCol->notNull?1:0);
1226 primarykey = (pCol->isPrimKey?1:0);
1227 autoinc = ((pTab->iPKey==iCol && pTab->autoInc)?1:0);
1228 }else{
1229 zDataType = "INTEGER";
1230 primarykey = 1;
1231 }
1232 if( !zCollSeq ){
1233 zCollSeq = "BINARY";
1234 }
1235
1236error_out:
1237 if( sqlite3SafetyOff(db) ){
1238 rc = SQLITE_MISUSE;
1239 }
1240
1241 /* Whether the function call succeeded or failed, set the output parameters
1242 ** to whatever their local counterparts contain. If an error did occur,
1243 ** this has the effect of zeroing all output parameters.
1244 */
1245 if( pzDataType ) *pzDataType = zDataType;
1246 if( pzCollSeq ) *pzCollSeq = zCollSeq;
1247 if( pNotNull ) *pNotNull = notnull;
1248 if( pPrimaryKey ) *pPrimaryKey = primarykey;
1249 if( pAutoinc ) *pAutoinc = autoinc;
1250
1251 if( SQLITE_OK==rc && !pTab ){
1252 sqlite3SetString(&zErrMsg, "no such table column: ", zTableName, ".",
1253 zColumnName, 0);
1254 rc = SQLITE_ERROR;
1255 }
1256 sqlite3Error(db, rc, (zErrMsg?"%s":0), zErrMsg);
1257 sqliteFree(zErrMsg);
1258 return sqlite3ApiExit(db, rc);
1259}
1260#endif