mirror of
https://github.com/RetroShare/RetroShare.git
synced 2024-10-01 02:35:48 -04:00
80f4686100
git-svn-id: http://svn.code.sf.net/p/retroshare/code/trunk@791 b45a01b8-16f6-495d-af2f-9b41ad6348cc
465 lines
10 KiB
C++
465 lines
10 KiB
C++
/*
|
|
* libretroshare/src/ft: ftserver3test.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".
|
|
*
|
|
*/
|
|
|
|
/*
|
|
* ftServer3Test - Test of the file transfer from a server level.
|
|
* Steps:
|
|
* 1) load shared directories into others, and let them be
|
|
transferred between clients.
|
|
* 2) search for local item on others.
|
|
* 3) request item on load server.
|
|
should transfer from all others simultaneously.
|
|
*/
|
|
|
|
#ifdef WIN32
|
|
#include "util/rswin.h"
|
|
#endif
|
|
|
|
#include "rsiface/rsexpr.h"
|
|
|
|
#include "ft/ftserver.h"
|
|
|
|
#include "ft/ftextralist.h"
|
|
#include "ft/ftdatamultiplex.h"
|
|
#include "ft/ftfilesearch.h"
|
|
|
|
#include "pqi/p3authmgr.h"
|
|
#include "pqi/p3connmgr.h"
|
|
|
|
#include "util/rsdebug.h"
|
|
|
|
#include "ft/pqitestor.h"
|
|
#include "util/rsdir.h"
|
|
#include "util/utest.h"
|
|
|
|
|
|
#include <sstream>
|
|
|
|
|
|
class TestData
|
|
{
|
|
public:
|
|
|
|
ftServer *loadServer;
|
|
std::list<ftServer *> otherServers;
|
|
std::list<std::string> extraList;
|
|
};
|
|
|
|
|
|
extern "C" void *do_server_test_thread(void *p);
|
|
|
|
|
|
void usage(char *name)
|
|
{
|
|
std::cerr << "Usage: " << name << " [-soa] [-p <peerId>] [-d <debugLvl>] [-e <extrafile>] [<path> [<path2> ... ]] ";
|
|
std::cerr << std::endl;
|
|
}
|
|
|
|
int main(int argc, char **argv)
|
|
{
|
|
int c;
|
|
uint32_t period = 1;
|
|
uint32_t debugLevel = 5;
|
|
bool debugStderr = true;
|
|
bool loadAll = false;
|
|
bool loadOthers = false;
|
|
|
|
std::list<std::string> fileList;
|
|
std::list<std::string> extraList;
|
|
std::list<std::string> peerIds;
|
|
std::map<std::string, ftServer *> mFtServers;
|
|
std::map<std::string, p3ConnectMgr *> mConnMgrs;
|
|
|
|
ftServer *mLoadServer = NULL;
|
|
std::list<ftServer *> mOtherServers;
|
|
std::list<std::string>::iterator eit;
|
|
|
|
#ifdef PTW32_STATIC_LIB
|
|
pthread_win32_process_attach_np();
|
|
#endif
|
|
|
|
#ifdef WIN32
|
|
// Windows Networking Init.
|
|
WORD wVerReq = MAKEWORD(2,2);
|
|
WSADATA wsaData;
|
|
|
|
if (0 != WSAStartup(wVerReq, &wsaData))
|
|
{
|
|
std::cerr << "Failed to Startup Windows Networking";
|
|
std::cerr << std::endl;
|
|
}
|
|
else
|
|
{
|
|
std::cerr << "Started Windows Networking";
|
|
std::cerr << std::endl;
|
|
}
|
|
|
|
#endif
|
|
|
|
|
|
while(-1 != (c = getopt(argc, argv, "aosd:p:e:")))
|
|
{
|
|
switch (c)
|
|
{
|
|
case 'p':
|
|
peerIds.push_back(optarg);
|
|
break;
|
|
case 'd':
|
|
debugLevel = atoi(optarg);
|
|
break;
|
|
case 's':
|
|
debugStderr = true;
|
|
break;
|
|
case 'e':
|
|
extraList.push_back(optarg);
|
|
break;
|
|
case 'a':
|
|
loadAll = true;
|
|
break;
|
|
case 'o':
|
|
loadOthers = true;
|
|
break;
|
|
default:
|
|
usage(argv[0]);
|
|
break;
|
|
}
|
|
}
|
|
|
|
/* do logging */
|
|
setOutputLevel(debugLevel);
|
|
|
|
if (optind >= argc)
|
|
{
|
|
std::cerr << "Missing Shared Directories" << std::endl;
|
|
usage(argv[0]);
|
|
}
|
|
|
|
for(; optind < argc; optind++)
|
|
{
|
|
std::cerr << "Adding: " << argv[optind] << std::endl;
|
|
fileList.push_back(std::string(argv[optind]));
|
|
}
|
|
|
|
/* We need to setup a series 2 - 4 different ftServers....
|
|
*
|
|
* Each one needs:
|
|
*
|
|
*
|
|
* A List of peerIds...
|
|
*/
|
|
|
|
std::list<std::string>::const_iterator it, jit;
|
|
|
|
std::list<pqiAuthDetails> baseFriendList, friendList;
|
|
std::list<pqiAuthDetails>::iterator fit;
|
|
|
|
/* Add in Serialiser Test
|
|
*/
|
|
RsSerialiser *rss = new RsSerialiser();
|
|
rss->addSerialType(new RsFileItemSerialiser());
|
|
rss->addSerialType(new RsCacheItemSerialiser());
|
|
rss->addSerialType(new RsServiceSerialiser());
|
|
|
|
P3Hub *testHub = new P3Hub(0, rss);
|
|
testHub->start();
|
|
|
|
/* Setup Base Friend Info */
|
|
for(it = peerIds.begin(); it != peerIds.end(); it++)
|
|
{
|
|
pqiAuthDetails pad;
|
|
pad.id = *it;
|
|
pad.name = *it;
|
|
pad.trustLvl = 5;
|
|
pad.ownsign = true;
|
|
pad.trusted = false;
|
|
|
|
baseFriendList.push_back(pad);
|
|
|
|
std::cerr << "ftserver1test::setup peer: " << *it;
|
|
std::cerr << std::endl;
|
|
}
|
|
|
|
std::ostringstream pname;
|
|
pname << "/tmp/rstst-" << time(NULL);
|
|
|
|
std::string basepath = pname.str();
|
|
RsDirUtil::checkCreateDirectory(basepath);
|
|
|
|
|
|
|
|
for(it = peerIds.begin(); it != peerIds.end(); it++)
|
|
{
|
|
friendList = baseFriendList;
|
|
/* remove current one */
|
|
for(fit = friendList.begin(); fit != friendList.end(); fit++)
|
|
{
|
|
if (fit->id == *it)
|
|
{
|
|
friendList.erase(fit);
|
|
break;
|
|
}
|
|
}
|
|
|
|
p3AuthMgr *authMgr = new p3DummyAuthMgr(*it, friendList);
|
|
p3ConnectMgr *connMgr = new p3ConnectMgr(authMgr);
|
|
mConnMgrs[*it] = connMgr;
|
|
|
|
|
|
for(fit = friendList.begin(); fit != friendList.end(); fit++)
|
|
{
|
|
/* add as peer to authMgr */
|
|
connMgr->addFriend(fit->id);
|
|
}
|
|
|
|
P3Pipe *pipe = new P3Pipe(); //(*it);
|
|
|
|
/* add server */
|
|
ftServer *server;
|
|
server = new ftServer(authMgr, connMgr);
|
|
mFtServers[*it] = server;
|
|
bool isOther;
|
|
if (!mLoadServer)
|
|
{
|
|
mLoadServer = server;
|
|
isOther = false;
|
|
}
|
|
else
|
|
{
|
|
mOtherServers.push_back(server);
|
|
isOther = true;
|
|
}
|
|
|
|
|
|
server->setP3Interface(pipe);
|
|
|
|
std::string configpath = basepath + "/" + *it;
|
|
RsDirUtil::checkCreateDirectory(configpath);
|
|
|
|
std::string cachepath = configpath + "/cache";
|
|
RsDirUtil::checkCreateDirectory(cachepath);
|
|
|
|
std::string partialspath = configpath + "/partials";
|
|
RsDirUtil::checkCreateDirectory(partialspath);
|
|
|
|
std::string downloadpath = configpath + "/downloads";
|
|
RsDirUtil::checkCreateDirectory(downloadpath);
|
|
|
|
std::string localpath = cachepath + "/local";
|
|
RsDirUtil::checkCreateDirectory(localpath);
|
|
|
|
std::string remotepath = cachepath + "/remote";
|
|
RsDirUtil::checkCreateDirectory(remotepath);
|
|
|
|
server->setConfigDirectory(configpath);
|
|
|
|
NotifyBase *base = NULL;
|
|
server->SetupFtServer(base);
|
|
|
|
testHub->addP3Pipe(*it, pipe, connMgr);
|
|
server->StartupThreads();
|
|
|
|
/* setup any extra bits */
|
|
server->setPartialsDirectory(partialspath);
|
|
server->setDownloadDirectory(downloadpath);
|
|
|
|
if ((loadAll) || (isOther && loadOthers))
|
|
{
|
|
server->setSharedDirectories(fileList);
|
|
for(eit = extraList.begin(); eit != extraList.end(); eit++)
|
|
{
|
|
server->ExtraFileHash(*eit, 3600, 0);
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
if ((mLoadServer) && (!loadOthers))
|
|
{
|
|
mLoadServer->setSharedDirectories(fileList);
|
|
for(eit = extraList.begin(); eit != extraList.end(); eit++)
|
|
{
|
|
mLoadServer->ExtraFileHash(*eit, 3600, 0);
|
|
}
|
|
}
|
|
|
|
|
|
/* stick your real test here */
|
|
std::map<std::string, ftServer *>::iterator sit;
|
|
std::map<std::string, p3ConnectMgr *>::iterator cit;
|
|
|
|
/* Start up test thread */
|
|
pthread_t tid;
|
|
TestData *mFt = new TestData;
|
|
|
|
/* set data */
|
|
mFt->loadServer = mLoadServer;
|
|
mFt->otherServers = mOtherServers;
|
|
mFt->extraList = extraList;
|
|
|
|
void *data = (void *) mFt;
|
|
pthread_create(&tid, 0, &do_server_test_thread, data);
|
|
pthread_detach(tid); /* so memory is reclaimed in linux */
|
|
|
|
while(1)
|
|
{
|
|
//std::cerr << "ftserver3test::sleep()";
|
|
//std::cerr << std::endl;
|
|
sleep(1);
|
|
|
|
/* tick the connmgrs */
|
|
for(sit = mFtServers.begin(); sit != mFtServers.end(); sit++)
|
|
{
|
|
/* update */
|
|
(sit->second)->tick();
|
|
}
|
|
|
|
for(cit = mConnMgrs.begin(); cit != mConnMgrs.end(); cit++)
|
|
{
|
|
/* update */
|
|
(cit->second)->tick();
|
|
}
|
|
}
|
|
}
|
|
|
|
/* So our actual test can run here.....
|
|
*
|
|
*/
|
|
|
|
INITTEST();
|
|
|
|
void *do_server_test_thread(void *data)
|
|
{
|
|
TestData *mFt = (TestData *) data;
|
|
time_t startTS = time(NULL);
|
|
|
|
std::cerr << "do_server_test_thread() running";
|
|
std::cerr << std::endl;
|
|
|
|
/* search Others for a suitable file
|
|
* (Tests GUI search functionality)
|
|
*/
|
|
if (mFt->otherServers.size() < 1)
|
|
{
|
|
std::cerr << "no Other Servers to search on";
|
|
std::cerr << std::endl;
|
|
exit(1);
|
|
return NULL;
|
|
}
|
|
|
|
|
|
for(int i = 0; i < 90; i++)
|
|
{
|
|
int age = time(NULL) - startTS;
|
|
std::cerr << "Waited " << age << " seconds to share caches";
|
|
std::cerr << std::endl;
|
|
sleep(1);
|
|
}
|
|
|
|
ftServer *oServer = *(mFt->otherServers.begin());
|
|
std::string oId = oServer->OwnId();
|
|
|
|
/* create Expression */
|
|
uint64_t minFileSize = 100000;
|
|
//SizeExpression se(Greater, minFileSize);
|
|
SizeExpression se(Smaller, minFileSize);
|
|
Expression *expr = &se;
|
|
|
|
std::list<FileDetail> results;
|
|
std::list<FileDetail>::iterator it;
|
|
|
|
oServer->SearchBoolExp(expr, results);
|
|
|
|
if (results.size() < 1)
|
|
{
|
|
std::cerr << "no Shared Files > " << minFileSize;
|
|
std::cerr << std::endl;
|
|
exit(1);
|
|
return NULL;
|
|
}
|
|
|
|
/* find the first remote entry */
|
|
FileDetail sFile;
|
|
bool foundFile = false;
|
|
for(it = results.begin();
|
|
(it != results.end()); it++)
|
|
{
|
|
std::cerr << "Shared File: " << it->name;
|
|
std::cerr << std::endl;
|
|
|
|
if (!foundFile)
|
|
{
|
|
if (it->id != mFt->loadServer->OwnId())
|
|
{
|
|
std::cerr << "Selected: " << it->name;
|
|
std::cerr << std::endl;
|
|
foundFile = true;
|
|
sFile = *it;
|
|
}
|
|
else
|
|
{
|
|
std::cerr << "LoadId: ";
|
|
std::cerr << mFt->loadServer->OwnId();
|
|
std::cerr << "FileId: ";
|
|
std::cerr << it->id;
|
|
std::cerr << std::endl;
|
|
}
|
|
}
|
|
}
|
|
|
|
if (!foundFile)
|
|
{
|
|
std::cerr << "Not Found Suitable File";
|
|
std::cerr << std::endl;
|
|
}
|
|
|
|
|
|
/*** Now Download it! ***/
|
|
std::list<std::string> srcIds;
|
|
//srcIds.push_back(sFile.id);
|
|
// Don't add srcId - to test whether the search works - or not
|
|
//srcIds.push_back(oId);
|
|
if (foundFile)
|
|
{
|
|
mFt->loadServer->FileRequest(sFile.name, sFile.hash,
|
|
sFile.size, "", 0, srcIds);
|
|
}
|
|
|
|
/* Give it a while to transfer */
|
|
for(int i = 0; i < 100; i++)
|
|
{
|
|
int age = time(NULL) - startTS;
|
|
std::cerr << "Waited " << age << " seconds for tranfer";
|
|
std::cerr << std::endl;
|
|
sleep(1);
|
|
}
|
|
|
|
FINALREPORT("Shared Directories, Bool Search, multi-source transfers");
|
|
exit(1);
|
|
}
|
|
|
|
|
|
|
|
|