[clang-tools-extra] 93bb994 - [clangd] Implement path and URI translation for remote index
Kirill Bobyrev via cfe-commits
cfe-commits at lists.llvm.org
Thu Jul 9 03:54:50 PDT 2020
Author: Kirill Bobyrev
Date: 2020-07-09T12:52:55+02:00
New Revision: 93bb9944cb577f0529636dc5acfba16026740962
URL: https://github.com/llvm/llvm-project/commit/93bb9944cb577f0529636dc5acfba16026740962
DIFF: https://github.com/llvm/llvm-project/commit/93bb9944cb577f0529636dc5acfba16026740962.diff
LOG: [clangd] Implement path and URI translation for remote index
Reviewers: sammccall
Reviewed By: sammccall
Subscribers: ormris, ilya-biryukov, MaskRay, jkorous, arphaman, kadircet, usaxena95, cfe-commits
Tags: #clang
Differential Revision: https://reviews.llvm.org/D82938
Added:
Modified:
clang-tools-extra/clangd/index/dex/dexp/Dexp.cpp
clang-tools-extra/clangd/index/remote/Client.cpp
clang-tools-extra/clangd/index/remote/Client.h
clang-tools-extra/clangd/index/remote/Index.proto
clang-tools-extra/clangd/index/remote/marshalling/Marshalling.cpp
clang-tools-extra/clangd/index/remote/marshalling/Marshalling.h
clang-tools-extra/clangd/index/remote/server/Server.cpp
clang-tools-extra/clangd/index/remote/unimplemented/UnimplementedClient.cpp
clang-tools-extra/clangd/unittests/remote/MarshallingTests.cpp
Removed:
################################################################################
diff --git a/clang-tools-extra/clangd/index/dex/dexp/Dexp.cpp b/clang-tools-extra/clangd/index/dex/dexp/Dexp.cpp
index 3e25da385d7a..6fc844c18931 100644
--- a/clang-tools-extra/clangd/index/dex/dexp/Dexp.cpp
+++ b/clang-tools-extra/clangd/index/dex/dexp/Dexp.cpp
@@ -32,6 +32,9 @@ llvm::cl::opt<std::string> IndexLocation(
llvm::cl::opt<std::string>
ExecCommand("c", llvm::cl::desc("Command to execute and then exit"));
+llvm::cl::opt<std::string> ProjectRoot("project-root",
+ llvm::cl::desc("Path to the project"));
+
static constexpr char Overview[] = R"(
This is an **experimental** interactive tool to process user-provided search
queries over given symbol collection obtained via clangd-indexer. The
@@ -326,7 +329,8 @@ struct {
std::unique_ptr<SymbolIndex> openIndex(llvm::StringRef Index) {
return Index.startswith("remote:")
- ? remote::getClient(Index.drop_front(strlen("remote:")))
+ ? remote::getClient(Index.drop_front(strlen("remote:")),
+ ProjectRoot)
: loadIndex(Index, /*UseDex=*/true);
}
diff --git a/clang-tools-extra/clangd/index/remote/Client.cpp b/clang-tools-extra/clangd/index/remote/Client.cpp
index c43afd2573a9..b46883193076 100644
--- a/clang-tools-extra/clangd/index/remote/Client.cpp
+++ b/clang-tools-extra/clangd/index/remote/Client.cpp
@@ -10,10 +10,12 @@
#include "Client.h"
#include "Index.grpc.pb.h"
+#include "index/Index.h"
#include "index/Serialization.h"
#include "marshalling/Marshalling.h"
#include "support/Logger.h"
#include "support/Trace.h"
+#include "llvm/ADT/StringRef.h"
#include <chrono>
@@ -27,6 +29,16 @@ class IndexClient : public clangd::SymbolIndex {
using StreamingCall = std::unique_ptr<grpc::ClientReader<ReplyT>> (
remote::SymbolIndex::Stub::*)(grpc::ClientContext *, const RequestT &);
+ template <typename ClangdRequestT, typename RequestT>
+ RequestT serializeRequest(ClangdRequestT Request) const {
+ return toProtobuf(Request);
+ }
+
+ template <>
+ FuzzyFindRequest serializeRequest(clangd::FuzzyFindRequest Request) const {
+ return toProtobuf(Request, ProjectRoot);
+ }
+
template <typename RequestT, typename ReplyT, typename ClangdRequestT,
typename CallbackT>
bool streamRPC(ClangdRequestT Request,
@@ -34,7 +46,7 @@ class IndexClient : public clangd::SymbolIndex {
CallbackT Callback) const {
bool FinalResult = false;
trace::Span Tracer(RequestT::descriptor()->name());
- const auto RPCRequest = toProtobuf(Request);
+ const auto RPCRequest = serializeRequest<ClangdRequestT, RequestT>(Request);
grpc::ClientContext Context;
std::chrono::system_clock::time_point Deadline =
std::chrono::system_clock::now() + DeadlineWaitingTime;
@@ -48,10 +60,11 @@ class IndexClient : public clangd::SymbolIndex {
FinalResult = Reply.final_result();
continue;
}
- auto Sym = fromProtobuf(Reply.stream_result(), &Strings);
- if (!Sym)
+ auto Response =
+ fromProtobuf(Reply.stream_result(), &Strings, ProjectRoot);
+ if (!Response)
elog("Received invalid {0}", ReplyT::descriptor()->name());
- Callback(*Sym);
+ Callback(*Response);
}
SPAN_ATTACH(Tracer, "status", Reader->Finish().ok());
return FinalResult;
@@ -59,9 +72,9 @@ class IndexClient : public clangd::SymbolIndex {
public:
IndexClient(
- std::shared_ptr<grpc::Channel> Channel,
+ std::shared_ptr<grpc::Channel> Channel, llvm::StringRef ProjectRoot,
std::chrono::milliseconds DeadlineTime = std::chrono::milliseconds(1000))
- : Stub(remote::SymbolIndex::NewStub(Channel)),
+ : Stub(remote::SymbolIndex::NewStub(Channel)), ProjectRoot(ProjectRoot),
DeadlineWaitingTime(DeadlineTime) {}
void lookup(const clangd::LookupRequest &Request,
@@ -86,22 +99,26 @@ class IndexClient : public clangd::SymbolIndex {
llvm::function_ref<void(const SymbolID &, const clangd::Symbol &)>)
const {}
- // IndexClient does not take any space since the data is stored on the server.
+ // IndexClient does not take any space since the data is stored on the
+ // server.
size_t estimateMemoryUsage() const { return 0; }
private:
std::unique_ptr<remote::SymbolIndex::Stub> Stub;
+ std::string ProjectRoot;
// Each request will be terminated if it takes too long.
std::chrono::milliseconds DeadlineWaitingTime;
};
} // namespace
-std::unique_ptr<clangd::SymbolIndex> getClient(llvm::StringRef Address) {
+std::unique_ptr<clangd::SymbolIndex> getClient(llvm::StringRef Address,
+ llvm::StringRef ProjectRoot) {
const auto Channel =
grpc::CreateChannel(Address.str(), grpc::InsecureChannelCredentials());
Channel->GetState(true);
- return std::unique_ptr<clangd::SymbolIndex>(new IndexClient(Channel));
+ return std::unique_ptr<clangd::SymbolIndex>(
+ new IndexClient(Channel, ProjectRoot));
}
} // namespace remote
diff --git a/clang-tools-extra/clangd/index/remote/Client.h b/clang-tools-extra/clangd/index/remote/Client.h
index 9708fdbe851a..7b4173af6cfa 100644
--- a/clang-tools-extra/clangd/index/remote/Client.h
+++ b/clang-tools-extra/clangd/index/remote/Client.h
@@ -10,6 +10,7 @@
#define LLVM_CLANG_TOOLS_EXTRA_CLANGD_INDEX_REMOTE_INDEX_H
#include "index/Index.h"
+#include "llvm/ADT/StringRef.h"
namespace clang {
namespace clangd {
@@ -17,12 +18,16 @@ namespace remote {
/// Returns an SymbolIndex client that passes requests to remote index located
/// at \p Address. The client allows synchronous RPC calls.
+/// \p IndexRoot is an absolute path on the local machine to the source tree
+/// described by the remote index. Paths returned by the index will be treated
+/// as relative to this directory.
///
/// This method attempts to resolve the address and establish the connection.
///
/// \returns nullptr if the address is not resolved during the function call or
/// if the project was compiled without Remote Index support.
-std::unique_ptr<clangd::SymbolIndex> getClient(llvm::StringRef Address);
+std::unique_ptr<clangd::SymbolIndex> getClient(llvm::StringRef Address,
+ llvm::StringRef IndexRoot);
} // namespace remote
} // namespace clangd
diff --git a/clang-tools-extra/clangd/index/remote/Index.proto b/clang-tools-extra/clangd/index/remote/Index.proto
index f3af29dcd588..99c4e3329d67 100644
--- a/clang-tools-extra/clangd/index/remote/Index.proto
+++ b/clang-tools-extra/clangd/index/remote/Index.proto
@@ -71,7 +71,7 @@ message Symbol {
string name = 3;
SymbolLocation definition = 4;
string scope = 5;
- SymbolLocation canonical_declarattion = 6;
+ SymbolLocation canonical_declaration = 6;
int32 references = 7;
uint32 origin = 8;
string signature = 9;
@@ -99,7 +99,10 @@ message SymbolInfo {
message SymbolLocation {
Position start = 1;
Position end = 2;
- string file_uri = 3;
+ // clangd::SymbolLocation stores FileURI, but the protocol transmits a the
+ // relative path. Because paths are
diff erent on the remote and local machines
+ // they will be translated in the marshalling layer.
+ string file_path = 3;
}
message Position {
diff --git a/clang-tools-extra/clangd/index/remote/marshalling/Marshalling.cpp b/clang-tools-extra/clangd/index/remote/marshalling/Marshalling.cpp
index e95132d000ec..db895bf039ba 100644
--- a/clang-tools-extra/clangd/index/remote/marshalling/Marshalling.cpp
+++ b/clang-tools-extra/clangd/index/remote/marshalling/Marshalling.cpp
@@ -7,6 +7,7 @@
//===----------------------------------------------------------------------===//
#include "Marshalling.h"
+#include "Headers.h"
#include "Index.pb.h"
#include "Protocol.h"
#include "index/Serialization.h"
@@ -16,7 +17,13 @@
#include "index/SymbolOrigin.h"
#include "support/Logger.h"
#include "clang/Index/IndexSymbol.h"
+#include "llvm/ADT/None.h"
+#include "llvm/ADT/Optional.h"
+#include "llvm/ADT/SmallString.h"
#include "llvm/ADT/SmallVector.h"
+#include "llvm/ADT/StringRef.h"
+#include "llvm/Support/Error.h"
+#include "llvm/Support/Path.h"
#include "llvm/Support/StringSaver.h"
namespace clang {
@@ -58,40 +65,66 @@ SymbolInfo toProtobuf(const clang::index::SymbolInfo &Info) {
return Result;
}
-clangd::SymbolLocation fromProtobuf(const SymbolLocation &Message,
- llvm::UniqueStringSaver *Strings) {
+llvm::Optional<clangd::SymbolLocation>
+fromProtobuf(const SymbolLocation &Message, llvm::UniqueStringSaver *Strings,
+ llvm::StringRef IndexRoot) {
clangd::SymbolLocation Location;
+ auto URIString = relativePathToURI(Message.file_path(), IndexRoot);
+ if (!URIString)
+ return llvm::None;
+ Location.FileURI = Strings->save(*URIString).begin();
Location.Start = fromProtobuf(Message.start());
Location.End = fromProtobuf(Message.end());
- Location.FileURI = Strings->save(Message.file_uri()).begin();
return Location;
}
-SymbolLocation toProtobuf(const clangd::SymbolLocation &Location) {
+llvm::Optional<SymbolLocation>
+toProtobuf(const clangd::SymbolLocation &Location, llvm::StringRef IndexRoot) {
remote::SymbolLocation Result;
+ auto RelativePath = uriToRelativePath(Location.FileURI, IndexRoot);
+ if (!RelativePath)
+ return llvm::None;
+ *Result.mutable_file_path() = *RelativePath;
*Result.mutable_start() = toProtobuf(Location.Start);
*Result.mutable_end() = toProtobuf(Location.End);
- *Result.mutable_file_uri() = Location.FileURI;
return Result;
}
-HeaderWithReferences
-toProtobuf(const clangd::Symbol::IncludeHeaderWithReferences &IncludeHeader) {
+llvm::Optional<HeaderWithReferences>
+toProtobuf(const clangd::Symbol::IncludeHeaderWithReferences &IncludeHeader,
+ llvm::StringRef IndexRoot) {
HeaderWithReferences Result;
- Result.set_header(IncludeHeader.IncludeHeader.str());
Result.set_references(IncludeHeader.References);
+ const std::string Header = IncludeHeader.IncludeHeader.str();
+ if (isLiteralInclude(Header)) {
+ Result.set_header(Header);
+ return Result;
+ }
+ auto RelativePath = uriToRelativePath(Header, IndexRoot);
+ if (!RelativePath)
+ return llvm::None;
+ Result.set_header(*RelativePath);
return Result;
}
-clangd::Symbol::IncludeHeaderWithReferences
-fromProtobuf(const HeaderWithReferences &Message) {
- return clangd::Symbol::IncludeHeaderWithReferences{Message.header(),
+llvm::Optional<clangd::Symbol::IncludeHeaderWithReferences>
+fromProtobuf(const HeaderWithReferences &Message,
+ llvm::UniqueStringSaver *Strings, llvm::StringRef IndexRoot) {
+ std::string Header = Message.header();
+ if (Header.front() != '<' && Header.front() != '"') {
+ auto URIString = relativePathToURI(Header, IndexRoot);
+ if (!URIString)
+ return llvm::None;
+ Header = *URIString;
+ }
+ return clangd::Symbol::IncludeHeaderWithReferences{Strings->save(Header),
Message.references()};
}
} // namespace
-clangd::FuzzyFindRequest fromProtobuf(const FuzzyFindRequest *Request) {
+clangd::FuzzyFindRequest fromProtobuf(const FuzzyFindRequest *Request,
+ llvm::StringRef IndexRoot) {
clangd::FuzzyFindRequest Result;
Result.Query = Request->query();
for (const auto &Scope : Request->scopes())
@@ -100,24 +133,29 @@ clangd::FuzzyFindRequest fromProtobuf(const FuzzyFindRequest *Request) {
if (Request->limit())
Result.Limit = Request->limit();
Result.RestrictForCodeCompletion = Request->restricted_for_code_completion();
- for (const auto &Path : Request->proximity_paths())
- Result.ProximityPaths.push_back(Path);
+ for (const auto &Path : Request->proximity_paths()) {
+ llvm::SmallString<256> LocalPath = llvm::StringRef(IndexRoot);
+ llvm::sys::path::append(LocalPath, Path);
+ Result.ProximityPaths.push_back(std::string(LocalPath));
+ }
for (const auto &Type : Request->preferred_types())
Result.ProximityPaths.push_back(Type);
return Result;
}
llvm::Optional<clangd::Symbol> fromProtobuf(const Symbol &Message,
- llvm::UniqueStringSaver *Strings) {
+ llvm::UniqueStringSaver *Strings,
+ llvm::StringRef IndexRoot) {
if (!Message.has_info() || !Message.has_definition() ||
- !Message.has_canonical_declarattion()) {
- elog("Cannot convert Symbol from Protobuf: {}", Message.ShortDebugString());
+ !Message.has_canonical_declaration()) {
+ elog("Cannot convert Symbol from Protobuf: {0}",
+ Message.ShortDebugString());
return llvm::None;
}
clangd::Symbol Result;
auto ID = SymbolID::fromStr(Message.id());
if (!ID) {
- elog("Cannot convert parse SymbolID {} from Protobuf: {}", ID.takeError(),
+ elog("Cannot parse SymbolID {0} given Protobuf: {1}", ID.takeError(),
Message.ShortDebugString());
return llvm::None;
}
@@ -125,9 +163,15 @@ llvm::Optional<clangd::Symbol> fromProtobuf(const Symbol &Message,
Result.SymInfo = fromProtobuf(Message.info());
Result.Name = Message.name();
Result.Scope = Message.scope();
- Result.Definition = fromProtobuf(Message.definition(), Strings);
- Result.CanonicalDeclaration =
- fromProtobuf(Message.canonical_declarattion(), Strings);
+ auto Definition = fromProtobuf(Message.definition(), Strings, IndexRoot);
+ if (!Definition)
+ return llvm::None;
+ Result.Definition = *Definition;
+ auto Declaration =
+ fromProtobuf(Message.canonical_declaration(), Strings, IndexRoot);
+ if (!Declaration)
+ return llvm::None;
+ Result.CanonicalDeclaration = *Declaration;
Result.References = Message.references();
Result.Origin = static_cast<clangd::SymbolOrigin>(Message.origin());
Result.Signature = Message.signature();
@@ -137,20 +181,26 @@ llvm::Optional<clangd::Symbol> fromProtobuf(const Symbol &Message,
Result.ReturnType = Message.return_type();
Result.Type = Message.type();
for (const auto &Header : Message.headers()) {
- Result.IncludeHeaders.push_back(fromProtobuf(Header));
+ auto SerializedHeader = fromProtobuf(Header, Strings, IndexRoot);
+ if (SerializedHeader)
+ Result.IncludeHeaders.push_back(*SerializedHeader);
}
Result.Flags = static_cast<clangd::Symbol::SymbolFlag>(Message.flags());
return Result;
}
llvm::Optional<clangd::Ref> fromProtobuf(const Ref &Message,
- llvm::UniqueStringSaver *Strings) {
+ llvm::UniqueStringSaver *Strings,
+ llvm::StringRef IndexRoot) {
if (!Message.has_location()) {
elog("Cannot convert Ref from Protobuf: {}", Message.ShortDebugString());
return llvm::None;
}
clangd::Ref Result;
- Result.Location = fromProtobuf(Message.location(), Strings);
+ auto Location = fromProtobuf(Message.location(), Strings, IndexRoot);
+ if (!Location)
+ return llvm::None;
+ Result.Location = *Location;
Result.Kind = static_cast<clangd::RefKind>(Message.kind());
return Result;
}
@@ -162,7 +212,8 @@ LookupRequest toProtobuf(const clangd::LookupRequest &From) {
return RPCRequest;
}
-FuzzyFindRequest toProtobuf(const clangd::FuzzyFindRequest &From) {
+FuzzyFindRequest toProtobuf(const clangd::FuzzyFindRequest &From,
+ llvm::StringRef IndexRoot) {
FuzzyFindRequest RPCRequest;
RPCRequest.set_query(From.Query);
for (const auto &Scope : From.Scopes)
@@ -171,8 +222,12 @@ FuzzyFindRequest toProtobuf(const clangd::FuzzyFindRequest &From) {
if (From.Limit)
RPCRequest.set_limit(*From.Limit);
RPCRequest.set_restricted_for_code_completion(From.RestrictForCodeCompletion);
- for (const auto &Path : From.ProximityPaths)
- RPCRequest.add_proximity_paths(Path);
+ for (const auto &Path : From.ProximityPaths) {
+ llvm::SmallString<256> RelativePath = llvm::StringRef(Path);
+ if (llvm::sys::path::replace_path_prefix(RelativePath, IndexRoot, ""))
+ RPCRequest.add_proximity_paths(llvm::sys::path::convert_to_slash(
+ RelativePath, llvm::sys::path::Style::posix));
+ }
for (const auto &Type : From.PreferredTypes)
RPCRequest.add_preferred_types(Type);
return RPCRequest;
@@ -188,15 +243,18 @@ RefsRequest toProtobuf(const clangd::RefsRequest &From) {
return RPCRequest;
}
-Symbol toProtobuf(const clangd::Symbol &From) {
+Symbol toProtobuf(const clangd::Symbol &From, llvm::StringRef IndexRoot) {
Symbol Result;
Result.set_id(From.ID.str());
*Result.mutable_info() = toProtobuf(From.SymInfo);
Result.set_name(From.Name.str());
- *Result.mutable_definition() = toProtobuf(From.Definition);
+ auto Definition = toProtobuf(From.Definition, IndexRoot);
+ if (Definition)
+ *Result.mutable_definition() = *Definition;
Result.set_scope(From.Scope.str());
- *Result.mutable_canonical_declarattion() =
- toProtobuf(From.CanonicalDeclaration);
+ auto Declaration = toProtobuf(From.CanonicalDeclaration, IndexRoot);
+ if (Declaration)
+ *Result.mutable_canonical_declaration() = *Declaration;
Result.set_references(From.References);
Result.set_origin(static_cast<uint32_t>(From.Origin));
Result.set_signature(From.Signature.str());
@@ -207,20 +265,82 @@ Symbol toProtobuf(const clangd::Symbol &From) {
Result.set_return_type(From.ReturnType.str());
Result.set_type(From.Type.str());
for (const auto &Header : From.IncludeHeaders) {
+ auto Serialized = toProtobuf(Header, IndexRoot);
+ if (!Serialized)
+ continue;
auto *NextHeader = Result.add_headers();
- *NextHeader = toProtobuf(Header);
+ *NextHeader = *Serialized;
}
Result.set_flags(static_cast<uint32_t>(From.Flags));
return Result;
}
-Ref toProtobuf(const clangd::Ref &From) {
+// FIXME(kirillbobyrev): A reference without location is invalid.
+// llvm::Optional<Ref> here and on the server side?
+Ref toProtobuf(const clangd::Ref &From, llvm::StringRef IndexRoot) {
Ref Result;
Result.set_kind(static_cast<uint32_t>(From.Kind));
- *Result.mutable_location() = toProtobuf(From.Location);
+ auto Location = toProtobuf(From.Location, IndexRoot);
+ if (Location)
+ *Result.mutable_location() = *Location;
return Result;
}
+llvm::Optional<std::string> relativePathToURI(llvm::StringRef RelativePath,
+ llvm::StringRef IndexRoot) {
+ assert(RelativePath == llvm::sys::path::convert_to_slash(
+ RelativePath, llvm::sys::path::Style::posix));
+ assert(IndexRoot == llvm::sys::path::convert_to_slash(IndexRoot));
+ assert(IndexRoot.endswith(llvm::sys::path::get_separator()));
+ if (RelativePath.empty())
+ return std::string();
+ if (llvm::sys::path::is_absolute(RelativePath)) {
+ elog("Remote index client got absolute path from server: {0}",
+ RelativePath);
+ return llvm::None;
+ }
+ if (llvm::sys::path::is_relative(IndexRoot)) {
+ elog("Remote index client got a relative path as index root: {0}",
+ IndexRoot);
+ return llvm::None;
+ }
+ llvm::SmallString<256> FullPath = IndexRoot;
+ llvm::sys::path::append(FullPath, RelativePath);
+ auto Result = URI::createFile(FullPath);
+ return Result.toString();
+}
+
+llvm::Optional<std::string> uriToRelativePath(llvm::StringRef URI,
+ llvm::StringRef IndexRoot) {
+ assert(IndexRoot.endswith(llvm::sys::path::get_separator()));
+ assert(IndexRoot == llvm::sys::path::convert_to_slash(IndexRoot));
+ assert(!IndexRoot.empty());
+ if (llvm::sys::path::is_relative(IndexRoot)) {
+ elog("Index root {0} is not absolute path", IndexRoot);
+ return llvm::None;
+ }
+ auto ParsedURI = URI::parse(URI);
+ if (!ParsedURI) {
+ elog("Remote index got bad URI from client {0}: {1}", URI,
+ ParsedURI.takeError());
+ return llvm::None;
+ }
+ if (ParsedURI->scheme() != "file") {
+ elog("Remote index got URI with scheme other than \"file\" {0}: {1}", URI,
+ ParsedURI->scheme());
+ return llvm::None;
+ }
+ llvm::SmallString<256> Result = ParsedURI->body();
+ if (!llvm::sys::path::replace_path_prefix(Result, IndexRoot, "")) {
+ elog("Can not get relative path from the URI {0} given the index root {1}",
+ URI, IndexRoot);
+ return llvm::None;
+ }
+ // Make sure the result has UNIX slashes.
+ return llvm::sys::path::convert_to_slash(Result,
+ llvm::sys::path::Style::posix);
+}
+
} // namespace remote
} // namespace clangd
} // namespace clang
diff --git a/clang-tools-extra/clangd/index/remote/marshalling/Marshalling.h b/clang-tools-extra/clangd/index/remote/marshalling/Marshalling.h
index 1bc25bf5a9de..86cc8fa272fc 100644
--- a/clang-tools-extra/clangd/index/remote/marshalling/Marshalling.h
+++ b/clang-tools-extra/clangd/index/remote/marshalling/Marshalling.h
@@ -6,7 +6,28 @@
//
//===----------------------------------------------------------------------===//
//
-// Transformations between native Clangd types and Protobuf-generated classes.
+// Marshalling provides translation between native clangd types into the
+// Protobuf-generated classes. Most translations are 1-to-1 and wrap variables
+// into appropriate Protobuf types.
+//
+// A notable exception is URI translation. Because paths to files are
diff erent
+// on indexing machine and client machine
+// ("/remote/machine/projects/llvm-project/llvm/include/HelloWorld.h" versus
+// "/usr/local/username/llvm-project/llvm/include/HelloWorld.h"), they need to
+// be converted appropriately. Remote machine strips the prefix from the
+// absolute path and passes paths relative to the project root over the wire
+// ("include/HelloWorld.h" in this example). The indexed project root is passed
+// to the remote server. Client receives this relative path and constructs a URI
+// that points to the relevant file in the filesystem. The relative path is
+// appended to IndexRoot to construct the full path and build the final URI.
+//
+// Index root is the absolute path to the project and includes a trailing slash.
+// The relative path passed over the wire has unix slashes.
+//
+// toProtobuf() functions serialize native clangd types and strip IndexRoot from
+// the file paths specific to indexing machine. fromProtobuf() functions
+// deserialize clangd types and translate relative paths into machine-native
+// URIs.
//
//===----------------------------------------------------------------------===//
@@ -15,24 +36,40 @@
#include "Index.pb.h"
#include "index/Index.h"
+#include "llvm/ADT/StringRef.h"
#include "llvm/Support/StringSaver.h"
namespace clang {
namespace clangd {
namespace remote {
-clangd::FuzzyFindRequest fromProtobuf(const FuzzyFindRequest *Request);
+clangd::FuzzyFindRequest fromProtobuf(const FuzzyFindRequest *Request,
+ llvm::StringRef IndexRoot);
llvm::Optional<clangd::Symbol> fromProtobuf(const Symbol &Message,
- llvm::UniqueStringSaver *Strings);
+ llvm::UniqueStringSaver *Strings,
+ llvm::StringRef IndexRoot);
llvm::Optional<clangd::Ref> fromProtobuf(const Ref &Message,
- llvm::UniqueStringSaver *Strings);
+ llvm::UniqueStringSaver *Strings,
+ llvm::StringRef IndexRoot);
LookupRequest toProtobuf(const clangd::LookupRequest &From);
-FuzzyFindRequest toProtobuf(const clangd::FuzzyFindRequest &From);
+FuzzyFindRequest toProtobuf(const clangd::FuzzyFindRequest &From,
+ llvm::StringRef IndexRoot);
RefsRequest toProtobuf(const clangd::RefsRequest &From);
-Ref toProtobuf(const clangd::Ref &From);
-Symbol toProtobuf(const clangd::Symbol &From);
+Ref toProtobuf(const clangd::Ref &From, llvm::StringRef IndexRoot);
+Symbol toProtobuf(const clangd::Symbol &From, llvm::StringRef IndexRoot);
+
+/// Translates \p RelativePath into the absolute path and builds URI for the
+/// user machine. This translation happens on the client side with the
+/// \p RelativePath received from remote index server and \p IndexRoot is
+/// provided by the client.
+llvm::Optional<std::string> relativePathToURI(llvm::StringRef RelativePath,
+ llvm::StringRef IndexRoot);
+/// Translates a URI from the server's backing index to a relative path suitable
+/// to send over the wire to the client.
+llvm::Optional<std::string> uriToRelativePath(llvm::StringRef URI,
+ llvm::StringRef IndexRoot);
} // namespace remote
} // namespace clangd
diff --git a/clang-tools-extra/clangd/index/remote/server/Server.cpp b/clang-tools-extra/clangd/index/remote/server/Server.cpp
index 4d84eb17210e..718d623a4845 100644
--- a/clang-tools-extra/clangd/index/remote/server/Server.cpp
+++ b/clang-tools-extra/clangd/index/remote/server/Server.cpp
@@ -11,6 +11,7 @@
#include "index/remote/marshalling/Marshalling.h"
#include "llvm/ADT/StringRef.h"
#include "llvm/Support/CommandLine.h"
+#include "llvm/Support/Path.h"
#include "llvm/Support/Signals.h"
#include <grpc++/grpc++.h>
@@ -31,6 +32,9 @@ awaits gRPC lookup requests from the client.
llvm::cl::opt<std::string> IndexPath(llvm::cl::desc("<INDEX FILE>"),
llvm::cl::Positional, llvm::cl::Required);
+llvm::cl::opt<std::string> IndexRoot(llvm::cl::desc("<PROJECT ROOT>"),
+ llvm::cl::Positional, llvm::cl::Required);
+
llvm::cl::opt<std::string> ServerAddress(
"server-address", llvm::cl::init("0.0.0.0:50051"),
llvm::cl::desc("Address of the invoked server. Defaults to 0.0.0.0:50051"));
@@ -41,8 +45,13 @@ std::unique_ptr<clangd::SymbolIndex> openIndex(llvm::StringRef Index) {
class RemoteIndexServer final : public SymbolIndex::Service {
public:
- RemoteIndexServer(std::unique_ptr<clangd::SymbolIndex> Index)
- : Index(std::move(Index)) {}
+ RemoteIndexServer(std::unique_ptr<clangd::SymbolIndex> Index,
+ llvm::StringRef IndexRoot)
+ : Index(std::move(Index)) {
+ llvm::SmallString<256> NativePath = IndexRoot;
+ llvm::sys::path::native(NativePath);
+ IndexedProjectRoot = std::string(NativePath);
+ }
private:
grpc::Status Lookup(grpc::ServerContext *Context,
@@ -57,7 +66,8 @@ class RemoteIndexServer final : public SymbolIndex::Service {
}
Index->lookup(Req, [&](const clangd::Symbol &Sym) {
LookupReply NextMessage;
- *NextMessage.mutable_stream_result() = toProtobuf(Sym);
+ *NextMessage.mutable_stream_result() =
+ toProtobuf(Sym, IndexedProjectRoot);
Reply->Write(NextMessage);
});
LookupReply LastMessage;
@@ -69,10 +79,11 @@ class RemoteIndexServer final : public SymbolIndex::Service {
grpc::Status FuzzyFind(grpc::ServerContext *Context,
const FuzzyFindRequest *Request,
grpc::ServerWriter<FuzzyFindReply> *Reply) override {
- const auto Req = fromProtobuf(Request);
+ const auto Req = fromProtobuf(Request, IndexedProjectRoot);
bool HasMore = Index->fuzzyFind(Req, [&](const clangd::Symbol &Sym) {
FuzzyFindReply NextMessage;
- *NextMessage.mutable_stream_result() = toProtobuf(Sym);
+ *NextMessage.mutable_stream_result() =
+ toProtobuf(Sym, IndexedProjectRoot);
Reply->Write(NextMessage);
});
FuzzyFindReply LastMessage;
@@ -92,7 +103,7 @@ class RemoteIndexServer final : public SymbolIndex::Service {
}
bool HasMore = Index->refs(Req, [&](const clangd::Ref &Reference) {
RefsReply NextMessage;
- *NextMessage.mutable_stream_result() = toProtobuf(Reference);
+ *NextMessage.mutable_stream_result() = toProtobuf(Reference, IndexRoot);
Reply->Write(NextMessage);
});
RefsReply LastMessage;
@@ -102,11 +113,12 @@ class RemoteIndexServer final : public SymbolIndex::Service {
}
std::unique_ptr<clangd::SymbolIndex> Index;
+ std::string IndexedProjectRoot;
};
void runServer(std::unique_ptr<clangd::SymbolIndex> Index,
const std::string &ServerAddress) {
- RemoteIndexServer Service(std::move(Index));
+ RemoteIndexServer Service(std::move(Index), IndexRoot);
grpc::EnableDefaultHealthCheckService(true);
grpc::ServerBuilder Builder;
@@ -128,6 +140,11 @@ int main(int argc, char *argv[]) {
llvm::cl::ParseCommandLineOptions(argc, argv, Overview);
llvm::sys::PrintStackTraceOnErrorSignal(argv[0]);
+ if (!llvm::sys::path::is_absolute(IndexRoot)) {
+ llvm::outs() << "Index root should be an absolute path.\n";
+ return -1;
+ }
+
std::unique_ptr<clang::clangd::SymbolIndex> Index = openIndex(IndexPath);
if (!Index) {
diff --git a/clang-tools-extra/clangd/index/remote/unimplemented/UnimplementedClient.cpp b/clang-tools-extra/clangd/index/remote/unimplemented/UnimplementedClient.cpp
index f0fb612f6332..a4d3127e0455 100644
--- a/clang-tools-extra/clangd/index/remote/unimplemented/UnimplementedClient.cpp
+++ b/clang-tools-extra/clangd/index/remote/unimplemented/UnimplementedClient.cpp
@@ -8,12 +8,14 @@
#include "index/remote/Client.h"
#include "support/Logger.h"
+#include "llvm/ADT/StringRef.h"
namespace clang {
namespace clangd {
namespace remote {
-std::unique_ptr<clangd::SymbolIndex> getClient(llvm::StringRef Address) {
+std::unique_ptr<clangd::SymbolIndex> getClient(llvm::StringRef Address,
+ llvm::StringRef IndexRoot) {
elog("Can't create SymbolIndex client without Remote Index support.");
return nullptr;
}
diff --git a/clang-tools-extra/clangd/unittests/remote/MarshallingTests.cpp b/clang-tools-extra/clangd/unittests/remote/MarshallingTests.cpp
index 8e68cfbdbeee..a14ff13150d3 100644
--- a/clang-tools-extra/clangd/unittests/remote/MarshallingTests.cpp
+++ b/clang-tools-extra/clangd/unittests/remote/MarshallingTests.cpp
@@ -7,85 +7,286 @@
//===----------------------------------------------------------------------===//
#include "../TestTU.h"
+#include "TestFS.h"
+#include "index/Index.h"
+#include "index/Ref.h"
#include "index/Serialization.h"
+#include "index/Symbol.h"
+#include "index/SymbolID.h"
#include "index/remote/marshalling/Marshalling.h"
+#include "clang/Index/IndexSymbol.h"
+#include "llvm/ADT/SmallString.h"
+#include "llvm/ADT/StringRef.h"
+#include "llvm/Support/Path.h"
#include "llvm/Support/StringSaver.h"
+#include "gmock/gmock.h"
#include "gtest/gtest.h"
+#include <cstring>
namespace clang {
namespace clangd {
namespace remote {
namespace {
+using llvm::sys::path::convert_to_slash;
+
+const char *testPathURI(llvm::StringRef Path,
+ llvm::UniqueStringSaver &Strings) {
+ auto URI = URI::createFile(testPath(Path));
+ return Strings.save(URI.toString()).begin();
+}
+
+TEST(RemoteMarshallingTest, URITranslation) {
+ llvm::BumpPtrAllocator Arena;
+ llvm::UniqueStringSaver Strings(Arena);
+ clangd::Ref Original;
+ Original.Location.FileURI =
+ testPathURI("remote/machine/projects/llvm-project/clang-tools-extra/"
+ "clangd/unittests/remote/MarshallingTests.cpp",
+ Strings);
+ auto Serialized =
+ toProtobuf(Original, testPath("remote/machine/projects/llvm-project/"));
+ EXPECT_EQ(Serialized.location().file_path(),
+ "clang-tools-extra/clangd/unittests/remote/MarshallingTests.cpp");
+ const std::string LocalIndexPrefix = testPath("local/machine/project/");
+ auto Deserialized = fromProtobuf(Serialized, &Strings,
+ testPath("home/my-projects/llvm-project/"));
+ EXPECT_TRUE(Deserialized);
+ EXPECT_EQ(Deserialized->Location.FileURI,
+ testPathURI("home/my-projects/llvm-project/clang-tools-extra/"
+ "clangd/unittests/remote/MarshallingTests.cpp",
+ Strings));
+
+ clangd::Ref WithInvalidURI;
+ // Invalid URI results in empty path.
+ WithInvalidURI.Location.FileURI = "This is not a URI";
+ Serialized = toProtobuf(WithInvalidURI, testPath("home/"));
+ EXPECT_EQ(Serialized.location().file_path(), "");
+
+ // Can not use URIs with scheme
diff erent from "file".
+ auto UnittestURI =
+ URI::create(testPath("project/lib/HelloWorld.cpp"), "unittest");
+ EXPECT_TRUE(bool(UnittestURI));
+ WithInvalidURI.Location.FileURI =
+ Strings.save(UnittestURI->toString()).begin();
+ Serialized = toProtobuf(WithInvalidURI, testPath("project/lib/"));
+ EXPECT_EQ(Serialized.location().file_path(), "");
+
+ Ref WithAbsolutePath;
+ *WithAbsolutePath.mutable_location()->mutable_file_path() =
+ "/usr/local/user/home/HelloWorld.cpp";
+ Deserialized = fromProtobuf(WithAbsolutePath, &Strings, LocalIndexPrefix);
+ // Paths transmitted over the wire can not be absolute, they have to be
+ // relative.
+ EXPECT_FALSE(Deserialized);
+}
+
TEST(RemoteMarshallingTest, SymbolSerialization) {
- const auto *Header = R"(
- // This is a class.
- class Foo {
- public:
- Foo();
-
- int Bar;
- private:
- double Number;
- };
- /// This is a function.
- char baz();
- template <typename T>
- T getT ();
- )";
- const auto TU = TestTU::withHeaderCode(Header);
- const auto Symbols = TU.headerSymbols();
- // Sanity check: there are more than 5 symbols available.
- EXPECT_GE(Symbols.size(), 5UL);
+ clangd::Symbol Sym;
+
+ auto ID = SymbolID::fromStr("057557CEBF6E6B2D");
+ EXPECT_TRUE(bool(ID));
+ Sym.ID = *ID;
+
+ index::SymbolInfo Info;
+ Info.Kind = index::SymbolKind::Function;
+ Info.SubKind = index::SymbolSubKind::AccessorGetter;
+ Info.Lang = index::SymbolLanguage::CXX;
+ Info.Properties = static_cast<index::SymbolPropertySet>(
+ index::SymbolProperty::TemplateSpecialization);
+ Sym.SymInfo = Info;
+
llvm::BumpPtrAllocator Arena;
llvm::UniqueStringSaver Strings(Arena);
- for (auto &Sym : Symbols) {
- const auto ProtobufMeessage = toProtobuf(Sym);
- const auto SymToProtobufAndBack = fromProtobuf(ProtobufMeessage, &Strings);
- EXPECT_TRUE(SymToProtobufAndBack.hasValue());
- EXPECT_EQ(toYAML(Sym), toYAML(*SymToProtobufAndBack));
- }
+
+ Sym.Name = Strings.save("Foo");
+ Sym.Scope = Strings.save("llvm::foo::bar::");
+
+ clangd::SymbolLocation Location;
+ Location.Start.setLine(1);
+ Location.Start.setColumn(15);
+ Location.End.setLine(3);
+ Location.End.setColumn(121);
+ Location.FileURI = testPathURI("home/Definition.cpp", Strings);
+ Sym.Definition = Location;
+
+ Location.Start.setLine(42);
+ Location.Start.setColumn(31);
+ Location.End.setLine(20);
+ Location.End.setColumn(400);
+ Location.FileURI = testPathURI("home/Declaration.h", Strings);
+ Sym.CanonicalDeclaration = Location;
+
+ Sym.References = 9000;
+ Sym.Origin = clangd::SymbolOrigin::Static;
+ Sym.Signature = Strings.save("(int X, char Y, Type T)");
+ Sym.TemplateSpecializationArgs = Strings.save("<int, char, bool, Type>");
+ Sym.CompletionSnippetSuffix =
+ Strings.save("({1: int X}, {2: char Y}, {3: Type T})");
+ Sym.Documentation = Strings.save("This is my amazing Foo constructor!");
+ Sym.ReturnType = Strings.save("Foo");
+
+ Sym.Flags = clangd::Symbol::SymbolFlag::IndexedForCodeCompletion;
+
+ // Check that symbols are exactly the same if the path to indexed project is
+ // the same on indexing machine and the client.
+ auto Serialized = toProtobuf(Sym, testPath("home/"));
+ auto Deserialized = fromProtobuf(Serialized, &Strings, testPath("home/"));
+ EXPECT_TRUE(Deserialized);
+ EXPECT_EQ(toYAML(Sym), toYAML(*Deserialized));
+ // Serialized paths are relative and have UNIX slashes.
+ EXPECT_EQ(convert_to_slash(Serialized.definition().file_path(),
+ llvm::sys::path::Style::posix),
+ Serialized.definition().file_path());
+ EXPECT_TRUE(
+ llvm::sys::path::is_relative(Serialized.definition().file_path()));
+
+ // Fail with an invalid URI.
+ Location.FileURI = "Not A URI";
+ Sym.Definition = Location;
+ Serialized = toProtobuf(Sym, testPath("home/"));
+ Deserialized = fromProtobuf(Serialized, &Strings, testPath("home/"));
+ EXPECT_FALSE(Deserialized);
+
+ // Schemes other than "file" can not be used.
+ auto UnittestURI = URI::create(testPath("home/SomePath.h"), "unittest");
+ EXPECT_TRUE(bool(UnittestURI));
+ Location.FileURI = Strings.save(UnittestURI->toString()).begin();
+ Sym.Definition = Location;
+ Serialized = toProtobuf(Sym, testPath("home/"));
+ Deserialized = fromProtobuf(Serialized, &Strings, testPath("home/"));
+ EXPECT_FALSE(Deserialized);
+
+ // Passing root that is not prefix of the original file path.
+ Location.FileURI = testPathURI("home/File.h", Strings);
+ Sym.Definition = Location;
+ // Check that the symbol is valid and passing the correct path works.
+ Serialized = toProtobuf(Sym, testPath("home/"));
+ Deserialized = fromProtobuf(Serialized, &Strings, testPath("home/"));
+ EXPECT_TRUE(Deserialized);
+ EXPECT_EQ(Deserialized->Definition.FileURI,
+ testPathURI("home/File.h", Strings));
+ // Fail with a wrong root.
+ Serialized = toProtobuf(Sym, testPath("nothome/"));
+ Deserialized = fromProtobuf(Serialized, &Strings, testPath("home/"));
+ EXPECT_FALSE(Deserialized);
}
-TEST(RemoteMarshallingTest, ReferenceSerialization) {
- TestTU TU;
- TU.HeaderCode = R"(
- int foo();
- int GlobalVariable = 42;
- class Foo {
- public:
- Foo();
-
- char Symbol = 'S';
- };
- template <typename T>
- T getT() { return T(); }
- )";
- TU.Code = R"(
- int foo() {
- ++GlobalVariable;
-
- Foo foo = Foo();
- if (foo.Symbol - 'a' == 42) {
- foo.Symbol = 'b';
- }
-
- const auto bar = getT<Foo>();
- }
- )";
- const auto References = TU.headerRefs();
+TEST(RemoteMarshallingTest, RefSerialization) {
+ clangd::Ref Ref;
+ Ref.Kind = clangd::RefKind::Spelled | clangd::RefKind::Declaration;
+
llvm::BumpPtrAllocator Arena;
llvm::UniqueStringSaver Strings(Arena);
- // Sanity check: there are more than 5 references available.
- EXPECT_GE(References.numRefs(), 5UL);
- for (const auto &SymbolWithRefs : References) {
- for (const auto &Ref : SymbolWithRefs.second) {
- const auto RefToProtobufAndBack = fromProtobuf(toProtobuf(Ref), &Strings);
- EXPECT_TRUE(RefToProtobufAndBack.hasValue());
- EXPECT_EQ(toYAML(Ref), toYAML(*RefToProtobufAndBack));
- }
- }
-} // namespace
+
+ clangd::SymbolLocation Location;
+ Location.Start.setLine(124);
+ Location.Start.setColumn(21);
+ Location.End.setLine(3213);
+ Location.End.setColumn(541);
+ Location.FileURI = testPathURI(
+ "llvm-project/llvm/clang-tools-extra/clangd/Protocol.h", Strings);
+ Ref.Location = Location;
+
+ auto Serialized = toProtobuf(Ref, testPath("llvm-project/"));
+ auto Deserialized =
+ fromProtobuf(Serialized, &Strings, testPath("llvm-project/"));
+ EXPECT_TRUE(Deserialized);
+ EXPECT_EQ(toYAML(Ref), toYAML(*Deserialized));
+}
+
+TEST(RemoteMarshallingTest, IncludeHeaderURIs) {
+ llvm::BumpPtrAllocator Arena;
+ llvm::UniqueStringSaver Strings(Arena);
+
+ llvm::SmallVector<clangd::Symbol::IncludeHeaderWithReferences, 1>
+ ValidHeaders;
+ clangd::Symbol::IncludeHeaderWithReferences Header;
+ Header.IncludeHeader = Strings.save(
+ URI::createFile("/usr/local/user/home/project/Header.h").toString());
+ Header.References = 21;
+ ValidHeaders.push_back(Header);
+ Header.IncludeHeader = Strings.save("<iostream>");
+ Header.References = 100;
+ ValidHeaders.push_back(Header);
+ Header.IncludeHeader = Strings.save("\"cstdio\"");
+ Header.References = 200;
+ ValidHeaders.push_back(Header);
+
+ llvm::SmallVector<clangd::Symbol::IncludeHeaderWithReferences, 1>
+ InvalidHeaders;
+ // This is an absolute path to a header: can not be transmitted over the wire.
+ Header.IncludeHeader = Strings.save(testPath("project/include/Common.h"));
+ Header.References = 42;
+ InvalidHeaders.push_back(Header);
+ // This is not a valid header: can not be transmitted over the wire;
+ Header.IncludeHeader = Strings.save("NotAHeader");
+ Header.References = 5;
+ InvalidHeaders.push_back(Header);
+
+ clangd::Symbol Sym;
+ // Fill in definition and declaration, Symbool will be invalid otherwise.
+ clangd::SymbolLocation Location;
+ Location.Start.setLine(1);
+ Location.Start.setColumn(2);
+ Location.End.setLine(3);
+ Location.End.setColumn(4);
+ Location.FileURI = testPathURI("File.h", Strings);
+ Sym.Definition = Location;
+ Sym.CanonicalDeclaration = Location;
+
+ // Try to serialize all headers but only valid ones will end up in Protobuf
+ // message.
+ auto AllHeaders = ValidHeaders;
+ AllHeaders.insert(AllHeaders.end(), InvalidHeaders.begin(),
+ InvalidHeaders.end());
+ Sym.IncludeHeaders = AllHeaders;
+
+ auto Serialized = toProtobuf(Sym, convert_to_slash("/"));
+ EXPECT_EQ(static_cast<size_t>(Serialized.headers_size()),
+ ValidHeaders.size());
+ auto Deserialized = fromProtobuf(Serialized, &Strings, convert_to_slash("/"));
+ EXPECT_TRUE(Deserialized);
+
+ Sym.IncludeHeaders = ValidHeaders;
+ EXPECT_EQ(toYAML(Sym), toYAML(*Deserialized));
+}
+
+TEST(RemoteMarshallingTest, FuzzyFindRequestSerialization) {
+ clangd::FuzzyFindRequest Request;
+ Request.ProximityPaths = {testPath("remote/Header.h"),
+ testPath("remote/subdir/OtherHeader.h"),
+ testPath("notremote/File.h"), "Not a Path."};
+ auto Serialized = toProtobuf(Request, testPath("remote/"));
+ EXPECT_EQ(Serialized.proximity_paths_size(), 2);
+ auto Deserialized = fromProtobuf(&Serialized, testPath("home/"));
+ EXPECT_THAT(Deserialized.ProximityPaths,
+ testing::ElementsAre(testPath("home/Header.h"),
+ testPath("home/subdir/OtherHeader.h")));
+}
+
+TEST(RemoteMarshallingTest, RelativePathToURITranslation) {
+ EXPECT_TRUE(relativePathToURI("lib/File.cpp", testPath("home/project/")));
+ // RelativePath can not be absolute.
+ EXPECT_FALSE(relativePathToURI("/lib/File.cpp", testPath("home/project/")));
+ // IndexRoot has to be absolute path.
+ EXPECT_FALSE(relativePathToURI("lib/File.cpp", "home/project/"));
+}
+
+TEST(RemoteMarshallingTest, URIToRelativePathTranslation) {
+ llvm::BumpPtrAllocator Arena;
+ llvm::UniqueStringSaver Strings(Arena);
+ EXPECT_TRUE(
+ uriToRelativePath(testPathURI("home/project/lib/File.cpp", Strings),
+ testPath("home/project/")));
+ // IndexRoot has to be absolute path.
+ EXPECT_FALSE(uriToRelativePath(
+ testPathURI("home/project/lib/File.cpp", Strings), "home/project/"));
+ // IndexRoot has to be be a prefix of the file path.
+ EXPECT_FALSE(
+ uriToRelativePath(testPathURI("home/project/lib/File.cpp", Strings),
+ testPath("home/other/project/")));
+}
} // namespace
} // namespace remote
More information about the cfe-commits
mailing list