• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 /*
2 ** 2001 September 15
3 **
4 ** The author disclaims copyright to this source code.  In place of
5 ** a legal notice, here is a blessing:
6 **
7 **    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.
10 **
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 */
17 #include "sqliteInt.h"
18 
19 #ifdef SQLITE_ENABLE_FTS3
20 # include "fts3.h"
21 #endif
22 #ifdef SQLITE_ENABLE_RTREE
23 # include "rtree.h"
24 #endif
25 #ifdef SQLITE_ENABLE_ICU
26 # include "sqliteicu.h"
27 #endif
28 
29 #ifndef SQLITE_AMALGAMATION
30 /* IMPLEMENTATION-OF: R-46656-45156 The sqlite3_version[] string constant
31 ** contains the text of SQLITE_VERSION macro.
32 */
33 const char sqlite3_version[] = SQLITE_VERSION;
34 #endif
35 
36 /* IMPLEMENTATION-OF: R-53536-42575 The sqlite3_libversion() function returns
37 ** a pointer to the to the sqlite3_version[] string constant.
38 */
sqlite3_libversion(void)39 const char *sqlite3_libversion(void){ return sqlite3_version; }
40 
41 /* IMPLEMENTATION-OF: R-63124-39300 The sqlite3_sourceid() function returns a
42 ** pointer to a string constant whose value is the same as the
43 ** SQLITE_SOURCE_ID C preprocessor macro.
44 */
sqlite3_sourceid(void)45 const char *sqlite3_sourceid(void){ return SQLITE_SOURCE_ID; }
46 
47 /* IMPLEMENTATION-OF: R-35210-63508 The sqlite3_libversion_number() function
48 ** returns an integer equal to SQLITE_VERSION_NUMBER.
49 */
sqlite3_libversion_number(void)50 int sqlite3_libversion_number(void){ return SQLITE_VERSION_NUMBER; }
51 
52 /* IMPLEMENTATION-OF: R-54823-41343 The sqlite3_threadsafe() function returns
53 ** zero if and only if SQLite was compiled mutexing code omitted due to
54 ** the SQLITE_THREADSAFE compile-time option being set to 0.
55 */
sqlite3_threadsafe(void)56 int sqlite3_threadsafe(void){ return SQLITE_THREADSAFE; }
57 
58 #if !defined(SQLITE_OMIT_TRACE) && defined(SQLITE_ENABLE_IOTRACE)
59 /*
60 ** If the following function pointer is not NULL and if
61 ** SQLITE_ENABLE_IOTRACE is enabled, then messages describing
62 ** I/O active are written using this function.  These messages
63 ** are intended for debugging activity only.
64 */
65 void (*sqlite3IoTrace)(const char*, ...) = 0;
66 #endif
67 
68 /*
69 ** If the following global variable points to a string which is the
70 ** name of a directory, then that directory will be used to store
71 ** temporary files.
72 **
73 ** See also the "PRAGMA temp_store_directory" SQL command.
74 */
75 char *sqlite3_temp_directory = 0;
76 
77 /*
78 ** Initialize SQLite.
79 **
80 ** This routine must be called to initialize the memory allocation,
81 ** VFS, and mutex subsystems prior to doing any serious work with
82 ** SQLite.  But as long as you do not compile with SQLITE_OMIT_AUTOINIT
83 ** this routine will be called automatically by key routines such as
84 ** sqlite3_open().
85 **
86 ** This routine is a no-op except on its very first call for the process,
87 ** or for the first call after a call to sqlite3_shutdown.
88 **
89 ** The first thread to call this routine runs the initialization to
90 ** completion.  If subsequent threads call this routine before the first
91 ** thread has finished the initialization process, then the subsequent
92 ** threads must block until the first thread finishes with the initialization.
93 **
94 ** The first thread might call this routine recursively.  Recursive
95 ** calls to this routine should not block, of course.  Otherwise the
96 ** initialization process would never complete.
97 **
98 ** Let X be the first thread to enter this routine.  Let Y be some other
99 ** thread.  Then while the initial invocation of this routine by X is
100 ** incomplete, it is required that:
101 **
102 **    *  Calls to this routine from Y must block until the outer-most
103 **       call by X completes.
104 **
105 **    *  Recursive calls to this routine from thread X return immediately
106 **       without blocking.
107 */
sqlite3_initialize(void)108 int sqlite3_initialize(void){
109   sqlite3_mutex *pMaster;                      /* The main static mutex */
110   int rc;                                      /* Result code */
111 
112 #ifdef SQLITE_OMIT_WSD
113   rc = sqlite3_wsd_init(4096, 24);
114   if( rc!=SQLITE_OK ){
115     return rc;
116   }
117 #endif
118 
119   /* If SQLite is already completely initialized, then this call
120   ** to sqlite3_initialize() should be a no-op.  But the initialization
121   ** must be complete.  So isInit must not be set until the very end
122   ** of this routine.
123   */
124   if( sqlite3GlobalConfig.isInit ) return SQLITE_OK;
125 
126   /* Make sure the mutex subsystem is initialized.  If unable to
127   ** initialize the mutex subsystem, return early with the error.
128   ** If the system is so sick that we are unable to allocate a mutex,
129   ** there is not much SQLite is going to be able to do.
130   **
131   ** The mutex subsystem must take care of serializing its own
132   ** initialization.
133   */
134   rc = sqlite3MutexInit();
135   if( rc ) return rc;
136 
137   /* Initialize the malloc() system and the recursive pInitMutex mutex.
138   ** This operation is protected by the STATIC_MASTER mutex.  Note that
139   ** MutexAlloc() is called for a static mutex prior to initializing the
140   ** malloc subsystem - this implies that the allocation of a static
141   ** mutex must not require support from the malloc subsystem.
142   */
143   pMaster = sqlite3MutexAlloc(SQLITE_MUTEX_STATIC_MASTER);
144   sqlite3_mutex_enter(pMaster);
145   sqlite3GlobalConfig.isMutexInit = 1;
146   if( !sqlite3GlobalConfig.isMallocInit ){
147     rc = sqlite3MallocInit();
148   }
149   if( rc==SQLITE_OK ){
150     sqlite3GlobalConfig.isMallocInit = 1;
151     if( !sqlite3GlobalConfig.pInitMutex ){
152       sqlite3GlobalConfig.pInitMutex =
153            sqlite3MutexAlloc(SQLITE_MUTEX_RECURSIVE);
154       if( sqlite3GlobalConfig.bCoreMutex && !sqlite3GlobalConfig.pInitMutex ){
155         rc = SQLITE_NOMEM;
156       }
157     }
158   }
159   if( rc==SQLITE_OK ){
160     sqlite3GlobalConfig.nRefInitMutex++;
161   }
162   sqlite3_mutex_leave(pMaster);
163 
164   /* If rc is not SQLITE_OK at this point, then either the malloc
165   ** subsystem could not be initialized or the system failed to allocate
166   ** the pInitMutex mutex. Return an error in either case.  */
167   if( rc!=SQLITE_OK ){
168     return rc;
169   }
170 
171   /* Do the rest of the initialization under the recursive mutex so
172   ** that we will be able to handle recursive calls into
173   ** sqlite3_initialize().  The recursive calls normally come through
174   ** sqlite3_os_init() when it invokes sqlite3_vfs_register(), but other
175   ** recursive calls might also be possible.
176   **
177   ** IMPLEMENTATION-OF: R-00140-37445 SQLite automatically serializes calls
178   ** to the xInit method, so the xInit method need not be threadsafe.
179   **
180   ** The following mutex is what serializes access to the appdef pcache xInit
181   ** methods.  The sqlite3_pcache_methods.xInit() all is embedded in the
182   ** call to sqlite3PcacheInitialize().
183   */
184   sqlite3_mutex_enter(sqlite3GlobalConfig.pInitMutex);
185   if( sqlite3GlobalConfig.isInit==0 && sqlite3GlobalConfig.inProgress==0 ){
186     FuncDefHash *pHash = &GLOBAL(FuncDefHash, sqlite3GlobalFunctions);
187     sqlite3GlobalConfig.inProgress = 1;
188     memset(pHash, 0, sizeof(sqlite3GlobalFunctions));
189     sqlite3RegisterGlobalFunctions();
190     if( sqlite3GlobalConfig.isPCacheInit==0 ){
191       rc = sqlite3PcacheInitialize();
192     }
193     if( rc==SQLITE_OK ){
194       sqlite3GlobalConfig.isPCacheInit = 1;
195       rc = sqlite3OsInit();
196     }
197     if( rc==SQLITE_OK ){
198       sqlite3PCacheBufferSetup( sqlite3GlobalConfig.pPage,
199           sqlite3GlobalConfig.szPage, sqlite3GlobalConfig.nPage);
200       sqlite3GlobalConfig.isInit = 1;
201     }
202     sqlite3GlobalConfig.inProgress = 0;
203   }
204   sqlite3_mutex_leave(sqlite3GlobalConfig.pInitMutex);
205 
206   /* Go back under the static mutex and clean up the recursive
207   ** mutex to prevent a resource leak.
208   */
209   sqlite3_mutex_enter(pMaster);
210   sqlite3GlobalConfig.nRefInitMutex--;
211   if( sqlite3GlobalConfig.nRefInitMutex<=0 ){
212     assert( sqlite3GlobalConfig.nRefInitMutex==0 );
213     sqlite3_mutex_free(sqlite3GlobalConfig.pInitMutex);
214     sqlite3GlobalConfig.pInitMutex = 0;
215   }
216   sqlite3_mutex_leave(pMaster);
217 
218   /* The following is just a sanity check to make sure SQLite has
219   ** been compiled correctly.  It is important to run this code, but
220   ** we don't want to run it too often and soak up CPU cycles for no
221   ** reason.  So we run it once during initialization.
222   */
223 #ifndef NDEBUG
224 #ifndef SQLITE_OMIT_FLOATING_POINT
225   /* This section of code's only "output" is via assert() statements. */
226   if ( rc==SQLITE_OK ){
227     u64 x = (((u64)1)<<63)-1;
228     double y;
229     assert(sizeof(x)==8);
230     assert(sizeof(x)==sizeof(y));
231     memcpy(&y, &x, 8);
232     assert( sqlite3IsNaN(y) );
233   }
234 #endif
235 #endif
236 
237   return rc;
238 }
239 
240 /*
241 ** Undo the effects of sqlite3_initialize().  Must not be called while
242 ** there are outstanding database connections or memory allocations or
243 ** while any part of SQLite is otherwise in use in any thread.  This
244 ** routine is not threadsafe.  But it is safe to invoke this routine
245 ** on when SQLite is already shut down.  If SQLite is already shut down
246 ** when this routine is invoked, then this routine is a harmless no-op.
247 */
sqlite3_shutdown(void)248 int sqlite3_shutdown(void){
249   if( sqlite3GlobalConfig.isInit ){
250     sqlite3_os_end();
251     sqlite3_reset_auto_extension();
252     sqlite3GlobalConfig.isInit = 0;
253   }
254   if( sqlite3GlobalConfig.isPCacheInit ){
255     sqlite3PcacheShutdown();
256     sqlite3GlobalConfig.isPCacheInit = 0;
257   }
258   if( sqlite3GlobalConfig.isMallocInit ){
259     sqlite3MallocEnd();
260     sqlite3GlobalConfig.isMallocInit = 0;
261   }
262   if( sqlite3GlobalConfig.isMutexInit ){
263     sqlite3MutexEnd();
264     sqlite3GlobalConfig.isMutexInit = 0;
265   }
266 
267   return SQLITE_OK;
268 }
269 
270 /*
271 ** This API allows applications to modify the global configuration of
272 ** the SQLite library at run-time.
273 **
274 ** This routine should only be called when there are no outstanding
275 ** database connections or memory allocations.  This routine is not
276 ** threadsafe.  Failure to heed these warnings can lead to unpredictable
277 ** behavior.
278 */
sqlite3_config(int op,...)279 int sqlite3_config(int op, ...){
280   va_list ap;
281   int rc = SQLITE_OK;
282 
283   /* sqlite3_config() shall return SQLITE_MISUSE if it is invoked while
284   ** the SQLite library is in use. */
285   if( sqlite3GlobalConfig.isInit ) return SQLITE_MISUSE_BKPT;
286 
287   va_start(ap, op);
288   switch( op ){
289 
290     /* Mutex configuration options are only available in a threadsafe
291     ** compile.
292     */
293 #if defined(SQLITE_THREADSAFE) && SQLITE_THREADSAFE>0
294     case SQLITE_CONFIG_SINGLETHREAD: {
295       /* Disable all mutexing */
296       sqlite3GlobalConfig.bCoreMutex = 0;
297       sqlite3GlobalConfig.bFullMutex = 0;
298       break;
299     }
300     case SQLITE_CONFIG_MULTITHREAD: {
301       /* Disable mutexing of database connections */
302       /* Enable mutexing of core data structures */
303       sqlite3GlobalConfig.bCoreMutex = 1;
304       sqlite3GlobalConfig.bFullMutex = 0;
305       break;
306     }
307     case SQLITE_CONFIG_SERIALIZED: {
308       /* Enable all mutexing */
309       sqlite3GlobalConfig.bCoreMutex = 1;
310       sqlite3GlobalConfig.bFullMutex = 1;
311       break;
312     }
313     case SQLITE_CONFIG_MUTEX: {
314       /* Specify an alternative mutex implementation */
315       sqlite3GlobalConfig.mutex = *va_arg(ap, sqlite3_mutex_methods*);
316       break;
317     }
318     case SQLITE_CONFIG_GETMUTEX: {
319       /* Retrieve the current mutex implementation */
320       *va_arg(ap, sqlite3_mutex_methods*) = sqlite3GlobalConfig.mutex;
321       break;
322     }
323 #endif
324 
325 
326     case SQLITE_CONFIG_MALLOC: {
327       /* Specify an alternative malloc implementation */
328       sqlite3GlobalConfig.m = *va_arg(ap, sqlite3_mem_methods*);
329       break;
330     }
331     case SQLITE_CONFIG_GETMALLOC: {
332       /* Retrieve the current malloc() implementation */
333       if( sqlite3GlobalConfig.m.xMalloc==0 ) sqlite3MemSetDefault();
334       *va_arg(ap, sqlite3_mem_methods*) = sqlite3GlobalConfig.m;
335       break;
336     }
337     case SQLITE_CONFIG_MEMSTATUS: {
338       /* Enable or disable the malloc status collection */
339       sqlite3GlobalConfig.bMemstat = va_arg(ap, int);
340       break;
341     }
342     case SQLITE_CONFIG_SCRATCH: {
343       /* Designate a buffer for scratch memory space */
344       sqlite3GlobalConfig.pScratch = va_arg(ap, void*);
345       sqlite3GlobalConfig.szScratch = va_arg(ap, int);
346       sqlite3GlobalConfig.nScratch = va_arg(ap, int);
347       break;
348     }
349     case SQLITE_CONFIG_PAGECACHE: {
350       /* Designate a buffer for page cache memory space */
351       sqlite3GlobalConfig.pPage = va_arg(ap, void*);
352       sqlite3GlobalConfig.szPage = va_arg(ap, int);
353       sqlite3GlobalConfig.nPage = va_arg(ap, int);
354       break;
355     }
356 
357     case SQLITE_CONFIG_PCACHE: {
358       /* Specify an alternative page cache implementation */
359       sqlite3GlobalConfig.pcache = *va_arg(ap, sqlite3_pcache_methods*);
360       break;
361     }
362 
363     case SQLITE_CONFIG_GETPCACHE: {
364       if( sqlite3GlobalConfig.pcache.xInit==0 ){
365         sqlite3PCacheSetDefault();
366       }
367       *va_arg(ap, sqlite3_pcache_methods*) = sqlite3GlobalConfig.pcache;
368       break;
369     }
370 
371 #if defined(SQLITE_ENABLE_MEMSYS3) || defined(SQLITE_ENABLE_MEMSYS5)
372     case SQLITE_CONFIG_HEAP: {
373       /* Designate a buffer for heap memory space */
374       sqlite3GlobalConfig.pHeap = va_arg(ap, void*);
375       sqlite3GlobalConfig.nHeap = va_arg(ap, int);
376       sqlite3GlobalConfig.mnReq = va_arg(ap, int);
377 
378       if( sqlite3GlobalConfig.mnReq<1 ){
379         sqlite3GlobalConfig.mnReq = 1;
380       }else if( sqlite3GlobalConfig.mnReq>(1<<12) ){
381         /* cap min request size at 2^12 */
382         sqlite3GlobalConfig.mnReq = (1<<12);
383       }
384 
385       if( sqlite3GlobalConfig.pHeap==0 ){
386         /* If the heap pointer is NULL, then restore the malloc implementation
387         ** back to NULL pointers too.  This will cause the malloc to go
388         ** back to its default implementation when sqlite3_initialize() is
389         ** run.
390         */
391         memset(&sqlite3GlobalConfig.m, 0, sizeof(sqlite3GlobalConfig.m));
392       }else{
393         /* The heap pointer is not NULL, then install one of the
394         ** mem5.c/mem3.c methods. If neither ENABLE_MEMSYS3 nor
395         ** ENABLE_MEMSYS5 is defined, return an error.
396         */
397 #ifdef SQLITE_ENABLE_MEMSYS3
398         sqlite3GlobalConfig.m = *sqlite3MemGetMemsys3();
399 #endif
400 #ifdef SQLITE_ENABLE_MEMSYS5
401         sqlite3GlobalConfig.m = *sqlite3MemGetMemsys5();
402 #endif
403       }
404       break;
405     }
406 #endif
407 
408     case SQLITE_CONFIG_LOOKASIDE: {
409       sqlite3GlobalConfig.szLookaside = va_arg(ap, int);
410       sqlite3GlobalConfig.nLookaside = va_arg(ap, int);
411       break;
412     }
413 
414     /* Record a pointer to the logger funcction and its first argument.
415     ** The default is NULL.  Logging is disabled if the function pointer is
416     ** NULL.
417     */
418     case SQLITE_CONFIG_LOG: {
419       /* MSVC is picky about pulling func ptrs from va lists.
420       ** http://support.microsoft.com/kb/47961
421       ** sqlite3GlobalConfig.xLog = va_arg(ap, void(*)(void*,int,const char*));
422       */
423       typedef void(*LOGFUNC_t)(void*,int,const char*);
424       sqlite3GlobalConfig.xLog = va_arg(ap, LOGFUNC_t);
425       sqlite3GlobalConfig.pLogArg = va_arg(ap, void*);
426       break;
427     }
428 
429     default: {
430       rc = SQLITE_ERROR;
431       break;
432     }
433   }
434   va_end(ap);
435   return rc;
436 }
437 
438 /*
439 ** Set up the lookaside buffers for a database connection.
440 ** Return SQLITE_OK on success.
441 ** If lookaside is already active, return SQLITE_BUSY.
442 **
443 ** The sz parameter is the number of bytes in each lookaside slot.
444 ** The cnt parameter is the number of slots.  If pStart is NULL the
445 ** space for the lookaside memory is obtained from sqlite3_malloc().
446 ** If pStart is not NULL then it is sz*cnt bytes of memory to use for
447 ** the lookaside memory.
448 */
setupLookaside(sqlite3 * db,void * pBuf,int sz,int cnt)449 static int setupLookaside(sqlite3 *db, void *pBuf, int sz, int cnt){
450   void *pStart;
451   if( db->lookaside.nOut ){
452     return SQLITE_BUSY;
453   }
454   /* Free any existing lookaside buffer for this handle before
455   ** allocating a new one so we don't have to have space for
456   ** both at the same time.
457   */
458   if( db->lookaside.bMalloced ){
459     sqlite3_free(db->lookaside.pStart);
460   }
461   /* The size of a lookaside slot needs to be larger than a pointer
462   ** to be useful.
463   */
464   if( sz<=(int)sizeof(LookasideSlot*) ) sz = 0;
465   if( cnt<0 ) cnt = 0;
466   if( sz==0 || cnt==0 ){
467     sz = 0;
468     pStart = 0;
469   }else if( pBuf==0 ){
470     sz = ROUNDDOWN8(sz); /* IMP: R-33038-09382 */
471     sqlite3BeginBenignMalloc();
472     pStart = sqlite3Malloc( sz*cnt );  /* IMP: R-61949-35727 */
473     sqlite3EndBenignMalloc();
474   }else{
475     sz = ROUNDDOWN8(sz); /* IMP: R-33038-09382 */
476     pStart = pBuf;
477   }
478   db->lookaside.pStart = pStart;
479   db->lookaside.pFree = 0;
480   db->lookaside.sz = (u16)sz;
481   if( pStart ){
482     int i;
483     LookasideSlot *p;
484     assert( sz > (int)sizeof(LookasideSlot*) );
485     p = (LookasideSlot*)pStart;
486     for(i=cnt-1; i>=0; i--){
487       p->pNext = db->lookaside.pFree;
488       db->lookaside.pFree = p;
489       p = (LookasideSlot*)&((u8*)p)[sz];
490     }
491     db->lookaside.pEnd = p;
492     db->lookaside.bEnabled = 1;
493     db->lookaside.bMalloced = pBuf==0 ?1:0;
494   }else{
495     db->lookaside.pEnd = 0;
496     db->lookaside.bEnabled = 0;
497     db->lookaside.bMalloced = 0;
498   }
499   return SQLITE_OK;
500 }
501 
502 /*
503 ** Return the mutex associated with a database connection.
504 */
sqlite3_db_mutex(sqlite3 * db)505 sqlite3_mutex *sqlite3_db_mutex(sqlite3 *db){
506   return db->mutex;
507 }
508 
509 /*
510 ** Configuration settings for an individual database connection
511 */
sqlite3_db_config(sqlite3 * db,int op,...)512 int sqlite3_db_config(sqlite3 *db, int op, ...){
513   va_list ap;
514   int rc;
515   va_start(ap, op);
516   switch( op ){
517     case SQLITE_DBCONFIG_LOOKASIDE: {
518       void *pBuf = va_arg(ap, void*); /* IMP: R-26835-10964 */
519       int sz = va_arg(ap, int);       /* IMP: R-47871-25994 */
520       int cnt = va_arg(ap, int);      /* IMP: R-04460-53386 */
521       rc = setupLookaside(db, pBuf, sz, cnt);
522       break;
523     }
524     default: {
525       static const struct {
526         int op;      /* The opcode */
527         u32 mask;    /* Mask of the bit in sqlite3.flags to set/clear */
528       } aFlagOp[] = {
529         { SQLITE_DBCONFIG_ENABLE_FKEY,    SQLITE_ForeignKeys    },
530         { SQLITE_DBCONFIG_ENABLE_TRIGGER, SQLITE_EnableTrigger  },
531       };
532       unsigned int i;
533       rc = SQLITE_ERROR; /* IMP: R-42790-23372 */
534       for(i=0; i<ArraySize(aFlagOp); i++){
535         if( aFlagOp[i].op==op ){
536           int onoff = va_arg(ap, int);
537           int *pRes = va_arg(ap, int*);
538           int oldFlags = db->flags;
539           if( onoff>0 ){
540             db->flags |= aFlagOp[i].mask;
541           }else if( onoff==0 ){
542             db->flags &= ~aFlagOp[i].mask;
543           }
544           if( oldFlags!=db->flags ){
545             sqlite3ExpirePreparedStatements(db);
546           }
547           if( pRes ){
548             *pRes = (db->flags & aFlagOp[i].mask)!=0;
549           }
550           rc = SQLITE_OK;
551           break;
552         }
553       }
554       break;
555     }
556   }
557   va_end(ap);
558   return rc;
559 }
560 
561 
562 /*
563 ** Return true if the buffer z[0..n-1] contains all spaces.
564 */
allSpaces(const char * z,int n)565 static int allSpaces(const char *z, int n){
566   while( n>0 && z[n-1]==' ' ){ n--; }
567   return n==0;
568 }
569 
570 /*
571 ** This is the default collating function named "BINARY" which is always
572 ** available.
573 **
574 ** If the padFlag argument is not NULL then space padding at the end
575 ** of strings is ignored.  This implements the RTRIM collation.
576 */
binCollFunc(void * padFlag,int nKey1,const void * pKey1,int nKey2,const void * pKey2)577 static int binCollFunc(
578   void *padFlag,
579   int nKey1, const void *pKey1,
580   int nKey2, const void *pKey2
581 ){
582   int rc, n;
583   n = nKey1<nKey2 ? nKey1 : nKey2;
584   rc = memcmp(pKey1, pKey2, n);
585   if( rc==0 ){
586     if( padFlag
587      && allSpaces(((char*)pKey1)+n, nKey1-n)
588      && allSpaces(((char*)pKey2)+n, nKey2-n)
589     ){
590       /* Leave rc unchanged at 0 */
591     }else{
592       rc = nKey1 - nKey2;
593     }
594   }
595   return rc;
596 }
597 
598 /*
599 ** Another built-in collating sequence: NOCASE.
600 **
601 ** This collating sequence is intended to be used for "case independant
602 ** comparison". SQLite's knowledge of upper and lower case equivalents
603 ** extends only to the 26 characters used in the English language.
604 **
605 ** At the moment there is only a UTF-8 implementation.
606 */
nocaseCollatingFunc(void * NotUsed,int nKey1,const void * pKey1,int nKey2,const void * pKey2)607 static int nocaseCollatingFunc(
608   void *NotUsed,
609   int nKey1, const void *pKey1,
610   int nKey2, const void *pKey2
611 ){
612   int r = sqlite3StrNICmp(
613       (const char *)pKey1, (const char *)pKey2, (nKey1<nKey2)?nKey1:nKey2);
614   UNUSED_PARAMETER(NotUsed);
615   if( 0==r ){
616     r = nKey1-nKey2;
617   }
618   return r;
619 }
620 
621 /*
622 ** Return the ROWID of the most recent insert
623 */
sqlite3_last_insert_rowid(sqlite3 * db)624 sqlite_int64 sqlite3_last_insert_rowid(sqlite3 *db){
625   return db->lastRowid;
626 }
627 
628 /*
629 ** Return the number of changes in the most recent call to sqlite3_exec().
630 */
sqlite3_changes(sqlite3 * db)631 int sqlite3_changes(sqlite3 *db){
632   return db->nChange;
633 }
634 
635 /*
636 ** Return the number of changes since the database handle was opened.
637 */
sqlite3_total_changes(sqlite3 * db)638 int sqlite3_total_changes(sqlite3 *db){
639   return db->nTotalChange;
640 }
641 
642 /*
643 ** Close all open savepoints. This function only manipulates fields of the
644 ** database handle object, it does not close any savepoints that may be open
645 ** at the b-tree/pager level.
646 */
sqlite3CloseSavepoints(sqlite3 * db)647 void sqlite3CloseSavepoints(sqlite3 *db){
648   while( db->pSavepoint ){
649     Savepoint *pTmp = db->pSavepoint;
650     db->pSavepoint = pTmp->pNext;
651     sqlite3DbFree(db, pTmp);
652   }
653   db->nSavepoint = 0;
654   db->nStatement = 0;
655   db->isTransactionSavepoint = 0;
656 }
657 
658 /*
659 ** Invoke the destructor function associated with FuncDef p, if any. Except,
660 ** if this is not the last copy of the function, do not invoke it. Multiple
661 ** copies of a single function are created when create_function() is called
662 ** with SQLITE_ANY as the encoding.
663 */
functionDestroy(sqlite3 * db,FuncDef * p)664 static void functionDestroy(sqlite3 *db, FuncDef *p){
665   FuncDestructor *pDestructor = p->pDestructor;
666   if( pDestructor ){
667     pDestructor->nRef--;
668     if( pDestructor->nRef==0 ){
669       pDestructor->xDestroy(pDestructor->pUserData);
670       sqlite3DbFree(db, pDestructor);
671     }
672   }
673 }
674 
675 /*
676 ** Close an existing SQLite database
677 */
sqlite3_close(sqlite3 * db)678 int sqlite3_close(sqlite3 *db){
679   HashElem *i;                    /* Hash table iterator */
680   int j;
681 
682   if( !db ){
683     return SQLITE_OK;
684   }
685   if( !sqlite3SafetyCheckSickOrOk(db) ){
686     return SQLITE_MISUSE_BKPT;
687   }
688   sqlite3_mutex_enter(db->mutex);
689 
690   /* Force xDestroy calls on all virtual tables */
691   sqlite3ResetInternalSchema(db, -1);
692 
693   /* If a transaction is open, the ResetInternalSchema() call above
694   ** will not have called the xDisconnect() method on any virtual
695   ** tables in the db->aVTrans[] array. The following sqlite3VtabRollback()
696   ** call will do so. We need to do this before the check for active
697   ** SQL statements below, as the v-table implementation may be storing
698   ** some prepared statements internally.
699   */
700   sqlite3VtabRollback(db);
701 
702   /* If there are any outstanding VMs, return SQLITE_BUSY. */
703   if( db->pVdbe ){
704     sqlite3Error(db, SQLITE_BUSY,
705         "unable to close due to unfinalised statements");
706     sqlite3_mutex_leave(db->mutex);
707     return SQLITE_BUSY;
708   }
709   assert( sqlite3SafetyCheckSickOrOk(db) );
710 
711   for(j=0; j<db->nDb; j++){
712     Btree *pBt = db->aDb[j].pBt;
713     if( pBt && sqlite3BtreeIsInBackup(pBt) ){
714       sqlite3Error(db, SQLITE_BUSY,
715           "unable to close due to unfinished backup operation");
716       sqlite3_mutex_leave(db->mutex);
717       return SQLITE_BUSY;
718     }
719   }
720 
721   /* Free any outstanding Savepoint structures. */
722   sqlite3CloseSavepoints(db);
723 
724   for(j=0; j<db->nDb; j++){
725     struct Db *pDb = &db->aDb[j];
726     if( pDb->pBt ){
727       sqlite3BtreeClose(pDb->pBt);
728       pDb->pBt = 0;
729       if( j!=1 ){
730         pDb->pSchema = 0;
731       }
732     }
733   }
734   sqlite3ResetInternalSchema(db, -1);
735 
736   /* Tell the code in notify.c that the connection no longer holds any
737   ** locks and does not require any further unlock-notify callbacks.
738   */
739   sqlite3ConnectionClosed(db);
740 
741   assert( db->nDb<=2 );
742   assert( db->aDb==db->aDbStatic );
743   for(j=0; j<ArraySize(db->aFunc.a); j++){
744     FuncDef *pNext, *pHash, *p;
745     for(p=db->aFunc.a[j]; p; p=pHash){
746       pHash = p->pHash;
747       while( p ){
748         functionDestroy(db, p);
749         pNext = p->pNext;
750         sqlite3DbFree(db, p);
751         p = pNext;
752       }
753     }
754   }
755   for(i=sqliteHashFirst(&db->aCollSeq); i; i=sqliteHashNext(i)){
756     CollSeq *pColl = (CollSeq *)sqliteHashData(i);
757     /* Invoke any destructors registered for collation sequence user data. */
758     for(j=0; j<3; j++){
759       if( pColl[j].xDel ){
760         pColl[j].xDel(pColl[j].pUser);
761       }
762     }
763     sqlite3DbFree(db, pColl);
764   }
765   sqlite3HashClear(&db->aCollSeq);
766 #ifndef SQLITE_OMIT_VIRTUALTABLE
767   for(i=sqliteHashFirst(&db->aModule); i; i=sqliteHashNext(i)){
768     Module *pMod = (Module *)sqliteHashData(i);
769     if( pMod->xDestroy ){
770       pMod->xDestroy(pMod->pAux);
771     }
772     sqlite3DbFree(db, pMod);
773   }
774   sqlite3HashClear(&db->aModule);
775 #endif
776 
777   sqlite3Error(db, SQLITE_OK, 0); /* Deallocates any cached error strings. */
778   if( db->pErr ){
779     sqlite3ValueFree(db->pErr);
780   }
781   sqlite3CloseExtensions(db);
782 
783   db->magic = SQLITE_MAGIC_ERROR;
784 
785   /* The temp-database schema is allocated differently from the other schema
786   ** objects (using sqliteMalloc() directly, instead of sqlite3BtreeSchema()).
787   ** So it needs to be freed here. Todo: Why not roll the temp schema into
788   ** the same sqliteMalloc() as the one that allocates the database
789   ** structure?
790   */
791   sqlite3DbFree(db, db->aDb[1].pSchema);
792   sqlite3_mutex_leave(db->mutex);
793   db->magic = SQLITE_MAGIC_CLOSED;
794   sqlite3_mutex_free(db->mutex);
795   assert( db->lookaside.nOut==0 );  /* Fails on a lookaside memory leak */
796   if( db->lookaside.bMalloced ){
797     sqlite3_free(db->lookaside.pStart);
798   }
799   sqlite3_free(db);
800   return SQLITE_OK;
801 }
802 
803 /*
804 ** Rollback all database files.
805 */
sqlite3RollbackAll(sqlite3 * db)806 void sqlite3RollbackAll(sqlite3 *db){
807   int i;
808   int inTrans = 0;
809   assert( sqlite3_mutex_held(db->mutex) );
810   sqlite3BeginBenignMalloc();
811   for(i=0; i<db->nDb; i++){
812     if( db->aDb[i].pBt ){
813       if( sqlite3BtreeIsInTrans(db->aDb[i].pBt) ){
814         inTrans = 1;
815       }
816       sqlite3BtreeRollback(db->aDb[i].pBt);
817       db->aDb[i].inTrans = 0;
818     }
819   }
820   sqlite3VtabRollback(db);
821   sqlite3EndBenignMalloc();
822 
823   if( db->flags&SQLITE_InternChanges ){
824     sqlite3ExpirePreparedStatements(db);
825     sqlite3ResetInternalSchema(db, -1);
826   }
827 
828   /* Any deferred constraint violations have now been resolved. */
829   db->nDeferredCons = 0;
830 
831   /* If one has been configured, invoke the rollback-hook callback */
832   if( db->xRollbackCallback && (inTrans || !db->autoCommit) ){
833     db->xRollbackCallback(db->pRollbackArg);
834   }
835 }
836 
837 /*
838 ** Return a static string that describes the kind of error specified in the
839 ** argument.
840 */
sqlite3ErrStr(int rc)841 const char *sqlite3ErrStr(int rc){
842   static const char* const aMsg[] = {
843     /* SQLITE_OK          */ "not an error",
844     /* SQLITE_ERROR       */ "SQL logic error or missing database",
845     /* SQLITE_INTERNAL    */ 0,
846     /* SQLITE_PERM        */ "access permission denied",
847     /* SQLITE_ABORT       */ "callback requested query abort",
848     /* SQLITE_BUSY        */ "database is locked",
849     /* SQLITE_LOCKED      */ "database table is locked",
850     /* SQLITE_NOMEM       */ "out of memory",
851     /* SQLITE_READONLY    */ "attempt to write a readonly database",
852     /* SQLITE_INTERRUPT   */ "interrupted",
853     /* SQLITE_IOERR       */ "disk I/O error",
854     /* SQLITE_CORRUPT     */ "database disk image is malformed",
855     /* SQLITE_NOTFOUND    */ "unknown operation",
856     /* SQLITE_FULL        */ "database or disk is full",
857     /* SQLITE_CANTOPEN    */ "unable to open database file",
858     /* SQLITE_PROTOCOL    */ "locking protocol",
859     /* SQLITE_EMPTY       */ "table contains no data",
860     /* SQLITE_SCHEMA      */ "database schema has changed",
861     /* SQLITE_TOOBIG      */ "string or blob too big",
862     /* SQLITE_CONSTRAINT  */ "constraint failed",
863     /* SQLITE_MISMATCH    */ "datatype mismatch",
864     /* SQLITE_MISUSE      */ "library routine called out of sequence",
865     /* SQLITE_NOLFS       */ "large file support is disabled",
866     /* SQLITE_AUTH        */ "authorization denied",
867     /* SQLITE_FORMAT      */ "auxiliary database format error",
868     /* SQLITE_RANGE       */ "bind or column index out of range",
869     /* SQLITE_NOTADB      */ "file is encrypted or is not a database",
870   };
871   rc &= 0xff;
872   if( ALWAYS(rc>=0) && rc<(int)(sizeof(aMsg)/sizeof(aMsg[0])) && aMsg[rc]!=0 ){
873     return aMsg[rc];
874   }else{
875     return "unknown error";
876   }
877 }
878 
879 /*
880 ** This routine implements a busy callback that sleeps and tries
881 ** again until a timeout value is reached.  The timeout value is
882 ** an integer number of milliseconds passed in as the first
883 ** argument.
884 */
sqliteDefaultBusyCallback(void * ptr,int count)885 static int sqliteDefaultBusyCallback(
886  void *ptr,               /* Database connection */
887  int count                /* Number of times table has been busy */
888 ){
889 #if SQLITE_OS_WIN || (defined(HAVE_USLEEP) && HAVE_USLEEP)
890   static const u8 delays[] =
891      { 1, 2, 5, 10, 15, 20, 25, 25,  25,  50,  50, 100 };
892   static const u8 totals[] =
893      { 0, 1, 3,  8, 18, 33, 53, 78, 103, 128, 178, 228 };
894 # define NDELAY ArraySize(delays)
895   sqlite3 *db = (sqlite3 *)ptr;
896   int timeout = db->busyTimeout;
897   int delay, prior;
898 
899   assert( count>=0 );
900   if( count < NDELAY ){
901     delay = delays[count];
902     prior = totals[count];
903   }else{
904     delay = delays[NDELAY-1];
905     prior = totals[NDELAY-1] + delay*(count-(NDELAY-1));
906   }
907   if( prior + delay > timeout ){
908     delay = timeout - prior;
909     if( delay<=0 ) return 0;
910   }
911   sqlite3OsSleep(db->pVfs, delay*1000);
912   return 1;
913 #else
914   sqlite3 *db = (sqlite3 *)ptr;
915   int timeout = ((sqlite3 *)ptr)->busyTimeout;
916   if( (count+1)*1000 > timeout ){
917     return 0;
918   }
919   sqlite3OsSleep(db->pVfs, 1000000);
920   return 1;
921 #endif
922 }
923 
924 /*
925 ** Invoke the given busy handler.
926 **
927 ** This routine is called when an operation failed with a lock.
928 ** If this routine returns non-zero, the lock is retried.  If it
929 ** returns 0, the operation aborts with an SQLITE_BUSY error.
930 */
sqlite3InvokeBusyHandler(BusyHandler * p)931 int sqlite3InvokeBusyHandler(BusyHandler *p){
932   int rc;
933   if( NEVER(p==0) || p->xFunc==0 || p->nBusy<0 ) return 0;
934   rc = p->xFunc(p->pArg, p->nBusy);
935   if( rc==0 ){
936     p->nBusy = -1;
937   }else{
938     p->nBusy++;
939   }
940   return rc;
941 }
942 
943 /*
944 ** This routine sets the busy callback for an Sqlite database to the
945 ** given callback function with the given argument.
946 */
sqlite3_busy_handler(sqlite3 * db,int (* xBusy)(void *,int),void * pArg)947 int sqlite3_busy_handler(
948   sqlite3 *db,
949   int (*xBusy)(void*,int),
950   void *pArg
951 ){
952   sqlite3_mutex_enter(db->mutex);
953   db->busyHandler.xFunc = xBusy;
954   db->busyHandler.pArg = pArg;
955   db->busyHandler.nBusy = 0;
956   sqlite3_mutex_leave(db->mutex);
957   return SQLITE_OK;
958 }
959 
960 #ifndef SQLITE_OMIT_PROGRESS_CALLBACK
961 /*
962 ** This routine sets the progress callback for an Sqlite database to the
963 ** given callback function with the given argument. The progress callback will
964 ** be invoked every nOps opcodes.
965 */
sqlite3_progress_handler(sqlite3 * db,int nOps,int (* xProgress)(void *),void * pArg)966 void sqlite3_progress_handler(
967   sqlite3 *db,
968   int nOps,
969   int (*xProgress)(void*),
970   void *pArg
971 ){
972   sqlite3_mutex_enter(db->mutex);
973   if( nOps>0 ){
974     db->xProgress = xProgress;
975     db->nProgressOps = nOps;
976     db->pProgressArg = pArg;
977   }else{
978     db->xProgress = 0;
979     db->nProgressOps = 0;
980     db->pProgressArg = 0;
981   }
982   sqlite3_mutex_leave(db->mutex);
983 }
984 #endif
985 
986 
987 /*
988 ** This routine installs a default busy handler that waits for the
989 ** specified number of milliseconds before returning 0.
990 */
sqlite3_busy_timeout(sqlite3 * db,int ms)991 int sqlite3_busy_timeout(sqlite3 *db, int ms){
992   if( ms>0 ){
993     db->busyTimeout = ms;
994     sqlite3_busy_handler(db, sqliteDefaultBusyCallback, (void*)db);
995   }else{
996     sqlite3_busy_handler(db, 0, 0);
997   }
998   return SQLITE_OK;
999 }
1000 
1001 /*
1002 ** Cause any pending operation to stop at its earliest opportunity.
1003 */
sqlite3_interrupt(sqlite3 * db)1004 void sqlite3_interrupt(sqlite3 *db){
1005   db->u1.isInterrupted = 1;
1006 }
1007 
1008 
1009 /*
1010 ** This function is exactly the same as sqlite3_create_function(), except
1011 ** that it is designed to be called by internal code. The difference is
1012 ** that if a malloc() fails in sqlite3_create_function(), an error code
1013 ** is returned and the mallocFailed flag cleared.
1014 */
sqlite3CreateFunc(sqlite3 * db,const char * zFunctionName,int nArg,int enc,void * pUserData,void (* xFunc)(sqlite3_context *,int,sqlite3_value **),void (* xStep)(sqlite3_context *,int,sqlite3_value **),void (* xFinal)(sqlite3_context *),FuncDestructor * pDestructor)1015 int sqlite3CreateFunc(
1016   sqlite3 *db,
1017   const char *zFunctionName,
1018   int nArg,
1019   int enc,
1020   void *pUserData,
1021   void (*xFunc)(sqlite3_context*,int,sqlite3_value **),
1022   void (*xStep)(sqlite3_context*,int,sqlite3_value **),
1023   void (*xFinal)(sqlite3_context*),
1024   FuncDestructor *pDestructor
1025 ){
1026   FuncDef *p;
1027   int nName;
1028 
1029   assert( sqlite3_mutex_held(db->mutex) );
1030   if( zFunctionName==0 ||
1031       (xFunc && (xFinal || xStep)) ||
1032       (!xFunc && (xFinal && !xStep)) ||
1033       (!xFunc && (!xFinal && xStep)) ||
1034       (nArg<-1 || nArg>SQLITE_MAX_FUNCTION_ARG) ||
1035       (255<(nName = sqlite3Strlen30( zFunctionName))) ){
1036     return SQLITE_MISUSE_BKPT;
1037   }
1038 
1039 #ifndef SQLITE_OMIT_UTF16
1040   /* If SQLITE_UTF16 is specified as the encoding type, transform this
1041   ** to one of SQLITE_UTF16LE or SQLITE_UTF16BE using the
1042   ** SQLITE_UTF16NATIVE macro. SQLITE_UTF16 is not used internally.
1043   **
1044   ** If SQLITE_ANY is specified, add three versions of the function
1045   ** to the hash table.
1046   */
1047   if( enc==SQLITE_UTF16 ){
1048     enc = SQLITE_UTF16NATIVE;
1049   }else if( enc==SQLITE_ANY ){
1050     int rc;
1051     rc = sqlite3CreateFunc(db, zFunctionName, nArg, SQLITE_UTF8,
1052          pUserData, xFunc, xStep, xFinal, pDestructor);
1053     if( rc==SQLITE_OK ){
1054       rc = sqlite3CreateFunc(db, zFunctionName, nArg, SQLITE_UTF16LE,
1055           pUserData, xFunc, xStep, xFinal, pDestructor);
1056     }
1057     if( rc!=SQLITE_OK ){
1058       return rc;
1059     }
1060     enc = SQLITE_UTF16BE;
1061   }
1062 #else
1063   enc = SQLITE_UTF8;
1064 #endif
1065 
1066   /* Check if an existing function is being overridden or deleted. If so,
1067   ** and there are active VMs, then return SQLITE_BUSY. If a function
1068   ** is being overridden/deleted but there are no active VMs, allow the
1069   ** operation to continue but invalidate all precompiled statements.
1070   */
1071   p = sqlite3FindFunction(db, zFunctionName, nName, nArg, (u8)enc, 0);
1072   if( p && p->iPrefEnc==enc && p->nArg==nArg ){
1073     if( db->activeVdbeCnt ){
1074       sqlite3Error(db, SQLITE_BUSY,
1075         "unable to delete/modify user-function due to active statements");
1076       assert( !db->mallocFailed );
1077       return SQLITE_BUSY;
1078     }else{
1079       sqlite3ExpirePreparedStatements(db);
1080     }
1081   }
1082 
1083   p = sqlite3FindFunction(db, zFunctionName, nName, nArg, (u8)enc, 1);
1084   assert(p || db->mallocFailed);
1085   if( !p ){
1086     return SQLITE_NOMEM;
1087   }
1088 
1089   /* If an older version of the function with a configured destructor is
1090   ** being replaced invoke the destructor function here. */
1091   functionDestroy(db, p);
1092 
1093   if( pDestructor ){
1094     pDestructor->nRef++;
1095   }
1096   p->pDestructor = pDestructor;
1097   p->flags = 0;
1098   p->xFunc = xFunc;
1099   p->xStep = xStep;
1100   p->xFinalize = xFinal;
1101   p->pUserData = pUserData;
1102   p->nArg = (u16)nArg;
1103   return SQLITE_OK;
1104 }
1105 
1106 /*
1107 ** Create new user functions.
1108 */
sqlite3_create_function(sqlite3 * db,const char * zFunc,int nArg,int enc,void * p,void (* xFunc)(sqlite3_context *,int,sqlite3_value **),void (* xStep)(sqlite3_context *,int,sqlite3_value **),void (* xFinal)(sqlite3_context *))1109 int sqlite3_create_function(
1110   sqlite3 *db,
1111   const char *zFunc,
1112   int nArg,
1113   int enc,
1114   void *p,
1115   void (*xFunc)(sqlite3_context*,int,sqlite3_value **),
1116   void (*xStep)(sqlite3_context*,int,sqlite3_value **),
1117   void (*xFinal)(sqlite3_context*)
1118 ){
1119   return sqlite3_create_function_v2(db, zFunc, nArg, enc, p, xFunc, xStep,
1120                                     xFinal, 0);
1121 }
1122 
sqlite3_create_function_v2(sqlite3 * db,const char * zFunc,int nArg,int enc,void * p,void (* xFunc)(sqlite3_context *,int,sqlite3_value **),void (* xStep)(sqlite3_context *,int,sqlite3_value **),void (* xFinal)(sqlite3_context *),void (* xDestroy)(void *))1123 int sqlite3_create_function_v2(
1124   sqlite3 *db,
1125   const char *zFunc,
1126   int nArg,
1127   int enc,
1128   void *p,
1129   void (*xFunc)(sqlite3_context*,int,sqlite3_value **),
1130   void (*xStep)(sqlite3_context*,int,sqlite3_value **),
1131   void (*xFinal)(sqlite3_context*),
1132   void (*xDestroy)(void *)
1133 ){
1134   int rc = SQLITE_ERROR;
1135   FuncDestructor *pArg = 0;
1136   sqlite3_mutex_enter(db->mutex);
1137   if( xDestroy ){
1138     pArg = (FuncDestructor *)sqlite3DbMallocZero(db, sizeof(FuncDestructor));
1139     if( !pArg ){
1140       xDestroy(p);
1141       goto out;
1142     }
1143     pArg->xDestroy = xDestroy;
1144     pArg->pUserData = p;
1145   }
1146   rc = sqlite3CreateFunc(db, zFunc, nArg, enc, p, xFunc, xStep, xFinal, pArg);
1147   if( pArg && pArg->nRef==0 ){
1148     assert( rc!=SQLITE_OK );
1149     xDestroy(p);
1150     sqlite3DbFree(db, pArg);
1151   }
1152 
1153  out:
1154   rc = sqlite3ApiExit(db, rc);
1155   sqlite3_mutex_leave(db->mutex);
1156   return rc;
1157 }
1158 
1159 #ifndef SQLITE_OMIT_UTF16
sqlite3_create_function16(sqlite3 * db,const void * zFunctionName,int nArg,int eTextRep,void * p,void (* xFunc)(sqlite3_context *,int,sqlite3_value **),void (* xStep)(sqlite3_context *,int,sqlite3_value **),void (* xFinal)(sqlite3_context *))1160 int sqlite3_create_function16(
1161   sqlite3 *db,
1162   const void *zFunctionName,
1163   int nArg,
1164   int eTextRep,
1165   void *p,
1166   void (*xFunc)(sqlite3_context*,int,sqlite3_value**),
1167   void (*xStep)(sqlite3_context*,int,sqlite3_value**),
1168   void (*xFinal)(sqlite3_context*)
1169 ){
1170   int rc;
1171   char *zFunc8;
1172   sqlite3_mutex_enter(db->mutex);
1173   assert( !db->mallocFailed );
1174   zFunc8 = sqlite3Utf16to8(db, zFunctionName, -1, SQLITE_UTF16NATIVE);
1175   rc = sqlite3CreateFunc(db, zFunc8, nArg, eTextRep, p, xFunc, xStep, xFinal,0);
1176   sqlite3DbFree(db, zFunc8);
1177   rc = sqlite3ApiExit(db, rc);
1178   sqlite3_mutex_leave(db->mutex);
1179   return rc;
1180 }
1181 #endif
1182 
1183 
1184 /*
1185 ** Declare that a function has been overloaded by a virtual table.
1186 **
1187 ** If the function already exists as a regular global function, then
1188 ** this routine is a no-op.  If the function does not exist, then create
1189 ** a new one that always throws a run-time error.
1190 **
1191 ** When virtual tables intend to provide an overloaded function, they
1192 ** should call this routine to make sure the global function exists.
1193 ** A global function must exist in order for name resolution to work
1194 ** properly.
1195 */
sqlite3_overload_function(sqlite3 * db,const char * zName,int nArg)1196 int sqlite3_overload_function(
1197   sqlite3 *db,
1198   const char *zName,
1199   int nArg
1200 ){
1201   int nName = sqlite3Strlen30(zName);
1202   int rc;
1203   sqlite3_mutex_enter(db->mutex);
1204   if( sqlite3FindFunction(db, zName, nName, nArg, SQLITE_UTF8, 0)==0 ){
1205     sqlite3CreateFunc(db, zName, nArg, SQLITE_UTF8,
1206                       0, sqlite3InvalidFunction, 0, 0, 0);
1207   }
1208   rc = sqlite3ApiExit(db, SQLITE_OK);
1209   sqlite3_mutex_leave(db->mutex);
1210   return rc;
1211 }
1212 
1213 #ifndef SQLITE_OMIT_TRACE
1214 /*
1215 ** Register a trace function.  The pArg from the previously registered trace
1216 ** is returned.
1217 **
1218 ** A NULL trace function means that no tracing is executes.  A non-NULL
1219 ** trace is a pointer to a function that is invoked at the start of each
1220 ** SQL statement.
1221 */
sqlite3_trace(sqlite3 * db,void (* xTrace)(void *,const char *),void * pArg)1222 void *sqlite3_trace(sqlite3 *db, void (*xTrace)(void*,const char*), void *pArg){
1223   void *pOld;
1224   sqlite3_mutex_enter(db->mutex);
1225   pOld = db->pTraceArg;
1226   db->xTrace = xTrace;
1227   db->pTraceArg = pArg;
1228   sqlite3_mutex_leave(db->mutex);
1229   return pOld;
1230 }
1231 /*
1232 ** Register a profile function.  The pArg from the previously registered
1233 ** profile function is returned.
1234 **
1235 ** A NULL profile function means that no profiling is executes.  A non-NULL
1236 ** profile is a pointer to a function that is invoked at the conclusion of
1237 ** each SQL statement that is run.
1238 */
sqlite3_profile(sqlite3 * db,void (* xProfile)(void *,const char *,sqlite_uint64),void * pArg)1239 void *sqlite3_profile(
1240   sqlite3 *db,
1241   void (*xProfile)(void*,const char*,sqlite_uint64),
1242   void *pArg
1243 ){
1244   void *pOld;
1245   sqlite3_mutex_enter(db->mutex);
1246   pOld = db->pProfileArg;
1247   db->xProfile = xProfile;
1248   db->pProfileArg = pArg;
1249   sqlite3_mutex_leave(db->mutex);
1250   return pOld;
1251 }
1252 #endif /* SQLITE_OMIT_TRACE */
1253 
1254 /*** EXPERIMENTAL ***
1255 **
1256 ** Register a function to be invoked when a transaction comments.
1257 ** If the invoked function returns non-zero, then the commit becomes a
1258 ** rollback.
1259 */
sqlite3_commit_hook(sqlite3 * db,int (* xCallback)(void *),void * pArg)1260 void *sqlite3_commit_hook(
1261   sqlite3 *db,              /* Attach the hook to this database */
1262   int (*xCallback)(void*),  /* Function to invoke on each commit */
1263   void *pArg                /* Argument to the function */
1264 ){
1265   void *pOld;
1266   sqlite3_mutex_enter(db->mutex);
1267   pOld = db->pCommitArg;
1268   db->xCommitCallback = xCallback;
1269   db->pCommitArg = pArg;
1270   sqlite3_mutex_leave(db->mutex);
1271   return pOld;
1272 }
1273 
1274 /*
1275 ** Register a callback to be invoked each time a row is updated,
1276 ** inserted or deleted using this database connection.
1277 */
sqlite3_update_hook(sqlite3 * db,void (* xCallback)(void *,int,char const *,char const *,sqlite_int64),void * pArg)1278 void *sqlite3_update_hook(
1279   sqlite3 *db,              /* Attach the hook to this database */
1280   void (*xCallback)(void*,int,char const *,char const *,sqlite_int64),
1281   void *pArg                /* Argument to the function */
1282 ){
1283   void *pRet;
1284   sqlite3_mutex_enter(db->mutex);
1285   pRet = db->pUpdateArg;
1286   db->xUpdateCallback = xCallback;
1287   db->pUpdateArg = pArg;
1288   sqlite3_mutex_leave(db->mutex);
1289   return pRet;
1290 }
1291 
1292 /*
1293 ** Register a callback to be invoked each time a transaction is rolled
1294 ** back by this database connection.
1295 */
sqlite3_rollback_hook(sqlite3 * db,void (* xCallback)(void *),void * pArg)1296 void *sqlite3_rollback_hook(
1297   sqlite3 *db,              /* Attach the hook to this database */
1298   void (*xCallback)(void*), /* Callback function */
1299   void *pArg                /* Argument to the function */
1300 ){
1301   void *pRet;
1302   sqlite3_mutex_enter(db->mutex);
1303   pRet = db->pRollbackArg;
1304   db->xRollbackCallback = xCallback;
1305   db->pRollbackArg = pArg;
1306   sqlite3_mutex_leave(db->mutex);
1307   return pRet;
1308 }
1309 
1310 #ifndef SQLITE_OMIT_WAL
1311 /*
1312 ** The sqlite3_wal_hook() callback registered by sqlite3_wal_autocheckpoint().
1313 ** Invoke sqlite3_wal_checkpoint if the number of frames in the log file
1314 ** is greater than sqlite3.pWalArg cast to an integer (the value configured by
1315 ** wal_autocheckpoint()).
1316 */
sqlite3WalDefaultHook(void * pClientData,sqlite3 * db,const char * zDb,int nFrame)1317 int sqlite3WalDefaultHook(
1318   void *pClientData,     /* Argument */
1319   sqlite3 *db,           /* Connection */
1320   const char *zDb,       /* Database */
1321   int nFrame             /* Size of WAL */
1322 ){
1323   if( nFrame>=SQLITE_PTR_TO_INT(pClientData) ){
1324     sqlite3BeginBenignMalloc();
1325     sqlite3_wal_checkpoint(db, zDb);
1326     sqlite3EndBenignMalloc();
1327   }
1328   return SQLITE_OK;
1329 }
1330 #endif /* SQLITE_OMIT_WAL */
1331 
1332 /*
1333 ** Configure an sqlite3_wal_hook() callback to automatically checkpoint
1334 ** a database after committing a transaction if there are nFrame or
1335 ** more frames in the log file. Passing zero or a negative value as the
1336 ** nFrame parameter disables automatic checkpoints entirely.
1337 **
1338 ** The callback registered by this function replaces any existing callback
1339 ** registered using sqlite3_wal_hook(). Likewise, registering a callback
1340 ** using sqlite3_wal_hook() disables the automatic checkpoint mechanism
1341 ** configured by this function.
1342 */
sqlite3_wal_autocheckpoint(sqlite3 * db,int nFrame)1343 int sqlite3_wal_autocheckpoint(sqlite3 *db, int nFrame){
1344 #ifdef SQLITE_OMIT_WAL
1345   UNUSED_PARAMETER(db);
1346   UNUSED_PARAMETER(nFrame);
1347 #else
1348   if( nFrame>0 ){
1349     sqlite3_wal_hook(db, sqlite3WalDefaultHook, SQLITE_INT_TO_PTR(nFrame));
1350   }else{
1351     sqlite3_wal_hook(db, 0, 0);
1352   }
1353 #endif
1354   return SQLITE_OK;
1355 }
1356 
1357 /*
1358 ** Register a callback to be invoked each time a transaction is written
1359 ** into the write-ahead-log by this database connection.
1360 */
sqlite3_wal_hook(sqlite3 * db,int (* xCallback)(void *,sqlite3 *,const char *,int),void * pArg)1361 void *sqlite3_wal_hook(
1362   sqlite3 *db,                    /* Attach the hook to this db handle */
1363   int(*xCallback)(void *, sqlite3*, const char*, int),
1364   void *pArg                      /* First argument passed to xCallback() */
1365 ){
1366 #ifndef SQLITE_OMIT_WAL
1367   void *pRet;
1368   sqlite3_mutex_enter(db->mutex);
1369   pRet = db->pWalArg;
1370   db->xWalCallback = xCallback;
1371   db->pWalArg = pArg;
1372   sqlite3_mutex_leave(db->mutex);
1373   return pRet;
1374 #else
1375   return 0;
1376 #endif
1377 }
1378 
1379 /*
1380 ** Checkpoint database zDb.
1381 */
sqlite3_wal_checkpoint_v2(sqlite3 * db,const char * zDb,int eMode,int * pnLog,int * pnCkpt)1382 int sqlite3_wal_checkpoint_v2(
1383   sqlite3 *db,                    /* Database handle */
1384   const char *zDb,                /* Name of attached database (or NULL) */
1385   int eMode,                      /* SQLITE_CHECKPOINT_* value */
1386   int *pnLog,                     /* OUT: Size of WAL log in frames */
1387   int *pnCkpt                     /* OUT: Total number of frames checkpointed */
1388 ){
1389 #ifdef SQLITE_OMIT_WAL
1390   return SQLITE_OK;
1391 #else
1392   int rc;                         /* Return code */
1393   int iDb = SQLITE_MAX_ATTACHED;  /* sqlite3.aDb[] index of db to checkpoint */
1394 
1395   /* Initialize the output variables to -1 in case an error occurs. */
1396   if( pnLog ) *pnLog = -1;
1397   if( pnCkpt ) *pnCkpt = -1;
1398 
1399   assert( SQLITE_CHECKPOINT_FULL>SQLITE_CHECKPOINT_PASSIVE );
1400   assert( SQLITE_CHECKPOINT_FULL<SQLITE_CHECKPOINT_RESTART );
1401   assert( SQLITE_CHECKPOINT_PASSIVE+2==SQLITE_CHECKPOINT_RESTART );
1402   if( eMode<SQLITE_CHECKPOINT_PASSIVE || eMode>SQLITE_CHECKPOINT_RESTART ){
1403     return SQLITE_MISUSE;
1404   }
1405 
1406   sqlite3_mutex_enter(db->mutex);
1407   if( zDb && zDb[0] ){
1408     iDb = sqlite3FindDbName(db, zDb);
1409   }
1410   if( iDb<0 ){
1411     rc = SQLITE_ERROR;
1412     sqlite3Error(db, SQLITE_ERROR, "unknown database: %s", zDb);
1413   }else{
1414     rc = sqlite3Checkpoint(db, iDb, eMode, pnLog, pnCkpt);
1415     sqlite3Error(db, rc, 0);
1416   }
1417   rc = sqlite3ApiExit(db, rc);
1418   sqlite3_mutex_leave(db->mutex);
1419   return rc;
1420 #endif
1421 }
1422 
1423 
1424 /*
1425 ** Checkpoint database zDb. If zDb is NULL, or if the buffer zDb points
1426 ** to contains a zero-length string, all attached databases are
1427 ** checkpointed.
1428 */
sqlite3_wal_checkpoint(sqlite3 * db,const char * zDb)1429 int sqlite3_wal_checkpoint(sqlite3 *db, const char *zDb){
1430   return sqlite3_wal_checkpoint_v2(db, zDb, SQLITE_CHECKPOINT_PASSIVE, 0, 0);
1431 }
1432 
1433 #ifndef SQLITE_OMIT_WAL
1434 /*
1435 ** Run a checkpoint on database iDb. This is a no-op if database iDb is
1436 ** not currently open in WAL mode.
1437 **
1438 ** If a transaction is open on the database being checkpointed, this
1439 ** function returns SQLITE_LOCKED and a checkpoint is not attempted. If
1440 ** an error occurs while running the checkpoint, an SQLite error code is
1441 ** returned (i.e. SQLITE_IOERR). Otherwise, SQLITE_OK.
1442 **
1443 ** The mutex on database handle db should be held by the caller. The mutex
1444 ** associated with the specific b-tree being checkpointed is taken by
1445 ** this function while the checkpoint is running.
1446 **
1447 ** If iDb is passed SQLITE_MAX_ATTACHED, then all attached databases are
1448 ** checkpointed. If an error is encountered it is returned immediately -
1449 ** no attempt is made to checkpoint any remaining databases.
1450 **
1451 ** Parameter eMode is one of SQLITE_CHECKPOINT_PASSIVE, FULL or RESTART.
1452 */
sqlite3Checkpoint(sqlite3 * db,int iDb,int eMode,int * pnLog,int * pnCkpt)1453 int sqlite3Checkpoint(sqlite3 *db, int iDb, int eMode, int *pnLog, int *pnCkpt){
1454   int rc = SQLITE_OK;             /* Return code */
1455   int i;                          /* Used to iterate through attached dbs */
1456   int bBusy = 0;                  /* True if SQLITE_BUSY has been encountered */
1457 
1458   assert( sqlite3_mutex_held(db->mutex) );
1459   assert( !pnLog || *pnLog==-1 );
1460   assert( !pnCkpt || *pnCkpt==-1 );
1461 
1462   for(i=0; i<db->nDb && rc==SQLITE_OK; i++){
1463     if( i==iDb || iDb==SQLITE_MAX_ATTACHED ){
1464       rc = sqlite3BtreeCheckpoint(db->aDb[i].pBt, eMode, pnLog, pnCkpt);
1465       pnLog = 0;
1466       pnCkpt = 0;
1467       if( rc==SQLITE_BUSY ){
1468         bBusy = 1;
1469         rc = SQLITE_OK;
1470       }
1471     }
1472   }
1473 
1474   return (rc==SQLITE_OK && bBusy) ? SQLITE_BUSY : rc;
1475 }
1476 #endif /* SQLITE_OMIT_WAL */
1477 
1478 /*
1479 ** This function returns true if main-memory should be used instead of
1480 ** a temporary file for transient pager files and statement journals.
1481 ** The value returned depends on the value of db->temp_store (runtime
1482 ** parameter) and the compile time value of SQLITE_TEMP_STORE. The
1483 ** following table describes the relationship between these two values
1484 ** and this functions return value.
1485 **
1486 **   SQLITE_TEMP_STORE     db->temp_store     Location of temporary database
1487 **   -----------------     --------------     ------------------------------
1488 **   0                     any                file      (return 0)
1489 **   1                     1                  file      (return 0)
1490 **   1                     2                  memory    (return 1)
1491 **   1                     0                  file      (return 0)
1492 **   2                     1                  file      (return 0)
1493 **   2                     2                  memory    (return 1)
1494 **   2                     0                  memory    (return 1)
1495 **   3                     any                memory    (return 1)
1496 */
sqlite3TempInMemory(const sqlite3 * db)1497 int sqlite3TempInMemory(const sqlite3 *db){
1498 #if SQLITE_TEMP_STORE==1
1499   return ( db->temp_store==2 );
1500 #endif
1501 #if SQLITE_TEMP_STORE==2
1502   return ( db->temp_store!=1 );
1503 #endif
1504 #if SQLITE_TEMP_STORE==3
1505   return 1;
1506 #endif
1507 #if SQLITE_TEMP_STORE<1 || SQLITE_TEMP_STORE>3
1508   return 0;
1509 #endif
1510 }
1511 
1512 /*
1513 ** Return UTF-8 encoded English language explanation of the most recent
1514 ** error.
1515 */
sqlite3_errmsg(sqlite3 * db)1516 const char *sqlite3_errmsg(sqlite3 *db){
1517   const char *z;
1518   if( !db ){
1519     return sqlite3ErrStr(SQLITE_NOMEM);
1520   }
1521   if( !sqlite3SafetyCheckSickOrOk(db) ){
1522     return sqlite3ErrStr(SQLITE_MISUSE_BKPT);
1523   }
1524   sqlite3_mutex_enter(db->mutex);
1525   if( db->mallocFailed ){
1526     z = sqlite3ErrStr(SQLITE_NOMEM);
1527   }else{
1528     z = (char*)sqlite3_value_text(db->pErr);
1529     assert( !db->mallocFailed );
1530     if( z==0 ){
1531       z = sqlite3ErrStr(db->errCode);
1532     }
1533   }
1534   sqlite3_mutex_leave(db->mutex);
1535   return z;
1536 }
1537 
1538 #ifndef SQLITE_OMIT_UTF16
1539 /*
1540 ** Return UTF-16 encoded English language explanation of the most recent
1541 ** error.
1542 */
sqlite3_errmsg16(sqlite3 * db)1543 const void *sqlite3_errmsg16(sqlite3 *db){
1544   static const u16 outOfMem[] = {
1545     'o', 'u', 't', ' ', 'o', 'f', ' ', 'm', 'e', 'm', 'o', 'r', 'y', 0
1546   };
1547   static const u16 misuse[] = {
1548     'l', 'i', 'b', 'r', 'a', 'r', 'y', ' ',
1549     'r', 'o', 'u', 't', 'i', 'n', 'e', ' ',
1550     'c', 'a', 'l', 'l', 'e', 'd', ' ',
1551     'o', 'u', 't', ' ',
1552     'o', 'f', ' ',
1553     's', 'e', 'q', 'u', 'e', 'n', 'c', 'e', 0
1554   };
1555 
1556   const void *z;
1557   if( !db ){
1558     return (void *)outOfMem;
1559   }
1560   if( !sqlite3SafetyCheckSickOrOk(db) ){
1561     return (void *)misuse;
1562   }
1563   sqlite3_mutex_enter(db->mutex);
1564   if( db->mallocFailed ){
1565     z = (void *)outOfMem;
1566   }else{
1567     z = sqlite3_value_text16(db->pErr);
1568     if( z==0 ){
1569       sqlite3ValueSetStr(db->pErr, -1, sqlite3ErrStr(db->errCode),
1570            SQLITE_UTF8, SQLITE_STATIC);
1571       z = sqlite3_value_text16(db->pErr);
1572     }
1573     /* A malloc() may have failed within the call to sqlite3_value_text16()
1574     ** above. If this is the case, then the db->mallocFailed flag needs to
1575     ** be cleared before returning. Do this directly, instead of via
1576     ** sqlite3ApiExit(), to avoid setting the database handle error message.
1577     */
1578     db->mallocFailed = 0;
1579   }
1580   sqlite3_mutex_leave(db->mutex);
1581   return z;
1582 }
1583 #endif /* SQLITE_OMIT_UTF16 */
1584 
1585 /*
1586 ** Return the most recent error code generated by an SQLite routine. If NULL is
1587 ** passed to this function, we assume a malloc() failed during sqlite3_open().
1588 */
sqlite3_errcode(sqlite3 * db)1589 int sqlite3_errcode(sqlite3 *db){
1590   if( db && !sqlite3SafetyCheckSickOrOk(db) ){
1591     return SQLITE_MISUSE_BKPT;
1592   }
1593   if( !db || db->mallocFailed ){
1594     return SQLITE_NOMEM;
1595   }
1596   return db->errCode & db->errMask;
1597 }
sqlite3_extended_errcode(sqlite3 * db)1598 int sqlite3_extended_errcode(sqlite3 *db){
1599   if( db && !sqlite3SafetyCheckSickOrOk(db) ){
1600     return SQLITE_MISUSE_BKPT;
1601   }
1602   if( !db || db->mallocFailed ){
1603     return SQLITE_NOMEM;
1604   }
1605   return db->errCode;
1606 }
1607 
1608 /*
1609 ** Create a new collating function for database "db".  The name is zName
1610 ** and the encoding is enc.
1611 */
createCollation(sqlite3 * db,const char * zName,u8 enc,u8 collType,void * pCtx,int (* xCompare)(void *,int,const void *,int,const void *),void (* xDel)(void *))1612 static int createCollation(
1613   sqlite3* db,
1614   const char *zName,
1615   u8 enc,
1616   u8 collType,
1617   void* pCtx,
1618   int(*xCompare)(void*,int,const void*,int,const void*),
1619   void(*xDel)(void*)
1620 ){
1621   CollSeq *pColl;
1622   int enc2;
1623   int nName = sqlite3Strlen30(zName);
1624 
1625   assert( sqlite3_mutex_held(db->mutex) );
1626 
1627   /* If SQLITE_UTF16 is specified as the encoding type, transform this
1628   ** to one of SQLITE_UTF16LE or SQLITE_UTF16BE using the
1629   ** SQLITE_UTF16NATIVE macro. SQLITE_UTF16 is not used internally.
1630   */
1631   enc2 = enc;
1632   testcase( enc2==SQLITE_UTF16 );
1633   testcase( enc2==SQLITE_UTF16_ALIGNED );
1634   if( enc2==SQLITE_UTF16 || enc2==SQLITE_UTF16_ALIGNED ){
1635     enc2 = SQLITE_UTF16NATIVE;
1636   }
1637   if( enc2<SQLITE_UTF8 || enc2>SQLITE_UTF16BE ){
1638     return SQLITE_MISUSE_BKPT;
1639   }
1640 
1641   /* Check if this call is removing or replacing an existing collation
1642   ** sequence. If so, and there are active VMs, return busy. If there
1643   ** are no active VMs, invalidate any pre-compiled statements.
1644   */
1645   pColl = sqlite3FindCollSeq(db, (u8)enc2, zName, 0);
1646   if( pColl && pColl->xCmp ){
1647     if( db->activeVdbeCnt ){
1648       sqlite3Error(db, SQLITE_BUSY,
1649         "unable to delete/modify collation sequence due to active statements");
1650       return SQLITE_BUSY;
1651     }
1652     sqlite3ExpirePreparedStatements(db);
1653 
1654     /* If collation sequence pColl was created directly by a call to
1655     ** sqlite3_create_collation, and not generated by synthCollSeq(),
1656     ** then any copies made by synthCollSeq() need to be invalidated.
1657     ** Also, collation destructor - CollSeq.xDel() - function may need
1658     ** to be called.
1659     */
1660     if( (pColl->enc & ~SQLITE_UTF16_ALIGNED)==enc2 ){
1661       CollSeq *aColl = sqlite3HashFind(&db->aCollSeq, zName, nName);
1662       int j;
1663       for(j=0; j<3; j++){
1664         CollSeq *p = &aColl[j];
1665         if( p->enc==pColl->enc ){
1666           if( p->xDel ){
1667             p->xDel(p->pUser);
1668           }
1669           p->xCmp = 0;
1670         }
1671       }
1672     }
1673   }
1674 
1675   pColl = sqlite3FindCollSeq(db, (u8)enc2, zName, 1);
1676   if( pColl==0 ) return SQLITE_NOMEM;
1677   pColl->xCmp = xCompare;
1678   pColl->pUser = pCtx;
1679   pColl->xDel = xDel;
1680   pColl->enc = (u8)(enc2 | (enc & SQLITE_UTF16_ALIGNED));
1681   pColl->type = collType;
1682   sqlite3Error(db, SQLITE_OK, 0);
1683   return SQLITE_OK;
1684 }
1685 
1686 
1687 /*
1688 ** This array defines hard upper bounds on limit values.  The
1689 ** initializer must be kept in sync with the SQLITE_LIMIT_*
1690 ** #defines in sqlite3.h.
1691 */
1692 static const int aHardLimit[] = {
1693   SQLITE_MAX_LENGTH,
1694   SQLITE_MAX_SQL_LENGTH,
1695   SQLITE_MAX_COLUMN,
1696   SQLITE_MAX_EXPR_DEPTH,
1697   SQLITE_MAX_COMPOUND_SELECT,
1698   SQLITE_MAX_VDBE_OP,
1699   SQLITE_MAX_FUNCTION_ARG,
1700   SQLITE_MAX_ATTACHED,
1701   SQLITE_MAX_LIKE_PATTERN_LENGTH,
1702   SQLITE_MAX_VARIABLE_NUMBER,
1703   SQLITE_MAX_TRIGGER_DEPTH,
1704 };
1705 
1706 /*
1707 ** Make sure the hard limits are set to reasonable values
1708 */
1709 #if SQLITE_MAX_LENGTH<100
1710 # error SQLITE_MAX_LENGTH must be at least 100
1711 #endif
1712 #if SQLITE_MAX_SQL_LENGTH<100
1713 # error SQLITE_MAX_SQL_LENGTH must be at least 100
1714 #endif
1715 #if SQLITE_MAX_SQL_LENGTH>SQLITE_MAX_LENGTH
1716 # error SQLITE_MAX_SQL_LENGTH must not be greater than SQLITE_MAX_LENGTH
1717 #endif
1718 #if SQLITE_MAX_COMPOUND_SELECT<2
1719 # error SQLITE_MAX_COMPOUND_SELECT must be at least 2
1720 #endif
1721 #if SQLITE_MAX_VDBE_OP<40
1722 # error SQLITE_MAX_VDBE_OP must be at least 40
1723 #endif
1724 #if SQLITE_MAX_FUNCTION_ARG<0 || SQLITE_MAX_FUNCTION_ARG>1000
1725 # error SQLITE_MAX_FUNCTION_ARG must be between 0 and 1000
1726 #endif
1727 #if SQLITE_MAX_ATTACHED<0 || SQLITE_MAX_ATTACHED>62
1728 # error SQLITE_MAX_ATTACHED must be between 0 and 62
1729 #endif
1730 #if SQLITE_MAX_LIKE_PATTERN_LENGTH<1
1731 # error SQLITE_MAX_LIKE_PATTERN_LENGTH must be at least 1
1732 #endif
1733 #if SQLITE_MAX_COLUMN>32767
1734 # error SQLITE_MAX_COLUMN must not exceed 32767
1735 #endif
1736 #if SQLITE_MAX_TRIGGER_DEPTH<1
1737 # error SQLITE_MAX_TRIGGER_DEPTH must be at least 1
1738 #endif
1739 
1740 
1741 /*
1742 ** Change the value of a limit.  Report the old value.
1743 ** If an invalid limit index is supplied, report -1.
1744 ** Make no changes but still report the old value if the
1745 ** new limit is negative.
1746 **
1747 ** A new lower limit does not shrink existing constructs.
1748 ** It merely prevents new constructs that exceed the limit
1749 ** from forming.
1750 */
sqlite3_limit(sqlite3 * db,int limitId,int newLimit)1751 int sqlite3_limit(sqlite3 *db, int limitId, int newLimit){
1752   int oldLimit;
1753 
1754 
1755   /* EVIDENCE-OF: R-30189-54097 For each limit category SQLITE_LIMIT_NAME
1756   ** there is a hard upper bound set at compile-time by a C preprocessor
1757   ** macro called SQLITE_MAX_NAME. (The "_LIMIT_" in the name is changed to
1758   ** "_MAX_".)
1759   */
1760   assert( aHardLimit[SQLITE_LIMIT_LENGTH]==SQLITE_MAX_LENGTH );
1761   assert( aHardLimit[SQLITE_LIMIT_SQL_LENGTH]==SQLITE_MAX_SQL_LENGTH );
1762   assert( aHardLimit[SQLITE_LIMIT_COLUMN]==SQLITE_MAX_COLUMN );
1763   assert( aHardLimit[SQLITE_LIMIT_EXPR_DEPTH]==SQLITE_MAX_EXPR_DEPTH );
1764   assert( aHardLimit[SQLITE_LIMIT_COMPOUND_SELECT]==SQLITE_MAX_COMPOUND_SELECT);
1765   assert( aHardLimit[SQLITE_LIMIT_VDBE_OP]==SQLITE_MAX_VDBE_OP );
1766   assert( aHardLimit[SQLITE_LIMIT_FUNCTION_ARG]==SQLITE_MAX_FUNCTION_ARG );
1767   assert( aHardLimit[SQLITE_LIMIT_ATTACHED]==SQLITE_MAX_ATTACHED );
1768   assert( aHardLimit[SQLITE_LIMIT_LIKE_PATTERN_LENGTH]==
1769                                                SQLITE_MAX_LIKE_PATTERN_LENGTH );
1770   assert( aHardLimit[SQLITE_LIMIT_VARIABLE_NUMBER]==SQLITE_MAX_VARIABLE_NUMBER);
1771   assert( aHardLimit[SQLITE_LIMIT_TRIGGER_DEPTH]==SQLITE_MAX_TRIGGER_DEPTH );
1772   assert( SQLITE_LIMIT_TRIGGER_DEPTH==(SQLITE_N_LIMIT-1) );
1773 
1774 
1775   if( limitId<0 || limitId>=SQLITE_N_LIMIT ){
1776     return -1;
1777   }
1778   oldLimit = db->aLimit[limitId];
1779   if( newLimit>=0 ){                   /* IMP: R-52476-28732 */
1780     if( newLimit>aHardLimit[limitId] ){
1781       newLimit = aHardLimit[limitId];  /* IMP: R-51463-25634 */
1782     }
1783     db->aLimit[limitId] = newLimit;
1784   }
1785   return oldLimit;                     /* IMP: R-53341-35419 */
1786 }
1787 
1788 /*
1789 ** This routine does the work of opening a database on behalf of
1790 ** sqlite3_open() and sqlite3_open16(). The database filename "zFilename"
1791 ** is UTF-8 encoded.
1792 */
openDatabase(const char * zFilename,sqlite3 ** ppDb,unsigned flags,const char * zVfs)1793 static int openDatabase(
1794   const char *zFilename, /* Database filename UTF-8 encoded */
1795   sqlite3 **ppDb,        /* OUT: Returned database handle */
1796   unsigned flags,        /* Operational flags */
1797   const char *zVfs       /* Name of the VFS to use */
1798 ){
1799   sqlite3 *db;
1800   int rc;
1801   int isThreadsafe;
1802 
1803   *ppDb = 0;
1804 #ifndef SQLITE_OMIT_AUTOINIT
1805   rc = sqlite3_initialize();
1806   if( rc ) return rc;
1807 #endif
1808 
1809   /* Only allow sensible combinations of bits in the flags argument.
1810   ** Throw an error if any non-sense combination is used.  If we
1811   ** do not block illegal combinations here, it could trigger
1812   ** assert() statements in deeper layers.  Sensible combinations
1813   ** are:
1814   **
1815   **  1:  SQLITE_OPEN_READONLY
1816   **  2:  SQLITE_OPEN_READWRITE
1817   **  6:  SQLITE_OPEN_READWRITE | SQLITE_OPEN_CREATE
1818   */
1819   assert( SQLITE_OPEN_READONLY  == 0x01 );
1820   assert( SQLITE_OPEN_READWRITE == 0x02 );
1821   assert( SQLITE_OPEN_CREATE    == 0x04 );
1822   testcase( (1<<(flags&7))==0x02 ); /* READONLY */
1823   testcase( (1<<(flags&7))==0x04 ); /* READWRITE */
1824   testcase( (1<<(flags&7))==0x40 ); /* READWRITE | CREATE */
1825   if( ((1<<(flags&7)) & 0x46)==0 ) return SQLITE_MISUSE;
1826 
1827   if( sqlite3GlobalConfig.bCoreMutex==0 ){
1828     isThreadsafe = 0;
1829   }else if( flags & SQLITE_OPEN_NOMUTEX ){
1830     isThreadsafe = 0;
1831   }else if( flags & SQLITE_OPEN_FULLMUTEX ){
1832     isThreadsafe = 1;
1833   }else{
1834     isThreadsafe = sqlite3GlobalConfig.bFullMutex;
1835   }
1836   if( flags & SQLITE_OPEN_PRIVATECACHE ){
1837     flags &= ~SQLITE_OPEN_SHAREDCACHE;
1838   }else if( sqlite3GlobalConfig.sharedCacheEnabled ){
1839     flags |= SQLITE_OPEN_SHAREDCACHE;
1840   }
1841 
1842   /* Remove harmful bits from the flags parameter
1843   **
1844   ** The SQLITE_OPEN_NOMUTEX and SQLITE_OPEN_FULLMUTEX flags were
1845   ** dealt with in the previous code block.  Besides these, the only
1846   ** valid input flags for sqlite3_open_v2() are SQLITE_OPEN_READONLY,
1847   ** SQLITE_OPEN_READWRITE, SQLITE_OPEN_CREATE, SQLITE_OPEN_SHAREDCACHE,
1848   ** SQLITE_OPEN_PRIVATECACHE, and some reserved bits.  Silently mask
1849   ** off all other flags.
1850   */
1851   flags &=  ~( SQLITE_OPEN_DELETEONCLOSE |
1852                SQLITE_OPEN_EXCLUSIVE |
1853                SQLITE_OPEN_MAIN_DB |
1854                SQLITE_OPEN_TEMP_DB |
1855                SQLITE_OPEN_TRANSIENT_DB |
1856                SQLITE_OPEN_MAIN_JOURNAL |
1857                SQLITE_OPEN_TEMP_JOURNAL |
1858                SQLITE_OPEN_SUBJOURNAL |
1859                SQLITE_OPEN_MASTER_JOURNAL |
1860                SQLITE_OPEN_NOMUTEX |
1861                SQLITE_OPEN_FULLMUTEX |
1862                SQLITE_OPEN_WAL
1863              );
1864 
1865   /* Allocate the sqlite data structure */
1866   db = sqlite3MallocZero( sizeof(sqlite3) );
1867   if( db==0 ) goto opendb_out;
1868   if( isThreadsafe ){
1869     db->mutex = sqlite3MutexAlloc(SQLITE_MUTEX_RECURSIVE);
1870     if( db->mutex==0 ){
1871       sqlite3_free(db);
1872       db = 0;
1873       goto opendb_out;
1874     }
1875   }
1876   sqlite3_mutex_enter(db->mutex);
1877   db->errMask = 0xff;
1878   db->nDb = 2;
1879   db->magic = SQLITE_MAGIC_BUSY;
1880   db->aDb = db->aDbStatic;
1881 
1882   assert( sizeof(db->aLimit)==sizeof(aHardLimit) );
1883   memcpy(db->aLimit, aHardLimit, sizeof(db->aLimit));
1884   db->autoCommit = 1;
1885   db->nextAutovac = -1;
1886   db->nextPagesize = 0;
1887   db->flags |= SQLITE_ShortColNames | SQLITE_AutoIndex | SQLITE_EnableTrigger
1888 #if SQLITE_DEFAULT_FILE_FORMAT<4
1889                  | SQLITE_LegacyFileFmt
1890 #endif
1891 #ifdef SQLITE_ENABLE_LOAD_EXTENSION
1892                  | SQLITE_LoadExtension
1893 #endif
1894 #if SQLITE_DEFAULT_RECURSIVE_TRIGGERS
1895                  | SQLITE_RecTriggers
1896 #endif
1897 #if defined(SQLITE_DEFAULT_FOREIGN_KEYS) && SQLITE_DEFAULT_FOREIGN_KEYS
1898                  | SQLITE_ForeignKeys
1899 #endif
1900       ;
1901   sqlite3HashInit(&db->aCollSeq);
1902 #ifndef SQLITE_OMIT_VIRTUALTABLE
1903   sqlite3HashInit(&db->aModule);
1904 #endif
1905 
1906   db->pVfs = sqlite3_vfs_find(zVfs);
1907   if( !db->pVfs ){
1908     rc = SQLITE_ERROR;
1909     sqlite3Error(db, rc, "no such vfs: %s", zVfs);
1910     goto opendb_out;
1911   }
1912 
1913   /* Add the default collation sequence BINARY. BINARY works for both UTF-8
1914   ** and UTF-16, so add a version for each to avoid any unnecessary
1915   ** conversions. The only error that can occur here is a malloc() failure.
1916   */
1917   createCollation(db, "BINARY", SQLITE_UTF8, SQLITE_COLL_BINARY, 0,
1918                   binCollFunc, 0);
1919   createCollation(db, "BINARY", SQLITE_UTF16BE, SQLITE_COLL_BINARY, 0,
1920                   binCollFunc, 0);
1921   createCollation(db, "BINARY", SQLITE_UTF16LE, SQLITE_COLL_BINARY, 0,
1922                   binCollFunc, 0);
1923   createCollation(db, "RTRIM", SQLITE_UTF8, SQLITE_COLL_USER, (void*)1,
1924                   binCollFunc, 0);
1925   if( db->mallocFailed ){
1926     goto opendb_out;
1927   }
1928   db->pDfltColl = sqlite3FindCollSeq(db, SQLITE_UTF8, "BINARY", 0);
1929   assert( db->pDfltColl!=0 );
1930 
1931   /* Also add a UTF-8 case-insensitive collation sequence. */
1932   createCollation(db, "NOCASE", SQLITE_UTF8, SQLITE_COLL_NOCASE, 0,
1933                   nocaseCollatingFunc, 0);
1934 
1935   /* Open the backend database driver */
1936   db->openFlags = flags;
1937   rc = sqlite3BtreeOpen(zFilename, db, &db->aDb[0].pBt, 0,
1938                         flags | SQLITE_OPEN_MAIN_DB);
1939   if( rc!=SQLITE_OK ){
1940     if( rc==SQLITE_IOERR_NOMEM ){
1941       rc = SQLITE_NOMEM;
1942     }
1943     sqlite3Error(db, rc, 0);
1944     goto opendb_out;
1945   }
1946   db->aDb[0].pSchema = sqlite3SchemaGet(db, db->aDb[0].pBt);
1947   db->aDb[1].pSchema = sqlite3SchemaGet(db, 0);
1948 
1949 
1950   /* The default safety_level for the main database is 'full'; for the temp
1951   ** database it is 'NONE'. This matches the pager layer defaults.
1952   */
1953   db->aDb[0].zName = "main";
1954   db->aDb[0].safety_level = 3;
1955   db->aDb[1].zName = "temp";
1956   db->aDb[1].safety_level = 1;
1957 
1958   db->magic = SQLITE_MAGIC_OPEN;
1959   if( db->mallocFailed ){
1960     goto opendb_out;
1961   }
1962 
1963   /* Register all built-in functions, but do not attempt to read the
1964   ** database schema yet. This is delayed until the first time the database
1965   ** is accessed.
1966   */
1967   sqlite3Error(db, SQLITE_OK, 0);
1968   sqlite3RegisterBuiltinFunctions(db);
1969 
1970   /* Load automatic extensions - extensions that have been registered
1971   ** using the sqlite3_automatic_extension() API.
1972   */
1973   sqlite3AutoLoadExtensions(db);
1974   rc = sqlite3_errcode(db);
1975   if( rc!=SQLITE_OK ){
1976     goto opendb_out;
1977   }
1978 
1979 #ifdef SQLITE_ENABLE_FTS1
1980   if( !db->mallocFailed ){
1981     extern int sqlite3Fts1Init(sqlite3*);
1982     rc = sqlite3Fts1Init(db);
1983   }
1984 #endif
1985 
1986 #ifdef SQLITE_ENABLE_FTS2
1987   if( !db->mallocFailed && rc==SQLITE_OK ){
1988     extern int sqlite3Fts2Init(sqlite3*);
1989     rc = sqlite3Fts2Init(db);
1990   }
1991 #endif
1992 
1993 #ifdef SQLITE_ENABLE_FTS3
1994   if( !db->mallocFailed && rc==SQLITE_OK ){
1995     rc = sqlite3Fts3Init(db);
1996   }
1997 #endif
1998 
1999 #ifdef SQLITE_ENABLE_ICU
2000   if( !db->mallocFailed && rc==SQLITE_OK ){
2001     rc = sqlite3IcuInit(db);
2002   }
2003 #endif
2004 
2005 #ifdef SQLITE_ENABLE_RTREE
2006   if( !db->mallocFailed && rc==SQLITE_OK){
2007     rc = sqlite3RtreeInit(db);
2008   }
2009 #endif
2010 
2011   sqlite3Error(db, rc, 0);
2012 
2013   /* -DSQLITE_DEFAULT_LOCKING_MODE=1 makes EXCLUSIVE the default locking
2014   ** mode.  -DSQLITE_DEFAULT_LOCKING_MODE=0 make NORMAL the default locking
2015   ** mode.  Doing nothing at all also makes NORMAL the default.
2016   */
2017 #ifdef SQLITE_DEFAULT_LOCKING_MODE
2018   db->dfltLockMode = SQLITE_DEFAULT_LOCKING_MODE;
2019   sqlite3PagerLockingMode(sqlite3BtreePager(db->aDb[0].pBt),
2020                           SQLITE_DEFAULT_LOCKING_MODE);
2021 #endif
2022 
2023   /* Enable the lookaside-malloc subsystem */
2024   setupLookaside(db, 0, sqlite3GlobalConfig.szLookaside,
2025                         sqlite3GlobalConfig.nLookaside);
2026 
2027   sqlite3_wal_autocheckpoint(db, SQLITE_DEFAULT_WAL_AUTOCHECKPOINT);
2028 
2029 opendb_out:
2030   if( db ){
2031     assert( db->mutex!=0 || isThreadsafe==0 || sqlite3GlobalConfig.bFullMutex==0 );
2032     sqlite3_mutex_leave(db->mutex);
2033   }
2034   rc = sqlite3_errcode(db);
2035   if( rc==SQLITE_NOMEM ){
2036     sqlite3_close(db);
2037     db = 0;
2038   }else if( rc!=SQLITE_OK ){
2039     db->magic = SQLITE_MAGIC_SICK;
2040   }
2041   *ppDb = db;
2042   return sqlite3ApiExit(0, rc);
2043 }
2044 
2045 /*
2046 ** Open a new database handle.
2047 */
sqlite3_open(const char * zFilename,sqlite3 ** ppDb)2048 int sqlite3_open(
2049   const char *zFilename,
2050   sqlite3 **ppDb
2051 ){
2052   return openDatabase(zFilename, ppDb,
2053                       SQLITE_OPEN_READWRITE | SQLITE_OPEN_CREATE, 0);
2054 }
sqlite3_open_v2(const char * filename,sqlite3 ** ppDb,int flags,const char * zVfs)2055 int sqlite3_open_v2(
2056   const char *filename,   /* Database filename (UTF-8) */
2057   sqlite3 **ppDb,         /* OUT: SQLite db handle */
2058   int flags,              /* Flags */
2059   const char *zVfs        /* Name of VFS module to use */
2060 ){
2061   return openDatabase(filename, ppDb, flags, zVfs);
2062 }
2063 
2064 #ifndef SQLITE_OMIT_UTF16
2065 /*
2066 ** Open a new database handle.
2067 */
sqlite3_open16(const void * zFilename,sqlite3 ** ppDb)2068 int sqlite3_open16(
2069   const void *zFilename,
2070   sqlite3 **ppDb
2071 ){
2072   char const *zFilename8;   /* zFilename encoded in UTF-8 instead of UTF-16 */
2073   sqlite3_value *pVal;
2074   int rc;
2075 
2076   assert( zFilename );
2077   assert( ppDb );
2078   *ppDb = 0;
2079 #ifndef SQLITE_OMIT_AUTOINIT
2080   rc = sqlite3_initialize();
2081   if( rc ) return rc;
2082 #endif
2083   pVal = sqlite3ValueNew(0);
2084   sqlite3ValueSetStr(pVal, -1, zFilename, SQLITE_UTF16NATIVE, SQLITE_STATIC);
2085   zFilename8 = sqlite3ValueText(pVal, SQLITE_UTF8);
2086   if( zFilename8 ){
2087     rc = openDatabase(zFilename8, ppDb,
2088                       SQLITE_OPEN_READWRITE | SQLITE_OPEN_CREATE, 0);
2089     assert( *ppDb || rc==SQLITE_NOMEM );
2090     if( rc==SQLITE_OK && !DbHasProperty(*ppDb, 0, DB_SchemaLoaded) ){
2091       ENC(*ppDb) = SQLITE_UTF16NATIVE;
2092     }
2093   }else{
2094     rc = SQLITE_NOMEM;
2095   }
2096   sqlite3ValueFree(pVal);
2097 
2098   return sqlite3ApiExit(0, rc);
2099 }
2100 #endif /* SQLITE_OMIT_UTF16 */
2101 
2102 /*
2103 ** Register a new collation sequence with the database handle db.
2104 */
sqlite3_create_collation(sqlite3 * db,const char * zName,int enc,void * pCtx,int (* xCompare)(void *,int,const void *,int,const void *))2105 int sqlite3_create_collation(
2106   sqlite3* db,
2107   const char *zName,
2108   int enc,
2109   void* pCtx,
2110   int(*xCompare)(void*,int,const void*,int,const void*)
2111 ){
2112   int rc;
2113   sqlite3_mutex_enter(db->mutex);
2114   assert( !db->mallocFailed );
2115   rc = createCollation(db, zName, (u8)enc, SQLITE_COLL_USER, pCtx, xCompare, 0);
2116   rc = sqlite3ApiExit(db, rc);
2117   sqlite3_mutex_leave(db->mutex);
2118   return rc;
2119 }
2120 
2121 /*
2122 ** Register a new collation sequence with the database handle db.
2123 */
sqlite3_create_collation_v2(sqlite3 * db,const char * zName,int enc,void * pCtx,int (* xCompare)(void *,int,const void *,int,const void *),void (* xDel)(void *))2124 int sqlite3_create_collation_v2(
2125   sqlite3* db,
2126   const char *zName,
2127   int enc,
2128   void* pCtx,
2129   int(*xCompare)(void*,int,const void*,int,const void*),
2130   void(*xDel)(void*)
2131 ){
2132   int rc;
2133   sqlite3_mutex_enter(db->mutex);
2134   assert( !db->mallocFailed );
2135   rc = createCollation(db, zName, (u8)enc, SQLITE_COLL_USER, pCtx, xCompare, xDel);
2136   rc = sqlite3ApiExit(db, rc);
2137   sqlite3_mutex_leave(db->mutex);
2138   return rc;
2139 }
2140 
2141 #ifndef SQLITE_OMIT_UTF16
2142 /*
2143 ** Register a new collation sequence with the database handle db.
2144 */
sqlite3_create_collation16(sqlite3 * db,const void * zName,int enc,void * pCtx,int (* xCompare)(void *,int,const void *,int,const void *))2145 int sqlite3_create_collation16(
2146   sqlite3* db,
2147   const void *zName,
2148   int enc,
2149   void* pCtx,
2150   int(*xCompare)(void*,int,const void*,int,const void*)
2151 ){
2152   int rc = SQLITE_OK;
2153   char *zName8;
2154   sqlite3_mutex_enter(db->mutex);
2155   assert( !db->mallocFailed );
2156   zName8 = sqlite3Utf16to8(db, zName, -1, SQLITE_UTF16NATIVE);
2157   if( zName8 ){
2158     rc = createCollation(db, zName8, (u8)enc, SQLITE_COLL_USER, pCtx, xCompare, 0);
2159     sqlite3DbFree(db, zName8);
2160   }
2161   rc = sqlite3ApiExit(db, rc);
2162   sqlite3_mutex_leave(db->mutex);
2163   return rc;
2164 }
2165 #endif /* SQLITE_OMIT_UTF16 */
2166 
2167 /*
2168 ** Register a collation sequence factory callback with the database handle
2169 ** db. Replace any previously installed collation sequence factory.
2170 */
sqlite3_collation_needed(sqlite3 * db,void * pCollNeededArg,void (* xCollNeeded)(void *,sqlite3 *,int eTextRep,const char *))2171 int sqlite3_collation_needed(
2172   sqlite3 *db,
2173   void *pCollNeededArg,
2174   void(*xCollNeeded)(void*,sqlite3*,int eTextRep,const char*)
2175 ){
2176   sqlite3_mutex_enter(db->mutex);
2177   db->xCollNeeded = xCollNeeded;
2178   db->xCollNeeded16 = 0;
2179   db->pCollNeededArg = pCollNeededArg;
2180   sqlite3_mutex_leave(db->mutex);
2181   return SQLITE_OK;
2182 }
2183 
2184 #ifndef SQLITE_OMIT_UTF16
2185 /*
2186 ** Register a collation sequence factory callback with the database handle
2187 ** db. Replace any previously installed collation sequence factory.
2188 */
sqlite3_collation_needed16(sqlite3 * db,void * pCollNeededArg,void (* xCollNeeded16)(void *,sqlite3 *,int eTextRep,const void *))2189 int sqlite3_collation_needed16(
2190   sqlite3 *db,
2191   void *pCollNeededArg,
2192   void(*xCollNeeded16)(void*,sqlite3*,int eTextRep,const void*)
2193 ){
2194   sqlite3_mutex_enter(db->mutex);
2195   db->xCollNeeded = 0;
2196   db->xCollNeeded16 = xCollNeeded16;
2197   db->pCollNeededArg = pCollNeededArg;
2198   sqlite3_mutex_leave(db->mutex);
2199   return SQLITE_OK;
2200 }
2201 #endif /* SQLITE_OMIT_UTF16 */
2202 
2203 #ifndef SQLITE_OMIT_DEPRECATED
2204 /*
2205 ** This function is now an anachronism. It used to be used to recover from a
2206 ** malloc() failure, but SQLite now does this automatically.
2207 */
sqlite3_global_recover(void)2208 int sqlite3_global_recover(void){
2209   return SQLITE_OK;
2210 }
2211 #endif
2212 
2213 /*
2214 ** Test to see whether or not the database connection is in autocommit
2215 ** mode.  Return TRUE if it is and FALSE if not.  Autocommit mode is on
2216 ** by default.  Autocommit is disabled by a BEGIN statement and reenabled
2217 ** by the next COMMIT or ROLLBACK.
2218 **
2219 ******* THIS IS AN EXPERIMENTAL API AND IS SUBJECT TO CHANGE ******
2220 */
sqlite3_get_autocommit(sqlite3 * db)2221 int sqlite3_get_autocommit(sqlite3 *db){
2222   return db->autoCommit;
2223 }
2224 
2225 /*
2226 ** The following routines are subtitutes for constants SQLITE_CORRUPT,
2227 ** SQLITE_MISUSE, SQLITE_CANTOPEN, SQLITE_IOERR and possibly other error
2228 ** constants.  They server two purposes:
2229 **
2230 **   1.  Serve as a convenient place to set a breakpoint in a debugger
2231 **       to detect when version error conditions occurs.
2232 **
2233 **   2.  Invoke sqlite3_log() to provide the source code location where
2234 **       a low-level error is first detected.
2235 */
sqlite3CorruptError(int lineno)2236 int sqlite3CorruptError(int lineno){
2237   testcase( sqlite3GlobalConfig.xLog!=0 );
2238   sqlite3_log(SQLITE_CORRUPT,
2239               "database corruption at line %d of [%.10s]",
2240               lineno, 20+sqlite3_sourceid());
2241   return SQLITE_CORRUPT;
2242 }
sqlite3MisuseError(int lineno)2243 int sqlite3MisuseError(int lineno){
2244   testcase( sqlite3GlobalConfig.xLog!=0 );
2245   sqlite3_log(SQLITE_MISUSE,
2246               "misuse at line %d of [%.10s]",
2247               lineno, 20+sqlite3_sourceid());
2248   return SQLITE_MISUSE;
2249 }
sqlite3CantopenError(int lineno)2250 int sqlite3CantopenError(int lineno){
2251   testcase( sqlite3GlobalConfig.xLog!=0 );
2252   sqlite3_log(SQLITE_CANTOPEN,
2253               "cannot open file at line %d of [%.10s]",
2254               lineno, 20+sqlite3_sourceid());
2255   return SQLITE_CANTOPEN;
2256 }
2257 
2258 
2259 #ifndef SQLITE_OMIT_DEPRECATED
2260 /*
2261 ** This is a convenience routine that makes sure that all thread-specific
2262 ** data for this thread has been deallocated.
2263 **
2264 ** SQLite no longer uses thread-specific data so this routine is now a
2265 ** no-op.  It is retained for historical compatibility.
2266 */
sqlite3_thread_cleanup(void)2267 void sqlite3_thread_cleanup(void){
2268 }
2269 #endif
2270 
2271 /*
2272 ** Return meta information about a specific column of a database table.
2273 ** See comment in sqlite3.h (sqlite.h.in) for details.
2274 */
2275 #ifdef SQLITE_ENABLE_COLUMN_METADATA
sqlite3_table_column_metadata(sqlite3 * db,const char * zDbName,const char * zTableName,const char * zColumnName,char const ** pzDataType,char const ** pzCollSeq,int * pNotNull,int * pPrimaryKey,int * pAutoinc)2276 int sqlite3_table_column_metadata(
2277   sqlite3 *db,                /* Connection handle */
2278   const char *zDbName,        /* Database name or NULL */
2279   const char *zTableName,     /* Table name */
2280   const char *zColumnName,    /* Column name */
2281   char const **pzDataType,    /* OUTPUT: Declared data type */
2282   char const **pzCollSeq,     /* OUTPUT: Collation sequence name */
2283   int *pNotNull,              /* OUTPUT: True if NOT NULL constraint exists */
2284   int *pPrimaryKey,           /* OUTPUT: True if column part of PK */
2285   int *pAutoinc               /* OUTPUT: True if column is auto-increment */
2286 ){
2287   int rc;
2288   char *zErrMsg = 0;
2289   Table *pTab = 0;
2290   Column *pCol = 0;
2291   int iCol;
2292 
2293   char const *zDataType = 0;
2294   char const *zCollSeq = 0;
2295   int notnull = 0;
2296   int primarykey = 0;
2297   int autoinc = 0;
2298 
2299   /* Ensure the database schema has been loaded */
2300   sqlite3_mutex_enter(db->mutex);
2301   sqlite3BtreeEnterAll(db);
2302   rc = sqlite3Init(db, &zErrMsg);
2303   if( SQLITE_OK!=rc ){
2304     goto error_out;
2305   }
2306 
2307   /* Locate the table in question */
2308   pTab = sqlite3FindTable(db, zTableName, zDbName);
2309   if( !pTab || pTab->pSelect ){
2310     pTab = 0;
2311     goto error_out;
2312   }
2313 
2314   /* Find the column for which info is requested */
2315   if( sqlite3IsRowid(zColumnName) ){
2316     iCol = pTab->iPKey;
2317     if( iCol>=0 ){
2318       pCol = &pTab->aCol[iCol];
2319     }
2320   }else{
2321     for(iCol=0; iCol<pTab->nCol; iCol++){
2322       pCol = &pTab->aCol[iCol];
2323       if( 0==sqlite3StrICmp(pCol->zName, zColumnName) ){
2324         break;
2325       }
2326     }
2327     if( iCol==pTab->nCol ){
2328       pTab = 0;
2329       goto error_out;
2330     }
2331   }
2332 
2333   /* The following block stores the meta information that will be returned
2334   ** to the caller in local variables zDataType, zCollSeq, notnull, primarykey
2335   ** and autoinc. At this point there are two possibilities:
2336   **
2337   **     1. The specified column name was rowid", "oid" or "_rowid_"
2338   **        and there is no explicitly declared IPK column.
2339   **
2340   **     2. The table is not a view and the column name identified an
2341   **        explicitly declared column. Copy meta information from *pCol.
2342   */
2343   if( pCol ){
2344     zDataType = pCol->zType;
2345     zCollSeq = pCol->zColl;
2346     notnull = pCol->notNull!=0;
2347     primarykey  = pCol->isPrimKey!=0;
2348     autoinc = pTab->iPKey==iCol && (pTab->tabFlags & TF_Autoincrement)!=0;
2349   }else{
2350     zDataType = "INTEGER";
2351     primarykey = 1;
2352   }
2353   if( !zCollSeq ){
2354     zCollSeq = "BINARY";
2355   }
2356 
2357 error_out:
2358   sqlite3BtreeLeaveAll(db);
2359 
2360   /* Whether the function call succeeded or failed, set the output parameters
2361   ** to whatever their local counterparts contain. If an error did occur,
2362   ** this has the effect of zeroing all output parameters.
2363   */
2364   if( pzDataType ) *pzDataType = zDataType;
2365   if( pzCollSeq ) *pzCollSeq = zCollSeq;
2366   if( pNotNull ) *pNotNull = notnull;
2367   if( pPrimaryKey ) *pPrimaryKey = primarykey;
2368   if( pAutoinc ) *pAutoinc = autoinc;
2369 
2370   if( SQLITE_OK==rc && !pTab ){
2371     sqlite3DbFree(db, zErrMsg);
2372     zErrMsg = sqlite3MPrintf(db, "no such table column: %s.%s", zTableName,
2373         zColumnName);
2374     rc = SQLITE_ERROR;
2375   }
2376   sqlite3Error(db, rc, (zErrMsg?"%s":0), zErrMsg);
2377   sqlite3DbFree(db, zErrMsg);
2378   rc = sqlite3ApiExit(db, rc);
2379   sqlite3_mutex_leave(db->mutex);
2380   return rc;
2381 }
2382 #endif
2383 
2384 /*
2385 ** Sleep for a little while.  Return the amount of time slept.
2386 */
sqlite3_sleep(int ms)2387 int sqlite3_sleep(int ms){
2388   sqlite3_vfs *pVfs;
2389   int rc;
2390   pVfs = sqlite3_vfs_find(0);
2391   if( pVfs==0 ) return 0;
2392 
2393   /* This function works in milliseconds, but the underlying OsSleep()
2394   ** API uses microseconds. Hence the 1000's.
2395   */
2396   rc = (sqlite3OsSleep(pVfs, 1000*ms)/1000);
2397   return rc;
2398 }
2399 
2400 /*
2401 ** Enable or disable the extended result codes.
2402 */
sqlite3_extended_result_codes(sqlite3 * db,int onoff)2403 int sqlite3_extended_result_codes(sqlite3 *db, int onoff){
2404   sqlite3_mutex_enter(db->mutex);
2405   db->errMask = onoff ? 0xffffffff : 0xff;
2406   sqlite3_mutex_leave(db->mutex);
2407   return SQLITE_OK;
2408 }
2409 
2410 /*
2411 ** Invoke the xFileControl method on a particular database.
2412 */
sqlite3_file_control(sqlite3 * db,const char * zDbName,int op,void * pArg)2413 int sqlite3_file_control(sqlite3 *db, const char *zDbName, int op, void *pArg){
2414   int rc = SQLITE_ERROR;
2415   int iDb;
2416   sqlite3_mutex_enter(db->mutex);
2417   if( zDbName==0 ){
2418     iDb = 0;
2419   }else{
2420     for(iDb=0; iDb<db->nDb; iDb++){
2421       if( strcmp(db->aDb[iDb].zName, zDbName)==0 ) break;
2422     }
2423   }
2424   if( iDb<db->nDb ){
2425     Btree *pBtree = db->aDb[iDb].pBt;
2426     if( pBtree ){
2427       Pager *pPager;
2428       sqlite3_file *fd;
2429       sqlite3BtreeEnter(pBtree);
2430       pPager = sqlite3BtreePager(pBtree);
2431       assert( pPager!=0 );
2432       fd = sqlite3PagerFile(pPager);
2433       assert( fd!=0 );
2434       if( op==SQLITE_FCNTL_FILE_POINTER ){
2435         *(sqlite3_file**)pArg = fd;
2436         rc = SQLITE_OK;
2437       }else if( fd->pMethods ){
2438         rc = sqlite3OsFileControl(fd, op, pArg);
2439       }else{
2440         rc = SQLITE_NOTFOUND;
2441       }
2442       sqlite3BtreeLeave(pBtree);
2443     }
2444   }
2445   sqlite3_mutex_leave(db->mutex);
2446   return rc;
2447 }
2448 
2449 /*
2450 ** Interface to the testing logic.
2451 */
sqlite3_test_control(int op,...)2452 int sqlite3_test_control(int op, ...){
2453   int rc = 0;
2454 #ifndef SQLITE_OMIT_BUILTIN_TEST
2455   va_list ap;
2456   va_start(ap, op);
2457   switch( op ){
2458 
2459     /*
2460     ** Save the current state of the PRNG.
2461     */
2462     case SQLITE_TESTCTRL_PRNG_SAVE: {
2463       sqlite3PrngSaveState();
2464       break;
2465     }
2466 
2467     /*
2468     ** Restore the state of the PRNG to the last state saved using
2469     ** PRNG_SAVE.  If PRNG_SAVE has never before been called, then
2470     ** this verb acts like PRNG_RESET.
2471     */
2472     case SQLITE_TESTCTRL_PRNG_RESTORE: {
2473       sqlite3PrngRestoreState();
2474       break;
2475     }
2476 
2477     /*
2478     ** Reset the PRNG back to its uninitialized state.  The next call
2479     ** to sqlite3_randomness() will reseed the PRNG using a single call
2480     ** to the xRandomness method of the default VFS.
2481     */
2482     case SQLITE_TESTCTRL_PRNG_RESET: {
2483       sqlite3PrngResetState();
2484       break;
2485     }
2486 
2487     /*
2488     **  sqlite3_test_control(BITVEC_TEST, size, program)
2489     **
2490     ** Run a test against a Bitvec object of size.  The program argument
2491     ** is an array of integers that defines the test.  Return -1 on a
2492     ** memory allocation error, 0 on success, or non-zero for an error.
2493     ** See the sqlite3BitvecBuiltinTest() for additional information.
2494     */
2495     case SQLITE_TESTCTRL_BITVEC_TEST: {
2496       int sz = va_arg(ap, int);
2497       int *aProg = va_arg(ap, int*);
2498       rc = sqlite3BitvecBuiltinTest(sz, aProg);
2499       break;
2500     }
2501 
2502     /*
2503     **  sqlite3_test_control(BENIGN_MALLOC_HOOKS, xBegin, xEnd)
2504     **
2505     ** Register hooks to call to indicate which malloc() failures
2506     ** are benign.
2507     */
2508     case SQLITE_TESTCTRL_BENIGN_MALLOC_HOOKS: {
2509       typedef void (*void_function)(void);
2510       void_function xBenignBegin;
2511       void_function xBenignEnd;
2512       xBenignBegin = va_arg(ap, void_function);
2513       xBenignEnd = va_arg(ap, void_function);
2514       sqlite3BenignMallocHooks(xBenignBegin, xBenignEnd);
2515       break;
2516     }
2517 
2518     /*
2519     **  sqlite3_test_control(SQLITE_TESTCTRL_PENDING_BYTE, unsigned int X)
2520     **
2521     ** Set the PENDING byte to the value in the argument, if X>0.
2522     ** Make no changes if X==0.  Return the value of the pending byte
2523     ** as it existing before this routine was called.
2524     **
2525     ** IMPORTANT:  Changing the PENDING byte from 0x40000000 results in
2526     ** an incompatible database file format.  Changing the PENDING byte
2527     ** while any database connection is open results in undefined and
2528     ** dileterious behavior.
2529     */
2530     case SQLITE_TESTCTRL_PENDING_BYTE: {
2531       rc = PENDING_BYTE;
2532 #ifndef SQLITE_OMIT_WSD
2533       {
2534         unsigned int newVal = va_arg(ap, unsigned int);
2535         if( newVal ) sqlite3PendingByte = newVal;
2536       }
2537 #endif
2538       break;
2539     }
2540 
2541     /*
2542     **  sqlite3_test_control(SQLITE_TESTCTRL_ASSERT, int X)
2543     **
2544     ** This action provides a run-time test to see whether or not
2545     ** assert() was enabled at compile-time.  If X is true and assert()
2546     ** is enabled, then the return value is true.  If X is true and
2547     ** assert() is disabled, then the return value is zero.  If X is
2548     ** false and assert() is enabled, then the assertion fires and the
2549     ** process aborts.  If X is false and assert() is disabled, then the
2550     ** return value is zero.
2551     */
2552     case SQLITE_TESTCTRL_ASSERT: {
2553       volatile int x = 0;
2554       assert( (x = va_arg(ap,int))!=0 );
2555       rc = x;
2556       break;
2557     }
2558 
2559 
2560     /*
2561     **  sqlite3_test_control(SQLITE_TESTCTRL_ALWAYS, int X)
2562     **
2563     ** This action provides a run-time test to see how the ALWAYS and
2564     ** NEVER macros were defined at compile-time.
2565     **
2566     ** The return value is ALWAYS(X).
2567     **
2568     ** The recommended test is X==2.  If the return value is 2, that means
2569     ** ALWAYS() and NEVER() are both no-op pass-through macros, which is the
2570     ** default setting.  If the return value is 1, then ALWAYS() is either
2571     ** hard-coded to true or else it asserts if its argument is false.
2572     ** The first behavior (hard-coded to true) is the case if
2573     ** SQLITE_TESTCTRL_ASSERT shows that assert() is disabled and the second
2574     ** behavior (assert if the argument to ALWAYS() is false) is the case if
2575     ** SQLITE_TESTCTRL_ASSERT shows that assert() is enabled.
2576     **
2577     ** The run-time test procedure might look something like this:
2578     **
2579     **    if( sqlite3_test_control(SQLITE_TESTCTRL_ALWAYS, 2)==2 ){
2580     **      // ALWAYS() and NEVER() are no-op pass-through macros
2581     **    }else if( sqlite3_test_control(SQLITE_TESTCTRL_ASSERT, 1) ){
2582     **      // ALWAYS(x) asserts that x is true. NEVER(x) asserts x is false.
2583     **    }else{
2584     **      // ALWAYS(x) is a constant 1.  NEVER(x) is a constant 0.
2585     **    }
2586     */
2587     case SQLITE_TESTCTRL_ALWAYS: {
2588       int x = va_arg(ap,int);
2589       rc = ALWAYS(x);
2590       break;
2591     }
2592 
2593     /*   sqlite3_test_control(SQLITE_TESTCTRL_RESERVE, sqlite3 *db, int N)
2594     **
2595     ** Set the nReserve size to N for the main database on the database
2596     ** connection db.
2597     */
2598     case SQLITE_TESTCTRL_RESERVE: {
2599       sqlite3 *db = va_arg(ap, sqlite3*);
2600       int x = va_arg(ap,int);
2601       sqlite3_mutex_enter(db->mutex);
2602       sqlite3BtreeSetPageSize(db->aDb[0].pBt, 0, x, 0);
2603       sqlite3_mutex_leave(db->mutex);
2604       break;
2605     }
2606 
2607     /*  sqlite3_test_control(SQLITE_TESTCTRL_OPTIMIZATIONS, sqlite3 *db, int N)
2608     **
2609     ** Enable or disable various optimizations for testing purposes.  The
2610     ** argument N is a bitmask of optimizations to be disabled.  For normal
2611     ** operation N should be 0.  The idea is that a test program (like the
2612     ** SQL Logic Test or SLT test module) can run the same SQL multiple times
2613     ** with various optimizations disabled to verify that the same answer
2614     ** is obtained in every case.
2615     */
2616     case SQLITE_TESTCTRL_OPTIMIZATIONS: {
2617       sqlite3 *db = va_arg(ap, sqlite3*);
2618       int x = va_arg(ap,int);
2619       db->flags = (x & SQLITE_OptMask) | (db->flags & ~SQLITE_OptMask);
2620       break;
2621     }
2622 
2623 #ifdef SQLITE_N_KEYWORD
2624     /* sqlite3_test_control(SQLITE_TESTCTRL_ISKEYWORD, const char *zWord)
2625     **
2626     ** If zWord is a keyword recognized by the parser, then return the
2627     ** number of keywords.  Or if zWord is not a keyword, return 0.
2628     **
2629     ** This test feature is only available in the amalgamation since
2630     ** the SQLITE_N_KEYWORD macro is not defined in this file if SQLite
2631     ** is built using separate source files.
2632     */
2633     case SQLITE_TESTCTRL_ISKEYWORD: {
2634       const char *zWord = va_arg(ap, const char*);
2635       int n = sqlite3Strlen30(zWord);
2636       rc = (sqlite3KeywordCode((u8*)zWord, n)!=TK_ID) ? SQLITE_N_KEYWORD : 0;
2637       break;
2638     }
2639 #endif
2640 
2641     /* sqlite3_test_control(SQLITE_TESTCTRL_PGHDRSZ)
2642     **
2643     ** Return the size of a pcache header in bytes.
2644     */
2645     case SQLITE_TESTCTRL_PGHDRSZ: {
2646       rc = sizeof(PgHdr);
2647       break;
2648     }
2649 
2650     /* sqlite3_test_control(SQLITE_TESTCTRL_SCRATCHMALLOC, sz, &pNew, pFree);
2651     **
2652     ** Pass pFree into sqlite3ScratchFree().
2653     ** If sz>0 then allocate a scratch buffer into pNew.
2654     */
2655     case SQLITE_TESTCTRL_SCRATCHMALLOC: {
2656       void *pFree, **ppNew;
2657       int sz;
2658       sz = va_arg(ap, int);
2659       ppNew = va_arg(ap, void**);
2660       pFree = va_arg(ap, void*);
2661       if( sz ) *ppNew = sqlite3ScratchMalloc(sz);
2662       sqlite3ScratchFree(pFree);
2663       break;
2664     }
2665 
2666   }
2667   va_end(ap);
2668 #endif /* SQLITE_OMIT_BUILTIN_TEST */
2669   return rc;
2670 }
2671