jumpserver/apps/assets/const/protocol.py

119 lines
3.1 KiB
Python

from django.db import models
from common.db.models import ChoicesMixin
__all__ = ['Protocol']
class Protocol(ChoicesMixin, models.TextChoices):
ssh = 'ssh', 'SSH'
rdp = 'rdp', 'RDP'
telnet = 'telnet', 'Telnet'
vnc = 'vnc', 'VNC'
mysql = 'mysql', 'MySQL'
mariadb = 'mariadb', 'MariaDB'
oracle = 'oracle', 'Oracle'
postgresql = 'postgresql', 'PostgreSQL'
sqlserver = 'sqlserver', 'SQLServer'
clickhouse = 'clickhouse', 'ClickHouse'
redis = 'redis', 'Redis'
mongodb = 'mongodb', 'MongoDB'
k8s = 'k8s', 'K8S'
http = 'http', 'HTTP'
_settings = None
@classmethod
def device_protocols(cls):
return {
cls.ssh: {
'port': 22,
'secret_types': ['password', 'ssh_key'],
'setting': {
'sftp_enabled': True,
'sftp_home': '/tmp',
}
},
cls.rdp: {
'port': 3389,
'secret_types': ['password'],
'setting': {
'console': True,
'security': 'any',
}
},
cls.vnc: {
'port': 5900,
'secret_types': ['password'],
},
cls.telnet: {
'port': 23,
'secret_types': ['password'],
},
}
@classmethod
def database_protocols(cls):
return {
cls.mysql: {
'port': 3306,
'secret_types': ['password'],
'setting': {
}
},
cls.mariadb: {
'port': 3306,
'secret_types': ['password'],
},
cls.postgresql: {
'port': 5432,
'secret_types': ['password'],
},
cls.oracle: {
'port': 1521,
'secret_types': ['password'],
},
cls.sqlserver: {
'port': 1433,
'secret_types': ['password'],
},
cls.clickhouse: {
'port': 9000,
'secret_types': ['password'],
},
cls.mongodb: {
'port': 27017,
'secret_types': ['password'],
},
cls.redis: {
'port': 6379,
'secret_types': ['password'],
},
}
@classmethod
def cloud_protocols(cls):
return {
cls.k8s: {
'port': 443,
'secret_types': ['token'],
},
cls.http: {
'port': 80,
'secret_types': ['password'],
'setting': {
'username_selector': 'input[type=text]',
'password_selector': 'input[type=password]',
'submit_selector': 'button[type=submit]',
}
},
}
@classmethod
def settings(cls):
return {
**cls.device_protocols(),
**cls.database_protocols(),
**cls.cloud_protocols()
}