j8wȀ N d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZ ej d
Z ej d
Z ej d
Z
ej d
Z ej e Z G d de Z G d d
e Z G d d ZeZ G d
d Z e Zd Z G d d Z G d de Z ddZ dS ) N )minutes)hours)days)weeksc e Zd ZdZd Zd Zd Zd Zd Zd Z G d d e
Z G d
d e Z G d d
e Z
G d de ZdS ) ServiceBasezBase service class.c d || _ d| _ d | _ | | | _ d S NF)_loop _should_stop
_main_task StoppedState_state)selfloops Q/opt/imunify360/venv/lib/python3.11/site-packages/defence360agent/utils/common.py__init__zServiceBase.__init__ s1
! ''-- c 4 | j S N)r startr s r r zServiceBase.start s { """r c 4 | j S r )r should_stopr s r r zServiceBase.should_stop s {&&(((r c D K | j d {V S r )r waitr s r r zServiceBase.wait" s, [%%'''''''''r c 4 | j S r )r
is_runningr s r r zServiceBase.is_running% s {%%'''r c K t r )NotImplementedErrorr s r _runzServiceBase._run( s !!r c , e Zd Zd Zd Zd Zd Zd ZdS )ServiceBase.Statec || _ dS )z:type obj: ServiceBaseN)_objr objs r r zServiceBase.State.__init__, s
DIIIr c d S r r s r r zServiceBase.State.start0 Dr c d S r r* r s r r z ServiceBase.State.should_stop3 r+ r c : K | j j }|r
| d {V d S d S r )r& r )r tasks r r zServiceBase.State.wait6 s7 9'D
r c dS r r* r s r r z ServiceBase.State.is_running; s 5r N)__name__
__module__ __qualname__r r r r r r* r r Stater$ + s_
r r3 c e Zd Zd Zd ZdS )ServiceBase.StoppedStatec p t | j | j _ d| j _ d S r )r r r& r r
)r futures r _on_stopz!ServiceBase.StoppedState._on_stop? s, * 7 7 B BDI %*DI " " "r c | j }|j | |_ |j | j t | |_ d S r )
r& r create_taskr" r add_done_callbackr8 r RunningStater r' s r r z ServiceBase.StoppedState.startC sY )C Y 2 2388:: > >CN N , ,T] ; ; ;$11#66CJJJr N)r0 r1 r2 r8 r r* r r r r5 > s2 + + + 7 7 7 7 7r r c e Zd Zd Zd ZdS )ServiceBase.RunningStatec | j }d|_ |j t | |_ d S NT)r& r
r cancelr
StoppingStater r' s r r z$ServiceBase.RunningState.should_stopJ s> )C #C N ! ! # # #$22377CJJJr c dS r@ r* r s r r z#ServiceBase.RunningState.is_runningP s 4r N)r0 r1 r2 r r r* r r r< r> I s2 8 8 8 r r< c e Zd Zd ZdS )ServiceBase.StoppingStatec t d )Nz9Cannot start stopping service. Please wait while it stop.)ProgrammingErrorr s r r z ServiceBase.StoppingState.startT s "K
r N)r0 r1 r2 r r* r r rB rE S s# r rB N)r0 r1 r2 __doc__r r r r r r" objectr3 r r< rB r* r r r r s . . . # # #) ) )( ( (( ( (" " " & 7 7 7 7 7u 7 7 7 u r r c e Zd ZdS )rG N)r0 r1 r2 r* r r rG rG Z s Dr rG c H e Zd ZdZej fdddZed Zd Z dS ) RateLimita3 Decorator to limit function calls to one per *period* seconds.
If less than *period* seconds have passed since the last call,
then the request to call the function is replace with an *on_drop*
call with the same arguments.
If *on_drop* is None [default] then the call is just dropped
N)on_dropc > d | _ || _ || _ || _ d S r )_next_call_time_period_timer_on_drop)r periodtimerrM s r r zRateLimit.__init__i s# #
r c N | j d u p | j | k S r )rO rQ r s r should_be_calledzRateLimit.should_be_calledo s,
D (
5#t{{}}4
r c t j fd }t j fd }t j r|n|S )Nc j r) j z _ | i |S j
j | i |S d S r rV rQ rP rO rR argskwargsfuncr s r wrapperz#RateLimit.__call__..wrapperw sa $
6'+{{}}t|'C$tT,V,,, *$t}d5f555 +*r c K j r/ j z _ | i | d {V S j
j | i |S d S r rY rZ s r
async_wrapperz)RateLimit.__call__..async_wrapper sw $
6'+{{}}t|'C$ !T4 26 2 22222222 *$t}d5f555 +*r ) functoolswrapsasyncioiscoroutinefunction)r r] r^ r` s `` r __call__zRateLimit.__call__v s 6 6 6 6 6
6
6 6 6 6 6
6 !( ;D A AN}}wNr )
r0 r1 r2 rH time monotonicr propertyrV re r* r r rL rL ^ st &*^
X
O O O O Or rL c e Zd Zd ZdddZdS )
CoalesceCallsc < t d | _ d | _ d S )Nz-inf)float call_time delayed_callr s r r zCoalesceCalls.__init__ s v r N)
done_callbackc fd}|S )a
Decorator to coalesce coroutine calls to one per *period* seconds.
Requests for a coroutine call in a given time period are coalesced:
If t is the time of the last call, then N call requests in the [t,
t+period) time interval results in a single call at the
t+period time iff N>0 i.e.,
if less than *period* seconds have passed since the last call,
then the calls are coalesced: (N-1) requests are dropped, Nth
requests is performed in *period* seconds.
It is unspecified which exact call is made if arguments differ.
If the call is not dropped then *done_callback* is attached
to the task when the coroutine is scheduled with the event loop.
Given `c` is the time of the last [actual] call (`loop.create_task()`)
And `T` is the coalesce time period
When a call request arrives at `t` time
Then
| call pending? | t>c+T | c<=t<=c+T | t.decorator..wrapper..log_exception s} >>++ 0@0@0L 33+H"+,,-1^^-=-=(, 0L0Lr c t d _ d_ | |i | }| n dS )z*Call & schedule the delayed coroutine now.zSchedule call %sN)loggerinforf rm rn r: r; ) coror[ r\ r. r| ro r} r r s r call_delayedzVCoalesceCalls.coalesce_calls..decorator..wrapper..call_delayed s KK 2I>>>%)YY[[DN(,D% ++DD$,A&,A,ABBD**(0 &
* r ziThere was a scheduled call (%s) but more than period (%r) seconds passed since the last call (%r, now=%r)zSSatisfy the call request soon: %s. No calls in more than %r seconds since the startz`Drop call request for %s, enforcing one call per %r seconds limit. Next call is in ~%.2f secondszQDelay call request: %s for ~%.2f seconds. Enforcing one call per %r seconds limitzNDrop call request for %s, reason: last call time (%r, now=%r) is in the future)getrc get_event_loopr0 rf rm rn strrA r warningr call_sooncall_at)r[ r\ args_reprr nowold_delayed_call_reprdelayr| r} r r ro rS r s @@@r r^ z@CoalesceCalls.coalesce_calls..decorator..wrapper s zz&))<"133D #6 #/3ttVV & " !
$ 0 8 8 8 H % ! " -1LL NV 3 ( " - -))) NN9! r )ra rb )r r^ ro rS r s ` r decoratorz/CoalesceCalls.coalesce_calls..decorator sQ
_T
"
"c
c
c
c
c
c
c
#
"c
J Nr r* )r rS ro r s ``` r coalesce_callsz CoalesceCalls.coalesce_calls s8 8g g g g g g g R r )r0 r1 r2 r r r* r r rj rj sH ! ! ! 7; E E E E E E Er rj c t j } | '| d rt j S | S )ziReturns readable name of the server.
It is sent to CLN and allows user to sort out his servers.
N localhost)socketgetfqdnlower
startswith gethostname)hostnames r get_hostnamer sH
~ H8>>++66{CC!### Or c > e Zd ZdZd
dZd Zd Zd Zd Zd Z d Z
dS ) VersionzAbstract base class for version numbering classes. Just provides
constructor (__init__) and reproducer (__repr__), because those
seem to be the same for all version numbering classes; and route
rich comparisons to _cmp.
Nc 8 |r| | d S d S r )parse)r vstrings r r zVersion.__init__3 s, JJw r c \ d | j j t | S )Nz {} ('{}'))format __class__r0 r r s r __repr__zVersion.__repr__7 s# !!$."93t99EEEr c N | | }|t u r|S |dk S Nr _cmpNotImplementedr othercs r __eq__zVersion.__eq__: , IIe HAv
r c N | | }|t u r|S |dk S r r r s r __lt__zVersion.__lt__@ , IIe H1u r c N | | }|t u r|S |dk S r r r s r __le__zVersion.__le__F r r c N | | }|t u r|S |dk S r r r s r __gt__zVersion.__gt__L r r c N | | }|t u r|S |dk S r r r s r __ge__zVersion.__ge__R r r r ) r0 r1 r2 rH r r r r r r r r* r r r r , s F F F r r c V e Zd ZdZ ej dej Zd Zd Z d Z
d Z dS ) LooseVersiona Version numbering for anarchists and software realists.
Implements the standard interface for version number classes as
described above. A version number consists of a series of numbers,
separated by either periods or strings of letters. When comparing
version numbers, the numeric components will be compared
numerically, and the alphabetic components lexically. The following
are all valid version numbers, in no particular order:
1.5.1
1.5.2b2
161
3.10a
8.02
3.4j
1996.07.12
3.2.pl0
3.1.1.6
2g6
11g
0.960923
2.2beta29
1.13++
5.5.kw
2.0b1pl0
In fact, there is no such thing as an invalid version number under
this scheme; the rules for comparison are simple and predictable,
but may not always give the results you want (for some definition
of "want").
z(\d+ | [a-z]+ | \.)c || _ d | j | D }t | D ](\ }} t | ||< # t
$ r Y %w xY w|| _ d S )Nc " g | ] }|r|d k
|
S ).r* ).0xs r
z&LooseVersion.parse.. s,
1
ABcAr )r component_resplit enumerateint
ValueErrorversion)r r
componentsir( s r r zLooseVersion.parse} s
(..w77
++ FAs
#C
1
" s A
A! A!c | j S r )r r s r __str__zLooseVersion.__str__ s
|r c & dt | z S )NzLooseVersion ('%s'))r r s r r zLooseVersion.__repr__ s $s4yy00r c t |t rt | }n t |t st S | j |j k rdS | j |j k rdS | j |j k rdS d S )Nr r )
isinstancer r r r )r r s r r zLooseVersion._cmp s eS ! ! " ''EEE<00 "! ! <5= ( (1 <%- ' '2 <%- ' '1 ( 'r N) r0 r1 r2 rH recompileVERBOSEr r r r r r* r r r r Z sr > 2: 4bjAAL" " " 1 1 1 r r c R t j | \ }}t j dk r |dk r| dz } t j | r| S |[t j dd }|9 t j d }n## t t f$ r t j }Y nw xY w|sdS | t j
}|D ]E}t j || }t j | r|c S FdS )zTries to find 'executable' in the directories listed in 'path'.
A string listing directories separated by 'os.pathsep'; defaults to
os.environ['PATH']. Returns the complete filename or None if not found.
win32z.exeNPATHCS_PATH)ospathsplitextsysplatformisfileenvironr confstrAttributeErrorr defpathr pathsepjoin)
executabler _extpathspfs r find_executabler s) W
j
)
)FAs cVmm &(
w~~j!! |z~~fd++ <
"z),,"J/
"
"
"z
" t JJrz " "E
GLLJ ' '
7>>! HHH 4s B B98B9r ) rc datetimera loggingr rf r r r timedelta
total_secondsMINUTEHOURDAYWEEK getLoggerr0 r rI r ExceptionrG rL
rate_limitrj webserver_gracefull_restartr r r r r* r r r sL
A & & & 4 4 6 6x"""0022ha ..00x"""0022 8 $ $D D D D D& D D DN y )O )O )O )O )O )O )O )OX
J J J J J J J JZ ,moo * * * * * * * *\D D D D D7 D D DP" " " " " "r