108 lines
2.9 KiB
Bash
Executable File
108 lines
2.9 KiB
Bash
Executable File
#!/bin/sh /etc/rc.common
|
|
# Copyright 2021 Stan Grishin (stangri@melmac.ca)
|
|
# shellcheck disable=SC3043,SC3060
|
|
|
|
# shellcheck disable=SC2034
|
|
START=80
|
|
# shellcheck disable=SC2034
|
|
USE_PROCD=1
|
|
|
|
readonly PKG_VERSION='dev-test'
|
|
readonly packageName='nebula-service'
|
|
readonly serviceName="$packageName $PKG_VERSION"
|
|
readonly sharedMemoryOutput="/dev/shm/$packageName-output"
|
|
readonly PROG=/usr/sbin/nebula
|
|
readonly _OK_='\033[0;32m\xe2\x9c\x93\033[0m'
|
|
readonly _FAIL_='\033[0;31m\xe2\x9c\x97\033[0m'
|
|
|
|
extra_command 'version' 'Show version information'
|
|
|
|
version() { echo "Version: $PKG_VERSION"; }
|
|
|
|
output() {
|
|
local msg memmsg logmsg
|
|
[ -t 1 ] && printf "%b" "$@"
|
|
msg="${1//$serviceName /service }";
|
|
if [ "$(printf "%b" "$msg" | wc -l)" -gt 0 ]; then
|
|
[ -s "$sharedMemoryOutput" ] && memmsg="$(cat "$sharedMemoryOutput")"
|
|
logmsg="$(printf "%b" "${memmsg}${msg}" | sed 's/\x1b\[[0-9;]*m//g')"
|
|
logger -t "$packageName" "$(printf "%b" "$logmsg")"
|
|
rm -f "$sharedMemoryOutput"
|
|
else
|
|
printf "%b" "$msg" >> "$sharedMemoryOutput"
|
|
fi
|
|
}
|
|
output_ok() { output "$_OK_"; }
|
|
output_okn() { output "${_OK_}\\n"; }
|
|
output_fail() { output "$_FAIL_"; }
|
|
output_failn() { output "${_FAIL_}\\n"; }
|
|
|
|
# https://gist.github.com/pkuczynski/8665367
|
|
# shellcheck disable=SC2086,SC2155
|
|
parse_yaml() {
|
|
local prefix=$2
|
|
local s='[[:space:]]*' w='[a-zA-Z0-9_-]*' fs="$(echo @|tr @ '\034'|tr -d '\015')"
|
|
sed -ne "s|^\($s\)\($w\)$s:$s\"\(.*\)\"$s\$|\1$fs\2$fs\3|p" \
|
|
-e "s|^\($s\)\($w\)$s:$s\(.*\)$s\$|\1$fs\2$fs\3|p" "$1" |
|
|
awk "-F$fs" '{
|
|
indent = length($1)/2;
|
|
vname[indent] = $2;
|
|
for (i in vname) {if (i > indent) {delete vname[i]}}
|
|
if (length($3) > 0) {
|
|
vn=""; for (i=0; i<indent; i++) {vn=(vn)(vname[i])("_")}
|
|
printf("%s%s%s=\"%s\"\n", "'$prefix'", vn, $2, $3);
|
|
}
|
|
}'
|
|
}
|
|
|
|
start_instance() {
|
|
local config_file="$1"
|
|
local yaml_listen_host yaml_listen_port
|
|
if [ ! -x "$PROG" ]; then
|
|
echo "Nebula binary not found! Please install 'nebula' package."
|
|
output_fail
|
|
return 1
|
|
fi
|
|
if [ ! -s "$config_file" ]; then
|
|
output_fail
|
|
return 1
|
|
fi
|
|
if ! eval "$(parse_yaml "$config_file" "yaml_")"; then
|
|
output_fail
|
|
return 1
|
|
else
|
|
procd_open_instance
|
|
procd_set_param command ${PROG} -config "${config_file}"
|
|
procd_set_param stderr 1
|
|
procd_set_param stdout 1
|
|
procd_set_param respawn
|
|
procd_open_data
|
|
json_add_array firewall
|
|
json_add_object ""
|
|
json_add_string type rule
|
|
json_add_string name "${config_file##*/}"
|
|
json_add_string src "*"
|
|
json_add_string dest_ip "${yaml_listen_host:-0.0.0.0}"
|
|
json_add_string dest_port "${yaml_listen_port:-4242}"
|
|
json_add_string proto udp
|
|
json_add_string target ACCEPT
|
|
json_close_object
|
|
json_close_array
|
|
procd_close_data
|
|
procd_close_instance
|
|
output_ok
|
|
fi
|
|
}
|
|
|
|
start_service() {
|
|
local f
|
|
output "Starting $packageName instances "
|
|
for f in /etc/nebula/*.yml; do
|
|
start_instance "$f"
|
|
done
|
|
output '\n'
|
|
}
|
|
|
|
service_started() { procd_set_config_changed firewall; }
|
|
service_stopped() { procd_set_config_changed firewall; }
|