From 1db755d190e2a6f099905d1ba11ca01c6b273dc9 Mon Sep 17 00:00:00 2001
From: danieldg <danieldg@e03df62e-2008-0410-955e-edbf42e46eb7>
Date: Sun, 18 Oct 2009 17:58:49 +0000
Subject: Add <define> configuration tag

git-svn-id: http://svn.inspircd.org/repository/trunk/inspircd@11907 e03df62e-2008-0410-955e-edbf42e46eb7
---
 src/configreader.cpp | 45 ++++++++++++++++++++++++++++++++++++---------
 1 file changed, 36 insertions(+), 9 deletions(-)

(limited to 'src')

diff --git a/src/configreader.cpp b/src/configreader.cpp
index 35196343d..b586db531 100644
--- a/src/configreader.cpp
+++ b/src/configreader.cpp
@@ -53,12 +53,17 @@ enum ParseFlags
 struct ParseStack
 {
 	std::vector<std::string> reading;
+	std::map<std::string, std::string> vars;
 	ConfigDataHash& output;
 	std::stringstream& errstr;
 
 	ParseStack(ServerConfig* conf)
 		: output(conf->config_data), errstr(conf->errstr)
-	{ }
+	{
+		vars["amp"] = "&";
+		vars["quot"] = "\"";
+		vars["newline"] = vars["nl"] = "\n";
+	}
 	bool ParseFile(const std::string& name, int flags);
 	bool ParseExec(const std::string& name, int flags);
 	void DoInclude(ConfigTag* includeTag, int flags);
@@ -161,13 +166,27 @@ struct Parser
 		while (1)
 		{
 			ch = next();
-			if (ch == '\\')
+			if (ch == '&')
 			{
-				ch = next();
-				if (ch == 'n')
-					ch = '\n';
-				else if (ch == 'r')
-					ch = '\r';
+				std::string varname;
+				while (1)
+				{
+					ch = next();
+					if (isalnum(ch))
+						varname.push_back(ch);
+					else if (ch == ';')
+						break;
+					else
+					{
+						stack.errstr << "Invalid XML entity name in value of <" + tag->tag + ":" + key + ">\n"
+							<< "To include an ampersand or quote, use &amp; or &quot;\n";
+						throw CoreException("Parse error");
+					}
+				}
+				std::map<std::string, std::string>::iterator var = stack.vars.find(varname);
+				if (var == stack.vars.end())
+					throw CoreException("Undefined XML entity reference '&" + varname + ";'");
+				value.append(var->second);
 			}
 			else if (ch == '"')
 				break;
@@ -194,13 +213,21 @@ struct Parser
 
 		while (kv());
 
-		if (tag->tag == "include")
+		if (name == "include")
 		{
 			stack.DoInclude(tag, flags);
 		}
+		else if (name == "define")
+		{
+			std::string varname = tag->getString("name");
+			std::string value = tag->getString("value");
+			if (varname.empty())
+				throw CoreException("Variable definition must include variable name");
+			stack.vars[varname] = value;
+		}
 		else
 		{
-			stack.output.insert(std::make_pair(tag->tag, tag));
+			stack.output.insert(std::make_pair(name, tag));
 		}
 		// this is not a leak; reference<> takes care of the delete
 		tag = NULL;
-- 
cgit v1.2.3