U /e@sdZddlmZddlZeeZddlmZm Z m Z m Z erJddl m Z ddl mZdZdd d d d d ZddddddZdd ddddZddddddZdS)zc Provide some utility functions useful for implementing different components in ``bokeh.server``. ) annotationsN) TYPE_CHECKINGListSequenceTuple)socket)netutil) bind_socketscheck_allowlistcreate_hosts_allowlist match_hostz str | NoneintzTuple[List[socket], int])addressportreturncCs`tj|p d|d}t|stdd|D}t|dks@td|}|rX||ksXt||fS)a Bind a socket to a port on an address. Args: address (str) : An address to bind a port on, e.g. ``"localhost"`` port (int) : A port number to bind. Pass 0 to have the OS automatically choose a free port. This function returns a 2-tuple with the new socket as the first element, and the port that was bound as the second. (Useful when passing 0 as a port number to bind any free port.) Returns: (socket, port) r)rrcSsh|]}|dqS)) getsockname).0sr5/tmp/pip-unpacked-wheel-f5fndrjf/bokeh/server/util.py Kszbind_sockets..rzMultiple ports assigned??)rr lenAssertionErrorpop)rrssZports actual_portrrrr 5s  r strz Sequence[str]bool)host allowlistrcs2dkrd|krdStfdd|DS)a Check a given request host against a allowlist. Args: host (str) : A host string to compare against a allowlist. If the host does not specify a port, then ``":80"`` is implicitly assumed. allowlist (seq[str]) : A list of host patterns to match against Returns: ``True``, if ``host`` matches any pattern in ``allowlist``, otherwise ``False`` ::80Tc3s|]}t|VqdS)N)r )rpatternrrr jsz"check_allowlist..)any)rr rr$rr Rs r zSequence[str] | Nonez List[str]) host_listrrc Cs|sdt|gSg}|D]}d|kr2td||dkrF||q|d}t|dkr|ddkrptd||d qt|d krzt|dWn tk rtd |YnX|ddkrtd||qtd |q|S) an This allowlist can be used to restrict websocket or other connections to only those explicitly originating from approved hosts. Args: host_list (seq[str]) : A list of string `` or `:` values to add to the allowlist. If no port is specified in a host string, then ``":80"`` is implicitly assumed. port (int) : If ``host_list`` is empty or ``None``, then the allowlist will be the single item list `` [ 'localhost:' ]`` If ``host_list`` is not empty, this parameter has no effect. Returns: list[str] Raises: ValueError, if host or port values are invalid Note: If any host in ``host_list`` contains a wildcard ``*`` a warning will be logged regarding permissive websocket connections. z localhost:*zHost wildcard %r will allow connections originating from multiple (or possibly all) hostnames or IPs. Use non-wildcard values to restrict access explicitlyr!rrzEmpty host valuer"zInvalid port in host value: %szInvalid host value: %s)rlogwarningappendsplitr ValueErrorr )r'rhostsrpartsrrrr ls6       r )rr#rcCsd}d|kr|dd\}}d}d|krD|dd\}}|dkrDd}|dk rX||krXdS|d}|d}t|t|krdSt||D]"\}}||ks|dkrqqdSqdS)aa Match a host string against a pattern Args: host (str) A hostname to compare to the given pattern pattern (str) A string representing a hostname pattern, possibly including wildcards for ip address octets or ports. This function will return ``True`` if the hostname matches the pattern, including any wildcards. If the pattern contains a port, the host string must also contain a matching port. Returns: bool Examples: >>> match_host('192.168.0.1:80', '192.168.0.1:80') True >>> match_host('192.168.0.1:80', '192.168.0.1') True >>> match_host('192.168.0.1:80', '192.168.0.1:8080') False >>> match_host('192.168.0.1', '192.168.0.2') False >>> match_host('192.168.0.1', '192.168.*.*') True >>> match_host('alice', 'alice') True >>> match_host('alice:80', 'alice') True >>> match_host('alice', 'bob') False >>> match_host('foo.example.com', 'foo.example.com.net') False >>> match_host('alice', '*') True >>> match_host('alice', '*:*') True >>> match_host('alice:80', '*') True >>> match_host('alice:80', '*:80') True >>> match_host('alice:8080', '*:80') False Nr!rr(F.T)rsplitr.rzip)rr# host_portZ pattern_portZ host_parts pattern_partshprrrr s&2  r )__doc__ __future__rlogging getLogger__name__r+typingrrrrrZtornador__all__r r r r rrrrs     ?