xref: /netbsd-src/external/bsd/openpam/dist/lib/libpam/pam_putenv.c (revision 7d62b00eb9ad855ffcd7da46b41e23feb5476fac)
1 /*	$NetBSD: pam_putenv.c,v 1.4 2021/08/11 09:11:04 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.4 2021/08/11 09:11:04 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;
67 	const char *p;
68 	size_t env_size;
69 	int i;
70 
71 	ENTER();
72 
73 	/* sanity checks */
74 	if ((p = strchr(namevalue, '=')) == NULL) {
75 		errno = EINVAL;
76 		RETURNC(PAM_SYSTEM_ERR);
77 	}
78 
79 	/* see if the variable is already in the environment */
80 	if ((i = openpam_findenv(pamh, namevalue,
81 	    (size_t)(p - namevalue))) >= 0) {
82 		char *q;
83 		if ((q = strdup(namevalue)) == NULL)
84 			RETURNC(PAM_BUF_ERR);
85 		FREE(pamh->env[i]);
86 		pamh->env[i] = q;
87 		RETURNC(PAM_SUCCESS);
88 	}
89 
90 	/* grow the environment list if necessary */
91 	if (pamh->env_count == pamh->env_size) {
92 		env_size = pamh->env_size * 2 + 1;
93 		env = realloc(pamh->env, sizeof(*env) * env_size);
94 		if (env == NULL)
95 			RETURNC(PAM_BUF_ERR);
96 		pamh->env = env;
97 		pamh->env_size = env_size;
98 	}
99 
100 	/* add the variable at the end */
101 	if ((pamh->env[pamh->env_count] = strdup(namevalue)) == NULL)
102 		RETURNC(PAM_BUF_ERR);
103 	++pamh->env_count;
104 	RETURNC(PAM_SUCCESS);
105 }
106 
107 /*
108  * Error codes:
109  *
110  *	PAM_SYSTEM_ERR
111  *	PAM_BUF_ERR
112  */
113 
114 /**
115  * The =pam_putenv function sets an environment variable.
116  * Its semantics are similar to those of =putenv, but it modifies the PAM
117  * context's environment list instead of the application's.
118  *
119  * >pam_getenv
120  * >pam_getenvlist
121  * >pam_setenv
122  */
123