averello/swift-mirror
1
0
Fork 0
mirror of https://github.com/apple/swift.git synced 2025-12-14 20:36:38 +01:00
Code Issues Packages Projects Releases Wiki Activity

[CusorInfo] Fix ASAN heap-use-after-free failure when reporting parent contexts.

Browse Source 
Resolves rdar://problem/74289832
This commit is contained in:
Nathan Hawes
2021-02-16 11:24:21 +10:00
parent 77d828d3c6
commit bba702940c
2 changed files with 18 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
tools/SourceKit/lib/SwiftLang/SwiftSourceDocInfo.cpp
View File

@@ -836,9 +836,7 @@ static bool passCursorInfoForDecl(SourceFile* SF,
SmallVector<symbolgraphgen::PathComponent, 4> PathComponents;
SmallVector<ParentInfo, 4> Parents;
unsigned SymbolGraphBegin = SS.size();
unsigned SymbolGraphEnd = SymbolGraphBegin;
if (SymbolGraph) {
symbolgraphgen::SymbolGraphOptions Options {
"",
@@ -851,16 +849,16 @@ static bool passCursorInfoForDecl(SourceFile* SF,
symbolgraphgen::printSymbolGraphForDecl(VD, BaseType,
InSynthesizedExtension, Options, OS,
PathComponents);
SymbolGraphEnd = SS.size();
for (auto &Component: PathComponents) {
unsigned USRStart = SS.size();
if (SwiftLangSupport::printUSR(Component.VD, OS))
continue;
StringRef USR{SS.begin()+USRStart, SS.size() - USRStart};
Parents.push_back({Component.Title, Component.Kind, USR});
}
}
unsigned SymbolGraphEnd = SS.size();
DelayedStringRetriever ParentUSRsOS(SS);
for (auto &Component: PathComponents) {
ParentUSRsOS.startPiece();
if (SwiftLangSupport::printUSR(Component.VD, OS))
ParentUSRsOS.startPiece(); // ignore any output if invalid
ParentUSRsOS.endPiece();
};
unsigned FullDeclBegin = SS.size();
{
@@ -995,6 +993,15 @@ static bool passCursorInfoForDecl(SourceFile* SF,
StringRef SymbolGraphJSON = StringRef(SS.begin()+SymbolGraphBegin,
SymbolGraphEnd-SymbolGraphBegin);
SmallVector<ParentInfo, 4> Parents;
auto ParentIt = PathComponents.begin();
ParentUSRsOS.retrieve([&](StringRef ParentUSR) {
assert(ParentIt != PathComponents.end());
if (!ParentUSR.empty())
Parents.push_back({ParentIt->Title, ParentIt->Kind, ParentUSR});
++ParentIt;
});
// If VD is the syntehsized property wrapper backing storage (_foo) or
// projected value ($foo) of a property (foo), base the location on that
// property instead.