XML reading/writing

1 files changed, 437 insertions, 17 deletions

diff --git a/client/Proxifier.cpp b/client/Proxifier.cpp
index 5ef213d..84553f7 100644
--- a/client/Proxifier.cpp
+++ b/client/Proxifier.cpp
@@ -18,7 +18,7 @@ Proxifier::Chain::Chain(): id(0)
 }
 
 
-Proxifier::Rule::Rule(): isEnabled(false), id(0)
+Proxifier::Rule::Rule(): isEnabled(false), actId(0)
 {
 }
 
@@ -37,7 +37,8 @@ Proxifier* Proxifier::GetInstance()
 
 Proxifier::Proxifier(): valid(false)
 {
-	Update();
+	filename = "Default.ppx";
+	ReadConfig();
 }
 
 bool Proxifier::IsValid()
@@ -45,23 +46,23 @@ bool Proxifier::IsValid()
 	return valid;
 }
 
-void Proxifier::Update()
+void Proxifier::ReadConfig()
 {
 	valid = false;
-	configDom = new QDomDocument("config");
-	QFile config("Default.ppx");
+	QDomDocument configDom("config");
+	QFile config(filename);
 	if (!config.open(QIODevice::ReadOnly))
 	{
 		Logger::Error("Can't open Proxifier configuration file\n");
 		return;
 	}
-	if (!configDom->setContent(&config))
+	if (!configDom.setContent(&config))
 	{
 		Logger::Fatal("Invalid Proxifier configuration file!");
 		return;
 	}
 
-	QDomElement root = configDom->documentElement();
+	QDomElement root = configDom.documentElement();
 	if (root.tagName() != "ProxifierProfile")
 	{
 		Logger::Fatal("Invalid Proxifier configuration file!");
@@ -73,28 +74,28 @@ void Proxifier::Update()
 	 */
 
 	// parse proxy list
-	if (!UpdateProxyList(root))
+	if (!ReadProxyList(root))
 	{
 		Logger::Fatal("Invalid Proxifier configuration file!");
 		return;
 	}
 	// parse chain list
-	if (!UpdateChainList(root))
+	if (!ReadChainList(root))
 	{
 		Logger::Fatal("Invalid Proxifier configuration file!");
 		return;
 	}
 	// parse rule list
-	if (!UpdateRuleList(root))
+	if (!ReadRuleList(root))
 	{
 		Logger::Fatal("Invalid Proxifier configuration file!");
 		return;
 	}
-
+	config.close();
 	valid = true;
 }
 
-bool Proxifier::UpdateProxyList(QDomElement& root)
+bool Proxifier::ReadProxyList(QDomElement& root)
 {
 	Logger::Trace("Processing Proxifier proxy list\n");
 	QDomElement proxyListElem = root.firstChildElement("ProxyList");
@@ -157,6 +158,7 @@ bool Proxifier::UpdateProxyList(QDomElement& root)
 			if (! authElem.hasChildNodes())
 			{
 				// in case if "NTLM + userLogin" = selfclosing auth tag
+				Logger::Trace("Auth section is empty\n");
 				proxyRecord.emptyAuth = true;
 			}
 			else
@@ -171,7 +173,8 @@ bool Proxifier::UpdateProxyList(QDomElement& root)
 		}
 		else
 		{
-			Logger::Trace("Auth section is empty\n");
+			Logger::Trace("Auth section is not present\n");
+			proxyRecord.useAuth = false;
 		}
 		Logger::Trace("Found proxy in proxifier config: id=%d\n", proxyRecord.id);
 		proxyList.push_back(proxyRecord);
@@ -182,7 +185,7 @@ bool Proxifier::UpdateProxyList(QDomElement& root)
 	return true;
 }
 
-bool Proxifier::UpdateChainList(QDomElement& root)
+bool Proxifier::ReadChainList(QDomElement& root)
 {
 	Logger::Trace("Processing Proxifier chain list\n");
 	QDomElement chainListElem = root.firstChildElement("ChainList");
@@ -242,7 +245,7 @@ bool Proxifier::UpdateChainList(QDomElement& root)
 	return true;
 }
 
-bool Proxifier::UpdateRuleList(QDomElement& root)
+bool Proxifier::ReadRuleList(QDomElement& root)
 {
 	Logger::Trace("Processing Proxifier rule list\n");
 	QDomElement ruleListElem = root.firstChildElement("RuleList");
@@ -295,8 +298,8 @@ bool Proxifier::UpdateRuleList(QDomElement& root)
 		QString actionTxt = actionElem.attribute("type");
 		if ((actionTxt == "Chain") || (actionTxt == "Proxy"))
 		{
-			QString idTxt = actionElem.text();
-			ruleRecord.id = idTxt.toInt();
+			QString actIdTxt = actionElem.text();
+			ruleRecord.actId = actIdTxt.toInt();
 		}
 		ruleRecord.action = actionTxt.toStdString();
 		Logger::Trace("Found rule in proxifier config: name=%s\n", ruleRecord.name.c_str());
@@ -308,3 +311,420 @@ bool Proxifier::UpdateRuleList(QDomElement& root)
 
 	return true;
 }
