1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
|
#include "pch.h"
#include "engine/r2engine.h"
AUTOHOOK_INIT()
// this is called from when our connection is rejected, this is the only case we're hooking this for
// clang-format off
AUTOHOOK(COM_ExplainDisconnection, engine.dll + 0x1342F0,
void,, (bool a1, const char* fmt, ...))
// clang-format on
{
va_list va;
va_start(va, fmt);
char buf[4096];
vsnprintf_s(buf, 4096, fmt, va);
va_end(va);
// slightly hacky comparison, but patching the function that calls this for reject would be worse
if (!strncmp(fmt, "Connection rejected: ", 21))
{
// when COM_ExplainDisconnection is called from engine.dll + 19ff1c for connection rejected, it doesn't
// call Host_Disconnect, which properly shuts down listen server
// not doing this gets our client in a pretty weird state so we need to shut it down manually here
// don't call Cbuf_Execute because we don't need this called immediately
R2::Cbuf_AddText(R2::Cbuf_GetCurrentPlayer(), "disconnect", R2::cmd_source_t::kCommandSrcCode);
}
return COM_ExplainDisconnection(a1, buf);
}
ON_DLL_LOAD_CLIENT("engine.dll", RejectConnectionFixes, (CModule module))
{
AUTOHOOK_DISPATCH()
}
|
