source: trunk/pxeconfig/pxeconfig @ 2

#!/usr/bin/env python
#
# Author: Bas van der Vlies <basv@sara.nl>
# Date  : 16 February 2002
#
# Tester: Walter de Jong <walter@sara.nl>
#
# CVS info
#  $Date: 2002/02/18 15:02:00 $
#  $Revision: 1.1 $
#
"""
With this program you can configure which PXE configuration file
to use when a node boots. The programm will ask the following questions:
  1) Network address (Class C-network address only)
  2) Starting nummber
  3) Ending number
  4) Which PXE config file to use. 

For example, if the answers are: 
  1) 10.168.44
  2) 2
  3) 4
  4) default.node_install

Then the result is that is create symbolic links in /tftpboot/pxelinux.cfg:
  0AA82C02 ----> default.node_install
  0AA82C03 ----> default.node_install
  0AA82C04 ----> default.node_install
"""

import string
import sys
import os
import glob

# DEBUG
#
DEBUG=0

# Constants
#
PXE_CONFIG_DIR='/tftpboot/pxelinux.cfg'
NADDR=0
PXEFILE=1
START=2
END=3

def choice_pxe_configfile():
  """
  Let user choice which pxeconfig file to use.
  """

  os.chdir(PXE_CONFIG_DIR)

  # Try to determine to which default file point to and 
  # if exists remove it from te list.
  #
  try:
    default_file = os.readlink('default')
  except OSError:
    default_file = None
    pass

  files = glob.glob('default.*')
  if not files:
    print 'There are no pxe config files starting with: default.'
    sys.exit(1)

  if default_file:
    files.remove(default_file)
 

  print 'Which pxe config file must we use: ?' 
  i = 1   
  for file in files:
    print "%d : %s" %(i,file)
    i = i +1

  while 1:
    index = raw_input('Choice a number: ')
    try:
      index = int(index)
    except ValueError:
      index = len(files) + 1

    # Is the user smart enough to select
    # the right value
    #
    if 0 < index <= len(files): break

  return files[index-1]

def create_links(list):
  """
  Create the links in the PXE_CONFIG_DIR,
    list : A list containing: network hex addres, pxe config file, 
           start number, end number
  """
  os.chdir(PXE_CONFIG_DIR)
  naddr = list[NADDR]
  pxe_filename = list[PXEFILE]
  for i in range(list[START], list[END]+1):
    haddr = '%s%02X' %(naddr, i)
    if DEBUG:
      print 'linking %s to %s' %(haddr, pxe_filename)
    os.symlink(pxe_filename, haddr)


def check_network(net):
  """
  This function check if the give network is a Class C-network and will
  convert the network address to a hex address if true.
  """
  d = string.split(net, '.')

  if len(d) != 3:
    print('Only C-class nerwork addresses are supported!!!')
    sys.exit(1)

  # Check if we have valid network values
  r = ''
  for i in d:
    i = check_number(i)
    r = '%s%02X' %(r,i)

  if DEBUG:
    print r

  return r

def check_number(number):
  """
  This functions checks if the input is between 0 < number < 255:
     number : is a string
  """
  try:
    n = int(number)
  except ValueError, detail:
    print detail
    sys.exit(1)

  if 0 <= n <= 255:
    return n
  else:
    print '%s is not a valid number, must be between 0 and 255' %number
    sys.exit(1)

def interactive():

  iptable = {}

  print __doc__

  network = raw_input('Give network addres (xxx.xxx.xxx): ') 
  naddr = check_network(network)

  start = raw_input('Starting number: ') 
  start = check_number(start)

  end = raw_input('Ending number: ') 
  end = check_number(end)

  pxe_filename = choice_pxe_configfile()

  iptable[network] = []
  iptable[network].append(naddr)
  iptable[network].append(pxe_filename)
  iptable[network].append(int(start))
  iptable[network].append(int(end))


  for net in iptable.keys():
    if DEBUG:
      print net, iptable[net]    
    create_links(iptable[net])

def main():
  interactive()

if __name__ == '__main__':
  main()