#!/usr/local/bin/python
import getopt, string, sys, telnetlib
 
def Usage():
  print """
Options: -p port        port number
         -s state       o or c to open/close the specified port
         -d port        destination port number (by default equal to p)
         -i ip          destination ip (by default 10.0.0.1)
         -n ip          the router IP address on the internal lan (by default 10.0.0.138)
         -u username    the username to connect to the alcatel
         -w password    the password to connect to the alcatel
 
Options -p and -w are not options.
If you don't specify -s, the status of the port will be displayed. (not yet implemented)
"""
 
def _main(argv):
 
  opt_port = ""
  opt_state = ""
  opt_dest_port = ""
  opt_dest_ip = "10.0.0.1"
  opt_router_ip = "10.0.0.138"
  opt_router_username = ""
  opt_router_password = ""
 
 
  #
  # parse the command line options
  #
  if len(argv) < 4:
    Usage()
    sys.exit(0)
 
  try:
    loweropts = "p:s:d:i:n:u:w:"
    opts, args = getopt.getopt(argv[1:], loweropts)
  except getopt.error, reason:
    print reason
    sys.exit(-1)
 
 
  for option in opts:
    (optname, optvalue) = option
    if optname == "-p":
      opt_port = optvalue
    elif optname == "-s":
      opt_state = optvalue
    elif optname == "-d":
      opt_dest_port = optvalue
    elif optname == "-i":
      opt_dest_ip = optvalue
    elif optname == "-n":
      opt_router_ip = optvalue
    elif optname == "-u":
      opt_router_username = optvalue
    elif optname == "-w":
      opt_router_password = optvalue
 
 
  if not opt_port:
    print "Error: port is mandatory"
    Usage()
    sys.exit(-1)
 
  if not opt_router_password:
    print "Error: password is mandatory"
    Usage()
    sys.exit(-1)
 
  if not opt_dest_port:
    opt_dest_port = opt_port
 
 
  # connect to the router
  # disables negotiation to prevent the connection being dropped
  telnetlib.IAC="\021"
  tn = telnetlib.Telnet(opt_router_ip)
 
  ret = tn.read_until("User : ", 5)
  if string.find(ret,"User : " )==-1:
      print "Did not get the user prompt before the 5 seconds timeout"+ret
      raise Exception("timeout for user prompt"+ret)
 
  print "User prompt found"
  tn.write("nobody\r" )
  ret = tn.read_until("assword : ", 5)
  if string.find(ret,"assword : " )==-1:
      print "Did not get the password prompt before the 5 seconds timeout"+ret
      raise Exception("timeout for password prompt"+ret)
 
  print "Password prompt found"
  tn.write(opt_router_password + "\r" )
  print "opt_router_password sent"
  ret = tn.read_until("=>", 5)
  if string.find(ret,"=>" )==-1:
      print "Did not get the prompt before the 5 seconds timeout"+ret
      raise Exception("timeout for prompt"+ret)
 
  if opt_state == "o":
    command = "create"
  elif opt_state == "c":
    command = "delete"
  else:
    raise Exception("no command" )
   
 
  line="nat %s protocol=%s inside_addr=%s inside_port=%s outside_addr=%s outside_port=%s\r"
  line=line % (command, "tcp", opt_dest_ip, opt_dest_port, '0.0.0.0', opt_port )
  tn.write(line)
  print line
  tn.read_until(line)
  ret = tn.read_until("=>" )
  print ret
   
  # no exit needed
  tn.close()
 
 
if __name__=="__main__":
 
  _main(sys.argv)