git.oblomov.eu Git - wine/blob - dlls/msi/where.c

   1 /*
   2  * Implementation of the Microsoft Installer (msi.dll)
   3  *
   4  * Copyright 2002 Mike McCormack for CodeWeavers
   5  *
   6  * This library is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2.1 of the License, or (at your option) any later version.
  10  *
  11  * This library is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public
  17  * License along with this library; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  19  */
  20
  21 #include <stdarg.h>
  22
  23 #include "windef.h"
  24 #include "winbase.h"
  25 #include "winerror.h"
  26 #include "wine/debug.h"
  27 #include "msi.h"
  28 #include "msiquery.h"
  29 #include "objbase.h"
  30 #include "objidl.h"
  31 #include "msipriv.h"
  32 #include "winnls.h"
  33
  34 #include "query.h"
  35
  36 WINE_DEFAULT_DEBUG_CHANNEL(msidb);
  37
  38 #define MSI_HASH_TABLE_SIZE 37
  39
  40 typedef struct tagMSIHASHENTRY
  41 {
  42     struct tagMSIHASHENTRY *next;
  43     UINT value;
  44     UINT row;
  45 } MSIHASHENTRY;
  46
  47 /* below is the query interface to a table */
  48
  49 typedef struct tagMSIWHEREVIEW
  50 {
  51     MSIVIEW        view;
  52     MSIDATABASE   *db;
  53     MSIVIEW       *table;
  54     UINT           row_count;
  55     MSIHASHENTRY **reorder;
  56     struct expr   *cond;
  57     UINT           rec_index;
  58 } MSIWHEREVIEW;
  59
  60 static void free_hash_table(MSIHASHENTRY **table)
  61 {
  62     MSIHASHENTRY *new, *old;
  63     int i;
  64
  65     if (!table)
  66         return;
  67
  68     for (i = 0; i < MSI_HASH_TABLE_SIZE; i++)
  69     {
  70         new = table[i];
  71
  72         while (new)
  73         {
  74             old = new;
  75             new = old->next;
  76             msi_free(old);
  77         }
  78
  79         table[i] = NULL;
  80     }
  81
  82     msi_free(table);
  83 }
  84
  85 static UINT find_entry_in_hash(MSIHASHENTRY **table, UINT row, UINT *val)
  86 {
  87     MSIHASHENTRY *entry;
  88
  89     if (!(entry = table[row % MSI_HASH_TABLE_SIZE]))
  90     {
  91         ERR("Row not found in hash table!\n");
  92         return ERROR_FUNCTION_FAILED;
  93     }
  94
  95     while (entry && entry->row != row)
  96         entry = entry->next;
  97
  98     if (entry) *val = entry->value;
  99     return ERROR_SUCCESS;
 100 }
 101
 102 static UINT add_entry_to_hash(MSIHASHENTRY **table, UINT row, UINT val)
 103 {
 104     MSIHASHENTRY *new = msi_alloc(sizeof(MSIHASHENTRY));
 105     MSIHASHENTRY *prev;
 106
 107     if (!new)
 108         return ERROR_OUTOFMEMORY;
 109
 110     new->next = NULL;
 111     new->value = val;
 112     new->row = row;
 113
 114     prev = table[row % MSI_HASH_TABLE_SIZE];
 115     if (prev)
 116         new->next = prev;
 117
 118     table[row % MSI_HASH_TABLE_SIZE] = new;
 119
 120     return ERROR_SUCCESS;
 121 }
 122
 123 static UINT WHERE_fetch_int( struct tagMSIVIEW *view, UINT row, UINT col, UINT *val )
 124 {
 125     MSIWHEREVIEW *wv = (MSIWHEREVIEW*)view;
 126     UINT r;
 127
 128     TRACE("%p %d %d %p\n", wv, row, col, val );
 129
 130     if( !wv->table )
 131         return ERROR_FUNCTION_FAILED;
 132
 133     if( row > wv->row_count )
 134         return ERROR_NO_MORE_ITEMS;
 135
 136     r = find_entry_in_hash(wv->reorder, row, &row);
 137     if (r != ERROR_SUCCESS)
 138         return r;
 139
 140     return wv->table->ops->fetch_int( wv->table, row, col, val );
 141 }
 142
 143 static UINT WHERE_fetch_stream( struct tagMSIVIEW *view, UINT row, UINT col, IStream **stm )
 144 {
 145     MSIWHEREVIEW *wv = (MSIWHEREVIEW*)view;
 146     UINT r;
 147
 148     TRACE("%p %d %d %p\n", wv, row, col, stm );
 149
 150     if( !wv->table )
 151         return ERROR_FUNCTION_FAILED;
 152
 153     if( row > wv->row_count )
 154         return ERROR_NO_MORE_ITEMS;
 155
 156     r = find_entry_in_hash(wv->reorder, row, &row);
 157     if (r != ERROR_SUCCESS)
 158         return r;
 159
 160     return wv->table->ops->fetch_stream( wv->table, row, col, stm );
 161 }
 162
 163 static UINT WHERE_get_row( struct tagMSIVIEW *view, UINT row, MSIRECORD **rec )
 164 {
 165     MSIWHEREVIEW *wv = (MSIWHEREVIEW *)view;
 166     UINT r;
 167
 168     TRACE("%p %d %p\n", wv, row, rec );
 169
 170     if (!wv->table)
 171         return ERROR_FUNCTION_FAILED;
 172
 173     if (row > wv->row_count)
 174         return ERROR_NO_MORE_ITEMS;
 175
 176     r = find_entry_in_hash(wv->reorder, row, &row);
 177     if (r != ERROR_SUCCESS)
 178         return r;
 179
 180     return wv->table->ops->get_row(view, row, rec);
 181 }
 182
 183 static UINT WHERE_set_row( struct tagMSIVIEW *view, UINT row, MSIRECORD *rec, UINT mask )
 184 {
 185     MSIWHEREVIEW *wv = (MSIWHEREVIEW*)view;
 186     UINT r;
 187
 188     TRACE("%p %d %p %08x\n", wv, row, rec, mask );
 189
 190     if( !wv->table )
 191          return ERROR_FUNCTION_FAILED;
 192
 193     if( row > wv->row_count )
 194         return ERROR_NO_MORE_ITEMS;
 195
 196     r = find_entry_in_hash(wv->reorder, row, &row);
 197     if (r != ERROR_SUCCESS)
 198         return r;
 199
 200     return wv->table->ops->set_row( wv->table, row, rec, mask );
 201 }
 202
 203 static INT INT_evaluate_binary( INT lval, UINT op, INT rval )
 204 {
 205     switch( op )
 206     {
 207     case OP_EQ:
 208         return ( lval == rval );
 209     case OP_AND:
 210         return ( lval && rval );
 211     case OP_OR:
 212         return ( lval || rval );
 213     case OP_GT:
 214         return ( lval > rval );
 215     case OP_LT:
 216         return ( lval < rval );
 217     case OP_LE:
 218         return ( lval <= rval );
 219     case OP_GE:
 220         return ( lval >= rval );
 221     case OP_NE:
 222         return ( lval != rval );
 223     default:
 224         ERR("Unknown operator %d\n", op );
 225     }
 226     return 0;
 227 }
 228
 229 static INT INT_evaluate_unary( INT lval, UINT op )
 230 {
 231     switch( op )
 232     {
 233     case OP_ISNULL:
 234         return ( !lval );
 235     case OP_NOTNULL:
 236         return ( lval );
 237     default:
 238         ERR("Unknown operator %d\n", op );
 239     }
 240     return 0;
 241 }
 242
 243 static const WCHAR *STRING_evaluate( MSIWHEREVIEW *wv, UINT row,
 244                                      const struct expr *expr,
 245                                      const MSIRECORD *record )
 246 {
 247     UINT val = 0, r;
 248
 249     switch( expr->type )
 250     {
 251     case EXPR_COL_NUMBER_STRING:
 252         r = wv->table->ops->fetch_int( wv->table, row, expr->u.col_number, &val );
 253         if( r != ERROR_SUCCESS )
 254             return NULL;
 255         return msi_string_lookup_id( wv->db->strings, val );
 256
 257     case EXPR_SVAL:
 258         return expr->u.sval;
 259
 260     case EXPR_WILDCARD:
 261         return MSI_RecordGetString( record, ++wv->rec_index );
 262
 263     default:
 264         ERR("Invalid expression type\n");
 265         break;
 266     }
 267     return NULL;
 268 }
 269
 270 static UINT STRCMP_Evaluate( MSIWHEREVIEW *wv, UINT row, const struct expr *cond,
 271                              INT *val, const MSIRECORD *record )
 272 {
 273     int sr;
 274     const WCHAR *l_str, *r_str;
 275
 276     l_str = STRING_evaluate( wv, row, cond->u.expr.left, record );
 277     r_str = STRING_evaluate( wv, row, cond->u.expr.right, record );
 278     if( l_str == r_str ||
 279         ((!l_str || !*l_str) && (!r_str || !*r_str)) )
 280         sr = 0;
 281     else if( l_str && ! r_str )
 282         sr = 1;
 283     else if( r_str && ! l_str )
 284         sr = -1;
 285     else
 286         sr = lstrcmpW( l_str, r_str );
 287
 288     *val = ( cond->u.expr.op == OP_EQ && ( sr == 0 ) ) ||
 289            ( cond->u.expr.op == OP_LT && ( sr < 0 ) ) ||
 290            ( cond->u.expr.op == OP_GT && ( sr > 0 ) );
 291
 292     return ERROR_SUCCESS;
 293 }
 294
 295 static UINT WHERE_evaluate( MSIWHEREVIEW *wv, UINT row,
 296                             struct expr *cond, INT *val, MSIRECORD *record )
 297 {
 298     UINT r, tval;
 299     INT lval, rval;
 300
 301     if( !cond )
 302         return ERROR_SUCCESS;
 303
 304     switch( cond->type )
 305     {
 306     case EXPR_COL_NUMBER:
 307         r = wv->table->ops->fetch_int( wv->table, row, cond->u.col_number, &tval );
 308         *val = tval - 0x8000;
 309         return ERROR_SUCCESS;
 310
 311     case EXPR_COL_NUMBER32:
 312         r = wv->table->ops->fetch_int( wv->table, row, cond->u.col_number, &tval );
 313         *val = tval - 0x80000000;
 314         return r;
 315
 316     case EXPR_UVAL:
 317         *val = cond->u.uval;
 318         return ERROR_SUCCESS;
 319
 320     case EXPR_COMPLEX:
 321         r = WHERE_evaluate( wv, row, cond->u.expr.left, &lval, record );
 322         if( r != ERROR_SUCCESS )
 323             return r;
 324         r = WHERE_evaluate( wv, row, cond->u.expr.right, &rval, record );
 325         if( r != ERROR_SUCCESS )
 326             return r;
 327         *val = INT_evaluate_binary( lval, cond->u.expr.op, rval );
 328         return ERROR_SUCCESS;
 329
 330     case EXPR_UNARY:
 331         r = wv->table->ops->fetch_int( wv->table, row, cond->u.expr.left->u.col_number, &tval );
 332         if( r != ERROR_SUCCESS )
 333             return r;
 334         *val = INT_evaluate_unary( tval, cond->u.expr.op );
 335         return ERROR_SUCCESS;
 336
 337     case EXPR_STRCMP:
 338         return STRCMP_Evaluate( wv, row, cond, val, record );
 339
 340     case EXPR_WILDCARD:
 341         *val = MSI_RecordGetInteger( record, ++wv->rec_index );
 342         return ERROR_SUCCESS;
 343
 344     default:
 345         ERR("Invalid expression type\n");
 346         break;
 347     }
 348
 349     return ERROR_SUCCESS;
 350 }
 351
 352 static UINT WHERE_execute( struct tagMSIVIEW *view, MSIRECORD *record )
 353 {
 354     MSIWHEREVIEW *wv = (MSIWHEREVIEW*)view;
 355     UINT count = 0, r, i;
 356     INT val;
 357     MSIVIEW *table = wv->table;
 358
 359     TRACE("%p %p\n", wv, record);
 360
 361     if( !table )
 362          return ERROR_FUNCTION_FAILED;
 363
 364     r = table->ops->execute( table, record );
 365     if( r != ERROR_SUCCESS )
 366         return r;
 367
 368     r = table->ops->get_dimensions( table, &count, NULL );
 369     if( r != ERROR_SUCCESS )
 370         return r;
 371
 372     free_hash_table(wv->reorder);
 373     wv->reorder = msi_alloc_zero(MSI_HASH_TABLE_SIZE * sizeof(MSIHASHENTRY *));
 374     if( !wv->reorder )
 375         return ERROR_OUTOFMEMORY;
 376
 377     wv->row_count = 0;
 378     if (wv->cond->type == EXPR_STRCMP)
 379     {
 380         MSIITERHANDLE handle = NULL;
 381         UINT row, value, col;
 382         struct expr *col_cond = wv->cond->u.expr.left;
 383         struct expr *val_cond = wv->cond->u.expr.right;
 384
 385         /* swap conditionals */
 386         if (col_cond->type != EXPR_COL_NUMBER_STRING)
 387         {
 388             val_cond = wv->cond->u.expr.left;
 389             col_cond = wv->cond->u.expr.right;
 390         }
 391
 392         if ((col_cond->type == EXPR_COL_NUMBER_STRING) && (val_cond->type == EXPR_SVAL))
 393         {
 394             col = col_cond->u.col_number;
 395             /* special case for "" - translate it into nil */
 396             if (!val_cond->u.sval[0])
 397                 value = 0;
 398             else
 399             {
 400                 r = msi_string2idW(wv->db->strings, val_cond->u.sval, &value);
 401                 if (r != ERROR_SUCCESS)
 402                 {
 403                     TRACE("no id for %s, assuming it doesn't exist in the table\n", debugstr_w(wv->cond->u.expr.right->u.sval));
 404                     return ERROR_SUCCESS;
 405                 }
 406             }
 407
 408             do
 409             {
 410                 r = table->ops->find_matching_rows(table, col, value, &row, &handle);
 411                 if (r == ERROR_SUCCESS)
 412                     add_entry_to_hash(wv->reorder, wv->row_count++, row);
 413             } while (r == ERROR_SUCCESS);
 414
 415             if (r == ERROR_NO_MORE_ITEMS)
 416                 return ERROR_SUCCESS;
 417             else
 418                 return r;
 419         }
 420         /* else fallback to slow case */
 421     }
 422
 423     for( i=0; i<count; i++ )
 424     {
 425         val = 0;
 426         wv->rec_index = 0;
 427         r = WHERE_evaluate( wv, i, wv->cond, &val, record );
 428         if( r != ERROR_SUCCESS )
 429             return r;
 430         if( val )
 431             add_entry_to_hash( wv->reorder, wv->row_count++, i );
 432     }
 433
 434     return ERROR_SUCCESS;
 435 }
 436
 437 static UINT WHERE_close( struct tagMSIVIEW *view )
 438 {
 439     MSIWHEREVIEW *wv = (MSIWHEREVIEW*)view;
 440
 441     TRACE("%p\n", wv );
 442
 443     if( !wv->table )
 444         return ERROR_FUNCTION_FAILED;
 445
 446     return wv->table->ops->close( wv->table );
 447 }
 448
 449 static UINT WHERE_get_dimensions( struct tagMSIVIEW *view, UINT *rows, UINT *cols )
 450 {
 451     MSIWHEREVIEW *wv = (MSIWHEREVIEW*)view;
 452
 453     TRACE("%p %p %p\n", wv, rows, cols );
 454
 455     if( !wv->table )
 456          return ERROR_FUNCTION_FAILED;
 457
 458     if( rows )
 459     {
 460         if( !wv->reorder )
 461             return ERROR_FUNCTION_FAILED;
 462         *rows = wv->row_count;
 463     }
 464
 465     return wv->table->ops->get_dimensions( wv->table, NULL, cols );
 466 }
 467
 468 static UINT WHERE_get_column_info( struct tagMSIVIEW *view,
 469                 UINT n, LPWSTR *name, UINT *type )
 470 {
 471     MSIWHEREVIEW *wv = (MSIWHEREVIEW*)view;
 472
 473     TRACE("%p %d %p %p\n", wv, n, name, type );
 474
 475     if( !wv->table )
 476          return ERROR_FUNCTION_FAILED;
 477
 478     return wv->table->ops->get_column_info( wv->table, n, name, type );
 479 }
 480
 481 static UINT WHERE_modify( struct tagMSIVIEW *view, MSIMODIFY eModifyMode,
 482                           MSIRECORD *rec, UINT row )
 483 {
 484     MSIWHEREVIEW *wv = (MSIWHEREVIEW*)view;
 485
 486     TRACE("%p %d %p\n", wv, eModifyMode, rec );
 487
 488     if( !wv->table )
 489          return ERROR_FUNCTION_FAILED;
 490
 491     return wv->table->ops->modify( wv->table, eModifyMode, rec, row );
 492 }
 493
 494 static UINT WHERE_delete( struct tagMSIVIEW *view )
 495 {
 496     MSIWHEREVIEW *wv = (MSIWHEREVIEW*)view;
 497
 498     TRACE("%p\n", wv );
 499
 500     if( wv->table )
 501         wv->table->ops->delete( wv->table );
 502     wv->table = 0;
 503
 504     free_hash_table(wv->reorder);
 505     wv->reorder = NULL;
 506     wv->row_count = 0;
 507
 508     msiobj_release( &wv->db->hdr );
 509     msi_free( wv );
 510
 511     return ERROR_SUCCESS;
 512 }
 513
 514 static UINT WHERE_find_matching_rows( struct tagMSIVIEW *view, UINT col,
 515     UINT val, UINT *row, MSIITERHANDLE *handle )
 516 {
 517     MSIWHEREVIEW *wv = (MSIWHEREVIEW*)view;
 518     UINT r;
 519
 520     TRACE("%p, %d, %u, %p\n", view, col, val, *handle);
 521
 522     if( !wv->table )
 523          return ERROR_FUNCTION_FAILED;
 524
 525     r = wv->table->ops->find_matching_rows( wv->table, col, val, row, handle );
 526     if (r != ERROR_SUCCESS)
 527         return r;
 528
 529     if( *row > wv->row_count )
 530         return ERROR_NO_MORE_ITEMS;
 531
 532     return find_entry_in_hash(wv->reorder, *row, row);
 533 }
 534
 535 static UINT WHERE_sort(struct tagMSIVIEW *view, column_info *columns)
 536 {
 537     MSIWHEREVIEW *wv = (MSIWHEREVIEW *)view;
 538
 539     TRACE("%p %p\n", view, columns);
 540
 541     return wv->table->ops->sort(wv->table, columns);
 542 }
 543
 544 static const MSIVIEWOPS where_ops =
 545 {
 546     WHERE_fetch_int,
 547     WHERE_fetch_stream,
 548     WHERE_get_row,
 549     WHERE_set_row,
 550     NULL,
 551     NULL,
 552     WHERE_execute,
 553     WHERE_close,
 554     WHERE_get_dimensions,
 555     WHERE_get_column_info,
 556     WHERE_modify,
 557     WHERE_delete,
 558     WHERE_find_matching_rows,
 559     NULL,
 560     NULL,
 561     NULL,
 562     NULL,
 563     WHERE_sort,
 564 };
 565
 566 static UINT WHERE_VerifyCondition( MSIDATABASE *db, MSIVIEW *table, struct expr *cond,
 567                                    UINT *valid )
 568 {
 569     UINT r, val = 0;
 570
 571     switch( cond->type )
 572     {
 573     case EXPR_COLUMN:
 574         r = VIEW_find_column( table, cond->u.column, &val );
 575         if( r == ERROR_SUCCESS )
 576         {
 577             UINT type = 0;
 578             r = table->ops->get_column_info( table, val, NULL, &type );
 579             if( r == ERROR_SUCCESS )
 580             {
 581                 if (type&MSITYPE_STRING)
 582                     cond->type = EXPR_COL_NUMBER_STRING;
 583                 else if ((type&0xff) == 4)
 584                     cond->type = EXPR_COL_NUMBER32;
 585                 else
 586                     cond->type = EXPR_COL_NUMBER;
 587                 cond->u.col_number = val;
 588                 *valid = 1;
 589             }
 590             else
 591                 *valid = 0;
 592         }
 593         else
 594         {
 595             *valid = 0;
 596             WARN("Couldn't find column %s\n", debugstr_w( cond->u.column ) );
 597         }
 598         break;
 599     case EXPR_COMPLEX:
 600         r = WHERE_VerifyCondition( db, table, cond->u.expr.left, valid );
 601         if( r != ERROR_SUCCESS )
 602             return r;
 603         if( !*valid )
 604             return ERROR_SUCCESS;
 605         r = WHERE_VerifyCondition( db, table, cond->u.expr.right, valid );
 606         if( r != ERROR_SUCCESS )
 607             return r;
 608
 609         /* check the type of the comparison */
 610         if( ( cond->u.expr.left->type == EXPR_SVAL ) ||
 611             ( cond->u.expr.left->type == EXPR_COL_NUMBER_STRING ) ||
 612             ( cond->u.expr.right->type == EXPR_SVAL ) ||
 613             ( cond->u.expr.right->type == EXPR_COL_NUMBER_STRING ) )
 614         {
 615             switch( cond->u.expr.op )
 616             {
 617             case OP_EQ:
 618             case OP_GT:
 619             case OP_LT:
 620                 break;
 621             default:
 622                 *valid = FALSE;
 623                 return ERROR_INVALID_PARAMETER;
 624             }
 625
 626             /* FIXME: check we're comparing a string to a column */
 627
 628             cond->type = EXPR_STRCMP;
 629         }
 630
 631         break;
 632     case EXPR_UNARY:
 633         if ( cond->u.expr.left->type != EXPR_COLUMN )
 634         {
 635             *valid = FALSE;
 636             return ERROR_INVALID_PARAMETER;
 637         }
 638         r = WHERE_VerifyCondition( db, table, cond->u.expr.left, valid );
 639         if( r != ERROR_SUCCESS )
 640             return r;
 641         break;
 642     case EXPR_IVAL:
 643         *valid = 1;
 644         cond->type = EXPR_UVAL;
 645         cond->u.uval = cond->u.ival;
 646         break;
 647     case EXPR_WILDCARD:
 648         *valid = 1;
 649         break;
 650     case EXPR_SVAL:
 651         *valid = 1;
 652         break;
 653     default:
 654         ERR("Invalid expression type\n");
 655         *valid = 0;
 656         break;
 657     }
 658
 659     return ERROR_SUCCESS;
 660 }
 661
 662 UINT WHERE_CreateView( MSIDATABASE *db, MSIVIEW **view, MSIVIEW *table,
 663                        struct expr *cond )
 664 {
 665     MSIWHEREVIEW *wv = NULL;
 666     UINT count = 0, r, valid = 0;
 667
 668     TRACE("%p\n", table );
 669
 670     r = table->ops->get_dimensions( table, NULL, &count );
 671     if( r != ERROR_SUCCESS )
 672     {
 673         ERR("can't get table dimensions\n");
 674         return r;
 675     }
 676
 677     if( cond )
 678     {
 679         r = WHERE_VerifyCondition( db, table, cond, &valid );
 680         if( r != ERROR_SUCCESS )
 681             return r;
 682         if( !valid )
 683             return ERROR_FUNCTION_FAILED;
 684     }
 685
 686     wv = msi_alloc_zero( sizeof *wv );
 687     if( !wv )
 688         return ERROR_FUNCTION_FAILED;
 689
 690     /* fill the structure */
 691     wv->view.ops = &where_ops;
 692     msiobj_addref( &db->hdr );
 693     wv->db = db;
 694     wv->table = table;
 695     wv->row_count = 0;
 696     wv->reorder = NULL;
 697     wv->cond = cond;
 698     wv->rec_index = 0;
 699     *view = (MSIVIEW*) wv;
 700
 701     return ERROR_SUCCESS;
 702 }