sourcemod-central: core/GameDataFetcher.h@450192701928

     1 /**

     2  * vim: set ts=4 :

     3  * =============================================================================

     4  * SourceMod

     5  * Copyright (C) 2004-2008 AlliedModders LLC.  All rights reserved.

     6  * =============================================================================

     7  *

     8  * This program is free software; you can redistribute it and/or modify it under

     9  * the terms of the GNU General Public License, version 3.0, as published by the

    10  * Free Software Foundation.

    11  *

    12  * This program is distributed in the hope that it will be useful, but WITHOUT

    13  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS

    14  * FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more

    15  * details.

    16  *

    17  * You should have received a copy of the GNU General Public License along with

    18  * this program.  If not, see <http://www.gnu.org/licenses/>.

    19  *

    20  * As a special exception, AlliedModders LLC gives you permission to link the

    21  * code of this program (as well as its derivative works) to "Half-Life 2," the

    22  * "Source Engine," the "SourcePawn JIT," and any Game MODs that run on software

    23  * by the Valve Corporation.  You must obey the GNU General Public License in

    24  * all respects for all other code used.  Additionally, AlliedModders LLC grants

    25  * this exception to all derivative works.  AlliedModders LLC defines further

    26  * exceptions, found in LICENSE.txt (as of this writing, version JULY-31-2007),

    27  * or <http://www.sourcemod.net/license.php>.

    28  *

    29  * Version: $Id$

    30  */

    32 #ifndef _INCLUDE_SOURCEMOD_GAMEDATAFETCHER_H_

    33 #define _INCLUDE_SOURCEMOD_GAMEDATAFETCHER_H_

    35 #include "sourcemod.h"

    36 #include "TextParsers.h"

    37 #include "IThreader.h"

    38 #include "Logger.h"

    39 #include "LibrarySys.h"

    40 #include "ThreadSupport.h"

    41 #include "sm_memtable.h"

    42 #include <sh_string.h>

    43 #include <sh_list.h>

    45 enum UpdateStatus

    46 {

    47 	Update_Unknown = 0,				/* Version wasn't recognised or version querying is unsupported */

    48 	Update_Current = 1,				/* Server is running latest version */

    49 	Update_NewBuild = 2,			/* Server is on a svn release and a newer version is available */

    50 	Update_MinorAvailable = 3,		/* Server is on a release and a minor release has superceeded it */

    51 	Update_MajorAvailable = 4,		/* Server is on a release and a major release has superceeded it */

    52 	Update_CriticalAvailable = 5,	/* A critical update has been released (security fixes etc) */

    53 };

    55 class MD5;

    57 class BuildMD5ableBuffer : public ITextListener_SMC

    58 {

    59 public:

    60 	BuildMD5ableBuffer();

    61 	~BuildMD5ableBuffer();

    62 	void ReadSMC_ParseStart();

    63 	SMCResult ReadSMC_KeyValue(const SMCStates *states, const char *key, const char *value);

    64 	SMCResult ReadSMC_NewSection(const SMCStates *states, const char *name);

    65 	void ReadSMC_ParseEnd(bool halted, bool failed);

    66 public:

    67 	MD5 *checksum;

    68 private:

    69 	BaseStringTable *stringTable;

    70 };

    72 struct FileData

    73 {

    74 	SourceHook::String *filename;

    75 	char checksum[33];

    76 };

    78 class FetcherThread : public IThread

    79 {

    80 public:

    81 	FetcherThread();

    82 	~FetcherThread();

    83 public:

    84 	void RunThread(IThreadHandle *pHandle);

    85 	void OnTerminate(IThreadHandle *pHandle, bool cancel);

    86 private:

    87 	int BuildGameDataQuery(char *buffer, int maxlen);

    88 	void ProcessGameDataQuery(int SocketDescriptor);

    89 	int RecvData(int socketDescriptor, char *buffer, int len);

    90 	int SendData(int socketDescriptor, char *buffer, int len);

    91 	int ConnectSocket();

    92 	void HandleUpdateStatus(UpdateStatus status, short version[4]);

    93 public:

    94 	SourceHook::CVector<FileData *> filenames;

    95 	bool needsRestart;

    96 	bool wasSuccess;

    97 private:

    98 	UpdateStatus updateStatus;

    99 	BaseMemTable *memtable;

   100 	short build[4];

   101 };

   103 extern bool g_blockGameDataLoad;

   105 #endif // _INCLUDE_SOURCEMOD_GAMEDATAFETCHER_H_

author	Fyren <fyrenmoo@gmail.com>
	Thu Feb 19 14:22:22 2009 +1300 (2009-02-19)
changeset 2589	450192701928
parent 2477	b51cbb84d3f0
permissions	-rw-r--r--