C++实现简单FTP客户端软件开发

#include "Winsock2.h"
#include "windows.h"
#include <iostream>
#include <string>
using namespace std;

#define RECV_PORT 3312    //接收端口
#define SEND_PORT 4302    //发送端口
#define DATA_PORT 3313  //数据发送端口
#pragma comment(lib, "wsock32.lib")

SOCKET sockClient, sockServer;
SOCKET dataClient, dataServer;

sockaddr_in dataAddr;
sockaddr_in severAddr;//服务器地址
sockaddr_in ClientAddr;//客户端地址
sockaddr_in dataClientAddr;//数据地址

int addrLen;        //地址长度
char fileName[20];    //文件名
char order[20];        //命令
char rbuff[1024];    //接收缓冲区
char sbuff[1024];    //发送缓冲区

char namePassword[1024] = "user 123456";    //用户名和密码

//***************函数声明***************

DWORD startSock();
DWORD createSocket();
int sendFileRecord(SOCKET datatcps, WIN32_FIND_DATA *pfd);
int sendFileList(SOCKET datatcps);
int sendFile(SOCKET datatcps, FILE* file);
DWORD connectProcess();

//***************函数声明***************
DWORD startSock() {//初始化winsock
    WSADATA WSAData;
    if (WSAStartup(MAKEWORD(2, 2), &WSAData) != 0) {
        cout << "初始化失败" << endl;
        return -1;
    }
    return 1;
}
DWORD createSocket() {
    sockClient = socket(AF_INET, SOCK_STREAM, 0);
    dataClient = socket(AF_INET, SOCK_STREAM, 0);
    if (sockClient == SOCKET_ERROR||dataClient == SOCKET_ERROR) {
        cout << "创建失败" << endl;
        WSACleanup();
        return -1;
    }
    dataAddr.sin_family = AF_INET;
    severAddr.sin_family = AF_INET;
    dataAddr.sin_addr.s_addr = htonl(INADDR_ANY);//监听任意地址
    dataAddr.sin_port = htons(DATA_PORT);
    severAddr.sin_addr.s_addr = htonl(INADDR_ANY);//监听任意地址
    //cout<<htonl(INADDR_ANY)<<endl;
    severAddr.sin_port = htons(RECV_PORT);
    if (bind(sockClient, (struct sockaddr FAR*)&severAddr, sizeof(severAddr)) == SOCKET_ERROR||bind(dataClient, (struct sockaddr FAR*)&dataAddr, sizeof(dataAddr)) == SOCKET_ERROR) {
        //bind函数用于将socket和地址结构绑定==
        cout << "绑定失败" << endl;
        return -1;
    }
    return 1;
}

DWORD mkdir(char fileName[]){
    char path[1000];
    GetCurrentDirectory(sizeof(path), path);//找到当前进程的当前目录

    strcat(path,"\\");
    strcat(path,fileName);
    cout<<path<<endl;
    bool flag = CreateDirectory(path,NULL);
    if (flag)
        cout<<"创建文件："<<fileName<<"成功"<<endl;
    else
        cout<<"创建文件："<<fileName<<"失败"<<endl;

}

