blob: 40a4a388eb2df0080038dac778e8b335f649c93f [file] [log] [blame]
drhcce7d172000-05-31 15:34:51 +00001/*
drhb19a2bc2001-09-16 00:13:26 +00002** 2001 September 15
drhcce7d172000-05-31 15:34:51 +00003**
drhb19a2bc2001-09-16 00:13:26 +00004** The author disclaims copyright to this source code. In place of
5** a legal notice, here is a blessing:
drhcce7d172000-05-31 15:34:51 +00006**
drhb19a2bc2001-09-16 00:13:26 +00007** 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.
drhcce7d172000-05-31 15:34:51 +000010**
11*************************************************************************
drh1ccde152000-06-17 13:12:39 +000012** This file contains routines used for analyzing expressions and
drhb19a2bc2001-09-16 00:13:26 +000013** for generating VDBE code that evaluates expressions in SQLite.
drhcce7d172000-05-31 15:34:51 +000014**
drha76b5df2002-02-23 02:32:10 +000015** $Id: expr.c,v 1.42 2002/02/23 02:32:10 drh Exp $
drhcce7d172000-05-31 15:34:51 +000016*/
17#include "sqliteInt.h"
18
drha2e00042002-01-22 03:13:42 +000019
20/*
drha76b5df2002-02-23 02:32:10 +000021** Construct a new expression node and return a pointer to it. Memory
22** for this node is obtained from sqliteMalloc(). The calling function
23** is responsible for making sure the node eventually gets freed.
24*/
25Expr *sqliteExpr(int op, Expr *pLeft, Expr *pRight, Token *pToken){
26 Expr *pNew;
27 pNew = sqliteMalloc( sizeof(Expr) );
28 if( pNew==0 ){
29 sqliteExprDelete(pLeft);
30 sqliteExprDelete(pRight);
31 return 0;
32 }
33 pNew->op = op;
34 pNew->pLeft = pLeft;
35 pNew->pRight = pRight;
36 if( pToken ){
37 pNew->token = *pToken;
38 }else{
39 pNew->token.z = 0;
40 pNew->token.n = 0;
41 }
42 if( pLeft && pRight ){
43 sqliteExprSpan(pNew, &pLeft->span, &pRight->span);
44 }else{
45 pNew->span = pNew->token;
46 }
47 return pNew;
48}
49
50/*
51** Set the Expr.token field of the given expression to span all
52** text between the two given tokens.
53*/
54void sqliteExprSpan(Expr *pExpr, Token *pLeft, Token *pRight){
55 if( pExpr ){
56 pExpr->span.z = pLeft->z;
57 pExpr->span.n = pRight->n + Addr(pRight->z) - Addr(pLeft->z);
58 }
59}
60
61/*
62** Construct a new expression node for a function with multiple
63** arguments.
64*/
65Expr *sqliteExprFunction(ExprList *pList, Token *pToken){
66 Expr *pNew;
67 pNew = sqliteMalloc( sizeof(Expr) );
68 if( pNew==0 ){
69 sqliteExprListDelete(pList);
70 return 0;
71 }
72 pNew->op = TK_FUNCTION;
73 pNew->pList = pList;
74 if( pToken ){
75 pNew->token = *pToken;
76 }else{
77 pNew->token.z = 0;
78 pNew->token.n = 0;
79 }
80 return pNew;
81}
82
83/*
drha2e00042002-01-22 03:13:42 +000084** Recursively delete an expression tree.
85*/
86void sqliteExprDelete(Expr *p){
87 if( p==0 ) return;
88 if( p->op!=TK_AS ){
89 if( p->pLeft ) sqliteExprDelete(p->pLeft);
90 if( p->pRight ) sqliteExprDelete(p->pRight);
91 }
92 if( p->pList ) sqliteExprListDelete(p->pList);
93 if( p->pSelect ) sqliteSelectDelete(p->pSelect);
94 sqliteFree(p);
95}
96
drhcce7d172000-05-31 15:34:51 +000097/*
drha76b5df2002-02-23 02:32:10 +000098** The following group of functions are used to translate the string
99** pointers of tokens in expression from one buffer to another.
100**
101** Normally, the Expr.token.z and Expr.span.z fields point into the
102** original input buffer of an SQL statement. This is usually OK
103** since the SQL statement is executed and the expression is deleted
104** before the input buffer is freed. Making the tokens point to the
105** original input buffer saves many calls to malloc() and thus helps
106** the library to run faster.
107**
108** But sometimes we need an expression to persist past the time when
109** the input buffer is freed. (Example: The SELECT clause of a
110** CREATE VIEW statement contains expressions that must persist for
111** the life of the view.) When that happens we have to make a
112** persistent copy of the input buffer and translate the Expr.token.z
113** and Expr.span.z fields to point to the copy rather than the
114** original input buffer. The following group of routines to that
115** translation.
116**
117** The "offset" parameter is the distance from the original input buffer
118** to the persistent copy. These routines recursively walk the entire
119** expression tree and shift all tokens by "offset" amount.
120**
121** The work of figuring out the appropriate "offset" and making the
122** presistent copy of the input buffer is done by the calling routine.
123*/
124void sqliteExprMoveStrings(Expr *p, int offset){
125 if( p==0 ) return;
126 if( p->token.z ) p->token.z += offset;
127 if( p->span.z ) p->span.z += offset;
128 if( p->pLeft ) sqliteExprMoveStrings(p->pLeft, offset);
129 if( p->pRight ) sqliteExprMoveStrings(p->pRight, offset);
130 if( p->pList ) sqliteExprListMoveStrings(p->pList, offset);
131 if( p->pSelect ) sqliteSelectMoveStrings(p->pSelect, offset);
132}
133void sqliteExprListMoveStrings(ExprList *pList, int offset){
134 int i;
135 if( pList==0 ) return;
136 for(i=0; i<pList->nExpr; i++){
137 sqliteExprMoveStrings(pList->a[i].pExpr, offset);
138 }
139}
140void sqliteSelectMoveStrings(Select *pSelect, int offset){
141 if( pSelect==0 ) return;
142 sqliteExprListMoveStrings(pSelect->pEList, offset);
143 sqliteExprMoveStrings(pSelect->pWhere, offset);
144 sqliteExprListMoveStrings(pSelect->pGroupBy, offset);
145 sqliteExprMoveStrings(pSelect->pHaving, offset);
146 sqliteExprListMoveStrings(pSelect->pOrderBy, offset);
147 sqliteSelectMoveStrings(pSelect->pPrior, offset);
148}
149
150/*
151** Add a new element to the end of an expression list. If pList is
152** initially NULL, then create a new expression list.
153*/
154ExprList *sqliteExprListAppend(ExprList *pList, Expr *pExpr, Token *pName){
155 int i;
156 if( pList==0 ){
157 pList = sqliteMalloc( sizeof(ExprList) );
158 if( pList==0 ){
159 sqliteExprDelete(pExpr);
160 return 0;
161 }
162 }
163 if( (pList->nExpr & 7)==0 ){
164 int n = pList->nExpr + 8;
165 struct ExprList_item *a;
166 a = sqliteRealloc(pList->a, n*sizeof(pList->a[0]));
167 if( a==0 ){
168 sqliteExprDelete(pExpr);
169 return pList;
170 }
171 pList->a = a;
172 }
173 if( pExpr || pName ){
174 i = pList->nExpr++;
175 pList->a[i].pExpr = pExpr;
176 pList->a[i].zName = 0;
177 if( pName ){
178 sqliteSetNString(&pList->a[i].zName, pName->z, pName->n, 0);
179 sqliteDequote(pList->a[i].zName);
180 }
181 }
182 return pList;
183}
184
185/*
186** Delete an entire expression list.
187*/
188void sqliteExprListDelete(ExprList *pList){
189 int i;
190 if( pList==0 ) return;
191 for(i=0; i<pList->nExpr; i++){
192 sqliteExprDelete(pList->a[i].pExpr);
193 sqliteFree(pList->a[i].zName);
194 }
195 sqliteFree(pList->a);
196 sqliteFree(pList);
197}
198
199/*
drhfef52082000-06-06 01:50:43 +0000200** Walk an expression tree. Return 1 if the expression is constant
201** and 0 if it involves variables.
202*/
drh92086432002-01-22 14:11:29 +0000203int sqliteExprIsConstant(Expr *p){
drhfef52082000-06-06 01:50:43 +0000204 switch( p->op ){
205 case TK_ID:
drh967e8b72000-06-21 13:59:10 +0000206 case TK_COLUMN:
drhfef52082000-06-06 01:50:43 +0000207 case TK_DOT:
208 return 0;
drh92086432002-01-22 14:11:29 +0000209 case TK_INTEGER:
210 case TK_FLOAT:
211 case TK_STRING:
212 return 1;
drhfef52082000-06-06 01:50:43 +0000213 default: {
drh92086432002-01-22 14:11:29 +0000214 if( p->pLeft && !sqliteExprIsConstant(p->pLeft) ) return 0;
215 if( p->pRight && !sqliteExprIsConstant(p->pRight) ) return 0;
drhfef52082000-06-06 01:50:43 +0000216 if( p->pList ){
217 int i;
218 for(i=0; i<p->pList->nExpr; i++){
drh92086432002-01-22 14:11:29 +0000219 if( !sqliteExprIsConstant(p->pList->a[i].pExpr) ) return 0;
drhfef52082000-06-06 01:50:43 +0000220 }
221 }
drh92086432002-01-22 14:11:29 +0000222 return p->pLeft!=0 || p->pRight!=0 || (p->pList && p->pList->nExpr>0);
drhfef52082000-06-06 01:50:43 +0000223 }
224 }
drh92086432002-01-22 14:11:29 +0000225 return 0;
drhfef52082000-06-06 01:50:43 +0000226}
227
228/*
drh4794b982000-06-06 13:54:14 +0000229** Walk the expression tree and process operators of the form:
230**
231** expr IN (SELECT ...)
232**
drh967e8b72000-06-21 13:59:10 +0000233** These operators have to be processed before column names are
drh4794b982000-06-06 13:54:14 +0000234** resolved because each such operator increments pParse->nTab
drh1ccde152000-06-17 13:12:39 +0000235** to reserve cursor numbers for its own use. But pParse->nTab
drhaacc5432002-01-06 17:07:40 +0000236** needs to be constant once we begin resolving column names. For
237** that reason, this procedure needs to be called on every expression
238** before sqliteExprResolveIds() is called on any expression.
drh4794b982000-06-06 13:54:14 +0000239**
240** Actually, the processing of IN-SELECT is only started by this
241** routine. This routine allocates a cursor number to the IN-SELECT
242** and then moves on. The code generation is done by
243** sqliteExprResolveIds() which must be called afterwards.
244*/
245void sqliteExprResolveInSelect(Parse *pParse, Expr *pExpr){
246 if( pExpr==0 ) return;
247 if( pExpr->op==TK_IN && pExpr->pSelect!=0 ){
248 pExpr->iTable = pParse->nTab++;
249 }else{
250 if( pExpr->pLeft ) sqliteExprResolveInSelect(pParse, pExpr->pLeft);
251 if( pExpr->pRight ) sqliteExprResolveInSelect(pParse, pExpr->pRight);
252 if( pExpr->pList ){
253 int i;
254 ExprList *pList = pExpr->pList;
255 for(i=0; i<pList->nExpr; i++){
256 sqliteExprResolveInSelect(pParse, pList->a[i].pExpr);
257 }
258 }
259 }
260}
261
262/*
drhc4a3c772001-04-04 11:48:57 +0000263** Return TRUE if the given string is a row-id column name.
264*/
265static int sqliteIsRowid(const char *z){
266 if( sqliteStrICmp(z, "_ROWID_")==0 ) return 1;
267 if( sqliteStrICmp(z, "ROWID")==0 ) return 1;
268 if( sqliteStrICmp(z, "OID")==0 ) return 1;
269 return 0;
270}
271
272/*
drhcce7d172000-05-31 15:34:51 +0000273** This routine walks an expression tree and resolves references to
drh967e8b72000-06-21 13:59:10 +0000274** table columns. Nodes of the form ID.ID or ID resolve into an
drhaacc5432002-01-06 17:07:40 +0000275** index to the table in the table list and a column offset. The
276** Expr.opcode for such nodes is changed to TK_COLUMN. The Expr.iTable
277** value is changed to the index of the referenced table in pTabList
278** plus the pParse->nTab value. This value will ultimately become the
279** VDBE cursor number for a cursor that is pointing into the referenced
280** table. The Expr.iColumn value is changed to the index of the column
281** of the referenced table. The Expr.iColumn value for the special
282** ROWID column is -1. Any INTEGER PRIMARY KEY column is tried as an
283** alias for ROWID.
drh19a775c2000-06-05 18:54:46 +0000284**
drhfef52082000-06-06 01:50:43 +0000285** We also check for instances of the IN operator. IN comes in two
286** forms:
287**
288** expr IN (exprlist)
289** and
290** expr IN (SELECT ...)
291**
292** The first form is handled by creating a set holding the list
293** of allowed values. The second form causes the SELECT to generate
294** a temporary table.
295**
296** This routine also looks for scalar SELECTs that are part of an expression.
drh19a775c2000-06-05 18:54:46 +0000297** If it finds any, it generates code to write the value of that select
298** into a memory cell.
drhcce7d172000-05-31 15:34:51 +0000299**
drh967e8b72000-06-21 13:59:10 +0000300** Unknown columns or tables provoke an error. The function returns
drhcce7d172000-05-31 15:34:51 +0000301** the number of errors seen and leaves an error message on pParse->zErrMsg.
302*/
drha2e00042002-01-22 03:13:42 +0000303int sqliteExprResolveIds(
304 Parse *pParse, /* The parser context */
305 IdList *pTabList, /* List of tables used to resolve column names */
306 ExprList *pEList, /* List of expressions used to resolve "AS" */
307 Expr *pExpr /* The expression to be analyzed. */
308){
drhdaffd0e2001-04-11 14:28:42 +0000309 if( pExpr==0 || pTabList==0 ) return 0;
drhcce7d172000-05-31 15:34:51 +0000310 switch( pExpr->op ){
drha2e00042002-01-22 03:13:42 +0000311 /* A lone identifier. Try and match it as follows:
312 **
313 ** 1. To the name of a column of one of the tables in pTabList
314 **
315 ** 2. To the right side of an AS keyword in the column list of
316 ** a SELECT statement. (For example, match against 'x' in
317 ** "SELECT a+b AS 'x' FROM t1".)
318 **
319 ** 3. One of the special names "ROWID", "OID", or "_ROWID_".
320 */
drhcce7d172000-05-31 15:34:51 +0000321 case TK_ID: {
drhc4a3c772001-04-04 11:48:57 +0000322 int cnt = 0; /* Number of matches */
323 int i; /* Loop counter */
drha76b5df2002-02-23 02:32:10 +0000324 char *z;
325 assert( pExpr->token.z );
326 z = sqliteStrNDup(pExpr->token.z, pExpr->token.n);
drh2f4392f2002-02-14 21:42:51 +0000327 sqliteDequote(z);
drhdaffd0e2001-04-11 14:28:42 +0000328 if( z==0 ) return 1;
drhcce7d172000-05-31 15:34:51 +0000329 for(i=0; i<pTabList->nId; i++){
330 int j;
331 Table *pTab = pTabList->a[i].pTab;
332 if( pTab==0 ) continue;
333 for(j=0; j<pTab->nCol; j++){
drh7020f652000-06-03 18:06:52 +0000334 if( sqliteStrICmp(pTab->aCol[j].zName, z)==0 ){
drhcce7d172000-05-31 15:34:51 +0000335 cnt++;
drh19a775c2000-06-05 18:54:46 +0000336 pExpr->iTable = i + pParse->nTab;
drh4a324312001-12-21 14:30:42 +0000337 if( j==pTab->iPKey ){
338 /* Substitute the record number for the INTEGER PRIMARY KEY */
339 pExpr->iColumn = -1;
340 }else{
341 pExpr->iColumn = j;
342 }
drha2e00042002-01-22 03:13:42 +0000343 pExpr->op = TK_COLUMN;
drhcce7d172000-05-31 15:34:51 +0000344 }
345 }
346 }
drha2e00042002-01-22 03:13:42 +0000347 if( cnt==0 && pEList!=0 ){
348 int j;
349 for(j=0; j<pEList->nExpr; j++){
350 char *zAs = pEList->a[j].zName;
351 if( zAs!=0 && sqliteStrICmp(zAs, z)==0 ){
352 cnt++;
353 assert( pExpr->pLeft==0 && pExpr->pRight==0 );
354 pExpr->op = TK_AS;
355 pExpr->iColumn = j;
356 pExpr->pLeft = pEList->a[j].pExpr;
357 }
358 }
359 }
drhc4a3c772001-04-04 11:48:57 +0000360 if( cnt==0 && sqliteIsRowid(z) ){
361 pExpr->iColumn = -1;
362 pExpr->iTable = pParse->nTab;
363 cnt = 1 + (pTabList->nId>1);
drha2e00042002-01-22 03:13:42 +0000364 pExpr->op = TK_COLUMN;
drhc4a3c772001-04-04 11:48:57 +0000365 }
drhcce7d172000-05-31 15:34:51 +0000366 sqliteFree(z);
367 if( cnt==0 ){
drh967e8b72000-06-21 13:59:10 +0000368 sqliteSetNString(&pParse->zErrMsg, "no such column: ", -1,
drhcce7d172000-05-31 15:34:51 +0000369 pExpr->token.z, pExpr->token.n, 0);
370 pParse->nErr++;
371 return 1;
372 }else if( cnt>1 ){
drh967e8b72000-06-21 13:59:10 +0000373 sqliteSetNString(&pParse->zErrMsg, "ambiguous column name: ", -1,
drhcce7d172000-05-31 15:34:51 +0000374 pExpr->token.z, pExpr->token.n, 0);
375 pParse->nErr++;
376 return 1;
377 }
drhcce7d172000-05-31 15:34:51 +0000378 break;
379 }
380
drh967e8b72000-06-21 13:59:10 +0000381 /* A table name and column name: ID.ID */
drhcce7d172000-05-31 15:34:51 +0000382 case TK_DOT: {
drhbed86902000-06-02 13:27:59 +0000383 int cnt = 0; /* Number of matches */
drhc4a3c772001-04-04 11:48:57 +0000384 int cntTab = 0; /* Number of matching tables */
drhbed86902000-06-02 13:27:59 +0000385 int i; /* Loop counter */
drhcce7d172000-05-31 15:34:51 +0000386 Expr *pLeft, *pRight; /* Left and right subbranches of the expr */
drhcce7d172000-05-31 15:34:51 +0000387 char *zLeft, *zRight; /* Text of an identifier */
388
389 pLeft = pExpr->pLeft;
390 pRight = pExpr->pRight;
drha76b5df2002-02-23 02:32:10 +0000391 assert( pLeft && pLeft->op==TK_ID && pLeft->token.z );
392 assert( pRight && pRight->op==TK_ID && pRight->token.z );
drh6e142f52000-06-08 13:36:40 +0000393 zLeft = sqliteStrNDup(pLeft->token.z, pLeft->token.n);
394 zRight = sqliteStrNDup(pRight->token.z, pRight->token.n);
drhdaffd0e2001-04-11 14:28:42 +0000395 if( zLeft==0 || zRight==0 ){
396 sqliteFree(zLeft);
397 sqliteFree(zRight);
398 return 1;
399 }
drh87c40e82001-07-23 14:33:02 +0000400 sqliteDequote(zLeft);
401 sqliteDequote(zRight);
drhc4a3c772001-04-04 11:48:57 +0000402 pExpr->iTable = -1;
drhcce7d172000-05-31 15:34:51 +0000403 for(i=0; i<pTabList->nId; i++){
404 int j;
405 char *zTab;
406 Table *pTab = pTabList->a[i].pTab;
407 if( pTab==0 ) continue;
408 if( pTabList->a[i].zAlias ){
409 zTab = pTabList->a[i].zAlias;
410 }else{
411 zTab = pTab->zName;
412 }
413 if( sqliteStrICmp(zTab, zLeft)!=0 ) continue;
drhc4a3c772001-04-04 11:48:57 +0000414 if( 0==(cntTab++) ) pExpr->iTable = i + pParse->nTab;
drhcce7d172000-05-31 15:34:51 +0000415 for(j=0; j<pTab->nCol; j++){
drh7020f652000-06-03 18:06:52 +0000416 if( sqliteStrICmp(pTab->aCol[j].zName, zRight)==0 ){
drhcce7d172000-05-31 15:34:51 +0000417 cnt++;
drh19a775c2000-06-05 18:54:46 +0000418 pExpr->iTable = i + pParse->nTab;
drh4a324312001-12-21 14:30:42 +0000419 if( j==pTab->iPKey ){
420 /* Substitute the record number for the INTEGER PRIMARY KEY */
421 pExpr->iColumn = -1;
422 }else{
423 pExpr->iColumn = j;
424 }
drhcce7d172000-05-31 15:34:51 +0000425 }
426 }
427 }
drhc4a3c772001-04-04 11:48:57 +0000428 if( cnt==0 && cntTab==1 && sqliteIsRowid(zRight) ){
429 cnt = 1;
430 pExpr->iColumn = -1;
431 }
drhcce7d172000-05-31 15:34:51 +0000432 sqliteFree(zLeft);
433 sqliteFree(zRight);
434 if( cnt==0 ){
drh967e8b72000-06-21 13:59:10 +0000435 sqliteSetNString(&pParse->zErrMsg, "no such column: ", -1,
drhcce7d172000-05-31 15:34:51 +0000436 pLeft->token.z, pLeft->token.n, ".", 1,
437 pRight->token.z, pRight->token.n, 0);
438 pParse->nErr++;
439 return 1;
440 }else if( cnt>1 ){
drh967e8b72000-06-21 13:59:10 +0000441 sqliteSetNString(&pParse->zErrMsg, "ambiguous column name: ", -1,
drhcce7d172000-05-31 15:34:51 +0000442 pLeft->token.z, pLeft->token.n, ".", 1,
443 pRight->token.z, pRight->token.n, 0);
444 pParse->nErr++;
445 return 1;
446 }
447 sqliteExprDelete(pLeft);
448 pExpr->pLeft = 0;
449 sqliteExprDelete(pRight);
450 pExpr->pRight = 0;
drh967e8b72000-06-21 13:59:10 +0000451 pExpr->op = TK_COLUMN;
drhcce7d172000-05-31 15:34:51 +0000452 break;
453 }
454
drhfef52082000-06-06 01:50:43 +0000455 case TK_IN: {
drhd8bc7082000-06-07 23:51:50 +0000456 Vdbe *v = sqliteGetVdbe(pParse);
drhfef52082000-06-06 01:50:43 +0000457 if( v==0 ) return 1;
drha2e00042002-01-22 03:13:42 +0000458 if( sqliteExprResolveIds(pParse, pTabList, pEList, pExpr->pLeft) ){
drhcfab11b2000-06-06 03:31:22 +0000459 return 1;
460 }
drhfef52082000-06-06 01:50:43 +0000461 if( pExpr->pSelect ){
462 /* Case 1: expr IN (SELECT ...)
463 **
464 ** Generate code to write the results of the select into a temporary
drh4794b982000-06-06 13:54:14 +0000465 ** table. The cursor number of the temporary table has already
466 ** been put in iTable by sqliteExprResolveInSelect().
drhfef52082000-06-06 01:50:43 +0000467 */
drhc6b52df2002-01-04 03:09:29 +0000468 sqliteVdbeAddOp(v, OP_OpenTemp, pExpr->iTable, 1);
drhfef52082000-06-06 01:50:43 +0000469 if( sqliteSelect(pParse, pExpr->pSelect, SRT_Set, pExpr->iTable) );
470 }else if( pExpr->pList ){
471 /* Case 2: expr IN (exprlist)
472 **
473 ** Create a set to put the exprlist values in. The Set id is stored
474 ** in iTable.
475 */
476 int i, iSet;
477 for(i=0; i<pExpr->pList->nExpr; i++){
478 Expr *pE2 = pExpr->pList->a[i].pExpr;
drh92086432002-01-22 14:11:29 +0000479 if( !sqliteExprIsConstant(pE2) ){
drhfef52082000-06-06 01:50:43 +0000480 sqliteSetString(&pParse->zErrMsg,
481 "right-hand side of IN operator must be constant", 0);
482 pParse->nErr++;
483 return 1;
484 }
drh4794b982000-06-06 13:54:14 +0000485 if( sqliteExprCheck(pParse, pE2, 0, 0) ){
486 return 1;
487 }
drhfef52082000-06-06 01:50:43 +0000488 }
489 iSet = pExpr->iTable = pParse->nSet++;
490 for(i=0; i<pExpr->pList->nExpr; i++){
491 Expr *pE2 = pExpr->pList->a[i].pExpr;
492 switch( pE2->op ){
493 case TK_FLOAT:
494 case TK_INTEGER:
495 case TK_STRING: {
drh99fcd712001-10-13 01:06:47 +0000496 int addr = sqliteVdbeAddOp(v, OP_SetInsert, iSet, 0);
drha76b5df2002-02-23 02:32:10 +0000497 assert( pE2->token.z );
drhfef52082000-06-06 01:50:43 +0000498 sqliteVdbeChangeP3(v, addr, pE2->token.z, pE2->token.n);
499 sqliteVdbeDequoteP3(v, addr);
500 break;
501 }
502 default: {
503 sqliteExprCode(pParse, pE2);
drh99fcd712001-10-13 01:06:47 +0000504 sqliteVdbeAddOp(v, OP_SetInsert, iSet, 0);
drhfef52082000-06-06 01:50:43 +0000505 break;
506 }
507 }
508 }
509 }
drhcfab11b2000-06-06 03:31:22 +0000510 break;
drhfef52082000-06-06 01:50:43 +0000511 }
512
drh19a775c2000-06-05 18:54:46 +0000513 case TK_SELECT: {
drhfef52082000-06-06 01:50:43 +0000514 /* This has to be a scalar SELECT. Generate code to put the
515 ** value of this select in a memory cell and record the number
drh967e8b72000-06-21 13:59:10 +0000516 ** of the memory cell in iColumn.
drhfef52082000-06-06 01:50:43 +0000517 */
drh967e8b72000-06-21 13:59:10 +0000518 pExpr->iColumn = pParse->nMem++;
519 if( sqliteSelect(pParse, pExpr->pSelect, SRT_Mem, pExpr->iColumn) ){
drh19a775c2000-06-05 18:54:46 +0000520 return 1;
521 }
522 break;
523 }
524
drhcce7d172000-05-31 15:34:51 +0000525 /* For all else, just recursively walk the tree */
526 default: {
drh4794b982000-06-06 13:54:14 +0000527 if( pExpr->pLeft
drha2e00042002-01-22 03:13:42 +0000528 && sqliteExprResolveIds(pParse, pTabList, pEList, pExpr->pLeft) ){
drhcce7d172000-05-31 15:34:51 +0000529 return 1;
530 }
531 if( pExpr->pRight
drha2e00042002-01-22 03:13:42 +0000532 && sqliteExprResolveIds(pParse, pTabList, pEList, pExpr->pRight) ){
drhcce7d172000-05-31 15:34:51 +0000533 return 1;
534 }
535 if( pExpr->pList ){
536 int i;
537 ExprList *pList = pExpr->pList;
538 for(i=0; i<pList->nExpr; i++){
drha2e00042002-01-22 03:13:42 +0000539 if( sqliteExprResolveIds(pParse,pTabList,pEList,pList->a[i].pExpr) ){
drhcce7d172000-05-31 15:34:51 +0000540 return 1;
541 }
542 }
543 }
544 }
545 }
546 return 0;
547}
548
549#if 0 /* NOT USED */
550/*
551** Compare a token against a string. Return TRUE if they match.
552*/
553static int sqliteTokenCmp(Token *pToken, const char *zStr){
554 int n = strlen(zStr);
555 if( n!=pToken->n ) return 0;
556 return sqliteStrNICmp(pToken->z, zStr, n)==0;
557}
558#endif
559
560/*
561** Convert a function name into its integer identifier. Return the
562** identifier. Return FN_Unknown if the function name is unknown.
563*/
564int sqliteFuncId(Token *pToken){
565 static const struct {
566 char *zName;
567 int len;
568 int id;
569 } aFunc[] = {
drhbf4133c2001-10-13 02:59:08 +0000570 { "count", 5, FN_Count },
571 { "min", 3, FN_Min },
572 { "max", 3, FN_Max },
573 { "sum", 3, FN_Sum },
574 { "avg", 3, FN_Avg },
drhbf4133c2001-10-13 02:59:08 +0000575 { "length", 6, FN_Length },
576 { "substr", 6, FN_Substr },
577 { "abs", 3, FN_Abs },
578 { "round", 5, FN_Round },
drhcce7d172000-05-31 15:34:51 +0000579 };
580 int i;
581 for(i=0; i<ArraySize(aFunc); i++){
582 if( aFunc[i].len==pToken->n
583 && sqliteStrNICmp(pToken->z, aFunc[i].zName, aFunc[i].len)==0 ){
584 return aFunc[i].id;
585 }
586 }
587 return FN_Unknown;
588}
589
590/*
591** Error check the functions in an expression. Make sure all
592** function names are recognized and all functions have the correct
593** number of arguments. Leave an error message in pParse->zErrMsg
594** if anything is amiss. Return the number of errors.
595**
596** if pIsAgg is not null and this expression is an aggregate function
597** (like count(*) or max(value)) then write a 1 into *pIsAgg.
598*/
599int sqliteExprCheck(Parse *pParse, Expr *pExpr, int allowAgg, int *pIsAgg){
600 int nErr = 0;
601 if( pExpr==0 ) return 0;
drhcce7d172000-05-31 15:34:51 +0000602 switch( pExpr->op ){
603 case TK_FUNCTION: {
604 int id = sqliteFuncId(&pExpr->token);
605 int n = pExpr->pList ? pExpr->pList->nExpr : 0;
606 int no_such_func = 0;
607 int too_many_args = 0;
608 int too_few_args = 0;
609 int is_agg = 0;
610 int i;
drh967e8b72000-06-21 13:59:10 +0000611 pExpr->iColumn = id;
drhcce7d172000-05-31 15:34:51 +0000612 switch( id ){
613 case FN_Unknown: {
614 no_such_func = 1;
615 break;
616 }
617 case FN_Count: {
618 no_such_func = !allowAgg;
619 too_many_args = n>1;
620 is_agg = 1;
621 break;
622 }
623 case FN_Max:
624 case FN_Min: {
625 too_few_args = allowAgg ? n<1 : n<2;
626 is_agg = n==1;
627 break;
628 }
drh22827922000-06-06 17:27:05 +0000629 case FN_Avg:
drhcce7d172000-05-31 15:34:51 +0000630 case FN_Sum: {
631 no_such_func = !allowAgg;
632 too_many_args = n>1;
633 too_few_args = n<1;
634 is_agg = 1;
635 break;
636 }
drhbf4133c2001-10-13 02:59:08 +0000637 case FN_Abs:
drh6ec27332000-08-28 15:51:43 +0000638 case FN_Length: {
639 too_few_args = n<1;
640 too_many_args = n>1;
641 break;
642 }
drhbf4133c2001-10-13 02:59:08 +0000643 case FN_Round: {
644 too_few_args = n<1;
645 too_many_args = n>2;
646 break;
647 }
drh6ec27332000-08-28 15:51:43 +0000648 case FN_Substr: {
649 too_few_args = n<3;
650 too_many_args = n>3;
651 break;
652 }
drhcce7d172000-05-31 15:34:51 +0000653 default: break;
654 }
655 if( no_such_func ){
656 sqliteSetNString(&pParse->zErrMsg, "no such function: ", -1,
657 pExpr->token.z, pExpr->token.n, 0);
658 pParse->nErr++;
659 nErr++;
660 }else if( too_many_args ){
661 sqliteSetNString(&pParse->zErrMsg, "too many arguments to function ",-1,
662 pExpr->token.z, pExpr->token.n, "()", 2, 0);
663 pParse->nErr++;
664 nErr++;
665 }else if( too_few_args ){
666 sqliteSetNString(&pParse->zErrMsg, "too few arguments to function ",-1,
667 pExpr->token.z, pExpr->token.n, "()", 2, 0);
668 pParse->nErr++;
669 nErr++;
670 }
drh22827922000-06-06 17:27:05 +0000671 if( is_agg ) pExpr->op = TK_AGG_FUNCTION;
drhcce7d172000-05-31 15:34:51 +0000672 if( is_agg && pIsAgg ) *pIsAgg = 1;
673 for(i=0; nErr==0 && i<n; i++){
drh4cfa7932000-06-08 15:10:46 +0000674 nErr = sqliteExprCheck(pParse, pExpr->pList->a[i].pExpr,
675 allowAgg && !is_agg, pIsAgg);
drhcce7d172000-05-31 15:34:51 +0000676 }
677 }
678 default: {
679 if( pExpr->pLeft ){
drh22827922000-06-06 17:27:05 +0000680 nErr = sqliteExprCheck(pParse, pExpr->pLeft, allowAgg, pIsAgg);
drhcce7d172000-05-31 15:34:51 +0000681 }
682 if( nErr==0 && pExpr->pRight ){
drh22827922000-06-06 17:27:05 +0000683 nErr = sqliteExprCheck(pParse, pExpr->pRight, allowAgg, pIsAgg);
drhcce7d172000-05-31 15:34:51 +0000684 }
drhfef52082000-06-06 01:50:43 +0000685 if( nErr==0 && pExpr->pList ){
686 int n = pExpr->pList->nExpr;
687 int i;
688 for(i=0; nErr==0 && i<n; i++){
drh22827922000-06-06 17:27:05 +0000689 Expr *pE2 = pExpr->pList->a[i].pExpr;
690 nErr = sqliteExprCheck(pParse, pE2, allowAgg, pIsAgg);
drhfef52082000-06-06 01:50:43 +0000691 }
692 }
drhcce7d172000-05-31 15:34:51 +0000693 break;
694 }
695 }
696 return nErr;
697}
698
699/*
700** Generate code into the current Vdbe to evaluate the given
drh1ccde152000-06-17 13:12:39 +0000701** expression and leave the result on the top of stack.
drhcce7d172000-05-31 15:34:51 +0000702*/
703void sqliteExprCode(Parse *pParse, Expr *pExpr){
704 Vdbe *v = pParse->pVdbe;
705 int op;
drhdaffd0e2001-04-11 14:28:42 +0000706 if( v==0 || pExpr==0 ) return;
drhcce7d172000-05-31 15:34:51 +0000707 switch( pExpr->op ){
708 case TK_PLUS: op = OP_Add; break;
709 case TK_MINUS: op = OP_Subtract; break;
710 case TK_STAR: op = OP_Multiply; break;
711 case TK_SLASH: op = OP_Divide; break;
712 case TK_AND: op = OP_And; break;
713 case TK_OR: op = OP_Or; break;
714 case TK_LT: op = OP_Lt; break;
715 case TK_LE: op = OP_Le; break;
716 case TK_GT: op = OP_Gt; break;
717 case TK_GE: op = OP_Ge; break;
718 case TK_NE: op = OP_Ne; break;
719 case TK_EQ: op = OP_Eq; break;
720 case TK_LIKE: op = OP_Like; break;
721 case TK_GLOB: op = OP_Glob; break;
722 case TK_ISNULL: op = OP_IsNull; break;
723 case TK_NOTNULL: op = OP_NotNull; break;
724 case TK_NOT: op = OP_Not; break;
725 case TK_UMINUS: op = OP_Negative; break;
drhbf4133c2001-10-13 02:59:08 +0000726 case TK_BITAND: op = OP_BitAnd; break;
727 case TK_BITOR: op = OP_BitOr; break;
728 case TK_BITNOT: op = OP_BitNot; break;
729 case TK_LSHIFT: op = OP_ShiftLeft; break;
730 case TK_RSHIFT: op = OP_ShiftRight; break;
731 case TK_REM: op = OP_Remainder; break;
drhcce7d172000-05-31 15:34:51 +0000732 default: break;
733 }
734 switch( pExpr->op ){
drh967e8b72000-06-21 13:59:10 +0000735 case TK_COLUMN: {
drh22827922000-06-06 17:27:05 +0000736 if( pParse->useAgg ){
drh99fcd712001-10-13 01:06:47 +0000737 sqliteVdbeAddOp(v, OP_AggGet, 0, pExpr->iAgg);
drhc4a3c772001-04-04 11:48:57 +0000738 }else if( pExpr->iColumn>=0 ){
drh99fcd712001-10-13 01:06:47 +0000739 sqliteVdbeAddOp(v, OP_Column, pExpr->iTable, pExpr->iColumn);
drhc4a3c772001-04-04 11:48:57 +0000740 }else{
drh99fcd712001-10-13 01:06:47 +0000741 sqliteVdbeAddOp(v, OP_Recno, pExpr->iTable, 0);
drh22827922000-06-06 17:27:05 +0000742 }
drhcce7d172000-05-31 15:34:51 +0000743 break;
744 }
drhef6764a2002-01-30 04:32:00 +0000745 case TK_FLOAT:
drhcce7d172000-05-31 15:34:51 +0000746 case TK_INTEGER: {
drh7a7c7392001-11-24 00:31:46 +0000747 sqliteVdbeAddOp(v, OP_String, 0, 0);
drha76b5df2002-02-23 02:32:10 +0000748 assert( pExpr->token.z );
drh7a7c7392001-11-24 00:31:46 +0000749 sqliteVdbeChangeP3(v, -1, pExpr->token.z, pExpr->token.n);
drhcce7d172000-05-31 15:34:51 +0000750 break;
751 }
drhcce7d172000-05-31 15:34:51 +0000752 case TK_STRING: {
drh99fcd712001-10-13 01:06:47 +0000753 int addr = sqliteVdbeAddOp(v, OP_String, 0, 0);
drha76b5df2002-02-23 02:32:10 +0000754 assert( pExpr->token.z );
drhcce7d172000-05-31 15:34:51 +0000755 sqliteVdbeChangeP3(v, addr, pExpr->token.z, pExpr->token.n);
756 sqliteVdbeDequoteP3(v, addr);
757 break;
758 }
759 case TK_NULL: {
drh99fcd712001-10-13 01:06:47 +0000760 sqliteVdbeAddOp(v, OP_String, 0, 0);
drhcce7d172000-05-31 15:34:51 +0000761 break;
762 }
763 case TK_AND:
764 case TK_OR:
765 case TK_PLUS:
766 case TK_STAR:
767 case TK_MINUS:
drhbf4133c2001-10-13 02:59:08 +0000768 case TK_REM:
769 case TK_BITAND:
770 case TK_BITOR:
drhcce7d172000-05-31 15:34:51 +0000771 case TK_SLASH: {
772 sqliteExprCode(pParse, pExpr->pLeft);
773 sqliteExprCode(pParse, pExpr->pRight);
drh99fcd712001-10-13 01:06:47 +0000774 sqliteVdbeAddOp(v, op, 0, 0);
drhcce7d172000-05-31 15:34:51 +0000775 break;
776 }
drhbf4133c2001-10-13 02:59:08 +0000777 case TK_LSHIFT:
778 case TK_RSHIFT: {
779 sqliteExprCode(pParse, pExpr->pRight);
780 sqliteExprCode(pParse, pExpr->pLeft);
781 sqliteVdbeAddOp(v, op, 0, 0);
782 break;
783 }
drh00400772000-06-16 20:51:26 +0000784 case TK_CONCAT: {
785 sqliteExprCode(pParse, pExpr->pLeft);
786 sqliteExprCode(pParse, pExpr->pRight);
drh99fcd712001-10-13 01:06:47 +0000787 sqliteVdbeAddOp(v, OP_Concat, 2, 0);
drh00400772000-06-16 20:51:26 +0000788 break;
789 }
drhcce7d172000-05-31 15:34:51 +0000790 case TK_LT:
791 case TK_LE:
792 case TK_GT:
793 case TK_GE:
794 case TK_NE:
795 case TK_EQ:
796 case TK_LIKE:
797 case TK_GLOB: {
798 int dest;
drh99fcd712001-10-13 01:06:47 +0000799 sqliteVdbeAddOp(v, OP_Integer, 1, 0);
drhcce7d172000-05-31 15:34:51 +0000800 sqliteExprCode(pParse, pExpr->pLeft);
801 sqliteExprCode(pParse, pExpr->pRight);
802 dest = sqliteVdbeCurrentAddr(v) + 2;
drh99fcd712001-10-13 01:06:47 +0000803 sqliteVdbeAddOp(v, op, 0, dest);
804 sqliteVdbeAddOp(v, OP_AddImm, -1, 0);
drhcce7d172000-05-31 15:34:51 +0000805 break;
806 }
drhcce7d172000-05-31 15:34:51 +0000807 case TK_UMINUS: {
drh6e142f52000-06-08 13:36:40 +0000808 assert( pExpr->pLeft );
drh7a7c7392001-11-24 00:31:46 +0000809 if( pExpr->pLeft->op==TK_FLOAT || pExpr->pLeft->op==TK_INTEGER ){
drh6e142f52000-06-08 13:36:40 +0000810 Token *p = &pExpr->pLeft->token;
811 char *z = sqliteMalloc( p->n + 2 );
812 sprintf(z, "-%.*s", p->n, p->z);
drh99fcd712001-10-13 01:06:47 +0000813 sqliteVdbeAddOp(v, OP_String, 0, 0);
814 sqliteVdbeChangeP3(v, -1, z, p->n+1);
drh6e142f52000-06-08 13:36:40 +0000815 sqliteFree(z);
816 break;
817 }
drh1ccde152000-06-17 13:12:39 +0000818 /* Fall through into TK_NOT */
drh6e142f52000-06-08 13:36:40 +0000819 }
drhbf4133c2001-10-13 02:59:08 +0000820 case TK_BITNOT:
drh6e142f52000-06-08 13:36:40 +0000821 case TK_NOT: {
drhcce7d172000-05-31 15:34:51 +0000822 sqliteExprCode(pParse, pExpr->pLeft);
drh99fcd712001-10-13 01:06:47 +0000823 sqliteVdbeAddOp(v, op, 0, 0);
drhcce7d172000-05-31 15:34:51 +0000824 break;
825 }
826 case TK_ISNULL:
827 case TK_NOTNULL: {
828 int dest;
drh99fcd712001-10-13 01:06:47 +0000829 sqliteVdbeAddOp(v, OP_Integer, 1, 0);
drhcce7d172000-05-31 15:34:51 +0000830 sqliteExprCode(pParse, pExpr->pLeft);
831 dest = sqliteVdbeCurrentAddr(v) + 2;
drh99fcd712001-10-13 01:06:47 +0000832 sqliteVdbeAddOp(v, op, 0, dest);
833 sqliteVdbeAddOp(v, OP_AddImm, -1, 0);
drhcce7d172000-05-31 15:34:51 +0000834 break;
835 }
drh22827922000-06-06 17:27:05 +0000836 case TK_AGG_FUNCTION: {
drh99fcd712001-10-13 01:06:47 +0000837 sqliteVdbeAddOp(v, OP_AggGet, 0, pExpr->iAgg);
drh967e8b72000-06-21 13:59:10 +0000838 if( pExpr->iColumn==FN_Avg ){
drh22827922000-06-06 17:27:05 +0000839 assert( pParse->iAggCount>=0 && pParse->iAggCount<pParse->nAgg );
drh99fcd712001-10-13 01:06:47 +0000840 sqliteVdbeAddOp(v, OP_AggGet, 0, pParse->iAggCount);
841 sqliteVdbeAddOp(v, OP_Divide, 0, 0);
drh22827922000-06-06 17:27:05 +0000842 }
843 break;
844 }
drhcce7d172000-05-31 15:34:51 +0000845 case TK_FUNCTION: {
drh967e8b72000-06-21 13:59:10 +0000846 int id = pExpr->iColumn;
drhcce7d172000-05-31 15:34:51 +0000847 int op;
848 int i;
849 ExprList *pList = pExpr->pList;
drh6ec27332000-08-28 15:51:43 +0000850 switch( id ){
drh6ec27332000-08-28 15:51:43 +0000851 case FN_Min:
852 case FN_Max: {
853 op = id==FN_Min ? OP_Min : OP_Max;
854 for(i=0; i<pList->nExpr; i++){
855 sqliteExprCode(pParse, pList->a[i].pExpr);
856 if( i>0 ){
drh99fcd712001-10-13 01:06:47 +0000857 sqliteVdbeAddOp(v, op, 0, 0);
drh6ec27332000-08-28 15:51:43 +0000858 }
859 }
860 break;
861 }
drhbf4133c2001-10-13 02:59:08 +0000862 case FN_Abs: {
863 sqliteExprCode(pParse, pList->a[0].pExpr);
864 sqliteVdbeAddOp(v, OP_AbsValue, 0, 0);
865 break;
866 }
867 case FN_Round: {
868 if( pList->nExpr==2 ){
869 sqliteExprCode(pParse, pList->a[1].pExpr);
870 }else{
871 sqliteVdbeAddOp(v, OP_Integer, 0, 0);
872 }
873 sqliteExprCode(pParse, pList->a[0].pExpr);
874 sqliteVdbeAddOp(v, OP_Precision, 0, 0);
875 break;
876 }
drh6ec27332000-08-28 15:51:43 +0000877 case FN_Length: {
878 sqliteExprCode(pParse, pList->a[0].pExpr);
drh99fcd712001-10-13 01:06:47 +0000879 sqliteVdbeAddOp(v, OP_Strlen, 0, 0);
drh6ec27332000-08-28 15:51:43 +0000880 break;
881 }
882 case FN_Substr: {
883 for(i=0; i<pList->nExpr; i++){
884 sqliteExprCode(pParse, pList->a[i].pExpr);
885 }
drh99fcd712001-10-13 01:06:47 +0000886 sqliteVdbeAddOp(v, OP_Substr, 0, 0);
drh6ec27332000-08-28 15:51:43 +0000887 break;
888 }
889 default: {
890 /* Can't happen! */
891 break;
drhcce7d172000-05-31 15:34:51 +0000892 }
893 }
894 break;
895 }
drh19a775c2000-06-05 18:54:46 +0000896 case TK_SELECT: {
drh99fcd712001-10-13 01:06:47 +0000897 sqliteVdbeAddOp(v, OP_MemLoad, pExpr->iColumn, 0);
drh19a775c2000-06-05 18:54:46 +0000898 break;
899 }
drhfef52082000-06-06 01:50:43 +0000900 case TK_IN: {
901 int addr;
drh99fcd712001-10-13 01:06:47 +0000902 sqliteVdbeAddOp(v, OP_Integer, 1, 0);
drhfef52082000-06-06 01:50:43 +0000903 sqliteExprCode(pParse, pExpr->pLeft);
904 addr = sqliteVdbeCurrentAddr(v);
905 if( pExpr->pSelect ){
drh99fcd712001-10-13 01:06:47 +0000906 sqliteVdbeAddOp(v, OP_Found, pExpr->iTable, addr+2);
drhfef52082000-06-06 01:50:43 +0000907 }else{
drh99fcd712001-10-13 01:06:47 +0000908 sqliteVdbeAddOp(v, OP_SetFound, pExpr->iTable, addr+2);
drhfef52082000-06-06 01:50:43 +0000909 }
drh99fcd712001-10-13 01:06:47 +0000910 sqliteVdbeAddOp(v, OP_AddImm, -1, 0);
drhfef52082000-06-06 01:50:43 +0000911 break;
912 }
913 case TK_BETWEEN: {
914 int lbl = sqliteVdbeMakeLabel(v);
drh99fcd712001-10-13 01:06:47 +0000915 sqliteVdbeAddOp(v, OP_Integer, 0, 0);
drhfef52082000-06-06 01:50:43 +0000916 sqliteExprIfFalse(pParse, pExpr, lbl);
drh99fcd712001-10-13 01:06:47 +0000917 sqliteVdbeAddOp(v, OP_AddImm, 1, 0);
drhfef52082000-06-06 01:50:43 +0000918 sqliteVdbeResolveLabel(v, lbl);
919 break;
920 }
drha2e00042002-01-22 03:13:42 +0000921 case TK_AS: {
922 sqliteExprCode(pParse, pExpr->pLeft);
923 break;
924 }
drhcce7d172000-05-31 15:34:51 +0000925 }
926 return;
927}
928
929/*
930** Generate code for a boolean expression such that a jump is made
931** to the label "dest" if the expression is true but execution
932** continues straight thru if the expression is false.
933*/
934void sqliteExprIfTrue(Parse *pParse, Expr *pExpr, int dest){
935 Vdbe *v = pParse->pVdbe;
936 int op = 0;
drhdaffd0e2001-04-11 14:28:42 +0000937 if( v==0 || pExpr==0 ) return;
drhcce7d172000-05-31 15:34:51 +0000938 switch( pExpr->op ){
939 case TK_LT: op = OP_Lt; break;
940 case TK_LE: op = OP_Le; break;
941 case TK_GT: op = OP_Gt; break;
942 case TK_GE: op = OP_Ge; break;
943 case TK_NE: op = OP_Ne; break;
944 case TK_EQ: op = OP_Eq; break;
945 case TK_LIKE: op = OP_Like; break;
946 case TK_GLOB: op = OP_Glob; break;
947 case TK_ISNULL: op = OP_IsNull; break;
948 case TK_NOTNULL: op = OP_NotNull; break;
949 default: break;
950 }
951 switch( pExpr->op ){
952 case TK_AND: {
953 int d2 = sqliteVdbeMakeLabel(v);
954 sqliteExprIfFalse(pParse, pExpr->pLeft, d2);
955 sqliteExprIfTrue(pParse, pExpr->pRight, dest);
956 sqliteVdbeResolveLabel(v, d2);
957 break;
958 }
959 case TK_OR: {
960 sqliteExprIfTrue(pParse, pExpr->pLeft, dest);
961 sqliteExprIfTrue(pParse, pExpr->pRight, dest);
962 break;
963 }
964 case TK_NOT: {
965 sqliteExprIfFalse(pParse, pExpr->pLeft, dest);
966 break;
967 }
968 case TK_LT:
969 case TK_LE:
970 case TK_GT:
971 case TK_GE:
972 case TK_NE:
973 case TK_EQ:
974 case TK_LIKE:
975 case TK_GLOB: {
976 sqliteExprCode(pParse, pExpr->pLeft);
977 sqliteExprCode(pParse, pExpr->pRight);
drh99fcd712001-10-13 01:06:47 +0000978 sqliteVdbeAddOp(v, op, 0, dest);
drhcce7d172000-05-31 15:34:51 +0000979 break;
980 }
981 case TK_ISNULL:
982 case TK_NOTNULL: {
983 sqliteExprCode(pParse, pExpr->pLeft);
drh99fcd712001-10-13 01:06:47 +0000984 sqliteVdbeAddOp(v, op, 0, dest);
drhcce7d172000-05-31 15:34:51 +0000985 break;
986 }
drhfef52082000-06-06 01:50:43 +0000987 case TK_IN: {
drhcfab11b2000-06-06 03:31:22 +0000988 sqliteExprCode(pParse, pExpr->pLeft);
drhfef52082000-06-06 01:50:43 +0000989 if( pExpr->pSelect ){
drh99fcd712001-10-13 01:06:47 +0000990 sqliteVdbeAddOp(v, OP_Found, pExpr->iTable, dest);
drhfef52082000-06-06 01:50:43 +0000991 }else{
drh99fcd712001-10-13 01:06:47 +0000992 sqliteVdbeAddOp(v, OP_SetFound, pExpr->iTable, dest);
drhfef52082000-06-06 01:50:43 +0000993 }
994 break;
995 }
996 case TK_BETWEEN: {
997 int lbl = sqliteVdbeMakeLabel(v);
998 sqliteExprCode(pParse, pExpr->pLeft);
drh99fcd712001-10-13 01:06:47 +0000999 sqliteVdbeAddOp(v, OP_Dup, 0, 0);
drhfef52082000-06-06 01:50:43 +00001000 sqliteExprCode(pParse, pExpr->pList->a[0].pExpr);
drh99fcd712001-10-13 01:06:47 +00001001 sqliteVdbeAddOp(v, OP_Lt, 0, lbl);
drhfef52082000-06-06 01:50:43 +00001002 sqliteExprCode(pParse, pExpr->pList->a[1].pExpr);
drh99fcd712001-10-13 01:06:47 +00001003 sqliteVdbeAddOp(v, OP_Le, 0, dest);
1004 sqliteVdbeAddOp(v, OP_Integer, 0, 0);
1005 sqliteVdbeResolveLabel(v, lbl);
1006 sqliteVdbeAddOp(v, OP_Pop, 1, 0);
drhfef52082000-06-06 01:50:43 +00001007 break;
1008 }
drhcce7d172000-05-31 15:34:51 +00001009 default: {
1010 sqliteExprCode(pParse, pExpr);
drh99fcd712001-10-13 01:06:47 +00001011 sqliteVdbeAddOp(v, OP_If, 0, dest);
drhcce7d172000-05-31 15:34:51 +00001012 break;
1013 }
1014 }
1015}
1016
1017/*
drh66b89c82000-11-28 20:47:17 +00001018** Generate code for a boolean expression such that a jump is made
drhcce7d172000-05-31 15:34:51 +00001019** to the label "dest" if the expression is false but execution
1020** continues straight thru if the expression is true.
1021*/
1022void sqliteExprIfFalse(Parse *pParse, Expr *pExpr, int dest){
1023 Vdbe *v = pParse->pVdbe;
1024 int op = 0;
drhdaffd0e2001-04-11 14:28:42 +00001025 if( v==0 || pExpr==0 ) return;
drhcce7d172000-05-31 15:34:51 +00001026 switch( pExpr->op ){
1027 case TK_LT: op = OP_Ge; break;
1028 case TK_LE: op = OP_Gt; break;
1029 case TK_GT: op = OP_Le; break;
1030 case TK_GE: op = OP_Lt; break;
1031 case TK_NE: op = OP_Eq; break;
1032 case TK_EQ: op = OP_Ne; break;
1033 case TK_LIKE: op = OP_Like; break;
1034 case TK_GLOB: op = OP_Glob; break;
1035 case TK_ISNULL: op = OP_NotNull; break;
1036 case TK_NOTNULL: op = OP_IsNull; break;
1037 default: break;
1038 }
1039 switch( pExpr->op ){
1040 case TK_AND: {
1041 sqliteExprIfFalse(pParse, pExpr->pLeft, dest);
1042 sqliteExprIfFalse(pParse, pExpr->pRight, dest);
1043 break;
1044 }
1045 case TK_OR: {
1046 int d2 = sqliteVdbeMakeLabel(v);
1047 sqliteExprIfTrue(pParse, pExpr->pLeft, d2);
1048 sqliteExprIfFalse(pParse, pExpr->pRight, dest);
1049 sqliteVdbeResolveLabel(v, d2);
1050 break;
1051 }
1052 case TK_NOT: {
1053 sqliteExprIfTrue(pParse, pExpr->pLeft, dest);
1054 break;
1055 }
1056 case TK_LT:
1057 case TK_LE:
1058 case TK_GT:
1059 case TK_GE:
1060 case TK_NE:
1061 case TK_EQ: {
1062 sqliteExprCode(pParse, pExpr->pLeft);
1063 sqliteExprCode(pParse, pExpr->pRight);
drh99fcd712001-10-13 01:06:47 +00001064 sqliteVdbeAddOp(v, op, 0, dest);
drhcce7d172000-05-31 15:34:51 +00001065 break;
1066 }
1067 case TK_LIKE:
1068 case TK_GLOB: {
1069 sqliteExprCode(pParse, pExpr->pLeft);
1070 sqliteExprCode(pParse, pExpr->pRight);
drh99fcd712001-10-13 01:06:47 +00001071 sqliteVdbeAddOp(v, op, 1, dest);
drhcce7d172000-05-31 15:34:51 +00001072 break;
1073 }
1074 case TK_ISNULL:
1075 case TK_NOTNULL: {
1076 sqliteExprCode(pParse, pExpr->pLeft);
drh99fcd712001-10-13 01:06:47 +00001077 sqliteVdbeAddOp(v, op, 0, dest);
drhcce7d172000-05-31 15:34:51 +00001078 break;
1079 }
drhfef52082000-06-06 01:50:43 +00001080 case TK_IN: {
drhcfab11b2000-06-06 03:31:22 +00001081 sqliteExprCode(pParse, pExpr->pLeft);
drhfef52082000-06-06 01:50:43 +00001082 if( pExpr->pSelect ){
drh99fcd712001-10-13 01:06:47 +00001083 sqliteVdbeAddOp(v, OP_NotFound, pExpr->iTable, dest);
drhfef52082000-06-06 01:50:43 +00001084 }else{
drh99fcd712001-10-13 01:06:47 +00001085 sqliteVdbeAddOp(v, OP_SetNotFound, pExpr->iTable, dest);
drhfef52082000-06-06 01:50:43 +00001086 }
1087 break;
1088 }
1089 case TK_BETWEEN: {
1090 int addr;
1091 sqliteExprCode(pParse, pExpr->pLeft);
drh99fcd712001-10-13 01:06:47 +00001092 sqliteVdbeAddOp(v, OP_Dup, 0, 0);
drhfef52082000-06-06 01:50:43 +00001093 sqliteExprCode(pParse, pExpr->pList->a[0].pExpr);
1094 addr = sqliteVdbeCurrentAddr(v);
drh99fcd712001-10-13 01:06:47 +00001095 sqliteVdbeAddOp(v, OP_Ge, 0, addr+3);
1096 sqliteVdbeAddOp(v, OP_Pop, 1, 0);
1097 sqliteVdbeAddOp(v, OP_Goto, 0, dest);
drhfef52082000-06-06 01:50:43 +00001098 sqliteExprCode(pParse, pExpr->pList->a[1].pExpr);
drh99fcd712001-10-13 01:06:47 +00001099 sqliteVdbeAddOp(v, OP_Gt, 0, dest);
drhfef52082000-06-06 01:50:43 +00001100 break;
1101 }
drhcce7d172000-05-31 15:34:51 +00001102 default: {
1103 sqliteExprCode(pParse, pExpr);
drh99fcd712001-10-13 01:06:47 +00001104 sqliteVdbeAddOp(v, OP_Not, 0, 0);
1105 sqliteVdbeAddOp(v, OP_If, 0, dest);
drhcce7d172000-05-31 15:34:51 +00001106 break;
1107 }
1108 }
1109}
drh22827922000-06-06 17:27:05 +00001110
1111/*
1112** Do a deep comparison of two expression trees. Return TRUE (non-zero)
1113** if they are identical and return FALSE if they differ in any way.
1114*/
drhd8bc7082000-06-07 23:51:50 +00001115int sqliteExprCompare(Expr *pA, Expr *pB){
drh22827922000-06-06 17:27:05 +00001116 int i;
1117 if( pA==0 ){
1118 return pB==0;
1119 }else if( pB==0 ){
1120 return 0;
1121 }
1122 if( pA->op!=pB->op ) return 0;
drhd8bc7082000-06-07 23:51:50 +00001123 if( !sqliteExprCompare(pA->pLeft, pB->pLeft) ) return 0;
1124 if( !sqliteExprCompare(pA->pRight, pB->pRight) ) return 0;
drh22827922000-06-06 17:27:05 +00001125 if( pA->pList ){
1126 if( pB->pList==0 ) return 0;
1127 if( pA->pList->nExpr!=pB->pList->nExpr ) return 0;
1128 for(i=0; i<pA->pList->nExpr; i++){
drhd8bc7082000-06-07 23:51:50 +00001129 if( !sqliteExprCompare(pA->pList->a[i].pExpr, pB->pList->a[i].pExpr) ){
drh22827922000-06-06 17:27:05 +00001130 return 0;
1131 }
1132 }
1133 }else if( pB->pList ){
1134 return 0;
1135 }
1136 if( pA->pSelect || pB->pSelect ) return 0;
1137 if( pA->token.z ){
1138 if( pB->token.z==0 ) return 0;
1139 if( pB->token.n!=pA->token.n ) return 0;
1140 if( sqliteStrNICmp(pA->token.z, pB->token.z, pA->token.n)!=0 ) return 0;
1141 }
1142 return 1;
1143}
1144
1145/*
1146** Add a new element to the pParse->aAgg[] array and return its index.
1147*/
1148static int appendAggInfo(Parse *pParse){
1149 if( (pParse->nAgg & 0x7)==0 ){
1150 int amt = pParse->nAgg + 8;
drh6d4abfb2001-10-22 02:58:08 +00001151 AggExpr *aAgg = sqliteRealloc(pParse->aAgg, amt*sizeof(pParse->aAgg[0]));
1152 if( aAgg==0 ){
drh22827922000-06-06 17:27:05 +00001153 return -1;
1154 }
drh6d4abfb2001-10-22 02:58:08 +00001155 pParse->aAgg = aAgg;
drh22827922000-06-06 17:27:05 +00001156 }
1157 memset(&pParse->aAgg[pParse->nAgg], 0, sizeof(pParse->aAgg[0]));
1158 return pParse->nAgg++;
1159}
1160
1161/*
1162** Analyze the given expression looking for aggregate functions and
1163** for variables that need to be added to the pParse->aAgg[] array.
1164** Make additional entries to the pParse->aAgg[] array as necessary.
1165**
1166** This routine should only be called after the expression has been
1167** analyzed by sqliteExprResolveIds() and sqliteExprCheck().
1168**
1169** If errors are seen, leave an error message in zErrMsg and return
1170** the number of errors.
1171*/
1172int sqliteExprAnalyzeAggregates(Parse *pParse, Expr *pExpr){
1173 int i;
1174 AggExpr *aAgg;
1175 int nErr = 0;
1176
1177 if( pExpr==0 ) return 0;
1178 switch( pExpr->op ){
drh967e8b72000-06-21 13:59:10 +00001179 case TK_COLUMN: {
drh22827922000-06-06 17:27:05 +00001180 aAgg = pParse->aAgg;
1181 for(i=0; i<pParse->nAgg; i++){
1182 if( aAgg[i].isAgg ) continue;
1183 if( aAgg[i].pExpr->iTable==pExpr->iTable
drh967e8b72000-06-21 13:59:10 +00001184 && aAgg[i].pExpr->iColumn==pExpr->iColumn ){
drh22827922000-06-06 17:27:05 +00001185 break;
1186 }
1187 }
1188 if( i>=pParse->nAgg ){
1189 i = appendAggInfo(pParse);
1190 if( i<0 ) return 1;
1191 pParse->aAgg[i].isAgg = 0;
1192 pParse->aAgg[i].pExpr = pExpr;
1193 }
drhaaf88722000-06-08 11:25:00 +00001194 pExpr->iAgg = i;
drh22827922000-06-06 17:27:05 +00001195 break;
1196 }
1197 case TK_AGG_FUNCTION: {
drh967e8b72000-06-21 13:59:10 +00001198 if( pExpr->iColumn==FN_Count || pExpr->iColumn==FN_Avg ){
drh22827922000-06-06 17:27:05 +00001199 if( pParse->iAggCount>=0 ){
1200 i = pParse->iAggCount;
1201 }else{
1202 i = appendAggInfo(pParse);
1203 if( i<0 ) return 1;
1204 pParse->aAgg[i].isAgg = 1;
1205 pParse->aAgg[i].pExpr = 0;
1206 pParse->iAggCount = i;
1207 }
drh967e8b72000-06-21 13:59:10 +00001208 if( pExpr->iColumn==FN_Count ){
drh22827922000-06-06 17:27:05 +00001209 pExpr->iAgg = i;
1210 break;
1211 }
1212 }
1213 aAgg = pParse->aAgg;
1214 for(i=0; i<pParse->nAgg; i++){
1215 if( !aAgg[i].isAgg ) continue;
drhd8bc7082000-06-07 23:51:50 +00001216 if( sqliteExprCompare(aAgg[i].pExpr, pExpr) ){
drh22827922000-06-06 17:27:05 +00001217 break;
1218 }
1219 }
1220 if( i>=pParse->nAgg ){
1221 i = appendAggInfo(pParse);
1222 if( i<0 ) return 1;
1223 pParse->aAgg[i].isAgg = 1;
1224 pParse->aAgg[i].pExpr = pExpr;
1225 }
1226 pExpr->iAgg = i;
1227 break;
1228 }
1229 default: {
1230 if( pExpr->pLeft ){
1231 nErr = sqliteExprAnalyzeAggregates(pParse, pExpr->pLeft);
1232 }
1233 if( nErr==0 && pExpr->pRight ){
1234 nErr = sqliteExprAnalyzeAggregates(pParse, pExpr->pRight);
1235 }
1236 if( nErr==0 && pExpr->pList ){
1237 int n = pExpr->pList->nExpr;
1238 int i;
1239 for(i=0; nErr==0 && i<n; i++){
1240 nErr = sqliteExprAnalyzeAggregates(pParse, pExpr->pList->a[i].pExpr);
1241 }
1242 }
1243 break;
1244 }
1245 }
1246 return nErr;
1247}