functions.hook

   1 #!/bin/bash
   2 ###############################################################################
   3 #                                                                             #
   4 # IPFire.org - A linux based firewall                                         #
   5 # Copyright (C) 2010  Michael Tremer & Christian Schmidt                      #
   6 #                                                                             #
   7 # This program is free software: you can redistribute it and/or modify        #
   8 # it under the terms of the GNU General Public License as published by        #
   9 # the Free Software Foundation, either version 3 of the License, or           #
  10 # (at your option) any later version.                                         #
  11 #                                                                             #
  12 # This program is distributed in the hope that it will be useful,             #
  13 # but WITHOUT ANY WARRANTY; without even the implied warranty of              #
  14 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the               #
  15 # GNU General Public License for more details.                                #
  16 #                                                                             #
  17 # You should have received a copy of the GNU General Public License           #
  18 # along with this program.  If not, see <http://www.gnu.org/licenses/>.       #
  19 #                                                                             #
  20 ###############################################################################
  21
  22 HOOK_COMMANDS_CONFIG="hook_create hook_down hook_status hook_up"
  23
  24 HOOK_COMMANDS_PORT="hook_create hook_down hook_hotplug hook_hotplug_rename \
  25         hook_info hook_status hook_up"
  26
  27 HOOK_COMMANDS_ZONE="hook_add hook_create hook_discover hook_down hook_edit hook_help \
  28         hook_info hook_remove hook_status hook_up \
  29         \
  30         hook_config_create hook_config_edit hook_config_remove hook_config_show \
  31         \
  32         hook_port hook_port_add hook_port_edit hook_port_remove hook_port_show \
  33         hook_port_status hook_port_up hook_port_down \
  34         \
  35         hook_ppp_ip_pre_up hook_ppp_ipv4_down hook_ppp_ipv4_up \
  36         hook_ipv6_down hook_ipv6_up hook_ppp_write_config"
  37
  38 function hook_dir() {
  39         local type=${1}
  40
  41         if [ -n "${type}" ]; then
  42                 type="/${type}s"
  43         fi
  44
  45         echo "${NETWORK_HOOKS_DIR}${type}"
  46 }
  47 NETWORK_HOOKS_DIR_ZONES="$(hook_dir zone)"
  48
  49 function hook_exists() {
  50         local type=${1}
  51         local hook=${2}
  52
  53         assert isset type
  54         assert isset hook
  55
  56         # Add the path prefix.
  57         hook="$(hook_dir ${type})/${hook}"
  58
  59         [ ! -d "${hook}" ] && [ -x "${hook}" ]
  60 }
  61
  62 function hook_exec() {
  63         local type="${1}"
  64         assert isset type
  65
  66         local hook="${2}"
  67         assert isset hook
  68
  69         local cmd="${3}"
  70         assert isset cmd
  71
  72         assert hook_exists "${type}" "${hook}"
  73         shift 3
  74
  75         # Complete the hook command by prepending "hook_"
  76         local hook_cmd="hook_${cmd}"
  77
  78         # Check if the hook action is valid.
  79         local valid_commands
  80         case "${type}" in
  81                 "config")
  82                         valid_commands="${HOOK_COMMANDS_CONFIG}"
  83                         ;;
  84                 "port")
  85                         valid_commands="${HOOK_COMMANDS_PORT}"
  86                         ;;
  87                 "zone")
  88                         valid_commands="${HOOK_COMMANDS_ZONE}"
  89                         ;;
  90         esac
  91         isset valid_commands && assert list_match "${hook_cmd}" ${valid_commands}
  92
  93         local hook_path="$(hook_dir ${type})/${hook}"
  94
  95         # For performance reasons, all hooks are executed
  96         # in a subshell and so will inherit the currently
  97         # running environment.
  98         (
  99                 # Set the name of the hook.
 100                 HOOK=$(basename ${hook})
 101
 102                 # Source the code of the hook.
 103                 source "${hook_path}"
 104
 105                 # Make sure HOOK is still properly set.
 106                 assert isset HOOK
 107
 108                 # Execute the requested command.
 109                 cmd "${hook_cmd}" "$@"
 110         )
 111         local ret=$?
 112
 113         case "${ret}" in
 114                 ${EXIT_COMMAND_NOT_FOUND})
 115                         log ERROR "Hook command not implemented: ${hook_command} ($@)"
 116                         exit ${EXIT_COMMAND_NOT_FOUND}
 117                         ;;
 118                 ${EXIT_ERROR_ASSERT})
 119                         log ERROR "Hook exited with an assertion error."
 120                         exit ${EXIT_ERROR_ASSERT}
 121                         ;;
 122         esac
 123
 124         return ${ret}
 125 }
 126
 127 function config_get_hook() {
 128         local config=${1}
 129
 130         assert isset config
 131         assert [ -e "${config}" ]
 132
 133         (
 134                 . ${config}
 135                 echo "${HOOK}"
 136         )
 137 }
 138
 139 function hook_zone_exists() {
 140         hook_exists zone $@
 141 }
 142
 143 function hook_zone_port_exists() {
 144         local hook_zone=${1}
 145         local hook_port=${2}
 146
 147         hook_exists zone "${hook_zone}.ports/${hook_port}"
 148 }
 149
 150 function hook_zone_config_exists() {
 151         local hook_zone=${1}
 152         local hook_config=${2}
 153
 154         hook_exists zone "${hook_zone}.configs/${hook_config}"
 155 }
 156
 157 function hook_zone_has_ports() {
 158         local hook=${1}
 159
 160         [ -d "${NETWORK_HOOKS_DIR_ZONES}/${hook}.ports" ]
 161 }
 162
 163 function hook_zone_has_configs() {
 164         local hook=${1}
 165
 166         [ -d "${NETWORK_HOOKS_DIR_ZONES}/${hook}.configs" ]
 167 }
 168
 169 function hook_zone_exec() {
 170         hook_exec zone $@
 171 }
 172
 173 function hook_zone_port_exec() {
 174         local hook_zone=${1}
 175         local hook_port=${2}
 176         shift 2
 177
 178         hook_zone_exec "${hook_zone}.ports/${hook_port}" $@
 179 }
 180
 181 function hook_zone_config_exec() {
 182         local hook_zone=${1}
 183         local hook_port=${2}
 184         shift 2
 185
 186         hook_zone_exec "${hook_zone}.configs/${hook_port}" $@
 187 }
 188
 189 function hook_zone_get_all() {
 190         local type=${1}
 191
 192         local hook
 193         for hook in $(hook_dir zone)/*; do
 194                 hook=$(basename ${hook})
 195                 hook_zone_exists ${hook} && echo "${hook}"
 196         done
 197 }
 198
 199 function hook_zone_ports_get_all() {
 200         local hook=${1}
 201
 202         if ! hook_exists zone ${hook}; then
 203                 error "Hook '${hook}' does not exist."
 204                 return ${EXIT_ERROR}
 205         fi
 206
 207         # If the zone hook has got no ports we exit silently
 208         if ! hook_zone_has_ports ${hook}; then
 209                 return ${EXIT_OK}
 210         fi
 211
 212         local h
 213         for h in $(hook_dir zone)/${hook}.ports/*; do
 214                 h=$(basename ${h})
 215                 if hook_zone_port_exists ${hook} ${h}; then
 216                         echo "${h}"
 217                 fi
 218         done
 219 }
 220
 221 function hook_zone_configs_get_all() {
 222         local hook=${1}
 223
 224         if ! hook_exists zone ${hook}; then
 225                 error "Hook '${hook}' does not exist."
 226                 return ${EXIT_ERROR}
 227         fi
 228
 229         # If the zone hook has got no configurations we exit silently
 230         if ! hook_zone_has_configs ${hook}; then
 231                 return ${EXIT_OK}
 232         fi
 233
 234         local h
 235         for h in $(hook_dir zone)/${hook}.configs/*; do
 236                 h=$(basename ${h})
 237                 if hook_zone_config_exists ${hook} ${h}; then
 238                         echo "${h}"
 239                 fi
 240         done
 241
 242         return ${EXIT_OK}
 243 }