a k#dtb@s(ddlmZddlZddlZddlZddlmmZddlm Z m Z ddl m Z m Z mZddlmZddlmZmZddlmZddlmZmZdd lmZmZmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%Gd d d e e&Z'Gd d d e$Z(e#dddZ)e#dddZ*e+e(GdddZ,Gddde,e eZ-Gddde,Z.Gddde,Z/Gddde/Z0Gddde-Z1Gddde1Z2Gdd d e/Z3Gd!d"d"e/Z4Gd#d$d$e1Z5Gd%d&d&e5Z6Gd'd(d(e2Z7Gd)d*d*e/Z8Gd+d,d,e/Z9Gd-d.d.e.Z:Gd/d0d0e.Z;Gd1d2d2e.Z|dr2|ddjvr2fdd}|St|dS)NZspawn_cs&j}||i|}||SN)radd)argskargskpdnamer4r&r'fMs  z!PipeEngine.__getattr__..f) startswithrAttributeError)r4attrr>r&r<r' __getattr__Hs    zPipeEngine.__getattr__cCs|Sr6)recvr4r&r&r' _read_cmdVszPipeEngine._read_cmdcCs||dSr6)sendr4_cmdr&r&r' _write_cmdZszPipeEngine._write_cmdcCsR|j|t|tr"|j|t|tr8|j|t|trN|j|dSr6) r,r7 isinstanceSourcer-Drainr.Sinkr/)r4piper&r&r' add_one_pipe^s      zPipeEngine.add_one_pipecs fddt}|||S)Ncs>|||j|jB|jB|jBD]}||vr"||q"dSr6)r7sourcessinks high_sources high_sinks)r;Zliqflattenr&r'rVjs z)PipeEngine.get_pipe_list..flatten)r+)r4rNplr&rUr' get_pipe_lisths  zPipeEngine.get_pipe_listcGs@t}|D]}|||O}q ||j8}|D]}||q,|Sr6)r+rXr,rO)r4rrWr;rTr&r&r'r0vs  zPipeEngine._add_pipesc Cstdzz(|jD] }|q|j}||tg}d}d}|r6|r^t|dkr6t|d}|D]}||ur| }|dkrd}qFn6|dkrd}n(|dkr|j|}||n t d |ql||vrlz | Wn6t y} zt d |j| WYd} ~ qld} ~ 00|rl||||qlqFWntyLYn0Wz.|jD]}|qXW|jtd n|jtd 0nHz.|jD]}|qW|jtd n|jtd 00dS) NzPipe engine thread started.TFg?XBAz3Unknown internal pipe engine command: %r. Ignoring.zpiping from %s failed: %szPipe engine thread stopped.)r debugr,startr-r7r+lenrrEr deliver Exception exceptionname exhaustedremoveKeyboardInterruptstopr1release) r4r;rPrdRUNZSTOP_IF_EXHAUSTEDfdsfdcmder&r&r'runsf               zPipeEngine.runcCs>|jdr0t|jdd}d|_|||_n tddS)NFzscapy.pipetool.PipeEnginetargetrcTzPipe engine already running) r1acquirerrndaemonr^r3r r])r4Z_tr&r&r'r^s  zPipeEngine.startcCs|jdddS)Nr[)rH)rgrDr&r&r' wait_and_stopszPipeEngine.wait_and_stoprZc Cszv|j\|jdurL|||jz|jWqVtyHYqV0n tdWdn1sj0YWnt yt dYn0dS)NzPipe engine thread not runningzInterrupted by user.) r2r3rIjoinr1rhrar r]rfr!rGr&r&r'rgs    , zPipeEngine.stopcGsZ|j|}|j6|jdur8|D] }|q |dWdn1sL0YdS)Nr\)r0r2r3r^rI)r4r_pipesr;r&r&r'r7s    zPipeEngine.addcKs&ddg}|jD]}|dt||jfq|d|d|jD]*}|jD]}|dt|t|fqPqF|d|d|jD]*}|jD]}|dt|t|fqq|d|d|jD]*}|jD]}|dt|t|fqq|d d |}t|fi|dS) Nzdigraph "pipe" {z node [shape=rectangle];z "%i" [label="%s"];rz" edge [color=blue, arrowhead=vee];z "%i" -> "%i";z' edge [color=purple, arrowhead=veevee];z% edge [color=red, arrowhead=diamond];} ) r,appendidrcrQrS trigger_sinksrtr )r4r9gr;shstsgraphr&r&r'rs*               zPipeEngine.graphN)rZ)__name__ __module__ __qualname__r classmethodr(r)r*rBrErIrOrXr0rnr^rsrgr7rr&r&r&r'r's$     /   rcseZdZfddZZS) _PipeMetacs*tdtt|||||}|tj|<|S)Nz Type[Pipe])rsuperr__new__rr)r"rcbasesdctc __class__r&r'rs  z_PipeMeta.__new__)rrrr __classcell__r&r&rr'rsr_SrM)bound_TS TriggerSinkc@speZdZdddZddZddZddd Zd d Zd d ZddZ ddZ ddZ ddZ ddZ ddZdS)PipeNcCsNt|_t|_t|_t|_t|_t|_|durDd|jj}||_ dS)Nz%s) r+rPrQrRrStrigger_sourcesrzrrrcr4rcr&r&r'r*s z Pipe.__init__cCs|jD]}||qdSr6)rQpushr4msgr|r&r&r'_sends z Pipe._sendcCs|jD]}||qdSr6)rS high_pushrr&r&r' _high_sends zPipe._high_sendcCs|jD]}||qdSr6)rz on_triggerrr&r&r'_triggers z Pipe._triggercCs|j||j||Sr6rQr7rPr4otherr&r&r'__gt__s  z Pipe.__gt__cCs|j||j||Sr6)rSr7rRrr&r&r' __rshift__#s  zPipe.__rshift__cCs|j||j||Sr6)rzr7rrr&r&r'__xor__)s  z Pipe.__xor__cCs t|Sr6)object__hash__rDr&r&r'r/sz Pipe.__hash__cCs t||Sr6)r__eq__rr&r&r'r3sz Pipe.__eq__cs>tjdd|jf}|js,|jr|dd7}|jrv|ddfdd|jDdf7}|d 7}|jr|dddfd d|jDf7}|d 7}|j s|j rz|dd7}|j r$|ddfd d|j Dd f7}|d 7}|j rl|dd dfdd|j Df7}|d 7}|j s|j r,|dd7}|j r|ddfdd|j Ddf7}|d 7}|j r|dddfdd|j Df7}|d 7}|d7}|S)Nz%s%s>z Pipe.__repr__..>#c3s|]}|jVqdSr6rrrr&r'rCr]c3s|]}|jVqdSr6rrrr&r'rIr>>c3s|]}|jVqdSr6rrrr&r'rNrc3s|]}|jVqdSr6rrrr&r'rTr^c3s|]}|jVqdSr6rrrr&r'rYr) r Z color_themepunctZ layer_namercrPrQrtZ field_valuerRrSrrz)r4r|r&rr'__repr__7sP " " " z Pipe.__repr__cCsdSr6r&rDr&r&r'r^_sz Pipe.startcCsdSr6r&rDr&r&r'rgcsz Pipe.stop)N)N)rrrr*rrrrrrrrrr^rgr&r&r&r'rs  (rc@s.eZdZd ddZddZddZdd ZdS) rKNcCs$tj||dt||d|_dSNrcF)rr*r is_exhaustedrr&r&r'r*is zSource.__init__cCstSr6)rrDr&r&r' _read_messageoszSource._read_messagecCs|j}||dSr6)rrr4rr&r&r'r`sszSource.delivercCs|jSr6rrDr&r&r'rdxszSource.exhausted)N)rrrr*rr`rdr&r&r&r'rKhs rKc@s eZdZdZddZddZdS)rLzRepeat messages from low/high entries to (resp.) low/high exits .. code:: +-------+ >>-|-------|->> | | >-|-------|-> +-------+ cCs||dSr6rrr&r&r'rsz Drain.pushcCs||dSr6rrr&r&r'rszDrain.high_pushNrrrrrrr&r&r&r'rL}s rLc@s@eZdZdZddZddZddZdd Zd d Zd d Z dS)rMz Does nothing; interface to extend for custom sinks. All sinks have the following constructor parameters: :param name: a human-readable name for the element :type name: str cCsdS)z Called by :py:class:`PipeEngine` when there is a new message for the low entry. :param msg: The message data :returns: None :rtype: None Nr&rr&r&r'rs z Sink.pushcCsdS)z Called by :py:class:`PipeEngine` when there is a new message for the high entry. :param msg: The message data :returns: None :rtype: None Nr&rr&r&r'rs zSink.high_pushcCs|j||j||Sr6rrr&r&r'__lt__s  z Sink.__lt__cCs|j||j||Sr6)rRr7rSrr&r&r' __lshift__s  zSink.__lshift__cCs||?||?|Sr6r&rr&r&r' __floordiv__szSink.__floordiv__cCs||k||k|Sr6r&rr&r&r'__mod__sz Sink.__mod__N) rrrrrrrrrrr&r&r&r'rMs  c@seZdZddZdS)rcCsdSr6r&rr&r&r'rszTriggerSink.on_triggerN)rrrrr&r&r&r'rsc@s6eZdZd ddZddZddZdd Zd d ZdS) AutoSourceNcCstj||ddSNr)rKr*rr&r&r'r*szAutoSource.__init__cCst||ddfdSNFrrFrr&r&r' _gen_dataszAutoSource._gen_datacCst||ddfdSNTFrrr&r&r'_gen_high_dataszAutoSource._gen_high_datacCst|ddS)N)NNTrrDr&r&r'_exhaustszAutoSource._exhaustcCs0|\}}}|r|r"||n ||dSr6)rCrr)r4rhighZexhaustr&r&r'r`s  zAutoSource.deliver)N)rrrr*rrrr`r&r&r&r'rs  rc@s.eZdZd ddZddZddZdd ZdS) ThreadGenSourceNcCstj||dd|_dSr)rr*rirr&r&r'r*szThreadGenSource.__init__cCsdSr6r&rDr&r&r'generateszThreadGenSource.generatecCsd|_t|jdddS)NTzscapy.pipetool.ThreadGenSourcero)rirrr^rDr&r&r'r^szThreadGenSource.startcCs d|_dSr)rirDr&r&r'rgszThreadGenSource.stop)N)rrrr*rr^rgr&r&r&r'rs rc@s eZdZdZddZddZdS) ConsoleSinkzPrint messages on low and high entries to ``stdout`` .. code:: +-------+ >>-|--. |->> | print | >-|--' |-> +-------+ cCstdt|dS)Nrr!reprrr&r&r'rszConsoleSink.pushcCstdt|dS)Nrrrr&r&r'rszConsoleSink.high_pushNrr&r&r&r'rs rc@s*eZdZdZd ddZddZdd ZdS) RawConsoleSinka>Print messages on low and high entries, using os.write .. code:: +-------+ >>-|--. |->> | write | >-|--' |-> +-------+ :param newlines: Include a new-line character after printing each packet. Defaults to True. :type newlines: bool NTcCstj||d||_d|_dS)NrrY)rMr*newlines _write_pipe)r4rcrr&r&r'r*(szRawConsoleSink.__init__cCs&|jr|d7}t|j|ddSNrwutf8roswriterencoderr&r&r'r.szRawConsoleSink.pushcCs&|jr|d7}t|j|ddSrrrr&r&r'r4szRawConsoleSink.high_push)NTrrrrr*rrr&r&r&r'rs rc@s eZdZdZddZddZdS) CLIFeederzSend messages from python command line: .. code:: +--------+ >>-| |->> | send() | >-| `----|-> +--------+ cCs||dSNrY)rrr&r&r'rFGs zCLIFeeder.sendcCs d|_dS)NTrrDr&r&r'closeLszCLIFeeder.closeN)rrrrrFrr&r&r&r'r;s rc@seZdZdZddZdS) CLIHighFeederzSend messages from python command line to high output: .. code:: +--------+ >>-| .----|->> | send() | >-| |-> +--------+ cCs||dSr)rrr&r&r'rF]s zCLIHighFeeder.sendN)rrrrrFr&r&r&r'rQs rc@s"eZdZdZdddZddZdS) PeriodicSourcezGenerage messages periodically on low exit: .. code:: +-------+ >>-| |->> | msg,T | >-| `----|-> +-------+ rNcCs>tj||dt|tttfs(|g|_n||_||_||_dSr) rr*rJlistr+tuplerperiodperiod2)r4rrrrcr&r&r'r*os  zPeriodicSource.__init__cCsT|jrPd}|jD]}d}||t|jq|rBd|_|t|jqdSr) rirrtimesleeprrrr)r4Z empty_genmr&r&r'rys  zPeriodicSource.generate)rN)rrrrr*rr&r&r&r'rcs rc@sheZdZdZdddZer,ddZdd Znd d Zd d Z ddZ ddZ ddZ ddZ ddZdS)TermSinka Prints messages on the low and high entries, on a separate terminal (xterm or cmd). .. code:: +-------+ >>-|--. |->> | print | >-|--' |-> +-------+ :param keepterm: Leave the terminal window open after :py:meth:`~Pipe.stop` is called. Defaults to True. :type keepterm: bool :param newlines: Include a new-line character after printing each packet. Defaults to True. :type newlines: bool :param openearly: Automatically starts the terminal when the constructor is called, rather than waiting for :py:meth:`~Pipe.start`. Defaults to True. :type openearly: bool NTcCs8tj||d||_||_||_d|_|jr4|dSr)rMr*keeptermr openearlyopenedr^)r4rcrrrr&r&r'r*szTermSink.__init__cCs|jsd|_t|_t|jd|jdur2dn|j|_d|j|jddf}tjt t j d|gtj d}| \}}t||_td |jdS) NTaZScapyz$app = Start-Process PowerShell -ArgumentList '-command &{$host.ui.RawUI.WindowTitle=\"%s\";Get-Content \"%s\" -wait}' -passthru; echo $app.Id\z\\ powershell)stdoutzPID: %d)rr _TermSink__fopenrrcreplace subprocessPopengetattrr progPIPE communicateintpidr!)r4rlprocoutput_r&r&r'_start_windowss   zTermSink._start_windowscCsN|jsJd|_ddl}d}|jj|d|j}|jj|d|jj|dS)NFrrY) rrctypesZwindllZkernel32Z OpenProcessrZTerminateProcessZ CloseHandle)r4rZPROCESS_TERMINATEhandler&r&r' _stop_windowsszTermSink._stop_windowscCsz|jsvd|_t\}|_dg}|jdur:|d|jg|jrJ|d|dd|gtj |dd|_ t |dS) NTZxtermz-titlez-holdz-ezcat <&%dF) close_fds) rrrNwdescrcextendrrxrrrr)r4Zrdescrlr&r&r' _start_unixs  zTermSink._start_unixcCs$|js d|_|j|jdSr)rrrkillwaitrDr&r&r' _stop_unixs zTermSink._stop_unixcCstr |S|SdSr6)rrrrDr&r&r'r^szTermSink.startcCstr |S|SdSr6)rrr rDr&r&r'rgsz TermSink.stopcCsH|jr|d7}tr2t|jd}|||nt|j|dS)Nrwr) rrrrrrrrr)r4r|rr&r&r'_prints   zTermSink._printcCs|t|dSr6r strrr&r&r'rsz TermSink.pushcCs|t|dSr6r rr&r&r'rszTermSink.high_push)NTTT)rrrrr*rrrrr r^rgr rrr&r&r&r'rs   rc@s4eZdZdZd ddZddZddZd d d ZdS) QueueSinka5 Collects messages on the low and high entries into a :py:class:`Queue`. Messages are dequeued with :py:meth:`recv`. Both high and low entries share the same :py:class:`Queue`. .. code:: +-------+ >>-|--. |->> | queue | >-|--' |-> +-------+ NcCs tj||dtjj|_dSr)rMr*sixmovesqueueQueuerTrr&r&r'r*szQueueSink.__init__cCs|j|dSr6rTputrr&r&r'rszQueueSink.pushcCs|j|dSr6rrr&r&r'rszQueueSink.high_pushTcCs2z|jj||dWStjjjy,YdS0dS)a^ Reads the next message from the queue. If no message is available in the queue, returns None. :param block: Blocks execution until a packet is available in the queue. Defaults to True. :type block: bool :param timeout: Controls how long to wait if ``block=True``. If None (the default), this method will wait forever. If a non-negative number, this is a number of seconds to wait before giving up (and returning None). :type timeout: None, int or float )blocktimeoutN)rTgetrrrEmpty)r4rrr&r&r'rC szQueueSink.recv)N)TN)rrrrr*rrrCr&r&r&r'r s  r c@s*eZdZdZd ddZddZddZdS) TransformDrainzApply a function to messages on low and high entry: .. code:: +-------+ >>-|--[f]--|->> | | >-|--[f]--|-> +-------+ NcCstj||d||_dSr)rLr*r>)r4r>rcr&r&r'r*BszTransformDrain.__init__cCs|||dSr6)rr>rr&r&r'rGszTransformDrain.pushcCs|||dSr6)rr>rr&r&r'rKszTransformDrain.high_push)Nrr&r&r&r'r6s rc@s eZdZdZddZddZdS)UpDrainzRepeat messages from low entry to high exit: .. code:: +-------+ >>-| ,--|->> | / | >-|--' |-> +-------+ cCs||dSr6rrr&r&r'r\sz UpDrain.pushcCsdSr6r&rr&r&r'r`szUpDrain.high_pushNrr&r&r&r'rPs rc@s eZdZdZddZddZdS) DownDrainzRepeat messages from high entry to low exit: .. code:: +-------+ >>-|--. |->> | \ | >-| `--|-> +-------+ cCsdSr6r&rr&r&r'rqszDownDrain.pushcCs||dSr6rrr&r&r'ruszDownDrain.high_pushNrr&r&r&r'res r)= __future__rrrrZscapy.libs.sixZlibsr threadingrrZscapy.automatonrrrZ scapy.constsrZ scapy.errorr r Z scapy.configr Z scapy.utilsr r Z scapy.compatrrrrrrrrrrrrr rrrr add_metaclassrrKrLrMrrrrrrrrrr rrrr&r&r&r'sD   8I  g:#$}2