286 lines
		
	
	
		
			7.4 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			286 lines
		
	
	
		
			7.4 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
/**
 | 
						|
 * vim: set ts=4 :
 | 
						|
 * =============================================================================
 | 
						|
 * SourceMod Sample Extension
 | 
						|
 * Copyright (C) 2004-2008 AlliedModders LLC.  All rights reserved.
 | 
						|
 * =============================================================================
 | 
						|
 *
 | 
						|
 * This program is free software; you can redistribute it and/or modify it under
 | 
						|
 * the terms of the GNU General Public License, version 3.0, as published by the
 | 
						|
 * Free Software Foundation.
 | 
						|
 * 
 | 
						|
 * 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, see <http://www.gnu.org/licenses/>.
 | 
						|
 *
 | 
						|
 * As a special exception, AlliedModders LLC gives you permission to link the
 | 
						|
 * code of this program (as well as its derivative works) to "Half-Life 2," the
 | 
						|
 * "Source Engine," the "SourcePawn JIT," and any Game MODs that run on software
 | 
						|
 * by the Valve Corporation.  You must obey the GNU General Public License in
 | 
						|
 * all respects for all other code used.  Additionally, AlliedModders LLC grants
 | 
						|
 * this exception to all derivative works.  AlliedModders LLC defines further
 | 
						|
 * exceptions, found in LICENSE.txt (as of this writing, version JULY-31-2007),
 | 
						|
 * or <http://www.sourcemod.net/license.php>.
 | 
						|
 *
 | 
						|
 * Version: $Id$
 | 
						|
 */
 | 
						|
 | 
						|
#include "extension.h"
 | 
						|
 | 
						|
/**
 | 
						|
 * @file extension.cpp
 | 
						|
 * @brief Implement extension code here.
 | 
						|
 */
 | 
						|
 | 
						|
Structs g_Structs;		/**< Global singleton for extension's main interface */
 | 
						|
 | 
						|
SMEXT_LINK(&g_Structs);
 | 
						|
 | 
						|
HandleType_t g_StructHandle = 0;
 | 
						|
StructHandler g_StructHandler;
 | 
						|
 | 
						|
IGameConfig *conf;
 | 
						|
 | 
						|
