amxmodx/amxmodx/CPlugin.cpp

/* AMX Mod X
*
* by the AMX Mod X Development Team
*  originally developed by OLO
*
*
*  This program is free software; you can redistribute it and/or modify it
*  under the terms of the GNU General Public License as published by the
*  Free Software Foundation; either version 2 of the License, or (at
*  your option) any later version.
*
*  This program is distributed in the hope that it will be useful, but
*  WITHOUT ANY WARRANTY; without even the implied warranty of
*  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
*  General Public License for more details.
*
*  You should have received a copy of the GNU General Public License
*  along with this program; if not, write to the Free Software Foundation,
*  Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
*
*  In addition, as a special exception, the author gives permission to
*  link the code of this program with the Half-Life Game Engine ("HL
*  Engine") and Modified Game Libraries ("MODs") developed by Valve,
*  L.L.C ("Valve"). You must obey the GNU General Public License in all
*  respects for all of the code used other than the HL Engine and MODs
*  from Valve. If you modify this file, you may extend this exception
*  to your version of the file, but you are not obligated to do so. If
*  you do not wish to do so, delete this exception statement from your
*  version.
*/

#include "amxmodx.h"
#include "CPlugin.h"
#include "CForward.h"
#include "CFile.h"
#include "amx.h"
#include "natives.h"

extern const char *no_function;

CPluginMngr::CPlugin* CPluginMngr::loadPlugin(const char* path, const char* name, char* error, int debug) {	
	CPlugin** a = &head;
	while( *a ) a = &(*a)->next;
	*a = new CPlugin( pCounter++ ,path,name,error, debug);
	return (*a);
}

void CPluginMngr::unloadPlugin( CPlugin** a ) {
	CPlugin* next = (*a)->next;
	delete *a;
	*a = next;
	--pCounter;
}

void CPluginMngr::Finalize()
{
	if (m_Finalized)
		return;
	pNatives = BuildNativeTable();

	CPlugin *a = head;
	while (a)
	{
		if (a->getStatusCode() == ps_running)
		{
			amx_Register(a->getAMX(), pNatives, -1);
			a->Finalize();
		}
		a=a->next;
	}
	m_Finalized = true;
}

int  CPluginMngr::loadPluginsFromFile( const char* filename )
{
	char file[256];
	FILE *fp = fopen(build_pathname_r(file, sizeof(file)-1, "%s",filename) , "rt");

	if ( !fp ) 
	{
		AMXXLOG_Log( "[AMXX] Plugins list not found (file \"%s\")",filename);
		return 1;
	}
	
	// Find now folder
	char pluginName[256], error[256], debug[256];
	int debugFlag = 0;
	const char *pluginsDir = get_localinfo("amxx_pluginsdir", "addons/amxmodx/plugins");
	
	String line;

	while ( !feof(fp) ) 
	{
		pluginName[0] = '\0';
		debug[0] = '\0';
		debugFlag = 0;
		line.clear();
		line._fread(fp);
		sscanf(line.c_str(),"%s %s",pluginName, debug);
		if (!isalnum(*pluginName))  continue;

		if (isalnum(*debug) && strcmp(debug, "debug") == 0)
		{
			debugFlag = 1;
		}

		CPlugin* plugin = loadPlugin( pluginsDir , pluginName  , error,  debugFlag);
		
		if (plugin->getStatusCode() == ps_bad_load)
		{
			char errorMsg[255];
			sprintf(errorMsg, "%s (plugin \"%s\")", error, pluginName);
			plugin->setError(errorMsg);
			AMXXLOG_Log("[AMXX] %s", plugin->getError());
		}
	}

	fclose(fp);

	return pCounter;
}

void CPluginMngr::clear() {
	CPlugin**a = &head;	
	while ( *a )
		unloadPlugin(a);
	m_Finalized = false;
	if (pNatives)
	{
		delete [] pNatives;
		pNatives = NULL;
	}
}

CPluginMngr::CPlugin* CPluginMngr::findPluginFast(AMX *amx) 
{ 
	return (CPlugin*)(amx->userdata[3]); 
}

