C++实现流式socket聊天程序

最新推荐文章于 2023-10-23 17:17:16 发布
最新推荐文章于 2023-10-23 17:17:16 发布
阅读量3k 收藏 32
点赞数 6
文章标签： c++ websocket
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/Roslin_v/article/details/127301484
版权

目录

协议设计

消息的类型

消息的语法

消息的语义

消息的处理

发送消息

接收消息

程序设计

模块的划分和功能

Client客户端

Server服务器

模块流程图

程序实现

辅助代码

client.cpp

server.cpp

程序测试

本文设计并实现了使用流式socket完成双人聊天程序，支持随时发送和接收消息。在实验中遇到的主要问题及解决方案有：

  • 创建一个线程负责接收消息，解决了因为recv是阻塞函数而造成主线程等待，只能发送和接收消息接替进行的问题；
  • 引入flag表示服务器或客户端是否在线，统一了接收消息的线程和主线程，实现了接收消息的线程收到对方下线的消息后主线程也能自动结束；
  • 每次写入前清空发送和接收消息的缓冲区，避免由于前一次缓冲区溢出而造成后一次发送和接收消息异常。

如果你也遇到了同样的bug，请继续看下去吧。

协议设计

消息的类型

  1. chat：服务器与客户端之间的普通聊天信息；
  2. exit：客户端断开与服务器的连接；
  3. offline：服务器下线。

消息的语法

消息的结构如下所示：

struct message {
    int type;
    string time;
    string msg;
};

其中，string类型的字段使用的是ASCII码，每个字段通过'\n'分割。由于本文实现的是两人聊天程序，所设计的消息的结构比较简单，如果是多人聊天程序，可以增加'from', ‘to’等字段。

消息的语义

  1. type：表示消息的类型；
  2. time：表示发送消息时的本地时间，格式为年-月-日 时:分:秒；
  3. msg：信息的具体内容，例如聊天的内容等。

消息的处理

发送消息

  1. 随时可以输入要发送的消息，按回车键视为一条完整的消息结束；
  2. 将一条完整的消息存入我们定义的message结构体；
  3. 将message结构体的字段转换为字符串数组，字段之间通过'\n'分割；
  4. 向对方发送消息；
  5. 如果消息类型为exit，在命令行中打印出提示消息，关闭相关的socket，客户端断开连接；
  6. 如果消息类型为offline，在命令行中打印出提示消息，等待客户端回复‘exit’的消息后，服务器才下线。

接收消息

  1. 将字符数组类型的消息解析为我们定义的message结构体；
  2. 如果消息类型为chat，在命令行中按照时间-发送者-消息内容的格式打印出消息；
  3. 如果消息类型为exit，在命令行中打印出提示消息，关闭相关的socket，并且服务器下线（多人聊天程序则不用下线）；
  4. 如果消息类型为offline，在命令行中打印出提示消息，关闭相关的socket，客户端断开连接，并向服务器发送exit消息类型。

程序设计

模块的划分和功能

Client客户端

  1. 加载环境并创建流式socket；
  2. 连接服务器：连接服务器并输出提示；
  3. 发送消息：接收通过命令行输入的消息，通过时间戳计算出发送消息的时间，判断消息类型，将消息存入定义的message结构体，并向服务器发送消息；
  4. 接收消息：创建线程，接收消息并进行相关处理，避免在主函数中调用recv函数造成主线程阻塞；
  5. 关闭socket并清理环境。

Server服务器

  1. 加载环境并创建流式socket；
  2. 绑定ip地址和服务器；
  3. 监听：等待客户端连接；
  4. 接收客户端的连接：连接服务器并输出提示；
  5. 创建新的socket：接收客户端的连接后，创建新的socket；
  6. 发送消息：接收通过命令行输入的消息，通过时间戳计算出发送消息的时间，判断消息类型，将消息存入定义的message结构体，并向客户端发送消息；
  7. 接收消息：创建线程，接收消息并进行相关处理，避免在主函数中调用recv函数造成主线程阻塞；
  8. 关闭socket并清理环境。

模块流程图

程序实现

辅助代码

#define _WINSOCK_DEPRECATED_NO_WARNINGS
#define _CRT_SECURE_NO_WARNINGS
#include <iostream>
#include <WinSock2.h>
#include <string>
#include <time.h>
#pragma comment(lib,"ws2_32.lib")

