a k#d2@s8dZddlZddlZddlZddlmZddlmZddlm Z ddl m Z ddl m Z ddlmZdd lmZdd lmZmZmZmZmZmZmZmZdd lmZdd lmZdd lm Z!ddlm"Z#ddl$m%Z&ddgZ'Gddde(Z)GdddZ*Gddde(Z+e+Z,Gddde#Z-Gddde Z.e.Z/dS)z Python-CAN CANSocket Wrapper. N)reduce)add)conf) SuperSocket)CAN)Packet)warning)ListTypeTupleDictAnyOptionalcastorb)queue)Message)CanError)BusABC)Bus CANSocketPythonCANSocketc@s@eZdZdZddZddZddZdd Zd d Zd d Z dS)PriotizedCanMessageanHelper object for comparison of CAN messages. If the timestamps of two messages are equal, the counter value of a priority counter, is used for comparison. It's only important that this priority counter always get increased for every CAN message in the receive heapq. This compensates a low resolution of `time.time()` on some operating systems. cCs||_||_dSN)msgcount)selfrrrY/var/www/html/gsl.tools/lib/python3.9/site-packages/scapy/contrib/cansocket_python_can.py__init__,szPriotizedCanMessage.__init__cCs*t|tsdS|jj|jjko(|j|jkSNF isinstancerr timestamprrotherrrr__eq__1s   zPriotizedCanMessage.__eq__cCs:t|tsdS|jj|jjkp8|jj|jjko8|j|jkSr r!r$rrr__lt__8s   zPriotizedCanMessage.__lt__cCs||kp||kSrrr$rrr__le__@szPriotizedCanMessage.__le__cCs ||k Srrr$rrr__gt__DszPriotizedCanMessage.__gt__cCs ||k Srrr$rrr__ge__HszPriotizedCanMessage.__ge__N) __name__ __module__ __qualname____doc__rr&r'r(r)r*rrrrr%src@s eZdZdZddZddZdS) SocketMapperzbInternal Helper class to map a python-can bus object to a list of SocketWrapper instances cCs||_||_dS)zInitializes the SocketMapper helper class :param bus: A python-can Bus object :param sockets: A list of SocketWrapper objects which want to receive messages from the provided python-can Bus object. N)bussockets)rr0r1rrrrQszSocketMapper.__init__c Csd}zP|jjdd}|dur"WdS|jD](}||r(|d7}|jt||q(Wqty}ztd|WYd}~qd}~00qdS)zMultiplexer function. Tries to receive from its python-can bus object. If a message is received, this message gets forwarded to all receive queues of the SocketWrapper objects. r)timeoutNz%[MUX] python-can exception caught: %s) r0recvr1_matches_filtersrx_queueputr Exceptionr)rZ prio_countrsockerrrmux\s  zSocketMapper.muxN)r+r,r-r.rr;rrrrr/Ms r/c@s:eZdZdZddZdddZddZd d Zd d Zd S) _SocketsPoolzCHelper class to organize all SocketWrapper and SocketMapper objectscCst|_t|_dSr)dictpool threadingLock pool_mutexrrrrrrsz_SocketsPool.__init__rc Cs|jdurtd|jzP|j|j}|j||jD],}||krHq:||sTq:|j t ||q:WnNt yt d|jYn0t y}zt d|WYd}~n d}~00Wdn1s0YdS)aXInternal send function. A given SocketWrapper wants to send a CAN message. The python-can Bus object is obtained from an internal pool of SocketMapper objects. The given message is sent on the python-can Bus object and also inserted into the message queues of all other SocketWrapper objects which are connected to the same python-can bus object by the SocketMapper. :param sender: SocketWrapper which initiated a send of a CAN message :param msg: CAN message to be sent :param prio: Priority count for internal heapq N'SocketWrapper.name should never be Nonez![SND] Socket %s not found in poolz%[SND] python-can exception caught: %s)name TypeErrorrAr>r0sendr1r5r6r7rKeyErrorr can_CanError)rZsenderrprioZmapperr9r:rrr internal_sendws       z_SocketsPool.internal_sendcCsB|j(|jD] }|qWdn1s40YdS)z]This calls the mux() function of all SocketMapper objects in this SocketPool N)rAr>valuesr;)rtrrrmultiplex_rx_packetssz!_SocketsPool.multiplex_rx_packetscOst|dddt|dd}|j||jvrz|j|}|j|dd|jD}|rr|jtt |||_ n&t |i|}||_ t ||g|j|<Wdn1s0YdS) aRegisters a SocketWrapper object. Every SocketWrapper describes to a python-can bus object. This python-can bus object can only exist once. In case this object already exists in this SocketsPool, organized by a SocketMapper object, the new SocketWrapper is inserted in the list of subscribers of the SocketMapper. Otherwise a new python-can Bus object is created from the provided args and kwargs and inserted, encapsulated in a SocketMapper, into this SocketsPool. :param socket: SocketWrapper object which needs to be registered. :param args: Arguments for the python-can Bus object :param kwargs: Keyword arguments for the python-can Bus object ZbustypeZunknown_bustype_ZchannelZunknown_channelcSsg|]}|jdur|jqSr)filters.0srrr s z)_SocketsPool.register..N) strgetrAr>r1appendr0Z set_filtersrrrDcan_Busr/)rsocketargskwargskrLrOr0rrrregisters   z_SocketsPool.registerc Cs|jdurtd|jhz6|j|j}|j||jsN|j|j|j=Wn typt d|jYn0Wdn1s0YdS)zUnregisters a SocketWrapper from its subscription to a SocketMapper. If a SocketMapper doesn't have any subscribers, the python-can Bus get shutdown. :param socket: SocketWrapper to be unregistered NrCz#Socket %s already removed from pool) rDrErAr>r1remover0shutdownrGr)rrXrLrrr unregisters     z_SocketsPool.unregisterN)r) r+r,r-r.rrJrMr\r_rrrrr<ps  " r<cs:eZdZdZfddZddZd ddZd d ZZS) SocketWrapperz6Helper class to wrap a python-can Bus object as socketcsHtt|j|i|t|_d|_d|_tj |g|Ri|dS)aHInitializes a new python-can based socket, described by the provided arguments and keyword arguments. This SocketWrapper gets automatically registered in the SocketsPool. :param args: Arguments for the python-can Bus object :param kwargs: Keyword arguments for the python-can Bus object Nr) superr`rr PriorityQueuer6rD prio_counter SocketsPoolr\)rrYrZ __class__rrrs  zSocketWrapper.__init__cCs@tz|jjd|d}|jdfWStjy:YdS0dS)apInternal blocking receive method, following the ``can_BusABC`` interface of python-can. This triggers the multiplex function of the general SocketsPool. :param timeout: Time to wait for a packet :return: Returns a tuple of either a can_Message or None and a bool to indicate if filtering was already applied. T)blockr2)NTN)rdrMr6rUrrEmpty)rr2pmrrr_recv_internals  zSocketWrapper._recv_internalNcCs"|jd7_t|||jdS)zSend function, following the ``can_BusABC`` interface of python-can. :param msg: Message to be sent. :param timeout: Not used. r3N)rcrdrJ)rrr2rrrrFszSocketWrapper.sendcCst|dS)zYShutdown function, following the ``can_BusABC`` interface of python-can. N)rdr_rBrrrr^szSocketWrapper.shutdown)N) r+r,r-r.rrjrFr^ __classcell__rrrerr`s   r`csTeZdZdZdZdZddZdddZd d Ze e j fd d Z fd dZ ZS)ralInitializes a python-can bus object as Scapy PythonCANSocket. All provided keyword arguments, except *basecls* are forwarded to the python-can can_Bus init function. For further details on python-can check: https://python-can.readthedocs.io/ Example: >>> socket = PythonCANSocket(bustype='socketcan', channel='vcan0', bitrate=250000) zIread/write packets at a given CAN interface using a python-can bus objectTcKs0tttt|dt|_tfi||_dS)Nbasecls) rrr rpoprrlr` can_iface)rrZrrrrszPythonCANSocket.__init__cCs|j}|jd>|jd>B|jd>B|jB}tjddrTt dt d|d}|j d >|j d >B|j d >B|jd >B}t d ||t|j}|j||jfS)z0Returns a tuple containing (cls, pkt_data, time)rz swap-byteszIrz!II)rnr4is_extended_idis_remote_frameis_error_framearbitration_idrZcontribsstructunpackpackdlcis_fderror_state_indicatorbitrate_switchbytesdatarlr#)rxrhdrr~Zpkt_datarrrrecv_raw#s zPythonCANSocket.recv_rawc Cst|}t|jdk|jdk|jdk|jt|dd@dkt|dd@dkt|dd@dk|j|ddd }t|_z |j|_Wnt yYn0|j |t |S)Nr3r) rxrwryrzrrrr~r) r can_Messageflags identifierrlengthtimer#Z sent_timeAttributeErrorrnrFlen)rrbxrrrrrF3s$      zPythonCANSocket.sendcCstdd|DS)a0This function is called during sendrecv() routine to select the available sockets. :param sockets: an array of sockets that need to be selected :returns: an array of sockets that were selected and the function to be called next to get the packets (i.g. recv) cSs&g|]}t|tr|jjs|qSr)r"rrnr6emptyrPrrrrSRs z*PythonCANSocket.select..)rdrM)r1ZremainrrrselectGs zPythonCANSocket.selectcs&|jr dStt||jdS)zCloses this socketN)closedrarclosernr^rBrerrrUszPythonCANSocket.close)ro)r+r,r-r.descZnonblocking_socketrrrF staticmethodrZrecv_poll_raterrrkrrrerrs   )0r.rr{r? functoolsroperatorrZ scapy.configrZscapy.supersocketrZscapy.layers.canrZ scapy.packetrZ scapy.errorrZ scapy.compatr r r r r rrrZscapy.libs.six.movesrZcanrrrrHrZ can_BusABCZ can.interfacerrW__all__objectrr/r<rdr`rrrrrr s0       (     (#g6N