mirror of
https://github.com/jumpserver/jumpserver.git
synced 2026-01-19 00:55:18 +00:00
164f48e131
* [Update] 统一url地址 * [Update] 修改api * [Update] 使用规范的签名 * [Update] 修改url * [Update] 修改swagger * [Update] 添加serializer class避免报错 * [Update] 修改token * [Update] 支持api key * [Update] 支持生成api key * [Update] 修改api重定向 * [Update] 修改翻译 * [Update] 添加说明文档 * [Update] 修复浏览器关闭后session不失效的问题 * [Update] 修改一些内容 * [Update] 修改 jms脚本 * [Update] 修改重定向 * [Update] 修改搜索trim * [Update] 修改搜索trim * [Update] 添加sys log * [Bugfix] 修改登陆错误 * [Update] 优化User操作private_token的接口 (#3091) * [Update] 优化User操作private_token的接口 * [Update] 优化User操作private_token的接口 2 * [Bugfix] 解决授权了一个节点，当移动节点后，被移动的节点下的资产会放到未分组节点下的问题 * [Update] 升级jquery * [Update] 默认使用page * [Update] 修改使用Orgmodel view set * [Update] 支持 nv的硬盘 https://github.com/jumpserver/jumpserver/issues/1804 * [UPdate] 解决命令执行宽度问题 * [Update] 优化节点 * [Update] 修改nodes过多时创建比较麻烦 * [Update] 修改导入 * [Update] 节点获取更新 * [Update] 修改nodes * [Update] nodes显示full value * [Update] 统一使用nodes select2 函数 * [Update] 修改磁盘大小小数 * [Update] 修改 Node service * [Update] 优化授权节点 * [Update] 修改 node permission * [Update] 修改asset permission * [Stash] * [Update] 修改node assets api * [Update] 修改tree service，支持资产数量 * [Update] 修改暂时完成 * [Update] 修改一些bug
98 lines
3.3 KiB
Python
98 lines
3.3 KiB
Python
# -*- coding: utf-8 -*-
|
|
#
|
|
|
|
import uuid
|
|
import random
|
|
|
|
import paramiko
|
|
|
|
from django.db import models
|
|
from django.utils.translation import ugettext_lazy as _
|
|
|
|
from orgs.mixins.models import OrgModelMixin
|
|
from .base import AssetUser
|
|
|
|
__all__ = ['Domain', 'Gateway']
|
|
|
|
|
|
class Domain(OrgModelMixin):
|
|
id = models.UUIDField(default=uuid.uuid4, primary_key=True)
|
|
name = models.CharField(max_length=128, unique=True, verbose_name=_('Name'))
|
|
comment = models.TextField(blank=True, verbose_name=_('Comment'))
|
|
date_created = models.DateTimeField(auto_now_add=True, null=True,
|
|
verbose_name=_('Date created'))
|
|
|
|
class Meta:
|
|
verbose_name = _("Domain")
|
|
|
|
def __str__(self):
|
|
return self.name
|
|
|
|
def has_gateway(self):
|
|
return self.gateway_set.filter(is_active=True).exists()
|
|
|
|
@property
|
|
def gateways(self):
|
|
return self.gateway_set.filter(is_active=True)
|
|
|
|
def random_gateway(self):
|
|
return random.choice(self.gateways)
|
|
|
|
|
|
class Gateway(AssetUser):
|
|
PROTOCOL_SSH = 'ssh'
|
|
PROTOCOL_RDP = 'rdp'
|
|
PROTOCOL_CHOICES = (
|
|
(PROTOCOL_SSH, 'ssh'),
|
|
(PROTOCOL_RDP, 'rdp'),
|
|
)
|
|
ip = models.GenericIPAddressField(max_length=32, verbose_name=_('IP'), db_index=True)
|
|
port = models.IntegerField(default=22, verbose_name=_('Port'))
|
|
protocol = models.CharField(choices=PROTOCOL_CHOICES, max_length=16, default=PROTOCOL_SSH, verbose_name=_("Protocol"))
|
|
domain = models.ForeignKey(Domain, on_delete=models.CASCADE, verbose_name=_("Domain"))
|
|
comment = models.CharField(max_length=128, blank=True, null=True, verbose_name=_("Comment"))
|
|
is_active = models.BooleanField(default=True, verbose_name=_("Is active"))
|
|
|
|
def __str__(self):
|
|
return self.name
|
|
|
|
class Meta:
|
|
unique_together = [('name', 'org_id')]
|
|
verbose_name = _("Gateway")
|
|
|
|
def test_connective(self, local_port=None):
|
|
if local_port is None:
|
|
local_port = self.port
|
|
client = paramiko.SSHClient()
|
|
client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
|
|
proxy = paramiko.SSHClient()
|
|
proxy.set_missing_host_key_policy(paramiko.AutoAddPolicy())
|
|
|
|
try:
|
|
proxy.connect(self.ip, port=self.port,
|
|
username=self.username,
|
|
password=self.password,
|
|
pkey=self.private_key_obj)
|
|
except(paramiko.AuthenticationException,
|
|
paramiko.BadAuthenticationType,
|
|
paramiko.SSHException,
|
|
paramiko.ssh_exception.NoValidConnectionsError) as e:
|
|
return False, str(e)
|
|
|
|
try:
|
|
sock = proxy.get_transport().open_channel(
|
|
'direct-tcpip', ('127.0.0.1', local_port), ('127.0.0.1', 0)
|
|
)
|
|
client.connect("127.0.0.1", port=local_port,
|
|
username=self.username,
|
|
password=self.password,
|
|
key_filename=self.private_key_file,
|
|
sock=sock,
|
|
timeout=5)
|
|
except (paramiko.SSHException, paramiko.ssh_exception.SSHException,
|
|
paramiko.AuthenticationException, TimeoutError) as e:
|
|
return False, str(e)
|
|
finally:
|
|
client.close()
|
|
return True, None
|