112 lines
4.7 KiB
C#
112 lines
4.7 KiB
C#
// This file is provided under The MIT License as part of RiptideNetworking.
|
|
// Copyright (c) Tom Weiland
|
|
// For additional information please see the included LICENSE.md file or view it on GitHub:
|
|
// https://github.com/RiptideNetworking/Riptide/blob/main/LICENSE.md
|
|
|
|
using System;
|
|
using System.Linq;
|
|
using System.Net;
|
|
using System.Net.Sockets;
|
|
|
|
namespace Riptide.Transports.Udp
|
|
{
|
|
/// <summary>A client which can connect to a <see cref="UdpServer"/>.</summary>
|
|
public class UdpClient : UdpPeer, IClient
|
|
{
|
|
/// <inheritdoc/>
|
|
public event EventHandler Connected;
|
|
/// <inheritdoc/>
|
|
public event EventHandler ConnectionFailed;
|
|
/// <inheritdoc/>
|
|
public event EventHandler<DataReceivedEventArgs> DataReceived;
|
|
|
|
/// <summary>The connection to the server.</summary>
|
|
private UdpConnection udpConnection;
|
|
|
|
/// <inheritdoc/>
|
|
public UdpClient(SocketMode mode = SocketMode.Both, int socketBufferSize = DefaultSocketBufferSize) : base(mode, socketBufferSize) { }
|
|
|
|
/// <inheritdoc/>
|
|
/// <remarks>Expects the host address to consist of an IP and port, separated by a colon. For example: <c>127.0.0.1:7777</c>.</remarks>
|
|
public bool Connect(string hostAddress, out Connection connection, out string connectError)
|
|
{
|
|
connectError = $"Invalid host address '{hostAddress}'! IP and port should be separated by a colon, for example: '127.0.0.1:7777'.";
|
|
if (!ParseHostAddress(hostAddress, out IPAddress ip, out ushort port))
|
|
{
|
|
connection = null;
|
|
return false;
|
|
}
|
|
|
|
if ((mode == SocketMode.IPv4Only && ip.AddressFamily == AddressFamily.InterNetworkV6) || (mode == SocketMode.IPv6Only && ip.AddressFamily == AddressFamily.InterNetwork))
|
|
{
|
|
// The IP address isn't in an acceptable format for the current socket mode
|
|
if (mode == SocketMode.IPv4Only)
|
|
connectError = "Connecting to IPv6 addresses is not allowed when running in IPv4 only mode!";
|
|
else
|
|
connectError = "Connecting to IPv4 addresses is not allowed when running in IPv6 only mode!";
|
|
|
|
connection = null;
|
|
return false;
|
|
}
|
|
|
|
OpenSocket();
|
|
|
|
connection = udpConnection = new UdpConnection(new IPEndPoint(mode == SocketMode.IPv4Only ? ip : ip.MapToIPv6(), port), this);
|
|
OnConnected(); // UDP is connectionless, so from the transport POV everything is immediately ready to send/receive data
|
|
return true;
|
|
}
|
|
|
|
/// <summary>Parses <paramref name="hostAddress"/> into <paramref name="ip"/> and <paramref name="port"/>, if possible.</summary>
|
|
/// <param name="hostAddress">The host address to parse.</param>
|
|
/// <param name="ip">The retrieved IP.</param>
|
|
/// <param name="port">The retrieved port.</param>
|
|
/// <returns>Whether or not <paramref name="hostAddress"/> was in a valid format.</returns>
|
|
private bool ParseHostAddress(string hostAddress, out IPAddress ip, out ushort port)
|
|
{
|
|
string[] ipAndPort = hostAddress.Split(':');
|
|
string ipString = "";
|
|
string portString = "";
|
|
if (ipAndPort.Length > 2)
|
|
{
|
|
// There was more than one ':' in the host address, might be IPv6
|
|
ipString = string.Join(":", ipAndPort.Take(ipAndPort.Length - 1));
|
|
portString = ipAndPort[ipAndPort.Length - 1];
|
|
}
|
|
else if (ipAndPort.Length == 2)
|
|
{
|
|
// IPv4
|
|
ipString = ipAndPort[0];
|
|
portString = ipAndPort[1];
|
|
}
|
|
|
|
port = 0; // Need to make sure a value is assigned in case IP parsing fails
|
|
return IPAddress.TryParse(ipString, out ip) && ushort.TryParse(portString, out port);
|
|
}
|
|
|
|
/// <inheritdoc/>
|
|
public void Disconnect()
|
|
{
|
|
CloseSocket();
|
|
}
|
|
|
|
/// <summary>Invokes the <see cref="Connected"/> event.</summary>
|
|
protected virtual void OnConnected()
|
|
{
|
|
Connected?.Invoke(this, EventArgs.Empty);
|
|
}
|
|
|
|
/// <summary>Invokes the <see cref="ConnectionFailed"/> event.</summary>
|
|
protected virtual void OnConnectionFailed()
|
|
{
|
|
ConnectionFailed?.Invoke(this, EventArgs.Empty);
|
|
}
|
|
|
|
/// <inheritdoc/>
|
|
protected override void OnDataReceived(byte[] dataBuffer, int amount, IPEndPoint fromEndPoint)
|
|
{
|
|
if (udpConnection.RemoteEndPoint.Equals(fromEndPoint) && !udpConnection.IsNotConnected)
|
|
DataReceived?.Invoke(this, new DataReceivedEventArgs(dataBuffer, amount, udpConnection));
|
|
}
|
|
}
|
|
}
|