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
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
|
"""
Misc heavy lifting functions for cobbler
Copyright 2006, Red Hat, Inc
Michael DeHaan <mdehaan@redhat.com>
This software may be freely redistributed under the terms of the GNU
general public license.
You should have received a copy of the GNU General Public License
along with this program; if not, write to the Free Software
Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
"""
import os
import re
import socket
import glob
import sub_process
_re_kernel = re.compile(r'vmlinuz(.*)')
_re_initrd = re.compile(r'initrd(.*).img')
def get_host_ip(ip):
"""
Return the IP encoding needed for the TFTP boot tree.
"""
handle = sub_process.Popen("/usr/bin/gethostip %s" % ip, shell=True, stdout=sub_process.PIPE)
out = handle.stdout
results = out.read()
return results.split(" ")[-1][0:8]
def find_system_identifier(strdata):
"""
If the input is a MAC or an IP, return that.
If it's not, resolve the hostname and return the IP.
pxe bootloaders don't work in hostnames
"""
if is_mac(strdata):
return strdata.upper()
if is_ip(strdata):
return strdata
return resolve_ip(strdata)
def is_ip(strdata):
"""
Return whether the argument is an IP address. ipv6 needs
to be added...
"""
# needs testcase
if strdata is None:
return False
if re.search(r'\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3}',strdata):
return True
return False
def is_mac(strdata):
"""
Return whether the argument is a mac address.
"""
# needs testcase
if strdata is None:
return False
if re.search(r'[A-F0-9]{2}:[A-F0-9]{2}:[A-F0-9]{2}:[A-F0-9]{2}:[A-F:0-9]{2}:[A-F:0-9]{2}',strdata, re.IGNORECASE):
return True
return False
def resolve_ip(strdata):
"""
Resolve the IP address and handle errors...
"""
try:
return socket.gethostbyname(strdata)
except:
return None
def find_matching_files(directory,regex):
"""
Find all files in a given directory that match a given regex.
Can't use glob directly as glob doesn't take regexen.
"""
files = glob.glob(os.path.join(directory,"*"))
results = []
for f in files:
if regex.match(os.path.basename(f)):
results.append(f)
return results
def find_highest_files(directory,unversioned,regex):
"""
Find the highest numbered file (kernel or initrd numbering scheme)
in a given directory that matches a given pattern. Used for
auto-booting the latest kernel in a directory.
"""
files = find_matching_files(directory, regex)
get_numbers = re.compile(r'(\d+).(\d+).(\d+)')
def max2(a, b):
"""Returns the larger of the two values"""
av = get_numbers.search(os.path.basename(a)).groups()
bv = get_numbers.search(os.path.basename(b)).groups()
ret = cmp(av[0], bv[0]) or cmp(av[1], bv[1]) or cmp(av[2], bv[2])
if ret < 0:
return b
return a
if len(files) > 0:
return reduce(max2, files)
# couldn't find a highest numbered file, but maybe there
# is just a 'vmlinuz' or an 'initrd.img' in this directory?
last_chance = os.path.join(directory,unversioned)
if os.path.exists(last_chance):
return last_chance
return None
def find_kernel(path):
"""
Given a directory or a filename, find if the path can be made
to resolve into a kernel, and return that full path if possible.
"""
if path is None:
return None
if os.path.isfile(path):
filename = os.path.basename(path)
if _re_kernel.match(filename):
return path
elif filename == "vmlinuz":
return path
elif os.path.isdir(path):
return find_highest_files(path,"vmlinuz",_re_kernel)
return None
def find_initrd(path):
"""
Given a directory or a filename, see if the path can be made
to resolve into an intird, return that full path if possible.
"""
# FUTURE: try to match kernel/initrd pairs?
if path is None:
return None
if os.path.isfile(path):
filename = os.path.basename(path)
if _re_initrd.match(filename):
return path
if filename == "initrd.img" or filename == "initrd":
return path
elif os.path.isdir(path):
return find_highest_files(path,"initrd.img",_re_initrd)
return None
def find_kickstart(url):
"""
Check if a kickstart url looks like an http, ftp, nfs or local path.
If a local path is used, cobbler will copy the kickstart and serve
it over http.
"""
if url is None:
return None
x = url.lower()
for y in ["http://","nfs://","ftp://","/"]:
if x.startswith(y):
if x.startswith("/") and not os.path.isfile(url):
return None
return url
return None
def input_string_or_hash(options,delim=","):
"""
Older cobbler files stored configurations in a flat way, such that all values for strings.
Newer versions of cobbler allow dictionaries. This function is used to allow loading
of older value formats so new users of cobbler aren't broken in an upgrade.
"""
if options is None:
return (True, {})
elif type(options) != dict:
new_dict = {}
tokens = options.split(delim)
for t in tokens:
tokens2 = t.split("=")
if len(tokens2) == 1 and tokens2[0] != '':
new_dict[tokens2[0]] = None
elif len(tokens2) == 2 and tokens2[0] != '':
new_dict[tokens2[0]] = tokens2[1]
else:
return (False, {})
new_dict.pop('', None)
return (True, new_dict)
else:
options.pop('',None)
return (True, options)
|
