mirror of
https://github.com/transmission/transmission
synced 2025-01-04 05:56:02 +00:00
bf156a97cf
* refactor: rename state enum type as tr_port_forwarding_state * chore: use snake_case for tr_shared struct fields * refactor: replace tr_shared with tr_port_forwarding * refactor: make tr_natpmp_state an enum class * refactor: uniform naming for port-forwarding module * refactor: move output-only parameters in tr_natpmp::pulse() to return struct * fix: use a nullptr multicastif if bindaddr is empty * chore: use PascalCase for enum class values * chore: clean up port-forwarding #includes * chore: remove unused tr_port_forwarding::peerPort()
79 lines
1.6 KiB
C++
79 lines
1.6 KiB
C++
// This file Copyright © 2007-2022 Mnemosyne LLC.
|
|
// It may be used under GPLv2 (SPDX: GPL-2.0-only), GPLv3 (SPDX: GPL-3.0-only),
|
|
// or any future license endorsed by Mnemosyne LLC.
|
|
// License text can be found in the licenses/ folder.
|
|
|
|
#pragma once
|
|
|
|
#ifndef LIBTRANSMISSION_PORT_FORWARDING_MODULE
|
|
#error only the libtransmission port forwarding module should #include this header.
|
|
#endif
|
|
|
|
#include <ctime> // time_t
|
|
#include <cstdint>
|
|
|
|
#include "transmission.h" // tr_port_forwarding_state
|
|
|
|
#include "natpmp.h"
|
|
#include "net.h" // tr_port
|
|
|
|
class tr_natpmp
|
|
{
|
|
public:
|
|
tr_natpmp()
|
|
{
|
|
natpmp_.s = static_cast<decltype(natpmp_.s)>(TR_BAD_SOCKET);
|
|
}
|
|
|
|
~tr_natpmp()
|
|
{
|
|
closenatpmp(&natpmp_);
|
|
}
|
|
|
|
[[nodiscard]] constexpr auto renewTime() const noexcept
|
|
{
|
|
return renew_time_;
|
|
}
|
|
|
|
struct PulseResult
|
|
{
|
|
tr_port_forwarding_state state = TR_PORT_ERROR;
|
|
|
|
tr_port public_port = {};
|
|
tr_port private_port = {};
|
|
};
|
|
|
|
PulseResult pulse(tr_port port, bool is_enabled);
|
|
|
|
private:
|
|
enum class State
|
|
{
|
|
Idle,
|
|
Err,
|
|
Discover,
|
|
RecvPub,
|
|
SendMap,
|
|
RecvMap,
|
|
SendUnmap,
|
|
RecvUnmap
|
|
};
|
|
|
|
static constexpr auto LifetimeSecs = uint32_t{ 3600 };
|
|
static constexpr auto CommandWaitSecs = time_t{ 8 };
|
|
|
|
[[nodiscard]] bool canSendCommand() const;
|
|
|
|
void setCommandTime();
|
|
|
|
natpmp_t natpmp_ = {};
|
|
|
|
tr_port public_port_ = {};
|
|
tr_port private_port_ = {};
|
|
|
|
time_t renew_time_ = 0;
|
|
time_t command_time_ = 0;
|
|
State state_ = State::Discover;
|
|
|
|
bool has_discovered_ = false;
|
|
bool is_mapped_ = false;
|
|
};
|