[llvm] 6d25dce - [llubi] Implement ptrtoaddr (#203771)

[via llvm-commits]

Author: Zhige Chen
Date: 2026-06-14T14:56:57Z
New Revision: 6d25dcecf8bbc4f3e0246af8f3c6d6e9f70b85d4

URL: https://github.com/llvm/llvm-project/commit/6d25dcecf8bbc4f3e0246af8f3c6d6e9f70b85d4
DIFF: https://github.com/llvm/llvm-project/commit/6d25dcecf8bbc4f3e0246af8f3c6d6e9f70b85d4.diff

LOG: [llubi] Implement ptrtoaddr (#203771)

This PR implements the `ptrtoaddr` instruction.

Added: 
    llvm/test/tools/llubi/ptrtoaddr.ll
    llvm/test/tools/llubi/ptrtoaddr_after_ptrtoint.ll
    llvm/test/tools/llubi/ptrtoaddr_no_expose.ll

Modified: 
    llvm/tools/llubi/lib/Interpreter.cpp

Removed: 
    


################################################################################
diff  --git a/llvm/test/tools/llubi/ptrtoaddr.ll b/llvm/test/tools/llubi/ptrtoaddr.ll
new file mode 100644
index 0000000000000..6def7c3b2fa0e
--- /dev/null
+++ b/llvm/test/tools/llubi/ptrtoaddr.ll
@@ -0,0 +1,25 @@
+; RUN: llubi --verbose < %s 2>&1 | FileCheck %s
+
+target datalayout = "e-p:64:64:64-p1:64:64:64:32"
+
+define void @main() {
+  %wide1 = inttoptr i64 4294967301 to ptr addrspace(1)
+  %repr1 = ptrtoint ptr addrspace(1) %wide1 to i64
+  %addr1 = ptrtoaddr ptr addrspace(1) %wide1 to i32
+
+  %wide2 = inttoptr <2 x i64> <i64 4294967301, i64 4294967303> to <2 x ptr addrspace(1)>
+  %repr2 = ptrtoint <2 x ptr addrspace(1)> %wide2 to <2 x i64>
+  %addr2 = ptrtoaddr <2 x ptr addrspace(1)> %wide2 to <2 x i32>
+
+  ret void
+}
+
+; CHECK: Entering function: main
+; CHECK-NEXT:   %wide1 = inttoptr i64 4294967301 to ptr addrspace(1) => ptr 0x100000005 [nullary]
+; CHECK-NEXT:   %repr1 = ptrtoint ptr addrspace(1) %wide1 to i64 => i64 4294967301
+; CHECK-NEXT:   %addr1 = ptrtoaddr ptr addrspace(1) %wide1 to i32 => i32 5
+; CHECK-NEXT:   %wide2 = inttoptr <2 x i64> <i64 4294967301, i64 4294967303> to <2 x ptr addrspace(1)> => { ptr 0x100000005 [nullary], ptr 0x100000007 [nullary] }
+; CHECK-NEXT:   %repr2 = ptrtoint <2 x ptr addrspace(1)> %wide2 to <2 x i64> => { i64 4294967301, i64 4294967303 }
+; CHECK-NEXT:   %addr2 = ptrtoaddr <2 x ptr addrspace(1)> %wide2 to <2 x i32> => { i32 5, i32 7 }
+; CHECK-NEXT:   ret void
+; CHECK-NEXT: Exiting function: main

diff  --git a/llvm/test/tools/llubi/ptrtoaddr_after_ptrtoint.ll b/llvm/test/tools/llubi/ptrtoaddr_after_ptrtoint.ll
new file mode 100644
index 0000000000000..5cbf4818b8b56
--- /dev/null
+++ b/llvm/test/tools/llubi/ptrtoaddr_after_ptrtoint.ll
@@ -0,0 +1,21 @@
+; RUN: llubi --verbose < %s 2>&1 | FileCheck %s
+
+target datalayout = "e-p:64:64:64"
+
+define void @main() {
+  %p = alloca i32
+  %i = ptrtoint ptr %p to i64
+  %a = ptrtoaddr ptr %p to i64
+  %p2 = inttoptr i64 %a to ptr
+  store i32 0, ptr %p2
+  ret void
+}
+
+; CHECK: Entering function: main
+; CHECK-NEXT:   %p = alloca i32, align 4 => ptr 0x8 [p]
+; CHECK-NEXT:   %i = ptrtoint ptr %p to i64 => i64 8
+; CHECK-NEXT:   %a = ptrtoaddr ptr %p to i64 => i64 8
+; CHECK-NEXT:   %p2 = inttoptr i64 %a to ptr => ptr 0x8 [wildcard]
+; CHECK-NEXT:   store i32 0, ptr %p2, align 4
+; CHECK-NEXT:   ret void
+; CHECK-NEXT: Exiting function: main

diff  --git a/llvm/test/tools/llubi/ptrtoaddr_no_expose.ll b/llvm/test/tools/llubi/ptrtoaddr_no_expose.ll
new file mode 100644
index 0000000000000..ccd75aeb77042
--- /dev/null
+++ b/llvm/test/tools/llubi/ptrtoaddr_no_expose.ll
@@ -0,0 +1,20 @@
+; RUN: not llubi --verbose < %s 2>&1 | FileCheck %s
+
+target datalayout = "e-p:64:64:64"
+
+define void @main() {
+  %alloc = alloca i32
+  %addr = ptrtoaddr ptr %alloc to i64
+  %cast = inttoptr i64 %addr to ptr
+  store i32 0, ptr %cast
+  ret void
+}
+
+; CHECK: Entering function: main
+; CHECK-NEXT:   %alloc = alloca i32, align 4 => ptr 0x8 [alloc]
+; CHECK-NEXT:   %addr = ptrtoaddr ptr %alloc to i64 => i64 8
+; CHECK-NEXT:   %cast = inttoptr i64 %addr to ptr => ptr 0x8 [nullary]
+; CHECK-NEXT: Stacktrace:
+; CHECK-NEXT: #0   store i32 0, ptr %cast, align 4 at @main {{.*}}
+; CHECK-NEXT: Immediate UB detected: Invalid memory access via a pointer with nullary provenance.
+; CHECK-NEXT: error: Execution of function 'main' failed.

diff  --git a/llvm/tools/llubi/lib/Interpreter.cpp b/llvm/tools/llubi/lib/Interpreter.cpp
index 5769d0a4a9c4d..5f7c48794245d 100644
--- a/llvm/tools/llubi/lib/Interpreter.cpp
+++ b/llvm/tools/llubi/lib/Interpreter.cpp
@@ -2246,8 +2246,6 @@ class InstExecutor : public InstVisitor<InstExecutor, void>,
     });
   }
 
-  // TODO: Add support for ptrtoaddr, which exposes nothing.
-
   void visitIntToPtr(IntToPtrInst &I) {
     return visitUnOp(I, [&](const AnyValue &V) -> AnyValue {
       if (V.isPoison())
@@ -2260,6 +2258,15 @@ class InstExecutor : public InstVisitor<InstExecutor, void>,
     });
   }
 
+  void visitPtrToAddr(PtrToAddrInst &I) {
+    unsigned BitWidth = I.getType()->getScalarSizeInBits();
+    return visitUnOp(I, [&](const AnyValue &V) -> AnyValue {
+      if (V.isPoison())
+        return AnyValue::poison();
+      return V.asPointer().address().trunc(BitWidth);
+    });
+  }
+
   void visitLoadInst(LoadInst &LI) {
     auto RetVal = load(getValue(LI.getPointerOperand()), LI.getAlign(),
                        LI.getType(), LI.hasMetadata(LLVMContext::MD_noundef));