Linux下基於socket多線程並發通信的實現

pthread_server.c

/***************************************************  
* 文件名：pthread_server.c  
* 文件描述：創建子線程來接收客戶端的數據  
***************************************************/
#include <sys/types.h>    
#include <sys/socket.h>    
#include <stdio.h>    
#include <netinet/in.h>    
#include <arpa/inet.h>    
#include <unistd.h>    
#include <stdlib.h>    
#include <pthread.h>    
void *rec_data(void *fd);    
int main(int argc,char *argv[])    
{    
       int server_sockfd;    
    int *client_sockfd;    
       int server_len, client_len;    
       struct sockaddr_in server_address;    
       struct sockaddr_in client_address;    
       struct sockaddr_in tempaddr;    
       int i,byte;    
       char char_recv,char_send;    
       socklen_t templen;    
       server_sockfd = socket(AF_INET, SOCK_STREAM, 0);//創建套接字    
           
       server_address.sin_family = AF_INET;    
       server_address.sin_addr.s_addr =  htonl(INADDR_ANY);    
       server_address.sin_port = htons(9734);    
       server_len = sizeof(server_address);    
                
       bind(server_sockfd, (struct sockaddr *)&server_address, server_len);//綁定套接字    
       templen = sizeof(struct sockaddr);    
           
       printf("server waiting for connect/n");    
       while(1){    
              pthread_t thread;//創建不同的子線程以區別不同的客戶端    
              client_sockfd = (int *)malloc(sizeof(int));    
              client_len = sizeof(client_address);    
              *client_sockfd = accept(server_sockfd,(struct sockaddr *)&client_address, (socklen_t *)&client_len);    
              if(-1==*client_sockfd){    
                     perror("accept");    
                     continue;    
              }    
              if(pthread_create(&thread, NULL, rec_data, client_sockfd)!=0)//創建子線程    
              {    
                     perror("pthread_create");    
                     break;    
              }    
       }    
       shutdown(*client_sockfd,2);    
       shutdown(server_sockfd,2);    
}    
/*****************************************  
* 函數名稱：rec_data  
* 功能描述：接受客戶端的數據  
* 參數列表：fd——連接套接字  
* 返回結果：void  
*****************************************/
void *rec_data(void *fd)    
{    
       int client_sockfd;    
       int i,byte;    
       char char_recv[100];//存放數據    
       client_sockfd=*((int*)fd);    
       for(;;)    
       {    
              if((byte=recv(client_sockfd,char_recv,100,0))==-1)    
              {    
                     perror("recv");    
                     exit(EXIT_FAILURE);     
              }    
              if(strcmp(char_recv, "exit")==0)//接受到exit時，跳出循環    
                     break;    
              printf("receive from client is %s/n",char_recv);//打印收到的數據    
       }    
       free(fd);    
       close(client_sockfd);    
       pthread_exit(NULL);    
}

pthread_client.c

/***************************************************  
* 文件名：pthread_client.c  
* 文件描述：創建子線程來接收客戶端的數據  
***************************************************/
#include <sys/types.h>    
#include <sys/socket.h>    
#include <stdio.h>    
#include <netinet/in.h>    
#include <arpa/inet.h>    
#include <unistd.h>    
#include <stdlib.h>    
int main(int argc,char *argv[])    
{    
       int sockfd;    
       int len;    
       struct sockaddr_in address;       
       int result;    
       int i,byte;    
       char char_send[100] = { 0 };    
    if((sockfd = socket(AF_INET, SOCK_STREAM, 0))==-1)    
       {    
              perror("socket");    
              exit(EXIT_FAILURE);    
       }    
    if(argc != 3){    
      printf("Usage: fileclient <address> <port>/n");//用法：文件名 服務器IP地址 服務器端口地址    
      return 0;    
   }    
   if((sockfd = socket(AF_INET,SOCK_STREAM,0)) == -1){    
       perror("sock");    
       exit(1);    
   }    
   bzero(&address,sizeof(address));    
   address.sin_family = AF_INET;    
   address.sin_port = htons(atoi(argv[2]));    
   inet_pton(AF_INET,argv[1],&address.sin_addr);    
len = sizeof(address);    
           
    if((result = connect(sockfd, (struct sockaddr *)&address, len))==-1)    
       {    
              perror("connect");    
              exit(EXIT_FAILURE);    
       }    
           
       for(;;)    
       {    
             scanf("%s", char_send);//輸入發送數據    
              fflush(stdin);//清除輸入緩存    
              if(strcmp(char_send, "exit")==0){//如果輸入exit，跳出循環    
                     if((byte=send(sockfd,char_send,100,0))==-1)    
                     {    
                            perror("send");    
                            exit(EXIT_FAILURE);    
                     }               
                     break;    
              }                      
              if((byte=send(sockfd,char_send,100,0))==-1)    
              {    
                     perror("send");    
                     exit(EXIT_FAILURE);    
              }    
       }    
    close(sockfd);    
    exit(0);    
}

編譯服務器端程序 pthread_server.c ：

$gcc pthread_server.c –o server –lpthread

編譯客戶端程序 pthread_client.c:

$gcc pthread_client.c –o client

編譯在開發板上跑的客戶端程序：

$arm-linux-gcc client.c –o arm_client

先啟動服務器端的程序 server:

$./server

打開另一個終端，啟動客戶端的程序 client:

$./client 127.0.0.1 9734

把 arm_client 放到開發板上，啟動 arm_client:

$./arm_client 219.222.170.9 9734

結果 :

服務器端：

tongs@tong's-desktop:~/c_c++_program/sock_inet_comm2$ ./server

server waiting for connect

receive from client is client

receive from client is client

receive from client is arm_client

receive from client is arm_client

客戶端：

tongs@tong's-desktop:~/c_c++_program/sock_inet_comm2$ ./client 127.0.0.1 9734

client

client

exit