summaryrefslogtreecommitdiffstats
blob: f022b9f3f22cf0615504d62e0b1a7cafb649050b (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
/*
 * Copyright (c) 2011 Tobias Markmann
 * Licensed under the simplified BSD license.
 * See Documentation/Licenses/BSD-simplified.txt for more information.
 */

/*
 * Copyright (c) 2013-2014 Isode Limited.
 * All rights reserved.
 * See the COPYING file for more information.
 */

#pragma once

#include <boost/shared_ptr.hpp>
#include <boost/optional/optional.hpp>

#include <Swiften/Base/API.h>
#include <Swiften/Base/Override.h>
#include <Swiften/Jingle/JingleContentID.h>
#include <Swiften/Elements/JingleFileTransferFileInfo.h>
#include <Swiften/FileTransfer/OutgoingFileTransfer.h>
#include <Swiften/FileTransfer/JingleFileTransfer.h>
#include <Swiften/FileTransfer/FileTransferOptions.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 JingleFileTransferFileInfo&,
				const FileTransferOptions&,
				CryptoProvider*);
			virtual ~OutgoingJingleFileTransfer();
			
			virtual void start() SWIFTEN_OVERRIDE;
			virtual void cancel() SWIFTEN_OVERRIDE;

		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;
			virtual void startTransferViaRemoteCandidate() SWIFTEN_OVERRIDE;
			virtual void startTransferViaLocalCandidate() SWIFTEN_OVERRIDE;
			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;
			JingleFileTransferFileInfo fileInfo;
			FileTransferOptions options;
			JingleContentID contentID;
			IncrementalBytestreamHashCalculator* hashCalculator;
			State state;
			bool candidateAcknowledged;

			boost::bsignals::connection processedBytesConnection;
			boost::bsignals::connection transferFinishedConnection;
	};

}