sxs: Implement surrogate data lookup in SxsLookupClrGuid().

dlls/sxs/sxs.c

@@ -60,6 +60,7 @@ typedef struct _SXS_GUID_INFORMATION_CLR
 #define SXS_LOOKUP_CLR_GUID_USE_ACTCTX     0x00000001
 #define SXS_LOOKUP_CLR_GUID_FIND_SURROGATE 0x00010000
 #define SXS_LOOKUP_CLR_GUID_FIND_CLR_CLASS 0x00020000
+#define SXS_LOOKUP_CLR_GUID_FIND_ANY       (SXS_LOOKUP_CLR_GUID_FIND_SURROGATE | SXS_LOOKUP_CLR_GUID_FIND_CLR_CLASS)
 
 struct comclassredirect_data
 {
@@ -96,27 +97,54 @@ struct clrclass_data
     DWORD res2[2];
 };
 
+struct clrsurrogate_data
+{
+    ULONG size;
+    DWORD res;
+    GUID  clsid;
+    ULONG version_offset;
+    ULONG version_len;
+    ULONG name_offset;
+    ULONG name_len;
+};
+
 BOOL WINAPI SxsLookupClrGuid(DWORD flags, GUID *clsid, HANDLE actctx, void *buffer, SIZE_T buffer_len,
                              SIZE_T *buffer_len_required)
 {
     ACTCTX_SECTION_KEYED_DATA guid_info = { sizeof(ACTCTX_SECTION_KEYED_DATA) };
     ACTIVATION_CONTEXT_ASSEMBLY_DETAILED_INFORMATION *assembly_info;
     SIZE_T bytes_assembly_info;
-    struct comclassredirect_data *redirect_data;
-    struct clrclass_data *class_data;
-    int len_version = 0, len_name, len_identity;
+    unsigned int len_version = 0, len_name, len_identity;
     const void *ptr_name, *ptr_version, *ptr_identity;
     SXS_GUID_INFORMATION_CLR *ret = buffer;
+    BOOL retval = FALSE;
     char *ret_strings;
 
-    TRACE("(%x, %s, %p, %p, %08lx, %p): stub\n", flags, wine_dbgstr_guid(clsid), actctx,
+    TRACE("%#x, %s, %p, %p, %lx, %p.\n", flags, wine_dbgstr_guid(clsid), actctx,
           buffer, buffer_len, buffer_len_required);
 
-    if (flags & ~SXS_LOOKUP_CLR_GUID_FIND_CLR_CLASS)
-        FIXME("Ignored flags: %x\n", flags & ~SXS_LOOKUP_CLR_GUID_FIND_CLR_CLASS);
+    if (flags & ~SXS_LOOKUP_CLR_GUID_FIND_ANY)
+        FIXME("Ignored flags: %x\n", flags & ~SXS_LOOKUP_CLR_GUID_FIND_ANY);
+
+    if (flags & SXS_LOOKUP_CLR_GUID_FIND_SURROGATE)
+    {
+        if ((retval = FindActCtxSectionGuid(FIND_ACTCTX_SECTION_KEY_RETURN_HACTCTX, 0,
+                ACTIVATION_CONTEXT_SECTION_CLR_SURROGATES, clsid, &guid_info)))
+        {
+            flags &= ~SXS_LOOKUP_CLR_GUID_FIND_CLR_CLASS;
+        }
+    }
+
+    if (!retval && (flags & SXS_LOOKUP_CLR_GUID_FIND_CLR_CLASS))
+    {
+        if ((retval = FindActCtxSectionGuid(FIND_ACTCTX_SECTION_KEY_RETURN_HACTCTX, 0,
+                ACTIVATION_CONTEXT_SECTION_COM_SERVER_REDIRECTION, clsid, &guid_info)))
+        {
+            flags &= ~SXS_LOOKUP_CLR_GUID_FIND_SURROGATE;
+        }
+    }
 
-    if (!FindActCtxSectionGuid(FIND_ACTCTX_SECTION_KEY_RETURN_HACTCTX, 0,
-                               ACTIVATION_CONTEXT_SECTION_COM_SERVER_REDIRECTION, clsid, &guid_info))
+    if (!retval)
     {
         SetLastError(ERROR_NOT_FOUND);
         return FALSE;
@@ -139,17 +167,30 @@ BOOL WINAPI SxsLookupClrGuid(DWORD flags, GUID *clsid, HANDLE actctx, void *buff
         return FALSE;
     }
 
-    redirect_data = guid_info.lpData;
-    class_data = (void *)((char*)redirect_data + redirect_data->clrdata_offset);
+    if (flags & SXS_LOOKUP_CLR_GUID_FIND_CLR_CLASS)
+    {
+        const struct comclassredirect_data *redirect_data = guid_info.lpData;
+        const struct clrclass_data *class_data;
 
-    ptr_identity = assembly_info->lpAssemblyEncodedAssemblyIdentity;
+        class_data = (void *)((char *)redirect_data + redirect_data->clrdata_offset);
         ptr_name = (char *)class_data + class_data->name_offset;
         ptr_version = (char *)class_data + class_data->version_offset;
-
-    len_identity = assembly_info->ulEncodedAssemblyIdentityLength + sizeof(WCHAR);
         len_name = class_data->name_len + sizeof(WCHAR);
-    if (class_data->version_len > 0)
+        if (class_data->version_len)
             len_version = class_data->version_len + sizeof(WCHAR);
+    }
+    else
+    {
+        const struct clrsurrogate_data *surrogate = guid_info.lpData;
+        ptr_name = (char *)surrogate + surrogate->name_offset;
+        ptr_version = (char *)surrogate + surrogate->version_offset;
+        len_name = surrogate->name_len + sizeof(WCHAR);
+        if (surrogate->version_len)
+            len_version = surrogate->version_len + sizeof(WCHAR);
+    }
+
+    ptr_identity = assembly_info->lpAssemblyEncodedAssemblyIdentity;
+    len_identity = assembly_info->ulEncodedAssemblyIdentityLength + sizeof(WCHAR);
 
     *buffer_len_required = sizeof(SXS_GUID_INFORMATION_CLR) + len_identity + len_version + len_name;
     if (!buffer || buffer_len < *buffer_len_required)
@@ -160,11 +201,12 @@ BOOL WINAPI SxsLookupClrGuid(DWORD flags, GUID *clsid, HANDLE actctx, void *buff
         return FALSE;
     }
 
-    ret->cbSize = sizeof(SXS_GUID_INFORMATION_CLR);
-    ret->dwFlags = SXS_GUID_INFORMATION_CLR_FLAG_IS_CLASS;
+    ret->cbSize = sizeof(*ret);
+    ret->dwFlags = flags & SXS_LOOKUP_CLR_GUID_FIND_CLR_CLASS ? SXS_GUID_INFORMATION_CLR_FLAG_IS_CLASS :
+            SXS_GUID_INFORMATION_CLR_FLAG_IS_SURROGATE;
 
     /* Copy strings into buffer */
-    ret_strings = (char *)ret + sizeof(SXS_GUID_INFORMATION_CLR);
+    ret_strings = (char *)ret + sizeof(*ret);
 
     memcpy(ret_strings, ptr_identity, len_identity);
     ret->pcwszAssemblyIdentity = (WCHAR *)ret_strings;

dlls/sxs/tests/comtest_dll.manifest

@@ -11,6 +11,11 @@
         name="DLL.Test"
         runtimeVersion="v4.0.0.0">
     </clrClass>
+    <clrSurrogate
+        clsid="{2e106e50-e7a4-4489-8538-83643f100fdd}"
+        name="Surrogate.Test"
+        runtimeVersion="v4.0.0.1">
+    </clrSurrogate>
     <file name="comtest.dll">
     </file>
 </assembly>

dlls/sxs/tests/interfaces.idl

@@ -33,3 +33,8 @@ interface ITest : IUnknown {
     uuid(2e106e50-e7a4-4489-8538-83643f100fdc),
 ]
 coclass Test { interface ITest; };
+
+[
+    uuid(2e106e50-e7a4-4489-8538-83643f100fdd),
+]
+coclass SurrogateTest { interface ITest; };

dlls/sxs/tests/sxs.c

@@ -86,6 +86,10 @@ static void run_test(void)
     BOOL ret;
     SXS_GUID_INFORMATION_CLR *info;
 
+    ret = SxsLookupClrGuid(SXS_LOOKUP_CLR_GUID_FIND_ANY, (GUID*)&CLSID_Test, NULL, NULL, 0, &buffer_size);
+    ok(!ret, "Unexpected return value %d.\n", ret);
+    ok(GetLastError() == ERROR_INSUFFICIENT_BUFFER, "Got %d\n", GetLastError());
+
     ret = SxsLookupClrGuid(SXS_LOOKUP_CLR_GUID_FIND_CLR_CLASS, (GUID*)&CLSID_Test, NULL, NULL, 0, &buffer_size);
     ok(ret == FALSE, "Got %d\n", ret);
     ok(GetLastError() == ERROR_INSUFFICIENT_BUFFER, "Got %d\n", GetLastError());
@@ -103,6 +107,29 @@ static void run_test(void)
            wine_dbgstr_w(info->pcwszRuntimeVersion));
 
     heap_free(info);
+
+    ret = SxsLookupClrGuid(SXS_LOOKUP_CLR_GUID_FIND_SURROGATE, (GUID *)&CLSID_SurrogateTest, NULL, NULL, 0, &buffer_size);
+    ok(!ret, "Unexpected return value %d.\n", ret);
+    ok(GetLastError() == ERROR_INSUFFICIENT_BUFFER, "Got %d\n", GetLastError());
+
+    info = heap_alloc(buffer_size);
+    ret = SxsLookupClrGuid(SXS_LOOKUP_CLR_GUID_FIND_SURROGATE, (GUID *)&CLSID_SurrogateTest, NULL, info,
+            buffer_size, &buffer_size);
+    ok(ret, "Unexpected return value %d.\n", ret);
+    ok(GetLastError() == 0, "Got %d\n", GetLastError());
+
+    ok(info->dwFlags == SXS_GUID_INFORMATION_CLR_FLAG_IS_SURROGATE, "Unexpected flags %#x.\n", info->dwFlags);
+    ok(!lstrcmpW(info->pcwszTypeName, L"Surrogate.Test"), "Unexpected typename %s.\n", wine_dbgstr_w(info->pcwszTypeName));
+    ok(!lstrcmpW(info->pcwszAssemblyIdentity, L"comtest,type=\"win32\",version=\"1.0.0.0\""),
+           "Unexpected assembly identity %s.\n", wine_dbgstr_w(info->pcwszAssemblyIdentity));
+    ok(!lstrcmpW(info->pcwszRuntimeVersion, L"v4.0.0.1"), "Unexpected runtime version %s.\n",
+           wine_dbgstr_w(info->pcwszRuntimeVersion));
+
+    heap_free(info);
+
+    ret = SxsLookupClrGuid(SXS_LOOKUP_CLR_GUID_FIND_ANY, (GUID *)&CLSID_SurrogateTest, NULL, NULL, 0, &buffer_size);
+    ok(!ret, "Unexpected return value %d.\n", ret);
+    ok(GetLastError() == ERROR_INSUFFICIENT_BUFFER, "Got %d\n", GetLastError());
 }
 
 static void prepare_and_run_test(void)
@@ -199,10 +226,16 @@ static void test_SxsLookupClrGuid(void)
     SIZE_T buffer_size;
     BOOL ret;
 
+    SetLastError(0xdeadbeef);
     ret = SxsLookupClrGuid(SXS_LOOKUP_CLR_GUID_FIND_CLR_CLASS, (GUID*)&CLSID_Test, NULL, NULL, 0, &buffer_size);
     ok(ret == FALSE, "Expected FALSE, got %d\n", ret);
     ok(GetLastError() == ERROR_NOT_FOUND, "Expected ERROR_NOT_FOUND, got %d\n", GetLastError());
 
+    SetLastError(0xdeadbeef);
+    ret = SxsLookupClrGuid(SXS_LOOKUP_CLR_GUID_FIND_SURROGATE, (GUID *)&CLSID_Test, NULL, NULL, 0, &buffer_size);
+    ok(!ret, "Unexpected return value %d.\n", ret);
+    ok(GetLastError() == ERROR_NOT_FOUND, "Expected ERROR_NOT_FOUND, got %d\n", GetLastError());
+
     run_child_process();
 }