dlls/kernelbase/main.c

   1 /*
   2  * Copyright 2016 Michael Müller
   3  * Copyright 2017 Andrey Gusev
   4  *
   5  * This library is free software; you can redistribute it and/or
   6  * modify it under the terms of the GNU Lesser General Public
   7  * License as published by the Free Software Foundation; either
   8  * version 2.1 of the License, or (at your option) any later version.
   9  *
  10  * This library is distributed in the hope that it will be useful,
  11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13  * Lesser General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU Lesser General Public
  16  * License along with this library; if not, write to the Free Software
  17  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  18  */
  19
  20 #define COBJMACROS
  21
  22 #include "ntstatus.h"
  23 #define WIN32_NO_STATUS
  24 #include "windows.h"
  25 #include "appmodel.h"
  26 #include "shlwapi.h"
  27 #include "perflib.h"
  28 #include "winternl.h"
  29
  30 #include "wine/debug.h"
  31 #include "kernelbase.h"
  32 #include "wine/heap.h"
  33
  34 WINE_DEFAULT_DEBUG_CHANNEL(kernelbase);
  35
  36
  37 BOOL is_wow64 = FALSE;
  38
  39 /***********************************************************************
  40  *           DllMain
  41  */
  42 BOOL WINAPI DllMain( HINSTANCE hinst, DWORD reason, LPVOID reserved )
  43 {
  44     if (reason == DLL_PROCESS_ATTACH)
  45     {
  46         DisableThreadLibraryCalls( hinst );
  47         IsWow64Process( GetCurrentProcess(), &is_wow64 );
  48         init_locale();
  49         init_startup_info( NtCurrentTeb()->Peb->ProcessParameters );
  50     }
  51     return TRUE;
  52 }
  53
  54
  55 /*************************************************************
  56  *            DllMainCRTStartup
  57  */
  58 BOOL WINAPI DllMainCRTStartup( HANDLE inst, DWORD reason, LPVOID reserved )
  59 {
  60     return DllMain( inst, reason, reserved );
  61 }
  62
  63
  64 /***********************************************************************
  65  *          AppPolicyGetProcessTerminationMethod (KERNELBASE.@)
  66  */
  67 LONG WINAPI AppPolicyGetProcessTerminationMethod(HANDLE token, AppPolicyProcessTerminationMethod *policy)
  68 {
  69     FIXME("%p, %p\n", token, policy);
  70
  71     if(policy)
  72         *policy = AppPolicyProcessTerminationMethod_ExitProcess;
  73
  74     return ERROR_SUCCESS;
  75 }
  76
  77 /***********************************************************************
  78  *          AppPolicyGetThreadInitializationType (KERNELBASE.@)
  79  */
  80 LONG WINAPI AppPolicyGetThreadInitializationType(HANDLE token, AppPolicyThreadInitializationType *policy)
  81 {
  82     FIXME("%p, %p\n", token, policy);
  83
  84     if(policy)
  85         *policy = AppPolicyThreadInitializationType_None;
  86
  87     return ERROR_SUCCESS;
  88 }
  89
  90 /***********************************************************************
  91  *          AppPolicyGetShowDeveloperDiagnostic (KERNELBASE.@)
  92  */
  93 LONG WINAPI AppPolicyGetShowDeveloperDiagnostic(HANDLE token, AppPolicyShowDeveloperDiagnostic *policy)
  94 {
  95     FIXME("%p, %p\n", token, policy);
  96
  97     if(policy)
  98         *policy = AppPolicyShowDeveloperDiagnostic_ShowUI;
  99
 100     return ERROR_SUCCESS;
 101 }
 102
 103 /***********************************************************************
 104  *          AppPolicyGetWindowingModel (KERNELBASE.@)
 105  */
 106 LONG WINAPI AppPolicyGetWindowingModel(HANDLE token, AppPolicyWindowingModel *policy)
 107 {
 108     FIXME("%p, %p\n", token, policy);
 109
 110     if(policy)
 111         *policy = AppPolicyWindowingModel_ClassicDesktop;
 112
 113     return ERROR_SUCCESS;
 114 }
 115
 116 /***********************************************************************
 117  *           PerfCreateInstance   (KERNELBASE.@)
 118  */
 119 PPERF_COUNTERSET_INSTANCE WINAPI PerfCreateInstance(HANDLE handle, LPCGUID guid,
 120                                                     const WCHAR *name, ULONG id)
 121 {
 122     FIXME("%p %s %s %u: stub\n", handle, debugstr_guid(guid), debugstr_w(name), id);
 123     return NULL;
 124 }
 125
 126 /***********************************************************************
 127  *           PerfDeleteInstance   (KERNELBASE.@)
 128  */
 129 ULONG WINAPI PerfDeleteInstance(HANDLE provider, PPERF_COUNTERSET_INSTANCE block)
 130 {
 131     FIXME("%p %p: stub\n", provider, block);
 132     return ERROR_CALL_NOT_IMPLEMENTED;
 133 }
 134
 135 /***********************************************************************
 136  *           PerfSetCounterSetInfo   (KERNELBASE.@)
 137  */
 138 ULONG WINAPI PerfSetCounterSetInfo(HANDLE handle, PPERF_COUNTERSET_INFO template, ULONG size)
 139 {
 140     FIXME("%p %p %u: stub\n", handle, template, size);
 141     return ERROR_CALL_NOT_IMPLEMENTED;
 142 }
 143
 144 /***********************************************************************
 145  *           PerfSetCounterRefValue   (KERNELBASE.@)
 146  */
 147 ULONG WINAPI PerfSetCounterRefValue(HANDLE provider, PPERF_COUNTERSET_INSTANCE instance,
 148                                     ULONG counterid, void *address)
 149 {
 150     FIXME("%p %p %u %p: stub\n", provider, instance, counterid, address);
 151     return ERROR_CALL_NOT_IMPLEMENTED;
 152 }
 153
 154 /***********************************************************************
 155  *           PerfStartProvider   (KERNELBASE.@)
 156  */
 157 ULONG WINAPI PerfStartProvider(GUID *guid, PERFLIBREQUEST callback, HANDLE *provider)
 158 {
 159     FIXME("%s %p %p: stub\n", debugstr_guid(guid), callback, provider);
 160     return ERROR_CALL_NOT_IMPLEMENTED;
 161 }
 162
 163 /***********************************************************************
 164  *           PerfStartProviderEx   (KERNELBASE.@)
 165  */
 166 ULONG WINAPI PerfStartProviderEx(GUID *guid, PPERF_PROVIDER_CONTEXT context, HANDLE *provider)
 167 {
 168     FIXME("%s %p %p: stub\n", debugstr_guid(guid), context, provider);
 169     return ERROR_CALL_NOT_IMPLEMENTED;
 170 }
 171
 172 /***********************************************************************
 173  *           PerfStopProvider   (KERNELBASE.@)
 174  */
 175 ULONG WINAPI PerfStopProvider(HANDLE handle)
 176 {
 177     FIXME("%p: stub\n", handle);
 178     return ERROR_CALL_NOT_IMPLEMENTED;
 179 }
 180
 181 /***********************************************************************
 182  *           QuirkIsEnabled   (KERNELBASE.@)
 183  */
 184 BOOL WINAPI QuirkIsEnabled(void *arg)
 185 {
 186     FIXME("(%p): stub\n", arg);
 187     return FALSE;
 188 }
 189
 190 /***********************************************************************
 191  *          QuirkIsEnabled3 (KERNELBASE.@)
 192  */
 193 BOOL WINAPI QuirkIsEnabled3(void *unk1, void *unk2)
 194 {
 195     static int once;
 196
 197     if (!once++)
 198         FIXME("(%p, %p) stub!\n", unk1, unk2);
 199
 200     return FALSE;
 201 }
 202
 203 /***********************************************************************
 204  *           WaitOnAddress   (KERNELBASE.@)
 205  */
 206 BOOL WINAPI WaitOnAddress(volatile void *addr, void *cmp, SIZE_T size, DWORD timeout)
 207 {
 208     LARGE_INTEGER to;
 209
 210     if (timeout != INFINITE)
 211     {
 212         to.QuadPart = -(LONGLONG)timeout * 10000;
 213         return set_ntstatus( RtlWaitOnAddress( (const void *)addr, cmp, size, &to ));
 214     }
 215     return set_ntstatus( RtlWaitOnAddress( (const void *)addr, cmp, size, NULL ));
 216 }
 217
 218 HRESULT WINAPI QISearch(void *base, const QITAB *table, REFIID riid, void **obj)
 219 {
 220     const QITAB *ptr;
 221     IUnknown *unk;
 222
 223     TRACE("%p, %p, %s, %p\n", base, table, debugstr_guid(riid), obj);
 224
 225     if (!obj)
 226         return E_POINTER;
 227
 228     for (ptr = table; ptr->piid; ++ptr)
 229     {
 230         TRACE("trying (offset %d) %s\n", ptr->dwOffset, debugstr_guid(ptr->piid));
 231         if (IsEqualIID(riid, ptr->piid))
 232         {
 233             unk = (IUnknown *)((BYTE *)base + ptr->dwOffset);
 234             TRACE("matched, returning (%p)\n", unk);
 235             *obj = unk;
 236             IUnknown_AddRef(unk);
 237             return S_OK;
 238         }
 239     }
 240
 241     if (IsEqualIID(riid, &IID_IUnknown))
 242     {
 243         unk = (IUnknown *)((BYTE *)base + table->dwOffset);
 244         TRACE("returning first for IUnknown (%p)\n", unk);
 245         *obj = unk;
 246         IUnknown_AddRef(unk);
 247         return S_OK;
 248     }
 249
 250     WARN("Not found %s.\n", debugstr_guid(riid));
 251     *obj = NULL;
 252     return E_NOINTERFACE;
 253 }
 254
 255 HRESULT WINAPI GetAcceptLanguagesA(LPSTR langbuf, DWORD *buflen)
 256 {
 257     DWORD buflenW, convlen;
 258     WCHAR *langbufW;
 259     HRESULT hr;
 260
 261     TRACE("%p, %p, *%p: %d\n", langbuf, buflen, buflen, buflen ? *buflen : -1);
 262
 263     if (!langbuf || !buflen || !*buflen)
 264         return E_FAIL;
 265
 266     buflenW = *buflen;
 267     langbufW = heap_alloc(sizeof(WCHAR) * buflenW);
 268     hr = GetAcceptLanguagesW(langbufW, &buflenW);
 269
 270     if (hr == S_OK)
 271     {
 272         convlen = WideCharToMultiByte(CP_ACP, 0, langbufW, -1, langbuf, *buflen, NULL, NULL);
 273         convlen--;  /* do not count the terminating 0 */
 274     }
 275     else  /* copy partial string anyway */
 276     {
 277         convlen = WideCharToMultiByte(CP_ACP, 0, langbufW, *buflen, langbuf, *buflen, NULL, NULL);
 278         if (convlen < *buflen)
 279         {
 280             langbuf[convlen] = 0;
 281             convlen--;  /* do not count the terminating 0 */
 282         }
 283         else
 284         {
 285             convlen = *buflen;
 286         }
 287     }
 288     *buflen = buflenW ? convlen : 0;
 289
 290     heap_free(langbufW);
 291     return hr;
 292 }
 293
 294 static HRESULT lcid_to_rfc1766(LCID lcid, WCHAR *rfc1766, INT len)
 295 {
 296     WCHAR buffer[6 /* MAX_RFC1766_NAME */];
 297     INT n = GetLocaleInfoW(lcid, LOCALE_SISO639LANGNAME, buffer, ARRAY_SIZE(buffer));
 298     INT i;
 299
 300     if (n)
 301     {
 302         i = PRIMARYLANGID(lcid);
 303         if ((((i == LANG_ENGLISH) || (i == LANG_CHINESE) || (i == LANG_ARABIC)) &&
 304             (SUBLANGID(lcid) == SUBLANG_DEFAULT)) ||
 305             (SUBLANGID(lcid) > SUBLANG_DEFAULT)) {
 306
 307             buffer[n - 1] = '-';
 308             i = GetLocaleInfoW(lcid, LOCALE_SISO3166CTRYNAME, buffer + n, ARRAY_SIZE(buffer) - n);
 309             if (!i)
 310                 buffer[n - 1] = '\0';
 311         }
 312         else
 313             i = 0;
 314
 315         LCMapStringW(LOCALE_USER_DEFAULT, LCMAP_LOWERCASE, buffer, n + i, rfc1766, len);
 316         return ((n + i) > len) ? E_INVALIDARG : S_OK;
 317     }
 318     return E_FAIL;
 319 }
 320
 321 HRESULT WINAPI GetAcceptLanguagesW(WCHAR *langbuf, DWORD *buflen)
 322 {
 323     static const WCHAR keyW[] = {
 324         'S','o','f','t','w','a','r','e','\\',
 325         'M','i','c','r','o','s','o','f','t','\\',
 326         'I','n','t','e','r','n','e','t',' ','E','x','p','l','o','r','e','r','\\',
 327         'I','n','t','e','r','n','a','t','i','o','n','a','l',0};
 328     static const WCHAR valueW[] = {'A','c','c','e','p','t','L','a','n','g','u','a','g','e',0};
 329     DWORD mystrlen, mytype;
 330     WCHAR *mystr;
 331     LCID mylcid;
 332     HKEY mykey;
 333     LONG lres;
 334     DWORD len;
 335
 336     TRACE("%p, %p, *%p: %d\n", langbuf, buflen, buflen, buflen ? *buflen : -1);
 337
 338     if (!langbuf || !buflen || !*buflen)
 339         return E_FAIL;
 340
 341     mystrlen = (*buflen > 20) ? *buflen : 20 ;
 342     len = mystrlen * sizeof(WCHAR);
 343     mystr = heap_alloc(len);
 344     mystr[0] = 0;
 345     RegOpenKeyExW(HKEY_CURRENT_USER, keyW, 0, KEY_QUERY_VALUE, &mykey);
 346     lres = RegQueryValueExW(mykey, valueW, 0, &mytype, (PBYTE)mystr, &len);
 347     RegCloseKey(mykey);
 348     len = lstrlenW(mystr);
 349
 350     if (!lres && (*buflen > len))
 351     {
 352         lstrcpyW(langbuf, mystr);
 353         *buflen = len;
 354         heap_free(mystr);
 355         return S_OK;
 356     }
 357
 358     /* Did not find a value in the registry or the user buffer is too small */
 359     mylcid = GetUserDefaultLCID();
 360     lcid_to_rfc1766(mylcid, mystr, mystrlen);
 361     len = lstrlenW(mystr);
 362
 363     memcpy(langbuf, mystr, min(*buflen, len + 1)*sizeof(WCHAR));
 364     heap_free(mystr);
 365
 366     if (*buflen > len)
 367     {
 368         *buflen = len;
 369         return S_OK;
 370     }
 371
 372     *buflen = 0;
 373     return E_NOT_SUFFICIENT_BUFFER;
 374 }