segatools2/platform/pcbid.c

60 lines
1.2 KiB
C
Raw Normal View History

#include <windows.h>
2019-08-20 22:44:57 +00:00
#include <assert.h>
#include <stddef.h>
#include <stdlib.h>
#include <string.h>
#include "hook/table.h"
2019-08-20 22:44:57 +00:00
#include "platform/pcbid.h"
#include "util/dprintf.h"
static BOOL WINAPI pcbid_GetComputerNameA(char *dest, uint32_t *len);
2019-08-20 22:44:57 +00:00
static struct pcbid_config pcbid_cfg;
static const struct hook_symbol pcbid_syms[] = {
{
.name = "GetComputerNameA",
.patch = pcbid_GetComputerNameA,
}
};
2019-08-20 22:44:57 +00:00
void pcbid_hook_init(const struct pcbid_config *cfg)
{
2019-08-20 22:44:57 +00:00
assert(cfg != NULL);
2019-08-20 22:44:57 +00:00
if (!cfg->enable) {
return;
}
2019-08-20 22:44:57 +00:00
memcpy(&pcbid_cfg, cfg, sizeof(*cfg));
hook_table_apply(NULL, "kernel32.dll", pcbid_syms, _countof(pcbid_syms));
}
static BOOL WINAPI pcbid_GetComputerNameA(char *dest, uint32_t *len)
{
2019-08-20 22:44:57 +00:00
size_t required;
if (dest == NULL || len == NULL) {
SetLastError(ERROR_INVALID_PARAMETER);
return FALSE;
}
2019-08-20 22:44:57 +00:00
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");
2019-08-20 22:44:57 +00:00
wcstombs_s(NULL, dest, *len, pcbid_cfg.serial_no, *len - 1);
return TRUE;
}