2 files changed, 12 insertions, 1 deletions
diff --git a/.bazelrc b/.bazelrc
index 717b2dc..ed993c5 100644
--- a/.bazelrc
+++ b/.bazelrc
@@ -6,6 +6,8 @@ common --experimental_repo_remote_exec
 common --cxxopt=-std=c++17
 common --host_cxxopt=-std=c++17
 common --copt=-w
+common --copt=-U_FORTIFY_SOURCE
+common --copt=-D_FORTIFY_SOURCE=2
 
 # Address sanitizer
 # bazel build --config asan
diff --git a/setup.py b/setup.py
index 92b3fb4..7fc714e 100644
--- a/setup.py
+++ b/setup.py
@@ -24,6 +24,12 @@ argparser.add_argument(
     choices=["asan", "ubsan"],
     required=False
 )
+argparser.add_argument(
+    "--tosa_checker_copt",
+    help="Build tosa_checker with addtional copt (comma separated string)",
+    default="",
+    required=False
+)
 args, unknown = argparser.parse_known_args()
 sys.argv = [sys.argv[0]] + unknown
 
@@ -66,7 +72,10 @@ class BazelBuildExtension(setuptools.command.build_ext.build_ext):
             commands += [
                 "--config={}".format(args.sanitizer)
             ]
-
+        if args.tosa_checker_copt:
+            commands += [
+                "--per_file_copt=tosa_checker/tosa_checker.*@{}".format(args.tosa_checker_copt)
+            ]
         commands += [
             # FIXME Some of the Bazel targets dependencies we use have
             # a 'friends' visibility, check if our Bazel target can be added