C# Socket编程实现简单的局域网聊天器的示例代码

时间：2020-12-22 18:02:16|栏目：.NET代码|点击：次

前言

最近在学习C# Socket相关的知识，学习之余，动手做了一个简单的局域网聊天器。有萌生做这个的想法，主要是由于之前家里两台电脑之间想要传输文件十分麻烦，需要借助QQ，微信或者其他第三方应用，基本都要登录，而且可能传输的文件还有大小限制，压缩问题。所以本聊天器的首要目标就是解决这两个问题，做到使用方便（双击启动即用），传文件无限制。

废话不多说，先上图。S-Chat是服务端，C-Chat是客户端，两者除了客户端首次启动后需要设置一下连接的IP地址外，无其他区别。操作与界面都完全相同，对于用户来说，基本不用在意谁是服务端谁是客户端。

编码

服务端监听接口

服务端主要负责开启监听线程，等待客户端接入

public void StartListen()
{
 // 创建Socket对象 new Socket(AddressFamily.InterNetwork, SocketType.Stream, ProtocolType.Tcp)
 Socket socket = GetSocket();
 // 将套接字与IPEndPoint绑定
 socket.Bind(this.GetIPEndPoint());
 // 开启监听 仅支持一个连接
 socket.Listen(1);
 // 开启线程等待客户端接入，避免堵塞
 Thread acceptThread = new Thread(new ThreadStart(TryAccept));
 acceptThread.IsBackground = true;
 acceptThread.Start();
}

public void TryAccept()
{
 Socket socket = GetSocket();
 while (true)
 {
  try
  {
   Socket connectedSocket = socket.Accept()
   this.ConnectedSocket = connectedSocket;
   OnConnect(); // 连接成功回调
   this.StartReceive(); // 开始接收线程
   break;
  }
  catch (Exception e)
  {
  }
 }
}

客户端连接接口

客户端主要负责开启连接线程，每隔2秒，自动尝试连接服务端

public void StartConnect()
{
 Thread connectThread = new Thread(new ThreadStart(TryConnect));
 connectThread.IsBackground = true;
 connectThread.Start();
}

public void TryConnect()
{
 Socket socket = GetSocket();
 while (true)
 {
  try
  {
   socket.Connect(this.GetIPEndPoint());
   this.ConnectedSocket = socket;
   OnConnect(); // 连接成功回调
   this.StartReceive();
   break;
  }
  catch (Exception e)
  {
   Thread.Sleep(TryConnectInterval); // 指定间隔后重新尝试连接
  }
 }
}

文字发送，文件发送，接收文字，接收文件等通用接口主要实现在 ChatBase 类中，是服务端与客户端的共同父类。

文字发送接口

发送数据的第一位表示发送信息的类型，0表示字符串文字，1表示文件

然后获取待发送字符串的长度，使用long类型表示，占用8个字节

共发送的字节数据可以表示为头部（类型 + 字符串字节长度，共9个字节）+ 实际字符串字节数据

public bool Send(string msg)
{
 if (ConnectedSocket != null && ConnectedSocket.Connected)
 {
  byte[] buffer = UTF8.GetBytes(msg); 
  byte[] len = BitConverter.GetBytes((long)buffer.Length); 
  byte[] content = new byte[1 + len.Length + buffer.Length]; 
  content[0] = (byte)ChatType.Str; // 发送信息类型，字符串
  Array.Copy(len, 0, content, 1, len.Length); // 字符串字节长度
  Array.Copy(buffer, 0, content, 1 + len.Length, buffer.Length); // 实际字符串字节数据
  try
  {
   ConnectedSocket.Send(content);
   return true;
  }
  catch (Exception e)
  {
  }
 }
 return false;
}

文件发送接口

与字符串发送相同的头部可以表示为（类型 + 文件长度，共9个字节）

还需要再加上待发送的文件名的长度，与文件名字节数据

共发送的字节数据可以表示为头部（类型 + 文件长度，共9个字节）+ 文件名头部（文件名长度 + 文件名字节数据）+ 实际文件数据

public bool SendFile(string path)
{
 if (ConnectedSocket != null && ConnectedSocket.Connected)
 {
  try
  {
   FileInfo fi = new FileInfo(path);
   byte[] len = BitConverter.GetBytes(fi.Length); 
   byte[] name = UTF8.GetBytes(fi.Name); 
   byte[] nameLen = BitConverter.GetBytes(name.Length); 
   byte[] head = new byte[1 + len.Length + nameLen.Length + name.Length];
   head[0] = (byte)ChatType.File; // 加上信息发送类型
   Array.Copy(len, 0, head, 1, len.Length); // 加上文件长度
   Array.Copy(nameLen, 0, head, 1 + len.Length, nameLen.Length); // 加上文件名长度
   Array.Copy(name, 0, head, 1 + len.Length + nameLen.Length, name.Length); // 加上文件名字节数据
   ConnectedSocket.SendFile(
    path,
    head,
    null,
    TransmitFileOptions.UseDefaultWorkerThread
   );
   return true;
  }
  catch(Exception e)
  {
  }
 }
 return false;
}

信息接收接口（文字与文件）

主要是解析接收到的字节数据，根据字符串或文件的类型进行处理

public void Receive()
{
 if (ConnectedSocket != null)
 {
  while (true)
  {
   try
   {
    // 读取公共头部
    byte[] head = new byte[9];
    ConnectedSocket.Receive(head, head.Length, SocketFlags.None);
    int len = BitConverter.ToInt32(head, 1);
    if (head[0] == (byte) ChatType.Str)
    {
     // 接收字符串
     byte[] buffer = new byte[len];
     ConnectedSocket.Receive(buffer, len, SocketFlags.None);
     OnReceive(ChatType.Str, UTF8.GetString(buffer));
    }
    else if(head[0] == (byte)ChatType.File)
    {
     // 接收文件
     if (!Directory.Exists(dirName))
     {
      Directory.CreateDirectory(dirName);
     }
     // 读取文件名信息
     byte[] nameLen = new byte[4];
     ConnectedSocket.Receive(nameLen, nameLen.Length, SocketFlags.None);
     byte[] name = new byte[BitConverter.ToInt32(nameLen, 0)];
     ConnectedSocket.Receive(name, name.Length, SocketFlags.None);
     string fileName = UTF8.GetString(name);
     // 读取文件内容并写入
     int readByte = 0;
     int count = 0;
     byte[] buffer = new byte[1024 * 8];
     string filePath = Path.Combine(dirName, fileName);
     if (File.Exists(filePath))
     {
      File.Delete(filePath);
     }
     using (FileStream fs = new FileStream(filePath, FileMode.Append, FileAccess.Write))
     {
      while (count != len)
      {
       int readLength = buffer.Length;
       if(len - count < readLength)
       {
        readLength = len - count;
       }
       readByte = ConnectedSocket.Receive(buffer, readLength, SocketFlags.None);
       fs.Write(buffer, 0, readByte);
       count += readByte;
      }
     }
     OnReceive(ChatType.File, fileName); 
    }
    else
    {
     // 未知类型
    }
   }
   catch (Exception e)
   {
   }
  }
 }
}

使用

第一次使用，客户端需要设置待连接的IP地址。之后再启动会自动连接

双击服务端exe启动，点击 设置 ，查看IP地址项