proper cidr validation server-side

This commit is contained in:
dekzter 2025-06-08 08:29:25 -04:00
parent 9f96529707
commit 789d29c97a
4 changed files with 78 additions and 12 deletions

View file

@ -1,19 +1,67 @@
# core/serializers.py
import json
import ipaddress
from rest_framework import serializers
from .models import UserAgent, StreamProfile, CoreSettings
from .models import UserAgent, StreamProfile, CoreSettings, NETWORK_ACCESS
class UserAgentSerializer(serializers.ModelSerializer):
class Meta:
model = UserAgent
fields = ['id', 'name', 'user_agent', 'description', 'is_active', 'created_at', 'updated_at']
fields = [
"id",
"name",
"user_agent",
"description",
"is_active",
"created_at",
"updated_at",
]
class StreamProfileSerializer(serializers.ModelSerializer):
class Meta:
model = StreamProfile
fields = ['id', 'name', 'command', 'parameters', 'is_active', 'user_agent', 'locked']
fields = [
"id",
"name",
"command",
"parameters",
"is_active",
"user_agent",
"locked",
]
class CoreSettingsSerializer(serializers.ModelSerializer):
class Meta:
model = CoreSettings
fields = '__all__'
fields = "__all__"
def update(self, instance, validated_data):
if instance.key == NETWORK_ACCESS:
errors = False
invalid = {}
value = json.loads(validated_data.get("value"))
for key, val in value.items():
cidrs = val.split(",")
for cidr in cidrs:
try:
ipaddress.ip_network(cidr)
except:
errors = True
if key not in invalid:
invalid[key] = []
invalid[key].append(cidr)
if errors:
# Perform CIDR validation
raise serializers.ValidationError(
{
"message": "Invalid CIDRs",
"value": invalid,
}
)
return super().update(instance, validated_data)