:<>|dZddlddlddlZddlZej dZGddeZGdd eZ Gd d e Z Gd d e Z GddeZ GddeZ GddeZGddeZGddeZy)a<This module implements all state handling during uploads and downloads, the main interface to which being the TftpState base class. The concept is simple. Each context object represents a single upload or download, and the state object in the context object represents the current state of that transfer. The state object has a handle() method that expects the next packet in the transfer, and returns a state object until the transfer is complete, at which point it returns None. That is, unless there is a fatal error, in which case a TftpException is returned instead.)*Nztftpy.TftpStatescNeZdZdZdZdZdZdZdZd dZ d Z d Z d Z d Z y) TftpStatezThe base class for the states.c||_y)zConstructor for setting up common instance variables. The involved file object is required, since in tftp there's always a file involved.N)contextselfrs 5/usr/lib/python3.12/site-packages/tftpy/TftpStates.py__init__zTftpState.__init__s  ctd)z|An abstract method for handling a packet. It is expected to return a TftpState object, either itself or a new state.zAbstract method)NotImplementedErrorr pktraddressrports r handlezTftpState.handle"s""344r ct|jjdkDr|j|jjr~t j d|j|j_|jjD]4}t j d|d|jj|6yt jdtdtd)zRThis method handles an OACK from the server, syncing any accepted options.rz!Successful negotiation of optionsz z = Failed to negotiate optionszNo options found in OACKN) lenoptionskeys match_optionsrloginfoerror TftpException)r rkeys r handleOACKzTftpState.handleOACK's s{{! "Q &  !5!56<='*{{ $<<//CHHc4<<3G3G3LMN0 78#$ABB :; ;r ci}|D]}|dk(rt||tkDr&tjdtzt||<Ct||tkr&tjdtzt||<~||||<|dk(rtj dd|d<tjd|ztj d||S) zjThis method takes a requested options list from a client, and returns the ones that are supported.blksizez;Client requested blksize greater than %d setting to maximumz8Client requested blksize less than %d setting to minimumtsizeztsize option is setrz Dropping unsupported option '%s'z$Returning these accepted options: %s)int MAX_BLKSIZErr MIN_BLKSIZEdebug)r raccepted_optionsoptions r returnSupportedOptionsz TftpState.returnSupportedOptions7s F"wv'+5HH24?@A/:$V,)K7HH24?@A/:$V,/6v$V,7" /0,- );fDE#$ 8:JKr cd}|jj}tr3t|k(r*ddl}tj d|j dd}|jj}|jjj|}tj dt|t||kr.tjd|jjzd}t}||_||_|jj xj"t|jz c_tj d |j|jj$j'|j)j*|jj,|jj.f|jj0r|jj1|||j_|S) zThis method sends the next DAT packet based on the data in the context. It returns a boolean indicating whether the transfer is finished.FrNz#Deliberately delaying 10 seconds... zRead %d bytes into bufferzReached EOF on file %sTzSending DAT packet %d)r next_block DELAY_BLOCKtimerr'sleep getBlocksizefileobjreadrrfile_to_transfer TftpPacketDATdata blocknumbermetricsbytessocksendtoencodebufferhosttidport packethooklast_pkt)r finishedr7r/datr"r=s r sendDATzTftpState.sendDATRsqll-- ;+5  II; < JJrN,,++-%%**73 -s6{; v; HH-,,//0 1Ho% ""c#((m3" )3??;   !4!4"&,,"3"3T\\5I5I!J L << " " LL # #C ( # r Nctjd|||jj}tj d|zt }||_|jjj|jj|jj|jjf||j_ y)zThis method sends an ack packet to the block number specified. If none is specified, it defaults to the next_block property in the parent context.z$In sendACK, passed blocknumber is %sNzSending ack to block %d)rr'rr-r TftpPacketACKr7r:r;r<r=r>r?rA)r r7ackpkts r sendACKzTftpState.sendACKqs 8+F  ,,11K *[89(   !7!7"&,,"3"3"&,,"6"6"8 9!' r ctjd|t}||_|jj dk(rtjdnh|jj j|jj|jj|jj f||j_ y)zgThis method uses the socket passed, and uses the errorcode to compose and send an error packet.z*In sendError, being asked to send error %dNz1Error packet received outside session. Discarding) rr' TftpPacketERR errorcoderr?r:r;r<r=r>rA)r rKerrpkts r sendErrorzTftpState.sendErrors > J$ <<  4 ' III J LL   $ $V]]_%;%;&*ll&7&7&*ll&:&:&< =!' r ctjd|jjt }|jj|_|jj j |jj|jj|jjf||j_ y)zSThis method sends an OACK packet with the options from the current context.zIn sendOACK with options %sN) rr'rrTftpPacketOACKr:r;r<r=r>r?rA)r rs r sendOACKzTftpState.sendOACKs /1E1EFll**    !4!4"&,,"3"3"&,,"6"6"8 9!$ r c,tjd|jjd||jjxj t |jjjz c_|jjj|jj|jj}|s|jj}|jjj|jjjj|jj|f|jjr0|jj|jjyy)z-Resend the last sent packet due to a timeout.zResending packet z on sessions N)rwarningrrAr8 resent_bytesrr=add_dupr?portr:r;r<r>r@)r sendto_ports r resendLastzTftpState.resendLasts  ||$$d, - ))S1F1F1M1M-NN) $$T\\%:%:;ll** ,,++K   !6!6!=!=!?!F!F"&,,"3"3[!A C << " " LL # #DLL$9$9 : #r ctjd|jztjd|jj |j|jj k(r*tjd|j|j |jxj dz c_tjdt|j|jjj|j|jjxjt|jz c_ t|j|jjkrOtjdy|j|jj kr|jdk(r?tjd |jt j"t%d tjd |jz|jjj'|tjd |j|j |jnEd |j|jj fz}tj)|t%|t+|jS) zVThis method handles a DAT packet during a client download, or a server upload.zHandling DAT packet - block %dzExpecting block %sz#Good, received block %d in sequencerzWriting %d bytes to output filezEnd of file detectedNrzThere is no block zero!zDropping duplicate block %dz#ACKing block %d again, just in casez.Whoa! Received future block %d but expected %d)rrr7r'rr-rHrr6r2writer8r9r1rRrM TftpErrors IllegalTftpOprrTrTftpStateExpectDAT)r rmsgs r handleDatzTftpState.handleDats 1COOCD & (?(?@ ??dll55 5 II;S__ M LLN LL # #q ( # II7SXX G LL & &sxx 0 LL & &#chh- 7 &388}t||88::/0 __t||66 6!# 56z778#$=>> KK5G H LL ( ( - II;S__ M LL )C??DLL$;$;<=C IIcN$ $"$,,//r N)__name__ __module__ __qualname____doc__r rr r*rDrHrMrPrWr^r r rrs7( 5 <  6> ' ' $;"%0r rceZdZdZdZdZy)TftpServerStatez!The base class for server states.c>tj||d|_yr_)rr full_pathr s r r zTftpServerState.__init__s4)r c z|j}d}|jjs)||j_tj d|ztj ddt i|j_|rNtj d||j|}|jjj|d}|jdk7rtjd|jj|k7s|jj|k7rk|jtjtj!d |jjd |jjd |d |d |Stj d |j"|j"j%|jj&r |j"}nMt(j*j-|jj&|j"j/d}t(j*j1||_tj d||j2j%t(j*j5|jj&t(j6zrtj dn?tjd|jtj8t;d|j"|j_|S)aSThis method performs initial setup for a server context transfer, put here to refactor code out of the TftpStateServerRecvRRQ and TftpStateServerRecvWRQ classes, since their initial setup is identical. The method returns a boolean, sendoack, to indicate whether it is required to send an OACK to the client.FzSetting tidport to %sz Setting default options, blksizer"zOptions requested: %sToctetz=Received non-octet mode request. I'll reply with binary data.zExpected traffic from :z but received it from z instead.zRequested filename is %s/zfull_path is %sz+requested file is in the server root - goodz2requested file is not within the server root - badz bad file path)rrr?rrr' DEF_BLKSIZEr*updatemoderRr>rUrMrZ UnknownTIDrfilename startswithrootospathjoinlstripabspathrhnormpathsepr[rr4)r rrrrsendoacksupported_optionsrhs r serverInitialzTftpServerState.serverInitials4 ++||###(DLL HH,u4 5 45!*K9   II-w 7 $ ; ;G D  LL ' '(9 :H 88w  KKW X <<   (DLL,=,=,F NN:00 1 II#||00#||00'$ & 'K ,cll; << " "4<<#4#4 5 I T\\%6%6 8K8KC8PQI 3 #Y/ >> $ $RWW%5%5dll6G6G%H266%Q R HHB C KKL M NN:33 40 0(+  %r N)r`rarbrcr r}rdr r rfrfs+Hr rfceZdZdZdZy)TftpStateServerRecvRRQz_This class represents the state of the TFTP server when it has just received an RRQ packet.ctjd|j|||}|j}tj d|zt j j|rt|d|j_ n|jjrtjd||jj|jj|||j_ |jjtjd|jtjt!d|ztj#d||jtjt!dj%||rd |jj&vr|jjj)d t j*t-|jjj/}|jjj)d d ||jj&d <|r|j1nEd |j_tjd |j5|j_t9|jS) z)Handle an initial RRQ packet as a server.z In TftpStateServerRecvRRQ.handlezOpening file %s for readingrbz'No such file %s but using dyn_file_func)rrz7dyn_file_func returned 'None', treating as FileNotFoundzFile not found: %szFile not found: {}r#rrz&No requested options, starting send...)rr'r}rhrrtruexistsopenrr2 dyn_file_funcr4rMrZ FileNotFoundrrRformatrseekSEEK_ENDstrtellrPr-rDpending_completeTftpStateExpectACK)r rrrr{rur#s r rzTftpStateServerRecvRRQ.handle)s 45%%c8U;~~ .56 77>>$ $(d#3DLL \\ ' ' II? F **4<<+H+HS[ch*i LL ||##+ )*z667#$84$?@@ KK,d 3 NN:22 3 4 ; ;D AB B 4<<#7#77 LL % %a 5 ,,1134E LL % %a +,1DLL  )  MMO'(DLL # II> ?,0LLNDLL )"$,,//r Nr`rarbrcrrdr r rr&s /0r rceZdZdZdZdZy)TftpStateServerRecvWRQz^This class represents the state of the TFTP server when it has just received a WRQ packet.c|jt|jjd}tj d||j tjdd}tj d||jj}|D]q}|stjj||}tjj|rtj d|\tj|dsy)zThe purpose of this method is to, if necessary, create all of the subdirectories leading up to the file to the written.Nzmake_subdirs: subpath is %sz dirs is %sz#%s is already an existing directoryi) rhrrrsrr'splitrtrzrurvisdirmkdir)r subpathdirscurrentdirs r make_subdirsz#TftpStateServerRecvWRQ.make_subdirs`s..T\\%6%6!7!89 /9}}RVV$Sb) ,%,,##C'',,w477==)IICWMHHWe, r cftjd|j|||}|j}|jj rg|jj ||j}|-|j tjtd|z||j_ ntjd|ztjj|r,tjd|jj z|j#t%|d|j_ |r&tjd|j'n%tjd|j)d|j_t-|jS) z)Handle an initial WRQ packet as a server.z In TftpStateServerRecvWRQ.handlezDynamic path %s not permittedzOpening file %s for writingz&File %s exists already, overwriting...wbzSending OACK to clientz4No requested options, expecting transfer to begin...r)rr'r}rhr upload_openrMrZAccessViolationrr2rrtrurrRr4rrrPrHr-r\)r rrrr{rufs r rzTftpStateServerRecvWRQ.handless; 45%%c8U;~~ << # # ((t||>s?G?DF F] + II- .)$,,7>>s?G?DF F NN:33 4 IC OP Pr Nrrdr r rrsHQr rceZdZdZdZy)rzThis class represents the state of the transfer when a DAT was just sent, and we are waiting for an ACK from the server. This class is the same one used by the client during the upload, and the server during the download.c2t|trtjd|jz|j j |jk(r|j jrtjdytjd|j xj dz c_tjd|j j |j|j _|S|j|j j krItjd|jz|j jj||Stjd|j jxjdz c_ |St|tr8tjd t!|zt#d t!|ztjd t!|z|S) z;Handle a packet, hopefully an ACK since we just sent a DAT.zReceived ACK for packet %dz&Received ACK to final DAT, we're done.NzGood ACK, sending next DATrzIncremented next_block to %dz#Received duplicate ACK for block %dzGOooh, time warp. Received ACK to packet we didn't send yet. Discarding.z!Received ERR packet from peer: %sz!Discarding unsupported packet: %s)rrFrr'r7rr-rrrDrRr8rTerrorsrJrrrrs r rzTftpStateExpectACK.handles c= ) II2S__D E||&&#//9<<00HHEFII:;LL++q0+II< //148LLNDLL1K4<<#:#:: Aoo&' $$,,S1 K 89 $$++q0+K ] + II9CHD E Cc#h NO O KK;c#hF GKr Nrrdr r rrs r rceZdZdZdZy)r\z)Just sent an ACK packet. Waiting for DAT.c$t|tr|j|St|tr*|j t j tdt|tr*|j t j tdt|tr6|j t j tdt|z|j t j tdt|z)z?Handle the packet in response to an ACK, which should be a DAT.z)Received ACK from peer when expecting DATz)Received WRQ from peer when expecting DATzReceived ERR from peer: z(Received unknown packet type from peer: ) rr5r^rFrMrZr[rrrJrrs r rzTftpStateExpectDAT.handles c= )>>#& &] + NN:33 4 KL L ] + NN:33 4 KL L ] + NN:33 4 :SX EF F NN:33 4 JSQTX UV Vr Nrrdr r r\r\s 3Wr r\ceZdZdZdZy)TftpStateSentWRQz&Just sent an WRQ packet for an upload.c|jjs'||j_tjd|t |t rtj d |j|tjd|j|j_ tjdt|jSt |t rtj dtjd|j"dk(rstjd tjd|j|j_ tjdt|jStj%d |j"ztjd |St |t&r-|jtj(td |zt |t*r*|jtj(td t |t,r*|jtj(td|jtj(td|z#t$r6tjd|jtjwxYw)r!Set remote port for session to %sReceived OACK from serverzSending first DAT packetz$Changing state to TftpStateExpectACKrzReceived ACK from serverz)Apparently the server ignored our optionsrz#Ack blocknumber is zero as expectedzDiscarding ACK to block %sz,Still waiting for valid response from serverzReceived ERR from server: %sz(Received RRQ from server while in uploadz(Received DAT from server while in upload,Received unknown packet type from server: %s)rr?rr'rrOrr rDrrrrrMrZFailedNegotiationrFr7rRrJr[rr5rs r rzTftpStateSentWRQ.handles||###(DLL II95 A c> * HH0 1 8$  4504  - @A)$,,77 ] + HH/ 0 IIA B!# ?@ 4504  - @A)$,,77 83??JK HI ] + NN:33 4 > DE E ] + NN:33 4 JK K ] + NN:33 4 JK K NN:33 4 NQT TU UO!  78z;;< s $J&&?K%Nrrdr r rrs 06r rceZdZdZdZy)TftpStateSentRRQzJust sent an RRQ packet.ct|jjs)||j_tjd|zt |t rxtjd |j |tjd|jdtjdt|jSt |t rhtjd |jj"r,tjd d t$i|j_|j'|St |t(r*|jtj*td t |t,r*|jtj*td t |t.rw|jtj*tjd||j0tj2k(r t5dtdj7||jtj*td|z#t$rF}tjdt|z|jtjd}~wwxYw)z6Handle the packet in response to an RRQ to the server.rrzSending ACK to OACKr)r7z$Changing state to TftpStateExpectDATzFailed to negotiate options: %sNzReceived DAT from serverz0Server ignored options, falling back to defaultsr"z*Received ACK from server while in downloadz*Received WRQ from server while in downloadzReceived ERR packet: %szFile not foundzReceived ERR from server: {}r)rr?rrrrOr r'rHr\rrrrMrZrr5rrmr^rFr[rrJrKrTftpFileNotFoundErrorr)r rrrerrs r rzTftpStateSentRRQ.handle,s||###(DLL HH85@ A c> * HH0 1 8$  /0  + @A)$,,77 ] + HH/ 0||##KL)2K'A $>>#& &] + NN:33 4 LM M ] + NN:33 4 LM M ] + NN:33 4 II/ 5}} 7 77+,<==#$B$I$I#$NOO NN:33 4 NQT TU US!  ;c#hFGz;;< s&I(( J71AJ22J7Nrrdr r rr*s "7r r)rc TftpSharedTftpPacketTypesrtlogging getLoggerrobjectrrfrrrrr\rrrdr r rs= g*+ v0v0pRiRj20_20n:0_:0~Q9Q*$$LWW28y8t9y9r