socket編程——一個簡單的例子

從一個簡單的使用TCP例子開始socket編程,其基本步驟如下:

server                                                  client

 

+++++++                                          ++++++++

創建socket                                          創建socket

+++++++                                          ++++++++

      |                                                         |

      |                                                         |

      |                                                         |

+++++++                                          ++++++++

地址賦值(                                           地址賦值(

自己的地址)                                        服務器地址)

+++++++                                          ++++++++

      |                                                         |

      |                                                         |

      |                                                         |

++++++++                                              |

用bind綁定                                                |

socket和地址                                             |

++++++++                                              |

      |                                                         |

      |                                                         |

      |                                                         |

+++++++                                                 |

listen                                                         |

+++++++                                                  |

      |                                                    ++++++++++

      |   <------------------------------          connect 服務器         

      |                                                    ++++++++++

+++++++                                                  |

accept                                                        |

+++++++                                                  |

      |                                                           |

      |                                                    +++++++++

      |                                                     recv 和send

      |                                                     進行數據處理

      |                                                     +++++++++

+++++++++                                                |

用accept得到                                                 |

的socket進行                                                 |

recv 和 send                                                 |

+++++++++                                                |

      |                                                             |

      |                                                             |

      |                                                             |

+++++++++                                        +++++++++

close socket                                         close socket

+++++++++                                        +++++++++

 

根據以上步驟,服務器端的代碼爲

  1. #include <stdio.h>  
  2. #include <string.h>  
  3. #include <sys/socket.h>  
  4. #include <netinet/in.h>  
  5. #include <stdlib.h>  
  6. #include <syslog.h>  
  7. #include <errno.h>  
  8. #define MAX_LISTEN_NUM 5  
  9. #define SEND_BUF_SIZE 100  
  10. #define RECV_BUF_SIZE 100  
  11. #define LISTEN_PORT 1010  
  12. int main()  
  13. {  
  14.   int listen_sock = 0;  
  15.   int app_sock = 0;  
  16.   struct sockaddr_in hostaddr;  
  17.   struct sockaddr_in clientaddr;  
  18.   int socklen = sizeof(clientaddr);  
  19.   char sendbuf[SEND_BUF_SIZE] = {0};  
  20.   char recvbuf[RECV_BUF_SIZE] = {0};  
  21.   int sendlen = 0;  
  22.   int recvlen = 0;  
  23.   int retlen = 0;  
  24.   int leftlen = 0;  
  25.   char *ptr = NULL;  
  26.   memset((void *)&hostaddr, 0, sizeof(hostaddr));  
  27.   memset((void *)&clientaddr, 0, sizeof(clientaddr));  
  28.   hostaddr.sin_family = AF_INET;  
  29.   hostaddr.sin_port = htons(LISTEN_PORT);  
  30.   hostaddr.sin_addr.s_addr = htonl(INADDR_ANY);  
  31.   listen_sock = socket(AF_INET, SOCK_STREAM, 0);  
  32.   if(listen_sock < 0)  
  33.   {  
  34.       syslog(LOG_ERR, "%s:%d, create socket failed", __FILE__, __LINE__);  
  35.       exit(1);  
  36.   }  
  37.   if(bind(listen_sock, (struct sockaddr *)&hostaddr, sizeof(hostaddr)) < 0)  
  38.   {  
  39.       syslog(LOG_ERR, "%s:%d, bind socket failed", __FILE__, __LINE__);  
  40.       exit(1);  
  41.   }  
  42.   if(listen(listen_sock, MAX_LISTEN_NUM) < 0)  
  43.   {  
  44.       syslog(LOG_ERR, "%s:%d, listen failed", __FILE__, __LINE__);  
  45.       exit(1);  
  46.   }  
  47.   while(1)  
  48.   {  
  49.       app_sock = accept(listen_sock, (struct sockaddr *)&clientaddr, &socklen);  
  50.       if(app_sock < 0)  
  51.      {  
  52.         syslog(LOG_ERR, "%s:%d, accept failed", __FILE__, __LINE__);  
  53.         exit(1);  
  54.      }  
  55.      sprintf(sendbuf, "welcome %s:%d here!/n", inet_ntoa(clientaddr.sin_addr.s_addr), clientaddr.sin_port);  
  56.      //send data  
  57.      sendlen = strlen(sendbuf) +1;  
  58.      retlen = 0;  
  59.      leftlen = sendlen;  
  60.      ptr = sendbuf;  
  61.      //while(leftlen)  
  62.      {  
  63.          retlen = send(app_sock, ptr, sendlen, 0);  
  64.       if(retlen < 0)  
  65.       {  
  66.           if(errno == EINTR)  
  67.             retlen = 0;  
  68.         else  
  69.             exit(1);  
  70.       }  
  71.       leftlen -= retlen;  
  72.       ptr += retlen;  
  73.      }  
  74.      //receive data  
  75.      recvlen = 0;  
  76.      retlen = 0;  
  77.      ptr = recvbuf;  
  78.      leftlen = RECV_BUF_SIZE -1;  
  79.      //do  
  80.      {  
  81.          retlen = recv(app_sock, ptr, leftlen, 0) ;  
  82.       if(retlen < 0)  
  83.       {  
  84.           if(errno == EINTR)  
  85.             retlen = 0;  
  86.         else  
  87.             exit(1);  
  88.       }  
  89.       recvlen += retlen;  
  90.       leftlen -= retlen;  
  91.       ptr += retlen;  
  92.      }  
  93.      //while(recvlen && leftlen);  
  94.      printf("receive data is : %s", recvbuf);  
  95.     close(app_sock);  
  96.   }  
  97.   close(listen_sock);  
  98.     
  99.   return 0;  
  100.     
  101.     
  102. }  

 

 

