9

我正在将IPv4应用程序移植到独立于 AF 的代码库（它应该与 IPv4 和 IPv6 一起使用）。现在我尽可能使用sockaddr_storage ，但是现在我必须设置（填充）一个sockaddr_storage。但我不知道正确的方法是什么。之前的代码是：

// defined in data_socket.h
struct sockaddr_in laddr;

现在有一个设置sin_addr和sin_port的函数：

void DataSocket::SetLocalAddr(const char *addr, const int port)
{
    this->laddr.sin_port = htons(port);
    if(addr != NULL)
        this->laddr.sin_addr.s_addr = inet_addr(addr);
    else
        this->laddr.sin_addr.s_addr = inet_addr("0.0.0.0");
}

如您所见，这是旧样式（使用 IPv4）。

现在我的更改如下。首先，我已更改sockaddr_in为sockaddr_storage

// defined in data_socket.h
struct sockaddr_storage laddr;

然后我更改了上面的代码以支持IPv4和IPv6：

void DataSocket::SetLocalAddr(const char *addr, const int port)
{ 
switch (this->GetAddrFamily(addr)) {
    case AF_INET:
        (struct sockaddr_in *) this->laddr.sin_port = htons(port);
        if(addr != NULL)
            inet_pton(AF_INET, addr, (struct sockaddr_in *) this->laddr.sin_addr);
        else
            inet_pton(AF_INET, "0.0.0.0", (struct sockaddr_in *) this->laddr.sin_addr);
        break;

    case AF_INET6:
        (struct sockaddr_in6 *) this->laddr.sin6_port = htons(port);
        if(addr != NULL)
            inet_pton(AF_INET6, addr, (struct sockaddr_in6 *) this->laddr.sin6_addr);
        else
            inet_pton(AF_INET6, "0:0:0:0:0:0:0:0", (struct sockaddr_in6 *) this->laddr.sin6_addr);
        break;

    default:
        return NULL;

}

}

在哪里GetAddrFamily()：

int DataSocket::GetAddrFamily(const char *addr)
{
    struct addrinfo hints, *res;
    int status, result;

    memset(&hints, 0, sizeof(hints));
    hints.ai_family = AF_UNSPEC;
    hints.ai_socktype = SOCK_STREAM;

    if ((status = getaddrinfo(addr, 0, &hints, &res)) != 0)
    {
        fprintf(stderr, "getaddrinfo: %s\n", gai_strerror(status));
        return false;
    }

    result = res->ai_family; // This might be AF_INET, AF_INET6,etc..
    freeaddrinfo(res); // We're done with res, free it up

    return result;
}

看来我的方法是复杂的。这是正确的方法吗？因为我已将 sockaddr_in 更改为 sockaddr_storage，实际上我只是想要这个问题的反面：Getting IPV4 address from a sockaddr structure

我试图找到最好的解决方案，例如这里：http ://www.kame.net/newsletter/19980604/它说永远不要使用inet_ntop()and ，但是其他一些（比如 Beej的inet_pton()网络教程）说应该使用用于基于IPv6的应用程序。inet_ntop()inet_pton()

我的实施方式是正确的还是应该改变它？

4

2 回答 2

5

我强烈建议让getaddrinfo所有繁重的工作，例如。

void DataSocket::SetLocalAddr(const char *addr, const unsigned short int port)
{
    struct addrinfo hints, *res;
    int status;
    char port_buffer[6];

    sprintf(port_buffer, "%hu", port);

    memset(&hints, 0, sizeof(hints));
    hints.ai_family = AF_UNSPEC;
    hints.ai_socktype = SOCK_STREAM;
    /* Setting AI_PASSIVE will give you a wildcard address if addr is NULL */
    hints.ai_flags = AI_NUMERICHOST | AI_NUMERICSERV | AI_PASSIVE;

    if ((status = getaddrinfo(addr, port_buffer, &hints, &res) != 0)
    {
        fprintf(stderr, "getaddrinfo: %s\n", gai_strerror(status));
        return;
    }

    /* Note, we're taking the first valid address, there may be more than one */
    memcpy(&this->laddr, res->ai_addr, res->ai_addrlen);

    freeaddrinfo(res);
}
于 2012-07-12T14:43:19.197 回答
1

如果我理解正确，您将投this给第一个成员？不要那样做，命名成员。

我还将通过{}为这两种情况引入范围和局部变量来使其更易于阅读，例如：

{
 struct sockaddr_in * in4 = reinterpret_cast< struct sockaddr_in * >(&this->addr);
 in4->laddr.sin_port = htons(port);
 ... etc
}

由于您使用的是 C++ 而不是 C，因此请使用 C++ 样式转换。C++ 中的 C 风格转换非常含糊。

于 2012-07-12T14:41:55.830 回答