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
|
"""
Misc heavy lifting functions for cobbler
Michael DeHaan <mdehaan@redhat.com>
"""
import os
import re
import socket
import glob
import subprocess
_re_kernel = re.compile(r'vmlinuz-(\d+)\.(\d+)\.(\d+)-(.*)')
_re_initrd = re.compile(r'initrd-(\d+)\.(\d+)\.(\d+)-(.*).img')
def get_host_ip(ip):
"""
Return the IP encoding needed for the TFTP boot tree.
"""
handle = subprocess.Popen("/usr/bin/gethostip %s" % ip, shell=True, stdout=subprocess.PIPE)
out = handle.stdout
results = out.read()
return results.split(" ")[-1]
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.
pxelinux doesn't work in hostnames
"""
if is_mac(strdata):
return strdata
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 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 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):
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 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 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(x):
return None
return url
return None
|
