o CD©i:ã@svddlZddlZddlZddlZddlZddlmZmZddlm Z ddl m Z Gdd„de ƒZ Gdd „d e e ƒZdS) éN)ÚreactorÚdefer)ÚDatagramProtocolé)ÚNBNSc@seZdZdZdS)ÚNetBIOSTimeoutzfRaised in NBNSProtocol via Deferred.errback method when queryName method has timeout waiting for replyN)Ú__name__Ú __module__Ú __qualname__Ú__doc__©r r úL/var/www/Datamplify/venv/lib/python3.10/site-packages/nmb/NetBIOSProtocol.pyrsrc@sTeZdZe d¡Zddd„Zdd„Zdd „Zdd d„Z ddd„Z dd„Z dd„Z dS)Ú NBNSProtocolzNMB.NBNSProtocolTrcCsJ||_i|_t ||¡|_|jr|j ¡ tjtj d¡t  d|j ¡dS)aß Instantiate a NBNSProtocol instance. This automatically calls reactor.listenUDP method to start listening for incoming packets, so you **must not** call the listenUDP method again. :param boolean broadcast: A boolean flag to indicate if we should setup the listening UDP port in broadcast mode :param integer listen_port: Specifies the UDP port number to bind to for listening. If zero, OS will automatically select a free port number. rN) Ú broadcastÚ pending_trnsrÚ listenUDPÚ transportÚ getHandleÚ setsockoptÚsocketÚ SOL_SOCKETÚ SO_BROADCASTÚ callLaterÚcleanupPendingTrns)ÚselfrÚ listen_portr r r Ú__init__s  zNBNSProtocol.__init__c CsZ|\}}| |¡\}}||jvr+|j |¡\}}} |tur$| |¡\}}|  |¡dSdS©N)Ú decodePacketrÚpopÚ NAME_QUERYÚdecodeIPQueryPacketÚcallback) rÚdataÚ from_infoÚhostÚportÚtrn_idÚretÚ_ÚipÚdr r r ÚdatagramReceiveds ûzNBNSProtocol.datagramReceivedcCs|j ¡ |||f¡dSr)rrÚsendto)rr#r*r&r r r Úwrite+szNBNSProtocol.writeÚé‰écCs†t dd¡} ||jvr n|dd@}q| ||¡}|jr"|s"d}n|s*|j d¡| |||¡t  ¡}t   ¡|||f|j|<|S)aX Send a query on the network and hopes that if machine matching the *name* will reply with its IP address. :param string ip: If the NBNSProtocol instance was instianted with broadcast=True, then this parameter can be an empty string. We will leave it to the OS to determine an appropriate broadcast address. If the NBNSProtocol instance was instianted with broadcast=False, then you should provide a target IP to send the query. :param integer port: The NetBIOS-NS port (IANA standard defines this port to be 137). You should not touch this parameter unless you know what you are doing. :param integer/float timeout: Number of seconds to wait for a reply, after which the returned Deferred instance will be called with a NetBIOSTimeout exception. :return: A *twisted.internet.defer.Deferred* instance. The callback function will be called with a list of IP addresses in dotted notation (aaa.bbb.ccc.ddd). On timeout, the errback function will be called with a Failure instance wrapping around a NetBIOSTimeout exception réÿÿTz zQqueryName: ip parameter is empty. OS might not transmit this query to the network) ÚrandomÚrandintrÚprepareNameQueryrÚlogÚwarningr.rÚDeferredÚtime)rÚnamer*r&Útimeoutr'r#r+r r r Ú queryName/s   ü   zNBNSProtocol.queryNamecsŽt dd¡} ||jvr n|dd@}q| |¡}| |||¡t ¡‰t ¡}| ˆj¡‡fdd„}|  |¡t   ¡|t |f|j|<ˆS)aU Send a query to the machine with *ip* and hopes that the machine will reply back with its name. The implementation of this function is contributed by Jason Anderson. :param string ip: If the NBNSProtocol instance was instianted with broadcast=True, then this parameter can be an empty string. We will leave it to the OS to determine an appropriate broadcast address. If the NBNSProtocol instance was instianted with broadcast=False, then you should provide a target IP to send the query. :param integer port: The NetBIOS-NS port (IANA standard defines this port to be 137). You should not touch this parameter unless you know what you are doing. :param integer/float timeout: Number of seconds to wait for a reply, after which the method will return None :return: A *twisted.internet.defer.Deferred* instance. The callback function will be called with a list of names of the machine at *ip*. On timeout, the errback function will be called with a Failure instance wrapping around a NetBIOSTimeout exception rr2Tcs.|durˆ tdd„tdd„|ƒƒ¡dSdS)NcSs|dS©Nrr ©Úsr r r Újsz@NBNSProtocol.queryIPForName..stripCode..cSs |dtkS)Nr)Ú TYPE_SERVERr>r r r r@jó )r"ÚmapÚfilter)r(©r+r r Ú stripCodehs"ÿz.NBNSProtocol.queryIPForName..stripCode) r3r4rÚprepareNetNameQueryr.rr8Ú addErrbackÚerrbackÚ addCallbackr9r )rr*r&r;r'r#Úd2rFr rEr ÚqueryIPForNameMs   ü    zNBNSProtocol.queryIPForNamecCst |¡dSr)rÚ stopProtocol©rr r r rMpszNBNSProtocol.stopProtocolcsRt ¡‰t‡fdd„ˆj ¡ƒ}‡fdd„}t||ƒˆjr't dˆj¡dSdS)Ncs |dˆkSr=r )r'Úinfo)Únowr r r@xrBz1NBNSProtocol.cleanupPendingTrns..cs8|\}\}}}ˆj|=z | t|ƒ¡WdSYdSr)rrIr)Úitemr'Ú expiry_timer:r+rNr r Ú expire_item{s  z4NBNSProtocol.cleanupPendingTrns..expire_itemr) r9rDrÚ iteritemsrCrrrr)rÚexpiredrSr )rPrr rss  ÿzNBNSProtocol.cleanupPendingTrnsN)Tr)r/r0r1)r0r1) rr r ÚloggingÚ getLoggerr6rr,r.r<rLrMrr r r r r s    # r)ÚosrVr3rr9Útwisted.internetrrÚtwisted.internet.protocolrÚbaserÚ Exceptionrrr r r r Ús (