blob: 6716256c0dbe5e57d3b93db5cf10a27d89493462 [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
drhd89bd002005-01-22 03:03:54 +000019/*
20** Return TRUE (non-zero) of the statement supplied as an argument needs
21** to be recompiled. A statement needs to be recompiled whenever the
22** execution environment changes in a way that would alter the program
23** that sqlite3_prepare() generates. For example, if new functions or
24** collating sequences are registered or if an authorizer function is
25** added or changed.
drhd89bd002005-01-22 03:03:54 +000026*/
27int sqlite3_expired(sqlite3_stmt *pStmt){
28 Vdbe *p = (Vdbe*)pStmt;
29 return p==0 || p->expired;
30}
31
drhe30f4422007-08-21 16:15:55 +000032/*
33** The following routine destroys a virtual machine that is created by
34** the sqlite3_compile() routine. The integer returned is an SQLITE_
35** success/failure code that describes the result of executing the virtual
36** machine.
37**
38** This routine sets the error code and string returned by
39** sqlite3_errcode(), sqlite3_errmsg() and sqlite3_errmsg16().
40*/
41int sqlite3_finalize(sqlite3_stmt *pStmt){
42 int rc;
43 if( pStmt==0 ){
44 rc = SQLITE_OK;
45 }else{
46 Vdbe *v = (Vdbe*)pStmt;
drh86f8c192007-08-22 00:39:19 +000047 sqlite3_mutex *mutex = v->db->mutex;
48 sqlite3_mutex_enter(mutex);
drhe30f4422007-08-21 16:15:55 +000049 rc = sqlite3VdbeFinalize(v);
drh86f8c192007-08-22 00:39:19 +000050 sqlite3_mutex_leave(mutex);
drhe30f4422007-08-21 16:15:55 +000051 }
52 return rc;
53}
54
55/*
56** Terminate the current execution of an SQL statement and reset it
57** back to its starting state so that it can be reused. A success code from
58** the prior execution is returned.
59**
60** This routine sets the error code and string returned by
61** sqlite3_errcode(), sqlite3_errmsg() and sqlite3_errmsg16().
62*/
63int sqlite3_reset(sqlite3_stmt *pStmt){
64 int rc;
65 if( pStmt==0 ){
66 rc = SQLITE_OK;
67 }else{
68 Vdbe *v = (Vdbe*)pStmt;
69 sqlite3_mutex_enter(v->db->mutex);
70 rc = sqlite3VdbeReset(v);
71 sqlite3VdbeMakeReady(v, -1, 0, 0, 0);
72 assert( (rc & (v->db->errMask))==rc );
73 sqlite3_mutex_leave(v->db->mutex);
74 }
75 return rc;
76}
77
78/*
79** Set all the parameters in the compiled SQL statement to NULL.
80*/
81int sqlite3_clear_bindings(sqlite3_stmt *pStmt){
82 int i;
83 int rc = SQLITE_OK;
84 Vdbe *v = (Vdbe*)pStmt;
85 sqlite3_mutex_enter(v->db->mutex);
86 for(i=1; rc==SQLITE_OK && i<=sqlite3_bind_parameter_count(pStmt); i++){
87 rc = sqlite3_bind_null(pStmt, i);
88 }
89 sqlite3_mutex_leave(v->db->mutex);
90 return rc;
91}
92
93
drh4f26d6c2004-05-26 23:25:30 +000094/**************************** sqlite3_value_ *******************************
95** The following routines extract information from a Mem or sqlite3_value
96** structure.
97*/
98const void *sqlite3_value_blob(sqlite3_value *pVal){
99 Mem *p = (Mem*)pVal;
100 if( p->flags & (MEM_Blob|MEM_Str) ){
drhb21c8cd2007-08-21 19:33:56 +0000101 sqlite3VdbeMemExpandBlob(p);
drh1f0feef2007-05-15 13:27:07 +0000102 p->flags &= ~MEM_Str;
103 p->flags |= MEM_Blob;
drh4f26d6c2004-05-26 23:25:30 +0000104 return p->z;
105 }else{
106 return sqlite3_value_text(pVal);
107 }
108}
109int sqlite3_value_bytes(sqlite3_value *pVal){
drhb21c8cd2007-08-21 19:33:56 +0000110 return sqlite3ValueBytes(pVal, SQLITE_UTF8);
drh4f26d6c2004-05-26 23:25:30 +0000111}
112int sqlite3_value_bytes16(sqlite3_value *pVal){
drhb21c8cd2007-08-21 19:33:56 +0000113 return sqlite3ValueBytes(pVal, SQLITE_UTF16NATIVE);
drh4f26d6c2004-05-26 23:25:30 +0000114}
115double sqlite3_value_double(sqlite3_value *pVal){
drh6a6124e2004-06-27 01:56:33 +0000116 return sqlite3VdbeRealValue((Mem*)pVal);
drh4f26d6c2004-05-26 23:25:30 +0000117}
118int sqlite3_value_int(sqlite3_value *pVal){
drh6a6124e2004-06-27 01:56:33 +0000119 return sqlite3VdbeIntValue((Mem*)pVal);
drh4f26d6c2004-05-26 23:25:30 +0000120}
drhefad9992004-06-22 12:13:55 +0000121sqlite_int64 sqlite3_value_int64(sqlite3_value *pVal){
drh6a6124e2004-06-27 01:56:33 +0000122 return sqlite3VdbeIntValue((Mem*)pVal);
drh4f26d6c2004-05-26 23:25:30 +0000123}
124const unsigned char *sqlite3_value_text(sqlite3_value *pVal){
drhb21c8cd2007-08-21 19:33:56 +0000125 return (const unsigned char *)sqlite3ValueText(pVal, SQLITE_UTF8);
drh4f26d6c2004-05-26 23:25:30 +0000126}
drh6c626082004-11-14 21:56:29 +0000127#ifndef SQLITE_OMIT_UTF16
drh4f26d6c2004-05-26 23:25:30 +0000128const void *sqlite3_value_text16(sqlite3_value* pVal){
drhb21c8cd2007-08-21 19:33:56 +0000129 return sqlite3ValueText(pVal, SQLITE_UTF16NATIVE);
drh4f26d6c2004-05-26 23:25:30 +0000130}
danielk1977d8123362004-06-12 09:25:12 +0000131const void *sqlite3_value_text16be(sqlite3_value *pVal){
drhb21c8cd2007-08-21 19:33:56 +0000132 return sqlite3ValueText(pVal, SQLITE_UTF16BE);
danielk1977d8123362004-06-12 09:25:12 +0000133}
134const void *sqlite3_value_text16le(sqlite3_value *pVal){
drhb21c8cd2007-08-21 19:33:56 +0000135 return sqlite3ValueText(pVal, SQLITE_UTF16LE);
danielk1977d8123362004-06-12 09:25:12 +0000136}
drh6c626082004-11-14 21:56:29 +0000137#endif /* SQLITE_OMIT_UTF16 */
drh4f26d6c2004-05-26 23:25:30 +0000138int sqlite3_value_type(sqlite3_value* pVal){
drhf4479502004-05-27 03:12:53 +0000139 return pVal->type;
drh4f26d6c2004-05-26 23:25:30 +0000140}
141
142/**************************** sqlite3_result_ *******************************
143** The following routines are used by user-defined functions to specify
144** the function result.
145*/
146void sqlite3_result_blob(
147 sqlite3_context *pCtx,
148 const void *z,
149 int n,
danielk1977d8123362004-06-12 09:25:12 +0000150 void (*xDel)(void *)
drh4f26d6c2004-05-26 23:25:30 +0000151){
drh5708d2d2005-06-22 10:53:59 +0000152 assert( n>=0 );
drh32bc3f62007-08-21 20:25:39 +0000153 assert( sqlite3_mutex_held(pCtx->s.db->mutex) );
drhb21c8cd2007-08-21 19:33:56 +0000154 sqlite3VdbeMemSetStr(&pCtx->s, z, n, 0, xDel);
drh4f26d6c2004-05-26 23:25:30 +0000155}
156void sqlite3_result_double(sqlite3_context *pCtx, double rVal){
drh32bc3f62007-08-21 20:25:39 +0000157 assert( sqlite3_mutex_held(pCtx->s.db->mutex) );
drh4f26d6c2004-05-26 23:25:30 +0000158 sqlite3VdbeMemSetDouble(&pCtx->s, rVal);
159}
160void sqlite3_result_error(sqlite3_context *pCtx, const char *z, int n){
drh32bc3f62007-08-21 20:25:39 +0000161 assert( sqlite3_mutex_held(pCtx->s.db->mutex) );
drh4f26d6c2004-05-26 23:25:30 +0000162 pCtx->isError = 1;
drhb21c8cd2007-08-21 19:33:56 +0000163 sqlite3VdbeMemSetStr(&pCtx->s, z, n, SQLITE_UTF8, SQLITE_TRANSIENT);
drh4f26d6c2004-05-26 23:25:30 +0000164}
danielk1977a1686c92006-01-23 07:52:37 +0000165#ifndef SQLITE_OMIT_UTF16
drh4f26d6c2004-05-26 23:25:30 +0000166void sqlite3_result_error16(sqlite3_context *pCtx, const void *z, int n){
drh32bc3f62007-08-21 20:25:39 +0000167 assert( sqlite3_mutex_held(pCtx->s.db->mutex) );
drh4f26d6c2004-05-26 23:25:30 +0000168 pCtx->isError = 1;
drhb21c8cd2007-08-21 19:33:56 +0000169 sqlite3VdbeMemSetStr(&pCtx->s, z, n, SQLITE_UTF16NATIVE, SQLITE_TRANSIENT);
drh4f26d6c2004-05-26 23:25:30 +0000170}
danielk1977a1686c92006-01-23 07:52:37 +0000171#endif
drhf4479502004-05-27 03:12:53 +0000172void sqlite3_result_int(sqlite3_context *pCtx, int iVal){
drh32bc3f62007-08-21 20:25:39 +0000173 assert( sqlite3_mutex_held(pCtx->s.db->mutex) );
drh4f26d6c2004-05-26 23:25:30 +0000174 sqlite3VdbeMemSetInt64(&pCtx->s, (i64)iVal);
175}
176void sqlite3_result_int64(sqlite3_context *pCtx, i64 iVal){
drh32bc3f62007-08-21 20:25:39 +0000177 assert( sqlite3_mutex_held(pCtx->s.db->mutex) );
drh4f26d6c2004-05-26 23:25:30 +0000178 sqlite3VdbeMemSetInt64(&pCtx->s, iVal);
179}
180void sqlite3_result_null(sqlite3_context *pCtx){
drh32bc3f62007-08-21 20:25:39 +0000181 assert( sqlite3_mutex_held(pCtx->s.db->mutex) );
drhf4479502004-05-27 03:12:53 +0000182 sqlite3VdbeMemSetNull(&pCtx->s);
drh4f26d6c2004-05-26 23:25:30 +0000183}
184void sqlite3_result_text(
185 sqlite3_context *pCtx,
186 const char *z,
187 int n,
danielk1977d8123362004-06-12 09:25:12 +0000188 void (*xDel)(void *)
drh4f26d6c2004-05-26 23:25:30 +0000189){
drh32bc3f62007-08-21 20:25:39 +0000190 assert( sqlite3_mutex_held(pCtx->s.db->mutex) );
drhb21c8cd2007-08-21 19:33:56 +0000191 sqlite3VdbeMemSetStr(&pCtx->s, z, n, SQLITE_UTF8, xDel);
drh4f26d6c2004-05-26 23:25:30 +0000192}
drh6c626082004-11-14 21:56:29 +0000193#ifndef SQLITE_OMIT_UTF16
drh4f26d6c2004-05-26 23:25:30 +0000194void sqlite3_result_text16(
195 sqlite3_context *pCtx,
196 const void *z,
197 int n,
danielk1977d8123362004-06-12 09:25:12 +0000198 void (*xDel)(void *)
drh4f26d6c2004-05-26 23:25:30 +0000199){
drh32bc3f62007-08-21 20:25:39 +0000200 assert( sqlite3_mutex_held(pCtx->s.db->mutex) );
drhb21c8cd2007-08-21 19:33:56 +0000201 sqlite3VdbeMemSetStr(&pCtx->s, z, n, SQLITE_UTF16NATIVE, xDel);
danielk1977d8123362004-06-12 09:25:12 +0000202}
203void sqlite3_result_text16be(
204 sqlite3_context *pCtx,
205 const void *z,
206 int n,
207 void (*xDel)(void *)
208){
drh32bc3f62007-08-21 20:25:39 +0000209 assert( sqlite3_mutex_held(pCtx->s.db->mutex) );
drhb21c8cd2007-08-21 19:33:56 +0000210 sqlite3VdbeMemSetStr(&pCtx->s, z, n, SQLITE_UTF16BE, xDel);
danielk1977d8123362004-06-12 09:25:12 +0000211}
212void sqlite3_result_text16le(
213 sqlite3_context *pCtx,
214 const void *z,
215 int n,
216 void (*xDel)(void *)
217){
drh32bc3f62007-08-21 20:25:39 +0000218 assert( sqlite3_mutex_held(pCtx->s.db->mutex) );
drhb21c8cd2007-08-21 19:33:56 +0000219 sqlite3VdbeMemSetStr(&pCtx->s, z, n, SQLITE_UTF16LE, xDel);
drh4f26d6c2004-05-26 23:25:30 +0000220}
drh6c626082004-11-14 21:56:29 +0000221#endif /* SQLITE_OMIT_UTF16 */
drh4f26d6c2004-05-26 23:25:30 +0000222void sqlite3_result_value(sqlite3_context *pCtx, sqlite3_value *pValue){
drh32bc3f62007-08-21 20:25:39 +0000223 assert( sqlite3_mutex_held(pCtx->s.db->mutex) );
drhb21c8cd2007-08-21 19:33:56 +0000224 sqlite3VdbeMemCopy(&pCtx->s, pValue);
drh4f26d6c2004-05-26 23:25:30 +0000225}
drhb026e052007-05-02 01:34:31 +0000226void sqlite3_result_zeroblob(sqlite3_context *pCtx, int n){
drh32bc3f62007-08-21 20:25:39 +0000227 assert( sqlite3_mutex_held(pCtx->s.db->mutex) );
drhb026e052007-05-02 01:34:31 +0000228 sqlite3VdbeMemSetZeroBlob(&pCtx->s, n);
229}
drh4f26d6c2004-05-26 23:25:30 +0000230
drha0206bc2007-05-08 15:15:02 +0000231/* Force an SQLITE_TOOBIG error. */
232void sqlite3_result_error_toobig(sqlite3_context *pCtx){
drh32bc3f62007-08-21 20:25:39 +0000233 assert( sqlite3_mutex_held(pCtx->s.db->mutex) );
drha0206bc2007-05-08 15:15:02 +0000234 sqlite3VdbeMemSetZeroBlob(&pCtx->s, SQLITE_MAX_LENGTH+1);
235}
236
danielk1977a1644fd2007-08-29 12:31:25 +0000237/* An SQLITE_NOMEM error. */
238void sqlite3_result_error_nomem(sqlite3_context *pCtx){
239 assert( sqlite3_mutex_held(pCtx->s.db->mutex) );
240 sqlite3VdbeMemSetNull(&pCtx->s);
241 pCtx->isError = 1;
242 pCtx->s.db->mallocFailed = 1;
243}
drh4f26d6c2004-05-26 23:25:30 +0000244
245/*
246** Execute the statement pStmt, either until a row of data is ready, the
247** statement is completely executed or an error occurs.
drhb900aaf2006-11-09 00:24:53 +0000248**
249** This routine implements the bulk of the logic behind the sqlite_step()
250** API. The only thing omitted is the automatic recompile if a
251** schema change has occurred. That detail is handled by the
252** outer sqlite3_step() wrapper procedure.
drh4f26d6c2004-05-26 23:25:30 +0000253*/
drhb900aaf2006-11-09 00:24:53 +0000254static int sqlite3Step(Vdbe *p){
drh9bb575f2004-09-06 17:24:11 +0000255 sqlite3 *db;
drh4f26d6c2004-05-26 23:25:30 +0000256 int rc;
257
danielk1977efaaf572006-01-16 11:29:19 +0000258 /* Assert that malloc() has not failed */
drh17435752007-08-16 04:30:38 +0000259 db = p->db;
260 assert( !db->mallocFailed );
danielk1977261919c2005-12-06 12:52:59 +0000261
drh1bcdb0c2004-08-28 14:49:46 +0000262 if( p==0 || p->magic!=VDBE_MAGIC_RUN ){
drh4f26d6c2004-05-26 23:25:30 +0000263 return SQLITE_MISUSE;
264 }
drh91b48aa2004-06-30 11:14:18 +0000265 if( p->aborted ){
266 return SQLITE_ABORT;
267 }
danielk1977a21c6b62005-01-24 10:25:59 +0000268 if( p->pc<=0 && p->expired ){
269 if( p->rc==SQLITE_OK ){
270 p->rc = SQLITE_SCHEMA;
271 }
drhb900aaf2006-11-09 00:24:53 +0000272 rc = SQLITE_ERROR;
273 goto end_of_step;
danielk1977a21c6b62005-01-24 10:25:59 +0000274 }
drh4f26d6c2004-05-26 23:25:30 +0000275 if( sqlite3SafetyOn(db) ){
276 p->rc = SQLITE_MISUSE;
277 return SQLITE_MISUSE;
278 }
danielk19771d850a72004-05-31 08:26:49 +0000279 if( p->pc<0 ){
drh15ca1df2006-07-26 13:43:30 +0000280 /* If there are no other statements currently running, then
281 ** reset the interrupt flag. This prevents a call to sqlite3_interrupt
282 ** from interrupting a statement that has not yet started.
283 */
284 if( db->activeVdbeCnt==0 ){
285 db->u1.isInterrupted = 0;
286 }
287
drh19e2d372005-08-29 23:00:03 +0000288#ifndef SQLITE_OMIT_TRACE
drhc16a03b2004-09-15 13:38:10 +0000289 /* Invoke the trace callback if there is one
290 */
drh19e2d372005-08-29 23:00:03 +0000291 if( db->xTrace && !db->init.busy ){
drhc16a03b2004-09-15 13:38:10 +0000292 assert( p->nOp>0 );
293 assert( p->aOp[p->nOp-1].opcode==OP_Noop );
294 assert( p->aOp[p->nOp-1].p3!=0 );
295 assert( p->aOp[p->nOp-1].p3type==P3_DYNAMIC );
296 sqlite3SafetyOff(db);
297 db->xTrace(db->pTraceArg, p->aOp[p->nOp-1].p3);
298 if( sqlite3SafetyOn(db) ){
299 p->rc = SQLITE_MISUSE;
300 return SQLITE_MISUSE;
301 }
302 }
drh19e2d372005-08-29 23:00:03 +0000303 if( db->xProfile && !db->init.busy ){
304 double rNow;
danielk1977b4b47412007-08-17 15:53:36 +0000305 sqlite3OsCurrentTime(db->pVfs, &rNow);
drh19e2d372005-08-29 23:00:03 +0000306 p->startTime = (rNow - (int)rNow)*3600.0*24.0*1000000000.0;
307 }
308#endif
drhc16a03b2004-09-15 13:38:10 +0000309
310 /* Print a copy of SQL as it is executed if the SQL_TRACE pragma is turned
311 ** on in debugging mode.
312 */
313#ifdef SQLITE_DEBUG
314 if( (db->flags & SQLITE_SqlTrace)!=0 ){
315 sqlite3DebugPrintf("SQL-trace: %s\n", p->aOp[p->nOp-1].p3);
316 }
317#endif /* SQLITE_DEBUG */
318
danielk19771d850a72004-05-31 08:26:49 +0000319 db->activeVdbeCnt++;
320 p->pc = 0;
321 }
drhb7f91642004-10-31 02:22:47 +0000322#ifndef SQLITE_OMIT_EXPLAIN
drh4f26d6c2004-05-26 23:25:30 +0000323 if( p->explain ){
324 rc = sqlite3VdbeList(p);
drhb7f91642004-10-31 02:22:47 +0000325 }else
326#endif /* SQLITE_OMIT_EXPLAIN */
327 {
drh4f26d6c2004-05-26 23:25:30 +0000328 rc = sqlite3VdbeExec(p);
329 }
330
331 if( sqlite3SafetyOff(db) ){
332 rc = SQLITE_MISUSE;
333 }
334
drh19e2d372005-08-29 23:00:03 +0000335#ifndef SQLITE_OMIT_TRACE
336 /* Invoke the profile callback if there is one
337 */
338 if( rc!=SQLITE_ROW && db->xProfile && !db->init.busy ){
339 double rNow;
340 u64 elapseTime;
341
danielk1977b4b47412007-08-17 15:53:36 +0000342 sqlite3OsCurrentTime(db->pVfs, &rNow);
drh19e2d372005-08-29 23:00:03 +0000343 elapseTime = (rNow - (int)rNow)*3600.0*24.0*1000000000.0 - p->startTime;
344 assert( p->nOp>0 );
345 assert( p->aOp[p->nOp-1].opcode==OP_Noop );
346 assert( p->aOp[p->nOp-1].p3!=0 );
347 assert( p->aOp[p->nOp-1].p3type==P3_DYNAMIC );
348 db->xProfile(db->pProfileArg, p->aOp[p->nOp-1].p3, elapseTime);
349 }
350#endif
351
danielk197797a227c2006-01-20 16:32:04 +0000352 sqlite3Error(p->db, rc, 0);
danielk197776e8d1a2006-01-18 18:22:43 +0000353 p->rc = sqlite3ApiExit(p->db, p->rc);
drhb900aaf2006-11-09 00:24:53 +0000354end_of_step:
drh4ac285a2006-09-15 07:28:50 +0000355 assert( (rc&0xff)==rc );
drhb900aaf2006-11-09 00:24:53 +0000356 if( p->zSql && (rc&0xff)<SQLITE_ROW ){
357 /* This behavior occurs if sqlite3_prepare_v2() was used to build
358 ** the prepared statement. Return error codes directly */
danielk1977612642d2007-07-12 13:18:05 +0000359 sqlite3Error(p->db, p->rc, 0);
drhb900aaf2006-11-09 00:24:53 +0000360 return p->rc;
361 }else{
362 /* This is for legacy sqlite3_prepare() builds and when the code
363 ** is SQLITE_ROW or SQLITE_DONE */
364 return rc;
365 }
366}
367
368/*
369** This is the top-level implementation of sqlite3_step(). Call
370** sqlite3Step() to do most of the work. If a schema error occurs,
371** call sqlite3Reprepare() and try again.
372*/
373#ifdef SQLITE_OMIT_PARSER
374int sqlite3_step(sqlite3_stmt *pStmt){
drhb21c8cd2007-08-21 19:33:56 +0000375 int rc;
376 Vdbe *v;
377 v = (Vdbe*)pStmt;
378 sqlite3_mutex_enter(v->db->mutex);
379 rc = sqlite3Step(v);
380 sqlite3_mutex_leave(v->db->mutex);
381 return rc;
drhb900aaf2006-11-09 00:24:53 +0000382}
383#else
384int sqlite3_step(sqlite3_stmt *pStmt){
385 int cnt = 0;
386 int rc;
387 Vdbe *v = (Vdbe*)pStmt;
drhb21c8cd2007-08-21 19:33:56 +0000388 sqlite3_mutex_enter(v->db->mutex);
drhb900aaf2006-11-09 00:24:53 +0000389 while( (rc = sqlite3Step(v))==SQLITE_SCHEMA
390 && cnt++ < 5
391 && sqlite3Reprepare(v) ){
392 sqlite3_reset(pStmt);
393 v->expired = 0;
394 }
drhb21c8cd2007-08-21 19:33:56 +0000395 sqlite3_mutex_leave(v->db->mutex);
danielk197776e8d1a2006-01-18 18:22:43 +0000396 return rc;
drh4f26d6c2004-05-26 23:25:30 +0000397}
drhb900aaf2006-11-09 00:24:53 +0000398#endif
drh4f26d6c2004-05-26 23:25:30 +0000399
400/*
drheb2e1762004-05-27 01:53:56 +0000401** Extract the user data from a sqlite3_context structure and return a
402** pointer to it.
403*/
404void *sqlite3_user_data(sqlite3_context *p){
405 assert( p && p->pFunc );
406 return p->pFunc->pUserData;
407}
408
409/*
drhb7481e72006-09-16 21:45:14 +0000410** The following is the implementation of an SQL function that always
411** fails with an error message stating that the function is used in the
412** wrong context. The sqlite3_overload_function() API might construct
413** SQL function that use this routine so that the functions will exist
414** for name resolution but are actually overloaded by the xFindFunction
415** method of virtual tables.
416*/
417void sqlite3InvalidFunction(
418 sqlite3_context *context, /* The function calling context */
419 int argc, /* Number of arguments to the function */
420 sqlite3_value **argv /* Value of each argument */
421){
422 const char *zName = context->pFunc->zName;
423 char *zErr;
danielk19771e536952007-08-16 10:09:01 +0000424 zErr = sqlite3MPrintf(0,
drhb7481e72006-09-16 21:45:14 +0000425 "unable to use function %s in the requested context", zName);
426 sqlite3_result_error(context, zErr, -1);
danielk19771e536952007-08-16 10:09:01 +0000427 sqlite3_free(zErr);
drhb7481e72006-09-16 21:45:14 +0000428}
429
430/*
drheb2e1762004-05-27 01:53:56 +0000431** Allocate or return the aggregate context for a user function. A new
432** context is allocated on the first call. Subsequent calls return the
433** same context that was returned on prior calls.
drheb2e1762004-05-27 01:53:56 +0000434*/
435void *sqlite3_aggregate_context(sqlite3_context *p, int nByte){
drhb21c8cd2007-08-21 19:33:56 +0000436 Mem *pMem;
drh825c6622005-09-08 19:01:05 +0000437 assert( p && p->pFunc && p->pFunc->xStep );
drhb21c8cd2007-08-21 19:33:56 +0000438 assert( sqlite3_mutex_held(p->s.db->mutex) );
439 pMem = p->pMem;
drha10a34b2005-09-07 22:09:48 +0000440 if( (pMem->flags & MEM_Agg)==0 ){
441 if( nByte==0 ){
442 assert( pMem->flags==MEM_Null );
443 pMem->z = 0;
drheb2e1762004-05-27 01:53:56 +0000444 }else{
drha10a34b2005-09-07 22:09:48 +0000445 pMem->flags = MEM_Agg;
danielk19771e536952007-08-16 10:09:01 +0000446 pMem->xDel = sqlite3_free;
drh3c024d62007-03-30 11:23:45 +0000447 pMem->u.pDef = p->pFunc;
drha10a34b2005-09-07 22:09:48 +0000448 if( nByte<=NBFS ){
449 pMem->z = pMem->zShort;
450 memset(pMem->z, 0, nByte);
451 }else{
drhb21c8cd2007-08-21 19:33:56 +0000452 pMem->z = sqlite3DbMallocZero(p->s.db, nByte);
drha10a34b2005-09-07 22:09:48 +0000453 }
drheb2e1762004-05-27 01:53:56 +0000454 }
455 }
drhabfcea22005-09-06 20:36:48 +0000456 return (void*)pMem->z;
drheb2e1762004-05-27 01:53:56 +0000457}
458
459/*
danielk1977682f68b2004-06-05 10:22:17 +0000460** Return the auxilary data pointer, if any, for the iArg'th argument to
461** the user-function defined by pCtx.
462*/
463void *sqlite3_get_auxdata(sqlite3_context *pCtx, int iArg){
drhb21c8cd2007-08-21 19:33:56 +0000464 VdbeFunc *pVdbeFunc;
465
466 assert( sqlite3_mutex_held(pCtx->s.db->mutex) );
467 pVdbeFunc = pCtx->pVdbeFunc;
danielk1977682f68b2004-06-05 10:22:17 +0000468 if( !pVdbeFunc || iArg>=pVdbeFunc->nAux || iArg<0 ){
469 return 0;
470 }
drhf92c7ff2004-06-19 15:40:23 +0000471 return pVdbeFunc->apAux[iArg].pAux;
danielk1977682f68b2004-06-05 10:22:17 +0000472}
473
474/*
475** Set the auxilary data pointer and delete function, for the iArg'th
476** argument to the user-function defined by pCtx. Any previous value is
477** deleted by calling the delete function specified when it was set.
478*/
479void sqlite3_set_auxdata(
480 sqlite3_context *pCtx,
481 int iArg,
482 void *pAux,
483 void (*xDelete)(void*)
484){
485 struct AuxData *pAuxData;
drhf92c7ff2004-06-19 15:40:23 +0000486 VdbeFunc *pVdbeFunc;
danielk1977e0fc5262007-07-26 06:50:05 +0000487 if( iArg<0 ) goto failed;
danielk1977682f68b2004-06-05 10:22:17 +0000488
drhb21c8cd2007-08-21 19:33:56 +0000489 assert( sqlite3_mutex_held(pCtx->s.db->mutex) );
drhf92c7ff2004-06-19 15:40:23 +0000490 pVdbeFunc = pCtx->pVdbeFunc;
491 if( !pVdbeFunc || pVdbeFunc->nAux<=iArg ){
danielk197731dad9d2007-08-16 11:36:15 +0000492 int nAux = (pVdbeFunc ? pVdbeFunc->nAux : 0);
drhf92c7ff2004-06-19 15:40:23 +0000493 int nMalloc = sizeof(VdbeFunc) + sizeof(struct AuxData)*iArg;
drh17435752007-08-16 04:30:38 +0000494 pVdbeFunc = sqlite3_realloc(pVdbeFunc, nMalloc);
495 if( !pVdbeFunc ){
drhb21c8cd2007-08-21 19:33:56 +0000496 pCtx->s.db->mallocFailed = 1;
drh17435752007-08-16 04:30:38 +0000497 goto failed;
498 }
drh53f733c2005-09-16 02:38:09 +0000499 pCtx->pVdbeFunc = pVdbeFunc;
danielk197731dad9d2007-08-16 11:36:15 +0000500 memset(&pVdbeFunc->apAux[nAux], 0, sizeof(struct AuxData)*(iArg+1-nAux));
drh998da3a2004-06-19 15:22:56 +0000501 pVdbeFunc->nAux = iArg+1;
502 pVdbeFunc->pFunc = pCtx->pFunc;
danielk1977682f68b2004-06-05 10:22:17 +0000503 }
504
drhf92c7ff2004-06-19 15:40:23 +0000505 pAuxData = &pVdbeFunc->apAux[iArg];
danielk1977682f68b2004-06-05 10:22:17 +0000506 if( pAuxData->pAux && pAuxData->xDelete ){
507 pAuxData->xDelete(pAuxData->pAux);
508 }
509 pAuxData->pAux = pAux;
510 pAuxData->xDelete = xDelete;
danielk1977e0fc5262007-07-26 06:50:05 +0000511 return;
512
513failed:
514 if( xDelete ){
515 xDelete(pAux);
516 }
danielk1977682f68b2004-06-05 10:22:17 +0000517}
518
519/*
drheb2e1762004-05-27 01:53:56 +0000520** Return the number of times the Step function of a aggregate has been
521** called.
522**
drhcf85a512006-02-09 18:35:29 +0000523** This function is deprecated. Do not use it for new code. It is
524** provide only to avoid breaking legacy code. New aggregate function
525** implementations should keep their own counts within their aggregate
526** context.
drheb2e1762004-05-27 01:53:56 +0000527*/
528int sqlite3_aggregate_count(sqlite3_context *p){
529 assert( p && p->pFunc && p->pFunc->xStep );
drhabfcea22005-09-06 20:36:48 +0000530 return p->pMem->n;
drheb2e1762004-05-27 01:53:56 +0000531}
532
533/*
drh4f26d6c2004-05-26 23:25:30 +0000534** Return the number of columns in the result set for the statement pStmt.
535*/
536int sqlite3_column_count(sqlite3_stmt *pStmt){
537 Vdbe *pVm = (Vdbe *)pStmt;
drh1bcdb0c2004-08-28 14:49:46 +0000538 return pVm ? pVm->nResColumn : 0;
drh4f26d6c2004-05-26 23:25:30 +0000539}
540
541/*
542** Return the number of values available from the current row of the
543** currently executing statement pStmt.
544*/
545int sqlite3_data_count(sqlite3_stmt *pStmt){
546 Vdbe *pVm = (Vdbe *)pStmt;
drh1bcdb0c2004-08-28 14:49:46 +0000547 if( pVm==0 || !pVm->resOnStack ) return 0;
drh4f26d6c2004-05-26 23:25:30 +0000548 return pVm->nResColumn;
549}
550
551
552/*
553** Check to see if column iCol of the given statement is valid. If
554** it is, return a pointer to the Mem for the value of that column.
555** If iCol is not valid, return a pointer to a Mem which has a value
556** of NULL.
557*/
558static Mem *columnMem(sqlite3_stmt *pStmt, int i){
drh32bc3f62007-08-21 20:25:39 +0000559 Vdbe *pVm;
560 int vals;
561 Mem *pOut;
562
563 pVm = (Vdbe *)pStmt;
564 if( pVm && pVm->resOnStack && i<pVm->nResColumn && i>=0 ){
565 sqlite3_mutex_enter(pVm->db->mutex);
566 vals = sqlite3_data_count(pStmt);
567 pOut = &pVm->pTos[(1-vals)+i];
568 }else{
drhb21c8cd2007-08-21 19:33:56 +0000569 static const Mem nullMem = {{0}, 0.0, 0, "", 0, MEM_Null, SQLITE_NULL };
drh27641702007-08-22 02:56:42 +0000570 if( pVm->db ){
571 sqlite3_mutex_enter(pVm->db->mutex);
572 sqlite3Error(pVm->db, SQLITE_RANGE, 0);
573 }
drh32bc3f62007-08-21 20:25:39 +0000574 pOut = (Mem*)&nullMem;
drh4f26d6c2004-05-26 23:25:30 +0000575 }
drh32bc3f62007-08-21 20:25:39 +0000576 return pOut;
drh4f26d6c2004-05-26 23:25:30 +0000577}
578
danielk19772e588c72005-12-09 14:25:08 +0000579/*
580** This function is called after invoking an sqlite3_value_XXX function on a
581** column value (i.e. a value returned by evaluating an SQL expression in the
582** select list of a SELECT statement) that may cause a malloc() failure. If
583** malloc() has failed, the threads mallocFailed flag is cleared and the result
584** code of statement pStmt set to SQLITE_NOMEM.
585**
drh32bc3f62007-08-21 20:25:39 +0000586** Specifically, this is called from within:
danielk19772e588c72005-12-09 14:25:08 +0000587**
588** sqlite3_column_int()
589** sqlite3_column_int64()
590** sqlite3_column_text()
591** sqlite3_column_text16()
592** sqlite3_column_real()
593** sqlite3_column_bytes()
594** sqlite3_column_bytes16()
595**
596** But not for sqlite3_column_blob(), which never calls malloc().
597*/
598static void columnMallocFailure(sqlite3_stmt *pStmt)
599{
600 /* If malloc() failed during an encoding conversion within an
601 ** sqlite3_column_XXX API, then set the return code of the statement to
602 ** SQLITE_NOMEM. The next call to _step() (if any) will return SQLITE_ERROR
603 ** and _finalize() will return NOMEM.
604 */
danielk197754f01982006-01-18 15:25:17 +0000605 Vdbe *p = (Vdbe *)pStmt;
drh32bc3f62007-08-21 20:25:39 +0000606 if( p ){
607 p->rc = sqlite3ApiExit(p->db, p->rc);
608 sqlite3_mutex_leave(p->db->mutex);
609 }
danielk19772e588c72005-12-09 14:25:08 +0000610}
611
drh4f26d6c2004-05-26 23:25:30 +0000612/**************************** sqlite3_column_ *******************************
613** The following routines are used to access elements of the current row
614** in the result set.
615*/
danielk1977c572ef72004-05-27 09:28:41 +0000616const void *sqlite3_column_blob(sqlite3_stmt *pStmt, int i){
danielk19772e588c72005-12-09 14:25:08 +0000617 const void *val;
danielk19772e588c72005-12-09 14:25:08 +0000618 val = sqlite3_value_blob( columnMem(pStmt,i) );
danielk1977c9cf9012007-05-30 10:36:47 +0000619 /* Even though there is no encoding conversion, value_blob() might
620 ** need to call malloc() to expand the result of a zeroblob()
621 ** expression.
622 */
623 columnMallocFailure(pStmt);
danielk19772e588c72005-12-09 14:25:08 +0000624 return val;
danielk1977c572ef72004-05-27 09:28:41 +0000625}
drh4f26d6c2004-05-26 23:25:30 +0000626int sqlite3_column_bytes(sqlite3_stmt *pStmt, int i){
danielk19772e588c72005-12-09 14:25:08 +0000627 int val = sqlite3_value_bytes( columnMem(pStmt,i) );
628 columnMallocFailure(pStmt);
629 return val;
drh4f26d6c2004-05-26 23:25:30 +0000630}
631int sqlite3_column_bytes16(sqlite3_stmt *pStmt, int i){
danielk19772e588c72005-12-09 14:25:08 +0000632 int val = sqlite3_value_bytes16( columnMem(pStmt,i) );
633 columnMallocFailure(pStmt);
634 return val;
drh4f26d6c2004-05-26 23:25:30 +0000635}
636double sqlite3_column_double(sqlite3_stmt *pStmt, int i){
danielk19772e588c72005-12-09 14:25:08 +0000637 double val = sqlite3_value_double( columnMem(pStmt,i) );
638 columnMallocFailure(pStmt);
639 return val;
drh4f26d6c2004-05-26 23:25:30 +0000640}
641int sqlite3_column_int(sqlite3_stmt *pStmt, int i){
danielk19772e588c72005-12-09 14:25:08 +0000642 int val = sqlite3_value_int( columnMem(pStmt,i) );
643 columnMallocFailure(pStmt);
644 return val;
drh4f26d6c2004-05-26 23:25:30 +0000645}
drhefad9992004-06-22 12:13:55 +0000646sqlite_int64 sqlite3_column_int64(sqlite3_stmt *pStmt, int i){
danielk19772e588c72005-12-09 14:25:08 +0000647 sqlite_int64 val = sqlite3_value_int64( columnMem(pStmt,i) );
648 columnMallocFailure(pStmt);
649 return val;
drh4f26d6c2004-05-26 23:25:30 +0000650}
651const unsigned char *sqlite3_column_text(sqlite3_stmt *pStmt, int i){
danielk19772e588c72005-12-09 14:25:08 +0000652 const unsigned char *val = sqlite3_value_text( columnMem(pStmt,i) );
653 columnMallocFailure(pStmt);
654 return val;
drh4f26d6c2004-05-26 23:25:30 +0000655}
drhd1e47332005-06-26 17:55:33 +0000656sqlite3_value *sqlite3_column_value(sqlite3_stmt *pStmt, int i){
drh32bc3f62007-08-21 20:25:39 +0000657 sqlite3_value *pOut = columnMem(pStmt, i);
658 columnMallocFailure(pStmt);
659 return pOut;
drhd1e47332005-06-26 17:55:33 +0000660}
drh6c626082004-11-14 21:56:29 +0000661#ifndef SQLITE_OMIT_UTF16
drh4f26d6c2004-05-26 23:25:30 +0000662const void *sqlite3_column_text16(sqlite3_stmt *pStmt, int i){
danielk19772e588c72005-12-09 14:25:08 +0000663 const void *val = sqlite3_value_text16( columnMem(pStmt,i) );
664 columnMallocFailure(pStmt);
665 return val;
drh4f26d6c2004-05-26 23:25:30 +0000666}
drh6c626082004-11-14 21:56:29 +0000667#endif /* SQLITE_OMIT_UTF16 */
drh4f26d6c2004-05-26 23:25:30 +0000668int sqlite3_column_type(sqlite3_stmt *pStmt, int i){
drh32bc3f62007-08-21 20:25:39 +0000669 int iType = sqlite3_value_type( columnMem(pStmt,i) );
670 columnMallocFailure(pStmt);
671 return iType;
drh4f26d6c2004-05-26 23:25:30 +0000672}
673
drh29d72102006-02-09 22:13:41 +0000674/* The following function is experimental and subject to change or
675** removal */
676/*int sqlite3_column_numeric_type(sqlite3_stmt *pStmt, int i){
677** return sqlite3_value_numeric_type( columnMem(pStmt,i) );
678**}
679*/
680
drh5e2517e2004-09-24 23:59:12 +0000681/*
682** Convert the N-th element of pStmt->pColName[] into a string using
683** xFunc() then return that string. If N is out of range, return 0.
drhe590fbd2005-05-20 19:36:01 +0000684**
685** There are up to 5 names for each column. useType determines which
686** name is returned. Here are the names:
687**
688** 0 The column name as it should be displayed for output
689** 1 The datatype name for the column
690** 2 The name of the database that the column derives from
691** 3 The name of the table that the column derives from
692** 4 The name of the table column that the result column derives from
693**
694** If the result is not a simple column reference (if it is an expression
695** or a constant) then useTypes 2, 3, and 4 return NULL.
drh5e2517e2004-09-24 23:59:12 +0000696*/
697static const void *columnName(
698 sqlite3_stmt *pStmt,
699 int N,
700 const void *(*xFunc)(Mem*),
701 int useType
702){
drh32bc3f62007-08-21 20:25:39 +0000703 const void *ret = 0;
drh5e2517e2004-09-24 23:59:12 +0000704 Vdbe *p = (Vdbe *)pStmt;
drh32bc3f62007-08-21 20:25:39 +0000705 int n;
706
drh5e2517e2004-09-24 23:59:12 +0000707
drh32bc3f62007-08-21 20:25:39 +0000708 if( p!=0 ){
709 n = sqlite3_column_count(pStmt);
710 if( N<n && N>=0 ){
711 N += useType*n;
drh153c62c2007-08-24 03:51:33 +0000712 sqlite3_mutex_enter(p->db->mutex);
drh32bc3f62007-08-21 20:25:39 +0000713 ret = xFunc(&p->aColName[N]);
714
drh32bc3f62007-08-21 20:25:39 +0000715 /* A malloc may have failed inside of the xFunc() call. If this
716 ** is the case, clear the mallocFailed flag and return NULL.
717 */
718 if( p->db && p->db->mallocFailed ){
719 p->db->mallocFailed = 0;
720 ret = 0;
721 }
drh153c62c2007-08-24 03:51:33 +0000722 sqlite3_mutex_leave(p->db->mutex);
drh32bc3f62007-08-21 20:25:39 +0000723 }
drh5e2517e2004-09-24 23:59:12 +0000724 }
danielk197700fd9572005-12-07 06:27:43 +0000725 return ret;
drh5e2517e2004-09-24 23:59:12 +0000726}
727
drh4f26d6c2004-05-26 23:25:30 +0000728/*
729** Return the name of the Nth column of the result set returned by SQL
730** statement pStmt.
731*/
732const char *sqlite3_column_name(sqlite3_stmt *pStmt, int N){
danielk1977955de522006-02-10 02:27:42 +0000733 return columnName(
734 pStmt, N, (const void*(*)(Mem*))sqlite3_value_text, COLNAME_NAME);
drh4f26d6c2004-05-26 23:25:30 +0000735}
drhe590fbd2005-05-20 19:36:01 +0000736#ifndef SQLITE_OMIT_UTF16
737const void *sqlite3_column_name16(sqlite3_stmt *pStmt, int N){
danielk1977955de522006-02-10 02:27:42 +0000738 return columnName(
739 pStmt, N, (const void*(*)(Mem*))sqlite3_value_text16, COLNAME_NAME);
drhe590fbd2005-05-20 19:36:01 +0000740}
741#endif
drh4f26d6c2004-05-26 23:25:30 +0000742
743/*
drh6c626082004-11-14 21:56:29 +0000744** Return the column declaration type (if applicable) of the 'i'th column
drhe590fbd2005-05-20 19:36:01 +0000745** of the result set of SQL statement pStmt.
drh6c626082004-11-14 21:56:29 +0000746*/
747const char *sqlite3_column_decltype(sqlite3_stmt *pStmt, int N){
danielk1977955de522006-02-10 02:27:42 +0000748 return columnName(
749 pStmt, N, (const void*(*)(Mem*))sqlite3_value_text, COLNAME_DECLTYPE);
drh6c626082004-11-14 21:56:29 +0000750}
drh6c626082004-11-14 21:56:29 +0000751#ifndef SQLITE_OMIT_UTF16
danielk197776d505b2004-05-28 13:13:02 +0000752const void *sqlite3_column_decltype16(sqlite3_stmt *pStmt, int N){
danielk1977955de522006-02-10 02:27:42 +0000753 return columnName(
754 pStmt, N, (const void*(*)(Mem*))sqlite3_value_text16, COLNAME_DECLTYPE);
drh4f26d6c2004-05-26 23:25:30 +0000755}
drh6c626082004-11-14 21:56:29 +0000756#endif /* SQLITE_OMIT_UTF16 */
drh4f26d6c2004-05-26 23:25:30 +0000757
danielk19774b1ae992006-02-10 03:06:10 +0000758#ifdef SQLITE_ENABLE_COLUMN_METADATA
drhe590fbd2005-05-20 19:36:01 +0000759/*
760** Return the name of the database from which a result column derives.
761** NULL is returned if the result column is an expression or constant or
762** anything else which is not an unabiguous reference to a database column.
763*/
764const char *sqlite3_column_database_name(sqlite3_stmt *pStmt, int N){
danielk1977955de522006-02-10 02:27:42 +0000765 return columnName(
766 pStmt, N, (const void*(*)(Mem*))sqlite3_value_text, COLNAME_DATABASE);
drhe590fbd2005-05-20 19:36:01 +0000767}
768#ifndef SQLITE_OMIT_UTF16
769const void *sqlite3_column_database_name16(sqlite3_stmt *pStmt, int N){
danielk1977955de522006-02-10 02:27:42 +0000770 return columnName(
771 pStmt, N, (const void*(*)(Mem*))sqlite3_value_text16, COLNAME_DATABASE);
drhe590fbd2005-05-20 19:36:01 +0000772}
773#endif /* SQLITE_OMIT_UTF16 */
774
775/*
776** Return the name of the table from which a result column derives.
777** NULL is returned if the result column is an expression or constant or
778** anything else which is not an unabiguous reference to a database column.
779*/
780const char *sqlite3_column_table_name(sqlite3_stmt *pStmt, int N){
danielk1977955de522006-02-10 02:27:42 +0000781 return columnName(
782 pStmt, N, (const void*(*)(Mem*))sqlite3_value_text, COLNAME_TABLE);
drhe590fbd2005-05-20 19:36:01 +0000783}
784#ifndef SQLITE_OMIT_UTF16
785const void *sqlite3_column_table_name16(sqlite3_stmt *pStmt, int N){
danielk1977955de522006-02-10 02:27:42 +0000786 return columnName(
787 pStmt, N, (const void*(*)(Mem*))sqlite3_value_text16, COLNAME_TABLE);
drhe590fbd2005-05-20 19:36:01 +0000788}
789#endif /* SQLITE_OMIT_UTF16 */
790
791/*
792** Return the name of the table column from which a result column derives.
793** NULL is returned if the result column is an expression or constant or
794** anything else which is not an unabiguous reference to a database column.
795*/
796const char *sqlite3_column_origin_name(sqlite3_stmt *pStmt, int N){
danielk1977955de522006-02-10 02:27:42 +0000797 return columnName(
798 pStmt, N, (const void*(*)(Mem*))sqlite3_value_text, COLNAME_COLUMN);
drhe590fbd2005-05-20 19:36:01 +0000799}
800#ifndef SQLITE_OMIT_UTF16
801const void *sqlite3_column_origin_name16(sqlite3_stmt *pStmt, int N){
danielk1977955de522006-02-10 02:27:42 +0000802 return columnName(
803 pStmt, N, (const void*(*)(Mem*))sqlite3_value_text16, COLNAME_COLUMN);
drhe590fbd2005-05-20 19:36:01 +0000804}
805#endif /* SQLITE_OMIT_UTF16 */
danielk19774b1ae992006-02-10 03:06:10 +0000806#endif /* SQLITE_ENABLE_COLUMN_METADATA */
drhe590fbd2005-05-20 19:36:01 +0000807
808
drh4f26d6c2004-05-26 23:25:30 +0000809/******************************* sqlite3_bind_ ***************************
810**
811** Routines used to attach values to wildcards in a compiled SQL statement.
812*/
813/*
814** Unbind the value bound to variable i in virtual machine p. This is the
815** the same as binding a NULL value to the column. If the "i" parameter is
816** out of range, then SQLITE_RANGE is returned. Othewise SQLITE_OK.
817**
818** The error code stored in database p->db is overwritten with the return
819** value in any case.
820*/
821static int vdbeUnbind(Vdbe *p, int i){
822 Mem *pVar;
drh1bcdb0c2004-08-28 14:49:46 +0000823 if( p==0 || p->magic!=VDBE_MAGIC_RUN || p->pc>=0 ){
drh473d1792005-06-06 17:54:55 +0000824 if( p ) sqlite3Error(p->db, SQLITE_MISUSE, 0);
drh4f26d6c2004-05-26 23:25:30 +0000825 return SQLITE_MISUSE;
826 }
827 if( i<1 || i>p->nVar ){
828 sqlite3Error(p->db, SQLITE_RANGE, 0);
829 return SQLITE_RANGE;
830 }
831 i--;
drh290c1942004-08-21 17:54:45 +0000832 pVar = &p->aVar[i];
danielk1977d8123362004-06-12 09:25:12 +0000833 sqlite3VdbeMemRelease(pVar);
drh4f26d6c2004-05-26 23:25:30 +0000834 pVar->flags = MEM_Null;
835 sqlite3Error(p->db, SQLITE_OK, 0);
836 return SQLITE_OK;
837}
838
839/*
drh5e2517e2004-09-24 23:59:12 +0000840** Bind a text or BLOB value.
drh4f26d6c2004-05-26 23:25:30 +0000841*/
drh5e2517e2004-09-24 23:59:12 +0000842static int bindText(
drh605264d2007-08-21 15:13:19 +0000843 sqlite3_stmt *pStmt, /* The statement to bind against */
844 int i, /* Index of the parameter to bind */
845 const void *zData, /* Pointer to the data to be bound */
846 int nData, /* Number of bytes of data to be bound */
847 void (*xDel)(void*), /* Destructor for the data */
848 int encoding /* Encoding for the data */
drh4f26d6c2004-05-26 23:25:30 +0000849){
850 Vdbe *p = (Vdbe *)pStmt;
851 Mem *pVar;
852 int rc;
853
drh605264d2007-08-21 15:13:19 +0000854 if( p==0 ){
855 return SQLITE_MISUSE;
856 }
857 sqlite3_mutex_enter(p->db->mutex);
drh4f26d6c2004-05-26 23:25:30 +0000858 rc = vdbeUnbind(p, i);
drh27641702007-08-22 02:56:42 +0000859 if( rc==SQLITE_OK && zData!=0 ){
860 pVar = &p->aVar[i-1];
861 rc = sqlite3VdbeMemSetStr(pVar, zData, nData, encoding, xDel);
862 if( rc==SQLITE_OK && encoding!=0 ){
863 rc = sqlite3VdbeChangeEncoding(pVar, ENC(p->db));
864 }
865 sqlite3Error(p->db, rc, 0);
866 rc = sqlite3ApiExit(p->db, rc);
drh4f26d6c2004-05-26 23:25:30 +0000867 }
drh605264d2007-08-21 15:13:19 +0000868 sqlite3_mutex_leave(p->db->mutex);
869 return rc;
drh4f26d6c2004-05-26 23:25:30 +0000870}
drh5e2517e2004-09-24 23:59:12 +0000871
872
873/*
874** Bind a blob value to an SQL statement variable.
875*/
876int sqlite3_bind_blob(
877 sqlite3_stmt *pStmt,
878 int i,
879 const void *zData,
880 int nData,
881 void (*xDel)(void*)
882){
883 return bindText(pStmt, i, zData, nData, xDel, 0);
884}
drh4f26d6c2004-05-26 23:25:30 +0000885int sqlite3_bind_double(sqlite3_stmt *pStmt, int i, double rValue){
886 int rc;
887 Vdbe *p = (Vdbe *)pStmt;
drh605264d2007-08-21 15:13:19 +0000888 sqlite3_mutex_enter(p->db->mutex);
drh4f26d6c2004-05-26 23:25:30 +0000889 rc = vdbeUnbind(p, i);
890 if( rc==SQLITE_OK ){
drh290c1942004-08-21 17:54:45 +0000891 sqlite3VdbeMemSetDouble(&p->aVar[i-1], rValue);
drh4f26d6c2004-05-26 23:25:30 +0000892 }
drh605264d2007-08-21 15:13:19 +0000893 sqlite3_mutex_leave(p->db->mutex);
danielk1977f4618892004-06-28 13:09:11 +0000894 return rc;
drh4f26d6c2004-05-26 23:25:30 +0000895}
896int sqlite3_bind_int(sqlite3_stmt *p, int i, int iValue){
drhefad9992004-06-22 12:13:55 +0000897 return sqlite3_bind_int64(p, i, (i64)iValue);
drh4f26d6c2004-05-26 23:25:30 +0000898}
drhefad9992004-06-22 12:13:55 +0000899int sqlite3_bind_int64(sqlite3_stmt *pStmt, int i, sqlite_int64 iValue){
drh4f26d6c2004-05-26 23:25:30 +0000900 int rc;
901 Vdbe *p = (Vdbe *)pStmt;
drh605264d2007-08-21 15:13:19 +0000902 sqlite3_mutex_enter(p->db->mutex);
drh4f26d6c2004-05-26 23:25:30 +0000903 rc = vdbeUnbind(p, i);
904 if( rc==SQLITE_OK ){
drh290c1942004-08-21 17:54:45 +0000905 sqlite3VdbeMemSetInt64(&p->aVar[i-1], iValue);
drh4f26d6c2004-05-26 23:25:30 +0000906 }
drh605264d2007-08-21 15:13:19 +0000907 sqlite3_mutex_leave(p->db->mutex);
drh4f26d6c2004-05-26 23:25:30 +0000908 return rc;
909}
drh605264d2007-08-21 15:13:19 +0000910int sqlite3_bind_null(sqlite3_stmt *pStmt, int i){
911 int rc;
912 Vdbe *p = (Vdbe*)pStmt;
913 sqlite3_mutex_enter(p->db->mutex);
914 rc = vdbeUnbind(p, i);
915 sqlite3_mutex_leave(p->db->mutex);
916 return rc;
drh4f26d6c2004-05-26 23:25:30 +0000917}
918int sqlite3_bind_text(
919 sqlite3_stmt *pStmt,
920 int i,
921 const char *zData,
922 int nData,
danielk1977d8123362004-06-12 09:25:12 +0000923 void (*xDel)(void*)
drh4f26d6c2004-05-26 23:25:30 +0000924){
drh5e2517e2004-09-24 23:59:12 +0000925 return bindText(pStmt, i, zData, nData, xDel, SQLITE_UTF8);
drh4f26d6c2004-05-26 23:25:30 +0000926}
drh6c626082004-11-14 21:56:29 +0000927#ifndef SQLITE_OMIT_UTF16
drh4f26d6c2004-05-26 23:25:30 +0000928int sqlite3_bind_text16(
929 sqlite3_stmt *pStmt,
930 int i,
931 const void *zData,
932 int nData,
danielk1977d8123362004-06-12 09:25:12 +0000933 void (*xDel)(void*)
drh4f26d6c2004-05-26 23:25:30 +0000934){
drh5e2517e2004-09-24 23:59:12 +0000935 return bindText(pStmt, i, zData, nData, xDel, SQLITE_UTF16NATIVE);
drh4f26d6c2004-05-26 23:25:30 +0000936}
drh6c626082004-11-14 21:56:29 +0000937#endif /* SQLITE_OMIT_UTF16 */
danielk197726e41442006-06-14 15:16:35 +0000938int sqlite3_bind_value(sqlite3_stmt *pStmt, int i, const sqlite3_value *pValue){
939 int rc;
940 Vdbe *p = (Vdbe *)pStmt;
drh605264d2007-08-21 15:13:19 +0000941 sqlite3_mutex_enter(p->db->mutex);
danielk197726e41442006-06-14 15:16:35 +0000942 rc = vdbeUnbind(p, i);
943 if( rc==SQLITE_OK ){
drhb21c8cd2007-08-21 19:33:56 +0000944 rc = sqlite3VdbeMemCopy(&p->aVar[i-1], pValue);
danielk197726e41442006-06-14 15:16:35 +0000945 }
drh605264d2007-08-21 15:13:19 +0000946 sqlite3_mutex_leave(p->db->mutex);
danielk197726e41442006-06-14 15:16:35 +0000947 return rc;
948}
drhb026e052007-05-02 01:34:31 +0000949int sqlite3_bind_zeroblob(sqlite3_stmt *pStmt, int i, int n){
950 int rc;
951 Vdbe *p = (Vdbe *)pStmt;
drh605264d2007-08-21 15:13:19 +0000952 sqlite3_mutex_enter(p->db->mutex);
drhb026e052007-05-02 01:34:31 +0000953 rc = vdbeUnbind(p, i);
954 if( rc==SQLITE_OK ){
955 sqlite3VdbeMemSetZeroBlob(&p->aVar[i-1], n);
956 }
drh605264d2007-08-21 15:13:19 +0000957 sqlite3_mutex_leave(p->db->mutex);
drhb026e052007-05-02 01:34:31 +0000958 return rc;
959}
drh75f6a032004-07-15 14:15:00 +0000960
961/*
962** Return the number of wildcards that can be potentially bound to.
963** This routine is added to support DBD::SQLite.
drh75f6a032004-07-15 14:15:00 +0000964*/
965int sqlite3_bind_parameter_count(sqlite3_stmt *pStmt){
drh92febd92004-08-20 18:34:20 +0000966 Vdbe *p = (Vdbe*)pStmt;
967 return p ? p->nVar : 0;
drh75f6a032004-07-15 14:15:00 +0000968}
drh895d7472004-08-20 16:02:39 +0000969
970/*
drhfa6bc002004-09-07 16:19:52 +0000971** Create a mapping from variable numbers to variable names
972** in the Vdbe.azVar[] array, if such a mapping does not already
973** exist.
drh895d7472004-08-20 16:02:39 +0000974*/
drhfa6bc002004-09-07 16:19:52 +0000975static void createVarMap(Vdbe *p){
drh895d7472004-08-20 16:02:39 +0000976 if( !p->okVar ){
drh605264d2007-08-21 15:13:19 +0000977 sqlite3_mutex_enter(p->db->mutex);
978 if( !p->okVar ){
979 int j;
980 Op *pOp;
981 for(j=0, pOp=p->aOp; j<p->nOp; j++, pOp++){
982 if( pOp->opcode==OP_Variable ){
983 assert( pOp->p1>0 && pOp->p1<=p->nVar );
984 p->azVar[pOp->p1-1] = pOp->p3;
985 }
drh895d7472004-08-20 16:02:39 +0000986 }
drh605264d2007-08-21 15:13:19 +0000987 p->okVar = 1;
drh895d7472004-08-20 16:02:39 +0000988 }
drh605264d2007-08-21 15:13:19 +0000989 sqlite3_mutex_leave(p->db->mutex);
drh895d7472004-08-20 16:02:39 +0000990 }
drhfa6bc002004-09-07 16:19:52 +0000991}
992
993/*
994** Return the name of a wildcard parameter. Return NULL if the index
995** is out of range or if the wildcard is unnamed.
996**
997** The result is always UTF-8.
998*/
999const char *sqlite3_bind_parameter_name(sqlite3_stmt *pStmt, int i){
1000 Vdbe *p = (Vdbe*)pStmt;
1001 if( p==0 || i<1 || i>p->nVar ){
1002 return 0;
1003 }
1004 createVarMap(p);
drh895d7472004-08-20 16:02:39 +00001005 return p->azVar[i-1];
1006}
drhfa6bc002004-09-07 16:19:52 +00001007
1008/*
1009** Given a wildcard parameter name, return the index of the variable
1010** with that name. If there is no variable with the given name,
1011** return 0.
1012*/
1013int sqlite3_bind_parameter_index(sqlite3_stmt *pStmt, const char *zName){
1014 Vdbe *p = (Vdbe*)pStmt;
1015 int i;
1016 if( p==0 ){
1017 return 0;
1018 }
1019 createVarMap(p);
drh6a8903c2004-12-10 18:00:04 +00001020 if( zName ){
1021 for(i=0; i<p->nVar; i++){
1022 const char *z = p->azVar[i];
1023 if( z && strcmp(z,zName)==0 ){
1024 return i+1;
1025 }
drhfa6bc002004-09-07 16:19:52 +00001026 }
1027 }
1028 return 0;
1029}
drhf8db1bc2005-04-22 02:38:37 +00001030
drh51942bc2005-06-12 22:01:42 +00001031/*
drhf8db1bc2005-04-22 02:38:37 +00001032** Transfer all bindings from the first statement over to the second.
1033** If the two statements contain a different number of bindings, then
1034** an SQLITE_ERROR is returned.
1035*/
1036int sqlite3_transfer_bindings(sqlite3_stmt *pFromStmt, sqlite3_stmt *pToStmt){
1037 Vdbe *pFrom = (Vdbe*)pFromStmt;
1038 Vdbe *pTo = (Vdbe*)pToStmt;
1039 int i, rc = SQLITE_OK;
1040 if( (pFrom->magic!=VDBE_MAGIC_RUN && pFrom->magic!=VDBE_MAGIC_HALT)
drh27641702007-08-22 02:56:42 +00001041 || (pTo->magic!=VDBE_MAGIC_RUN && pTo->magic!=VDBE_MAGIC_HALT)
1042 || pTo->db!=pFrom->db ){
drhf8db1bc2005-04-22 02:38:37 +00001043 return SQLITE_MISUSE;
1044 }
1045 if( pFrom->nVar!=pTo->nVar ){
1046 return SQLITE_ERROR;
1047 }
drh27641702007-08-22 02:56:42 +00001048 sqlite3_mutex_enter(pTo->db->mutex);
drhf8db1bc2005-04-22 02:38:37 +00001049 for(i=0; rc==SQLITE_OK && i<pFrom->nVar; i++){
danielk1977950f0542006-01-18 05:51:57 +00001050 sqlite3MallocDisallow();
drhb21c8cd2007-08-21 19:33:56 +00001051 rc = sqlite3VdbeMemMove(&pTo->aVar[i], &pFrom->aVar[i]);
danielk1977950f0542006-01-18 05:51:57 +00001052 sqlite3MallocAllow();
drhf8db1bc2005-04-22 02:38:37 +00001053 }
drh27641702007-08-22 02:56:42 +00001054 sqlite3_mutex_leave(pTo->db->mutex);
drh4ac285a2006-09-15 07:28:50 +00001055 assert( rc==SQLITE_OK || rc==SQLITE_NOMEM );
drhf8db1bc2005-04-22 02:38:37 +00001056 return rc;
1057}
drh51942bc2005-06-12 22:01:42 +00001058
1059/*
1060** Return the sqlite3* database handle to which the prepared statement given
1061** in the argument belongs. This is the same database handle that was
1062** the first argument to the sqlite3_prepare() that was used to create
1063** the statement in the first place.
1064*/
1065sqlite3 *sqlite3_db_handle(sqlite3_stmt *pStmt){
1066 return pStmt ? ((Vdbe*)pStmt)->db : 0;
1067}