1 /*	$NetBSD: pam_setenv.c,v 1.4 2023/06/30 21:46:21 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 
38 #ifdef HAVE_CONFIG_H
39 # include "config.h"
40 #endif
41 
42 #include <sys/cdefs.h>
43 __RCSID("$NetBSD: pam_setenv.c,v 1.4 2023/06/30 21:46:21 christos Exp $");
44 
45 #include <errno.h>
46 #include <stdlib.h>
47 #include <stdio.h>
48 #include <string.h>
49 
50 #include <security/pam_appl.h>
51 
52 #include "openpam_impl.h"
53 #include "openpam_asprintf.h"
54 
55 /*
56  * OpenPAM extension
57  *
58  * Set the value of an environment variable
59  * Mirrors setenv(3)
60  */
61 
62 int
pam_setenv(pam_handle_t * pamh,const char * name,const char * value,int overwrite)63 pam_setenv(pam_handle_t *pamh,
64 	const char *name,
65 	const char *value,
66 	int overwrite)
67 {
68 	char *env;
69 	int r;
70 
71 	ENTER();
72 
73 	/* sanity checks */
74 	if (*name == '\0' || strchr(name, '=') != NULL) {
75 		errno = EINVAL;
76 		RETURNC(PAM_SYSTEM_ERR);
77 	}
78 
79 	/* is it already there? */
80 	if (!overwrite && openpam_findenv(pamh, name, strlen(name)) >= 0)
81 		RETURNC(PAM_SUCCESS);
82 
83 	/* set it... */
84 	if (asprintf(&env, "%s=%s", name, value) < 0)
85 		RETURNC(PAM_BUF_ERR);
86 	r = pam_putenv(pamh, env);
87 	FREE(env);
88 	RETURNC(r);
89 }
90 
91 /*
92  * Error codes:
93  *
94  *	=pam_putenv
95  *	PAM_SYSTEM_ERR
96  *	PAM_BUF_ERR
97  */
98 
99 /**
100  * The =pam_setenv function sets an environment variable.
101  * Its semantics are similar to those of =setenv, but it modifies the PAM
102  * context's environment list instead of the application's.
103  *
104  * >pam_getenv
105  * >pam_getenvlist
106  * >pam_putenv
107  */
108