Minor fix in shell mode

1 files changed, 13 insertions, 3 deletions

diff --git a/rex.exp b/rex.exp
index 42026f2..312beb7 100755
--- a/rex.exp
+++ b/rex.exp
@@ -1677,7 +1677,10 @@ if {![config_option copy]} {
     exit [eval runcprev $config(command)]
 }
 
-if {![info exists config(hosts)] || [llength $config(hosts)] == 0} {
+if {[lindex $config(command) 0] == "shell" && [llength $config(command)] == 2} {
+    lappend config(hosts) [lindex $config(command) 1]
+    set config(option,shell) 1
+} elseif {![info exists config(hosts)] || [llength $config(hosts)] == 0} {
     terror "no host list"
     listgroups
     exit 1
@@ -1710,9 +1713,16 @@ if [config_option copy] {
 	debug 1 "... in sudo mode"
     }
     debug 1 "... to $config(hosts)"
-} elseif {[regexp {shell[ \t][ \t]*(.*)} "$config(command)" dummy host]} {
+} elseif [config_option shell] {
     global config
-    debug 2 "entering shell mode"
+
+    if {[llength $config(hosts)] > 1} {
+	terror "only one hostname is allowed in shell mode"
+	exit 1
+    }
+    set host [lindex $config(hosts) 0]
+    
+    debug 2 "entering shell on $host"
     set password_sent 0
     spawn -noecho ssh [hostuser $host]@$host
     expect {