diff options
Diffstat (limited to 'libapol/tests/user-tests.c')
-rw-r--r-- | libapol/tests/user-tests.c | 159 |
1 files changed, 159 insertions, 0 deletions
diff --git a/libapol/tests/user-tests.c b/libapol/tests/user-tests.c new file mode 100644 index 0000000..2d912c0 --- /dev/null +++ b/libapol/tests/user-tests.c @@ -0,0 +1,159 @@ +/** + * @file + * + * Test the user queries. + * + * @author Jeremy A. Mowery jmowery@tresys.com + * @author Jason Tang jtang@tresys.com + * + * Copyright (C) 2007 Tresys Technology, LLC + * + * This library is free software; you can redistribute it and/or + * modify it under the terms of the GNU Lesser General Public + * License as published by the Free Software Foundation; either + * version 2.1 of the License, or (at your option) any later version. + * + * This library is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU + * Lesser General Public License for more details. + * + * You should have received a copy of the GNU Lesser General Public + * License along with this library; if not, write to the Free Software + * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA + */ + +#include <config.h> + +#include <CUnit/CUnit.h> +#include <apol/user-query.h> +#include <apol/policy.h> +#include <apol/policy-path.h> +#include <stdbool.h> + +#define SOURCE_POLICY TEST_POLICIES "/setools/apol/user_mls_testing_policy.conf" + +static apol_policy_t *sp = NULL; +static qpol_policy_t *qp = NULL; + +static void user_basic(void) +{ + apol_user_query_t *q = apol_user_query_create(); + CU_ASSERT_PTR_NOT_NULL_FATAL(q); + + apol_vector_t *v = NULL; + CU_ASSERT(apol_user_get_by_query(sp, q, &v) == 0); + CU_ASSERT(v != NULL && apol_vector_get_size(v) == 10); + apol_vector_destroy(&v); + + apol_user_query_set_role(sp, q, "object_r"); + CU_ASSERT(apol_user_get_by_query(sp, q, &v) == 0); + CU_ASSERT(v != NULL && apol_vector_get_size(v) == 10); + apol_vector_destroy(&v); + + apol_user_query_set_user(sp, q, "sys"); + CU_ASSERT(apol_user_get_by_query(sp, q, &v) == 0); + CU_ASSERT(v != NULL && apol_vector_get_size(v) == 0); + apol_vector_destroy(&v); + + apol_user_query_set_user(sp, q, NULL); + apol_user_query_set_role(sp, q, "staff_r"); + CU_ASSERT(apol_user_get_by_query(sp, q, &v) == 0); + CU_ASSERT(v != NULL && apol_vector_get_size(v) == 3); + bool found_staff = false, found_rick = false, found_simple = false; + for (size_t i = 0; i < apol_vector_get_size(v); i++) { + qpol_user_t *u = (qpol_user_t *) apol_vector_get_element(v, i); + const char *name; + qpol_user_get_name(qp, u, &name); + if (strcmp(name, "staff_u") == 0) { + found_staff = true; + } else if (strcmp(name, "rick_u") == 0) { + found_rick = true; + } else if (strcmp(name, "simple_u") == 0) { + found_simple = true; + } else { + CU_ASSERT(0); + } + } + CU_ASSERT(found_staff && found_rick && found_simple); + apol_vector_destroy(&v); + + apol_user_query_set_role(sp, q, "not_in_the_policy_r"); + CU_ASSERT(apol_user_get_by_query(sp, q, &v) == 0); + CU_ASSERT(v != NULL && apol_vector_get_size(v) == 0); + apol_vector_destroy(&v); + + apol_user_query_destroy(&q); +} + +static void user_regex(void) +{ + apol_user_query_t *q = apol_user_query_create(); + CU_ASSERT_PTR_NOT_NULL_FATAL(q); + apol_user_query_set_regex(sp, q, 1); + + apol_user_query_set_user(sp, q, "*"); + apol_vector_t *v = NULL; + CU_ASSERT(apol_user_get_by_query(sp, q, &v) < 0 && v == NULL); + + apol_user_query_set_user(sp, q, "st"); + CU_ASSERT(apol_user_get_by_query(sp, q, &v) == 0); + CU_ASSERT(v != NULL && apol_vector_get_size(v) == 3); + bool found_staff = false, found_system = false, found_guest = false; + for (size_t i = 0; i < apol_vector_get_size(v); i++) { + qpol_user_t *u = (qpol_user_t *) apol_vector_get_element(v, i); + const char *name; + qpol_user_get_name(qp, u, &name); + if (strcmp(name, "staff_u") == 0) { + found_staff = true; + } else if (strcmp(name, "system_u") == 0) { + found_system = true; + } else if (strcmp(name, "guest_u") == 0) { + found_guest = true; + } else { + CU_ASSERT(0); + } + } + CU_ASSERT(found_staff && found_system && found_guest); + apol_vector_destroy(&v); + + apol_user_query_set_user(sp, q, NULL); + apol_user_query_set_role(sp, q, "user_r"); + CU_ASSERT(apol_user_get_by_query(sp, q, &v) == 0); + CU_ASSERT(v != NULL && apol_vector_get_size(v) == 3); + apol_vector_destroy(&v); + + apol_user_query_destroy(&q); +} + +CU_TestInfo user_tests[] = { + {"basic query", user_basic} + , + {"regex query", user_regex} + , + CU_TEST_INFO_NULL +}; + +int user_init() +{ + apol_policy_path_t *ppath = apol_policy_path_create(APOL_POLICY_PATH_TYPE_MONOLITHIC, SOURCE_POLICY, NULL); + if (ppath == NULL) { + return 1; + } + + if ((sp = apol_policy_create_from_policy_path(ppath, 0, NULL, NULL)) == NULL) { + apol_policy_path_destroy(&ppath); + return 1; + } + apol_policy_path_destroy(&ppath); + + qp = apol_policy_get_qpol(sp); + + return 0; +} + +int user_cleanup() +{ + apol_policy_destroy(&sp); + return 0; +} |