summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorJi Won Shin <jiwonshin@google.com>2018-09-04 14:39:49 -0700
committerandroid-build-merger <android-build-merger@google.com>2018-09-04 14:39:49 -0700
commit4330bce1890ae456754a8f9572c1244924e4ff28 (patch)
tree3b5822ef894a7b92868dac5fec19a80b6355c2d8
parente4776c4255ef874aa06c783abdb599942d893970 (diff)
parent56ca31a154cc8ac72f2f74c339f852242e2ebe7b (diff)
downloadfuzz-pie-qpr3-release.tar.gz
am: 56ca31a154 Change-Id: I933972b85da374eea83744861b9b1af73d993e7f
-rw-r--r--kernel/syzkaller/syzkaller_test.py9
-rw-r--r--template/libfuzzer_test/libfuzzer_test.py13
2 files changed, 21 insertions, 1 deletions
diff --git a/kernel/syzkaller/syzkaller_test.py b/kernel/syzkaller/syzkaller_test.py
index 5d30d73..d951592 100644
--- a/kernel/syzkaller/syzkaller_test.py
+++ b/kernel/syzkaller/syzkaller_test.py
@@ -143,6 +143,15 @@ class SyzkallerTest(base_test.BaseTestClass):
asserts.fail('%s caused crash in our device.',
test_case._test_name)
+ def tearDownClass(self):
+ """Removes the temporary directory used for Syzkaller."""
+ logging.debug('Temporary directory %s is being deleted',
+ self._env['temp_dir'])
+ try:
+ shutil.rmtree(self._env['temp_dir'])
+ except OSError as e:
+ logging.exception(e)
+
def generateKernelFuzzerTests(self):
"""Runs kernel fuzzer tests."""
self.FetchSyzkaller()
diff --git a/template/libfuzzer_test/libfuzzer_test.py b/template/libfuzzer_test/libfuzzer_test.py
index ba4ea03..28137c5 100644
--- a/template/libfuzzer_test/libfuzzer_test.py
+++ b/template/libfuzzer_test/libfuzzer_test.py
@@ -17,6 +17,7 @@
import logging
import os
+import shutil
from vts.runners.host import asserts
from vts.runners.host import base_test
@@ -183,7 +184,8 @@ class LibFuzzerTest(base_test.BaseTestClass):
inuse_seed = self.RetrieveCorpusSeed(test_case)
if inuse_seed == 'locked':
# skip this test case
- logging.warning('test case locked, skipping testcase %s.', test_case.test_name)
+ logging.warning('test case locked, skipping testcase %s.',
+ test_case.test_name)
return
fuzz_cmd = '"%s"' % test_case.GetRunCommand()
@@ -275,6 +277,15 @@ class LibFuzzerTest(base_test.BaseTestClass):
elif exit_code != config.ExitCode.FUZZER_TEST_PASS:
asserts.fail('%s failed abnormally.' % test_case.test_name)
+ def tearDownClass(self):
+ """Removes the temporary directory used for corpus management."""
+ logging.debug('Temporary directory %s is being deleted',
+ self._temp_dir)
+ try:
+ shutil.rmtree(self._temp_dir)
+ except OSError as e:
+ logging.exception(e)
+
def generateFuzzerTests(self):
"""Runs fuzzer tests."""
self.runGeneratedTests(