Subversion Repositories openrisc_2011-10-31

#!/bin/bash
 
# Copyright (C) 2011 Embecosm Limited
 
# Contributor Jeremy Bennett <jeremy.bennett@embecosm.com>
 
# This file is a script to start up a group of Linux instances
 
# 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 3 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, see <http://www.gnu.org/licenses/>.          
 
# ------------------------------------------------------------------------------
 
# See if telnet or ftp is working
 
# We call the program with a sequence
 
# command (telnet or ftp)
# wait for prompt ("telnet> "or "ftp> ")
# send open <ip address>
# wait for prompt ("login: " or "Name (<ip address>:<user>): ")
# send root
# wait for prompt ("# " or "ftp> ")
# send exit
# wait for closing message ("Connection closed by foreign host." or
#                           "221 Operation successful")
 
# @param[in] $1  The command to use (ftp or telnet)
# @param[in] $2  The IP address
# @param[in] $3  The connection prompt (regular expression)
# @param[in] $4  The operational prompt (regular expression)
# @param[in] $5  The closing message (regular expression)
 
# @return  0 on success, non-zero otherwise.
function check_remote_command {
 
    command=$1
    ip=$2
    conn_prompt=$3
    op_prompt=$4
    close_mess=$5
 
    expect -f - <<EOF > /dev/null 2>&1
    spawn "${command}"
 
    set timeout 30
    expect {
        "${command}> " {}
        timeout    {exit 1}
    }
 
    send "open ${ip}\n"
 
    expect {
        -re "${conn_prompt}" {}
        timeout   {exit 2}
    }
 
    send "root\n"
 
    expect {
        -re "${op_prompt}" {}
        timeout   {exit 3}
    }
 
    send "exit\n"
 
    expect {
        -re "${close_mess}" {exit 0}
        timeout {exit 4}
    }
EOF
 
    return $?
}
 
 
# See if telnet is working.
 
# We call check_remote_command with appropriate arguments
 
# @param[in] $1  The IP address
 
# @return  0 on success, non-zero otherwise.
function check_telnet {
    check_remote_command telnet $1 "^.* login: " "^.*# " \
        "Connection closed by foreign host."
 
    return $?
}
 
 
# See if FTP is working.
 
# We call check_remote_command with appropriate arguments
 
# @param[in] $1  The IP address
 
# @return  0 on success, non-zero otherwise.
 
function check_ftp {
    check_remote_command ftp $1 "Name \\\($1:.*\\\): " "ftp> " \
        "221 Operation successful"
 
    return $?
}
 
 
# See if telnet and FTP are working.
 
# We combine calls to check_telnet and check_ftp. Note that we do check both,
# even though we really only need to check one for failure. However in the
# future the additional information may be useful.
 
# @param[in] $1  The IP address
 
# @return  0 on success, non-zero (10 x telnet failure + ftp failure)
#          otherwise.
function check_telnet_ftp {
    check_telnet $1
    res_telnet=$?
    check_ftp $1
    res_ftp=$?
 
    return $(( $res_telnet * 10 + $res_ftp ))
}
 
 
# IP list excluding all known IPs
ip_list="192.168.0.2-39,41-59,61-79,82-127,129,130,133-255"
ip_file=`dirname ${DEJAGNU}`/ip-avail.txt
tmp_file=/tmp/find-linuxes-$$
 
# Keep using nmap until we get the same number of IP addresses twice running.
nmap -sP -n ${ip_list} | \
    sed -n -e 's/Nmap scan report for //p' | \
    sort > ${ip_file}
len1=`wc -l ${ip_file} | cut -d " " -f 1`
nmap -sP -n ${ip_list} | \
    sed -n -e 's/Nmap scan report for //p' | \
    sort > ${ip_file}
len2=`wc -l ${ip_file} | cut -d " " -f 1`
 
while [ ${len1} -ne ${len2} ]
do
    len1=${len2}
    nmap -sP -n ${ip_list} | \
	sed -n -e 's/Nmap scan report for //p' | \
	sort > ${ip_file}
    len2=`wc -l ${ip_file} | cut -d " " -f 1`
done
 
# If we have a -c flag, check that the targets respond to telnet and ftp
if [ $# > 0 -a "x$1" == "x-c" ]
then
    mv ${ip_file} ${tmp_file}
    touch ${ip_file}
 
    len=`wc -l ${tmp_file} | cut -d " " -f 1`
    while [ ${len} -gt 0 ]
    do
	# Get each IP address in turn
	ip=`tail -${len} ${tmp_file} | head -1`
	len=$(( ${len} - 1 ))
 
	check_telnet_ftp ${ip}
	res=$?
 
	if [ ${res} -eq 0 ]
	then
	    echo ${ip} >> ${ip_file}
	else
	    echo "${ip} failed to respond to telnet/FTP: result ${res}"
	fi
    done
 
    rm ${tmp_file}
fi
 
# Print the results
cat ${ip_file}