客戶端代碼爲:

  1. #include <stdio.h>  
  2. #include <string.h>  
  3. #include <sys/socket.h>  
  4. #include <netinet/in.h>  
  5. #include <syslog.h>  
  6. #include <errno.h>  
  7. #include <stdlib.h>  
  8. #define MAX_LISTEN_NUM 5  
  9. #define SEND_BUF_SIZE 100  
  10. #define RECV_BUF_SIZE 100  
  11. #define SERVER_PORT 1010  
  12. int main()  
  13. {  
  14.     int sock_fd = 0;  
  15.     char recvbuf[RECV_BUF_SIZE] = {0};  
  16.     char sendbuf[SEND_BUF_SIZE] = {0};  
  17.     int recvlen = 0;  
  18.     int retlen = 0;  
  19.     int sendlen = 0;  
  20.     int leftlen = 0;  
  21.     char *ptr = NULL;  
  22.     struct sockaddr_in ser_addr;  
  23.       
  24.     memset(&ser_addr, 0, sizeof(ser_addr));  
  25.     ser_addr.sin_family = AF_INET;  
  26.     inet_aton("127.0.0.1", (struct in_addr *)&ser_addr.sin_addr);  
  27.     ser_addr.sin_port = htons(SERVER_PORT);  
  28.     sock_fd = socket(AF_INET, SOCK_STREAM, 0);  
  29.     if(sock_fd < 0)  
  30.     {  
  31.         syslog(LOG_ERR, "%s:%d, create socket failed", __FILE__, __LINE__);  
  32.         exit(1);  
  33.     }  
  34.     if(connect(sock_fd, (struct sockaddr *)&ser_addr, sizeof(ser_addr)) < 0)  
  35.     {  
  36.         syslog(LOG_ERR, "%s:%d, connect socket failed", __FILE__, __LINE__);  
  37.         exit(1);  
  38.     }  
  39.      //receive data  
  40.      recvlen = 0;  
  41.      retlen = 0;  
  42.      ptr = recvbuf;  
  43.      leftlen = RECV_BUF_SIZE -1;  
  44.      //do  
  45.      {  
  46.          retlen = recv(sock_fd, ptr, leftlen, 0) ;  
  47.       if(retlen < 0)  
  48.       {  
  49.           if(errno == EINTR)  
  50.             retlen = 0;  
  51.         else  
  52.             exit(1);  
  53.       }  
  54.       recvlen += retlen;  
  55.       leftlen -= retlen;  
  56.       ptr += retlen;  
  57.      }  
  58.      //while(recvlen && leftlen);  
  59.      printf("receive data is : %s", recvbuf);  
  60.      sprintf(sendbuf, "hello server/n");  
  61.      //send data  
  62.      sendlen = strlen(sendbuf) +1;  
  63.      retlen = 0;  
  64.      leftlen = sendlen;  
  65.      ptr = sendbuf;  
  66.     // while(leftlen)  
  67.      {  
  68.          retlen = send(sock_fd, ptr, sendlen, 0);  
  69.       if(retlen < 0)  
  70.       {  
  71.           if(errno == EINTR)  
  72.             retlen = 0;  
  73.         else  
  74.             exit(1);  
  75.       }  
  76.       leftlen -= retlen;  
  77.       ptr += retlen;  
  78.      }  
  79.      close(sock_fd);  
  80.       
  81. }  

 

 

現在一個簡單的使用tcp的socket通信的例子已經完成了,這裏有幾個需要說明的問題

1)頭文件:

sys/socket.h   包含了socket相關的函數,如socket,send 和recv, 以及struct sockaddr等

netinet/in.h    包含了地址結構,如struct sockaddr_in

errno.h           包含了errno 和 EINTR

syslog.h         包含了syslog相關的信息,其打印結果在/var/log/messages裏面

 

2)socket地址

