blob: fa28ba1b6ab6a094b8d4f011e2233270d6d00d27 [file] [log] [blame]
Rafael Espindolaf1fc3822013-06-26 19:33:03 +00001//===- llvm/Support/Unix/Path.inc - Unix Path Implementation ----*- C++ -*-===//
Michael J. Spencerebad2f92010-11-29 22:28:51 +00002//
3// The LLVM Compiler Infrastructure
4//
5// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9//
Rafael Espindolaf1fc3822013-06-26 19:33:03 +000010// This file implements the Unix specific implementation of the Path API.
Michael J. Spencerebad2f92010-11-29 22:28:51 +000011//
12//===----------------------------------------------------------------------===//
13
14//===----------------------------------------------------------------------===//
15//=== WARNING: Implementation here must contain only generic UNIX code that
16//=== is guaranteed to work on *all* UNIX variants.
17//===----------------------------------------------------------------------===//
18
19#include "Unix.h"
Duncan P. N. Exon Smith91d3cfe2016-04-05 20:45:04 +000020#include <limits.h>
21#include <stdio.h>
Michael J. Spencer9fc1d9d2010-12-01 19:32:01 +000022#if HAVE_SYS_STAT_H
23#include <sys/stat.h>
24#endif
25#if HAVE_FCNTL_H
26#include <fcntl.h>
27#endif
Taewook Ohd9153272016-06-13 15:54:56 +000028#ifdef HAVE_UNISTD_H
29#include <unistd.h>
30#endif
Nick Kledzik18497e92012-06-20 00:28:54 +000031#ifdef HAVE_SYS_MMAN_H
32#include <sys/mman.h>
33#endif
Michael J. Spencer52714862011-01-05 16:38:57 +000034#if HAVE_DIRENT_H
35# include <dirent.h>
36# define NAMLEN(dirent) strlen((dirent)->d_name)
37#else
38# define dirent direct
39# define NAMLEN(dirent) (dirent)->d_namlen
40# if HAVE_SYS_NDIR_H
41# include <sys/ndir.h>
42# endif
43# if HAVE_SYS_DIR_H
44# include <sys/dir.h>
45# endif
46# if HAVE_NDIR_H
47# include <ndir.h>
48# endif
49#endif
Michael J. Spencer9fc1d9d2010-12-01 19:32:01 +000050
Zachary Turnere48ace62017-03-10 17:39:21 +000051#include <pwd.h>
52
Rafael Espindola4601c462013-06-26 05:25:44 +000053#ifdef __APPLE__
54#include <mach-o/dyld.h>
Taewook Ohd9153272016-06-13 15:54:56 +000055#include <sys/attr.h>
Rafael Espindola4601c462013-06-26 05:25:44 +000056#endif
57
Simon Pilgrimf2fbf432016-11-20 13:47:59 +000058// Both stdio.h and cstdio are included via different paths and
Joerg Sonnenbergerc0697302012-08-10 10:56:09 +000059// stdcxx's cstdio doesn't include stdio.h, so it doesn't #undef the macros
60// either.
61#undef ferror
62#undef feof
63
Sylvestre Ledru14ada942012-04-11 15:35:36 +000064// For GNU Hurd
65#if defined(__GNU__) && !defined(PATH_MAX)
66# define PATH_MAX 4096
67#endif
68
Mehdi Aminie2d8f1b2016-04-01 00:18:08 +000069#include <sys/types.h>
Zachary Turner392ed9d2017-02-21 20:55:47 +000070#if !defined(__APPLE__) && !defined(__OpenBSD__) && !defined(__FreeBSD__) && \
71 !defined(__linux__)
Mehdi Aminie2d8f1b2016-04-01 00:18:08 +000072#include <sys/statvfs.h>
73#define STATVFS statvfs
Zachary Turner392ed9d2017-02-21 20:55:47 +000074#define FSTATVFS fstatvfs
Mehdi Aminie2d8f1b2016-04-01 00:18:08 +000075#define STATVFS_F_FRSIZE(vfs) vfs.f_frsize
76#else
Zachary Turner392ed9d2017-02-21 20:55:47 +000077#if defined(__OpenBSD__) || defined(__FreeBSD__)
Mehdi Aminie2d8f1b2016-04-01 00:18:08 +000078#include <sys/param.h>
Mehdi Amini3ba84ca2016-04-08 16:45:05 +000079#include <sys/mount.h>
Zachary Turner392ed9d2017-02-21 20:55:47 +000080#elif defined(__linux__)
Michal Gorny5ddd2a52017-02-22 18:09:15 +000081#if defined(HAVE_LINUX_MAGIC_H)
Zachary Turner392ed9d2017-02-21 20:55:47 +000082#include <linux/magic.h>
Michal Gorny5ddd2a52017-02-22 18:09:15 +000083#else
84#if defined(HAVE_LINUX_NFS_FS_H)
85#include <linux/nfs_fs.h>
86#endif
87#if defined(HAVE_LINUX_SMB_H)
88#include <linux/smb.h>
89#endif
90#endif
Mehdi Aminie2d8f1b2016-04-01 00:18:08 +000091#include <sys/vfs.h>
92#else
93#include <sys/mount.h>
94#endif
95#define STATVFS statfs
Zachary Turner392ed9d2017-02-21 20:55:47 +000096#define FSTATVFS fstatfs
Mehdi Aminie2d8f1b2016-04-01 00:18:08 +000097#define STATVFS_F_FRSIZE(vfs) static_cast<uint64_t>(vfs.f_bsize)
98#endif
99
Zachary Turner392ed9d2017-02-21 20:55:47 +0000100#if defined(__NetBSD__)
101#define STATVFS_F_FLAG(vfs) (vfs).f_flag
102#else
103#define STATVFS_F_FLAG(vfs) (vfs).f_flags
104#endif
Mehdi Aminie2d8f1b2016-04-01 00:18:08 +0000105
Michael J. Spencer45710402010-12-03 01:21:28 +0000106using namespace llvm;
107
Michael J. Spencerebad2f92010-11-29 22:28:51 +0000108namespace llvm {
109namespace sys {
Michael J. Spencer20daa282010-12-07 01:22:31 +0000110namespace fs {
Dimitry Andricebc87792017-05-17 19:33:10 +0000111#if defined(__FreeBSD__) || defined (__NetBSD__) || defined(__Bitrig__) || \
112 defined(__OpenBSD__) || defined(__minix) || defined(__FreeBSD_kernel__) || \
Chandler Carruth2aff7502016-07-19 22:46:39 +0000113 defined(__linux__) || defined(__CYGWIN__) || defined(__DragonFly__) || \
114 defined(_AIX)
Duncan P. N. Exon Smith91d3cfe2016-04-05 20:45:04 +0000115static int
Sylvestre Ledru21e674722013-12-09 16:27:00 +0000116test_dir(char ret[PATH_MAX], const char *dir, const char *bin)
Mehdi Aminie2d8f1b2016-04-01 00:18:08 +0000117{
Rafael Espindolae03dfd92013-06-26 05:01:35 +0000118 struct stat sb;
Sylvestre Ledru21e674722013-12-09 16:27:00 +0000119 char fullpath[PATH_MAX];
Rafael Espindolae03dfd92013-06-26 05:01:35 +0000120
Sylvestre Ledru21e674722013-12-09 16:27:00 +0000121 snprintf(fullpath, PATH_MAX, "%s/%s", dir, bin);
Hans Wennborg083ca9b2015-10-06 23:24:35 +0000122 if (!realpath(fullpath, ret))
123 return 1;
Sylvestre Ledru21e674722013-12-09 16:27:00 +0000124 if (stat(fullpath, &sb) != 0)
Hans Wennborg083ca9b2015-10-06 23:24:35 +0000125 return 1;
Rafael Espindolae03dfd92013-06-26 05:01:35 +0000126
Hans Wennborg083ca9b2015-10-06 23:24:35 +0000127 return 0;
Rafael Espindolae03dfd92013-06-26 05:01:35 +0000128}
129
Duncan P. N. Exon Smith91d3cfe2016-04-05 20:45:04 +0000130static char *
Rafael Espindolae03dfd92013-06-26 05:01:35 +0000131getprogpath(char ret[PATH_MAX], const char *bin)
132{
Sylvestre Ledru21e674722013-12-09 16:27:00 +0000133 char *pv, *s, *t;
Rafael Espindolae03dfd92013-06-26 05:01:35 +0000134
135 /* First approach: absolute path. */
136 if (bin[0] == '/') {
Sylvestre Ledru21e674722013-12-09 16:27:00 +0000137 if (test_dir(ret, "/", bin) == 0)
Hans Wennborg083ca9b2015-10-06 23:24:35 +0000138 return ret;
139 return nullptr;
Rafael Espindolae03dfd92013-06-26 05:01:35 +0000140 }
141
142 /* Second approach: relative path. */
Hans Wennborg083ca9b2015-10-06 23:24:35 +0000143 if (strchr(bin, '/')) {
Sylvestre Ledru21e674722013-12-09 16:27:00 +0000144 char cwd[PATH_MAX];
Hans Wennborg083ca9b2015-10-06 23:24:35 +0000145 if (!getcwd(cwd, PATH_MAX))
146 return nullptr;
Sylvestre Ledru21e674722013-12-09 16:27:00 +0000147 if (test_dir(ret, cwd, bin) == 0)
Hans Wennborg083ca9b2015-10-06 23:24:35 +0000148 return ret;
149 return nullptr;
Rafael Espindolae03dfd92013-06-26 05:01:35 +0000150 }
151
152 /* Third approach: $PATH */
Hans Wennborg083ca9b2015-10-06 23:24:35 +0000153 if ((pv = getenv("PATH")) == nullptr)
154 return nullptr;
Rafael Espindolae03dfd92013-06-26 05:01:35 +0000155 s = pv = strdup(pv);
Hans Wennborg083ca9b2015-10-06 23:24:35 +0000156 if (!pv)
157 return nullptr;
158 while ((t = strsep(&s, ":")) != nullptr) {
Sylvestre Ledru21e674722013-12-09 16:27:00 +0000159 if (test_dir(ret, t, bin) == 0) {
Rafael Espindolae03dfd92013-06-26 05:01:35 +0000160 free(pv);
Hans Wennborg083ca9b2015-10-06 23:24:35 +0000161 return ret;
Rafael Espindolae03dfd92013-06-26 05:01:35 +0000162 }
163 }
164 free(pv);
Hans Wennborg083ca9b2015-10-06 23:24:35 +0000165 return nullptr;
Rafael Espindolae03dfd92013-06-26 05:01:35 +0000166}
Dimitry Andricebc87792017-05-17 19:33:10 +0000167#endif // __FreeBSD__ || __NetBSD__ || __FreeBSD_kernel__
Rafael Espindolae03dfd92013-06-26 05:01:35 +0000168
169/// GetMainExecutable - Return the path to the main executable, given the
170/// value of argv[0] from program startup.
171std::string getMainExecutable(const char *argv0, void *MainAddr) {
172#if defined(__APPLE__)
173 // On OS X the executable path is saved to the stack by dyld. Reading it
174 // from there is much faster than calling dladdr, especially for large
175 // binaries with symbols.
176 char exe_path[MAXPATHLEN];
177 uint32_t size = sizeof(exe_path);
178 if (_NSGetExecutablePath(exe_path, &size) == 0) {
179 char link_path[MAXPATHLEN];
180 if (realpath(exe_path, link_path))
Rafael Espindola4601c462013-06-26 05:25:44 +0000181 return link_path;
Rafael Espindolae03dfd92013-06-26 05:01:35 +0000182 }
Dimitry Andricebc87792017-05-17 19:33:10 +0000183#elif defined(__FreeBSD__) || defined (__NetBSD__) || defined(__Bitrig__) || \
184 defined(__OpenBSD__) || defined(__minix) || defined(__DragonFly__) || \
185 defined(__FreeBSD_kernel__) || defined(_AIX)
Rafael Espindolae03dfd92013-06-26 05:01:35 +0000186 char exe_path[PATH_MAX];
187
188 if (getprogpath(exe_path, argv0) != NULL)
Rafael Espindola2c6f4fe2013-06-26 06:10:32 +0000189 return exe_path;
Rafael Espindolae03dfd92013-06-26 05:01:35 +0000190#elif defined(__linux__) || defined(__CYGWIN__)
191 char exe_path[MAXPATHLEN];
192 StringRef aPath("/proc/self/exe");
193 if (sys::fs::exists(aPath)) {
194 // /proc is not always mounted under Linux (chroot for example).
195 ssize_t len = readlink(aPath.str().c_str(), exe_path, sizeof(exe_path));
196 if (len >= 0)
Alexey Samsonov4c798452014-12-29 20:59:02 +0000197 return std::string(exe_path, len);
Rafael Espindolae03dfd92013-06-26 05:01:35 +0000198 } else {
199 // Fall back to the classical detection.
Hans Wennborg083ca9b2015-10-06 23:24:35 +0000200 if (getprogpath(exe_path, argv0))
201 return exe_path;
Rafael Espindolae03dfd92013-06-26 05:01:35 +0000202 }
Omair Javaidf5d560bc842017-02-02 01:17:49 +0000203#elif defined(HAVE_DLFCN_H) && defined(HAVE_DLADDR)
Rafael Espindolae03dfd92013-06-26 05:01:35 +0000204 // Use dladdr to get executable path if available.
205 Dl_info DLInfo;
206 int err = dladdr(MainAddr, &DLInfo);
207 if (err == 0)
Rafael Espindola2c6f4fe2013-06-26 06:10:32 +0000208 return "";
Rafael Espindolae03dfd92013-06-26 05:01:35 +0000209
210 // If the filename is a symlink, we need to resolve and return the location of
211 // the actual executable.
212 char link_path[MAXPATHLEN];
213 if (realpath(DLInfo.dli_fname, link_path))
Rafael Espindola2c6f4fe2013-06-26 06:10:32 +0000214 return link_path;
Rafael Espindolae03dfd92013-06-26 05:01:35 +0000215#else
216#error GetMainExecutable is not implemented on this host yet.
217#endif
218 return "";
219}
Michael J. Spencerebad2f92010-11-29 22:28:51 +0000220
Pavel Labath757ca882016-10-24 10:59:17 +0000221TimePoint<> file_status::getLastAccessedTime() const {
222 return toTimePoint(fs_st_atime);
Mehdi Amini1e39ef32016-03-25 07:30:21 +0000223}
224
Pavel Labath757ca882016-10-24 10:59:17 +0000225TimePoint<> file_status::getLastModificationTime() const {
226 return toTimePoint(fs_st_mtime);
Rafael Espindoladb5d8fe2013-06-20 18:42:04 +0000227}
228
Rafael Espindolad1230992013-07-29 21:55:38 +0000229UniqueID file_status::getUniqueID() const {
Rafael Espindola7f822a92013-07-29 21:26:49 +0000230 return UniqueID(fs_st_dev, fs_st_ino);
231}
232
Zachary Turner5821a3b2017-03-20 23:55:20 +0000233uint32_t file_status::getLinkCount() const {
234 return fs_st_nlinks;
235}
236
Mehdi Aminie2d8f1b2016-04-01 00:18:08 +0000237ErrorOr<space_info> disk_space(const Twine &Path) {
238 struct STATVFS Vfs;
239 if (::STATVFS(Path.str().c_str(), &Vfs))
240 return std::error_code(errno, std::generic_category());
241 auto FrSize = STATVFS_F_FRSIZE(Vfs);
242 space_info SpaceInfo;
243 SpaceInfo.capacity = static_cast<uint64_t>(Vfs.f_blocks) * FrSize;
244 SpaceInfo.free = static_cast<uint64_t>(Vfs.f_bfree) * FrSize;
245 SpaceInfo.available = static_cast<uint64_t>(Vfs.f_bavail) * FrSize;
246 return SpaceInfo;
247}
248
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000249std::error_code current_path(SmallVectorImpl<char> &result) {
Rafael Espindola6ee16382013-08-10 00:50:57 +0000250 result.clear();
251
252 const char *pwd = ::getenv("PWD");
253 llvm::sys::fs::file_status PWDStatus, DotStatus;
254 if (pwd && llvm::sys::path::is_absolute(pwd) &&
255 !llvm::sys::fs::status(pwd, PWDStatus) &&
256 !llvm::sys::fs::status(".", DotStatus) &&
257 PWDStatus.getUniqueID() == DotStatus.getUniqueID()) {
258 result.append(pwd, pwd + strlen(pwd));
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000259 return std::error_code();
Rafael Espindola6ee16382013-08-10 00:50:57 +0000260 }
261
Sylvestre Ledru14ada942012-04-11 15:35:36 +0000262#ifdef MAXPATHLEN
Andrew Tricka4ec5b22011-03-24 16:43:37 +0000263 result.reserve(MAXPATHLEN);
Sylvestre Ledru14ada942012-04-11 15:35:36 +0000264#else
265// For GNU Hurd
266 result.reserve(1024);
267#endif
Michael J. Spencerebad2f92010-11-29 22:28:51 +0000268
Michael J. Spencer20daa282010-12-07 01:22:31 +0000269 while (true) {
Craig Toppere73658d2014-04-28 04:05:08 +0000270 if (::getcwd(result.data(), result.capacity()) == nullptr) {
Michael J. Spencer20daa282010-12-07 01:22:31 +0000271 // See if there was a real error.
Rafael Espindola882ce87b2014-05-31 02:29:28 +0000272 if (errno != ENOMEM)
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000273 return std::error_code(errno, std::generic_category());
Michael J. Spencer20daa282010-12-07 01:22:31 +0000274 // Otherwise there just wasn't enough space.
275 result.reserve(result.capacity() * 2);
276 } else
277 break;
278 }
Michael J. Spencerebad2f92010-11-29 22:28:51 +0000279
280 result.set_size(strlen(result.data()));
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000281 return std::error_code();
Michael J. Spencerebad2f92010-11-29 22:28:51 +0000282}
283
Pavel Labath2f096092017-01-24 10:32:03 +0000284std::error_code set_current_path(const Twine &path) {
285 SmallString<128> path_storage;
286 StringRef p = path.toNullTerminatedStringRef(path_storage);
287
288 if (::chdir(p.begin()) == -1)
289 return std::error_code(errno, std::generic_category());
290
291 return std::error_code();
292}
293
Frederic Riss6b9396c2015-08-06 21:04:55 +0000294std::error_code create_directory(const Twine &path, bool IgnoreExisting,
295 perms Perms) {
Michael J. Spencer31e310c2010-12-03 05:42:11 +0000296 SmallString<128> path_storage;
297 StringRef p = path.toNullTerminatedStringRef(path_storage);
298
Frederic Riss6b9396c2015-08-06 21:04:55 +0000299 if (::mkdir(p.begin(), Perms) == -1) {
Rafael Espindola882ce87b2014-05-31 02:29:28 +0000300 if (errno != EEXIST || !IgnoreExisting)
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000301 return std::error_code(errno, std::generic_category());
Rafael Espindola5c20ac02014-02-23 13:56:14 +0000302 }
Michael J. Spencer31e310c2010-12-03 05:42:11 +0000303
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000304 return std::error_code();
Michael J. Spencer31e310c2010-12-03 05:42:11 +0000305}
306
Rafael Espindola83f858e2014-03-11 18:40:24 +0000307// Note that we are using symbolic link because hard links are not supported by
308// all filesystems (SMB doesn't).
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000309std::error_code create_link(const Twine &to, const Twine &from) {
Michael J. Spencere0c45602010-12-03 05:58:41 +0000310 // Get arguments.
311 SmallString<128> from_storage;
312 SmallString<128> to_storage;
313 StringRef f = from.toNullTerminatedStringRef(from_storage);
314 StringRef t = to.toNullTerminatedStringRef(to_storage);
315
Rafael Espindola83f858e2014-03-11 18:40:24 +0000316 if (::symlink(t.begin(), f.begin()) == -1)
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000317 return std::error_code(errno, std::generic_category());
Michael J. Spencere0c45602010-12-03 05:58:41 +0000318
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000319 return std::error_code();
Michael J. Spencere0c45602010-12-03 05:58:41 +0000320}
321
Mehdi Amini8e13bc42016-12-14 04:56:42 +0000322std::error_code create_hard_link(const Twine &to, const Twine &from) {
323 // Get arguments.
324 SmallString<128> from_storage;
325 SmallString<128> to_storage;
326 StringRef f = from.toNullTerminatedStringRef(from_storage);
327 StringRef t = to.toNullTerminatedStringRef(to_storage);
328
329 if (::link(t.begin(), f.begin()) == -1)
330 return std::error_code(errno, std::generic_category());
331
332 return std::error_code();
333}
334
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000335std::error_code remove(const Twine &path, bool IgnoreNonExisting) {
Michael J. Spencer6e74e112010-12-03 17:53:43 +0000336 SmallString<128> path_storage;
337 StringRef p = path.toNullTerminatedStringRef(path_storage);
338
Rafael Espindola8cd62b02013-06-17 20:35:51 +0000339 struct stat buf;
Argyrios Kyrtzidis37575692014-03-21 01:25:37 +0000340 if (lstat(p.begin(), &buf) != 0) {
Rafael Espindola882ce87b2014-05-31 02:29:28 +0000341 if (errno != ENOENT || !IgnoreNonExisting)
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000342 return std::error_code(errno, std::generic_category());
343 return std::error_code();
Rafael Espindola8cd62b02013-06-17 20:35:51 +0000344 }
345
346 // Note: this check catches strange situations. In all cases, LLVM should
347 // only be involved in the creation and deletion of regular files. This
348 // check ensures that what we're trying to erase is a regular file. It
349 // effectively prevents LLVM from erasing things like /dev/null, any block
350 // special file, or other things that aren't "regular" files.
Argyrios Kyrtzidis37575692014-03-21 01:25:37 +0000351 if (!S_ISREG(buf.st_mode) && !S_ISDIR(buf.st_mode) && !S_ISLNK(buf.st_mode))
Rafael Espindola2a826e42014-06-13 17:20:48 +0000352 return make_error_code(errc::operation_not_permitted);
Rafael Espindola8cd62b02013-06-17 20:35:51 +0000353
Michael J. Spencer6e74e112010-12-03 17:53:43 +0000354 if (::remove(p.begin()) == -1) {
Rafael Espindola882ce87b2014-05-31 02:29:28 +0000355 if (errno != ENOENT || !IgnoreNonExisting)
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000356 return std::error_code(errno, std::generic_category());
Rafael Espindola5c20ac02014-02-23 13:56:14 +0000357 }
Michael J. Spencer6e74e112010-12-03 17:53:43 +0000358
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000359 return std::error_code();
Michael J. Spencer6e74e112010-12-03 17:53:43 +0000360}
361
Zachary Turner392ed9d2017-02-21 20:55:47 +0000362static bool is_local_impl(struct STATVFS &Vfs) {
363#if defined(__linux__)
Michal Gorny5ddd2a52017-02-22 18:09:15 +0000364#ifndef NFS_SUPER_MAGIC
365#define NFS_SUPER_MAGIC 0x6969
366#endif
367#ifndef SMB_SUPER_MAGIC
368#define SMB_SUPER_MAGIC 0x517B
369#endif
Zachary Turner6bc2dac2017-02-21 21:13:10 +0000370#ifndef CIFS_MAGIC_NUMBER
371#define CIFS_MAGIC_NUMBER 0xFF534D42
372#endif
Zachary Turner392ed9d2017-02-21 20:55:47 +0000373 switch ((uint32_t)Vfs.f_type) {
374 case NFS_SUPER_MAGIC:
375 case SMB_SUPER_MAGIC:
376 case CIFS_MAGIC_NUMBER:
377 return false;
378 default:
379 return true;
380 }
Nuno Lopes7a1bbd42017-03-09 13:43:31 +0000381#elif defined(__CYGWIN__)
382 // Cygwin doesn't expose this information; would need to use Win32 API.
383 return false;
Zachary Turner392ed9d2017-02-21 20:55:47 +0000384#else
385 return !!(STATVFS_F_FLAG(Vfs) & MNT_LOCAL);
386#endif
387}
388
389std::error_code is_local(const Twine &Path, bool &Result) {
390 struct STATVFS Vfs;
391 if (::STATVFS(Path.str().c_str(), &Vfs))
392 return std::error_code(errno, std::generic_category());
393
394 Result = is_local_impl(Vfs);
395 return std::error_code();
396}
397
398std::error_code is_local(int FD, bool &Result) {
399 struct STATVFS Vfs;
400 if (::FSTATVFS(FD, &Vfs))
401 return std::error_code(errno, std::generic_category());
402
403 Result = is_local_impl(Vfs);
404 return std::error_code();
405}
406
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000407std::error_code rename(const Twine &from, const Twine &to) {
Michael J. Spencer409f5562010-12-03 17:53:55 +0000408 // Get arguments.
409 SmallString<128> from_storage;
410 SmallString<128> to_storage;
411 StringRef f = from.toNullTerminatedStringRef(from_storage);
412 StringRef t = to.toNullTerminatedStringRef(to_storage);
413
Rafael Espindolab6fea4c2013-07-17 03:33:41 +0000414 if (::rename(f.begin(), t.begin()) == -1)
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000415 return std::error_code(errno, std::generic_category());
Michael J. Spencer409f5562010-12-03 17:53:55 +0000416
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000417 return std::error_code();
Michael J. Spencer409f5562010-12-03 17:53:55 +0000418}
419
Rafael Espindola59aaa6c2014-12-12 17:55:12 +0000420std::error_code resize_file(int FD, uint64_t Size) {
Rafael Espindola3816c532016-07-19 20:19:56 +0000421#if defined(HAVE_POSIX_FALLOCATE)
422 // If we have posix_fallocate use it. Unlike ftruncate it always allocates
423 // space, so we get an error if the disk is full.
Joerg Sonnenbergerecfb8762017-05-05 17:55:58 +0000424 if (int Err = ::posix_fallocate(FD, 0, Size)) {
425 if (Err != EOPNOTSUPP)
426 return std::error_code(Err, std::generic_category());
427 }
428#endif
Rafael Espindola3816c532016-07-19 20:19:56 +0000429 // Use ftruncate as a fallback. It may or may not allocate space. At least on
430 // OS X with HFS+ it does.
Rafael Espindola59aaa6c2014-12-12 17:55:12 +0000431 if (::ftruncate(FD, Size) == -1)
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000432 return std::error_code(errno, std::generic_category());
Michael J. Spencerc20a0322010-12-03 17:54:07 +0000433
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000434 return std::error_code();
Michael J. Spencerc20a0322010-12-03 17:54:07 +0000435}
436
Duncan P. N. Exon Smith91d3cfe2016-04-05 20:45:04 +0000437static int convertAccessMode(AccessMode Mode) {
Rafael Espindola281f23a2014-09-11 20:30:02 +0000438 switch (Mode) {
439 case AccessMode::Exist:
440 return F_OK;
441 case AccessMode::Write:
442 return W_OK;
443 case AccessMode::Execute:
444 return R_OK | X_OK; // scripts also need R_OK.
445 }
446 llvm_unreachable("invalid enum");
447}
Michael J. Spencer45710402010-12-03 01:21:28 +0000448
Rafael Espindola281f23a2014-09-11 20:30:02 +0000449std::error_code access(const Twine &Path, AccessMode Mode) {
450 SmallString<128> PathStorage;
451 StringRef P = Path.toNullTerminatedStringRef(PathStorage);
452
453 if (::access(P.begin(), convertAccessMode(Mode)) == -1)
454 return std::error_code(errno, std::generic_category());
455
456 if (Mode == AccessMode::Execute) {
457 // Don't say that directories are executable.
458 struct stat buf;
459 if (0 != stat(P.begin(), &buf))
460 return errc::permission_denied;
461 if (!S_ISREG(buf.st_mode))
462 return errc::permission_denied;
463 }
Michael J. Spencer45710402010-12-03 01:21:28 +0000464
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000465 return std::error_code();
Michael J. Spencer45710402010-12-03 01:21:28 +0000466}
467
Reid Kleckner89d4b1a2015-09-10 23:28:06 +0000468bool can_execute(const Twine &Path) {
469 return !access(Path, AccessMode::Execute);
470}
471
Michael J. Spencer203d7802011-12-12 06:04:28 +0000472bool equivalent(file_status A, file_status B) {
473 assert(status_known(A) && status_known(B));
Sylvestre Ledru3099f4bd2012-04-23 16:37:23 +0000474 return A.fs_st_dev == B.fs_st_dev &&
475 A.fs_st_ino == B.fs_st_ino;
Michael J. Spencer203d7802011-12-12 06:04:28 +0000476}
477
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000478std::error_code equivalent(const Twine &A, const Twine &B, bool &result) {
Michael J. Spencer203d7802011-12-12 06:04:28 +0000479 file_status fsA, fsB;
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000480 if (std::error_code ec = status(A, fsA))
481 return ec;
482 if (std::error_code ec = status(B, fsB))
483 return ec;
Michael J. Spencer203d7802011-12-12 06:04:28 +0000484 result = equivalent(fsA, fsB);
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000485 return std::error_code();
Michael J. Spencer376d3872010-12-03 18:49:13 +0000486}
487
Zachary Turnere48ace62017-03-10 17:39:21 +0000488static void expandTildeExpr(SmallVectorImpl<char> &Path) {
489 StringRef PathStr(Path.begin(), Path.size());
490 if (PathStr.empty() || !PathStr.startswith("~"))
491 return;
492
493 PathStr = PathStr.drop_front();
Zachary Turner18753342017-03-16 22:34:18 +0000494 StringRef Expr =
495 PathStr.take_until([](char c) { return path::is_separator(c); });
Zachary Turnere48ace62017-03-10 17:39:21 +0000496 StringRef Remainder = PathStr.substr(Expr.size() + 1);
497 SmallString<128> Storage;
498 if (Expr.empty()) {
499 // This is just ~/..., resolve it to the current user's home dir.
500 if (!path::home_directory(Storage)) {
501 // For some reason we couldn't get the home directory. Just exit.
502 return;
503 }
504
505 // Overwrite the first character and insert the rest.
506 Path[0] = Storage[0];
507 Path.insert(Path.begin() + 1, Storage.begin() + 1, Storage.end());
508 return;
509 }
510
511 // This is a string of the form ~username/, look up this user's entry in the
512 // password database.
513 struct passwd *Entry = nullptr;
514 std::string User = Expr.str();
515 Entry = ::getpwnam(User.c_str());
516
517 if (!Entry) {
518 // Unable to look up the entry, just return back the original path.
519 return;
520 }
521
522 Storage = Remainder;
523 Path.clear();
524 Path.append(Entry->pw_dir, Entry->pw_dir + strlen(Entry->pw_dir));
525 llvm::sys::path::append(Path, Storage);
526}
527
Duncan P. N. Exon Smith91d3cfe2016-04-05 20:45:04 +0000528static std::error_code fillStatus(int StatRet, const struct stat &Status,
529 file_status &Result) {
Rafael Espindola77021c92013-07-16 03:20:13 +0000530 if (StatRet != 0) {
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000531 std::error_code ec(errno, std::generic_category());
Rafael Espindola2a826e42014-06-13 17:20:48 +0000532 if (ec == errc::no_such_file_or_directory)
Rafael Espindola77021c92013-07-16 03:20:13 +0000533 Result = file_status(file_type::file_not_found);
Michael J. Spencerdb5576a2010-12-04 00:32:40 +0000534 else
Rafael Espindola77021c92013-07-16 03:20:13 +0000535 Result = file_status(file_type::status_error);
Michael J. Spencerdb5576a2010-12-04 00:32:40 +0000536 return ec;
537 }
538
Rafael Espindola9da91a02013-07-16 02:55:33 +0000539 file_type Type = file_type::type_unknown;
Michael J. Spencerdb5576a2010-12-04 00:32:40 +0000540
Rafael Espindola77021c92013-07-16 03:20:13 +0000541 if (S_ISDIR(Status.st_mode))
Rafael Espindola9da91a02013-07-16 02:55:33 +0000542 Type = file_type::directory_file;
Rafael Espindola77021c92013-07-16 03:20:13 +0000543 else if (S_ISREG(Status.st_mode))
Rafael Espindola9da91a02013-07-16 02:55:33 +0000544 Type = file_type::regular_file;
Rafael Espindola77021c92013-07-16 03:20:13 +0000545 else if (S_ISBLK(Status.st_mode))
Rafael Espindola9da91a02013-07-16 02:55:33 +0000546 Type = file_type::block_file;
Rafael Espindola77021c92013-07-16 03:20:13 +0000547 else if (S_ISCHR(Status.st_mode))
Rafael Espindola9da91a02013-07-16 02:55:33 +0000548 Type = file_type::character_file;
Rafael Espindola77021c92013-07-16 03:20:13 +0000549 else if (S_ISFIFO(Status.st_mode))
Rafael Espindola9da91a02013-07-16 02:55:33 +0000550 Type = file_type::fifo_file;
Rafael Espindola77021c92013-07-16 03:20:13 +0000551 else if (S_ISSOCK(Status.st_mode))
Rafael Espindola9da91a02013-07-16 02:55:33 +0000552 Type = file_type::socket_file;
Zachary Turner3b9fb882017-03-07 17:48:47 +0000553 else if (S_ISLNK(Status.st_mode))
554 Type = file_type::symlink_file;
Rafael Espindola9da91a02013-07-16 02:55:33 +0000555
James Henderson566fdf42017-03-16 11:22:09 +0000556 perms Perms = static_cast<perms>(Status.st_mode) & all_perms;
Zachary Turner5821a3b2017-03-20 23:55:20 +0000557 Result = file_status(Type, Perms, Status.st_dev, Status.st_nlink,
558 Status.st_ino, Status.st_atime, Status.st_mtime,
559 Status.st_uid, Status.st_gid, Status.st_size);
Michael J. Spencer203d7802011-12-12 06:04:28 +0000560
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000561 return std::error_code();
Michael J. Spencerdb5576a2010-12-04 00:32:40 +0000562}
563
Zachary Turner82dd5422017-03-07 16:10:10 +0000564std::error_code status(const Twine &Path, file_status &Result, bool Follow) {
Rafael Espindola77021c92013-07-16 03:20:13 +0000565 SmallString<128> PathStorage;
566 StringRef P = Path.toNullTerminatedStringRef(PathStorage);
567
568 struct stat Status;
Zachary Turner82dd5422017-03-07 16:10:10 +0000569 int StatRet = (Follow ? ::stat : ::lstat)(P.begin(), &Status);
Rafael Espindola77021c92013-07-16 03:20:13 +0000570 return fillStatus(StatRet, Status, Result);
571}
572
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000573std::error_code status(int FD, file_status &Result) {
Rafael Espindola77021c92013-07-16 03:20:13 +0000574 struct stat Status;
575 int StatRet = ::fstat(FD, &Status);
Aaron Ballman345012d2017-03-13 12:24:51 +0000576 return fillStatus(StatRet, Status, Result);
577}
578
James Henderson566fdf42017-03-16 11:22:09 +0000579std::error_code setPermissions(const Twine &Path, perms Permissions) {
580 SmallString<128> PathStorage;
581 StringRef P = Path.toNullTerminatedStringRef(PathStorage);
582
583 if (::chmod(P.begin(), Permissions))
584 return std::error_code(errno, std::generic_category());
585 return std::error_code();
586}
587
Aaron Ballman345012d2017-03-13 12:24:51 +0000588std::error_code setLastModificationAndAccessTime(int FD, TimePoint<> Time) {
589#if defined(HAVE_FUTIMENS)
590 timespec Times[2];
Pavel Labath757ca882016-10-24 10:59:17 +0000591 Times[0] = Times[1] = sys::toTimeSpec(Time);
Eric Christophera24dc7f2013-07-04 01:10:38 +0000592 if (::futimens(FD, Times))
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000593 return std::error_code(errno, std::generic_category());
594 return std::error_code();
Eric Christophera24dc7f2013-07-04 01:10:38 +0000595#elif defined(HAVE_FUTIMES)
Rafael Espindola4a3365c2013-06-20 20:56:14 +0000596 timeval Times[2];
Pavel Labath676a8752016-10-24 14:19:28 +0000597 Times[0] = Times[1] = sys::toTimeVal(
598 std::chrono::time_point_cast<std::chrono::microseconds>(Time));
Rafael Espindola4a3365c2013-06-20 20:56:14 +0000599 if (::futimes(FD, Times))
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000600 return std::error_code(errno, std::generic_category());
601 return std::error_code();
Alp Tokerb30f01e2013-12-11 15:42:33 +0000602#else
603#warning Missing futimes() and futimens()
Alp Tokerb792a012014-06-30 18:57:04 +0000604 return make_error_code(errc::function_not_supported);
Alp Tokerb30f01e2013-12-11 15:42:33 +0000605#endif
Rafael Espindola4a3365c2013-06-20 20:56:14 +0000606}
607
Rafael Espindola7eb1f182014-12-11 20:12:55 +0000608std::error_code mapped_file_region::init(int FD, uint64_t Offset,
609 mapmode Mode) {
Rafael Espindolac69f13b2014-12-12 18:13:23 +0000610 assert(Size != 0);
Michael J. Spenceref2284f2012-08-15 19:05:47 +0000611
612 int flags = (Mode == readwrite) ? MAP_SHARED : MAP_PRIVATE;
613 int prot = (Mode == readonly) ? PROT_READ : (PROT_READ | PROT_WRITE);
Zachary Turner842972b2017-02-22 21:24:06 +0000614#if defined(__APPLE__)
615 //----------------------------------------------------------------------
616 // Newer versions of MacOSX have a flag that will allow us to read from
617 // binaries whose code signature is invalid without crashing by using
618 // the MAP_RESILIENT_CODESIGN flag. Also if a file from removable media
619 // is mapped we can avoid crashing and return zeroes to any pages we try
620 // to read if the media becomes unavailable by using the
621 // MAP_RESILIENT_MEDIA flag. These flags are only usable when mapping
622 // with PROT_READ, so take care not to specify them otherwise.
623 //----------------------------------------------------------------------
624 if (Mode == readonly) {
625#if defined(MAP_RESILIENT_CODESIGN)
626 flags |= MAP_RESILIENT_CODESIGN;
627#endif
628#if defined(MAP_RESILIENT_MEDIA)
629 flags |= MAP_RESILIENT_MEDIA;
630#endif
631 }
632#endif // #if defined (__APPLE__)
Zachary Turner392ed9d2017-02-21 20:55:47 +0000633
Craig Toppere73658d2014-04-28 04:05:08 +0000634 Mapping = ::mmap(nullptr, Size, prot, flags, FD, Offset);
Michael J. Spenceref2284f2012-08-15 19:05:47 +0000635 if (Mapping == MAP_FAILED)
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000636 return std::error_code(errno, std::generic_category());
637 return std::error_code();
Michael J. Spenceref2284f2012-08-15 19:05:47 +0000638}
639
Rafael Espindola7eb1f182014-12-11 20:12:55 +0000640mapped_file_region::mapped_file_region(int fd, mapmode mode, uint64_t length,
641 uint64_t offset, std::error_code &ec)
642 : Size(length), Mapping() {
Michael J. Spenceref2284f2012-08-15 19:05:47 +0000643 // Make sure that the requested size fits within SIZE_T.
644 if (length > std::numeric_limits<size_t>::max()) {
Rafael Espindola2a826e42014-06-13 17:20:48 +0000645 ec = make_error_code(errc::invalid_argument);
Michael J. Spenceref2284f2012-08-15 19:05:47 +0000646 return;
647 }
648
Rafael Espindola7eb1f182014-12-11 20:12:55 +0000649 ec = init(fd, offset, mode);
Michael J. Spenceref2284f2012-08-15 19:05:47 +0000650 if (ec)
Craig Toppere73658d2014-04-28 04:05:08 +0000651 Mapping = nullptr;
Michael J. Spenceref2284f2012-08-15 19:05:47 +0000652}
653
654mapped_file_region::~mapped_file_region() {
655 if (Mapping)
656 ::munmap(Mapping, Size);
657}
658
Michael J. Spenceref2284f2012-08-15 19:05:47 +0000659uint64_t mapped_file_region::size() const {
660 assert(Mapping && "Mapping failed but used anyway!");
661 return Size;
662}
663
664char *mapped_file_region::data() const {
665 assert(Mapping && "Mapping failed but used anyway!");
Michael J. Spenceref2284f2012-08-15 19:05:47 +0000666 return reinterpret_cast<char*>(Mapping);
667}
668
669const char *mapped_file_region::const_data() const {
670 assert(Mapping && "Mapping failed but used anyway!");
671 return reinterpret_cast<const char*>(Mapping);
672}
673
674int mapped_file_region::alignment() {
Rafael Espindolac0610bf2014-12-04 16:59:36 +0000675 return Process::getPageSize();
Michael J. Spenceref2284f2012-08-15 19:05:47 +0000676}
677
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000678std::error_code detail::directory_iterator_construct(detail::DirIterState &it,
Zachary Turner260bda32017-03-08 22:49:32 +0000679 StringRef path,
680 bool follow_symlinks) {
Michael J. Spencer52714862011-01-05 16:38:57 +0000681 SmallString<128> path_null(path);
682 DIR *directory = ::opendir(path_null.c_str());
Craig Toppere73658d2014-04-28 04:05:08 +0000683 if (!directory)
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000684 return std::error_code(errno, std::generic_category());
Michael J. Spencer52714862011-01-05 16:38:57 +0000685
686 it.IterationHandle = reinterpret_cast<intptr_t>(directory);
687 // Add something for replace_filename to replace.
688 path::append(path_null, ".");
Zachary Turner260bda32017-03-08 22:49:32 +0000689 it.CurrentEntry = directory_entry(path_null.str(), follow_symlinks);
Michael J. Spencer52714862011-01-05 16:38:57 +0000690 return directory_iterator_increment(it);
691}
692
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000693std::error_code detail::directory_iterator_destruct(detail::DirIterState &it) {
Michael J. Spencer52714862011-01-05 16:38:57 +0000694 if (it.IterationHandle)
695 ::closedir(reinterpret_cast<DIR *>(it.IterationHandle));
696 it.IterationHandle = 0;
697 it.CurrentEntry = directory_entry();
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000698 return std::error_code();
Michael J. Spencer52714862011-01-05 16:38:57 +0000699}
700
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000701std::error_code detail::directory_iterator_increment(detail::DirIterState &it) {
Michael J. Spencer52714862011-01-05 16:38:57 +0000702 errno = 0;
703 dirent *cur_dir = ::readdir(reinterpret_cast<DIR *>(it.IterationHandle));
Craig Toppere73658d2014-04-28 04:05:08 +0000704 if (cur_dir == nullptr && errno != 0) {
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000705 return std::error_code(errno, std::generic_category());
Craig Toppere73658d2014-04-28 04:05:08 +0000706 } else if (cur_dir != nullptr) {
Michael J. Spencer52714862011-01-05 16:38:57 +0000707 StringRef name(cur_dir->d_name, NAMLEN(cur_dir));
708 if ((name.size() == 1 && name[0] == '.') ||
709 (name.size() == 2 && name[0] == '.' && name[1] == '.'))
710 return directory_iterator_increment(it);
711 it.CurrentEntry.replace_filename(name);
712 } else
713 return directory_iterator_destruct(it);
714
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000715 return std::error_code();
Michael J. Spencer52714862011-01-05 16:38:57 +0000716}
717
Taewook Ohd9153272016-06-13 15:54:56 +0000718#if !defined(F_GETPATH)
719static bool hasProcSelfFD() {
720 // If we have a /proc filesystem mounted, we can quickly establish the
721 // real name of the file with readlink
722 static const bool Result = (::access("/proc/self/fd", R_OK) == 0);
723 return Result;
724}
725#endif
726
727std::error_code openFileForRead(const Twine &Name, int &ResultFD,
728 SmallVectorImpl<char> *RealPath) {
Rafael Espindolaa0d9b6b2013-07-17 14:58:25 +0000729 SmallString<128> Storage;
730 StringRef P = Name.toNullTerminatedStringRef(Storage);
Pavel Labathf726dfa2017-01-24 10:57:01 +0000731 int OpenFlags = O_RDONLY;
732#ifdef O_CLOEXEC
733 OpenFlags |= O_CLOEXEC;
734#endif
735 while ((ResultFD = open(P.begin(), OpenFlags)) < 0) {
Rafael Espindolaa0d9b6b2013-07-17 14:58:25 +0000736 if (errno != EINTR)
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000737 return std::error_code(errno, std::generic_category());
Rafael Espindolaa0d9b6b2013-07-17 14:58:25 +0000738 }
Pavel Labathf726dfa2017-01-24 10:57:01 +0000739#ifndef O_CLOEXEC
740 int r = fcntl(ResultFD, F_SETFD, FD_CLOEXEC);
741 (void)r;
742 assert(r == 0 && "fcntl(F_SETFD, FD_CLOEXEC) failed");
743#endif
Taewook Ohd9153272016-06-13 15:54:56 +0000744 // Attempt to get the real name of the file, if the user asked
745 if(!RealPath)
746 return std::error_code();
747 RealPath->clear();
748#if defined(F_GETPATH)
749 // When F_GETPATH is availble, it is the quickest way to get
750 // the real path name.
751 char Buffer[MAXPATHLEN];
752 if (::fcntl(ResultFD, F_GETPATH, Buffer) != -1)
753 RealPath->append(Buffer, Buffer + strlen(Buffer));
754#else
755 char Buffer[PATH_MAX];
756 if (hasProcSelfFD()) {
757 char ProcPath[64];
758 snprintf(ProcPath, sizeof(ProcPath), "/proc/self/fd/%d", ResultFD);
759 ssize_t CharCount = ::readlink(ProcPath, Buffer, sizeof(Buffer));
760 if (CharCount > 0)
761 RealPath->append(Buffer, Buffer + CharCount);
762 } else {
763 // Use ::realpath to get the real path name
764 if (::realpath(P.begin(), Buffer) != nullptr)
765 RealPath->append(Buffer, Buffer + strlen(Buffer));
766 }
767#endif
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000768 return std::error_code();
Rafael Espindolaa0d9b6b2013-07-17 14:58:25 +0000769}
Nick Kledzik18497e92012-06-20 00:28:54 +0000770
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000771std::error_code openFileForWrite(const Twine &Name, int &ResultFD,
Rafael Espindola67080ce2013-07-19 15:02:03 +0000772 sys::fs::OpenFlags Flags, unsigned Mode) {
773 // Verify that we don't have both "append" and "excl".
774 assert((!(Flags & sys::fs::F_Excl) || !(Flags & sys::fs::F_Append)) &&
775 "Cannot specify both 'excl' and 'append' file creation flags!");
776
Pavel Labathf726dfa2017-01-24 10:57:01 +0000777 int OpenFlags = O_CREAT;
778
779#ifdef O_CLOEXEC
780 OpenFlags |= O_CLOEXEC;
781#endif
Rafael Espindola7a0b6402014-02-24 03:07:41 +0000782
783 if (Flags & F_RW)
784 OpenFlags |= O_RDWR;
785 else
786 OpenFlags |= O_WRONLY;
Rafael Espindola67080ce2013-07-19 15:02:03 +0000787
788 if (Flags & F_Append)
789 OpenFlags |= O_APPEND;
790 else
791 OpenFlags |= O_TRUNC;
792
793 if (Flags & F_Excl)
794 OpenFlags |= O_EXCL;
795
796 SmallString<128> Storage;
797 StringRef P = Name.toNullTerminatedStringRef(Storage);
798 while ((ResultFD = open(P.begin(), OpenFlags, Mode)) < 0) {
799 if (errno != EINTR)
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000800 return std::error_code(errno, std::generic_category());
Rafael Espindola67080ce2013-07-19 15:02:03 +0000801 }
Pavel Labathf726dfa2017-01-24 10:57:01 +0000802#ifndef O_CLOEXEC
803 int r = fcntl(ResultFD, F_SETFD, FD_CLOEXEC);
804 (void)r;
805 assert(r == 0 && "fcntl(F_SETFD, FD_CLOEXEC) failed");
806#endif
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000807 return std::error_code();
Rafael Espindola67080ce2013-07-19 15:02:03 +0000808}
809
Taewook Ohd9153272016-06-13 15:54:56 +0000810std::error_code getPathFromOpenFD(int FD, SmallVectorImpl<char> &ResultPath) {
811 if (FD < 0)
812 return make_error_code(errc::bad_file_descriptor);
813
814#if defined(F_GETPATH)
815 // When F_GETPATH is availble, it is the quickest way to get
816 // the path from a file descriptor.
817 ResultPath.reserve(MAXPATHLEN);
818 if (::fcntl(FD, F_GETPATH, ResultPath.begin()) == -1)
819 return std::error_code(errno, std::generic_category());
820
821 ResultPath.set_size(strlen(ResultPath.begin()));
822#else
823 // If we have a /proc filesystem mounted, we can quickly establish the
824 // real name of the file with readlink. Otherwise, we don't know how to
825 // get the filename from a file descriptor. Give up.
826 if (!fs::hasProcSelfFD())
827 return make_error_code(errc::function_not_supported);
828
829 ResultPath.reserve(PATH_MAX);
830 char ProcPath[64];
831 snprintf(ProcPath, sizeof(ProcPath), "/proc/self/fd/%d", FD);
832 ssize_t CharCount = ::readlink(ProcPath, ResultPath.begin(), ResultPath.capacity());
833 if (CharCount < 0)
834 return std::error_code(errno, std::generic_category());
835
836 // Was the filename truncated?
837 if (static_cast<size_t>(CharCount) == ResultPath.capacity()) {
838 // Use lstat to get the size of the filename
839 struct stat sb;
840 if (::lstat(ProcPath, &sb) < 0)
841 return std::error_code(errno, std::generic_category());
842
843 ResultPath.reserve(sb.st_size + 1);
844 CharCount = ::readlink(ProcPath, ResultPath.begin(), ResultPath.capacity());
845 if (CharCount < 0)
846 return std::error_code(errno, std::generic_category());
847
848 // Test for race condition: did the link size change?
849 if (CharCount > sb.st_size)
850 return std::error_code(ENAMETOOLONG, std::generic_category());
851 }
852 ResultPath.set_size(static_cast<size_t>(CharCount));
853#endif
854 return std::error_code();
855}
856
Zachary Turner260bda32017-03-08 22:49:32 +0000857template <typename T>
858static std::error_code remove_directories_impl(const T &Entry,
859 bool IgnoreErrors) {
860 std::error_code EC;
861 directory_iterator Begin(Entry, EC, false);
862 directory_iterator End;
863 while (Begin != End) {
864 auto &Item = *Begin;
865 file_status st;
866 EC = Item.status(st);
867 if (EC && !IgnoreErrors)
868 return EC;
869
870 if (is_directory(st)) {
871 EC = remove_directories_impl(Item, IgnoreErrors);
872 if (EC && !IgnoreErrors)
873 return EC;
874 }
875
876 EC = fs::remove(Item.path(), true);
877 if (EC && !IgnoreErrors)
878 return EC;
879
880 Begin.increment(EC);
881 if (EC && !IgnoreErrors)
882 return EC;
883 }
884 return std::error_code();
885}
886
887std::error_code remove_directories(const Twine &path, bool IgnoreErrors) {
888 auto EC = remove_directories_impl(path, IgnoreErrors);
889 if (EC && !IgnoreErrors)
890 return EC;
891 EC = fs::remove(path, true);
892 if (EC && !IgnoreErrors)
893 return EC;
894 return std::error_code();
895}
896
Zachary Turnere48ace62017-03-10 17:39:21 +0000897std::error_code real_path(const Twine &path, SmallVectorImpl<char> &dest,
898 bool expand_tilde) {
899 dest.clear();
900 if (path.isTriviallyEmpty())
901 return std::error_code();
902
903 if (expand_tilde) {
904 SmallString<128> Storage;
905 path.toVector(Storage);
906 expandTildeExpr(Storage);
907 return real_path(Storage, dest, false);
908 }
909
910 int fd;
911 std::error_code EC = openFileForRead(path, fd, &dest);
912
913 if (EC)
914 return EC;
915 ::close(fd);
916 return std::error_code();
917}
918
Michael J. Spencer9fc1d9d2010-12-01 19:32:01 +0000919} // end namespace fs
Peter Collingbournef7d41012014-01-31 23:46:06 +0000920
921namespace path {
922
923bool home_directory(SmallVectorImpl<char> &result) {
Zachary Turnera3cf70b2017-03-22 15:24:59 +0000924 char *RequestedDir = getenv("HOME");
925 if (!RequestedDir) {
926 struct passwd *pw = getpwuid(getuid());
927 if (pw && pw->pw_dir)
928 RequestedDir = pw->pw_dir;
Peter Collingbournef7d41012014-01-31 23:46:06 +0000929 }
Zachary Turnera3cf70b2017-03-22 15:24:59 +0000930 if (!RequestedDir)
931 return false;
932
933 result.clear();
934 result.append(RequestedDir, RequestedDir + strlen(RequestedDir));
935 return true;
Peter Collingbournef7d41012014-01-31 23:46:06 +0000936}
937
Duncan P. N. Exon Smith91d3cfe2016-04-05 20:45:04 +0000938static bool getDarwinConfDir(bool TempDir, SmallVectorImpl<char> &Result) {
Pawel Bylica0e97e5c2015-11-02 09:49:17 +0000939 #if defined(_CS_DARWIN_USER_TEMP_DIR) && defined(_CS_DARWIN_USER_CACHE_DIR)
940 // On Darwin, use DARWIN_USER_TEMP_DIR or DARWIN_USER_CACHE_DIR.
941 // macros defined in <unistd.h> on darwin >= 9
942 int ConfName = TempDir ? _CS_DARWIN_USER_TEMP_DIR
943 : _CS_DARWIN_USER_CACHE_DIR;
944 size_t ConfLen = confstr(ConfName, nullptr, 0);
945 if (ConfLen > 0) {
946 do {
947 Result.resize(ConfLen);
948 ConfLen = confstr(ConfName, Result.data(), Result.size());
949 } while (ConfLen > 0 && ConfLen != Result.size());
950
951 if (ConfLen > 0) {
952 assert(Result.back() == 0);
953 Result.pop_back();
954 return true;
955 }
956
957 Result.clear();
958 }
959 #endif
960 return false;
961}
962
Duncan P. N. Exon Smith91d3cfe2016-04-05 20:45:04 +0000963static bool getUserCacheDir(SmallVectorImpl<char> &Result) {
Sean Silvaa915a162016-03-24 22:27:27 +0000964 // First try using XDG_CACHE_HOME env variable,
Pawel Bylica0e97e5c2015-11-02 09:49:17 +0000965 // as specified in XDG Base Directory Specification at
966 // http://standards.freedesktop.org/basedir-spec/basedir-spec-latest.html
Sean Silvaa915a162016-03-24 22:27:27 +0000967 if (const char *XdgCacheDir = std::getenv("XDG_CACHE_HOME")) {
Pawel Bylica0e97e5c2015-11-02 09:49:17 +0000968 Result.clear();
Sean Silvaa915a162016-03-24 22:27:27 +0000969 Result.append(XdgCacheDir, XdgCacheDir + strlen(XdgCacheDir));
Pawel Bylica0e97e5c2015-11-02 09:49:17 +0000970 return true;
971 }
972
973 // Try Darwin configuration query
974 if (getDarwinConfDir(false, Result))
975 return true;
976
977 // Use "$HOME/.cache" if $HOME is available
978 if (home_directory(Result)) {
979 append(Result, ".cache");
980 return true;
981 }
982
983 return false;
984}
Pawel Bylica0e97e5c2015-11-02 09:49:17 +0000985
Duncan P. N. Exon Smith91d3cfe2016-04-05 20:45:04 +0000986static const char *getEnvTempDir() {
Rafael Espindola016a6d52014-08-26 14:47:52 +0000987 // Check whether the temporary directory is specified by an environment
988 // variable.
989 const char *EnvironmentVariables[] = {"TMPDIR", "TMP", "TEMP", "TEMPDIR"};
990 for (const char *Env : EnvironmentVariables) {
991 if (const char *Dir = std::getenv(Env))
992 return Dir;
993 }
994
995 return nullptr;
996}
997
Duncan P. N. Exon Smith91d3cfe2016-04-05 20:45:04 +0000998static const char *getDefaultTempDir(bool ErasedOnReboot) {
Rafael Espindola016a6d52014-08-26 14:47:52 +0000999#ifdef P_tmpdir
Benjamin Kramer870d9512014-08-27 11:47:52 +00001000 if ((bool)P_tmpdir)
Rafael Espindola016a6d52014-08-26 14:47:52 +00001001 return P_tmpdir;
1002#endif
1003
1004 if (ErasedOnReboot)
1005 return "/tmp";
1006 return "/var/tmp";
1007}
1008
1009void system_temp_directory(bool ErasedOnReboot, SmallVectorImpl<char> &Result) {
1010 Result.clear();
1011
1012 if (ErasedOnReboot) {
1013 // There is no env variable for the cache directory.
1014 if (const char *RequestedDir = getEnvTempDir()) {
1015 Result.append(RequestedDir, RequestedDir + strlen(RequestedDir));
1016 return;
1017 }
1018 }
1019
Pawel Bylica0e97e5c2015-11-02 09:49:17 +00001020 if (getDarwinConfDir(ErasedOnReboot, Result))
1021 return;
Rafael Espindola016a6d52014-08-26 14:47:52 +00001022
1023 const char *RequestedDir = getDefaultTempDir(ErasedOnReboot);
1024 Result.append(RequestedDir, RequestedDir + strlen(RequestedDir));
1025}
1026
Peter Collingbournef7d41012014-01-31 23:46:06 +00001027} // end namespace path
1028
Michael J. Spencerebad2f92010-11-29 22:28:51 +00001029} // end namespace sys
1030} // end namespace llvm