blob: 38997fabc989ec317aa0ec56a038c250853eee7a [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;
drh7e8b8482008-01-23 03:03:05 +000047#ifndef SQLITE_MUTEX_NOOP
drh86f8c192007-08-22 00:39:19 +000048 sqlite3_mutex *mutex = v->db->mutex;
drh7e8b8482008-01-23 03:03:05 +000049#endif
drh86f8c192007-08-22 00:39:19 +000050 sqlite3_mutex_enter(mutex);
drhe30f4422007-08-21 16:15:55 +000051 rc = sqlite3VdbeFinalize(v);
drh86f8c192007-08-22 00:39:19 +000052 sqlite3_mutex_leave(mutex);
drhe30f4422007-08-21 16:15:55 +000053 }
54 return rc;
55}
56
57/*
58** Terminate the current execution of an SQL statement and reset it
59** back to its starting state so that it can be reused. A success code from
60** the prior execution is returned.
61**
62** This routine sets the error code and string returned by
63** sqlite3_errcode(), sqlite3_errmsg() and sqlite3_errmsg16().
64*/
65int sqlite3_reset(sqlite3_stmt *pStmt){
66 int rc;
67 if( pStmt==0 ){
68 rc = SQLITE_OK;
69 }else{
70 Vdbe *v = (Vdbe*)pStmt;
71 sqlite3_mutex_enter(v->db->mutex);
72 rc = sqlite3VdbeReset(v);
73 sqlite3VdbeMakeReady(v, -1, 0, 0, 0);
74 assert( (rc & (v->db->errMask))==rc );
75 sqlite3_mutex_leave(v->db->mutex);
76 }
77 return rc;
78}
79
80/*
81** Set all the parameters in the compiled SQL statement to NULL.
82*/
83int sqlite3_clear_bindings(sqlite3_stmt *pStmt){
84 int i;
85 int rc = SQLITE_OK;
drh7e8b8482008-01-23 03:03:05 +000086#ifndef SQLITE_MUTEX_NOOP
87 sqlite3_mutex *mutex = ((Vdbe*)pStmt)->db->mutex;
88#endif
89 sqlite3_mutex_enter(mutex);
drhe30f4422007-08-21 16:15:55 +000090 for(i=1; rc==SQLITE_OK && i<=sqlite3_bind_parameter_count(pStmt); i++){
91 rc = sqlite3_bind_null(pStmt, i);
92 }
drh7e8b8482008-01-23 03:03:05 +000093 sqlite3_mutex_leave(mutex);
drhe30f4422007-08-21 16:15:55 +000094 return rc;
95}
96
97
drh4f26d6c2004-05-26 23:25:30 +000098/**************************** sqlite3_value_ *******************************
99** The following routines extract information from a Mem or sqlite3_value
100** structure.
101*/
102const void *sqlite3_value_blob(sqlite3_value *pVal){
103 Mem *p = (Mem*)pVal;
104 if( p->flags & (MEM_Blob|MEM_Str) ){
drhb21c8cd2007-08-21 19:33:56 +0000105 sqlite3VdbeMemExpandBlob(p);
drh1f0feef2007-05-15 13:27:07 +0000106 p->flags &= ~MEM_Str;
107 p->flags |= MEM_Blob;
drh4f26d6c2004-05-26 23:25:30 +0000108 return p->z;
109 }else{
110 return sqlite3_value_text(pVal);
111 }
112}
113int sqlite3_value_bytes(sqlite3_value *pVal){
drhb21c8cd2007-08-21 19:33:56 +0000114 return sqlite3ValueBytes(pVal, SQLITE_UTF8);
drh4f26d6c2004-05-26 23:25:30 +0000115}
116int sqlite3_value_bytes16(sqlite3_value *pVal){
drhb21c8cd2007-08-21 19:33:56 +0000117 return sqlite3ValueBytes(pVal, SQLITE_UTF16NATIVE);
drh4f26d6c2004-05-26 23:25:30 +0000118}
119double sqlite3_value_double(sqlite3_value *pVal){
drh6a6124e2004-06-27 01:56:33 +0000120 return sqlite3VdbeRealValue((Mem*)pVal);
drh4f26d6c2004-05-26 23:25:30 +0000121}
122int sqlite3_value_int(sqlite3_value *pVal){
drh6a6124e2004-06-27 01:56:33 +0000123 return sqlite3VdbeIntValue((Mem*)pVal);
drh4f26d6c2004-05-26 23:25:30 +0000124}
drhefad9992004-06-22 12:13:55 +0000125sqlite_int64 sqlite3_value_int64(sqlite3_value *pVal){
drh6a6124e2004-06-27 01:56:33 +0000126 return sqlite3VdbeIntValue((Mem*)pVal);
drh4f26d6c2004-05-26 23:25:30 +0000127}
128const unsigned char *sqlite3_value_text(sqlite3_value *pVal){
drhb21c8cd2007-08-21 19:33:56 +0000129 return (const unsigned char *)sqlite3ValueText(pVal, SQLITE_UTF8);
drh4f26d6c2004-05-26 23:25:30 +0000130}
drh6c626082004-11-14 21:56:29 +0000131#ifndef SQLITE_OMIT_UTF16
drh4f26d6c2004-05-26 23:25:30 +0000132const void *sqlite3_value_text16(sqlite3_value* pVal){
drhb21c8cd2007-08-21 19:33:56 +0000133 return sqlite3ValueText(pVal, SQLITE_UTF16NATIVE);
drh4f26d6c2004-05-26 23:25:30 +0000134}
danielk1977d8123362004-06-12 09:25:12 +0000135const void *sqlite3_value_text16be(sqlite3_value *pVal){
drhb21c8cd2007-08-21 19:33:56 +0000136 return sqlite3ValueText(pVal, SQLITE_UTF16BE);
danielk1977d8123362004-06-12 09:25:12 +0000137}
138const void *sqlite3_value_text16le(sqlite3_value *pVal){
drhb21c8cd2007-08-21 19:33:56 +0000139 return sqlite3ValueText(pVal, SQLITE_UTF16LE);
danielk1977d8123362004-06-12 09:25:12 +0000140}
drh6c626082004-11-14 21:56:29 +0000141#endif /* SQLITE_OMIT_UTF16 */
drh4f26d6c2004-05-26 23:25:30 +0000142int sqlite3_value_type(sqlite3_value* pVal){
drhf4479502004-05-27 03:12:53 +0000143 return pVal->type;
drh4f26d6c2004-05-26 23:25:30 +0000144}
145
146/**************************** sqlite3_result_ *******************************
147** The following routines are used by user-defined functions to specify
148** the function result.
149*/
150void sqlite3_result_blob(
151 sqlite3_context *pCtx,
152 const void *z,
153 int n,
danielk1977d8123362004-06-12 09:25:12 +0000154 void (*xDel)(void *)
drh4f26d6c2004-05-26 23:25:30 +0000155){
drh5708d2d2005-06-22 10:53:59 +0000156 assert( n>=0 );
drh32bc3f62007-08-21 20:25:39 +0000157 assert( sqlite3_mutex_held(pCtx->s.db->mutex) );
drhb21c8cd2007-08-21 19:33:56 +0000158 sqlite3VdbeMemSetStr(&pCtx->s, z, n, 0, xDel);
drh4f26d6c2004-05-26 23:25:30 +0000159}
160void sqlite3_result_double(sqlite3_context *pCtx, double rVal){
drh32bc3f62007-08-21 20:25:39 +0000161 assert( sqlite3_mutex_held(pCtx->s.db->mutex) );
drh4f26d6c2004-05-26 23:25:30 +0000162 sqlite3VdbeMemSetDouble(&pCtx->s, rVal);
163}
164void sqlite3_result_error(sqlite3_context *pCtx, const char *z, int n){
drh32bc3f62007-08-21 20:25:39 +0000165 assert( sqlite3_mutex_held(pCtx->s.db->mutex) );
drh69544ec2008-02-06 14:11:34 +0000166 pCtx->isError = SQLITE_ERROR;
drhb21c8cd2007-08-21 19:33:56 +0000167 sqlite3VdbeMemSetStr(&pCtx->s, z, n, SQLITE_UTF8, SQLITE_TRANSIENT);
drh4f26d6c2004-05-26 23:25:30 +0000168}
danielk1977a1686c92006-01-23 07:52:37 +0000169#ifndef SQLITE_OMIT_UTF16
drh4f26d6c2004-05-26 23:25:30 +0000170void sqlite3_result_error16(sqlite3_context *pCtx, const void *z, int n){
drh32bc3f62007-08-21 20:25:39 +0000171 assert( sqlite3_mutex_held(pCtx->s.db->mutex) );
drh69544ec2008-02-06 14:11:34 +0000172 pCtx->isError = SQLITE_ERROR;
drhb21c8cd2007-08-21 19:33:56 +0000173 sqlite3VdbeMemSetStr(&pCtx->s, z, n, SQLITE_UTF16NATIVE, SQLITE_TRANSIENT);
drh4f26d6c2004-05-26 23:25:30 +0000174}
danielk1977a1686c92006-01-23 07:52:37 +0000175#endif
drhf4479502004-05-27 03:12:53 +0000176void sqlite3_result_int(sqlite3_context *pCtx, int 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, (i64)iVal);
179}
180void sqlite3_result_int64(sqlite3_context *pCtx, i64 iVal){
drh32bc3f62007-08-21 20:25:39 +0000181 assert( sqlite3_mutex_held(pCtx->s.db->mutex) );
drh4f26d6c2004-05-26 23:25:30 +0000182 sqlite3VdbeMemSetInt64(&pCtx->s, iVal);
183}
184void sqlite3_result_null(sqlite3_context *pCtx){
drh32bc3f62007-08-21 20:25:39 +0000185 assert( sqlite3_mutex_held(pCtx->s.db->mutex) );
drhf4479502004-05-27 03:12:53 +0000186 sqlite3VdbeMemSetNull(&pCtx->s);
drh4f26d6c2004-05-26 23:25:30 +0000187}
188void sqlite3_result_text(
189 sqlite3_context *pCtx,
190 const char *z,
191 int n,
danielk1977d8123362004-06-12 09:25:12 +0000192 void (*xDel)(void *)
drh4f26d6c2004-05-26 23:25:30 +0000193){
drh32bc3f62007-08-21 20:25:39 +0000194 assert( sqlite3_mutex_held(pCtx->s.db->mutex) );
drhb21c8cd2007-08-21 19:33:56 +0000195 sqlite3VdbeMemSetStr(&pCtx->s, z, n, SQLITE_UTF8, xDel);
drh4f26d6c2004-05-26 23:25:30 +0000196}
drh6c626082004-11-14 21:56:29 +0000197#ifndef SQLITE_OMIT_UTF16
drh4f26d6c2004-05-26 23:25:30 +0000198void sqlite3_result_text16(
199 sqlite3_context *pCtx,
200 const void *z,
201 int n,
danielk1977d8123362004-06-12 09:25:12 +0000202 void (*xDel)(void *)
drh4f26d6c2004-05-26 23:25:30 +0000203){
drh32bc3f62007-08-21 20:25:39 +0000204 assert( sqlite3_mutex_held(pCtx->s.db->mutex) );
drhb21c8cd2007-08-21 19:33:56 +0000205 sqlite3VdbeMemSetStr(&pCtx->s, z, n, SQLITE_UTF16NATIVE, xDel);
danielk1977d8123362004-06-12 09:25:12 +0000206}
207void sqlite3_result_text16be(
208 sqlite3_context *pCtx,
209 const void *z,
210 int n,
211 void (*xDel)(void *)
212){
drh32bc3f62007-08-21 20:25:39 +0000213 assert( sqlite3_mutex_held(pCtx->s.db->mutex) );
drhb21c8cd2007-08-21 19:33:56 +0000214 sqlite3VdbeMemSetStr(&pCtx->s, z, n, SQLITE_UTF16BE, xDel);
danielk1977d8123362004-06-12 09:25:12 +0000215}
216void sqlite3_result_text16le(
217 sqlite3_context *pCtx,
218 const void *z,
219 int n,
220 void (*xDel)(void *)
221){
drh32bc3f62007-08-21 20:25:39 +0000222 assert( sqlite3_mutex_held(pCtx->s.db->mutex) );
drhb21c8cd2007-08-21 19:33:56 +0000223 sqlite3VdbeMemSetStr(&pCtx->s, z, n, SQLITE_UTF16LE, xDel);
drh4f26d6c2004-05-26 23:25:30 +0000224}
drh6c626082004-11-14 21:56:29 +0000225#endif /* SQLITE_OMIT_UTF16 */
drh4f26d6c2004-05-26 23:25:30 +0000226void sqlite3_result_value(sqlite3_context *pCtx, sqlite3_value *pValue){
drh32bc3f62007-08-21 20:25:39 +0000227 assert( sqlite3_mutex_held(pCtx->s.db->mutex) );
drhb21c8cd2007-08-21 19:33:56 +0000228 sqlite3VdbeMemCopy(&pCtx->s, pValue);
drh4f26d6c2004-05-26 23:25:30 +0000229}
drhb026e052007-05-02 01:34:31 +0000230void sqlite3_result_zeroblob(sqlite3_context *pCtx, int n){
drh32bc3f62007-08-21 20:25:39 +0000231 assert( sqlite3_mutex_held(pCtx->s.db->mutex) );
drhb026e052007-05-02 01:34:31 +0000232 sqlite3VdbeMemSetZeroBlob(&pCtx->s, n);
233}
drh69544ec2008-02-06 14:11:34 +0000234void sqlite3_result_error_code(sqlite3_context *pCtx, int errCode){
235 pCtx->isError = errCode;
236}
drh4f26d6c2004-05-26 23:25:30 +0000237
drha0206bc2007-05-08 15:15:02 +0000238/* Force an SQLITE_TOOBIG error. */
239void sqlite3_result_error_toobig(sqlite3_context *pCtx){
drh32bc3f62007-08-21 20:25:39 +0000240 assert( sqlite3_mutex_held(pCtx->s.db->mutex) );
drhbb4957f2008-03-20 14:03:29 +0000241 pCtx->isError = SQLITE_TOOBIG;
242 sqlite3VdbeMemSetStr(&pCtx->s, "string or blob too big", -1,
243 SQLITE_UTF8, SQLITE_STATIC);
drha0206bc2007-05-08 15:15:02 +0000244}
245
danielk1977a1644fd2007-08-29 12:31:25 +0000246/* An SQLITE_NOMEM error. */
247void sqlite3_result_error_nomem(sqlite3_context *pCtx){
248 assert( sqlite3_mutex_held(pCtx->s.db->mutex) );
249 sqlite3VdbeMemSetNull(&pCtx->s);
drh69544ec2008-02-06 14:11:34 +0000250 pCtx->isError = SQLITE_NOMEM;
danielk1977a1644fd2007-08-29 12:31:25 +0000251 pCtx->s.db->mallocFailed = 1;
252}
drh4f26d6c2004-05-26 23:25:30 +0000253
254/*
255** Execute the statement pStmt, either until a row of data is ready, the
256** statement is completely executed or an error occurs.
drhb900aaf2006-11-09 00:24:53 +0000257**
258** This routine implements the bulk of the logic behind the sqlite_step()
259** API. The only thing omitted is the automatic recompile if a
260** schema change has occurred. That detail is handled by the
261** outer sqlite3_step() wrapper procedure.
drh4f26d6c2004-05-26 23:25:30 +0000262*/
drhb900aaf2006-11-09 00:24:53 +0000263static int sqlite3Step(Vdbe *p){
drh9bb575f2004-09-06 17:24:11 +0000264 sqlite3 *db;
drh4f26d6c2004-05-26 23:25:30 +0000265 int rc;
266
danielk1977a185ddb2007-11-15 16:04:15 +0000267 assert(p);
268 if( p->magic!=VDBE_MAGIC_RUN ){
drhf1bfe9a2007-10-17 01:44:20 +0000269 return SQLITE_MISUSE;
270 }
271
danielk1977efaaf572006-01-16 11:29:19 +0000272 /* Assert that malloc() has not failed */
drh17435752007-08-16 04:30:38 +0000273 db = p->db;
274 assert( !db->mallocFailed );
danielk1977261919c2005-12-06 12:52:59 +0000275
drh91b48aa2004-06-30 11:14:18 +0000276 if( p->aborted ){
277 return SQLITE_ABORT;
278 }
danielk1977a21c6b62005-01-24 10:25:59 +0000279 if( p->pc<=0 && p->expired ){
280 if( p->rc==SQLITE_OK ){
281 p->rc = SQLITE_SCHEMA;
282 }
drhb900aaf2006-11-09 00:24:53 +0000283 rc = SQLITE_ERROR;
284 goto end_of_step;
danielk1977a21c6b62005-01-24 10:25:59 +0000285 }
drh4f26d6c2004-05-26 23:25:30 +0000286 if( sqlite3SafetyOn(db) ){
287 p->rc = SQLITE_MISUSE;
288 return SQLITE_MISUSE;
289 }
danielk19771d850a72004-05-31 08:26:49 +0000290 if( p->pc<0 ){
drh15ca1df2006-07-26 13:43:30 +0000291 /* If there are no other statements currently running, then
292 ** reset the interrupt flag. This prevents a call to sqlite3_interrupt
293 ** from interrupting a statement that has not yet started.
294 */
295 if( db->activeVdbeCnt==0 ){
296 db->u1.isInterrupted = 0;
297 }
298
drh19e2d372005-08-29 23:00:03 +0000299#ifndef SQLITE_OMIT_TRACE
drh19e2d372005-08-29 23:00:03 +0000300 if( db->xProfile && !db->init.busy ){
301 double rNow;
danielk1977b4b47412007-08-17 15:53:36 +0000302 sqlite3OsCurrentTime(db->pVfs, &rNow);
drh19e2d372005-08-29 23:00:03 +0000303 p->startTime = (rNow - (int)rNow)*3600.0*24.0*1000000000.0;
304 }
305#endif
drhc16a03b2004-09-15 13:38:10 +0000306
danielk19771d850a72004-05-31 08:26:49 +0000307 db->activeVdbeCnt++;
308 p->pc = 0;
309 }
drhb7f91642004-10-31 02:22:47 +0000310#ifndef SQLITE_OMIT_EXPLAIN
drh4f26d6c2004-05-26 23:25:30 +0000311 if( p->explain ){
312 rc = sqlite3VdbeList(p);
drhb7f91642004-10-31 02:22:47 +0000313 }else
314#endif /* SQLITE_OMIT_EXPLAIN */
315 {
drh4f26d6c2004-05-26 23:25:30 +0000316 rc = sqlite3VdbeExec(p);
317 }
318
319 if( sqlite3SafetyOff(db) ){
320 rc = SQLITE_MISUSE;
321 }
322
drh19e2d372005-08-29 23:00:03 +0000323#ifndef SQLITE_OMIT_TRACE
324 /* Invoke the profile callback if there is one
325 */
drh949f9cd2008-01-12 21:35:57 +0000326 if( rc!=SQLITE_ROW && db->xProfile && !db->init.busy && p->nOp>0
327 && p->aOp[0].opcode==OP_Trace && p->aOp[0].p4.z!=0 ){
drh19e2d372005-08-29 23:00:03 +0000328 double rNow;
329 u64 elapseTime;
330
danielk1977b4b47412007-08-17 15:53:36 +0000331 sqlite3OsCurrentTime(db->pVfs, &rNow);
drh19e2d372005-08-29 23:00:03 +0000332 elapseTime = (rNow - (int)rNow)*3600.0*24.0*1000000000.0 - p->startTime;
drh949f9cd2008-01-12 21:35:57 +0000333 db->xProfile(db->pProfileArg, p->aOp[0].p4.z, elapseTime);
drh19e2d372005-08-29 23:00:03 +0000334 }
335#endif
336
danielk197797a227c2006-01-20 16:32:04 +0000337 sqlite3Error(p->db, rc, 0);
danielk197776e8d1a2006-01-18 18:22:43 +0000338 p->rc = sqlite3ApiExit(p->db, p->rc);
drhb900aaf2006-11-09 00:24:53 +0000339end_of_step:
drh4ac285a2006-09-15 07:28:50 +0000340 assert( (rc&0xff)==rc );
drhb900aaf2006-11-09 00:24:53 +0000341 if( p->zSql && (rc&0xff)<SQLITE_ROW ){
342 /* This behavior occurs if sqlite3_prepare_v2() was used to build
343 ** the prepared statement. Return error codes directly */
danielk1977612642d2007-07-12 13:18:05 +0000344 sqlite3Error(p->db, p->rc, 0);
drhb900aaf2006-11-09 00:24:53 +0000345 return p->rc;
346 }else{
347 /* This is for legacy sqlite3_prepare() builds and when the code
348 ** is SQLITE_ROW or SQLITE_DONE */
349 return rc;
350 }
351}
352
353/*
354** This is the top-level implementation of sqlite3_step(). Call
355** sqlite3Step() to do most of the work. If a schema error occurs,
356** call sqlite3Reprepare() and try again.
357*/
358#ifdef SQLITE_OMIT_PARSER
359int sqlite3_step(sqlite3_stmt *pStmt){
danielk1977a185ddb2007-11-15 16:04:15 +0000360 int rc = SQLITE_MISUSE;
361 if( pStmt ){
362 Vdbe *v;
363 v = (Vdbe*)pStmt;
364 sqlite3_mutex_enter(v->db->mutex);
365 rc = sqlite3Step(v);
366 sqlite3_mutex_leave(v->db->mutex);
367 }
drhb21c8cd2007-08-21 19:33:56 +0000368 return rc;
drhb900aaf2006-11-09 00:24:53 +0000369}
370#else
371int sqlite3_step(sqlite3_stmt *pStmt){
danielk1977a185ddb2007-11-15 16:04:15 +0000372 int rc = SQLITE_MISUSE;
373 if( pStmt ){
374 int cnt = 0;
375 Vdbe *v = (Vdbe*)pStmt;
376 sqlite3 *db = v->db;
377 sqlite3_mutex_enter(db->mutex);
378 while( (rc = sqlite3Step(v))==SQLITE_SCHEMA
379 && cnt++ < 5
380 && sqlite3Reprepare(v) ){
381 sqlite3_reset(pStmt);
382 v->expired = 0;
danielk19778e556522007-11-13 10:30:24 +0000383 }
danielk1977a185ddb2007-11-15 16:04:15 +0000384 if( rc==SQLITE_SCHEMA && v->zSql && db->pErr ){
385 /* This case occurs after failing to recompile an sql statement.
386 ** The error message from the SQL compiler has already been loaded
387 ** into the database handle. This block copies the error message
388 ** from the database handle into the statement and sets the statement
389 ** program counter to 0 to ensure that when the statement is
390 ** finalized or reset the parser error message is available via
391 ** sqlite3_errmsg() and sqlite3_errcode().
392 */
393 const char *zErr = (const char *)sqlite3_value_text(db->pErr);
394 sqlite3_free(v->zErrMsg);
395 if( !db->mallocFailed ){
396 v->zErrMsg = sqlite3DbStrDup(db, zErr);
397 } else {
398 v->zErrMsg = 0;
399 v->rc = SQLITE_NOMEM;
400 }
401 }
402 rc = sqlite3ApiExit(db, rc);
403 sqlite3_mutex_leave(db->mutex);
danielk19778e556522007-11-13 10:30:24 +0000404 }
danielk197776e8d1a2006-01-18 18:22:43 +0000405 return rc;
drh4f26d6c2004-05-26 23:25:30 +0000406}
drhb900aaf2006-11-09 00:24:53 +0000407#endif
drh4f26d6c2004-05-26 23:25:30 +0000408
409/*
drheb2e1762004-05-27 01:53:56 +0000410** Extract the user data from a sqlite3_context structure and return a
411** pointer to it.
412*/
413void *sqlite3_user_data(sqlite3_context *p){
414 assert( p && p->pFunc );
415 return p->pFunc->pUserData;
416}
417
418/*
drhfa4a4b92008-03-19 21:45:51 +0000419** Extract the user data from a sqlite3_context structure and return a
420** pointer to it.
421*/
422sqlite3 *sqlite3_context_db_handle(sqlite3_context *p){
423 assert( p && p->pFunc );
424 return p->s.db;
425}
426
427/*
drhb7481e72006-09-16 21:45:14 +0000428** The following is the implementation of an SQL function that always
429** fails with an error message stating that the function is used in the
430** wrong context. The sqlite3_overload_function() API might construct
431** SQL function that use this routine so that the functions will exist
432** for name resolution but are actually overloaded by the xFindFunction
433** method of virtual tables.
434*/
435void sqlite3InvalidFunction(
436 sqlite3_context *context, /* The function calling context */
437 int argc, /* Number of arguments to the function */
438 sqlite3_value **argv /* Value of each argument */
439){
440 const char *zName = context->pFunc->zName;
441 char *zErr;
danielk19771e536952007-08-16 10:09:01 +0000442 zErr = sqlite3MPrintf(0,
drhb7481e72006-09-16 21:45:14 +0000443 "unable to use function %s in the requested context", zName);
444 sqlite3_result_error(context, zErr, -1);
danielk19771e536952007-08-16 10:09:01 +0000445 sqlite3_free(zErr);
drhb7481e72006-09-16 21:45:14 +0000446}
447
448/*
drheb2e1762004-05-27 01:53:56 +0000449** Allocate or return the aggregate context for a user function. A new
450** context is allocated on the first call. Subsequent calls return the
451** same context that was returned on prior calls.
drheb2e1762004-05-27 01:53:56 +0000452*/
453void *sqlite3_aggregate_context(sqlite3_context *p, int nByte){
drhb21c8cd2007-08-21 19:33:56 +0000454 Mem *pMem;
drh825c6622005-09-08 19:01:05 +0000455 assert( p && p->pFunc && p->pFunc->xStep );
drhb21c8cd2007-08-21 19:33:56 +0000456 assert( sqlite3_mutex_held(p->s.db->mutex) );
457 pMem = p->pMem;
drha10a34b2005-09-07 22:09:48 +0000458 if( (pMem->flags & MEM_Agg)==0 ){
459 if( nByte==0 ){
460 assert( pMem->flags==MEM_Null );
461 pMem->z = 0;
drheb2e1762004-05-27 01:53:56 +0000462 }else{
drha10a34b2005-09-07 22:09:48 +0000463 pMem->flags = MEM_Agg;
danielk19771e536952007-08-16 10:09:01 +0000464 pMem->xDel = sqlite3_free;
drh3c024d62007-03-30 11:23:45 +0000465 pMem->u.pDef = p->pFunc;
danielk1977d949fd32007-11-05 12:46:04 +0000466 pMem->z = sqlite3DbMallocZero(p->s.db, nByte);
drheb2e1762004-05-27 01:53:56 +0000467 }
468 }
drhabfcea22005-09-06 20:36:48 +0000469 return (void*)pMem->z;
drheb2e1762004-05-27 01:53:56 +0000470}
471
472/*
danielk1977682f68b2004-06-05 10:22:17 +0000473** Return the auxilary data pointer, if any, for the iArg'th argument to
474** the user-function defined by pCtx.
475*/
476void *sqlite3_get_auxdata(sqlite3_context *pCtx, int iArg){
drhb21c8cd2007-08-21 19:33:56 +0000477 VdbeFunc *pVdbeFunc;
478
479 assert( sqlite3_mutex_held(pCtx->s.db->mutex) );
480 pVdbeFunc = pCtx->pVdbeFunc;
danielk1977682f68b2004-06-05 10:22:17 +0000481 if( !pVdbeFunc || iArg>=pVdbeFunc->nAux || iArg<0 ){
482 return 0;
483 }
drhf92c7ff2004-06-19 15:40:23 +0000484 return pVdbeFunc->apAux[iArg].pAux;
danielk1977682f68b2004-06-05 10:22:17 +0000485}
486
487/*
488** Set the auxilary data pointer and delete function, for the iArg'th
489** argument to the user-function defined by pCtx. Any previous value is
490** deleted by calling the delete function specified when it was set.
491*/
492void sqlite3_set_auxdata(
493 sqlite3_context *pCtx,
494 int iArg,
495 void *pAux,
496 void (*xDelete)(void*)
497){
498 struct AuxData *pAuxData;
drhf92c7ff2004-06-19 15:40:23 +0000499 VdbeFunc *pVdbeFunc;
danielk1977e0fc5262007-07-26 06:50:05 +0000500 if( iArg<0 ) goto failed;
danielk1977682f68b2004-06-05 10:22:17 +0000501
drhb21c8cd2007-08-21 19:33:56 +0000502 assert( sqlite3_mutex_held(pCtx->s.db->mutex) );
drhf92c7ff2004-06-19 15:40:23 +0000503 pVdbeFunc = pCtx->pVdbeFunc;
504 if( !pVdbeFunc || pVdbeFunc->nAux<=iArg ){
danielk197731dad9d2007-08-16 11:36:15 +0000505 int nAux = (pVdbeFunc ? pVdbeFunc->nAux : 0);
drhf92c7ff2004-06-19 15:40:23 +0000506 int nMalloc = sizeof(VdbeFunc) + sizeof(struct AuxData)*iArg;
danielk197726783a52007-08-29 14:06:22 +0000507 pVdbeFunc = sqlite3DbRealloc(pCtx->s.db, pVdbeFunc, nMalloc);
drh17435752007-08-16 04:30:38 +0000508 if( !pVdbeFunc ){
drh17435752007-08-16 04:30:38 +0000509 goto failed;
510 }
drh53f733c2005-09-16 02:38:09 +0000511 pCtx->pVdbeFunc = pVdbeFunc;
danielk197731dad9d2007-08-16 11:36:15 +0000512 memset(&pVdbeFunc->apAux[nAux], 0, sizeof(struct AuxData)*(iArg+1-nAux));
drh998da3a2004-06-19 15:22:56 +0000513 pVdbeFunc->nAux = iArg+1;
514 pVdbeFunc->pFunc = pCtx->pFunc;
danielk1977682f68b2004-06-05 10:22:17 +0000515 }
516
drhf92c7ff2004-06-19 15:40:23 +0000517 pAuxData = &pVdbeFunc->apAux[iArg];
danielk1977682f68b2004-06-05 10:22:17 +0000518 if( pAuxData->pAux && pAuxData->xDelete ){
519 pAuxData->xDelete(pAuxData->pAux);
520 }
521 pAuxData->pAux = pAux;
522 pAuxData->xDelete = xDelete;
danielk1977e0fc5262007-07-26 06:50:05 +0000523 return;
524
525failed:
526 if( xDelete ){
527 xDelete(pAux);
528 }
danielk1977682f68b2004-06-05 10:22:17 +0000529}
530
531/*
drheb2e1762004-05-27 01:53:56 +0000532** Return the number of times the Step function of a aggregate has been
533** called.
534**
drhcf85a512006-02-09 18:35:29 +0000535** This function is deprecated. Do not use it for new code. It is
536** provide only to avoid breaking legacy code. New aggregate function
537** implementations should keep their own counts within their aggregate
538** context.
drheb2e1762004-05-27 01:53:56 +0000539*/
540int sqlite3_aggregate_count(sqlite3_context *p){
541 assert( p && p->pFunc && p->pFunc->xStep );
drhabfcea22005-09-06 20:36:48 +0000542 return p->pMem->n;
drheb2e1762004-05-27 01:53:56 +0000543}
544
545/*
drh4f26d6c2004-05-26 23:25:30 +0000546** Return the number of columns in the result set for the statement pStmt.
547*/
548int sqlite3_column_count(sqlite3_stmt *pStmt){
549 Vdbe *pVm = (Vdbe *)pStmt;
drh1bcdb0c2004-08-28 14:49:46 +0000550 return pVm ? pVm->nResColumn : 0;
drh4f26d6c2004-05-26 23:25:30 +0000551}
552
553/*
554** Return the number of values available from the current row of the
555** currently executing statement pStmt.
556*/
557int sqlite3_data_count(sqlite3_stmt *pStmt){
558 Vdbe *pVm = (Vdbe *)pStmt;
drhd4e70eb2008-01-02 00:34:36 +0000559 if( pVm==0 || pVm->pResultSet==0 ) return 0;
drh4f26d6c2004-05-26 23:25:30 +0000560 return pVm->nResColumn;
561}
562
563
564/*
565** Check to see if column iCol of the given statement is valid. If
566** it is, return a pointer to the Mem for the value of that column.
567** If iCol is not valid, return a pointer to a Mem which has a value
568** of NULL.
569*/
570static Mem *columnMem(sqlite3_stmt *pStmt, int i){
drh32bc3f62007-08-21 20:25:39 +0000571 Vdbe *pVm;
572 int vals;
573 Mem *pOut;
574
575 pVm = (Vdbe *)pStmt;
drhd4e70eb2008-01-02 00:34:36 +0000576 if( pVm && pVm->pResultSet!=0 && i<pVm->nResColumn && i>=0 ){
drh32bc3f62007-08-21 20:25:39 +0000577 sqlite3_mutex_enter(pVm->db->mutex);
578 vals = sqlite3_data_count(pStmt);
drhd4e70eb2008-01-02 00:34:36 +0000579 pOut = &pVm->pResultSet[i];
drh32bc3f62007-08-21 20:25:39 +0000580 }else{
drhb21c8cd2007-08-21 19:33:56 +0000581 static const Mem nullMem = {{0}, 0.0, 0, "", 0, MEM_Null, SQLITE_NULL };
drh27641702007-08-22 02:56:42 +0000582 if( pVm->db ){
583 sqlite3_mutex_enter(pVm->db->mutex);
584 sqlite3Error(pVm->db, SQLITE_RANGE, 0);
585 }
drh32bc3f62007-08-21 20:25:39 +0000586 pOut = (Mem*)&nullMem;
drh4f26d6c2004-05-26 23:25:30 +0000587 }
drh32bc3f62007-08-21 20:25:39 +0000588 return pOut;
drh4f26d6c2004-05-26 23:25:30 +0000589}
590
danielk19772e588c72005-12-09 14:25:08 +0000591/*
592** This function is called after invoking an sqlite3_value_XXX function on a
593** column value (i.e. a value returned by evaluating an SQL expression in the
594** select list of a SELECT statement) that may cause a malloc() failure. If
595** malloc() has failed, the threads mallocFailed flag is cleared and the result
596** code of statement pStmt set to SQLITE_NOMEM.
597**
drh32bc3f62007-08-21 20:25:39 +0000598** Specifically, this is called from within:
danielk19772e588c72005-12-09 14:25:08 +0000599**
600** sqlite3_column_int()
601** sqlite3_column_int64()
602** sqlite3_column_text()
603** sqlite3_column_text16()
604** sqlite3_column_real()
605** sqlite3_column_bytes()
606** sqlite3_column_bytes16()
607**
608** But not for sqlite3_column_blob(), which never calls malloc().
609*/
610static void columnMallocFailure(sqlite3_stmt *pStmt)
611{
612 /* If malloc() failed during an encoding conversion within an
613 ** sqlite3_column_XXX API, then set the return code of the statement to
614 ** SQLITE_NOMEM. The next call to _step() (if any) will return SQLITE_ERROR
615 ** and _finalize() will return NOMEM.
616 */
danielk197754f01982006-01-18 15:25:17 +0000617 Vdbe *p = (Vdbe *)pStmt;
drh32bc3f62007-08-21 20:25:39 +0000618 if( p ){
619 p->rc = sqlite3ApiExit(p->db, p->rc);
620 sqlite3_mutex_leave(p->db->mutex);
621 }
danielk19772e588c72005-12-09 14:25:08 +0000622}
623
drh4f26d6c2004-05-26 23:25:30 +0000624/**************************** sqlite3_column_ *******************************
625** The following routines are used to access elements of the current row
626** in the result set.
627*/
danielk1977c572ef72004-05-27 09:28:41 +0000628const void *sqlite3_column_blob(sqlite3_stmt *pStmt, int i){
danielk19772e588c72005-12-09 14:25:08 +0000629 const void *val;
danielk19772e588c72005-12-09 14:25:08 +0000630 val = sqlite3_value_blob( columnMem(pStmt,i) );
danielk1977c9cf9012007-05-30 10:36:47 +0000631 /* Even though there is no encoding conversion, value_blob() might
632 ** need to call malloc() to expand the result of a zeroblob()
633 ** expression.
634 */
635 columnMallocFailure(pStmt);
danielk19772e588c72005-12-09 14:25:08 +0000636 return val;
danielk1977c572ef72004-05-27 09:28:41 +0000637}
drh4f26d6c2004-05-26 23:25:30 +0000638int sqlite3_column_bytes(sqlite3_stmt *pStmt, int i){
danielk19772e588c72005-12-09 14:25:08 +0000639 int val = sqlite3_value_bytes( columnMem(pStmt,i) );
640 columnMallocFailure(pStmt);
641 return val;
drh4f26d6c2004-05-26 23:25:30 +0000642}
643int sqlite3_column_bytes16(sqlite3_stmt *pStmt, int i){
danielk19772e588c72005-12-09 14:25:08 +0000644 int val = sqlite3_value_bytes16( columnMem(pStmt,i) );
645 columnMallocFailure(pStmt);
646 return val;
drh4f26d6c2004-05-26 23:25:30 +0000647}
648double sqlite3_column_double(sqlite3_stmt *pStmt, int i){
danielk19772e588c72005-12-09 14:25:08 +0000649 double val = sqlite3_value_double( columnMem(pStmt,i) );
650 columnMallocFailure(pStmt);
651 return val;
drh4f26d6c2004-05-26 23:25:30 +0000652}
653int sqlite3_column_int(sqlite3_stmt *pStmt, int i){
danielk19772e588c72005-12-09 14:25:08 +0000654 int val = sqlite3_value_int( columnMem(pStmt,i) );
655 columnMallocFailure(pStmt);
656 return val;
drh4f26d6c2004-05-26 23:25:30 +0000657}
drhefad9992004-06-22 12:13:55 +0000658sqlite_int64 sqlite3_column_int64(sqlite3_stmt *pStmt, int i){
danielk19772e588c72005-12-09 14:25:08 +0000659 sqlite_int64 val = sqlite3_value_int64( columnMem(pStmt,i) );
660 columnMallocFailure(pStmt);
661 return val;
drh4f26d6c2004-05-26 23:25:30 +0000662}
663const unsigned char *sqlite3_column_text(sqlite3_stmt *pStmt, int i){
danielk19772e588c72005-12-09 14:25:08 +0000664 const unsigned char *val = sqlite3_value_text( columnMem(pStmt,i) );
665 columnMallocFailure(pStmt);
666 return val;
drh4f26d6c2004-05-26 23:25:30 +0000667}
drhd1e47332005-06-26 17:55:33 +0000668sqlite3_value *sqlite3_column_value(sqlite3_stmt *pStmt, int i){
drh32bc3f62007-08-21 20:25:39 +0000669 sqlite3_value *pOut = columnMem(pStmt, i);
670 columnMallocFailure(pStmt);
671 return pOut;
drhd1e47332005-06-26 17:55:33 +0000672}
drh6c626082004-11-14 21:56:29 +0000673#ifndef SQLITE_OMIT_UTF16
drh4f26d6c2004-05-26 23:25:30 +0000674const void *sqlite3_column_text16(sqlite3_stmt *pStmt, int i){
danielk19772e588c72005-12-09 14:25:08 +0000675 const void *val = sqlite3_value_text16( columnMem(pStmt,i) );
676 columnMallocFailure(pStmt);
677 return val;
drh4f26d6c2004-05-26 23:25:30 +0000678}
drh6c626082004-11-14 21:56:29 +0000679#endif /* SQLITE_OMIT_UTF16 */
drh4f26d6c2004-05-26 23:25:30 +0000680int sqlite3_column_type(sqlite3_stmt *pStmt, int i){
drh32bc3f62007-08-21 20:25:39 +0000681 int iType = sqlite3_value_type( columnMem(pStmt,i) );
682 columnMallocFailure(pStmt);
683 return iType;
drh4f26d6c2004-05-26 23:25:30 +0000684}
685
drh29d72102006-02-09 22:13:41 +0000686/* The following function is experimental and subject to change or
687** removal */
688/*int sqlite3_column_numeric_type(sqlite3_stmt *pStmt, int i){
689** return sqlite3_value_numeric_type( columnMem(pStmt,i) );
690**}
691*/
692
drh5e2517e2004-09-24 23:59:12 +0000693/*
694** Convert the N-th element of pStmt->pColName[] into a string using
695** xFunc() then return that string. If N is out of range, return 0.
drhe590fbd2005-05-20 19:36:01 +0000696**
697** There are up to 5 names for each column. useType determines which
698** name is returned. Here are the names:
699**
700** 0 The column name as it should be displayed for output
701** 1 The datatype name for the column
702** 2 The name of the database that the column derives from
703** 3 The name of the table that the column derives from
704** 4 The name of the table column that the result column derives from
705**
706** If the result is not a simple column reference (if it is an expression
707** or a constant) then useTypes 2, 3, and 4 return NULL.
drh5e2517e2004-09-24 23:59:12 +0000708*/
709static const void *columnName(
710 sqlite3_stmt *pStmt,
711 int N,
712 const void *(*xFunc)(Mem*),
713 int useType
714){
drh32bc3f62007-08-21 20:25:39 +0000715 const void *ret = 0;
drh5e2517e2004-09-24 23:59:12 +0000716 Vdbe *p = (Vdbe *)pStmt;
drh32bc3f62007-08-21 20:25:39 +0000717 int n;
718
drh5e2517e2004-09-24 23:59:12 +0000719
drh32bc3f62007-08-21 20:25:39 +0000720 if( p!=0 ){
721 n = sqlite3_column_count(pStmt);
722 if( N<n && N>=0 ){
723 N += useType*n;
drh153c62c2007-08-24 03:51:33 +0000724 sqlite3_mutex_enter(p->db->mutex);
drh32bc3f62007-08-21 20:25:39 +0000725 ret = xFunc(&p->aColName[N]);
726
drh32bc3f62007-08-21 20:25:39 +0000727 /* A malloc may have failed inside of the xFunc() call. If this
728 ** is the case, clear the mallocFailed flag and return NULL.
729 */
730 if( p->db && p->db->mallocFailed ){
731 p->db->mallocFailed = 0;
732 ret = 0;
733 }
drh153c62c2007-08-24 03:51:33 +0000734 sqlite3_mutex_leave(p->db->mutex);
drh32bc3f62007-08-21 20:25:39 +0000735 }
drh5e2517e2004-09-24 23:59:12 +0000736 }
danielk197700fd9572005-12-07 06:27:43 +0000737 return ret;
drh5e2517e2004-09-24 23:59:12 +0000738}
739
drh4f26d6c2004-05-26 23:25:30 +0000740/*
741** Return the name of the Nth column of the result set returned by SQL
742** statement pStmt.
743*/
744const char *sqlite3_column_name(sqlite3_stmt *pStmt, int N){
danielk1977955de522006-02-10 02:27:42 +0000745 return columnName(
746 pStmt, N, (const void*(*)(Mem*))sqlite3_value_text, COLNAME_NAME);
drh4f26d6c2004-05-26 23:25:30 +0000747}
drhe590fbd2005-05-20 19:36:01 +0000748#ifndef SQLITE_OMIT_UTF16
749const void *sqlite3_column_name16(sqlite3_stmt *pStmt, int N){
danielk1977955de522006-02-10 02:27:42 +0000750 return columnName(
751 pStmt, N, (const void*(*)(Mem*))sqlite3_value_text16, COLNAME_NAME);
drhe590fbd2005-05-20 19:36:01 +0000752}
753#endif
drh4f26d6c2004-05-26 23:25:30 +0000754
755/*
drh6c626082004-11-14 21:56:29 +0000756** Return the column declaration type (if applicable) of the 'i'th column
drhe590fbd2005-05-20 19:36:01 +0000757** of the result set of SQL statement pStmt.
drh6c626082004-11-14 21:56:29 +0000758*/
759const char *sqlite3_column_decltype(sqlite3_stmt *pStmt, int N){
danielk1977955de522006-02-10 02:27:42 +0000760 return columnName(
761 pStmt, N, (const void*(*)(Mem*))sqlite3_value_text, COLNAME_DECLTYPE);
drh6c626082004-11-14 21:56:29 +0000762}
drh6c626082004-11-14 21:56:29 +0000763#ifndef SQLITE_OMIT_UTF16
danielk197776d505b2004-05-28 13:13:02 +0000764const void *sqlite3_column_decltype16(sqlite3_stmt *pStmt, int N){
danielk1977955de522006-02-10 02:27:42 +0000765 return columnName(
766 pStmt, N, (const void*(*)(Mem*))sqlite3_value_text16, COLNAME_DECLTYPE);
drh4f26d6c2004-05-26 23:25:30 +0000767}
drh6c626082004-11-14 21:56:29 +0000768#endif /* SQLITE_OMIT_UTF16 */
drh4f26d6c2004-05-26 23:25:30 +0000769
danielk19774b1ae992006-02-10 03:06:10 +0000770#ifdef SQLITE_ENABLE_COLUMN_METADATA
drhe590fbd2005-05-20 19:36:01 +0000771/*
772** Return the name of the database from which a result column derives.
773** NULL is returned if the result column is an expression or constant or
774** anything else which is not an unabiguous reference to a database column.
775*/
776const char *sqlite3_column_database_name(sqlite3_stmt *pStmt, int N){
danielk1977955de522006-02-10 02:27:42 +0000777 return columnName(
778 pStmt, N, (const void*(*)(Mem*))sqlite3_value_text, COLNAME_DATABASE);
drhe590fbd2005-05-20 19:36:01 +0000779}
780#ifndef SQLITE_OMIT_UTF16
781const void *sqlite3_column_database_name16(sqlite3_stmt *pStmt, int N){
danielk1977955de522006-02-10 02:27:42 +0000782 return columnName(
783 pStmt, N, (const void*(*)(Mem*))sqlite3_value_text16, COLNAME_DATABASE);
drhe590fbd2005-05-20 19:36:01 +0000784}
785#endif /* SQLITE_OMIT_UTF16 */
786
787/*
788** Return the name of the table from which a result column derives.
789** NULL is returned if the result column is an expression or constant or
790** anything else which is not an unabiguous reference to a database column.
791*/
792const char *sqlite3_column_table_name(sqlite3_stmt *pStmt, int N){
danielk1977955de522006-02-10 02:27:42 +0000793 return columnName(
794 pStmt, N, (const void*(*)(Mem*))sqlite3_value_text, COLNAME_TABLE);
drhe590fbd2005-05-20 19:36:01 +0000795}
796#ifndef SQLITE_OMIT_UTF16
797const void *sqlite3_column_table_name16(sqlite3_stmt *pStmt, int N){
danielk1977955de522006-02-10 02:27:42 +0000798 return columnName(
799 pStmt, N, (const void*(*)(Mem*))sqlite3_value_text16, COLNAME_TABLE);
drhe590fbd2005-05-20 19:36:01 +0000800}
801#endif /* SQLITE_OMIT_UTF16 */
802
803/*
804** Return the name of the table column from which a result column derives.
805** NULL is returned if the result column is an expression or constant or
806** anything else which is not an unabiguous reference to a database column.
807*/
808const char *sqlite3_column_origin_name(sqlite3_stmt *pStmt, int N){
danielk1977955de522006-02-10 02:27:42 +0000809 return columnName(
810 pStmt, N, (const void*(*)(Mem*))sqlite3_value_text, COLNAME_COLUMN);
drhe590fbd2005-05-20 19:36:01 +0000811}
812#ifndef SQLITE_OMIT_UTF16
813const void *sqlite3_column_origin_name16(sqlite3_stmt *pStmt, int N){
danielk1977955de522006-02-10 02:27:42 +0000814 return columnName(
815 pStmt, N, (const void*(*)(Mem*))sqlite3_value_text16, COLNAME_COLUMN);
drhe590fbd2005-05-20 19:36:01 +0000816}
817#endif /* SQLITE_OMIT_UTF16 */
danielk19774b1ae992006-02-10 03:06:10 +0000818#endif /* SQLITE_ENABLE_COLUMN_METADATA */
drhe590fbd2005-05-20 19:36:01 +0000819
820
drh4f26d6c2004-05-26 23:25:30 +0000821/******************************* sqlite3_bind_ ***************************
822**
823** Routines used to attach values to wildcards in a compiled SQL statement.
824*/
825/*
826** Unbind the value bound to variable i in virtual machine p. This is the
827** the same as binding a NULL value to the column. If the "i" parameter is
828** out of range, then SQLITE_RANGE is returned. Othewise SQLITE_OK.
829**
830** The error code stored in database p->db is overwritten with the return
831** value in any case.
832*/
833static int vdbeUnbind(Vdbe *p, int i){
834 Mem *pVar;
drh1bcdb0c2004-08-28 14:49:46 +0000835 if( p==0 || p->magic!=VDBE_MAGIC_RUN || p->pc>=0 ){
drh473d1792005-06-06 17:54:55 +0000836 if( p ) sqlite3Error(p->db, SQLITE_MISUSE, 0);
drh4f26d6c2004-05-26 23:25:30 +0000837 return SQLITE_MISUSE;
838 }
839 if( i<1 || i>p->nVar ){
840 sqlite3Error(p->db, SQLITE_RANGE, 0);
841 return SQLITE_RANGE;
842 }
843 i--;
drh290c1942004-08-21 17:54:45 +0000844 pVar = &p->aVar[i];
danielk1977d8123362004-06-12 09:25:12 +0000845 sqlite3VdbeMemRelease(pVar);
drh4f26d6c2004-05-26 23:25:30 +0000846 pVar->flags = MEM_Null;
847 sqlite3Error(p->db, SQLITE_OK, 0);
848 return SQLITE_OK;
849}
850
851/*
drh5e2517e2004-09-24 23:59:12 +0000852** Bind a text or BLOB value.
drh4f26d6c2004-05-26 23:25:30 +0000853*/
drh5e2517e2004-09-24 23:59:12 +0000854static int bindText(
drh605264d2007-08-21 15:13:19 +0000855 sqlite3_stmt *pStmt, /* The statement to bind against */
856 int i, /* Index of the parameter to bind */
857 const void *zData, /* Pointer to the data to be bound */
858 int nData, /* Number of bytes of data to be bound */
859 void (*xDel)(void*), /* Destructor for the data */
860 int encoding /* Encoding for the data */
drh4f26d6c2004-05-26 23:25:30 +0000861){
862 Vdbe *p = (Vdbe *)pStmt;
863 Mem *pVar;
864 int rc;
865
drh605264d2007-08-21 15:13:19 +0000866 if( p==0 ){
867 return SQLITE_MISUSE;
868 }
869 sqlite3_mutex_enter(p->db->mutex);
drh4f26d6c2004-05-26 23:25:30 +0000870 rc = vdbeUnbind(p, i);
drh27641702007-08-22 02:56:42 +0000871 if( rc==SQLITE_OK && zData!=0 ){
872 pVar = &p->aVar[i-1];
873 rc = sqlite3VdbeMemSetStr(pVar, zData, nData, encoding, xDel);
874 if( rc==SQLITE_OK && encoding!=0 ){
875 rc = sqlite3VdbeChangeEncoding(pVar, ENC(p->db));
876 }
877 sqlite3Error(p->db, rc, 0);
878 rc = sqlite3ApiExit(p->db, rc);
drh4f26d6c2004-05-26 23:25:30 +0000879 }
drh605264d2007-08-21 15:13:19 +0000880 sqlite3_mutex_leave(p->db->mutex);
881 return rc;
drh4f26d6c2004-05-26 23:25:30 +0000882}
drh5e2517e2004-09-24 23:59:12 +0000883
884
885/*
886** Bind a blob value to an SQL statement variable.
887*/
888int sqlite3_bind_blob(
889 sqlite3_stmt *pStmt,
890 int i,
891 const void *zData,
892 int nData,
893 void (*xDel)(void*)
894){
895 return bindText(pStmt, i, zData, nData, xDel, 0);
896}
drh4f26d6c2004-05-26 23:25:30 +0000897int sqlite3_bind_double(sqlite3_stmt *pStmt, int i, double rValue){
898 int rc;
899 Vdbe *p = (Vdbe *)pStmt;
drh605264d2007-08-21 15:13:19 +0000900 sqlite3_mutex_enter(p->db->mutex);
drh4f26d6c2004-05-26 23:25:30 +0000901 rc = vdbeUnbind(p, i);
902 if( rc==SQLITE_OK ){
drh290c1942004-08-21 17:54:45 +0000903 sqlite3VdbeMemSetDouble(&p->aVar[i-1], rValue);
drh4f26d6c2004-05-26 23:25:30 +0000904 }
drh605264d2007-08-21 15:13:19 +0000905 sqlite3_mutex_leave(p->db->mutex);
danielk1977f4618892004-06-28 13:09:11 +0000906 return rc;
drh4f26d6c2004-05-26 23:25:30 +0000907}
908int sqlite3_bind_int(sqlite3_stmt *p, int i, int iValue){
drhefad9992004-06-22 12:13:55 +0000909 return sqlite3_bind_int64(p, i, (i64)iValue);
drh4f26d6c2004-05-26 23:25:30 +0000910}
drhefad9992004-06-22 12:13:55 +0000911int sqlite3_bind_int64(sqlite3_stmt *pStmt, int i, sqlite_int64 iValue){
drh4f26d6c2004-05-26 23:25:30 +0000912 int rc;
913 Vdbe *p = (Vdbe *)pStmt;
drh605264d2007-08-21 15:13:19 +0000914 sqlite3_mutex_enter(p->db->mutex);
drh4f26d6c2004-05-26 23:25:30 +0000915 rc = vdbeUnbind(p, i);
916 if( rc==SQLITE_OK ){
drh290c1942004-08-21 17:54:45 +0000917 sqlite3VdbeMemSetInt64(&p->aVar[i-1], iValue);
drh4f26d6c2004-05-26 23:25:30 +0000918 }
drh605264d2007-08-21 15:13:19 +0000919 sqlite3_mutex_leave(p->db->mutex);
drh4f26d6c2004-05-26 23:25:30 +0000920 return rc;
921}
drh605264d2007-08-21 15:13:19 +0000922int sqlite3_bind_null(sqlite3_stmt *pStmt, int i){
923 int rc;
924 Vdbe *p = (Vdbe*)pStmt;
925 sqlite3_mutex_enter(p->db->mutex);
926 rc = vdbeUnbind(p, i);
927 sqlite3_mutex_leave(p->db->mutex);
928 return rc;
drh4f26d6c2004-05-26 23:25:30 +0000929}
930int sqlite3_bind_text(
931 sqlite3_stmt *pStmt,
932 int i,
933 const char *zData,
934 int nData,
danielk1977d8123362004-06-12 09:25:12 +0000935 void (*xDel)(void*)
drh4f26d6c2004-05-26 23:25:30 +0000936){
drh5e2517e2004-09-24 23:59:12 +0000937 return bindText(pStmt, i, zData, nData, xDel, SQLITE_UTF8);
drh4f26d6c2004-05-26 23:25:30 +0000938}
drh6c626082004-11-14 21:56:29 +0000939#ifndef SQLITE_OMIT_UTF16
drh4f26d6c2004-05-26 23:25:30 +0000940int sqlite3_bind_text16(
941 sqlite3_stmt *pStmt,
942 int i,
943 const void *zData,
944 int nData,
danielk1977d8123362004-06-12 09:25:12 +0000945 void (*xDel)(void*)
drh4f26d6c2004-05-26 23:25:30 +0000946){
drh5e2517e2004-09-24 23:59:12 +0000947 return bindText(pStmt, i, zData, nData, xDel, SQLITE_UTF16NATIVE);
drh4f26d6c2004-05-26 23:25:30 +0000948}
drh6c626082004-11-14 21:56:29 +0000949#endif /* SQLITE_OMIT_UTF16 */
danielk197726e41442006-06-14 15:16:35 +0000950int sqlite3_bind_value(sqlite3_stmt *pStmt, int i, const sqlite3_value *pValue){
951 int rc;
952 Vdbe *p = (Vdbe *)pStmt;
drh605264d2007-08-21 15:13:19 +0000953 sqlite3_mutex_enter(p->db->mutex);
danielk197726e41442006-06-14 15:16:35 +0000954 rc = vdbeUnbind(p, i);
955 if( rc==SQLITE_OK ){
drhb21c8cd2007-08-21 19:33:56 +0000956 rc = sqlite3VdbeMemCopy(&p->aVar[i-1], pValue);
danielk197726e41442006-06-14 15:16:35 +0000957 }
danielk1977a7a8e142008-02-13 18:25:27 +0000958 rc = sqlite3ApiExit(p->db, rc);
drh605264d2007-08-21 15:13:19 +0000959 sqlite3_mutex_leave(p->db->mutex);
danielk197726e41442006-06-14 15:16:35 +0000960 return rc;
961}
drhb026e052007-05-02 01:34:31 +0000962int sqlite3_bind_zeroblob(sqlite3_stmt *pStmt, int i, int n){
963 int rc;
964 Vdbe *p = (Vdbe *)pStmt;
drh605264d2007-08-21 15:13:19 +0000965 sqlite3_mutex_enter(p->db->mutex);
drhb026e052007-05-02 01:34:31 +0000966 rc = vdbeUnbind(p, i);
967 if( rc==SQLITE_OK ){
968 sqlite3VdbeMemSetZeroBlob(&p->aVar[i-1], n);
969 }
drh605264d2007-08-21 15:13:19 +0000970 sqlite3_mutex_leave(p->db->mutex);
drhb026e052007-05-02 01:34:31 +0000971 return rc;
972}
drh75f6a032004-07-15 14:15:00 +0000973
974/*
975** Return the number of wildcards that can be potentially bound to.
976** This routine is added to support DBD::SQLite.
drh75f6a032004-07-15 14:15:00 +0000977*/
978int sqlite3_bind_parameter_count(sqlite3_stmt *pStmt){
drh92febd92004-08-20 18:34:20 +0000979 Vdbe *p = (Vdbe*)pStmt;
980 return p ? p->nVar : 0;
drh75f6a032004-07-15 14:15:00 +0000981}
drh895d7472004-08-20 16:02:39 +0000982
983/*
drhfa6bc002004-09-07 16:19:52 +0000984** Create a mapping from variable numbers to variable names
985** in the Vdbe.azVar[] array, if such a mapping does not already
986** exist.
drh895d7472004-08-20 16:02:39 +0000987*/
drhfa6bc002004-09-07 16:19:52 +0000988static void createVarMap(Vdbe *p){
drh895d7472004-08-20 16:02:39 +0000989 if( !p->okVar ){
drh605264d2007-08-21 15:13:19 +0000990 sqlite3_mutex_enter(p->db->mutex);
991 if( !p->okVar ){
992 int j;
993 Op *pOp;
994 for(j=0, pOp=p->aOp; j<p->nOp; j++, pOp++){
995 if( pOp->opcode==OP_Variable ){
996 assert( pOp->p1>0 && pOp->p1<=p->nVar );
danielk19772dca4ac2008-01-03 11:50:29 +0000997 p->azVar[pOp->p1-1] = pOp->p4.z;
drh605264d2007-08-21 15:13:19 +0000998 }
drh895d7472004-08-20 16:02:39 +0000999 }
drh605264d2007-08-21 15:13:19 +00001000 p->okVar = 1;
drh895d7472004-08-20 16:02:39 +00001001 }
drh605264d2007-08-21 15:13:19 +00001002 sqlite3_mutex_leave(p->db->mutex);
drh895d7472004-08-20 16:02:39 +00001003 }
drhfa6bc002004-09-07 16:19:52 +00001004}
1005
1006/*
1007** Return the name of a wildcard parameter. Return NULL if the index
1008** is out of range or if the wildcard is unnamed.
1009**
1010** The result is always UTF-8.
1011*/
1012const char *sqlite3_bind_parameter_name(sqlite3_stmt *pStmt, int i){
1013 Vdbe *p = (Vdbe*)pStmt;
1014 if( p==0 || i<1 || i>p->nVar ){
1015 return 0;
1016 }
1017 createVarMap(p);
drh895d7472004-08-20 16:02:39 +00001018 return p->azVar[i-1];
1019}
drhfa6bc002004-09-07 16:19:52 +00001020
1021/*
1022** Given a wildcard parameter name, return the index of the variable
1023** with that name. If there is no variable with the given name,
1024** return 0.
1025*/
1026int sqlite3_bind_parameter_index(sqlite3_stmt *pStmt, const char *zName){
1027 Vdbe *p = (Vdbe*)pStmt;
1028 int i;
1029 if( p==0 ){
1030 return 0;
1031 }
1032 createVarMap(p);
drh6a8903c2004-12-10 18:00:04 +00001033 if( zName ){
1034 for(i=0; i<p->nVar; i++){
1035 const char *z = p->azVar[i];
1036 if( z && strcmp(z,zName)==0 ){
1037 return i+1;
1038 }
drhfa6bc002004-09-07 16:19:52 +00001039 }
1040 }
1041 return 0;
1042}
drhf8db1bc2005-04-22 02:38:37 +00001043
drh51942bc2005-06-12 22:01:42 +00001044/*
drhf8db1bc2005-04-22 02:38:37 +00001045** Transfer all bindings from the first statement over to the second.
1046** If the two statements contain a different number of bindings, then
1047** an SQLITE_ERROR is returned.
1048*/
1049int sqlite3_transfer_bindings(sqlite3_stmt *pFromStmt, sqlite3_stmt *pToStmt){
1050 Vdbe *pFrom = (Vdbe*)pFromStmt;
1051 Vdbe *pTo = (Vdbe*)pToStmt;
1052 int i, rc = SQLITE_OK;
1053 if( (pFrom->magic!=VDBE_MAGIC_RUN && pFrom->magic!=VDBE_MAGIC_HALT)
drh27641702007-08-22 02:56:42 +00001054 || (pTo->magic!=VDBE_MAGIC_RUN && pTo->magic!=VDBE_MAGIC_HALT)
1055 || pTo->db!=pFrom->db ){
drhf8db1bc2005-04-22 02:38:37 +00001056 return SQLITE_MISUSE;
1057 }
1058 if( pFrom->nVar!=pTo->nVar ){
1059 return SQLITE_ERROR;
1060 }
drh27641702007-08-22 02:56:42 +00001061 sqlite3_mutex_enter(pTo->db->mutex);
drhf8db1bc2005-04-22 02:38:37 +00001062 for(i=0; rc==SQLITE_OK && i<pFrom->nVar; i++){
drh643167f2008-01-22 21:30:53 +00001063 sqlite3VdbeMemMove(&pTo->aVar[i], &pFrom->aVar[i]);
drhf8db1bc2005-04-22 02:38:37 +00001064 }
drh27641702007-08-22 02:56:42 +00001065 sqlite3_mutex_leave(pTo->db->mutex);
drh4ac285a2006-09-15 07:28:50 +00001066 assert( rc==SQLITE_OK || rc==SQLITE_NOMEM );
drhf8db1bc2005-04-22 02:38:37 +00001067 return rc;
1068}
drh51942bc2005-06-12 22:01:42 +00001069
1070/*
1071** Return the sqlite3* database handle to which the prepared statement given
1072** in the argument belongs. This is the same database handle that was
1073** the first argument to the sqlite3_prepare() that was used to create
1074** the statement in the first place.
1075*/
1076sqlite3 *sqlite3_db_handle(sqlite3_stmt *pStmt){
1077 return pStmt ? ((Vdbe*)pStmt)->db : 0;
1078}