source: trunk/packages/sipb-xen-dhcp/files/usr/sbin/sipb-xen-dhcpserver @ 1042

Last change on this file since 1042 was 1042, checked in by broder, 16 years ago

Remove the hack to reconnect to the database if it goes away

  • Property svn:executable set to *
File size: 9.6 KB
RevLine 
[189]1#!/usr/bin/python
[191]2import sys
[189]3import pydhcplib
[190]4import pydhcplib.dhcp_network
[189]5from pydhcplib.dhcp_packet import *
6from pydhcplib.type_hw_addr import hwmac
7from pydhcplib.type_ipv4 import ipv4
[191]8from pydhcplib.type_strlist import strlist
[202]9import socket
10import IN
[189]11
[1034]12import syslog as s
[189]13
[202]14import time
[1033]15from invirt import database
16from invirt.config import structs as config
[189]17
[1033]18dhcp_options = {'subnet_mask': config.dhcp.netmask,
19                'router': config.dhcp.gateway,
20                'domain_name_server': ','.join(config.dhcp.dns),
[202]21                'ip_address_lease_time': 60*60*24}
[191]22
[189]23class DhcpBackend:
[1033]24    def __init__(self):
25        database.connect()
[202]26    def findNIC(self, mac):
[1033]27        database.clear_cache()
[1042]28        return database.NIC.query().filter_by(mac_addr=mac).one()
[202]29    def find_interface(self, packet):
30        chaddr = hwmac(packet.GetHardwareAddress())
31        nic = self.findNIC(str(chaddr))
32        if nic is None or nic.ip is None:
[1042]33            return None
[202]34        ipstr = ''.join(reversed(['%02X' % i for i in ipv4(nic.ip).list()]))
35        for line in open('/proc/net/route'):
36            parts = line.split()
37            if parts[1] == ipstr:
[1034]38                s.syslog(s.LOG_DEBUG, "find_interface found "+str(nic.ip)+" on "+parts[0])
[1033]39                return parts[0]
[1042]40        return None
[202]41                           
42    def getParameters(self, **extra):
43        all_options=dict(dhcp_options)
44        all_options.update(extra)
[191]45        options = {}
[202]46        for parameter, value in all_options.iteritems():
47            if value is None:
48                continue
[191]49            option_type = DhcpOptionsTypes[DhcpOptions[parameter]]
[189]50
[191]51            if option_type == "ipv4" :
52                # this is a single ip address
53                options[parameter] = map(int,value.split("."))
54            elif option_type == "ipv4+" :
55                # this is multiple ip address
56                iplist = value.split(",")
57                opt = []
58                for single in iplist :
[202]59                    opt.extend(ipv4(single).list())
[191]60                options[parameter] = opt
61            elif option_type == "32-bits" :
62                # This is probably a number...
63                digit = int(value)
64                options[parameter] = [digit>>24&0xFF,(digit>>16)&0xFF,(digit>>8)&0xFF,digit&0xFF]
65            elif option_type == "16-bits" :
66                digit = int(value)
67                options[parameter] = [(digit>>8)&0xFF,digit&0xFF]
68
69            elif option_type == "char" :
70                digit = int(value)
71                options[parameter] = [digit&0xFF]
72
73            elif option_type == "bool" :
74                if value=="False" or value=="false" or value==0 :
75                    options[parameter] = [0]
76                else : options[parameter] = [1]
77                   
78            elif option_type == "string" :
79                options[parameter] = strlist(value).list()
[377]80           
81            elif option_type == "RFC3397" :
82                parsed_value = ""
83                for item in value:
84                    components = item.split('.')
85                    item_fmt = "".join(chr(len(elt)) + elt for elt in components) + "\x00"
86                    parsed_value += item_fmt
[191]87               
[377]88                options[parameter] = strlist(parsed_value).list()
89           
[191]90            else :
91                options[parameter] = strlist(value).list()
92        return options
93
[189]94    def Discover(self, packet):
[1034]95        s.syslog(s.LOG_DEBUG, "dhcp_backend : Discover ")
[189]96        chaddr = hwmac(packet.GetHardwareAddress())
[202]97        nic = self.findNIC(str(chaddr))
[230]98        if nic is None or nic.machine is None:
[202]99            return False
100        ip = nic.ip
101        if ip is None:  #Deactivated?
102            return False
[377]103
104        options = {}
[252]105        if nic.hostname and '.' in nic.hostname:
[377]106            options['host_name'], options['domain_name'] = nic.hostname.split('.', 1)
[252]107        elif nic.machine.name:
[377]108            options['host_name'] = nic.machine.name
[1033]109            options['domain_name'] = config.dns.domains[0]
[252]110        else:
111            hostname = None
[377]112        if DhcpOptions['domain_search'] in packet.GetOption('parameter_request_list'):
113            options['host_name'] += '.' + options['domain_name']
114            del options['domain_name']
[1033]115            options['domain_search'] = [config.dhcp.search_domain]
[189]116        if ip is not None:
117            ip = ipv4(ip)
[1034]118            s.syslog(s.LOG_DEBUG,"dhcp_backend : Discover result = "+str(ip))
[377]119            packet_parameters = self.getParameters(**options)
[189]120
121            # FIXME: Other offer parameters go here
122            packet_parameters["yiaddr"] = ip.list()
123           
124            packet.SetMultipleOptions(packet_parameters)
125            return True
126        return False
127       
128    def Request(self, packet):
[1034]129        s.syslog(s.LOG_DEBUG, "dhcp_backend : Request")
[189]130       
131        discover = self.Discover(packet)
132       
133        chaddr = hwmac(packet.GetHardwareAddress())
134        request = packet.GetOption("request_ip_address")
[202]135        if not request:
136            request = packet.GetOption("ciaddr")
[189]137        yiaddr = packet.GetOption("yiaddr")
138
139        if not discover:
[1034]140            s.syslog(s.LOG_INFO,"Unknown MAC address: "+str(chaddr))
[189]141            return False
142       
143        if yiaddr!="0.0.0.0" and yiaddr == request :
[1034]144            s.syslog(s.LOG_INFO,"Ack ip "+str(yiaddr)+" for "+str(chaddr))
[189]145            return True
146        else:
[1034]147            s.syslog(s.LOG_INFO,"Requested ip "+str(request)+" not available for "+str(chaddr))
[189]148        return False
149
150    def Decline(self, packet):
151        pass
152    def Release(self, packet):
153        pass
154   
155
156class DhcpServer(pydhcplib.dhcp_network.DhcpServer):
157    def __init__(self, backend, options = {'client_listenport':68,'server_listenport':67}):
158        pydhcplib.dhcp_network.DhcpServer.__init__(self,"0.0.0.0",options["client_listen_port"],options["server_listen_port"],)
159        self.backend = backend
[1034]160        s.syslog(s.LOG_DEBUG, "__init__ DhcpServer")
[189]161
[202]162    def SendDhcpPacketTo(self, To, packet):
[1033]163        intf = self.backend.find_interface(packet)
[202]164        if intf:
165            out_socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
166            out_socket.setsockopt(socket.SOL_SOCKET, socket.SO_BROADCAST,1)
167            out_socket.setsockopt(socket.SOL_SOCKET, IN.SO_BINDTODEVICE, intf)
168            #out_socket.bind((ip, self.listen_port))
169            ret = out_socket.sendto(packet.EncodePacket(), (To,self.emit_port))
170            out_socket.close()
171            return ret
172        else:
173            return self.dhcp_socket.sendto(packet.EncodePacket(),(To,self.emit_port))
174
[189]175    def SendPacket(self, packet):
[190]176        """Encode and send the packet."""
[189]177       
178        giaddr = packet.GetOption('giaddr')
179
180        # in all case, if giaddr is set, send packet to relay_agent
181        # network address defines by giaddr
182        if giaddr!=[0,0,0,0] :
183            agent_ip = ".".join(map(str,giaddr))
184            self.SendDhcpPacketTo(agent_ip,packet)
[1034]185            s.syslog(s.LOG_DEBUG, "SendPacket to agent : "+agent_ip)
[189]186
187        # FIXME: This shouldn't broadcast if it has an IP address to send
188        # it to instead. See RFC2131 part 4.1 for full details
189        else :
[1034]190            s.syslog(s.LOG_DEBUG, "No agent, broadcast packet.")
[189]191            self.SendDhcpPacketTo("255.255.255.255",packet)
192           
193
194    def HandleDhcpDiscover(self, packet):
195        """Build and send DHCPOFFER packet in response to DHCPDISCOVER
196        packet."""
197
198        logmsg = "Get DHCPDISCOVER packet from " + hwmac(packet.GetHardwareAddress()).str()
199
[1034]200        s.syslog(s.LOG_INFO, logmsg)
[189]201        offer = DhcpPacket()
202        offer.CreateDhcpOfferPacketFrom(packet)
203       
[191]204        if self.backend.Discover(offer):
[189]205            self.SendPacket(offer)
206        # FIXME : what if false ?
207
208
209    def HandleDhcpRequest(self, packet):
210        """Build and send DHCPACK or DHCPNACK packet in response to
211        DHCPREQUEST packet. 4 types of DHCPREQUEST exists."""
212
213        ip = packet.GetOption("request_ip_address")
214        sid = packet.GetOption("server_identifier")
215        ciaddr = packet.GetOption("ciaddr")
[202]216        #packet.PrintHeaders()
217        #packet.PrintOptions()
[189]218
219        if sid != [0,0,0,0] and ciaddr == [0,0,0,0] :
[1034]220            s.syslog(s.LOG_INFO, "Get DHCPREQUEST_SELECTING_STATE packet")
[189]221
222        elif sid == [0,0,0,0] and ciaddr == [0,0,0,0] and ip :
[1034]223            s.syslog(s.LOG_INFO, "Get DHCPREQUEST_INITREBOOT_STATE packet")
[189]224
225        elif sid == [0,0,0,0] and ciaddr != [0,0,0,0] and not ip :
[1034]226            s.syslog(s.LOG_INFO,"Get DHCPREQUEST_INITREBOOT_STATE packet")
[189]227
[1034]228        else : s.syslog(s.LOG_INFO,"Get DHCPREQUEST_UNKNOWN_STATE packet : not implemented")
[189]229
230        if self.backend.Request(packet) : packet.TransformToDhcpAckPacket()
231        else : packet.TransformToDhcpNackPacket()
232
233        self.SendPacket(packet)
234
235
236
237    # FIXME: These are not yet implemented.
238    def HandleDhcpDecline(self, packet):
[1034]239        s.syslog(s.LOG_INFO, "Get DHCPDECLINE packet")
[189]240        self.backend.Decline(packet)
241       
242    def HandleDhcpRelease(self, packet):
[1034]243        s.syslog(s.LOG_INFO,"Get DHCPRELEASE packet")
[189]244        self.backend.Release(packet)
245       
246    def HandleDhcpInform(self, packet):
[1034]247        s.syslog(s.LOG_INFO, "Get DHCPINFORM packet")
[189]248
249        if self.backend.Request(packet) :
250            packet.TransformToDhcpAckPacket()
251            # FIXME : Remove lease_time from options
252            self.SendPacket(packet)
253
254        # FIXME : what if false ?
255
256if '__main__' == __name__:
257    options = { "server_listen_port":67,
258                "client_listen_port":68,
259                "listen_address":"0.0.0.0"}
[1033]260    backend = DhcpBackend()
[189]261    server = DhcpServer(backend, options)
262
263    while True : server.GetNextDhcpPacket()
Note: See TracBrowser for help on using the repository browser.