#!/bin/bash
# Copyright (c) 1999-2017, Parallels International GmbH
# Copyright (c) 2017-2019 Virtuozzo International GmbH. All rights reserved.
#
# This file is part of OpenVZ libraries. OpenVZ is free software; you can
# redistribute it and/or modify it under the terms of the GNU Lesser General
# Public License as published by the Free Software Foundation; either version 2
# 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
# Lesser General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
# 02110-1301, USA.
#
# Our contact details: Virtuozzo International GmbH, Vordergasse 59, 8200
# Schaffhausen, Switzerland.
#
# This script setup routing traffic accounting for given VPS.
# For usage info see vz-net_add(5) man page.
#
# Parameters are passed in environment variables.
# Required parameters:
#   VEID          - VPS id
# Optional parameters:
#   IP_ADDR       - IP address(es) to add
#                   (several addresses should be divided by space)
#   VE_STATE      - state of VPS; could be one of:
#                     starting | stopping | running | stopped
. /etc/vz/vz.conf
. /usr/libexec/libvzctl/scripts/vz-functions

vzcheckvar VEID

[ -z "${IP_ADDR}" ] && exit 0

######## Common stuff - route, ARP, iptables ###########
# Set routing, iptables, ARP...
vzgetnetdev

vzarpipdetect "$IP_ADDR"
for IP in $IP_ADDR; do
	vzaddrouting $IP
	vzarp add $IP
done
vzarpipset "$IP_ADDR"
# Save ip address information
mkdir -p ${VE_STATE_DIR} >/dev/null 2>&1
if [ "${VE_STATE}" == "starting" ]; then
  echo -n  "${IP_ADDR} " > ${VE_STATE_DIR}/${VEID}
elif [ "${VE_STATE}" == "running" ]; then
  echo -n "${IP_ADDR} " >> ${VE_STATE_DIR}/${VEID}
fi

exit 0
# end of script

