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