//! Macro support for format strings //! //! These structures are used when parsing format strings for the compiler. //! Parsing does not happen at runtime: structures of `std::fmt::rt` are //! generated instead. #![doc( html_root_url = "https://doc.rust-lang.org/nightly/nightly-rustc/", html_playground_url = "https://play.rust-lang.org/", test(attr(deny(warnings))) )] #![deny(rustc::untranslatable_diagnostic)] #![deny(rustc::diagnostic_outside_of_impl)] // We want to be able to build this crate with a stable compiler, so no // `#![feature]` attributes should be added. pub use Alignment::*; pub use Count::*; pub use Flag::*; pub use Piece::*; pub use Position::*; use std::iter; use std::str; use std::string; // Note: copied from rustc_span /// Range inside of a `Span` used for diagnostics when we only have access to relative positions. #[derive(Copy, Clone, PartialEq, Eq, Debug)] pub struct InnerSpan { pub start: usize, pub end: usize, } impl InnerSpan { pub fn new(start: usize, end: usize) -> InnerSpan { InnerSpan { start, end } } } /// The type of format string that we are parsing. #[derive(Copy, Clone, Debug, Eq, PartialEq)] pub enum ParseMode { /// A normal format string as per `format_args!`. Format, /// An inline assembly template string for `asm!`. InlineAsm, } #[derive(Copy, Clone)] struct InnerOffset(usize); impl InnerOffset { fn to(self, end: InnerOffset) -> InnerSpan { InnerSpan::new(self.0, end.0) } } /// A piece is a portion of the format string which represents the next part /// to emit. These are emitted as a stream by the `Parser` class. #[derive(Copy, Clone, Debug, PartialEq)] pub enum Piece<'a> { /// A literal string which should directly be emitted String(&'a str), /// This describes that formatting should process the next argument (as /// specified inside) for emission. NextArgument(Argument<'a>), } /// Representation of an argument specification. #[derive(Copy, Clone, Debug, PartialEq)] pub struct Argument<'a> { /// Where to find this argument pub position: Position<'a>, /// The span of the position indicator. Includes any whitespace in implicit /// positions (`{ }`). pub position_span: InnerSpan, /// How to format the argument pub format: FormatSpec<'a>, } /// Specification for the formatting of an argument in the format string. #[derive(Copy, Clone, Debug, PartialEq)] pub struct FormatSpec<'a> { /// Optionally specified character to fill alignment with. pub fill: Option, /// Optionally specified alignment. pub align: Alignment, /// Packed version of various flags provided. pub flags: u32, /// The integer precision to use. pub precision: Count<'a>, /// The span of the precision formatting flag (for diagnostics). pub precision_span: Option, /// The string width requested for the resulting format. pub width: Count<'a>, /// The span of the width formatting flag (for diagnostics). pub width_span: Option, /// The descriptor string representing the name of the format desired for /// this argument, this can be empty or any number of characters, although /// it is required to be one word. pub ty: &'a str, /// The span of the descriptor string (for diagnostics). pub ty_span: Option, } /// Enum describing where an argument for a format can be located. #[derive(Copy, Clone, Debug, PartialEq)] pub enum Position<'a> { /// The argument is implied to be located at an index ArgumentImplicitlyIs(usize), /// The argument is located at a specific index given in the format, ArgumentIs(usize), /// The argument has a name. ArgumentNamed(&'a str), } impl Position<'_> { pub fn index(&self) -> Option { match self { ArgumentIs(i, ..) | ArgumentImplicitlyIs(i) => Some(*i), _ => None, } } } /// Enum of alignments which are supported. #[derive(Copy, Clone, Debug, PartialEq)] pub enum Alignment { /// The value will be aligned to the left. AlignLeft, /// The value will be aligned to the right. AlignRight, /// The value will be aligned in the center. AlignCenter, /// The value will take on a default alignment. AlignUnknown, } /// Various flags which can be applied to format strings. The meaning of these /// flags is defined by the formatters themselves. #[derive(Copy, Clone, Debug, PartialEq)] pub enum Flag { /// A `+` will be used to denote positive numbers. FlagSignPlus, /// A `-` will be used to denote negative numbers. This is the default. FlagSignMinus, /// An alternate form will be used for the value. In the case of numbers, /// this means that the number will be prefixed with the supplied string. FlagAlternate, /// For numbers, this means that the number will be padded with zeroes, /// and the sign (`+` or `-`) will precede them. FlagSignAwareZeroPad, /// For Debug / `?`, format integers in lower-case hexadecimal. FlagDebugLowerHex, /// For Debug / `?`, format integers in upper-case hexadecimal. FlagDebugUpperHex, } /// A count is used for the precision and width parameters of an integer, and /// can reference either an argument or a literal integer. #[derive(Copy, Clone, Debug, PartialEq)] pub enum Count<'a> { /// The count is specified explicitly. CountIs(usize), /// The count is specified by the argument with the given name. CountIsName(&'a str, InnerSpan), /// The count is specified by the argument at the given index. CountIsParam(usize), /// The count is specified by a star (like in `{:.*}`) that refers to the argument at the given index. CountIsStar(usize), /// The count is implied and cannot be explicitly specified. CountImplied, } pub struct ParseError { pub description: string::String, pub note: Option, pub label: string::String, pub span: InnerSpan, pub secondary_label: Option<(string::String, InnerSpan)>, pub should_be_replaced_with_positional_argument: bool, } /// The parser structure for interpreting the input format string. This is /// modeled as an iterator over `Piece` structures to form a stream of tokens /// being output. /// /// This is a recursive-descent parser for the sake of simplicity, and if /// necessary there's probably lots of room for improvement performance-wise. pub struct Parser<'a> { mode: ParseMode, input: &'a str, cur: iter::Peekable>, /// Error messages accumulated during parsing pub errors: Vec, /// Current position of implicit positional argument pointer pub curarg: usize, /// `Some(raw count)` when the string is "raw", used to position spans correctly style: Option, /// Start and end byte offset of every successfully parsed argument pub arg_places: Vec, /// Characters that need to be shifted skips: Vec, /// Span of the last opening brace seen, used for error reporting last_opening_brace: Option, /// Whether the source string is comes from `println!` as opposed to `format!` or `print!` append_newline: bool, /// Whether this formatting string is a literal or it comes from a macro. pub is_literal: bool, /// Start position of the current line. cur_line_start: usize, /// Start and end byte offset of every line of the format string. Excludes /// newline characters and leading whitespace. pub line_spans: Vec, } impl<'a> Iterator for Parser<'a> { type Item = Piece<'a>; fn next(&mut self) -> Option> { if let Some(&(pos, c)) = self.cur.peek() { match c { '{' => { let curr_last_brace = self.last_opening_brace; let byte_pos = self.to_span_index(pos); let lbrace_end = self.to_span_index(pos + 1); self.last_opening_brace = Some(byte_pos.to(lbrace_end)); self.cur.next(); if self.consume('{') { self.last_opening_brace = curr_last_brace; Some(String(self.string(pos + 1))) } else { let arg = self.argument(lbrace_end); if let Some(rbrace_byte_idx) = self.must_consume('}') { let lbrace_inner_offset = self.to_span_index(pos); let rbrace_inner_offset = self.to_span_index(rbrace_byte_idx); if self.is_literal { self.arg_places.push( lbrace_inner_offset.to(InnerOffset(rbrace_inner_offset.0 + 1)), ); } } else { self.suggest_positional_arg_instead_of_captured_arg(arg); } Some(NextArgument(arg)) } } '}' => { self.cur.next(); if self.consume('}') { Some(String(self.string(pos + 1))) } else { let err_pos = self.to_span_index(pos); self.err_with_note( "unmatched `}` found", "unmatched `}`", "if you intended to print `}`, you can escape it using `}}`", err_pos.to(err_pos), ); None } } _ => Some(String(self.string(pos))), } } else { if self.is_literal { let span = self.span(self.cur_line_start, self.input.len()); if self.line_spans.last() != Some(&span) { self.line_spans.push(span); } } None } } } impl<'a> Parser<'a> { /// Creates a new parser for the given format string pub fn new( s: &'a str, style: Option, snippet: Option, append_newline: bool, mode: ParseMode, ) -> Parser<'a> { let (skips, is_literal) = find_skips_from_snippet(snippet, style); Parser { mode, input: s, cur: s.char_indices().peekable(), errors: vec![], curarg: 0, style, arg_places: vec![], skips, last_opening_brace: None, append_newline, is_literal, cur_line_start: 0, line_spans: vec![], } } /// Notifies of an error. The message doesn't actually need to be of type /// String, but I think it does when this eventually uses conditions so it /// might as well start using it now. fn err, S2: Into>( &mut self, description: S1, label: S2, span: InnerSpan, ) { self.errors.push(ParseError { description: description.into(), note: None, label: label.into(), span, secondary_label: None, should_be_replaced_with_positional_argument: false, }); } /// Notifies of an error. The message doesn't actually need to be of type /// String, but I think it does when this eventually uses conditions so it /// might as well start using it now. fn err_with_note< S1: Into, S2: Into, S3: Into, >( &mut self, description: S1, label: S2, note: S3, span: InnerSpan, ) { self.errors.push(ParseError { description: description.into(), note: Some(note.into()), label: label.into(), span, secondary_label: None, should_be_replaced_with_positional_argument: false, }); } /// Optionally consumes the specified character. If the character is not at /// the current position, then the current iterator isn't moved and `false` is /// returned, otherwise the character is consumed and `true` is returned. fn consume(&mut self, c: char) -> bool { self.consume_pos(c).is_some() } /// Optionally consumes the specified character. If the character is not at /// the current position, then the current iterator isn't moved and `None` is /// returned, otherwise the character is consumed and the current position is /// returned. fn consume_pos(&mut self, c: char) -> Option { if let Some(&(pos, maybe)) = self.cur.peek() { if c == maybe { self.cur.next(); return Some(pos); } } None } fn to_span_index(&self, pos: usize) -> InnerOffset { let mut pos = pos; // This handles the raw string case, the raw argument is the number of # // in r###"..."### (we need to add one because of the `r`). let raw = self.style.map_or(0, |raw| raw + 1); for skip in &self.skips { if pos > *skip { pos += 1; } else if pos == *skip && raw == 0 { pos += 1; } else { break; } } InnerOffset(raw + pos + 1) } fn span(&self, start_pos: usize, end_pos: usize) -> InnerSpan { let start = self.to_span_index(start_pos); let end = self.to_span_index(end_pos); start.to(end) } /// Forces consumption of the specified character. If the character is not /// found, an error is emitted. fn must_consume(&mut self, c: char) -> Option { self.ws(); if let Some(&(pos, maybe)) = self.cur.peek() { if c == maybe { self.cur.next(); Some(pos) } else { let pos = self.to_span_index(pos); let description = format!("expected `'}}'`, found `{:?}`", maybe); let label = "expected `}`".to_owned(); let (note, secondary_label) = if c == '}' { ( Some( "if you intended to print `{`, you can escape it using `{{`".to_owned(), ), self.last_opening_brace .map(|sp| ("because of this opening brace".to_owned(), sp)), ) } else { (None, None) }; self.errors.push(ParseError { description, note, label, span: pos.to(pos), secondary_label, should_be_replaced_with_positional_argument: false, }); None } } else { let description = format!("expected `{:?}` but string was terminated", c); // point at closing `"` let pos = self.input.len() - if self.append_newline { 1 } else { 0 }; let pos = self.to_span_index(pos); if c == '}' { let label = format!("expected `{:?}`", c); let (note, secondary_label) = if c == '}' { ( Some( "if you intended to print `{`, you can escape it using `{{`".to_owned(), ), self.last_opening_brace .map(|sp| ("because of this opening brace".to_owned(), sp)), ) } else { (None, None) }; self.errors.push(ParseError { description, note, label, span: pos.to(pos), secondary_label, should_be_replaced_with_positional_argument: false, }); } else { self.err(description, format!("expected `{:?}`", c), pos.to(pos)); } None } } /// Consumes all whitespace characters until the first non-whitespace character fn ws(&mut self) { while let Some(&(_, c)) = self.cur.peek() { if c.is_whitespace() { self.cur.next(); } else { break; } } } /// Parses all of a string which is to be considered a "raw literal" in a /// format string. This is everything outside of the braces. fn string(&mut self, start: usize) -> &'a str { // we may not consume the character, peek the iterator while let Some(&(pos, c)) = self.cur.peek() { match c { '{' | '}' => { return &self.input[start..pos]; } '\n' if self.is_literal => { self.line_spans.push(self.span(self.cur_line_start, pos)); self.cur_line_start = pos + 1; self.cur.next(); } _ => { if self.is_literal && pos == self.cur_line_start && c.is_whitespace() { self.cur_line_start = pos + c.len_utf8(); } self.cur.next(); } } } &self.input[start..self.input.len()] } /// Parses an `Argument` structure, or what's contained within braces inside the format string. fn argument(&mut self, start: InnerOffset) -> Argument<'a> { let pos = self.position(); let end = self .cur .clone() .find(|(_, ch)| !ch.is_whitespace()) .map_or(start, |(end, _)| self.to_span_index(end)); let position_span = start.to(end); let format = match self.mode { ParseMode::Format => self.format(), ParseMode::InlineAsm => self.inline_asm(), }; // Resolve position after parsing format spec. let pos = match pos { Some(position) => position, None => { let i = self.curarg; self.curarg += 1; ArgumentImplicitlyIs(i) } }; Argument { position: pos, position_span, format } } /// Parses a positional argument for a format. This could either be an /// integer index of an argument, a named argument, or a blank string. /// Returns `Some(parsed_position)` if the position is not implicitly /// consuming a macro argument, `None` if it's the case. fn position(&mut self) -> Option> { if let Some(i) = self.integer() { Some(ArgumentIs(i)) } else { match self.cur.peek() { Some(&(_, c)) if rustc_lexer::is_id_start(c) => Some(ArgumentNamed(self.word())), // This is an `ArgumentNext`. // Record the fact and do the resolution after parsing the // format spec, to make things like `{:.*}` work. _ => None, } } } fn current_pos(&mut self) -> usize { if let Some(&(pos, _)) = self.cur.peek() { pos } else { self.input.len() } } /// Parses a format specifier at the current position, returning all of the /// relevant information in the `FormatSpec` struct. fn format(&mut self) -> FormatSpec<'a> { let mut spec = FormatSpec { fill: None, align: AlignUnknown, flags: 0, precision: CountImplied, precision_span: None, width: CountImplied, width_span: None, ty: &self.input[..0], ty_span: None, }; if !self.consume(':') { return spec; } // fill character if let Some(&(_, c)) = self.cur.peek() { if let Some((_, '>' | '<' | '^')) = self.cur.clone().nth(1) { spec.fill = Some(c); self.cur.next(); } } // Alignment if self.consume('<') { spec.align = AlignLeft; } else if self.consume('>') { spec.align = AlignRight; } else if self.consume('^') { spec.align = AlignCenter; } // Sign flags if self.consume('+') { spec.flags |= 1 << (FlagSignPlus as u32); } else if self.consume('-') { spec.flags |= 1 << (FlagSignMinus as u32); } // Alternate marker if self.consume('#') { spec.flags |= 1 << (FlagAlternate as u32); } // Width and precision let mut havewidth = false; if self.consume('0') { // small ambiguity with '0$' as a format string. In theory this is a // '0' flag and then an ill-formatted format string with just a '$' // and no count, but this is better if we instead interpret this as // no '0' flag and '0$' as the width instead. if let Some(end) = self.consume_pos('$') { spec.width = CountIsParam(0); spec.width_span = Some(self.span(end - 1, end + 1)); havewidth = true; } else { spec.flags |= 1 << (FlagSignAwareZeroPad as u32); } } if !havewidth { let start = self.current_pos(); spec.width = self.count(start); if spec.width != CountImplied { let end = self.current_pos(); spec.width_span = Some(self.span(start, end)); } } if let Some(start) = self.consume_pos('.') { if self.consume('*') { // Resolve `CountIsNextParam`. // We can do this immediately as `position` is resolved later. let i = self.curarg; self.curarg += 1; spec.precision = CountIsStar(i); } else { spec.precision = self.count(start + 1); } let end = self.current_pos(); spec.precision_span = Some(self.span(start, end)); } let ty_span_start = self.current_pos(); // Optional radix followed by the actual format specifier if self.consume('x') { if self.consume('?') { spec.flags |= 1 << (FlagDebugLowerHex as u32); spec.ty = "?"; } else { spec.ty = "x"; } } else if self.consume('X') { if self.consume('?') { spec.flags |= 1 << (FlagDebugUpperHex as u32); spec.ty = "?"; } else { spec.ty = "X"; } } else if self.consume('?') { spec.ty = "?"; } else { spec.ty = self.word(); if !spec.ty.is_empty() { let ty_span_end = self.current_pos(); spec.ty_span = Some(self.span(ty_span_start, ty_span_end)); } } spec } /// Parses an inline assembly template modifier at the current position, returning the modifier /// in the `ty` field of the `FormatSpec` struct. fn inline_asm(&mut self) -> FormatSpec<'a> { let mut spec = FormatSpec { fill: None, align: AlignUnknown, flags: 0, precision: CountImplied, precision_span: None, width: CountImplied, width_span: None, ty: &self.input[..0], ty_span: None, }; if !self.consume(':') { return spec; } let ty_span_start = self.current_pos(); spec.ty = self.word(); if !spec.ty.is_empty() { let ty_span_end = self.current_pos(); spec.ty_span = Some(self.span(ty_span_start, ty_span_end)); } spec } /// Parses a `Count` parameter at the current position. This does not check /// for 'CountIsNextParam' because that is only used in precision, not /// width. fn count(&mut self, start: usize) -> Count<'a> { if let Some(i) = self.integer() { if self.consume('$') { CountIsParam(i) } else { CountIs(i) } } else { let tmp = self.cur.clone(); let word = self.word(); if word.is_empty() { self.cur = tmp; CountImplied } else if let Some(end) = self.consume_pos('$') { let name_span = self.span(start, end); CountIsName(word, name_span) } else { self.cur = tmp; CountImplied } } } /// Parses a word starting at the current position. A word is the same as /// Rust identifier, except that it can't start with `_` character. fn word(&mut self) -> &'a str { let start = match self.cur.peek() { Some(&(pos, c)) if rustc_lexer::is_id_start(c) => { self.cur.next(); pos } _ => { return ""; } }; let mut end = None; while let Some(&(pos, c)) = self.cur.peek() { if rustc_lexer::is_id_continue(c) { self.cur.next(); } else { end = Some(pos); break; } } let end = end.unwrap_or(self.input.len()); let word = &self.input[start..end]; if word == "_" { self.err_with_note( "invalid argument name `_`", "invalid argument name", "argument name cannot be a single underscore", self.span(start, end), ); } word } fn integer(&mut self) -> Option { let mut cur: usize = 0; let mut found = false; let mut overflow = false; let start = self.current_pos(); while let Some(&(_, c)) = self.cur.peek() { if let Some(i) = c.to_digit(10) { let (tmp, mul_overflow) = cur.overflowing_mul(10); let (tmp, add_overflow) = tmp.overflowing_add(i as usize); if mul_overflow || add_overflow { overflow = true; } cur = tmp; found = true; self.cur.next(); } else { break; } } if overflow { let end = self.current_pos(); let overflowed_int = &self.input[start..end]; self.err( format!( "integer `{}` does not fit into the type `usize` whose range is `0..={}`", overflowed_int, usize::MAX ), "integer out of range for `usize`", self.span(start, end), ); } if found { Some(cur) } else { None } } fn suggest_positional_arg_instead_of_captured_arg(&mut self, arg: Argument<'a>) { if let Some(end) = self.consume_pos('.') { let byte_pos = self.to_span_index(end); let start = InnerOffset(byte_pos.0 + 1); let field = self.argument(start); // We can only parse `foo.bar` field access, any deeper nesting, // or another type of expression, like method calls, are not supported if !self.consume('}') { return; } if let ArgumentNamed(_) = arg.position { if let ArgumentNamed(_) = field.position { self.errors.insert( 0, ParseError { description: "field access isn't supported".to_string(), note: None, label: "not supported".to_string(), span: InnerSpan::new(arg.position_span.start, field.position_span.end), secondary_label: None, should_be_replaced_with_positional_argument: true, }, ); } } } } } /// Finds the indices of all characters that have been processed and differ between the actual /// written code (code snippet) and the `InternedString` that gets processed in the `Parser` /// in order to properly synthesise the intra-string `Span`s for error diagnostics. fn find_skips_from_snippet( snippet: Option, str_style: Option, ) -> (Vec, bool) { let snippet = match snippet { Some(ref s) if s.starts_with('"') || s.starts_with("r\"") || s.starts_with("r#") => s, _ => return (vec![], false), }; if str_style.is_some() { return (vec![], true); } let snippet = &snippet[1..snippet.len() - 1]; let mut s = snippet.char_indices(); let mut skips = vec![]; while let Some((pos, c)) = s.next() { match (c, s.clone().next()) { // skip whitespace and empty lines ending in '\\' ('\\', Some((next_pos, '\n'))) => { skips.push(pos); skips.push(next_pos); let _ = s.next(); while let Some((pos, c)) = s.clone().next() { if matches!(c, ' ' | '\n' | '\t') { skips.push(pos); let _ = s.next(); } else { break; } } } ('\\', Some((next_pos, 'n' | 't' | 'r' | '0' | '\\' | '\'' | '\"'))) => { skips.push(next_pos); let _ = s.next(); } ('\\', Some((_, 'x'))) => { for _ in 0..3 { // consume `\xAB` literal if let Some((pos, _)) = s.next() { skips.push(pos); } else { break; } } } ('\\', Some((_, 'u'))) => { if let Some((pos, _)) = s.next() { skips.push(pos); } if let Some((next_pos, next_c)) = s.next() { if next_c == '{' { // consume up to 6 hexanumeric chars let digits_len = s.clone().take(6).take_while(|(_, c)| c.is_digit(16)).count(); let len_utf8 = s .as_str() .get(..digits_len) .and_then(|digits| u32::from_str_radix(digits, 16).ok()) .and_then(char::from_u32) .map_or(1, char::len_utf8); // Skip the digits, for chars that encode to more than 1 utf-8 byte // exclude as many digits as it is greater than 1 byte // // So for a 3 byte character, exclude 2 digits let required_skips = digits_len.saturating_sub(len_utf8.saturating_sub(1)); // skip '{' and '}' also for pos in (next_pos..).take(required_skips + 2) { skips.push(pos) } s.nth(digits_len); } else if next_c.is_digit(16) { skips.push(next_pos); // We suggest adding `{` and `}` when appropriate, accept it here as if // it were correct let mut i = 0; // consume up to 6 hexanumeric chars while let (Some((next_pos, c)), _) = (s.next(), i < 6) { if c.is_digit(16) { skips.push(next_pos); } else { break; } i += 1; } } } } _ => {} } } (skips, true) } #[cfg(test)] mod tests;