Windows——获取一个可用的端口

作者：小 琛
欢迎转载，请标明出处

很多涉及到网络的业务场景下，需要给当前服务一个本机可用的端口，写死端口自然是不可取的，会像定时炸弹一样，因此最标准的解决方式就是：动态获取一个当前可以使用的端口号，将该端口提供给目标业务

思路描述

借助socket实现，当我们使用socket去bind端口时，如果将端口号sin_port定为0，系统则会自动给我们一个当前可用的端口，关闭socket后，这个端口就是我们需要的

1. winsock初始化
2. 初始化一个AF_INET、SOCK_STREAM的套接字
3. sockaddr_in的属性设置中将port定为0
4. bind成功后，使用getsockname获取对应信息，拿到系统分配的sin_port
5. 关闭该套接字，返回上一步中的sin_port

代码实现

#include <Windows.h>

#pragma comment(lib, "ws2_32.lib")

bool getAvaliablePort(unsigned short& port) {
    WSADATA wsadata;
    if (WSAStartup(MAKEWORD(2, 2), &wsadata) != 0) {
        return false;
    }
    SOCKET sock = socket(AF_INET, SOCK_STREAM, 0);

    struct sockaddr_in addr;
    addr.sin_family = AF_INET;
    addr.sin_addr.s_addr = htonl(INADDR_ANY);
    addr.sin_port = 0;

    struct sockaddr_in connAddr;
    int len = sizeof connAddr;

    if (::bind(sock, (SOCKADDR*)&addr, sizeof addr) != 0) {
        return false;
    }

    if (getsockname(sock, (SOCKADDR*)&connAddr, &len) != 0) {
        return false;
    }

    port = ntohs(connAddr.sin_port);

    if (closesocket(sock) != 0) {
        return false;
    }

    return true;
}