DWORD delFile(char fileName[]){
    char path[1000];
    GetCurrentDirectory(sizeof(path), path);//找到当前进程的当前目录

    strcat(path,"\\");
    strcat(path,fileName);
   // cout<<path<<endl;
    bool flag = RemoveDirectory(path);
    if (flag)
        cout<<"删除文件："<<fileName<<"成功"<<endl;
    else
        cout<<"删除文件："<<fileName<<"失败"<<endl;
}
DWORD connectProcess() {
    addrLen = sizeof(ClientAddr);//addrLen是对象地址的长度

    if (listen(sockClient, 10) < 0|| listen(dataClient, 10)<0) {//让套接字进入被动监听状态，参数2为请求队列的最大长度
        cout << "监听失败" << endl;
        return -1;
    }
    cout << "服务器正在监听中…" << endl;
    while (1) {
        //accept取出队列头部的连接请求
        //sockclient是处于监听的套接字
        //ClientAddr 是监听的对象地址

        sockServer = accept(sockClient, (struct sockaddr FAR*)&ClientAddr, &addrLen);
        while (1) {

            memset(rbuff, 0, sizeof(rbuff));
            memset(sbuff, 0, sizeof(sbuff));
            if (recv(sockServer, rbuff, sizeof(rbuff), 0) <= 0) {
                continue;
            }
            cout << endl << "获取并执行的命令：" << rbuff << endl;
            if (strncmp(rbuff, "get", 3) == 0) {//将文件发给客户端
                strcpy(fileName, rbuff + 4);
                FILE* file;//定义一个文件访问指针
                //处理下载文件请求
                file = fopen(fileName, "rb");//二进制打开文件，只允许读
                if (file) {
                    sprintf(sbuff, "get %s", fileName);
                    if (!send(sockServer, sbuff, sizeof(sbuff), 0)) {
                        cout<<"发送失败"<<endl;
                        fclose(file);
                        return 0;
                    }
                    else {//创建额外数据连接传送数据
                        if (!sendFile(dataClient, file)) {
                            return 0;
                        }
                        fclose(file);
                    }
                }else {
                    strcpy(sbuff, "无法打开文件\n");
                    cout<<"无法打开文件"<<endl;
                    if (!send(sockServer, sbuff, sizeof(sbuff), 0)) {
                        return 0;
                    }
                }
            }//get
            else if (strncmp(rbuff, "put", 3) == 0) {//从客户端上传上来文件

                cout<<"debug1"<<endl;
                FILE* fd;
                int cnt;
                strcpy(fileName, rbuff + 4);
                fd = fopen(fileName, "wb");

                if (fd == NULL) {
                    cout << "无法打开文件" << fileName << endl;
                    return 0;
                }
                sprintf(sbuff, "put %s", fileName);
                if (!send(sockServer, sbuff, sizeof(sbuff), 0)) {
                    fclose(fd);
                    return 0;
                }
                memset(sbuff, '\0', sizeof(rbuff));
                int dataAddLen = sizeof (dataClientAddr);
                dataServer = accept(dataClient, (struct sockaddr FAR*)&dataClientAddr, &dataAddLen);
                while ((cnt = recv(dataServer, rbuff, sizeof(rbuff), 0)) > 0) {
                    fwrite(rbuff, sizeof(char), cnt, fd);//把cnt个数据长度为char的数据从rbuff输入到fd指向的文件
                }
                cout << "成功获得文件" << fileName << endl;
                closesocket(dataServer);
                fclose(fd);
            }//put
            else if (strncmp(rbuff, "pwd", 3) == 0) {
                char path[1000];
                GetCurrentDirectory(sizeof(path), path);//找到当前进程的当前目录
                strcpy(sbuff, path);
                send(sockServer, sbuff, sizeof(sbuff), 0);
            }//pwd
            else if (strncmp(rbuff, "ls", 2) == 0) {
                strcpy(sbuff, rbuff);
                send(sockServer, sbuff, sizeof(sbuff), 0);
                sendFileList(dataClient);
            }//dir
            else if(strncmp(rbuff, "mkdir", 5)==0){
                strcpy(fileName, rbuff + 6);
                strcpy(sbuff,rbuff);
                send(sockServer, sbuff, sizeof (sbuff),0);//发送回信息
                mkdir(fileName);
            }//mkdir
            else if(strncmp(rbuff,"del", 3)==0){
                strcpy(fileName, rbuff + 4);//获得要删的文件名
                strcpy(sbuff,rbuff);
                send(sockServer, sbuff, sizeof (sbuff),0);//发送回信息
                delFile(fileName);
            }//del
            else if (strncmp(rbuff, "cd", 2) == 0) {
                strcpy(fileName, rbuff + 3);
                strcpy(sbuff, rbuff);
                send(sockServer, sbuff, sizeof(sbuff), 0);
                char path[1000];
                GetCurrentDirectory(sizeof(path), path);//找到当前进程的当前目录
                strcat(path,"\\");
                strcat(path,fileName);
                SetCurrentDirectory(path);//设置当前目录
            }//cd
            else if (strncmp(rbuff, "user", 4) == 0) {
                char tbuff[1024];
                strcpy(tbuff, rbuff + 5);
                strcat(tbuff, " ");
                memset(rbuff, '\0', sizeof(rbuff));
                strcpy(sbuff, "成功获取用户名\0");
                send(sockServer, sbuff, sizeof(sbuff), 0);

                recv(sockServer, rbuff, sizeof(rbuff), 0);
                cout << endl << "获取并执行的命令：" << rbuff << endl;
                strcat(tbuff, rbuff + 5);
                if (strcmp(tbuff, namePassword) == 0) {//验证是否正确并返回数据给客户端
                    send(sockServer, "right\0", sizeof(sbuff), 0);
                }else {
                    send(sockServer, "wrong\0", sizeof(sbuff), 0);
                }
            }//user pass

        }
        closesocket(sockServer);
    }
}
int sendFile(SOCKET datatcps, FILE* file) {
    int dataAddLen = sizeof (dataClientAddr);
    dataServer = accept(datatcps, (struct sockaddr FAR*)&dataClientAddr, &dataAddLen);
    cout << "正在发送文件…" << endl;
    memset(sbuff, '\0', sizeof(sbuff));
    while(1) {//从文件中循环读取数据并发送至客户端
        int len = fread(sbuff, 1, sizeof(sbuff), file);//把file指针指向的文件中的内容读取到sbuff中
       //cout<<"sbuff内容："<<sbuff<<endl;
        if (send(dataServer, sbuff, len, 0) == SOCKET_ERROR) {
            cout << "连接失败" << endl;
            closesocket(dataServer);
            return 0;
        }
        if (len < sizeof(sbuff)) {//文件传送结束
            break;
        }
    }
    closesocket(dataServer);
    cout << "发送成功" << endl;
    return 1;
}
int sendFileList(SOCKET datatcps) {
    int dataAddLen = sizeof (dataClientAddr);
    dataServer = accept(datatcps, (struct sockaddr FAR*)&dataClientAddr, &dataAddLen);
    HANDLE hff;                                //建立一个线程
    WIN32_FIND_DATA fd;                        //搜索文件
    hff = FindFirstFile("*", &fd);            //查找文件来把待操作文件的相关属性读取到WIN32_FIND_DATA结构中去
    if (hff == INVALID_HANDLE_VALUE) {        //发生错误
        const char *errStr = "列出文件列表时发生错误\n";
        cout << *errStr << endl;
        if (send(dataServer, errStr, strlen(errStr), 0) == SOCKET_ERROR) {
            cout << "发送失败" << endl;
        }
        closesocket(dataServer);
        return 0;
    }
    BOOL flag = TRUE;
    while (flag) {//发送文件信息

        if (!sendFileRecord(dataServer, &fd)) {
            closesocket(dataServer);
            return 0;
        }
        flag = FindNextFile(hff, &fd);//查找下一个文件
    }
    closesocket(dataServer);
    return 1;
}
int sendFileRecord(SOCKET datatcps, WIN32_FIND_DATA *pfd) {//发送当前的文件记录
    char fileRecord[MAX_PATH + 32];

    FILETIME ft;                        //文件的建立时间
    FileTimeToLocalFileTime(&pfd -> ftLastWriteTime, &ft);//Converts a file time to a local file time.

    SYSTEMTIME lastWriteTime;
    FileTimeToSystemTime(&ft, &lastWriteTime);

    const char *dir = pfd -> dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY ? "<DIR>" : " ";
    sprintf(fileRecord, "%04d-%02d-%02d %02d:%02d %5s %10d   %-20s\n",
        lastWriteTime.wYear,
        lastWriteTime.wMonth,
        lastWriteTime.wDay,
        lastWriteTime.wHour,
        lastWriteTime.wMinute,
        dir,
        pfd -> nFileSizeLow,
        pfd -> cFileName
    );
    if (send(datatcps, fileRecord, strlen(fileRecord), 0) == SOCKET_ERROR) {
        //通过datatcps接口发送fileRecord数据，成功返回发送的字节数
        cout << "发送失败" << endl;
        return 0;
    }
    return 1;
}
int main(){

    if (startSock() == -1 || createSocket() == -1 || connectProcess() == -1) {
        return -1;
    }
    return 1;
}

