362 lines
11 KiB
Bash
Executable File
362 lines
11 KiB
Bash
Executable File
#!/bin/bash
|
|
set -euo pipefail
|
|
[[ -n "${DEBUG:-}" ]] && set -x
|
|
|
|
PATH="/bin:/usr/bin:/sbin:/usr/sbin"
|
|
|
|
UFW_ACTION="${1:-help}"
|
|
|
|
GREP_REGEXP_INSTANCE_NAME="[-_.[:alnum:]]\\+"
|
|
DEFAULT_PROTO=tcp
|
|
|
|
function ufw-docker--status() {
|
|
ufw-docker--list "$GREP_REGEXP_INSTANCE_NAME"
|
|
}
|
|
|
|
function ufw-docker--list() {
|
|
local INSTANCE_NAME="$1"
|
|
local INSTANCE_PORT="${2:-}"
|
|
local PROTO="${3:-${DEFAULT_PROTO}}"
|
|
|
|
if [[ -z "$INSTANCE_PORT" ]]; then
|
|
INSTANCE_PORT="[[:digit:]]\\+"
|
|
PROTO="\\(tcp\\|udp\\)"
|
|
fi
|
|
ufw status numbered | grep "# allow ${INSTANCE_NAME}\\( ${INSTANCE_PORT}\\/${PROTO}\\)\\?\$"
|
|
}
|
|
|
|
function ufw-docker--list-number() {
|
|
ufw-docker--list "$@" | sed -e 's/^\[[[:blank:]]*\([[:digit:]]\+\)\].*/\1/'
|
|
}
|
|
|
|
function ufw-docker--delete() {
|
|
for UFW_NUMBER in $(ufw-docker--list-number "$@" | sort -rn); do
|
|
echo "delete \"$UFW_NUMBER\""
|
|
echo y | ufw delete "$UFW_NUMBER" || true
|
|
done
|
|
}
|
|
|
|
function ufw-docker--allow() {
|
|
local INSTANCE_NAME="$1"
|
|
local INSTANCE_PORT="$2"
|
|
local PROTO="$3"
|
|
|
|
docker inspect "$INSTANCE_NAME" &>/dev/null ||
|
|
die "Docker instance \"$INSTANCE_NAME\" doesn't exist."
|
|
|
|
mapfile -t INSTANCE_IP_ADDRESSES < <(docker inspect --format='{{range .NetworkSettings.Networks}}{{.IPAddress}}{{"\n"}}{{end}}' "$INSTANCE_NAME" 2>/dev/null | remove_blank_lines)
|
|
|
|
[[ -z "${INSTANCE_IP_ADDRESSES:-}" ]] && die "Could not find a running instance \"$INSTANCE_NAME\"."
|
|
|
|
mapfile -t PORT_PROTO_LIST < <(docker inspect --format='{{range $p, $conf := .NetworkSettings.Ports}}{{with $conf}}{{$p}}{{"\n"}}{{end}}{{end}}' "$INSTANCE_NAME" | remove_blank_lines)
|
|
|
|
if [[ "${UFW_DOCKER_FORCE_ADD:-}" = "yes" ]]; then
|
|
for IP in "${INSTANCE_IP_ADDRESSES[@]}"; do
|
|
ufw-docker--add-rule "$INSTANCE_NAME" "$IP" "${INSTANCE_PORT}" "${PROTO}"
|
|
done
|
|
return
|
|
fi
|
|
|
|
if [[ -z "${PORT_PROTO_LIST[@]:-}" ]]; then
|
|
err "\"$INSTANCE_NAME\" doesn't have any published ports."
|
|
return 1
|
|
fi
|
|
|
|
RETVAL=1
|
|
for PORT_PROTO in "${PORT_PROTO_LIST[@]}"; do
|
|
if [[ -z "$INSTANCE_PORT" || "$PORT_PROTO" = "${INSTANCE_PORT}/${PROTO}" ]]; then
|
|
for IP in "${INSTANCE_IP_ADDRESSES[@]}"; do
|
|
ufw-docker--add-rule "$INSTANCE_NAME" "$IP" "${PORT_PROTO%/*}" "${PORT_PROTO#*/}"
|
|
RETVAL="$?"
|
|
done
|
|
fi
|
|
done
|
|
if [[ "$RETVAL" -ne 0 ]]; then
|
|
err "Fail to add rule(s), cannot find the published port ${INSTANCE_PORT}/${PROTO} of instance \"${INSTANCE_NAME}\" or cannot update outdated rule(s)."
|
|
fi
|
|
return "$RETVAL"
|
|
}
|
|
|
|
function ufw-docker--add-service-rule() {
|
|
declare service_id="$1"
|
|
declare port="${2%/*}"
|
|
declare proto="${2#*/}"
|
|
|
|
declare target_ip_port="$(iptables -t nat -L DOCKER-INGRESS | grep -E "^DNAT\\s+${proto}\\s+.+\\sto:[.0-9]+:${port}\$" | grep -Eo "[.0-9]+:${port}\$")"
|
|
|
|
[[ -z "$target_ip_port" ]] && die "Could not find VIP of service ${service_id}."
|
|
|
|
ufw-docker--add-rule "$service_id" "${target_ip_port%:*}" "$port" "$proto"
|
|
}
|
|
|
|
function ufw-docker--add-rule() {
|
|
local INSTANCE_NAME="$1"
|
|
local INSTANCE_IP_ADDRESS="$2"
|
|
local PORT="$3"
|
|
local PROTO="$4"
|
|
|
|
echo "allow ${INSTANCE_NAME} ${PORT}/${PROTO}"
|
|
typeset -a UFW_OPTS
|
|
UFW_OPTS=(route allow proto "${PROTO}"
|
|
from any to "$INSTANCE_IP_ADDRESS")
|
|
comment="allow ${INSTANCE_NAME}"
|
|
[[ -n "$PORT" ]] && {
|
|
UFW_OPTS+=(port "${PORT}")
|
|
comment=("$comment ${PORT}/${PROTO}")
|
|
}
|
|
UFW_OPTS+=(comment "$comment")
|
|
|
|
if ufw-docker--list "$INSTANCE_NAME" "$PORT" "$PROTO" &>/dev/null; then
|
|
ufw --dry-run "${UFW_OPTS[@]}" | grep "^Skipping" && return 0
|
|
err "Remove outdated rule."
|
|
ufw-docker--delete "$INSTANCE_NAME" "$PORT" "$PROTO"
|
|
fi
|
|
echo " ${UFW_OPTS[@]}"
|
|
ufw "${UFW_OPTS[@]}"
|
|
}
|
|
|
|
function ufw-docker--instance-name() {
|
|
local INSTANCE_ID="$1"
|
|
{
|
|
{
|
|
docker inspect --format='{{.Name}}' "$INSTANCE_ID" 2>/dev/null | sed -e 's,^/,,' |
|
|
grep "^${GREP_REGEXP_INSTANCE_NAME}\$" 2>/dev/null
|
|
} || echo -n "$INSTANCE_ID";
|
|
} | remove_blank_lines
|
|
}
|
|
|
|
function ufw-docker--service() {
|
|
declare service_action="${1:-help}"
|
|
case "$service_action" in
|
|
delete)
|
|
shift || true
|
|
if [[ "${1:?Invalid 'delete' command syntax.}" != "allow" ]]; then
|
|
die "\"delete\" command only support removing allowed rules"
|
|
fi
|
|
shift || true
|
|
declare service_id_or_name="${1:?Missing swarm service name or service ID}"
|
|
declare service_name="$(docker service inspect "$service_id_or_name" --format '{{.Spec.Name}}')"
|
|
|
|
"ufw-docker--service-${service_action}" "${service_name}"
|
|
;;
|
|
allow)
|
|
shift || true
|
|
declare service_id_or_name="${1:?Missing swarm service name or service ID}"
|
|
declare service_name="$(docker service inspect "$service_id_or_name" --format '{{.Spec.Name}}')"
|
|
declare service_port="${2:?Missing the port number, such as '80/tcp'.}"
|
|
|
|
"ufw-docker--service-${service_action}" "${service_name}" "${service_port}"
|
|
;;
|
|
*)
|
|
ufw-docker--help
|
|
;;
|
|
esac
|
|
}
|
|
|
|
ufw_docker_agent=ufw-docker-agent
|
|
ufw_docker_agent_image="${ufw_docker_agent_image:-chaifeng/${ufw_docker_agent}:181003}"
|
|
|
|
function ufw-docker--get-service-id() {
|
|
declare service_name="$1"
|
|
docker service inspect "${service_name}" --format "{{.ID}}"
|
|
}
|
|
|
|
function ufw-docker--get-service-name() {
|
|
declare service_name="$1"
|
|
docker service inspect "${service_name}" --format "{{.Spec.Name}}"
|
|
}
|
|
|
|
function ufw-docker--service-allow() {
|
|
declare service_name="$1"
|
|
declare service_port="$2"
|
|
declare service_proto=tcp
|
|
|
|
if [[ -n "$service_port" ]] &&
|
|
! grep -E '^[0-9]+(/(tcp|udp))?$' <<< "$service_port" &>/dev/null; then
|
|
die "Invalid port syntax: $service_port"
|
|
return 1
|
|
fi
|
|
|
|
if [[ "$service_port" = */* ]]; then
|
|
service_proto="${service_port#*/}"
|
|
service_port="${service_port%/*}"
|
|
fi
|
|
|
|
declare service_id="$(ufw-docker--get-service-id "${service_name}")"
|
|
service_name="$(ufw-docker--get-service-name "${service_name}")"
|
|
|
|
declare -a service_env
|
|
|
|
exec 9< <(docker service inspect "$service_name" \
|
|
--format '{{range .Endpoint.Spec.Ports}}{{.PublishedPort}} {{.TargetPort}}/{{.Protocol}}{{"\n"}}{{end}}')
|
|
while read -u 9 port target_port; do
|
|
if [[ "$target_port" = "${service_port}/${service_proto}" ]]; then
|
|
service_env="ufw_public_${service_id}=${service_name}/${port}/${service_proto}"
|
|
break;
|
|
fi
|
|
done
|
|
exec 9<&-
|
|
|
|
if [[ -z "${service_env:-}" ]]; then
|
|
die "Service $service_name does not publish port $service_port."
|
|
return 1
|
|
fi
|
|
|
|
if ! docker service inspect "$ufw_docker_agent" &>/dev/null; then
|
|
err "Not found ufw-docker-agent service, creating ..."
|
|
docker service create --name "$ufw_docker_agent" --mode global \
|
|
--mount type=bind,source=/var/run/docker.sock,target=/var/run/docker.sock \
|
|
--mount type=bind,source=/etc/ufw,target=/etc/ufw,readonly=true \
|
|
--env ufw_docker_agent_image="${ufw_docker_agent_image}" \
|
|
--env DEBUG="${DEBUG:-}" \
|
|
--env "${service_env}" \
|
|
"${ufw_docker_agent_image}"
|
|
else
|
|
docker service update --update-parallelism=0 \
|
|
--env-add ufw_docker_agent_image="${ufw_docker_agent_image}" \
|
|
--env-add DEBUG="${DEBUG:-}" \
|
|
--env-add "${service_env}" \
|
|
--image "${ufw_docker_agent_image}" \
|
|
"${ufw_docker_agent}"
|
|
fi
|
|
}
|
|
|
|
function ufw-docker--service-delete() {
|
|
declare service_name="$1"
|
|
|
|
declare service_id="$(ufw-docker--get-service-id "${service_name}")"
|
|
service_name="$(ufw-docker--get-service-name "${service_name}")"
|
|
declare service_env="ufw_public_${service_id}=${service_name}/deny"
|
|
|
|
docker service update --update-parallelism=0 \
|
|
--env-add ufw_docker_agent_image="${ufw_docker_agent_image}" \
|
|
--env-add "${service_env}" \
|
|
--image "${ufw_docker_agent_image}" \
|
|
"${ufw_docker_agent}"
|
|
}
|
|
|
|
function ufw-docker--raw-command() {
|
|
ufw "$@"
|
|
}
|
|
|
|
function ufw-docker--install() {
|
|
if ! grep "^# BEGIN UFW AND DOCKER\$" /etc/ufw/after.rules &>/dev/null; then
|
|
err "Back up /etc/ufw/after.rules"
|
|
cp /etc/ufw/after.rules /etc/ufw/after.rules-ufw-docker~"$(date '+%Y-%m-%d-%H%M%S').bak"
|
|
cat <<-\EOF | tee -a /etc/ufw/after.rules
|
|
# BEGIN UFW AND DOCKER
|
|
*filter
|
|
:ufw-user-forward - [0:0]
|
|
:DOCKER-USER - [0:0]
|
|
-A DOCKER-USER -j RETURN -s 10.0.0.0/8
|
|
-A DOCKER-USER -j RETURN -s 172.16.0.0/12
|
|
-A DOCKER-USER -j RETURN -s 192.168.0.0/16
|
|
|
|
-A DOCKER-USER -j ufw-user-forward
|
|
|
|
-A DOCKER-USER -j DROP -d 192.168.0.0/16 -p tcp -m tcp --tcp-flags FIN,SYN,RST,ACK SYN
|
|
-A DOCKER-USER -j DROP -d 10.0.0.0/8 -p tcp -m tcp --tcp-flags FIN,SYN,RST,ACK SYN
|
|
-A DOCKER-USER -j DROP -d 172.16.0.0/12 -p tcp -m tcp --tcp-flags FIN,SYN,RST,ACK SYN
|
|
-A DOCKER-USER -j DROP -d 192.168.0.0/16 -p udp -m udp --dport 0:32767
|
|
-A DOCKER-USER -j DROP -d 10.0.0.0/8 -p udp -m udp --dport 0:32767
|
|
-A DOCKER-USER -j DROP -d 172.16.0.0/12 -p udp -m udp --dport 0:32767
|
|
|
|
-A DOCKER-USER -j RETURN
|
|
COMMIT
|
|
# END UFW AND DOCKER
|
|
EOF
|
|
err "Please restart UFW service manually."
|
|
fi
|
|
}
|
|
|
|
function ufw-docker--help() {
|
|
cat <<-EOF >&2
|
|
Usage:
|
|
ufw-docker <list|allow> [docker-instance-id-or-name [port[/tcp|/udp]]]
|
|
ufw-docker delete allow [docker-instance-id-or-name [port[/tcp|/udp]]]
|
|
|
|
ufw-docker service allow <swarm-service-id-or-name <port</tcp|/udp>>>
|
|
ufw-docker service delete allow <swarm-service-id-or-name>
|
|
|
|
ufw-docker <status|install|help>
|
|
|
|
Examples:
|
|
ufw-docker help
|
|
ufw-docker install
|
|
|
|
ufw-docker status
|
|
|
|
ufw-docker list httpd
|
|
|
|
|
|
ufw-docker allow httpd
|
|
ufw-docker allow httpd 80
|
|
ufw-docker allow httpd 80/tcp
|
|
|
|
ufw-docker delete allow httpd
|
|
ufw-docker delete allow httpd 80/tcp
|
|
|
|
|
|
ufw-docker service allow httpd 80/tcp
|
|
|
|
ufw-docker service delete allow httpd
|
|
EOF
|
|
}
|
|
|
|
function remove_blank_lines() {
|
|
sed '/^[[:blank:]]*$/d'
|
|
}
|
|
|
|
function err() {
|
|
echo "$@" >&2
|
|
}
|
|
|
|
function die() {
|
|
err "ERROR:" "$@"
|
|
exit 1
|
|
}
|
|
|
|
if ! ufw status 2>/dev/null | grep -Fq "Status: active" ; then
|
|
die "UFW is disabled or you are not root user."
|
|
fi
|
|
|
|
case "$UFW_ACTION" in
|
|
delete)
|
|
shift || true
|
|
if [[ "${1:?Invalid 'delete' command syntax.}" != "allow" ]]; then
|
|
die "\"delete\" command only support removing allowed rules"
|
|
fi
|
|
;&
|
|
list|allow)
|
|
shift || true
|
|
|
|
INSTANCE_ID="${1:?Docker instance name/ID cannot be empty.}"
|
|
INSTANCE_NAME="$(ufw-docker--instance-name "$INSTANCE_ID")"
|
|
shift || true
|
|
|
|
INSTANCE_PORT="${1:-}"
|
|
if [[ -n "$INSTANCE_PORT" && ! "$INSTANCE_PORT" =~ [0-9]+(/(tcp|udp))? ]]; then
|
|
die "invalid port syntax: \"$INSTANCE_PORT\"."
|
|
fi
|
|
|
|
PROTO="$DEFAULT_PROTO"
|
|
if [[ "$INSTANCE_PORT" = */udp ]]; then
|
|
PROTO=udp
|
|
fi
|
|
|
|
INSTANCE_PORT="${INSTANCE_PORT%/*}"
|
|
|
|
"ufw-docker--$UFW_ACTION" "$INSTANCE_NAME" "$INSTANCE_PORT" "$PROTO"
|
|
;;
|
|
service|raw-command|add-service-rule)
|
|
shift || true
|
|
"ufw-docker--$UFW_ACTION" "$@"
|
|
;;
|
|
status|install)
|
|
ufw-docker--"$UFW_ACTION"
|
|
;;
|
|
*)
|
|
ufw-docker--help
|
|
;;
|
|
esac
|