diff options
Diffstat (limited to 'Swiften/Examples/NetworkTool/main.cpp')
| -rw-r--r-- | Swiften/Examples/NetworkTool/main.cpp | 81 | 
1 files changed, 81 insertions, 0 deletions
diff --git a/Swiften/Examples/NetworkTool/main.cpp b/Swiften/Examples/NetworkTool/main.cpp new file mode 100644 index 0000000..fa19e14 --- /dev/null +++ b/Swiften/Examples/NetworkTool/main.cpp @@ -0,0 +1,81 @@ +/* + * 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> + +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(); +	} +}  | 
 Swift