mirror of git://erdgeist.org/opentracker
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
157 lines
4.5 KiB
157 lines
4.5 KiB
/* This software was written by Dirk Engling <erdgeist@erdgeist.org> |
|
It is considered beerware. Prost. Skol. Cheers or whatever. |
|
|
|
$id$ */ |
|
|
|
#ifndef __OT_TRACKERLOGIC_H__ |
|
#define __OT_TRACKERLOGIC_H__ |
|
|
|
#include <sys/types.h> |
|
#include <sys/time.h> |
|
#include <time.h> |
|
#include <stdint.h> |
|
|
|
typedef uint8_t ot_hash[20]; |
|
typedef time_t ot_time; |
|
typedef char ot_ip6[16]; |
|
#ifdef WANT_V6 |
|
#define OT_IP_SIZE 16 |
|
#define PEERS_BENCODED "6:peers6" |
|
#else |
|
#define OT_IP_SIZE 4 |
|
#define PEERS_BENCODED "5:peers" |
|
#endif |
|
|
|
/* Some tracker behaviour tunable */ |
|
#define OT_CLIENT_TIMEOUT 30 |
|
#define OT_CLIENT_TIMEOUT_CHECKINTERVAL 10 |
|
#define OT_CLIENT_TIMEOUT_SEND (60*15) |
|
#define OT_CLIENT_REQUEST_INTERVAL (60*30) |
|
#define OT_CLIENT_REQUEST_VARIATION (60*6) |
|
|
|
#define OT_TORRENT_TIMEOUT_HOURS 24 |
|
#define OT_TORRENT_TIMEOUT (60*OT_TORRENT_TIMEOUT_HOURS) |
|
|
|
#define OT_CLIENT_REQUEST_INTERVAL_RANDOM ( OT_CLIENT_REQUEST_INTERVAL - OT_CLIENT_REQUEST_VARIATION/2 + (int)( random( ) % OT_CLIENT_REQUEST_VARIATION ) ) |
|
|
|
/* If peers come back before 10 minutes, don't live sync them */ |
|
#define OT_CLIENT_SYNC_RENEW_BOUNDARY 10 |
|
|
|
/* Number of tracker admin ip addresses allowed */ |
|
#define OT_ADMINIP_MAX 64 |
|
#define OT_MAX_THREADS 16 |
|
|
|
#define OT_PEER_TIMEOUT 45 |
|
|
|
/* 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_BITS 10 |
|
|
|
#define OT_BUCKET_COUNT (1<<OT_BUCKET_COUNT_BITS) |
|
#define OT_BUCKET_COUNT_SHIFT (32-OT_BUCKET_COUNT_BITS) |
|
|
|
/* From opentracker.c */ |
|
extern time_t g_now_seconds; |
|
extern volatile int g_opentracker_running; |
|
#define g_now_minutes (g_now_seconds/60) |
|
|
|
extern uint32_t g_tracker_id; |
|
typedef enum { FLAG_TCP, FLAG_UDP, FLAG_MCA, FLAG_SELFPIPE } PROTO_FLAG; |
|
|
|
typedef struct { |
|
uint8_t data[OT_IP_SIZE+2+2]; |
|
} ot_peer; |
|
static const uint8_t PEER_FLAG_SEEDING = 0x80; |
|
static const uint8_t PEER_FLAG_COMPLETED = 0x40; |
|
static const uint8_t PEER_FLAG_STOPPED = 0x20; |
|
static const uint8_t PEER_FLAG_FROM_SYNC = 0x10; |
|
static const uint8_t PEER_FLAG_LEECHING = 0x00; |
|
|
|
#ifdef WANT_V6 |
|
#define OT_SETIP(peer,ip) memcpy((peer),(ip),(OT_IP_SIZE)) |
|
#else |
|
#define OT_SETIP(peer,ip) memcpy((peer),(((uint8_t*)ip)+12),(OT_IP_SIZE)) |
|
#endif |
|
#define OT_SETPORT(peer,port) memcpy(((uint8_t*)(peer))+(OT_IP_SIZE),(port),2) |
|
#define OT_PEERFLAG(peer) (((uint8_t*)(peer))[(OT_IP_SIZE)+2]) |
|
#define OT_PEERTIME(peer) (((uint8_t*)(peer))[(OT_IP_SIZE)+3]) |
|
|
|
#define OT_HASH_COMPARE_SIZE (sizeof(ot_hash)) |
|
#define OT_PEER_COMPARE_SIZE ((OT_IP_SIZE)+2) |
|
|
|
struct ot_peerlist; |
|
typedef struct ot_peerlist ot_peerlist; |
|
typedef struct { |
|
ot_hash hash; |
|
ot_peerlist *peer_list; |
|
} ot_torrent; |
|
|
|
#include "ot_vector.h" |
|
|
|
struct ot_peerlist { |
|
ot_time base; |
|
size_t seed_count; |
|
size_t peer_count; |
|
size_t down_count; |
|
/* normal peers vector or |
|
pointer to ot_vector[32] buckets if data != NULL and space == 0 |
|
*/ |
|
ot_vector peers; |
|
}; |
|
#define OT_PEERLIST_HASBUCKETS(peer_list) ((peer_list)->peers.size > (peer_list)->peers.space) |
|
|
|
struct ot_workstruct { |
|
/* Thread specific, static */ |
|
char *inbuf; |
|
#define G_INBUF_SIZE 8192 |
|
char *outbuf; |
|
#define G_OUTBUF_SIZE 8192 |
|
#ifdef _DEBUG_HTTPERROR |
|
char *debugbuf; |
|
#define G_DEBUGBUF_SIZE 8192 |
|
#endif |
|
|
|
/* HTTP specific, non static */ |
|
char *request; |
|
ssize_t request_size; |
|
char *reply; |
|
ssize_t reply_size; |
|
}; |
|
|
|
/* |
|
Exported functions |
|
*/ |
|
|
|
#ifdef WANT_SYNC_LIVE |
|
#define WANT_SYNC |
|
#endif |
|
|
|
#ifdef WANT_SYNC |
|
#define WANT_SYNC_PARAM( param ) , param |
|
#else |
|
#define WANT_SYNC_PARAM( param ) |
|
#endif |
|
|
|
#if defined WANT_V6 && defined WANT_LOG_NETWORKS |
|
#undef WANT_LOG_NETWORKS |
|
#endif |
|
|
|
void trackerlogic_init( ); |
|
void trackerlogic_deinit( void ); |
|
void exerr( char * message ); |
|
|
|
/* add_peer_to_torrent does only release the torrent bucket if from_sync is set, |
|
otherwise it is released in return_peers_for_torrent */ |
|
#define add_peer_to_torrent(hash,peer,proto) add_peer_to_torrent_and_return_peers(hash,peer,proto,0,NULL) |
|
size_t add_peer_to_torrent_and_return_peers( ot_hash hash, ot_peer *peer, PROTO_FLAG proto, size_t amount, char * reply ); |
|
size_t remove_peer_from_torrent( ot_hash hash, ot_peer *peer, char *reply, PROTO_FLAG proto ); |
|
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 ); |
|
|
|
/* torrent iterator */ |
|
void iterate_all_torrents( int (*for_each)( ot_torrent* torrent, uintptr_t data ), uintptr_t data ); |
|
|
|
/* Helper, before it moves to its own object */ |
|
void free_peerlist( ot_peerlist *peer_list ); |
|
|
|
#endif
|
|
|