From b9889299b18225da43a0974e55a7278fb95b9462 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Traian-Florin=20=C8=98erb=C4=83nu=C8=9B=C4=83?=
 <traian.serbanuta@gmail.com>
Date: Tue, 17 Dec 2024 11:36:46 +0200
Subject: [PATCH] pyk: added --llvm-hidden-visibility attribute

---
 pyk/src/pyk/__main__.py      | 5 +++++
 pyk/src/pyk/cli/args.py      | 9 +++++++++
 pyk/src/pyk/ktool/kompile.py | 6 ++++++
 3 files changed, 20 insertions(+)

diff --git a/pyk/src/pyk/__main__.py b/pyk/src/pyk/__main__.py
index 81d09295acb..65cbe7ebcb2 100644
--- a/pyk/src/pyk/__main__.py
+++ b/pyk/src/pyk/__main__.py
@@ -324,6 +324,7 @@ def exec_kompile(options: KompileCommandOptions) -> None:
         kompile_dict['llvm_kompile_output'] = options.llvm_kompile_output
         kompile_dict['llvm_proof_hint_instrumentation'] = options.llvm_proof_hint_instrumentation
         kompile_dict['llvm_proof_hint_debugging'] = options.llvm_proof_hint_debugging
+        kompile_dict['llvm_hidden_visibility'] = options.llvm_hidden_visibility
     elif len(options.ccopts) > 0:
         raise ValueError(f'Option `-ccopt` requires `--backend llvm`, not: --backend {options.backend.value}')
     elif options.enable_search:
@@ -344,6 +345,10 @@ def exec_kompile(options: KompileCommandOptions) -> None:
         raise ValueError(
             f'Option `--llvm-proof-hint-debugging` requires `--backend llvm`, not: --backend {options.backend.value}'
         )
+    elif options.llvm_hidden_visibility:
+        raise ValueError(
+            f'Option `--llvm-hidden-visibility` requires `--backend llvm`, not: --backend {options.backend.value}'
+        )
 
     try:
         Kompile.from_dict(kompile_dict)(
diff --git a/pyk/src/pyk/cli/args.py b/pyk/src/pyk/cli/args.py
index 468f7da072a..e9114b38b56 100644
--- a/pyk/src/pyk/cli/args.py
+++ b/pyk/src/pyk/cli/args.py
@@ -194,6 +194,7 @@ class KompileOptions(Options):
     llvm_kompile_output: Path | None
     llvm_proof_hint_instrumentation: bool
     llvm_proof_hint_debugging: bool
+    llvm_hidden_visibility: bool
     read_only: bool
     o0: bool
     o1: bool
@@ -220,6 +221,7 @@ def default() -> dict[str, Any]:
             'llvm_kompile_output': None,
             'llvm_proof_hint_instrumentation': False,
             'llvm_proof_hint_debugging': False,
+            'llvm_hidden_visibility': False,
             'read_only': False,
             'o0': False,
             'o1': False,
@@ -479,6 +481,13 @@ def kompile_args(self) -> ArgumentParser:
             action='store_true',
             help='Enable additional proof hint debugging information in LLVM backend kompilation.',
         )
+        args.add_argument(
+            '--llvm-hidden-visibility',
+            dest='llvm_hidden_visibility',
+            default=None,
+            action='store_true',
+            help='Whether to make all symbols hidden by default in LLVM backend kompilation.',
+        )
 
         args.add_argument(
             '--no-exc-wrap',
diff --git a/pyk/src/pyk/ktool/kompile.py b/pyk/src/pyk/ktool/kompile.py
index a75b790c1c4..d7cad9680dd 100644
--- a/pyk/src/pyk/ktool/kompile.py
+++ b/pyk/src/pyk/ktool/kompile.py
@@ -428,6 +428,7 @@ class LLVMKompile(Kompile):
     enable_llvm_debug: bool
     llvm_proof_hint_instrumentation: bool
     llvm_proof_hint_debugging: bool
+    llvm_hidden_visibility: bool
     llvm_mutable_bytes: bool
     iterated_threshold: Fraction | None
     heuristic: str | None
@@ -445,6 +446,7 @@ def __init__(
         enable_llvm_debug: bool = False,
         llvm_proof_hint_instrumentation: bool = False,
         llvm_proof_hint_debugging: bool = False,
+        llvm_hidden_visibility: bool = False,
         llvm_mutable_bytes: bool = False,
         iterated_threshold: Fraction | None = None,
         heuristic: str | None = None,
@@ -468,6 +470,7 @@ def __init__(
         object.__setattr__(self, 'enable_llvm_debug', enable_llvm_debug)
         object.__setattr__(self, 'llvm_proof_hint_instrumentation', llvm_proof_hint_instrumentation)
         object.__setattr__(self, 'llvm_proof_hint_debugging', llvm_proof_hint_debugging)
+        object.__setattr__(self, 'llvm_hidden_visibility', llvm_hidden_visibility)
         object.__setattr__(self, 'llvm_mutable_bytes', llvm_mutable_bytes)
         object.__setattr__(self, 'iterated_threshold', iterated_threshold)
         object.__setattr__(self, 'heuristic', heuristic)
@@ -507,6 +510,9 @@ def args(self) -> list[str]:
         if self.llvm_proof_hint_debugging:
             args += ['--llvm-proof-hint-debugging']
 
+        if self.llvm_hidden_visibility:
+            args += ['--llvm-hidden-visibility']
+
         if self.llvm_mutable_bytes:
             args += ['--llvm-mutable-bytes']