Home | History | Annotate | Line # | Download | only in t
t_pam_env.c revision 1.1
      1 /*-
      2  * Copyright (c) 2019 Dag-Erling Smrgrav
      3  * All rights reserved.
      4  *
      5  * Redistribution and use in source and binary forms, with or without
      6  * modification, are permitted provided that the following conditions
      7  * are met:
      8  * 1. Redistributions of source code must retain the above copyright
      9  *    notice, this list of conditions and the following disclaimer.
     10  * 2. Redistributions in binary form must reproduce the above copyright
     11  *    notice, this list of conditions and the following disclaimer in the
     12  *    documentation and/or other materials provided with the distribution.
     13  * 3. The name of the author may not be used to endorse or promote
     14  *    products derived from this software without specific prior written
     15  *    permission.
     16  *
     17  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
     18  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
     19  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
     20  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
     21  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
     22  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
     23  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
     24  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
     25  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
     26  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
     27  * SUCH DAMAGE.
     28  *
     29  * $OpenPAM$
     30  */
     31 
     32 #ifdef HAVE_CONFIG_H
     33 # include "config.h"
     34 #endif
     35 
     36 #include <stdint.h>
     37 #include <stdio.h>
     38 #include <unistd.h>
     39 
     40 #include <cryb/test.h>
     41 
     42 #include <security/pam_appl.h>
     43 #include <security/openpam.h>
     44 
     45 #include "t_pam_err.h"
     46 
     47 #define T_ENV_NAME	"MAGIC_WORDS"
     48 #define T_ENV_VALUE	"SQUEAMISH OSSIFRAGE"
     49 #define T_ENV_NAMEVALUE	T_ENV_NAME "=" T_ENV_VALUE
     50 
     51 struct pam_conv t_null_pamc;
     52 
     53 
     54 /***************************************************************************
     56  * Tests
     57  */
     58 
     59 static int
     60 t_env_empty(char **desc CRYB_UNUSED, void *arg CRYB_UNUSED)
     61 {
     62 	pam_handle_t *pamh;
     63 	char **envlist;
     64 	int pam_err, ret;
     65 
     66 	ret = 1;
     67 	pam_err = pam_start("t_pam_env", "test", &t_null_pamc, &pamh);
     68 	t_assert(pam_err == PAM_SUCCESS);
     69 	envlist = pam_getenvlist(pamh);
     70 	ret &= t_is_not_null(envlist);
     71 	if (envlist != NULL) {
     72 		ret &= t_is_null(*envlist);
     73 		openpam_free_envlist(envlist);
     74 	}
     75 	pam_end(pamh, pam_err);
     76 	return (ret);
     77 }
     78 
     79 static int
     80 t_putenv_simple(char **desc CRYB_UNUSED, void *arg CRYB_UNUSED)
     81 {
     82 	pam_handle_t *pamh;
     83 	char **envlist;
     84 	int pam_err, ret;
     85 
     86 	ret = 1;
     87 	pam_err = pam_start("t_pam_env", "test", &t_null_pamc, &pamh);
     88 	t_assert(pam_err == PAM_SUCCESS);
     89 	pam_err = pam_putenv(pamh, T_ENV_NAMEVALUE);
     90 	ret &= t_compare_pam_err(PAM_SUCCESS, pam_err);
     91 	envlist = pam_getenvlist(pamh);
     92 	ret &= t_is_not_null(envlist);
     93 	if (envlist != NULL) {
     94 		ret &= t_compare_str(T_ENV_NAMEVALUE, envlist[0])
     95 		    & t_is_null(envlist[1]);
     96 		openpam_free_envlist(envlist);
     97 	}
     98 	pam_end(pamh, pam_err);
     99 	return (ret);
    100 }
    101 
    102 static int
    103 t_setenv_simple(char **desc CRYB_UNUSED, void *arg CRYB_UNUSED)
    104 {
    105 	pam_handle_t *pamh;
    106 	char **envlist;
    107 	int pam_err, ret;
    108 
    109 	ret = 1;
    110 	pam_err = pam_start("t_pam_env", "test", &t_null_pamc, &pamh);
    111 	t_assert(pam_err == PAM_SUCCESS);
    112 	pam_err = pam_setenv(pamh, T_ENV_NAME, T_ENV_VALUE, 0);
    113 	ret &= t_compare_pam_err(PAM_SUCCESS, pam_err);
    114 	envlist = pam_getenvlist(pamh);
    115 	ret &= t_is_not_null(envlist);
    116 	if (envlist != NULL) {
    117 		ret &= t_compare_str(T_ENV_NAMEVALUE, envlist[0])
    118 		    & t_is_null(envlist[1]);
    119 		openpam_free_envlist(envlist);
    120 	}
    121 	pam_end(pamh, pam_err);
    122 	return (ret);
    123 }
    124 
    125 static int
    126 t_getenv_empty(char **desc CRYB_UNUSED, void *arg CRYB_UNUSED)
    127 {
    128 	pam_handle_t *pamh;
    129 	const char *value;
    130 	int pam_err, ret;
    131 
    132 	ret = 1;
    133 	pam_err = pam_start("t_pam_env", "test", &t_null_pamc, &pamh);
    134 	t_assert(pam_err == PAM_SUCCESS);
    135 	value = pam_getenv(pamh, T_ENV_NAME);
    136 	ret &= t_compare_str(NULL, value);
    137 	pam_end(pamh, pam_err);
    138 	return (ret);
    139 }
    140 
    141 static int
    142 t_getenv_simple_miss(char **desc CRYB_UNUSED, void *arg CRYB_UNUSED)
    143 {
    144 	pam_handle_t *pamh;
    145 	const char *value;
    146 	int pam_err, ret;
    147 
    148 	ret = 1;
    149 	pam_err = pam_start("t_pam_env", "test", &t_null_pamc, &pamh);
    150 	t_assert(pam_err == PAM_SUCCESS);
    151 	pam_err = pam_setenv(pamh, T_ENV_NAME, T_ENV_VALUE, 0);
    152 	t_assert(pam_err == PAM_SUCCESS);
    153 	value = pam_getenv(pamh, "XYZZY");
    154 	ret &= t_compare_str(NULL, value);
    155 	pam_end(pamh, pam_err);
    156 	return (ret);
    157 }
    158 
    159 static int
    160 t_getenv_simple_hit(char **desc CRYB_UNUSED, void *arg CRYB_UNUSED)
    161 {
    162 	pam_handle_t *pamh;
    163 	const char *value;
    164 	int pam_err, ret;
    165 
    166 	ret = 1;
    167 	pam_err = pam_start("t_pam_env", "test", &t_null_pamc, &pamh);
    168 	t_assert(pam_err == PAM_SUCCESS);
    169 	pam_err = pam_setenv(pamh, T_ENV_NAME, T_ENV_VALUE, 0);
    170 	t_assert(pam_err == PAM_SUCCESS);
    171 	value = pam_getenv(pamh, T_ENV_NAME);
    172 	ret &= t_compare_str(T_ENV_VALUE, value);
    173 	pam_end(pamh, pam_err);
    174 	return (ret);
    175 }
    176 
    177 
    178 /***************************************************************************
    180  * Boilerplate
    181  */
    182 
    183 static int
    184 t_prepare(int argc CRYB_UNUSED, char *argv[] CRYB_UNUSED)
    185 {
    186 
    187 	openpam_set_feature(OPENPAM_FALLBACK_TO_OTHER, 0);
    188 
    189 	t_add_test(t_env_empty, NULL, "initially empty");
    190 	t_add_test(t_putenv_simple, NULL, "put - simple");
    191 	t_add_test(t_setenv_simple, NULL, "set - simple");
    192 	t_add_test(t_getenv_empty, NULL, "get - empty");
    193 	t_add_test(t_getenv_simple_miss, NULL, "get - simple (miss)");
    194 	t_add_test(t_getenv_simple_hit, NULL, "get - simple (hit)");
    195 
    196 	return (0);
    197 }
    198 
    199 int
    200 main(int argc, char *argv[])
    201 {
    202 
    203 	t_main(t_prepare, NULL, argc, argv);
    204 }
    205