drh | 7589723 | 2000-05-29 14:26:00 +0000 | [diff] [blame] | 1 | /* |
drh | b19a2bc | 2001-09-16 00:13:26 +0000 | [diff] [blame] | 2 | ** 2001 September 15 |
drh | 7589723 | 2000-05-29 14:26:00 +0000 | [diff] [blame] | 3 | ** |
drh | b19a2bc | 2001-09-16 00:13:26 +0000 | [diff] [blame] | 4 | ** The author disclaims copyright to this source code. In place of |
| 5 | ** a legal notice, here is a blessing: |
drh | 7589723 | 2000-05-29 14:26:00 +0000 | [diff] [blame] | 6 | ** |
drh | b19a2bc | 2001-09-16 00:13:26 +0000 | [diff] [blame] | 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. |
drh | 7589723 | 2000-05-29 14:26:00 +0000 | [diff] [blame] | 10 | ** |
| 11 | ************************************************************************* |
| 12 | ** An tokenizer for SQL |
| 13 | ** |
| 14 | ** This file contains C code that splits an SQL input string up into |
| 15 | ** individual tokens and sends those tokens one-by-one over to the |
| 16 | ** parser for analysis. |
drh | 7589723 | 2000-05-29 14:26:00 +0000 | [diff] [blame] | 17 | */ |
| 18 | #include "sqliteInt.h" |
drh | dcc581c | 2000-05-30 13:44:19 +0000 | [diff] [blame] | 19 | #include <stdlib.h> |
drh | 7589723 | 2000-05-29 14:26:00 +0000 | [diff] [blame] | 20 | |
drh | 34dcee6 | 2016-02-08 19:15:48 +0000 | [diff] [blame] | 21 | /* Character classes for tokenizing |
| 22 | ** |
| 23 | ** In the sqlite3GetToken() function, a switch() on aiClass[c] is implemented |
| 24 | ** using a lookup table, whereas a switch() directly on c uses a binary search. |
| 25 | ** The lookup table is much faster. To maximize speed, and to ensure that |
| 26 | ** a lookup table is used, all of the classes need to be small integers and |
| 27 | ** all of them need to be used within the switch. |
| 28 | */ |
drh | e96f361 | 2016-02-08 19:36:46 +0000 | [diff] [blame] | 29 | #define CC_X 0 /* The letter 'x', or start of BLOB literal */ |
drh | d1032f9 | 2020-11-27 20:56:16 +0000 | [diff] [blame] | 30 | #define CC_KYWD0 1 /* First letter of a keyword */ |
| 31 | #define CC_KYWD 2 /* Alphabetics or '_'. Usable in a keyword */ |
drh | 8974331 | 2016-02-08 02:30:50 +0000 | [diff] [blame] | 32 | #define CC_DIGIT 3 /* Digits */ |
| 33 | #define CC_DOLLAR 4 /* '$' */ |
| 34 | #define CC_VARALPHA 5 /* '@', '#', ':'. Alphabetic SQL variables */ |
| 35 | #define CC_VARNUM 6 /* '?'. Numeric SQL variables */ |
| 36 | #define CC_SPACE 7 /* Space characters */ |
| 37 | #define CC_QUOTE 8 /* '"', '\'', or '`'. String literals, quoted ids */ |
| 38 | #define CC_QUOTE2 9 /* '['. [...] style quoted ids */ |
| 39 | #define CC_PIPE 10 /* '|'. Bitwise OR or concatenate */ |
| 40 | #define CC_MINUS 11 /* '-'. Minus or SQL-style comment */ |
| 41 | #define CC_LT 12 /* '<'. Part of < or <= or <> */ |
| 42 | #define CC_GT 13 /* '>'. Part of > or >= */ |
| 43 | #define CC_EQ 14 /* '='. Part of = or == */ |
| 44 | #define CC_BANG 15 /* '!'. Part of != */ |
| 45 | #define CC_SLASH 16 /* '/'. / or c-style comment */ |
| 46 | #define CC_LP 17 /* '(' */ |
| 47 | #define CC_RP 18 /* ')' */ |
| 48 | #define CC_SEMI 19 /* ';' */ |
| 49 | #define CC_PLUS 20 /* '+' */ |
| 50 | #define CC_STAR 21 /* '*' */ |
| 51 | #define CC_PERCENT 22 /* '%' */ |
| 52 | #define CC_COMMA 23 /* ',' */ |
| 53 | #define CC_AND 24 /* '&' */ |
| 54 | #define CC_TILDA 25 /* '~' */ |
| 55 | #define CC_DOT 26 /* '.' */ |
drh | d1032f9 | 2020-11-27 20:56:16 +0000 | [diff] [blame] | 56 | #define CC_ID 27 /* unicode characters usable in IDs */ |
| 57 | #define CC_ILLEGAL 28 /* Illegal character */ |
| 58 | #define CC_NUL 29 /* 0x00 */ |
drh | ba9ebc2 | 2021-04-24 12:24:08 +0000 | [diff] [blame] | 59 | #define CC_BOM 30 /* First byte of UTF8 BOM: 0xEF 0xBB 0xBF */ |
drh | 8974331 | 2016-02-08 02:30:50 +0000 | [diff] [blame] | 60 | |
| 61 | static const unsigned char aiClass[] = { |
drh | 34dcee6 | 2016-02-08 19:15:48 +0000 | [diff] [blame] | 62 | #ifdef SQLITE_ASCII |
drh | 8974331 | 2016-02-08 02:30:50 +0000 | [diff] [blame] | 63 | /* x0 x1 x2 x3 x4 x5 x6 x7 x8 x9 xa xb xc xd xe xf */ |
drh | d1032f9 | 2020-11-27 20:56:16 +0000 | [diff] [blame] | 64 | /* 0x */ 29, 28, 28, 28, 28, 28, 28, 28, 28, 7, 7, 28, 7, 7, 28, 28, |
| 65 | /* 1x */ 28, 28, 28, 28, 28, 28, 28, 28, 28, 28, 28, 28, 28, 28, 28, 28, |
drh | 8974331 | 2016-02-08 02:30:50 +0000 | [diff] [blame] | 66 | /* 2x */ 7, 15, 8, 5, 4, 22, 24, 8, 17, 18, 21, 20, 23, 11, 26, 16, |
| 67 | /* 3x */ 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 5, 19, 12, 14, 13, 6, |
| 68 | /* 4x */ 5, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, |
drh | d1032f9 | 2020-11-27 20:56:16 +0000 | [diff] [blame] | 69 | /* 5x */ 1, 1, 1, 1, 1, 1, 1, 1, 0, 2, 2, 9, 28, 28, 28, 2, |
drh | 8974331 | 2016-02-08 02:30:50 +0000 | [diff] [blame] | 70 | /* 6x */ 8, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, |
drh | d1032f9 | 2020-11-27 20:56:16 +0000 | [diff] [blame] | 71 | /* 7x */ 1, 1, 1, 1, 1, 1, 1, 1, 0, 2, 2, 28, 10, 28, 25, 28, |
drh | ba9ebc2 | 2021-04-24 12:24:08 +0000 | [diff] [blame] | 72 | /* 8x */ 27, 27, 27, 27, 27, 27, 27, 27, 27, 27, 27, 27, 27, 27, 27, 27, |
| 73 | /* 9x */ 27, 27, 27, 27, 27, 27, 27, 27, 27, 27, 27, 27, 27, 27, 27, 27, |
| 74 | /* Ax */ 27, 27, 27, 27, 27, 27, 27, 27, 27, 27, 27, 27, 27, 27, 27, 27, |
| 75 | /* Bx */ 27, 27, 27, 27, 27, 27, 27, 27, 27, 27, 27, 27, 27, 27, 27, 27, |
| 76 | /* Cx */ 27, 27, 27, 27, 27, 27, 27, 27, 27, 27, 27, 27, 27, 27, 27, 27, |
| 77 | /* Dx */ 27, 27, 27, 27, 27, 27, 27, 27, 27, 27, 27, 27, 27, 27, 27, 27, |
| 78 | /* Ex */ 27, 27, 27, 27, 27, 27, 27, 27, 27, 27, 27, 27, 27, 27, 27, 30, |
| 79 | /* Fx */ 27, 27, 27, 27, 27, 27, 27, 27, 27, 27, 27, 27, 27, 27, 27, 27 |
drh | 34dcee6 | 2016-02-08 19:15:48 +0000 | [diff] [blame] | 80 | #endif |
| 81 | #ifdef SQLITE_EBCDIC |
| 82 | /* x0 x1 x2 x3 x4 x5 x6 x7 x8 x9 xa xb xc xd xe xf */ |
drh | d1032f9 | 2020-11-27 20:56:16 +0000 | [diff] [blame] | 83 | /* 0x */ 29, 28, 28, 28, 28, 7, 28, 28, 28, 28, 28, 28, 7, 7, 28, 28, |
larrybr | 4cf34a5 | 2021-03-19 15:02:59 +0000 | [diff] [blame] | 84 | /* 1x */ 28, 28, 28, 28, 28, 7, 28, 28, 28, 28, 28, 28, 28, 28, 28, 28, |
drh | d1032f9 | 2020-11-27 20:56:16 +0000 | [diff] [blame] | 85 | /* 2x */ 28, 28, 28, 28, 28, 7, 28, 28, 28, 28, 28, 28, 28, 28, 28, 28, |
| 86 | /* 3x */ 28, 28, 28, 28, 28, 28, 28, 28, 28, 28, 28, 28, 28, 28, 28, 28, |
| 87 | /* 4x */ 7, 28, 28, 28, 28, 28, 28, 28, 28, 28, 28, 26, 12, 17, 20, 10, |
| 88 | /* 5x */ 24, 28, 28, 28, 28, 28, 28, 28, 28, 28, 15, 4, 21, 18, 19, 28, |
| 89 | /* 6x */ 11, 16, 28, 28, 28, 28, 28, 28, 28, 28, 28, 23, 22, 2, 13, 6, |
| 90 | /* 7x */ 28, 28, 28, 28, 28, 28, 28, 28, 28, 8, 5, 5, 5, 8, 14, 8, |
| 91 | /* 8x */ 28, 1, 1, 1, 1, 1, 1, 1, 1, 1, 28, 28, 28, 28, 28, 28, |
| 92 | /* 9x */ 28, 1, 1, 1, 1, 1, 1, 1, 1, 1, 28, 28, 28, 28, 28, 28, |
| 93 | /* Ax */ 28, 25, 1, 1, 1, 1, 1, 0, 2, 2, 28, 28, 28, 28, 28, 28, |
| 94 | /* Bx */ 28, 28, 28, 28, 28, 28, 28, 28, 28, 28, 9, 28, 28, 28, 28, 28, |
| 95 | /* Cx */ 28, 1, 1, 1, 1, 1, 1, 1, 1, 1, 28, 28, 28, 28, 28, 28, |
| 96 | /* Dx */ 28, 1, 1, 1, 1, 1, 1, 1, 1, 1, 28, 28, 28, 28, 28, 28, |
| 97 | /* Ex */ 28, 28, 1, 1, 1, 1, 1, 0, 2, 2, 28, 28, 28, 28, 28, 28, |
| 98 | /* Fx */ 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 28, 28, 28, 28, 28, 28, |
drh | 34dcee6 | 2016-02-08 19:15:48 +0000 | [diff] [blame] | 99 | #endif |
drh | 8974331 | 2016-02-08 02:30:50 +0000 | [diff] [blame] | 100 | }; |
| 101 | |
drh | 7589723 | 2000-05-29 14:26:00 +0000 | [diff] [blame] | 102 | /* |
drh | 34dcee6 | 2016-02-08 19:15:48 +0000 | [diff] [blame] | 103 | ** The charMap() macro maps alphabetic characters (only) into their |
drh | 9b8f447 | 2006-04-04 01:54:55 +0000 | [diff] [blame] | 104 | ** lower-case ASCII equivalent. On ASCII machines, this is just |
| 105 | ** an upper-to-lower case map. On EBCDIC machines we also need |
drh | 34dcee6 | 2016-02-08 19:15:48 +0000 | [diff] [blame] | 106 | ** to adjust the encoding. The mapping is only valid for alphabetics |
| 107 | ** which are the only characters for which this feature is used. |
| 108 | ** |
| 109 | ** Used by keywordhash.h |
drh | 9b8f447 | 2006-04-04 01:54:55 +0000 | [diff] [blame] | 110 | */ |
| 111 | #ifdef SQLITE_ASCII |
| 112 | # define charMap(X) sqlite3UpperToLower[(unsigned char)X] |
| 113 | #endif |
| 114 | #ifdef SQLITE_EBCDIC |
| 115 | # define charMap(X) ebcdicToAscii[(unsigned char)X] |
| 116 | const unsigned char ebcdicToAscii[] = { |
| 117 | /* 0 1 2 3 4 5 6 7 8 9 A B C D E F */ |
| 118 | 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, /* 0x */ |
| 119 | 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, /* 1x */ |
| 120 | 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, /* 2x */ |
| 121 | 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, /* 3x */ |
| 122 | 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, /* 4x */ |
| 123 | 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, /* 5x */ |
| 124 | 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 95, 0, 0, /* 6x */ |
| 125 | 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, /* 7x */ |
| 126 | 0, 97, 98, 99,100,101,102,103,104,105, 0, 0, 0, 0, 0, 0, /* 8x */ |
| 127 | 0,106,107,108,109,110,111,112,113,114, 0, 0, 0, 0, 0, 0, /* 9x */ |
| 128 | 0, 0,115,116,117,118,119,120,121,122, 0, 0, 0, 0, 0, 0, /* Ax */ |
| 129 | 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, /* Bx */ |
| 130 | 0, 97, 98, 99,100,101,102,103,104,105, 0, 0, 0, 0, 0, 0, /* Cx */ |
| 131 | 0,106,107,108,109,110,111,112,113,114, 0, 0, 0, 0, 0, 0, /* Dx */ |
| 132 | 0, 0,115,116,117,118,119,120,121,122, 0, 0, 0, 0, 0, 0, /* Ex */ |
| 133 | 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, /* Fx */ |
| 134 | }; |
| 135 | #endif |
| 136 | |
| 137 | /* |
drh | 52fb6d7 | 2004-11-03 03:59:57 +0000 | [diff] [blame] | 138 | ** The sqlite3KeywordCode function looks up an identifier to determine if |
| 139 | ** it is a keyword. If it is a keyword, the token code of that keyword is |
drh | 7589723 | 2000-05-29 14:26:00 +0000 | [diff] [blame] | 140 | ** returned. If the input is not a keyword, TK_ID is returned. |
drh | 2090a0e | 2004-10-07 19:03:01 +0000 | [diff] [blame] | 141 | ** |
| 142 | ** The implementation of this routine was generated by a program, |
drh | e96f361 | 2016-02-08 19:36:46 +0000 | [diff] [blame] | 143 | ** mkkeywordhash.c, located in the tool subdirectory of the distribution. |
drh | 52fb6d7 | 2004-11-03 03:59:57 +0000 | [diff] [blame] | 144 | ** The output of the mkkeywordhash.c program is written into a file |
drh | 73b211a | 2005-01-18 04:00:42 +0000 | [diff] [blame] | 145 | ** named keywordhash.h and then included into this source file by |
drh | 52fb6d7 | 2004-11-03 03:59:57 +0000 | [diff] [blame] | 146 | ** the #include below. |
drh | 7589723 | 2000-05-29 14:26:00 +0000 | [diff] [blame] | 147 | */ |
drh | 73b211a | 2005-01-18 04:00:42 +0000 | [diff] [blame] | 148 | #include "keywordhash.h" |
drh | 7589723 | 2000-05-29 14:26:00 +0000 | [diff] [blame] | 149 | |
drh | 40f20f7 | 2004-10-23 05:10:18 +0000 | [diff] [blame] | 150 | |
drh | 98808ba | 2001-10-18 12:34:46 +0000 | [diff] [blame] | 151 | /* |
drh | 9b8f447 | 2006-04-04 01:54:55 +0000 | [diff] [blame] | 152 | ** If X is a character that can be used in an identifier then |
| 153 | ** IdChar(X) will be true. Otherwise it is false. |
drh | 98808ba | 2001-10-18 12:34:46 +0000 | [diff] [blame] | 154 | ** |
drh | 9b8f447 | 2006-04-04 01:54:55 +0000 | [diff] [blame] | 155 | ** For ASCII, any character with the high-order bit set is |
| 156 | ** allowed in an identifier. For 7-bit characters, |
| 157 | ** sqlite3IsIdChar[X] must be 1. |
| 158 | ** |
| 159 | ** For EBCDIC, the rules are more complex but have the same |
| 160 | ** end result. |
drh | a0d1f66 | 2005-01-11 17:59:47 +0000 | [diff] [blame] | 161 | ** |
| 162 | ** Ticket #1066. the SQL standard does not allow '$' in the |
peter.d.reid | 60ec914 | 2014-09-06 16:39:46 +0000 | [diff] [blame] | 163 | ** middle of identifiers. But many SQL implementations do. |
drh | a0d1f66 | 2005-01-11 17:59:47 +0000 | [diff] [blame] | 164 | ** SQLite will allow '$' in identifiers for compatibility. |
| 165 | ** But the feature is undocumented. |
drh | 98808ba | 2001-10-18 12:34:46 +0000 | [diff] [blame] | 166 | */ |
drh | 9b8f447 | 2006-04-04 01:54:55 +0000 | [diff] [blame] | 167 | #ifdef SQLITE_ASCII |
drh | af2b572 | 2009-11-16 15:11:51 +0000 | [diff] [blame] | 168 | #define IdChar(C) ((sqlite3CtypeMap[(unsigned char)C]&0x46)!=0) |
drh | 9b8f447 | 2006-04-04 01:54:55 +0000 | [diff] [blame] | 169 | #endif |
| 170 | #ifdef SQLITE_EBCDIC |
drh | 46c99e0 | 2007-08-27 23:26:59 +0000 | [diff] [blame] | 171 | const char sqlite3IsEbcdicIdChar[] = { |
drh | 9b8f447 | 2006-04-04 01:54:55 +0000 | [diff] [blame] | 172 | /* x0 x1 x2 x3 x4 x5 x6 x7 x8 x9 xA xB xC xD xE xF */ |
| 173 | 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0, 0, /* 4x */ |
| 174 | 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 0, 0, 0, 0, /* 5x */ |
| 175 | 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 0, 1, 0, 0, /* 6x */ |
| 176 | 0, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0, 0, 0, /* 7x */ |
| 177 | 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 1, 1, 1, 0, /* 8x */ |
| 178 | 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 1, 0, 1, 0, /* 9x */ |
| 179 | 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 0, /* Ax */ |
| 180 | 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, /* Bx */ |
| 181 | 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, /* Cx */ |
| 182 | 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, /* Dx */ |
| 183 | 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, /* Ex */ |
| 184 | 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 0, /* Fx */ |
| 185 | }; |
drh | 46c99e0 | 2007-08-27 23:26:59 +0000 | [diff] [blame] | 186 | #define IdChar(C) (((c=C)>=0x42 && sqlite3IsEbcdicIdChar[c-0x40])) |
drh | 9b8f447 | 2006-04-04 01:54:55 +0000 | [diff] [blame] | 187 | #endif |
drh | f5ed7ad | 2015-06-15 14:43:25 +0000 | [diff] [blame] | 188 | |
drh | 38d9964 | 2018-08-18 18:27:18 +0000 | [diff] [blame] | 189 | /* Make the IdChar function accessible from ctime.c and alter.c */ |
drh | 97348b3 | 2014-09-25 02:44:29 +0000 | [diff] [blame] | 190 | int sqlite3IsIdChar(u8 c){ return IdChar(c); } |
drh | 9b8f447 | 2006-04-04 01:54:55 +0000 | [diff] [blame] | 191 | |
dan | 34a7d79 | 2018-06-29 20:43:33 +0000 | [diff] [blame] | 192 | #ifndef SQLITE_OMIT_WINDOWFUNC |
dan | 59ff425 | 2018-06-29 17:44:52 +0000 | [diff] [blame] | 193 | /* |
| 194 | ** Return the id of the next token in string (*pz). Before returning, set |
| 195 | ** (*pz) to point to the byte following the parsed token. |
dan | 59ff425 | 2018-06-29 17:44:52 +0000 | [diff] [blame] | 196 | */ |
dan | 6e2210e | 2018-06-30 18:54:56 +0000 | [diff] [blame] | 197 | static int getToken(const unsigned char **pz){ |
dan | 59ff425 | 2018-06-29 17:44:52 +0000 | [diff] [blame] | 198 | const unsigned char *z = *pz; |
dan | 6e2210e | 2018-06-30 18:54:56 +0000 | [diff] [blame] | 199 | int t; /* Token type to return */ |
| 200 | do { |
| 201 | z += sqlite3GetToken(z, &t); |
| 202 | }while( t==TK_SPACE ); |
| 203 | if( t==TK_ID |
| 204 | || t==TK_STRING |
| 205 | || t==TK_JOIN_KW |
| 206 | || t==TK_WINDOW |
| 207 | || t==TK_OVER |
| 208 | || sqlite3ParserFallback(t)==TK_ID |
| 209 | ){ |
| 210 | t = TK_ID; |
dan | 59ff425 | 2018-06-29 17:44:52 +0000 | [diff] [blame] | 211 | } |
| 212 | *pz = z; |
dan | 6e2210e | 2018-06-30 18:54:56 +0000 | [diff] [blame] | 213 | return t; |
dan | 59ff425 | 2018-06-29 17:44:52 +0000 | [diff] [blame] | 214 | } |
| 215 | |
| 216 | /* |
dan | 6e2210e | 2018-06-30 18:54:56 +0000 | [diff] [blame] | 217 | ** The following three functions are called immediately after the tokenizer |
| 218 | ** reads the keywords WINDOW, OVER and FILTER, respectively, to determine |
| 219 | ** whether the token should be treated as a keyword or an SQL identifier. |
| 220 | ** This cannot be handled by the usual lemon %fallback method, due to |
| 221 | ** the ambiguity in some constructions. e.g. |
dan | 59ff425 | 2018-06-29 17:44:52 +0000 | [diff] [blame] | 222 | ** |
dan | 6e2210e | 2018-06-30 18:54:56 +0000 | [diff] [blame] | 223 | ** SELECT sum(x) OVER ... |
dan | 59ff425 | 2018-06-29 17:44:52 +0000 | [diff] [blame] | 224 | ** |
dan | 6e2210e | 2018-06-30 18:54:56 +0000 | [diff] [blame] | 225 | ** In the above, "OVER" might be a keyword, or it might be an alias for the |
| 226 | ** sum(x) expression. If a "%fallback ID OVER" directive were added to |
| 227 | ** grammar, then SQLite would always treat "OVER" as an alias, making it |
| 228 | ** impossible to call a window-function without a FILTER clause. |
| 229 | ** |
| 230 | ** WINDOW is treated as a keyword if: |
| 231 | ** |
| 232 | ** * the following token is an identifier, or a keyword that can fallback |
| 233 | ** to being an identifier, and |
| 234 | ** * the token after than one is TK_AS. |
| 235 | ** |
| 236 | ** OVER is a keyword if: |
| 237 | ** |
| 238 | ** * the previous token was TK_RP, and |
| 239 | ** * the next token is either TK_LP or an identifier. |
| 240 | ** |
| 241 | ** FILTER is a keyword if: |
| 242 | ** |
| 243 | ** * the previous token was TK_RP, and |
| 244 | ** * the next token is TK_LP. |
dan | 59ff425 | 2018-06-29 17:44:52 +0000 | [diff] [blame] | 245 | */ |
dan | 769309f | 2018-06-29 19:54:51 +0000 | [diff] [blame] | 246 | static int analyzeWindowKeyword(const unsigned char *z){ |
dan | 59ff425 | 2018-06-29 17:44:52 +0000 | [diff] [blame] | 247 | int t; |
dan | 6e2210e | 2018-06-30 18:54:56 +0000 | [diff] [blame] | 248 | t = getToken(&z); |
| 249 | if( t!=TK_ID ) return TK_ID; |
| 250 | t = getToken(&z); |
| 251 | if( t!=TK_AS ) return TK_ID; |
| 252 | return TK_WINDOW; |
| 253 | } |
| 254 | static int analyzeOverKeyword(const unsigned char *z, int lastToken){ |
| 255 | if( lastToken==TK_RP ){ |
| 256 | int t = getToken(&z); |
| 257 | if( t==TK_LP || t==TK_ID ) return TK_OVER; |
dan | 59ff425 | 2018-06-29 17:44:52 +0000 | [diff] [blame] | 258 | } |
dan | 6e2210e | 2018-06-30 18:54:56 +0000 | [diff] [blame] | 259 | return TK_ID; |
| 260 | } |
| 261 | static int analyzeFilterKeyword(const unsigned char *z, int lastToken){ |
| 262 | if( lastToken==TK_RP && getToken(&z)==TK_LP ){ |
| 263 | return TK_FILTER; |
| 264 | } |
| 265 | return TK_ID; |
dan | 59ff425 | 2018-06-29 17:44:52 +0000 | [diff] [blame] | 266 | } |
drh | c7bf571 | 2018-07-09 22:49:01 +0000 | [diff] [blame] | 267 | #endif /* SQLITE_OMIT_WINDOWFUNC */ |
drh | 98808ba | 2001-10-18 12:34:46 +0000 | [diff] [blame] | 268 | |
drh | 7589723 | 2000-05-29 14:26:00 +0000 | [diff] [blame] | 269 | /* |
drh | e96f361 | 2016-02-08 19:36:46 +0000 | [diff] [blame] | 270 | ** Return the length (in bytes) of the token that begins at z[0]. |
drh | 61b487d | 2003-09-12 02:08:14 +0000 | [diff] [blame] | 271 | ** Store the token type in *tokenType before returning. |
drh | 7589723 | 2000-05-29 14:26:00 +0000 | [diff] [blame] | 272 | */ |
drh | 35b5a33 | 2008-04-05 18:41:42 +0000 | [diff] [blame] | 273 | int sqlite3GetToken(const unsigned char *z, int *tokenType){ |
drh | aa756b0 | 2004-09-25 15:25:26 +0000 | [diff] [blame] | 274 | int i, c; |
drh | e96f361 | 2016-02-08 19:36:46 +0000 | [diff] [blame] | 275 | switch( aiClass[*z] ){ /* Switch on the character-class of the first byte |
| 276 | ** of the token. See the comment on the CC_ defines |
| 277 | ** above. */ |
drh | 8974331 | 2016-02-08 02:30:50 +0000 | [diff] [blame] | 278 | case CC_SPACE: { |
drh | 19f81f6 | 2009-06-09 18:01:37 +0000 | [diff] [blame] | 279 | testcase( z[0]==' ' ); |
| 280 | testcase( z[0]=='\t' ); |
| 281 | testcase( z[0]=='\n' ); |
| 282 | testcase( z[0]=='\f' ); |
| 283 | testcase( z[0]=='\r' ); |
danielk1977 | 78ca0e7 | 2009-01-20 16:53:39 +0000 | [diff] [blame] | 284 | for(i=1; sqlite3Isspace(z[i]); i++){} |
drh | 7589723 | 2000-05-29 14:26:00 +0000 | [diff] [blame] | 285 | *tokenType = TK_SPACE; |
| 286 | return i; |
| 287 | } |
drh | 8974331 | 2016-02-08 02:30:50 +0000 | [diff] [blame] | 288 | case CC_MINUS: { |
drh | 7589723 | 2000-05-29 14:26:00 +0000 | [diff] [blame] | 289 | if( z[1]=='-' ){ |
drh | aa756b0 | 2004-09-25 15:25:26 +0000 | [diff] [blame] | 290 | for(i=2; (c=z[i])!=0 && c!='\n'; i++){} |
drh | d3479b9 | 2009-12-14 17:42:12 +0000 | [diff] [blame] | 291 | *tokenType = TK_SPACE; /* IMP: R-22934-25134 */ |
drh | 7589723 | 2000-05-29 14:26:00 +0000 | [diff] [blame] | 292 | return i; |
| 293 | } |
| 294 | *tokenType = TK_MINUS; |
| 295 | return 1; |
| 296 | } |
drh | 8974331 | 2016-02-08 02:30:50 +0000 | [diff] [blame] | 297 | case CC_LP: { |
drh | dab3518 | 2003-09-27 13:39:38 +0000 | [diff] [blame] | 298 | *tokenType = TK_LP; |
| 299 | return 1; |
drh | 7589723 | 2000-05-29 14:26:00 +0000 | [diff] [blame] | 300 | } |
drh | 8974331 | 2016-02-08 02:30:50 +0000 | [diff] [blame] | 301 | case CC_RP: { |
drh | 7589723 | 2000-05-29 14:26:00 +0000 | [diff] [blame] | 302 | *tokenType = TK_RP; |
| 303 | return 1; |
| 304 | } |
drh | 8974331 | 2016-02-08 02:30:50 +0000 | [diff] [blame] | 305 | case CC_SEMI: { |
drh | 7589723 | 2000-05-29 14:26:00 +0000 | [diff] [blame] | 306 | *tokenType = TK_SEMI; |
| 307 | return 1; |
| 308 | } |
drh | 8974331 | 2016-02-08 02:30:50 +0000 | [diff] [blame] | 309 | case CC_PLUS: { |
drh | 7589723 | 2000-05-29 14:26:00 +0000 | [diff] [blame] | 310 | *tokenType = TK_PLUS; |
| 311 | return 1; |
| 312 | } |
drh | 8974331 | 2016-02-08 02:30:50 +0000 | [diff] [blame] | 313 | case CC_STAR: { |
drh | 7589723 | 2000-05-29 14:26:00 +0000 | [diff] [blame] | 314 | *tokenType = TK_STAR; |
| 315 | return 1; |
| 316 | } |
drh | 8974331 | 2016-02-08 02:30:50 +0000 | [diff] [blame] | 317 | case CC_SLASH: { |
drh | 66105a8 | 2002-08-27 14:28:29 +0000 | [diff] [blame] | 318 | if( z[1]!='*' || z[2]==0 ){ |
| 319 | *tokenType = TK_SLASH; |
| 320 | return 1; |
| 321 | } |
drh | aa756b0 | 2004-09-25 15:25:26 +0000 | [diff] [blame] | 322 | for(i=3, c=z[2]; (c!='*' || z[i]!='/') && (c=z[i])!=0; i++){} |
| 323 | if( c ) i++; |
drh | d3479b9 | 2009-12-14 17:42:12 +0000 | [diff] [blame] | 324 | *tokenType = TK_SPACE; /* IMP: R-22934-25134 */ |
drh | 66105a8 | 2002-08-27 14:28:29 +0000 | [diff] [blame] | 325 | return i; |
drh | 7589723 | 2000-05-29 14:26:00 +0000 | [diff] [blame] | 326 | } |
drh | 8974331 | 2016-02-08 02:30:50 +0000 | [diff] [blame] | 327 | case CC_PERCENT: { |
drh | bf4133c | 2001-10-13 02:59:08 +0000 | [diff] [blame] | 328 | *tokenType = TK_REM; |
| 329 | return 1; |
| 330 | } |
drh | 8974331 | 2016-02-08 02:30:50 +0000 | [diff] [blame] | 331 | case CC_EQ: { |
drh | 7589723 | 2000-05-29 14:26:00 +0000 | [diff] [blame] | 332 | *tokenType = TK_EQ; |
| 333 | return 1 + (z[1]=='='); |
| 334 | } |
drh | 8974331 | 2016-02-08 02:30:50 +0000 | [diff] [blame] | 335 | case CC_LT: { |
drh | aa756b0 | 2004-09-25 15:25:26 +0000 | [diff] [blame] | 336 | if( (c=z[1])=='=' ){ |
drh | 7589723 | 2000-05-29 14:26:00 +0000 | [diff] [blame] | 337 | *tokenType = TK_LE; |
| 338 | return 2; |
drh | aa756b0 | 2004-09-25 15:25:26 +0000 | [diff] [blame] | 339 | }else if( c=='>' ){ |
drh | 7589723 | 2000-05-29 14:26:00 +0000 | [diff] [blame] | 340 | *tokenType = TK_NE; |
| 341 | return 2; |
drh | aa756b0 | 2004-09-25 15:25:26 +0000 | [diff] [blame] | 342 | }else if( c=='<' ){ |
drh | bf4133c | 2001-10-13 02:59:08 +0000 | [diff] [blame] | 343 | *tokenType = TK_LSHIFT; |
| 344 | return 2; |
drh | 7589723 | 2000-05-29 14:26:00 +0000 | [diff] [blame] | 345 | }else{ |
| 346 | *tokenType = TK_LT; |
| 347 | return 1; |
| 348 | } |
| 349 | } |
drh | 8974331 | 2016-02-08 02:30:50 +0000 | [diff] [blame] | 350 | case CC_GT: { |
drh | aa756b0 | 2004-09-25 15:25:26 +0000 | [diff] [blame] | 351 | if( (c=z[1])=='=' ){ |
drh | 7589723 | 2000-05-29 14:26:00 +0000 | [diff] [blame] | 352 | *tokenType = TK_GE; |
| 353 | return 2; |
drh | aa756b0 | 2004-09-25 15:25:26 +0000 | [diff] [blame] | 354 | }else if( c=='>' ){ |
drh | bf4133c | 2001-10-13 02:59:08 +0000 | [diff] [blame] | 355 | *tokenType = TK_RSHIFT; |
| 356 | return 2; |
drh | 7589723 | 2000-05-29 14:26:00 +0000 | [diff] [blame] | 357 | }else{ |
| 358 | *tokenType = TK_GT; |
| 359 | return 1; |
| 360 | } |
| 361 | } |
drh | 8974331 | 2016-02-08 02:30:50 +0000 | [diff] [blame] | 362 | case CC_BANG: { |
drh | 7589723 | 2000-05-29 14:26:00 +0000 | [diff] [blame] | 363 | if( z[1]!='=' ){ |
| 364 | *tokenType = TK_ILLEGAL; |
drh | b2bddbb | 2016-02-18 14:49:28 +0000 | [diff] [blame] | 365 | return 1; |
drh | 7589723 | 2000-05-29 14:26:00 +0000 | [diff] [blame] | 366 | }else{ |
| 367 | *tokenType = TK_NE; |
| 368 | return 2; |
| 369 | } |
| 370 | } |
drh | 8974331 | 2016-02-08 02:30:50 +0000 | [diff] [blame] | 371 | case CC_PIPE: { |
drh | 0040077 | 2000-06-16 20:51:26 +0000 | [diff] [blame] | 372 | if( z[1]!='|' ){ |
drh | bf4133c | 2001-10-13 02:59:08 +0000 | [diff] [blame] | 373 | *tokenType = TK_BITOR; |
drh | 0040077 | 2000-06-16 20:51:26 +0000 | [diff] [blame] | 374 | return 1; |
| 375 | }else{ |
| 376 | *tokenType = TK_CONCAT; |
| 377 | return 2; |
| 378 | } |
| 379 | } |
drh | 8974331 | 2016-02-08 02:30:50 +0000 | [diff] [blame] | 380 | case CC_COMMA: { |
drh | 7589723 | 2000-05-29 14:26:00 +0000 | [diff] [blame] | 381 | *tokenType = TK_COMMA; |
| 382 | return 1; |
| 383 | } |
drh | 8974331 | 2016-02-08 02:30:50 +0000 | [diff] [blame] | 384 | case CC_AND: { |
drh | bf4133c | 2001-10-13 02:59:08 +0000 | [diff] [blame] | 385 | *tokenType = TK_BITAND; |
| 386 | return 1; |
| 387 | } |
drh | 8974331 | 2016-02-08 02:30:50 +0000 | [diff] [blame] | 388 | case CC_TILDA: { |
drh | bf4133c | 2001-10-13 02:59:08 +0000 | [diff] [blame] | 389 | *tokenType = TK_BITNOT; |
| 390 | return 1; |
| 391 | } |
drh | 8974331 | 2016-02-08 02:30:50 +0000 | [diff] [blame] | 392 | case CC_QUOTE: { |
drh | 7589723 | 2000-05-29 14:26:00 +0000 | [diff] [blame] | 393 | int delim = z[0]; |
drh | 19f81f6 | 2009-06-09 18:01:37 +0000 | [diff] [blame] | 394 | testcase( delim=='`' ); |
| 395 | testcase( delim=='\'' ); |
| 396 | testcase( delim=='"' ); |
drh | aa756b0 | 2004-09-25 15:25:26 +0000 | [diff] [blame] | 397 | for(i=1; (c=z[i])!=0; i++){ |
| 398 | if( c==delim ){ |
drh | 7589723 | 2000-05-29 14:26:00 +0000 | [diff] [blame] | 399 | if( z[i+1]==delim ){ |
| 400 | i++; |
| 401 | }else{ |
| 402 | break; |
| 403 | } |
| 404 | } |
| 405 | } |
drh | a33cb5f | 2008-08-07 13:05:34 +0000 | [diff] [blame] | 406 | if( c=='\'' ){ |
drh | eef8b55 | 2005-10-23 11:29:40 +0000 | [diff] [blame] | 407 | *tokenType = TK_STRING; |
| 408 | return i+1; |
drh | a33cb5f | 2008-08-07 13:05:34 +0000 | [diff] [blame] | 409 | }else if( c!=0 ){ |
| 410 | *tokenType = TK_ID; |
| 411 | return i+1; |
drh | eef8b55 | 2005-10-23 11:29:40 +0000 | [diff] [blame] | 412 | }else{ |
| 413 | *tokenType = TK_ILLEGAL; |
| 414 | return i; |
| 415 | } |
drh | 7589723 | 2000-05-29 14:26:00 +0000 | [diff] [blame] | 416 | } |
drh | 8974331 | 2016-02-08 02:30:50 +0000 | [diff] [blame] | 417 | case CC_DOT: { |
drh | 7681618 | 2005-08-23 11:31:26 +0000 | [diff] [blame] | 418 | #ifndef SQLITE_OMIT_FLOATING_POINT |
danielk1977 | 78ca0e7 | 2009-01-20 16:53:39 +0000 | [diff] [blame] | 419 | if( !sqlite3Isdigit(z[1]) ) |
drh | 7681618 | 2005-08-23 11:31:26 +0000 | [diff] [blame] | 420 | #endif |
| 421 | { |
| 422 | *tokenType = TK_DOT; |
| 423 | return 1; |
| 424 | } |
| 425 | /* If the next character is a digit, this is a floating point |
| 426 | ** number that begins with ".". Fall thru into the next case */ |
drh | 08b9208 | 2020-08-10 14:18:00 +0000 | [diff] [blame] | 427 | /* no break */ deliberate_fall_through |
drh | 7589723 | 2000-05-29 14:26:00 +0000 | [diff] [blame] | 428 | } |
drh | 8974331 | 2016-02-08 02:30:50 +0000 | [diff] [blame] | 429 | case CC_DIGIT: { |
drh | 19f81f6 | 2009-06-09 18:01:37 +0000 | [diff] [blame] | 430 | testcase( z[0]=='0' ); testcase( z[0]=='1' ); testcase( z[0]=='2' ); |
| 431 | testcase( z[0]=='3' ); testcase( z[0]=='4' ); testcase( z[0]=='5' ); |
| 432 | testcase( z[0]=='6' ); testcase( z[0]=='7' ); testcase( z[0]=='8' ); |
| 433 | testcase( z[0]=='9' ); |
drh | c837e70 | 2000-06-08 16:26:24 +0000 | [diff] [blame] | 434 | *tokenType = TK_INTEGER; |
drh | 28e048c | 2014-07-23 01:26:51 +0000 | [diff] [blame] | 435 | #ifndef SQLITE_OMIT_HEX_INTEGER |
| 436 | if( z[0]=='0' && (z[1]=='x' || z[1]=='X') && sqlite3Isxdigit(z[2]) ){ |
| 437 | for(i=3; sqlite3Isxdigit(z[i]); i++){} |
| 438 | return i; |
| 439 | } |
| 440 | #endif |
danielk1977 | 78ca0e7 | 2009-01-20 16:53:39 +0000 | [diff] [blame] | 441 | for(i=0; sqlite3Isdigit(z[i]); i++){} |
drh | b7f9164 | 2004-10-31 02:22:47 +0000 | [diff] [blame] | 442 | #ifndef SQLITE_OMIT_FLOATING_POINT |
drh | 7681618 | 2005-08-23 11:31:26 +0000 | [diff] [blame] | 443 | if( z[i]=='.' ){ |
| 444 | i++; |
danielk1977 | 78ca0e7 | 2009-01-20 16:53:39 +0000 | [diff] [blame] | 445 | while( sqlite3Isdigit(z[i]) ){ i++; } |
drh | c837e70 | 2000-06-08 16:26:24 +0000 | [diff] [blame] | 446 | *tokenType = TK_FLOAT; |
| 447 | } |
| 448 | if( (z[i]=='e' || z[i]=='E') && |
danielk1977 | 78ca0e7 | 2009-01-20 16:53:39 +0000 | [diff] [blame] | 449 | ( sqlite3Isdigit(z[i+1]) |
| 450 | || ((z[i+1]=='+' || z[i+1]=='-') && sqlite3Isdigit(z[i+2])) |
drh | 7589723 | 2000-05-29 14:26:00 +0000 | [diff] [blame] | 451 | ) |
drh | c837e70 | 2000-06-08 16:26:24 +0000 | [diff] [blame] | 452 | ){ |
| 453 | i += 2; |
danielk1977 | 78ca0e7 | 2009-01-20 16:53:39 +0000 | [diff] [blame] | 454 | while( sqlite3Isdigit(z[i]) ){ i++; } |
drh | 7589723 | 2000-05-29 14:26:00 +0000 | [diff] [blame] | 455 | *tokenType = TK_FLOAT; |
drh | 7589723 | 2000-05-29 14:26:00 +0000 | [diff] [blame] | 456 | } |
drh | b7f9164 | 2004-10-31 02:22:47 +0000 | [diff] [blame] | 457 | #endif |
drh | 67dd901 | 2006-08-12 12:33:14 +0000 | [diff] [blame] | 458 | while( IdChar(z[i]) ){ |
| 459 | *tokenType = TK_ILLEGAL; |
| 460 | i++; |
| 461 | } |
drh | 7589723 | 2000-05-29 14:26:00 +0000 | [diff] [blame] | 462 | return i; |
| 463 | } |
drh | 8974331 | 2016-02-08 02:30:50 +0000 | [diff] [blame] | 464 | case CC_QUOTE2: { |
drh | aa756b0 | 2004-09-25 15:25:26 +0000 | [diff] [blame] | 465 | for(i=1, c=z[0]; c!=']' && (c=z[i])!=0; i++){} |
drh | 4b2f936 | 2008-01-22 23:37:09 +0000 | [diff] [blame] | 466 | *tokenType = c==']' ? TK_ID : TK_ILLEGAL; |
drh | 2f4392f | 2002-02-14 21:42:51 +0000 | [diff] [blame] | 467 | return i; |
| 468 | } |
drh | 8974331 | 2016-02-08 02:30:50 +0000 | [diff] [blame] | 469 | case CC_VARNUM: { |
drh | 5045789 | 2003-09-06 01:10:47 +0000 | [diff] [blame] | 470 | *tokenType = TK_VARIABLE; |
danielk1977 | 78ca0e7 | 2009-01-20 16:53:39 +0000 | [diff] [blame] | 471 | for(i=1; sqlite3Isdigit(z[i]); i++){} |
drh | fa6bc00 | 2004-09-07 16:19:52 +0000 | [diff] [blame] | 472 | return i; |
drh | 5045789 | 2003-09-06 01:10:47 +0000 | [diff] [blame] | 473 | } |
drh | 8974331 | 2016-02-08 02:30:50 +0000 | [diff] [blame] | 474 | case CC_DOLLAR: |
| 475 | case CC_VARALPHA: { |
drh | 288d37f | 2005-06-22 08:48:06 +0000 | [diff] [blame] | 476 | int n = 0; |
drh | f59b12f | 2014-01-11 03:54:05 +0000 | [diff] [blame] | 477 | testcase( z[0]=='$' ); testcase( z[0]=='@' ); |
| 478 | testcase( z[0]==':' ); testcase( z[0]=='#' ); |
drh | 9d74b4c | 2004-08-24 15:23:34 +0000 | [diff] [blame] | 479 | *tokenType = TK_VARIABLE; |
drh | 288d37f | 2005-06-22 08:48:06 +0000 | [diff] [blame] | 480 | for(i=1; (c=z[i])!=0; i++){ |
| 481 | if( IdChar(c) ){ |
| 482 | n++; |
| 483 | #ifndef SQLITE_OMIT_TCL_VARIABLE |
| 484 | }else if( c=='(' && n>0 ){ |
| 485 | do{ |
| 486 | i++; |
danielk1977 | 78ca0e7 | 2009-01-20 16:53:39 +0000 | [diff] [blame] | 487 | }while( (c=z[i])!=0 && !sqlite3Isspace(c) && c!=')' ); |
drh | 288d37f | 2005-06-22 08:48:06 +0000 | [diff] [blame] | 488 | if( c==')' ){ |
drh | 895d747 | 2004-08-20 16:02:39 +0000 | [diff] [blame] | 489 | i++; |
| 490 | }else{ |
drh | 288d37f | 2005-06-22 08:48:06 +0000 | [diff] [blame] | 491 | *tokenType = TK_ILLEGAL; |
drh | 895d747 | 2004-08-20 16:02:39 +0000 | [diff] [blame] | 492 | } |
drh | 288d37f | 2005-06-22 08:48:06 +0000 | [diff] [blame] | 493 | break; |
| 494 | }else if( c==':' && z[i+1]==':' ){ |
| 495 | i++; |
| 496 | #endif |
| 497 | }else{ |
| 498 | break; |
drh | 895d747 | 2004-08-20 16:02:39 +0000 | [diff] [blame] | 499 | } |
| 500 | } |
drh | 288d37f | 2005-06-22 08:48:06 +0000 | [diff] [blame] | 501 | if( n==0 ) *tokenType = TK_ILLEGAL; |
drh | 895d747 | 2004-08-20 16:02:39 +0000 | [diff] [blame] | 502 | return i; |
drh | b7f9164 | 2004-10-31 02:22:47 +0000 | [diff] [blame] | 503 | } |
drh | d1032f9 | 2020-11-27 20:56:16 +0000 | [diff] [blame] | 504 | case CC_KYWD0: { |
drh | e96f361 | 2016-02-08 19:36:46 +0000 | [diff] [blame] | 505 | for(i=1; aiClass[z[i]]<=CC_KYWD; i++){} |
| 506 | if( IdChar(z[i]) ){ |
| 507 | /* This token started out using characters that can appear in keywords, |
| 508 | ** but z[i] is a character not allowed within keywords, so this must |
| 509 | ** be an identifier instead */ |
| 510 | i++; |
| 511 | break; |
| 512 | } |
| 513 | *tokenType = TK_ID; |
dan | 769309f | 2018-06-29 19:54:51 +0000 | [diff] [blame] | 514 | return keywordCode((char*)z, i, tokenType); |
drh | e96f361 | 2016-02-08 19:36:46 +0000 | [diff] [blame] | 515 | } |
drh | 8974331 | 2016-02-08 02:30:50 +0000 | [diff] [blame] | 516 | case CC_X: { |
dan | a73086d | 2016-02-24 16:14:07 +0000 | [diff] [blame] | 517 | #ifndef SQLITE_OMIT_BLOB_LITERAL |
drh | 19f81f6 | 2009-06-09 18:01:37 +0000 | [diff] [blame] | 518 | testcase( z[0]=='x' ); testcase( z[0]=='X' ); |
drh | 4b2f936 | 2008-01-22 23:37:09 +0000 | [diff] [blame] | 519 | if( z[1]=='\'' ){ |
danielk1977 | 3fd0a73 | 2004-05-27 13:35:19 +0000 | [diff] [blame] | 520 | *tokenType = TK_BLOB; |
drh | 3a61a5a | 2011-06-03 21:34:45 +0000 | [diff] [blame] | 521 | for(i=2; sqlite3Isxdigit(z[i]); i++){} |
| 522 | if( z[i]!='\'' || i%2 ){ |
| 523 | *tokenType = TK_ILLEGAL; |
| 524 | while( z[i] && z[i]!='\'' ){ i++; } |
danielk1977 | c572ef7 | 2004-05-27 09:28:41 +0000 | [diff] [blame] | 525 | } |
drh | 3a61a5a | 2011-06-03 21:34:45 +0000 | [diff] [blame] | 526 | if( z[i] ) i++; |
danielk1977 | c572ef7 | 2004-05-27 09:28:41 +0000 | [diff] [blame] | 527 | return i; |
| 528 | } |
dan | a73086d | 2016-02-24 16:14:07 +0000 | [diff] [blame] | 529 | #endif |
drh | e96f361 | 2016-02-08 19:36:46 +0000 | [diff] [blame] | 530 | /* If it is not a BLOB literal, then it must be an ID, since no |
| 531 | ** SQL keywords start with the letter 'x'. Fall through */ |
drh | 08b9208 | 2020-08-10 14:18:00 +0000 | [diff] [blame] | 532 | /* no break */ deliberate_fall_through |
danielk1977 | c572ef7 | 2004-05-27 09:28:41 +0000 | [diff] [blame] | 533 | } |
drh | d1032f9 | 2020-11-27 20:56:16 +0000 | [diff] [blame] | 534 | case CC_KYWD: |
drh | 8974331 | 2016-02-08 02:30:50 +0000 | [diff] [blame] | 535 | case CC_ID: { |
| 536 | i = 1; |
| 537 | break; |
| 538 | } |
drh | ba9ebc2 | 2021-04-24 12:24:08 +0000 | [diff] [blame] | 539 | case CC_BOM: { |
| 540 | if( z[1]==0xbb && z[2]==0xbf ){ |
| 541 | *tokenType = TK_SPACE; |
| 542 | return 3; |
| 543 | } |
| 544 | i = 1; |
| 545 | break; |
| 546 | } |
dan | 2b5f152 | 2018-06-30 20:00:35 +0000 | [diff] [blame] | 547 | case CC_NUL: { |
| 548 | *tokenType = TK_ILLEGAL; |
| 549 | return 0; |
| 550 | } |
drh | 98808ba | 2001-10-18 12:34:46 +0000 | [diff] [blame] | 551 | default: { |
drh | 8974331 | 2016-02-08 02:30:50 +0000 | [diff] [blame] | 552 | *tokenType = TK_ILLEGAL; |
| 553 | return 1; |
drh | 7589723 | 2000-05-29 14:26:00 +0000 | [diff] [blame] | 554 | } |
drh | 7589723 | 2000-05-29 14:26:00 +0000 | [diff] [blame] | 555 | } |
drh | 34dcee6 | 2016-02-08 19:15:48 +0000 | [diff] [blame] | 556 | while( IdChar(z[i]) ){ i++; } |
drh | 8974331 | 2016-02-08 02:30:50 +0000 | [diff] [blame] | 557 | *tokenType = TK_ID; |
| 558 | return i; |
drh | 7589723 | 2000-05-29 14:26:00 +0000 | [diff] [blame] | 559 | } |
| 560 | |
| 561 | /* |
| 562 | ** Run the parser on the given SQL string. The parser structure is |
drh | b19a2bc | 2001-09-16 00:13:26 +0000 | [diff] [blame] | 563 | ** passed in. An SQLITE_ status code is returned. If an error occurs |
drh | fb45d8c | 2008-07-08 00:06:49 +0000 | [diff] [blame] | 564 | ** then an and attempt is made to write an error message into |
| 565 | ** memory obtained from sqlite3_malloc() and to make *pzErrMsg point to that |
| 566 | ** error message. |
drh | 7589723 | 2000-05-29 14:26:00 +0000 | [diff] [blame] | 567 | */ |
danielk1977 | 4adee20 | 2004-05-08 08:23:19 +0000 | [diff] [blame] | 568 | int sqlite3RunParser(Parse *pParse, const char *zSql, char **pzErrMsg){ |
drh | d9da78a | 2009-03-24 15:08:09 +0000 | [diff] [blame] | 569 | int nErr = 0; /* Number of errors encountered */ |
drh | d9da78a | 2009-03-24 15:08:09 +0000 | [diff] [blame] | 570 | void *pEngine; /* The LEMON-generated LALR(1) parser */ |
drh | 9b38d66 | 2017-03-07 14:38:52 +0000 | [diff] [blame] | 571 | int n = 0; /* Length of the next token token */ |
drh | d9da78a | 2009-03-24 15:08:09 +0000 | [diff] [blame] | 572 | int tokenType; /* type of the next token */ |
| 573 | int lastTokenParsed = -1; /* type of the previous token */ |
drh | d9da78a | 2009-03-24 15:08:09 +0000 | [diff] [blame] | 574 | sqlite3 *db = pParse->db; /* The database connection */ |
| 575 | int mxSqlLen; /* Max length of an SQL string */ |
dan | 488b558 | 2021-11-16 13:36:50 +0000 | [diff] [blame] | 576 | Parse *pParentParse = 0; /* Outer parse context, if any */ |
drh | d26cc54 | 2017-01-28 20:46:37 +0000 | [diff] [blame] | 577 | #ifdef sqlite3Parser_ENGINEALWAYSONSTACK |
drh | 53b2459 | 2017-03-30 17:13:37 +0000 | [diff] [blame] | 578 | yyParser sEngine; /* Space to hold the Lemon-generated Parser object */ |
drh | d26cc54 | 2017-01-28 20:46:37 +0000 | [diff] [blame] | 579 | #endif |
drh | 2b454e0 | 2019-02-26 16:11:46 +0000 | [diff] [blame] | 580 | VVA_ONLY( u8 startedWithOom = db->mallocFailed ); |
drh | 7589723 | 2000-05-29 14:26:00 +0000 | [diff] [blame] | 581 | |
drh | 96c707a | 2015-02-13 16:36:14 +0000 | [diff] [blame] | 582 | assert( zSql!=0 ); |
drh | d9da78a | 2009-03-24 15:08:09 +0000 | [diff] [blame] | 583 | mxSqlLen = db->aLimit[SQLITE_LIMIT_SQL_LENGTH]; |
drh | 4f7d3a5 | 2013-06-27 23:54:02 +0000 | [diff] [blame] | 584 | if( db->nVdbeActive==0 ){ |
dan | 892edb6 | 2020-03-30 13:35:05 +0000 | [diff] [blame] | 585 | AtomicStore(&db->u1.isInterrupted, 0); |
drh | 15ca1df | 2006-07-26 13:43:30 +0000 | [diff] [blame] | 586 | } |
drh | 4c50439 | 2000-10-16 22:06:40 +0000 | [diff] [blame] | 587 | pParse->rc = SQLITE_OK; |
drh | e7f3f3e | 2009-07-03 22:54:36 +0000 | [diff] [blame] | 588 | pParse->zTail = zSql; |
drh | fb45d8c | 2008-07-08 00:06:49 +0000 | [diff] [blame] | 589 | assert( pzErrMsg!=0 ); |
drh | 9b74706 | 2019-01-31 01:39:01 +0000 | [diff] [blame] | 590 | #ifdef SQLITE_DEBUG |
| 591 | if( db->flags & SQLITE_ParserTrace ){ |
| 592 | printf("parser: [[[%s]]]\n", zSql); |
| 593 | sqlite3ParserTrace(stdout, "parser: "); |
| 594 | }else{ |
| 595 | sqlite3ParserTrace(0, 0); |
| 596 | } |
| 597 | #endif |
drh | d26cc54 | 2017-01-28 20:46:37 +0000 | [diff] [blame] | 598 | #ifdef sqlite3Parser_ENGINEALWAYSONSTACK |
drh | 53b2459 | 2017-03-30 17:13:37 +0000 | [diff] [blame] | 599 | pEngine = &sEngine; |
drh | fb32c44 | 2018-04-21 13:51:42 +0000 | [diff] [blame] | 600 | sqlite3ParserInit(pEngine, pParse); |
drh | d26cc54 | 2017-01-28 20:46:37 +0000 | [diff] [blame] | 601 | #else |
drh | fb32c44 | 2018-04-21 13:51:42 +0000 | [diff] [blame] | 602 | pEngine = sqlite3ParserAlloc(sqlite3Malloc, pParse); |
drh | 7589723 | 2000-05-29 14:26:00 +0000 | [diff] [blame] | 603 | if( pEngine==0 ){ |
drh | 4a642b6 | 2016-02-05 01:55:27 +0000 | [diff] [blame] | 604 | sqlite3OomFault(db); |
mistachkin | fad3039 | 2016-02-13 23:43:46 +0000 | [diff] [blame] | 605 | return SQLITE_NOMEM_BKPT; |
drh | 7589723 | 2000-05-29 14:26:00 +0000 | [diff] [blame] | 606 | } |
drh | d26cc54 | 2017-01-28 20:46:37 +0000 | [diff] [blame] | 607 | #endif |
drh | fa6bc00 | 2004-09-07 16:19:52 +0000 | [diff] [blame] | 608 | assert( pParse->pNewTable==0 ); |
| 609 | assert( pParse->pNewTrigger==0 ); |
| 610 | assert( pParse->nVar==0 ); |
drh | 9bf755c | 2016-12-23 03:59:31 +0000 | [diff] [blame] | 611 | assert( pParse->pVList==0 ); |
dan | 488b558 | 2021-11-16 13:36:50 +0000 | [diff] [blame] | 612 | pParentParse = db->pParse; |
drh | 1cf1975 | 2019-02-08 14:55:30 +0000 | [diff] [blame] | 613 | db->pParse = pParse; |
drh | 167fbbe | 2016-08-10 14:40:00 +0000 | [diff] [blame] | 614 | while( 1 ){ |
dan | d437ac0 | 2018-06-29 20:21:24 +0000 | [diff] [blame] | 615 | n = sqlite3GetToken((u8*)zSql, &tokenType); |
| 616 | mxSqlLen -= n; |
| 617 | if( mxSqlLen<0 ){ |
| 618 | pParse->rc = SQLITE_TOOBIG; |
| 619 | break; |
drh | e5c941b | 2007-05-08 13:58:26 +0000 | [diff] [blame] | 620 | } |
dan | 34a7d79 | 2018-06-29 20:43:33 +0000 | [diff] [blame] | 621 | #ifndef SQLITE_OMIT_WINDOWFUNC |
| 622 | if( tokenType>=TK_WINDOW ){ |
dan | 6e2210e | 2018-06-30 18:54:56 +0000 | [diff] [blame] | 623 | assert( tokenType==TK_SPACE || tokenType==TK_OVER || tokenType==TK_FILTER |
dan | 34a7d79 | 2018-06-29 20:43:33 +0000 | [diff] [blame] | 624 | || tokenType==TK_ILLEGAL || tokenType==TK_WINDOW |
| 625 | ); |
| 626 | #else |
drh | 0739e45 | 2015-11-09 02:08:09 +0000 | [diff] [blame] | 627 | if( tokenType>=TK_SPACE ){ |
| 628 | assert( tokenType==TK_SPACE || tokenType==TK_ILLEGAL ); |
drh | c7bf571 | 2018-07-09 22:49:01 +0000 | [diff] [blame] | 629 | #endif /* SQLITE_OMIT_WINDOWFUNC */ |
dan | 892edb6 | 2020-03-30 13:35:05 +0000 | [diff] [blame] | 630 | if( AtomicLoad(&db->u1.isInterrupted) ){ |
drh | 0739e45 | 2015-11-09 02:08:09 +0000 | [diff] [blame] | 631 | pParse->rc = SQLITE_INTERRUPT; |
drh | 7589723 | 2000-05-29 14:26:00 +0000 | [diff] [blame] | 632 | break; |
| 633 | } |
dan | d437ac0 | 2018-06-29 20:21:24 +0000 | [diff] [blame] | 634 | if( tokenType==TK_SPACE ){ |
| 635 | zSql += n; |
| 636 | continue; |
| 637 | } |
| 638 | if( zSql[0]==0 ){ |
| 639 | /* Upon reaching the end of input, call the parser two more times |
| 640 | ** with tokens TK_SEMI and 0, in that order. */ |
| 641 | if( lastTokenParsed==TK_SEMI ){ |
| 642 | tokenType = 0; |
| 643 | }else if( lastTokenParsed==0 ){ |
| 644 | break; |
| 645 | }else{ |
| 646 | tokenType = TK_SEMI; |
| 647 | } |
| 648 | n = 0; |
dan | 34a7d79 | 2018-06-29 20:43:33 +0000 | [diff] [blame] | 649 | #ifndef SQLITE_OMIT_WINDOWFUNC |
| 650 | }else if( tokenType==TK_WINDOW ){ |
dan | 6e2210e | 2018-06-30 18:54:56 +0000 | [diff] [blame] | 651 | assert( n==6 ); |
dan | 34a7d79 | 2018-06-29 20:43:33 +0000 | [diff] [blame] | 652 | tokenType = analyzeWindowKeyword((const u8*)&zSql[6]); |
dan | 6e2210e | 2018-06-30 18:54:56 +0000 | [diff] [blame] | 653 | }else if( tokenType==TK_OVER ){ |
| 654 | assert( n==4 ); |
| 655 | tokenType = analyzeOverKeyword((const u8*)&zSql[4], lastTokenParsed); |
| 656 | }else if( tokenType==TK_FILTER ){ |
| 657 | assert( n==6 ); |
| 658 | tokenType = analyzeFilterKeyword((const u8*)&zSql[6], lastTokenParsed); |
drh | c7bf571 | 2018-07-09 22:49:01 +0000 | [diff] [blame] | 659 | #endif /* SQLITE_OMIT_WINDOWFUNC */ |
dan | d437ac0 | 2018-06-29 20:21:24 +0000 | [diff] [blame] | 660 | }else{ |
drh | 9b38d66 | 2017-03-07 14:38:52 +0000 | [diff] [blame] | 661 | sqlite3ErrorMsg(pParse, "unrecognized token: \"%.*s\"", n, zSql); |
drh | 7589723 | 2000-05-29 14:26:00 +0000 | [diff] [blame] | 662 | break; |
drh | 32eb7b4 | 2003-01-07 01:44:37 +0000 | [diff] [blame] | 663 | } |
drh | 7589723 | 2000-05-29 14:26:00 +0000 | [diff] [blame] | 664 | } |
dan | d437ac0 | 2018-06-29 20:21:24 +0000 | [diff] [blame] | 665 | pParse->sLastToken.z = zSql; |
| 666 | pParse->sLastToken.n = n; |
| 667 | sqlite3Parser(pEngine, tokenType, pParse->sLastToken); |
| 668 | lastTokenParsed = tokenType; |
| 669 | zSql += n; |
drh | 1cf1975 | 2019-02-08 14:55:30 +0000 | [diff] [blame] | 670 | assert( db->mallocFailed==0 || pParse->rc!=SQLITE_OK || startedWithOom ); |
| 671 | if( pParse->rc!=SQLITE_OK ) break; |
drh | 7589723 | 2000-05-29 14:26:00 +0000 | [diff] [blame] | 672 | } |
drh | 0be0cf6 | 2015-04-15 08:37:42 +0000 | [diff] [blame] | 673 | assert( nErr==0 ); |
drh | ec424a5 | 2008-07-25 15:39:03 +0000 | [diff] [blame] | 674 | #ifdef YYTRACKMAXSTACKDEPTH |
drh | af89fe6 | 2015-03-23 17:25:18 +0000 | [diff] [blame] | 675 | sqlite3_mutex_enter(sqlite3MallocMutex()); |
drh | b02392e | 2015-10-15 15:28:56 +0000 | [diff] [blame] | 676 | sqlite3StatusHighwater(SQLITE_STATUS_PARSER_STACK, |
drh | ec424a5 | 2008-07-25 15:39:03 +0000 | [diff] [blame] | 677 | sqlite3ParserStackPeak(pEngine) |
| 678 | ); |
drh | af89fe6 | 2015-03-23 17:25:18 +0000 | [diff] [blame] | 679 | sqlite3_mutex_leave(sqlite3MallocMutex()); |
drh | ec424a5 | 2008-07-25 15:39:03 +0000 | [diff] [blame] | 680 | #endif /* YYDEBUG */ |
drh | d26cc54 | 2017-01-28 20:46:37 +0000 | [diff] [blame] | 681 | #ifdef sqlite3Parser_ENGINEALWAYSONSTACK |
| 682 | sqlite3ParserFinalize(pEngine); |
| 683 | #else |
drh | 1743575 | 2007-08-16 04:30:38 +0000 | [diff] [blame] | 684 | sqlite3ParserFree(pEngine, sqlite3_free); |
drh | d26cc54 | 2017-01-28 20:46:37 +0000 | [diff] [blame] | 685 | #endif |
drh | 1743575 | 2007-08-16 04:30:38 +0000 | [diff] [blame] | 686 | if( db->mallocFailed ){ |
mistachkin | fad3039 | 2016-02-13 23:43:46 +0000 | [diff] [blame] | 687 | pParse->rc = SQLITE_NOMEM_BKPT; |
drh | 71c697e | 2004-08-08 23:39:19 +0000 | [diff] [blame] | 688 | } |
drh | b86ccfb | 2003-01-28 23:13:10 +0000 | [diff] [blame] | 689 | if( pParse->rc!=SQLITE_OK && pParse->rc!=SQLITE_DONE && pParse->zErrMsg==0 ){ |
drh | 22c17b8 | 2015-05-15 04:13:15 +0000 | [diff] [blame] | 690 | pParse->zErrMsg = sqlite3MPrintf(db, "%s", sqlite3ErrStr(pParse->rc)); |
drh | b86ccfb | 2003-01-28 23:13:10 +0000 | [diff] [blame] | 691 | } |
drh | 19f81f6 | 2009-06-09 18:01:37 +0000 | [diff] [blame] | 692 | assert( pzErrMsg!=0 ); |
drh | 7589723 | 2000-05-29 14:26:00 +0000 | [diff] [blame] | 693 | if( pParse->zErrMsg ){ |
drh | 19f81f6 | 2009-06-09 18:01:37 +0000 | [diff] [blame] | 694 | *pzErrMsg = pParse->zErrMsg; |
drh | 875ad8c | 2018-06-21 23:53:54 +0000 | [diff] [blame] | 695 | sqlite3_log(pParse->rc, "%s in \"%s\"", |
| 696 | *pzErrMsg, pParse->zTail); |
drh | b86ccfb | 2003-01-28 23:13:10 +0000 | [diff] [blame] | 697 | pParse->zErrMsg = 0; |
drh | 4b2f936 | 2008-01-22 23:37:09 +0000 | [diff] [blame] | 698 | nErr++; |
drh | 7589723 | 2000-05-29 14:26:00 +0000 | [diff] [blame] | 699 | } |
drh | 875ad8c | 2018-06-21 23:53:54 +0000 | [diff] [blame] | 700 | pParse->zTail = zSql; |
drh | 2958a4e | 2004-11-12 03:56:15 +0000 | [diff] [blame] | 701 | if( pParse->pVdbe && pParse->nErr>0 && pParse->nested==0 ){ |
danielk1977 | 4adee20 | 2004-05-08 08:23:19 +0000 | [diff] [blame] | 702 | sqlite3VdbeDelete(pParse->pVdbe); |
drh | 7589723 | 2000-05-29 14:26:00 +0000 | [diff] [blame] | 703 | pParse->pVdbe = 0; |
| 704 | } |
danielk1977 | c00da10 | 2006-01-07 13:21:04 +0000 | [diff] [blame] | 705 | #ifndef SQLITE_OMIT_SHARED_CACHE |
| 706 | if( pParse->nested==0 ){ |
drh | 633e6d5 | 2008-07-28 19:34:53 +0000 | [diff] [blame] | 707 | sqlite3DbFree(db, pParse->aTableLock); |
danielk1977 | c00da10 | 2006-01-07 13:21:04 +0000 | [diff] [blame] | 708 | pParse->aTableLock = 0; |
| 709 | pParse->nTableLock = 0; |
| 710 | } |
| 711 | #endif |
drh | 4f3dd15 | 2008-04-28 18:46:43 +0000 | [diff] [blame] | 712 | #ifndef SQLITE_OMIT_VIRTUALTABLE |
drh | b975598 | 2010-07-24 16:34:37 +0000 | [diff] [blame] | 713 | sqlite3_free(pParse->apVtabLock); |
drh | 4f3dd15 | 2008-04-28 18:46:43 +0000 | [diff] [blame] | 714 | #endif |
danielk1977 | 7e6ebfb | 2006-06-12 11:24:37 +0000 | [diff] [blame] | 715 | |
dan | cf8f289 | 2018-08-09 20:47:01 +0000 | [diff] [blame] | 716 | if( !IN_SPECIAL_PARSE ){ |
danielk1977 | 7e6ebfb | 2006-06-12 11:24:37 +0000 | [diff] [blame] | 717 | /* If the pParse->declareVtab flag is set, do not delete any table |
| 718 | ** structure built up in pParse->pNewTable. The calling code (see vtab.c) |
| 719 | ** will take responsibility for freeing the Table structure. |
| 720 | */ |
dan | 1feeaed | 2010-07-23 15:41:47 +0000 | [diff] [blame] | 721 | sqlite3DeleteTable(db, pParse->pNewTable); |
danielk1977 | 7e6ebfb | 2006-06-12 11:24:37 +0000 | [diff] [blame] | 722 | } |
dan | c9461ec | 2018-08-29 21:00:16 +0000 | [diff] [blame] | 723 | if( !IN_RENAME_OBJECT ){ |
dan | 5496d6a | 2018-08-13 17:14:26 +0000 | [diff] [blame] | 724 | sqlite3DeleteTrigger(db, pParse->pNewTrigger); |
| 725 | } |
drh | 9bf755c | 2016-12-23 03:59:31 +0000 | [diff] [blame] | 726 | sqlite3DbFree(db, pParse->pVList); |
dan | 488b558 | 2021-11-16 13:36:50 +0000 | [diff] [blame] | 727 | db->pParse = pParentParse; |
drh | f3151f0 | 2015-04-16 20:27:09 +0000 | [diff] [blame] | 728 | assert( nErr==0 || pParse->rc!=SQLITE_OK ); |
drh | 7589723 | 2000-05-29 14:26:00 +0000 | [diff] [blame] | 729 | return nErr; |
| 730 | } |
drh | 643d855 | 2018-12-10 16:00:57 +0000 | [diff] [blame] | 731 | |
| 732 | |
| 733 | #ifdef SQLITE_ENABLE_NORMALIZE |
| 734 | /* |
| 735 | ** Insert a single space character into pStr if the current string |
| 736 | ** ends with an identifier |
| 737 | */ |
| 738 | static void addSpaceSeparator(sqlite3_str *pStr){ |
| 739 | if( pStr->nChar && sqlite3IsIdChar(pStr->zText[pStr->nChar-1]) ){ |
| 740 | sqlite3_str_append(pStr, " ", 1); |
| 741 | } |
| 742 | } |
| 743 | |
| 744 | /* |
| 745 | ** Compute a normalization of the SQL given by zSql[0..nSql-1]. Return |
| 746 | ** the normalization in space obtained from sqlite3DbMalloc(). Or return |
| 747 | ** NULL if anything goes wrong or if zSql is NULL. |
| 748 | */ |
| 749 | char *sqlite3Normalize( |
| 750 | Vdbe *pVdbe, /* VM being reprepared */ |
drh | 1a6c2b1 | 2018-12-10 20:01:40 +0000 | [diff] [blame] | 751 | const char *zSql /* The original SQL string */ |
drh | 643d855 | 2018-12-10 16:00:57 +0000 | [diff] [blame] | 752 | ){ |
| 753 | sqlite3 *db; /* The database connection */ |
| 754 | int i; /* Next unread byte of zSql[] */ |
| 755 | int n; /* length of current token */ |
| 756 | int tokenType; /* type of current token */ |
mistachkin | 844b900 | 2019-02-02 01:27:45 +0000 | [diff] [blame] | 757 | int prevType = 0; /* Previous non-whitespace token */ |
drh | 643d855 | 2018-12-10 16:00:57 +0000 | [diff] [blame] | 758 | int nParen; /* Number of nested levels of parentheses */ |
| 759 | int iStartIN; /* Start of RHS of IN operator in z[] */ |
| 760 | int nParenAtIN; /* Value of nParent at start of RHS of IN operator */ |
mistachkin | 16fd04c | 2019-10-16 17:46:22 +0000 | [diff] [blame] | 761 | u32 j; /* Bytes of normalized SQL generated so far */ |
drh | 643d855 | 2018-12-10 16:00:57 +0000 | [diff] [blame] | 762 | sqlite3_str *pStr; /* The normalized SQL string under construction */ |
| 763 | |
drh | 643d855 | 2018-12-10 16:00:57 +0000 | [diff] [blame] | 764 | db = sqlite3VdbeDb(pVdbe); |
| 765 | tokenType = -1; |
| 766 | nParen = iStartIN = nParenAtIN = 0; |
| 767 | pStr = sqlite3_str_new(db); |
drh | 1a6c2b1 | 2018-12-10 20:01:40 +0000 | [diff] [blame] | 768 | assert( pStr!=0 ); /* sqlite3_str_new() never returns NULL */ |
| 769 | for(i=0; zSql[i] && pStr->accError==0; i+=n){ |
drh | 643d855 | 2018-12-10 16:00:57 +0000 | [diff] [blame] | 770 | if( tokenType!=TK_SPACE ){ |
| 771 | prevType = tokenType; |
| 772 | } |
| 773 | n = sqlite3GetToken((unsigned char*)zSql+i, &tokenType); |
| 774 | if( NEVER(n<=0) ) break; |
| 775 | switch( tokenType ){ |
| 776 | case TK_SPACE: { |
| 777 | break; |
| 778 | } |
| 779 | case TK_NULL: { |
| 780 | if( prevType==TK_IS || prevType==TK_NOT ){ |
| 781 | sqlite3_str_append(pStr, " NULL", 5); |
| 782 | break; |
| 783 | } |
| 784 | /* Fall through */ |
| 785 | } |
| 786 | case TK_STRING: |
| 787 | case TK_INTEGER: |
| 788 | case TK_FLOAT: |
| 789 | case TK_VARIABLE: |
| 790 | case TK_BLOB: { |
| 791 | sqlite3_str_append(pStr, "?", 1); |
| 792 | break; |
| 793 | } |
| 794 | case TK_LP: { |
| 795 | nParen++; |
| 796 | if( prevType==TK_IN ){ |
| 797 | iStartIN = pStr->nChar; |
| 798 | nParenAtIN = nParen; |
| 799 | } |
| 800 | sqlite3_str_append(pStr, "(", 1); |
| 801 | break; |
| 802 | } |
| 803 | case TK_RP: { |
| 804 | if( iStartIN>0 && nParen==nParenAtIN ){ |
mistachkin | 16fd04c | 2019-10-16 17:46:22 +0000 | [diff] [blame] | 805 | assert( pStr->nChar>=(u32)iStartIN ); |
drh | 643d855 | 2018-12-10 16:00:57 +0000 | [diff] [blame] | 806 | pStr->nChar = iStartIN+1; |
| 807 | sqlite3_str_append(pStr, "?,?,?", 5); |
| 808 | iStartIN = 0; |
| 809 | } |
| 810 | nParen--; |
| 811 | sqlite3_str_append(pStr, ")", 1); |
| 812 | break; |
| 813 | } |
| 814 | case TK_ID: { |
drh | 9042ff2 | 2018-12-10 16:49:33 +0000 | [diff] [blame] | 815 | iStartIN = 0; |
drh | 643d855 | 2018-12-10 16:00:57 +0000 | [diff] [blame] | 816 | j = pStr->nChar; |
| 817 | if( sqlite3Isquote(zSql[i]) ){ |
| 818 | char *zId = sqlite3DbStrNDup(db, zSql+i, n); |
| 819 | int nId; |
| 820 | int eType = 0; |
| 821 | if( zId==0 ) break; |
| 822 | sqlite3Dequote(zId); |
| 823 | if( zSql[i]=='"' && sqlite3VdbeUsesDoubleQuotedString(pVdbe, zId) ){ |
| 824 | sqlite3_str_append(pStr, "?", 1); |
| 825 | sqlite3DbFree(db, zId); |
| 826 | break; |
| 827 | } |
| 828 | nId = sqlite3Strlen30(zId); |
| 829 | if( sqlite3GetToken((u8*)zId, &eType)==nId && eType==TK_ID ){ |
| 830 | addSpaceSeparator(pStr); |
| 831 | sqlite3_str_append(pStr, zId, nId); |
| 832 | }else{ |
| 833 | sqlite3_str_appendf(pStr, "\"%w\"", zId); |
| 834 | } |
| 835 | sqlite3DbFree(db, zId); |
| 836 | }else{ |
| 837 | addSpaceSeparator(pStr); |
| 838 | sqlite3_str_append(pStr, zSql+i, n); |
| 839 | } |
| 840 | while( j<pStr->nChar ){ |
| 841 | pStr->zText[j] = sqlite3Tolower(pStr->zText[j]); |
| 842 | j++; |
| 843 | } |
| 844 | break; |
| 845 | } |
drh | 9042ff2 | 2018-12-10 16:49:33 +0000 | [diff] [blame] | 846 | case TK_SELECT: { |
| 847 | iStartIN = 0; |
| 848 | /* fall through */ |
| 849 | } |
drh | 643d855 | 2018-12-10 16:00:57 +0000 | [diff] [blame] | 850 | default: { |
| 851 | if( sqlite3IsIdChar(zSql[i]) ) addSpaceSeparator(pStr); |
| 852 | j = pStr->nChar; |
| 853 | sqlite3_str_append(pStr, zSql+i, n); |
| 854 | while( j<pStr->nChar ){ |
| 855 | pStr->zText[j] = sqlite3Toupper(pStr->zText[j]); |
| 856 | j++; |
| 857 | } |
| 858 | break; |
| 859 | } |
| 860 | } |
| 861 | } |
drh | 9042ff2 | 2018-12-10 16:49:33 +0000 | [diff] [blame] | 862 | if( tokenType!=TK_SEMI ) sqlite3_str_append(pStr, ";", 1); |
drh | 643d855 | 2018-12-10 16:00:57 +0000 | [diff] [blame] | 863 | return sqlite3_str_finish(pStr); |
| 864 | } |
| 865 | #endif /* SQLITE_ENABLE_NORMALIZE */ |