package/network/utils/wireguard-tools/files/wireguard.sh

   1 #!/bin/sh
   2 # Copyright 2016-2017 Dan Luedtke <mail@danrl.com>
   3 # Licensed to the public under the Apache License 2.0.
   4
   5 WG=/usr/bin/wg
   6 if [ ! -x $WG ]; then
   7         logger -t "wireguard" "error: missing wireguard-tools (${WG})"
   8         exit 0
   9 fi
  10
  11 [ -n "$INCLUDE_ONLY" ] || {
  12         . /lib/functions.sh
  13         . ../netifd-proto.sh
  14         init_proto "$@"
  15 }
  16
  17 proto_wireguard_init_config() {
  18         proto_config_add_string "private_key"
  19         proto_config_add_int "listen_port"
  20         proto_config_add_int "mtu"
  21         proto_config_add_string "fwmark"
  22         available=1
  23         no_proto_task=1
  24 }
  25
  26 proto_wireguard_setup_peer() {
  27         local peer_config="$1"
  28
  29         local public_key
  30         local preshared_key
  31         local allowed_ips
  32         local route_allowed_ips
  33         local endpoint_host
  34         local endpoint_port
  35         local persistent_keepalive
  36
  37         config_get public_key "${peer_config}" "public_key"
  38         config_get preshared_key "${peer_config}" "preshared_key"
  39         config_get allowed_ips "${peer_config}" "allowed_ips"
  40         config_get_bool route_allowed_ips "${peer_config}" "route_allowed_ips" 0
  41         config_get endpoint_host "${peer_config}" "endpoint_host"
  42         config_get endpoint_port "${peer_config}" "endpoint_port"
  43         config_get persistent_keepalive "${peer_config}" "persistent_keepalive"
  44
  45         if [ -z "$public_key" ]; then
  46                 echo "Skipping peer config $peer_config because public key is not defined."
  47                 return 0
  48         fi
  49
  50         echo "[Peer]" >> "${wg_cfg}"
  51         echo "PublicKey=${public_key}" >> "${wg_cfg}"
  52         if [ "${preshared_key}" ]; then
  53                 echo "PresharedKey=${preshared_key}" >> "${wg_cfg}"
  54         fi
  55         for allowed_ip in $allowed_ips; do
  56                 echo "AllowedIPs=${allowed_ip}" >> "${wg_cfg}"
  57         done
  58         if [ "${endpoint_host}" ]; then
  59                 case "${endpoint_host}" in
  60                         *:*)
  61                                 endpoint="[${endpoint_host}]"
  62                                 ;;
  63                         *)
  64                                 endpoint="${endpoint_host}"
  65                                 ;;
  66                 esac
  67                 if [ "${endpoint_port}" ]; then
  68                         endpoint="${endpoint}:${endpoint_port}"
  69                 else
  70                         endpoint="${endpoint}:51820"
  71                 fi
  72                 echo "Endpoint=${endpoint}" >> "${wg_cfg}"
  73         fi
  74         if [ "${persistent_keepalive}" ]; then
  75                 echo "PersistentKeepalive=${persistent_keepalive}" >> "${wg_cfg}"
  76         fi
  77
  78         if [ ${route_allowed_ips} -ne 0 ]; then
  79                 for allowed_ip in ${allowed_ips}; do
  80                         case "${allowed_ip}" in
  81                                 *:*/*)
  82                                         proto_add_ipv6_route "${allowed_ip%%/*}" "${allowed_ip##*/}"
  83                                         ;;
  84                                 *.*/*)
  85                                         proto_add_ipv4_route "${allowed_ip%%/*}" "${allowed_ip##*/}"
  86                                         ;;
  87                                 *:*)
  88                                         proto_add_ipv6_route "${allowed_ip%%/*}" "128"
  89                                         ;;
  90                                 *.*)
  91                                         proto_add_ipv4_route "${allowed_ip%%/*}" "32"
  92                                         ;;
  93                         esac
  94                 done
  95         fi
  96 }
  97
  98 proto_wireguard_setup() {
  99         local config="$1"
 100         local wg_dir="/tmp/wireguard"
 101         local wg_cfg="${wg_dir}/${config}"
 102
 103         local private_key
 104         local listen_port
 105         local mtu
 106
 107         config_load network
 108         config_get private_key "${config}" "private_key"
 109         config_get listen_port "${config}" "listen_port"
 110         config_get addresses "${config}" "addresses"
 111         config_get mtu "${config}" "mtu"
 112         config_get fwmark "${config}" "fwmark"
 113         config_get ip6prefix "${config}" "ip6prefix"
 114         config_get nohostroute "${config}" "nohostroute"
 115
 116         ip link del dev "${config}" 2>/dev/null
 117         ip link add dev "${config}" type wireguard
 118
 119         if [ "${mtu}" ]; then
 120                 ip link set mtu "${mtu}" dev "${config}"
 121         fi
 122
 123         proto_init_update "${config}" 1
 124
 125         umask 077
 126         mkdir -p "${wg_dir}"
 127         echo "[Interface]" > "${wg_cfg}"
 128         echo "PrivateKey=${private_key}" >> "${wg_cfg}"
 129         if [ "${listen_port}" ]; then
 130                 echo "ListenPort=${listen_port}" >> "${wg_cfg}"
 131         fi
 132         if [ "${fwmark}" ]; then
 133                 echo "FwMark=${fwmark}" >> "${wg_cfg}"
 134         fi
 135         config_foreach proto_wireguard_setup_peer "wireguard_${config}"
 136
 137         # apply configuration file
 138         ${WG} setconf ${config} "${wg_cfg}"
 139         WG_RETURN=$?
 140
 141         rm -f "${wg_cfg}"
 142
 143         if [ ${WG_RETURN} -ne 0 ]; then
 144                 sleep 5
 145                 proto_setup_failed "${config}"
 146                 exit 1
 147         fi
 148
 149         for address in ${addresses}; do
 150                 case "${address}" in
 151                         *:*/*)
 152                                 proto_add_ipv6_address "${address%%/*}" "${address##*/}"
 153                                 ;;
 154                         *.*/*)
 155                                 proto_add_ipv4_address "${address%%/*}" "${address##*/}"
 156                                 ;;
 157                         *:*)
 158                                 proto_add_ipv6_address "${address%%/*}" "128"
 159                                 ;;
 160                         *.*)
 161                                 proto_add_ipv4_address "${address%%/*}" "32"
 162                                 ;;
 163                 esac
 164         done
 165
 166         for prefix in ${ip6prefix}; do
 167                 proto_add_ipv6_prefix "$prefix"
 168         done
 169
 170         # endpoint dependency
 171         if [ "${nohostroute}" != "1" ]; then
 172                 wg show "${config}" endpoints | \
 173                 sed -E 's/\[?([0-9.:a-f]+)\]?:([0-9]+)/\1 \2/' | \
 174                 while IFS=$'\t ' read -r key address port; do
 175                         [ -n "${port}" ] || continue
 176                         proto_add_host_dependency "${config}" "${address}"
 177                 done
 178         fi
 179
 180         proto_send_update "${config}"
 181 }
 182
 183 proto_wireguard_teardown() {
 184         local config="$1"
 185         ip link del dev "${config}" >/dev/null 2>&1
 186 }
 187
 188 [ -n "$INCLUDE_ONLY" ] || {
 189         add_protocol wireguard
 190 }