Fix the riscv test

Author: cla7aye15I4nd

tests/test_riscv.py

@@ -7,38 +7,42 @@
 sys.path.append("..")
 
 from qiling.core import Qiling
-from qiling.const import QL_VERBOSE
+from qiling.const import QL_VERBOSE, QL_INTERCEPT
 from qiling.extensions.pipe import SimpleOutStream
 
 class RISCVTest(unittest.TestCase):
     def test_riscv32_hello_linux(self):
         stdout = SimpleOutStream(1)
         ql = Qiling(['../examples/rootfs/riscv32_linux/bin/hello'], '../examples/rootfs/riscv32_linux/', 
-                    verbose=QL_VERBOSE.DEBUG, stdout=stdout)
-
-        ql.run(end=0x66112) # avoid close 0, 1, 2
+                    verbose=QL_VERBOSE.DEFAULT, stdout=stdout)
+        
+        def close(ql, fd):
+            return 0
+        ql.set_syscall("close", close, QL_INTERCEPT.CALL)
+        ql.run()
         self.assertTrue(stdout.read() == b'Hello, World!\n')
 
         del ql
 
     def test_riscv64_hello_linux(self):
         stdout = SimpleOutStream(1)
         ql = Qiling(['../examples/rootfs/riscv64_linux/bin/hello'], '../examples/rootfs/riscv64_linux/', 
-                    verbose=QL_VERBOSE.DEBUG, stdout=stdout)
-
+                    verbose=QL_VERBOSE.DEFAULT, stdout=stdout)
 
-        ql.run(end=0x1249e) # avoid close 0, 1, 2
+        def close(ql, fd):
+            return 0
+        ql.set_syscall("close", close, QL_INTERCEPT.CALL)
+        ql.run()
         self.assertTrue(stdout.read() == b'Hello, World!\n')
 
         del ql
 
     def test_riscv64_hello_dyn_linux(self):
         stdout = SimpleOutStream(1)
         ql = Qiling(['../examples/rootfs/riscv64_linux/bin/hello-linux'], '../examples/rootfs/riscv64_linux/', 
-                    verbose=QL_VERBOSE.DEBUG, stdout=stdout)
-
+                    verbose=QL_VERBOSE.DEFAULT, stdout=stdout)
 
-        ql.run(end=0x1249e) # avoid close 0, 1, 2
+        ql.run()
         self.assertTrue(stdout.read() == b'Hello, World!\n')
 
         del ql