Vagrantfile

# -*- mode: ruby -*-
# vi: set ft=ruby :

# In Windows/MinGW we need to use NUL instead of /dev/null. This does not apply
# in Cygwin (where 'host_os' below would return as 'cygwin').
require 'rbconfig'
is_windows = (RbConfig::CONFIG['host_os'] =~ /mswin|mingw/)
if is_windows
  devnull = 'NUL'
else
  devnull = '/dev/null'
end

################################################################################
# Custom configurations
################################################################################

# Load local configurations.
local_vagrantfile = File.expand_path('../Vagrantfile.local', __FILE__)
load local_vagrantfile if File.exists?(local_vagrantfile)

# Load team configurations.
team_vagrantfile = File.expand_path('../Vagrantfile.team', __FILE__)
load team_vagrantfile if File.exists?(team_vagrantfile)

################################################################################
# Provisioning Scripts
################################################################################

# Use the current Git SHA1 as the VM version number.
$vm_version = `git rev-parse HEAD`

# Configure the VM to use the proxy used by the host machine. Configuration is
# done at the system level (vs per-user).
$configure_proxy = <<EOT
if [ "#{$http_proxy}" == "" ]; then
    exit 0
fi

echo "Configuring proxy as '#{$http_proxy}'."

export http_proxy="#{$http_proxy}"
export https_proxy="$http_proxy"

sed -i -e '/https?_proxy *=/d' /etc/environment
echo "http_proxy=$http_proxy" >> /etc/environment
echo "https_proxy=$https_proxy" >> /etc/environment

sed -i -e '/env_keep =/d' /etc/sudoers
echo "Defaults env_keep = \\"http_proxy https_proxy ftp_proxy\\"" >> /etc/sudoers

if [ -f /etc/apt/apt.conf ]; then
    sed -i -e '/Acquire::http::Proxy/d' /etc/apt/apt.conf
fi
echo "Acquire::http::Proxy \\"$http_proxy\\";" >> /etc/apt/apt.conf
EOT

# Install required tools.
$install_apt_packages = <<EOT
echo "Installing required apt packages."

apt-get update
apt-get install -y git gcc g++ gdb pkg-config python-pip python-dev \
                   linux-image-extra-virtual \
                   #{$team_apt_packages} #{$local_apt_packages}

if [ "#{$enable_gui_mode}" = "true" ]; then
    # Note that the following lines are necessary to resolve errors that come
    # up when installing ubuntu-desktop below (in Vagrant only).
    /usr/share/debconf/fix_db.pl
    dpkg-reconfigure dictionaries-common

    apt-get install -y ubuntu-desktop
fi

yes | pip install numpy
EOT

$install_pip_packages = <<EOT
echo "Installing required pip packages."

yes | pip install #{$team_pip_packages} #{$local_pip_packages}
EOT



################################################################################
# VM Configuration
################################################################################

def provisionFile(path, config)
  if not File.exist?(path)
    return
  end
  if File.directory?(path)
    config.vm.provision "shell", privileged: false, inline: "mkdir #{path}"
    for file in Dir[File.expand_path(path) + "/*"]
      privisionFile(file, config)
    end
  else
    config.vm.provision "file", source: path, destination: path
  end
end

Vagrant.configure(2) do |config|
  # Every Vagrant development environment requires a box. You can search for
  # boxes at https://atlas.hashicorp.com/search.
  config.vm.box = "ubuntu/xenial64"

  # Use local setting for insecure box downloads.
  config.vm.box_download_insecure = $download_insecure

  # Configure the VM network settings.
  if $network_bridged
    if $network_ip == ""
      config.vm.network "public_network"
    else
      config.vm.network "public_network", ip: "#{$network_ip}"
    end
  else
    # NAT settings (e.g., port forwarding) would go here.
  end

  # VirtualBox-specific configuration.
  config.vm.provider "virtualbox" do |v|
    # Specify the name of the VM in the VirtualBox GUI.
    v.name = $vm_name

    # Set to 'true' to enable the a UI for the VM. By default, Vagrant VMs run
    # in headless mode.
    #
    # Note: This may be necessary if using CodeComposer.
    v.gui = $enable_gui_mode

    # Specify the number of CPU cores to assign to the VM.
    v.cpus = $num_cpus

    # Specify the amount of memory to allocate to the VM (in MB).
    v.memory = $memory_size

    # Configure USB settings.
    v.customize ["modifyvm", :id, "--usb", "on"]
    v.customize ["modifyvm", :id, "--usbehci", "on"]

    # Add filter for programmer.
    v.customize ["usbfilter", "add", "0", "--target", :id,
                 "--name", "ATMEL AVRDRAGON",
                 "--manufacturer", "ATMEL", "--product", "AVRDRAGON"]

    # Add filter for USB to RS232 converters.
    v.customize ["usbfilter", "add", "0", "--target", :id,
                 "--name", "FTDI TTL232R-3V3",
                 "--manufacturer", "FTDI", "--product", "TTL232R-3V3"]
  end

  # Configure the machine hostname.
  config.vm.hostname = $hostname

  # Enable X11 forwarding.
  config.ssh.forward_x11 = true
  config.ssh.forward_agent = true

  # Share an additional folder to the guest VM. The first argument is
  # the path on the host to the actual folder. The second argument is
  # the path on the guest to mount the folder. And the optional third
  # argument is a set of non-required options.
  config.vm.synced_folder "./", "/home/ubuntu/ectf"

  # Provision the VM.
  if $ssh_public_key != ""
    config.vm.provision "file",
                        source: "#{$ssh_key_location}/#{$ssh_public_key}",
                        destination: "~/.ssh/#{$ssh_public_key}"
  end

  for file in $files
    provisionFile(file, config)
  end

  if $http_proxy != ""
    config.vm.provision "shell", inline: $configure_proxy
  end

  # Configure custom system-level settings.
  config.vm.provision "shell", inline: $configure_team_settings
  config.vm.provision "shell", inline: $configure_local_settings

  # Install dependencies.
  config.vm.provision "shell", inline: $install_apt_packages
  config.vm.provision "shell", inline: $install_pip_packages

  # Configure installed tools.
  config.vm.provision "shell", inline: $configure_team_tools
  config.vm.provision "shell", inline: $configure_local_tools

  # Store the version of the Vagrant configuration used to provision the VM.
  config.vm.provision "shell", privileged: false,
                      inline: "echo -n \"#{$vm_version}\" > /home/ubuntu/.vm_version"
end