X-Git-Url: https://iankelling.org/git/?a=blobdiff_plain;f=rootsshsync;h=c4cebbf5660b56b7c305d184a46dbc826ffb1377;hb=12e7a0f3de4f4f1b1f8e56e7cbe2e972027e0a42;hp=c55014921b8af67fd2c8d4f3554feb64bfadb218;hpb=343a048be4b32d18540c6531048d2b9a006774ae;p=distro-setup

diff --git a/rootsshsync b/rootsshsync
index c550149..c4cebbf 100755
--- a/rootsshsync
+++ b/rootsshsync
@@ -27,7 +27,12 @@ fi
 mkdir -p $dest
 chmod 700 $dest
 
-user_ssh_dir=$(eval echo ~${SUDO_USER:-$USER})/.ssh
+user=$(id -un)
+if [[ $SUDO_USER ]]; then
+  user=$SUDO_USER
+fi
+
+user_ssh_dir=$(eval echo ~$user)/.ssh
 
 # remove broken links, or else rsync has error about them.
 find $user_ssh_dir -xtype l -exec rm '{}' \;
@@ -35,10 +40,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 /etc/initramfs-tools/root/.ssh
+  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