| #!/bin/bash |
| |
| ## |
| # Copyright 2015 Telefónica Investigación y Desarrollo, S.A.U. |
| # This file is part of openvim |
| # All Rights Reserved. |
| # |
| # Licensed under the Apache License, Version 2.0 (the "License"); you may |
| # not use this file except in compliance with the License. You may obtain |
| # a copy of the License at |
| # |
| # http://www.apache.org/licenses/LICENSE-2.0 |
| # |
| # Unless required by applicable law or agreed to in writing, software |
| # distributed under the License is distributed on an "AS IS" BASIS, WITHOUT |
| # WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the |
| # License for the specific language governing permissions and limitations |
| # under the License. |
| # |
| # For those usages not covered by the Apache License, Version 2.0 please |
| # contact with: nfvlabs@tid.es |
| ## |
| |
| #Get configuration of a host for using it as a compute node |
| |
| function usage(){ |
| |
| echo -e "\nUsage: $0 --user-name <user> --ip=<X.X.X.X> --cores=<core_number> --huge-pages-1G=<huga_pages_number> --nb-10GB-interfaces==<interface_number>" |
| echo -e "Generate a develop host yaml to be used for openvim host-add\n" |
| echo -e " --user-name -u <user>* Server OAM Ip" |
| echo -e " --ip -i <ip>* Server hostname" |
| echo -e " --cores -c <cores> Numa Cores available must be an odd number and bigger or equal to 4." |
| echo -e " --huge-pages-1G -hp <huge_pages_number> Must be an odd number and bigger or equal to 16. 4GiB of memory will be reserved for the host OS, the rest will be used by VM." |
| echo -e " --nb-10GB-interfaces -ni <nb-10GB-interfaces> Dataplane interfaces must be an odd number and bigger or equal to 4." |
| echo -e " * Mandatory arguments" |
| echo -e "\n" |
| echo -e "The output will be a server descriptor with two numas and resources (memory, cores and interfaces) equally distributed between them." |
| echo -e "Each interface (physical funtion) will have defined 8 SR-IOV (virtual functions)." |
| |
| exit 1 |
| } |
| |
| function get_hash_value() { echo `eval echo $\{\`echo $USER[$IP]\`\}`; } |
| |
| function get_mac(){ |
| seed=$1 |
| b1=$((seed%16)); seed=$((seed/16)) |
| b2=$((seed%16)); seed=$((seed/16)) |
| b3=$((seed%16)); seed=$((seed/16)) |
| b4=$((seed%16)); seed=$((seed/16)) |
| b5=$((seed%16)); seed=$((seed/16)) |
| mac=`printf "%02X:%02X:%02X:%02X:%02X:%02X" 2 $b5 $b4 $b3 $b2 $b1` |
| echo $mac |
| } |
| |
| function _parse_opts(){ |
| |
| #help argument |
| if [ -n "$option_help" ]; |
| then |
| usage |
| fi |
| |
| #User argument |
| [ -z "$option_user_name" ] && echo -e "ERROR: User argument is mandatory, --user-name=<user>\n" && usage |
| USER_NAME=${option_user_name} |
| |
| [ -z "$option_ip" ] && echo -e "ERROR: OAM IP argument is mandatory, --ip=<X.X.X.X>\n" && usage |
| IP=${option_ip} |
| |
| #TODO to be checl diference between real cores an numa cores |
| #cores argument |
| REAL_CORES=$(grep -c "^processor" "/proc/cpuinfo") |
| if [ -z "$option_cores" ] ; then |
| CORES=REAL_CORES |
| else |
| CORES=${option_cores} |
| fi |
| |
| #Ensure the core user input is big enough |
| ([ $((CORES%2)) -ne 0 ]) && echo -e "ERROR: Wrong number of cores\n" && usage |
| |
| MEMORY=$(($(grep MemTotal /proc/meminfo | awk '{print $2}') /1024/1024)) |
| if [ -z "$option_huge_pages_1G" ] ; then |
| HUGE_PAGES_MEMORY=0 |
| else |
| HUGE_PAGES_MEMORY=${option_huge_pages_1G} |
| fi |
| #Ensure the memory user input is big enough |
| #([ $((MEMORY%2)) -ne 0 ] || [ $MEMORY -lt 16 ] ) && echo -e "ERROR: Wrong number of memory\n" && usage |
| |
| #nb_10GB_interfaces argument |
| if [ -z "$nb_10GB_interfaces" ] ; then |
| INTERFACES=8 |
| else |
| INTERFACES=${nb_10GB_interfaces} |
| fi |
| ([ $((INTERFACES%2)) -ne 0 ] || [ $INTERFACES -lt 4 ] ) && echo -e "ERROR: Wrong number of interfaces\n" && usage |
| |
| # Parameter by default |
| NUMAS=1 |
| ([ $((NUMAS%2)) -ne 0 ]) && NUMAS=1 |
| } |
| |
| function _generate_compute_develope_yaml(){ |
| |
| _yaml_init |
| |
| FEATURES_LIST="lps,dioc,hwsv,tlbps,ht,lps,64b,iommu" |
| |
| #Generate a cpu topology for 4 numas with hyperthreading |
| #in this developing/fake server all cores can be used |
| #TODO check if this calculation is correct |
| echo2file "#This file was created by $0" |
| echo2file "#for adding this compute node to openvim" |
| echo2file "#copy this file to openvim controller and run" |
| echo2file "#openvim host-add <this>" |
| echo2file |
| echo2file "host:" |
| echo2file " name: $HOST_NAME" |
| echo2file " user: $USER_NAME" |
| echo2file " ip_name: $IP" |
| echo2file "host-data:" |
| echo2file " name: $HOST_NAME" |
| echo2file " user: $USER_NAME" |
| echo2file " ip_name: $IP" |
| echo2file " ranking: 100" |
| echo2file " description: $HOST_NAME" |
| echo2file " features: $FEATURES_LIST" |
| echo2file " numas:" |
| |
| numa=0 |
| last_iface=0 |
| iface_counter=0 |
| |
| while [ $numa -lt $NUMAS ] |
| do |
| |
| HUGE_PAGES=$((HUGE_PAGES_MEMORY/2-2)) |
| ([ ${HUGE_PAGES} -lt -1 ]) && HUGE_PAGES=0 |
| |
| echo2file " - numa_socket: $numa" |
| #MEMORY |
| echo2file " hugepages: $((HUGE_PAGES))" |
| echo2file " memory: $((MEMORY/$NUMAS))" |
| #CORES |
| echo2file " cores:" |
| |
| for((n_core=0;n_core<$REAL_CORES;n_core++)) |
| do |
| THREAD_ID=$(($n_core+1)) |
| CORE_ID=$(($((${n_core}+${numa}))/2)) |
| echo2file " - core_id: ${CORE_ID}" |
| echo2file " thread_id: ${THREAD_ID}" |
| [ $CORE_ID -eq 0 ] && echo2file " status: noteligible" |
| |
| thread_counter=$((thread_counter+1)) |
| done |
| # GENERATE INTERFACES INFORMATION AND PRINT IT |
| seed=$RANDOM |
| echo2file " interfaces:" |
| for ((iface=0;iface<$INTERFACES;iface+=2)) |
| do |
| name="iface$iface_counter" |
| bus=$((iface+last_iface)) |
| bus=$((iface)) |
| pci=`printf "0000:%02X:00.0" $bus` |
| mac=`get_mac $seed` |
| seed=$((seed+1)) |
| |
| echo2file " - source_name: $name" |
| echo2file " Mbps: 10000" |
| echo2file " pci: \"$pci\"" |
| echo2file " mac: \"$mac\"" |
| echo2file " sriovs:" |
| |
| PCI_COUNTER=0 |
| for((nb_sriov=0;nb_sriov<8;nb_sriov++)) |
| do |
| #PCI_COUNTER=$((PCI_COUNTER+2)) |
| #echo2file "nb_sriov "$nb_sriov |
| #echo2file "PCI_COUNTER "$PCI_COUNTER |
| |
| pci=`printf "0000:%02X:10.%i" $bus $nb_sriov` |
| mac=`get_mac $seed` |
| seed=$((seed+1)) |
| echo2file " - source_name: eth$nb_sriov" |
| echo2file " mac: \"$mac\"" |
| echo2file " pci: \"$pci\"" |
| done |
| iface_counter=$((iface_counter+1)) |
| done |
| last_iface=$(((numa+1)*127/NUMAS+5)) #made-up formula for more realistic pci numbers |
| numa=$((numa+1)) |
| done |
| } |
| |
| function _yaml_init(){ |
| echo -n > host-develope.yaml |
| } |
| |
| function echo2file(){ |
| echo "${1}" |
| echo "${1}" >> host-develope.yaml |
| } |
| |
| function _get_opts() |
| { |
| options="$1" |
| shift |
| |
| get_argument="" |
| #reset variables |
| params="" |
| for option_group in $options |
| do |
| _name=${option_group%%:*} |
| _name=${_name%=} |
| _name=${_name//-/_} |
| eval option_${_name}='""' |
| done |
| |
| while [[ $# -gt 0 ]] |
| do |
| argument="$1" |
| shift |
| if [[ -n $get_argument ]] |
| then |
| [[ ${argument:0:1} == "-" ]] && echo "option '-$option' requires an argument" >&2 && return 1 |
| eval ${get_argument}='"$argument"' |
| #echo option $get_argument with argument |
| get_argument="" |
| continue |
| fi |
| |
| |
| #short options |
| if [[ ${argument:0:1} == "-" ]] && [[ ${argument:1:1} != "-" ]] && [[ ${#argument} -ge 2 ]] |
| then |
| index=0 |
| while index=$((index+1)) && [[ $index -lt ${#argument} ]] |
| do |
| option=${argument:$index:1} |
| bad_option=y |
| for option_group in $options |
| do |
| _name="" |
| for o in $(echo $option_group | tr ":=" " ") |
| do |
| [[ -z "$_name" ]] && _name=${o//-/_} |
| #echo option $option versus $o |
| if [[ "$option" == "${o}" ]] |
| then |
| eval option_${_name}='${option_'${_name}'}-' |
| bad_option=n |
| if [[ ${option_group:${#option_group}-1} != "=" ]] |
| then |
| continue |
| fi |
| if [[ ${#argument} -gt $((index+1)) ]] |
| then |
| eval option_${_name}='"${argument:$((index+1))}"' |
| index=${#argument} |
| else |
| get_argument=option_${_name} |
| #echo next should be argument $argument |
| fi |
| |
| break |
| fi |
| done |
| done |
| [[ $bad_option == y ]] && echo "invalid argument '-$option'? Type -h for help" >&2 && return 1 |
| done |
| elif [[ ${argument:0:2} == "--" ]] && [[ ${#argument} -ge 3 ]] |
| then |
| option=${argument:2} |
| option_argument=${option#*=} |
| option_name=${option%%=*} |
| [[ "$option_name" == "$option" ]] && option_argument="" |
| bad_option=y |
| for option_group in $options |
| do |
| _name="" |
| for o in $(echo $option_group | tr ":=" " ") |
| do |
| [[ -z "$_name" ]] && _name=${o//-/_} |
| #echo option $option versus $o |
| if [[ "$option_name" == "${o}" ]] |
| then |
| bad_option=n |
| if [[ ${option_group:${#option_group}-1} != "=" ]] |
| then #not an argument |
| [[ -n "${option_argument}" ]] && echo "option '--${option%%=*}' do not accept an argument " >&2 && return 1 |
| eval option_${_name}='"${option_'${_name}'}-"' |
| elif [[ -n "${option_argument}" ]] |
| then |
| eval option_${_name}='"${option_argument}"' |
| else |
| get_argument=option_${_name} |
| #echo next should be argument $argument |
| fi |
| break |
| fi |
| done |
| done |
| [[ $bad_option == y ]] && echo "invalid argument '-$option'? Type -h for help" >&2 && return 1 |
| elif [[ ${argument:0:2} == "--" ]] |
| then |
| option__="$*" |
| bad_option=y |
| for o in $options |
| do |
| if [[ "$o" == "--" ]] |
| then |
| bad_option=n |
| option__=" $*" |
| break |
| fi |
| done |
| [[ $bad_option == y ]] && echo "invalid argument '--'? Type -h for help" >&2 && return 1 |
| break |
| else |
| params="$params ${argument}" |
| fi |
| |
| done |
| |
| [[ -n "$get_argument" ]] && echo "option '-$option' requires an argument" >&2 && return 1 |
| return 0 |
| } |
| |
| #process options |
| DIRNAME=$(readlink -f ${BASH_SOURCE[0]}) |
| DIRNAME=$(dirname $DIRNAME) |
| |
| #source ${DIRNAME}/get-options.sh "help:h user:u= ip:i= cores:c= huge-pages-1G:hp= nb-10GB-interfaces:ni=" $* |
| _get_opts "help:h user-name:u= ip:i= cores:c= huge-pages-1G:hp= nb-10GB-interfaces:ni=" $* || exit 1 |
| _parse_opts |
| |
| #check root privileges |
| [ "${USER}" != "root" ] && echo "Needed root privileges" >&2 && exit 2 |
| #check root privileges and non a root user behind |
| |
| |
| HOST_NAME=`cat /etc/hostname` |
| |
| _generate_compute_develope_yaml |
| |