Page Menu
Home
VyOS Platform
Search
Configure Global Search
Log In
Files
F118309721
ntp.py
No One
Temporary
Actions
Download File
Edit File
Delete File
View Transforms
Subscribe
Flag For Later
Award Token
Size
3 KB
Referenced Files
None
Subscribers
None
ntp.py
View Options
#!/usr/bin/env python3
#
# Copyright (C) 2018-2023 VyOS maintainers and contributors
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License version 2 or later 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/>.
import
os
from
vyos.config
import
Config
from
vyos.configdict
import
is_node_changed
from
vyos.configverify
import
verify_vrf
from
vyos.configverify
import
verify_interface_exists
from
vyos.util
import
call
from
vyos.util
import
chmod_750
from
vyos.util
import
get_interface_config
from
vyos.template
import
render
from
vyos
import
ConfigError
from
vyos
import
airbag
airbag
.
enable
()
config_file
=
r'/run/chrony/chrony.conf'
systemd_override
=
r'/run/systemd/system/chrony.service.d/override.conf'
user_group
=
'_chrony'
def
get_config
(
config
=
None
):
if
config
:
conf
=
config
else
:
conf
=
Config
()
base
=
[
'service'
,
'ntp'
]
if
not
conf
.
exists
(
base
):
return
None
ntp
=
conf
.
get_config_dict
(
base
,
key_mangling
=
(
'-'
,
'_'
),
get_first_key
=
True
)
ntp
[
'config_file'
]
=
config_file
ntp
[
'user'
]
=
user_group
tmp
=
is_node_changed
(
conf
,
base
+
[
'vrf'
])
if
tmp
:
ntp
.
update
({
'restart_required'
:
{}})
return
ntp
def
verify
(
ntp
):
# bail out early - looks like removal from running config
if
not
ntp
:
return
None
if
'server'
not
in
ntp
:
raise
ConfigError
(
'NTP server not configured'
)
verify_vrf
(
ntp
)
if
'interface'
in
ntp
:
# If ntpd should listen on a given interface, ensure it exists
for
interface
in
ntp
[
'interface'
]:
verify_interface_exists
(
interface
)
# If we run in a VRF, our interface must belong to this VRF, too
if
'vrf'
in
ntp
:
tmp
=
get_interface_config
(
interface
)
vrf_name
=
ntp
[
'vrf'
]
if
'master'
not
in
tmp
or
tmp
[
'master'
]
!=
vrf_name
:
raise
ConfigError
(
f
'NTP runs in VRF "{vrf_name}" - "{interface}" '
\
f
'does not belong to this VRF!'
)
return
None
def
generate
(
ntp
):
# bail out early - looks like removal from running config
if
not
ntp
:
return
None
render
(
config_file
,
'chrony/chrony.conf.j2'
,
ntp
,
user
=
user_group
,
group
=
user_group
)
render
(
systemd_override
,
'chrony/override.conf.j2'
,
ntp
,
user
=
user_group
,
group
=
user_group
)
# Ensure proper permission for chrony command socket
config_dir
=
os
.
path
.
dirname
(
config_file
)
chmod_750
(
config_dir
)
return
None
def
apply
(
ntp
):
systemd_service
=
'chrony.service'
# Reload systemd manager configuration
call
(
'systemctl daemon-reload'
)
if
not
ntp
:
# NTP support is removed in the commit
call
(
f
'systemctl stop {systemd_service}'
)
if
os
.
path
.
exists
(
config_file
):
os
.
unlink
(
config_file
)
if
os
.
path
.
isfile
(
systemd_override
):
os
.
unlink
(
systemd_override
)
return
# we need to restart the service if e.g. the VRF name changed
systemd_action
=
'reload-or-restart'
if
'restart_required'
in
ntp
:
systemd_action
=
'restart'
call
(
f
'systemctl {systemd_action} {systemd_service}'
)
return
None
if
__name__
==
'__main__'
:
try
:
c
=
get_config
()
verify
(
c
)
generate
(
c
)
apply
(
c
)
except
ConfigError
as
e
:
print
(
e
)
exit
(
1
)
File Metadata
Details
Attached
Mime Type
text/x-script.python
Expires
Sat, Oct 17, 12:51 AM (1 d, 15 h)
Storage Engine
blob
Storage Format
Raw Data
Storage Handle
4352572
Default Alt Text
ntp.py (3 KB)
Attached To
Mode
rVYOSONEX vyos-1x
Attached
Detach File
Event Timeline
Log In to Comment