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

   1 /*
   2  * Copyright (C) 2005 Mike McCormack for CodeWeavers
   3  *
   4  * A test program for MSI database files.
   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 #include <stdio.h>
  21
  22 #include <windows.h>
  23 #include <msi.h>
  24 #include <msiquery.h>
  25
  26 #include "wine/test.h"
  27
  28 static const char *msifile = "winetest.msi";
  29
  30 static void test_msidatabase(void)
  31 {
  32     MSIHANDLE hdb = 0;
  33     UINT res;
  34
  35     DeleteFile(msifile);
  36
  37     /* create an empty database */
  38     res = MsiOpenDatabase(msifile, MSIDBOPEN_CREATE, &hdb );
  39     ok( res == ERROR_SUCCESS , "Failed to create database\n" );
  40
  41     res = MsiDatabaseCommit( hdb );
  42     ok( res == ERROR_SUCCESS , "Failed to commit database\n" );
  43
  44     res = MsiCloseHandle( hdb );
  45     ok( res == ERROR_SUCCESS , "Failed to close database\n" );
  46
  47     res = DeleteFile( msifile );
  48     ok( res == TRUE, "Falled to delete database\n" );
  49 }
  50
  51 static void test_msiinsert(void)
  52 {
  53     MSIHANDLE hdb = 0, hview = 0, hrec = 0;
  54     UINT r;
  55     const char *query;
  56     char buf[80];
  57     DWORD sz;
  58
  59     DeleteFile(msifile);
  60
  61     /* just MsiOpenDatabase should not create a file */
  62     r = MsiOpenDatabase(msifile, MSIDBOPEN_CREATE, &hdb);
  63     ok(r == ERROR_SUCCESS, "MsiOpenDatabase failed\n");
  64
  65     /* create a table */
  66     query = "CREATE TABLE `phone` ( "
  67             "`id` INT, `name` CHAR(32), `number` CHAR(32) "
  68             "PRIMARY KEY `id`)";
  69     r = MsiDatabaseOpenView(hdb, query, &hview);
  70     ok(r == ERROR_SUCCESS, "MsiDatabaseOpenView failed\n");
  71     r = MsiViewExecute(hview, 0);
  72     ok(r == ERROR_SUCCESS, "MsiViewExecute failed\n");
  73     r = MsiViewClose(hview);
  74     ok(r == ERROR_SUCCESS, "MsiViewClose failed\n");
  75     r = MsiCloseHandle(hview);
  76     ok(r == ERROR_SUCCESS, "MsiCloseHandle failed\n");
  77
  78     /* insert a value into it */
  79     query = "INSERT INTO `phone` ( `id`, `name`, `number` )"
  80         "VALUES('1', 'Abe', '8675309')";
  81     r = MsiDatabaseOpenView(hdb, query, &hview);
  82     ok(r == ERROR_SUCCESS, "MsiDatabaseOpenView failed\n");
  83     r = MsiViewExecute(hview, 0);
  84     ok(r == ERROR_SUCCESS, "MsiViewExecute failed\n");
  85     r = MsiViewClose(hview);
  86     ok(r == ERROR_SUCCESS, "MsiViewClose failed\n");
  87     r = MsiCloseHandle(hview);
  88     ok(r == ERROR_SUCCESS, "MsiCloseHandle failed\n");
  89
  90     query = "SELECT * FROM `phone`";
  91     r = MsiDatabaseOpenView(hdb, query, &hview);
  92     ok(r == ERROR_SUCCESS, "MsiDatabaseOpenView failed\n");
  93     r = MsiViewExecute(hview, 0);
  94     ok(r == ERROR_SUCCESS, "MsiViewExecute failed\n");
  95     r = MsiViewFetch(hview, &hrec);
  96     ok(r == ERROR_SUCCESS, "MsiViewFetch failed\n");
  97
  98     /* check the record contains what we put in it */
  99     r = MsiRecordGetFieldCount(hrec);
 100     ok(r == 3, "record count wrong\n");
 101
 102     r = MsiRecordGetInteger(hrec, 1);
 103     ok(r == 1, "field 1 contents wrong\n");
 104     sz = sizeof buf;
 105     r = MsiRecordGetString(hrec, 2, buf, &sz);
 106     ok(r == ERROR_SUCCESS, "field 2 content fetch failed\n");
 107     ok(!strcmp(buf,"Abe"), "field 2 content incorrect\n");
 108     sz = sizeof buf;
 109     r = MsiRecordGetString(hrec, 3, buf, &sz);
 110     ok(r == ERROR_SUCCESS, "field 3 content fetch failed\n");
 111     ok(!strcmp(buf,"8675309"), "field 3 content incorrect\n");
 112
 113     r = MsiViewClose(hview);
 114     ok(r == ERROR_SUCCESS, "MsiViewClose failed\n");
 115     r = MsiCloseHandle(hview);
 116     ok(r == ERROR_SUCCESS, "MsiCloseHandle failed\n");
 117     r = MsiCloseHandle(hrec);
 118     ok(r == ERROR_SUCCESS, "MsiCloseHandle failed\n");
 119
 120     r = MsiDatabaseCommit(hdb);
 121     ok(r == ERROR_SUCCESS, "MsiDatabaseCommit failed\n");
 122
 123     r = MsiCloseHandle(hdb);
 124     ok(r == ERROR_SUCCESS, "MsiCloseHandle failed\n");
 125
 126     r = DeleteFile(msifile);
 127     ok(r == TRUE, "file didn't exist after commit\n");
 128 }
 129
 130 typedef UINT (WINAPI *fnMsiDecomposeDescriptorA)(LPCSTR, LPCSTR, LPSTR, LPSTR, DWORD *);
 131 fnMsiDecomposeDescriptorA MsiDecomposeDescriptorA;
 132
 133 static void test_msidecomposedesc(void)
 134 {
 135     char prod[MAX_FEATURE_CHARS+1], comp[MAX_FEATURE_CHARS+1], feature[MAX_FEATURE_CHARS+1];
 136     const char *desc;
 137     UINT r;
 138     DWORD len;
 139     HMODULE hmod;
 140
 141     hmod = GetModuleHandle("msi.dll");
 142     if (!hmod)
 143         return;
 144     MsiDecomposeDescriptorA = (fnMsiDecomposeDescriptorA)
 145         GetProcAddress(hmod, "MsiDecomposeDescriptorA");
 146     if (!MsiDecomposeDescriptorA)
 147         return;
 148
 149     /* test a valid feature descriptor */
 150     desc = "']gAVn-}f(ZXfeAR6.jiFollowTheWhiteRabbit>3w2x^IGfe?CxI5heAvk.";
 151     len = 0;
 152     r = MsiDecomposeDescriptorA(desc, prod, feature, comp, &len);
 153     ok(r == ERROR_SUCCESS, "returned an error\n");
 154     ok(len == strlen(desc), "length was wrong\n");
 155     ok(strcmp(prod,"{90110409-6000-11D3-8CFE-0150048383C9}")==0, "product wrong\n");
 156     ok(strcmp(feature,"FollowTheWhiteRabbit")==0, "feature wrong\n");
 157     ok(strcmp(comp,"{A7CD68DB-EF74-49C8-FBB2-A7C463B2AC24}")==0,"component wrong\n");
 158
 159     /* test an invalid feature descriptor with too many characters */
 160     desc = "']gAVn-}f(ZXfeAR6.ji"
 161            "ThisWillFailIfTheresMoreThanAGuidsChars>"
 162            "3w2x^IGfe?CxI5heAvk.";
 163     len = 0;
 164     r = MsiDecomposeDescriptorA(desc, prod, feature, comp, &len);
 165     ok(r == ERROR_INVALID_PARAMETER, "returned wrong error\n");
 166
 167     /*
 168      * Test a valid feature descriptor with the
 169      * maximum number of characters and some trailing characters.
 170      */
 171     desc = "']gAVn-}f(ZXfeAR6.ji"
 172            "ThisWillWorkIfTheresLTEThanAGuidsChars>"
 173            "3w2x^IGfe?CxI5heAvk."
 174            "extra";
 175     len = 0;
 176     r = MsiDecomposeDescriptorA(desc, prod, feature, comp, &len);
 177     ok(r == ERROR_SUCCESS, "returned wrong error\n");
 178     ok(len == (strlen(desc) - strlen("extra")), "length wrong\n");
 179 }
 180
 181 static UINT try_query_param( MSIHANDLE hdb, LPCSTR szQuery, MSIHANDLE hrec )
 182 {
 183     MSIHANDLE htab = 0;
 184     UINT res;
 185
 186     res = MsiDatabaseOpenView( hdb, szQuery, &htab );
 187     if(res == ERROR_SUCCESS )
 188     {
 189         UINT r;
 190
 191         r = MsiViewExecute( htab, hrec );
 192         if(r != ERROR_SUCCESS )
 193             res = r;
 194
 195         r = MsiViewClose( htab );
 196         if(r != ERROR_SUCCESS )
 197             res = r;
 198
 199         r = MsiCloseHandle( htab );
 200         if(r != ERROR_SUCCESS )
 201             res = r;
 202     }
 203     return res;
 204 }
 205
 206 static UINT try_query( MSIHANDLE hdb, LPCSTR szQuery )
 207 {
 208     return try_query_param( hdb, szQuery, 0 );
 209 }
 210
 211 static UINT try_insert_query( MSIHANDLE hdb, LPCSTR szQuery )
 212 {
 213     MSIHANDLE hrec = 0;
 214     UINT r;
 215
 216     hrec = MsiCreateRecord( 1 );
 217     MsiRecordSetString( hrec, 1, "Hello");
 218
 219     r = try_query_param( hdb, szQuery, hrec );
 220
 221     MsiCloseHandle( hrec );
 222     return r;
 223 }
 224
 225 static void test_msibadqueries(void)
 226 {
 227     MSIHANDLE hdb = 0;
 228     UINT r;
 229
 230     DeleteFile(msifile);
 231
 232     /* just MsiOpenDatabase should not create a file */
 233     r = MsiOpenDatabase(msifile, MSIDBOPEN_CREATE, &hdb);
 234     ok(r == ERROR_SUCCESS, "MsiOpenDatabase failed\n");
 235
 236     r = MsiDatabaseCommit( hdb );
 237     ok(r == ERROR_SUCCESS , "Failed to commit database\n");
 238
 239     r = MsiCloseHandle( hdb );
 240     ok(r == ERROR_SUCCESS , "Failed to close database\n");
 241
 242     /* open it readonly */
 243     r = MsiOpenDatabase(msifile, MSIDBOPEN_READONLY, &hdb );
 244     ok(r == ERROR_SUCCESS , "Failed to open database r/o\n");
 245
 246     /* add a table to it */
 247     r = try_query( hdb, "select * from _Tables");
 248     ok(r == ERROR_SUCCESS , "query 1 failed\n");
 249
 250     r = MsiCloseHandle( hdb );
 251     ok(r == ERROR_SUCCESS , "Failed to close database r/o\n");
 252
 253     /* open it read/write */
 254     r = MsiOpenDatabase(msifile, MSIDBOPEN_TRANSACT, &hdb );
 255     ok(r == ERROR_SUCCESS , "Failed to open database r/w\n");
 256
 257     /* a bunch of test queries that fail with the native MSI */
 258
 259     r = try_query( hdb, "CREATE TABLE");
 260     ok(r == ERROR_BAD_QUERY_SYNTAX , "invalid query 2a return code\n");
 261
 262     r = try_query( hdb, "CREATE TABLE `a`");
 263     ok(r == ERROR_BAD_QUERY_SYNTAX , "invalid query 2b return code\n");
 264
 265     r = try_query( hdb, "CREATE TABLE `a` ()");
 266     ok(r == ERROR_BAD_QUERY_SYNTAX , "invalid query 2c return code\n");
 267
 268     r = try_query( hdb, "CREATE TABLE `a` (`b`)");
 269     ok(r == ERROR_BAD_QUERY_SYNTAX , "invalid query 2d return code\n");
 270
 271     r = try_query( hdb, "CREATE TABLE `a` (`b` CHAR(72) )");
 272     ok(r == ERROR_BAD_QUERY_SYNTAX , "invalid query 2e return code\n");
 273
 274     r = try_query( hdb, "CREATE TABLE `a` (`b` CHAR(72) NOT NULL)");
 275     ok(r == ERROR_BAD_QUERY_SYNTAX , "invalid query 2f return code\n");
 276
 277     r = try_query( hdb, "CREATE TABLE `a` (`b` CHAR(72) NOT NULL PRIMARY)");
 278     ok(r == ERROR_BAD_QUERY_SYNTAX , "invalid query 2g return code\n");
 279
 280     r = try_query( hdb, "CREATE TABLE `a` (`b` CHAR(72) NOT NULL PRIMARY KEY)");
 281     ok(r == ERROR_BAD_QUERY_SYNTAX , "invalid query 2h return code\n");
 282
 283     r = try_query( hdb, "CREATE TABLE `a` (`b` CHAR(72) NOT NULL PRIMARY KEY)");
 284     ok(r == ERROR_BAD_QUERY_SYNTAX , "invalid query 2i return code\n");
 285
 286     r = try_query( hdb, "CREATE TABLE `a` (`b` CHAR(72) NOT NULL PRIMARY KEY 'b')");
 287     ok(r == ERROR_BAD_QUERY_SYNTAX , "invalid query 2j return code\n");
 288
 289     r = try_query( hdb, "CREATE TABLE `a` (`b` CHAR(72) NOT NULL PRIMARY KEY `b')");
 290     ok(r == ERROR_BAD_QUERY_SYNTAX , "invalid query 2k return code\n");
 291
 292     r = try_query( hdb, "CREATE TABLE `a` (`b` CHAR(72) NOT NULL PRIMARY KEY `b')");
 293     ok(r == ERROR_BAD_QUERY_SYNTAX , "invalid query 2l return code\n");
 294
 295     r = try_query( hdb, "CREATE TABLE `a` (`b` CHA(72) NOT NULL PRIMARY KEY `b`)");
 296     ok(r == ERROR_BAD_QUERY_SYNTAX , "invalid query 2m return code\n");
 297
 298     r = try_query( hdb, "CREATE TABLE `a` (`b` CHAR(-1) NOT NULL PRIMARY KEY `b`)");
 299     ok(r == ERROR_BAD_QUERY_SYNTAX , "invalid query 2n return code\n");
 300
 301     r = try_query( hdb, "CREATE TABLE `a` (`b` CHAR(720) NOT NULL PRIMARY KEY `b`)");
 302     ok(r == ERROR_BAD_QUERY_SYNTAX , "invalid query 2o return code\n");
 303
 304     r = try_query( hdb, "CREATE TABLE `a` (`b` CHAR(72) NOT NULL KEY `b`)");
 305     ok(r == ERROR_BAD_QUERY_SYNTAX , "invalid query 2p return code\n");
 306
 307     r = try_query( hdb, "CREATE TABLE `a` (`` CHAR(72) NOT NULL PRIMARY KEY `b`)");
 308     ok(r == ERROR_BAD_QUERY_SYNTAX , "invalid query 2p return code\n");
 309
 310     r = try_query( hdb, "CREATE TABLE `a` (`b` CHAR(72) NOT NULL PRIMARY KEY `b`)");
 311     ok(r == ERROR_SUCCESS , "valid query 2z failed\n");
 312
 313     r = try_query( hdb, "CREATE TABLE `a` (`b` CHAR(72) NOT NULL PRIMARY KEY `b`)");
 314     ok(r == ERROR_BAD_QUERY_SYNTAX , "created same table again\n");
 315
 316     r = try_query( hdb, "CREATE TABLE `aa` (`b` CHAR(72) NOT NULL, `c` "
 317           "CHAR(72), `d` CHAR(255) NOT NULL LOCALIZABLE PRIMARY KEY `b`)");
 318     ok(r == ERROR_SUCCESS , "query 4 failed\n");
 319
 320     r = MsiDatabaseCommit( hdb );
 321     ok(r == ERROR_SUCCESS , "Failed to commit database after write\n");
 322
 323     r = try_query( hdb, "CREATE TABLE `blah` (`foo` CHAR(72) NOT NULL "
 324                           "PRIMARY KEY `foo`)");
 325     ok(r == ERROR_SUCCESS , "query 4 failed\n");
 326
 327     r = try_insert_query( hdb, "insert into a  ( `b` ) VALUES ( ? )");
 328     ok(r == ERROR_SUCCESS , "failed to insert record in db\n");
 329
 330     r = MsiDatabaseCommit( hdb );
 331     ok(r == ERROR_SUCCESS , "Failed to commit database after write\n");
 332
 333     r = try_query( hdb, "CREATE TABLE `boo` (`foo` CHAR(72) NOT NULL "
 334                           "PRIMARY KEY `ba`)");
 335     ok(r != ERROR_SUCCESS , "query 5 succeeded\n");
 336
 337     r = try_query( hdb,"CREATE TABLE `bee` (`foo` CHAR(72) NOT NULL )");
 338     ok(r != ERROR_SUCCESS , "query 6 succeeded\n");
 339
 340     r = try_query( hdb, "CREATE TABLE `temp` (`t` CHAR(72) NOT NULL "
 341                           "PRIMARY KEY `t`)");
 342     ok(r == ERROR_SUCCESS , "query 7 failed\n");
 343
 344     r = try_query( hdb, "CREATE TABLE `c` (`b` CHAR NOT NULL PRIMARY KEY `b`)");
 345     ok(r == ERROR_SUCCESS , "query 8 failed\n");
 346
 347     r = MsiCloseHandle( hdb );
 348     ok(r == ERROR_SUCCESS , "Failed to close database transact\n");
 349
 350     r = DeleteFile( msifile );
 351     ok(r == TRUE, "file didn't exist after commit\n");
 352 }
 353
 354 static UINT run_query( MSIHANDLE hdb, const char *query )
 355 {
 356     MSIHANDLE hview = 0;
 357     UINT r;
 358
 359     r = MsiDatabaseOpenView(hdb, query, &hview);
 360     if( r != ERROR_SUCCESS )
 361         return r;
 362
 363     r = MsiViewExecute(hview, 0);
 364     if( r == ERROR_SUCCESS )
 365         r = MsiViewClose(hview);
 366     MsiCloseHandle(hview);
 367     return r;
 368 }
 369
 370 static void test_viewmodify(void)
 371 {
 372     MSIHANDLE hdb = 0, hview = 0, hrec = 0;
 373     UINT r;
 374     const char *query;
 375     char buffer[0x100];
 376     DWORD sz;
 377
 378     DeleteFile(msifile);
 379
 380     /* just MsiOpenDatabase should not create a file */
 381     r = MsiOpenDatabase(msifile, MSIDBOPEN_CREATE, &hdb);
 382     ok(r == ERROR_SUCCESS, "MsiOpenDatabase failed\n");
 383
 384     query = "CREATE TABLE `phone` ( "
 385             "`id` INT, `name` CHAR(32), `number` CHAR(32) "
 386             "PRIMARY KEY `id`)";
 387     r = run_query( hdb, query );
 388     ok(r == ERROR_SUCCESS, "query failed\n");
 389
 390     /* check what the error function reports without doing anything */
 391     sz = 0;
 392     /* passing NULL as the 3rd param make function to crash on older platforms */
 393     r = MsiViewGetError( 0, NULL, &sz );
 394     ok(r == MSIDBERROR_INVALIDARG, "MsiViewGetError return\n");
 395
 396     /* open a view */
 397     query = "SELECT * FROM `phone`";
 398     r = MsiDatabaseOpenView(hdb, query, &hview);
 399     ok(r == ERROR_SUCCESS, "MsiDatabaseOpenView failed\n");
 400
 401     /* see what happens with a good hview and bad args */
 402     r = MsiViewGetError( hview, NULL, NULL );
 403     ok(r == MSIDBERROR_INVALIDARG, "MsiViewGetError return\n");
 404     r = MsiViewGetError( hview, buffer, NULL );
 405     ok(r == MSIDBERROR_INVALIDARG, "MsiViewGetError return\n");
 406
 407     /* see what happens with a zero length buffer */
 408     sz = 0;
 409     buffer[0] = 'x';
 410     r = MsiViewGetError( hview, buffer, &sz );
 411     ok(r == MSIDBERROR_MOREDATA, "MsiViewGetError return\n");
 412     ok(buffer[0] == 'x', "buffer cleared\n");
 413     ok(sz == 0, "size not zero\n");
 414
 415     /* ok this one is strange */
 416     sz = 0;
 417     r = MsiViewGetError( hview, NULL, &sz );
 418     ok(r == MSIDBERROR_NOERROR, "MsiViewGetError return\n");
 419     ok(sz == 0, "size not zero\n");
 420
 421     /* see if it really has an error */
 422     sz = sizeof buffer;
 423     buffer[0] = 'x';
 424     r = MsiViewGetError( hview, buffer, &sz );
 425     ok(r == MSIDBERROR_NOERROR, "MsiViewGetError return\n");
 426     ok(buffer[0] == 0, "buffer not cleared\n");
 427     ok(sz == 0, "size not zero\n");
 428
 429     r = MsiViewExecute(hview, 0);
 430     ok(r == ERROR_SUCCESS, "MsiViewExecute failed\n");
 431
 432     /* try some invalid records */
 433     r = MsiViewModify(hview, MSIMODIFY_INSERT, 0 );
 434     ok(r == ERROR_INVALID_HANDLE, "MsiViewModify failed\n");
 435     r = MsiViewModify(hview, -1, 0 );
 436     ok(r == ERROR_INVALID_HANDLE, "MsiViewModify failed\n");
 437
 438     /* try an small record */
 439     hrec = MsiCreateRecord(1);
 440     r = MsiViewModify(hview, -1, hrec );
 441     ok(r == ERROR_INVALID_DATA, "MsiViewModify failed\n");
 442
 443     r = MsiCloseHandle(hrec);
 444     ok(r == ERROR_SUCCESS, "failed to close record\n");
 445
 446     /* insert a valid record */
 447     hrec = MsiCreateRecord(3);
 448
 449     r = MsiRecordSetInteger(hrec, 2, 1);
 450     ok(r == ERROR_SUCCESS, "failed to set integer\n");
 451     r = MsiRecordSetString(hrec, 2, "bob");
 452     ok(r == ERROR_SUCCESS, "failed to set integer\n");
 453     r = MsiRecordSetString(hrec, 3, "7654321");
 454     ok(r == ERROR_SUCCESS, "failed to set integer\n");
 455
 456     r = MsiViewExecute(hview, 0);
 457     ok(r == ERROR_SUCCESS, "MsiViewExecute failed\n");
 458     r = MsiViewModify(hview, MSIMODIFY_INSERT_TEMPORARY, hrec );
 459     ok(r == ERROR_SUCCESS, "MsiViewModify failed\n");
 460
 461     /* insert the same thing again */
 462     r = MsiViewExecute(hview, 0);
 463     ok(r == ERROR_SUCCESS, "MsiViewExecute failed\n");
 464
 465     /* should fail ... */
 466     todo_wine {
 467     r = MsiViewModify(hview, MSIMODIFY_INSERT_TEMPORARY, hrec );
 468     ok(r == ERROR_FUNCTION_FAILED, "MsiViewModify failed\n");
 469     }
 470
 471     r = MsiCloseHandle(hrec);
 472     ok(r == ERROR_SUCCESS, "failed to close record\n");
 473
 474     r = MsiViewClose(hview);
 475     ok(r == ERROR_SUCCESS, "MsiViewClose failed\n");
 476     r = MsiCloseHandle(hview);
 477     ok(r == ERROR_SUCCESS, "MsiCloseHandle failed\n");
 478
 479     r = MsiCloseHandle( hdb );
 480     ok(r == ERROR_SUCCESS, "MsiOpenDatabase close failed\n");
 481 }
 482
 483 START_TEST(db)
 484 {
 485     test_msidatabase();
 486     test_msiinsert();
 487     test_msidecomposedesc();
 488     test_msibadqueries();
 489     test_viewmodify();
 490 }