1 /*
2 * Copyright (c) 2023 Huawei Device Co., Ltd.
3 * Licensed under the Apache License, Version 2.0 (the "License");
4 * you may not use this file except in compliance with the License.
5 * You may obtain a copy of the License at
6 *
7 * http://www.apache.org/licenses/LICENSE-2.0
8 *
9 * Unless required by applicable law or agreed to in writing, software
10 * distributed under the License is distributed on an "AS IS" BASIS,
11 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 * See the License for the specific language governing permissions and
13 * limitations under the License.
14 */
15
16 //! stacktrace tools for Rust.
17
18 use std::ffi::CStr;
19 use std::os::raw::{c_char, c_int, c_uint};
20 use std::string::String;
21 use std::sync::Mutex;
22
23 static TRACE_MUTEX : Mutex<i32> = Mutex::new(0);
24
25 #[link(name = "backtrace_local")]
26 extern "C" {
PrintTrace(fd : c_int) -> bool27 fn PrintTrace(fd : c_int) -> bool;
GetTrace(skip_frame_num : c_uint, max_frame_num : c_uint) -> *const c_char28 fn GetTrace(skip_frame_num : c_uint, max_frame_num : c_uint) -> *const c_char;
29 }
30
31 /// Print Rust trace into File
print_trace(fd : i32) -> bool32 pub fn print_trace(fd : i32) -> bool {
33 unsafe {
34 PrintTrace(fd)
35 }
36 }
37
38 /// Get Rust trace by returned parameter
39 #[allow(unused_variables)]
get_trace(is_crash : bool) -> String40 pub fn get_trace(is_crash : bool) -> String {
41 unsafe {
42 let mutex = TRACE_MUTEX.lock().unwrap();
43 let mut skip_frame_num = 0;
44 let max_frame_num = 256;
45 if is_crash {
46 skip_frame_num = 8;
47 }
48 let trace = GetTrace(skip_frame_num, max_frame_num);
49 CStr::from_ptr(trace).to_str().unwrap().to_owned()
50 }
51 }