mirror of
https://github.com/transmission/transmission
synced 2024-12-28 10:38:45 +00:00
ba26e79afe
Add a shim between the handshake code and the rest of the codebase to improve decoupling so that a followup PR can add handshake unit tests. The handshake code no longer directly relies on tr_torrent, tr_session, tr_dht, or tr_peerMgr.
78 lines
2 KiB
C++
78 lines
2 KiB
C++
// This file Copyright © 2007-2022 Mnemosyne LLC.
|
|
// It may be used under GPLv2 (SPDX: GPL-2.0), GPLv3 (SPDX: GPL-3.0),
|
|
// or any future license endorsed by Mnemosyne LLC.
|
|
// License text can be found in the licenses/ folder.
|
|
|
|
#pragma once
|
|
|
|
#ifndef __TRANSMISSION__
|
|
#error only libtransmission should #include this header.
|
|
#endif
|
|
|
|
#include <optional>
|
|
#include <memory>
|
|
|
|
#include "transmission.h"
|
|
|
|
#include "net.h" // tr_address
|
|
|
|
/** @addtogroup peers Peers
|
|
@{ */
|
|
|
|
class tr_peerIo;
|
|
|
|
/** @brief opaque struct holding hanshake state information.
|
|
freed when the handshake is completed. */
|
|
struct tr_handshake;
|
|
struct event_base;
|
|
|
|
struct tr_handshake_result
|
|
{
|
|
struct tr_handshake* handshake;
|
|
tr_peerIo* io;
|
|
bool readAnythingFromPeer;
|
|
bool isConnected;
|
|
void* userData;
|
|
std::optional<tr_peer_id_t> peer_id;
|
|
};
|
|
|
|
class tr_handshake_mediator
|
|
{
|
|
public:
|
|
struct torrent_info
|
|
{
|
|
tr_sha1_digest_t info_hash;
|
|
tr_peer_id_t client_peer_id;
|
|
tr_torrent_id_t id;
|
|
bool is_done;
|
|
};
|
|
|
|
[[nodiscard]] virtual std::optional<torrent_info> torrentInfo(tr_sha1_digest_t const& info_hash) const = 0;
|
|
|
|
[[nodiscard]] virtual std::optional<torrent_info> torrentInfoFromObfuscated(tr_sha1_digest_t const& info_hash) const = 0;
|
|
|
|
[[nodiscard]] virtual event_base* eventBase() const = 0;
|
|
|
|
[[nodiscard]] virtual bool isDHTEnabled() const = 0;
|
|
|
|
[[nodiscard]] virtual bool isPeerKnownSeed(tr_torrent_id_t tor_id, tr_address addr) const = 0;
|
|
|
|
virtual void setUTPFailed(tr_sha1_digest_t const& info_hash, tr_address) = 0;
|
|
};
|
|
|
|
/* returns true on success, false on error */
|
|
using tr_handshake_done_func = bool (*)(tr_handshake_result const& result);
|
|
|
|
/** @brief create a new handshake */
|
|
tr_handshake* tr_handshakeNew(
|
|
std::shared_ptr<tr_handshake_mediator> mediator,
|
|
tr_peerIo* io,
|
|
tr_encryption_mode encryption_mode,
|
|
tr_handshake_done_func when_done,
|
|
void* when_done_user_data);
|
|
|
|
void tr_handshakeAbort(tr_handshake* handshake);
|
|
|
|
tr_peerIo* tr_handshakeStealIO(tr_handshake* handshake);
|
|
|
|
/** @} */
|