base chat module

This commit is contained in:
2023-02-07 18:15:24 +09:00
parent f74090a49e
commit f97220d7c2
10 changed files with 223 additions and 0 deletions

9
Chat/Core/Core.csproj Normal file
View File

@@ -0,0 +1,9 @@
<Project Sdk="Microsoft.NET.Sdk">
<PropertyGroup>
<TargetFramework>net6.0</TargetFramework>
<ImplicitUsings>enable</ImplicitUsings>
<Nullable>enable</Nullable>
</PropertyGroup>
</Project>

35
Chat/Core/PClient.cs Normal file
View File

@@ -0,0 +1,35 @@
using System;
using System.Collections.Generic;
using System.Linq;
using System.Net;
using System.Net.Sockets;
using System.Text;
using System.Threading.Tasks;
namespace Core
{
public class PClient
{
private Socket socket = new Socket(AddressFamily.InterNetwork, SocketType.Stream, ProtocolType.Tcp);
private EndPoint endPoint;
public PClient(string ip, int port)
{
endPoint = new IPEndPoint(IPAddress.Parse(ip), port);
}
public async Task StartAsync()
{
await socket.ConnectAsync(endPoint);
while (true)
{
string str = Console.ReadLine() ?? string.Empty;
byte[] dataBuffer = Encoding.UTF8.GetBytes(str);
byte[] headerBuffer = BitConverter.GetBytes(IPAddress.HostToNetworkOrder((short)dataBuffer.Length));
await socket.SendAsync(headerBuffer, SocketFlags.None);
await socket.SendAsync(dataBuffer, SocketFlags.None);
}
}
}
}

69
Chat/Core/PServer.cs Normal file
View File

@@ -0,0 +1,69 @@
using System;
using System.Collections.Generic;
using System.Linq;
using System.Net;
using System.Net.Sockets;
using System.Text;
using System.Threading.Tasks;
namespace Core
{
public class PServer
{
private Socket serverSocket = new Socket(AddressFamily.InterNetwork, SocketType.Stream, ProtocolType.Tcp);
public PServer(string ip, int port, int backlog)
{
IPEndPoint endPoint = new IPEndPoint(IPAddress.Parse(ip), port);
serverSocket.Bind(endPoint);
serverSocket.Listen(backlog);
}
public async Task StartAsync()
{
while (true)
{
Socket clientSocket = await serverSocket.AcceptAsync();
Console.WriteLine($"[{DateTime.Now}] Accept client: {clientSocket.RemoteEndPoint}");
ThreadPool.QueueUserWorkItem(RunAsync, clientSocket);
}
}
private async void RunAsync(object? sender)
{
if (sender == null)
return;
Socket clientSocket = (Socket)sender;
byte[] headerBuffer = new byte[2];
while (true)
{
// header buffer
int n1 = await clientSocket.ReceiveAsync(headerBuffer, SocketFlags.None);
if (n1 < 1)
{
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;
}
Console.WriteLine($"[{DateTime.Now}] {clientSocket.RemoteEndPoint} client: {Encoding.UTF8.GetString(dataBuffer)}");
}
}
}
}