dlls/sxs/sxs.c

   1 /*
   2  * sxs main
   3  *
   4  * Copyright 2007 EA Durbin
   5  *
   6  * This library is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2.1 of the License, or (at your option) any later version.
  10  *
  11  * This library is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public
  17  * License along with this library; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  19  */
  20
  21 #include <stdarg.h>
  22
  23 #include "windef.h"
  24 #include "winbase.h"
  25
  26 #include "wine/heap.h"
  27 #include "wine/debug.h"
  28
  29 WINE_DEFAULT_DEBUG_CHANNEL(sxs);
  30
  31 /***********************************************************************
  32  *             DllMain   (SXS.@)
  33  *
  34  */
  35 BOOL WINAPI DllMain(HINSTANCE hinstDLL, DWORD fdwReason, LPVOID lpvReserved)
  36 {
  37     switch(fdwReason)
  38     {
  39     case DLL_WINE_PREATTACH:
  40         return FALSE;  /* prefer native version */
  41     case DLL_PROCESS_ATTACH:
  42         DisableThreadLibraryCalls( hinstDLL );
  43         break;
  44     }
  45     return TRUE;
  46 }
  47
  48 typedef struct _SXS_GUID_INFORMATION_CLR
  49 {
  50     DWORD cbSize;
  51     DWORD dwFlags;
  52     PCWSTR pcwszRuntimeVersion;
  53     PCWSTR pcwszTypeName;
  54     PCWSTR pcwszAssemblyIdentity;
  55 } SXS_GUID_INFORMATION_CLR, *PSXS_GUID_INFORMATION_CLR;
  56
  57 #define SXS_GUID_INFORMATION_CLR_FLAG_IS_SURROGATE 0x1
  58 #define SXS_GUID_INFORMATION_CLR_FLAG_IS_CLASS     0x2
  59
  60 #define SXS_LOOKUP_CLR_GUID_USE_ACTCTX     0x00000001
  61 #define SXS_LOOKUP_CLR_GUID_FIND_SURROGATE 0x00010000
  62 #define SXS_LOOKUP_CLR_GUID_FIND_CLR_CLASS 0x00020000
  63 #define SXS_LOOKUP_CLR_GUID_FIND_ANY       (SXS_LOOKUP_CLR_GUID_FIND_SURROGATE | SXS_LOOKUP_CLR_GUID_FIND_CLR_CLASS)
  64
  65 struct comclassredirect_data
  66 {
  67     ULONG size;
  68     ULONG flags;
  69     DWORD model;
  70     GUID  clsid;
  71     GUID  alias;
  72     GUID  clsid2;
  73     GUID  tlbid;
  74     ULONG name_len;
  75     ULONG name_offset;
  76     ULONG progid_len;
  77     ULONG progid_offset;
  78     ULONG clrdata_len;
  79     ULONG clrdata_offset;
  80     DWORD miscstatus;
  81     DWORD miscstatuscontent;
  82     DWORD miscstatusthumbnail;
  83     DWORD miscstatusicon;
  84     DWORD miscstatusdocprint;
  85 };
  86
  87 struct clrclass_data
  88 {
  89     ULONG size;
  90     DWORD res[2];
  91     ULONG module_len;
  92     ULONG module_offset;
  93     ULONG name_len;
  94     ULONG name_offset;
  95     ULONG version_len;
  96     ULONG version_offset;
  97     DWORD res2[2];
  98 };
  99
 100 struct clrsurrogate_data
 101 {
 102     ULONG size;
 103     DWORD res;
 104     GUID  clsid;
 105     ULONG version_offset;
 106     ULONG version_len;
 107     ULONG name_offset;
 108     ULONG name_len;
 109 };
 110
 111 BOOL WINAPI SxsLookupClrGuid(DWORD flags, GUID *clsid, HANDLE actctx, void *buffer, SIZE_T buffer_len,
 112                              SIZE_T *buffer_len_required)
 113 {
 114     ACTCTX_SECTION_KEYED_DATA guid_info = { sizeof(ACTCTX_SECTION_KEYED_DATA) };
 115     ACTIVATION_CONTEXT_ASSEMBLY_DETAILED_INFORMATION *assembly_info = NULL;
 116     SIZE_T bytes_assembly_info;
 117     unsigned int len_version = 0, len_name, len_identity;
 118     const void *ptr_name, *ptr_version, *ptr_identity;
 119     SXS_GUID_INFORMATION_CLR *ret = buffer;
 120     BOOL retval = FALSE;
 121     char *ret_strings;
 122     ULONG_PTR cookie;
 123
 124     TRACE("%#x, %s, %p, %p, %lx, %p.\n", flags, wine_dbgstr_guid(clsid), actctx,
 125           buffer, buffer_len, buffer_len_required);
 126
 127     if (flags & SXS_LOOKUP_CLR_GUID_USE_ACTCTX)
 128     {
 129         if (!ActivateActCtx(actctx, &cookie))
 130         {
 131             WARN("Failed to activate context.\n");
 132             return FALSE;
 133         }
 134     }
 135
 136     if (flags & SXS_LOOKUP_CLR_GUID_FIND_SURROGATE)
 137     {
 138         if ((retval = FindActCtxSectionGuid(FIND_ACTCTX_SECTION_KEY_RETURN_HACTCTX, NULL,
 139                 ACTIVATION_CONTEXT_SECTION_CLR_SURROGATES, clsid, &guid_info)))
 140         {
 141             flags &= ~SXS_LOOKUP_CLR_GUID_FIND_CLR_CLASS;
 142         }
 143     }
 144
 145     if (!retval && (flags & SXS_LOOKUP_CLR_GUID_FIND_CLR_CLASS))
 146     {
 147         if ((retval = FindActCtxSectionGuid(FIND_ACTCTX_SECTION_KEY_RETURN_HACTCTX, NULL,
 148                 ACTIVATION_CONTEXT_SECTION_COM_SERVER_REDIRECTION, clsid, &guid_info)))
 149         {
 150             flags &= ~SXS_LOOKUP_CLR_GUID_FIND_SURROGATE;
 151         }
 152     }
 153
 154     if (!retval)
 155     {
 156         SetLastError(ERROR_NOT_FOUND);
 157         goto out;
 158     }
 159
 160     retval = QueryActCtxW(0, guid_info.hActCtx, &guid_info.ulAssemblyRosterIndex,
 161             AssemblyDetailedInformationInActivationContext, NULL, 0, &bytes_assembly_info);
 162     if (!retval && GetLastError() != ERROR_INSUFFICIENT_BUFFER)
 163     {
 164         goto out;
 165     }
 166
 167     assembly_info = heap_alloc(bytes_assembly_info);
 168     if (!(retval = QueryActCtxW(0, guid_info.hActCtx, &guid_info.ulAssemblyRosterIndex,
 169             AssemblyDetailedInformationInActivationContext, assembly_info,
 170             bytes_assembly_info, &bytes_assembly_info)))
 171     {
 172         goto out;
 173     }
 174
 175     if (flags & SXS_LOOKUP_CLR_GUID_FIND_CLR_CLASS)
 176     {
 177         const struct comclassredirect_data *redirect_data = guid_info.lpData;
 178         const struct clrclass_data *class_data;
 179
 180         class_data = (void *)((char *)redirect_data + redirect_data->clrdata_offset);
 181         ptr_name = (char *)class_data + class_data->name_offset;
 182         ptr_version = (char *)class_data + class_data->version_offset;
 183         len_name = class_data->name_len + sizeof(WCHAR);
 184         if (class_data->version_len)
 185             len_version = class_data->version_len + sizeof(WCHAR);
 186     }
 187     else
 188     {
 189         const struct clrsurrogate_data *surrogate = guid_info.lpData;
 190         ptr_name = (char *)surrogate + surrogate->name_offset;
 191         ptr_version = (char *)surrogate + surrogate->version_offset;
 192         len_name = surrogate->name_len + sizeof(WCHAR);
 193         if (surrogate->version_len)
 194             len_version = surrogate->version_len + sizeof(WCHAR);
 195     }
 196
 197     ptr_identity = assembly_info->lpAssemblyEncodedAssemblyIdentity;
 198     len_identity = assembly_info->ulEncodedAssemblyIdentityLength + sizeof(WCHAR);
 199
 200     *buffer_len_required = sizeof(*ret) + len_identity + len_version + len_name;
 201     if (!buffer || buffer_len < *buffer_len_required)
 202     {
 203         SetLastError(ERROR_INSUFFICIENT_BUFFER);
 204         retval = FALSE;
 205         goto out;
 206     }
 207
 208     ret->cbSize = sizeof(*ret);
 209     ret->dwFlags = flags & SXS_LOOKUP_CLR_GUID_FIND_CLR_CLASS ? SXS_GUID_INFORMATION_CLR_FLAG_IS_CLASS :
 210             SXS_GUID_INFORMATION_CLR_FLAG_IS_SURROGATE;
 211
 212     /* Copy strings into buffer */
 213     ret_strings = (char *)ret + sizeof(*ret);
 214
 215     memcpy(ret_strings, ptr_identity, len_identity);
 216     ret->pcwszAssemblyIdentity = (WCHAR *)ret_strings;
 217     ret_strings += len_identity;
 218
 219     memcpy(ret_strings, ptr_name, len_name);
 220     ret->pcwszTypeName = (WCHAR *)ret_strings;
 221     ret_strings += len_name;
 222
 223     if (len_version > 0)
 224     {
 225         memcpy(ret_strings, ptr_version, len_version);
 226         ret->pcwszRuntimeVersion = (WCHAR *)ret_strings;
 227     }
 228     else
 229         ret->pcwszRuntimeVersion = NULL;
 230
 231     SetLastError(0);
 232
 233 out:
 234     ReleaseActCtx(guid_info.hActCtx);
 235
 236     if (flags & SXS_LOOKUP_CLR_GUID_USE_ACTCTX)
 237         DeactivateActCtx(0, cookie);
 238
 239     heap_free(assembly_info);
 240     return retval;
 241 }