tzh
2024-08-22 c7d0944258c7d0943aa7b2211498fd612971ce27
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
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
#include <netinet/in.h>
#include <stdlib.h>
#include <inttypes.h>
 
#include "debug.h"
#include "context.h"
#include "handle.h"
 
#include <sepol/ibpkey_record.h>
#include <sepol/policydb/policydb.h>
#include "ibpkey_internal.h"
 
/* Create a low level ibpkey structure from
 * a high level representation
 */
static int ibpkey_from_record(sepol_handle_t *handle,
                 const policydb_t *policydb,
                 ocontext_t **ibpkey, const sepol_ibpkey_t *data)
{
   ocontext_t *tmp_ibpkey = NULL;
   context_struct_t *tmp_con = NULL;
   char *subnet_prefix_buf = NULL;
   int low = sepol_ibpkey_get_low(data);
   int high = sepol_ibpkey_get_high(data);
 
   tmp_ibpkey = (ocontext_t *)calloc(1, sizeof(*tmp_ibpkey));
   if (!tmp_ibpkey)
       goto omem;
 
   tmp_ibpkey->u.ibpkey.subnet_prefix = sepol_ibpkey_get_subnet_prefix_bytes(data);
 
   /* Pkey range */
   tmp_ibpkey->u.ibpkey.low_pkey = low;
   tmp_ibpkey->u.ibpkey.high_pkey = high;
   if (tmp_ibpkey->u.ibpkey.low_pkey > tmp_ibpkey->u.ibpkey.high_pkey) {
       ERR(handle, "low ibpkey %d exceeds high ibpkey %d",
           tmp_ibpkey->u.ibpkey.low_pkey, tmp_ibpkey->u.ibpkey.high_pkey);
       goto err;
   }
 
   /* Context */
   if (context_from_record(handle, policydb, &tmp_con,
               sepol_ibpkey_get_con(data)) < 0)
       goto err;
   context_cpy(&tmp_ibpkey->context[0], tmp_con);
   context_destroy(tmp_con);
   free(tmp_con);
   tmp_con = NULL;
 
   *ibpkey = tmp_ibpkey;
   return STATUS_SUCCESS;
 
omem:
   ERR(handle, "out of memory");
 
err:
   if (tmp_ibpkey) {
       context_destroy(&tmp_ibpkey->context[0]);
       free(tmp_ibpkey);
   }
   context_destroy(tmp_con);
   free(tmp_con);
   free(subnet_prefix_buf);
   ERR(handle, "could not create ibpkey structure");
   return STATUS_ERR;
}
 
static int ibpkey_to_record(sepol_handle_t *handle,
               const policydb_t *policydb,
               ocontext_t *ibpkey, sepol_ibpkey_t **record)
{
   context_struct_t *con = &ibpkey->context[0];
   sepol_context_t *tmp_con = NULL;
   sepol_ibpkey_t *tmp_record = NULL;
 
   if (sepol_ibpkey_create(handle, &tmp_record) < 0)
       goto err;
 
   sepol_ibpkey_set_subnet_prefix_bytes(tmp_record,
                        ibpkey->u.ibpkey.subnet_prefix);
 
   sepol_ibpkey_set_range(tmp_record, ibpkey->u.ibpkey.low_pkey,
                  ibpkey->u.ibpkey.high_pkey);
 
   if (context_to_record(handle, policydb, con, &tmp_con) < 0)
       goto err;
 
   if (sepol_ibpkey_set_con(handle, tmp_record, tmp_con) < 0)
       goto err;
 
   sepol_context_free(tmp_con);
   *record = tmp_record;
   return STATUS_SUCCESS;
 
err:
   ERR(handle, "could not convert ibpkey to record");
   sepol_context_free(tmp_con);
   sepol_ibpkey_free(tmp_record);
   return STATUS_ERR;
}
 
/* Return the number of ibpkeys */
extern int sepol_ibpkey_count(sepol_handle_t *handle __attribute__ ((unused)),
                 const sepol_policydb_t *p, unsigned int *response)
{
   unsigned int count = 0;
   ocontext_t *c, *head;
   const policydb_t *policydb = &p->p;
 
   head = policydb->ocontexts[OCON_IBPKEY];
   for (c = head; c; c = c->next)
       count++;
 
   *response = count;
 
   return STATUS_SUCCESS;
}
 
