(z* d Z ddlZddlZddlmZ ddlmZ ddlmZm Z m
Z
ddl Z ddl m
Z
mZ ej e Z G d de Z G d d
Z G d d e ZdS )
u
This program is free software: you can redistribute it and/or modify it under
the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License,
or (at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
See the GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program. If not, see .
Copyright © 2019 Cloud Linux Software Inc.
This software is also available under ImunifyAV commercial license,
see
N)StrEnum)Path)AnyDictList) check_run
nice_iteratorc e Zd ZdZdZdS ) PatchmanCustomIntegrationTypefilescriptN)__name__
__module__ __qualname__FILESCRIPT U/opt/imunify360/venv/lib/python3.11/site-packages/imav/patchman/custom_integration.pyr r # s D
FFFr r c e Zd ZdZdddddddddddddddddg ddd d
ddd Zd ed
efdZed Zed Z de
de
fdZ de
fdZ de
eeef fdZdS )PatchmanCustomIntegrationa`
Patchman custom integration uses pre-defined user files that can be
either files with info about users (file integration) or scripts (script
integration). This class is designed to retrieve the users and provide them
to the DB the same way panels do.
More info:
https://docs.imunify360.com/Patchman_custom_integration.pdf
stringT)typerequired)r r nullablebooleaninteger) )r allowedr dict)homediremailparentlanguage suspendedleveldomainsintegration_type
metadata_pathc " || _ || _ d S N)_integration_type_metadata_path)selfr* r+ s r __init__z"PatchmanCustomIntegration.__init__= s !1 +r c * t | j S r- )r r. r0 s r r* z*PatchmanCustomIntegration.integration_typeA s ,T-CDDDr c * t | j S r- )r r/ r3 s r r+ z'PatchmanCustomIntegration.metadata_pathE s D'(((r
users_datareturnc t j | j }i }| D ]A\ }}| | s"t
d||j <|||< B|S )z0
Return only valid *users_data*
z Skip not valid data for '%s': %s)cerberus ValidatorUSER_DATA_SCHEMAitemsvalidateloggerwarningerrors)r0 r5 validatorresultusernamedatas r r< z"PatchmanCustomIntegration.validateI s &t'<== ( . . 0 0 $ $NHd %%d++
6 $
#F8
r c K i }| j t j k rx| j st d t
t | j g d {V }t j |
}n,| j st d t | j
d d 2 3 d {V } | d 5 }t j | }d d d n # 1 swxY w Y n=# t j $ r+ t" dt | Y w xY wt' | d k r)t" d
t | | | 6 | | S ) Nz$Provided metadata path is not a filez)Provided metadata path is not a directoryz*.json2 )
chunk_sizezutf-8)encodingzSkip invalid JSON file: %sr z(Skip %s: expect one JSON object per file)r* r r r+ is_file PatchmanCustomIntegrationErrorr strjsonloadsdecodeis_dirr globopenloadJSONDecodeErrorr= r> lenupdater< )r0 r5 rA user_data_filefrC s r _get_users_dataz)PatchmanCustomIntegration._get_users_dataZ s<
$A$H H H%--//
4: %c$*<&=&=%>????????F FMMOO44JJ %,,..
4? )6"''11b) ) )
(
(
(
(
(
(
(n ',,g,>> ,! #y||, , , , , , , , , , , , , , ,+ NN4c.6I6I H
t99>>NNBN++ !!$''''#)$ }}Z(((s< #F=+D.D" D."D& &D.)D& *D..7E('E(c \ K g }| d {V }| D ]S\ }}| di }d | D }| ||d| T|S # t $ r t
$ r
}t |d }~ww xY w)Nr) c g | ]
\ }}||d S ))domainpathsr ).0 domain_namer[ s r
z. s4 * U +U;; r )rB r) )rW r; popappendrI Exception)r0 users
metadata_dictrB details orig_domainsr) es r patchman_usersz(PatchmanCustomIntegration.patchman_users s 8E"&"6"6"8"8 8 8 8 8 8 8M%2%8%8%:%:
!' &{{9b 9 9 .:.@.@.B.B !)gIII L- 8 8 80a 7 8s BB
B+ B&&B+N)r r r __doc__r: rJ r1 propertyr* r+ r" r< rW r r r rg r r r r r ( sC %$77"55#4HH %488 'T::# tLL "55 , ,S , , , , E E XE ) ) X)4 D "#)t #) #) #) #)J8d4S>&: 8 8 8 8 8 8r r c e Zd ZdS )rI N)r r r r r r rI rI s Dr rI )rh rK loggingenumr pathlibr typingr r r r8 defence360agent.utilsr r getLoggerr r= r r ra rI r r r rq s' * " " " " " " " " " " : : : : : : : : 8 $ $ G
i8 i8 i8 i8 i8 i8 i8 i8X Y r