52 lines
1.7 KiB
Bash
Executable File
52 lines
1.7 KiB
Bash
Executable File
#!/bin/bash
|
|
# FILE: peer_list
|
|
# DESCRIPTION: Add a new peer
|
|
# USAGE: add remote_ip
|
|
# ERRORS:
|
|
# 3: bad args/usage
|
|
# 4: config file not found
|
|
# 5: Wireguard not installed
|
|
# 9: Token file not found
|
|
# 10: Failed to get peer data from wg
|
|
# 11: Failed to generate new IPs
|
|
# 12: Servers file doesn't exist
|
|
# 15: Failed to add user to local wireguard
|
|
# 16: Failed to add user to federated server
|
|
# 17: User not found
|
|
|
|
CONFIG_FILE='/etc/wgapi/config'
|
|
if ! [ ${#} -eq 1 ]; then
|
|
printf 'ERROR! Bad input: %s %s\n' "${0}" "${*}" >>"${LOGFILE}"
|
|
exit 3
|
|
fi; if ! [ -x '/usr/bin/wg' ]; then
|
|
printf 'ERROR! %s could not find /usr/bin/wg\n' "${0}" >>"${LOGFILE}"
|
|
exit 5
|
|
fi; if ! [ -f "${CONFIG_FILE}" ]; then
|
|
printf 'ERROR! %s could not find %s!\n' "${0}" "${CONFIG_FILE}" >>"${LOGFILE}"
|
|
exit 4
|
|
fi
|
|
source "${CONFIG_FILE}"
|
|
if ! [ -f "${TOKENS_FILE}" ]; then
|
|
printf 'ERROR! %s could not find %s!\n' "${0}" "${TOKENS_FILE}" >>"${LOGFILE}"
|
|
exit 9
|
|
fi
|
|
ip="${1}"
|
|
printf '%s requested peer listing...\n' "${ip}" >>"${LOGFILE}"
|
|
|
|
# Create token if needed
|
|
token="$(grep "${ip}" "${TOKENS_FILE}" | cut -f2)"
|
|
if [ "${token}" == "" ]; then
|
|
printf 'Creating token for %s...\n' "${ip}" >>"${LOGFILE}"
|
|
token="$(</dev/urandom tr -dc '[:alnum:]' | fold -w 64 | head -n 1)"
|
|
printf '%s\t%s\n' "${ip}" "${token}" >>"${TOKENS_FILE}"
|
|
fi
|
|
|
|
# Get peer list from wireguard
|
|
if peers="[$("${LIB_DIR}/wg_peer_list" "${ip}" json)]"; then
|
|
printf '{"token":"%s","peers":%s}' "${token}" "${peers}" | "${LIB_DIR}/http_res" 200 'application/json'
|
|
printf 'Sent peers to user %s\n' "${ip}" >>"${LOGFILE}"
|
|
else
|
|
printf 'ERROR: Failed to lookup user: %s\n' "${ip}" >>"${LOGFILE}"
|
|
printf 'Failed to lookup user' | "${LIB_DIR}/http_res" 500
|
|
fi
|
|
|