github
/
jumpserver
mirror of https://github.com/jumpserver/jumpserver
1
0
Fork 0
Code Issues Releases Wiki Activity

feat: rdp协议新增username表达式，匹配更多特殊字符

pull/7594/head v2.17.2
halo 2021-12-20 18:24:16 +08:00 committed by Jiangjie.Bai
parent c8b049349c
commit 5adbbeb868
2 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
apps/assets/serializers/system_user.py
View File

@ -4,7 +4,7 @@ from django.db.models import Count
from common.mixins.serializers import BulkSerializerMixin from common.mixins.serializers import BulkSerializerMixin
from common.utils import ssh_pubkey_gen from common.utils import ssh_pubkey_gen
from common.validators import alphanumeric_re, alphanumeric_cn_re from common.validators import alphanumeric_re, alphanumeric_cn_re, alphanumeric_win_re
from orgs.mixins.serializers import BulkOrgResourceModelSerializer from orgs.mixins.serializers import BulkOrgResourceModelSerializer
from ..models import SystemUser, Asset from ..models import SystemUser, Asset
from .utils import validate_password_contains_left_double_curly_bracket from .utils import validate_password_contains_left_double_curly_bracket
@ -107,9 +107,12 @@ class SystemUserSerializer(AuthSerializerMixin, BulkOrgResourceModelSerializer):
def validate_username(self, username): def validate_username(self, username):
protocol = self.get_initial_value("protocol") protocol = self.get_initial_value("protocol")
if username: if username:
regx = alphanumeric_re
if protocol == SystemUser.Protocol.telnet: if protocol == SystemUser.Protocol.telnet:
regx = alphanumeric_cn_re regx = alphanumeric_cn_re
elif protocol == SystemUser.Protocol.rdp:
regx = alphanumeric_win_re
else:
regx = alphanumeric_re
if not regx.match(username): if not regx.match(username):
raise serializers.ValidationError(_('Special char not allowed')) raise serializers.ValidationError(_('Special char not allowed'))
return username return username

2
apps/common/validators.py
View File

@ -17,6 +17,8 @@ alphanumeric_re = re.compile(r'^[0-9a-zA-Z_@\-\.]*$')
alphanumeric_cn_re = re.compile(r'^[0-9a-zA-Z_@\-\.\u4E00-\u9FA5]*$') alphanumeric_cn_re = re.compile(r'^[0-9a-zA-Z_@\-\.\u4E00-\u9FA5]*$')
alphanumeric_win_re = re.compile(r'^[0-9a-zA-Z_@#%&~\^\$\-\.\u4E00-\u9FA5]*$')
class ProjectUniqueValidator(UniqueTogetherValidator): class ProjectUniqueValidator(UniqueTogetherValidator):
def __call__(self, attrs, serializer): def __call__(self, attrs, serializer):