TCP通信多线程网络聊天程序

程序功能

要求：编写一个网络聊天程序，要求采用数据流的套接口编程。程序分为服务器端与客户端。服务器端最大同时连接10个客户端。
服务器端可以响应多个客户端的请求，每个客户端之间可以相互通信，由服务器实现转发。服务器端显示所有客户端的通信，并有脏话过滤功能（比如用户输入“Fuck”，过滤后为“**”）。客户端通过用户名实现不同用户间通信(发送消息格式“用户名” “消息内容””)。

多程线网络聊天程序视频

程序流程图

程序源代码

服务端

#include <stdlib.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <unistd.h>
#include <pthread.h>
#include <stdio.h>
#include <string.h>
#include <arpa/inet.h>

#define PORT 8848
#define BACKLOG 10
#define MAXDATASIZE 128
#define MAXUSER 9

void fuck_filter(const char* buf);
int select_user(char uname);
void process_cli(int connectfd, struct sockaddr_in client);
void* creat_conn(void* arg);

struct ARG {
       int connfd;
       struct sockaddr_in client;
};

struct USER {
       int connfd;
       char name;
}user[MAXUSER+1];

static int u_len = 0;

///////////////////////////////// 

int main(void)
{
      int listenfd, connectfd;
      pthread_t thread;         //id of thread
      struct ARG *arg;
      struct sockaddr_in server; //server's address info
      struct sockaddr_in client; //client's
      int sin_size;
      int opt;

      //create tcp socket
      printf("socket.... ");
      if ((listenfd = socket(AF_INET, SOCK_STREAM, 0)) == -1){
          perror("creating socket failed.");
          exit(1);
      }

      opt = SO_REUSEADDR;
      setsockopt(listenfd, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof(opt));      
      /*setsockopt()用来设置参数s所指定的socket状态。
      参数level代表欲设置的网络层，一般设成SOL_SOCKET以存取socket层。
      参数optname代表欲设置的选项，有下列几种数值:SO_REUSEADDR 允许在bind（）过程中本地地址可重复使用
        参数  optval代表欲设置的值，参数optlen则为optval的长度。
      返回值  成功则返回0，若有错误则返回-1，错误原因存于errno。*/ 

      bzero(&server,sizeof(server));
      server.sin_family = AF_INET;
      server.sin_port = htons(PORT);
      server.sin_addr.s_addr = htonl(INADDR_ANY);
      printf("bind.... ");

      if(bind(listenfd,(struct sockaddr *)&server,sizeof(struct sockaddr)) == -1){
          perror("bind error.");
          exit(1);
      }  

      printf("listen.... ");
      if(listen(listenfd,BACKLOG) == -1) {
          perror("listen() error ");
          exit(1);
      }
      sin_size = sizeof(struct sockaddr_in);
      sleep(1);
      while(1){
          //accept() using main thread
          printf("accepting....\n ");
          if((connectfd = accept(listenfd,(struct sockaddr *)&client,(socklen_t*)&sin_size)) == -1) {
              perror("accept() error ");
              exit(1);
          }
          //超过系统最大用户容量的处理???
          printf("cread ptheard arg\n");    
          arg =  malloc(sizeof(struct ARG));
          arg->connfd = connectfd;
          memcpy((void *)&arg->client, &client, sizeof(client));

          //invoke start_routine to handle this thread

          if(pthread_create(&thread, NULL, creat_conn, (void*)arg)){
              perror("pthread_creat() error");
              exit(1);
          }          
      }
      close(listenfd);    
      return (0);
}


////////////////////////// 

void process_cli(int connectfd, struct sockaddr_in client)
{

    int num,ufd,i;
    int tid = pthread_self();
    char recvbuf[MAXDATASIZE], sendbuf[MAXDATASIZE];
    char cli_name[MAXDATASIZE];
    char name;

    printf("IP: %s. \n",inet_ntoa(client.sin_addr) );
    //get client's name from client

    user[u_len].connfd = connectfd;
    user[u_len].name = 'a' + u_len;

    printf("u_fd:%d, u_name:%c\n",user[u_len].connfd,user[u_len].name);
    //******???
    if(u_len++ > MAXUSER) u_len = 0;    

    while(1){
        memset(recvbuf,0,MAXDATASIZE);  
        num = recv(connectfd, recvbuf, MAXDATASIZE, 0);
        printf("num=%d\n",num);
        if(num == 0 || num == -1) {
            close(connectfd);
            perror("Client disconnected. ");
            break;
        }
        printf("[%d]INFO: %s \n", tid, recvbuf );
        //脏话过滤
        fuck_filter(recvbuf);
        //选择用户
        name = recvbuf[0];
        //群发处理
        if(name == 'x')
        {
            for(i=0;i<= MAXUSER;i++)
            {       
                if(user[i].connfd>0)
                    send(user[i].connfd, recvbuf, strlen(recvbuf), 0);              
            }           
        }

        ufd = select_user(name);//得到用户的socket 
        if(ufd > 0){
            //通过ufd,向指定用户发送
            send(ufd, recvbuf, strlen(recvbuf), 0);
        }
     }
}

