1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
|
# Software Name: MOON
# Version: 5.4
# SPDX-FileCopyrightText: Copyright (c) 2018-2020 Orange and its contributors
# SPDX-License-Identifier: Apache-2.0
# This software is distributed under the 'Apache License 2.0',
# the text of which is available at 'http://www.apache.org/licenses/LICENSE-2.0.txt'
# or see the "LICENSE" file for more details.
"""
Plugin to request OpenStack infrastructure:
- Keystone
- Nova
"""
import logging
import os
import atexit
import signal
import sys
import time
from moon_manager import pip_driver
from moon_manager import db_driver
from moon_manager.api import configuration
import hug.interface
LOGGER = logging.getLogger("moon.manager.plugins.daemon")
PLUGIN_TYPE = "daemon"
db_driver.init()
class OpenstackDaemon(object):
@staticmethod
def _update_subjects():
k_users = []
for manager in pip_driver.InformationManager["subjects"]:
k_users += list(manager.get_items().values())[0]
user_ids = {user["id"].replace("-", ""): user for user in k_users}
moon_subjects = db_driver.PolicyManager.get_subjects(moon_user_id=None, policy_id=None)
moon_subject_ids = moon_subjects.keys()
for user_id in user_ids:
if user_id not in moon_subject_ids:
value = {"name": user_ids[user_id]["name"], "description": ""}
db_driver.PolicyManager.add_subject(
moon_user_id=None, policy_id=None, value=value, perimeter_id=user_id)
@staticmethod
def _update_objects():
k_objects = []
for manager in pip_driver.InformationManager["objects"]:
k_objects += list(manager.get_items().values())[0]
object_ids = {object_["id"].replace("-", ""): object_ for object_ in k_objects}
moon_objects = db_driver.PolicyManager.get_objects(moon_user_id=None, policy_id=None)
moon_object_ids = moon_objects.keys()
for object_id in object_ids:
if object_id not in moon_object_ids:
value = {"name": object_ids[object_id]["name"], "description": ""}
db_driver.PolicyManager.add_object(
moon_user_id=None, policy_id=None, value=value, perimeter_id=object_id)
@staticmethod
def update():
OpenstackDaemon._update_subjects()
OpenstackDaemon._update_objects()
def daemonize(pidfile, logfile):
try:
with open(pidfile, 'r') as pf:
pid = int(pf.read().strip())
except IOError:
pid = None
if pid:
message = "pidfile {0} already exist. " + \
"Daemon already running?\n"
sys.stderr.write(message.format(pidfile))
sys.exit(1)
try:
pid = os.fork()
if pid > 0:
# exit first parent
sys.exit(0)
except OSError as err:
sys.stderr.write('fork #1 failed: {0}\n'.format(err))
sys.exit(1)
# decouple from parent environment
os.chdir('/')
os.setsid()
os.umask(0)
# do second fork
try:
pid = os.fork()
if pid > 0:
# exit from second parent
sys.exit(0)
except OSError as err:
sys.stderr.write('fork #2 failed: {0}\n'.format(err))
sys.exit(1)
# redirect standard file descriptors
sys.stdout.flush()
sys.stderr.flush()
si = open(os.devnull, 'r')
so = open(logfile, 'a+')
se = open(logfile, 'a+')
os.dup2(si.fileno(), sys.stdin.fileno())
os.dup2(so.fileno(), sys.stdout.fileno())
os.dup2(se.fileno(), sys.stderr.fileno())
# write pidfile
atexit.register(os.remove, pidfile)
pid = str(os.getpid())
with open(pidfile, 'w+') as f:
f.write(pid + '\n')
def kill_daemon(pidfile):
"""Stop the daemon."""
# Get the pid from the pidfile
try:
with open(pidfile, 'r') as pf:
pid = int(pf.read().strip())
except IOError:
pid = None
if not pid:
message = "pidfile {0} does not exist. " + \
"Daemon not running?\n"
sys.stderr.write(message.format(pidfile))
return # not an error in a restart
# Try killing the daemon process
try:
while 1:
os.kill(pid, signal.SIGTERM)
time.sleep(0.1)
except OSError as err:
e = str(err.args)
if e.find("No such process") > 0:
if os.path.exists(pidfile):
os.remove(pidfile)
else:
print(str(err.args))
sys.exit(1)
@hug.cli("start_daemon")
def run():
""" start the auto-update service """
daemon_conf = configuration.get_configuration("information").get("daemon")
daemonize(daemon_conf["pid_file"], daemon_conf["log_file"])
for category in pip_driver.InformationManager:
for manager in pip_driver.InformationManager[category]:
manager.set_auth()
while True:
OpenstackDaemon.update()
time.sleep(1)
@hug.cli("stop_daemon")
def stop():
""" stop the auto-update service """
pid_file = configuration.get_configuration("information").get("daemon").get("pid_file")
kill_daemon(pid_file)
|
