blob: b485aa74833e8eb05aee4a347b9c63a18cc09e1c [file] [log] [blame]
drh4f26d6c2004-05-26 23:25:30 +00001/*
2** 2004 May 26
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**
13** This file contains code use to implement APIs that are part of the
14** VDBE.
15*/
16#include "sqliteInt.h"
17#include "vdbeInt.h"
18
shaneeec556d2008-10-12 00:27:53 +000019#ifndef SQLITE_OMIT_DEPRECATED
drhd89bd002005-01-22 03:03:54 +000020/*
21** Return TRUE (non-zero) of the statement supplied as an argument needs
22** to be recompiled. A statement needs to be recompiled whenever the
23** execution environment changes in a way that would alter the program
24** that sqlite3_prepare() generates. For example, if new functions or
25** collating sequences are registered or if an authorizer function is
26** added or changed.
drhd89bd002005-01-22 03:03:54 +000027*/
28int sqlite3_expired(sqlite3_stmt *pStmt){
29 Vdbe *p = (Vdbe*)pStmt;
30 return p==0 || p->expired;
31}
shaneeec556d2008-10-12 00:27:53 +000032#endif
drhd89bd002005-01-22 03:03:54 +000033
drhe30f4422007-08-21 16:15:55 +000034/*
drh413c3d32010-02-23 20:11:56 +000035** Check on a Vdbe to make sure it has not been finalized. Log
36** an error and return true if it has been finalized (or is otherwise
37** invalid). Return false if it is ok.
38*/
39static int vdbeSafety(Vdbe *p){
40 if( p->db==0 ){
41 sqlite3_log(SQLITE_MISUSE, "API called with finalized prepared statement");
42 return 1;
43 }else{
44 return 0;
45 }
46}
47static int vdbeSafetyNotNull(Vdbe *p){
48 if( p==0 ){
49 sqlite3_log(SQLITE_MISUSE, "API called with NULL prepared statement");
50 return 1;
51 }else{
52 return vdbeSafety(p);
53 }
54}
55
56/*
drhe30f4422007-08-21 16:15:55 +000057** The following routine destroys a virtual machine that is created by
58** the sqlite3_compile() routine. The integer returned is an SQLITE_
59** success/failure code that describes the result of executing the virtual
60** machine.
61**
62** This routine sets the error code and string returned by
63** sqlite3_errcode(), sqlite3_errmsg() and sqlite3_errmsg16().
64*/
65int sqlite3_finalize(sqlite3_stmt *pStmt){
66 int rc;
67 if( pStmt==0 ){
drh65bafa62010-09-29 01:54:00 +000068 /* IMPLEMENTATION-OF: R-57228-12904 Invoking sqlite3_finalize() on a NULL
69 ** pointer is a harmless no-op. */
drhe30f4422007-08-21 16:15:55 +000070 rc = SQLITE_OK;
71 }else{
72 Vdbe *v = (Vdbe*)pStmt;
danielk1977238746a2009-03-19 18:51:06 +000073 sqlite3 *db = v->db;
drh0ae0bfd2010-01-31 15:14:37 +000074#if SQLITE_THREADSAFE
75 sqlite3_mutex *mutex;
76#endif
drh413c3d32010-02-23 20:11:56 +000077 if( vdbeSafety(v) ) return SQLITE_MISUSE_BKPT;
drh18472fa2008-10-07 15:25:48 +000078#if SQLITE_THREADSAFE
drh0ae0bfd2010-01-31 15:14:37 +000079 mutex = v->db->mutex;
drh7e8b8482008-01-23 03:03:05 +000080#endif
drh86f8c192007-08-22 00:39:19 +000081 sqlite3_mutex_enter(mutex);
drhe30f4422007-08-21 16:15:55 +000082 rc = sqlite3VdbeFinalize(v);
danielk1977238746a2009-03-19 18:51:06 +000083 rc = sqlite3ApiExit(db, rc);
drh86f8c192007-08-22 00:39:19 +000084 sqlite3_mutex_leave(mutex);
drhe30f4422007-08-21 16:15:55 +000085 }
86 return rc;
87}
88
89/*
90** Terminate the current execution of an SQL statement and reset it
91** back to its starting state so that it can be reused. A success code from
92** the prior execution is returned.
93**
94** This routine sets the error code and string returned by
95** sqlite3_errcode(), sqlite3_errmsg() and sqlite3_errmsg16().
96*/
97int sqlite3_reset(sqlite3_stmt *pStmt){
98 int rc;
99 if( pStmt==0 ){
100 rc = SQLITE_OK;
101 }else{
102 Vdbe *v = (Vdbe*)pStmt;
103 sqlite3_mutex_enter(v->db->mutex);
drhc890fec2008-08-01 20:10:08 +0000104 rc = sqlite3VdbeReset(v);
dane0af83a2009-09-08 19:15:01 +0000105 sqlite3VdbeMakeReady(v, -1, 0, 0, 0, 0, 0);
drhe30f4422007-08-21 16:15:55 +0000106 assert( (rc & (v->db->errMask))==rc );
danielk1977238746a2009-03-19 18:51:06 +0000107 rc = sqlite3ApiExit(v->db, rc);
drhe30f4422007-08-21 16:15:55 +0000108 sqlite3_mutex_leave(v->db->mutex);
109 }
110 return rc;
111}
112
113/*
114** Set all the parameters in the compiled SQL statement to NULL.
115*/
116int sqlite3_clear_bindings(sqlite3_stmt *pStmt){
117 int i;
118 int rc = SQLITE_OK;
drh7297d1f2008-05-09 14:39:44 +0000119 Vdbe *p = (Vdbe*)pStmt;
drh18472fa2008-10-07 15:25:48 +0000120#if SQLITE_THREADSAFE
drh7e8b8482008-01-23 03:03:05 +0000121 sqlite3_mutex *mutex = ((Vdbe*)pStmt)->db->mutex;
122#endif
123 sqlite3_mutex_enter(mutex);
drh7297d1f2008-05-09 14:39:44 +0000124 for(i=0; i<p->nVar; i++){
125 sqlite3VdbeMemRelease(&p->aVar[i]);
126 p->aVar[i].flags = MEM_Null;
drhe30f4422007-08-21 16:15:55 +0000127 }
danc94b8592009-10-20 07:01:24 +0000128 if( p->isPrepareV2 && p->expmask ){
129 p->expired = 1;
130 }
drh7e8b8482008-01-23 03:03:05 +0000131 sqlite3_mutex_leave(mutex);
drhe30f4422007-08-21 16:15:55 +0000132 return rc;
133}
134
135
drh4f26d6c2004-05-26 23:25:30 +0000136/**************************** sqlite3_value_ *******************************
137** The following routines extract information from a Mem or sqlite3_value
138** structure.
139*/
140const void *sqlite3_value_blob(sqlite3_value *pVal){
141 Mem *p = (Mem*)pVal;
142 if( p->flags & (MEM_Blob|MEM_Str) ){
drhb21c8cd2007-08-21 19:33:56 +0000143 sqlite3VdbeMemExpandBlob(p);
drh1f0feef2007-05-15 13:27:07 +0000144 p->flags &= ~MEM_Str;
145 p->flags |= MEM_Blob;
drh42262532010-09-08 16:30:36 +0000146 return p->n ? p->z : 0;
drh4f26d6c2004-05-26 23:25:30 +0000147 }else{
148 return sqlite3_value_text(pVal);
149 }
150}
151int sqlite3_value_bytes(sqlite3_value *pVal){
drhb21c8cd2007-08-21 19:33:56 +0000152 return sqlite3ValueBytes(pVal, SQLITE_UTF8);
drh4f26d6c2004-05-26 23:25:30 +0000153}
154int sqlite3_value_bytes16(sqlite3_value *pVal){
drhb21c8cd2007-08-21 19:33:56 +0000155 return sqlite3ValueBytes(pVal, SQLITE_UTF16NATIVE);
drh4f26d6c2004-05-26 23:25:30 +0000156}
157double sqlite3_value_double(sqlite3_value *pVal){
drh6a6124e2004-06-27 01:56:33 +0000158 return sqlite3VdbeRealValue((Mem*)pVal);
drh4f26d6c2004-05-26 23:25:30 +0000159}
160int sqlite3_value_int(sqlite3_value *pVal){
drhb27b7f52008-12-10 18:03:45 +0000161 return (int)sqlite3VdbeIntValue((Mem*)pVal);
drh4f26d6c2004-05-26 23:25:30 +0000162}
drhefad9992004-06-22 12:13:55 +0000163sqlite_int64 sqlite3_value_int64(sqlite3_value *pVal){
drh6a6124e2004-06-27 01:56:33 +0000164 return sqlite3VdbeIntValue((Mem*)pVal);
drh4f26d6c2004-05-26 23:25:30 +0000165}
166const unsigned char *sqlite3_value_text(sqlite3_value *pVal){
drhb21c8cd2007-08-21 19:33:56 +0000167 return (const unsigned char *)sqlite3ValueText(pVal, SQLITE_UTF8);
drh4f26d6c2004-05-26 23:25:30 +0000168}
drh6c626082004-11-14 21:56:29 +0000169#ifndef SQLITE_OMIT_UTF16
drh4f26d6c2004-05-26 23:25:30 +0000170const void *sqlite3_value_text16(sqlite3_value* pVal){
drhb21c8cd2007-08-21 19:33:56 +0000171 return sqlite3ValueText(pVal, SQLITE_UTF16NATIVE);
drh4f26d6c2004-05-26 23:25:30 +0000172}
danielk1977d8123362004-06-12 09:25:12 +0000173const void *sqlite3_value_text16be(sqlite3_value *pVal){
drhb21c8cd2007-08-21 19:33:56 +0000174 return sqlite3ValueText(pVal, SQLITE_UTF16BE);
danielk1977d8123362004-06-12 09:25:12 +0000175}
176const void *sqlite3_value_text16le(sqlite3_value *pVal){
drhb21c8cd2007-08-21 19:33:56 +0000177 return sqlite3ValueText(pVal, SQLITE_UTF16LE);
danielk1977d8123362004-06-12 09:25:12 +0000178}
drh6c626082004-11-14 21:56:29 +0000179#endif /* SQLITE_OMIT_UTF16 */
drh4f26d6c2004-05-26 23:25:30 +0000180int sqlite3_value_type(sqlite3_value* pVal){
drhf4479502004-05-27 03:12:53 +0000181 return pVal->type;
drh4f26d6c2004-05-26 23:25:30 +0000182}
183
184/**************************** sqlite3_result_ *******************************
185** The following routines are used by user-defined functions to specify
186** the function result.
drh4c8555f2009-06-25 01:47:11 +0000187**
188** The setStrOrError() funtion calls sqlite3VdbeMemSetStr() to store the
189** result as a string or blob but if the string or blob is too large, it
190** then sets the error code to SQLITE_TOOBIG
drh4f26d6c2004-05-26 23:25:30 +0000191*/
drh4c8555f2009-06-25 01:47:11 +0000192static void setResultStrOrError(
193 sqlite3_context *pCtx, /* Function context */
194 const char *z, /* String pointer */
195 int n, /* Bytes in string, or negative */
196 u8 enc, /* Encoding of z. 0 for BLOBs */
197 void (*xDel)(void*) /* Destructor function */
198){
199 if( sqlite3VdbeMemSetStr(&pCtx->s, z, n, enc, xDel)==SQLITE_TOOBIG ){
200 sqlite3_result_error_toobig(pCtx);
201 }
202}
drh4f26d6c2004-05-26 23:25:30 +0000203void sqlite3_result_blob(
204 sqlite3_context *pCtx,
205 const void *z,
206 int n,
danielk1977d8123362004-06-12 09:25:12 +0000207 void (*xDel)(void *)
drh4f26d6c2004-05-26 23:25:30 +0000208){
drh5708d2d2005-06-22 10:53:59 +0000209 assert( n>=0 );
drh32bc3f62007-08-21 20:25:39 +0000210 assert( sqlite3_mutex_held(pCtx->s.db->mutex) );
drh4c8555f2009-06-25 01:47:11 +0000211 setResultStrOrError(pCtx, z, n, 0, xDel);
drh4f26d6c2004-05-26 23:25:30 +0000212}
213void sqlite3_result_double(sqlite3_context *pCtx, double rVal){
drh32bc3f62007-08-21 20:25:39 +0000214 assert( sqlite3_mutex_held(pCtx->s.db->mutex) );
drh4f26d6c2004-05-26 23:25:30 +0000215 sqlite3VdbeMemSetDouble(&pCtx->s, rVal);
216}
217void sqlite3_result_error(sqlite3_context *pCtx, const char *z, int n){
drh32bc3f62007-08-21 20:25:39 +0000218 assert( sqlite3_mutex_held(pCtx->s.db->mutex) );
drh69544ec2008-02-06 14:11:34 +0000219 pCtx->isError = SQLITE_ERROR;
drhb21c8cd2007-08-21 19:33:56 +0000220 sqlite3VdbeMemSetStr(&pCtx->s, z, n, SQLITE_UTF8, SQLITE_TRANSIENT);
drh4f26d6c2004-05-26 23:25:30 +0000221}
danielk1977a1686c92006-01-23 07:52:37 +0000222#ifndef SQLITE_OMIT_UTF16
drh4f26d6c2004-05-26 23:25:30 +0000223void sqlite3_result_error16(sqlite3_context *pCtx, const void *z, int n){
drh32bc3f62007-08-21 20:25:39 +0000224 assert( sqlite3_mutex_held(pCtx->s.db->mutex) );
drh69544ec2008-02-06 14:11:34 +0000225 pCtx->isError = SQLITE_ERROR;
drhb21c8cd2007-08-21 19:33:56 +0000226 sqlite3VdbeMemSetStr(&pCtx->s, z, n, SQLITE_UTF16NATIVE, SQLITE_TRANSIENT);
drh4f26d6c2004-05-26 23:25:30 +0000227}
danielk1977a1686c92006-01-23 07:52:37 +0000228#endif
drhf4479502004-05-27 03:12:53 +0000229void sqlite3_result_int(sqlite3_context *pCtx, int iVal){
drh32bc3f62007-08-21 20:25:39 +0000230 assert( sqlite3_mutex_held(pCtx->s.db->mutex) );
drh4f26d6c2004-05-26 23:25:30 +0000231 sqlite3VdbeMemSetInt64(&pCtx->s, (i64)iVal);
232}
233void sqlite3_result_int64(sqlite3_context *pCtx, i64 iVal){
drh32bc3f62007-08-21 20:25:39 +0000234 assert( sqlite3_mutex_held(pCtx->s.db->mutex) );
drh4f26d6c2004-05-26 23:25:30 +0000235 sqlite3VdbeMemSetInt64(&pCtx->s, iVal);
236}
237void sqlite3_result_null(sqlite3_context *pCtx){
drh32bc3f62007-08-21 20:25:39 +0000238 assert( sqlite3_mutex_held(pCtx->s.db->mutex) );
drhf4479502004-05-27 03:12:53 +0000239 sqlite3VdbeMemSetNull(&pCtx->s);
drh4f26d6c2004-05-26 23:25:30 +0000240}
241void sqlite3_result_text(
242 sqlite3_context *pCtx,
243 const char *z,
244 int n,
danielk1977d8123362004-06-12 09:25:12 +0000245 void (*xDel)(void *)
drh4f26d6c2004-05-26 23:25:30 +0000246){
drh32bc3f62007-08-21 20:25:39 +0000247 assert( sqlite3_mutex_held(pCtx->s.db->mutex) );
drh4c8555f2009-06-25 01:47:11 +0000248 setResultStrOrError(pCtx, z, n, SQLITE_UTF8, xDel);
drh4f26d6c2004-05-26 23:25:30 +0000249}
drh6c626082004-11-14 21:56:29 +0000250#ifndef SQLITE_OMIT_UTF16
drh4f26d6c2004-05-26 23:25:30 +0000251void sqlite3_result_text16(
252 sqlite3_context *pCtx,
253 const void *z,
254 int n,
danielk1977d8123362004-06-12 09:25:12 +0000255 void (*xDel)(void *)
drh4f26d6c2004-05-26 23:25:30 +0000256){
drh32bc3f62007-08-21 20:25:39 +0000257 assert( sqlite3_mutex_held(pCtx->s.db->mutex) );
drh4c8555f2009-06-25 01:47:11 +0000258 setResultStrOrError(pCtx, z, n, SQLITE_UTF16NATIVE, xDel);
danielk1977d8123362004-06-12 09:25:12 +0000259}
260void sqlite3_result_text16be(
261 sqlite3_context *pCtx,
262 const void *z,
263 int n,
264 void (*xDel)(void *)
265){
drh32bc3f62007-08-21 20:25:39 +0000266 assert( sqlite3_mutex_held(pCtx->s.db->mutex) );
drh4c8555f2009-06-25 01:47:11 +0000267 setResultStrOrError(pCtx, z, n, SQLITE_UTF16BE, xDel);
danielk1977d8123362004-06-12 09:25:12 +0000268}
269void sqlite3_result_text16le(
270 sqlite3_context *pCtx,
271 const void *z,
272 int n,
273 void (*xDel)(void *)
274){
drh32bc3f62007-08-21 20:25:39 +0000275 assert( sqlite3_mutex_held(pCtx->s.db->mutex) );
drh4c8555f2009-06-25 01:47:11 +0000276 setResultStrOrError(pCtx, z, n, SQLITE_UTF16LE, xDel);
drh4f26d6c2004-05-26 23:25:30 +0000277}
drh6c626082004-11-14 21:56:29 +0000278#endif /* SQLITE_OMIT_UTF16 */
drh4f26d6c2004-05-26 23:25:30 +0000279void sqlite3_result_value(sqlite3_context *pCtx, sqlite3_value *pValue){
drh32bc3f62007-08-21 20:25:39 +0000280 assert( sqlite3_mutex_held(pCtx->s.db->mutex) );
drhb21c8cd2007-08-21 19:33:56 +0000281 sqlite3VdbeMemCopy(&pCtx->s, pValue);
drh4f26d6c2004-05-26 23:25:30 +0000282}
drhb026e052007-05-02 01:34:31 +0000283void sqlite3_result_zeroblob(sqlite3_context *pCtx, int n){
drh32bc3f62007-08-21 20:25:39 +0000284 assert( sqlite3_mutex_held(pCtx->s.db->mutex) );
drhb026e052007-05-02 01:34:31 +0000285 sqlite3VdbeMemSetZeroBlob(&pCtx->s, n);
286}
drh69544ec2008-02-06 14:11:34 +0000287void sqlite3_result_error_code(sqlite3_context *pCtx, int errCode){
288 pCtx->isError = errCode;
drh51c7d862009-04-27 18:46:06 +0000289 if( pCtx->s.flags & MEM_Null ){
290 sqlite3VdbeMemSetStr(&pCtx->s, sqlite3ErrStr(errCode), -1,
291 SQLITE_UTF8, SQLITE_STATIC);
292 }
drh69544ec2008-02-06 14:11:34 +0000293}
drh4f26d6c2004-05-26 23:25:30 +0000294
drha0206bc2007-05-08 15:15:02 +0000295/* Force an SQLITE_TOOBIG error. */
296void sqlite3_result_error_toobig(sqlite3_context *pCtx){
drh32bc3f62007-08-21 20:25:39 +0000297 assert( sqlite3_mutex_held(pCtx->s.db->mutex) );
drhbb4957f2008-03-20 14:03:29 +0000298 pCtx->isError = SQLITE_TOOBIG;
299 sqlite3VdbeMemSetStr(&pCtx->s, "string or blob too big", -1,
300 SQLITE_UTF8, SQLITE_STATIC);
drha0206bc2007-05-08 15:15:02 +0000301}
302
danielk1977a1644fd2007-08-29 12:31:25 +0000303/* An SQLITE_NOMEM error. */
304void sqlite3_result_error_nomem(sqlite3_context *pCtx){
305 assert( sqlite3_mutex_held(pCtx->s.db->mutex) );
306 sqlite3VdbeMemSetNull(&pCtx->s);
drh69544ec2008-02-06 14:11:34 +0000307 pCtx->isError = SQLITE_NOMEM;
danielk1977a1644fd2007-08-29 12:31:25 +0000308 pCtx->s.db->mallocFailed = 1;
309}
drh4f26d6c2004-05-26 23:25:30 +0000310
dan5cf53532010-05-01 16:40:20 +0000311/*
312** This function is called after a transaction has been committed. It
313** invokes callbacks registered with sqlite3_wal_hook() as required.
314*/
drh7ed91f22010-04-29 22:34:07 +0000315static int doWalCallbacks(sqlite3 *db){
dan8d22a172010-04-19 18:03:51 +0000316 int rc = SQLITE_OK;
dan5cf53532010-05-01 16:40:20 +0000317#ifndef SQLITE_OMIT_WAL
318 int i;
dan8d22a172010-04-19 18:03:51 +0000319 for(i=0; i<db->nDb; i++){
320 Btree *pBt = db->aDb[i].pBt;
321 if( pBt ){
drh7ed91f22010-04-29 22:34:07 +0000322 int nEntry = sqlite3PagerWalCallback(sqlite3BtreePager(pBt));
drh5def0842010-05-05 20:00:25 +0000323 if( db->xWalCallback && nEntry>0 && rc==SQLITE_OK ){
324 rc = db->xWalCallback(db->pWalArg, db, db->aDb[i].zName, nEntry);
dan8d22a172010-04-19 18:03:51 +0000325 }
326 }
327 }
dan5cf53532010-05-01 16:40:20 +0000328#endif
dan8d22a172010-04-19 18:03:51 +0000329 return rc;
330}
331
drh4f26d6c2004-05-26 23:25:30 +0000332/*
333** Execute the statement pStmt, either until a row of data is ready, the
334** statement is completely executed or an error occurs.
drhb900aaf2006-11-09 00:24:53 +0000335**
336** This routine implements the bulk of the logic behind the sqlite_step()
337** API. The only thing omitted is the automatic recompile if a
338** schema change has occurred. That detail is handled by the
339** outer sqlite3_step() wrapper procedure.
drh4f26d6c2004-05-26 23:25:30 +0000340*/
drhb900aaf2006-11-09 00:24:53 +0000341static int sqlite3Step(Vdbe *p){
drh9bb575f2004-09-06 17:24:11 +0000342 sqlite3 *db;
drh4f26d6c2004-05-26 23:25:30 +0000343 int rc;
344
danielk1977a185ddb2007-11-15 16:04:15 +0000345 assert(p);
346 if( p->magic!=VDBE_MAGIC_RUN ){
drh3674bfd2010-04-17 12:53:19 +0000347 /* We used to require that sqlite3_reset() be called before retrying
drh602acb42011-01-17 17:42:37 +0000348 ** sqlite3_step() after any error or after SQLITE_DONE. But beginning
349 ** with version 3.7.0, we changed this so that sqlite3_reset() would
350 ** be called automatically instead of throwing the SQLITE_MISUSE error.
351 ** This "automatic-reset" change is not technically an incompatibility,
352 ** since any application that receives an SQLITE_MISUSE is broken by
353 ** definition.
354 **
355 ** Nevertheless, some published applications that were originally written
356 ** for version 3.6.23 or earlier do in fact depend on SQLITE_MISUSE
357 ** returns, and the so were broken by the automatic-reset change. As a
358 ** a work-around, the SQLITE_OMIT_AUTORESET compile-time restores the
359 ** legacy behavior of returning SQLITE_MISUSE for cases where the
360 ** previous sqlite3_step() returned something other than a SQLITE_LOCKED
361 ** or SQLITE_BUSY error.
drh3674bfd2010-04-17 12:53:19 +0000362 */
drh602acb42011-01-17 17:42:37 +0000363#ifdef SQLITE_OMIT_AUTORESET
364 if( p->rc==SQLITE_BUSY || p->rc==SQLITE_LOCKED ){
365 sqlite3_reset((sqlite3_stmt*)p);
366 }else{
367 return SQLITE_MISUSE_BKPT;
368 }
369#else
drh3674bfd2010-04-17 12:53:19 +0000370 sqlite3_reset((sqlite3_stmt*)p);
drh602acb42011-01-17 17:42:37 +0000371#endif
drhf1bfe9a2007-10-17 01:44:20 +0000372 }
373
dan14d4cc42010-01-20 14:25:37 +0000374 /* Check that malloc() has not failed. If it has, return early. */
drh17435752007-08-16 04:30:38 +0000375 db = p->db;
drhc456e572008-08-11 18:44:58 +0000376 if( db->mallocFailed ){
dan14d4cc42010-01-20 14:25:37 +0000377 p->rc = SQLITE_NOMEM;
drhc456e572008-08-11 18:44:58 +0000378 return SQLITE_NOMEM;
379 }
danielk1977261919c2005-12-06 12:52:59 +0000380
danielk1977a21c6b62005-01-24 10:25:59 +0000381 if( p->pc<=0 && p->expired ){
drhbee80652010-02-25 21:27:58 +0000382 p->rc = SQLITE_SCHEMA;
drhb900aaf2006-11-09 00:24:53 +0000383 rc = SQLITE_ERROR;
384 goto end_of_step;
danielk1977a21c6b62005-01-24 10:25:59 +0000385 }
danielk19771d850a72004-05-31 08:26:49 +0000386 if( p->pc<0 ){
drh15ca1df2006-07-26 13:43:30 +0000387 /* If there are no other statements currently running, then
388 ** reset the interrupt flag. This prevents a call to sqlite3_interrupt
389 ** from interrupting a statement that has not yet started.
390 */
391 if( db->activeVdbeCnt==0 ){
392 db->u1.isInterrupted = 0;
393 }
394
dan1da40a32009-09-19 17:00:31 +0000395 assert( db->writeVdbeCnt>0 || db->autoCommit==0 || db->nDeferredCons==0 );
396
drh19e2d372005-08-29 23:00:03 +0000397#ifndef SQLITE_OMIT_TRACE
drh19e2d372005-08-29 23:00:03 +0000398 if( db->xProfile && !db->init.busy ){
drhb7e8ea22010-05-03 14:32:30 +0000399 sqlite3OsCurrentTimeInt64(db->pVfs, &p->startTime);
drh19e2d372005-08-29 23:00:03 +0000400 }
401#endif
drhc16a03b2004-09-15 13:38:10 +0000402
danielk19771d850a72004-05-31 08:26:49 +0000403 db->activeVdbeCnt++;
dan6566ebe2011-03-16 09:49:14 +0000404 if( p->readOnly==0 ) db->writeVdbeCnt++;
danielk19771d850a72004-05-31 08:26:49 +0000405 p->pc = 0;
406 }
drhb7f91642004-10-31 02:22:47 +0000407#ifndef SQLITE_OMIT_EXPLAIN
drh4f26d6c2004-05-26 23:25:30 +0000408 if( p->explain ){
409 rc = sqlite3VdbeList(p);
drhb7f91642004-10-31 02:22:47 +0000410 }else
411#endif /* SQLITE_OMIT_EXPLAIN */
412 {
dan27381bd2011-01-22 13:32:29 +0000413 db->vdbeExecCnt++;
drh4f26d6c2004-05-26 23:25:30 +0000414 rc = sqlite3VdbeExec(p);
dan27381bd2011-01-22 13:32:29 +0000415 db->vdbeExecCnt--;
drh4f26d6c2004-05-26 23:25:30 +0000416 }
417
drh19e2d372005-08-29 23:00:03 +0000418#ifndef SQLITE_OMIT_TRACE
419 /* Invoke the profile callback if there is one
420 */
danielk19776ab3a2e2009-02-19 14:39:25 +0000421 if( rc!=SQLITE_ROW && db->xProfile && !db->init.busy && p->zSql ){
drhb7e8ea22010-05-03 14:32:30 +0000422 sqlite3_int64 iNow;
423 sqlite3OsCurrentTimeInt64(db->pVfs, &iNow);
drhdf0db0f2010-07-29 10:07:21 +0000424 db->xProfile(db->pProfileArg, p->zSql, (iNow - p->startTime)*1000000);
drh19e2d372005-08-29 23:00:03 +0000425 }
426#endif
427
dan8d22a172010-04-19 18:03:51 +0000428 if( rc==SQLITE_DONE ){
429 assert( p->rc==SQLITE_OK );
drh7ed91f22010-04-29 22:34:07 +0000430 p->rc = doWalCallbacks(db);
dan8d22a172010-04-19 18:03:51 +0000431 if( p->rc!=SQLITE_OK ){
432 rc = SQLITE_ERROR;
433 }
434 }
435
drh80cc85b2008-07-23 21:07:25 +0000436 db->errCode = rc;
danielk1977357864e2009-03-25 15:43:08 +0000437 if( SQLITE_NOMEM==sqlite3ApiExit(p->db, p->rc) ){
438 p->rc = SQLITE_NOMEM;
drhb900aaf2006-11-09 00:24:53 +0000439 }
danielk1977357864e2009-03-25 15:43:08 +0000440end_of_step:
441 /* At this point local variable rc holds the value that should be
442 ** returned if this statement was compiled using the legacy
443 ** sqlite3_prepare() interface. According to the docs, this can only
444 ** be one of the values in the first assert() below. Variable p->rc
445 ** contains the value that would be returned if sqlite3_finalize()
446 ** were called on statement p.
447 */
448 assert( rc==SQLITE_ROW || rc==SQLITE_DONE || rc==SQLITE_ERROR
449 || rc==SQLITE_BUSY || rc==SQLITE_MISUSE
450 );
451 assert( p->rc!=SQLITE_ROW && p->rc!=SQLITE_DONE );
452 if( p->isPrepareV2 && rc!=SQLITE_ROW && rc!=SQLITE_DONE ){
453 /* If this statement was prepared using sqlite3_prepare_v2(), and an
454 ** error has occured, then return the error code in p->rc to the
455 ** caller. Set the error code in the database handle to the same value.
456 */
457 rc = db->errCode = p->rc;
458 }
459 return (rc&db->errMask);
drhb900aaf2006-11-09 00:24:53 +0000460}
461
462/*
463** This is the top-level implementation of sqlite3_step(). Call
464** sqlite3Step() to do most of the work. If a schema error occurs,
465** call sqlite3Reprepare() and try again.
466*/
drhb900aaf2006-11-09 00:24:53 +0000467int sqlite3_step(sqlite3_stmt *pStmt){
drha6129fa2010-02-24 17:15:19 +0000468 int rc = SQLITE_OK; /* Result from sqlite3Step() */
469 int rc2 = SQLITE_OK; /* Result from sqlite3Reprepare() */
470 Vdbe *v = (Vdbe*)pStmt; /* the prepared statement */
471 int cnt = 0; /* Counter to prevent infinite loop of reprepares */
472 sqlite3 *db; /* The database connection */
473
drh413c3d32010-02-23 20:11:56 +0000474 if( vdbeSafetyNotNull(v) ){
475 return SQLITE_MISUSE_BKPT;
danielk19778e556522007-11-13 10:30:24 +0000476 }
drh413c3d32010-02-23 20:11:56 +0000477 db = v->db;
478 sqlite3_mutex_enter(db->mutex);
479 while( (rc = sqlite3Step(v))==SQLITE_SCHEMA
480 && cnt++ < 5
drha6129fa2010-02-24 17:15:19 +0000481 && (rc2 = rc = sqlite3Reprepare(v))==SQLITE_OK ){
drh413c3d32010-02-23 20:11:56 +0000482 sqlite3_reset(pStmt);
483 v->expired = 0;
484 }
drhbee80652010-02-25 21:27:58 +0000485 if( rc2!=SQLITE_OK && ALWAYS(v->isPrepareV2) && ALWAYS(db->pErr) ){
drh413c3d32010-02-23 20:11:56 +0000486 /* This case occurs after failing to recompile an sql statement.
487 ** The error message from the SQL compiler has already been loaded
488 ** into the database handle. This block copies the error message
489 ** from the database handle into the statement and sets the statement
490 ** program counter to 0 to ensure that when the statement is
491 ** finalized or reset the parser error message is available via
492 ** sqlite3_errmsg() and sqlite3_errcode().
493 */
494 const char *zErr = (const char *)sqlite3_value_text(db->pErr);
495 sqlite3DbFree(db, v->zErrMsg);
496 if( !db->mallocFailed ){
497 v->zErrMsg = sqlite3DbStrDup(db, zErr);
drha6129fa2010-02-24 17:15:19 +0000498 v->rc = rc2;
drh413c3d32010-02-23 20:11:56 +0000499 } else {
500 v->zErrMsg = 0;
drha6129fa2010-02-24 17:15:19 +0000501 v->rc = rc = SQLITE_NOMEM;
drh413c3d32010-02-23 20:11:56 +0000502 }
503 }
504 rc = sqlite3ApiExit(db, rc);
505 sqlite3_mutex_leave(db->mutex);
danielk197776e8d1a2006-01-18 18:22:43 +0000506 return rc;
drh4f26d6c2004-05-26 23:25:30 +0000507}
508
509/*
drheb2e1762004-05-27 01:53:56 +0000510** Extract the user data from a sqlite3_context structure and return a
511** pointer to it.
512*/
513void *sqlite3_user_data(sqlite3_context *p){
514 assert( p && p->pFunc );
515 return p->pFunc->pUserData;
516}
517
518/*
drhfa4a4b92008-03-19 21:45:51 +0000519** Extract the user data from a sqlite3_context structure and return a
520** pointer to it.
drh9f129f42010-08-31 15:27:32 +0000521**
522** IMPLEMENTATION-OF: R-46798-50301 The sqlite3_context_db_handle() interface
523** returns a copy of the pointer to the database connection (the 1st
524** parameter) of the sqlite3_create_function() and
525** sqlite3_create_function16() routines that originally registered the
526** application defined function.
drhfa4a4b92008-03-19 21:45:51 +0000527*/
528sqlite3 *sqlite3_context_db_handle(sqlite3_context *p){
529 assert( p && p->pFunc );
530 return p->s.db;
531}
532
533/*
drhb7481e72006-09-16 21:45:14 +0000534** The following is the implementation of an SQL function that always
535** fails with an error message stating that the function is used in the
536** wrong context. The sqlite3_overload_function() API might construct
537** SQL function that use this routine so that the functions will exist
538** for name resolution but are actually overloaded by the xFindFunction
539** method of virtual tables.
540*/
541void sqlite3InvalidFunction(
542 sqlite3_context *context, /* The function calling context */
danielk197762c14b32008-11-19 09:05:26 +0000543 int NotUsed, /* Number of arguments to the function */
544 sqlite3_value **NotUsed2 /* Value of each argument */
drhb7481e72006-09-16 21:45:14 +0000545){
546 const char *zName = context->pFunc->zName;
547 char *zErr;
danielk197762c14b32008-11-19 09:05:26 +0000548 UNUSED_PARAMETER2(NotUsed, NotUsed2);
drhbc6160b2009-04-08 15:45:31 +0000549 zErr = sqlite3_mprintf(
drhb7481e72006-09-16 21:45:14 +0000550 "unable to use function %s in the requested context", zName);
551 sqlite3_result_error(context, zErr, -1);
danielk19771e536952007-08-16 10:09:01 +0000552 sqlite3_free(zErr);
drhb7481e72006-09-16 21:45:14 +0000553}
554
555/*
drheb2e1762004-05-27 01:53:56 +0000556** Allocate or return the aggregate context for a user function. A new
557** context is allocated on the first call. Subsequent calls return the
558** same context that was returned on prior calls.
drheb2e1762004-05-27 01:53:56 +0000559*/
560void *sqlite3_aggregate_context(sqlite3_context *p, int nByte){
drhb21c8cd2007-08-21 19:33:56 +0000561 Mem *pMem;
drh825c6622005-09-08 19:01:05 +0000562 assert( p && p->pFunc && p->pFunc->xStep );
drhb21c8cd2007-08-21 19:33:56 +0000563 assert( sqlite3_mutex_held(p->s.db->mutex) );
564 pMem = p->pMem;
drhd68eee02009-12-11 03:44:18 +0000565 testcase( nByte<0 );
drha10a34b2005-09-07 22:09:48 +0000566 if( (pMem->flags & MEM_Agg)==0 ){
drhd68eee02009-12-11 03:44:18 +0000567 if( nByte<=0 ){
danielk19775f096132008-03-28 15:44:09 +0000568 sqlite3VdbeMemReleaseExternal(pMem);
569 pMem->flags = MEM_Null;
drha10a34b2005-09-07 22:09:48 +0000570 pMem->z = 0;
drheb2e1762004-05-27 01:53:56 +0000571 }else{
danielk19775f096132008-03-28 15:44:09 +0000572 sqlite3VdbeMemGrow(pMem, nByte, 0);
drha10a34b2005-09-07 22:09:48 +0000573 pMem->flags = MEM_Agg;
drh3c024d62007-03-30 11:23:45 +0000574 pMem->u.pDef = p->pFunc;
danielk19775f096132008-03-28 15:44:09 +0000575 if( pMem->z ){
576 memset(pMem->z, 0, nByte);
577 }
drheb2e1762004-05-27 01:53:56 +0000578 }
579 }
drhabfcea22005-09-06 20:36:48 +0000580 return (void*)pMem->z;
drheb2e1762004-05-27 01:53:56 +0000581}
582
583/*
danielk1977682f68b2004-06-05 10:22:17 +0000584** Return the auxilary data pointer, if any, for the iArg'th argument to
585** the user-function defined by pCtx.
586*/
587void *sqlite3_get_auxdata(sqlite3_context *pCtx, int iArg){
drhb21c8cd2007-08-21 19:33:56 +0000588 VdbeFunc *pVdbeFunc;
589
590 assert( sqlite3_mutex_held(pCtx->s.db->mutex) );
591 pVdbeFunc = pCtx->pVdbeFunc;
danielk1977682f68b2004-06-05 10:22:17 +0000592 if( !pVdbeFunc || iArg>=pVdbeFunc->nAux || iArg<0 ){
593 return 0;
594 }
drhf92c7ff2004-06-19 15:40:23 +0000595 return pVdbeFunc->apAux[iArg].pAux;
danielk1977682f68b2004-06-05 10:22:17 +0000596}
597
598/*
599** Set the auxilary data pointer and delete function, for the iArg'th
600** argument to the user-function defined by pCtx. Any previous value is
601** deleted by calling the delete function specified when it was set.
602*/
603void sqlite3_set_auxdata(
604 sqlite3_context *pCtx,
605 int iArg,
606 void *pAux,
607 void (*xDelete)(void*)
608){
609 struct AuxData *pAuxData;
drhf92c7ff2004-06-19 15:40:23 +0000610 VdbeFunc *pVdbeFunc;
danielk1977e0fc5262007-07-26 06:50:05 +0000611 if( iArg<0 ) goto failed;
danielk1977682f68b2004-06-05 10:22:17 +0000612
drhb21c8cd2007-08-21 19:33:56 +0000613 assert( sqlite3_mutex_held(pCtx->s.db->mutex) );
drhf92c7ff2004-06-19 15:40:23 +0000614 pVdbeFunc = pCtx->pVdbeFunc;
615 if( !pVdbeFunc || pVdbeFunc->nAux<=iArg ){
danielk197731dad9d2007-08-16 11:36:15 +0000616 int nAux = (pVdbeFunc ? pVdbeFunc->nAux : 0);
drhf92c7ff2004-06-19 15:40:23 +0000617 int nMalloc = sizeof(VdbeFunc) + sizeof(struct AuxData)*iArg;
danielk197726783a52007-08-29 14:06:22 +0000618 pVdbeFunc = sqlite3DbRealloc(pCtx->s.db, pVdbeFunc, nMalloc);
drh17435752007-08-16 04:30:38 +0000619 if( !pVdbeFunc ){
drh17435752007-08-16 04:30:38 +0000620 goto failed;
621 }
drh53f733c2005-09-16 02:38:09 +0000622 pCtx->pVdbeFunc = pVdbeFunc;
danielk197731dad9d2007-08-16 11:36:15 +0000623 memset(&pVdbeFunc->apAux[nAux], 0, sizeof(struct AuxData)*(iArg+1-nAux));
drh998da3a2004-06-19 15:22:56 +0000624 pVdbeFunc->nAux = iArg+1;
625 pVdbeFunc->pFunc = pCtx->pFunc;
danielk1977682f68b2004-06-05 10:22:17 +0000626 }
627
drhf92c7ff2004-06-19 15:40:23 +0000628 pAuxData = &pVdbeFunc->apAux[iArg];
danielk1977682f68b2004-06-05 10:22:17 +0000629 if( pAuxData->pAux && pAuxData->xDelete ){
630 pAuxData->xDelete(pAuxData->pAux);
631 }
632 pAuxData->pAux = pAux;
633 pAuxData->xDelete = xDelete;
danielk1977e0fc5262007-07-26 06:50:05 +0000634 return;
635
636failed:
637 if( xDelete ){
638 xDelete(pAux);
639 }
danielk1977682f68b2004-06-05 10:22:17 +0000640}
641
shaneeec556d2008-10-12 00:27:53 +0000642#ifndef SQLITE_OMIT_DEPRECATED
danielk1977682f68b2004-06-05 10:22:17 +0000643/*
drheb2e1762004-05-27 01:53:56 +0000644** Return the number of times the Step function of a aggregate has been
645** called.
646**
drhcf85a512006-02-09 18:35:29 +0000647** This function is deprecated. Do not use it for new code. It is
648** provide only to avoid breaking legacy code. New aggregate function
649** implementations should keep their own counts within their aggregate
650** context.
drheb2e1762004-05-27 01:53:56 +0000651*/
652int sqlite3_aggregate_count(sqlite3_context *p){
shanee34c6472009-03-05 04:23:47 +0000653 assert( p && p->pMem && p->pFunc && p->pFunc->xStep );
drhabfcea22005-09-06 20:36:48 +0000654 return p->pMem->n;
drheb2e1762004-05-27 01:53:56 +0000655}
shaneeec556d2008-10-12 00:27:53 +0000656#endif
drheb2e1762004-05-27 01:53:56 +0000657
658/*
drh4f26d6c2004-05-26 23:25:30 +0000659** Return the number of columns in the result set for the statement pStmt.
660*/
661int sqlite3_column_count(sqlite3_stmt *pStmt){
662 Vdbe *pVm = (Vdbe *)pStmt;
drh1bcdb0c2004-08-28 14:49:46 +0000663 return pVm ? pVm->nResColumn : 0;
drh4f26d6c2004-05-26 23:25:30 +0000664}
665
666/*
667** Return the number of values available from the current row of the
668** currently executing statement pStmt.
669*/
670int sqlite3_data_count(sqlite3_stmt *pStmt){
671 Vdbe *pVm = (Vdbe *)pStmt;
drhd4e70eb2008-01-02 00:34:36 +0000672 if( pVm==0 || pVm->pResultSet==0 ) return 0;
drh4f26d6c2004-05-26 23:25:30 +0000673 return pVm->nResColumn;
674}
675
dan46c47d42011-03-01 18:42:07 +0000676/*
677** Return a pointer to static memory containing an SQL NULL value.
678*/
679static const Mem *columnNullValue(void){
680 /* Even though the Mem structure contains an element
681 ** of type i64, on certain architecture (x86) with certain compiler
682 ** switches (-Os), gcc may align this Mem object on a 4-byte boundary
683 ** instead of an 8-byte one. This all works fine, except that when
684 ** running with SQLITE_DEBUG defined the SQLite code sometimes assert()s
685 ** that a Mem structure is located on an 8-byte boundary. To prevent
686 ** this assert() from failing, when building with SQLITE_DEBUG defined
687 ** using gcc, force nullMem to be 8-byte aligned using the magical
688 ** __attribute__((aligned(8))) macro. */
689 static const Mem nullMem
690#if defined(SQLITE_DEBUG) && defined(__GNUC__)
691 __attribute__((aligned(8)))
692#endif
693 = {0, "", (double)0, {0}, 0, MEM_Null, SQLITE_NULL, 0, 0, 0 };
694 return &nullMem;
695}
drh4f26d6c2004-05-26 23:25:30 +0000696
697/*
698** Check to see if column iCol of the given statement is valid. If
699** it is, return a pointer to the Mem for the value of that column.
700** If iCol is not valid, return a pointer to a Mem which has a value
701** of NULL.
702*/
703static Mem *columnMem(sqlite3_stmt *pStmt, int i){
drh32bc3f62007-08-21 20:25:39 +0000704 Vdbe *pVm;
drh32bc3f62007-08-21 20:25:39 +0000705 Mem *pOut;
706
707 pVm = (Vdbe *)pStmt;
drhd4e70eb2008-01-02 00:34:36 +0000708 if( pVm && pVm->pResultSet!=0 && i<pVm->nResColumn && i>=0 ){
drh32bc3f62007-08-21 20:25:39 +0000709 sqlite3_mutex_enter(pVm->db->mutex);
drhd4e70eb2008-01-02 00:34:36 +0000710 pOut = &pVm->pResultSet[i];
drh32bc3f62007-08-21 20:25:39 +0000711 }else{
danielk197741f5b042009-06-06 14:13:26 +0000712 /* If the value passed as the second argument is out of range, return
dan46c47d42011-03-01 18:42:07 +0000713 ** a pointer to a static Mem object that contains the value SQL NULL.
714 */
drh9b4ea4a2009-04-10 20:32:00 +0000715 if( pVm && ALWAYS(pVm->db) ){
drh27641702007-08-22 02:56:42 +0000716 sqlite3_mutex_enter(pVm->db->mutex);
717 sqlite3Error(pVm->db, SQLITE_RANGE, 0);
718 }
dan46c47d42011-03-01 18:42:07 +0000719 pOut = (Mem*)columnNullValue();
drh4f26d6c2004-05-26 23:25:30 +0000720 }
drh32bc3f62007-08-21 20:25:39 +0000721 return pOut;
drh4f26d6c2004-05-26 23:25:30 +0000722}
723
danielk19772e588c72005-12-09 14:25:08 +0000724/*
725** This function is called after invoking an sqlite3_value_XXX function on a
726** column value (i.e. a value returned by evaluating an SQL expression in the
727** select list of a SELECT statement) that may cause a malloc() failure. If
728** malloc() has failed, the threads mallocFailed flag is cleared and the result
729** code of statement pStmt set to SQLITE_NOMEM.
730**
drh32bc3f62007-08-21 20:25:39 +0000731** Specifically, this is called from within:
danielk19772e588c72005-12-09 14:25:08 +0000732**
733** sqlite3_column_int()
734** sqlite3_column_int64()
735** sqlite3_column_text()
736** sqlite3_column_text16()
737** sqlite3_column_real()
738** sqlite3_column_bytes()
739** sqlite3_column_bytes16()
drh42262532010-09-08 16:30:36 +0000740** sqiite3_column_blob()
danielk19772e588c72005-12-09 14:25:08 +0000741*/
742static void columnMallocFailure(sqlite3_stmt *pStmt)
743{
744 /* If malloc() failed during an encoding conversion within an
745 ** sqlite3_column_XXX API, then set the return code of the statement to
746 ** SQLITE_NOMEM. The next call to _step() (if any) will return SQLITE_ERROR
747 ** and _finalize() will return NOMEM.
748 */
danielk197754f01982006-01-18 15:25:17 +0000749 Vdbe *p = (Vdbe *)pStmt;
drh32bc3f62007-08-21 20:25:39 +0000750 if( p ){
751 p->rc = sqlite3ApiExit(p->db, p->rc);
752 sqlite3_mutex_leave(p->db->mutex);
753 }
danielk19772e588c72005-12-09 14:25:08 +0000754}
755
drh4f26d6c2004-05-26 23:25:30 +0000756/**************************** sqlite3_column_ *******************************
757** The following routines are used to access elements of the current row
758** in the result set.
759*/
danielk1977c572ef72004-05-27 09:28:41 +0000760const void *sqlite3_column_blob(sqlite3_stmt *pStmt, int i){
danielk19772e588c72005-12-09 14:25:08 +0000761 const void *val;
danielk19772e588c72005-12-09 14:25:08 +0000762 val = sqlite3_value_blob( columnMem(pStmt,i) );
danielk1977c9cf9012007-05-30 10:36:47 +0000763 /* Even though there is no encoding conversion, value_blob() might
764 ** need to call malloc() to expand the result of a zeroblob()
765 ** expression.
766 */
767 columnMallocFailure(pStmt);
danielk19772e588c72005-12-09 14:25:08 +0000768 return val;
danielk1977c572ef72004-05-27 09:28:41 +0000769}
drh4f26d6c2004-05-26 23:25:30 +0000770int sqlite3_column_bytes(sqlite3_stmt *pStmt, int i){
danielk19772e588c72005-12-09 14:25:08 +0000771 int val = sqlite3_value_bytes( columnMem(pStmt,i) );
772 columnMallocFailure(pStmt);
773 return val;
drh4f26d6c2004-05-26 23:25:30 +0000774}
775int sqlite3_column_bytes16(sqlite3_stmt *pStmt, int i){
danielk19772e588c72005-12-09 14:25:08 +0000776 int val = sqlite3_value_bytes16( columnMem(pStmt,i) );
777 columnMallocFailure(pStmt);
778 return val;
drh4f26d6c2004-05-26 23:25:30 +0000779}
780double sqlite3_column_double(sqlite3_stmt *pStmt, int i){
danielk19772e588c72005-12-09 14:25:08 +0000781 double val = sqlite3_value_double( columnMem(pStmt,i) );
782 columnMallocFailure(pStmt);
783 return val;
drh4f26d6c2004-05-26 23:25:30 +0000784}
785int sqlite3_column_int(sqlite3_stmt *pStmt, int i){
danielk19772e588c72005-12-09 14:25:08 +0000786 int val = sqlite3_value_int( columnMem(pStmt,i) );
787 columnMallocFailure(pStmt);
788 return val;
drh4f26d6c2004-05-26 23:25:30 +0000789}
drhefad9992004-06-22 12:13:55 +0000790sqlite_int64 sqlite3_column_int64(sqlite3_stmt *pStmt, int i){
danielk19772e588c72005-12-09 14:25:08 +0000791 sqlite_int64 val = sqlite3_value_int64( columnMem(pStmt,i) );
792 columnMallocFailure(pStmt);
793 return val;
drh4f26d6c2004-05-26 23:25:30 +0000794}
795const unsigned char *sqlite3_column_text(sqlite3_stmt *pStmt, int i){
danielk19772e588c72005-12-09 14:25:08 +0000796 const unsigned char *val = sqlite3_value_text( columnMem(pStmt,i) );
797 columnMallocFailure(pStmt);
798 return val;
drh4f26d6c2004-05-26 23:25:30 +0000799}
drhd1e47332005-06-26 17:55:33 +0000800sqlite3_value *sqlite3_column_value(sqlite3_stmt *pStmt, int i){
danielk1977d0ffa1e2008-10-13 10:37:49 +0000801 Mem *pOut = columnMem(pStmt, i);
802 if( pOut->flags&MEM_Static ){
803 pOut->flags &= ~MEM_Static;
804 pOut->flags |= MEM_Ephem;
805 }
drh32bc3f62007-08-21 20:25:39 +0000806 columnMallocFailure(pStmt);
danielk1977d0ffa1e2008-10-13 10:37:49 +0000807 return (sqlite3_value *)pOut;
drhd1e47332005-06-26 17:55:33 +0000808}
drh6c626082004-11-14 21:56:29 +0000809#ifndef SQLITE_OMIT_UTF16
drh4f26d6c2004-05-26 23:25:30 +0000810const void *sqlite3_column_text16(sqlite3_stmt *pStmt, int i){
danielk19772e588c72005-12-09 14:25:08 +0000811 const void *val = sqlite3_value_text16( columnMem(pStmt,i) );
812 columnMallocFailure(pStmt);
813 return val;
drh4f26d6c2004-05-26 23:25:30 +0000814}
drh6c626082004-11-14 21:56:29 +0000815#endif /* SQLITE_OMIT_UTF16 */
drh4f26d6c2004-05-26 23:25:30 +0000816int sqlite3_column_type(sqlite3_stmt *pStmt, int i){
drh32bc3f62007-08-21 20:25:39 +0000817 int iType = sqlite3_value_type( columnMem(pStmt,i) );
818 columnMallocFailure(pStmt);
819 return iType;
drh4f26d6c2004-05-26 23:25:30 +0000820}
821
drh29d72102006-02-09 22:13:41 +0000822/* The following function is experimental and subject to change or
823** removal */
824/*int sqlite3_column_numeric_type(sqlite3_stmt *pStmt, int i){
825** return sqlite3_value_numeric_type( columnMem(pStmt,i) );
826**}
827*/
828
drh5e2517e2004-09-24 23:59:12 +0000829/*
830** Convert the N-th element of pStmt->pColName[] into a string using
831** xFunc() then return that string. If N is out of range, return 0.
drhe590fbd2005-05-20 19:36:01 +0000832**
833** There are up to 5 names for each column. useType determines which
834** name is returned. Here are the names:
835**
836** 0 The column name as it should be displayed for output
837** 1 The datatype name for the column
838** 2 The name of the database that the column derives from
839** 3 The name of the table that the column derives from
840** 4 The name of the table column that the result column derives from
841**
842** If the result is not a simple column reference (if it is an expression
843** or a constant) then useTypes 2, 3, and 4 return NULL.
drh5e2517e2004-09-24 23:59:12 +0000844*/
845static const void *columnName(
846 sqlite3_stmt *pStmt,
847 int N,
848 const void *(*xFunc)(Mem*),
849 int useType
850){
drh32bc3f62007-08-21 20:25:39 +0000851 const void *ret = 0;
drh5e2517e2004-09-24 23:59:12 +0000852 Vdbe *p = (Vdbe *)pStmt;
drh32bc3f62007-08-21 20:25:39 +0000853 int n;
drhc6c7fd52009-04-08 23:05:28 +0000854 sqlite3 *db = p->db;
drh32bc3f62007-08-21 20:25:39 +0000855
drh9b4ea4a2009-04-10 20:32:00 +0000856 assert( db!=0 );
857 n = sqlite3_column_count(pStmt);
858 if( N<n && N>=0 ){
859 N += useType*n;
860 sqlite3_mutex_enter(db->mutex);
861 assert( db->mallocFailed==0 );
862 ret = xFunc(&p->aColName[N]);
863 /* A malloc may have failed inside of the xFunc() call. If this
864 ** is the case, clear the mallocFailed flag and return NULL.
865 */
866 if( db->mallocFailed ){
867 db->mallocFailed = 0;
868 ret = 0;
drh32bc3f62007-08-21 20:25:39 +0000869 }
drh9b4ea4a2009-04-10 20:32:00 +0000870 sqlite3_mutex_leave(db->mutex);
drh5e2517e2004-09-24 23:59:12 +0000871 }
danielk197700fd9572005-12-07 06:27:43 +0000872 return ret;
drh5e2517e2004-09-24 23:59:12 +0000873}
874
drh4f26d6c2004-05-26 23:25:30 +0000875/*
876** Return the name of the Nth column of the result set returned by SQL
877** statement pStmt.
878*/
879const char *sqlite3_column_name(sqlite3_stmt *pStmt, int N){
danielk1977955de522006-02-10 02:27:42 +0000880 return columnName(
881 pStmt, N, (const void*(*)(Mem*))sqlite3_value_text, COLNAME_NAME);
drh4f26d6c2004-05-26 23:25:30 +0000882}
drhe590fbd2005-05-20 19:36:01 +0000883#ifndef SQLITE_OMIT_UTF16
884const void *sqlite3_column_name16(sqlite3_stmt *pStmt, int N){
danielk1977955de522006-02-10 02:27:42 +0000885 return columnName(
886 pStmt, N, (const void*(*)(Mem*))sqlite3_value_text16, COLNAME_NAME);
drhe590fbd2005-05-20 19:36:01 +0000887}
888#endif
drh4f26d6c2004-05-26 23:25:30 +0000889
890/*
drh3f913572008-03-22 01:07:17 +0000891** Constraint: If you have ENABLE_COLUMN_METADATA then you must
892** not define OMIT_DECLTYPE.
893*/
894#if defined(SQLITE_OMIT_DECLTYPE) && defined(SQLITE_ENABLE_COLUMN_METADATA)
895# error "Must not define both SQLITE_OMIT_DECLTYPE \
896 and SQLITE_ENABLE_COLUMN_METADATA"
897#endif
898
899#ifndef SQLITE_OMIT_DECLTYPE
900/*
drh6c626082004-11-14 21:56:29 +0000901** Return the column declaration type (if applicable) of the 'i'th column
drhe590fbd2005-05-20 19:36:01 +0000902** of the result set of SQL statement pStmt.
drh6c626082004-11-14 21:56:29 +0000903*/
904const char *sqlite3_column_decltype(sqlite3_stmt *pStmt, int N){
danielk1977955de522006-02-10 02:27:42 +0000905 return columnName(
906 pStmt, N, (const void*(*)(Mem*))sqlite3_value_text, COLNAME_DECLTYPE);
drh6c626082004-11-14 21:56:29 +0000907}
drh6c626082004-11-14 21:56:29 +0000908#ifndef SQLITE_OMIT_UTF16
danielk197776d505b2004-05-28 13:13:02 +0000909const void *sqlite3_column_decltype16(sqlite3_stmt *pStmt, int N){
danielk1977955de522006-02-10 02:27:42 +0000910 return columnName(
911 pStmt, N, (const void*(*)(Mem*))sqlite3_value_text16, COLNAME_DECLTYPE);
drh4f26d6c2004-05-26 23:25:30 +0000912}
drh6c626082004-11-14 21:56:29 +0000913#endif /* SQLITE_OMIT_UTF16 */
drh3f913572008-03-22 01:07:17 +0000914#endif /* SQLITE_OMIT_DECLTYPE */
drh4f26d6c2004-05-26 23:25:30 +0000915
danielk19774b1ae992006-02-10 03:06:10 +0000916#ifdef SQLITE_ENABLE_COLUMN_METADATA
drhe590fbd2005-05-20 19:36:01 +0000917/*
918** Return the name of the database from which a result column derives.
919** NULL is returned if the result column is an expression or constant or
920** anything else which is not an unabiguous reference to a database column.
921*/
922const char *sqlite3_column_database_name(sqlite3_stmt *pStmt, int N){
danielk1977955de522006-02-10 02:27:42 +0000923 return columnName(
924 pStmt, N, (const void*(*)(Mem*))sqlite3_value_text, COLNAME_DATABASE);
drhe590fbd2005-05-20 19:36:01 +0000925}
926#ifndef SQLITE_OMIT_UTF16
927const void *sqlite3_column_database_name16(sqlite3_stmt *pStmt, int N){
danielk1977955de522006-02-10 02:27:42 +0000928 return columnName(
929 pStmt, N, (const void*(*)(Mem*))sqlite3_value_text16, COLNAME_DATABASE);
drhe590fbd2005-05-20 19:36:01 +0000930}
931#endif /* SQLITE_OMIT_UTF16 */
932
933/*
934** Return the name of the table from which a result column derives.
935** NULL is returned if the result column is an expression or constant or
936** anything else which is not an unabiguous reference to a database column.
937*/
938const char *sqlite3_column_table_name(sqlite3_stmt *pStmt, int N){
danielk1977955de522006-02-10 02:27:42 +0000939 return columnName(
940 pStmt, N, (const void*(*)(Mem*))sqlite3_value_text, COLNAME_TABLE);
drhe590fbd2005-05-20 19:36:01 +0000941}
942#ifndef SQLITE_OMIT_UTF16
943const void *sqlite3_column_table_name16(sqlite3_stmt *pStmt, int N){
danielk1977955de522006-02-10 02:27:42 +0000944 return columnName(
945 pStmt, N, (const void*(*)(Mem*))sqlite3_value_text16, COLNAME_TABLE);
drhe590fbd2005-05-20 19:36:01 +0000946}
947#endif /* SQLITE_OMIT_UTF16 */
948
949/*
950** Return the name of the table column from which a result column derives.
951** NULL is returned if the result column is an expression or constant or
952** anything else which is not an unabiguous reference to a database column.
953*/
954const char *sqlite3_column_origin_name(sqlite3_stmt *pStmt, int N){
danielk1977955de522006-02-10 02:27:42 +0000955 return columnName(
956 pStmt, N, (const void*(*)(Mem*))sqlite3_value_text, COLNAME_COLUMN);
drhe590fbd2005-05-20 19:36:01 +0000957}
958#ifndef SQLITE_OMIT_UTF16
959const void *sqlite3_column_origin_name16(sqlite3_stmt *pStmt, int N){
danielk1977955de522006-02-10 02:27:42 +0000960 return columnName(
961 pStmt, N, (const void*(*)(Mem*))sqlite3_value_text16, COLNAME_COLUMN);
drhe590fbd2005-05-20 19:36:01 +0000962}
963#endif /* SQLITE_OMIT_UTF16 */
danielk19774b1ae992006-02-10 03:06:10 +0000964#endif /* SQLITE_ENABLE_COLUMN_METADATA */
drhe590fbd2005-05-20 19:36:01 +0000965
966
drh4f26d6c2004-05-26 23:25:30 +0000967/******************************* sqlite3_bind_ ***************************
968**
969** Routines used to attach values to wildcards in a compiled SQL statement.
970*/
971/*
972** Unbind the value bound to variable i in virtual machine p. This is the
973** the same as binding a NULL value to the column. If the "i" parameter is
974** out of range, then SQLITE_RANGE is returned. Othewise SQLITE_OK.
975**
drh488e7b62008-10-06 12:46:43 +0000976** A successful evaluation of this routine acquires the mutex on p.
977** the mutex is released if any kind of error occurs.
978**
drh4f26d6c2004-05-26 23:25:30 +0000979** The error code stored in database p->db is overwritten with the return
980** value in any case.
981*/
982static int vdbeUnbind(Vdbe *p, int i){
983 Mem *pVar;
drh413c3d32010-02-23 20:11:56 +0000984 if( vdbeSafetyNotNull(p) ){
985 return SQLITE_MISUSE_BKPT;
986 }
drh488e7b62008-10-06 12:46:43 +0000987 sqlite3_mutex_enter(p->db->mutex);
988 if( p->magic!=VDBE_MAGIC_RUN || p->pc>=0 ){
989 sqlite3Error(p->db, SQLITE_MISUSE, 0);
990 sqlite3_mutex_leave(p->db->mutex);
drh413c3d32010-02-23 20:11:56 +0000991 sqlite3_log(SQLITE_MISUSE,
992 "bind on a busy prepared statement: [%s]", p->zSql);
993 return SQLITE_MISUSE_BKPT;
drh4f26d6c2004-05-26 23:25:30 +0000994 }
995 if( i<1 || i>p->nVar ){
996 sqlite3Error(p->db, SQLITE_RANGE, 0);
drh488e7b62008-10-06 12:46:43 +0000997 sqlite3_mutex_leave(p->db->mutex);
drh4f26d6c2004-05-26 23:25:30 +0000998 return SQLITE_RANGE;
999 }
1000 i--;
drh290c1942004-08-21 17:54:45 +00001001 pVar = &p->aVar[i];
danielk1977d8123362004-06-12 09:25:12 +00001002 sqlite3VdbeMemRelease(pVar);
drh4f26d6c2004-05-26 23:25:30 +00001003 pVar->flags = MEM_Null;
1004 sqlite3Error(p->db, SQLITE_OK, 0);
dan937d0de2009-10-15 18:35:38 +00001005
dan1d2ce4f2009-10-19 18:11:09 +00001006 /* If the bit corresponding to this variable in Vdbe.expmask is set, then
1007 ** binding a new value to this variable invalidates the current query plan.
drha7044002010-09-14 18:22:59 +00001008 **
1009 ** IMPLEMENTATION-OF: R-48440-37595 If the specific value bound to host
1010 ** parameter in the WHERE clause might influence the choice of query plan
1011 ** for a statement, then the statement will be automatically recompiled,
1012 ** as if there had been a schema change, on the first sqlite3_step() call
1013 ** following any change to the bindings of that parameter.
dan1d2ce4f2009-10-19 18:11:09 +00001014 */
drh823e09a2009-10-19 20:15:38 +00001015 if( p->isPrepareV2 &&
1016 ((i<32 && p->expmask & ((u32)1 << i)) || p->expmask==0xffffffff)
1017 ){
dan937d0de2009-10-15 18:35:38 +00001018 p->expired = 1;
1019 }
drh4f26d6c2004-05-26 23:25:30 +00001020 return SQLITE_OK;
1021}
1022
1023/*
drh5e2517e2004-09-24 23:59:12 +00001024** Bind a text or BLOB value.
drh4f26d6c2004-05-26 23:25:30 +00001025*/
drh5e2517e2004-09-24 23:59:12 +00001026static int bindText(
drh605264d2007-08-21 15:13:19 +00001027 sqlite3_stmt *pStmt, /* The statement to bind against */
1028 int i, /* Index of the parameter to bind */
1029 const void *zData, /* Pointer to the data to be bound */
1030 int nData, /* Number of bytes of data to be bound */
1031 void (*xDel)(void*), /* Destructor for the data */
drhb27b7f52008-12-10 18:03:45 +00001032 u8 encoding /* Encoding for the data */
drh4f26d6c2004-05-26 23:25:30 +00001033){
1034 Vdbe *p = (Vdbe *)pStmt;
1035 Mem *pVar;
1036 int rc;
1037
1038 rc = vdbeUnbind(p, i);
drh488e7b62008-10-06 12:46:43 +00001039 if( rc==SQLITE_OK ){
1040 if( zData!=0 ){
1041 pVar = &p->aVar[i-1];
1042 rc = sqlite3VdbeMemSetStr(pVar, zData, nData, encoding, xDel);
1043 if( rc==SQLITE_OK && encoding!=0 ){
1044 rc = sqlite3VdbeChangeEncoding(pVar, ENC(p->db));
1045 }
1046 sqlite3Error(p->db, rc, 0);
1047 rc = sqlite3ApiExit(p->db, rc);
drh27641702007-08-22 02:56:42 +00001048 }
drh488e7b62008-10-06 12:46:43 +00001049 sqlite3_mutex_leave(p->db->mutex);
drh94bb2ba2010-10-14 01:16:32 +00001050 }else if( xDel!=SQLITE_STATIC && xDel!=SQLITE_TRANSIENT ){
drh6fec9ee2010-10-12 02:13:32 +00001051 xDel((void*)zData);
drh4f26d6c2004-05-26 23:25:30 +00001052 }
drh605264d2007-08-21 15:13:19 +00001053 return rc;
drh4f26d6c2004-05-26 23:25:30 +00001054}
drh5e2517e2004-09-24 23:59:12 +00001055
1056
1057/*
1058** Bind a blob value to an SQL statement variable.
1059*/
1060int sqlite3_bind_blob(
1061 sqlite3_stmt *pStmt,
1062 int i,
1063 const void *zData,
1064 int nData,
1065 void (*xDel)(void*)
1066){
1067 return bindText(pStmt, i, zData, nData, xDel, 0);
1068}
drh4f26d6c2004-05-26 23:25:30 +00001069int sqlite3_bind_double(sqlite3_stmt *pStmt, int i, double rValue){
1070 int rc;
1071 Vdbe *p = (Vdbe *)pStmt;
drh4f26d6c2004-05-26 23:25:30 +00001072 rc = vdbeUnbind(p, i);
1073 if( rc==SQLITE_OK ){
drh290c1942004-08-21 17:54:45 +00001074 sqlite3VdbeMemSetDouble(&p->aVar[i-1], rValue);
drh488e7b62008-10-06 12:46:43 +00001075 sqlite3_mutex_leave(p->db->mutex);
drh4f26d6c2004-05-26 23:25:30 +00001076 }
danielk1977f4618892004-06-28 13:09:11 +00001077 return rc;
drh4f26d6c2004-05-26 23:25:30 +00001078}
1079int sqlite3_bind_int(sqlite3_stmt *p, int i, int iValue){
drhefad9992004-06-22 12:13:55 +00001080 return sqlite3_bind_int64(p, i, (i64)iValue);
drh4f26d6c2004-05-26 23:25:30 +00001081}
drhefad9992004-06-22 12:13:55 +00001082int sqlite3_bind_int64(sqlite3_stmt *pStmt, int i, sqlite_int64 iValue){
drh4f26d6c2004-05-26 23:25:30 +00001083 int rc;
1084 Vdbe *p = (Vdbe *)pStmt;
1085 rc = vdbeUnbind(p, i);
1086 if( rc==SQLITE_OK ){
drh290c1942004-08-21 17:54:45 +00001087 sqlite3VdbeMemSetInt64(&p->aVar[i-1], iValue);
drh488e7b62008-10-06 12:46:43 +00001088 sqlite3_mutex_leave(p->db->mutex);
drh4f26d6c2004-05-26 23:25:30 +00001089 }
1090 return rc;
1091}
drh605264d2007-08-21 15:13:19 +00001092int sqlite3_bind_null(sqlite3_stmt *pStmt, int i){
1093 int rc;
1094 Vdbe *p = (Vdbe*)pStmt;
drh605264d2007-08-21 15:13:19 +00001095 rc = vdbeUnbind(p, i);
drh488e7b62008-10-06 12:46:43 +00001096 if( rc==SQLITE_OK ){
1097 sqlite3_mutex_leave(p->db->mutex);
1098 }
drh605264d2007-08-21 15:13:19 +00001099 return rc;
drh4f26d6c2004-05-26 23:25:30 +00001100}
1101int sqlite3_bind_text(
1102 sqlite3_stmt *pStmt,
1103 int i,
1104 const char *zData,
1105 int nData,
danielk1977d8123362004-06-12 09:25:12 +00001106 void (*xDel)(void*)
drh4f26d6c2004-05-26 23:25:30 +00001107){
drh5e2517e2004-09-24 23:59:12 +00001108 return bindText(pStmt, i, zData, nData, xDel, SQLITE_UTF8);
drh4f26d6c2004-05-26 23:25:30 +00001109}
drh6c626082004-11-14 21:56:29 +00001110#ifndef SQLITE_OMIT_UTF16
drh4f26d6c2004-05-26 23:25:30 +00001111int sqlite3_bind_text16(
1112 sqlite3_stmt *pStmt,
1113 int i,
1114 const void *zData,
1115 int nData,
danielk1977d8123362004-06-12 09:25:12 +00001116 void (*xDel)(void*)
drh4f26d6c2004-05-26 23:25:30 +00001117){
drh5e2517e2004-09-24 23:59:12 +00001118 return bindText(pStmt, i, zData, nData, xDel, SQLITE_UTF16NATIVE);
drh4f26d6c2004-05-26 23:25:30 +00001119}
drh6c626082004-11-14 21:56:29 +00001120#endif /* SQLITE_OMIT_UTF16 */
danielk197726e41442006-06-14 15:16:35 +00001121int sqlite3_bind_value(sqlite3_stmt *pStmt, int i, const sqlite3_value *pValue){
1122 int rc;
drh29def562009-04-14 12:43:33 +00001123 switch( pValue->type ){
drh29def562009-04-14 12:43:33 +00001124 case SQLITE_INTEGER: {
1125 rc = sqlite3_bind_int64(pStmt, i, pValue->u.i);
1126 break;
1127 }
1128 case SQLITE_FLOAT: {
1129 rc = sqlite3_bind_double(pStmt, i, pValue->r);
1130 break;
1131 }
1132 case SQLITE_BLOB: {
1133 if( pValue->flags & MEM_Zero ){
1134 rc = sqlite3_bind_zeroblob(pStmt, i, pValue->u.nZero);
1135 }else{
1136 rc = sqlite3_bind_blob(pStmt, i, pValue->z, pValue->n,SQLITE_TRANSIENT);
1137 }
1138 break;
1139 }
1140 case SQLITE_TEXT: {
drhac80db72009-04-14 12:58:20 +00001141 rc = bindText(pStmt,i, pValue->z, pValue->n, SQLITE_TRANSIENT,
1142 pValue->enc);
1143 break;
1144 }
1145 default: {
1146 rc = sqlite3_bind_null(pStmt, i);
drh29def562009-04-14 12:43:33 +00001147 break;
1148 }
danielk197726e41442006-06-14 15:16:35 +00001149 }
1150 return rc;
1151}
drhb026e052007-05-02 01:34:31 +00001152int sqlite3_bind_zeroblob(sqlite3_stmt *pStmt, int i, int n){
1153 int rc;
1154 Vdbe *p = (Vdbe *)pStmt;
1155 rc = vdbeUnbind(p, i);
1156 if( rc==SQLITE_OK ){
1157 sqlite3VdbeMemSetZeroBlob(&p->aVar[i-1], n);
drh488e7b62008-10-06 12:46:43 +00001158 sqlite3_mutex_leave(p->db->mutex);
drhb026e052007-05-02 01:34:31 +00001159 }
1160 return rc;
1161}
drh75f6a032004-07-15 14:15:00 +00001162
1163/*
1164** Return the number of wildcards that can be potentially bound to.
1165** This routine is added to support DBD::SQLite.
drh75f6a032004-07-15 14:15:00 +00001166*/
1167int sqlite3_bind_parameter_count(sqlite3_stmt *pStmt){
drh92febd92004-08-20 18:34:20 +00001168 Vdbe *p = (Vdbe*)pStmt;
1169 return p ? p->nVar : 0;
drh75f6a032004-07-15 14:15:00 +00001170}
drh895d7472004-08-20 16:02:39 +00001171
1172/*
drhfa6bc002004-09-07 16:19:52 +00001173** Create a mapping from variable numbers to variable names
1174** in the Vdbe.azVar[] array, if such a mapping does not already
1175** exist.
drh895d7472004-08-20 16:02:39 +00001176*/
drhfa6bc002004-09-07 16:19:52 +00001177static void createVarMap(Vdbe *p){
drh895d7472004-08-20 16:02:39 +00001178 if( !p->okVar ){
drh0f5ea0b2009-04-09 14:02:44 +00001179 int j;
1180 Op *pOp;
drh605264d2007-08-21 15:13:19 +00001181 sqlite3_mutex_enter(p->db->mutex);
drh0f5ea0b2009-04-09 14:02:44 +00001182 /* The race condition here is harmless. If two threads call this
1183 ** routine on the same Vdbe at the same time, they both might end
1184 ** up initializing the Vdbe.azVar[] array. That is a little extra
1185 ** work but it results in the same answer.
1186 */
1187 for(j=0, pOp=p->aOp; j<p->nOp; j++, pOp++){
1188 if( pOp->opcode==OP_Variable ){
1189 assert( pOp->p1>0 && pOp->p1<=p->nVar );
1190 p->azVar[pOp->p1-1] = pOp->p4.z;
drh895d7472004-08-20 16:02:39 +00001191 }
1192 }
drh0f5ea0b2009-04-09 14:02:44 +00001193 p->okVar = 1;
drh605264d2007-08-21 15:13:19 +00001194 sqlite3_mutex_leave(p->db->mutex);
drh895d7472004-08-20 16:02:39 +00001195 }
drhfa6bc002004-09-07 16:19:52 +00001196}
1197
1198/*
1199** Return the name of a wildcard parameter. Return NULL if the index
1200** is out of range or if the wildcard is unnamed.
1201**
1202** The result is always UTF-8.
1203*/
1204const char *sqlite3_bind_parameter_name(sqlite3_stmt *pStmt, int i){
1205 Vdbe *p = (Vdbe*)pStmt;
1206 if( p==0 || i<1 || i>p->nVar ){
1207 return 0;
1208 }
1209 createVarMap(p);
drh895d7472004-08-20 16:02:39 +00001210 return p->azVar[i-1];
1211}
drhfa6bc002004-09-07 16:19:52 +00001212
1213/*
1214** Given a wildcard parameter name, return the index of the variable
1215** with that name. If there is no variable with the given name,
1216** return 0.
1217*/
drh5f18a222009-11-26 14:01:53 +00001218int sqlite3VdbeParameterIndex(Vdbe *p, const char *zName, int nName){
drhfa6bc002004-09-07 16:19:52 +00001219 int i;
1220 if( p==0 ){
1221 return 0;
1222 }
1223 createVarMap(p);
drh6a8903c2004-12-10 18:00:04 +00001224 if( zName ){
1225 for(i=0; i<p->nVar; i++){
1226 const char *z = p->azVar[i];
drh5f18a222009-11-26 14:01:53 +00001227 if( z && memcmp(z,zName,nName)==0 && z[nName]==0 ){
drh6a8903c2004-12-10 18:00:04 +00001228 return i+1;
1229 }
drhfa6bc002004-09-07 16:19:52 +00001230 }
1231 }
1232 return 0;
1233}
drh5f18a222009-11-26 14:01:53 +00001234int sqlite3_bind_parameter_index(sqlite3_stmt *pStmt, const char *zName){
1235 return sqlite3VdbeParameterIndex((Vdbe*)pStmt, zName, sqlite3Strlen30(zName));
1236}
drhf8db1bc2005-04-22 02:38:37 +00001237
drh51942bc2005-06-12 22:01:42 +00001238/*
drhf8db1bc2005-04-22 02:38:37 +00001239** Transfer all bindings from the first statement over to the second.
drhf8db1bc2005-04-22 02:38:37 +00001240*/
shane145834a2008-09-04 12:03:42 +00001241int sqlite3TransferBindings(sqlite3_stmt *pFromStmt, sqlite3_stmt *pToStmt){
drhf8db1bc2005-04-22 02:38:37 +00001242 Vdbe *pFrom = (Vdbe*)pFromStmt;
1243 Vdbe *pTo = (Vdbe*)pToStmt;
drh0f5ea0b2009-04-09 14:02:44 +00001244 int i;
1245 assert( pTo->db==pFrom->db );
1246 assert( pTo->nVar==pFrom->nVar );
drh27641702007-08-22 02:56:42 +00001247 sqlite3_mutex_enter(pTo->db->mutex);
drh0f5ea0b2009-04-09 14:02:44 +00001248 for(i=0; i<pFrom->nVar; i++){
drh643167f2008-01-22 21:30:53 +00001249 sqlite3VdbeMemMove(&pTo->aVar[i], &pFrom->aVar[i]);
drhf8db1bc2005-04-22 02:38:37 +00001250 }
drh27641702007-08-22 02:56:42 +00001251 sqlite3_mutex_leave(pTo->db->mutex);
drh0f5ea0b2009-04-09 14:02:44 +00001252 return SQLITE_OK;
drhf8db1bc2005-04-22 02:38:37 +00001253}
drh51942bc2005-06-12 22:01:42 +00001254
shaneeec556d2008-10-12 00:27:53 +00001255#ifndef SQLITE_OMIT_DEPRECATED
drh51942bc2005-06-12 22:01:42 +00001256/*
shane145834a2008-09-04 12:03:42 +00001257** Deprecated external interface. Internal/core SQLite code
1258** should call sqlite3TransferBindings.
drh0f5ea0b2009-04-09 14:02:44 +00001259**
1260** Is is misuse to call this routine with statements from different
1261** database connections. But as this is a deprecated interface, we
1262** will not bother to check for that condition.
1263**
1264** If the two statements contain a different number of bindings, then
1265** an SQLITE_ERROR is returned. Nothing else can go wrong, so otherwise
1266** SQLITE_OK is returned.
shane145834a2008-09-04 12:03:42 +00001267*/
1268int sqlite3_transfer_bindings(sqlite3_stmt *pFromStmt, sqlite3_stmt *pToStmt){
drh0f5ea0b2009-04-09 14:02:44 +00001269 Vdbe *pFrom = (Vdbe*)pFromStmt;
1270 Vdbe *pTo = (Vdbe*)pToStmt;
1271 if( pFrom->nVar!=pTo->nVar ){
1272 return SQLITE_ERROR;
1273 }
danc94b8592009-10-20 07:01:24 +00001274 if( pTo->isPrepareV2 && pTo->expmask ){
1275 pTo->expired = 1;
1276 }
1277 if( pFrom->isPrepareV2 && pFrom->expmask ){
1278 pFrom->expired = 1;
1279 }
shane145834a2008-09-04 12:03:42 +00001280 return sqlite3TransferBindings(pFromStmt, pToStmt);
1281}
shaneeec556d2008-10-12 00:27:53 +00001282#endif
shane145834a2008-09-04 12:03:42 +00001283
1284/*
drh51942bc2005-06-12 22:01:42 +00001285** Return the sqlite3* database handle to which the prepared statement given
1286** in the argument belongs. This is the same database handle that was
1287** the first argument to the sqlite3_prepare() that was used to create
1288** the statement in the first place.
1289*/
1290sqlite3 *sqlite3_db_handle(sqlite3_stmt *pStmt){
1291 return pStmt ? ((Vdbe*)pStmt)->db : 0;
1292}
drhbb5a9c32008-06-19 02:52:25 +00001293
1294/*
drhf03d9cc2010-11-16 23:10:25 +00001295** Return true if the prepared statement is guaranteed to not modify the
1296** database.
1297*/
1298int sqlite3_stmt_readonly(sqlite3_stmt *pStmt){
1299 return pStmt ? ((Vdbe*)pStmt)->readOnly : 1;
1300}
1301
1302/*
drhbb5a9c32008-06-19 02:52:25 +00001303** Return a pointer to the next prepared statement after pStmt associated
1304** with database connection pDb. If pStmt is NULL, return the first
1305** prepared statement for the database connection. Return NULL if there
1306** are no more.
1307*/
1308sqlite3_stmt *sqlite3_next_stmt(sqlite3 *pDb, sqlite3_stmt *pStmt){
1309 sqlite3_stmt *pNext;
1310 sqlite3_mutex_enter(pDb->mutex);
1311 if( pStmt==0 ){
1312 pNext = (sqlite3_stmt*)pDb->pVdbe;
1313 }else{
1314 pNext = (sqlite3_stmt*)((Vdbe*)pStmt)->pNext;
1315 }
1316 sqlite3_mutex_leave(pDb->mutex);
1317 return pNext;
1318}
drhd1d38482008-10-07 23:46:38 +00001319
1320/*
1321** Return the value of a status counter for a prepared statement
1322*/
1323int sqlite3_stmt_status(sqlite3_stmt *pStmt, int op, int resetFlag){
1324 Vdbe *pVdbe = (Vdbe*)pStmt;
1325 int v = pVdbe->aCounter[op-1];
1326 if( resetFlag ) pVdbe->aCounter[op-1] = 0;
1327 return v;
1328}
dan46c47d42011-03-01 18:42:07 +00001329
dan37db03b2011-03-16 19:59:18 +00001330/*
1331** This function is called from within a pre-update callback to retrieve
1332** a field of the row currently being updated or deleted.
1333*/
dan46c47d42011-03-01 18:42:07 +00001334int sqlite3_preupdate_old(sqlite3 *db, int iIdx, sqlite3_value **ppValue){
1335 PreUpdate *p = db->pPreUpdate;
1336 int rc = SQLITE_OK;
1337
dan37db03b2011-03-16 19:59:18 +00001338 /* Test that this call is being made from within an SQLITE_DELETE or
1339 ** SQLITE_UPDATE pre-update callback, and that iIdx is within range. */
dan46c47d42011-03-01 18:42:07 +00001340 if( !p || p->op==SQLITE_INSERT ){
1341 rc = SQLITE_MISUSE_BKPT;
1342 goto preupdate_old_out;
1343 }
1344 if( iIdx>=p->pCsr->nField || iIdx<0 ){
1345 rc = SQLITE_RANGE;
1346 goto preupdate_old_out;
1347 }
1348
dan37db03b2011-03-16 19:59:18 +00001349 /* If the old.* record has not yet been loaded into memory, do so now. */
dan46c47d42011-03-01 18:42:07 +00001350 if( p->pUnpacked==0 ){
dan12ca0b52011-03-21 16:17:42 +00001351 u32 nRec;
1352 u8 *aRec;
dan46c47d42011-03-01 18:42:07 +00001353
dan12ca0b52011-03-21 16:17:42 +00001354 rc = sqlite3BtreeDataSize(p->pCsr->pCursor, &nRec);
dan46c47d42011-03-01 18:42:07 +00001355 if( rc!=SQLITE_OK ) goto preupdate_old_out;
dan12ca0b52011-03-21 16:17:42 +00001356 aRec = sqlite3DbMallocRaw(db, nRec);
1357 if( !aRec ) goto preupdate_old_out;
1358 rc = sqlite3BtreeData(p->pCsr->pCursor, 0, nRec, aRec);
1359 if( rc==SQLITE_OK ){
1360 p->pUnpacked = sqlite3VdbeRecordUnpack(&p->keyinfo, nRec, aRec, 0, 0);
1361 if( !p->pUnpacked ) rc = SQLITE_NOMEM;
1362 }
dan46c47d42011-03-01 18:42:07 +00001363 if( rc!=SQLITE_OK ){
dan12ca0b52011-03-21 16:17:42 +00001364 sqlite3DbFree(db, aRec);
dan46c47d42011-03-01 18:42:07 +00001365 goto preupdate_old_out;
1366 }
dan12ca0b52011-03-21 16:17:42 +00001367 p->aRecord = aRec;
dan46c47d42011-03-01 18:42:07 +00001368 }
1369
1370 if( iIdx>=p->pUnpacked->nField ){
1371 *ppValue = (sqlite3_value *)columnNullValue();
1372 }else{
1373 *ppValue = &p->pUnpacked->aMem[iIdx];
dan319eeb72011-03-19 08:38:50 +00001374 if( iIdx==p->iPKey ){
1375 sqlite3VdbeMemSetInt64(*ppValue, p->iKey1);
1376 }
dan46c47d42011-03-01 18:42:07 +00001377 sqlite3VdbeMemStoreType(*ppValue);
1378 }
1379
1380 preupdate_old_out:
1381 sqlite3Error(db, rc, 0);
1382 return sqlite3ApiExit(db, rc);
1383}
1384
dan37db03b2011-03-16 19:59:18 +00001385/*
1386** This function is called from within a pre-update callback to retrieve
1387** the number of columns in the row being updated, deleted or inserted.
1388*/
dan46c47d42011-03-01 18:42:07 +00001389int sqlite3_preupdate_count(sqlite3 *db){
1390 PreUpdate *p = db->pPreUpdate;
1391 return (p ? p->pCsr->nField : 0);
1392}
1393
dan37db03b2011-03-16 19:59:18 +00001394/*
dan1e7a2d42011-03-22 18:45:29 +00001395** This function is designed to be called from within a pre-update callback
1396** only. It returns zero if the change that caused the callback was made
1397** immediately by a user SQL statement. Or, if the change was made by a
1398** trigger program, it returns the number of trigger programs currently
1399** on the stack (1 for a top-level trigger, 2 for a trigger fired by a
1400** top-level trigger etc.).
1401**
1402** For the purposes of the previous paragraph, a foreign key CASCADE, SET NULL
1403** or SET DEFAULT action is considered a trigger.
1404*/
1405int sqlite3_preupdate_depth(sqlite3 *db){
1406 PreUpdate *p = db->pPreUpdate;
1407 return (p ? p->v->nFrame : 0);
1408}
1409
1410/*
dan37db03b2011-03-16 19:59:18 +00001411** This function is called from within a pre-update callback to retrieve
1412** a field of the row currently being updated or inserted.
1413*/
1414int sqlite3_preupdate_new(sqlite3 *db, int iIdx, sqlite3_value **ppValue){
dan46c47d42011-03-01 18:42:07 +00001415 PreUpdate *p = db->pPreUpdate;
1416 int rc = SQLITE_OK;
dan37db03b2011-03-16 19:59:18 +00001417 Mem *pMem;
dan46c47d42011-03-01 18:42:07 +00001418
dan37db03b2011-03-16 19:59:18 +00001419 if( !p || p->op==SQLITE_DELETE ){
dan46c47d42011-03-01 18:42:07 +00001420 rc = SQLITE_MISUSE_BKPT;
dan37db03b2011-03-16 19:59:18 +00001421 goto preupdate_new_out;
dan46c47d42011-03-01 18:42:07 +00001422 }
1423 if( iIdx>=p->pCsr->nField || iIdx<0 ){
1424 rc = SQLITE_RANGE;
dan37db03b2011-03-16 19:59:18 +00001425 goto preupdate_new_out;
dan46c47d42011-03-01 18:42:07 +00001426 }
dan46c47d42011-03-01 18:42:07 +00001427
dan37db03b2011-03-16 19:59:18 +00001428 if( p->op==SQLITE_INSERT ){
1429 /* For an INSERT, memory cell p->iNewReg contains the serialized record
1430 ** that is being inserted. Deserialize it. */
1431 UnpackedRecord *pUnpack = p->pNewUnpacked;
1432 if( !pUnpack ){
1433 Mem *pData = &p->v->aMem[p->iNewReg];
1434 rc = sqlite3VdbeMemExpandBlob(pData);
1435 if( rc!=SQLITE_OK ) goto preupdate_new_out;
1436 pUnpack = sqlite3VdbeRecordUnpack(&p->keyinfo, pData->n, pData->z, 0, 0);
1437 if( !pUnpack ){
1438 rc = SQLITE_NOMEM;
1439 goto preupdate_new_out;
1440 }
1441 p->pNewUnpacked = pUnpack;
1442 }
1443 if( iIdx>=pUnpack->nField ){
1444 pMem = (sqlite3_value *)columnNullValue();
1445 }else{
1446 pMem = &pUnpack->aMem[iIdx];
dan319eeb72011-03-19 08:38:50 +00001447 if( iIdx==p->iPKey ){
1448 sqlite3VdbeMemSetInt64(pMem, p->iKey2);
1449 }
dan37db03b2011-03-16 19:59:18 +00001450 sqlite3VdbeMemStoreType(pMem);
1451 }
1452 }else{
1453 /* For an UPDATE, memory cell (p->iNewReg+1+iIdx) contains the required
1454 ** value. Make a copy of the cell contents and return a pointer to it.
1455 ** It is not safe to return a pointer to the memory cell itself as the
1456 ** caller may modify the value text encoding.
1457 */
1458 assert( p->op==SQLITE_UPDATE );
1459 if( !p->aNew ){
1460 p->aNew = (Mem *)sqlite3DbMallocZero(db, sizeof(Mem) * p->pCsr->nField);
1461 if( !p->aNew ){
1462 rc = SQLITE_NOMEM;
1463 goto preupdate_new_out;
1464 }
1465 }
drh304637c2011-03-18 16:47:27 +00001466 assert( iIdx>=0 && iIdx<p->pCsr->nField );
dan37db03b2011-03-16 19:59:18 +00001467 pMem = &p->aNew[iIdx];
1468 if( pMem->flags==0 ){
dan319eeb72011-03-19 08:38:50 +00001469 if( iIdx==p->iPKey ){
1470 sqlite3VdbeMemSetInt64(pMem, p->iKey2);
1471 }else{
1472 rc = sqlite3VdbeMemCopy(pMem, &p->v->aMem[p->iNewReg+1+iIdx]);
1473 if( rc!=SQLITE_OK ) goto preupdate_new_out;
1474 }
dan37db03b2011-03-16 19:59:18 +00001475 sqlite3VdbeMemStoreType(pMem);
1476 }
1477 }
1478 *ppValue = pMem;
1479
1480 preupdate_new_out:
dan46c47d42011-03-01 18:42:07 +00001481 sqlite3Error(db, rc, 0);
1482 return sqlite3ApiExit(db, rc);
1483}