mirror of
https://github.com/RetroShare/RetroShare.git
synced 2024-10-01 02:35:48 -04:00
d43a131c04
- servers now compute the map in a separate thread - CRC32Maps are kept in cache for 30 mins - CRC32Maps requests cannot be used to overflood a server anymore since their number is limited. - Transfer modules now send keep alive packets to maintain tunnels when asking for a CRC32Map git-svn-id: http://svn.code.sf.net/p/retroshare/code/trunk@4661 b45a01b8-16f6-495d-af2f-9b41ad6348cc
145 lines
4.8 KiB
C++
145 lines
4.8 KiB
C++
/*
|
|
* libretroshare/src/ft/ ftfilecreator.h
|
|
*
|
|
* File Transfer for RetroShare.
|
|
*
|
|
* Copyright 2008 by Robert Fernie.
|
|
*
|
|
* This library is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU Library General Public
|
|
* License Version 2 as published by the Free Software Foundation.
|
|
*
|
|
* This library 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
|
|
* Library General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Library General Public
|
|
* License along with this library; if not, write to the Free Software
|
|
* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
|
|
* USA.
|
|
*
|
|
* Please report all bugs and problems to "retroshare@lunamutt.com".
|
|
*
|
|
*/
|
|
|
|
#ifndef FT_FILE_CREATOR_HEADER
|
|
#define FT_FILE_CREATOR_HEADER
|
|
|
|
/*
|
|
* ftFileCreator
|
|
*
|
|
* TODO: Serialiser Load / Save.
|
|
*
|
|
*/
|
|
#include "ftfileprovider.h"
|
|
#include "ftchunkmap.h"
|
|
#include <map>
|
|
|
|
class ZeroInitCounter
|
|
{
|
|
public:
|
|
ZeroInitCounter(): cnt(0) {}
|
|
uint32_t cnt ;
|
|
};
|
|
|
|
class ftFileCreator: public ftFileProvider
|
|
{
|
|
public:
|
|
|
|
ftFileCreator(const std::string& savepath, uint64_t size, const std::string& hash,bool assume_availability);
|
|
|
|
~ftFileCreator();
|
|
|
|
/* overloaded from FileProvider */
|
|
virtual bool getFileData(const std::string& peer_id,uint64_t offset, uint32_t &chunk_size, void *data);
|
|
bool finished() ;
|
|
uint64_t getRecvd();
|
|
|
|
/// (temporarily) close the file, to save file descriptors.
|
|
void closeFile() ;
|
|
|
|
void getChunkMap(FileChunksInfo& info) ;
|
|
|
|
void setChunkStrategy(FileChunksInfo::ChunkStrategy s) ;
|
|
FileChunksInfo::ChunkStrategy getChunkStrategy() ;
|
|
|
|
// Computes a sha1sum of the partial file, to check that the data is overall consistent.
|
|
// This function is not mutexed. This is a bit dangerous, but otherwise we might stuck the GUI for a
|
|
// long time. Therefore, we must pay attention not to call this function
|
|
// at a time file_name nor hash can be modified, which is quite easy.
|
|
|
|
bool hashReceivedData(std::string& hash) ;
|
|
|
|
// Checks the CRC32 of all chunks against the given CRC32 map. Re-flag the bad chunks as being void.
|
|
// bad_chunks: count of achieved chunks that don't match the CRC
|
|
// incomplete_chunks: count of any bad or not yet downloaded chunk
|
|
//
|
|
bool crossCheckChunkMap(const CRC32Map& ref,uint32_t& bad_chunks,uint32_t& incomplete_chunks) ;
|
|
/*
|
|
* creation functions for FileCreator
|
|
*/
|
|
|
|
// Gets a new variable-sized chunk of size "size_hint" from the given peer id. The returned size, "size" is
|
|
// at most equal to size_hint. chunk_map_needed is set if
|
|
// - no chunkmap info is available. In such a case, the chunk info is irrelevant and false is returned.
|
|
// - the chunk info is too old. In tis case, true is returned, and the chunks info can be used.
|
|
//
|
|
bool getMissingChunk(const std::string& peer_id,uint32_t size_hint,uint64_t& offset, uint32_t& size,bool& is_chunk_map_too_old);
|
|
|
|
// Takes care of purging any inactive chunks. This should be called regularly, because some peers may disconnect
|
|
// and let inactive chunks not finished.
|
|
//
|
|
void removeInactiveChunks() ;
|
|
|
|
// removes the designated file source from the chunkmap.
|
|
void removeFileSource(const std::string& peer_id) ;
|
|
|
|
// Returns resets the time stamp of the last data receive.
|
|
time_t lastRecvTimeStamp() ;
|
|
void resetRecvTimeStamp() ;
|
|
time_t creationTimeStamp() ;
|
|
|
|
// actually store data in the file, and update chunks info
|
|
//
|
|
bool addFileData(uint64_t offset, uint32_t chunk_size, void *data);
|
|
|
|
// Load/save the availability map for the file being downloaded, in a compact/compressed form.
|
|
// This is used for
|
|
// - loading and saving info about the current transfers
|
|
// - getting info about current chunks for the GUI
|
|
// - sending availability info to the peers for which we also are a source
|
|
//
|
|
virtual void getAvailabilityMap(CompressedChunkMap& cmap) ;
|
|
void setAvailabilityMap(const CompressedChunkMap& cmap) ;
|
|
|
|
// This is called when receiving the availability map from a source peer, for the file being handled.
|
|
//
|
|
void setSourceMap(const std::string& peer_id,const CompressedChunkMap& map) ;
|
|
|
|
protected:
|
|
|
|
virtual int locked_initializeFileAttrs();
|
|
|
|
private:
|
|
|
|
bool locked_printChunkMap();
|
|
int locked_notifyReceived(uint64_t offset, uint32_t chunk_size);
|
|
/*
|
|
* structure to track missing chunks
|
|
*/
|
|
|
|
uint64_t mStart;
|
|
uint64_t mEnd;
|
|
|
|
std::map<uint64_t, ftChunk> mChunks;
|
|
std::map<std::string,ZeroInitCounter> mChunksPerPeer ;
|
|
|
|
ChunkMap chunkMap ;
|
|
|
|
time_t _last_recv_time_t ; /// last time stamp when data was received. Used for queue control.
|
|
time_t _creation_time ; /// time at which the file creator was created. Used to spot long-inactive transfers.
|
|
};
|
|
|
|
#endif // FT_FILE_CREATOR_HEADER
|