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
|
# Misc heavy lifting functions for cobbler
#
# Michael DeHaan <mdehaan@redhat.com>
import config
import os
import re
import socket
import glob
class BootUtil:
def __init__(self,api,config):
self.api = api
self.config = config
self.re_kernel = re.compile(r'vmlinuz-(\d+)\.(\d+)\.(\d+)-(.*)')
self.re_initrd = re.compile(r'initrd-(\d+)\.(\d+)\.(\d+)-(.*).img')
def find_system_identifier(self,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 self.is_mac(strdata):
return strdata
if self.is_ip(strdata):
return strdata
return self.resolve_ip(strdata)
def is_ip(self,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(self,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(self,strdata):
"""
Resolve the IP address and handle errors...
"""
try:
return socket.gethostbyname(strdata)
except:
return None
def find_matching_files(self,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(self,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 = self.find_matching_files(directory, regex)
get_numbers = re.compile(r'(\d+).(\d+).(\d+)')
def sort(a,b):
av = get_numbers.search(os.path.basename(a)).groups()
bv = get_numbers.search(os.path.basename(b)).groups()
if av[0]<bv[0]: return -1
elif av[0]>bv[0]: return 1
elif av[1]<bv[1]: return -1
elif av[1]>bv[1]: return 1
elif av[2]<bv[2]: return -1
elif av[2]>bv[2]: return 1
return 0
if len(files) > 0:
return sorted(files, sort)[-1]
else:
# 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(self,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 self.re_kernel.match(filename):
return path
if filename == "vmlinuz":
return path
elif os.path.isdir(path):
return self.find_highest_files(path,"vmlinuz",self.re_kernel)
return None
def find_initrd(self,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: add another function to see if kernel and initrd have matched numbers (and throw a warning?)
if os.path.isfile(path):
filename = os.path.basename(path)
if self.re_initrd.match(filename):
return path
if filename == "initrd.img" or filename == "initrd":
return path
elif os.path.isdir(path):
return self.find_highest_files(path,"initrd.img",self.re_initrd)
return None
def find_kickstart(self,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
|
