1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
|
/*
* Copyright 2022 the Pacemaker project contributors
*
* The version control history for this file may have further details.
*
* This source code is licensed under the GNU General Public License version 2
* or later (GPLv2+) WITHOUT ANY WARRANTY.
*/
#include <crm_internal.h>
#include <crm/common/unittest_internal.h>
#include "crmcommon_private.h"
#include "mock_private.h"
#include <pwd.h>
#include <sys/types.h>
static void
calloc_fails(void **state)
{
uid_t uid;
gid_t gid;
pcmk__mock_calloc = true; // calloc() will return NULL
expect_value(__wrap_calloc, nmemb, 1);
expect_value(__wrap_calloc, size, PCMK__PW_BUFFER_LEN);
assert_int_equal(crm_user_lookup("hauser", &uid, &gid), -ENOMEM);
pcmk__mock_calloc = false; // Use real calloc()
}
static void
getpwnam_r_fails(void **state)
{
uid_t uid;
gid_t gid;
// Set getpwnam_r() return value and result parameter
pcmk__mock_getpwnam_r = true;
expect_string(__wrap_getpwnam_r, name, "hauser");
expect_any(__wrap_getpwnam_r, pwd);
expect_any(__wrap_getpwnam_r, buf);
expect_value(__wrap_getpwnam_r, buflen, PCMK__PW_BUFFER_LEN);
expect_any(__wrap_getpwnam_r, result);
will_return(__wrap_getpwnam_r, EIO);
will_return(__wrap_getpwnam_r, NULL);
assert_int_equal(crm_user_lookup("hauser", &uid, &gid), -EIO);
pcmk__mock_getpwnam_r = false;
}
static void
no_matching_pwent(void **state)
{
uid_t uid;
gid_t gid;
// Set getpwnam_r() return value and result parameter
pcmk__mock_getpwnam_r = true;
expect_string(__wrap_getpwnam_r, name, "hauser");
expect_any(__wrap_getpwnam_r, pwd);
expect_any(__wrap_getpwnam_r, buf);
expect_value(__wrap_getpwnam_r, buflen, PCMK__PW_BUFFER_LEN);
expect_any(__wrap_getpwnam_r, result);
will_return(__wrap_getpwnam_r, 0);
will_return(__wrap_getpwnam_r, NULL);
assert_int_equal(crm_user_lookup("hauser", &uid, &gid), -EINVAL);
pcmk__mock_getpwnam_r = false;
}
static void
entry_found(void **state)
{
uid_t uid;
gid_t gid;
/* We don't care about any of the other fields of the password entry, so just
* leave them blank.
*/
struct passwd returned_ent = { .pw_uid = 1000, .pw_gid = 1000 };
/* Test getpwnam_r returning a valid passwd entry, but we don't pass uid or gid. */
// Set getpwnam_r() return value and result parameter
pcmk__mock_getpwnam_r = true;
expect_string(__wrap_getpwnam_r, name, "hauser");
expect_any(__wrap_getpwnam_r, pwd);
expect_any(__wrap_getpwnam_r, buf);
expect_value(__wrap_getpwnam_r, buflen, PCMK__PW_BUFFER_LEN);
expect_any(__wrap_getpwnam_r, result);
will_return(__wrap_getpwnam_r, 0);
will_return(__wrap_getpwnam_r, &returned_ent);
assert_int_equal(crm_user_lookup("hauser", NULL, NULL), 0);
/* Test getpwnam_r returning a valid passwd entry, and we do pass uid and gid. */
// Set getpwnam_r() return value and result parameter
expect_string(__wrap_getpwnam_r, name, "hauser");
expect_any(__wrap_getpwnam_r, pwd);
expect_any(__wrap_getpwnam_r, buf);
expect_value(__wrap_getpwnam_r, buflen, PCMK__PW_BUFFER_LEN);
expect_any(__wrap_getpwnam_r, result);
will_return(__wrap_getpwnam_r, 0);
will_return(__wrap_getpwnam_r, &returned_ent);
assert_int_equal(crm_user_lookup("hauser", &uid, &gid), 0);
assert_int_equal(uid, 1000);
assert_int_equal(gid, 1000);
pcmk__mock_getpwnam_r = false;
}
PCMK__UNIT_TEST(NULL, NULL,
cmocka_unit_test(calloc_fails),
cmocka_unit_test(getpwnam_r_fails),
cmocka_unit_test(no_matching_pwent),
cmocka_unit_test(entry_found))
|
