202 lines
7.5 KiB
C
202 lines
7.5 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_PEER_MGR_H
|
|
#define TR_PEER_MGR_H
|
|
|
|
#include <inttypes.h> /* uint16_t */
|
|
|
|
#ifdef WIN32
|
|
#include <winsock2.h> /* struct in_addr */
|
|
#endif
|
|
|
|
#include "net.h" /* tr_address */
|
|
#include "peer-common.h"
|
|
#include "quark.h"
|
|
|
|
/**
|
|
* @addtogroup peers Peers
|
|
* @{
|
|
*/
|
|
|
|
struct UTPSocket;
|
|
struct tr_peer_stat;
|
|
struct tr_torrent;
|
|
typedef struct tr_peerMgr tr_peerMgr;
|
|
|
|
/* added_f's bitwise-or'ed flags */
|
|
enum
|
|
{
|
|
/* true if the peer supports encryption */
|
|
ADDED_F_ENCRYPTION_FLAG = 1,
|
|
|
|
/* true if the peer is a seed or partial seed */
|
|
ADDED_F_SEED_FLAG = 2,
|
|
|
|
/* true if the peer supports uTP */
|
|
ADDED_F_UTP_FLAGS = 4,
|
|
|
|
/* true if the peer has holepunch support */
|
|
ADDED_F_HOLEPUNCH = 8,
|
|
|
|
/* true if the peer telling us about this peer
|
|
* initiated the connection (implying that it is connectible) */
|
|
ADDED_F_CONNECTABLE = 16
|
|
};
|
|
|
|
typedef struct tr_pex
|
|
{
|
|
tr_address addr;
|
|
tr_port port; /* this field is in network byte order */
|
|
uint8_t flags;
|
|
}
|
|
tr_pex;
|
|
|
|
struct peer_atom;
|
|
struct tr_peerIo;
|
|
struct tr_peerMsgs;
|
|
struct tr_swarm;
|
|
|
|
static inline bool
|
|
tr_isPex (const tr_pex * pex)
|
|
{
|
|
return pex && tr_address_is_valid (&pex->addr);
|
|
}
|
|
|
|
const tr_address * tr_peerAddress (const tr_peer *);
|
|
|
|
int tr_pexCompare (const void * a, const void * b);
|
|
|
|
tr_peerMgr * tr_peerMgrNew (tr_session * session);
|
|
|
|
void tr_peerMgrFree (tr_peerMgr * manager);
|
|
|
|
bool tr_peerMgrPeerIsSeed (const tr_torrent * tor,
|
|
const tr_address * addr);
|
|
|
|
void tr_peerMgrSetUtpSupported (tr_torrent * tor,
|
|
const tr_address * addr);
|
|
|
|
void tr_peerMgrSetUtpFailed (tr_torrent * tor,
|
|
const tr_address * addr,
|
|
bool failed);
|
|
|
|
void tr_peerMgrGetNextRequests (tr_torrent * torrent,
|
|
tr_peer * peer,
|
|
int numwant,
|
|
tr_block_index_t * setme,
|
|
int * numgot,
|
|
bool get_intervals);
|
|
|
|
bool tr_peerMgrDidPeerRequest (const tr_torrent * torrent,
|
|
const tr_peer * peer,
|
|
tr_block_index_t block);
|
|
|
|
void tr_peerMgrRebuildRequests (tr_torrent * torrent);
|
|
|
|
void tr_peerMgrAddIncoming (tr_peerMgr * manager,
|
|
tr_address * addr,
|
|
tr_port port,
|
|
int socket,
|
|
struct UTPSocket * utp_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);
|
|
|
|
tr_pex * tr_peerMgrCompact6ToPex (const void * compact,
|
|
size_t compactLen,
|
|
const uint8_t * added_f,
|
|
size_t added_f_len,
|
|
size_t * pexCount);
|
|
|
|
tr_pex * tr_peerMgrArrayToPex (const void * array,
|
|
size_t arrayLen,
|
|
size_t * setme_pex_count);
|
|
|
|
/**
|
|
* @param seedProbability [0..100] for likelihood that the peer is a seed; -1 for unknown
|
|
*/
|
|
void tr_peerMgrAddPex (tr_torrent * tor,
|
|
uint8_t from,
|
|
const tr_pex * pex,
|
|
int8_t seedProbability);
|
|
|
|
void tr_peerMgrMarkAllAsSeeds (tr_torrent * tor);
|
|
|
|
enum
|
|
{
|
|
TR_PEERS_CONNECTED,
|
|
TR_PEERS_INTERESTING
|
|
};
|
|
|
|
int tr_peerMgrGetPeers (tr_torrent * tor,
|
|
tr_pex ** setme_pex,
|
|
uint8_t address_type,
|
|
uint8_t peer_list_mode,
|
|
int max_peer_count);
|
|
|
|
void tr_peerMgrStartTorrent (tr_torrent * tor);
|
|
|
|
void tr_peerMgrStopTorrent (tr_torrent * tor);
|
|
|
|
void tr_peerMgrAddTorrent (tr_peerMgr * manager,
|
|
struct tr_torrent * tor);
|
|
|
|
void tr_peerMgrRemoveTorrent (tr_torrent * tor);
|
|
|
|
void tr_peerMgrTorrentAvailability (const tr_torrent * tor,
|
|
int8_t * tab,
|
|
unsigned int tabCount);
|
|
|
|
uint64_t tr_peerMgrGetDesiredAvailable (const tr_torrent * tor);
|
|
|
|
void tr_peerMgrOnTorrentGotMetainfo (tr_torrent * tor);
|
|
|
|
void tr_peerMgrOnBlocklistChanged (tr_peerMgr * manager);
|
|
|
|
void tr_peerMgrTorrentStats (tr_torrent * tor,
|
|
int * setmePeersConnected,
|
|
int * setmeWebseedsSendingToUs,
|
|
int * setmePeersSendingToUs,
|
|
int * setmePeersGettingFromUs,
|
|
int * setmePeersFrom); /* TR_PEER_FROM__MAX */
|
|
|
|
struct tr_peer_stat * tr_peerMgrPeerStats (const tr_torrent * tor,
|
|
int * setmeCount);
|
|
|
|
double * tr_peerMgrWebSpeeds_KBps (const tr_torrent * tor);
|
|
|
|
unsigned int tr_peerGetPieceSpeed_Bps (const tr_peer * peer,
|
|
uint64_t now,
|
|
tr_direction direction);
|
|
|
|
void tr_peerMgrClearInterest (tr_torrent * tor);
|
|
|
|
void tr_peerMgrGotBadPiece (tr_torrent * tor,
|
|
tr_piece_index_t pieceIndex);
|
|
|
|
void tr_peerMgrPieceCompleted (tr_torrent * tor,
|
|
tr_piece_index_t pieceIndex);
|
|
|
|
|
|
|
|
/* @} */
|
|
|
|
#endif
|