Merge pull request #461 from negativeExponent/libchdr

[pcsx_rearmed.git] / deps / lzma-16.04 / C / DllSecur.c

/* DllSecur.c -- DLL loading security\r
2016-10-04 : Igor Pavlov : Public domain */\r
\r
#include "Precomp.h"\r
\r
#ifdef _WIN32\r
\r
#include <windows.h>\r
\r
#include "DllSecur.h"\r
\r
#ifndef UNDER_CE\r
\r
typedef BOOL (WINAPI *Func_SetDefaultDllDirectories)(DWORD DirectoryFlags);\r
\r
#define MY_LOAD_LIBRARY_SEARCH_USER_DIRS 0x400\r
#define MY_LOAD_LIBRARY_SEARCH_SYSTEM32  0x800\r
\r
static const char * const g_Dlls =\r
  #ifndef _CONSOLE\r
  "UXTHEME\0"\r
  #endif\r
  "USERENV\0"\r
  "SETUPAPI\0"\r
  "APPHELP\0"\r
  "PROPSYS\0"\r
  "DWMAPI\0"\r
  "CRYPTBASE\0"\r
  "OLEACC\0"\r
  "CLBCATQ\0"\r
  ;\r
\r
#endif\r
\r
void LoadSecurityDlls()\r
{\r
  #ifndef UNDER_CE\r
  \r
  wchar_t buf[MAX_PATH + 100];\r
\r
  {\r
    // at Vista (ver 6.0) : CoCreateInstance(CLSID_ShellLink, ...) doesn't work after SetDefaultDllDirectories() : Check it ???\r
    OSVERSIONINFO vi;\r
    vi.dwOSVersionInfoSize = sizeof(vi);\r
    if (!GetVersionEx(&vi) || vi.dwMajorVersion != 6 || vi.dwMinorVersion != 0)\r
    {\r
      Func_SetDefaultDllDirectories setDllDirs = (Func_SetDefaultDllDirectories)\r
          GetProcAddress(GetModuleHandle(TEXT("kernel32.dll")), "SetDefaultDllDirectories");\r
      if (setDllDirs)\r
        if (setDllDirs(MY_LOAD_LIBRARY_SEARCH_SYSTEM32 | MY_LOAD_LIBRARY_SEARCH_USER_DIRS))\r
          return;\r
    }\r
  }\r
\r
  {\r
    unsigned len = GetSystemDirectoryW(buf, MAX_PATH + 2);\r
    if (len == 0 || len > MAX_PATH)\r
      return;\r
  }\r
  {\r
    const char *dll;\r
    unsigned pos = (unsigned)lstrlenW(buf);\r
\r
    if (buf[pos - 1] != '\\')\r
      buf[pos++] = '\\';\r
    \r
    for (dll = g_Dlls; dll[0] != 0;)\r
    {\r
      unsigned k = 0;\r
      for (;;)\r
      {\r
        char c = *dll++;\r
        buf[pos + k] = c;\r
        k++;\r
        if (c == 0)\r
          break;\r
      }\r
\r
      lstrcatW(buf, L".dll");\r
      LoadLibraryExW(buf, NULL, LOAD_WITH_ALTERED_SEARCH_PATH);\r
    }\r
  }\r
  \r
  #endif\r
}\r
\r
#endif\r