/* Check if a ibpkey exists */
int sepol_ibpkey_exists(sepol_handle_t *handle __attribute__ ((unused)),
           const sepol_policydb_t *p,
           const sepol_ibpkey_key_t *key, int *response)
{
   const policydb_t *policydb = &p->p;
   ocontext_t *c, *head;
   int low, high;
   uint64_t subnet_prefix;
 
   sepol_ibpkey_key_unpack(key, &subnet_prefix, &low, &high);
 
   head = policydb->ocontexts[OCON_IBPKEY];
   for (c = head; c; c = c->next) {
       uint64_t subnet_prefix2 = c->u.ibpkey.subnet_prefix;
       uint16_t low2 = c->u.ibpkey.low_pkey;
       uint16_t high2 = c->u.ibpkey.high_pkey;
 
       if (low2 == low &&
           high2 == high &&
           subnet_prefix == subnet_prefix2) {
           *response = 1;
           return STATUS_SUCCESS;
       }
   }
 
   *response = 0;
   return STATUS_SUCCESS;
}
 
/* Query a ibpkey */
int sepol_ibpkey_query(sepol_handle_t *handle,
              const sepol_policydb_t *p,
              const sepol_ibpkey_key_t *key, sepol_ibpkey_t **response)
{
   const policydb_t *policydb = &p->p;
   ocontext_t *c, *head;
   int low, high;
   uint64_t subnet_prefix;
 
   sepol_ibpkey_key_unpack(key, &subnet_prefix, &low, &high);
 
   head = policydb->ocontexts[OCON_IBPKEY];
   for (c = head; c; c = c->next) {
       uint64_t subnet_prefix2 = c->u.ibpkey.subnet_prefix;
       int low2 = c->u.ibpkey.low_pkey;
       int high2 = c->u.ibpkey.high_pkey;
 
       if (low2 == low &&
           high2 == high &&
           subnet_prefix == subnet_prefix2) {
           if (ibpkey_to_record(handle, policydb, c, response) < 0)
               goto err;
           return STATUS_SUCCESS;
       }
   }
 
   *response = NULL;
   return STATUS_SUCCESS;
 
err:
   ERR(handle, "could not query ibpkey subnet prefix: %#" PRIx64 " range %u - %u exists",
       subnet_prefix, low, high);
   return STATUS_ERR;
}
 
/* Load a ibpkey into policy */
int sepol_ibpkey_modify(sepol_handle_t *handle,
           sepol_policydb_t *p,
           const sepol_ibpkey_key_t *key, const sepol_ibpkey_t *data)
{
   policydb_t *policydb = &p->p;
   ocontext_t *ibpkey = NULL;
   int low, high;
   uint64_t subnet_prefix;
 
   sepol_ibpkey_key_unpack(key, &subnet_prefix, &low, &high);
 
   if (ibpkey_from_record(handle, policydb, &ibpkey, data) < 0)
       goto err;
 
   /* Attach to context list */
   ibpkey->next = policydb->ocontexts[OCON_IBPKEY];
   policydb->ocontexts[OCON_IBPKEY] = ibpkey;
 
   return STATUS_SUCCESS;
 
err:
   ERR(handle, "could not load ibpkey subnet prefix: %#" PRIx64 " range %u - %u exists",
       subnet_prefix, low, high);
   if (ibpkey) {
       context_destroy(&ibpkey->context[0]);
       free(ibpkey);
   }
   return STATUS_ERR;
}
 
int sepol_ibpkey_iterate(sepol_handle_t *handle,
            const sepol_policydb_t *p,
            int (*fn)(const sepol_ibpkey_t *ibpkey,
                  void *fn_arg), void *arg)
{
   const policydb_t *policydb = &p->p;
   ocontext_t *c, *head;
   sepol_ibpkey_t *ibpkey = NULL;
 
   head = policydb->ocontexts[OCON_IBPKEY];
   for (c = head; c; c = c->next) {
       int status;
 
       if (ibpkey_to_record(handle, policydb, c, &ibpkey) < 0)
           goto err;
 
       /* Invoke handler */
       status = fn(ibpkey, arg);
       if (status < 0)
           goto err;
 
       sepol_ibpkey_free(ibpkey);
       ibpkey = NULL;
 
       /* Handler requested exit */
       if (status > 0)
           break;
   }
 
   return STATUS_SUCCESS;
 
err:
   ERR(handle, "could not iterate over ibpkeys");
   sepol_ibpkey_free(ibpkey);
   return STATUS_ERR;
}