core/UserMessages.h
author David Anderson <dvander@alliedmods.net>
Sun Oct 14 21:31:39 2007 +0000 (2007-10-14)
changeset 1558 95a3d8bf618d
parent 1330 ab94316bb52b
child 1690 7cc4f79a0416
permissions -rw-r--r--
all pdbs are now packaged with windows
     1 /**
     2  * vim: set ts=4 :
     3  * =============================================================================
     4  * SourceMod
     5  * Copyright (C) 2004-2007 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_CUSERMESSAGES_H_
    33 #define _INCLUDE_SOURCEMOD_CUSERMESSAGES_H_
    34 
    35 #include "ShareSys.h"
    36 #include <IUserMessages.h>
    37 #include "sourcemm_api.h"
    38 #include "sm_trie.h"
    39 #include "CellRecipientFilter.h"
    40 
    41 using namespace SourceHook;
    42 using namespace SourceMod;
    43 
    44 #define INVALID_MESSAGE_ID -1
    45 
    46 struct ListenerInfo
    47 {
    48 	IUserMessageListener *Callback;
    49 	bool IsHooked;
    50 	bool KillMe;
    51 };
    52 
    53 typedef List<ListenerInfo *> MsgList;
    54 typedef List<ListenerInfo *>::iterator MsgIter;
    55 
    56 class UserMessages : 
    57 	public IUserMessages,
    58 	public SMGlobalClass
    59 {
    60 public:
    61 	UserMessages();
    62 	~UserMessages();
    63 public: //SMGlobalClass
    64 	void OnSourceModStartup(bool late);
    65 	void OnSourceModAllInitialized();
    66 	void OnSourceModAllShutdown();
    67 public: //IUserMessages
    68 	int GetMessageIndex(const char *msg);
    69 	bool GetMessageName(int msgid, char *buffer, size_t maxlength) const;
    70 	bool HookUserMessage(int msg_id, IUserMessageListener *pListener, bool intercept=false);
    71 	bool UnhookUserMessage(int msg_id, IUserMessageListener *pListener, bool intercept=false);
    72 	bf_write *StartMessage(int msg_id, cell_t players[], unsigned int playersNum, int flags);
    73 	bool EndMessage();
    74 public:
    75 	bf_write *OnStartMessage_Pre(IRecipientFilter *filter, int msg_type);
    76 	bf_write *OnStartMessage_Post(IRecipientFilter *filter, int msg_type);
    77 	void OnMessageEnd_Pre();
    78 	void OnMessageEnd_Post();
    79 private:
    80 	void _DecRefCounter();
    81 private:
    82 	List<ListenerInfo *> m_msgHooks[255];
    83 	List<ListenerInfo *> m_msgIntercepts[255];
    84 	CStack<ListenerInfo *> m_FreeListeners;
    85 	unsigned char m_pBase[2500];
    86 	IRecipientFilter *m_CurRecFilter;
    87 	bf_write m_InterceptBuffer;
    88 	bf_write *m_OrigBuffer;
    89 	bf_read m_ReadBuffer;
    90 	size_t m_HookCount;
    91 	bool m_InHook;
    92 	bool m_BlockEndPost;
    93 	bool m_FallbackSearch;
    94 
    95 	Trie *m_Names;
    96 	CellRecipientFilter m_CellRecFilter;
    97 	bool m_InExec;
    98 	int m_CurFlags;
    99 	int m_CurId;
   100 };
   101 
   102 extern UserMessages g_UserMsgs;
   103 
   104 #endif //_INCLUDE_SOURCEMOD_CUSERMESSAGES_H_