mirror of
https://github.com/netbirdio/netbird.git
synced 2026-03-31 06:34:19 -04:00
ead1c618ba
optimizes the NetBird client startup process by avoiding unnecessary login commands when the peer is already authenticated. The changes increase the default login timeout and expand the log message patterns used to detect successful authentication. - Increased default login timeout from 1 to 5 seconds for more reliable authentication detection - Enhanced log pattern matching to detect both registration and ready states - Added extended regex support for more flexible pattern matching
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:="5"}
|
|
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 -E -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|management connection state READY'; 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 "$@"
|