dvander@370: /**
ds@653:  * vim: set ts=4 :
ds@1330:  * =============================================================================
dvander@1231:  * SourceMod
dvander@1231:  * Copyright (C) 2004-2007 AlliedModders LLC.  All rights reserved.
ds@1330:  * =============================================================================
dvander@370:  *
ds@1330:  * This program is free software; you can redistribute it and/or modify it under
ds@1330:  * the terms of the GNU General Public License, version 3.0, as published by the
ds@1330:  * Free Software Foundation.
dvander@1231:  * 
ds@1330:  * This program is distributed in the hope that it will be useful, but WITHOUT
ds@1330:  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
ds@1330:  * FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
ds@1330:  * details.
dvander@1231:  *
ds@1330:  * You should have received a copy of the GNU General Public License along with
ds@1330:  * this program.  If not, see <http://www.gnu.org/licenses/>.
dvander@1231:  *
ds@1330:  * As a special exception, AlliedModders LLC gives you permission to link the
ds@1330:  * code of this program (as well as its derivative works) to "Half-Life 2," the
ds@1330:  * "Source Engine," the "SourcePawn JIT," and any Game MODs that run on software
ds@1330:  * by the Valve Corporation.  You must obey the GNU General Public License in
ds@1330:  * all respects for all other code used.  Additionally, AlliedModders LLC grants
ds@1330:  * this exception to all derivative works.  AlliedModders LLC defines further
ds@1330:  * exceptions, found in LICENSE.txt (as of this writing, version JULY-31-2007),
ds@1330:  * or <http://www.sourcemod.net/license.php>.
dvander@370:  *
dvander@370:  * Version: $Id$
dvander@370:  */
dvander@370: 
dvander@203: #include <stdarg.h>
dvander@203: #include <string.h>
dvander@203: #include "PluginInfoDatabase.h"
dvander@203: #include "PluginSys.h"
dvander@203: 
dvander@203: void PluginSettings::Init()
dvander@203: {
dvander@203: 	name = -1;
dvander@203: 	pause_val = false;
dvander@203: 	type_val = PluginType_MapUpdated;
dvander@203: 	optarray = -1;
dvander@203: 	opts_num = 0;
dvander@203: 	opts_size = 0;
dvander@203: }
dvander@203: 
dvander@203: /**
dvander@203:  * :TODO: write the logger, make these errors log instead of being static
dvander@203:  * NOTE: once we do that, we will have to change some code to ignore sections
dvander@203:  */
dvander@203: 
dvander@203: CPluginInfoDatabase::CPluginInfoDatabase()
dvander@203: {
dvander@203: 	m_strtab = NULL;
dvander@203: 	m_infodb_count = 0;
dvander@203: 	m_infodb_size = 0;
dvander@203: 	m_infodb = -1;
dvander@203: }
dvander@203: 
dvander@203: CPluginInfoDatabase::~CPluginInfoDatabase()
dvander@203: {
dvander@203: 	delete m_strtab;
dvander@203: }
dvander@203: 
dvander@203: void CPluginInfoDatabase::ReadSMC_ParseStart()
dvander@203: {
dvander@203: 	/* Create or reset our string table */
dvander@203: 	if (m_strtab)
dvander@203: 	{
dvander@203: 		m_strtab->Reset();
dvander@203: 	} else {
dvander@203: 		m_strtab = new BaseStringTable(1024);
dvander@203: 	}
dvander@203: 
dvander@203: 	/* Set our internal states to the beginning */
dvander@203: 	in_plugins = false;
dvander@203: 	cur_plugin = -1;
dvander@203: 	in_options = false;
dvander@264: 	m_infodb_size = 0;
dvander@264: 	m_infodb_count = 0;
dvander@264: 	m_infodb = -1;
dvander@203: }
dvander@203: 
dvander@203: SMCParseResult CPluginInfoDatabase::MakeError(const char *fmt, ...)
dvander@203: {
dvander@203: 	char buffer[512];
dvander@203: 	va_list ap;
dvander@203: 
dvander@203: 	va_start(ap, fmt);
dvander@203: 	vsnprintf(buffer, sizeof(buffer), fmt, ap);
dvander@203: 	va_end(ap);
dvander@203: 
dvander@203: 	m_errmsg = m_strtab->AddString(buffer);
dvander@203: 
dvander@203: 	return SMCParse_HaltFail;
dvander@203: }
dvander@203: 
dvander@203: unsigned int CPluginInfoDatabase::GetSettingsNum()
dvander@203: {
dvander@203: 	return m_infodb_count;
dvander@203: }
dvander@203: 
dvander@203: PluginSettings *CPluginInfoDatabase::GetSettingsIfMatch(unsigned int index, const char *filename)
dvander@203: {
dvander@203: 	BaseMemTable *memtab = m_strtab->GetMemTable();
dvander@205: 	int *table = (int *)memtab->GetAddress(m_infodb);
dvander@203: 
dvander@203: 	if (!table || index >= m_infodb_count)
dvander@203: 	{
dvander@203: 		return NULL;
dvander@203: 	}
dvander@203: 
dvander@205: 	PluginSettings *plugin = (PluginSettings *)memtab->GetAddress(table[index]);
dvander@205: 
dvander@205: 	const char *name = m_strtab->GetString(plugin->name);
dvander@203: 
dvander@203: 	if (!name)
dvander@203: 	{
dvander@203: 		return NULL;
dvander@203: 	}
dvander@203: 
dvander@211: 	if (!g_PluginSys.TestAliasMatch(name, filename))
dvander@203: 	{
dvander@203: 		return NULL;
dvander@203: 	}
dvander@203: 
dvander@205: 	return plugin;
dvander@203: }
dvander@203: 
dvander@203: void CPluginInfoDatabase::GetOptionsForPlugin(PluginSettings *settings, unsigned int opt_num, const char **key, const char **val)
dvander@203: {
dvander@203: 	PluginOpts *table = (PluginOpts *)m_strtab->GetMemTable()->GetAddress(settings->optarray);
dvander@203: 	if (!table)
dvander@203: 	{
dvander@203: 		*key = NULL;
dvander@203: 		*val = NULL;
dvander@203: 		return;
dvander@203: 	}
dvander@203: 
dvander@203: 	if (opt_num >= settings->opts_num)
dvander@203: 	{
dvander@203: 		*key = NULL;
dvander@203: 		*val = NULL;
dvander@203: 		return;
dvander@203: 	}
dvander@203: 
dvander@203: 	*key = m_strtab->GetString(table[opt_num].key);
dvander@203: 	*val = m_strtab->GetString(table[opt_num].val);
dvander@203: }
dvander@203: 
dvander@203: SMCParseResult CPluginInfoDatabase::ReadSMC_KeyValue(const char *key, 
dvander@203: 													 const char *value, 
dvander@203: 													 bool key_quotes, 
dvander@203: 													 bool value_quotes)
dvander@203: {
dvander@203: 	if (cur_plugin != -1)
dvander@203: 	{
dvander@203: 		PluginSettings *plugin = (PluginSettings *)m_strtab->GetMemTable()->GetAddress(cur_plugin);
dvander@203: 		if (!in_options)
dvander@203: 		{
dvander@203: 			if (strcmp(key, "pause") == 0)
dvander@203: 			{
dvander@203: 				if (strcasecmp(value, "yes") == 0)
dvander@203: 				{
dvander@203: 					plugin->pause_val = true;
dvander@203: 				} else {
dvander@203: 					plugin->pause_val = false;
dvander@203: 				}
dvander@203: 			} else if (strcmp(key, "lifetime") == 0) {
dvander@203: 				if (strcasecmp(value, "private") == 0)
dvander@203: 				{
dvander@203: 					plugin->type_val = PluginType_Private;
dvander@203: 				} else if (strcasecmp(value, "mapsync") == 0) {
dvander@203: 					plugin->type_val = PluginType_MapUpdated;
dvander@203: 				} else if (strcasecmp(value, "maponly") == 0) {
dvander@203: 					plugin->type_val = PluginType_MapOnly;
dvander@203: 				} else if (strcasecmp(value, "global") == 0) {
dvander@203: 					plugin->type_val = PluginType_Global;
dvander@203: 				} else {
dvander@203: 					return MakeError("Unknown value for key \"lifetime\": \"%s\"", value);
dvander@203: 				}
dvander@203: 			} else {
dvander@203: 				return MakeError("Unknown property key: \"%s\"", key);
dvander@203: 			}
dvander@203: 		} else {
dvander@203: 			/* Cache every option, valid or not */
dvander@203: 			int keyidx = m_strtab->AddString(key);
dvander@203: 			int validx = m_strtab->AddString(value);
dvander@203: 			PluginOpts *table;
dvander@203: 			BaseMemTable *memtab = m_strtab->GetMemTable();
dvander@203: 			if (plugin->opts_num + 1 > plugin->opts_size)
dvander@203: 			{
dvander@203: 				unsigned int oldsize = plugin->opts_size;
dvander@203: 				if (oldsize == 0)
dvander@203: 				{
dvander@203: 					//right now we don't have many
dvander@203: 					plugin->opts_size = 2;
dvander@203: 				} else {
dvander@203: 					plugin->opts_size *= 2;
dvander@203: 				}
dvander@203: 				int newidx = memtab->CreateMem(plugin->opts_size * sizeof(PluginOpts), (void **)&table);
dvander@293: 				/* in case it resized */
dvander@293: 				plugin = (PluginSettings *)memtab->GetAddress(cur_plugin);
dvander@203: 				if (plugin->optarray != -1)
dvander@203: 				{
dvander@203: 					void *oldtable = memtab->GetAddress(plugin->optarray);
dvander@203: 					memcpy(table, oldtable, oldsize * sizeof(PluginOpts));
dvander@203: 				}
dvander@203: 				plugin->optarray = newidx;
dvander@203: 			} else {
dvander@203: 				table = (PluginOpts *)memtab->GetAddress(plugin->optarray);
dvander@203: 			}
dvander@203: 			PluginOpts *opt = &table[plugin->opts_num++];
dvander@203: 			opt->key = keyidx;
dvander@203: 			opt->val = validx;
dvander@203: 		}
dvander@203: 	} else if (in_plugins) {
dvander@203: 		return MakeError("Unknown property key: \"%s\"", key);
dvander@203: 	} else {
dvander@203: 		/* Ignore anything we don't know about! */
dvander@203: 	}
dvander@203: 
dvander@203: 	return SMCParse_Continue;
dvander@203: }
dvander@203: 
dvander@203: SMCParseResult CPluginInfoDatabase::ReadSMC_LeavingSection()
dvander@203: {
dvander@203: 	if (in_plugins)
dvander@203: 	{
dvander@205: 		if (cur_plugin != -1)
dvander@203: 		{
dvander@203: 			if (in_options)
dvander@203: 			{
dvander@203: 				in_options = false;
dvander@203: 			} else {
dvander@203: 				/* If the plugin is ending, add it to the table */
dvander@203: 				BaseMemTable *memtab = m_strtab->GetMemTable();
dvander@203: 				int *table;
dvander@203: 				if (m_infodb_count + 1 > m_infodb_size)
dvander@203: 				{
dvander@203: 					unsigned int oldsize = m_infodb_size;
dvander@203: 					if (!m_infodb_size)
dvander@203: 					{
dvander@203: 						m_infodb_size = 8;
dvander@203: 					} else {
dvander@203: 						m_infodb_size *= 2;
dvander@203: 					}
dvander@203: 					int newidx = memtab->CreateMem(m_infodb_size, (void **)&table);
dvander@203: 					if (m_infodb != -1)
dvander@203: 					{
dvander@203: 						void *oldtable = (int *)memtab->GetAddress(m_infodb);
dvander@203: 						memcpy(table, oldtable, oldsize * sizeof(int));
dvander@203: 					}
dvander@203: 					m_infodb = newidx;
dvander@203: 				} else {
dvander@203: 					table = (int *)memtab->GetAddress(m_infodb);
dvander@203: 				}
dvander@203: 				/* Assign to table and scrap the current plugin */
dvander@203: 				table[m_infodb_count++] = cur_plugin;
dvander@203: 				cur_plugin = -1;
dvander@203: 			}
dvander@203: 		} else {
dvander@203: 			in_plugins = false;
dvander@203: 		}
dvander@203: 	}
dvander@203: 
dvander@203: 	return SMCParse_Continue;
dvander@203: }
dvander@203: 
dvander@203: SMCParseResult CPluginInfoDatabase::ReadSMC_NewSection(const char *name, bool opt_quotes)
dvander@203: {
dvander@203: 	if (!in_plugins)
dvander@203: 	{
dvander@203: 		/* If we're not in the main Plugins section, and we don't get it for the name, error out */
dvander@203: 		if (strcmp(name, "Plugins") != 0)
dvander@203: 		{
dvander@203: 			return MakeError("Unknown root section: \"%s\"", name);
dvander@203: 		} else {
dvander@203: 			/* Otherwise set our states */
dvander@203: 			in_plugins = true;
dvander@203: 			cur_plugin = -1;
dvander@203: 			in_options = false;
dvander@203: 		}
dvander@203: 	} else {
dvander@203: 		if (cur_plugin == -1)
dvander@203: 		{
dvander@203: 			/* If we get a plugin node and we don't have a current plugin, create a new one */
dvander@203: 			PluginSettings *plugin;
dvander@203: 			cur_plugin = m_strtab->GetMemTable()->CreateMem(sizeof(PluginSettings), (void **)&plugin);
dvander@203: 			plugin->Init();
dvander@203: 			plugin->name = m_strtab->AddString(name);
dvander@203: 			in_options = false;
dvander@203: 		} else {
dvander@203: 			if (!in_options && strcmp(name, "Options") == 0)
dvander@203: 			{
dvander@203: 				in_options = true;
dvander@203: 			} else {
dvander@203: 				return MakeError("Unknown plugin sub-section: \"%s\"", name);
dvander@203: 			}
dvander@203: 		}
dvander@203: 	}
dvander@203: 
dvander@203: 	return SMCParse_Continue;
dvander@203: }