[APPHELP_APITEST] Add tests for SdbGetMatchingExe, to test matching on version resour...

svn path=/trunk/; revision=75205

diff --git a/rostests/apitests/apphelp/CMakeLists.txt b/rostests/apitests/apphelp/CMakeLists.txt
index 454723f..ec218d2 100644 (file)
--- a/rostests/apitests/apphelp/CMakeLists.txt
+++ b/rostests/apitests/apphelp/CMakeLists.txt
@@ -1,14 +1,24 @@
 project(appcompat)
 add_definitions(-D__ROS_LONG64__ -DWINETEST_USE_DBGSTR_LONGLONG)
 
 project(appcompat)
 add_definitions(-D__ROS_LONG64__ -DWINETEST_USE_DBGSTR_LONGLONG)
 

+include_directories(${CMAKE_CURRENT_BINARY_DIR})
+

 list(APPEND SOURCE
     apphelp.c
     data.c
     db.cpp
     env.c
     layerapi.c
 list(APPEND SOURCE
     apphelp.c
     data.c
     db.cpp
     env.c
     layerapi.c

-    testlist.c)
+    testlist.c
+    testdata.rc
+    testdb.xml)

 
 

+add_custom_command(OUTPUT ${CMAKE_CURRENT_BINARY_DIR}/testdb.sdb
+                   COMMAND native-xml2sdb -i ${CMAKE_CURRENT_SOURCE_DIR}/testdb.xml -o ${CMAKE_CURRENT_BINARY_DIR}/testdb.sdb
+                   DEPENDS ${CMAKE_CURRENT_SOURCE_DIR}/testdb.xml native-xml2sdb)
+                   
+add_custom_target(testdb DEPENDS ${CMAKE_CURRENT_BINARY_DIR}/testdb.sdb)
+add_rc_deps(testdata.rc testdb)

 add_executable(apphelp_apitest ${SOURCE})
 set_module_type(apphelp_apitest win32cui)
 target_link_libraries(apphelp_apitest ${PSEH_LIB})
 add_executable(apphelp_apitest ${SOURCE})
 set_module_type(apphelp_apitest win32cui)
 target_link_libraries(apphelp_apitest ${PSEH_LIB})

diff --git a/rostests/apitests/apphelp/db.cpp b/rostests/apitests/apphelp/db.cpp
index ad3f053..8fb1bc6 100644 (file)
--- a/rostests/apitests/apphelp/db.cpp
+++ b/rostests/apitests/apphelp/db.cpp
@@ -1332,6 +1332,147 @@ static void test_MatchApplications(void)
     ok(ret, "RemoveDirectoryW error: %d\n", GetLastError());
 }
 
     ok(ret, "RemoveDirectoryW error: %d\n", GetLastError());
 }
 

