1 files changed, 101 insertions, 93 deletions
diff --git a/Swiften/FileTransfer/OutgoingJingleFileTransfer.h b/Swiften/FileTransfer/OutgoingJingleFileTransfer.h
index d1f4dc4..6db4e40 100644
--- a/Swiften/FileTransfer/OutgoingJingleFileTransfer.h
+++ b/Swiften/FileTransfer/OutgoingJingleFileTransfer.h
@@ -5,109 +5,117 @@
  */
 
 /*
- * Copyright (c) 2013 Remko Tronçon
- * Licensed under the GNU General Public License.
+ * Copyright (c) 2013-2017 Isode Limited.
+ * All rights reserved.
  * See the COPYING file for more information.
  */
 
 #pragma once
 
-#include <boost/shared_ptr.hpp>
+#include <memory>
+
 #include <boost/optional/optional.hpp>
 
 #include <Swiften/Base/API.h>
-#include <Swiften/Base/Override.h>
-#include <Swiften/Jingle/JingleContentID.h>
-#include <Swiften/Elements/StreamInitiationFileInfo.h>
-#include <Swiften/FileTransfer/OutgoingFileTransfer.h>
-#include <Swiften/FileTransfer/JingleFileTransfer.h>
+#include <Swiften/Elements/JingleFileTransferFileInfo.h>
 #include <Swiften/FileTransfer/FileTransferOptions.h>
