mirror of
https://github.com/Telecominfraproject/wlan-lanforge-scripts.git
synced 2025-11-02 03:37:55 +00:00
lf_cisco_power.py: LCS-45 Read AP for Power values, updates
This commit is contained in:
Chuck SmileyRekiere
326
cisco_ap_ctl.py
326
cisco_ap_ctl.py
@@ -31,9 +31,9 @@ import pexpect
|
||||
|
||||
default_host = "localhost"
|
||||
default_ports = {
|
||||
"serial": None,
|
||||
"ssh": 22,
|
||||
"telnet": 23
|
||||
"serial": None,
|
||||
"ssh": 22,
|
||||
"telnet": 23
|
||||
}
|
||||
NL = "\n"
|
||||
CR = "\r\n"
|
||||
@@ -48,17 +48,18 @@ logfile = "stdout"
|
||||
# ^\s+1\s+6\s+\S+\s+\S+\s+(\S+)\s+(\S+)\s+(\S+)\s+(\S+)\s+(\S+)\s+(\S+)\s+(\S+)\s+(\S+)\s+(\S+)\s+(\S+)\s+(\S+)\s+(\S+)
|
||||
|
||||
def usage():
|
||||
print("$0 used connect to Cisco AP:")
|
||||
print("-a|--ap: AP to act upon")
|
||||
print("-d|--dest: destination host")
|
||||
print("-o|--port: destination port")
|
||||
print("-u|--user: AP login name")
|
||||
print("-p|--pass: AP password")
|
||||
print("-s|--scheme (serial|telnet|ssh): connect to controller via serial, ssh or telnet")
|
||||
print("--tty Serial port for accessing AP")
|
||||
print("-l|--log file: log messages here")
|
||||
print("-b|--band: a (5Ghz) or b (2.4Ghz) or abgn for dual-band 2.4Ghz AP")
|
||||
print("-h|--help")
|
||||
print("$0 used connect to Cisco AP:")
|
||||
print("-a|--ap: AP to act upon")
|
||||
print("-d|--dest: destination host")
|
||||
print("-o|--port: destination port")
|
||||
print("-u|--user: AP login name")
|
||||
print("-p|--pass: AP password")
|
||||
print("-s|--scheme (serial|telnet|ssh): connect to controller via serial, ssh or telnet")
|
||||
print("--tty Serial port for accessing AP")
|
||||
print("-l|--log file: log messages here")
|
||||
print("-b|--band: a (5Ghz) or b (2.4Ghz) or abgn for dual-band 2.4Ghz AP")
|
||||
print("-z|--action: action")
|
||||
print("-h|--help")
|
||||
|
||||
# see https://stackoverflow.com/a/13306095/11014343
|
||||
class FileAdapter(object):
|
||||
@@ -74,204 +75,111 @@ class FileAdapter(object):
|
||||
|
||||
def main():
|
||||
|
||||
global logfile
|
||||
global logfile
|
||||
|
||||
parser = argparse.ArgumentParser(description="Cisco AP Control Script")
|
||||
parser.add_argument("-a", "--ap", type=str, help="select AP")
|
||||
parser.add_argument("-d", "--dest", type=str, help="address of the AP 172.19.27.55")
|
||||
parser.add_argument("-o", "--port", type=int, help="control port on the AP, 2008")
|
||||
parser.add_argument("-u", "--user", type=str, help="credential login/username, admin")
|
||||
parser.add_argument("-p", "--passwd", type=str, help="credential password Wnbulab@123")
|
||||
parser.add_argument("-s", "--scheme", type=str, choices=["serial", "ssh", "telnet"], help="Connect via serial, ssh or telnet")
|
||||
parser.add_argument("-t", "--tty", type=str, help="tty serial device for connecting to AP")
|
||||
parser.add_argument("-l", "--log", type=str, help="logfile for messages, stdout means output to console")
|
||||
parser = argparse.ArgumentParser(description="Cisco AP Control Script")
|
||||
parser.add_argument("-a", "--prompt", type=str, help="ap prompt")
|
||||
parser.add_argument("-d", "--dest", type=str, help="address of the AP 172.19.27.55")
|
||||
parser.add_argument("-o", "--port", type=int, help="control port on the AP, 2008")
|
||||
parser.add_argument("-u", "--user", type=str, help="credential login/username, admin")
|
||||
parser.add_argument("-p", "--passwd", type=str, help="credential password Wnbulab@123")
|
||||
parser.add_argument("-s", "--scheme", type=str, choices=["serial", "ssh", "telnet"], help="Connect via serial, ssh or telnet")
|
||||
parser.add_argument("-t", "--tty", type=str, help="tty serial device for connecting to AP")
|
||||
parser.add_argument("-l", "--log", type=str, help="logfile for messages, stdout means output to console")
|
||||
parser.add_argument("-z", "--action", type=str, help="action, current action is powercfg")
|
||||
|
||||
args = None
|
||||
try:
|
||||
args = parser.parse_args()
|
||||
host = args.dest
|
||||
scheme = args.scheme
|
||||
port = (default_ports[scheme], args.port)[args.port != None]
|
||||
user = args.user
|
||||
passwd = args.passwd
|
||||
if (args.log != None):
|
||||
logfile = args.log
|
||||
filehandler = None
|
||||
except Exception as e:
|
||||
logging.exception(e)
|
||||
usage()
|
||||
exit(2)
|
||||
args = None
|
||||
try:
|
||||
args = parser.parse_args()
|
||||
host = args.dest
|
||||
scheme = args.scheme
|
||||
port = (default_ports[scheme], args.port)[args.port != None]
|
||||
user = args.user
|
||||
passwd = args.passwd
|
||||
if (args.log != None):
|
||||
logfile = args.log
|
||||
filehandler = None
|
||||
except Exception as e:
|
||||
logging.exception(e)
|
||||
usage()
|
||||
exit(2)
|
||||
console_handler = logging.StreamHandler()
|
||||
formatter = logging.Formatter(FORMAT)
|
||||
logg = logging.getLogger(__name__)
|
||||
logg.setLevel(logging.DEBUG)
|
||||
file_handler = None
|
||||
if (logfile is not None):
|
||||
if (logfile != "stdout"):
|
||||
file_handler = logging.FileHandler(logfile, "w")
|
||||
file_handler.setLevel(logging.DEBUG)
|
||||
file_handler.setFormatter(formatter)
|
||||
logg.addHandler(file_handler)
|
||||
logging.basicConfig(format=FORMAT, handlers=[file_handler])
|
||||
else:
|
||||
# stdout logging
|
||||
logging.basicConfig(format=FORMAT, handlers=[console_handler])
|
||||
egg = None # think "eggpect"
|
||||
ser = None
|
||||
try:
|
||||
if (scheme == "serial"):
|
||||
#eggspect = pexpect.fdpexpect.fdspan(telcon, logfile=sys.stdout.buffer)
|
||||
import serial
|
||||
from pexpect_serial import SerialSpawn
|
||||
ser = serial.Serial(args.tty, 9600, timeout=5)
|
||||
print("Created serial connection on %s, open: %s"%(args.tty, ser.is_open))
|
||||
egg = SerialSpawn(ser)
|
||||
egg.logfile = FileAdapter(logg)
|
||||
time.sleep(1)
|
||||
elif (scheme == "ssh"):
|
||||
if (port is None):
|
||||
port = 22
|
||||
cmd = "ssh -p%d %s@%s"%(port, user, host)
|
||||
logg.info("Spawn: "+cmd+NL)
|
||||
egg = pexpect.spawn(cmd)
|
||||
#egg.logfile_read = sys.stdout.buffer
|
||||
egg.logfile = FileAdapter(logg)
|
||||
i = egg.expect(["password:", "continue connecting (yes/no)?"], timeout=3)
|
||||
time.sleep(0.1)
|
||||
if i == 1:
|
||||
egg.sendline('yes')
|
||||
egg.expect('password:')
|
||||
sleep(0.1)
|
||||
egg.sendline(args.passwd)
|
||||
elif (scheme == "telnet"):
|
||||
if (port is None):
|
||||
port = 23
|
||||
cmd = "telnet {} {}".format(host, port)
|
||||
logg.info("Spawn: "+cmd+NL)
|
||||
egg = pexpect.spawn(cmd)
|
||||
egg.logfile = FileAdapter(logg)
|
||||
# Will login below as needed.
|
||||
else:
|
||||
usage()
|
||||
exit(1)
|
||||
except Exception as e:
|
||||
logging.exception(e)
|
||||
|
||||
ap_prompt = "{}>".format(args.prompt)
|
||||
ap_hash = "{}#".format(args.prompt)
|
||||
egg.sendline()
|
||||
|
||||
console_handler = logging.StreamHandler()
|
||||
formatter = logging.Formatter(FORMAT)
|
||||
logg = logging.getLogger(__name__)
|
||||
logg.setLevel(logging.DEBUG)
|
||||
file_handler = None
|
||||
if (logfile is not None):
|
||||
if (logfile != "stdout"):
|
||||
file_handler = logging.FileHandler(logfile, "w")
|
||||
file_handler.setLevel(logging.DEBUG)
|
||||
file_handler.setFormatter(formatter)
|
||||
logg.addHandler(file_handler)
|
||||
logging.basicConfig(format=FORMAT, handlers=[file_handler])
|
||||
else:
|
||||
# stdout logging
|
||||
logging.basicConfig(format=FORMAT, handlers=[console_handler])
|
||||
|
||||
egg = None # think "eggpect"
|
||||
ser = None
|
||||
try:
|
||||
if (scheme == "serial"):
|
||||
#eggspect = pexpect.fdpexpect.fdspan(telcon, logfile=sys.stdout.buffer)
|
||||
import serial
|
||||
from pexpect_serial import SerialSpawn
|
||||
ser = serial.Serial(args.tty, 9600, timeout=5)
|
||||
print("Created serial connection on %s, open: %s"%(args.tty, ser.is_open))
|
||||
egg = SerialSpawn(ser);
|
||||
egg.logfile = FileAdapter(logg)
|
||||
time.sleep(1)
|
||||
|
||||
elif (scheme == "ssh"):
|
||||
if (port is None):
|
||||
port = 22
|
||||
cmd = "ssh -p%d %s@%s"%(port, user, host)
|
||||
logg.info("Spawn: "+cmd+NL)
|
||||
egg = pexpect.spawn(cmd)
|
||||
#egg.logfile_read = sys.stdout.buffer
|
||||
egg.logfile = FileAdapter(logg)
|
||||
i = egg.expect(["password:", "continue connecting (yes/no)?"], timeout=3)
|
||||
time.sleep(0.1)
|
||||
if i == 1:
|
||||
egg.sendline('yes')
|
||||
egg.expect('password:')
|
||||
sleep(0.1)
|
||||
egg.sendline(args.passwd)
|
||||
|
||||
elif (scheme == "telnet"):
|
||||
if (port is None):
|
||||
port = 23
|
||||
cmd = "telnet %s %d"%(host, port)
|
||||
logg.info("Spawn: "+cmd+NL)
|
||||
egg = pexpect.spawn(cmd)
|
||||
egg.logfile = FileAdapter(logg)
|
||||
# Will login below as needed.
|
||||
else:
|
||||
usage()
|
||||
exit(1)
|
||||
except Exception as e:
|
||||
logging.exception(e);
|
||||
|
||||
ap_prompt = "%s>"%(args.ap)
|
||||
ap_hash = "%s#"%(args.ap)
|
||||
egg.sendline()
|
||||
|
||||
i = egg.expect(['Password:', 'Username:', ap_prompt, 'u-boot>', ap_hash], timeout=3)
|
||||
j = 0
|
||||
if i == 0:
|
||||
egg.sendline()
|
||||
i = egg.expect(['Password:', 'Username:', ap_prompt, 'u-boot>', ap_hash], timeout=10)
|
||||
|
||||
i -= 1
|
||||
|
||||
if i == 0:
|
||||
time.sleep(1)
|
||||
egg.sendline(args.user)
|
||||
time.sleep(1)
|
||||
egg.expect('Password:')
|
||||
egg.sendline(args.passwd)
|
||||
|
||||
if (i <= 1):
|
||||
egg.expect(ap_prompt)
|
||||
egg.sendline("en")
|
||||
egg.expect("Password:")
|
||||
egg.sendline(args.passwd)
|
||||
egg.expect("#")
|
||||
|
||||
# Seems this cannot work on factory images, so comment it out. It is one-time
|
||||
# thing, user must do it manually.
|
||||
#logg.info('Login to Devshell and issue the updt command')
|
||||
#egg.sendline('devshell')
|
||||
#egg.expect("#", timeout=5)
|
||||
#egg.sendline("updt_util -i /lib/firmware/%s uboot"%(args.binfile))
|
||||
# if updt_util is not available, we must already be in the netbooted cookie shell
|
||||
#egg.expect("#")
|
||||
#egg.sendline('exit')
|
||||
#egg.expect("#", timeout=5)
|
||||
|
||||
logg.info("Reload the AP and bring it to u-boot")
|
||||
egg.sendline('reload')
|
||||
egg.expect(r"confirm")
|
||||
egg.sendline('yes')
|
||||
time.sleep(10)
|
||||
egg.expect(r'Hit ESC key to stop autoboot:', timeout=100)
|
||||
egg.sendline('\x1b')
|
||||
egg.expect(r"u-boot>")
|
||||
egg.sendline('printenv')
|
||||
time.sleep(5)
|
||||
egg.expect(r"u-boot>")
|
||||
|
||||
if (i <= 2):
|
||||
logg.info("Set the env variable and download bcm.bin from TFTP server")
|
||||
time.sleep(1)
|
||||
egg.sendline("setenv ipaddr %s"%(args.ip))
|
||||
time.sleep(1)
|
||||
egg.expect(r"u-boot>")
|
||||
egg.sendline("setenv netmask %s"%(args.mask))
|
||||
time.sleep(1)
|
||||
egg.expect(r"u-boot>")
|
||||
egg.sendline("setenv serverip %s"%(args.tftp_server))
|
||||
time.sleep(2)
|
||||
egg.expect(r"u-boot>")
|
||||
egg.sendline("setenv gatewayip %s"%(args.gw))
|
||||
time.sleep(2)
|
||||
egg.expect(r"u-boot>")
|
||||
egg.sendline('setenv tftpdir /')
|
||||
time.sleep(2)
|
||||
egg.expect(r"u-boot>")
|
||||
egg.sendline('saveenv')
|
||||
time.sleep(10)
|
||||
egg.expect("u-boot>", timeout=5)
|
||||
egg.sendline('printenv')
|
||||
time.sleep(10)
|
||||
egg.expect("u-boot>", timeout=15)
|
||||
|
||||
egg.sendline("netboot %s"%(args.netboot_binfile))
|
||||
while True:
|
||||
time.sleep(10)
|
||||
k = egg.expect(["Username:", "Unable to download image"], timeout=240)
|
||||
if (k == 0):
|
||||
break
|
||||
if (k == 1):
|
||||
egg.sendline("netboot %s"%(args.netboot_binfile))
|
||||
continue
|
||||
break
|
||||
|
||||
|
||||
logg.info("Setting up the cookie from devshell and rebooting the AP")
|
||||
|
||||
# Log in again
|
||||
egg.sendline()
|
||||
time.sleep(0.1)
|
||||
egg.expect('Username:', timeout=3)
|
||||
time.sleep(0.1)
|
||||
egg.sendline(args.user)
|
||||
time.sleep(0.1)
|
||||
egg.expect('Password:')
|
||||
egg.sendline(args.passwd)
|
||||
|
||||
egg.expect(ap_prompt)
|
||||
egg.sendline("en")
|
||||
egg.expect("Password:")
|
||||
egg.sendline(args.passwd)
|
||||
egg.expect("#")
|
||||
|
||||
|
||||
egg.sendline('show controllers dot11Radio 1 powercfg | g T1')
|
||||
egg.expect("#", timeout=5)
|
||||
# ctlr.execute(cn_cmd)
|
||||
|
||||
|
||||
# ----- ----- ----- ----- ----- ----- ----- ----- ----- ----- ----- -----
|
||||
egg.sendline()
|
||||
time.sleep(0.1)
|
||||
egg.expect('Username:', timeout=3)
|
||||
time.sleep(0.1)
|
||||
egg.sendline(args.user)
|
||||
time.sleep(0.1)
|
||||
egg.expect('Password:')
|
||||
egg.sendline(args.passwd)
|
||||
egg.expect(ap_prompt)
|
||||
egg.sendline("en")
|
||||
egg.expect("Password:")
|
||||
egg.sendline(args.passwd)
|
||||
egg.expect("#")
|
||||
egg.sendline('show controllers dot11Radio 1 powercfg | g T1')
|
||||
egg.expect("#", timeout=5)
|
||||
# ctlr.execute(cn_cmd)
|
||||
----- ----- ----- ----- ----- ----- ----- ----- ----- ----- ----- -----
|
||||
if __name__ == '__main__':
|
||||
main()
|
||||
|
||||
|
||||
Reference in New Issue
Block a user