import unittest
import cwe_checker_testlib


class TestCwe416(unittest.TestCase):

    def setUp(self):
        self.target = '416'
        self.string = b'Use After Free'
        self.check_name = 'Memory'

    def test_cwe416_01_x64_gcc(self):
        expect_res = 1
        res = cwe_checker_testlib.execute_and_check_occurence(
            self.target, self.target, 'x64', 'gcc', self.string, self.check_name)
        self.assertEqual(res, expect_res)

    def test_cwe416_01_x64_clang(self):
        expect_res = 1
        res = cwe_checker_testlib.execute_and_check_occurence(
            self.target, self.target, 'x64', 'clang', self.string, self.check_name)
        self.assertEqual(res, expect_res)

    @unittest.skip('FIXME: Needs pointer alignment tracking, or else SP = SP & Const loses the stack offset')
    def test_cwe416_01_x86_gcc(self):
        expect_res = 1
        res = cwe_checker_testlib.execute_and_check_occurence(
            self.target, self.target, 'x86', 'gcc', self.string, self.check_name)
        self.assertEqual(res, expect_res)

    def test_cwe416_01_x86_clang(self):
        expect_res = 1
        res = cwe_checker_testlib.execute_and_check_occurence(
            self.target, self.target, 'x86', 'clang', self.string, self.check_name)
        self.assertEqual(res, expect_res)

    def test_cwe416_01_arm_gcc(self):
        expect_res = 1
        res = cwe_checker_testlib.execute_and_check_occurence(
            self.target, self.target, 'arm', 'gcc', self.string, self.check_name)
        self.assertEqual(res, expect_res)

    def test_cwe416_01_arm_clang(self):
        expect_res = 1
        res = cwe_checker_testlib.execute_and_check_occurence(
            self.target, self.target, 'arm', 'clang', self.string, self.check_name)
        self.assertEqual(res, expect_res)

    @unittest.skip('Not supported by BAP. (no recognizable code backtrace)')
    def test_cwe416_01_aarch64_gcc(self):
        expect_res = 1
        res = cwe_checker_testlib.execute_and_check_occurence(
            self.target, self.target, 'aarch64', 'gcc', self.string, self.check_name)
        self.assertEqual(res, expect_res)

    @unittest.skip('Not supported by BAP. (no recognizable code backtrace)')
    def test_cwe416_01_aarch64_clang(self):
        expect_res = 1
        res = cwe_checker_testlib.execute_and_check_occurence(
            self.target, self.target, 'aarch64', 'clang', self.string, self.check_name)
        self.assertEqual(res, expect_res)

    @unittest.skip("FIXME: Check again when BAP handles the ZERO register of MIPS.")
    def test_cwe416_01_mips_gcc(self):
        expect_res = 1
        res = cwe_checker_testlib.execute_and_check_occurence(
            self.target, self.target, 'mips', 'gcc', self.string, self.check_name)
        self.assertEqual(res, expect_res)

    @unittest.skip("FIXME: Check again when BAP handles the ZERO register of MIPS.")
    def test_cwe416_01_mips_clang(self):
        expect_res = 1
        res = cwe_checker_testlib.execute_and_check_occurence(
            self.target, self.target, 'mips', 'clang', self.string, self.check_name)
        self.assertEqual(res, expect_res)

    @unittest.skip("FIXME: Check again when BAP handles the ZERO register of MIPS.")
    def test_cwe416_01_mipsel_gcc(self):
        expect_res = 1
        res = cwe_checker_testlib.execute_and_check_occurence(
            self.target, self.target, 'mipsel', 'gcc', self.string, self.check_name)
        self.assertEqual(res, expect_res)

    @unittest.skip("FIXME: Check again when BAP handles the ZERO register of MIPS.")
    def test_cwe416_01_mipsel_clang(self):
        expect_res = 1
        res = cwe_checker_testlib.execute_and_check_occurence(
            self.target, self.target, 'mipsel', 'clang', self.string, self.check_name)
        self.assertEqual(res, expect_res)

    @unittest.skip("BAP does not recognize extern calls")
    def test_cwe416_01_mips64_gcc(self):
        expect_res = 1
        res = cwe_checker_testlib.execute_and_check_occurence(
            self.target, self.target, 'mips64', 'gcc', self.string, self.check_name)
        self.assertEqual(res, expect_res)

    @unittest.skip("BAP does not recognize extern calls")
    def test_cwe416_01_mips64_clang(self):
        expect_res = 1
        res = cwe_checker_testlib.execute_and_check_occurence(
            self.target, self.target, 'mips64', 'clang', self.string, self.check_name)
        self.assertEqual(res, expect_res)

    @unittest.skip("BAP does not recognize extern calls")
    def test_cwe416_01_mips64el_gcc(self):
        expect_res = 1
        res = cwe_checker_testlib.execute_and_check_occurence(
            self.target, self.target, 'mips64el', 'gcc', self.string, self.check_name)
        self.assertEqual(res, expect_res)

    @unittest.skip("BAP does not recognize extern calls")
    def test_cwe416_01_mips64el_clang(self):
        expect_res = 1
        res = cwe_checker_testlib.execute_and_check_occurence(
            self.target, self.target, 'mips64el', 'clang', self.string, self.check_name)
        self.assertEqual(res, expect_res)

    #@unittest.skip('FIXME: Check again when moved to BAP 2.1')
    def test_cwe416_01_ppc_gcc(self):
        expect_res = 1
        res = cwe_checker_testlib.execute_and_check_occurence(
            self.target, self.target, 'ppc', 'gcc', self.string, self.check_name)
        self.assertEqual(res, expect_res)

    @unittest.skip('Dynamic Symbol calls are mangled by BAP')
    def test_cwe416_01_ppc64_gcc(self):
        expect_res = 1
        res = cwe_checker_testlib.execute_and_check_occurence(
            self.target, self.target, 'ppc64', 'gcc', self.string, self.check_name)
        self.assertEqual(res, expect_res)

    @unittest.skip('Dynamic Symbol calls are mangled by BAP')
    def test_cwe416_01_ppc64_clang(self):
        expect_res = 1
        res = cwe_checker_testlib.execute_and_check_occurence(
            self.target, self.target, 'ppc64', 'clang', self.string, self.check_name)
        self.assertEqual(res, expect_res)

    @unittest.skip('Dynamic Symbol calls are mangled by BAP')
    def test_cwe416_01_ppc64le_gcc(self):
        expect_res = 1
        res = cwe_checker_testlib.execute_and_check_occurence(
            self.target, self.target, 'ppc64le', 'gcc', self.string, self.check_name)
        self.assertEqual(res, expect_res)

    @unittest.skip('Dynamic Symbol calls are mangled by BAP')
    def test_cwe416_01_ppc64le_clang(self):
        expect_res = 1
        res = cwe_checker_testlib.execute_and_check_occurence(
            self.target, self.target, 'ppc64le', 'clang', self.string, self.check_name)
        self.assertEqual(res, expect_res)

    @unittest.skip("FIXME")
    def test_cwe416_01_x86_mingw_gcc(self):
        expect_res = 2
        res = cwe_checker_testlib.execute_and_check_occurence(
            self.target, self.target, 'x86', 'mingw32-gcc', self.string, self.check_name)
        self.assertEqual(res, expect_res)

    @unittest.skip('FIXME')
    def test_cwe416_01_x64_mingw_gcc(self):
        expect_res = 1
        res = cwe_checker_testlib.execute_and_check_occurence(
            self.target, self.target, 'x64', 'mingw32-gcc', self.string, self.check_name)
        self.assertEqual(res, expect_res)