欢迎来到军工软件开发人才培养基地——学到牛牛

TCP多线程聊天室

时间:2024-05-06 07:01:10 来源:学到牛牛

 

【有一些复杂问题,先不发表出来】

基于 tcp 实现群聊功能,本项目设计是在Linux环境下基于套接字(Socket)和多线程编程进行开发的简易聊天室

 

 

TCP前言

传输控制协议(TCP)是互联网协议组的主要协议之一。它起源于最初的网络实施,在网络实施中,它补充了互联网协议。因此,整个套件通常被称为TCP/IP。TCP在通过IP网络通信的主机上运行的应用程序之间提供可靠、有序且经过错误检查的八位字节流传递。万维网、电子邮件、远程管理和文件传输等主要互联网应用都依赖于TCP。不需要可靠数据流服务的应用程序可以使用用户数据报协议(User Datagram Protocol ,UDP),该协议提供无连接数据报服务,强调降低延迟而不是可靠性。

多线程

在计算机体系结构中,多线程是由操作系统支持的中央处理器(或多核处理器中的单核)同时具有执行多个线程能力的方法。这种方法不同于多处理。在多线程应用程序中,线程共享单个或多个内核的资源,包括计算单元、中央处理器缓存和转换查找缓冲器(TLB)。

线程和进程

线程:一个程序中的多个执行路径。

进程:运行中的程序

Linux系统下的多线程遵循POSIX线程接口,称为 pthread。编写Linux下的多线程程序,需要使用头文件pthread.h,连接时需要使用库libpthread.a。

头文件

#include <pthread.h>

一个重要的线程创建函数原型:

int pthread_create(pthread_t *restrict tidp,const pthread_attr_t *restrict  attr, void *(*start_rtn)(void),void *restrict arg); //创建一个新线程

成功创建返回0,错误则返回其它

聊天室源码

Sever.c

#include <stdio.h>

#include <string.h>

#include <sys/types.h>

#include <stdlib.h>

#include <unistd.h>

#include <netinet/in.h>

#include <arpa/inet.h>

#include <sys/socket.h>

#include <pthread.h>

/*

int sockfd;//服务器socket

int fds[100];//客户端的socketfd,100个元素,fds[0]~fds[99]

int size =100 ;//用来控制进入聊天室的人数为100以内

char* IP = "127.0.0.1"; //这里要改成自己的虚拟机IP

short PORT = 10222;

*/

typedef struct sockaddr SA;

int sockfd;//服务器socket

int fds[100];//客户端的socketfd,100个元素,fds[0]~fds[99]

 

void init()

{

    short PORT = 8848;

    char* IP = "127.0.0.1"; //这里要改成自己的虚拟机IP

    int size =100 ;//用来控制进入聊天室的人数为100以内

    sockfd = socket(PF_INET,SOCK_STREAM,0);

    if (sockfd == -1){

        perror("创建socket失败");

        exit(-1);

    }

    struct sockaddr_in addr;

    addr.sin_family = PF_INET;

    addr.sin_port = htons(PORT);

    addr.sin_addr.s_addr = inet_addr(IP);

    if (bind(sockfd,(SA*)&addr,sizeof(addr)) == -1){

        perror("绑定失败");

        exit(-1);

    }

    if (listen(sockfd,100) == -1){

        perror("设置监听失败");

        exit(-1);

    }

}

 

void SendMsgToAll(char* msg)

{

    int size =100 ;//用来控制进入聊天室的人数为100以内

    int i;

    for (i = 0;i < size;i++){

        if (fds[i] != 0){

            printf("sendto%d\n",fds[i]);

            printf("msg = %s\n",msg);

            send(fds[i],msg,strlen(msg),0);

        }

    }

}

 

void* service_thread(void* p)

{

    int size =100 ;//用来控制进入聊天室的人数为100以内

    int fd = *(int*)p;

    printf("pthread = %d\n",fd);

    while(1){

        char buf[100] = {};

        if (recv(fd,buf,sizeof(buf),0) <= 0){

            int i;

            for (i = 0;i < size;i++){

                if (fd == fds[i]){

                    fds[i] = 0;

                    break;

                }

            }

                printf("退出:fd = %dquit\n",fd);

                pthread_exit((void*)i);

        }

        //把服务器接受到的信息发给所有的客户端

        SendMsgToAll(buf);

    }

}

 

void service()

{

    int size =100 ;//用来控制进入聊天室的人数为100以内

    printf("服务器启动\n");

    while(1){

        struct sockaddr_in fromaddr;

        socklen_t len = sizeof(fromaddr);

        int fd = accept(sockfd,(SA*)&fromaddr,&len);

        if (fd == -1){

            printf("客户端连接出错...\n");

            continue;

        }

        int i = 0;

        for (i = 0;i < size;i++){

            if (fds[i] == 0){

                //记录客户端的socket

                fds[i] = fd;

                printf("fd = %d\n",fd);

                //有客户端连接之后,启动线程给此客户服务

                pthread_t tid;

                pthread_create(&tid,0,service_thread,&fd);

                break;

            }

        if (size == i){

            //发送给客户端说聊天室满了

            char* str = "对不起,聊天室已经满了!";

            send(fd,str,strlen(str),0); 

            close(fd);

        }

        }

    }

}

 

int main()

{

    init();

    service();

}

 

Client.c

#include <stdio.h>

#include <stdlib.h>

#include <string.h>

#include <unistd.h>

#include <sys/socket.h>

#include <netinet/in.h>

#include <arpa/inet.h>

#include <pthread.h>

#include <unistd.h>

 

int sockfd;//客户端socket

char* IP = "127.0.0.1";//改成自己的IP

short PORT = 8848;//服务器服务端口

typedef struct sockaddr SA;

char name[30];

 

void init()

{

    sockfd = socket(PF_INET,SOCK_STREAM,0);

    struct sockaddr_in addr;

    addr.sin_family = PF_INET;

    addr.sin_port = htons(PORT);

    addr.sin_addr.s_addr = inet_addr(IP);

    if (connect(sockfd,(SA*)&addr,sizeof(addr)) == -1){

        perror("无法连接到服务器");

        exit(-1);

    }

    printf("客户端启动成功\n");

}

 

void start()

{

    pthread_t id;

    void* recv_thread(void*);

    pthread_create(&id,0,recv_thread,0);

    char buf2[100] = {};

    sprintf(buf2,"%s进入了聊天室",name);

    send(sockfd,buf2,strlen(buf2),0);

    while(1){

        char buf[100] = {};

        scanf("%s",buf);

        char msg[131] = {};

        sprintf(msg,"%s:%s",name,buf);

        send(sockfd,msg,strlen(msg),0);

        if (strcmp(buf,"bye") == 0){

            memset(buf2,0,sizeof(buf2));

            sprintf(buf2,"%s退出了聊天室",name);

            send(sockfd,buf2,strlen(buf2),0);

            break;

        }

    }

    close(sockfd);

}

 

void* recv_thread(void* p)

{

    while(1){

        char buf[100] = {};

        if (recv(sockfd,buf,sizeof(buf),0) <= 0){

            return;

        }

        printf("%s\n",buf);

    }

}

 

int main()

{

    init();

    printf("请输入您的名字:");

 //   scanf("%s",name);

    read( 1,"%s",1024);

    start();

    return 0;

}