allow raw borrow on just union field access

Author: Kivooeo

crates/hir-ty/src/diagnostics/unsafe_check.rs

@@ -296,6 +296,29 @@ impl<'db> UnsafeVisitor<'db> {
 
                         return;
                     }
+                    Expr::Field { .. } => {
+                        if matches!(
+                            self.infer.field_resolution(*expr),
+                            Some(Either::Left(FieldId { parent: VariantId::UnionId(_), .. }))
+                        ) {
+                            match &self.body.exprs[*expr] {
+                                Expr::Field { expr, .. } => {
+                                    // Visit the base expression (e.g., `self` in `self.field`) for safety,
+                                    // but don't trigger the union field access error since we're just
+                                    // creating a raw pointer, not actually reading the field
+                                    self.walk_expr(*expr);
+                                }
+                                _ => {
+                                    self.body.walk_child_exprs_without_pats(*expr, |child| {
+                                        // If it's not a field access for some reason, fall back to normal walking
+                                        // This shouldn't happen based on how this function is called
+                                        self.walk_expr(child)
+                                    });
+                                }
+                            }
+                            return;
+                        }
+                    }
                     _ => (),
                 }
             }

crates/ide-diagnostics/src/handlers/missing_unsafe.rs

@@ -834,6 +834,41 @@ fn bar(mut v: Union2) {
         )
     }
 
+    #[test]
+    fn raw_deref_on_union_field() {
+        check_diagnostics(
+            r#"
+fn main() {
+    union U1 {
+        a: u8
+    }
+    let x = U1 { a: 3 };
+
+    let a = x.a;
+         // ^^^ 💡 error: access to union field is unsafe and requires an unsafe function or block
+
+
+    let b = &raw const x.a;
+
+    let tmp = Vec::from([1, 2, 3]);
+
+    let c = &raw const tmp[x.a];
+                        // ^^^ 💡 error: access to union field is unsafe and requires an unsafe function or block
+
+    union URef {
+        p: &'static mut i32,
+    }
+
+    fn deref_union_field(u: URef) {
+        // Not an assignment but an access to the union field!
+        *(u.p) = 13;
+       // ^^^ 💡 error: access to union field is unsafe and requires an unsafe function or block
+    }
+}
+"#,
+        )
+    }
+
     #[test]
     fn raw_ref_reborrow_is_safe() {
         check_diagnostics(

crates/ide/src/syntax_highlighting/test_data/highlight_unsafe.html

@@ -96,7 +96,7 @@
 
         <span class="variable">u</span><span class="operator">.</span><span class="field unsafe">field</span><span class="semicolon">;</span>
         <span class="operator">&</span><span class="variable">u</span><span class="operator">.</span><span class="field unsafe">field</span><span class="semicolon">;</span>
-        <span class="operator">&</span><span class="keyword">raw</span> <span class="keyword const">const</span> <span class="variable">u</span><span class="operator">.</span><span class="field unsafe">field</span><span class="semicolon">;</span>
+        <span class="operator">&</span><span class="keyword">raw</span> <span class="keyword const">const</span> <span class="variable">u</span><span class="operator">.</span><span class="field">field</span><span class="semicolon">;</span>
         <span class="comment">// this should be safe!</span>
         <span class="keyword">let</span> <span class="union">Union</span> <span class="brace">{</span> <span class="field">field</span><span class="colon">:</span> <span class="punctuation">_</span> <span class="brace">}</span><span class="semicolon">;</span>
         <span class="comment">// but not these</span>