RetroShare/tests/librssimulator/peer/FakeLinkMgr.h
2018-11-11 14:24:45 +01:00

92 lines
3.2 KiB
C++

/*******************************************************************************
* librssimulator/peer/: FakeLinkMgr.h *
* *
* libretroshare: retroshare core library *
* *
* Copyright (C) 2018, Retroshare team <retroshare.team@gmailcom> *
* *
* This program is free software: you can redistribute it and/or modify *
* it under the terms of the GNU Affero General Public License as *
* published by the Free Software Foundation, either version 3 of the *
* License, or (at your option) any later version. *
* *
* This program 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 Lesser General Public License for more details. *
* *
* You should have received a copy of the GNU Lesser General Public License *
* along with this program. If not, see <https://www.gnu.org/licenses/>. *
* *
******************************************************************************/
#pragma once
#include <iostream>
#include <list>
#include <retroshare/rsids.h>
#include <pqi/p3linkmgr.h>
class FakePeerListStatus
{
public:
bool mOnline;
};
class FakeLinkMgr: public p3LinkMgrIMPL
{
public:
FakeLinkMgr(const RsPeerId& own_id,const std::list<RsPeerId>& friends, bool peersOnline)
: p3LinkMgrIMPL(NULL,NULL), mOwnId(own_id), mFriends()
{
std::list<RsPeerId>::const_iterator it;
for(it = friends.begin(); it != friends.end(); it++)
{
setOnlineStatus(*it, peersOnline);
}
}
virtual const RsPeerId& getOwnId() { return mOwnId; }
virtual void getOnlineList(std::list<RsPeerId>& lst)
{
std::map<RsPeerId, FakePeerListStatus>::iterator it;
for(it = mFriends.begin(); it != mFriends.end(); it++)
{
if (it->second.mOnline)
{
lst.push_back(it->first);
}
}
}
virtual void getFriendList(std::list<RsPeerId> &ssl_peers)
{
std::map<RsPeerId, FakePeerListStatus>::iterator it;
for(it = mFriends.begin(); it != mFriends.end(); it++)
{
ssl_peers.push_back(it->first);
}
}
virtual uint32_t getLinkType(const RsPeerId&) { return RS_NET_CONN_TCP_ALL | RS_NET_CONN_SPEED_NORMAL; }
virtual bool getPeerName(const RsPeerId &ssl_id, std::string &name) { name = ssl_id.toStdString() ; return true ;}
// functions to manipulate status.
virtual void setOnlineStatus(RsPeerId id, bool online)
{
FakePeerListStatus status;
status.mOnline = online;
mFriends[id] = status;
}
private:
RsPeerId mOwnId;
std::map<RsPeerId, FakePeerListStatus> mFriends;
};