1 /* $NetBSD: pam_ssh.c,v 1.4 2005/02/27 01:16:27 christos Exp $ */ 2 3 /*- 4 * Copyright (c) 2003 Networks Associates Technology, Inc. 5 * All rights reserved. 6 * 7 * This software was developed for the FreeBSD Project by ThinkSec AS and 8 * NAI Labs, the Security Research Division of Network Associates, Inc. 9 * under DARPA/SPAWAR contract N66001-01-C-8035 ("CBOSS"), as part of the 10 * DARPA CHATS research program. 11 * 12 * Redistribution and use in source and binary forms, with or without 13 * modification, are permitted provided that the following conditions 14 * are met: 15 * 1. Redistributions of source code must retain the above copyright 16 * notice, this list of conditions and the following disclaimer. 17 * 2. Redistributions in binary form must reproduce the above copyright 18 * notice, this list of conditions and the following disclaimer in the 19 * documentation and/or other materials provided with the distribution. 20 * 3. The name of the author may not be used to endorse or promote 21 * products derived from this software without specific prior written 22 * permission. 23 * 24 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND 25 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 26 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 27 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 28 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 29 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 30 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 31 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 32 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 33 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 34 * SUCH DAMAGE. 35 */ 36 37 #include <sys/cdefs.h> 38 #ifdef __FreeBSD__ 39 __FBSDID("$FreeBSD: src/lib/libpam/modules/pam_ssh/pam_ssh.c,v 1.40 2004/02/10 10:13:21 des Exp $"); 40 #else 41 __RCSID("$NetBSD: pam_ssh.c,v 1.4 2005/02/27 01:16:27 christos Exp $"); 42 #endif 43 44 #include <sys/param.h> 45 #include <sys/wait.h> 46 47 #include <errno.h> 48 #include <fcntl.h> 49 #include <paths.h> 50 #include <pwd.h> 51 #include <signal.h> 52 #include <stdio.h> 53 #include <string.h> 54 #include <unistd.h> 55 56 #define PAM_SM_AUTH 57 #define PAM_SM_SESSION 58 59 #include <security/pam_appl.h> 60 #include <security/pam_modules.h> 61 #include <security/openpam.h> 62 63 #include <openssl/evp.h> 64 65 #include "key.h" 66 #include "authfd.h" 67 #include "authfile.h" 68 69 extern char **environ; 70 71 struct pam_ssh_key { 72 Key *key; 73 char *comment; 74 }; 75 76 static const char *pam_ssh_prompt = "SSH passphrase: "; 77 static const char *pam_ssh_have_keys = "pam_ssh_have_keys"; 78 79 static const char *pam_ssh_keyfiles[] = { 80 ".ssh/identity", /* SSH1 RSA key */ 81 ".ssh/id_rsa", /* SSH2 RSA key */ 82 ".ssh/id_dsa", /* SSH2 DSA key */ 83 NULL 84 }; 85 86 static const char *pam_ssh_agent = "/usr/bin/ssh-agent"; 87 static const char *pam_ssh_agent_argv[] = { "ssh_agent", "-s", NULL }; 88 static const char *pam_ssh_agent_envp[] = { NULL }; 89 90 /* 91 * Attempts to load a private key from the specified file in the specified 92 * directory, using the specified passphrase. If successful, returns a 93 * struct pam_ssh_key containing the key and its comment. 94 */ 95 static struct pam_ssh_key * 96 pam_ssh_load_key(const char *dir, const char *kfn, const char *passphrase) 97 { 98 struct pam_ssh_key *psk; 99 char fn[PATH_MAX]; 100 char *comment; 101 Key *key; 102 103 if (snprintf(fn, sizeof(fn), "%s/%s", dir, kfn) > (int)sizeof(fn)) 104 return (NULL); 105 comment = NULL; 106 key = key_load_private(fn, passphrase, &comment); 107 if (key == NULL) { 108 openpam_log(PAM_LOG_DEBUG, "failed to load key from %s\n", fn); 109 return (NULL); 110 } 111 112 openpam_log(PAM_LOG_DEBUG, "loaded '%s' from %s\n", comment, fn); 113 if ((psk = malloc(sizeof(*psk))) == NULL) { 114 key_free(key); 115 free(comment); 116 return (NULL); 117 } 118 psk->key = key; 119 psk->comment = comment; 120 return (psk); 121 } 122 123 /* 124 * Wipes a private key and frees the associated resources. 125 */ 126 static void 127 pam_ssh_free_key(pam_handle_t *pamh __unused, 128 void *data, int pam_err __unused) 129 { 130 struct pam_ssh_key *psk; 131 132 psk = data; 133 key_free(psk->key); 134 free(psk->comment); 135 free(psk); 136 } 137 138 PAM_EXTERN int 139 pam_sm_authenticate(pam_handle_t *pamh, int flags __unused, 140 int argc __unused, const char *argv[] __unused) 141 { 142 const char **kfn, *passphrase, *user; 143 struct passwd *pwd; 144 struct pam_ssh_key *psk; 145 int nkeys, pam_err, pass; 146 147 /* PEM is not loaded by default */ 148 OpenSSL_add_all_algorithms(); 149 150 /* get user name and home directory */ 151 pam_err = pam_get_user(pamh, &user, NULL); 152 if (pam_err != PAM_SUCCESS) 153 return (pam_err); 154 pwd = getpwnam(user); 155 if (pwd == NULL) 156 return (PAM_USER_UNKNOWN); 157 if (pwd->pw_dir == NULL) 158 return (PAM_AUTH_ERR); 159 160 /* switch to user credentials */ 161 pam_err = openpam_borrow_cred(pamh, pwd); 162 if (pam_err != PAM_SUCCESS) 163 return (pam_err); 164 165 pass = (pam_get_item(pamh, PAM_AUTHTOK, 166 (const void **)__UNCONST(&passphrase)) == PAM_SUCCESS); 167 load_keys: 168 /* get passphrase */ 169 pam_err = pam_get_authtok(pamh, PAM_AUTHTOK, 170 &passphrase, pam_ssh_prompt); 171 if (pam_err != PAM_SUCCESS) { 172 openpam_restore_cred(pamh); 173 return (pam_err); 174 } 175 176 /* try to load keys from all keyfiles we know of */ 177 nkeys = 0; 178 for (kfn = pam_ssh_keyfiles; *kfn != NULL; ++kfn) { 179 psk = pam_ssh_load_key(pwd->pw_dir, *kfn, passphrase); 180 if (psk != NULL) { 181 pam_set_data(pamh, *kfn, psk, pam_ssh_free_key); 182 ++nkeys; 183 } 184 } 185 186 /* 187 * If we tried an old token and didn't get anything, and 188 * try_first_pass was specified, try again after prompting the 189 * user for a new passphrase. 190 */ 191 if (nkeys == 0 && pass == 1 && 192 openpam_get_option(pamh, "try_first_pass") != NULL) { 193 pam_set_item(pamh, PAM_AUTHTOK, NULL); 194 pass = 0; 195 goto load_keys; 196 } 197 198 /* switch back to arbitrator credentials before returning */ 199 openpam_restore_cred(pamh); 200 201 /* no keys? */ 202 if (nkeys == 0) 203 return (PAM_AUTH_ERR); 204 205 pam_set_data(pamh, pam_ssh_have_keys, NULL, NULL); 206 return (PAM_SUCCESS); 207 } 208 209 PAM_EXTERN int 210 pam_sm_setcred(pam_handle_t *pamh __unused, int flags __unused, 211 int argc __unused, const char *argv[] __unused) 212 { 213 214 return (PAM_SUCCESS); 215 } 216 217 /* 218 * Parses a line from ssh-agent's output. 219 */ 220 static void 221 pam_ssh_process_agent_output(pam_handle_t *pamh, FILE *f) 222 { 223 char *line, *p, *key, *val; 224 size_t len; 225 226 while ((line = fgetln(f, &len)) != NULL) { 227 if (len < 4 || strncmp(line, "SSH_", 4) != 0) 228 continue; 229 230 /* find equal sign at end of key */ 231 for (p = key = line; p < line + len; ++p) 232 if (*p == '=') 233 break; 234 if (p == line + len || *p != '=') 235 continue; 236 *p = '\0'; 237 238 /* find semicolon at end of value */ 239 for (val = ++p; p < line + len; ++p) 240 if (*p == ';') 241 break; 242 if (p == line + len || *p != ';') 243 continue; 244 *p = '\0'; 245 246 /* store key-value pair in environment */ 247 openpam_log(PAM_LOG_DEBUG, "got %s: %s", key, val); 248 pam_setenv(pamh, key, val, 1); 249 } 250 } 251 252 /* 253 * Starts an ssh agent and stores the environment variables derived from 254 * its output. 255 */ 256 static int 257 pam_ssh_start_agent(pam_handle_t *pamh, struct passwd *pwd) 258 { 259 int agent_pipe[2]; 260 pid_t pid; 261 FILE *f; 262 263 /* get a pipe which we will use to read the agent's output */ 264 if (pipe(agent_pipe) == -1) 265 return (PAM_SYSTEM_ERR); 266 267 /* start the agent */ 268 openpam_log(PAM_LOG_DEBUG, "starting an ssh agent"); 269 pid = fork(); 270 if (pid == (pid_t)-1) { 271 /* failed */ 272 close(agent_pipe[0]); 273 close(agent_pipe[1]); 274 return (PAM_SYSTEM_ERR); 275 } 276 if (pid == 0) { 277 #ifndef F_CLOSEM 278 int fd; 279 #endif 280 /* child: drop privs, close fds and start agent */ 281 if (setgid(pwd->pw_gid) == -1) { 282 openpam_log(PAM_LOG_DEBUG, "%s: Cannot setgid %d (%m)", 283 __FUNCTION__, (int)pwd->pw_gid); 284 goto done; 285 } 286 if (initgroups(pwd->pw_name, pwd->pw_gid) == -1) { 287 openpam_log(PAM_LOG_DEBUG, 288 "%s: Cannot initgroups for %s (%m)", 289 __FUNCTION__, pwd->pw_name); 290 goto done; 291 } 292 if (setuid(pwd->pw_uid) == -1) { 293 openpam_log(PAM_LOG_DEBUG, "%s: Cannot setuid %d (%m)", 294 __FUNCTION__, (int)pwd->pw_uid); 295 goto done; 296 } 297 (void)close(STDIN_FILENO); 298 (void)open(_PATH_DEVNULL, O_RDONLY); 299 (void)dup2(agent_pipe[1], STDOUT_FILENO); 300 (void)dup2(agent_pipe[1], STDERR_FILENO); 301 #ifdef F_CLOSEM 302 (void)fcntl(3, F_CLOSEM, 0); 303 #else 304 for (fd = 3; fd < getdtablesize(); ++fd) 305 (void)close(fd); 306 #endif 307 (void)execve(pam_ssh_agent, 308 (char **)__UNCONST(pam_ssh_agent_argv), 309 (char **)__UNCONST(pam_ssh_agent_envp)); 310 done: 311 _exit(127); 312 } 313 314 /* parent */ 315 close(agent_pipe[1]); 316 if ((f = fdopen(agent_pipe[0], "r")) == NULL) 317 return (PAM_SYSTEM_ERR); 318 pam_ssh_process_agent_output(pamh, f); 319 fclose(f); 320 321 return (PAM_SUCCESS); 322 } 323 324 /* 325 * Adds previously stored keys to a running agent. 326 */ 327 static int 328 pam_ssh_add_keys_to_agent(pam_handle_t *pamh) 329 { 330 AuthenticationConnection *ac; 331 struct pam_ssh_key *psk; 332 const char **kfn; 333 char **envlist, **env; 334 int pam_err; 335 336 /* switch to PAM environment */ 337 envlist = environ; 338 if ((environ = pam_getenvlist(pamh)) == NULL) { 339 openpam_log(PAM_LOG_DEBUG, "%s: cannot get envlist", 340 __FUNCTION__); 341 environ = envlist; 342 return (PAM_SYSTEM_ERR); 343 } 344 345 /* get a connection to the agent */ 346 if ((ac = ssh_get_authentication_connection()) == NULL) { 347 openpam_log(PAM_LOG_DEBUG, 348 "%s: cannot get authentication connection", 349 __FUNCTION__); 350 pam_err = PAM_SYSTEM_ERR; 351 goto end; 352 } 353 354 /* look for keys to add to it */ 355 for (kfn = pam_ssh_keyfiles; *kfn != NULL; ++kfn) { 356 pam_err = pam_get_data(pamh, *kfn, (void **)__UNCONST(&psk)); 357 if (pam_err == PAM_SUCCESS && psk != NULL) { 358 if (ssh_add_identity(ac, psk->key, psk->comment)) 359 openpam_log(PAM_LOG_DEBUG, 360 "added %s to ssh agent", psk->comment); 361 else 362 openpam_log(PAM_LOG_DEBUG, "failed " 363 "to add %s to ssh agent", psk->comment); 364 /* we won't need the key again, so wipe it */ 365 pam_set_data(pamh, *kfn, NULL, NULL); 366 } 367 } 368 pam_err = PAM_SUCCESS; 369 end: 370 /* disconnect from agent */ 371 if (ac != NULL) 372 ssh_close_authentication_connection(ac); 373 374 /* switch back to original environment */ 375 for (env = environ; *env != NULL; ++env) 376 free(*env); 377 free(environ); 378 environ = envlist; 379 380 return (pam_err); 381 } 382 383 PAM_EXTERN int 384 pam_sm_open_session(pam_handle_t *pamh, int flags __unused, 385 int argc __unused, const char *argv[] __unused) 386 { 387 struct passwd *pwd; 388 const char *user; 389 void *data; 390 int pam_err = PAM_SUCCESS; 391 392 /* no keys, no work */ 393 if (pam_get_data(pamh, pam_ssh_have_keys, &data) != PAM_SUCCESS && 394 openpam_get_option(pamh, "want_agent") == NULL) 395 return (PAM_SUCCESS); 396 397 /* switch to user credentials */ 398 pam_err = pam_get_user(pamh, &user, NULL); 399 if (pam_err != PAM_SUCCESS) 400 return (pam_err); 401 pwd = getpwnam(user); 402 if (pwd == NULL) 403 return (PAM_USER_UNKNOWN); 404 405 /* start the agent */ 406 pam_err = pam_ssh_start_agent(pamh, pwd); 407 if (pam_err != PAM_SUCCESS) 408 return pam_err; 409 410 pam_err = openpam_borrow_cred(pamh, pwd); 411 if (pam_err != PAM_SUCCESS) 412 return pam_err; 413 414 /* we have an agent, see if we can add any keys to it */ 415 pam_err = pam_ssh_add_keys_to_agent(pamh); 416 if (pam_err != PAM_SUCCESS) { 417 /* XXX ignore failures */ 418 openpam_log(PAM_LOG_DEBUG, "failed adding keys to ssh agent"); 419 pam_err = PAM_SUCCESS; 420 } 421 422 openpam_restore_cred(pamh); 423 return pam_err; 424 } 425 426 PAM_EXTERN int 427 pam_sm_close_session(pam_handle_t *pamh, int flags __unused, 428 int argc __unused, const char *argv[] __unused) 429 { 430 const char *ssh_agent_pid; 431 char *end; 432 int status; 433 pid_t pid; 434 435 if ((ssh_agent_pid = pam_getenv(pamh, "SSH_AGENT_PID")) == NULL) { 436 openpam_log(PAM_LOG_DEBUG, "no ssh agent"); 437 return (PAM_SUCCESS); 438 } 439 pid = (pid_t)strtol(ssh_agent_pid, &end, 10); 440 if (*ssh_agent_pid == '\0' || *end != '\0') { 441 openpam_log(PAM_LOG_DEBUG, "invalid ssh agent pid"); 442 return (PAM_SESSION_ERR); 443 } 444 openpam_log(PAM_LOG_DEBUG, "killing ssh agent %d", (int)pid); 445 if (kill(pid, SIGTERM) == -1 || 446 (waitpid(pid, &status, 0) == -1 && errno != ECHILD)) 447 return (PAM_SYSTEM_ERR); 448 return (PAM_SUCCESS); 449 } 450 451 PAM_MODULE_ENTRY("pam_ssh"); 452