recv()只读取1个字节(使用winsock实现FTP)

Question

我正在尝试使用winsock实现一个简单的FTP客户端。我在下载文件时遇到问题。这是我目前使用的代码：

bool FTPHandler::downloadFile(const char * remoteFilePath, const char * filePath) {
    if (!isConnected()) {
         setErrorMsg("Not connected, imposible to upload file...");
         return false;
     }   
     if (usePasiveMode) {
         this->pasivePort = makeConectionPasive();
         if (this->pasivePort == -1) {
             //error msg will be setted by makeConectionPasive()
             return false;
         }
     } else {
         setErrorMsg("Unable to upload file not in pasive mode :S");
         return false;
     }
     char * fileName = new char[500];
     getFileName(remoteFilePath,fileName);     
     // Default name and path := current directory and same name as remote.
     if (filePath == NULL) {
          filePath = fileName;
     }
     if (!setDirectory(remoteFilePath)) {
         return false;
     }

    char msg[OTHER_BUF_SIZE];
    char serverMsg[SERVER_BUF_SIZE];
    sprintf(msg,"%s%sn",RETR_MSG,fileName);
    send(sock, msg, strlen(msg), 0);
     SOCKET passSocket;
     SOCKADDR_IN passServer;
     passSocket = socket(PF_INET, SOCK_STREAM, IPPROTO_TCP);
     if (passSocket == INVALID_SOCKET) {
         WSACleanup();  
         sprintf(errorMsg,"Error trying to create socket (WSA error code: %d)",WSAGetLastError());
         return false;
     }
     passServer.sin_family = PF_INET;
     passServer.sin_port =   htons(this->pasivePort);
     passServer.sin_addr = *((struct in_addr *)gethostbyname(this->host)->h_addr);
     memset(server.sin_zero,0,8);
     int errorCode = connect(passSocket, (LPSOCKADDR) &passServer, sizeof(struct sockaddr));
     int tries = 0;
     while (errorCode == SOCKET_ERROR) {
           tries++;
           if (tries >= MAX_TRIES) {
               closesocket(passSocket);
               sprintf(errorMsg,"Error trying to create socket");
               WSACleanup();
               return false;
           }
     }
     char * buffer = (char *) malloc(CHUNK_SIZE);
     ofstream f(filePath);
     Sleep(WAIT_TIME);
     while (int readBytes = ***recv(passSocket, buffer, CHUNK_SIZE, 0)***>0) {
           buffer[readBytes] = '';
           f.write(buffer,readBytes);
     }
     f.close();
     Sleep(WAIT_TIME);
     recv(sock, serverMsg, OTHER_BUF_SIZE, 0);
     if (!startWith(serverMsg, FILE_STATUS_OKEY_CODE)) {
         sprintf(errorMsg,"Bad response: %s",serverMsg);
         return false;
     }
     return true;
}

最后一个recv（）多次返回1个字节，然后方法结束，应该在1Kb左右的文件只有23个字节。

为什么recv没有读取孔文件？

Answer 1

此代码中存在各种逻辑漏洞和错误/缺失错误处理。一般来说，您确实需要清理这些代码。

您将错误的sizeof()值传递给connect()，如果connect()失败，则无法正确处理错误（重试循环无效）。您需要使用sizeof(sockaddr_in)或sizeof(passServer)而不是sizeof(sockaddr)。您也没有正确初始化passServer。

您没有检查recv()是否有错误。在recv()实际读取CHUCK_SIZE字节数的可能性很小的情况下，当你将空字节写入缓冲区时，缓冲区溢出会破坏内存（你不需要这样做），因为你正在将其写入超过缓冲区的边界。

如果connect()失败，或者recv()失败并出现除服务器端启动的断开连接以外的任何错误，则不会告诉服务器中止传输。

告诉服务器进入被动模式后，在发送RETR命令之前，您需要连接到服务器告诉您的IP/端口（而不仅仅是端口）。

不要忘记向服务器发送TYPE命令，这样它就知道发送文件字节的格式，例如ASCII文本的TYPE A和二进制数据的TYPE I。如果试图以错误的格式传输文件，可能会损坏数据。FTP的默认TYPE是ASCII，而不是二进制。

最后，由于您显然不知道如何有效地对套接字进行编程，我建议您使用WinInet库的FTP部分，而不是直接使用WinSock，例如FtpGetFile()函数。让WinInet为您处理传输FTP文件的详细信息。