a k#d@sZdZddlmZddlmZddlmZddlmZm Z m Z m Z m Z ddl mZddlmZmZmZdZd ee<d Zd Zd ZGd ddeZGdddeZGdddeZGdddeZGdddeZGdddeZGdddeZGdddeZGdddeZej eej!eej"eej#eej$eej%eej&eiZ'eeeedeeeedd S)!a- MACControl ~~~~~~~~~~ :author: Thomas Tannhaeuser, hecke@naberius.de :description: This module provides Scapy layers for the MACControl protocol messages: - Pause - Gate - Report - Register/REQ/ACK - Class Based Flow Control normative references: - IEEE 802.3x :NOTES: - this is based on the MACControl dissector used by Wireshark (https://github.com/wireshark/wireshark/blob/master/epan/dissectors/packet-maccontrol.c) )orb) ETHER_TYPES)Scapy_Exception)IntField ByteField ByteEnumField ShortFieldBitField)Packet)EtherDot1Q bind_layersiZ MAC_CONTROLc@seZdZdZdZdZdZdZdZdZ dZ ed ed ed ed ed e de diZ dZ dZ dZdZe d e dedediZddZddZddZdS) MACControlz01:80:c2:00:00:01rrripauseZgatereportz register reqregisterZ register_ackzclass based flow controlZ deregisterZackZnackcCsFz&t|dd>t|d}t|WSty8Yn0t||S)Nrr)rMAC_CTRL_CLASSESKeyErrorr guess_payload_class)selfpayloadZop_coderP/var/www/html/gsl.tools/lib/python3.9/site-packages/scapy/contrib/mac_control.pyrTs   zMACControl.guess_payload_classcCsB|j}d}|r(t|tr(|d7}|j}q |r>t|tr>|d7}|S)zY get the total size of all under layers :return: number of bytes rr)Z underlayer isinstancer r )rZ under_layerunder_layers_sizerrr_get_underlayers_size^sz MACControl._get_underlayers_sizecCs8|s0|}t||}|dkr0|dd|S||S)z add padding to the frame if required. note that padding is only added if pay is None/empty. this allows us to add # noqa: E501 any payload after the MACControl* PDU if needed (piggybacking). @)r#len)rZpktZpayr" frame_sizerrr post_buildrs  zMACControl.post_buildN)__name__ __module__ __qualname__ZDEFAULT_DST_MAC OP_CODE_PAUSE OP_CODE_GATEOP_CODE_REPORTOP_CODE_REGISTER_REQOP_CODE_REGISTEROP_CODE_REGISTER_ACK OP_CODE_CLASS_BASED_FLOW_CONTROLZOP_CODESZ FLAG_REGISTERZFLAG_DEREGISTERZFLAG_ACKZ FLAG_NACKREGISTER_FLAGSrr#r(rrrrr0s8  rc@s eZdZdS)MACControlInvalidSpeedExceptionN)r)r*r+rrrrr4sr4c@s.eZdZedejeddgZefddZdS)MACControlPause_op_code pause_timercCs<z|jtdtdtdi|WSty6tdYn0dS)a+ get pause time for given link speed in seconds :param speed: select link speed to get the pause time for, must be ETHER_SPEED_MBIT_[10,100,1000] # noqa: E501 :return: pause time in seconds :raises MACControlInvalidSpeedException: on invalid speed selector gH ?g:0y>g& .>zKInvalid speed selector given. Must be one of ETHER_SPEED_MBIT_[10,100,1000]N)r7ETHER_SPEED_MBIT_10ETHER_SPEED_MBIT_100ETHER_SPEED_MBIT_1000rr4)rspeedrrrget_pause_times  zMACControlPause.get_pause_timeN) r)r*r+rrr, fields_descr:r<rrrrr5s r5c@s"eZdZedejeddgZdS)MACControlGater6 timestamprN)r)r*r+rrr-rr=rrrrr>s r>c@s6eZdZedejeddeddeje ddgZ dS)MACControlReportr6r?rflagsZpending_grantsN) r)r*r+rrr.rrr3rr=rrrrr@s   r@c@sFeZdZedejeddeddeddejedde ddgZ dS) MACControlRegisterReqr6r?rZ assigned_portrAZ sync_timeZechoed_pending_grantsN) r)r*r+rrr/rrr3rr=rrrrrBs  rBc@s>eZdZedejeddeddejeddeddgZ dS)MACControlRegisterr6r?rrAechoed_assigned_portechoed_sync_timeN) r)r*r+rrr0rrr3r=rrrrrCs   rCc@s>eZdZedejeddeddejeddeddgZ dS)MACControlRegisterAckr6r?rrArDrEN) r)r*r+rrr1rrr3r=rrrrrFs   rFc@seZdZedejeddedddedddedddeddded dded dded dded dded deddeddeddeddeddeddeddgZdS)MACControlClassBasedFlowControlr6 _reservedrZ c7_enabledrZ c6_enabledZ c5_enabledZ c4_enabledZ c3_enabledZ c2_enabledZ c1_enabledZ c0_enabledZ c0_pause_timeZ c1_pause_timeZ c2_pause_timeZ c3_pause_timeZ c4_pause_timeZ c5_pause_timeZ c6_pause_timeZ c7_pause_timeN) r)r*r+rrr2rr r=rrrrrGs&         rG)typeN)(__doc__Z scapy.compatrZ scapy.datarZ scapy.errorrZ scapy.fieldsrrrrr Zscapy.layers.dot11r Zscapy.layers.l2r r r ZMAC_CONTROL_ETHER_TYPEr8r9r:rr4r5r>r@rBrCrFrGr,r-r.r/r0r1r2rrrrrs<    U