a k#do*@sRddlZddlZddlmZmZmZmZmZddlm Z ddl m Z m Z m Z mZmZmZmZmZmZmZmZddlmZmZddlmZddlmZedZd Zd Zd Z d Z!d Z"dddddZ#dZ$d Z%dZ&dZ'Gddde Z(GdddeZ)Gddde)Z*dddddZ+Gddde Z,Gddde Z-Gdd d e Z.Gd!d"d"e Z/dS)#N)OptionalListTupleAnyType)Packet) BitField FlagsField StrLenFieldThreeBytesField XBitFieldConditionalField BitEnumField ByteField XByteFieldBitFieldLenFieldStrField)chborb)CAN)Scapy_Exceptionzscapy.contrib.isotpilisinglefirstZ consecutiveZ flow_control)r 0c@sTeZdZdZdZeddgZejgdZddZ ddZ d d Z e dd d Z d S)ISOTPaPacket class for ISOTP messages. This class contains additional slots for source address (tx_id), destination address (rx_id), extended source address (ext_address) and extended destination address (rx_ext_address) information. This information gets filled from ISOTPSockets or the ISOTPMessageBuilder, if it is available. Address information is not used for Packet comparison. :param args: Arguments for Packet init, for example bytes string :param kwargs: Keyword arguments for Packet init. data)tx_idrx_id ext_addressrx_ext_addresscOs\|dd|_|dd|_|dd|_|dd|_tj|g|Ri||dS)Nr#r$r%r&)popr#r$r%r&r__init__validate_fields)selfargskwargsr-W/var/www/html/gsl.tools/lib/python3.9/site-packages/scapy/contrib/isotp/isotp_packet.pyr(8s zISOTP.__init__cCs|jdur*d|jkr tks*ntd|jdurTd|jkrJtksTntd|jdur~d|jkrtdks~ntd|jdurd|jkrdksntddS)znHelper function to validate information in tx_id, rx_id, ext_address and rx_ext_address slots Nrz#tx_id is not a valid CAN identifierz#rx_id is not a valid CAN identifierzext_address is not a bytezrx_ext_address is not a byte)r#CAN_MAX_IDENTIFIERrr$r%r&)r*r-r-r.r)As    zISOTP.validate_fieldsc Os d}|jdurd}t|jtkr(tdt|j|krtdt|j|j}|jrftd|j|}|jdusz|jdkrt|j|d}nt|jd|d }|gSt|jt krtd t|jd }ntd d t|j}|jrtd|j|}d t|}|jd|}|jdus$|jdkr8t|j||d}nt|jd||d }d} |g} |t|jkr|j|||}td| dt }| d7} |t|7}|jrtd|j|}|jdus|jdkrt|j||d}nt|jd||d }| |qV| S)z~Helper function to fragment an ISOTP message into multiple CAN frames. :return: A list of CAN frames NzToo much data in ISOTP messageBi) identifierr!extended)r4flagsr!z>Hiz>HIrrrbr) r&lenr!ISOTP_MAX_DLEN_2015rstructpackr$rISOTP_MAX_DLENN_PCI_CFappend) r*r+ZkargsZdata_bytes_in_frameZ frame_datapktZ frame_headeridxframenZpktsr-r-r.fragmentSsV   zISOTP.fragmentNcsddlm}t|dkr td|djtfdd|DrJtd||}||g}|D]:}|durz|j dus|d ur|j dus|durd| |qd|sdSt|d krtd |dS) aHelper function to defragment a list of CAN frames to one ISOTP message :param can_frames: A list of CAN frames :param use_extended_addressing: Specify if extended ISO-TP addressing is used in the packets for defragmentation. :return: An ISOTP message containing the data of the CAN frames or None r)ISOTPMessageBuilderz%ISOTP.defragment called with 0 framesc3s|]}|jkVqdSN)r4).0rAdstr-r. r"z#ISOTP.defragment..z+Not all CAN frames have the same identifierTNFrzkMore than one ISOTP frame could be defragmented from the provided CAN frames, only returning the first one.) Zscapy.contrib.isotp.isotp_utilsrDr8rr4any log_isotpwarningfeedr&r>)Z can_framesZuse_extended_addressingrDparserresultspr-rGr. defragments6       zISOTP.defragment)N)__name__ __module__ __qualname____doc__namer fields_descr __slots__r(r)rC staticmethodrQr-r-r-r.r 's  >r c@sVeZdZdZedddgdedddedded dgZd d Z d d Z ddZ dS) ISOTPHeaderr6rrerrorZremote_transmission_requestr5r4lengthNreservedcCs|dfSrEr-)r*rPr-r-r.extract_paddingszISOTPHeader.extract_paddingcCs6|jdur.|ddtt||dd}||S)zL This will set the ByteField 'length' to the correct value. Nr^rr8)r*r?payr-r-r. post_builds $zISOTPHeader.post_buildcCs@t|dd@d?}|dkr tS|dkr,tS|dkr8tStSdS)zISO-TP encodes the frame type in the first nibble of a frame. This is used to determine the payload_class :param payload: payload bytes string :return: Type of payload class rrarrN)rISOTP_SFISOTP_FFISOTP_CFISOTP_FC)r*payloadtr-r-r.guess_payload_classszISOTPHeader.guess_payload_class) rRrSrTrVr r rr rWr`rermr-r-r-r.rZs   rZc@sNeZdZdZedddgdedddedd ed ded dgZ d d Z d S) ISOTPHeaderEAZISOTPHeaderExtendedAddressr6rrr[r4r]r^Nr_Zextended_addresscCs:|jdur2|ddtt|d|dd}||S)z This will set the ByteField 'length' to the correct value. 'chb(len(pay) + 1)' is required, because the field 'extended_address' is counted as payload on the CAN layer Nrarrbrc)r*rPrdr-r-r.res (zISOTPHeaderEA.post_build) rRrSrTrVr r rr rrWrer-r-r-r.rns  rnc@s>eZdZdZedddeedddddedd d d d gZdS) rgZISOTPSingleFrametyperra message_sizeNr!)Z length_ofr"cCs|jSrErpr?r-r-r. r"zISOTP_SF.)Z length_from) rRrSrTrVr ISOTP_TYPErr rWr-r-r-r.rgs  rgc@sJeZdZdZedddeedddeeddd d d ed d ddgZ dS)rhZISOTPFirstFramerorrarpr Zextended_message_sizercCs |jdkS)Nrrqrrr-r-r.rsr"zISOTP_FF.r!r"r3fmtN) rRrSrTrVrrtrr rrWr-r-r-r.rhs    rhc@s6eZdZdZedddeedddeddd d gZd S) riZISOTPConsecutiveFramerorraindexrr!r"r3rvN) rRrSrTrVrrtrrrWr-r-r-r.ris    ric @sDeZdZdZedddeedddddd d ed ded dgZd S)rjZISOTPFlowControlFramerorraZfc_flagrcontinuewaitabort)rrr block_sizeZseparation_timeN)rRrSrTrVrrtrrWr-r-r-r.rj#s  rj)0r:loggingZ scapy.compatrrrrrZ scapy.packetrZ scapy.fieldsrr r r r r rrrrrrrZscapy.layers.canrZ scapy.errorr getLoggerrKr0ZCAN_MTUZ CAN_MAX_DLENr9r<Z ISOTP_TYPESZN_PCI_SFZN_PCI_FFr=ZN_PCI_FCr rZrnrtrgrhrirjr-r-r-r. sF 4   +