mirror of
https://github.com/netbirdio/netbird.git
synced 2025-08-07 22:39:14 +02:00
af8687579b
This will allow running netbird commands (including debugging) against the daemon and provide a flow similar to non-container usages. It will by default both log to file and stderr so it can be handled more uniformly in container-native environments.
106 lines
2.9 KiB
Bash
Executable File
106 lines
2.9 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
set -eEuo pipefail
|
|
|
|
: ${NB_ENTRYPOINT_SERVICE_TIMEOUT:="5"}
|
|
: ${NB_ENTRYPOINT_LOGIN_TIMEOUT:="1"}
|
|
NETBIRD_BIN="${NETBIRD_BIN:-"netbird"}"
|
|
export NB_LOG_FILE="${NB_LOG_FILE:-"console,/var/log/netbird/client.log"}"
|
|
service_pids=()
|
|
log_file_path=""
|
|
|
|
_log() {
|
|
# mimic Go logger's output for easier parsing
|
|
# 2025-04-15T21:32:00+08:00 INFO client/internal/config.go:495: setting notifications to disabled by default
|
|
printf "$(date -Isec) ${1} ${BASH_SOURCE[1]}:${BASH_LINENO[1]}: ${2}\n" "${@:3}" >&2
|
|
}
|
|
|
|
info() {
|
|
_log INFO "$@"
|
|
}
|
|
|
|
warn() {
|
|
_log WARN "$@"
|
|
}
|
|
|
|
on_exit() {
|
|
info "Shutting down NetBird daemon..."
|
|
if test "${#service_pids[@]}" -gt 0; then
|
|
info "terminating service process IDs: ${service_pids[@]@Q}"
|
|
kill -TERM "${service_pids[@]}" 2>/dev/null || true
|
|
wait "${service_pids[@]}" 2>/dev/null || true
|
|
else
|
|
info "there are no service processes to terminate"
|
|
fi
|
|
}
|
|
|
|
wait_for_message() {
|
|
local timeout="${1}" message="${2}"
|
|
if test "${timeout}" -eq 0; then
|
|
info "not waiting for log line ${message@Q} due to zero timeout."
|
|
elif test -n "${log_file_path}"; then
|
|
info "waiting for log line ${message@Q} for ${timeout} seconds..."
|
|
grep -q "${message}" <(timeout "${timeout}" tail -F "${log_file_path}" 2>/dev/null)
|
|
else
|
|
info "log file unsupported, sleeping for ${timeout} seconds..."
|
|
sleep "${timeout}"
|
|
fi
|
|
}
|
|
|
|
locate_log_file() {
|
|
local log_files_string="${1}"
|
|
|
|
while read -r log_file; do
|
|
case "${log_file}" in
|
|
console | syslog) ;;
|
|
*)
|
|
log_file_path="${log_file}"
|
|
return
|
|
;;
|
|
esac
|
|
done < <(sed 's#,#\n#g' <<<"${log_files_string}")
|
|
|
|
warn "log files parsing for ${log_files_string@Q} is not supported by debug bundles"
|
|
warn "please consider removing the \$NB_LOG_FILE or setting it to real file, before gathering debug bundles."
|
|
}
|
|
|
|
wait_for_daemon_startup() {
|
|
local timeout="${1}"
|
|
|
|
if test -n "${log_file_path}"; then
|
|
if ! wait_for_message "${timeout}" "started daemon server"; then
|
|
warn "log line containing 'started daemon server' not found after ${timeout} seconds"
|
|
warn "daemon failed to start, exiting..."
|
|
exit 1
|
|
fi
|
|
else
|
|
warn "daemon service startup not discovered, sleeping ${timeout} instead"
|
|
sleep "${timeout}"
|
|
fi
|
|
}
|
|
|
|
login_if_needed() {
|
|
local timeout="${1}"
|
|
|
|
if test -n "${log_file_path}" && wait_for_message "${timeout}" 'peer has been successfully registered'; then
|
|
info "already logged in, skipping 'netbird up'..."
|
|
else
|
|
info "logging in..."
|
|
"${NETBIRD_BIN}" up
|
|
fi
|
|
}
|
|
|
|
main() {
|
|
trap 'on_exit' SIGTERM SIGINT EXIT
|
|
"${NETBIRD_BIN}" service run &
|
|
service_pids+=("$!")
|
|
info "registered new service process 'netbird service run', currently running: ${service_pids[@]@Q}"
|
|
|
|
locate_log_file "${NB_LOG_FILE}"
|
|
wait_for_daemon_startup "${NB_ENTRYPOINT_SERVICE_TIMEOUT}"
|
|
login_if_needed "${NB_ENTRYPOINT_LOGIN_TIMEOUT}"
|
|
|
|
wait "${service_pids[@]}"
|
|
}
|
|
|
|
main "$@"
|