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 

#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;
}

你可能感兴趣的:(Windows,网络,网络,windows,c++)