對於IPv4來說,其地址用的是struct sockaddr_in,具體結構如下

  1. struct in_addr {  
  2.   in_addr_t   s_addr;           /* 32-bit IPv4 address */  
  3.                                 /* network byte ordered */  
  4. };  
  5.   
  6. struct sockaddr_in {  
  7.   uint8_t         sin_len;      /* length of structure (16) */  
  8.   sa_family_t     sin_family;   /* AF_INET */  
  9.   in_port_t       sin_port;     /* 16-bit TCP or UDP port number */  
  10.                                 /* network byte ordered */  
  11.   struct in_addr  sin_addr;     /* 32-bit IPv4 address */  
  12.                                 /* network byte ordered */  
  13.   char            sin_zero[8];  /* unused */  
  14. };  
其中sin_len我們一般不關注,也不填(只有在使用routing socket的時候纔用到,被內核用來處理各種協議簇的地址結構)。 bind, connect, sendto, 和 sendmsg會把socket地址從程序傳遞給內核; 而accept, recvfrom, recvmsg, getpeername, 和 getsockname會把地址從內核傳遞給程序。因爲不同協議簇的地址結構是不一樣的,所以必須要有一個通用的指針來傳遞地址,對於ANSI C來說我們一般使用void *,但是socket產生早於ANSI C,所以也就沒有使用這個機制,而是使用一個通用的地址結構struct sockaddr來處理的
  1. struct sockaddr {  
  2.   uint8_t      sa_len;  
  3.   sa_family_t  sa_family;    /* address family: AF_xxx value */  
  4.   char         sa_data[14];  /* protocol-specific address */  
  5. };  
IPv6的socket地址爲struct sockaddr_in6
  1. struct in6_addr {  
  2.   uint8_t  s6_addr[16];          /* 128-bit IPv6 address */  
  3.                                  /* network byte ordered */  
  4. };  
  5.   
  6. #define SIN6_LEN      /* required for compile-time tests */  
  7.   
  8. struct sockaddr_in6 {  
  9.   uint8_t         sin6_len;      /* length of this struct (28) */  
  10.   sa_family_t     sin6_family;   /* AF_INET6 */  
  11.   in_port_t       sin6_port;     /* transport layer port# */  
  12.                                  /* network byte ordered */  
  13.   uint32_t        sin6_flowinfo; /* flow information, undefined */  
  14.   struct in6_addr sin6_addr;     /* IPv6 address */  
  15.                                  /* network byte ordered */  
  16.   uint32_t        sin6_scope_id; /* set of interfaces for a scope */  
  17. };  
對於sockaddr-in6來說,我們不能用通用的地址struct sockaddr來存儲了,而是產用新的通用地址結構struct sockaddr_storage,這個結構足夠大可以存儲任何系統支持的地址。
  1. struct sockaddr_storage {  
  2.   uint8_t      ss_len;       /* length of this struct (implementation dependent) */  
  3.   sa_family_t  ss_family;    /* address family: AF_xxx value */  
  4.   /* implementation-dependent elements to provide: 
  5.    * a) alignment sufficient to fulfill the alignment requirements of 
  6.    *    all socket address types that the system support 
  7.    * b) enough storage to hold any type of socket address that the 
  8.    *    system supports. 
  9.    */  
  10. };  
幾種常見的地址結構 3) 相關函數的的length 對於從程序傳地址給內核的函數(如connect),其長度是一個整型值,告訴內核要copy的地址長度。 對於從內核傳遞給程序的函數(如accpt),其長度是一個整型指針,是一個value-result參數。有兩個目的:一告訴內核地址結構的長度,讓內核在copy的時候不要超過這個長度;二返回內核真正copy的長度。 4)字節序 socket相關的函數都是使用網絡字節序 5)地址轉換函數 inet_aton, inet_ntoa, and inet_addr把IPv4字符串地址轉爲32位的網絡字節序地址 inet_ptonand inet_ntop可以轉換IPv4和IPv6的地址 6)listen中的backlog 要知道這個值的含義先用說一下,對於一個listen的socket,有兩個隊列:一個是incomplete connection隊列(僅僅收到SYN);一個是complete connection隊列(三次握手完成)。accept函數就是在complete connection隊列中取一個socket。backlog就是指隊列的個數,但不行的是各個地方都沒有明確定義這個值,沒有說明究竟代表了哪個隊列,或是兩個隊列之和。一般來說可以 同時處理的連接數是backlog的1.5倍,很多地方都用5. 7) getsockname 和 getpeername 這兩個函數可以與socket關聯的地址,getsockname 和 getpeername分別得到自己和對端的地址
  1. int getsockname(int sockfd, struct sockaddr *localaddr, socklen_t *addrlen)  
  2.   
  3. int getpeername(int sockfd , struct sockaddr * peeraddr , socklen_t * addrlen );   
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章