Question

我已经完成了一些代码以从DNS检索地址，并且发生了崩溃。这是我的代码：

#include <ws2tcpip.h>
#include <winSock2.h>

// Need to link with Ws2_32.lib
#pragma comment(lib, "ws2_32.lib")
#include <memory>
#include <random>

bool resolveHost(const wchar_t* hostname, uint16_t port);

int main(int argc, char *argv[]) {
    WSADATA wsaData;

    WSAStartup(
      MAKEWORD(2,2),
      &wsaData
    );
    auto l = resolveHost(L"your own DNS here", 80);
}

std::unique_ptr<addrinfoW, decltype (&FreeAddrInfoW)> getAddrInfo(const wchar_t* hostname, quint16 defaultPort)
{
    addrinfoW* result;
    // zero initialization
    addrinfoW hints = {};
    hints.ai_socktype = SOCK_STREAM;
    wchar_t port[9];
    int error = swprintf_s(port, sizeof (port), L"%hu", defaultPort);
    if(error == -1)
    {
        printf("Invalid call to swprintf_s");
        return std::unique_ptr<addrinfoW, decltype (&FreeAddrInfoW)>{nullptr, &FreeAddrInfoW};
    }
    /* resolve the domain name into a list of addresses */
    error = GetAddrInfoW(hostname, port, &hints, &result);
    if (error != 0)
    {
        wprintf(gai_strerror(error));
        return std::unique_ptr<addrinfoW, decltype (&FreeAddrInfoW)>{nullptr, &FreeAddrInfoW};
    }
// here, when I return, I have a stack overflow
    return std::move(std::unique_ptr<addrinfoW, decltype (&FreeAddrInfoW)>{result, &FreeAddrInfoW});
}

bool resolveHost(const wchar_t hostname, uint16_t port)
{
    auto guard {getAddrInfo(hostname, port)};
    return true;
}

当getAddrInfo返回时，VS 2015出现堆栈溢出（我没有其他版本），我也不明白为什么。有人可以帮我吗？

Answer 1

您以字节为单位传递数组大小，而不是在swprintf_s中传递数组元素计数，这可能会导致堆栈损坏。在调试模式下，使用garbage marker 0xFE终止null后，它将填充缓冲区的其余部分。请注意，swprintf_s有适当的重载，可以减小c样式数组的大小，因此无需显式传递它。

最后一个return语句中也不应有std::move(。

尝试使用VS2015和addrinfoW

1 个答案: