File: infoflow-tests.c

package info (click to toggle)
setools 3.3.7-3
  • links: PTS, VCS
  • area: main
  • in suites: wheezy
  • size: 9,840 kB
  • sloc: ansic: 82,247; tcl: 13,145; cpp: 4,885; makefile: 1,603; yacc: 779; lex: 296; python: 57; sh: 50
file content (127 lines) | stat: -rw-r--r-- 3,778 bytes parent folder | download | duplicates (4)
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
/**
 *  @file
 *
 *  Test the information flow analysis code.
 *
 *  @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/infoflow-analysis.h>
#include <apol/perm-map.h>
#include <apol/policy.h>
#include <apol/policy-path.h>
#include <stdbool.h>
#include <string.h>

#define BIG_POLICY TEST_POLICIES "/snapshots/fc4_targeted.policy.conf"
#define PERMMAP TOP_SRCDIR "/apol/perm_maps/apol_perm_mapping_ver19"

static apol_policy_t *p = NULL;

static void infoflow_direct_overview(void)
{
	apol_infoflow_analysis_t *ia = apol_infoflow_analysis_create();
	CU_ASSERT_PTR_NOT_NULL_FATAL(ia);
	int retval;
	retval = apol_infoflow_analysis_set_mode(p, ia, APOL_INFOFLOW_MODE_DIRECT);
	CU_ASSERT(retval == 0);
	retval = apol_infoflow_analysis_set_dir(p, ia, APOL_INFOFLOW_IN);
	CU_ASSERT(retval == 0);
	retval = apol_infoflow_analysis_set_type(p, ia, "agp_device_t");
	CU_ASSERT(retval == 0);

	apol_vector_t *v = NULL;
	apol_infoflow_graph_t *g = NULL;
	// no permmap loaded, so analysis run will abort with error
	retval = apol_infoflow_analysis_do(p, ia, &v, &g);
	CU_ASSERT(retval < 0);

	retval = apol_policy_open_permmap(p, PERMMAP);
	CU_ASSERT(retval == 0);

	retval = apol_infoflow_analysis_do(p, ia, &v, &g);
	CU_ASSERT(retval == 0);
	CU_ASSERT_PTR_NOT_NULL(v);
	CU_ASSERT(apol_vector_get_size(v) > 0);
	CU_ASSERT_PTR_NOT_NULL(g);

	apol_infoflow_analysis_destroy(&ia);
	apol_vector_destroy(&v);
	apol_infoflow_graph_destroy(&g);
}

static void infoflow_trans_overview(void)
{
	apol_infoflow_analysis_t *ia = apol_infoflow_analysis_create();
	CU_ASSERT_PTR_NOT_NULL_FATAL(ia);
	int retval;
	retval = apol_infoflow_analysis_set_mode(p, ia, APOL_INFOFLOW_MODE_DIRECT);
	CU_ASSERT(retval == 0);
	retval = apol_infoflow_analysis_set_dir(p, ia, APOL_INFOFLOW_IN);
	CU_ASSERT(retval == 0);
	retval = apol_infoflow_analysis_set_type(p, ia, "local_login_t");
	CU_ASSERT(retval == 0);

	apol_vector_t *v = NULL;
	apol_infoflow_graph_t *g = NULL;
	// permmap was loaded by infoflow_direct_overview()
	retval = apol_infoflow_analysis_do(p, ia, &v, &g);
	CU_ASSERT(retval == 0);
	CU_ASSERT_PTR_NOT_NULL(v);
	CU_ASSERT(apol_vector_get_size(v) > 0);
	CU_ASSERT_PTR_NOT_NULL(g);

	apol_infoflow_analysis_destroy(&ia);
	apol_vector_destroy(&v);
	apol_infoflow_graph_destroy(&g);
}

CU_TestInfo infoflow_tests[] = {
	{"infoflow direct overview", infoflow_direct_overview}
	,
	{"infoflow trans overview", infoflow_trans_overview}
	,
	CU_TEST_INFO_NULL
};

int infoflow_init()
{
	apol_policy_path_t *ppath = apol_policy_path_create(APOL_POLICY_PATH_TYPE_MONOLITHIC, BIG_POLICY, NULL);
	if (ppath == NULL) {
		return 1;
	}

	if ((p = apol_policy_create_from_policy_path(ppath, QPOL_POLICY_OPTION_NO_NEVERALLOWS, NULL, NULL)) == NULL) {
		apol_policy_path_destroy(&ppath);
		return 1;
	}
	apol_policy_path_destroy(&ppath);

	return 0;
}

int infoflow_cleanup()
{
	apol_policy_destroy(&p);
	return 0;
}