binder: Move cleanup_object from Allocation to AllocationView.

Signed-off-by: Wedson Almeida Filho <[email protected]>

Author: wedsonaf

drivers/android/allocation.rs

@@ -112,33 +112,6 @@ impl<'a> Allocation<'a> {
     pub(crate) fn set_info(&mut self, info: AllocationInfo) {
         self.allocation_info = Some(info);
     }
-
-    fn cleanup_object(&self, index_offset: usize, view: &AllocationView) -> Result {
-        let offset = self.read(index_offset)?;
-        let header = view.read::<bindings::binder_object_header>(offset)?;
-        // TODO: Handle other types.
-        match header.type_ {
-            BINDER_TYPE_WEAK_BINDER | BINDER_TYPE_BINDER => {
-                let obj = view.read::<bindings::flat_binder_object>(offset)?;
-                let strong = header.type_ == BINDER_TYPE_BINDER;
-                // SAFETY: The type is `BINDER_TYPE_{WEAK_}BINDER`, so the `binder` field is
-                // populated.
-                let ptr = unsafe { obj.__bindgen_anon_1.binder } as usize;
-                let cookie = obj.cookie as usize;
-                self.process.update_node(ptr, cookie, strong, false);
-                Ok(())
-            }
-            BINDER_TYPE_WEAK_HANDLE | BINDER_TYPE_HANDLE => {
-                let obj = view.read::<bindings::flat_binder_object>(offset)?;
-                let strong = header.type_ == BINDER_TYPE_HANDLE;
-                // SAFETY: The type is `BINDER_TYPE_{WEAK_}HANDLE`, so the `handle` field is
-                // populated.
-                let handle = unsafe { obj.__bindgen_anon_1.handle } as _;
-                self.process.update_ref(handle, false, strong)
-            }
-            _ => Ok(()),
-        }
-    }
 }
 
 impl Drop for Allocation<'_> {
@@ -150,7 +123,7 @@ impl Drop for Allocation<'_> {
         if let Some(info) = &self.allocation_info {
             let view = AllocationView::new(self, info.offsets.start);
             for i in info.offsets.clone().step_by(size_of::<usize>()) {
-                if self.cleanup_object(i, &view).is_err() {
+                if view.cleanup_object(i).is_err() {
                     pr_warn!("Error cleaning up object at offset {}\n", i)
                 }
             }
@@ -250,4 +223,31 @@ impl<'a> AllocationView<'a> {
         }
         Ok(())
     }
+
+    fn cleanup_object(&self, index_offset: usize) -> Result {
+        let offset = self.alloc.read(index_offset)?;
+        let header = self.read::<bindings::binder_object_header>(offset)?;
+        // TODO: Handle other types.
+        match header.type_ {
+            BINDER_TYPE_WEAK_BINDER | BINDER_TYPE_BINDER => {
+                let obj = self.read::<bindings::flat_binder_object>(offset)?;
+                let strong = header.type_ == BINDER_TYPE_BINDER;
+                // SAFETY: The type is `BINDER_TYPE_{WEAK_}BINDER`, so the `binder` field is
+                // populated.
+                let ptr = unsafe { obj.__bindgen_anon_1.binder } as usize;
+                let cookie = obj.cookie as usize;
+                self.alloc.process.update_node(ptr, cookie, strong, false);
+                Ok(())
+            }
+            BINDER_TYPE_WEAK_HANDLE | BINDER_TYPE_HANDLE => {
+                let obj = self.read::<bindings::flat_binder_object>(offset)?;
+                let strong = header.type_ == BINDER_TYPE_HANDLE;
+                // SAFETY: The type is `BINDER_TYPE_{WEAK_}HANDLE`, so the `handle` field is
+                // populated.
+                let handle = unsafe { obj.__bindgen_anon_1.handle } as _;
+                self.alloc.process.update_ref(handle, false, strong)
+            }
+            _ => Ok(()),
+        }
+    }
 }