mirror of
https://github.com/valitydev/salt.git
synced 2024-11-09 01:36:48 +00:00
358 lines
9.3 KiB
Python
358 lines
9.3 KiB
Python
# -*- coding: utf-8 -*-
|
|
'''
|
|
This module contains the function calls to execute command line scripts
|
|
'''
|
|
|
|
from __future__ import absolute_import
|
|
|
|
# Import python libs
|
|
from __future__ import print_function
|
|
import os
|
|
import sys
|
|
import traceback
|
|
import logging
|
|
import threading
|
|
import time
|
|
from random import randint
|
|
|
|
# Import salt libs
|
|
from salt.exceptions import SaltSystemExit, SaltClientError, SaltReqTimeoutError
|
|
import salt.defaults.exitcodes # pylint: disable=unused-import
|
|
|
|
|
|
log = logging.getLogger(__name__)
|
|
|
|
|
|
def _handle_interrupt(exc, original_exc, hardfail=False, trace=''):
|
|
'''
|
|
if hardfailing:
|
|
If we got the original stacktrace, log it
|
|
If all cases, raise the original exception
|
|
but this is logically part the initial
|
|
stack.
|
|
else just let salt exit gracefully
|
|
|
|
'''
|
|
if hardfail:
|
|
if trace:
|
|
log.error(trace)
|
|
raise original_exc
|
|
else:
|
|
raise exc
|
|
|
|
|
|
def salt_master():
|
|
'''
|
|
Start the salt master.
|
|
'''
|
|
import salt.cli.daemons
|
|
master = salt.cli.daemons.Master()
|
|
master.start()
|
|
|
|
|
|
def minion_process(queue):
|
|
'''
|
|
Start a minion process
|
|
'''
|
|
import salt.cli.daemons
|
|
# salt_minion spawns this function in a new process
|
|
|
|
def suicide_when_without_parent(parent_pid):
|
|
'''
|
|
Have the minion suicide if the parent process is gone
|
|
|
|
NOTE: there is a small race issue where the parent PID could be replace
|
|
with another process with the same PID!
|
|
'''
|
|
while True:
|
|
time.sleep(5)
|
|
try:
|
|
# check pid alive (Unix only trick!)
|
|
os.kill(parent_pid, 0)
|
|
except OSError:
|
|
sys.exit(999)
|
|
if not salt.utils.is_windows():
|
|
thread = threading.Thread(target=suicide_when_without_parent, args=(os.getppid(),))
|
|
thread.start()
|
|
|
|
restart = False
|
|
minion = None
|
|
try:
|
|
minion = salt.cli.daemons.Minion()
|
|
minion.start()
|
|
except (Exception, SaltClientError, SaltReqTimeoutError, SaltSystemExit) as exc:
|
|
log.error(exc)
|
|
restart = True
|
|
except SystemExit as exc:
|
|
restart = False
|
|
|
|
if restart is True:
|
|
log.warn('** Restarting minion **')
|
|
delay = 60
|
|
if minion is not None:
|
|
if hasattr(minion, 'config'):
|
|
delay = minion.config.get('random_reauth_delay', 60)
|
|
random_delay = randint(1, delay)
|
|
log.info('Sleeping random_reauth_delay of {0} seconds'.format(random_delay))
|
|
# preform delay after minion resources have been cleaned
|
|
queue.put(random_delay)
|
|
else:
|
|
queue.put(0)
|
|
|
|
|
|
def salt_minion():
|
|
'''
|
|
Start the salt minion.
|
|
'''
|
|
import salt.cli.daemons
|
|
import multiprocessing
|
|
if '' in sys.path:
|
|
sys.path.remove('')
|
|
|
|
if salt.utils.is_windows():
|
|
minion = salt.cli.daemons.Minion()
|
|
minion.start()
|
|
return
|
|
|
|
if '--disable-keepalive' in sys.argv:
|
|
sys.argv.remove('--disable-keepalive')
|
|
minion = salt.cli.daemons.Minion()
|
|
minion.start()
|
|
return
|
|
|
|
# keep one minion subprocess running
|
|
while True:
|
|
try:
|
|
queue = multiprocessing.Queue()
|
|
except Exception:
|
|
# This breaks in containers
|
|
minion = salt.cli.daemons.Minion()
|
|
minion.start()
|
|
return
|
|
process = multiprocessing.Process(target=minion_process, args=(queue,))
|
|
process.start()
|
|
try:
|
|
process.join()
|
|
try:
|
|
restart_delay = queue.get(block=False)
|
|
except Exception:
|
|
if process.exitcode == 0:
|
|
# Minion process ended naturally, Ctrl+C or --version
|
|
break
|
|
restart_delay = 60
|
|
if restart_delay == 0:
|
|
# Minion process ended naturally, Ctrl+C, --version, etc.
|
|
break
|
|
# delay restart to reduce flooding and allow network resources to close
|
|
time.sleep(restart_delay)
|
|
except KeyboardInterrupt:
|
|
break
|
|
# need to reset logging because new minion objects
|
|
# cause extra log handlers to accumulate
|
|
rlogger = logging.getLogger()
|
|
for handler in rlogger.handlers:
|
|
rlogger.removeHandler(handler)
|
|
logging.basicConfig()
|
|
|
|
|
|
def salt_syndic():
|
|
'''
|
|
Start the salt syndic.
|
|
'''
|
|
import salt.cli.daemons
|
|
pid = os.getpid()
|
|
try:
|
|
syndic = salt.cli.daemons.Syndic()
|
|
syndic.start()
|
|
except KeyboardInterrupt:
|
|
os.kill(pid, 15)
|
|
|
|
|
|
def salt_key():
|
|
'''
|
|
Manage the authentication keys with salt-key.
|
|
'''
|
|
import salt.cli.key
|
|
client = None
|
|
try:
|
|
client = salt.cli.key.SaltKey()
|
|
client.run()
|
|
except KeyboardInterrupt as err:
|
|
trace = traceback.format_exc()
|
|
try:
|
|
hardcrash = client.options.hard_crash
|
|
except (AttributeError, KeyError):
|
|
hardcrash = False
|
|
_handle_interrupt(
|
|
SystemExit('\nExiting gracefully on Ctrl-c'),
|
|
err,
|
|
hardcrash, trace=trace)
|
|
|
|
|
|
def salt_cp():
|
|
'''
|
|
Publish commands to the salt system from the command line on the
|
|
master.
|
|
'''
|
|
import salt.cli.cp
|
|
client = None
|
|
try:
|
|
client = salt.cli.cp.SaltCPCli()
|
|
client.run()
|
|
except KeyboardInterrupt as err:
|
|
trace = traceback.format_exc()
|
|
try:
|
|
hardcrash = client.options.hard_crash
|
|
except (AttributeError, KeyError):
|
|
hardcrash = False
|
|
_handle_interrupt(
|
|
SystemExit('\nExiting gracefully on Ctrl-c'),
|
|
err,
|
|
hardcrash, trace=trace)
|
|
|
|
|
|
def salt_call():
|
|
'''
|
|
Directly call a salt command in the modules, does not require a running
|
|
salt minion to run.
|
|
'''
|
|
import salt.cli.call
|
|
if '' in sys.path:
|
|
sys.path.remove('')
|
|
client = None
|
|
try:
|
|
client = salt.cli.call.SaltCall()
|
|
client.run()
|
|
except KeyboardInterrupt as err:
|
|
trace = traceback.format_exc()
|
|
try:
|
|
hardcrash = client.options.hard_crash
|
|
except (AttributeError, KeyError):
|
|
hardcrash = False
|
|
_handle_interrupt(
|
|
SystemExit('\nExiting gracefully on Ctrl-c'),
|
|
err,
|
|
hardcrash, trace=trace)
|
|
|
|
|
|
def salt_run():
|
|
'''
|
|
Execute a salt convenience routine.
|
|
'''
|
|
import salt.cli.run
|
|
if '' in sys.path:
|
|
sys.path.remove('')
|
|
client = None
|
|
try:
|
|
client = salt.cli.run.SaltRun()
|
|
client.run()
|
|
except KeyboardInterrupt as err:
|
|
trace = traceback.format_exc()
|
|
try:
|
|
hardcrash = client.options.hard_crash
|
|
except (AttributeError, KeyError):
|
|
hardcrash = False
|
|
_handle_interrupt(
|
|
SystemExit('\nExiting gracefully on Ctrl-c'),
|
|
err,
|
|
hardcrash, trace=trace)
|
|
|
|
|
|
def salt_ssh():
|
|
'''
|
|
Execute the salt-ssh system
|
|
'''
|
|
import salt.cli.ssh
|
|
if '' in sys.path:
|
|
sys.path.remove('')
|
|
client = None
|
|
try:
|
|
client = salt.cli.ssh.SaltSSH()
|
|
client.run()
|
|
except KeyboardInterrupt as err:
|
|
trace = traceback.format_exc()
|
|
try:
|
|
hardcrash = client.options.hard_crash
|
|
except (AttributeError, KeyError):
|
|
hardcrash = False
|
|
_handle_interrupt(
|
|
SystemExit('\nExiting gracefully on Ctrl-c'),
|
|
err,
|
|
hardcrash, trace=trace)
|
|
except SaltClientError as err:
|
|
trace = traceback.format_exc()
|
|
try:
|
|
hardcrash = client.options.hard_crash
|
|
except (AttributeError, KeyError):
|
|
hardcrash = False
|
|
_handle_interrupt(
|
|
SystemExit(err),
|
|
err,
|
|
hardcrash, trace=trace)
|
|
|
|
|
|
def salt_cloud():
|
|
'''
|
|
The main function for salt-cloud
|
|
'''
|
|
try:
|
|
import salt.cloud.cli
|
|
has_saltcloud = True
|
|
except ImportError:
|
|
# No salt cloud on Windows
|
|
has_saltcloud = False
|
|
if '' in sys.path:
|
|
sys.path.remove('')
|
|
|
|
if not has_saltcloud:
|
|
print('salt-cloud is not available in this system')
|
|
sys.exit(salt.defaults.exitcodes.EX_UNAVAILABLE)
|
|
|
|
client = None
|
|
try:
|
|
client = salt.cloud.cli.SaltCloud()
|
|
client.run()
|
|
except KeyboardInterrupt as err:
|
|
trace = traceback.format_exc()
|
|
try:
|
|
hardcrash = client.options.hard_crash
|
|
except (AttributeError, KeyError):
|
|
hardcrash = False
|
|
_handle_interrupt(
|
|
SystemExit('\nExiting gracefully on Ctrl-c'),
|
|
err,
|
|
hardcrash, trace=trace)
|
|
|
|
|
|
def salt_api():
|
|
'''
|
|
The main function for salt-api
|
|
'''
|
|
import salt.cli.api
|
|
sapi = salt.cli.api.SaltAPI() # pylint: disable=E1120
|
|
sapi.run()
|
|
|
|
|
|
def salt_main():
|
|
'''
|
|
Publish commands to the salt system from the command line on the
|
|
master.
|
|
'''
|
|
import salt.cli.salt
|
|
if '' in sys.path:
|
|
sys.path.remove('')
|
|
client = None
|
|
try:
|
|
client = salt.cli.salt.SaltCMD()
|
|
client.run()
|
|
except KeyboardInterrupt as err:
|
|
trace = traceback.format_exc()
|
|
try:
|
|
hardcrash = client.options.hard_crash
|
|
except (AttributeError, KeyError):
|
|
hardcrash = False
|
|
_handle_interrupt(
|
|
SystemExit('\nExiting gracefully on Ctrl-c'),
|
|
err,
|
|
hardcrash, trace=trace)
|