wg^> d dl m Z mZ d dlZd dlZd dlZd dlZd dlmZ ddlm Z ej Z ej Z
G d d Z G d d Zy)
)datetimetimezoneN)parse_qs )ConnectionRefusedErrorc e Zd Zd ZddZd Zy) EventBufferc i | _ y N)bufferselfs S/home/mcse/projects/flask/flask-venv/lib/python3.12/site-packages/socketio/admin.py__init__zEventBuffer.__init__ s c t t j dz }| d| }|| j vr|||d| j |< y | j | dxx |z
cc< y )N ;) timestamptypecountr )inttimer )r r r r keys r pushzEventBuffer.push sg $t+ 1TF# dkk ! & DKK
KK W % . %r c n | j }i | _ |j D cg c] }| c}S c c}w r )r values)r r values r
get_and_clearzEventBuffer.get_and_clear s, #)==?3%333s 2N)r )__name__
__module__ __qualname__r r r r r r r
s
/4r r c e Zd Z ddZd Zd Zd Zd ZddZddZ dd Z
d
Z d Z d Z
dd
Zd Z ddZd Zd ZddZd Zd Zd Zd ZddZy)InstrumentedServerNc j | t d || _ || _ || _ || _ |xsF t | j j d r | j j j nt | _ || _
|| _ t | _
d| _ d| _ | j! y)zInstrument the Socket.IO server for monitoring with the `Socket.IO
Admin UI `_.
Nzauth must be specifiedhost_id)
ValueErrorsioauthadmin_namespace read_onlyhasattrmanagerr' HOSTNAME server_idmodeserver_stats_intervalr event_bufferstop_stats_event
stats_task
instrument)r r) r* r1 r, r0 namespacer2 s r r zInstrumentedServer.__init__$ s
< 56 6 ( ""
(/0@0@)(LDHH $ $
%:" 'M !%
r c B | j j d| j | j | j dk( rI| j
s| j j d| j | j | j j d| j | j | j j d| j | j | j j d| j | j i | j j _ | j j | j _
| j | j _ | j j j | j j _ | j | j j _ | j j j" | j j _ | j& | j j _ | j j j( | j j _ | j, | j j _ | j j. j d| j0 | j j. j d| j2 d d
lm} | j j. j8 | j j. _ | j< | j j. _ |j> |_ tC jD | jF jH | |_ |jJ |_& tC jD | jF jN | |_% | j dk( rA|jP |_) tC jD | jF jT | |_( y y ) Nconnectr7 developmentemitjoinleave _disconnect
disconnectr Socket)+r) on
admin_connectr+ r1 r,
admin_emitadmin_enter_roomadmin_leave_roomadmin_disconnectr. _timestamps_trigger_event"_InstrumentedServer__trigger_eventbasic_enter_room%_InstrumentedServer__basic_enter_room_basic_enter_roombasic_leave_room%_InstrumentedServer__basic_leave_room_basic_leave_roomr< _InstrumentedServer__emit_emiteio_handle_eio_connect_handle_eio_disconnectengineio.socketrB _ok_InstrumentedServer__ok_eio_http_responsehandle_post_request(_InstrumentedServer__handle_post_request functools
partialmethod __class___eio_handle_post_request_websocket_handler&_InstrumentedServer__websocket_handler_eio_websocket_handler
_send_ping _InstrumentedServer__send_ping_eio_send_pingr rB s r r6 z InstrumentedServer.instrument> s It 1 1 " 2 2 4 99
%>> FDOO&*&:&: < FD$9$9&*&:&: < GT%:%:&*&:&: < M4+@+@&*&:&: < ,.DHH ( (,xx'>'>DHH $&*&9&9DHH # 11
HH /040F0FDHH - 11
HH /040F0FDHH - '+hh&6&6&;&;DHH #$(JJDHH !
4#;#;< d&A&AB + HHLL , , 22 '-'A'A$%.%<%< NN 3 3T&;" &,%>%>"$-$;$; NN 1 14%9! 99
%!'!2!2F ) 7 7--t!5F &r c | j dk( r| j j | j _ | j j j
| j j _ | j j j | j j _ | j j j | j j _
| j j j | j j _
ddlm} |j |_ |j$ |_ | j dk( r|j( |_ y y )Nr; r rA )r1 r) rK rJ r. rM rL rP rO rR r< rT rY rX rW rB r\ r[ rb ra re rd rg s r uninstrumentz InstrumentedServer.uninstrumenty s 99
%&*hh&>&>DHH # 33
HH - 33
HH -$(HH$4$4$;$;DHH ! 88<<,, *%+%A%A"$*$>$>! 99
% & 2 2F &r c F j rsd}t j t r| j k( }n:t j t r| j v }n j | }|s t d fd} j
j
|| y )NFzauthentication failedc f j j d dg}j s|g dz
}j dk( r|j d j j dd|i| j j dk( rg }j j j D ]R }j j j |d D ]' \ } }|j j | || ) T j j d || j y y )
Ng?AGGREGATED_EVENTS)EMITJOINLEAVE
DISCONNECTMJOINMLEAVE MDISCONNECTr;
ALL_EVENTSconfigsupportedFeaturestor7 all_sockets) r) sleepr, r1 appendr< r+ r. get_namespacesget_participantsserialize_socket)sidfeaturesry nspeio_sidr s r ru z0InstrumentedServer.admin_connect..config s( HHNN3 ,,H>> 6 6yyM) - HHMM(%8($C D,@,@
B yyM) 88++::< FC(,(8(8(I(I )' F W#** 1 1#sG DFFF
m[S(,(<(< > *r )r*
isinstancedictlistr r) start_background_task)r r environ client_auth
authenticatedru s ` r rD z InstrumentedServer.admin_connect s 99 !M$))T* +tyy 8
DIIt, +tyy 8
$ + 6
,-DEE >0
&&vs3r c B | j j |||| y )Nrw )r) r< )r _r7 room_filtereventdatas r rE z InstrumentedServer.admin_emit s
eTkY
Gr c | j j j || D ]# \ }}| j j ||| % y Nr: )r) r. r}
enter_roomr r r7 roomr r s r rF z#InstrumentedServer.admin_enter_room J hh&&77;( @FC HH TY ? @r c | j j j || D ]# \ }}| j j ||| % y r )r) r. r}
leave_roomr s r rG z#InstrumentedServer.admin_leave_room r r c | j j j || D ]" \ }}| j j || $ y r )r) r. r} r@ )r r r7 closer r s r rH z#InstrumentedServer.admin_disconnect sF hh&&77;( :FC HH y 9 :r c | j r5| j j | j j y y r )r5 r4 setr= r
s r shutdownzInstrumentedServer.shutdown s0 ?? ! ! % % ' OO " r c
t j }|d }|dk( r| j j j || }|| j j j |< | j ||| }| j j
d|t j |t j j f| j n|dk( r| j j j |= |d }| j j
d|||t j |t j j f| j nc| j j
d|||g|dd t j |t j j f| j | j j ||g| S ) Nr r9 socket_connectedr: r@ r socket_disconnectedevent_received)r r) r. eio_sid_from_sidrI r~ r< r
fromtimestampr utc isoformatr+ rK ) r r r7 argstr r serialized_socketreasons r rJ z!InstrumentedServer._trigger_event s IIK1g I hh&&77YGG01DHH ( ( - $ 5 5c9g N HHMM,!&&q(,,7AAC/ --
/ l
" ,,S1!WF HHMM/&&q(,,7AAC 2
--
/
HHMM*"ab"&&q(,,7AAC -
--
/ (txx''y@4@@r c 4 t d D ]{ }| j j d | j j j | j
r.| j j
d||dd| j y } y # t $ r Y w xY w)N socket_updated websocket)idr transportr: ) ranger) rz rT _get_socketupgradedr< r+ KeyError)r r r r7 r s r _check_for_upgradez%InstrumentedServer._check_for_upgrade s q A HHNN1
88<<++G4==HHMM"2 ! (%05 "&!5!5 " 7
>
s AB BBc | j j j |||| }|r[| j j d|||t j
t j j f| j |S )N room_joinedr: )
r) r. rM r< r nowr r r r+ )r r r7 r r rets r rN z$InstrumentedServer._basic_enter_room sx hh 11#y$29; HHMM- X\\*446 *
--
/
r c
|r[| j j d|||t j t j
j
f| j | j j j ||| S )N room_leftr: )
r) r< r r r r r r+ r. rP )r r r7 r s r rQ z$InstrumentedServer._basic_leave_room so HHMM+ X\\*446 (
--
/ xx 223 4HHr c
| j j j |||f|||d|}|| j k7 rt |t
r|gt
| z n||g} t |t s|g}| j j j || D ]e \ }
} |
|vs | j j d||
| t j t j j f| j g |S )N)r skip_sidcallback
event_sentr: )r) r. rR r+ r tupler r} r< r r r r r ) r r r r7 r r r kwargsr
event_datar r s r rS zInstrumentedServer._emit s %dhh %%eT9 04/7(0(.0 ,, ,1;D%1H%4:-T]
h- $: ((**;;ItL
7Qh&HHMM,!" X\\2<<> 1
"&!5!5 " 7
7
r c . | j S| j j j | _ | j j | j
| _ | j j d | j j || S )N
rawConnection)
r4 r) rT create_eventr _emit_server_statsr5 r3 r rU )r r r s r rU z&InstrumentedServer._handle_eio_connect ss ($(HHLL$=$=$?D ! "hh < <'' )DO
/xx++GW==r c p | j j d | j j || S )NrawDisconnection)r3 r r) rV )r r r s r rV z)InstrumentedServer._handle_eio_disconnect$ s. 12xx..w??r c | j j j ||| }| j j d | j j dt |d |S )N)packetsheaders jsonp_index
packetsOutbytesOutresponse)r) rT rY r3 r len)r r r r r s r rZ z%InstrumentedServer._eio_http_response( s^ hhll ,7 9 |, z3s:+?@
r c | j | }|j j d |j j dt |j dd |S )N packetsInbytesInCONTENT_LENGTHr )r\ r3 r r get)socketr r r s r r` z+InstrumentedServer._eio_handle_post_request/ sW *f**73 {+ s7;;'7;< >
r c fd}fd}|j |_ t j || |_ |j |_ t j || |_ | j | S )Nc j j d j j dt | | j |g|i |S )Nr r )r3 r r _InstrumentedServer__send)wsr r r r s r _sendz8InstrumentedServer._eio_websocket_handler.._send7 sL " "< 0 " ":s4y 9 299T3D3F3 3r c | j }j j d j j dt |xs d |S )Nr r )_InstrumentedServer__waitr3 r r )r r r s r _waitz8InstrumentedServer._eio_websocket_handler.._wait< sE ))+C " "; / " "9c#)n =Jr )sendr r] partialwaitr rb )r r r r r s ` r rc z)InstrumentedServer._eio_websocket_handler6 s` 4
GG ##E2.GG ##E2.)v))"--r c | j }t j }|j j j D ] }|j j j || }|s,|j
||| }|j j d|t j |t j j f|j | j S )Nr r: )r r r) r. r| sid_from_eio_sidr~ r< r r r r r r+ re )r r r r r7 r r s r rf z!InstrumentedServer._eio_send_pingH s ** IIK ))88: 3I((""33GYGC$($9$9#y:A%C!
0% **1hll;EEG3 " 1 1 3 3 "v!!##r c t j }t | j j j }|j | j j s| j j | j | j j d| j t t t j |z
t | j j j t | j j j j# D cg c] }|j$ s| c} | j& j) |D cg c]P }|t | j j* j, j/ |d g i j/ d g dR c}d| j0 | j j sy y c c}w c c}w )N server_stats)name socketsCount)serverIdhostnamepiduptime clientsCountpollingClientsCountaggregatedEvents
namespacesr: )r r r) handlerskeyssortr4 is_setrz r2 r< r0 r/ PIDr rT socketsr r r3 r r. roomsr r+ )r
start_timer sr s r r z%InstrumentedServer._emit_server_statsV s| YY[
$((++0023
''..0 HHNN455 6 HHMM. NN $ ))+
2 #DHHLL$8$8 9'* $ 4 4 ; ; = (1 JJ (() %)$5$5$C$C$E
( ) $'(8(8(>(>(B(BdBZ))),T2 %8 )+ --
/ ''..00( )s
0G0(AG5c |%| j j j | }| j j j | }| j j
j
|i }|| j j j v r#| j j j | nd}|||j rdnd|i |j
dd |j D ci c]* \ }}|j d r|dd j |, c}}t |j
dd j D ci c] \ }}|t | d k( r|d n| c}}|j
d
d d k( |j
d d |d
z |r2t j |t j" j% ndd| j j j' || dS c c}}w c c}}w )Nr r polling REMOTE_ADDRr HTTP_r QUERY_STRINGr zwsgi.url_schemehttps PATH_INFOr )addressr querysecureurlissuedr )r clientIdr r r handshaker )r) r. r rT r r r rI r items
startswithlowerr r r r r r r get_rooms) r r r7 r r r tmkvs r r~ z#InstrumentedServer.serialize_socketm s ?hh&&77DAq!SVq[QqTa7 >!++&7<G {{;3 t) !..r8<<@JJL XX%%//Y?%
6>s 1/G: H )Nr; FNz/admin r )NNN)r r! r" r r6 ri rD rE rF rG rH r rJ r rN rQ rS rU rV rZ r` rc rf r r~ r# r r r% r% # s EJKL495v3 $4LH@
@
:
#
A:
I AE(>@.$ $ /.
r r% )r r r] osr r urllib.parser
exceptionsr gethostnamer/ getpidr r r% r# r r r sM '
! . 6 biik4 4,d
d
r