Max/TeamTofuShop_segatools
1
0
Fork 0
mirror of https://gitea.tendokyu.moe/TeamTofuShop/segatools.git synced 2026-09-22 22:37:59 +03:00
Code Issues Packages Projects Releases Wiki Activity

reg: allow opening keys multiple times

Browse Source
This commit is contained in:
kyoubate-haruka
2026-05-29 12:06:27 +02:00
parent 268f252d3e
commit de82acf782
1 changed files with 22 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files
common/hooklib/reg.c
+22 -11
View File
@@ -12,12 +12,15 @@
#include "util/dprintf.h"
#include "util/str.h"
#define MAX_CONCURRENT_REG_HANDLES 4
struct reg_hook_key {
HKEY root;
const wchar_t *name;
const struct reg_hook_val *vals;
size_t nvals;
HKEY handle;
HKEY handles[MAX_CONCURRENT_REG_HANDLES];
size_t nhandles;
};
/* Helper functions */
@@ -344,7 +347,7 @@ static LRESULT reg_hook_propagate_hr(HRESULT hr)
static struct reg_hook_key *reg_hook_match_key_locked(HKEY handle)
{
struct reg_hook_key *key;
size_t i;
size_t i, j;
if (handle == NULL || handle == INVALID_HANDLE_VALUE) {
return NULL;
@@ -353,8 +356,10 @@ static struct reg_hook_key *reg_hook_match_key_locked(HKEY handle)
for (i = 0 ; i < reg_hook_nkeys ; i++) {
key = &reg_hook_keys[i];
if (key->handle == handle) {
return key;
for (j = 0; j < key->nhandles; j++) {
if (key->handles[j] == handle) {
return key;
}
}
}
@@ -414,9 +419,9 @@ static LSTATUS reg_hook_open_locked(
return ERROR_SUCCESS;
}
/* Assume only one handle will be open at a time */
/* Assume only MAX_CONCURRENT_REG_HANDLES handles will be open at a time */
if (key->handle != NULL) {
if (key->nhandles > MAX_CONCURRENT_REG_HANDLES) {
return ERROR_SHARING_VIOLATION;
}
@@ -433,7 +438,7 @@ static LSTATUS reg_hook_open_locked(
out);
if (err == ERROR_SUCCESS) {
key->handle = *out;
key->handles[key->nhandles++] = *out;
}
return err;
@@ -511,16 +516,22 @@ static LSTATUS WINAPI hook_RegCreateKeyExW(
static LSTATUS WINAPI hook_RegCloseKey(HKEY handle)
{
struct reg_hook_key *key;
size_t i;
size_t i, j, k;
EnterCriticalSection(&reg_hook_lock);
for (i = 0 ; i < reg_hook_nkeys ; i++) {
key = &reg_hook_keys[i];
if (key->handle == handle) {
//dprintf("Registry: Closed virtual key %S\n", key->name);
key->handle = NULL;
for (j = 0; j < key->nhandles; j++) {
if (key->handles[j] == handle) {
for (k = j; k < MAX_CONCURRENT_REG_HANDLES - 1; k++) {
key->handles[k] = key->handles[k + 1];
}
key->handles[MAX_CONCURRENT_REG_HANDLES - 1] = NULL;
key->nhandles--;
//dprintf("Registry: Closed virtual key %S\n", key->name);
}
}
}