extend pyo3-ffi-check to check functions

pyo3-ffi-check/build.rs

@@ -32,6 +32,8 @@ fn main() {
     println!("cargo:rerun-if-changed=definitions");
     println!("cargo:rerun-if-changed=../pyo3-ffi");
 
+    pyo3_build_config::use_pyo3_cfgs();
+
     // Because `pyo3-ffi` is a dependency, libpython is linked, this ensures `main.rs` can run.
     // Slightly needless (no symbols from libpython are actually called), but simple to do.
     pyo3_build_config::add_libpython_rpath_link_args();

pyo3-ffi-check/definitions/build.rs

@@ -19,6 +19,19 @@ impl bindgen::callbacks::ParseCallbacks for ParseCallbacks {
     }
 }
 
+#[derive(Debug)]
+struct PyPyReplaceCallbacks;
+
+impl bindgen::callbacks::ParseCallbacks for PyPyReplaceCallbacks {
+    fn item_name(&self, item_info: ItemInfo<'_>) -> Option<String> {
+        if item_info.name.starts_with("PyPy") || item_info.name.starts_with("_PyPy") {
+            Some(item_info.name.replacen("PyPy", "Py", 1))
+        } else {
+            None
+        }
+    }
+}
+
 fn main() {
     let config = pyo3_build_config::get();
 
@@ -44,11 +57,20 @@ fn main() {
 
     println!("cargo:rerun-if-changed=wrapper.h");
 
-    let bindings = bindgen::Builder::default()
+    let mut builder = bindgen::Builder::default()
         .header("wrapper.h")
         .clang_args(clang_args)
         .parse_callbacks(Box::new(bindgen::CargoCallbacks::new()))
-        .parse_callbacks(Box::new(ParseCallbacks))
+        .parse_callbacks(Box::new(ParseCallbacks));
+
+    if matches!(
+        config.implementation,
+        pyo3_build_config::PythonImplementation::PyPy
+    ) {
+        builder = builder.parse_callbacks(Box::new(PyPyReplaceCallbacks));
+    }
+
+    let bindings = builder
         // blocklist some values which apparently have conflicting definitions on unix
         .blocklist_item("FP_NORMAL")
         .blocklist_item("FP_SUBNORMAL")