Commit 21bf3b80666c14c9b2a2e1ec984a6b4bb23a5bb7 | zig

author	Robin Voetter <robin@voetter.nl> 2021-10-26 00:51:46+02:00
committer	Robin Voetter <robin@voetter.nl> 2021-10-26 01:24:14+02:00
log	21bf3b80666c14c9b2a2e1ec984a6b4bb23a5bb7
tree	36e906cf5c4a40a715ed53f058feaccacf73ab8a
parent	4eb7b28700b23d8465a36e364e60394b2a1da41b

2 files changed, 11 insertions(+), 6 deletions(-)

src/codegen/llvm.zig+6-5

 @@ -2450,6 +2450,12 @@ pub const FuncGen = struct {
         const operand = try self.resolveInst(un_op);
         const operand_ty = self.air.typeOf(un_op);
         const optional_ty = if (operand_is_ptr) operand_ty.childType() else operand_ty;
+        if (optional_ty.isPtrLikeOptional()) {
+            const optional_llvm_ty = try self.dg.llvmType(optional_ty);
+            const loaded = if (operand_is_ptr) self.builder.buildLoad(operand, "") else operand;
+            return self.builder.buildICmp(pred, loaded, optional_llvm_ty.constNull(), "");
+        }
         var buf: Type.Payload.ElemType = undefined;
         const payload_ty = optional_ty.optionalChild(&buf);
         if (!payload_ty.hasCodeGenBits()) {
 @@ -2459,11 +2465,6 @@ pub const FuncGen = struct {
                 return operand;
             }
         }
-        if (optional_ty.isPtrLikeOptional()) {
-            const optional_llvm_ty = try self.dg.llvmType(optional_ty);
-            const loaded = if (operand_is_ptr) self.builder.buildLoad(operand, "") else operand;
-            return self.builder.buildICmp(pred, loaded, optional_llvm_ty.constNull(), "");
-        }
         if (operand_is_ptr or isByRef(optional_ty)) {
             const index_type = self.context.intType(32);

src/type.zig+5-1

 @@ -2347,11 +2347,13 @@ pub const Type = extern union {
         }
     }
-    /// Asserts that the type is an optional
+    /// Asserts that the type is an optional or a pointer that can be null.
     pub fn isPtrLikeOptional(self: Type) bool {
         switch (self.tag()) {
             .optional_single_const_pointer,
             .optional_single_mut_pointer,
+            .c_const_pointer,
+            .c_mut_pointer,
             => return true,
             .optional => {
 @@ -2367,6 +2369,8 @@ pub const Type = extern union {
                     .Many, .One => return !info.@"allowzero",
                 }
             },
+            .pointer => return self.castTag(.pointer).?.data.size == .C,
             else => unreachable,
         }
     }