c225824bde
[breaking-change]: instead of a closure like `|| -> i32 22`, prefer `|| -> i32 { 22 }`. Fixes #23420.
1175 lines
44 KiB
Rust
1175 lines
44 KiB
Rust
// Copyright 2012-2014 The Rust Project Developers. See the COPYRIGHT
|
|
// file at the top-level directory of this distribution and at
|
|
// http://rust-lang.org/COPYRIGHT.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
|
|
// http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
|
|
// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
|
|
// option. This file may not be copied, modified, or distributed
|
|
// except according to those terms.
|
|
|
|
//! The main parser interface
|
|
|
|
use ast;
|
|
use codemap::{Span, CodeMap, FileMap};
|
|
use diagnostic::{SpanHandler, mk_span_handler, default_handler, Auto};
|
|
use parse::attr::ParserAttr;
|
|
use parse::parser::Parser;
|
|
use ptr::P;
|
|
|
|
use std::cell::{Cell, RefCell};
|
|
use std::fs::File;
|
|
use std::io::Read;
|
|
use std::iter;
|
|
use std::num::Int;
|
|
use std::path::{Path, PathBuf};
|
|
use std::rc::Rc;
|
|
use std::str;
|
|
|
|
#[macro_use]
|
|
pub mod parser;
|
|
|
|
pub mod lexer;
|
|
pub mod token;
|
|
pub mod attr;
|
|
|
|
pub mod common;
|
|
pub mod classify;
|
|
pub mod obsolete;
|
|
|
|
/// Info about a parsing session.
|
|
pub struct ParseSess {
|
|
pub span_diagnostic: SpanHandler, // better be the same as the one in the reader!
|
|
/// Used to determine and report recursive mod inclusions
|
|
included_mod_stack: RefCell<Vec<PathBuf>>,
|
|
pub node_id: Cell<ast::NodeId>,
|
|
}
|
|
|
|
pub fn new_parse_sess() -> ParseSess {
|
|
ParseSess {
|
|
span_diagnostic: mk_span_handler(default_handler(Auto, None, true), CodeMap::new()),
|
|
included_mod_stack: RefCell::new(Vec::new()),
|
|
node_id: Cell::new(1),
|
|
}
|
|
}
|
|
|
|
pub fn new_parse_sess_special_handler(sh: SpanHandler) -> ParseSess {
|
|
ParseSess {
|
|
span_diagnostic: sh,
|
|
included_mod_stack: RefCell::new(Vec::new()),
|
|
node_id: Cell::new(1),
|
|
}
|
|
}
|
|
|
|
impl ParseSess {
|
|
pub fn next_node_id(&self) -> ast::NodeId {
|
|
self.reserve_node_ids(1)
|
|
}
|
|
pub fn reserve_node_ids(&self, count: ast::NodeId) -> ast::NodeId {
|
|
let v = self.node_id.get();
|
|
|
|
match v.checked_add(count) {
|
|
Some(next) => { self.node_id.set(next); }
|
|
None => panic!("Input too large, ran out of node ids!")
|
|
}
|
|
|
|
v
|
|
}
|
|
}
|
|
|
|
// a bunch of utility functions of the form parse_<thing>_from_<source>
|
|
// where <thing> includes crate, expr, item, stmt, tts, and one that
|
|
// uses a HOF to parse anything, and <source> includes file and
|
|
// source_str.
|
|
|
|
pub fn parse_crate_from_file(
|
|
input: &Path,
|
|
cfg: ast::CrateConfig,
|
|
sess: &ParseSess
|
|
) -> ast::Crate {
|
|
new_parser_from_file(sess, cfg, input).parse_crate_mod()
|
|
// why is there no p.abort_if_errors here?
|
|
}
|
|
|
|
pub fn parse_crate_attrs_from_file(
|
|
input: &Path,
|
|
cfg: ast::CrateConfig,
|
|
sess: &ParseSess
|
|
) -> Vec<ast::Attribute> {
|
|
new_parser_from_file(sess, cfg, input).parse_inner_attributes()
|
|
}
|
|
|
|
pub fn parse_crate_from_source_str(name: String,
|
|
source: String,
|
|
cfg: ast::CrateConfig,
|
|
sess: &ParseSess)
|
|
-> ast::Crate {
|
|
let mut p = new_parser_from_source_str(sess,
|
|
cfg,
|
|
name,
|
|
source);
|
|
maybe_aborted(p.parse_crate_mod(),p)
|
|
}
|
|
|
|
pub fn parse_crate_attrs_from_source_str(name: String,
|
|
source: String,
|
|
cfg: ast::CrateConfig,
|
|
sess: &ParseSess)
|
|
-> Vec<ast::Attribute> {
|
|
let mut p = new_parser_from_source_str(sess,
|
|
cfg,
|
|
name,
|
|
source);
|
|
maybe_aborted(p.parse_inner_attributes(), p)
|
|
}
|
|
|
|
pub fn parse_expr_from_source_str(name: String,
|
|
source: String,
|
|
cfg: ast::CrateConfig,
|
|
sess: &ParseSess)
|
|
-> P<ast::Expr> {
|
|
let mut p = new_parser_from_source_str(sess, cfg, name, source);
|
|
maybe_aborted(p.parse_expr(), p)
|
|
}
|
|
|
|
pub fn parse_item_from_source_str(name: String,
|
|
source: String,
|
|
cfg: ast::CrateConfig,
|
|
sess: &ParseSess)
|
|
-> Option<P<ast::Item>> {
|
|
let mut p = new_parser_from_source_str(sess, cfg, name, source);
|
|
maybe_aborted(p.parse_item(),p)
|
|
}
|
|
|
|
pub fn parse_meta_from_source_str(name: String,
|
|
source: String,
|
|
cfg: ast::CrateConfig,
|
|
sess: &ParseSess)
|
|
-> P<ast::MetaItem> {
|
|
let mut p = new_parser_from_source_str(sess, cfg, name, source);
|
|
maybe_aborted(p.parse_meta_item(),p)
|
|
}
|
|
|
|
pub fn parse_stmt_from_source_str(name: String,
|
|
source: String,
|
|
cfg: ast::CrateConfig,
|
|
sess: &ParseSess)
|
|
-> Option<P<ast::Stmt>> {
|
|
let mut p = new_parser_from_source_str(
|
|
sess,
|
|
cfg,
|
|
name,
|
|
source
|
|
);
|
|
maybe_aborted(p.parse_stmt(), p)
|
|
}
|
|
|
|
// Note: keep in sync with `with_hygiene::parse_tts_from_source_str`
|
|
// until #16472 is resolved.
|
|
//
|
|
// Warning: This parses with quote_depth > 0, which is not the default.
|
|
pub fn parse_tts_from_source_str(name: String,
|
|
source: String,
|
|
cfg: ast::CrateConfig,
|
|
sess: &ParseSess)
|
|
-> Vec<ast::TokenTree> {
|
|
let mut p = new_parser_from_source_str(
|
|
sess,
|
|
cfg,
|
|
name,
|
|
source
|
|
);
|
|
p.quote_depth += 1;
|
|
// right now this is re-creating the token trees from ... token trees.
|
|
maybe_aborted(p.parse_all_token_trees(),p)
|
|
}
|
|
|
|
// Note: keep in sync with `with_hygiene::new_parser_from_source_str`
|
|
// until #16472 is resolved.
|
|
// Create a new parser from a source string
|
|
pub fn new_parser_from_source_str<'a>(sess: &'a ParseSess,
|
|
cfg: ast::CrateConfig,
|
|
name: String,
|
|
source: String)
|
|
-> Parser<'a> {
|
|
filemap_to_parser(sess, string_to_filemap(sess, source, name), cfg)
|
|
}
|
|
|
|
/// Create a new parser, handling errors as appropriate
|
|
/// if the file doesn't exist
|
|
pub fn new_parser_from_file<'a>(sess: &'a ParseSess,
|
|
cfg: ast::CrateConfig,
|
|
path: &Path) -> Parser<'a> {
|
|
filemap_to_parser(sess, file_to_filemap(sess, path, None), cfg)
|
|
}
|
|
|
|
/// Given a session, a crate config, a path, and a span, add
|
|
/// the file at the given path to the codemap, and return a parser.
|
|
/// On an error, use the given span as the source of the problem.
|
|
pub fn new_sub_parser_from_file<'a>(sess: &'a ParseSess,
|
|
cfg: ast::CrateConfig,
|
|
path: &Path,
|
|
owns_directory: bool,
|
|
module_name: Option<String>,
|
|
sp: Span) -> Parser<'a> {
|
|
let mut p = filemap_to_parser(sess, file_to_filemap(sess, path, Some(sp)), cfg);
|
|
p.owns_directory = owns_directory;
|
|
p.root_module_name = module_name;
|
|
p
|
|
}
|
|
|
|
// Note: keep this in sync with `with_hygiene::filemap_to_parser` until
|
|
// #16472 is resolved.
|
|
/// Given a filemap and config, return a parser
|
|
pub fn filemap_to_parser<'a>(sess: &'a ParseSess,
|
|
filemap: Rc<FileMap>,
|
|
cfg: ast::CrateConfig) -> Parser<'a> {
|
|
tts_to_parser(sess, filemap_to_tts(sess, filemap), cfg)
|
|
}
|
|
|
|
// must preserve old name for now, because quote! from the *existing*
|
|
// compiler expands into it
|
|
pub fn new_parser_from_tts<'a>(sess: &'a ParseSess,
|
|
cfg: ast::CrateConfig,
|
|
tts: Vec<ast::TokenTree>) -> Parser<'a> {
|
|
tts_to_parser(sess, tts, cfg)
|
|
}
|
|
|
|
|
|
// base abstractions
|
|
|
|
/// Given a session and a path and an optional span (for error reporting),
|
|
/// add the path to the session's codemap and return the new filemap.
|
|
pub fn file_to_filemap(sess: &ParseSess, path: &Path, spanopt: Option<Span>)
|
|
-> Rc<FileMap> {
|
|
let err = |msg: &str| {
|
|
match spanopt {
|
|
Some(sp) => sess.span_diagnostic.span_fatal(sp, msg),
|
|
None => sess.span_diagnostic.handler().fatal(msg),
|
|
}
|
|
};
|
|
let mut bytes = Vec::new();
|
|
match File::open(path).and_then(|mut f| f.read_to_end(&mut bytes)) {
|
|
Ok(..) => {}
|
|
Err(e) => {
|
|
err(&format!("couldn't read {:?}: {}", path.display(), e));
|
|
unreachable!();
|
|
}
|
|
};
|
|
match str::from_utf8(&bytes[..]).ok() {
|
|
Some(s) => {
|
|
string_to_filemap(sess, s.to_string(),
|
|
path.to_str().unwrap().to_string())
|
|
}
|
|
None => {
|
|
err(&format!("{:?} is not UTF-8 encoded", path.display()));
|
|
unreachable!();
|
|
}
|
|
}
|
|
}
|
|
|
|
/// Given a session and a string, add the string to
|
|
/// the session's codemap and return the new filemap
|
|
pub fn string_to_filemap(sess: &ParseSess, source: String, path: String)
|
|
-> Rc<FileMap> {
|
|
sess.span_diagnostic.cm.new_filemap(path, source)
|
|
}
|
|
|
|
// Note: keep this in sync with `with_hygiene::filemap_to_tts` (apart
|
|
// from the StringReader constructor), until #16472 is resolved.
|
|
/// Given a filemap, produce a sequence of token-trees
|
|
pub fn filemap_to_tts(sess: &ParseSess, filemap: Rc<FileMap>)
|
|
-> Vec<ast::TokenTree> {
|
|
// it appears to me that the cfg doesn't matter here... indeed,
|
|
// parsing tt's probably shouldn't require a parser at all.
|
|
let cfg = Vec::new();
|
|
let srdr = lexer::StringReader::new(&sess.span_diagnostic, filemap);
|
|
let mut p1 = Parser::new(sess, cfg, box srdr);
|
|
p1.parse_all_token_trees()
|
|
}
|
|
|
|
/// Given tts and cfg, produce a parser
|
|
pub fn tts_to_parser<'a>(sess: &'a ParseSess,
|
|
tts: Vec<ast::TokenTree>,
|
|
cfg: ast::CrateConfig) -> Parser<'a> {
|
|
let trdr = lexer::new_tt_reader(&sess.span_diagnostic, None, None, tts);
|
|
let mut p = Parser::new(sess, cfg, box trdr);
|
|
p.check_unknown_macro_variable();
|
|
p
|
|
}
|
|
|
|
// FIXME (Issue #16472): The `with_hygiene` mod should go away after
|
|
// ToToken impls are revised to go directly to token-trees.
|
|
pub mod with_hygiene {
|
|
use ast;
|
|
use codemap::FileMap;
|
|
use parse::parser::Parser;
|
|
use std::rc::Rc;
|
|
use super::ParseSess;
|
|
use super::{maybe_aborted, string_to_filemap, tts_to_parser};
|
|
|
|
// Note: keep this in sync with `super::parse_tts_from_source_str` until
|
|
// #16472 is resolved.
|
|
//
|
|
// Warning: This parses with quote_depth > 0, which is not the default.
|
|
pub fn parse_tts_from_source_str(name: String,
|
|
source: String,
|
|
cfg: ast::CrateConfig,
|
|
sess: &ParseSess) -> Vec<ast::TokenTree> {
|
|
let mut p = new_parser_from_source_str(
|
|
sess,
|
|
cfg,
|
|
name,
|
|
source
|
|
);
|
|
p.quote_depth += 1;
|
|
// right now this is re-creating the token trees from ... token trees.
|
|
maybe_aborted(p.parse_all_token_trees(),p)
|
|
}
|
|
|
|
// Note: keep this in sync with `super::new_parser_from_source_str` until
|
|
// #16472 is resolved.
|
|
// Create a new parser from a source string
|
|
fn new_parser_from_source_str<'a>(sess: &'a ParseSess,
|
|
cfg: ast::CrateConfig,
|
|
name: String,
|
|
source: String) -> Parser<'a> {
|
|
filemap_to_parser(sess, string_to_filemap(sess, source, name), cfg)
|
|
}
|
|
|
|
// Note: keep this in sync with `super::filemap_to_parserr` until
|
|
// #16472 is resolved.
|
|
/// Given a filemap and config, return a parser
|
|
fn filemap_to_parser<'a>(sess: &'a ParseSess,
|
|
filemap: Rc<FileMap>,
|
|
cfg: ast::CrateConfig) -> Parser<'a> {
|
|
tts_to_parser(sess, filemap_to_tts(sess, filemap), cfg)
|
|
}
|
|
|
|
// Note: keep this in sync with `super::filemap_to_tts` until
|
|
// #16472 is resolved.
|
|
/// Given a filemap, produce a sequence of token-trees
|
|
fn filemap_to_tts(sess: &ParseSess, filemap: Rc<FileMap>)
|
|
-> Vec<ast::TokenTree> {
|
|
// it appears to me that the cfg doesn't matter here... indeed,
|
|
// parsing tt's probably shouldn't require a parser at all.
|
|
use super::lexer::make_reader_with_embedded_idents as make_reader;
|
|
let cfg = Vec::new();
|
|
let srdr = make_reader(&sess.span_diagnostic, filemap);
|
|
let mut p1 = Parser::new(sess, cfg, box srdr);
|
|
p1.parse_all_token_trees()
|
|
}
|
|
}
|
|
|
|
/// Abort if necessary
|
|
pub fn maybe_aborted<T>(result: T, p: Parser) -> T {
|
|
p.abort_if_errors();
|
|
result
|
|
}
|
|
|
|
/// Parse a string representing a character literal into its final form.
|
|
/// Rather than just accepting/rejecting a given literal, unescapes it as
|
|
/// well. Can take any slice prefixed by a character escape. Returns the
|
|
/// character and the number of characters consumed.
|
|
pub fn char_lit(lit: &str) -> (char, isize) {
|
|
use std::{num, char};
|
|
|
|
let mut chars = lit.chars();
|
|
let c = match (chars.next(), chars.next()) {
|
|
(Some(c), None) if c != '\\' => return (c, 1),
|
|
(Some('\\'), Some(c)) => match c {
|
|
'"' => Some('"'),
|
|
'n' => Some('\n'),
|
|
'r' => Some('\r'),
|
|
't' => Some('\t'),
|
|
'\\' => Some('\\'),
|
|
'\'' => Some('\''),
|
|
'0' => Some('\0'),
|
|
_ => { None }
|
|
},
|
|
_ => panic!("lexer accepted invalid char escape `{}`", lit)
|
|
};
|
|
|
|
match c {
|
|
Some(x) => return (x, 2),
|
|
None => { }
|
|
}
|
|
|
|
let msg = format!("lexer should have rejected a bad character escape {}", lit);
|
|
let msg2 = &msg[..];
|
|
|
|
fn esc(len: usize, lit: &str) -> Option<(char, isize)> {
|
|
num::from_str_radix(&lit[2..len], 16).ok()
|
|
.and_then(char::from_u32)
|
|
.map(|x| (x, len as isize))
|
|
}
|
|
|
|
let unicode_escape = || -> Option<(char, isize)> {
|
|
if lit.as_bytes()[2] == b'{' {
|
|
let idx = lit.find('}').expect(msg2);
|
|
let subslice = &lit[3..idx];
|
|
num::from_str_radix(subslice, 16).ok()
|
|
.and_then(char::from_u32)
|
|
.map(|x| (x, subslice.chars().count() as isize + 4))
|
|
} else {
|
|
esc(6, lit)
|
|
}
|
|
};
|
|
|
|
// Unicode escapes
|
|
return match lit.as_bytes()[1] as char {
|
|
'x' | 'X' => esc(4, lit),
|
|
'u' => unicode_escape(),
|
|
'U' => esc(10, lit),
|
|
_ => None,
|
|
}.expect(msg2);
|
|
}
|
|
|
|
/// Parse a string representing a string literal into its final form. Does
|
|
/// unescaping.
|
|
pub fn str_lit(lit: &str) -> String {
|
|
debug!("parse_str_lit: given {}", lit.escape_default());
|
|
let mut res = String::with_capacity(lit.len());
|
|
|
|
// FIXME #8372: This could be a for-loop if it didn't borrow the iterator
|
|
let error = |i| format!("lexer should have rejected {} at {}", lit, i);
|
|
|
|
/// Eat everything up to a non-whitespace
|
|
fn eat<'a>(it: &mut iter::Peekable<str::CharIndices<'a>>) {
|
|
loop {
|
|
match it.peek().map(|x| x.1) {
|
|
Some(' ') | Some('\n') | Some('\r') | Some('\t') => {
|
|
it.next();
|
|
},
|
|
_ => { break; }
|
|
}
|
|
}
|
|
}
|
|
|
|
let mut chars = lit.char_indices().peekable();
|
|
loop {
|
|
match chars.next() {
|
|
Some((i, c)) => {
|
|
match c {
|
|
'\\' => {
|
|
let ch = chars.peek().unwrap_or_else(|| {
|
|
panic!("{}", error(i))
|
|
}).1;
|
|
|
|
if ch == '\n' {
|
|
eat(&mut chars);
|
|
} else if ch == '\r' {
|
|
chars.next();
|
|
let ch = chars.peek().unwrap_or_else(|| {
|
|
panic!("{}", error(i))
|
|
}).1;
|
|
|
|
if ch != '\n' {
|
|
panic!("lexer accepted bare CR");
|
|
}
|
|
eat(&mut chars);
|
|
} else {
|
|
// otherwise, a normal escape
|
|
let (c, n) = char_lit(&lit[i..]);
|
|
for _ in 0..n - 1 { // we don't need to move past the first \
|
|
chars.next();
|
|
}
|
|
res.push(c);
|
|
}
|
|
},
|
|
'\r' => {
|
|
let ch = chars.peek().unwrap_or_else(|| {
|
|
panic!("{}", error(i))
|
|
}).1;
|
|
|
|
if ch != '\n' {
|
|
panic!("lexer accepted bare CR");
|
|
}
|
|
chars.next();
|
|
res.push('\n');
|
|
}
|
|
c => res.push(c),
|
|
}
|
|
},
|
|
None => break
|
|
}
|
|
}
|
|
|
|
res.shrink_to_fit(); // probably not going to do anything, unless there was an escape.
|
|
debug!("parse_str_lit: returning {}", res);
|
|
res
|
|
}
|
|
|
|
/// Parse a string representing a raw string literal into its final form. The
|
|
/// only operation this does is convert embedded CRLF into a single LF.
|
|
pub fn raw_str_lit(lit: &str) -> String {
|
|
debug!("raw_str_lit: given {}", lit.escape_default());
|
|
let mut res = String::with_capacity(lit.len());
|
|
|
|
// FIXME #8372: This could be a for-loop if it didn't borrow the iterator
|
|
let mut chars = lit.chars().peekable();
|
|
loop {
|
|
match chars.next() {
|
|
Some(c) => {
|
|
if c == '\r' {
|
|
if *chars.peek().unwrap() != '\n' {
|
|
panic!("lexer accepted bare CR");
|
|
}
|
|
chars.next();
|
|
res.push('\n');
|
|
} else {
|
|
res.push(c);
|
|
}
|
|
},
|
|
None => break
|
|
}
|
|
}
|
|
|
|
res.shrink_to_fit();
|
|
res
|
|
}
|
|
|
|
// check if `s` looks like i32 or u1234 etc.
|
|
fn looks_like_width_suffix(first_chars: &[char], s: &str) -> bool {
|
|
s.len() > 1 &&
|
|
first_chars.contains(&s.char_at(0)) &&
|
|
s[1..].chars().all(|c| '0' <= c && c <= '9')
|
|
}
|
|
|
|
fn filtered_float_lit(data: token::InternedString, suffix: Option<&str>,
|
|
sd: &SpanHandler, sp: Span) -> ast::Lit_ {
|
|
debug!("filtered_float_lit: {}, {:?}", data, suffix);
|
|
match suffix {
|
|
Some("f32") => ast::LitFloat(data, ast::TyF32),
|
|
Some("f64") => ast::LitFloat(data, ast::TyF64),
|
|
Some(suf) => {
|
|
if suf.len() >= 2 && looks_like_width_suffix(&['f'], suf) {
|
|
// if it looks like a width, lets try to be helpful.
|
|
sd.span_err(sp, &*format!("illegal width `{}` for float literal, \
|
|
valid widths are 32 and 64", &suf[1..]));
|
|
} else {
|
|
sd.span_err(sp, &*format!("illegal suffix `{}` for float literal, \
|
|
valid suffixes are `f32` and `f64`", suf));
|
|
}
|
|
|
|
ast::LitFloatUnsuffixed(data)
|
|
}
|
|
None => ast::LitFloatUnsuffixed(data)
|
|
}
|
|
}
|
|
pub fn float_lit(s: &str, suffix: Option<&str>, sd: &SpanHandler, sp: Span) -> ast::Lit_ {
|
|
debug!("float_lit: {:?}, {:?}", s, suffix);
|
|
// FIXME #2252: bounds checking float literals is deferred until trans
|
|
let s = s.chars().filter(|&c| c != '_').collect::<String>();
|
|
let data = token::intern_and_get_ident(&*s);
|
|
filtered_float_lit(data, suffix, sd, sp)
|
|
}
|
|
|
|
/// Parse a string representing a byte literal into its final form. Similar to `char_lit`
|
|
pub fn byte_lit(lit: &str) -> (u8, usize) {
|
|
let err = |i| format!("lexer accepted invalid byte literal {} step {}", lit, i);
|
|
|
|
if lit.len() == 1 {
|
|
(lit.as_bytes()[0], 1)
|
|
} else {
|
|
assert!(lit.as_bytes()[0] == b'\\', err(0));
|
|
let b = match lit.as_bytes()[1] {
|
|
b'"' => b'"',
|
|
b'n' => b'\n',
|
|
b'r' => b'\r',
|
|
b't' => b'\t',
|
|
b'\\' => b'\\',
|
|
b'\'' => b'\'',
|
|
b'0' => b'\0',
|
|
_ => {
|
|
match ::std::num::from_str_radix::<u64>(&lit[2..4], 16).ok() {
|
|
Some(c) =>
|
|
if c > 0xFF {
|
|
panic!(err(2))
|
|
} else {
|
|
return (c as u8, 4)
|
|
},
|
|
None => panic!(err(3))
|
|
}
|
|
}
|
|
};
|
|
return (b, 2);
|
|
}
|
|
}
|
|
|
|
pub fn binary_lit(lit: &str) -> Rc<Vec<u8>> {
|
|
let mut res = Vec::with_capacity(lit.len());
|
|
|
|
// FIXME #8372: This could be a for-loop if it didn't borrow the iterator
|
|
let error = |i| format!("lexer should have rejected {} at {}", lit, i);
|
|
|
|
/// Eat everything up to a non-whitespace
|
|
fn eat<'a, I: Iterator<Item=(usize, u8)>>(it: &mut iter::Peekable<I>) {
|
|
loop {
|
|
match it.peek().map(|x| x.1) {
|
|
Some(b' ') | Some(b'\n') | Some(b'\r') | Some(b'\t') => {
|
|
it.next();
|
|
},
|
|
_ => { break; }
|
|
}
|
|
}
|
|
}
|
|
|
|
// binary literals *must* be ASCII, but the escapes don't have to be
|
|
let mut chars = lit.bytes().enumerate().peekable();
|
|
loop {
|
|
match chars.next() {
|
|
Some((i, b'\\')) => {
|
|
let em = error(i);
|
|
match chars.peek().expect(&em).1 {
|
|
b'\n' => eat(&mut chars),
|
|
b'\r' => {
|
|
chars.next();
|
|
if chars.peek().expect(&em).1 != b'\n' {
|
|
panic!("lexer accepted bare CR");
|
|
}
|
|
eat(&mut chars);
|
|
}
|
|
_ => {
|
|
// otherwise, a normal escape
|
|
let (c, n) = byte_lit(&lit[i..]);
|
|
// we don't need to move past the first \
|
|
for _ in 0..n - 1 {
|
|
chars.next();
|
|
}
|
|
res.push(c);
|
|
}
|
|
}
|
|
},
|
|
Some((i, b'\r')) => {
|
|
let em = error(i);
|
|
if chars.peek().expect(&em).1 != b'\n' {
|
|
panic!("lexer accepted bare CR");
|
|
}
|
|
chars.next();
|
|
res.push(b'\n');
|
|
}
|
|
Some((_, c)) => res.push(c),
|
|
None => break,
|
|
}
|
|
}
|
|
|
|
Rc::new(res)
|
|
}
|
|
|
|
pub fn integer_lit(s: &str, suffix: Option<&str>, sd: &SpanHandler, sp: Span) -> ast::Lit_ {
|
|
// s can only be ascii, byte indexing is fine
|
|
|
|
let s2 = s.chars().filter(|&c| c != '_').collect::<String>();
|
|
let mut s = &s2[..];
|
|
|
|
debug!("integer_lit: {}, {:?}", s, suffix);
|
|
|
|
let mut base = 10;
|
|
let orig = s;
|
|
let mut ty = ast::UnsuffixedIntLit(ast::Plus);
|
|
|
|
if s.char_at(0) == '0' && s.len() > 1 {
|
|
match s.char_at(1) {
|
|
'x' => base = 16,
|
|
'o' => base = 8,
|
|
'b' => base = 2,
|
|
_ => { }
|
|
}
|
|
}
|
|
|
|
// 1f64 and 2f32 etc. are valid float literals.
|
|
match suffix {
|
|
Some(suf) if looks_like_width_suffix(&['f'], suf) => {
|
|
match base {
|
|
16 => sd.span_err(sp, "hexadecimal float literal is not supported"),
|
|
8 => sd.span_err(sp, "octal float literal is not supported"),
|
|
2 => sd.span_err(sp, "binary float literal is not supported"),
|
|
_ => ()
|
|
}
|
|
let ident = token::intern_and_get_ident(&*s);
|
|
return filtered_float_lit(ident, suffix, sd, sp)
|
|
}
|
|
_ => {}
|
|
}
|
|
|
|
if base != 10 {
|
|
s = &s[2..];
|
|
}
|
|
|
|
if let Some(suf) = suffix {
|
|
if suf.is_empty() { sd.span_bug(sp, "found empty literal suffix in Some")}
|
|
ty = match suf {
|
|
"isize" => ast::SignedIntLit(ast::TyIs(false), ast::Plus),
|
|
"i8" => ast::SignedIntLit(ast::TyI8, ast::Plus),
|
|
"i16" => ast::SignedIntLit(ast::TyI16, ast::Plus),
|
|
"i32" => ast::SignedIntLit(ast::TyI32, ast::Plus),
|
|
"i64" => ast::SignedIntLit(ast::TyI64, ast::Plus),
|
|
"usize" => ast::UnsignedIntLit(ast::TyUs(false)),
|
|
"u8" => ast::UnsignedIntLit(ast::TyU8),
|
|
"u16" => ast::UnsignedIntLit(ast::TyU16),
|
|
"u32" => ast::UnsignedIntLit(ast::TyU32),
|
|
"u64" => ast::UnsignedIntLit(ast::TyU64),
|
|
"i" | "is" => ast::SignedIntLit(ast::TyIs(true), ast::Plus),
|
|
"u" | "us" => ast::UnsignedIntLit(ast::TyUs(true)),
|
|
_ => {
|
|
// i<digits> and u<digits> look like widths, so lets
|
|
// give an error message along those lines
|
|
if looks_like_width_suffix(&['i', 'u'], suf) {
|
|
sd.span_err(sp, &*format!("illegal width `{}` for integer literal; \
|
|
valid widths are 8, 16, 32 and 64",
|
|
&suf[1..]));
|
|
} else {
|
|
sd.span_err(sp, &*format!("illegal suffix `{}` for numeric literal", suf));
|
|
sd.fileline_help(sp, "the suffix must be one of the integral types \
|
|
(`u32`, `isize`, etc)");
|
|
}
|
|
|
|
ty
|
|
}
|
|
}
|
|
}
|
|
|
|
debug!("integer_lit: the type is {:?}, base {:?}, the new string is {:?}, the original \
|
|
string was {:?}, the original suffix was {:?}", ty, base, s, orig, suffix);
|
|
|
|
let res: u64 = match ::std::num::from_str_radix(s, base).ok() {
|
|
Some(r) => r,
|
|
None => { sd.span_err(sp, "int literal is too large"); 0 }
|
|
};
|
|
|
|
// adjust the sign
|
|
let sign = ast::Sign::new(res);
|
|
match ty {
|
|
ast::SignedIntLit(t, _) => ast::LitInt(res, ast::SignedIntLit(t, sign)),
|
|
ast::UnsuffixedIntLit(_) => ast::LitInt(res, ast::UnsuffixedIntLit(sign)),
|
|
us@ast::UnsignedIntLit(_) => ast::LitInt(res, us)
|
|
}
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod test {
|
|
use super::*;
|
|
use std::rc::Rc;
|
|
use codemap::{Span, BytePos, Pos, Spanned, NO_EXPANSION};
|
|
use owned_slice::OwnedSlice;
|
|
use ast;
|
|
use abi;
|
|
use attr::{first_attr_value_str_by_name, AttrMetaMethods};
|
|
use parse;
|
|
use parse::parser::Parser;
|
|
use parse::token::{str_to_ident};
|
|
use print::pprust::item_to_string;
|
|
use ptr::P;
|
|
use util::parser_testing::{string_to_tts, string_to_parser};
|
|
use util::parser_testing::{string_to_expr, string_to_item, string_to_stmt};
|
|
|
|
// produce a codemap::span
|
|
fn sp(a: u32, b: u32) -> Span {
|
|
Span {lo: BytePos(a), hi: BytePos(b), expn_id: NO_EXPANSION}
|
|
}
|
|
|
|
#[test] fn path_exprs_1() {
|
|
assert!(string_to_expr("a".to_string()) ==
|
|
P(ast::Expr{
|
|
id: ast::DUMMY_NODE_ID,
|
|
node: ast::ExprPath(None, ast::Path {
|
|
span: sp(0, 1),
|
|
global: false,
|
|
segments: vec!(
|
|
ast::PathSegment {
|
|
identifier: str_to_ident("a"),
|
|
parameters: ast::PathParameters::none(),
|
|
}
|
|
),
|
|
}),
|
|
span: sp(0, 1)
|
|
}))
|
|
}
|
|
|
|
#[test] fn path_exprs_2 () {
|
|
assert!(string_to_expr("::a::b".to_string()) ==
|
|
P(ast::Expr {
|
|
id: ast::DUMMY_NODE_ID,
|
|
node: ast::ExprPath(None, ast::Path {
|
|
span: sp(0, 6),
|
|
global: true,
|
|
segments: vec!(
|
|
ast::PathSegment {
|
|
identifier: str_to_ident("a"),
|
|
parameters: ast::PathParameters::none(),
|
|
},
|
|
ast::PathSegment {
|
|
identifier: str_to_ident("b"),
|
|
parameters: ast::PathParameters::none(),
|
|
}
|
|
)
|
|
}),
|
|
span: sp(0, 6)
|
|
}))
|
|
}
|
|
|
|
#[should_panic]
|
|
#[test] fn bad_path_expr_1() {
|
|
string_to_expr("::abc::def::return".to_string());
|
|
}
|
|
|
|
// check the token-tree-ization of macros
|
|
#[test]
|
|
fn string_to_tts_macro () {
|
|
let tts = string_to_tts("macro_rules! zip (($a)=>($a))".to_string());
|
|
let tts: &[ast::TokenTree] = &tts[..];
|
|
match tts {
|
|
[ast::TtToken(_, token::Ident(name_macro_rules, token::Plain)),
|
|
ast::TtToken(_, token::Not),
|
|
ast::TtToken(_, token::Ident(name_zip, token::Plain)),
|
|
ast::TtDelimited(_, ref macro_delimed)]
|
|
if name_macro_rules.as_str() == "macro_rules"
|
|
&& name_zip.as_str() == "zip" => {
|
|
match ¯o_delimed.tts[..] {
|
|
[ast::TtDelimited(_, ref first_delimed),
|
|
ast::TtToken(_, token::FatArrow),
|
|
ast::TtDelimited(_, ref second_delimed)]
|
|
if macro_delimed.delim == token::Paren => {
|
|
match &first_delimed.tts[..] {
|
|
[ast::TtToken(_, token::Dollar),
|
|
ast::TtToken(_, token::Ident(name, token::Plain))]
|
|
if first_delimed.delim == token::Paren
|
|
&& name.as_str() == "a" => {},
|
|
_ => panic!("value 3: {:?}", **first_delimed),
|
|
}
|
|
match &second_delimed.tts[..] {
|
|
[ast::TtToken(_, token::Dollar),
|
|
ast::TtToken(_, token::Ident(name, token::Plain))]
|
|
if second_delimed.delim == token::Paren
|
|
&& name.as_str() == "a" => {},
|
|
_ => panic!("value 4: {:?}", **second_delimed),
|
|
}
|
|
},
|
|
_ => panic!("value 2: {:?}", **macro_delimed),
|
|
}
|
|
},
|
|
_ => panic!("value: {:?}",tts),
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn string_to_tts_1() {
|
|
let tts = string_to_tts("fn a (b : i32) { b; }".to_string());
|
|
|
|
let expected = vec![
|
|
ast::TtToken(sp(0, 2),
|
|
token::Ident(str_to_ident("fn"),
|
|
token::IdentStyle::Plain)),
|
|
ast::TtToken(sp(3, 4),
|
|
token::Ident(str_to_ident("a"),
|
|
token::IdentStyle::Plain)),
|
|
ast::TtDelimited(
|
|
sp(5, 14),
|
|
Rc::new(ast::Delimited {
|
|
delim: token::DelimToken::Paren,
|
|
open_span: sp(5, 6),
|
|
tts: vec![
|
|
ast::TtToken(sp(6, 7),
|
|
token::Ident(str_to_ident("b"),
|
|
token::IdentStyle::Plain)),
|
|
ast::TtToken(sp(8, 9),
|
|
token::Colon),
|
|
ast::TtToken(sp(10, 13),
|
|
token::Ident(str_to_ident("i32"),
|
|
token::IdentStyle::Plain)),
|
|
],
|
|
close_span: sp(13, 14),
|
|
})),
|
|
ast::TtDelimited(
|
|
sp(15, 21),
|
|
Rc::new(ast::Delimited {
|
|
delim: token::DelimToken::Brace,
|
|
open_span: sp(15, 16),
|
|
tts: vec![
|
|
ast::TtToken(sp(17, 18),
|
|
token::Ident(str_to_ident("b"),
|
|
token::IdentStyle::Plain)),
|
|
ast::TtToken(sp(18, 19),
|
|
token::Semi)
|
|
],
|
|
close_span: sp(20, 21),
|
|
}))
|
|
];
|
|
|
|
assert_eq!(tts, expected);
|
|
}
|
|
|
|
#[test] fn ret_expr() {
|
|
assert!(string_to_expr("return d".to_string()) ==
|
|
P(ast::Expr{
|
|
id: ast::DUMMY_NODE_ID,
|
|
node:ast::ExprRet(Some(P(ast::Expr{
|
|
id: ast::DUMMY_NODE_ID,
|
|
node:ast::ExprPath(None, ast::Path{
|
|
span: sp(7, 8),
|
|
global: false,
|
|
segments: vec!(
|
|
ast::PathSegment {
|
|
identifier: str_to_ident("d"),
|
|
parameters: ast::PathParameters::none(),
|
|
}
|
|
),
|
|
}),
|
|
span:sp(7,8)
|
|
}))),
|
|
span:sp(0,8)
|
|
}))
|
|
}
|
|
|
|
#[test] fn parse_stmt_1 () {
|
|
assert!(string_to_stmt("b;".to_string()) ==
|
|
P(Spanned{
|
|
node: ast::StmtExpr(P(ast::Expr {
|
|
id: ast::DUMMY_NODE_ID,
|
|
node: ast::ExprPath(None, ast::Path {
|
|
span:sp(0,1),
|
|
global:false,
|
|
segments: vec!(
|
|
ast::PathSegment {
|
|
identifier: str_to_ident("b"),
|
|
parameters: ast::PathParameters::none(),
|
|
}
|
|
),
|
|
}),
|
|
span: sp(0,1)}),
|
|
ast::DUMMY_NODE_ID),
|
|
span: sp(0,1)}))
|
|
|
|
}
|
|
|
|
fn parser_done(p: Parser){
|
|
assert_eq!(p.token.clone(), token::Eof);
|
|
}
|
|
|
|
#[test] fn parse_ident_pat () {
|
|
let sess = new_parse_sess();
|
|
let mut parser = string_to_parser(&sess, "b".to_string());
|
|
assert!(parser.parse_pat()
|
|
== P(ast::Pat{
|
|
id: ast::DUMMY_NODE_ID,
|
|
node: ast::PatIdent(ast::BindByValue(ast::MutImmutable),
|
|
Spanned{ span:sp(0, 1),
|
|
node: str_to_ident("b")
|
|
},
|
|
None),
|
|
span: sp(0,1)}));
|
|
parser_done(parser);
|
|
}
|
|
|
|
// check the contents of the tt manually:
|
|
#[test] fn parse_fundecl () {
|
|
// this test depends on the intern order of "fn" and "i32"
|
|
assert_eq!(string_to_item("fn a (b : i32) { b; }".to_string()),
|
|
Some(
|
|
P(ast::Item{ident:str_to_ident("a"),
|
|
attrs:Vec::new(),
|
|
id: ast::DUMMY_NODE_ID,
|
|
node: ast::ItemFn(P(ast::FnDecl {
|
|
inputs: vec!(ast::Arg{
|
|
ty: P(ast::Ty{id: ast::DUMMY_NODE_ID,
|
|
node: ast::TyPath(None, ast::Path{
|
|
span:sp(10,13),
|
|
global:false,
|
|
segments: vec!(
|
|
ast::PathSegment {
|
|
identifier:
|
|
str_to_ident("i32"),
|
|
parameters: ast::PathParameters::none(),
|
|
}
|
|
),
|
|
}),
|
|
span:sp(10,13)
|
|
}),
|
|
pat: P(ast::Pat {
|
|
id: ast::DUMMY_NODE_ID,
|
|
node: ast::PatIdent(
|
|
ast::BindByValue(ast::MutImmutable),
|
|
Spanned{
|
|
span: sp(6,7),
|
|
node: str_to_ident("b")},
|
|
None
|
|
),
|
|
span: sp(6,7)
|
|
}),
|
|
id: ast::DUMMY_NODE_ID
|
|
}),
|
|
output: ast::DefaultReturn(sp(15, 15)),
|
|
variadic: false
|
|
}),
|
|
ast::Unsafety::Normal,
|
|
abi::Rust,
|
|
ast::Generics{ // no idea on either of these:
|
|
lifetimes: Vec::new(),
|
|
ty_params: OwnedSlice::empty(),
|
|
where_clause: ast::WhereClause {
|
|
id: ast::DUMMY_NODE_ID,
|
|
predicates: Vec::new(),
|
|
}
|
|
},
|
|
P(ast::Block {
|
|
stmts: vec!(P(Spanned{
|
|
node: ast::StmtSemi(P(ast::Expr{
|
|
id: ast::DUMMY_NODE_ID,
|
|
node: ast::ExprPath(None,
|
|
ast::Path{
|
|
span:sp(17,18),
|
|
global:false,
|
|
segments: vec!(
|
|
ast::PathSegment {
|
|
identifier:
|
|
str_to_ident(
|
|
"b"),
|
|
parameters:
|
|
ast::PathParameters::none(),
|
|
}
|
|
),
|
|
}),
|
|
span: sp(17,18)}),
|
|
ast::DUMMY_NODE_ID),
|
|
span: sp(17,19)})),
|
|
expr: None,
|
|
id: ast::DUMMY_NODE_ID,
|
|
rules: ast::DefaultBlock, // no idea
|
|
span: sp(15,21),
|
|
})),
|
|
vis: ast::Inherited,
|
|
span: sp(0,21)})));
|
|
}
|
|
|
|
#[test] fn parse_use() {
|
|
let use_s = "use foo::bar::baz;";
|
|
let vitem = string_to_item(use_s.to_string()).unwrap();
|
|
let vitem_s = item_to_string(&*vitem);
|
|
assert_eq!(&vitem_s[..], use_s);
|
|
|
|
let use_s = "use foo::bar as baz;";
|
|
let vitem = string_to_item(use_s.to_string()).unwrap();
|
|
let vitem_s = item_to_string(&*vitem);
|
|
assert_eq!(&vitem_s[..], use_s);
|
|
}
|
|
|
|
#[test] fn parse_extern_crate() {
|
|
let ex_s = "extern crate foo;";
|
|
let vitem = string_to_item(ex_s.to_string()).unwrap();
|
|
let vitem_s = item_to_string(&*vitem);
|
|
assert_eq!(&vitem_s[..], ex_s);
|
|
|
|
let ex_s = "extern crate \"foo\" as bar;";
|
|
let vitem = string_to_item(ex_s.to_string()).unwrap();
|
|
let vitem_s = item_to_string(&*vitem);
|
|
assert_eq!(&vitem_s[..], ex_s);
|
|
}
|
|
|
|
fn get_spans_of_pat_idents(src: &str) -> Vec<Span> {
|
|
let item = string_to_item(src.to_string()).unwrap();
|
|
|
|
struct PatIdentVisitor {
|
|
spans: Vec<Span>
|
|
}
|
|
impl<'v> ::visit::Visitor<'v> for PatIdentVisitor {
|
|
fn visit_pat(&mut self, p: &'v ast::Pat) {
|
|
match p.node {
|
|
ast::PatIdent(_ , ref spannedident, _) => {
|
|
self.spans.push(spannedident.span.clone());
|
|
}
|
|
_ => {
|
|
::visit::walk_pat(self, p);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
let mut v = PatIdentVisitor { spans: Vec::new() };
|
|
::visit::walk_item(&mut v, &*item);
|
|
return v.spans;
|
|
}
|
|
|
|
#[test] fn span_of_self_arg_pat_idents_are_correct() {
|
|
|
|
let srcs = ["impl z { fn a (&self, &myarg: i32) {} }",
|
|
"impl z { fn a (&mut self, &myarg: i32) {} }",
|
|
"impl z { fn a (&'a self, &myarg: i32) {} }",
|
|
"impl z { fn a (self, &myarg: i32) {} }",
|
|
"impl z { fn a (self: Foo, &myarg: i32) {} }",
|
|
];
|
|
|
|
for &src in &srcs {
|
|
let spans = get_spans_of_pat_idents(src);
|
|
let Span{ lo, hi, .. } = spans[0];
|
|
assert!("self" == &src[lo.to_usize()..hi.to_usize()],
|
|
"\"{}\" != \"self\". src=\"{}\"",
|
|
&src[lo.to_usize()..hi.to_usize()], src)
|
|
}
|
|
}
|
|
|
|
#[test] fn parse_exprs () {
|
|
// just make sure that they parse....
|
|
string_to_expr("3 + 4".to_string());
|
|
string_to_expr("a::z.froob(b,&(987+3))".to_string());
|
|
}
|
|
|
|
#[test] fn attrs_fix_bug () {
|
|
string_to_item("pub fn mk_file_writer(path: &Path, flags: &[FileFlag])
|
|
-> Result<Box<Writer>, String> {
|
|
#[cfg(windows)]
|
|
fn wb() -> c_int {
|
|
(O_WRONLY | libc::consts::os::extra::O_BINARY) as c_int
|
|
}
|
|
|
|
#[cfg(unix)]
|
|
fn wb() -> c_int { O_WRONLY as c_int }
|
|
|
|
let mut fflags: c_int = wb();
|
|
}".to_string());
|
|
}
|
|
|
|
#[test] fn crlf_doc_comments() {
|
|
let sess = new_parse_sess();
|
|
|
|
let name = "<source>".to_string();
|
|
let source = "/// doc comment\r\nfn foo() {}".to_string();
|
|
let item = parse_item_from_source_str(name.clone(), source, Vec::new(), &sess).unwrap();
|
|
let doc = first_attr_value_str_by_name(&item.attrs, "doc").unwrap();
|
|
assert_eq!(&doc[..], "/// doc comment");
|
|
|
|
let source = "/// doc comment\r\n/// line 2\r\nfn foo() {}".to_string();
|
|
let item = parse_item_from_source_str(name.clone(), source, Vec::new(), &sess).unwrap();
|
|
let docs = item.attrs.iter().filter(|a| &*a.name() == "doc")
|
|
.map(|a| a.value_str().unwrap().to_string()).collect::<Vec<_>>();
|
|
let b: &[_] = &["/// doc comment".to_string(), "/// line 2".to_string()];
|
|
assert_eq!(&docs[..], b);
|
|
|
|
let source = "/** doc comment\r\n * with CRLF */\r\nfn foo() {}".to_string();
|
|
let item = parse_item_from_source_str(name, source, Vec::new(), &sess).unwrap();
|
|
let doc = first_attr_value_str_by_name(&item.attrs, "doc").unwrap();
|
|
assert_eq!(&doc[..], "/** doc comment\n * with CRLF */");
|
|
}
|
|
|
|
#[test]
|
|
fn ttdelim_span() {
|
|
let sess = parse::new_parse_sess();
|
|
let expr = parse::parse_expr_from_source_str("foo".to_string(),
|
|
"foo!( fn main() { body } )".to_string(), vec![], &sess);
|
|
|
|
let tts = match expr.node {
|
|
ast::ExprMac(ref mac) => {
|
|
let ast::MacInvocTT(_, ref tts, _) = mac.node;
|
|
tts.clone()
|
|
}
|
|
_ => panic!("not a macro"),
|
|
};
|
|
|
|
let span = tts.iter().rev().next().unwrap().get_span();
|
|
|
|
match sess.span_diagnostic.cm.span_to_snippet(span) {
|
|
Ok(s) => assert_eq!(&s[..], "{ body }"),
|
|
Err(_) => panic!("could not get snippet"),
|
|
}
|
|
}
|
|
}
|