]> git.lizzy.rs Git - rust.git/blob - src/libcore/panic.rs
rustc_span: return an impl Iterator instead of a Vec from macro_backtrace.
[rust.git] / src / libcore / panic.rs
1 //! Panic support in the standard library.
2
3 #![stable(feature = "core_panic_info", since = "1.41.0")]
4
5 use crate::any::Any;
6 use crate::fmt;
7
8 /// A struct providing information about a panic.
9 ///
10 /// `PanicInfo` structure is passed to a panic hook set by the [`set_hook`]
11 /// function.
12 ///
13 /// [`set_hook`]: ../../std/panic/fn.set_hook.html
14 ///
15 /// # Examples
16 ///
17 /// ```should_panic
18 /// use std::panic;
19 ///
20 /// panic::set_hook(Box::new(|panic_info| {
21 ///     if let Some(s) = panic_info.payload().downcast_ref::<&str>() {
22 ///         println!("panic occurred: {:?}", s);
23 ///     } else {
24 ///         println!("panic occurred");
25 ///     }
26 /// }));
27 ///
28 /// panic!("Normal panic");
29 /// ```
30 #[lang = "panic_info"]
31 #[stable(feature = "panic_hooks", since = "1.10.0")]
32 #[derive(Debug)]
33 pub struct PanicInfo<'a> {
34     payload: &'a (dyn Any + Send),
35     message: Option<&'a fmt::Arguments<'a>>,
36     location: &'a Location<'a>,
37 }
38
39 impl<'a> PanicInfo<'a> {
40     #[unstable(
41         feature = "panic_internals",
42         reason = "internal details of the implementation of the `panic!` \
43                          and related macros",
44         issue = "none"
45     )]
46     #[doc(hidden)]
47     #[inline]
48     pub fn internal_constructor(
49         message: Option<&'a fmt::Arguments<'a>>,
50         location: &'a Location<'a>,
51     ) -> Self {
52         struct NoPayload;
53         PanicInfo { location, message, payload: &NoPayload }
54     }
55
56     #[unstable(
57         feature = "panic_internals",
58         reason = "internal details of the implementation of the `panic!` \
59                          and related macros",
60         issue = "none"
61     )]
62     #[doc(hidden)]
63     #[inline]
64     pub fn set_payload(&mut self, info: &'a (dyn Any + Send)) {
65         self.payload = info;
66     }
67
68     /// Returns the payload associated with the panic.
69     ///
70     /// This will commonly, but not always, be a `&'static str` or [`String`].
71     ///
72     /// [`String`]: ../../std/string/struct.String.html
73     ///
74     /// # Examples
75     ///
76     /// ```should_panic
77     /// use std::panic;
78     ///
79     /// panic::set_hook(Box::new(|panic_info| {
80     ///     println!("panic occurred: {:?}", panic_info.payload().downcast_ref::<&str>().unwrap());
81     /// }));
82     ///
83     /// panic!("Normal panic");
84     /// ```
85     #[stable(feature = "panic_hooks", since = "1.10.0")]
86     pub fn payload(&self) -> &(dyn Any + Send) {
87         self.payload
88     }
89
90     /// If the `panic!` macro from the `core` crate (not from `std`)
91     /// was used with a formatting string and some additional arguments,
92     /// returns that message ready to be used for example with [`fmt::write`]
93     ///
94     /// [`fmt::write`]: ../fmt/fn.write.html
95     #[unstable(feature = "panic_info_message", issue = "66745")]
96     pub fn message(&self) -> Option<&fmt::Arguments<'_>> {
97         self.message
98     }
99
100     /// Returns information about the location from which the panic originated,
101     /// if available.
102     ///
103     /// This method will currently always return [`Some`], but this may change
104     /// in future versions.
105     ///
106     /// [`Some`]: ../../std/option/enum.Option.html#variant.Some
107     ///
108     /// # Examples
109     ///
110     /// ```should_panic
111     /// use std::panic;
112     ///
113     /// panic::set_hook(Box::new(|panic_info| {
114     ///     if let Some(location) = panic_info.location() {
115     ///         println!("panic occurred in file '{}' at line {}", location.file(),
116     ///             location.line());
117     ///     } else {
118     ///         println!("panic occurred but can't get location information...");
119     ///     }
120     /// }));
121     ///
122     /// panic!("Normal panic");
123     /// ```
124     #[stable(feature = "panic_hooks", since = "1.10.0")]
125     pub fn location(&self) -> Option<&Location<'_>> {
126         // NOTE: If this is changed to sometimes return None,
127         // deal with that case in std::panicking::default_hook and std::panicking::begin_panic_fmt.
128         Some(&self.location)
129     }
130 }
131
132 #[stable(feature = "panic_hook_display", since = "1.26.0")]
133 impl fmt::Display for PanicInfo<'_> {
134     fn fmt(&self, formatter: &mut fmt::Formatter<'_>) -> fmt::Result {
135         formatter.write_str("panicked at ")?;
136         if let Some(message) = self.message {
137             write!(formatter, "'{}', ", message)?
138         } else if let Some(payload) = self.payload.downcast_ref::<&'static str>() {
139             write!(formatter, "'{}', ", payload)?
140         }
141         // NOTE: we cannot use downcast_ref::<String>() here
142         // since String is not available in libcore!
143         // The payload is a String when `std::panic!` is called with multiple arguments,
144         // but in that case the message is also available.
145
146         self.location.fmt(formatter)
147     }
148 }
149
150 /// A struct containing information about the location of a panic.
151 ///
152 /// This structure is created by the [`location`] method of [`PanicInfo`].
153 ///
154 /// [`location`]: ../../std/panic/struct.PanicInfo.html#method.location
155 /// [`PanicInfo`]: ../../std/panic/struct.PanicInfo.html
156 ///
157 /// # Examples
158 ///
159 /// ```should_panic
160 /// use std::panic;
161 ///
162 /// panic::set_hook(Box::new(|panic_info| {
163 ///     if let Some(location) = panic_info.location() {
164 ///         println!("panic occurred in file '{}' at line {}", location.file(), location.line());
165 ///     } else {
166 ///         println!("panic occurred but can't get location information...");
167 ///     }
168 /// }));
169 ///
170 /// panic!("Normal panic");
171 /// ```
172 #[lang = "panic_location"]
173 #[derive(Debug)]
174 #[stable(feature = "panic_hooks", since = "1.10.0")]
175 pub struct Location<'a> {
176     file: &'a str,
177     line: u32,
178     col: u32,
179 }
180
181 impl<'a> Location<'a> {
182     /// Returns the source location of the caller of this function. If that function's caller is
183     /// annotated then its call location will be returned, and so on up the stack to the first call
184     /// within a non-tracked function body.
185     ///
186     /// # Examples
187     ///
188     /// ```
189     /// #![feature(track_caller)]
190     /// use core::panic::Location;
191     ///
192     /// /// Returns the [`Location`] at which it is called.
193     /// #[track_caller]
194     /// fn get_caller_location() -> &'static Location<'static> {
195     ///     Location::caller()
196     /// }
197     ///
198     /// /// Returns a [`Location`] from within this function's definition.
199     /// fn get_just_one_location() -> &'static Location<'static> {
200     ///     get_caller_location()
201     /// }
202     ///
203     /// let fixed_location = get_just_one_location();
204     /// assert_eq!(fixed_location.file(), file!());
205     /// assert_eq!(fixed_location.line(), 15);
206     /// assert_eq!(fixed_location.column(), 5);
207     ///
208     /// // running the same untracked function in a different location gives us the same result
209     /// let second_fixed_location = get_just_one_location();
210     /// assert_eq!(fixed_location.file(), second_fixed_location.file());
211     /// assert_eq!(fixed_location.line(), second_fixed_location.line());
212     /// assert_eq!(fixed_location.column(), second_fixed_location.column());
213     ///
214     /// let this_location = get_caller_location();
215     /// assert_eq!(this_location.file(), file!());
216     /// assert_eq!(this_location.line(), 29);
217     /// assert_eq!(this_location.column(), 21);
218     ///
219     /// // running the tracked function in a different location produces a different value
220     /// let another_location = get_caller_location();
221     /// assert_eq!(this_location.file(), another_location.file());
222     /// assert_ne!(this_location.line(), another_location.line());
223     /// assert_ne!(this_location.column(), another_location.column());
224     /// ```
225     #[unstable(
226         feature = "track_caller",
227         reason = "uses #[track_caller] which is not yet stable",
228         issue = "47809"
229     )]
230     #[track_caller]
231     pub const fn caller() -> &'static Location<'static> {
232         crate::intrinsics::caller_location()
233     }
234 }
235
236 impl<'a> Location<'a> {
237     #![unstable(
238         feature = "panic_internals",
239         reason = "internal details of the implementation of the `panic!` \
240                           and related macros",
241         issue = "none"
242     )]
243     #[doc(hidden)]
244     pub const fn internal_constructor(file: &'a str, line: u32, col: u32) -> Self {
245         Location { file, line, col }
246     }
247
248     /// Returns the name of the source file from which the panic originated.
249     ///
250     /// # Examples
251     ///
252     /// ```should_panic
253     /// use std::panic;
254     ///
255     /// panic::set_hook(Box::new(|panic_info| {
256     ///     if let Some(location) = panic_info.location() {
257     ///         println!("panic occurred in file '{}'", location.file());
258     ///     } else {
259     ///         println!("panic occurred but can't get location information...");
260     ///     }
261     /// }));
262     ///
263     /// panic!("Normal panic");
264     /// ```
265     #[stable(feature = "panic_hooks", since = "1.10.0")]
266     pub fn file(&self) -> &str {
267         self.file
268     }
269
270     /// Returns the line number from which the panic originated.
271     ///
272     /// # Examples
273     ///
274     /// ```should_panic
275     /// use std::panic;
276     ///
277     /// panic::set_hook(Box::new(|panic_info| {
278     ///     if let Some(location) = panic_info.location() {
279     ///         println!("panic occurred at line {}", location.line());
280     ///     } else {
281     ///         println!("panic occurred but can't get location information...");
282     ///     }
283     /// }));
284     ///
285     /// panic!("Normal panic");
286     /// ```
287     #[stable(feature = "panic_hooks", since = "1.10.0")]
288     pub fn line(&self) -> u32 {
289         self.line
290     }
291
292     /// Returns the column from which the panic originated.
293     ///
294     /// # Examples
295     ///
296     /// ```should_panic
297     /// use std::panic;
298     ///
299     /// panic::set_hook(Box::new(|panic_info| {
300     ///     if let Some(location) = panic_info.location() {
301     ///         println!("panic occurred at column {}", location.column());
302     ///     } else {
303     ///         println!("panic occurred but can't get location information...");
304     ///     }
305     /// }));
306     ///
307     /// panic!("Normal panic");
308     /// ```
309     #[stable(feature = "panic_col", since = "1.25.0")]
310     pub fn column(&self) -> u32 {
311         self.col
312     }
313 }
314
315 #[stable(feature = "panic_hook_display", since = "1.26.0")]
316 impl fmt::Display for Location<'_> {
317     fn fmt(&self, formatter: &mut fmt::Formatter<'_>) -> fmt::Result {
318         write!(formatter, "{}:{}:{}", self.file, self.line, self.col)
319     }
320 }
321
322 /// An internal trait used by libstd to pass data from libstd to `panic_unwind`
323 /// and other panic runtimes. Not intended to be stabilized any time soon, do
324 /// not use.
325 #[unstable(feature = "std_internals", issue = "none")]
326 #[doc(hidden)]
327 pub unsafe trait BoxMeUp {
328     /// Take full ownership of the contents.
329     /// The return type is actually `Box<dyn Any + Send>`, but we cannot use `Box` in libcore.
330     ///
331     /// After this method got called, only some dummy default value is left in `self`.
332     /// Calling this method twice, or calling `get` after calling this method, is an error.
333     ///
334     /// The argument is borrowed because the panic runtime (`__rust_start_panic`) only
335     /// gets a borrowed `dyn BoxMeUp`.
336     fn take_box(&mut self) -> *mut (dyn Any + Send);
337
338     /// Just borrow the contents.
339     fn get(&mut self) -> &(dyn Any + Send);
340 }