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