mirror of
https://github.com/RetroShare/RetroShare.git
synced 2024-10-01 02:35:48 -04:00
646603d71d
Merge branch 'maxUploadRate' git-svn-id: http://svn.code.sf.net/p/retroshare/code/trunk@1293 b45a01b8-16f6-495d-af2f-9b41ad6348cc
161 lines
3.7 KiB
C++
161 lines
3.7 KiB
C++
/*
|
|
* "$Id: pqihandler.h,v 1.10 2007-03-31 09:41:32 rmf24 Exp $"
|
|
*
|
|
* 3P/PQI network interface for RetroShare.
|
|
*
|
|
* Copyright 2004-2006 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 MRK_PQI_HANDLER_HEADER
|
|
#define MRK_PQI_HANDLER_HEADER
|
|
|
|
#include "pqi/pqi.h"
|
|
#include "pqi/pqisecurity.h"
|
|
|
|
#include "util/rsthreads.h"
|
|
|
|
#include <map>
|
|
#include <list>
|
|
|
|
class SearchModule
|
|
{
|
|
public:
|
|
std::string peerid;
|
|
PQInterface *pqi;
|
|
SecurityPolicy *sp;
|
|
};
|
|
|
|
// Presents a P3 Face to the world!
|
|
// and funnels data through to a PQInterface.
|
|
//
|
|
class pqihandler: public P3Interface
|
|
{
|
|
public:
|
|
pqihandler(SecurityPolicy *Global);
|
|
bool AddSearchModule(SearchModule *mod);
|
|
bool RemoveSearchModule(SearchModule *mod);
|
|
|
|
// P3Interface.
|
|
virtual int SearchSpecific(RsCacheRequest *ns);
|
|
virtual int SendSearchResult(RsCacheItem *);
|
|
|
|
// inputs.
|
|
virtual RsCacheRequest * RequestedSearch();
|
|
virtual RsCacheItem * GetSearchResult();
|
|
|
|
// file i/o
|
|
virtual int SendFileRequest(RsFileRequest *ns);
|
|
virtual int SendFileData(RsFileData *ns);
|
|
virtual RsFileRequest * GetFileRequest();
|
|
virtual RsFileData * GetFileData();
|
|
|
|
// Rest of P3Interface
|
|
virtual int tick();
|
|
virtual int status();
|
|
|
|
// Service Data Interface
|
|
virtual int SendRsRawItem(RsRawItem *);
|
|
virtual RsRawItem *GetRsRawItem();
|
|
|
|
// rate control.
|
|
//indiv rate is deprecated
|
|
//void setMaxIndivRate(bool in, float val);
|
|
//float getMaxIndivRate(bool in);
|
|
void setMaxRate(bool in, float val);
|
|
float getMaxRate(bool in);
|
|
|
|
void getCurrentRates(float &in, float &out);
|
|
|
|
|
|
protected:
|
|
/* check to be overloaded by those that can
|
|
* generates warnings otherwise
|
|
*/
|
|
|
|
int HandleRsItem(RsItem *ns, int allowglobal);
|
|
|
|
virtual int locked_checkOutgoingRsItem(RsItem *item, int global);
|
|
int locked_GetItems();
|
|
void locked_SortnStoreItem(RsItem *item);
|
|
|
|
RsMutex coreMtx; /* MUTEX */
|
|
|
|
std::map<std::string, SearchModule *> mods;
|
|
SecurityPolicy *globsec;
|
|
|
|
// Temporary storage...
|
|
std::list<RsItem *> in_result, in_search,
|
|
in_request, in_data, in_service;
|
|
|
|
private:
|
|
|
|
// rate control.
|
|
int UpdateRates();
|
|
void locked_StoreCurrentRates(float in, float out);
|
|
|
|
float rateIndiv_in;
|
|
float rateIndiv_out;
|
|
float rateMax_in;
|
|
float rateMax_out;
|
|
|
|
float rateTotal_in;
|
|
float rateTotal_out;
|
|
};
|
|
|
|
//inline void pqihandler::setMaxIndivRate(bool in, float val)
|
|
//{
|
|
// RsStackMutex stack(coreMtx); /**************** LOCKED MUTEX ****************/
|
|
// if (in)
|
|
// rateIndiv_in = val;
|
|
// else
|
|
// rateIndiv_out = val;
|
|
// return;
|
|
//}
|
|
//
|
|
//inline float pqihandler::getMaxIndivRate(bool in)
|
|
//{
|
|
// RsStackMutex stack(coreMtx); /**************** LOCKED MUTEX ****************/
|
|
// if (in)
|
|
// return rateIndiv_in;
|
|
// else
|
|
// return rateIndiv_out;
|
|
//}
|
|
|
|
inline void pqihandler::setMaxRate(bool in, float val)
|
|
{
|
|
RsStackMutex stack(coreMtx); /**************** LOCKED MUTEX ****************/
|
|
if (in)
|
|
rateMax_in = val;
|
|
else
|
|
rateMax_out = val;
|
|
return;
|
|
}
|
|
|
|
inline float pqihandler::getMaxRate(bool in)
|
|
{
|
|
RsStackMutex stack(coreMtx); /**************** LOCKED MUTEX ****************/
|
|
if (in)
|
|
return rateMax_in;
|
|
else
|
|
return rateMax_out;
|
|
}
|
|
|
|
#endif // MRK_PQI_HANDLER_HEADER
|