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