usr.bin/ssh-copy-id/ssh-copy-id.sh

   1 #!/bin/sh
   2 #-
   3 # Copyright (c) 2012 Eitan Adler
   4 # All rights reserved.
   5 #
   6 # Redistribution and use in source and binary forms, with or without
   7 # modification, are permitted provided that the following conditions
   8 # are met:
   9 # 1. Redistributions of source code must retain the above copyright
  10 #    notice, this list of conditions and the following disclaimer
  11 #    in this position and unchanged.
  12 # 2. Redistributions in binary form must reproduce the above copyright
  13 #    notice, this list of conditions and the following disclaimer in the
  14 #    documentation and/or other materials provided with the distribution.
  15 #
  16 # THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
  17 # ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  18 # IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  19 # ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
  20 # FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  21 # DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  22 # OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  23 # HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  24 # LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  25 # OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  26 # SUCH DAMAGE.
  27 #
  28 # $FreeBSD$
  29
  30 usage() {
  31         echo "usage: ssh-copy-id [-l] [-i keyfile] [-o option] [-p port] [user@]hostname" >&2
  32         exit 1
  33 }
  34
  35 sendkey() {
  36         local h="$1"
  37         local k="$2"
  38         printf "%s\n" "$k" | ssh $port -S none $options "$user$h" /bin/sh -c \'' \
  39                 set -e; \
  40                 umask 077; \
  41                 keyfile=$HOME/.ssh/authorized_keys ; \
  42                 mkdir -p -- "$HOME/.ssh/" ; \
  43                 while read alg key comment ; do \
  44                         [ -n "$key" ] || continue; \
  45                         if ! grep -sqwF "$key" "$keyfile"; then \
  46                                 printf "$alg $key $comment\n" >> "$keyfile" ; \
  47                         fi ; \
  48                 done \
  49         '\'
  50 }
  51
  52 agentKeys() {
  53         keys="$(ssh-add -L | grep -v 'The agent has no identities.')$nl$keys"
  54 }
  55
  56 keys=""
  57 host=""
  58 hasarg=""
  59 user=""
  60 port=""
  61 nl="
  62 "
  63 options=""
  64
  65 IFS=$nl
  66
  67 while getopts 'i:lo:p:' arg; do
  68         case $arg in
  69         i)
  70                 hasarg="x"
  71                 if [ -r "$OPTARG" ]; then
  72                         keys="$(cat -- "$OPTARG")$nl$keys"
  73                 else
  74                         echo "File $OPTARG not found" >&2
  75                         exit 1
  76                 fi
  77                 ;;
  78         l)
  79                 hasarg="x"
  80                 agentKeys
  81                 ;;
  82         p)
  83                 port=-p$nl$OPTARG
  84                 ;;
  85         o)
  86                 options=$options$nl-o$nl$OPTARG
  87                 ;;
  88         *)
  89                 usage
  90                 ;;
  91         esac
  92 done >&2
  93
  94 shift $((OPTIND-1))
  95
  96 if [ -z "$hasarg" ]; then
  97         agentKeys
  98 fi
  99 if [ -z "$keys" ] || [ "$keys" = "$nl" ]; then
 100         echo "no keys found" >&2
 101         exit 1
 102 fi
 103 if [ "$#" -eq 0 ]; then
 104         usage
 105 fi
 106
 107 for host in "$@"; do
 108         sendkey "$host" "$keys"
 109 done