1. #include<stdio.h>  
2. #include<stdlib.h>  
3. #include<string.h>  
4. #include<unistd.h>  
5. #include<sys/types.h>  
6. #include<sys/socket.h>  
7. #include<netinet/in.h>  
8. #include<arpa/inet.h>  
9. #include<fcntl.h>  
10. #include<dirent.h>  
11. #include<errno.h>  
12.   
13. #define N 128   
14. #define PORT_NUM 8888  
15.   
16. typedef struct sockaddr SA;  
17.   
18. void ProcessList(int connfd)  
19. {  
20.     char buf[N];  
21.     DIR *mydir;  
22.     struct dirent *myitem;  
23.   
24.     mydir = opendir(".");  
25.     while((myitem = readdir(mydir)) != NULL){  
26.         if((strcmp(myitem->d_name, ".") == 0) || (strcmp(myitem->d_name, "..") == 0))  
27.             continue;  
28.         strcpy(buf, myitem->d_name);  
29.         send( connfd, buf, N, 0);  
30.     }  
31.     close(connfd);  
32.   
33.     return;  
34. }  
35.   
36. void ProcessGet(int connfd, char buf[])  
37. {  
38.     int fd,nbyte;  
39.   
40.     if(( fd = open(buf+1, O_RDONLY)) < 0){  
41.         fprintf(stderr, "fail to open %s: %s\n",buf+1,strerror(errno));  
42.         buf[0] = ‘N‘;  
43.         send(connfd, buf, N, 0);  
44.         return;  
45.     }  
46.     buf[0] = ‘Y‘;  
47.     send(connfd, buf, N, 0);  
48.     while(( nbyte = read( fd, buf, N)) > 0){  
49.         send(connfd, buf, nbyte, 0);      
50.     }  
51.   
52.     close(connfd);  
53.   
54.     return;  
55. }  
56.   
57. void ProcessPut(int connfd, char buf[])  
58. {  
59.     int fd, nbyte;  
60.   
61.     if(( fd = open(buf+1, O_WRONLY | O_CREAT | O_TRUNC, 0666)) < 0){  
62.         printf("fail to create %s on server\n",buf+1);  
63.         return;  
64.     }  
65.   
66.     while(( nbyte = recv( connfd, buf, N, 0)) > 0){  
67.         write(fd, buf, nbyte);    
68.     }  
69.     close(fd);  
70.   
71.     return;  
72. }  
73.   
74. int main(int argc, char *argv[])  
75. {  
76.     int listenfd, connfd;  
77.     int optval = 1;  
78.     char buf[N];  
79.     struct sockaddr_in server_addr;  
80.   
81.     if(( listenfd = socket( AF_INET, SOCK_STREAM, 0)) < 0 ){  
82.         fprintf(stderr, "fail to socket: %s\n",strerror(errno));  
83.         exit(-1);  
84.     }  
85.   
86. #ifdef _DEBUG_  
87.     printf("socket is %d\n", listenfd);  
88. #endif  
89.   
90.     memset(&server_addr, 0, sizeof(server_addr));  
91.     server_addr.sin_family = AF_INET;  
92.     server_addr.sin_port = htons(PORT_NUM);  
93.     server_addr.sin_addr.s_addr = htonl(INADDR_ANY);  
94.       
95.     setsockopt(listenfd, SOL_SOCKET, SO_REUSEADDR, &optval, sizeof(optval));  
96.   
97.     if( bind( listenfd, (SA *)(&server_addr), sizeof(server_addr)) < 0){  
98.         perror("fail to bind");  
99.         exit(-1);  
100.     }  
101.     listen( listenfd, 5);  
102.   
103.     while(1){  
104.         if(( connfd = accept(listenfd, NULL, NULL)) < 0){  
105.             perror("fail to accept");  
106.             break;  
107.         }     
108.         recv( connfd, buf, N, 0);  
109.   
110.         switch(buf[0]){  
111.             case ‘L‘: ProcessList(connfd);  
112.                 break;  
113.             case ‘G‘: ProcessGet(connfd, buf);  
114.                 break;  
115.             case ‘P‘: ProcessPut(connfd, buf);  
116.                 break;  
117.             default: printf("Input ");  
118.         }  
119.         close(connfd);  
120.     }  
121.   
122.     return 0;  
123. }  

[cpp] view plain copy

 

