blob: ffe89b24c80235443ad308e148623447415e680e [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
drh4f26d6c2004-05-26 23:25:30 +000032/**************************** sqlite3_value_ *******************************
33** The following routines extract information from a Mem or sqlite3_value
34** structure.
35*/
36const void *sqlite3_value_blob(sqlite3_value *pVal){
37 Mem *p = (Mem*)pVal;
38 if( p->flags & (MEM_Blob|MEM_Str) ){
danielk19771e536952007-08-16 10:09:01 +000039 sqlite3VdbeMemExpandBlob(0, p);
drh1f0feef2007-05-15 13:27:07 +000040 p->flags &= ~MEM_Str;
41 p->flags |= MEM_Blob;
drh4f26d6c2004-05-26 23:25:30 +000042 return p->z;
43 }else{
44 return sqlite3_value_text(pVal);
45 }
46}
47int sqlite3_value_bytes(sqlite3_value *pVal){
danielk19771e536952007-08-16 10:09:01 +000048 return sqlite3ValueBytes(0, pVal, SQLITE_UTF8);
drh4f26d6c2004-05-26 23:25:30 +000049}
50int sqlite3_value_bytes16(sqlite3_value *pVal){
danielk19771e536952007-08-16 10:09:01 +000051 return sqlite3ValueBytes(0, pVal, SQLITE_UTF16NATIVE);
drh4f26d6c2004-05-26 23:25:30 +000052}
53double sqlite3_value_double(sqlite3_value *pVal){
drh6a6124e2004-06-27 01:56:33 +000054 return sqlite3VdbeRealValue((Mem*)pVal);
drh4f26d6c2004-05-26 23:25:30 +000055}
56int sqlite3_value_int(sqlite3_value *pVal){
drh6a6124e2004-06-27 01:56:33 +000057 return sqlite3VdbeIntValue((Mem*)pVal);
drh4f26d6c2004-05-26 23:25:30 +000058}
drhefad9992004-06-22 12:13:55 +000059sqlite_int64 sqlite3_value_int64(sqlite3_value *pVal){
drh6a6124e2004-06-27 01:56:33 +000060 return sqlite3VdbeIntValue((Mem*)pVal);
drh4f26d6c2004-05-26 23:25:30 +000061}
62const unsigned char *sqlite3_value_text(sqlite3_value *pVal){
danielk19771e536952007-08-16 10:09:01 +000063 return (const unsigned char *)sqlite3ValueText(0, pVal, SQLITE_UTF8);
drh4f26d6c2004-05-26 23:25:30 +000064}
drh6c626082004-11-14 21:56:29 +000065#ifndef SQLITE_OMIT_UTF16
drh4f26d6c2004-05-26 23:25:30 +000066const void *sqlite3_value_text16(sqlite3_value* pVal){
danielk19771e536952007-08-16 10:09:01 +000067 return sqlite3ValueText(0, pVal, SQLITE_UTF16NATIVE);
drh4f26d6c2004-05-26 23:25:30 +000068}
danielk1977d8123362004-06-12 09:25:12 +000069const void *sqlite3_value_text16be(sqlite3_value *pVal){
danielk19771e536952007-08-16 10:09:01 +000070 return sqlite3ValueText(0, pVal, SQLITE_UTF16BE);
danielk1977d8123362004-06-12 09:25:12 +000071}
72const void *sqlite3_value_text16le(sqlite3_value *pVal){
danielk19771e536952007-08-16 10:09:01 +000073 return sqlite3ValueText(0, pVal, SQLITE_UTF16LE);
danielk1977d8123362004-06-12 09:25:12 +000074}
drh6c626082004-11-14 21:56:29 +000075#endif /* SQLITE_OMIT_UTF16 */
drh4f26d6c2004-05-26 23:25:30 +000076int sqlite3_value_type(sqlite3_value* pVal){
drhf4479502004-05-27 03:12:53 +000077 return pVal->type;
drh4f26d6c2004-05-26 23:25:30 +000078}
drh29d72102006-02-09 22:13:41 +000079/* sqlite3_value_numeric_type() defined in vdbe.c */
drh4f26d6c2004-05-26 23:25:30 +000080
81/**************************** sqlite3_result_ *******************************
82** The following routines are used by user-defined functions to specify
83** the function result.
84*/
85void sqlite3_result_blob(
86 sqlite3_context *pCtx,
87 const void *z,
88 int n,
danielk1977d8123362004-06-12 09:25:12 +000089 void (*xDel)(void *)
drh4f26d6c2004-05-26 23:25:30 +000090){
drh5708d2d2005-06-22 10:53:59 +000091 assert( n>=0 );
danielk19771e536952007-08-16 10:09:01 +000092 sqlite3VdbeMemSetStr(0, &pCtx->s, z, n, 0, xDel);
drh4f26d6c2004-05-26 23:25:30 +000093}
94void sqlite3_result_double(sqlite3_context *pCtx, double rVal){
95 sqlite3VdbeMemSetDouble(&pCtx->s, rVal);
96}
97void sqlite3_result_error(sqlite3_context *pCtx, const char *z, int n){
98 pCtx->isError = 1;
danielk19771e536952007-08-16 10:09:01 +000099 sqlite3VdbeMemSetStr(0, &pCtx->s, z, n, SQLITE_UTF8, SQLITE_TRANSIENT);
drh4f26d6c2004-05-26 23:25:30 +0000100}
danielk1977a1686c92006-01-23 07:52:37 +0000101#ifndef SQLITE_OMIT_UTF16
drh4f26d6c2004-05-26 23:25:30 +0000102void sqlite3_result_error16(sqlite3_context *pCtx, const void *z, int n){
103 pCtx->isError = 1;
danielk19771e536952007-08-16 10:09:01 +0000104 sqlite3VdbeMemSetStr(0, &pCtx->s, z, n, SQLITE_UTF16NATIVE, SQLITE_TRANSIENT);
drh4f26d6c2004-05-26 23:25:30 +0000105}
danielk1977a1686c92006-01-23 07:52:37 +0000106#endif
drhf4479502004-05-27 03:12:53 +0000107void sqlite3_result_int(sqlite3_context *pCtx, int iVal){
drh4f26d6c2004-05-26 23:25:30 +0000108 sqlite3VdbeMemSetInt64(&pCtx->s, (i64)iVal);
109}
110void sqlite3_result_int64(sqlite3_context *pCtx, i64 iVal){
111 sqlite3VdbeMemSetInt64(&pCtx->s, iVal);
112}
113void sqlite3_result_null(sqlite3_context *pCtx){
drhf4479502004-05-27 03:12:53 +0000114 sqlite3VdbeMemSetNull(&pCtx->s);
drh4f26d6c2004-05-26 23:25:30 +0000115}
116void sqlite3_result_text(
117 sqlite3_context *pCtx,
118 const char *z,
119 int n,
danielk1977d8123362004-06-12 09:25:12 +0000120 void (*xDel)(void *)
drh4f26d6c2004-05-26 23:25:30 +0000121){
danielk19771e536952007-08-16 10:09:01 +0000122 sqlite3VdbeMemSetStr(0, &pCtx->s, z, n, SQLITE_UTF8, xDel);
drh4f26d6c2004-05-26 23:25:30 +0000123}
drh6c626082004-11-14 21:56:29 +0000124#ifndef SQLITE_OMIT_UTF16
drh4f26d6c2004-05-26 23:25:30 +0000125void sqlite3_result_text16(
126 sqlite3_context *pCtx,
127 const void *z,
128 int n,
danielk1977d8123362004-06-12 09:25:12 +0000129 void (*xDel)(void *)
drh4f26d6c2004-05-26 23:25:30 +0000130){
danielk19771e536952007-08-16 10:09:01 +0000131 sqlite3VdbeMemSetStr(0, &pCtx->s, z, n, SQLITE_UTF16NATIVE, xDel);
danielk1977d8123362004-06-12 09:25:12 +0000132}
133void sqlite3_result_text16be(
134 sqlite3_context *pCtx,
135 const void *z,
136 int n,
137 void (*xDel)(void *)
138){
danielk19771e536952007-08-16 10:09:01 +0000139 sqlite3VdbeMemSetStr(0, &pCtx->s, z, n, SQLITE_UTF16BE, xDel);
danielk1977d8123362004-06-12 09:25:12 +0000140}
141void sqlite3_result_text16le(
142 sqlite3_context *pCtx,
143 const void *z,
144 int n,
145 void (*xDel)(void *)
146){
danielk19771e536952007-08-16 10:09:01 +0000147 sqlite3VdbeMemSetStr(0, &pCtx->s, z, n, SQLITE_UTF16LE, xDel);
drh4f26d6c2004-05-26 23:25:30 +0000148}
drh6c626082004-11-14 21:56:29 +0000149#endif /* SQLITE_OMIT_UTF16 */
drh4f26d6c2004-05-26 23:25:30 +0000150void sqlite3_result_value(sqlite3_context *pCtx, sqlite3_value *pValue){
danielk19771e536952007-08-16 10:09:01 +0000151 sqlite3VdbeMemCopy(0, &pCtx->s, pValue);
drh4f26d6c2004-05-26 23:25:30 +0000152}
drhb026e052007-05-02 01:34:31 +0000153void sqlite3_result_zeroblob(sqlite3_context *pCtx, int n){
154 sqlite3VdbeMemSetZeroBlob(&pCtx->s, n);
155}
drh4f26d6c2004-05-26 23:25:30 +0000156
drha0206bc2007-05-08 15:15:02 +0000157/* Force an SQLITE_TOOBIG error. */
158void sqlite3_result_error_toobig(sqlite3_context *pCtx){
159 sqlite3VdbeMemSetZeroBlob(&pCtx->s, SQLITE_MAX_LENGTH+1);
160}
161
drh4f26d6c2004-05-26 23:25:30 +0000162
163/*
164** Execute the statement pStmt, either until a row of data is ready, the
165** statement is completely executed or an error occurs.
drhb900aaf2006-11-09 00:24:53 +0000166**
167** This routine implements the bulk of the logic behind the sqlite_step()
168** API. The only thing omitted is the automatic recompile if a
169** schema change has occurred. That detail is handled by the
170** outer sqlite3_step() wrapper procedure.
drh4f26d6c2004-05-26 23:25:30 +0000171*/
drhb900aaf2006-11-09 00:24:53 +0000172static int sqlite3Step(Vdbe *p){
drh9bb575f2004-09-06 17:24:11 +0000173 sqlite3 *db;
drh4f26d6c2004-05-26 23:25:30 +0000174 int rc;
175
danielk1977efaaf572006-01-16 11:29:19 +0000176 /* Assert that malloc() has not failed */
drh17435752007-08-16 04:30:38 +0000177 db = p->db;
178 assert( !db->mallocFailed );
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 if( sqlite3SafetyOn(db) ){
194 p->rc = SQLITE_MISUSE;
195 return SQLITE_MISUSE;
196 }
danielk19771d850a72004-05-31 08:26:49 +0000197 if( p->pc<0 ){
drh15ca1df2006-07-26 13:43:30 +0000198 /* If there are no other statements currently running, then
199 ** reset the interrupt flag. This prevents a call to sqlite3_interrupt
200 ** from interrupting a statement that has not yet started.
201 */
202 if( db->activeVdbeCnt==0 ){
203 db->u1.isInterrupted = 0;
204 }
205
drh19e2d372005-08-29 23:00:03 +0000206#ifndef SQLITE_OMIT_TRACE
drhc16a03b2004-09-15 13:38:10 +0000207 /* Invoke the trace callback if there is one
208 */
drh19e2d372005-08-29 23:00:03 +0000209 if( db->xTrace && !db->init.busy ){
drhc16a03b2004-09-15 13:38:10 +0000210 assert( p->nOp>0 );
211 assert( p->aOp[p->nOp-1].opcode==OP_Noop );
212 assert( p->aOp[p->nOp-1].p3!=0 );
213 assert( p->aOp[p->nOp-1].p3type==P3_DYNAMIC );
214 sqlite3SafetyOff(db);
215 db->xTrace(db->pTraceArg, p->aOp[p->nOp-1].p3);
216 if( sqlite3SafetyOn(db) ){
217 p->rc = SQLITE_MISUSE;
218 return SQLITE_MISUSE;
219 }
220 }
drh19e2d372005-08-29 23:00:03 +0000221 if( db->xProfile && !db->init.busy ){
222 double rNow;
danielk1977b4b47412007-08-17 15:53:36 +0000223 sqlite3OsCurrentTime(db->pVfs, &rNow);
drh19e2d372005-08-29 23:00:03 +0000224 p->startTime = (rNow - (int)rNow)*3600.0*24.0*1000000000.0;
225 }
226#endif
drhc16a03b2004-09-15 13:38:10 +0000227
228 /* Print a copy of SQL as it is executed if the SQL_TRACE pragma is turned
229 ** on in debugging mode.
230 */
231#ifdef SQLITE_DEBUG
232 if( (db->flags & SQLITE_SqlTrace)!=0 ){
233 sqlite3DebugPrintf("SQL-trace: %s\n", p->aOp[p->nOp-1].p3);
234 }
235#endif /* SQLITE_DEBUG */
236
danielk19771d850a72004-05-31 08:26:49 +0000237 db->activeVdbeCnt++;
238 p->pc = 0;
239 }
drhb7f91642004-10-31 02:22:47 +0000240#ifndef SQLITE_OMIT_EXPLAIN
drh4f26d6c2004-05-26 23:25:30 +0000241 if( p->explain ){
242 rc = sqlite3VdbeList(p);
drhb7f91642004-10-31 02:22:47 +0000243 }else
244#endif /* SQLITE_OMIT_EXPLAIN */
245 {
drh4f26d6c2004-05-26 23:25:30 +0000246 rc = sqlite3VdbeExec(p);
247 }
248
249 if( sqlite3SafetyOff(db) ){
250 rc = SQLITE_MISUSE;
251 }
252
drh19e2d372005-08-29 23:00:03 +0000253#ifndef SQLITE_OMIT_TRACE
254 /* Invoke the profile callback if there is one
255 */
256 if( rc!=SQLITE_ROW && db->xProfile && !db->init.busy ){
257 double rNow;
258 u64 elapseTime;
259
danielk1977b4b47412007-08-17 15:53:36 +0000260 sqlite3OsCurrentTime(db->pVfs, &rNow);
drh19e2d372005-08-29 23:00:03 +0000261 elapseTime = (rNow - (int)rNow)*3600.0*24.0*1000000000.0 - p->startTime;
262 assert( p->nOp>0 );
263 assert( p->aOp[p->nOp-1].opcode==OP_Noop );
264 assert( p->aOp[p->nOp-1].p3!=0 );
265 assert( p->aOp[p->nOp-1].p3type==P3_DYNAMIC );
266 db->xProfile(db->pProfileArg, p->aOp[p->nOp-1].p3, elapseTime);
267 }
268#endif
269
danielk197797a227c2006-01-20 16:32:04 +0000270 sqlite3Error(p->db, rc, 0);
danielk197776e8d1a2006-01-18 18:22:43 +0000271 p->rc = sqlite3ApiExit(p->db, p->rc);
drhb900aaf2006-11-09 00:24:53 +0000272end_of_step:
drh4ac285a2006-09-15 07:28:50 +0000273 assert( (rc&0xff)==rc );
drhb900aaf2006-11-09 00:24:53 +0000274 if( p->zSql && (rc&0xff)<SQLITE_ROW ){
275 /* This behavior occurs if sqlite3_prepare_v2() was used to build
276 ** the prepared statement. Return error codes directly */
danielk1977612642d2007-07-12 13:18:05 +0000277 sqlite3Error(p->db, p->rc, 0);
drhb900aaf2006-11-09 00:24:53 +0000278 return p->rc;
279 }else{
280 /* This is for legacy sqlite3_prepare() builds and when the code
281 ** is SQLITE_ROW or SQLITE_DONE */
282 return rc;
283 }
284}
285
286/*
287** This is the top-level implementation of sqlite3_step(). Call
288** sqlite3Step() to do most of the work. If a schema error occurs,
289** call sqlite3Reprepare() and try again.
290*/
291#ifdef SQLITE_OMIT_PARSER
292int sqlite3_step(sqlite3_stmt *pStmt){
293 return sqlite3Step((Vdbe*)pStmt);
294}
295#else
296int sqlite3_step(sqlite3_stmt *pStmt){
297 int cnt = 0;
298 int rc;
299 Vdbe *v = (Vdbe*)pStmt;
300 while( (rc = sqlite3Step(v))==SQLITE_SCHEMA
301 && cnt++ < 5
302 && sqlite3Reprepare(v) ){
303 sqlite3_reset(pStmt);
304 v->expired = 0;
305 }
danielk197776e8d1a2006-01-18 18:22:43 +0000306 return rc;
drh4f26d6c2004-05-26 23:25:30 +0000307}
drhb900aaf2006-11-09 00:24:53 +0000308#endif
drh4f26d6c2004-05-26 23:25:30 +0000309
310/*
drheb2e1762004-05-27 01:53:56 +0000311** Extract the user data from a sqlite3_context structure and return a
312** pointer to it.
313*/
314void *sqlite3_user_data(sqlite3_context *p){
315 assert( p && p->pFunc );
316 return p->pFunc->pUserData;
317}
318
319/*
drhb7481e72006-09-16 21:45:14 +0000320** The following is the implementation of an SQL function that always
321** fails with an error message stating that the function is used in the
322** wrong context. The sqlite3_overload_function() API might construct
323** SQL function that use this routine so that the functions will exist
324** for name resolution but are actually overloaded by the xFindFunction
325** method of virtual tables.
326*/
327void sqlite3InvalidFunction(
328 sqlite3_context *context, /* The function calling context */
329 int argc, /* Number of arguments to the function */
330 sqlite3_value **argv /* Value of each argument */
331){
332 const char *zName = context->pFunc->zName;
333 char *zErr;
danielk19771e536952007-08-16 10:09:01 +0000334 zErr = sqlite3MPrintf(0,
drhb7481e72006-09-16 21:45:14 +0000335 "unable to use function %s in the requested context", zName);
336 sqlite3_result_error(context, zErr, -1);
danielk19771e536952007-08-16 10:09:01 +0000337 sqlite3_free(zErr);
drhb7481e72006-09-16 21:45:14 +0000338}
339
340/*
drheb2e1762004-05-27 01:53:56 +0000341** Allocate or return the aggregate context for a user function. A new
342** context is allocated on the first call. Subsequent calls return the
343** same context that was returned on prior calls.
drheb2e1762004-05-27 01:53:56 +0000344*/
345void *sqlite3_aggregate_context(sqlite3_context *p, int nByte){
drhabfcea22005-09-06 20:36:48 +0000346 Mem *pMem = p->pMem;
drh825c6622005-09-08 19:01:05 +0000347 assert( p && p->pFunc && p->pFunc->xStep );
drha10a34b2005-09-07 22:09:48 +0000348 if( (pMem->flags & MEM_Agg)==0 ){
349 if( nByte==0 ){
350 assert( pMem->flags==MEM_Null );
351 pMem->z = 0;
drheb2e1762004-05-27 01:53:56 +0000352 }else{
drha10a34b2005-09-07 22:09:48 +0000353 pMem->flags = MEM_Agg;
danielk19771e536952007-08-16 10:09:01 +0000354 pMem->xDel = sqlite3_free;
drh3c024d62007-03-30 11:23:45 +0000355 pMem->u.pDef = p->pFunc;
drha10a34b2005-09-07 22:09:48 +0000356 if( nByte<=NBFS ){
357 pMem->z = pMem->zShort;
358 memset(pMem->z, 0, nByte);
359 }else{
drh17435752007-08-16 04:30:38 +0000360 pMem->z = sqlite3DbMallocZero(p->db, nByte);
drha10a34b2005-09-07 22:09:48 +0000361 }
drheb2e1762004-05-27 01:53:56 +0000362 }
363 }
drhabfcea22005-09-06 20:36:48 +0000364 return (void*)pMem->z;
drheb2e1762004-05-27 01:53:56 +0000365}
366
367/*
danielk1977682f68b2004-06-05 10:22:17 +0000368** Return the auxilary data pointer, if any, for the iArg'th argument to
369** the user-function defined by pCtx.
370*/
371void *sqlite3_get_auxdata(sqlite3_context *pCtx, int iArg){
372 VdbeFunc *pVdbeFunc = pCtx->pVdbeFunc;
373 if( !pVdbeFunc || iArg>=pVdbeFunc->nAux || iArg<0 ){
374 return 0;
375 }
drhf92c7ff2004-06-19 15:40:23 +0000376 return pVdbeFunc->apAux[iArg].pAux;
danielk1977682f68b2004-06-05 10:22:17 +0000377}
378
379/*
380** Set the auxilary data pointer and delete function, for the iArg'th
381** argument to the user-function defined by pCtx. Any previous value is
382** deleted by calling the delete function specified when it was set.
383*/
384void sqlite3_set_auxdata(
385 sqlite3_context *pCtx,
386 int iArg,
387 void *pAux,
388 void (*xDelete)(void*)
389){
390 struct AuxData *pAuxData;
drhf92c7ff2004-06-19 15:40:23 +0000391 VdbeFunc *pVdbeFunc;
danielk1977e0fc5262007-07-26 06:50:05 +0000392 if( iArg<0 ) goto failed;
danielk1977682f68b2004-06-05 10:22:17 +0000393
drhf92c7ff2004-06-19 15:40:23 +0000394 pVdbeFunc = pCtx->pVdbeFunc;
395 if( !pVdbeFunc || pVdbeFunc->nAux<=iArg ){
danielk197731dad9d2007-08-16 11:36:15 +0000396 int nAux = (pVdbeFunc ? pVdbeFunc->nAux : 0);
drhf92c7ff2004-06-19 15:40:23 +0000397 int nMalloc = sizeof(VdbeFunc) + sizeof(struct AuxData)*iArg;
drh17435752007-08-16 04:30:38 +0000398 pVdbeFunc = sqlite3_realloc(pVdbeFunc, nMalloc);
399 if( !pVdbeFunc ){
400 pCtx->db->mallocFailed = 1;
401 goto failed;
402 }
drh53f733c2005-09-16 02:38:09 +0000403 pCtx->pVdbeFunc = pVdbeFunc;
danielk197731dad9d2007-08-16 11:36:15 +0000404 memset(&pVdbeFunc->apAux[nAux], 0, sizeof(struct AuxData)*(iArg+1-nAux));
drh998da3a2004-06-19 15:22:56 +0000405 pVdbeFunc->nAux = iArg+1;
406 pVdbeFunc->pFunc = pCtx->pFunc;
danielk1977682f68b2004-06-05 10:22:17 +0000407 }
408
drhf92c7ff2004-06-19 15:40:23 +0000409 pAuxData = &pVdbeFunc->apAux[iArg];
danielk1977682f68b2004-06-05 10:22:17 +0000410 if( pAuxData->pAux && pAuxData->xDelete ){
411 pAuxData->xDelete(pAuxData->pAux);
412 }
413 pAuxData->pAux = pAux;
414 pAuxData->xDelete = xDelete;
danielk1977e0fc5262007-07-26 06:50:05 +0000415 return;
416
417failed:
418 if( xDelete ){
419 xDelete(pAux);
420 }
danielk1977682f68b2004-06-05 10:22:17 +0000421}
422
423/*
drheb2e1762004-05-27 01:53:56 +0000424** Return the number of times the Step function of a aggregate has been
425** called.
426**
drhcf85a512006-02-09 18:35:29 +0000427** This function is deprecated. Do not use it for new code. It is
428** provide only to avoid breaking legacy code. New aggregate function
429** implementations should keep their own counts within their aggregate
430** context.
drheb2e1762004-05-27 01:53:56 +0000431*/
432int sqlite3_aggregate_count(sqlite3_context *p){
433 assert( p && p->pFunc && p->pFunc->xStep );
drhabfcea22005-09-06 20:36:48 +0000434 return p->pMem->n;
drheb2e1762004-05-27 01:53:56 +0000435}
436
437/*
drh4f26d6c2004-05-26 23:25:30 +0000438** Return the number of columns in the result set for the statement pStmt.
439*/
440int sqlite3_column_count(sqlite3_stmt *pStmt){
441 Vdbe *pVm = (Vdbe *)pStmt;
drh1bcdb0c2004-08-28 14:49:46 +0000442 return pVm ? pVm->nResColumn : 0;
drh4f26d6c2004-05-26 23:25:30 +0000443}
444
445/*
446** Return the number of values available from the current row of the
447** currently executing statement pStmt.
448*/
449int sqlite3_data_count(sqlite3_stmt *pStmt){
450 Vdbe *pVm = (Vdbe *)pStmt;
drh1bcdb0c2004-08-28 14:49:46 +0000451 if( pVm==0 || !pVm->resOnStack ) return 0;
drh4f26d6c2004-05-26 23:25:30 +0000452 return pVm->nResColumn;
453}
454
455
456/*
457** Check to see if column iCol of the given statement is valid. If
458** it is, return a pointer to the Mem for the value of that column.
459** If iCol is not valid, return a pointer to a Mem which has a value
460** of NULL.
461*/
462static Mem *columnMem(sqlite3_stmt *pStmt, int i){
463 Vdbe *pVm = (Vdbe *)pStmt;
464 int vals = sqlite3_data_count(pStmt);
drhb21f87d2007-06-19 10:58:24 +0000465 if( pVm==0 || pVm->resOnStack==0 || i>=pVm->nResColumn || i<0 ){
drh7a521cf2007-04-25 18:23:52 +0000466 static const Mem nullMem = {{0}, 0.0, "", 0, MEM_Null, SQLITE_NULL };
drh4f26d6c2004-05-26 23:25:30 +0000467 sqlite3Error(pVm->db, SQLITE_RANGE, 0);
drh0f7eb612006-08-08 13:51:43 +0000468 return (Mem*)&nullMem;
drh4f26d6c2004-05-26 23:25:30 +0000469 }
470 return &pVm->pTos[(1-vals)+i];
471}
472
danielk19772e588c72005-12-09 14:25:08 +0000473/*
474** This function is called after invoking an sqlite3_value_XXX function on a
475** column value (i.e. a value returned by evaluating an SQL expression in the
476** select list of a SELECT statement) that may cause a malloc() failure. If
477** malloc() has failed, the threads mallocFailed flag is cleared and the result
478** code of statement pStmt set to SQLITE_NOMEM.
479**
480** Specificly, this is called from within:
481**
482** sqlite3_column_int()
483** sqlite3_column_int64()
484** sqlite3_column_text()
485** sqlite3_column_text16()
486** sqlite3_column_real()
487** sqlite3_column_bytes()
488** sqlite3_column_bytes16()
489**
490** But not for sqlite3_column_blob(), which never calls malloc().
491*/
492static void columnMallocFailure(sqlite3_stmt *pStmt)
493{
494 /* If malloc() failed during an encoding conversion within an
495 ** sqlite3_column_XXX API, then set the return code of the statement to
496 ** SQLITE_NOMEM. The next call to _step() (if any) will return SQLITE_ERROR
497 ** and _finalize() will return NOMEM.
498 */
danielk197754f01982006-01-18 15:25:17 +0000499 Vdbe *p = (Vdbe *)pStmt;
500 p->rc = sqlite3ApiExit(0, p->rc);
danielk19772e588c72005-12-09 14:25:08 +0000501}
502
drh4f26d6c2004-05-26 23:25:30 +0000503/**************************** sqlite3_column_ *******************************
504** The following routines are used to access elements of the current row
505** in the result set.
506*/
danielk1977c572ef72004-05-27 09:28:41 +0000507const void *sqlite3_column_blob(sqlite3_stmt *pStmt, int i){
danielk19772e588c72005-12-09 14:25:08 +0000508 const void *val;
danielk19772e588c72005-12-09 14:25:08 +0000509 val = sqlite3_value_blob( columnMem(pStmt,i) );
danielk1977c9cf9012007-05-30 10:36:47 +0000510 /* Even though there is no encoding conversion, value_blob() might
511 ** need to call malloc() to expand the result of a zeroblob()
512 ** expression.
513 */
514 columnMallocFailure(pStmt);
danielk19772e588c72005-12-09 14:25:08 +0000515 return val;
danielk1977c572ef72004-05-27 09:28:41 +0000516}
drh4f26d6c2004-05-26 23:25:30 +0000517int sqlite3_column_bytes(sqlite3_stmt *pStmt, int i){
danielk19772e588c72005-12-09 14:25:08 +0000518 int val = sqlite3_value_bytes( columnMem(pStmt,i) );
519 columnMallocFailure(pStmt);
520 return val;
drh4f26d6c2004-05-26 23:25:30 +0000521}
522int sqlite3_column_bytes16(sqlite3_stmt *pStmt, int i){
danielk19772e588c72005-12-09 14:25:08 +0000523 int val = sqlite3_value_bytes16( columnMem(pStmt,i) );
524 columnMallocFailure(pStmt);
525 return val;
drh4f26d6c2004-05-26 23:25:30 +0000526}
527double sqlite3_column_double(sqlite3_stmt *pStmt, int i){
danielk19772e588c72005-12-09 14:25:08 +0000528 double val = sqlite3_value_double( columnMem(pStmt,i) );
529 columnMallocFailure(pStmt);
530 return val;
drh4f26d6c2004-05-26 23:25:30 +0000531}
532int sqlite3_column_int(sqlite3_stmt *pStmt, int i){
danielk19772e588c72005-12-09 14:25:08 +0000533 int val = sqlite3_value_int( columnMem(pStmt,i) );
534 columnMallocFailure(pStmt);
535 return val;
drh4f26d6c2004-05-26 23:25:30 +0000536}
drhefad9992004-06-22 12:13:55 +0000537sqlite_int64 sqlite3_column_int64(sqlite3_stmt *pStmt, int i){
danielk19772e588c72005-12-09 14:25:08 +0000538 sqlite_int64 val = sqlite3_value_int64( columnMem(pStmt,i) );
539 columnMallocFailure(pStmt);
540 return val;
drh4f26d6c2004-05-26 23:25:30 +0000541}
542const unsigned char *sqlite3_column_text(sqlite3_stmt *pStmt, int i){
danielk19772e588c72005-12-09 14:25:08 +0000543 const unsigned char *val = sqlite3_value_text( columnMem(pStmt,i) );
544 columnMallocFailure(pStmt);
545 return val;
drh4f26d6c2004-05-26 23:25:30 +0000546}
drhd1e47332005-06-26 17:55:33 +0000547sqlite3_value *sqlite3_column_value(sqlite3_stmt *pStmt, int i){
548 return columnMem(pStmt, i);
549}
drh6c626082004-11-14 21:56:29 +0000550#ifndef SQLITE_OMIT_UTF16
drh4f26d6c2004-05-26 23:25:30 +0000551const void *sqlite3_column_text16(sqlite3_stmt *pStmt, int i){
danielk19772e588c72005-12-09 14:25:08 +0000552 const void *val = sqlite3_value_text16( columnMem(pStmt,i) );
553 columnMallocFailure(pStmt);
554 return val;
drh4f26d6c2004-05-26 23:25:30 +0000555}
drh6c626082004-11-14 21:56:29 +0000556#endif /* SQLITE_OMIT_UTF16 */
drh4f26d6c2004-05-26 23:25:30 +0000557int sqlite3_column_type(sqlite3_stmt *pStmt, int i){
558 return sqlite3_value_type( columnMem(pStmt,i) );
559}
560
drh29d72102006-02-09 22:13:41 +0000561/* The following function is experimental and subject to change or
562** removal */
563/*int sqlite3_column_numeric_type(sqlite3_stmt *pStmt, int i){
564** return sqlite3_value_numeric_type( columnMem(pStmt,i) );
565**}
566*/
567
drh5e2517e2004-09-24 23:59:12 +0000568/*
569** Convert the N-th element of pStmt->pColName[] into a string using
570** xFunc() then return that string. If N is out of range, return 0.
drhe590fbd2005-05-20 19:36:01 +0000571**
572** There are up to 5 names for each column. useType determines which
573** name is returned. Here are the names:
574**
575** 0 The column name as it should be displayed for output
576** 1 The datatype name for the column
577** 2 The name of the database that the column derives from
578** 3 The name of the table that the column derives from
579** 4 The name of the table column that the result column derives from
580**
581** If the result is not a simple column reference (if it is an expression
582** or a constant) then useTypes 2, 3, and 4 return NULL.
drh5e2517e2004-09-24 23:59:12 +0000583*/
584static const void *columnName(
585 sqlite3_stmt *pStmt,
586 int N,
587 const void *(*xFunc)(Mem*),
588 int useType
589){
danielk197700fd9572005-12-07 06:27:43 +0000590 const void *ret;
drh5e2517e2004-09-24 23:59:12 +0000591 Vdbe *p = (Vdbe *)pStmt;
592 int n = sqlite3_column_count(pStmt);
593
594 if( p==0 || N>=n || N<0 ){
595 return 0;
596 }
drhe590fbd2005-05-20 19:36:01 +0000597 N += useType*n;
danielk197700fd9572005-12-07 06:27:43 +0000598 ret = xFunc(&p->aColName[N]);
599
600 /* A malloc may have failed inside of the xFunc() call. If this is the case,
601 ** clear the mallocFailed flag and return NULL.
602 */
danielk197754f01982006-01-18 15:25:17 +0000603 sqlite3ApiExit(0, 0);
danielk197700fd9572005-12-07 06:27:43 +0000604 return ret;
drh5e2517e2004-09-24 23:59:12 +0000605}
606
drh4f26d6c2004-05-26 23:25:30 +0000607/*
608** Return the name of the Nth column of the result set returned by SQL
609** statement pStmt.
610*/
611const char *sqlite3_column_name(sqlite3_stmt *pStmt, int N){
danielk1977955de522006-02-10 02:27:42 +0000612 return columnName(
613 pStmt, N, (const void*(*)(Mem*))sqlite3_value_text, COLNAME_NAME);
drh4f26d6c2004-05-26 23:25:30 +0000614}
drhe590fbd2005-05-20 19:36:01 +0000615#ifndef SQLITE_OMIT_UTF16
616const void *sqlite3_column_name16(sqlite3_stmt *pStmt, int N){
danielk1977955de522006-02-10 02:27:42 +0000617 return columnName(
618 pStmt, N, (const void*(*)(Mem*))sqlite3_value_text16, COLNAME_NAME);
drhe590fbd2005-05-20 19:36:01 +0000619}
620#endif
drh4f26d6c2004-05-26 23:25:30 +0000621
622/*
drh6c626082004-11-14 21:56:29 +0000623** Return the column declaration type (if applicable) of the 'i'th column
drhe590fbd2005-05-20 19:36:01 +0000624** of the result set of SQL statement pStmt.
drh6c626082004-11-14 21:56:29 +0000625*/
626const char *sqlite3_column_decltype(sqlite3_stmt *pStmt, int N){
danielk1977955de522006-02-10 02:27:42 +0000627 return columnName(
628 pStmt, N, (const void*(*)(Mem*))sqlite3_value_text, COLNAME_DECLTYPE);
drh6c626082004-11-14 21:56:29 +0000629}
drh6c626082004-11-14 21:56:29 +0000630#ifndef SQLITE_OMIT_UTF16
danielk197776d505b2004-05-28 13:13:02 +0000631const void *sqlite3_column_decltype16(sqlite3_stmt *pStmt, int N){
danielk1977955de522006-02-10 02:27:42 +0000632 return columnName(
633 pStmt, N, (const void*(*)(Mem*))sqlite3_value_text16, COLNAME_DECLTYPE);
drh4f26d6c2004-05-26 23:25:30 +0000634}
drh6c626082004-11-14 21:56:29 +0000635#endif /* SQLITE_OMIT_UTF16 */
drh4f26d6c2004-05-26 23:25:30 +0000636
danielk19774b1ae992006-02-10 03:06:10 +0000637#ifdef SQLITE_ENABLE_COLUMN_METADATA
drhe590fbd2005-05-20 19:36:01 +0000638/*
639** Return the name of the database from which a result column derives.
640** NULL is returned if the result column is an expression or constant or
641** anything else which is not an unabiguous reference to a database column.
642*/
643const char *sqlite3_column_database_name(sqlite3_stmt *pStmt, int N){
danielk1977955de522006-02-10 02:27:42 +0000644 return columnName(
645 pStmt, N, (const void*(*)(Mem*))sqlite3_value_text, COLNAME_DATABASE);
drhe590fbd2005-05-20 19:36:01 +0000646}
647#ifndef SQLITE_OMIT_UTF16
648const void *sqlite3_column_database_name16(sqlite3_stmt *pStmt, int N){
danielk1977955de522006-02-10 02:27:42 +0000649 return columnName(
650 pStmt, N, (const void*(*)(Mem*))sqlite3_value_text16, COLNAME_DATABASE);
drhe590fbd2005-05-20 19:36:01 +0000651}
652#endif /* SQLITE_OMIT_UTF16 */
653
654/*
655** Return the name of the table from which a result column derives.
656** NULL is returned if the result column is an expression or constant or
657** anything else which is not an unabiguous reference to a database column.
658*/
659const char *sqlite3_column_table_name(sqlite3_stmt *pStmt, int N){
danielk1977955de522006-02-10 02:27:42 +0000660 return columnName(
661 pStmt, N, (const void*(*)(Mem*))sqlite3_value_text, COLNAME_TABLE);
drhe590fbd2005-05-20 19:36:01 +0000662}
663#ifndef SQLITE_OMIT_UTF16
664const void *sqlite3_column_table_name16(sqlite3_stmt *pStmt, int N){
danielk1977955de522006-02-10 02:27:42 +0000665 return columnName(
666 pStmt, N, (const void*(*)(Mem*))sqlite3_value_text16, COLNAME_TABLE);
drhe590fbd2005-05-20 19:36:01 +0000667}
668#endif /* SQLITE_OMIT_UTF16 */
669
670/*
671** Return the name of the table column from which a result column derives.
672** NULL is returned if the result column is an expression or constant or
673** anything else which is not an unabiguous reference to a database column.
674*/
675const char *sqlite3_column_origin_name(sqlite3_stmt *pStmt, int N){
danielk1977955de522006-02-10 02:27:42 +0000676 return columnName(
677 pStmt, N, (const void*(*)(Mem*))sqlite3_value_text, COLNAME_COLUMN);
drhe590fbd2005-05-20 19:36:01 +0000678}
679#ifndef SQLITE_OMIT_UTF16
680const void *sqlite3_column_origin_name16(sqlite3_stmt *pStmt, int N){
danielk1977955de522006-02-10 02:27:42 +0000681 return columnName(
682 pStmt, N, (const void*(*)(Mem*))sqlite3_value_text16, COLNAME_COLUMN);
drhe590fbd2005-05-20 19:36:01 +0000683}
684#endif /* SQLITE_OMIT_UTF16 */
danielk19774b1ae992006-02-10 03:06:10 +0000685#endif /* SQLITE_ENABLE_COLUMN_METADATA */
drhe590fbd2005-05-20 19:36:01 +0000686
687
drh4f26d6c2004-05-26 23:25:30 +0000688/******************************* sqlite3_bind_ ***************************
689**
690** Routines used to attach values to wildcards in a compiled SQL statement.
691*/
692/*
693** Unbind the value bound to variable i in virtual machine p. This is the
694** the same as binding a NULL value to the column. If the "i" parameter is
695** out of range, then SQLITE_RANGE is returned. Othewise SQLITE_OK.
696**
697** The error code stored in database p->db is overwritten with the return
698** value in any case.
699*/
700static int vdbeUnbind(Vdbe *p, int i){
701 Mem *pVar;
drh1bcdb0c2004-08-28 14:49:46 +0000702 if( p==0 || p->magic!=VDBE_MAGIC_RUN || p->pc>=0 ){
drh473d1792005-06-06 17:54:55 +0000703 if( p ) sqlite3Error(p->db, SQLITE_MISUSE, 0);
drh4f26d6c2004-05-26 23:25:30 +0000704 return SQLITE_MISUSE;
705 }
706 if( i<1 || i>p->nVar ){
707 sqlite3Error(p->db, SQLITE_RANGE, 0);
708 return SQLITE_RANGE;
709 }
710 i--;
drh290c1942004-08-21 17:54:45 +0000711 pVar = &p->aVar[i];
danielk1977d8123362004-06-12 09:25:12 +0000712 sqlite3VdbeMemRelease(pVar);
drh4f26d6c2004-05-26 23:25:30 +0000713 pVar->flags = MEM_Null;
714 sqlite3Error(p->db, SQLITE_OK, 0);
715 return SQLITE_OK;
716}
717
718/*
drh5e2517e2004-09-24 23:59:12 +0000719** Bind a text or BLOB value.
drh4f26d6c2004-05-26 23:25:30 +0000720*/
drh5e2517e2004-09-24 23:59:12 +0000721static int bindText(
drh605264d2007-08-21 15:13:19 +0000722 sqlite3_stmt *pStmt, /* The statement to bind against */
723 int i, /* Index of the parameter to bind */
724 const void *zData, /* Pointer to the data to be bound */
725 int nData, /* Number of bytes of data to be bound */
726 void (*xDel)(void*), /* Destructor for the data */
727 int encoding /* Encoding for the data */
drh4f26d6c2004-05-26 23:25:30 +0000728){
729 Vdbe *p = (Vdbe *)pStmt;
730 Mem *pVar;
731 int rc;
732
drh605264d2007-08-21 15:13:19 +0000733 if( p==0 ){
734 return SQLITE_MISUSE;
735 }
736 sqlite3_mutex_enter(p->db->mutex);
drh4f26d6c2004-05-26 23:25:30 +0000737 rc = vdbeUnbind(p, i);
drhfeac5f82004-08-01 00:10:45 +0000738 if( rc || zData==0 ){
drh4f26d6c2004-05-26 23:25:30 +0000739 return rc;
740 }
drh290c1942004-08-21 17:54:45 +0000741 pVar = &p->aVar[i-1];
drh605264d2007-08-21 15:13:19 +0000742 rc = sqlite3VdbeMemSetStr(p->db, pVar, zData, nData, encoding, xDel);
drh5e2517e2004-09-24 23:59:12 +0000743 if( rc==SQLITE_OK && encoding!=0 ){
danielk19771e536952007-08-16 10:09:01 +0000744 rc = sqlite3VdbeChangeEncoding(p->db, pVar, ENC(p->db));
drh5e2517e2004-09-24 23:59:12 +0000745 }
danielk19771e536952007-08-16 10:09:01 +0000746 sqlite3Error(p->db, rc, 0);
drh605264d2007-08-21 15:13:19 +0000747 rc = sqlite3ApiExit(p->db, rc);
748 sqlite3_mutex_leave(p->db->mutex);
749 return rc;
drh4f26d6c2004-05-26 23:25:30 +0000750}
drh5e2517e2004-09-24 23:59:12 +0000751
752
753/*
754** Bind a blob value to an SQL statement variable.
755*/
756int sqlite3_bind_blob(
757 sqlite3_stmt *pStmt,
758 int i,
759 const void *zData,
760 int nData,
761 void (*xDel)(void*)
762){
763 return bindText(pStmt, i, zData, nData, xDel, 0);
764}
drh4f26d6c2004-05-26 23:25:30 +0000765int sqlite3_bind_double(sqlite3_stmt *pStmt, int i, double rValue){
766 int rc;
767 Vdbe *p = (Vdbe *)pStmt;
drh605264d2007-08-21 15:13:19 +0000768 sqlite3_mutex_enter(p->db->mutex);
drh4f26d6c2004-05-26 23:25:30 +0000769 rc = vdbeUnbind(p, i);
770 if( rc==SQLITE_OK ){
drh290c1942004-08-21 17:54:45 +0000771 sqlite3VdbeMemSetDouble(&p->aVar[i-1], rValue);
drh4f26d6c2004-05-26 23:25:30 +0000772 }
drh605264d2007-08-21 15:13:19 +0000773 sqlite3_mutex_leave(p->db->mutex);
danielk1977f4618892004-06-28 13:09:11 +0000774 return rc;
drh4f26d6c2004-05-26 23:25:30 +0000775}
776int sqlite3_bind_int(sqlite3_stmt *p, int i, int iValue){
drhefad9992004-06-22 12:13:55 +0000777 return sqlite3_bind_int64(p, i, (i64)iValue);
drh4f26d6c2004-05-26 23:25:30 +0000778}
drhefad9992004-06-22 12:13:55 +0000779int sqlite3_bind_int64(sqlite3_stmt *pStmt, int i, sqlite_int64 iValue){
drh4f26d6c2004-05-26 23:25:30 +0000780 int rc;
781 Vdbe *p = (Vdbe *)pStmt;
drh605264d2007-08-21 15:13:19 +0000782 sqlite3_mutex_enter(p->db->mutex);
drh4f26d6c2004-05-26 23:25:30 +0000783 rc = vdbeUnbind(p, i);
784 if( rc==SQLITE_OK ){
drh290c1942004-08-21 17:54:45 +0000785 sqlite3VdbeMemSetInt64(&p->aVar[i-1], iValue);
drh4f26d6c2004-05-26 23:25:30 +0000786 }
drh605264d2007-08-21 15:13:19 +0000787 sqlite3_mutex_leave(p->db->mutex);
drh4f26d6c2004-05-26 23:25:30 +0000788 return rc;
789}
drh605264d2007-08-21 15:13:19 +0000790int sqlite3_bind_null(sqlite3_stmt *pStmt, int i){
791 int rc;
792 Vdbe *p = (Vdbe*)pStmt;
793 sqlite3_mutex_enter(p->db->mutex);
794 rc = vdbeUnbind(p, i);
795 sqlite3_mutex_leave(p->db->mutex);
796 return rc;
drh4f26d6c2004-05-26 23:25:30 +0000797}
798int sqlite3_bind_text(
799 sqlite3_stmt *pStmt,
800 int i,
801 const char *zData,
802 int nData,
danielk1977d8123362004-06-12 09:25:12 +0000803 void (*xDel)(void*)
drh4f26d6c2004-05-26 23:25:30 +0000804){
drh5e2517e2004-09-24 23:59:12 +0000805 return bindText(pStmt, i, zData, nData, xDel, SQLITE_UTF8);
drh4f26d6c2004-05-26 23:25:30 +0000806}
drh6c626082004-11-14 21:56:29 +0000807#ifndef SQLITE_OMIT_UTF16
drh4f26d6c2004-05-26 23:25:30 +0000808int sqlite3_bind_text16(
809 sqlite3_stmt *pStmt,
810 int i,
811 const void *zData,
812 int nData,
danielk1977d8123362004-06-12 09:25:12 +0000813 void (*xDel)(void*)
drh4f26d6c2004-05-26 23:25:30 +0000814){
drh5e2517e2004-09-24 23:59:12 +0000815 return bindText(pStmt, i, zData, nData, xDel, SQLITE_UTF16NATIVE);
drh4f26d6c2004-05-26 23:25:30 +0000816}
drh6c626082004-11-14 21:56:29 +0000817#endif /* SQLITE_OMIT_UTF16 */
danielk197726e41442006-06-14 15:16:35 +0000818int sqlite3_bind_value(sqlite3_stmt *pStmt, int i, const sqlite3_value *pValue){
819 int rc;
820 Vdbe *p = (Vdbe *)pStmt;
drh605264d2007-08-21 15:13:19 +0000821 sqlite3_mutex_enter(p->db->mutex);
danielk197726e41442006-06-14 15:16:35 +0000822 rc = vdbeUnbind(p, i);
823 if( rc==SQLITE_OK ){
danielk19771e536952007-08-16 10:09:01 +0000824 rc = sqlite3VdbeMemCopy(0, &p->aVar[i-1], pValue);
danielk197726e41442006-06-14 15:16:35 +0000825 }
drh605264d2007-08-21 15:13:19 +0000826 sqlite3_mutex_leave(p->db->mutex);
danielk197726e41442006-06-14 15:16:35 +0000827 return rc;
828}
drhb026e052007-05-02 01:34:31 +0000829int sqlite3_bind_zeroblob(sqlite3_stmt *pStmt, int i, int n){
830 int rc;
831 Vdbe *p = (Vdbe *)pStmt;
drh605264d2007-08-21 15:13:19 +0000832 sqlite3_mutex_enter(p->db->mutex);
drhb026e052007-05-02 01:34:31 +0000833 rc = vdbeUnbind(p, i);
834 if( rc==SQLITE_OK ){
835 sqlite3VdbeMemSetZeroBlob(&p->aVar[i-1], n);
836 }
drh605264d2007-08-21 15:13:19 +0000837 sqlite3_mutex_leave(p->db->mutex);
drhb026e052007-05-02 01:34:31 +0000838 return rc;
839}
drh75f6a032004-07-15 14:15:00 +0000840
841/*
842** Return the number of wildcards that can be potentially bound to.
843** This routine is added to support DBD::SQLite.
drh75f6a032004-07-15 14:15:00 +0000844*/
845int sqlite3_bind_parameter_count(sqlite3_stmt *pStmt){
drh92febd92004-08-20 18:34:20 +0000846 Vdbe *p = (Vdbe*)pStmt;
847 return p ? p->nVar : 0;
drh75f6a032004-07-15 14:15:00 +0000848}
drh895d7472004-08-20 16:02:39 +0000849
850/*
drhfa6bc002004-09-07 16:19:52 +0000851** Create a mapping from variable numbers to variable names
852** in the Vdbe.azVar[] array, if such a mapping does not already
853** exist.
drh895d7472004-08-20 16:02:39 +0000854*/
drhfa6bc002004-09-07 16:19:52 +0000855static void createVarMap(Vdbe *p){
drh895d7472004-08-20 16:02:39 +0000856 if( !p->okVar ){
drh605264d2007-08-21 15:13:19 +0000857 sqlite3_mutex_enter(p->db->mutex);
858 if( !p->okVar ){
859 int j;
860 Op *pOp;
861 for(j=0, pOp=p->aOp; j<p->nOp; j++, pOp++){
862 if( pOp->opcode==OP_Variable ){
863 assert( pOp->p1>0 && pOp->p1<=p->nVar );
864 p->azVar[pOp->p1-1] = pOp->p3;
865 }
drh895d7472004-08-20 16:02:39 +0000866 }
drh605264d2007-08-21 15:13:19 +0000867 p->okVar = 1;
drh895d7472004-08-20 16:02:39 +0000868 }
drh605264d2007-08-21 15:13:19 +0000869 sqlite3_mutex_leave(p->db->mutex);
drh895d7472004-08-20 16:02:39 +0000870 }
drhfa6bc002004-09-07 16:19:52 +0000871}
872
873/*
874** Return the name of a wildcard parameter. Return NULL if the index
875** is out of range or if the wildcard is unnamed.
876**
877** The result is always UTF-8.
878*/
879const char *sqlite3_bind_parameter_name(sqlite3_stmt *pStmt, int i){
880 Vdbe *p = (Vdbe*)pStmt;
881 if( p==0 || i<1 || i>p->nVar ){
882 return 0;
883 }
884 createVarMap(p);
drh895d7472004-08-20 16:02:39 +0000885 return p->azVar[i-1];
886}
drhfa6bc002004-09-07 16:19:52 +0000887
888/*
889** Given a wildcard parameter name, return the index of the variable
890** with that name. If there is no variable with the given name,
891** return 0.
892*/
893int sqlite3_bind_parameter_index(sqlite3_stmt *pStmt, const char *zName){
894 Vdbe *p = (Vdbe*)pStmt;
895 int i;
896 if( p==0 ){
897 return 0;
898 }
899 createVarMap(p);
drh6a8903c2004-12-10 18:00:04 +0000900 if( zName ){
901 for(i=0; i<p->nVar; i++){
902 const char *z = p->azVar[i];
903 if( z && strcmp(z,zName)==0 ){
904 return i+1;
905 }
drhfa6bc002004-09-07 16:19:52 +0000906 }
907 }
908 return 0;
909}
drhf8db1bc2005-04-22 02:38:37 +0000910
drh51942bc2005-06-12 22:01:42 +0000911/*
drhf8db1bc2005-04-22 02:38:37 +0000912** Transfer all bindings from the first statement over to the second.
913** If the two statements contain a different number of bindings, then
914** an SQLITE_ERROR is returned.
915*/
916int sqlite3_transfer_bindings(sqlite3_stmt *pFromStmt, sqlite3_stmt *pToStmt){
917 Vdbe *pFrom = (Vdbe*)pFromStmt;
918 Vdbe *pTo = (Vdbe*)pToStmt;
919 int i, rc = SQLITE_OK;
920 if( (pFrom->magic!=VDBE_MAGIC_RUN && pFrom->magic!=VDBE_MAGIC_HALT)
921 || (pTo->magic!=VDBE_MAGIC_RUN && pTo->magic!=VDBE_MAGIC_HALT) ){
922 return SQLITE_MISUSE;
923 }
924 if( pFrom->nVar!=pTo->nVar ){
925 return SQLITE_ERROR;
926 }
927 for(i=0; rc==SQLITE_OK && i<pFrom->nVar; i++){
danielk1977950f0542006-01-18 05:51:57 +0000928 sqlite3MallocDisallow();
danielk19771e536952007-08-16 10:09:01 +0000929 rc = sqlite3VdbeMemMove(0, &pTo->aVar[i], &pFrom->aVar[i]);
danielk1977950f0542006-01-18 05:51:57 +0000930 sqlite3MallocAllow();
drhf8db1bc2005-04-22 02:38:37 +0000931 }
drh4ac285a2006-09-15 07:28:50 +0000932 assert( rc==SQLITE_OK || rc==SQLITE_NOMEM );
drhf8db1bc2005-04-22 02:38:37 +0000933 return rc;
934}
drh51942bc2005-06-12 22:01:42 +0000935
936/*
937** Return the sqlite3* database handle to which the prepared statement given
938** in the argument belongs. This is the same database handle that was
939** the first argument to the sqlite3_prepare() that was used to create
940** the statement in the first place.
941*/
942sqlite3 *sqlite3_db_handle(sqlite3_stmt *pStmt){
943 return pStmt ? ((Vdbe*)pStmt)->db : 0;
944}