336 lines
9.9 KiB
C
336 lines
9.9 KiB
C
/*
|
|
* This file Copyright (C) Mnemosyne LLC
|
|
*
|
|
* This file is licensed by the GPL version 2. Works owned by the
|
|
* Transmission project are granted a special exemption to clause 2(b)
|
|
* so that the bulk of its code can remain under the MIT license.
|
|
* This exemption does not extend to derived works not owned by
|
|
* the Transmission project.
|
|
*
|
|
* $Id$
|
|
*/
|
|
|
|
#ifndef __TRANSMISSION__
|
|
#error only libtransmission should #include this header.
|
|
#endif
|
|
|
|
#ifndef TR_INTERNAL_H
|
|
#define TR_INTERNAL_H 1
|
|
|
|
#define TR_NAME "Transmission"
|
|
|
|
#ifndef UNUSED
|
|
#ifdef __GNUC__
|
|
#define UNUSED __attribute__ ( ( unused ) )
|
|
#else
|
|
#define UNUSED
|
|
#endif
|
|
#endif
|
|
|
|
#include "bandwidth.h"
|
|
#include "bencode.h"
|
|
#include "bitfield.h"
|
|
#include "utils.h"
|
|
|
|
typedef enum { TR_NET_OK, TR_NET_ERROR, TR_NET_WAIT } tr_tristate_t;
|
|
|
|
typedef enum {
|
|
TR_AUTO_SWITCH_UNUSED,
|
|
TR_AUTO_SWITCH_ON,
|
|
TR_AUTO_SWITCH_OFF,
|
|
} tr_auto_switch_state_t;
|
|
|
|
enum
|
|
{
|
|
PEER_ID_LEN = 20
|
|
};
|
|
|
|
void tr_peerIdInit( uint8_t * setme );
|
|
|
|
struct event_base;
|
|
struct evdns_base;
|
|
|
|
struct tr_address;
|
|
struct tr_announcer;
|
|
struct tr_announcer_udp;
|
|
struct tr_bindsockets;
|
|
struct tr_cache;
|
|
struct tr_fdInfo;
|
|
|
|
typedef void ( tr_web_config_func )( tr_session * session, void * curl_pointer, const char * url, void * user_data );
|
|
|
|
struct tr_turtle_info
|
|
{
|
|
/* TR_UP and TR_DOWN speed limits */
|
|
unsigned int speedLimit_Bps[2];
|
|
|
|
/* is turtle mode on right now? */
|
|
bool isEnabled;
|
|
|
|
/* does turtle mode turn itself on and off at given times? */
|
|
bool isClockEnabled;
|
|
|
|
/* when clock mode is on, minutes after midnight to turn on turtle mode */
|
|
int beginMinute;
|
|
|
|
/* when clock mode is on, minutes after midnight to turn off turtle mode */
|
|
int endMinute;
|
|
|
|
/* only use clock mode on these days of the week */
|
|
tr_sched_day days;
|
|
|
|
/* called when isEnabled changes */
|
|
tr_altSpeedFunc * callback;
|
|
|
|
/* the callback's user_data argument */
|
|
void * callbackUserData;
|
|
|
|
/* the callback's changedByUser argument.
|
|
* indicates whether the change came from the user or from the clock. */
|
|
bool changedByUser;
|
|
|
|
/* bitfield of all the minutes in a week.
|
|
* Each bit's value indicates whether the scheduler wants turtle
|
|
* limits on or off at that given minute in the week. */
|
|
tr_bitfield minutes;
|
|
|
|
/* recent action that was done by turtle's automatic switch */
|
|
tr_auto_switch_state_t autoTurtleState;
|
|
};
|
|
|
|
/** @brief handle to an active libtransmission session */
|
|
struct tr_session
|
|
{
|
|
bool isPortRandom;
|
|
bool isPexEnabled;
|
|
bool isDHTEnabled;
|
|
bool isUTPEnabled;
|
|
bool isLPDEnabled;
|
|
bool isBlocklistEnabled;
|
|
bool isPrefetchEnabled;
|
|
bool isTorrentDoneScriptEnabled;
|
|
bool isClosed;
|
|
bool isIncompleteFileNamingEnabled;
|
|
bool isRatioLimited;
|
|
bool isIdleLimited;
|
|
bool isIncompleteDirEnabled;
|
|
bool pauseAddedTorrent;
|
|
bool deleteSourceTorrent;
|
|
bool scrapePausedTorrents;
|
|
|
|
tr_benc removedTorrents;
|
|
|
|
bool stalledEnabled;
|
|
bool queueEnabled[2];
|
|
int queueSize[2];
|
|
int queueStalledMinutes;
|
|
|
|
int umask;
|
|
|
|
unsigned int speedLimit_Bps[2];
|
|
bool speedLimitEnabled[2];
|
|
|
|
struct tr_turtle_info turtle;
|
|
|
|
struct tr_fdInfo * fdInfo;
|
|
|
|
int magicNumber;
|
|
|
|
tr_encryption_mode encryptionMode;
|
|
|
|
tr_preallocation_mode preallocationMode;
|
|
|
|
struct event_base * event_base;
|
|
struct evdns_base * evdns_base;
|
|
struct tr_event_handle * events;
|
|
|
|
uint16_t peerLimit;
|
|
uint16_t peerLimitPerTorrent;
|
|
|
|
int uploadSlotsPerTorrent;
|
|
|
|
/* The UDP sockets used for the DHT and uTP. */
|
|
tr_port udp_port;
|
|
int udp_socket;
|
|
int udp6_socket;
|
|
unsigned char * udp6_bound;
|
|
struct event *udp_event;
|
|
struct event *udp6_event;
|
|
|
|
/* The open port on the local machine for incoming peer requests */
|
|
tr_port private_peer_port;
|
|
|
|
/**
|
|
* The open port on the public device for incoming peer requests.
|
|
* This is usually the same as private_peer_port but can differ
|
|
* if the public device is a router and it decides to use a different
|
|
* port than the one requested by Transmission.
|
|
*/
|
|
tr_port public_peer_port;
|
|
|
|
tr_port randomPortLow;
|
|
tr_port randomPortHigh;
|
|
|
|
int peerSocketTOS;
|
|
char * peer_congestion_algorithm;
|
|
|
|
int torrentCount;
|
|
tr_torrent * torrentList;
|
|
|
|
char * torrentDoneScript;
|
|
|
|
char * tag;
|
|
char * configDir;
|
|
char * downloadDir;
|
|
char * resumeDir;
|
|
char * torrentDir;
|
|
char * incompleteDir;
|
|
|
|
char * blocklist_url;
|
|
|
|
struct tr_list * blocklists;
|
|
struct tr_peerMgr * peerMgr;
|
|
struct tr_shared * shared;
|
|
|
|
struct tr_cache * cache;
|
|
|
|
struct tr_lock * lock;
|
|
|
|
struct tr_web * web;
|
|
|
|
struct tr_rpc_server * rpcServer;
|
|
tr_rpc_func rpc_func;
|
|
void * rpc_func_user_data;
|
|
|
|
struct tr_stats_handle * sessionStats;
|
|
|
|
struct tr_announcer * announcer;
|
|
struct tr_announcer_udp * announcer_udp;
|
|
|
|
tr_benc * metainfoLookup;
|
|
|
|
struct event * nowTimer;
|
|
struct event * saveTimer;
|
|
|
|
/* monitors the "global pool" speeds */
|
|
struct tr_bandwidth bandwidth;
|
|
|
|
float desiredRatio;
|
|
|
|
uint16_t idleLimitMinutes;
|
|
|
|
struct tr_bindinfo * public_ipv4;
|
|
struct tr_bindinfo * public_ipv6;
|
|
|
|
uint8_t peer_id[PEER_ID_LEN+1];
|
|
};
|
|
|
|
static inline tr_port
|
|
tr_sessionGetPublicPeerPort( const tr_session * session )
|
|
{
|
|
return session->public_peer_port;
|
|
}
|
|
|
|
static inline const uint8_t*
|
|
tr_getPeerId( tr_session * session )
|
|
{
|
|
return session->peer_id;
|
|
}
|
|
|
|
bool tr_sessionAllowsDHT( const tr_session * session );
|
|
|
|
bool tr_sessionAllowsLPD( const tr_session * session );
|
|
|
|
const char * tr_sessionFindTorrentFile( const tr_session * session,
|
|
const char * hashString );
|
|
|
|
void tr_sessionSetTorrentFile( tr_session * session,
|
|
const char * hashString,
|
|
const char * filename );
|
|
|
|
bool tr_sessionIsAddressBlocked( const tr_session * session,
|
|
const struct tr_address * addr );
|
|
|
|
void tr_sessionLock( tr_session * );
|
|
|
|
void tr_sessionUnlock( tr_session * );
|
|
|
|
bool tr_sessionIsLocked( const tr_session * );
|
|
|
|
const struct tr_address* tr_sessionGetPublicAddress( const tr_session * session,
|
|
int tr_af_type,
|
|
bool * is_default_value );
|
|
|
|
|
|
struct tr_bindsockets * tr_sessionGetBindSockets( tr_session * );
|
|
|
|
int tr_sessionCountTorrents( const tr_session * session );
|
|
|
|
enum
|
|
{
|
|
SESSION_MAGIC_NUMBER = 3845,
|
|
};
|
|
|
|
static inline bool tr_isSession( const tr_session * session )
|
|
{
|
|
return ( session != NULL ) && ( session->magicNumber == SESSION_MAGIC_NUMBER );
|
|
}
|
|
|
|
static inline bool tr_isPreallocationMode( tr_preallocation_mode m )
|
|
{
|
|
return ( m == TR_PREALLOCATE_NONE )
|
|
|| ( m == TR_PREALLOCATE_SPARSE )
|
|
|| ( m == TR_PREALLOCATE_FULL );
|
|
}
|
|
|
|
static inline bool tr_isEncryptionMode( tr_encryption_mode m )
|
|
{
|
|
return ( m == TR_CLEAR_PREFERRED )
|
|
|| ( m == TR_ENCRYPTION_PREFERRED )
|
|
|| ( m == TR_ENCRYPTION_REQUIRED );
|
|
}
|
|
|
|
static inline bool tr_isPriority( tr_priority_t p )
|
|
{
|
|
return ( p == TR_PRI_LOW )
|
|
|| ( p == TR_PRI_NORMAL )
|
|
|| ( p == TR_PRI_HIGH );
|
|
}
|
|
|
|
/***
|
|
****
|
|
***/
|
|
|
|
static inline unsigned int
|
|
toSpeedBytes ( unsigned int KBps ) { return KBps * tr_speed_K; }
|
|
static inline double
|
|
toSpeedKBps ( unsigned int Bps ) { return Bps / (double)tr_speed_K; }
|
|
|
|
static inline uint64_t
|
|
toMemBytes ( unsigned int MB ) { uint64_t B = tr_mem_K * tr_mem_K; B *= MB; return B; }
|
|
static inline int
|
|
toMemMB ( uint64_t B ) { return B / ( tr_mem_K * tr_mem_K ); }
|
|
|
|
/**
|
|
**/
|
|
|
|
unsigned int tr_sessionGetSpeedLimit_Bps( const tr_session *, tr_direction );
|
|
unsigned int tr_sessionGetAltSpeed_Bps ( const tr_session *, tr_direction );
|
|
unsigned int tr_sessionGetRawSpeed_Bps ( const tr_session *, tr_direction );
|
|
unsigned int tr_sessionGetPieceSpeed_Bps( const tr_session *, tr_direction );
|
|
|
|
void tr_sessionSetSpeedLimit_Bps( tr_session *, tr_direction, unsigned int Bps );
|
|
void tr_sessionSetAltSpeed_Bps ( tr_session *, tr_direction, unsigned int Bps );
|
|
|
|
bool tr_sessionGetActiveSpeedLimit_Bps( const tr_session * session,
|
|
tr_direction dir,
|
|
unsigned int * setme );
|
|
|
|
tr_torrent * tr_sessionGetNextQueuedSeed( tr_session * session );
|
|
tr_torrent * tr_sessionGetNextQueuedTorrent( tr_session * session, tr_direction );
|
|
|
|
int tr_sessionCountQueueFreeSlots( tr_session * session, tr_direction );
|
|
|
|
|
|
#endif
|