1 /* $NetBSD: pam_putenv.c,v 1.3 2017/05/06 19:50:10 christos Exp $ */ 2 3 /*- 4 * Copyright (c) 2002-2003 Networks Associates Technology, Inc. 5 * Copyright (c) 2004-2017 Dag-Erling Smørgrav 6 * All rights reserved. 7 * 8 * This software was developed for the FreeBSD Project by ThinkSec AS and 9 * Network Associates Laboratories, the Security Research Division of 10 * Network Associates, Inc. under DARPA/SPAWAR contract N66001-01-C-8035 11 * ("CBOSS"), as part of the DARPA CHATS research program. 12 * 13 * Redistribution and use in source and binary forms, with or without 14 * modification, are permitted provided that the following conditions 15 * are met: 16 * 1. Redistributions of source code must retain the above copyright 17 * notice, this list of conditions and the following disclaimer. 18 * 2. Redistributions in binary form must reproduce the above copyright 19 * notice, this list of conditions and the following disclaimer in the 20 * documentation and/or other materials provided with the distribution. 21 * 3. The name of the author may not be used to endorse or promote 22 * products derived from this software without specific prior written 23 * permission. 24 * 25 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND 26 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 27 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 28 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 29 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 30 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 31 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 32 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 33 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 34 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 35 * SUCH DAMAGE. 36 * 37 * $OpenPAM: pam_putenv.c 938 2017-04-30 21:34:42Z des $ 38 */ 39 40 #ifdef HAVE_CONFIG_H 41 # include "config.h" 42 #endif 43 44 #include <sys/cdefs.h> 45 __RCSID("$NetBSD: pam_putenv.c,v 1.3 2017/05/06 19:50:10 christos Exp $"); 46 47 #include <errno.h> 48 #include <stdlib.h> 49 #include <string.h> 50 51 #include <security/pam_appl.h> 52 53 #include "openpam_impl.h" 54 55 /* 56 * XSSO 4.2.1 57 * XSSO 6 page 56 58 * 59 * Set the value of an environment variable 60 */ 61 62 int 63 pam_putenv(pam_handle_t *pamh, 64 const char *namevalue) 65 { 66 char **env, *p; 67 size_t env_size; 68 int i; 69 70 ENTER(); 71 72 /* sanity checks */ 73 if ((p = strchr(namevalue, '=')) == NULL) { 74 errno = EINVAL; 75 RETURNC(PAM_SYSTEM_ERR); 76 } 77 78 /* see if the variable is already in the environment */ 79 if ((i = openpam_findenv(pamh, namevalue, 80 (size_t)(p - namevalue))) >= 0) { 81 if ((p = strdup(namevalue)) == NULL) 82 RETURNC(PAM_BUF_ERR); 83 FREE(pamh->env[i]); 84 pamh->env[i] = p; 85 RETURNC(PAM_SUCCESS); 86 } 87 88 /* grow the environment list if necessary */ 89 if (pamh->env_count == pamh->env_size) { 90 env_size = pamh->env_size * 2 + 1; 91 env = realloc(pamh->env, sizeof(*env) * env_size); 92 if (env == NULL) 93 RETURNC(PAM_BUF_ERR); 94 pamh->env = env; 95 pamh->env_size = env_size; 96 } 97 98 /* add the variable at the end */ 99 if ((pamh->env[pamh->env_count] = strdup(namevalue)) == NULL) 100 RETURNC(PAM_BUF_ERR); 101 ++pamh->env_count; 102 RETURNC(PAM_SUCCESS); 103 } 104 105 /* 106 * Error codes: 107 * 108 * PAM_SYSTEM_ERR 109 * PAM_BUF_ERR 110 */ 111 112 /** 113 * The =pam_putenv function sets an environment variable. 114 * Its semantics are similar to those of =putenv, but it modifies the PAM 115 * context's environment list instead of the application's. 116 * 117 * >pam_getenv 118 * >pam_getenvlist 119 * >pam_setenv 120 */ 121