OLD | NEW |
| (Empty) |
1 /* | |
2 ** 2008 August 16 | |
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 ** This file contains routines used for walking the parser tree for | |
13 ** an SQL statement. | |
14 */ | |
15 #include "sqliteInt.h" | |
16 #include <stdlib.h> | |
17 #include <string.h> | |
18 | |
19 | |
20 /* | |
21 ** Walk an expression tree. Invoke the callback once for each node | |
22 ** of the expression, while descending. (In other words, the callback | |
23 ** is invoked before visiting children.) | |
24 ** | |
25 ** The return value from the callback should be one of the WRC_* | |
26 ** constants to specify how to proceed with the walk. | |
27 ** | |
28 ** WRC_Continue Continue descending down the tree. | |
29 ** | |
30 ** WRC_Prune Do not descend into child nodes. But allow | |
31 ** the walk to continue with sibling nodes. | |
32 ** | |
33 ** WRC_Abort Do no more callbacks. Unwind the stack and | |
34 ** return the top-level walk call. | |
35 ** | |
36 ** The return value from this routine is WRC_Abort to abandon the tree walk | |
37 ** and WRC_Continue to continue. | |
38 */ | |
39 int sqlite3WalkExpr(Walker *pWalker, Expr *pExpr){ | |
40 int rc; | |
41 if( pExpr==0 ) return WRC_Continue; | |
42 testcase( ExprHasProperty(pExpr, EP_TokenOnly) ); | |
43 testcase( ExprHasProperty(pExpr, EP_Reduced) ); | |
44 rc = pWalker->xExprCallback(pWalker, pExpr); | |
45 if( rc==WRC_Continue | |
46 && !ExprHasProperty(pExpr,EP_TokenOnly) ){ | |
47 if( sqlite3WalkExpr(pWalker, pExpr->pLeft) ) return WRC_Abort; | |
48 if( sqlite3WalkExpr(pWalker, pExpr->pRight) ) return WRC_Abort; | |
49 if( ExprHasProperty(pExpr, EP_xIsSelect) ){ | |
50 if( sqlite3WalkSelect(pWalker, pExpr->x.pSelect) ) return WRC_Abort; | |
51 }else{ | |
52 if( sqlite3WalkExprList(pWalker, pExpr->x.pList) ) return WRC_Abort; | |
53 } | |
54 } | |
55 return rc & WRC_Abort; | |
56 } | |
57 | |
58 /* | |
59 ** Call sqlite3WalkExpr() for every expression in list p or until | |
60 ** an abort request is seen. | |
61 */ | |
62 int sqlite3WalkExprList(Walker *pWalker, ExprList *p){ | |
63 int i; | |
64 struct ExprList_item *pItem; | |
65 if( p ){ | |
66 for(i=p->nExpr, pItem=p->a; i>0; i--, pItem++){ | |
67 if( sqlite3WalkExpr(pWalker, pItem->pExpr) ) return WRC_Abort; | |
68 } | |
69 } | |
70 return WRC_Continue; | |
71 } | |
72 | |
73 /* | |
74 ** Walk all expressions associated with SELECT statement p. Do | |
75 ** not invoke the SELECT callback on p, but do (of course) invoke | |
76 ** any expr callbacks and SELECT callbacks that come from subqueries. | |
77 ** Return WRC_Abort or WRC_Continue. | |
78 */ | |
79 int sqlite3WalkSelectExpr(Walker *pWalker, Select *p){ | |
80 if( sqlite3WalkExprList(pWalker, p->pEList) ) return WRC_Abort; | |
81 if( sqlite3WalkExpr(pWalker, p->pWhere) ) return WRC_Abort; | |
82 if( sqlite3WalkExprList(pWalker, p->pGroupBy) ) return WRC_Abort; | |
83 if( sqlite3WalkExpr(pWalker, p->pHaving) ) return WRC_Abort; | |
84 if( sqlite3WalkExprList(pWalker, p->pOrderBy) ) return WRC_Abort; | |
85 if( sqlite3WalkExpr(pWalker, p->pLimit) ) return WRC_Abort; | |
86 if( sqlite3WalkExpr(pWalker, p->pOffset) ) return WRC_Abort; | |
87 return WRC_Continue; | |
88 } | |
89 | |
90 /* | |
91 ** Walk the parse trees associated with all subqueries in the | |
92 ** FROM clause of SELECT statement p. Do not invoke the select | |
93 ** callback on p, but do invoke it on each FROM clause subquery | |
94 ** and on any subqueries further down in the tree. Return | |
95 ** WRC_Abort or WRC_Continue; | |
96 */ | |
97 int sqlite3WalkSelectFrom(Walker *pWalker, Select *p){ | |
98 SrcList *pSrc; | |
99 int i; | |
100 struct SrcList_item *pItem; | |
101 | |
102 pSrc = p->pSrc; | |
103 if( ALWAYS(pSrc) ){ | |
104 for(i=pSrc->nSrc, pItem=pSrc->a; i>0; i--, pItem++){ | |
105 if( sqlite3WalkSelect(pWalker, pItem->pSelect) ){ | |
106 return WRC_Abort; | |
107 } | |
108 } | |
109 } | |
110 return WRC_Continue; | |
111 } | |
112 | |
113 /* | |
114 ** Call sqlite3WalkExpr() for every expression in Select statement p. | |
115 ** Invoke sqlite3WalkSelect() for subqueries in the FROM clause and | |
116 ** on the compound select chain, p->pPrior. | |
117 ** | |
118 ** If it is not NULL, the xSelectCallback() callback is invoked before | |
119 ** the walk of the expressions and FROM clause. The xSelectCallback2() | |
120 ** method, if it is not NULL, is invoked following the walk of the | |
121 ** expressions and FROM clause. | |
122 ** | |
123 ** Return WRC_Continue under normal conditions. Return WRC_Abort if | |
124 ** there is an abort request. | |
125 ** | |
126 ** If the Walker does not have an xSelectCallback() then this routine | |
127 ** is a no-op returning WRC_Continue. | |
128 */ | |
129 int sqlite3WalkSelect(Walker *pWalker, Select *p){ | |
130 int rc; | |
131 if( p==0 || (pWalker->xSelectCallback==0 && pWalker->xSelectCallback2==0) ){ | |
132 return WRC_Continue; | |
133 } | |
134 rc = WRC_Continue; | |
135 pWalker->walkerDepth++; | |
136 while( p ){ | |
137 if( pWalker->xSelectCallback ){ | |
138 rc = pWalker->xSelectCallback(pWalker, p); | |
139 if( rc ) break; | |
140 } | |
141 if( sqlite3WalkSelectExpr(pWalker, p) | |
142 || sqlite3WalkSelectFrom(pWalker, p) | |
143 ){ | |
144 pWalker->walkerDepth--; | |
145 return WRC_Abort; | |
146 } | |
147 if( pWalker->xSelectCallback2 ){ | |
148 pWalker->xSelectCallback2(pWalker, p); | |
149 } | |
150 p = p->pPrior; | |
151 } | |
152 pWalker->walkerDepth--; | |
153 return rc & WRC_Abort; | |
154 } | |
OLD | NEW |