Why Gemfury?
Push, build, and install
RubyGems
npm packages
Python packages
Maven artifacts
PHP packages
Go Modules
Debian packages
RPM packages
NuGet packages
dfinimundi / wlanpi-profiler deb
Repository URL to install this package:
|
Version:
1.0.11 ▾
|
wlanpi-profiler
/
opt
/
wlanpi-profiler
/
lib
/
python3.7
/
site-packages
/
profiler
/
__pycache__
/
interface.cpython-37.pyc
|
|---|
B
P¥1呅vã @ s d Z ddlZddlZddlZddlmZ ddlmZmZ ddl m
Z
ddl m Z m
Z
G dd d eZG d
d d ZG d d
d
ZdS )zB
profiler.interface
~~~~~~~~~~~~~~~~~~
wlan interface data class
é N)Ú
namedtuple)ÚDictÚListé )Ú _20MHZ_FREQUENCY_CHANNEL_MAP)Úflag_last_objectÚ run_commandc @ s e Zd ZdZdS )ÚInterfaceErrorzQCustom exception used when there are problems staging the interface for injectionN)Ú__name__Ú
__module__Ú __qualname__Ú__doc__© r r ú/build/wlanpi-profiler-7IIg1Q/wlanpi-profiler-1.0.11/debian/wlanpi-profiler/opt/wlanpi-profiler/lib/python3.7/site-packages/profiler/interface.pyr s r c @ s e Zd ZdZdd ZdS )ÚInterfaceInformationz$Base class for Interface Informationc C s. || _ || _|| _|| _|| _|| _|| _d S )N)ÚphyÚ interfaceÚmodeÚdriverÚdriver_versionÚ firmware_revÚchipset)Úselfr r r r r r r r r r Ú__init__" s
z InterfaceInformation.__init__N)r
r r r
r r r r r r s r c @ s¨ e Zd ZdZdd Zdd Zedd Zdd d
d Zdd d d
Z dd ddZ
dd ddZ e d ddZ
e d ddZdBeedddZdd ddZeed d d ZdCdd d d Zeed d!d"Zdd d#d$Zed d%d&Zed d'd(Zed d)d*Zed d+d,Z ed d-d.Z ed d/d0Z ed d1d2Z ed3d4 Z ed5d6 Z!edDd7d8Z"dEed d:d;Z#ee d d<d=Z$ed d>d?Z%dFed d@dAZ&dS )GÚ InterfacezWLAN Interface data classc C sB t | jj ¡ ¡| _d| _d | _d | _d| _ g | _
d| _ d| _ d S )NÚ F)
ÚloggingÚ getLoggerÚ __class__r
ÚlowerÚlogÚnameÚchannelÚ frequencyÚ requires_vifÚphysÚno_interface_prepÚremoved)r r r r r 8 s zInterface.__init__c C s | j std| | j ¡| _| | j ¡}| |¡| _| |¡| _| | j ¡| _
| | j ¡ | ¡ | _
d| j
| _d| _| jr´tdd| j dg}| || j ¡| _| || j ¡| _nbd| jkr| j d| _| j| j krü| j d | j| j | j
¡ td
n| j d | j| j
¡ d | _| js*td
| j | js>td| j | j d| j| j¡ | ¡ | _| ¡ | _ | j dkr~td| j | ¡ | _ | ¡ | !¡ dS )z Perform setup for the interfacezinterface name not setr r ÚiwÚdevÚinfoÚ88XXauÚmonzCproposed %s interface matches provided %s and already maps to phy%sziwlwifi requires use of a separate monitor interface. did you already handle interface staging and mean to run with --noprep option?znew %s will map to phy%sTz unknown channel setting for %sz unknown frequency setting for %sz/frequency is set to %s which maps to channel %s)ÚmanagedÚmonitorz$%s is mode is not managed or monitorN)"r! r Ú
get_driverr Úget_ethtool_infoÚget_driver_versionr Úget_firmware_revisionZfirmware_revisionÚ get_chipsetr Úcheck_interface_stackÚ
get_phy_idÚphy_idr r, r& r Ú
get_frequencyr# Ú get_channelr" r ÚwarningÚdebugr$ Úget_macÚmacÚget_moder Ú
get_operstateÚ operstateÚchecksÚ log_debug)r Ú
eth_tool_infoÚiw_dev_iface_infor r r ÚsetupB sT
zInterface.setupc C sZ g }xB| D ]:}t t||tr0| t||¡ q
| tt||¡ q
W tdd |D S )z<Find all matching attributes in a list and return max lengthc s s | ]}t |V qd S )N)Úlen)Ú.0Úxr r r ú <genexpr> s z-Interface.get_attr_max_len.<locals>.<genexpr>)Ú
isinstanceÚgetattrÚstrÚappendÚmax)Z
searchListÚattrZ_listÚ_objr r r Úget_attr_max_lenw s
zInterface.get_attr_max_lenN)Úreturnc C sÆ t ddg}t| | t ddg¡| _| j d| j¡ g }x| jD ]~}xx|jD ]n}| |j¡}| |j¡}|
|¡}| |¡}| |j¡} | j
|jd}
| td|j |j|
|||| ¡ qPW qDW | ¡ | dtd d
d d d
dd¡ | dtddddddd¡ t |d¡} t |d¡} t |d¡}
t |d¡}t |d¡}t |d¡}t |d¡}t ddg}t| d}xF|D ]>}|dj|j|j|j|j|j|j|j | | |
||||d7 }qxW t| d S ) zPrint wiphys to the screenÚ lsb_releasez-ar( r) zphys: %s)Úifacer r ZPHYr ZModeZDriverZ
DriverVersionZ FirmwareRevZChipsetr ú r r r r r r Úunamer z{0:<{phy_len}} {1:<{interface_len}} {2:<{mode_len}} {3:<{driver_len}} {4:<{driverv_len}} {5:<{firmwarer_len}} {6:<{chipset_len}}
)Úphy_lenÚ
interface_lenÚmode_lenÚ
driver_lenÚ driverv_lenÚ
firmwarer_lenÚ chipset_lenN) r ÚprintÚbuild_iw_phy_listr% r r: Ú
interfacesr0 r! r/ r1 r2 r3 r= rL r r6 ÚreverseÚinsertr rP Úformatr r r r r r r )r rR Úifacesr rS rB r r r r r rV rW rX rY rZ r[ r\ rU Úoutr r r Úprint_interface_information sz
z%Interface.print_interface_informationc C sx t dddg}dd | d¡D }dd |¡krTd | jkrt| j d
¡ | j d ¡ n | j d d
|¡¡ | j d¡ dS )z*Check and report the set regulatory domainr( ÚregÚgetc S s g | ]}d |kr|qS )Zcountryr )rF Úliner r r ú
<listcomp>Ï s z.Interface.check_reg_domain.<locals>.<listcomp>Ú
ZUNSETr ZiwlwifizBreg domain appears unset. consider setting it with 'iw reg set XX'zZhttps://wireless.wiki.kernel.org/en/users/documentation/iw#updating_your_regulatory_domainzreg domain set to %srT z see 'iw reg get' for detailsN)r ÚsplitÚjoinr r r9 r: )r Zregdomain_resultZ regdomainr r r Úcheck_reg_domainÌ s
zInterface.check_reg_domainc C s` ddd| j dgdd| j dgddd| j dgg}x(|D ] }| j d d
|¡¡ t| q8W d S ) z.Delete monitor interface and restore interfaceÚipÚlinkÚsetÚdownr( r) ÚdelÚupzrun: %srT N)r, r! r r: rl r )r ÚcommandsÚcmdr r r Úreset_interfaceÜ s
zInterface.reset_interfacec C st ddd| j dgdd| j dddgddd| j d gd| j d
gg}x,|D ]$}| j d d |¡¡ t|d
d qHW dS )z4Perform scan in attempt to enable a disabled channelrn ro rp rq r( r) Útyper- rs Úscanzrun: %srT T)Zsuppress_outputN)r! r r: rl r )r Ziwlwifi_scan_commandsru r r r rx ç s
zInterface.scanc C sL ddd| j dgdd| j dddgddd| j d gd| j dd
| j d ggS ) z+Retrieve generic interface staging commandsrn ro rp rq r( r) rw r. rs r" ÚHT20)r! r" )r r r r Ú get_generic_staging_commandsó s z&Interface.get_generic_staging_commandsc C sZ d| j dd| j ddddg dd d
| j d gdd d
| j d gd| j d
d
| j dgg}|S )z5Retrieve interface staging commands for iwlwifi cardsr( r Úaddrw r. ÚflagsÚnonern ro rp rs rq Úfreqry )r r, r! r# )r Úcmdsr r r Ú get_iwlwifi_staging_commandsü s z&Interface.get_iwlwifi_staging_commandsF)r~ rQ c C s | |¡}|rx|| ¡ D ]p\}}xf|D ]^}t|t|jkr&|jsH|jr&|r|jrf| j d|j|j¡ |jr| j d|j|j¡ dS q&W qW dS )zDCheck iw phy channels for disabled or No IR and return True if foundz Channel is disabled for %s (%s)zZNo IR found in iw channel information for %s (%s) which means injection will not function!TF) Úget_channels_statusÚitemsÚintr~ ÚdisabledÚno_irr r9 Úch)r r~ Úiw_phy_channelsÚverboseZchannels_statusZ_bandÚchannelsr r r r Ú#check_for_disabled_or_noir_channels s&
z-Interface.check_for_disabled_or_noir_channelsc C s t ddg}|r.| j d| ¡ d dd¡¡ t ddg}|rP| j d | ¡ ¡ t d
d g}|rr| j d | ¡ ¡ dd | j d
g}t | g }d| jkr¢| ¡ }nB| ¡ }|
| j t d
d| j dg¡rä| j d| j| j ¡ |
¡ xp|D ]h}| j dd |¡¡ d|krJt | ¡ }d|krR| j |¡ d|krRt| j dqêt | qêW | j| jd| _d| jkrztd| j
| j t d
d| j dgdd d S ) z4Prepare the interface for monitor mode and injectionZwpa_cliz-vzwpa_cli version is %sr zwpa_cli r rn z-Vz%sr( z --versionz-iÚ terminater+ r r zBinitiating scan on %s because checks found disabled or No IR on %szrun: %srT r. znon-zeroz
not supportedz( does not appear to support monitor mode)rS z interface is not in monitor modeT)r N)r r r: Ú
splitlinesÚreplaceÚstripr! r rz r r r# r rx rl r r= r, r )r Zwpa_cli_versionZ
ip_versionZ
iw_versionZ wpa_cli_cmdr ru Ústdoutr r r Ústage_interface- sR
zInterface.stage_interfacec C sÈ t t ¡ d d ¡}| r"d| kr0| d¡ i S d}d}d}d}d}d}d}i } g }
tdd d
d d g} xZt| ¡ D ]H\} }
| ¡ ¡ } |r®d}|
d
¡r®| d¡d }qv|rêd}|
d¡rê| ¡ d }| ¡ d dd¡ dd¡}qv|
d¡rød}|
d¡r^|
| ||||¡ d}d}d}d}d | kr6d}| ¡ d }| ¡ d dd¡ dd¡}|
d
¡r |
| ||||¡ |
| |< g }
d}d}| d¡d }|
rv|
| ||||¡ |
| |< qvW | S )z:Run `iw phy phyX channels` and analyze channel informationr é zcommand failedz$unable to detect valid channels fromr FTr" r~ r r
r zband rT r Ú*ú[ú]zno ir)r r ÚinspectÚstackr9 r r r r r Ú
startswithrk r rL )r r r~ r r
Zbandr Z
first_bandZfirst_channel_in_bandZbandsr r" rh Ú last_liner r r r n sd
z Interface.get_channels_statusc C sl | j s
|r(d| jkr(| j d| j| j¡ | j s2|r`| j}| jrD| j}d| jkr`| j d|| j¡ | ¡ dS )z*Perform self checks and warn as neccessaryr. z3%s mode is in %s mode when we expected monitor moders z*%s operating state is %s when we expect upN) r& r r r9 r! r$ r, r? rm )r Zstagedr! r r r r@ ¨ s
zInterface.checks)r rQ c C s g }xDt d¡D ]6}t j d|¡}t j |¡rdt |¡kr| |¡ qW ||krx| j d|d |¡¡ t| dn| j d|¡ |S dS )z]Check that the interface we've been asked to run on actually exists and has an mac80211 stackz/sys/class/netZphy80211zW%s interface does not support the mac80211 stack. here are some interfaces which do: %sz, z% is not detected as a valid interfacez%s has a mac80211 stackN)
ÚosÚlistdirÚpathrl ÚisdirrL r r9 r r: )r r Zdiscovered_interfacesrS Z
iface_pathr r r r4 ¿ s z Interface.check_interface_stackc C s$ | j d| j| j| j| j| j¡ dS )z Send debug information to loggerzAmac: %s, channel: %s, driver: %s, driver-version: %s, chipset: %sN)r r: r< r" r r r )r r r r rA Ô s zInterface.log_debugc C s t dd| g}| ¡ S )z(Gather ethtool information for interfaceÚethtoolz-i)r r )r rS r r r r r0 ß s zInterface.get_ethtool_infoc C s( t ddd| dg}| d¡d ¡ S )z'Gather driver information for interfaceÚreadlinkz-fz/sys/class/net/z/device/driverú/éÿÿÿÿ)r rk r )r rS r r r r r/ ä s zInterface.get_driverc C s8 d}x.| ¡ ¡ D ] }| d¡r| dd¡ ¡ }qW |S )z#Gather driver version for interfacer zversion:)r r r r r )r rB rd rh r r r r1 ë s
z Interface.get_driver_versionc C s8 d}x.| ¡ ¡ D ] }| d¡r| dd¡ ¡ }qW |S )z,Gather driver firmware version for interfacer zfirmware-version:)r r r r r )r rB rd rh r r r r2 ó s
z Interface.get_firmware_revisionc C sZ ddddddddd d
d d d
ddddg}x |D ]}||kr,| |d¡}q,W d | ¡ ¡}|S )zRemove extraneous wordszWireless LAN ControllerszNetwork ConnectionzWireless Adapterz WLAN Adapterú(ú)z Corporation.Z CorporationzCorp.ZCorpzInc.ZIncz Technology,Z
Technologyú,ú"z rT )r rl rk )r r ÚwordsÚwordr r r Úcleanup_chipsetû s,
zInterface.cleanup_chipsetc C sZ t dd| dg}| d¡d }d}|dkr| d¡d dd
d
d¡}t d d | g}| d¡d
dd
¡ }| |¡}|S |dkrþt dd| dg ¡ }t dd| dg ¡ }t dd | d| dg}| d¡d
¡ ¡ d }| |¡}|S |dkrRt dd| dg ¡ }t dd| dg ¡ }| d| dkrVd}nd}|S )z(Gather chipset information for interfaceÚcatz/sys/class/net/z/device/modaliasú:r r Zusbr é
ÚpZlsusbz-dé é NZpciz/device/vendorz/device/deviceZlspciz-qZsdioz
0x02d0:0xa9a6zBroadcom 43430ÚUnknown)r rk r r r§ r )r rS ZmodaliasZbusr ZbusinfoÚvendorZdevicer r r r3 s6
zInterface.get_chipsetc C s" t dd| j dg}| ¡ ¡ S )z(Gather MAC address for a given interfacer¨ z/sys/class/net/z/address)r r! r r )r r< r r r r; 9 s zInterface.get_macc C s t j| |ddS )z0Determine which frequency the interfac is set toT)r7 )r Úparse_iw_dev_iface_info)rC rS r r r r7 > s zInterface.get_frequencyc C s t j| |ddS )z/Determine which channel the interface is set toT)r8 )r r° )rC rS r r r r8 E s zInterface.get_channelc C sÎ t t ¡ d d ¡}x²| ¡ D ]¦}| ¡ ¡ }d|kr t| d¡d d¡d }t| d¡d d¡d dd ¡}t
|d¡}||kr| d
||¡ |r°|
d ||¡ |S |r |
d ||¡ |S q W d
S )z;Determine what channel or frequency the interface is set tor r r" r£ rT r r¬ r¡ r z9iw reported a different channel (%s) than our lookup (%s)z.get_frequency returns %s from `iw dev %s info`z,get_channel returns %s from `iw dev %s info`N)r r r r r r r r rk r r rg r9 r: ) rC rS r7 r8 r rh r" r~ Úrespr r r r° L s2 $ z!Interface.parse_iw_dev_iface_infor c C s* |s
| j }tdd| dg}| ¡ ¡ S )a
Get the current operating state of the interface.
https://www.kernel.org/doc/Documentation/ABI/testing/sysfs-class-net
What: /sys/class/net/<iface>/operstate
Date: March 2006
KernelVersion: 2.6.17
Contact: netdev@vger.kernel.org
Description:
Indicates the interface RFC2863 operational state as a string.
Possible values are:
"unknown", "notpresent", "down", "lowerlayerdown", "testing",
"dormant", "up".
r¨ z/sys/class/net/z
/operstate)r! r r r )r rS r? r r r r> n s zInterface.get_operstatec C sÖ t t ¡ d d ¡}tdddddg}tdd d
g}g }g }d }d }d }d } d }
d } xxt| ¡ D ]f\} }
| ¡ ¡ } |r¦|
rg S | d
¡r¦d}|
d¡d }qf| d¡rød| ksÀd| krÐ| d| |¡ qf|rè| ||| |
| ¡ |
d¡d }qf| d¡r|
d¡d } qf| d¡r0|
d¡d }
qf| d¡rL|
d¡d } qf| d
¡r¦| ||| |
| ¡ | |||¡ d }d }g }d } d }
d } |
d¡d ¡ }|
rf| ||| |
| ¡ | |||¡ qfW |S )zCreate map of phy to ifacer r rS r! ÚifindexÚaddrrw r r6 r_ Tr zphy#Fú#r z
interface Zunnamedz
non-netdevz skipping %s in phy%s detectionrT zifindex zaddr ztype )
r r r r r r r r r r rk r: rL )Ziw_devsr rS r r% rc Z first_phyr6 Z
iface_namer² r³ Ú_typerh Z is_last_liner r r r^ sf
zInterface.build_iw_phy_listc C sb | tddg¡| _d}xD| jD ]:}x4|jD ]*}| j|jkr,| j d|j|j¡ |j}q,W q W |S )zCCheck and determines the phy# for the interface name of this objectr( r) r zphy%s maps to provided %s)r^ r r% r_ r! r r: r6 )r r6 r rS r r r r5 Æ s zInterface.get_phy_idc C s |s
| j }tdd| dg}d}y t|}W n tk
rB |S X |dkrRd}n(|dkr`d}n|d krnd}n |d
krzd}| ¡ S ) z?Get the current mode of the interface {unknown/managed/monitor}r¨ z/sys/class/net/z/typeÚunknownr r- i! r. i" i# )r! r r Ú
ValueErrorr )r rS Z_interface_typer rµ r r r r= Ò s" zInterface.get_mode)F)F)FF)r )r )'r
r r r
r rD Ú staticmethodrP re rm rv rx r rz r r Úboolr r r r r@ rK r4 rA r0 r/ r1 r2 r§ r3 r; r7 r8 r° r> r^ r5 r= r r r r r 5 sD
5 J A9 # C r )r
r r r Ú collectionsr Útypingr r Ú constantsr Zhelpersr r Ú Exceptionr r r r r r r Ú<module>
s