1 files changed, 86 insertions, 0 deletions
diff --git a/Swiften/Network/MiniUPnPInterface.cpp b/Swiften/Network/MiniUPnPInterface.cpp
new file mode 100644
index 0000000..f6e3b5d
--- /dev/null
+++ b/Swiften/Network/MiniUPnPInterface.cpp
@@ -0,0 +1,86 @@
+/*
+ * Copyright (c) 2011 Tobias Markmann
+ * Licensed under the simplified BSD license.
+ * See Documentation/Licenses/BSD-simplified.txt for more information.
+ */
+
+#include <Swiften/Network/MiniUPnPInterface.h>
+
+#include <upnpcommands.h>
+#include <upnperrors.h>
+#include <boost/lexical_cast.hpp>
+
+#include <Swiften/Base/Log.h>
+
+namespace Swift {
+
+MiniUPnPInterface::MiniUPnPInterface() : isValid(false) {
+	int error = 0;
+	deviceList = upnpDiscover(1500 /* timeout in ms */, 0, 0, 0, 0 /* do IPv6? */, &error);
+	if (!deviceList) {
+		return;
+	}
+
+	char lanAddress[64];
+	if (!UPNP_GetValidIGD(deviceList, &urls, &data, lanAddress, sizeof(lanAddress))) {
+		return;
+	}
+	localAddress = std::string(lanAddress);
+	isValid = true;
+}
+
+MiniUPnPInterface::~MiniUPnPInterface() {
+	if (isValid) {
+		FreeUPNPUrls(&urls);
+	}
+	freeUPNPDevlist(deviceList);
+}
+
+boost::optional<HostAddress> MiniUPnPInterface::getPublicIP() {
+	if (!isValid) {
+		return boost::optional<HostAddress>();
+	}
+	char externalIPAddress[40];
+	int ret = UPNP_GetExternalIPAddress(urls.controlURL, data.first.servicetype, externalIPAddress);
+	if (ret != UPNPCOMMAND_SUCCESS) {
+		return boost::optional<HostAddress>();
+	}
+	else {
+		return HostAddress(std::string(externalIPAddress));
+	}
+}
+
+boost::optional<NATPortMapping> MiniUPnPInterface::addPortForward(int actualLocalPort, int actualPublicPort) {
+	if (!isValid) {
+		return boost::optional<NATPortMapping>();
+	}
+
+	NATPortMapping mapping(actualLocalPort, actualPublicPort, NATPortMapping::TCP);
+
+	std::string publicPort = boost::lexical_cast<std::string>(mapping.publicPort);
+	std::string localPort = boost::lexical_cast<std::string>(mapping.localPort);
+	std::string leaseSeconds = boost::lexical_cast<std::string>(mapping.leaseInSeconds);
+
+	int ret = UPNP_AddPortMapping(urls.controlURL, data.first.servicetype, publicPort.c_str(), localPort.c_str(), localAddress.c_str(), 0, mapping.protocol == NATPortMapping::TCP ? "TCP" : "UDP", 0, leaseSeconds.c_str());
+	if (ret == UPNPCOMMAND_SUCCESS) {
+		return mapping;
+	}
+	else {
+		return boost::optional<NATPortMapping>();
+	}
+}
+
+bool MiniUPnPInterface::removePortForward(const NATPortMapping& mapping) {
+	if (!isValid) {
+		return false;
+	}
+
+	std::string publicPort = boost::lexical_cast<std::string>(mapping.publicPort);
+	std::string localPort = boost::lexical_cast<std::string>(mapping.localPort);
+	std::string leaseSeconds = boost::lexical_cast<std::string>(mapping.leaseInSeconds);
+
+	int ret = UPNP_DeletePortMapping(urls.controlURL, data.first.servicetype, publicPort.c_str(), mapping.protocol == NATPortMapping::TCP ? "TCP" : "UDP", 0);
+	return ret == UPNPCOMMAND_SUCCESS;
+}
+
+}