2013-11-02 10:35:33 -04:00
/*
* libretroshare / src / services : p3grouter . cc
*
* Services for RetroShare .
*
* Copyright 2013 by Cyril Soler
*
* 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 " csoler@users.sourceforge.net " .
*
*/
2014-03-17 16:56:06 -04:00
////////////////////////////////////////////////////////////////////////////////////////////////////////////
//
// Decentralized routing
// =====================
//
// Use cases:
// - Peer A asks for B's key, for which he has the signature, or the ID.
// - Peer A wants to send a private msg to peer C, for which he has the public key
// - Peer A wants to contact a channel's owner, a group owner, a forum owner, etc.
// - Peer C needs to route msg/key requests from unknown peer, to unknown peer so that the information
// eventually reach their destination.
//
// Main idea: Each peer holds a local routing table, a matrix with probabilities that each friend
// is a correct path for a given key ID.
//
// The routing tables are updated as messages go back and forth. Successful
// interactions feed the routing table with information of where to route the
// packets.
//
// The routing is kept probabilistic, meaning that the optimal route is not
// always chosen, but the randomness helps updating the routing probabilities.
//
// Services that might use the router (All services really...)
// - Identity manager (p3Identity)
// - asks identities i.e. RSA public keys (i.e. sends dentity requests through router)
// - Messenger
// - sends/receives messages to distant peers
// - Channels, forums, posted, etc.
// - send messages to the origin of the channel/forum/posted
//
// GUI
// - a debug panel should show the routing info: probabilities for all known IDs
// - routing probabilities for a given ID accordign to who's connected
//
// Decentralized routing algorithm:
// - tick() method
// * calls autoWash(), send() and receive()
//
// - message passing
// - upward:
// * Forward msg to friends according to probabilities.
// * If all equal, send to all friends (or a rando subset of them).
// * keep the local routing info in a cache that is saved (Which peer issued the msg)
// - which probability was used to chose this friend (will be useful
// to compute the routing contribution if the msg is ACK-ed)
2014-03-29 10:18:05 -04:00
//
// Two probabilities are computed:
// - routing probabilities among connected friends
// * this is computed by the routing matrix
// - branching factor N
// * depends on the depth of the items
// * depends on the distribution of probabilities (min and max)
//
// Once computed,
// - the item is forwarded randomly to N peers drawn from the list of connected peers with the given probabilities.
// - the depth of the item is incremented randomly
2014-03-17 16:56:06 -04:00
//
// - downward: look into routing cache. If info not present, drop the item.
// Forward item into stored direction.
//
// - routing probability computation: count number of times a reliable info is obtained from
// which direction for which identity
// * the count is a floating point number, since weights can be assigned to each info
// (especially for importance sampling)
// * init: all friends have equal count of 0 (or 1, well, we'll have to make this right).
// * We use importance sampling, meaning that when peer relays a msg from ID:
// count[ID, peer] += 1.0 / importance
//
// ... where importance was the probability of chosing peer for the
// route upward.
//
// * probability of forward is proportional to count.
//
// - routing cache
// * this cache stores messages IDs (like turtle router) but is saved on disk
// * it is used to remember where to send back responses to messages, and
// with what probability the route was chosen.
// * cache items have a TTL and the cache is cleaned regularly.
//
// - routing matrix
// * the structure is fed by other services, when they receive key IDs.
// * stores for each identity the count of how many times each peer gave reliable info for that ID.
// That information should be enough to route packets in the correct direction.
// * saved to disk.
// * all contributions should have a time stamp. Regularly, the oldest contributions are removed.
//
// struct RoutingMatrixHitEntry
// {
// float weight ;
// time_t time_stamp ;
// }
// typedef std::map<std::string,std::list<RoutingMatrixHitEntry> > RSAKeyRoutingMap ;
//
// class RoutingMatrix
// {
// public:
// // Computes the routing probabilities for this id for the given list of friends.
// // the computation accounts for the time at which the info was received and the
// // weight of each routing hit record.
// //
// bool computeRoutingProbabilities(RSAKeyIDType id, const std::vector<SSLIdType>& friends,
// std::vector<float>& probas) const ;
//
// // remove oldest entries.
// bool autoWash() ;
//
// // Record one routing clue. The events can possibly be merged in time buckets.
// //
// bool addRoutingEvent(RSAKeyIDType id,const SSLIdType& which friend) ;
//
// private:
// std::map<RSAKeyIDType, RSAKeyRoutingMap> _known_keys ;
// };
//
// - Routed packets: we use a common packet type for all services:
//
// We need two abstract item types:
//
// * Data packet
// - packet unique ID (sha1, or uint64_t)
// - destination ID (for Dn packets, the destination is the source!)
// - packet type: Id request, Message, etc.
// - packet service ID (Can be messenging, channels, etc).
// - packet data (void* + size_t)
// - flags (such as ACK or response required, and packet direction)
2014-03-29 10:18:05 -04:00
// - routed directions and probabilities
2014-03-17 16:56:06 -04:00
// * ACK packet.
// - packet unique ID (the id of the corresponding data)
// - flags (reason for ACK. Could be data delivered, or error, too far, etc)
//
// - Data storage packets
// * We need storage packets for the matrix states.
// * General routing options info?
//
// - estimated memory cost
// For each identity, the matrix needs
// - hits for each friend peer with time stamps. That means 8 bytes per hit.
// That is for 1000 identities, having at most 100 hits each (We keep
// the hits below a maximum. 100 seems ok.), that is 1000*100*8 < 1MB. Not much.
//
// - Main difficulties:
// * have a good re-try strategy if a msg does not arrive.
// * handle peer availability. In forward mode: easy. In backward mode:
// difficult. We should wait, and send back the packet if possible.
// * robustness
// * security: avoid flooding, and message alteration.
//
// - Questions to be solved
// * how do we talk to other services?
// - keep a list of services?
//
// - in practice, services will need to send requests, and expect responses.
// * gxs (p3identity) asks for a key, gxs (p3identity) should get the key.
// * msg service wants to send a distant msg, or msg receives a distant msg.
//
// => we need abstract packets and service ids.
//
////////////////////////////////////////////////////////////////////////////////////////////////////////////
2013-11-02 10:35:33 -04:00
# include "util/rsrandom.h"
# include "pqi/p3linkmgr.h"
# include "serialiser/rsconfigitems.h"
# include "p3grouter.h"
# include "grouteritems.h"
2013-11-08 17:22:40 -05:00
# include "groutertypes.h"
2013-12-24 11:41:07 -05:00
# include "grouterclientservice.h"
2013-11-02 10:35:33 -04:00
2014-03-29 10:18:05 -04:00
const std : : string p3GRouter : : SERVICE_INFO_APP_NAME = " Global Router " ;
2013-11-02 10:35:33 -04:00
p3GRouter : : p3GRouter ( p3LinkMgr * lm )
2014-03-29 10:18:05 -04:00
: p3Service ( ) , p3Config ( ) , mLinkMgr ( lm ) , grMtx ( " GRouter " )
2013-11-02 10:35:33 -04:00
{
addSerialType ( new RsGRouterSerialiser ( ) ) ;
2013-11-05 16:15:26 -05:00
2013-12-28 16:47:15 -05:00
_changed = false ;
2013-11-02 10:35:33 -04:00
}
int p3GRouter : : tick ( )
{
static time_t last_autowash_time = 0 ;
static time_t last_publish_campaign_time = 0 ;
2013-11-05 16:15:26 -05:00
static time_t last_debug_output_time = 0 ;
2013-12-28 16:47:15 -05:00
static time_t last_config_changed = 0 ;
2013-11-02 10:35:33 -04:00
time_t now = time ( NULL ) ;
if ( now > last_autowash_time + RS_GROUTER_AUTOWASH_PERIOD )
{
2013-11-05 16:15:26 -05:00
// route pending objects
//
2013-12-22 12:04:13 -05:00
routePendingObjects ( ) ;
2013-11-05 16:15:26 -05:00
2013-11-02 10:35:33 -04:00
last_autowash_time = now ;
autoWash ( ) ;
}
2013-11-05 16:15:26 -05:00
// Handle incoming items
2013-11-02 10:35:33 -04:00
//
handleIncoming ( ) ;
2013-11-05 16:15:26 -05:00
// Advertise published keys
2013-11-02 10:35:33 -04:00
//
if ( now > last_publish_campaign_time + RS_GROUTER_PUBLISH_CAMPAIGN_PERIOD )
{
last_publish_campaign_time = now ;
2013-12-27 15:06:47 -05:00
2014-03-29 10:18:05 -04:00
//publishKeys() ; // we don't publish keys anymore.
//
2013-11-05 16:15:26 -05:00
_routing_matrix . updateRoutingProbabilities ( ) ;
2013-11-02 10:35:33 -04:00
}
2013-11-05 16:15:26 -05:00
// Debug dump everything
//
if ( now > last_debug_output_time + RS_GROUTER_DEBUG_OUTPUT_PERIOD )
{
last_debug_output_time = now ;
debugDump ( ) ;
}
2013-12-28 16:47:15 -05:00
// If content has changed, save config, at most every RS_GROUTER_MIN_CONFIG_SAVE_PERIOD seconds appart
// Otherwise, always save at least every RS_GROUTER_MAX_CONFIG_SAVE_PERIOD seconds
//
if ( _changed & & now > last_config_changed + RS_GROUTER_MIN_CONFIG_SAVE_PERIOD )
{
std : : cerr < < " p3GRouter::tick(): triggering config save. " < < std : : endl ;
_changed = false ;
last_config_changed = now ;
IndicateConfigChanged ( ) ;
}
2013-11-02 10:35:33 -04:00
return 0 ;
}
RsSerialiser * p3GRouter : : setupSerialiser ( )
{
RsSerialiser * rss = new RsSerialiser ;
rss - > addSerialType ( new RsGRouterSerialiser ) ;
rss - > addSerialType ( new RsGeneralConfigSerialiser ( ) ) ;
return rss ;
}
void p3GRouter : : autoWash ( )
{
RsStackMutex mtx ( grMtx ) ;
2013-12-27 15:06:47 -05:00
std : : cerr < < " p3GRouter::autoWash(): cleaning old entried. " < < std : : endl ;
2013-11-02 10:35:33 -04:00
// cleanup cache
2013-12-27 15:06:47 -05:00
time_t now = time ( NULL ) ;
for ( std : : map < GRouterKeyPropagationId , time_t > : : iterator it ( _key_diffusion_time_stamps . begin ( ) ) ; it ! = _key_diffusion_time_stamps . end ( ) ; )
if ( it - > second + RS_GROUTER_KEY_DIFFUSION_MAX_KEEP < now )
{
std : : cerr < < " Removing key diffusion time stamp " < < it - > second < < " for diffusion id " < < std : : hex < < it - > first < < std : : dec < < std : : endl ;
std : : map < GRouterKeyPropagationId , time_t > : : iterator tmp ( it ) ;
+ + tmp ;
_key_diffusion_time_stamps . erase ( it ) ;
it = tmp ;
}
else
+ + it ;
// look into pending items.
std : : cerr < < " Pending key diffusion items: " < < _key_diffusion_items . size ( ) < < std : : endl ;
std : : cerr < < " Pending messages to route : " < < _pending_messages . size ( ) < < std : : endl ;
2013-11-02 10:35:33 -04:00
}
2013-12-22 12:04:13 -05:00
void p3GRouter : : routePendingObjects ( )
2013-11-02 10:35:33 -04:00
{
2013-12-27 15:06:47 -05:00
RsStackMutex mtx ( grMtx ) ;
2013-12-22 12:04:13 -05:00
time_t now = time ( NULL ) ;
std : : cerr < < " p3GRouter::routeObjects() triage phase: " < < std : : endl ;
std : : cerr < < " Cached Items : " < < _pending_messages . size ( ) < < std : : endl ;
2014-03-17 16:56:06 -04:00
std : : list < RsPeerId > lst ;
2014-03-29 10:18:05 -04:00
mLinkMgr - > getOnlineList ( lst ) ;
2014-03-17 16:56:06 -04:00
RsPeerId own_id ( mLinkMgr - > getOwnId ( ) ) ;
2013-12-22 12:04:13 -05:00
2014-03-29 10:18:05 -04:00
std : : vector < RsPeerId > pids ;
for ( std : : list < RsPeerId > : : const_iterator it ( lst . begin ( ) ) ; it ! = lst . end ( ) ; + + it )
pids . push_back ( * it ) ;
2013-12-22 12:04:13 -05:00
for ( std : : map < GRouterMsgPropagationId , GRouterRoutingInfo > : : iterator it ( _pending_messages . begin ( ) ) ; it ! = _pending_messages . end ( ) ; )
2014-03-29 10:18:05 -04:00
if ( ( it - > second . status_flags = = RS_GROUTER_ROUTING_STATE_PEND ) | | ( it - > second . status_flags = = RS_GROUTER_ROUTING_STATE_SENT & & it - > second . tried_friends . front ( ) . time_stamp + RS_GROUTER_ROUTING_WAITING_TIME < now ) )
2013-12-22 12:04:13 -05:00
{
std : : cerr < < " Msg id: " < < std : : hex < < it - > first < < std : : dec < < std : : endl ;
std : : cerr < < " Origin: " < < it - > second . origin . toStdString ( ) < < std : : endl ;
std : : cerr < < " Last : " < < it - > second . tried_friends . front ( ) . friend_id . toStdString ( ) < < std : : endl ;
std : : cerr < < " Time : " < < it - > second . tried_friends . front ( ) . time_stamp < < std : : endl ;
std : : cerr < < " Flags : " < < it - > second . status_flags < < std : : endl ;
2014-03-29 10:18:05 -04:00
std : : cerr < < " Dist : " < < it - > second . data_item - > randomized_distance < < std : : endl ;
2013-12-22 12:04:13 -05:00
std : : cerr < < " Probabilities: " < < std : : endl ;
2014-03-29 10:18:05 -04:00
std : : vector < float > probas ; // friends probabilities for online friend list.
RsPeerId routed_friend ; // friend chosen for the next hop
bool should_remove = false ; // should we remove this from the map?
2013-12-22 12:04:13 -05:00
2014-03-29 10:18:05 -04:00
// Retrieve probabilities for this key. This call always succeeds. If no route is known, all probabilities become equal.
2013-12-22 12:04:13 -05:00
//
2014-03-29 10:18:05 -04:00
_routing_matrix . computeRoutingProbabilities ( it - > second . data_item - > destination_key , pids , probas ) ;
2013-12-22 12:04:13 -05:00
2014-03-29 10:18:05 -04:00
// Compute the branching factor.
2013-12-22 12:04:13 -05:00
2014-03-29 10:18:05 -04:00
int N = computeBranchingFactor ( pids , probas , it - > second . data_item - > randomized_distance ) ;
2013-12-22 12:04:13 -05:00
2014-03-29 10:18:05 -04:00
// Now use this to select N random peers according to the given probabilities
2013-12-22 12:04:13 -05:00
2014-03-29 10:18:05 -04:00
std : : set < uint32_t > routing_friend_indices = computeRoutingFriends ( pids , probas , N ) ;
2013-12-22 12:04:13 -05:00
2014-03-29 10:18:05 -04:00
std : : cerr < < " Routing statistics: " < < std : : endl ;
2013-12-22 12:04:13 -05:00
2014-03-29 10:18:05 -04:00
// Actually send the item.
2013-11-02 10:35:33 -04:00
2014-03-29 10:18:05 -04:00
for ( std : : set < uint32_t > : : const_iterator its ( routing_friend_indices . begin ( ) ) ; its ! = routing_friend_indices . end ( ) ; + + its )
2013-12-22 12:04:13 -05:00
{
2014-03-29 10:18:05 -04:00
std : : cerr < < " Friend : " < < ( * its ) < < std : : endl ;
2013-12-22 12:04:13 -05:00
// make a deep copy of the item
RsGRouterGenericDataItem * new_item = it - > second . data_item - > duplicate ( ) ;
// update cache entry
FriendTrialRecord ftr ;
ftr . time_stamp = now ;
2014-03-29 10:18:05 -04:00
ftr . friend_id = pids [ * its ] ;
ftr . probability = probas [ * its ] ;
ftr . nb_friends = probas . size ( ) ;
2013-12-22 12:04:13 -05:00
it - > second . tried_friends . push_front ( ftr ) ;
2013-12-24 11:41:07 -05:00
it - > second . status_flags = RS_GROUTER_ROUTING_STATE_SENT ;
2013-12-22 12:04:13 -05:00
2014-03-29 10:18:05 -04:00
std : : cerr < < " Routing probability: " < < ftr . probability < < std : : endl ;
std : : cerr < < " Sending... " < < std : : endl ;
2013-12-22 12:04:13 -05:00
// send
2014-03-29 10:18:05 -04:00
new_item - > PeerId ( pids [ * its ] ) ;
new_item - > randomized_distance + = computeRandomDistanceIncrement ( pids [ * its ] , new_item - > destination_key ) ;
2013-12-22 12:04:13 -05:00
sendItem ( new_item ) ;
}
if ( should_remove )
{
// We remove from the map. That means the RsItem* has been transfered to somewhere else.
//
std : : cerr < < " Removing item from pending items " < < std : : endl ;
std : : map < GRouterMsgPropagationId , GRouterRoutingInfo > : : iterator tmp ( it ) ;
delete it - > second . data_item ;
+ + tmp ;
_pending_messages . erase ( it ) ;
it = tmp ;
}
else
+ + it ;
}
else
2014-03-29 10:18:05 -04:00
{
2013-12-22 12:04:13 -05:00
std : : cerr < < " Skipping " < < std : : hex < < it - > first < < std : : dec < < " , dest= " < < it - > second . data_item - > destination_key . toStdString ( ) < < " , state = " < < it - > second . status_flags < < " , stamp= " < < it - > second . tried_friends . front ( ) . time_stamp < < " - " < < it - > second . tried_friends . front ( ) . friend_id . toStdString ( ) < < std : : endl ;
2014-03-29 10:18:05 -04:00
+ + it ;
}
}
uint32_t p3GRouter : : computeRandomDistanceIncrement ( const RsPeerId & pid , const GRouterKeyId & destination_key )
{
// This computes a consistent random bias between 0 and 255, which only depends on the
// destination key and the friend the item is going to be routed through.
// Makes it much harder for attakcers to figure out what is going on with
// distances in the network, and makes statistics about multiple sending
// attempts impossible.
//
static uint64_t random_salt = RSRandom : : random_u64 ( ) ;
static const int total_size = RsPeerId : : SIZE_IN_BYTES + GRouterKeyId : : SIZE_IN_BYTES + sizeof ( random_salt ) ;
unsigned char tmpmem [ total_size ] ;
* ( uint64_t * ) & tmpmem [ 0 ] = random_salt ;
memcpy ( & tmpmem [ sizeof ( random_salt ) ] , pid . toByteArray ( ) , RsPeerId : : SIZE_IN_BYTES ) ;
memcpy ( & tmpmem [ sizeof ( random_salt ) + RsPeerId : : SIZE_IN_BYTES ] , destination_key . toByteArray ( ) , GRouterKeyId : : SIZE_IN_BYTES ) ;
return RsDirUtil : : sha1sum ( tmpmem , total_size ) . toByteArray ( ) [ 5 ] ;
}
uint32_t p3GRouter : : computeBranchingFactor ( const std : : vector < RsPeerId > & friends , const std : : vector < float > & probas , uint32_t dist )
{
// This should be made a bit more adaptive ;-)
//
return 1 ;
}
std : : set < uint32_t > p3GRouter : : computeRoutingFriends ( const std : : vector < RsPeerId > & pids , const std : : vector < float > & probas , uint32_t N )
{
// We draw N friends according to the routing probabilitites that are passed as parameter.
//
std : : set < uint32_t > res ;
if ( probas . empty ( ) )
return res ;
res . insert ( 0 ) ;
return res ;
2013-11-02 10:35:33 -04:00
}
2013-12-22 12:04:13 -05:00
2013-11-02 10:35:33 -04:00
void p3GRouter : : publishKeys ( )
{
2014-03-29 10:18:05 -04:00
# ifdef SUSPENDED
2013-12-27 15:06:47 -05:00
RsStackMutex mtx ( grMtx ) ;
2013-11-02 10:35:33 -04:00
// Go through list of published keys
// broadcast a publishKeyItem for each of them.
time_t now = time ( NULL ) ;
for ( std : : map < GRouterKeyId , GRouterPublishedKeyInfo > : : iterator it ( _owned_key_ids . begin ( ) ) ; it ! = _owned_key_ids . end ( ) ; + + it )
{
GRouterPublishedKeyInfo & info ( it - > second ) ;
if ( now > info . last_published_time + RS_GROUTER_PUBLISH_KEY_TIME_INTERVAL )
{
// publish this key
2013-11-05 16:15:26 -05:00
std : : cerr < < " Publishing this key: " < < std : : endl ;
std : : cerr < < " Key id : " < < it - > first . toStdString ( ) < < std : : endl ;
std : : cerr < < " Service id : " < < std : : hex < < info . service_id < < std : : dec < < std : : endl ;
std : : cerr < < " Description : " < < info . description_string < < std : : endl ;
2013-11-02 10:35:33 -04:00
RsGRouterPublishKeyItem item ;
item . diffusion_id = RSRandom : : random_u32 ( ) ;
item . published_key = it - > first ;
item . service_id = info . service_id ;
2013-11-05 16:15:26 -05:00
item . randomized_distance = drand48 ( ) ;
2014-03-29 10:18:05 -04:00
item . fingerprint . clear ( ) ; // not set
2013-11-02 10:35:33 -04:00
item . description_string = info . description_string ;
2014-03-17 16:56:06 -04:00
item . PeerId ( RsPeerId ( ) ) ; // no peer id => key is forwarded to all friends.
2013-11-02 10:35:33 -04:00
2013-12-27 15:06:47 -05:00
locked_forwardKey ( item ) ;
2013-11-02 10:35:33 -04:00
2013-11-05 16:15:26 -05:00
info . last_published_time = now ;
2013-11-02 10:35:33 -04:00
}
}
2014-03-29 10:18:05 -04:00
# endif
2013-11-02 10:35:33 -04:00
}
2013-12-27 15:06:47 -05:00
void p3GRouter : : locked_forwardKey ( const RsGRouterPublishKeyItem & item )
{
2014-03-17 16:56:06 -04:00
std : : list < RsPeerId > connected_peers ;
2013-12-27 15:06:47 -05:00
mLinkMgr - > getOnlineList ( connected_peers ) ;
std : : cerr < < " Forwarding key item to all available friends... " < < std : : endl ;
// get list of connected friends, and broadcast to all of them
//
2014-03-17 16:56:06 -04:00
for ( std : : list < RsPeerId > : : const_iterator it ( connected_peers . begin ( ) ) ; it ! = connected_peers . end ( ) ; + + it )
2013-12-27 15:06:47 -05:00
if ( item . PeerId ( ) ! = * it )
{
std : : cerr < < " sending to " < < ( * it ) < < std : : endl ;
RsGRouterPublishKeyItem * itm = new RsGRouterPublishKeyItem ( item ) ;
itm - > PeerId ( * it ) ;
// we should randomise the depth
sendItem ( itm ) ;
}
else
std : : cerr < < " Not forwarding to source id " < < item . PeerId ( ) < < std : : endl ;
}
2014-03-29 10:18:05 -04:00
bool p3GRouter : : registerKey ( const GRouterKeyId & key , const GRouterServiceId & client_id , const std : : string & description )
2013-11-02 10:35:33 -04:00
{
RsStackMutex mtx ( grMtx ) ;
2014-03-29 10:18:05 -04:00
if ( _registered_services . find ( client_id ) = = _registered_services . end ( ) )
{
std : : cerr < < __PRETTY_FUNCTION__ < < " : unable to register key " < < key < < " for client id " < < client_id < < " : client id is not known. " < < std : : endl ;
return false ;
}
2013-11-02 10:35:33 -04:00
GRouterPublishedKeyInfo info ;
info . service_id = client_id ;
2014-01-30 17:14:37 -05:00
info . description_string = description . substr ( 0 , 20 ) ;
2014-03-29 10:18:05 -04:00
//info.last_published_time = 0 ; // means never published, se it will be re-published soon.
2013-11-02 10:35:33 -04:00
_owned_key_ids [ key ] = info ;
2013-11-05 16:15:26 -05:00
std : : cerr < < " Registered the following key: " < < std : : endl ;
std : : cerr < < " Key id : " < < key . toStdString ( ) < < std : : endl ;
std : : cerr < < " Client id : " < < std : : hex < < client_id < < std : : dec < < std : : endl ;
2014-01-30 17:14:37 -05:00
std : : cerr < < " Description : " < < info . description_string < < std : : endl ;
2013-11-05 16:15:26 -05:00
2013-11-02 10:35:33 -04:00
return true ;
}
2014-01-03 17:41:20 -05:00
bool p3GRouter : : unregisterKey ( const GRouterKeyId & key )
{
RsStackMutex mtx ( grMtx ) ;
std : : map < GRouterKeyId , GRouterPublishedKeyInfo > : : iterator it = _owned_key_ids . find ( key ) ;
if ( it = = _owned_key_ids . end ( ) )
{
std : : cerr < < " p3GRouter::unregisterKey(): key " < < key . toStdString ( ) < < " not found. " < < std : : endl ;
return false ;
}
std : : cerr < < " p3GRouter::unregistered the following key: " < < std : : endl ;
std : : cerr < < " Key id : " < < key . toStdString ( ) < < std : : endl ;
std : : cerr < < " Client id : " < < std : : hex < < it - > second . service_id < < std : : dec < < std : : endl ;
std : : cerr < < " Description : " < < it - > second . description_string < < std : : endl ;
_owned_key_ids . erase ( it ) ;
return true ;
}
2013-11-02 10:35:33 -04:00
void p3GRouter : : handleIncoming ( )
{
RsItem * item ;
while ( NULL ! = ( item = recvItem ( ) ) )
{
switch ( item - > PacketSubType ( ) )
{
2013-12-22 12:04:13 -05:00
case RS_PKT_SUBTYPE_GROUTER_PUBLISH_KEY : handleRecvPublishKeyItem ( dynamic_cast < RsGRouterPublishKeyItem * > ( item ) ) ;
break ;
2013-11-02 10:35:33 -04:00
2013-12-22 12:04:13 -05:00
case RS_PKT_SUBTYPE_GROUTER_DATA : handleRecvDataItem ( dynamic_cast < RsGRouterGenericDataItem * > ( item ) ) ;
break ;
2013-11-02 10:35:33 -04:00
2013-12-22 12:04:13 -05:00
case RS_PKT_SUBTYPE_GROUTER_ACK : handleRecvACKItem ( dynamic_cast < RsGRouterACKItem * > ( item ) ) ;
break ;
2013-11-02 10:35:33 -04:00
default :
std : : cerr < < " (EE) " < < __PRETTY_FUNCTION__ < < " : Unhandled item type " < < item - > PacketSubType ( ) < < std : : endl ;
}
delete item ;
}
}
void p3GRouter : : handleRecvPublishKeyItem ( RsGRouterPublishKeyItem * item )
{
2013-12-27 15:06:47 -05:00
RsStackMutex mtx ( grMtx ) ;
2013-11-02 10:35:33 -04:00
std : : cerr < < " Received key publish item for key : " < < std : : endl ;
std : : cerr < < " diffusion = " < < std : : hex < < item - > diffusion_id < < std : : dec < < std : : endl ;
std : : cerr < < " key id = " < < item - > published_key . toStdString ( ) < < std : : endl ;
std : : cerr < < " service id = " < < std : : hex < < item - > service_id < < std : : dec < < std : : endl ;
std : : cerr < < " distance = " < < item - > randomized_distance < < std : : endl ;
std : : cerr < < " description= " < < item - > description_string < < std : : endl ;
// update the route matrix
2014-03-29 10:18:05 -04:00
_routing_matrix . addRoutingClue ( item - > published_key , RsPeerId ( item - > PeerId ( ) ) , 1 ) ;
2013-11-02 10:35:33 -04:00
// forward the key to other peers according to key forwarding cache
std : : map < GRouterKeyPropagationId , time_t > : : iterator it = _key_diffusion_time_stamps . find ( item - > diffusion_id ) ;
2013-12-27 15:06:47 -05:00
bool found = ( it ! = _key_diffusion_time_stamps . end ( ) ) ; // found. We don't propagate further
2013-11-02 10:35:33 -04:00
_key_diffusion_time_stamps [ item - > diffusion_id ] = time ( NULL ) ; // always stamp
if ( found )
2013-12-27 15:06:47 -05:00
{
std : : cerr < < " Key diffusion item already in cache. Not forwardign further. " < < std : : endl ;
2013-11-02 10:35:33 -04:00
return ;
2013-12-27 15:06:47 -05:00
}
2013-11-02 10:35:33 -04:00
// Propagate the item to all other online friends. We don't do this right now, but push items in a queue.
// Doing this we can control the amount of key propagation and avoid flooding.
2013-12-27 15:06:47 -05:00
locked_forwardKey ( * item ) ;
2013-12-28 16:47:15 -05:00
_changed = true ;
2013-11-02 10:35:33 -04:00
}
void p3GRouter : : handleRecvACKItem ( RsGRouterACKItem * item )
{
2013-12-27 15:06:47 -05:00
RsStackMutex mtx ( grMtx ) ;
2014-03-29 10:18:05 -04:00
std : : cerr < < " Received ACK item, mid= " < < std : : hex < < item - > mid < < std : : dec < < " , ACK type = " < < item - > state < < std : : endl ;
2013-12-22 12:04:13 -05:00
// find the item in the pendign list,
// - if not found, drop.
//
// ...and act appropriately:
// - item was
// - if we're origin
// notify the client service
// else
// remove item
//
std : : map < GRouterMsgPropagationId , GRouterRoutingInfo > : : iterator it ( _pending_messages . find ( item - > mid ) ) ;
2013-11-02 10:35:33 -04:00
2013-12-22 12:04:13 -05:00
if ( it = = _pending_messages . end ( ) )
{
std : : cerr < < " p3GRouter::handleRecvACKItem(): cannot find entry for message id " < < std : : hex < < item - > mid < < std : : dec < < " . Dropping it. " < < std : : endl ;
return ;
}
2013-11-02 10:35:33 -04:00
switch ( item - > state )
{
case RS_GROUTER_ACK_STATE_RECEIVED_INDIRECTLY :
2013-12-22 12:04:13 -05:00
case RS_GROUTER_ACK_STATE_RECEIVED :
// Notify the origin. This is the main route and it was successful.
2014-03-29 10:18:05 -04:00
std : : cerr < < " updating routing matrix. " < < std : : endl ;
it - > second . status_flags = RS_GROUTER_ROUTING_STATE_ARVD ;
{
# warning UNFINISHED code.
// Now compute the weight for that particular item. See with what probabilities it was chosen.
//
float weight = ( item - > state = = RS_GROUTER_ACK_STATE_RECEIVED ) ? 1.0f : 0.5 ;
std : : cerr < < " weight = " < < weight < < std : : endl ;
2013-12-22 12:04:13 -05:00
2014-03-29 10:18:05 -04:00
_routing_matrix . addRoutingClue ( it - > second . data_item - > destination_key , item - > PeerId ( ) , weight ) ;
}
if ( it - > second . origin ! = mLinkMgr - > getOwnId ( ) )
{
std : : cerr < < " forwarding ACK to origin: " < < it - > second . origin . toStdString ( ) < < std : : endl ;
sendACK ( it - > second . origin , item - > mid , item - > state ) ;
}
2013-11-02 10:35:33 -04:00
break ;
case RS_GROUTER_ACK_STATE_GIVEN_UP : // route is bad. We forward back and update the routing matrix.
break ;
}
2013-12-22 12:04:13 -05:00
2014-03-29 10:18:05 -04:00
if ( it - > second . origin = = mLinkMgr - > getOwnId ( ) )
2013-12-22 12:04:13 -05:00
{
// find the client service and notify it.
std : : cerr < < " We're owner: should notify client id " < < std : : endl ;
}
2014-03-29 10:18:05 -04:00
// Just decrement the list of tried friends
2013-12-22 12:04:13 -05:00
//
2014-03-29 10:18:05 -04:00
for ( std : : list < FriendTrialRecord > : : iterator it2 ( it - > second . tried_friends . begin ( ) ) ; it2 ! = it - > second . tried_friends . end ( ) ; + + it2 )
if ( ( * it2 ) . friend_id = = item - > PeerId ( ) )
{
std : : cerr < < " Removing friend try for peer " < < item - > PeerId ( ) < < " . " < < it - > second . tried_friends . size ( ) < < " tries left. " < < std : : endl ;
it - > second . tried_friends . erase ( it2 ) ;
break ;
}
if ( it - > second . tried_friends . empty ( ) )
{
delete it - > second . data_item ;
_pending_messages . erase ( it ) ;
std : : cerr < < " No tries left. Removing item from pending list. " < < std : : endl ;
}
2013-11-02 10:35:33 -04:00
}
void p3GRouter : : handleRecvDataItem ( RsGRouterGenericDataItem * item )
{
2013-12-27 15:06:47 -05:00
RsStackMutex mtx ( grMtx ) ;
2014-03-29 10:18:05 -04:00
std : : cerr < < " Received data item for key " < < item - > destination_key < < " , distance = " < < item - > randomized_distance < < std : : endl ;
// check the item depth. If too large, send a ACK back.
if ( item - > randomized_distance > GROUTER_ITEM_MAX_TRAVEL_DISTANCE )
{
std : : cerr < < " Distance is too large: " < < item - > randomized_distance < < " units. Item is dropped. " < < std : : endl ;
sendACK ( item - > PeerId ( ) , item - > routing_id , RS_GROUTER_ACK_STATE_TOO_FAR ) ;
return ;
}
2013-11-02 10:35:33 -04:00
2013-12-24 11:41:07 -05:00
// Do we have this item in the cache already?
// - if not, add in the pending items
// - if yet. Ignore, or send ACK for shorter route.
std : : map < GRouterKeyId , GRouterPublishedKeyInfo > : : const_iterator it = _owned_key_ids . find ( item - > destination_key ) ;
2014-03-29 10:18:05 -04:00
std : : map < GRouterMsgPropagationId , GRouterRoutingInfo > : : iterator itr = _pending_messages . find ( item - > routing_id ) ;
2013-12-24 11:41:07 -05:00
RsGRouterGenericDataItem * item_copy = NULL ;
if ( itr ! = _pending_messages . end ( ) )
{
std : : cerr < < " Item is already there. Nothing to do. Should we update the cache? " < < std : : endl ;
item_copy = itr - > second . data_item ;
}
2014-03-29 10:18:05 -04:00
else // item is not known. Store it into pending msgs. We make a copy, since the item will be deleted otherwise.
2013-12-24 11:41:07 -05:00
{
std : : cerr < < " Item is new. Storing in cache as pending messages. " < < std : : endl ;
GRouterRoutingInfo info ;
info . data_item = item - > duplicate ( ) ;
item_copy = info . data_item ;
2014-03-17 16:56:06 -04:00
info . origin = RsPeerId ( item - > PeerId ( ) ) ;
2013-12-24 11:41:07 -05:00
info . received_time = time ( NULL ) ;
_pending_messages [ item - > routing_id ] = info ;
2014-03-29 10:18:05 -04:00
itr = _pending_messages . find ( item - > routing_id ) ;
2013-12-24 11:41:07 -05:00
}
// Is the item for us? If so, find the client service and send the item back.
2014-03-29 10:18:05 -04:00
//
2013-12-24 11:41:07 -05:00
if ( it ! = _owned_key_ids . end ( ) )
2014-03-29 10:18:05 -04:00
{
if ( itr - > second . status_flags = = RS_GROUTER_ROUTING_STATE_ARVD )
sendACK ( item - > PeerId ( ) , item - > routing_id , RS_GROUTER_ACK_STATE_RECEIVED_INDIRECTLY ) ;
else
2013-12-24 11:41:07 -05:00
{
2014-03-29 10:18:05 -04:00
sendACK ( item - > PeerId ( ) , item - > routing_id , RS_GROUTER_ACK_STATE_RECEIVED ) ;
itr - > second . status_flags = RS_GROUTER_ROUTING_STATE_ARVD ;
2013-12-24 11:41:07 -05:00
std : : map < GRouterServiceId , GRouterClientService * > : : const_iterator its = _registered_services . find ( it - > second . service_id ) ;
if ( its ! = _registered_services . end ( ) )
{
std : : cerr < < " Key is owned by us. Notifying service for this item. " < < std : : endl ;
2014-03-29 10:18:05 -04:00
its - > second - > receiveGRouterData ( it - > first , item_copy ) ;
2013-12-24 11:41:07 -05:00
}
else
std : : cerr < < " (EE) weird situation. No service registered for a key that we own. Key id = " < < item - > destination_key . toStdString ( ) < < " , service id = " < < it - > second . service_id < < std : : endl ;
}
2014-03-29 10:18:05 -04:00
}
2013-12-24 11:41:07 -05:00
else
2014-03-29 10:18:05 -04:00
{
std : : cerr < < " item is not for us. Storing in pending mode. " < < std : : endl ;
itr - > second . status_flags = RS_GROUTER_ROUTING_STATE_PEND ;
}
2013-12-28 16:47:15 -05:00
_changed = true ;
2013-11-02 10:35:33 -04:00
}
2013-12-27 15:06:47 -05:00
bool p3GRouter : : registerClientService ( const GRouterServiceId & id , GRouterClientService * service )
{
RsStackMutex mtx ( grMtx ) ;
_registered_services [ id ] = service ;
return true ;
}
2013-12-24 11:41:07 -05:00
void p3GRouter : : sendData ( const GRouterKeyId & destination , RsGRouterGenericDataItem * item )
2013-11-02 10:35:33 -04:00
{
2013-12-27 15:06:47 -05:00
RsStackMutex mtx ( grMtx ) ;
2013-12-24 11:41:07 -05:00
// push the item into pending messages.
//
GRouterRoutingInfo info ;
info . data_item = item ;
info . status_flags = RS_GROUTER_ROUTING_STATE_PEND ;
2014-03-17 16:56:06 -04:00
info . origin = RsPeerId ( mLinkMgr - > getOwnId ( ) ) ;
2014-03-29 10:18:05 -04:00
info . data_item - > randomized_distance = 0 ;
2013-12-24 11:41:07 -05:00
// Make sure we have a unique id (at least locally).
//
GRouterMsgPropagationId propagation_id ;
2013-12-27 15:06:47 -05:00
do { propagation_id = RSRandom : : random_u32 ( ) ; } while ( _pending_messages . find ( propagation_id ) ! = _pending_messages . end ( ) ) ;
2013-12-24 11:41:07 -05:00
2014-03-29 10:18:05 -04:00
item - > destination_key = destination ;
item - > routing_id = propagation_id ;
2013-12-24 11:41:07 -05:00
std : : cerr < < " p3GRouter::sendGRouterData(): pushing the followign item in the msg pending list: " < < std : : endl ;
std : : cerr < < " data_item.size = " < < info . data_item - > data_size < < std : : endl ;
std : : cerr < < " data_item.byte = " < < info . data_item - > data_bytes < < std : : endl ;
2014-03-29 10:18:05 -04:00
std : : cerr < < " destination = " < < info . data_item - > destination_key < < std : : endl ;
2013-12-24 11:41:07 -05:00
std : : cerr < < " status = " < < info . status_flags < < std : : endl ;
2014-03-29 10:18:05 -04:00
std : : cerr < < " distance = " < < info . data_item - > randomized_distance < < std : : endl ;
2013-12-24 11:41:07 -05:00
std : : cerr < < " origin = " < < info . origin . toStdString ( ) < < std : : endl ;
std : : cerr < < " Recv time = " < < info . received_time < < std : : endl ;
_pending_messages [ propagation_id ] = info ;
2013-11-02 10:35:33 -04:00
}
2014-03-17 16:56:06 -04:00
void p3GRouter : : sendACK ( const RsPeerId & peer , GRouterMsgPropagationId mid , uint32_t ack_flags )
2013-12-22 12:04:13 -05:00
{
RsGRouterACKItem * item = new RsGRouterACKItem ;
item - > state = ack_flags ;
item - > mid = mid ;
2014-03-17 16:56:06 -04:00
item - > PeerId ( peer ) ;
2013-12-22 12:04:13 -05:00
sendItem ( item ) ;
}
2013-11-02 10:35:33 -04:00
bool p3GRouter : : loadList ( std : : list < RsItem * > & items )
{
2013-12-27 15:06:47 -05:00
RsStackMutex mtx ( grMtx ) ;
std : : cerr < < " p3GRouter::loadList() : " < < std : : endl ;
_routing_matrix . loadList ( items ) ;
// remove all existing objects.
//
std : : cerr < < " removing all existing items ( " < < _pending_messages . size ( ) < < " items to delete). " < < std : : endl ;
for ( std : : map < GRouterMsgPropagationId , GRouterRoutingInfo > : : iterator it ( _pending_messages . begin ( ) ) ; it ! = _pending_messages . end ( ) ; + + it )
delete it - > second . data_item ;
_pending_messages . clear ( ) ;
for ( std : : list < RsItem * > : : const_iterator it ( items . begin ( ) ) ; it ! = items . end ( ) ; + + it )
{
RsGRouterRoutingInfoItem * itm1 = NULL ;
if ( NULL ! = ( itm1 = dynamic_cast < RsGRouterRoutingInfoItem * > ( * it ) ) )
{
_pending_messages [ itm1 - > data_item - > routing_id ] = * itm1 ;
_pending_messages [ itm1 - > data_item - > routing_id ] . data_item = itm1 - > data_item ; // avoids duplication.
itm1 - > data_item = NULL ; // prevents deletion.
}
delete * it ;
}
return true ;
2013-11-02 10:35:33 -04:00
}
2013-12-27 15:06:47 -05:00
bool p3GRouter : : saveList ( bool & cleanup , std : : list < RsItem * > & items )
2013-11-02 10:35:33 -04:00
{
2013-12-22 12:04:13 -05:00
// We save
// - the routing clues
// - the pending items
2013-12-27 15:06:47 -05:00
cleanup = true ; // the client should delete the items.
std : : cerr < < " p3GRouter::saveList()... " < < std : : endl ;
std : : cerr < < " saving routing clues. " < < std : : endl ;
_routing_matrix . saveList ( items ) ;
std : : cerr < < " saving pending items. " < < std : : endl ;
for ( std : : map < GRouterMsgPropagationId , GRouterRoutingInfo > : : const_iterator it ( _pending_messages . begin ( ) ) ; it ! = _pending_messages . end ( ) ; + + it )
{
RsGRouterRoutingInfoItem * item = new RsGRouterRoutingInfoItem ;
* ( GRouterRoutingInfo * ) item = it - > second ; // copy all members
item - > data_item = it - > second . data_item - > duplicate ( ) ; // deep copy, because we call delete on the object, and the item might be removed before we handle it in the client.
items . push_back ( item ) ;
}
return true ;
2013-11-02 10:35:33 -04:00
}
2014-03-29 10:18:05 -04:00
bool p3GRouter : : getRoutingMatrixInfo ( RsGRouter : : GRouterRoutingMatrixInfo & info )
{
info . per_friend_probabilities . clear ( ) ;
info . friend_ids . clear ( ) ;
info . published_keys . clear ( ) ;
std : : list < RsPeerId > ids ;
mLinkMgr - > getOnlineList ( ids ) ;
RsStackMutex mtx ( grMtx ) ;
info . published_keys = _owned_key_ids ;
for ( std : : list < RsPeerId > : : const_iterator it ( ids . begin ( ) ) ; it ! = ids . end ( ) ; + + it )
info . friend_ids . push_back ( * it ) ;
std : : vector < GRouterKeyId > known_keys ;
std : : vector < float > probas ;
_routing_matrix . getListOfKnownKeys ( known_keys ) ;
for ( uint32_t i = 0 ; i < known_keys . size ( ) ; + + i )
{
_routing_matrix . computeRoutingProbabilities ( known_keys [ i ] , info . friend_ids , probas ) ;
info . per_friend_probabilities [ known_keys [ i ] ] = probas ;
}
return true ;
}
bool p3GRouter : : getRoutingCacheInfo ( std : : vector < GRouterRoutingCacheInfo > & infos )
{
RsStackMutex mtx ( grMtx ) ;
infos . clear ( ) ;
for ( std : : map < GRouterMsgPropagationId , GRouterRoutingInfo > : : const_iterator it ( _pending_messages . begin ( ) ) ; it ! = _pending_messages . end ( ) ; + + it )
{
infos . push_back ( GRouterRoutingCacheInfo ( ) ) ;
GRouterRoutingCacheInfo & cinfo ( infos . back ( ) ) ;
cinfo . mid = it - > first ;
cinfo . local_origin = it - > second . origin ;
cinfo . destination = it - > second . data_item - > destination_key ;
cinfo . time_stamp = it - > second . received_time ;
cinfo . status = it - > second . status_flags ;
cinfo . data_size = it - > second . data_item - > data_size ;
}
return true ;
}
2013-11-05 16:15:26 -05:00
// Dump everything
//
void p3GRouter : : debugDump ( )
{
2013-12-27 15:06:47 -05:00
RsStackMutex mtx ( grMtx ) ;
2013-11-05 16:15:26 -05:00
time_t now = time ( NULL ) ;
std : : cerr < < " Full dump of Global Router state: " < < std : : endl ;
std : : cerr < < " Owned keys : " < < std : : endl ;
for ( std : : map < GRouterKeyId , GRouterPublishedKeyInfo > : : const_iterator it ( _owned_key_ids . begin ( ) ) ; it ! = _owned_key_ids . end ( ) ; + + it )
{
std : : cerr < < " Key id : " < < it - > first . toStdString ( ) < < std : : endl ;
std : : cerr < < " Service id : " < < std : : hex < < it - > second . service_id < < std : : dec < < std : : endl ;
std : : cerr < < " Description : " < < it - > second . description_string < < std : : endl ;
2014-03-29 10:18:05 -04:00
//std::cerr << " Last published: " << now - it->second.last_published_time << " secs ago" << std::endl;
2013-11-05 16:15:26 -05:00
}
std : : cerr < < " Registered services: " < < std : : endl ;
for ( std : : map < GRouterServiceId , GRouterClientService * > : : const_iterator it ( _registered_services . begin ( ) ) ; it ! = _registered_services . end ( ) ; + + it )
std : : cerr < < " " < < std : : hex < < it - > first < < " " < < std : : dec < < ( void * ) it - > second < < std : : endl ;
std : : cerr < < " Key diffusion cache: " < < std : : endl ;
for ( std : : map < GRouterKeyPropagationId , time_t > : : const_iterator it ( _key_diffusion_time_stamps . begin ( ) ) ; it ! = _key_diffusion_time_stamps . end ( ) ; + + it )
std : : cerr < < " " < < std : : hex < < it - > first < < " " < < std : : dec < < now - it - > second < < " secs ago " < < std : : endl ;
std : : cerr < < " Key diffusion items: " < < std : : endl ;
std : : cerr < < " [Not shown yet] " < < std : : endl ;
std : : cerr < < " Data items: " < < std : : endl ;
2014-03-29 10:18:05 -04:00
static std : : string statusString [ 4 ] = { " Unkn " , " Pend " , " Sent " , " Ackn " } ;
for ( std : : map < GRouterMsgPropagationId , GRouterRoutingInfo > : : iterator it ( _pending_messages . begin ( ) ) ; it ! = _pending_messages . end ( ) ; + + it )
std : : cerr < < " Msg id: " < < std : : hex < < it - > first < < std : : dec
< < " Local Origin: " < < it - > second . origin . toStdString ( )
< < " Destination: " < < it - > second . data_item - > destination_key
< < " Time : " < < now - it - > second . tried_friends . front ( ) . time_stamp < < " secs ago. "
< < " Status: " < < statusString [ it - > second . status_flags ] < < std : : endl ;
// << " Last : " << it->second.tried_friends.front().friend_id.toStdString() << std::endl;
// << " Probabilities: " << std::endl;
2013-11-05 16:15:26 -05:00
std : : cerr < < " Routing matrix: " < < std : : endl ;
_routing_matrix . debugDump ( ) ;
}
2013-11-02 10:35:33 -04:00