uzix/KernelEx
1
0
Fork 0
mirror of https://github.com/UzixLS/KernelEx.git synced 2025-07-19 07:21:20 +03:00
Code Issues Projects Releases Wiki Activity

import KernelEx-4.5-RC5

Browse Source
This commit is contained in:
UzixLS
2018-11-03 16:22:49 +03:00
parent e8dd043a25
commit 309977e788
76 changed files with 2275 additions and 544 deletions
Download Patch File Download Diff File Expand all files Collapse all files

101
apilibs/kexbasen/shell32/SHGetFolderPath.c
View File

@ -1,6 +1,6 @@
/*
* KernelEx
* Copyright (C) 2009, Xeno86
* Copyright (C) 2009-2010, Xeno86, Tihiy
*
* This file is part of KernelEx source code.
*
@ -24,15 +24,79 @@
#include "kexcoresdk.h"
#include "folderfix.h"
#include <pshpack1.h>
typedef struct
{
BYTE jmp;
DWORD func;
} LONGJMP, *PLONGJMP;
#include <poppack.h>
typedef HRESULT (WINAPI *SHGetFolderPathA_t)(HWND, int, HANDLE, DWORD, LPSTR);
typedef HRESULT (WINAPI *SHGetFolderPathW_t)(HWND, int, HANDLE, DWORD, LPWSTR);
static PROC LoadShfolderProc(const char* proc)
static HMODULE hShfolder;
static HMODULE hShell32;
static SHGetFolderPathA_t SHGetFolderPathA_pfn = (SHGetFolderPathA_t)-1;
static SHGetFolderPathW_t SHGetFolderPathW_pfn = (SHGetFolderPathW_t)-1;
static BYTE prev_entry[sizeof(LONGJMP)];
static BOOL APIENTRY shfolder_entry(HINSTANCE inst, DWORD reason, BOOL load_static)
{
if (reason == DLL_PROCESS_DETACH)
{
DBGPRINTF(("kexbasen: shfolder detached\n"));
SHGetFolderPathA_pfn = (SHGetFolderPathA_t)-1;
SHGetFolderPathW_pfn = (SHGetFolderPathW_t)-1;
hShfolder = NULL;
}
return TRUE;
}
/*
* Protection against buggy apps which call FreeLibrary on
* a DLL handle too many times.
*/
static void protect_shfolder()
{
DWORD fold;
DWORD fnew;
DWORD entry_addr;
PLONGJMP ljmp;
DWORD dwShfolder = (DWORD) hShfolder;
IMAGE_DOS_HEADER* dosh = (IMAGE_DOS_HEADER*) hShfolder;
IMAGE_NT_HEADERS* nth = (IMAGE_NT_HEADERS*) (dwShfolder + dosh->e_lfanew);
entry_addr = dwShfolder + nth->OptionalHeader.AddressOfEntryPoint;
memcpy(prev_entry, (PVOID) entry_addr, sizeof(LONGJMP));
VirtualProtect((PVOID) entry_addr, sizeof(LONGJMP), PAGE_READWRITE, &fold);
ljmp = (PLONGJMP) entry_addr;
ljmp->jmp = 0xe9; //jmp near rel
ljmp->func = (DWORD) shfolder_entry - (entry_addr + sizeof(LONGJMP));
VirtualProtect((PVOID) entry_addr, sizeof(LONGJMP), fold, &fnew);
}
static void unprotect_shfolder()
{
DWORD fold;
DWORD fnew;
DWORD entry_addr;
DWORD dwShfolder = (DWORD) hShfolder;
IMAGE_DOS_HEADER* dosh = (IMAGE_DOS_HEADER*) hShfolder;
IMAGE_NT_HEADERS* nth = (IMAGE_NT_HEADERS*) (dwShfolder + dosh->e_lfanew);
entry_addr = dwShfolder + nth->OptionalHeader.AddressOfEntryPoint;
VirtualProtect((PVOID) entry_addr, sizeof(LONGJMP), PAGE_READWRITE, &fold);
memcpy((PVOID) entry_addr, prev_entry, sizeof(LONGJMP));
VirtualProtect((PVOID) entry_addr, sizeof(LONGJMP), fold, &fnew);
}
static PROC LoadShProc(LPCSTR proc)
{
static const char ShfolderFn[] = "SHFOLDER.DLL";
static const char Shell32Fn[] = "SHELL32.DLL";
static HMODULE hShfolder;
static HMODULE hShell32;
PROC ret = NULL;
DWORD lasterr = GetLastError();
@ -45,19 +109,32 @@ static PROC LoadShfolderProc(const char* proc)
if (!ret)
{
if (!hShfolder)
{
hShfolder = LoadLibrary(ShfolderFn);
if (hShfolder) ret = kexGetProcAddress(hShfolder, proc);
if (hShfolder)
protect_shfolder();
}
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)
void uninit_SHGetFolderPath()
{
if (!hShfolder) return;
unprotect_shfolder();
hShfolder = NULL;
}
/* MAKE_EXPORT SHGetFolderPathA_new=SHGetFolderPathA */
HRESULT WINAPI SHGetFolderPathA_new(
HWND hwndOwner, int nFolder, HANDLE hToken, DWORD dwFlags, LPSTR pszPath)
{
static SHGetFolderPathA_t SHGetFolderPathA_pfn = (SHGetFolderPathA_t)-1;
if (SHGetFolderPathA_pfn == (void*)-1)
SHGetFolderPathA_pfn = (SHGetFolderPathA_t) LoadShfolderProc("SHGetFolderPathA");
SHGetFolderPathA_pfn = (SHGetFolderPathA_t) LoadShProc("SHGetFolderPathA");
if (SHGetFolderPathA_pfn == NULL)
return E_NOTIMPL;
nFolder = folder_fix(nFolder);
@ -65,11 +142,11 @@ HRESULT WINAPI SHGetFolderPathA_new(HWND hwndOwner, int nFolder, HANDLE hToken,
}
/* MAKE_EXPORT SHGetFolderPathW_new=SHGetFolderPathW */
HRESULT WINAPI SHGetFolderPathW_new(HWND hwndOwner, int nFolder, HANDLE hToken, DWORD dwFlags, LPWSTR pszPath)
HRESULT WINAPI SHGetFolderPathW_new(
HWND hwndOwner, int nFolder, HANDLE hToken, DWORD dwFlags, LPWSTR pszPath)
{
static SHGetFolderPathW_t SHGetFolderPathW_pfn = (SHGetFolderPathW_t)-1;
if (SHGetFolderPathW_pfn == (void*)-1)
SHGetFolderPathW_pfn = (SHGetFolderPathW_t) LoadShfolderProc("SHGetFolderPathW");
SHGetFolderPathW_pfn = (SHGetFolderPathW_t) LoadShProc("SHGetFolderPathW");
if (SHGetFolderPathW_pfn == NULL)
return E_NOTIMPL;
nFolder = folder_fix(nFolder);