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/>.

# ------------------------------------------------------------------------------

# Start up a batch of linux sessions ready for testing.

# There are a number of ways of starting up vmlinux to force a particular IP

# address to be used, which are specified by the first argument

#   DHCP   - Linux will get an IP address via DHCP. In this case we must poll

#            the router to get the newly created IP address.

#   STATIC - We force Linux to chose a static address, by an explicit ifconfig

#            in the etc/init.d/rcS of the initramfs. We need to select an IP

#            address that is not currently being used.

# In both cases, potentially we get in a mess if someone else is starting up

# sessions at this time.

# ------------------------------------------------------------------------------

# Tell the user how to use this command

# If we are doing STATIC startup, the <count> must be followed by <count> IP

# addresses.

# ------------------------------------------------------------------------------

function bad_args {

    echo "Usage: start-linuxes.sh DHCP|STATIC <count> <dir> [ <ip> <ip> ... ]"

    echo "  <count>   - Number of Linux sessions to start up"

    echo "  <dir>     - Linux directory"

    echo "  <ip>      - IP addresses (only if STATIC)"

}

# ------------------------------------------------------------------------------

# Check the args

# Remote machine name is optional. For argument usage, see function bad_args.

# @param[in] $1      Type of processing (DHCP or STATIC)

# @pmara[in] $2      Count of Linux instances to start

# @param[in] $3      Linux directory

# @param[in] $4 ...  IP addresses if static startup

# ------------------------------------------------------------------------------

function get_args {

    # Check We have at least the first two arguments

    if [ $# -lt 3 ]

    then

        bad_args

        exit  1

    fi

    type=$1

    shift

    count=$1

    shift

    linux_dir=$1

    shift

    # Check type and get static IP addresses

    case "x${type}" in

        "xSTATIC" )

            # Get all the IP addresses

            for (( i=0 ; i < ${count} ; i++ ))

            do

                if [ "x${i}" == "x" ]

                then

                    bad_args

                    exit  1

                fi

                ip_list[${i}]=$1

                shift

            done

            ;;

        "xDHCP" )

            # No extra args for DHCP

            ;;

        "x*" )

            # Not recognized

            bad_args

            exit  1

            ;;

    esac

}

# ------------------------------------------------------------------------------

# Allocate an unused TAP device

# We need /dev/tap?? that is not being used at present. We have a custom

# program to do this.

# @return  The TAP to use.

# ------------------------------------------------------------------------------

function allocate_tap {

    for (( i=0 ; ${i} < 20 ; i=${i} + 1 ))

    do

        if ./check-tap "tap${i}"

        then

            echo "tap${i}"

            exit

        fi

    done

}

# ------------------------------------------------------------------------------

# Start up one STATIC instance

# We need to edit the IP address into rcS, rebuild Linux and start Linux using

# the next available TAP.

# This could go wrong if someone else starts up a machine at the same time. We

# could reserve an unlikely block to avoid this, and flock this code, to avoid

# another instance causing trouble. For now we leave it unprotected.

# @param[in] $1  An index number of this instance

# @param[in] $2  The IP address to use

# @return  The Linux process ID

# ------------------------------------------------------------------------------

function start_one_static {

    ip=$2

    # A new Linux instance will require a new TAP in its config

    tap=`allocate_tap $*`

    # Edit the Or1ksim configuration file to use the allocated TAP

    sed -i -e "s/^\( *tap_dev *= *\).*$/\1\"$tap\"/" \

        ${linux_dir}/arch/openrisc/or1ksim_eth.cfg

    # Edit the Linux rcS file and rebuild

    sed -i -e "s/ifconfig eth0.*up/ifconfig eth0 ${ip} up/" \

        ${linux_dir}/arch/openrisc/support/initramfs/etc/init.d/rcS

    cd ${linux_dir}

    make vmlinux ARCH=openrisc CROSS_COMPILE=/opt/or32-new/bin/or32-elf- \

        > /dev/null 2>&1

    if [ $? != 0 ];

    then

        echo "start-linux.sh: 'make vmlinux' failed"

        exit 1

    fi

    cd - > /dev/null 2>&1

    # Start up Linux and allow time for the ramdisk to unpack, by waiting

    # until ping to the IP address is OK.

    or32-elf-sim -f ${linux_dir}/arch/openrisc/or1ksim_eth.cfg \

        ${linux_dir}/vmlinux \

        > `hostname`-linux-console-$$-$1.log 2>&1 & linux_pid=$!

    until ping -c 1 -q ${ip} > /dev/null 2>&1

    do

        sleep 1

    done

    # make us nicer than normal, then return the new process ID

    renice +1 ${linux_pid} > /dev/null 2>&1

    echo "${linux_pid}"

}

# ------------------------------------------------------------------------------

# Start up one DHCP instance

# Start up Linux using the next available TAP.

# @param[in] $1  An index number of this instance

# @return  The Linux process ID

# ------------------------------------------------------------------------------

function start_one_dhcp {

    # A new Linux instance will require a new TAP in its config

    tap=`allocate_tap $*`

    # Edit the Or1ksim configuration file to use the allocated TAP

    sed -i -e "s/^\( *tap_dev *= *\).*$/\1\"$tap\"/" \

        ${linux_dir}/arch/openrisc/or1ksim_eth.cfg

    # Start up Linux

    or32-elf-sim -f ${linux_dir}/arch/openrisc/or1ksim_eth.cfg \

        ${linux_dir}/vmlinux > `hostname`-linux-output-$$-$1 2>&1 & linux_pid=$!

    # make us nicer than normal, then return the new process ID

    renice +1 ${linux_pid} > /dev/null 2>&1

    echo "${linux_pid}"

}

# ------------------------------------------------------------------------------

# Start up using DHCP on the local machine

# We wait for 30 seconds between each Linux startup, to avoid flooding the

# DHCP server.

# @return  A list of IP address of the Linux instances

# ------------------------------------------------------------------------------

function start_all_dhcp {

    echo "DHCP startup in progress..."

    for (( i=0 ; ${i} < ${count} ; i=${i} + 1 ))

    do

        one_res=`start_one_dhcp ${i}`

        res="${res} ${one_res}"

        sleep 30

    done

    echo ${res}

}

# ------------------------------------------------------------------------------

# Start up using STATIC on the local machine

# ------------------------------------------------------------------------------

function start_all_static {

    echo "STATIC startup in progress..."

    for (( i=0 ; ${i} < ${count} ; i=${i} + 1 ))

    do

        one_res=`start_one_static ${i} ${ip_list[${i}]}`

        res="${res} ${one_res}"

    done

    echo ${res}

}

# ------------------------------------------------------------------------------

# Main program

# ------------------------------------------------------------------------------

# Some useful files

tmp1=/tmp/start-linuxes-1-$$

tmp2=/tmp/start-linuxes-2-$$

# Check the arguments

get_args $*

# Start Linuxes

case "${type}" in

    "DHCP" )

        start_all_dhcp

        ;;

    "STATIC" )

        start_all_static

        ;;

    "*" )

        echo "Abort at line ${LINENO} in ${BASH_SOURCE[0]}"

        exit 255

        ;;

esac