dvander@1945: /**
dvander@1945:  * vim: set ts=4 :
dvander@1945:  * =============================================================================
dvander@1945:  * SourceMod Sample Extension
ds@1947:  * Copyright (C) 2004-2008 AlliedModders LLC.  All rights reserved.
dvander@1945:  * =============================================================================
dvander@1945:  *
dvander@1945:  * This program is free software; you can redistribute it and/or modify it under
dvander@1945:  * the terms of the GNU General Public License, version 3.0, as published by the
dvander@1945:  * Free Software Foundation.
dvander@1945:  * 
dvander@1945:  * This program is distributed in the hope that it will be useful, but WITHOUT
dvander@1945:  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
dvander@1945:  * FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
dvander@1945:  * details.
dvander@1945:  *
dvander@1945:  * You should have received a copy of the GNU General Public License along with
dvander@1945:  * this program.  If not, see <http://www.gnu.org/licenses/>.
dvander@1945:  *
dvander@1945:  * As a special exception, AlliedModders LLC gives you permission to link the
dvander@1945:  * code of this program (as well as its derivative works) to "Half-Life 2," the
dvander@1945:  * "Source Engine," the "SourcePawn JIT," and any Game MODs that run on software
dvander@1945:  * by the Valve Corporation.  You must obey the GNU General Public License in
dvander@1945:  * all respects for all other code used.  Additionally, AlliedModders LLC grants
dvander@1945:  * this exception to all derivative works.  AlliedModders LLC defines further
dvander@1945:  * exceptions, found in LICENSE.txt (as of this writing, version JULY-31-2007),
dvander@1945:  * or <http://www.sourcemod.net/license.php>.
dvander@1945:  *
dvander@1945:  * Version: $Id$
dvander@1945:  */
dvander@1945: 
dvander@1945: #include "extension.h"
dvander@1945: 
dvander@1945: #include <sh_string.h>
dvander@1945: #include "pcre.h"
dvander@1945: #include "CRegEx.h"
dvander@1945: using namespace SourceHook;
dvander@1945: 
dvander@1945: /**
dvander@1945:  * @file extension.cpp
dvander@1945:  * @brief Implement extension code here.
dvander@1945:  */
dvander@1945: 
dvander@1945: RegexExtension g_RegexExtension;		/**< Global singleton for extension's main interface */
dvander@1945: 
dvander@1945: SMEXT_LINK(&g_RegexExtension);
dvander@1945: 
dvander@1945: RegexHandler g_RegexHandler;
dvander@1945: HandleType_t g_RegexHandle=0;
dvander@1945: 
dvander@1945: 
dvander@1945: 
dvander@1945: bool RegexExtension::SDK_OnLoad(char *error, size_t err_max, bool late)
dvander@1945: {
dvander@1945: 	g_pShareSys->AddNatives(myself,regex_natives);
dvander@1945: 	g_RegexHandle = g_pHandleSys->CreateType("Regex", &g_RegexHandler, 0, NULL, NULL, myself->GetIdentity(), NULL);
dvander@1945: 	return true;
dvander@1945: }
dvander@1945: 
dvander@1945: void RegexExtension::SDK_OnUnload()
dvander@1945: {
dvander@1945: 	g_pHandleSys->RemoveType(g_RegexHandle, myself->GetIdentity());
dvander@1945: 
dvander@1945: }
dvander@1945: 
dvander@1945: static cell_t CompileRegex(IPluginContext *pCtx, const cell_t *params)
dvander@1945: {
dvander@1945: 	char *regex;
dvander@1945: 	pCtx->LocalToString(params[1], &regex);
dvander@1945: 
dvander@1945: 	RegEx *x = new RegEx();
dvander@1945: 	
dvander@1945: 	if (x->Compile(regex, params[2]) == 0)
dvander@1945: 	{
dvander@1945: 		cell_t *eOff;
dvander@1945: 		pCtx->LocalToPhysAddr(params[5], &eOff);
dvander@1945: 		const char *err = x->mError;
dvander@1945: 		*eOff = x->mErrorOffset;
dvander@1945: 		pCtx->StringToLocal(params[3], params[4], err ? err:"unknown");
dvander@1945: 		return 0;
dvander@1945: 	}
dvander@1945: 
dvander@1945: 	return g_pHandleSys->CreateHandle(g_RegexHandle, (void*)x, pCtx->GetIdentity(), myself->GetIdentity(), NULL);
dvander@1945: }
dvander@1945: 
dvander@1945: 
dvander@1945: static cell_t MatchRegex(IPluginContext *pCtx, const cell_t *params)
dvander@1945: {
dvander@1945: 	Handle_t hndl = static_cast<Handle_t>(params[1]);
dvander@1945: 	HandleError err;
dvander@1945: 	HandleSecurity sec;
dvander@1945: 	sec.pOwner = NULL;
dvander@1945: 	sec.pIdentity = myself->GetIdentity();
dvander@1945: 
dvander@1945: 	RegEx *x;
dvander@1945: 
dvander@1945: 	if ((err=g_pHandleSys->ReadHandle(hndl, g_RegexHandle, &sec, (void **)&x)) != HandleError_None)
dvander@1945: 	{
dvander@1945: 		return pCtx->ThrowNativeError("Invalid file handle %x (error %d)", hndl, err);
dvander@1945: 	}
dvander@1945: 
dvander@1945: 	if (!x)
dvander@1945: 	{
dvander@1945: 		pCtx->ThrowNativeError("Regex data not found\n");
dvander@1945: 
dvander@1945: 		return 0;
dvander@1945: 	}
dvander@1945: 
dvander@1945: 	char *str;
dvander@1945: 	pCtx->LocalToString(params[2], &str);
dvander@1945: 
dvander@1945: 	int e = x->Match(str);
dvander@1945: 
dvander@1945: 	if (e == -1)
dvander@1945: 	{
dvander@1945: 		/* there was a match error.  move on. */
dvander@1945: 		cell_t *res;
dvander@1945: 		pCtx->LocalToPhysAddr(params[3], &res);
dvander@1945: 		*res = x->mErrorOffset;
dvander@1945: 		/* only clear the match results, since the regex object
dvander@1945: 		   may still be referenced later */
dvander@1945: 		x->ClearMatch();
dvander@1945: 
dvander@1945: 		return -1;
dvander@1945: 	}
dvander@1945: 	else if (e == 0) 
dvander@1945: 	{
dvander@1945: 		/* only clear the match results, since the regex object
dvander@1945: 		   may still be referenced later */
dvander@1945: 		x->ClearMatch();
dvander@1945: 
dvander@1945: 		return 0;
dvander@1945: 	} 
dvander@1945: 	else 
dvander@1945: 	{
dvander@1945: 		return x->mSubStrings;
dvander@1945: 	}
dvander@1945: }
dvander@1945: 
dvander@1945: static cell_t GetRegexSubString(IPluginContext *pCtx, const cell_t *params)
dvander@1945: {
dvander@1945: 	Handle_t hndl=static_cast<Handle_t>(params[1]);
dvander@1945: 	HandleError err;
dvander@1945: 	HandleSecurity sec;
dvander@1945: 	sec.pOwner=NULL;
dvander@1945: 	sec.pIdentity=myself->GetIdentity();
dvander@1945: 
dvander@1945: 	RegEx *x;
dvander@1945: 
dvander@1945: 	if ((err=g_pHandleSys->ReadHandle(hndl, g_RegexHandle, &sec, (void **)&x)) != HandleError_None)
dvander@1945: 	{
dvander@1945: 		return pCtx->ThrowNativeError("Invalid file handle %x (error %d)", hndl, err);
dvander@1945: 	}
dvander@1945: 
dvander@1945: 	if (!x)
dvander@1945: 	{
dvander@1945: 		pCtx->ThrowNativeError("Regex data not found\n");
dvander@1945: 		return 0;
dvander@1945: 	}
dvander@1945: 
dvander@1945: 	static char buffer[4096];
dvander@1945: 	const char *ret=x->GetSubstring(params[2], buffer, sizeof(buffer));
dvander@1945: 
dvander@1945: 	if(!ret)
dvander@1945: 	{
dvander@1945: 		return 0;
dvander@1945: 	}
dvander@1945: 
dvander@1945: 	pCtx->StringToLocalUTF8(params[3], params[4], ret, NULL);
dvander@1945: 
dvander@1945: 	return 1;
dvander@1945: }
dvander@1945: 
dvander@1945: void RegexHandler::OnHandleDestroy(HandleType_t type, void *object)
dvander@1945: {
dvander@1945: 	RegEx *x = (RegEx *)object;
dvander@1945: 
dvander@1945: 	x->Clear();
dvander@1945: 	delete x;
dvander@1945: }
dvander@1945: 
dvander@1945: const sp_nativeinfo_t regex_natives[] = 
dvander@1945: {
dvander@1945: 	{"GetRegexSubString",			GetRegexSubString},
dvander@1945: 	{"MatchRegex",					MatchRegex},
dvander@1945: 	{"CompileRegex",				CompileRegex},
dvander@1945: 	{NULL,							NULL},
dvander@1945: };