blob: 1c3b120215791bf12cecc27037112a776c75a37a [file] [log] [blame]
drh75897232000-05-29 14:26:00 +00001/*
drhb19a2bc2001-09-16 00:13:26 +00002** 2001 September 15
drh75897232000-05-29 14:26:00 +00003**
drhb19a2bc2001-09-16 00:13:26 +00004** The author disclaims copyright to this source code. In place of
5** a legal notice, here is a blessing:
drh75897232000-05-29 14:26:00 +00006**
drhb19a2bc2001-09-16 00:13:26 +00007** May you do good and not evil.
8** May you find forgiveness for yourself and forgive others.
9** May you share freely, never taking more than you give.
drh75897232000-05-29 14:26:00 +000010**
11*************************************************************************
12** Main file for the SQLite library. The routines in this file
13** implement the programmer interface to the library. Routines in
14** other files are for internal use by SQLite and should not be
15** accessed by users of the library.
16**
drh18706c02003-06-23 15:15:03 +000017** $Id: main.c,v 1.137 2003/06/23 15:15:03 drh Exp $
drh75897232000-05-29 14:26:00 +000018*/
19#include "sqliteInt.h"
drh8cfbf082001-09-19 13:22:39 +000020#include "os.h"
drhce9079c2002-05-15 14:17:44 +000021#include <ctype.h>
drh75897232000-05-29 14:26:00 +000022
23/*
drhc2311722002-07-19 17:46:38 +000024** A pointer to this structure is used to communicate information
25** from sqliteInit into the sqliteInitCallback.
26*/
27typedef struct {
28 sqlite *db; /* The database being initialized */
29 char **pzErrMsg; /* Error message stored here */
30} InitData;
31
drh8bf8dc92003-05-17 17:35:10 +000032/*
33** Fill the InitData structure with an error message that indicates
34** that the database is corrupt.
35*/
36static void corruptSchema(InitData *pData){
37 sqliteSetString(pData->pzErrMsg, "malformed database schema", 0);
38}
drhc2311722002-07-19 17:46:38 +000039
40/*
drh75897232000-05-29 14:26:00 +000041** This is the callback routine for the code that initializes the
drh382c0242001-10-06 16:33:02 +000042** database. See sqliteInit() below for additional information.
43**
44** Each callback contains the following information:
drh28037572000-08-02 13:47:41 +000045**
drh4a324312001-12-21 14:30:42 +000046** argv[0] = "file-format" or "schema-cookie" or "table" or "index"
drhe3c41372001-09-17 20:25:58 +000047** argv[1] = table or index name or meta statement type.
48** argv[2] = root page number for table or index. NULL for meta.
drhe78e8282003-01-19 03:59:45 +000049** argv[3] = SQL text for a CREATE TABLE or CREATE INDEX statement.
drh1c2d8412003-03-31 00:30:47 +000050** argv[4] = "1" for temporary files, "0" for main database, "2" or more
51** for auxiliary database files.
drhd78eeee2001-09-13 16:18:53 +000052**
drh75897232000-05-29 14:26:00 +000053*/
drhc2311722002-07-19 17:46:38 +000054static
55int sqliteInitCallback(void *pInit, int argc, char **argv, char **azColName){
56 InitData *pData = (InitData*)pInit;
drh75897232000-05-29 14:26:00 +000057 Parse sParse;
drhd78eeee2001-09-13 16:18:53 +000058 int nErr = 0;
drh75897232000-05-29 14:26:00 +000059
drhe0bc4042002-06-25 01:09:11 +000060 assert( argc==5 );
drh8bf8dc92003-05-17 17:35:10 +000061 if( argv[0]==0 ){
62 corruptSchema(pData);
63 return 1;
64 }
drhd78eeee2001-09-13 16:18:53 +000065 switch( argv[0][0] ){
drh17f71932002-02-21 12:01:27 +000066 case 'v':
drhd78eeee2001-09-13 16:18:53 +000067 case 'i':
drh17f71932002-02-21 12:01:27 +000068 case 't': { /* CREATE TABLE, CREATE INDEX, or CREATE VIEW statements */
drh8bf8dc92003-05-17 17:35:10 +000069 if( argv[2]==0 || argv[4]==0 ){
70 corruptSchema(pData);
71 return 1;
72 }
drhadbca9c2001-09-27 15:11:53 +000073 if( argv[3] && argv[3][0] ){
drh17f71932002-02-21 12:01:27 +000074 /* Call the parser to process a CREATE TABLE, INDEX or VIEW.
drh382c0242001-10-06 16:33:02 +000075 ** But because sParse.initFlag is set to 1, no VDBE code is generated
76 ** or executed. All the parser does is build the internal data
drh17f71932002-02-21 12:01:27 +000077 ** structures that describe the table, index, or view.
drh382c0242001-10-06 16:33:02 +000078 */
drhadbca9c2001-09-27 15:11:53 +000079 memset(&sParse, 0, sizeof(sParse));
drhc2311722002-07-19 17:46:38 +000080 sParse.db = pData->db;
drhadbca9c2001-09-27 15:11:53 +000081 sParse.initFlag = 1;
drhd24cc422003-03-27 12:51:24 +000082 sParse.iDb = atoi(argv[4]);
drhadbca9c2001-09-27 15:11:53 +000083 sParse.newTnum = atoi(argv[2]);
drhb86ccfb2003-01-28 23:13:10 +000084 sParse.useCallback = 1;
drhc2311722002-07-19 17:46:38 +000085 sqliteRunParser(&sParse, argv[3], pData->pzErrMsg);
drhadbca9c2001-09-27 15:11:53 +000086 }else{
drh382c0242001-10-06 16:33:02 +000087 /* If the SQL column is blank it means this is an index that
88 ** was created to be the PRIMARY KEY or to fulfill a UNIQUE
drhaacc5432002-01-06 17:07:40 +000089 ** constraint for a CREATE TABLE. The index should have already
drh382c0242001-10-06 16:33:02 +000090 ** been created when we processed the CREATE TABLE. All we have
drhaacc5432002-01-06 17:07:40 +000091 ** to do here is record the root page number for that index.
drh382c0242001-10-06 16:33:02 +000092 */
drhd24cc422003-03-27 12:51:24 +000093 int iDb;
94 Index *pIndex;
95
96 iDb = atoi(argv[4]);
97 assert( iDb>=0 && iDb<pData->db->nDb );
98 pIndex = sqliteFindIndex(pData->db, argv[1], pData->db->aDb[iDb].zName);
drhadbca9c2001-09-27 15:11:53 +000099 if( pIndex==0 || pIndex->tnum!=0 ){
drhda9e0342002-01-10 14:31:48 +0000100 /* This can occur if there exists an index on a TEMP table which
101 ** has the same name as another index on a permanent index. Since
102 ** the permanent table is hidden by the TEMP table, we can also
103 ** safely ignore the index on the permanent table.
104 */
105 /* Do Nothing */;
drhadbca9c2001-09-27 15:11:53 +0000106 }else{
107 pIndex->tnum = atoi(argv[2]);
108 }
109 }
drhd78eeee2001-09-13 16:18:53 +0000110 break;
111 }
112 default: {
113 /* This can not happen! */
114 nErr = 1;
115 assert( nErr==0 );
116 }
drh28037572000-08-02 13:47:41 +0000117 }
drh75897232000-05-29 14:26:00 +0000118 return nErr;
119}
120
121/*
drh491791a2002-07-18 00:34:09 +0000122** This is a callback procedure used to reconstruct a table. The
123** name of the table to be reconstructed is passed in as argv[0].
124**
125** This routine is used to automatically upgrade a database from
126** format version 1 or 2 to version 3. The correct operation of
127** this routine relys on the fact that no indices are used when
128** copying a table out to a temporary file.
129*/
drhc2311722002-07-19 17:46:38 +0000130static
131int upgrade_3_callback(void *pInit, int argc, char **argv, char **NotUsed){
132 InitData *pData = (InitData*)pInit;
drh491791a2002-07-18 00:34:09 +0000133 int rc;
drh8e5ba842002-07-18 01:27:17 +0000134 Table *pTab;
135 Trigger *pTrig;
drh26b3e1b2002-07-19 18:52:40 +0000136 char *zErr = 0;
drh491791a2002-07-18 00:34:09 +0000137
drhd24cc422003-03-27 12:51:24 +0000138 pTab = sqliteFindTable(pData->db, argv[0], 0);
drhf573c992002-07-31 00:32:50 +0000139 assert( pTab!=0 );
140 assert( sqliteStrICmp(pTab->zName, argv[0])==0 );
drh8e5ba842002-07-18 01:27:17 +0000141 if( pTab ){
142 pTrig = pTab->pTrigger;
143 pTab->pTrigger = 0; /* Disable all triggers before rebuilding the table */
144 }
drhc2311722002-07-19 17:46:38 +0000145 rc = sqlite_exec_printf(pData->db,
drh491791a2002-07-18 00:34:09 +0000146 "CREATE TEMP TABLE sqlite_x AS SELECT * FROM '%q'; "
147 "DELETE FROM '%q'; "
148 "INSERT INTO '%q' SELECT * FROM sqlite_x; "
149 "DROP TABLE sqlite_x;",
drhc2311722002-07-19 17:46:38 +0000150 0, 0, &zErr, argv[0], argv[0], argv[0]);
151 if( zErr ){
152 sqliteSetString(pData->pzErrMsg, zErr, 0);
153 sqlite_freemem(zErr);
154 }
drhf573c992002-07-31 00:32:50 +0000155
156 /* If an error occurred in the SQL above, then the transaction will
157 ** rollback which will delete the internal symbol tables. This will
158 ** cause the structure that pTab points to be deleted. In case that
159 ** happened, we need to refetch pTab.
160 */
drhd24cc422003-03-27 12:51:24 +0000161 pTab = sqliteFindTable(pData->db, argv[0], 0);
drhf573c992002-07-31 00:32:50 +0000162 if( pTab ){
163 assert( sqliteStrICmp(pTab->zName, argv[0])==0 );
164 pTab->pTrigger = pTrig; /* Re-enable triggers */
165 }
drh491791a2002-07-18 00:34:09 +0000166 return rc!=SQLITE_OK;
167}
168
169
170
171/*
drh58b95762000-06-02 01:17:37 +0000172** Attempt to read the database schema and initialize internal
drh1c2d8412003-03-31 00:30:47 +0000173** data structures for a single database file. The index of the
174** database file is given by iDb. iDb==0 is used for the main
175** database. iDb==1 should never be used. iDb>=2 is used for
176** auxiliary databases. Return one of the SQLITE_ error codes to
drh58b95762000-06-02 01:17:37 +0000177** indicate success or failure.
drh75897232000-05-29 14:26:00 +0000178*/
drh1c2d8412003-03-31 00:30:47 +0000179static int sqliteInitOne(sqlite *db, int iDb, char **pzErrMsg){
drh58b95762000-06-02 01:17:37 +0000180 int rc;
drhe0bc4042002-06-25 01:09:11 +0000181 BtCursor *curMain;
182 int size;
183 Table *pTab;
184 char *azArg[6];
drh1c2d8412003-03-31 00:30:47 +0000185 char zDbNum[30];
drhe0bc4042002-06-25 01:09:11 +0000186 int meta[SQLITE_N_BTREE_META];
187 Parse sParse;
drhc2311722002-07-19 17:46:38 +0000188 InitData initData;
drh58b95762000-06-02 01:17:37 +0000189
190 /*
191 ** The master database table has a structure like this
192 */
drh75897232000-05-29 14:26:00 +0000193 static char master_schema[] =
drhe0bc4042002-06-25 01:09:11 +0000194 "CREATE TABLE sqlite_master(\n"
195 " type text,\n"
196 " name text,\n"
197 " tbl_name text,\n"
198 " rootpage integer,\n"
199 " sql text\n"
200 ")"
201 ;
202 static char temp_master_schema[] =
203 "CREATE TEMP TABLE sqlite_temp_master(\n"
drh75897232000-05-29 14:26:00 +0000204 " type text,\n"
205 " name text,\n"
206 " tbl_name text,\n"
drhadbca9c2001-09-27 15:11:53 +0000207 " rootpage integer,\n"
drh75897232000-05-29 14:26:00 +0000208 " sql text\n"
209 ")"
210 ;
211
drhe0bc4042002-06-25 01:09:11 +0000212 /* The following SQL will read the schema from the master tables.
213 ** The first version works with SQLite file formats 2 or greater.
214 ** The second version is for format 1 files.
drh75897232000-05-29 14:26:00 +0000215 **
drhe0bc4042002-06-25 01:09:11 +0000216 ** Beginning with file format 2, the rowid for new table entries
217 ** (including entries in sqlite_master) is an increasing integer.
218 ** So for file format 2 and later, we can play back sqlite_master
219 ** and all the CREATE statements will appear in the right order.
220 ** But with file format 1, table entries were random and so we
221 ** have to make sure the CREATE TABLEs occur before their corresponding
222 ** CREATE INDEXs. (We don't have to deal with CREATE VIEW or
223 ** CREATE TRIGGER in file format 1 because those constructs did
224 ** not exist then.)
drh75897232000-05-29 14:26:00 +0000225 */
drhe0bc4042002-06-25 01:09:11 +0000226 static char init_script[] =
227 "SELECT type, name, rootpage, sql, 1 FROM sqlite_temp_master "
228 "UNION ALL "
229 "SELECT type, name, rootpage, sql, 0 FROM sqlite_master";
230 static char older_init_script[] =
231 "SELECT type, name, rootpage, sql, 1 FROM sqlite_temp_master "
232 "UNION ALL "
233 "SELECT type, name, rootpage, sql, 0 FROM sqlite_master "
234 "WHERE type='table' "
235 "UNION ALL "
236 "SELECT type, name, rootpage, sql, 0 FROM sqlite_master "
237 "WHERE type='index'";
drh17f71932002-02-21 12:01:27 +0000238
drh603240c2002-03-05 01:11:12 +0000239
drh1c2d8412003-03-31 00:30:47 +0000240 assert( iDb>=0 && iDb!=1 && iDb<db->nDb );
241
drhe0bc4042002-06-25 01:09:11 +0000242 /* Construct the schema tables: sqlite_master and sqlite_temp_master
drh58b95762000-06-02 01:17:37 +0000243 */
drhe0bc4042002-06-25 01:09:11 +0000244 azArg[0] = "table";
245 azArg[1] = MASTER_NAME;
246 azArg[2] = "2";
247 azArg[3] = master_schema;
drh1c2d8412003-03-31 00:30:47 +0000248 sprintf(zDbNum, "%d", iDb);
249 azArg[4] = zDbNum;
drhe0bc4042002-06-25 01:09:11 +0000250 azArg[5] = 0;
drhc2311722002-07-19 17:46:38 +0000251 initData.db = db;
252 initData.pzErrMsg = pzErrMsg;
253 sqliteInitCallback(&initData, 5, azArg, 0);
drhd24cc422003-03-27 12:51:24 +0000254 pTab = sqliteFindTable(db, MASTER_NAME, "main");
drhe0bc4042002-06-25 01:09:11 +0000255 if( pTab ){
256 pTab->readOnly = 1;
drhd8bc7082000-06-07 23:51:50 +0000257 }
drh1c2d8412003-03-31 00:30:47 +0000258 if( iDb==0 ){
259 azArg[1] = TEMP_MASTER_NAME;
260 azArg[3] = temp_master_schema;
261 azArg[4] = "1";
262 sqliteInitCallback(&initData, 5, azArg, 0);
263 pTab = sqliteFindTable(db, TEMP_MASTER_NAME, "temp");
264 if( pTab ){
265 pTab->readOnly = 1;
266 }
drhe0bc4042002-06-25 01:09:11 +0000267 }
268
269 /* Create a cursor to hold the database open
270 */
drh1c2d8412003-03-31 00:30:47 +0000271 if( db->aDb[iDb].pBt==0 ) return SQLITE_OK;
272 rc = sqliteBtreeCursor(db->aDb[iDb].pBt, 2, 0, &curMain);
drh92ed08a2002-07-30 18:43:40 +0000273 if( rc ){
drh1e0ccab2002-08-29 23:59:47 +0000274 sqliteSetString(pzErrMsg, sqlite_error_string(rc), 0);
drh92ed08a2002-07-30 18:43:40 +0000275 return rc;
276 }
drhe0bc4042002-06-25 01:09:11 +0000277
278 /* Get the database meta information
279 */
drh1c2d8412003-03-31 00:30:47 +0000280 rc = sqliteBtreeGetMeta(db->aDb[iDb].pBt, meta);
drhe0bc4042002-06-25 01:09:11 +0000281 if( rc ){
drh1e0ccab2002-08-29 23:59:47 +0000282 sqliteSetString(pzErrMsg, sqlite_error_string(rc), 0);
drhe0bc4042002-06-25 01:09:11 +0000283 sqliteBtreeCloseCursor(curMain);
284 return rc;
285 }
drh1c2d8412003-03-31 00:30:47 +0000286 db->aDb[iDb].schema_cookie = meta[1];
287 if( iDb==0 ){
288 db->next_cookie = meta[1];
289 db->file_format = meta[2];
290 size = meta[3];
291 if( size==0 ){ size = MAX_PAGES; }
292 db->cache_size = size;
293 db->safety_level = meta[4];
294 if( db->safety_level==0 ) db->safety_level = 2;
drhe0bc4042002-06-25 01:09:11 +0000295
drh1c2d8412003-03-31 00:30:47 +0000296 /*
297 ** file_format==1 Version 2.1.0.
298 ** file_format==2 Version 2.2.0. Add support for INTEGER PRIMARY KEY.
299 ** file_format==3 Version 2.6.0. Fix empty-string index bug.
300 ** file_format==4 Version 2.7.0. Add support for separate numeric and
301 ** text datatypes.
302 */
303 if( db->file_format==0 ){
304 /* This happens if the database was initially empty */
305 db->file_format = 4;
306 }else if( db->file_format>4 ){
307 sqliteBtreeCloseCursor(curMain);
308 sqliteSetString(pzErrMsg, "unsupported file format", 0);
309 return SQLITE_ERROR;
310 }
drh2d458342003-04-05 03:42:26 +0000311 }else if( db->file_format!=meta[2] || db->file_format<4 ){
312 assert( db->file_format>=4 );
313 if( meta[2]==0 ){
314 sqliteSetString(pzErrMsg, "cannot attach empty database: ",
315 db->aDb[iDb].zName, 0);
316 }else{
317 sqliteSetString(pzErrMsg, "incompatible file format in auxiliary "
318 "database: ", db->aDb[iDb].zName, 0);
319 }
drh1c2d8412003-03-31 00:30:47 +0000320 sqliteBtreeClose(db->aDb[iDb].pBt);
321 db->aDb[iDb].pBt = 0;
322 return SQLITE_FORMAT;
drh28037572000-08-02 13:47:41 +0000323 }
drhade106c2003-06-02 18:17:33 +0000324 sqliteBtreeSetCacheSize(db->aDb[iDb].pBt, db->cache_size);
drh1c2d8412003-03-31 00:30:47 +0000325 sqliteBtreeSetSafetyLevel(db->aDb[iDb].pBt, meta[4]==0 ? 2 : meta[4]);
drhaacc5432002-01-06 17:07:40 +0000326
drhe0bc4042002-06-25 01:09:11 +0000327 /* Read the schema information out of the schema tables
drhaacc5432002-01-06 17:07:40 +0000328 */
drhe0bc4042002-06-25 01:09:11 +0000329 memset(&sParse, 0, sizeof(sParse));
330 sParse.db = db;
drhe0bc4042002-06-25 01:09:11 +0000331 sParse.xCallback = sqliteInitCallback;
drhc2311722002-07-19 17:46:38 +0000332 sParse.pArg = (void*)&initData;
drhe0bc4042002-06-25 01:09:11 +0000333 sParse.initFlag = 1;
drhb86ccfb2003-01-28 23:13:10 +0000334 sParse.useCallback = 1;
drh1c2d8412003-03-31 00:30:47 +0000335 if( iDb==0 ){
336 sqliteRunParser(&sParse,
337 db->file_format>=2 ? init_script : older_init_script,
338 pzErrMsg);
339 }else{
340 char *zSql = 0;
341 sqliteSetString(&zSql,
342 "SELECT type, name, rootpage, sql, ", zDbNum, " FROM \"",
343 db->aDb[iDb].zName, "\".sqlite_master", 0);
344 sqliteRunParser(&sParse, zSql, pzErrMsg);
345 sqliteFree(zSql);
346 }
347 sqliteBtreeCloseCursor(curMain);
drhe0bc4042002-06-25 01:09:11 +0000348 if( sqlite_malloc_failed ){
349 sqliteSetString(pzErrMsg, "out of memory", 0);
350 sParse.rc = SQLITE_NOMEM;
drh1c2d8412003-03-31 00:30:47 +0000351 sqliteResetInternalSchema(db, 0);
drhe0bc4042002-06-25 01:09:11 +0000352 }
353 if( sParse.rc==SQLITE_OK ){
drh8bf8dc92003-05-17 17:35:10 +0000354 DbSetProperty(db, iDb, DB_SchemaLoaded);
355 if( iDb==0 ){
356 DbSetProperty(db, 1, DB_SchemaLoaded);
357 }
drh1c2d8412003-03-31 00:30:47 +0000358 }else{
359 sqliteResetInternalSchema(db, iDb);
360 }
361 return sParse.rc;
362}
363
364/*
365** Initialize all database files - the main database file, the file
366** used to store temporary tables, and any additional database files
367** created using ATTACH statements. Return a success code. If an
368** error occurs, write an error message into *pzErrMsg.
369**
370** After the database is initialized, the SQLITE_Initialized
371** bit is set in the flags field of the sqlite structure. An
372** attempt is made to initialize the database as soon as it
373** is opened. If that fails (perhaps because another process
374** has the sqlite_master table locked) than another attempt
375** is made the first time the database is accessed.
376*/
377int sqliteInit(sqlite *db, char **pzErrMsg){
378 int i, rc;
379
380 assert( (db->flags & SQLITE_Initialized)==0 );
381 rc = SQLITE_OK;
382 for(i=0; rc==SQLITE_OK && i<db->nDb; i++){
drh8bf8dc92003-05-17 17:35:10 +0000383 if( DbHasProperty(db, i, DB_SchemaLoaded) ) continue;
384 assert( i!=1 ); /* Should have been initialized together with 0 */
drh1c2d8412003-03-31 00:30:47 +0000385 rc = sqliteInitOne(db, i, pzErrMsg);
386 }
387 if( rc==SQLITE_OK ){
drh58b95762000-06-02 01:17:37 +0000388 db->flags |= SQLITE_Initialized;
drh5e00f6c2001-09-13 13:46:56 +0000389 sqliteCommitInternalChanges(db);
drhe0bc4042002-06-25 01:09:11 +0000390 }else{
391 db->flags &= ~SQLITE_Initialized;
drh58b95762000-06-02 01:17:37 +0000392 }
drh1c2d8412003-03-31 00:30:47 +0000393 return rc;
drh58b95762000-06-02 01:17:37 +0000394}
395
396/*
drhb217a572000-08-22 13:40:18 +0000397** The version of the library
398*/
drh096c4972002-07-19 19:03:41 +0000399const char rcsid[] = "@(#) \044Id: SQLite version " SQLITE_VERSION " $";
drh3d0b5592000-08-22 13:40:51 +0000400const char sqlite_version[] = SQLITE_VERSION;
drhb217a572000-08-22 13:40:18 +0000401
402/*
drh297ecf12001-04-05 15:57:13 +0000403** Does the library expect data to be encoded as UTF-8 or iso8859? The
404** following global constant always lets us know.
405*/
406#ifdef SQLITE_UTF8
drhfbc3eab2001-04-06 16:13:42 +0000407const char sqlite_encoding[] = "UTF-8";
drh297ecf12001-04-05 15:57:13 +0000408#else
drhfbc3eab2001-04-06 16:13:42 +0000409const char sqlite_encoding[] = "iso8859";
drh297ecf12001-04-05 15:57:13 +0000410#endif
411
412/*
drh58b95762000-06-02 01:17:37 +0000413** Open a new SQLite database. Construct an "sqlite" structure to define
414** the state of this database and return a pointer to that structure.
415**
416** An attempt is made to initialize the in-memory data structures that
417** hold the database schema. But if this fails (because the schema file
418** is locked) then that step is deferred until the first call to
419** sqlite_exec().
420*/
421sqlite *sqlite_open(const char *zFilename, int mode, char **pzErrMsg){
422 sqlite *db;
drhd24cc422003-03-27 12:51:24 +0000423 int rc, i;
drh58b95762000-06-02 01:17:37 +0000424
425 /* Allocate the sqlite data structure */
drh75897232000-05-29 14:26:00 +0000426 db = sqliteMalloc( sizeof(sqlite) );
427 if( pzErrMsg ) *pzErrMsg = 0;
drhdaffd0e2001-04-11 14:28:42 +0000428 if( db==0 ) goto no_mem_on_open;
drh1c928532002-01-31 15:54:21 +0000429 db->onError = OE_Default;
drh5cf8e8c2002-02-19 22:42:05 +0000430 db->priorNewRowid = 0;
drh247be432002-05-10 05:44:55 +0000431 db->magic = SQLITE_MAGIC_BUSY;
drh001bbcb2003-03-19 03:14:00 +0000432 db->nDb = 2;
433 db->aDb = db->aDbStatic;
drhd24cc422003-03-27 12:51:24 +0000434 sqliteHashInit(&db->aFunc, SQLITE_HASH_STRING, 1);
435 for(i=0; i<db->nDb; i++){
436 sqliteHashInit(&db->aDb[i].tblHash, SQLITE_HASH_STRING, 0);
437 sqliteHashInit(&db->aDb[i].idxHash, SQLITE_HASH_STRING, 0);
438 sqliteHashInit(&db->aDb[i].trigHash, SQLITE_HASH_STRING, 0);
439 sqliteHashInit(&db->aDb[i].aFKey, SQLITE_HASH_STRING, 1);
440 }
drh75897232000-05-29 14:26:00 +0000441
442 /* Open the backend database driver */
drh5e8e1982003-06-04 15:48:33 +0000443 if( zFilename[0]==':' && strcmp(zFilename,":memory:")==0 ){
444 db->temp_store = 2;
445 }
drh144f9ea2003-04-16 01:28:16 +0000446 rc = sqliteBtreeFactory(db, zFilename, 0, MAX_PAGES, &db->aDb[0].pBt);
drh5e00f6c2001-09-13 13:46:56 +0000447 if( rc!=SQLITE_OK ){
448 switch( rc ){
449 default: {
drhaacc5432002-01-06 17:07:40 +0000450 sqliteSetString(pzErrMsg, "unable to open database: ", zFilename, 0);
drh5e00f6c2001-09-13 13:46:56 +0000451 }
452 }
drh75897232000-05-29 14:26:00 +0000453 sqliteFree(db);
drh5edc3122001-09-13 21:53:09 +0000454 sqliteStrRealloc(pzErrMsg);
drhbe0072d2001-09-13 14:46:09 +0000455 return 0;
drh75897232000-05-29 14:26:00 +0000456 }
drh001bbcb2003-03-19 03:14:00 +0000457 db->aDb[0].zName = "main";
drh113088e2003-03-20 01:16:58 +0000458 db->aDb[1].zName = "temp";
drh75897232000-05-29 14:26:00 +0000459
drh58b95762000-06-02 01:17:37 +0000460 /* Attempt to read the schema */
drh8c90ebb2002-09-05 23:21:37 +0000461 sqliteRegisterBuiltinFunctions(db);
drh58b95762000-06-02 01:17:37 +0000462 rc = sqliteInit(db, pzErrMsg);
drhc67980b2002-06-14 20:54:14 +0000463 db->magic = SQLITE_MAGIC_OPEN;
drhdaffd0e2001-04-11 14:28:42 +0000464 if( sqlite_malloc_failed ){
drh6d4abfb2001-10-22 02:58:08 +0000465 sqlite_close(db);
drhdaffd0e2001-04-11 14:28:42 +0000466 goto no_mem_on_open;
467 }else if( rc!=SQLITE_OK && rc!=SQLITE_BUSY ){
drh58b95762000-06-02 01:17:37 +0000468 sqlite_close(db);
drh5edc3122001-09-13 21:53:09 +0000469 sqliteStrRealloc(pzErrMsg);
drh58b95762000-06-02 01:17:37 +0000470 return 0;
drhaacc5432002-01-06 17:07:40 +0000471 }else if( pzErrMsg ){
drhdaffd0e2001-04-11 14:28:42 +0000472 sqliteFree(*pzErrMsg);
drhbed86902000-06-02 13:27:59 +0000473 *pzErrMsg = 0;
drh75897232000-05-29 14:26:00 +0000474 }
drh491791a2002-07-18 00:34:09 +0000475
476 /* If the database is in formats 1 or 2, then upgrade it to
477 ** version 3. This will reconstruct all indices. If the
478 ** upgrade fails for any reason (ex: out of disk space, database
drhe78e8282003-01-19 03:59:45 +0000479 ** is read only, interrupt received, etc.) then refuse to open.
drh491791a2002-07-18 00:34:09 +0000480 */
drh92ed08a2002-07-30 18:43:40 +0000481 if( rc==SQLITE_OK && db->file_format<3 ){
drhc2311722002-07-19 17:46:38 +0000482 char *zErr = 0;
483 InitData initData;
drh491791a2002-07-18 00:34:09 +0000484 int meta[SQLITE_N_BTREE_META];
485
drhc2311722002-07-19 17:46:38 +0000486 initData.db = db;
487 initData.pzErrMsg = &zErr;
drh491791a2002-07-18 00:34:09 +0000488 db->file_format = 3;
489 rc = sqlite_exec(db,
490 "BEGIN; SELECT name FROM sqlite_master WHERE type='table';",
491 upgrade_3_callback,
drhc2311722002-07-19 17:46:38 +0000492 &initData,
drh491791a2002-07-18 00:34:09 +0000493 &zErr);
494 if( rc==SQLITE_OK ){
drh001bbcb2003-03-19 03:14:00 +0000495 sqliteBtreeGetMeta(db->aDb[0].pBt, meta);
drha9e99ae2002-08-13 23:02:57 +0000496 meta[2] = 4;
drh001bbcb2003-03-19 03:14:00 +0000497 sqliteBtreeUpdateMeta(db->aDb[0].pBt, meta);
drh491791a2002-07-18 00:34:09 +0000498 sqlite_exec(db, "COMMIT", 0, 0, 0);
499 }
500 if( rc!=SQLITE_OK ){
501 sqliteSetString(pzErrMsg,
502 "unable to upgrade database to the version 2.6 format",
503 zErr ? ": " : 0, zErr, 0);
drhc2311722002-07-19 17:46:38 +0000504 sqlite_freemem(zErr);
drh491791a2002-07-18 00:34:09 +0000505 sqliteStrRealloc(pzErrMsg);
506 sqlite_close(db);
507 return 0;
508 }
drhc2311722002-07-19 17:46:38 +0000509 sqlite_freemem(zErr);
drh491791a2002-07-18 00:34:09 +0000510 }
511
512 /* Return a pointer to the newly opened database structure */
drh75897232000-05-29 14:26:00 +0000513 return db;
drhdaffd0e2001-04-11 14:28:42 +0000514
515no_mem_on_open:
516 sqliteSetString(pzErrMsg, "out of memory", 0);
517 sqliteStrRealloc(pzErrMsg);
518 return 0;
drh75897232000-05-29 14:26:00 +0000519}
520
521/*
drhaf9ff332002-01-16 21:00:27 +0000522** Return the ROWID of the most recent insert
523*/
524int sqlite_last_insert_rowid(sqlite *db){
525 return db->lastRowid;
526}
527
528/*
drhc8d30ac2002-04-12 10:08:59 +0000529** Return the number of changes in the most recent call to sqlite_exec().
530*/
531int sqlite_changes(sqlite *db){
532 return db->nChange;
533}
534
535/*
drh50e5dad2001-09-15 00:57:28 +0000536** Close an existing SQLite database
537*/
538void sqlite_close(sqlite *db){
drh8e0a2f92002-02-23 23:45:45 +0000539 HashElem *i;
drh001bbcb2003-03-19 03:14:00 +0000540 int j;
drh94e92032003-02-16 22:21:32 +0000541 db->want_to_close = 1;
542 if( sqliteSafetyCheck(db) || sqliteSafetyOn(db) ){
543 /* printf("DID NOT CLOSE\n"); fflush(stdout); */
544 return;
545 }
drh247be432002-05-10 05:44:55 +0000546 db->magic = SQLITE_MAGIC_CLOSED;
drh001bbcb2003-03-19 03:14:00 +0000547 for(j=0; j<db->nDb; j++){
548 if( db->aDb[j].pBt ){
549 sqliteBtreeClose(db->aDb[j].pBt);
drh1c2d8412003-03-31 00:30:47 +0000550 db->aDb[j].pBt = 0;
drh001bbcb2003-03-19 03:14:00 +0000551 }
drh113088e2003-03-20 01:16:58 +0000552 if( j>=2 ){
553 sqliteFree(db->aDb[j].zName);
drh1c2d8412003-03-31 00:30:47 +0000554 db->aDb[j].zName = 0;
drh113088e2003-03-20 01:16:58 +0000555 }
drhf57b3392001-10-08 13:22:32 +0000556 }
drh1c2d8412003-03-31 00:30:47 +0000557 sqliteResetInternalSchema(db, 0);
558 assert( db->nDb<=2 );
559 assert( db->aDb==db->aDbStatic );
drh0bce8352002-02-28 00:41:10 +0000560 for(i=sqliteHashFirst(&db->aFunc); i; i=sqliteHashNext(i)){
561 FuncDef *pFunc, *pNext;
562 for(pFunc = (FuncDef*)sqliteHashData(i); pFunc; pFunc=pNext){
drh8e0a2f92002-02-23 23:45:45 +0000563 pNext = pFunc->pNext;
564 sqliteFree(pFunc);
565 }
566 }
drh0bce8352002-02-28 00:41:10 +0000567 sqliteHashClear(&db->aFunc);
drh75897232000-05-29 14:26:00 +0000568 sqliteFree(db);
569}
570
571/*
drh001bbcb2003-03-19 03:14:00 +0000572** Rollback all database files.
573*/
574void sqliteRollbackAll(sqlite *db){
575 int i;
576 for(i=0; i<db->nDb; i++){
577 if( db->aDb[i].pBt ){
578 sqliteBtreeRollback(db->aDb[i].pBt);
579 db->aDb[i].inTrans = 0;
580 }
581 }
582 sqliteRollbackInternalChanges(db);
583}
584
585/*
drhb86ccfb2003-01-28 23:13:10 +0000586** This routine does the work of either sqlite_exec() or sqlite_compile().
587** It works like sqlite_exec() if pVm==NULL and it works like sqlite_compile()
588** otherwise.
drh75897232000-05-29 14:26:00 +0000589*/
drhb86ccfb2003-01-28 23:13:10 +0000590static int sqliteMain(
drh75897232000-05-29 14:26:00 +0000591 sqlite *db, /* The database on which the SQL executes */
drh9f71c2e2001-11-03 23:57:09 +0000592 const char *zSql, /* The SQL to be executed */
drh75897232000-05-29 14:26:00 +0000593 sqlite_callback xCallback, /* Invoke this callback routine */
594 void *pArg, /* First argument to xCallback() */
drhb86ccfb2003-01-28 23:13:10 +0000595 const char **pzTail, /* OUT: Next statement after the first */
596 sqlite_vm **ppVm, /* OUT: The virtual machine */
597 char **pzErrMsg /* OUT: Write error messages here */
drh75897232000-05-29 14:26:00 +0000598){
599 Parse sParse;
drh75897232000-05-29 14:26:00 +0000600
drh799550b2003-01-18 17:04:08 +0000601 if( pzErrMsg ) *pzErrMsg = 0;
602 if( sqliteSafetyOn(db) ) goto exec_misuse;
drh58b95762000-06-02 01:17:37 +0000603 if( (db->flags & SQLITE_Initialized)==0 ){
drh1e0ccab2002-08-29 23:59:47 +0000604 int rc, cnt = 1;
605 while( (rc = sqliteInit(db, pzErrMsg))==SQLITE_BUSY
606 && db->xBusyCallback && db->xBusyCallback(db->pBusyArg, "", cnt++)!=0 ){}
drhdaffd0e2001-04-11 14:28:42 +0000607 if( rc!=SQLITE_OK ){
608 sqliteStrRealloc(pzErrMsg);
drh247be432002-05-10 05:44:55 +0000609 sqliteSafetyOff(db);
drhdaffd0e2001-04-11 14:28:42 +0000610 return rc;
611 }
drh7bc09d32002-11-01 01:55:36 +0000612 if( pzErrMsg ){
613 sqliteFree(*pzErrMsg);
614 *pzErrMsg = 0;
615 }
drh58b95762000-06-02 01:17:37 +0000616 }
drh491791a2002-07-18 00:34:09 +0000617 if( db->file_format<3 ){
618 sqliteSafetyOff(db);
619 sqliteSetString(pzErrMsg, "obsolete database file format", 0);
620 return SQLITE_ERROR;
621 }
drh326dce72003-01-29 14:06:07 +0000622 if( db->pVdbe==0 ){ db->nChange = 0; }
drh75897232000-05-29 14:26:00 +0000623 memset(&sParse, 0, sizeof(sParse));
624 sParse.db = db;
625 sParse.xCallback = xCallback;
626 sParse.pArg = pArg;
drhb86ccfb2003-01-28 23:13:10 +0000627 sParse.useCallback = ppVm==0;
drhb86ccfb2003-01-28 23:13:10 +0000628 if( db->xTrace ) db->xTrace(db->pTraceArg, zSql);
drh4c504392000-10-16 22:06:40 +0000629 sqliteRunParser(&sParse, zSql, pzErrMsg);
drhdaffd0e2001-04-11 14:28:42 +0000630 if( sqlite_malloc_failed ){
631 sqliteSetString(pzErrMsg, "out of memory", 0);
632 sParse.rc = SQLITE_NOMEM;
drh001bbcb2003-03-19 03:14:00 +0000633 sqliteRollbackAll(db);
drh1c2d8412003-03-31 00:30:47 +0000634 sqliteResetInternalSchema(db, 0);
drh001bbcb2003-03-19 03:14:00 +0000635 db->flags &= ~SQLITE_InTrans;
drhdaffd0e2001-04-11 14:28:42 +0000636 }
drh326dce72003-01-29 14:06:07 +0000637 if( sParse.rc==SQLITE_DONE ) sParse.rc = SQLITE_OK;
drhb798fa62002-09-03 19:43:23 +0000638 if( sParse.rc!=SQLITE_OK && pzErrMsg && *pzErrMsg==0 ){
639 sqliteSetString(pzErrMsg, sqlite_error_string(sParse.rc), 0);
640 }
drhdaffd0e2001-04-11 14:28:42 +0000641 sqliteStrRealloc(pzErrMsg);
drh50e5dad2001-09-15 00:57:28 +0000642 if( sParse.rc==SQLITE_SCHEMA ){
drh1c2d8412003-03-31 00:30:47 +0000643 sqliteResetInternalSchema(db, 0);
drh50e5dad2001-09-15 00:57:28 +0000644 }
drhb86ccfb2003-01-28 23:13:10 +0000645 if( sParse.useCallback==0 ){
646 assert( ppVm );
drh326dce72003-01-29 14:06:07 +0000647 *ppVm = (sqlite_vm*)sParse.pVdbe;
648 *pzTail = sParse.zTail;
drhb86ccfb2003-01-28 23:13:10 +0000649 }
drhc22bd472002-05-10 13:14:07 +0000650 if( sqliteSafetyOff(db) ) goto exec_misuse;
drh4c504392000-10-16 22:06:40 +0000651 return sParse.rc;
drhc22bd472002-05-10 13:14:07 +0000652
653exec_misuse:
654 if( pzErrMsg ){
655 *pzErrMsg = 0;
656 sqliteSetString(pzErrMsg, sqlite_error_string(SQLITE_MISUSE), 0);
657 sqliteStrRealloc(pzErrMsg);
658 }
659 return SQLITE_MISUSE;
drh75897232000-05-29 14:26:00 +0000660}
drh2dfbbca2000-07-28 14:32:48 +0000661
662/*
drhb86ccfb2003-01-28 23:13:10 +0000663** Execute SQL code. Return one of the SQLITE_ success/failure
664** codes. Also write an error message into memory obtained from
665** malloc() and make *pzErrMsg point to that message.
666**
667** If the SQL is a query, then for each row in the query result
668** the xCallback() function is called. pArg becomes the first
669** argument to xCallback(). If xCallback=NULL then no callback
670** is invoked, even for queries.
671*/
672int sqlite_exec(
673 sqlite *db, /* The database on which the SQL executes */
674 const char *zSql, /* The SQL to be executed */
675 sqlite_callback xCallback, /* Invoke this callback routine */
676 void *pArg, /* First argument to xCallback() */
677 char **pzErrMsg /* Write error messages here */
678){
679 return sqliteMain(db, zSql, xCallback, pArg, 0, 0, pzErrMsg);
680}
681
682/*
683** Compile a single statement of SQL into a virtual machine. Return one
684** of the SQLITE_ success/failure codes. Also write an error message into
685** memory obtained from malloc() and make *pzErrMsg point to that message.
686*/
687int sqlite_compile(
688 sqlite *db, /* The database on which the SQL executes */
689 const char *zSql, /* The SQL to be executed */
690 const char **pzTail, /* OUT: Next statement after the first */
691 sqlite_vm **ppVm, /* OUT: The virtual machine */
692 char **pzErrMsg /* OUT: Write error messages here */
693){
694 return sqliteMain(db, zSql, 0, 0, pzTail, ppVm, pzErrMsg);
695}
696
697/*
698** The following routine destroys a virtual machine that is created by
699** the sqlite_compile() routine.
700**
701** The integer returned is an SQLITE_ success/failure code that describes
702** the result of executing the virtual machine. An error message is
703** written into memory obtained from malloc and *pzErrMsg is made to
704** point to that error if pzErrMsg is not NULL. The calling routine
705** should use sqlite_freemem() to delete the message when it has finished
706** with it.
707*/
708int sqlite_finalize(
709 sqlite_vm *pVm, /* The virtual machine to be destroyed */
710 char **pzErrMsg /* OUT: Write error messages here */
711){
drh483750b2003-01-29 18:46:51 +0000712 int rc = sqliteVdbeFinalize((Vdbe*)pVm, pzErrMsg);
713 sqliteStrRealloc(pzErrMsg);
714 return rc;
drhb86ccfb2003-01-28 23:13:10 +0000715}
716
717/*
drhc22bd472002-05-10 13:14:07 +0000718** Return a static string that describes the kind of error specified in the
719** argument.
drh247be432002-05-10 05:44:55 +0000720*/
drhc22bd472002-05-10 13:14:07 +0000721const char *sqlite_error_string(int rc){
722 const char *z;
723 switch( rc ){
724 case SQLITE_OK: z = "not an error"; break;
725 case SQLITE_ERROR: z = "SQL logic error or missing database"; break;
726 case SQLITE_INTERNAL: z = "internal SQLite implementation flaw"; break;
727 case SQLITE_PERM: z = "access permission denied"; break;
728 case SQLITE_ABORT: z = "callback requested query abort"; break;
729 case SQLITE_BUSY: z = "database is locked"; break;
730 case SQLITE_LOCKED: z = "database table is locked"; break;
731 case SQLITE_NOMEM: z = "out of memory"; break;
732 case SQLITE_READONLY: z = "attempt to write a readonly database"; break;
733 case SQLITE_INTERRUPT: z = "interrupted"; break;
734 case SQLITE_IOERR: z = "disk I/O error"; break;
735 case SQLITE_CORRUPT: z = "database disk image is malformed"; break;
736 case SQLITE_NOTFOUND: z = "table or record not found"; break;
737 case SQLITE_FULL: z = "database is full"; break;
738 case SQLITE_CANTOPEN: z = "unable to open database file"; break;
739 case SQLITE_PROTOCOL: z = "database locking protocol failure"; break;
740 case SQLITE_EMPTY: z = "table contains no data"; break;
741 case SQLITE_SCHEMA: z = "database schema has changed"; break;
742 case SQLITE_TOOBIG: z = "too much data for one table row"; break;
743 case SQLITE_CONSTRAINT: z = "constraint failed"; break;
744 case SQLITE_MISMATCH: z = "datatype mismatch"; break;
745 case SQLITE_MISUSE: z = "library routine called out of sequence";break;
drh8766c342002-11-09 00:33:15 +0000746 case SQLITE_NOLFS: z = "kernel lacks large file support"; break;
drhed6c8672003-01-12 18:02:16 +0000747 case SQLITE_AUTH: z = "authorization denied"; break;
jplyon892f6712003-06-12 08:59:00 +0000748 case SQLITE_FORMAT: z = "auxiliary database format error"; break;
drhc22bd472002-05-10 13:14:07 +0000749 default: z = "unknown error"; break;
drh247be432002-05-10 05:44:55 +0000750 }
drhc22bd472002-05-10 13:14:07 +0000751 return z;
drh247be432002-05-10 05:44:55 +0000752}
753
754/*
drh2dfbbca2000-07-28 14:32:48 +0000755** This routine implements a busy callback that sleeps and tries
756** again until a timeout value is reached. The timeout value is
757** an integer number of milliseconds passed in as the first
758** argument.
759*/
drhdaffd0e2001-04-11 14:28:42 +0000760static int sqliteDefaultBusyCallback(
drh2dfbbca2000-07-28 14:32:48 +0000761 void *Timeout, /* Maximum amount of time to wait */
762 const char *NotUsed, /* The name of the table that is busy */
763 int count /* Number of times table has been busy */
764){
drh8cfbf082001-09-19 13:22:39 +0000765#if SQLITE_MIN_SLEEP_MS==1
766 int delay = 10;
drh2dfbbca2000-07-28 14:32:48 +0000767 int prior_delay = 0;
768 int timeout = (int)Timeout;
769 int i;
770
771 for(i=1; i<count; i++){
772 prior_delay += delay;
773 delay = delay*2;
drh8cfbf082001-09-19 13:22:39 +0000774 if( delay>=1000 ){
775 delay = 1000;
776 prior_delay += 1000*(count - i - 1);
drh2dfbbca2000-07-28 14:32:48 +0000777 break;
778 }
779 }
drh3109e022001-10-09 13:46:01 +0000780 if( prior_delay + delay > timeout ){
781 delay = timeout - prior_delay;
drh2dfbbca2000-07-28 14:32:48 +0000782 if( delay<=0 ) return 0;
783 }
drh8cfbf082001-09-19 13:22:39 +0000784 sqliteOsSleep(delay);
drh2dfbbca2000-07-28 14:32:48 +0000785 return 1;
786#else
787 int timeout = (int)Timeout;
788 if( (count+1)*1000 > timeout ){
789 return 0;
790 }
drh8cfbf082001-09-19 13:22:39 +0000791 sqliteOsSleep(1000);
drh2dfbbca2000-07-28 14:32:48 +0000792 return 1;
793#endif
794}
795
796/*
797** This routine sets the busy callback for an Sqlite database to the
798** given callback function with the given argument.
799*/
800void sqlite_busy_handler(
801 sqlite *db,
802 int (*xBusy)(void*,const char*,int),
803 void *pArg
804){
805 db->xBusyCallback = xBusy;
806 db->pBusyArg = pArg;
807}
808
809/*
810** This routine installs a default busy handler that waits for the
811** specified number of milliseconds before returning 0.
812*/
813void sqlite_busy_timeout(sqlite *db, int ms){
814 if( ms>0 ){
drhdaffd0e2001-04-11 14:28:42 +0000815 sqlite_busy_handler(db, sqliteDefaultBusyCallback, (void*)ms);
drh2dfbbca2000-07-28 14:32:48 +0000816 }else{
817 sqlite_busy_handler(db, 0, 0);
818 }
819}
drh4c504392000-10-16 22:06:40 +0000820
821/*
822** Cause any pending operation to stop at its earliest opportunity.
823*/
824void sqlite_interrupt(sqlite *db){
825 db->flags |= SQLITE_Interrupt;
826}
drhfa86c412002-02-02 15:01:15 +0000827
828/*
829** Windows systems should call this routine to free memory that
830** is returned in the in the errmsg parameter of sqlite_open() when
831** SQLite is a DLL. For some reason, it does not work to call free()
832** directly.
833**
834** Note that we need to call free() not sqliteFree() here, since every
835** string that is exported from SQLite should have already passed through
836** sqliteStrRealloc().
837*/
838void sqlite_freemem(void *p){ free(p); }
839
840/*
841** Windows systems need functions to call to return the sqlite_version
drhe78e8282003-01-19 03:59:45 +0000842** and sqlite_encoding strings since they are unable to access constants
843** within DLLs.
drhfa86c412002-02-02 15:01:15 +0000844*/
845const char *sqlite_libversion(void){ return sqlite_version; }
846const char *sqlite_libencoding(void){ return sqlite_encoding; }
drh8e0a2f92002-02-23 23:45:45 +0000847
848/*
849** Create new user-defined functions. The sqlite_create_function()
850** routine creates a regular function and sqlite_create_aggregate()
851** creates an aggregate function.
852**
853** Passing a NULL xFunc argument or NULL xStep and xFinalize arguments
854** disables the function. Calling sqlite_create_function() with the
855** same name and number of arguments as a prior call to
856** sqlite_create_aggregate() disables the prior call to
857** sqlite_create_aggregate(), and vice versa.
858**
859** If nArg is -1 it means that this function will accept any number
860** of arguments, including 0.
861*/
862int sqlite_create_function(
863 sqlite *db, /* Add the function to this database connection */
864 const char *zName, /* Name of the function to add */
865 int nArg, /* Number of arguments */
drh1350b032002-02-27 19:00:20 +0000866 void (*xFunc)(sqlite_func*,int,const char**), /* The implementation */
867 void *pUserData /* User data */
drh8e0a2f92002-02-23 23:45:45 +0000868){
drh0bce8352002-02-28 00:41:10 +0000869 FuncDef *p;
drh4b59ab52002-08-24 18:24:51 +0000870 int nName;
drhc22bd472002-05-10 13:14:07 +0000871 if( db==0 || zName==0 || sqliteSafetyCheck(db) ) return 1;
drh4b59ab52002-08-24 18:24:51 +0000872 nName = strlen(zName);
873 if( nName>255 ) return 1;
874 p = sqliteFindFunction(db, zName, nName, nArg, 1);
drh4e0f9952002-02-27 01:53:13 +0000875 if( p==0 ) return 1;
drh8e0a2f92002-02-23 23:45:45 +0000876 p->xFunc = xFunc;
877 p->xStep = 0;
878 p->xFinalize = 0;
drh1350b032002-02-27 19:00:20 +0000879 p->pUserData = pUserData;
drh8e0a2f92002-02-23 23:45:45 +0000880 return 0;
881}
882int sqlite_create_aggregate(
883 sqlite *db, /* Add the function to this database connection */
884 const char *zName, /* Name of the function to add */
885 int nArg, /* Number of arguments */
drh1350b032002-02-27 19:00:20 +0000886 void (*xStep)(sqlite_func*,int,const char**), /* The step function */
887 void (*xFinalize)(sqlite_func*), /* The finalizer */
888 void *pUserData /* User data */
drh8e0a2f92002-02-23 23:45:45 +0000889){
drh0bce8352002-02-28 00:41:10 +0000890 FuncDef *p;
drh4b59ab52002-08-24 18:24:51 +0000891 int nName;
drhc22bd472002-05-10 13:14:07 +0000892 if( db==0 || zName==0 || sqliteSafetyCheck(db) ) return 1;
drh4b59ab52002-08-24 18:24:51 +0000893 nName = strlen(zName);
894 if( nName>255 ) return 1;
895 p = sqliteFindFunction(db, zName, nName, nArg, 1);
drh4e0f9952002-02-27 01:53:13 +0000896 if( p==0 ) return 1;
drh8e0a2f92002-02-23 23:45:45 +0000897 p->xFunc = 0;
898 p->xStep = xStep;
899 p->xFinalize = xFinalize;
drh1350b032002-02-27 19:00:20 +0000900 p->pUserData = pUserData;
drh8e0a2f92002-02-23 23:45:45 +0000901 return 0;
902}
drhc9b84a12002-06-20 11:36:48 +0000903
904/*
drh411995d2002-06-25 19:31:18 +0000905** Change the datatype for all functions with a given name. See the
906** header comment for the prototype of this function in sqlite.h for
907** additional information.
drhc9b84a12002-06-20 11:36:48 +0000908*/
909int sqlite_function_type(sqlite *db, const char *zName, int dataType){
910 FuncDef *p = (FuncDef*)sqliteHashFind(&db->aFunc, zName, strlen(zName));
911 while( p ){
912 p->dataType = dataType;
913 p = p->pNext;
914 }
drhf46f9052002-06-22 02:33:38 +0000915 return SQLITE_OK;
drhc9b84a12002-06-20 11:36:48 +0000916}
drh411995d2002-06-25 19:31:18 +0000917
918/*
drh18de4822003-01-16 16:28:53 +0000919** Register a trace function. The pArg from the previously registered trace
920** is returned.
921**
922** A NULL trace function means that no tracing is executes. A non-NULL
923** trace is a pointer to a function that is invoked at the start of each
924** sqlite_exec().
925*/
926void *sqlite_trace(sqlite *db, void (*xTrace)(void*,const char*), void *pArg){
drh18de4822003-01-16 16:28:53 +0000927 void *pOld = db->pTraceArg;
928 db->xTrace = xTrace;
929 db->pTraceArg = pArg;
930 return pOld;
drh0d1a6432003-04-03 15:46:04 +0000931}
paulb0208cc2003-04-13 18:26:49 +0000932
933/*
drh13bff812003-04-15 01:19:47 +0000934** This routine is called to create a connection to a database BTree
935** driver. If zFilename is the name of a file, then that file is
936** opened and used. If zFilename is the magic name ":memory:" then
937** the database is stored in memory (and is thus forgotten as soon as
938** the connection is closed.) If zFilename is NULL then the database
939** is for temporary use only and is deleted as soon as the connection
940** is closed.
941**
drh13bff812003-04-15 01:19:47 +0000942** A temporary database can be either a disk file (that is automatically
943** deleted when the file is closed) or a set of red-black trees held in memory,
944** depending on the values of the TEMP_STORE compile-time macro and the
945** db->temp_store variable, according to the following chart:
946**
947** TEMP_STORE db->temp_store Location of temporary database
948** ---------- -------------- ------------------------------
949** 0 any file
950** 1 1 file
951** 1 2 memory
952** 1 0 file
953** 2 1 file
954** 2 2 memory
955** 2 0 memory
956** 3 any memory
paulb0208cc2003-04-13 18:26:49 +0000957*/
958int sqliteBtreeFactory(
959 const sqlite *db, /* Main database when opening aux otherwise 0 */
960 const char *zFilename, /* Name of the file containing the BTree database */
961 int omitJournal, /* if TRUE then do not journal this file */
962 int nCache, /* How many pages in the page cache */
963 Btree **ppBtree){ /* Pointer to new Btree object written here */
964
paulb0208cc2003-04-13 18:26:49 +0000965 assert( ppBtree != 0);
966
drh13bff812003-04-15 01:19:47 +0000967#ifndef SQLITE_OMIT_INMEMORYDB
968 if( zFilename==0 ){
969 if (TEMP_STORE == 0) {
paulb0208cc2003-04-13 18:26:49 +0000970 /* Always use file based temporary DB */
971 return sqliteBtreeOpen(0, omitJournal, nCache, ppBtree);
drh13bff812003-04-15 01:19:47 +0000972 } else if (TEMP_STORE == 1 || TEMP_STORE == 2) {
paulb0208cc2003-04-13 18:26:49 +0000973 /* Switch depending on compile-time and/or runtime settings. */
drh13bff812003-04-15 01:19:47 +0000974 int location = db->temp_store==0 ? TEMP_STORE : db->temp_store;
paulb0208cc2003-04-13 18:26:49 +0000975
976 if (location == 1) {
977 return sqliteBtreeOpen(zFilename, omitJournal, nCache, ppBtree);
978 } else {
drh8bf8dc92003-05-17 17:35:10 +0000979 return sqliteRbtreeOpen(0, 0, 0, ppBtree);
paulb0208cc2003-04-13 18:26:49 +0000980 }
981 } else {
982 /* Always use in-core DB */
drh8bf8dc92003-05-17 17:35:10 +0000983 return sqliteRbtreeOpen(0, 0, 0, ppBtree);
paulb0208cc2003-04-13 18:26:49 +0000984 }
drh13bff812003-04-15 01:19:47 +0000985 }else if( zFilename[0]==':' && strcmp(zFilename,":memory:")==0 ){
drh8bf8dc92003-05-17 17:35:10 +0000986 return sqliteRbtreeOpen(0, 0, 0, ppBtree);
drh13bff812003-04-15 01:19:47 +0000987 }else
988#endif
989 {
paulb0208cc2003-04-13 18:26:49 +0000990 return sqliteBtreeOpen(zFilename, omitJournal, nCache, ppBtree);
991 }
992}