#include <Winsock.h>
#include <windows.h>
#include <time.h>
#include <stdio.h>
#include <iostream>
using namespace std;

#define RECV_PORT 3312    //接收端口
#define SEND_PORT 4302    //发送端口
#define DATA_PORT 3313  //数据接受端口

#pragma comment(lib, "wsock32.lib")    //加载ws2_32.dll，它是Windows Sockets应用程序接口， 用于支持Internet和网络应用程序。

SOCKET sockClient;        //客户端对象
sockaddr_in serverAddr;    //服务器地址
sockaddr_in dataAddr;
SOCKET dataClient;      //数据对象

char inputIP[20];        //存储输入的服务器IP
char fileName[20];        //文件名
char rbuff[1024];        //接收缓冲区
char sbuff[1024];        //发送缓冲区
bool checkFlag = false;            //标志是否通过登陆

//***********************函数声明***********************
DWORD startSock();                            //启动winsock并初始化
DWORD createSocket(SOCKET &mySocket);                        //创建socket
DWORD callServer(SOCKET &mySocket,sockaddr_in addr);                            //发送连接请求

int command();                             //执行命令
void help();                                //菜单
void list(SOCKET &sockfd);                    //列出远方当前目录
DWORD sendTCP(char data[]);                    //发送要执行的命令至服务端
int user();                                    //上传用户名
int pass();                                    //上传密码
int sendFile(SOCKET &datatcps, FILE* file);    //put 传送给远方一个文件
//***********************函数声明***********************


