1 files changed, 85 insertions, 0 deletions
diff --git a/Swiften/Examples/NetworkTool/main.cpp b/Swiften/Examples/NetworkTool/main.cpp
new file mode 100644
index 0000000..ace38fd
--- /dev/null
+++ b/Swiften/Examples/NetworkTool/main.cpp
@@ -0,0 +1,85 @@
+/*
+ * Copyright (c) 2011 Remko Tronçon
+ * Licensed under the GNU General Public License v3.
+ * See Documentation/Licenses/GPLv3.txt for more information.
+ */
+
+#include <iostream>
+#include <boost/lexical_cast.hpp>
+
+#include <Swiften/EventLoop/SimpleEventLoop.h>
+#include <Swiften/Network/PlatformNATTraversalWorker.h>
+#include <Swiften/Network/NATTraversalGetPublicIPRequest.h>
+#include <Swiften/Network/NATTraversalForwardPortRequest.h>
+#include <Swiften/Network/NATTraversalRemovePortForwardingRequest.h>
+#include <Swiften/Network/PlatformNetworkEnvironment.h>
+
+using namespace Swift;
+
+SimpleEventLoop eventLoop;
+
+void handleGetPublicIPRequestResponse(const boost::optional<HostAddress>& result) {
+	if (result) {
+		std::cerr << "Result: " << result->toString() << std::endl;;
+	}
+	else {
+		std::cerr << "No result" << std::endl;
+	}
+	eventLoop.stop();
+}
+
+void handleGetForwardPortRequestResponse(const boost::optional<NATPortMapping>& result) {
+	if (result) {
+		std::cerr << "Result: " << result->publicPort << " -> " << result->localPort << std::endl;;
+	}
+	else {
+		std::cerr << "No result" << std::endl;
+	}
+	eventLoop.stop();
+}
+
+void handleRemovePortForwardingRequestResponse(bool result) {
+	if (result) {
+		std::cerr << "Result: OK" << std::endl;
+	}
+	else {
+		std::cerr << "Result: ERROR" << std::endl;
+	}
+	eventLoop.stop();
+}
+
+int main(int argc, char* argv[]) {
+	if (argc < 2) {
+		std::cerr << "Invalid parameters" << std::endl;
+		return -1;
+	}
+
+	PlatformNATTraversalWorker natTraverser(&eventLoop);
+	if (std::string(argv[1]) == "get-public-ip") {
+		boost::shared_ptr<NATTraversalGetPublicIPRequest> query = natTraverser.createGetPublicIPRequest();
+		query->onResult.connect(boost::bind(&handleGetPublicIPRequestResponse, _1));
+		query->run();
+		eventLoop.run();
+	}
+	else if (std::string(argv[1]) == "add-port-forward") {
+		if (argc < 4) {
+			std::cerr << "Invalid parameters" << std::endl;
+		}
+		boost::shared_ptr<NATTraversalForwardPortRequest> query = natTraverser.createForwardPortRequest(boost::lexical_cast<int>(argv[2]), boost::lexical_cast<int>(argv[3]));
+		query->onResult.connect(boost::bind(&handleGetForwardPortRequestResponse, _1));
+		query->run();
+		eventLoop.run();
+	}
+	else if (std::string(argv[1]) == "remove-port-forward") {
+		if (argc < 4) {
+			std::cerr << "Invalid parameters" << std::endl;
+		}
+		boost::shared_ptr<NATTraversalRemovePortForwardingRequest> query = natTraverser.createRemovePortForwardingRequest(boost::lexical_cast<int>(argv[2]), boost::lexical_cast<int>(argv[3]));
+		query->onResult.connect(boost::bind(&handleRemovePortForwardingRequestResponse, _1));
+		query->run();
+		eventLoop.run();
+	}
+	else if (std::string(argv[1]) == "get-local-ip") {
+		std::cout << PlatformNetworkEnvironment().getLocalAddress().toString() << std::endl;
+	}
+}