Delete RelayHandler.cs
This commit is contained in:
cxxpxr
GitHub
parent
7491c3339a
commit
6bf574a791
1 changed files with 0 additions and 340 deletions
340
RelayHandler.cs
340
RelayHandler.cs
|
|
@ -1,340 +0,0 @@
|
||||||
using System;
|
|
||||||
using System.Buffers;
|
|
||||||
using System.Collections.Generic;
|
|
||||||
using System.Net;
|
|
||||||
using System.Text;
|
|
||||||
|
|
||||||
namespace LightReflectiveMirror
|
|
||||||
{
|
|
||||||
public class RelayHandler
|
|
||||||
{
|
|
||||||
public List<Room> rooms = new List<Room>();
|
|
||||||
private List<int> _pendingAuthentication = new List<int>();
|
|
||||||
private ArrayPool<byte> _sendBuffers;
|
|
||||||
private int _maxPacketSize = 0;
|
|
||||||
|
|
||||||
public RelayHandler(int maxPacketSize)
|
|
||||||
{
|
|
||||||
this._maxPacketSize = maxPacketSize;
|
|
||||||
_sendBuffers = ArrayPool<byte>.Create(maxPacketSize, 50);
|
|
||||||
}
|
|
||||||
|
|
||||||
public void ClientConnected(int clientId)
|
|
||||||
{
|
|
||||||
_pendingAuthentication.Add(clientId);
|
|
||||||
var buffer = _sendBuffers.Rent(1);
|
|
||||||
int pos = 0;
|
|
||||||
buffer.WriteByte(ref pos, (byte)OpCodes.AuthenticationRequest);
|
|
||||||
Program.transport.ServerSend(clientId, 0, new ArraySegment<byte>(buffer, 0, pos));
|
|
||||||
_sendBuffers.Return(buffer);
|
|
||||||
}
|
|
||||||
|
|
||||||
public void HandleMessage(int clientId, ArraySegment<byte> segmentData, int channel)
|
|
||||||
{
|
|
||||||
try
|
|
||||||
{
|
|
||||||
var data = segmentData.Array;
|
|
||||||
int pos = segmentData.Offset;
|
|
||||||
|
|
||||||
OpCodes opcode = (OpCodes)data.ReadByte(ref pos);
|
|
||||||
|
|
||||||
if (_pendingAuthentication.Contains(clientId))
|
|
||||||
{
|
|
||||||
if (opcode == OpCodes.AuthenticationResponse)
|
|
||||||
{
|
|
||||||
string authResponse = data.ReadString(ref pos);
|
|
||||||
if (authResponse == Program.conf.AuthenticationKey)
|
|
||||||
{
|
|
||||||
_pendingAuthentication.Remove(clientId);
|
|
||||||
int writePos = 0;
|
|
||||||
var sendBuffer = _sendBuffers.Rent(1);
|
|
||||||
sendBuffer.WriteByte(ref writePos, (byte)OpCodes.Authenticated);
|
|
||||||
Program.transport.ServerSend(clientId, 0, new ArraySegment<byte>(sendBuffer, 0, writePos));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
switch (opcode)
|
|
||||||
{
|
|
||||||
case OpCodes.CreateRoom:
|
|
||||||
CreateRoom(clientId, data.ReadInt(ref pos), data.ReadString(ref pos), data.ReadBool(ref pos), data.ReadString(ref pos), data.ReadBool(ref pos), data.ReadString(ref pos), data.ReadBool(ref pos), data.ReadInt(ref pos));
|
|
||||||
break;
|
|
||||||
case OpCodes.RequestID:
|
|
||||||
SendClientID(clientId);
|
|
||||||
break;
|
|
||||||
case OpCodes.LeaveRoom:
|
|
||||||
LeaveRoom(clientId);
|
|
||||||
break;
|
|
||||||
case OpCodes.JoinServer:
|
|
||||||
JoinRoom(clientId, data.ReadInt(ref pos), data.ReadBool(ref pos), data.ReadString(ref pos));
|
|
||||||
break;
|
|
||||||
case OpCodes.KickPlayer:
|
|
||||||
LeaveRoom(data.ReadInt(ref pos), clientId);
|
|
||||||
break;
|
|
||||||
case OpCodes.SendData:
|
|
||||||
ProcessData(clientId, data.ReadBytes(ref pos), channel, data.ReadInt(ref pos));
|
|
||||||
break;
|
|
||||||
case OpCodes.UpdateRoomData:
|
|
||||||
var plyRoom = GetRoomForPlayer(clientId);
|
|
||||||
|
|
||||||
if (plyRoom == null)
|
|
||||||
return;
|
|
||||||
|
|
||||||
bool newName = data.ReadBool(ref pos);
|
|
||||||
if (newName)
|
|
||||||
plyRoom.serverName = data.ReadString(ref pos);
|
|
||||||
|
|
||||||
bool newData = data.ReadBool(ref pos);
|
|
||||||
if (newData)
|
|
||||||
plyRoom.serverData = data.ReadString(ref pos);
|
|
||||||
|
|
||||||
bool newPublicStatus = data.ReadBool(ref pos);
|
|
||||||
if (newPublicStatus)
|
|
||||||
plyRoom.isPublic = data.ReadBool(ref pos);
|
|
||||||
|
|
||||||
bool newPlayerCap = data.ReadBool(ref pos);
|
|
||||||
if (newPlayerCap)
|
|
||||||
plyRoom.maxPlayers = data.ReadInt(ref pos);
|
|
||||||
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
catch
|
|
||||||
{
|
|
||||||
// Do Nothing. Client probably sent some invalid data.
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
public void HandleDisconnect(int clientId) => LeaveRoom(clientId);
|
|
||||||
|
|
||||||
void ProcessData(int clientId, byte[] clientData, int channel, int sendTo = -1)
|
|
||||||
{
|
|
||||||
Room playersRoom = GetRoomForPlayer(clientId);
|
|
||||||
|
|
||||||
if(playersRoom != null)
|
|
||||||
{
|
|
||||||
Room room = playersRoom;
|
|
||||||
|
|
||||||
if(room.hostId == clientId)
|
|
||||||
{
|
|
||||||
if (room.clients.Contains(sendTo))
|
|
||||||
{
|
|
||||||
int pos = 0;
|
|
||||||
byte[] sendBuffer = _sendBuffers.Rent(_maxPacketSize);
|
|
||||||
|
|
||||||
sendBuffer.WriteByte(ref pos, (byte)OpCodes.GetData);
|
|
||||||
sendBuffer.WriteBytes(ref pos, clientData);
|
|
||||||
|
|
||||||
Program.transport.ServerSend(sendTo, channel, new ArraySegment<byte>(sendBuffer, 0, pos));
|
|
||||||
_sendBuffers.Return(sendBuffer);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
// We are not the host, so send the data to the host.
|
|
||||||
int pos = 0;
|
|
||||||
byte[] sendBuffer = _sendBuffers.Rent(_maxPacketSize);
|
|
||||||
|
|
||||||
sendBuffer.WriteByte(ref pos, (byte)OpCodes.GetData);
|
|
||||||
sendBuffer.WriteBytes(ref pos, clientData);
|
|
||||||
sendBuffer.WriteInt(ref pos, clientId);
|
|
||||||
|
|
||||||
Program.transport.ServerSend(room.hostId, channel, new ArraySegment<byte>(sendBuffer, 0, pos));
|
|
||||||
_sendBuffers.Return(sendBuffer);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
Room GetRoomForPlayer(int clientId)
|
|
||||||
{
|
|
||||||
for(int i = 0; i < rooms.Count; i++)
|
|
||||||
{
|
|
||||||
if (rooms[i].hostId == clientId)
|
|
||||||
return rooms[i];
|
|
||||||
|
|
||||||
if (rooms[i].clients.Contains(clientId))
|
|
||||||
return rooms[i];
|
|
||||||
}
|
|
||||||
|
|
||||||
return null;
|
|
||||||
}
|
|
||||||
|
|
||||||
void JoinRoom(int clientId, int serverId, bool canDirectConnect, string localIP)
|
|
||||||
{
|
|
||||||
LeaveRoom(clientId);
|
|
||||||
|
|
||||||
for(int i = 0; i < rooms.Count; i++)
|
|
||||||
{
|
|
||||||
if(rooms[i].serverId == serverId)
|
|
||||||
{
|
|
||||||
if(rooms[i].clients.Count < rooms[i].maxPlayers)
|
|
||||||
{
|
|
||||||
rooms[i].clients.Add(clientId);
|
|
||||||
|
|
||||||
int sendJoinPos = 0;
|
|
||||||
byte[] sendJoinBuffer = _sendBuffers.Rent(500);
|
|
||||||
|
|
||||||
if (canDirectConnect && Program.instance.NATConnections.ContainsKey(clientId))
|
|
||||||
{
|
|
||||||
sendJoinBuffer.WriteByte(ref sendJoinPos, (byte)OpCodes.DirectConnectIP);
|
|
||||||
|
|
||||||
if (Program.instance.NATConnections[clientId].Address.Equals(rooms[i].hostIP.Address))
|
|
||||||
sendJoinBuffer.WriteString(ref sendJoinPos, rooms[i].hostLocalIP == localIP ? "127.0.0.1" : rooms[i].hostLocalIP);
|
|
||||||
else
|
|
||||||
sendJoinBuffer.WriteString(ref sendJoinPos, rooms[i].hostIP.Address.ToString());
|
|
||||||
|
|
||||||
sendJoinBuffer.WriteInt(ref sendJoinPos, rooms[i].useNATPunch ? rooms[i].hostIP.Port : rooms[i].port);
|
|
||||||
|
|
||||||
Program.transport.ServerSend(clientId, 0, new ArraySegment<byte>(sendJoinBuffer, 0, sendJoinPos));
|
|
||||||
|
|
||||||
sendJoinPos = 0;
|
|
||||||
sendJoinBuffer.WriteByte(ref sendJoinPos, (byte)OpCodes.DirectConnectIP);
|
|
||||||
Console.WriteLine(Program.instance.NATConnections[clientId].Address.ToString());
|
|
||||||
sendJoinBuffer.WriteString(ref sendJoinPos, Program.instance.NATConnections[clientId].Address.ToString());
|
|
||||||
sendJoinBuffer.WriteInt(ref sendJoinPos, Program.instance.NATConnections[clientId].Port);
|
|
||||||
|
|
||||||
Program.transport.ServerSend(rooms[i].hostId, 0, new ArraySegment<byte>(sendJoinBuffer, 0, sendJoinPos));
|
|
||||||
|
|
||||||
_sendBuffers.Return(sendJoinBuffer);
|
|
||||||
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
|
|
||||||
sendJoinBuffer.WriteByte(ref sendJoinPos, (byte)OpCodes.ServerJoined);
|
|
||||||
sendJoinBuffer.WriteInt(ref sendJoinPos, clientId);
|
|
||||||
|
|
||||||
Program.transport.ServerSend(clientId, 0, new ArraySegment<byte>(sendJoinBuffer, 0, sendJoinPos));
|
|
||||||
Program.transport.ServerSend(rooms[i].hostId, 0, new ArraySegment<byte>(sendJoinBuffer, 0, sendJoinPos));
|
|
||||||
_sendBuffers.Return(sendJoinBuffer);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// If it got to here, then the server was not found, or full. Tell the client.
|
|
||||||
int pos = 0;
|
|
||||||
byte[] sendBuffer = _sendBuffers.Rent(1);
|
|
||||||
|
|
||||||
sendBuffer.WriteByte(ref pos, (byte)OpCodes.ServerLeft);
|
|
||||||
|
|
||||||
Program.transport.ServerSend(clientId, 0, new ArraySegment<byte>(sendBuffer, 0, pos));
|
|
||||||
_sendBuffers.Return(sendBuffer);
|
|
||||||
}
|
|
||||||
|
|
||||||
void CreateRoom(int clientId, int maxPlayers, string serverName, bool isPublic, string serverData, bool useDirectConnect, string hostLocalIP, bool useNatPunch, int port)
|
|
||||||
{
|
|
||||||
LeaveRoom(clientId);
|
|
||||||
|
|
||||||
IPEndPoint hostIP = null;
|
|
||||||
Program.instance.NATConnections.TryGetValue(clientId, out hostIP);
|
|
||||||
|
|
||||||
Room room = new Room
|
|
||||||
{
|
|
||||||
hostId = clientId,
|
|
||||||
maxPlayers = maxPlayers,
|
|
||||||
serverName = serverName,
|
|
||||||
isPublic = isPublic,
|
|
||||||
serverData = serverData,
|
|
||||||
clients = new List<int>(),
|
|
||||||
serverId = GetRandomServerID(),
|
|
||||||
hostIP = hostIP,
|
|
||||||
hostLocalIP = hostLocalIP,
|
|
||||||
supportsDirectConnect = hostIP == null ? false : useDirectConnect,
|
|
||||||
port = port,
|
|
||||||
useNATPunch = useNatPunch
|
|
||||||
};
|
|
||||||
|
|
||||||
rooms.Add(room);
|
|
||||||
|
|
||||||
int pos = 0;
|
|
||||||
byte[] sendBuffer = _sendBuffers.Rent(5);
|
|
||||||
|
|
||||||
sendBuffer.WriteByte(ref pos, (byte)OpCodes.RoomCreated);
|
|
||||||
sendBuffer.WriteInt(ref pos, clientId);
|
|
||||||
|
|
||||||
Program.transport.ServerSend(clientId, 0, new ArraySegment<byte>(sendBuffer, 0, pos));
|
|
||||||
_sendBuffers.Return(sendBuffer);
|
|
||||||
}
|
|
||||||
|
|
||||||
void LeaveRoom(int clientId, int requiredHostId = -1)
|
|
||||||
{
|
|
||||||
for(int i = 0; i < rooms.Count; i++)
|
|
||||||
{
|
|
||||||
if(rooms[i].hostId == clientId)
|
|
||||||
{
|
|
||||||
int pos = 0;
|
|
||||||
byte[] sendBuffer = _sendBuffers.Rent(1);
|
|
||||||
sendBuffer.WriteByte(ref pos, (byte)OpCodes.ServerLeft);
|
|
||||||
|
|
||||||
for(int x = 0; x < rooms[i].clients.Count; x++)
|
|
||||||
Program.transport.ServerSend(rooms[i].clients[x], 0, new ArraySegment<byte>(sendBuffer, 0, pos));
|
|
||||||
|
|
||||||
_sendBuffers.Return(sendBuffer);
|
|
||||||
rooms[i].clients.Clear();
|
|
||||||
rooms.RemoveAt(i);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
if (requiredHostId >= 0 && rooms[i].hostId != requiredHostId)
|
|
||||||
continue;
|
|
||||||
|
|
||||||
if(rooms[i].clients.RemoveAll(x => x == clientId) > 0)
|
|
||||||
{
|
|
||||||
int pos = 0;
|
|
||||||
byte[] sendBuffer = _sendBuffers.Rent(5);
|
|
||||||
|
|
||||||
sendBuffer.WriteByte(ref pos, (byte)OpCodes.PlayerDisconnected);
|
|
||||||
sendBuffer.WriteInt(ref pos, clientId);
|
|
||||||
|
|
||||||
Program.transport.ServerSend(rooms[i].hostId, 0, new ArraySegment<byte>(sendBuffer, 0, pos));
|
|
||||||
_sendBuffers.Return(sendBuffer);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
void SendClientID(int clientId)
|
|
||||||
{
|
|
||||||
int pos = 0;
|
|
||||||
byte[] sendBuffer = _sendBuffers.Rent(5);
|
|
||||||
|
|
||||||
sendBuffer.WriteByte(ref pos, (byte)OpCodes.GetID);
|
|
||||||
sendBuffer.WriteInt(ref pos, clientId);
|
|
||||||
|
|
||||||
Program.transport.ServerSend(clientId, 0, new ArraySegment<byte>(sendBuffer, 0, pos));
|
|
||||||
_sendBuffers.Return(sendBuffer);
|
|
||||||
}
|
|
||||||
|
|
||||||
int GetRandomServerID()
|
|
||||||
{
|
|
||||||
Random rand = new Random();
|
|
||||||
int temp = rand.Next(int.MinValue, int.MaxValue);
|
|
||||||
|
|
||||||
while (DoesServerIdExist(temp))
|
|
||||||
temp = rand.Next(int.MinValue, int.MaxValue);
|
|
||||||
|
|
||||||
return temp;
|
|
||||||
}
|
|
||||||
|
|
||||||
bool DoesServerIdExist(int id)
|
|
||||||
{
|
|
||||||
for (int i = 0; i < rooms.Count; i++)
|
|
||||||
if (rooms[i].serverId == id)
|
|
||||||
return true;
|
|
||||||
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
public enum OpCodes
|
|
||||||
{
|
|
||||||
Default = 0, RequestID = 1, JoinServer = 2, SendData = 3, GetID = 4, ServerJoined = 5, GetData = 6, CreateRoom = 7, ServerLeft = 8, PlayerDisconnected = 9, RoomCreated = 10,
|
|
||||||
LeaveRoom = 11, KickPlayer = 12, AuthenticationRequest = 13, AuthenticationResponse = 14, Authenticated = 17, UpdateRoomData = 18, ServerConnectionData = 19, RequestNATConnection = 20,
|
|
||||||
DirectConnectIP = 21
|
|
||||||
}
|
|
||||||
}
|
|
||||||
Loading…
Reference in a new issue