blob: 0bf74727a1264160a55de61939d60a1a03b40783 [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**
drh382c0242001-10-06 16:33:02 +000015** $Id: expr.c,v 1.29 2001/10/06 16:33:03 drh Exp $
drhcce7d172000-05-31 15:34:51 +000016*/
17#include "sqliteInt.h"
18
19/*
drhfef52082000-06-06 01:50:43 +000020** Walk an expression tree. Return 1 if the expression is constant
21** and 0 if it involves variables.
22*/
23static int isConstant(Expr *p){
24 switch( p->op ){
25 case TK_ID:
drh967e8b72000-06-21 13:59:10 +000026 case TK_COLUMN:
drhfef52082000-06-06 01:50:43 +000027 case TK_DOT:
28 return 0;
29 default: {
30 if( p->pLeft && !isConstant(p->pLeft) ) return 0;
31 if( p->pRight && !isConstant(p->pRight) ) return 0;
32 if( p->pList ){
33 int i;
34 for(i=0; i<p->pList->nExpr; i++){
35 if( !isConstant(p->pList->a[i].pExpr) ) return 0;
36 }
37 }
38 break;
39 }
40 }
41 return 1;
42}
43
44/*
drh4794b982000-06-06 13:54:14 +000045** Walk the expression tree and process operators of the form:
46**
47** expr IN (SELECT ...)
48**
drh967e8b72000-06-21 13:59:10 +000049** These operators have to be processed before column names are
drh4794b982000-06-06 13:54:14 +000050** resolved because each such operator increments pParse->nTab
drh1ccde152000-06-17 13:12:39 +000051** to reserve cursor numbers for its own use. But pParse->nTab
drh967e8b72000-06-21 13:59:10 +000052** needs to be constant once we begin resolving column names.
drh4794b982000-06-06 13:54:14 +000053**
54** Actually, the processing of IN-SELECT is only started by this
55** routine. This routine allocates a cursor number to the IN-SELECT
56** and then moves on. The code generation is done by
57** sqliteExprResolveIds() which must be called afterwards.
58*/
59void sqliteExprResolveInSelect(Parse *pParse, Expr *pExpr){
60 if( pExpr==0 ) return;
61 if( pExpr->op==TK_IN && pExpr->pSelect!=0 ){
62 pExpr->iTable = pParse->nTab++;
63 }else{
64 if( pExpr->pLeft ) sqliteExprResolveInSelect(pParse, pExpr->pLeft);
65 if( pExpr->pRight ) sqliteExprResolveInSelect(pParse, pExpr->pRight);
66 if( pExpr->pList ){
67 int i;
68 ExprList *pList = pExpr->pList;
69 for(i=0; i<pList->nExpr; i++){
70 sqliteExprResolveInSelect(pParse, pList->a[i].pExpr);
71 }
72 }
73 }
74}
75
76/*
drhc4a3c772001-04-04 11:48:57 +000077** Return TRUE if the given string is a row-id column name.
78*/
79static int sqliteIsRowid(const char *z){
80 if( sqliteStrICmp(z, "_ROWID_")==0 ) return 1;
81 if( sqliteStrICmp(z, "ROWID")==0 ) return 1;
82 if( sqliteStrICmp(z, "OID")==0 ) return 1;
83 return 0;
84}
85
86/*
drhcce7d172000-05-31 15:34:51 +000087** This routine walks an expression tree and resolves references to
drh967e8b72000-06-21 13:59:10 +000088** table columns. Nodes of the form ID.ID or ID resolve into an
89** index to the table in the table list and a column offset. The opcode
90** for such nodes is changed to TK_COLUMN. The iTable value is changed
drh19a775c2000-06-05 18:54:46 +000091** to the index of the referenced table in pTabList plus the pParse->nTab
drh967e8b72000-06-21 13:59:10 +000092** value. The iColumn value is changed to the index of the column of the
drh19a775c2000-06-05 18:54:46 +000093** referenced table.
94**
drhfef52082000-06-06 01:50:43 +000095** We also check for instances of the IN operator. IN comes in two
96** forms:
97**
98** expr IN (exprlist)
99** and
100** expr IN (SELECT ...)
101**
102** The first form is handled by creating a set holding the list
103** of allowed values. The second form causes the SELECT to generate
104** a temporary table.
105**
106** This routine also looks for scalar SELECTs that are part of an expression.
drh19a775c2000-06-05 18:54:46 +0000107** If it finds any, it generates code to write the value of that select
108** into a memory cell.
drhcce7d172000-05-31 15:34:51 +0000109**
drh967e8b72000-06-21 13:59:10 +0000110** Unknown columns or tables provoke an error. The function returns
drhcce7d172000-05-31 15:34:51 +0000111** the number of errors seen and leaves an error message on pParse->zErrMsg.
112*/
113int sqliteExprResolveIds(Parse *pParse, IdList *pTabList, Expr *pExpr){
drhdaffd0e2001-04-11 14:28:42 +0000114 if( pExpr==0 || pTabList==0 ) return 0;
drhcce7d172000-05-31 15:34:51 +0000115 switch( pExpr->op ){
116 /* A lone identifier */
117 case TK_ID: {
drhc4a3c772001-04-04 11:48:57 +0000118 int cnt = 0; /* Number of matches */
119 int i; /* Loop counter */
drh6e142f52000-06-08 13:36:40 +0000120 char *z = sqliteStrNDup(pExpr->token.z, pExpr->token.n);
drhdaffd0e2001-04-11 14:28:42 +0000121 if( z==0 ) return 1;
drhcce7d172000-05-31 15:34:51 +0000122 for(i=0; i<pTabList->nId; i++){
123 int j;
124 Table *pTab = pTabList->a[i].pTab;
125 if( pTab==0 ) continue;
126 for(j=0; j<pTab->nCol; j++){
drh7020f652000-06-03 18:06:52 +0000127 if( sqliteStrICmp(pTab->aCol[j].zName, z)==0 ){
drhcce7d172000-05-31 15:34:51 +0000128 cnt++;
drh19a775c2000-06-05 18:54:46 +0000129 pExpr->iTable = i + pParse->nTab;
drh967e8b72000-06-21 13:59:10 +0000130 pExpr->iColumn = j;
drhcce7d172000-05-31 15:34:51 +0000131 }
132 }
133 }
drhc4a3c772001-04-04 11:48:57 +0000134 if( cnt==0 && sqliteIsRowid(z) ){
135 pExpr->iColumn = -1;
136 pExpr->iTable = pParse->nTab;
137 cnt = 1 + (pTabList->nId>1);
138 }
drhcce7d172000-05-31 15:34:51 +0000139 sqliteFree(z);
140 if( cnt==0 ){
drh967e8b72000-06-21 13:59:10 +0000141 sqliteSetNString(&pParse->zErrMsg, "no such column: ", -1,
drhcce7d172000-05-31 15:34:51 +0000142 pExpr->token.z, pExpr->token.n, 0);
143 pParse->nErr++;
144 return 1;
145 }else if( cnt>1 ){
drh967e8b72000-06-21 13:59:10 +0000146 sqliteSetNString(&pParse->zErrMsg, "ambiguous column name: ", -1,
drhcce7d172000-05-31 15:34:51 +0000147 pExpr->token.z, pExpr->token.n, 0);
148 pParse->nErr++;
149 return 1;
150 }
drh967e8b72000-06-21 13:59:10 +0000151 pExpr->op = TK_COLUMN;
drhcce7d172000-05-31 15:34:51 +0000152 break;
153 }
154
drh967e8b72000-06-21 13:59:10 +0000155 /* A table name and column name: ID.ID */
drhcce7d172000-05-31 15:34:51 +0000156 case TK_DOT: {
drhbed86902000-06-02 13:27:59 +0000157 int cnt = 0; /* Number of matches */
drhc4a3c772001-04-04 11:48:57 +0000158 int cntTab = 0; /* Number of matching tables */
drhbed86902000-06-02 13:27:59 +0000159 int i; /* Loop counter */
drhcce7d172000-05-31 15:34:51 +0000160 Expr *pLeft, *pRight; /* Left and right subbranches of the expr */
drhcce7d172000-05-31 15:34:51 +0000161 char *zLeft, *zRight; /* Text of an identifier */
162
163 pLeft = pExpr->pLeft;
164 pRight = pExpr->pRight;
165 assert( pLeft && pLeft->op==TK_ID );
166 assert( pRight && pRight->op==TK_ID );
drh6e142f52000-06-08 13:36:40 +0000167 zLeft = sqliteStrNDup(pLeft->token.z, pLeft->token.n);
168 zRight = sqliteStrNDup(pRight->token.z, pRight->token.n);
drhdaffd0e2001-04-11 14:28:42 +0000169 if( zLeft==0 || zRight==0 ){
170 sqliteFree(zLeft);
171 sqliteFree(zRight);
172 return 1;
173 }
drh87c40e82001-07-23 14:33:02 +0000174 sqliteDequote(zLeft);
175 sqliteDequote(zRight);
drhc4a3c772001-04-04 11:48:57 +0000176 pExpr->iTable = -1;
drhcce7d172000-05-31 15:34:51 +0000177 for(i=0; i<pTabList->nId; i++){
178 int j;
179 char *zTab;
180 Table *pTab = pTabList->a[i].pTab;
181 if( pTab==0 ) continue;
182 if( pTabList->a[i].zAlias ){
183 zTab = pTabList->a[i].zAlias;
184 }else{
185 zTab = pTab->zName;
186 }
187 if( sqliteStrICmp(zTab, zLeft)!=0 ) continue;
drhc4a3c772001-04-04 11:48:57 +0000188 if( 0==(cntTab++) ) pExpr->iTable = i + pParse->nTab;
drhcce7d172000-05-31 15:34:51 +0000189 for(j=0; j<pTab->nCol; j++){
drh7020f652000-06-03 18:06:52 +0000190 if( sqliteStrICmp(pTab->aCol[j].zName, zRight)==0 ){
drhcce7d172000-05-31 15:34:51 +0000191 cnt++;
drh19a775c2000-06-05 18:54:46 +0000192 pExpr->iTable = i + pParse->nTab;
drh967e8b72000-06-21 13:59:10 +0000193 pExpr->iColumn = j;
drhcce7d172000-05-31 15:34:51 +0000194 }
195 }
196 }
drhc4a3c772001-04-04 11:48:57 +0000197 if( cnt==0 && cntTab==1 && sqliteIsRowid(zRight) ){
198 cnt = 1;
199 pExpr->iColumn = -1;
200 }
drhcce7d172000-05-31 15:34:51 +0000201 sqliteFree(zLeft);
202 sqliteFree(zRight);
203 if( cnt==0 ){
drh967e8b72000-06-21 13:59:10 +0000204 sqliteSetNString(&pParse->zErrMsg, "no such column: ", -1,
drhcce7d172000-05-31 15:34:51 +0000205 pLeft->token.z, pLeft->token.n, ".", 1,
206 pRight->token.z, pRight->token.n, 0);
207 pParse->nErr++;
208 return 1;
209 }else if( cnt>1 ){
drh967e8b72000-06-21 13:59:10 +0000210 sqliteSetNString(&pParse->zErrMsg, "ambiguous column name: ", -1,
drhcce7d172000-05-31 15:34:51 +0000211 pLeft->token.z, pLeft->token.n, ".", 1,
212 pRight->token.z, pRight->token.n, 0);
213 pParse->nErr++;
214 return 1;
215 }
216 sqliteExprDelete(pLeft);
217 pExpr->pLeft = 0;
218 sqliteExprDelete(pRight);
219 pExpr->pRight = 0;
drh967e8b72000-06-21 13:59:10 +0000220 pExpr->op = TK_COLUMN;
drhcce7d172000-05-31 15:34:51 +0000221 break;
222 }
223
drhfef52082000-06-06 01:50:43 +0000224 case TK_IN: {
drhd8bc7082000-06-07 23:51:50 +0000225 Vdbe *v = sqliteGetVdbe(pParse);
drhfef52082000-06-06 01:50:43 +0000226 if( v==0 ) return 1;
drhcfab11b2000-06-06 03:31:22 +0000227 if( sqliteExprResolveIds(pParse, pTabList, pExpr->pLeft) ){
228 return 1;
229 }
drhfef52082000-06-06 01:50:43 +0000230 if( pExpr->pSelect ){
231 /* Case 1: expr IN (SELECT ...)
232 **
233 ** Generate code to write the results of the select into a temporary
drh4794b982000-06-06 13:54:14 +0000234 ** table. The cursor number of the temporary table has already
235 ** been put in iTable by sqliteExprResolveInSelect().
drhfef52082000-06-06 01:50:43 +0000236 */
drhbe0072d2001-09-13 14:46:09 +0000237 sqliteVdbeAddOp(v, OP_OpenTemp, pExpr->iTable, 0, 0, 0);
drhfef52082000-06-06 01:50:43 +0000238 if( sqliteSelect(pParse, pExpr->pSelect, SRT_Set, pExpr->iTable) );
239 }else if( pExpr->pList ){
240 /* Case 2: expr IN (exprlist)
241 **
242 ** Create a set to put the exprlist values in. The Set id is stored
243 ** in iTable.
244 */
245 int i, iSet;
246 for(i=0; i<pExpr->pList->nExpr; i++){
247 Expr *pE2 = pExpr->pList->a[i].pExpr;
drhfef52082000-06-06 01:50:43 +0000248 if( !isConstant(pE2) ){
249 sqliteSetString(&pParse->zErrMsg,
250 "right-hand side of IN operator must be constant", 0);
251 pParse->nErr++;
252 return 1;
253 }
drh4794b982000-06-06 13:54:14 +0000254 if( sqliteExprCheck(pParse, pE2, 0, 0) ){
255 return 1;
256 }
drhfef52082000-06-06 01:50:43 +0000257 }
258 iSet = pExpr->iTable = pParse->nSet++;
259 for(i=0; i<pExpr->pList->nExpr; i++){
260 Expr *pE2 = pExpr->pList->a[i].pExpr;
261 switch( pE2->op ){
262 case TK_FLOAT:
263 case TK_INTEGER:
264 case TK_STRING: {
265 int addr = sqliteVdbeAddOp(v, OP_SetInsert, iSet, 0, 0, 0);
266 sqliteVdbeChangeP3(v, addr, pE2->token.z, pE2->token.n);
267 sqliteVdbeDequoteP3(v, addr);
268 break;
269 }
270 default: {
271 sqliteExprCode(pParse, pE2);
272 sqliteVdbeAddOp(v, OP_SetInsert, iSet, 0, 0, 0);
273 break;
274 }
275 }
276 }
277 }
drhcfab11b2000-06-06 03:31:22 +0000278 break;
drhfef52082000-06-06 01:50:43 +0000279 }
280
drh19a775c2000-06-05 18:54:46 +0000281 case TK_SELECT: {
drhfef52082000-06-06 01:50:43 +0000282 /* This has to be a scalar SELECT. Generate code to put the
283 ** value of this select in a memory cell and record the number
drh967e8b72000-06-21 13:59:10 +0000284 ** of the memory cell in iColumn.
drhfef52082000-06-06 01:50:43 +0000285 */
drh967e8b72000-06-21 13:59:10 +0000286 pExpr->iColumn = pParse->nMem++;
287 if( sqliteSelect(pParse, pExpr->pSelect, SRT_Mem, pExpr->iColumn) ){
drh19a775c2000-06-05 18:54:46 +0000288 return 1;
289 }
290 break;
291 }
292
drhcce7d172000-05-31 15:34:51 +0000293 /* For all else, just recursively walk the tree */
294 default: {
drh4794b982000-06-06 13:54:14 +0000295 if( pExpr->pLeft
296 && sqliteExprResolveIds(pParse, pTabList, pExpr->pLeft) ){
drhcce7d172000-05-31 15:34:51 +0000297 return 1;
298 }
299 if( pExpr->pRight
drh4794b982000-06-06 13:54:14 +0000300 && sqliteExprResolveIds(pParse, pTabList, pExpr->pRight) ){
drhcce7d172000-05-31 15:34:51 +0000301 return 1;
302 }
303 if( pExpr->pList ){
304 int i;
305 ExprList *pList = pExpr->pList;
306 for(i=0; i<pList->nExpr; i++){
307 if( sqliteExprResolveIds(pParse, pTabList, pList->a[i].pExpr) ){
308 return 1;
309 }
310 }
311 }
312 }
313 }
314 return 0;
315}
316
317#if 0 /* NOT USED */
318/*
319** Compare a token against a string. Return TRUE if they match.
320*/
321static int sqliteTokenCmp(Token *pToken, const char *zStr){
322 int n = strlen(zStr);
323 if( n!=pToken->n ) return 0;
324 return sqliteStrNICmp(pToken->z, zStr, n)==0;
325}
326#endif
327
328/*
329** Convert a function name into its integer identifier. Return the
330** identifier. Return FN_Unknown if the function name is unknown.
331*/
332int sqliteFuncId(Token *pToken){
333 static const struct {
334 char *zName;
335 int len;
336 int id;
337 } aFunc[] = {
338 { "count", 5, FN_Count },
339 { "min", 3, FN_Min },
340 { "max", 3, FN_Max },
341 { "sum", 3, FN_Sum },
drh22827922000-06-06 17:27:05 +0000342 { "avg", 3, FN_Avg },
drh0bdaf622000-06-11 23:50:13 +0000343 { "fcnt", 4, FN_Fcnt }, /* Used for testing only */
drh6ec27332000-08-28 15:51:43 +0000344 { "length", 6, FN_Length},
345 { "substr", 6, FN_Substr},
drhcce7d172000-05-31 15:34:51 +0000346 };
347 int i;
348 for(i=0; i<ArraySize(aFunc); i++){
349 if( aFunc[i].len==pToken->n
350 && sqliteStrNICmp(pToken->z, aFunc[i].zName, aFunc[i].len)==0 ){
351 return aFunc[i].id;
352 }
353 }
354 return FN_Unknown;
355}
356
357/*
358** Error check the functions in an expression. Make sure all
359** function names are recognized and all functions have the correct
360** number of arguments. Leave an error message in pParse->zErrMsg
361** if anything is amiss. Return the number of errors.
362**
363** if pIsAgg is not null and this expression is an aggregate function
364** (like count(*) or max(value)) then write a 1 into *pIsAgg.
365*/
366int sqliteExprCheck(Parse *pParse, Expr *pExpr, int allowAgg, int *pIsAgg){
367 int nErr = 0;
368 if( pExpr==0 ) return 0;
drhcce7d172000-05-31 15:34:51 +0000369 switch( pExpr->op ){
370 case TK_FUNCTION: {
371 int id = sqliteFuncId(&pExpr->token);
372 int n = pExpr->pList ? pExpr->pList->nExpr : 0;
373 int no_such_func = 0;
374 int too_many_args = 0;
375 int too_few_args = 0;
376 int is_agg = 0;
377 int i;
drh967e8b72000-06-21 13:59:10 +0000378 pExpr->iColumn = id;
drhcce7d172000-05-31 15:34:51 +0000379 switch( id ){
380 case FN_Unknown: {
381 no_such_func = 1;
382 break;
383 }
384 case FN_Count: {
385 no_such_func = !allowAgg;
386 too_many_args = n>1;
387 is_agg = 1;
388 break;
389 }
390 case FN_Max:
391 case FN_Min: {
392 too_few_args = allowAgg ? n<1 : n<2;
393 is_agg = n==1;
394 break;
395 }
drh22827922000-06-06 17:27:05 +0000396 case FN_Avg:
drhcce7d172000-05-31 15:34:51 +0000397 case FN_Sum: {
398 no_such_func = !allowAgg;
399 too_many_args = n>1;
400 too_few_args = n<1;
401 is_agg = 1;
402 break;
403 }
drh6ec27332000-08-28 15:51:43 +0000404 case FN_Length: {
405 too_few_args = n<1;
406 too_many_args = n>1;
407 break;
408 }
409 case FN_Substr: {
410 too_few_args = n<3;
411 too_many_args = n>3;
412 break;
413 }
drh0bdaf622000-06-11 23:50:13 +0000414 /* The "fcnt(*)" function always returns the number of fetch
415 ** operations that have occurred so far while processing the
416 ** SQL statement. This information can be used by test procedures
417 ** to verify that indices are being used properly to minimize
418 ** searching. All arguments to fcnt() are ignored. fcnt() has
419 ** no use (other than testing) that we are aware of.
420 */
421 case FN_Fcnt: {
422 n = 0;
423 break;
424 }
drh6ec27332000-08-28 15:51:43 +0000425
drhcce7d172000-05-31 15:34:51 +0000426 default: break;
427 }
428 if( no_such_func ){
429 sqliteSetNString(&pParse->zErrMsg, "no such function: ", -1,
430 pExpr->token.z, pExpr->token.n, 0);
431 pParse->nErr++;
432 nErr++;
433 }else if( too_many_args ){
434 sqliteSetNString(&pParse->zErrMsg, "too many arguments to function ",-1,
435 pExpr->token.z, pExpr->token.n, "()", 2, 0);
436 pParse->nErr++;
437 nErr++;
438 }else if( too_few_args ){
439 sqliteSetNString(&pParse->zErrMsg, "too few arguments to function ",-1,
440 pExpr->token.z, pExpr->token.n, "()", 2, 0);
441 pParse->nErr++;
442 nErr++;
443 }
drh22827922000-06-06 17:27:05 +0000444 if( is_agg ) pExpr->op = TK_AGG_FUNCTION;
drhcce7d172000-05-31 15:34:51 +0000445 if( is_agg && pIsAgg ) *pIsAgg = 1;
446 for(i=0; nErr==0 && i<n; i++){
drh4cfa7932000-06-08 15:10:46 +0000447 nErr = sqliteExprCheck(pParse, pExpr->pList->a[i].pExpr,
448 allowAgg && !is_agg, pIsAgg);
drhcce7d172000-05-31 15:34:51 +0000449 }
450 }
451 default: {
452 if( pExpr->pLeft ){
drh22827922000-06-06 17:27:05 +0000453 nErr = sqliteExprCheck(pParse, pExpr->pLeft, allowAgg, pIsAgg);
drhcce7d172000-05-31 15:34:51 +0000454 }
455 if( nErr==0 && pExpr->pRight ){
drh22827922000-06-06 17:27:05 +0000456 nErr = sqliteExprCheck(pParse, pExpr->pRight, allowAgg, pIsAgg);
drhcce7d172000-05-31 15:34:51 +0000457 }
drhfef52082000-06-06 01:50:43 +0000458 if( nErr==0 && pExpr->pList ){
459 int n = pExpr->pList->nExpr;
460 int i;
461 for(i=0; nErr==0 && i<n; i++){
drh22827922000-06-06 17:27:05 +0000462 Expr *pE2 = pExpr->pList->a[i].pExpr;
463 nErr = sqliteExprCheck(pParse, pE2, allowAgg, pIsAgg);
drhfef52082000-06-06 01:50:43 +0000464 }
465 }
drhcce7d172000-05-31 15:34:51 +0000466 break;
467 }
468 }
469 return nErr;
470}
471
472/*
473** Generate code into the current Vdbe to evaluate the given
drh1ccde152000-06-17 13:12:39 +0000474** expression and leave the result on the top of stack.
drhcce7d172000-05-31 15:34:51 +0000475*/
476void sqliteExprCode(Parse *pParse, Expr *pExpr){
477 Vdbe *v = pParse->pVdbe;
478 int op;
drhdaffd0e2001-04-11 14:28:42 +0000479 if( v==0 || pExpr==0 ) return;
drhcce7d172000-05-31 15:34:51 +0000480 switch( pExpr->op ){
481 case TK_PLUS: op = OP_Add; break;
482 case TK_MINUS: op = OP_Subtract; break;
483 case TK_STAR: op = OP_Multiply; break;
484 case TK_SLASH: op = OP_Divide; break;
485 case TK_AND: op = OP_And; break;
486 case TK_OR: op = OP_Or; break;
487 case TK_LT: op = OP_Lt; break;
488 case TK_LE: op = OP_Le; break;
489 case TK_GT: op = OP_Gt; break;
490 case TK_GE: op = OP_Ge; break;
491 case TK_NE: op = OP_Ne; break;
492 case TK_EQ: op = OP_Eq; break;
493 case TK_LIKE: op = OP_Like; break;
494 case TK_GLOB: op = OP_Glob; break;
495 case TK_ISNULL: op = OP_IsNull; break;
496 case TK_NOTNULL: op = OP_NotNull; break;
497 case TK_NOT: op = OP_Not; break;
498 case TK_UMINUS: op = OP_Negative; break;
499 default: break;
500 }
501 switch( pExpr->op ){
drh967e8b72000-06-21 13:59:10 +0000502 case TK_COLUMN: {
drh22827922000-06-06 17:27:05 +0000503 if( pParse->useAgg ){
504 sqliteVdbeAddOp(v, OP_AggGet, 0, pExpr->iAgg, 0, 0);
drhc4a3c772001-04-04 11:48:57 +0000505 }else if( pExpr->iColumn>=0 ){
drhbe0072d2001-09-13 14:46:09 +0000506 sqliteVdbeAddOp(v, OP_Column, pExpr->iTable, pExpr->iColumn, 0, 0);
drhc4a3c772001-04-04 11:48:57 +0000507 }else{
drh3fc190c2001-09-14 03:24:23 +0000508 sqliteVdbeAddOp(v, OP_Recno, pExpr->iTable, 0, 0, 0);
drh22827922000-06-06 17:27:05 +0000509 }
drhcce7d172000-05-31 15:34:51 +0000510 break;
511 }
512 case TK_INTEGER: {
513 int i = atoi(pExpr->token.z);
514 sqliteVdbeAddOp(v, OP_Integer, i, 0, 0, 0);
515 break;
516 }
517 case TK_FLOAT: {
518 int addr = sqliteVdbeAddOp(v, OP_String, 0, 0, 0, 0);
519 sqliteVdbeChangeP3(v, addr, pExpr->token.z, pExpr->token.n);
520 break;
521 }
522 case TK_STRING: {
523 int addr = sqliteVdbeAddOp(v, OP_String, 0, 0, 0, 0);
524 sqliteVdbeChangeP3(v, addr, pExpr->token.z, pExpr->token.n);
525 sqliteVdbeDequoteP3(v, addr);
526 break;
527 }
528 case TK_NULL: {
drh382c0242001-10-06 16:33:02 +0000529 sqliteVdbeAddOp(v, OP_String, 0, 0, 0, 0);
drhcce7d172000-05-31 15:34:51 +0000530 break;
531 }
532 case TK_AND:
533 case TK_OR:
534 case TK_PLUS:
535 case TK_STAR:
536 case TK_MINUS:
537 case TK_SLASH: {
538 sqliteExprCode(pParse, pExpr->pLeft);
539 sqliteExprCode(pParse, pExpr->pRight);
540 sqliteVdbeAddOp(v, op, 0, 0, 0, 0);
541 break;
542 }
drh00400772000-06-16 20:51:26 +0000543 case TK_CONCAT: {
544 sqliteExprCode(pParse, pExpr->pLeft);
545 sqliteExprCode(pParse, pExpr->pRight);
546 sqliteVdbeAddOp(v, OP_Concat, 2, 0, 0, 0);
547 break;
548 }
drhcce7d172000-05-31 15:34:51 +0000549 case TK_LT:
550 case TK_LE:
551 case TK_GT:
552 case TK_GE:
553 case TK_NE:
554 case TK_EQ:
555 case TK_LIKE:
556 case TK_GLOB: {
557 int dest;
558 sqliteVdbeAddOp(v, OP_Integer, 1, 0, 0, 0);
559 sqliteExprCode(pParse, pExpr->pLeft);
560 sqliteExprCode(pParse, pExpr->pRight);
561 dest = sqliteVdbeCurrentAddr(v) + 2;
562 sqliteVdbeAddOp(v, op, 0, dest, 0, 0);
563 sqliteVdbeAddOp(v, OP_AddImm, -1, 0, 0, 0);
564 break;
565 }
drhcce7d172000-05-31 15:34:51 +0000566 case TK_UMINUS: {
drh6e142f52000-06-08 13:36:40 +0000567 assert( pExpr->pLeft );
568 if( pExpr->pLeft->op==TK_INTEGER ){
569 int i = atoi(pExpr->pLeft->token.z);
570 sqliteVdbeAddOp(v, OP_Integer, -i, 0, 0, 0);
571 break;
572 }else if( pExpr->pLeft->op==TK_FLOAT ){
573 Token *p = &pExpr->pLeft->token;
574 char *z = sqliteMalloc( p->n + 2 );
575 sprintf(z, "-%.*s", p->n, p->z);
576 sqliteVdbeAddOp(v, OP_String, 0, 0, z, 0);
577 sqliteFree(z);
578 break;
579 }
drh1ccde152000-06-17 13:12:39 +0000580 /* Fall through into TK_NOT */
drh6e142f52000-06-08 13:36:40 +0000581 }
582 case TK_NOT: {
drhcce7d172000-05-31 15:34:51 +0000583 sqliteExprCode(pParse, pExpr->pLeft);
584 sqliteVdbeAddOp(v, op, 0, 0, 0, 0);
585 break;
586 }
587 case TK_ISNULL:
588 case TK_NOTNULL: {
589 int dest;
drh8be51132000-06-03 19:19:41 +0000590 sqliteVdbeAddOp(v, OP_Integer, 1, 0, 0, 0);
drhcce7d172000-05-31 15:34:51 +0000591 sqliteExprCode(pParse, pExpr->pLeft);
592 dest = sqliteVdbeCurrentAddr(v) + 2;
593 sqliteVdbeAddOp(v, op, 0, dest, 0, 0);
drh8be51132000-06-03 19:19:41 +0000594 sqliteVdbeAddOp(v, OP_AddImm, -1, 0, 0, 0);
drhcce7d172000-05-31 15:34:51 +0000595 break;
596 }
drh22827922000-06-06 17:27:05 +0000597 case TK_AGG_FUNCTION: {
598 sqliteVdbeAddOp(v, OP_AggGet, 0, pExpr->iAgg, 0, 0);
drh967e8b72000-06-21 13:59:10 +0000599 if( pExpr->iColumn==FN_Avg ){
drh22827922000-06-06 17:27:05 +0000600 assert( pParse->iAggCount>=0 && pParse->iAggCount<pParse->nAgg );
601 sqliteVdbeAddOp(v, OP_AggGet, 0, pParse->iAggCount, 0, 0);
602 sqliteVdbeAddOp(v, OP_Divide, 0, 0, 0, 0);
603 }
604 break;
605 }
drhcce7d172000-05-31 15:34:51 +0000606 case TK_FUNCTION: {
drh967e8b72000-06-21 13:59:10 +0000607 int id = pExpr->iColumn;
drhcce7d172000-05-31 15:34:51 +0000608 int op;
609 int i;
610 ExprList *pList = pExpr->pList;
drh6ec27332000-08-28 15:51:43 +0000611 switch( id ){
612 case FN_Fcnt: {
613 sqliteVdbeAddOp(v, OP_Fcnt, 0, 0, 0, 0);
614 break;
615 }
616 case FN_Min:
617 case FN_Max: {
618 op = id==FN_Min ? OP_Min : OP_Max;
619 for(i=0; i<pList->nExpr; i++){
620 sqliteExprCode(pParse, pList->a[i].pExpr);
621 if( i>0 ){
622 sqliteVdbeAddOp(v, op, 0, 0, 0, 0);
623 }
624 }
625 break;
626 }
627 case FN_Length: {
628 sqliteExprCode(pParse, pList->a[0].pExpr);
629 sqliteVdbeAddOp(v, OP_Strlen, 0, 0, 0, 0);
630 break;
631 }
632 case FN_Substr: {
633 for(i=0; i<pList->nExpr; i++){
634 sqliteExprCode(pParse, pList->a[i].pExpr);
635 }
636 sqliteVdbeAddOp(v, OP_Substr, 0, 0, 0, 0);
637 break;
638 }
639 default: {
640 /* Can't happen! */
641 break;
drhcce7d172000-05-31 15:34:51 +0000642 }
643 }
644 break;
645 }
drh19a775c2000-06-05 18:54:46 +0000646 case TK_SELECT: {
drh967e8b72000-06-21 13:59:10 +0000647 sqliteVdbeAddOp(v, OP_MemLoad, pExpr->iColumn, 0, 0, 0);
drh19a775c2000-06-05 18:54:46 +0000648 break;
649 }
drhfef52082000-06-06 01:50:43 +0000650 case TK_IN: {
651 int addr;
drh4794b982000-06-06 13:54:14 +0000652 sqliteVdbeAddOp(v, OP_Integer, 1, 0, 0, 0);
drhfef52082000-06-06 01:50:43 +0000653 sqliteExprCode(pParse, pExpr->pLeft);
654 addr = sqliteVdbeCurrentAddr(v);
655 if( pExpr->pSelect ){
656 sqliteVdbeAddOp(v, OP_Found, pExpr->iTable, addr+2, 0, 0);
657 }else{
658 sqliteVdbeAddOp(v, OP_SetFound, pExpr->iTable, addr+2, 0, 0);
659 }
drh4794b982000-06-06 13:54:14 +0000660 sqliteVdbeAddOp(v, OP_AddImm, -1, 0, 0, 0);
drhfef52082000-06-06 01:50:43 +0000661 break;
662 }
663 case TK_BETWEEN: {
664 int lbl = sqliteVdbeMakeLabel(v);
665 sqliteVdbeAddOp(v, OP_Integer, 0, 0, 0, 0);
666 sqliteExprIfFalse(pParse, pExpr, lbl);
667 sqliteVdbeAddOp(v, OP_AddImm, 1, 0, 0, 0);
668 sqliteVdbeResolveLabel(v, lbl);
669 break;
670 }
drhcce7d172000-05-31 15:34:51 +0000671 }
672 return;
673}
674
675/*
676** Generate code for a boolean expression such that a jump is made
677** to the label "dest" if the expression is true but execution
678** continues straight thru if the expression is false.
679*/
680void sqliteExprIfTrue(Parse *pParse, Expr *pExpr, int dest){
681 Vdbe *v = pParse->pVdbe;
682 int op = 0;
drhdaffd0e2001-04-11 14:28:42 +0000683 if( v==0 || pExpr==0 ) return;
drhcce7d172000-05-31 15:34:51 +0000684 switch( pExpr->op ){
685 case TK_LT: op = OP_Lt; break;
686 case TK_LE: op = OP_Le; break;
687 case TK_GT: op = OP_Gt; break;
688 case TK_GE: op = OP_Ge; break;
689 case TK_NE: op = OP_Ne; break;
690 case TK_EQ: op = OP_Eq; break;
691 case TK_LIKE: op = OP_Like; break;
692 case TK_GLOB: op = OP_Glob; break;
693 case TK_ISNULL: op = OP_IsNull; break;
694 case TK_NOTNULL: op = OP_NotNull; break;
695 default: break;
696 }
697 switch( pExpr->op ){
698 case TK_AND: {
699 int d2 = sqliteVdbeMakeLabel(v);
700 sqliteExprIfFalse(pParse, pExpr->pLeft, d2);
701 sqliteExprIfTrue(pParse, pExpr->pRight, dest);
702 sqliteVdbeResolveLabel(v, d2);
703 break;
704 }
705 case TK_OR: {
706 sqliteExprIfTrue(pParse, pExpr->pLeft, dest);
707 sqliteExprIfTrue(pParse, pExpr->pRight, dest);
708 break;
709 }
710 case TK_NOT: {
711 sqliteExprIfFalse(pParse, pExpr->pLeft, dest);
712 break;
713 }
714 case TK_LT:
715 case TK_LE:
716 case TK_GT:
717 case TK_GE:
718 case TK_NE:
719 case TK_EQ:
720 case TK_LIKE:
721 case TK_GLOB: {
722 sqliteExprCode(pParse, pExpr->pLeft);
723 sqliteExprCode(pParse, pExpr->pRight);
724 sqliteVdbeAddOp(v, op, 0, dest, 0, 0);
725 break;
726 }
727 case TK_ISNULL:
728 case TK_NOTNULL: {
729 sqliteExprCode(pParse, pExpr->pLeft);
730 sqliteVdbeAddOp(v, op, 0, dest, 0, 0);
731 break;
732 }
drhfef52082000-06-06 01:50:43 +0000733 case TK_IN: {
drhcfab11b2000-06-06 03:31:22 +0000734 sqliteExprCode(pParse, pExpr->pLeft);
drhfef52082000-06-06 01:50:43 +0000735 if( pExpr->pSelect ){
736 sqliteVdbeAddOp(v, OP_Found, pExpr->iTable, dest, 0, 0);
737 }else{
738 sqliteVdbeAddOp(v, OP_SetFound, pExpr->iTable, dest, 0, 0);
739 }
740 break;
741 }
742 case TK_BETWEEN: {
743 int lbl = sqliteVdbeMakeLabel(v);
744 sqliteExprCode(pParse, pExpr->pLeft);
745 sqliteVdbeAddOp(v, OP_Dup, 0, 0, 0, 0);
746 sqliteExprCode(pParse, pExpr->pList->a[0].pExpr);
747 sqliteVdbeAddOp(v, OP_Lt, 0, lbl, 0, 0);
748 sqliteExprCode(pParse, pExpr->pList->a[1].pExpr);
749 sqliteVdbeAddOp(v, OP_Le, 0, dest, 0, 0);
750 sqliteVdbeAddOp(v, OP_Integer, 0, 0, 0, 0);
751 sqliteVdbeAddOp(v, OP_Pop, 1, 0, 0, lbl);
752 break;
753 }
drhcce7d172000-05-31 15:34:51 +0000754 default: {
755 sqliteExprCode(pParse, pExpr);
756 sqliteVdbeAddOp(v, OP_If, 0, dest, 0, 0);
757 break;
758 }
759 }
760}
761
762/*
drh66b89c82000-11-28 20:47:17 +0000763** Generate code for a boolean expression such that a jump is made
drhcce7d172000-05-31 15:34:51 +0000764** to the label "dest" if the expression is false but execution
765** continues straight thru if the expression is true.
766*/
767void sqliteExprIfFalse(Parse *pParse, Expr *pExpr, int dest){
768 Vdbe *v = pParse->pVdbe;
769 int op = 0;
drhdaffd0e2001-04-11 14:28:42 +0000770 if( v==0 || pExpr==0 ) return;
drhcce7d172000-05-31 15:34:51 +0000771 switch( pExpr->op ){
772 case TK_LT: op = OP_Ge; break;
773 case TK_LE: op = OP_Gt; break;
774 case TK_GT: op = OP_Le; break;
775 case TK_GE: op = OP_Lt; break;
776 case TK_NE: op = OP_Eq; break;
777 case TK_EQ: op = OP_Ne; break;
778 case TK_LIKE: op = OP_Like; break;
779 case TK_GLOB: op = OP_Glob; break;
780 case TK_ISNULL: op = OP_NotNull; break;
781 case TK_NOTNULL: op = OP_IsNull; break;
782 default: break;
783 }
784 switch( pExpr->op ){
785 case TK_AND: {
786 sqliteExprIfFalse(pParse, pExpr->pLeft, dest);
787 sqliteExprIfFalse(pParse, pExpr->pRight, dest);
788 break;
789 }
790 case TK_OR: {
791 int d2 = sqliteVdbeMakeLabel(v);
792 sqliteExprIfTrue(pParse, pExpr->pLeft, d2);
793 sqliteExprIfFalse(pParse, pExpr->pRight, dest);
794 sqliteVdbeResolveLabel(v, d2);
795 break;
796 }
797 case TK_NOT: {
798 sqliteExprIfTrue(pParse, pExpr->pLeft, dest);
799 break;
800 }
801 case TK_LT:
802 case TK_LE:
803 case TK_GT:
804 case TK_GE:
805 case TK_NE:
806 case TK_EQ: {
807 sqliteExprCode(pParse, pExpr->pLeft);
808 sqliteExprCode(pParse, pExpr->pRight);
809 sqliteVdbeAddOp(v, op, 0, dest, 0, 0);
810 break;
811 }
812 case TK_LIKE:
813 case TK_GLOB: {
814 sqliteExprCode(pParse, pExpr->pLeft);
815 sqliteExprCode(pParse, pExpr->pRight);
816 sqliteVdbeAddOp(v, op, 1, dest, 0, 0);
817 break;
818 }
819 case TK_ISNULL:
820 case TK_NOTNULL: {
821 sqliteExprCode(pParse, pExpr->pLeft);
822 sqliteVdbeAddOp(v, op, 0, dest, 0, 0);
823 break;
824 }
drhfef52082000-06-06 01:50:43 +0000825 case TK_IN: {
drhcfab11b2000-06-06 03:31:22 +0000826 sqliteExprCode(pParse, pExpr->pLeft);
drhfef52082000-06-06 01:50:43 +0000827 if( pExpr->pSelect ){
828 sqliteVdbeAddOp(v, OP_NotFound, pExpr->iTable, dest, 0, 0);
829 }else{
830 sqliteVdbeAddOp(v, OP_SetNotFound, pExpr->iTable, dest, 0, 0);
831 }
832 break;
833 }
834 case TK_BETWEEN: {
835 int addr;
836 sqliteExprCode(pParse, pExpr->pLeft);
837 sqliteVdbeAddOp(v, OP_Dup, 0, 0, 0, 0);
838 sqliteExprCode(pParse, pExpr->pList->a[0].pExpr);
839 addr = sqliteVdbeCurrentAddr(v);
840 sqliteVdbeAddOp(v, OP_Ge, 0, addr+3, 0, 0);
841 sqliteVdbeAddOp(v, OP_Pop, 1, 0, 0, 0);
842 sqliteVdbeAddOp(v, OP_Goto, 0, dest, 0, 0);
843 sqliteExprCode(pParse, pExpr->pList->a[1].pExpr);
844 sqliteVdbeAddOp(v, OP_Gt, 0, dest, 0, 0);
845 break;
846 }
drhcce7d172000-05-31 15:34:51 +0000847 default: {
848 sqliteExprCode(pParse, pExpr);
849 sqliteVdbeAddOp(v, OP_Not, 0, 0, 0, 0);
850 sqliteVdbeAddOp(v, OP_If, 0, dest, 0, 0);
851 break;
852 }
853 }
854}
drh22827922000-06-06 17:27:05 +0000855
856/*
857** Do a deep comparison of two expression trees. Return TRUE (non-zero)
858** if they are identical and return FALSE if they differ in any way.
859*/
drhd8bc7082000-06-07 23:51:50 +0000860int sqliteExprCompare(Expr *pA, Expr *pB){
drh22827922000-06-06 17:27:05 +0000861 int i;
862 if( pA==0 ){
863 return pB==0;
864 }else if( pB==0 ){
865 return 0;
866 }
867 if( pA->op!=pB->op ) return 0;
drhd8bc7082000-06-07 23:51:50 +0000868 if( !sqliteExprCompare(pA->pLeft, pB->pLeft) ) return 0;
869 if( !sqliteExprCompare(pA->pRight, pB->pRight) ) return 0;
drh22827922000-06-06 17:27:05 +0000870 if( pA->pList ){
871 if( pB->pList==0 ) return 0;
872 if( pA->pList->nExpr!=pB->pList->nExpr ) return 0;
873 for(i=0; i<pA->pList->nExpr; i++){
drhd8bc7082000-06-07 23:51:50 +0000874 if( !sqliteExprCompare(pA->pList->a[i].pExpr, pB->pList->a[i].pExpr) ){
drh22827922000-06-06 17:27:05 +0000875 return 0;
876 }
877 }
878 }else if( pB->pList ){
879 return 0;
880 }
881 if( pA->pSelect || pB->pSelect ) return 0;
882 if( pA->token.z ){
883 if( pB->token.z==0 ) return 0;
884 if( pB->token.n!=pA->token.n ) return 0;
885 if( sqliteStrNICmp(pA->token.z, pB->token.z, pA->token.n)!=0 ) return 0;
886 }
887 return 1;
888}
889
890/*
891** Add a new element to the pParse->aAgg[] array and return its index.
892*/
893static int appendAggInfo(Parse *pParse){
894 if( (pParse->nAgg & 0x7)==0 ){
895 int amt = pParse->nAgg + 8;
896 pParse->aAgg = sqliteRealloc(pParse->aAgg, amt*sizeof(pParse->aAgg[0]));
897 if( pParse->aAgg==0 ){
drhdaffd0e2001-04-11 14:28:42 +0000898 pParse->nAgg = 0;
drh22827922000-06-06 17:27:05 +0000899 return -1;
900 }
901 }
902 memset(&pParse->aAgg[pParse->nAgg], 0, sizeof(pParse->aAgg[0]));
903 return pParse->nAgg++;
904}
905
906/*
907** Analyze the given expression looking for aggregate functions and
908** for variables that need to be added to the pParse->aAgg[] array.
909** Make additional entries to the pParse->aAgg[] array as necessary.
910**
911** This routine should only be called after the expression has been
912** analyzed by sqliteExprResolveIds() and sqliteExprCheck().
913**
914** If errors are seen, leave an error message in zErrMsg and return
915** the number of errors.
916*/
917int sqliteExprAnalyzeAggregates(Parse *pParse, Expr *pExpr){
918 int i;
919 AggExpr *aAgg;
920 int nErr = 0;
921
922 if( pExpr==0 ) return 0;
923 switch( pExpr->op ){
drh967e8b72000-06-21 13:59:10 +0000924 case TK_COLUMN: {
drh22827922000-06-06 17:27:05 +0000925 aAgg = pParse->aAgg;
926 for(i=0; i<pParse->nAgg; i++){
927 if( aAgg[i].isAgg ) continue;
928 if( aAgg[i].pExpr->iTable==pExpr->iTable
drh967e8b72000-06-21 13:59:10 +0000929 && aAgg[i].pExpr->iColumn==pExpr->iColumn ){
drh22827922000-06-06 17:27:05 +0000930 break;
931 }
932 }
933 if( i>=pParse->nAgg ){
934 i = appendAggInfo(pParse);
935 if( i<0 ) return 1;
936 pParse->aAgg[i].isAgg = 0;
937 pParse->aAgg[i].pExpr = pExpr;
938 }
drhaaf88722000-06-08 11:25:00 +0000939 pExpr->iAgg = i;
drh22827922000-06-06 17:27:05 +0000940 break;
941 }
942 case TK_AGG_FUNCTION: {
drh967e8b72000-06-21 13:59:10 +0000943 if( pExpr->iColumn==FN_Count || pExpr->iColumn==FN_Avg ){
drh22827922000-06-06 17:27:05 +0000944 if( pParse->iAggCount>=0 ){
945 i = pParse->iAggCount;
946 }else{
947 i = appendAggInfo(pParse);
948 if( i<0 ) return 1;
949 pParse->aAgg[i].isAgg = 1;
950 pParse->aAgg[i].pExpr = 0;
951 pParse->iAggCount = i;
952 }
drh967e8b72000-06-21 13:59:10 +0000953 if( pExpr->iColumn==FN_Count ){
drh22827922000-06-06 17:27:05 +0000954 pExpr->iAgg = i;
955 break;
956 }
957 }
958 aAgg = pParse->aAgg;
959 for(i=0; i<pParse->nAgg; i++){
960 if( !aAgg[i].isAgg ) continue;
drhd8bc7082000-06-07 23:51:50 +0000961 if( sqliteExprCompare(aAgg[i].pExpr, pExpr) ){
drh22827922000-06-06 17:27:05 +0000962 break;
963 }
964 }
965 if( i>=pParse->nAgg ){
966 i = appendAggInfo(pParse);
967 if( i<0 ) return 1;
968 pParse->aAgg[i].isAgg = 1;
969 pParse->aAgg[i].pExpr = pExpr;
970 }
971 pExpr->iAgg = i;
972 break;
973 }
974 default: {
975 if( pExpr->pLeft ){
976 nErr = sqliteExprAnalyzeAggregates(pParse, pExpr->pLeft);
977 }
978 if( nErr==0 && pExpr->pRight ){
979 nErr = sqliteExprAnalyzeAggregates(pParse, pExpr->pRight);
980 }
981 if( nErr==0 && pExpr->pList ){
982 int n = pExpr->pList->nExpr;
983 int i;
984 for(i=0; nErr==0 && i<n; i++){
985 nErr = sqliteExprAnalyzeAggregates(pParse, pExpr->pList->a[i].pExpr);
986 }
987 }
988 break;
989 }
990 }
991 return nErr;
992}