a k#d\@sdZddlmZddlmZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlZddlmZddlmZmZmZddlmmZddlmZmZmZddlmZdd lmZm Z m!Z!m"Z"m#Z#d d iZ$gd Z%d dZ&e'e(dfddZ)ddZ*e&dZ+e&dZ,ddZ-d7ddZ.d8ddZ/d9ddZ0d:ddZ1d;d!d"Z2d#d$Z3d%d&Z4dd-d.Z7d?d/d0Z8d@d2d3Z9ddde j:fd4d5Z;es  rTc Cstd|zDt|(}tt||d||Wdn1sD0YWnZty}z |rftd||WYd}~n,d}~0ty|rt d|Yn0dS)aRead a config file: execute a python file while loading scapy, that may contain some pre-configured values. If _globals or _locals are specified, they will be updated with the loaded vars. This allows an external program to use the function. Otherwise, vars are only available from inside the scapy console. params: - _globals: the globals() vars - _locals: the locals() vars - interactive: specified whether or not errors should be printed using the scapy console or raised. ex, content of a config.py file: 'conf.verb = 42 ' Manual loading: >>> _read_config_file("./config.py")) >>> conf.verb 2 zLoading config file [%s]execNz!Cannot read config file [%s] [%s]z+Error during evaluation of config file [%s]) rdebugopenrcompilereadIOErrorwarning Exception exception)rZ_globalsZ_locals interactiveZcfgferrr_read_config_fileIs"  &" r'cCs|ddko|dvS)z5Returns whether or not a variable should be imported.r_)rangemapr)krrr_validate_localtsr,z.scapy_prestart.pyz.scapy_startup.pycCstdtddS)NzUsage: scapy.py [-s sessionfile] [-c new_startup_file] [-p new_prestart_file] [-C] [-P] [-H] Args: -H: header-less start -C: do not read startup file -P: do not read pre-startup file r)printsysexitrrrr_usage~sr0cCs|durtjjj}zt|}d|jvrZ|jdD]$}|durH|||j|||<q2n8t|jD]*\}}t|rf|dur|||||<qfWn"t yt j d|ddYn0dS)zLoads a Python module to make variables, objects and functions available globally. The idea is to load the module using importlib, then copy the symbols to the global symbol table. N__all__zLoading module %sT)exc_info) sixmovesbuiltins__dict__ importlib import_moduleappend iteritemsr,r#rerror)module globals_dict symb_listmodnamesymrrr_loads       rBcCstd|||ddS)z[Loads a Scapy module to make variables, objects and functions available globally. zscapy.modules.r=r>N)rBr@r=r>rrr load_modulesrEcCstdt||||ddS)zaLoads a Scapy layer module to make variables, objects and functions available globally. z scapy.layers.rCN)rB LAYER_ALIASESgetrDrrr load_layersrHc Csxz$td|td|||dWnNtyr}z6zt|||dWnty\|Yn0WYd}~n d}~00dS)zLoads a Scapy contrib module to make variables, objects and functions available globally. If no contrib module can be found with the given name, try to find a layer module, since a contrib module may become a layer module. scapy.contrib.rCN)r7r8rB ImportErrorrH)r@r=r>r&rrr load_contribs    rKFc Cs|durd}n"d|vr0d|vr0|ds0|d7}g}tjtjtd}tjdkrttj|d|}tj |d d }ntj||}t |}|D]P}| tjj d  d d }| drq|dr|dd}dd|d}tj|dd} | D]} | ddkrq| d} | dkrZ| d7} | d| } | | | } | | dd}||| <|ddkrnq|dr|dr||qqWdn1s0Y|r|ddkrq|dr|dstd|q|jddd |r|S|D]}td!|qdSdS)"zShow the list of all existing contribs. :param name: filter to search the contribs :param ret: whether the function should return a dict instead of printing it :returns: None or a dictionary containing the results if ret=True Nz*.py*?z.pycontrib)z**T) recursive.zcontrib.__) descriptionstatusr@replace)errorsr#rI=rXskiprWz(Module %s is missing its contrib infos !cSs|dS)Nr@r)xrrrzlist_contrib..)keyz1%(name)-20s: %(description)-40s status=%(status)s)endswithrrrdirname__file__r. version_infoglobiglobrYsep partition startswithiorfindstripr9rsortr-)r@ret_debugresultsZdir_pathiteratorfr?descfdlinepqrcvaluerrr list_contribs^           (r|cCsVd|vr i|d<|d<i|d<zddlm}|j|WntyPYn0dS)z)Updates IPython session with a custom oneZ_ohOutInr get_ipythonN)IPythonruser_nsupdater#)sessionrrrrupdate_ipython_session!s  rcCs"ddttddjDS)z"Load Scapy and return all builtinscSsi|]\}}t|r||qSr)r,).0r+vrrr 1sz#_scapy_builtins..z.allscapy)r3r:r7r8r6copyrrrr_scapy_builtins.s rrVc Csddlm}ddlm}m}|s<|j}|s<|jdd|_}td||szddl m }|j }Wn t yt jjjd}Yn0|std d S|d g}gd }|} t| D]} | | } | dd kr| | =qt| dr| jdr| | =qt| |r| | =q| |vs"| |vr*| | =qt| ttjfr| dd kr^td| t| | | | =qzt||dWntyYn0t !|d} t jj"#| | || $d S)a Save current Scapy session to the file specified in the fname arg. params: - fname: file to save the scapy session in - session: scapy session to use. If None, the console one will be used - pickleProto: pickle proto version (default: -1 = latest)r)utils)conf ConfClassT)ZkeepzSaving session into [%s]r scapy_sessionzNo session found ?!N _scpybuiltins)rr~r}r( __module__rz[%s] (%s) can't be saved.z.bakwb)%rr scapy.configrrrZ get_temp_filerinforrrr#r3r4r5r6r;rGrlisthasattrrrl isinstancetypetypes ModuleTyper"rrenamergziprcPickledumpclose) fnamerZ pickleProtorrrrignoreZ hard_ignoreZ to_be_savedr+irurrr save_session8sT            rc sddlm}|dur|j}ztjjt|d}Wn@t ytztjjt|d}Wnt ynYn0Yn0tjj j d| fdddD  |ttd |dS) zLoad current Scapy session from the file specified in the fname arg. This will erase any existing session. params: - fname: file to load the scapy session fromrrNrbrcsi|]}||qSrr)rr+rrrrrbz load_session..rzLoaded session [%s])rrrr3r4rloadrrr!r5r6rclearrrr)rrsrrr load_sessionqs     rcCsddlm}|dur|j}ztjjt|d}Wn&t yZtjjt|d}Yn0tjj j d}| |t |dS)zUpdate current Scapy session from the file specified in the fname arg. params: - fname: file to load the scapy session fromrrNrr)rrrr3r4rrrrr!r5r6rr)rrrrrrrupdate_sessions   rcCsddlm}i}|r8zt|WntyBtd|Yn0zRztjj t |d}Wn&t ytjj t |d}Yn0td|WnZtyd}t||Yn:tytd|Yntytd|Yn0|r(d |vr ||d ||_||d <n||_n||_d |i}nd |i}t}||||d <|tjjjd <|durtjjjd |t||r|SdS) NrrzNew session [%s]rzUsing existing session [%s]z5Error opening Python3 pickled session on Python2 [%s]zError opening session [%s]z-Error opening session [%s]. Attribute missingrrr)rrrrrrrr3r4rrrrr! ValueErrorr;EOFErrorAttributeError configurerrrkeysr5r6r) session_namemydictrqrSESSIONmsgZscapy_builtinsrrr init_sessionsV              rNcCs|d}|d8}g}g}dd}|rr|rF||t|dd|krX||dq"|dd|g}q"|r|dd|g}|dd|t|d |f|S) zhThis function processes a quote and returns a string that is ready to be used in the fancy prompt.  cSstdd|Dt|dS)Ncss|]}t|VqdSN)len)reltrrr rbz/_prepare_quote.._len..r^)sumr)rxrrr_lensz_prepare_quote.._lenrr^z | %sz | %s-- %srP)splitrr9popr)quoteauthormax_len_quotelinescur_linerrrr_prepare_quotes   "rc stdddlmd_t_|dur2|_t}t }d}|durLt j }zt |ddd}|dD]\}} |dkrt qj|d krd _d_tj_qj|d kr| }qj|d kr| }qj|d krd}qj|dkr| }qj|dkrd}qj|dkrjtdjd_qjt|ddkr,t dd|dWn<t jyj} z t| t dWYd} ~ n d} ~ 00t j ddt _ t||dd} |rt|dd|rt|ddjrƈjdvr zddl} ddlm} Wn:tyt dt!rt dt"_d }Yn0d}nd }jrddl#m$}|pBddk}gd}gd}ddd d!d"j%d d#d d$d g }|r|}d%d&|d'd(D}dg|dg}n*t&t'\}}|(t)||d)d*|*d d+d,d-t+j,j-fd.d-|Dfd/d-|Ddd0D}n d1j%}|dur6|d+7}||7}|r|d2| j.}zdd3l/m0}Wn^tyt d4z| d | d5|d6gd7Wn"t1yt2j3|| d8Yn0Yn0|}zdd9lm4}|stWn0ty t5|j6d:d |j6_7d|j6_8Yn0t9| j.dd;kr:d<j%|j6_:d |j;_|j?_@||jA_Bz| || d=Wn&tCtDfyt2j3|| d8Yn0nt2j3|| d8jErtFjE| dS)>z! Starts Scapy's console. alwaysrrTNr^z hs:Cc:Pp:d:Hz-hz-HFz-sz-cz-Cz-pz-Pz-d zToo many parameters : [%s]r)rrq)r%)ipythonauto) start_ipythonzaIPython not available. Using standard Python shell instead. AutoCompletion, History are disabled.z$On Windows, colors are also disabled)get_terminal_widthTK)& z& aSPY//YASa z& apyyyyCY//////////YCa z& sY//////YSpcs scpCY//Pp z& ayp ayyyyyyySCP//Pp syY//C z& AYAsAYYYYYYYY///Ps cY//Sz& pCCCCY//p cSSps y//Yz& SPPPP///a pP///AC//Yz& A//A cyP////Cz& p///Ac sC///az& P////YCpc A//Az& scccccp///pSP///p p//Yz& sY/////////y caa S//Pz& cayCyayP//Ya pY/Yaz& sY/PsY////YCc aC//Yp z& sc sccaCY//PCypaapyCP//YSs z& spCPY//////YPSps z& ccaacs r) z .SYPACCCSASYY zP /SCS/CCS ACSz /A ACz A/PS /SPPSz YP (SCz SPS/A. SCz Y/PACC PPz PY*AYC CAAz YYCY//SCYP rVz |z | Welcome to Scapyz | Version %sz$ | https://github.com/secdev/scapyz | Have fun!cSsg|]}|ddqS)rSNr)rr`rrr rbzinteract..rOr')r css|]\}}||VqdSrr)rlogobannerrrrrszinteract..c3s|]}j|VqdSr) color_themerrrxrrrrrbc3s|]}j|VqdSr)rsuccessrrrrrrb) fillvaluezWelcome to Scapy (%s)z using IPython %s )ConfigzFtraitlets not available. Some Scapy shell features won't be available.z print("""z"""))Zdisplay_bannerrZ exec_lines)rlocalr)shellrzScapy %s)configr)Gwarnings simplefilterrrr%rrZlogLevelDEFAULT_STARTUP_FILEDEFAULT_PRESTART_FILEr.argvgetoptr0Z fancy_promptverbloggingWARNINGmaxr GetoptErrorrrr;r/rr'Zinteractive_shelllowerrrrJr"r r Z scapy.utilsrversionrQUOTESextendrr9r3r4 zip_longest __version__Ztraitlets.config.loaderrr#codeinteractrr TerminalInteractiveShellZ confirm_exitZ separate_inintZterm_title_formatZ CompleterZuse_jediZ term_titleZhistfileZHistoryAccessorZ hist_fileZInteractiveShellZbanner1r TypeErrorrr)rrZmybannerZloglevelZ STARTUP_FILEZ PRESTART_FILEroptsoptparamrrrrZIPYTHONrZ mini_bannerZthe_logoZ the_logo_miniZ the_bannerrrZ banner_textrrcfgrrrrrs                      r__main__)NN)NN)NN)NN)NFF)rVNr)N)N)NF)r)=__doc__ __future__rrr.rrrrrhr7rmrrrrandomrZ scapy.errorrrrZscapy.libs.sixZlibsr3Z scapy.themesrr r Z scapy.constsr Z scapy.compatr r rrrrFrrglobalslocalsr'r,rrr0rBrErHrKr|rrrrrrrINFOr__name__rrrrsf        +   I  9   > T