+static BOOL write_raw_file(const WCHAR* FileName, const void* Data, DWORD Size)
+{
+    BOOL Success;
+    DWORD dwWritten;
+    HANDLE Handle = CreateFileW(FileName, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
+
+    if (Handle == INVALID_HANDLE_VALUE)
+    {
+        skip("Failed to create temp file %ls, error %u\n", FileName, GetLastError());
+        return FALSE;
+    }
+    Success = WriteFile(Handle, Data, Size, &dwWritten, NULL);
+    ok(Success == TRUE, "WriteFile failed with %u\n", GetLastError());
+    ok(dwWritten == Size, "WriteFile wrote %u bytes instead of %u\n", dwWritten, Size);
+    CloseHandle(Handle);
+    return Success && (dwWritten == Size);
+}
+
+static bool extract_resource(const WCHAR* Filename, LPCWSTR ResourceName)
+{
+    HMODULE hMod = GetModuleHandleW(NULL);
+    HRSRC hRsrc = FindResourceW(hMod, ResourceName, MAKEINTRESOURCEW(RT_RCDATA));
+    ok(!!hRsrc, "Unable to find %s\n", wine_dbgstr_w(ResourceName));
+    if (!hRsrc)
+        return false;
+
+    HGLOBAL hGlobal = LoadResource(hMod, hRsrc);
+    DWORD Size = SizeofResource(hMod, hRsrc);
+    LPVOID pData = LockResource(hGlobal);
+
+    ok(Size && !!pData, "Unable to load %s\n", wine_dbgstr_w(ResourceName));
+    if (!Size || !pData)
+        return false;
+
+    BOOL Written = write_raw_file(Filename, pData, Size);
+    UnlockResource(pData);
+    return Written;
+}
+
+template<typename SDBQUERYRESULT_T>
+static BOOL test_match_ex(const WCHAR* workdir, HSDB hsdb, int cur)
+{
+    WCHAR exename[MAX_PATH];
+    WCHAR* Vendor;
+    SDBQUERYRESULT_T query;
+    TAGID tagid, exetag;
+    BOOL ret, Succeed;
+    PDB pdb;
+
+    memset(&query, 0xab, sizeof(query));
+
+    swprintf(exename, L"%s\\test_match%d.exe", workdir, cur);
+
+    ret = pSdbTagRefToTagID(hsdb, 0, &pdb, &tagid);
+    ok(pdb != NULL && pdb != (PDB)0x12345678, "Expected pdb to be set to a valid pdb, was: %p\n", pdb);
+
+    tagid = pSdbFindFirstTag(pdb, TAGID_ROOT, TAG_DATABASE);
+    ok(tagid != TAGID_NULL, "Expected to get a valid TAG_DATABASE\n");
+
+    exetag = pSdbFindFirstNamedTag(pdb, tagid, TAG_EXE, TAG_NAME, exename + wcslen(workdir) + 1);
+
+    if (!exetag)
+    {
+        /* Test done */
+        return FALSE;
+    }
+
+    tagid = pSdbFindFirstTag(pdb, exetag, TAG_VENDOR);
+    Vendor = pSdbGetStringTagPtr(pdb, tagid);
+    Succeed = tagid != TAGID_NULL && Vendor && !wcsicmp(Vendor, L"Succeed");
+
+    test_create_exe(exename, 0);
+
+    ret = pSdbGetMatchingExe(hsdb, exename, NULL, NULL, 0, (SDBQUERYRESULT_VISTA*)&query);
+    DWORD exe_count = Succeed ? 1 : 0;
+
+    if (Succeed)
+        ok(ret, "SdbGetMatchingExe should not fail for %d.\n", cur);
+    else
+        ok(!ret, "SdbGetMatchingExe should not succeed for %d.\n", cur);
+
+    ok(query.dwExeCount == exe_count, "Expected dwExeCount to be %d, was %d for %d\n", exe_count, query.dwExeCount, cur);
+    DeleteFileW(exename);
+    /* Try the next file */
+    return TRUE;
+}
+
+
+template<typename SDBQUERYRESULT_T>
+static void test_MatchApplicationsEx(void)
+{
+    WCHAR workdir[MAX_PATH], dbpath[MAX_PATH];
+    BOOL ret;
+    HSDB hsdb;
+
+    ret = GetTempPathW(_countof(workdir), workdir);
+    ok(ret, "GetTempPathW error: %d\n", GetLastError());
+    lstrcatW(workdir, L"apphelp_test");
+
+    ret = CreateDirectoryW(workdir, NULL);
+    ok(ret, "CreateDirectoryW error: %d\n", GetLastError());
+
+    /* SdbInitDatabase needs an nt-path */
+    swprintf(dbpath, L"\\??\\%s\\test.sdb", workdir);
+
+    if (extract_resource(dbpath + 4, MAKEINTRESOURCEW(101)))
+    {
+        hsdb = pSdbInitDatabase(HID_DATABASE_FULLPATH, dbpath);
+
+        ok(hsdb != NULL, "Expected a valid database handle\n");
+
+        if (!hsdb)
+        {
+            skip("SdbInitDatabase not implemented?\n");
+        }
+        else
+        {
+            size_t n;
+            /* now that our enviroment is setup, let's go ahead and run the actual tests.. */
+            for (n = 0;; ++n)
+            {
+                if (!test_match_ex<SDBQUERYRESULT_T>(workdir, hsdb, n))
+                    break;
+            }
+            pSdbReleaseDatabase(hsdb);
+        }
+    }
+    else
+    {
+        ok(0, "Unable to extract database\n");
+    }
+
+    DeleteFileW(dbpath + 4);
+
+    ret = RemoveDirectoryW(workdir);
+    ok(ret, "RemoveDirectoryW error: %d\n", GetLastError());
+}
+
+
+
+

 static void test_TagRef(void)
 {
     WCHAR tmpdir[MAX_PATH], dbpath[MAX_PATH];
 static void test_TagRef(void)
 {
     WCHAR tmpdir[MAX_PATH], dbpath[MAX_PATH];


@@ -1660,9 +1801,11 @@ START_TEST(db)
     {
     case 1:
         test_MatchApplications<SDBQUERYRESULT_2k3>();
     {
     case 1:
         test_MatchApplications<SDBQUERYRESULT_2k3>();

+        test_MatchApplicationsEx<SDBQUERYRESULT_2k3>();

         break;
     case 2:
         test_MatchApplications<SDBQUERYRESULT_VISTA>();
         break;
     case 2:
         test_MatchApplications<SDBQUERYRESULT_VISTA>();

+        test_MatchApplicationsEx<SDBQUERYRESULT_VISTA>();

         break;
     default:
         skip("Skipping tests with SDBQUERYRESULT due to a wrong size reported\n");
         break;
     default:
         skip("Skipping tests with SDBQUERYRESULT due to a wrong size reported\n");

diff --git a/rostests/apitests/apphelp/testdata.rc b/rostests/apitests/apphelp/testdata.rc
new file mode 100644 (file)
index 0000000..2c973f0
--- /dev/null
+++ b/rostests/apitests/apphelp/testdata.rc
@@ -0,0 +1,2 @@
+
+101 RCDATA "testdb.sdb"

diff --git a/rostests/apitests/apphelp/testdb.xml b/rostests/apitests/apphelp/testdb.xml
new file mode 100644 (file)
index 0000000..9158d32
--- /dev/null
+++ b/rostests/apitests/apphelp/testdb.xml
@@ -0,0 +1,235 @@
+<SDB>
+    <DATABASE>
+        <NAME>ReactOS test database</NAME>
+        <OS_PLATFORM >1</OS_PLATFORM>
+        <DATABASE_ID>{551F8E78-A9DA-44AC-A24C-5A8145317BC7}</DATABASE_ID>
+        <LIBRARY>
+        </LIBRARY>
+
+        <!-- Verify that we are able to match this -->
+        <EXE>
+            <NAME>test_match0.exe</NAME>
+            <APP_NAME>Generic name</APP_NAME>
+            <VENDOR>Succeed</VENDOR>
+            <MATCHING_FILE>
+                <NAME>*</NAME>
+                <SIZE>2048</SIZE>
+                <CHECKSUM>0x178BD629</CHECKSUM>
+                <PRODUCT_VERSION>1.0.0.1</PRODUCT_VERSION>
+                <!-- BIN_PRODUCT_VERSION? -->
+                <FILE_VERSION>1.0.0.0</FILE_VERSION>
+                <!-- BIN_FILE_VERSION? -->
+                <FILE_DESCRIPTION>FileDescription</FILE_DESCRIPTION>
+                <COMPANY_NAME>CompanyName</COMPANY_NAME>
+                <ORIGINAL_FILENAME>OriginalFilename</ORIGINAL_FILENAME>
+                <INTERNAL_NAME>InternalName</INTERNAL_NAME>
+            </MATCHING_FILE>
+        </EXE>
+        <!-- Verify that we are able to match this -->
+        <EXE>
+            <NAME>test_match1.exe</NAME>
+            <APP_NAME>Generic name</APP_NAME>
+            <VENDOR>Succeed</VENDOR>
+            <MATCHING_FILE>
+                <NAME>*</NAME>
+                <SIZE>2048</SIZE>
+                <CHECKSUM>0x178BD629</CHECKSUM>
+                <PRODUCT_VERSION>1.0.0.1</PRODUCT_VERSION>
+                <!-- BIN_PRODUCT_VERSION? -->
+                <FILE_VERSION>1.0.0.0</FILE_VERSION>
+                <!-- BIN_FILE_VERSION? -->
+                <FILE_DESCRIPTION>FILEDESCRIPTION</FILE_DESCRIPTION>
+                <COMPANY_NAME>companyname</COMPANY_NAME>
+                <ORIGINAL_FILENAME>ORIGINALFILENAME</ORIGINAL_FILENAME>
+                <INTERNAL_NAME>internalname</INTERNAL_NAME>
+            </MATCHING_FILE>
+        </EXE>
+        <!-- SIZE wrong -->
+        <EXE>
+            <NAME>test_match2.exe</NAME>
+            <APP_NAME>Generic name</APP_NAME>
+            <VENDOR>Generic Description</VENDOR>
+            <MATCHING_FILE>
+                <NAME>*</NAME>
+                <SIZE>2047</SIZE>
+                <CHECKSUM>0x178BD629</CHECKSUM>
+                <PRODUCT_VERSION>1.0.0.1</PRODUCT_VERSION>
+                <FILE_VERSION>1.0.0.0</FILE_VERSION>
+                <FILE_DESCRIPTION>FileDescription</FILE_DESCRIPTION>
+                <COMPANY_NAME>CompanyName</COMPANY_NAME>
+                <ORIGINAL_FILENAME>OriginalFilename</ORIGINAL_FILENAME>
+                <INTERNAL_NAME>InternalName</INTERNAL_NAME>
+            </MATCHING_FILE>
+        </EXE>
+        <!-- CHECKSUM wrong -->
+        <EXE>
+            <NAME>test_match3.exe</NAME>
+            <APP_NAME>Generic name</APP_NAME>
+            <VENDOR>Generic Description</VENDOR>
+            <MATCHING_FILE>
+                <NAME>*</NAME>
+                <SIZE>2048</SIZE>
+                <CHECKSUM>0x111111</CHECKSUM>
+                <PRODUCT_VERSION>1.0.0.1</PRODUCT_VERSION>
+                <FILE_VERSION>1.0.0.0</FILE_VERSION>
+                <FILE_DESCRIPTION>FileDescription</FILE_DESCRIPTION>
+                <COMPANY_NAME>CompanyName</COMPANY_NAME>
+                <ORIGINAL_FILENAME>OriginalFilename</ORIGINAL_FILENAME>
+                <INTERNAL_NAME>InternalName</INTERNAL_NAME>
+            </MATCHING_FILE>
+        </EXE>
+        <!-- PRODUCT_VERSION too high -->
+        <EXE>
+            <NAME>test_match4.exe</NAME>
+            <APP_NAME>Generic name</APP_NAME>
+            <VENDOR>Generic Description</VENDOR>
+            <MATCHING_FILE>
+                <NAME>*</NAME>
+                <SIZE>2048</SIZE>
+                <CHECKSUM>0x178BD629</CHECKSUM>
+                <PRODUCT_VERSION>1.1.1.1</PRODUCT_VERSION>
+                <FILE_VERSION>1.0.0.0</FILE_VERSION>
+                <FILE_DESCRIPTION>FileDescription</FILE_DESCRIPTION>
+                <COMPANY_NAME>CompanyName</COMPANY_NAME>
+                <ORIGINAL_FILENAME>OriginalFilename</ORIGINAL_FILENAME>
+                <INTERNAL_NAME>InternalName</INTERNAL_NAME>
+            </MATCHING_FILE>
+        </EXE>
+        <!-- PRODUCT_VERSION too low -->
+        <EXE>
+            <NAME>test_match5.exe</NAME>
+            <APP_NAME>Generic name</APP_NAME>
+            <VENDOR>Generic Description</VENDOR>
+            <MATCHING_FILE>
+                <NAME>*</NAME>
+                <SIZE>2048</SIZE>
+                <CHECKSUM>0x178BD629</CHECKSUM>
+                <PRODUCT_VERSION>1.0.0.0</PRODUCT_VERSION>
+                <FILE_VERSION>1.0.0.0</FILE_VERSION>
+                <FILE_DESCRIPTION>FileDescription</FILE_DESCRIPTION>
+                <COMPANY_NAME>CompanyName</COMPANY_NAME>
+                <ORIGINAL_FILENAME>OriginalFilename</ORIGINAL_FILENAME>
+                <INTERNAL_NAME>InternalName</INTERNAL_NAME>
+            </MATCHING_FILE>
+        </EXE>
+        <!-- FILE_VERSION wrong -->
+        <EXE>
+            <NAME>test_match6.exe</NAME>
+            <APP_NAME>Generic name</APP_NAME>
+            <VENDOR>Generic Description</VENDOR>
+            <MATCHING_FILE>
+                <NAME>*</NAME>
+                <SIZE>2048</SIZE>
+                <CHECKSUM>0x178BD629</CHECKSUM>
+                <PRODUCT_VERSION>1.0.0.1</PRODUCT_VERSION>
+                <FILE_VERSION>1.1.1.1</FILE_VERSION>
+                <FILE_DESCRIPTION>FileDescription</FILE_DESCRIPTION>
+                <COMPANY_NAME>CompanyName</COMPANY_NAME>
+                <ORIGINAL_FILENAME>OriginalFilename</ORIGINAL_FILENAME>
+                <INTERNAL_NAME>InternalName</INTERNAL_NAME>
+            </MATCHING_FILE>
+        </EXE>
+        <!-- FILE_DESCRIPTION wrong -->
+        <EXE>
+            <NAME>test_match7.exe</NAME>
+            <APP_NAME>Generic name</APP_NAME>
+            <VENDOR>Generic Description</VENDOR>
+            <MATCHING_FILE>
+                <NAME>*</NAME>
+                <SIZE>2048</SIZE>
+                <CHECKSUM>0x178BD629</CHECKSUM>
+                <PRODUCT_VERSION>1.0.0.1</PRODUCT_VERSION>
+                <FILE_VERSION>1.0.0.0</FILE_VERSION>
+                <FILE_DESCRIPTION>Wrong Description</FILE_DESCRIPTION>
+                <COMPANY_NAME>CompanyName</COMPANY_NAME>
+                <ORIGINAL_FILENAME>OriginalFilename</ORIGINAL_FILENAME>
+                <INTERNAL_NAME>InternalName</INTERNAL_NAME>
+            </MATCHING_FILE>
+        </EXE>
+        <!-- COMPANY_NAME wrong -->
+        <EXE>
+            <NAME>test_match8.exe</NAME>
+            <APP_NAME>Generic name</APP_NAME>
+            <VENDOR>Generic Description</VENDOR>
+            <MATCHING_FILE>
+                <NAME>*</NAME>
+                <SIZE>2048</SIZE>
+                <CHECKSUM>0x178BD629</CHECKSUM>
+                <PRODUCT_VERSION>1.0.0.1</PRODUCT_VERSION>
+                <FILE_VERSION>1.0.0.0</FILE_VERSION>
+                <FILE_DESCRIPTION>FileDescription</FILE_DESCRIPTION>
+                <COMPANY_NAME>Wrong CompanyName</COMPANY_NAME>
+                <ORIGINAL_FILENAME>OriginalFilename</ORIGINAL_FILENAME>
+                <INTERNAL_NAME>InternalName</INTERNAL_NAME>
+            </MATCHING_FILE>
+        </EXE>
+        <!-- ORIGINAL_FILENAME wrong -->
+        <EXE>
+            <NAME>test_match9.exe</NAME>
+            <APP_NAME>Generic name</APP_NAME>
+            <VENDOR>Generic Description</VENDOR>
+            <MATCHING_FILE>
+                <NAME>*</NAME>
+                <SIZE>2048</SIZE>
+                <CHECKSUM>0x178BD629</CHECKSUM>
+                <PRODUCT_VERSION>1.0.0.1</PRODUCT_VERSION>
+                <FILE_VERSION>1.0.0.0</FILE_VERSION>
+                <FILE_DESCRIPTION>FileDescription</FILE_DESCRIPTION>
+                <COMPANY_NAME>CompanyName</COMPANY_NAME>
+                <ORIGINAL_FILENAME>Wrong OriginalFilename</ORIGINAL_FILENAME>
+                <INTERNAL_NAME>InternalName</INTERNAL_NAME>
+            </MATCHING_FILE>
+        </EXE>
+        <!-- INTERNAL_NAME wrong -->
+        <EXE>
+            <NAME>test_match10.exe</NAME>
+            <APP_NAME>Generic name</APP_NAME>
+            <VENDOR>Generic Description</VENDOR>
+            <MATCHING_FILE>
+                <NAME>*</NAME>
+                <SIZE>2048</SIZE>
+                <CHECKSUM>0x178BD629</CHECKSUM>
+                <PRODUCT_VERSION>1.0.0.1</PRODUCT_VERSION>
+                <FILE_VERSION>1.0.0.0</FILE_VERSION>
+                <FILE_DESCRIPTION>FileDescription</FILE_DESCRIPTION>
+                <COMPANY_NAME>CompanyName</COMPANY_NAME>
+                <ORIGINAL_FILENAME>OriginalFilename</ORIGINAL_FILENAME>
+                <INTERNAL_NAME>wrong InternalName</INTERNAL_NAME>
+            </MATCHING_FILE>
+        </EXE>
+        <!-- Show that strings are checked for the full length(1) -->
+        <EXE>
+            <NAME>test_match11.exe</NAME>
+            <APP_NAME>Generic name</APP_NAME>
+            <VENDOR>Generic Description</VENDOR>
+            <MATCHING_FILE>
+                <NAME>*</NAME>
+                <SIZE>2048</SIZE>
+                <CHECKSUM>0x178BD629</CHECKSUM>
+                <PRODUCT_VERSION>1.0.0.1</PRODUCT_VERSION>
+                <FILE_VERSION>1.0.0.0</FILE_VERSION>
+                <FILE_DESCRIPTION>FileDescription</FILE_DESCRIPTION>
+                <COMPANY_NAME>CompanyName</COMPANY_NAME>
+                <ORIGINAL_FILENAME>OriginalFilename</ORIGINAL_FILENAME>
+                <INTERNAL_NAME>InternalName wrong</INTERNAL_NAME>
+            </MATCHING_FILE>
+        </EXE>
+        <!-- Show that strings are checked for the full length(2) -->
+        <EXE>
+            <NAME>test_match12.exe</NAME>
+            <APP_NAME>Generic name</APP_NAME>
+            <VENDOR>Generic Description</VENDOR>
+            <MATCHING_FILE>
+                <NAME>*</NAME>
+                <SIZE>2048</SIZE>
+                <CHECKSUM>0x178BD629</CHECKSUM>
+                <PRODUCT_VERSION>1.0.0.1</PRODUCT_VERSION>
+                <FILE_VERSION>1.0.0.0</FILE_VERSION>
+                <FILE_DESCRIPTION>FileDescription</FILE_DESCRIPTION>
+                <COMPANY_NAME>CompanyName</COMPANY_NAME>
+                <ORIGINAL_FILENAME>OriginalFilename</ORIGINAL_FILENAME>
+                <INTERNAL_NAME>Internal</INTERNAL_NAME>
+            </MATCHING_FILE>
+        </EXE>
+    </DATABASE>
+</SDB>