Fix the mcu test

Author: cla7aye15I4nd

tests/test_mcu.py

@@ -9,13 +9,13 @@
 
 from qiling.core import Qiling
 from qiling.const import QL_VERBOSE
-from qiling.extensions.soc.stm32f4 import stm32f407_env, stm32f411_env
-from qiling.extensions.soc.gd32vf1 import gd32vf103_env
+from qiling.extensions.soc.stm32f4 import stm32f407, stm32f411
+from qiling.extensions.soc.gd32vf1 import gd32vf103
 
 class MCUTest(unittest.TestCase):
     def test_mcu_led_stm32f411(self):
         ql = Qiling(["../examples/rootfs/mcu/stm32f411/rand_blink.hex"],                    
-                    archtype="cortex_m", env=stm32f411_env, verbose=QL_VERBOSE.DISASM)
+                    archtype="cortex_m", env=stm32f411, verbose=QL_VERBOSE.DISASM)
 
         # Set verbose=QL_VERBOSE.DEFAULT to find warning
         ql.run(count=1000)
@@ -24,7 +24,7 @@ def test_mcu_led_stm32f411(self):
 
     def test_mcu_usart_output_stm32f411(self):
         ql = Qiling(["../examples/rootfs/mcu/stm32f411/hello_usart.hex"],                    
-                    archtype="cortex_m", env=stm32f411_env, verbose=QL_VERBOSE.DEFAULT)        
+                    archtype="cortex_m", env=stm32f411, verbose=QL_VERBOSE.DEFAULT)        
 
         ql.hw.create('usart2')
         ql.hw.create('rcc')
@@ -38,7 +38,7 @@ def test_mcu_usart_output_stm32f411(self):
 
     def test_mcu_usart_input_stm32f411(self):
         ql = Qiling(["../examples/rootfs/mcu/stm32f411/md5_server.hex"],                    
-            archtype="cortex_m", env=stm32f411_env, verbose=QL_VERBOSE.OFF)
+            archtype="cortex_m", env=stm32f411, verbose=QL_VERBOSE.OFF)
 
         ql.hw.create('usart2')
         ql.hw.create('rcc')
@@ -59,7 +59,7 @@ def test_mcu_usart_input_stm32f411(self):
 
     def test_mcu_patch_stm32f411(self):
         ql = Qiling(["../examples/rootfs/mcu/stm32f411/patch_test.hex"],                    
-                    archtype="cortex_m", env=stm32f411_env, verbose=QL_VERBOSE.DEFAULT)
+                    archtype="cortex_m", env=stm32f411, verbose=QL_VERBOSE.DEFAULT)
 
         ql.hw.create('usart2')
         ql.hw.create('rcc')
@@ -72,7 +72,7 @@ def test_mcu_patch_stm32f411(self):
 
     def test_mcu_freertos_stm32f411(self):
         ql = Qiling(["../examples/rootfs/mcu/stm32f411/os-demo.elf"],
-            archtype="cortex_m", env=stm32f411_env, verbose=QL_VERBOSE.DEBUG)
+            archtype="cortex_m", env=stm32f411, verbose=QL_VERBOSE.DEBUG)
 
         ql.hw.create('usart2')
         ql.hw.create('rcc')
@@ -94,7 +94,7 @@ def counter():
 
     def test_mcu_dma_stm32f411(self):
         ql = Qiling(["../examples/rootfs/mcu/stm32f411/dma-clock.elf"],                    
-            archtype="cortex_m", env=stm32f411_env, verbose=QL_VERBOSE.DEBUG)
+            archtype="cortex_m", env=stm32f411, verbose=QL_VERBOSE.DEBUG)
 
         ql.hw.create('usart2')
         ql.hw.create('dma1')
@@ -112,7 +112,7 @@ def test_mcu_dma_stm32f411(self):
 
     def test_mcu_i2c_stm32f411(self):
         ql = Qiling(["../examples/rootfs/mcu/stm32f411/i2c-lcd.bin", 0x8000000],
-            archtype="cortex_m", env=stm32f411_env, verbose=QL_VERBOSE.DEBUG)
+            archtype="cortex_m", env=stm32f411, verbose=QL_VERBOSE.DEBUG)
 
         ql.hw.create('i2c1')
         ql.hw.create('rcc')
@@ -144,7 +144,7 @@ def step(self):
 
     def test_mcu_spi_stm32f411(self):
         ql = Qiling(["../examples/rootfs/mcu/stm32f411/spi-test.bin", 0x8000000],
-            archtype="cortex_m", env=stm32f411_env, verbose=QL_VERBOSE.DEBUG)
+            archtype="cortex_m", env=stm32f411, verbose=QL_VERBOSE.DEBUG)
 
         ql.hw.create('spi1')
         ql.hw.create('rcc')
@@ -158,7 +158,7 @@ def test_mcu_spi_stm32f411(self):
 
     def test_mcu_led_rust_stm32f411(self):
         ql = Qiling(["../examples/rootfs/mcu/stm32f411/led-rust.hex"],
-                    archtype="cortex_m", env=gd32vf103_env, profile="profiles/stm32f411.yml", verbose=QL_VERBOSE.DEBUG)
+                    archtype="cortex_m", env=gd32vf103, profile="profiles/stm32f411.yml", verbose=QL_VERBOSE.DEBUG)
 
         count = 0
         def counter():
@@ -175,7 +175,7 @@ def counter():
 
     def test_mcu_uart_rust_stm32f411(self): 
         ql = Qiling(["../examples/rootfs/mcu/stm32f411/uart-rust.hex"],
-                    archtype="cortex_m", env=stm32f411_env, verbose=QL_VERBOSE.DEBUG)
+                    archtype="cortex_m", env=stm32f411, verbose=QL_VERBOSE.DEBUG)
 
         ## cover env by profiles
 
@@ -192,7 +192,7 @@ def test_mcu_uart_rust_stm32f411(self):
     def test_mcu_hacklock_stm32f407(self):
         def crack(passwd):
             ql = Qiling(["../examples/rootfs/mcu/stm32f407/backdoorlock.hex"],                    
-                                archtype="cortex_m", env=stm32f407_env, verbose=QL_VERBOSE.OFF)
+                                archtype="cortex_m", env=stm32f407, verbose=QL_VERBOSE.OFF)
 
             ql.hw.create('spi2')
             ql.hw.create('gpioe')
@@ -221,7 +221,7 @@ def crack(passwd):
 
     def test_mcu_blink_gd32vf103(self):
         ql = Qiling(['../examples/rootfs/mcu/gd32vf103/blink.hex'], archtype="riscv64", 
-                    env=gd32vf103_env, verbose=QL_VERBOSE.DEFAULT)
+                    env=gd32vf103, verbose=QL_VERBOSE.DEFAULT)
 
         ql.hw.create('rcu')
         ql.hw.create('gpioa')