mirror of
https://github.com/RetroShare/RetroShare.git
synced 2024-10-01 02:35:48 -04:00
7a2ae6e979
Removed SearchInterface that was an historical remain that make no sense anymore Fixed some clang warning Fix includes on some files
153 lines
3.8 KiB
C++
153 lines
3.8 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 <stdint.h> // for uint32_t
|
|
#include <time.h> // for time_t, NULL
|
|
#include <list> // for list
|
|
#include <map> // for map
|
|
|
|
#include "pqi/pqi.h" // for P3Interface, pqiPublisher
|
|
#include "retroshare/rstypes.h" // for RsPeerId
|
|
#include "util/rsthreads.h" // for RsStackMutex, RsMutex
|
|
|
|
class PQInterface;
|
|
class RSTrafficClue;
|
|
class RsBwRates;
|
|
class RsItem;
|
|
class RsRawItem;
|
|
|
|
class SearchModule
|
|
{
|
|
public:
|
|
SearchModule() : pqi(NULL) {}
|
|
|
|
RsPeerId peerid ;
|
|
PQInterface *pqi;
|
|
};
|
|
|
|
// Presents a P3 Face to the world!
|
|
// and funnels data through to a PQInterface.
|
|
//
|
|
class pqihandler: public P3Interface, public pqiPublisher
|
|
{
|
|
public:
|
|
pqihandler();
|
|
|
|
/**** Overloaded from pqiPublisher ****/
|
|
virtual bool sendItem(RsRawItem *item)
|
|
{
|
|
return SendRsRawItem(item);
|
|
}
|
|
|
|
bool AddSearchModule(SearchModule *mod);
|
|
bool RemoveSearchModule(SearchModule *mod);
|
|
|
|
// Rest of P3Interface
|
|
virtual int tick();
|
|
virtual int status();
|
|
|
|
// Service Data Interface
|
|
virtual int SendRsRawItem(RsRawItem *);
|
|
#ifdef TO_BE_REMOVED
|
|
virtual RsRawItem *GetRsRawItem();
|
|
#endif
|
|
|
|
// rate control.
|
|
//void setMaxRate(const RsPeerId& pid,bool in, uint32_t val_kBs);
|
|
void setMaxRate(bool in, float val);
|
|
float getMaxRate(bool in);
|
|
|
|
void setMaxRates(const RsPeerId& pid,bool in,float val) ;
|
|
float getMaxRates(const RsPeerId& pid,bool in) ;
|
|
|
|
void getCurrentRates(float &in, float &out);
|
|
|
|
// TESTING INTERFACE.
|
|
int ExtractRates(std::map<RsPeerId, RsBwRates> &ratemap, RsBwRates &totals);
|
|
int ExtractTrafficInfo(std::list<RSTrafficClue> &out_lst, std::list<RSTrafficClue> &in_lst);
|
|
|
|
protected:
|
|
/* check to be overloaded by those that can
|
|
* generates warnings otherwise
|
|
*/
|
|
|
|
int locked_HandleRsItem(RsItem *ns, uint32_t& size);
|
|
bool queueOutRsItem(RsItem *) ;
|
|
|
|
#ifdef TO_BE_REMOVED
|
|
int locked_GetItems();
|
|
void locked_SortnStoreItem(RsItem *item);
|
|
#endif
|
|
|
|
RsMutex coreMtx; /* MUTEX */
|
|
|
|
std::map<RsPeerId, SearchModule *> mods;
|
|
|
|
std::list<RsItem *> 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;
|
|
|
|
uint32_t nb_ticks ;
|
|
time_t last_m ;
|
|
time_t mLastRateCapUpdate ;
|
|
float ticks_per_sec ;
|
|
};
|
|
|
|
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
|