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__
/
helpers.cpython-37.pyc
|
|---|
B
°Æä^x.,ã
@ sâ d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl
Z
ddl Z ddl Z ddl
mZ ddlmZ ddlmZ ddlmZ ddlmZmZmZmZ y ddlZW nN ek
r Z z.e j dkràe d n
e e e e j ¡ W ddZ [ X Y nX d
d d d
dddddg Z!xBe!D ]:Z"e #e"¡dkr$e de" d e d e e j ¡ q$W ddl$m$Z$ ddl%m&Z&m'Z' dZ(dddd Z)e*e+d d d Z,e*e*d d!d"Z-e+dd#d$Z.ej/dd%d&Z0e*e1dd'd(d)Z2d*d+ Z3ej4ed,d-d.Z5e*ej4d/d0d1Z6e1dd2d3Z7e1dd4d5Z8ee1d,d6d7Z9dRe:e*d9d:d;Z;e1dd<d=Z<edd,d>d?Z=e+e>d@dAdBZ?G dCdD dDej@ZAe*ddEdFZBe*dGdHdIZCdJdK ZDedd,dLdMZEeG dNdO dOZFe1ddPdQZGdS )Sza
profiler.helpers
~~~~~~~~~~~~~~~~
provides init functions that are used to help setup the app.
é N)Ú b64encode)Ú dataclass)Ú strtobool)Úctime)ÚAnyÚDictÚListÚUnionÚmanufz required module manuf not found.ZtcpdumpÚiwÚipZethtoolZlspciZlsusbZmodprobeZmodinfoZwpa_cliz It looks like you do not have z installed.z3Please install using your distro's package manager.é )Ú __version__)ÚCHANNELSÚ CONFIG_FILEz/var/www/html/profiler)Úreturnc C s| | j r(| j dkrt j}| j dkr.t j}nt j}| jr:t j}dddddiid|dd d
d id dg|d
id}t j |¡ dS )z Configure and set logging levelsÚdebugÚwarningr
FÚstandardÚformatz1%(asctime)s [%(levelname)s] %(name)s: %(message)sÚdefaultzlogging.StreamHandlerzext://sys.stdout)ÚlevelÚ formatterÚclassÚstreamÚ )Úhandlersr )ÚversionÚdisable_existing_loggersÚ
formattersr ÚloggersN)ÚloggingÚDEBUGÚWARNINGÚINFOr ÚconfigÚ
dictConfig)ÚargsZ
logging_levelZdefault_logging© r( ú/build/wlanpi-profiler-7IIg1Q/wlanpi-profiler-1.0.11/debian/wlanpi-profiler/opt/wlanpi-profiler/lib/python3.7/site-packages/profiler/helpers.pyÚ setup_loggerH s$
r* )Úvaluer c s4 t | t fddt ¡ D r& S td dS )zCheck if channel is validc 3 s | ]} |kV qd S )Nr( )Ú.0Úband)Úchr( r) ú <genexpr>j s zchannel.<locals>.<genexpr>z%s is not a valid channelN)ÚintÚanyr ÚvaluesÚ
ValueError)r+ r( )r. r) Úchannelg s r4 )Ússidr c C s t | dkrtd| | S )zCheck if SSID is validé z %s length is greater than 32)Úlenr3 )r5 r( r( r) r5 o s r5 c C sv y t | } W n tk
r( tdY nX dddg}x0|D ](}|d | krZ|d kr:n q:| S q:W td| |dS ) z(Check if the provided frequency is validz%s is not a number)il i´ )i< i )iC iË r r
z*%s not found in these frequency ranges: %sN)r0 r3 )ÚfreqZ freq_rangesr- r( r( r) Ú frequencyv s
r9 c C s t jt jdd} | jddddt jd | ¡ }|jdd td
d |jd d
tdd | jdddd | ¡ }|jddtdd | jdt dt
dd | jddd d d d |jd d!ddd"d | jd#d$ddd%d | jd&d'd(d)d* | jd+d,ddd-d |jd.d/ddd0d | ¡ }|jd1d2ddt jd |jd3d4ddd5d | ¡ }|jd6d7ddt jd |jd8d9ddd:d | jd;d<ddt jd | jd=d>ddt jd | jd?d@ddt jd | jdAdBdddCd | jdDdEdFdGdH | jdIdJdddKd | jdLdMdddNd | jdOdPdQt dR | S )Sz(Set default values and handle arg parserzdwlanpi-profiler is an 802.11 client capabilities profiler. Read the manual with: man wlanpi-profiler)Úformatter_classÚ descriptionz--pytestZpytestÚ
store_trueF)ÚdestÚactionr Úhelpz-cr4 z set the channel to broadcast on)r= Útyper? z-fr8 z!set the frequency to broadcast onz-iÚ interfacez"set network interface for profiler)r= r? z-sr5 zset profiler SSID namez--configÚFILEz<customize path for configuration file (default: %(default)s))r@ Úmetavarr r? z --files_pathÚPATHÚ
files_pathz/var/www/html/profilerzZcustomize default directory where analysis is saved on local system (default: %(default)s))rC r= r r? z--hostname_ssidÚ
hostname_ssidz>use the WLAN Pi's hostname as SSID name (default: %(default)s)z--debugr zenable debug logging outputz --loggingzchange logging outputú?)r r )r? ÚnargsÚchoicesz--noprepZno_interface_prepz4disable interface preperation (default: %(default)s)z--noAPÚ listen_onlyz*enable Rx only mode (default: %(default)s)z--11rÚ
ft_enabledz--no11rÚ ft_disabledz/turn off 802.11r Fast Transition (FT) reportingz--11axÚ
he_enabledz--no11axÚ he_disabledz0turn off 802.11ax High Efficiency (HE) reportingz--cleanÚcleanz--filesÚfilesz--yesÚyesz --oui_updateZ
oui_updatez?initiates update of OUI database (requires Internet connection)z--readZPCAPÚ
pcap_analysisz5read and analyze association request frames from pcap)rC r= r? z--no_bpf_filtersZno_bpf_filterszFremoves BPF filters from sniffer() but may impact profiler performancez--list_interfacesZlist_interfacesz2print out a list of interfaces with an 80211 stackz --versionz-Vr )r> r ) ÚargparseÚArgumentParserÚRawDescriptionHelpFormatterÚ add_argumentÚSUPPRESSÚ add_mutually_exclusive_groupr4 r9 r5 Ústrr r )ÚparserZfrequency_groupZ
ssid_groupZ dot11r_groupZ
dot11ax_groupr( r( r) Ú setup_parser s
r[ )Ú directoryÚ acknowledgedr c C s t t ¡ d d ¡}ddlm} t|| d¡}| dd dd |D ¡¡ |rVn&t
d ¡ ¡ d
d
d ks|t
d ¡ yjxdt || ¡D ]R}|| || }| ¡ rÀtd
| | ¡ | ¡ rtd| t |¡ qW W n tk
r | d¡ Y nX d
S )zPurge files recursivelyr é )ÚPathÚ*z Delete the following files: %sz, c S s g | ] }t |qS r( )rY )r, Úxr( r( r) ú
<listcomp>- s z!files_cleanup.<locals>.<listcomp>zAre you sure? (y/n): Nr
ÚyzRemoving file: zRemoving directory: zissue removing files)r! Ú getLoggerÚinspectÚstackÚpathlibr_ ÚlistÚrglobr ÚjoinÚinputÚlowerÚstripÚsysÚexitÚosÚlistdirÚis_fileÚprintÚunlinkÚis_dirÚshutilÚrmtreeÚOSErrorÚ exception)r\ r] Úlogr_ ÚresultÚ_pathÚpathr( r( r) Ú
files_cleanup& s&
r~ c C s@ t t ¡ d d ¡}tj | j¡r8t| j}t |}n|
d| j¡ i }d|krZi |d< d|d krrd|d d< d|d krd|d d< d |d kr¢d
|d d < |d d ¡rÀt
¡ |d d< | jrÔ| j|d d< | jrö| j|d d < d|d d< n d|d d < | jr| j|d d < | jr0| j|d d< n| jrHt
¡ |d d< | jr\d
|d d< | jrr| j|d d< | jrd
|d d< | jr| j|d d< | jr²| j|d d< | jrÈ| j|d d< | jrà| j|d d< n t|d d< y.| d¡ d¡}|r t|}||d d< W n t k
r: |
d¡ Y nX |S )zICreate the configuration (SSID, channel, interface, etc) for the Profilerr r^ zcan not find config at %sÚGENERALr4 é$ r5 zWLAN PirA Zwlan0rF r9 FrL rN rJ rR rE z(config.ini does not have channel defined) r! rd re rf rp r} Úisfiler% Ú load_configÚ convert_configparser_to_dictr ÚgetÚsocketÚ gethostnamer4 r8 rA r5 rF rK rL rM rN rJ rR rE Ú
FILES_PATHr0 ÚKeyError)r' rz rZ r% r. r( r( r) Ú setup_configA sf
r )r% r c
C sl i }xb| ¡ D ]V}i ||< xH| |¡D ]:\}}ytt|}W n tk
rR Y nX ||| |< q&W qW |S )a
Convert ConfigParser object to dictionary.
The resulting dictionary has sections as keys which point to a dict of the
section options as key => value pairs.
If there is a string representation of truth, it is converted from str to bool.
)ÚsectionsÚitemsÚboolr r3 )r% Ú_dictÚsectionÚkeyÚ_valuer( r( r) r s r )Ú config_filer c C s t ¡ }| | ¡ |S )z!Load in config from external file)Ú configparserÚ ConfigParserÚread)r r% r( r( r) r ¡ s
r c C s¼ t t ¡ d d ¡}t| s"dS y`| d¡ d¡}|r@t| | d¡ d¡}|r\t| | d¡ d¡}|rxt| t | W n4 t
k
r¶ | dt
¡ d ¡ t tj¡ Y nX d
S ) z$Validate minimum config to run is OKr r^ Fr r5 r4 r9 z%sr
T)r! rd re rf Úcheck_config_missingr r5 r4 r9 Ú verify_reporting_directoriesr3 Úerrorrn Úexc_inforo ÚsignalÚSIGABRT)r% rz Z_ssidr. r8 r( r( r) Úvalidate¨ s$ r c s t fdddD S )zECheck if MAC Address <format>:'00:00:00:00:00:00' is locally assignedc 3 s | ]}| ¡ d kV qdS )r
N)rl )r, Úlocal)Úmacr( r) r/ Æ s z is_randomized.<locals>.<genexpr>)Ú2Ú6ÚaÚe)r1 )r r( )r r) Ú
is_randomizedÄ s r¢ c C s t t ¡ d d ¡}yPd| kr(td| d ¡ }d|krDtdd|krTtdd |krdtd
W n( tk
r | d t ¡ d ¡ d
S X dS )z)Check that the minimal config items existr r^ r z*missing general section from configurationrA z missing interface from configr4 zmissing channel from configr5 zmissing ssid from configz%sr
FT) r! rd re rf r Úkeysr rn r )r% rz Úoptionsr( r( r) r É s r F)Úcmdr c C s4 t j| ddddd}|s0|jr$|jS |jr0|jS dS )zMRun a single CLI command with subprocess and return stdout or stderr responsezutf-8FT)ÚencodingÚshellÚcheckÚcapture_outputzBcompleted process return code is non-zero with no stdout or stderr)Ú
subprocessÚrunÚstdoutÚstderr)r¥ Zsuppress_outputÚcpr( r( r) Ú run_commandÝ s r¯ c C sÞ t t ¡ d d ¡} ytj tjd d¡}t j
d}| d|¡ | d|¡ | dt tj
|¡¡ | d¡ td |d
g}| d t|¡ d |kr°| dt tj
|¡¡ W n& tk
rØ | d
¡ td dS X dS )z9Manuf wrapper to update manuf OUI flat file from Internetr r^ r
z
/bin/manufz OUI database is located at %szmanuf is located at %sz manuf file last modified at: %sz running 'sudo manuf --update'Zsudoz--updatez%sÚURLErrorz7problem updating manuf. make sure manuf is installed...z
exiting...FT)r! rd re rf rp r} rj r
Ú__path__rn ÚprefixÚinfor Úgetmtimer¯ rY rx ry rs )rz Z flat_fileZmanuf_locationÚoutr( r( r) Ú update_manufð s*
r¶ c C s t t ¡ d d ¡}d| kr| d d¡}tj |¡sH| t |¡¡ tj
|d¡}tj |¡sr| t |¡¡ tj
|d¡}tj |¡s| t |¡¡ dS )z3Check reporting directories exist and create if notr r^ r rE ZclientsZreportsN) r! rd re rf r rp r} Úisdirr Úmakedirsrj )r% rz rE Z clients_dirZ reports_dirr( r( r) r s r )r4 r c C sD | dkr d}| dk r"d| d }n| dkr6d| d }|j dddS ) zGTake a channel number, converts it to a frequency, and finally to bytesé i´ ig é i é Úlittle)Ú byteorder)Úto_bytes)r4 r8 r( r( r) Úget_frequency_bytes s r¿ c @ s e Zd ZdZdd ZdS )Ú
Base64EncoderzA Base64 encoder for JSONc C s$ t |trt| ¡ S tj | |¡S )z Perform default Base64 encode)Ú
isinstanceÚbytesr ÚdecodeÚjsonÚ JSONEncoderr )ÚselfÚobjr( r( r) r 1 s
zBase64Encoder.defaultN)Ú__name__Ú
__module__Ú __qualname__Ú__doc__r r( r( r( r) rÀ + s rÀ c C s d} ybt dP}| ¡ ¡ }x<|D ]4}d|kr"d | d¡d dd¡ d d¡ ¡ ¡} q"W W d
Q R X W n tk
rz Y nX | S ) z Retrieve system image versonÚunknownz/etc/wlanpi-releaseÚVERSIONz{0}ú=r
ú"r ú'N)Úopenr Ú
splitlinesr ÚsplitÚreplacerm rx )Zwlanpi_versionÚ_fileÚlinesÚliner( r( r) Úget_wlanpi_version8 s
6rØ )r5 c C sJ t t ¡ d d ¡}d}t|d }| | ¡ | d| ¡ W dQ R X dS )z&Update SSID record on local filesystemr r^ z/etc/wlanpi-profiler/ssidÚwz updated ssid record with: %sN)r! rd re rf rÑ Úwriter )r5 rz Zssid_file_pathrÕ r( r( r) Úupdate_ssid_recordH s
rÛ c c s: t | } t| }x| D ]}|dfV |}qW |dfV dS )z0Treat the last object in an iterable differentlyFTN)ÚiterÚnext)ÚseqZ_aÚ_br( r( r) Úflag_last_objectS s
rà c C sÔ | d d }| d d¡dkrâg }| d| d| d d d| d d d
¡ | d ¡ | d ¡ | d ¡ | d
¡ | d¡ | d¡ | d¡ tt|td}tdd| x|D ] }t| q¾W td| d nîg }| d d }| d| d| d d d| d d d
¡ | d ¡ | d ¡ | d ¡ | d| ¡ | d¡ | d¡ | d| d d ¡ | d¡ tt|td}tdd| x|D ]}t| q¬W td| d d S ) z)Create message to display to users screenr rA rJ Tz,Starting profiler in listen only mode using z on r4 z (r9 ú)ú zGetting started:zK - Associate your Wi-Fi client to *any* SSID on the channel/frequency abovez5 - Any detected association requests will be profiledz? - Capabilities may vary depending on SSID and AP configurationz: - Results are then saved locally and printed on the shell)r Ú
ú~r5 zStarting a fake AP using z on channel z. - Associate your Wi-Fi client to *our* SSID: z' - Enter any random password to connectz( - Authentication will fail, which is OKz/ - We should receive an association request to r N)r Úappendr7 Úmaxrs )r% rA rµ Z
header_lenr× r5 r( r( r) Úgenerate_run_message] sH *
*
rç c @ sX e Zd ZU dZdZeed< dZeee f ed< dZ
eed< dZ ee ee e f ed< dS ) Ú
Capabilityz"Define custom fields for reportingr Únamer+ Údb_keyr Údb_valueN)
rÈ rÉ rÊ rË ré rY Ú__annotations__r+ r r0 rê rë r r( r( r( r) rè s
rè c C s | d|> @ dkS )z.Retrieve bit value from byte at provided indexr
r r( )ZbytevalÚindexr( r( r) Úget_bit s rî )F)HrË rS r re rÄ r! Úlogging.configrp rv r r
rª rn Úbase64r Z dataclassesr Údistutils.utilr Útimer Útypingr r r r r
ÚModuleNotFoundErrorr ré rs ro r Z__toolsZtoolÚwhichr Ú constantsr r r r* rY r0 r4 r5 r9 rT r[ r r~ r r r r r r¢ r rh r¯ r¶ r r r¿ rÅ rÀ rØ rÛ rà rç rè rî r( r( r( r) Ú<module>
s
K
/