zoukankan      html  css  js  c++  java
  • C# Socket编程实现简单的局域网聊天器

    前言

    最近在学习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地址。之后再启动会自动连接

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

      2. 双击客户端exe启动,点击设置,在IP地址项,输入服务端查看到的IP地址

    • 设置成功后,等待大约一两秒,应用cion变成绿色,即表示连接成功,可以正常发送文字和文件了

    • 可以点击选择文件(支持选择多个文件),发送文件

    • 支持直接拖拽文件到输入框,发送文件

    • 支持Ctrl+Enter快捷键发送

    • 接收到的文件自动存放在exe所在目录的ChatFiles文件夹下

    注意事项

    • 客户端服务端需要在同一个局域网下才能实现连接
    • 服务端IP地址是不支持修改的,自动读取本机的IP地址

    源码

    • 完整代码放在GitHub上,点击查看
    • 预编译好的可运行exe程序,在仓库的Release目录,也可以直接通过百度云下载,提取码v4pe
  • 相关阅读:
    Apache JMeter 入门教程
    nacos 配置中心 & 服务发现 使用
    Apache Flink能用来做什么?
    使用etcd watch做服务发现
    服务的注册与发现(Consul、zookeeper、etcd、eureka、Nacos)
    四层、七层负载均衡的区别
    Apollo核心概念之“Namespace”
    Charles的原理、作用
    GIN框架中间件 C.NEXT() C.ABORT() C.SET() C.GET 跨中间件取值 GIN中间件中使用GOROUTINE
    gin内置验证器使用
  • 原文地址:https://www.cnblogs.com/iwiniwin/p/12501969.html
Copyright © 2011-2022 走看看