extensions/clientprefs/extension.h
author David Anderson <dvander@alliedmods.net>
Mon Aug 10 22:16:58 2009 -0400 (2009-08-10)
changeset 2767 c1f37afdcbb9
parent 2373 01dd8a057823
child 2783 5f333027ef04
permissions -rw-r--r--
Fixed incorrect escaping in clientprefs SQLite queries (bug 3904, r+a12=pred).
     1 /**
     2  * vim: set ts=4 :
     3  * =============================================================================
     4  * SourceMod Client Preferences Extension
     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_EXTENSION_PROPER_H_
    33 #define _INCLUDE_SOURCEMOD_EXTENSION_PROPER_H_
    34 
    35 #define MAXCLIENTS 64
    36 
    37 #include <stdlib.h>
    38 #include <stdarg.h>
    39 #include "smsdk_ext.h"
    40 #include "sh_list.h"
    41 #include "cookie.h"
    42 #include "menus.h"
    43 #include "query.h"
    44 
    45 enum DbDriver
    46 {
    47 	Driver_MySQL,
    48 	Driver_SQLite
    49 };
    50 
    51 #define MAX_TRANSLATE_PARAMS		32
    52 
    53 class TQueryOp;
    54 
    55 /**
    56  * @brief Sample implementation of the SDK Extension.
    57  * Note: Uncomment one of the pre-defined virtual functions in order to use it.
    58  */
    59 class ClientPrefs : public SDKExtension
    60 {
    61 public:
    62 	/**
    63 	 * @brief This is called after the initial loading sequence has been processed.
    64 	 *
    65 	 * @param error		Error message buffer.
    66 	 * @param maxlength	Size of error message buffer.
    67 	 * @param late		Whether or not the module was loaded after map load.
    68 	 * @return			True to succeed loading, false to fail.
    69 	 */
    70 	virtual bool SDK_OnLoad(char *error, size_t maxlength, bool late);
    71 	
    72 	/**
    73 	 * @brief This is called right before the extension is unloaded.
    74 	 */
    75 	virtual void SDK_OnUnload();
    76 
    77 	/**
    78 	 * @brief This is called once all known extensions have been loaded.
    79 	 * Note: It is is a good idea to add natives here, if any are provided.
    80 	 */
    81 	virtual void SDK_OnAllLoaded();
    82 
    83 	virtual bool QueryInterfaceDrop(SMInterface *pInterface);
    84 
    85 	virtual void NotifyInterfaceDrop(SMInterface *pInterface);
    86 
    87 	void DatabaseConnect();
    88 
    89 	bool AddQueryToQueue(TQueryOp *query);
    90 	void ProcessQueryCache();
    91 
    92 	/**
    93 	 * @brief Called when the pause state is changed.
    94 	 */
    95 	//virtual void SDK_OnPauseChange(bool paused);
    96 
    97 	/**
    98 	 * @brief this is called when Core wants to know if your extension is working.
    99 	 *
   100 	 * @param error		Error message buffer.
   101 	 * @param maxlength	Size of error message buffer.
   102 	 * @return			True if working, false otherwise.
   103 	 */
   104 	//virtual bool QueryRunning(char *error, size_t maxlength);
   105 public:
   106 #if defined SMEXT_CONF_METAMOD
   107 	/**
   108 	 * @brief Called when Metamod is attached, before the extension version is called.
   109 	 *
   110 	 * @param error			Error buffer.
   111 	 * @param maxlength		Maximum size of error buffer.
   112 	 * @param late			Whether or not Metamod considers this a late load.
   113 	 * @return				True to succeed, false to fail.
   114 	 */
   115 	//virtual bool SDK_OnMetamodLoad(ISmmAPI *ismm, char *error, size_t maxlength, bool late);
   116 
   117 	/**
   118 	 * @brief Called when Metamod is detaching, after the extension version is called.
   119 	 * NOTE: By default this is blocked unless sent from SourceMod.
   120 	 *
   121 	 * @param error			Error buffer.
   122 	 * @param maxlength		Maximum size of error buffer.
   123 	 * @return				True to succeed, false to fail.
   124 	 */
   125 	//virtual bool SDK_OnMetamodUnload(char *error, size_t maxlength);
   126 
   127 	/**
   128 	 * @brief Called when Metamod's pause state is changing.
   129 	 * NOTE: By default this is blocked unless sent from SourceMod.
   130 	 *
   131 	 * @param paused		Pause state being set.
   132 	 * @param error			Error buffer.
   133 	 * @param maxlength		Maximum size of error buffer.
   134 	 * @return				True to succeed, false to fail.
   135 	 */
   136 	//virtual bool SDK_OnMetamodPauseChange(bool paused, char *error, size_t maxlength);
   137 #endif
   138 public:
   139 	IDBDriver *Driver;
   140 	IDatabase *Database;
   141 	bool databaseLoading;
   142 	IPhraseCollection *phrases;
   143 	const DatabaseInfo *DBInfo;
   144 
   145 	IMutex *cookieMutex;
   146 
   147 private:
   148 	SourceHook::List<TQueryOp *> cachedQueries;
   149 	IMutex *queryMutex;
   150 };
   151 
   152 class CookieTypeHandler : public IHandleTypeDispatch
   153 {
   154 public:
   155 	void OnHandleDestroy(HandleType_t type, void *object)
   156 	{
   157 		/* No delete needed since Cookies are persistent */
   158 	}
   159 };
   160 
   161 class CookieIteratorHandler : public IHandleTypeDispatch
   162 {
   163 public:
   164 	void OnHandleDestroy(HandleType_t type, void *object)
   165 	{
   166 		delete (SourceHook::List<Cookie *>::iterator *)object;
   167 	}
   168 };
   169 
   170 size_t UTIL_Format(char *buffer, size_t maxlength, const char *fmt, ...);
   171 
   172 extern sp_nativeinfo_t g_ClientPrefNatives[];
   173 
   174 extern ClientPrefs g_ClientPrefs;
   175 extern HandleType_t g_CookieType;
   176 extern CookieTypeHandler g_CookieTypeHandler;
   177 
   178 extern HandleType_t g_CookieIterator;
   179 extern CookieIteratorHandler g_CookieIteratorHandler;
   180 
   181 bool Translate(char *buffer, 
   182 				   size_t maxlength,
   183 				   const char *format,
   184 				   unsigned int numparams,
   185 				   size_t *pOutLength,
   186 				   ...);
   187 
   188 extern DbDriver g_DriverType;
   189 
   190 #endif // _INCLUDE_SOURCEMOD_EXTENSION_PROPER_H_
   191