summaryrefslogtreecommitdiff
path: root/replay/sv_replaycontext.h
blob: 672494b7333cb637da62df73be4cca583b89efcf (plain) (blame)
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
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
//========= Copyright Valve Corporation, All rights reserved. ============//
//
//=======================================================================================//

#ifndef SV_REPLAYCONTEXT_H
#define SV_REPLAYCONTEXT_H
#ifdef _WIN32
#pragma once
#endif

//----------------------------------------------------------------------------------------

#include "shared_replaycontext.h"
#include "replay/iserverreplaycontext.h"
#include "sv_recordingsessionmanager.h"
#include "sv_recordingsessionblockmanager.h"
#include "errorsystem.h"

//----------------------------------------------------------------------------------------

class CSessionRecorder;
class CBaseRecordingSessionBlock;
class IRecordingSessionManager;
class IThreadPool;
class CFileserverCleaner;

//----------------------------------------------------------------------------------------

class CServerReplayContext : public IServerReplayContext,
							 public IErrorReporter
{
public:
	LINK_TO_SHARED_REPLAYCONTEXT_IMP();

	CServerReplayContext();
	~CServerReplayContext();

	virtual bool			Init( CreateInterfaceFn fnFactory );
	virtual void			Shutdown();

	virtual void			Think();	// Called by engine

	virtual void			OnPublishFailed();
	void					DoSanityCheckNow();

	void					UpdateFileserverIPFromHostname( const char *pHostname );
	void					UpdateFileserverProxyIPFromHostname( const char *pHostname );

	//
	// IErrorReporter
	//
	virtual void			ReportErrorsToUser( wchar_t *pErrorText );

	//
	// IServerReplayContext
	//
	virtual void			FlagForConVarSanityCheck();
	virtual IGameEvent		*CreateReplaySessionInfoEvent();
	virtual	IReplaySessionRecorder	*GetSessionRecorder();
	virtual const char		*GetLocalFileServerPath() const;
	virtual void			CreateSessionOnClient( int nClientSlot );

	const char				*GetServerSubDirName() const;

	CSessionRecorder		*m_pSessionRecorder;
	CFileserverCleaner		*m_pFileserverCleaner;

	char					m_szFileserverIP[16];		// Fileserver's IP, cached any time "replay_fileserver_offload_hostname" is modified.
	char					m_szFileserverProxyIP[16];	// Proxy's IP, cached any time "replay_fileserver_offload_proxy_host" is modified.

private:
	void					CleanTmpDir();
	void					ConVarSanityThink();

	float					m_flConVarSanityCheckTime; 
	bool					m_bShouldAbortRecording;
};

//----------------------------------------------------------------------------------------

extern CServerReplayContext *g_pServerReplayContext;

//----------------------------------------------------------------------------------------

inline CServerRecordingSessionManager *SV_GetRecordingSessionManager()
{
	return static_cast< CServerRecordingSessionManager * >( g_pServerReplayContext->GetRecordingSessionManager() );
}

inline CServerRecordingSessionBlockManager *SV_GetRecordingSessionBlockManager()
{
	return static_cast< CServerRecordingSessionBlockManager * >( g_pServerReplayContext->GetRecordingSessionBlockManager() );
}

inline CSessionRecorder *SV_GetSessionRecorder()
{
	return g_pServerReplayContext->m_pSessionRecorder;
}

inline CFileserverCleaner *SV_GetFileserverCleaner()
{
	return g_pServerReplayContext->m_pFileserverCleaner;
}

inline const char *SV_GetBasePath()
{
	return g_pServerReplayContext->m_pShared->m_strBasePath;
}

inline IThreadPool *SV_GetThreadPool()
{
	return g_pServerReplayContext->m_pShared->m_pThreadPool;
}

inline char const *SV_GetFileserverIP()
{
	return g_pServerReplayContext->m_szFileserverIP;
}

inline char const *SV_GetFileserverProxyIP()
{
	return g_pServerReplayContext->m_szFileserverProxyIP;
}

CServerRecordingSession *SV_GetRecordingSessionInProgress();
const char	*SV_GetTmpDir();	// Get "replay/server/tmp/"
bool		SV_IsOffloadingEnabled();

class CJob;
bool		SV_RunJobToCompletion( CJob *pJob );	// NOTE: Adds to thread pool first

//----------------------------------------------------------------------------------------

#endif	// SV_REPLAYCONTEXT_H