android_system_sepolicy/tests/fc_sort_test.py
Thiébaud Weksteen f24b457dd0 Migrate tests/ to Python 3
In general, it appears that libselinux and libsepol interpret paths and
contexts as bytes. For instance, selabel_file(5) mentions about the path
field of file_contexts:

  Strings representing paths are processed as bytes (as opposed to
  Unicode), meaning that non-ASCII characters are not matched
  by a single wildcard.

libsepol also uses primitives such as strchr[1], which explicitly
operate at the byte level (see strchr(3)). However, practically, Android
paths and contexts all uses ASCII characters.

Use the str type (i.e., Unicode) for all Python code to avoid a larger
refactoring. Ensure we convert to bytes for inputs and outputs of
libsepolwrap.so. The encoding "ascii" is used, which will raise an error
should a context or type contain non-ASCII characters.

Update headers to match development/docs/copyright-templates.

[1] https://cs.android.com/android/platform/superproject/+/master:external/selinux/libsepol/src/context_record.c;l=224;drc=454466e2e49fd99f36db78396e604962b8682cb4

Bug: 200119288
Test: lunch aosp_bramble-userdebug && m
Test: atest --host fc_sort_test
Test: manually run searchpolicy
Change-Id: I72d41a35f90b2d4112e481cd8d7408764a6c8132
2021-12-01 21:45:13 +00:00

60 lines
2.1 KiB
Python

# Copyright 2021 The Android Open Source Project
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
import unittest
import fc_sort
class FcSortTest(unittest.TestCase):
def testGetStemLen(self):
self.assertEqual(fc_sort.get_stem_len("/data"), 5)
self.assertEqual(fc_sort.get_stem_len("/data/system"), 12)
self.assertEqual(fc_sort.get_stem_len("/data/(system)?"), 6)
def testIsMeta(self):
self.assertEqual(fc_sort.is_meta("/data"), False)
self.assertEqual(fc_sort.is_meta("/data$"), True)
self.assertEqual(fc_sort.is_meta(r"\$data"), False)
def testLesserThan(self):
n1 = fc_sort.FileContextsNode.create("/data u:object_r:rootfs:s0")
# shorter stem_len
n2 = fc_sort.FileContextsNode.create("/d u:object_r:rootfs:s0")
# is meta
n3 = fc_sort.FileContextsNode.create("/data/l(/.*)? u:object_r:log:s0")
# with file_type
n4 = fc_sort.FileContextsNode.create("/data -- u:object_r:rootfs:s0")
contexts = [n1, n2, n3, n4]
contexts.sort()
self.assertEqual(contexts, [n3, n2, n1, n4])
def testReadFileContexts(self):
content = """# comment
/ u:object_r:rootfs:s0
# another comment
/adb_keys u:object_r:adb_keys_file:s0
"""
fcs = fc_sort.read_file_contexts(content.splitlines())
self.assertEqual(len(fcs), 2)
self.assertEqual(fcs[0].path, "/")
self.assertEqual(fcs[0].type, "rootfs")
self.assertEqual(fcs[1].path, "/adb_keys")
self.assertEqual(fcs[1].type, "adb_keys_file")
if __name__ == '__main__':
unittest.main(verbosity=2)