mirror of https://github.com/jumpserver/jumpserver
68 lines
2.4 KiB
Python
68 lines
2.4 KiB
Python
import copy
|
|
|
|
from django.utils.translation import gettext_lazy as _
|
|
from rest_framework import serializers
|
|
|
|
from accounts.const import SecretType, DEFAULT_PASSWORD_RULES
|
|
from common.utils import ssh_key_gen, random_string
|
|
from common.utils import validate_ssh_private_key, parse_ssh_private_key_str
|
|
|
|
|
|
class SecretGenerator:
|
|
def __init__(self, secret_strategy, secret_type, password_rules=None):
|
|
self.secret_strategy = secret_strategy
|
|
self.secret_type = secret_type
|
|
self.password_rules = password_rules
|
|
|
|
@staticmethod
|
|
def generate_ssh_key():
|
|
private_key, __ = ssh_key_gen()
|
|
return private_key
|
|
|
|
def generate_password(self):
|
|
password_rules = self.password_rules
|
|
if not password_rules or not isinstance(password_rules, dict):
|
|
password_rules = {}
|
|
rules = copy.deepcopy(DEFAULT_PASSWORD_RULES)
|
|
rules.update(password_rules)
|
|
rules = {
|
|
'length': rules['length'],
|
|
'lower': rules['lowercase'],
|
|
'upper': rules['uppercase'],
|
|
'digit': rules['digit'],
|
|
'special_char': rules['symbol']
|
|
}
|
|
return random_string(**rules)
|
|
|
|
def get_secret(self):
|
|
if self.secret_type == SecretType.SSH_KEY:
|
|
secret = self.generate_ssh_key()
|
|
elif self.secret_type == SecretType.PASSWORD:
|
|
secret = self.generate_password()
|
|
else:
|
|
raise ValueError("Secret must be set")
|
|
return secret
|
|
|
|
|
|
def validate_password_for_ansible(password):
|
|
""" 校验 Ansible 不支持的特殊字符 """
|
|
# validate password contains left double curly bracket
|
|
# check password not contains `{{`
|
|
# Ansible 推送的时候不支持
|
|
if '{{' in password:
|
|
raise serializers.ValidationError(_('Password can not contains `{{` '))
|
|
if '{%' in password:
|
|
raise serializers.ValidationError(_('Password can not contains `{%` '))
|
|
# Ansible Windows 推送的时候不支持
|
|
if "'" in password:
|
|
raise serializers.ValidationError(_("Password can not contains `'` "))
|
|
if '"' in password:
|
|
raise serializers.ValidationError(_('Password can not contains `"` '))
|
|
|
|
|
|
def validate_ssh_key(ssh_key, passphrase=None):
|
|
valid = validate_ssh_private_key(ssh_key, password=passphrase)
|
|
if not valid:
|
|
raise serializers.ValidationError(_("private key invalid or passphrase error"))
|
|
return parse_ssh_private_key_str(ssh_key, passphrase)
|