U e5dD@sdZddgZddlmZddlZddlZddlZddlZddlZej dkrXddl Z dZ n ddl Z dZ ej ejBZd Ze r~d Znd Zd d ZGdddZdZGdddZdS)zProvides shared memory for direct access across processes. The API of this package is currently provisional. Refer to the documentation for details. SharedMemory ShareableList)partialNntFTz/psm_Zwnsm_cCs"tttd}tt|}|S)z6Create a random filename for the shared memory object.)_SHM_SAFE_NAME_LENGTHlen_SHM_NAME_PREFIXsecretsZ token_hex)nbytesnamer5/usr/lib64/python3.8/multiprocessing/shared_memory.py_make_filename&src@seZdZdZdZdZdZdZej Z dZ e r.dndZ ddd Zd d Zd d ZddZeddZeddZeddZddZddZdS)raCreates a new shared memory block or attaches to an existing shared memory block. Every shared memory block is assigned a unique name. This enables one process to create a shared memory block with a particular name so that a different process can attach to that same shared memory block using that same name. As a resource for sharing data across processes, shared memory blocks may outlive the original process that created them. When one process no longer needs access to a shared memory block that might still be needed by other processes, the close() method should be called. When a shared memory block is no longer needed by any process, the unlink() method should be called to ensure proper cleanup.NiTFrc Csl|dkstd|r0ttjB|_|dkr0td|dkrL|jtj@sLtdtrH|dkrt}ztj ||j|j d|_ Wnt k rYqZYnX||_ qqZn.|jrd|n|}tj ||j|j d|_ ||_ z<|r|rt|j |t|j }|j}t|j ||_Wn tk r*|YnXddlm}||j d n|r|dkr^tn|}ttjtjtj|d ?d @|d @|}zXt}|tjkr|dk rt t j!t"t j!|tjnWqNtjd ||d |_W5t|X||_ qVqNnX||_ t#tj$d|}zt%|tj$ddd} W5t|Xt&| }tjd ||d |_||_'t(|j|_)dS)Nrz!'size' must be a positive integerz4'size' must be a positive number different from zeroz&'name' can only be None if create=True)mode/)register shared_memory lr)ZtagnameF)* ValueError_O_CREXosO_RDWR_flagsO_EXCL _USE_POSIXr _posixshmemZshm_open_mode_fdFileExistsError_name_prepend_leading_slash ftruncatefstatst_sizemmap_mmapOSErrorunlinkresource_trackerr_winapiZCreateFileMappingZINVALID_HANDLE_VALUEZNULLZPAGE_READWRITEZ CloseHandleZ GetLastErrorZERROR_ALREADY_EXISTSerrnoZEEXISTstrerrorZOpenFileMappingZ FILE_MAP_READZ MapViewOfFileZVirtualQuerySize_size memoryview_buf) selfr createsizeZstatsrZ temp_nameZh_mapZlast_error_codeZp_bufrrr__init__Is             zSharedMemory.__init__cCs&z |Wntk r YnXdSN)closer*r3rrr__del__s zSharedMemory.__del__cCs|j|jd|jffS)NF) __class__r r5r9rrr __reduce__s zSharedMemory.__reduce__cCs|jjd|jd|jdS)N(z, size=))r;__name__r r5r9rrr__repr__szSharedMemory.__repr__cCs|jS)z4A memoryview of contents of the shared memory block.)r2r9rrrbufszSharedMemory.bufcCs.|j}tr*|jr*|jdr*|jdd}|S)z4Unique name that identifies the shared memory block.rrN)r#rr$ startswith)r3Z reported_namerrrr s   zSharedMemory.namecCs|jS)zSize in bytes.)r0r9rrrr5szSharedMemory.sizecCsX|jdk r|jd|_|jdk r4|jd|_trT|jdkrTt|jd|_dS)zkCloses access to the shared memory from this instance but does not destroy the shared memory block.Nrr)r2releaser)r8rr!rr9rrrr8s     zSharedMemory.closecCs2tr.|jr.ddlm}t|j||jddS)zRequests that the underlying shared memory block be destroyed. In order to ensure proper cleanup of resources, unlink should be called once (and only once) across all processes which have access to the shared memory block.r) unregisterrN)rr#r,rDrZ shm_unlink)r3rDrrrr+s   zSharedMemory.unlink)NFr)r? __module__ __qualname____doc__r#r!r)r2rrrr rr$r6r:r<r@propertyrAr r5r8r+rrrrr0s(  l     utf8c @seZdZdZedededededdj diZ dZ d d d d d d d d dZ e ddZd6ddddZddZddZddZddZddZddZd d!Zd"d#Zed$d%Zed&d'Zed(d)Zed*d+Zed,d-Zed.d/Zed0d1Zd2d3Z d4d5Z!dS)7raPattern for a mutable list-like object shareable via a shared memory block. It differs from the built-in list type in that these lists can not change their overall length (i.e. no append, insert, etc.) Because values are packed into a memoryview as bytes, the struct packing format for any storable value must require no more than 8 characters to describe its format.qdzxxxxxxx?z%dsNzxxxxxx?xcCs|Sr7rvaluerrr zShareableList.cCs|dtSN)rstripdecode _encodingrMrrrrO rPcCs |dSrQ)rSrMrrrrO rPcCsdSr7r)Z_valuerrrrO rP)rrrcCs:t|ttdjfsdSt|tr$dSt|tr2dSdSdS)zUsed in concert with _back_transforms_mapping to convert values into the appropriate Python objects when retrieving them from the list as well as when storing them.NrrrrV) isinstancestrbytesr;rMrrr_extract_recreation_codes  z&ShareableList._extract_recreation_coder csr|dk rvfdd|D}t|_tfdd|D_fdd|D}tdjd|jj }nd}|dk r|dkrt |_ nt |d |d _ |dk rNt tj djj jd jfjtj d|j jjffd d|Dtj jj jjffd d|Dtj j j jjf|n t_tjj jd_dS)NcsPg|]H}t|ttfs$jt|n&jt|jt|jdfqS)r)rWrXrY_types_mappingtype _alignmentr .0itemr9rr s   z*ShareableList.__init__..c3s0|](}|ddkrjnt|ddVqdS)rsN)r^int)r`fmtr9rr *sz)ShareableList.__init__..csg|]}|qSr)rZr_r9rrrb.srJrLT)r4r5rc3s&|]}t|tr|n|VqdSr7)rWrXencoder`v_encrrrfMsc3s|]}|VqdSr7)rhrirkrrrfSs)r _list_lentuple_allocated_bytesstructZcalcsize_format_size_metainfojoin_format_packing_metainfo_format_back_transform_codesrshmrU pack_intorA_offset_data_start_offset_packing_formats_offset_back_transform_codes unpack_from)r3Zsequencer Z_formatsZ_recreation_codesZrequested_sizer)rlr3rr6sz        zShareableList.__init__cCsj|dkr |n||j}||jks*|jdkr2tdtd|jj|j|dd}|d}|t }|S)z>Gets the packing format for a single value stored in the list.r Requested position out of range.8srLrR) rm IndexErrorrprzrurArxrSrTrU)r3positionrjre fmt_as_strrrr_get_packing_formatds   z!ShareableList._get_packing_formatcCs\|dkr |n||j}||jks*|jdkr2tdtd|jj|j|d}|j|}|S)z9Gets the back transformation function for a single value.rr{b)rmr}rprzrurAry_back_transforms_mapping)r3r~transform_codeZtransform_functionrrr_get_back_transformts z!ShareableList._get_back_transformcCs~|dkr |n||j}||jks*|jdkr2tdtd|jj|j|d|t| |}td|jj|j ||dS)zvSets the packing format and back transformation code for a single value in the list at the specified position.rr{r|rLrN) rmr}rprvrurArxrhrUrZry)r3r~rrNrrrr!_set_packing_format_and_transforms   z/ShareableList._set_packing_format_and_transformcCsjz6|jt|jd|}t|||jj|\}Wntk rRtdYnX| |}||}|S)Nzindex out of range) rwsumrorprzrrurAr}r)r3r~offsetrjZback_transformrrr __getitem__s  zShareableList.__getitem__cCsz&|jt|jd|}||}Wntk rBtdYnXt|ttfsf|jt |}|}nZt|trz| t n|}t ||j|krt d|ddkr|}n|jt|j|f}||||t||jj||dS)Nzassignment index out of rangez(bytes/str item exceeds available storagerrc)rwrrorr}rWrXrYr\r]rhrUr rrrprvrurA)r3r~rNrZcurrent_formatZ new_formatZ encoded_valuerrr __setitem__s6  zShareableList.__setitem__cCst|j|jjddfS)Nr[r)rr;rur r9rrrr<szShareableList.__reduce__cCstd|jjddS)NrJr)rprzrurAr9rrr__len__szShareableList.__len__cCs"|jjdt|d|jjdS)Nr=z, name=r>)r;r?listrur r9rrrr@szShareableList.__repr__csdfddtjDS)z>The struct packing format used by all currently stored values.rgc3s|]}|VqdSr7)r)r`ir9rrrfsz'ShareableList.format..)rrrangermr9rr9rformatszShareableList.formatcCs |jdS)z=The struct packing format used for metainfo on storage sizes.rJrmr9rrrrqsz#ShareableList._format_size_metainfocCs d|jS)z?The struct packing format used for the values' packing formats.r|rr9rrrrssz&ShareableList._format_packing_metainfocCs d|jS)z?The struct packing format used for the values' back transforms.rrr9rrrrtsz*ShareableList._format_back_transform_codescCs|jddS)NrrLrr9rrrrwsz ShareableList._offset_data_startcCs|jt|jSr7)rwrror9rrrrxsz%ShareableList._offset_packing_formatscCs|j|jdS)NrL)rxrmr9rrrrysz*ShareableList._offset_back_transform_codescstfdd|DS)zCL.count(value) -> integer -- return number of occurrences of value.c3s|]}|kVqdSr7r)r`entryrMrrrfsz&ShareableList.count..)r)r3rNrrMrcountszShareableList.countcCs4t|D]\}}||kr|Sqt|ddS)zpL.index(value) -> integer -- return first index of value. Raises ValueError if the value is not present.z not in this containerN) enumerater)r3rNr~rrrrindexs zShareableList.index)N)"r?rErFrGrdfloatboolrXrYr;r\r^r staticmethodrZr6rrrrrr<rr@rHrrqrsrtrwrxryrrrrrrrs^  F       )rG__all__ functoolsrr(rr.rpr r r-rrO_CREATrrrr rrrUrrrrrs,    E