1 files changed, 45 insertions, 0 deletions
diff --git a/Swiften/FileTransfer/ConnectivityManager.h b/Swiften/FileTransfer/ConnectivityManager.h
new file mode 100644
index 0000000..c094c02
--- /dev/null
+++ b/Swiften/FileTransfer/ConnectivityManager.h
@@ -0,0 +1,45 @@
+/*
+ * Copyright (c) 2011 Tobias Markmann
+ * Licensed under the simplified BSD license.
+ * See Documentation/Licenses/BSD-simplified.txt for more information.
+ */
+
+#pragma once
+
+#include <vector>
+#include <set>
+
+#include <boost/optional.hpp>
+
+#include <Swiften/Network/HostAddressPort.h>
+#include <Swiften/Network/NATTraverser.h>
+#include <Swiften/Network/NATTraversalForwardPortRequest.h>
+#include <Swiften/Network/NATPortMapping.h>
+
+namespace Swift {
+
+class NATTraverser;
+
+class ConnectivityManager {
+public:
+	ConnectivityManager(NATTraverser*);
+	~ConnectivityManager();
+public:
+	void addListeningPort(int port);
+	void removeListeningPort(int port);
+
+	std::vector<HostAddressPort> getHostAddressPorts() const;
+	std::vector<HostAddressPort> getAssistedHostAddressPorts() const;
+
+private:
+	void natTraversalGetPublicIPResult(boost::optional<HostAddress> address);
+	void natTraversalForwardPortResult(boost::optional<NATPortMapping> mapping);
+
+private:
+	NATTraverser* natTraversalWorker;
+
+	std::set<int> ports;
+	boost::optional<HostAddress> publicAddress;
+};
+
+}