Zd,(dZddlZddlmZddlmZdxZZdZ dZ d Z d Z d Z d Zd ZdZejddkZejZddlZddlmZmZddlmZmZmZddlmZddlm Z ddZ!ddZ"ddZ#GddZ$ddgZ%dS)zzmq Socket classN)ffi)libc,tjd|S)Nzsize_t*)rnewlengths 7lib/python3.11/site-packages/zmq/backend/cffi/socket.pyr sF)C)CcltjdttjdfSNz uint64_t*uint64_trrnspsizeofr r r r s'cgk22C :8N8N4O4OPr cltjdttjdfSNzint64_t*int64_trrr r r r s'SWZ00#cj6K6K2L2LMr cltjdttjdfSNzint*intrrr r r r s%376??C 50A0A,B,BCr cxtjd|zttjd|zfS)Nchar[%d]charrrs r r r s5GJ& !"" 6V#$$"r cTtjd|tjdfSrrrrvals r r r s!CGK$=$=sz*?U?U#Vr cTtjd|tjdfSrrrs r r r s!37:s#;#;SZ =R=R"Sr cTtjd|tjdfSrrrs r r r s!!5!5sz%7H7H Ir cftjd|dzz|tjd|zfS)Nrrrr)r r s r r r s3GJ&1* %s++Jv)r ZMQ_FD_T) SocketOption_OptType)ZMQError _check_rc_check_version)Frame)_retry_sys_callct|dtj}|tjkstr|tjkrt S|tjkrt|StSN _opt_type) getattrr'rint64 ZMQ_FD_64BITfdnew_int64_pointerbytesnew_binary_datanew_int_pointer)optionr opt_types r new_pointer_from_optr:)shv{HL99H8>!!l!x8;7N7N """ X^ # #v&&&   r c t|}|j}n#t$rtj}YnwxYw|tjkrt j||ddSt |dS)Nr)r&r/ ValueErrorr'rr5rbuffer)r8 opt_pointerr r9s r value_from_opt_pointerr?5s$f%%#    <  8>!!z+v..qqq11;q>""" 22ct|dtj}|tjkstr|tjkrt |S|tjkrt||St|Sr.) r0r'rr1r2r3value_int64_pointerr5value_binary_datavalue_int_pointer)r8valuer r9s r initialize_opt_pointerrFFslv{HL99H8>!!l!x8;7N7N"5))) X^ # # /// '''r ceZdZdZdZdZdZdZdZdZ ddZ e dZ dZ e dZddZd Zd Zd Zd Zd ZdZdZdZddZddZddZdS)SocketNFrc&| tj}||_||_|r"t jd||_d|_n&d|_tj |j ||_|jtj krtd|_ dS)Nzvoid *TF)zmqCOPY_THRESHOLDcopy_thresholdcontextrcast _zmq_socket_shadowC zmq_socket_zmq_ctxNULLr(_closed)selfrM socket_typeshadowrLs r __init__zSocket.__init__Ys  ! /N,  K"x&99D DLL DL |G,<== = (!&))F'HH  $ $ $ |HHH  $ eU # # 8>)) 9F BCCCZZF#9&%#P#P      HWo . .       s?AAc t|}|j}n#t$rtj}YnwxYwt |d\}}t tj|j ||||d}t|||}|tjj kr/|tj kr|dr |dd}|S)Nrrrs)r&r/r<r'rr:r,rQzmq_getsockoptrOr?rJ ROUTING_IDr5endswith)rVr8r9rc_sizet_pointerszvs r r_z Socket.gets (!&))F'HH  $ $ $ |HHH  $,@s+S+S+S( d.   Q  "6?B ? ? c&1 1 1HN** 5!!+#2#Ar@ctjd}t|ts!t |}tjd|}t j|t|}t|t j t j ||t|tt j ||j|t j|}t|dS)zSend a copy of a bufferable zmq_msg_t*zchar[]N)rrrur5 memoryviewtobytesrQzmq_msg_init_sizerr)memcpy zmq_msg_datar, zmq_msg_sendrO zmq_msg_close)rVbufflagszmq_msg c_messagerorc2s r _send_copyzSocket._send_copys',''#u%% ,S//))++CGHc**  #c(( 3 3"  (()SXX>>>1A5IIIog&&#r c|}|j}ttj||j||j}||S)z1Send a Frame on this socket in a non-copy manner.) fast_copyrr,rQrrOtrackerrp)rVframer frame_copyrrs r _send_framezSocket._send_framesS__&& $1A5III$r c&t|trtd|r+t|ts|||Sd}t|tr|r|jst d|}na|jrAt|}|j |jkr"|||tj St|||j}d}| ||}|r| |S)Nz.Message must be in bytes, not a unicode objectFzNot a tracked message)trackrLT)rurvrr+rrr<rLrnbytesrJ_FINISHED_TRACKERrrp) rVdatarcopyr close_framerrrs r sendz Socket.sends dC  NLMM M   4// ??4// /K$&& #>>$%<===&5$T**CzD$777U333"44d%@STTT" &&ue44G  Nr Tc|r)tjd}tj|nt j|}|j} ttj||j |n%#t$r|rtj |wxYw|s|Stj tj |tj|}|dd}tj |}t||S)Nr)r)rrrQ zmq_msg_initrJr+rr, zmq_msg_recvrO Exceptionrr=r zmq_msg_sizer)) rVrrrrr_buffer_bytesros r recvz Socket.recv.s  $gl++G N7 # # # #IE***EmG  ANGT5Eu M M M M    )(((    L*Q^G44anW6M6MNN _W % %"  s !A++"B rsctdd|dkr tj}| tj}t |t r|d}tj |j ||dS)as.monitor(addr, flags) Start publishing socket events on inproc. See libzmq docs for zmq_monitor for details. Note: requires libzmq >= 3.2 Parameters ---------- addr : str The inproc url used for monitoring. Passing None as the addr will cause an existing socket monitor to be deregistered. events : int [default: zmq.EVENT_ALL] The zmq event bitmask for which events will be sent to the monitor. rmonitorrNrr) r*rJ EVENT_ALLrrTrurvrwrQzmq_socket_monitorrO)rVaddreventss r rzSocket.monitorFsq$ vy))) A::]F <8D dC  ';;v&&D T-tV<<<<rs CCCPPMMCC WVSSII sz*%%* )1)++ 000000009999999999"""""" ! ! ! !####"((((O=O=O=O=O=O=O=O=d ' (r