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
|
# Copyright 2019 VyOS maintainers and contributors <maintainers@vyos.io>
#
# This library is free software; you can redistribute it and/or
# modify it under the terms of the GNU Lesser General Public
# License as published by the Free Software Foundation; either
# version 2.1 of the License, or (at your option) any later version.
#
# This library 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
# Lesser General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public
# License along with this library. If not, see <http://www.gnu.org/licenses/>.
import os
import sys
def message(message, flag='', destination=sys.stdout):
"""
print a debug message line on stdout if debugging is enabled for the flag
also log it to a file if the flag 'log' is enabled
message: the message to print
flag: which flag must be set for it to print
destination: which file like object to write to (default: sys.stdout)
returns if any message was logged or not
"""
enable = enabled(flag)
if enable:
destination.write(_format(flag,message))
# the log flag is special as it logs all the commands
# executed to a log
logfile = _logfile('log', '/tmp/developer-log')
if not logfile:
return enable
try:
# at boot the file is created as root:vyattacfg
# at runtime the file is created as user:vyattacfg
# the default permission are 644
mask = os.umask(0o113)
with open(logfile, 'a') as f:
f.write(_format('log', message))
finally:
os.umask(mask)
return enable
def enabled(flag):
"""
a flag can be set by touching the file in /tmp or /config
The current flags are:
- developer: the code will drop into PBD on un-handled exception
- log: the code will log all command to a file
- ifconfig: when modifying an interface,
prints command with result and sysfs access on stdout for interface
- command: print command run with result
Having the flag setup on the filesystem is required to have
debuging at boot time, however, setting the flag via environment
does not require a seek to the filesystem and is more efficient
it can be done on the shell on via .bashrc for the user
The function returns an empty string if the flag was not set otherwise
the function returns either the file or environment name used to set it up
"""
# this is to force all new flags to be registered here to be
# documented both here and a reminder to update readthedocs :-)
if flag not in ['developer', 'log', 'ifconfig', 'command']:
return ''
return _fromenv(flag) or _fromfile(flag)
def _format(flag, message):
"""
format a log message
"""
return f'DEBUG/{flag.upper():<7} {message}\n'
def _fromenv(flag):
"""
check if debugging is set for this flag via environment
For a given debug flag named "test"
The presence of the environment VYOS_TEST_DEBUG (uppercase) enables it
return empty string if not
return content of env value it is
"""
flagname = f'VYOS_{flag.upper()}_DEBUG'
flagenv = os.environ.get(flagname, None)
if flagenv is None:
return ''
return flagenv
def _fromfile(flag):
"""
Check if debug exist for a given debug flag name
Check is a debug flag was set by the user. the flag can be set either:
- in /tmp for a non-persistent presence between reboot
- in /config for always on (an existence at boot time)
For a given debug flag named "test"
The presence of the file vyos.test.debug (all lowercase) enables it
The function returns an empty string if the flag was not set otherwise
the function returns the full flagname
"""
for folder in ('/tmp', '/config'):
flagfile = f'{folder}/vyos.{flag}.debug'
if os.path.isfile(flagfile):
return flagfile
return ''
def _contentenv(flag):
return os.environ.get(f'VYOS_{flag.upper()}_DEBUG', '').strip()
def _contentfile(flag):
"""
Check if debug exist for a given debug flag name
Check is a debug flag was set by the user. the flag can be set either:
- in /tmp for a non-persistent presence between reboot
- in /config for always on (an existence at boot time)
For a given debug flag named "test"
The presence of the file vyos.test.debug (all lowercase) enables it
The function returns an empty string if the flag was not set otherwise
the function returns the full flagname
"""
for folder in ('/tmp', '/config'):
flagfile = f'{folder}/vyos.{flag}.debug'
if not os.path.isfile(flagfile):
continue
with open(flagfile) as f:
return f.readline().strip()
return ''
def _logfile(flag, default):
"""
return the name of the file to use for logging when the flag 'log' is set
if it could not be established or the location is invalid it returns
an empty string
"""
# For log we return the location of the log file
log_location = _contentenv(flag) or _contentfile(flag)
# it was not set
if not log_location:
return ''
# Make sure that the logs can only be in /tmp, /var/log, or /tmp
if not log_location.startswith('/tmp/') and \
not log_location.startswith('/config/') and \
not log_location.startswith('/var/log/'):
return default
if '..' in log_location:
return default
return log_location
|
