forked from Hay1tsme/segatools
71 lines
1.4 KiB
C
71 lines
1.4 KiB
C
#include <windows.h>
|
|
|
|
#include <assert.h>
|
|
#include <stddef.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
|
|
#include "hook/table.h"
|
|
|
|
#include "platform/pcbid.h"
|
|
|
|
#include "util/dprintf.h"
|
|
|
|
static BOOL WINAPI pcbid_GetComputerNameA(char *dest, uint32_t *len);
|
|
|
|
static struct pcbid_config pcbid_cfg;
|
|
|
|
static const struct hook_symbol pcbid_syms[] = {
|
|
{
|
|
.name = "GetComputerNameA",
|
|
.patch = pcbid_GetComputerNameA,
|
|
}
|
|
};
|
|
|
|
HRESULT pcbid_hook_init(const struct pcbid_config *cfg)
|
|
{
|
|
assert(cfg != NULL);
|
|
|
|
if (!cfg->enable) {
|
|
return S_FALSE;
|
|
}
|
|
|
|
if (wcslen(cfg->serial_no) != 15) {
|
|
dprintf("Pcbid: ERROR: Must be 15 chars! ex: ACAE01A99999999\n");
|
|
|
|
return E_INVALIDARG;
|
|
}
|
|
|
|
memcpy(&pcbid_cfg, cfg, sizeof(*cfg));
|
|
hook_table_apply(NULL, "kernel32.dll", pcbid_syms, _countof(pcbid_syms));
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
static BOOL WINAPI pcbid_GetComputerNameA(char *dest, uint32_t *len)
|
|
{
|
|
size_t required;
|
|
|
|
if (dest == NULL || len == NULL) {
|
|
SetLastError(ERROR_INVALID_PARAMETER);
|
|
|
|
return FALSE;
|
|
}
|
|
|
|
wcstombs_s(&required, NULL, 0, pcbid_cfg.serial_no, 0);
|
|
|
|
if (required > *len) {
|
|
SetLastError(ERROR_INSUFFICIENT_BUFFER);
|
|
|
|
return FALSE;
|
|
}
|
|
|
|
dprintf("Pcbid: Get PCB serial\n");
|
|
|
|
wcstombs_s(NULL, dest, *len, pcbid_cfg.serial_no, *len - 1);
|
|
SetLastError(ERROR_SUCCESS);
|
|
*len = required - 1;
|
|
|
|
return TRUE;
|
|
}
|