Skip to content

Commit

Permalink
implement rudp send and send raw
Browse files Browse the repository at this point in the history
  • Loading branch information
@alec1o
alec1o committed Jul 2, 2024
1 parent 35b0f61 commit a5a6fdc
Show file tree
Hide file tree
Showing 2 changed files with 132 additions and 14 deletions.
142 changes: 130 additions & 12 deletions src/rudp/partials/RUDP.ClientTo.cs
View file
Original file line number Diff line number Diff line change
Expand Up @@ -23,6 +23,8 @@ private class ClientTo : IRUDP.ClientTo
private bool _isOpeningOrClosing, _isClosed;
private readonly bool _isServer;
private int _openTimeout;
private MessageId _messageId;
private readonly object _nextIdLock = new object();

private struct Config
{
Expand All @@ -40,6 +42,14 @@ private class Package
public sbyte Type { get; set; }
}

private class MessageId
{
public uint Reliable { get; set; }
public uint ReliableUnordered { get; set; }
public uint Unreliable { get; }
public uint UnreliableOrdered { get; set; }
}


private ClientTo()
{
Expand Down Expand Up @@ -136,8 +146,10 @@ ref endpoint
// connected successful
Host = new Host(_socket.RemoteEndPoint);

_messageId = new MessageId();

_isClosed = false;

On.OnOpen?.Invoke(null, null);
}
catch (Exception e)
Expand Down Expand Up @@ -168,7 +180,7 @@ public Task Close()

for (var i = 0; i < Config.CloseRepeat; i++)
{
SendRaw(_isServer ? Host : null, ref data);
SendRaw(Host, ref data);
}
}
catch (Exception e)
Expand Down Expand Up @@ -202,42 +214,104 @@ public Task Close()

public void Data(byte[] data, MessageType messageType)
{
Send(null, data, messageType);
if (data == null || data.Length <= 0) return;

Send(data, messageType);
}

public void Data(string data, MessageType messageType)
{
Send(null, data.GetBytes(), messageType);
if (data == null || data.Length <= 0) return;

Send(data.GetBytes(), messageType);
}

public void Data(string data, MessageType messageType, Encoding encoding)
{
Send(null, data.GetBytes(encoding), messageType);
if (data == null || data.Length <= 0) return;

Send(data.GetBytes(encoding), messageType);
}

public void Event(string name, byte[] data, MessageType messageType)
{
Send(null, NetlyEnvironment.EventManager.Create(name, data), messageType);
if (data == null || data.Length <= 0) return;

Send(NetlyEnvironment.EventManager.Create(name, data), messageType);
}

public void Event(string name, string data, MessageType messageType)
{
Send(null, NetlyEnvironment.EventManager.Create(name, data.GetBytes()), messageType);
if (data == null || data.Length <= 0 || name == null || name.Length <= 0) return;

Send(NetlyEnvironment.EventManager.Create(name, data.GetBytes()), messageType);
}

public void Event(string name, string data, MessageType messageType, Encoding encoding)
{
Send(null, NetlyEnvironment.EventManager.Create(name, data.GetBytes(encoding)), messageType);
if (data == null || data.Length <= 0 || name == null || name.Length <= 0) return;

Send(NetlyEnvironment.EventManager.Create(name, data.GetBytes(encoding)), messageType);
}

private void Send(Host host, byte[] data, MessageType messageType)

private void Send(byte[] bytes, MessageType messageType)
{
throw new NotImplementedException();
Send(Host, ref bytes, messageType);
}

private void SendRaw(Host host, ref byte[] data)
private void Send(Host host, ref byte[] bytes, MessageType messageType)
{
throw new NotImplementedException();
var package = new Package
{
Type = (sbyte)messageType,
Data = bytes,
Id = GetNextId(messageType)
};

GetBufferFromPackage(ref package, out var buffer);

SendRaw(host, ref buffer);
}

private void SendRaw(Host host, ref byte[] bytes)
{
if (!IsOpened) return;

try
{
if (_isServer)
{
// this way of send just work on windows and linux, except macOS (maybe iOs)
_socket?.BeginSendTo
(
bytes,
0,
bytes.Length,
SocketFlags.None,
host.EndPoint,
null,
null
);
}
else
{
// this way of send just work on windows and linux, include macOS and iOs
_socket?.BeginSend
(
bytes,
0,
bytes.Length,
SocketFlags.None,
null,
null
);
}
}
catch (Exception e)
{
NetlyEnvironment.Logger.Create(e);
}
}

private bool IsConnected()
Expand Down Expand Up @@ -267,6 +341,50 @@ public int GetOpenTimeout()
{
return _openTimeout;
}

private uint GetNextId(MessageType messageType)
{
lock (_nextIdLock)
{
switch (messageType)
{
case MessageType.Reliable:
{
_messageId.Reliable++;
return _messageId.Reliable;
}
case MessageType.ReliableUnordered:
{
_messageId.ReliableUnordered++;
return _messageId.ReliableUnordered;
}
case MessageType.Unreliable:
{
return _messageId.Unreliable;
}
case MessageType.UnreliableOrdered:
{
_messageId.UnreliableOrdered++;
return _messageId.UnreliableOrdered;
}
default:
{
throw new InvalidOperationException(messageType.ToString());
}
}
}
}

private void GetBufferFromPackage(ref Package package, out byte[] buffer)
{
var primitive = new Primitive();

primitive.Add.Class(package);

buffer = primitive.GetBytes();

primitive.Reset();
}
}
}
}
Expand Down
4 changes: 2 additions & 2 deletions src/rudp/partials/RUDP.MessageType.cs
View file
Original file line number Diff line number Diff line change
Expand Up @@ -3,14 +3,14 @@ namespace Netly
public static partial class RUDP
{
public const MessageType Reliable = MessageType.Reliable;
public const MessageType ReliableOrdered = MessageType.ReliableOrdered;
public const MessageType ReliableOrdered = MessageType.ReliableUnordered;
public const MessageType Unreliable = MessageType.Unreliable;
public const MessageType UnreliableOrdered = MessageType.UnreliableOrdered;

public enum MessageType : sbyte
{
Reliable = 1,
ReliableOrdered = -1,
ReliableUnordered = -1,
Unreliable = 2,
UnreliableOrdered = -2,
}
Expand Down

0 comments on commit a5a6fdc

Please sign in to comment.