o CD©i¡ã@shddlZddlZddlZddlZddlZddlZddlmZmZddl m Z m Z m Z Gdd„deƒZ dS)éNé)ÚNBNSÚNotConnectedError)Ú TYPE_CLIENTÚ TYPE_SERVERÚTYPE_WORKSTATIONc@sTeZdZe d¡Zddd„Zdd„Zdd „Zdd d„Z ddd„Z dd„Z dd„Z dS)ÚNetBIOSz NMB.NetBIOSTrcCsN||_t tjtj¡|_|jr|j tjtjd¡|r%|j d|f¡dSdS)a| Instantiate a NetBIOS instance, and creates a IPv4 UDP socket to listen/send NBNS packets. :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. rÚN) Ú broadcastÚsocketÚAF_INETÚ SOCK_DGRAMÚsockÚ setsockoptÚ SOL_SOCKETÚ SO_BROADCASTÚbind)Úselfr Ú listen_port©rúD/var/www/Datamplify/venv/lib/python3.10/site-packages/nmb/NetBIOS.pyÚ__init__ sÿzNetBIOS.__init__cCs|j ¡d|_dS)z· Close the underlying and free resources. The NetBIOS instance should not be used to perform any operations after this method returns. :return: None N)rÚclose)rrrrrs  z NetBIOS.closecCs$|jsJdƒ‚|j |||f¡dS)NúSocket is already closed)rÚsendto)rÚdataÚipÚportrrrÚwrite#sz NetBIOS.writer é‰écCs`|jsJdƒ‚t dd¡}| ||¡}|jr|sd}n|s#|j d¡| |||¡| ||¡S)ac 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 method will return None :return: A list of IP addresses in dotted notation (aaa.bbb.ccc.ddd). On timeout, returns None. rréÿÿz zQqueryName: ip parameter is empty. OS might not transmit this query to the network) rÚrandomÚrandintÚprepareNameQueryr ÚlogÚwarningrÚ_pollForNetBIOSPacket)rÚnamerrÚtimeoutÚtrn_idrrrrÚ queryName's     zNetBIOS.queryNamecCsd|jsJdƒ‚t dd¡}| |d¡}| |||¡| ||¡}|r0ttdd„tdd„|ƒƒƒSdS) a© 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 list of string containing the names of the machine at *ip*. On timeout, returns None. rrr!FcSs|dS)Nrr©ÚsrrrÚQsz(NetBIOS.queryIPForName..cSs |dtkS)Nr)rr,rrrr.Qs N) rr"r#ÚprepareNetNameQueryrÚ_pollForQueryPacketÚlistÚmapÚfilter)rrrr)r*rÚretrrrÚqueryIPForName>s   zNetBIOS.queryIPForNamec Cóìt ¡|} zAt ¡|}|dkrWdSt |j ¡ggg|¡\}}}|s)WdS|j d¡\}}t|ƒdkr9t‚| |¡\}} ||krG| WSWn,tjyt} zt | ƒt urh| dt j krg| dt j krg| ‚n| ‚WYd} ~ nd} ~ wwq©NTrr!)ÚtimeÚselectrÚfilenoÚrecvfromÚlenrÚ decodePacketÚerrorÚtypeÚtupleÚerrnoÚEINTRÚEAGAIN© rÚ wait_trn_idr)Úend_timeÚ_timeoutÚreadyÚ_rr*r4Úexrrrr'Yó2   ÿ €€ûîzNetBIOS._pollForNetBIOSPacketc Cr6r7)r8r9rr:r;r<rÚdecodeIPQueryPacketr>r?r@rArBrCrDrrrr0wrKzNetBIOS._pollForQueryPacketN)Tr)r rr )rr ) Ú__name__Ú __module__Ú __qualname__ÚloggingÚ getLoggerr%rrrr+r5r'r0rrrrrs     r)ÚosrPr"r r8r9ÚbaserrÚ nmb_constantsrrrrrrrrÚs0