BOOL CNetwork::SendPush(GGUID* pGUID, DWORD nIndex)
{
CSingleLock pLock( &Network.m_pSection );
if ( ! pLock.Lock( 250 ) ) return TRUE;
if ( ! IsListening() ) return FALSE;
GGUID pGUID2 = *pGUID;
SOCKADDR_IN pEndpoint;
CNeighbour* pOrigin;
int nCount = 0;
while ( GetNodeRoute( &pGUID2, &pOrigin, &pEndpoint ) )
{
if ( pOrigin != NULL && pOrigin->m_nProtocol == PROTOCOL_G1 )
{
CG1Packet* pPacket = CG1Packet::New( G1_PACKET_PUSH,
Settings.Gnutella1.MaximumTTL - 1 );
pPacket->Write( pGUID, 16 );
pPacket->WriteLongLE( nIndex );
pPacket->WriteLongLE( m_pHost.sin_addr.S_un.S_addr );
pPacket->WriteShortLE( htons( m_pHost.sin_port ) );
pOrigin->Send( pPacket );
}
else
{
CG2Packet* pPacket = CG2Packet::New( G2_PACKET_PUSH, TRUE );
pPacket->WritePacket( G2_PACKET_TO, 16 );
pPacket->Write( pGUID, 16 );
pPacket->WriteByte( 0 );
pPacket->WriteLongLE( m_pHost.sin_addr.S_un.S_addr );
pPacket->WriteShortBE( htons( m_pHost.sin_port ) );
if ( pOrigin != NULL )
{
pOrigin->Send( pPacket );
}
else
{
Datagrams.Send( &pEndpoint, pPacket );
}
}
pGUID2.n[15] ++;
nCount++;
}
return nCount > 0;
}
void CLocalSearch::CreatePacketG2()
{
CG2Packet* pPacket = CG2Packet::New( G2_PACKET_HIT, TRUE );
m_pPacket = pPacket;
pPacket->WritePacket( "GU", 16 );
GGUID tmp( MyProfile.GUID );
pPacket->Write( &tmp, sizeof(GGUID) );
if ( TRUE /* Network.IsListening() */ )
{
pPacket->WritePacket( "NA", 6 );
pPacket->WriteLongLE( Network.m_pHost.sin_addr.S_un.S_addr );
pPacket->WriteShortBE( htons( Network.m_pHost.sin_port ) );
}
pPacket->WritePacket( "V", 4 );
pPacket->WriteString( SHAREAZA_VENDOR_A, FALSE );
if ( ! Network.IsStable() || ! Datagrams.IsStable() )
{
pPacket->WritePacket( "FW", 0 );
}
{
CSingleLock pNetLock( &Network.m_pSection );
if ( pNetLock.Lock( 50 ) )
{
for ( POSITION pos = Neighbours.GetIterator() ; pos ; )
{
CNeighbour* pNeighbour = Neighbours.GetNext( pos );
if ( pNeighbour->m_nNodeType != ntLeaf &&
pNeighbour->m_nProtocol == PROTOCOL_G2 )
{
pPacket->WritePacket( "NH", 6 );
pPacket->WriteLongLE( pNeighbour->m_pHost.sin_addr.S_un.S_addr );
pPacket->WriteShortBE( htons( pNeighbour->m_pHost.sin_port ) );
}
}
}
}
if ( ! Uploads.m_bStable ) pPacket->WritePacket( "UNSTA", 0 );
CSingleLock pQueueLock( &UploadQueues.m_pSection );
int nQueue = 1;
if ( pQueueLock.Lock() )
{
for ( POSITION pos = UploadQueues.GetIterator() ; pos ; nQueue++ )
{
CUploadQueue* pQueue = UploadQueues.GetNext( pos );
pPacket->WritePacket( "HG", ( 4 + 7 ) + 2, TRUE );
pPacket->WritePacket( "SS", 7 );
pPacket->WriteShortBE( pQueue->GetQueuedCount() + pQueue->GetTransferCount() );
pPacket->WriteByte( pQueue->GetTransferCount( TRUE ) );
pPacket->WriteLongBE( pQueue->GetPredictedBandwidth() * 8 / 1024 );
pPacket->WriteByte( 0 );
pPacket->WriteByte( nQueue );
}
pQueueLock.Unlock();
}
CString strNick = MyProfile.GetNick();
if ( strNick.GetLength() > 32 ) strNick = strNick.Left( 32 );
if ( strNick.GetLength() )
{
int nNick = pPacket->GetStringLen( strNick );
pPacket->WritePacket( "UPRO", nNick + 6, TRUE );
pPacket->WritePacket( "NICK", nNick );
pPacket->WriteString( strNick, FALSE );
}
if ( Settings.Community.ServeProfile ) pPacket->WritePacket( "BUP", 0 );
if ( Settings.Community.ServeFiles ) pPacket->WritePacket( "BH", 0 );
if ( Settings.Community.ChatEnable ) pPacket->WritePacket( "PCH", 0 );
}
// Takes a GUID, and a neighbour to except from the packet we will make
// Makes a Gnutella2 query web packet, containing the IP addresses of the computers we are connected to and from the Gnutella2 host cache
CG2Packet* CNeighboursWithG2::CreateQueryWeb(const Hashes::Guid& oGUID, bool bWithHubs, CNeighbour* pExcept, bool bDone)
{
// Make a new Gnutella2 Query Ack packet
CG2Packet* pPacket = CG2Packet::New( G2_PACKET_QUERY_ACK, TRUE );
if ( ! pPacket )
return NULL;
// Start it with the text "TS" and the time now
const DWORD tNow = static_cast< DWORD >( time( NULL ) ); // Number of seconds since 1970
pPacket->WritePacket( G2_PACKET_TIMESTAMP, 4 );
pPacket->WriteLongBE( tNow );
// Write in header information about us
pPacket->WritePacket( G2_PACKET_FROM_ADDRESS, 4 );
pPacket->WriteLongLE( Network.m_pHost.sin_addr.S_un.S_addr );
pPacket->WritePacket( G2_PACKET_RETRY_AFTER, 4 );
pPacket->WriteLongBE( Settings.Gnutella2.QueryThrottle );
if ( bDone )
{
pPacket->WritePacket( G2_PACKET_QUERY_DONE, 8 );
pPacket->WriteLongLE( Network.m_pHost.sin_addr.S_un.S_addr );
pPacket->WriteShortBE( htons( Network.m_pHost.sin_port ) );
if ( bWithHubs )
{
const WORD nLeafs = (WORD)GetCount( PROTOCOL_G2, nrsConnected, ntLeaf );
pPacket->WriteShortBE( nLeafs );
// Loop through the connected computers
for ( POSITION pos = GetIterator() ; pos ; )
{
// Get the neighbour object at this position, and move pos to the next one
CG2Neighbour* pNeighbour = (CG2Neighbour*)GetNext( pos );
// If this neighbour is running Gnutella2 software
if ( pNeighbour->m_nProtocol == PROTOCOL_G2 && // The remote computer is running Gnutella2 software, and
pNeighbour->m_nNodeType != ntLeaf && // Our connection to it is not down to a leaf, and
pNeighbour->m_nState >= nrsConnected && // We've finished the handshake with it, and
pNeighbour != pExcept ) // This isn't the computer the caller warned us to except
{
// Write information about this connected computer into the packet
pPacket->WritePacket( G2_PACKET_QUERY_DONE, 8 );
pPacket->WriteLongLE( pNeighbour->m_pHost.sin_addr.S_un.S_addr );
pPacket->WriteShortBE( htons( pNeighbour->m_pHost.sin_port ) );
pPacket->WriteShortBE( (WORD)pNeighbour->m_nLeafCount );
}
}
// Will put up to 3 or 25 IP addresses in the packet:
// If caller didn't give us a computer to ignore make nCount 3, if it did give us an except make nCount 25
int nCount = ( pExcept == NULL ) ? 3 : 25;
CQuickLock oLock( HostCache.Gnutella2.m_pSection );
// Loop, starting with the newest entry in the Gnutella2 host cache, then stepping to the one before that
for ( CHostCacheIterator i = HostCache.Gnutella2.Begin() ; i != HostCache.Gnutella2.End() ; ++i )
{
CHostCacheHostPtr pHost = (*i);
// If this host cache entry is good
if ( pHost->CanQuote( tNow ) && // If this host cache entry hasn't expired, and
Get( pHost->m_pAddress ) == NULL && // We're connected to that IP address right now, and
HubHorizonPool.Find( &pHost->m_pAddress ) == NULL ) // The IP address is also in the hub horizon pool
{
// Add the IP address to the packet we're making, to encourage recipient to try this address
pPacket->WritePacket( G2_PACKET_QUERY_SEARCH, 10 );
pPacket->WriteLongLE( pHost->m_pAddress.S_un.S_addr );
pPacket->WriteShortBE( pHost->m_nPort );
pPacket->WriteLongBE( pHost->Seen() );
// Lower the count, if it is then 0, leave the loop
if ( ! --nCount ) break;
}
}
// Give the packet we're making to our own hub horizon pool
HubHorizonPool.AddHorizonHubs( pPacket );
}
else // No hubs
{
pPacket->WriteShortBE( 0 );
}
}
// Finish the packet with a 0 byte and the guid the caller gave us, and return it
pPacket->WriteByte( 0 );
pPacket->Write( oGUID );
return pPacket;
}
CG2Packet* CLocalSearch::CreatePacketG2()
{
CG2Packet* pPacket = CG2Packet::New( G2_PACKET_HIT, TRUE );
pPacket->WritePacket( G2_PACKET_NODE_GUID, Hashes::Guid::byteCount );
pPacket->Write( Hashes::Guid( MyProfile.oGUID ) );
pPacket->WritePacket( G2_PACKET_NODE_ADDRESS, 6 );
pPacket->WriteLongLE( Network.m_pHost.sin_addr.s_addr );
pPacket->WriteShortBE( htons( Network.m_pHost.sin_port ) );
pPacket->WritePacket( G2_PACKET_VENDOR, 4 );
pPacket->WriteString( VENDOR_CODE, FALSE );
if ( Network.IsFirewalled() )
{
pPacket->WritePacket( G2_PACKET_PEER_FIREWALLED, 0 );
}
{
CSingleLock pNetLock( &Network.m_pSection );
if ( pNetLock.Lock( 50 ) )
{
for ( POSITION pos = Neighbours.GetIterator() ; pos ; )
{
CNeighbour* pNeighbour = Neighbours.GetNext( pos );
if ( pNeighbour->m_nNodeType != ntLeaf &&
pNeighbour->m_nProtocol == PROTOCOL_G2 )
{
pPacket->WritePacket( G2_PACKET_NEIGHBOUR_HUB, 6 );
pPacket->WriteLongLE( pNeighbour->m_pHost.sin_addr.s_addr );
pPacket->WriteShortBE( htons( pNeighbour->m_pHost.sin_port ) );
}
}
}
}
if ( ! Uploads.m_bStable ) pPacket->WritePacket( G2_PACKET_PEER_UNSTABLE, 0 );
CSingleLock pQueueLock( &UploadQueues.m_pSection );
int nQueue = 1;
if ( pQueueLock.Lock() )
{
for ( POSITION pos = UploadQueues.GetIterator() ; pos ; nQueue++ )
{
CUploadQueue* pQueue = UploadQueues.GetNext( pos );
pPacket->WritePacket( G2_PACKET_HIT_GROUP, ( 4 + 7 ) + 2, TRUE );
pPacket->WritePacket( G2_PACKET_PEER_STATUS, 7 );
pPacket->WriteShortBE( WORD( pQueue->GetQueuedCount() + pQueue->GetTransferCount() ) );
pPacket->WriteByte( BYTE( pQueue->GetTransferCount( TRUE ) ) );
pPacket->WriteLongBE( pQueue->GetPredictedBandwidth() * 8 / 1024 );
pPacket->WriteByte( 0 );
pPacket->WriteByte( BYTE( nQueue ) );
}
pQueueLock.Unlock();
}
CString strNick = MyProfile.GetNick();
if ( strNick.GetLength() > 32 ) strNick = strNick.Left( 32 );
if ( strNick.GetLength() )
{
int nNick = pPacket->GetStringLen( strNick );
pPacket->WritePacket( G2_PACKET_PROFILE, nNick + 6, TRUE );
pPacket->WritePacket( G2_PACKET_NICK, nNick );
pPacket->WriteString( strNick, FALSE );
}
if ( Settings.Community.ServeProfile ) pPacket->WritePacket( G2_PACKET_BROWSE_PROFILE, 0 );
if ( Settings.Community.ServeFiles ) pPacket->WritePacket( G2_PACKET_BROWSE_HOST, 0 );
if ( Settings.Community.ChatEnable ) pPacket->WritePacket( G2_PACKET_PEER_CHAT, 0 );
return pPacket;
}
CG2Packet* CNeighboursWithG2::CreateQueryWeb(GGUID* pGUID, CNeighbour* pExcept)
{
CG2Packet* pPacket = CG2Packet::New( G2_PACKET_QUERY_ACK, TRUE );
DWORD tNow = time( NULL );
pPacket->WritePacket( "TS", 4 );
pPacket->WriteLongBE( tNow );
theApp.Message( MSG_DEBUG, _T("Creating a query acknowledgement:") );
pPacket->WritePacket( "D", 8 );
pPacket->WriteLongLE( Network.m_pHost.sin_addr.S_un.S_addr );
pPacket->WriteShortBE( htons( Network.m_pHost.sin_port ) );
pPacket->WriteShortBE( GetCount( PROTOCOL_G2, nrsConnected, ntLeaf ) );
for ( POSITION pos = GetIterator() ; pos ; )
{
CG2Neighbour* pNeighbour = (CG2Neighbour*)GetNext( pos );
if ( pNeighbour->m_nProtocol == PROTOCOL_G2 &&
pNeighbour->m_nNodeType != ntLeaf &&
pNeighbour->m_nState >= nrsConnected &&
pNeighbour != pExcept )
{
pPacket->WritePacket( "D", 8 );
pPacket->WriteLongLE( pNeighbour->m_pHost.sin_addr.S_un.S_addr );
pPacket->WriteShortBE( htons( pNeighbour->m_pHost.sin_port ) );
pPacket->WriteShortBE( (WORD)pNeighbour->m_nLeafCount );
theApp.Message( MSG_DEBUG, _T(" Done neighbour %s"),
(LPCTSTR)pNeighbour->m_sAddress );
}
}
int nCount = ( pExcept == NULL ) ? 3 : 25;
for ( CHostCacheHost* pHost = HostCache.Gnutella2.GetNewest() ; pHost ; pHost = pHost->m_pPrevTime )
{
if ( pHost->CanQuote( tNow ) &&
Get( &pHost->m_pAddress ) == NULL &&
HubHorizonPool.Find( &pHost->m_pAddress ) == NULL )
{
pPacket->WritePacket( "S", 10 );
pPacket->WriteLongLE( pHost->m_pAddress.S_un.S_addr );
pPacket->WriteShortBE( pHost->m_nPort );
pPacket->WriteLongBE( pHost->m_tSeen );
theApp.Message( MSG_DEBUG, _T(" Try cached hub %s"),
(LPCTSTR)CString( inet_ntoa( pHost->m_pAddress ) ) );
if ( ! --nCount ) break;
}
}
HubHorizonPool.AddHorizonHubs( pPacket );
pPacket->WriteByte( 0 );
pPacket->Write( pGUID, sizeof(GGUID) );
return pPacket;
}