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