boot.rb

#!/usr/bin/env ruby

MODE = 2 # Modes 1: set boot record; 2: Remove boot record
RECORD = 1 # Record 1: Windows; 2: Linux; 3: Image Cast; 4: Image Recive

BASE_ADDRESS = "143.60"
VLAN = "74"
START_IP = "1"
END_IP = "255"


###############################################################################
# NO EDIT BELOW!
###############################################################################

def ip2hex(ipaddress)
  `gethostip #{ipaddress}`.match(/(\w{8})$/)
end

ip = START_IP.to_i

while ip < END_IP.to_i do
  hex = ip2hex([BASE_ADDRESS, VLAN, ip.to_s].join('.'))
  if(MODE == 1)
    if(RECORD == 1)
      `ln -s /srv/tftp/pxe_configs/boot_windows /srv/tftp/pxelinux.cfg/#{hex}`
    elsif(RECORD == 2)
      `ln -s /srv/tftp/pxe_configs/debian /srv/tftp/pxelinux.cfg/#{hex}`
    elsif(RECORD == 3)
      `ln -s /srv/tftp/pxe_configs/udpcast_send_image /srv/tftp/pxelinux.cfg/#{hex}`
    elsif(RECORD == 4)
      `ln -s /srv/tftp/pxe_configs/udpcast_take_image /srv/tftp/pxelinux.cfg/#{hex}`
    end
  elsif(MODE == 2)
    links = Dir["/srv/tftp/pxelinux.cfg/[0-9A-F]*"]
    links.each { |n| `rm #{n}` }
  end
  ip = ip + 1
end