1 /************************************************************************* 2 > File Name: my_socket.h 3 > Author:Monica 4 > Mail:liling222@126.com 5 > Created Time: Thu 03 Jul 2014 10:40:54 AM CST 6 ************************************************************************/ 7 8 #ifndef __MY_SOCKET_H__ 9 #define __MY_SOCKET_H__ 10 11 #include <stdio.h> 12 #include <sys/types.h> 13 #include <sys/socket.h> 14 #include <sys/stat.h> 15 #include <arpa/inet.h> 16 #include <netinet/in.h> 17 #include <fcntl.h> 18 #include <stdlib.h> 19 #include <string.h> 20 #include <unistd.h> 21 #include "msg.h" 22 23 int listenfd_init(char* ip, char* port); 24 25 26 #endif
1 /************************************************************************* 2 > File Name: server.c 3 > Author: Monica 4 > Mail:liling222@126.com 5 > Created Time: Thu 03 Jul 2014 10:02:48 AM CST 6 ************************************************************************/ 7 8 #include "my_socket.h" 9 10 int listenfd_init(char* ip, char* port){ 11 12 int fd_server; 13 int reuse = 1 ; 14 struct sockaddr_in server_addr; 15 fd_server = socket(AF_INET,SOCK_STREAM, 0); 16 if(fd_server == -1) 17 { 18 perror("socket"); 19 exit(-1); 20 } 21 if(setsockopt(fd_server,SOL_SOCKET,SO_REUSEADDR,&reuse,sizeof(int)) == -1) 22 { 23 perror("setsockopt"); 24 close(fd_server); 25 exit(-1); 26 } 27 memset(&server_addr, 0 , sizeof(server_addr)) ; 28 server_addr.sin_family = AF_INET ; 29 server_addr.sin_port = htons(atoi(port)); 30 server_addr.sin_addr.s_addr = inet_addr(ip); 31 if( bind(fd_server,(struct sockaddr*)&server_addr,sizeof(server_addr)) == -1) 32 { 33 perror("bind"); 34 close(fd_server); 35 exit(-1); 36 } 37 if(listen(fd_server,5) == -1) 38 { 39 perror("listen"); 40 close(fd_server); 41 exit(-1); 42 } 43 return fd_server; 44 }
1 #include "my_socket.h" 2 3 void * thread_handle(void* fd_client); 4 int main(int argc , char* argv[])//exe ip port 5 { 6 int fd_server,fd_client; 7 struct sockaddr_in server_addr,client_addr ; 8 int addrlen = sizeof(client_addr); 9 int tid; 10 if(argc != 3) 11 { 12 printf("usge : exe ip port"); 13 exit(-1); 14 } 15 fd_server = listenfd_init(argv[1], argv[2]); 16 while(fd_client = accept(fd_server,(struct sockaddr*)&client_addr,&addrlen)){ 17 printf("a client connect IP:%s PORT:%d ",inet_ntoa(client_addr.sin_addr),ntohs(client_addr.sin_port)); 18 pthread_create(&tid, NULL, thread_handle, (void *)fd_client); 19 } 20 close(fd_server); 21 return 0; 22 } 23 void* thread_handle(void* fd_client){ 24 pthread_detach(pthread_self()); 25 MSG recv_msg, snd_msg; 26 int read_n; 27 int fd_file; 28 int fd_cli = (int)fd_client; 29 memset(&recv_msg, 0, sizeof(recv_msg)); 30 recv(fd_cli, &recv_msg.s_len, MSG_LEN , 0) ; 31 recv(fd_cli, recv_msg.s_buf, recv_msg.s_len, 0) ; 32 printf("recv msg :%s ", recv_msg.s_buf); 33 fd_file = open(recv_msg.s_buf, O_RDONLY); 34 while(memset(&snd_msg, 0, sizeof(snd_msg)), (read_n = read(fd_file, snd_msg.s_buf, MSG_SIZE)) > 0) 35 { 36 snd_msg.s_len = read_n ; 37 send(fd_cli, &snd_msg, snd_msg.s_len + MSG_LEN ,0); 38 } 39 snd_msg.s_len = 0 ; 40 send(fd_cli, &snd_msg, snd_msg.s_len + MSG_LEN ,0); 41 close(fd_file); 42 close(fd_cli); 43 }
1 /************************************************************************* 2 > File Name: client.c 3 > Author:Monica 4 > Mail:liling222@126.com 5 > Created Time: Thu 03 Jul 2014 10:46:20 AM CST 6 ************************************************************************/ 7 8 #include "my_socket.h" 9 int main(int argc, char* argv[]){ 10 int fd_client; 11 struct sockaddr_in server_addr; 12 MSG recv_msg, snd_msg; 13 int fd_file; 14 int recv_len = 0 ; 15 char filename[128]; 16 server_addr.sin_family = AF_INET; 17 server_addr.sin_addr.s_addr = inet_addr(argv[1]); 18 server_addr.sin_port = htons(atoi(argv[2])); 19 if((fd_client = socket(AF_INET, SOCK_STREAM, 0)) == -1){ 20 perror("socket"); 21 exit(-1); 22 } 23 if(connect(fd_client, (struct sockaddr*)&server_addr, sizeof(server_addr)) == -1){ 24 perror("connect"); 25 close(fd_client); 26 exit(-1); 27 } 28 printf("inputs filename:"); 29 scanf("%s", snd_msg.s_buf); 30 snd_msg.s_len = strlen(snd_msg.s_buf); 31 send(fd_client,&snd_msg, snd_msg.s_len+MSG_LEN, 0 ); 32 sprintf(filename, "download_%s", snd_msg.s_buf); 33 fd_file = open(filename, O_WRONLY|O_CREAT, 0666); 34 while(1){ 35 printf("begin recv "); 36 memset(&recv_msg, 0, sizeof(recv_msg)); 37 recv(fd_client, &recv_msg.s_len, MSG_LEN, 0); 38 if(recv_msg.s_len >0){ 39 recv(fd_client, recv_msg.s_buf, MSG_SIZE, 0); 40 write(fd_file, recv_msg.s_buf, strlen(recv_msg.s_buf)); 41 recv_len += recv_msg.s_len; 42 printf("download...%d ", recv_len); 43 } 44 else{ 45 break; 46 } 47 } 48 close(fd_file); 49 close(fd_client); 50 return 0; 51 }