core/logic/GameConfigs.h
author David Anderson <dvander@alliedmods.net>
Sun Aug 25 11:59:45 2013 -0700 (2013-08-25)
changeset 4064 d7ca1e28949d
parent 4063 7b7372572e85
child 4114 58146f5766f0
permissions -rw-r--r--
Switch HandleSys off KTrie (bug 5884 part 2, r=ds).
     1 /**
     2  * vim: set ts=4 sw=4 tw=99 noet:
     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  */
    31 
    32 #ifndef _INCLUDE_SOURCEMOD_CGAMECONFIGS_H_
    33 #define _INCLUDE_SOURCEMOD_CGAMECONFIGS_H_
    34 
    35 #include "common_logic.h"
    36 #include <IGameConfigs.h>
    37 #include <ITextParsers.h>
    38 #include "sm_memtable.h"
    39 #include <am-refcounting.h>
    40 #include <sm_stringhashmap.h>
    41 #include <sm_namehashset.h>
    42 
    43 using namespace SourceMod;
    44 using namespace SourceHook;
    45 
    46 class SendProp;
    47 
    48 class CGameConfig : 
    49 	public ITextListener_SMC,
    50 	public IGameConfig,
    51 	public ke::Refcounted<CGameConfig>
    52 {
    53 	friend class GameConfigManager;
    54 public:
    55 	CGameConfig(const char *file, const char *engine = NULL);
    56 	~CGameConfig();
    57 public:
    58 	bool Reparse(char *error, size_t maxlength);
    59 	bool EnterFile(const char *file, char *error, size_t maxlength);
    60 	void SetBaseEngine(const char *engine);
    61 	void SetParseEngine(const char *engine);
    62 public: //ITextListener_SMC
    63 	SMCResult ReadSMC_NewSection(const SMCStates *states, const char *name);
    64 	SMCResult ReadSMC_KeyValue(const SMCStates *states, const char *key, const char *value);
    65 	SMCResult ReadSMC_LeavingSection(const SMCStates *states);
    66 public: //IGameConfig
    67 	const char *GetKeyValue(const char *key);
    68 	bool GetOffset(const char *key, int *value);
    69 	SendProp *GetSendProp(const char *key);
    70 	bool GetMemSig(const char *key, void **addr);
    71 	bool GetAddress(const char *key, void **addr);
    72 public: //NameHashSet
    73 	static inline bool matches(const char *key, const CGameConfig *value)
    74 	{
    75 		return strcmp(key, value->m_File) == 0;
    76 	}
    77 private:
    78 	ke::AutoPtr<BaseStringTable> m_pStrings;
    79 	char m_File[PLATFORM_MAX_PATH];
    80 	char m_CurFile[PLATFORM_MAX_PATH];
    81 	StringHashMap<int> m_Offsets;
    82 	StringHashMap<SendProp *> m_Props;
    83 	StringHashMap<int> m_Keys;
    84 	StringHashMap<void *> m_Sigs;
    85 	/* Parse states */
    86 	int m_ParseState;
    87 	unsigned int m_IgnoreLevel;
    88 	char m_Class[64];
    89 	char m_Prop[64];
    90 	char m_offset[64];
    91 	char m_Game[256];
    92 	bool bShouldBeReadingDefault;
    93 	bool had_game;
    94 	bool matched_game;
    95 	bool had_engine;
    96 	bool matched_engine;
    97 
    98 	/* Custom Sections */
    99 	unsigned int m_CustomLevel;
   100 	ITextListener_SMC *m_CustomHandler;
   101 
   102 	/* Support for reading Addresses */
   103 	struct AddressConf
   104 	{
   105 		char signatureName[64];
   106 		int readCount;
   107 		int read[8];
   108 
   109 		AddressConf(char *sigName, unsigned sigLength, unsigned readCount, int *read);
   110 
   111 		AddressConf() {}
   112 	};
   113 
   114 	char m_Address[64];
   115 	char m_AddressSignature[64];
   116 	int m_AddressReadCount;
   117 	int m_AddressRead[8];
   118 	StringHashMap<AddressConf> m_Addresses;
   119 	const char *m_pEngine;
   120 	const char *m_pBaseEngine;
   121 };
   122 
   123 class GameConfigManager : 
   124 	public IGameConfigManager,
   125 	public SMGlobalClass
   126 {
   127 public:
   128 	GameConfigManager();
   129 	~GameConfigManager();
   130 public: //IGameConfigManager
   131 	bool LoadGameConfigFile(const char *file, IGameConfig **pConfig, char *error, size_t maxlength);
   132 	void CloseGameConfigFile(IGameConfig *cfg);
   133 	IGameConfig *ReadHandle(Handle_t hndl,
   134 		IdentityToken_t *ident,
   135 		HandleError *err);
   136 	void AddUserConfigHook(const char *sectionname, ITextListener_SMC *listener);
   137 	void RemoveUserConfigHook(const char *sectionname, ITextListener_SMC *listener);
   138 	void AcquireLock();
   139 	void ReleaseLock();
   140 public: //SMGlobalClass
   141 	void OnSourceModStartup(bool late);
   142 	void OnSourceModAllInitialized();
   143 	void OnSourceModAllShutdown();
   144 public:
   145 	void RemoveCachedConfig(CGameConfig *config);
   146 private:
   147 	NameHashSet<CGameConfig *> m_Lookup;
   148 public:
   149 	StringHashMap<ITextListener_SMC *> m_customHandlers;
   150 };
   151 
   152 extern GameConfigManager g_GameConfigs;
   153 extern IGameConfig *g_pGameConf;
   154 
   155 #endif //_INCLUDE_SOURCEMOD_CGAMECONFIGS_H_