Linux网络编程--FTP云盘项目

FTP云盘

FTP就是文件传输协议。用于互联网双向传输，控制文件下载空间在服务器复制文件从本地计算机或本地上传文件复制到服务器上的空间。

server

#include <stdio.h>
#include <sys/wait.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <sys/stat.h>
#include <fcntl.h>
​
​
void choosecmd(char *cmd,int c_fd);
int choose(char * cmd);
char *Fenge(char cmd[128]);
void Putbuf(char cmd[128],int  c_fd);
​
int main(int argc,char **argv)
{
        int s_fd;
        int c_fd;
        char readbuf[128];
                char msg[128] = {0};
​
        struct sockaddr_in s_addr;
        struct sockaddr_in c_addr;
​
        memset(&s_addr,0,sizeof(struct sockaddr_in));
        memset(&c_addr,0,sizeof(struct sockaddr_in));
​
        if(argc != 3){
                printf("param isn't right\n");
        }
​
        //1.socket
        s_fd = socket(AF_INET,SOCK_STREAM,0);
        if(s_fd == -1){
                perror("socked");
                exit(-1);
        }
​
        s_addr.sin_family = AF_INET;
        s_addr.sin_port = htons(atoi(argv[2]));
        //s_addr.sin_addr.s_addr = "127.0.0.1";  (error)
        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);
​
        printf("wait connect.....\n");
​
        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));//打印客户端的ip地址，因为是网络格式，我们要转换成字符串格式
​
                if(fork() > 0)
                {
                        wait(NULL);//fangzhi z+
                }
​
                if(fork() == 0)
                {
                        while(1)
                        {
                                memset(readbuf,0,sizeof(readbuf));
                                read(c_fd,readbuf,sizeof(readbuf));
                                choosecmd(readbuf,c_fd);//选择你要执行的哪条指令
​
                        }
                }
        }
​
        return 0;
}
​
void Putbuf(char cmd[128],int  c_fd)
{
        int fd;
        char *p;
        char putbuf[8000] = {0};
​
        p = Fenge(cmd);
        read(c_fd,putbuf,sizeof(putbuf));
​
        fd = open(p,O_CREAT|O_RDWR,0600);
        write(fd,putbuf,sizeof(putbuf));
        close(fd);
        memset(putbuf,0,sizeof(putbuf));
        memset(p,0,sizeof(p));
        printf("-------------已接收----------\n");
}
​
char *Fenge(char cmd[128])
{
        char *p;
        p = strtok(cmd," ");
        p = strtok(NULL," ");
​
        return p;
}
​
​
int choose(char * cmd)
{
        if(!strcmp(cmd,"ls"))
        {
                return 1;
        }
        else if(strstr(cmd,"cd")!=NULL )
        {
                return 2;
        }
        else if(strstr(cmd,"get")!=NULL )
        {
                return 3;
        }
        else if(strstr(cmd,"put")!=NULL )
        {
                return 4;
        }
        if(!strcmp(cmd,"lls"))
        {
                return 5;
        }
        if(!strcmp(cmd,"lcd"))
        {
                return 6;
        }
        if(!strcmp(cmd,"close"))
        {
                return 7;
        }
}
​
​
void choosecmd(char *cmd,int c_fd)
{
        int ret = choose(cmd);//判断该指令是否存在
        FILE *fd;
        int fd1;
        char lsbuf[8000] = {0};
        char getbuf[8000] = {0};
        char *p = (char*)malloc(1024);
        char *p1 = (char*)malloc(1024);
​
​
        switch(ret)
        {
                case 1:
                        fd = popen("ls","r");
                        fread(lsbuf,sizeof(lsbuf),1,fd);
                        write(c_fd,lsbuf,sizeof(lsbuf));
                        memset(lsbuf,0,sizeof(lsbuf));
                        printf("----------ls succeed------\n");
                        break;
​
                case 2:
                        p = Fenge(cmd);
                        chdir(p);//执行到这个目录
                        memset(p,0,sizeof(p));
                        printf("------------进入成功--------\n");
                        break;
                case 3:
                        p1 = Fenge(cmd);
                        if(access(p1,F_OK) == -1)//判断该文件是否存在
                        {
                                exit(-1);
                        }
                        else
                        {
                                fd1 = open(p1,O_RDWR);
                                read(fd1,getbuf,sizeof(getbuf));
                                write(c_fd,getbuf,sizeof(getbuf));
                                close(fd1);
                                memset(p1,0,sizeof(p1));
                                memset(getbuf,0,sizeof(getbuf));
                                printf("-------------已发送----------\n");
                        }
​
                        break;
                case 4:
                        Putbuf(cmd,c_fd);
                        break;
                case 5:
                        printf("------------lls---------\n");
                        break;
                case 6:
                        printf("------------lcd---------\n");
                        break;
                case 7:
                        printf("客户端已退出......\n");
​
                default:
                        break;
​
        }
}       

