Subversion Repositories or1k

# Copyright (C) 97, 98, 1999 Free Software Foundation, Inc.

# This program is free software; you can redistribute it and/or modify

# it under the terms of the GNU General Public License as published by

# the Free Software Foundation; either version 2 of the License, or

# (at your option) any later version.

#

# This program is distributed in the hope that it will be useful,

# but WITHOUT ANY WARRANTY; without even the implied warranty of

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

# GNU General Public License for more details.

#

# You should have received a copy of the GNU General Public License

# along with this program; if not, write to the Free Software

# Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.

# Please email any bugs, comments, and/or additions to this file to:

# DejaGnu@cygnus.com

#

# Connect using telnet. This takes two arguments. The first one is the

# hostname, and the second is the optional port number. This sets

# the fileid field in the config array, and returns -1 for error, or the

# spawn id.

#

proc telnet_open { hostname args } {

    global verbose

    global connectmode

    global spawn_id

    global timeout

    global board_info

    set raw 0;

    if { [llength $args] > 0 } {

        if { [lindex $args 0] == "raw" } {

            set raw 1;

        }

    }

    set port 23

    if [board_info $hostname exists name] {

        set connhost [board_info $hostname name]

    } else {

        set connhost $hostname

    }

    if [board_info $connhost exists hostname] {

        set hostname [board_info $connhost hostname];

    }

    if [file exists /usr/kerberos/bin/telnet] {

        set telnet /usr/kerberos/bin/telnet;

    } else {

        set telnet telnet;

    }

    # Instead of unsetting it, let's return it. One connection at a

    # time, please.

    if [board_info $connhost exists fileid] {

        return [board_info $connhost fileid];

    }

    # get the hostname and port number from the config array

    if [board_info $connhost exists netport] {

        set type $hostname

        set hosttmp [split [board_info $connhost netport] ":"]

        set hostname [lindex $hosttmp 0]

        if { [llength $hosttmp] > 1 } {

            set port [lindex $hosttmp 1]

        }

        unset hosttmp

    } else {

        set type target

    }

    if [board_info $connhost exists shell_prompt] {

        set shell_prompt [board_info $connhost shell_prompt]

    }

    if ![info exists shell_prompt] {    # if no prompt, then set it to something generic

        set shell_prompt ".*> "

    }

    set tries 0

    set result -1

    set need_respawn 1;

    verbose "Starting a telnet connection to $hostname:$port $shell_prompt" 2

    while { $result < 0 && $tries <= 3 } {

        if { $need_respawn } {

            set need_respawn 0;

            spawn $telnet $hostname $port;

        }

        expect {

            "Trying " {

                exp_continue;

            }

            -re "$shell_prompt.*$" {

                verbose "Got prompt\n"

                set result 0

            }

            -re "nt Name:|ogin:" {

                if [board_info $connhost exists telnet_username] {

                    exp_send "[board_info $connhost telnet_username]\n";

                    exp_continue;

                }

                if [board_info $connhost exists username] {

                    exp_send "[board_info $connhost username]\n";

                    exp_continue;

                }

                perror "telnet: need to login"

                break

            }

            "assword:" {

                if [board_info $connhost exists telnet_password] {

                    exp_send "[board_info $connhost telnet_password]\n";

                    exp_continue;

                }

                if [board_info $connhost exists password] {

                    exp_send "[board_info $connhost password]\n";

                    exp_continue;

                }

                perror "telnet: need a password"

                break

            }

            -re "advance.*y/n.*\\?" {

                exp_send "n\n";

                exp_continue;

            }

            -re {([Aa]dvanced|[Ss]imple) or ([Ss]imple|[Aa]dvanced)} {

                exp_send "simple\n";

                exp_continue;

            }

            "Connected to" {

                exp_continue

            }

            "unknown host" {

                exp_send "\003"

                perror "telnet: unknown host"

                break

            }

            "VxWorks Boot" {

                exp_send "@\n";

                sleep 20;

                exp_continue;

            }

            -re "Escape character is.*\\.\[\r\n\]" {

                if { $raw || [board_info $connhost exists dont_wait_for_prompt] } {

                    set result 0;

                } else {

                    if [board_info $connhost exists send_initial_cr] {

                        exp_send "\n"

                    }

                    exp_continue

                }

            }

            "has logged on from" {

                exp_continue

            }

            "You have no Kerberos tickets" {

                warning "telnet: no kerberos Tickets, please kinit"

                break

            }

            -re "Connection refused.*$" {

                catch "exp_send \"\003\"" foo;

                sleep 5;

                warning "telnet: connection refused."

            }

            -re "Sorry, this system is engaged.*" {

                exp_send "\003"

                warning "telnet: already connected."

            }

            "Connection closed by foreign host.*$" {

                warning "telnet: connection closed by foreign host."

                break

            }

            -re "\[\r\n\]+" {

                exp_continue

            }

            timeout {

                exp_send "\n"

            }

            eof {

                warning "telnet: got unexpected EOF from telnet."

                catch close;

                catch wait;

                set need_respawn 1;

                sleep 5;

            }

        }

        incr tries

    }

    # we look for this here again cause it means something went wrong, and

    # it doesn't always show up in the expect in buffer till the server times out.

    if [info exists expect_out(buffer)] {

        if [regexp "assword:|ogin:" $expect_out(buffer)] {

            perror "telnet: need to supply a login and password."

        }

    }

    if { $result < 0 } {

        catch close

        catch wait

        set spawn_id -1

    }

    if { $spawn_id >= 0 } {

        verbose "setting board_info($connhost,fileid) to $spawn_id" 3

        set board_info($connhost,fileid) $spawn_id

    }

    return $spawn_id

}

#

# Put the telnet connection into binary mode.

#

proc telnet_binary { hostname } {

    if [board_info $hostname exists fileid] {

        remote_send $hostname "";

        remote_expect $hostname 5 {

            -re "telnet> *$" {}

            default {}

        }

        remote_send $hostname "set binary\n"

        remote_expect $hostname 5 {

            -re "Format is .*telnet> *$" {

                remote_send $hostname "toggle binary\n";

                exp_continue;

            }

            -re "Negotiating network ascii.*telnet> *$" {

                remote_send $hostname "toggle binary\n";

                exp_continue;

            }

            -re "Negotiating binary.*\[\r\n\].*$" { }

            -re "binary.*unknown argument.*telnet> *$" {

                remote_send $hostname "mode character\n";

            }

            -re "Already operating in binary.*\[\r\n\].*$" { }

            timeout {

                warning "Never got binary response from telnet."

            }

        }

    }

}

proc telnet_transmit { dest file args } {

    return [standard_transmit $dest $file];

}