TCP 文件服务器实现功能
主要实现文件的上传,下载,查看功能。
TCP文件服务器server.c
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/types.h> /* See NOTES */
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <sys/types.h>
#include <dirent.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#define N 128
// ./a.out 192.168.1.57 8000
// argv[0] argv[1] aegv[2]
int process_list(int newsockfd,char *buf)
{
DIR * dirp;
int ret;
struct dirent * direntp;
dirp = opendir("./");
if(dirp == NULL)
{
perror("opendir");
exit(-1);
}
while( (direntp = readdir(dirp)) != NULL)
{
if(strncmp(direntp->d_name,".",1) == 0 ) continue;
strcpy(buf,direntp->d_name);
ret = write(newsockfd,buf,N);
if(ret < 0)
{
perror("write");
exit(-1);
}
}
close(newsockfd);
return 0;
}
int process_get(int newsockfd,char *buf)
{
int ret,fd,nbytes;
fd = open(buf+4,O_RDWR);
if(fd < 0)
{
perror("open");
exit(-1);
}
while( (nbytes= read(fd,buf,N)) > 0)
{
ret = write(newsockfd,buf,nbytes);
if(ret < 0)
{
perror("write");
exit(-1);
}
}
close(newsockfd);
return 0;
}
int process_put(int newsockfd,char *buf)
{
int ret,fd,nbytes;
fd = open(buf+4,O_RDWR|O_CREAT|O_TRUNC,0664);
if(fd < 0)
{
perror("open");
exit(-1);
}
while( (nbytes= read(newsockfd,buf,N)) > 0)
{
ret = write(fd,buf,nbytes);
if(ret < 0)
{
perror("write");
exit(-1);
}
}
close(newsockfd);
close(fd);
return 0;
}
int main(int argc, const char *argv[])
{
char buf[N] ={0};
int sockfd,ret,newsockfd;
struct sockaddr_in myaddr,client;
if(argc != 3)
{
printf("运行程序时带入ip 和 port\n");
exit(-1);
}
socklen_t addrlen = sizeof(client);
sockfd = socket(AF_INET,SOCK_STREAM,0);
if(sockfd < 0)
{
perror("socket");
exit(-1);
}
printf("sockfd=%d\n",sockfd);
memset(&myaddr,0,sizeof(myaddr));
myaddr.sin_family = AF_INET ;
myaddr.sin_port = htons(atoi(argv[2]));
myaddr.sin_addr.s_addr = inet_addr(argv[1]);
ret = bind(sockfd,(struct sockaddr *)&myaddr,sizeof(myaddr)) ; // 给socket 一个固定的ip 和端口
if(ret < 0)
{
perror("bind");
exit(-1);
}
ret = listen(sockfd,10);
if(ret < 0)
{
perror("listen");
exit(-1);
}
while(1)
{
newsockfd = accept(sockfd,(struct sockaddr *)&client,&addrlen) ;
if(newsockfd < 0)
{
printf("accept");
exit(-1);
}
printf("newsockfd = %d\n",newsockfd);
ret = read(newsockfd,buf,N);
if(ret < 0)
{
printf("read");
exit(-1);
}
printf("buf=%s\n",buf);
if( strncmp(buf,"list",4) == 0 )
{
process_list(newsockfd,buf);
}
else if( strncmp(buf,"get",3) == 0 )
{
process_get(newsockfd,buf);
}
else if( strncmp(buf,"put",3) == 0 )
{
process_put(newsockfd,buf);
}
}
return 0;
}
TCP文件服务器client.c
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/types.h> /* See NOTES */
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#define N 128
// ./client 192.168.1.57 8000
// list
// get file
// put file
// help
int process_list(struct sockaddr_in *addr,char * buf)
{
int ret,sockfd,nbytes;
sockfd = socket(AF_INET,SOCK_STREAM,0);
if(sockfd < 0)
{
perror("socket");
exit(-1);
}
ret = connect(sockfd,(struct sockaddr *)addr,sizeof(*addr)) ;
if(ret < 0)
{
perror("connect");
exit(-1);
}
ret = write(sockfd,buf,N);
if(ret < 0)
{
perror("write");
exit(-1);
}
while( (nbytes= read(sockfd,buf,N)) > 0)
{
printf("%s\n",buf);
}
close(sockfd);
return 0;
}
int process_get(struct sockaddr_in *addr,char * buf)
{
int ret,sockfd,nbytes,fd;
sockfd = socket(AF_INET,SOCK_STREAM,0);
if(sockfd < 0)
{
perror("socket");
exit(-1);
}
ret = connect(sockfd,(struct sockaddr *)addr,sizeof(*addr)) ;
if(ret < 0)
{
perror("connect");
exit(-1);
}
ret = write(sockfd,buf,N);
if(ret < 0)
{
perror("write");
exit(-1);
}
fd = open(buf+4,O_RDWR|O_CREAT|O_TRUNC,0664);
if(fd < 0)
{
perror("open");
exit(-1);
}
while( (nbytes= read(sockfd,buf,N)) > 0)
{
ret = write(fd,buf,nbytes);
if(ret < 0)
{
perror("write");
exit(-1);
}
}
close(sockfd);
return 0;
}
int process_put(struct sockaddr_in *addr,char * buf)
{
int ret,sockfd,nbytes,fd;
sockfd = socket(AF_INET,SOCK_STREAM,0);
if(sockfd < 0)
{
perror("socket");
exit(-1);
}
ret = connect(sockfd,(struct sockaddr *)addr,sizeof(*addr)) ;
if(ret < 0)
{
perror("connect");
exit(-1);
}
ret = write(sockfd,buf,N);
if(ret < 0)
{
perror("write");
exit(-1);
}
fd = open(buf+4,O_RDWR);
if(fd < 0)
{
perror("open");
exit(-1);
}
while( (nbytes= read(fd,buf,N)) > 0)
{
ret = write(sockfd,buf,nbytes);
if(ret < 0)
{
perror("write");
exit(-1);
}
}
close(sockfd);
close(fd);
return 0;
}
int main(int argc, const char *argv[])
{
char buf[N] = {0};
int sockfd,ret;
struct sockaddr_in server,client;
if(argc != 3)
{
printf("运行程序时带入ip 和 port\n");
exit(-1);
}
memset(&server,0,sizeof(server));
server.sin_family = AF_INET ;
server.sin_port = htons( atoi(argv[2]) );
server.sin_addr.s_addr = inet_addr(argv[1]);
while(1)
{
printf(">:");
fgets(buf,N,stdin);
buf[strlen(buf)-1] = '\0';
if( strncmp(buf,"list",4) == 0 )
{
process_list(&server,buf);
}
else if( strncmp(buf,"get",3) == 0 )
{
process_get(&server,buf);
}
else if( strncmp(buf,"put",3) == 0 )
{
process_put(&server,buf);
}
else if( strncmp(buf,"quit",4) == 0 )
{
break;
}
else if( strncmp(buf,"exit",4) == 0 )
{
break;
}
else
{
printf("命令错误,请重新输入\n");
}
}
#if 0
printf(">:");
fgets(buf,N,stdin);
ret = write(sockfd,buf,N) ;
if(ret < 0)
{
perror("write");
exit(-1);
}
ret = read(sockfd,buf,N);
if(ret < 0)
{
printf("read");
exit(-1);
}
printf("%s\n",buf);
close(sockfd);
#endif
return 0;
}
运行client输入list实现查看功能
输入get+文件名实现下载
输入put+文件名实现上传