Linux网络编程：socket文件传输范例

转自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: lisong.shine@qq.com  
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: lisong.shine@qq.com  
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. }