基于Linux的TCP网络聊天室

1.实验项目名称：基于Linux的TCP网络聊天室

2.实验目的：通过TCP完成多用户群聊和私聊功能。

3.实验过程：

         通过socket建立用户连接并传送用户输入的信息，分别来写客户端和服务器端，利用多线程来实现多用户模式，服务器端随时准备接收客户端发送的消息，并判断该消息类型（私聊或群聊）来进行对应的转发工作，客户端随时接受来自服务器端的消息，从而实现消息的同步。

（1）开启服务器。

 

（2）开启客户端，输入用户昵称，客户端开始与服务器建立连接。

 

（3）群聊功能，一名用户发送消息，聊天室的其他成员均可收到此消息。

 

 

（4）私聊功能，一位用户@其他用户，则此消息只有被@的用户可以收到。

 

 

用户的信息借助结构体来存放，群聊功能通过遍历结构体，把消息发送给每一名用户，私聊功能通过对用户所发消息进行判断，若消息中存在@其他用户则把该消息私发给指定用户。

另外，私聊功能函数可以进行对各种@操作的判断，如用户所@的人不存在，则把此消息作为群消息发出，@其他用户后没有添加空格，则把此消息视为群发消息。

 

源码：

头文件

//header.h
#ifndef _A_H
#define _A_H
#include <stdio.h>
#include <pthread.h>
#include <sys/socket.h> 
#include <netinet/in.h>
#include <arpa/inet.h>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>
#include <signal.h>
#include <pthread.h>
struct client{
    char name[30];    //客户端名字 
    int fds;        //客户端socket描述符 
};
struct client c[100]={0};//接收客户端100个 
int in;            //数组下标 
char *IP ="127.0.0.1"; //ip
short PORT=10222;
char NAME[20]={};    //@的名字 
int FLAG=0;        //标记是群发还是私聊 
#endif

服务器端

//tcp_s.c
#include "header.h"
int sockfd;            //服务器socket 
char tempName[20]={};
//初始化服务器网络
void init(){
    printf("start chat room...");
    sockfd = socket(PF_INET,SOCK_STREAM,0);
    if(sockfd == -1){
        perror("create socket false\n");
        exit(-1);
    }
    struct sockaddr_in addr;    //网络通信地址结构 
    addr.sin_family = PF_INET;    //协议簇 
    addr.sin_port = htons(PORT);    //端口
    addr.sin_addr.s_addr = inet_addr(IP);    //IP地址
    if(bind(sockfd,(struct sockaddr*)&addr,sizeof(addr)) == -1){
        perror("bind false\n");
        exit(-1);
    } 
    if(listen(sockfd,100) == -1){
        perror("Setup listening failed\n");
        exit(-1); 
    }
    printf("Server initialization succeeded!\n");
} 

分发消息

void SendMessage(char *msg){
    int flag=0;        //标记@符号是否为私聊 
    int exit=0;        //比较@的用户是否存在 
    int i=0,j=0;
    while(msg[i]!=':'){
        if(i>=strlen(msg))
            break;
        i++;
    }                        //找到@符号位置 
    if(msg[i+1] == '@'){
        i++;
        while(msg[i]!=32){
            if(i>=strlen(msg)){        //@符号不是私聊作用 
                flag=1;
                break;
            }                
            NAME[j]=msg[i+1];
            i++;
            j++;
        }
        if(flag){        //群发 
            for(i=0;i<in;i++){
                printf("send to %d\n",c[i].fds);
                send(c[i].fds,msg,strlen(msg),0);
            }        
        }
        else{            //私发 
            for(i=0;i<in;i++){
                int k=0;
                for(k=0;k<strlen(NAME)-1;k++){
                    tempName[k]=NAME[k];
                }                
                if(strcmp(c[i].name,tempName)==0){                
                    send(c[i].fds,msg,strlen(msg),0);
                    exit=1;  
                    break;    
                }        
            }
        }
    } 
    else{ 
        exit = 1;
        for(i=0;i<in;i++){
            printf("send to %d\n",c[i].fds);
            send(c[i].fds,msg,strlen(msg),0);
        }
    }    
    if(exit == 0){
        for(i=0;i<in;i++){
            printf("send to %d\n",c[i].fds);
            send(c[i].fds,msg,strlen(msg),0);
        }
    }    
}

线程函数中进行通信/接收客户端消息，分发给所有客户端 

void *server_thread(void *p){
    char name[30]={};
    if(recv(c[in].fds,name,sizeof(name),0)>0){
        name[strlen(name)]='\0';
        strcpy(c[in].name,name);
    }
    in++;
    char tip[100]={}; 
    sprintf(tip,"%s join in the chat room\n",c[in-1].name);
    SendMessage(tip);
    int fd = *(int*)p;
    printf("pthread = %d\n",fd);
    while(1){
        char buf[100]={};
        if(recv(fd,buf,sizeof(buf),0) == 0){
            //表示退出连接 
            printf("fd = %d left the chat room\n",fd);
            int i,j;
            char name[20]={};
            int flag = 1;
            for(i=0;i<in;i++){
                if(c[i].fds == fd){
                    strcpy(name,c[i].name);
                    i++;
                    flag = 0;
                }
                if(flag != 1){
                    c[i-1].fds = c[i].fds;
                    strcpy(c[i-1].name,c[i].name);
                }
            }
            c[i].fds = 0;
            strcpy(c[i].name,"");
            in--;
            char msg[100];
            sprintf(msg,"%s left the chat room\n",name);
            SendMessage(msg);
            close(fd);
            break;
        }
        SendMessage(buf); 
    }
}

等待客户端连接，启动服务器的服务

void server(){
    printf("Server starts service!\n");
    while(1){
        struct sockaddr_in fromaddr;    //存储客户端通信地址 
        socklen_t len = sizeof(fromaddr);
        int fd = accept(sockfd,(struct sockaddr*)&fromaddr,&len);
        if(fd == -1){
            perror("Client connection failed!\n");
            continue;            
        }
        c[in].fds = fd;
        pthread_t pid;
        pthread_create(&pid,0,server_thread,&fd);
    }    
} 
void sig_close(){
    close(sockfd);
    printf("Server close\n");
    exit(0);
}
int main()
{
    signal(SIGINT,sig_close);
    init();
    server();
    return 0;
}

客户端

//tcp_c.c
#include "header.h" 
char name[30];   //name
int sockfd;        //客户端socket 
int mutax=0;    //互斥变量 
void init(){
    printf("Client starts\n");
    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,(struct sockaddr*)&addr,sizeof (addr))==-1){
        perror("connect failed\n");
        exit(-1);
    }
}  

通信

void start(){
    pthread_t pid;
    void* receive_thread(void*);
    pthread_create(&pid,0,receive_thread,0);
    while(1){
        char buf[100]={};
        gets(buf); 
        char msg[100]={};
        if(mutax){
            mutax=0;
            continue;
        }
        sprintf(msg,"%s said:%s",name,buf);
        send(sockfd,msg,strlen(msg),0);  
}

}
void* receive_thread(void *p){
    while(1){
        char buf[100]={};
        if(recv(sockfd,buf,sizeof(buf),0)<=0){
            break;
        }
        printf("%s\n",buf);
        }
}
void sig_close(){
    close(sockfd);
    exit(0);
}
int main(){
    signal(SIGINT,sig_close);
    printf("Please input your name:");
    scanf("%s",name);
    mutax=1;
    init();
    if(mutax)
        send(sockfd,name,strlen(name),0);
    start();
    return 0;
}