From b361b7f1a5bd1d7a0a66b908a65c1fdfe65ae3bf Mon Sep 17 00:00:00 2001 From: Daniel Campbell Date: Wed, 3 Feb 2016 20:44:53 +1300 Subject: Changed macro spans in CSVs to point to the macro name, bugfixed nested spans --- src/libsyntax/codemap.rs | 21 ++++++++++++++++++++- 1 file changed, 20 insertions(+), 1 deletion(-) (limited to 'src/libsyntax') diff --git a/src/libsyntax/codemap.rs b/src/libsyntax/codemap.rs index 9da5e1e3881..74302e2f6a0 100644 --- a/src/libsyntax/codemap.rs +++ b/src/libsyntax/codemap.rs @@ -1052,9 +1052,18 @@ impl CodeMap { /// the macro callsite that expanded to it. pub fn source_callsite(&self, sp: Span) -> Span { let mut span = sp; + // Special case - if a macro is parsed as an argument to another macro, the source + // callsite is the first callsite, which is also source-equivalent to the span. + let mut first = true; while span.expn_id != NO_EXPANSION && span.expn_id != COMMAND_LINE_EXPN { if let Some(callsite) = self.with_expn_info(span.expn_id, |ei| ei.map(|ei| ei.call_site.clone())) { + if first && span.source_equal(&callsite) { + if self.lookup_char_pos(span.lo).file.is_real_file() { + return Span { expn_id: NO_EXPANSION, .. span }; + } + } + first = false; span = callsite; } else { @@ -1071,10 +1080,20 @@ impl CodeMap { /// corresponding to the source callsite. pub fn source_callee(&self, sp: Span) -> Option { let mut span = sp; + // Special case - if a macro is parsed as an argument to another macro, the source + // callsite is source-equivalent to the span, and the source callee is the first callee. + let mut first = true; while let Some(callsite) = self.with_expn_info(span.expn_id, |ei| ei.map(|ei| ei.call_site.clone())) { + if first && span.source_equal(&callsite) { + if self.lookup_char_pos(span.lo).file.is_real_file() { + return self.with_expn_info(span.expn_id, + |ei| ei.map(|ei| ei.callee.clone())); + } + } + first = false; if let Some(_) = self.with_expn_info(callsite.expn_id, - |ei| ei.map(|ei| ei.call_site.clone())) { + |ei| ei.map(|ei| ei.call_site.clone())) { span = callsite; } else { -- cgit 1.4.1-3-g733a5