自己定义一个.h文件,#include “config.h”
config.h文件
#define LS 0 //宏定义
#define GET 1
#define PWD 2
#define IFGO 3
#define LCD 4
#define LLS 5
#define CD 6
#define PUT 7
#define QUIT 8
#define DOFILE 9
struct Msg
{
int type;
char data[1024]; //存放指令名称
char secondBuf[128]; //存放内容
};
FTP服务器
server.c
#include <stdio.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <stdlib.h>
#include <string.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <unistd.h>
#include "config.h"
/*解析指令*/
int get_cmd_type(char *cmd)
{
if(!strcmp("ls",cmd)) return LS; //0 字符串比较,两个字符串相等时返回0,这里取!,即为1
if(!strcmp("quit",cmd)) return QUIT;//8
if(!strcmp("pwd",cmd)) return PWD; //2
if(strstr(cmd,"cd")!=NULL) return CD; //6字符串中存在cd
if(strstr(cmd,"get")!=NULL) return GET;//1客户端输入的cmd中存在get
if(strstr(cmd,"put")!=NULL) return PUT; //7客户端输入的cmd中存在put
return 100;
}
/*字符串提取*/
char *getDesDir(char *cmsg)
{
char *p;
p = strtok(cmsg," "); //以空格来分隔,获取第一个字符串;将第一个空格修改为了'/0',然后返回第一个字符串
p = strtok(NULL," "); //获取第二个串开始,第一个参数都要写成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); //把字符串传到get_cmd_type转换成我们想要的整型数
switch(ret){
case LS: //列出服务端当前文件夹所有文件
case PWD: //列出服务端当前路径
msg.type = 0;
FILE *r = popen(msg.data, "r"); //打开
fread(msg.data, sizeof(msg.data), 1, r); //把msg结果读出来,这里的r是文件描述符是指针
write(fd, &msg, sizeof(msg)); //写到fd(fd是socket通道),发送给客户端
break;
case CD: //进入文件夹,需要分割字符串
msg.type = 1;
//分割字符串获取到cd后面的字符串
char *dir = getDesDir(msg.data); //调用getDesDir来获取cd后面的字符串
printf("dir:%s\n",dir);
chdir(dir); //chdir 是C语言中的一个系统调用函数(同cd),用于改变当前工作目录,其参数为 dir 目标目录
break;
case GET: //获取文件
file = getDesDir(msg.data); //调用getDesDir来获取cd后面的文件(字符串)
if(access(file, F_OK) == -1){ //判断文件是否存在
strcpy(msg.data, "No This File!"); //如果没有此文件,把No This File!拷贝到msg的信息,提示客户没有
write(fd, &msg, sizeof(dataBuf)); //返回给客户端
}else{
msg.type = DOFILE; //如果有此文件,让客户端创文件
fdfile = open(file,O_RDWR); //打开此文件
read(fdfile, dataBuf, sizeof(dataBuf)); //把文件所有信息读到dataBuf
close(fdfile);
strcpy(msg.data,dataBuf); //把dataBuf拷贝到想要发出去的消息的msg结构体里的data数组里
write(fd, &msg, sizeof(msg)); //把读到的文件信息发回去
}
break;
case PUT: //放入文件
fdfile = open(getDesDir(msg.data),O_RDWR|O_CREAT, 0666);//打开客户端发送过来的文件,如果客户端没有此文件则创建此文件
write(fdfile, msg.secondBuf, strlen(msg.secondBuf));//此时msg结构体的data存放的是指令,secondBuf存放的是文件
的内容
close(fdfile);
break;
case QUIT://退出
printf("client Quit!\n");
exit(-1);
}
}
int main(int argc,char **argv)
{
int s_fd;
int c_fd;
int n_read;
char readBuf[128];
struct sockaddr_in s_addr;
struct sockaddr_in c_addr;
struct Msg msg;
if(argc != 3){
printf("param is not good\n");
exit(-1);
}
memset(&s_addr, 0, sizeof(struct sockaddr_in));
memset(&c_addr, 0, sizeof(struct sockaddr_in));
//1. socket
s_fd = socket(AF_INET, SOCK_STREAM, 0);
if(s_fd == -1){
perror("socket");
exit(-1);
}
s_addr.sin_family = AF_INET;
s_addr.sin_port = htons(atoi(argv[2]));
inet_aton(argv[1], &s_addr.sin_addr);
//2. bind
bind(s_fd, (struct sockaddr *)&s_addr, sizeof(struct sockaddr_in));
//3. listen
listen(s_fd, 10);
//4. accept连接
int clen = sizeof(struct sockaddr_in);
while(1){
c_fd = accept(s_fd, (struct sockaddr *)&c_addr,&clen);
if(c_fd == -1){
perror("accept");
}
printf("get connect:%s\n",inet_ntoa(c_addr.sin_addr));
if(fork() == 0){ //当新的客户端接入时创建一个子进程
while(1){
memset(msg.data, 0, sizeof(msg.data)); //清空数据
n_read = read(c_fd, &msg, sizeof(msg)); //首先是读取客户端传过来的指令,读不到就等待
if(n_read == 0){ //当读到的字节数为0时,代表客户端已经退出
printf("client out\n");
break;
}else if(n_read > 0){ //当读到的字节数>0代表读到真正有效的数据,
msg_handler(msg, c_fd); //msg和c_fd传到msg_handler函数
}
}
}
}
close(c_fd);
close(s_fd);
return 0;
}
FTP客户端
client.c
#include <stdio.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <stdlib.h>
#include <string.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <unistd.h>
#include "config.h"
/*字符串处理*/
char * getdir(char *cmd)
{
char *p;
p = strtok(cmd," "); //以空格来分隔,获取第一个字符串
p = strtok(NULL," "); //获取第二个串开始,第一个参数都要写 NULL
return p;
}
int get_cmd_type(char *cmd)
{
if(!strcmp("quit",cmd)) return QUIT; //两字符串相等返回0,取非
if(!strcmp("ls",cmd)) return LS;
if(!strcmp("lls",cmd)) return LLS;
if(!strcmp("pwd",cmd)) return LS;
if(strstr(cmd,"lcd")!=NULL) return LCD;
if(strstr(cmd,"cd")!=NULL) return CD;
if(strstr(cmd,"get")!=NULL) return GET;
if(strstr(cmd,"put")!=NULL) return PUT;
return -1; //没有符合的字符串时返回-1
}
/*指令相应的操作*/
int cmd_handler(struct Msg msg, int fd)
{
char *dir = NULL;
char buf[32];
int ret;
int filefd;
ret = get_cmd_type(msg.data); //把字符串变成整型数
switch(ret){
case LS:
case CD:
case PWD:
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)); //如果有此文件,把数据读到msg结构体的secondBuf
close(filefd);
write(fd, &msg, sizeof(msg)); //发送给服务端
}
break;
case LLS: //列出客户端当前文件夹所有文件
system("ls");
break;
case LCD: //进入客户端XX文件夹
dir = getdir(msg.data);
chdir(dir);
break;
case QUIT: //退出
strcpy(msg.data, "quit");
write(fd, &msg, sizeof(msg));
close(fd);
exit(-1);
}
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("server is out,quit\n");
exit(-1);
}
else if(msgget.type == DOFILE){ //服务器接收到get指令回返回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 argc,char **argv)
{
int c_fd;
struct sockaddr_in c_addr;
struct Msg msg;
memset(&c_addr, 0, sizeof(struct sockaddr_in));
if(argc != 3){ //参数判断
printf("param is not good\n");
exit(-1);
}
//1. socket
c_fd = socket(AF_INET, SOCK_STREAM, 0);
if(c_fd == -1){
perror("socket");
exit(-1);
}
c_addr.sin_family = AF_INET;
c_addr.sin_port = htons(atoi(argv[2]));
inet_aton(argv[1], &c_addr.sin_addr);
//2. connect,客户端不需要去创建子进程
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); //接入以后通过while不断获取用户的输入,存放在结构体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");
printf(">");
fflush(stdout);
continue;
}
handler_server_message(c_fd, msg);
}
return 0;
}