1 files changed, 7 insertions, 3 deletions
diff --git a/Swiften/ScreenSharing/ScreenSharing.h b/Swiften/ScreenSharing/ScreenSharing.h
index b1756b9..5289d3c 100644
--- a/Swiften/ScreenSharing/ScreenSharing.h
+++ b/Swiften/ScreenSharing/ScreenSharing.h
@@ -8,6 +8,7 @@
 
 #include <Swiften/Base/boost_bsignals.h>
 #include <Swiften/Base/IDGenerator.h>
+#include <Swiften/Elements/JinglePayload.h>
 
 #include <boost/shared_ptr.hpp>
 
@@ -16,6 +17,7 @@ namespace Swift {
 	class JingleRawUDPTransportPayload;
 	class UDPSocketFactory;
 	class UDPSocket;
+	class RTPSession;
 
 	class ScreenSharing {
 		public:
@@ -34,24 +36,26 @@ namespace Swift {
 			};
 
 		public:
-			ScreenSharing(boost::shared_ptr<JingleSession> session, UDPSocketFactory* udpSocketFactory);
+			ScreenSharing(boost::shared_ptr<JingleSession> jingleSession, UDPSocketFactory* udpSocketFactory);
 			virtual ~ScreenSharing();
 
 			virtual void cancel() = 0;
+			void stop();
 
 		public:
 			boost::signal<void (SCState)> onStateChange;
 
 		protected:
 			bool addBestCandidate(boost::shared_ptr<JingleRawUDPTransportPayload> transport);
+			void handleSessionTerminateReceived(boost::optional<JinglePayload::Reason> reason);
 
 		protected:
 			IDGenerator idGenerator;
 			boost::shared_ptr<UDPSocket> serverSocket;
 			boost::shared_ptr<UDPSocket> clientSocket;
+			RTPSession* rtpSession;
 
-			boost::shared_ptr<JingleSession> session;
+			boost::shared_ptr<JingleSession> jingleSession;
 			UDPSocketFactory* udpSocketFactory;
-
 	};
 }