ECHO网络程序的演变史 (7) --- TCP多线程聊天系统

Andrew Huang 转载请注明作者及联络方式

一.聊天实现原理
--------------------------------------------------------------

二.实现的代码
--------------------------------------------------------------

聊天服务器代码 tcp_chat_svr.c
 

/*
  TCP MultiThread Chat Server
 
 Author :Andrew Huang

 */

#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>

#include <pthread.h>



#include <stdio.h>
#include <string.h>

#define MAX_USER 60

static pthread_mutex_t chat_lock;
static int user_list[MAX_USER]; //以client_fd 为下标，如果对应socket在线，值为1,下线为0



static void * thread_fn(void *arg)
{
   int client_fd = (int)arg;
   int len,fd;
        char buf[1024];

     fprintf(stdout,"thread %d begin\n",client_fd);

     while(1)
        {
 
               len = recv(client_fd,buf,sizeof(buf),0);
                  if(len <= 0)
                            break;
                    
                 fprintf(stdout,"recv client:%s\n",buf);

                   //对所有在线的socket进行轮播

                 pthread_mutex_lock(&chat_lock);
                    for(fd=3; fd< MAX_USER;fd++)
                         {
                         if((user_list[fd] == 0) || (fd == client_fd))
                                      continue;

                             if(send(fd,buf,len,0)<0)
                                       continue;
                          }
                 pthread_mutex_unlock(&chat_lock);

         
       }

               pthread_mutex_lock(&chat_lock);
                  user_list[client_fd] = 0;
               pthread_mutex_unlock(&chat_lock);

                close(client_fd);
                 fprintf(stdout,"CLOSE socket %d\n",client_fd);
}


int main(int argc ,char * argv[])
{
  unsigned short port = 9000;
   
   int listen_fd ;
   int client_fd;
   pthread_t pth;

   char * welcome_msg = "Welcome to Chat Server ,ver 0.1";
    char * exit_msg = "user to max!";

   struct sockaddr_in addr; //ipv4的地址结构

   struct sockaddr_in client_addr ;
   socklen_t addr_len;


   int len ;
  
   if(argc >1)
     {
         port = (unsigned short)atoi(argv[1]);
     }

  //第一步：创建一个IPV4,TCP，可以运行所有协议的SOCKET

  listen_fd = socket(PF_INET,SOCK_STREAM,0);
  if(listen_fd == -1)
   {
        perror("socket");
         return -1;
   }


    {
         int val = 1;
         setsockopt(listen_fd, SOL_SOCKET, SO_REUSEADDR, (char *) &val, sizeof (val));
    }


   //第二步:用bind来指定端口

   memset(&addr,0,sizeof(addr));
    addr.sin_port = htons(port); //转成网络序的端口

   addr.sin_family = AF_INET; //ipv4的地址类型

   //addr.sin_addr.s_addr = inet_addr("0.0.0.0"); //bind 的sin_addr表示在哪一个网卡上有效,

   //绑定在0.0.0.0上表示这个SOCKET在所有网卡侦听

   addr.sin_addr.s_addr = INADDR_ANY; //INADDR_ANY = 0 = inet_addr("0.0.0.0")


    if(bind(listen_fd,(const struct sockaddr *)&addr,sizeof(addr)) == -1)
       {
          perror("bind");
            close(listen_fd);
          return -2;
       }

   //第三步： 通知内核里的TCP／IP协议栈，让这个SOCKET开始侦听网络上的请求

   // listen只是通知一下，通知成功立即执行后面代码

     if(listen(listen_fd,20) == -1)
       {
             perror("listen");
        close(listen_fd);
             return -3;
       }


      fprintf(stdout,"TCP Chat Server listen on %d\n",port);

      //初始化锁

     pthread_mutex_init(&chat_lock,NULL);
      memset(user_list,0,sizeof(user_list));
   
      
     while(1)
       {
           memset(&client_addr,0,sizeof(client_addr));
           client_addr.sin_family = AF_INET;
           addr_len = sizeof(client_addr); //必须赋值！


           //第四步：服务器在侦听客户端请求

                printf("accept client ...\n");
           client_fd = accept(listen_fd,(struct sockaddr *)&client_addr,&addr_len);
             if(client_fd == -1)
               {
                  perror("accept");
                     continue;
               }

             fprintf(stdout,"client socket %d,addr %s,port %d\n",client_fd,
                    inet_ntoa(client_addr.sin_addr),ntohs(client_addr.sin_port));

                 if(client_fd > MAX_USER)
                          {
                              send(client_fd,exit_msg,strlen(exit_msg),0);
                              close(client_fd);
                              continue;
                           }

                 if(send(client_fd,welcome_msg,strlen(welcome_msg),0)<=0)
                         {
                            perror("send");
                         }
              //把上线用户加入在线用户表

               pthread_mutex_lock(&chat_lock);
                  
                   printf("user %d online\n",client_fd);
                  user_list[client_fd] = 1;
               pthread_mutex_unlock(&chat_lock);

              

              pthread_create(&pth,NULL,thread_fn,(void*)client_fd);

          
       }

     printf("exit");
      close(listen_fd);

}


