o vvXj~@sdZddlmZmZddlZddlZddlmZddlm Z Gddde Z Gdd d e Z Gd d d e Z d d ZeZGddde ZGddde ZdS)z(SASL mechanisms for AMQP authentication.)absolute_importunicode_literalsN)BytesIO) _write_tablec@s$eZdZdZeddZddZdS)SASLzThe base class for all amqp SASL authentication mechanisms. You should sub-class this if you're implementing your own authentication. cCt)z2Return a bytes containing the SASL mechanism name.NotImplementedErrorselfr F/var/www/html/myproject/venv/lib/python3.10/site-packages/amqp/sasl.py mechanismszSASL.mechanismcCr)z@Return the first response to a SASL challenge as a bytes object.rr connectionr r r startsz SASL.startN)__name__ __module__ __qualname____doc__propertyrrr r r r r s   rc@$eZdZdZdZddZddZdS)PLAINzbPLAIN SASL authentication mechanism. See https://tools.ietf.org/html/rfc4616 for details sPLAINcC|||_|_dSNusernamepasswordr rrr r r __init__#zPLAIN.__init__cCs^|jdus |jdur tSt}|d||jd|d||jd|S)Nzutf-8)rrNotImplementedrwriteencodegetvaluer rlogin_responser r r r&s  z PLAIN.startNrrrrrrrr r r r r  rc@r)AMQPLAINzhAMQPLAIN SASL authentication mechanism. This is a non-standard mechanism used by AMQP servers. sAMQPLAINcCrrrrr r r r9r zAMQPLAIN.__init__cCsF|jdus |jdur tSt}t|j|jd|jg|ddS)N)sLOGINsPASSWORD)rrr"rrr#r%r&r r r r<szAMQPLAIN.startNr(r r r r r*1r)r*csTz ddlddlWntyGdddt}|YSwGfdddt}|S)Nrc@s*eZdZdZdZ  d ddZddZdS) z)_get_gssapi_mechanism..FakeGSSAPIz7A no-op SASL mechanism for when gssapi isn't available.NamqpFcSs|stddS)Nz?You need to install the `gssapi` module for GSSAPI SASL supportrr client_nameservicerdns fail_softr r r rPs z2_get_gssapi_mechanism..FakeGSSAPI.__init__cSstSr)r"r r r r rWz/_get_gssapi_mechanism..FakeGSSAPI.startNr,FFr(r r r r FakeGSSAPIKs r4cs6eZdZdZdZ  d ddZdd Zfd d ZdS) z%_get_gssapi_mechanism..GSSAPIzsGSSAPI SASL authentication mechanism. See https://tools.ietf.org/html/rfc4752 for details sGSSAPINr,FcSs4|r t|ts |d}||_||_||_||_dS)Nascii) isinstancebytesr$r.r1r/r0r-r r r rcs   z._get_gssapi_mechanism..GSSAPI.__init__cSs^|jj}|jr|jtjtjfvr|}t|d\}}}n|jj }t |t s-| d}|S)Nrr5) transportsockr0familysocketAF_INETAF_INET6 getpeername gethostbyaddrhostr6r7r$)r rr9peerhostname_r r r get_hostnamels  z2_get_gssapi_mechanism..GSSAPI.get_hostnamecsz0|jrj|jd}nd}||}d|j|gjj}j||d}| dWSj j j yB|j rAtYSw)N)name@)rEcreds)r. CredentialsNamerDjoinr/NameTypehostbased_serviceSecurityContextsteprawmiscGSSErrorr1r")r rrGrBrEcontextgssapir r rxs"   z+_get_gssapi_mechanism..GSSAPI.startr3)rrrrrrrDrr rSr r GSSAPI[s   rU)rTgssapi.raw.misc ImportErrorr)r4rUr rSr _get_gssapi_mechanismFs  .rXc@seZdZdZdZddZdS)EXTERNALzEXTERNAL SASL mechanism. Enables external authentication, i.e. not handled through this protocol. Only passes 'EXTERNAL' as authentication mechanism, but no further authentication data. sEXTERNALcCsdS)Nr rr r r rr2zEXTERNAL.startN)rrrrrrr r r r rYs rYc@s$eZdZdZdZddZddZdS)RAWzA generic custom SASL mechanism. This mechanism takes a mechanism name and response to send to the server, so can be used for simple custom authentication schemes. NcCs:t|tsJt|tsJ|||_|_tdtdS)NznPassing login_method and login_response to Connection is deprecated. Please implement a SASL subclass instead.)r6r7rresponsewarningswarnDeprecationWarning)r rr\r r r rs z RAW.__init__cCs|jSr)r\rr r r rsz RAW.startr(r r r r r[s  r[)r __future__rrr;r]ioramqp.serializationrobjectrrr*rXrUrYr[r r r r s  F