verify script now works if root user has password

This commit is contained in:
Alex Kaplan 2015-10-12 16:00:27 -07:00
parent 310f6fe4a0
commit 5b168e4985
1 changed files with 22 additions and 5 deletions

View File

@ -6,16 +6,30 @@ import serial
import re import re
#------------------------------------------------------------------ #------------------------------------------------------------------
def answer_prompt(sio,prompt_to_wait_for,answer_to_write): def answer_prompt(sio,prompt_to_wait_for,answer_to_write,send_cr=True):
#------------------------------------------------------------------ #------------------------------------------------------------------
sio.flush()
prompt_found = False prompt_found = False
data = '' data = ''
while not prompt_found: #if send_cr:
data += sio.read(100); #sio.write(unicode('\n'))
d='something'
while not len(d)==0:
d = sio.read(2000);
data += d
print '-' * 50 print '-' * 50
print ' %d bytes read' % (len(data)) print ' %d bytes read' % (len(data))
print '-' * 50 print '-' * 50
print data print data
while not prompt_found:
d = sio.read(100);
data += d
print '-' * 50
print ' %d bytes read' % (len(data))
print '-' * 50
if(data[:-1].endswith(prompt_to_wait_for)): if(data[:-1].endswith(prompt_to_wait_for)):
sio.write(unicode(answer_to_write+'\n')) sio.write(unicode(answer_to_write+'\n'))
print '-' * 50 print '-' * 50
@ -23,8 +37,10 @@ def answer_prompt(sio,prompt_to_wait_for,answer_to_write):
print '-' * 50 print '-' * 50
prompt_found = True prompt_found = True
else: else:
if send_cr:
sio.write(unicode('\n')) sio.write(unicode('\n'))
sio.flush() sio.flush()
#sys.stdin.readline()
#------------------------------------------------------------------ #------------------------------------------------------------------
def scanfor(sio,regexp_to_scan_for,answer_to_write): def scanfor(sio,regexp_to_scan_for,answer_to_write):
@ -64,6 +80,7 @@ def main():
#login #login
answer_prompt(sio,'login:','root') answer_prompt(sio,'login:','root')
answer_prompt(sio,'Password:','chip',False)
answer_prompt(sio,'#','hwtest') answer_prompt(sio,'#','hwtest')
d=scanfor(sio,r'.*### [^#]+ ###.*','poweroff') d=scanfor(sio,r'.*### [^#]+ ###.*','poweroff')