1#!/usr/bin/env python3
2# -*- coding: utf-8 -*-
3#
4# Copyright (c) 2024 Huawei Device Co., Ltd.
5# Licensed under the Apache License, Version 2.0 (the "License");
6# you may not use this file except in compliance with the License.
7# You may obtain a copy of the License at
8#
9#     http://www.apache.org/licenses/LICENSE-2.0
10#
11# Unless required by applicable law or agreed to in writing, software
12# distributed under the License is distributed on an "AS IS" BASIS,
13# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14# See the License for the specific language governing permissions and
15# limitations under the License.
16
17from devicetest.core.test_case import TestCase, Step
18import re
19
20# @tc.number: C-ALL-SECURITY-0304
21# @tc.name: testRealRandomGenerate
22# @tc.desc: 【C-ALL-SECURITY-0304】必须支持硬件真随机数生成，满足 FIPS140-2 随机测试标准。
23class testRealRandomGenerate(TestCase):
24
25    def __init__(self, controllers):
26        self.TAG = self.__class__.__name__
27        super().__init__(self.TAG, controllers)
28
29    def setup(self):
30        Step("Setup")
31
32    def process(self):
33        Step("Process")
34        randomFile = self.device1.execute_shell_command("ls  /dev/random | wc -l").strip()
35        assert randomFile == '1'
36        urandomFile = self.device1.execute_shell_command("ls  /dev/urandom | wc -l").strip()
37        assert urandomFile == '1'
38        pattern = r'[a-z0-9]{10}'
39        try:
40            randomNumber = self.device1.execute_shell_command("cat /dev/random | head -n 10 | md5sum | head -c 10", timeout = 1000).strip()
41        except:
42            randomNumber = ''
43        if not re.match(pattern, randomNumber):
44            randomNumber = self.device1.execute_shell_command("cat /dev/urandom | head -n 10 | md5sum | head -c 10").strip()
45        assert re.match(pattern, randomNumber)
46
47    def teardown(self):
48        Step("Teardown")