blob: 4e121059f83b98bdbae8faf1f95c2c952ddc8d44 [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);
drh9310ef22007-04-27 17:16:20 +000041 if( (p->flags & MEM_Term)==0 ){
42 p->flags &= ~MEM_Str;
43 }
drh4f26d6c2004-05-26 23:25:30 +000044 return p->z;
45 }else{
46 return sqlite3_value_text(pVal);
47 }
48}
49int sqlite3_value_bytes(sqlite3_value *pVal){
danielk197713073932004-06-30 11:54:06 +000050 return sqlite3ValueBytes(pVal, SQLITE_UTF8);
drh4f26d6c2004-05-26 23:25:30 +000051}
52int sqlite3_value_bytes16(sqlite3_value *pVal){
danielk197713073932004-06-30 11:54:06 +000053 return sqlite3ValueBytes(pVal, SQLITE_UTF16NATIVE);
drh4f26d6c2004-05-26 23:25:30 +000054}
55double sqlite3_value_double(sqlite3_value *pVal){
drh6a6124e2004-06-27 01:56:33 +000056 return sqlite3VdbeRealValue((Mem*)pVal);
drh4f26d6c2004-05-26 23:25:30 +000057}
58int sqlite3_value_int(sqlite3_value *pVal){
drh6a6124e2004-06-27 01:56:33 +000059 return sqlite3VdbeIntValue((Mem*)pVal);
drh4f26d6c2004-05-26 23:25:30 +000060}
drhefad9992004-06-22 12:13:55 +000061sqlite_int64 sqlite3_value_int64(sqlite3_value *pVal){
drh6a6124e2004-06-27 01:56:33 +000062 return sqlite3VdbeIntValue((Mem*)pVal);
drh4f26d6c2004-05-26 23:25:30 +000063}
64const unsigned char *sqlite3_value_text(sqlite3_value *pVal){
danielk197700fd9572005-12-07 06:27:43 +000065 return (const unsigned char *)sqlite3ValueText(pVal, SQLITE_UTF8);
drh4f26d6c2004-05-26 23:25:30 +000066}
drh6c626082004-11-14 21:56:29 +000067#ifndef SQLITE_OMIT_UTF16
drh4f26d6c2004-05-26 23:25:30 +000068const void *sqlite3_value_text16(sqlite3_value* pVal){
danielk1977dc8453f2004-06-12 00:42:34 +000069 return sqlite3ValueText(pVal, SQLITE_UTF16NATIVE);
drh4f26d6c2004-05-26 23:25:30 +000070}
danielk1977d8123362004-06-12 09:25:12 +000071const void *sqlite3_value_text16be(sqlite3_value *pVal){
72 return sqlite3ValueText(pVal, SQLITE_UTF16BE);
73}
74const void *sqlite3_value_text16le(sqlite3_value *pVal){
75 return sqlite3ValueText(pVal, SQLITE_UTF16LE);
76}
drh6c626082004-11-14 21:56:29 +000077#endif /* SQLITE_OMIT_UTF16 */
drh4f26d6c2004-05-26 23:25:30 +000078int sqlite3_value_type(sqlite3_value* pVal){
drhf4479502004-05-27 03:12:53 +000079 return pVal->type;
drh4f26d6c2004-05-26 23:25:30 +000080}
drh29d72102006-02-09 22:13:41 +000081/* sqlite3_value_numeric_type() defined in vdbe.c */
drh4f26d6c2004-05-26 23:25:30 +000082
83/**************************** sqlite3_result_ *******************************
84** The following routines are used by user-defined functions to specify
85** the function result.
86*/
87void sqlite3_result_blob(
88 sqlite3_context *pCtx,
89 const void *z,
90 int n,
danielk1977d8123362004-06-12 09:25:12 +000091 void (*xDel)(void *)
drh4f26d6c2004-05-26 23:25:30 +000092){
drh5708d2d2005-06-22 10:53:59 +000093 assert( n>=0 );
danielk1977d8123362004-06-12 09:25:12 +000094 sqlite3VdbeMemSetStr(&pCtx->s, z, n, 0, xDel);
drh4f26d6c2004-05-26 23:25:30 +000095}
96void sqlite3_result_double(sqlite3_context *pCtx, double rVal){
97 sqlite3VdbeMemSetDouble(&pCtx->s, rVal);
98}
99void sqlite3_result_error(sqlite3_context *pCtx, const char *z, int n){
100 pCtx->isError = 1;
danielk1977d8123362004-06-12 09:25:12 +0000101 sqlite3VdbeMemSetStr(&pCtx->s, z, n, SQLITE_UTF8, SQLITE_TRANSIENT);
drh4f26d6c2004-05-26 23:25:30 +0000102}
danielk1977a1686c92006-01-23 07:52:37 +0000103#ifndef SQLITE_OMIT_UTF16
drh4f26d6c2004-05-26 23:25:30 +0000104void sqlite3_result_error16(sqlite3_context *pCtx, const void *z, int n){
105 pCtx->isError = 1;
danielk1977d8123362004-06-12 09:25:12 +0000106 sqlite3VdbeMemSetStr(&pCtx->s, z, n, SQLITE_UTF16NATIVE, SQLITE_TRANSIENT);
drh4f26d6c2004-05-26 23:25:30 +0000107}
danielk1977a1686c92006-01-23 07:52:37 +0000108#endif
drhf4479502004-05-27 03:12:53 +0000109void sqlite3_result_int(sqlite3_context *pCtx, int iVal){
drh4f26d6c2004-05-26 23:25:30 +0000110 sqlite3VdbeMemSetInt64(&pCtx->s, (i64)iVal);
111}
112void sqlite3_result_int64(sqlite3_context *pCtx, i64 iVal){
113 sqlite3VdbeMemSetInt64(&pCtx->s, iVal);
114}
115void sqlite3_result_null(sqlite3_context *pCtx){
drhf4479502004-05-27 03:12:53 +0000116 sqlite3VdbeMemSetNull(&pCtx->s);
drh4f26d6c2004-05-26 23:25:30 +0000117}
118void sqlite3_result_text(
119 sqlite3_context *pCtx,
120 const char *z,
121 int n,
danielk1977d8123362004-06-12 09:25:12 +0000122 void (*xDel)(void *)
drh4f26d6c2004-05-26 23:25:30 +0000123){
danielk1977d8123362004-06-12 09:25:12 +0000124 sqlite3VdbeMemSetStr(&pCtx->s, z, n, SQLITE_UTF8, xDel);
drh4f26d6c2004-05-26 23:25:30 +0000125}
drh6c626082004-11-14 21:56:29 +0000126#ifndef SQLITE_OMIT_UTF16
drh4f26d6c2004-05-26 23:25:30 +0000127void sqlite3_result_text16(
128 sqlite3_context *pCtx,
129 const void *z,
130 int n,
danielk1977d8123362004-06-12 09:25:12 +0000131 void (*xDel)(void *)
drh4f26d6c2004-05-26 23:25:30 +0000132){
danielk1977d8123362004-06-12 09:25:12 +0000133 sqlite3VdbeMemSetStr(&pCtx->s, z, n, SQLITE_UTF16NATIVE, xDel);
134}
135void sqlite3_result_text16be(
136 sqlite3_context *pCtx,
137 const void *z,
138 int n,
139 void (*xDel)(void *)
140){
141 sqlite3VdbeMemSetStr(&pCtx->s, z, n, SQLITE_UTF16BE, xDel);
142}
143void sqlite3_result_text16le(
144 sqlite3_context *pCtx,
145 const void *z,
146 int n,
147 void (*xDel)(void *)
148){
149 sqlite3VdbeMemSetStr(&pCtx->s, z, n, SQLITE_UTF16LE, xDel);
drh4f26d6c2004-05-26 23:25:30 +0000150}
drh6c626082004-11-14 21:56:29 +0000151#endif /* SQLITE_OMIT_UTF16 */
drh4f26d6c2004-05-26 23:25:30 +0000152void sqlite3_result_value(sqlite3_context *pCtx, sqlite3_value *pValue){
153 sqlite3VdbeMemCopy(&pCtx->s, pValue);
154}
drhb026e052007-05-02 01:34:31 +0000155void sqlite3_result_zeroblob(sqlite3_context *pCtx, int n){
156 sqlite3VdbeMemSetZeroBlob(&pCtx->s, n);
157}
drh4f26d6c2004-05-26 23:25:30 +0000158
159
160/*
161** Execute the statement pStmt, either until a row of data is ready, the
162** statement is completely executed or an error occurs.
drhb900aaf2006-11-09 00:24:53 +0000163**
164** This routine implements the bulk of the logic behind the sqlite_step()
165** API. The only thing omitted is the automatic recompile if a
166** schema change has occurred. That detail is handled by the
167** outer sqlite3_step() wrapper procedure.
drh4f26d6c2004-05-26 23:25:30 +0000168*/
drhb900aaf2006-11-09 00:24:53 +0000169static int sqlite3Step(Vdbe *p){
drh9bb575f2004-09-06 17:24:11 +0000170 sqlite3 *db;
drh4f26d6c2004-05-26 23:25:30 +0000171 int rc;
172
danielk1977efaaf572006-01-16 11:29:19 +0000173 /* Assert that malloc() has not failed */
danielk19779e128002006-01-18 16:51:35 +0000174 assert( !sqlite3MallocFailed() );
danielk1977261919c2005-12-06 12:52:59 +0000175
drh1bcdb0c2004-08-28 14:49:46 +0000176 if( p==0 || p->magic!=VDBE_MAGIC_RUN ){
drh4f26d6c2004-05-26 23:25:30 +0000177 return SQLITE_MISUSE;
178 }
drh91b48aa2004-06-30 11:14:18 +0000179 if( p->aborted ){
180 return SQLITE_ABORT;
181 }
danielk1977a21c6b62005-01-24 10:25:59 +0000182 if( p->pc<=0 && p->expired ){
183 if( p->rc==SQLITE_OK ){
184 p->rc = SQLITE_SCHEMA;
185 }
drhb900aaf2006-11-09 00:24:53 +0000186 rc = SQLITE_ERROR;
187 goto end_of_step;
danielk1977a21c6b62005-01-24 10:25:59 +0000188 }
drh4f26d6c2004-05-26 23:25:30 +0000189 db = p->db;
190 if( sqlite3SafetyOn(db) ){
191 p->rc = SQLITE_MISUSE;
192 return SQLITE_MISUSE;
193 }
danielk19771d850a72004-05-31 08:26:49 +0000194 if( p->pc<0 ){
drh15ca1df2006-07-26 13:43:30 +0000195 /* If there are no other statements currently running, then
196 ** reset the interrupt flag. This prevents a call to sqlite3_interrupt
197 ** from interrupting a statement that has not yet started.
198 */
199 if( db->activeVdbeCnt==0 ){
200 db->u1.isInterrupted = 0;
201 }
202
drh19e2d372005-08-29 23:00:03 +0000203#ifndef SQLITE_OMIT_TRACE
drhc16a03b2004-09-15 13:38:10 +0000204 /* Invoke the trace callback if there is one
205 */
drh19e2d372005-08-29 23:00:03 +0000206 if( db->xTrace && !db->init.busy ){
drhc16a03b2004-09-15 13:38:10 +0000207 assert( p->nOp>0 );
208 assert( p->aOp[p->nOp-1].opcode==OP_Noop );
209 assert( p->aOp[p->nOp-1].p3!=0 );
210 assert( p->aOp[p->nOp-1].p3type==P3_DYNAMIC );
211 sqlite3SafetyOff(db);
212 db->xTrace(db->pTraceArg, p->aOp[p->nOp-1].p3);
213 if( sqlite3SafetyOn(db) ){
214 p->rc = SQLITE_MISUSE;
215 return SQLITE_MISUSE;
216 }
217 }
drh19e2d372005-08-29 23:00:03 +0000218 if( db->xProfile && !db->init.busy ){
219 double rNow;
drh66560ad2006-01-06 14:32:19 +0000220 sqlite3OsCurrentTime(&rNow);
drh19e2d372005-08-29 23:00:03 +0000221 p->startTime = (rNow - (int)rNow)*3600.0*24.0*1000000000.0;
222 }
223#endif
drhc16a03b2004-09-15 13:38:10 +0000224
225 /* Print a copy of SQL as it is executed if the SQL_TRACE pragma is turned
226 ** on in debugging mode.
227 */
228#ifdef SQLITE_DEBUG
229 if( (db->flags & SQLITE_SqlTrace)!=0 ){
230 sqlite3DebugPrintf("SQL-trace: %s\n", p->aOp[p->nOp-1].p3);
231 }
232#endif /* SQLITE_DEBUG */
233
danielk19771d850a72004-05-31 08:26:49 +0000234 db->activeVdbeCnt++;
235 p->pc = 0;
236 }
drhb7f91642004-10-31 02:22:47 +0000237#ifndef SQLITE_OMIT_EXPLAIN
drh4f26d6c2004-05-26 23:25:30 +0000238 if( p->explain ){
239 rc = sqlite3VdbeList(p);
drhb7f91642004-10-31 02:22:47 +0000240 }else
241#endif /* SQLITE_OMIT_EXPLAIN */
242 {
drh4f26d6c2004-05-26 23:25:30 +0000243 rc = sqlite3VdbeExec(p);
244 }
245
246 if( sqlite3SafetyOff(db) ){
247 rc = SQLITE_MISUSE;
248 }
249
drh19e2d372005-08-29 23:00:03 +0000250#ifndef SQLITE_OMIT_TRACE
251 /* Invoke the profile callback if there is one
252 */
253 if( rc!=SQLITE_ROW && db->xProfile && !db->init.busy ){
254 double rNow;
255 u64 elapseTime;
256
drh66560ad2006-01-06 14:32:19 +0000257 sqlite3OsCurrentTime(&rNow);
drh19e2d372005-08-29 23:00:03 +0000258 elapseTime = (rNow - (int)rNow)*3600.0*24.0*1000000000.0 - p->startTime;
259 assert( p->nOp>0 );
260 assert( p->aOp[p->nOp-1].opcode==OP_Noop );
261 assert( p->aOp[p->nOp-1].p3!=0 );
262 assert( p->aOp[p->nOp-1].p3type==P3_DYNAMIC );
263 db->xProfile(db->pProfileArg, p->aOp[p->nOp-1].p3, elapseTime);
264 }
265#endif
266
danielk197797a227c2006-01-20 16:32:04 +0000267 sqlite3Error(p->db, rc, 0);
danielk197776e8d1a2006-01-18 18:22:43 +0000268 p->rc = sqlite3ApiExit(p->db, p->rc);
drhb900aaf2006-11-09 00:24:53 +0000269end_of_step:
drh4ac285a2006-09-15 07:28:50 +0000270 assert( (rc&0xff)==rc );
drhb900aaf2006-11-09 00:24:53 +0000271 if( p->zSql && (rc&0xff)<SQLITE_ROW ){
272 /* This behavior occurs if sqlite3_prepare_v2() was used to build
273 ** the prepared statement. Return error codes directly */
274 return p->rc;
275 }else{
276 /* This is for legacy sqlite3_prepare() builds and when the code
277 ** is SQLITE_ROW or SQLITE_DONE */
278 return rc;
279 }
280}
281
282/*
283** This is the top-level implementation of sqlite3_step(). Call
284** sqlite3Step() to do most of the work. If a schema error occurs,
285** call sqlite3Reprepare() and try again.
286*/
287#ifdef SQLITE_OMIT_PARSER
288int sqlite3_step(sqlite3_stmt *pStmt){
289 return sqlite3Step((Vdbe*)pStmt);
290}
291#else
292int sqlite3_step(sqlite3_stmt *pStmt){
293 int cnt = 0;
294 int rc;
295 Vdbe *v = (Vdbe*)pStmt;
296 while( (rc = sqlite3Step(v))==SQLITE_SCHEMA
297 && cnt++ < 5
298 && sqlite3Reprepare(v) ){
299 sqlite3_reset(pStmt);
300 v->expired = 0;
301 }
danielk197776e8d1a2006-01-18 18:22:43 +0000302 return rc;
drh4f26d6c2004-05-26 23:25:30 +0000303}
drhb900aaf2006-11-09 00:24:53 +0000304#endif
drh4f26d6c2004-05-26 23:25:30 +0000305
306/*
drheb2e1762004-05-27 01:53:56 +0000307** Extract the user data from a sqlite3_context structure and return a
308** pointer to it.
309*/
310void *sqlite3_user_data(sqlite3_context *p){
311 assert( p && p->pFunc );
312 return p->pFunc->pUserData;
313}
314
315/*
drhb7481e72006-09-16 21:45:14 +0000316** The following is the implementation of an SQL function that always
317** fails with an error message stating that the function is used in the
318** wrong context. The sqlite3_overload_function() API might construct
319** SQL function that use this routine so that the functions will exist
320** for name resolution but are actually overloaded by the xFindFunction
321** method of virtual tables.
322*/
323void sqlite3InvalidFunction(
324 sqlite3_context *context, /* The function calling context */
325 int argc, /* Number of arguments to the function */
326 sqlite3_value **argv /* Value of each argument */
327){
328 const char *zName = context->pFunc->zName;
329 char *zErr;
330 zErr = sqlite3MPrintf(
331 "unable to use function %s in the requested context", zName);
332 sqlite3_result_error(context, zErr, -1);
333 sqliteFree(zErr);
334}
335
336/*
drheb2e1762004-05-27 01:53:56 +0000337** Allocate or return the aggregate context for a user function. A new
338** context is allocated on the first call. Subsequent calls return the
339** same context that was returned on prior calls.
drheb2e1762004-05-27 01:53:56 +0000340*/
341void *sqlite3_aggregate_context(sqlite3_context *p, int nByte){
drhabfcea22005-09-06 20:36:48 +0000342 Mem *pMem = p->pMem;
drh825c6622005-09-08 19:01:05 +0000343 assert( p && p->pFunc && p->pFunc->xStep );
drha10a34b2005-09-07 22:09:48 +0000344 if( (pMem->flags & MEM_Agg)==0 ){
345 if( nByte==0 ){
346 assert( pMem->flags==MEM_Null );
347 pMem->z = 0;
drheb2e1762004-05-27 01:53:56 +0000348 }else{
drha10a34b2005-09-07 22:09:48 +0000349 pMem->flags = MEM_Agg;
drh5360ad32005-09-08 00:13:27 +0000350 pMem->xDel = sqlite3FreeX;
drh3c024d62007-03-30 11:23:45 +0000351 pMem->u.pDef = p->pFunc;
drha10a34b2005-09-07 22:09:48 +0000352 if( nByte<=NBFS ){
353 pMem->z = pMem->zShort;
354 memset(pMem->z, 0, nByte);
355 }else{
356 pMem->z = sqliteMalloc( nByte );
357 }
drheb2e1762004-05-27 01:53:56 +0000358 }
359 }
drhabfcea22005-09-06 20:36:48 +0000360 return (void*)pMem->z;
drheb2e1762004-05-27 01:53:56 +0000361}
362
363/*
danielk1977682f68b2004-06-05 10:22:17 +0000364** Return the auxilary data pointer, if any, for the iArg'th argument to
365** the user-function defined by pCtx.
366*/
367void *sqlite3_get_auxdata(sqlite3_context *pCtx, int iArg){
368 VdbeFunc *pVdbeFunc = pCtx->pVdbeFunc;
369 if( !pVdbeFunc || iArg>=pVdbeFunc->nAux || iArg<0 ){
370 return 0;
371 }
drhf92c7ff2004-06-19 15:40:23 +0000372 return pVdbeFunc->apAux[iArg].pAux;
danielk1977682f68b2004-06-05 10:22:17 +0000373}
374
375/*
376** Set the auxilary data pointer and delete function, for the iArg'th
377** argument to the user-function defined by pCtx. Any previous value is
378** deleted by calling the delete function specified when it was set.
379*/
380void sqlite3_set_auxdata(
381 sqlite3_context *pCtx,
382 int iArg,
383 void *pAux,
384 void (*xDelete)(void*)
385){
386 struct AuxData *pAuxData;
drhf92c7ff2004-06-19 15:40:23 +0000387 VdbeFunc *pVdbeFunc;
danielk1977682f68b2004-06-05 10:22:17 +0000388 if( iArg<0 ) return;
389
drhf92c7ff2004-06-19 15:40:23 +0000390 pVdbeFunc = pCtx->pVdbeFunc;
391 if( !pVdbeFunc || pVdbeFunc->nAux<=iArg ){
392 int nMalloc = sizeof(VdbeFunc) + sizeof(struct AuxData)*iArg;
drh53f733c2005-09-16 02:38:09 +0000393 pVdbeFunc = sqliteRealloc(pVdbeFunc, nMalloc);
drh998da3a2004-06-19 15:22:56 +0000394 if( !pVdbeFunc ) return;
drh53f733c2005-09-16 02:38:09 +0000395 pCtx->pVdbeFunc = pVdbeFunc;
drh0e3d7472004-06-19 17:33:07 +0000396 memset(&pVdbeFunc->apAux[pVdbeFunc->nAux], 0,
397 sizeof(struct AuxData)*(iArg+1-pVdbeFunc->nAux));
drh998da3a2004-06-19 15:22:56 +0000398 pVdbeFunc->nAux = iArg+1;
399 pVdbeFunc->pFunc = pCtx->pFunc;
danielk1977682f68b2004-06-05 10:22:17 +0000400 }
401
drhf92c7ff2004-06-19 15:40:23 +0000402 pAuxData = &pVdbeFunc->apAux[iArg];
danielk1977682f68b2004-06-05 10:22:17 +0000403 if( pAuxData->pAux && pAuxData->xDelete ){
404 pAuxData->xDelete(pAuxData->pAux);
405 }
406 pAuxData->pAux = pAux;
407 pAuxData->xDelete = xDelete;
408}
409
410/*
drheb2e1762004-05-27 01:53:56 +0000411** Return the number of times the Step function of a aggregate has been
412** called.
413**
drhcf85a512006-02-09 18:35:29 +0000414** This function is deprecated. Do not use it for new code. It is
415** provide only to avoid breaking legacy code. New aggregate function
416** implementations should keep their own counts within their aggregate
417** context.
drheb2e1762004-05-27 01:53:56 +0000418*/
419int sqlite3_aggregate_count(sqlite3_context *p){
420 assert( p && p->pFunc && p->pFunc->xStep );
drhabfcea22005-09-06 20:36:48 +0000421 return p->pMem->n;
drheb2e1762004-05-27 01:53:56 +0000422}
423
424/*
drh4f26d6c2004-05-26 23:25:30 +0000425** Return the number of columns in the result set for the statement pStmt.
426*/
427int sqlite3_column_count(sqlite3_stmt *pStmt){
428 Vdbe *pVm = (Vdbe *)pStmt;
drh1bcdb0c2004-08-28 14:49:46 +0000429 return pVm ? pVm->nResColumn : 0;
drh4f26d6c2004-05-26 23:25:30 +0000430}
431
432/*
433** Return the number of values available from the current row of the
434** currently executing statement pStmt.
435*/
436int sqlite3_data_count(sqlite3_stmt *pStmt){
437 Vdbe *pVm = (Vdbe *)pStmt;
drh1bcdb0c2004-08-28 14:49:46 +0000438 if( pVm==0 || !pVm->resOnStack ) return 0;
drh4f26d6c2004-05-26 23:25:30 +0000439 return pVm->nResColumn;
440}
441
442
443/*
444** Check to see if column iCol of the given statement is valid. If
445** it is, return a pointer to the Mem for the value of that column.
446** If iCol is not valid, return a pointer to a Mem which has a value
447** of NULL.
448*/
449static Mem *columnMem(sqlite3_stmt *pStmt, int i){
450 Vdbe *pVm = (Vdbe *)pStmt;
451 int vals = sqlite3_data_count(pStmt);
452 if( i>=vals || i<0 ){
drh7a521cf2007-04-25 18:23:52 +0000453 static const Mem nullMem = {{0}, 0.0, "", 0, MEM_Null, SQLITE_NULL };
drh4f26d6c2004-05-26 23:25:30 +0000454 sqlite3Error(pVm->db, SQLITE_RANGE, 0);
drh0f7eb612006-08-08 13:51:43 +0000455 return (Mem*)&nullMem;
drh4f26d6c2004-05-26 23:25:30 +0000456 }
457 return &pVm->pTos[(1-vals)+i];
458}
459
danielk19772e588c72005-12-09 14:25:08 +0000460/*
461** This function is called after invoking an sqlite3_value_XXX function on a
462** column value (i.e. a value returned by evaluating an SQL expression in the
463** select list of a SELECT statement) that may cause a malloc() failure. If
464** malloc() has failed, the threads mallocFailed flag is cleared and the result
465** code of statement pStmt set to SQLITE_NOMEM.
466**
467** Specificly, this is called from within:
468**
469** sqlite3_column_int()
470** sqlite3_column_int64()
471** sqlite3_column_text()
472** sqlite3_column_text16()
473** sqlite3_column_real()
474** sqlite3_column_bytes()
475** sqlite3_column_bytes16()
476**
477** But not for sqlite3_column_blob(), which never calls malloc().
478*/
479static void columnMallocFailure(sqlite3_stmt *pStmt)
480{
481 /* If malloc() failed during an encoding conversion within an
482 ** sqlite3_column_XXX API, then set the return code of the statement to
483 ** SQLITE_NOMEM. The next call to _step() (if any) will return SQLITE_ERROR
484 ** and _finalize() will return NOMEM.
485 */
danielk197754f01982006-01-18 15:25:17 +0000486 Vdbe *p = (Vdbe *)pStmt;
487 p->rc = sqlite3ApiExit(0, p->rc);
danielk19772e588c72005-12-09 14:25:08 +0000488}
489
drh4f26d6c2004-05-26 23:25:30 +0000490/**************************** sqlite3_column_ *******************************
491** The following routines are used to access elements of the current row
492** in the result set.
493*/
danielk1977c572ef72004-05-27 09:28:41 +0000494const void *sqlite3_column_blob(sqlite3_stmt *pStmt, int i){
danielk19772e588c72005-12-09 14:25:08 +0000495 const void *val;
496 sqlite3MallocDisallow();
497 val = sqlite3_value_blob( columnMem(pStmt,i) );
498 sqlite3MallocAllow();
499 return val;
danielk1977c572ef72004-05-27 09:28:41 +0000500}
drh4f26d6c2004-05-26 23:25:30 +0000501int sqlite3_column_bytes(sqlite3_stmt *pStmt, int i){
danielk19772e588c72005-12-09 14:25:08 +0000502 int val = sqlite3_value_bytes( columnMem(pStmt,i) );
503 columnMallocFailure(pStmt);
504 return val;
drh4f26d6c2004-05-26 23:25:30 +0000505}
506int sqlite3_column_bytes16(sqlite3_stmt *pStmt, int i){
danielk19772e588c72005-12-09 14:25:08 +0000507 int val = sqlite3_value_bytes16( columnMem(pStmt,i) );
508 columnMallocFailure(pStmt);
509 return val;
drh4f26d6c2004-05-26 23:25:30 +0000510}
511double sqlite3_column_double(sqlite3_stmt *pStmt, int i){
danielk19772e588c72005-12-09 14:25:08 +0000512 double val = sqlite3_value_double( columnMem(pStmt,i) );
513 columnMallocFailure(pStmt);
514 return val;
drh4f26d6c2004-05-26 23:25:30 +0000515}
516int sqlite3_column_int(sqlite3_stmt *pStmt, int i){
danielk19772e588c72005-12-09 14:25:08 +0000517 int val = sqlite3_value_int( columnMem(pStmt,i) );
518 columnMallocFailure(pStmt);
519 return val;
drh4f26d6c2004-05-26 23:25:30 +0000520}
drhefad9992004-06-22 12:13:55 +0000521sqlite_int64 sqlite3_column_int64(sqlite3_stmt *pStmt, int i){
danielk19772e588c72005-12-09 14:25:08 +0000522 sqlite_int64 val = sqlite3_value_int64( columnMem(pStmt,i) );
523 columnMallocFailure(pStmt);
524 return val;
drh4f26d6c2004-05-26 23:25:30 +0000525}
526const unsigned char *sqlite3_column_text(sqlite3_stmt *pStmt, int i){
danielk19772e588c72005-12-09 14:25:08 +0000527 const unsigned char *val = sqlite3_value_text( columnMem(pStmt,i) );
528 columnMallocFailure(pStmt);
529 return val;
drh4f26d6c2004-05-26 23:25:30 +0000530}
drhd1e47332005-06-26 17:55:33 +0000531sqlite3_value *sqlite3_column_value(sqlite3_stmt *pStmt, int i){
532 return columnMem(pStmt, i);
533}
drh6c626082004-11-14 21:56:29 +0000534#ifndef SQLITE_OMIT_UTF16
drh4f26d6c2004-05-26 23:25:30 +0000535const void *sqlite3_column_text16(sqlite3_stmt *pStmt, int i){
danielk19772e588c72005-12-09 14:25:08 +0000536 const void *val = sqlite3_value_text16( columnMem(pStmt,i) );
537 columnMallocFailure(pStmt);
538 return val;
drh4f26d6c2004-05-26 23:25:30 +0000539}
drh6c626082004-11-14 21:56:29 +0000540#endif /* SQLITE_OMIT_UTF16 */
drh4f26d6c2004-05-26 23:25:30 +0000541int sqlite3_column_type(sqlite3_stmt *pStmt, int i){
542 return sqlite3_value_type( columnMem(pStmt,i) );
543}
544
drh29d72102006-02-09 22:13:41 +0000545/* The following function is experimental and subject to change or
546** removal */
547/*int sqlite3_column_numeric_type(sqlite3_stmt *pStmt, int i){
548** return sqlite3_value_numeric_type( columnMem(pStmt,i) );
549**}
550*/
551
drh5e2517e2004-09-24 23:59:12 +0000552/*
553** Convert the N-th element of pStmt->pColName[] into a string using
554** xFunc() then return that string. If N is out of range, return 0.
drhe590fbd2005-05-20 19:36:01 +0000555**
556** There are up to 5 names for each column. useType determines which
557** name is returned. Here are the names:
558**
559** 0 The column name as it should be displayed for output
560** 1 The datatype name for the column
561** 2 The name of the database that the column derives from
562** 3 The name of the table that the column derives from
563** 4 The name of the table column that the result column derives from
564**
565** If the result is not a simple column reference (if it is an expression
566** or a constant) then useTypes 2, 3, and 4 return NULL.
drh5e2517e2004-09-24 23:59:12 +0000567*/
568static const void *columnName(
569 sqlite3_stmt *pStmt,
570 int N,
571 const void *(*xFunc)(Mem*),
572 int useType
573){
danielk197700fd9572005-12-07 06:27:43 +0000574 const void *ret;
drh5e2517e2004-09-24 23:59:12 +0000575 Vdbe *p = (Vdbe *)pStmt;
576 int n = sqlite3_column_count(pStmt);
577
578 if( p==0 || N>=n || N<0 ){
579 return 0;
580 }
drhe590fbd2005-05-20 19:36:01 +0000581 N += useType*n;
danielk197700fd9572005-12-07 06:27:43 +0000582 ret = xFunc(&p->aColName[N]);
583
584 /* A malloc may have failed inside of the xFunc() call. If this is the case,
585 ** clear the mallocFailed flag and return NULL.
586 */
danielk197754f01982006-01-18 15:25:17 +0000587 sqlite3ApiExit(0, 0);
danielk197700fd9572005-12-07 06:27:43 +0000588 return ret;
drh5e2517e2004-09-24 23:59:12 +0000589}
590
drh4f26d6c2004-05-26 23:25:30 +0000591/*
592** Return the name of the Nth column of the result set returned by SQL
593** statement pStmt.
594*/
595const char *sqlite3_column_name(sqlite3_stmt *pStmt, int N){
danielk1977955de522006-02-10 02:27:42 +0000596 return columnName(
597 pStmt, N, (const void*(*)(Mem*))sqlite3_value_text, COLNAME_NAME);
drh4f26d6c2004-05-26 23:25:30 +0000598}
drhe590fbd2005-05-20 19:36:01 +0000599#ifndef SQLITE_OMIT_UTF16
600const void *sqlite3_column_name16(sqlite3_stmt *pStmt, int N){
danielk1977955de522006-02-10 02:27:42 +0000601 return columnName(
602 pStmt, N, (const void*(*)(Mem*))sqlite3_value_text16, COLNAME_NAME);
drhe590fbd2005-05-20 19:36:01 +0000603}
604#endif
drh4f26d6c2004-05-26 23:25:30 +0000605
606/*
drh6c626082004-11-14 21:56:29 +0000607** Return the column declaration type (if applicable) of the 'i'th column
drhe590fbd2005-05-20 19:36:01 +0000608** of the result set of SQL statement pStmt.
drh6c626082004-11-14 21:56:29 +0000609*/
610const char *sqlite3_column_decltype(sqlite3_stmt *pStmt, int N){
danielk1977955de522006-02-10 02:27:42 +0000611 return columnName(
612 pStmt, N, (const void*(*)(Mem*))sqlite3_value_text, COLNAME_DECLTYPE);
drh6c626082004-11-14 21:56:29 +0000613}
drh6c626082004-11-14 21:56:29 +0000614#ifndef SQLITE_OMIT_UTF16
danielk197776d505b2004-05-28 13:13:02 +0000615const void *sqlite3_column_decltype16(sqlite3_stmt *pStmt, int N){
danielk1977955de522006-02-10 02:27:42 +0000616 return columnName(
617 pStmt, N, (const void*(*)(Mem*))sqlite3_value_text16, COLNAME_DECLTYPE);
drh4f26d6c2004-05-26 23:25:30 +0000618}
drh6c626082004-11-14 21:56:29 +0000619#endif /* SQLITE_OMIT_UTF16 */
drh4f26d6c2004-05-26 23:25:30 +0000620
danielk19774b1ae992006-02-10 03:06:10 +0000621#ifdef SQLITE_ENABLE_COLUMN_METADATA
drhe590fbd2005-05-20 19:36:01 +0000622/*
623** Return the name of the database from which a result column derives.
624** NULL is returned if the result column is an expression or constant or
625** anything else which is not an unabiguous reference to a database column.
626*/
627const char *sqlite3_column_database_name(sqlite3_stmt *pStmt, int N){
danielk1977955de522006-02-10 02:27:42 +0000628 return columnName(
629 pStmt, N, (const void*(*)(Mem*))sqlite3_value_text, COLNAME_DATABASE);
drhe590fbd2005-05-20 19:36:01 +0000630}
631#ifndef SQLITE_OMIT_UTF16
632const void *sqlite3_column_database_name16(sqlite3_stmt *pStmt, int N){
danielk1977955de522006-02-10 02:27:42 +0000633 return columnName(
634 pStmt, N, (const void*(*)(Mem*))sqlite3_value_text16, COLNAME_DATABASE);
drhe590fbd2005-05-20 19:36:01 +0000635}
636#endif /* SQLITE_OMIT_UTF16 */
637
638/*
639** Return the name of the table 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_table_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_TABLE);
drhe590fbd2005-05-20 19:36:01 +0000646}
647#ifndef SQLITE_OMIT_UTF16
648const void *sqlite3_column_table_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_TABLE);
drhe590fbd2005-05-20 19:36:01 +0000651}
652#endif /* SQLITE_OMIT_UTF16 */
653
654/*
655** Return the name of the table column 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_origin_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_COLUMN);
drhe590fbd2005-05-20 19:36:01 +0000662}
663#ifndef SQLITE_OMIT_UTF16
664const void *sqlite3_column_origin_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_COLUMN);
drhe590fbd2005-05-20 19:36:01 +0000667}
668#endif /* SQLITE_OMIT_UTF16 */
danielk19774b1ae992006-02-10 03:06:10 +0000669#endif /* SQLITE_ENABLE_COLUMN_METADATA */
drhe590fbd2005-05-20 19:36:01 +0000670
671
drh4f26d6c2004-05-26 23:25:30 +0000672/******************************* sqlite3_bind_ ***************************
673**
674** Routines used to attach values to wildcards in a compiled SQL statement.
675*/
676/*
677** Unbind the value bound to variable i in virtual machine p. This is the
678** the same as binding a NULL value to the column. If the "i" parameter is
679** out of range, then SQLITE_RANGE is returned. Othewise SQLITE_OK.
680**
681** The error code stored in database p->db is overwritten with the return
682** value in any case.
683*/
684static int vdbeUnbind(Vdbe *p, int i){
685 Mem *pVar;
drh1bcdb0c2004-08-28 14:49:46 +0000686 if( p==0 || p->magic!=VDBE_MAGIC_RUN || p->pc>=0 ){
drh473d1792005-06-06 17:54:55 +0000687 if( p ) sqlite3Error(p->db, SQLITE_MISUSE, 0);
drh4f26d6c2004-05-26 23:25:30 +0000688 return SQLITE_MISUSE;
689 }
690 if( i<1 || i>p->nVar ){
691 sqlite3Error(p->db, SQLITE_RANGE, 0);
692 return SQLITE_RANGE;
693 }
694 i--;
drh290c1942004-08-21 17:54:45 +0000695 pVar = &p->aVar[i];
danielk1977d8123362004-06-12 09:25:12 +0000696 sqlite3VdbeMemRelease(pVar);
drh4f26d6c2004-05-26 23:25:30 +0000697 pVar->flags = MEM_Null;
698 sqlite3Error(p->db, SQLITE_OK, 0);
699 return SQLITE_OK;
700}
701
702/*
drh5e2517e2004-09-24 23:59:12 +0000703** Bind a text or BLOB value.
drh4f26d6c2004-05-26 23:25:30 +0000704*/
drh5e2517e2004-09-24 23:59:12 +0000705static int bindText(
drhf4479502004-05-27 03:12:53 +0000706 sqlite3_stmt *pStmt,
drh4f26d6c2004-05-26 23:25:30 +0000707 int i,
708 const void *zData,
709 int nData,
drh5e2517e2004-09-24 23:59:12 +0000710 void (*xDel)(void*),
711 int encoding
drh4f26d6c2004-05-26 23:25:30 +0000712){
713 Vdbe *p = (Vdbe *)pStmt;
714 Mem *pVar;
715 int rc;
716
717 rc = vdbeUnbind(p, i);
drhfeac5f82004-08-01 00:10:45 +0000718 if( rc || zData==0 ){
drh4f26d6c2004-05-26 23:25:30 +0000719 return rc;
720 }
drh290c1942004-08-21 17:54:45 +0000721 pVar = &p->aVar[i-1];
drh5e2517e2004-09-24 23:59:12 +0000722 rc = sqlite3VdbeMemSetStr(pVar, zData, nData, encoding, xDel);
drh5e2517e2004-09-24 23:59:12 +0000723 if( rc==SQLITE_OK && encoding!=0 ){
danielk197714db2662006-01-09 16:12:04 +0000724 rc = sqlite3VdbeChangeEncoding(pVar, ENC(p->db));
drh5e2517e2004-09-24 23:59:12 +0000725 }
danielk197754f01982006-01-18 15:25:17 +0000726
danielk1977771151b2006-01-17 13:21:40 +0000727 sqlite3Error(((Vdbe *)pStmt)->db, rc, 0);
danielk197754f01982006-01-18 15:25:17 +0000728 return sqlite3ApiExit(((Vdbe *)pStmt)->db, rc);
drh4f26d6c2004-05-26 23:25:30 +0000729}
drh5e2517e2004-09-24 23:59:12 +0000730
731
732/*
733** Bind a blob value to an SQL statement variable.
734*/
735int sqlite3_bind_blob(
736 sqlite3_stmt *pStmt,
737 int i,
738 const void *zData,
739 int nData,
740 void (*xDel)(void*)
741){
742 return bindText(pStmt, i, zData, nData, xDel, 0);
743}
drh4f26d6c2004-05-26 23:25:30 +0000744int sqlite3_bind_double(sqlite3_stmt *pStmt, int i, double rValue){
745 int rc;
746 Vdbe *p = (Vdbe *)pStmt;
drh4f26d6c2004-05-26 23:25:30 +0000747 rc = vdbeUnbind(p, i);
748 if( rc==SQLITE_OK ){
drh290c1942004-08-21 17:54:45 +0000749 sqlite3VdbeMemSetDouble(&p->aVar[i-1], rValue);
drh4f26d6c2004-05-26 23:25:30 +0000750 }
danielk1977f4618892004-06-28 13:09:11 +0000751 return rc;
drh4f26d6c2004-05-26 23:25:30 +0000752}
753int sqlite3_bind_int(sqlite3_stmt *p, int i, int iValue){
drhefad9992004-06-22 12:13:55 +0000754 return sqlite3_bind_int64(p, i, (i64)iValue);
drh4f26d6c2004-05-26 23:25:30 +0000755}
drhefad9992004-06-22 12:13:55 +0000756int sqlite3_bind_int64(sqlite3_stmt *pStmt, int i, sqlite_int64 iValue){
drh4f26d6c2004-05-26 23:25:30 +0000757 int rc;
758 Vdbe *p = (Vdbe *)pStmt;
759 rc = vdbeUnbind(p, i);
760 if( rc==SQLITE_OK ){
drh290c1942004-08-21 17:54:45 +0000761 sqlite3VdbeMemSetInt64(&p->aVar[i-1], iValue);
drh4f26d6c2004-05-26 23:25:30 +0000762 }
763 return rc;
764}
765int sqlite3_bind_null(sqlite3_stmt* p, int i){
766 return vdbeUnbind((Vdbe *)p, i);
767}
768int sqlite3_bind_text(
769 sqlite3_stmt *pStmt,
770 int i,
771 const char *zData,
772 int nData,
danielk1977d8123362004-06-12 09:25:12 +0000773 void (*xDel)(void*)
drh4f26d6c2004-05-26 23:25:30 +0000774){
drh5e2517e2004-09-24 23:59:12 +0000775 return bindText(pStmt, i, zData, nData, xDel, SQLITE_UTF8);
drh4f26d6c2004-05-26 23:25:30 +0000776}
drh6c626082004-11-14 21:56:29 +0000777#ifndef SQLITE_OMIT_UTF16
drh4f26d6c2004-05-26 23:25:30 +0000778int sqlite3_bind_text16(
779 sqlite3_stmt *pStmt,
780 int i,
781 const void *zData,
782 int nData,
danielk1977d8123362004-06-12 09:25:12 +0000783 void (*xDel)(void*)
drh4f26d6c2004-05-26 23:25:30 +0000784){
drh5e2517e2004-09-24 23:59:12 +0000785 return bindText(pStmt, i, zData, nData, xDel, SQLITE_UTF16NATIVE);
drh4f26d6c2004-05-26 23:25:30 +0000786}
drh6c626082004-11-14 21:56:29 +0000787#endif /* SQLITE_OMIT_UTF16 */
danielk197726e41442006-06-14 15:16:35 +0000788int sqlite3_bind_value(sqlite3_stmt *pStmt, int i, const sqlite3_value *pValue){
789 int rc;
790 Vdbe *p = (Vdbe *)pStmt;
791 rc = vdbeUnbind(p, i);
792 if( rc==SQLITE_OK ){
793 sqlite3VdbeMemCopy(&p->aVar[i-1], pValue);
794 }
795 return rc;
796}
drhb026e052007-05-02 01:34:31 +0000797int sqlite3_bind_zeroblob(sqlite3_stmt *pStmt, int i, int n){
798 int rc;
799 Vdbe *p = (Vdbe *)pStmt;
800 rc = vdbeUnbind(p, i);
801 if( rc==SQLITE_OK ){
802 sqlite3VdbeMemSetZeroBlob(&p->aVar[i-1], n);
803 }
804 return rc;
805}
drh75f6a032004-07-15 14:15:00 +0000806
807/*
808** Return the number of wildcards that can be potentially bound to.
809** This routine is added to support DBD::SQLite.
drh75f6a032004-07-15 14:15:00 +0000810*/
811int sqlite3_bind_parameter_count(sqlite3_stmt *pStmt){
drh92febd92004-08-20 18:34:20 +0000812 Vdbe *p = (Vdbe*)pStmt;
813 return p ? p->nVar : 0;
drh75f6a032004-07-15 14:15:00 +0000814}
drh895d7472004-08-20 16:02:39 +0000815
816/*
drhfa6bc002004-09-07 16:19:52 +0000817** Create a mapping from variable numbers to variable names
818** in the Vdbe.azVar[] array, if such a mapping does not already
819** exist.
drh895d7472004-08-20 16:02:39 +0000820*/
drhfa6bc002004-09-07 16:19:52 +0000821static void createVarMap(Vdbe *p){
drh895d7472004-08-20 16:02:39 +0000822 if( !p->okVar ){
823 int j;
824 Op *pOp;
825 for(j=0, pOp=p->aOp; j<p->nOp; j++, pOp++){
826 if( pOp->opcode==OP_Variable ){
827 assert( pOp->p1>0 && pOp->p1<=p->nVar );
828 p->azVar[pOp->p1-1] = pOp->p3;
829 }
830 }
831 p->okVar = 1;
832 }
drhfa6bc002004-09-07 16:19:52 +0000833}
834
835/*
836** Return the name of a wildcard parameter. Return NULL if the index
837** is out of range or if the wildcard is unnamed.
838**
839** The result is always UTF-8.
840*/
841const char *sqlite3_bind_parameter_name(sqlite3_stmt *pStmt, int i){
842 Vdbe *p = (Vdbe*)pStmt;
843 if( p==0 || i<1 || i>p->nVar ){
844 return 0;
845 }
846 createVarMap(p);
drh895d7472004-08-20 16:02:39 +0000847 return p->azVar[i-1];
848}
drhfa6bc002004-09-07 16:19:52 +0000849
850/*
851** Given a wildcard parameter name, return the index of the variable
852** with that name. If there is no variable with the given name,
853** return 0.
854*/
855int sqlite3_bind_parameter_index(sqlite3_stmt *pStmt, const char *zName){
856 Vdbe *p = (Vdbe*)pStmt;
857 int i;
858 if( p==0 ){
859 return 0;
860 }
861 createVarMap(p);
drh6a8903c2004-12-10 18:00:04 +0000862 if( zName ){
863 for(i=0; i<p->nVar; i++){
864 const char *z = p->azVar[i];
865 if( z && strcmp(z,zName)==0 ){
866 return i+1;
867 }
drhfa6bc002004-09-07 16:19:52 +0000868 }
869 }
870 return 0;
871}
drhf8db1bc2005-04-22 02:38:37 +0000872
drh51942bc2005-06-12 22:01:42 +0000873/*
drhf8db1bc2005-04-22 02:38:37 +0000874** Transfer all bindings from the first statement over to the second.
875** If the two statements contain a different number of bindings, then
876** an SQLITE_ERROR is returned.
877*/
878int sqlite3_transfer_bindings(sqlite3_stmt *pFromStmt, sqlite3_stmt *pToStmt){
879 Vdbe *pFrom = (Vdbe*)pFromStmt;
880 Vdbe *pTo = (Vdbe*)pToStmt;
881 int i, rc = SQLITE_OK;
882 if( (pFrom->magic!=VDBE_MAGIC_RUN && pFrom->magic!=VDBE_MAGIC_HALT)
883 || (pTo->magic!=VDBE_MAGIC_RUN && pTo->magic!=VDBE_MAGIC_HALT) ){
884 return SQLITE_MISUSE;
885 }
886 if( pFrom->nVar!=pTo->nVar ){
887 return SQLITE_ERROR;
888 }
889 for(i=0; rc==SQLITE_OK && i<pFrom->nVar; i++){
danielk1977950f0542006-01-18 05:51:57 +0000890 sqlite3MallocDisallow();
drhf8db1bc2005-04-22 02:38:37 +0000891 rc = sqlite3VdbeMemMove(&pTo->aVar[i], &pFrom->aVar[i]);
danielk1977950f0542006-01-18 05:51:57 +0000892 sqlite3MallocAllow();
drhf8db1bc2005-04-22 02:38:37 +0000893 }
drh4ac285a2006-09-15 07:28:50 +0000894 assert( rc==SQLITE_OK || rc==SQLITE_NOMEM );
drhf8db1bc2005-04-22 02:38:37 +0000895 return rc;
896}
drh51942bc2005-06-12 22:01:42 +0000897
898/*
899** Return the sqlite3* database handle to which the prepared statement given
900** in the argument belongs. This is the same database handle that was
901** the first argument to the sqlite3_prepare() that was used to create
902** the statement in the first place.
903*/
904sqlite3 *sqlite3_db_handle(sqlite3_stmt *pStmt){
905 return pStmt ? ((Vdbe*)pStmt)->db : 0;
906}