Proxifier config parsing

1 files changed, 310 insertions, 0 deletions

diff --git a/client/Proxifier.cpp b/client/Proxifier.cpp
new file mode 100644
index 0000000..5ef213d
--- /dev/null
+++ b/client/Proxifier.cpp
@@ -0,0 +1,310 @@
+
+#include <QtCore>
+#include <QtXml>
+#include "Logger.h"
+#include "Proxifier.h"
+
+
+Proxifier* Proxifier::instance = NULL;
+
+
+Proxifier::Proxy::Proxy(): port(0), id(0), option(0), emptyAuth(true)
+{
+}
+
+
+Proxifier::Chain::Chain(): id(0)
+{
+}
+
+
+Proxifier::Rule::Rule(): isEnabled(false), id(0)
+{
+}
+
+
+/**
+ * Proxifier class
+ */
+Proxifier* Proxifier::GetInstance()
+{
+	if (instance == NULL)
+	{
+		instance = new Proxifier;
+	}
+	return instance;
+}
+
+Proxifier::Proxifier(): valid(false)
+{
+	Update();
+}
+
+bool Proxifier::IsValid()
+{
+	return valid;
+}
+
+void Proxifier::Update()
+{
+	valid = false;
+	configDom = new QDomDocument("config");
+	QFile config("Default.ppx");
+	if (!config.open(QIODevice::ReadOnly))
+	{
+		Logger::Error("Can't open Proxifier configuration file\n");
+		return;
+	}
+	if (!configDom->setContent(&config))
+	{
+		Logger::Fatal("Invalid Proxifier configuration file!");
+		return;
+	}
+
+	QDomElement root = configDom->documentElement();
+	if (root.tagName() != "ProxifierProfile")
+	{
+		Logger::Fatal("Invalid Proxifier configuration file!");
+		return;
+	}
+	QDomElement options = root.firstChildElement("Options");
+	/**
+	 * @todo	process Options if needed
+	 */
+
+	// parse proxy list
+	if (!UpdateProxyList(root))
+	{
+		Logger::Fatal("Invalid Proxifier configuration file!");
+		return;
+	}
+	// parse chain list
+	if (!UpdateChainList(root))
+	{
+		Logger::Fatal("Invalid Proxifier configuration file!");
+		return;
+	}
+	// parse rule list
+	if (!UpdateRuleList(root))
+	{
+		Logger::Fatal("Invalid Proxifier configuration file!");
+		return;
+	}
+
+	valid = true;
+}
+
+bool Proxifier::UpdateProxyList(QDomElement& root)
+{
+	Logger::Trace("Processing Proxifier proxy list\n");
+	QDomElement proxyListElem = root.firstChildElement("ProxyList");
+	if (proxyListElem.isNull())
+	{
+		return false;
+	}
+
+	proxyList.clear();
+	QDomElement proxy = proxyListElem.firstChildElement("Proxy");
+	while (!proxy.isNull())
+	{
+		bool ok = true;		// used when converting values to int
+		Proxy proxyRecord;
+
+		// check whether attributes are present
+		if (! (proxy.hasAttribute("id") && proxy.hasAttribute("type")))
+		{
+			Logger::Error("Invalid Proxy entry\n");
+			proxy = proxy.nextSiblingElement("Proxy");
+			continue;
+		}
+		// proxy id
+		QString id = proxy.attribute("id");
+		proxyRecord.id = id.toInt(&ok);
+		if (!ok)
+		{
+			Logger::Error("Invalid Proxy id\n");
+			proxy = proxy.nextSiblingElement("Proxy");
+			continue;
+		}
+		//proxy type
+		proxyRecord.type = proxy.attribute("type").toStdString();
+		// proxy address
+		QDomElement addrElem = proxy.firstChildElement("Address");
+		proxyRecord.host = addrElem.text().toStdString();
+		// proxy port
+		QDomElement portElem = proxy.firstChildElement("Port");
+		QString port = portElem.text();
+		proxyRecord.port = port.toShort(&ok);
+		if (!ok)
+		{
+			Logger::Error("Invalid Proxy port\n");
+			proxy = proxy.nextSiblingElement("Proxy");
+			continue;
+		}
+		// proxy options
+		QDomElement optElem = proxy.firstChildElement("Options");
+		QString opt = optElem.text();
+		proxyRecord.option = opt.toInt(&ok);
+		if (!ok)
+		{
+			Logger::Error("Invalid Proxy option\n");
+			continue;
+		}
+		// auth section
+		QDomElement authElem = proxy.firstChildElement("Authentication");
+		if (!authElem.isNull())
+		{
+			if (! authElem.hasChildNodes())
+			{
+				// in case if "NTLM + userLogin" = selfclosing auth tag
+				proxyRecord.emptyAuth = true;
+			}
+			else
+			{
+				// username
+				QDomElement userElem = authElem.firstChildElement("Username");
+				proxyRecord.login = userElem.text().toStdString();
+				// pass
+				QDomElement passElem = authElem.firstChildElement("Password");
+				proxyRecord.password =  passElem.text().toStdString();
+			}
+		}
+		else
+		{
+			Logger::Trace("Auth section is empty\n");
+		}
+		Logger::Trace("Found proxy in proxifier config: id=%d\n", proxyRecord.id);
+		proxyList.push_back(proxyRecord);
+
+		// process next proxy element
+		proxy = proxy.nextSiblingElement("Proxy");
+	}
+	return true;
+}
+
+bool Proxifier::UpdateChainList(QDomElement& root)
+{
+	Logger::Trace("Processing Proxifier chain list\n");
+	QDomElement chainListElem = root.firstChildElement("ChainList");
+	if (chainListElem.isNull())
+	{
+		return false;
+	}
+
+	chainList.clear();
+	QDomElement chain = chainListElem.firstChildElement("Chain");
+	while (!chain.isNull())
+	{
+		bool ok;
+		Chain chainRecord;
+
+		// chain id
+		if (!chain.hasAttribute("id"))
+		{
+			Logger::Error("Invalid Chain entry\n");
+			chain = chain.nextSiblingElement("Chain");
+			continue;
+		}
+		QString id = chain.attribute("id");
+		chainRecord.id = id.toInt(&ok);
+		if (!ok)
+		{
+			Logger::Error("Invalid Chain id\n");
+			chain = chain.nextSiblingElement("Chain");
+			continue;
+		}
+		// chain name
+		QDomElement nameElem = chain.firstChildElement("Name");
+		chainRecord.name = nameElem.text().toStdString();
+		//chain proxies
+		QDomElement proxyElem = chain.firstChildElement("Proxy");
+		while (!proxyElem.isNull())
+		{
+		    Logger::Trace("Processing Chain proxy\n");
+			bool isEnabled = false;
+			QString enAttr = proxyElem.attribute("enabled");
+			if (enAttr == "true")
+			{
+				isEnabled = true;
+			}
+			QString idTxt = proxyElem.text();
+			int id = idTxt.toInt();
+			chainRecord.proxies[id] = isEnabled;
+			// proceed to the next chain proxy
+			proxyElem = proxyElem.nextSiblingElement("Proxy");
+		}
+		Logger::Trace("Found chain in proxifier config: id=%d\n", chainRecord.id);
+		chainList.push_back(chainRecord);
+
+		// proceed to the next chain
+		chain = chain.nextSiblingElement("Chain");
+	}
+	return true;
+}
+
+bool Proxifier::UpdateRuleList(QDomElement& root)
+{
+	Logger::Trace("Processing Proxifier rule list\n");
+	QDomElement ruleListElem = root.firstChildElement("RuleList");
+	if (ruleListElem.isNull())
+	{
+		return false;
+	}
+
+	ruleList.clear();
+	QDomElement rule = ruleListElem.firstChildElement("Rule");
+	while (!rule.isNull())
+	{
+		Rule ruleRecord;
+
+		// rule enabled attr
+		if (!rule.hasAttribute("enabled"))
+		{
+			Logger::Error("Invalid Rule entry\n");
+			rule = rule.nextSiblingElement("Rule");
+			continue;
+		}
+		QString enAttr = rule.attribute("enabled");
+		if (enAttr == "true")
+		{
+			ruleRecord.isEnabled = true;
+		}
+		// rule name
+		QDomElement nameElem = rule.firstChildElement("Name");
+		ruleRecord.name = nameElem.text().toStdString();
+		// rule applications
+		QDomElement appsElem = rule.firstChildElement("Applications");
+		if (!appsElem.isNull())
+		{
+			ruleRecord.apps = appsElem.text().toStdString();
+		}
+		// rule targets
+		QDomElement targetsElem = rule.firstChildElement("Targets");
+		if (!targetsElem.isNull())
+		{
+			ruleRecord.targets = targetsElem.text().toStdString();
+		}
+		// rule ports
+		QDomElement portsElem = rule.firstChildElement("Ports");
+		if (!portsElem.isNull())
+		{
+			ruleRecord.ports = targetsElem.text().toStdString();
+		}
+		// rule action
+		QDomElement actionElem = rule.firstChildElement("Action");
+		QString actionTxt = actionElem.attribute("type");
+		if ((actionTxt == "Chain") || (actionTxt == "Proxy"))
+		{
+			QString idTxt = actionElem.text();
+			ruleRecord.id = idTxt.toInt();
+		}
+		ruleRecord.action = actionTxt.toStdString();
+		Logger::Trace("Found rule in proxifier config: name=%s\n", ruleRecord.name.c_str());
+		ruleList.push_back(ruleRecord);
+
+		// proceed to the next rule
+		rule = rule.nextSiblingElement("Rule");
+	}
+
+	return true;
+}