1use super::parser::PolicyCursor;
6use super::{
7 Array, Counted, Parse, PolicyValidationContext, Validate, ValidateArray, array_type,
8 array_type_validate_deref_both,
9};
10use crate::policy::error::{ParseError, ValidateError};
11
12use zerocopy::{FromBytes, Immutable, KnownLayout, Unaligned, little_endian as le};
13
14pub(super) const SELINUX_MAGIC: u32 = 0xf97cff8c;
15
16pub(super) const POLICYDB_STRING_MAX_LENGTH: u32 = 32;
17pub(super) const POLICYDB_SIGNATURE: &[u8] = b"SE Linux";
18
19pub(super) const POLICYDB_VERSION_MIN: u32 = 30;
20pub(super) const POLICYDB_VERSION_MAX: u32 = 33;
21
22pub(super) const CONFIG_MLS_FLAG: u32 = 1;
23pub(super) const CONFIG_HANDLE_UNKNOWN_REJECT_FLAG: u32 = 1 << 1;
24pub(super) const CONFIG_HANDLE_UNKNOWN_ALLOW_FLAG: u32 = 1 << 2;
25pub(super) const CONFIG_HANDLE_UNKNOWN_MASK: u32 =
26 CONFIG_HANDLE_UNKNOWN_REJECT_FLAG | CONFIG_HANDLE_UNKNOWN_ALLOW_FLAG;
27
28#[derive(Clone, Debug, KnownLayout, FromBytes, Immutable, PartialEq, Unaligned)]
29#[repr(C, packed)]
30pub(super) struct Magic(le::U32);
31
32impl Validate for Magic {
33 type Error = ValidateError;
34
35 fn validate(&self, _context: &mut PolicyValidationContext) -> Result<(), Self::Error> {
36 let found_magic = self.0.get();
37 if found_magic != SELINUX_MAGIC {
38 Err(ValidateError::InvalidMagic { found_magic })
39 } else {
40 Ok(())
41 }
42 }
43}
44
45array_type!(Signature, SignatureMetadata, Vec<u8>);
46
47array_type_validate_deref_both!(Signature);
48
49impl ValidateArray<SignatureMetadata, u8> for Signature {
50 type Error = ValidateError;
51
52 fn validate_array(
53 _context: &mut PolicyValidationContext,
54 _metadata: &SignatureMetadata,
55 items: &[u8],
56 ) -> Result<(), Self::Error> {
57 if items != POLICYDB_SIGNATURE {
58 Err(ValidateError::InvalidSignature { found_signature: items.to_owned() })
59 } else {
60 Ok(())
61 }
62 }
63}
64
65#[derive(Clone, Debug, KnownLayout, FromBytes, Immutable, PartialEq, Unaligned)]
66#[repr(C, packed)]
67pub(super) struct SignatureMetadata(le::U32);
68
69impl Validate for SignatureMetadata {
70 type Error = ValidateError;
71
72 fn validate(&self, _context: &mut PolicyValidationContext) -> Result<(), Self::Error> {
74 let found_length = self.0.get();
75 if found_length > POLICYDB_STRING_MAX_LENGTH {
76 Err(ValidateError::InvalidSignatureLength { found_length })
77 } else {
78 Ok(())
79 }
80 }
81}
82
83impl Counted for SignatureMetadata {
84 fn count(&self) -> u32 {
85 self.0.get()
86 }
87}
88
89#[derive(Clone, Debug, KnownLayout, FromBytes, Immutable, PartialEq, Unaligned)]
90#[repr(C, packed)]
91pub(super) struct PolicyVersion(le::U32);
92
93impl PolicyVersion {
94 pub fn policy_version(&self) -> u32 {
95 self.0.get()
96 }
97}
98
99impl Validate for PolicyVersion {
100 type Error = ValidateError;
101
102 fn validate(&self, _context: &mut PolicyValidationContext) -> Result<(), Self::Error> {
103 let found_policy_version = self.0.get();
104 if found_policy_version < POLICYDB_VERSION_MIN
105 || found_policy_version > POLICYDB_VERSION_MAX
106 {
107 Err(ValidateError::InvalidPolicyVersion { found_policy_version })
108 } else {
109 Ok(())
110 }
111 }
112}
113
114#[derive(Debug)]
115pub(super) struct Config {
116 handle_unknown: HandleUnknown,
117
118 #[allow(dead_code)]
119 config: le::U32,
120}
121
122impl Config {
123 pub fn handle_unknown(&self) -> HandleUnknown {
124 self.handle_unknown
125 }
126}
127
128impl Parse for Config {
129 type Error = ParseError;
130
131 fn parse(bytes: PolicyCursor) -> Result<(Self, PolicyCursor), Self::Error> {
132 let num_bytes = bytes.len();
133 let (config, tail) =
134 PolicyCursor::parse::<le::U32>(bytes).ok_or(ParseError::MissingData {
135 type_name: "Config",
136 type_size: std::mem::size_of::<le::U32>(),
137 num_bytes,
138 })?;
139
140 let found_config = config.get();
141 if found_config & CONFIG_MLS_FLAG == 0 {
142 return Err(ParseError::ConfigMissingMlsFlag { found_config });
143 }
144 let handle_unknown = try_handle_unknown_fom_config(found_config)?;
145
146 Ok((Self { handle_unknown, config }, tail))
147 }
148}
149
150impl Validate for Config {
151 type Error = anyhow::Error;
152
153 fn validate(&self, _context: &mut PolicyValidationContext) -> Result<(), Self::Error> {
156 Ok(())
157 }
158}
159
160#[derive(Copy, Clone, Debug, PartialEq)]
161pub enum HandleUnknown {
162 Deny,
163 Reject,
164 Allow,
165}
166
167fn try_handle_unknown_fom_config(config: u32) -> Result<HandleUnknown, ParseError> {
168 match config & CONFIG_HANDLE_UNKNOWN_MASK {
169 CONFIG_HANDLE_UNKNOWN_ALLOW_FLAG => Ok(HandleUnknown::Allow),
170 CONFIG_HANDLE_UNKNOWN_REJECT_FLAG => Ok(HandleUnknown::Reject),
171 0 => Ok(HandleUnknown::Deny),
172 _ => Err(ParseError::InvalidHandleUnknownConfigurationBits {
173 masked_bits: (config & CONFIG_HANDLE_UNKNOWN_MASK),
174 }),
175 }
176}
177
178#[derive(Clone, Debug, KnownLayout, FromBytes, Immutable, PartialEq, Unaligned)]
179#[repr(C, packed)]
180pub(super) struct Counts {
181 symbols_count: le::U32,
182 object_context_count: le::U32,
183}
184
185impl Validate for Counts {
186 type Error = anyhow::Error;
187
188 fn validate(&self, _context: &mut PolicyValidationContext) -> Result<(), Self::Error> {
190 Ok(())
191 }
192}
193
194#[cfg(test)]
195mod tests {
196 use super::*;
197
198 use crate::policy::parser::PolicyCursor;
199 use crate::policy::testing::{as_parse_error, as_validate_error};
200 use std::sync::Arc;
201
202 macro_rules! validate_test {
203 ($parse_output:ident, $data:expr, $result:tt, $check_impl:block) => {{
204 let data = Arc::new($data);
205 let mut context = crate::policy::PolicyValidationContext { data: data.clone() };
206 fn check_by_value(
207 $result: Result<(), <$parse_output as crate::policy::Validate>::Error>,
208 ) {
209 $check_impl
210 }
211
212 let (by_value_parsed, _tail) = $parse_output::parse(PolicyCursor::new(data.clone()))
213 .expect("successful parse for validate test");
214 let by_value_result = by_value_parsed.validate(&mut context);
215 check_by_value(by_value_result);
216 }};
217 }
218
219 #[test]
221 fn no_magic() {
222 let mut bytes = [SELINUX_MAGIC.to_le_bytes().as_slice()].concat();
223 bytes.pop();
225 let data = Arc::new(bytes);
226 assert_eq!(None, PolicyCursor::parse::<Magic>(PolicyCursor::new(data)),);
227 }
228
229 #[test]
230 fn invalid_magic() {
231 let mut bytes = [SELINUX_MAGIC.to_le_bytes().as_slice()].concat();
232 bytes[0] = bytes[0] + 1;
234 let bytes = bytes;
235 let expected_invalid_magic =
236 u32::from_le_bytes(bytes.clone().as_slice().try_into().unwrap());
237
238 let data = Arc::new(bytes);
239 let mut context = crate::policy::PolicyValidationContext { data: data.clone() };
240 let (magic, tail) =
241 PolicyCursor::parse::<Magic>(PolicyCursor::new(data.clone())).expect("magic");
242 assert_eq!(0, tail.len());
243 assert_eq!(
244 Err(ValidateError::InvalidMagic { found_magic: expected_invalid_magic }),
245 magic.validate(&mut context)
246 );
247 }
248
249 #[test]
250 fn invalid_signature_length() {
251 const INVALID_SIGNATURE_LENGTH: u32 = POLICYDB_STRING_MAX_LENGTH + 1;
252 let bytes: Vec<u8> = [
253 INVALID_SIGNATURE_LENGTH.to_le_bytes().as_slice(),
254 [42u8; INVALID_SIGNATURE_LENGTH as usize].as_slice(),
255 ]
256 .concat();
257
258 validate_test!(Signature, bytes, result, {
259 assert_eq!(
260 Some(ValidateError::InvalidSignatureLength {
261 found_length: INVALID_SIGNATURE_LENGTH
262 }),
263 result.err().map(as_validate_error),
264 );
265 });
266 }
267
268 #[test]
269 fn missing_signature() {
270 let bytes = [(1 as u32).to_le_bytes().as_slice()].concat();
271 match Signature::parse(PolicyCursor::new(Arc::new(bytes))).err().map(as_parse_error) {
272 Some(ParseError::MissingData { type_name: "u8", type_size: 1, num_bytes: 0 }) => {}
273 parse_err => {
274 assert!(false, "Expected Some(MissingData...), but got {:?}", parse_err);
275 }
276 }
277 }
278
279 #[test]
280 fn invalid_signature() {
281 const INVALID_SIGNATURE: &[u8] = b"TE Linux";
283
284 let bytes =
285 [(INVALID_SIGNATURE.len() as u32).to_le_bytes().as_slice(), INVALID_SIGNATURE].concat();
286
287 validate_test!(Signature, bytes, result, {
288 assert_eq!(
289 Some(ValidateError::InvalidSignature {
290 found_signature: INVALID_SIGNATURE.to_owned()
291 }),
292 result.err().map(as_validate_error),
293 );
294 });
295 }
296
297 #[test]
298 fn invalid_policy_version() {
299 let bytes = [(POLICYDB_VERSION_MIN - 1).to_le_bytes().as_slice()].concat();
300 let data = Arc::new(bytes);
301 let mut context = crate::policy::PolicyValidationContext { data: data.clone() };
302 let (policy_version, tail) =
303 PolicyCursor::parse::<PolicyVersion>(PolicyCursor::new(data.clone())).expect("magic");
304 assert_eq!(0, tail.len());
305 assert_eq!(
306 Err(ValidateError::InvalidPolicyVersion {
307 found_policy_version: POLICYDB_VERSION_MIN - 1
308 }),
309 policy_version.validate(&mut context)
310 );
311
312 let bytes = [(POLICYDB_VERSION_MAX + 1).to_le_bytes().as_slice()].concat();
313 let data = Arc::new(bytes);
314 let mut context = crate::policy::PolicyValidationContext { data: data.clone() };
315 let (policy_version, tail) =
316 PolicyCursor::parse::<PolicyVersion>(PolicyCursor::new(data.clone())).expect("magic");
317 assert_eq!(0, tail.len());
318 assert_eq!(
319 Err(ValidateError::InvalidPolicyVersion {
320 found_policy_version: POLICYDB_VERSION_MAX + 1
321 }),
322 policy_version.validate(&mut context)
323 );
324 }
325
326 #[test]
327 fn config_missing_mls_flag() {
328 let bytes = [(!CONFIG_MLS_FLAG).to_le_bytes().as_slice()].concat();
329 match Config::parse(PolicyCursor::new(Arc::new(bytes))).err() {
330 Some(ParseError::ConfigMissingMlsFlag { .. }) => {}
331 parse_err => {
332 assert!(false, "Expected Some(ConfigMissingMlsFlag...), but got {:?}", parse_err);
333 }
334 }
335 }
336
337 #[test]
338 fn invalid_handle_unknown() {
339 let bytes = [(CONFIG_MLS_FLAG
340 | CONFIG_HANDLE_UNKNOWN_ALLOW_FLAG
341 | CONFIG_HANDLE_UNKNOWN_REJECT_FLAG)
342 .to_le_bytes()
343 .as_slice()]
344 .concat();
345 assert_eq!(
346 Some(ParseError::InvalidHandleUnknownConfigurationBits {
347 masked_bits: CONFIG_HANDLE_UNKNOWN_ALLOW_FLAG | CONFIG_HANDLE_UNKNOWN_REJECT_FLAG
348 }),
349 Config::parse(PolicyCursor::new(Arc::new(bytes))).err()
350 );
351 }
352}