【linux网络编程】C++实现本地简易聊天室服务器（100行代码）

ylxc_tu

已于 2024-09-21 21:02:43 修改

阅读量429

点赞数 3

文章标签：服务器 linux 网络 c++

于 2024-09-20 20:12:22 首次发布

本文链接：https://blog.csdn.net/ylxc_tu/article/details/142395628

版权

学习了多线程与Socket套接字后，打算写一个类似qq的本地群聊聊天室作为练习，于是就写了下面的这些代码。运行环境为linux，编程语言为c++11。

话不多说直接上代码。

服务器代码：

#include <iostream>
#include <cstring>
#include <vector>
#include <unistd.h>
#include <fcntl.h>
#include <pthread.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <queue>
#include <utility>

using namespace std;
#define SERV_PORT 5555                        // port number

queue<pair<pthread_t,string>>WriteQueue;      //this helps to contain the message which has been read and need to be send to other sockets.
vector<pair<pthread_t,int>>Thread;            //the second of the pair is the file describer of the socket.
pthread_mutex_t mutex;

void* WriteMessageOut(void *)
{
    while(1)
    {   
        while(!WriteQueue.empty())
        {
            pthread_mutex_lock(&mutex);
        
            pthread_t sender=WriteQueue.front().first;
            string s=WriteQueue.front().second;
            WriteQueue.pop();
            s="user "+to_string(sender%1000)+":"+s;
            cout<<s<<endl;

            for(auto pair:Thread)
            {
                if(sender!=pair.first)
                {
                    int fd=pair.second;
                    char str[BUFSIZ];
                    strcpy(str,s.c_str());
                    write(fd,str,strlen(str));
                }
            }
            pthread_mutex_unlock(&mutex);
        }
        usleep(1000);                       //sleep for 1000 microseconds.(weimiao)
    }
    return NULL;
}

void* ReadMessageIn(void *arg)
{
    int fd=*reinterpret_cast<int*>(arg);
    int ret;
    char buf[BUFSIZ];
    while((ret=read(fd, buf, BUFSIZ))>0)
    {
        buf[ret] = '\0';
        string s(buf);
        WriteQueue.emplace(pthread_self(),s);
    }
    close(fd);
    string s="user " +to_string(pthread_self() % 1000)+" exited.\n";
    cout << s << endl;
    WriteQueue.emplace(pthread_self(), s);
    pthread_exit(0);
    return NULL;
}

pthread_t CreateThread(void*(*f)(void *),void *arg)
{
    pthread_t ret;
    pthread_create(&ret,NULL,f,arg);
    //pthread_detach(ret);                //分离该线程，防止僵尸线程
    return ret;
}

int main()
{
    int sid;
    int ret;
    int fd;

    pthread_mutex_init(&mutex, NULL);

    sid=socket(AF_INET,SOCK_STREAM,0);
    
    struct sockaddr_in servaddr;
    servaddr.sin_family=AF_INET;
    servaddr.sin_port=htons(SERV_PORT);
    servaddr.sin_addr.s_addr=htonl(INADDR_ANY);

    ret=bind(sid,(struct sockaddr*)&servaddr,sizeof(servaddr));

    listen(sid,128);

    pthread_t WriteIn=CreateThread(WriteMessageOut,NULL);

    cout<<"Server Started."<<endl;
    while((fd=accept(sid,NULL,NULL))!=-1)
    {
      pthread_t pid_thread =CreateThread(ReadMessageIn, (void *)&fd);
      Thread.emplace_back(pid_thread, fd);
      string s = "user " + to_string(pid_thread % 1000) + " connected.\n";
      cout << s<<endl;
      WriteQueue.emplace(pid_thread, s);
    }

    close(sid);
    pthread_mutex_destroy(&mutex);

    return 0;
}