first part of peer sorting in friend server

This commit is contained in:
csoler 2021-11-05 16:52:00 +01:00
parent 62655779e5
commit af6dee088c
4 changed files with 94 additions and 8 deletions

View File

@ -32,8 +32,6 @@ FsBioInterface::FsBioInterface(int socket)
int FsBioInterface::tick() int FsBioInterface::tick()
{ {
std::cerr << "ticking FsNetworkInterface" << std::endl;
// 2 - read incoming data pending on existing connections // 2 - read incoming data pending on existing connections
char inBuffer[1025]; char inBuffer[1025];
@ -57,7 +55,7 @@ int FsBioInterface::tick()
return mTotalBufferBytes; return mTotalBufferBytes;
} }
std::cerr << "clintConnt: " << mCLintConnt << ", readbytes: " << readbytes << std::endl; RsDbg() << "clintConnt: " << mCLintConnt << ", readbytes: " << readbytes ;
// display some debug info // display some debug info
@ -77,7 +75,7 @@ int FsBioInterface::tick()
mTotalBufferBytes += readbytes; mTotalBufferBytes += readbytes;
mTotalReadBytes += readbytes; mTotalReadBytes += readbytes;
std::cerr << "Socket: " << mCLintConnt << ". Total read: " << mTotalReadBytes << ". Buffer size: " << mTotalBufferBytes << std::endl ; RsDbg() << "Socket: " << mCLintConnt << ". Total read: " << mTotalReadBytes << ". Buffer size: " << mTotalBufferBytes ;
} }
return mTotalBufferBytes; return mTotalBufferBytes;

View File

