JFIF $
d d 7
Viewing File: /opt/imunify360/venv/lib/python3.11/site-packages/defence360agent/simple_rpc/schema.py
from cerberus.errors import BaseErrorHandler, BasicErrorHandler
from defence360agent.contracts.config import UserType
from defence360agent.rpc_tools.middleware import (
add_eula,
add_license,
add_license_user,
add_version,
collect_warnings,
counts,
default_to_items,
max_count,
preserve_remote_addr,
send_command_invoke_message,
)
from defence360agent.rpc_tools.utils import prepare_schema
class ErrorHandler(BaseErrorHandler):
messages = BasicErrorHandler.messages.copy()
def collect_errors(self, error):
if error.child_errors:
for err in error.child_errors:
yield from self.collect_errors(err)
else:
# avoid abstract error: required field
yield "field: '{}', value: '{}', error: {}".format(
error.field,
error.value,
self.messages.get(error.code, "").format(
*error.info,
constraint=error.constraint,
field=error.field,
value=error.value
),
)
def __call__(self, errors):
string_representation = []
for error in errors:
for info in self.collect_errors(error):
string_representation.append(info)
return string_representation
def init_validator(schema_validator, validate_middleware, schema_paths):
_validator = schema_validator(
prepare_schema(schema_paths),
error_handler=ErrorHandler,
)
# NOTE: it is processed in the reversed order, see _apply_middleware
_middleware = {
None: [
(send_command_invoke_message, (UserType.ROOT, UserType.NON_ROOT)),
# validation before processing the data
(
validate_middleware(_validator),
(UserType.ROOT, UserType.NON_ROOT),
),
# inject license for root
(add_license, (UserType.ROOT,)),
# inject license for regular user
(add_license_user, (UserType.NON_ROOT,)),
# inject eula
(add_eula, (UserType.ROOT,)),
# inject version
(add_version, (UserType.ROOT, UserType.NON_ROOT)),
# add warnings if any
(collect_warnings, (UserType.ROOT, UserType.NON_ROOT)),
# for backward compatibility
(default_to_items, (UserType.ROOT, UserType.NON_ROOT)),
],
("whitelist", "ip", "list"): [
(counts, (UserType.ROOT, UserType.NON_ROOT))
],
("blacklist", "ip", "list"): [
(max_count, (UserType.ROOT, UserType.NON_ROOT))
],
("graylist", "ip", "list"): [
(counts, (UserType.ROOT, UserType.NON_ROOT))
],
("whitelist", "ip", "add"): [
(preserve_remote_addr, (UserType.ROOT, UserType.NON_ROOT))
],
("blacklist", "ip", "add"): [
(preserve_remote_addr, (UserType.ROOT, UserType.NON_ROOT))
],
("whitelist", "country", "list"): [
(max_count, (UserType.ROOT, UserType.NON_ROOT))
],
("blacklist", "country", "list"): [
(max_count, (UserType.ROOT, UserType.NON_ROOT))
],
("graylist", "country", "list"): [
(max_count, (UserType.ROOT, UserType.NON_ROOT))
],
("blacklist",): [(counts, (UserType.ROOT, UserType.NON_ROOT))],
("whitelist",): [(counts, (UserType.ROOT, UserType.NON_ROOT))],
("whitelisted-crawlers", "list"): [
(max_count, (UserType.ROOT, UserType.NON_ROOT))
],
("blocked-port", "list"): [
(counts, (UserType.ROOT, UserType.NON_ROOT))
],
("blocked-port-ip", "list"): [
(max_count, (UserType.ROOT, UserType.NON_ROOT))
],
("rules", "list-disabled"): [
(max_count, (UserType.ROOT, UserType.NON_ROOT))
],
("proactive", "ignore", "list"): [
(max_count, (UserType.ROOT, UserType.NON_ROOT))
],
("feature-management", "show"): [
(max_count, (UserType.ROOT, UserType.NON_ROOT))
],
("ip-list", "synced"): [(counts, (UserType.ROOT, UserType.NON_ROOT))],
("ip-list", "local", "list"): [
(counts, (UserType.ROOT, UserType.NON_ROOT))
],
("ip-list", "local", "add"): [
(preserve_remote_addr, (UserType.ROOT, UserType.NON_ROOT))
],
("ip-list", "local", "delete"): [
(preserve_remote_addr, (UserType.ROOT, UserType.NON_ROOT))
],
}
_middleware_exclude = {
("enable-plugin",): [add_eula],
("disable-plugin",): [add_eula],
("switch-max-webserver",): [add_eula],
("install-vendors",): [add_eula],
("uninstall-vendors",): [add_eula],
("add-sudouser",): [add_eula],
("delete-sudouser",): [add_eula],
("doctor",): [add_eula],
("captcha", "update-localizations"): [add_eula],
("captcha", "compile-localizations"): [add_eula],
("update",): [add_eula],
("kcarectl", "disable-auto-update"): [add_eula],
("kcarectl", "enable-auto-update"): [add_eula],
("kcarectl", "plugin-info"): [add_eula],
("register",): [add_eula],
("unregister",): [add_eula],
("rstatus",): [add_eula],
("update-license",): [add_eula],
("3rdparty", "list"): [add_eula],
("admin-emails",): [add_eula],
("list-docroots",): [add_eula],
("features", "list"): [add_eula],
("features", "status"): [add_eula],
("features", "install"): [add_eula],
("features", "remove"): [add_eula],
("feature-management", "native", "enable"): [add_eula],
("feature-management", "native", "disable"): [add_eula],
("feature-management", "native", "status"): [add_eula],
("import", "wblist"): [add_eula],
("rules", "update-app-specific-rules"): [add_eula],
("support", "send"): [add_eula],
("3rdparty", "conflicts"): [add_eula],
("smtp-blocking", "reset"): [add_eula],
("smtp-blocking", "sync"): [add_eula],
("malware", "on-demand", "check-detached"): [add_eula],
("checkdb",): [add_eula],
("restore-configs",): [add_eula],
("patchman", "users"): [add_eula],
("patchman", "register"): [add_eula],
("patchman", "install"): [add_eula],
("patchman", "migrate"): [add_eula],
("patchman", "uninstall"): [add_eula],
("patchman", "status"): [add_eula],
("patchman", "install", "realtime"): [add_eula],
("patchman", "uninstall", "realtime"): [add_eula],
("analyst-cleanup", "request"): [add_eula],
("analyst-cleanup", "get-requests"): [add_eula],
("analyst-cleanup", "is-allowed"): [add_eula],
}
return _validator, _middleware, _middleware_exclude
Back to Directory
nL+D550H?Mx ,D"v]qv;6*Zqn)ZP 0 !1 A "#a$2Qr D8 a Ri[f\mIykIw0 cuFcRı?lO7к_f˓[C$殷WF<_W ԣs KcëIzyQy /_LKℂ;C",pFA: /]=H ~,ls/9ć:[=/#f;)x {ٛEQ )~ =𘙲r*2~ a _V=' kumF D}KYYC)({ *g&f`툪ry`=^cJ.I](*`wq 1dđ#̩͑0;H]u搂@:~וKL Nsh}OIR*8:2 !lDJVo( 3=M(zȰ+i*NA r6KnSl )!JJӁ* %݉?|D}d5:eP0R;{$X'xF@.ÊB {,WJuQɲRI;9QE琯62fT.DUJ;*cP A\ILNj!J۱+O\ ͔]ޒ SJȧc%ANolՎprULZԛerE2=XDXgVQeӓk yP7U*omQIs,K`)6\G3t?pgjrmۛجwluGtf h9uyP0D;Uڽ"OXlif$)&|ML0Zrm1[HXPlPR0'G=i2N +0e2]] 9VTPO7h(F*癈 '=QVZDF,d߬~TX
G[`le69CR(!S2!P
<0x < !1AQ "Raq02Br#SCTb
?Ζ"]mH5WR7k.ۛ!}Q~+yԏz|@T20S~Kek *zFf^2X*(@8r?CIuI|֓>^ExLgNUY+{.RѪ
τVYTD I62'8Y2 7'\TP.6d&˦@Vqi|8- OΕ]ʔ U=TL8=;6c| !qfF3aů&~$l}'NWUs$Uk^SV:U# 6w++s&r+nڐ{@29 gL
u"TÙM=6 (^"7r} =6YݾlCuhquympǦ
G jhsǜNlɻ}o7 #S6aw4!OSrD57%|?x>L
|/nD6?/8w#[)L7+6〼T ATg ! %5MmZ/c-{1_Je"|^$'O&ޱմ Trb$ w)R$&
N1EtdU3Uȉ1pM" N*(DN yd96.(jQ)X
5cQɎMyW? Q*!R>6=7)Xj5`J]e 8%t!+'!1Q 5 !1 AQaqё#2"0BRb ? Gt^## .llQT $v,,m 㵜5 ubV =sY+@d{N! dnO< .-B; _wJt6;QJd.Qc%p{ 1,sNDdFHI0ГoXшe黅XۢF:)[FGXƹ/w_cMeD,ʡcc. WDtA$j@:) -# u
c1<@ۗ9F)KJ-hpP]_x[qBlbp ʖw q"LFGd ƶ*s+ډ_Zc"? %t[IP 6J]#=ɺVvvCGsGh1 >)6|ey?Lӣm,4GWUi` ]uJVoV DG< SB 6ϏQ@ TiUly OU0kfV~~}SZ@*WUUi##; s/[ =!7}" WN]'(L! ~y5g9T̅JkbM' +s:S + B)v@Mj
e Cf jE 0Y\QnzG1д~Wo{T 9?`Rmyhsy3!HAD]mc1~2L Su7xT;j$`}4->L#vzŏ ILS ֭ T { rjGKC; bpU= -`BsK.SFw4Mq]ZdHS0)tLg