+
+int Proxifier::GetNextId()
+{
+	if (!valid)
+	{
+		return -1;
+	}
+	int max = 100; // first default proxifier ID
+	for (unsigned i = 0; i < proxyList.size(); i++)
+	{
+		if (proxyList[i].id > max)
+		{
+			max = proxyList[i].id;
+		}
+	}
+	for (unsigned i = 0; i < chainList.size(); i++)
+	{
+		if (chainList[i].id > max)
+		{
+			max = chainList[i].id;
+		}
+	}
+	return max;
+}
+
+bool Proxifier::AddProxy(Proxy& proxy, int* id)
+{
+	Logger::Trace("Adding new proxy to Proxifier configuration\n");
+	if (!valid)
+	{
+		Logger::Error("Current Proxifier configuration is not valid. Update first!\n");
+		return false;
+	}
+	QDomDocument configDom("config");
+	QFile config(filename);
+	if (!config.open(QIODevice::WriteOnly))
+	{
+		Logger::Error("Can't open Proxifier configuration file\n");
+		return false;
+	}
+	if (!configDom.setContent(&config))
+	{
+		Logger::Fatal("Invalid Proxifier configuration file!");
+		return false;
+	}
+	QDomElement root = configDom.documentElement();
+	if (root.tagName() != "ProxifierProfile")
+	{
+		Logger::Error("Invalid Proxifier configuration file!");
+		return false;
+	}
+	QDomElement proxyListElem = root.firstChildElement("ProxyList");
+	if (proxyListElem.isNull())
+	{
+		return false;
+	}
+	
+	// create proxy record
+	// root element
+	QDomElement proxyElem = configDom.createElement("Proxy");
+	*id = GetNextId();
+	if (*id == -1)
+	{
+		Logger::Error("Can't get valid ID number!\n");
+		return false;
+	}
+	proxyElem.setAttribute("id", *id);
+	proxyElem.setAttribute("type", QString::fromLocal8Bit(proxy.type.c_str()));
+	// address element
+	QDomElement addrElem = configDom.createElement("Address");
+	QDomText addrTxt = configDom.createTextNode(QString::fromLocal8Bit(proxy.host.c_str()));
+	addrElem.appendChild(addrTxt);
+	proxyElem.appendChild(addrElem);
+	// port element
+	QDomElement portElem = configDom.createElement("Port");
+	QString portStr;
+	QDomText portTxt = configDom.createTextNode(portStr.setNum(proxy.port));
+	portElem.appendChild(portTxt);
+	proxyElem.appendChild(portElem);
+	// option element
+	QDomElement optsElem = configDom.createElement("Options");
+	QString optsStr;
+	QDomText optsTxt = configDom.createTextNode(optsStr.setNum(proxy.option));
+	optsElem.appendChild(optsTxt);
+	proxyElem.appendChild(optsElem);
+	// auth element
+	if (proxy.useAuth)
+	{
+		QDomElement authElem = configDom.createElement("Authentication");
+		authElem.setAttribute("enabled", "true");
+		if (!proxy.emptyAuth)
+		{
+			// username
+			QDomElement userElem = configDom.createElement("Username");
+			QDomText userTxt = configDom.createTextNode(QString::fromLocal8Bit(proxy.login.c_str()));
+			userElem.appendChild(userTxt);
+			authElem.appendChild(userElem);
+			// password
+			QDomElement passElem = configDom.createElement("Password");
+			QDomText passTxt = configDom.createTextNode(QString::fromLocal8Bit(proxy.password.c_str()));
+			passElem.appendChild(passTxt);
+			authElem.appendChild(passElem);
+		}
+		proxyElem.appendChild(authElem);
+	}
+	proxyListElem.appendChild(proxyElem);
+	QTextStream(&config) << configDom.toString(indent);
+	config.close();
+	return true;
+}
+
+bool Proxifier::AddChain(string& name, int *id)
+{
+	QDomDocument configDom("config");
+	QFile config(filename);
+	if (!config.open(QIODevice::WriteOnly))
+	{
+		Logger::Error("Can't open Proxifier configuration file\n");
+		return false;
+	}
+	if (!configDom.setContent(&config))
+	{
+		Logger::Fatal("Invalid Proxifier configuration file!");
+		return false;
+	}
+	QDomElement root = configDom.documentElement();
+	if (root.tagName() != "ProxifierProfile")
+	{
+		Logger::Error("Invalid Proxifier configuration file!");
+		return false;
+	}
+	QDomElement chainListElem = root.firstChildElement("ChainList");
+	if (chainListElem.isNull())
+	{
+		return false;
+	}
+	
+	QDomElement chainElem = configDom.createElement("Chain");
+	*id = GetNextId();
+	if (*id == -1)
+	{
+		Logger::Error("Can't get valid ID number!\n");
+		return false;
+	}
+	chainElem.setAttribute("id", *id);
+	QDomElement nameElem = configDom.createElement("Name");
+	QDomText nameTxt = configDom.createTextNode(QString::fromLocal8Bit(name.c_str()));
+	nameElem.appendChild(nameTxt);
+	chainElem.appendChild(nameElem);
+	chainListElem.appendChild(chainElem);
+	
+	QTextStream(&config) << configDom.toString(indent);
+	config.close();
+	return true;
+}
+
+bool Proxifier::AddChainProxy(int proxyId, int chainId)
+{
+	QDomDocument configDom("config");
+	QFile config(filename);
+	if (!config.open(QIODevice::WriteOnly))
+	{
+		Logger::Error("Can't open Proxifier configuration file\n");
+		return false;
+	}
+	if (!configDom.setContent(&config))
+	{
+		Logger::Fatal("Invalid Proxifier configuration file!");
+		return false;
+	}
+	QDomElement root = configDom.documentElement();
+	if (root.tagName() != "ProxifierProfile")
+	{
+		Logger::Error("Invalid Proxifier configuration file!");
+		return false;
+	}
+	QDomElement chainListElem = root.firstChildElement("ChainList");
+	if (chainListElem.isNull())
+	{
+		return false;
+	}
+	
+	QDomElement chainElem = chainListElem.firstChildElement("Chain");
+	while (!chainElem.isNull())
+	{
+		bool ok;
+		if (!chainElem.hasAttribute("id"))
+		{
+			Logger::Error("Invalid Chain entry\n");
+			chainElem = chainElem.nextSiblingElement("Chain");
+			continue;
+		}
+		QString idStr = chainElem.attribute("id");
+		int id = idStr.toInt(&ok);
+		if (!ok)
+		{
+			Logger::Error("Invalid Chain id\n");
+			chainElem = chainElem.nextSiblingElement("Chain");
+			continue;
+		}
+		if (id == chainId)
+		{
+			// chain found => add proxy
+			QDomElement proxyElem = configDom.createElement("Proxy");
+			proxyElem.setAttribute("enabled", "true");
+			QString idTxt;
+			idTxt.setNum(proxyId);
+			QDomText nameTxt = configDom.createTextNode(idTxt);
+			proxyElem.appendChild(nameTxt);
+			chainElem.appendChild(proxyElem);
+			break;
+		}
+	}
+	
+	QTextStream(&config) << configDom.toString(indent);
+	config.close();
+	return true;
+}
+
+bool Proxifier::RemoveChainProxy(int proxyId, int chainId)
+{
+	QDomDocument configDom("config");
+	QFile config(filename);
+	if (!config.open(QIODevice::WriteOnly))
+	{
+		Logger::Error("Can't open Proxifier configuration file\n");
+		return false;
+	}
+	if (!configDom.setContent(&config))
+	{
+		Logger::Fatal("Invalid Proxifier configuration file!");
+		return false;
+	}
+	QDomElement root = configDom.documentElement();
+	if (root.tagName() != "ProxifierProfile")
+	{
+		Logger::Error("Invalid Proxifier configuration file!");
+		return false;
+	}
+	QDomElement chainListElem = root.firstChildElement("ChainList");
+	if (chainListElem.isNull())
+	{
+		return false;
+	}
+	
+	QDomElement chainElem = chainListElem.firstChildElement("Chain");
+	while (!chainElem.isNull())
+	{
+		bool ok;
+		if (!chainElem.hasAttribute("id"))
+		{
+			Logger::Error("Invalid Chain entry\n");
+			chainElem = chainElem.nextSiblingElement("Chain");
+			continue;
+		}
+		QString idStr = chainElem.attribute("id");
+		int id = idStr.toInt(&ok);
+		if (!ok)
+		{
+			Logger::Error("Invalid Chain id\n");
+			chainElem = chainElem.nextSiblingElement("Chain");
+			continue;
+		}
+		if (id == chainId)
+		{
+			// chain found => find proxy
+			QDomElement proxyElem = chainElem.firstChildElement("Proxy");
+			while (!proxyElem.isNull())
+			{
+				QString idTxt = proxyElem.text();
+				int id = idTxt.toInt();
+				if (id == proxyId)
+				{
+					chainElem.removeChild(proxyElem);
+					break;
+				}
+			}
+			break;
+		}
+	}
+	
+	QTextStream(&config) << configDom.toString(indent);
+	config.close();	
+	return true;
+}
+
+
+bool Proxifier::AddRule(Rule& rule)
+{
+	QDomDocument configDom("config");
+	QFile config(filename);
+	if (!config.open(QIODevice::WriteOnly))
+	{
+		Logger::Error("Can't open Proxifier configuration file\n");
+		return false;
+	}
+	if (!configDom.setContent(&config))
+	{
+		Logger::Fatal("Invalid Proxifier configuration file!");
+		return false;
+	}
+	QDomElement root = configDom.documentElement();
+	if (root.tagName() != "ProxifierProfile")
+	{
+		Logger::Error("Invalid Proxifier configuration file!");
+		return false;
+	}
+	QDomElement ruleListElem = root.firstChildElement("RuleList");
+	if (ruleListElem.isNull())
+	{
+		return false;
+	}
+	
+	QDomElement ruleElem = configDom.createElement("Rule");
+	if (rule.isEnabled)
+	{
+		ruleElem.setAttribute("enabled", "true");
+	}
+	else
+	{
+		ruleElem.setAttribute("enabled", "false");
+	}
+	// name element
+	QDomElement nameElem = configDom.createElement("Name");
+	QDomText nameTxt = configDom.createTextNode(QString::fromLocal8Bit(rule.name.c_str()));
+	nameElem.appendChild(nameTxt);
+	ruleElem.appendChild(nameElem);
+	// applications element
+	if (! rule.apps.empty())
+	{
+		QDomElement appsElem = configDom.createElement("Applications");
+		QDomText appsTxt = configDom.createTextNode(QString::fromLocal8Bit(rule.apps.c_str()));
+		appsElem.appendChild(appsTxt);
+		ruleElem.appendChild(appsElem);
+	}
+	// targets element
+	if (! rule.targets.empty())
+	{
+		QDomElement targetsElem = configDom.createElement("Applications");
+		QDomText targetsTxt = configDom.createTextNode(QString::fromLocal8Bit(rule.targets.c_str()));
+		targetsElem.appendChild(targetsTxt);
+		ruleElem.appendChild(targetsElem);
+	}
+	// ports element
+	if (! rule.targets.empty())
+	{
+		QDomElement portsElem = configDom.createElement("Ports");
+		QDomText portsTxt = configDom.createTextNode(QString::fromLocal8Bit(rule.ports.c_str()));
+		portsElem.appendChild(portsTxt);
+		ruleElem.appendChild(portsElem);
+	}
+	// action element
+	if (rule.action.empty())
+	{
+		Logger::Error("Invalid action value: %s\n", rule.action.c_str());
+		return false;
+	}
+	QDomElement actionElem = configDom.createElement("Action");
+	QString actionType = QString::fromLocal8Bit(rule.action.c_str());
+	actionElem.setAttribute("type", actionType);
+	if ((actionType == "Chain") || (actionType == "Proxy"))
+	{
+		QString actionId;
+		actionId.setNum(rule.actId);
+		QDomText portsTxt = configDom.createTextNode(actionId);
+	}
+	ruleElem.appendChild(actionElem);
+	
+	QTextStream(&config) << configDom.toString(indent);
+	config.close();
+	return true;
+}
+
+bool Proxifier::RemoveRule(string name)
+{
+	// !names only compared!
+	QDomDocument configDom("config");
+	QFile config(filename);
+	if (!config.open(QIODevice::WriteOnly))
+	{
+		Logger::Error("Can't open Proxifier configuration file\n");
+		return false;
+	}
+	if (!configDom.setContent(&config))
+	{
+		Logger::Fatal("Invalid Proxifier configuration file!");
+		return false;
+	}
+	QDomElement root = configDom.documentElement();
+	if (root.tagName() != "ProxifierProfile")
+	{
+		Logger::Error("Invalid Proxifier configuration file!");
+		return false;
+	}
+	QDomElement ruleListElem = root.firstChildElement("ChainList");
+	if (ruleListElem.isNull())
+	{
+		return false;
+	}
+	
+	QDomElement ruleElem = ruleListElem.firstChildElement("Rule");
+	while (!ruleElem.isNull())
+	{
+		QDomElement nameElem = ruleElem.firstChildElement("Name");
+		QString nameStr = nameElem.text();
+		if (nameStr == QString::fromLocal8Bit(name.c_str()))
+		{
+			ruleListElem.removeChild(ruleElem);
+			break;
+		}
+	}
+	
+	QTextStream(&config) << configDom.toString(indent);
+	config.close();
+	return true;
+}
+