Linux網路程式設計:socket檔案傳輸範例

NO IMAGE

轉自http://blog.csdn.net/lisonglisonglisong/article/details/22699675

基於TCP流協議的socket網路檔案傳輸Demo:

實現:C語言
功能:檔案傳輸(可以傳任何格式的檔案)

[cpp] view
plain
 copy

  1. /************************************************************************* 
  2.     > File Name: Server.c 
  3.     > Author: SongLee 
  4.     > E-mail: [email protected]  
  5.     > Created Time: 2014年03月13日 星期四 22時17分43秒 
  6.     > Personal Blog: http://songlee24.github.io/ 
  7.  ************************************************************************/  
  8.   
  9. #include<netinet/in.h>  // sockaddr_in  
  10. #include<sys/types.h>   // socket  
  11. #include<sys/socket.h>  // socket  
  12. #include<stdio.h>       // printf  
  13. #include<stdlib.h>      // exit  
  14. #include<string.h>      // bzero  
  15.   
  16. #define SERVER_PORT 8000  
  17. #define LENGTH_OF_LISTEN_QUEUE 20  
  18. #define BUFFER_SIZE 1024  
  19. #define FILE_NAME_MAX_SIZE 512  
  20.   
  21. int main(void)  
  22. {  
  23.     // 宣告並初始化一個伺服器端的socket地址結構  
  24.     struct sockaddr_in server_addr;  
  25.     bzero(&server_addr, sizeof(server_addr));  
  26.     server_addr.sin_family = AF_INET;  
  27.     server_addr.sin_addr.s_addr = htons(INADDR_ANY);
     //任意IP地址
  28.     server_addr.sin_port = htons(SERVER_PORT);  
  29.   
  30.     // 建立socket,若成功,返回socket描述符  
  31.     int server_socket_fd = socket(PF_INET, SOCK_STREAM, 0);  
  32.     if(server_socket_fd < 0)  
  33.     {  
  34.         perror(“Create Socket Failed:”);  
  35.         exit(1);  
  36.     }  
  37.     int opt = 1;  
  38.     setsockopt(server_socket_fd, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof(opt));  
  39.   
  40.     // 繫結socket和socket地址結構  
  41.     if(-1 == (bind(server_socket_fd, (struct sockaddr*)&server_addr, sizeof(server_addr))))  
  42.     {  
  43.         perror(“Server Bind Failed:”);  
  44.         exit(1);  
  45.     }  
  46.       
  47.     // socket監聽  
  48.     if(-1 == (listen(server_socket_fd, LENGTH_OF_LISTEN_QUEUE)))
     //LENGTH_OF_LISTEN_QUEUE
  49. //參考http://blog.csdn.net/wm_1991/article/details/52176287
  50.     {  
  51.         perror(“Server Listen Failed:”);  
  52.         exit(1);  
  53.     }  
  54.   
  55.     while(1)  
  56.     {  
  57.         // 定義客戶端的socket地址結構  
  58.         struct sockaddr_in client_addr;  
  59.         socklen_t client_addr_length = sizeof(client_addr);  
  60.   
  61.         // 接受連線請求,返回一個新的socket(描述符),這個新socket用於同連線的客戶端通訊  
  62.         // accept函式會把連線到的客戶端資訊寫到client_addr中  
  63.         int new_server_socket_fd = accept(server_socket_fd, (struct sockaddr*)&client_addr, &client_addr_length);  
  64.         if(new_server_socket_fd < 0)  
  65.         {  
  66.             perror(“Server Accept Failed:”);  
  67.             break;  
  68.         }  
  69.   
  70.         // recv函式接收資料到緩衝區buffer中  
  71.         char buffer[BUFFER_SIZE];  
  72.         bzero(buffer, BUFFER_SIZE);  
  73.         if(recv(new_server_socket_fd, buffer, BUFFER_SIZE, 0) < 0)
     //BUFFER_SIZE=1024
  74.         {  
  75.             perror(“Server Recieve Data Failed:”);  
  76.             break;  
  77.         }  
  78.   
  79.         // 然後從buffer(緩衝區)拷貝到file_name中  
  80.         char file_name[FILE_NAME_MAX_SIZE 1];  
  81.         bzero(file_name, FILE_NAME_MAX_SIZE 1);  
  82.         strncpy(file_name, buffer, strlen(buffer)>FILE_NAME_MAX_SIZE?FILE_NAME_MAX_SIZE:strlen(buffer));  
  83.         printf(“%s\n”, file_name);  
  84.   
  85.         // 開啟檔案並讀取檔案資料  
  86.         FILE *fp = fopen(file_name, “r”);  
  87.         if(NULL == fp)  
  88.         {  
  89.             printf(“File:%s Not Found\n”, file_name);  
  90.         }  
  91.         else  
  92.         {  
  93.             bzero(buffer, BUFFER_SIZE);  
  94.             int length = 0;  
  95.             // 每讀取一段資料,便將其傳送給客戶端,迴圈直到檔案讀完為止  
  96.             while((length = fread(buffer, sizeof(char), BUFFER_SIZE, fp)) > 0)  
  97.             {  
  98.                 if(send(new_server_socket_fd, buffer, length, 0) < 0)  
  99.                 {  
  100.                     printf(“Send File:%s Failed./n”, file_name);  
  101.                     break;  
  102.                 }  
  103.                 bzero(buffer, BUFFER_SIZE);  
  104.             }  
  105.   
  106.             // 關閉檔案  
  107.             fclose(fp);  
  108.             printf(“File:%s Transfer Successful!\n”, file_name);  
  109.         }  
  110.         // 關閉與客戶端的連線  
  111.         close(new_server_socket_fd);  
  112.     }  
  113.     // 關閉監聽用的socket  
  114.     close(server_socket_fd);  
  115.     return 0;  
  116. }  
