crm_user_lookup_test.c

Go to the documentation of this file.

/*
 * 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 Lesser General Public License
 * version 2.1 or later (LGPLv2.1+) WITHOUT ANY WARRANTY.
 */
 
#include <crm_internal.h>
#include "mock_private.h"
 
#include <pwd.h>
#include <stdarg.h>
#include <stddef.h>
#include <stdint.h>
#include <setjmp.h>
#include <cmocka.h>
#include <sys/types.h>
 
void *
__wrap_calloc(size_t nmemb, size_t size)
{
    int fail = mock_type(int);
 
    if (fail) {
        return mock_ptr_type(void *);
    } else {
        return __real_calloc(nmemb, size);
    }
}
 
int
__wrap_getpwnam_r(const char *name, struct passwd *pwd, char *buf, size_t buflen,
                  struct passwd **result)
{
    int retval = mock_type(int);
 
    *result = mock_ptr_type(struct passwd *);
 
    return retval;
}
 
static void
calloc_fails(void **state)
{
    uid_t uid;
    gid_t gid;
 
    /* Test calloc() returning NULL. */
 
    will_return(__wrap_calloc, 1);                      // calloc() should fail
    will_return(__wrap_calloc, NULL);                   // calloc() return value
 
    assert_int_equal(crm_user_lookup("hauser", &uid, &gid), -ENOMEM);
}
 
static void
getpwnam_r_fails(void **state)
{
    uid_t uid;
    gid_t gid;
 
    will_return_always(__wrap_calloc, 0);               // calloc() should never fail
 
    will_return(__wrap_getpwnam_r, EIO);                // getpwnam_r() return value
    will_return(__wrap_getpwnam_r, NULL);               // result parameter to getpwnam_r()
 
    assert_int_equal(crm_user_lookup("hauser", &uid, &gid), -EIO);
}
 
static void
no_matching_pwent(void **state)
{
    uid_t uid;
    gid_t gid;
 
    will_return_always(__wrap_calloc, 0);               // calloc() should never fail
 
    will_return(__wrap_getpwnam_r, 0);                  // getpwnam_r() return value
    will_return(__wrap_getpwnam_r, NULL);               // result parameter to getpwnam_r()
 
    assert_int_equal(crm_user_lookup("hauser", &uid, &gid), -EINVAL);
}
 
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 };
 
    will_return_always(__wrap_calloc, 0);               // calloc() should never fail
 
    /* Test getpwnam_r returning a valid passwd entry, but we don't pass uid or gid. */
 
    will_return(__wrap_getpwnam_r, 0);                  // getpwnam_r() return value
    will_return(__wrap_getpwnam_r, &returned_ent);      // result parameter to getpwnam_r()
 
    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. */
 
    will_return(__wrap_getpwnam_r, 0);                  // getpwnam_r() return value
    will_return(__wrap_getpwnam_r, &returned_ent);      // result parameter to getpwnam_r()
 
    assert_int_equal(crm_user_lookup("hauser", &uid, &gid), 0);
    assert_int_equal(uid, 1000);
    assert_int_equal(gid, 1000);
}
 
int main(int argc, char **argv)
{
    const struct CMUnitTest tests[] = {
        cmocka_unit_test(calloc_fails),
        cmocka_unit_test(getpwnam_r_fails),
        cmocka_unit_test(no_matching_pwent),
        cmocka_unit_test(entry_found),
    };
 
    cmocka_set_message_output(CM_OUTPUT_TAP);
    return cmocka_run_group_tests(tests, NULL, NULL);
}