Add domain name and url options to DHCP server

Author: rcvalle

lib/rex/proto/dhcp/constants.rb

@@ -21,6 +21,8 @@ module DHCP
 OpRouter = 3
 OpDns = 6
 OpHostname = 0x0c
+OpDomainname = 0x0f
+OpURL = 0x72
 OpEnd = 0xff
 
 PXEMagic = "\xF1\x00\x74\x7E"

lib/rex/proto/dhcp/server.rb

@@ -94,6 +94,9 @@ def initialize(hash, context = {})
     self.pxealtconfigfile = "update0"
     self.pxepathprefix = ""
     self.pxereboottime = 2000
+
+    self.domainname = hash['DOMAINNAME'] if hash.include?('DOMAINNAME')
+    self.url = hash['URL'] if hash.include?('URL')
   end
 
   def report(&block)
@@ -126,7 +129,7 @@ def set_option(opts)
     allowed_options = [
       :serveOnce, :pxealtconfigfile, :servePXE, :relayip, :leasetime, :dnsserv,
       :pxeconfigfile, :pxepathprefix, :pxereboottime, :router,
-      :give_hostname, :served_hostname, :served_over, :serveOnlyPXE
+      :give_hostname, :served_hostname, :served_over, :serveOnlyPXE, :domainname, :url
     ]
 
     opts.each_pair { |k,v|
@@ -154,7 +157,7 @@ def send_packet(ip, pkt)
   attr_accessor :sock, :thread, :myfilename, :ipstring, :served, :serveOnce
   attr_accessor :current_ip, :start_ip, :end_ip, :broadcasta, :netmaskn
   attr_accessor :servePXE, :pxeconfigfile, :pxealtconfigfile, :pxepathprefix, :pxereboottime, :serveOnlyPXE
-  attr_accessor :give_hostname, :served_hostname, :served_over, :reporter
+  attr_accessor :give_hostname, :served_hostname, :served_over, :reporter, :domainname, :url
 
 protected
 
@@ -317,6 +320,8 @@ def dispatch_request(from, buf)
         pkt << dhcpoption(OpHostname, send_hostname)
       end
     end
+    pkt << dhcpoption(OpDomainname, self.domainname) if self.domainname
+    pkt << dhcpoption(OpURL, self.url) if self.url
     pkt << dhcpoption(OpEnd)
 
     pkt << ("\x00" * 32) #padding