Updated to final KDE3 ktorrent release (2.2.6)

git-svn-id: svn://anonsvn.kde.org/home/kde/branches/trinity/applications/ktorrent@1077377 283d02a7-25f6-0310-bc7c-ecb5cbfe19da

1 files changed, 207 insertions, 0 deletions

diff --git a/libktorrent/torrent/chunkdownload.h b/libktorrent/torrent/chunkdownload.h
new file mode 100644
index 0000000..4119a5b
--- /dev/null
+++ b/libktorrent/torrent/chunkdownload.h
@@ -0,0 +1,207 @@
+/***************************************************************************
+ *   Copyright (C) 2005 by Joris Guisson                                   *
+ *   joris.guisson@gmail.com                                               *
+ *                                                                         *
+ *   This program is free software; you can redistribute it and/or modify  *
+ *   it under the terms of the GNU General Public License as published by  *
+ *   the Free Software Foundation; either version 2 of the License, or     *
+ *   (at your option) any later version.                                   *
+ *                                                                         *
+ *   This program is distributed in the hope that it will be useful,       *
+ *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
+ *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
+ *   GNU General Public License for more details.                          *
+ *                                                                         *
+ *   You should have received a copy of the GNU General Public License     *
+ *   along with this program; if not, write to the                         *
+ *   Free Software Foundation, Inc.,                                       *
+ *   51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.             *
+ ***************************************************************************/
+#ifndef BTCHUNKDOWNLOAD_H
+#define BTCHUNKDOWNLOAD_H
+
+#include <set>
+#include <qobject.h>
+#include <qptrlist.h>
+#include <util/timer.h>
+#include <util/ptrmap.h>
+#include <util/sha1hashgen.h>
+#include <interfaces/chunkdownloadinterface.h>
+#include <util/bitset.h>
+#include "globals.h"
+#include "peerid.h"
+
+
+namespace bt
+{
+	
+	class File;
+	class Chunk;
+	class Piece;
+	class Peer;
+	class Request;
+	class PeerDownloader;
+	class DownloadStatus;
+	
+	struct ChunkDownloadHeader
+	{
+		Uint32 index;
+		Uint32 num_bits;
+		Uint32 buffered;
+	};
+	
+	
+	
+	
+	/**
+	 * @author Joris Guisson
+	 * @brief Handles the download off one Chunk off a Peer
+	 * 
+	 * This class handles the download of one Chunk.
+	*/
+	class ChunkDownload : public QObject,public kt::ChunkDownloadInterface 
+	{
+		Q_OBJECT
+	public:
+		/**
+		 * Constructor, set the chunk and the PeerManager.
+		 * @param chunk The Chunk
+		 */
+		ChunkDownload(Chunk* chunk);
+		
+		virtual ~ChunkDownload();
+
+		/// Get the chunk
+		Chunk* getChunk() {return chunk;}
+	
+		/// Get the total number of pieces
+		Uint32 getTotalPieces() const {return num;}
+		
+		/// Get the number of pieces downloaded
+		Uint32 getPiecesDownloaded() const {return num_downloaded;}
+
+		/// Get the number of bytes downloaded.
+		Uint32 bytesDownloaded() const;
+		
+		/// Get the index of the chunk
+		Uint32 getChunkIndex() const;
+		
+		/// Get the current peer
+		const Peer* getCurrentPeer() const;
+		
+		/// Get the PeerID of the current peer
+		QString getCurrentPeerID() const;
+		
+		/// Get the download speed
+		Uint32 getDownloadSpeed() const;
+
+		/// Get download stats
+		void getStats(Stats & s);
+		
+		/// See if a chunkdownload is idle (i.e. has no downloaders)
+		bool isIdle() const {return pdown.count() == 0;}
+		
+		/**
+		 * A Piece has arived.
+		 * @param p The Piece
+		 * @param ok Wether or not the piece was needed
+		 * @return true If Chunk is complete
+		 */
+		bool piece(const Piece & p,bool & ok);	
+		
+		/**
+		 * Assign the downloader to download from.
+		 * @param pd The downloader
+		 * @return true if the peer was asigned, false if not
+		 */
+		bool assignPeer(PeerDownloader* pd);
+		
+		Uint32 getNumDownloaders() const {return pdown.count();}
+
+		/**
+		 * A Peer has been killed. We need to remove it's
+		 * PeerDownloader.
+		 * @param pd The PeerDownloader
+		 */
+		void peerKilled(PeerDownloader* pd);
+
+		/**
+		 * Save to a File
+		 * @param file The File
+		 */
+		void save(File & file);
+		
+		/**
+		 * Load from a File
+		 * @param file The File
+		 */
+		bool load(File & file,ChunkDownloadHeader & hdr);
+
+		/**
+		 * Cancel all requests.
+		 */
+		void cancelAll();
+
+		/**
+		 * When a Chunk is downloaded, this function checks if all
+		 * pieces are delivered by the same peer and if so sets
+		 * that peers' ID.
+		 * @param pid The peers' ID (!= PeerID)
+		 * @return true if there is only one downloader
+		 */
+		bool getOnlyDownloader(Uint32 & pid);
+		
+		/// See if a PeerDownloader is assigned to this chunk
+		bool containsPeer(PeerDownloader *pd) {return pdown.contains(pd);} 
+		
+		/// See if the download is choked (i.e. all downloaders are choked)
+		bool isChoked() const;
+		
+		/// Release all PD's and clear the requested chunks
+		void releaseAllPDs();
+		
+		/// Send requests to peers
+		void update();
+		
+		/// See if this CD hasn't been active in the last update
+		bool needsToBeUpdated() const {return timer.getElapsedSinceUpdate() > 60 * 1000;}
+		
+		/// Get the SHA1 hash of the downloaded chunk
+		SHA1Hash getHash() const {return hash_gen.get();}
+		
+		/// Are we using the continous hashing feature for this chunk
+		bool usingContinuousHashing() const;
+
+	private slots:
+		void sendRequests(PeerDownloader* pd);
+		void sendCancels(PeerDownloader* pd);
+		void endgameCancel(const Piece & p);
+		void onTimeout(const Request & r);
+		void onRejected(const Request & r);
+		
+	private:
+		void notDownloaded(const Request & r,bool reject);
+		void updateHash();
+		
+	private:		
+		BitSet pieces;
+		QValueList<Uint32> piece_queue;
+		Chunk* chunk;
+		Uint32 num;
+		Uint32 num_downloaded;
+		Uint32 last_size;
+		Timer timer;
+		QPtrList<PeerDownloader> pdown;
+		PtrMap<Uint32,DownloadStatus> dstatus;
+		std::set<Uint32> piece_providers;
+		
+
+		SHA1HashGen hash_gen;
+		Uint32 num_pieces_in_hash;
+
+		friend File & operator << (File & out,const ChunkDownload & cd);
+		friend File & operator >> (File & in,ChunkDownload & cd);
+	};
+}
+
+#endif