146 lines
4.2 KiB
Plaintext
146 lines
4.2 KiB
Plaintext
|
#compdef ufw
|
||
|
|
# ------------------------------------------------------------------------------
|
||
|
|
# Copyright (c) 2016 Github zsh-users - http://github.com/zsh-users
|
||
|
|
# All rights reserved.
|
||
|
|
#
|
||
|
|
# Redistribution and use in source and binary forms, with or without
|
||
|
|
# modification, are permitted provided that the following conditions are met:
|
||
|
|
# * Redistributions of source code must retain the above copyright
|
||
|
|
# notice, this list of conditions and the following disclaimer.
|
||
|
|
# * Redistributions in binary form must reproduce the above copyright
|
||
|
|
# notice, this list of conditions and the following disclaimer in the
|
||
|
|
# documentation and/or other materials provided with the distribution.
|
||
|
|
# * Neither the name of the zsh-users nor the
|
||
|
|
# names of its contributors may be used to endorse or promote products
|
||
|
|
# derived from this software without specific prior written permission.
|
||
|
|
#
|
||
|
|
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
|
||
|
|
# ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
|
||
|
|
# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
|
||
|
|
# DISCLAIMED. IN NO EVENT SHALL ZSH-USERS BE LIABLE FOR ANY
|
||
|
|
# DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
|
||
|
|
# (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
|
||
|
|
# LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
|
||
|
|
# ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
||
|
|
# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
|
||
|
|
# SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||
|
|
# ------------------------------------------------------------------------------
|
||
|
|
# Description
|
||
|
|
# -----------
|
||
|
|
#
|
||
|
|
# Completion script for The Uncomplicated Firewall (ufw). (https://launchpad.net/ufw).
|
||
|
|
#
|
||
|
|
# ------------------------------------------------------------------------------
|
||
|
|
# Authors
|
||
|
|
# -------
|
||
|
|
#
|
||
|
|
# * Jindřich Pilař (https://github.com/JindrichPilar)
|
||
|
|
#
|
||
|
|
# ------------------------------------------------------------------------------
|
||
|
|
|
||
|
|
|
||
|
|
_ufw_logging() {
|
||
|
|
|
||
|
|
local params additional second
|
||
|
|
second=$words[2]
|
||
|
|
|
||
|
|
if [ ! -z $second ]; then
|
||
|
|
return
|
||
|
|
fi
|
||
|
|
|
||
|
|
params=(
|
||
|
|
"on"
|
||
|
|
"off"
|
||
|
|
)
|
||
|
|
|
||
|
|
additional=(
|
||
|
|
"low"
|
||
|
|
"medium"
|
||
|
|
"high"
|
||
|
|
"full"
|
||
|
|
)
|
||
|
|
|
||
|
|
_describe -t params 'on/off' params
|
||
|
|
_describe -t additional 'level' additional
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
_ufw_delete() {
|
||
|
|
local rules complrules second
|
||
|
|
|
||
|
|
second=$words[2]
|
||
|
|
|
||
|
|
if [ ! -z $second ]; then
|
||
|
|
return
|
||
|
|
fi
|
||
|
|
|
||
|
|
complrules=()
|
||
|
|
rules=("${(f)$(ufw status | tr -s ' ' | tail -n +5 | tr -s '\n')}")
|
||
|
|
|
||
|
|
for ((i=1; i<=${#rules[@]}; i++)); do
|
||
|
|
complrules+=("$i:$rules[i]");
|
||
|
|
done
|
||
|
|
|
||
|
|
_describe -t complrules 'Rules' complrules
|
||
|
|
}
|
||
|
|
|
||
|
|
_ufw() {
|
||
|
|
local curcontext="$curcontext" ret=1
|
||
|
|
local -a state line commands
|
||
|
|
|
||
|
|
commands=(
|
||
|
|
"enable:enable the firewall"
|
||
|
|
"disable:disable the firewall"
|
||
|
|
"default:set default policy"
|
||
|
|
"logging:set logging level"
|
||
|
|
"allow:add allow rule"
|
||
|
|
"deny:add deny rule"
|
||
|
|
"reject:add reject rule"
|
||
|
|
"limit:add limit rule"
|
||
|
|
"delete:delete rule"
|
||
|
|
"insert:insert rule at position"
|
||
|
|
"route:add route rule"
|
||
|
|
"reload:reload firewall"
|
||
|
|
"reset:reset firewall"
|
||
|
|
"status:show firewall status"
|
||
|
|
"show:show firewall report"
|
||
|
|
"version:display version information"
|
||
|
|
"prepend:add rule before all of the same type"
|
||
|
|
)
|
||
|
|
|
||
|
|
_arguments -C -s -S -n \
|
||
|
|
'(- 1 *)'--version"[display version information]: :->full" \
|
||
|
|
'(- 1 *)'{-h,--help}'[display usage information]: :->full' \
|
||
|
|
'(- 1 *)'--dry-run"[don't modify anything, just show the changes]: :->cmds" \
|
||
|
|
'1:cmd:->cmds' \
|
||
|
|
'*:: :->args' && ret=0
|
||
|
|
|
||
|
|
case "$state" in
|
||
|
|
(cmds)
|
||
|
|
_describe -t commands 'commands' commands
|
||
|
|
;;
|
||
|
|
(args)
|
||
|
|
local cmd
|
||
|
|
cmd=$words[1]
|
||
|
|
case "$cmd" in
|
||
|
|
(logging)
|
||
|
|
_ufw_logging && ret=0
|
||
|
|
;;
|
||
|
|
(delete)
|
||
|
|
_ufw_delete && ret=0
|
||
|
|
;;
|
||
|
|
(*)
|
||
|
|
_default && ret=0
|
||
|
|
;;
|
||
|
|
esac
|
||
|
|
;;
|
||
|
|
(*)
|
||
|
|
;;
|
||
|
|
esac
|
||
|
|
|
||
|
|
return ret
|
||
|
|
}
|
||
|
|
|
||
|
|
_ufw
|
||
|
|
|