本项目实现了客户端从服务器端下载文件,上传文件,查看客户端及服务器端目录,服务器端和客户端的CD操作。
1.服务器端客户端的连接流程。
ps:fork()常用在服务器端accept之后创建等待客户端的接入;以及调用exec函数,所以不用一直创建fork()函数。
2.服务器端代码
2.1config.h
#define LS 0
#define GET 1
#define CD 2
#define IFGO 3
#define LCD 4
#define LLS 5
#define PWD 6
#define PUT 7
#define QUIT 8
#define DOFILE 9
struct Msg
{
int type;
char data[1024];
char secondBuf[128];
};
~
2.2server
#include <stdio.h>
#include <sys/types.h> /* See NOTES */
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <errno.h>
#include <unistd.h>
#include <string.h>
#include "config.h"
#include <sys/stat.h>
#include <fcntl.h>
#include <stdlib.h>
#include <unistd.h>
int get_cmd_type(char *cmd)
{
if(!strcmp("ls",cmd)) return LS; //如果相比不等与0返回LS
if(!strcmp("pwd",cmd)) return PWD; //两个字符串比较用strcmp
if(!strcmp("quit",cmd)) return QUIT;
if(strstr(cmd,"cd")!=NULL) return CD; //如果cmd,"cd"相比如果cd是cmd的字串则返回的是cmd
if(strstr(cmd,"get")!=NULL) return GET; //否则返回NULL
if(strstr(cmd,"put")!=NULL) return PUT;
return 100;
}
char *getDesDir(char *cmsg) //字符串分割
{
char *p;
p = strtok(cmsg," ");
p = strtok(NULL," ");
return p;
}
void msg_handler(struct Msg msg,int fd)
{
char dataBuf[1024] = {0};
char *file = NULL;
int fdfile;
printf("cmd:%s\n",msg.data);
int ret = get_cmd_type(msg.data);
switch(ret){
case LS:
case PWD:{
// msg.type = 0;
FILE *r = popen(msg.data,"r"); //客户端传来数据调用POPEN开始执行
fread(msg.data,sizeof(msg.data),1,r);
write(fd,&msg,sizeof(msg)); //写fd回给客户端
break;}
case CD:{
// msg.type = 1;
char *dir = getDesDir(msg.data); //getDesDir例如cd或put等指令,(msg.data)例如aa,bb
printf("dir:%s\n",dir); //例如cd test/ dir就是test
chdir(dir); //不能用system会启动另外一个程序
break;}
case GET:
file = getDesDir(msg.data);
if(access(file,F_OK) == -1){ //判断文件是否存在,file文件名,F_OK默认宏
strcpy(msg.data,"No This File!");
write(fd,&msg,sizeof(msg));
}else{
msg.type = DOFILE;
fdfile = open(file,O_RDWR);
read(fdfile,dataBuf,sizeof(dataBuf));
close(fdfile);
break;
case QUIT:
printf("client quit!\n");
exit(-1);
}
}
int main(int argv,char **argc)
{
//2.bind
struct sockaddr_in s_addr;
memset(&s_addr,0,sizeof(struct sockaddr_in));
//4.accept
struct sockaddr_in c_addr;
memset(&c_addr,0,sizeof(struct sockaddr_in));
//5.write
struct Msg msg;
//6.read
char readBuf[128] = {0};
//1.socket
//int socket(int domain, int type, int protocol);
int s_fd;
int c_fd;
int n_read;
s_fd = socket(AF_INET,SOCK_STREAM,0);
if(s_fd == -1){
perror("socket");
printf("open socket failed\n");
exit(-1);
}
//2.bind
//int bind(int sockfd, const struct sockaddr *addr,socklen_t addrlen);
s_addr.sin_family = AF_INET;
s_addr.sin_port = htons(atoi(argc[2]));
inet_aton(argc[1],&s_addr.sin_addr);
bind(s_fd,(struct sockaddr *)&s_addr,sizeof(struct sockaddr));
//3.listen
//int listen(int sockfd, int backlog);
listen(s_fd,10);
//4.accept
//int accept(int sockfd, struct sockaddr *addr, socklen_t *addrlen);
int clen = sizeof(struct sockaddr_in);
while(1){
int c_fd = accept(s_fd,(struct sockaddr *)&c_addr,&clen);
if(c_fd == -1){
perror("accept");
exit(-1);
}
printf("connect cilent %s\n",inet_ntoa(c_addr.sin_addr));
//5.write
//ssize_t write(int fd, const void *buf, size_t count);
if(fork() == 0){
while(1){
memset(msg.data,0,sizeof(msg.data));
n_read = read(c_fd,&msg,sizeof(msg));
if(n_read == 0){
printf("client out\n");
break;
}else if(n_read > 0){
msg_handler(msg,c_fd);
}
}
}
}
close(c_fd);
close(s_fd);
return 0;
}
3.服务器端
client
#include <stdio.h>
#include <sys/types.h> /* See NOTES */
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <errno.h>
#include <unistd.h>
#include <string.h>
#include "config.h"
#include <sys/stat.h>
#include <fcntl.h>
#include <stdlib.h>
#include <unistd.h>
char *getdir(char *cmd)
{
char *p;
p = strtok(cmd," ");
p = strtok(NULL," ");
return p;
}
int get_cmd_type(char *cmd)
{
if(strstr(cmd,"lcd")) return LCD;
if(!strcmp("quit",cmd)) return QUIT;
if(!strcmp("ls",cmd)) return LS;
if(!strcmp("lls",cmd)) return LLS;
if(!strcmp("pwd",cmd)) return LS;
if(strstr(cmd,"cd")) return CD;
if(strstr(cmd,"get")) return GET;
if(strstr(cmd,"put")) return PUT;
return -1;
}
int cmd_handler(struct Msg msg,int fd)
{
char *dir = NULL;
char buf[32];
int filefd;
int ret;
ret = get_cmd_type(msg.data);
switch(ret){
case LS:
case PWD:
case CD:
msg.type = 0;
write(fd,&msg,sizeof(msg));
break;
case GET:
msg.type = 2;
write(fd,&msg,sizeof(msg));
break;
case PUT:
strcpy(buf,msg.data);
dir = getdir(buf);
if(access(dir,F_OK) == -1){
printf("%s not exsit\n",dir);
}else{
filefd = open(dir,O_RDWR);
read(filefd,msg.secondBuf,sizeof(msg.secondBuf));
close(filefd);
write(fd,&msg,sizeof(msg));
}
break;
case LLS:
system("ls");
break;
case LCD:
dir = getdir(msg.data);
chdir(dir);
break;
}
}
return ret;
}
void handler_server_message(int c_fd,struct Msg msg)
{
int n_read;
struct Msg msgget;
int newfilefd;
n_read = read(c_fd,&msgget,sizeof(msgget));
if(n_read == 0){
printf("sever is out,quit\n");
exit(-1);
}
else if(msgget.type == DOFILE){
char *p = getdir(msg.data);
newfilefd = open(p,O_RDWR|O_CREAT,0600);
write(newfilefd,msgget.data,strlen(msgget.data));
putchar('>');
fflush(stdout);
}
else{
printf("---------------------------------\n");
printf("\n%s\n",msgget.data);
printf("---------------------------------\n");
putchar('>');
fflush(stdout);
}
}
int main(int argv,char **argc)
{
//2.bind
struct sockaddr_in s_addr;
memset(&s_addr,0,sizeof(struct sockaddr_in));
//4.accept
struct sockaddr_in c_addr;
memset(&c_addr,0,sizeof(struct sockaddr_in));
//5.write
struct Msg msg;
//6.read
char readBuf[128] = {0};
//1.socket
//int socket(int domain, int type, int protocol);
int c_fd;
int n_read;
c_fd = socket(AF_INET,SOCK_STREAM,0);
if(c_fd == -1){
perror("socket");
printf("open socket failed\n");
exit(-1);
}
//2.connect
//int connect(int sockfd, const struct sockaddr *addr,socklen_t addrlen);
c_addr.sin_family = AF_INET;
c_addr.sin_port = htons(atoi(argc[2]));
inet_aton(argc[1],&c_addr.sin_addr);
if(connect(c_fd,(struct sockaddr *)&c_addr,sizeof(struct sockaddr)) == -1){
perror("connect");
exit(-1);
}
printf("connect...\n");
int mark = 0;
while(1){
memset(msg.data,0,sizeof(msg.data));
if(mark == 0) printf(">");
gets(msg.data);
if(strlen(msg.data) == 0){
if(mark == 1){
printf(">");
}
continue;
}
mark = 1;
int ret = cmd_handler(msg,c_fd);
if(ret > IFGO){
putchar('>');
fflush(stdout);
continue;
}
if(ret == -1){
printf("command not \n");
putchar('>');
fflush(stdout);
continue;
}
handler_server_message(c_fd,msg);
}
close(c_fd);
return 0;
}
总结:本次FTP的编写让我对客户端和服务器端代码编写进一步的认识,然后通过函数的调用也熟悉了一下c代码的编写,当然对于结构体的传参以及popen的使用还是比较生疏,需要巩固这一块的知识点。