reactos/dll/win32/propsys/propsys_main.c

   1 /*
   2  * propsys main
   3  *
   4  * Copyright 1997, 2002 Alexandre Julliard
   5  * Copyright 2008 James Hawkins
   6  *
   7  * This library is free software; you can redistribute it and/or
   8  * modify it under the terms of the GNU Lesser General Public
   9  * License as published by the Free Software Foundation; either
  10  * version 2.1 of the License, or (at your option) any later version.
  11  *
  12  * This library is distributed in the hope that it will be useful,
  13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  15  * Lesser General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU Lesser General Public
  18  * License along with this library; if not, write to the Free Software
  19  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  20  */
  21
  22 #include "propsys_private.h"
  23
  24 #include <rpcproxy.h>
  25
  26 static HINSTANCE propsys_hInstance;
  27
  28 BOOL WINAPI DllMain(HINSTANCE hinstDLL, DWORD fdwReason, LPVOID lpvReserved)
  29 {
  30     TRACE("(0x%p, %d, %p)\n", hinstDLL, fdwReason, lpvReserved);
  31
  32     switch (fdwReason)
  33     {
  34         case DLL_WINE_PREATTACH:
  35             return FALSE;    /* prefer native version */
  36         case DLL_PROCESS_ATTACH:
  37             propsys_hInstance = hinstDLL;
  38             DisableThreadLibraryCalls(hinstDLL);
  39             break;
  40     }
  41
  42     return TRUE;
  43 }
  44
  45 HRESULT WINAPI DllRegisterServer(void)
  46 {
  47     return __wine_register_resources( propsys_hInstance );
  48 }
  49
  50 HRESULT WINAPI DllUnregisterServer(void)
  51 {
  52     return __wine_unregister_resources( propsys_hInstance );
  53 }
  54
  55 static HRESULT WINAPI ClassFactory_QueryInterface(IClassFactory *iface, REFIID riid, void **ppv)
  56 {
  57     *ppv = NULL;
  58
  59     if(IsEqualGUID(&IID_IUnknown, riid)) {
  60         TRACE("(%p)->(IID_IUnknown %p)\n", iface, ppv);
  61         *ppv = iface;
  62     }else if(IsEqualGUID(&IID_IClassFactory, riid)) {
  63         TRACE("(%p)->(IID_IClassFactory %p)\n", iface, ppv);
  64         *ppv = iface;
  65     }
  66
  67     if(*ppv) {
  68         IUnknown_AddRef((IUnknown*)*ppv);
  69         return S_OK;
  70     }
  71
  72     FIXME("(%p)->(%s %p)\n", iface, debugstr_guid(riid), ppv);
  73     return E_NOINTERFACE;
  74 }
  75
  76 static ULONG WINAPI ClassFactory_AddRef(IClassFactory *iface)
  77 {
  78     TRACE("(%p)\n", iface);
  79     return 2;
  80 }
  81
  82 static ULONG WINAPI ClassFactory_Release(IClassFactory *iface)
  83 {
  84     TRACE("(%p)\n", iface);
  85     return 1;
  86 }
  87
  88 static HRESULT WINAPI ClassFactory_LockServer(IClassFactory *iface, BOOL fLock)
  89 {
  90     TRACE("(%p)->(%x)\n", iface, fLock);
  91
  92     return S_OK;
  93 }
  94
  95 static HRESULT WINAPI InMemoryPropertyStoreFactory_CreateInstance(IClassFactory *iface, IUnknown *outer,
  96         REFIID riid, void **ppv)
  97 {
  98     TRACE("(%p %s %p)\n", outer, debugstr_guid(riid), ppv);
  99
 100     return PropertyStore_CreateInstance(outer, riid, ppv);
 101 }
 102
 103 static const IClassFactoryVtbl InMemoryPropertyStoreFactoryVtbl = {
 104     ClassFactory_QueryInterface,
 105     ClassFactory_AddRef,
 106     ClassFactory_Release,
 107     InMemoryPropertyStoreFactory_CreateInstance,
 108     ClassFactory_LockServer
 109 };
 110
 111 static IClassFactory InMemoryPropertyStoreFactory = { &InMemoryPropertyStoreFactoryVtbl };
 112
 113 HRESULT WINAPI DllGetClassObject(REFCLSID rclsid, REFIID riid, LPVOID *ppv)
 114 {
 115     if(IsEqualGUID(&CLSID_InMemoryPropertyStore, rclsid)) {
 116         TRACE("(CLSID_InMemoryPropertyStore %s %p)\n", debugstr_guid(riid), ppv);
 117         return IClassFactory_QueryInterface(&InMemoryPropertyStoreFactory, riid, ppv);
 118     }
 119
 120     FIXME("%s %s %p\n", debugstr_guid(rclsid), debugstr_guid(riid), ppv);
 121     return CLASS_E_CLASSNOTAVAILABLE;
 122 }
 123
 124 HRESULT WINAPI DllCanUnloadNow(void)
 125 {
 126     return S_FALSE;
 127 }
 128
 129 HRESULT WINAPI PSRegisterPropertySchema(PCWSTR path)
 130 {
 131     FIXME("%s stub\n", debugstr_w(path));
 132
 133     return S_OK;
 134 }
 135
 136 HRESULT WINAPI PSUnregisterPropertySchema(PCWSTR path)
 137 {
 138     FIXME("%s stub\n", debugstr_w(path));
 139
 140     return E_NOTIMPL;
 141 }
 142
 143 HRESULT WINAPI PSGetPropertyDescription(REFPROPERTYKEY propkey, REFIID riid, void **ppv)
 144 {
 145     FIXME("%p, %p, %p\n", propkey, riid, ppv);
 146     return E_NOTIMPL;
 147 }
 148
 149 HRESULT WINAPI PSRefreshPropertySchema(void)
 150 {
 151     FIXME("\n");
 152     return S_OK;
 153 }
 154
 155 HRESULT WINAPI PSStringFromPropertyKey(REFPROPERTYKEY pkey, LPWSTR psz, UINT cch)
 156 {
 157     static const WCHAR guid_fmtW[] = {'{','%','0','8','X','-','%','0','4','X','-',
 158                                       '%','0','4','X','-','%','0','2','X','%','0','2','X','-',
 159                                       '%','0','2','X','%','0','2','X','%','0','2','X',
 160                                       '%','0','2','X','%','0','2','X','%','0','2','X','}',0};
 161     static const WCHAR pid_fmtW[] = {'%','u',0};
 162
 163     WCHAR pidW[PKEY_PIDSTR_MAX + 1];
 164     LPWSTR p = psz;
 165     int len;
 166
 167     TRACE("(%p, %p, %u)\n", pkey, psz, cch);
 168
 169     if (!psz)
 170         return E_POINTER;
 171
 172     /* GUIDSTRING_MAX accounts for null terminator, +1 for space character. */
 173     if (cch <= GUIDSTRING_MAX + 1)
 174         return HRESULT_FROM_WIN32(ERROR_INSUFFICIENT_BUFFER);
 175
 176     if (!pkey)
 177     {
 178         psz[0] = '\0';
 179         return HRESULT_FROM_WIN32(ERROR_INSUFFICIENT_BUFFER);
 180     }
 181
 182     sprintfW(psz, guid_fmtW, pkey->fmtid.Data1, pkey->fmtid.Data2,
 183              pkey->fmtid.Data3, pkey->fmtid.Data4[0], pkey->fmtid.Data4[1],
 184              pkey->fmtid.Data4[2], pkey->fmtid.Data4[3], pkey->fmtid.Data4[4],
 185              pkey->fmtid.Data4[5], pkey->fmtid.Data4[6], pkey->fmtid.Data4[7]);
 186
 187     /* Overwrite the null terminator with the space character. */
 188     p += GUIDSTRING_MAX - 1;
 189     *p++ = ' ';
 190     cch -= GUIDSTRING_MAX - 1 + 1;
 191
 192     len = sprintfW(pidW, pid_fmtW, pkey->pid);
 193
 194     if (cch >= len + 1)
 195     {
 196         strcpyW(p, pidW);
 197         return S_OK;
 198     }
 199     else
 200     {
 201         WCHAR *ptr = pidW + len - 1;
 202
 203         psz[0] = '\0';
 204         *p++ = '\0';
 205         cch--;
 206
 207         /* Replicate a quirk of the native implementation where the contents
 208          * of the property ID string are written backwards to the output
 209          * buffer, skipping the rightmost digit. */
 210         if (cch)
 211         {
 212             ptr--;
 213             while (cch--)
 214                 *p++ = *ptr--;
 215         }
 216
 217         return HRESULT_FROM_WIN32(ERROR_INSUFFICIENT_BUFFER);
 218     }
 219 }
 220
 221 static const BYTE hex2bin[] =
 222 {
 223     0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,        /* 0x00 */
 224     0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,        /* 0x10 */
 225     0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,        /* 0x20 */
 226     0,1,2,3,4,5,6,7,8,9,0,0,0,0,0,0,        /* 0x30 */
 227     0,10,11,12,13,14,15,0,0,0,0,0,0,0,0,0,  /* 0x40 */
 228     0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,        /* 0x50 */
 229     0,10,11,12,13,14,15                     /* 0x60 */
 230 };
 231
 232 static BOOL validate_indices(LPCWSTR s, int min, int max)
 233 {
 234     int i;
 235
 236     for (i = min; i <= max; i++)
 237     {
 238         if (!s[i])
 239             return FALSE;
 240
 241         if (i == 0)
 242         {
 243             if (s[i] != '{')
 244                 return FALSE;
 245         }
 246         else if (i == 9 || i == 14 || i == 19 || i == 24)
 247         {
 248             if (s[i] != '-')
 249                 return FALSE;
 250         }
 251         else if (i == 37)
 252         {
 253             if (s[i] != '}')
 254                 return FALSE;
 255         }
 256         else
 257         {
 258             if (s[i] > 'f' || (!hex2bin[s[i]] && s[i] != '0'))
 259                 return FALSE;
 260         }
 261     }
 262
 263     return TRUE;
 264 }
 265
 266 /* Adapted from CLSIDFromString helper in dlls/ole32/compobj.c and
 267  * UuidFromString in dlls/rpcrt4/rpcrt4_main.c. */
 268 static BOOL string_to_guid(LPCWSTR s, LPGUID id)
 269 {
 270     /* in form {XXXXXXXX-XXXX-XXXX-XXXX-XXXXXXXXXXXX} */
 271
 272     if (!validate_indices(s, 0, 8)) return FALSE;
 273     id->Data1 = (hex2bin[s[1]] << 28 | hex2bin[s[2]] << 24 | hex2bin[s[3]] << 20 | hex2bin[s[4]] << 16 |
 274                  hex2bin[s[5]] << 12 | hex2bin[s[6]] << 8  | hex2bin[s[7]] << 4  | hex2bin[s[8]]);
 275     if (!validate_indices(s, 9, 14)) return FALSE;
 276     id->Data2 = hex2bin[s[10]] << 12 | hex2bin[s[11]] << 8 | hex2bin[s[12]] << 4 | hex2bin[s[13]];
 277     if (!validate_indices(s, 15, 19)) return FALSE;
 278     id->Data3 = hex2bin[s[15]] << 12 | hex2bin[s[16]] << 8 | hex2bin[s[17]] << 4 | hex2bin[s[18]];
 279
 280     /* these are just sequential bytes */
 281
 282     if (!validate_indices(s, 20, 21)) return FALSE;
 283     id->Data4[0] = hex2bin[s[20]] << 4 | hex2bin[s[21]];
 284     if (!validate_indices(s, 22, 24)) return FALSE;
 285     id->Data4[1] = hex2bin[s[22]] << 4 | hex2bin[s[23]];
 286
 287     if (!validate_indices(s, 25, 26)) return FALSE;
 288     id->Data4[2] = hex2bin[s[25]] << 4 | hex2bin[s[26]];
 289     if (!validate_indices(s, 27, 28)) return FALSE;
 290     id->Data4[3] = hex2bin[s[27]] << 4 | hex2bin[s[28]];
 291     if (!validate_indices(s, 29, 30)) return FALSE;
 292     id->Data4[4] = hex2bin[s[29]] << 4 | hex2bin[s[30]];
 293     if (!validate_indices(s, 31, 32)) return FALSE;
 294     id->Data4[5] = hex2bin[s[31]] << 4 | hex2bin[s[32]];
 295     if (!validate_indices(s, 33, 34)) return FALSE;
 296     id->Data4[6] = hex2bin[s[33]] << 4 | hex2bin[s[34]];
 297     if (!validate_indices(s, 35, 37)) return FALSE;
 298     id->Data4[7] = hex2bin[s[35]] << 4 | hex2bin[s[36]];
 299
 300     return TRUE;
 301 }
 302
 303 HRESULT WINAPI PSPropertyKeyFromString(LPCWSTR pszString, PROPERTYKEY *pkey)
 304 {
 305     int has_minus = 0, has_comma = 0;
 306
 307     TRACE("(%s, %p)\n", debugstr_w(pszString), pkey);
 308
 309     if (!pszString || !pkey)
 310         return E_POINTER;
 311
 312     memset(pkey, 0, sizeof(PROPERTYKEY));
 313
 314     if (!string_to_guid(pszString, &pkey->fmtid))
 315         return E_INVALIDARG;
 316
 317     pszString += GUIDSTRING_MAX - 1;
 318
 319     if (!*pszString)
 320         return E_INVALIDARG;
 321
 322     /* Only the space seems to be recognized as whitespace. The comma is only
 323      * recognized once and processing terminates if another comma is found. */
 324     while (*pszString == ' ' || *pszString == ',')
 325     {
 326         if (*pszString == ',')
 327         {
 328             if (has_comma)
 329                 return S_OK;
 330             else
 331                 has_comma = 1;
 332         }
 333         pszString++;
 334     }
 335
 336     if (!*pszString)
 337         return E_INVALIDARG;
 338
 339     /* Only two minus signs are recognized if no comma is detected. The first
 340      * sign is ignored, and the second is interpreted. If a comma is detected
 341      * before the minus sign, then only one minus sign counts, and property ID
 342      * interpretation begins with the next character. */
 343     if (has_comma)
 344     {
 345         if (*pszString == '-')
 346         {
 347             has_minus = 1;
 348             pszString++;
 349         }
 350     }
 351     else
 352     {
 353         if (*pszString == '-')
 354             pszString++;
 355
 356         /* Skip any intermediate spaces after the first minus sign. */
 357         while (*pszString == ' ')
 358             pszString++;
 359
 360         if (*pszString == '-')
 361         {
 362             has_minus = 1;
 363             pszString++;
 364         }
 365
 366         /* Skip any remaining spaces after minus sign. */
 367         while (*pszString == ' ')
 368             pszString++;
 369     }
 370
 371     /* Overflow is not checked. */
 372     while (isdigitW(*pszString))
 373     {
 374         pkey->pid *= 10;
 375         pkey->pid += (*pszString - '0');
 376         pszString++;
 377     }
 378
 379     if (has_minus)
 380         pkey->pid = ~pkey->pid + 1;
 381
 382     return S_OK;
 383 }