8d48cd6009
This was causing a test failure in one of LLDB's tests which specifically dealt with a limitation in LLVM's implementation of home_directory() that LLDB's own implementation had worked around. This limitation has been addressed in r298513 on the LLVM side, so the failing test (which is now unnecessary as the limitation no longer exists) was removed in r298519, allowing this patch to be re-submitted without modification. llvm-svn: 298526
90 lines
2.4 KiB
C++
90 lines
2.4 KiB
C++
//===--------------------- TildeExpressionResolver.cpp ----------*- C++ -*-===//
|
|
//
|
|
// The LLVM Compiler Infrastructure
|
|
//
|
|
// This file is distributed under the University of Illinois Open Source
|
|
// License. See LICENSE.TXT for details.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "lldb/Utility/TildeExpressionResolver.h"
|
|
|
|
#include "llvm/ADT/SmallString.h"
|
|
#include "llvm/Support/FileSystem.h"
|
|
#include "llvm/Support/Path.h"
|
|
|
|
#if !defined(LLVM_ON_WIN32)
|
|
#include <pwd.h>
|
|
#endif
|
|
|
|
using namespace lldb_private;
|
|
using namespace llvm;
|
|
|
|
namespace fs = llvm::sys::fs;
|
|
namespace path = llvm::sys::path;
|
|
|
|
TildeExpressionResolver::~TildeExpressionResolver() {}
|
|
|
|
bool StandardTildeExpressionResolver::ResolveExact(
|
|
StringRef Expr, SmallVectorImpl<char> &Output) {
|
|
// We expect the tilde expression to be ONLY the expression itself, and
|
|
// contain no separators.
|
|
assert(!llvm::any_of(Expr, [](char c) { return path::is_separator(c); }));
|
|
assert(Expr.empty() || Expr[0] == '~');
|
|
|
|
return !fs::real_path(Expr, Output, true);
|
|
}
|
|
|
|
bool StandardTildeExpressionResolver::ResolvePartial(StringRef Expr,
|
|
StringSet<> &Output) {
|
|
// We expect the tilde expression to be ONLY the expression itself, and
|
|
// contain no separators.
|
|
assert(!llvm::any_of(Expr, [](char c) { return path::is_separator(c); }));
|
|
assert(Expr.empty() || Expr[0] == '~');
|
|
|
|
Output.clear();
|
|
#if defined(LLVM_ON_WIN32) || defined(__ANDROID__)
|
|
return false;
|
|
#else
|
|
if (Expr.empty())
|
|
return false;
|
|
|
|
SmallString<32> Buffer("~");
|
|
setpwent();
|
|
struct passwd *user_entry;
|
|
Expr = Expr.drop_front();
|
|
|
|
while ((user_entry = getpwent()) != NULL) {
|
|
StringRef ThisName(user_entry->pw_name);
|
|
if (!ThisName.startswith(Expr))
|
|
continue;
|
|
|
|
Buffer.resize(1);
|
|
Buffer.append(ThisName);
|
|
Buffer.append(path::get_separator());
|
|
Output.insert(Buffer);
|
|
}
|
|
|
|
return true;
|
|
#endif
|
|
}
|
|
|
|
bool TildeExpressionResolver::ResolveFullPath(
|
|
StringRef Expr, llvm::SmallVectorImpl<char> &Output) {
|
|
Output.clear();
|
|
if (!Expr.startswith("~")) {
|
|
Output.append(Expr.begin(), Expr.end());
|
|
return false;
|
|
}
|
|
|
|
namespace path = llvm::sys::path;
|
|
StringRef Left =
|
|
Expr.take_until([](char c) { return path::is_separator(c); });
|
|
|
|
if (!ResolveExact(Left, Output))
|
|
return false;
|
|
|
|
Output.append(Expr.begin() + Left.size(), Expr.end());
|
|
return true;
|
|
}
|