Properly scope getopt for subcommands.
[monkeysphere.git] / src / monkeysphere
1 #!/bin/bash
2
3 # monkeysphere: MonkeySphere client tool
4 #
5 # The monkeysphere scripts are written by:
6 # Jameson Rollins <jrollins@fifthhorseman.net>
7 #
8 # They are Copyright 2008, and are all released under the GPL, version 3
9 # or later.
10
11 ########################################################################
12 PGRM=$(basename $0)
13
14 SHARE=${MONKEYSPHERE_SHARE:-"/usr/share/monkeysphere"}
15 export SHARE
16 . "${SHARE}/common" || exit 1
17
18 # date in UTF format if needed
19 DATE=$(date -u '+%FT%T')
20
21 # unset some environment variables that could screw things up
22 unset GREP_OPTIONS
23
24 # default return code
25 RETURN=0
26
27 # set the file creation mask to be only owner rw
28 umask 077
29
30 ########################################################################
31 # FUNCTIONS
32 ########################################################################
33
34 usage() {
35 cat <<EOF
36 usage: $PGRM <subcommand> [options] [args]
37 MonkeySphere client tool.
38
39 subcommands:
40   update-known_hosts (k) [HOST]...  update known_hosts file
41   update-authorized_keys (a)        update authorized_keys file
42   gen-subkey (g) KEYID              generate an 'a' capable subkey
43     -l|--length BITS                  key length in bits (2048)
44     -e|--expire EXPIRE                date to expire
45   help (h,?)                        this help
46
47 EOF
48 }
49
50 # generate a subkey with the 'a' usage flags set
51 # FIXME: this needs some tweaking to clean it up
52 gen_subkey(){
53     local keyLength
54     local keyExpire
55     local keyID
56     local gpgOut
57     local userID
58
59     # set default key parameter values
60     keyLength=
61     keyExpire=
62
63     # get options
64     TEMP=$(getopt -o l:e: -l length:,expire: -n "$PGRM" -- "$@")
65
66     if [ $? != 0 ] ; then
67         exit 1
68     fi
69
70     # Note the quotes around `$TEMP': they are essential!
71     eval set -- "$TEMP"
72
73     while true ; do
74         case "$1" in
75             -l|--length)
76                 keyLength="$2"
77                 shift 2
78                 ;;
79             -e|--expire)
80                 keyExpire="$2"
81                 shift 2
82                 ;;
83             --)
84                 shift
85                 ;;
86             *)
87                 break
88                 ;;
89         esac
90     done
91
92     keyID="$1"
93
94     gpgOut=$(gpg --quiet --fixed-list-mode --list-keys --with-colons \
95         "$keyID" 2> /dev/null)
96
97     # fail if there only "tru" lines are output from gpg, which
98     # indicates the key was not found.
99     if [ -z "$(echo "$gpgOut" | grep -v '^tru:')" ] ; then
100         failure "Key ID '$keyID' not found."
101     fi
102
103     # fail if multiple pub lines are returned, which means the id given
104     # is not unique
105     if [ $(echo "$gpgOut" | grep '^pub:' | wc -l) -gt '1' ] ; then
106         failure "Key ID '$keyID' is not unique."
107     fi
108
109     # prompt if an authentication subkey already exists
110     if echo "$gpgOut" | egrep "^(pub|sub):" | cut -d: -f 12 | grep -q a ; then
111         echo "An authentication subkey already exists for key '$keyID'."
112         read -p "Are you sure you would like to generate another one? (y/N) " OK; OK=${OK:N}
113         if [ "${OK/y/Y}" != 'Y' ] ; then
114             failure "aborting."
115         fi
116     fi
117
118     # set subkey defaults
119     # prompt about key expiration if not specified
120     if [ -z "$keyExpire" ] ; then
121         cat <<EOF
122 Please specify how long the key should be valid.
123          0 = key does not expire
124       <n>  = key expires in n days
125       <n>w = key expires in n weeks
126       <n>m = key expires in n months
127       <n>y = key expires in n years
128 EOF
129         while [ -z "$keyExpire" ] ; do
130             read -p "Key is valid for? (0) " keyExpire
131             if ! test_gpg_expire ${keyExpire:=0} ; then
132                 echo "invalid value"
133                 unset keyExpire
134             fi
135         done
136     elif ! test_gpg_expire "$keyExpire" ; then
137         failure "invalid key expiration value '$keyExpire'."
138     fi
139
140     # generate the list of commands that will be passed to edit-key
141     editCommands=$(cat <<EOF
142 addkey
143 7
144 S
145 E
146 A
147 Q
148 $keyLength
149 $keyExpire
150 save
151 EOF
152 )
153
154     log "generating subkey..."
155     echo "$editCommands" | gpg --expert --command-fd 0 --edit-key "$keyID"
156     log "done."
157 }
158
159 ########################################################################
160 # MAIN
161 ########################################################################
162
163 # unset variables that should be defined only in config file
164 unset KEYSERVER
165 unset CHECK_KEYSERVER
166 unset KNOWN_HOSTS
167 unset HASH_KNOWN_HOSTS
168 unset AUTHORIZED_KEYS
169
170 # load global config
171 [ -r "${ETC}/monkeysphere.conf" ] && . "${ETC}/monkeysphere.conf"
172
173 # set monkeysphere home directory
174 MONKEYSPHERE_HOME=${MONKEYSPHERE_HOME:="${HOME}/.config/monkeysphere"}
175 mkdir -p -m 0700 "$MONKEYSPHERE_HOME"
176
177 # load local config
178 [ -e ${MONKEYSPHERE_CONFIG:="${MONKEYSPHERE_HOME}/monkeysphere.conf"} ] && . "$MONKEYSPHERE_CONFIG"
179
180 # set empty config variables with ones from the environment, or from
181 # config file, or with defaults
182 GNUPGHOME=${MONKEYSPHERE_GNUPGHOME:=${GNUPGHOME:="${HOME}/.gnupg"}}
183 KEYSERVER=${MONKEYSPHERE_KEYSERVER:=${KEYSERVER:="subkeys.pgp.net"}}
184 CHECK_KEYSERVER=${MONKEYSPHERE_CHECK_KEYSERVER:=${CHECK_KEYSERVER:="true"}}
185 KNOWN_HOSTS=${MONKEYSPHERE_KNOWN_HOSTS:=${KNOWN_HOSTS:="${HOME}/.ssh/known_hosts"}}
186 HASH_KNOWN_HOSTS=${MONKEYSPHERE_HASH_KNOWN_HOSTS:=${HASH_KNOWN_HOSTS:="true"}}
187 AUTHORIZED_KEYS=${MONKEYSPHERE_AUTHORIZED_KEYS:=${AUTHORIZED_KEYS:="${HOME}/.ssh/authorized_keys"}}
188
189 # other variables not in config file
190 AUTHORIZED_USER_IDS=${MONKEYSPHERE_AUTHORIZED_USER_IDS:="${MONKEYSPHERE_HOME}/authorized_user_ids"}
191 REQUIRED_HOST_KEY_CAPABILITY=${MONKEYSPHERE_REQUIRED_HOST_KEY_CAPABILITY:="a"}
192 REQUIRED_USER_KEY_CAPABILITY=${MONKEYSPHERE_REQUIRED_USER_KEY_CAPABILITY:="a"}
193
194 # export GNUPGHOME and make sure gpg home exists with proper
195 # permissions
196 export GNUPGHOME
197 mkdir -p -m 0700 "$GNUPGHOME"
198
199 # get subcommand
200 COMMAND="$1"
201 [ "$COMMAND" ] || failure "Type '$PGRM help' for usage."
202 shift
203
204 case $COMMAND in
205     'update-known_hosts'|'update-known-hosts'|'k')
206         MODE='known_hosts'
207
208         if ! check_key_file_permissions "$USER" "$KNOWN_HOSTS" ; then
209             failure "Improper permissions on known_hosts file."
210         fi
211
212         # if hosts are specified on the command line, process just
213         # those hosts
214         if [ "$1" ] ; then
215             update_known_hosts "$@"
216             RETURN="$?"
217
218         # otherwise, if no hosts are specified, process every host
219         # in the user's known_hosts file
220         else
221             if [ ! -s "$KNOWN_HOSTS" ] ; then
222                 failure "known_hosts file '$KNOWN_HOSTS' is empty or does not exist."
223             fi
224
225             process_known_hosts
226             RETURN="$?"
227         fi
228         ;;
229
230     'update-authorized_keys'|'update-authorized-keys'|'a')
231         MODE='authorized_keys'
232
233         # fail if the authorized_user_ids file is empty
234         if [ ! -s "$AUTHORIZED_USER_IDS" ] ; then
235             failure "authorized_user_ids file '$AUTHORIZED_USER_IDS' is empty or does not exist."
236         fi
237
238         if ! check_key_file_permissions "$USER" "$AUTHORIZED_USER_IDS" ; then
239             failure "Improper permissions on authorized_user_ids file."
240         fi
241
242         # process authorized_user_ids file
243         process_authorized_user_ids "$AUTHORIZED_USER_IDS"
244         RETURN="$?"
245         ;;
246
247     'gen-subkey'|'g')
248         keyID="$1"
249         if [ -z "$keyID" ] ; then
250             failure "You must specify the key ID of your primary key."
251         fi
252         gen_subkey "$keyID"
253         ;;
254
255     'help'|'h'|'?')
256         usage
257         ;;
258
259     *)
260         failure "Unknown command: '$COMMAND'
261 Type '$PGRM help' for usage."
262         ;;
263 esac
264
265 exit "$RETURN"