#!/usr/bin/expect
###########################################################################
# Catfeeder telnet control script                v0.2 07/11/2009          #
# Usage: catfeeder.exp forward <time> | reverse <time> | cutout | reset   #
#                      | test | startforward | startreverse | stop        #
#                                                                         #
# Mathew J. Newton                          http://www.newtonnet.co.uk    #
###########################################################################

log_file -a /var/log/catfeeder.log  ;# Log all output
log_user 0                          ;# Set to 0 for no STDOUT output

# Set variables
set command [lindex $argv 0] ;# Set 'command' to be the first received arg
set time [lindex $argv 1]    ;# Set 'time' to be the second received arg
set password "<removed>"     ;# Set the telnet password
set timestamp [timestamp -format "%Y-%m-%d %X"] ;# Set timestamp format
send_log "\n\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\nStart: $timestamp Received arguments: $command $time\n\n" ;# Stamp logfile
set force_conservative 0      ;# Pause briefly before each 'send'
if {$force_conservative} {
	set send_slow {1 .1}
	proc send {ignore arg} {
		sleep .1
		exp_send -s -- $arg
	}
}
set timeout -1                ;# Disable timeout (waiting for expected output)

# Check that a valid operation argument has been passed - if not then abort
if {$command != "forward" && $command != "reverse" && $command != "cutout" && $command != "reset" && $command != "test" && $command != "startforward" && $command != "startreverse" && $command != "stop"} then {
 send_error "\nAborted. Wrong or missing command - no action taken!\nUsage: catfeeder.exp forward <time> | reverse <time> | cutout | reset\n(Command received: $command)\n"
 exit 1
}

# Common login sequence
 spawn telnet 192.168.0.55
 expect -exact "Press any key to continue."
 #send -- " "
 #expect -exact "Enter Selection:"
 send -- "k"
 expect -exact "CatFeeder>"
 send -- "en\r"
 expect -exact "Enter password:"
 send -- "$password\r"
 expect -exact "CatFeeder#"
 send -- "conf t\r"

# Forward feed control (i.e. port 0/20 on for x seconds)
if {$command == "forward"} {
 expect -exact "CatFeeder(config)#"
 send -- "int e 0/21\r"
 expect -exact "CatFeeder(config-if)#"
 send -- "shut\r"
 expect -exact "CatFeeder(config-if)#" 
 send -- "int e 0/20\r"
 expect -exact "CatFeeder(config-if)#"
 send -- "no shut\r"
 expect -exact "CatFeeder(config-if)#"
 send_log " \[Pausing for $time seconds...\] "
 sleep $time
 send -- "shut\r"
 expect -exact "CatFeeder(config-if)#"
 #puts "Feeding complete."
 send -- "exit\r"
 expect -exact "CatFeeder(config)#"
 send -- "exit\r"
 expect -exact "CatFeeder#"
 send -- "exit\r"
 expect eof
 send_log "\n\[Last action taken: $timestamp - Fed for $time seconds\]\n\n"
 exit 0
}

# Reverse feed control (i.e. port 0/21 on for x seconds)
if {$command == "reverse"} {
 expect -exact "CatFeeder(config)#"
 send -- "int e 0/20\r"
 expect -exact "CatFeeder(config-if)#"
 send -- "shut\r"
 expect -exact "CatFeeder(config-if)#" 
 send -- "int e 0/21\r"
 expect -exact "CatFeeder(config-if)#"
 send -- "no shut\r"
 expect -exact "CatFeeder(config-if)#"
 send_log " \[Pausing for $time seconds...\] "
 sleep $time
 send -- "shut\r"
 expect -exact "CatFeeder(config-if)#"
 send -- "exit\r"
 expect -exact "CatFeeder(config)#"
 send -- "exit\r"
 expect -exact "CatFeeder#"
 send -- "exit\r"
 expect eof
 send_log "\n\[Last action taken: $timestamp - Fed (in reverse) for $time seconds\]\n\n"
 exit 0
}

# Cutout (i.e. port 0/22 on)
if {$command == "cutout"} {
 expect -exact "CatFeeder(config)#"
 send -- "int e 0/22\r"
 expect -exact "CatFeeder(config-if)#"
 send -- "no shut\r"
 expect -exact "CatFeeder(config-if)#"
 send -- "exit\r"
 expect -exact "CatFeeder(config)#"
 send -- "exit\r"
 expect -exact "CatFeeder#"
 send -- "exit\r"
 expect eof
 send_log "\n\[Last action taken: $timestamp - Cutout mode (feeding disabled) - use 'reset' to clear ports\]\n\n"
 exit 0
}

