nfvbenchvm/dib/build-image.sh

   1 #!/bin/bash
   2 #
   3 # A shell script to build the VPP VM image or NFVbench+TRex VM image using diskinage-builder
   4 #
   5 # The following packages must be installed prior to using this script:
   6 # sudo apt-get -y install python-virtualenv qemu-utils kpartx
   7
   8 usage() {
   9     echo "Usage: $0 [-l] [-g] [-v]"
  10     echo "   -l    build NFVbench loop VM image"
  11     echo "   -g    build NFVbench generator image"
  12     echo "   -v    verify only (build but do not push to google storage)"
  13     exit 1
  14 }
  15
  16 verify_only=0
  17 generator_only=0
  18 loopvm_only=0
  19 __prefix__=""
  20 # ----------------------------------------------------------------------------
  21 # Parse command line options and configure the script
  22 # ----------------------------------------------------------------------------
  23
  24 while getopts ":hglv" opt; do
  25     case $opt in
  26         h)
  27             usage
  28             exit 0
  29             ;;
  30         g)
  31             generator_only=1
  32             ;;
  33         l)
  34             loopvm_only=1
  35             ;;
  36         v)
  37             verify_only=1
  38             ;;
  39         ?)
  40             usage
  41             exit 1
  42             ;;
  43     esac
  44 done
  45
  46 set -e
  47
  48 # Artifact URL
  49 gs_url=artifacts.opnfv.org/nfvbench/images
  50
  51 # image version number
  52 __version__=0.13
  53 loopvm_image_name=nfvbenchvm_centos-$__version__
  54 generator_image_name=nfvbenchvm_centos-generator-$__version__
  55
  56 function build_image {
  57     # if image exists skip building
  58     echo "Checking if image exists in google storage..."
  59     if  command -v gsutil >/dev/null; then
  60         if gsutil -q stat gs://$gs_url/$1.qcow2; then
  61             echo "Image already exists at http://$gs_url/$1.qcow2"
  62             echo "Build is skipped"
  63             exit 0
  64         fi
  65         echo "Image does not exist in google storage, starting build..."
  66         echo
  67     else
  68         echo "Cannot check image availability in OPNFV artifact repository (gsutil not available)"
  69     fi
  70
  71     # check if image is already built locally
  72     if [ -f $1.qcow2 ]; then
  73         echo "Image $1.qcow2 already exists locally"
  74     else
  75
  76         # install diskimage-builder
  77         if [ -d dib-venv ]; then
  78            . dib-venv/bin/activate
  79         else
  80            virtualenv dib-venv
  81            . dib-venv/bin/activate
  82            pip install diskimage-builder
  83         fi
  84
  85         # Add nfvbenchvm_centos elements directory to the DIB elements path
  86         export ELEMENTS_PATH=`pwd`/elements
  87
  88         # canned user/password for direct login
  89         export DIB_DEV_USER_USERNAME=nfvbench
  90         export DIB_DEV_USER_PASSWORD=nfvbench
  91         export DIB_DEV_USER_PWDLESS_SUDO=Y
  92
  93         # Set the data sources to have ConfigDrive only
  94         export DIB_CLOUD_INIT_DATASOURCES="ConfigDrive"
  95
  96         # Configure VPP REPO
  97         export DIB_YUM_REPO_CONF=$ELEMENTS_PATH/nfvbenchvm/fdio-release.repo
  98
  99         # Use ELRepo to have latest kernel
 100         # only for loop vm image
 101         if [ $1 = $loopvm_image_name ]; then
 102            export DIB_USE_ELREPO_KERNEL=True
 103            export DIB_DEV_IMAGE=loopvm
 104         else
 105            export DIB_USE_ELREPO_KERNEL=False
 106            export DIB_DEV_IMAGE=generator
 107            # get current git branch to build image with current code
 108            export GIT_BRANCH=$(git rev-parse --abbrev-ref HEAD)
 109            # retrieve TREX_VER from Dockerfile
 110            export TREX_VER=$(awk '/ENV TREX_VER/ {print $3}' ../../docker/Dockerfile | sed 's/"//g' | sed 's/\r//g')
 111         fi
 112
 113         echo "Building $1.qcow2..."
 114         time disk-image-create -o $1 centos7 nfvbenchvm
 115     fi
 116
 117     ls -l $1.qcow2
 118
 119     if [ $verify_only -eq 1 ]; then
 120         echo "Image verification SUCCESS"
 121         echo "NO upload to google storage (-v)"
 122     else
 123         if command -v gsutil >/dev/null; then
 124             echo "Uploading $1.qcow2..."
 125             gsutil cp $1.qcow2 gs://$gs_url/$1.qcow2
 126             echo "You can access to image at http://$gs_url/$1.qcow2"
 127         else
 128             echo "Cannot upload new image to the OPNFV artifact repository (gsutil not available)"
 129             exit 1
 130         fi
 131     fi
 132 }
 133
 134
 135 if [ ! $generator_only -eq 1 ] && [ ! $loopvm_only -eq 1 ]; then
 136    echo "Build loop VM image"
 137    build_image $loopvm_image_name
 138    echo "Build generator image"
 139    build_image $generator_image_name
 140 else
 141     if [ $loopvm_only -eq 1 ]; then
 142        echo "Build loop VM image"
 143        build_image $loopvm_image_name
 144     fi
 145     if [ $generator_only -eq 1 ]; then
 146        echo "Build generator image"
 147        build_image $generator_image_name
 148     fi
 149 fi