dlls/msi/msiquery.c

   1 /*
   2  * Implementation of the Microsoft Installer (msi.dll)
   3  *
   4  * Copyright 2002-2005 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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  19  */
  20
  21 #include <stdarg.h>
  22
  23 #define COBJMACROS
  24
  25 #include "windef.h"
  26 #include "winbase.h"
  27 #include "winerror.h"
  28 #include "wine/debug.h"
  29 #include "wine/unicode.h"
  30 #include "msi.h"
  31 #include "msiquery.h"
  32 #include "objbase.h"
  33 #include "objidl.h"
  34 #include "msipriv.h"
  35 #include "winnls.h"
  36
  37 #include "query.h"
  38
  39 WINE_DEFAULT_DEBUG_CHANNEL(msi);
  40
  41 static void MSI_CloseView( MSIOBJECTHDR *arg )
  42 {
  43     MSIQUERY *query = (MSIQUERY*) arg;
  44     struct list *ptr, *t;
  45
  46     if( query->view && query->view->ops->delete )
  47         query->view->ops->delete( query->view );
  48     msiobj_release( &query->db->hdr );
  49
  50     LIST_FOR_EACH_SAFE( ptr, t, &query->mem )
  51     {
  52         msi_free( ptr );
  53     }
  54 }
  55
  56 UINT VIEW_find_column( MSIVIEW *table, LPCWSTR name, UINT *n )
  57 {
  58     LPWSTR col_name;
  59     UINT i, count, r;
  60
  61     r = table->ops->get_dimensions( table, NULL, &count );
  62     if( r != ERROR_SUCCESS )
  63         return r;
  64
  65     for( i=1; i<=count; i++ )
  66     {
  67         INT x;
  68
  69         col_name = NULL;
  70         r = table->ops->get_column_info( table, i, &col_name, NULL );
  71         if( r != ERROR_SUCCESS )
  72             return r;
  73         x = lstrcmpW( name, col_name );
  74         msi_free( col_name );
  75         if( !x )
  76         {
  77             *n = i;
  78             return ERROR_SUCCESS;
  79         }
  80     }
  81
  82     return ERROR_INVALID_PARAMETER;
  83 }
  84
  85 UINT WINAPI MsiDatabaseOpenViewA(MSIHANDLE hdb,
  86               LPCSTR szQuery, MSIHANDLE *phView)
  87 {
  88     UINT r;
  89     LPWSTR szwQuery;
  90
  91     TRACE("%ld %s %p\n", hdb, debugstr_a(szQuery), phView);
  92
  93     if( szQuery )
  94     {
  95         szwQuery = strdupAtoW( szQuery );
  96         if( !szwQuery )
  97             return ERROR_FUNCTION_FAILED;
  98     }
  99     else
 100         szwQuery = NULL;
 101
 102     r = MsiDatabaseOpenViewW( hdb, szwQuery, phView);
 103
 104     msi_free( szwQuery );
 105     return r;
 106 }
 107
 108 UINT MSI_DatabaseOpenViewW(MSIDATABASE *db,
 109               LPCWSTR szQuery, MSIQUERY **pView)
 110 {
 111     MSIQUERY *query;
 112     UINT r;
 113
 114     TRACE("%s %p\n", debugstr_w(szQuery), pView);
 115
 116     if( !szQuery)
 117         return ERROR_INVALID_PARAMETER;
 118
 119     /* pre allocate a handle to hold a pointer to the view */
 120     query = alloc_msiobject( MSIHANDLETYPE_VIEW, sizeof (MSIQUERY),
 121                               MSI_CloseView );
 122     if( !query )
 123         return ERROR_FUNCTION_FAILED;
 124
 125     msiobj_addref( &db->hdr );
 126     query->row = 0;
 127     query->db = db;
 128     query->view = NULL;
 129     list_init( &query->mem );
 130
 131     r = MSI_ParseSQL( db, szQuery, &query->view, &query->mem );
 132     if( r == ERROR_SUCCESS )
 133     {
 134         msiobj_addref( &query->hdr );
 135         *pView = query;
 136     }
 137
 138     msiobj_release( &query->hdr );
 139     return r;
 140 }
 141
 142 UINT MSI_OpenQuery( MSIDATABASE *db, MSIQUERY **view, LPCWSTR fmt, ... )
 143 {
 144     UINT r;
 145     int size = 100, res;
 146     LPWSTR query;
 147
 148     /* construct the string */
 149     for (;;)
 150     {
 151         va_list va;
 152         query = msi_alloc( size*sizeof(WCHAR) );
 153         va_start(va, fmt);
 154         res = vsnprintfW(query, size, fmt, va);
 155         va_end(va);
 156         if (res == -1) size *= 2;
 157         else if (res >= size) size = res + 1;
 158         else break;
 159         msi_free( query );
 160     }
 161     /* perform the query */
 162     r = MSI_DatabaseOpenViewW(db, query, view);
 163     msi_free(query);
 164     return r;
 165 }
 166
 167 UINT MSI_IterateRecords( MSIQUERY *view, DWORD *count,
 168                          record_func func, LPVOID param )
 169 {
 170     MSIRECORD *rec = NULL;
 171     UINT r, n = 0, max = 0;
 172
 173     r = MSI_ViewExecute( view, NULL );
 174     if( r != ERROR_SUCCESS )
 175         return r;
 176
 177     if( count )
 178         max = *count;
 179
 180     /* iterate a query */
 181     for( n = 0; (max == 0) || (n < max); n++ )
 182     {
 183         r = MSI_ViewFetch( view, &rec );
 184         if( r != ERROR_SUCCESS )
 185             break;
 186         if (func)
 187             r = func( rec, param );
 188         msiobj_release( &rec->hdr );
 189         if( r != ERROR_SUCCESS )
 190             break;
 191     }
 192
 193     MSI_ViewClose( view );
 194
 195     if( count )
 196         *count = n;
 197
 198     if( r == ERROR_NO_MORE_ITEMS )
 199         r = ERROR_SUCCESS;
 200
 201     return r;
 202 }
 203
 204 /* return a single record from a query */
 205 MSIRECORD *MSI_QueryGetRecord( MSIDATABASE *db, LPCWSTR fmt, ... )
 206 {
 207     MSIRECORD *rec = NULL;
 208     MSIQUERY *view = NULL;
 209     UINT r;
 210     int size = 100, res;
 211     LPWSTR query;
 212
 213     /* construct the string */
 214     for (;;)
 215     {
 216         va_list va;
 217         query = msi_alloc( size*sizeof(WCHAR) );
 218         va_start(va, fmt);
 219         res = vsnprintfW(query, size, fmt, va);
 220         va_end(va);
 221         if (res == -1) size *= 2;
 222         else if (res >= size) size = res + 1;
 223         else break;
 224         msi_free( query );
 225     }
 226     /* perform the query */
 227     r = MSI_DatabaseOpenViewW(db, query, &view);
 228     msi_free(query);
 229
 230     if( r == ERROR_SUCCESS )
 231     {
 232         MSI_ViewExecute( view, NULL );
 233         MSI_ViewFetch( view, &rec );
 234         MSI_ViewClose( view );
 235         msiobj_release( &view->hdr );
 236     }
 237     return rec;
 238 }
 239
 240 UINT WINAPI MsiDatabaseOpenViewW(MSIHANDLE hdb,
 241               LPCWSTR szQuery, MSIHANDLE *phView)
 242 {
 243     MSIDATABASE *db;
 244     MSIQUERY *query = NULL;
 245     UINT ret;
 246
 247     TRACE("%s %p\n", debugstr_w(szQuery), phView);
 248
 249     db = msihandle2msiinfo( hdb, MSIHANDLETYPE_DATABASE );
 250     if( !db )
 251         return ERROR_INVALID_HANDLE;
 252
 253     ret = MSI_DatabaseOpenViewW( db, szQuery, &query );
 254     if( ret == ERROR_SUCCESS )
 255     {
 256         *phView = alloc_msihandle( &query->hdr );
 257         msiobj_release( &query->hdr );
 258     }
 259     msiobj_release( &db->hdr );
 260
 261     return ret;
 262 }
 263
 264 UINT MSI_ViewFetch(MSIQUERY *query, MSIRECORD **prec)
 265 {
 266     MSIVIEW *view;
 267     MSIRECORD *rec;
 268     UINT row_count = 0, col_count = 0, i, ival, ret, type;
 269
 270     TRACE("%p %p\n", query, prec );
 271
 272     view = query->view;
 273     if( !view )
 274         return ERROR_FUNCTION_FAILED;
 275
 276     ret = view->ops->get_dimensions( view, &row_count, &col_count );
 277     if( ret )
 278         return ret;
 279     if( !col_count )
 280         return ERROR_INVALID_PARAMETER;
 281
 282     if( query->row >= row_count )
 283         return ERROR_NO_MORE_ITEMS;
 284
 285     rec = MSI_CreateRecord( col_count );
 286     if( !rec )
 287         return ERROR_FUNCTION_FAILED;
 288
 289     for( i=1; i<=col_count; i++ )
 290     {
 291         ret = view->ops->get_column_info( view, i, NULL, &type );
 292         if( ret )
 293         {
 294             ERR("Error getting column type for %d\n", i );
 295             continue;
 296         }
 297         if (!MSITYPE_IS_BINARY(type))
 298         {
 299             ret = view->ops->fetch_int( view, query->row, i, &ival );
 300             if( ret )
 301             {
 302                 ERR("Error fetching data for %d\n", i );
 303                 continue;
 304             }
 305             if( ! (type & MSITYPE_VALID ) )
 306                 ERR("Invalid type!\n");
 307
 308             /* check if it's nul (0) - if so, don't set anything */
 309             if( !ival )
 310                 continue;
 311
 312             if( type & MSITYPE_STRING )
 313             {
 314                 LPWSTR sval;
 315
 316                 sval = MSI_makestring( query->db, ival );
 317                 MSI_RecordSetStringW( rec, i, sval );
 318                 msi_free( sval );
 319             }
 320             else
 321             {
 322                 if( (type & MSI_DATASIZEMASK) == 2 )
 323                     MSI_RecordSetInteger( rec, i, ival - (1<<15) );
 324                 else
 325                     MSI_RecordSetInteger( rec, i, ival - (1<<31) );
 326             }
 327         }
 328         else
 329         {
 330             IStream *stm = NULL;
 331
 332             ret = view->ops->fetch_stream( view, query->row, i, &stm );
 333             if( ( ret == ERROR_SUCCESS ) && stm )
 334             {
 335                 MSI_RecordSetIStream( rec, i, stm );
 336                 IStream_Release( stm );
 337             }
 338             else
 339                 ERR("failed to get stream\n");
 340         }
 341     }
 342     query->row ++;
 343
 344     *prec = rec;
 345
 346     return ERROR_SUCCESS;
 347 }
 348
 349 UINT WINAPI MsiViewFetch(MSIHANDLE hView, MSIHANDLE *record)
 350 {
 351     MSIQUERY *query;
 352     MSIRECORD *rec = NULL;
 353     UINT ret;
 354
 355     TRACE("%ld %p\n", hView, record);
 356
 357     query = msihandle2msiinfo( hView, MSIHANDLETYPE_VIEW );
 358     if( !query )
 359         return ERROR_INVALID_HANDLE;
 360     ret = MSI_ViewFetch( query, &rec );
 361     if( ret == ERROR_SUCCESS )
 362     {
 363         *record = alloc_msihandle( &rec->hdr );
 364         msiobj_release( &rec->hdr );
 365     }
 366     msiobj_release( &query->hdr );
 367     return ret;
 368 }
 369
 370 UINT MSI_ViewClose(MSIQUERY *query)
 371 {
 372     MSIVIEW *view;
 373
 374     TRACE("%p\n", query );
 375
 376     view = query->view;
 377     if( !view )
 378         return ERROR_FUNCTION_FAILED;
 379     if( !view->ops->close )
 380         return ERROR_FUNCTION_FAILED;
 381
 382     return view->ops->close( view );
 383 }
 384
 385 UINT WINAPI MsiViewClose(MSIHANDLE hView)
 386 {
 387     MSIQUERY *query;
 388     UINT ret;
 389
 390     TRACE("%ld\n", hView );
 391
 392     query = msihandle2msiinfo( hView, MSIHANDLETYPE_VIEW );
 393     if( !query )
 394         return ERROR_INVALID_HANDLE;
 395
 396     ret = MSI_ViewClose( query );
 397     msiobj_release( &query->hdr );
 398     return ret;
 399 }
 400
 401 UINT MSI_ViewExecute(MSIQUERY *query, MSIRECORD *rec )
 402 {
 403     MSIVIEW *view;
 404
 405     TRACE("%p %p\n", query, rec);
 406
 407     view = query->view;
 408     if( !view )
 409         return ERROR_FUNCTION_FAILED;
 410     if( !view->ops->execute )
 411         return ERROR_FUNCTION_FAILED;
 412     query->row = 0;
 413
 414     return view->ops->execute( view, rec );
 415 }
 416
 417 UINT WINAPI MsiViewExecute(MSIHANDLE hView, MSIHANDLE hRec)
 418 {
 419     MSIQUERY *query;
 420     MSIRECORD *rec = NULL;
 421     UINT ret;
 422
 423     TRACE("%ld %ld\n", hView, hRec);
 424
 425     query = msihandle2msiinfo( hView, MSIHANDLETYPE_VIEW );
 426     if( !query )
 427         return ERROR_INVALID_HANDLE;
 428
 429     if( hRec )
 430     {
 431         rec = msihandle2msiinfo( hRec, MSIHANDLETYPE_RECORD );
 432         if( !rec )
 433         {
 434             ret = ERROR_INVALID_HANDLE;
 435             goto out;
 436         }
 437     }
 438
 439     msiobj_lock( &rec->hdr );
 440     ret = MSI_ViewExecute( query, rec );
 441     msiobj_unlock( &rec->hdr );
 442
 443 out:
 444     msiobj_release( &query->hdr );
 445     if( rec )
 446         msiobj_release( &rec->hdr );
 447
 448     return ret;
 449 }
 450
 451 static UINT msi_set_record_type_string( MSIRECORD *rec, UINT field, UINT type )
 452 {
 453     static const WCHAR fmt[] = { '%','d',0 };
 454     WCHAR szType[0x10];
 455
 456     if (MSITYPE_IS_BINARY(type))
 457         szType[0] = 'v';
 458     else if (type & MSITYPE_LOCALIZABLE)
 459         szType[0] = 'l';
 460     else if (type & MSITYPE_STRING)
 461         szType[0] = 's';
 462     else
 463         szType[0] = 'i';
 464     if (type & MSITYPE_NULLABLE)
 465         szType[0] &= ~0x20;
 466
 467     sprintfW( &szType[1], fmt, (type&0xff) );
 468
 469     TRACE("type %04x -> %s\n", type, debugstr_w(szType) );
 470
 471     return MSI_RecordSetStringW( rec, field, szType );
 472 }
 473
 474 UINT WINAPI MsiViewGetColumnInfo(MSIHANDLE hView, MSICOLINFO info, MSIHANDLE *hRec)
 475 {
 476     MSIVIEW *view = NULL;
 477     MSIQUERY *query = NULL;
 478     MSIRECORD *rec = NULL;
 479     UINT r = ERROR_FUNCTION_FAILED, i, count = 0, type;
 480     LPWSTR name;
 481
 482     TRACE("%ld %d %p\n", hView, info, hRec);
 483
 484     if( !hRec )
 485         return ERROR_INVALID_PARAMETER;
 486
 487     if( info != MSICOLINFO_NAMES && info != MSICOLINFO_TYPES )
 488         return ERROR_INVALID_PARAMETER;
 489
 490     query = msihandle2msiinfo( hView, MSIHANDLETYPE_VIEW );
 491     if( !query )
 492         return ERROR_INVALID_HANDLE;
 493
 494     view = query->view;
 495     if( !view )
 496         goto out;
 497
 498     if( !view->ops->get_dimensions )
 499         goto out;
 500
 501     r = view->ops->get_dimensions( view, NULL, &count );
 502     if( r )
 503         goto out;
 504     if( !count )
 505     {
 506         r = ERROR_INVALID_PARAMETER;
 507         goto out;
 508     }
 509
 510     rec = MSI_CreateRecord( count );
 511     if( !rec )
 512     {
 513         r = ERROR_FUNCTION_FAILED;
 514         goto out;
 515     }
 516
 517     for( i=0; i<count; i++ )
 518     {
 519         name = NULL;
 520         r = view->ops->get_column_info( view, i+1, &name, &type );
 521         if( r != ERROR_SUCCESS )
 522             continue;
 523         if (info == MSICOLINFO_NAMES)
 524             MSI_RecordSetStringW( rec, i+1, name );
 525         else
 526             msi_set_record_type_string( rec, i+1, type);
 527         msi_free( name );
 528     }
 529
 530     *hRec = alloc_msihandle( &rec->hdr );
 531
 532 out:
 533     msiobj_release( &query->hdr );
 534     if( rec )
 535         msiobj_release( &rec->hdr );
 536
 537     return r;
 538 }
 539
 540 UINT WINAPI MsiViewModify( MSIHANDLE hView, MSIMODIFY eModifyMode,
 541                 MSIHANDLE hRecord)
 542 {
 543     MSIVIEW *view = NULL;
 544     MSIQUERY *query = NULL;
 545     MSIRECORD *rec = NULL;
 546     UINT r = ERROR_FUNCTION_FAILED;
 547
 548     TRACE("%ld %x %ld\n", hView, eModifyMode, hRecord);
 549
 550     query = msihandle2msiinfo( hView, MSIHANDLETYPE_VIEW );
 551     if( !query )
 552         return ERROR_INVALID_HANDLE;
 553
 554     view = query->view;
 555     if( !view )
 556         goto out;
 557
 558     if( !view->ops->modify )
 559         goto out;
 560
 561     rec = msihandle2msiinfo( hRecord, MSIHANDLETYPE_RECORD );
 562     if( !rec )
 563     {
 564         r = ERROR_INVALID_HANDLE;
 565         goto out;
 566     }
 567
 568     r = view->ops->modify( view, eModifyMode, rec );
 569
 570 out:
 571     msiobj_release( &query->hdr );
 572     if( rec )
 573         msiobj_release( &rec->hdr );
 574
 575     return r;
 576 }
 577
 578 MSIDBERROR WINAPI MsiViewGetErrorW( MSIHANDLE handle, LPWSTR szColumnNameBuffer,
 579                               DWORD *pcchBuf )
 580 {
 581     MSIQUERY *query = NULL;
 582     static const WCHAR szError[] = { 0 };
 583     MSIDBERROR r = MSIDBERROR_NOERROR;
 584     int len;
 585
 586     FIXME("%ld %p %p - returns empty error string\n",
 587           handle, szColumnNameBuffer, pcchBuf );
 588
 589     if( !pcchBuf )
 590         return MSIDBERROR_INVALIDARG;
 591
 592     query = msihandle2msiinfo( handle, MSIHANDLETYPE_VIEW );
 593     if( !query )
 594         return MSIDBERROR_INVALIDARG;
 595
 596     len = lstrlenW( szError );
 597     if( szColumnNameBuffer )
 598     {
 599         if( *pcchBuf > len )
 600             lstrcpyW( szColumnNameBuffer, szError );
 601         else
 602             r = MSIDBERROR_MOREDATA;
 603     }
 604     *pcchBuf = len;
 605
 606     msiobj_release( &query->hdr );
 607     return r;
 608 }
 609
 610 MSIDBERROR WINAPI MsiViewGetErrorA( MSIHANDLE handle, LPSTR szColumnNameBuffer,
 611                               DWORD *pcchBuf )
 612 {
 613     static const CHAR szError[] = { 0 };
 614     MSIQUERY *query = NULL;
 615     MSIDBERROR r = MSIDBERROR_NOERROR;
 616     int len;
 617
 618     FIXME("%ld %p %p - returns empty error string\n",
 619           handle, szColumnNameBuffer, pcchBuf );
 620
 621     if( !pcchBuf )
 622         return MSIDBERROR_INVALIDARG;
 623
 624     query = msihandle2msiinfo( handle, MSIHANDLETYPE_VIEW );
 625     if( !query )
 626         return MSIDBERROR_INVALIDARG;
 627
 628     len = lstrlenA( szError );
 629     if( szColumnNameBuffer )
 630     {
 631         if( *pcchBuf > len )
 632             lstrcpyA( szColumnNameBuffer, szError );
 633         else
 634             r = MSIDBERROR_MOREDATA;
 635     }
 636     *pcchBuf = len;
 637
 638     msiobj_release( &query->hdr );
 639     return r;
 640 }
 641
 642 DEFINE_GUID( CLSID_MsiTransform, 0x000c1082, 0x0000, 0x0000, 0xc0,0x00,0x00,0x00,0x00,0x00,0x00,0x46);
 643
 644 static UINT MSI_DatabaseApplyTransformW( MSIDATABASE *db,
 645                  LPCWSTR szTransformFile, int iErrorCond )
 646 {
 647     UINT r;
 648     IStorage *stg = NULL;
 649
 650     TRACE("%p %s %d\n", db, debugstr_w(szTransformFile), iErrorCond);
 651
 652     r = StgOpenStorage( szTransformFile, NULL,
 653            STGM_DIRECT|STGM_READ|STGM_SHARE_DENY_WRITE, NULL, 0, &stg);
 654     if( r )
 655         return r;
 656
 657     if( TRACE_ON( msi ) )
 658         enum_stream_names( stg );
 659
 660     r = msi_table_apply_transform( db, stg );
 661
 662     IStorage_Release( stg );
 663
 664     return r;
 665 }
 666
 667 UINT WINAPI MsiDatabaseApplyTransformW( MSIHANDLE hdb,
 668                  LPCWSTR szTransformFile, int iErrorCond)
 669 {
 670     MSIDATABASE *db;
 671     UINT r;
 672
 673     db = msihandle2msiinfo( hdb, MSIHANDLETYPE_DATABASE );
 674     if( !db )
 675         return ERROR_INVALID_HANDLE;
 676
 677     r = MSI_DatabaseApplyTransformW( db, szTransformFile, iErrorCond );
 678     msiobj_release( &db->hdr );
 679     return r;
 680 }
 681
 682 UINT WINAPI MsiDatabaseApplyTransformA( MSIHANDLE hdb,
 683                  LPCSTR szTransformFile, int iErrorCond)
 684 {
 685     LPWSTR wstr;
 686     UINT ret;
 687
 688     TRACE("%ld %s %d\n", hdb, debugstr_a(szTransformFile), iErrorCond);
 689
 690     wstr = strdupAtoW( szTransformFile );
 691     if( szTransformFile && !wstr )
 692         return ERROR_NOT_ENOUGH_MEMORY;
 693
 694     ret = MsiDatabaseApplyTransformW( hdb, wstr, iErrorCond);
 695
 696     msi_free( wstr );
 697
 698     return ret;
 699 }
 700
 701 UINT WINAPI MsiDatabaseGenerateTransformA( MSIHANDLE hdb, MSIHANDLE hdbref,
 702                  LPCSTR szTransformFile, int iReserved1, int iReserved2 )
 703 {
 704     FIXME("%ld %ld %s %d %d\n", hdb, hdbref,
 705            debugstr_a(szTransformFile), iReserved1, iReserved2);
 706     return ERROR_CALL_NOT_IMPLEMENTED;
 707 }
 708
 709 UINT WINAPI MsiDatabaseGenerateTransformW( MSIHANDLE hdb, MSIHANDLE hdbref,
 710                  LPCWSTR szTransformFile, int iReserved1, int iReserved2 )
 711 {
 712     FIXME("%ld %ld %s %d %d\n", hdb, hdbref,
 713            debugstr_w(szTransformFile), iReserved1, iReserved2);
 714     return ERROR_CALL_NOT_IMPLEMENTED;
 715 }
 716
 717 UINT WINAPI MsiDatabaseCommit( MSIHANDLE hdb )
 718 {
 719     MSIDATABASE *db;
 720     UINT r;
 721
 722     TRACE("%ld\n", hdb);
 723
 724     db = msihandle2msiinfo( hdb, MSIHANDLETYPE_DATABASE );
 725     if( !db )
 726         return ERROR_INVALID_HANDLE;
 727
 728     /* FIXME: lock the database */
 729
 730     r = MSI_CommitTables( db );
 731
 732     /* FIXME: unlock the database */
 733
 734     msiobj_release( &db->hdr );
 735
 736     return r;
 737 }
 738
 739 struct msi_primary_key_record_info
 740 {
 741     DWORD n;
 742     MSIRECORD *rec;
 743 };
 744
 745 static UINT msi_primary_key_iterator( MSIRECORD *rec, LPVOID param )
 746 {
 747     struct msi_primary_key_record_info *info = param;
 748     LPCWSTR name;
 749     DWORD type;
 750
 751     type = MSI_RecordGetInteger( rec, 4 );
 752     if( type & MSITYPE_KEY )
 753     {
 754         info->n++;
 755         if( info->rec )
 756         {
 757             name = MSI_RecordGetString( rec, 3 );
 758             MSI_RecordSetStringW( info->rec, info->n, name );
 759         }
 760     }
 761
 762     return ERROR_SUCCESS;
 763 }
 764
 765 UINT MSI_DatabaseGetPrimaryKeys( MSIDATABASE *db,
 766                 LPCWSTR table, MSIRECORD **prec )
 767 {
 768     static const WCHAR sql[] = {
 769         's','e','l','e','c','t',' ','*',' ',
 770         'f','r','o','m',' ','`','_','C','o','l','u','m','n','s','`',' ',
 771         'w','h','e','r','e',' ',
 772         '`','T','a','b','l','e','`',' ','=',' ','\'','%','s','\'',0 };
 773     struct msi_primary_key_record_info info;
 774     MSIQUERY *query = NULL;
 775     MSIVIEW *view;
 776     UINT r;
 777
 778     r = MSI_OpenQuery( db, &query, sql, table );
 779     if( r != ERROR_SUCCESS )
 780         return r;
 781
 782     view = query->view;
 783
 784     /* count the number of primary key records */
 785     info.n = 0;
 786     info.rec = 0;
 787     r = MSI_IterateRecords( query, 0, msi_primary_key_iterator, &info );
 788     if( r == ERROR_SUCCESS )
 789     {
 790         TRACE("Found %ld primary keys\n", info.n );
 791
 792         /* allocate a record and fill in the names of the tables */
 793         info.rec = MSI_CreateRecord( info.n );
 794         info.n = 0;
 795         r = MSI_IterateRecords( query, 0, msi_primary_key_iterator, &info );
 796         if( r == ERROR_SUCCESS )
 797             *prec = info.rec;
 798         else
 799             msiobj_release( &info.rec->hdr );
 800     }
 801     msiobj_release( &query->hdr );
 802
 803     return r;
 804 }
 805
 806 UINT WINAPI MsiDatabaseGetPrimaryKeysW( MSIHANDLE hdb,
 807                     LPCWSTR table, MSIHANDLE* phRec )
 808 {
 809     MSIRECORD *rec = NULL;
 810     MSIDATABASE *db;
 811     UINT r;
 812
 813     TRACE("%ld %s %p\n", hdb, debugstr_w(table), phRec);
 814
 815     db = msihandle2msiinfo( hdb, MSIHANDLETYPE_DATABASE );
 816     if( !db )
 817         return ERROR_INVALID_HANDLE;
 818
 819     r = MSI_DatabaseGetPrimaryKeys( db, table, &rec );
 820     if( r == ERROR_SUCCESS )
 821     {
 822         *phRec = alloc_msihandle( &rec->hdr );
 823         msiobj_release( &rec->hdr );
 824     }
 825     msiobj_release( &db->hdr );
 826
 827     return r;
 828 }
 829
 830 UINT WINAPI MsiDatabaseGetPrimaryKeysA(MSIHANDLE hdb,
 831                     LPCSTR table, MSIHANDLE* phRec)
 832 {
 833     LPWSTR szwTable = NULL;
 834     UINT r;
 835
 836     TRACE("%ld %s %p\n", hdb, debugstr_a(table), phRec);
 837
 838     if( table )
 839     {
 840         szwTable = strdupAtoW( table );
 841         if( !szwTable )
 842             return ERROR_OUTOFMEMORY;
 843     }
 844     r = MsiDatabaseGetPrimaryKeysW( hdb, szwTable, phRec );
 845     msi_free( szwTable );
 846
 847     return r;
 848 }
 849
 850 UINT WINAPI MsiDatabaseIsTablePersistentA(
 851               MSIHANDLE hDatabase, LPSTR szTableName)
 852 {
 853     FIXME("%lx %s\n", hDatabase, debugstr_a(szTableName));
 854     return ERROR_CALL_NOT_IMPLEMENTED;
 855 }
 856
 857 UINT WINAPI MsiDatabaseIsTablePersistentW(
 858               MSIHANDLE hDatabase, LPWSTR szTableName)
 859 {
 860     FIXME("%lx %s\n", hDatabase, debugstr_w(szTableName));
 861     return ERROR_CALL_NOT_IMPLEMENTED;
 862 }