c++ 網路編程（二）TCP/IP linux 下多進程socket通訊 多個用戶端與單個服務端互動代碼實現回聲伺服器

標籤：ons   用戶端   %s   pid_t   編程   相互   檔案描述符   add   執行   

 

原文aircraft

原文連結：https://www.cnblogs.com/DOMLX/p/9612820.html

 

 

LINUX下：

 

一.服務端代碼

 

下面用了多個close來關閉檔案描述符，可能有的小夥伴會有疑惑。。。。我就說一句，建立進程的時候會把父進程的資源都複製 一份，而你這個子進程只需要保留自己需要處理的資源，其他的自然要關閉掉，

不然父親一個兒子一個 待會打起來怎麼辦  嘿嘿

 

注意了：就像進程間的通訊需要屬於作業系統的資源管道來進行，通訊端也屬於作業系統，所以建立新進程也還是只有原來的那個，複製的資源只不過是檔案描述符而已，我們關閉的也是這個檔案描述符

//基於多進程的並發伺服器實現//註：子進程會複製父進程擁有的所有資源#include <stdio.h>#include <stdlib.h>#include <string.h>#include <unistd.h>#include <arpa/inet.h>#include <sys/socket.h>#include <signal.h>#include <sys/wait.h> #define BUF_SIZE 30 void error_handling(char *message);void read_childproc(int sig); int main(int argc, char *argv[]){  //定義TCP串連變數  int serv_sock;  int clnt_sock;  struct sockaddr_in serv_addr;  struct sockaddr_in clnt_addr;  socklen_t clnt_addr_size;  int str_len;  char buf[BUF_SIZE];  //定義訊號處理變數  pid_t pid;  struct sigaction act;  int state;    if(argc!=2)  {    exit(1);  }  //配置訊號處理函數  act.sa_handler=read_childproc;  sigemptyset(&act.sa_mask);  act.sa_flags=0;  sigaction(SIGCHLD, &act, 0);    //TCP通訊端配置  serv_sock=socket(PF_INET, SOCK_STREAM, 0);  if(serv_sock == -1)    error_handling("socket error!");   memset(&serv_addr, 0, sizeof(serv_addr));  serv_addr.sin_family = AF_INET;    //IPV4協議族  serv_addr.sin_addr.s_addr = htonl(INADDR_ANY);    //主機位元組序（host）轉換成網路位元組序（net）（大端序）  serv_addr.sin_port = htons(atoi(argv[1]));    //連接埠號碼  if(bind(serv_sock, (struct sockaddr*) &serv_addr, sizeof(serv_addr)) == -1)    error_handling("bind error");   if(listen(serv_sock, 5) == -1)    error_handling("listen error");    while(1)  {    clnt_addr_size = sizeof(clnt_addr);    clnt_sock=accept(serv_sock, (struct sockaddr*) &clnt_addr, &clnt_addr_size);    if(clnt_sock == -1)      continue;    else      puts("new client connected...");        pid=fork();    //建立新進程    if(pid==-1)    {      close(clnt_sock);      continue;    }    if(pid==0)    //子進程運列區域    {      close(serv_sock);    //在子進程中要關閉伺服器通訊端檔案描述符      while((str_len=read(clnt_sock, buf, BUF_SIZE))!=0)    write(clnt_sock, buf, str_len);            close(clnt_sock);//執行完關閉自己的檔案描述符      puts("client disconnected...");      return 0;    }    else    //父進程運列區域    {      //調用fork函數後，要將無關的通訊端檔案描述符關閉掉      close(clnt_sock);    }  }    close(serv_sock);  return 0;} void error_handling(char *message){  fputs(message, stderr);  fputc(‘\n‘, stderr);  exit(1);} //一旦有子進程結束就調用這個函數void read_childproc(int sig){  int status;  pid_t id=waitpid(-1, &status, WNOHANG);//等待子進程終止  if(WIFEXITED(status))  {    printf("remove proc id: %d \n", id);    printf("child send: %d \n", WEXITSTATUS(status));  }}

 

二.用戶端代碼

這裡說一下這裡用多進程分割I/O（輸入/輸出）,是為了代碼的分割提高程式最佳化，在輸入資料的時候不需要考慮輸出，在一個地方不用寫兩個地方的代碼，雖然代碼可能變多了，但是程式確實最佳化了，老經驗的程式員就能體會到了

 

然後為什麼write_routine裡還要調用shutdown給服務端傳輸EOF，MAIN函數最後不是有close可以向服務端發送嗎？？？  這是因為我們建立了子進程，沒有辦法通過一次調用close傳遞EOF，不然會出大問題的！！  所以自己在子進程裏手工調用shutdown發送EOF，告訴服務端：“嘿哥們，我差不多要涼涼了      下輩子有緣再見吧QAQ”  哈哈哈哈哈哈哈

//分割IO實現分割資料的收發過程//父進程負責接收，子進程負責發送#include <stdio.h>#include <stdlib.h>#include <string.h>#include <unistd.h>#include <arpa/inet.h>#include <sys/socket.h> #define BUF_SIZE 30 void error_handling(char *message);void read_routine(int sock, char *buf);void write_routine(int sock, char *buf); int main(int argc, char *argv[]){  int sock;  pid_t pid;  struct sockaddr_in serv_addr;  int str_len;  char buf[BUF_SIZE];    if(argc!=3)  {    exit(1);  }    sock=socket(PF_INET, SOCK_STREAM, 0);  if(sock == -1)    error_handling("socket error!");    memset(&serv_addr, 0, sizeof(serv_addr));  serv_addr.sin_family = AF_INET;  serv_addr.sin_addr.s_addr = inet_addr(argv[1]);  serv_addr.sin_port = htons(atoi(argv[2]));   if(connect(sock, (struct sockaddr*) &serv_addr, sizeof(serv_addr)) == -1)    error_handling("connect error");  else    puts("Connected.....");    pid=fork();  if(pid==0)    //子進程寫    write_routine(sock, buf);  else    //父進程讀    read_routine(sock, buf);    close(sock);  return 0;} void error_handling(char *message){  fputs(message, stderr);  fputc(‘\n‘, stderr);  exit(1);} //這裡實現讀入有關代碼void read_routine(int sock, char* buf){  while(1)  {    int str_len=read(sock, buf, BUF_SIZE);    if(str_len==0)      return;    //接受到EOF結束符時返回        buf[str_len]=0;    printf("message from server: %s \n", buf);  }} //這裡負責實現輸出有關代碼void write_routine(int sock, char* buf){  while(1)  {    fgets(buf, BUF_SIZE, stdin);    if(!strcmp(buf, "q\n") || !strcmp(buf, "Q\n"))    {      shutdown(sock, SHUT_WR);      return;    }    write(sock, buf, strlen(buf));  }}

 同時多進程服務端也是有缺點的，每建立一個進程就代表大量的運算與記憶體空間佔用，相互進程資料交換也很麻煩。。。那麼怎麼解決呢，在我後面的部落格也許會給出答案-----hhhhhhh

 

參考書籍《TCP/IP網路編程---尹聖雨》

c++ 網路編程（二）TCP/IP linux 下多進程socket通訊 多個用戶端與單個服務端互動代碼實現回聲伺服器