[cpp] view
plain
 copy

  1. /************************************************************************* 
  2.     > File Name: Client.c 
  3.     > Author: SongLee 
  4.     > E-mail: [email protected]  
  5.     > Created Time: 2014年03月14日 星期五 09時41分46秒 
  6.     > Personal Blog: http://songlee24.github.io/ 
  7.  ************************************************************************/  
  8.   
  9. #include<netinet/in.h>   // sockaddr_in  
  10. #include<sys/types.h>    // socket  
  11. #include<sys/socket.h>   // socket  
  12. #include<stdio.h>        // printf  
  13. #include<stdlib.h>       // exit  
  14. #include<string.h>       // bzero  
  15.   
  16. #define SERVER_PORT 8000  
  17. #define BUFFER_SIZE 1024  
  18. #define FILE_NAME_MAX_SIZE 512  
  19.   
  20. int main()  
  21. {  
  22.     // 宣告並初始化一個客戶端的socket地址結構  
  23.     struct sockaddr_in client_addr;  
  24.     bzero(&client_addr, sizeof(client_addr));  
  25.     client_addr.sin_family = AF_INET;  
  26.     client_addr.sin_addr.s_addr = htons(INADDR_ANY);  
  27.     client_addr.sin_port = htons(0);  
  28.   
  29.     // 建立socket,若成功,返回socket描述符  
  30.     int client_socket_fd = socket(AF_INET, SOCK_STREAM, 0);  
  31.     if(client_socket_fd < 0)  
  32.     {  
  33.         perror(“Create Socket Failed:”);  
  34.         exit(1);  
  35.     }  
  36.   
  37.     // 繫結客戶端的socket和客戶端的socket地址結構 非必需  
  38.     if(-1 == (bind(client_socket_fd, (struct sockaddr*)&client_addr, sizeof(client_addr))))  
  39.     {  
  40.         perror(“Client Bind Failed:”);  
  41.         exit(1);  
  42.     }  
  43.   
  44.     // 宣告一個伺服器端的socket地址結構,並用伺服器那邊的IP地址及埠對其進行初始化,用於後面的連線  
  45.     struct sockaddr_in server_addr;  
  46.     bzero(&server_addr, sizeof(server_addr));  
  47.     server_addr.sin_family = AF_INET;  
  48. //inet_pton:將“點分十進位制” -> “二進位制整數”
    //int inet_pton(int af, const char *src, void *dst);
    //這個函式轉換字串到網路地址,第一個引數af是地址簇,第二個引數*src是來源地址,第三個引數* dst接收轉換後//的資料。
  49.     if(inet_pton(AF_INET, “127.0.0.1”, &server_addr.sin_addr) == 0)  
  50.     {  
  51.         perror(“Server IP Address Error:”);  
  52.         exit(1);  
  53.     }  
  54.     server_addr.sin_port = htons(SERVER_PORT);  
  55.     socklen_t server_addr_length = sizeof(server_addr);  
  56.   
  57.     // 向伺服器發起連線,連線成功後client_socket_fd代表了客戶端和伺服器的一個socket連線  
  58.     if(connect(client_socket_fd, (struct sockaddr*)&server_addr, server_addr_length) < 0)  
  59.     {  
  60.         perror(“Can Not Connect To Server IP:”);  
  61.         exit(0);  
  62.     }  
  63.   
  64.     // 輸入檔名 並放到緩衝區buffer中等待傳送  
  65.     char file_name[FILE_NAME_MAX_SIZE 1];  
  66.     bzero(file_name, FILE_NAME_MAX_SIZE 1);  
  67.     printf(“Please Input File Name On Server:\t”);  
  68.     scanf(“%s”, file_name);  
  69.   
  70.     char buffer[BUFFER_SIZE];  
  71.     bzero(buffer, BUFFER_SIZE);  
  72.     strncpy(buffer, file_name, strlen(file_name)>BUFFER_SIZE?BUFFER_SIZE:strlen(file_name));  
  73.       
  74.     // 向伺服器傳送buffer中的資料  
  75.     if(send(client_socket_fd, buffer, BUFFER_SIZE, 0) < 0)  
  76.     {  
  77.         perror(“Send File Name Failed:”);  
  78.         exit(1);  
  79.     }  
  80.   
  81.     // 開啟檔案,準備寫入  
  82.     FILE *fp = fopen(file_name, “w”);  
  83.     if(NULL == fp)  
  84.     {  
  85.         printf(“File:\t%s Can Not Open To Write\n”, file_name);  
  86.         exit(1);  
  87.     }  
  88.   
  89.     // 從伺服器接收資料到buffer中  
  90.     // 每接收一段資料,便將其寫入檔案中,迴圈直到檔案接收完並寫完為止  
  91.     bzero(buffer, BUFFER_SIZE);  
  92.     int length = 0;  
  93.     while((length = recv(client_socket_fd, buffer, BUFFER_SIZE, 0)) > 0)  
  94.     {  
  95.         if(fwrite(buffer, sizeof(char), length, fp) < length)  
  96.         {  
  97.             printf(“File:\t%s Write Failed\n”, file_name);  
  98.             break;  
  99.         }  
  100.         bzero(buffer, BUFFER_SIZE);  
  101.     }  
  102.   
  103.     // 接收成功後,關閉檔案,關閉socket  
  104.     printf(“Receive File:\t%s From Server IP Successful!\n”, file_name);  
  105.     close(fp);  
  106.     close(client_socket_fd);  
  107.     return 0;  
  108. }