// 表示是否在线，0表示下线，1表示在线
int flag = 0;
​
// 消息类型
enum Type {
    CHAT = 1,
    EXIT,
    OFFLINE
};
​
// 自定义消息结构
struct message {
    Type type;
    string time;
    string msg;
};
​
// message类型转换为string类型，字段间以'\n'为分隔符
string msgToString(message m) {
    string s;
    if (m.type == CHAT) {
        s = '1';
    }
    else if (m.type == EXIT) {
        s = '2';
    }
    else if (m.type == OFFLINE) {
        s = '3';
    }
    s.append("\n");
    s.append(m.time);
    s.append("\n");
    s.append(m.msg);
    s.append("\n");
    return s;
}
​
// string类型转换为message类型，字段间以'\n'为分隔符
message stringToMsg(string s) {
    message m;
    if (s[0] == '1') {
        m.type = CHAT;
    }
    else if (s[0] == '2') {
        m.type = EXIT;
    }
    else if (s[0] == '3') {
        m.type = OFFLINE;
    }
    int i = 2;
    while (s[i] != '\n') {
        i++;
    }
    m.time = s.substr(2, i - 2);
    m.msg = s.substr(i + 1);
    return m;
}

client.cpp

// 负责接收消息的线程
DWORD WINAPI clientThread(LPVOID IpParameter) {
    SOCKET sockClient = *(SOCKET*)IpParameter;
    // 接收消息的缓冲区，记得每次要清空
    char recvBuf[1024];
    memset(recvBuf, 0, sizeof(recvBuf));
    int recvLen = 0;
    // 当客户端还在线时，持续接收消息
    while (flag) {
        recvLen = recv(sockClient, recvBuf, 1024, 0);
        if (recvLen > 0) {
            string s = recvBuf;
            message r = stringToMsg(s);
            message m;
            char tmp[32] = { NULL };
            time_t t;
            switch (r.type){
            // 输出聊天消息
            case 1:
                cout << r.time << " " << "Server: " << r.msg;
                break;
            // 服务器下线，客户端向服务器发送断开连接的消息，并下线
            case 3:
                cout << "---------------------------------------------------" << endl;
                cout << "[SERVER IS OFFLINE! PRESS ENTER TO CLOSE CONNECTION!]" << endl;
                m.type = EXIT;
                t = time(0);
                strftime(tmp, sizeof(tmp), "%Y-%m-%d %H:%M:%S", localtime(&t));
                m.time = tmp;
                m.msg = "Client closed connection.";
                char sendBuf[1024];
                memset(sendBuf, 0, sizeof(sendBuf));
                strcpy_s(sendBuf, msgToString(m).c_str());
                send(sockClient, sendBuf, sizeof(sendBuf), 0);
                flag = 0;
                break;
            default:
                break;
            }
        }
        memset(recvBuf, 0, sizeof(recvBuf));
    }
    return 0;
}
​
int main() {
    // 加载环境
    WSADATA wsaData;
    if (WSAStartup(MAKEWORD(2, 2), &wsaData) != 0) {
        cout << "WSA STARTUP ERROR：" << GetLastError() << endl;
        return 0;
    }
    else {
        cout << "[WSA STARTED UP!]" << endl;
    }
​
    // 创建流式套接字
    SOCKET sockClient = socket(AF_INET, SOCK_STREAM, 0);
    if (sockClient == INVALID_SOCKET) {
        cout << "SOCKET ERROR：" << GetLastError() << endl;
        return 0;
    }
    else {
        cout << "[SOCKET BUILT!]" << endl;
    }
​
    // 连接服务器
    sockaddr_in addrServer;
    addrServer.sin_family = AF_INET;
    addrServer.sin_port = htons(8000);
    addrServer.sin_addr.S_un.S_addr = inet_addr("127.0.0.1");
    if (connect(sockClient, (SOCKADDR*)&addrServer, sizeof(SOCKADDR))) {
        cout << "CONNECT ERROR：" << GetLastError() << endl;
        return 0;
    }
    else {
        cout << "[CONNECTION BUILT! ENTER exit TO CLOSE CONNECTION!]" << endl;
        cout << "---------------------------------------------------" << endl;
        flag = 1;
    }
​
    // 接收和发送消息
    char recvBuf[1024];
    char sendBuf[1024];
    memset(recvBuf, 0, sizeof(recvBuf));
    memset(sendBuf, 0, sizeof(sendBuf));
    message m;
    // 创建线程负责接收消息
    CloseHandle(CreateThread(NULL, 0, clientThread, (LPVOID)&sockClient, 0, 0));
    // 当客户端还在线时，随时可以发送消息
    while (flag) {
        char in[1000];
        cin.getline(in, 1000);
        m.msg = in;
        // 通过时间戳计算发送时间
        char tmp[32] = { NULL };
        time_t t = time(0);
        strftime(tmp, sizeof(tmp), "%Y-%m-%d %H:%M:%S", localtime(&t));
        m.time = tmp;
        if (m.msg == "exit") {
            m.type = EXIT;
        }
        else {
            m.type = CHAT;
        }
        strcpy_s(sendBuf, msgToString(m).c_str());
        send(sockClient, sendBuf, sizeof(sendBuf), 0);
        memset(sendBuf, 0, sizeof(sendBuf));
        // 输入exit断开连接
        if (m.msg == "exit") {
            cout << "---------------------------------------------------" << endl;
            cout << "[CONNECTION CLOSED!]" << endl;
            flag = 0;
            break;
        }
        /*
        recvLen = recv(sockClient, recvBuf, 1024, 0);
        if (recvLen > 0) {
            cout << recvBuf << endl;
        }
        memset(recvBuf, 0, sizeof(recvBuf));
        */
    }
​
    // 关闭监听套接字
    closesocket(sockClient);
    cout << "[SOCKET CLOSED!]" << endl;
​
    //清理环境
    WSACleanup();
    cout << "[WSA CLEANED UP!]" << endl;
​
    system("pause");
    return 0;
}

