コード例 #1
0
ファイル: integrity.py プロジェクト: xme/CAPEv2
            args = dict(
                (arg['name'], arg['value']) for arg in row['arguments'])
            buf += args['Buffer']

        self.assertEqual(buf, 'hello!!1\r\n')

    def test_file_write(self):
        report = self.run_analysis("""
            open('a.txt', 'wb').write('Hello World')
        """)
        self.assertTrue('dropped' in report)
        self.assertEqual(len(report['dropped']), 1)
        self.assertEqual(
            open(report['dropped'][0]['path']).read(), 'Hello World')


if __name__ == '__main__':
    parser = argparse.ArgumentParser()
    # parser.add_argument("vmname", type=str, required=False, help="Name of the Virtual Machine to check.")
    args = parser.parse_args()

    # Clean the database.
    subprocess.Popen([os.path.join(CUCKOO_ROOT, 'utils', 'clean.sh')],
                     cwd=CUCKOO_ROOT).wait()

    cuckoo_init(quiet=True)

    db = Database()

    unittest.main()
コード例 #2
0
ファイル: integrity.py プロジェクト: 0day29/cuckoo
            args = dict((arg['name'], arg['value']) for arg in row['arguments'])
            buf += args['Buffer']

        self.assertEqual(buf, 'hello!!1\r\n')

    def test_file_write(self):
        report = self.run_analysis("""
            open('a.txt', 'wb').write('Hello World')
        """)
        self.assertTrue('dropped' in report)
        self.assertEqual(len(report['dropped']), 1)
        self.assertEqual(open(report['dropped'][0]['path']).read(),
                         'Hello World')


if __name__ == '__main__':
    parser = argparse.ArgumentParser()
    # parser.add_argument("vmname", type=str, required=False, help="Name of the Virtual Machine to check.")
    args = parser.parse_args()

    # Clean the database.
    subprocess.Popen([os.path.join(CUCKOO_ROOT, 'utils', 'clean.sh')],
                     cwd=CUCKOO_ROOT).wait()

    cuckoo_init(quiet=True)

    db = Database()

    unittest.main()