RSiFddlZddlZddlZddlZddlZddlmZddlmZddlmZddl m Z ddl m Z ddl m Z ddl m Z dd lmZdd lmZdd lmZd d gZGdd e ZGdd eZdS)N)_FileReadWriteError)_GiveUpOnSendfile) _RetryError)_ERRNOS_DISCONNECTED) _ERRNOS_RETRY) AsyncChat)timer)debug)logger)strerror DTPHandlerThrottledDTPHandlerceZdZdZdZdZdZdZdZeZ dZ dZ dZ d Z d Zd Zd Zd ZdZdZdZdZdZdZeZdZdZdZdZdZdZdS)r a*Class handling server-data-transfer-process (server-DTP, see RFC-959) managing data-transfer operations involving sending and receiving data. Class attributes: - (int) timeout: the timeout which roughly is the maximum time we permit data transfers to stall for with no progress. If the timeout triggers, the remote client will be kicked off (defaults 300). - (int) ac_in_buffer_size: incoming data buffer size (defaults 65536) - (int) ac_out_buffer_size: outgoing data buffer size (defaults 65536) i,ic(||_d|_d|_d|_d|_d|_d|_|j|_|j|_d|_ d|_ d|_ t|_ d|_d|_d|_d|_d|_ t'j|||jn#t,$r{}t'j|t/j|j||jt2jkrYd}~dS|Yd}~dSd}~wwxYw|js|dS|jr3|j|j|j|j|_dSdS)zInitialize the command channel. - (instance) sock: the socket object instance of the newly established connection. - (instance) cmd_channel: the command channel class instance. NFr)ioloop_errback) cmd_channelfile_objreceivetransfer_finishedtot_bytes_senttot_bytes_receivedcmdlog log_exception _data_wrapper _lastdata_had_crr _start_time_resp_offset_filefd_idler _initializedr__init__rOSErrorsocketcloseerrnoEINVAL handle_error connectedtimeout call_everyhandle_timeout)selfsockrerrs O/home/jrussi/.local/lib/python3.11/site-packages/pyftpdlib/handlers/ftp/data.pyr'zDTPHandler.__init__/s'  !&"#?(6!  77    !   tT+2D E E E E E     fmook.@     JJLLLyEL((       FFFFF ~  JJLLL F < +00 d1D) __class____name__r get_repr_infor2s r5__repr__zDTPHandler.__repr__as9 N # # #   * *$ * 7 7 7 7  c|jjsdS|jt|jdsdS |jn#t t f$rYdSwxYw|jjdkrdSdS)NFfilenoiT)r use_sendfilerhasattrrAr( ValueError _current_typer=s r5rCzDTPHandler.use_sendfileis, 5 =  x(H(H 5  M " " " "$   55    )S 0 05tsAAAcd|_||jj|jj|_t j||dS)NT)r&modify_ioloop_eventsrWRITE_wanted_io_eventsrpush)r2datas r5rKzDTPHandler.push}sH  !!$+"3444!%!2tT"""""r?cd|_||jj|jj|_|ro|j|_|j |_ | |j |_ dS#t$rYnwxYwtd|t!j||dS)NTzstarting transfer using send())r&rHrrIrJrCfiletellr#rrAr$initiate_sendfile initiate_sendrr rpush_with_producer)r2producers r5rRzDTPHandler.push_with_producers  !!$+"3444!%!2      #=--//DL=//11DL &&(((&*%;" %     .555$T844444s B++ B87B8cDtj|dSN)asynchat async_chatclose_when_doner=s r5rXzDTPHandler.close_when_dones++D11111r?cDtj|dSrU)rVrWrQr=s r5rQzDTPHandler.initiate_sends))$/////r?c tj|j|j|j|j}|dkr*||dS|xj|z c_|xj|z c_dS#t$r}|j tvs|j tj krYd}~dS|j tvr|n(|jdkrtjdt |Yd}~dSd}~wwxYw)zA wrapper around sendfile.rNz3sendfile() failed; falling back on using plain send)ossendfile_filenor$r#ac_out_buffer_sizediscard_buffers handle_closerr(r+rEBUSYrr warningr)r2sentr4s r5rPzDTPHandler.initiate_sendfiles> ,;   ' D&qyy$$&&&!!##### $ ##t+####'   yM))SY%+-E-E222!!####$))I(S0$##### s+A?? D #D2A DDc|jrd|z}|drd|_|dd}nd|_|dttjdS)zThe data wrapper used for receiving data in ASCII mode on systems using a single line terminator, handling those cases where CRLF (' ') gets delivered in two chunks.  TNFs ascii)r endswithreplacebytesr[linesep)r2chunks r5_posix_ascii_data_wrapperz$DTPHandler._posix_ascii_data_wrappersh < "EME >>%  !DL#2#JEE DL}}WeBJ&@&@AAAr?c d|_||jj|jj|_||_|dkr%t jdkrd|_n*|j |_n|dkrd|_ntdd|_ dS)zEnable receiving of data over the channel. Depending on the TYPE currently in use it creates an appropriate wrapper for the incoming data. - (str) type: current transfer type, 'a' (ASCII) or 'i' (binary). Taz NrBzunsupported type) r&rHrREADrJrr[rkrrm TypeErrorr)r2typers r5enable_receivingzDTPHandler.enable_receivings! !!$+"2333!%!1 3;;zV##%)""%)%C"" S[[!%D  .// / r?c |j|jzS)z'Return the number of transmitted bytes.)rrr=s r5get_transmitted_bytesz DTPHandler.get_transmitted_bytess"T%<<>/B10B1cr|js|jrd|_nt|jdk|_ |jrdt jf|_n8|}dt|dt jf|_| dS#| wxYwdS)z!Called when the socket is closed.Trz226 Transfer complete.z426 Transfer aborted; z bytes transmitted.N) _closedrrr~rr r r"ruintr*)r2 tot_bytess r5r`zDTPHandler.handle_closeWs| | F)-&&),T-?)@)@A)E& ) ":FL!IDJJ $ : : < __str__rCrKrRrXrQrPrmrsrurwryr{rrhandle_read_eventrrr1r-r`r*rr?r5r r s G000d   G(### 555"222000,,,@BBB ,===***111  :888*$   GGG / / /444&8%-%-%-%-%-r?cdeZdZdZdZdZdZfdZdZdZ fdZ fdZ d Z d Z fd ZxZS) raA DTPHandler subclass which wraps sending and receiving in a data counter and temporarily "sleeps" the channel so that you burst to no more than x Kb/sec average. - (int) read_limit: the maximum number of bytes to read (receive) in one second (defaults to 0 == no limit). - (int) write_limit: the maximum number of bytes to write (send) in one second (defaults to 0 == no limit). - (bool) auto_sized_buffers: this option only applies when read and/or write limits are specified. When enabled it bumps down the data buffer sizes so that they are never greater than read and write limits which results in a less bursty and smoother throughput (default: True). rTct||d|_d|_d|_d|_|jrn|jr0|j|jkr |xjdzc_|j|jk |j r0|j |j kr |xj dzc_ |j |j k t|j|_t|j |_ dS)NrF) superr' _timenext _datacountsleeping _throttlerauto_sized_buffers read_limitr write_limitr^r)r2r3rr:s r5r'zThrottledDTPHandler.__init__s {+++   " 1 0,t>>**a/**,t>> 1-0@@@++q0++-0@@@!$T%;!">r?c6t|SrU)r r>r=s r5r>zThrottledDTPHandler.__repr__s""4(((r?cdS)NFrr=s r5rCz ThrottledDTPHandler.use_sendfilesur?ct|}|jr(|t ||j|SrU)rrr_throttle_bandwidthr~)r2 buffer_sizerlr:s r5rzThrottledDTPHandler.recvsF [)) ? B  $ $SZZ A A A r?ct|}|jr|||j|SrU)rr{rr)r2rLnum_sentr:s r5r{zThrottledDTPHandler.sendsB77<<%%   A  $ $Xt/? @ @ @r?cf|j'|jjs|jdSdSdSrU)rrrr=s r5_cancel_throttlerz%ThrottledDTPHandler._cancel_throttlers< ? &t/H & O " " $ $ $ $ $ ' & & &r?cNxj|z c_j|krd_t}j|z dz}|dkrTfd}j||j_|dz_dSdS)zkA method which counts data transmitted so that you burst to no more than x Kb/sec average. rrctjr jj}n jj}|dS)N)events)rrrprI add_channel)eventr2s r5unsleepz8ThrottledDTPHandler._throttle_bandwidth..unsleeps@|2 $ 0 $ 1$$E$22222r?rrN) rr r del_channelrr call_laterr-r)r2 len_chunk max_speednowsleepforrs` r5rz'ThrottledDTPHandler._throttle_bandwidths 9$ ?i ' 'DO''C,1H!||33333  """&&((("&+"8"8g0A#9##!1WDNNN% ( 'r?cp|tdSrU)rrr*)r2r:s r5r*zThrottledDTPHandler.closes*      r?)r;rrrrrrr'r>rCrr{rrr* __classcell__)r:s@r5rrs"JK????? )))  %%%%%%2r?)rVr+r[r)rpyftpdlib.exceptionsrrrpyftpdlib.iolooprrrr pyftpdlib.logr r pyftpdlib.utilsr __all__r rrr?r5rsg  444444222222,,,,,,111111******&&&&&&"""""" $$$$$$ . /~-~-~-~-~-~-~-~-B WWWWW*WWWWWr?