mirror of
https://github.com/Py-KMS-Organization/py-kms.git
synced 2024-09-19 15:28:52 -04:00
474c5feb6d
Added multi-ip healthcheck support Cleanup of logging inside container-scripts Removed "nc" from containers
81 lines
3.5 KiB
Python
Executable file
81 lines
3.5 KiB
Python
Executable file
#!/usr/bin/python3 -u
|
|
|
|
# Need root privileges to change timezone, and user uid/gid, file/folder ownernship
|
|
|
|
import grp
|
|
import logging
|
|
import os
|
|
import pwd
|
|
import subprocess
|
|
import sys
|
|
import signal
|
|
import time
|
|
|
|
PYTHON3 = '/usr/bin/python3'
|
|
dbPath = os.path.join(os.sep, 'home', 'py-kms', 'db') # Do not include the database file name, as we must correct the folder permissions (the db file is recursively reachable)
|
|
|
|
def change_uid_grp(logger):
|
|
if os.geteuid() != 0:
|
|
logger.info(f'not root user, cannot change uid/gid.')
|
|
return None
|
|
user_db_entries = pwd.getpwnam("py-kms")
|
|
user_grp_db_entries = grp.getgrnam("users")
|
|
uid = int(user_db_entries.pw_uid)
|
|
gid = int(user_grp_db_entries.gr_gid)
|
|
new_gid = int(os.getenv('GID', str(gid)))
|
|
new_uid = int(os.getenv('UID', str(uid)))
|
|
os.chown("/home/py-kms", new_uid, new_gid)
|
|
os.chown("/usr/bin/start.py", new_uid, new_gid)
|
|
if os.path.isdir(dbPath):
|
|
# Corret permissions recursively, as to access the database file, also its parent folder must be accessible
|
|
logger.debug(f'Correcting owner permissions on {dbPath}.')
|
|
os.chown(dbPath, new_uid, new_gid)
|
|
for root, dirs, files in os.walk(dbPath):
|
|
for dName in dirs:
|
|
dPath = os.path.join(root, dName)
|
|
logger.debug(f'Correcting owner permissions on {dPath}.')
|
|
os.chown(dPath, new_uid, new_gid)
|
|
for fName in files:
|
|
fPath = os.path.join(root, fName)
|
|
logger.debug(f'Correcting owner permissions on {fPath}.')
|
|
os.chown(fPath, new_uid, new_gid)
|
|
logger.debug(subprocess.check_output(['ls', '-la', dbPath]).decode())
|
|
if 'LOGFILE' in os.environ and os.path.exists(os.environ['LOGFILE']):
|
|
# Oh, the user also wants a custom log file -> make sure start.py can access it by setting the correct permissions (777)
|
|
os.chmod(os.environ['LOGFILE'], 0o777)
|
|
logger.error(str(subprocess.check_output(['ls', '-la', os.environ['LOGFILE']])))
|
|
logger.info("Setting gid to '%s'." % str(new_gid))
|
|
os.setgid(new_gid)
|
|
|
|
logger.info("Setting uid to '%s'." % str(new_uid))
|
|
os.setuid(new_uid)
|
|
|
|
def change_tz(logger):
|
|
tz = os.getenv('TZ', 'etc/UTC')
|
|
# TZ is not symlinked and defined TZ exists
|
|
if tz not in os.readlink('/etc/localtime') and os.path.isfile('/usr/share/zoneinfo/' + tz) and hasattr(time, 'tzset'):
|
|
logger.info("Setting timzeone to %s" % tz )
|
|
# time.tzet() should be called on Unix, but doesn't exist on Windows.
|
|
time.tzset()
|
|
|
|
if __name__ == "__main__":
|
|
log_level_bootstrap = log_level = os.getenv('LOGLEVEL', 'INFO')
|
|
if log_level_bootstrap == "MININFO":
|
|
log_level_bootstrap = "INFO"
|
|
loggersrv = logging.getLogger('entrypoint.py')
|
|
loggersrv.setLevel(log_level_bootstrap)
|
|
streamhandler = logging.StreamHandler(sys.stdout)
|
|
streamhandler.setLevel(log_level_bootstrap)
|
|
formatter = logging.Formatter(fmt = '\x1b[94m%(asctime)s %(levelname)-8s %(message)s', datefmt = '%a, %d %b %Y %H:%M:%S',)
|
|
streamhandler.setFormatter(formatter)
|
|
loggersrv.addHandler(streamhandler)
|
|
loggersrv.info("Log level: %s" % log_level)
|
|
loggersrv.debug("user id: %s" % os.getuid())
|
|
|
|
change_tz(loggersrv)
|
|
childProcess = subprocess.Popen(PYTHON3 + " -u /usr/bin/start.py", preexec_fn=change_uid_grp(loggersrv), shell=True)
|
|
def shutdown(signum, frame):
|
|
loggersrv.info("Received signal %s, shutting down..." % signum)
|
|
childProcess.terminate() # This will also cause communicate() from below to continue
|
|
signal.signal(signal.SIGTERM, shutdown) # This signal will be sent by Docker to request shutdown
|
|
childProcess.communicate()
|