mirror of
https://github.com/UzixLS/KernelEx.git
synced 2025-07-19 07:21:20 +03:00
81 lines
2.8 KiB
C
81 lines
2.8 KiB
C
/*
|
|
* KernelEx
|
|
* Copyright (C) 2009, Xeno86
|
|
*
|
|
* This file is part of KernelEx source code.
|
|
*
|
|
* KernelEx is free software; you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published
|
|
* by the Free Software Foundation; version 2 of the License.
|
|
*
|
|
* KernelEx is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with GNU Make; see the file COPYING. If not, write to
|
|
* the Free Software Foundation, 675 Mass Ave, Cambridge, MA 02139, USA.
|
|
*
|
|
*/
|
|
|
|
#include <windows.h>
|
|
#include "kexcoresdk.h"
|
|
|
|
typedef HRESULT (WINAPI *SHGetFolderPathA_t)(HWND, int, HANDLE, DWORD, LPSTR);
|
|
typedef HRESULT (WINAPI *SHGetFolderPathW_t)(HWND, int, HANDLE, DWORD, LPWSTR);
|
|
|
|
static SHGetFolderPathA_t SHGetFolderPathA_pfn = (SHGetFolderPathA_t)-1;
|
|
static SHGetFolderPathW_t SHGetFolderPathW_pfn = (SHGetFolderPathW_t)-1;
|
|
|
|
BOOL init_shfolder(void)
|
|
{
|
|
DWORD lasterr = GetLastError();
|
|
HMODULE hShell32 = GetModuleHandle("SHELL32.DLL");
|
|
PROC pfn = kexGetProcAddress(hShell32, "SHGetFolderPathA");
|
|
if (pfn)
|
|
SHGetFolderPathA_pfn = (SHGetFolderPathA_t) pfn;
|
|
pfn = kexGetProcAddress(hShell32, "SHGetFolderPathW");
|
|
if (pfn)
|
|
SHGetFolderPathW_pfn = (SHGetFolderPathW_t) pfn;
|
|
SetLastError(lasterr);
|
|
return TRUE;
|
|
}
|
|
|
|
static PROC LoadShfolderProc(const char* proc)
|
|
{
|
|
static const char ShfolderFn[] = "SHFOLDER.DLL";
|
|
static HMODULE hShfolder;
|
|
PROC ret = NULL;
|
|
DWORD lasterr = GetLastError();
|
|
|
|
if (!hShfolder)
|
|
{
|
|
hShfolder = GetModuleHandle(ShfolderFn);
|
|
if (!hShfolder) hShfolder = LoadLibrary(ShfolderFn);
|
|
if (hShfolder) ret = kexGetProcAddress(hShfolder, proc);
|
|
}
|
|
SetLastError(lasterr);
|
|
return ret;
|
|
}
|
|
|
|
/* MAKE_EXPORT SHGetFolderPathA_new=SHGetFolderPathA */
|
|
HRESULT WINAPI SHGetFolderPathA_new(HWND hwndOwner, int nFolder, HANDLE hToken, DWORD dwFlags, LPSTR pszPath)
|
|
{
|
|
if (SHGetFolderPathA_pfn == (void*)-1)
|
|
SHGetFolderPathA_pfn = (SHGetFolderPathA_t) LoadShfolderProc("SHGetFolderPathA");
|
|
if (SHGetFolderPathA_pfn == NULL)
|
|
return E_NOTIMPL;
|
|
return SHGetFolderPathA_pfn(hwndOwner, nFolder, hToken, dwFlags, pszPath);
|
|
}
|
|
|
|
/* MAKE_EXPORT SHGetFolderPathW_new=SHGetFolderPathW */
|
|
HRESULT WINAPI SHGetFolderPathW_new(HWND hwndOwner, int nFolder, HANDLE hToken, DWORD dwFlags, LPWSTR pszPath)
|
|
{
|
|
if (SHGetFolderPathW_pfn == (void*)-1)
|
|
SHGetFolderPathW_pfn = (SHGetFolderPathW_t) LoadShfolderProc("SHGetFolderPathW");
|
|
if (SHGetFolderPathW_pfn == NULL)
|
|
return E_NOTIMPL;
|
|
return SHGetFolderPathW_pfn(hwndOwner, nFolder, hToken, dwFlags, pszPath);
|
|
}
|