xref: /freebsd/contrib/openpam/lib/libpam/pam_setenv.c (revision f4b37ed0f8b307b1f3f0f630ca725d68f1dff30d)
1 /*-
2  * Copyright (c) 2002-2003 Networks Associates Technology, Inc.
3  * Copyright (c) 2004-2011 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  * $Id: pam_setenv.c 648 2013-03-05 17:54:27Z des $
36  */
37 
38 #ifdef HAVE_CONFIG_H
39 # include "config.h"
40 #endif
41 
42 #include <stdlib.h>
43 #include <stdio.h>
44 #include <string.h>
45 
46 #include <security/pam_appl.h>
47 
48 #include "openpam_impl.h"
49 #include "openpam_asprintf.h"
50 
51 /*
52  * OpenPAM extension
53  *
54  * Set the value of an environment variable
55  * Mirrors setenv(3)
56  */
57 
58 int
59 pam_setenv(pam_handle_t *pamh,
60 	const char *name,
61 	const char *value,
62 	int overwrite)
63 {
64 	char *env;
65 	int r;
66 
67 	ENTER();
68 	if (pamh == NULL)
69 		RETURNC(PAM_SYSTEM_ERR);
70 
71 	/* sanity checks */
72 	if (name == NULL || value == NULL || strchr(name, '=') != NULL)
73 		RETURNC(PAM_SYSTEM_ERR);
74 
75 	/* is it already there? */
76 	if (!overwrite && openpam_findenv(pamh, name, strlen(name)) >= 0)
77 		RETURNC(PAM_SUCCESS);
78 
79 	/* set it... */
80 	if (asprintf(&env, "%s=%s", name, value) < 0)
81 		RETURNC(PAM_BUF_ERR);
82 	r = pam_putenv(pamh, env);
83 	FREE(env);
84 	RETURNC(r);
85 }
86 
87 /*
88  * Error codes:
89  *
90  *	=pam_putenv
91  *	PAM_SYSTEM_ERR
92  *	PAM_BUF_ERR
93  */
94 
95 /**
96  * The =pam_setenv function sets an environment variable.
97  * Its semantics are similar to those of =setenv, but it modifies the PAM
98  * context's environment list instead of the application's.
99  *
100  * >pam_getenv
101  * >pam_getenvlist
102  * >pam_putenv
103  */
104