1#!/usr/bin/env python3 2# -*- coding: utf-8 -*- 3#*************************************************************************** 4# _ _ ____ _ 5# Project ___| | | | _ \| | 6# / __| | | | |_) | | 7# | (__| |_| | _ <| |___ 8# \___|\___/|_| \_\_____| 9# 10# Copyright (C) Daniel Stenberg, <daniel@haxx.se>, et al. 11# 12# This software is licensed as described in the file COPYING, which 13# you should have received as part of this distribution. The terms 14# are also available at https://curl.se/docs/copyright.html. 15# 16# You may opt to use, copy, modify, merge, publish, distribute and/or sell 17# copies of the Software, and permit persons to whom the Software is 18# furnished to do so, under the terms of the COPYING file. 19# 20# This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY 21# KIND, either express or implied. 22# 23# SPDX-License-Identifier: curl 24# 25########################################################################### 26# 27import logging 28import socket 29from typing import Dict 30 31log = logging.getLogger(__name__) 32 33 34def alloc_ports(port_specs: Dict[str, int]) -> Dict[str, int]: 35 ports = {} 36 socks = [] 37 for name, ptype in port_specs.items(): 38 try: 39 s = socket.socket(type=ptype) 40 s.bind(('', 0)) 41 ports[name] = s.getsockname()[1] 42 socks.append(s) 43 except Exception as e: 44 raise e 45 for s in socks: 46 s.close() 47 return ports 48