//***********************函数定义***********************
DWORD startSock() { //启动winsock并初始化
    WSADATA WSAData;
    char a[20];
    memset(a, 0, sizeof(a));
    if (WSAStartup(MAKEWORD(2, 2), &WSAData) != 0) { //加载winsock版本
        cout << "sock初始化失败" << endl;
        return -1;
    }
    if (strncmp(inputIP, a, sizeof(a)) == 0) {
        cout << "请输入要连接的服务器IP：";
        cin >> inputIP;
    }
    //设置地址结构
    serverAddr.sin_family = AF_INET;                    //表明底层是使用的哪种通信协议来递交数据的，AF_INET表示使用 TCP/IPv4 地址族进行通信
    serverAddr.sin_addr.s_addr = inet_addr(inputIP);    //指定服务器IP，十进制转化成二进制IPV4地址
    serverAddr.sin_port = htons(RECV_PORT);                //设置端口号，htons用于将主机字节序改为网络字节序

    dataAddr.sin_family = AF_INET;
    dataAddr.sin_addr.s_addr = inet_addr(inputIP);
    dataAddr.sin_port = htons(DATA_PORT);//数据端口和控制端口不一样

    return 1;
}
DWORD createSocket(SOCKET &mySocket) { //创建socket
    //要使用套接字，首先必须调用socket()函数创建一个套接字描述符，就如同操作文件时，首先得调用fopen()函数打开一个文件。
    mySocket = socket(AF_INET, SOCK_STREAM, 0);//当scoket函数成功调用时返回一个新的SOCKET(Socket Descriptor) //SOCK_STREAM（流式套接字）:Tcp连接，提供序列化的、可靠的、双向连接的字节流。支持带外数据传输
    if (mySocket == SOCKET_ERROR) {
        cout << "创建socket失败" << endl;
        WSACleanup();//终止Ws2_32.dll 的使用
        return -1;
    }
    return 1;
}
DWORD callServer(SOCKET &mySocket,sockaddr_in addr) { //发送连接请求
    createSocket(mySocket);
    if (connect(mySocket, (struct sockaddr*)&addr, sizeof(addr)) == SOCKET_ERROR) {//connect()创建与指定外部端口的连接
        cout << "连接失败" << endl;
        memset(inputIP, 0, sizeof(inputIP));
        return -1;
    }
    return 1;
}
void help() { //帮助菜单
    cout << "        ___________________________________________  " << endl
         << "       |                FTP帮助菜单                 |   " << endl
         << "       | 1、get 下载文件 [输入格式: get 文件名 ]       |   " << endl
         << "       | 2、put 上传文件 [输入格式：put 文件名]         |   " << endl
         << "       | 3、pwd 显示当前文件夹的绝对路径                |    " << endl
         << "       | 4、ls 显示远方当前目录的文件                 |   " << endl
         << "       | 5、mkdir 新建文件夹 [输入格式：mkdir 文件名]   |   " << endl
         << "       | 6、del 删除文件夹 [输入格式：del 文件名]       |   " << endl
         << "       | 7、cd  改变远方当前目录和路径                 |   " << endl
         << "       |         进入下级目录: cd 路径名              |   " << endl
         << "       |         进入上级目录: c ..                  |   " << endl
         << "       | 8、? 或者 help 进入帮助菜单                  |   " << endl
         << "       | 9、quit 退出FTP                            |   " << endl
         << "       |___________________________________________|    " << endl;
}
DWORD sendTCP(char data[]) { //发送要执行的命令至服务端
    int length = send(sockClient, data, strlen(data), 0);
    if (length <= 0) {
        cout << "发送命令至服务端失败" << endl;
        closesocket(sockClient);//当不使用socket()创建的套接字时，应该调用closesocket()函数将它关闭，就如同调用fclose()函数关闭一个文件，用来进行套接字资源的释放。
        WSACleanup();
        return -1;
    }
    return 1;
}
int sendFile(SOCKET &datatcps, FILE* file) { //put 传送给远方一个文件
    callServer(datatcps,dataAddr);
    cout << "正在传输文件…" << endl;
    memset(sbuff, '\0', sizeof(sbuff));
    while (1) { //从文件中循环读取数据并发送
        int len = fread(sbuff, 1, sizeof(sbuff), file); //fread从file文件读取sizeof(sbuff)长度的数据到sbuff，返回成功读取的数据个数

        if (send(datatcps, sbuff, sizeof(rbuff), 0) == SOCKET_ERROR) {
            cout << "与客户端的连接中断" << endl;
            closesocket(datatcps);
            return 0;
        }
        if (len < sizeof(sbuff)) {
            break;
        }
    }
    closesocket(datatcps);
    cout << "传输完成" << endl;
    return 1;
}
void list(SOCKET &sockfd) { //列出远方当前目录
    callServer(sockfd,dataAddr);
    int nRead;
    memset(sbuff, '\0', sizeof(sbuff));
    while (1) {
        nRead = recv(sockfd, rbuff, sizeof(rbuff), 0);
        //recv通过sockClient套接口接受数据存入rbuff缓冲区，返回接收到的字节数
        if (nRead == SOCKET_ERROR) {
            cout << "读取时发生错误" << endl;
            exit(1);
        }
        if (nRead == 0) { //数据读取结束
            break;
        }
        cout<<"nRead长度"<<nRead<<endl;
        //显示数据
        rbuff[nRead] = '\0';
        cout << rbuff << endl;
    }
    cout<<"读取结束"<<endl;
    closesocket(sockfd);
}
int  user() {
    char operation[10], name[20];        //操作与文件名
    char order[30] = "\0";                //输入的命令
    char buff[80];                        //用来存储经过字符串格式化的order
    cout << "请输入用户名指令（user 用户名）：";
    strcpy(operation,"user");
    cin >> name;
    strcat(order, operation), strcat(order, " "), strcat(order, name);
    sprintf(buff, order);
    sendTCP(buff);                                    //发送指令
    recv(sockClient, rbuff, sizeof(rbuff), 0);        //接收信息
    cout << rbuff << endl;
    return 1;
}
int pass() {
    char operation[10], name[20];        //操作与文件名
    char order[30] = "\0";                //输入的命令
    char buff[80];                        //用来存储经过字符串格式化的order
    cout << "请输入密码指令（pass 密码）：" ;
    strcpy(operation,"pass");
    cin >> name;
    strcat(order, operation), strcat(order, " "), strcat(order, name);
    sprintf(buff, order);
    sendTCP(buff);                                    //发送指令
    recv(sockClient, rbuff, sizeof(rbuff), 0);        //接收信息
    cout << rbuff << endl;
    if (strcmp(rbuff, "wrong") == 0) {
        return 0;
    }
    return 1;
}
//***********************函数定义***********************

