Server : ApacheSystem : Linux host.digitalbabaji.in 4.18.0-513.11.1.el8_9.x86_64 #1 SMP Wed Jan 17 02:00:40 EST 2024 x86_64 User : addictionfreeind ( 1003) PHP Version : 7.2.34 Disable Function : exec,passthru,shell_exec,system Directory : /opt/imunify360/venv/lib64/python3.11/site-packages/defence360agent/utils/ |
Upload File : |
"""JSON encoders to help with sending messages to server."""
import json
from ipaddress import IPv4Address, IPv4Network, IPv6Address, IPv6Network
from playhouse.shortcuts import model_to_dict
from defence360agent.model import Model
def ip_net_to_string(net) -> str:
"""
IPv4Network('192.168.1.1/32') -> '192.168.1.1'
IPv4Network('192.168.1.0/24') -> '192.168.1.0/24'
"""
if not int(net.hostmask):
return str(net.network_address)
return str(net)
class IPEncoder(json.JSONEncoder):
def default(self, obj):
if isinstance(obj, (IPv4Network, IPv6Network)):
return ip_net_to_string(obj)
if isinstance(obj, (IPv4Address, IPv6Address)):
return str(obj)
return json.JSONEncoder.default(self, obj)
class ServerJSONEncoder(IPEncoder):
def default(self, obj):
if isinstance(obj, Model):
return model_to_dict(obj)
return super().default(obj)