Df9A2dZddlZddlZddlZddlZddlZddlZddlZddlZ ddl Z ddl Z ddl Z ddl mZddlmZmZddlmZmZmZmZddlmZmZmZddlmZdd lmZmZmZepeZ e j!"e#ed  d Z$e j!"ed Z%e j&ddZ'e j&dZ(e j&dZ)dZ*dZ+e j!,de-e.e/e0fdZ1dZ2e j!,deefdZ3e$dZ4e$dZ5dZ6dZ7GddeZ8e j&d Z9e$e%e j!:d!d"d#Z;e j!"e d$ e$e%d%Ze j!"e#ed- d. e j!,deefd/Z?eZ@ese@se j!Ae>Z>e j&d0ZBe j&d1ZCe j&d2ZDe j!,d3d4d5eEd6fd4d7eEd6fd8d9d4eEd6eEd6fdd5eEd6fdd7eEd6fd:deEd6eEd6ff d;ZFe j!,dZGd?ZHdS)@zTests for the HTTP server.N) DefaultGc)btonntob)IS_LINUXIS_MACOS IS_WINDOWS SYS_PLATFORM)IS_UID_GID_RESOLVABLEGateway HTTPServer) ThreadPool)ANY_INTERFACE_IPV4ANY_INTERFACE_IPV6EPHEMERAL_PORTAF_UNIXz:UNIX domain sockets are only available under UNIX-based OSreasonz9Peercreds lookup does not work under macOS/BSD currently.)abstractfile)paramsc^djdit}||S)9Check that bound UNIX socket address is stored in server.zunix_{request.param}_sock)formatlocalsgetfixturevalue)requestnames G/var/www/media/lib/python3.11/site-packages/cheroot/test/test_server.pyunix_sock_filer!,s5 . & - 9 9 9 9D  " "4 ( ((c ts-tjdtddt ttj f S)z'Return an abstract UNIX socket address.z2{os} does not support an abstract socket namespace)osr"scheroot-test-socket) rpytestskiprr joinrstruuiduuid4decoderr"r unix_abstract_sockr,3sx    %vv66    88" S    vxxr"c#Ktj\}}|Vtj|tj|dS)zYield a unix file socket.N)tempfilemkstempr$closeunlink) tmp_sock_fhtmp_sock_fnames r unix_file_sockr4AsM#+"2"4"4K H[Inr"cTtttft}|jrJ|jjrJ||jsJ|jjsJ|jjD] }|jsJ ||jjrJ|jrJdS)zBCheck that prepare() makes the server ready, and stop() clears it. bind_addrgatewayN) r rrr readyrequests_threadspreparestop) httpserverthrs r test_prepare_makes_server_readyr@Ls%~6J "++++     ''''"+yOO"++++r"ctttft}|t j|j}|| d| sJ| | d| rJdS)0Check that stop() interrupts running of serve().r6target?N) r rrr r< threadingThreadservestartr'is_aliver=)r> serve_threads r test_stop_interrupts_serverLcs%~6J #:+;<< interrupt_msgraise_marker_sentinelresult_qs r rKz+test_server_interrupt..serve_threads 4         4 4 41vv&& 2333333333'&&&&& 4sA(AArCrEN)rr)r*objectr rrr queueQueuer<rFrGrIr'rJ interrupt get_nowait)rMrKr>rSrTrUs` @@@@r test_server_interruptr[wsQ,22 2EEM"HH%~6J {}}H444444444#<888Lc  " """"#7=11Jc$$&&&&&    $9 9 9 9 9 9 9r"ctttft}|d}||jj_tj |j }| | |j trdndz|rJ|jjrJ|dS)rBr6ctrP)KeyboardInterrupt)argskwargss r raise_keyboard_interruptzTtest_serving_is_false_and_stop_returns_after_ctrlc..raise_keyboard_interrupts!!!r"rCrN)r rrr r< _connections _selectorselectrFrGrHrIr'expiration_interval IS_SLOW_ENVrJ_servingr=)r>rarKs r 2test_serving_is_false_and_stop_returns_after_ctrlcris%~6J """0HJ%,#:+;<<s r test_bind_addr_inetrosT!!7N";<s r test_bind_addr_unixrss1!!.11J  > 1 1 1 1 1 1r"cJ||}|j|ksJdS)zBCheck that bound UNIX abstract socket address is stored in server.Nrq)rnr,r>s r test_bind_addr_unix_abstractrus3!!"455J  #5 5 5 5 5 5 5r"z/peer_creds/idsz/peer_creds/textsc$eZdZfdZdZxZS) _TestGatewayc|j}|j}t|j}|tkrQ|j|j|jf}|d tt|dS|tkr6|d |j |jfdStt |S)N|!)reqconnruriPEERCRED_IDS_URIpeer_pidpeer_uidpeer_gid send_payloadr'mapr(PEERCRED_TEXTS_URI peer_user peer_groupsuperrwrespond)selfr{r|req_uri peer_creds __class__s r rz_TestGateway.respondshxsw-- & & & t}DJ   chhs3 ';';<< = = = F * * *   chh'HII J J J F\4((00222r"c|j}d|_||t |dS)Ns200 OK)r{statusensure_headers_sentwriter)rpayloadr{s r rz_TestGateway.send_payloads@h  !!! $w--     r")__name__ __module__ __qualname__rr __classcell__)rs@r rwrwsG 3 3 3 3 3!!!!!!!r"rwcV||}t|_d|_|S)z3Construct a test server with ``peercreds_enabled``.T)rmrwr8peercreds_enabledrrs r peercreds_enabled_serverrs-!!.11J%J#'J r")reruns reruns_delayc|}|j}t|tr|}tj|d}djdit}tj tj tj f}d tt|}t!j5t%j|t(z|}||j|ksJt%j|t.z|}|jdksJ ddddS#1swxYwYdS) z2Check that ``PEERCRED`` lookup works when enabled.safehttp+unix://{quoted}rytimeoutiNr)r7 isinstancebytesr+urllibparsequoterrr$getpidgetuidgetgidr'rr(requests_unixsocket monkeypatchr:getr~raise_for_statustextr status_code) http_request_timeoutrr>r7quoted unix_base_uriexpected_peercredspeercreds_resppeercreds_text_resps r test_peercreds_unix_sockrs *J$I)U##'$$&& \     3 3F1*1==FHH==Mbikk29;;>#c+=">">??  ( * * 6 6! , ,(    '')))"&88888&l . .(   #.#55555 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6s A+EE EzDModules `grp` and `pwd` are not available under the current platformc|}d|_|j}t|tr|}t j|d}djd it}ddl }ddl }| tjj|tjjf}dt)t*|}t-j5t1j|t4z|} | | j|ksJ ddddS#1swxYwYdS) z6Check that ``PEERCRED`` resolution works when enabled.TrrrrNrzrr)peercreds_resolve_enabledr7rrr+rrrrrgrppwdgetpwuidr$rpw_namegetgrgidrgr_namer'rr(rrr:rrrr) rrr>r7rrrrexpected_textcredsrs r $test_peercreds_unix_sock_with_lookupr"s*J+/J($I)U##'$$&& \     3 3F1*1==FHH==MJJJJJJ RY[[!!) RY[[!!)#c+=">">??  ( * *>>&l . .(    ,,..."'+====== >>>>>>>>>>>>>>>>>>s AEE"Ez\This regression test is for a Linux bug, and the resource module is not available on Windowsresource_limit)ii)r)indirectmany_open_socketsc|jjfdt_|j_|dt jdksJt fdjDsJdS)a>Test the server does not crash with a high file-descriptor value. This test shouldn't cause a server crash when trying to access file-descriptor higher than 1024. The earlier implementation used to rely on ``select()`` syscall that doesn't support file descriptors with numbers higher than 1024. c|j|j|SrP)filenosaddsocketfileno)r|_old_process_connnative_process_conns r rzAtest_high_number_of_file_descriptors..native_process_conngs9#'' (:(:(<(<===  &&&r"/rc3$K|] }|kV dSrPr).0fnrs r z7test_high_number_of_file_descriptors..us(JJr^#JJJJJJr"N)server_instance process_connsetrconnectlenany)native_server_clientrrrs `@@r $test_high_number_of_file_descriptorsrJs6-<I''''''#&%%8K(5  %%% "* + +a / / / / JJJJ.A.IJJJ J JJJJJJr" SO_REUSEPORTz5socket.SO_REUSEPORT is not supported on this platformc tj|tdd}tj|}|tjtjd||tft|ddtd}| |d}| | | |jJdS)z5Check that port initialized externally can be reused.rrlNrT)r7r8 reuse_portr<)r getaddrinfor setsockopt SOL_SOCKETrbindr getsocknamer spyr<r=r0 spy_exception)rnrjmockerfamilysserverrs r test_reuse_portrxs  8 8 ;A >F fALL"F$7;;;FFG^ $%%% --//"1"%w4F **VY ' 'C NN KKMMMGGIII   $ $ $ $ $r"c#Kt5dVddddS#1swxYwYdS)z7Disable garbage collection when this fixture is in use.N)rnogcrr"r _garbage_binrs      s 599c#LKtjdd}||j\}}||j|jdz|f |jV||j||fdS#||j||fwxYw)z7Set the resource limit two times bigger then requested.resourcez&The "resource" module is Unix-specificrrN)r% importorskip getrlimit RLIMIT_NOFILE setrlimitparam)rr soft_limit hard_limits r rrs"7H &//0FGGJ    J' Mm 81J 3KLLLLL81J 3KLLLLs  BB#c#K|dg} t|D]D}tj}||||krnE|d}||ksJ|V|D]}|dS#|D]}|wxYw)z 0)rrr )rrlr )rrr )rrmin=0 must be > 0)rrr )rrlr )rrr )rlrKExpected an integer or the infinity value for the `max` argument but got 0.)rlrEzMExpected an integer or the infinity value for the `max` argument but got 0.5.)rrr )r1zMExpected an integer or the infinity value for the `max` argument but got '1'.)rrlzmax=1 must be > min=2ctjttf|5t d||ddddS#1swxYwYdS)zTest that a ThreadPool rejects invalid min/max values. The ThreadPool should raise an error with the proper message when initialized with an invalid min+max number of threads. matchNr)r%raises ValueError TypeErrorrr s r #test_threadpool_invalid_threadrangers0  I.e < < <                        sAA Ac|tddtd}|tjt d5|ddddS#1swxYwYdS)zTest for ThreadPool multi-start behavior. Tests that when calling start() on a ThreadPool multiple times raises a :exc:`RuntimeError` _spawn_workerc,tjdS)NT)r9)typesSimpleNamespace)rs r z7test_threadpool_multistart_validation..(s%'d333r"N)rz%Threadpools can only be started once.r)setattrrrIr%r RuntimeError)rrs r %test_threadpool_multistart_validationrs33 4 BHHJJJ |+R S S S  sA??BB)I__doc__r$rWrr.rFrr) urllib.parserr%r:rpypytools.gc.customr_compatrrrrr r rr r r workers.threadpoolrtestingrrrrgmarkskipifhasattrunix_only_sock_testnon_macos_sock_testfixturer!r,r4r@rL parametrizeIOErrorr^OSErrorrr[rirorsrur~rrwrflakyrr usefixturesrrISSUE511forkedrrrfloatrrrrr"r r3s}   )))))) BBBBBBBBBBBB??????????++++++$* k(( """ G) k(( F) +,,,))-,)       .'''(  !:!:!:H8 555222666%(!!!!!7!!!*!!,,66-,6< ( >> >B : ! ,--KK.- K@''' B  % %  % (+1;+=+=,,,(  MMM2   85   E%LL   Be  EE%LL%%,,' Be  Be  EE%LL%%,,'( ' ')( ')$   %$  r"