如何使用'getaddrinfo'为所有接口选择默认空闲端口?

时间:2013-10-17 15:47:20

标签: c++ ipv6 ipv4 getaddrinfo

我正在尝试让服务器在双栈模式下侦听IPv4和IPv6。

我想要IPv4和IPv6服务器的相同端口号,我希望它在随机选择的端口上(使用端口“0”)

当我绑定时,每个服务器获得不同的端口,我希望它是相同的。

所以我认为它应该由getaddrinfo函数来完成。

但是当我给它“0”端口时,它在addrinfo结果中保持“0”,这是因为每个绑定都给了我不同的端口号。

我的问题:有没有办法告诉getaddrinfo选择一个在所有接口上都是空闲的空闲端口,然后将给定地址绑定到所有接口?

如果没有,是否有其他方法可以找到空闲端口号? (没有约束,失败时停止)

请参阅以下代码:

编辑:现在代码可以在VS 10上完全编译。

#ifdef WIN32
#include <WS2tcpip.h>
#pragma comment(lib, "ws2_32.lib")
#else
#include <sys/types.h>
#include <sys/socket.h>
#include <netdb.h>
#include <unistd.h>
#define closesocket close
#endif
#include <iostream>
#include <vector>  
#include <stdio.h>
#include <string>

int GetAddressFamily()
{
    return AF_UNSPEC;
}

std::string ipaddress(addrinfo* info)
{
    std::string retval;   
    char temp[260];
    socklen_t addrlen = (socklen_t)info->ai_addrlen;
    int res = ::getnameinfo(info->ai_addr, addrlen, temp, 256, NULL, 0, NI_NUMERICHOST);
    if(res){
        std::cout<<gai_strerrorA(res)<<std::endl;
    }else{
        retval = temp;
    }
    return retval;
}

int getport(addrinfo* info)
{
    int retval=0;

    if (info->ai_family == AF_INET) {
        retval = htons(((struct sockaddr_in*)(info->ai_addr))->sin_port);
    }else{
        retval = htons(((struct sockaddr_in6*)(info->ai_addr))->sin6_port);
    }
    return retval;
}

int main()
{
    char *hostName = NULL; //GetHostName();
    int portNum = 0;
#ifdef WIN32
    WSADATA w;
    if(0 != WSAStartup(MAKEWORD(2, 2), &w))
    {
       std::cerr<<" WSAStartup() failed \n";
       return -1;
    } 
#endif
    addrinfo hints,*results,*tmp;
    memset(&hints,0,sizeof(hints));
    hints.ai_family = GetAddressFamily();
    hints.ai_socktype = SOCK_STREAM;
    hints.ai_flags = AI_NUMERICSERV;
    if(hostName){
        hints.ai_flags |= AI_CANONNAME; 
        //AI_CANONNAME - fills ai_cannonname in address.
    }else{
        hints.ai_flags |= AI_PASSIVE;
        //AI_PASSIVE - give ADDR_ANY and IN6ADDR_ANY_INIT when hostName is NULL 
    }
    char portbuff[40];
    sprintf(portbuff,"%u",portNum);

    int res = ::getaddrinfo(hostName, portbuff,&hints, &results); 
    if(res){
        std::cerr<<gai_strerrorA(res)<<std::endl;
    }else{
        std::vector<int> sockets;
        for(tmp = results; tmp ; tmp=tmp->ai_next){
            std::cout<<ipaddress(tmp).c_str()<<" : "<<port(tmp)<<std::endl;
            int s = socket(tmp->ai_family,tmp->ai_socktype,tmp->ai_protocol);
            if(s != -1){
                res = bind(s, tmp->ai_addr, (int)tmp->ai_addrlen);
                if(res != -1){
                    sockaddr_storage addr;
                    socklen_t len =sizeof(addr);
                    int res = getsockname(s, (struct sockaddr *)&addr, &len);
                    std::cout<<"Bound to port: ";
                    if(addr.ss_family == AF_INET){
                        std::cout<<htons(((sockaddr_in*)&addr)->sin_port)<<std::endl;
                    }else{
                        std::cout<<htons(((sockaddr_in6*)&addr)->sin6_port)<<std::endl;
                    }
                    sockets.push_back(s);
                }
            }
        }
        for(int i=0;i<sockets.size();i++){
            closesocket(sockets[i]);
        }
    }
    ::freeaddrinfo(results);
    return 0;
}

EDIT2:我现在的解决方案:

我在第一次成功绑定后添加了以下函数,并将给定端口设置为addrinfo list:

void setport(addrinfo* info,int port)
{
   for(addrinfo* tmp = info; tmp ; tmp=tmp->ai_next){
   if (tmp->ai_family == AF_INET) {
     ((struct sockaddr_in*)(tmp->ai_addr))->sin_port = htons(port);
   }else{
     ((struct sockaddr_in6*)(tmp->ai_addr))->sin6_port = htons(port);
   }
}

成功绑定后应该调用它:

port = getport(result)
//...after bind:   
if(port == 0) {
   port = printed value after succesful bind
   setport(result, port)
}

1 个答案:

答案 0 :(得分:2)

我认为您最好的选择是使用禁用IPV6_V6ONLY的IPv6套接字。然后,该单个套接字将同时接受IPv6和IPv4连接。 IPv4客户端将地址设置为映射地址::ffff:a.b.c.d

这里没有任何理由使用getaddrinfo,因为你没有找到任何东西。这样的事情应该做(未经测试):

int s = socket(AF_INET6, SOCK_STREAM, 0);
if (s < 0)
    throw std::system_error(errno, generic_category());

const int off = 0;
if (setsockopt(s, IPPROTO_IPV6, IPV6_V6ONLY, &off, sizeof(off)) {
    close(s);
    throw std::system_error(errno, generic_category());
}

struct sockaddr_in6 addr{};
socklen_t alen = sizeof(addr);
if (bind(s, static_cast<struct sockaddr*>(&addr), alen)) {
    close(s);
    throw std::system_error(errno, generic_category());
}

getsockname(s, static_cast<struct sockaddr*>(&addr), &alen);
int port = ntohs(addr.sin6_port);

PS。 始终IPV6_V6ONLY选项设置为您希望的任何值都是一个好主意,因为默认值在操作系统之间有所不同。