mirror of
https://github.com/RetroShare/RetroShare.git
synced 2024-10-01 02:35:48 -04:00
Completed most of ftserver class.
Also changes to make everything fit together. changes to Interface to support srcId list. git-svn-id: http://svn.code.sf.net/p/retroshare/code/trunk@675 b45a01b8-16f6-495d-af2f-9b41ad6348cc
This commit is contained in:
parent
d7e1a90c28
commit
189dcbb482
@ -7,13 +7,17 @@ RS_TOP_DIR = ..
|
|||||||
include $(RS_TOP_DIR)/scripts/config.mk
|
include $(RS_TOP_DIR)/scripts/config.mk
|
||||||
###############################################################
|
###############################################################
|
||||||
|
|
||||||
RSOBJ = ftdata.o ftfileprovider.o ftfilecreator.o ftextralist.o ftdatamultiplex.o ftfilesearch.o fttransfermodule.o
|
RSOBJ = ftdata.o ftfileprovider.o ftfilecreator.o ftextralist.o ftdatamultiplex.o ftfilesearch.o fttransfermodule.o ftdbase.o ftserver.o
|
||||||
|
|
||||||
|
#ftcontroller.o pqitestor.o
|
||||||
|
|
||||||
#ftserver.o
|
|
||||||
|
|
||||||
TESTOBJ = ftfileprovidertest.o ftfilecreatortest.o ftextralisttest.o ftdataplextest.o
|
TESTOBJ = ftfileprovidertest.o ftfilecreatortest.o ftextralisttest.o ftdataplextest.o
|
||||||
|
#ftserver1test.o
|
||||||
|
|
||||||
|
|
||||||
TESTS = ftfileprovidertest ftfilecreatortest ftextralisttest ftdataplextest
|
TESTS = ftfileprovidertest ftfilecreatortest ftextralisttest ftdataplextest
|
||||||
|
#ftserver1test
|
||||||
|
|
||||||
all: librs tests
|
all: librs tests
|
||||||
|
|
||||||
@ -29,6 +33,9 @@ ftextralisttest : ftextralisttest.o
|
|||||||
ftdataplextest : ftdataplextest.o
|
ftdataplextest : ftdataplextest.o
|
||||||
$(CC) $(CFLAGS) -o ftdataplextest ftdataplextest.o $(LIBS)
|
$(CC) $(CFLAGS) -o ftdataplextest ftdataplextest.o $(LIBS)
|
||||||
|
|
||||||
|
ftserver1test : ftserver1test.o
|
||||||
|
$(CC) $(CFLAGS) -o ftserver1test ftserver1test.o $(LIBS)
|
||||||
|
|
||||||
###############################################################
|
###############################################################
|
||||||
include $(RS_TOP_DIR)/scripts/rules.mk
|
include $(RS_TOP_DIR)/scripts/rules.mk
|
||||||
###############################################################
|
###############################################################
|
||||||
|
@ -23,9 +23,6 @@
|
|||||||
*
|
*
|
||||||
*/
|
*/
|
||||||
|
|
||||||
#ifndef FT_CONTROLLER_HEADER
|
|
||||||
#define FT_CONTROLLER_HEADER
|
|
||||||
|
|
||||||
/*
|
/*
|
||||||
* ftController
|
* ftController
|
||||||
*
|
*
|
||||||
@ -38,20 +35,34 @@
|
|||||||
*
|
*
|
||||||
*/
|
*/
|
||||||
|
|
||||||
class ftController: public CacheTransfer, public RsThread, public pqiMonitor, public p3Config
|
#include "ft/ftcontroller.h"
|
||||||
|
|
||||||
|
#include "ft/ftfilecreator.h"
|
||||||
|
#include "ft/fttransfermodule.h"
|
||||||
|
#include "ft/ftsearch.h"
|
||||||
|
#include "ft/ftdatamultiplex.h"
|
||||||
|
|
||||||
|
#include "util/rsdir.h"
|
||||||
|
|
||||||
|
#include "pqi/p3connmgr.h"
|
||||||
|
|
||||||
|
|
||||||
|
#warning CONFIG_FT_CONTROL Not defined in p3cfgmgr.h
|
||||||
|
|
||||||
|
const uint32_t CONFIG_FT_CONTROL = 1;
|
||||||
|
|
||||||
|
ftController::ftController(CacheStrapper *cs, ftDataMultiplex *dm, std::string configDir)
|
||||||
|
:CacheTransfer(cs), p3Config(CONFIG_FT_CONTROL)
|
||||||
{
|
{
|
||||||
public:
|
/* TODO */
|
||||||
|
}
|
||||||
|
|
||||||
/* Setup */
|
void ftController::setFtSearch(ftSearch *search)
|
||||||
ftController::ftController(std::string configDir);
|
|
||||||
|
|
||||||
void ftController::setFtSearch(ftSearch *search)
|
|
||||||
{
|
{
|
||||||
mSearch = search;
|
mSearch = search;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
void ftController::run()
|
||||||
virtual void ftController::run()
|
|
||||||
{
|
{
|
||||||
/* check the queues */
|
/* check the queues */
|
||||||
}
|
}
|
||||||
@ -68,6 +79,9 @@ void ftController::checkDownloadQueue()
|
|||||||
|
|
||||||
bool ftController::completeFile(std::string hash)
|
bool ftController::completeFile(std::string hash)
|
||||||
{
|
{
|
||||||
|
|
||||||
|
#if 0
|
||||||
|
|
||||||
RsStackMutex stack(ctrlMutex); /******* LOCKED ********/
|
RsStackMutex stack(ctrlMutex); /******* LOCKED ********/
|
||||||
|
|
||||||
std::map<std::string, ftFileControl>::iterator it;
|
std::map<std::string, ftFileControl>::iterator it;
|
||||||
@ -88,8 +102,7 @@ bool ftController::completeFile(std::string hash)
|
|||||||
|
|
||||||
/* done - cleanup */
|
/* done - cleanup */
|
||||||
fc->mTransfer->done();
|
fc->mTransfer->done();
|
||||||
mClientModule->removeTransferModule(fc->mTransfer);
|
mDataplex->removeTransferModule(fc->mTransfer->hash());
|
||||||
mServerModule->removeFileCreator(fc->mCreator);
|
|
||||||
|
|
||||||
delete fc->mTransfer;
|
delete fc->mTransfer;
|
||||||
fc->mTransfer = NULL;
|
fc->mTransfer = NULL;
|
||||||
@ -97,13 +110,15 @@ bool ftController::completeFile(std::string hash)
|
|||||||
delete fc->mCreator;
|
delete fc->mCreator;
|
||||||
fc->mCreator = NULL;
|
fc->mCreator = NULL;
|
||||||
|
|
||||||
fc->state = COMPLETE;
|
fc->mState = COMPLETE;
|
||||||
|
|
||||||
/* switch map */
|
/* switch map */
|
||||||
mCompleted[fc->hash] = *fc;
|
mCompleted[fc->mHash] = *fc;
|
||||||
mDownloads.erase(it);
|
mDownloads.erase(it);
|
||||||
|
|
||||||
return true;
|
return true;
|
||||||
|
|
||||||
|
#endif
|
||||||
}
|
}
|
||||||
|
|
||||||
/***************************************************************/
|
/***************************************************************/
|
||||||
@ -112,11 +127,18 @@ bool ftController::completeFile(std::string hash)
|
|||||||
|
|
||||||
bool ftController::FileRequest(std::string fname, std::string hash,
|
bool ftController::FileRequest(std::string fname, std::string hash,
|
||||||
uint64_t size, std::string dest, uint32_t flags,
|
uint64_t size, std::string dest, uint32_t flags,
|
||||||
std::list<std::string> srcIds)
|
std::list<std::string> &srcIds)
|
||||||
{
|
{
|
||||||
/* check if we have the file */
|
|
||||||
|
|
||||||
if (ftSearch->findFile(LOCAL))
|
#if 0 /*** FIX ME !!!**************/
|
||||||
|
|
||||||
|
/* check if we have the file */
|
||||||
|
FileInfo info;
|
||||||
|
|
||||||
|
if (mSearch->search(hash, size,
|
||||||
|
RS_FILE_HINTS_LOCAL |
|
||||||
|
RS_FILE_HINTS_EXTRA |
|
||||||
|
RS_FILE_HINTS_SPEC_ONLY, info))
|
||||||
{
|
{
|
||||||
/* have it already */
|
/* have it already */
|
||||||
/* add in as completed transfer */
|
/* add in as completed transfer */
|
||||||
@ -124,9 +146,11 @@ bool ftController::FileRequest(std::string fname, std::string hash,
|
|||||||
}
|
}
|
||||||
|
|
||||||
/* do a source search - for any extra sources */
|
/* do a source search - for any extra sources */
|
||||||
if (ftSearch->findFile(REMOTE))
|
if (mSearch->search(hash, size,
|
||||||
|
RS_FILE_HINTS_REMOTE |
|
||||||
|
RS_FILE_HINTS_SPEC_ONLY, info))
|
||||||
{
|
{
|
||||||
|
/* do something with results */
|
||||||
}
|
}
|
||||||
|
|
||||||
std::map<std::string, ftTransferModule *> mTransfers;
|
std::map<std::string, ftTransferModule *> mTransfers;
|
||||||
@ -136,27 +160,28 @@ bool ftController::FileRequest(std::string fname, std::string hash,
|
|||||||
std::string savepath = mDownloadPath + "/" + fname;
|
std::string savepath = mDownloadPath + "/" + fname;
|
||||||
std::string chunker = "";
|
std::string chunker = "";
|
||||||
ftFileCreator *fc = new ftFileCreator(savepath, size, hash, chunker);
|
ftFileCreator *fc = new ftFileCreator(savepath, size, hash, chunker);
|
||||||
ftTransferModule = *tm = new ftTransferModule(fc, mClientModule);
|
ftTransferModule *tm = new ftTransferModule(fc, mDataplex);
|
||||||
|
|
||||||
/* add into maps */
|
/* add into maps */
|
||||||
ftFileControl ftfc(fname, size, hash, flags, fc, tm);
|
ftFileControl ftfc(fname, size, hash, flags, fc, tm);
|
||||||
|
|
||||||
/* add to ClientModule */
|
/* add to ClientModule */
|
||||||
mClientModule->addTransferModule(tm);
|
mDataplex->addTransferModule(tm, fc);
|
||||||
|
|
||||||
/* now add source peers (and their current state) */
|
/* now add source peers (and their current state) */
|
||||||
tm->setFileSources(srcIds);
|
tm->setFileSources(srcIds);
|
||||||
|
|
||||||
/* get current state for transfer module */
|
/* get current state for transfer module */
|
||||||
|
std::list<std::string>::iterator it;
|
||||||
for(it = srcIds.begin(); it != srcIds.end(); it++)
|
for(it = srcIds.begin(); it != srcIds.end(); it++)
|
||||||
{
|
{
|
||||||
if (mConnMgr->isOnline(*it))
|
if (mConnMgr->isOnline(*it))
|
||||||
{
|
{
|
||||||
tm->setPeer(*it, TRICKLE | ONLINE);
|
tm->setPeer(*it, RS_FILE_RATE_TRICKLE | RS_FILE_PEER_ONLINE);
|
||||||
}
|
}
|
||||||
else
|
else
|
||||||
{
|
{
|
||||||
tm->setPeer(*it, OFFLINE);
|
tm->setPeer(*it, RS_FILE_PEER_OFFLINE);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -165,12 +190,26 @@ bool ftController::FileRequest(std::string fname, std::string hash,
|
|||||||
mDownloads[hash] = ftfc;
|
mDownloads[hash] = ftfc;
|
||||||
mSlowQueue.push_back(hash);
|
mSlowQueue.push_back(hash);
|
||||||
|
|
||||||
|
#endif
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
bool ftController::FileCancel(std::string hash);
|
bool ftController::FileCancel(std::string hash)
|
||||||
bool ftController::FileControl(std::string hash, uint32_t flags);
|
{
|
||||||
bool ftController::FileClearCompleted();
|
/* TODO */
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool ftController::FileControl(std::string hash, uint32_t flags)
|
||||||
|
{
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool ftController::FileClearCompleted()
|
||||||
|
{
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
/* get Details of File Transfers */
|
/* get Details of File Transfers */
|
||||||
bool ftController::FileDownloads(std::list<std::string> &hashs)
|
bool ftController::FileDownloads(std::list<std::string> &hashs)
|
||||||
@ -180,7 +219,7 @@ bool ftController::FileDownloads(std::list<std::string> &hashs)
|
|||||||
std::map<std::string, ftFileControl>::iterator it;
|
std::map<std::string, ftFileControl>::iterator it;
|
||||||
for(it = mDownloads.begin(); it != mDownloads.end(); it++)
|
for(it = mDownloads.begin(); it != mDownloads.end(); it++)
|
||||||
{
|
{
|
||||||
hashes.push_back(it->second.hash);
|
hashs.push_back(it->second.mHash);
|
||||||
}
|
}
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
@ -200,8 +239,10 @@ bool ftController::setDownloadDirectory(std::string path)
|
|||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
|
|
||||||
bool ftController::setPartialsDirectory(std::string path);
|
bool ftController::setPartialsDirectory(std::string path)
|
||||||
{
|
{
|
||||||
|
#if 0 /*** FIX ME !!!**************/
|
||||||
|
|
||||||
/* check it is not a subdir of download / shared directories (BAD) - TODO */
|
/* check it is not a subdir of download / shared directories (BAD) - TODO */
|
||||||
|
|
||||||
/* check if it exists */
|
/* check if it exists */
|
||||||
@ -220,7 +261,9 @@ bool ftController::setPartialsDirectory(std::string path);
|
|||||||
}
|
}
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
return false;
|
return false;
|
||||||
|
#endif
|
||||||
}
|
}
|
||||||
|
|
||||||
std::string ftController::getDownloadDirectory()
|
std::string ftController::getDownloadDirectory()
|
||||||
@ -266,6 +309,9 @@ bool ftController::FileDetails(std::string hash, FileInfo &info)
|
|||||||
*/
|
*/
|
||||||
void ftController::statusChange(const std::list<pqipeer> &plist)
|
void ftController::statusChange(const std::list<pqipeer> &plist)
|
||||||
{
|
{
|
||||||
|
|
||||||
|
#if 0 /*** FIX ME !!!**************/
|
||||||
|
|
||||||
RsStackMutex stack(ctrlMutex); /******* LOCKED ********/
|
RsStackMutex stack(ctrlMutex); /******* LOCKED ********/
|
||||||
|
|
||||||
/* add online to all downloads */
|
/* add online to all downloads */
|
||||||
@ -278,11 +324,11 @@ void ftController::statusChange(const std::list<pqipeer> &plist)
|
|||||||
{
|
{
|
||||||
if (pit->actions | RS_PEER_CONNECTED)
|
if (pit->actions | RS_PEER_CONNECTED)
|
||||||
{
|
{
|
||||||
((it->second).tm)->setPeer(ONLINE | TRICKLE);
|
((it->second).mTransfer)->setPeer(RS_FILE_PEER_ONLINE | RS_FILE_RATE_TRICKLE);
|
||||||
}
|
}
|
||||||
else if (pit->actions | RS_PEER_DISCONNECTED)
|
else if (pit->actions | RS_PEER_DISCONNECTED)
|
||||||
{
|
{
|
||||||
((it->second).tm)->setPeer(OFFLINE);
|
((it->second).mTransfer)->setPeer(RS_FILE_PEER_OFFLINE);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -294,71 +340,33 @@ void ftController::statusChange(const std::list<pqipeer> &plist)
|
|||||||
{
|
{
|
||||||
/* add in */
|
/* add in */
|
||||||
|
|
||||||
((it->second).tm)->setPeer(ONLINE | TRICKLE);
|
((it->second).mTransfer)->setPeer(RS_FILE_PEER_ONLINE | RS_FILE_RATE_TRICKLE);
|
||||||
}
|
}
|
||||||
else if (pit->actions | RS_PEER_DISCONNECTED)
|
else if (pit->actions | RS_PEER_DISCONNECTED)
|
||||||
{
|
{
|
||||||
((it->second).tm)->setPeer(OFFLINE);
|
((it->second).mTransfer)->setPeer(RS_FILE_PEER_OFFLINE);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#endif
|
||||||
|
|
||||||
|
}
|
||||||
|
/* p3Config Interface */
|
||||||
|
RsSerialiser *ftController::setupSerialiser()
|
||||||
|
{
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
std::list<RsItem *> ftController::saveList(bool &cleanup)
|
||||||
|
{
|
||||||
|
std::list<RsItem *> emptyList;
|
||||||
|
return emptyList;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
bool ftController::loadList(std::list<RsItem *> load)
|
||||||
/* p3Config Interface */
|
|
||||||
protected:
|
|
||||||
virtual RsSerialiser *setupSerialiser();
|
|
||||||
virtual std::list<RsItem *> saveList(bool &cleanup);
|
|
||||||
virtual bool loadList(std::list<RsItem *> load);
|
|
||||||
|
|
||||||
private:
|
|
||||||
|
|
||||||
/* RunTime Functions */
|
|
||||||
|
|
||||||
/* pointers to other components */
|
|
||||||
|
|
||||||
ftSearch *mSearch;
|
|
||||||
|
|
||||||
RsMutex ctrlMutex;
|
|
||||||
|
|
||||||
std::list<FileDetails> incomingQueue;
|
|
||||||
std::map<std::string, FileDetails> mCompleted;
|
|
||||||
|
|
||||||
class ftFileControl
|
|
||||||
{
|
{
|
||||||
public:
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
ftFileControl(std::string fname, uint64_t size, std::string hash,
|
|
||||||
uint32_t flags, ftFileCreator *fc, ftTransferModule *tm);
|
|
||||||
|
|
||||||
std::string mName,
|
|
||||||
uint64_t mSize;
|
|
||||||
std::string mHash,
|
|
||||||
uint32_t mFlags;
|
|
||||||
ftFileCreator *mCreator;
|
|
||||||
ftTransferModule *mTransfer;
|
|
||||||
};
|
|
||||||
|
|
||||||
class ftPeerControl
|
|
||||||
{
|
|
||||||
std::string peerId;
|
|
||||||
std::map<std::string, uint32_t> priority;
|
|
||||||
uint32_t currentBandwidth;
|
|
||||||
uint32_t maxBandwidth;
|
|
||||||
};
|
|
||||||
|
|
||||||
std::map<std::string, ftFileControl> mDownloads;
|
|
||||||
|
|
||||||
/* A Bunch of Queues */
|
|
||||||
std::map<std::string, std::string> mStreamQueue;
|
|
||||||
std::map<std::string, std::string> mFastQueue;
|
|
||||||
std::list<std::string> mSlowQueue;
|
|
||||||
std::list<std::string> mTrickleQueue;
|
|
||||||
|
|
||||||
std::string mConfigPath;
|
|
||||||
std::string mDownloadPath;
|
|
||||||
std::string mPartialPath;
|
|
||||||
};
|
|
||||||
|
|
||||||
#endif
|
|
||||||
|
@ -41,13 +41,41 @@
|
|||||||
class ftFileCreator;
|
class ftFileCreator;
|
||||||
class ftTransferModule;
|
class ftTransferModule;
|
||||||
class ftFileProvider;
|
class ftFileProvider;
|
||||||
|
class ftSearch;
|
||||||
|
class ftDataMultiplex;
|
||||||
|
|
||||||
|
#include "dbase/cachestrapper.h"
|
||||||
|
#include "util/rsthreads.h"
|
||||||
|
#include "pqi/pqimonitor.h"
|
||||||
|
#include "pqi/p3cfgmgr.h"
|
||||||
|
|
||||||
|
#include "rsiface/rsfiles.h"
|
||||||
|
|
||||||
|
#include <map>
|
||||||
|
|
||||||
|
class ftFileControl
|
||||||
|
{
|
||||||
|
public:
|
||||||
|
|
||||||
|
ftFileControl();
|
||||||
|
ftFileControl(std::string fname, uint64_t size, std::string hash,
|
||||||
|
uint32_t flags, ftFileCreator *fc, ftTransferModule *tm);
|
||||||
|
|
||||||
|
ftTransferModule * mTransfer;
|
||||||
|
ftFileCreator * mCreator;
|
||||||
|
uint32_t mState;
|
||||||
|
std::string mHash;
|
||||||
|
std::string mName;
|
||||||
|
uint64_t mSize;
|
||||||
|
};
|
||||||
|
|
||||||
|
|
||||||
class ftController: public CacheTransfer, public RsThread, public pqiMonitor, public p3Config
|
class ftController: public CacheTransfer, public RsThread, public pqiMonitor, public p3Config
|
||||||
{
|
{
|
||||||
public:
|
public:
|
||||||
|
|
||||||
/* Setup */
|
/* Setup */
|
||||||
ftController(std::string configDir);
|
ftController(CacheStrapper *cs, ftDataMultiplex *dm, std::string configDir);
|
||||||
|
|
||||||
void setFtSearch(ftSearch *);
|
void setFtSearch(ftSearch *);
|
||||||
|
|
||||||
@ -59,7 +87,7 @@ virtual void run();
|
|||||||
|
|
||||||
bool FileRequest(std::string fname, std::string hash,
|
bool FileRequest(std::string fname, std::string hash,
|
||||||
uint64_t size, std::string dest, uint32_t flags,
|
uint64_t size, std::string dest, uint32_t flags,
|
||||||
std::list<std::string> sourceIds);
|
std::list<std::string> &sourceIds);
|
||||||
|
|
||||||
bool FileCancel(std::string hash);
|
bool FileCancel(std::string hash);
|
||||||
bool FileControl(std::string hash, uint32_t flags);
|
bool FileControl(std::string hash, uint32_t flags);
|
||||||
@ -93,18 +121,24 @@ virtual bool loadList(std::list<RsItem *> load);
|
|||||||
private:
|
private:
|
||||||
|
|
||||||
/* RunTime Functions */
|
/* RunTime Functions */
|
||||||
|
void checkDownloadQueue();
|
||||||
|
bool completeFile(std::string hash);
|
||||||
|
|
||||||
/* pointers to other components */
|
/* pointers to other components */
|
||||||
|
|
||||||
ftSearch *mSearch;
|
ftSearch *mSearch;
|
||||||
|
ftDataMultiplex *mDataplex;
|
||||||
|
|
||||||
RsMutex ctrlMutex;
|
RsMutex ctrlMutex;
|
||||||
|
|
||||||
std::list<FileInfo> incomingQueue;
|
std::list<FileInfo> incomingQueue;
|
||||||
std::map<std::string, FileInfo> mCompleted;
|
std::map<std::string, FileInfo> mCompleted;
|
||||||
|
|
||||||
std::map<std::string, ftTransferModule *> mTransfers;
|
|
||||||
std::map<std::string, ftFileCreator *> mFileCreators;
|
std::map<std::string, ftFileControl> mDownloads;
|
||||||
|
|
||||||
|
//std::map<std::string, ftTransferModule *> mTransfers;
|
||||||
|
//std::map<std::string, ftFileCreator *> mFileCreators;
|
||||||
|
|
||||||
std::string mConfigPath;
|
std::string mConfigPath;
|
||||||
std::string mDownloadPath;
|
std::string mDownloadPath;
|
||||||
|
@ -77,11 +77,11 @@ bool ftDataMultiplex::addTransferModule(ftTransferModule *mod, ftFileCreator *f)
|
|||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
bool ftDataMultiplex::removeTransferModule(ftTransferModule *mod, ftFileCreator *f)
|
bool ftDataMultiplex::removeTransferModule(std::string hash)
|
||||||
{
|
{
|
||||||
RsStackMutex stack(dataMtx); /******* LOCK MUTEX ******/
|
RsStackMutex stack(dataMtx); /******* LOCK MUTEX ******/
|
||||||
std::map<std::string, ftClient>::iterator it;
|
std::map<std::string, ftClient>::iterator it;
|
||||||
if (mClients.end() == (it = mClients.find(mod->hash())))
|
if (mClients.end() == (it = mClients.find(hash)))
|
||||||
{
|
{
|
||||||
/* error */
|
/* error */
|
||||||
return false;
|
return false;
|
||||||
@ -91,6 +91,51 @@ bool ftDataMultiplex::removeTransferModule(ftTransferModule *mod, ftFileCreator
|
|||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
bool ftDataMultiplex::FileUploads(std::list<std::string> &hashs)
|
||||||
|
{
|
||||||
|
RsStackMutex stack(dataMtx); /******* LOCK MUTEX ******/
|
||||||
|
std::map<std::string, ftFileProvider *>::iterator sit;
|
||||||
|
for(sit = mServers.begin(); sit != mServers.end(); sit++)
|
||||||
|
{
|
||||||
|
hashs.push_back(sit->first);
|
||||||
|
}
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool ftDataMultiplex::FileDownloads(std::list<std::string> &hashs)
|
||||||
|
{
|
||||||
|
RsStackMutex stack(dataMtx); /******* LOCK MUTEX ******/
|
||||||
|
std::map<std::string, ftClient>::iterator cit;
|
||||||
|
for(cit = mClients.begin(); cit != mClients.end(); cit++)
|
||||||
|
{
|
||||||
|
hashs.push_back(cit->first);
|
||||||
|
}
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
bool ftDataMultiplex::FileDetails(std::string hash, uint32_t hintsflag, FileInfo &info)
|
||||||
|
{
|
||||||
|
RsStackMutex stack(dataMtx); /******* LOCK MUTEX ******/
|
||||||
|
std::map<std::string, ftFileProvider *>::iterator sit;
|
||||||
|
sit = mServers.find(hash);
|
||||||
|
if (sit != mServers.end())
|
||||||
|
{
|
||||||
|
(sit->second)->FileDetails(info);
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
std::map<std::string, ftClient>::iterator cit;
|
||||||
|
if (mClients.end() != (cit = mClients.find(hash)))
|
||||||
|
{
|
||||||
|
//(cit->second).mModule->FileDetails(info);
|
||||||
|
(cit->second).mCreator->FileDetails(info);
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
/* data interface */
|
/* data interface */
|
||||||
|
|
||||||
/*************** SEND INTERFACE (calls ftDataSend) *******************/
|
/*************** SEND INTERFACE (calls ftDataSend) *******************/
|
||||||
|
@ -45,6 +45,8 @@ class ftSearch;
|
|||||||
#include "util/rsthreads.h"
|
#include "util/rsthreads.h"
|
||||||
|
|
||||||
#include "ft/ftdata.h"
|
#include "ft/ftdata.h"
|
||||||
|
#include "rsiface/rsfiles.h"
|
||||||
|
|
||||||
|
|
||||||
class ftClient
|
class ftClient
|
||||||
{
|
{
|
||||||
@ -86,9 +88,13 @@ class ftDataMultiplex: public ftDataRecv, public RsQueueThread
|
|||||||
|
|
||||||
/* ftController Interface */
|
/* ftController Interface */
|
||||||
bool addTransferModule(ftTransferModule *mod, ftFileCreator *f);
|
bool addTransferModule(ftTransferModule *mod, ftFileCreator *f);
|
||||||
bool removeTransferModule(ftTransferModule *mod, ftFileCreator *f);
|
bool removeTransferModule(std::string hash);
|
||||||
|
|
||||||
/* data interface */
|
/* data interface */
|
||||||
|
/* get Details of File Transfers */
|
||||||
|
bool FileUploads(std::list<std::string> &hashs);
|
||||||
|
bool FileDownloads(std::list<std::string> &hashs);
|
||||||
|
bool FileDetails(std::string hash, uint32_t hintsflag, FileInfo &info);
|
||||||
|
|
||||||
/*************** SEND INTERFACE (calls ftDataSend) *******************/
|
/*************** SEND INTERFACE (calls ftDataSend) *******************/
|
||||||
|
|
||||||
|
104
libretroshare/src/ft/ftdbase.cc
Normal file
104
libretroshare/src/ft/ftdbase.cc
Normal file
@ -0,0 +1,104 @@
|
|||||||
|
/*
|
||||||
|
* libretroshare/src/ft: ftdbase.cc
|
||||||
|
*
|
||||||
|
* 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".
|
||||||
|
*
|
||||||
|
*/
|
||||||
|
|
||||||
|
#include "ft/ftdbase.h"
|
||||||
|
#include "util/rsdir.h"
|
||||||
|
|
||||||
|
|
||||||
|
ftFiStore::ftFiStore(CacheStrapper *cs, CacheTransfer *cft, NotifyBase *cb_in,
|
||||||
|
RsPeerId ownid, std::string cachedir)
|
||||||
|
:FileIndexStore(cs, cft, cb_in, ownid, cachedir)
|
||||||
|
{
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool ftFiStore::search(std::string hash, uint64_t size, uint32_t hintflags, FileInfo &info) const
|
||||||
|
{
|
||||||
|
std::list<FileDetail> results;
|
||||||
|
std::list<FileDetail>::iterator it;
|
||||||
|
|
||||||
|
if (SearchHash(hash, results))
|
||||||
|
{
|
||||||
|
for(it = results.begin(); it != results.end(); it++)
|
||||||
|
{
|
||||||
|
if (it->size == size)
|
||||||
|
{
|
||||||
|
/*
|
||||||
|
*/
|
||||||
|
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
ftFiMonitor::ftFiMonitor(CacheStrapper *cs, std::string cachedir, std::string pid)
|
||||||
|
:FileIndexMonitor(cs, cachedir, pid)
|
||||||
|
{
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool ftFiMonitor::search(std::string hash, uint64_t size, uint32_t hintflags, FileInfo &info) const
|
||||||
|
{
|
||||||
|
uint64_t fsize;
|
||||||
|
std::string path;
|
||||||
|
|
||||||
|
if (findLocalFile(hash, path, fsize))
|
||||||
|
{
|
||||||
|
/* fill in details */
|
||||||
|
|
||||||
|
info.size = fsize;
|
||||||
|
info.fname = RsDirUtil::getTopDir(path);
|
||||||
|
info.path = path;
|
||||||
|
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
return false;
|
||||||
|
};
|
||||||
|
|
||||||
|
ftCacheStrapper::ftCacheStrapper(p3AuthMgr *am, p3ConnectMgr *cm)
|
||||||
|
:CacheStrapper(am, cm)
|
||||||
|
{
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
/* overloaded search function */
|
||||||
|
bool ftCacheStrapper::search(std::string hash, uint64_t size, uint32_t hintflags, FileInfo &info) const
|
||||||
|
{
|
||||||
|
CacheData data;
|
||||||
|
if (findCache(hash, data))
|
||||||
|
{
|
||||||
|
/* ... */
|
||||||
|
info.size = data.size;
|
||||||
|
info.fname = data.name;
|
||||||
|
info.path = data.path + "/" + data.name;
|
||||||
|
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
75
libretroshare/src/ft/ftdbase.h
Normal file
75
libretroshare/src/ft/ftdbase.h
Normal file
@ -0,0 +1,75 @@
|
|||||||
|
/*
|
||||||
|
* libretroshare/src/ft: ftdbase.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_DBASE_INTERFACE_HEADER
|
||||||
|
#define FT_DBASE_INTERFACE_HEADER
|
||||||
|
|
||||||
|
/*
|
||||||
|
* ftdbase.
|
||||||
|
*
|
||||||
|
* Wrappers for the Cache/FiStore/FiMonitor classes.
|
||||||
|
* So they work in the ft world.
|
||||||
|
*/
|
||||||
|
|
||||||
|
#include "ft/ftsearch.h"
|
||||||
|
|
||||||
|
#include "dbase/fistore.h"
|
||||||
|
#include "dbase/fimonitor.h"
|
||||||
|
#include "dbase/cachestrapper.h"
|
||||||
|
|
||||||
|
|
||||||
|
class ftFiStore: public FileIndexStore, public ftSearch
|
||||||
|
{
|
||||||
|
public:
|
||||||
|
ftFiStore(CacheStrapper *cs, CacheTransfer *cft, NotifyBase *cb_in,
|
||||||
|
RsPeerId ownid, std::string cachedir);
|
||||||
|
|
||||||
|
/* overloaded search function */
|
||||||
|
virtual bool search(std::string hash, uint64_t size, uint32_t hintflags, FileInfo &info) const;
|
||||||
|
};
|
||||||
|
|
||||||
|
class ftFiMonitor: public FileIndexMonitor, public ftSearch
|
||||||
|
{
|
||||||
|
public:
|
||||||
|
ftFiMonitor(CacheStrapper *cs, std::string cachedir, std::string pid);
|
||||||
|
|
||||||
|
/* overloaded search function */
|
||||||
|
virtual bool search(std::string hash, uint64_t size, uint32_t hintflags, FileInfo &info) const;
|
||||||
|
|
||||||
|
};
|
||||||
|
|
||||||
|
class ftCacheStrapper: public CacheStrapper, public ftSearch
|
||||||
|
{
|
||||||
|
public:
|
||||||
|
ftCacheStrapper(p3AuthMgr *am, p3ConnectMgr *cm);
|
||||||
|
|
||||||
|
/* overloaded search function */
|
||||||
|
virtual bool search(std::string hash, uint64_t size, uint32_t hintflags, FileInfo &info) const;
|
||||||
|
|
||||||
|
};
|
||||||
|
|
||||||
|
|
||||||
|
#endif
|
||||||
|
|
@ -108,7 +108,7 @@ void ftExtraList::hashAFile()
|
|||||||
|
|
||||||
/* hash it! */
|
/* hash it! */
|
||||||
std::string name, hash;
|
std::string name, hash;
|
||||||
uint64_t size;
|
//uint64_t size;
|
||||||
if (RsDirUtil::hashFile(details.info.path, details.info.fname,
|
if (RsDirUtil::hashFile(details.info.path, details.info.fname,
|
||||||
details.info.hash, details.info.size))
|
details.info.hash, details.info.size))
|
||||||
{
|
{
|
||||||
@ -204,7 +204,7 @@ bool ftExtraList::cleanupOldFiles()
|
|||||||
for(it = mFiles.begin(); it != mFiles.end(); it++)
|
for(it = mFiles.begin(); it != mFiles.end(); it++)
|
||||||
{
|
{
|
||||||
/* check timestamps */
|
/* check timestamps */
|
||||||
if (it->second.start + it->second.period < now)
|
if (it->second.start + it->second.period < (unsigned) now)
|
||||||
{
|
{
|
||||||
toRemove.push_back(it->first);
|
toRemove.push_back(it->first);
|
||||||
}
|
}
|
||||||
|
@ -1,5 +1,7 @@
|
|||||||
#include "ftfileprovider.h"
|
#include "ftfileprovider.h"
|
||||||
|
|
||||||
|
#include "util/rsdir.h"
|
||||||
|
|
||||||
ftFileProvider::ftFileProvider(std::string path, uint64_t size, std::string
|
ftFileProvider::ftFileProvider(std::string path, uint64_t size, std::string
|
||||||
hash) : total_size(size), hash(hash), file_name(path), fd(NULL) {
|
hash) : total_size(size), hash(hash), file_name(path), fd(NULL) {
|
||||||
|
|
||||||
@ -21,6 +23,19 @@ uint64_t ftFileProvider::getFileSize()
|
|||||||
return total_size;
|
return total_size;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
bool ftFileProvider::FileDetails(FileInfo &info)
|
||||||
|
{
|
||||||
|
info.hash = hash;
|
||||||
|
info.size = total_size;
|
||||||
|
info.path = file_name;
|
||||||
|
info.fname = RsDirUtil::getTopDir(file_name);
|
||||||
|
|
||||||
|
/* Use req_loc / req_size to estimate data rate */
|
||||||
|
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
bool ftFileProvider::getFileData(uint64_t offset, uint32_t chunk_size, void *data)
|
bool ftFileProvider::getFileData(uint64_t offset, uint32_t chunk_size, void *data)
|
||||||
{
|
{
|
||||||
RsStackMutex stack(ftPMutex); /********** STACK LOCKED MTX ******/
|
RsStackMutex stack(ftPMutex); /********** STACK LOCKED MTX ******/
|
||||||
|
@ -33,14 +33,19 @@
|
|||||||
#include <iostream>
|
#include <iostream>
|
||||||
#include <stdint.h>
|
#include <stdint.h>
|
||||||
#include "util/rsthreads.h"
|
#include "util/rsthreads.h"
|
||||||
|
#include "rsiface/rsfiles.h"
|
||||||
|
|
||||||
class ftFileProvider
|
class ftFileProvider
|
||||||
{
|
{
|
||||||
public:
|
public:
|
||||||
ftFileProvider(std::string path, uint64_t size, std::string hash);
|
ftFileProvider(std::string path, uint64_t size, std::string hash);
|
||||||
virtual ~ftFileProvider();
|
virtual ~ftFileProvider();
|
||||||
virtual bool getFileData(uint64_t offset, uint32_t chunk_size, void *data);
|
virtual bool getFileData(uint64_t offset, uint32_t chunk_size, void *data);
|
||||||
|
virtual bool FileDetails(FileInfo &info);
|
||||||
std::string getHash();
|
std::string getHash();
|
||||||
uint64_t getFileSize();
|
uint64_t getFileSize();
|
||||||
|
|
||||||
|
|
||||||
protected:
|
protected:
|
||||||
virtual int initializeFileAttrs();
|
virtual int initializeFileAttrs();
|
||||||
uint64_t total_size;
|
uint64_t total_size;
|
||||||
|
@ -32,22 +32,24 @@ const int ftserverzone = 29539;
|
|||||||
#include "ft/ftcontroller.h"
|
#include "ft/ftcontroller.h"
|
||||||
#include "ft/ftdatamultiplex.h"
|
#include "ft/ftdatamultiplex.h"
|
||||||
|
|
||||||
#include "dbase/cachestrapper.h"
|
|
||||||
#include "dbase/fimonitor.h"
|
// Includes CacheStrapper / FiMonitor / FiStore for us.
|
||||||
#include "dbase/fistore.h"
|
|
||||||
|
#include "ft/ftdbase.h"
|
||||||
|
|
||||||
#include "pqi/pqi.h"
|
#include "pqi/pqi.h"
|
||||||
#include "pqi/p3connmgr.h"
|
#include "pqi/p3connmgr.h"
|
||||||
|
|
||||||
|
#include "serialiser/rsserviceids.h"
|
||||||
|
|
||||||
#include <iostream>
|
#include <iostream>
|
||||||
#include <sstream>
|
#include <sstream>
|
||||||
|
|
||||||
/* Setup */
|
/* Setup */
|
||||||
ftServer::ftServer(CacheStrapper *cStrapper, p3ConnectMgr *connMgr)
|
ftServer::ftServer(p3AuthMgr *authMgr, p3ConnectMgr *connMgr)
|
||||||
:mCacheStrapper(cStrapper), mConnMgr(connMgr)
|
:mAuthMgr(authMgr), mConnMgr(connMgr)
|
||||||
{
|
{
|
||||||
|
mCacheStrapper = new ftCacheStrapper(authMgr, connMgr);
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
void ftServer::setConfigDirectory(std::string path)
|
void ftServer::setConfigDirectory(std::string path)
|
||||||
@ -69,19 +71,16 @@ void ftServer::addConfigComponents(p3ConfigMgr *mgr)
|
|||||||
}
|
}
|
||||||
|
|
||||||
/* Final Setup (once everything is assigned) */
|
/* Final Setup (once everything is assigned) */
|
||||||
void ftServer::SetupFtServer()
|
void ftServer::SetupFtServer(NotifyBase *cb)
|
||||||
{
|
{
|
||||||
/* make Controller */
|
|
||||||
mFtController = new ftController(config_dir);
|
|
||||||
NotifyBase *cb = getNotify();
|
|
||||||
|
|
||||||
/* setup FiStore/Monitor */
|
/* setup FiStore/Monitor */
|
||||||
std::string localcachedir = config_dir + "/cache/local";
|
std::string localcachedir = mConfigPath + "/cache/local";
|
||||||
std::string remotecachedir = config_dir + "/cache/remote";
|
std::string remotecachedir = mConfigPath + "/cache/remote";
|
||||||
std::string ownId = mConnMgr->getOwnId();
|
std::string ownId = mConnMgr->getOwnId();
|
||||||
|
|
||||||
mFiStore = new FileIndexStore(mCacheStrapper, mFtController, cb, ownId, remotecachedir);
|
mFiStore = new ftFiStore(mCacheStrapper, mFtController, cb, ownId, remotecachedir);
|
||||||
mFiMon = new FileIndexMonitor(mCacheStrapper, localcachedir, ownId);
|
mFiMon = new ftFiMonitor(mCacheStrapper, localcachedir, ownId);
|
||||||
|
|
||||||
/* now add the set to the cachestrapper */
|
/* now add the set to the cachestrapper */
|
||||||
CachePair cp(mFiMon, mFiStore, CacheId(RS_SERVICE_TYPE_FILE_INDEX, 0));
|
CachePair cp(mFiMon, mFiStore, CacheId(RS_SERVICE_TYPE_FILE_INDEX, 0));
|
||||||
@ -91,15 +90,22 @@ void ftServer::SetupFtServer()
|
|||||||
mFtExtra = new ftExtraList();
|
mFtExtra = new ftExtraList();
|
||||||
|
|
||||||
mFtSearch = new ftFileSearch();
|
mFtSearch = new ftFileSearch();
|
||||||
//mFtSearch->addSearchMode(mCacheStrapper, RS_FILE_HINTS_CACHE);
|
mFtSearch->addSearchMode(mCacheStrapper, RS_FILE_HINTS_CACHE);
|
||||||
mFtSearch->addSearchMode(mFtExtra, RS_FILE_HINTS_EXTRA);
|
mFtSearch->addSearchMode(mFtExtra, RS_FILE_HINTS_EXTRA);
|
||||||
//mFtSearch->addSearchMode(mFiMon, RS_FILE_HINTS_LOCAL);
|
mFtSearch->addSearchMode(mFiMon, RS_FILE_HINTS_LOCAL);
|
||||||
//mFtSearch->addSearchMode(mFiStore, RS_FILE_HINTS_REMOTE);
|
mFtSearch->addSearchMode(mFiStore, RS_FILE_HINTS_REMOTE);
|
||||||
|
|
||||||
|
/* Transport */
|
||||||
|
mFtDataplex = new ftDataMultiplex(this, mFtSearch);
|
||||||
|
|
||||||
|
/* make Controller */
|
||||||
|
mFtController = new ftController(mCacheStrapper, mFtDataplex, mConfigPath);
|
||||||
mFtController -> setFtSearch(mFtSearch);
|
mFtController -> setFtSearch(mFtSearch);
|
||||||
mFtController -> setSaveBasePath(save_dir);
|
std::string tmppath = "./";
|
||||||
|
mFtController->setPartialsDirectory(tmppath);
|
||||||
|
mFtController->setDownloadDirectory(tmppath);
|
||||||
|
|
||||||
mFtDataplex = ftDataMultiplex(this, mFtSearch);
|
mConnMgr->addMonitor(mFtController);
|
||||||
|
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
@ -147,10 +153,11 @@ CacheTransfer *ftServer::getCacheTransfer()
|
|||||||
/********************** Controller Access **********************/
|
/********************** Controller Access **********************/
|
||||||
/***************************************************************/
|
/***************************************************************/
|
||||||
|
|
||||||
bool ftServer::FileRequest(std::string fname, std::string hash,
|
bool ftServer::FileRequest(std::string fname, std::string hash, uint32_t size,
|
||||||
uint32_t size, std::string dest, uint32_t flags)
|
std::string dest, uint32_t flags, std::list<std::string> srcIds)
|
||||||
{
|
{
|
||||||
return mFtController->FileRequest(fname, hash, size, dest, flags);
|
return mFtController->FileRequest(fname, hash, size,
|
||||||
|
dest, flags, srcIds);
|
||||||
}
|
}
|
||||||
|
|
||||||
bool ftServer::FileCancel(std::string hash)
|
bool ftServer::FileCancel(std::string hash)
|
||||||
@ -168,11 +175,6 @@ bool ftServer::FileClearCompleted()
|
|||||||
return mFtController->FileClearCompleted();
|
return mFtController->FileClearCompleted();
|
||||||
}
|
}
|
||||||
|
|
||||||
/* get Details of File Transfers */
|
|
||||||
bool ftServer::FileDownloads(std::list<std::string> &hashs)
|
|
||||||
{
|
|
||||||
return mFtController->FileDownloads(hashs);
|
|
||||||
}
|
|
||||||
|
|
||||||
/* Directory Handling */
|
/* Directory Handling */
|
||||||
void ftServer::setDownloadDirectory(std::string path)
|
void ftServer::setDownloadDirectory(std::string path)
|
||||||
@ -200,9 +202,14 @@ std::string ftServer::getPartialsDirectory()
|
|||||||
/************************* Other Access ************************/
|
/************************* Other Access ************************/
|
||||||
/***************************************************************/
|
/***************************************************************/
|
||||||
|
|
||||||
|
bool ftServer::FileDownloads(std::list<std::string> &hashs)
|
||||||
|
{
|
||||||
|
return mFtDataplex->FileDownloads(hashs);
|
||||||
|
}
|
||||||
|
|
||||||
bool ftServer::FileUploads(std::list<std::string> &hashs)
|
bool ftServer::FileUploads(std::list<std::string> &hashs)
|
||||||
{
|
{
|
||||||
return mFtDataplex->FileUploads(hashes);
|
return mFtDataplex->FileUploads(hashs);
|
||||||
}
|
}
|
||||||
|
|
||||||
bool ftServer::FileDetails(std::string hash, uint32_t hintflags, FileInfo &info)
|
bool ftServer::FileDetails(std::string hash, uint32_t hintflags, FileInfo &info)
|
||||||
@ -210,16 +217,16 @@ bool ftServer::FileDetails(std::string hash, uint32_t hintflags, FileInfo &info)
|
|||||||
bool found = false;
|
bool found = false;
|
||||||
if (hintflags | RS_FILE_HINTS_DOWNLOAD)
|
if (hintflags | RS_FILE_HINTS_DOWNLOAD)
|
||||||
{
|
{
|
||||||
found = mFtController->FileDetails(hash, info);
|
found = mFtDataplex->FileDetails(hash, hintflags, info);
|
||||||
}
|
}
|
||||||
else if (hintflags | RS_FILE_HINTS_UPLOAD)
|
else if (hintflags | RS_FILE_HINTS_UPLOAD)
|
||||||
{
|
{
|
||||||
found = mFtDataplex->FileDetails(hash, info);
|
found = mFtDataplex->FileDetails(hash, hintflags, info);
|
||||||
}
|
}
|
||||||
|
|
||||||
if (!found)
|
if (!found)
|
||||||
{
|
{
|
||||||
mFtSearch->FileDetails(hash, hintflags, info);
|
mFtSearch->search(hash, 0, hintflags, info);
|
||||||
}
|
}
|
||||||
return found;
|
return found;
|
||||||
}
|
}
|
||||||
@ -301,17 +308,43 @@ bool ftServer::InDirectoryCheck()
|
|||||||
|
|
||||||
bool ftServer::getSharedDirectories(std::list<std::string> &dirs)
|
bool ftServer::getSharedDirectories(std::list<std::string> &dirs)
|
||||||
{
|
{
|
||||||
return mFiMon->getSharedDirectories(dirs);
|
mFiMon->getSharedDirectories(dirs);
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool ftServer::setSharedDirectories(std::list<std::string> &dirs)
|
||||||
|
{
|
||||||
|
mFiMon->setSharedDirectories(dirs);
|
||||||
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
bool ftServer::addSharedDirectory(std::string dir)
|
bool ftServer::addSharedDirectory(std::string dir)
|
||||||
{
|
{
|
||||||
return mFiMon->addSharedDirectory(dir);
|
std::list<std::string> dirList;
|
||||||
|
mFiMon->getSharedDirectories(dirList);
|
||||||
|
dirList.push_back(dir);
|
||||||
|
|
||||||
|
mFiMon->setSharedDirectories(dirList);
|
||||||
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
bool ftServer::removeSharedDirectory(std::string dir)
|
bool ftServer::removeSharedDirectory(std::string dir)
|
||||||
{
|
{
|
||||||
return mFiMon->removeSharedDirectory(dir);
|
std::list<std::string> dirList;
|
||||||
|
std::list<std::string>::iterator it;
|
||||||
|
|
||||||
|
mFiMon->getSharedDirectories(dirList);
|
||||||
|
|
||||||
|
if (dirList.end() != (it =
|
||||||
|
std::find(dirList.begin(), dirList.end(), dir)))
|
||||||
|
{
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
dirList.erase(it);
|
||||||
|
mFiMon->setSharedDirectories(dirList);
|
||||||
|
|
||||||
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
@ -324,7 +357,6 @@ bool ftServer::removeSharedDirectory(std::string dir)
|
|||||||
/**************** Config Interface *****************************/
|
/**************** Config Interface *****************************/
|
||||||
/***************************************************************/
|
/***************************************************************/
|
||||||
|
|
||||||
protected:
|
|
||||||
/* Key Functions to be overloaded for Full Configuration */
|
/* Key Functions to be overloaded for Full Configuration */
|
||||||
RsSerialiser *ftServer::setupSerialiser()
|
RsSerialiser *ftServer::setupSerialiser()
|
||||||
{
|
{
|
||||||
@ -372,6 +404,8 @@ bool ftServer::sendDataRequest(std::string peerId, std::string hash,
|
|||||||
rfi->chunksize = chunksize; /* ftr->chunk; */
|
rfi->chunksize = chunksize; /* ftr->chunk; */
|
||||||
|
|
||||||
mP3iface->SendFileRequest(rfi);
|
mP3iface->SendFileRequest(rfi);
|
||||||
|
|
||||||
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
const uint32_t MAX_FT_CHUNK = 32 * 1024; /* 32K */
|
const uint32_t MAX_FT_CHUNK = 32 * 1024; /* 32K */
|
||||||
@ -425,6 +459,8 @@ bool ftServer::sendData(std::string peerId, std::string hash, uint64_t size,
|
|||||||
|
|
||||||
/* clean up data */
|
/* clean up data */
|
||||||
free(data);
|
free(data);
|
||||||
|
|
||||||
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
@ -433,9 +469,6 @@ int ftServer::tick()
|
|||||||
rslog(RSL_DEBUG_BASIC, ftserverzone,
|
rslog(RSL_DEBUG_BASIC, ftserverzone,
|
||||||
"filedexserver::tick()");
|
"filedexserver::tick()");
|
||||||
|
|
||||||
/* the new Cache Hack() */
|
|
||||||
FileStoreTick();
|
|
||||||
|
|
||||||
if (mP3iface == NULL)
|
if (mP3iface == NULL)
|
||||||
{
|
{
|
||||||
std::ostringstream out;
|
std::ostringstream out;
|
||||||
@ -471,6 +504,8 @@ bool ftServer::handleInputQueues()
|
|||||||
{
|
{
|
||||||
handleCacheData();
|
handleCacheData();
|
||||||
handleFileData();
|
handleFileData();
|
||||||
|
return true;
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
bool ftServer::handleCacheData()
|
bool ftServer::handleCacheData()
|
||||||
@ -554,6 +589,7 @@ bool ftServer::handleCacheData()
|
|||||||
mP3iface -> SendSearchResult(ci);
|
mP3iface -> SendSearchResult(ci);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
@ -45,7 +45,7 @@
|
|||||||
|
|
||||||
#include "ft/ftdata.h"
|
#include "ft/ftdata.h"
|
||||||
#include "rsiface/rsfiles.h"
|
#include "rsiface/rsfiles.h"
|
||||||
#include "dbase/cachestrapper.h"
|
//#include "dbase/cachestrapper.h"
|
||||||
|
|
||||||
#include "pqi/pqi.h"
|
#include "pqi/pqi.h"
|
||||||
#include "pqi/p3cfgmgr.h"
|
#include "pqi/p3cfgmgr.h"
|
||||||
@ -55,8 +55,12 @@ class p3ConnectMgr;
|
|||||||
class p3AuthMgr;
|
class p3AuthMgr;
|
||||||
|
|
||||||
class CacheStrapper;
|
class CacheStrapper;
|
||||||
class FileIndexStore;
|
class CacheTransfer;
|
||||||
class FileIndexMonitor;
|
|
||||||
|
class NotifyBase; /* needed by FiStore */
|
||||||
|
class ftCacheStrapper;
|
||||||
|
class ftFiStore;
|
||||||
|
class ftFiMonitor;
|
||||||
|
|
||||||
class ftController;
|
class ftController;
|
||||||
class ftExtraList;
|
class ftExtraList;
|
||||||
@ -73,7 +77,7 @@ class ftServer: public RsFiles, public ftDataSend
|
|||||||
/******************** Setup ************************************/
|
/******************** Setup ************************************/
|
||||||
/***************************************************************/
|
/***************************************************************/
|
||||||
|
|
||||||
ftServer(CacheStrapper *cStrapper, p3ConnectMgr *connMgr);
|
ftServer(p3AuthMgr *authMgr, p3ConnectMgr *connMgr);
|
||||||
|
|
||||||
/* Assign important variables */
|
/* Assign important variables */
|
||||||
void setConfigDirectory(std::string path);
|
void setConfigDirectory(std::string path);
|
||||||
@ -88,7 +92,9 @@ CacheStrapper *getCacheStrapper();
|
|||||||
CacheTransfer *getCacheTransfer();
|
CacheTransfer *getCacheTransfer();
|
||||||
|
|
||||||
/* Final Setup (once everything is assigned) */
|
/* Final Setup (once everything is assigned) */
|
||||||
void SetupFtServer();
|
//void SetupFtServer();
|
||||||
|
void SetupFtServer(NotifyBase *cb);
|
||||||
|
|
||||||
void StartupThreads();
|
void StartupThreads();
|
||||||
|
|
||||||
/***************************************************************/
|
/***************************************************************/
|
||||||
@ -99,8 +105,8 @@ void StartupThreads();
|
|||||||
/***
|
/***
|
||||||
* Control of Downloads
|
* Control of Downloads
|
||||||
***/
|
***/
|
||||||
virtual bool FileRequest(std::string fname, std::string hash,
|
virtual bool FileRequest(std::string fname, std::string hash, uint32_t size,
|
||||||
uint32_t size, std::string dest, uint32_t flags);
|
std::string dest, uint32_t flags, std::list<std::string> srcIds);
|
||||||
virtual bool FileCancel(std::string hash);
|
virtual bool FileCancel(std::string hash);
|
||||||
virtual bool FileControl(std::string hash, uint32_t flags);
|
virtual bool FileControl(std::string hash, uint32_t flags);
|
||||||
virtual bool FileClearCompleted();
|
virtual bool FileClearCompleted();
|
||||||
@ -148,6 +154,7 @@ virtual std::string getDownloadDirectory();
|
|||||||
virtual std::string getPartialsDirectory();
|
virtual std::string getPartialsDirectory();
|
||||||
|
|
||||||
virtual bool getSharedDirectories(std::list<std::string> &dirs);
|
virtual bool getSharedDirectories(std::list<std::string> &dirs);
|
||||||
|
virtual bool setSharedDirectories(std::list<std::string> &dirs);
|
||||||
virtual bool addSharedDirectory(std::string dir);
|
virtual bool addSharedDirectory(std::string dir);
|
||||||
virtual bool removeSharedDirectory(std::string dir);
|
virtual bool removeSharedDirectory(std::string dir);
|
||||||
|
|
||||||
@ -203,15 +210,15 @@ virtual int check_dBUpdate();
|
|||||||
p3AuthMgr *mAuthMgr;
|
p3AuthMgr *mAuthMgr;
|
||||||
p3ConnectMgr *mConnMgr;
|
p3ConnectMgr *mConnMgr;
|
||||||
|
|
||||||
CacheStrapper *mCacheStrapper;
|
ftCacheStrapper *mCacheStrapper;
|
||||||
|
ftFiStore *mFiStore;
|
||||||
|
ftFiMonitor *mFiMon;
|
||||||
|
|
||||||
ftController *mFtController;
|
ftController *mFtController;
|
||||||
ftExtraList *mFtExtra;
|
ftExtraList *mFtExtra;
|
||||||
|
|
||||||
ftDataMultiplex *mFtDataplex;
|
ftDataMultiplex *mFtDataplex;
|
||||||
|
|
||||||
FileIndexStore *mFiStore;
|
|
||||||
FileIndexMonitor *mFiMon;
|
|
||||||
|
|
||||||
ftFileSearch *mFtSearch;
|
ftFileSearch *mFtSearch;
|
||||||
|
|
||||||
|
@ -38,16 +38,19 @@ extern RsFiles *rsFiles;
|
|||||||
|
|
||||||
class Expression;
|
class Expression;
|
||||||
|
|
||||||
const uint32_t RS_FILE_CTRL_PAUSE = 0x0100;
|
/* These are used mainly by ftController at the moment */
|
||||||
const uint32_t RS_FILE_CTRL_START = 0x0200;
|
const uint32_t RS_FILE_CTRL_PAUSE = 0x00000100;
|
||||||
|
const uint32_t RS_FILE_CTRL_START = 0x00000200;
|
||||||
|
|
||||||
const uint32_t RS_FILE_CTRL_TRICKLE = 0x0001;
|
const uint32_t RS_FILE_RATE_TRICKLE = 0x00000001;
|
||||||
const uint32_t RS_FILE_CTRL_SLOW = 0x0002;
|
const uint32_t RS_FILE_RATE_SLOW = 0x00000002;
|
||||||
const uint32_t RS_FILE_CTRL_STANDARD = 0x0003;
|
const uint32_t RS_FILE_RATE_STANDARD = 0x00000003;
|
||||||
const uint32_t RS_FILE_CTRL_FAST = 0x0004;
|
const uint32_t RS_FILE_RATE_FAST = 0x00000004;
|
||||||
const uint32_t RS_FILE_CTRL_STREAM_AUDIO = 0x0005;
|
const uint32_t RS_FILE_RATE_STREAM_AUDIO = 0x00000005;
|
||||||
const uint32_t RS_FILE_CTRL_STREAM_VIDEO = 0x0006;
|
const uint32_t RS_FILE_RATE_STREAM_VIDEO = 0x00000006;
|
||||||
|
|
||||||
|
const uint32_t RS_FILE_PEER_ONLINE = 0x00001000;
|
||||||
|
const uint32_t RS_FILE_PEER_OFFLINE = 0x00002000;
|
||||||
|
|
||||||
/************************************
|
/************************************
|
||||||
* Used To indicate where to search.
|
* Used To indicate where to search.
|
||||||
@ -86,8 +89,8 @@ virtual ~RsFiles() { return; }
|
|||||||
/***
|
/***
|
||||||
* Control of Downloads.
|
* Control of Downloads.
|
||||||
***/
|
***/
|
||||||
virtual bool FileRequest(std::string fname, std::string hash,
|
virtual bool FileRequest(std::string fname, std::string hash, uint32_t size,
|
||||||
uint32_t size, std::string dest, uint32_t flags) = 0;
|
std::string dest, uint32_t flags, std::list<std::string> srcIds) = 0;
|
||||||
virtual bool FileCancel(std::string hash) = 0;
|
virtual bool FileCancel(std::string hash) = 0;
|
||||||
virtual bool FileControl(std::string hash, uint32_t flags) = 0;
|
virtual bool FileControl(std::string hash, uint32_t flags) = 0;
|
||||||
virtual bool FileClearCompleted() = 0;
|
virtual bool FileClearCompleted() = 0;
|
||||||
|
@ -141,8 +141,8 @@ bool p3Files::FileControl(std::string hash, uint32_t flags)
|
|||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
|
|
||||||
bool p3Files::FileRequest(std::string fname, std::string hash,
|
bool p3Files::FileRequest(std::string fname, std::string hash, uint32_t size,
|
||||||
uint32_t size, std::string dest, uint32_t flags)
|
std::string dest, uint32_t flags, std::list<std::string> srcIds)
|
||||||
{
|
{
|
||||||
lockRsCore(); /* LOCKED */
|
lockRsCore(); /* LOCKED */
|
||||||
|
|
||||||
|
@ -57,8 +57,9 @@ virtual bool FileUploads(std::list<std::string> &hashs);
|
|||||||
virtual bool FileDetails(std::string hash, uint32_t hintflags, FileInfo &info);
|
virtual bool FileDetails(std::string hash, uint32_t hintflags, FileInfo &info);
|
||||||
|
|
||||||
/* 2) Control of Downloads. */
|
/* 2) Control of Downloads. */
|
||||||
virtual bool FileRequest(std::string fname, std::string hash,
|
virtual bool FileRequest(std::string fname, std::string hash, uint32_t size,
|
||||||
uint32_t size, std::string dest, uint32_t flags);
|
std::string dest, uint32_t flags,
|
||||||
|
std::list<std::string> srcIds);
|
||||||
virtual bool FileCancel(std::string hash);
|
virtual bool FileCancel(std::string hash);
|
||||||
virtual bool FileControl(std::string hash, uint32_t flags);
|
virtual bool FileControl(std::string hash, uint32_t flags);
|
||||||
virtual bool FileClearCompleted();
|
virtual bool FileClearCompleted();
|
||||||
|
Loading…
Reference in New Issue
Block a user