blob: 2ef0d3c11cb8399b08a3d579cc5c5f3fdd1308d7 [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"
drh19e2d372005-08-29 23:00:03 +000018#include "os.h"
drh4f26d6c2004-05-26 23:25:30 +000019
drhd89bd002005-01-22 03:03:54 +000020/*
21** Return TRUE (non-zero) of the statement supplied as an argument needs
22** to be recompiled. A statement needs to be recompiled whenever the
23** execution environment changes in a way that would alter the program
24** that sqlite3_prepare() generates. For example, if new functions or
25** collating sequences are registered or if an authorizer function is
26** added or changed.
drhd89bd002005-01-22 03:03:54 +000027*/
28int sqlite3_expired(sqlite3_stmt *pStmt){
29 Vdbe *p = (Vdbe*)pStmt;
30 return p==0 || p->expired;
31}
32
drh4f26d6c2004-05-26 23:25:30 +000033/**************************** sqlite3_value_ *******************************
34** The following routines extract information from a Mem or sqlite3_value
35** structure.
36*/
37const void *sqlite3_value_blob(sqlite3_value *pVal){
38 Mem *p = (Mem*)pVal;
39 if( p->flags & (MEM_Blob|MEM_Str) ){
drhb026e052007-05-02 01:34:31 +000040 sqlite3VdbeMemExpandBlob(p);
drh1f0feef2007-05-15 13:27:07 +000041 p->flags &= ~MEM_Str;
42 p->flags |= MEM_Blob;
drh4f26d6c2004-05-26 23:25:30 +000043 return p->z;
44 }else{
45 return sqlite3_value_text(pVal);
46 }
47}
48int sqlite3_value_bytes(sqlite3_value *pVal){
danielk197713073932004-06-30 11:54:06 +000049 return sqlite3ValueBytes(pVal, SQLITE_UTF8);
drh4f26d6c2004-05-26 23:25:30 +000050}
51int sqlite3_value_bytes16(sqlite3_value *pVal){
danielk197713073932004-06-30 11:54:06 +000052 return sqlite3ValueBytes(pVal, SQLITE_UTF16NATIVE);
drh4f26d6c2004-05-26 23:25:30 +000053}
54double sqlite3_value_double(sqlite3_value *pVal){
drh6a6124e2004-06-27 01:56:33 +000055 return sqlite3VdbeRealValue((Mem*)pVal);
drh4f26d6c2004-05-26 23:25:30 +000056}
57int sqlite3_value_int(sqlite3_value *pVal){
drh6a6124e2004-06-27 01:56:33 +000058 return sqlite3VdbeIntValue((Mem*)pVal);
drh4f26d6c2004-05-26 23:25:30 +000059}
drhefad9992004-06-22 12:13:55 +000060sqlite_int64 sqlite3_value_int64(sqlite3_value *pVal){
drh6a6124e2004-06-27 01:56:33 +000061 return sqlite3VdbeIntValue((Mem*)pVal);
drh4f26d6c2004-05-26 23:25:30 +000062}
63const unsigned char *sqlite3_value_text(sqlite3_value *pVal){
danielk197700fd9572005-12-07 06:27:43 +000064 return (const unsigned char *)sqlite3ValueText(pVal, SQLITE_UTF8);
drh4f26d6c2004-05-26 23:25:30 +000065}
drh6c626082004-11-14 21:56:29 +000066#ifndef SQLITE_OMIT_UTF16
drh4f26d6c2004-05-26 23:25:30 +000067const void *sqlite3_value_text16(sqlite3_value* pVal){
danielk1977dc8453f2004-06-12 00:42:34 +000068 return sqlite3ValueText(pVal, SQLITE_UTF16NATIVE);
drh4f26d6c2004-05-26 23:25:30 +000069}
danielk1977d8123362004-06-12 09:25:12 +000070const void *sqlite3_value_text16be(sqlite3_value *pVal){
71 return sqlite3ValueText(pVal, SQLITE_UTF16BE);
72}
73const void *sqlite3_value_text16le(sqlite3_value *pVal){
74 return sqlite3ValueText(pVal, SQLITE_UTF16LE);
75}
drh6c626082004-11-14 21:56:29 +000076#endif /* SQLITE_OMIT_UTF16 */
drh4f26d6c2004-05-26 23:25:30 +000077int sqlite3_value_type(sqlite3_value* pVal){
drhf4479502004-05-27 03:12:53 +000078 return pVal->type;
drh4f26d6c2004-05-26 23:25:30 +000079}
drh29d72102006-02-09 22:13:41 +000080/* sqlite3_value_numeric_type() defined in vdbe.c */
drh4f26d6c2004-05-26 23:25:30 +000081
82/**************************** sqlite3_result_ *******************************
83** The following routines are used by user-defined functions to specify
84** the function result.
85*/
86void sqlite3_result_blob(
87 sqlite3_context *pCtx,
88 const void *z,
89 int n,
danielk1977d8123362004-06-12 09:25:12 +000090 void (*xDel)(void *)
drh4f26d6c2004-05-26 23:25:30 +000091){
drh5708d2d2005-06-22 10:53:59 +000092 assert( n>=0 );
danielk1977d8123362004-06-12 09:25:12 +000093 sqlite3VdbeMemSetStr(&pCtx->s, z, n, 0, xDel);
drh4f26d6c2004-05-26 23:25:30 +000094}
95void sqlite3_result_double(sqlite3_context *pCtx, double rVal){
96 sqlite3VdbeMemSetDouble(&pCtx->s, rVal);
97}
98void sqlite3_result_error(sqlite3_context *pCtx, const char *z, int n){
99 pCtx->isError = 1;
danielk1977d8123362004-06-12 09:25:12 +0000100 sqlite3VdbeMemSetStr(&pCtx->s, z, n, SQLITE_UTF8, SQLITE_TRANSIENT);
drh4f26d6c2004-05-26 23:25:30 +0000101}
danielk1977a1686c92006-01-23 07:52:37 +0000102#ifndef SQLITE_OMIT_UTF16
drh4f26d6c2004-05-26 23:25:30 +0000103void sqlite3_result_error16(sqlite3_context *pCtx, const void *z, int n){
104 pCtx->isError = 1;
danielk1977d8123362004-06-12 09:25:12 +0000105 sqlite3VdbeMemSetStr(&pCtx->s, z, n, SQLITE_UTF16NATIVE, SQLITE_TRANSIENT);
drh4f26d6c2004-05-26 23:25:30 +0000106}
danielk1977a1686c92006-01-23 07:52:37 +0000107#endif
drhf4479502004-05-27 03:12:53 +0000108void sqlite3_result_int(sqlite3_context *pCtx, int iVal){
drh4f26d6c2004-05-26 23:25:30 +0000109 sqlite3VdbeMemSetInt64(&pCtx->s, (i64)iVal);
110}
111void sqlite3_result_int64(sqlite3_context *pCtx, i64 iVal){
112 sqlite3VdbeMemSetInt64(&pCtx->s, iVal);
113}
114void sqlite3_result_null(sqlite3_context *pCtx){
drhf4479502004-05-27 03:12:53 +0000115 sqlite3VdbeMemSetNull(&pCtx->s);
drh4f26d6c2004-05-26 23:25:30 +0000116}
117void sqlite3_result_text(
118 sqlite3_context *pCtx,
119 const char *z,
120 int n,
danielk1977d8123362004-06-12 09:25:12 +0000121 void (*xDel)(void *)
drh4f26d6c2004-05-26 23:25:30 +0000122){
danielk1977d8123362004-06-12 09:25:12 +0000123 sqlite3VdbeMemSetStr(&pCtx->s, z, n, SQLITE_UTF8, xDel);
drh4f26d6c2004-05-26 23:25:30 +0000124}
drh6c626082004-11-14 21:56:29 +0000125#ifndef SQLITE_OMIT_UTF16
drh4f26d6c2004-05-26 23:25:30 +0000126void sqlite3_result_text16(
127 sqlite3_context *pCtx,
128 const void *z,
129 int n,
danielk1977d8123362004-06-12 09:25:12 +0000130 void (*xDel)(void *)
drh4f26d6c2004-05-26 23:25:30 +0000131){
danielk1977d8123362004-06-12 09:25:12 +0000132 sqlite3VdbeMemSetStr(&pCtx->s, z, n, SQLITE_UTF16NATIVE, xDel);
133}
134void sqlite3_result_text16be(
135 sqlite3_context *pCtx,
136 const void *z,
137 int n,
138 void (*xDel)(void *)
139){
140 sqlite3VdbeMemSetStr(&pCtx->s, z, n, SQLITE_UTF16BE, xDel);
141}
142void sqlite3_result_text16le(
143 sqlite3_context *pCtx,
144 const void *z,
145 int n,
146 void (*xDel)(void *)
147){
148 sqlite3VdbeMemSetStr(&pCtx->s, z, n, SQLITE_UTF16LE, xDel);
drh4f26d6c2004-05-26 23:25:30 +0000149}
drh6c626082004-11-14 21:56:29 +0000150#endif /* SQLITE_OMIT_UTF16 */
drh4f26d6c2004-05-26 23:25:30 +0000151void sqlite3_result_value(sqlite3_context *pCtx, sqlite3_value *pValue){
152 sqlite3VdbeMemCopy(&pCtx->s, pValue);
153}
drhb026e052007-05-02 01:34:31 +0000154void sqlite3_result_zeroblob(sqlite3_context *pCtx, int n){
155 sqlite3VdbeMemSetZeroBlob(&pCtx->s, n);
156}
drh4f26d6c2004-05-26 23:25:30 +0000157
drha0206bc2007-05-08 15:15:02 +0000158/* Force an SQLITE_TOOBIG error. */
159void sqlite3_result_error_toobig(sqlite3_context *pCtx){
160 sqlite3VdbeMemSetZeroBlob(&pCtx->s, SQLITE_MAX_LENGTH+1);
161}
162
drh4f26d6c2004-05-26 23:25:30 +0000163
164/*
165** Execute the statement pStmt, either until a row of data is ready, the
166** statement is completely executed or an error occurs.
drhb900aaf2006-11-09 00:24:53 +0000167**
168** This routine implements the bulk of the logic behind the sqlite_step()
169** API. The only thing omitted is the automatic recompile if a
170** schema change has occurred. That detail is handled by the
171** outer sqlite3_step() wrapper procedure.
drh4f26d6c2004-05-26 23:25:30 +0000172*/
drhb900aaf2006-11-09 00:24:53 +0000173static int sqlite3Step(Vdbe *p){
drh9bb575f2004-09-06 17:24:11 +0000174 sqlite3 *db;
drh4f26d6c2004-05-26 23:25:30 +0000175 int rc;
176
danielk1977efaaf572006-01-16 11:29:19 +0000177 /* Assert that malloc() has not failed */
danielk19779e128002006-01-18 16:51:35 +0000178 assert( !sqlite3MallocFailed() );
danielk1977261919c2005-12-06 12:52:59 +0000179
drh1bcdb0c2004-08-28 14:49:46 +0000180 if( p==0 || p->magic!=VDBE_MAGIC_RUN ){
drh4f26d6c2004-05-26 23:25:30 +0000181 return SQLITE_MISUSE;
182 }
drh91b48aa2004-06-30 11:14:18 +0000183 if( p->aborted ){
184 return SQLITE_ABORT;
185 }
danielk1977a21c6b62005-01-24 10:25:59 +0000186 if( p->pc<=0 && p->expired ){
187 if( p->rc==SQLITE_OK ){
188 p->rc = SQLITE_SCHEMA;
189 }
drhb900aaf2006-11-09 00:24:53 +0000190 rc = SQLITE_ERROR;
191 goto end_of_step;
danielk1977a21c6b62005-01-24 10:25:59 +0000192 }
drh4f26d6c2004-05-26 23:25:30 +0000193 db = p->db;
194 if( sqlite3SafetyOn(db) ){
195 p->rc = SQLITE_MISUSE;
196 return SQLITE_MISUSE;
197 }
danielk19771d850a72004-05-31 08:26:49 +0000198 if( p->pc<0 ){
drh15ca1df2006-07-26 13:43:30 +0000199 /* If there are no other statements currently running, then
200 ** reset the interrupt flag. This prevents a call to sqlite3_interrupt
201 ** from interrupting a statement that has not yet started.
202 */
203 if( db->activeVdbeCnt==0 ){
204 db->u1.isInterrupted = 0;
205 }
206
drh19e2d372005-08-29 23:00:03 +0000207#ifndef SQLITE_OMIT_TRACE
drhc16a03b2004-09-15 13:38:10 +0000208 /* Invoke the trace callback if there is one
209 */
drh19e2d372005-08-29 23:00:03 +0000210 if( db->xTrace && !db->init.busy ){
drhc16a03b2004-09-15 13:38:10 +0000211 assert( p->nOp>0 );
212 assert( p->aOp[p->nOp-1].opcode==OP_Noop );
213 assert( p->aOp[p->nOp-1].p3!=0 );
214 assert( p->aOp[p->nOp-1].p3type==P3_DYNAMIC );
215 sqlite3SafetyOff(db);
216 db->xTrace(db->pTraceArg, p->aOp[p->nOp-1].p3);
217 if( sqlite3SafetyOn(db) ){
218 p->rc = SQLITE_MISUSE;
219 return SQLITE_MISUSE;
220 }
221 }
drh19e2d372005-08-29 23:00:03 +0000222 if( db->xProfile && !db->init.busy ){
223 double rNow;
drh66560ad2006-01-06 14:32:19 +0000224 sqlite3OsCurrentTime(&rNow);
drh19e2d372005-08-29 23:00:03 +0000225 p->startTime = (rNow - (int)rNow)*3600.0*24.0*1000000000.0;
226 }
227#endif
drhc16a03b2004-09-15 13:38:10 +0000228
229 /* Print a copy of SQL as it is executed if the SQL_TRACE pragma is turned
230 ** on in debugging mode.
231 */
232#ifdef SQLITE_DEBUG
233 if( (db->flags & SQLITE_SqlTrace)!=0 ){
234 sqlite3DebugPrintf("SQL-trace: %s\n", p->aOp[p->nOp-1].p3);
235 }
236#endif /* SQLITE_DEBUG */
237
danielk19771d850a72004-05-31 08:26:49 +0000238 db->activeVdbeCnt++;
239 p->pc = 0;
240 }
drhb7f91642004-10-31 02:22:47 +0000241#ifndef SQLITE_OMIT_EXPLAIN
drh4f26d6c2004-05-26 23:25:30 +0000242 if( p->explain ){
243 rc = sqlite3VdbeList(p);
drhb7f91642004-10-31 02:22:47 +0000244 }else
245#endif /* SQLITE_OMIT_EXPLAIN */
246 {
drh4f26d6c2004-05-26 23:25:30 +0000247 rc = sqlite3VdbeExec(p);
248 }
249
250 if( sqlite3SafetyOff(db) ){
251 rc = SQLITE_MISUSE;
252 }
253
drh19e2d372005-08-29 23:00:03 +0000254#ifndef SQLITE_OMIT_TRACE
255 /* Invoke the profile callback if there is one
256 */
257 if( rc!=SQLITE_ROW && db->xProfile && !db->init.busy ){
258 double rNow;
259 u64 elapseTime;
260
drh66560ad2006-01-06 14:32:19 +0000261 sqlite3OsCurrentTime(&rNow);
drh19e2d372005-08-29 23:00:03 +0000262 elapseTime = (rNow - (int)rNow)*3600.0*24.0*1000000000.0 - p->startTime;
263 assert( p->nOp>0 );
264 assert( p->aOp[p->nOp-1].opcode==OP_Noop );
265 assert( p->aOp[p->nOp-1].p3!=0 );
266 assert( p->aOp[p->nOp-1].p3type==P3_DYNAMIC );
267 db->xProfile(db->pProfileArg, p->aOp[p->nOp-1].p3, elapseTime);
268 }
269#endif
270
danielk197797a227c2006-01-20 16:32:04 +0000271 sqlite3Error(p->db, rc, 0);
danielk197776e8d1a2006-01-18 18:22:43 +0000272 p->rc = sqlite3ApiExit(p->db, p->rc);
drhb900aaf2006-11-09 00:24:53 +0000273end_of_step:
drh4ac285a2006-09-15 07:28:50 +0000274 assert( (rc&0xff)==rc );
drhb900aaf2006-11-09 00:24:53 +0000275 if( p->zSql && (rc&0xff)<SQLITE_ROW ){
276 /* This behavior occurs if sqlite3_prepare_v2() was used to build
277 ** the prepared statement. Return error codes directly */
danielk1977612642d2007-07-12 13:18:05 +0000278 sqlite3Error(p->db, p->rc, 0);
drhb900aaf2006-11-09 00:24:53 +0000279 return p->rc;
280 }else{
281 /* This is for legacy sqlite3_prepare() builds and when the code
282 ** is SQLITE_ROW or SQLITE_DONE */
283 return rc;
284 }
285}
286
287/*
288** This is the top-level implementation of sqlite3_step(). Call
289** sqlite3Step() to do most of the work. If a schema error occurs,
290** call sqlite3Reprepare() and try again.
291*/
292#ifdef SQLITE_OMIT_PARSER
293int sqlite3_step(sqlite3_stmt *pStmt){
294 return sqlite3Step((Vdbe*)pStmt);
295}
296#else
297int sqlite3_step(sqlite3_stmt *pStmt){
298 int cnt = 0;
299 int rc;
300 Vdbe *v = (Vdbe*)pStmt;
301 while( (rc = sqlite3Step(v))==SQLITE_SCHEMA
302 && cnt++ < 5
303 && sqlite3Reprepare(v) ){
304 sqlite3_reset(pStmt);
305 v->expired = 0;
306 }
danielk197776e8d1a2006-01-18 18:22:43 +0000307 return rc;
drh4f26d6c2004-05-26 23:25:30 +0000308}
drhb900aaf2006-11-09 00:24:53 +0000309#endif
drh4f26d6c2004-05-26 23:25:30 +0000310
311/*
drheb2e1762004-05-27 01:53:56 +0000312** Extract the user data from a sqlite3_context structure and return a
313** pointer to it.
314*/
315void *sqlite3_user_data(sqlite3_context *p){
316 assert( p && p->pFunc );
317 return p->pFunc->pUserData;
318}
319
320/*
drhb7481e72006-09-16 21:45:14 +0000321** The following is the implementation of an SQL function that always
322** fails with an error message stating that the function is used in the
323** wrong context. The sqlite3_overload_function() API might construct
324** SQL function that use this routine so that the functions will exist
325** for name resolution but are actually overloaded by the xFindFunction
326** method of virtual tables.
327*/
328void sqlite3InvalidFunction(
329 sqlite3_context *context, /* The function calling context */
330 int argc, /* Number of arguments to the function */
331 sqlite3_value **argv /* Value of each argument */
332){
333 const char *zName = context->pFunc->zName;
334 char *zErr;
335 zErr = sqlite3MPrintf(
336 "unable to use function %s in the requested context", zName);
337 sqlite3_result_error(context, zErr, -1);
338 sqliteFree(zErr);
339}
340
341/*
drheb2e1762004-05-27 01:53:56 +0000342** Allocate or return the aggregate context for a user function. A new
343** context is allocated on the first call. Subsequent calls return the
344** same context that was returned on prior calls.
drheb2e1762004-05-27 01:53:56 +0000345*/
346void *sqlite3_aggregate_context(sqlite3_context *p, int nByte){
drhabfcea22005-09-06 20:36:48 +0000347 Mem *pMem = p->pMem;
drh825c6622005-09-08 19:01:05 +0000348 assert( p && p->pFunc && p->pFunc->xStep );
drha10a34b2005-09-07 22:09:48 +0000349 if( (pMem->flags & MEM_Agg)==0 ){
350 if( nByte==0 ){
351 assert( pMem->flags==MEM_Null );
352 pMem->z = 0;
drheb2e1762004-05-27 01:53:56 +0000353 }else{
drha10a34b2005-09-07 22:09:48 +0000354 pMem->flags = MEM_Agg;
drh5360ad32005-09-08 00:13:27 +0000355 pMem->xDel = sqlite3FreeX;
drh3c024d62007-03-30 11:23:45 +0000356 pMem->u.pDef = p->pFunc;
drha10a34b2005-09-07 22:09:48 +0000357 if( nByte<=NBFS ){
358 pMem->z = pMem->zShort;
359 memset(pMem->z, 0, nByte);
360 }else{
361 pMem->z = sqliteMalloc( nByte );
362 }
drheb2e1762004-05-27 01:53:56 +0000363 }
364 }
drhabfcea22005-09-06 20:36:48 +0000365 return (void*)pMem->z;
drheb2e1762004-05-27 01:53:56 +0000366}
367
368/*
danielk1977682f68b2004-06-05 10:22:17 +0000369** Return the auxilary data pointer, if any, for the iArg'th argument to
370** the user-function defined by pCtx.
371*/
372void *sqlite3_get_auxdata(sqlite3_context *pCtx, int iArg){
373 VdbeFunc *pVdbeFunc = pCtx->pVdbeFunc;
374 if( !pVdbeFunc || iArg>=pVdbeFunc->nAux || iArg<0 ){
375 return 0;
376 }
drhf92c7ff2004-06-19 15:40:23 +0000377 return pVdbeFunc->apAux[iArg].pAux;
danielk1977682f68b2004-06-05 10:22:17 +0000378}
379
380/*
381** Set the auxilary data pointer and delete function, for the iArg'th
382** argument to the user-function defined by pCtx. Any previous value is
383** deleted by calling the delete function specified when it was set.
384*/
385void sqlite3_set_auxdata(
386 sqlite3_context *pCtx,
387 int iArg,
388 void *pAux,
389 void (*xDelete)(void*)
390){
391 struct AuxData *pAuxData;
drhf92c7ff2004-06-19 15:40:23 +0000392 VdbeFunc *pVdbeFunc;
danielk1977e0fc5262007-07-26 06:50:05 +0000393 if( iArg<0 ) goto failed;
danielk1977682f68b2004-06-05 10:22:17 +0000394
drhf92c7ff2004-06-19 15:40:23 +0000395 pVdbeFunc = pCtx->pVdbeFunc;
396 if( !pVdbeFunc || pVdbeFunc->nAux<=iArg ){
397 int nMalloc = sizeof(VdbeFunc) + sizeof(struct AuxData)*iArg;
drh53f733c2005-09-16 02:38:09 +0000398 pVdbeFunc = sqliteRealloc(pVdbeFunc, nMalloc);
danielk1977e0fc5262007-07-26 06:50:05 +0000399 if( !pVdbeFunc ) goto failed;
drh53f733c2005-09-16 02:38:09 +0000400 pCtx->pVdbeFunc = pVdbeFunc;
drh0e3d7472004-06-19 17:33:07 +0000401 memset(&pVdbeFunc->apAux[pVdbeFunc->nAux], 0,
402 sizeof(struct AuxData)*(iArg+1-pVdbeFunc->nAux));
drh998da3a2004-06-19 15:22:56 +0000403 pVdbeFunc->nAux = iArg+1;
404 pVdbeFunc->pFunc = pCtx->pFunc;
danielk1977682f68b2004-06-05 10:22:17 +0000405 }
406
drhf92c7ff2004-06-19 15:40:23 +0000407 pAuxData = &pVdbeFunc->apAux[iArg];
danielk1977682f68b2004-06-05 10:22:17 +0000408 if( pAuxData->pAux && pAuxData->xDelete ){
409 pAuxData->xDelete(pAuxData->pAux);
410 }
411 pAuxData->pAux = pAux;
412 pAuxData->xDelete = xDelete;
danielk1977e0fc5262007-07-26 06:50:05 +0000413 return;
414
415failed:
416 if( xDelete ){
417 xDelete(pAux);
418 }
danielk1977682f68b2004-06-05 10:22:17 +0000419}
420
421/*
drheb2e1762004-05-27 01:53:56 +0000422** Return the number of times the Step function of a aggregate has been
423** called.
424**
drhcf85a512006-02-09 18:35:29 +0000425** This function is deprecated. Do not use it for new code. It is
426** provide only to avoid breaking legacy code. New aggregate function
427** implementations should keep their own counts within their aggregate
428** context.
drheb2e1762004-05-27 01:53:56 +0000429*/
430int sqlite3_aggregate_count(sqlite3_context *p){
431 assert( p && p->pFunc && p->pFunc->xStep );
drhabfcea22005-09-06 20:36:48 +0000432 return p->pMem->n;
drheb2e1762004-05-27 01:53:56 +0000433}
434
435/*
drh4f26d6c2004-05-26 23:25:30 +0000436** Return the number of columns in the result set for the statement pStmt.
437*/
438int sqlite3_column_count(sqlite3_stmt *pStmt){
439 Vdbe *pVm = (Vdbe *)pStmt;
drh1bcdb0c2004-08-28 14:49:46 +0000440 return pVm ? pVm->nResColumn : 0;
drh4f26d6c2004-05-26 23:25:30 +0000441}
442
443/*
444** Return the number of values available from the current row of the
445** currently executing statement pStmt.
446*/
447int sqlite3_data_count(sqlite3_stmt *pStmt){
448 Vdbe *pVm = (Vdbe *)pStmt;
drh1bcdb0c2004-08-28 14:49:46 +0000449 if( pVm==0 || !pVm->resOnStack ) return 0;
drh4f26d6c2004-05-26 23:25:30 +0000450 return pVm->nResColumn;
451}
452
453
454/*
455** Check to see if column iCol of the given statement is valid. If
456** it is, return a pointer to the Mem for the value of that column.
457** If iCol is not valid, return a pointer to a Mem which has a value
458** of NULL.
459*/
460static Mem *columnMem(sqlite3_stmt *pStmt, int i){
461 Vdbe *pVm = (Vdbe *)pStmt;
462 int vals = sqlite3_data_count(pStmt);
drhb21f87d2007-06-19 10:58:24 +0000463 if( pVm==0 || pVm->resOnStack==0 || i>=pVm->nResColumn || i<0 ){
drh7a521cf2007-04-25 18:23:52 +0000464 static const Mem nullMem = {{0}, 0.0, "", 0, MEM_Null, SQLITE_NULL };
drh4f26d6c2004-05-26 23:25:30 +0000465 sqlite3Error(pVm->db, SQLITE_RANGE, 0);
drh0f7eb612006-08-08 13:51:43 +0000466 return (Mem*)&nullMem;
drh4f26d6c2004-05-26 23:25:30 +0000467 }
468 return &pVm->pTos[(1-vals)+i];
469}
470
danielk19772e588c72005-12-09 14:25:08 +0000471/*
472** This function is called after invoking an sqlite3_value_XXX function on a
473** column value (i.e. a value returned by evaluating an SQL expression in the
474** select list of a SELECT statement) that may cause a malloc() failure. If
475** malloc() has failed, the threads mallocFailed flag is cleared and the result
476** code of statement pStmt set to SQLITE_NOMEM.
477**
478** Specificly, this is called from within:
479**
480** sqlite3_column_int()
481** sqlite3_column_int64()
482** sqlite3_column_text()
483** sqlite3_column_text16()
484** sqlite3_column_real()
485** sqlite3_column_bytes()
486** sqlite3_column_bytes16()
487**
488** But not for sqlite3_column_blob(), which never calls malloc().
489*/
490static void columnMallocFailure(sqlite3_stmt *pStmt)
491{
492 /* If malloc() failed during an encoding conversion within an
493 ** sqlite3_column_XXX API, then set the return code of the statement to
494 ** SQLITE_NOMEM. The next call to _step() (if any) will return SQLITE_ERROR
495 ** and _finalize() will return NOMEM.
496 */
danielk197754f01982006-01-18 15:25:17 +0000497 Vdbe *p = (Vdbe *)pStmt;
498 p->rc = sqlite3ApiExit(0, p->rc);
danielk19772e588c72005-12-09 14:25:08 +0000499}
500
drh4f26d6c2004-05-26 23:25:30 +0000501/**************************** sqlite3_column_ *******************************
502** The following routines are used to access elements of the current row
503** in the result set.
504*/
danielk1977c572ef72004-05-27 09:28:41 +0000505const void *sqlite3_column_blob(sqlite3_stmt *pStmt, int i){
danielk19772e588c72005-12-09 14:25:08 +0000506 const void *val;
danielk19772e588c72005-12-09 14:25:08 +0000507 val = sqlite3_value_blob( columnMem(pStmt,i) );
danielk1977c9cf9012007-05-30 10:36:47 +0000508 /* Even though there is no encoding conversion, value_blob() might
509 ** need to call malloc() to expand the result of a zeroblob()
510 ** expression.
511 */
512 columnMallocFailure(pStmt);
danielk19772e588c72005-12-09 14:25:08 +0000513 return val;
danielk1977c572ef72004-05-27 09:28:41 +0000514}
drh4f26d6c2004-05-26 23:25:30 +0000515int sqlite3_column_bytes(sqlite3_stmt *pStmt, int i){
danielk19772e588c72005-12-09 14:25:08 +0000516 int val = sqlite3_value_bytes( columnMem(pStmt,i) );
517 columnMallocFailure(pStmt);
518 return val;
drh4f26d6c2004-05-26 23:25:30 +0000519}
520int sqlite3_column_bytes16(sqlite3_stmt *pStmt, int i){
danielk19772e588c72005-12-09 14:25:08 +0000521 int val = sqlite3_value_bytes16( columnMem(pStmt,i) );
522 columnMallocFailure(pStmt);
523 return val;
drh4f26d6c2004-05-26 23:25:30 +0000524}
525double sqlite3_column_double(sqlite3_stmt *pStmt, int i){
danielk19772e588c72005-12-09 14:25:08 +0000526 double val = sqlite3_value_double( columnMem(pStmt,i) );
527 columnMallocFailure(pStmt);
528 return val;
drh4f26d6c2004-05-26 23:25:30 +0000529}
530int sqlite3_column_int(sqlite3_stmt *pStmt, int i){
danielk19772e588c72005-12-09 14:25:08 +0000531 int val = sqlite3_value_int( columnMem(pStmt,i) );
532 columnMallocFailure(pStmt);
533 return val;
drh4f26d6c2004-05-26 23:25:30 +0000534}
drhefad9992004-06-22 12:13:55 +0000535sqlite_int64 sqlite3_column_int64(sqlite3_stmt *pStmt, int i){
danielk19772e588c72005-12-09 14:25:08 +0000536 sqlite_int64 val = sqlite3_value_int64( columnMem(pStmt,i) );
537 columnMallocFailure(pStmt);
538 return val;
drh4f26d6c2004-05-26 23:25:30 +0000539}
540const unsigned char *sqlite3_column_text(sqlite3_stmt *pStmt, int i){
danielk19772e588c72005-12-09 14:25:08 +0000541 const unsigned char *val = sqlite3_value_text( columnMem(pStmt,i) );
542 columnMallocFailure(pStmt);
543 return val;
drh4f26d6c2004-05-26 23:25:30 +0000544}
drhd1e47332005-06-26 17:55:33 +0000545sqlite3_value *sqlite3_column_value(sqlite3_stmt *pStmt, int i){
546 return columnMem(pStmt, i);
547}
drh6c626082004-11-14 21:56:29 +0000548#ifndef SQLITE_OMIT_UTF16
drh4f26d6c2004-05-26 23:25:30 +0000549const void *sqlite3_column_text16(sqlite3_stmt *pStmt, int i){
danielk19772e588c72005-12-09 14:25:08 +0000550 const void *val = sqlite3_value_text16( columnMem(pStmt,i) );
551 columnMallocFailure(pStmt);
552 return val;
drh4f26d6c2004-05-26 23:25:30 +0000553}
drh6c626082004-11-14 21:56:29 +0000554#endif /* SQLITE_OMIT_UTF16 */
drh4f26d6c2004-05-26 23:25:30 +0000555int sqlite3_column_type(sqlite3_stmt *pStmt, int i){
556 return sqlite3_value_type( columnMem(pStmt,i) );
557}
558
drh29d72102006-02-09 22:13:41 +0000559/* The following function is experimental and subject to change or
560** removal */
561/*int sqlite3_column_numeric_type(sqlite3_stmt *pStmt, int i){
562** return sqlite3_value_numeric_type( columnMem(pStmt,i) );
563**}
564*/
565
drh5e2517e2004-09-24 23:59:12 +0000566/*
567** Convert the N-th element of pStmt->pColName[] into a string using
568** xFunc() then return that string. If N is out of range, return 0.
drhe590fbd2005-05-20 19:36:01 +0000569**
570** There are up to 5 names for each column. useType determines which
571** name is returned. Here are the names:
572**
573** 0 The column name as it should be displayed for output
574** 1 The datatype name for the column
575** 2 The name of the database that the column derives from
576** 3 The name of the table that the column derives from
577** 4 The name of the table column that the result column derives from
578**
579** If the result is not a simple column reference (if it is an expression
580** or a constant) then useTypes 2, 3, and 4 return NULL.
drh5e2517e2004-09-24 23:59:12 +0000581*/
582static const void *columnName(
583 sqlite3_stmt *pStmt,
584 int N,
585 const void *(*xFunc)(Mem*),
586 int useType
587){
danielk197700fd9572005-12-07 06:27:43 +0000588 const void *ret;
drh5e2517e2004-09-24 23:59:12 +0000589 Vdbe *p = (Vdbe *)pStmt;
590 int n = sqlite3_column_count(pStmt);
591
592 if( p==0 || N>=n || N<0 ){
593 return 0;
594 }
drhe590fbd2005-05-20 19:36:01 +0000595 N += useType*n;
danielk197700fd9572005-12-07 06:27:43 +0000596 ret = xFunc(&p->aColName[N]);
597
598 /* A malloc may have failed inside of the xFunc() call. If this is the case,
599 ** clear the mallocFailed flag and return NULL.
600 */
danielk197754f01982006-01-18 15:25:17 +0000601 sqlite3ApiExit(0, 0);
danielk197700fd9572005-12-07 06:27:43 +0000602 return ret;
drh5e2517e2004-09-24 23:59:12 +0000603}
604
drh4f26d6c2004-05-26 23:25:30 +0000605/*
606** Return the name of the Nth column of the result set returned by SQL
607** statement pStmt.
608*/
609const char *sqlite3_column_name(sqlite3_stmt *pStmt, int N){
danielk1977955de522006-02-10 02:27:42 +0000610 return columnName(
611 pStmt, N, (const void*(*)(Mem*))sqlite3_value_text, COLNAME_NAME);
drh4f26d6c2004-05-26 23:25:30 +0000612}
drhe590fbd2005-05-20 19:36:01 +0000613#ifndef SQLITE_OMIT_UTF16
614const void *sqlite3_column_name16(sqlite3_stmt *pStmt, int N){
danielk1977955de522006-02-10 02:27:42 +0000615 return columnName(
616 pStmt, N, (const void*(*)(Mem*))sqlite3_value_text16, COLNAME_NAME);
drhe590fbd2005-05-20 19:36:01 +0000617}
618#endif
drh4f26d6c2004-05-26 23:25:30 +0000619
620/*
drh6c626082004-11-14 21:56:29 +0000621** Return the column declaration type (if applicable) of the 'i'th column
drhe590fbd2005-05-20 19:36:01 +0000622** of the result set of SQL statement pStmt.
drh6c626082004-11-14 21:56:29 +0000623*/
624const char *sqlite3_column_decltype(sqlite3_stmt *pStmt, int N){
danielk1977955de522006-02-10 02:27:42 +0000625 return columnName(
626 pStmt, N, (const void*(*)(Mem*))sqlite3_value_text, COLNAME_DECLTYPE);
drh6c626082004-11-14 21:56:29 +0000627}
drh6c626082004-11-14 21:56:29 +0000628#ifndef SQLITE_OMIT_UTF16
danielk197776d505b2004-05-28 13:13:02 +0000629const void *sqlite3_column_decltype16(sqlite3_stmt *pStmt, int N){
danielk1977955de522006-02-10 02:27:42 +0000630 return columnName(
631 pStmt, N, (const void*(*)(Mem*))sqlite3_value_text16, COLNAME_DECLTYPE);
drh4f26d6c2004-05-26 23:25:30 +0000632}
drh6c626082004-11-14 21:56:29 +0000633#endif /* SQLITE_OMIT_UTF16 */
drh4f26d6c2004-05-26 23:25:30 +0000634
danielk19774b1ae992006-02-10 03:06:10 +0000635#ifdef SQLITE_ENABLE_COLUMN_METADATA
drhe590fbd2005-05-20 19:36:01 +0000636/*
637** Return the name of the database from which a result column derives.
638** NULL is returned if the result column is an expression or constant or
639** anything else which is not an unabiguous reference to a database column.
640*/
641const char *sqlite3_column_database_name(sqlite3_stmt *pStmt, int N){
danielk1977955de522006-02-10 02:27:42 +0000642 return columnName(
643 pStmt, N, (const void*(*)(Mem*))sqlite3_value_text, COLNAME_DATABASE);
drhe590fbd2005-05-20 19:36:01 +0000644}
645#ifndef SQLITE_OMIT_UTF16
646const void *sqlite3_column_database_name16(sqlite3_stmt *pStmt, int N){
danielk1977955de522006-02-10 02:27:42 +0000647 return columnName(
648 pStmt, N, (const void*(*)(Mem*))sqlite3_value_text16, COLNAME_DATABASE);
drhe590fbd2005-05-20 19:36:01 +0000649}
650#endif /* SQLITE_OMIT_UTF16 */
651
652/*
653** Return the name of the table from which a result column derives.
654** NULL is returned if the result column is an expression or constant or
655** anything else which is not an unabiguous reference to a database column.
656*/
657const char *sqlite3_column_table_name(sqlite3_stmt *pStmt, int N){
danielk1977955de522006-02-10 02:27:42 +0000658 return columnName(
659 pStmt, N, (const void*(*)(Mem*))sqlite3_value_text, COLNAME_TABLE);
drhe590fbd2005-05-20 19:36:01 +0000660}
661#ifndef SQLITE_OMIT_UTF16
662const void *sqlite3_column_table_name16(sqlite3_stmt *pStmt, int N){
danielk1977955de522006-02-10 02:27:42 +0000663 return columnName(
664 pStmt, N, (const void*(*)(Mem*))sqlite3_value_text16, COLNAME_TABLE);
drhe590fbd2005-05-20 19:36:01 +0000665}
666#endif /* SQLITE_OMIT_UTF16 */
667
668/*
669** Return the name of the table column from which a result column derives.
670** NULL is returned if the result column is an expression or constant or
671** anything else which is not an unabiguous reference to a database column.
672*/
673const char *sqlite3_column_origin_name(sqlite3_stmt *pStmt, int N){
danielk1977955de522006-02-10 02:27:42 +0000674 return columnName(
675 pStmt, N, (const void*(*)(Mem*))sqlite3_value_text, COLNAME_COLUMN);
drhe590fbd2005-05-20 19:36:01 +0000676}
677#ifndef SQLITE_OMIT_UTF16
678const void *sqlite3_column_origin_name16(sqlite3_stmt *pStmt, int N){
danielk1977955de522006-02-10 02:27:42 +0000679 return columnName(
680 pStmt, N, (const void*(*)(Mem*))sqlite3_value_text16, COLNAME_COLUMN);
drhe590fbd2005-05-20 19:36:01 +0000681}
682#endif /* SQLITE_OMIT_UTF16 */
danielk19774b1ae992006-02-10 03:06:10 +0000683#endif /* SQLITE_ENABLE_COLUMN_METADATA */
drhe590fbd2005-05-20 19:36:01 +0000684
685
drh4f26d6c2004-05-26 23:25:30 +0000686/******************************* sqlite3_bind_ ***************************
687**
688** Routines used to attach values to wildcards in a compiled SQL statement.
689*/
690/*
691** Unbind the value bound to variable i in virtual machine p. This is the
692** the same as binding a NULL value to the column. If the "i" parameter is
693** out of range, then SQLITE_RANGE is returned. Othewise SQLITE_OK.
694**
695** The error code stored in database p->db is overwritten with the return
696** value in any case.
697*/
698static int vdbeUnbind(Vdbe *p, int i){
699 Mem *pVar;
drh1bcdb0c2004-08-28 14:49:46 +0000700 if( p==0 || p->magic!=VDBE_MAGIC_RUN || p->pc>=0 ){
drh473d1792005-06-06 17:54:55 +0000701 if( p ) sqlite3Error(p->db, SQLITE_MISUSE, 0);
drh4f26d6c2004-05-26 23:25:30 +0000702 return SQLITE_MISUSE;
703 }
704 if( i<1 || i>p->nVar ){
705 sqlite3Error(p->db, SQLITE_RANGE, 0);
706 return SQLITE_RANGE;
707 }
708 i--;
drh290c1942004-08-21 17:54:45 +0000709 pVar = &p->aVar[i];
danielk1977d8123362004-06-12 09:25:12 +0000710 sqlite3VdbeMemRelease(pVar);
drh4f26d6c2004-05-26 23:25:30 +0000711 pVar->flags = MEM_Null;
712 sqlite3Error(p->db, SQLITE_OK, 0);
713 return SQLITE_OK;
714}
715
716/*
drh5e2517e2004-09-24 23:59:12 +0000717** Bind a text or BLOB value.
drh4f26d6c2004-05-26 23:25:30 +0000718*/
drh5e2517e2004-09-24 23:59:12 +0000719static int bindText(
drhf4479502004-05-27 03:12:53 +0000720 sqlite3_stmt *pStmt,
drh4f26d6c2004-05-26 23:25:30 +0000721 int i,
722 const void *zData,
723 int nData,
drh5e2517e2004-09-24 23:59:12 +0000724 void (*xDel)(void*),
725 int encoding
drh4f26d6c2004-05-26 23:25:30 +0000726){
727 Vdbe *p = (Vdbe *)pStmt;
728 Mem *pVar;
729 int rc;
730
731 rc = vdbeUnbind(p, i);
drhfeac5f82004-08-01 00:10:45 +0000732 if( rc || zData==0 ){
drh4f26d6c2004-05-26 23:25:30 +0000733 return rc;
734 }
drh290c1942004-08-21 17:54:45 +0000735 pVar = &p->aVar[i-1];
drh5e2517e2004-09-24 23:59:12 +0000736 rc = sqlite3VdbeMemSetStr(pVar, zData, nData, encoding, xDel);
drh5e2517e2004-09-24 23:59:12 +0000737 if( rc==SQLITE_OK && encoding!=0 ){
danielk197714db2662006-01-09 16:12:04 +0000738 rc = sqlite3VdbeChangeEncoding(pVar, ENC(p->db));
drh5e2517e2004-09-24 23:59:12 +0000739 }
danielk197754f01982006-01-18 15:25:17 +0000740
danielk1977771151b2006-01-17 13:21:40 +0000741 sqlite3Error(((Vdbe *)pStmt)->db, rc, 0);
danielk197754f01982006-01-18 15:25:17 +0000742 return sqlite3ApiExit(((Vdbe *)pStmt)->db, rc);
drh4f26d6c2004-05-26 23:25:30 +0000743}
drh5e2517e2004-09-24 23:59:12 +0000744
745
746/*
747** Bind a blob value to an SQL statement variable.
748*/
749int sqlite3_bind_blob(
750 sqlite3_stmt *pStmt,
751 int i,
752 const void *zData,
753 int nData,
754 void (*xDel)(void*)
755){
756 return bindText(pStmt, i, zData, nData, xDel, 0);
757}
drh4f26d6c2004-05-26 23:25:30 +0000758int sqlite3_bind_double(sqlite3_stmt *pStmt, int i, double rValue){
759 int rc;
760 Vdbe *p = (Vdbe *)pStmt;
drh4f26d6c2004-05-26 23:25:30 +0000761 rc = vdbeUnbind(p, i);
762 if( rc==SQLITE_OK ){
drh290c1942004-08-21 17:54:45 +0000763 sqlite3VdbeMemSetDouble(&p->aVar[i-1], rValue);
drh4f26d6c2004-05-26 23:25:30 +0000764 }
danielk1977f4618892004-06-28 13:09:11 +0000765 return rc;
drh4f26d6c2004-05-26 23:25:30 +0000766}
767int sqlite3_bind_int(sqlite3_stmt *p, int i, int iValue){
drhefad9992004-06-22 12:13:55 +0000768 return sqlite3_bind_int64(p, i, (i64)iValue);
drh4f26d6c2004-05-26 23:25:30 +0000769}
drhefad9992004-06-22 12:13:55 +0000770int sqlite3_bind_int64(sqlite3_stmt *pStmt, int i, sqlite_int64 iValue){
drh4f26d6c2004-05-26 23:25:30 +0000771 int rc;
772 Vdbe *p = (Vdbe *)pStmt;
773 rc = vdbeUnbind(p, i);
774 if( rc==SQLITE_OK ){
drh290c1942004-08-21 17:54:45 +0000775 sqlite3VdbeMemSetInt64(&p->aVar[i-1], iValue);
drh4f26d6c2004-05-26 23:25:30 +0000776 }
777 return rc;
778}
779int sqlite3_bind_null(sqlite3_stmt* p, int i){
780 return vdbeUnbind((Vdbe *)p, i);
781}
782int sqlite3_bind_text(
783 sqlite3_stmt *pStmt,
784 int i,
785 const char *zData,
786 int nData,
danielk1977d8123362004-06-12 09:25:12 +0000787 void (*xDel)(void*)
drh4f26d6c2004-05-26 23:25:30 +0000788){
drh5e2517e2004-09-24 23:59:12 +0000789 return bindText(pStmt, i, zData, nData, xDel, SQLITE_UTF8);
drh4f26d6c2004-05-26 23:25:30 +0000790}
drh6c626082004-11-14 21:56:29 +0000791#ifndef SQLITE_OMIT_UTF16
drh4f26d6c2004-05-26 23:25:30 +0000792int sqlite3_bind_text16(
793 sqlite3_stmt *pStmt,
794 int i,
795 const void *zData,
796 int nData,
danielk1977d8123362004-06-12 09:25:12 +0000797 void (*xDel)(void*)
drh4f26d6c2004-05-26 23:25:30 +0000798){
drh5e2517e2004-09-24 23:59:12 +0000799 return bindText(pStmt, i, zData, nData, xDel, SQLITE_UTF16NATIVE);
drh4f26d6c2004-05-26 23:25:30 +0000800}
drh6c626082004-11-14 21:56:29 +0000801#endif /* SQLITE_OMIT_UTF16 */
danielk197726e41442006-06-14 15:16:35 +0000802int sqlite3_bind_value(sqlite3_stmt *pStmt, int i, const sqlite3_value *pValue){
803 int rc;
804 Vdbe *p = (Vdbe *)pStmt;
805 rc = vdbeUnbind(p, i);
806 if( rc==SQLITE_OK ){
807 sqlite3VdbeMemCopy(&p->aVar[i-1], pValue);
808 }
809 return rc;
810}
drhb026e052007-05-02 01:34:31 +0000811int sqlite3_bind_zeroblob(sqlite3_stmt *pStmt, int i, int n){
812 int rc;
813 Vdbe *p = (Vdbe *)pStmt;
814 rc = vdbeUnbind(p, i);
815 if( rc==SQLITE_OK ){
816 sqlite3VdbeMemSetZeroBlob(&p->aVar[i-1], n);
817 }
818 return rc;
819}
drh75f6a032004-07-15 14:15:00 +0000820
821/*
822** Return the number of wildcards that can be potentially bound to.
823** This routine is added to support DBD::SQLite.
drh75f6a032004-07-15 14:15:00 +0000824*/
825int sqlite3_bind_parameter_count(sqlite3_stmt *pStmt){
drh92febd92004-08-20 18:34:20 +0000826 Vdbe *p = (Vdbe*)pStmt;
827 return p ? p->nVar : 0;
drh75f6a032004-07-15 14:15:00 +0000828}
drh895d7472004-08-20 16:02:39 +0000829
830/*
drhfa6bc002004-09-07 16:19:52 +0000831** Create a mapping from variable numbers to variable names
832** in the Vdbe.azVar[] array, if such a mapping does not already
833** exist.
drh895d7472004-08-20 16:02:39 +0000834*/
drhfa6bc002004-09-07 16:19:52 +0000835static void createVarMap(Vdbe *p){
drh895d7472004-08-20 16:02:39 +0000836 if( !p->okVar ){
837 int j;
838 Op *pOp;
839 for(j=0, pOp=p->aOp; j<p->nOp; j++, pOp++){
840 if( pOp->opcode==OP_Variable ){
841 assert( pOp->p1>0 && pOp->p1<=p->nVar );
842 p->azVar[pOp->p1-1] = pOp->p3;
843 }
844 }
845 p->okVar = 1;
846 }
drhfa6bc002004-09-07 16:19:52 +0000847}
848
849/*
850** Return the name of a wildcard parameter. Return NULL if the index
851** is out of range or if the wildcard is unnamed.
852**
853** The result is always UTF-8.
854*/
855const char *sqlite3_bind_parameter_name(sqlite3_stmt *pStmt, int i){
856 Vdbe *p = (Vdbe*)pStmt;
857 if( p==0 || i<1 || i>p->nVar ){
858 return 0;
859 }
860 createVarMap(p);
drh895d7472004-08-20 16:02:39 +0000861 return p->azVar[i-1];
862}
drhfa6bc002004-09-07 16:19:52 +0000863
864/*
865** Given a wildcard parameter name, return the index of the variable
866** with that name. If there is no variable with the given name,
867** return 0.
868*/
869int sqlite3_bind_parameter_index(sqlite3_stmt *pStmt, const char *zName){
870 Vdbe *p = (Vdbe*)pStmt;
871 int i;
872 if( p==0 ){
873 return 0;
874 }
875 createVarMap(p);
drh6a8903c2004-12-10 18:00:04 +0000876 if( zName ){
877 for(i=0; i<p->nVar; i++){
878 const char *z = p->azVar[i];
879 if( z && strcmp(z,zName)==0 ){
880 return i+1;
881 }
drhfa6bc002004-09-07 16:19:52 +0000882 }
883 }
884 return 0;
885}
drhf8db1bc2005-04-22 02:38:37 +0000886
drh51942bc2005-06-12 22:01:42 +0000887/*
drhf8db1bc2005-04-22 02:38:37 +0000888** Transfer all bindings from the first statement over to the second.
889** If the two statements contain a different number of bindings, then
890** an SQLITE_ERROR is returned.
891*/
892int sqlite3_transfer_bindings(sqlite3_stmt *pFromStmt, sqlite3_stmt *pToStmt){
893 Vdbe *pFrom = (Vdbe*)pFromStmt;
894 Vdbe *pTo = (Vdbe*)pToStmt;
895 int i, rc = SQLITE_OK;
896 if( (pFrom->magic!=VDBE_MAGIC_RUN && pFrom->magic!=VDBE_MAGIC_HALT)
897 || (pTo->magic!=VDBE_MAGIC_RUN && pTo->magic!=VDBE_MAGIC_HALT) ){
898 return SQLITE_MISUSE;
899 }
900 if( pFrom->nVar!=pTo->nVar ){
901 return SQLITE_ERROR;
902 }
903 for(i=0; rc==SQLITE_OK && i<pFrom->nVar; i++){
danielk1977950f0542006-01-18 05:51:57 +0000904 sqlite3MallocDisallow();
drhf8db1bc2005-04-22 02:38:37 +0000905 rc = sqlite3VdbeMemMove(&pTo->aVar[i], &pFrom->aVar[i]);
danielk1977950f0542006-01-18 05:51:57 +0000906 sqlite3MallocAllow();
drhf8db1bc2005-04-22 02:38:37 +0000907 }
drh4ac285a2006-09-15 07:28:50 +0000908 assert( rc==SQLITE_OK || rc==SQLITE_NOMEM );
drhf8db1bc2005-04-22 02:38:37 +0000909 return rc;
910}
drh51942bc2005-06-12 22:01:42 +0000911
912/*
913** Return the sqlite3* database handle to which the prepared statement given
914** in the argument belongs. This is the same database handle that was
915** the first argument to the sqlite3_prepare() that was used to create
916** the statement in the first place.
917*/
918sqlite3 *sqlite3_db_handle(sqlite3_stmt *pStmt){
919 return pStmt ? ((Vdbe*)pStmt)->db : 0;
920}