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