2012-07-12 93 views
7

我移植一个的IPv4应用到AF-独立代码库(它应该与IPv4和IPv6工作)。现在我正在使用sockaddr_storage只要我可以,但现在我必须设置(填充)sockaddr_storage。但我不知道正确的方法是什么。上面的代码是:设置的IPv4/IPv6地址和端口一个struct sockaddr_storage的结构

// defined in data_socket.h 
struct sockaddr_in laddr; 

现在有这个功能,设置sin_addrsin_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_insockaddr_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到struct sockaddr_storage的,其实我只是想这个问题的反面:Getting IPV4 address from a sockaddr structure

我试图找到最佳的解决方案,例如在这里:http://www.kame.net/newsletter/19980604/它说,从来没有使用inet_ntop()inet_pton(),但有些人(比如Beej的网络教程)说inet_ntop()inet_pton()应该用于IPv6的应用程序。

我的执行方式是否正确或应该更改?

+1

这就是为什么我喜欢'boost :: asio :: ip :: address',它将'boost :: asio :: ip :: address_v4'和'boost :: asio :: ip :: address_v6'。 – Chad 2012-07-12 14:00:16

+1

当C获得成员函数时让我知道。 – Puppy 2012-07-12 15:59:35

+0

@DeadMG哎呀抱歉'''标签 – 2012-07-12 16:04:32

回答

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); 
} 
+0

感谢您的代码。但是我对'memcpy(&this-> laddr,res-> ai_addr,res-> ai_addrlen);'line'感到困惑。 'res-> ai_addr'是一个sockaddr结构,我猜它会自动转换为'sockaddr_in'或'sockaddr_in6'?它如何存储到'this-> laddr'中(因为'this-> laddr'是'sockaddr_storage'结构)? – 2012-07-12 15:11:44

+0

我想我应该在'getaddrinfo'中使用'port_buffer'而不是'port',因为'getaddrinfo()'除了const char *(又名字符串)? – 2012-07-12 15:19:32

+1

'res-> ai_addr'指向'struct sockaddr'的一些变体,可以是'sockaddr_in'或'sockaddr_in6'。没有转换正在进行,结构只是被原样复制。 'struct sockaddr_storage'的大小和布局允许将其转换为任何'sockaddr'变体,所有这些变体都共享一个标头。重新:传递一个字符串作为端口,我不会打扰,但它是你的代码。 (另外,固定错字,getaddrinfo本来就是缓冲区) – Hasturkun 2012-07-12 15:56:20

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风格的转换远不够模糊。

相关问题