1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
|
//===--- TransProtectedScope.cpp - Transformations to ARC mode ------------===//
//
// The LLVM Compiler Infrastructure
//
// This file is distributed under the University of Illinois Open Source
// License. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//
//
// Adds brackets in case statements that "contain" initialization of retaining
// variable, thus emitting the "switch case is in protected scope" error.
//
//===----------------------------------------------------------------------===//
#include "Transforms.h"
#include "Internals.h"
#include "clang/Sema/SemaDiagnostic.h"
using namespace clang;
using namespace arcmt;
using namespace trans;
namespace {
struct CaseInfo {
SwitchCase *SC;
SourceRange Range;
bool FixedBypass;
CaseInfo() : SC(0), FixedBypass(false) {}
CaseInfo(SwitchCase *S, SourceRange Range)
: SC(S), Range(Range), FixedBypass(false) {}
};
class CaseCollector : public RecursiveASTVisitor<CaseCollector> {
llvm::SmallVectorImpl<CaseInfo> &Cases;
public:
CaseCollector(llvm::SmallVectorImpl<CaseInfo> &Cases)
: Cases(Cases) { }
bool VisitSwitchStmt(SwitchStmt *S) {
SourceLocation NextLoc = S->getLocEnd();
SwitchCase *Curr = S->getSwitchCaseList();
// We iterate over case statements in reverse source-order.
while (Curr) {
Cases.push_back(CaseInfo(Curr,SourceRange(Curr->getLocStart(), NextLoc)));
NextLoc = Curr->getLocStart();
Curr = Curr->getNextSwitchCase();
}
return true;
}
};
} // anonymous namespace
static bool isInRange(FullSourceLoc Loc, SourceRange R) {
return !Loc.isBeforeInTranslationUnitThan(R.getBegin()) &&
Loc.isBeforeInTranslationUnitThan(R.getEnd());
}
static bool handleProtectedNote(const StoredDiagnostic &Diag,
llvm::SmallVectorImpl<CaseInfo> &Cases,
TransformActions &TA) {
assert(Diag.getLevel() == DiagnosticsEngine::Note);
for (unsigned i = 0; i != Cases.size(); i++) {
CaseInfo &info = Cases[i];
if (isInRange(Diag.getLocation(), info.Range)) {
TA.clearDiagnostic(Diag.getID(), Diag.getLocation());
if (!info.FixedBypass) {
TA.insertAfterToken(info.SC->getColonLoc(), " {");
TA.insert(info.Range.getEnd(), "}\n");
info.FixedBypass = true;
}
return true;
}
}
return false;
}
static void handleProtectedScopeError(CapturedDiagList::iterator &DiagI,
CapturedDiagList::iterator DiagE,
llvm::SmallVectorImpl<CaseInfo> &Cases,
TransformActions &TA) {
Transaction Trans(TA);
assert(DiagI->getID() == diag::err_switch_into_protected_scope);
SourceLocation ErrLoc = DiagI->getLocation();
bool handledAllNotes = true;
++DiagI;
for (; DiagI != DiagE && DiagI->getLevel() == DiagnosticsEngine::Note;
++DiagI) {
if (!handleProtectedNote(*DiagI, Cases, TA))
handledAllNotes = false;
}
if (handledAllNotes)
TA.clearDiagnostic(diag::err_switch_into_protected_scope, ErrLoc);
}
void ProtectedScopeTraverser::traverseBody(BodyContext &BodyCtx) {
MigrationPass &Pass = BodyCtx.getMigrationContext().Pass;
SmallVector<CaseInfo, 16> Cases;
CaseCollector(Cases).TraverseStmt(BodyCtx.getTopStmt());
SourceRange BodyRange = BodyCtx.getTopStmt()->getSourceRange();
const CapturedDiagList &DiagList = Pass.getDiags();
CapturedDiagList::iterator I = DiagList.begin(), E = DiagList.end();
while (I != E) {
if (I->getID() == diag::err_switch_into_protected_scope &&
isInRange(I->getLocation(), BodyRange)) {
handleProtectedScopeError(I, E, Cases, Pass.TA);
continue;
}
++I;
}
}
|