@ -168,6 +168,15 @@ struct t_RsGenericIdType
return ret; return ret;
} }
inline Id_t operator^ (const Id_t& fp) const
{
Id_t ret;
for(uint32_t i=0; i < ID_SIZE_IN_BYTES; ++i)
ret.bytes[i] = bytes[i] ^ fp.bytes[i];
return ret;
}
inline bool isNull() const inline bool isNull() const
{ {
for(uint32_t i=0; i < SIZE_IN_BYTES; ++i) for(uint32_t i=0; i < SIZE_IN_BYTES; ++i)

View File

@ -12,6 +12,7 @@
static const rstime_t MAXIMUM_PEER_INACTIVE_DELAY = 600; static const rstime_t MAXIMUM_PEER_INACTIVE_DELAY = 600;
static const rstime_t DELAY_BETWEEN_TWO_AUTOWASH = 60; static const rstime_t DELAY_BETWEEN_TWO_AUTOWASH = 60;
static const rstime_t DELAY_BETWEEN_TWO_DEBUG_PRINT = 10; static const rstime_t DELAY_BETWEEN_TWO_DEBUG_PRINT = 10;
static const uint32_t MAXIMUM_PEERS_TO_REQUEST = 10;
void FriendServer::threadTick() void FriendServer::threadTick()
{ {
@ -69,8 +70,6 @@ void FriendServer::handleClientPublish(const RsFriendServerClientPublishItem *it
RsDbg() << *item ; RsDbg() << *item ;
// First of all, read PGP key and short invites, parse them, and check that they contain the same information // First of all, read PGP key and short invites, parse them, and check that they contain the same information
RsPeerId pid;
RsPgpFingerprint fpr;
std::map<RsPeerId,PeerInfo>::iterator pi = handleIncomingClientData(item->pgp_public_key_b64,item->short_invite); std::map<RsPeerId,PeerInfo>::iterator pi = handleIncomingClientData(item->pgp_public_key_b64,item->short_invite);
@ -87,6 +86,9 @@ void FriendServer::handleClientPublish(const RsFriendServerClientPublishItem *it
mni->SendItem(sr_item); mni->SendItem(sr_item);
// Update the list of closest peers for all peers currently in the database.
updateClosestPeers(pi->first,pi->second.pgp_fingerprint);
} }
catch(std::exception& e) catch(std::exception& e)
{ {
@ -129,6 +131,30 @@ std::map<std::string, bool> FriendServer::computeListOfFriendInvites(uint32_t nb
break; break;
} }
// Strategy: we want to return the same set of friends for a given PGP profile key.
// Still, using some closest distance strategy, the n-closest peers for profile A is not the
// same set than the n-closest peers for profile B. We have multiple options:
//
// Option 1:
//
// (1) for each profile, keep the list of n-closest peers updated (when a new peer if added/removed all lists are updated)
//
// When a peer asks for k friends, read from (1), until the number of collected peers
// reaches the requested value. Then when a peer receives a connection request, ask the friend server if the
// peer has been sent your own cert.
//
// Option 2:
//
// (1) for each profile, keep the list of n-closest peers updated (when a new peer if added/removed all lists are updated)
// (2) for each profile, keep the list of which peers have been sent this profile already
//
// When a peer asks for k friends, read from (2) first, then (1), until the number of collected peers
// reaches the requested value.
//
// Drawbacks: it's not clear which list of friends you're going to get eventually, but in the end the peers that will be
// sent that are not in the n-closest list will need to be checked, so they will be known whatsoever.
//
return res; return res;
} }
@ -214,8 +240,26 @@ void FriendServer::handleClientRemove(const RsFriendServerClientRemoveItem *item
RsDbg() << " Nonce is correct: " << std::hex << item->nonce << std::dec << ". Removing peer " << item->peer_id ; RsDbg() << " Nonce is correct: " << std::hex << item->nonce << std::dec << ". Removing peer " << item->peer_id ;
mCurrentClientPeers.erase(it); mCurrentClientPeers.erase(it);
// Also remove that peer from all n-closest lists
for(auto& it:mCurrentClientPeers)
for(auto pit(it.second.closest_peers.begin());pit!=it.second.closest_peers.end();)
if(pit->second == item->peer_id)
{
auto tmp(pit);
++tmp;
it.second.closest_peers.erase(pit);
pit=tmp;
}
else
++pit;
} }
PeerInfo::PeerDistance FriendServer::computePeerDistance(const RsPgpFingerprint& p1,const RsPgpFingerprint& p2)
{
return (p1 ^ p2)^mRandomPeerBias;
}
FriendServer::FriendServer(const std::string& base_dir) FriendServer::FriendServer(const std::string& base_dir)
{ {
RsDbg() << "Creating friend server." ; RsDbg() << "Creating friend server." ;
@ -230,6 +274,10 @@ FriendServer::FriendServer(const std::string& base_dir)
std::string pgp_trustdb_path = RsDirUtil::makePath(base_dir,"pgp_trustdb") ; // not used. std::string pgp_trustdb_path = RsDirUtil::makePath(base_dir,"pgp_trustdb") ; // not used.
mPgpHandler = new PGPHandler(pgp_public_keyring_path,pgp_private_keyring_path,pgp_trustdb_path,pgp_lock_path); mPgpHandler = new PGPHandler(pgp_public_keyring_path,pgp_private_keyring_path,pgp_trustdb_path,pgp_lock_path);
// Random bias. Should be cryptographically safe.
mRandomPeerBias = RsPgpFingerprint::random();
} }
void FriendServer::run() void FriendServer::run()
@ -262,18 +310,38 @@ void FriendServer::autoWash()
RsDbg() << "done." ; RsDbg() << "done." ;
} }
void FriendServer::updateClosestPeers(const RsPeerId& pid,const RsPgpFingerprint& fpr)
{
for(auto& it:mCurrentClientPeers)
{
PeerInfo::PeerDistance d = computePeerDistance(fpr,it.second.pgp_fingerprint);
it.second.closest_peers.insert(std::make_pair(d,pid));
if(it.second.closest_peers.size() > MAXIMUM_PEERS_TO_REQUEST)
it.second.closest_peers.erase(std::prev(it.second.closest_peers.end()));
}
}
void FriendServer::debugPrint() void FriendServer::debugPrint()
{ {
RsDbg() << "========== FriendServer statistics ============"; RsDbg() << "========== FriendServer statistics ============";
RsDbg() << " Base directory: "<< mBaseDirectory; RsDbg() << " Base directory: "<< mBaseDirectory;
RsDbg() << " Random peer bias: "<< mRandomPeerBias;
RsDbg() << " Network interface: "; RsDbg() << " Network interface: ";
RsDbg() << " Current peers: " << mCurrentClientPeers.size() ; RsDbg() << " Max peers in n-closest list: " << MAXIMUM_PEERS_TO_REQUEST;
RsDbg() << " Current active peers: " << mCurrentClientPeers.size() ;
rstime_t now = time(nullptr); rstime_t now = time(nullptr);
for(auto& it:mCurrentClientPeers) for(const auto& it:mCurrentClientPeers)
{
RsDbg() << " " << it.first << ": nonce=" << std::hex << it.second.last_nonce << std::dec << ", last contact: " << now - it.second.last_connection_TS << " secs ago."; RsDbg() << " " << it.first << ": nonce=" << std::hex << it.second.last_nonce << std::dec << ", last contact: " << now - it.second.last_connection_TS << " secs ago.";
for(const auto& pit:it.second.closest_peers)
RsDbg() << " " << pit.second << " distance=" << pit.first ;
}
RsDbg() << "==============================================="; RsDbg() << "===============================================";
} }

