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 void test_msidatabase(void)
  29 {
  30     MSIHANDLE hdb = 0;
  31     CHAR szName[] = "C:\\mytest.msi";
  32     UINT res;
  33
  34     DeleteFile(szName);
  35
  36     /* create an empty database */
  37     res = MsiOpenDatabase(szName, MSIDBOPEN_CREATE, &hdb );
  38     ok( res == ERROR_SUCCESS , "Failed to create database\n" );
  39
  40     res = MsiDatabaseCommit( hdb );
  41     ok( res == ERROR_SUCCESS , "Failed to commit database\n" );
  42
  43     res = MsiCloseHandle( hdb );
  44     ok( res == ERROR_SUCCESS , "Failed to close database\n" );
  45
  46     res = DeleteFile( szName );
  47     ok( res == TRUE, "Falled to delete database\n" );
  48 }
  49
  50 static void test_msiinsert(void)
  51 {
  52     const char *msifile = "winetest.msi";
  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()
 226 {
 227     const char *msifile = "winetest.msi";
 228     MSIHANDLE hdb = 0;
 229     UINT r;
 230
 231     DeleteFile(msifile);
 232
 233     /* just MsiOpenDatabase should not create a file */
 234     r = MsiOpenDatabase(msifile, MSIDBOPEN_CREATE, &hdb);
 235     ok(r == ERROR_SUCCESS, "MsiOpenDatabase failed\n");
 236
 237     r = MsiDatabaseCommit( hdb );
 238     ok(r == ERROR_SUCCESS , "Failed to commit database\n");
 239
 240     r = MsiCloseHandle( hdb );
 241     ok(r == ERROR_SUCCESS , "Failed to close database\n");
 242
 243     /* open it readonly */
 244     r = MsiOpenDatabase(msifile, MSIDBOPEN_READONLY, &hdb );
 245     ok(r == ERROR_SUCCESS , "Failed to open database r/o\n");
 246
 247     /* add a table to it */
 248     r = try_query( hdb, "select * from _Tables");
 249     ok(r == ERROR_SUCCESS , "query 1 failed\n");
 250
 251     r = MsiCloseHandle( hdb );
 252     ok(r == ERROR_SUCCESS , "Failed to close database r/o\n");
 253
 254     /* open it read/write */
 255     r = MsiOpenDatabase(msifile, MSIDBOPEN_TRANSACT, &hdb );
 256     ok(r == ERROR_SUCCESS , "Failed to open database r/w\n");
 257
 258     /* a bunch of test queries that fail with the native MSI */
 259
 260     r = try_query( hdb, "CREATE TABLE");
 261     ok(r == ERROR_BAD_QUERY_SYNTAX , "invalid query 2a return code\n");
 262
 263     r = try_query( hdb, "CREATE TABLE `a`");
 264     ok(r == ERROR_BAD_QUERY_SYNTAX , "invalid query 2b return code\n");
 265
 266     r = try_query( hdb, "CREATE TABLE `a` ()");
 267     ok(r == ERROR_BAD_QUERY_SYNTAX , "invalid query 2c return code\n");
 268
 269     r = try_query( hdb, "CREATE TABLE `a` (`b`)");
 270     ok(r == ERROR_BAD_QUERY_SYNTAX , "invalid query 2d return code\n");
 271
 272     r = try_query( hdb, "CREATE TABLE `a` (`b` CHAR(72) )");
 273     ok(r == ERROR_BAD_QUERY_SYNTAX , "invalid query 2e return code\n");
 274
 275     r = try_query( hdb, "CREATE TABLE `a` (`b` CHAR(72) NOT NULL)");
 276     ok(r == ERROR_BAD_QUERY_SYNTAX , "invalid query 2f return code\n");
 277
 278     r = try_query( hdb, "CREATE TABLE `a` (`b` CHAR(72) NOT NULL PRIMARY)");
 279     ok(r == ERROR_BAD_QUERY_SYNTAX , "invalid query 2g return code\n");
 280
 281     r = try_query( hdb, "CREATE TABLE `a` (`b` CHAR(72) NOT NULL PRIMARY KEY)");
 282     ok(r == ERROR_BAD_QUERY_SYNTAX , "invalid query 2h return code\n");
 283
 284     r = try_query( hdb, "CREATE TABLE `a` (`b` CHAR(72) NOT NULL PRIMARY KEY)");
 285     ok(r == ERROR_BAD_QUERY_SYNTAX , "invalid query 2i return code\n");
 286
 287     r = try_query( hdb, "CREATE TABLE `a` (`b` CHAR(72) NOT NULL PRIMARY KEY 'b')");
 288     ok(r == ERROR_BAD_QUERY_SYNTAX , "invalid query 2j return code\n");
 289
 290     r = try_query( hdb, "CREATE TABLE `a` (`b` CHAR(72) NOT NULL PRIMARY KEY `b')");
 291     ok(r == ERROR_BAD_QUERY_SYNTAX , "invalid query 2k return code\n");
 292
 293     r = try_query( hdb, "CREATE TABLE `a` (`b` CHAR(72) NOT NULL PRIMARY KEY `b')");
 294     ok(r == ERROR_BAD_QUERY_SYNTAX , "invalid query 2l return code\n");
 295
 296     r = try_query( hdb, "CREATE TABLE `a` (`b` CHA(72) NOT NULL PRIMARY KEY `b`)");
 297     ok(r == ERROR_BAD_QUERY_SYNTAX , "invalid query 2m return code\n");
 298
 299     r = try_query( hdb, "CREATE TABLE `a` (`b` CHAR(-1) NOT NULL PRIMARY KEY `b`)");
 300     ok(r == ERROR_BAD_QUERY_SYNTAX , "invalid query 2n return code\n");
 301
 302     r = try_query( hdb, "CREATE TABLE `a` (`b` CHAR(720) NOT NULL PRIMARY KEY `b`)");
 303     ok(r == ERROR_BAD_QUERY_SYNTAX , "invalid query 2o return code\n");
 304
 305     r = try_query( hdb, "CREATE TABLE `a` (`b` CHAR(72) NOT NULL KEY `b`)");
 306     ok(r == ERROR_BAD_QUERY_SYNTAX , "invalid query 2p return code\n");
 307
 308     r = try_query( hdb, "CREATE TABLE `a` (`` CHAR(72) NOT NULL PRIMARY KEY `b`)");
 309     ok(r == ERROR_BAD_QUERY_SYNTAX , "invalid query 2p return code\n");
 310
 311     r = try_query( hdb, "CREATE TABLE `a` (`b` CHAR(72) NOT NULL PRIMARY KEY `b`)");
 312     ok(r == ERROR_SUCCESS , "valid query 2z failed\n");
 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\n");
 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\n");
 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 }