159 lines
5.2 KiB
C
159 lines
5.2 KiB
C
/*
|
|
* This file Copyright (C) 2007-2008 Charles Kerr <charles@rebelbase.com>
|
|
*
|
|
* 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 TR_PEER_MGR_H
|
|
#define TR_PEER_MGR_H
|
|
|
|
#include <inttypes.h> /* uint16_t */
|
|
|
|
#ifdef WIN32
|
|
#include <winsock2.h> /* struct in_addr */
|
|
#else
|
|
#include <netinet/in.h> /* struct in_addr */
|
|
#endif
|
|
|
|
struct in_addr;
|
|
struct tr_handle;
|
|
struct tr_peer_stat;
|
|
struct tr_torrent;
|
|
typedef struct tr_peerMgr tr_peerMgr;
|
|
|
|
enum
|
|
{
|
|
/* corresponds to ut_pex's added.f flags */
|
|
ADDED_F_ENCRYPTION_FLAG = 1,
|
|
|
|
/* corresponds to ut_pex's added.f flags */
|
|
ADDED_F_SEED_FLAG = 2,
|
|
};
|
|
|
|
typedef struct tr_pex
|
|
{
|
|
struct in_addr in_addr;
|
|
uint16_t port;
|
|
uint8_t flags;
|
|
}
|
|
tr_pex;
|
|
|
|
int tr_pexCompare( const void * a,
|
|
const void * b );
|
|
|
|
tr_peerMgr* tr_peerMgrNew( struct tr_handle * );
|
|
|
|
void tr_peerMgrFree( tr_peerMgr * manager );
|
|
|
|
int tr_peerMgrPeerIsSeed(
|
|
const tr_peerMgr * mgr,
|
|
const uint8_t *
|
|
torrentHash,
|
|
const struct in_addr * addr );
|
|
|
|
void tr_peerMgrAddIncoming( tr_peerMgr * manager,
|
|
struct in_addr * addr,
|
|
uint16_t port,
|
|
int socket );
|
|
|
|
tr_pex * tr_peerMgrCompactToPex( const void * compact,
|
|
size_t compactLen,
|
|
const uint8_t * added_f,
|
|
size_t added_f_len,
|
|
size_t * setme_pex_count );
|
|
|
|
void tr_peerMgrAddPex( tr_peerMgr * manager,
|
|
const uint8_t * torrentHash,
|
|
uint8_t from,
|
|
const tr_pex * pex );
|
|
|
|
void tr_peerMgrSetBlame( tr_peerMgr * manager,
|
|
const uint8_t * torrentHash,
|
|
tr_piece_index_t pieceIndex,
|
|
int success );
|
|
|
|
int tr_peerMgrGetPeers( tr_peerMgr * manager,
|
|
const uint8_t * torrentHash,
|
|
tr_pex ** setme_pex );
|
|
|
|
void tr_peerMgrStartTorrent( tr_peerMgr * manager,
|
|
const uint8_t * torrentHash );
|
|
|
|
void tr_peerMgrStopTorrent( tr_peerMgr * manager,
|
|
const uint8_t * torrentHash );
|
|
|
|
void tr_peerMgrAddTorrent( tr_peerMgr * manager,
|
|
struct tr_torrent * tor );
|
|
|
|
void tr_peerMgrRemoveTorrent( tr_peerMgr * manager,
|
|
const uint8_t * torrentHash );
|
|
|
|
void tr_peerMgrTorrentAvailability(
|
|
const tr_peerMgr * manager,
|
|
const uint8_t *
|
|
torrentHash,
|
|
int8_t * tab,
|
|
unsigned int
|
|
tabCount );
|
|
|
|
struct tr_bitfield* tr_peerMgrGetAvailable( const tr_peerMgr * manager,
|
|
const uint8_t * torrentHash );
|
|
|
|
int tr_peerMgrHasConnections(
|
|
const tr_peerMgr * manager,
|
|
const uint8_t *
|
|
torrentHash );
|
|
|
|
void tr_peerMgrTorrentStats(
|
|
const tr_peerMgr * manager,
|
|
const uint8_t * torrentHash,
|
|
int *
|
|
setmePeersKnown,
|
|
int *
|
|
setmePeersConnected,
|
|
int *
|
|
setmeSeedsConnected,
|
|
int *
|
|
setmeWebseedsSendingToUs,
|
|
int *
|
|
setmePeersSendingToUs,
|
|
int *
|
|
setmePeersGettingFromUs,
|
|
int *
|
|
setmePeersFrom, /* <-- array of
|
|
TR_PEER_FROM__MAX
|
|
*/
|
|
double *
|
|
setmeRateToClient,
|
|
double *
|
|
setmeRateToPeers );
|
|
|
|
double tr_peerMgrGetRate( const tr_peerMgr * manager,
|
|
tr_direction direction );
|
|
|
|
struct tr_peer_stat * tr_peerMgrPeerStats( const tr_peerMgr * manager,
|
|
const uint8_t * torrentHash,
|
|
int * setmeCount );
|
|
|
|
float* tr_peerMgrWebSpeeds( const tr_peerMgr * manager,
|
|
const uint8_t * torrentHash );
|
|
|
|
|
|
struct tr_bitfield * tr_peerMgrGenerateAllowedSet(
|
|
const uint32_t setCount,
|
|
const uint32_t
|
|
pieceCount,
|
|
const uint8_t
|
|
infohash[20],
|
|
const struct in_addr *
|
|
ip );
|
|
|
|
|
|
#endif
|