/* * Copyright (C) 2005 Mike McCormack for CodeWeavers * * A test program for MSI database files. * * This library is free software; you can redistribute it and/or * modify it under the terms of the GNU Lesser General Public * License as published by the Free Software Foundation; either * version 2.1 of the License, or (at your option) any later version. * * This library is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU * Lesser General Public License for more details. * * You should have received a copy of the GNU Lesser General Public * License along with this library; if not, write to the Free Software * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA */ #define COBJMACROS #include #include #include #include #include #include #include #ifdef _WIN32 #include #include #endif #include #include "test.h" #ifdef _WIN32 #define mkdir(dir, mode) mkdir(dir) #else typedef uint16_t WCHAR; #define O_BINARY 0 #endif static const char *msifile = "winetest-db.msi"; static const char *msifile2 = "winetst2-db.msi"; static const char *mstfile = "winetst-db.mst"; static const WCHAR msifileW[] = {'w','i','n','e','t','e','s','t','-','d','b','.','m','s','i',0}; static void test_msidatabase(void) { LibmsiDatabase *hdb = 0; LibmsiDatabase *hdb2 = 0; unsigned res; unlink(msifile); hdb = libmsi_database_new( msifile, LIBMSI_DB_FLAGS_TRANSACT, msifile2, NULL); ok(!hdb, "expected failure\n"); /* create an empty database */ hdb = libmsi_database_new(msifile, LIBMSI_DB_FLAGS_CREATE, NULL, NULL ); ok(hdb , "Failed to create database\n" ); res = libmsi_database_commit(hdb, NULL); ok(res, "Failed to commit database\n"); ok( -1 != access( msifile, F_OK ), "database should exist\n"); g_object_unref( hdb ); hdb2 = libmsi_database_new( msifile, LIBMSI_DB_FLAGS_TRANSACT, msifile2, NULL ); ok(hdb2 , "Failed to open database\n" ); res = libmsi_database_commit(hdb2, NULL); ok(res , "Failed to commit database\n"); ok( -1 != access( msifile2, F_OK ), "database should exist\n"); g_object_unref( hdb2 ); hdb2 = libmsi_database_new( msifile, LIBMSI_DB_FLAGS_TRANSACT, msifile2, NULL ); ok( hdb2 , "Failed to open database\n" ); g_object_unref( hdb2 ); ok( -1 == access( msifile2, F_OK ), "uncommitted database should not exist\n"); hdb2 = libmsi_database_new( msifile, LIBMSI_DB_FLAGS_TRANSACT, msifile2, NULL ); ok( hdb2 , "Failed to close database\n" ); res = libmsi_database_commit(hdb2, NULL); ok(res, "Failed to commit database\n"); g_object_unref( hdb2 ); ok( -1 != access( msifile2, F_OK ), "committed database should exist\n"); hdb = libmsi_database_new( msifile, LIBMSI_DB_FLAGS_READONLY, NULL, NULL ); ok(hdb , "Failed to open database\n" ); res = libmsi_database_commit(hdb, NULL); ok(res, "Failed to commit database\n"); g_object_unref( hdb ); hdb = libmsi_database_new( msifile, LIBMSI_DB_FLAGS_TRANSACT, NULL, NULL ); ok(hdb , "Failed to open database\n" ); g_object_unref( hdb ); ok( -1 != access( msifile, F_OK ), "database should exist\n"); unlink( msifile ); /* LIBMSI_DB_FLAGS_CREATE deletes the database if MsiCommitDatabase isn't called */ hdb = libmsi_database_new( msifile, LIBMSI_DB_FLAGS_CREATE, NULL, NULL ); ok(hdb , "Failed to open database\n" ); g_object_unref( hdb ); ok( -1 == access( msifile, F_OK ), "database should not exist\n"); hdb = libmsi_database_new( msifile, LIBMSI_DB_FLAGS_CREATE, NULL, NULL ); ok(hdb , "Failed to open database\n" ); res = libmsi_database_commit(hdb, NULL); ok(res, "Failed to commit database\n", res); ok( -1 != access( msifile, F_OK ), "database should exist\n"); g_object_unref( hdb ); res = unlink( msifile2 ); ok( res == 0, "Failed to delete database\n" ); res = unlink( msifile ); ok( res == 0, "Failed to delete database\n" ); } int do_query(LibmsiDatabase *hdb, const char *sql, LibmsiRecord **rec) { GError *error = NULL; LibmsiQuery *hquery = 0; int ret = LIBMSI_RESULT_SUCCESS; /* open a select query */ hquery = libmsi_query_new(hdb, sql, &error); if (error) goto error; if (!libmsi_query_execute(hquery, 0, &error)) goto error; *rec = libmsi_query_fetch(hquery, &error); if (error) goto error; libmsi_query_close(hquery, NULL); g_object_unref(hquery); return ret; error: ret = error->code; g_clear_error(&error); return ret; } static unsigned run_query( LibmsiDatabase *hdb, LibmsiRecord *hrec, const char *sql ) { GError *error = NULL; LibmsiQuery *hquery = 0; unsigned r = LIBMSI_RESULT_SUCCESS; hquery = libmsi_query_new(hdb, sql, &error); if (error) goto end; if (!libmsi_query_execute(hquery, hrec, &error) || !libmsi_query_close(hquery, &error)) goto end; end: if (error) r = error->code; g_clear_error(&error); if (hquery) g_object_unref(hquery); return r; } static unsigned create_component_table( LibmsiDatabase *hdb ) { return run_query( hdb, 0, "CREATE TABLE `Component` ( " "`Component` CHAR(72) NOT NULL, " "`ComponentId` CHAR(38), " "`Directory_` CHAR(72) NOT NULL, " "`Attributes` SHORT NOT NULL, " "`Condition` CHAR(255), " "`KeyPath` CHAR(72) " "PRIMARY KEY `Component`)" ); } static unsigned create_custom_action_table( LibmsiDatabase *hdb ) { return run_query( hdb, 0, "CREATE TABLE `CustomAction` ( " "`Action` CHAR(72) NOT NULL, " "`Type` SHORT NOT NULL, " "`Source` CHAR(72), " "`Target` CHAR(255) " "PRIMARY KEY `Action`)" ); } static unsigned create_directory_table( LibmsiDatabase *hdb ) { return run_query( hdb, 0, "CREATE TABLE `Directory` ( " "`Directory` CHAR(255) NOT NULL, " "`Directory_Parent` CHAR(255), " "`DefaultDir` CHAR(255) NOT NULL " "PRIMARY KEY `Directory`)" ); } static unsigned create_feature_components_table( LibmsiDatabase *hdb ) { return run_query( hdb, 0, "CREATE TABLE `FeatureComponents` ( " "`Feature_` CHAR(38) NOT NULL, " "`Component_` CHAR(72) NOT NULL " "PRIMARY KEY `Feature_`, `Component_` )" ); } static unsigned create_std_dlls_table( LibmsiDatabase *hdb ) { return run_query( hdb, 0, "CREATE TABLE `StdDlls` ( " "`File` CHAR(255) NOT NULL, " "`Binary_` CHAR(72) NOT NULL " "PRIMARY KEY `File` )" ); } static unsigned create_binary_table( LibmsiDatabase *hdb ) { return run_query( hdb, 0, "CREATE TABLE `Binary` ( " "`Name` CHAR(72) NOT NULL, " "`Data` CHAR(72) NOT NULL " "PRIMARY KEY `Name` )" ); } #define make_add_entry(type, qtext) \ static unsigned add##_##type##_##entry( LibmsiDatabase *hdb, const char *values ) \ { \ char insert[] = qtext; \ char *sql; \ unsigned sz, r; \ sz = strlen(values) + sizeof insert; \ sql = malloc(sz); \ sprintf(sql,insert,values); \ r = run_query( hdb, 0, sql ); \ free(sql); \ return r; \ } make_add_entry(component, "INSERT INTO `Component` " "(`Component`, `ComponentId`, `Directory_`, " "`Attributes`, `Condition`, `KeyPath`) VALUES( %s )") make_add_entry(custom_action, "INSERT INTO `CustomAction` " "(`Action`, `Type`, `Source`, `Target`) VALUES( %s )") make_add_entry(feature_components, "INSERT INTO `FeatureComponents` " "(`Feature_`, `Component_`) VALUES( %s )") make_add_entry(std_dlls, "INSERT INTO `StdDlls` (`File`, `Binary_`) VALUES( %s )") make_add_entry(binary, "INSERT INTO `Binary` (`Name`, `Data`) VALUES( %s )") static void query_check_no_more(LibmsiQuery *query) { GError *error = NULL; LibmsiRecord *hrec; hrec = libmsi_query_fetch(query, &error); g_assert_no_error(error); ok(hrec == NULL, "hrec should be null\n"); } static void test_msiinsert(void) { LibmsiDatabase *hdb = 0; LibmsiQuery *hquery = 0; LibmsiQuery *hquery2 = 0; LibmsiRecord *hrec = 0; gchar *str; unsigned r; const char *sql; char buf[80]; unsigned sz; GError *error = NULL; unlink(msifile); /* just libmsi_database_open should not create a file */ hdb = libmsi_database_new(msifile, LIBMSI_DB_FLAGS_CREATE, NULL, NULL); ok(hdb, "libmsi_database_new failed\n"); /* create a table */ sql = "CREATE TABLE `phone` ( " "`id` INT, `name` CHAR(32), `number` CHAR(32) " "PRIMARY KEY `id`)"; hquery = libmsi_query_new(hdb, sql, &error); ok(hquery, "libmsi_database_open_query failed\n"); r = libmsi_query_execute(hquery, 0, NULL); ok(r, "libmsi_query_execute failed\n"); r = libmsi_query_close(hquery, NULL); ok(r, "libmsi_query_close failed\n"); g_object_unref(hquery); sql = "SELECT * FROM phone WHERE number = '8675309'"; hquery2 = libmsi_query_new(hdb, sql, &error); ok(hquery2, "libmsi_database_open_query failed\n"); r = libmsi_query_execute(hquery2, 0, NULL); ok(r, "libmsi_query_execute failed\n"); query_check_no_more(hquery2); /* insert a value into it */ sql = "INSERT INTO `phone` ( `id`, `name`, `number` )" "VALUES('1', 'Abe', '8675309')"; hquery = libmsi_query_new(hdb, sql, &error); ok(hquery, "libmsi_database_open_query failed\n"); r = libmsi_query_execute(hquery, 0, NULL); ok(r, "libmsi_query_execute failed\n"); r = libmsi_query_close(hquery, NULL); ok(r, "libmsi_query_close failed\n"); g_object_unref(hquery); query_check_no_more(hquery2); r = libmsi_query_execute(hquery2, 0, NULL); ok(r, "libmsi_query_execute failed\n"); hrec = libmsi_query_fetch(hquery2, NULL); ok(hrec, "libmsi_query_fetch failed\n"); g_object_unref(hrec); r = libmsi_query_close(hquery2, NULL); ok(r, "libmsi_query_close failed\n"); g_object_unref(hquery2); sql = "SELECT * FROM `phone` WHERE `id` = 1"; r = do_query(hdb, sql, &hrec); ok(r == LIBMSI_RESULT_SUCCESS, "libmsi_query_fetch failed\n"); /* check the record contains what we put in it */ r = libmsi_record_get_field_count(hrec); ok(r == 3, "record count wrong\n"); r = libmsi_record_is_null(hrec, 0); ok(r == true, "Expected true, got %d\n", r); r = libmsi_record_get_int(hrec, 1); ok(r == 1, "field 1 contents wrong\n"); check_record_string (hrec, 2, "Abe"); check_record_string (hrec, 3, "8675309"); g_object_unref(hrec); /* open a select query */ hrec = NULL; sql = "SELECT * FROM `phone` WHERE `id` >= 10"; r = do_query(hdb, sql, &hrec); ok(r == LIBMSI_RESULT_SUCCESS, "libmsi_query_fetch failed\n"); ok(hrec == NULL, "hrec should be null\n"); sql = "SELECT * FROM `phone` WHERE `id` < 0"; r = do_query(hdb, sql, &hrec); ok(r == LIBMSI_RESULT_SUCCESS, "libmsi_query_fetch failed\n"); ok(hrec == NULL, "hrec should be null\n"); sql = "SELECT * FROM `phone` WHERE `id` <= 0"; r = do_query(hdb, sql, &hrec); ok(r == LIBMSI_RESULT_SUCCESS, "libmsi_query_fetch failed\n"); ok(hrec == NULL, "hrec should be null\n"); sql = "SELECT * FROM `phone` WHERE `id` <> 1"; r = do_query(hdb, sql, &hrec); ok(r == LIBMSI_RESULT_SUCCESS, "libmsi_query_fetch failed\n"); ok(hrec == NULL, "hrec should be null\n"); sql = "SELECT * FROM `phone` WHERE `id` > 10"; r = do_query(hdb, sql, &hrec); ok(r == LIBMSI_RESULT_SUCCESS, "libmsi_query_fetch failed\n"); ok(hrec == NULL, "hrec should be null\n"); /* now try a few bad INSERT xqueries */ sql = "INSERT INTO `phone` ( `id`, `name`, `number` )" "VALUES(?, ?)"; hquery = libmsi_query_new(hdb, sql, &error); g_assert_error(error, LIBMSI_RESULT_ERROR, LIBMSI_RESULT_BAD_QUERY_SYNTAX); g_clear_error(&error); /* construct a record to insert */ hrec = libmsi_record_new(4); r = libmsi_record_set_int(hrec, 1, 2); ok(r, "libmsi_record_set_int failed\n"); r = libmsi_record_set_string(hrec, 2, "Adam"); ok(r, "libmsi_record_set_string failed\n"); r = libmsi_record_set_string(hrec, 3, "96905305"); ok(r, "libmsi_record_set_string failed\n"); /* insert another value, using a record and wildcards */ sql = "INSERT INTO `phone` ( `id`, `name`, `number` )" "VALUES(?, ?, ?)"; hquery = libmsi_query_new(hdb, sql, &error); ok(hquery, "libmsi_database_open_query failed\n"); r = libmsi_query_execute(hquery, hrec, NULL); ok(r, "libmsi_query_execute failed\n"); r = libmsi_query_close(hquery, NULL); ok(r, "libmsi_query_close failed\n"); g_object_unref(hquery); g_object_unref(hrec); hrec = libmsi_query_fetch(0, NULL); ok(!hrec, "libmsi_query_fetch must fail\n"); r = libmsi_database_commit(hdb, NULL); ok(r, "libmsi_database_commit failed\n"); g_object_unref(hdb); r = unlink(msifile); ok(r == 0, "file didn't exist after commit\n"); } static unsigned try_query_param( LibmsiDatabase *hdb, const char *szQuery, LibmsiRecord *hrec ) { GError *error = NULL; LibmsiQuery *htab = 0; unsigned res = LIBMSI_RESULT_SUCCESS; htab = libmsi_query_new( hdb, szQuery, &error ); if(!htab) goto end; if (!libmsi_query_execute (htab, hrec, &error)) goto end; if (!libmsi_query_close (htab, &error)) goto end; end: if (htab) g_object_unref( htab ); if (error) res = error->code; g_clear_error(&error); return res; } static unsigned try_query( LibmsiDatabase *hdb, const char *szQuery ) { return try_query_param( hdb, szQuery, 0 ); } static unsigned try_insert_query( LibmsiDatabase *hdb, const char *szQuery ) { LibmsiRecord *hrec = 0; unsigned r; hrec = libmsi_record_new( 1 ); libmsi_record_set_string( hrec, 1, "Hello"); r = try_query_param( hdb, szQuery, hrec ); g_object_unref( hrec ); return r; } static void test_msibadqueries(void) { LibmsiDatabase *hdb = 0; unsigned r; unlink(msifile); /* just libmsi_database_open should not create a file */ hdb = libmsi_database_new(msifile, LIBMSI_DB_FLAGS_CREATE, NULL, NULL); ok(hdb, "libmsi_database_open failed\n"); r = libmsi_database_commit(hdb, NULL); ok(r, "Failed to commit database\n"); g_object_unref( hdb ); /* open it readonly */ hdb = libmsi_database_new(msifile, LIBMSI_DB_FLAGS_READONLY, NULL, NULL ); ok(hdb , "Failed to open database r/o\n"); /* add a table to it */ r = try_query( hdb, "select * from _Tables"); ok(r == LIBMSI_RESULT_SUCCESS , "query 1 failed\n"); g_object_unref( hdb ); /* open it read/write */ hdb = libmsi_database_new(msifile, LIBMSI_DB_FLAGS_TRANSACT, NULL, NULL ); ok(hdb , "Failed to open database r/w\n"); /* a bunch of test queries that fail with the native MSI */ r = try_query( hdb, "CREATE TABLE"); ok(r == LIBMSI_RESULT_BAD_QUERY_SYNTAX , "invalid query 2a return code\n"); r = try_query( hdb, "CREATE TABLE `a`"); ok(r == LIBMSI_RESULT_BAD_QUERY_SYNTAX , "invalid query 2b return code\n"); r = try_query( hdb, "CREATE TABLE `a` ()"); ok(r == LIBMSI_RESULT_BAD_QUERY_SYNTAX , "invalid query 2c return code\n"); r = try_query( hdb, "CREATE TABLE `a` (`b`)"); ok(r == LIBMSI_RESULT_BAD_QUERY_SYNTAX , "invalid query 2d return code\n"); r = try_query( hdb, "CREATE TABLE `a` (`b` CHAR(72) )"); ok(r == LIBMSI_RESULT_BAD_QUERY_SYNTAX , "invalid query 2e return code\n"); r = try_query( hdb, "CREATE TABLE `a` (`b` CHAR(72) NOT NULL)"); ok(r == LIBMSI_RESULT_BAD_QUERY_SYNTAX , "invalid query 2f return code\n"); r = try_query( hdb, "CREATE TABLE `a` (`b` CHAR(72) NOT NULL PRIMARY)"); ok(r == LIBMSI_RESULT_BAD_QUERY_SYNTAX , "invalid query 2g return code\n"); r = try_query( hdb, "CREATE TABLE `a` (`b` CHAR(72) NOT NULL PRIMARY KEY)"); ok(r == LIBMSI_RESULT_BAD_QUERY_SYNTAX , "invalid query 2h return code\n"); r = try_query( hdb, "CREATE TABLE `a` (`b` CHAR(72) NOT NULL PRIMARY KEY)"); ok(r == LIBMSI_RESULT_BAD_QUERY_SYNTAX , "invalid query 2i return code\n"); r = try_query( hdb, "CREATE TABLE `a` (`b` CHAR(72) NOT NULL PRIMARY KEY 'b')"); ok(r == LIBMSI_RESULT_BAD_QUERY_SYNTAX , "invalid query 2j return code\n"); r = try_query( hdb, "CREATE TABLE `a` (`b` CHAR(72) NOT NULL PRIMARY KEY `b')"); ok(r == LIBMSI_RESULT_BAD_QUERY_SYNTAX , "invalid query 2k return code\n"); r = try_query( hdb, "CREATE TABLE `a` (`b` CHAR(72) NOT NULL PRIMARY KEY `b')"); ok(r == LIBMSI_RESULT_BAD_QUERY_SYNTAX , "invalid query 2l return code\n"); r = try_query( hdb, "CREATE TABLE `a` (`b` CHA(72) NOT NULL PRIMARY KEY `b`)"); ok(r == LIBMSI_RESULT_BAD_QUERY_SYNTAX , "invalid query 2m return code\n"); r = try_query( hdb, "CREATE TABLE `a` (`b` CHAR(-1) NOT NULL PRIMARY KEY `b`)"); ok(r == LIBMSI_RESULT_BAD_QUERY_SYNTAX , "invalid query 2n return code\n"); r = try_query( hdb, "CREATE TABLE `a` (`b` CHAR(720) NOT NULL PRIMARY KEY `b`)"); ok(r == LIBMSI_RESULT_BAD_QUERY_SYNTAX , "invalid query 2o return code\n"); r = try_query( hdb, "CREATE TABLE `a` (`b` CHAR(72) NOT NULL KEY `b`)"); ok(r == LIBMSI_RESULT_BAD_QUERY_SYNTAX , "invalid query 2p return code\n"); r = try_query( hdb, "CREATE TABLE `a` (`` CHAR(72) NOT NULL PRIMARY KEY `b`)"); ok(r == LIBMSI_RESULT_BAD_QUERY_SYNTAX , "invalid query 2p return code\n"); r = try_query( hdb, "CREATE TABLE `a` (`b` CHAR(72) NOT NULL PRIMARY KEY `b`)"); ok(r == LIBMSI_RESULT_SUCCESS , "valid query 2z failed\n"); r = try_query( hdb, "CREATE TABLE `a` (`b` CHAR(72) NOT NULL PRIMARY KEY `b`)"); ok(r == LIBMSI_RESULT_BAD_QUERY_SYNTAX , "created same table again\n"); r = try_query( hdb, "CREATE TABLE `aa` (`b` CHAR(72) NOT NULL, `c` " "CHAR(72), `d` CHAR(255) NOT NULL LOCALIZABLE PRIMARY KEY `b`)"); ok(r == LIBMSI_RESULT_SUCCESS , "query 4 failed\n"); r = libmsi_database_commit(hdb, NULL); ok(r, "Failed to commit database after write\n"); r = try_query( hdb, "CREATE TABLE `blah` (`foo` CHAR(72) NOT NULL " "PRIMARY KEY `foo`)"); ok(r == LIBMSI_RESULT_SUCCESS , "query 4 failed\n"); r = try_insert_query( hdb, "insert into a ( `b` ) VALUES ( ? )"); ok(r == LIBMSI_RESULT_SUCCESS , "failed to insert record in db\n"); r = libmsi_database_commit(hdb, NULL); ok(r, "Failed to commit database after write\n"); r = try_query( hdb, "CREATE TABLE `boo` (`foo` CHAR(72) NOT NULL " "PRIMARY KEY `ba`)"); ok(r != LIBMSI_RESULT_SUCCESS , "query 5 succeeded\n"); r = try_query( hdb,"CREATE TABLE `bee` (`foo` CHAR(72) NOT NULL )"); ok(r != LIBMSI_RESULT_SUCCESS , "query 6 succeeded\n"); r = try_query( hdb, "CREATE TABLE `temp` (`t` CHAR(72) NOT NULL " "PRIMARY KEY `t`)"); ok(r == LIBMSI_RESULT_SUCCESS , "query 7 failed\n"); r = try_query( hdb, "CREATE TABLE `c` (`b` CHAR NOT NULL PRIMARY KEY `b`)"); ok(r == LIBMSI_RESULT_SUCCESS , "query 8 failed\n"); r = try_query( hdb, "select * from c"); ok(r == LIBMSI_RESULT_SUCCESS , "query failed\n"); r = try_query( hdb, "select * from c where b = 'x"); ok(r == LIBMSI_RESULT_BAD_QUERY_SYNTAX, "query failed\n"); r = try_query( hdb, "select * from c where b = 'x'"); ok(r == LIBMSI_RESULT_SUCCESS, "query failed\n"); r = try_query( hdb, "select * from 'c'"); ok(r == LIBMSI_RESULT_BAD_QUERY_SYNTAX, "query failed\n"); r = try_query( hdb, "select * from ''"); ok(r == LIBMSI_RESULT_BAD_QUERY_SYNTAX, "query failed\n"); r = try_query( hdb, "select * from c where b = x"); ok(r == LIBMSI_RESULT_BAD_QUERY_SYNTAX, "query failed\n"); r = try_query( hdb, "select * from c where b = \"x\""); ok(r == LIBMSI_RESULT_BAD_QUERY_SYNTAX, "query failed\n"); r = try_query( hdb, "select * from c where b = 'x'"); ok(r == LIBMSI_RESULT_SUCCESS, "query failed\n"); r = try_query( hdb, "select * from c where b = '\"x'"); ok(r == LIBMSI_RESULT_SUCCESS, "query failed\n"); if (0) /* FIXME: this query causes trouble with other tests */ { r = try_query( hdb, "select * from c where b = '\\\'x'"); ok(r == LIBMSI_RESULT_BAD_QUERY_SYNTAX, "query failed\n"); } r = try_query( hdb, "select * from 'c'"); ok(r == LIBMSI_RESULT_BAD_QUERY_SYNTAX, "query failed\n"); r = try_query( hdb, "select `c`.`b` from `c` order by `c`.`order`"); ok( r == LIBMSI_RESULT_BAD_QUERY_SYNTAX, "query failed: %u\n", r ); r = try_query( hdb, "select `c`.b` from `c`"); ok( r == LIBMSI_RESULT_SUCCESS, "query failed: %u\n", r ); r = try_query( hdb, "select `c`.`b from `c`"); ok( r == LIBMSI_RESULT_BAD_QUERY_SYNTAX, "query failed: %u\n", r ); r = try_query( hdb, "select `c`.b from `c`"); ok( r == LIBMSI_RESULT_SUCCESS, "query failed: %u\n", r ); r = try_query( hdb, "select `c.`b` from `c`"); ok( r == LIBMSI_RESULT_BAD_QUERY_SYNTAX, "query failed: %u\n", r ); r = try_query( hdb, "select c`.`b` from `c`"); ok( r == LIBMSI_RESULT_SUCCESS, "query failed: %u\n", r ); r = try_query( hdb, "select c.`b` from `c`"); ok( r == LIBMSI_RESULT_SUCCESS, "query failed: %u\n", r ); r = try_query( hdb, "select `c`.`b` from c`"); ok( r == LIBMSI_RESULT_SUCCESS, "query failed: %u\n", r ); r = try_query( hdb, "select `c`.`b` from `c"); ok( r == LIBMSI_RESULT_BAD_QUERY_SYNTAX, "query failed: %u\n", r ); r = try_query( hdb, "select `c`.`b` from c"); ok( r == LIBMSI_RESULT_SUCCESS, "query failed: %u\n", r ); r = try_query( hdb, "CREATE TABLE `\5a` (`b` CHAR NOT NULL PRIMARY KEY `b`)" ); ok( r == LIBMSI_RESULT_SUCCESS , "query failed: %u\n", r ); r = try_query( hdb, "SELECT * FROM \5a" ); ok( r == LIBMSI_RESULT_SUCCESS , "query failed: %u\n", r ); r = try_query( hdb, "CREATE TABLE `a\5` (`b` CHAR NOT NULL PRIMARY KEY `b`)" ); ok( r == LIBMSI_RESULT_SUCCESS , "query failed: %u\n", r ); r = try_query( hdb, "SELECT * FROM a\5" ); ok( r == LIBMSI_RESULT_SUCCESS , "query failed: %u\n", r ); r = try_query( hdb, "CREATE TABLE `-a` (`b` CHAR NOT NULL PRIMARY KEY `b`)" ); ok( r == LIBMSI_RESULT_SUCCESS , "query failed: %u\n", r ); r = try_query( hdb, "SELECT * FROM -a" ); todo_wine ok( r == LIBMSI_RESULT_SUCCESS , "query failed: %u\n", r ); r = try_query( hdb, "CREATE TABLE `a-` (`b` CHAR NOT NULL PRIMARY KEY `b`)" ); ok( r == LIBMSI_RESULT_SUCCESS , "query failed: %u\n", r ); r = try_query( hdb, "SELECT * FROM a-" ); ok( r == LIBMSI_RESULT_SUCCESS , "query failed: %u\n", r ); g_object_unref( hdb ); r = unlink( msifile ); ok(r == 0, "file didn't exist after commit\n"); } static LibmsiDatabase *create_db(void) { LibmsiDatabase *hdb = 0; unsigned res; unlink(msifile); /* create an empty database */ hdb = libmsi_database_new(msifile, LIBMSI_DB_FLAGS_CREATE, NULL, NULL ); ok(hdb , "Failed to create database\n" ); res = libmsi_database_commit(hdb, NULL); ok(res, "Failed to commit database\n"); return hdb; } static void test_getcolinfo(void) { LibmsiDatabase *hdb; LibmsiQuery *hquery = 0; LibmsiRecord *rec = 0; unsigned r; unsigned sz; char buffer[0x20]; /* create an empty db */ hdb = create_db(); ok( hdb, "failed to create db\n"); /* tables should be present */ hquery = libmsi_query_new(hdb, "select * from _Tables", NULL); ok(hquery, "failed to open query\n"); r = libmsi_query_execute(hquery, 0, NULL); ok( r, "failed to execute query\n"); /* check that NAMES works */ rec = libmsi_query_get_column_info( hquery, LIBMSI_COL_INFO_NAMES, NULL ); ok(rec, "failed to get names\n"); check_record_string (rec, 1, "Name"); g_object_unref( rec ); /* check that TYPES works */ rec = libmsi_query_get_column_info( hquery, LIBMSI_COL_INFO_TYPES, NULL ); ok(rec, "failed to get names\n"); check_record_string (rec, 1, "s64"); g_object_unref( rec ); /* check that invalid values fail */ rec = libmsi_query_get_column_info( hquery, 100, NULL); ok(!rec, "returned a record\n"); rec = libmsi_query_get_column_info( 0, LIBMSI_COL_INFO_TYPES, NULL); ok(!rec, "wrong return value\n"); r = libmsi_query_close(hquery, NULL); ok(r, "failed to close query\n"); g_object_unref(hquery); g_object_unref(hdb); } static LibmsiRecord *get_column_info(LibmsiDatabase *hdb, const char *sql, LibmsiColInfo type) { LibmsiQuery *hquery = 0; LibmsiRecord *rec = 0; unsigned r; hquery = libmsi_query_new(hdb, sql, NULL); if(!hquery) return NULL; r = libmsi_query_execute(hquery, 0, NULL); if( r ) { rec = libmsi_query_get_column_info( hquery, type, NULL ); } libmsi_query_close(hquery, NULL); g_object_unref(hquery); return rec; } static unsigned get_columns_table_type(LibmsiDatabase *hdb, const char *table, unsigned field) { GError *error = NULL; LibmsiQuery *hquery = 0; LibmsiRecord *rec = 0; unsigned r, type = 0; char sql[0x100]; sprintf(sql, "select * from `_Columns` where `Table` = '%s'", table ); hquery = libmsi_query_new(hdb, sql, NULL); if (!hquery) return type; r = libmsi_query_execute(hquery, 0, NULL); if( r ) { while (1) { rec = libmsi_query_fetch(hquery, NULL); if (!rec) break; r = libmsi_record_get_int( rec, 2 ); if (r == field) type = libmsi_record_get_int( rec, 4 ); g_object_unref( rec ); } } libmsi_query_close(hquery, NULL); g_object_unref(hquery); return type; } static bool check_record( LibmsiRecord *rec, unsigned field, const char *val ) { bool result; gchar *str; str = libmsi_record_get_string(rec, field); result = g_strcmp0(val, str) == 0; g_free(str); return result; } static void test_querygetcolumninfo(void) { LibmsiDatabase *hdb = 0; LibmsiRecord *rec; unsigned r; hdb = create_db(); ok( hdb, "failed to create db\n"); r = run_query( hdb, 0, "CREATE TABLE `Properties` " "( `Property` CHAR(255), " " `Value` CHAR(1), " " `Intvalue` INT, " " `Integervalue` INTEGER, " " `Shortvalue` SHORT, " " `Longvalue` LONG, " " `Longcharvalue` LONGCHAR " " PRIMARY KEY `Property`)" ); ok( r == LIBMSI_RESULT_SUCCESS , "Failed to create table\n" ); /* check the column types */ rec = get_column_info( hdb, "select * from `Properties`", LIBMSI_COL_INFO_TYPES ); ok( rec, "failed to get column info record\n" ); ok( check_record( rec, 1, "S255"), "wrong record type\n"); ok( check_record( rec, 2, "S1"), "wrong record type\n"); ok( check_record( rec, 3, "I2"), "wrong record type\n"); ok( check_record( rec, 4, "I2"), "wrong record type\n"); ok( check_record( rec, 5, "I2"), "wrong record type\n"); ok( check_record( rec, 6, "I4"), "wrong record type\n"); ok( check_record( rec, 7, "S0"), "wrong record type\n"); g_object_unref( rec ); /* check the type in _Columns */ ok( 0x3dff == get_columns_table_type(hdb, "Properties", 1 ), "_columns table wrong\n"); ok( 0x1d01 == get_columns_table_type(hdb, "Properties", 2 ), "_columns table wrong\n"); ok( 0x1502 == get_columns_table_type(hdb, "Properties", 3 ), "_columns table wrong\n"); ok( 0x1502 == get_columns_table_type(hdb, "Properties", 4 ), "_columns table wrong\n"); ok( 0x1502 == get_columns_table_type(hdb, "Properties", 5 ), "_columns table wrong\n"); ok( 0x1104 == get_columns_table_type(hdb, "Properties", 6 ), "_columns table wrong\n"); ok( 0x1d00 == get_columns_table_type(hdb, "Properties", 7 ), "_columns table wrong\n"); /* now try the names */ rec = get_column_info( hdb, "select * from `Properties`", LIBMSI_COL_INFO_NAMES ); ok( rec, "failed to get column info record\n" ); ok( check_record( rec, 1, "Property"), "wrong record type\n"); ok( check_record( rec, 2, "Value"), "wrong record type\n"); ok( check_record( rec, 3, "Intvalue"), "wrong record type\n"); ok( check_record( rec, 4, "Integervalue"), "wrong record type\n"); ok( check_record( rec, 5, "Shortvalue"), "wrong record type\n"); ok( check_record( rec, 6, "Longvalue"), "wrong record type\n"); ok( check_record( rec, 7, "Longcharvalue"), "wrong record type\n"); g_object_unref( rec ); r = run_query( hdb, 0, "CREATE TABLE `Binary` " "( `Name` CHAR(255), `Data` OBJECT PRIMARY KEY `Name`)" ); ok( r == LIBMSI_RESULT_SUCCESS , "Failed to create table\n" ); /* check the column types */ rec = get_column_info( hdb, "select * from `Binary`", LIBMSI_COL_INFO_TYPES ); ok( rec, "failed to get column info record\n" ); ok( check_record( rec, 1, "S255"), "wrong record type\n"); ok( check_record( rec, 2, "V0"), "wrong record type\n"); g_object_unref( rec ); /* check the type in _Columns */ ok( 0x3dff == get_columns_table_type(hdb, "Binary", 1 ), "_columns table wrong\n"); ok( 0x1900 == get_columns_table_type(hdb, "Binary", 2 ), "_columns table wrong\n"); /* now try the names */ rec = get_column_info( hdb, "select * from `Binary`", LIBMSI_COL_INFO_NAMES ); ok( rec, "failed to get column info record\n" ); ok( check_record( rec, 1, "Name"), "wrong record type\n"); ok( check_record( rec, 2, "Data"), "wrong record type\n"); g_object_unref( rec ); r = run_query( hdb, 0, "CREATE TABLE `UIText` " "( `Key` CHAR(72) NOT NULL, `Text` CHAR(255) LOCALIZABLE PRIMARY KEY `Key`)" ); ok( r == LIBMSI_RESULT_SUCCESS , "Failed to create table\n" ); ok( 0x2d48 == get_columns_table_type(hdb, "UIText", 1 ), "_columns table wrong\n"); ok( 0x1fff == get_columns_table_type(hdb, "UIText", 2 ), "_columns table wrong\n"); rec = get_column_info( hdb, "select * from `UIText`", LIBMSI_COL_INFO_NAMES ); ok( rec, "failed to get column info record\n" ); ok( check_record( rec, 1, "Key"), "wrong record type\n"); ok( check_record( rec, 2, "Text"), "wrong record type\n"); g_object_unref( rec ); rec = get_column_info( hdb, "select * from `UIText`", LIBMSI_COL_INFO_TYPES ); ok( rec, "failed to get column info record\n" ); ok( check_record( rec, 1, "s72"), "wrong record type\n"); ok( check_record( rec, 2, "L255"), "wrong record type\n"); g_object_unref( rec ); g_object_unref( hdb ); } static void test_msiexport(void) { LibmsiDatabase *hdb = 0; LibmsiQuery *hquery = 0; unsigned r; const char *sql; int fd; const char file[] = "phone.txt"; char buffer[0x100]; unsigned length; const char expected[] = "id\tname\tnumber\r\n" "I2\tS32\tS32\r\n" "phone\tid\r\n" "1\tAbe\t8675309\r\n"; unlink(msifile); /* just libmsi_database_open should not create a file */ hdb = libmsi_database_new(msifile, LIBMSI_DB_FLAGS_CREATE, NULL, NULL); ok(hdb, "libmsi_database_open failed\n"); /* create a table */ sql = "CREATE TABLE `phone` ( " "`id` INT, `name` CHAR(32), `number` CHAR(32) " "PRIMARY KEY `id`)"; hquery = libmsi_query_new(hdb, sql, NULL); ok(hquery, "libmsi_database_open_query failed\n"); r = libmsi_query_execute(hquery, 0, NULL); ok(r, "libmsi_query_execute failed\n"); r = libmsi_query_close(hquery, NULL); ok(r, "libmsi_query_close failed\n"); g_object_unref(hquery); /* insert a value into it */ sql = "INSERT INTO `phone` ( `id`, `name`, `number` )" "VALUES('1', 'Abe', '8675309')"; hquery = libmsi_query_new(hdb, sql, NULL); ok(hquery, "libmsi_database_open_query failed\n"); r = libmsi_query_execute(hquery, 0, NULL); ok(r, "libmsi_query_execute failed\n"); r = libmsi_query_close(hquery, NULL); ok(r, "libmsi_query_close failed\n"); g_object_unref(hquery); fd = open(file, O_WRONLY | O_BINARY | O_CREAT, 0644); ok(fd != -1, "open failed\n"); r = libmsi_database_export(hdb, "phone", fd, NULL); ok(r, "libmsi_database_export failed\n"); close(fd); g_object_unref(hdb); /* check the data that was written */ length = 0; memset(buffer, 0, sizeof buffer); fd = open(file, O_RDONLY | O_BINARY); if (fd != -1) { length = read(fd, buffer, sizeof buffer); close(fd); unlink(file); } else ok(0, "failed to open file %s\n", file); ok( length == strlen(expected), "length of data wrong\n"); ok( g_str_equal(buffer, expected), "data doesn't match\n"); unlink(msifile); } static void test_longstrings(void) { const char insert_query[] = "INSERT INTO `strings` ( `id`, `val` ) VALUES('1', 'Z')"; char *str; LibmsiDatabase *hdb = 0; LibmsiQuery *hquery = 0; LibmsiRecord *hrec = 0; unsigned len; unsigned r; const unsigned STRING_LENGTH = 0x10005; unlink(msifile); /* just libmsi_database_open should not create a file */ hdb = libmsi_database_new(msifile, LIBMSI_DB_FLAGS_CREATE, NULL, NULL); ok(hdb, "libmsi_database_open failed\n"); /* create a table */ r = try_query( hdb, "CREATE TABLE `strings` ( `id` INT, `val` CHAR(0) PRIMARY KEY `id`)"); ok(r == LIBMSI_RESULT_SUCCESS, "query failed\n"); /* try a insert a very long string */ str = malloc(STRING_LENGTH+sizeof insert_query); len = strchr(insert_query, 'Z') - insert_query; strcpy(str, insert_query); memset(str+len, 'Z', STRING_LENGTH); strcpy(str+len+STRING_LENGTH, insert_query+len+1); r = try_query( hdb, str ); ok(r == LIBMSI_RESULT_SUCCESS, "libmsi_database_open_query failed\n"); r = libmsi_database_commit(hdb, NULL); ok(r, "libmsi_database_commit failed\n"); g_object_unref(hdb); hdb = libmsi_database_new(msifile, LIBMSI_DB_FLAGS_READONLY, NULL, NULL); ok(hdb, "libmsi_database_open failed\n"); hquery = libmsi_query_new(hdb, "select * from `strings` where `id` = 1", NULL); ok(hquery, "libmsi_database_open_query failed\n"); r = libmsi_query_execute(hquery, 0, NULL); ok(r, "libmsi_query_execute failed\n"); hrec = libmsi_query_fetch(hquery, NULL); ok(hrec, "libmsi_query_fetch failed"); libmsi_query_close(hquery, NULL); g_object_unref(hquery); str[len+STRING_LENGTH] = '\0'; check_record_string (hrec, 2, str+len); g_free(str); g_object_unref(hrec); g_object_unref(hdb); unlink(msifile); } static void create_file_data(const char *name, const char *data, unsigned size) { int file; file = open(name, O_CREAT | O_WRONLY | O_BINARY, 0644); if (file == -1) return; write(file, data, size ? size : strlen(data)); if (size == 0) write(file, "\n", strlen("\n")); assert(size == 0 || size == lseek(file, 0, SEEK_CUR)); close(file); } #define create_file(name) create_file_data(name, name, 0) static void test_streamtable(void) { GError *error = NULL; GInputStream *in; LibmsiDatabase *hdb = 0; LibmsiRecord *rec; LibmsiQuery *query; LibmsiSummaryInfo *hsi; char file[256]; char buf[256]; unsigned size; unsigned r; hdb = create_db(); ok( hdb, "failed to create db\n"); r = run_query( hdb, 0, "CREATE TABLE `Properties` " "( `Property` CHAR(255), `Value` CHAR(1) PRIMARY KEY `Property`)" ); ok( r == LIBMSI_RESULT_SUCCESS , "Failed to create table\n" ); r = run_query( hdb, 0, "INSERT INTO `Properties` " "( `Value`, `Property` ) VALUES ( 'Prop', 'value' )" ); ok( r == LIBMSI_RESULT_SUCCESS, "Failed to add to table\n" ); r = libmsi_database_commit(hdb, NULL); ok(r, "Failed to commit database\n"); g_object_unref( hdb ); hdb = libmsi_database_new(msifile, LIBMSI_DB_FLAGS_TRANSACT, NULL, NULL ); ok(hdb , "Failed to open database\n" ); /* check the column types */ rec = get_column_info( hdb, "select * from `_Streams`", LIBMSI_COL_INFO_TYPES ); ok( rec, "failed to get column info record\n" ); ok( check_record( rec, 1, "s62"), "wrong record type\n"); ok( check_record( rec, 2, "V0"), "wrong record type\n"); g_object_unref( rec ); /* now try the names */ rec = get_column_info( hdb, "select * from `_Streams`", LIBMSI_COL_INFO_NAMES ); ok( rec, "failed to get column info record\n" ); ok( check_record( rec, 1, "Name"), "wrong record type\n"); ok( check_record( rec, 2, "Data"), "wrong record type\n"); g_object_unref( rec ); query = libmsi_query_new( hdb, "SELECT * FROM `_Streams` WHERE `Name` = '\5SummaryInformation'", NULL ); ok(query, "Failed to open database query\n"); r = libmsi_query_execute( query, 0 , NULL); ok( r, "Failed to execute query: %u\n", r ); query_check_no_more(query); libmsi_query_close(query, NULL); g_object_unref( query ); /* create a summary information stream */ hsi = libmsi_summary_info_new (hdb, 1, NULL); ok(hsi, "Failed to get summary information handle: %u\n", r ); libmsi_summary_info_set_int(hsi, LIBMSI_PROPERTY_SECURITY, 2, &error); ok(!error, "Failed to set property\n"); r = libmsi_summary_info_persist( hsi, NULL ); ok(r, "Failed to save summary information: %u\n", r ); g_object_unref( hsi ); query = NULL; query = libmsi_query_new( hdb, "SELECT * FROM `_Streams` WHERE `Name` = '\5SummaryInformation'", NULL ); ok(query, "Failed to open database query\n"); r = libmsi_query_execute( query, 0 , NULL); ok( r, "Failed to execute query: %u\n", r ); rec = libmsi_query_fetch(query, NULL); ok(rec, "Unexpected result\n"); g_object_unref( rec ); libmsi_query_close(query, NULL); g_object_unref( query ); /* insert a file into the _Streams table */ create_file( "test.txt" ); rec = libmsi_record_new( 2 ); libmsi_record_set_string( rec, 1, "data" ); r = libmsi_record_load_stream( rec, 2, "test.txt" ); ok(r, "Failed to add stream data to the record: %d\n", r); unlink("test.txt"); query = libmsi_query_new( hdb, "INSERT INTO `_Streams` ( `Name`, `Data` ) VALUES ( ?, ? )", NULL ); ok(query, "Failed to open database query\n"); r = libmsi_query_execute( query, rec , NULL); ok( r, "Failed to execute query: %d\n", r); g_object_unref( rec ); libmsi_query_close(query, NULL); g_object_unref( query ); /* insert another one */ create_file( "test1.txt" ); rec = libmsi_record_new( 2 ); libmsi_record_set_string( rec, 1, "data1" ); r = libmsi_record_load_stream( rec, 2, "test1.txt" ); ok(r, "Failed to add stream data to the record: %d\n", r); unlink("test1.txt"); query = libmsi_query_new( hdb, "INSERT INTO `_Streams` ( `Name`, `Data` ) VALUES ( ?, ? )", NULL ); ok(query, "Failed to open database query\n"); r = libmsi_query_execute( query, rec , NULL); ok( r, "Failed to execute query: %d\n", r); g_object_unref( rec ); libmsi_query_close(query, NULL); g_object_unref( query ); query = libmsi_query_new( hdb, "SELECT `Name`, `Data` FROM `_Streams` WHERE `Name` = 'data'", NULL ); ok(query, "Failed to open database query\n"); r = libmsi_query_execute( query, 0 , NULL); ok( r, "Failed to execute query: %d\n", r); rec = libmsi_query_fetch(query, NULL); ok(rec, "Failed to fetch record\n"); check_record_string (rec, 1, "data"); size = sizeof(buf); memset(buf, 0, sizeof(buf)); in = libmsi_record_get_stream(rec, 2); ok(in, "Failed to get stream\n"); size = g_input_stream_read(in, buf, sizeof(buf), NULL, NULL); ok( g_str_equal(buf, "test.txt\n"), "Expected 'test.txt\\n', got %s\n", buf); g_object_unref(in); g_object_unref( rec ); libmsi_query_close(query, NULL); g_object_unref( query ); query = libmsi_query_new( hdb, "SELECT `Name`, `Data` FROM `_Streams` WHERE `Name` = 'data1'", NULL); ok(query, "Failed to open database query\n"); r = libmsi_query_execute( query, 0 , NULL); ok( r, "Failed to execute query: %d\n", r); rec = libmsi_query_fetch(query, NULL); ok(rec, "Expected result"); check_record_string (rec, 1, "data1"); size = sizeof(buf); memset(buf, 0, sizeof(buf)); in = libmsi_record_get_stream(rec, 2); ok(in, "Failed to get stream\n"); size = g_input_stream_read(in, buf, sizeof(buf), NULL, NULL); ok( g_str_equal(buf, "test1.txt\n"), "Expected 'test1.txt\\n', got %s\n", buf); g_object_unref(in); g_object_unref( rec ); libmsi_query_close(query, NULL); g_object_unref( query ); /* perform an update */ create_file( "test2.txt" ); rec = libmsi_record_new( 1 ); r = libmsi_record_load_stream( rec, 1, "test2.txt" ); ok(r, "Failed to add stream data to the record: %d\n", r); unlink("test2.txt"); query = libmsi_query_new( hdb, "UPDATE `_Streams` SET `Data` = ? WHERE `Name` = 'data1'", NULL); ok(query, "Failed to open database query\n"); r = libmsi_query_execute( query, rec , NULL); ok( r, "Failed to execute query: %d\n", r); g_object_unref( rec ); libmsi_query_close(query, NULL); g_object_unref( query ); query = libmsi_query_new( hdb, "SELECT `Name`, `Data` FROM `_Streams` WHERE `Name` = 'data1'", NULL); ok(query, "Failed to open database query\n"); r = libmsi_query_execute( query, 0 , NULL); ok( r, "Failed to execute query: %d\n", r); rec = libmsi_query_fetch(query, NULL); ok(rec, "Failed to fetch record\n"); check_record_string (rec, 1, "data1"); size = sizeof(buf); memset(buf, 0, sizeof(buf)); in = libmsi_record_get_stream(rec, 2); ok(in, "Failed to get stream\n"); size = g_input_stream_read(in, buf, sizeof(buf), NULL, NULL); todo_wine ok( g_str_equal(buf, "test2.txt\n"), "Expected 'test2.txt\\n', got %s\n", buf); g_object_unref(in); g_object_unref( rec ); libmsi_query_close(query, NULL); g_object_unref( query ); r = run_query( hdb, 0, "DELETE FROM `_Streams` WHERE `Name` = 'data1'" ); ok( r == LIBMSI_RESULT_SUCCESS, "Cannot create Binary table: %d\n", r ); query = libmsi_query_new( hdb, "SELECT `Name`, `Data` FROM `_Streams` WHERE `Name` = 'data1'", NULL); ok(query, "Failed to open database query\n"); r = libmsi_query_execute( query, 0 , NULL); ok( r, "Failed to execute query: %d\n", r); query_check_no_more(query); libmsi_query_close(query, NULL); g_object_unref( query ); g_object_unref( hdb ); unlink(msifile); } static void test_binary(void) { GInputStream *in; LibmsiDatabase *hdb = 0; LibmsiRecord *rec; char file[256]; char buf[256]; unsigned size; const char *sql; unsigned r; /* insert a file into the Binary table */ hdb = libmsi_database_new(msifile, LIBMSI_DB_FLAGS_CREATE, NULL, NULL ); ok(hdb , "Failed to open database\n" ); sql = "CREATE TABLE `Binary` ( `Name` CHAR(72) NOT NULL, `ID` INT NOT NULL, `Data` OBJECT PRIMARY KEY `Name`, `ID`)"; r = run_query( hdb, 0, sql ); ok( r == LIBMSI_RESULT_SUCCESS, "Cannot create Binary table: %d\n", r ); create_file( "test.txt" ); rec = libmsi_record_new( 1 ); r = libmsi_record_load_stream( rec, 1, "test.txt" ); ok(r, "Failed to add stream data to the record: %d\n", r); unlink( "test.txt" ); sql = "INSERT INTO `Binary` ( `Name`, `ID`, `Data` ) VALUES ( 'filename1', 1, ? )"; r = run_query( hdb, rec, sql ); ok( r == LIBMSI_RESULT_SUCCESS, "Insert into Binary table failed: %d\n", r ); g_object_unref( rec ); r = libmsi_database_commit(hdb, NULL); ok(r, "Failed to commit database\n" ); g_object_unref( hdb ); /* read file from the Stream table */ hdb = libmsi_database_new( msifile, LIBMSI_DB_FLAGS_READONLY, NULL, NULL ); ok(hdb , "Failed to open database\n" ); sql = "SELECT * FROM `_Streams`"; r = do_query( hdb, sql, &rec ); ok( r == LIBMSI_RESULT_SUCCESS, "SELECT query failed: %d\n", r ); check_record_string (rec, 1, "Binary.filename1.1"); size = sizeof(buf); memset( buf, 0, sizeof(buf) ); in = libmsi_record_get_stream(rec, 2); ok(in, "Failed to get stream\n"); size = g_input_stream_read(in, buf, sizeof(buf), NULL, NULL); ok( g_str_equal(buf, "test.txt\n"), "Expected 'test.txt\\n', got %s\n", buf ); g_object_unref(in); g_object_unref( rec ); /* read file from the Binary table */ sql = "SELECT * FROM `Binary`"; r = do_query( hdb, sql, &rec ); ok( r == LIBMSI_RESULT_SUCCESS, "SELECT query failed: %d\n", r ); check_record_string (rec, 1, "filename1"); size = sizeof(buf); memset( buf, 0, sizeof(buf) ); in = libmsi_record_get_stream(rec, 3); ok(in, "Failed to get stream\n"); size = g_input_stream_read(in, buf, sizeof(buf), NULL, NULL); ok( g_str_equal(buf, "test.txt\n"), "Expected 'test.txt\\n', got %s\n", buf ); g_object_unref(in); g_object_unref( rec ); g_object_unref( hdb ); unlink( msifile ); } static void test_where_not_in_selected(void) { LibmsiDatabase *hdb = 0; LibmsiRecord *rec; LibmsiQuery *query; const char *sql; unsigned r; hdb = create_db(); ok( hdb, "failed to create db\n"); r = run_query(hdb, 0, "CREATE TABLE `IESTable` (" "`Action` CHAR(64), " "`Condition` CHAR(64), " "`Sequence` LONG PRIMARY KEY `Sequence`)"); ok( r == LIBMSI_RESULT_SUCCESS, "Cannot create IESTable table: %d\n", r); r = run_query(hdb, 0, "CREATE TABLE `CATable` (" "`Action` CHAR(64), " "`Type` LONG PRIMARY KEY `Type`)"); ok( r == LIBMSI_RESULT_SUCCESS, "Cannot create CATable table: %d\n", r); r = run_query(hdb, 0, "INSERT INTO `IESTable` " "( `Action`, `Condition`, `Sequence`) " "VALUES ( 'clean', 'cond4', 4)"); ok( r == LIBMSI_RESULT_SUCCESS, "cannot add entry to IESTable table:%d\n", r ); r = run_query(hdb, 0, "INSERT INTO `IESTable` " "( `Action`, `Condition`, `Sequence`) " "VALUES ( 'depends', 'cond1', 1)"); ok( r == LIBMSI_RESULT_SUCCESS, "cannot add entry to IESTable table:%d\n", r ); r = run_query(hdb, 0, "INSERT INTO `IESTable` " "( `Action`, `Condition`, `Sequence`) " "VALUES ( 'build', 'cond2', 2)"); ok( r == LIBMSI_RESULT_SUCCESS, "cannot add entry to IESTable table:%d\n", r ); r = run_query(hdb, 0, "INSERT INTO `IESTable` " "( `Action`, `Condition`, `Sequence`) " "VALUES ( 'build2', 'cond6', 6)"); ok( r == LIBMSI_RESULT_SUCCESS, "cannot add entry to IESTable table:%d\n", r ); r = run_query(hdb, 0, "INSERT INTO `IESTable` " "( `Action`, `Condition`, `Sequence`) " "VALUES ( 'build', 'cond3', 3)"); ok(r == LIBMSI_RESULT_SUCCESS, "cannot add entry to IESTable table:%d\n", r ); r = run_query(hdb, 0, "INSERT INTO `CATable` " "( `Action`, `Type` ) " "VALUES ( 'build', 32)"); ok(r == LIBMSI_RESULT_SUCCESS, "cannot add entry to CATable table:%d\n", r ); r = run_query(hdb, 0, "INSERT INTO `CATable` " "( `Action`, `Type` ) " "VALUES ( 'depends', 64)"); ok(r == LIBMSI_RESULT_SUCCESS, "cannot add entry to CATable table:%d\n", r ); r = run_query(hdb, 0, "INSERT INTO `CATable` " "( `Action`, `Type` ) " "VALUES ( 'clean', 63)"); ok(r == LIBMSI_RESULT_SUCCESS, "cannot add entry to CATable table:%d\n", r ); r = run_query(hdb, 0, "INSERT INTO `CATable` " "( `Action`, `Type` ) " "VALUES ( 'build2', 34)"); ok(r == LIBMSI_RESULT_SUCCESS, "cannot add entry to CATable table:%d\n", r ); query = NULL; sql = "Select IESTable.Condition from CATable, IESTable where " "CATable.Action = IESTable.Action and CATable.Type = 32"; query = libmsi_query_new(hdb, sql, NULL); ok(query, "failed to open query: %d\n"); r = libmsi_query_execute(query, 0, NULL); ok( r, "failed to execute query: %d\n", r ); rec = libmsi_query_fetch(query, NULL); ok(rec, "failed to fetch query\n" ); ok( check_record( rec, 1, "cond2"), "wrong condition\n"); g_object_unref( rec ); rec = libmsi_query_fetch(query, NULL); ok(rec, "failed to fetch query\n"); ok( check_record( rec, 1, "cond3"), "wrong condition\n"); g_object_unref( rec ); libmsi_query_close(query, NULL); g_object_unref(query); g_object_unref( hdb ); unlink(msifile); } static void test_where(void) { LibmsiDatabase *hdb = 0; LibmsiRecord *rec; LibmsiQuery *query; const char *sql; unsigned r; unsigned size; char buf[256]; unsigned count; hdb = create_db(); ok( hdb, "failed to create db\n"); r = run_query( hdb, 0, "CREATE TABLE `Media` (" "`DiskId` SHORT NOT NULL, " "`LastSequence` LONG, " "`DiskPrompt` CHAR(64) LOCALIZABLE, " "`Cabinet` CHAR(255), " "`VolumeLabel` CHAR(32), " "`Source` CHAR(72) " "PRIMARY KEY `DiskId`)" ); ok( r == LIBMSI_RESULT_SUCCESS, "cannot create Media table: %d\n", r ); r = run_query( hdb, 0, "INSERT INTO `Media` " "( `DiskId`, `LastSequence`, `DiskPrompt`, `Cabinet`, `VolumeLabel`, `Source` ) " "VALUES ( 1, 0, '', 'zero.cab', '', '' )" ); ok( r == LIBMSI_RESULT_SUCCESS, "cannot add file to the Media table: %d\n", r ); r = run_query( hdb, 0, "INSERT INTO `Media` " "( `DiskId`, `LastSequence`, `DiskPrompt`, `Cabinet`, `VolumeLabel`, `Source` ) " "VALUES ( 2, 1, '', 'one.cab', '', '' )" ); ok( r == LIBMSI_RESULT_SUCCESS, "cannot add file to the Media table: %d\n", r ); r = run_query( hdb, 0, "INSERT INTO `Media` " "( `DiskId`, `LastSequence`, `DiskPrompt`, `Cabinet`, `VolumeLabel`, `Source` ) " "VALUES ( 3, 2, '', 'two.cab', '', '' )" ); ok( r == LIBMSI_RESULT_SUCCESS, "cannot add file to the Media table: %d\n", r ); sql = "SELECT * FROM `Media`"; r = do_query(hdb, sql, &rec); ok(r == LIBMSI_RESULT_SUCCESS, "libmsi_query_fetch failed: %d\n", r); ok( check_record( rec, 4, "zero.cab"), "wrong cabinet\n"); g_object_unref( rec ); rec = NULL; sql = "SELECT * FROM `Media` WHERE `LastSequence` >= 1"; r = do_query(hdb, sql, &rec); ok(r == LIBMSI_RESULT_SUCCESS, "libmsi_query_fetch failed: %d\n", r); ok( check_record( rec, 4, "one.cab"), "wrong cabinet\n"); r = libmsi_record_get_int(rec, 1); ok( 2 == r, "field wrong\n"); r = libmsi_record_get_int(rec, 2); ok( 1 == r, "field wrong\n"); g_object_unref( rec ); rec = NULL; sql = "SELECT `DiskId` FROM `Media` WHERE `LastSequence` >= 1 AND DiskId >= 0"; query = libmsi_query_new(hdb, sql, NULL); ok(query, "failed to open query: %d\n", r ); r = libmsi_query_execute(query, 0, NULL); ok( r, "failed to execute query: %d\n", r ); rec = libmsi_query_fetch(query, NULL); ok(rec, "failed to fetch query\n"); count = libmsi_record_get_field_count( rec ); ok( count == 1, "Expected 1 record fields, got %d\n", count ); check_record_string(rec, 1, "2"); g_object_unref(rec); rec = NULL; rec = libmsi_query_fetch(query, NULL); ok(rec, "failed to fetch query: %d\n", r ); check_record_string(rec, 1, "3"); g_object_unref(rec); rec = NULL; query_check_no_more(query); libmsi_query_close(query, NULL); g_object_unref(query); sql = "SELECT * FROM `Media` WHERE `DiskPrompt` IS NULL"; r = do_query(hdb, sql, &rec); ok( r == LIBMSI_RESULT_SUCCESS, "query failed: %d\n", r ); g_object_unref( rec ); rec = 0; sql = "SELECT * FROM `Media` WHERE `DiskPrompt` < 'Cabinet'"; r = do_query(hdb, sql, &rec); ok( r == LIBMSI_RESULT_BAD_QUERY_SYNTAX, "query failed: %d\n", r ); ok(rec == NULL, "Must be null"); sql = "SELECT * FROM `Media` WHERE `DiskPrompt` > 'Cabinet'"; r = do_query(hdb, sql, &rec); ok( r == LIBMSI_RESULT_BAD_QUERY_SYNTAX, "query failed: %d\n", r ); ok(rec == NULL, "Must be null"); sql = "SELECT * FROM `Media` WHERE `DiskPrompt` <> 'Cabinet'"; r = do_query(hdb, sql, &rec); ok( r == LIBMSI_RESULT_SUCCESS, "query failed: %d\n", r ); g_object_unref( rec ); rec = NULL; sql = "SELECT * FROM `Media` WHERE `DiskPrompt` = 'Cabinet'"; r = do_query(hdb, sql, &rec); ok( r == LIBMSI_RESULT_SUCCESS, "query failed: %d\n", r ); ok(rec == NULL, "Must be null"); rec = libmsi_record_new(1); libmsi_record_set_string(rec, 1, ""); sql = "SELECT * FROM `Media` WHERE `DiskPrompt` = ?"; query = libmsi_query_new(hdb, sql, NULL); ok(query, "Expected LIBMSI_RESULT_SUCCESS, got %d\n", r); r = libmsi_query_execute(query, rec, NULL); ok(r, "Expected LIBMSI_RESULT_SUCCESS, got %d\n", r); g_object_unref(rec); rec = libmsi_query_fetch(query, NULL); ok(rec, "Expected result\n"); g_object_unref(rec); libmsi_query_close(query, NULL); g_object_unref(query); g_object_unref( hdb ); unlink(msifile); } static const char test_data[] = "FirstPrimaryColumn\tSecondPrimaryColumn\tShortInt\tShortIntNullable\tLongInt\tLongIntNullable\tString\tLocalizableString\tLocalizableStringNullable\n" "s255\ti2\ti2\tI2\ti4\tI4\tS255\tS0\ts0\n" "TestTable\tFirstPrimaryColumn\n" "stringage\t5\t2\t\t2147483640\t-2147483640\tanother string\tlocalizable\tduh\n"; static const char two_primary[] = "PrimaryOne\tPrimaryTwo\n" "s255\ts255\n" "TwoPrimary\tPrimaryOne\tPrimaryTwo\n" "papaya\tleaf\n" "papaya\tflower\n"; static const char endlines1[] = "A\tB\tC\tD\tE\tF\r\n" "s72\ts72\ts72\ts72\ts72\ts72\n" "Table\tA\r\n" "a\tb\tc\td\te\tf\n" "g\th\ti\t\rj\tk\tl\r\n"; static const char endlines2[] = "A\tB\tC\tD\tE\tF\r" "s72\ts72\ts72\ts72\ts72\ts72\n" "Table2\tA\r\n" "a\tb\tc\td\te\tf\n" "g\th\ti\tj\tk\tl\r\n"; static const char suminfo[] = "PropertyId\tValue\n" "i2\tl255\n" "_SummaryInformation\tPropertyId\n" "1\t1252\n" "2\tInstaller Database\n" "3\tInstaller description\n" "4\tWineHQ\n" "5\tInstaller\n" "6\tInstaller comments\n" "7\tIntel;1033,2057\n" "9\t{12345678-1234-1234-1234-123456789012}\n" "12\t2009/04/12 15:46:11\n" "13\t2009/04/12 15:46:11\n" "14\t200\n" "15\t2\n" "18\tVim\n" "19\t2\n"; static void write_file(const char *filename, const char *data, int data_size) { int file; file = open(filename, O_CREAT | O_WRONLY | O_BINARY, 0644); if (file == -1) return; write(file, data, data_size); close(file); } static unsigned add_table_to_db(LibmsiDatabase *hdb, const char *table_data) { GError *err = NULL; unsigned r = LIBMSI_RESULT_SUCCESS; write_file("temp_file", table_data, (strlen(table_data) - 1) * sizeof(char)); if (!libmsi_database_import(hdb, "temp_file", &err)) r = err->code; g_clear_error(&err); unlink("temp_file"); return r; } static void test_suminfo_import(void) { GError *error = NULL; GArray *props; LibmsiDatabase *hdb; LibmsiSummaryInfo *hsi; LibmsiQuery *query = 0; const char *sql; unsigned r, type; const char *str_value; int int_value; guint64 ft_value; hdb = libmsi_database_new(msifile, LIBMSI_DB_FLAGS_CREATE, NULL, NULL); ok(hdb, "Expected LIBMSI_RESULT_SUCCESS, got %u\n", r); r = add_table_to_db(hdb, suminfo); ok(r == LIBMSI_RESULT_SUCCESS, "Expected LIBMSI_RESULT_SUCCESS, got %u\n", r); /* _SummaryInformation is not imported as a regular table... */ query = NULL; sql = "SELECT * FROM `_SummaryInformation`"; query = libmsi_query_new(hdb, sql, &error); g_assert_error(error, LIBMSI_RESULT_ERROR, LIBMSI_RESULT_BAD_QUERY_SYNTAX); ok(query == NULL, "Must be null"); g_clear_error(&error); /* ...its data is added to the special summary information stream */ hsi = libmsi_summary_info_new(hdb, 0, NULL); ok(hsi, "libmsi_summary_info_new() failed"); props = libmsi_summary_info_get_properties (hsi); ok(props->len == 14, "Expected 14, got %u\n", props->len); g_array_unref (props); int_value = libmsi_summary_info_get_int (hsi, LIBMSI_PROPERTY_CODEPAGE, &error); ok(!error, "Expected success\n"); ok(int_value == 1252, "Expected 1252, got %d\n", int_value); str_value = libmsi_summary_info_get_string (hsi, LIBMSI_PROPERTY_TITLE, &error); ok(!error, "Expected no error\n"); g_assert_cmpstr (str_value, ==, "Installer Database"); str_value = libmsi_summary_info_get_string (hsi, LIBMSI_PROPERTY_SUBJECT, &error); ok(!error, "Expected no error\n"); g_assert_cmpstr (str_value, ==, "Installer description"); str_value = libmsi_summary_info_get_string (hsi, LIBMSI_PROPERTY_AUTHOR, &error); ok(!error, "Expected no error\n"); g_assert_cmpstr (str_value, ==, "WineHQ"); str_value = libmsi_summary_info_get_string (hsi, LIBMSI_PROPERTY_KEYWORDS, &error); ok(!error, "Expected no error\n"); g_assert_cmpstr (str_value, ==, "Installer"); str_value = libmsi_summary_info_get_string (hsi, LIBMSI_PROPERTY_COMMENTS, &error); ok(!error, "Expected no error\n"); g_assert_cmpstr (str_value, ==, "Installer comments"); str_value = libmsi_summary_info_get_string (hsi, LIBMSI_PROPERTY_TEMPLATE, &error); ok(!error, "Expected no error\n"); g_assert_cmpstr (str_value, ==, "Intel;1033,2057"); str_value = libmsi_summary_info_get_string (hsi, LIBMSI_PROPERTY_UUID, &error); ok(!error, "Expected no error\n"); g_assert_cmpstr (str_value, ==, "{12345678-1234-1234-1234-123456789012}"); ft_value = libmsi_summary_info_get_filetime (hsi, LIBMSI_PROPERTY_CREATED_TM, &error); ok(!error, "Expected no error\n"); ft_value = libmsi_summary_info_get_filetime (hsi, LIBMSI_PROPERTY_LASTSAVED_TM, &error); ok(!error, "Expected no error\n"); int_value = libmsi_summary_info_get_int (hsi, LIBMSI_PROPERTY_VERSION, &error); ok(!error, "Expected success\n"); ok(int_value == 200, "Expected 200, got %d\n", int_value); int_value = libmsi_summary_info_get_int (hsi, LIBMSI_PROPERTY_SOURCE, &error); ok(!error, "Expected success\n"); ok(int_value == 2, "Expected 2, got %d\n", int_value); int_value = libmsi_summary_info_get_int (hsi, LIBMSI_PROPERTY_SECURITY, &error); ok(!error, "Expected success\n"); ok(int_value == 2, "Expected 2, got %d\n", int_value); str_value = libmsi_summary_info_get_string (hsi, LIBMSI_PROPERTY_APPNAME, &error); ok(!error, "Expected no error\n"); g_assert_cmpstr (str_value, ==, "Vim"); g_object_unref(hsi); g_object_unref(hdb); unlink(msifile); } static void test_msiimport(void) { LibmsiDatabase *hdb; LibmsiQuery *query; LibmsiRecord *rec; const char *sql; unsigned r, count; signed int i; hdb = libmsi_database_new(msifile, LIBMSI_DB_FLAGS_CREATE, NULL, NULL); ok(hdb, "Expected LIBMSI_RESULT_SUCCESS, got %d\n", r); r = add_table_to_db(hdb, test_data); ok(r == LIBMSI_RESULT_SUCCESS, "Expected LIBMSI_RESULT_SUCCESS, got %d\n", r); r = add_table_to_db(hdb, two_primary); ok(r == LIBMSI_RESULT_SUCCESS, "Expected LIBMSI_RESULT_SUCCESS, got %d\n", r); r = add_table_to_db(hdb, endlines1); ok(r == LIBMSI_RESULT_SUCCESS, "Expected LIBMSI_RESULT_SUCCESS, got %d\n", r); r = add_table_to_db(hdb, endlines2); ok(r == LIBMSI_RESULT_FUNCTION_FAILED, "Expected LIBMSI_RESULT_FUNCTION_FAILED, got %d\n", r); query = NULL; sql = "SELECT * FROM `TestTable`"; query = libmsi_query_new(hdb, sql, NULL); ok(query, "Expected LIBMSI_RESULT_SUCCESS\n"); rec = NULL; rec = libmsi_query_get_column_info(query, LIBMSI_COL_INFO_NAMES, NULL); ok(rec, "Expected result\n"); count = libmsi_record_get_field_count(rec); ok(count == 9, "Expected 9, got %d\n", count); ok(check_record(rec, 1, "FirstPrimaryColumn"), "Expected FirstPrimaryColumn\n"); ok(check_record(rec, 2, "SecondPrimaryColumn"), "Expected SecondPrimaryColumn\n"); ok(check_record(rec, 3, "ShortInt"), "Expected ShortInt\n"); ok(check_record(rec, 4, "ShortIntNullable"), "Expected ShortIntNullalble\n"); ok(check_record(rec, 5, "LongInt"), "Expected LongInt\n"); ok(check_record(rec, 6, "LongIntNullable"), "Expected LongIntNullalble\n"); ok(check_record(rec, 7, "String"), "Expected String\n"); ok(check_record(rec, 8, "LocalizableString"), "Expected LocalizableString\n"); ok(check_record(rec, 9, "LocalizableStringNullable"), "Expected LocalizableStringNullable\n"); g_object_unref(rec); rec = NULL; rec = libmsi_query_get_column_info(query, LIBMSI_COL_INFO_TYPES, NULL); ok(rec, "Expected result\n"); count = libmsi_record_get_field_count(rec); ok(count == 9, "Expected 9, got %d\n", count); ok(check_record(rec, 1, "s255"), "Expected s255\n"); ok(check_record(rec, 2, "i2"), "Expected i2\n"); ok(check_record(rec, 3, "i2"), "Expected i2\n"); ok(check_record(rec, 4, "I2"), "Expected I2\n"); ok(check_record(rec, 5, "i4"), "Expected i4\n"); ok(check_record(rec, 6, "I4"), "Expected I4\n"); ok(check_record(rec, 7, "S255"), "Expected S255\n"); ok(check_record(rec, 8, "S0"), "Expected S0\n"); ok(check_record(rec, 9, "s0"), "Expected s0\n"); g_object_unref(rec); sql = "SELECT * FROM `TestTable`"; r = do_query(hdb, sql, &rec); ok(r == LIBMSI_RESULT_SUCCESS, "Expected LIBMSI_RESULT_SUCCESS, got %d\n", r); ok(check_record(rec, 1, "stringage"), "Expected 'stringage'\n"); ok(check_record(rec, 7, "another string"), "Expected 'another string'\n"); ok(check_record(rec, 8, "localizable"), "Expected 'localizable'\n"); ok(check_record(rec, 9, "duh"), "Expected 'duh'\n"); i = libmsi_record_get_int(rec, 2); ok(i == 5, "Expected 5, got %d\n", i); i = libmsi_record_get_int(rec, 3); ok(i == 2, "Expected 2, got %d\n", i); i = libmsi_record_get_int(rec, 4); ok(i == LIBMSI_NULL_INT, "Expected LIBMSI_NULL_INT, got %d\n", i); i = libmsi_record_get_int(rec, 5); ok(i == 2147483640, "Expected 2147483640, got %d\n", i); i = libmsi_record_get_int(rec, 6); ok(i == -2147483640, "Expected -2147483640, got %d\n", i); g_object_unref(rec); libmsi_query_close(query, NULL); g_object_unref(query); query = NULL; sql = "SELECT * FROM `TwoPrimary`"; query = libmsi_query_new(hdb, sql, NULL); ok(query, "Expected LIBMSI_RESULT_SUCCESS\n"); rec = libmsi_query_get_column_info(query, LIBMSI_COL_INFO_NAMES, NULL); ok(rec, "Expected result\n"); count = libmsi_record_get_field_count(rec); ok(count == 2, "Expected 2, got %d\n", count); ok(check_record(rec, 1, "PrimaryOne"), "Expected PrimaryOne\n"); ok(check_record(rec, 2, "PrimaryTwo"), "Expected PrimaryTwo\n"); g_object_unref(rec); rec = libmsi_query_get_column_info(query, LIBMSI_COL_INFO_TYPES, NULL); ok(rec, "Expected result\n"); count = libmsi_record_get_field_count(rec); ok(count == 2, "Expected 2, got %d\n", count); ok(check_record(rec, 1, "s255"), "Expected s255\n"); ok(check_record(rec, 2, "s255"), "Expected s255\n"); g_object_unref(rec); r = libmsi_query_execute(query, 0, NULL); ok(r, "Expected LIBMSI_RESULT_SUCCESS, got %d\n", r); rec = libmsi_query_fetch(query, NULL); ok(rec, "Expected result\n"); ok(check_record(rec, 1, "papaya"), "Expected 'papaya'\n"); ok(check_record(rec, 2, "leaf"), "Expected 'leaf'\n"); g_object_unref(rec); rec = libmsi_query_fetch(query, NULL); ok(rec, "Expected result\n"); ok(check_record(rec, 1, "papaya"), "Expected 'papaya'\n"); ok(check_record(rec, 2, "flower"), "Expected 'flower'\n"); g_object_unref(rec); query_check_no_more(query); r = libmsi_query_close(query, NULL); ok(r, "Expected LIBMSI_RESULT_SUCCESS, got %d\n", r); g_object_unref(query); query = NULL; sql = "SELECT * FROM `Table`"; query = libmsi_query_new(hdb, sql, NULL); ok(query, "Expected LIBMSI_RESULT_SUCCESS\n"); rec = libmsi_query_get_column_info(query, LIBMSI_COL_INFO_NAMES, NULL); ok(rec, "Expected result\n"); count = libmsi_record_get_field_count(rec); ok(count == 6, "Expected 6, got %d\n", count); ok(check_record(rec, 1, "A"), "Expected A\n"); ok(check_record(rec, 2, "B"), "Expected B\n"); ok(check_record(rec, 3, "C"), "Expected C\n"); ok(check_record(rec, 4, "D"), "Expected D\n"); ok(check_record(rec, 5, "E"), "Expected E\n"); ok(check_record(rec, 6, "F"), "Expected F\n"); g_object_unref(rec); rec = libmsi_query_get_column_info(query, LIBMSI_COL_INFO_TYPES, NULL); ok(rec, "Expected result\n"); count = libmsi_record_get_field_count(rec); ok(count == 6, "Expected 6, got %d\n", count); ok(check_record(rec, 1, "s72"), "Expected s72\n"); ok(check_record(rec, 2, "s72"), "Expected s72\n"); ok(check_record(rec, 3, "s72"), "Expected s72\n"); ok(check_record(rec, 4, "s72"), "Expected s72\n"); ok(check_record(rec, 5, "s72"), "Expected s72\n"); ok(check_record(rec, 6, "s72"), "Expected s72\n"); g_object_unref(rec); libmsi_query_close(query, NULL); g_object_unref(query); query = NULL; sql = "SELECT * FROM `Table`"; query = libmsi_query_new(hdb, sql, NULL); ok(query, "Expected LIBMSI_RESULT_SUCCESS\n"); r = libmsi_query_execute(query, 0, NULL); ok(r, "Expected LIBMSI_RESULT_SUCCESS, got %d\n", r); rec = libmsi_query_fetch(query, NULL); ok(rec, "Expected result\n"); ok(check_record(rec, 1, "a"), "Expected 'a'\n"); ok(check_record(rec, 2, "b"), "Expected 'b'\n"); ok(check_record(rec, 3, "c"), "Expected 'c'\n"); ok(check_record(rec, 4, "d"), "Expected 'd'\n"); ok(check_record(rec, 5, "e"), "Expected 'e'\n"); ok(check_record(rec, 6, "f"), "Expected 'f'\n"); g_object_unref(rec); rec = libmsi_query_fetch(query, NULL); ok(rec, "Expected result\n"); ok(check_record(rec, 1, "g"), "Expected 'g'\n"); ok(check_record(rec, 2, "h"), "Expected 'h'\n"); ok(check_record(rec, 3, "i"), "Expected 'i'\n"); ok(check_record(rec, 4, "j"), "Expected 'j'\n"); ok(check_record(rec, 5, "k"), "Expected 'k'\n"); ok(check_record(rec, 6, "l"), "Expected 'l'\n"); g_object_unref(rec); query_check_no_more(query); libmsi_query_close(query, NULL); g_object_unref(query); g_object_unref(hdb); unlink(msifile); } static const char bin_import_dat[] = "Name\tData\r\n" "s72\tV0\r\n" "Binary\tName\r\n" "filename1\tfilename1.ibd\r\n"; static void test_binary_import(void) { GInputStream *in; LibmsiDatabase *hdb = 0; LibmsiRecord *rec; char file[256]; char buf[256]; unsigned size; const char *sql; unsigned r; /* create files to import */ write_file("bin_import.idt", bin_import_dat, (sizeof(bin_import_dat) - 1) * sizeof(char)); mkdir("Binary", 0755); create_file_data("Binary/filename1.ibd", "just some words", 15); /* import files into database */ hdb = libmsi_database_new(msifile, LIBMSI_DB_FLAGS_CREATE, NULL, NULL); ok(hdb , "Failed to open database\n"); r = libmsi_database_import(hdb, "bin_import.idt", NULL); ok(r , "Failed to import Binary table\n"); /* read file from the Binary table */ sql = "SELECT * FROM `Binary`"; r = do_query(hdb, sql, &rec); ok(r == LIBMSI_RESULT_SUCCESS, "SELECT query failed: %d\n", r); check_record_string(rec, 1, "filename1"); size = sizeof(buf); memset(buf, 0, size); in = libmsi_record_get_stream(rec, 2); ok(in, "Failed to get stream\n"); size = g_input_stream_read(in, buf, sizeof(buf), NULL, NULL); ok(g_str_equal(buf, "just some words"), "Expected 'just some words', got %s\n", buf); g_object_unref(in); g_object_unref(rec); g_object_unref(hdb); unlink("bin_import/filename1.ibd"); rmdir("bin_import"); unlink("bin_import.idt"); } static void test_markers(void) { LibmsiDatabase *hdb; LibmsiRecord *rec; const char *sql; unsigned r; hdb = create_db(); ok( hdb, "failed to create db\n"); rec = libmsi_record_new(3); libmsi_record_set_string(rec, 1, "Table"); libmsi_record_set_string(rec, 2, "Apples"); libmsi_record_set_string(rec, 3, "Oranges"); /* try a legit create */ sql = "CREATE TABLE `Table` ( `One` SHORT NOT NULL, `Two` CHAR(255) PRIMARY KEY `One`)"; r = run_query(hdb, 0, sql); ok(r == LIBMSI_RESULT_SUCCESS, "Expected LIBMSI_RESULT_SUCCESS, got %d\n", r); g_object_unref(rec); /* try table name as marker */ rec = libmsi_record_new(1); libmsi_record_set_string(rec, 1, "Fable"); sql = "CREATE TABLE `?` ( `One` SHORT NOT NULL, `Two` CHAR(255) PRIMARY KEY `One`)"; r = run_query(hdb, rec, sql); ok(r == LIBMSI_RESULT_SUCCESS, "Expected LIBMSI_RESULT_SUCCESS, got %d\n", r); /* verify that we just created a table called '?', not 'Fable' */ r = try_query(hdb, "SELECT * from `Fable`"); ok(r == LIBMSI_RESULT_BAD_QUERY_SYNTAX, "Expected LIBMSI_RESULT_BAD_QUERY_SYNTAX, got %d\n", r); r = try_query(hdb, "SELECT * from `?`"); ok(r == LIBMSI_RESULT_SUCCESS, "Expected LIBMSI_RESULT_SUCCESS, got %d\n", r); /* try table name as marker without backticks */ libmsi_record_set_string(rec, 1, "Mable"); sql = "CREATE TABLE ? ( `One` SHORT NOT NULL, `Two` CHAR(255) PRIMARY KEY `One`)"; r = run_query(hdb, rec, sql); ok(r == LIBMSI_RESULT_BAD_QUERY_SYNTAX, "Expected LIBMSI_RESULT_BAD_QUERY_SYNTAX, got %d\n", r); /* try one column name as marker */ libmsi_record_set_string(rec, 1, "One"); sql = "CREATE TABLE `Mable` ( `?` SHORT NOT NULL, `Two` CHAR(255) PRIMARY KEY `One`)"; r = run_query(hdb, rec, sql); ok(r == LIBMSI_RESULT_BAD_QUERY_SYNTAX, "Expected LIBMSI_RESULT_BAD_QUERY_SYNTAX, got %d\n", r); g_object_unref(rec); /* try column names as markers */ rec = libmsi_record_new(2); libmsi_record_set_string(rec, 1, "One"); libmsi_record_set_string(rec, 2, "Two"); sql = "CREATE TABLE `Mable` ( `?` SHORT NOT NULL, `?` CHAR(255) PRIMARY KEY `One`)"; r = run_query(hdb, rec, sql); ok(r == LIBMSI_RESULT_BAD_QUERY_SYNTAX, "Expected LIBMSI_RESULT_BAD_QUERY_SYNTAX, got %d\n", r); g_object_unref(rec); /* try names with backticks */ rec = libmsi_record_new(3); libmsi_record_set_string(rec, 1, "One"); libmsi_record_set_string(rec, 2, "Two"); libmsi_record_set_string(rec, 3, "One"); sql = "CREATE TABLE `Mable` ( `?` SHORT NOT NULL, `?` CHAR(255) PRIMARY KEY `?`)"; r = run_query(hdb, rec, sql); ok(r == LIBMSI_RESULT_BAD_QUERY_SYNTAX, "Expected LIBMSI_RESULT_BAD_QUERY_SYNTAX, got %d\n", r); /* try names with backticks, minus definitions */ sql = "CREATE TABLE `Mable` ( `?`, `?` PRIMARY KEY `?`)"; r = run_query(hdb, rec, sql); ok(r == LIBMSI_RESULT_BAD_QUERY_SYNTAX, "Expected LIBMSI_RESULT_BAD_QUERY_SYNTAX, got %d\n", r); /* try names without backticks */ sql = "CREATE TABLE `Mable` ( ? SHORT NOT NULL, ? CHAR(255) PRIMARY KEY ?)"; r = run_query(hdb, rec, sql); ok(r == LIBMSI_RESULT_BAD_QUERY_SYNTAX, "Expected LIBMSI_RESULT_BAD_QUERY_SYNTAX, got %d\n", r); g_object_unref(rec); /* try one long marker */ rec = libmsi_record_new(1); libmsi_record_set_string(rec, 1, "`One` SHORT NOT NULL, `Two` CHAR(255) PRIMARY KEY `One`"); sql = "CREATE TABLE `Mable` ( ? )"; r = run_query(hdb, rec, sql); ok(r == LIBMSI_RESULT_BAD_QUERY_SYNTAX, "Expected LIBMSI_RESULT_BAD_QUERY_SYNTAX, got %d\n", r); g_object_unref(rec); /* try all names as markers */ rec = libmsi_record_new(4); libmsi_record_set_string(rec, 1, "Mable"); libmsi_record_set_string(rec, 2, "One"); libmsi_record_set_string(rec, 3, "Two"); libmsi_record_set_string(rec, 4, "One"); sql = "CREATE TABLE `?` ( `?` SHORT NOT NULL, `?` CHAR(255) PRIMARY KEY `?`)"; r = run_query(hdb, rec, sql); ok(r == LIBMSI_RESULT_BAD_QUERY_SYNTAX, "Expected LIBMSI_RESULT_BAD_QUERY_SYNTAX, got %d\n", r); g_object_unref(rec); /* try a legit insert */ sql = "INSERT INTO `Table` ( `One`, `Two` ) VALUES ( 5, 'hello' )"; r = run_query(hdb, 0, sql); ok(r == LIBMSI_RESULT_SUCCESS, "Expected LIBMSI_RESULT_SUCCESS, got %d\n", r); r = try_query(hdb, "SELECT * from `Table`"); ok(r == LIBMSI_RESULT_SUCCESS, "Expected LIBMSI_RESULT_SUCCESS, got %d\n", r); /* try values as markers */ rec = libmsi_record_new(2); libmsi_record_set_int(rec, 1, 4); libmsi_record_set_string(rec, 2, "hi"); sql = "INSERT INTO `Table` ( `One`, `Two` ) VALUES ( ?, '?' )"; r = run_query(hdb, rec, sql); ok(r == LIBMSI_RESULT_SUCCESS, "Expected LIBMSI_RESULT_SUCCESS, got %d\n", r); g_object_unref(rec); /* try column names and values as markers */ rec = libmsi_record_new(4); libmsi_record_set_string(rec, 1, "One"); libmsi_record_set_string(rec, 2, "Two"); libmsi_record_set_int(rec, 3, 5); libmsi_record_set_string(rec, 4, "hi"); sql = "INSERT INTO `Table` ( `?`, `?` ) VALUES ( ?, '?' )"; r = run_query(hdb, rec, sql); ok(r == LIBMSI_RESULT_BAD_QUERY_SYNTAX, "Expected LIBMSI_RESULT_BAD_QUERY_SYNTAX, got %d\n", r); g_object_unref(rec); /* try column names as markers */ rec = libmsi_record_new(2); libmsi_record_set_string(rec, 1, "One"); libmsi_record_set_string(rec, 2, "Two"); sql = "INSERT INTO `Table` ( `?`, `?` ) VALUES ( 3, 'yellow' )"; r = run_query(hdb, rec, sql); ok(r == LIBMSI_RESULT_BAD_QUERY_SYNTAX, "Expected LIBMSI_RESULT_BAD_QUERY_SYNTAX, got %d\n", r); g_object_unref(rec); /* try table name as a marker */ rec = libmsi_record_new(1); libmsi_record_set_string(rec, 1, "Table"); sql = "INSERT INTO `?` ( `One`, `Two` ) VALUES ( 2, 'green' )"; r = run_query(hdb, rec, sql); ok(r == LIBMSI_RESULT_SUCCESS, "Expected LIBMSI_RESULT_SUCCESS, got %d\n", r); g_object_unref(rec); /* try table name and values as markers */ rec = libmsi_record_new(3); libmsi_record_set_string(rec, 1, "Table"); libmsi_record_set_int(rec, 2, 10); libmsi_record_set_string(rec, 3, "haha"); sql = "INSERT INTO `?` ( `One`, `Two` ) VALUES ( ?, '?' )"; r = run_query(hdb, rec, sql); ok(r == LIBMSI_RESULT_FUNCTION_FAILED, "Expected LIBMSI_RESULT_FUNCTION_FAILED, got %d\n", r); g_object_unref(rec); /* try all markers */ rec = libmsi_record_new(5); libmsi_record_set_string(rec, 1, "Table"); libmsi_record_set_string(rec, 1, "One"); libmsi_record_set_string(rec, 1, "Two"); libmsi_record_set_int(rec, 2, 10); libmsi_record_set_string(rec, 3, "haha"); sql = "INSERT INTO `?` ( `?`, `?` ) VALUES ( ?, '?' )"; r = run_query(hdb, rec, sql); ok(r == LIBMSI_RESULT_BAD_QUERY_SYNTAX, "Expected LIBMSI_RESULT_BAD_QUERY_SYNTAX, got %d\n", r); g_object_unref(rec); /* insert an integer as a string */ rec = libmsi_record_new(2); libmsi_record_set_string(rec, 1, "11"); libmsi_record_set_string(rec, 2, "hi"); sql = "INSERT INTO `Table` ( `One`, `Two` ) VALUES ( ?, '?' )"; r = run_query(hdb, rec, sql); ok(r == LIBMSI_RESULT_SUCCESS, "Expected LIBMSI_RESULT_SUCCESS, got %d\n", r); g_object_unref(rec); /* leave off the '' for the string */ rec = libmsi_record_new(2); libmsi_record_set_int(rec, 1, 12); libmsi_record_set_string(rec, 2, "hi"); sql = "INSERT INTO `Table` ( `One`, `Two` ) VALUES ( ?, ? )"; r = run_query(hdb, rec, sql); ok(r == LIBMSI_RESULT_SUCCESS, "Expected LIBMSI_RESULT_SUCCESS, got %d\n", r); g_object_unref(rec); g_object_unref(hdb); unlink(msifile); } #define MY_NQUERIES 4000 /* Largest installer I've seen uses < 2k */ static void test_handle_limit(void) { int i; LibmsiDatabase *hdb; LibmsiQuery *hqueries[MY_NQUERIES]; unsigned r; /* create an empty db */ hdb = create_db(); ok( hdb, "failed to create db\n"); memset(hqueries, 0, sizeof(hqueries)); for (i=0; i | D | C | A | E | B | * --------------------- --------------------- * * set primary key `E` * --------------------- --------------------- * | D | C | A | E | B | -> | D | E | A | C | B | * --------------------- --------------------- */ sql = "CREATE TABLE `T` ( `B` SHORT NOT NULL, `C` SHORT NOT NULL, " "`A` CHAR(255), `E` INT, `D` CHAR(255) NOT NULL " "PRIMARY KEY `D`, `E`)"; r = run_query(hdb, 0, sql); ok(r == LIBMSI_RESULT_SUCCESS, "Expected LIBMSI_RESULT_SUCCESS, got %d\n", r); query = NULL; sql = "SELECT * FROM `T`"; query = libmsi_query_new(hdb, sql, NULL); ok(query, "Expected LIBMSI_RESULT_SUCCESS\n"); rec = libmsi_query_get_column_info(query, LIBMSI_COL_INFO_TYPES, NULL); ok(rec, "Expected result\n"); sz = sizeof(buf); strcpy(buf, "kiwi"); check_record_string(rec, 1, "s255"); check_record_string(rec, 2, "I2"); check_record_string(rec, 3, "S255"); check_record_string(rec, 4, "i2"); check_record_string(rec, 5, "i2"); g_object_unref(rec); rec = libmsi_query_get_column_info(query, LIBMSI_COL_INFO_NAMES, NULL); ok(rec, "Expected result\n"); check_record_string(rec, 1, "D"); check_record_string(rec, 2, "E"); check_record_string(rec, 3, "A"); check_record_string(rec, 4, "C"); check_record_string(rec, 5, "B"); g_object_unref(rec); libmsi_query_close(query, NULL); g_object_unref(query); sql = "INSERT INTO `T` ( `B`, `C`, `A`, `E`, `D` ) " "VALUES ( 1, 2, 'a', 3, 'bc' )"; r = run_query(hdb, 0, sql); ok(r == LIBMSI_RESULT_SUCCESS, "Expected LIBMSI_RESULT_SUCCESS, got %d\n", r); sql = "SELECT * FROM `T`"; r = do_query(hdb, sql, &rec); ok(r == LIBMSI_RESULT_SUCCESS, "Expected LIBMSI_RESULT_SUCCESS, got %d\n", r); check_record_string(rec, 1, "bc"); r = libmsi_record_get_int(rec, 2); ok(r == 3, "Expected 3, got %d\n", r); check_record_string(rec, 3, "a"); r = libmsi_record_get_int(rec, 4); ok(r == 2, "Expected 2, got %d\n", r); r = libmsi_record_get_int(rec, 5); ok(r == 1, "Expected 1, got %d\n", r); g_object_unref(rec); query = NULL; sql = "SELECT * FROM `_Columns` WHERE `Table` = 'T'"; query = libmsi_query_new(hdb, sql, NULL); ok(query, "Expected LIBMSI_RESULT_SUCCESS\n"); r = libmsi_query_execute(query, 0, NULL); ok(r, "Expected LIBMSI_RESULT_SUCCESS, got %d\n", r); rec = libmsi_query_fetch(query, NULL); ok(rec, "Expected result\n"); check_record_string(rec, 1, "T"); r = libmsi_record_get_int(rec, 2); ok(r == 1, "Expected 1, got %d\n", r); check_record_string(rec, 3, "D"); g_object_unref(rec); rec = libmsi_query_fetch(query, NULL); ok(rec, "Expected result\n"); check_record_string(rec, 1, "T"); r = libmsi_record_get_int(rec, 2); ok(r == 2, "Expected 2, got %d\n", r); check_record_string(rec, 3, "E"); g_object_unref(rec); rec = libmsi_query_fetch(query, NULL); ok(rec, "Expected result\n"); check_record_string(rec, 1, "T"); r = libmsi_record_get_int(rec, 2); ok(r == 3, "Expected 3, got %d\n", r); check_record_string(rec, 3, "A"); g_object_unref(rec); rec = libmsi_query_fetch(query, NULL); ok(rec, "Expected result\n"); check_record_string(rec, 1, "T"); r = libmsi_record_get_int(rec, 2); ok(r == 4, "Expected 4, got %d\n", r); check_record_string(rec, 3, "C"); g_object_unref(rec); rec = libmsi_query_fetch(query, NULL); ok(rec, "Expected result\n"); check_record_string(rec, 1, "T"); r = libmsi_record_get_int(rec, 2); ok(r == 5, "Expected 5, got %d\n", r); check_record_string(rec, 3, "B"); g_object_unref(rec); query_check_no_more(query); libmsi_query_close(query, NULL); g_object_unref(query); sql = "CREATE TABLE `Z` ( `B` SHORT NOT NULL, `C` SHORT NOT NULL, " "`A` CHAR(255), `E` INT, `D` CHAR(255) NOT NULL " "PRIMARY KEY `C`, `A`, `D`)"; r = run_query(hdb, 0, sql); ok(r == LIBMSI_RESULT_SUCCESS, "Expected LIBMSI_RESULT_SUCCESS, got %d\n", r); query = NULL; sql = "SELECT * FROM `Z`"; query = libmsi_query_new(hdb, sql, NULL); ok(query, "Expected LIBMSI_RESULT_SUCCESS\n"); rec = libmsi_query_get_column_info(query, LIBMSI_COL_INFO_TYPES, NULL); ok(rec, "Expected result\n"); check_record_string(rec, 1, "i2"); check_record_string(rec, 2, "S255"); check_record_string(rec, 3, "s255"); check_record_string(rec, 4, "I2"); check_record_string(rec, 5, "i2"); g_object_unref(rec); rec = libmsi_query_get_column_info(query, LIBMSI_COL_INFO_NAMES, NULL); ok(rec, "Expected result\n"); check_record_string(rec, 1, "C"); check_record_string(rec, 2, "A"); check_record_string(rec, 3, "D"); check_record_string(rec, 4, "E"); check_record_string(rec, 5, "B"); g_object_unref(rec); libmsi_query_close(query, NULL); g_object_unref(query); sql = "INSERT INTO `Z` ( `B`, `C`, `A`, `E`, `D` ) " "VALUES ( 1, 2, 'a', 3, 'bc' )"; r = run_query(hdb, 0, sql); ok(r == LIBMSI_RESULT_SUCCESS, "Expected LIBMSI_RESULT_SUCCESS, got %d\n", r); sql = "SELECT * FROM `Z`"; r = do_query(hdb, sql, &rec); ok(r == LIBMSI_RESULT_SUCCESS, "Expected LIBMSI_RESULT_SUCCESS, got %d\n", r); r = libmsi_record_get_int(rec, 1); ok(r == 2, "Expected 2, got %d\n", r); check_record_string(rec, 2, "a"); check_record_string(rec, 3, "bc"); r = libmsi_record_get_int(rec, 4); ok(r == 3, "Expected 3, got %d\n", r); r = libmsi_record_get_int(rec, 5); ok(r == 1, "Expected 1, got %d\n", r); g_object_unref(rec); query = NULL; sql = "SELECT * FROM `_Columns` WHERE `Table` = 'T'"; query = libmsi_query_new(hdb, sql, NULL); ok(query, "Expected LIBMSI_RESULT_SUCCESS\n"); r = libmsi_query_execute(query, 0, NULL); ok(r, "Expected LIBMSI_RESULT_SUCCESS, got %d\n", r); rec = libmsi_query_fetch(query, NULL); ok(rec, "Expected result\n"); check_record_string(rec, 1, "T"); r = libmsi_record_get_int(rec, 2); ok(r == 1, "Expected 1, got %d\n", r); check_record_string(rec, 3, "D"); g_object_unref(rec); rec = libmsi_query_fetch(query, NULL); ok(rec, "Expected result\n"); check_record_string(rec, 1, "T"); r = libmsi_record_get_int(rec, 2); ok(r == 2, "Expected 2, got %d\n", r); check_record_string(rec, 3, "E"); g_object_unref(rec); rec = libmsi_query_fetch(query, NULL); ok(rec, "Expected result\n"); check_record_string(rec, 1, "T"); r = libmsi_record_get_int(rec, 2); ok(r == 3, "Expected 3, got %d\n", r); check_record_string(rec, 3, "A"); g_object_unref(rec); rec = libmsi_query_fetch(query, NULL); ok(rec, "Expected result\n"); check_record_string(rec, 1, "T"); r = libmsi_record_get_int(rec, 2); ok(r == 4, "Expected 4, got %d\n", r); check_record_string(rec, 3, "C"); g_object_unref(rec); rec = libmsi_query_fetch(query, NULL); ok(rec, "Expected result\n"); check_record_string(rec, 1, "T"); r = libmsi_record_get_int(rec, 2); ok(r == 5, "Expected 5, got %d\n", r); check_record_string(rec, 3, "B"); g_object_unref(rec); query_check_no_more(query); libmsi_query_close(query, NULL); g_object_unref(query); g_object_unref(hdb); unlink(msifile); } static void test_createtable(void) { LibmsiDatabase *hdb; LibmsiQuery *htab = 0; LibmsiRecord *hrec = 0; const char *sql; unsigned res; unsigned size; char buffer[0x20]; gchar *str; hdb = create_db(); ok(hdb, "failed to create db\n"); sql = "CREATE TABLE `blah` (`foo` CHAR(72) NOT NULL PRIMARY KEY `foo`)"; htab = libmsi_query_new( hdb, sql, NULL); ok(htab, "Expected LIBMSI_RESULT_SUCCESS\n"); if(res == LIBMSI_RESULT_SUCCESS ) { res = libmsi_query_execute( htab, hrec , NULL); ok(res, "Expected LIBMSI_RESULT_SUCCESS, got %d\n", res); hrec = libmsi_query_get_column_info( htab, LIBMSI_COL_INFO_NAMES, NULL ); todo_wine ok(hrec, "Expected result\n"); str = libmsi_record_get_string(hrec, 1); todo_wine ok(str, "Expected string\n"); g_free(str); g_object_unref( hrec ); res = libmsi_query_close(htab , NULL); ok(res, "Expected LIBMSI_RESULT_SUCCESS, got %d\n", res); g_object_unref( htab ); } sql = "CREATE TABLE `a` (`b` INT PRIMARY KEY `b`)"; htab = libmsi_query_new( hdb, sql, NULL); ok(htab, "Expected LIBMSI_RESULT_SUCCESS\n"); if(res == LIBMSI_RESULT_SUCCESS ) { res = libmsi_query_execute( htab, 0 , NULL); ok(res, "Expected LIBMSI_RESULT_SUCCESS, got %d\n", res); res = libmsi_query_close(htab, NULL); ok(res, "Expected LIBMSI_RESULT_SUCCESS, got %d\n", res); g_object_unref( htab ); sql = "SELECT * FROM `a`"; htab = libmsi_query_new( hdb, sql, NULL); ok(htab, "Expected LIBMSI_RESULT_SUCCESS\n"); hrec = libmsi_query_get_column_info( htab, LIBMSI_COL_INFO_NAMES, NULL ); ok(hrec, "Expected result\n"); check_record_string(hrec, 1, "b"); g_object_unref( hrec ); res = libmsi_query_close(htab , NULL); ok(res, "Expected LIBMSI_RESULT_SUCCESS, got %d\n", res); g_object_unref( htab ); res = libmsi_database_commit(hdb, NULL); ok(res, "Expected LIBMSI_RESULT_SUCCESS\n"); g_object_unref(hdb); hdb = libmsi_database_new(msifile, LIBMSI_DB_FLAGS_TRANSACT, NULL, NULL); ok(hdb, "Expected LIBMSI_RESULT_SUCCESS, got %d\n", res); sql = "SELECT * FROM `a`"; htab = libmsi_query_new(hdb, sql, NULL); ok(htab, "Expected LIBMSI_RESULT_SUCCESS\n"); hrec = libmsi_query_get_column_info( htab, LIBMSI_COL_INFO_NAMES, NULL ); ok(hrec, "Expected result\n"); check_record_string(hrec, 1, "b"); g_object_unref( hrec ); res = libmsi_query_close(htab , NULL); ok(res, "Expected LIBMSI_RESULT_SUCCESS, got %d\n", res); g_object_unref( htab ); } res = libmsi_database_commit(hdb, NULL); ok(res, "Expected LIBMSI_RESULT_SUCCESS\n"); g_object_unref(hdb); unlink(msifile); } static void test_embedded_nulls(void) { static const char control_table[] = "Dialog\tText\n" "s72\tL0\n" "Control\tDialog\n" "LicenseAgreementDlg\ttext\x11\x19text\0text"; unsigned r, sz; LibmsiDatabase *hdb; LibmsiRecord *hrec; char buffer[32]; gchar *str; hdb = libmsi_database_new( msifile, LIBMSI_DB_FLAGS_CREATE, NULL, NULL); ok(hdb, "failed to open database %u\n", r ); write_file( "temp_file", control_table, sizeof(control_table) ); r = libmsi_database_import( hdb, "temp_file", NULL); ok(r, "libmsi_database_import() failed\n"); unlink( "temp_file" ); r = do_query( hdb, "SELECT `Text` FROM `Control` WHERE `Dialog` = 'LicenseAgreementDlg'", &hrec ); ok( r == LIBMSI_RESULT_SUCCESS, "query failed %u\n", r ); str = libmsi_record_get_string( hrec, 1); ok(str, "failed to get string\n" ); ok( !memcmp( "text\r\ntext\ntext", str, sizeof("text\r\ntext\ntext") - 1 ), "wrong buffer contents \"%s\"\n", str ); g_free(str); g_object_unref( hrec ); g_object_unref( hdb ); unlink( msifile ); } static void test_select_column_names(void) { LibmsiDatabase *hdb = 0; LibmsiRecord *rec; LibmsiRecord *rec2; LibmsiQuery *query; char buffer[32]; unsigned r, size; unlink(msifile); hdb = libmsi_database_new( msifile, LIBMSI_DB_FLAGS_CREATE, NULL, NULL); ok(hdb, "failed to open database: %u\n", r ); r = try_query( hdb, "CREATE TABLE `t` (`a` CHAR NOT NULL, `b` CHAR PRIMARY KEY `a`)"); ok( r == LIBMSI_RESULT_SUCCESS , "query failed: %u\n", r ); r = try_query( hdb, "SELECT `t`.`b` FROM `t` WHERE `t`.`b` = `x`" ); ok( r == LIBMSI_RESULT_BAD_QUERY_SYNTAX, "query failed: %u\n", r ); r = try_query( hdb, "SELECT '', `t`.`b` FROM `t` WHERE `t`.`b` = 'x'" ); ok( r == LIBMSI_RESULT_SUCCESS , "query failed: %u\n", r ); r = try_query( hdb, "SELECT *, `t`.`b` FROM `t` WHERE `t`.`b` = 'x'" ); todo_wine ok( r == LIBMSI_RESULT_SUCCESS, "query failed: %u\n", r ); r = try_query( hdb, "SELECT 'b', `t`.`b` FROM `t` WHERE `t`.`b` = 'x'" ); ok( r == LIBMSI_RESULT_SUCCESS , "query failed: %u\n", r ); r = try_query( hdb, "SELECT `t`.`b`, '' FROM `t` WHERE `t`.`b` = 'x'" ); ok( r == LIBMSI_RESULT_SUCCESS , "query failed: %u\n", r ); r = try_query( hdb, "SELECT `t`.`b`, '' FROM `t` WHERE `t`.`b` = 'x' ORDER BY `b`" ); ok( r == LIBMSI_RESULT_SUCCESS , "query failed: %u\n", r ); r = try_query( hdb, "SELECT `t`.`b`, '' FROM `t` WHERE `t`.`b` = 'x' ORDER BY 'b'" ); ok( r == LIBMSI_RESULT_BAD_QUERY_SYNTAX, "query failed: %u\n", r ); r = try_query( hdb, "SELECT 't'.'b' FROM `t` WHERE `t`.`b` = 'x'" ); ok( r == LIBMSI_RESULT_BAD_QUERY_SYNTAX, "query failed: %u\n", r ); r = try_query( hdb, "SELECT 'b' FROM `t` WHERE `t`.`b` = 'x'" ); ok( r == LIBMSI_RESULT_SUCCESS , "query failed: %u\n", r ); r = try_query( hdb, "INSERT INTO `t` ( `a`, `b` ) VALUES( '1', '2' )" ); ok( r == LIBMSI_RESULT_SUCCESS , "query failed: %u\n", r ); r = try_query( hdb, "INSERT INTO `t` ( `a`, `b` ) VALUES( '3', '4' )" ); ok( r == LIBMSI_RESULT_SUCCESS , "query failed: %u\n", r ); query = libmsi_query_new( hdb, "SELECT '' FROM `t`", NULL); ok(query, "failed to open database query\n"); r = libmsi_query_execute( query, 0 , NULL); ok( r, "failed to execute query: %u\n", r ); rec = libmsi_query_fetch(query, NULL); ok(rec, "Expected result\n"); r = libmsi_record_get_field_count( rec ); ok( r == 1, "got %u\n", r ); rec2 = libmsi_query_get_column_info( query, LIBMSI_COL_INFO_NAMES, NULL ); ok(rec2, "unexpected result\n"); r = libmsi_record_get_field_count( rec2 ); ok( r == 1, "got %u\n", r ); check_record_string(rec2, 1, ""); g_object_unref( rec2 ); rec2 = libmsi_query_get_column_info( query, LIBMSI_COL_INFO_TYPES, NULL ); ok(rec2, "unexpected result\n"); r = libmsi_record_get_field_count( rec2 ); ok( r == 1, "got %u\n", r ); check_record_string( rec2, 1, "f0"); g_object_unref( rec2 ); check_record_string( rec, 1, ""); g_object_unref( rec ); rec = libmsi_query_fetch(query, NULL); ok(rec, "Expected result\n"); check_record_string(rec, 1, ""); g_object_unref( rec ); query_check_no_more(query); libmsi_query_close(query, NULL); g_object_unref( query ); query = libmsi_query_new( hdb, "SELECT `a`, '' FROM `t`", NULL); ok(query, "failed to open database query\n"); r = libmsi_query_execute( query, 0 , NULL); ok( r, "failed to execute query: %u\n", r ); rec = libmsi_query_fetch(query, NULL); ok(rec, "Expected result\n"); r = libmsi_record_get_field_count( rec ); ok( r == 2, "got %u\n", r ); check_record_string(rec, 1, "1"); g_object_unref( rec ); rec = libmsi_query_fetch(query, NULL); ok(rec, "Expected result\n"); check_record_string( rec, 2, ""); g_object_unref( rec ); query_check_no_more(query); libmsi_query_close(query, NULL); g_object_unref( query ); query = libmsi_query_new( hdb, "SELECT '', `a` FROM `t`", NULL); ok(query, "failed to open database query\n"); r = libmsi_query_execute( query, 0 , NULL); ok( r, "failed to execute query: %u\n", r ); rec = libmsi_query_fetch(query, NULL); ok(rec, "Expected result\n"); r = libmsi_record_get_field_count( rec ); ok( r == 2, "got %u\n", r ); check_record_string( rec, 1, ""); check_record_string( rec, 2, "1"); g_object_unref( rec ); rec = libmsi_query_fetch(query, NULL); ok(rec, "Expected result\n"); check_record_string( rec, 1, ""); check_record_string(rec, 2, "3"); g_object_unref( rec ); query_check_no_more(query); libmsi_query_close(query, NULL); g_object_unref( query ); query = libmsi_query_new( hdb, "SELECT `a`, '', `b` FROM `t`", NULL ); ok(query, "failed to open database query\n"); r = libmsi_query_execute( query, 0 , NULL); ok( r, "failed to execute query: %u\n", r ); rec = libmsi_query_fetch(query, NULL); ok(rec, "Expected result\n"); r = libmsi_record_get_field_count( rec ); ok( r == 3, "got %u\n", r ); check_record_string(rec, 1, "1"); check_record_string(rec, 2, ""); check_record_string(rec, 3, "2"); g_object_unref( rec ); rec = libmsi_query_fetch(query, NULL); ok(rec, "Expected result\n"); check_record_string(rec, 1, "3"); check_record_string(rec, 2, ""); check_record_string(rec, 3, "4"); g_object_unref( rec ); query_check_no_more(query); libmsi_query_close(query, NULL); g_object_unref( query ); r = try_query( hdb, "SELECT '' FROM `t` WHERE `t`.`b` = 'x'" ); ok( r == LIBMSI_RESULT_SUCCESS , "query failed: %u\n", r ); r = try_query( hdb, "SELECT `` FROM `t` WHERE `t`.`b` = 'x'" ); todo_wine ok( r == LIBMSI_RESULT_BAD_QUERY_SYNTAX, "query failed: %u\n", r ); r = try_query( hdb, "SELECT `b` FROM 't' WHERE `t`.`b` = 'x'" ); ok( r == LIBMSI_RESULT_BAD_QUERY_SYNTAX, "query failed: %u\n", r ); r = try_query( hdb, "SELECT `b` FROM `t` WHERE 'b' = 'x'" ); ok( r == LIBMSI_RESULT_BAD_QUERY_SYNTAX, "query failed: %u\n", r ); r = try_query( hdb, "SELECT `t`.`b`, `` FROM `t` WHERE `t`.`b` = 'x'" ); todo_wine ok( r == LIBMSI_RESULT_BAD_QUERY_SYNTAX, "query failed: %u\n", r ); g_object_unref( hdb ); } void main() { #if !GLIB_CHECK_VERSION(2,35,1) g_type_init (); #endif test_msidatabase(); test_msiinsert(); test_msibadqueries(); test_querygetcolumninfo(); test_getcolinfo(); test_msiexport(); test_longstrings(); test_streamtable(); test_binary(); test_where_not_in_selected(); test_where(); test_msiimport(); test_binary_import(); test_markers(); test_handle_limit(); #if 0 test_try_transform(); #endif test_join(); test_temporary_table(); test_alter(); test_integers(); test_update(); test_special_tables(); test_tables_order(); test_rows_order(); test_select_markers(); #if 0 test_stringtable(); #endif test_defaultdatabase(); test_order(); #if 0 test_deleterow(); #endif test_quotes(); test_carriagereturn(); test_noquotes(); test_forcecodepage(); #if 0 test_storages_table(); #endif test_droptable(); #if 0 test_dbmerge(); #endif test_select_with_tablenames(); test_insertorder(); test_columnorder(); test_suminfo_import(); #if 0 test_createtable(); #endif test_collation(); test_embedded_nulls(); test_select_column_names(); }