83 lines
1.6 KiB
C
83 lines
1.6 KiB
C
|
#include <windows.h>
|
||
|
|
|
||
|
|
#include <assert.h>
|
||
|
|
#include <stdbool.h>
|
||
|
|
#include <stdlib.h>
|
||
|
|
|
||
|
|
#include "hook/com-proxy.h"
|
||
|
|
#include "hook/table.h"
|
||
|
|
|
||
|
|
#include "hooklib/config.h"
|
||
|
|
#include "hooklib/dll.h"
|
||
|
|
#include "hooklib/dvd.h"
|
||
|
|
|
||
|
|
#include "util/dprintf.h"
|
||
|
|
|
||
|
|
/* API hooks */
|
||
|
|
|
||
|
|
static DWORD WINAPI hook_QueryDosDeviceW(
|
||
|
|
const wchar_t *lpDeviceName,
|
||
|
|
wchar_t *lpTargetPath,
|
||
|
|
DWORD ucchMax);
|
||
|
|
|
||
|
|
/* Link pointers */
|
||
|
|
|
||
|
|
static DWORD (WINAPI *next_QueryDosDeviceW)(
|
||
|
|
const wchar_t *lpDeviceName,
|
||
|
|
wchar_t *lpTargetPath,
|
||
|
|
DWORD ucchMax);
|
||
|
|
|
||
|
|
static bool dvd_hook_initted;
|
||
|
|
static struct dvd_config dvd_config;
|
||
|
|
|
||
|
|
static const struct hook_symbol dvd_hooks[] = {
|
||
|
|
{
|
||
|
|
.name = "QueryDosDeviceW",
|
||
|
|
.patch = hook_QueryDosDeviceW,
|
||
|
|
.link = (void **) &next_QueryDosDeviceW
|
||
|
|
},
|
||
|
|
};
|
||
|
|
|
||
|
|
void dvd_hook_init(const struct dvd_config *cfg, HINSTANCE self)
|
||
|
|
{
|
||
|
|
assert(cfg != NULL);
|
||
|
|
|
||
|
|
if (!cfg->enable) {
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (dvd_hook_initted) {
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
|
||
|
|
dvd_hook_initted = true;
|
||
|
|
|
||
|
|
memcpy(&dvd_config, cfg, sizeof(*cfg));
|
||
|
|
hook_table_apply(NULL, "kernel32.dll", dvd_hooks, _countof(dvd_hooks));
|
||
|
|
dprintf("DVD: hook enabled.\n");
|
||
|
|
}
|
||
|
|
|
||
|
|
DWORD WINAPI hook_QueryDosDeviceW(
|
||
|
|
const wchar_t *lpDeviceName,
|
||
|
|
wchar_t *lpTargetPath,
|
||
|
|
DWORD ucchMax)
|
||
|
|
{
|
||
|
|
DWORD ok;
|
||
|
|
wchar_t *p_dest;
|
||
|
|
wchar_t *dvd_string = L"CdRom";
|
||
|
|
|
||
|
|
ok = next_QueryDosDeviceW(
|
||
|
|
lpDeviceName,
|
||
|
|
lpTargetPath,
|
||
|
|
ucchMax);
|
||
|
|
|
||
|
|
p_dest = wcsstr (lpTargetPath, dvd_string);
|
||
|
|
|
||
|
|
if ( p_dest != NULL ) {
|
||
|
|
dprintf("DVD: Hiding DVD drive.\n");
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
return ok;
|
||
|
|
}
|