src/functions/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_dir() {
  23         local type=${1}
  24
  25         if [ -n "${type}" ]; then
  26                 type="/${type}s"
  27         fi
  28
  29         echo "${NETWORK_HOOKS_DIR}${type}"
  30 }
  31 NETWORK_HOOKS_DIR_ZONES="$(hook_dir zone)"
  32
  33 hook_exists() {
  34         local type=${1}
  35         local hook=${2}
  36
  37         assert isset type
  38         assert isset hook
  39
  40         # Add the path prefix.
  41         hook="$(hook_dir ${type})/${hook}"
  42
  43         [ ! -d "${hook}" ] && [ -x "${hook}" ]
  44 }
  45
  46 hook_exec() {
  47         local type="${1}"
  48         assert isset type
  49
  50         local hook="${2}"
  51         assert isset hook
  52
  53         local cmd="${3}"
  54         assert isset cmd
  55
  56         assert hook_exists "${type}" "${hook}"
  57         shift 3
  58
  59         # Complete the hook command by prepending "hook_"
  60         local hook_cmd="hook_${cmd}"
  61
  62         # Check if the hook action is valid.
  63         assert hook_valid_command "${type}" "${cmd}"
  64
  65         local hook_path="$(hook_dir ${type})/${hook}"
  66
  67         # For performance reasons, all hooks are executed
  68         # in a subshell and so will inherit the currently
  69         # running environment.
  70         (
  71                 # Set the name of the hook.
  72                 HOOK=$(basename ${hook})
  73
  74                 # Source the code of the hook.
  75                 source "${hook_path}"
  76
  77                 # Make sure HOOK is still properly set.
  78                 assert isset HOOK
  79
  80                 # Execute the requested command.
  81                 "${hook_cmd}" "$@"
  82         )
  83         local ret=$?
  84
  85         case "${ret}" in
  86                 ${EXIT_COMMAND_NOT_FOUND}|${EXIT_NOT_SUPPORTED})
  87                         log ERROR "Hook '${hook}' does not implement the method '${cmd}':"
  88                         log ERROR "  arguments: $@"
  89                         return ${EXIT_COMMAND_NOT_FOUND}
  90                         ;;
  91                 ${EXIT_ERROR_ASSERT})
  92                         log ERROR "Hook exited with an assertion error."
  93                         return ${EXIT_ERROR_ASSERT}
  94                         ;;
  95         esac
  96
  97         return ${ret}
  98 }
  99
 100 hook_list() {
 101         local type="${1}"
 102
 103         local dir="$(hook_dir "${type}")"
 104         assert isset dir
 105
 106         local hook
 107         for hook in $(list_directory "${dir}"); do
 108                 if hook_exists "${type}" "${hook}"; then
 109                         echo "${hook}"
 110                 fi
 111         done
 112 }
 113
 114 # The default help function.
 115 hook_help() {
 116         # If no man page has been configured, we print an error message.
 117         if [ -z "${HOOK_MANPAGE}" ]; then
 118                 error "There is no help available for hook '${HOOK}'. Exiting."
 119                 exit ${EXIT_ERROR}
 120         fi
 121
 122         cli_show_man "${HOOK_MANPAGE}"
 123
 124         exit $?
 125 }
 126
 127 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 hook_zone_exists() {
 140         hook_exists zone "$@"
 141 }
 142
 143 hook_zone_exec() {
 144         hook_exec zone "$@"
 145 }
 146
 147 hook_zone_get_all() {
 148         hook_list zone
 149 }
 150
 151 hook_config_exists() {
 152         hook_exists config "$@"
 153 }
 154
 155 hook_config_exec() {
 156         hook_exec config "$@"
 157 }
 158
 159 hook_config_get_all() {
 160         hook_list config
 161 }
 162
 163 hook_valid_command() {
 164         local type="${1}"
 165         local cmd="${2}"
 166
 167         case "${type}" in
 168                 config)
 169                         hook_valid_command_config "${cmd}"
 170                         return ${?}
 171                         ;;
 172                 port)
 173                         hook_valid_command_port "${cmd}"
 174                         return ${?}
 175                         ;;
 176                 zone)
 177                         hook_valid_command_zone "${cmd}"
 178                         return ${?}
 179                         ;;
 180         esac
 181
 182         return ${EXIT_FALSE}
 183 }
 184
 185 hook_valid_command_config() {
 186         local cmd="${1}"
 187
 188         case "${cmd}" in
 189                 new|destroy|edit|up|down|status|hid)
 190                         return ${EXIT_TRUE}
 191                         ;;
 192         esac
 193
 194         return ${EXIT_FALSE}
 195 }
 196
 197 hook_valid_command_port() {
 198         local cmd="${1}"
 199
 200         case "${cmd}" in
 201                 # Configuration hooks
 202                 new|edit|destroy)
 203                         return ${EXIT_TRUE}
 204                         ;;
 205
 206                 # Control hooks
 207                 create|remove|up|down)
 208                         return ${EXIT_TRUE}
 209                         ;;
 210
 211                 # Hotplug
 212                 hotplug|hotplug_rename)
 213                         return ${EXIT_TRUE}
 214                         ;;
 215
 216                 # Status
 217                 status|children)
 218                         return ${EXIT_TRUE}
 219                         ;;
 220         esac
 221
 222         return ${EXIT_FALSE}
 223 }
 224
 225 hook_valid_command_zone() {
 226         local cmd="${1}"
 227
 228         case "${cmd}" in
 229                 # Configuration hooks
 230                 new|edit|destroy)
 231                         return ${EXIT_TRUE}
 232                         ;;
 233
 234                 config_new|config_destroy|config_edit|config_show)
 235                         return ${EXIT_TRUE}
 236                         ;;
 237
 238                 # Control hooks
 239                 up|down)
 240                         return ${EXIT_TRUE}
 241                         ;;
 242
 243                 # Hotplug
 244                 hotplug)
 245                         return ${EXIT_TRUE}
 246                         ;;
 247
 248                 # Ports
 249                 port_attach|port_detach|port_edit|port_create|port_remove|port_status|port_up|port_down)
 250                         return ${EXIT_TRUE}
 251                         ;;
 252
 253                 # Status
 254                 status|info|help)
 255                         return ${EXIT_TRUE}
 256                         ;;
 257
 258                 # Discovery
 259                 discover)
 260                         return ${EXIT_TRUE}
 261                         ;;
 262
 263                 # PPP
 264                 ppp_ip_pre_up|ppp_ipv[64]_up|ppp_ipv[64]_down|ppp_write_config)
 265                         return ${EXIT_TRUE}
 266                         ;;
 267         esac
 268
 269         return ${EXIT_FALSE}
 270 }