mirror of
https://github.com/dolphin-emu/dolphin.git
synced 2026-01-30 19:13:09 +00:00
a14c88ba67
Yellow squiggly lines begone! Done automatically on .cpp files through `run-clang-tidy`, with manual corrections to the mistakes. If an import is directly used, but is technically unnecessary since it's recursively imported by something else, it is *not* removed. The tool doesn't touch .h files, so I did some of them by hand while fixing errors due to old recursive imports. Not everything is removed, but the cleanup should be substantial enough. Because this done on Linux, code that isn't used on it is mostly untouched. (Hopefully no open PR is depending on these imports...)
115 lines
3.0 KiB
C++
115 lines
3.0 KiB
C++
// SPDX-License-Identifier: CC0-1.0
|
|
|
|
#pragma once
|
|
|
|
#include <cstddef>
|
|
#include <list>
|
|
#include <memory>
|
|
#include <string>
|
|
#include <string_view>
|
|
|
|
#include <enet/enet.h>
|
|
|
|
#include "Common/CommonTypes.h"
|
|
#include "Common/ENet.h"
|
|
#include "Common/TraversalProto.h"
|
|
|
|
namespace Common
|
|
{
|
|
class TraversalClientClient
|
|
{
|
|
public:
|
|
virtual ~TraversalClientClient() = default;
|
|
virtual void OnTraversalStateChanged() = 0;
|
|
virtual void OnConnectReady(ENetAddress addr) = 0;
|
|
virtual void OnConnectFailed(TraversalConnectFailedReason reason) = 0;
|
|
virtual void OnTtlDetermined(u8 ttl) = 0;
|
|
};
|
|
|
|
class TraversalClient
|
|
{
|
|
public:
|
|
enum class State
|
|
{
|
|
Connecting,
|
|
Connected,
|
|
Failure,
|
|
};
|
|
enum class FailureReason
|
|
{
|
|
BadHost = 0x300,
|
|
VersionTooOld,
|
|
ServerForgotAboutUs,
|
|
SocketSendError,
|
|
ResendTimeout,
|
|
};
|
|
TraversalClient(ENetHost* netHost, const std::string& server, const u16 port,
|
|
const u16 port_alt = 0);
|
|
~TraversalClient();
|
|
|
|
TraversalHostId GetHostID() const;
|
|
TraversalInetAddress GetExternalAddress() const;
|
|
State GetState() const;
|
|
FailureReason GetFailureReason() const;
|
|
|
|
bool HasFailed() const { return m_State == State::Failure; }
|
|
bool IsConnecting() const { return m_State == State::Connecting; }
|
|
bool IsConnected() const { return m_State == State::Connected; }
|
|
|
|
void Reset();
|
|
void ConnectToClient(std::string_view host);
|
|
void ReconnectToServer();
|
|
void Update();
|
|
void HandleResends();
|
|
|
|
TraversalClientClient* m_Client = nullptr;
|
|
|
|
private:
|
|
struct OutgoingTraversalPacketInfo
|
|
{
|
|
TraversalPacket packet;
|
|
int tries;
|
|
u32 sendTime;
|
|
};
|
|
void HandleServerPacket(TraversalPacket* packet);
|
|
// called from NetHost
|
|
bool TestPacket(u8* data, size_t size, ENetAddress* from);
|
|
void ResendPacket(OutgoingTraversalPacketInfo* info);
|
|
TraversalRequestId SendTraversalPacket(const TraversalPacket& packet);
|
|
void OnFailure(FailureReason reason);
|
|
void HandlePing();
|
|
static int ENET_CALLBACK InterceptCallback(ENetHost* host, ENetEvent* event);
|
|
|
|
void NewTraversalTest();
|
|
void HandleTraversalTest();
|
|
|
|
ENetHost* m_NetHost;
|
|
TraversalHostId m_HostId{};
|
|
TraversalInetAddress m_external_address{};
|
|
State m_State{};
|
|
FailureReason m_FailureReason{};
|
|
TraversalRequestId m_ConnectRequestId = 0;
|
|
bool m_PendingConnect = false;
|
|
std::list<OutgoingTraversalPacketInfo> m_OutgoingTraversalPackets;
|
|
ENetAddress m_ServerAddress{};
|
|
std::string m_Server;
|
|
u16 m_port;
|
|
u16 m_portAlt;
|
|
u32 m_PingTime = 0;
|
|
|
|
ENetSocket m_TestSocket = ENET_SOCKET_NULL;
|
|
TraversalRequestId m_TestRequestId = 0;
|
|
u8 m_ttl = 2;
|
|
bool m_ttlReady = false;
|
|
};
|
|
|
|
extern std::unique_ptr<TraversalClient> g_TraversalClient;
|
|
// the NetHost connected to the TraversalClient.
|
|
extern ENet::ENetHostPtr g_MainNetHost;
|
|
|
|
// Create g_TraversalClient and g_MainNetHost if necessary.
|
|
bool EnsureTraversalClient(const std::string& server, u16 server_port, u16 server_port_alt = 0,
|
|
u16 listen_port = 0);
|
|
void ReleaseTraversalClient();
|
|
} // namespace Common
|