1. #include<stdio.h>  
2. #include<stdlib.h>  
3. #include<unistd.h>  
4. #include<string.h>  
5. #include<fcntl.h>  
6. #include<errno.h>  
7. #include<sys/types.h>  
8. #include<sys/socket.h>  
9. #include<netinet/in.h>  
10. #include<arpa/inet.h>  
11.   
12. #define N 128  
13. #define PORT_NUM 8888  
14.   
15. typedef struct sockaddr SA;   
16.   
17. void PrintHelp()  
18. {  
19.     printf("help: display help info\n");  
20.     printf("list: get file list of server\n");  
21.     printf("get : get <file>\n");  
22.     printf("put : put <file>\n");  
23.     printf("quit: quit the client\n");  
24.   
25.     return;  
26. }  
27.   
28. void ProcessList(struct sockaddr_in server_addr)  
29. {  
30.     int sockfd, nbyte;  
31.     char buf[N];  
32.   
33.     if(( sockfd = socket( AF_INET, SOCK_STREAM, 0)) < 0){  
34.         printf("fail to list\n");  
35.         return;  
36.     }  
37.   
38.     if( connect(sockfd, (SA *)(&server_addr), sizeof(server_addr)) < 0){  
39.         printf("fail to connect server\n");  
40.         goto ERROR_1;  
41.     }  
42.   
43.     strcpy(buf, "L");  
44.     send(sockfd, buf, N, 0);  
45.   
46.     while(( nbyte = recv( sockfd, buf, N, 0)) != 0){  
47.         printf("%s\n",buf);   
48.     }  
49.   
50. ERROR_1:  
51.     close(sockfd);  
52.   
53.     return;  
54. }  
55.   
56. void ProcessGet(struct sockaddr_in server_addr, char command[])  
57. {  
58.     int sockfd, nbyte, fd;  
59.     char buf[N];  
60.   
61.     if(( sockfd = socket( AF_INET, SOCK_STREAM, 0)) < 0){  
62.         printf("fail to get\n");  
63.         return;  
64.     }  
65.   
66.     if( connect( sockfd, (SA *)(&server_addr), sizeof(server_addr)) < 0){  
67.         printf("fail to connect server\n");  
68.         goto ERROR_2;  
69.     }  
70.   
71.     sprintf(buf, "G%s", command+4);  
72.     send(sockfd, buf, N, 0);  
73.     recv(sockfd, buf, N, 0);  
74.   
75.     if(buf[0] == ‘N‘){  
76.         printf("No such file on server\n");  
77.         goto ERROR_2;  
78.     }  
79.   
80.     if(( fd = open(command+4, O_WRONLY | O_CREAT | O_TRUNC, 0666)) < 0){  
81.         printf("fail to create local file %s\n",command+4);  
82.         goto ERROR_2;  
83.     }  
84.   
85.     while(( nbyte = recv(sockfd, buf, N, 0)) > 0){  
86.         write(fd, buf, nbyte);    
87.     }  
88.     close(fd);  
89.   
90. ERROR_2:  
91.     close(sockfd);  
92.   
93.     return;  
94. }  
95.   
96. void ProcessPut(struct sockaddr_in server_addr, char command[])  
97. {  
98.     int sockfd, fd, nbyte;  
99.     char buf[N];  
100.   
101.     if(( sockfd = socket( AF_INET, SOCK_STREAM, 0)) < 0){  
102.         printf("fail to get\n");  
103.         return;  
104.     }  
105.       
106.     if( connect(sockfd,(SA *)(&server_addr),sizeof(server_addr)) < 0){  
107.         printf("fail to connect server\n");  
108.         goto ERROR_3;  
109.     }  
110.   
111.     if((fd = open(command+4, O_RDONLY)) < 0){  
112.         printf("fail to open %s\n",command+4);  
113.         goto ERROR_3;  
114.     }  
115.     sprintf(buf, "P%s", command+4);  
116.     send(sockfd, buf, N, 0);  
117.   
118.     while(( nbyte = read(fd, buf, N)) > 0){  
119.         send(sockfd, buf, nbyte, 0);      
120.     }  
121.     close(fd);  
122.   
123. ERROR_3:  
124.     close(sockfd);  
125.   
126.     return;  
127. }  
128.   
129. int main(int argc, char *argv[])  
130. {  
131.     int sockfd, fd, nbyte;  
132.     char command[32];  
133.     struct sockaddr_in server_addr;  
134.   
135.     if(argc < 3){  
136.         printf("Usage: %s <server_ip> <port>\n",argv[0]);  
137.         exit(-1);  
138.     }  
139.   
140.     if(( sockfd = socket( AF_INET, SOCK_STREAM, 0)) < 0){  
141.         fprintf(stderr, "fail to socket: %s\n", strerror(errno));  
142.         exit(-1);  
143.     }  
144.   
145. #ifdef _DEBUG_  
146.     printf("socket is %d\n",sockfd);  
147. #endif  
148.       
149.     memset(&server_addr, 0, sizeof(server_addr));  
150.     server_addr.sin_family = AF_INET;  
151.     server_addr.sin_port = htons(atoi(argv[2]));  
152.     server_addr.sin_addr.s_addr = inet_addr(argv[1]);  
153.     //server_addr.sin_addr.s_addr = htonl(INADDR_ANY);  
154.   
155.     while(1){  
156.         printf("<client>");  
157.         fgets(command, 32, stdin);  
158.         command[strlen(command)-1] = ‘\0‘;  
159.         if(strcmp( command, "help") == 0){  
160.             PrintHelp();      
161.         }  
162.         else if(strcmp( command, "list") == 0){  
163.             ProcessList(server_addr);     
164.         }  
165.         else if(strncmp( command, "get", 3) == 0){  
166.             ProcessGet(server_addr, command);     
167.         }  
168.         else if(strncmp( command, "put", 3) == 0){  
169.             ProcessPut(server_addr, command);  
170.         }  
171.         else if(strcmp( command, "quit") == 0){  
172.             printf("Bye\n");  
173.             break;  
174.         }  
175.         else{  
176.             printf("Wrong command, ‘help‘ for command list.\n");      
177.         }  
178.     }  
179.   
180.     return 0;  
181. }  

3.验证结果（ubuntu14.04）