1 files changed, 6 insertions, 5 deletions
diff --git a/Swiften/FileTransfer/ConnectivityManager.h b/Swiften/FileTransfer/ConnectivityManager.h
index c094c02..c70cb5a 100644
--- a/Swiften/FileTransfer/ConnectivityManager.h
+++ b/Swiften/FileTransfer/ConnectivityManager.h
@@ -25,11 +25,11 @@ public:
 	ConnectivityManager(NATTraverser*);
 	~ConnectivityManager();
 public:
-	void addListeningPort(int port);
-	void removeListeningPort(int port);
+	void addListeningPort(int port, NATPortMapping::Protocol protocol);
+	void removeListeningPort(int port, NATPortMapping::Protocol protocol);
 
-	std::vector<HostAddressPort> getHostAddressPorts() const;
-	std::vector<HostAddressPort> getAssistedHostAddressPorts() const;
+	std::vector<HostAddressPort> getHostAddressPorts(NATPortMapping::Protocol protocol) const;
+	std::vector<HostAddressPort> getAssistedHostAddressPorts(NATPortMapping::Protocol protocol) const;
 
 private:
 	void natTraversalGetPublicIPResult(boost::optional<HostAddress> address);
@@ -38,7 +38,8 @@ private:
 private:
 	NATTraverser* natTraversalWorker;
 
-	std::set<int> ports;
+	std::set<int> udpPorts;
+	std::set<int> tcpPorts;
 	boost::optional<HostAddress> publicAddress;
 };