forked from uncloud/uncloud
43 lines
1.3 KiB
Python
43 lines
1.3 KiB
Python
import base64
|
|
|
|
from django.contrib.auth import get_user_model
|
|
from django.utils.translation import gettext_lazy as _
|
|
from rest_framework import serializers
|
|
|
|
from .models import *
|
|
from .services import *
|
|
|
|
class WireGuardVPNSerializer(serializers.ModelSerializer):
|
|
address = serializers.CharField(read_only=True)
|
|
vpn_server = serializers.CharField(read_only=True)
|
|
vpn_server_public_key = serializers.CharField(read_only=True)
|
|
network_mask = serializers.IntegerField()
|
|
|
|
class Meta:
|
|
model = WireGuardVPN
|
|
fields = [ 'wireguard_public_key', 'address', 'network_mask', 'vpn_server',
|
|
'vpn_server_public_key' ]
|
|
|
|
extra_kwargs = {
|
|
'network_mask': {'write_only': True }
|
|
}
|
|
|
|
|
|
def validate_wireguard_public_key(self, value):
|
|
msg = _("Supplied key is not a valid wireguard public key")
|
|
|
|
"""
|
|
FIXME: verify that this does not create broken wireguard config files,
|
|
i.e. contains \n or similar!
|
|
We might even need to be more strict to not break wireguard...
|
|
"""
|
|
|
|
try:
|
|
base64.standard_b64decode(value)
|
|
except Exception as e:
|
|
raise serializers.ValidationError(msg)
|
|
|
|
if '\n' in value:
|
|
raise serializers.ValidationError(msg)
|
|
|
|
return value
|