1 files changed, 110 insertions, 3 deletions
diff --git a/Swiften/FileTransfer/IncomingJingleFileTransfer.h b/Swiften/FileTransfer/IncomingJingleFileTransfer.h
index d69449e..4ae0bfb 100644
--- a/Swiften/FileTransfer/IncomingJingleFileTransfer.h
+++ b/Swiften/FileTransfer/IncomingJingleFileTransfer.h
@@ -7,21 +7,128 @@
 #pragma once
 
 #include <boost/shared_ptr.hpp>
+#include <boost/cstdint.hpp>
 
-#include <Swiften/Jingle/IncomingJingleSession.h>
+#include <Swiften/Base/IDGenerator.h>
+#include <Swiften/Network/Timer.h>
+#include <Swiften/Jingle/JingleSession.h>
+#include <Swiften/Jingle/JingleContentID.h>
 #include <Swiften/FileTransfer/IncomingFileTransfer.h>
+#include <Swiften/FileTransfer/JingleTransport.h>
+#include <Swiften/FileTransfer/JingleIncomingIBBTransport.h>
+#include <Swiften/FileTransfer/SOCKS5BytestreamClientSession.h>
+#include <Swiften/FileTransfer/SOCKS5BytestreamServerSession.h>
+#include <Swiften/Elements/JingleContentPayload.h>
+#include <Swiften/Elements/JingleFileTransferDescription.h>
+#include <Swiften/Elements/JingleIBBTransportPayload.h>
+#include <Swiften/Elements/JingleS5BTransportPayload.h>
+#include <Swiften/Elements/S5BProxyRequest.h>
+#include <Swiften/Elements/ErrorPayload.h>
 
 namespace Swift {
+	class IQRouter;
+	class RemoteJingleTransportCandidateSelectorFactory;
+	class LocalJingleTransportCandidateGeneratorFactory;
+	class RemoteJingleTransportCandidateSelector;
+	class LocalJingleTransportCandidateGenerator;
+	class SOCKS5BytestreamRegistry;
+	class SOCKS5BytestreamProxy;
+	class IncrementalBytestreamHashCalculator;
+
 	class IncomingJingleFileTransfer : public IncomingFileTransfer {
 		public:
 			typedef boost::shared_ptr<IncomingJingleFileTransfer> ref;
+			enum State {
+				Initial,
+				CreatingInitialTransports,
+				NegotiatingTransport,
+				Transferring,
+				WaitingForFallbackOrTerminate,
+				Terminated
+			};
 
-			IncomingJingleFileTransfer(IncomingJingleSession::ref session);
+			IncomingJingleFileTransfer(
+					const JID& recipient,
+					JingleSession::ref,
+					JingleContentPayload::ref content,
+					RemoteJingleTransportCandidateSelectorFactory*,
+					LocalJingleTransportCandidateGeneratorFactory*,
+					IQRouter* router,
+					SOCKS5BytestreamRegistry* bytestreamRegistry,
+					SOCKS5BytestreamProxy* bytestreamProxy,
+					TimerFactory*);
+			~IncomingJingleFileTransfer();
 
 			virtual void accept(WriteBytestream::ref);
+			virtual const JID& getSender() const;
+			virtual const JID& getRecipient() const;
+			void cancel();
+
+		private:
+			void handleSessionTerminateReceived(boost::optional<JinglePayload::Reason>);
+			void handleSessionInfoReceived(JinglePayload::ref);
+			void handleTransportReplaceReceived(const JingleContentID&, JingleTransportPayload::ref);
+			void handleTransportInfoReceived(const JingleContentID&, JingleTransportPayload::ref);
+			void handleLocalTransportCandidatesGenerated(JingleTransportPayload::ref candidates);
+			void handleRemoteTransportCandidateSelectFinished(JingleTransportPayload::ref candidate);
+			void setActiveTransport(JingleTransport::ref transport);
+			void handleTransportDataReceived(const std::vector<unsigned char>& data);
+			void handleWriteStreamDataReceived(const std::vector<unsigned char>& data);
+			void stopActiveTransport();
+			void checkCandidateSelected();
+			JingleIncomingIBBTransport::ref createIBBTransport(JingleIBBTransportPayload::ref ibbTransport);
+			JingleContentID getContentID() const;
+			void checkIfAllDataReceived();
+			bool verifyReceviedData();
+			void finishOffTransfer();
+			void handleTransferFinished(boost::optional<FileTransferError>);
+
+		private:
+			typedef std::map<std::string, JingleS5BTransportPayload::Candidate> CandidateMap;
+
+		private:
+			void activateProxySession(const JID &proxy);
+			void handleActivateProxySessionResult(boost::shared_ptr<S5BProxyRequest> request, ErrorPayload::ref error);
+			void proxySessionReady(const JID& proxy, bool error);
+
+		private:
+			void useOurCandidateChoiceForTransfer(JingleS5BTransportPayload::Candidate candidate);
+			void useTheirCandidateChoiceForTransfer(JingleS5BTransportPayload::Candidate candidate);
+			void decideOnUsedTransport();
+			void fillCandidateMap(CandidateMap& map, JingleS5BTransportPayload::ref s5bPayload);
 
 		private:
-			IncomingJingleSession::ref session;
+			JID ourJID;
+			JingleSession::ref session;
+			IQRouter* router;
+			TimerFactory* timerFactory;
+			JingleContentPayload::ref initialContent;
+			State state;
+			JingleFileTransferDescription::ref description;
 			WriteBytestream::ref stream;
+			boost::uintmax_t receivedBytes;
+			IncrementalBytestreamHashCalculator* hashCalculator;
+			Timer::ref waitOnHashTimer;
+			IDGenerator idGenerator;
+
+			RemoteJingleTransportCandidateSelector* candidateSelector;
+			LocalJingleTransportCandidateGenerator* candidateGenerator;
+			SOCKS5BytestreamRegistry* s5bRegistry;
+			SOCKS5BytestreamProxy* s5bProxy;
+			bool remoteTransportCandidateSelectFinished;
+			JingleTransportPayload::ref selectedRemoteTransportCandidate;
+			bool localTransportCandidateSelectFinished;
+			JingleTransportPayload::ref selectedLocalTransportCandidate;
+
+			JingleS5BTransportPayload::ref ourCandidate;
+			JingleS5BTransportPayload::ref theirCandidate;
+			CandidateMap ourCandidates;
+			CandidateMap theirCandidates;
+			SOCKS5BytestreamClientSession::ref clientSession;
+			std::string s5bDestination;
+			std::string s5bSessionID;
+			SOCKS5BytestreamServerSession* serverSession;
+
+			JingleTransport::ref activeTransport;
 	};
 }