U  W[û ã@s~dZddlmZddlmZddlmZddlmZddl m Z ddl m Z Gd d „d ej eƒZGd d „d ejƒZd d„ZdS)z@ Protocol wrapper that provides HAProxy PROXY protocol support. é)Úpolicies)Ú interfaces)Ú_WrapperServerEndpointé)ÚInvalidProxyHeader)ÚV1Parser)ÚV2Parserc@s0eZdZdZdd„Zdd„Zdd„Zdd „Zd S) ÚHAProxyProtocolWrappera A Protocol wrapper that provides HAProxy support. This protocol reads the PROXY stream header, v1 or v2, parses the provided connection data, and modifies the behavior of getPeer and getHost to return the data provided by the PROXY header. cCs tj |||¡d|_d|_dS©N)rÚProtocolWrapperÚ__init__Ú _proxyInfoÚ_parser)ÚselfÚfactoryÚwrappedProtocol©rúD/usr/lib/python3/dist-packages/twisted/protocols/haproxy/_wrapper.pyr szHAProxyProtocolWrapper.__init__cCsÜ|jdk r|j |¡S|jdkr”t|ƒdkr`|dd…tjkr`t|dd…ƒd@dkr`tƒ|_n4t|ƒdkrˆ|dd…tj krˆtƒ|_n |  ¡dSz&|j  |¡\|_}|r¸|j |¡Wnt k rÖ|  ¡YnXdS)Néé é éðé éé) r rÚ dataReceivedrÚlenrZPREFIXÚordrZPROXYSTRZloseConnectionZfeedr)rÚdataZ remainingrrrr#s(    ÿþý  z#HAProxyProtocolWrapper.dataReceivedcCs |jr|jjr|jjS|j ¡Sr )r ÚsourceÚ transportÚgetPeer©rrrrr!<szHAProxyProtocolWrapper.getPeercCs |jr|jjr|jjS|j ¡Sr )r Z destinationr ÚgetHostr"rrrr#BszHAProxyProtocolWrapper.getHostN)Ú__name__Ú __module__Ú __qualname__Ú__doc__r rr!r#rrrrr s r c@seZdZdZeZdd„ZdS)ÚHAProxyWrappingFactoryzL A Factory wrapper that adds PROXY protocol support to connections. cCs.tj |j¡r|j ¡}n |jjj}d|fS)z“ Annotate the wrapped factory's log prefix with some text indicating the PROXY protocol is in use. @rtype: C{str} z %s (PROXY))rZILoggingContextZ providedByZwrappedFactoryÚ logPrefixÚ __class__r$)rr)rrrr)Os  z HAProxyWrappingFactory.logPrefixN)r$r%r&r'r Zprotocolr)rrrrr(Isr(cCs t|tƒS)a± Wrap an endpoint with PROXY protocol support, so that the transport's C{getHost} and C{getPeer} methods reflect the attributes of the proxied connection rather than the underlying connection. @param wrappedEndpoint: The underlying listening endpoint. @type wrappedEndpoint: L{IStreamServerEndpoint} @return: a new listening endpoint that speaks the PROXY protocol. @rtype: L{IStreamServerEndpoint} )rr()ZwrappedEndpointrrrÚ proxyEndpoint^s r+N)r'Ztwisted.protocolsrZtwisted.internetrZtwisted.internet.endpointsrÚ _exceptionsrZ _v1parserrZ _v2parserrr Úobjectr ZWrappingFactoryr(r+rrrrÚs      5