1 //===--- FS.h - File system related utils ------------------------*- C++-*-===// 2 // 3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 4 // See https://llvm.org/LICENSE.txt for license information. 5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 6 // 7 //===----------------------------------------------------------------------===// 8 9 #ifndef LLVM_CLANG_TOOLS_EXTRA_CLANGD_FS_H 10 #define LLVM_CLANG_TOOLS_EXTRA_CLANGD_FS_H 11 12 #include "support/Path.h" 13 #include "clang/Basic/LLVM.h" 14 #include "llvm/ADT/Optional.h" 15 #include "llvm/ADT/StringMap.h" 16 #include "llvm/Support/VirtualFileSystem.h" 17 18 namespace clang { 19 namespace clangd { 20 21 /// Records status information for files open()ed or stat()ed during preamble 22 /// build (except for the main file), so we can avoid stat()s on the underlying 23 /// FS when reusing the preamble. For example, code completion can re-stat files 24 /// when getting FileID for source locations stored in preamble (e.g. checking 25 /// whether a location is in the main file). 26 /// 27 /// The cache is keyed by absolute path of file name in cached status, as this 28 /// is what preamble stores. 29 /// 30 /// The cache is not thread-safe when updates happen, so the use pattern should 31 /// be: 32 /// - One FS writes to the cache from one thread (or several but strictly 33 /// sequenced), e.g. when building preamble. 34 /// - Sequence point (no writes after this point, no reads before). 35 /// - Several FSs can read from the cache, e.g. code completions. 36 /// 37 /// Note that the cache is only valid when reusing preamble. 38 class PreambleFileStatusCache { 39 public: 40 /// \p MainFilePath is the absolute path of the main source file this preamble 41 /// corresponds to. The stat for the main file will not be cached. 42 PreambleFileStatusCache(llvm::StringRef MainFilePath); 43 44 void update(const llvm::vfs::FileSystem &FS, llvm::vfs::Status S); 45 46 /// \p Path is a path stored in preamble. 47 llvm::Optional<llvm::vfs::Status> lookup(llvm::StringRef Path) const; 48 49 /// Returns a VFS that collects file status. 50 /// Only cache stats for files that exist because 51 /// 1) we only care about existing files when reusing preamble, unlike 52 /// building preamble. 53 /// 2) we use the file name in the Status as the cache key. 54 /// 55 /// Note that the returned VFS should not outlive the cache. 56 IntrusiveRefCntPtr<llvm::vfs::FileSystem> 57 getProducingFS(IntrusiveRefCntPtr<llvm::vfs::FileSystem> FS); 58 59 /// Returns a VFS that uses the cache collected. 60 /// 61 /// Note that the returned VFS should not outlive the cache. 62 IntrusiveRefCntPtr<llvm::vfs::FileSystem> 63 getConsumingFS(IntrusiveRefCntPtr<llvm::vfs::FileSystem> FS) const; 64 65 private: 66 std::string MainFilePath; 67 llvm::StringMap<llvm::vfs::Status> StatCache; 68 }; 69 70 /// Returns a version of \p File that doesn't contain dots and dot dots. 71 /// e.g /a/b/../c -> /a/c 72 /// /a/b/./c -> /a/b/c 73 /// FIXME: We should avoid encountering such paths in clangd internals by 74 /// filtering everything we get over LSP, CDB, etc. 75 Path removeDots(PathRef File); 76 77 } // namespace clangd 78 } // namespace clang 79 80 #endif // LLVM_CLANG_TOOLS_EXTRA_CLANGD_FS_H 81