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