mirror of
https://github.com/fire1ce/UDM-Persistent-SSH-Keys.git
synced 2023-10-10 11:36:54 +00:00
7d6d086ff3
Ensure the target path exists before writing the keys so there's no errors for the user. Dropped a temp variable as well.
37 lines
870 B
Bash
Executable File
37 lines
870 B
Bash
Executable File
#!/bin/sh
|
|
|
|
# Get DataDir location
|
|
DATA_DIR="/mnt/data"
|
|
case "$(ubnt-device-info firmware || true)" in
|
|
1*)
|
|
DATA_DIR="/mnt/data"
|
|
;;
|
|
2*)
|
|
DATA_DIR="/data"
|
|
;;
|
|
3*)
|
|
DATA_DIR="/data"
|
|
;;
|
|
*)
|
|
echo "ERROR: No persistent storage found." 1>&2
|
|
exit 1
|
|
;;
|
|
esac
|
|
|
|
user_authorized_keys_file="$DATA_DIR/ssh/authorized_keys"
|
|
udm_authorized_keys_file="/root/.ssh/authorized_keys"
|
|
|
|
# if $DATA_DIR/ssh/authorized_keys is missing print error message
|
|
if [ ! -f "$user_authorized_keys_file" ]; then
|
|
echo "ERROR: $user_authorized_keys_file is missing"
|
|
exit 1
|
|
fi
|
|
|
|
# Reads the user file line by line and adds the key to the authorized_keys file
|
|
mkdir -p "$(dirname ${udm_authorized_keys_file})"
|
|
while IFS= read -r line || [ -n "$line" ]; do
|
|
echo $line >>$udm_authorized_keys_file
|
|
done <"$user_authorized_keys_file"
|
|
|
|
exit 0
|