exception
This commit is contained in:
@@ -27,9 +27,16 @@ namespace Core
|
||||
{
|
||||
while (true)
|
||||
{
|
||||
Socket clientSocket = await serverSocket.AcceptAsync();
|
||||
Console.WriteLine($"[{DateTime.Now}] Accept client: {clientSocket.RemoteEndPoint}");
|
||||
ThreadPool.QueueUserWorkItem(RunAsync, clientSocket);
|
||||
try
|
||||
{
|
||||
Socket clientSocket = await serverSocket.AcceptAsync();
|
||||
Console.WriteLine($"[{DateTime.Now}] Accept client: {clientSocket.RemoteEndPoint}");
|
||||
ThreadPool.QueueUserWorkItem(RunAsync, clientSocket);
|
||||
}
|
||||
catch (Exception ex)
|
||||
{
|
||||
Console.WriteLine(ex);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@@ -45,143 +52,175 @@ namespace Core
|
||||
string nickname = "";
|
||||
string roomName = "";
|
||||
|
||||
while (true)
|
||||
try
|
||||
{
|
||||
// header buffer
|
||||
int n1 = await clientSocket.ReceiveAsync(headerBuffer, SocketFlags.None);
|
||||
if (n1 < 1)
|
||||
while (true)
|
||||
{
|
||||
Console.WriteLine($"[{DateTime.Now}] Disconnect client - {clientSocket.RemoteEndPoint}");
|
||||
clientSocket.Dispose();
|
||||
return;
|
||||
}
|
||||
else if (n1 == 1)
|
||||
{
|
||||
await clientSocket.ReceiveAsync(new ArraySegment<byte>(headerBuffer, 1, 1), SocketFlags.None);
|
||||
}
|
||||
|
||||
// data buffer
|
||||
short dataSize = IPAddress.NetworkToHostOrder(BitConverter.ToInt16(headerBuffer));
|
||||
byte[] dataBuffer = new byte[dataSize];
|
||||
|
||||
int totalRecv = 0;
|
||||
while (totalRecv < dataSize)
|
||||
{
|
||||
int n2 = await clientSocket.ReceiveAsync(new ArraySegment<byte>(dataBuffer, totalRecv, dataSize - totalRecv), SocketFlags.None);
|
||||
totalRecv += n2;
|
||||
}
|
||||
|
||||
PacketType packetType = (PacketType)IPAddress.NetworkToHostOrder(BitConverter.ToInt16(dataBuffer));
|
||||
if (packetType == PacketType.LoginRequest)
|
||||
{
|
||||
LoginRequestPacket requestPacket = new LoginRequestPacket(dataBuffer);
|
||||
|
||||
bool result = Clients.TryAdd(requestPacket.Id, clientSocket);
|
||||
Console.WriteLine($"[{DateTime.Now}] LoginRequest - Id: {requestPacket.Id}, Nickname: {requestPacket.NickName}");
|
||||
|
||||
id = requestPacket.Id;
|
||||
nickname = requestPacket.NickName;
|
||||
|
||||
LoginResponsePacket responsePacket = new LoginResponsePacket(result ? StatusCode.Success : StatusCode.Failed);
|
||||
await clientSocket.SendAsync(responsePacket.Serialize(), SocketFlags.None);
|
||||
}
|
||||
else if (packetType == PacketType.CreateRoomRequest)
|
||||
{
|
||||
CreateRoomRequestPacket requestPacket = new CreateRoomRequestPacket(dataBuffer);
|
||||
Room room = new Room();
|
||||
if (Rooms.TryAdd(requestPacket.RoomName, room))
|
||||
// header buffer
|
||||
int n1 = await clientSocket.ReceiveAsync(headerBuffer, SocketFlags.None);
|
||||
if (n1 < 1)
|
||||
{
|
||||
roomName = requestPacket.RoomName;
|
||||
room.Users.TryAdd(id, nickname);
|
||||
Console.WriteLine($"[{DateTime.Now}] CreateRoomRequest - RoomName: {roomName}, id: {id}, nickname: {nickname}");
|
||||
Console.WriteLine($"[{DateTime.Now}] Disconnect client - {clientSocket.RemoteEndPoint}");
|
||||
await Remove(id, nickname, roomName, clientSocket);
|
||||
return;
|
||||
}
|
||||
else if (n1 == 1)
|
||||
{
|
||||
await clientSocket.ReceiveAsync(new ArraySegment<byte>(headerBuffer, 1, 1), SocketFlags.None);
|
||||
}
|
||||
|
||||
CreateRoomResponsePacket responsePacket = new CreateRoomResponsePacket(StatusCode.Success);
|
||||
// data buffer
|
||||
short dataSize = IPAddress.NetworkToHostOrder(BitConverter.ToInt16(headerBuffer));
|
||||
byte[] dataBuffer = new byte[dataSize];
|
||||
|
||||
int totalRecv = 0;
|
||||
while (totalRecv < dataSize)
|
||||
{
|
||||
int n2 = await clientSocket.ReceiveAsync(new ArraySegment<byte>(dataBuffer, totalRecv, dataSize - totalRecv), SocketFlags.None);
|
||||
totalRecv += n2;
|
||||
}
|
||||
|
||||
PacketType packetType = (PacketType)IPAddress.NetworkToHostOrder(BitConverter.ToInt16(dataBuffer));
|
||||
if (packetType == PacketType.LoginRequest)
|
||||
{
|
||||
LoginRequestPacket requestPacket = new LoginRequestPacket(dataBuffer);
|
||||
|
||||
bool result = Clients.TryAdd(requestPacket.Id, clientSocket);
|
||||
Console.WriteLine($"[{DateTime.Now}] LoginRequest - Id: {requestPacket.Id}, Nickname: {requestPacket.NickName}");
|
||||
|
||||
id = requestPacket.Id;
|
||||
nickname = requestPacket.NickName;
|
||||
|
||||
LoginResponsePacket responsePacket = new LoginResponsePacket(result ? StatusCode.Success : StatusCode.Failed);
|
||||
await clientSocket.SendAsync(responsePacket.Serialize(), SocketFlags.None);
|
||||
}
|
||||
else
|
||||
else if (packetType == PacketType.CreateRoomRequest)
|
||||
{
|
||||
Console.WriteLine($"[{DateTime.Now}] CreateRoomRequest failed - RoomName: {requestPacket.RoomName}");
|
||||
|
||||
CreateRoomResponsePacket responsePacket = new CreateRoomResponsePacket(StatusCode.Failed);
|
||||
await clientSocket.SendAsync(responsePacket.Serialize(), SocketFlags.None);
|
||||
}
|
||||
}
|
||||
else if (packetType == PacketType.RoomListRequest)
|
||||
{
|
||||
RoomListResponsePacket packet = new RoomListResponsePacket(Rooms.Keys);
|
||||
await clientSocket.SendAsync(packet.Serialize(), SocketFlags.None);
|
||||
}
|
||||
else if (packetType == PacketType.EnterRoomRequest)
|
||||
{
|
||||
EnterRoomRequestPacket requestPacket = new EnterRoomRequestPacket(dataBuffer);
|
||||
if (Rooms.TryGetValue(requestPacket.RoomName, out var room))
|
||||
{
|
||||
roomName = requestPacket.RoomName;
|
||||
room.Users.TryAdd(id, nickname);
|
||||
Console.WriteLine($"[{DateTime.Now}] EnterRoomRequest - RoomName: {roomName}, id: {id}, nickname: {nickname}");
|
||||
|
||||
EnterRoomResponsePacket responsePacket = new EnterRoomResponsePacket(StatusCode.Success);
|
||||
await clientSocket.SendAsync(responsePacket.Serialize(), SocketFlags.None);
|
||||
|
||||
await Task.Delay(100);
|
||||
foreach (var user in room.Users)
|
||||
CreateRoomRequestPacket requestPacket = new CreateRoomRequestPacket(dataBuffer);
|
||||
Room room = new Room();
|
||||
if (Rooms.TryAdd(requestPacket.RoomName, room))
|
||||
{
|
||||
if (user.Value == nickname)
|
||||
continue;
|
||||
roomName = requestPacket.RoomName;
|
||||
room.Users.TryAdd(id, nickname);
|
||||
Console.WriteLine($"[{DateTime.Now}] CreateRoomRequest - RoomName: {roomName}, id: {id}, nickname: {nickname}");
|
||||
|
||||
// add me to other user
|
||||
if (Clients.TryGetValue(user.Key, out var otherClient))
|
||||
CreateRoomResponsePacket responsePacket = new CreateRoomResponsePacket(StatusCode.Success);
|
||||
await clientSocket.SendAsync(responsePacket.Serialize(), SocketFlags.None);
|
||||
}
|
||||
else
|
||||
{
|
||||
Console.WriteLine($"[{DateTime.Now}] CreateRoomRequest failed - RoomName: {requestPacket.RoomName}");
|
||||
|
||||
CreateRoomResponsePacket responsePacket = new CreateRoomResponsePacket(StatusCode.Failed);
|
||||
await clientSocket.SendAsync(responsePacket.Serialize(), SocketFlags.None);
|
||||
}
|
||||
}
|
||||
else if (packetType == PacketType.RoomListRequest)
|
||||
{
|
||||
RoomListResponsePacket packet = new RoomListResponsePacket(Rooms.Keys);
|
||||
await clientSocket.SendAsync(packet.Serialize(), SocketFlags.None);
|
||||
}
|
||||
else if (packetType == PacketType.EnterRoomRequest)
|
||||
{
|
||||
EnterRoomRequestPacket requestPacket = new EnterRoomRequestPacket(dataBuffer);
|
||||
if (Rooms.TryGetValue(requestPacket.RoomName, out var room))
|
||||
{
|
||||
roomName = requestPacket.RoomName;
|
||||
room.Users.TryAdd(id, nickname);
|
||||
Console.WriteLine($"[{DateTime.Now}] EnterRoomRequest - RoomName: {roomName}, id: {id}, nickname: {nickname}");
|
||||
|
||||
EnterRoomResponsePacket responsePacket = new EnterRoomResponsePacket(StatusCode.Success);
|
||||
await clientSocket.SendAsync(responsePacket.Serialize(), SocketFlags.None);
|
||||
|
||||
await Task.Delay(100);
|
||||
foreach (var user in room.Users)
|
||||
{
|
||||
UserEnterPacket enterPacket = new UserEnterPacket(nickname);
|
||||
await otherClient.SendAsync(enterPacket.Serialize(), SocketFlags.None);
|
||||
if (user.Value == nickname)
|
||||
continue;
|
||||
|
||||
// add me to other user
|
||||
if (Clients.TryGetValue(user.Key, out var otherClient))
|
||||
{
|
||||
UserEnterPacket enterPacket = new UserEnterPacket(nickname);
|
||||
await otherClient.SendAsync(enterPacket.Serialize(), SocketFlags.None);
|
||||
}
|
||||
|
||||
// add other user to me
|
||||
UserEnterPacket packet = new UserEnterPacket(user.Value);
|
||||
await clientSocket.SendAsync(packet.Serialize(), SocketFlags.None);
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
Console.WriteLine($"[{DateTime.Now}] EnterRoomRequest failed - RoomName: {requestPacket.RoomName}");
|
||||
|
||||
EnterRoomResponsePacket responsePacket = new EnterRoomResponsePacket(StatusCode.Failed);
|
||||
await clientSocket.SendAsync(responsePacket.Serialize(), SocketFlags.None);
|
||||
|
||||
// add other user to me
|
||||
UserEnterPacket packet = new UserEnterPacket(user.Value);
|
||||
await clientSocket.SendAsync(packet.Serialize(), SocketFlags.None);
|
||||
}
|
||||
}
|
||||
else
|
||||
else if (packetType == PacketType.UserLeave)
|
||||
{
|
||||
Console.WriteLine($"[{DateTime.Now}] EnterRoomRequest failed - RoomName: {requestPacket.RoomName}");
|
||||
|
||||
EnterRoomResponsePacket responsePacket = new EnterRoomResponsePacket(StatusCode.Failed);
|
||||
await clientSocket.SendAsync(responsePacket.Serialize(), SocketFlags.None);
|
||||
|
||||
}
|
||||
}
|
||||
else if (packetType == PacketType.UserLeave)
|
||||
{
|
||||
UserLeavePacket packet = new UserLeavePacket(dataBuffer);
|
||||
if (Rooms.TryGetValue(roomName, out var room))
|
||||
{
|
||||
room.Users.TryRemove(id, out _);
|
||||
|
||||
if (room.Users.IsEmpty)
|
||||
Rooms.TryRemove(roomName, out _);
|
||||
|
||||
roomName = "";
|
||||
|
||||
foreach (var user in room.Users)
|
||||
UserLeavePacket packet = new UserLeavePacket(dataBuffer);
|
||||
if (Rooms.TryGetValue(roomName, out var room))
|
||||
{
|
||||
if (Clients.TryGetValue(user.Key, out var otherClient))
|
||||
await otherClient.SendAsync(packet.Serialize(), SocketFlags.None);
|
||||
room.Users.TryRemove(id, out _);
|
||||
|
||||
if (room.Users.IsEmpty)
|
||||
Rooms.TryRemove(roomName, out _);
|
||||
|
||||
roomName = "";
|
||||
|
||||
foreach (var user in room.Users)
|
||||
{
|
||||
if (Clients.TryGetValue(user.Key, out var otherClient))
|
||||
await otherClient.SendAsync(packet.Serialize(), SocketFlags.None);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
else if (packetType == PacketType.Chat)
|
||||
{
|
||||
ChatPacket packet = new ChatPacket(dataBuffer);
|
||||
if (Rooms.TryGetValue(roomName, out var room))
|
||||
else if (packetType == PacketType.Chat)
|
||||
{
|
||||
foreach (var user in room.Users)
|
||||
ChatPacket packet = new ChatPacket(dataBuffer);
|
||||
if (Rooms.TryGetValue(roomName, out var room))
|
||||
{
|
||||
if (Clients.TryGetValue(user.Key, out var otherClient))
|
||||
await otherClient.SendAsync(packet.Serialize(), SocketFlags.None);
|
||||
foreach (var user in room.Users)
|
||||
{
|
||||
if (Clients.TryGetValue(user.Key, out var otherClient))
|
||||
await otherClient.SendAsync(packet.Serialize(), SocketFlags.None);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
catch (Exception ex)
|
||||
{
|
||||
Console.WriteLine(ex);
|
||||
await Remove(id, nickname, roomName, clientSocket);
|
||||
}
|
||||
}
|
||||
|
||||
private async Task Remove(string id, string nickname, string roomName, Socket clientSocket)
|
||||
{
|
||||
Clients.TryRemove(id, out _);
|
||||
if (Rooms.TryGetValue(roomName, out var room))
|
||||
{
|
||||
UserLeavePacket packet = new UserLeavePacket(nickname);
|
||||
|
||||
room.Users.TryRemove(id, out _);
|
||||
|
||||
if (room.Users.IsEmpty)
|
||||
Rooms.TryRemove(roomName, out _);
|
||||
|
||||
roomName = "";
|
||||
|
||||
foreach (var user in room.Users)
|
||||
{
|
||||
if (Clients.TryGetValue(user.Key, out var otherClient))
|
||||
await otherClient.SendAsync(packet.Serialize(), SocketFlags.None);
|
||||
}
|
||||
}
|
||||
|
||||
clientSocket.Dispose();
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user