diff options
Diffstat (limited to 'Rewrite/Rewriter.cpp')
-rw-r--r-- | Rewrite/Rewriter.cpp | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/Rewrite/Rewriter.cpp b/Rewrite/Rewriter.cpp index 87a5deaf1a..514b82235a 100644 --- a/Rewrite/Rewriter.cpp +++ b/Rewrite/Rewriter.cpp @@ -13,6 +13,7 @@ //===----------------------------------------------------------------------===// #include "clang/Rewrite/Rewriter.h" +#include "clang/Lex/Lexer.h" #include "clang/Basic/SourceManager.h" using namespace clang; @@ -156,6 +157,10 @@ int Rewriter::getRangeSize(SourceRange Range) const { if (StartFileID != EndFileID) return -1; + // Adjust the end offset to the end of the last token, instead of being the + // start of the last token. + EndOff += Lexer::MeasureTokenLength(Range.getEnd(), *SourceMgr); + return EndOff-StartOff; } |