int command(){
    char operation[10], name[20];        //操作与文件名
    char order[30] = "\0";                //输入的命令
    char buff[80];                        //用来存储经过字符串格式化的order
    FILE *fd1, *fd2;                    //File协议主要用于访问本地计算机中的文件，fd指针指向要访问的目标文件
    int cnt;
    //发送连接请求成功，初始化数据
    memset(operation, 0, sizeof(operation));
    memset(name, 0, sizeof(name));
    memset(order, 0, sizeof(order));
    memset(buff, 0, sizeof(buff));
    memset(rbuff, 0, sizeof(rbuff));
    memset(sbuff, 0, sizeof(sbuff));
        cout << endl << "请输入要执行的指令: ";
        cin >> operation;
//        cout<<"why:"<<operation<<endl;
        if (strncmp(operation, "get", 3) == 0 || strncmp(operation, "put", 3) == 0 ||
            strncmp(operation, "cd", 2) == 0 || strncmp(operation, "mkdir", 5) == 0 || strncmp(operation, "del", 3) ==0){ ///需要输入文件名的功能
            cin >> name;
        } else if (strncmp(operation, "quit", 4) == 0) { ///退出功能
            cout << "感谢您的使用" << endl;
            return 1;
        } else if (strncmp(operation, "?", 1) == 0 || strncmp(operation, "help", 4) == 0) { ///帮助菜单功能
            help();
        }else if(strncmp(operation,"ls",2)==0|| strncmp(operation,"pwd",3)==0){

        }
        else{
            cout<<"非法输入"<<endl;
            return 0;
        }


        //将指令整合进order，并存放进buff
        strcat(order, operation), strcat(order, " "), strcat(order, name);

        sprintf(buff, order);
        sendTCP(buff);                                    //发送指令
        recv(sockClient, rbuff, sizeof(rbuff), 0);        //接收信息
        cout << rbuff << endl;                            //pwd功能在这里已经实现
        if (strncmp(rbuff, "get", 3) == 0) {            ///下载功能
            callServer(dataClient,dataAddr);
            fd1 = fopen(name, "wb");                    //用二进制的方式打开文件，wb表示打开或新建一个二进制文件（只允许写数据）
            if (fd1 == NULL) {
                cout << "打开或者新建 " << name << "文件失败" << endl;
                return 1;
            }
            memset(rbuff, '\0', sizeof(rbuff));
            while ((cnt = recv(dataClient, rbuff, sizeof(rbuff), 0)) > 0) {
               // cout<<"缓冲区"<<rbuff<<endl<<"长度"<<cnt<<endl;
                fwrite(rbuff, sizeof(rbuff), 1, fd1);    //C 库函数 size_t fwrite(const void *ptr, size_t size, size_t nmemb, FILE *stream) 把 ptr 所指向的数组中的数据写入到给定流 stream 中。
            }
            closesocket(dataClient);
            fclose(fd1);                                //关闭文件
        }//get
        else if (strncmp(rbuff, "put", 3) == 0) { ///上传功能
          //  cout<<"执行"<<endl;
            strcpy(fileName, rbuff + 4);
            fd2 = fopen(fileName, "rb");                //打开一个二进制文件，文件必须存在，只允许读
            if (fd2) { //成功打开
               // cout<<"debug0"<<endl;
                if (!sendFile(dataClient, fd2)) {
                    cout << "发送失败" << endl;
                    return 1;
                }
                fclose(fd2);
            } else {
                strcpy(sbuff, "无法打开文件\n");
                if (!send(sockClient, sbuff, sizeof(sbuff), 0)) {
                    return 1;
                }
            }

        }//put
        else if (strncmp(rbuff, "ls", 2) == 0) { ///ls功能
            list(dataClient);
        }//dir

        return 0;

}

int main() {
    while (1) {
        startSock();                //启动winsock并初始化
        if (callServer(sockClient,serverAddr) == -1) {    //发送连接请求失败
            continue;
        }
        cout << "发送连接请求成功" << endl;

//        if (checkFlag == false) {//登陆
//            if (user() && pass()) {
//                checkFlag = true;
//            }
//            continue;
//        }
        checkFlag = true;

        if(checkFlag){
            help();
        }
        while(checkFlag){
            if(command()){
                break;
            }
        }
        cout<<"命令输入结束"<<endl;
        closesocket(sockClient);    //关闭连接
        WSACleanup();                //释放Winsock

        return 0;

    }
}
/*
192.168.0.100
user gyc
pass 123456
pwd
cd Debug
get 110.txt
*/