95 строки
2.9 KiB
Python
Executable File
95 строки
2.9 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
import argparse
|
|
import pexpect
|
|
import sys
|
|
import time
|
|
|
|
|
|
def main():
|
|
|
|
parser = argparse.ArgumentParser(description='test_login cmdline parser')
|
|
parser.add_argument('-u', default="admin", help='login user name')
|
|
parser.add_argument('-P', default="YourPaSsWoRd", help='login password')
|
|
parser.add_argument('-N', default="Test@2022", help='new password')
|
|
parser.add_argument('-p', type=int, default=9000, help='local port')
|
|
|
|
args = parser.parse_args()
|
|
|
|
login_prompt = 'sonic login:'
|
|
passwd_prompt = 'Password:'
|
|
cmd_prompt = "{}@sonic:~\$ $".format(args.u)
|
|
grub_selection = "The highlighted entry will be executed"
|
|
firsttime_prompt = 'firsttime_exit'
|
|
passwd_change_prompt = ['Current password:', 'New password:', 'Retype new password:']
|
|
|
|
i = 0
|
|
while True:
|
|
try:
|
|
p = pexpect.spawn("telnet 127.0.0.1 {}".format(args.p), timeout=600, logfile=sys.stdout, encoding='utf-8')
|
|
break
|
|
except Exception as e:
|
|
print(str(e))
|
|
i += 1
|
|
if i == 10:
|
|
raise
|
|
time.sleep(1)
|
|
|
|
# select default SONiC Image
|
|
p.expect(grub_selection)
|
|
p.sendline()
|
|
# bootup sonic image
|
|
while True:
|
|
i = p.expect([login_prompt, passwd_prompt, firsttime_prompt, cmd_prompt])
|
|
if i == 0:
|
|
# send user name
|
|
p.sendline(args.u)
|
|
elif i == 1:
|
|
# send password
|
|
p.sendline(args.P)
|
|
# Check for password change prompt
|
|
try:
|
|
p.expect('Current password:', timeout=2)
|
|
except pexpect.TIMEOUT:
|
|
break
|
|
else:
|
|
# send old password for password prompt
|
|
p.sendline(args.P)
|
|
p.expect(passwd_change_prompt[1])
|
|
# send new password
|
|
p.sendline(args.N)
|
|
p.expect(passwd_change_prompt[2])
|
|
# retype new password
|
|
p.sendline(args.N)
|
|
time.sleep(1)
|
|
# Restore default password
|
|
p.sendline('passwd {}'.format(args.u))
|
|
p.expect(passwd_change_prompt[0])
|
|
p.sendline(args.N)
|
|
p.expect(passwd_change_prompt[1])
|
|
p.sendline(args.P)
|
|
p.expect(passwd_change_prompt[2])
|
|
p.sendline(args.P)
|
|
break
|
|
elif i == 2:
|
|
# fix a login timeout issue, caused by the login_prompt message mixed with the output message of the rc.local
|
|
time.sleep(1)
|
|
p.sendline()
|
|
else:
|
|
break
|
|
|
|
# check version
|
|
time.sleep(5)
|
|
p.sendline('uptime')
|
|
p.expect([cmd_prompt])
|
|
p.sendline('show version')
|
|
p.expect([cmd_prompt])
|
|
p.sendline('show ip bgp sum')
|
|
p.expect([cmd_prompt])
|
|
p.sendline('sync')
|
|
p.expect([cmd_prompt])
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|