aboutsummaryrefslogtreecommitdiff
path: root/NorthstarDedicatedTest/playlist.cpp
blob: 096f4f3d414db7edab3ca93a89142c04e6560aa0 (plain)
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
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
#include "pch.h"
#include "playlist.h"
#include "concommand.h"
#include "convar.h"
#include "gameutils.h"
#include "hookutils.h"
#include "dedicated.h"

typedef char(*Onclc_SetPlaylistVarOverrideType)(void* a1, void* a2);
Onclc_SetPlaylistVarOverrideType Onclc_SetPlaylistVarOverride;

ConVar* Cvar_ns_use_clc_SetPlaylistVarOverride;

void SetPlaylistCommand(const CCommand& args)
{
	if (args.ArgC() < 2)
		return;

	SetCurrentPlaylist(args.Arg(1));
}

void SetPlaylistVarOverrideCommand(const CCommand& args)
{
	if (args.ArgC() < 3)
		return;

	SetPlaylistVarOverride(args.Arg(1), args.Arg(2));
}

char Onclc_SetPlaylistVarOverrideHook(void* a1, void* a2)
{
	// the private_match playlist is the only situation where there should be any legitimate sending of this netmessage
	// todo: check mp_lobby here too
	if (!Cvar_ns_use_clc_SetPlaylistVarOverride->m_nValue || strcmp(GetCurrentPlaylistName(), "private_match"))
		return 1;

	return Onclc_SetPlaylistVarOverride(a1, a2);
}

void InitialisePlaylistHooks(HMODULE baseAddress)
{
	RegisterConCommand("setplaylist", SetPlaylistCommand, "Sets the current playlist", FCVAR_NONE);
	RegisterConCommand("setplaylistvaroverride", SetPlaylistVarOverrideCommand, "Sets a playlist var override", FCVAR_NONE);
	// note: clc_SetPlaylistVarOverride is pretty insecure, since it allows for entirely arbitrary playlist var overrides to be sent to the server
	// this is somewhat restricted on custom servers to prevent it being done outside of private matches, but ideally it should be disabled altogether, since the custom menus won't use it anyway
	// this should only really be accepted if you want vanilla client compatibility
	Cvar_ns_use_clc_SetPlaylistVarOverride = RegisterConVar("ns_use_clc_SetPlaylistVarOverride", "0", FCVAR_GAMEDLL, "Whether the server should accept clc_SetPlaylistVarOverride messages");

	HookEnabler hook;
	ENABLER_CREATEHOOK(hook, (char*)baseAddress + 0x222180, &Onclc_SetPlaylistVarOverrideHook, reinterpret_cast<LPVOID*>(&Onclc_SetPlaylistVarOverride));

	// patch to prevent clc_SetPlaylistVarOverride from being able to crash servers if we reach max overrides due to a call to Error (why is this possible respawn, wtf)
	// todo: add a warning for this
	{
		void* ptr = (char*)baseAddress + 0x18ED8D;
		TempReadWrite rw(ptr);
		*((char*)ptr) = 0xC3; // jmp => ret
	}

	// on dedicated servers, force SetPlaylistVarOverride to always set the server's override list, irregardless of whether server is running yet
	if (IsDedicated())
	{
		// SetPlaylistVarOverride
		void* ptr = (char*)baseAddress + 0x18ED17;
		TempReadWrite rw(ptr);

		*((char*)ptr) = 0x90; // jl => nop
		*((char*)ptr + 1) = 0x90;
		*((char*)ptr + 2) = 0x90;
		*((char*)ptr + 3) = 0x90;
		*((char*)ptr + 4) = 0x90;
		*((char*)ptr + 5) = 0x90;
	}
}