o ?Og(%@sddlmZddlZddlZddlZddlZddlZddlZddlZddl m Z gdZ ej dkp@e edo@e edo@e ejd ZGd d d ejZejZd8d d Zej dkr~e gd7Z ddlZd9ddddZddZddZddZGdddeZn#e gd7Z ddlZej dkZddZdd Zd!dZd"dZd#d$Zd%d&ZGd'd(d(Z ee!e j"ed)d*Z#ee!e$j%e#ee!e&j'e#d+d,Z(d-d.Z)eej*e(ej dkrd/d0Z+d1d2Z,eeje+nd3d0Z+d4d2Z,eeje+Gd5d6d6ed7Z-dS):)ABCMetaN)context) send_handle recv_handleForkingPicklerregisterdumpwin32ZCMSG_LEN SCM_RIGHTSsendmsgcsJeZdZdZiZejZfddZe ddZ e d ddZ e j Z ZS) rz)Pickler subclass used by multiprocessing.cs*tj||j|_|j|jdSN)super__init___copyreg_dispatch_tablecopydispatch_tableupdate_extra_reducersselfargs __class__@/opt/alt/python310/lib64/python3.10/multiprocessing/reduction.pyr&s  zForkingPickler.__init__cCs||j|<dS)z&Register a reduce function for a type.N)r)clstypereducerrrr+szForkingPickler.registerNcCs t}|||||Sr )ioBytesIOr getbuffer)robjprotocolZbufrrrdumps0szForkingPickler.dumpsr )__name__ __module__ __qualname____doc__rcopyregrrr classmethodrr$pickleloads __classcell__rrrrr!s   rcCst|||dS)z3Replacement for pickle.dump() using ForkingPickler.N)rr )r"filer#rrrr :sr ) DupHandle duplicate steal_handleF)source_processcCs6t}|dur |}|dur|}t|||d|tjS)zrrrrds rc@s"eZdZdZdddZddZdS)r/zPicklable wrapper for a handle.Nc Csf|durt}ttjd|}ztt|||dd|_Wt|nt|w||_ ||_ dS)NFr) osgetpidr3r8r9r5r4_handler;_access_pid)rr7accesspidprocrrrrjs  zDupHandle.__init__c Cs^|jtkr |jSttjd|j}zt||jt|j dtj Wt |St |w)z1Get the handle. This should only be called once.F) rGrCrDrEr3r8r9r5r4rFr:r;)rrJrrrrBys   zDupHandle.detachr )r%r&r'r(rrBrrrrr/hs  r/)DupFdsendfdsrecvfdsdarwincCsZtd|}tt|dg}||gtjtj|fgtr)|ddkr+t ddSdS)z,Send an array of fds over an AF_UNIX socket.irAz%did not receive acknowledgement of fdN) arraybyteslenr socket SOL_SOCKETr ACKNOWLEDGErA RuntimeError)sockZfdsmsgrrrrLs rLc Cstd}|j|}|dt|\}}}}|s|stz[tr%|dt|dkr3t dt||d\}} } |tj kro| tj krtt| |jdkrOt | | t|d|dkrjtdt||dt|WSWt dWt dt tfyYt dw) z/Receive an array of fds over an AF_UNIX socket.rOrrQzreceived %d items of ancdatarrPz Len is {0:n} but msg[0] is {1!r}zInvalid data received)rRitemsizeZrecvmsgrUZ CMSG_SPACEEOFErrorrWr=rTrXrVr ValueErrorZ frombytesAssertionErrorformatlist IndexError) rYsizeaZ bytes_sizerZZancdataflagsZaddrZ cmsg_levelZ cmsg_typeZ cmsg_datarrrrMsB           rMcCsHt|tjtj}t||gWddS1swYdSr<)rUfromfdfilenoAF_UNIX SOCK_STREAMrL)r>r7r?srrrrs"cCsHt|tjtj}t|ddWdS1swYdS)r@rrN)rUrerfrgrhrM)r>rirrrrs $cCsBt}|dur|||Strddlm}||Std)zReturn a wrapper for an fd.Nr)resource_sharerz&SCM_RIGHTS appears not to be available)rZget_spawning_popenrKZduplicate_for_childHAVE_SEND_HANDLErjr])fdZ popen_objrjrrrrKs  rKcCs.|jdurt|j|jjffSt|j|jjffSr )__self__getattrr__func__r%mrrr_reduce_methods rsc@seZdZddZdS)_CcCsdSr r)rrrrfsz_C.fN)r%r&r'rurrrrrts rtcCst|j|jffSr )ro __objclass__r%rqrrr_reduce_method_descriptorsrwcCst|j|j|jp iffSr )_rebuild_partialfuncrkeywords)prrr_reduce_partialr|cCstj|g|Ri|Sr ) functoolspartial)ryrrzrrrrxr}rxcCsddlm}t||ffS)Nr) DupSocket)rjr_rebuild_socket)rirrrr_reduce_sockets rcCs|Sr )rB)ZdsrrrrsrcCs"t|}t||j|j|jffSr )rKrfrfamilyrproto)ridfrrrrs cCs|}tj||||dS)N)rf)rBrU)rrrrrmrrrrsc@sdeZdZdZeZeZeZeZeZe j dkre Z e Z e Z neZeZeZeZeZeZeZeZddZdS)AbstractReducerzAbstract base class for use in implementing a Reduction class suitable for use in replacing the standard reduction mechanism used in multiprocessing.r cGsNtttjttttjttttj ttt j t tt j tdSr )rrrtrursr`appendrwint__add__r~rr|rUrrrrrrs  zAbstractReducer.__init__N)r%r&r'r(rrr rrsysplatformr1r0r/rLrMrKrsrwrxrrrrrrrrs(  r) metaclassr )NF).abcrr)r~rrCr+rUrrlr__all__rhasattrrkZPicklerrrr r3r0r1rrobjectr/rRrWrLrMrKrsrtrrurwr`rrrr|rxrrrrrrrrsl             #