|
|
|
|
@ -22,8 +22,7 @@
@@ -22,8 +22,7 @@
|
|
|
|
|
#include <sys/stat.h> |
|
|
|
|
#endif |
|
|
|
|
|
|
|
|
|
// GLOBAL VARIABLES
|
|
|
|
|
//
|
|
|
|
|
/* GLOBAL VARIABLES */ |
|
|
|
|
static ot_vector all_torrents[256]; |
|
|
|
|
|
|
|
|
|
#ifdef WANT_CLOSED_TRACKER |
|
|
|
|
@ -36,36 +35,37 @@ int g_check_blacklist = 1;
@@ -36,36 +35,37 @@ int g_check_blacklist = 1;
|
|
|
|
|
static ot_torrent* const OT_TORRENT_ON_BLACKLIST = (ot_torrent*)2; |
|
|
|
|
#endif |
|
|
|
|
|
|
|
|
|
// This function gives us a binary search that returns a pointer, even if
|
|
|
|
|
// no exact match is found. In that case it sets exactmatch 0 and gives
|
|
|
|
|
// calling functions the chance to insert data
|
|
|
|
|
//
|
|
|
|
|
static void *binary_search( const void *key, const void *base, unsigned long member_count, const unsigned long member_size, |
|
|
|
|
int compare_size, int *exactmatch ) { |
|
|
|
|
/* This function gives us a binary search that returns a pointer, even if
|
|
|
|
|
no exact match is found. In that case it sets exactmatch 0 and gives |
|
|
|
|
calling functions the chance to insert data |
|
|
|
|
*/ |
|
|
|
|
static void *binary_search( const void * const key, const void * base, const size_t member_count, const size_t member_size, |
|
|
|
|
size_t compare_size, int *exactmatch ) { |
|
|
|
|
size_t mc = member_count; |
|
|
|
|
ot_byte *lookat = ((ot_byte*)base) + member_size * (member_count >> 1); |
|
|
|
|
*exactmatch = 1; |
|
|
|
|
|
|
|
|
|
while( member_count ) { |
|
|
|
|
while( mc ) { |
|
|
|
|
int cmp = memcmp( lookat, key, compare_size); |
|
|
|
|
if (cmp == 0) return (void *)lookat; |
|
|
|
|
if (cmp < 0) { |
|
|
|
|
base = (void*)(lookat + member_size); |
|
|
|
|
--member_count; |
|
|
|
|
--mc; |
|
|
|
|
} |
|
|
|
|
member_count >>= 1; |
|
|
|
|
lookat = ((ot_byte*)base) + member_size * (member_count >> 1); |
|
|
|
|
mc >>= 1; |
|
|
|
|
lookat = ((ot_byte*)base) + member_size * (mc >> 1); |
|
|
|
|
} |
|
|
|
|
*exactmatch = 0; |
|
|
|
|
return (void*)lookat; |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
// Converter function from memory to human readable hex strings
|
|
|
|
|
// * definitely not thread safe!!!
|
|
|
|
|
//
|
|
|
|
|
/* Converter function from memory to human readable hex strings
|
|
|
|
|
- definitely not thread safe!!! |
|
|
|
|
*/ |
|
|
|
|
char ths[2+2*20]="-";char*to_hex(ot_byte*s){char*m="0123456789ABCDEF";char*e=ths+41;char*t=ths+1;while(t<e){*t++=m[*s>>4];*t++=m[*s++&15];}*t=0;return ths+1;} |
|
|
|
|
|
|
|
|
|
static void *vector_find_or_insert( ot_vector *vector, void *key, size_t member_size, int compare_size, int *exactmatch ) { |
|
|
|
|
ot_byte *match = BINARY_FIND( key, vector->data, vector->size, member_size, compare_size, exactmatch ); |
|
|
|
|
ot_byte *match = binary_search( key, vector->data, vector->size, member_size, compare_size, exactmatch ); |
|
|
|
|
|
|
|
|
|
if( *exactmatch ) return match; |
|
|
|
|
|
|
|
|
|
@ -74,13 +74,13 @@ static void *vector_find_or_insert( ot_vector *vector, void *key, size_t member_
@@ -74,13 +74,13 @@ static void *vector_find_or_insert( ot_vector *vector, void *key, size_t member_
|
|
|
|
|
ot_byte *new_data = realloc( vector->data, new_space * member_size ); |
|
|
|
|
if( !new_data ) return NULL; |
|
|
|
|
|
|
|
|
|
// Adjust pointer if it moved by realloc
|
|
|
|
|
/* Adjust pointer if it moved by realloc */ |
|
|
|
|
match = new_data + (match - (ot_byte*)vector->data); |
|
|
|
|
|
|
|
|
|
vector->data = new_data; |
|
|
|
|
vector->space = new_space; |
|
|
|
|
} |
|
|
|
|
MEMMOVE( match + member_size, match, ((ot_byte*)vector->data) + member_size * vector->size - match ); |
|
|
|
|
memmove( match + member_size, match, ((ot_byte*)vector->data) + member_size * vector->size - match ); |
|
|
|
|
vector->size++; |
|
|
|
|
return match; |
|
|
|
|
} |
|
|
|
|
@ -91,11 +91,11 @@ static int vector_remove_peer( ot_vector *vector, ot_peer *peer ) {
@@ -91,11 +91,11 @@ static int vector_remove_peer( ot_vector *vector, ot_peer *peer ) {
|
|
|
|
|
ot_peer *match; |
|
|
|
|
|
|
|
|
|
if( !vector->size ) return 0; |
|
|
|
|
match = BINARY_FIND( peer, vector->data, vector->size, sizeof( ot_peer ), OT_PEER_COMPARE_SIZE, &exactmatch ); |
|
|
|
|
match = binary_search( peer, vector->data, vector->size, sizeof( ot_peer ), OT_PEER_COMPARE_SIZE, &exactmatch ); |
|
|
|
|
|
|
|
|
|
if( !exactmatch ) return 0; |
|
|
|
|
exactmatch = ( OT_FLAG( match ) & PEER_FLAG_SEEDING ) ? 2 : 1; |
|
|
|
|
MEMMOVE( match, match + 1, sizeof(ot_peer) * ( end - match - 1 ) ); |
|
|
|
|
memmove( match, match + 1, sizeof(ot_peer) * ( end - match - 1 ) ); |
|
|
|
|
if( ( --vector->size * OT_VECTOR_SHRINK_THRESH < vector->space ) && ( vector->space > OT_VECTOR_MIN_MEMBERS ) ) { |
|
|
|
|
vector->space /= OT_VECTOR_SHRINK_RATIO; |
|
|
|
|
vector->data = realloc( vector->data, vector->space * sizeof( ot_peer ) ); |
|
|
|
|
@ -117,15 +117,15 @@ static int vector_remove_torrent( ot_vector *vector, ot_hash *hash ) {
@@ -117,15 +117,15 @@ static int vector_remove_torrent( ot_vector *vector, ot_hash *hash ) {
|
|
|
|
|
ot_torrent *match; |
|
|
|
|
|
|
|
|
|
if( !vector->size ) return 0; |
|
|
|
|
match = BINARY_FIND( hash, vector->data, vector->size, sizeof( ot_torrent ), OT_HASH_COMPARE_SIZE, &exactmatch ); |
|
|
|
|
match = binary_search( hash, vector->data, vector->size, sizeof( ot_torrent ), OT_HASH_COMPARE_SIZE, &exactmatch ); |
|
|
|
|
|
|
|
|
|
if( !exactmatch ) return 0; |
|
|
|
|
|
|
|
|
|
// If this is being called after a unsuccessful malloc() for peer_list
|
|
|
|
|
// in add_peer_to_torrent, match->peer_list actually might be NULL
|
|
|
|
|
/* If this is being called after a unsuccessful malloc() for peer_list
|
|
|
|
|
in add_peer_to_torrent, match->peer_list actually might be NULL */ |
|
|
|
|
if( match->peer_list) free_peerlist( match->peer_list ); |
|
|
|
|
|
|
|
|
|
MEMMOVE( match, match + 1, sizeof(ot_torrent) * ( end - match - 1 ) ); |
|
|
|
|
memmove( match, match + 1, sizeof(ot_torrent) * ( end - match - 1 ) ); |
|
|
|
|
if( ( --vector->size * OT_VECTOR_SHRINK_THRESH < vector->space ) && ( vector->space > OT_VECTOR_MIN_MEMBERS ) ) { |
|
|
|
|
vector->space /= OT_VECTOR_SHRINK_RATIO; |
|
|
|
|
vector->data = realloc( vector->data, vector->space * sizeof( ot_torrent ) ); |
|
|
|
|
@ -133,22 +133,21 @@ static int vector_remove_torrent( ot_vector *vector, ot_hash *hash ) {
@@ -133,22 +133,21 @@ static int vector_remove_torrent( ot_vector *vector, ot_hash *hash ) {
|
|
|
|
|
return 1; |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
// Returns 1, if torrent is gone, 0 otherwise
|
|
|
|
|
/* Returns 1, if torrent is gone, 0 otherwise */ |
|
|
|
|
static int clean_peerlist( ot_peerlist *peer_list ) { |
|
|
|
|
long timedout = NOW-peer_list->base; |
|
|
|
|
int i; |
|
|
|
|
int i, timedout = (int)( NOW - peer_list->base ); |
|
|
|
|
|
|
|
|
|
if( !timedout ) return 0; |
|
|
|
|
if( timedout > OT_POOLS_COUNT ) timedout = OT_POOLS_COUNT; |
|
|
|
|
|
|
|
|
|
for( i=OT_POOLS_COUNT-timedout; i<OT_POOLS_COUNT; ++i ) |
|
|
|
|
for( i = OT_POOLS_COUNT - timedout; i < OT_POOLS_COUNT; ++i ) |
|
|
|
|
free( peer_list->peers[i].data); |
|
|
|
|
|
|
|
|
|
MEMMOVE( peer_list->peers + timedout, peer_list->peers, sizeof( ot_vector ) * (OT_POOLS_COUNT-timedout) ); |
|
|
|
|
memmove( peer_list->peers + timedout, peer_list->peers, sizeof( ot_vector ) * (OT_POOLS_COUNT-timedout) ); |
|
|
|
|
byte_zero( peer_list->peers, sizeof( ot_vector ) * timedout ); |
|
|
|
|
|
|
|
|
|
MEMMOVE( peer_list->seed_count + timedout, peer_list->seed_count, sizeof( unsigned long ) * (OT_POOLS_COUNT-timedout) ); |
|
|
|
|
byte_zero( peer_list->seed_count, sizeof( unsigned long ) * timedout ); |
|
|
|
|
memmove( peer_list->seed_count + timedout, peer_list->seed_count, sizeof( size_t ) * ( OT_POOLS_COUNT - timedout) ); |
|
|
|
|
byte_zero( peer_list->seed_count, sizeof( size_t ) * timedout ); |
|
|
|
|
|
|
|
|
|
peer_list->base = NOW; |
|
|
|
|
return timedout == OT_POOLS_COUNT; |
|
|
|
|
@ -178,8 +177,8 @@ ot_torrent *add_peer_to_torrent( ot_hash *hash, ot_peer *peer ) {
@@ -178,8 +177,8 @@ ot_torrent *add_peer_to_torrent( ot_hash *hash, ot_peer *peer ) {
|
|
|
|
|
if( !torrent ) return NULL; |
|
|
|
|
|
|
|
|
|
if( !exactmatch ) { |
|
|
|
|
// Create a new torrent entry, then
|
|
|
|
|
MEMMOVE( &torrent->hash, hash, sizeof( ot_hash ) ); |
|
|
|
|
/* Create a new torrent entry, then */ |
|
|
|
|
memmove( &torrent->hash, hash, sizeof( ot_hash ) ); |
|
|
|
|
|
|
|
|
|
torrent->peer_list = malloc( sizeof (ot_peerlist) ); |
|
|
|
|
if( !torrent->peer_list ) { |
|
|
|
|
@ -195,11 +194,11 @@ ot_torrent *add_peer_to_torrent( ot_hash *hash, ot_peer *peer ) {
@@ -195,11 +194,11 @@ ot_torrent *add_peer_to_torrent( ot_hash *hash, ot_peer *peer ) {
|
|
|
|
|
peer_pool = &torrent->peer_list->peers[0]; |
|
|
|
|
peer_dest = vector_find_or_insert( peer_pool, (void*)peer, sizeof( ot_peer ), OT_PEER_COMPARE_SIZE, &exactmatch ); |
|
|
|
|
|
|
|
|
|
// If we hadn't had a match in current pool, create peer there and
|
|
|
|
|
// remove it from all older pools
|
|
|
|
|
/* If we hadn't had a match in current pool, create peer there and
|
|
|
|
|
remove it from all older pools */ |
|
|
|
|
if( !exactmatch ) { |
|
|
|
|
int i; |
|
|
|
|
MEMMOVE( peer_dest, peer, sizeof( ot_peer ) ); |
|
|
|
|
memmove( peer_dest, peer, sizeof( ot_peer ) ); |
|
|
|
|
if( OT_FLAG(peer) & PEER_FLAG_SEEDING ) |
|
|
|
|
torrent->peer_list->seed_count[0]++; |
|
|
|
|
for( i=1; i<OT_POOLS_COUNT; ++i ) { |
|
|
|
|
@ -221,27 +220,30 @@ ot_torrent *add_peer_to_torrent( ot_hash *hash, ot_peer *peer ) {
@@ -221,27 +220,30 @@ ot_torrent *add_peer_to_torrent( ot_hash *hash, ot_peer *peer ) {
|
|
|
|
|
return torrent; |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
// Compiles a list of random peers for a torrent
|
|
|
|
|
// * reply must have enough space to hold 24+6*amount bytes
|
|
|
|
|
// * Selector function can be anything, maybe test for seeds, etc.
|
|
|
|
|
// * RANDOM may return huge values
|
|
|
|
|
// * does not yet check not to return self
|
|
|
|
|
//
|
|
|
|
|
size_t return_peers_for_torrent( ot_torrent *torrent, unsigned long amount, char *reply ) { |
|
|
|
|
/* Compiles a list of random peers for a torrent
|
|
|
|
|
* reply must have enough space to hold 24+6*amount bytes |
|
|
|
|
* Selector function can be anything, maybe test for seeds, etc. |
|
|
|
|
* RANDOM may return huge values |
|
|
|
|
* does not yet check not to return self |
|
|
|
|
*/ |
|
|
|
|
size_t return_peers_for_torrent( ot_torrent *torrent, unsigned int amount, char *reply ) { |
|
|
|
|
char *r = reply; |
|
|
|
|
unsigned long peer_count, seed_count, index; |
|
|
|
|
signed long pool_offset = -1, pool_index = 0; |
|
|
|
|
signed long wert = -1; |
|
|
|
|
unsigned int peer_count, seed_count, index; |
|
|
|
|
int pool_offset = -1, pool_index = 0, wert = -1; |
|
|
|
|
|
|
|
|
|
#ifdef WANT_CLOSED_TRACKER |
|
|
|
|
if( torrent == OT_TORRENT_NOT_ON_WHITELIST ) { |
|
|
|
|
return( FORMAT_FORMAT_STRING( reply, "d14:failure reason43:This torrent is not served by this tracker.e" ) ); |
|
|
|
|
const char * const notvalid = "d14:failure reason43:This torrent is not served by this tracker.e"; |
|
|
|
|
memmove( reply, notvalid, sizeof(notvalid)); |
|
|
|
|
return sizeof(notvalid); |
|
|
|
|
} |
|
|
|
|
#endif |
|
|
|
|
|
|
|
|
|
#ifdef WANT_BLACKLIST |
|
|
|
|
if( torrent == OT_TORRENT_ON_BLACKLIST ) { |
|
|
|
|
return( FORMAT_FORMAT_STRING( reply, "d14:failure reason29:This torrent is black listed.e" ) ); |
|
|
|
|
const char * const blacklisted = "d14:failure reason29:This torrent is black listed.e"; |
|
|
|
|
memmove( reply, blacklisted, sizeof(blacklisted)); |
|
|
|
|
return sizeof(blacklisted); |
|
|
|
|
} |
|
|
|
|
#endif |
|
|
|
|
|
|
|
|
|
@ -251,23 +253,19 @@ size_t return_peers_for_torrent( ot_torrent *torrent, unsigned long amount, char
@@ -251,23 +253,19 @@ size_t return_peers_for_torrent( ot_torrent *torrent, unsigned long amount, char
|
|
|
|
|
} |
|
|
|
|
if( peer_count < amount ) amount = peer_count; |
|
|
|
|
|
|
|
|
|
r += FORMAT_FORMAT_STRING( r, "d8:completei%lie10:incompletei%lie8:intervali600e5:peers%li:", seed_count, peer_count-seed_count, 6*amount ); |
|
|
|
|
r += sprintf( r, "d8:completei%ie10:incompletei%ie8:intervali600e5:peers%i:", seed_count, peer_count-seed_count, 6*amount ); |
|
|
|
|
for( index = 0; index < amount; ++index ) { |
|
|
|
|
double step = 1.8*((double)( peer_count - wert - 1 ))/((double)( amount - index )); |
|
|
|
|
int off = random() % (int)floor( step ); |
|
|
|
|
off = 1 + ( off % ( peer_count - wert - 1 )); |
|
|
|
|
wert += off; pool_offset += off; |
|
|
|
|
|
|
|
|
|
// In some rare occasions random gets the last peer a round to early
|
|
|
|
|
// correct that and return last peer twice
|
|
|
|
|
// if( wert >= peer_count ) { wert--; pool_offset--; }
|
|
|
|
|
|
|
|
|
|
while( pool_offset >= torrent->peer_list->peers[pool_index].size ) { |
|
|
|
|
pool_offset -= torrent->peer_list->peers[pool_index].size; |
|
|
|
|
pool_index++; |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
MEMMOVE( r, ((ot_peer*)torrent->peer_list->peers[pool_index].data) + pool_offset, 6 ); |
|
|
|
|
memmove( r, ((ot_peer*)torrent->peer_list->peers[pool_index].data) + pool_offset, 6 ); |
|
|
|
|
r += 6; |
|
|
|
|
} |
|
|
|
|
*r++ = 'e'; |
|
|
|
|
@ -275,12 +273,12 @@ size_t return_peers_for_torrent( ot_torrent *torrent, unsigned long amount, char
@@ -275,12 +273,12 @@ size_t return_peers_for_torrent( ot_torrent *torrent, unsigned long amount, char
|
|
|
|
|
return r - reply; |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
// Fetches scrape info for a specific torrent
|
|
|
|
|
/* Fetches scrape info for a specific torrent */ |
|
|
|
|
size_t return_scrape_for_torrent( ot_hash *hash, char *reply ) { |
|
|
|
|
char *r = reply; |
|
|
|
|
int exactmatch, peers = 0, seeds = 0, i; |
|
|
|
|
ot_vector *torrents_list = &all_torrents[*hash[0]]; |
|
|
|
|
ot_torrent *torrent = BINARY_FIND( hash, torrents_list->data, torrents_list->size, sizeof( ot_torrent ), OT_HASH_COMPARE_SIZE, &exactmatch ); |
|
|
|
|
ot_torrent *torrent = binary_search( hash, torrents_list->data, torrents_list->size, sizeof( ot_torrent ), OT_HASH_COMPARE_SIZE, &exactmatch ); |
|
|
|
|
|
|
|
|
|
if( !exactmatch ) return 0; |
|
|
|
|
clean_peerlist( torrent->peer_list ); |
|
|
|
|
@ -290,8 +288,8 @@ size_t return_scrape_for_torrent( ot_hash *hash, char *reply ) {
@@ -290,8 +288,8 @@ size_t return_scrape_for_torrent( ot_hash *hash, char *reply ) {
|
|
|
|
|
seeds += torrent->peer_list->seed_count[i]; |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
MEMMOVE( r, "d5:filesd20:", 12 ); MEMMOVE( r+12, hash, 20 ); |
|
|
|
|
r += FORMAT_FORMAT_STRING( r+32, "d8:completei%de10:downloadedi%lde10:incompletei%deeee", seeds, torrent->peer_list->downloaded, peers-seeds ) + 32; |
|
|
|
|
memmove( r, "d5:filesd20:", 12 ); memmove( r+12, hash, 20 ); |
|
|
|
|
r += sprintf( r+32, "d8:completei%de10:downloadedi%de10:incompletei%deeee", seeds, torrent->peer_list->downloaded, peers-seeds ) + 32; |
|
|
|
|
|
|
|
|
|
return r - reply; |
|
|
|
|
} |
|
|
|
|
@ -299,11 +297,11 @@ size_t return_scrape_for_torrent( ot_hash *hash, char *reply ) {
@@ -299,11 +297,11 @@ size_t return_scrape_for_torrent( ot_hash *hash, char *reply ) {
|
|
|
|
|
void remove_peer_from_torrent( ot_hash *hash, ot_peer *peer ) { |
|
|
|
|
int exactmatch, i; |
|
|
|
|
ot_vector *torrents_list = &all_torrents[*hash[0]]; |
|
|
|
|
ot_torrent *torrent = BINARY_FIND( hash, torrents_list->data, torrents_list->size, sizeof( ot_torrent ), OT_HASH_COMPARE_SIZE, &exactmatch ); |
|
|
|
|
ot_torrent *torrent = binary_search( hash, torrents_list->data, torrents_list->size, sizeof( ot_torrent ), OT_HASH_COMPARE_SIZE, &exactmatch ); |
|
|
|
|
|
|
|
|
|
if( !exactmatch ) return; |
|
|
|
|
|
|
|
|
|
// Maybe this does the job
|
|
|
|
|
/* Maybe this does the job */ |
|
|
|
|
if( clean_peerlist( torrent->peer_list ) ) { |
|
|
|
|
#ifdef WANT_CLOSED_TRACKER |
|
|
|
|
if( !g_closedtracker ) |
|
|
|
|
@ -332,7 +330,7 @@ int init_logic( char *serverdir ) {
@@ -332,7 +330,7 @@ int init_logic( char *serverdir ) {
|
|
|
|
|
|
|
|
|
|
srandom( time(NULL)); |
|
|
|
|
|
|
|
|
|
// Initialize control structures
|
|
|
|
|
/* Initialize control structures */ |
|
|
|
|
byte_zero( all_torrents, sizeof (all_torrents)); |
|
|
|
|
|
|
|
|
|
return 0; |
|
|
|
|
@ -340,7 +338,8 @@ int init_logic( char *serverdir ) {
@@ -340,7 +338,8 @@ int init_logic( char *serverdir ) {
|
|
|
|
|
|
|
|
|
|
void deinit_logic( ) { |
|
|
|
|
int i, j; |
|
|
|
|
// Free all torrents...
|
|
|
|
|
|
|
|
|
|
/* Free all torrents... */ |
|
|
|
|
for(i=0; i<256; ++i ) { |
|
|
|
|
if( all_torrents[i].size ) { |
|
|
|
|
ot_torrent *torrents_list = (ot_torrent*)all_torrents[i].data; |
|
|
|
|
|
erdgeist
18 years ago