a mzf9A@s0dZddlZddlZddlZddlZddlZddlZddlZddlZ ddl Z ddl Z ddl Z ddl mZddlmZmZddlmZmZmZmZddlmZmZmZddlmZdd lmZmZmZepeZ e j!j"e#ed  d d Z$e j!j"ed d Z%e j&ddddZ'e j&ddZ(e j&ddZ)ddZ*ddZ+e j!,de-e.e/e0fddZ1ddZ2e j!,deefd d!Z3e$d"d#Z4e$d$d%Z5d&Z6d'Z7Gd(d)d)eZ8e j&d*d+Z9e$e%e j!j:d,dd-d.d/Z;e j!j"e d0d e$e%d1d2Ze j!j"e#ed; dZ?eZ@esxe@sxe j!Ae>Z>e j&d?d@ZBe j&dAd4ZCe j&dBd8ZDe j!,dCdDdEeEdFfdDdGeEdFfdHdIdDeEdFeEdFfddEeEdFfddGeEdFfdJdeEdFeEdFff dKdLZFe j!,dMdNdOdPZGdQdRZHdS)SzTests 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.)Zabstractfile)paramscCsdjfit}||S)9Check that bound UNIX socket address is stored in server.zunix_{request.param}_sock)formatlocalsgetfixturevalue)requestnamerF/var/www/media/lib/python3.9/site-packages/cheroot/test/test_server.pyunix_sock_file,srcCs4tstdjtdddttt f S)z'Return an abstract UNIX socket address.z2{os} does not support an abstract socket namespace)osscheroot-test-socket) rpytestskiprr joinrstruuiduuid4decoderrrrunix_abstract_sock3sr(ccs*t\}}|Vt|t|dS)zYield a unix file socket.N)tempfilemkstemprcloseunlink)Z tmp_sock_fhZtmp_sock_fnamerrrunix_file_sockAs  r-cCs~tttftd}|jrJ|jjr&J||js8J|jjsDJ|jjD]}|jsLJqL||jjrpJ|jrzJdS)zBCheck that prepare() makes the server ready, and stop() clears it. bind_addrgatewayN) r rrr readyrequests_threadspreparestop) httpserverZthrrrrtest_prepare_makes_server_readyLs       r7cCsftttftd}|tj|jd}|| d| sDJ| | d| rbJdS)0Check that stop() interrupts running of serve().r.target?N) r rrr r4 threadingThreadservestartr#is_aliver5)r6 serve_threadrrrtest_stop_interrupts_servecs   rBexc_clscsdjtdttttftdt fdd} t j |d}| |d|stJ_|d|rJusJdS) z0Check that assigning interrupt stops the server.zshould catch {uuid!s})r%r.c sLz Wn:yF}z"t|kr2WYd}~n d}~00dSN)r>r$put)erCr6Z interrupt_msgZraise_marker_sentinelZresult_qrrrAs   z+test_server_interrupt..serve_threadr9r;N)rr%r&objectr rrr queueQueuer4r<r=r?r#r@Z interrupt get_nowait)rCrArrGrtest_server_interruptws"       rLcCs|tttftd}|dd}||jj_tj |j d}| | |j trPdnd|rdJ|jjrpJ|dS)r8r.c_s tdSrD)KeyboardInterrupt)argskwargsrrrraise_keyboard_interruptszTtest_serving_is_false_and_stop_returns_after_ctrlc..raise_keyboard_interruptr9rN)r rrr r4Z _connections _selectorselectr<r=r>r?r#Zexpiration_interval IS_SLOW_ENVr@Z_servingr5)r6rPrArrr2test_serving_is_false_and_stop_returns_after_ctrlcs   rUip_addrcCs6||tf}|jd|ks J|jdtks2JdS)z0Check that bound IP address is stored in server.rN)sendrr/) http_serverrVr6rrrtest_bind_addr_inets rZcCs||}|j|ksJdS)rNrXr/rYrr6rrrtest_bind_addr_unixs r]cCs||}|j|ksJdS)zBCheck that bound UNIX abstract socket address is stored in server.Nr[)rYr(r6rrrtest_bind_addr_unix_abstracts r^z/peer_creds/idsz/peer_creds/textscs$eZdZfddZddZZS) _TestGatewaycsz|j}|j}t|j}|tkrH|j|j|jf}|d t t |dS|t krl|d |j |jfdStt|S)N|!)reqconnruriPEERCRED_IDS_URIZpeer_pidZpeer_uidZpeer_gid send_payloadr#mapr$PEERCRED_TEXTS_URIZ peer_userZ peer_groupsuperr_respond)selfrbrcZreq_uriZ peer_creds __class__rrrjs z_TestGateway.respondcCs&|j}d|_||t|dS)Ns200 OK)rbstatusZensure_headers_sentwriter)rkpayloadrbrrrrfsz_TestGateway.send_payload)__name__ __module__ __qualname__rjrf __classcell__rrrlrr_s r_cCs||}t|_d|_|S)z3Construct a test server with ``peercreds_enabled``.T)rXr_r0Zpeercreds_enabledr\rrrpeercreds_enabled_servers ru)ZrerunsZ reruns_delayc Cs|}|j}t|tr|}tjj|dd}djfit}t t t f}d tt|}tXtj|t|d}||j|ksJtj|t|d}|jdksJWdn1s0YdS)z2Check that ``PEERCRED`` lookup works when enabled.safehttp+unix://{quoted}r`timeoutiN)r/ isinstancebytesr'urllibparsequoterrrgetpidgetuidgetgidr#rgr$requests_unixsocket monkeypatchr2getreraise_for_statustextrh status_code) http_request_timeoutrur6r/quoted unix_base_uriZexpected_peercredsZpeercreds_resppeercreds_text_resprrrtest_peercreds_unix_socks(  rzDModules `grp` and `pwd` are not available under the current platformc Cs|}d|_|j}t|tr"|}tjj|dd}djfit }ddl }ddl }| t j|t jf}dtt|}t8tj|t|d} | | j|ksJWdn1s0YdS) z6Check that ``PEERCRED`` resolution works when enabled.TrwrxrzrNrar{)Zpeercreds_resolve_enabledr/r}r~r'rrrrrgrppwdgetpwuidrrpw_namegetgrgidrgr_namer#rgr$rrr2rrhrr) rrur6r/rrrrZexpected_textcredsrrrr$test_peercreds_unix_sock_with_lookup"s(   rz\This regression test is for a Linux bug, and the resource module is not available on Windowsresource_limit)iir)Zindirectmany_open_socketscsb|jjfddt_|j_|dtjdksBJtfddjDs^JdS)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. csj|j|SrD)filenosaddsocketfileno)rc)_old_process_connnative_process_connrrrgszAtest_high_number_of_file_descriptors..native_process_conn/rc3s|]}|kVqdSrDr).0fnrrr ur z7test_high_number_of_file_descriptors..N)Zserver_instanceZ process_connsetrconnectlenany)Znative_server_clientrr)rrrr$test_high_number_of_file_descriptorsJs r SO_REUSEPORTz5socket.SO_REUSEPORT is not supported on this platformcCst|tdd}t|}|tjtjd||tft|ddt dd}| |d}| | | |jdusJdS)z5Check that port initialized externally can be reused.rrWNrT)r/r0 reuse_portr4)r getaddrinfor setsockopt SOL_SOCKETrbindr getsocknamer spyr4r5r+Z spy_exception)rYrVZmockerfamilysserverrrrrtest_reuse_portxs   rccs4tdVWdn1s&0YdS)z7Disable garbage collection when this fixture is in use.N)rZnogcrrrr _garbage_bins rc csntjddd}||j\}}||j|jd|fz|jVW||j||fn||j||f0dS)z7Set the resource limit two times bigger then requested.resourcez&The "resource" module is Unix-specificrrN)r!Z importorskipZ getrlimitZ RLIMIT_NOFILEZ setrlimitparam)rrZ soft_limitZ hard_limitrrrrs  c cs|dg}zdt|D]&}t}||||krq@q|d}||ksXJ|VW|D] }|qdn|D] }|qx0dS)z 0)rrr)rrWr)rrr)rrmin=0 must be > 0)rrr)rrWr)rrr)rWrKExpected an integer or the infinity value for the `max` argument but got 0.)rWr;zMExpected 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'.)rrWzmax=1 must be > min=2cCsDtjttf|dtd||dWdn1s60YdS)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 TypeErrorr rrrr#test_threadpool_invalid_threadranges rcCs^|tdddtdd}|tjtdd|Wdn1sP0YdS)zTest for ThreadPool multi-start behavior. Tests that when calling start() on a ThreadPool multiple times raises a :exc:`RuntimeError` Z _spawn_workercSs tjddS)NT)r1)typesSimpleNamespace)rrrr(r z7test_threadpool_multistart_validation..N)rz%Threadpools can only be started once.r)setattrr r?r!r RuntimeError)rrrrr%test_threadpool_multistart_validations r)I__doc__rrIrr)r<rr% urllib.parserr!r2rZpypytools.gc.customr_compatrrrrrr rr r r Zworkers.threadpoolr ZtestingrrrrTmarkZskipifhasattrZunix_only_sock_testZnon_macos_sock_testZfixturerr(r-r7rBZ parametrizeIOErrorrMOSErrorrrLrUrZr]r^rerhr_ruZflakyrrZ usefixturesrrZISSUE511Zforkedrrrfloatrrrrrrrs        $    !