iankelling.org Git - distro-setup/blob - rootsshsync

   1 #!/bin/bash
   2 # Copyright (C) 2016 Ian Kelling
   3
   4 # Licensed under the Apache License, Version 2.0 (the "License");
   5 # you may not use this file except in compliance with the License.
   6 # You may obtain a copy of the License at
   7
   8 #     http://www.apache.org/licenses/LICENSE-2.0
   9
  10 # Unless required by applicable law or agreed to in writing, software
  11 # distributed under the License is distributed on an "AS IS" BASIS,
  12 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13 # See the License for the specific language governing permissions and
  14 # limitations under the License.
  15
  16 set -eE -o pipefail
  17 trap 'echo "$0:$LINENO:error: \"$BASH_COMMAND\" returned $?" >&2' ERR
  18
  19 [[ $EUID == 0 ]] || exec sudo "$BASH_SOURCE" "$@"
  20
  21 dest=/root/.ssh
  22
  23 # leftover
  24 if [[ -L $dest ]]; then
  25   rm $dest
  26 fi
  27 mkdir -p $dest
  28 chmod 700 $dest
  29
  30 user_ssh_dir=$(eval echo ~${SUDO_USER:-$USER})/.ssh
  31
  32 # remove broken links, or else rsync has error about them.
  33 find $user_ssh_dir -xtype l -exec rm '{}' \;
  34 # -t times, so it won't rewrite the file every time,
  35 # -L resolve links
  36 rsync -rtL --delete $user_ssh_dir/ $dest
  37 chown -R root:root /root/.ssh
  38 d=/etc/initramfs-tools
  39 if [[ -e $d ]] && ! diff -q /root/.ssh/authorized_keys $d/root/.ssh/authorized_keys &>/dev/null; then
  40   mkdir -p /etc/initramfs-tools/root/.ssh
  41   chmod 700 $d/root $d/root/.ssh
  42   cp -p /root/.ssh/authorized_keys $d/root/.ssh/authorized_keys
  43   update-initramfs -u -k all
  44 fi