contrib/openpam/lib/pam_setenv.c

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