blob: 88ec9dae6269e0915eb2006e27e69d35bcdebd72 [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
danielk1977dfb316d2008-03-26 18:34:43 +000019#ifdef SQLITE_ENABLE_MEMORY_MANAGEMENT
20/*
21** The following structure contains pointers to the end points of a
22** doubly-linked list of all compiled SQL statements that may be holding
23** buffers eligible for release when the sqlite3_release_memory() interface is
24** invoked. Access to this list is protected by the SQLITE_MUTEX_STATIC_LRU2
25** mutex.
26**
27** Statements are added to the end of this list when sqlite3_reset() is
28** called. They are removed either when sqlite3_step() or sqlite3_finalize()
29** is called. When statements are added to this list, the associated
30** register array (p->aMem[1..p->nMem]) may contain dynamic buffers that
31** can be freed using sqlite3VdbeReleaseMemory().
32**
33** When statements are added or removed from this list, the mutex
34** associated with the Vdbe being added or removed (Vdbe.db->mutex) is
35** already held. The LRU2 mutex is then obtained, blocking if necessary,
36** the linked-list pointers manipulated and the LRU2 mutex relinquished.
37*/
38struct StatementLruList {
39 Vdbe *pFirst;
40 Vdbe *pLast;
41};
42static struct StatementLruList sqlite3LruStatements;
43
44/*
45** Check that the list looks to be internally consistent. This is used
46** as part of an assert() statement as follows:
47**
48** assert( stmtLruCheck() );
49*/
drh26e4a8b2008-05-01 17:16:52 +000050#ifndef NDEBUG
danielk1977dfb316d2008-03-26 18:34:43 +000051static int stmtLruCheck(){
52 Vdbe *p;
53 for(p=sqlite3LruStatements.pFirst; p; p=p->pLruNext){
54 assert(p->pLruNext || p==sqlite3LruStatements.pLast);
55 assert(!p->pLruNext || p->pLruNext->pLruPrev==p);
56 assert(p->pLruPrev || p==sqlite3LruStatements.pFirst);
57 assert(!p->pLruPrev || p->pLruPrev->pLruNext==p);
58 }
59 return 1;
60}
drh26e4a8b2008-05-01 17:16:52 +000061#endif
danielk1977dfb316d2008-03-26 18:34:43 +000062
63/*
64** Add vdbe p to the end of the statement lru list. It is assumed that
65** p is not already part of the list when this is called. The lru list
66** is protected by the SQLITE_MUTEX_STATIC_LRU mutex.
67*/
68static void stmtLruAdd(Vdbe *p){
69 sqlite3_mutex_enter(sqlite3_mutex_alloc(SQLITE_MUTEX_STATIC_LRU2));
70
71 if( p->pLruPrev || p->pLruNext || sqlite3LruStatements.pFirst==p ){
72 sqlite3_mutex_leave(sqlite3_mutex_alloc(SQLITE_MUTEX_STATIC_LRU2));
73 return;
74 }
75
76 assert( stmtLruCheck() );
77
78 if( !sqlite3LruStatements.pFirst ){
79 assert( !sqlite3LruStatements.pLast );
80 sqlite3LruStatements.pFirst = p;
81 sqlite3LruStatements.pLast = p;
82 }else{
83 assert( !sqlite3LruStatements.pLast->pLruNext );
84 p->pLruPrev = sqlite3LruStatements.pLast;
85 sqlite3LruStatements.pLast->pLruNext = p;
86 sqlite3LruStatements.pLast = p;
87 }
88
89 assert( stmtLruCheck() );
90
91 sqlite3_mutex_leave(sqlite3_mutex_alloc(SQLITE_MUTEX_STATIC_LRU2));
92}
93
94/*
95** Assuming the SQLITE_MUTEX_STATIC_LRU2 mutext is already held, remove
96** statement p from the least-recently-used statement list. If the
97** statement is not currently part of the list, this call is a no-op.
98*/
99static void stmtLruRemoveNomutex(Vdbe *p){
100 if( p->pLruPrev || p->pLruNext || p==sqlite3LruStatements.pFirst ){
101 assert( stmtLruCheck() );
102 if( p->pLruNext ){
103 p->pLruNext->pLruPrev = p->pLruPrev;
104 }else{
105 sqlite3LruStatements.pLast = p->pLruPrev;
106 }
107 if( p->pLruPrev ){
108 p->pLruPrev->pLruNext = p->pLruNext;
109 }else{
110 sqlite3LruStatements.pFirst = p->pLruNext;
111 }
112 p->pLruNext = 0;
113 p->pLruPrev = 0;
114 assert( stmtLruCheck() );
115 }
116}
117
118/*
119** Assuming the SQLITE_MUTEX_STATIC_LRU2 mutext is not held, remove
120** statement p from the least-recently-used statement list. If the
121** statement is not currently part of the list, this call is a no-op.
122*/
123static void stmtLruRemove(Vdbe *p){
124 sqlite3_mutex_enter(sqlite3_mutex_alloc(SQLITE_MUTEX_STATIC_LRU2));
125 stmtLruRemoveNomutex(p);
126 sqlite3_mutex_leave(sqlite3_mutex_alloc(SQLITE_MUTEX_STATIC_LRU2));
127}
128
129/*
130** Try to release n bytes of memory by freeing buffers associated
131** with the memory registers of currently unused vdbes.
132*/
133int sqlite3VdbeReleaseMemory(int n){
134 Vdbe *p;
135 Vdbe *pNext;
136 int nFree = 0;
137
138 sqlite3_mutex_enter(sqlite3_mutex_alloc(SQLITE_MUTEX_STATIC_LRU2));
139 for(p=sqlite3LruStatements.pFirst; p && nFree<n; p=pNext){
140 pNext = p->pLruNext;
141
142 /* For each statement handle in the lru list, attempt to obtain the
143 ** associated database mutex. If it cannot be obtained, continue
144 ** to the next statement handle. It is not possible to block on
145 ** the database mutex - that could cause deadlock.
146 */
147 if( SQLITE_OK==sqlite3_mutex_try(p->db->mutex) ){
148 nFree += sqlite3VdbeReleaseBuffers(p);
149 stmtLruRemoveNomutex(p);
150 sqlite3_mutex_leave(p->db->mutex);
151 }
152 }
153 sqlite3_mutex_leave(sqlite3_mutex_alloc(SQLITE_MUTEX_STATIC_LRU2));
154
155 return nFree;
156}
157
158/*
159** Call sqlite3Reprepare() on the statement. Remove it from the
160** lru list before doing so, as Reprepare() will free all the
161** memory register buffers anyway.
162*/
163int vdbeReprepare(Vdbe *p){
164 stmtLruRemove(p);
165 return sqlite3Reprepare(p);
166}
167
168#else /* !SQLITE_ENABLE_MEMORY_MANAGEMENT */
169 #define stmtLruRemove(x)
170 #define stmtLruAdd(x)
171 #define vdbeReprepare(x) sqlite3Reprepare(x)
172#endif
173
174
drhd89bd002005-01-22 03:03:54 +0000175/*
176** Return TRUE (non-zero) of the statement supplied as an argument needs
177** to be recompiled. A statement needs to be recompiled whenever the
178** execution environment changes in a way that would alter the program
179** that sqlite3_prepare() generates. For example, if new functions or
180** collating sequences are registered or if an authorizer function is
181** added or changed.
drhd89bd002005-01-22 03:03:54 +0000182*/
183int sqlite3_expired(sqlite3_stmt *pStmt){
184 Vdbe *p = (Vdbe*)pStmt;
185 return p==0 || p->expired;
186}
187
drhe30f4422007-08-21 16:15:55 +0000188/*
189** The following routine destroys a virtual machine that is created by
190** the sqlite3_compile() routine. The integer returned is an SQLITE_
191** success/failure code that describes the result of executing the virtual
192** machine.
193**
194** This routine sets the error code and string returned by
195** sqlite3_errcode(), sqlite3_errmsg() and sqlite3_errmsg16().
196*/
197int sqlite3_finalize(sqlite3_stmt *pStmt){
198 int rc;
199 if( pStmt==0 ){
200 rc = SQLITE_OK;
201 }else{
202 Vdbe *v = (Vdbe*)pStmt;
drh7e8b8482008-01-23 03:03:05 +0000203#ifndef SQLITE_MUTEX_NOOP
drh86f8c192007-08-22 00:39:19 +0000204 sqlite3_mutex *mutex = v->db->mutex;
drh7e8b8482008-01-23 03:03:05 +0000205#endif
drh86f8c192007-08-22 00:39:19 +0000206 sqlite3_mutex_enter(mutex);
danielk1977dfb316d2008-03-26 18:34:43 +0000207 stmtLruRemove(v);
drhe30f4422007-08-21 16:15:55 +0000208 rc = sqlite3VdbeFinalize(v);
drh86f8c192007-08-22 00:39:19 +0000209 sqlite3_mutex_leave(mutex);
drhe30f4422007-08-21 16:15:55 +0000210 }
211 return rc;
212}
213
214/*
215** Terminate the current execution of an SQL statement and reset it
216** back to its starting state so that it can be reused. A success code from
217** the prior execution is returned.
218**
219** This routine sets the error code and string returned by
220** sqlite3_errcode(), sqlite3_errmsg() and sqlite3_errmsg16().
221*/
222int sqlite3_reset(sqlite3_stmt *pStmt){
223 int rc;
224 if( pStmt==0 ){
225 rc = SQLITE_OK;
226 }else{
227 Vdbe *v = (Vdbe*)pStmt;
228 sqlite3_mutex_enter(v->db->mutex);
danielk197765710b12008-04-14 15:15:22 +0000229 rc = sqlite3VdbeReset(v, 1);
danielk1977dfb316d2008-03-26 18:34:43 +0000230 stmtLruAdd(v);
drhe30f4422007-08-21 16:15:55 +0000231 sqlite3VdbeMakeReady(v, -1, 0, 0, 0);
232 assert( (rc & (v->db->errMask))==rc );
233 sqlite3_mutex_leave(v->db->mutex);
234 }
235 return rc;
236}
237
238/*
239** Set all the parameters in the compiled SQL statement to NULL.
240*/
241int sqlite3_clear_bindings(sqlite3_stmt *pStmt){
242 int i;
243 int rc = SQLITE_OK;
drh7297d1f2008-05-09 14:39:44 +0000244 Vdbe *p = (Vdbe*)pStmt;
drh7e8b8482008-01-23 03:03:05 +0000245#ifndef SQLITE_MUTEX_NOOP
246 sqlite3_mutex *mutex = ((Vdbe*)pStmt)->db->mutex;
247#endif
248 sqlite3_mutex_enter(mutex);
drh7297d1f2008-05-09 14:39:44 +0000249 for(i=0; i<p->nVar; i++){
250 sqlite3VdbeMemRelease(&p->aVar[i]);
251 p->aVar[i].flags = MEM_Null;
drhe30f4422007-08-21 16:15:55 +0000252 }
drh7e8b8482008-01-23 03:03:05 +0000253 sqlite3_mutex_leave(mutex);
drhe30f4422007-08-21 16:15:55 +0000254 return rc;
255}
256
257
drh4f26d6c2004-05-26 23:25:30 +0000258/**************************** sqlite3_value_ *******************************
259** The following routines extract information from a Mem or sqlite3_value
260** structure.
261*/
262const void *sqlite3_value_blob(sqlite3_value *pVal){
263 Mem *p = (Mem*)pVal;
264 if( p->flags & (MEM_Blob|MEM_Str) ){
drhb21c8cd2007-08-21 19:33:56 +0000265 sqlite3VdbeMemExpandBlob(p);
drh1f0feef2007-05-15 13:27:07 +0000266 p->flags &= ~MEM_Str;
267 p->flags |= MEM_Blob;
drh4f26d6c2004-05-26 23:25:30 +0000268 return p->z;
269 }else{
270 return sqlite3_value_text(pVal);
271 }
272}
273int sqlite3_value_bytes(sqlite3_value *pVal){
drhb21c8cd2007-08-21 19:33:56 +0000274 return sqlite3ValueBytes(pVal, SQLITE_UTF8);
drh4f26d6c2004-05-26 23:25:30 +0000275}
276int sqlite3_value_bytes16(sqlite3_value *pVal){
drhb21c8cd2007-08-21 19:33:56 +0000277 return sqlite3ValueBytes(pVal, SQLITE_UTF16NATIVE);
drh4f26d6c2004-05-26 23:25:30 +0000278}
279double sqlite3_value_double(sqlite3_value *pVal){
drh6a6124e2004-06-27 01:56:33 +0000280 return sqlite3VdbeRealValue((Mem*)pVal);
drh4f26d6c2004-05-26 23:25:30 +0000281}
282int sqlite3_value_int(sqlite3_value *pVal){
drh6a6124e2004-06-27 01:56:33 +0000283 return sqlite3VdbeIntValue((Mem*)pVal);
drh4f26d6c2004-05-26 23:25:30 +0000284}
drhefad9992004-06-22 12:13:55 +0000285sqlite_int64 sqlite3_value_int64(sqlite3_value *pVal){
drh6a6124e2004-06-27 01:56:33 +0000286 return sqlite3VdbeIntValue((Mem*)pVal);
drh4f26d6c2004-05-26 23:25:30 +0000287}
288const unsigned char *sqlite3_value_text(sqlite3_value *pVal){
drhb21c8cd2007-08-21 19:33:56 +0000289 return (const unsigned char *)sqlite3ValueText(pVal, SQLITE_UTF8);
drh4f26d6c2004-05-26 23:25:30 +0000290}
drh6c626082004-11-14 21:56:29 +0000291#ifndef SQLITE_OMIT_UTF16
drh4f26d6c2004-05-26 23:25:30 +0000292const void *sqlite3_value_text16(sqlite3_value* pVal){
drhb21c8cd2007-08-21 19:33:56 +0000293 return sqlite3ValueText(pVal, SQLITE_UTF16NATIVE);
drh4f26d6c2004-05-26 23:25:30 +0000294}
danielk1977d8123362004-06-12 09:25:12 +0000295const void *sqlite3_value_text16be(sqlite3_value *pVal){
drhb21c8cd2007-08-21 19:33:56 +0000296 return sqlite3ValueText(pVal, SQLITE_UTF16BE);
danielk1977d8123362004-06-12 09:25:12 +0000297}
298const void *sqlite3_value_text16le(sqlite3_value *pVal){
drhb21c8cd2007-08-21 19:33:56 +0000299 return sqlite3ValueText(pVal, SQLITE_UTF16LE);
danielk1977d8123362004-06-12 09:25:12 +0000300}
drh6c626082004-11-14 21:56:29 +0000301#endif /* SQLITE_OMIT_UTF16 */
drh4f26d6c2004-05-26 23:25:30 +0000302int sqlite3_value_type(sqlite3_value* pVal){
drhf4479502004-05-27 03:12:53 +0000303 return pVal->type;
drh4f26d6c2004-05-26 23:25:30 +0000304}
305
306/**************************** sqlite3_result_ *******************************
307** The following routines are used by user-defined functions to specify
308** the function result.
309*/
310void sqlite3_result_blob(
311 sqlite3_context *pCtx,
312 const void *z,
313 int n,
danielk1977d8123362004-06-12 09:25:12 +0000314 void (*xDel)(void *)
drh4f26d6c2004-05-26 23:25:30 +0000315){
drh5708d2d2005-06-22 10:53:59 +0000316 assert( n>=0 );
drh32bc3f62007-08-21 20:25:39 +0000317 assert( sqlite3_mutex_held(pCtx->s.db->mutex) );
drhb21c8cd2007-08-21 19:33:56 +0000318 sqlite3VdbeMemSetStr(&pCtx->s, z, n, 0, xDel);
drh4f26d6c2004-05-26 23:25:30 +0000319}
320void sqlite3_result_double(sqlite3_context *pCtx, double rVal){
drh32bc3f62007-08-21 20:25:39 +0000321 assert( sqlite3_mutex_held(pCtx->s.db->mutex) );
drh4f26d6c2004-05-26 23:25:30 +0000322 sqlite3VdbeMemSetDouble(&pCtx->s, rVal);
323}
324void sqlite3_result_error(sqlite3_context *pCtx, const char *z, int n){
drh32bc3f62007-08-21 20:25:39 +0000325 assert( sqlite3_mutex_held(pCtx->s.db->mutex) );
drh69544ec2008-02-06 14:11:34 +0000326 pCtx->isError = SQLITE_ERROR;
drhb21c8cd2007-08-21 19:33:56 +0000327 sqlite3VdbeMemSetStr(&pCtx->s, z, n, SQLITE_UTF8, SQLITE_TRANSIENT);
drh4f26d6c2004-05-26 23:25:30 +0000328}
danielk1977a1686c92006-01-23 07:52:37 +0000329#ifndef SQLITE_OMIT_UTF16
drh4f26d6c2004-05-26 23:25:30 +0000330void sqlite3_result_error16(sqlite3_context *pCtx, const void *z, int n){
drh32bc3f62007-08-21 20:25:39 +0000331 assert( sqlite3_mutex_held(pCtx->s.db->mutex) );
drh69544ec2008-02-06 14:11:34 +0000332 pCtx->isError = SQLITE_ERROR;
drhb21c8cd2007-08-21 19:33:56 +0000333 sqlite3VdbeMemSetStr(&pCtx->s, z, n, SQLITE_UTF16NATIVE, SQLITE_TRANSIENT);
drh4f26d6c2004-05-26 23:25:30 +0000334}
danielk1977a1686c92006-01-23 07:52:37 +0000335#endif
drhf4479502004-05-27 03:12:53 +0000336void sqlite3_result_int(sqlite3_context *pCtx, int iVal){
drh32bc3f62007-08-21 20:25:39 +0000337 assert( sqlite3_mutex_held(pCtx->s.db->mutex) );
drh4f26d6c2004-05-26 23:25:30 +0000338 sqlite3VdbeMemSetInt64(&pCtx->s, (i64)iVal);
339}
340void sqlite3_result_int64(sqlite3_context *pCtx, i64 iVal){
drh32bc3f62007-08-21 20:25:39 +0000341 assert( sqlite3_mutex_held(pCtx->s.db->mutex) );
drh4f26d6c2004-05-26 23:25:30 +0000342 sqlite3VdbeMemSetInt64(&pCtx->s, iVal);
343}
344void sqlite3_result_null(sqlite3_context *pCtx){
drh32bc3f62007-08-21 20:25:39 +0000345 assert( sqlite3_mutex_held(pCtx->s.db->mutex) );
drhf4479502004-05-27 03:12:53 +0000346 sqlite3VdbeMemSetNull(&pCtx->s);
drh4f26d6c2004-05-26 23:25:30 +0000347}
348void sqlite3_result_text(
349 sqlite3_context *pCtx,
350 const char *z,
351 int n,
danielk1977d8123362004-06-12 09:25:12 +0000352 void (*xDel)(void *)
drh4f26d6c2004-05-26 23:25:30 +0000353){
drh32bc3f62007-08-21 20:25:39 +0000354 assert( sqlite3_mutex_held(pCtx->s.db->mutex) );
drhb21c8cd2007-08-21 19:33:56 +0000355 sqlite3VdbeMemSetStr(&pCtx->s, z, n, SQLITE_UTF8, xDel);
drh4f26d6c2004-05-26 23:25:30 +0000356}
drh6c626082004-11-14 21:56:29 +0000357#ifndef SQLITE_OMIT_UTF16
drh4f26d6c2004-05-26 23:25:30 +0000358void sqlite3_result_text16(
359 sqlite3_context *pCtx,
360 const void *z,
361 int n,
danielk1977d8123362004-06-12 09:25:12 +0000362 void (*xDel)(void *)
drh4f26d6c2004-05-26 23:25:30 +0000363){
drh32bc3f62007-08-21 20:25:39 +0000364 assert( sqlite3_mutex_held(pCtx->s.db->mutex) );
drhb21c8cd2007-08-21 19:33:56 +0000365 sqlite3VdbeMemSetStr(&pCtx->s, z, n, SQLITE_UTF16NATIVE, xDel);
danielk1977d8123362004-06-12 09:25:12 +0000366}
367void sqlite3_result_text16be(
368 sqlite3_context *pCtx,
369 const void *z,
370 int n,
371 void (*xDel)(void *)
372){
drh32bc3f62007-08-21 20:25:39 +0000373 assert( sqlite3_mutex_held(pCtx->s.db->mutex) );
drhb21c8cd2007-08-21 19:33:56 +0000374 sqlite3VdbeMemSetStr(&pCtx->s, z, n, SQLITE_UTF16BE, xDel);
danielk1977d8123362004-06-12 09:25:12 +0000375}
376void sqlite3_result_text16le(
377 sqlite3_context *pCtx,
378 const void *z,
379 int n,
380 void (*xDel)(void *)
381){
drh32bc3f62007-08-21 20:25:39 +0000382 assert( sqlite3_mutex_held(pCtx->s.db->mutex) );
drhb21c8cd2007-08-21 19:33:56 +0000383 sqlite3VdbeMemSetStr(&pCtx->s, z, n, SQLITE_UTF16LE, xDel);
drh4f26d6c2004-05-26 23:25:30 +0000384}
drh6c626082004-11-14 21:56:29 +0000385#endif /* SQLITE_OMIT_UTF16 */
drh4f26d6c2004-05-26 23:25:30 +0000386void sqlite3_result_value(sqlite3_context *pCtx, sqlite3_value *pValue){
drh32bc3f62007-08-21 20:25:39 +0000387 assert( sqlite3_mutex_held(pCtx->s.db->mutex) );
drhb21c8cd2007-08-21 19:33:56 +0000388 sqlite3VdbeMemCopy(&pCtx->s, pValue);
drh4f26d6c2004-05-26 23:25:30 +0000389}
drhb026e052007-05-02 01:34:31 +0000390void sqlite3_result_zeroblob(sqlite3_context *pCtx, int n){
drh32bc3f62007-08-21 20:25:39 +0000391 assert( sqlite3_mutex_held(pCtx->s.db->mutex) );
drhb026e052007-05-02 01:34:31 +0000392 sqlite3VdbeMemSetZeroBlob(&pCtx->s, n);
393}
drh69544ec2008-02-06 14:11:34 +0000394void sqlite3_result_error_code(sqlite3_context *pCtx, int errCode){
395 pCtx->isError = errCode;
396}
drh4f26d6c2004-05-26 23:25:30 +0000397
drha0206bc2007-05-08 15:15:02 +0000398/* Force an SQLITE_TOOBIG error. */
399void sqlite3_result_error_toobig(sqlite3_context *pCtx){
drh32bc3f62007-08-21 20:25:39 +0000400 assert( sqlite3_mutex_held(pCtx->s.db->mutex) );
drhbb4957f2008-03-20 14:03:29 +0000401 pCtx->isError = SQLITE_TOOBIG;
402 sqlite3VdbeMemSetStr(&pCtx->s, "string or blob too big", -1,
403 SQLITE_UTF8, SQLITE_STATIC);
drha0206bc2007-05-08 15:15:02 +0000404}
405
danielk1977a1644fd2007-08-29 12:31:25 +0000406/* An SQLITE_NOMEM error. */
407void sqlite3_result_error_nomem(sqlite3_context *pCtx){
408 assert( sqlite3_mutex_held(pCtx->s.db->mutex) );
409 sqlite3VdbeMemSetNull(&pCtx->s);
drh69544ec2008-02-06 14:11:34 +0000410 pCtx->isError = SQLITE_NOMEM;
danielk1977a1644fd2007-08-29 12:31:25 +0000411 pCtx->s.db->mallocFailed = 1;
412}
drh4f26d6c2004-05-26 23:25:30 +0000413
414/*
415** Execute the statement pStmt, either until a row of data is ready, the
416** statement is completely executed or an error occurs.
drhb900aaf2006-11-09 00:24:53 +0000417**
418** This routine implements the bulk of the logic behind the sqlite_step()
419** API. The only thing omitted is the automatic recompile if a
420** schema change has occurred. That detail is handled by the
421** outer sqlite3_step() wrapper procedure.
drh4f26d6c2004-05-26 23:25:30 +0000422*/
drhb900aaf2006-11-09 00:24:53 +0000423static int sqlite3Step(Vdbe *p){
drh9bb575f2004-09-06 17:24:11 +0000424 sqlite3 *db;
drh4f26d6c2004-05-26 23:25:30 +0000425 int rc;
426
danielk1977a185ddb2007-11-15 16:04:15 +0000427 assert(p);
428 if( p->magic!=VDBE_MAGIC_RUN ){
drhf1bfe9a2007-10-17 01:44:20 +0000429 return SQLITE_MISUSE;
430 }
431
danielk1977efaaf572006-01-16 11:29:19 +0000432 /* Assert that malloc() has not failed */
drh17435752007-08-16 04:30:38 +0000433 db = p->db;
434 assert( !db->mallocFailed );
danielk1977261919c2005-12-06 12:52:59 +0000435
drh91b48aa2004-06-30 11:14:18 +0000436 if( p->aborted ){
437 return SQLITE_ABORT;
438 }
danielk1977a21c6b62005-01-24 10:25:59 +0000439 if( p->pc<=0 && p->expired ){
440 if( p->rc==SQLITE_OK ){
441 p->rc = SQLITE_SCHEMA;
442 }
drhb900aaf2006-11-09 00:24:53 +0000443 rc = SQLITE_ERROR;
444 goto end_of_step;
danielk1977a21c6b62005-01-24 10:25:59 +0000445 }
drh4f26d6c2004-05-26 23:25:30 +0000446 if( sqlite3SafetyOn(db) ){
447 p->rc = SQLITE_MISUSE;
448 return SQLITE_MISUSE;
449 }
danielk19771d850a72004-05-31 08:26:49 +0000450 if( p->pc<0 ){
drh15ca1df2006-07-26 13:43:30 +0000451 /* If there are no other statements currently running, then
452 ** reset the interrupt flag. This prevents a call to sqlite3_interrupt
453 ** from interrupting a statement that has not yet started.
454 */
455 if( db->activeVdbeCnt==0 ){
456 db->u1.isInterrupted = 0;
457 }
458
drh19e2d372005-08-29 23:00:03 +0000459#ifndef SQLITE_OMIT_TRACE
drh19e2d372005-08-29 23:00:03 +0000460 if( db->xProfile && !db->init.busy ){
461 double rNow;
danielk1977b4b47412007-08-17 15:53:36 +0000462 sqlite3OsCurrentTime(db->pVfs, &rNow);
drh19e2d372005-08-29 23:00:03 +0000463 p->startTime = (rNow - (int)rNow)*3600.0*24.0*1000000000.0;
464 }
465#endif
drhc16a03b2004-09-15 13:38:10 +0000466
danielk19771d850a72004-05-31 08:26:49 +0000467 db->activeVdbeCnt++;
468 p->pc = 0;
danielk1977dfb316d2008-03-26 18:34:43 +0000469 stmtLruRemove(p);
danielk19771d850a72004-05-31 08:26:49 +0000470 }
drhb7f91642004-10-31 02:22:47 +0000471#ifndef SQLITE_OMIT_EXPLAIN
drh4f26d6c2004-05-26 23:25:30 +0000472 if( p->explain ){
473 rc = sqlite3VdbeList(p);
drhb7f91642004-10-31 02:22:47 +0000474 }else
475#endif /* SQLITE_OMIT_EXPLAIN */
476 {
drh4f26d6c2004-05-26 23:25:30 +0000477 rc = sqlite3VdbeExec(p);
478 }
479
480 if( sqlite3SafetyOff(db) ){
481 rc = SQLITE_MISUSE;
482 }
483
drh19e2d372005-08-29 23:00:03 +0000484#ifndef SQLITE_OMIT_TRACE
485 /* Invoke the profile callback if there is one
486 */
drh949f9cd2008-01-12 21:35:57 +0000487 if( rc!=SQLITE_ROW && db->xProfile && !db->init.busy && p->nOp>0
488 && p->aOp[0].opcode==OP_Trace && p->aOp[0].p4.z!=0 ){
drh19e2d372005-08-29 23:00:03 +0000489 double rNow;
490 u64 elapseTime;
491
danielk1977b4b47412007-08-17 15:53:36 +0000492 sqlite3OsCurrentTime(db->pVfs, &rNow);
drh19e2d372005-08-29 23:00:03 +0000493 elapseTime = (rNow - (int)rNow)*3600.0*24.0*1000000000.0 - p->startTime;
drh949f9cd2008-01-12 21:35:57 +0000494 db->xProfile(db->pProfileArg, p->aOp[0].p4.z, elapseTime);
drh19e2d372005-08-29 23:00:03 +0000495 }
496#endif
497
danielk197797a227c2006-01-20 16:32:04 +0000498 sqlite3Error(p->db, rc, 0);
danielk197776e8d1a2006-01-18 18:22:43 +0000499 p->rc = sqlite3ApiExit(p->db, p->rc);
drhb900aaf2006-11-09 00:24:53 +0000500end_of_step:
drh4ac285a2006-09-15 07:28:50 +0000501 assert( (rc&0xff)==rc );
drhb900aaf2006-11-09 00:24:53 +0000502 if( p->zSql && (rc&0xff)<SQLITE_ROW ){
503 /* This behavior occurs if sqlite3_prepare_v2() was used to build
504 ** the prepared statement. Return error codes directly */
danielk1977612642d2007-07-12 13:18:05 +0000505 sqlite3Error(p->db, p->rc, 0);
drhb900aaf2006-11-09 00:24:53 +0000506 return p->rc;
507 }else{
508 /* This is for legacy sqlite3_prepare() builds and when the code
509 ** is SQLITE_ROW or SQLITE_DONE */
510 return rc;
511 }
512}
513
514/*
515** This is the top-level implementation of sqlite3_step(). Call
516** sqlite3Step() to do most of the work. If a schema error occurs,
517** call sqlite3Reprepare() and try again.
518*/
519#ifdef SQLITE_OMIT_PARSER
520int sqlite3_step(sqlite3_stmt *pStmt){
danielk1977a185ddb2007-11-15 16:04:15 +0000521 int rc = SQLITE_MISUSE;
522 if( pStmt ){
523 Vdbe *v;
524 v = (Vdbe*)pStmt;
525 sqlite3_mutex_enter(v->db->mutex);
526 rc = sqlite3Step(v);
527 sqlite3_mutex_leave(v->db->mutex);
528 }
drhb21c8cd2007-08-21 19:33:56 +0000529 return rc;
drhb900aaf2006-11-09 00:24:53 +0000530}
531#else
532int sqlite3_step(sqlite3_stmt *pStmt){
danielk1977a185ddb2007-11-15 16:04:15 +0000533 int rc = SQLITE_MISUSE;
534 if( pStmt ){
535 int cnt = 0;
536 Vdbe *v = (Vdbe*)pStmt;
537 sqlite3 *db = v->db;
538 sqlite3_mutex_enter(db->mutex);
539 while( (rc = sqlite3Step(v))==SQLITE_SCHEMA
540 && cnt++ < 5
danielk1977dfb316d2008-03-26 18:34:43 +0000541 && vdbeReprepare(v) ){
danielk1977a185ddb2007-11-15 16:04:15 +0000542 sqlite3_reset(pStmt);
543 v->expired = 0;
danielk19778e556522007-11-13 10:30:24 +0000544 }
danielk1977a185ddb2007-11-15 16:04:15 +0000545 if( rc==SQLITE_SCHEMA && v->zSql && db->pErr ){
546 /* This case occurs after failing to recompile an sql statement.
547 ** The error message from the SQL compiler has already been loaded
548 ** into the database handle. This block copies the error message
549 ** from the database handle into the statement and sets the statement
550 ** program counter to 0 to ensure that when the statement is
551 ** finalized or reset the parser error message is available via
552 ** sqlite3_errmsg() and sqlite3_errcode().
553 */
554 const char *zErr = (const char *)sqlite3_value_text(db->pErr);
555 sqlite3_free(v->zErrMsg);
556 if( !db->mallocFailed ){
557 v->zErrMsg = sqlite3DbStrDup(db, zErr);
558 } else {
559 v->zErrMsg = 0;
560 v->rc = SQLITE_NOMEM;
561 }
562 }
563 rc = sqlite3ApiExit(db, rc);
564 sqlite3_mutex_leave(db->mutex);
danielk19778e556522007-11-13 10:30:24 +0000565 }
danielk197776e8d1a2006-01-18 18:22:43 +0000566 return rc;
drh4f26d6c2004-05-26 23:25:30 +0000567}
drhb900aaf2006-11-09 00:24:53 +0000568#endif
drh4f26d6c2004-05-26 23:25:30 +0000569
570/*
drheb2e1762004-05-27 01:53:56 +0000571** Extract the user data from a sqlite3_context structure and return a
572** pointer to it.
573*/
574void *sqlite3_user_data(sqlite3_context *p){
575 assert( p && p->pFunc );
576 return p->pFunc->pUserData;
577}
578
579/*
drhfa4a4b92008-03-19 21:45:51 +0000580** Extract the user data from a sqlite3_context structure and return a
581** pointer to it.
582*/
583sqlite3 *sqlite3_context_db_handle(sqlite3_context *p){
584 assert( p && p->pFunc );
585 return p->s.db;
586}
587
588/*
drhb7481e72006-09-16 21:45:14 +0000589** The following is the implementation of an SQL function that always
590** fails with an error message stating that the function is used in the
591** wrong context. The sqlite3_overload_function() API might construct
592** SQL function that use this routine so that the functions will exist
593** for name resolution but are actually overloaded by the xFindFunction
594** method of virtual tables.
595*/
596void sqlite3InvalidFunction(
597 sqlite3_context *context, /* The function calling context */
598 int argc, /* Number of arguments to the function */
599 sqlite3_value **argv /* Value of each argument */
600){
601 const char *zName = context->pFunc->zName;
602 char *zErr;
danielk19771e536952007-08-16 10:09:01 +0000603 zErr = sqlite3MPrintf(0,
drhb7481e72006-09-16 21:45:14 +0000604 "unable to use function %s in the requested context", zName);
605 sqlite3_result_error(context, zErr, -1);
danielk19771e536952007-08-16 10:09:01 +0000606 sqlite3_free(zErr);
drhb7481e72006-09-16 21:45:14 +0000607}
608
609/*
drheb2e1762004-05-27 01:53:56 +0000610** Allocate or return the aggregate context for a user function. A new
611** context is allocated on the first call. Subsequent calls return the
612** same context that was returned on prior calls.
drheb2e1762004-05-27 01:53:56 +0000613*/
614void *sqlite3_aggregate_context(sqlite3_context *p, int nByte){
drhb21c8cd2007-08-21 19:33:56 +0000615 Mem *pMem;
drh825c6622005-09-08 19:01:05 +0000616 assert( p && p->pFunc && p->pFunc->xStep );
drhb21c8cd2007-08-21 19:33:56 +0000617 assert( sqlite3_mutex_held(p->s.db->mutex) );
618 pMem = p->pMem;
drha10a34b2005-09-07 22:09:48 +0000619 if( (pMem->flags & MEM_Agg)==0 ){
620 if( nByte==0 ){
danielk19775f096132008-03-28 15:44:09 +0000621 sqlite3VdbeMemReleaseExternal(pMem);
622 pMem->flags = MEM_Null;
drha10a34b2005-09-07 22:09:48 +0000623 pMem->z = 0;
drheb2e1762004-05-27 01:53:56 +0000624 }else{
danielk19775f096132008-03-28 15:44:09 +0000625 sqlite3VdbeMemGrow(pMem, nByte, 0);
drha10a34b2005-09-07 22:09:48 +0000626 pMem->flags = MEM_Agg;
drh3c024d62007-03-30 11:23:45 +0000627 pMem->u.pDef = p->pFunc;
danielk19775f096132008-03-28 15:44:09 +0000628 if( pMem->z ){
629 memset(pMem->z, 0, nByte);
630 }
drheb2e1762004-05-27 01:53:56 +0000631 }
632 }
drhabfcea22005-09-06 20:36:48 +0000633 return (void*)pMem->z;
drheb2e1762004-05-27 01:53:56 +0000634}
635
636/*
danielk1977682f68b2004-06-05 10:22:17 +0000637** Return the auxilary data pointer, if any, for the iArg'th argument to
638** the user-function defined by pCtx.
639*/
640void *sqlite3_get_auxdata(sqlite3_context *pCtx, int iArg){
drhb21c8cd2007-08-21 19:33:56 +0000641 VdbeFunc *pVdbeFunc;
642
643 assert( sqlite3_mutex_held(pCtx->s.db->mutex) );
644 pVdbeFunc = pCtx->pVdbeFunc;
danielk1977682f68b2004-06-05 10:22:17 +0000645 if( !pVdbeFunc || iArg>=pVdbeFunc->nAux || iArg<0 ){
646 return 0;
647 }
drhf92c7ff2004-06-19 15:40:23 +0000648 return pVdbeFunc->apAux[iArg].pAux;
danielk1977682f68b2004-06-05 10:22:17 +0000649}
650
651/*
652** Set the auxilary data pointer and delete function, for the iArg'th
653** argument to the user-function defined by pCtx. Any previous value is
654** deleted by calling the delete function specified when it was set.
655*/
656void sqlite3_set_auxdata(
657 sqlite3_context *pCtx,
658 int iArg,
659 void *pAux,
660 void (*xDelete)(void*)
661){
662 struct AuxData *pAuxData;
drhf92c7ff2004-06-19 15:40:23 +0000663 VdbeFunc *pVdbeFunc;
danielk1977e0fc5262007-07-26 06:50:05 +0000664 if( iArg<0 ) goto failed;
danielk1977682f68b2004-06-05 10:22:17 +0000665
drhb21c8cd2007-08-21 19:33:56 +0000666 assert( sqlite3_mutex_held(pCtx->s.db->mutex) );
drhf92c7ff2004-06-19 15:40:23 +0000667 pVdbeFunc = pCtx->pVdbeFunc;
668 if( !pVdbeFunc || pVdbeFunc->nAux<=iArg ){
danielk197731dad9d2007-08-16 11:36:15 +0000669 int nAux = (pVdbeFunc ? pVdbeFunc->nAux : 0);
drhf92c7ff2004-06-19 15:40:23 +0000670 int nMalloc = sizeof(VdbeFunc) + sizeof(struct AuxData)*iArg;
danielk197726783a52007-08-29 14:06:22 +0000671 pVdbeFunc = sqlite3DbRealloc(pCtx->s.db, pVdbeFunc, nMalloc);
drh17435752007-08-16 04:30:38 +0000672 if( !pVdbeFunc ){
drh17435752007-08-16 04:30:38 +0000673 goto failed;
674 }
drh53f733c2005-09-16 02:38:09 +0000675 pCtx->pVdbeFunc = pVdbeFunc;
danielk197731dad9d2007-08-16 11:36:15 +0000676 memset(&pVdbeFunc->apAux[nAux], 0, sizeof(struct AuxData)*(iArg+1-nAux));
drh998da3a2004-06-19 15:22:56 +0000677 pVdbeFunc->nAux = iArg+1;
678 pVdbeFunc->pFunc = pCtx->pFunc;
danielk1977682f68b2004-06-05 10:22:17 +0000679 }
680
drhf92c7ff2004-06-19 15:40:23 +0000681 pAuxData = &pVdbeFunc->apAux[iArg];
danielk1977682f68b2004-06-05 10:22:17 +0000682 if( pAuxData->pAux && pAuxData->xDelete ){
683 pAuxData->xDelete(pAuxData->pAux);
684 }
685 pAuxData->pAux = pAux;
686 pAuxData->xDelete = xDelete;
danielk1977e0fc5262007-07-26 06:50:05 +0000687 return;
688
689failed:
690 if( xDelete ){
691 xDelete(pAux);
692 }
danielk1977682f68b2004-06-05 10:22:17 +0000693}
694
695/*
drheb2e1762004-05-27 01:53:56 +0000696** Return the number of times the Step function of a aggregate has been
697** called.
698**
drhcf85a512006-02-09 18:35:29 +0000699** This function is deprecated. Do not use it for new code. It is
700** provide only to avoid breaking legacy code. New aggregate function
701** implementations should keep their own counts within their aggregate
702** context.
drheb2e1762004-05-27 01:53:56 +0000703*/
704int sqlite3_aggregate_count(sqlite3_context *p){
705 assert( p && p->pFunc && p->pFunc->xStep );
drhabfcea22005-09-06 20:36:48 +0000706 return p->pMem->n;
drheb2e1762004-05-27 01:53:56 +0000707}
708
709/*
drh4f26d6c2004-05-26 23:25:30 +0000710** Return the number of columns in the result set for the statement pStmt.
711*/
712int sqlite3_column_count(sqlite3_stmt *pStmt){
713 Vdbe *pVm = (Vdbe *)pStmt;
drh1bcdb0c2004-08-28 14:49:46 +0000714 return pVm ? pVm->nResColumn : 0;
drh4f26d6c2004-05-26 23:25:30 +0000715}
716
717/*
718** Return the number of values available from the current row of the
719** currently executing statement pStmt.
720*/
721int sqlite3_data_count(sqlite3_stmt *pStmt){
722 Vdbe *pVm = (Vdbe *)pStmt;
drhd4e70eb2008-01-02 00:34:36 +0000723 if( pVm==0 || pVm->pResultSet==0 ) return 0;
drh4f26d6c2004-05-26 23:25:30 +0000724 return pVm->nResColumn;
725}
726
727
728/*
729** Check to see if column iCol of the given statement is valid. If
730** it is, return a pointer to the Mem for the value of that column.
731** If iCol is not valid, return a pointer to a Mem which has a value
732** of NULL.
733*/
734static Mem *columnMem(sqlite3_stmt *pStmt, int i){
drh32bc3f62007-08-21 20:25:39 +0000735 Vdbe *pVm;
736 int vals;
737 Mem *pOut;
738
739 pVm = (Vdbe *)pStmt;
drhd4e70eb2008-01-02 00:34:36 +0000740 if( pVm && pVm->pResultSet!=0 && i<pVm->nResColumn && i>=0 ){
drh32bc3f62007-08-21 20:25:39 +0000741 sqlite3_mutex_enter(pVm->db->mutex);
742 vals = sqlite3_data_count(pStmt);
drhd4e70eb2008-01-02 00:34:36 +0000743 pOut = &pVm->pResultSet[i];
drh32bc3f62007-08-21 20:25:39 +0000744 }else{
rse867780a2008-03-29 12:39:39 +0000745 static const Mem nullMem = {{0}, 0.0, 0, "", 0, MEM_Null, SQLITE_NULL, 0, 0, 0 };
drh27641702007-08-22 02:56:42 +0000746 if( pVm->db ){
747 sqlite3_mutex_enter(pVm->db->mutex);
748 sqlite3Error(pVm->db, SQLITE_RANGE, 0);
749 }
drh32bc3f62007-08-21 20:25:39 +0000750 pOut = (Mem*)&nullMem;
drh4f26d6c2004-05-26 23:25:30 +0000751 }
drh32bc3f62007-08-21 20:25:39 +0000752 return pOut;
drh4f26d6c2004-05-26 23:25:30 +0000753}
754
danielk19772e588c72005-12-09 14:25:08 +0000755/*
756** This function is called after invoking an sqlite3_value_XXX function on a
757** column value (i.e. a value returned by evaluating an SQL expression in the
758** select list of a SELECT statement) that may cause a malloc() failure. If
759** malloc() has failed, the threads mallocFailed flag is cleared and the result
760** code of statement pStmt set to SQLITE_NOMEM.
761**
drh32bc3f62007-08-21 20:25:39 +0000762** Specifically, this is called from within:
danielk19772e588c72005-12-09 14:25:08 +0000763**
764** sqlite3_column_int()
765** sqlite3_column_int64()
766** sqlite3_column_text()
767** sqlite3_column_text16()
768** sqlite3_column_real()
769** sqlite3_column_bytes()
770** sqlite3_column_bytes16()
771**
772** But not for sqlite3_column_blob(), which never calls malloc().
773*/
774static void columnMallocFailure(sqlite3_stmt *pStmt)
775{
776 /* If malloc() failed during an encoding conversion within an
777 ** sqlite3_column_XXX API, then set the return code of the statement to
778 ** SQLITE_NOMEM. The next call to _step() (if any) will return SQLITE_ERROR
779 ** and _finalize() will return NOMEM.
780 */
danielk197754f01982006-01-18 15:25:17 +0000781 Vdbe *p = (Vdbe *)pStmt;
drh32bc3f62007-08-21 20:25:39 +0000782 if( p ){
783 p->rc = sqlite3ApiExit(p->db, p->rc);
784 sqlite3_mutex_leave(p->db->mutex);
785 }
danielk19772e588c72005-12-09 14:25:08 +0000786}
787
drh4f26d6c2004-05-26 23:25:30 +0000788/**************************** sqlite3_column_ *******************************
789** The following routines are used to access elements of the current row
790** in the result set.
791*/
danielk1977c572ef72004-05-27 09:28:41 +0000792const void *sqlite3_column_blob(sqlite3_stmt *pStmt, int i){
danielk19772e588c72005-12-09 14:25:08 +0000793 const void *val;
danielk19772e588c72005-12-09 14:25:08 +0000794 val = sqlite3_value_blob( columnMem(pStmt,i) );
danielk1977c9cf9012007-05-30 10:36:47 +0000795 /* Even though there is no encoding conversion, value_blob() might
796 ** need to call malloc() to expand the result of a zeroblob()
797 ** expression.
798 */
799 columnMallocFailure(pStmt);
danielk19772e588c72005-12-09 14:25:08 +0000800 return val;
danielk1977c572ef72004-05-27 09:28:41 +0000801}
drh4f26d6c2004-05-26 23:25:30 +0000802int sqlite3_column_bytes(sqlite3_stmt *pStmt, int i){
danielk19772e588c72005-12-09 14:25:08 +0000803 int val = sqlite3_value_bytes( columnMem(pStmt,i) );
804 columnMallocFailure(pStmt);
805 return val;
drh4f26d6c2004-05-26 23:25:30 +0000806}
807int sqlite3_column_bytes16(sqlite3_stmt *pStmt, int i){
danielk19772e588c72005-12-09 14:25:08 +0000808 int val = sqlite3_value_bytes16( columnMem(pStmt,i) );
809 columnMallocFailure(pStmt);
810 return val;
drh4f26d6c2004-05-26 23:25:30 +0000811}
812double sqlite3_column_double(sqlite3_stmt *pStmt, int i){
danielk19772e588c72005-12-09 14:25:08 +0000813 double val = sqlite3_value_double( columnMem(pStmt,i) );
814 columnMallocFailure(pStmt);
815 return val;
drh4f26d6c2004-05-26 23:25:30 +0000816}
817int sqlite3_column_int(sqlite3_stmt *pStmt, int i){
danielk19772e588c72005-12-09 14:25:08 +0000818 int val = sqlite3_value_int( columnMem(pStmt,i) );
819 columnMallocFailure(pStmt);
820 return val;
drh4f26d6c2004-05-26 23:25:30 +0000821}
drhefad9992004-06-22 12:13:55 +0000822sqlite_int64 sqlite3_column_int64(sqlite3_stmt *pStmt, int i){
danielk19772e588c72005-12-09 14:25:08 +0000823 sqlite_int64 val = sqlite3_value_int64( columnMem(pStmt,i) );
824 columnMallocFailure(pStmt);
825 return val;
drh4f26d6c2004-05-26 23:25:30 +0000826}
827const unsigned char *sqlite3_column_text(sqlite3_stmt *pStmt, int i){
danielk19772e588c72005-12-09 14:25:08 +0000828 const unsigned char *val = sqlite3_value_text( columnMem(pStmt,i) );
829 columnMallocFailure(pStmt);
830 return val;
drh4f26d6c2004-05-26 23:25:30 +0000831}
drhd1e47332005-06-26 17:55:33 +0000832sqlite3_value *sqlite3_column_value(sqlite3_stmt *pStmt, int i){
drh32bc3f62007-08-21 20:25:39 +0000833 sqlite3_value *pOut = columnMem(pStmt, i);
834 columnMallocFailure(pStmt);
835 return pOut;
drhd1e47332005-06-26 17:55:33 +0000836}
drh6c626082004-11-14 21:56:29 +0000837#ifndef SQLITE_OMIT_UTF16
drh4f26d6c2004-05-26 23:25:30 +0000838const void *sqlite3_column_text16(sqlite3_stmt *pStmt, int i){
danielk19772e588c72005-12-09 14:25:08 +0000839 const void *val = sqlite3_value_text16( columnMem(pStmt,i) );
840 columnMallocFailure(pStmt);
841 return val;
drh4f26d6c2004-05-26 23:25:30 +0000842}
drh6c626082004-11-14 21:56:29 +0000843#endif /* SQLITE_OMIT_UTF16 */
drh4f26d6c2004-05-26 23:25:30 +0000844int sqlite3_column_type(sqlite3_stmt *pStmt, int i){
drh32bc3f62007-08-21 20:25:39 +0000845 int iType = sqlite3_value_type( columnMem(pStmt,i) );
846 columnMallocFailure(pStmt);
847 return iType;
drh4f26d6c2004-05-26 23:25:30 +0000848}
849
drh29d72102006-02-09 22:13:41 +0000850/* The following function is experimental and subject to change or
851** removal */
852/*int sqlite3_column_numeric_type(sqlite3_stmt *pStmt, int i){
853** return sqlite3_value_numeric_type( columnMem(pStmt,i) );
854**}
855*/
856
drh5e2517e2004-09-24 23:59:12 +0000857/*
858** Convert the N-th element of pStmt->pColName[] into a string using
859** xFunc() then return that string. If N is out of range, return 0.
drhe590fbd2005-05-20 19:36:01 +0000860**
861** There are up to 5 names for each column. useType determines which
862** name is returned. Here are the names:
863**
864** 0 The column name as it should be displayed for output
865** 1 The datatype name for the column
866** 2 The name of the database that the column derives from
867** 3 The name of the table that the column derives from
868** 4 The name of the table column that the result column derives from
869**
870** If the result is not a simple column reference (if it is an expression
871** or a constant) then useTypes 2, 3, and 4 return NULL.
drh5e2517e2004-09-24 23:59:12 +0000872*/
873static const void *columnName(
874 sqlite3_stmt *pStmt,
875 int N,
876 const void *(*xFunc)(Mem*),
877 int useType
878){
drh32bc3f62007-08-21 20:25:39 +0000879 const void *ret = 0;
drh5e2517e2004-09-24 23:59:12 +0000880 Vdbe *p = (Vdbe *)pStmt;
drh32bc3f62007-08-21 20:25:39 +0000881 int n;
882
drh5e2517e2004-09-24 23:59:12 +0000883
drh32bc3f62007-08-21 20:25:39 +0000884 if( p!=0 ){
885 n = sqlite3_column_count(pStmt);
886 if( N<n && N>=0 ){
887 N += useType*n;
drh153c62c2007-08-24 03:51:33 +0000888 sqlite3_mutex_enter(p->db->mutex);
drh32bc3f62007-08-21 20:25:39 +0000889 ret = xFunc(&p->aColName[N]);
890
drh32bc3f62007-08-21 20:25:39 +0000891 /* A malloc may have failed inside of the xFunc() call. If this
892 ** is the case, clear the mallocFailed flag and return NULL.
893 */
894 if( p->db && p->db->mallocFailed ){
895 p->db->mallocFailed = 0;
896 ret = 0;
897 }
drh153c62c2007-08-24 03:51:33 +0000898 sqlite3_mutex_leave(p->db->mutex);
drh32bc3f62007-08-21 20:25:39 +0000899 }
drh5e2517e2004-09-24 23:59:12 +0000900 }
danielk197700fd9572005-12-07 06:27:43 +0000901 return ret;
drh5e2517e2004-09-24 23:59:12 +0000902}
903
drh4f26d6c2004-05-26 23:25:30 +0000904/*
905** Return the name of the Nth column of the result set returned by SQL
906** statement pStmt.
907*/
908const char *sqlite3_column_name(sqlite3_stmt *pStmt, int N){
danielk1977955de522006-02-10 02:27:42 +0000909 return columnName(
910 pStmt, N, (const void*(*)(Mem*))sqlite3_value_text, COLNAME_NAME);
drh4f26d6c2004-05-26 23:25:30 +0000911}
drhe590fbd2005-05-20 19:36:01 +0000912#ifndef SQLITE_OMIT_UTF16
913const void *sqlite3_column_name16(sqlite3_stmt *pStmt, int N){
danielk1977955de522006-02-10 02:27:42 +0000914 return columnName(
915 pStmt, N, (const void*(*)(Mem*))sqlite3_value_text16, COLNAME_NAME);
drhe590fbd2005-05-20 19:36:01 +0000916}
917#endif
drh4f26d6c2004-05-26 23:25:30 +0000918
919/*
drh3f913572008-03-22 01:07:17 +0000920** Constraint: If you have ENABLE_COLUMN_METADATA then you must
921** not define OMIT_DECLTYPE.
922*/
923#if defined(SQLITE_OMIT_DECLTYPE) && defined(SQLITE_ENABLE_COLUMN_METADATA)
924# error "Must not define both SQLITE_OMIT_DECLTYPE \
925 and SQLITE_ENABLE_COLUMN_METADATA"
926#endif
927
928#ifndef SQLITE_OMIT_DECLTYPE
929/*
drh6c626082004-11-14 21:56:29 +0000930** Return the column declaration type (if applicable) of the 'i'th column
drhe590fbd2005-05-20 19:36:01 +0000931** of the result set of SQL statement pStmt.
drh6c626082004-11-14 21:56:29 +0000932*/
933const char *sqlite3_column_decltype(sqlite3_stmt *pStmt, int N){
danielk1977955de522006-02-10 02:27:42 +0000934 return columnName(
935 pStmt, N, (const void*(*)(Mem*))sqlite3_value_text, COLNAME_DECLTYPE);
drh6c626082004-11-14 21:56:29 +0000936}
drh6c626082004-11-14 21:56:29 +0000937#ifndef SQLITE_OMIT_UTF16
danielk197776d505b2004-05-28 13:13:02 +0000938const void *sqlite3_column_decltype16(sqlite3_stmt *pStmt, int N){
danielk1977955de522006-02-10 02:27:42 +0000939 return columnName(
940 pStmt, N, (const void*(*)(Mem*))sqlite3_value_text16, COLNAME_DECLTYPE);
drh4f26d6c2004-05-26 23:25:30 +0000941}
drh6c626082004-11-14 21:56:29 +0000942#endif /* SQLITE_OMIT_UTF16 */
drh3f913572008-03-22 01:07:17 +0000943#endif /* SQLITE_OMIT_DECLTYPE */
drh4f26d6c2004-05-26 23:25:30 +0000944
danielk19774b1ae992006-02-10 03:06:10 +0000945#ifdef SQLITE_ENABLE_COLUMN_METADATA
drhe590fbd2005-05-20 19:36:01 +0000946/*
947** Return the name of the database from which a result column derives.
948** NULL is returned if the result column is an expression or constant or
949** anything else which is not an unabiguous reference to a database column.
950*/
951const char *sqlite3_column_database_name(sqlite3_stmt *pStmt, int N){
danielk1977955de522006-02-10 02:27:42 +0000952 return columnName(
953 pStmt, N, (const void*(*)(Mem*))sqlite3_value_text, COLNAME_DATABASE);
drhe590fbd2005-05-20 19:36:01 +0000954}
955#ifndef SQLITE_OMIT_UTF16
956const void *sqlite3_column_database_name16(sqlite3_stmt *pStmt, int N){
danielk1977955de522006-02-10 02:27:42 +0000957 return columnName(
958 pStmt, N, (const void*(*)(Mem*))sqlite3_value_text16, COLNAME_DATABASE);
drhe590fbd2005-05-20 19:36:01 +0000959}
960#endif /* SQLITE_OMIT_UTF16 */
961
962/*
963** Return the name of the table from which a result column derives.
964** NULL is returned if the result column is an expression or constant or
965** anything else which is not an unabiguous reference to a database column.
966*/
967const char *sqlite3_column_table_name(sqlite3_stmt *pStmt, int N){
danielk1977955de522006-02-10 02:27:42 +0000968 return columnName(
969 pStmt, N, (const void*(*)(Mem*))sqlite3_value_text, COLNAME_TABLE);
drhe590fbd2005-05-20 19:36:01 +0000970}
971#ifndef SQLITE_OMIT_UTF16
972const void *sqlite3_column_table_name16(sqlite3_stmt *pStmt, int N){
danielk1977955de522006-02-10 02:27:42 +0000973 return columnName(
974 pStmt, N, (const void*(*)(Mem*))sqlite3_value_text16, COLNAME_TABLE);
drhe590fbd2005-05-20 19:36:01 +0000975}
976#endif /* SQLITE_OMIT_UTF16 */
977
978/*
979** Return the name of the table column from which a result column derives.
980** NULL is returned if the result column is an expression or constant or
981** anything else which is not an unabiguous reference to a database column.
982*/
983const char *sqlite3_column_origin_name(sqlite3_stmt *pStmt, int N){
danielk1977955de522006-02-10 02:27:42 +0000984 return columnName(
985 pStmt, N, (const void*(*)(Mem*))sqlite3_value_text, COLNAME_COLUMN);
drhe590fbd2005-05-20 19:36:01 +0000986}
987#ifndef SQLITE_OMIT_UTF16
988const void *sqlite3_column_origin_name16(sqlite3_stmt *pStmt, int N){
danielk1977955de522006-02-10 02:27:42 +0000989 return columnName(
990 pStmt, N, (const void*(*)(Mem*))sqlite3_value_text16, COLNAME_COLUMN);
drhe590fbd2005-05-20 19:36:01 +0000991}
992#endif /* SQLITE_OMIT_UTF16 */
danielk19774b1ae992006-02-10 03:06:10 +0000993#endif /* SQLITE_ENABLE_COLUMN_METADATA */
drhe590fbd2005-05-20 19:36:01 +0000994
995
drh4f26d6c2004-05-26 23:25:30 +0000996/******************************* sqlite3_bind_ ***************************
997**
998** Routines used to attach values to wildcards in a compiled SQL statement.
999*/
1000/*
1001** Unbind the value bound to variable i in virtual machine p. This is the
1002** the same as binding a NULL value to the column. If the "i" parameter is
1003** out of range, then SQLITE_RANGE is returned. Othewise SQLITE_OK.
1004**
1005** The error code stored in database p->db is overwritten with the return
1006** value in any case.
1007*/
1008static int vdbeUnbind(Vdbe *p, int i){
1009 Mem *pVar;
drh1bcdb0c2004-08-28 14:49:46 +00001010 if( p==0 || p->magic!=VDBE_MAGIC_RUN || p->pc>=0 ){
drh473d1792005-06-06 17:54:55 +00001011 if( p ) sqlite3Error(p->db, SQLITE_MISUSE, 0);
drh4f26d6c2004-05-26 23:25:30 +00001012 return SQLITE_MISUSE;
1013 }
1014 if( i<1 || i>p->nVar ){
1015 sqlite3Error(p->db, SQLITE_RANGE, 0);
1016 return SQLITE_RANGE;
1017 }
1018 i--;
drh290c1942004-08-21 17:54:45 +00001019 pVar = &p->aVar[i];
danielk1977d8123362004-06-12 09:25:12 +00001020 sqlite3VdbeMemRelease(pVar);
drh4f26d6c2004-05-26 23:25:30 +00001021 pVar->flags = MEM_Null;
1022 sqlite3Error(p->db, SQLITE_OK, 0);
1023 return SQLITE_OK;
1024}
1025
1026/*
drh5e2517e2004-09-24 23:59:12 +00001027** Bind a text or BLOB value.
drh4f26d6c2004-05-26 23:25:30 +00001028*/
drh5e2517e2004-09-24 23:59:12 +00001029static int bindText(
drh605264d2007-08-21 15:13:19 +00001030 sqlite3_stmt *pStmt, /* The statement to bind against */
1031 int i, /* Index of the parameter to bind */
1032 const void *zData, /* Pointer to the data to be bound */
1033 int nData, /* Number of bytes of data to be bound */
1034 void (*xDel)(void*), /* Destructor for the data */
1035 int encoding /* Encoding for the data */
drh4f26d6c2004-05-26 23:25:30 +00001036){
1037 Vdbe *p = (Vdbe *)pStmt;
1038 Mem *pVar;
1039 int rc;
1040
drh605264d2007-08-21 15:13:19 +00001041 if( p==0 ){
1042 return SQLITE_MISUSE;
1043 }
1044 sqlite3_mutex_enter(p->db->mutex);
drh4f26d6c2004-05-26 23:25:30 +00001045 rc = vdbeUnbind(p, i);
drh27641702007-08-22 02:56:42 +00001046 if( rc==SQLITE_OK && zData!=0 ){
1047 pVar = &p->aVar[i-1];
1048 rc = sqlite3VdbeMemSetStr(pVar, zData, nData, encoding, xDel);
1049 if( rc==SQLITE_OK && encoding!=0 ){
1050 rc = sqlite3VdbeChangeEncoding(pVar, ENC(p->db));
1051 }
1052 sqlite3Error(p->db, rc, 0);
1053 rc = sqlite3ApiExit(p->db, rc);
drh4f26d6c2004-05-26 23:25:30 +00001054 }
drh605264d2007-08-21 15:13:19 +00001055 sqlite3_mutex_leave(p->db->mutex);
1056 return rc;
drh4f26d6c2004-05-26 23:25:30 +00001057}
drh5e2517e2004-09-24 23:59:12 +00001058
1059
1060/*
1061** Bind a blob value to an SQL statement variable.
1062*/
1063int sqlite3_bind_blob(
1064 sqlite3_stmt *pStmt,
1065 int i,
1066 const void *zData,
1067 int nData,
1068 void (*xDel)(void*)
1069){
1070 return bindText(pStmt, i, zData, nData, xDel, 0);
1071}
drh4f26d6c2004-05-26 23:25:30 +00001072int sqlite3_bind_double(sqlite3_stmt *pStmt, int i, double rValue){
1073 int rc;
1074 Vdbe *p = (Vdbe *)pStmt;
drh605264d2007-08-21 15:13:19 +00001075 sqlite3_mutex_enter(p->db->mutex);
drh4f26d6c2004-05-26 23:25:30 +00001076 rc = vdbeUnbind(p, i);
1077 if( rc==SQLITE_OK ){
drh290c1942004-08-21 17:54:45 +00001078 sqlite3VdbeMemSetDouble(&p->aVar[i-1], rValue);
drh4f26d6c2004-05-26 23:25:30 +00001079 }
drh605264d2007-08-21 15:13:19 +00001080 sqlite3_mutex_leave(p->db->mutex);
danielk1977f4618892004-06-28 13:09:11 +00001081 return rc;
drh4f26d6c2004-05-26 23:25:30 +00001082}
1083int sqlite3_bind_int(sqlite3_stmt *p, int i, int iValue){
drhefad9992004-06-22 12:13:55 +00001084 return sqlite3_bind_int64(p, i, (i64)iValue);
drh4f26d6c2004-05-26 23:25:30 +00001085}
drhefad9992004-06-22 12:13:55 +00001086int sqlite3_bind_int64(sqlite3_stmt *pStmt, int i, sqlite_int64 iValue){
drh4f26d6c2004-05-26 23:25:30 +00001087 int rc;
1088 Vdbe *p = (Vdbe *)pStmt;
drh605264d2007-08-21 15:13:19 +00001089 sqlite3_mutex_enter(p->db->mutex);
drh4f26d6c2004-05-26 23:25:30 +00001090 rc = vdbeUnbind(p, i);
1091 if( rc==SQLITE_OK ){
drh290c1942004-08-21 17:54:45 +00001092 sqlite3VdbeMemSetInt64(&p->aVar[i-1], iValue);
drh4f26d6c2004-05-26 23:25:30 +00001093 }
drh605264d2007-08-21 15:13:19 +00001094 sqlite3_mutex_leave(p->db->mutex);
drh4f26d6c2004-05-26 23:25:30 +00001095 return rc;
1096}
drh605264d2007-08-21 15:13:19 +00001097int sqlite3_bind_null(sqlite3_stmt *pStmt, int i){
1098 int rc;
1099 Vdbe *p = (Vdbe*)pStmt;
1100 sqlite3_mutex_enter(p->db->mutex);
1101 rc = vdbeUnbind(p, i);
1102 sqlite3_mutex_leave(p->db->mutex);
1103 return rc;
drh4f26d6c2004-05-26 23:25:30 +00001104}
1105int sqlite3_bind_text(
1106 sqlite3_stmt *pStmt,
1107 int i,
1108 const char *zData,
1109 int nData,
danielk1977d8123362004-06-12 09:25:12 +00001110 void (*xDel)(void*)
drh4f26d6c2004-05-26 23:25:30 +00001111){
drh5e2517e2004-09-24 23:59:12 +00001112 return bindText(pStmt, i, zData, nData, xDel, SQLITE_UTF8);
drh4f26d6c2004-05-26 23:25:30 +00001113}
drh6c626082004-11-14 21:56:29 +00001114#ifndef SQLITE_OMIT_UTF16
drh4f26d6c2004-05-26 23:25:30 +00001115int sqlite3_bind_text16(
1116 sqlite3_stmt *pStmt,
1117 int i,
1118 const void *zData,
1119 int nData,
danielk1977d8123362004-06-12 09:25:12 +00001120 void (*xDel)(void*)
drh4f26d6c2004-05-26 23:25:30 +00001121){
drh5e2517e2004-09-24 23:59:12 +00001122 return bindText(pStmt, i, zData, nData, xDel, SQLITE_UTF16NATIVE);
drh4f26d6c2004-05-26 23:25:30 +00001123}
drh6c626082004-11-14 21:56:29 +00001124#endif /* SQLITE_OMIT_UTF16 */
danielk197726e41442006-06-14 15:16:35 +00001125int sqlite3_bind_value(sqlite3_stmt *pStmt, int i, const sqlite3_value *pValue){
1126 int rc;
1127 Vdbe *p = (Vdbe *)pStmt;
drh605264d2007-08-21 15:13:19 +00001128 sqlite3_mutex_enter(p->db->mutex);
danielk197726e41442006-06-14 15:16:35 +00001129 rc = vdbeUnbind(p, i);
1130 if( rc==SQLITE_OK ){
drhb21c8cd2007-08-21 19:33:56 +00001131 rc = sqlite3VdbeMemCopy(&p->aVar[i-1], pValue);
danielk197726e41442006-06-14 15:16:35 +00001132 }
danielk1977a7a8e142008-02-13 18:25:27 +00001133 rc = sqlite3ApiExit(p->db, rc);
drh605264d2007-08-21 15:13:19 +00001134 sqlite3_mutex_leave(p->db->mutex);
danielk197726e41442006-06-14 15:16:35 +00001135 return rc;
1136}
drhb026e052007-05-02 01:34:31 +00001137int sqlite3_bind_zeroblob(sqlite3_stmt *pStmt, int i, int n){
1138 int rc;
1139 Vdbe *p = (Vdbe *)pStmt;
drh605264d2007-08-21 15:13:19 +00001140 sqlite3_mutex_enter(p->db->mutex);
drhb026e052007-05-02 01:34:31 +00001141 rc = vdbeUnbind(p, i);
1142 if( rc==SQLITE_OK ){
1143 sqlite3VdbeMemSetZeroBlob(&p->aVar[i-1], n);
1144 }
drh605264d2007-08-21 15:13:19 +00001145 sqlite3_mutex_leave(p->db->mutex);
drhb026e052007-05-02 01:34:31 +00001146 return rc;
1147}
drh75f6a032004-07-15 14:15:00 +00001148
1149/*
1150** Return the number of wildcards that can be potentially bound to.
1151** This routine is added to support DBD::SQLite.
drh75f6a032004-07-15 14:15:00 +00001152*/
1153int sqlite3_bind_parameter_count(sqlite3_stmt *pStmt){
drh92febd92004-08-20 18:34:20 +00001154 Vdbe *p = (Vdbe*)pStmt;
1155 return p ? p->nVar : 0;
drh75f6a032004-07-15 14:15:00 +00001156}
drh895d7472004-08-20 16:02:39 +00001157
1158/*
drhfa6bc002004-09-07 16:19:52 +00001159** Create a mapping from variable numbers to variable names
1160** in the Vdbe.azVar[] array, if such a mapping does not already
1161** exist.
drh895d7472004-08-20 16:02:39 +00001162*/
drhfa6bc002004-09-07 16:19:52 +00001163static void createVarMap(Vdbe *p){
drh895d7472004-08-20 16:02:39 +00001164 if( !p->okVar ){
drh605264d2007-08-21 15:13:19 +00001165 sqlite3_mutex_enter(p->db->mutex);
1166 if( !p->okVar ){
1167 int j;
1168 Op *pOp;
1169 for(j=0, pOp=p->aOp; j<p->nOp; j++, pOp++){
1170 if( pOp->opcode==OP_Variable ){
1171 assert( pOp->p1>0 && pOp->p1<=p->nVar );
danielk19772dca4ac2008-01-03 11:50:29 +00001172 p->azVar[pOp->p1-1] = pOp->p4.z;
drh605264d2007-08-21 15:13:19 +00001173 }
drh895d7472004-08-20 16:02:39 +00001174 }
drh605264d2007-08-21 15:13:19 +00001175 p->okVar = 1;
drh895d7472004-08-20 16:02:39 +00001176 }
drh605264d2007-08-21 15:13:19 +00001177 sqlite3_mutex_leave(p->db->mutex);
drh895d7472004-08-20 16:02:39 +00001178 }
drhfa6bc002004-09-07 16:19:52 +00001179}
1180
1181/*
1182** Return the name of a wildcard parameter. Return NULL if the index
1183** is out of range or if the wildcard is unnamed.
1184**
1185** The result is always UTF-8.
1186*/
1187const char *sqlite3_bind_parameter_name(sqlite3_stmt *pStmt, int i){
1188 Vdbe *p = (Vdbe*)pStmt;
1189 if( p==0 || i<1 || i>p->nVar ){
1190 return 0;
1191 }
1192 createVarMap(p);
drh895d7472004-08-20 16:02:39 +00001193 return p->azVar[i-1];
1194}
drhfa6bc002004-09-07 16:19:52 +00001195
1196/*
1197** Given a wildcard parameter name, return the index of the variable
1198** with that name. If there is no variable with the given name,
1199** return 0.
1200*/
1201int sqlite3_bind_parameter_index(sqlite3_stmt *pStmt, const char *zName){
1202 Vdbe *p = (Vdbe*)pStmt;
1203 int i;
1204 if( p==0 ){
1205 return 0;
1206 }
1207 createVarMap(p);
drh6a8903c2004-12-10 18:00:04 +00001208 if( zName ){
1209 for(i=0; i<p->nVar; i++){
1210 const char *z = p->azVar[i];
1211 if( z && strcmp(z,zName)==0 ){
1212 return i+1;
1213 }
drhfa6bc002004-09-07 16:19:52 +00001214 }
1215 }
1216 return 0;
1217}
drhf8db1bc2005-04-22 02:38:37 +00001218
drh51942bc2005-06-12 22:01:42 +00001219/*
drhf8db1bc2005-04-22 02:38:37 +00001220** Transfer all bindings from the first statement over to the second.
1221** If the two statements contain a different number of bindings, then
1222** an SQLITE_ERROR is returned.
1223*/
1224int sqlite3_transfer_bindings(sqlite3_stmt *pFromStmt, sqlite3_stmt *pToStmt){
1225 Vdbe *pFrom = (Vdbe*)pFromStmt;
1226 Vdbe *pTo = (Vdbe*)pToStmt;
1227 int i, rc = SQLITE_OK;
1228 if( (pFrom->magic!=VDBE_MAGIC_RUN && pFrom->magic!=VDBE_MAGIC_HALT)
drh27641702007-08-22 02:56:42 +00001229 || (pTo->magic!=VDBE_MAGIC_RUN && pTo->magic!=VDBE_MAGIC_HALT)
1230 || pTo->db!=pFrom->db ){
drhf8db1bc2005-04-22 02:38:37 +00001231 return SQLITE_MISUSE;
1232 }
1233 if( pFrom->nVar!=pTo->nVar ){
1234 return SQLITE_ERROR;
1235 }
drh27641702007-08-22 02:56:42 +00001236 sqlite3_mutex_enter(pTo->db->mutex);
drhf8db1bc2005-04-22 02:38:37 +00001237 for(i=0; rc==SQLITE_OK && i<pFrom->nVar; i++){
drh643167f2008-01-22 21:30:53 +00001238 sqlite3VdbeMemMove(&pTo->aVar[i], &pFrom->aVar[i]);
drhf8db1bc2005-04-22 02:38:37 +00001239 }
drh27641702007-08-22 02:56:42 +00001240 sqlite3_mutex_leave(pTo->db->mutex);
drh4ac285a2006-09-15 07:28:50 +00001241 assert( rc==SQLITE_OK || rc==SQLITE_NOMEM );
drhf8db1bc2005-04-22 02:38:37 +00001242 return rc;
1243}
drh51942bc2005-06-12 22:01:42 +00001244
1245/*
1246** Return the sqlite3* database handle to which the prepared statement given
1247** in the argument belongs. This is the same database handle that was
1248** the first argument to the sqlite3_prepare() that was used to create
1249** the statement in the first place.
1250*/
1251sqlite3 *sqlite3_db_handle(sqlite3_stmt *pStmt){
1252 return pStmt ? ((Vdbe*)pStmt)->db : 0;
1253}