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_COMMANDS_CONFIG="hook_create hook_down hook_status hook_remove hook_up"
  23
  24 HOOK_COMMANDS_PORT="hook_create hook_down hook_edit hook_hotplug \
  25         hook_hotplug_rename 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}|${EXIT_NOT_SUPPORTED})
 115                         log ERROR "Hook '${hook}' does not implement the method '${cmd}':"
 116                         log ERROR "  arguments: $@"
 117                         exit ${EXIT_COMMAND_NOT_FOUND}
 118                         ;;
 119                 ${EXIT_ERROR_ASSERT})
 120                         log ERROR "Hook exited with an assertion error."
 121                         exit ${EXIT_ERROR_ASSERT}
 122                         ;;
 123         esac
 124
 125         return ${ret}
 126 }
 127
 128 function hook_list() {
 129         local type="${1}"
 130
 131         local dir="$(hook_dir "${type}")"
 132         assert isset dir
 133
 134         local hook
 135         for hook in ${dir}/*; do
 136                 hook="$(basename "${hook}")"
 137
 138                 if hook_exists "${type}" "${hook}"; then
 139                         echo "${hook}"
 140                 fi
 141         done
 142 }
 143
 144 function config_get_hook() {
 145         local config=${1}
 146
 147         assert isset config
 148         assert [ -e "${config}" ]
 149
 150         (
 151                 . ${config}
 152                 echo "${HOOK}"
 153         )
 154 }
 155
 156 function hook_zone_exists() {
 157         hook_exists zone $@
 158 }
 159
 160 function hook_zone_exec() {
 161         hook_exec zone $@
 162 }
 163
 164 function hook_zone_get_all() {
 165         hook_list zone
 166 }
 167
 168 function hook_config_exists() {
 169         hook_exists config $@
 170 }
 171
 172 function hook_config_exec() {
 173         hook_exec config $@
 174 }
 175
 176 function hook_config_get_all() {
 177         hook_list config
 178 }