mirror of git://erdgeist.org/opentracker
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
158 lines
4.5 KiB
C
158 lines
4.5 KiB
C
/* 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
|