mirror of https://github.com/jumpserver/jumpserver
263 lines
8.7 KiB
Python
263 lines
8.7 KiB
Python
# ~*~ coding: utf-8 ~*~
|
|
|
|
from itertools import groupby
|
|
from celery import shared_task
|
|
from common.db.utils import get_object_if_need, get_objects_if_need, get_objects
|
|
from django.utils.translation import ugettext as _
|
|
from django.db.models import Empty
|
|
|
|
from common.utils import encrypt_password, get_logger
|
|
from assets.models import SystemUser, Asset
|
|
from orgs.utils import org_aware_func
|
|
from . import const
|
|
from .utils import clean_ansible_task_hosts, group_asset_by_platform
|
|
|
|
|
|
logger = get_logger(__file__)
|
|
__all__ = [
|
|
'push_system_user_util', 'push_system_user_to_assets',
|
|
'push_system_user_to_assets_manual', 'push_system_user_a_asset_manual',
|
|
]
|
|
|
|
|
|
def _split_by_comma(raw: str):
|
|
try:
|
|
return [i.strip() for i in raw.split(',')]
|
|
except AttributeError:
|
|
return []
|
|
|
|
|
|
def _dump_args(args: dict):
|
|
return ' '.join([f'{k}={v}' for k, v in args.items() if v is not Empty])
|
|
|
|
|
|
def get_push_unixlike_system_user_tasks(system_user, username=None):
|
|
if username is None:
|
|
username = system_user.username
|
|
password = system_user.password
|
|
public_key = system_user.public_key
|
|
comment = system_user.name
|
|
|
|
groups = _split_by_comma(system_user.system_groups)
|
|
|
|
if groups:
|
|
groups = '"%s"' % ','.join(groups)
|
|
|
|
add_user_args = {
|
|
'name': username,
|
|
'shell': system_user.shell or Empty,
|
|
'state': 'present',
|
|
'home': system_user.home or Empty,
|
|
'groups': groups or Empty,
|
|
'comment': comment
|
|
}
|
|
|
|
tasks = [
|
|
{
|
|
'name': 'Add user {}'.format(username),
|
|
'action': {
|
|
'module': 'user',
|
|
'args': _dump_args(add_user_args),
|
|
}
|
|
},
|
|
{
|
|
'name': 'Add group {}'.format(username),
|
|
'action': {
|
|
'module': 'group',
|
|
'args': 'name={} state=present'.format(username),
|
|
}
|
|
}
|
|
]
|
|
if not system_user.home:
|
|
tasks.extend([
|
|
{
|
|
'name': 'Check home dir exists',
|
|
'action': {
|
|
'module': 'stat',
|
|
'args': 'path=/home/{}'.format(username)
|
|
},
|
|
'register': 'home_existed'
|
|
},
|
|
{
|
|
'name': "Set home dir permission",
|
|
'action': {
|
|
'module': 'file',
|
|
'args': "path=/home/{0} owner={0} group={0} mode=700".format(username)
|
|
},
|
|
'when': 'home_existed.stat.exists == true'
|
|
}
|
|
])
|
|
if password:
|
|
tasks.append({
|
|
'name': 'Set {} password'.format(username),
|
|
'action': {
|
|
'module': 'user',
|
|
'args': 'name={} shell={} state=present password={}'.format(
|
|
username, system_user.shell,
|
|
encrypt_password(password, salt="K3mIlKK"),
|
|
),
|
|
}
|
|
})
|
|
if public_key:
|
|
tasks.append({
|
|
'name': 'Set {} authorized key'.format(username),
|
|
'action': {
|
|
'module': 'authorized_key',
|
|
'args': "user={} state=present key='{}'".format(
|
|
username, public_key
|
|
)
|
|
}
|
|
})
|
|
if system_user.sudo:
|
|
sudo = system_user.sudo.replace('\r\n', '\n').replace('\r', '\n')
|
|
sudo_list = sudo.split('\n')
|
|
sudo_tmp = []
|
|
for s in sudo_list:
|
|
sudo_tmp.append(s.strip(','))
|
|
sudo = ','.join(sudo_tmp)
|
|
tasks.append({
|
|
'name': 'Set {} sudo setting'.format(username),
|
|
'action': {
|
|
'module': 'lineinfile',
|
|
'args': "dest=/etc/sudoers state=present regexp='^{0} ALL=' "
|
|
"line='{0} ALL=(ALL) NOPASSWD: {1}' "
|
|
"validate='visudo -cf %s'".format(username, sudo)
|
|
}
|
|
})
|
|
|
|
return tasks
|
|
|
|
|
|
def get_push_windows_system_user_tasks(system_user, username=None):
|
|
if username is None:
|
|
username = system_user.username
|
|
password = system_user.password
|
|
groups = {'Users', 'Remote Desktop Users'}
|
|
if system_user.system_groups:
|
|
groups.update(_split_by_comma(system_user.system_groups))
|
|
groups = ','.join(groups)
|
|
|
|
tasks = []
|
|
if not password:
|
|
logger.error("Error: no password found")
|
|
return tasks
|
|
task = {
|
|
'name': 'Add user {}'.format(username),
|
|
'action': {
|
|
'module': 'win_user',
|
|
'args': 'fullname={} '
|
|
'name={} '
|
|
'password={} '
|
|
'state=present '
|
|
'update_password=always '
|
|
'password_expired=no '
|
|
'password_never_expires=yes '
|
|
'groups="{}" '
|
|
'groups_action=add '
|
|
''.format(username, username, password, groups),
|
|
}
|
|
}
|
|
tasks.append(task)
|
|
return tasks
|
|
|
|
|
|
def get_push_system_user_tasks(system_user, platform="unixlike", username=None):
|
|
"""
|
|
:param system_user:
|
|
:param platform:
|
|
:param username: 当动态时，近推送某个
|
|
:return:
|
|
"""
|
|
get_task_map = {
|
|
"unixlike": get_push_unixlike_system_user_tasks,
|
|
"windows": get_push_windows_system_user_tasks,
|
|
}
|
|
get_tasks = get_task_map.get(platform, get_push_unixlike_system_user_tasks)
|
|
if not system_user.username_same_with_user:
|
|
return get_tasks(system_user)
|
|
tasks = []
|
|
# 仅推送这个username
|
|
if username is not None:
|
|
tasks.extend(get_tasks(system_user, username))
|
|
return tasks
|
|
users = system_user.users.all().values_list('username', flat=True)
|
|
print(_("System user is dynamic: {}").format(list(users)))
|
|
for _username in users:
|
|
tasks.extend(get_tasks(system_user, _username))
|
|
return tasks
|
|
|
|
|
|
@org_aware_func("system_user")
|
|
def push_system_user_util(system_user, assets, task_name, username=None):
|
|
from ops.utils import update_or_create_ansible_task
|
|
hosts = clean_ansible_task_hosts(assets, system_user=system_user)
|
|
if not hosts:
|
|
return {}
|
|
|
|
platform_hosts_map = {}
|
|
hosts_sorted = sorted(hosts, key=group_asset_by_platform)
|
|
platform_hosts = groupby(hosts_sorted, key=group_asset_by_platform)
|
|
for i in platform_hosts:
|
|
platform_hosts_map[i[0]] = list(i[1])
|
|
|
|
def run_task(_tasks, _hosts):
|
|
if not _tasks:
|
|
return
|
|
task, created = update_or_create_ansible_task(
|
|
task_name=task_name, hosts=_hosts, tasks=_tasks, pattern='all',
|
|
options=const.TASK_OPTIONS, run_as_admin=True,
|
|
)
|
|
task.run()
|
|
|
|
for platform, _hosts in platform_hosts_map.items():
|
|
if not _hosts:
|
|
continue
|
|
print(_("Start push system user for platform: [{}]").format(platform))
|
|
print(_("Hosts count: {}").format(len(_hosts)))
|
|
|
|
# 如果没有特殊密码设置，就不需要单独推送某台机器了
|
|
if not system_user.has_special_auth(username=username):
|
|
logger.debug("System user not has special auth")
|
|
tasks = get_push_system_user_tasks(system_user, platform, username=username)
|
|
run_task(tasks, _hosts)
|
|
continue
|
|
|
|
for _host in _hosts:
|
|
system_user.load_asset_special_auth(_host, username=username)
|
|
tasks = get_push_system_user_tasks(system_user, platform, username=username)
|
|
run_task(tasks, [_host])
|
|
|
|
|
|
@shared_task(queue="ansible")
|
|
def push_system_user_to_assets_manual(system_user, username=None):
|
|
system_user = get_object_if_need(SystemUser, system_user)
|
|
assets = system_user.get_related_assets()
|
|
task_name = _("Push system users to assets: {}").format(system_user.name)
|
|
return push_system_user_util(system_user, assets, task_name=task_name, username=username)
|
|
|
|
|
|
@shared_task(queue="ansible")
|
|
def push_system_user_a_asset_manual(system_user, asset, username=None):
|
|
if username is None:
|
|
username = system_user.username
|
|
task_name = _("Push system users to asset: {}({}) => {}").format(
|
|
system_user.name, username, asset
|
|
)
|
|
return push_system_user_util(system_user, [asset], task_name=task_name, username=username)
|
|
|
|
|
|
@shared_task(queue="ansible")
|
|
def push_system_user_to_assets(system_user_id, assets_id, username=None):
|
|
system_user = SystemUser.objects.get(id=system_user_id)
|
|
assets = get_objects(Asset, assets_id)
|
|
task_name = _("Push system users to assets: {}").format(system_user.name)
|
|
return push_system_user_util(system_user, assets, task_name, username=username)
|
|
|
|
# @shared_task
|
|
# @register_as_period_task(interval=3600)
|
|
# @after_app_ready_start
|
|
# @after_app_shutdown_clean_periodic
|
|
# def push_system_user_period():
|
|
# for system_user in SystemUser.objects.all():
|
|
# push_system_user_related_nodes(system_user)
|