#!/usr/bin/expect
############################################################################
# WRT54GL Catfeeder control script                         v0.2 22/12/2009 #
# Usage: wrtcatfeeder.exp <command> <time> <address>                       #
# where <command> = forward | reverse | cutout | reset |                   #
#                   startforward | startreverse | stop                     #
#       <time>    = time argument (for 'forward' and 'reverse')            #
#       <address> = IP address of user (sent by the web interface)         #
# (Note: <time> should be set to arbitrary value if <address> is used)     #
#                                                                          #
# Mathew J. Newton                             http://www.newtonnet.co.uk  #
#                                                                          #
# LEDs: White SES = Forward feed, Orange SES = Reverse feed, DMZ = Cutout  #
#                                                                          #
# NOTE: The LEDs on the WRT54GL use inverse logic i.e. ON = 0v, OFF = 3.3v #
#       Hence, to activate a function, turn its associated LED *off*       #
#       Boot scripts on the WRT54GL will set each LED on by default        #
############################################################################

log_file -a /var/log/wrtcatfeeder.log  ;# Log all output
log_user 1                             ;# 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 address [lindex $argv 2] ;# Set 'address' to be the third received arg
set user "<removed>"              ;# Set login username
set deviceip "<removed>"  ;# Set IP address (can be IPv4 or IPv6)
# Note: If you change the address then ensure SSH can still connect because it might sit there querying the new/changed fingerprint (which might stall a script)
set password "<removed>"    ;# Set login password
set timestamp [timestamp -format "%Y-%m-%d %X"] ;# Set timestamp format

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)

send_log "\n\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\#\nStart: $timestamp Received arguments: $command $time\n\n" ;# Stamp logfile

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

# Common login sequence
 # 26/05/2011 - '-o StrictHostKeyChecking=no' added whilst playing with changing IP address on the WRT54GL (IPv6 address particularly) otherwise ssh sits ther asking yes/no to storing the fingerprint for the (new) IP connection
 spawn ssh -o StrictHostKeyChecking=no $user@$deviceip
 expect -exact "password:"
 send -- "$password\r"
 expect -exact ":~\# "

# Forward feed control (i.e. white Cisco LED off for x seconds)
if {$command == "forward"} {
 send -- "echo '0' > /proc/diag/led/ses_white\r"
 expect -exact ":~\# "
 send_log "\[Pausing for $time seconds...\] "
 sleep $time
 send -- "echo '1' > /proc/diag/led/ses_white\r"
 expect -exact ":~\# "
 send -- "exit\r"
 expect eof
 send_log "\n\[ Last action taken: $timestamp - Fed for $time seconds by $address \]\n\n"
 exit 0
}

# Reverse feed control (i.e. orange Cisco LED off for x seconds)
if {$command == "reverse"} {
 send -- "echo '0' > /proc/diag/led/ses_orange\r"
 expect -exact ":~\# "
 send_log "\[Pausing for $time seconds...\] "
 sleep $time
 send -- "echo '1' > /proc/diag/led/ses_orange\r"
 expect -exact ":~\# "
 send -- "exit\r"
 expect eof
 send_log "\n\[ Last action taken: $timestamp - Fed (in reverse) for $time seconds by $address \]\n\n"
 exit 0
}

# Cutout (i.e. DMZ LED off, white/orange LEDs on)
if {$command == "cutout"} {
 send -- "echo '0' > /proc/diag/led/dmz\r"
 expect -exact ":~\# "
 send -- "echo '1' > /proc/diag/led/ses_white\r"
 expect -exact ":~\# "
 send -- "echo '1' > /proc/diag/led/ses_orange\r"
 expect -exact ":~\# "
 send -- "exit\r"
 expect eof
 send_log "\n\[ Last action taken: $timestamp - Cutout mode (feeding disabled) by $address - use 'reset' to clear cutout state \]\n\n"
 exit 0
}

# Reset (i.e. DMZ LED on, white/orange LEDs on)
if {$command == "reset"} {
 send -- "echo '1' > /proc/diag/led/dmz\r"
 expect -exact ":~\# "
 send -- "echo '1' > /proc/diag/led/ses_white\r"
 expect -exact ":~\# "
 send -- "echo '1' > /proc/diag/led/ses_orange\r"
 expect -exact ":~\# "
 send -- "exit\r"
 expect eof
 send_log "\n\[ Last action taken: $timestamp - All ports reset (feeding enabled) by $address \]\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"} {
 send -- "echo '1' > /proc/diag/led/ses_orange\r"
 expect -exact ":~\# "
 send -- "echo '0' > /proc/diag/led/ses_white\r"
 expect -exact ":~\# "
 send -- "exit\r"
 expect eof
 send_log "\n\[ Last action taken: $timestamp - Feeding (forward) started indefinitely by $address ! 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"} {
 send -- "echo '1' > /proc/diag/led/ses_white\r"
 expect -exact ":~\# "
 send -- "echo '0' > /proc/diag/led/ses_orange\r"
 expect -exact ":~\# "
 send -- "exit\r"
 expect eof
 send_log "\n\[ Last action taken: $timestamp - Feeding (reverse) started indefinitely by $address ! Use 'stop' to cancel!\]\n\n"
 exit 0
}

# Stop indefinite feeding (i.e. resets both ports)
if {$command == "stop"} {
 send -- "echo '1' > /proc/diag/led/ses_white\r"
 expect -exact ":~\# "
 send -- "echo '1' > /proc/diag/led/ses_orange\r"
 expect -exact ":~\# "
 send -- "exit\r"
 expect eof
 send_log "\n\[ Last action taken: $timestamp - Feeding stopped manually by $address \]\n\n"
 exit 0
}

# We shouldn't get this far so if we did then there's a problem
send_log "\n\n\[ WARNING: There's a problem with the control script - we really shouldn't have got this far because we should only be acting on known commands and then actively exiting. Command received: $command $time $address \]\n\n"
exit 1