core/ShareSys.h
author Nicholas Hastings <skamonkey@gmail.com>
Mon Apr 11 22:30:03 2011 -0400 (2011-04-11)
changeset 3238 fff6e001bf9b
parent 2841 dde3dd0b4b8b
permissions -rw-r--r--
Removed gEntList failure notify on Windows. (bug 4832, r=fyren).
     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  */
    31 
    32 #ifndef _INCLUDE_SOURCEMOD_SHARESYSTEM_H_
    33 #define _INCLUDE_SOURCEMOD_SHARESYSTEM_H_
    34 
    35 #include <IShareSys.h>
    36 #include <IHandleSys.h>
    37 #include <sh_list.h>
    38 #include <sm_trie_tpl.h>
    39 #include "sm_globals.h"
    40 #include "sourcemod.h"
    41 
    42 using namespace SourceHook;
    43 
    44 namespace SourceMod
    45 {
    46 	struct IdentityToken_t
    47 	{
    48 		Handle_t ident;
    49 		void *ptr;
    50 		IdentityType_t type;
    51 	};
    52 };
    53 
    54 struct IfaceInfo
    55 {
    56 	bool operator ==(const IfaceInfo &info)
    57 	{
    58 		return (info.iface == iface && info.owner == owner);
    59 	}
    60 	SMInterface *iface;
    61 	IExtension *owner;
    62 };
    63 
    64 class CNativeOwner;
    65 class CPlugin;
    66 struct NativeEntry;
    67 
    68 struct ReplaceNative
    69 {
    70 	CNativeOwner *owner;
    71 	SPVM_NATIVE_FUNC func;
    72 };
    73 
    74 struct FakeNative
    75 {
    76 	char name[64];
    77 	IPluginContext *ctx;
    78 	IPluginFunction *call;
    79 };
    80 
    81 struct NativeEntry
    82 {
    83 	CNativeOwner *owner;
    84 	SPVM_NATIVE_FUNC func;
    85 	const char *name;
    86 	ReplaceNative replacement;
    87 	FakeNative *fake;
    88 };
    89 
    90 struct Capability
    91 {
    92 	IExtension *ext;
    93 	IFeatureProvider *provider;
    94 };
    95 
    96 class ShareSystem :
    97 	public IShareSys,
    98 	public SMGlobalClass,
    99 	public IHandleTypeDispatch
   100 {
   101 	friend class CNativeOwner;
   102 public:
   103 	ShareSystem();
   104 public: //IShareSys
   105 	bool AddInterface(IExtension *myself, SMInterface *pIface);
   106 	bool RequestInterface(const char *iface_name, 
   107 		unsigned int iface_vers,
   108 		IExtension *myself,
   109 		SMInterface **pIface);
   110 	void AddNatives(IExtension *myself, const sp_nativeinfo_t *natives);
   111 	IdentityType_t CreateIdentType(const char *name);
   112 	IdentityType_t FindIdentType(const char *name);
   113 	IdentityToken_t *CreateIdentity(IdentityType_t type, void *ptr);
   114 	void DestroyIdentType(IdentityType_t type);
   115 	void DestroyIdentity(IdentityToken_t *identity);
   116 	void AddDependency(IExtension *myself, const char *filename, bool require, bool autoload);
   117 	void RegisterLibrary(IExtension *myself, const char *name);
   118 	void OverrideNatives(IExtension *myself, const sp_nativeinfo_t *natives);
   119 	void AddCapabilityProvider(IExtension *myself, IFeatureProvider *provider,
   120 		                       const char *name);
   121 	void DropCapabilityProvider(IExtension *myself, IFeatureProvider *provider,
   122 		                        const char *name);
   123 public: //SMGlobalClass
   124 	/* Pre-empt in case anything tries to register idents early */
   125 	void Initialize();
   126 	void OnSourceModShutdown();
   127 public: //IHandleTypeDispatch
   128 	void OnHandleDestroy(HandleType_t type, void *object);
   129 public:
   130 	IdentityToken_t *CreateCoreIdentity();
   131 	void RemoveInterfaces(IExtension *pExtension);
   132 	FeatureStatus TestFeature(IPluginRuntime *pRuntime, FeatureType feature, const char *name);
   133 	FeatureStatus TestNative(IPluginRuntime *pRuntime, const char *name);
   134 	FeatureStatus TestCap(const char *name);
   135 public:
   136 	inline IdentityToken_t *GetIdentRoot()
   137 	{
   138 		return &m_IdentRoot;
   139 	}
   140 public:
   141 	void BindNativesToPlugin(CPlugin *pPlugin, bool bCoreOnly);
   142 	void BindNativeToPlugin(CPlugin *pPlugin, NativeEntry *pEntry);
   143 	NativeEntry *AddFakeNative(IPluginFunction *pFunc, const char *name, SPVM_FAKENATIVE_FUNC func);
   144 	NativeEntry *FindNative(const char *name);
   145 private:
   146 	NativeEntry *AddNativeToCache(CNativeOwner *pOwner, const sp_nativeinfo_t *ntv);
   147 	void ClearNativeFromCache(CNativeOwner *pOwner, const char *name);
   148 	void BindNativeToPlugin(CPlugin *pPlugin, 
   149 		sp_native_t *ntv, 
   150 		uint32_t index, 
   151 		NativeEntry *pEntry);
   152 private:
   153 	List<IfaceInfo> m_Interfaces;
   154 	HandleType_t m_TypeRoot;
   155 	IdentityToken_t m_IdentRoot;
   156 	HandleType_t m_IfaceType;
   157 	IdentityType_t m_CoreType;
   158 	KTrie<NativeEntry *> m_NtvCache;
   159 	KTrie<Capability> m_caps;
   160 };
   161 
   162 extern ShareSystem g_ShareSys;
   163 
   164 #endif //_INCLUDE_SOURCEMOD_SHARESYSTEM_H_