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.7~rc2 ▾
|
wlanpi-profiler
/
opt
/
wlanpi-profiler
/
lib
/
python3.7
/
site-packages
/
profiler
/
__pycache__
/
helpers.cpython-37.pyc
|
|---|
B
ø uã
@ 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 ee!d
d d Z"ye"d
dgZ#W n* e$k
rR e d e e j ¡ Y nX ye"ddgZ#W n* e$k
r e d e e j ¡ Y nX ddl%m%Z% ddl&m'Z'm(Z( dZ)ddddZ*e!e+dddZ,e!e!d d d Z-ej.dd d Z/e!e0dd!d"d#Z1edd$d%Z2ej3ed&d'd(Z4e!ej3d)d*d+Z5e0dd,d-Z6e0dd.d/Z7ee0d&d0d1Z8e0dd2d3Z9edd&d4d5Z:e+e;d6d7d8Z<G d9d: d:ej=Z>e!dd;d<Z?d=d> Z@edd&d?d@ZAeG dAdB dBZBe0ddCdDZCdS )Eza
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.)ÚcmdÚreturnc C s: t j| ddddd}|jdkr6|jr*|jS |jr6|jS dS )z3Run an arbitrary CLI command and return the resultszutf-8FT)ÚencodingÚshellÚcheckÚcapture_outputr z)completed process return code is non-zero)Ú
subprocessÚrunÚ
returncodeÚstdoutÚstderr)r Úcp© r ú/build/wlanpi-profiler-CSvHZ8/wlanpi-profiler-1.0.7~rc2/debian/wlanpi-profiler/opt/wlanpi-profiler/lib/python3.7/site-packages/profiler/helpers.pyÚ run_cli_cmd- s
r Ztcpdumpz --versionzRproblem checking tcpdump version. is tcpdump installed and functioning? exiting...ÚnetstatzRproblem checking netstat version. is netstat installed and functioning? exiting...é )Ú __version__)ÚCHANNELSÚ CONFIG_FILEz/var/www/html/profiler)r c C sp | j r(| j dkrt j}| j dkr.t j}nt 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ÚINFOÚconfigÚ
dictConfig)ÚargsZ
logging_levelZdefault_loggingr r r Ú setup_loggerX s
r5 )Úvaluer c sJ t | d } dkr t|t fddt ¡ D r> S t|dS )zCheck if channel is validz %s is not a valid channel valuer c 3 s | ]} |kV qd S )Nr )Ú.0Zband)Úchannelr r ú <genexpr>{ s z check_channel.<locals>.<genexpr>N)ÚintÚ
ValueErrorÚanyr Úvalues)r6 Ú error_msgr )r8 r Ú
check_channelu s r? )Ússidr c C s t | dkrtd| | S )zCheck if SSID is validé z %s length is greater than 32)Úlenr; )r@ r r r Ú
check_ssid s rC c C sÒ t jt jdd} | jddddt jd | jdd td
d | jd d
dd | ¡ }|jddtdd | jdtdt dd | jdddddd |jd d ddd d | jd d!d"d#d$ | jd%d&ddd'd |jd(d)ddd*d | ¡ }|jd+d,ddt jd |jd-d.ddd/d | ¡ }|jd0d1ddt jd |jd2d3ddd4d | jd5d6ddt jd | jd7d8ddt jd | jd9d:ddt jd | jd;d<ddt jd | jd=d>ddd?d | jd@dAdBdCdD | jdEdFdGt
dH | S )Iz(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-cr8 z)set the operating channel to broadcast on)rG ÚtyperI z-iÚ interfacez"set network interface for profiler)rG rI z-sr@ zset profiler SSID namez--configÚFILEz<customize path for configuration file (default: %(default)s))rJ Úmetavarr# rI z --files_pathÚPATHÚ
files_pathz/var/www/html/profilerzZcustomize default directory where analysis is saved on local system (default: %(default)s))rM rG r# rI z--hostname_ssidÚ
hostname_ssidz>use the WLAN Pi's hostname as SSID name (default: %(default)s)z --loggingzchange logging outputú?)r r )rI Ú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--no_sniffer_filterZno_sniffer_filterz--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)rM rG rI z --versionz-Vr* )rH r* ) ÚargparseÚArgumentParserÚRawDescriptionHelpFormatterÚ add_argumentÚSUPPRESSr? Ú add_mutually_exclusive_grouprC Ústrr r )ÚparserZ
ssid_groupZ dot11r_groupZ
dot11ax_groupr r r Ú setup_parser sæ
re )Ú 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 )rc )r7 Ú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Úpathlibri ÚlistÚrglobr ÚjoinÚinputÚlowerÚstripÚsysÚexitÚosÚlistdirÚis_fileÚprintÚunlinkÚis_dirÚshutilÚrmtreeÚOSErrorÚ exception)rf rg Úlogri ÚresultÚ_pathÚpathr r r Ú
files_cleanup s&
r c C s¼ t t ¡ d d ¡}d}tj | j¡r@d}t| j}t |}n|
d| j¡ i }d|krbi |d< |sd|d d< d |d d
< d |d d < |d d
¡r¨t
¡ |d d< | jr¼| j|d d
< | jrÐ| j|d d < | jræ| j|d d< n| jrüt
¡ |d d< | jrd|d d< | jr&| j|d d< | jr:d|d d< | jrP| j|d d< | jrf| j|d d< | jr|| j|d d< | jr| j|d d< n t|d d< t|d d
|d d
< |S )zICreate the configuration (SSID, channel, interface, etc) for the Profilerr rh FTzcan not find config at %sÚGENERALzWLAN Pir@ é$ r8 Zwlan0rK rP rV rX rT r\ rO )r. rn ro rp rz r Úisfiler2 Ú load_configÚ convert_configparser_to_dictr ÚgetÚsocketÚ gethostnamer8 rK r@ rP rU rV rW rX rT r\ rO Ú
FILES_PATHr: )r4 r Z config_foundrd r2 r r r Ú setup_config/ sR
r )r2 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 r; )r2 Ú_dictÚsectionÚkeyÚ_valuer r r r n s r )Ú config_filer c C s t ¡ }| | ¡ |S )z!Load in config from external file)Ú configparserÚ ConfigParserÚread)r r2 r r r r s
r c C s t t ¡ d d ¡}t| s"dS y4t| d¡ d¡ t| d¡ d¡ t| W n4 t k
r |
dt ¡ d ¡ t
tj¡ Y nX d S )
z$Validate minimum config to run is OKr rh Fr r@ r8 z%sr T)r. rn ro rp Úcheck_config_missingrC r r? Ú verify_reporting_directoriesr; Úerrorrx Úexc_infory ÚsignalÚSIGABRT)r2 r 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)rv )r7 Úlocal)Úmacr r r9 s z is_randomized.<locals>.<genexpr>)Ú2Ú6ÚaÚe)r< )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 rh r z*missing general section from configurationrK z missing interface from configr8 zmissing channel from configr@ zmissing ssid from configz%sr FT) r. rn ro rp ÚKeyErrorÚkeysr rx r¡ )r2 r Úoptionsr r r r £ s r c C sÆ t t ¡ d d ¡} ytj tjd d¡}| d|¡ | dt
tj |¡¡ | d¡ t dt
j dd g}| d
t|¡ | dt
tj |¡¡ W n& tk
rÀ | d ¡ td d
S X dS )z9Manuf wrapper to update manuf OUI flat file from Internetr rh r
zmanuf file is located at %sz manuf file last modified at: %sz running 'sudo manuf --update'Zsudoz
/bin/manufz--updatez%sz7problem updating manuf. make sure manuf is installed...z
exiting...FT)r. rn ro rp rz r rt r
Ú__path__Úinfor Úgetmtimer rx Úprefixrc r r r} )r Z flat_filer r 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 rh r rO ZclientsZreportsN) r. rn ro rp r rz r Úisdirr Úmakedirsrt )r2 r rO Z clients_dirZ reports_dirr r r r Ï s r )r8 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)r8 Úfreqr 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# õ 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Úreplacerw r )Zwlanpi_versionÚ_fileÚlinesÚliner r r Úget_wlanpi_versionü s
6rÖ 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_object s
rÛ c C sØ | d d¡dkrÔg }| 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 }|r | d| d| d¡ n| d| ¡ | d| ¡ | d ¡ | d
¡ | d ¡ | d| ¡ | d¡ | d
¡ | d¡ tt|td}tdd| x|D ]}t| q°W td| d dS )z)Create message to display to users screenr rT Tz&Starting profiler in listen only mode:z) - Listening for association frames with rK z on channel r8 z2 - Results are saved locally and printed to screenú z
Instructions:z4 - Associate your Wi-Fi client to any AP on channel z5 - We should passively detect the association request)r Ú
ú~r@ z$Starting profiler AP with interface ú:z - Our fake AP SSID: z, - Associate your Wi-Fi client to our SSID: z' - Enter any random password to connectzJThe client will fail authentication but should send an association requestrÌ N)r ÚappendrB Úmaxr} )r2 ÚoutZ
header_lenrÕ r8 rK r@ r r r Úgenerate_run_message sL
"
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( Únamer6 Údb_keyr Údb_valueN)
rÆ rÇ rÈ rÉ rå rc Ú__annotations__r6 r r: ræ rç r r r r r rä H s
rä c C s | d|> @ dkS )z.Retrieve bit value from byte at provided indexr r r )ZbytevalÚindexr r r Úget_bitR s rê )DrÉ r] r ro r r. Úlogging.configrz r r¢ r r rx Úbase64r Z dataclassesr Údistutils.utilr Útimer Útypingr r r r r
ÚModuleNotFoundErrorr rå r} ry r£ rc r r
r r Ú constantsr r r r5 r: r? rC r^ re r r r r r r r¤ r« r r³ r rÀ r½ rà r¾ rÖ rÛ rã rä rê r r r r Ú<module>
sz
?
2