////////////////////////////// 
int select_user(char uname)
{
    int ufd = 0;
    int i;
    for(i=0;i<= MAXUSER;i++)
    {

        if(uname==user[i].name)
            return user[i].connfd;

    }

    return ufd;
}




///////////////////////////// 

void* creat_conn(void* arg)
{
    int tid = pthread_self();
    struct ARG * n_arg = arg;
//  printf("\nTid[%d]", tid);
    process_cli(n_arg->connfd, n_arg->client);
}

/////////////////////////// 
void fuck_filter(const char* buf)
{
    char buf_fuck[] = "fuck";
    char * p = NULL;
    do{
        p = strstr(buf,buf_fuck);
//      printf("==%d==",p);
        if(p != 0){
            *p = '*'; 
            *(++p) = '*';
            *(++p) = '*';
            *(++p) = '*';
        }
    }while(p != 0);
}

客户端

/* cthread.c */
#include <stdio.h>
#include <sys/types.h>
#include <netinet/in.h>
#include <sys/socket.h>
#include <netdb.h>
#include <unistd.h>
#include <string.h>
#include <arpa/inet.h>
#include <errno.h>
#include <stdlib.h>
#include <pthread.h>

#define PORT 8848
#define MAXDATASIZE 128

void* recv_proc(void* arg);
void* send_proc(void* arg);
int get_line(char *msg);


int main(int argc, char *argv[])
{
    int fd;
    pthread_t th1,th2;
    struct hostent *he;
    struct sockaddr_in server; //server's address info
    char *defaultIP = "127.0.0.1";
    //???
    if(argc != 2) {
        if((he = gethostbyname(defaultIP)) == NULL){
            perror("gethostbyname() error");
            exit(1);
        }
    }else{
       if((he = gethostbyname(argv[1])) == NULL){
            perror("gethostbyname() error");
            exit(1);
            }
    }

    if((fd = socket(AF_INET, SOCK_STREAM, 0)) == -1){
        perror("socket() error");
        exit(1);
    }

    bzero(&server , sizeof(server));
    server.sin_family = AF_INET;
    server.sin_port = htons(PORT);
    server.sin_addr = *((struct in_addr *)he->h_addr);

    if(connect(fd, (struct sockaddr *)&server,sizeof(struct sockaddr)) == -1){
        perror("connect() error");
        exit(1);
    }               

    if(pthread_create(&th1, NULL, recv_proc, (void*)&fd)){
        perror("recv pthread_creat() error");
        close(fd);      
        exit(1);
    }   

    if(pthread_create(&th2, NULL, send_proc, (void*)&fd)){
        perror("send pthread_creat() error");
        close(fd);          
        exit(1);
    }   
    pthread_join(th1,NULL);
    pthread_join(th2,NULL);    
    return 0;

}


void* recv_proc(void* arg)
{
    char recvline[MAXDATASIZE];
    int numbytes;
    int sockfd;

    printf("recv from server. \n");
    sockfd = *((int*)arg);  
    while(1){
        if((numbytes = recv(sockfd, recvline, MAXDATASIZE, 0)) > 0){       
            recvline[numbytes] = '\0';
            printf("\n[recvice:] %s\n", recvline);
            //clean output buffer
            memset(recvline, '0', strlen(recvline)-1);
        }else{
            //printf("recv error/null. ");          
        }           
    }   

}

void* send_proc(void* arg)
{
    char sendline[MAXDATASIZE];
    int numbytes;
    int len;
    int sockfd;

    printf("send to server. \n");   
    sockfd = *((int*)arg);

    while(1)
    {
        //send message to server
        //when the string is not NULL , send another!
        while((len =get_line(sendline)) != 0){
            sendline[len] = '\0';
            send(sockfd, sendline, strlen(sendline), 0);
        }
    }

}

int get_line(char *msg)
{
    int i=0;
    char temp;

    printf("Input message:");   
    fflush(stdout);
    while (1){ 
        temp = getchar();
        if (temp == '\r' || temp == '\n') {return i ;}  
        msg[i]=temp;
        if(msg[i]==13){
                msg[i]=0;
                break;
        }
        fflush(stdout);
        i++;
    }
}