summaryrefslogtreecommitdiff
path: root/replay/cl_sessionblockdownloader.h
diff options
context:
space:
mode:
Diffstat (limited to 'replay/cl_sessionblockdownloader.h')
-rw-r--r--replay/cl_sessionblockdownloader.h64
1 files changed, 64 insertions, 0 deletions
diff --git a/replay/cl_sessionblockdownloader.h b/replay/cl_sessionblockdownloader.h
new file mode 100644
index 0000000..354ef3f
--- /dev/null
+++ b/replay/cl_sessionblockdownloader.h
@@ -0,0 +1,64 @@
+//========= Copyright Valve Corporation, All rights reserved. ============//
+//
+//----------------------------------------------------------------------------------------
+
+#ifndef CL_SESSIONBLOCKDOWNLOADER_H
+#define CL_SESSIONBLOCKDOWNLOADER_H
+#ifdef _WIN32
+#pragma once
+#endif
+
+//----------------------------------------------------------------------------------------
+
+#include "utllinkedlist.h"
+#include "cl_downloader.h"
+#include "basethinker.h"
+#include "replay/replayhandle.h"
+
+//----------------------------------------------------------------------------------------
+
+class CClientRecordingSession;
+
+//----------------------------------------------------------------------------------------
+
+//
+// Downloads multiple session blocks concurrently and updates their status.
+//
+class CSessionBlockDownloader : public CBaseThinker,
+ public IDownloadHandler
+{
+public:
+ CSessionBlockDownloader();
+
+ void Shutdown();
+ void Think();
+
+ // If pSession is NULL, all remaining downloads will affected.
+ void AbortDownloadsAndCleanup( CClientRecordingSession *pSession );
+
+private:
+ bool AtMaxConcurrentDownloads() const;
+
+ //
+ // CBaseThinker
+ //
+ virtual float GetNextThinkTime() const;
+
+ //
+ // IDownloadHandler
+ //
+ virtual void OnConnecting( CHttpDownloader *pDownloader );
+ virtual void OnFetch( CHttpDownloader *pDownloader );
+ virtual void OnDownloadComplete( CHttpDownloader *pDownloader, const unsigned char *pData );
+
+ static int sm_nNumCurrentDownloads;
+
+ CUtlLinkedList< CHttpDownloader *, int > m_lstDownloaders;
+ int m_nMaxBlock;
+
+ friend class CErrorSystem;
+};
+
+//----------------------------------------------------------------------------------------
+
+#endif // CL_SESSIONBLOCKDOWNLOADER_H