linuxC语言tftp文件服务站程序编写

linux文件服务站类似于tftp

功能：ls 查看服务端文件列表
pwd查看服务端路径
cd 切换服务端路径
get 从服务端获取文件
put上传文件到服务端
quit客户端退出
lls查看客户端文件列表
lcd 客户端切换路径
lpwd查看客户端路径
1.客户端 client.c

#include <stdlib.h>
#include <sys/types.h>          /* See NOTES */
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <string.h>
#include "config.h"
#include <sys/stat.h>
#include <fcntl.h>
int get_cmd(char *cmd){
        if(!strcmp(cmd,"ls")) return LS;
        if(!strcmp(cmd,"pwd")) return PWD;
        if(!strcmp(cmd,"quit")) return QUIT;
        if(!strcmp(cmd,"lls"))  return LLS;
        if(!strcmp(cmd,"lpwd")) return LPWD;
        if(strstr(cmd,"cd")!=NULL) return CD;
        if(strstr(cmd,"get")!=NULL) return GET;
        if(strstr(cmd,"put")!=NULL) return PUT;
        if(strstr(cmd,"lcd")!=NULL) return LCD;
        return -1;
}
char *getdir(char *cmd){
        char *p=NULL;
        p=strtok(cmd," ");
        p=strtok(NULL," ");
        return p;
}

int cmd_handler(struct Msg msg,int c_fd){
        int ret=get_cmd(msg.data);
        char *dir=NULL;

        int fd;
        char buf[1024]={0};
        switch(ret){
                case PWD:
                case LS:
                case CD:
                        msg.type=1;
                        write(c_fd,&msg,sizeof(msg));
                        break;
                case GET:
                        msg.type=2;
                           write(c_fd,&msg,sizeof(msg));
                        break;
                case PUT:
                        strcpy(buf,msg.data);
                        dir=getdir(buf);
                        if(access(dir,F_OK)==-1){
                                printf("dir : %sno this file\n",dir);


                        }else{
                                fd=open(dir,O_RDWR);
                                read(fd,msg.secondBuf,sizeof(msg.secondBuf));
                                close(fd);

                                write(c_fd,&msg,sizeof(msg));
                        }
                        break;
                case LLS:
                        system("ls");
                        break;
                case LPWD:
                        system("pwd");
                        break;
                case LCD:
                        dir=getdir(msg.data);
                        chdir(dir);
                        break;
                case QUIT:
                        write(c_fd,&msg,sizeof(msg));
                        exit(-1);
                        break;

        }
        return ret;
}
void handler_server_message(int c_fd,struct Msg msg){
        struct Msg msgget;
        char *dir=NULL;
        int newfd;
        memset(&msgget,0,sizeof(msgget));
        int n_read=read(c_fd,&msgget,sizeof(msgget));
        if(n_read==0){
                printf("the server is quit\n");
                exit(-1);
        }
        if(msgget.type==DOFILE){
                dir=getdir(msg.data);
                newfd=open(dir,O_RDWR|O_CREAT,0666);
                write(newfd,msgget.data,strlen(msgget.data));
                close(newfd);
                putchar('>');
                fflush(stdout);

        }else{
                printf("----------------------------\n");
                printf("\n%s\n",msgget.data);
                printf("----------------------------\n");
                putchar('>');
                fflush(stdout);
        }


}
int main(int argc,char **argv){
        if(argc!=3){
                perror("canshu:");
                exit(-1);

        }
        int c_fd;
        struct Msg msg;
        struct sockaddr_in c_addr;
        memset(&c_addr,0,sizeof(struct sockaddr_in));
        c_fd=socket(AF_INET,SOCK_STREAM,0);
        c_addr.sin_family=AF_INET;
        c_addr.sin_port=htons(atoi(argv[2]));
        inet_aton(argv[1],&c_addr.sin_addr);
        connect(c_fd,(struct sockaddr*)&c_addr,sizeof(struct sockaddr_in));
        printf("connect...\n");
        int mark=0;
        while(1){

                memset(&msg,0,sizeof(msg));
                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;
                }
                else if(ret==-1){
                        printf("no this cmd\n");
                        printf(">");
                        fflush(stdout);
                        continue;

                }else{
                     handler_server_message(c_fd,msg);
                }
        }




        return 0;
}
                                                                                                                                                                                                                                

