From 15ace721ea4ec21d23feb536d92789d0e238b7af Mon Sep 17 00:00:00 2001
From: Lily Foote <code@lilyf.org>
Date: Tue, 30 Jul 2024 19:57:30 +0100
Subject: [PATCH] Add missing #[allow(unsafe_code)] attributes

Fixes #4394.
---
 newsfragments/4396.fixed.md         | 1 +
 pyo3-macros-backend/src/pyclass.rs  | 1 +
 pyo3-macros-backend/src/pymethod.rs | 2 ++
 src/impl_/pyclass.rs                | 6 +++++-
 src/tests/hygiene/mod.rs            | 1 +
 src/types/mod.rs                    | 2 ++
 6 files changed, 12 insertions(+), 1 deletion(-)
 create mode 100644 newsfragments/4396.fixed.md

diff --git a/newsfragments/4396.fixed.md b/newsfragments/4396.fixed.md
new file mode 100644
index 00000000000..285358ad526
--- /dev/null
+++ b/newsfragments/4396.fixed.md
@@ -0,0 +1 @@
+Hide confusing warnings about unsafe usage in `#[pyclass]` implementation.
diff --git a/pyo3-macros-backend/src/pyclass.rs b/pyo3-macros-backend/src/pyclass.rs
index 6fba5b7e23e..2936f05cd44 100644
--- a/pyo3-macros-backend/src/pyclass.rs
+++ b/pyo3-macros-backend/src/pyclass.rs
@@ -1744,6 +1744,7 @@ fn impl_pytypeinfo(cls: &syn::Ident, attr: &PyClassArgs, ctx: &Ctx) -> TokenStre
     };
 
     quote! {
+        #[allow(unsafe_code)]
         unsafe impl #pyo3_path::type_object::PyTypeInfo for #cls {
             const NAME: &'static str = #cls_name;
             const MODULE: ::std::option::Option<&'static str> = #module;
diff --git a/pyo3-macros-backend/src/pymethod.rs b/pyo3-macros-backend/src/pymethod.rs
index 77cc9ed5cc6..584dd8f04dd 100644
--- a/pyo3-macros-backend/src/pymethod.rs
+++ b/pyo3-macros-backend/src/pymethod.rs
@@ -772,6 +772,7 @@ pub fn impl_py_getter_def(
                     use #pyo3_path::impl_::pyclass::Probe;
 
                     struct Offset;
+                    #[allow(unsafe_code)]
                     unsafe impl #pyo3_path::impl_::pyclass::OffsetCalculator<#cls, #ty> for Offset {
                         fn offset() -> usize {
                             #pyo3_path::impl_::pyclass::class_offset::<#cls>() +
@@ -779,6 +780,7 @@ pub fn impl_py_getter_def(
                         }
                     }
 
+                    #[allow(unsafe_code)]
                     const GENERATOR: #pyo3_path::impl_::pyclass::PyClassGetterGenerator::<
                         #cls,
                         #ty,
diff --git a/src/impl_/pyclass.rs b/src/impl_/pyclass.rs
index 7ad772f0496..112bc54417f 100644
--- a/src/impl_/pyclass.rs
+++ b/src/impl_/pyclass.rs
@@ -346,6 +346,7 @@ slot_fragment_trait! {
 #[macro_export]
 macro_rules! generate_pyclass_getattro_slot {
     ($cls:ty) => {{
+        #[allow(unsafe_code)]
         unsafe extern "C" fn __wrap(
             _slf: *mut $crate::ffi::PyObject,
             attr: *mut $crate::ffi::PyObject,
@@ -429,6 +430,7 @@ macro_rules! define_pyclass_setattr_slot {
         #[macro_export]
         macro_rules! $generate_macro {
             ($cls:ty) => {{
+                #[allow(unsafe_code)]
                 unsafe extern "C" fn __wrap(
                     _slf: *mut $crate::ffi::PyObject,
                     attr: *mut $crate::ffi::PyObject,
@@ -545,6 +547,7 @@ macro_rules! define_pyclass_binary_operator_slot {
         #[macro_export]
         macro_rules! $generate_macro {
             ($cls:ty) => {{
+                #[allow(unsafe_code)]
                 unsafe extern "C" fn __wrap(
                     _slf: *mut $crate::ffi::PyObject,
                     _other: *mut $crate::ffi::PyObject,
@@ -737,6 +740,7 @@ slot_fragment_trait! {
 #[macro_export]
 macro_rules! generate_pyclass_pow_slot {
     ($cls:ty) => {{
+        #[allow(unsafe_code)]
         unsafe extern "C" fn __wrap(
             _slf: *mut $crate::ffi::PyObject,
             _other: *mut $crate::ffi::PyObject,
@@ -861,7 +865,7 @@ macro_rules! generate_pyclass_richcompare_slot {
     ($cls:ty) => {{
         #[allow(unknown_lints, non_local_definitions)]
         impl $cls {
-            #[allow(non_snake_case)]
+            #[allow(non_snake_case, unsafe_code)]
             unsafe extern "C" fn __pymethod___richcmp____(
                 slf: *mut $crate::ffi::PyObject,
                 other: *mut $crate::ffi::PyObject,
diff --git a/src/tests/hygiene/mod.rs b/src/tests/hygiene/mod.rs
index c950e18da94..9bf89161b24 100644
--- a/src/tests/hygiene/mod.rs
+++ b/src/tests/hygiene/mod.rs
@@ -1,5 +1,6 @@
 #![no_implicit_prelude]
 #![allow(dead_code, unused_variables, clippy::unnecessary_wraps)]
+#![deny(unsafe_code)]
 
 // The modules in this test are used to check PyO3 macro expansion is hygienic. By locating the test
 // inside the crate the global `::pyo3` namespace is not available, so in combination with
diff --git a/src/types/mod.rs b/src/types/mod.rs
index fdf73434864..2ba64f46566 100644
--- a/src/types/mod.rs
+++ b/src/types/mod.rs
@@ -133,6 +133,7 @@ macro_rules! pyobject_native_type_named (
             }
         }
 
+        #[allow(unsafe_code)]
         unsafe impl<$($generics,)*> $crate::AsPyPointer for $name {
             /// Gets the underlying FFI pointer, returns a borrowed pointer.
             #[inline]
@@ -160,6 +161,7 @@ macro_rules! pyobject_native_static_type_object(
 #[macro_export]
 macro_rules! pyobject_native_type_info(
     ($name:ty, $typeobject:expr, $module:expr $(, #checkfunction=$checkfunction:path)? $(;$generics:ident)*) => {
+        #[allow(unsafe_code)]
         unsafe impl<$($generics,)*> $crate::type_object::PyTypeInfo for $name {
             const NAME: &'static str = stringify!($name);
             const MODULE: ::std::option::Option<&'static str> = $module;