a zeM{@sddlmZmZmZmZmZmZmZmZddl Z ddl Z ddl Z ddl Z ddl Z ddlZddlZedZejZGddde jZGddde jZedkre dS) )TESTFN import_moduleunlinkrequires_2G_4G gc_collect cpython_onlyNmmapc@seZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ ddZddZddZddZdd Zd!d"Zd#d$Zd%d&Zd'd(Zd)d*Zd+d,Zeeed-d.d/d0Zd1d2Zd3d4Zd5d6Z ee!j"d7kd8d9d:Z#e$ee!j"d7kd8d;d<Z%ee!j"d7kd8d=d>Z&ee!j"d7kd8d?d@Z'dAdBZ(dCdDZ)dEdFZ*dGdHZ+e,e!j"d7kdIdJdKZ-dLdMZ.dNdOZ/dPdQZ0eeejdRdSdTdUZ1dVS)W MmapTestscCstjtrttdSN)ospathexistsrrselfrB/opt/bitninja-python-dojo/embedded/lib/python3.9/test/test_mmap.pysetUps zMmapTests.setUpcCs&zttWnty Yn0dSr )r rrOSErrorrrrrtearDowns zMmapTests.tearDowncCsttd}zR|dt|d|dtd|t|dt}W|n |0tt |}| | dt| t |dt| |dd| |ddd| t|jt || t|jt |dd|d<d |tdtdd<| |dd| |ddd | |td td d |td|}|durr|dn2|d\}}||}| |t| |td|dd| |d|dd | |d|dd| |t || t|jd| t|jd d| t|jt | d dz|dWnty\Ynr0| t |d| t|jddttd}z(|dd| |dW|n |0| |d|dS)Nzbw+foors3bars3sfoobars [A-Za-z]+zregex match on mmap failed!*rb)openrwritePAGESIZEflushr filenoclosestrtype assertEqualfindlen assertRaises IndexError __getitem__ __setitem__researchfailspanseektell ValueErrorresize SystemErrorsize)rfmtpmatchstartendlengthrrr test_basicsd            zMmapTests.test_basicc Cszd}ttd}|d|Wdn1s20Yttd}tj||tjd}||ddd|dzd||dd<WntyYn 0|dz d|d <WntyYn 0|dz| d d |d WntyYn 0|dz| d d | d Wnty>Yn 0|dz| d |Wn&t ypYntyYn 0|d ttd&}|| d|dWdn1s0YWdn1s0Yttd}zt||d}Wn,ty@tjdr<|dYn"0tjdsZ|d|tjdrttd}||Wdn1s0YWdn1s0Yttd`}tj||tjd}d||dd<||ddd|d||Wdn1s80Yttd}| }Wdn1sl0Y||d|dttd}tj||tjd}d ||dd<||ddd |d|ttd&}|| d|dWdn1s0Y|t|j d ||Wdn1sP0Yttd*}|jttj||ddWdn1s0Ytjdkrvttd4}|jttj||tjtjtjdWdn1s0Ytjttdd B}ttdL}tj|||d}|t|jd|t|j d |Wdn1sl0YdS)N wbar$accessz#Readonly memory map data incorrect.bz$Able to write to readonly memory maprabcdrz"Able to resize readonly memory mapz*Readonly memory map data file was modifiedr+brwinz0Opening mmap with size+1 should work on Windows.z1Opening mmap with size+1 should raise ValueError.cz5Write-through memory map memory not updated properly.z8Write-through memory map data file not updated properly.z4Copy-on-write memory map data not written correctly.z4Copy-on-write test data file should not be modified.posix)flagsprotrJZ PROT_EXECrTsabcdef)r%rr&r r) ACCESS_READr- TypeErrorr6r8 write_byter;r<readr:sysplatform startswithr*truncate ACCESS_WRITEr( ACCESS_COPYr0r nameZ MAP_PRIVATE PROT_READgetattr)rmapsizefpr>r?stuffrTrrrtest_access_parameter{s ,           D   J ( (   $( :  & zMmapTests.test_access_parametercCs|ttjdddS)N)r0rr rrrrtest_bad_file_descszMmapTests.test_bad_file_desccCsttd>}d}t|}|||t||}Wdn1sN0Yt|dD]R}t||dD]>}|||}|| || ||| |ddqvqd| dS)Nwb+saabaacdeefaarxr!) r%rr/r&r(r r)ranger-r.r*)rr>datanr?rBZfinishslicerrrtest_tougher_finds  . zMmapTests.test_tougher_findcCsttd>}d}t|}|||t||}Wdn1sN0Y||dd||dd||dddd||ddd||dddd||ddd d||t dddS) Nrj one two onesonersonesr!rrg) r%rr/r&r(r r)r-r. bytearrayrr>rmrnr?rrr test_find_ends  .zMmapTests.test_find_endcCsttd>}d}t|}|||t||}Wdn1sN0Y||dd||dd||dddd||dddd||dd dd||dd dd||t dddS) Nrjrqrrrssone rr!rgr) r%rr/r&r(r r)r-rfindrtrurrr test_rfind s  .zMmapTests.test_rfindcCsttd}|ddWdn1s.0Yttd6}tj|dtjd}||Wdn1s~0YdS)NrjrHr$rI)r%rr&r r)rVr*rr>mfrrrtest_double_close2s  , zMmapTests.test_double_closec Csttd}|ddWdn1s.0Yttdh}t|d8}|t|dd||dddWdn1s0YWdn1s0YdS)Nrjrymzrb+rz Map size should equal file size.)r%rr&r r)r-r/rYrzrrrtest_entire_file<s , zMmapTests.test_entire_filec Csttd}|ddWdn1s.0YttdX}tj|ddtjd }|t|jdWdn1s0YWdn1s0YdS) NrGir}r$rryoffsetrJi8) r%rr&r r)rVr0r1r2rzrrrtest_length_0_offsetFs  , zMmapTests.test_length_0_offsetcCsttd}|ddWdn1s.0Yttd*}|jttj|dddWdn1sr0YdS)NrGir}w+brir)r%rr&r0r:r r)rr>rrrtest_length_0_large_offsetRs  , z$MmapTests.test_length_0_large_offsetc Cs$ttd^}|d|t|d}|ddd||dddd|Wdn1sn0Yd}t t |D]}t t |D]}t t |t ||D]v}|d||||||||d}td t |}||dd<||||||dd||qqqtd d }gd } t | | | D]4\} }} z|| || WntyYn0qPgd } | D] \} }} |t|j| || q|td d }|t|jddd|t|jd dd |t|jdd d |ddd |ddddS)Nrjs ABCDEabcderFrs ABCDEABCDEz'Map move should have duplicated front 5 0123456789r!d)ir!rrr))r!r!r!)r!r!r)r!rr!)rr!r!)r!rr)rr!r)rrr!rr)r%rr&r(r r)mover-r*rlr/max itertoolsproductr:r0) rr>r{rmdestsrccountZexpectedr?offsetssourcer=rrr test_move\sD  &,    zMmapTests.test_movecCs^tdt}ttD]}|||ddqttD]$}|d@}|||<||||q4dS)Nr!rz+anonymously mmap'ed contents should be zero)r r'rlr-)rr?xbrrrtest_anonymouss    zMmapTests.test_anonymouscCs&tdd}||j|ttd|d||ttd|d||ttdd|d||d|d||dttdd|d||dttdd|d||d ttdd|d ||d ttd ddS) Nr!rrsrrQrrg i) r addCleanupr*r&bytesrlr8r-rYrr?rrr test_read_alls"         zMmapTests.test_read_allcCsPtdd}||j|t|jd|t|jd|t|jgddS)Nr!rfoog@)rrr)r rr*r0rWrYrrrrtest_read_invalid_args   zMmapTests.test_read_invalid_argc Cstttd}tdt|}||dd<||dd|ddddddtjddd d f }|D]@}|D]6}|ddD]$}|||||||||qxqhq`dS Nr!rrri,rgii)rreversedrlr r/r-rZmaxsize)rsr?indicesrBstopsteprrrtest_extended_getslices  z MmapTests.test_extended_getslicec Cstttd}tdt|}ddddddtjddd d f }|D]}|D]}|ddD]z}||dd<||dd|t|}||||}tt|}|||||<|||||<||ddt|qXqHq@dSr) rrrlr r/rZrr-list) rrr?rrBrrLrmrrrtest_extended_set_del_slices  z%MmapTests.test_extended_set_del_slicecCsB|d||d|d|d|t|dS)Nrrrr)r&r(r r))rr>halfsizerrrmake_mmap_files  zMmapTests.make_mmap_filec Cs^ttd}|ttd.}|jtdtj|dtjdWdn1sP0YdS)Nrr$zcannot mmap an empty filerrI)r%rr*assertRaisesRegexr:r r)rVrrrrtest_empty_files   zMmapTests.test_empty_filecCsttd}ztj}|||}|||d}ttd}dD]N}z$tj|||d}|ddWntt t fyYqF0|ddqF|ttd}tj||||d}||ddd |z| d Wnt yYn~0|t |d |t|jd d||ddd ttd }|dd|||d ||||d |W|zttWntyYn0n.|zttWntyYn00dS) NrrrN)rgr!Nrrrrrr"r#r$)r%rr ALLOCATIONGRANULARITYrr*r)r-r:rW OverflowErrorr;r<r/r0r8r9r=r rr)rr>rr?rcrrrr test_offsetsV        zMmapTests.test_offsetcCsGdddtj}|tdS)Nc@seZdZddZdS)z*MmapTests.test_subclass..anon_mmapc_stjj|dg|Ri|S)Nr!)r __new__)klassargskwargsrrrr#sz2MmapTests.test_subclass..anon_mmap.__new__N)__name__ __module__ __qualname__rrrrr anon_mmap"sr)r r')rrrrr test_subclass!szMmapTests.test_subclassrazneeds mmap.PROT_READcCsd}ttd}|d|Wdn1s20Yttd6}tj||tjd}|t|jdWdn1s0YdS)NrFrGrHr$rUr)r%rr&r r)rar0rW)rrcrdr>r?rrrtest_prot_readonly's  , zMmapTests.test_prot_readonlycCs|tjtdSr )assertIsr errorrrrrr test_error0szMmapTests.test_errorcCsd}ttd"}|dt|Wdn1s60Yttd$}t|t|}Wdn1st0Ytt|D]6}|||| |||||dq| t |j d||dd|| dtt|D]<}||||| |||||dq| t |j | d||dd ||d | d|d ||d ||ddd |td ||d||ddd| t |jddS)NrrGrkrNrxrrs345rrs 012bar6789sbazrs 012barbaz9sba)r%rr&r/r r)rlr-r9rXr0r:r8 read_byterYrt)rrmrdr>r?irrrtest_io_methods3s8 0 2    zMmapTests.test_io_methodscCsVdD]L}tdd}||||d||d||||qdS)N)rr!rr)r rXr-r8rr*)rrr?rrrtest_non_ascii_byteUs   zMmapTests.test_non_ascii_bytentzrequires WindowscCsd}d}tjdt|dd}||dd<tjdt|dd}||dd<||dd|||dd|||tjdt|dd}||dd<tjdt|dd}||dd<||dd|||dd|||dS)Nrs abcdefghijr!rtagnameZboo)r r/r-r*)rZdata1Zdata2m1m2rrr test_tagname^s$    zMmapTests.test_tagnamecCsHtdd}d}tjdd|d}|t|t|t|ddS)Nr!rrrr)r r-rZ getsizeofr/)rrrrrrr test_sizeofxs   zMmapTests.test_sizeofcCstjdddd}ztjddddddWn Yn0|ttd}|dWdn1sl0Yttd}t|d }|z|d Wn Yn0z|ddWn Yn0|dS) Nr!rrrGs xxxxxxxxxxrNr)r r*r%rr&r)r;)rr?rdr>rrrtest_crasher_on_windowss( ( z!MmapTests.test_crasher_on_windowscCs^t}zF|t t|d}Wdn1s:0YW|n |0dS)NrF)socketr0rr r)r*)rrr?rrrtest_invalid_descriptors  0z!MmapTests.test_invalid_descriptorcCsHtdd}||jWdn1s.0Y||jdS)Nr!rF)r Z assertFalseclosed assertTruerrrrtest_context_managers*zMmapTests.test_context_managerc Cs||t@}tdd}tWdn1s20YWdn1sP0Y||jtd||jddS)Nr!rFz)wrong exception raised in context managerzcontext manager failed)r0 Exceptionr rZassertIsInstance exceptionrr)rexcr?rrrtest_context_manager_exceptions @ z(MmapTests.test_context_manager_exceptioncCs>tdd}t|}|||~t||ddS)Nr!r)r weakrefrefrr)rmmwrrrr test_weakrefs   zMmapTests.test_weakrefcCsXtdd}||dd||dd||dd||d d dS) Nr!rrrrkrsyzrpythonr)r r-r&)rrrrr0test_write_returning_the_number_of_bytes_writtens  z:MmapTests.test_write_returning_the_number_of_bytes_writtenz(cannot resize anonymous mmaps on WindowscCstdd}||j|dz|dWntyL|dYn0||dd|t |j |t |j d|t |j d dS) Nr!i rrhzresizing not supportedrr rL) r rr*rYr;r<ZskipTestr-r0r:rrXr&rrrrtest_resize_past_poss    zMmapTests.test_resize_past_poscCsttdd}|t||Wdn1s40Y|t|dWdn1sf0YdS)Nr!rr)r r0rWrrrrtest_concat_repeat_exceptions   & z&MmapTests.test_concat_repeat_exceptioncCsZtdd}||j|d|}||tjdrV| t |jdt ddS)Nr!rrlinuxr) r rr*r&r(Z assertIsNonerZr[r\r0rr/)rrresultrrrtest_flush_return_values     z!MmapTests.test_flush_return_valuec Cstd}td}d}tdd|D}t||D]Z\}}d|}||}d} d|d |d |d f} ttd } | || t | | D]\} } t t | }t j | |||d }| | |t|}||||d| ||dt|||dt| ||dt|Wdn1sL0Y|t|}||qWdq41s0Yq4dS)Nzr\S+), length=(?P\d+), pos=(?P\d+), offset=(?P\d+)>z)2rri@Bicss"|]}|dtjtjVqdS)rN)r r).0rcrrr s z&MmapTests.test_repr..rH)ZACCESS_DEFAULTrVr_r^rrFrrQrj)rJrrJrDposr)r4compiletuplezipr%rr&r(rrrbr r)r8rAreprZassertIsNotNoner-groupr+)rZopen_mmap_repr_patZclosed_mmap_repr_patZmapsizesrrrcrmrDZaccessesZ positionsrdrJrZaccintrrArrr test_reprsB      6zMmapTests.test_reprmadvisez needs madvisecCsxdt}td|}|td|tj|Wdn1sD0Y|td|tjdWdn1s~0Y|td |tjddWdn1s0Y|td"|tjttjWdn1s0Y| |tjd| |tjtd| |tjt|d| |tjddd| |tjd|ddS)Nrr!zmadvise start out of boundszmadvise length invalidrzmadvise length too large) r'r rr:rZ MADV_NORMALrrZrr-)rr=r?rrr test_madvise s ,,.0zMmapTests.test_madviseN)2rrrrrrErfrirprvrxr|r~rrrrrrrrrrrrunittest skipUnlesshasattrr rrrrr r`rr rrrrrrrZskipIfrrrrrrrrrr s^^}    0  9 "         $r c@speZdZddZddZddZddZd d Zd d Ze e j e kd ddZe e j e kd ddZdS)LargeMmapTestscCs ttdSr rrrrrrr!szLargeMmapTests.setUpcCs ttdSr rrrrrr$szLargeMmapTests.tearDownc Cstjdddkstjdkr.tddtdttd}z |||||WnFt t t fyz | Wnt t fyYn0t dYn0|S) NrrOdarwinZ largefilez-test requires %s bytes and a long time to runrz*filesystem does not have largefile support)rZr[rr+r%rr8r&r(rrr:r*rZSkipTest)rZ num_zeroestailr>rrr_make_test_file's       zLargeMmapTests._make_test_filec Csx|ddX}tj|ddtjd }||ddWdn1sL0YWdn1sj0YdS)Nl rlri )rr r)rVr-rr>r?rrrtest_large_offset8sz LargeMmapTests.test_large_offsetc Cs|dd}tjdkr|t&tj|dtjdWdn1sN0Y|t&tj|dtjdWdn1s0Ytj|dtjd }| | dWdn1s0YWdn1s0YdS)NlrrrIrry) rrZrr0rr r)rVr:r-r=rrrrtest_large_filesize=s  4 4z"LargeMmapTests.test_large_filesizec Csd}|t|d}|t|}|||Z}tj|dtjd$}|||||Wdn1sn0YWdn1s0YdS)Ns DEARdear rrrI)r/rr r)rVr-)rboundaryrrBrCr>r?rrr_test_around_boundaryKs  z$LargeMmapTests._test_around_boundaryz!test cannot run on 32-bit systemscCs|tdSr )rrrrrrtest_around_2GBSszLargeMmapTests.test_around_2GBcCs|tdSr )rrrrrrtest_around_4GBWszLargeMmapTests.test_around_4GBN)rrrrrrrrrrrrZrrrrrrrrrs r__main__)Z test.supportrrrrrrrr rr r4rrrZrr r'ZTestCaser rrmainrrrrs&(=