sourcemod-central: core/UserMessages.h@bf5f9aa53abe

     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_CUSERMESSAGES_H_

    33 #define _INCLUDE_SOURCEMOD_CUSERMESSAGES_H_

    35 #include "ShareSys.h"

    36 #include <IUserMessages.h>

    37 #include "sourcemm_api.h"

    38 #include "sm_trie.h"

    39 #include "CellRecipientFilter.h"

    41 using namespace SourceHook;

    42 using namespace SourceMod;

    44 #define INVALID_MESSAGE_ID -1

    46 struct ListenerInfo

    47 {

    48 	IUserMessageListener *Callback;

    49 	bool IsHooked;

    50 	bool KillMe;

    51 	bool IsNew;

    52 };

    54 typedef List<ListenerInfo *> MsgList;

    55 typedef List<ListenerInfo *>::iterator MsgIter;

    57 class UserMessages :

    58 	public IUserMessages,

    59 	public SMGlobalClass

    60 {

    61 public:

    62 	UserMessages();

    63 	~UserMessages();

    64 public: //SMGlobalClass

    65 	void OnSourceModStartup(bool late);

    66 	void OnSourceModAllInitialized();

    67 	void OnSourceModAllShutdown();

    68 public: //IUserMessages

    69 	int GetMessageIndex(const char *msg);

    70 	bool GetMessageName(int msgid, char *buffer, size_t maxlength) const;

    71 	bool HookUserMessage(int msg_id, IUserMessageListener *pListener, bool intercept=false);

    72 	bool UnhookUserMessage(int msg_id, IUserMessageListener *pListener, bool intercept=false);

    73 	bf_write *StartMessage(int msg_id, const cell_t players[], unsigned int playersNum, int flags);

    74 	bool EndMessage();

    75 	bool HookUserMessage2(int msg_id,

    76 		IUserMessageListener *pListener,

    77 		bool intercept=false);

    78 	bool UnhookUserMessage2(int msg_id,

    79 		IUserMessageListener *pListener,

    80 		bool intercept=false);

    81 public:

    82 #if (SOURCE_ENGINE == SE_LEFT4DEAD) || (SOURCE_ENGINE == SE_LEFT4DEAD2)

    83 	bf_write *OnStartMessage_Pre(IRecipientFilter *filter, int msg_type, const char *msg_name);

    84 	bf_write *OnStartMessage_Post(IRecipientFilter *filter, int msg_type, const char *msg_name);

    85 #else

    86 	bf_write *OnStartMessage_Pre(IRecipientFilter *filter, int msg_type);

    87 	bf_write *OnStartMessage_Post(IRecipientFilter *filter, int msg_type);

    88 #endif

    89 	void OnMessageEnd_Pre();

    90 	void OnMessageEnd_Post();

    91 private:

    92 	bool InternalHook(int msg_id, IUserMessageListener *pListener, bool intercept, bool isNew);

    93 	bool InternalUnhook(int msg_id, IUserMessageListener *pListener, bool intercept, bool isNew);

    94 	void _DecRefCounter();

    95 private:

    96 	List<ListenerInfo *> m_msgHooks[255];

    97 	List<ListenerInfo *> m_msgIntercepts[255];

    98 	CStack<ListenerInfo *> m_FreeListeners;

    99 	unsigned char m_pBase[2500];

   100 	IRecipientFilter *m_CurRecFilter;

   101 	bf_write m_InterceptBuffer;

   102 	bf_write *m_OrigBuffer;

   103 	bf_read m_ReadBuffer;

   104 	size_t m_HookCount;

   105 	bool m_InHook;

   106 	bool m_BlockEndPost;

   107 	bool m_FallbackSearch;

   109 	Trie *m_Names;

   110 	CellRecipientFilter m_CellRecFilter;

   111 	bool m_InExec;

   112 	int m_CurFlags;

   113 	int m_CurId;

   114 };

   116 extern UserMessages g_UserMsgs;

   118 #endif //_INCLUDE_SOURCEMOD_CUSERMESSAGES_H_

author	Nicholas Hastings <skamonkey@gmail.com>
	Mon Feb 08 19:57:20 2010 -0800 (2010-02-08)
changeset 2914	bf5f9aa53abe
parent 2847	1daf209d9dd2
child 3099	1b7bf8509625
permissions	-rw-r--r--