network: Some code cleanup.

[people/stevee/network.git] / hooks / ports / virtual

#!/bin/bash
###############################################################################
#                                                                             #
# IPFire.org - A linux based firewall                                         #
# Copyright (C) 2010  Michael Tremer & Christian Schmidt                      #
#                                                                             #
# 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/>.       #
#                                                                             #
###############################################################################

. /lib/network/header-port

HOOK_SETTINGS="HOOK DEVICE DEVICE_MAC DEVICE_VID"

DEVICE_MAC=$(mac_generate)

function _check() {
	assert isset DEVICE
	assert ismac DEVICE_MAC
	assert isinteger DEVICE_VID

	if [ ${DEVICE_VID} -gt 4096 ]; then
		error "DEVICE_VID is greater than 4096."
		exit ${EXIT_ERROR}
	fi

	local reserved
	for reserved in 0 4095; do
		if [ "${DEVICE_VID}" = "${reserved}" ]; then
			error "DEVICE_VID=${reserved} is reserved."
			exit ${EXIT_ERROR}
		fi
	done
}

function _create() {
	while [ $# -gt 0 ]; do
		case "${1}" in
			--device=*)
				DEVICE=${1#--device=}
				;;
			--mac=*)
				DEVICE_MAC=${1#--mac=}
				;;
			--id=*)
				DEVICE_VID=${1#--id=}
				;;
			*)
				warning "Unknown argument '${1}'"
				;;
		esac
		shift
	done

	local port="${DEVICE}v${DEVICE_VID}"

	config_write $(port_file ${port}) ${HOOK_SETTINGS}

	exit ${EXIT_OK}
}

function _edit() {
	local port=${1}
	shift

	assert isset port

	config_read $(port_file ${port})

	while [ $# -gt 0 ]; do
		case "${1}" in
			--mac=*)
				DEVICE_MAC=${1#--mac=}
				;;
			*)
				warning "Unknown argument '${1}'"
				;;
		esac
		shift
	done

	config_write $(port_file ${port}) ${HOOK_SETTINGS}

	exit ${EXIT_OK}	
}

function _up() {
	local port=${1}

	assert isset port

	config_read $(port_file ${port})

	if ! device_exists ${port}; then
		virtual_create ${DEVICE} ${DEVICE_VID} ${DEVICE_MAC}
	fi

	exit ${EXIT_OK}
}

function _down() {
	local port=${1}

	assert isset port

	config_read $(port_file ${port})

	if ! device_exists ${port}; then
		exit ${EXIT_OK}
	fi

	virtual_remove ${port}

	exit ${EXIT_OK}
}

function _status() {
	local zone=${1}
	local port=${2}

config_read $(zone_dir ${zone})/${port}

	local device=$(devicify ${DEVICE_MAC})

	printf "        %-10s - " "${device}"
	if ! device_is_up ${device}; then
		echo -ne "${COLOUR_DOWN}   DOWN   ${COLOUR_NORMAL}"
	else
		local state=$(stp_port_state ${zone} ${device})
		local colour="COLOUR_STP_${state}"
		printf "${!colour}%10s${COLOUR_NORMAL}" ${state}
	fi

	echo -n " - DSR: $(stp_port_designated_root ${zone} ${device})"
	echo -n " - Cost: $(stp_port_pathcost ${zone} ${device})"
	echo

	exit ${EXIT_OK}
}

run $@