core/logic/ShareSys.h
author David Anderson <dvander@alliedmods.net>
Fri Aug 30 10:16:28 2013 -0700 (2013-08-30)
changeset 4109 10eb68d668af
parent 4108 dabaea5ab45c
permissions -rw-r--r--
Use Refcounted to manage to NativeEntry (bug 5852 part 4, 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_SHARESYSTEM_H_
    33 #define _INCLUDE_SOURCEMOD_SHARESYSTEM_H_
    34 
    35 #include <IShareSys.h>
    36 #include <IHandleSys.h>
    37 #include <am-string.h>
    38 #include <am-utility.h>
    39 #include <am-refcounting.h>
    40 #include <sh_list.h>
    41 #include <sm_stringhashmap.h>
    42 #include <sm_namehashset.h>
    43 #include "common_logic.h"
    44 #include "Native.h"
    45 
    46 using namespace SourceHook;
    47 
    48 namespace SourceMod
    49 {
    50 	struct IdentityToken_t
    51 	{
    52 		Handle_t ident;
    53 		void *ptr;
    54 		IdentityType_t type;
    55 	};
    56 };
    57 
    58 struct IfaceInfo
    59 {
    60 	bool operator ==(const IfaceInfo &info)
    61 	{
    62 		return (info.iface == iface && info.owner == owner);
    63 	}
    64 	SMInterface *iface;
    65 	IExtension *owner;
    66 };
    67 
    68 class CNativeOwner;
    69 class CPlugin;
    70 
    71 struct Capability
    72 {
    73 	IExtension *ext;
    74 	IFeatureProvider *provider;
    75 };
    76 
    77 class ShareSystem :
    78 	public IShareSys,
    79 	public SMGlobalClass,
    80 	public IHandleTypeDispatch
    81 {
    82 	friend class CNativeOwner;
    83 public:
    84 	ShareSystem();
    85 public: //IShareSys
    86 	bool AddInterface(IExtension *myself, SMInterface *pIface);
    87 	bool RequestInterface(const char *iface_name, 
    88 		unsigned int iface_vers,
    89 		IExtension *myself,
    90 		SMInterface **pIface);
    91 	void AddNatives(IExtension *myself, const sp_nativeinfo_t *natives);
    92 	IdentityType_t CreateIdentType(const char *name);
    93 	IdentityType_t FindIdentType(const char *name);
    94 	IdentityToken_t *CreateIdentity(IdentityType_t type, void *ptr);
    95 	void DestroyIdentType(IdentityType_t type);
    96 	void DestroyIdentity(IdentityToken_t *identity);
    97 	void AddDependency(IExtension *myself, const char *filename, bool require, bool autoload);
    98 	void RegisterLibrary(IExtension *myself, const char *name);
    99 	void OverrideNatives(IExtension *myself, const sp_nativeinfo_t *natives);
   100 	void AddCapabilityProvider(IExtension *myself, IFeatureProvider *provider,
   101 		                       const char *name);
   102 	void DropCapabilityProvider(IExtension *myself, IFeatureProvider *provider,
   103 		                        const char *name);
   104 public: //SMGlobalClass
   105 	/* Pre-empt in case anything tries to register idents early */
   106 	void Initialize();
   107 	void OnSourceModShutdown();
   108 public: //IHandleTypeDispatch
   109 	void OnHandleDestroy(HandleType_t type, void *object);
   110 public:
   111 	IdentityToken_t *CreateCoreIdentity();
   112 	void RemoveInterfaces(IExtension *pExtension);
   113 	FeatureStatus TestFeature(IPluginRuntime *pRuntime, FeatureType feature, const char *name);
   114 	FeatureStatus TestNative(IPluginRuntime *pRuntime, const char *name);
   115 	FeatureStatus TestCap(const char *name);
   116 public:
   117 	inline IdentityToken_t *GetIdentRoot()
   118 	{
   119 		return &m_IdentRoot;
   120 	}
   121 public:
   122 	void BindNativesToPlugin(CPlugin *pPlugin, bool bCoreOnly);
   123 	void BindNativeToPlugin(CPlugin *pPlugin, const ke::Ref<Native> &pEntry);
   124 	ke::PassRef<Native> AddFakeNative(IPluginFunction *pFunc, const char *name, SPVM_FAKENATIVE_FUNC func);
   125 	ke::PassRef<Native> FindNative(const char *name);
   126 private:
   127 	ke::PassRef<Native> AddNativeToCache(CNativeOwner *pOwner, const sp_nativeinfo_t *ntv);
   128 	void ClearNativeFromCache(CNativeOwner *pOwner, const char *name);
   129 	void BindNativeToPlugin(CPlugin *pPlugin, sp_native_t *ntv,  uint32_t index, const ke::Ref<Native> &pEntry);
   130 private:
   131 	typedef NameHashSet<ke::Ref<Native>, Native> NativeCache;
   132 
   133 	List<IfaceInfo> m_Interfaces;
   134 	HandleType_t m_TypeRoot;
   135 	IdentityToken_t m_IdentRoot;
   136 	HandleType_t m_IfaceType;
   137 	IdentityType_t m_CoreType;
   138 	NativeCache m_NtvCache;
   139 	StringHashMap<Capability> m_caps;
   140 };
   141 
   142 extern ShareSystem g_ShareSys;
   143 
   144 #endif //_INCLUDE_SOURCEMOD_SHARESYSTEM_H_