transmission/libtransmission/port-forwarding-natpmp.h

81 lines
1.7 KiB
C
Raw Normal View History

// This file Copyright © 2007-2023 Mnemosyne LLC.
2022-08-08 18:05:39 +00:00
// 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.
2006-09-25 18:37:45 +00:00
#pragma once
#ifndef LIBTRANSMISSION_PORT_FORWARDING_MODULE
#error only the libtransmission port forwarding module should #include this header.
#endif
2022-04-25 01:49:52 +00:00
#include <ctime> // time_t
2022-08-17 16:08:36 +00:00
#include <cstdint>
2022-04-25 01:49:52 +00:00
#include "natpmp.h"
2023-07-08 15:24:03 +00:00
#include "libtransmission/transmission.h" // tr_port_forwarding_state
#include "libtransmission/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 local_port = {};
tr_port advertised_port = {};
};
PulseResult pulse(tr_port local_port, bool is_enabled);
2006-09-25 18:37:45 +00:00
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 local_port_ = {};
tr_port advertised_port_ = {};
time_t renew_time_ = 0;
time_t command_time_ = 0;
State state_ = State::Discover;
bool has_discovered_ = false;
bool is_mapped_ = false;
};