Rust find an available tcp port 获取一个可用的 TCP 端口

fn get_available_port() -> u16 {
    std::net::TcpListener::bind("0.0.0.0:0")
        .unwrap()
        .local_addr()
        .unwrap()
        .port()
}

TcpListener::bind() 函数内部会调用 c::bind() 和 c::listen(), 其实我们只需要 c::bind() 即可

不过 get_available_port() 返回后，这个 listenner 也就被 drop 了，所以被 listen 一下问题不大

python 版本：

import socket

def get_available_port():
    with socket.socket() as s:
        s.bind(("0.0.0.0", 0))
        return s.getsockname()[1]

C 版本:

#include <stdio.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <string.h>


int main(int argc, char** argv) {
	struct sockaddr_in addr;
	memset(&addr, 0, sizeof(addr));
	addr.sin_family = AF_INET;

	int sock = socket(AF_INET, SOCK_STREAM, 0);
	if(sock == -1) {
		perror("socket");
		return -1;
	}

	if(bind(sock,(struct sockaddr*)&addr, sizeof(addr))) {
		perror("bind");
		return -1;
	}

	socklen_t len = sizeof(addr);
	if(getsockname(sock, (struct sockaddr*)&addr, &len)) {
		perror("getsockname");
		return -1;
	}

	printf("available tcp port: %d\n", addr.sin_port);

	return 0;
}