From 465e9acd2c52d4218e6f278efc95ec25328c1f86 Mon Sep 17 00:00:00 2001
From: Kamil Lepek <kamil.lepek94@gmail.com>
Date: Mon, 15 Apr 2019 13:09:07 +0200
Subject: [PATCH] Add randomizers for ctypes and string in pyocf

---
 tests/functional/tests/security/__init__.py |  0
 tests/functional/tests/security/conftest.py | 83 +++++++++++++++++++++
 2 files changed, 83 insertions(+)
 create mode 100644 tests/functional/tests/security/__init__.py
 create mode 100644 tests/functional/tests/security/conftest.py

diff --git a/tests/functional/tests/security/__init__.py b/tests/functional/tests/security/__init__.py
new file mode 100644
index 0000000..e69de29
diff --git a/tests/functional/tests/security/conftest.py b/tests/functional/tests/security/conftest.py
new file mode 100644
index 0000000..5df3e81
--- /dev/null
+++ b/tests/functional/tests/security/conftest.py
@@ -0,0 +1,83 @@
+#
+# Copyright(c) 2019 Intel Corporation
+# SPDX-License-Identifier: BSD-3-Clause-Clear
+#
+
+import os
+import sys
+import random
+import string
+from ctypes import (
+    c_uint64,
+    c_uint32,
+    c_uint16,
+    c_int,
+    c_uint
+)
+
+import pytest
+
+sys.path.append(os.path.join(os.path.dirname(__file__), os.path.pardir))
+
+
+def generate_random_numbers(c_type):
+    type_dict = {
+        c_uint16: [0, c_uint16(-1).value],
+        c_uint32: [0, c_uint32(-1).value],
+        c_uint64: [0, c_uint64(-1).value],
+        c_int: [int(-c_uint(-1).value / 2) - 1, int(c_uint(-1).value / 2)]
+    }
+
+    values = []
+    for i in range(0, 1000):
+        values.append(random.randint(type_dict[c_type][0], type_dict[c_type][1]))
+    return values
+
+
+def generate_random_strings():
+    values = []
+    for t in [string.digits,
+              string.ascii_letters + string.digits,
+              string.ascii_lowercase,
+              string.ascii_uppercase,
+              string.printable,
+              string.punctuation,
+              string.hexdigits]:
+        for i in range(0, 100):
+            values.append(''.join(random.choice(t) for _ in range(random.randint(0, 20))))
+    return values
+
+
+def get_random_ints(c_type):
+    for value in generate_random_numbers(c_type):
+        yield value
+
+
+def get_random_strings():
+    for value in generate_random_strings():
+        yield value
+
+
+@pytest.fixture(params=get_random_ints(c_uint16))
+def c_uint16_randomize(request):
+    return request.param
+
+
+@pytest.fixture(params=get_random_ints(c_uint32))
+def c_uint32_randomize(request):
+    return request.param
+
+
+@pytest.fixture(params=get_random_ints(c_uint64))
+def c_uint64_randomize(request):
+    return request.param
+
+
+@pytest.fixture(params=get_random_ints(c_int))
+def c_int_randomize(request):
+    return request.param
+
+
+@pytest.fixture(params=get_random_strings())
+def string_randomize(request):
+    return request.param