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
|
# vi: ts=4 expandtab
#
# Copyright (C) 2015 Canonical Ltd.
#
# Author: Scott Moser <scott.moser@canonical.com>
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License version 3, as
# published by the Free Software Foundation.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
"""
fan module allows configuration of Ubuntu Fan
https://wiki.ubuntu.com/FanNetworking
Example config:
#cloud-config
fan:
config: |
# fan 240
10.0.0.0/8 eth0/16 dhcp
10.0.0.0/8 eth1/16 dhcp off
# fan 241
241.0.0.0/8 eth0/16 dhcp
config_path: /etc/network/fan
If cloud-init sees a 'fan' entry in cloud-config it will
a.) write 'config_path' with the contents
b.) install the package 'ubuntu-fan' if it is not installed
c.) ensure the service is started (or restarted if was previously running)
"""
from cloudinit import log as logging
from cloudinit.settings import PER_INSTANCE
from cloudinit import util
LOG = logging.getLogger(__name__)
frequency = PER_INSTANCE
BUILTIN_CFG = {
'config': None,
'config_path': '/etc/network/fan',
}
def stop_update_start(service, config_file, content, systemd=False):
if systemd:
cmds = {'stop': ['systemctl', 'stop', service],
'start': ['systemctl', 'start', service],
'enable': ['systemctl', 'enable', service]}
else:
cmds = {'stop': ['service', 'stop'],
'start': ['service', 'start']}
def run(cmd, msg):
try:
return util.subp(cmd, capture=True)
except util.ProcessExecutionError as e:
LOG.warn("failed: %s (%s): %s", service, cmd, e)
return False
stop_failed = not run(cmds['stop'], msg='stop %s' % service)
if not content.endswith('\n'):
content += '\n'
util.write_file(config_file, content, omode="w")
ret = run(cmds['start'], msg='start %s' % service)
if ret and stop_failed:
LOG.warn("success: %s started", service)
if 'enable' in cmds:
ret = run(cmds['enable'], msg='enable %s' % service)
return ret
def handle(name, cfg, cloud, log, args):
cfgin = cfg.get('fan')
if not cfgin:
cfgin = {}
mycfg = util.mergemanydict([cfgin, BUILTIN_CFG])
if not mycfg.get('config'):
LOG.debug("%s: no 'fan' config entry. disabling", name)
return
util.write_file(mycfg.get('config_path'), mycfg.get('config'), omode="w")
distro = cloud.distro
if not util.which('fanctl'):
distro.install_packages(['ubuntu-fan'])
stop_update_start(
service='ubuntu-fan', config_file=mycfg.get('config_path'),
content=mycfg.get('config'), systemd=distro.uses_systemd())
|
