套接字下载太少或太多的网页

时间:2012-11-16 03:11:30

标签: c++ sockets winsock2

为什么我的代码只下载了半个网页?有时它下载网页大小的4倍:S

我找不到有什么问题,这就是我要问的原因。基本上,我连接到套接字,发送我的请求并将响应读取到缓冲区。我尝试将其保存到文件并将其打印到屏幕上,但它会打印并保存不完整的数据或过多的数据。我不确定它是否是缓冲区溢出或我做错了什么。

有什么想法吗?

#define _WIN32_WINNT 0x501

#include <iostream>
#include <winsock2.h>
#include <ws2tcpip.h>
#include <stdio.h>
#include <fstream>

using namespace std;

void Get(string WebPage)
{
    WSADATA wsaData;
    string Address;
    struct addrinfo *result;
    struct sockaddr_in  *sockaddr_ipv4;

    char Buffer[50000] = {0};

    string Header = "GET / HTTP/1.1\r\n";
    Header += "Host: " + WebPage + "\r\n";
    Header += "Connection: close\r\n";
    Header += "\r\n";

    if (WSAStartup(MAKEWORD(2,2), &wsaData) != 0) return;

    SOCKET Socket = socket(AF_INET,SOCK_STREAM,IPPROTO_TCP);

    getaddrinfo(WebPage.c_str(), NULL, NULL, &result);
    if (result->ai_family == AF_INET)
    {
        sockaddr_ipv4 = (struct sockaddr_in *) result->ai_addr;
        Address = inet_ntoa(sockaddr_ipv4->sin_addr);
    }
    freeaddrinfo(result);


    SOCKADDR_IN SockAddr;
    memset(&SockAddr, 0, sizeof(SockAddr));
    SockAddr.sin_port = htons(80);
    SockAddr.sin_family = AF_INET;
    SockAddr.sin_addr.s_addr = inet_addr(Address.c_str());

    if(connect(Socket,(SOCKADDR*)(&SockAddr),sizeof(SockAddr)) == SOCKET_ERROR) return;

    if (send(Socket, Header.c_str(), Header.size(), 0) == SOCKET_ERROR) return;
    shutdown(Socket, SD_SEND);

    std::string Response;

    while(true)
    {
        int Val = recv(Socket, Buffer, sizeof(Buffer), 0);
        if (Val == 0)
            break;
        else if (Val == SOCKET_ERROR)
        {
            cout<<"Error!";
        }
        else
        {
            Response += Buffer;
        }
    }

    closesocket(Socket);
    WSACleanup();

    ofstream File;
    File.open("C:/Saved.html");
    File<<Response;
    File.close();
}

int main()
{
    Get("villavu.com");
}

1 个答案:

答案 0 :(得分:3)

编辑:recv不是null为您终止数据 - 您需要写下您收到的数据量,而不仅仅是+ =它。


您的回复中是否有二进制数据?如果是这样,

Response += Buffer;

将停在第一个空字符处。我会使用向量来存储来自recv的数据:

vector<char> recvBuffer(50000);

int bytesReceived = recv(socket, recvBuffer.data(), recvBuffer.size(), 0);
//error checking
recvBuffer.resize(bytesReceived);

再次将收到的数据存储在另一个矢量中,然后将其复制回来。

vector<char> pageContents;

pageContents.insert(pageContents.end(), recvBuffer.begin(), recvBuffer.end());

但这并不能解释你的4x数据。

我看到的另一个问题是,在使用缓冲区后,您不会将其清零。

IOW:您需要写下您收到的数据量,而不仅仅是+ =数组。