2.服务端 server.c

2. 服务端 server.c
#include <stdio.h>
#include <stdlib.h>
#include <sys/types.h>          /* See NOTES */
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <string.h>
#include "config.h"
#include <sys/stat.h>
#include <fcntl.h>
int cmd_get(char *cmd){
        if(!strcmp(cmd,"ls")) return LS;
        if(!strcmp(cmd,"pwd")) return PWD;
        if(!strcmp(cmd,"quit")) return QUIT;

        if(strstr(cmd,"cd")!=NULL) return CD;
        if(strstr(cmd,"get")!=NULL) return GET;
        if(strstr(cmd,"put")!=NULL) return PUT;
        return 100;
}
char *getdir(char *cmd){
        char *p=NULL;
        p=strtok(cmd," ");
        p=strtok(NULL," ");
        return p;
}
void cmd_handler(struct Msg msg,int c_fd){
        int ret=cmd_get(msg.data);
        FILE *r;
        char *dir=NULL;
        char buf[1024]={0};
        int fd;
        printf("cmd:%s\n",msg.data);
        switch(ret){

                case LS:
                case PWD:
                        r=popen(msg.data,"r");

                        fread(msg.data,sizeof(msg.data),1,r);
                        fclose(r);
                        write(c_fd,&msg,sizeof(msg));
                        break;
                case QUIT:
                                printf("client quit\n");
                        exit(-1);
                        break;
                case CD:
                        dir=getdir(msg.data);
                        chdir(dir);
                        break;
                case GET:
                        dir=getdir(msg.data);

                        if(access(dir,F_OK)==-1){
                                strcpy(msg.data,"NO this file") ;
                                write(c_fd,&msg,sizeof(msg));
                        }else{
                                msg.type=DOFILE;
                                fd=open(dir,O_RDWR);
                                read(fd,msg.data,sizeof(msg.data));
                                close(fd);
                                write(c_fd,&msg,sizeof(msg));
                        }
                        break;
                case PUT:

                        dir=getdir(msg.data);

                        fd=open(dir,O_RDWR|O_CREAT,0666);
                        write(fd,msg.secondBuf,sizeof(msg.secondBuf));
                        close(fd);
                        break;

        }

}
int main(int argc,char **argv){

        int s_fd;
        int c_fd;
        int n_read;
        struct sockaddr_in s_addr;
        struct sockaddr_in c_addr;
        struct Msg msg;
        memset(&s_addr,0,sizeof(struct sockaddr_in));
        memset(&s_addr,0,sizeof(struct sockaddr_in));
        s_fd=socket(AF_INET,SOCK_STREAM,0);
        s_addr.sin_family=AF_INET;
        s_addr.sin_port=htons(atoi(argv[2]));
        inet_aton(argv[1],&s_addr.sin_addr);
        bind(s_fd,(struct sockaddr*)&s_addr,sizeof(struct sockaddr_in));
        listen(s_fd,10);


        int len=sizeof(struct sockaddr_in);
        while(1){
                c_fd=accept(s_fd,(struct sockaddr*)&c_addr,&len);
                if(c_fd==-1){
                        perror("connect:");
                        exit(-1);
                }
                printf("get connect IP:%s\n",inet_ntoa(s_addr.sin_addr));
                if(fork()==0){
                        while(1){

                                memset(&msg,0,sizeof(msg));
                                n_read=read(c_fd,&msg,sizeof(msg));
                                if(n_read==0){
                                        printf(" client quit\n");
                                        break;
                                }else{
                                        cmd_handler(msg,c_fd);
                                }
                        }

                }

        }
                                                                                                                                                          
  return 0;
}

3. 头文件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
#define LPWD  10
struct Msg{
        int type;
        char data[1024];
        char secondBuf[1024];
};