client

#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 <unistd.h>
#include <sys/stat.h>
#include <fcntl.h>
​
​
int choose(char *cmd);
void choosecmd(char *cmd,int c_fd);
char * Fenge(char cmd[128]);
void Putbuf(char cmd[128],int c_fd);
​
int main(int argc,char **argv)
{
​
        if(argc != 3)
        {
                printf("error");
                exit(-1);
​
        }
​
        int c_fd;
        char writebuf[1024] ={0};
​
        struct sockaddr_in c_addr;
        memset(&c_addr,0,sizeof(struct sockaddr_in));
​
        //1.socket
        c_fd = socket(AF_INET,SOCK_STREAM,0);
        if(c_fd == -1){
                perror("socked");
                exit(-1);
        }
​
        c_addr.sin_family = AF_INET;
        c_addr.sin_port = htons(atoi(argv[2]));
        //s_addr.sin_addr.s_addr = "127.0.0.1";  //(不是这样写的)
        inet_aton(argv[1],&c_addr.sin_addr);
​
        //2.connect
        if(connect(c_fd,(struct sockaddr *)&c_addr,sizeof(struct sockaddr_in)) == -1){
                perror("connect");
                exit(-1);
        }
​
        //3.send
        while(1)
        {
              memset(writebuf,0,sizeof(writebuf));
              printf("input:");
              gets(writebuf);
              printf("--->:\n");
              write(c_fd,writebuf,sizeof(writebuf));
              choosecmd(writebuf,c_fd);
​
      }
​
        return 0;
}
​
void Putbuf(char cmd[128],int c_fd)
{
        int fd;
        char *p;
        char putbuf[8000] = {0};
        p = Fenge(cmd);
​
        if(access(p,F_OK) == -1)//判断文件是否存在
        {
                printf("NO file\n");
        }
        else
        {
                fd = open(p,O_RDWR,0600);
                read(fd,putbuf,sizeof(putbuf));
                write(c_fd,putbuf,sizeof(putbuf));
                close(fd);
                memset(putbuf,0,sizeof(putbuf));
                memset(p,0,sizeof(p));
                printf("------------上传成功----------\n");
        }
​
}
​
char * Fenge(char cmd[128])
{
        char * p;
        //p = (char *)malloc(1024);
        p = strtok(cmd," ");
        p  = strtok(NULL," ");
​
        return p;
​
}
​
int choose(char cmd[128])
{
        if(!strcmp(cmd,"ls"))//看服务器的所有文件
        {
                return 1;
        }
        else if(!strcmp(cmd,"lls"))//看客户端的所有文件
        {
                return 2;
        }
        else if(strstr(cmd,"lcd")!=NULL)//进入客户端的其他文件目录
        {
​
                return 3;
        }
        else if(strstr(cmd,"cd")!=NULL)//服务器进入其他文件目录
        {
​
                return 4;
        }
        else if(strstr(cmd,"get")!=NULL)//下载文件
        {
​
                return 5;
        }
        else if(strstr(cmd,"put")!=NULL)//上传文件
        {
​
                return 6;
        }
        if(!strcmp(cmd,"close"))//close server
        {
                return 7;
        }
​
}
​
void choosecmd(char *cmd,int c_fd)
{
        int fd;
        int ret = choose(cmd);//判断该指令是否存在
        char lsbuf[8000] = {0};
        char getbuf[8000] = {0};
        char * p = (char *)malloc(128);
        char * p1 = (char *)malloc(128);
        switch(ret)
        {
​
        case 1:
                read(c_fd,lsbuf,sizeof(lsbuf));
                printf("%s\n",lsbuf);
                memset(lsbuf,0,sizeof(lsbuf));
                break;
        case 2:
                system("ls");
                break;
​
​
        case 3:
                p = Fenge(cmd);
                chdir(p);
                memset(p,0,sizeof(p));
                break;
        case 4:
                printf("-------------服务器-------------\n");
                break;
        case 5:
                p1 = Fenge(cmd);
                read(c_fd,getbuf,sizeof(getbuf));
                fd = open(p1,O_RDWR|O_CREAT,0600);
                write(fd,getbuf,sizeof(getbuf));
                close(fd);
                printf("get success!\n");
                memset(p1,0,sizeof(p1));
                memset(getbuf,0,sizeof(getbuf));
                printf("-------------下载成功--------------\n");
                break;
        case 6:
                Putbuf(cmd,c_fd);
                break;
        case 7:
                close(c_fd);
                printf("-----------close succeed--------\n");
                exit(0);
        default:
                printf("input error!\n");
                break;
​
        }
​
}