a k#dO @s@ddlZddlZddlmZddlmZmZmZmZm Z m Z ddl m Z ddlm Z ddlmZddlmZddlmZdd lmZmZmZmZed Zd d Zd'ddZddZddZedddd ddfddZ eddedddd dfddZ!edd edddddd dd df ddZ"d(d d!Z#d)d#d$Z$d*d%d&Z%dS)+N)Event)IterableOptionalUnionListTupleDict)Packet)orb)CAN) SuperSocket) PYTHON_CAN) ISOTPHeader ISOTPHeaderEAISOTP_FFISOTPzscapy.contrib.isotpcCs6t||d}t||dD]}||_||qdS)a+Send multiple packets with extended addresses at once. This function is used for scanning with extended addresses. It sends multiple packets at once. The number of packets is defined in the number_of_packets variable. It only iterates the extended ID, NOT the actual CAN ID of the packet. This method is used in extended scan function. :param sock: CAN interface to send packets :param ext_id: Extended ISOTP-Address :param packet: Template Packet :param number_of_packets: number of packets to send in one batch N)minrangeextended_addresssend)sockext_idpacketZnumber_of_packetsZend_idirX/var/www/html/gsl.tools/lib/python3.9/site-packages/scapy/contrib/isotp/isotp_scanner.pysend_multiple_extsrFcCsJ|rtt}d|_d|_ntt}d|_|r:d|_||_d|_|S)zCraft ISO-TP packet :param identifier: identifier of crafted packet :param extended: boolean if packet uses extended address :param extended_can_id: boolean if CAN should use extended Ids :return: Crafted Packet rssextendedd)rrrdatarflags identifierZ message_size)r#rextended_can_idpktrrrget_isotp_packet0s   r&c Csi}|D]V\}}|d}|d}||vr>|g|gf||<q ||\}}||g||gf||<q |D]}||d}||d}t|dkrqhddt|dd|ddD}tdd t|dd|ddDrhtd |D] } || =qqhdS) zFilter to remove periodic packets from packet_dict ISOTP-Filter for periodic packets (same ID, always same time-gaps) Deletes periodic packets in packet_dict :param packet_dict: Dictionary, where the filter is applied rrcSs$g|]\}}t|jt|jqSr)floattime).0p1p2rrr csz+filter_periodic_packets..Ncss"|]\}}t||dkVqdS)gMbP?N)abs)r*t1t2rrr ez*filter_periodic_packets..z5[i] Identifier 0x%03x seems to be periodic. Filtered.)itemslenzipall log_isotpinfo) Z packet_dictZ filter_dictkeyvaluer%ZidnZkey_lstZpkt_lsttgkrrrfilter_periodic_packetsIs,     (r>c Cs|jr|jdkrdS|dur*|j|vr*dSz|r4dnd}t|j|d?}t|j|d@}|dkrd|krxdkrnnJtd |j|t|tr||jf||<qt|tr| |qt d n|dur| |jWn8t y}zt d |t |WYd}~n d}~00dS) aCallback for sniff function when packet received If received packet is a FlowControl and not in noise_ids append it to id_list. :param id_value: packet id of send packet :param id_list: list of received IDs :param noise_ids: list of packet IDs which will not be considered when received during scan :param extended: boolean if extended scan :param packet: received packet rNrrr'zNFound flow-control frame from identifier 0x%03x when testing identifier 0x%03xzUnknown type of id_listz+Unknown message Exception: %s on packet: %s)r"r#r r!r8r9 isinstancedictlistappend TypeError Exception exceptionrepr) Zid_valueid_list noise_idsrrindexZ isotp_pciZisotp_fcerrr get_isotp_fcls0      rNig?Tcst|D]T|dur"|r"q`r0vr0q |td||jfdd|ddq |shStD]~}|dur|rqttd|d|ddD]J|dur|rqv|td||jfd d|d ddqqvS) a6Scan and return dictionary of detections ISOTP-Scan - NO extended IDs found_packets = Dictionary with Send-to-ID as key and a tuple (received packet, Recv_ID) :param sock: socket for can interface :param scan_range: hexadecimal range of IDs to scan. Default is 0x0 - 0x7ff :param noise_ids: list of packet IDs which will not be tested during scan :param sniff_time: time the scan waits for isotp flow control responses after sending a first frame :param extended_can_id: Send extended can frames :param verify_results: Verify scan results. This will cause a second scan of all possible candidates for ISOTP Sockets :param stop_event: Event object to asynchronously stop the scan :return: Dictionary with all found packets NFcstd|SNFrNr%)rK return_valuesr;rrszscan..ZprntimeoutstorerrArcstd|SrOrPrQ)cleaned_ret_valrKr;rrrSs )rCis_setrr&sniffkeysrmax)r scan_rangerK sniff_timer$verify_results stop_eventZ tested_idr)rWrKrRr;rscans0   ra c sRt|p d}|D]8}r&|vr&qt|d|d} gt|} t| d| d|D]P|durl|rlqt|| ||jfdd|d d d t|qTtt D]|dur|rqtt d dt |d d D]X} |dur |r q| | _ |d>| | | |jfdd|d d d qqqS)aScan with ISOTP extended addresses and return dictionary of detections If an answer-packet found -> slow scan with single packages with extended ID 0 - 255 found_packets = Dictionary with Send-to-ID as key and a tuple (received packet, Recv_ID) :param sock: socket for can interface :param scan_range: hexadecimal range of IDs to scan. Default is 0x0 - 0x7ff :param scan_block_size: count of packets send at once :param extended_scan_range: range to search for extended ISOTP addresses :param noise_ids: list of packet IDs which will not be tested during scan :param sniff_time: time the scan waits for isotp flow control responses after sending a first frame :param extended_can_id: Send extended can frames :param stop_event: Event object to asynchronously stop the scan :return: Dictionary with all found packets rT)rr$rr.Ncstd|SNTrP)p) ext_isotp_idrJrKrrrSszscan_extended..r'FrTrArccstd|SrdrPrQ)full_idrKrRrrrSs)rCr&rDrrYrrZr)sleepsetr\rrr) rr]Zscan_block_sizeextended_scan_rangerKr^r$r`r;r%rrr)rfrhrJrKrRr scan_extendedsD       rmrAc  s| rttjtdtdddj|fddd} ttdd | D} |rpt ||| ||| d }nt || ||| | d }t ||d krt ||S|d krt |||St||p|S)aScan for ISOTP Sockets on a bus and return findings Scan for ISOTP Sockets in the defined range and returns found sockets in a specified format. The format can be: - text: human readable output - code: python code for copy&paste - sockets: if output format is not specified, ISOTPSockets will be created and returned in a list :param sock: CANSocket object to communicate with the bus under scan :param scan_range: range of CAN-Identifiers to scan. Default is 0x0 - 0x7ff :param extended_addressing: scan with ISOTP extended addressing :param extended_scan_range: range for ISOTP extended addressing values :param noise_listen_time: seconds to listen for default communication on the bus :param sniff_time: time the scan waits for isotp flow control responses after sending a first frame :param output_format: defines the format of the returned results (text, code or sockets). Provide a string e.g. "text". Default is "socket". :param can_interface: interface used to create the returned code/sockets :param extended_can_id: Use Extended CAN-Frames :param verify_results: Verify scan results. This will cause a second scan of all possible candidates for ISOTP Sockets :param verbose: displays information during scan :param stop_event: Event object to asynchronously stop the scan :return: zFiltering background noise...i#s)r#r!cs SN)rrZ dummy_pktrrrrSOr3zisotp_scan..)rUZstarted_callbackcss|] }|jVqdSrn)r#)r*r%rrrr2Qr3zisotp_scan..)rkrKr^r$r`)rKr^r$r_r`textcode)r8setLevelloggingDEBUGr9r rZrDrjrmrar>generate_text_outputgenerate_code_outputgenerate_isotp_list)rr]extended_addressingrkZnoise_listen_timer^Z output_format can_interfacer$r_verboser`Zbackground_pktsrK found_packetsrror isotp_scansH+     r|c Cs|sdSdt|}|D]}|r|d}||d}tt||djd}|dt|t|t||dj|t||df7}n0|dt|t||djt||df7}||djdk}|r|d7}n|d 7}|d 7}q|S) aVGenerate a human readable output from the result of the `scan` or the `scan_extended` function. :param found_packets: result of the `scan` or `scan_extended` function :param extended_addressing: print results from a scan with ISOTP extended addressing :return: human readable scan results zNo packets found.z& Found %s ISOTP-FlowControl Packet(s):rcrz Send to ID: %s Send to extended ID: %s Received ID: %s Received extended ID: %s Message: %szQ Send to ID: %s Received ID: %s Message: %srgz Padding enabledz No Padding )r5hexr r!r#rIlength)r{rxrppacksend_idsend_extrpaddingrrrruos4       ruifacec Csd}|s |Sdt}|D]}|r|d}||d}t||djd}|d||t||dj||djdk||f7}q|d||t||dj||djdkf7}q||S)aGenerate a copy&past-able output from the result of the `scan` or the `scan_extended` function. :param found_packets: result of the `scan` or `scan_extended` function :param can_interface: description string for a CAN interface to be used for the creation of the output. :param extended_addressing: print results from a scan with ISOTP extended addressing :return: Python-code as string to generate all found sockets zr import can conf.contribs['CANSocket'] = {'use-python-can': %s} load_contrib('cansocket') load_contrib('isotp') rcrzjISOTPSocket(%s, tx_id=0x%x, rx_id=0x%x, padding=%s, ext_address=0x%x, rx_ext_address=0x%x, basecls=ISOTP) rgzCISOTPSocket(%s, tx_id=0x%x, rx_id=0x%x, padding=%s, basecls=ISOTP) )r r r!intr#r) r{ryrxresultheaderrrrrrrrrvs4   rvc Csddlm}g}|D]}||d}|j}|jdkr8dnd}|r|d?} t|| d} t|jd} |||| | || |tdq|} |||| ||tdq|S) aGenerate a list of ISOTPSocket objects from the result of the `scan` or the `scan_extended` function. :param found_packets: result of the `scan` or `scan_extended` function :param can_interface: description string for a CAN interface to be used for the creation of the output. :param extended_addressing: print results from a scan with ISOTP extended addressing :return: A list of all found ISOTPSockets r) ISOTPSocketrgTFrc)tx_idZ ext_addressrx_idZrx_ext_addressrbasecls)rrrr) Zscapy.contrib.isotprr#rrr r!rEr) r{ryrxrZ socket_listrr%Zdest_idpadZ source_idZ source_extZdest_extrrrrws.      rw)rFF)F)rF)F)&rsr) threadingrZ scapy.compatrrrrrrZ scapy.packetr r Zscapy.layers.canr Zscapy.supersocketr Zscapy.contrib.cansocketr Z scapy.contrib.isotp.isotp_packetrrrr getLoggerr8rr&r>rNrrarmr|rurvrwrrrr s^         #2 7 F W - /