Bfi$dZdZdZdZdZdZdZddlmZdd lZdd l Z dd l m Z dd l m Z ejeZGd d eZy )zPyOpenSSL utilities including HTTPSSocket class which wraps PyOpenSSL SSL connection into a httplib-like interface suitable for use with urllib2 z P J Kershawz21/12/10z2(C) 2012 Science and Technology Facilities Councilz-BSD - see LICENSE file in top-level directoryzPhilip.Kershaw@stfc.ac.ukz$Id$)datetimeN)BytesIO)SSLceZdZdZdZd!dZdZedZejdZdZ d Z d Z d Z d Zd ZdZdZdZdZdZdZdZdZefdZdZdZdZdZdZdZdZdZ dZ!d Z"y)" SSLSocketa:SSL Socket class wraps pyOpenSSL's SSL.Connection class implementing the makefile method so that it is compatible with the standard socket interface and usable with httplib. @cvar default_buf_size: default buffer size for recv operations in the makefile method @type default_buf_size: int i Nc|||_nptjtjtj|_|jjtjtj dt j||j|_|jj|_ d|_ y)zCreate SSL socket object @param ctx: SSL context @type ctx: OpenSSL.SSL.Context @param sock: underlying socket object @type sock: socket.socket Nr) socketAF_INET SOCK_STREAM setsockopt SOL_SOCKET SO_REUSEADDRr Connection_SSLSocket__ssl_conn __class__default_buf_sizebuf_size_makefile_refs)selfctxsocks l/var/www/html/BankruptcyAI-uat/bankruptcy-ai/venv/lib/python3.12/site-packages/ndg/httpsclient/ssl_socket.py__init__zSSLSocket.__init__!s~  DK --8J8JKDK KK " "6#4#4f6I6I1 M..dkk:77 c$|jy)zCClose underlying socket when this object goes out of scope N)closers r__del__zSSLSocket.__del__3s  rc|jS)1Buffer size for makefile method recv() operations)_SSLSocket__buf_sizers rrzSSLSocket.buf_size8src`t|tstdt|z||_y)r!z1Expecting int type for "buf_size"; got %r insteadN) isinstanceint TypeErrortyper")rvalues rrzSSLSocket.buf_size=s3%%-/3E{;< <rc|jdkr |jjy|xjdzc_y#tjtj f$rYywxYw)zVShutdown the SSL connection and call the close method of the underlying socketr N)rrshutdownrError SysCallErrorrs rrzSSLSocket.closeEs]    " ((*   1 $  IIs//0  sA#A('A(c:|jj|y)zSet the shutdown state of the Connection. @param mode: bit vector of either or both of SENT_SHUTDOWN and RECEIVED_SHUTDOWN N)r set_shutdownrmodes rr.zSSLSocket.set_shutdownQs $$T*rc6|jjS)zGet the shutdown state of the Connection. @return: bit vector of either or both of SENT_SHUTDOWN and RECEIVED_SHUTDOWN )r get_shutdownrs rr2zSSLSocket.get_shutdownXs ++--rc:|jj|y)zbind to the given address - calls method of the underlying socket @param addr: address/port number tuple @type addr: tupleN)rbindraddrs rr4zSSLSocket.bind_s T"rc:|jj|y)zListen for connections made to the socket. @param backlog: specifies the maximum number of queued connections and should be at least 1; the maximum value is system-dependent (usually 5). @param backlog: int N)rlisten)rbacklogs rr8zSSLSocket.listenes w'rc8|jjy)zlSet the connection to work in server mode. The handshake will be handled automatically by read/writeN)rset_accept_staters rr;zSSLSocket.set_accept_statens ((*rc6|jjS)zAccept an SSL connection. @return: pair (ssl, addr) where ssl is a new SSL connection object and addr is the address bound to the other end of the SSL connection. @rtype: tuple )racceptrs rr=zSSLSocket.acceptss%%''rc8|jjy)zlSet the connection to work in client mode. The handshake will be handled automatically by read/writeN)rset_connect_staters rr?zSSLSocket.set_connect_state|s ))+rc:|jj|y)zCall the connect method of the underlying socket and set up SSL on the socket, using the Context object supplied to this Connection object at creation. @param addr: address/port number pair @type addr: tuple N)rconnectr5s rrAzSSLSocket.connects %rc6|jjS)aSend the shutdown message to the Connection. @param how: for socket.socket this flag determines whether read, write or both type operations are supported. OpenSSL.SSL.Connection doesn't support this so this parameter is IGNORED @return: true if the shutdown message exchange is completed and false otherwise (in which case you call recv() or send() when the connection becomes readable/writeable. @rtype: bool )rr*)rhows rr*zSSLSocket.shutdowns''))rc6|jjS)z-Renegotiate this connection's SSL parameters.)r renegotiaters rrEzSSLSocket.renegotiate**,,rc6|jjS)zk@return: numbers of bytes that can be safely read from the SSL buffer. @rtype: int )rpendingrs rrHzSSLSocket.pendings &&((rc8|jj|S)zSend data to the socket. Nb. The optional flags argument is ignored. - retained for compatibility with socket.socket interface @param data: data to send down the socket @type data: string )rsend)rdata flags_args rrJzSSLSocket.sends##D))rc:|jj|yN)rsendall)rrKs rrOzSSLSocket.sendalls %rc8|jj|S)zReceive data from the Connection. @param size: The maximum amount of data to be received at once @type size: int @return: data received. @rtype: string )rrecv)rsizes rrQzSSLSocket.recvs##D))rc:|jj|y)z}Set this connection's underlying socket blocking _mode_. @param mode: blocking mode @type mode: int N)r setblockingr/s rrTzSSLSocket.setblockings ##D)rc6|jjS)z_ @return: file descriptor number for the underlying socket @rtype: int )rfilenors rrVzSSLSocket.filenos %%''rc4|jj|S)z%See socket.socket.getsockopt )r getsockoptrargss rrXzSSLSocket.getsockopts*t))400rc4|jj|S)zsSee socket.socket.setsockopt @return: value of the given socket option @rtype: int/string )rr rYs rr zSSLSocket.setsockopts *t))400rc6|jjS)z(Return the SSL state of this connection.)r state_stringrs rr]zSSLSocket.state_strings++--rcB|xjdz c_|j}d}t}tj} |j j |}|r4|dz }|j||j j |}|r4tjtjkr,tjd|tj|z |j!d|S#tjtjf$rYwxYw)ahSpecific to Python socket API and required by httplib: convert response into a file-like object. This implementation reads using recv and copies the output into a StringIO buffer to simulate a file object for consumption by httplib Nb. Ignoring optional file open mode (StringIO is generic and will open for read and write unless a string is passed to the constructor) and buffer size - httplib set a zero buffer size which results in recv reading nothing @return: file object for data returned from socket @rtype: cStringIO.StringO r rz-Socket.makefile %d recv calls completed in %s)rrrrutcnowrrQwriterZeroReturnErrorr,loggetEffectiveLevelloggingDEBUGdebugseek)rrZ _buf_sizeistream startTimedats rmakefilezSSLSocket.makefiles q MM OO%  //&&y1C1 S!oo**95  "gmm 3 IIEqoo')3 5  A ##S%5%56    sAC88#DDc6|jjS)zC @return: the socket's own address @rtype: )r getsocknamers rrozSSLSocket.getsocknames **,,rc6|jjS)zJ @return: remote address to which the socket is connected )r getpeernamers rrqzSSLSocket.getpeernames**,,rc6|jjS)z=Retrieve the Context object associated with this Connection. )r get_contextrs rrszSSLSocket.get_contextrFrc6|jjS)z0Retrieve the other side's certificate (if any) )rget_peer_certificaters rruzSSLSocket.get_peer_certificates3355rrN)#__name__ __module__ __qualname____doc__rrrpropertyrsetterrr.r2r4r8r;r=r?rAr*rErHrJrOrQrTrVrXr r]rmrorqrsrurrrrs $ __   %+.# (+ (, & *-)*&)**(1 1.,\-- -6rr)ry __author____date__ __copyright__ __license__ __contact__ __revision__rrdr iorOpenSSLr getLoggerrvrbobjectrr|rrrsY D = )   g!D6D6r