客户端代码 tcp_chat_cli.c
 

/*
  TCP MultiThread Chat Client
 
 Author :Andrew Huang

 */


#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <pthread.h>

#include <stdio.h>
#include <string.h>

static int run = 1;

void * thread_fn(void * arg)
{
   int sock = (int)arg;
   char buf[1024];
    int len;

   while(run)
   {
       len = recv(sock,buf,sizeof(buf),0);
           if(len <=0)
                 break;
             
          buf[len] = 0;
          printf("SERVER:%s\n",buf);
             
   }

    printf("close thread\n");
   //close(sock);

}


int main(int argc ,char * argv[])
{
   unsigned short port = 9000;
   
   int sock ;
    char ipaddr[32] = "127.0.0.1";
    char buf[32] = "hello server";
    struct sockaddr_in serv_addr;
     pthread_t pth;
    int len;

     if(argc >1)
     {
         port = (unsigned short)atoi(argv[1]);
     }

    if(argc >2)
     {
          memset(ipaddr,0,sizeof(ipaddr));
         strncpy(ipaddr,argv[2],sizeof(ipaddr)-1);
     }

    if(argc > 3)
       {
            memset(buf,0,sizeof(buf));
         strncpy(buf,argv[3],sizeof(buf)-1);
        }


   printf("connect server %s:%d,buf %s\n",ipaddr,port,buf);

  //第一步：创建一个IPV4,TCP，可以运行所有协议的SOCKET

   sock = socket(PF_INET,SOCK_STREAM,0);
   if(sock == -1)
      {
          perror("socket");
          return -1;
      }


    //第二步:对于客户端,port可用任意一个编号,如果不使用bind，协议栈会自动分配没用的端口0

    // 省略bind ....


   //第三步：用connect()去联接服务器

    memset(&serv_addr,0,sizeof(serv_addr));
    serv_addr.sin_family = AF_INET;
    serv_addr.sin_port = htons(port);
    serv_addr.sin_addr.s_addr = inet_addr(ipaddr);
 
     //printf("ipaddr %s,s_addr %x\n",ipaddr,ntohl( serv_addr.sin_addr.s_addr));


     if(connect(sock,(struct sockaddr *)&serv_addr,sizeof(serv_addr)) == -1)
        {
            perror("connect");
              close(sock);
              return -2;
        }

      pthread_create(&pth,NULL,thread_fn,(void *)sock);

      while(1)
        {
           
          //键盘输入

         if(fgets(buf,sizeof(buf)-1,stdin)== NULL)
                 break;

           //如果直接输入回车则退出

          if(buf[0]== '\n')
                {
                   run = 0;
                   break;
                 }

           

        if(send(sock,buf,strlen(buf)+1,0)<= 0)
           {
              break;
           }

    

          }
              //关闭socket

            close(sock);
           
   
}