server.cpp

DWORD WINAPI serverThread(LPVOID IpParameter) {
    SOCKET sockConn = *(SOCKET*)IpParameter;
    char recvBuf[1024];
    memset(recvBuf, 0, sizeof(recvBuf));
    int recvLen = 0;
    while (flag) {
        recvLen = recv(sockConn, recvBuf, 1024, 0);
        if (recvLen > 0) {
            string s = recvBuf;
            message r = stringToMsg(s);
            switch (r.type) {
            case 1:
                cout << r.time << " " << "Client: " << r.msg;
                break;
            case 2:
                cout << "---------------------------------------------------" << endl;
                cout << "[CLIENT CLOSED CONNECTION! PRESS ENTER TO CLOSE CONNECTION!]" << endl;
                flag = 0;
                break;
            default:
                break;
            }
        }
        memset(recvBuf, 0, sizeof(recvBuf));
    }
    return 0;
}
​
int main() {
    WSADATA wsaData;
    if (WSAStartup(MAKEWORD(2, 2), &wsaData) != 0) {
        cout << "WSASTARTUP ERROR：" << GetLastError() << endl;
        return 0;
    }
    else {
        cout << "[WSA STARTED UP!]" << endl;
    }
​
    SOCKET sockServer = socket(AF_INET, SOCK_STREAM, 0);
    if (sockServer == INVALID_SOCKET) {
        cout << "SOCKET ERROR：" << GetLastError() << endl;
        return 0;
    }
    else {
        cout << "[SOCKET BUILT!]" << endl;
    }
​
    // 绑定ip地址和服务器
    sockaddr_in addrServer;
    memset(&addrServer, 0, sizeof(sockaddr_in));
    addrServer.sin_family = AF_INET;
    addrServer.sin_port = htons(8000);
    addrServer.sin_addr.S_un.S_addr = inet_addr("127.0.0.1");
    if (bind(sockServer, (SOCKADDR*)&addrServer, sizeof(SOCKADDR)) == SOCKET_ERROR) {
        cout << "BIND ERROR:" << WSAGetLastError() << endl;
        return 0;
    }
    else {
        cout << "[BINDED. SERVER UP!]" << endl;
        flag = 1;
    }
​
    // 监听
    listen(sockServer, 5);
    cout << "[LISTENING!]" << endl;
​
    // 当服务器在线时，接收客户端的连接请求
    while (flag) {
        sockaddr_in addrClient;
        int len = sizeof(sockaddr_in);
        // 接收客户端的连接请求并创建新的套接字
        SOCKET sockConn = accept(sockServer, (SOCKADDR*)&addrClient, &len);
        if (sockConn != INVALID_SOCKET) {
            cout << "[ACCEPT CONNECTION REQUEST! ENTER off TO CLOSE CONNECTION!]" << endl;
            cout << "---------------------------------------------------" << endl;
            char recvBuf[1024];
            char sendBuf[1024];
            memset(recvBuf, 0, sizeof(recvBuf));
            memset(sendBuf, 0, sizeof(sendBuf));
            message m;
            CloseHandle(CreateThread(NULL, 0, serverThread, (LPVOID)&sockConn, 0, 0));
            while (flag) {
                char in[1000];
                cin.getline(in, 1000);
                m.msg = in;
                char tmp[32] = { NULL };
                time_t t = time(0);
                strftime(tmp, sizeof(tmp), "%Y-%m-%d %H:%M:%S", localtime(&t));
                m.time = tmp;
                if (m.msg == "off") {
                    m.type = OFFLINE;
                }
                else {
                    m.type = CHAT;
                }
                strcpy_s(sendBuf, msgToString(m).c_str());
                send(sockConn, sendBuf, sizeof(sendBuf), 0);
                memset(sendBuf, 0, sizeof(sendBuf));
                if (m.msg == "off") {
                    cout << "---------------------------------------------------" << endl;
                    cout << "[CONNECTION CLOSED!]" << endl;
                    flag = 0;
                    break;
                }
            }
        }
    
        closesocket(sockConn);
    }
    
    closesocket(sockServer);
    cout << "[SOCKET CLOSED!]" << endl;
​
    WSACleanup();
    cout << "[WSA CLEANED UP!]" << endl;
​
    system("pause");
    return 0;
}

