diff --git a/trackerlogic.c b/trackerlogic.c index 134c504..80fc23e 100644 --- a/trackerlogic.c +++ b/trackerlogic.c @@ -18,8 +18,13 @@ #include #include "scan.h" #include "byte.h" +#include "mutex.h" /* GLOBAL VARIABLES */ + +/* We maintain a list of 1024 pointers to sorted list of ot_torrent structs + Sort key is, of course, its hash */ +#define OT_BUCKET_COUNT 1024 static ot_vector all_torrents[OT_BUCKET_COUNT]; static ot_time all_torrents_clean[OT_BUCKET_COUNT]; #if defined ( WANT_BLACKLISTING ) || defined( WANT_CLOSED_TRACKER ) @@ -29,6 +34,28 @@ static ot_vector accesslist; static int clean_single_torrent( ot_torrent *torrent ); +/* these functions protect our buckets from other threads that + try to commit announces or clean up */ +static ot_vector *lock_bucket_by_hash( ot_hash *hash ) { + unsigned char *local_hash = hash[0]; + int bucket = ( local_hash[0] << 2 ) | ( local_hash[1] >> 6 ); + + /* Can block */ + mutex_bucket_lock( bucket ); + + return all_torrents + bucket; +} + +static void *unlock_bucket_by_hash( ot_hash *hash ) { + unsigned char *local_hash = hash[0]; + int bucket = ( local_hash[0] << 2 ) | ( local_hash[1] >> 6 ); + mutex_bucket_unlock( bucket ); + + /* To make caller's code look better, allow + return unlock_bucket_by_hash() */ + return NULL; +} + /* Converter function from memory to human readable hex strings - definitely not thread safe!!! */ @@ -149,7 +176,7 @@ ot_torrent *add_peer_to_torrent( ot_hash *hash, ot_peer *peer WANT_TRACKER_SYNC int exactmatch; ot_torrent *torrent; ot_peer *peer_dest; - ot_vector *torrents_list = hash_to_bucket( all_torrents, hash ), *peer_pool; + ot_vector *torrents_list = lock_bucket_by_hash( hash ), *peer_pool; int base_pool = 0; #ifdef WANT_ACCESS_CONTROL @@ -160,11 +187,12 @@ ot_torrent *add_peer_to_torrent( ot_hash *hash, ot_peer *peer WANT_TRACKER_SYNC #endif if( exactmatch ) - return NULL; + return unlock_bucket_by_hash( hash ); #endif torrent = vector_find_or_insert( torrents_list, (void*)hash, sizeof( ot_torrent ), OT_HASH_COMPARE_SIZE, &exactmatch ); - if( !torrent ) return NULL; + if( !torrent ) + return unlock_bucket_by_hash( hash ); if( !exactmatch ) { /* Create a new torrent entry, then */ @@ -172,7 +200,7 @@ ot_torrent *add_peer_to_torrent( ot_hash *hash, ot_peer *peer WANT_TRACKER_SYNC if( !( torrent->peer_list = malloc( sizeof (ot_peerlist) ) ) ) { vector_remove_torrent( torrents_list, torrent ); - return NULL; + return unlock_bucket_by_hash( hash ); } byte_zero( torrent->peer_list, sizeof( ot_peerlist ) ); @@ -189,8 +217,10 @@ ot_torrent *add_peer_to_torrent( ot_hash *hash, ot_peer *peer WANT_TRACKER_SYNC /* Check, whether peer already is in current pool, do nothing if so */ peer_pool = &torrent->peer_list->peers[0]; binary_search( peer, peer_pool->data, peer_pool->size, sizeof(ot_peer), OT_PEER_COMPARE_SIZE, &exactmatch ); - if( exactmatch ) + if( exactmatch ) { + unlock_bucket_by_hash( hash ); return torrent; + } base_pool = 1; } #endif @@ -220,6 +250,7 @@ ot_torrent *add_peer_to_torrent( ot_hash *hash, ot_peer *peer WANT_TRACKER_SYNC torrent->peer_list->seed_count--; case 1: default: torrent->peer_list->peer_count--; + unlock_bucket_by_hash( hash ); return torrent; } } @@ -240,6 +271,7 @@ ot_torrent *add_peer_to_torrent( ot_hash *hash, ot_peer *peer WANT_TRACKER_SYNC memmove( peer_dest, peer, sizeof( ot_peer ) ); } + unlock_bucket_by_hash( hash ); return torrent; } @@ -249,11 +281,19 @@ ot_torrent *add_peer_to_torrent( ot_hash *hash, ot_peer *peer WANT_TRACKER_SYNC * RANDOM may return huge values * does not yet check not to return self */ -size_t return_peers_for_torrent( ot_torrent *torrent, size_t amount, char *reply, int is_tcp ) { +size_t return_peers_for_torrent( ot_hash *hash, size_t amount, char *reply, int is_tcp ) { char *r = reply; + int exactmatch; + ot_vector *torrents_list = lock_bucket_by_hash( hash ); + ot_torrent *torrent = binary_search( hash, torrents_list->data, torrents_list->size, sizeof( ot_torrent ), OT_HASH_COMPARE_SIZE, &exactmatch ); ot_peerlist *peer_list = torrent->peer_list; size_t index; + if( !torrent ) { + unlock_bucket_by_hash( hash ); + return 0; + } + if( peer_list->peer_count < amount ) amount = peer_list->peer_count; @@ -300,6 +340,7 @@ size_t return_peers_for_torrent( ot_torrent *torrent, size_t amount, char *reply if( is_tcp ) *r++ = 'e'; + unlock_bucket_by_hash( hash ); return r - reply; } @@ -385,8 +426,8 @@ size_t return_memstat_for_tracker( char **reply ) { /* Fetches scrape info for a specific torrent */ size_t return_udp_scrape_for_torrent( ot_hash *hash, char *reply ) { - int exactmatch ; - ot_vector *torrents_list = hash_to_bucket( all_torrents, hash ); + int exactmatch; + ot_vector *torrents_list = lock_bucket_by_hash( hash ); ot_torrent *torrent = binary_search( hash, torrents_list->data, torrents_list->size, sizeof( ot_torrent ), OT_HASH_COMPARE_SIZE, &exactmatch ); if( !exactmatch ) { @@ -403,6 +444,7 @@ size_t return_udp_scrape_for_torrent( ot_hash *hash, char *reply ) { r[2] = htonl( torrent->peer_list->peer_count-torrent->peer_list->seed_count ); } } + unlock_bucket_by_hash( hash ); return 12; } @@ -415,17 +457,19 @@ size_t return_tcp_scrape_for_torrent( ot_hash *hash_list, int amount, char *repl for( i=0; idata, torrents_list->size, sizeof( ot_torrent ), OT_HASH_COMPARE_SIZE, &exactmatch ); - if( !exactmatch ) continue; - if( clean_single_torrent( torrent ) ) { - vector_remove_torrent( torrents_list, torrent ); - } else { - memmove( r, "20:", 3 ); memmove( r+3, hash, 20 ); - r += sprintf( r+23, "d8:completei%zde10:downloadedi%zde10:incompletei%zdee", - torrent->peer_list->seed_count, torrent->peer_list->down_count, torrent->peer_list->peer_count-torrent->peer_list->seed_count ) + 23; + if( exactmatch ) { + if( clean_single_torrent( torrent ) ) { + vector_remove_torrent( torrents_list, torrent ); + } else { + memmove( r, "20:", 3 ); memmove( r+3, hash, 20 ); + r += sprintf( r+23, "d8:completei%zde10:downloadedi%zde10:incompletei%zdee", + torrent->peer_list->seed_count, torrent->peer_list->down_count, torrent->peer_list->peer_count-torrent->peer_list->seed_count ) + 23; + } } + unlock_bucket_by_hash( hash ); } *r++ = 'e'; *r++ = 'e'; @@ -806,11 +850,13 @@ size_t return_stats_for_slash24s_old( char *reply, size_t amount, ot_dword thres size_t remove_peer_from_torrent( ot_hash *hash, ot_peer *peer, char *reply, int is_tcp ) { int exactmatch; size_t index; - ot_vector *torrents_list = hash_to_bucket( all_torrents, hash ); + ot_vector *torrents_list = lock_bucket_by_hash( hash ); ot_torrent *torrent = binary_search( hash, torrents_list->data, torrents_list->size, sizeof( ot_torrent ), OT_HASH_COMPARE_SIZE, &exactmatch ); ot_peerlist *peer_list; if( !exactmatch ) { + unlock_bucket_by_hash( hash ); + if( is_tcp ) return sprintf( reply, "d8:completei0e10:incompletei0e8:intervali%ie5:peers0:e", OT_CLIENT_REQUEST_INTERVAL_RANDOM ); @@ -834,17 +880,22 @@ size_t remove_peer_from_torrent( ot_hash *hash, ot_peer *peer, char *reply, int exit_loop: - if( is_tcp ) - return sprintf( reply, "d8:completei%zde10:incompletei%zde8:intervali%ie5:peers0:e", peer_list->seed_count, peer_list->peer_count - peer_list->seed_count, OT_CLIENT_REQUEST_INTERVAL_RANDOM ); + if( is_tcp ) { + size_t reply_size = sprintf( reply, "d8:completei%zde10:incompletei%zde8:intervali%ie5:peers0:e", peer_list->seed_count, peer_list->peer_count - peer_list->seed_count, OT_CLIENT_REQUEST_INTERVAL_RANDOM ); + unlock_bucket_by_hash( hash ); + return reply_size; + } /* else { Handle UDP reply */ ((ot_dword*)reply)[2] = htonl( OT_CLIENT_REQUEST_INTERVAL_RANDOM ); ((ot_dword*)reply)[3] = peer_list->peer_count - peer_list->seed_count; ((ot_dword*)reply)[4] = peer_list->seed_count; + + unlock_bucket_by_hash( hash ); return (size_t)20; } -int init_logic( const char * const serverdir ) { +int trackerlogic_init( const char * const serverdir ) { if( serverdir && chdir( serverdir ) ) { fprintf( stderr, "Could not chdir() to %s\n", serverdir ); return -1; @@ -855,10 +906,12 @@ int init_logic( const char * const serverdir ) { /* Initialize control structures */ byte_zero( all_torrents, sizeof( all_torrents ) ); + mutex_init( ); + return 0; } -void deinit_logic( void ) { +void trackerlogic_deinit( void ) { int i; size_t j; @@ -873,6 +926,8 @@ void deinit_logic( void ) { } byte_zero( all_torrents, sizeof (all_torrents)); byte_zero( all_torrents_clean, sizeof (all_torrents_clean)); + + mutex_deinit( ); } #ifdef WANT_ACCESS_CONTROL diff --git a/trackerlogic.h b/trackerlogic.h index 67b14e1..86737ab 100644 --- a/trackerlogic.h +++ b/trackerlogic.h @@ -46,14 +46,6 @@ typedef struct { #define OT_ADMINIP_MAX 64 #define OT_MAX_THREADS 16 -/* We maintain a list of 4096 pointers to sorted list of ot_torrent structs - Sort key is, of course, its hash */ -#define OT_BUCKET_COUNT 1024 -static inline ot_vector *hash_to_bucket( ot_vector *vectors, ot_hash *hash ) { - unsigned char *local_hash = hash[0]; - return vectors + ( ( local_hash[0] << 2 ) | ( local_hash[1] >> 6 ) ); -} - /* This list points to 9 pools of peers each grouped in five-minute-intervals thus achieving a timeout of 2700s or 45 minutes These pools are sorted by its binary content */ @@ -105,14 +97,14 @@ typedef struct { #define WANT_TRACKER_SYNC_PARAM( param ) #endif -int init_logic( const char * const serverdir ); -void deinit_logic( void ); +int trackerlogic_init( const char * const serverdir ); +void trackerlogic_deinit( void ); -enum { STATS_MRTG, STATS_TOP5, STATS_DMEM, STATS_TCP, STATS_UDP, STATS_SLASH24S, STATS_SLASH24S_OLD, SYNC_IN, SYNC_OUT, STATS_FULLSCRAPE }; +enum { STATS_CONNS, STATS_PEERS, STATS_TOP5, STATS_DMEM, STATS_TCP, STATS_UDP, STATS_SLASH24S, SYNC_IN, SYNC_OUT, STATS_FULLSCRAPE }; ot_torrent *add_peer_to_torrent( ot_hash *hash, ot_peer *peer WANT_TRACKER_SYNC_PARAM( int from_changeset ) ); size_t remove_peer_from_torrent( ot_hash *hash, ot_peer *peer, char *reply, int is_tcp ); -size_t return_peers_for_torrent( ot_torrent *torrent, size_t amount, char *reply, int is_tcp ); +size_t return_peers_for_torrent( ot_hash *hash, size_t amount, char *reply, int is_tcp ); size_t return_fullscrape_for_tracker( char **reply ); size_t return_tcp_scrape_for_torrent( ot_hash *hash, int amount, char *reply ); size_t return_udp_scrape_for_torrent( ot_hash *hash, char *reply );