size_t UTIL_Format(char *buffer, size_t maxlength, const char *fmt, ...)
 | 
						|
{
 | 
						|
	va_list ap;
 | 
						|
	va_start(ap, fmt);
 | 
						|
	size_t len = vsnprintf(buffer, maxlength, fmt, ap);
 | 
						|
	va_end(ap);
 | 
						|
 | 
						|
	if (len >= maxlength)
 | 
						|
	{
 | 
						|
		buffer[maxlength - 1] = '\0';
 | 
						|
		return (maxlength - 1);
 | 
						|
	}
 | 
						|
	else
 | 
						|
	{
 | 
						|
		return len;
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
void Structs::SDK_OnAllLoaded()
 | 
						|
{
 | 
						|
	g_StructHandle = handlesys->CreateType("Struct", 
 | 
						|
		&g_StructHandler, 
 | 
						|
		0, 
 | 
						|
		NULL, 
 | 
						|
		NULL, 
 | 
						|
		myself->GetIdentity(), 
 | 
						|
		NULL);
 | 
						|
 | 
						|
	gameconfs->AddUserConfigHook("Structs", this);
 | 
						|
 | 
						|
	sharesys->AddNatives(myself, MyNatives);
 | 
						|
 | 
						|
 | 
						|
	char error[100];
 | 
						|
	if (!gameconfs->LoadGameConfigFile("structs.gamedata", &conf, error, sizeof(error)))
 | 
						|
	{
 | 
						|
		g_pSM->LogError(myself, "Parsing Failed!");
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
void Structs::SDK_OnUnload()
 | 
						|
{
 | 
						|
	handlesys->RemoveType(g_StructHandle, myself->GetIdentity());
 | 
						|
	gameconfs->CloseGameConfigFile(conf);
 | 
						|
	gameconfs->RemoveUserConfigHook("Structs", this);
 | 
						|
}
 | 
						|
 | 
						|
bool Structs::SDK_OnLoad( char *error, size_t maxlength, bool late )
 | 
						|
{
 | 
						|
	sharesys->AddInterface(myself, &g_StructManager);
 | 
						|
 | 
						|
	m_typeLookup.insert("int", Member_Int32);
 | 
						|
	m_typeLookup.insert("int*", Member_Int32Pointer);
 | 
						|
	m_typeLookup.insert("float", Member_Float);
 | 
						|
	m_typeLookup.insert("float*", Member_FloatPointer);
 | 
						|
	m_typeLookup.insert("char", Member_Char);
 | 
						|
	m_typeLookup.insert("char*", Member_CharPointer);
 | 
						|
	m_typeLookup.insert("Vector", Member_Vector);
 | 
						|
	m_typeLookup.insert("Vector*", Member_VectorPointer);
 | 
						|
	m_typeLookup.insert("ent", Member_EHandle);
 | 
						|
	m_typeLookup.insert("ent*", Member_EHandlePointer);
 | 
						|
 | 
						|
	return true;
 | 
						|
}
 | 
						|
 | 
						|
void Structs::ReadSMC_ParseStart()
 | 
						|
{
 | 
						|
	m_bInStruct = false;
 | 
						|
	m_currentStruct = NULL;
 | 
						|
	m_bInMember = false;
 | 
						|
	m_currentMember = NULL;
 | 
						|
}
 | 
						|
 | 
						|
SourceMod::SMCResult Structs::ReadSMC_NewSection( const SMCStates *states, const char *name )
 | 
						|
{
 | 
						|
	if (!m_bInStruct)
 | 
						|
	{
 | 
						|
		m_currentStruct = new StructInfo();
 | 
						|
		UTIL_Format(m_currentStruct->name, sizeof(m_currentStruct->name), "%s", name);
 | 
						|
		m_bInStruct = true;
 | 
						|
 | 
						|
		return SMCResult_Continue;
 | 
						|
	}
 | 
						|
 | 
						|
	if (!m_bInMember)
 | 
						|
	{
 | 
						|
		m_bInMember = true;
 | 
						|
		m_currentMember = new MemberInfo();
 | 
						|
		UTIL_Format(m_currentMember->name, sizeof(m_currentMember->name), "%s", name);
 | 
						|
 | 
						|
		return SMCResult_Continue;
 | 
						|
	}
 | 
						|
 | 
						|
	g_pSM->LogMessage(myself, "Cannot nest within a member: line: %i col: %i", states->line, states->col);
 | 
						|
 | 
						|
	return SMCResult_HaltFail;
 | 
						|
}
 | 
						|
 | 
						|
SourceMod::SMCResult Structs::ReadSMC_KeyValue( const SMCStates *states, const char *key, const char *value )
 | 
						|
{
 | 
						|
	if (!m_bInStruct)
 | 
						|
	{
 | 
						|
		//hrm..
 | 
						|
		g_pSM->LogMessage(myself, "Unknown value %s: line: %i col: %i", key, states->line, states->col);
 | 
						|
		return SMCResult_HaltFail;
 | 
						|
	}
 | 
						|
 | 
						|
	if (m_bInMember)
 | 
						|
	{
 | 
						|
		if (strcmp(key, "type") == 0)
 | 
						|
		{
 | 
						|
			MemberType *pType = m_typeLookup.retrieve(value);
 | 
						|
 | 
						|
			if (pType == NULL)
 | 
						|
			{
 | 
						|
				//invalid type
 | 
						|
				g_pSM->LogMessage(myself, "Invalid Type: line: %i col: %i", states->line, states->col);
 | 
						|
				return SMCResult_HaltFail;
 | 
						|
			}
 | 
						|
 | 
						|
			/* Note all 'int' types are assumed to be 32bit for the moment */
 | 
						|
			m_currentMember->type = *pType;
 | 
						|
		}
 | 
						|
		else if (strcmp(key, "size") == 0)
 | 
						|
		{
 | 
						|
			m_currentMember->size = atoi(value);
 | 
						|
		}
 | 
						|
#if defined WIN32
 | 
						|
		else if (strcmp(key, "windows") == 0)
 | 
						|
#else
 | 
						|
		else if (strcmp(key, "linux") == 0)
 | 
						|
#endif
 | 
						|
		{
 | 
						|
			m_currentMember->offset = atoi(value);
 | 
						|
		}
 | 
						|
	}
 | 
						|
	else
 | 
						|
	{
 | 
						|
		if (strcmp(key, "size") == 0)
 | 
						|
		{
 | 
						|
			m_currentStruct->size = atoi(value);
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	return SMCResult_Continue;
 | 
						|
}
 | 
						|
 | 
						|
SourceMod::SMCResult Structs::ReadSMC_LeavingSection( const SMCStates *states )
 | 
						|
{
 | 
						|
	if (m_bInMember)
 | 
						|
	{
 | 
						|
		/** Check for missing or invalid parameters */
 | 
						|
		if (m_currentMember->type == Member_Unknown || m_currentMember->offset == -1)
 | 
						|
		{
 | 
						|
			g_pSM->LogMessage(myself, "Missing offset or type: line: %i col: %i", states->line, states->col);
 | 
						|
			return SMCResult_HaltFail;
 | 
						|
		}
 | 
						|
 | 
						|
		if (m_currentMember->size == -1)
 | 
						|
		{
 | 
						|
			switch (m_currentMember->type)
 | 
						|
			{
 | 
						|
				case Member_Float:
 | 
						|
				case Member_FloatPointer:
 | 
						|
				case Member_Vector:
 | 
						|
				case Member_VectorPointer:
 | 
						|
				case Member_EHandle:
 | 
						|
				case Member_EHandlePointer:
 | 
						|
				{
 | 
						|
					break;
 | 
						|
				}
 | 
						|
 | 
						|
				default:
 | 
						|
				{
 | 
						|
					g_pSM->LogMessage(myself, "Missing size: line: %i col: %i", states->line, states->col);
 | 
						|
					return SMCResult_HaltFail;
 | 
						|
				}
 | 
						|
			}
 | 
						|
		}
 | 
						|
 | 
						|
		/* Work out the int size */
 | 
						|
		if (m_currentMember->type == Member_Int32)
 | 
						|
		{
 | 
						|
			if (m_currentMember->size == 1)
 | 
						|
			{
 | 
						|
				m_currentMember->type = Member_Int8;
 | 
						|
			} 
 | 
						|
			else if (m_currentMember->size == 2)
 | 
						|
			{
 | 
						|
				m_currentMember->type = Member_Int16;
 | 
						|
			} 
 | 
						|
			else if (m_currentMember->size == 4)
 | 
						|
			{
 | 
						|
				m_currentMember->type = Member_Int32;
 | 
						|
			} 
 | 
						|
			else
 | 
						|
			{
 | 
						|
				g_pSM->LogMessage(myself, "Invalid int size %i: line: %i col: %i", m_currentMember->size, states->line, states->col);
 | 
						|
				return SMCResult_HaltFail;
 | 
						|
			}
 | 
						|
 | 
						|
		} 
 | 
						|
		else if (m_currentMember->type == Member_Int32)
 | 
						|
		{
 | 
						|
			if (m_currentMember->size == 1)
 | 
						|
			{
 | 
						|
				m_currentMember->type = Member_Int8Pointer;
 | 
						|
			} 
 | 
						|
			else if (m_currentMember->size == 2)
 | 
						|
			{
 | 
						|
				m_currentMember->type = Member_Int16Pointer;
 | 
						|
			} 
 | 
						|
			else if (m_currentMember->size == 4)
 | 
						|
			{
 | 
						|
				m_currentMember->type = Member_Int32Pointer;
 | 
						|
			} 
 | 
						|
			else
 | 
						|
			{
 | 
						|
				g_pSM->LogMessage(myself, "Invalid int size %i: line: %i col: %i", m_currentMember->size, states->line, states->col);
 | 
						|
				return SMCResult_HaltFail;
 | 
						|
			}
 | 
						|
		}
 | 
						|
 | 
						|
		m_currentStruct->AddMember(m_currentMember->name, m_currentMember);
 | 
						|
 | 
						|
		m_bInMember = false;
 | 
						|
		m_currentMember = NULL;
 | 
						|
 | 
						|
		return SMCResult_Continue;
 | 
						|
	}
 | 
						|
 | 
						|
	g_StructManager.AddStruct(m_currentStruct->name, m_currentStruct);
 | 
						|
 | 
						|
	m_bInStruct = false;
 | 
						|
	m_currentStruct = NULL;
 | 
						|
 | 
						|
	return SMCResult_Continue;
 | 
						|
}
 |