libsemanage/tests/test_ibendport.c - platform/external/selinux - Git at Google

 /*
  * Authors: Jan Zarsky <[email protected]>
  *
  * Copyright (C) 2019 Red Hat, Inc.
  *
  * 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 "utilities.h"
 #include "test_ibendport.h"

 #define IBENDPORT_COUNT 3
 #define IBENDPORT_1_NAME "mlx4_0"
 #define IBENDPORT_1_PORT 1
 #define IBENDPORT_1_CON "system_u:object_r:first_ibendport_t:s0"
 #define IBENDPORT_2_NAME "mlx4_1"
 #define IBENDPORT_2_PORT 2
 #define IBENDPORT_2_CON "system_u:object_r:second_ibendport_second_t:s0"
 #define IBENDPORT_3_NAME "mlx4_1"
 #define IBENDPORT_3_PORT 3
 #define IBENDPORT_3_CON "system_u:object_r:third_ibendport_second_t:s0"

 /* ibendports_policy.h */
 static void test_ibendport_query(void);
 static void test_ibendport_exists(void);
 static void test_ibendport_count(void);
 static void test_ibendport_iterate(void);
 static void test_ibendport_list(void);

 /* ibendports_local.h */
 static void test_ibendport_modify_del_query_local(void);
 static void test_ibendport_exists_local(void);
 static void test_ibendport_count_local(void);
 static void test_ibendport_iterate_local(void);
 static void test_ibendport_list_local(void);

 extern semanage_handle_t *sh;

 int ibendport_test_init(void)
 {
 	if (create_test_store() < 0) {
 		fprintf(stderr, "Could not create test store\n");
 		return 1;
 	}

 	if (write_test_policy_from_file("test_ibendport.policy") < 0) {
 		fprintf(stderr, "Could not write test policy\n");
 		return 1;
 	}

 	return 0;
 }

 int ibendport_test_cleanup(void)
 {
 	if (destroy_test_store() < 0) {
 		fprintf(stderr, "Could not destroy test store\n");
 		return 1;
 	}

 	return 0;
 }

 int ibendport_add_tests(CU_pSuite suite)
 {
 	CU_add_test(suite, "ibendport_query", test_ibendport_query);
 	CU_add_test(suite, "ibendport_exists", test_ibendport_exists);
 	CU_add_test(suite, "ibendport_count", test_ibendport_count);
 	CU_add_test(suite, "ibendport_iterate", test_ibendport_iterate);
 	CU_add_test(suite, "ibendport_list", test_ibendport_list);

 	CU_add_test(suite, "ibendport_modify_del_query_local",
 		    test_ibendport_modify_del_query_local);
 	CU_add_test(suite, "ibendport_exists_local",
 		    test_ibendport_exists_local);
 	CU_add_test(suite, "ibendport_count_local", test_ibendport_count_local);
 	CU_add_test(suite, "ibendport_iterate_local",
 		    test_ibendport_iterate_local);
 	CU_add_test(suite, "ibendport_list_local", test_ibendport_list_local);

 	return 0;
 }

 /* Helpers */

 static semanage_ibendport_t *get_ibendport_nth(int idx)
 {
 	semanage_ibendport_t **records;
 	semanage_ibendport_t *ibendport;
 	unsigned int count;

 	if (idx == I_NULL)
 		return NULL;

 	CU_ASSERT_FATAL(semanage_ibendport_list(sh, &records, &count) >= 0);
 	CU_ASSERT_FATAL(count >= (unsigned int) idx + 1);

 	ibendport = records[idx];

 	for (unsigned int i = 0; i < count; i++)
 		if (i != (unsigned int) idx)
 			semanage_ibendport_free(records[i]);

 	free(records);

 	return ibendport;
 }

 static semanage_ibendport_key_t *get_ibendport_key_nth(int idx)
 {
 	semanage_ibendport_key_t *key;
 	semanage_ibendport_t *ibendport;
 	int res;

 	if (idx == I_NULL)
 		return NULL;

 	ibendport = get_ibendport_nth(idx);

 	res = semanage_ibendport_key_extract(sh, ibendport, &key);

 	CU_ASSERT_FATAL(res >= 0);
 	CU_ASSERT_PTR_NOT_NULL_FATAL(key);

 	semanage_ibendport_free(ibendport);

 	return key;
 }

 static void add_local_ibendport(int idx)
 {
 	semanage_ibendport_t *ibendport;
 	semanage_ibendport_key_t *key = NULL;

 	ibendport = get_ibendport_nth(idx);

 	CU_ASSERT_FATAL(semanage_ibendport_key_extract(sh, ibendport,
 						       &key) >= 0);
 	CU_ASSERT_PTR_NOT_NULL_FATAL(key);

 	CU_ASSERT_FATAL(semanage_ibendport_modify_local(sh, key,
 							ibendport) >= 0);

 	semanage_ibendport_key_free(key);
 	semanage_ibendport_free(ibendport);
 }

 static void delete_local_ibendport(int idx)
 {
 	semanage_ibendport_key_t *key = NULL;
 	key = get_ibendport_key_nth(idx);
 	CU_ASSERT_FATAL(semanage_ibendport_del_local(sh, key) >= 0);

 	semanage_ibendport_key_free(key);
 }

 /* Function semanage_ibendport_query */
 static void test_ibendport_query(void)
 {
 	semanage_ibendport_t *ibendport = NULL;
 	semanage_ibendport_t *ibendport_exp = NULL;
 	semanage_ibendport_key_t *key = NULL;
 	semanage_context_t *con = NULL;
 	semanage_context_t *con_exp = NULL;
 	char *name;
 	char *name_exp;

 	/* setup */
 	setup_handle(SH_CONNECT);
 	key = get_ibendport_key_nth(I_FIRST);
 	ibendport_exp = get_ibendport_nth(I_FIRST);

 	/* test */
 	CU_ASSERT(semanage_ibendport_query(sh, key, &ibendport) >= 0);
 	CU_ASSERT_PTR_NOT_NULL_FATAL(ibendport);

 	CU_ASSERT(semanage_ibendport_get_ibdev_name(sh, ibendport, &name) >= 0);
 	CU_ASSERT_PTR_NOT_NULL_FATAL(name);
 	CU_ASSERT(semanage_ibendport_get_ibdev_name(sh, ibendport_exp,
 						    &name_exp) >= 0);
 	CU_ASSERT_PTR_NOT_NULL_FATAL(name_exp);
 	assert(name);
 	CU_ASSERT_STRING_EQUAL(name, name_exp);

 	CU_ASSERT(semanage_ibendport_get_port(ibendport) ==
 		  semanage_ibendport_get_port(ibendport_exp));

 	con = semanage_ibendport_get_con(ibendport);
 	con_exp = semanage_ibendport_get_con(ibendport_exp);
 	CU_ASSERT_PTR_NOT_NULL_FATAL(con);
 	CU_ASSERT_PTR_NOT_NULL_FATAL(con_exp);
 	CU_ASSERT_CONTEXT_EQUAL(con, con_exp);

 	/* cleanup */
 	free(name_exp);
 	free(name);
 	semanage_ibendport_key_free(key);
 	semanage_ibendport_free(ibendport);
 	semanage_ibendport_free(ibendport_exp);
 	cleanup_handle(SH_CONNECT);
 }

 /* Function semanage_ibendport_exists */
 static void test_ibendport_exists(void)
 {
 	semanage_ibendport_key_t *key1 = NULL;
 	semanage_ibendport_key_t *key2 = NULL;
 	int resp = 42;

 	/* setup */
 	setup_handle(SH_CONNECT);
 	key1 = get_ibendport_key_nth(I_FIRST);
 	CU_ASSERT(semanage_ibendport_key_create(sh, "asdf", 1, &key2) >= 0);

 	/* test */
 	CU_ASSERT(semanage_ibendport_exists(sh, key1, &resp) >= 0);
 	CU_ASSERT(resp);

 	CU_ASSERT(semanage_ibendport_exists(sh, key2, &resp) >= 0);
 	CU_ASSERT(!resp);

 	/* cleanup */
 	semanage_ibendport_key_free(key1);
 	semanage_ibendport_key_free(key2);
 	cleanup_handle(SH_CONNECT);
 }

 /* Function semanage_ibendport_count */
 static void test_ibendport_count(void)
 {
 	unsigned int count = 42;

 	/* setup */
 	setup_handle(SH_CONNECT);

 	/* test */
 	CU_ASSERT(semanage_ibendport_count(sh, &count) >= 0);
 	CU_ASSERT(count == IBENDPORT_COUNT);

 	/* cleanup */
 	cleanup_handle(SH_CONNECT);
 }

 /* Function semanage_ibendport_iterate */
 unsigned int helper_ibendport_iterate_counter = 0;

 static int helper_ibendport_iterate(const semanage_ibendport_t *ibendport,
 			     void *fn_arg)
 {
 	CU_ASSERT(fn_arg == (void *) 42);
 	helper_ibendport_iterate_counter++;
 	return 0;
 }

 static int helper_ibendport_iterate_error(const semanage_ibendport_t *ibendport,
 				   void *fn_arg)
 {
 	CU_ASSERT(fn_arg == (void *) 42);
 	helper_ibendport_iterate_counter++;
 	return -1;
 }

 static int helper_ibendport_iterate_break(const semanage_ibendport_t *ibendport,
 				   void *fn_arg)
 {
 	CU_ASSERT(fn_arg == (void *) 42);
 	helper_ibendport_iterate_counter++;
 	return 1;
 }

 static void test_ibendport_iterate(void)
 {
 	/* setup */
 	setup_handle(SH_CONNECT);

 	/* test */
 	helper_ibendport_iterate_counter = 0;
 	CU_ASSERT(semanage_ibendport_iterate(sh, helper_ibendport_iterate,
 					     (void *) 42) >= 0);
 	CU_ASSERT(helper_ibendport_iterate_counter == IBENDPORT_COUNT);

 	/* test function which returns error */
 	helper_ibendport_iterate_counter = 0;
 	CU_ASSERT(semanage_ibendport_iterate(sh, helper_ibendport_iterate_error,
 					     (void *) 42) < 0);
 	CU_ASSERT(helper_ibendport_iterate_counter == 1);

 	/* test function which requests break */
 	helper_ibendport_iterate_counter = 0;
 	CU_ASSERT(semanage_ibendport_iterate(sh, helper_ibendport_iterate_break,
 					     (void *) 42) >= 0);
 	CU_ASSERT(helper_ibendport_iterate_counter == 1);

 	/* cleanup */
 	cleanup_handle(SH_CONNECT);
 }

 /* Function semanage_ibendport_list */
 static void test_ibendport_list(void)
 {
 	semanage_ibendport_t **records = NULL;
 	unsigned int count = 42;
 	char *name = NULL;
 	semanage_context_t *con = NULL;

 	/* setup */
 	setup_handle(SH_CONNECT);

 	/* test */
 	CU_ASSERT(semanage_ibendport_list(sh, &records, &count) >= 0);

 	CU_ASSERT_PTR_NOT_NULL_FATAL(records);
 	assert(records);
 	CU_ASSERT(count == IBENDPORT_COUNT);

 	for (unsigned int i = 0; i < count; i++) {
 		CU_ASSERT_PTR_NOT_NULL_FATAL(records[i]);
 		CU_ASSERT(semanage_ibendport_get_ibdev_name(sh, records[i],
 							    &name) >= 0);
 		con = semanage_ibendport_get_con(records[i]);
 		CU_ASSERT_PTR_NOT_NULL_FATAL(con);
 		free(name);
 	}

 	/* cleanup */
 	for (unsigned int i = 0; i < count; i++)
 		semanage_ibendport_free(records[i]);

 	free(records);
 	cleanup_handle(SH_CONNECT);
 }

 /* Function semanage_ibendport_modify_local, semanage_ibendport_del_local,
  * semanage_ibendport_query_local
  */
 static void test_ibendport_modify_del_query_local(void)
 {
 	semanage_ibendport_t *ibendport;
 	semanage_ibendport_t *ibendport_local;
 	semanage_ibendport_key_t *key = NULL;

 	/* setup */
 	setup_handle(SH_TRANS);
 	ibendport = get_ibendport_nth(I_FIRST);
 	CU_ASSERT(semanage_ibendport_key_extract(sh, ibendport, &key) >= 0);
 	CU_ASSERT_PTR_NOT_NULL(key);

 	/* test */
 	CU_ASSERT(semanage_ibendport_modify_local(sh, key, ibendport) >= 0);

 	/* write changes to file */
 	helper_commit();
 	helper_begin_transaction();

 	CU_ASSERT(semanage_ibendport_query_local(sh, key,
 						 &ibendport_local) >= 0);
 	CU_ASSERT_PTR_NOT_NULL_FATAL(ibendport_local);
 	semanage_ibendport_free(ibendport_local);

 	CU_ASSERT(semanage_ibendport_del_local(sh, key) >= 0);
 	CU_ASSERT(semanage_ibendport_query_local(sh, key,
 						 &ibendport_local) < 0);

 	/* cleanup */
 	semanage_ibendport_key_free(key);
 	semanage_ibendport_free(ibendport);
 	cleanup_handle(SH_TRANS);
 }

 /* Function semanage_ibendport_exists_local */
 static void test_ibendport_exists_local(void)
 {
 	semanage_ibendport_key_t *key1 = NULL;
 	semanage_ibendport_key_t *key2 = NULL;
 	int resp = 42;

 	/* setup */
 	setup_handle(SH_TRANS);
 	add_local_ibendport(I_FIRST);
 	key1 = get_ibendport_key_nth(I_FIRST);
 	key2 = get_ibendport_key_nth(I_SECOND);

 	/* test */
 	CU_ASSERT(semanage_ibendport_exists_local(sh, key1, &resp) >= 0);
 	CU_ASSERT(resp);

 	CU_ASSERT(semanage_ibendport_exists_local(sh, key2, &resp) >= 0);
 	CU_ASSERT(!resp);

 	/* cleanup */
 	CU_ASSERT(semanage_ibendport_del_local(sh, key1) >= 0);
 	semanage_ibendport_key_free(key1);
 	semanage_ibendport_key_free(key2);
 	cleanup_handle(SH_TRANS);
 }

 /* Function semanage_ibendport_count_local */
 static void test_ibendport_count_local(void)
 {
 	unsigned int count = 42;

 	/* setup */
 	setup_handle(SH_TRANS);

 	/* test */
 	CU_ASSERT(semanage_ibendport_count_local(sh, &count) >= 0);
 	CU_ASSERT(count == 0);

 	add_local_ibendport(I_FIRST);
 	CU_ASSERT(semanage_ibendport_count_local(sh, &count) >= 0);
 	CU_ASSERT(count == 1);

 	add_local_ibendport(I_SECOND);
 	CU_ASSERT(semanage_ibendport_count_local(sh, &count) >= 0);
 	CU_ASSERT(count == 2);

 	delete_local_ibendport(I_SECOND);
 	CU_ASSERT(semanage_ibendport_count_local(sh, &count) >= 0);
 	CU_ASSERT(count == 1);

 	delete_local_ibendport(I_FIRST);
 	CU_ASSERT(semanage_ibendport_count_local(sh, &count) >= 0);
 	CU_ASSERT(count == 0);

 	/* cleanup */
 	cleanup_handle(SH_TRANS);
 }

 /* Function semanage_ibendport_iterate_local */
 unsigned int helper_ibendport_iterate_local_counter = 0;

 static int helper_ibendport_iterate_local(const semanage_ibendport_t *ibendport,
 				   void *fn_arg)
 {
 	CU_ASSERT(fn_arg == (void *) 42);
 	helper_ibendport_iterate_local_counter++;
 	return 0;
 }

 static int helper_ibendport_iterate_local_error(const semanage_ibendport_t *ibendport,
 					 void *fn_arg)
 {
 	CU_ASSERT(fn_arg == (void *) 42);
 	helper_ibendport_iterate_local_counter++;
 	return -1;
 }

 static int helper_ibendport_iterate_local_break(const semanage_ibendport_t *ibendport,
 					 void *fn_arg)
 {
 	CU_ASSERT(fn_arg == (void *) 42);
 	helper_ibendport_iterate_local_counter++;
 	return 1;
 }

 static void test_ibendport_iterate_local(void)
 {
 	/* setup */
 	setup_handle(SH_TRANS);
 	add_local_ibendport(I_FIRST);
 	add_local_ibendport(I_SECOND);
 	add_local_ibendport(I_THIRD);

 	/* test */
 	helper_ibendport_iterate_local_counter = 0;
 	CU_ASSERT(semanage_ibendport_iterate_local(sh,
 			     helper_ibendport_iterate_local, (void *) 42) >= 0);
 	CU_ASSERT(helper_ibendport_iterate_local_counter == 3);

 	/* test function which returns error */
 	helper_ibendport_iterate_local_counter = 0;
 	CU_ASSERT(semanage_ibendport_iterate_local(sh,
 			helper_ibendport_iterate_local_error, (void *) 42) < 0);
 	CU_ASSERT(helper_ibendport_iterate_local_counter == 1);

 	/* test function which requests break */
 	helper_ibendport_iterate_local_counter = 0;
 	CU_ASSERT(semanage_ibendport_iterate_local(sh,
 		       helper_ibendport_iterate_local_break, (void *) 42) >= 0);

 	/* cleanup */
 	delete_local_ibendport(I_FIRST);
 	delete_local_ibendport(I_SECOND);
 	delete_local_ibendport(I_THIRD);
 	cleanup_handle(SH_TRANS);
 }

 /* Function semanage_ibendport_list_local */
 static void test_ibendport_list_local(void)
 {
 	semanage_ibendport_t **records = NULL;
 	unsigned int count = 42;
 	char *name = NULL;
 	semanage_context_t *con = NULL;

 	/* setup */
 	setup_handle(SH_TRANS);
 	add_local_ibendport(I_FIRST);
 	add_local_ibendport(I_SECOND);
 	add_local_ibendport(I_THIRD);

 	/* test */
 	CU_ASSERT(semanage_ibendport_list_local(sh, &records, &count) >= 0);
 	CU_ASSERT_PTR_NOT_NULL_FATAL(records);
 	assert(records);
 	CU_ASSERT(count == 3);

 	for (unsigned int i = 0; i < count; i++) {
 		CU_ASSERT_PTR_NOT_NULL_FATAL(records[i]);
 		CU_ASSERT(semanage_ibendport_get_ibdev_name(sh, records[i],
 								   &name) >= 0);
 		con = semanage_ibendport_get_con(records[i]);
 		CU_ASSERT_PTR_NOT_NULL_FATAL(con);
 		free(name);
 	}

 	/* cleanup */
 	for (unsigned int i = 0; i < count; i++)
 		semanage_ibendport_free(records[i]);

 	free(records);
 	delete_local_ibendport(I_FIRST);
 	delete_local_ibendport(I_SECOND);
 	delete_local_ibendport(I_THIRD);
 	cleanup_handle(SH_TRANS);
 }
	/*
	* Authors: Jan Zarsky <[email protected]>
	*
	* Copyright (C) 2019 Red Hat, Inc.
	*
	* 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 "utilities.h"
	#include "test_ibendport.h"

	#define IBENDPORT_COUNT 3
	#define IBENDPORT_1_NAME "mlx4_0"
	#define IBENDPORT_1_PORT 1
	#define IBENDPORT_1_CON "system_u:object_r:first_ibendport_t:s0"
	#define IBENDPORT_2_NAME "mlx4_1"
	#define IBENDPORT_2_PORT 2
	#define IBENDPORT_2_CON "system_u:object_r:second_ibendport_second_t:s0"
	#define IBENDPORT_3_NAME "mlx4_1"
	#define IBENDPORT_3_PORT 3
	#define IBENDPORT_3_CON "system_u:object_r:third_ibendport_second_t:s0"

	/* ibendports_policy.h */
	static void test_ibendport_query(void);
	static void test_ibendport_exists(void);
	static void test_ibendport_count(void);
	static void test_ibendport_iterate(void);
	static void test_ibendport_list(void);

	/* ibendports_local.h */
	static void test_ibendport_modify_del_query_local(void);
	static void test_ibendport_exists_local(void);
	static void test_ibendport_count_local(void);
	static void test_ibendport_iterate_local(void);
	static void test_ibendport_list_local(void);

	extern semanage_handle_t *sh;

	int ibendport_test_init(void)
	{
	if (create_test_store() < 0) {
	fprintf(stderr, "Could not create test store\n");
	return 1;
	}

	if (write_test_policy_from_file("test_ibendport.policy") < 0) {
	fprintf(stderr, "Could not write test policy\n");
	return 1;
	}

	return 0;
	}

	int ibendport_test_cleanup(void)
	{
	if (destroy_test_store() < 0) {
	fprintf(stderr, "Could not destroy test store\n");
	return 1;
	}

	return 0;
	}

	int ibendport_add_tests(CU_pSuite suite)
	{
	CU_add_test(suite, "ibendport_query", test_ibendport_query);
	CU_add_test(suite, "ibendport_exists", test_ibendport_exists);
	CU_add_test(suite, "ibendport_count", test_ibendport_count);
	CU_add_test(suite, "ibendport_iterate", test_ibendport_iterate);
	CU_add_test(suite, "ibendport_list", test_ibendport_list);

	CU_add_test(suite, "ibendport_modify_del_query_local",
	test_ibendport_modify_del_query_local);
	CU_add_test(suite, "ibendport_exists_local",
	test_ibendport_exists_local);
	CU_add_test(suite, "ibendport_count_local", test_ibendport_count_local);
	CU_add_test(suite, "ibendport_iterate_local",
	test_ibendport_iterate_local);
	CU_add_test(suite, "ibendport_list_local", test_ibendport_list_local);

	return 0;
	}

	/* Helpers */

	static semanage_ibendport_t *get_ibendport_nth(int idx)
	{
	semanage_ibendport_t **records;
	semanage_ibendport_t *ibendport;
	unsigned int count;

	if (idx == I_NULL)
	return NULL;

	CU_ASSERT_FATAL(semanage_ibendport_list(sh, &records, &count) >= 0);
	CU_ASSERT_FATAL(count >= (unsigned int) idx + 1);

	ibendport = records[idx];

	for (unsigned int i = 0; i < count; i++)
	if (i != (unsigned int) idx)
	semanage_ibendport_free(records[i]);

	free(records);

	return ibendport;
	}

	static semanage_ibendport_key_t *get_ibendport_key_nth(int idx)
	{
	semanage_ibendport_key_t *key;
	semanage_ibendport_t *ibendport;
	int res;

	if (idx == I_NULL)
	return NULL;

	ibendport = get_ibendport_nth(idx);

	res = semanage_ibendport_key_extract(sh, ibendport, &key);

	CU_ASSERT_FATAL(res >= 0);
	CU_ASSERT_PTR_NOT_NULL_FATAL(key);

	semanage_ibendport_free(ibendport);

	return key;
	}

	static void add_local_ibendport(int idx)
	{
	semanage_ibendport_t *ibendport;
	semanage_ibendport_key_t *key = NULL;

	ibendport = get_ibendport_nth(idx);

	CU_ASSERT_FATAL(semanage_ibendport_key_extract(sh, ibendport,
	&key) >= 0);
	CU_ASSERT_PTR_NOT_NULL_FATAL(key);

	CU_ASSERT_FATAL(semanage_ibendport_modify_local(sh, key,
	ibendport) >= 0);

	semanage_ibendport_key_free(key);
	semanage_ibendport_free(ibendport);
	}

	static void delete_local_ibendport(int idx)
	{
	semanage_ibendport_key_t *key = NULL;
	key = get_ibendport_key_nth(idx);
	CU_ASSERT_FATAL(semanage_ibendport_del_local(sh, key) >= 0);

	semanage_ibendport_key_free(key);
	}

	/* Function semanage_ibendport_query */
	static void test_ibendport_query(void)
	{
	semanage_ibendport_t *ibendport = NULL;
	semanage_ibendport_t *ibendport_exp = NULL;
	semanage_ibendport_key_t *key = NULL;
	semanage_context_t *con = NULL;
	semanage_context_t *con_exp = NULL;
	char *name;
	char *name_exp;

	/* setup */
	setup_handle(SH_CONNECT);
	key = get_ibendport_key_nth(I_FIRST);
	ibendport_exp = get_ibendport_nth(I_FIRST);

	/* test */
	CU_ASSERT(semanage_ibendport_query(sh, key, &ibendport) >= 0);
	CU_ASSERT_PTR_NOT_NULL_FATAL(ibendport);

	CU_ASSERT(semanage_ibendport_get_ibdev_name(sh, ibendport, &name) >= 0);
	CU_ASSERT_PTR_NOT_NULL_FATAL(name);
	CU_ASSERT(semanage_ibendport_get_ibdev_name(sh, ibendport_exp,
	&name_exp) >= 0);
	CU_ASSERT_PTR_NOT_NULL_FATAL(name_exp);
	assert(name);
	CU_ASSERT_STRING_EQUAL(name, name_exp);

	CU_ASSERT(semanage_ibendport_get_port(ibendport) ==
	semanage_ibendport_get_port(ibendport_exp));

	con = semanage_ibendport_get_con(ibendport);
	con_exp = semanage_ibendport_get_con(ibendport_exp);
	CU_ASSERT_PTR_NOT_NULL_FATAL(con);
	CU_ASSERT_PTR_NOT_NULL_FATAL(con_exp);
	CU_ASSERT_CONTEXT_EQUAL(con, con_exp);

	/* cleanup */
	free(name_exp);
	free(name);
	semanage_ibendport_key_free(key);
	semanage_ibendport_free(ibendport);
	semanage_ibendport_free(ibendport_exp);
	cleanup_handle(SH_CONNECT);
	}

	/* Function semanage_ibendport_exists */
	static void test_ibendport_exists(void)
	{
	semanage_ibendport_key_t *key1 = NULL;
	semanage_ibendport_key_t *key2 = NULL;
	int resp = 42;

	/* setup */
	setup_handle(SH_CONNECT);
	key1 = get_ibendport_key_nth(I_FIRST);
	CU_ASSERT(semanage_ibendport_key_create(sh, "asdf", 1, &key2) >= 0);

	/* test */
	CU_ASSERT(semanage_ibendport_exists(sh, key1, &resp) >= 0);
	CU_ASSERT(resp);

	CU_ASSERT(semanage_ibendport_exists(sh, key2, &resp) >= 0);
	CU_ASSERT(!resp);

	/* cleanup */
	semanage_ibendport_key_free(key1);
	semanage_ibendport_key_free(key2);
	cleanup_handle(SH_CONNECT);
	}

	/* Function semanage_ibendport_count */
	static void test_ibendport_count(void)
	{
	unsigned int count = 42;

	/* setup */
	setup_handle(SH_CONNECT);

	/* test */
	CU_ASSERT(semanage_ibendport_count(sh, &count) >= 0);
	CU_ASSERT(count == IBENDPORT_COUNT);

	/* cleanup */
	cleanup_handle(SH_CONNECT);
	}

	/* Function semanage_ibendport_iterate */
	unsigned int helper_ibendport_iterate_counter = 0;

	static int helper_ibendport_iterate(const semanage_ibendport_t *ibendport,
	void *fn_arg)
	{
	CU_ASSERT(fn_arg == (void *) 42);
	helper_ibendport_iterate_counter++;
	return 0;
	}

	static int helper_ibendport_iterate_error(const semanage_ibendport_t *ibendport,
	void *fn_arg)
	{
	CU_ASSERT(fn_arg == (void *) 42);
	helper_ibendport_iterate_counter++;
	return -1;
	}

	static int helper_ibendport_iterate_break(const semanage_ibendport_t *ibendport,
	void *fn_arg)
	{
	CU_ASSERT(fn_arg == (void *) 42);
	helper_ibendport_iterate_counter++;
	return 1;
	}

	static void test_ibendport_iterate(void)
	{
	/* setup */
	setup_handle(SH_CONNECT);

	/* test */
	helper_ibendport_iterate_counter = 0;
	CU_ASSERT(semanage_ibendport_iterate(sh, helper_ibendport_iterate,
	(void *) 42) >= 0);
	CU_ASSERT(helper_ibendport_iterate_counter == IBENDPORT_COUNT);

	/* test function which returns error */
	helper_ibendport_iterate_counter = 0;
	CU_ASSERT(semanage_ibendport_iterate(sh, helper_ibendport_iterate_error,
	(void *) 42) < 0);
	CU_ASSERT(helper_ibendport_iterate_counter == 1);

	/* test function which requests break */
	helper_ibendport_iterate_counter = 0;
	CU_ASSERT(semanage_ibendport_iterate(sh, helper_ibendport_iterate_break,
	(void *) 42) >= 0);
	CU_ASSERT(helper_ibendport_iterate_counter == 1);

	/* cleanup */
	cleanup_handle(SH_CONNECT);
	}

	/* Function semanage_ibendport_list */
	static void test_ibendport_list(void)
	{
	semanage_ibendport_t **records = NULL;
	unsigned int count = 42;
	char *name = NULL;
	semanage_context_t *con = NULL;

	/* setup */
	setup_handle(SH_CONNECT);

	/* test */
	CU_ASSERT(semanage_ibendport_list(sh, &records, &count) >= 0);

	CU_ASSERT_PTR_NOT_NULL_FATAL(records);
	assert(records);
	CU_ASSERT(count == IBENDPORT_COUNT);

	for (unsigned int i = 0; i < count; i++) {
	CU_ASSERT_PTR_NOT_NULL_FATAL(records[i]);
	CU_ASSERT(semanage_ibendport_get_ibdev_name(sh, records[i],
	&name) >= 0);
	con = semanage_ibendport_get_con(records[i]);
	CU_ASSERT_PTR_NOT_NULL_FATAL(con);
	free(name);
	}

	/* cleanup */
	for (unsigned int i = 0; i < count; i++)
	semanage_ibendport_free(records[i]);

	free(records);
	cleanup_handle(SH_CONNECT);
	}

	/* Function semanage_ibendport_modify_local, semanage_ibendport_del_local,
	* semanage_ibendport_query_local
	*/
	static void test_ibendport_modify_del_query_local(void)
	{
	semanage_ibendport_t *ibendport;
	semanage_ibendport_t *ibendport_local;
	semanage_ibendport_key_t *key = NULL;

	/* setup */
	setup_handle(SH_TRANS);
	ibendport = get_ibendport_nth(I_FIRST);
	CU_ASSERT(semanage_ibendport_key_extract(sh, ibendport, &key) >= 0);
	CU_ASSERT_PTR_NOT_NULL(key);

	/* test */
	CU_ASSERT(semanage_ibendport_modify_local(sh, key, ibendport) >= 0);

	/* write changes to file */
	helper_commit();
	helper_begin_transaction();

	CU_ASSERT(semanage_ibendport_query_local(sh, key,
	&ibendport_local) >= 0);
	CU_ASSERT_PTR_NOT_NULL_FATAL(ibendport_local);
	semanage_ibendport_free(ibendport_local);

	CU_ASSERT(semanage_ibendport_del_local(sh, key) >= 0);
	CU_ASSERT(semanage_ibendport_query_local(sh, key,
	&ibendport_local) < 0);

	/* cleanup */
	semanage_ibendport_key_free(key);
	semanage_ibendport_free(ibendport);
	cleanup_handle(SH_TRANS);
	}

	/* Function semanage_ibendport_exists_local */
	static void test_ibendport_exists_local(void)
	{
	semanage_ibendport_key_t *key1 = NULL;
	semanage_ibendport_key_t *key2 = NULL;
	int resp = 42;

	/* setup */
	setup_handle(SH_TRANS);
	add_local_ibendport(I_FIRST);
	key1 = get_ibendport_key_nth(I_FIRST);
	key2 = get_ibendport_key_nth(I_SECOND);

	/* test */
	CU_ASSERT(semanage_ibendport_exists_local(sh, key1, &resp) >= 0);
	CU_ASSERT(resp);

	CU_ASSERT(semanage_ibendport_exists_local(sh, key2, &resp) >= 0);
	CU_ASSERT(!resp);

	/* cleanup */
	CU_ASSERT(semanage_ibendport_del_local(sh, key1) >= 0);
	semanage_ibendport_key_free(key1);
	semanage_ibendport_key_free(key2);
	cleanup_handle(SH_TRANS);
	}

	/* Function semanage_ibendport_count_local */
	static void test_ibendport_count_local(void)
	{
	unsigned int count = 42;

	/* setup */
	setup_handle(SH_TRANS);

	/* test */
	CU_ASSERT(semanage_ibendport_count_local(sh, &count) >= 0);
	CU_ASSERT(count == 0);

	add_local_ibendport(I_FIRST);
	CU_ASSERT(semanage_ibendport_count_local(sh, &count) >= 0);
	CU_ASSERT(count == 1);

	add_local_ibendport(I_SECOND);
	CU_ASSERT(semanage_ibendport_count_local(sh, &count) >= 0);
	CU_ASSERT(count == 2);

	delete_local_ibendport(I_SECOND);
	CU_ASSERT(semanage_ibendport_count_local(sh, &count) >= 0);
	CU_ASSERT(count == 1);

	delete_local_ibendport(I_FIRST);
	CU_ASSERT(semanage_ibendport_count_local(sh, &count) >= 0);
	CU_ASSERT(count == 0);

	/* cleanup */
	cleanup_handle(SH_TRANS);
	}

	/* Function semanage_ibendport_iterate_local */
	unsigned int helper_ibendport_iterate_local_counter = 0;

	static int helper_ibendport_iterate_local(const semanage_ibendport_t *ibendport,
	void *fn_arg)
	{
	CU_ASSERT(fn_arg == (void *) 42);
	helper_ibendport_iterate_local_counter++;
	return 0;
	}

	static int helper_ibendport_iterate_local_error(const semanage_ibendport_t *ibendport,
	void *fn_arg)
	{
	CU_ASSERT(fn_arg == (void *) 42);
	helper_ibendport_iterate_local_counter++;
	return -1;
	}

	static int helper_ibendport_iterate_local_break(const semanage_ibendport_t *ibendport,
	void *fn_arg)
	{
	CU_ASSERT(fn_arg == (void *) 42);
	helper_ibendport_iterate_local_counter++;
	return 1;
	}

	static void test_ibendport_iterate_local(void)
	{
	/* setup */
	setup_handle(SH_TRANS);
	add_local_ibendport(I_FIRST);
	add_local_ibendport(I_SECOND);
	add_local_ibendport(I_THIRD);

	/* test */
	helper_ibendport_iterate_local_counter = 0;
	CU_ASSERT(semanage_ibendport_iterate_local(sh,
	helper_ibendport_iterate_local, (void *) 42) >= 0);
	CU_ASSERT(helper_ibendport_iterate_local_counter == 3);

	/* test function which returns error */
	helper_ibendport_iterate_local_counter = 0;
	CU_ASSERT(semanage_ibendport_iterate_local(sh,
	helper_ibendport_iterate_local_error, (void *) 42) < 0);
	CU_ASSERT(helper_ibendport_iterate_local_counter == 1);

	/* test function which requests break */
	helper_ibendport_iterate_local_counter = 0;
	CU_ASSERT(semanage_ibendport_iterate_local(sh,
	helper_ibendport_iterate_local_break, (void *) 42) >= 0);

	/* cleanup */
	delete_local_ibendport(I_FIRST);
	delete_local_ibendport(I_SECOND);
	delete_local_ibendport(I_THIRD);
	cleanup_handle(SH_TRANS);
	}

	/* Function semanage_ibendport_list_local */
	static void test_ibendport_list_local(void)
	{
	semanage_ibendport_t **records = NULL;
	unsigned int count = 42;
	char *name = NULL;
	semanage_context_t *con = NULL;

	/* setup */
	setup_handle(SH_TRANS);
	add_local_ibendport(I_FIRST);
	add_local_ibendport(I_SECOND);
	add_local_ibendport(I_THIRD);

	/* test */
	CU_ASSERT(semanage_ibendport_list_local(sh, &records, &count) >= 0);
	CU_ASSERT_PTR_NOT_NULL_FATAL(records);
	assert(records);
	CU_ASSERT(count == 3);

	for (unsigned int i = 0; i < count; i++) {
	CU_ASSERT_PTR_NOT_NULL_FATAL(records[i]);
	CU_ASSERT(semanage_ibendport_get_ibdev_name(sh, records[i],
	&name) >= 0);
	con = semanage_ibendport_get_con(records[i]);
	CU_ASSERT_PTR_NOT_NULL_FATAL(con);
	free(name);
	}

	/* cleanup */
	for (unsigned int i = 0; i < count; i++)
	semanage_ibendport_free(records[i]);

	free(records);
	delete_local_ibendport(I_FIRST);
	delete_local_ibendport(I_SECOND);
	delete_local_ibendport(I_THIRD);
	cleanup_handle(SH_TRANS);
	}