CCIE.pl

Kod: Zaznacz cały

#!/usr/bin/env python 

import paramiko
import time

host = '1.1.1.1'
user = 'admin'
pass = 'password'
interface = 'fa0/1'

connect = paramiko.SSHClient()
connect.set_missing_host_key_policy(paramiko.AutoAddPolicy())
connect.connect(host, username=user, password=pass, look_for_keys=False, allow_agent=False)


shell = connect.invoke_shell()

shell.send('\n')
time.sleep(2)
shell.send('conf t\n')
time.sleep(2)
shell.send('int %s\n') % interface
time.sleep(2)
shell.send('shut\n') 
time.sleep(2)
shell.send('end\n') 
connect.close()

Kod: Zaznacz cały

def do_enable(self):
		output = ''
		self.remote_conn.send('enable\n')
		output = self.wait_for_password()
		if not "ERROR" in output:
			if opt.enable:
				self.remote_conn.send(opt.enable+'\n')
				output = self.wait_for_enable()
				if not "ERROR" in output:
					print "[+] Enable password correct"
					temp = output.splitlines()
					self.prompt = temp[-1]
					print self.prompt
					if '#' in self.prompt:
						#print "True returned"
						return True
				else:
					print "[-] Enable failed"
					return False
		if '#' in self.prompt:
				return True
						
	'''Core method used to check prompts each time when command is sent
	   Includes while loop that recievs packets and check prompt '''

	def wait_for_output(self,new_prompt):
		complete = False
		temp = ''
		packet = ''
		while not complete:
			packet += self.remote_conn.recv(65535)
			#print "IN WAIT FOR OUTPUT PROMPT: ",self.prompt
			'''!!!! print packet important for debbuging problems !!!'''
			#print packet				
			if new_prompt in packet:
				self.prompt = new_prompt
				#print "NEW_PROMPT:",self.prompt
				complete = True
			if self.prompt in packet:
				#print "Error"
				#print "SELF PROMPT:",self.prompt
				complete = True
		#print packet
		return packet

	'''Method used to check if "Password" appears after enable command'''

	def wait_for_password(self):
		complete = False
		temp = ''
		packet = ''
		#print "wait_for_password"
		while not complete:
			packet += self.remote_conn.recv(65535)
			#print "PACKET: ",packet
			if "Password" in packet:
				complete = True
			if "%" in packet:
				complete = True
				return "ERROR"
			if "#" in packet:
				complete =True
		#print packet
		return packet

	'''Method checks prompt for enable password'''
	
	def wait_for_enable(self):
		complete = False
		temp = ''
		packet = ''
		#print "wait_for_password"
		while not complete:
			packet += self.remote_conn.recv(65535)
			#print "PACKET: ",packet
			if "#" in packet:
				complete = True
			if ">" in packet:
				complete = True
				return "ERROR"
		return packet

Kod: Zaznacz cały

#!/usr/bin/expect

set user user
set workingdir .
set pass haslo
set LOG [open "$workingdir/RESULT" a]
set RTR [open "$workingdir/RTR_LIST" r]
set PROMPT "#|]"
set timeout 60000
match_max -d 2000000

while {[gets $RTR router] != -1} {
   if {[ string range $router 0 0 ] != "#"} {
set timeout 30
#set LOG [open "$workingdir/RESULT" a]
spawn ssh $user@$router
expect "password:"
send "$pass\r"
expect -re $PROMPT

send "sh ver | inc System image \r"
expect -re $PROMPT
puts $LOG "$expect_out(buffer)"
send "sh ver | inc uptime \r"
expect -re $PROMPT
puts $LOG "$expect_out(buffer)"
send "quit \r"
    }
}
close $RTR; close $LOG