View File

@ -32,10 +32,14 @@ class RsFriendServerClientPublishItem;
struct PeerInfo struct PeerInfo
{ {
typedef RsPgpFingerprint PeerDistance;
RsPgpFingerprint pgp_fingerprint; RsPgpFingerprint pgp_fingerprint;
std::string short_certificate; std::string short_certificate;
rstime_t last_connection_TS; rstime_t last_connection_TS;
uint64_t last_nonce; uint64_t last_nonce;
std::map<PeerDistance,RsPeerId> closest_peers;
}; };
class FriendServer : public RsTickingThread class FriendServer : public RsTickingThread
@ -54,12 +58,18 @@ private:
void handleClientRemove(const RsFriendServerClientRemoveItem *item); void handleClientRemove(const RsFriendServerClientRemoveItem *item);
void handleClientPublish(const RsFriendServerClientPublishItem *item); void handleClientPublish(const RsFriendServerClientPublishItem *item);
// Updates for each peer in the database, the list of closest peers w.r.t. some arbitrary distance.
void updateClosestPeers(const RsPeerId& pid,const RsPgpFingerprint& fpr);
// Adds the incoming peer data to the list of current clients and returns the // Adds the incoming peer data to the list of current clients and returns the
std::map<RsPeerId,PeerInfo>::iterator handleIncomingClientData(const std::string& pgp_public_key_b64,const std::string& short_invite_b64); std::map<RsPeerId,PeerInfo>::iterator handleIncomingClientData(const std::string& pgp_public_key_b64,const std::string& short_invite_b64);
// Computes the appropriate list of short invites to send to a given peer. // Computes the appropriate list of short invites to send to a given peer.
std::map<std::string,bool> computeListOfFriendInvites(uint32_t nb_reqs_invites,const RsPeerId& pid,const RsPgpFingerprint& fpr); std::map<std::string,bool> computeListOfFriendInvites(uint32_t nb_reqs_invites,const RsPeerId& pid,const RsPgpFingerprint& fpr);
// Compute the distance between peers using the random bias (It's not really a distance though. I'm not sure about the triangular inequality).
PeerInfo::PeerDistance computePeerDistance(const RsPgpFingerprint &p1, const RsPgpFingerprint &p2);
void autoWash(); void autoWash();
void debugPrint(); void debugPrint();
@ -69,6 +79,7 @@ private:
PGPHandler *mPgpHandler; PGPHandler *mPgpHandler;
std::string mBaseDirectory; std::string mBaseDirectory;
RsPgpFingerprint mRandomPeerBias;
std::map<RsPeerId, PeerInfo> mCurrentClientPeers; std::map<RsPeerId, PeerInfo> mCurrentClientPeers;
}; };