dlls/msi/tokenize.c

   1 /*
   2 ** 2001 September 15
   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 ** 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.
  17 */
  18
  19 #include <ctype.h>
  20 #include <stdarg.h>
  21 #include <stdlib.h>
  22
  23 #include "windef.h"
  24 #include "winbase.h"
  25 #include "wine/debug.h"
  26 #include "winnls.h"
  27 #include "query.h"
  28 #include "sql.tab.h"
  29
  30 WINE_DEFAULT_DEBUG_CHANNEL(msi);
  31
  32 /*
  33 ** All the keywords of the SQL language are stored as in a hash
  34 ** table composed of instances of the following structure.
  35 */
  36 typedef struct Keyword Keyword;
  37 struct Keyword {
  38   const char *zName;             /* The keyword name */
  39   int tokenType;           /* The token value for this keyword */
  40 };
  41
  42 /*
  43 ** These are the keywords
  44 */
  45 static const Keyword aKeywordTable[] = {
  46   { "ABORT", TK_ABORT },
  47   { "AFTER", TK_AFTER },
  48   { "ALL", TK_ALL },
  49   { "AND", TK_AND },
  50   { "AS", TK_AS },
  51   { "ASC", TK_ASC },
  52   { "BEFORE", TK_BEFORE },
  53   { "BEGIN", TK_BEGIN },
  54   { "BETWEEN", TK_BETWEEN },
  55   { "BY", TK_BY },
  56   { "CASCADE", TK_CASCADE },
  57   { "CASE", TK_CASE },
  58   { "CHAR", TK_CHAR },
  59   { "CHARACTER", TK_CHAR },
  60   { "CHECK", TK_CHECK },
  61   { "CLUSTER", TK_CLUSTER },
  62   { "COLLATE", TK_COLLATE },
  63   { "COMMIT", TK_COMMIT },
  64   { "CONFLICT", TK_CONFLICT },
  65   { "CONSTRAINT", TK_CONSTRAINT },
  66   { "COPY", TK_COPY },
  67   { "CREATE", TK_CREATE },
  68   { "CROSS", TK_JOIN_KW },
  69   { "DEFAULT", TK_DEFAULT },
  70   { "DEFERRED", TK_DEFERRED },
  71   { "DEFERRABLE", TK_DEFERRABLE },
  72   { "DELETE", TK_DELETE },
  73   { "DELIMITERS", TK_DELIMITERS },
  74   { "DESC", TK_DESC },
  75   { "DISTINCT", TK_DISTINCT },
  76   { "DROP", TK_DROP },
  77   { "END", TK_END },
  78   { "EACH", TK_EACH },
  79   { "ELSE", TK_ELSE },
  80   { "EXCEPT", TK_EXCEPT },
  81   { "EXPLAIN", TK_EXPLAIN },
  82   { "FAIL", TK_FAIL },
  83   { "FOR", TK_FOR },
  84   { "FOREIGN", TK_FOREIGN },
  85   { "FROM", TK_FROM },
  86   { "FULL", TK_JOIN_KW },
  87   { "GLOB", TK_GLOB },
  88   { "GROUP", TK_GROUP },
  89   { "HAVING", TK_HAVING },
  90   { "HOLD", TK_HOLD },
  91   { "IGNORE", TK_IGNORE },
  92   { "IMMEDIATE", TK_IMMEDIATE },
  93   { "IN", TK_IN },
  94   { "INDEX", TK_INDEX },
  95   { "INITIALLY", TK_INITIALLY },
  96   { "INNER", TK_JOIN_KW },
  97   { "INSERT", TK_INSERT },
  98   { "INSTEAD", TK_INSTEAD },
  99   { "INT", TK_INT },
 100   { "INTERSECT", TK_INTERSECT },
 101   { "INTO", TK_INTO },
 102   { "IS", TK_IS },
 103   { "ISNULL", TK_ISNULL },
 104   { "JOIN", TK_JOIN },
 105   { "KEY", TK_KEY },
 106   { "LEFT", TK_JOIN_KW },
 107   { "LIKE", TK_LIKE },
 108   { "LIMIT", TK_LIMIT },
 109   { "LOCALIZABLE", TK_LOCALIZABLE },
 110   { "LONG", TK_LONG },
 111   { "LONGCHAR", TK_LONGCHAR },
 112   { "MATCH", TK_MATCH },
 113   { "NATURAL", TK_JOIN_KW },
 114   { "NOT", TK_NOT },
 115   { "NOTNULL", TK_NOTNULL },
 116   { "NULL", TK_NULL },
 117   { "OBJECT", TK_OBJECT },
 118   { "OF", TK_OF },
 119   { "OFFSET", TK_OFFSET },
 120   { "ON", TK_ON },
 121   { "OR", TK_OR },
 122   { "ORDER", TK_ORDER },
 123   { "OUTER", TK_JOIN_KW },
 124   { "PRAGMA", TK_PRAGMA },
 125   { "PRIMARY", TK_PRIMARY },
 126   { "RAISE", TK_RAISE },
 127   { "REFERENCES", TK_REFERENCES },
 128   { "REPLACE", TK_REPLACE },
 129   { "RESTRICT", TK_RESTRICT },
 130   { "RIGHT", TK_JOIN_KW },
 131   { "ROLLBACK", TK_ROLLBACK },
 132   { "ROW", TK_ROW },
 133   { "SELECT", TK_SELECT },
 134   { "SET", TK_SET },
 135   { "SHORT", TK_SHORT },
 136   { "STATEMENT", TK_STATEMENT },
 137   { "TABLE", TK_TABLE },
 138   { "TEMP", TK_TEMP },
 139   { "TEMPORARY", TK_TEMP },
 140   { "THEN", TK_THEN },
 141   { "TRANSACTION", TK_TRANSACTION },
 142   { "TRIGGER", TK_TRIGGER },
 143   { "UNION", TK_UNION },
 144   { "UNIQUE", TK_UNIQUE },
 145   { "UPDATE", TK_UPDATE },
 146   { "USING", TK_USING },
 147   { "VACUUM", TK_VACUUM },
 148   { "VALUES", TK_VALUES },
 149   { "VIEW", TK_VIEW },
 150   { "WHEN", TK_WHEN },
 151   { "WHERE", TK_WHERE },
 152 };
 153
 154 #define KEYWORD_COUNT ( sizeof aKeywordTable/sizeof (Keyword) )
 155
 156 /*
 157 ** This function looks up an identifier to determine if it is a
 158 ** keyword.  If it is a keyword, the token code of that keyword is
 159 ** returned.  If the input is not a keyword, TK_ID is returned.
 160 */
 161 int sqliteKeywordCode(const WCHAR *z, int n){
 162   UINT i, len;
 163   char buffer[0x10];
 164
 165   len = WideCharToMultiByte( CP_ACP, 0, z, n, buffer, sizeof buffer, NULL, NULL );
 166   for(i=0; i<len; i++)
 167       buffer[i] = toupper(buffer[i]);
 168   for(i=0; i<KEYWORD_COUNT; i++)
 169   {
 170       if(memcmp(buffer, aKeywordTable[i].zName, len))
 171           continue;
 172       if(strlen(aKeywordTable[i].zName) == len )
 173           return aKeywordTable[i].tokenType;
 174   }
 175   return TK_ID;
 176 }
 177
 178
 179 /*
 180 ** If X is a character that can be used in an identifier then
 181 ** isIdChar[X] will be 1.  Otherwise isIdChar[X] will be 0.
 182 **
 183 ** In this implementation, an identifier can be a string of
 184 ** alphabetic characters, digits, and "_" plus any character
 185 ** with the high-order bit set.  The latter rule means that
 186 ** any sequence of UTF-8 characters or characters taken from
 187 ** an extended ISO8859 character set can form an identifier.
 188 */
 189 static const char isIdChar[] = {
 190 /* x0 x1 x2 x3 x4 x5 x6 x7 x8 x9 xA xB xC xD xE xF */
 191     0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  /* 0x */
 192     0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  /* 1x */
 193     0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,  /* 2x */
 194     1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0, 0,  /* 3x */
 195     0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,  /* 4x */
 196     1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 0, 0, 1,  /* 5x */
 197     0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,  /* 6x */
 198     1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0,  /* 7x */
 199     1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,  /* 8x */
 200     1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,  /* 9x */
 201     1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,  /* Ax */
 202     1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,  /* Bx */
 203     1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,  /* Cx */
 204     1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,  /* Dx */
 205     1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,  /* Ex */
 206     1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,  /* Fx */
 207 };
 208
 209
 210 /*
 211 ** Return the length of the token that begins at z[0].  Return
 212 ** -1 if the token is (or might be) incomplete.  Store the token
 213 ** type in *tokenType before returning.
 214 */
 215 int sqliteGetToken(const WCHAR *z, int *tokenType){
 216   int i;
 217   switch( *z ){
 218     case ' ': case '\t': case '\n': case '\f': case '\r': {
 219       for(i=1; isspace(z[i]); i++){}
 220       *tokenType = TK_SPACE;
 221       return i;
 222     }
 223     case '-': {
 224       if( z[1]==0 ) return -1;
 225       if( z[1]=='-' ){
 226         for(i=2; z[i] && z[i]!='\n'; i++){}
 227         *tokenType = TK_COMMENT;
 228         return i;
 229       }
 230       *tokenType = TK_MINUS;
 231       return 1;
 232     }
 233     case '(': {
 234       if( z[1]=='+' && z[2]==')' ){
 235         *tokenType = TK_ORACLE_OUTER_JOIN;
 236         return 3;
 237       }else{
 238         *tokenType = TK_LP;
 239         return 1;
 240       }
 241     }
 242     case ')': {
 243       *tokenType = TK_RP;
 244       return 1;
 245     }
 246     case ';': {
 247       *tokenType = TK_SEMI;
 248       return 1;
 249     }
 250     case '+': {
 251       *tokenType = TK_PLUS;
 252       return 1;
 253     }
 254     case '*': {
 255       *tokenType = TK_STAR;
 256       return 1;
 257     }
 258     case '/': {
 259       if( z[1]!='*' || z[2]==0 ){
 260         *tokenType = TK_SLASH;
 261         return 1;
 262       }
 263       for(i=3; z[i] && (z[i]!='/' || z[i-1]!='*'); i++){}
 264       if( z[i] ) i++;
 265       *tokenType = TK_COMMENT;
 266       return i;
 267     }
 268     case '%': {
 269       *tokenType = TK_REM;
 270       return 1;
 271     }
 272     case '=': {
 273       *tokenType = TK_EQ;
 274       return 1 + (z[1]=='=');
 275     }
 276     case '<': {
 277       if( z[1]=='=' ){
 278         *tokenType = TK_LE;
 279         return 2;
 280       }else if( z[1]=='>' ){
 281         *tokenType = TK_NE;
 282         return 2;
 283       }else if( z[1]=='<' ){
 284         *tokenType = TK_LSHIFT;
 285         return 2;
 286       }else{
 287         *tokenType = TK_LT;
 288         return 1;
 289       }
 290     }
 291     case '>': {
 292       if( z[1]=='=' ){
 293         *tokenType = TK_GE;
 294         return 2;
 295       }else if( z[1]=='>' ){
 296         *tokenType = TK_RSHIFT;
 297         return 2;
 298       }else{
 299         *tokenType = TK_GT;
 300         return 1;
 301       }
 302     }
 303     case '!': {
 304       if( z[1]!='=' ){
 305         *tokenType = TK_ILLEGAL;
 306         return 2;
 307       }else{
 308         *tokenType = TK_NE;
 309         return 2;
 310       }
 311     }
 312     case '|': {
 313       if( z[1]!='|' ){
 314         *tokenType = TK_BITOR;
 315         return 1;
 316       }else{
 317         *tokenType = TK_CONCAT;
 318         return 2;
 319       }
 320     }
 321     case '?': {
 322       *tokenType = TK_WILDCARD;
 323       return 1;
 324     }
 325     case ',': {
 326       *tokenType = TK_COMMA;
 327       return 1;
 328     }
 329     case '&': {
 330       *tokenType = TK_BITAND;
 331       return 1;
 332     }
 333     case '~': {
 334       *tokenType = TK_BITNOT;
 335       return 1;
 336     }
 337     case '`': case '\'': case '"': {
 338       int delim = z[0];
 339       for(i=1; z[i]; i++){
 340         if( z[i]==delim ){
 341           if( z[i+1]==delim ){
 342             i++;
 343           }else{
 344             break;
 345           }
 346         }
 347       }
 348       if( z[i] ) i++;
 349       *tokenType = TK_STRING;
 350       return i;
 351     }
 352     case '.': {
 353       if( !isdigit(z[1]) ){
 354         *tokenType = TK_DOT;
 355         return 1;
 356       }
 357       /* Fall thru into the next case */
 358     }
 359     case '0': case '1': case '2': case '3': case '4':
 360     case '5': case '6': case '7': case '8': case '9': {
 361       *tokenType = TK_INTEGER;
 362       for(i=1; isdigit(z[i]); i++){}
 363       if( z[i]=='.' ){
 364         i++;
 365         while( isdigit(z[i]) ){ i++; }
 366         *tokenType = TK_FLOAT;
 367       }
 368       if( (z[i]=='e' || z[i]=='E') &&
 369            ( isdigit(z[i+1])
 370             || ((z[i+1]=='+' || z[i+1]=='-') && isdigit(z[i+2]))
 371            )
 372       ){
 373         i += 2;
 374         while( isdigit(z[i]) ){ i++; }
 375         *tokenType = TK_FLOAT;
 376       }else if( z[0]=='.' ){
 377         *tokenType = TK_FLOAT;
 378       }
 379       return i;
 380     }
 381     case '[': {
 382       for(i=1; z[i] && z[i-1]!=']'; i++){}
 383       *tokenType = TK_ID;
 384       return i;
 385     }
 386     default: {
 387       if( !isIdChar[*z] ){
 388         break;
 389       }
 390       for(i=1; isIdChar[z[i]]; i++){}
 391       *tokenType = sqliteKeywordCode(z, i);
 392       return i;
 393     }
 394   }
 395   *tokenType = TK_ILLEGAL;
 396   return 1;
 397 }