+#include <Swiften/FileTransfer/JingleFileTransfer.h>
+#include <Swiften/FileTransfer/OutgoingFileTransfer.h>
+#include <Swiften/Jingle/JingleContentID.h>
+#include <Swiften/Network/Timer.h>
 
 namespace Swift {
-	class ReadBytestream;
-	class IDGenerator;
-	class IncrementalBytestreamHashCalculator;
-	class CryptoProvider;
-	class FileTransferTransporter;
-	class FileTransferTransporterFactory;
-	class TransportSession;
-
-	class SWIFTEN_API OutgoingJingleFileTransfer : public OutgoingFileTransfer, public JingleFileTransfer {
-		public:
-			OutgoingJingleFileTransfer(
-				const JID& to,
-				boost::shared_ptr<JingleSession>,
-				boost::shared_ptr<ReadBytestream>,
-				FileTransferTransporterFactory*,
-				IDGenerator*,
-				const StreamInitiationFileInfo&,
-				const FileTransferOptions&,
-				CryptoProvider*);
-			virtual ~OutgoingJingleFileTransfer();
-			
-			void start();
-			void cancel();
-
-		private:
-			enum State {
-				Initial,
-				GeneratingInitialLocalCandidates,	
-				WaitingForAccept,
-				TryingCandidates,
-				WaitingForPeerProxyActivate,
-				WaitingForLocalProxyActivate,
-				WaitingForCandidateAcknowledge,
-				FallbackRequested,
-				Transferring,
-				Finished
-			};
-
-			virtual void handleSessionAcceptReceived(const JingleContentID&, boost::shared_ptr<JingleDescription>, boost::shared_ptr<JingleTransportPayload>) SWIFTEN_OVERRIDE;
-			virtual void handleSessionTerminateReceived(boost::optional<JinglePayload::Reason> reason) SWIFTEN_OVERRIDE;
-			virtual void handleTransportAcceptReceived(const JingleContentID&, boost::shared_ptr<JingleTransportPayload>) SWIFTEN_OVERRIDE;
-			void startTransferViaRemoteCandidate();
-			void startTransferViaLocalCandidate();
-			void startTransferringIfCandidateAcknowledged();
-
-			virtual void handleLocalTransportCandidatesGenerated(const std::string& s5bSessionID, const std::vector<JingleS5BTransportPayload::Candidate>&) SWIFTEN_OVERRIDE;
-			virtual void handleTransportInfoAcknowledged(const std::string& id) SWIFTEN_OVERRIDE;
-
-			virtual JingleContentID getContentID() const SWIFTEN_OVERRIDE;
-
-			virtual void terminate(JinglePayload::Reason::Type reason) SWIFTEN_OVERRIDE;
-
-			virtual void fallback() SWIFTEN_OVERRIDE;
-			void handleTransferFinished(boost::optional<FileTransferError>);
-
-			void sendSessionInfoHash();
-
-			virtual void startTransferring(boost::shared_ptr<TransportSession>) SWIFTEN_OVERRIDE;
-
-			virtual bool hasPriorityOnCandidateTie() const SWIFTEN_OVERRIDE;
-			virtual bool isWaitingForPeerProxyActivate() const SWIFTEN_OVERRIDE;
-			virtual bool isWaitingForLocalProxyActivate() const SWIFTEN_OVERRIDE;
-			virtual bool isTryingCandidates() const SWIFTEN_OVERRIDE;
-			virtual boost::shared_ptr<TransportSession> createLocalCandidateSession() SWIFTEN_OVERRIDE;
-			virtual boost::shared_ptr<TransportSession> createRemoteCandidateSession() SWIFTEN_OVERRIDE;
-
-			void stopAll();
-			void setState(State state);
-			void setFinishedState(FileTransfer::State::Type, const boost::optional<FileTransferError>& error);
-
-			static FileTransfer::State::Type getExternalState(State state);
-
-		private:
-			IDGenerator* idGenerator;
-			boost::shared_ptr<ReadBytestream> stream;
-			StreamInitiationFileInfo fileInfo;
-			FileTransferOptions options;
-			JingleContentID contentID;
-			IncrementalBytestreamHashCalculator* hashCalculator;
-			State state;
-			bool candidateAcknowledged;
-
-			boost::bsignals::connection processedBytesConnection;
-			boost::bsignals::connection transferFinishedConnection;
-	};
+    class CryptoProvider;
+    class FileTransferTransporterFactory;
+    class IDGenerator;
+    class IncrementalBytestreamHashCalculator;
+    class ReadBytestream;
+    class TimerFactory;
+    class TransportSession;
+
+    class SWIFTEN_API OutgoingJingleFileTransfer : public OutgoingFileTransfer, public JingleFileTransfer {
+        public:
+            OutgoingJingleFileTransfer(
+                const JID& to,
+                std::shared_ptr<JingleSession>,
+                std::shared_ptr<ReadBytestream>,
+                FileTransferTransporterFactory*,
+                TimerFactory*,
+                IDGenerator*,
+                const JingleFileTransferFileInfo&,
+                const FileTransferOptions&,
+                CryptoProvider*);
+            virtual ~OutgoingJingleFileTransfer() override;
+
+            virtual void start() override;
+            virtual void cancel() override;
+
+        private:
+            enum State {
+                Initial,
+                GeneratingInitialLocalCandidates,
+                WaitingForAccept,
+                TryingCandidates,
+                WaitingForPeerProxyActivate,
+                WaitingForLocalProxyActivate,
+                WaitingForCandidateAcknowledge,
+                FallbackRequested,
+                Transferring,
+                WaitForTermination,
+                Finished
+            };
+
+            virtual void handleSessionAcceptReceived(const JingleContentID&, std::shared_ptr<JingleDescription>, std::shared_ptr<JingleTransportPayload>) override;
+            virtual void handleSessionTerminateReceived(boost::optional<JinglePayload::Reason> reason) override;
+            virtual void handleTransportAcceptReceived(const JingleContentID&, std::shared_ptr<JingleTransportPayload>) override;
+            virtual void handleTransportRejectReceived(const JingleContentID &, std::shared_ptr<JingleTransportPayload>) override;
+            virtual void startTransferViaRemoteCandidate() override;
+            virtual void startTransferViaLocalCandidate() override;
+            void startTransferringIfCandidateAcknowledged();
+
+            virtual void handleLocalTransportCandidatesGenerated(const std::string& s5bSessionID, const std::vector<JingleS5BTransportPayload::Candidate>&, const std::string& dstAddr) override;
+            virtual void handleTransportInfoAcknowledged(const std::string& id) override;
+
+            virtual JingleContentID getContentID() const override;
+
+            virtual void terminate(JinglePayload::Reason::Type reason) override;
+
+            virtual void fallback() override;
+            void handleTransferFinished(boost::optional<FileTransferError>);
+
+            void sendSessionInfoHash();
+
+            virtual void startTransferring(std::shared_ptr<TransportSession>) override;
+
+            virtual bool hasPriorityOnCandidateTie() const override;
+            virtual bool isWaitingForPeerProxyActivate() const override;
+            virtual bool isWaitingForLocalProxyActivate() const override;
+            virtual bool isTryingCandidates() const override;
+            virtual std::shared_ptr<TransportSession> createLocalCandidateSession() override;
+            virtual std::shared_ptr<TransportSession> createRemoteCandidateSession() override;
+
+            void handleWaitForRemoteTerminationTimeout();
+
+            void stopAll();
+            void setInternalState(State state);
+            void setFinishedState(FileTransfer::State::Type, const boost::optional<FileTransferError>& error);
+
+            static FileTransfer::State::Type getExternalState(State state);
+
+        private:
+            IDGenerator* idGenerator;
+            std::shared_ptr<ReadBytestream> stream;
+            JingleFileTransferFileInfo fileInfo;
+            FileTransferOptions options;
+            JingleContentID contentID;
+            IncrementalBytestreamHashCalculator* hashCalculator;
+            State state;
+            bool candidateAcknowledged;
+
+            Timer::ref waitForRemoteTermination;
+
+            boost::signals2::connection processedBytesConnection;
+            boost::signals2::connection transferFinishedConnection;
+    };
 
 }