程序测试

对程序进行测试，可以看到服务器和客户端可以正常通信，通信消息带有时间标签，并且可以正常退出。

Roslin_v
关注
  • 6
    点赞
  • 踩
  • 32
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
C++入门教程（18）socket 实现简单聊天室
CV战士的日常
03-19 9117
本节通过socket实现一个简单的聊天室功能 聊天室中如果有人说话，则服务器负责将内容传送给聊天室的其他人 那么就需要客户端和服务端两个程序，客户端负责发送消息，服务端负责接收和转发消息 客户端代码： #include <sys/types.h> #include <sys/socket.h> #include <stdio.h> #include <netinet/in.h> #include <arpa/inet.h> #in
c语言实现局域网聊天程序,局域网聊天的程序(C++版)
weixin_42515045的博客
05-17 768
该楼层疑似违规已被系统折叠隐藏此楼查看此楼//server#include#include#include#include#include#definePORT100#definesocklen_tint#pragmacomment(lib,"wsock32.lib")SOCKETs;SOCKETclient;sockaddr_infrom;voidServerThr...
socket编程1：以实现小型聊天室为例（附C++代码）
weixin_50941083的博客
05-31 2631
以实现一个小型聊天室为例，学习socket编程的相关知识。
tcp/ip c++ 即时通讯聊天室(四）实现公网传输
Platonic_hacker的博客
06-08 1999
正常来讲之前三篇文章都是在讲如何在局域网内进行传输数据，那么我们如果想让数据在公网上传输便需要内网穿透技术。 内网穿透技术原理很简单，借用另一个能连上公网的服务器当作跳板连接你的服务端与公网。（大佬们可以用服务器自己搭建一个穿透端口）我们这里则直接利用网上提供的tcp内网穿透服务来实现我们的这一目的。 这里我们利用Sunny-Ngrok的免费tcp穿透服务（因为是免费的所以速度较慢，有钱的可以选择付费的），但对于此聊天所需要的流量来讲已经足够了。 首先我们在sunny-ngrok官网上注册一个账号，直接邮
C++实现聊天程序(QQ类型)
热门推荐
shaohuiyang的博客
02-09 1万+
服务端#include&lt;iostream&gt; #include&lt;WinSock2.h&gt; #pragma comment(lib,"ws2_32.lib") using namespace std; void initialization(); int main(){ //定义长度变量 int send_len=0; int recv_len=0; int len=0; ...
C++基于socket的多线程聊天程序
03-17
Socket分为流式（SOCK_STREAM，对应TCP）和数据报（SOCK_DGRAM，对应UDP）两种类型，本程序可能使用了流式socket，因为TCP提供了面向连接、可靠的数据传输服务，适合群聊应用。 2. **多线程**：多线程技术允许程序...
c++实现socket通信
11-03
- **Socket类型**：Socket分为两种主要类型，流式Socket（TCP）和数据报Socket（UDP）。TCP提供面向连接的服务，确保数据的可靠传输；UDP则为无连接服务，数据传输速度快但不保证顺序。 2. **创建Socket**： - ...
C++ 实现json形式的Socket传输图片
08-11
本文将深入探讨如何使用C++实现基于JSON格式的Socket传输图片。首先，我们要了解涉及的知识点： 1. **C++**: C++是一种强大的、面向对象的编程语言，广泛用于系统软件、应用软件、游戏开发和高性能计算等领域。它的...
基于Socket的MFC聊天室程序
12-18
总的来说，基于Socket的MFC聊天室程序是一个结合了网络编程和GUI编程的项目，它展示了如何利用MFC来构建用户友好的界面，同时通过Socket实现网络通信，满足多人实时交流的需求。通过学习和理解这个项目，开发者可以...
拟qq聊天程序 socket实现
04-15
总结来说，"拟qq聊天程序 socket实现"是一个涵盖了Socket编程、多线程、MFC UI设计、网络通信协议等多个IT知识点的项目。通过学习和实践这样的项目，开发者可以提升网络编程和客户端-服务器应用开发的能力。
视频推流调试.rar java PHP c++ 推流调试工具，socket开发
06-17
视频推流调试.rar java PHP c++ 推流调试工具，socket开发
VC++ socket 聊天室客户端
01-30
详细可看这篇我的博客http://blog.csdn.net/lightspear/article/details/54790683
Socket聊天室C++
12-22
c++编写的网络聊天室，有服务端和客户端，在同一个局域网内可以进行聊天。
c++聊天小程序
zylg
07-01 2173
纯c++写的聊天小程序 涉及的技术 c++ socket c++ stl c++ thread 设计的原理 以一个结构体的形式存储客户端,用vector存取存在的客户端,开启多线程处理逻辑 效果 源代码 server #include<thread> #include <WinSock2.h> #include<iostream> #include<...
c++网络编程实战入门（实现聊天室）
weixin_36328380的博客
04-04 7780
epoll 使用详解 https://www.cnblogs.com/fnlingnzb-learner/p/5835573.html
socket——流式套接字编程实例及注意事项
cprimesplus的博客
02-21 782
流式套接字基本操作步骤 C端（主动端） 获取socket 给socket取得地址（可省） 发送连接 收发消息 关闭 S端（被动端） 获取socket 给socket取得地址 将socket设置为监听模式 接受连接 收发消息 关闭 ...
C/C++项目之网络聊天室在线聊天系统
最新发布
weixin_52542873的博客
10-23 1587
多客户端聊天室内容：1、C/S架构分析2、TCP协议编程模型3、多线程实现并发4、网络聊天室制作TCP协议：通信方式IP协议：区分同一网络上的不同主机TCP协议：1、确定协议版本 1、确定协议版本2、创建socket 2、创建socket3、确定协议地址簇 3、获取服务器协议地址4、绑定5、监听6、接收客户端连接 4、连接服务器。
onvif协议
04-28 1万+
onvif协议是视频监控领域共同制订的一套标准，以探测同一局域网内摄像头信息。 onvif开源代码： #include <stdio.h> #include <stdlib.h> #include <string.h> #include <sys/socket.h> #include <netinet/in.h> #include...
糖儿飞教你学C++ Socket网络编程——6.控制台版的TCP通信程序
03-09 3421
根据图2-1的TCP通信程序的流程，下面编程实现一个控制台版的TCP通信程序，程序分为服务器端和客户端，双方可以相互发送消息，运行效果如图2-4所示。 图2-4 控制台版的TCP通信程序（左图为服务器端，右图为客户端） 2.2.1服务器端程序的制作 服务器端程序的制作步骤如下： 1. 在VC6中新建工程，选择“Win32 Console Application”，输入工程名（如TCP...
socket编程 C++
09-14
在C语言中进行socket编程，需要使用socket函数来创建一个套接字。该函数的原型如下所示： #include <sys/types.h> #include <sys/socket.h> int socket(int domain, int type, int protocol); 其中，domain参数指定了使用的协议族，常见的有AF_INET（IPv4）和AF_INET6（IPv6）。type参数指定了套接字的类型，常见的有SOCK_STREAM（流式套接字，用于TCP）和SOCK_DGRAM（数据报套接字，用于UDP）。protocol参数通常为0，表示自动选择合适的协议。 例如，要创建一个IPv4的TCP套接字，可以使用以下代码： int sockfd = socket(AF_INET, SOCK_STREAM, 0); 需要注意的是，socket函数返回一个整数值，表示套接字的文件描述符。如果返回值小于0，则表示创建套接字失败。 可以参考引用中提到的博文，了解更多关于C语言实现udp客户端、tcp客户端和tcp服务端的代码实现。
评论 2
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值