CPluginMngr::CPlugin* CPluginMngr::findPlugin(AMX *amx) {
	CPlugin*a = head;
	while ( a && &a->amx != amx )
		a=a->next;
	return a;
}
	
CPluginMngr::CPlugin* CPluginMngr::findPlugin(int index){
	CPlugin*a = head;
	while ( a && index--)
		a=a->next;
	return a;
}
	
CPluginMngr::CPlugin* CPluginMngr::findPlugin(const char* name) {
	if (!name) return 0;
	int len = strlen(name);
	if (!len) return 0;
	CPlugin*a = head;
	while( a && strncmp(a->name.c_str(), name,len) )
		a=a->next;
	return a;
}

const char* CPluginMngr::CPlugin::getStatus() const {
	switch(status){
	case ps_running: 
			{
					if (m_Debug)
					{
						return "debug";
					} else {
						return "running";
					}
					break;
			}
	case ps_paused: return "paused";
	case ps_bad_load: return "bad load";
	case ps_stopped: return "stopped";
	case ps_locked: return "locked";
	}
	return "error";
}

CPluginMngr::CPlugin::CPlugin(int i, const char* p,const char* n, char* e, int d) : name(n), title(n) {
	const char* unk = "unknown";
	title.assign(unk);
	author.assign(unk);
	version.assign(unk);
	char file[256];
	char* path = build_pathname_r(file, sizeof(file)-1, "%s/%s",p,n);
	code = 0;
	memset(&amx, 0, sizeof(AMX));
	int err = load_amxscript(&amx,&code,path,e, d);
	if ( err == AMX_ERR_NONE )
	{
		status = ps_running;
	} else {
		status = ps_bad_load;
	}
	amx.userdata[3] = this;
	paused_fun = 0;
	next = 0;
	id = i;
	if (status == ps_running)
	{
		m_PauseFwd = registerSPForwardByName(&amx, "plugin_pause");
		m_UnpauseFwd = registerSPForwardByName(&amx, "plugin_unpause");
		if (amx.flags & AMX_FLAG_DEBUG)
		{
			m_Debug = true;
		} else {
			m_Debug = false;
		}
	}
}

CPluginMngr::CPlugin::~CPlugin( )
{
	unload_amxscript( &amx, &code );
}

void CPluginMngr::CPlugin::Finalize()
{
	char buffer[128];

	int old_status = status;
	if (CheckModules(&amx, buffer))
	{
		if ( amx_Register(&amx, core_Natives, -1) != AMX_ERR_NONE )
		{
			status = ps_bad_load;
			sprintf(buffer, "Plugin uses an unknown function (name \"%s\") - check your modules.ini.", no_function);
			errorMsg.assign(buffer);
			amx.error = AMX_ERR_NOTFOUND;
		}
	} else {
		status = ps_bad_load;
		errorMsg.assign(buffer);
		amx.error = AMX_ERR_NOTFOUND;
	}
	if (old_status != status)
	{
		AMXXLOG_Log("[AMXX] Plugin \"%s\" failed to load: %s", name.c_str(), errorMsg.c_str());
	}
}

void CPluginMngr::CPlugin::pauseFunction( int id ) { 
	if (isValid()){
		paused_fun |= (1<<id);
		g_commands.clearBufforedInfo();
	}
}

void CPluginMngr::CPlugin::unpauseFunction( int id ) {
	if (isValid()) {
		paused_fun &= ~(1<<id); 
		g_commands.clearBufforedInfo();
	}
}

void CPluginMngr::CPlugin::setStatus( int a   ) { 
	status = a; 
	g_commands.clearBufforedInfo(); // ugly way
}

// Pause a plugin
void CPluginMngr::CPlugin::pausePlugin()
{
	if (isValid())
	{
		// call plugin_pause if provided
		if (m_PauseFwd != -1)
			executeForwards(m_PauseFwd);
	
		setStatus(ps_paused);
	}
}

// Unpause a plugin
void CPluginMngr::CPlugin::unpausePlugin()
{
	if (isValid())
	{
		// set status first so the function will be marked executable

		setStatus(ps_running);
		// call plugin_unpause if provided
		if (m_UnpauseFwd != -1)
			executeForwards(m_UnpauseFwd);
	}
}