From b771d52128cfb8244b067b19e13c425d7cfe0997 Mon Sep 17 00:00:00 2001
From: YangKeao <yangkeao@chunibyo.icu>
Date: Thu, 19 Jan 2023 04:41:24 -0500
Subject: [PATCH] update the prost generated file and remove malloc_free (#188)

* update prost file

Signed-off-by: YangKeao <yangkeao@chunibyo.icu>

* remove malloc free test

Signed-off-by: YangKeao <yangkeao@chunibyo.icu>

Signed-off-by: YangKeao <yangkeao@chunibyo.icu>
---
 examples/malloc_hook.rs     | 109 ---------------------------------
 proto/perftools.profiles.rs |   8 +++
 src/collector.rs            |  87 ---------------------------
 src/profiler.rs             | 116 ------------------------------------
 4 files changed, 8 insertions(+), 312 deletions(-)
 delete mode 100644 examples/malloc_hook.rs

diff --git a/examples/malloc_hook.rs b/examples/malloc_hook.rs
deleted file mode 100644
index 56bea852..00000000
--- a/examples/malloc_hook.rs
+++ /dev/null
@@ -1,109 +0,0 @@
-// Copyright 2019 TiKV Project Authors. Licensed under Apache-2.0.
-
-extern crate libc;
-
-use std::ffi::c_void;
-
-#[cfg(not(target_env = "gnu"))]
-#[allow(clippy::wrong_self_convention)]
-#[allow(non_upper_case_globals)]
-static mut __malloc_hook: Option<extern "C" fn(size: usize) -> *mut c_void> = None;
-
-#[cfg(target_arch = "riscv64")]
-#[allow(clippy::wrong_self_convention)]
-#[allow(non_upper_case_globals)]
-static mut __malloc_hook: Option<extern "C" fn(size: usize) -> *mut c_void> = None;
-
-extern "C" {
-    #[cfg(target_env = "gnu")]
-    #[cfg(not(target_arch = "riscv64"))]
-    static mut __malloc_hook: Option<extern "C" fn(size: usize) -> *mut c_void>;
-
-    fn malloc(size: usize) -> *mut c_void;
-}
-
-static mut FLAG: bool = false;
-
-extern "C" fn malloc_hook(size: usize) -> *mut c_void {
-    unsafe {
-        FLAG = true;
-    }
-    remove_hook();
-
-    let bt = backtrace::Backtrace::new();
-    println!("{:?}", bt);
-    let p = unsafe { malloc(size) };
-
-    set_hook();
-
-    p
-}
-
-fn set_hook() {
-    unsafe {
-        __malloc_hook = Some(malloc_hook);
-    }
-}
-
-fn remove_hook() {
-    unsafe {
-        __malloc_hook = None;
-    }
-}
-
-#[inline(never)]
-fn is_prime_number(v: usize, prime_numbers: &[usize]) -> bool {
-    if v < 10000 {
-        let r = prime_numbers.binary_search(&v);
-        return r.is_ok();
-    }
-
-    for n in prime_numbers {
-        if v % n == 0 {
-            return false;
-        }
-    }
-
-    true
-}
-
-#[inline(never)]
-fn prepare_prime_numbers() -> Vec<usize> {
-    // bootstrap: Generate a prime table of 0..10000
-    let mut prime_number_table: [bool; 10000] = [true; 10000];
-    prime_number_table[0] = false;
-    prime_number_table[1] = false;
-    for i in 2..10000 {
-        if prime_number_table[i] {
-            let mut v = i * 2;
-            while v < 10000 {
-                prime_number_table[v] = false;
-                v += i;
-            }
-        }
-    }
-    let mut prime_numbers = vec![];
-    for (i, exist) in prime_number_table.iter().enumerate().skip(2) {
-        if *exist {
-            prime_numbers.push(i);
-        }
-    }
-    prime_numbers
-}
-
-fn main() {
-    let prime_numbers = prepare_prime_numbers();
-
-    let _ = pprof::ProfilerGuard::new(100).unwrap();
-
-    loop {
-        let mut _v = 0;
-
-        set_hook();
-        for i in 2..50000 {
-            if is_prime_number(i, &prime_numbers) {
-                _v += 1;
-            }
-        }
-    }
-}
diff --git a/proto/perftools.profiles.rs b/proto/perftools.profiles.rs
index 3d3e432e..22453330 100644
--- a/proto/perftools.profiles.rs
+++ b/proto/perftools.profiles.rs
@@ -1,5 +1,6 @@
 // f9f855b960d01b292a3c2642e263e6156d52631e78e0177fe51416ed5bbecc81  proto/profile.proto
 
+#[allow(clippy::derive_partial_eq_without_eq)]
 #[derive(Clone, PartialEq, ::prost::Message)]
 pub struct Profile {
     /// A description of the samples associated with each Sample.value.
@@ -65,6 +66,7 @@ pub struct Profile {
     pub default_sample_type: i64,
 }
 /// ValueType describes the semantics and measurement units of a value.
+#[allow(clippy::derive_partial_eq_without_eq)]
 #[derive(Clone, PartialEq, ::prost::Message)]
 pub struct ValueType {
     /// Rename it from type to ty to avoid using keyword in Rust.
@@ -80,6 +82,7 @@ pub struct ValueType {
 /// context. The program context is typically a stack trace, perhaps
 /// augmented with auxiliary information like the thread-id, some
 /// indicator of a higher level request being handled etc.
+#[allow(clippy::derive_partial_eq_without_eq)]
 #[derive(Clone, PartialEq, ::prost::Message)]
 pub struct Sample {
     /// The ids recorded here correspond to a Profile.location.id.
@@ -99,6 +102,7 @@ pub struct Sample {
     #[prost(message, repeated, tag = "3")]
     pub label: ::prost::alloc::vec::Vec<Label>,
 }
+#[allow(clippy::derive_partial_eq_without_eq)]
 #[derive(Clone, PartialEq, ::prost::Message)]
 pub struct Label {
     /// Index into string table
@@ -123,6 +127,7 @@ pub struct Label {
     #[prost(int64, tag = "4")]
     pub num_unit: i64,
 }
+#[allow(clippy::derive_partial_eq_without_eq)]
 #[derive(Clone, PartialEq, ::prost::Message)]
 pub struct Mapping {
     /// Unique nonzero id for the mapping.
@@ -162,6 +167,7 @@ pub struct Mapping {
     pub has_inline_frames: bool,
 }
 /// Describes function and line table debug information.
+#[allow(clippy::derive_partial_eq_without_eq)]
 #[derive(Clone, PartialEq, ::prost::Message)]
 pub struct Location {
     /// Unique nonzero id for the location.  A profile could use
@@ -197,6 +203,7 @@ pub struct Location {
     #[prost(bool, tag = "5")]
     pub is_folded: bool,
 }
+#[allow(clippy::derive_partial_eq_without_eq)]
 #[derive(Clone, PartialEq, ::prost::Message)]
 pub struct Line {
     /// The id of the corresponding profile.Function for this line.
@@ -206,6 +213,7 @@ pub struct Line {
     #[prost(int64, tag = "2")]
     pub line: i64,
 }
+#[allow(clippy::derive_partial_eq_without_eq)]
 #[derive(Clone, PartialEq, ::prost::Message)]
 pub struct Function {
     /// Unique nonzero id for the function.
diff --git a/src/collector.rs b/src/collector.rs
index 207de5a4..41b34139 100644
--- a/src/collector.rs
+++ b/src/collector.rs
@@ -360,90 +360,3 @@ mod tests {
         }
     }
 }
-
-#[cfg(test)]
-#[cfg(target_os = "linux")]
-mod malloc_free_test {
-    use super::*;
-    use std::cell::RefCell;
-    use std::collections::BTreeMap;
-    use std::ffi::c_void;
-
-    #[cfg(not(target_env = "gnu"))]
-    #[allow(clippy::wrong_self_convention)]
-    #[allow(non_upper_case_globals)]
-    static mut __malloc_hook: Option<extern "C" fn(size: usize) -> *mut c_void> = None;
-
-    #[cfg(target_arch = "riscv64")]
-    #[allow(clippy::wrong_self_convention)]
-    #[allow(non_upper_case_globals)]
-    static mut __malloc_hook: Option<extern "C" fn(size: usize) -> *mut c_void> = None;
-
-    extern "C" {
-        #[cfg(target_env = "gnu")]
-        #[cfg(not(target_arch = "riscv64"))]
-        static mut __malloc_hook: Option<extern "C" fn(size: usize) -> *mut c_void>;
-
-        fn malloc(size: usize) -> *mut c_void;
-    }
-
-    thread_local! {
-        static FLAG: RefCell<bool> = RefCell::new(false);
-    }
-
-    extern "C" fn malloc_hook(size: usize) -> *mut c_void {
-        unsafe {
-            __malloc_hook = None;
-        }
-
-        FLAG.with(|flag| {
-            flag.replace(true);
-        });
-        let p = unsafe { malloc(size) };
-
-        unsafe {
-            __malloc_hook = Some(malloc_hook);
-        }
-
-        p
-    }
-
-    #[test]
-    fn malloc_free() {
-        let mut collector = Collector::new().unwrap();
-        let mut real_map = BTreeMap::new();
-
-        unsafe {
-            __malloc_hook = Some(malloc_hook);
-        }
-
-        for item in 0..(1 << 10) * 4 {
-            for _ in 0..(item % 4) {
-                collector.add(item, 1).unwrap();
-            }
-        }
-        unsafe {
-            __malloc_hook = None;
-        }
-
-        FLAG.with(|flag| {
-            assert!(!*flag.borrow());
-        });
-
-        collector.try_iter().unwrap().for_each(|entry| {
-            test_utils::add_map(&mut real_map, entry);
-        });
-
-        for item in 0..(1 << 10) * 4 {
-            let count = (item % 4) as isize;
-            match real_map.get(&item) {
-                Some(value) => {
-                    assert_eq!(count, *value);
-                }
-                None => {
-                    assert_eq!(count, 0);
-                }
-            }
-        }
-    }
-}
diff --git a/src/profiler.rs b/src/profiler.rs
index c78f0dc2..294b8c55 100644
--- a/src/profiler.rs
+++ b/src/profiler.rs
@@ -464,119 +464,3 @@ impl Profiler {
         if let Ok(()) = self.data.add(frames, 1) {}
     }
 }
-
-#[cfg(test)]
-#[cfg(target_os = "linux")]
-mod tests {
-    use super::*;
-
-    use std::cell::RefCell;
-    use std::ffi::c_void;
-    use std::ptr::null_mut;
-
-    #[cfg(not(target_env = "gnu"))]
-    #[allow(clippy::wrong_self_convention)]
-    #[allow(non_upper_case_globals)]
-    static mut __malloc_hook: Option<extern "C" fn(size: usize) -> *mut c_void> = None;
-
-    #[cfg(target_arch = "riscv64")]
-    #[allow(clippy::wrong_self_convention)]
-    #[allow(non_upper_case_globals)]
-    static mut __malloc_hook: Option<extern "C" fn(size: usize) -> *mut c_void> = None;
-
-    extern "C" {
-        #[cfg(target_env = "gnu")]
-        #[cfg(not(target_arch = "riscv64"))]
-        static mut __malloc_hook: Option<extern "C" fn(size: usize) -> *mut c_void>;
-
-        fn malloc(size: usize) -> *mut c_void;
-    }
-
-    thread_local! {
-        static FLAG: RefCell<bool> = RefCell::new(false);
-    }
-
-    extern "C" fn malloc_hook(size: usize) -> *mut c_void {
-        unsafe {
-            __malloc_hook = None;
-        }
-
-        FLAG.with(|flag| {
-            flag.replace(true);
-        });
-        let p = unsafe { malloc(size) };
-
-        unsafe {
-            __malloc_hook = Some(malloc_hook);
-        }
-
-        p
-    }
-
-    #[inline(never)]
-    fn is_prime_number(v: usize, prime_numbers: &[usize]) -> bool {
-        if v < 10000 {
-            let r = prime_numbers.binary_search(&v);
-            return r.is_ok();
-        }
-
-        for n in prime_numbers {
-            if v % n == 0 {
-                return false;
-            }
-        }
-
-        true
-    }
-
-    #[inline(never)]
-    fn prepare_prime_numbers() -> Vec<usize> {
-        // bootstrap: Generate a prime table of 0..10000
-        let mut prime_number_table: [bool; 10000] = [true; 10000];
-        prime_number_table[0] = false;
-        prime_number_table[1] = false;
-        for i in 2..10000 {
-            if prime_number_table[i] {
-                let mut v = i * 2;
-                while v < 10000 {
-                    prime_number_table[v] = false;
-                    v += i;
-                }
-            }
-        }
-        let mut prime_numbers = vec![];
-        for (i, item) in prime_number_table.iter().enumerate().skip(2) {
-            if *item {
-                prime_numbers.push(i);
-            }
-        }
-        prime_numbers
-    }
-
-    #[cfg(target_os = "linux")]
-    #[test]
-    fn malloc_free() {
-        trigger_lazy();
-
-        let prime_numbers = prepare_prime_numbers();
-
-        let mut _v = 0;
-
-        unsafe {
-            __malloc_hook = Some(malloc_hook);
-        }
-        for i in 2..50000 {
-            if is_prime_number(i, &prime_numbers) {
-                _v += 1;
-                perf_signal_handler(27, null_mut(), null_mut());
-            }
-        }
-        unsafe {
-            __malloc_hook = None;
-        }
-
-        FLAG.with(|flag| {
-            assert!(!*flag.borrow());
-        });
-    }
-}