X-Git-Url: https://iankelling.org/git/?p=distro-setup;a=blobdiff_plain;f=rootsshsync;h=668f8a3d7ed931ea2a65130c050b9c7e9326ca66;hp=97d2d2f779919413058d70b1ec8282a2a468682e;hb=79b274fcd8bfa556133ab13270e84b40aebe8468;hpb=eb9b839bb5a91c60cc4f6eb9d7e38ffbf73f0e90

diff --git a/rootsshsync b/rootsshsync
index 97d2d2f..668f8a3 100755
--- a/rootsshsync
+++ b/rootsshsync
@@ -16,18 +16,27 @@
 set -eE -o pipefail
 trap 'echo "$0:$LINENO:error: \"$BASH_COMMAND\" returned $?" >&2' ERR
 
-[[ $EUID == 0 ]] || exec sudo "$BASH_SOURCE" "$@"
-
-if test -e /q/root/.ssh; then
-    dest=/q/root/.ssh
-    /a/exe/lnf $dest /root
-else
-    dest=/root/.ssh
-    mkdir -p $dest
-    chmod 700 $dest
+[[ $EUID == 0 ]] || exec sudo -E "${BASH_SOURCE[0]}" "$@"
+
+dest=/root/.ssh
+
+# leftover
+if [[ -L $dest ]]; then
+  rm $dest
+fi
+mkdir -p $dest
+chmod 700 $dest
+
+user=$(id -un)
+if [[ $SUDO_USER ]]; then
+  user=$SUDO_USER
 fi
 
-user_ssh_dir=$(eval echo ~${SUDO_USER:-$USER})/.ssh
+user_ssh_dir=$(eval echo ~$user)/.ssh
+if [[ ! -s $user_ssh_dir/authorized_keys ]]; then
+  echo missing $user_ssh_dir/authorized_keys. bad sign. bailing >&2
+  exit 1
+fi
 
 # remove broken links, or else rsync has error about them.
 find $user_ssh_dir -xtype l -exec rm '{}' \;
@@ -35,3 +44,13 @@ find $user_ssh_dir -xtype l -exec rm '{}' \;
 # -L resolve links
 rsync -rtL --delete $user_ssh_dir/ $dest
 chown -R root:root /root/.ssh
+
+
+d=/etc/initramfs-tools
+if [[ -e $d ]] && ! diff -q /root/.ssh/authorized_keys $d/root/.ssh/authorized_keys &>/dev/null; then
+  mkdir -p $d/root/.ssh /etc/dropbear-initramfs
+  chmod 700 $d/root $d/root/.ssh
+  cp -p /root/.ssh/authorized_keys $d/root/.ssh/authorized_keys
+  cp -p /root/.ssh/authorized_keys /etc/dropbear-initramfs
+  update-initramfs -u -k all
+fi