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
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
|
"""
Builds out DHCP info
This is the code behind 'cobbler sync'.
Copyright 2006-2008, 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 os.path
import shutil
import time
import sub_process
import sys
import glob
import traceback
import errno
import popen2
from shlex import shlex
import utils
from cexceptions import *
import templar
import item_distro
import item_profile
import item_repo
import item_system
from utils import _
class DHCPGen:
"""
Handles conversion of internal state to the tftpboot tree layout
"""
def __init__(self,config,verbose=False):
"""
Constructor
"""
self.verbose = verbose
self.config = config
self.api = config.api
self.distros = config.distros()
self.profiles = config.profiles()
self.systems = config.systems()
self.settings = config.settings()
self.repos = config.repos()
self.templar = templar.Templar(config)
def writeDHCPLease(self,port,host,ip,mac):
"""writeDHCPLease(port,host,ip,mac)
Use DHCP's API to create a DHCP entry in the /var/lib/dhcpd/dhcpd.leases file """
#Code from http://svn.osgdc.org/browse/kusu/kusu/trunk/src/kits/base/packages/kusu-base-installer/lib/kusu/nodefun.py?r=3025
fromchild, tochild = popen2.popen2("/usr/bin/omshell")
tochild.write("port %s\n" % port)
tochild.flush()
tochild.write("connect\n")
tochild.flush()
tochild.write("new host\n")
tochild.flush()
tochild.write('set name = \"%s\"\n' % host)
tochild.flush()
tochild.write("set ip-address = %s\n" % ip)
tochild.flush()
tochild.write("set hardware-address = %s\n" % mac)
tochild.flush()
tochild.write("set hardware-type = 1\n")
tochild.flush()
tochild.write("create\n")
tochild.flush()
tochild.close()
fromchild.close()
def removeDHCPLease(self,port,host):
"""removeDHCPLease(port,host)
Use DHCP's API to delete a DHCP entry in the /var/lib/dhcpd/dhcpd.leases file """
fromchild, tochild = popen2.popen2("/usr/bin/omshell")
tochild.write("port %s\n" % port)
tochild.flush()
tochild.write("connect\n")
tochild.flush()
tochild.write("new host\n")
tochild.flush()
tochild.write('set name = \"%s\"\n' % host)
tochild.flush()
tochild.write("open\n") # opens register with host information
tochild.flush()
tochild.write("remove\n")
tochild.flush()
tochild.close()
fromchild.close()
def write_dhcp_file(self):
"""
DHCP files are written when manage_dhcp is set in
/var/lib/cobbler/settings.
"""
settings_file = self.settings.dhcpd_conf
template_file = "/etc/cobbler/dhcp.template"
mode = self.settings.manage_dhcp_mode.lower()
if mode == "dnsmasq":
settings_file = self.settings.dnsmasq_conf
template_file = "/etc/cobbler/dnsmasq.template"
try:
f2 = open(template_file,"r")
except:
raise CX(_("error writing template to file: %s") % template_file)
template_data = ""
template_data = f2.read()
f2.close()
# build each per-system definition
# as configured, this only works for ISC, patches accepted
# from those that care about Itanium. elilo seems to be unmaintained
# so additional maintaince in other areas may be required to keep
# this working.
elilo = os.path.basename(self.settings.bootloaders["ia64"])
system_definitions = {}
counter = 0
# Clean system definitions in /var/lib/dhcpd/dhcpd.leases just in
# case to avoid conflicts with the hosts we're defining and to clean
# possible removed hosts (only if using OMAPI)
#
# Pablo Iranzo Gómez (Pablo.Iranzo@redhat.com)
if self.settings.omapi and self.settings.omapi_port:
file = open('/var/lib/dhcpd/dhcpd.leases')
item = shlex(file)
while 1:
elem = item.get_token()
if not elem:
break
if elem == 'host':
hostremove = item.get_token()
self.removeDHCPLease(self.settings.omapi_port,hostremove)
# we used to just loop through each system, but now we must loop
# through each network interface of each system.
for system in self.systems:
profile = system.get_conceptual_parent()
distro = profile.get_conceptual_parent()
for (name, interface) in system.interfaces.iteritems():
mac = interface["mac_address"]
ip = interface["ip_address"]
host = interface["hostname"]
if mac is None or mac == "":
# can't write a DHCP entry for this system
continue
counter = counter + 1
systxt = ""
if mode == "isc":
# the label the entry after the hostname if possible
if host is not None and host != "":
systxt = "\nhost %s {\n" % host
if self.settings.isc_set_host_name:
systxt = systxt + " option host-name = \"%s\";\n" % host
else:
systxt = "\nhost generic%d {\n" % counter
if distro.arch == "ia64":
# can't use pxelinux.0 anymore
systxt = systxt + " filename \"/%s\";\n" % elilo
systxt = systxt + " hardware ethernet %s;\n" % mac
if ip is not None and ip != "":
systxt = systxt + " fixed-address %s;\n" % ip
systxt = systxt + "}\n"
# If we have all values defined and we're using omapi,
# we will just create entries dinamically into DHCPD
# without requiring a restart (but file will be written
# as usual for having it working after restart)
if ip is not None and ip != "":
if mac is not None and mac != "":
if host is not None and host != "":
if self.settings.omapi and self.settings.omapi_port:
self.removeDHCPLease(self.settings.omapi_port,host)
self.writeDHCPLease(self.settings.omapi_port,host,ip,mac)
else:
# dnsmasq. don't have to write IP and other info here, but we do tag
# each MAC based on the arch of it's distro, if it needs something other
# than pxelinux.0 -- for these arches, and these arches only, a dnsmasq
# reload (full "cobbler sync") would be required after adding the system
# to cobbler, just to tag this relationship.
if ip is not None and ip != "":
if distro.arch.lower() == "ia64":
systxt = "dhcp-host=net:ia64," + ip + "\n"
# support for other arches needs modifications here
else:
systxt = ""
dhcp_tag = interface["dhcp_tag"]
if dhcp_tag == "":
dhcp_tag = "default"
if not system_definitions.has_key(dhcp_tag):
system_definitions[dhcp_tag] = ""
system_definitions[dhcp_tag] = system_definitions[dhcp_tag] + systxt
# we are now done with the looping through each interface of each system
metadata = {
"insert_cobbler_system_definitions" : system_definitions.get("default",""),
"date" : time.asctime(time.gmtime()),
"cobbler_server" : self.settings.server,
"next_server" : self.settings.next_server,
"elilo" : elilo
}
# now add in other DHCP expansions that are not tagged with "default"
for x in system_definitions.keys():
if x == "default":
continue
metadata["insert_cobbler_system_definitions_%s" % x] = system_definitions[x]
self.templar.render(template_data, metadata, settings_file, None)
def regen_ethers(self):
# dnsmasq knows how to read this database of MACs -> IPs, so we'll keep it up to date
# every time we add a system.
# read 'man ethers' for format info
fh = open("/etc/ethers","w+")
for sys in self.systems:
for (name, interface) in sys.interfaces.iteritems():
mac = interface["mac_address"]
ip = interface["ip_address"]
if mac is None or mac == "":
# can't write this w/o a MAC address
continue
if ip is not None and ip != "":
fh.write(mac.upper() + "\t" + ip + "\n")
fh.close()
def regen_hosts(self):
# dnsmasq knows how to read this database for host info
# (other things may also make use of this later)
fh = open("/var/lib/cobbler/cobbler_hosts","w+")
for sys in self.systems:
for (name, interface) in sys.interfaces.iteritems():
mac = interface["mac_address"]
host = interface["hostname"]
ip = interface["ip_address"]
if mac is None or mac == "":
continue
if host is not None and host != "" and ip is not None and ip != "":
fh.write(ip + "\t" + host + "\n")
fh.close()
|