# Reset (i.e. all ports shut down)
if {$command == "reset"} {
 expect -exact "CatFeeder(config)#"
 send -- "int e 0/20\r"
 expect -exact "CatFeeder(config-if)#"
 send -- "shut\r"
 expect -exact "CatFeeder(config-if)#"
 send -- "int e 0/21\r"
 expect -exact "CatFeeder(config-if)#"
 send -- "shut\r"
 expect -exact "CatFeeder(config-if)#"
 send -- "int e 0/22\r"
 expect -exact "CatFeeder(config-if)#"
 send -- "shut\r"
 expect -exact "CatFeeder(config-if)#"
 send -- "exit\r"
 expect -exact "CatFeeder(config)#"
 send -- "exit\r"
 expect -exact "CatFeeder#"
 send -- "exit\r"
 expect eof
 send_log "\n\[Last action taken: $timestamp - All ports reset (feeding enabled)\]\n\n"
 exit 0
}

# Test (Cycle each output on/off (staying open for 2 seconds each))
if {$command == "test"} {
 expect -exact "CatFeeder(config)#"
 send -- "int e 0/20\r"
 expect -exact "CatFeeder(config-if)#"
 send -- "no shut\r"
 expect -exact "CatFeeder(config-if)#"
 send_log " \[Pausing for 2 seconds...\] "
 sleep 2
 send -- "shut\r"
 expect -exact "CatFeeder(config-if)#"
 send -- "int e 0/21\r"
 expect -exact "CatFeeder(config-if)#"
 send -- "no shut\r"
 expect -exact "CatFeeder(config-if)#"
 send_log " \[Pausing for 2 seconds...\] "
 sleep 2
 send -- "shut\r"
 send -- "int e 0/22\r"
 expect -exact "CatFeeder(config-if)#"
 send -- "no shut\r"
 expect -exact "CatFeeder(config-if)#"
 send_log " \[Pausing for 2 second...\] "
 sleep 2
 expect -exact "CatFeeder(config-if)#"
 send -- "shut\r"
 expect -exact "CatFeeder(config-if)#"
 send -- "exit\r"
 expect -exact "CatFeeder(config)#"
 send -- "exit\r"
 expect -exact "CatFeeder#"
 send -- "exit\r"
 expect eof
 send_log "\n\[Last action taken: $timestamp - Test sequence (each port open for 2 seconds in turn)\]\n\n"
 exit 0
}

# Start forward feed (i.e. feed indefinitely until stopped!)
# Resets the reverse port in case it had previously been activated
if {$command == "startforward"} {
 expect -exact "CatFeeder(config)#"
 send -- "int e 0/21\r"
 expect -exact "CatFeeder(config-if)#"
 send -- "shut\r"
 expect -exact "CatFeeder(config-if)#"
 send -- "int e 0/20\r"
 expect -exact "CatFeeder(config-if)#"
 send -- "no shut\r"
 expect -exact "CatFeeder(config-if)#"
 send -- "exit\r"
 expect -exact "CatFeeder(config)#"
 send -- "exit\r"
 expect -exact "CatFeeder#"
 send -- "exit\r"
 expect eof
 send_log "\n\[Last action taken: $timestamp - Feeding (forward) started indefinitely! Use 'stop' to cancel!\]\n\n"
 exit 0
}

# Start reverse feed (i.e. feed indefinitely until stopped!)
# Resets the forward port in case it was had previously been opened
if {$command == "startreverse"} {
 expect -exact "CatFeeder(config)#"
 send -- "int e 0/20\r"
 expect -exact "CatFeeder(config-if)#"
 send -- "shut\r"
 expect -exact "CatFeeder(config-if)#"
 send -- "int e 0/21\r"
 expect -exact "CatFeeder(config-if)#"
 send -- "no shut\r"
 expect -exact "CatFeeder(config-if)#"
 send -- "exit\r"
 expect -exact "CatFeeder(config)#"
 send -- "exit\r"
 expect -exact "CatFeeder#"
 send -- "exit\r"
 expect eof
 send_log "\n\[Last action taken: $timestamp - Feeding (reverse) started indefinitely! Use 'stop' to cancel!\]\n\n"
 exit 0
}

# Stop indefinite feeding (i.e. resets both ports)
if {$command == "stop"} {
 expect -exact "CatFeeder(config)#"
 send -- "int e 0/20\r"
 expect -exact "CatFeeder(config-if)#"
 send -- "shut\r"
 expect -exact "CatFeeder(config-if)#"
 send -- "int e 0/21\r"
 expect -exact "CatFeeder(config-if)#"
 send -- "shut\r"
 expect -exact "CatFeeder(config-if)#"
 send -- "exit\r"
 expect -exact "CatFeeder(config)#"
 send -- "exit\r"
 expect -exact "CatFeeder#"
 send -- "exit\r"
 expect eof
 send_log "\n\[Last action taken: $timestamp - Feeding stopped manually\]\n\n"
 exit 0
}