dlls/sxs/tests/sxs.c

   1 /*
   2  * Copyright 2018 Fabian Maurer
   3  *
   4  * This library is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU Lesser General Public
   6  * License as published by the Free Software Foundation; either
   7  * version 2.1 of the License, or (at your option) any later version.
   8  *
   9  * This library is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * Lesser General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU Lesser General Public
  15  * License along with this library; if not, write to the Free Software
  16  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  17  */
  18
  19 #include <stdio.h>
  20
  21 #define COBJMACROS
  22
  23 #include <windows.h>
  24 #include <winsxs.h>
  25 #include <corerror.h>
  26 #include "shlwapi.h"
  27
  28 #include "wine/test.h"
  29
  30 #include "initguid.h"
  31 #include "interfaces.h"
  32
  33 #define SXS_LOOKUP_CLR_GUID_USE_ACTCTX     0x00000001
  34 #define SXS_LOOKUP_CLR_GUID_FIND_SURROGATE 0x00010000
  35 #define SXS_LOOKUP_CLR_GUID_FIND_CLR_CLASS 0x00020000
  36 #define SXS_LOOKUP_CLR_GUID_FIND_ANY       (SXS_LOOKUP_CLR_GUID_FIND_SURROGATE | SXS_LOOKUP_CLR_GUID_FIND_CLR_CLASS)
  37
  38 #define SXS_GUID_INFORMATION_CLR_FLAG_IS_SURROGATE  0x00000001
  39 #define SXS_GUID_INFORMATION_CLR_FLAG_IS_CLASS      0x00000002
  40
  41 typedef struct _SXS_GUID_INFORMATION_CLR
  42 {
  43     DWORD cbSize;
  44     DWORD dwFlags;
  45     PCWSTR pcwszRuntimeVersion;
  46     PCWSTR pcwszTypeName;
  47     PCWSTR pcwszAssemblyIdentity;
  48 } SXS_GUID_INFORMATION_CLR;
  49
  50 /* Defined in sxs.dll, but not found in any header */
  51 BOOL WINAPI SxsLookupClrGuid(DWORD flags, GUID *clsid, HANDLE actctx, void *buffer, SIZE_T buffer_len, SIZE_T *buffer_len_required);
  52
  53 static BOOL write_resource_file(const char *path_tmp, const char *name_res, const char *name_file, char *path_file)
  54 {
  55     HRSRC rsrc;
  56     void *rsrc_data;
  57     DWORD rsrc_size;
  58     BOOL ret;
  59     HANDLE hfile;
  60
  61     path_file[0] = 0;
  62     rsrc = FindResourceA(GetModuleHandleA(NULL), name_res, (LPCSTR)RT_RCDATA);
  63     if (!rsrc) return FALSE;
  64
  65     rsrc_data = LockResource(LoadResource(GetModuleHandleA(NULL), rsrc));
  66     if (!rsrc_data) return FALSE;
  67
  68     rsrc_size = SizeofResource(GetModuleHandleA(NULL), rsrc);
  69     if (!rsrc_size) return FALSE;
  70
  71     strcpy(path_file, path_tmp);
  72     PathAppendA(path_file, name_file);
  73     hfile = CreateFileA(path_file, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, 0);
  74     if (hfile == INVALID_HANDLE_VALUE) return FALSE;
  75
  76     ret = WriteFile(hfile, rsrc_data, rsrc_size, &rsrc_size, NULL);
  77
  78     CloseHandle(hfile);
  79     return ret;
  80 }
  81
  82 static void run_test(void)
  83 {
  84     SIZE_T buffer_size;
  85     BOOL ret;
  86     SXS_GUID_INFORMATION_CLR *info;
  87
  88     SetLastError(0xdeadbeef);
  89     ret = SxsLookupClrGuid(SXS_LOOKUP_CLR_GUID_FIND_ANY | SXS_LOOKUP_CLR_GUID_USE_ACTCTX, (GUID *)&CLSID_Test,
  90             NULL, NULL, 0, &buffer_size);
  91     ok(!ret, "Unexpected return value %d.\n", ret);
  92     ok(GetLastError() == ERROR_NOT_FOUND, "Unexpected error %ld.\n", GetLastError());
  93
  94     SetLastError(0xdeadbeef);
  95     ret = SxsLookupClrGuid(SXS_LOOKUP_CLR_GUID_FIND_ANY | SXS_LOOKUP_CLR_GUID_USE_ACTCTX, (GUID *)&CLSID_Test,
  96             NULL, NULL, 0, &buffer_size);
  97     ok(!ret, "Unexpected return value %d.\n", ret);
  98     ok(GetLastError() == ERROR_NOT_FOUND, "Unexpected error %ld.\n", GetLastError());
  99
 100     SetLastError(0xdeadbeef);
 101     ret = SxsLookupClrGuid(SXS_LOOKUP_CLR_GUID_FIND_ANY, (GUID*)&CLSID_Test, NULL, NULL, 0, &buffer_size);
 102     ok(!ret, "Unexpected return value %d.\n", ret);
 103     ok(GetLastError() == ERROR_INSUFFICIENT_BUFFER, "Got %ld\n", GetLastError());
 104
 105     SetLastError(0xdeadbeef);
 106     ret = SxsLookupClrGuid(SXS_LOOKUP_CLR_GUID_FIND_CLR_CLASS, (GUID*)&CLSID_Test, NULL, NULL, 0, &buffer_size);
 107     ok(ret == FALSE, "Got %d\n", ret);
 108     ok(GetLastError() == ERROR_INSUFFICIENT_BUFFER, "Got %ld\n", GetLastError());
 109
 110     info = malloc(buffer_size);
 111     SetLastError(0xdeadbeef);
 112     ret = SxsLookupClrGuid(SXS_LOOKUP_CLR_GUID_FIND_CLR_CLASS, (GUID*)&CLSID_Test, NULL, info, buffer_size, &buffer_size);
 113     ok(ret == TRUE, "Got %d\n", ret);
 114     ok(GetLastError() == 0, "Got %ld\n", GetLastError());
 115
 116     ok(info->dwFlags == SXS_GUID_INFORMATION_CLR_FLAG_IS_CLASS, "Got %ld\n", info->dwFlags);
 117     ok(!lstrcmpW(info->pcwszTypeName, L"DLL.Test"), "Unexpected typename %s.\n", wine_dbgstr_w(info->pcwszTypeName));
 118     ok(!lstrcmpW(info->pcwszAssemblyIdentity, L"comtest,type=\"win32\",version=\"1.0.0.0\""),
 119            "Unexpected assembly identity %s.\n", wine_dbgstr_w(info->pcwszAssemblyIdentity));
 120     ok(!lstrcmpW(info->pcwszRuntimeVersion, L"v4.0.0.0"), "Unexpected runtime version %s.\n",
 121            wine_dbgstr_w(info->pcwszRuntimeVersion));
 122
 123     free(info);
 124
 125     SetLastError(0xdeadbeef);
 126     ret = SxsLookupClrGuid(SXS_LOOKUP_CLR_GUID_FIND_SURROGATE, (GUID *)&CLSID_SurrogateTest, NULL, NULL, 0, &buffer_size);
 127     ok(!ret, "Unexpected return value %d.\n", ret);
 128     ok(GetLastError() == ERROR_INSUFFICIENT_BUFFER, "Got %ld\n", GetLastError());
 129
 130     info = malloc(buffer_size);
 131     SetLastError(0xdeadbeef);
 132     ret = SxsLookupClrGuid(SXS_LOOKUP_CLR_GUID_FIND_SURROGATE, (GUID *)&CLSID_SurrogateTest, NULL, info,
 133             buffer_size, &buffer_size);
 134     ok(ret, "Unexpected return value %d.\n", ret);
 135     ok(GetLastError() == 0, "Got %ld\n", GetLastError());
 136
 137     ok(info->dwFlags == SXS_GUID_INFORMATION_CLR_FLAG_IS_SURROGATE, "Unexpected flags %#lx.\n", info->dwFlags);
 138     ok(!lstrcmpW(info->pcwszTypeName, L"Surrogate.Test"), "Unexpected typename %s.\n", wine_dbgstr_w(info->pcwszTypeName));
 139     ok(!lstrcmpW(info->pcwszAssemblyIdentity, L"comtest,type=\"win32\",version=\"1.0.0.0\""),
 140            "Unexpected assembly identity %s.\n", wine_dbgstr_w(info->pcwszAssemblyIdentity));
 141     ok(!lstrcmpW(info->pcwszRuntimeVersion, L"v4.0.0.1"), "Unexpected runtime version %s.\n",
 142            wine_dbgstr_w(info->pcwszRuntimeVersion));
 143
 144     free(info);
 145
 146     SetLastError(0xdeadbeef);
 147     ret = SxsLookupClrGuid(SXS_LOOKUP_CLR_GUID_FIND_ANY, (GUID *)&CLSID_SurrogateTest, NULL, NULL, 0, &buffer_size);
 148     ok(!ret, "Unexpected return value %d.\n", ret);
 149     ok(GetLastError() == ERROR_INSUFFICIENT_BUFFER, "Got %ld\n", GetLastError());
 150 }
 151
 152 static void prepare_and_run_test(void)
 153 {
 154     char path_tmp[MAX_PATH];
 155     char path_manifest_dll[MAX_PATH];
 156     char path_manifest_exe[MAX_PATH];
 157     BOOL success;
 158     ACTCTXA context = {0};
 159     ULONG_PTR cookie;
 160     HANDLE handle_context = INVALID_HANDLE_VALUE;
 161
 162     GetTempPathA(MAX_PATH, path_tmp);
 163
 164     if (!write_resource_file(path_tmp, "comtest_exe.manifest", "exe.manifest", path_manifest_exe))
 165     {
 166         ok(0, "Failed to create file for testing\n");
 167         goto cleanup;
 168     }
 169
 170     if (!write_resource_file(path_tmp, "comtest_dll.manifest", "comtest.manifest", path_manifest_dll))
 171     {
 172         ok(0, "Failed to create file for testing\n");
 173         goto cleanup;
 174     }
 175
 176     context.cbSize = sizeof(ACTCTXA);
 177     context.lpSource = path_manifest_exe;
 178     context.lpAssemblyDirectory = path_tmp;
 179     context.dwFlags = ACTCTX_FLAG_ASSEMBLY_DIRECTORY_VALID;
 180
 181     handle_context = CreateActCtxA(&context);
 182     ok(handle_context != INVALID_HANDLE_VALUE, "CreateActCtxA failed: %ld\n", GetLastError());
 183
 184     if (handle_context == INVALID_HANDLE_VALUE)
 185     {
 186         ok(0, "Failed to create activation context\n");
 187         goto cleanup;
 188     }
 189
 190     success = ActivateActCtx(handle_context, &cookie);
 191     ok(success, "ActivateActCtx failed: %ld\n", GetLastError());
 192
 193     run_test();
 194
 195 cleanup:
 196     if (handle_context != INVALID_HANDLE_VALUE)
 197     {
 198         success = DeactivateActCtx(0, cookie);
 199         ok(success, "DeactivateActCtx failed: %ld\n", GetLastError());
 200         ReleaseActCtx(handle_context);
 201     }
 202     if (*path_manifest_exe)
 203     {
 204         success = DeleteFileA(path_manifest_exe);
 205         ok(success, "DeleteFileA failed: %ld\n", GetLastError());
 206     }
 207     if(*path_manifest_dll)
 208     {
 209         success = DeleteFileA(path_manifest_dll);
 210         ok(success, "DeleteFileA failed: %ld\n", GetLastError());
 211     }
 212 }
 213
 214 static void run_child_process(void)
 215 {
 216     char cmdline[MAX_PATH];
 217     char exe[MAX_PATH];
 218     char **argv;
 219     PROCESS_INFORMATION pi;
 220     STARTUPINFOA si = { 0 };
 221     BOOL ret;
 222
 223     winetest_get_mainargs(&argv);
 224
 225     if (strstr(argv[0], ".exe"))
 226         sprintf(exe, "%s", argv[0]);
 227     else
 228         sprintf(exe, "%s.exe", argv[0]);
 229     sprintf(cmdline, "\"%s\" %s %s", argv[0], argv[1], "subtest");
 230
 231     si.cb = sizeof(si);
 232     ret = CreateProcessA(exe, cmdline, NULL, NULL, FALSE, 0, NULL, NULL, &si, &pi);
 233     ok(ret, "Could not create process: %lu\n", GetLastError());
 234
 235     wait_child_process(pi.hProcess);
 236
 237     CloseHandle(pi.hThread);
 238     CloseHandle(pi.hProcess);
 239 }
 240
 241 static void test_SxsLookupClrGuid(void)
 242 {
 243     SIZE_T buffer_size;
 244     BOOL ret;
 245
 246     SetLastError(0xdeadbeef);
 247     ret = SxsLookupClrGuid(SXS_LOOKUP_CLR_GUID_FIND_CLR_CLASS, (GUID*)&CLSID_Test, NULL, NULL, 0, &buffer_size);
 248     ok(ret == FALSE, "Expected FALSE, got %d\n", ret);
 249     ok(GetLastError() == ERROR_NOT_FOUND, "Expected ERROR_NOT_FOUND, got %ld\n", GetLastError());
 250
 251     SetLastError(0xdeadbeef);
 252     ret = SxsLookupClrGuid(SXS_LOOKUP_CLR_GUID_FIND_CLR_CLASS | SXS_LOOKUP_CLR_GUID_USE_ACTCTX, (GUID *)&CLSID_Test,
 253             NULL, NULL, 0, &buffer_size);
 254     ok(!ret, "Unexpected return value %d.\n", ret);
 255     ok(GetLastError() == ERROR_NOT_FOUND, "Expected ERROR_NOT_FOUND, got %ld\n", GetLastError());
 256
 257     SetLastError(0xdeadbeef);
 258     ret = SxsLookupClrGuid(SXS_LOOKUP_CLR_GUID_FIND_SURROGATE, (GUID *)&CLSID_Test, NULL, NULL, 0, &buffer_size);
 259     ok(!ret, "Unexpected return value %d.\n", ret);
 260     ok(GetLastError() == ERROR_NOT_FOUND, "Expected ERROR_NOT_FOUND, got %ld\n", GetLastError());
 261
 262     run_child_process();
 263 }
 264
 265 START_TEST(sxs)
 266 {
 267     char **argv;
 268     int argc = winetest_get_mainargs(&argv);
 269     if (argc > 2)
 270     {
 271         prepare_and_run_test();
 272         return;
 273     }
 274
 275     test_SxsLookupClrGuid();
 276 }