TCP编程实现LINUX平台文件传输工程

Posted 2020-07-30

server端代码

 1 #include <stdio.h>  
 2 #include <string.h>  
 3 #include <stdlib.h>  
 4 #include <string.h>  
 5 #include <unistd.h>  
 6 #include <sys/types.h>  
 7 #include <sys/socket.h>  
 8 #include <netinet/in.h>  
 9 #include <arpa/inet.h>  
10   
11 #define    MAXLINE        1024  
12   
13 void usage(char *command)  
14 {  
15     printf("usage :%s portnum filename\n", command);  
16     exit(0);  
17 }  
18 int main(int argc,char **argv)   
19 {  
20     struct sockaddr_in     serv_addr;  
21     struct sockaddr_in     clie_addr;  
22     char                   buf[MAXLINE];  
23     int                    sock_id;  
24     int                    link_id;  
25     int                    recv_len;  
26     int                    write_leng;  
27     int                    clie_addr_len;  
28     FILE                   *fp;  
29   
30     if (argc != 3) {  
31         usage(argv[0]);  
32     }  
33 /*<-----------------------------------------socket---------------------------------------------->*/
34     if ((sock_id = socket(AF_INET, SOCK_STREAM, 0)) < 0) 
35     {  
36         perror("Create socket failed\n");  
37         exit(0);  
38     }  
39 /*<-----------------------------------------bind------------------------------------------------->*/
40     memset(&serv_addr, 0, sizeof(serv_addr));  
41     serv_addr.sin_family = AF_INET;  
42     serv_addr.sin_port = htons(atoi(argv[1]));  
43     serv_addr.sin_addr.s_addr = htonl(INADDR_ANY);  
44   
45     if (bind(sock_id, (struct sockaddr *)&serv_addr, sizeof(serv_addr)) < 0 ) 
46     {  
47         perror("Bind socket failed\n");  
48         exit(0);  
49     }  
50 /*<-----------------------------------------listen----------------------------------------------->*/
51     if (-1 == listen(sock_id, 10)) 
52     {  
53         perror("Listen socket failed\n");  
54         exit(0);  
55     }  
56 /*<-------------------------------------server receive part---------------------------------->*/
57     while (1) {  
58         if ((fp = fopen(argv[2], "w")) == NULL) 
59         {  
60             perror("Open file failed\n");  
61             exit(0);  
62         }  
63         lseek(fp, 0, SEEK_END);
64         clie_addr_len = sizeof(clie_addr); 
65 /*<-----------------------------------------accept----------------------------------------------->*/        
66         link_id = accept(sock_id, (struct sockaddr *)&clie_addr, &clie_addr_len);  
67         if (-1 == link_id) {  
68             perror("Accept socket failed\n");  
69             exit(0);  
70         }  
71         bzero(buf, MAXLINE);  
72         while (recv_len = recv(link_id, buf, MAXLINE, 0)) 
73         {  
74             if(recv_len < 0) 
75             {  
76                 printf("Recieve Data From Server Failed!\n");  
77                 break;  
78             }  
79             printf("#");  
80             write_leng = fwrite(buf, sizeof(char), recv_len, fp);  
81             if (write_leng < recv_len) 
82             {  
83                 printf("Write file failed\n");  
84                 break;  
85             }  
86             bzero(buf,MAXLINE);  
87         }  
88         printf("\nFinish Recieve\n");  
89         fclose(fp);  
90         close(link_id);  
91     }  
92     close(sock_id);   
93     return 0;  
94 }