mirror of
https://github.com/Telecominfraproject/wlan-testing.git
synced 2025-10-30 02:22:44 +00:00
Uncommented apnos.py and openwrt_ctl.py to prevent quali reservations from failing (#748)
This commit is contained in:
dutta-rohan
1382
libs/apnos/apnos.py
1382
libs/apnos/apnos.py
File diff suppressed because it is too large
Load Diff
@@ -16,360 +16,360 @@
|
||||
# --value2 password-for-10.28.3.100 --log stdout --scheme serial --prompt root@Open
|
||||
# '''
|
||||
#
|
||||
# import sys
|
||||
#
|
||||
# if sys.version_info[0] != 3:
|
||||
# print("This script requires Python 3")
|
||||
# exit()
|
||||
#
|
||||
# import re
|
||||
# import logging
|
||||
# import time
|
||||
# from time import sleep
|
||||
# import pprint
|
||||
# import telnetlib
|
||||
# import argparse
|
||||
# import pexpect
|
||||
#
|
||||
# default_host = "localhost"
|
||||
# default_ports = {
|
||||
# "serial": None,
|
||||
# "ssh": 22,
|
||||
# "telnet": 23
|
||||
# }
|
||||
# NL = "\n"
|
||||
# CR = "\r\n"
|
||||
# Q = '"'
|
||||
# A = "'"
|
||||
# FORMAT = '%(asctime)s %(name)s %(levelname)s: %(message)s'
|
||||
# prompt = "root@OpenWrt:"
|
||||
#
|
||||
#
|
||||
# def usage():
|
||||
# print("$0 used connect to OpenWrt AP or similar Linux machine:")
|
||||
# print("-d|--dest IP address of the OpenWrt AP, for ssh/telnet scheme")
|
||||
# print("-o|--port IP port of the OpenWrt AP, for ssh/telnet scheme")
|
||||
# print("-t|--tty Serial port, if using serial scheme")
|
||||
# print("-u|--user login name")
|
||||
# print("-p|--pass password")
|
||||
# print("--prompt Prompt to look for when commands are done (default: root@OpenWrt)")
|
||||
# print("-s|--scheme (serial|telnet|ssh): connect via serial, ssh or telnet")
|
||||
# print("-l|--log file log messages here")
|
||||
# print("--action (logread | journalctl | lurk | sysupgrade | download | upload | reboot | cmd | ssh-tunnel")
|
||||
# print("--value (option to help complete the action")
|
||||
# print("--value2 (option to help complete the action, dest filename for download, passwd for ssh-tunnel")
|
||||
# print("-h|--help")
|
||||
#
|
||||
#
|
||||
# # see https://stackoverflow.com/a/13306095/11014343
|
||||
# class FileAdapter(object):
|
||||
# def __init__(self, logger):
|
||||
# self.logger = logger
|
||||
#
|
||||
# def write(self, data):
|
||||
# # NOTE: data can be a partial line, multiple lines
|
||||
# data = data.strip() # ignore leading/trailing whitespace
|
||||
# if data: # non-blank
|
||||
# self.logger.info(data)
|
||||
#
|
||||
# def flush(self):
|
||||
# pass # leave it to logging to flush properly
|
||||
#
|
||||
#
|
||||
# def main():
|
||||
# global prompt
|
||||
#
|
||||
# parser = argparse.ArgumentParser(description="OpenWrt AP Control Script")
|
||||
# parser.add_argument("-d", "--dest", type=str, help="address of the cisco controller_tests")
|
||||
# parser.add_argument("-o", "--port", type=int, help="control port on the controller_tests")
|
||||
# parser.add_argument("-u", "--user", type=str, help="credential login/username")
|
||||
# parser.add_argument("-p", "--passwd", type=str, help="credential password")
|
||||
# parser.add_argument("-P", "--prompt", type=str, help="Prompt to look for")
|
||||
# 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")
|
||||
# parser.add_argument("-l", "--log", type=str, help="logfile for messages, stdout means output to console")
|
||||
# parser.add_argument("--action", type=str, help="perform action",
|
||||
# choices=["logread", "journalctl", "lurk", "sysupgrade", "sysupgrade-n", "download", "upload",
|
||||
# "reboot", "cmd", "ssh-tunnel"])
|
||||
# parser.add_argument("--value", type=str, help="set value")
|
||||
# parser.add_argument("--value2", type=str, help="set value2")
|
||||
# tty = None
|
||||
#
|
||||
# args = None
|
||||
# try:
|
||||
# args = parser.parse_args()
|
||||
# host = args.dest
|
||||
# scheme = args.scheme
|
||||
# port = args.port
|
||||
# # port = (default_ports[scheme], args.port)[args.port != None]
|
||||
# user = args.user
|
||||
# passwd = args.passwd
|
||||
# logfile = args.log
|
||||
# tty = args.tty;
|
||||
# if (args.prompt != None):
|
||||
# prompt = args.prompt
|
||||
# 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])
|
||||
#
|
||||
# CCPROMPT = prompt
|
||||
#
|
||||
# ser = None
|
||||
# egg = None # think "eggpect"
|
||||
# try:
|
||||
# if (scheme == "serial"):
|
||||
# # eggspect = pexpect.fdpexpect.fdspan(telcon, logfile=sys.stdout.buffer)
|
||||
# import serial
|
||||
# from pexpect_serial import SerialSpawn
|
||||
# ser = serial.Serial(tty, 115200, timeout=5)
|
||||
#
|
||||
# egg = SerialSpawn(ser);
|
||||
# egg.logfile = FileAdapter(logg)
|
||||
# egg.sendline(NL)
|
||||
# has_reset = False
|
||||
# try:
|
||||
# logg.info("prompt: %s user: %s passwd: %s" % (prompt, user, passwd))
|
||||
# while True:
|
||||
# i = egg.expect([prompt, "Please press Enter to activate", "login:", "Password:", "IPQ6018#"],
|
||||
# timeout=3)
|
||||
# logg.info("expect-0: %i" % (i))
|
||||
# if (i == 0):
|
||||
# logg.info("Found prompt, login complete.")
|
||||
# break
|
||||
# if (i == 1):
|
||||
# logg.info("Sending newline")
|
||||
# egg.setdline(NL)
|
||||
# if (i == 2):
|
||||
# logg.info("Sending username: %s" % (user))
|
||||
# egg.sendline(user)
|
||||
# if (i == 3):
|
||||
# logg.info("Sending password: %s" % (passwd))
|
||||
# egg.sendline(passwd)
|
||||
# if (i == 4): # in bootloader
|
||||
# if has_reset:
|
||||
# logg.info("ERROR: Have reset once already, back in bootloader?")
|
||||
# sys.exit(1)
|
||||
# has_reset = True
|
||||
# logg.info("In boot loader, will reset and sleep 30 seconds")
|
||||
# egg.sendline("reset")
|
||||
# time.sleep(30)
|
||||
# egg.sendline(NL)
|
||||
#
|
||||
# except Exception as e:
|
||||
# # maybe something like 'logread -f' is running?
|
||||
# # send ctrl-c
|
||||
# egg.send(chr(3))
|
||||
#
|
||||
# elif (scheme == "ssh"):
|
||||
# # Not implemented/tested currently. --Ben
|
||||
# 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(passwd)
|
||||
#
|
||||
# elif (scheme == "telnet"):
|
||||
# # Not implemented/tested currently. --Ben
|
||||
# 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)
|
||||
# time.sleep(0.1)
|
||||
# egg.sendline(' ')
|
||||
# egg.expect('User\:')
|
||||
# egg.sendline(user)
|
||||
# egg.expect('Password\:')
|
||||
# egg.sendline(passwd)
|
||||
# egg.sendline('config paging disable')
|
||||
# else:
|
||||
# usage()
|
||||
# exit(1)
|
||||
# except Exception as e:
|
||||
# logging.exception(e);
|
||||
#
|
||||
# command = None
|
||||
#
|
||||
# CLOSEDBYREMOTE = "closed by remote host."
|
||||
# CLOSEDCX = "Connection to .* closed."
|
||||
#
|
||||
# try:
|
||||
# egg.expect(CCPROMPT)
|
||||
# except Exception as e:
|
||||
# egg.sendline(NL)
|
||||
#
|
||||
# TO = 10
|
||||
# wait_forever = False
|
||||
#
|
||||
# # Clean pending output
|
||||
# egg.sendline("echo __hello__")
|
||||
# egg.expect("__hello__")
|
||||
# egg.expect(CCPROMPT)
|
||||
#
|
||||
# logg.info("Action[%s] Value[%s] Value2[%s]" % (args.action, args.value, args.value2))
|
||||
#
|
||||
# if (args.action == "reboot"):
|
||||
# command = "reboot"
|
||||
# TO = 60
|
||||
#
|
||||
# if (args.action == "cmd"):
|
||||
# if (args.value is None):
|
||||
# raise Exception("cmd requires value to be set.")
|
||||
# command = "%s" % (args.value)
|
||||
#
|
||||
# if (args.action == "logread"):
|
||||
# command = "logread -f"
|
||||
# TO = 1
|
||||
# wait_forever = True
|
||||
#
|
||||
# if (args.action == "journalctl"):
|
||||
# command = "journalctl -f"
|
||||
# TO = 1
|
||||
# wait_forever = True
|
||||
#
|
||||
# if (args.action == "lurk"):
|
||||
# command = "date"
|
||||
# TO = 1
|
||||
# wait_forever = True
|
||||
#
|
||||
# if (args.action == "ssh-tunnel"):
|
||||
# command = "%s" % (args.value)
|
||||
# passwd = "%s" % (args.value2)
|
||||
# logg.info("Command[%s]" % command)
|
||||
# egg.sendline(command);
|
||||
#
|
||||
# i = egg.expect(["password:", "Do you want to continue connecting"], timeout=5)
|
||||
# if i == 1:
|
||||
# egg.sendline("y")
|
||||
# egg.expect("password:", timeout=5)
|
||||
# egg.sendline(passwd)
|
||||
# egg.expect(CCPROMPT, timeout=20)
|
||||
# return
|
||||
#
|
||||
# if ((args.action == "sysupgrade") or (args.action == "sysupgrade-n")):
|
||||
# command = "scp %s /tmp/new_img.bin" % (args.value)
|
||||
# logg.info("Command[%s]" % command)
|
||||
# egg.sendline(command);
|
||||
#
|
||||
# i = egg.expect(["password:", "Do you want to continue connecting"], timeout=5)
|
||||
# if i == 1:
|
||||
# egg.sendline("y")
|
||||
# egg.expect("password:", timeout=5)
|
||||
# egg.sendline("lanforge")
|
||||
# egg.expect(CCPROMPT, timeout=20)
|
||||
# if (args.action == "sysupgrade-n"):
|
||||
# egg.sendline("sysupgrade -n /tmp/new_img.bin")
|
||||
# else:
|
||||
# egg.sendline("sysupgrade /tmp/new_img.bin")
|
||||
# egg.expect("link becomes ready", timeout=100)
|
||||
# return
|
||||
#
|
||||
# if (args.action == "download"):
|
||||
# command = "scp %s /tmp/%s" % (args.value, args.value2)
|
||||
# logg.info("Command[%s]" % command)
|
||||
# egg.sendline(command);
|
||||
#
|
||||
# i = egg.expect(["password:", "Do you want to continue connecting", "Network unreachable"], timeout=5)
|
||||
# if i == 2:
|
||||
# print("Network unreachable, wait 15 seconds and try again.")
|
||||
# time.sleep(15)
|
||||
# command = "scp %s /tmp/%s" % (args.value, args.value2)
|
||||
# logg.info("Command[%s]" % command)
|
||||
# egg.sendline(command);
|
||||
#
|
||||
# i = egg.expect(["password:", "Do you want to continue connecting", "Network unreachable"], timeout=5)
|
||||
# if i == 2:
|
||||
# print("ERROR: Could not connect to LANforge to get download file")
|
||||
# exit(2)
|
||||
# if i == 1:
|
||||
# egg.sendline("y")
|
||||
# egg.expect("password:", timeout=5)
|
||||
# egg.sendline("lanforge")
|
||||
# egg.expect(CCPROMPT, timeout=20)
|
||||
# return
|
||||
#
|
||||
# if (args.action == "upload"):
|
||||
# command = "scp %s %s" % (args.value, args.value2)
|
||||
# logg.info("Command[%s]" % command)
|
||||
# egg.sendline(command);
|
||||
#
|
||||
# i = egg.expect(["password:", "Do you want to continue connecting", "Network unreachable"], timeout=5)
|
||||
# if i == 2:
|
||||
# print("Network unreachable, wait 15 seconds and try again.")
|
||||
# time.sleep(15)
|
||||
# command = "scp /tmp/%s %s" % (args.value, args.value2)
|
||||
# logg.info("Command[%s]" % command)
|
||||
# egg.sendline(command);
|
||||
#
|
||||
# i = egg.expect(["password:", "Do you want to continue connecting", "Network unreachable"], timeout=5)
|
||||
# if i == 2:
|
||||
# print("ERROR: Could not connect to LANforge to put upload file")
|
||||
# exit(2)
|
||||
# if i == 1:
|
||||
# egg.sendline("y")
|
||||
# egg.expect("password:", timeout=5)
|
||||
# egg.sendline("lanforge")
|
||||
# egg.expect(CCPROMPT, timeout=20)
|
||||
# return
|
||||
#
|
||||
# if (command is None):
|
||||
# logg.info("No command specified, going to log out.")
|
||||
# else:
|
||||
# logg.info("Command[%s]" % command)
|
||||
# egg.sendline(command);
|
||||
# while True:
|
||||
# try:
|
||||
# i = egg.expect([CCPROMPT, "kmodloader: done loading kernel", "\n"], timeout=TO)
|
||||
# print(egg.before.decode('utf-8', 'ignore'))
|
||||
# if i == 1:
|
||||
# egg.sendline(' ')
|
||||
# egg.expect(CCPROMPT, timeout=20)
|
||||
# print(egg.before.decode('utf-8', 'ignore'))
|
||||
# if i == 2: # new line of text, just print and continue
|
||||
# continue
|
||||
#
|
||||
# if not wait_forever:
|
||||
# break
|
||||
#
|
||||
# except Exception as e:
|
||||
# # Some commands take a long time (logread -f)
|
||||
# if not wait_forever:
|
||||
# logging.exception(e)
|
||||
# break
|
||||
import sys
|
||||
|
||||
if sys.version_info[0] != 3:
|
||||
print("This script requires Python 3")
|
||||
exit()
|
||||
|
||||
import re
|
||||
import logging
|
||||
import time
|
||||
from time import sleep
|
||||
import pprint
|
||||
import telnetlib
|
||||
import argparse
|
||||
import pexpect
|
||||
|
||||
default_host = "localhost"
|
||||
default_ports = {
|
||||
"serial": None,
|
||||
"ssh": 22,
|
||||
"telnet": 23
|
||||
}
|
||||
NL = "\n"
|
||||
CR = "\r\n"
|
||||
Q = '"'
|
||||
A = "'"
|
||||
FORMAT = '%(asctime)s %(name)s %(levelname)s: %(message)s'
|
||||
prompt = "root@OpenWrt:"
|
||||
|
||||
|
||||
def usage():
|
||||
print("$0 used connect to OpenWrt AP or similar Linux machine:")
|
||||
print("-d|--dest IP address of the OpenWrt AP, for ssh/telnet scheme")
|
||||
print("-o|--port IP port of the OpenWrt AP, for ssh/telnet scheme")
|
||||
print("-t|--tty Serial port, if using serial scheme")
|
||||
print("-u|--user login name")
|
||||
print("-p|--pass password")
|
||||
print("--prompt Prompt to look for when commands are done (default: root@OpenWrt)")
|
||||
print("-s|--scheme (serial|telnet|ssh): connect via serial, ssh or telnet")
|
||||
print("-l|--log file log messages here")
|
||||
print("--action (logread | journalctl | lurk | sysupgrade | download | upload | reboot | cmd | ssh-tunnel")
|
||||
print("--value (option to help complete the action")
|
||||
print("--value2 (option to help complete the action, dest filename for download, passwd for ssh-tunnel")
|
||||
print("-h|--help")
|
||||
|
||||
|
||||
# see https://stackoverflow.com/a/13306095/11014343
|
||||
class FileAdapter(object):
|
||||
def __init__(self, logger):
|
||||
self.logger = logger
|
||||
|
||||
def write(self, data):
|
||||
# NOTE: data can be a partial line, multiple lines
|
||||
data = data.strip() # ignore leading/trailing whitespace
|
||||
if data: # non-blank
|
||||
self.logger.info(data)
|
||||
|
||||
def flush(self):
|
||||
pass # leave it to logging to flush properly
|
||||
|
||||
|
||||
def main():
|
||||
global prompt
|
||||
|
||||
parser = argparse.ArgumentParser(description="OpenWrt AP Control Script")
|
||||
parser.add_argument("-d", "--dest", type=str, help="address of the cisco controller_tests")
|
||||
parser.add_argument("-o", "--port", type=int, help="control port on the controller_tests")
|
||||
parser.add_argument("-u", "--user", type=str, help="credential login/username")
|
||||
parser.add_argument("-p", "--passwd", type=str, help="credential password")
|
||||
parser.add_argument("-P", "--prompt", type=str, help="Prompt to look for")
|
||||
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")
|
||||
parser.add_argument("-l", "--log", type=str, help="logfile for messages, stdout means output to console")
|
||||
parser.add_argument("--action", type=str, help="perform action",
|
||||
choices=["logread", "journalctl", "lurk", "sysupgrade", "sysupgrade-n", "download", "upload",
|
||||
"reboot", "cmd", "ssh-tunnel"])
|
||||
parser.add_argument("--value", type=str, help="set value")
|
||||
parser.add_argument("--value2", type=str, help="set value2")
|
||||
tty = None
|
||||
|
||||
args = None
|
||||
try:
|
||||
args = parser.parse_args()
|
||||
host = args.dest
|
||||
scheme = args.scheme
|
||||
port = args.port
|
||||
# port = (default_ports[scheme], args.port)[args.port != None]
|
||||
user = args.user
|
||||
passwd = args.passwd
|
||||
logfile = args.log
|
||||
tty = args.tty;
|
||||
if (args.prompt != None):
|
||||
prompt = args.prompt
|
||||
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])
|
||||
|
||||
CCPROMPT = prompt
|
||||
|
||||
ser = None
|
||||
egg = None # think "eggpect"
|
||||
try:
|
||||
if (scheme == "serial"):
|
||||
# eggspect = pexpect.fdpexpect.fdspan(telcon, logfile=sys.stdout.buffer)
|
||||
import serial
|
||||
from pexpect_serial import SerialSpawn
|
||||
ser = serial.Serial(tty, 115200, timeout=5)
|
||||
|
||||
egg = SerialSpawn(ser);
|
||||
egg.logfile = FileAdapter(logg)
|
||||
egg.sendline(NL)
|
||||
has_reset = False
|
||||
try:
|
||||
logg.info("prompt: %s user: %s passwd: %s" % (prompt, user, passwd))
|
||||
while True:
|
||||
i = egg.expect([prompt, "Please press Enter to activate", "login:", "Password:", "IPQ6018#"],
|
||||
timeout=3)
|
||||
logg.info("expect-0: %i" % (i))
|
||||
if (i == 0):
|
||||
logg.info("Found prompt, login complete.")
|
||||
break
|
||||
if (i == 1):
|
||||
logg.info("Sending newline")
|
||||
egg.setdline(NL)
|
||||
if (i == 2):
|
||||
logg.info("Sending username: %s" % (user))
|
||||
egg.sendline(user)
|
||||
if (i == 3):
|
||||
logg.info("Sending password: %s" % (passwd))
|
||||
egg.sendline(passwd)
|
||||
if (i == 4): # in bootloader
|
||||
if has_reset:
|
||||
logg.info("ERROR: Have reset once already, back in bootloader?")
|
||||
sys.exit(1)
|
||||
has_reset = True
|
||||
logg.info("In boot loader, will reset and sleep 30 seconds")
|
||||
egg.sendline("reset")
|
||||
time.sleep(30)
|
||||
egg.sendline(NL)
|
||||
|
||||
except Exception as e:
|
||||
# maybe something like 'logread -f' is running?
|
||||
# send ctrl-c
|
||||
egg.send(chr(3))
|
||||
|
||||
elif (scheme == "ssh"):
|
||||
# Not implemented/tested currently. --Ben
|
||||
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(passwd)
|
||||
|
||||
elif (scheme == "telnet"):
|
||||
# Not implemented/tested currently. --Ben
|
||||
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)
|
||||
time.sleep(0.1)
|
||||
egg.sendline(' ')
|
||||
egg.expect('User\:')
|
||||
egg.sendline(user)
|
||||
egg.expect('Password\:')
|
||||
egg.sendline(passwd)
|
||||
egg.sendline('config paging disable')
|
||||
else:
|
||||
usage()
|
||||
exit(1)
|
||||
except Exception as e:
|
||||
logging.exception(e);
|
||||
|
||||
command = None
|
||||
|
||||
CLOSEDBYREMOTE = "closed by remote host."
|
||||
CLOSEDCX = "Connection to .* closed."
|
||||
|
||||
try:
|
||||
egg.expect(CCPROMPT)
|
||||
except Exception as e:
|
||||
egg.sendline(NL)
|
||||
|
||||
TO = 10
|
||||
wait_forever = False
|
||||
|
||||
# Clean pending output
|
||||
egg.sendline("echo __hello__")
|
||||
egg.expect("__hello__")
|
||||
egg.expect(CCPROMPT)
|
||||
|
||||
logg.info("Action[%s] Value[%s] Value2[%s]" % (args.action, args.value, args.value2))
|
||||
|
||||
if (args.action == "reboot"):
|
||||
command = "reboot"
|
||||
TO = 60
|
||||
|
||||
if (args.action == "cmd"):
|
||||
if (args.value is None):
|
||||
raise Exception("cmd requires value to be set.")
|
||||
command = "%s" % (args.value)
|
||||
|
||||
if (args.action == "logread"):
|
||||
command = "logread -f"
|
||||
TO = 1
|
||||
wait_forever = True
|
||||
|
||||
if (args.action == "journalctl"):
|
||||
command = "journalctl -f"
|
||||
TO = 1
|
||||
wait_forever = True
|
||||
|
||||
if (args.action == "lurk"):
|
||||
command = "date"
|
||||
TO = 1
|
||||
wait_forever = True
|
||||
|
||||
if (args.action == "ssh-tunnel"):
|
||||
command = "%s" % (args.value)
|
||||
passwd = "%s" % (args.value2)
|
||||
logg.info("Command[%s]" % command)
|
||||
egg.sendline(command);
|
||||
|
||||
i = egg.expect(["password:", "Do you want to continue connecting"], timeout=5)
|
||||
if i == 1:
|
||||
egg.sendline("y")
|
||||
egg.expect("password:", timeout=5)
|
||||
egg.sendline(passwd)
|
||||
egg.expect(CCPROMPT, timeout=20)
|
||||
return
|
||||
|
||||
if ((args.action == "sysupgrade") or (args.action == "sysupgrade-n")):
|
||||
command = "scp %s /tmp/new_img.bin" % (args.value)
|
||||
logg.info("Command[%s]" % command)
|
||||
egg.sendline(command);
|
||||
|
||||
i = egg.expect(["password:", "Do you want to continue connecting"], timeout=5)
|
||||
if i == 1:
|
||||
egg.sendline("y")
|
||||
egg.expect("password:", timeout=5)
|
||||
egg.sendline("lanforge")
|
||||
egg.expect(CCPROMPT, timeout=20)
|
||||
if (args.action == "sysupgrade-n"):
|
||||
egg.sendline("sysupgrade -n /tmp/new_img.bin")
|
||||
else:
|
||||
egg.sendline("sysupgrade /tmp/new_img.bin")
|
||||
egg.expect("link becomes ready", timeout=100)
|
||||
return
|
||||
|
||||
if (args.action == "download"):
|
||||
command = "scp %s /tmp/%s" % (args.value, args.value2)
|
||||
logg.info("Command[%s]" % command)
|
||||
egg.sendline(command);
|
||||
|
||||
i = egg.expect(["password:", "Do you want to continue connecting", "Network unreachable"], timeout=5)
|
||||
if i == 2:
|
||||
print("Network unreachable, wait 15 seconds and try again.")
|
||||
time.sleep(15)
|
||||
command = "scp %s /tmp/%s" % (args.value, args.value2)
|
||||
logg.info("Command[%s]" % command)
|
||||
egg.sendline(command);
|
||||
|
||||
i = egg.expect(["password:", "Do you want to continue connecting", "Network unreachable"], timeout=5)
|
||||
if i == 2:
|
||||
print("ERROR: Could not connect to LANforge to get download file")
|
||||
exit(2)
|
||||
if i == 1:
|
||||
egg.sendline("y")
|
||||
egg.expect("password:", timeout=5)
|
||||
egg.sendline("lanforge")
|
||||
egg.expect(CCPROMPT, timeout=20)
|
||||
return
|
||||
|
||||
if (args.action == "upload"):
|
||||
command = "scp %s %s" % (args.value, args.value2)
|
||||
logg.info("Command[%s]" % command)
|
||||
egg.sendline(command);
|
||||
|
||||
i = egg.expect(["password:", "Do you want to continue connecting", "Network unreachable"], timeout=5)
|
||||
if i == 2:
|
||||
print("Network unreachable, wait 15 seconds and try again.")
|
||||
time.sleep(15)
|
||||
command = "scp /tmp/%s %s" % (args.value, args.value2)
|
||||
logg.info("Command[%s]" % command)
|
||||
egg.sendline(command);
|
||||
|
||||
i = egg.expect(["password:", "Do you want to continue connecting", "Network unreachable"], timeout=5)
|
||||
if i == 2:
|
||||
print("ERROR: Could not connect to LANforge to put upload file")
|
||||
exit(2)
|
||||
if i == 1:
|
||||
egg.sendline("y")
|
||||
egg.expect("password:", timeout=5)
|
||||
egg.sendline("lanforge")
|
||||
egg.expect(CCPROMPT, timeout=20)
|
||||
return
|
||||
|
||||
if (command is None):
|
||||
logg.info("No command specified, going to log out.")
|
||||
else:
|
||||
logg.info("Command[%s]" % command)
|
||||
egg.sendline(command);
|
||||
while True:
|
||||
try:
|
||||
i = egg.expect([CCPROMPT, "kmodloader: done loading kernel", "\n"], timeout=TO)
|
||||
print(egg.before.decode('utf-8', 'ignore'))
|
||||
if i == 1:
|
||||
egg.sendline(' ')
|
||||
egg.expect(CCPROMPT, timeout=20)
|
||||
print(egg.before.decode('utf-8', 'ignore'))
|
||||
if i == 2: # new line of text, just print and continue
|
||||
continue
|
||||
|
||||
if not wait_forever:
|
||||
break
|
||||
|
||||
except Exception as e:
|
||||
# Some commands take a long time (logread -f)
|
||||
if not wait_forever:
|
||||
logging.exception(e)
|
||||
break
|
||||
#
|
||||
#
|
||||
# # ----- ----- ----- ----- ----- ----- ----- ----- ----- ----- ----- -----
|
||||
# if __name__ == '__main__':
|
||||
# main()
|
||||
if __name__ == '__main__':
|
||||
main()
|
||||
#
|
||||
# ####
|
||||
# ####
|
||||
|
||||
Reference in New Issue
Block a user