blob: 76e1ed2d418ab67b1a1b3b9f3400a473dda03fa2 [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
Rafael Espindola4601c462013-06-26 05:25:44 +000051#ifdef __APPLE__
52#include <mach-o/dyld.h>
Taewook Ohd9153272016-06-13 15:54:56 +000053#include <sys/attr.h>
Rafael Espindola4601c462013-06-26 05:25:44 +000054#endif
55
Simon Pilgrimf2fbf432016-11-20 13:47:59 +000056// Both stdio.h and cstdio are included via different paths and
Joerg Sonnenbergerc0697302012-08-10 10:56:09 +000057// stdcxx's cstdio doesn't include stdio.h, so it doesn't #undef the macros
58// either.
59#undef ferror
60#undef feof
61
Sylvestre Ledru14ada942012-04-11 15:35:36 +000062// For GNU Hurd
63#if defined(__GNU__) && !defined(PATH_MAX)
64# define PATH_MAX 4096
65#endif
66
Mehdi Aminie2d8f1b2016-04-01 00:18:08 +000067#include <sys/types.h>
Zachary Turner392ed9d2017-02-21 20:55:47 +000068#if !defined(__APPLE__) && !defined(__OpenBSD__) && !defined(__FreeBSD__) && \
69 !defined(__linux__)
Mehdi Aminie2d8f1b2016-04-01 00:18:08 +000070#include <sys/statvfs.h>
71#define STATVFS statvfs
Zachary Turner392ed9d2017-02-21 20:55:47 +000072#define FSTATVFS fstatvfs
Mehdi Aminie2d8f1b2016-04-01 00:18:08 +000073#define STATVFS_F_FRSIZE(vfs) vfs.f_frsize
74#else
Zachary Turner392ed9d2017-02-21 20:55:47 +000075#if defined(__OpenBSD__) || defined(__FreeBSD__)
Mehdi Aminie2d8f1b2016-04-01 00:18:08 +000076#include <sys/param.h>
Mehdi Amini3ba84ca2016-04-08 16:45:05 +000077#include <sys/mount.h>
Zachary Turner392ed9d2017-02-21 20:55:47 +000078#elif defined(__linux__)
Michal Gorny5ddd2a52017-02-22 18:09:15 +000079#if defined(HAVE_LINUX_MAGIC_H)
Zachary Turner392ed9d2017-02-21 20:55:47 +000080#include <linux/magic.h>
Michal Gorny5ddd2a52017-02-22 18:09:15 +000081#else
82#if defined(HAVE_LINUX_NFS_FS_H)
83#include <linux/nfs_fs.h>
84#endif
85#if defined(HAVE_LINUX_SMB_H)
86#include <linux/smb.h>
87#endif
88#endif
Mehdi Aminie2d8f1b2016-04-01 00:18:08 +000089#include <sys/vfs.h>
90#else
91#include <sys/mount.h>
92#endif
93#define STATVFS statfs
Zachary Turner392ed9d2017-02-21 20:55:47 +000094#define FSTATVFS fstatfs
Mehdi Aminie2d8f1b2016-04-01 00:18:08 +000095#define STATVFS_F_FRSIZE(vfs) static_cast<uint64_t>(vfs.f_bsize)
96#endif
97
Zachary Turner392ed9d2017-02-21 20:55:47 +000098#if defined(__NetBSD__)
99#define STATVFS_F_FLAG(vfs) (vfs).f_flag
100#else
101#define STATVFS_F_FLAG(vfs) (vfs).f_flags
102#endif
Mehdi Aminie2d8f1b2016-04-01 00:18:08 +0000103
Michael J. Spencer45710402010-12-03 01:21:28 +0000104using namespace llvm;
105
Michael J. Spencerebad2f92010-11-29 22:28:51 +0000106namespace llvm {
107namespace sys {
Michael J. Spencer20daa282010-12-07 01:22:31 +0000108namespace fs {
Rafael Espindolae03dfd92013-06-26 05:01:35 +0000109#if defined(__FreeBSD__) || defined (__NetBSD__) || defined(__Bitrig__) || \
110 defined(__OpenBSD__) || defined(__minix) || defined(__FreeBSD_kernel__) || \
Chandler Carruth2aff7502016-07-19 22:46:39 +0000111 defined(__linux__) || defined(__CYGWIN__) || defined(__DragonFly__) || \
112 defined(_AIX)
Duncan P. N. Exon Smith91d3cfe2016-04-05 20:45:04 +0000113static int
Sylvestre Ledru21e674722013-12-09 16:27:00 +0000114test_dir(char ret[PATH_MAX], const char *dir, const char *bin)
Mehdi Aminie2d8f1b2016-04-01 00:18:08 +0000115{
Rafael Espindolae03dfd92013-06-26 05:01:35 +0000116 struct stat sb;
Sylvestre Ledru21e674722013-12-09 16:27:00 +0000117 char fullpath[PATH_MAX];
Rafael Espindolae03dfd92013-06-26 05:01:35 +0000118
Sylvestre Ledru21e674722013-12-09 16:27:00 +0000119 snprintf(fullpath, PATH_MAX, "%s/%s", dir, bin);
Hans Wennborg083ca9b2015-10-06 23:24:35 +0000120 if (!realpath(fullpath, ret))
121 return 1;
Sylvestre Ledru21e674722013-12-09 16:27:00 +0000122 if (stat(fullpath, &sb) != 0)
Hans Wennborg083ca9b2015-10-06 23:24:35 +0000123 return 1;
Rafael Espindolae03dfd92013-06-26 05:01:35 +0000124
Hans Wennborg083ca9b2015-10-06 23:24:35 +0000125 return 0;
Rafael Espindolae03dfd92013-06-26 05:01:35 +0000126}
127
Duncan P. N. Exon Smith91d3cfe2016-04-05 20:45:04 +0000128static char *
Rafael Espindolae03dfd92013-06-26 05:01:35 +0000129getprogpath(char ret[PATH_MAX], const char *bin)
130{
Sylvestre Ledru21e674722013-12-09 16:27:00 +0000131 char *pv, *s, *t;
Rafael Espindolae03dfd92013-06-26 05:01:35 +0000132
133 /* First approach: absolute path. */
134 if (bin[0] == '/') {
Sylvestre Ledru21e674722013-12-09 16:27:00 +0000135 if (test_dir(ret, "/", bin) == 0)
Hans Wennborg083ca9b2015-10-06 23:24:35 +0000136 return ret;
137 return nullptr;
Rafael Espindolae03dfd92013-06-26 05:01:35 +0000138 }
139
140 /* Second approach: relative path. */
Hans Wennborg083ca9b2015-10-06 23:24:35 +0000141 if (strchr(bin, '/')) {
Sylvestre Ledru21e674722013-12-09 16:27:00 +0000142 char cwd[PATH_MAX];
Hans Wennborg083ca9b2015-10-06 23:24:35 +0000143 if (!getcwd(cwd, PATH_MAX))
144 return nullptr;
Sylvestre Ledru21e674722013-12-09 16:27:00 +0000145 if (test_dir(ret, cwd, bin) == 0)
Hans Wennborg083ca9b2015-10-06 23:24:35 +0000146 return ret;
147 return nullptr;
Rafael Espindolae03dfd92013-06-26 05:01:35 +0000148 }
149
150 /* Third approach: $PATH */
Hans Wennborg083ca9b2015-10-06 23:24:35 +0000151 if ((pv = getenv("PATH")) == nullptr)
152 return nullptr;
Rafael Espindolae03dfd92013-06-26 05:01:35 +0000153 s = pv = strdup(pv);
Hans Wennborg083ca9b2015-10-06 23:24:35 +0000154 if (!pv)
155 return nullptr;
156 while ((t = strsep(&s, ":")) != nullptr) {
Sylvestre Ledru21e674722013-12-09 16:27:00 +0000157 if (test_dir(ret, t, bin) == 0) {
Rafael Espindolae03dfd92013-06-26 05:01:35 +0000158 free(pv);
Hans Wennborg083ca9b2015-10-06 23:24:35 +0000159 return ret;
Rafael Espindolae03dfd92013-06-26 05:01:35 +0000160 }
161 }
162 free(pv);
Hans Wennborg083ca9b2015-10-06 23:24:35 +0000163 return nullptr;
Rafael Espindolae03dfd92013-06-26 05:01:35 +0000164}
165#endif // __FreeBSD__ || __NetBSD__ || __FreeBSD_kernel__
166
167/// GetMainExecutable - Return the path to the main executable, given the
168/// value of argv[0] from program startup.
169std::string getMainExecutable(const char *argv0, void *MainAddr) {
170#if defined(__APPLE__)
171 // On OS X the executable path is saved to the stack by dyld. Reading it
172 // from there is much faster than calling dladdr, especially for large
173 // binaries with symbols.
174 char exe_path[MAXPATHLEN];
175 uint32_t size = sizeof(exe_path);
176 if (_NSGetExecutablePath(exe_path, &size) == 0) {
177 char link_path[MAXPATHLEN];
178 if (realpath(exe_path, link_path))
Rafael Espindola4601c462013-06-26 05:25:44 +0000179 return link_path;
Rafael Espindolae03dfd92013-06-26 05:01:35 +0000180 }
181#elif defined(__FreeBSD__) || defined (__NetBSD__) || defined(__Bitrig__) || \
Rafael Espindolaaca97392013-10-31 14:35:00 +0000182 defined(__OpenBSD__) || defined(__minix) || defined(__DragonFly__) || \
Chandler Carruth2aff7502016-07-19 22:46:39 +0000183 defined(__FreeBSD_kernel__) || defined(_AIX)
Rafael Espindolae03dfd92013-06-26 05:01:35 +0000184 char exe_path[PATH_MAX];
185
186 if (getprogpath(exe_path, argv0) != NULL)
Rafael Espindola2c6f4fe2013-06-26 06:10:32 +0000187 return exe_path;
Rafael Espindolae03dfd92013-06-26 05:01:35 +0000188#elif defined(__linux__) || defined(__CYGWIN__)
189 char exe_path[MAXPATHLEN];
190 StringRef aPath("/proc/self/exe");
191 if (sys::fs::exists(aPath)) {
192 // /proc is not always mounted under Linux (chroot for example).
193 ssize_t len = readlink(aPath.str().c_str(), exe_path, sizeof(exe_path));
194 if (len >= 0)
Alexey Samsonov4c798452014-12-29 20:59:02 +0000195 return std::string(exe_path, len);
Rafael Espindolae03dfd92013-06-26 05:01:35 +0000196 } else {
197 // Fall back to the classical detection.
Hans Wennborg083ca9b2015-10-06 23:24:35 +0000198 if (getprogpath(exe_path, argv0))
199 return exe_path;
Rafael Espindolae03dfd92013-06-26 05:01:35 +0000200 }
Omair Javaidf5d560bc842017-02-02 01:17:49 +0000201#elif defined(HAVE_DLFCN_H) && defined(HAVE_DLADDR)
Rafael Espindolae03dfd92013-06-26 05:01:35 +0000202 // Use dladdr to get executable path if available.
203 Dl_info DLInfo;
204 int err = dladdr(MainAddr, &DLInfo);
205 if (err == 0)
Rafael Espindola2c6f4fe2013-06-26 06:10:32 +0000206 return "";
Rafael Espindolae03dfd92013-06-26 05:01:35 +0000207
208 // If the filename is a symlink, we need to resolve and return the location of
209 // the actual executable.
210 char link_path[MAXPATHLEN];
211 if (realpath(DLInfo.dli_fname, link_path))
Rafael Espindola2c6f4fe2013-06-26 06:10:32 +0000212 return link_path;
Rafael Espindolae03dfd92013-06-26 05:01:35 +0000213#else
214#error GetMainExecutable is not implemented on this host yet.
215#endif
216 return "";
217}
Michael J. Spencerebad2f92010-11-29 22:28:51 +0000218
Pavel Labath757ca882016-10-24 10:59:17 +0000219TimePoint<> file_status::getLastAccessedTime() const {
220 return toTimePoint(fs_st_atime);
Mehdi Amini1e39ef32016-03-25 07:30:21 +0000221}
222
Pavel Labath757ca882016-10-24 10:59:17 +0000223TimePoint<> file_status::getLastModificationTime() const {
224 return toTimePoint(fs_st_mtime);
Rafael Espindoladb5d8fe2013-06-20 18:42:04 +0000225}
226
Rafael Espindolad1230992013-07-29 21:55:38 +0000227UniqueID file_status::getUniqueID() const {
Rafael Espindola7f822a92013-07-29 21:26:49 +0000228 return UniqueID(fs_st_dev, fs_st_ino);
229}
230
Mehdi Aminie2d8f1b2016-04-01 00:18:08 +0000231ErrorOr<space_info> disk_space(const Twine &Path) {
232 struct STATVFS Vfs;
233 if (::STATVFS(Path.str().c_str(), &Vfs))
234 return std::error_code(errno, std::generic_category());
235 auto FrSize = STATVFS_F_FRSIZE(Vfs);
236 space_info SpaceInfo;
237 SpaceInfo.capacity = static_cast<uint64_t>(Vfs.f_blocks) * FrSize;
238 SpaceInfo.free = static_cast<uint64_t>(Vfs.f_bfree) * FrSize;
239 SpaceInfo.available = static_cast<uint64_t>(Vfs.f_bavail) * FrSize;
240 return SpaceInfo;
241}
242
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000243std::error_code current_path(SmallVectorImpl<char> &result) {
Rafael Espindola6ee16382013-08-10 00:50:57 +0000244 result.clear();
245
246 const char *pwd = ::getenv("PWD");
247 llvm::sys::fs::file_status PWDStatus, DotStatus;
248 if (pwd && llvm::sys::path::is_absolute(pwd) &&
249 !llvm::sys::fs::status(pwd, PWDStatus) &&
250 !llvm::sys::fs::status(".", DotStatus) &&
251 PWDStatus.getUniqueID() == DotStatus.getUniqueID()) {
252 result.append(pwd, pwd + strlen(pwd));
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000253 return std::error_code();
Rafael Espindola6ee16382013-08-10 00:50:57 +0000254 }
255
Sylvestre Ledru14ada942012-04-11 15:35:36 +0000256#ifdef MAXPATHLEN
Andrew Tricka4ec5b22011-03-24 16:43:37 +0000257 result.reserve(MAXPATHLEN);
Sylvestre Ledru14ada942012-04-11 15:35:36 +0000258#else
259// For GNU Hurd
260 result.reserve(1024);
261#endif
Michael J. Spencerebad2f92010-11-29 22:28:51 +0000262
Michael J. Spencer20daa282010-12-07 01:22:31 +0000263 while (true) {
Craig Toppere73658d2014-04-28 04:05:08 +0000264 if (::getcwd(result.data(), result.capacity()) == nullptr) {
Michael J. Spencer20daa282010-12-07 01:22:31 +0000265 // See if there was a real error.
Rafael Espindola882ce87b2014-05-31 02:29:28 +0000266 if (errno != ENOMEM)
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000267 return std::error_code(errno, std::generic_category());
Michael J. Spencer20daa282010-12-07 01:22:31 +0000268 // Otherwise there just wasn't enough space.
269 result.reserve(result.capacity() * 2);
270 } else
271 break;
272 }
Michael J. Spencerebad2f92010-11-29 22:28:51 +0000273
274 result.set_size(strlen(result.data()));
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000275 return std::error_code();
Michael J. Spencerebad2f92010-11-29 22:28:51 +0000276}
277
Pavel Labath2f096092017-01-24 10:32:03 +0000278std::error_code set_current_path(const Twine &path) {
279 SmallString<128> path_storage;
280 StringRef p = path.toNullTerminatedStringRef(path_storage);
281
282 if (::chdir(p.begin()) == -1)
283 return std::error_code(errno, std::generic_category());
284
285 return std::error_code();
286}
287
Frederic Riss6b9396c2015-08-06 21:04:55 +0000288std::error_code create_directory(const Twine &path, bool IgnoreExisting,
289 perms Perms) {
Michael J. Spencer31e310c2010-12-03 05:42:11 +0000290 SmallString<128> path_storage;
291 StringRef p = path.toNullTerminatedStringRef(path_storage);
292
Frederic Riss6b9396c2015-08-06 21:04:55 +0000293 if (::mkdir(p.begin(), Perms) == -1) {
Rafael Espindola882ce87b2014-05-31 02:29:28 +0000294 if (errno != EEXIST || !IgnoreExisting)
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000295 return std::error_code(errno, std::generic_category());
Rafael Espindola5c20ac02014-02-23 13:56:14 +0000296 }
Michael J. Spencer31e310c2010-12-03 05:42:11 +0000297
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000298 return std::error_code();
Michael J. Spencer31e310c2010-12-03 05:42:11 +0000299}
300
Rafael Espindola83f858e2014-03-11 18:40:24 +0000301// Note that we are using symbolic link because hard links are not supported by
302// all filesystems (SMB doesn't).
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000303std::error_code create_link(const Twine &to, const Twine &from) {
Michael J. Spencere0c45602010-12-03 05:58:41 +0000304 // Get arguments.
305 SmallString<128> from_storage;
306 SmallString<128> to_storage;
307 StringRef f = from.toNullTerminatedStringRef(from_storage);
308 StringRef t = to.toNullTerminatedStringRef(to_storage);
309
Rafael Espindola83f858e2014-03-11 18:40:24 +0000310 if (::symlink(t.begin(), f.begin()) == -1)
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000311 return std::error_code(errno, std::generic_category());
Michael J. Spencere0c45602010-12-03 05:58:41 +0000312
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000313 return std::error_code();
Michael J. Spencere0c45602010-12-03 05:58:41 +0000314}
315
Mehdi Amini8e13bc42016-12-14 04:56:42 +0000316std::error_code create_hard_link(const Twine &to, const Twine &from) {
317 // Get arguments.
318 SmallString<128> from_storage;
319 SmallString<128> to_storage;
320 StringRef f = from.toNullTerminatedStringRef(from_storage);
321 StringRef t = to.toNullTerminatedStringRef(to_storage);
322
323 if (::link(t.begin(), f.begin()) == -1)
324 return std::error_code(errno, std::generic_category());
325
326 return std::error_code();
327}
328
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000329std::error_code remove(const Twine &path, bool IgnoreNonExisting) {
Michael J. Spencer6e74e112010-12-03 17:53:43 +0000330 SmallString<128> path_storage;
331 StringRef p = path.toNullTerminatedStringRef(path_storage);
332
Rafael Espindola8cd62b02013-06-17 20:35:51 +0000333 struct stat buf;
Argyrios Kyrtzidis37575692014-03-21 01:25:37 +0000334 if (lstat(p.begin(), &buf) != 0) {
Rafael Espindola882ce87b2014-05-31 02:29:28 +0000335 if (errno != ENOENT || !IgnoreNonExisting)
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000336 return std::error_code(errno, std::generic_category());
337 return std::error_code();
Rafael Espindola8cd62b02013-06-17 20:35:51 +0000338 }
339
340 // Note: this check catches strange situations. In all cases, LLVM should
341 // only be involved in the creation and deletion of regular files. This
342 // check ensures that what we're trying to erase is a regular file. It
343 // effectively prevents LLVM from erasing things like /dev/null, any block
344 // special file, or other things that aren't "regular" files.
Argyrios Kyrtzidis37575692014-03-21 01:25:37 +0000345 if (!S_ISREG(buf.st_mode) && !S_ISDIR(buf.st_mode) && !S_ISLNK(buf.st_mode))
Rafael Espindola2a826e42014-06-13 17:20:48 +0000346 return make_error_code(errc::operation_not_permitted);
Rafael Espindola8cd62b02013-06-17 20:35:51 +0000347
Michael J. Spencer6e74e112010-12-03 17:53:43 +0000348 if (::remove(p.begin()) == -1) {
Rafael Espindola882ce87b2014-05-31 02:29:28 +0000349 if (errno != ENOENT || !IgnoreNonExisting)
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000350 return std::error_code(errno, std::generic_category());
Rafael Espindola5c20ac02014-02-23 13:56:14 +0000351 }
Michael J. Spencer6e74e112010-12-03 17:53:43 +0000352
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000353 return std::error_code();
Michael J. Spencer6e74e112010-12-03 17:53:43 +0000354}
355
Zachary Turner392ed9d2017-02-21 20:55:47 +0000356static bool is_local_impl(struct STATVFS &Vfs) {
357#if defined(__linux__)
Michal Gorny5ddd2a52017-02-22 18:09:15 +0000358#ifndef NFS_SUPER_MAGIC
359#define NFS_SUPER_MAGIC 0x6969
360#endif
361#ifndef SMB_SUPER_MAGIC
362#define SMB_SUPER_MAGIC 0x517B
363#endif
Zachary Turner6bc2dac2017-02-21 21:13:10 +0000364#ifndef CIFS_MAGIC_NUMBER
365#define CIFS_MAGIC_NUMBER 0xFF534D42
366#endif
Zachary Turner392ed9d2017-02-21 20:55:47 +0000367 switch ((uint32_t)Vfs.f_type) {
368 case NFS_SUPER_MAGIC:
369 case SMB_SUPER_MAGIC:
370 case CIFS_MAGIC_NUMBER:
371 return false;
372 default:
373 return true;
374 }
375#else
376 return !!(STATVFS_F_FLAG(Vfs) & MNT_LOCAL);
377#endif
378}
379
380std::error_code is_local(const Twine &Path, bool &Result) {
381 struct STATVFS Vfs;
382 if (::STATVFS(Path.str().c_str(), &Vfs))
383 return std::error_code(errno, std::generic_category());
384
385 Result = is_local_impl(Vfs);
386 return std::error_code();
387}
388
389std::error_code is_local(int FD, bool &Result) {
390 struct STATVFS Vfs;
391 if (::FSTATVFS(FD, &Vfs))
392 return std::error_code(errno, std::generic_category());
393
394 Result = is_local_impl(Vfs);
395 return std::error_code();
396}
397
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000398std::error_code rename(const Twine &from, const Twine &to) {
Michael J. Spencer409f5562010-12-03 17:53:55 +0000399 // Get arguments.
400 SmallString<128> from_storage;
401 SmallString<128> to_storage;
402 StringRef f = from.toNullTerminatedStringRef(from_storage);
403 StringRef t = to.toNullTerminatedStringRef(to_storage);
404
Rafael Espindolab6fea4c2013-07-17 03:33:41 +0000405 if (::rename(f.begin(), t.begin()) == -1)
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000406 return std::error_code(errno, std::generic_category());
Michael J. Spencer409f5562010-12-03 17:53:55 +0000407
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000408 return std::error_code();
Michael J. Spencer409f5562010-12-03 17:53:55 +0000409}
410
Rafael Espindola59aaa6c2014-12-12 17:55:12 +0000411std::error_code resize_file(int FD, uint64_t Size) {
Rafael Espindola3816c532016-07-19 20:19:56 +0000412#if defined(HAVE_POSIX_FALLOCATE)
413 // If we have posix_fallocate use it. Unlike ftruncate it always allocates
414 // space, so we get an error if the disk is full.
415 if (int Err = ::posix_fallocate(FD, 0, Size))
416 return std::error_code(Err, std::generic_category());
417#else
418 // Use ftruncate as a fallback. It may or may not allocate space. At least on
419 // OS X with HFS+ it does.
Rafael Espindola59aaa6c2014-12-12 17:55:12 +0000420 if (::ftruncate(FD, Size) == -1)
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000421 return std::error_code(errno, std::generic_category());
Rafael Espindola3816c532016-07-19 20:19:56 +0000422#endif
Michael J. Spencerc20a0322010-12-03 17:54:07 +0000423
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000424 return std::error_code();
Michael J. Spencerc20a0322010-12-03 17:54:07 +0000425}
426
Duncan P. N. Exon Smith91d3cfe2016-04-05 20:45:04 +0000427static int convertAccessMode(AccessMode Mode) {
Rafael Espindola281f23a2014-09-11 20:30:02 +0000428 switch (Mode) {
429 case AccessMode::Exist:
430 return F_OK;
431 case AccessMode::Write:
432 return W_OK;
433 case AccessMode::Execute:
434 return R_OK | X_OK; // scripts also need R_OK.
435 }
436 llvm_unreachable("invalid enum");
437}
Michael J. Spencer45710402010-12-03 01:21:28 +0000438
Rafael Espindola281f23a2014-09-11 20:30:02 +0000439std::error_code access(const Twine &Path, AccessMode Mode) {
440 SmallString<128> PathStorage;
441 StringRef P = Path.toNullTerminatedStringRef(PathStorage);
442
443 if (::access(P.begin(), convertAccessMode(Mode)) == -1)
444 return std::error_code(errno, std::generic_category());
445
446 if (Mode == AccessMode::Execute) {
447 // Don't say that directories are executable.
448 struct stat buf;
449 if (0 != stat(P.begin(), &buf))
450 return errc::permission_denied;
451 if (!S_ISREG(buf.st_mode))
452 return errc::permission_denied;
453 }
Michael J. Spencer45710402010-12-03 01:21:28 +0000454
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000455 return std::error_code();
Michael J. Spencer45710402010-12-03 01:21:28 +0000456}
457
Reid Kleckner89d4b1a2015-09-10 23:28:06 +0000458bool can_execute(const Twine &Path) {
459 return !access(Path, AccessMode::Execute);
460}
461
Michael J. Spencer203d7802011-12-12 06:04:28 +0000462bool equivalent(file_status A, file_status B) {
463 assert(status_known(A) && status_known(B));
Sylvestre Ledru3099f4bd2012-04-23 16:37:23 +0000464 return A.fs_st_dev == B.fs_st_dev &&
465 A.fs_st_ino == B.fs_st_ino;
Michael J. Spencer203d7802011-12-12 06:04:28 +0000466}
467
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000468std::error_code equivalent(const Twine &A, const Twine &B, bool &result) {
Michael J. Spencer203d7802011-12-12 06:04:28 +0000469 file_status fsA, fsB;
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000470 if (std::error_code ec = status(A, fsA))
471 return ec;
472 if (std::error_code ec = status(B, fsB))
473 return ec;
Michael J. Spencer203d7802011-12-12 06:04:28 +0000474 result = equivalent(fsA, fsB);
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000475 return std::error_code();
Michael J. Spencer376d3872010-12-03 18:49:13 +0000476}
477
Duncan P. N. Exon Smith91d3cfe2016-04-05 20:45:04 +0000478static std::error_code fillStatus(int StatRet, const struct stat &Status,
479 file_status &Result) {
Rafael Espindola77021c92013-07-16 03:20:13 +0000480 if (StatRet != 0) {
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000481 std::error_code ec(errno, std::generic_category());
Rafael Espindola2a826e42014-06-13 17:20:48 +0000482 if (ec == errc::no_such_file_or_directory)
Rafael Espindola77021c92013-07-16 03:20:13 +0000483 Result = file_status(file_type::file_not_found);
Michael J. Spencerdb5576a2010-12-04 00:32:40 +0000484 else
Rafael Espindola77021c92013-07-16 03:20:13 +0000485 Result = file_status(file_type::status_error);
Michael J. Spencerdb5576a2010-12-04 00:32:40 +0000486 return ec;
487 }
488
Rafael Espindola9da91a02013-07-16 02:55:33 +0000489 file_type Type = file_type::type_unknown;
Michael J. Spencerdb5576a2010-12-04 00:32:40 +0000490
Rafael Espindola77021c92013-07-16 03:20:13 +0000491 if (S_ISDIR(Status.st_mode))
Rafael Espindola9da91a02013-07-16 02:55:33 +0000492 Type = file_type::directory_file;
Rafael Espindola77021c92013-07-16 03:20:13 +0000493 else if (S_ISREG(Status.st_mode))
Rafael Espindola9da91a02013-07-16 02:55:33 +0000494 Type = file_type::regular_file;
Rafael Espindola77021c92013-07-16 03:20:13 +0000495 else if (S_ISBLK(Status.st_mode))
Rafael Espindola9da91a02013-07-16 02:55:33 +0000496 Type = file_type::block_file;
Rafael Espindola77021c92013-07-16 03:20:13 +0000497 else if (S_ISCHR(Status.st_mode))
Rafael Espindola9da91a02013-07-16 02:55:33 +0000498 Type = file_type::character_file;
Rafael Espindola77021c92013-07-16 03:20:13 +0000499 else if (S_ISFIFO(Status.st_mode))
Rafael Espindola9da91a02013-07-16 02:55:33 +0000500 Type = file_type::fifo_file;
Rafael Espindola77021c92013-07-16 03:20:13 +0000501 else if (S_ISSOCK(Status.st_mode))
Rafael Espindola9da91a02013-07-16 02:55:33 +0000502 Type = file_type::socket_file;
Zachary Turner3b9fb882017-03-07 17:48:47 +0000503 else if (S_ISLNK(Status.st_mode))
504 Type = file_type::symlink_file;
Rafael Espindola9da91a02013-07-16 02:55:33 +0000505
Rafael Espindola77021c92013-07-16 03:20:13 +0000506 perms Perms = static_cast<perms>(Status.st_mode);
507 Result =
Mehdi Amini1e39ef32016-03-25 07:30:21 +0000508 file_status(Type, Perms, Status.st_dev, Status.st_ino, Status.st_atime,
509 Status.st_mtime, Status.st_uid, Status.st_gid,
510 Status.st_size);
Michael J. Spencer203d7802011-12-12 06:04:28 +0000511
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000512 return std::error_code();
Michael J. Spencerdb5576a2010-12-04 00:32:40 +0000513}
514
Zachary Turner82dd5422017-03-07 16:10:10 +0000515std::error_code status(const Twine &Path, file_status &Result, bool Follow) {
Rafael Espindola77021c92013-07-16 03:20:13 +0000516 SmallString<128> PathStorage;
517 StringRef P = Path.toNullTerminatedStringRef(PathStorage);
518
519 struct stat Status;
Zachary Turner82dd5422017-03-07 16:10:10 +0000520 int StatRet = (Follow ? ::stat : ::lstat)(P.begin(), &Status);
Rafael Espindola77021c92013-07-16 03:20:13 +0000521 return fillStatus(StatRet, Status, Result);
522}
523
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000524std::error_code status(int FD, file_status &Result) {
Rafael Espindola77021c92013-07-16 03:20:13 +0000525 struct stat Status;
526 int StatRet = ::fstat(FD, &Status);
527 return fillStatus(StatRet, Status, Result);
528}
529
Pavel Labath757ca882016-10-24 10:59:17 +0000530std::error_code setLastModificationAndAccessTime(int FD, TimePoint<> Time) {
Eric Christophera24dc7f2013-07-04 01:10:38 +0000531#if defined(HAVE_FUTIMENS)
532 timespec Times[2];
Pavel Labath757ca882016-10-24 10:59:17 +0000533 Times[0] = Times[1] = sys::toTimeSpec(Time);
Eric Christophera24dc7f2013-07-04 01:10:38 +0000534 if (::futimens(FD, Times))
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000535 return std::error_code(errno, std::generic_category());
536 return std::error_code();
Eric Christophera24dc7f2013-07-04 01:10:38 +0000537#elif defined(HAVE_FUTIMES)
Rafael Espindola4a3365c2013-06-20 20:56:14 +0000538 timeval Times[2];
Pavel Labath676a8752016-10-24 14:19:28 +0000539 Times[0] = Times[1] = sys::toTimeVal(
540 std::chrono::time_point_cast<std::chrono::microseconds>(Time));
Rafael Espindola4a3365c2013-06-20 20:56:14 +0000541 if (::futimes(FD, Times))
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000542 return std::error_code(errno, std::generic_category());
543 return std::error_code();
Alp Tokerb30f01e2013-12-11 15:42:33 +0000544#else
545#warning Missing futimes() and futimens()
Alp Tokerb792a012014-06-30 18:57:04 +0000546 return make_error_code(errc::function_not_supported);
Alp Tokerb30f01e2013-12-11 15:42:33 +0000547#endif
Rafael Espindola4a3365c2013-06-20 20:56:14 +0000548}
549
Rafael Espindola7eb1f182014-12-11 20:12:55 +0000550std::error_code mapped_file_region::init(int FD, uint64_t Offset,
551 mapmode Mode) {
Rafael Espindolac69f13b2014-12-12 18:13:23 +0000552 assert(Size != 0);
Michael J. Spenceref2284f2012-08-15 19:05:47 +0000553
554 int flags = (Mode == readwrite) ? MAP_SHARED : MAP_PRIVATE;
555 int prot = (Mode == readonly) ? PROT_READ : (PROT_READ | PROT_WRITE);
Zachary Turner842972b2017-02-22 21:24:06 +0000556#if defined(__APPLE__)
557 //----------------------------------------------------------------------
558 // Newer versions of MacOSX have a flag that will allow us to read from
559 // binaries whose code signature is invalid without crashing by using
560 // the MAP_RESILIENT_CODESIGN flag. Also if a file from removable media
561 // is mapped we can avoid crashing and return zeroes to any pages we try
562 // to read if the media becomes unavailable by using the
563 // MAP_RESILIENT_MEDIA flag. These flags are only usable when mapping
564 // with PROT_READ, so take care not to specify them otherwise.
565 //----------------------------------------------------------------------
566 if (Mode == readonly) {
567#if defined(MAP_RESILIENT_CODESIGN)
568 flags |= MAP_RESILIENT_CODESIGN;
569#endif
570#if defined(MAP_RESILIENT_MEDIA)
571 flags |= MAP_RESILIENT_MEDIA;
572#endif
573 }
574#endif // #if defined (__APPLE__)
Zachary Turner392ed9d2017-02-21 20:55:47 +0000575
Craig Toppere73658d2014-04-28 04:05:08 +0000576 Mapping = ::mmap(nullptr, Size, prot, flags, FD, Offset);
Michael J. Spenceref2284f2012-08-15 19:05:47 +0000577 if (Mapping == MAP_FAILED)
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000578 return std::error_code(errno, std::generic_category());
579 return std::error_code();
Michael J. Spenceref2284f2012-08-15 19:05:47 +0000580}
581
Rafael Espindola7eb1f182014-12-11 20:12:55 +0000582mapped_file_region::mapped_file_region(int fd, mapmode mode, uint64_t length,
583 uint64_t offset, std::error_code &ec)
584 : Size(length), Mapping() {
Michael J. Spenceref2284f2012-08-15 19:05:47 +0000585 // Make sure that the requested size fits within SIZE_T.
586 if (length > std::numeric_limits<size_t>::max()) {
Rafael Espindola2a826e42014-06-13 17:20:48 +0000587 ec = make_error_code(errc::invalid_argument);
Michael J. Spenceref2284f2012-08-15 19:05:47 +0000588 return;
589 }
590
Rafael Espindola7eb1f182014-12-11 20:12:55 +0000591 ec = init(fd, offset, mode);
Michael J. Spenceref2284f2012-08-15 19:05:47 +0000592 if (ec)
Craig Toppere73658d2014-04-28 04:05:08 +0000593 Mapping = nullptr;
Michael J. Spenceref2284f2012-08-15 19:05:47 +0000594}
595
596mapped_file_region::~mapped_file_region() {
597 if (Mapping)
598 ::munmap(Mapping, Size);
599}
600
Michael J. Spenceref2284f2012-08-15 19:05:47 +0000601uint64_t mapped_file_region::size() const {
602 assert(Mapping && "Mapping failed but used anyway!");
603 return Size;
604}
605
606char *mapped_file_region::data() const {
607 assert(Mapping && "Mapping failed but used anyway!");
Michael J. Spenceref2284f2012-08-15 19:05:47 +0000608 return reinterpret_cast<char*>(Mapping);
609}
610
611const char *mapped_file_region::const_data() const {
612 assert(Mapping && "Mapping failed but used anyway!");
613 return reinterpret_cast<const char*>(Mapping);
614}
615
616int mapped_file_region::alignment() {
Rafael Espindolac0610bf2014-12-04 16:59:36 +0000617 return Process::getPageSize();
Michael J. Spenceref2284f2012-08-15 19:05:47 +0000618}
619
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000620std::error_code detail::directory_iterator_construct(detail::DirIterState &it,
Michael J. Spencer0a7625d2011-12-08 22:50:09 +0000621 StringRef path){
Michael J. Spencer52714862011-01-05 16:38:57 +0000622 SmallString<128> path_null(path);
623 DIR *directory = ::opendir(path_null.c_str());
Craig Toppere73658d2014-04-28 04:05:08 +0000624 if (!directory)
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000625 return std::error_code(errno, std::generic_category());
Michael J. Spencer52714862011-01-05 16:38:57 +0000626
627 it.IterationHandle = reinterpret_cast<intptr_t>(directory);
628 // Add something for replace_filename to replace.
629 path::append(path_null, ".");
630 it.CurrentEntry = directory_entry(path_null.str());
631 return directory_iterator_increment(it);
632}
633
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000634std::error_code detail::directory_iterator_destruct(detail::DirIterState &it) {
Michael J. Spencer52714862011-01-05 16:38:57 +0000635 if (it.IterationHandle)
636 ::closedir(reinterpret_cast<DIR *>(it.IterationHandle));
637 it.IterationHandle = 0;
638 it.CurrentEntry = directory_entry();
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000639 return std::error_code();
Michael J. Spencer52714862011-01-05 16:38:57 +0000640}
641
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000642std::error_code detail::directory_iterator_increment(detail::DirIterState &it) {
Michael J. Spencer52714862011-01-05 16:38:57 +0000643 errno = 0;
644 dirent *cur_dir = ::readdir(reinterpret_cast<DIR *>(it.IterationHandle));
Craig Toppere73658d2014-04-28 04:05:08 +0000645 if (cur_dir == nullptr && errno != 0) {
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000646 return std::error_code(errno, std::generic_category());
Craig Toppere73658d2014-04-28 04:05:08 +0000647 } else if (cur_dir != nullptr) {
Michael J. Spencer52714862011-01-05 16:38:57 +0000648 StringRef name(cur_dir->d_name, NAMLEN(cur_dir));
649 if ((name.size() == 1 && name[0] == '.') ||
650 (name.size() == 2 && name[0] == '.' && name[1] == '.'))
651 return directory_iterator_increment(it);
652 it.CurrentEntry.replace_filename(name);
653 } else
654 return directory_iterator_destruct(it);
655
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000656 return std::error_code();
Michael J. Spencer52714862011-01-05 16:38:57 +0000657}
658
Taewook Ohd9153272016-06-13 15:54:56 +0000659#if !defined(F_GETPATH)
660static bool hasProcSelfFD() {
661 // If we have a /proc filesystem mounted, we can quickly establish the
662 // real name of the file with readlink
663 static const bool Result = (::access("/proc/self/fd", R_OK) == 0);
664 return Result;
665}
666#endif
667
668std::error_code openFileForRead(const Twine &Name, int &ResultFD,
669 SmallVectorImpl<char> *RealPath) {
Rafael Espindolaa0d9b6b2013-07-17 14:58:25 +0000670 SmallString<128> Storage;
671 StringRef P = Name.toNullTerminatedStringRef(Storage);
Pavel Labathf726dfa2017-01-24 10:57:01 +0000672 int OpenFlags = O_RDONLY;
673#ifdef O_CLOEXEC
674 OpenFlags |= O_CLOEXEC;
675#endif
676 while ((ResultFD = open(P.begin(), OpenFlags)) < 0) {
Rafael Espindolaa0d9b6b2013-07-17 14:58:25 +0000677 if (errno != EINTR)
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000678 return std::error_code(errno, std::generic_category());
Rafael Espindolaa0d9b6b2013-07-17 14:58:25 +0000679 }
Pavel Labathf726dfa2017-01-24 10:57:01 +0000680#ifndef O_CLOEXEC
681 int r = fcntl(ResultFD, F_SETFD, FD_CLOEXEC);
682 (void)r;
683 assert(r == 0 && "fcntl(F_SETFD, FD_CLOEXEC) failed");
684#endif
Taewook Ohd9153272016-06-13 15:54:56 +0000685 // Attempt to get the real name of the file, if the user asked
686 if(!RealPath)
687 return std::error_code();
688 RealPath->clear();
689#if defined(F_GETPATH)
690 // When F_GETPATH is availble, it is the quickest way to get
691 // the real path name.
692 char Buffer[MAXPATHLEN];
693 if (::fcntl(ResultFD, F_GETPATH, Buffer) != -1)
694 RealPath->append(Buffer, Buffer + strlen(Buffer));
695#else
696 char Buffer[PATH_MAX];
697 if (hasProcSelfFD()) {
698 char ProcPath[64];
699 snprintf(ProcPath, sizeof(ProcPath), "/proc/self/fd/%d", ResultFD);
700 ssize_t CharCount = ::readlink(ProcPath, Buffer, sizeof(Buffer));
701 if (CharCount > 0)
702 RealPath->append(Buffer, Buffer + CharCount);
703 } else {
704 // Use ::realpath to get the real path name
705 if (::realpath(P.begin(), Buffer) != nullptr)
706 RealPath->append(Buffer, Buffer + strlen(Buffer));
707 }
708#endif
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000709 return std::error_code();
Rafael Espindolaa0d9b6b2013-07-17 14:58:25 +0000710}
Nick Kledzik18497e92012-06-20 00:28:54 +0000711
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000712std::error_code openFileForWrite(const Twine &Name, int &ResultFD,
Rafael Espindola67080ce2013-07-19 15:02:03 +0000713 sys::fs::OpenFlags Flags, unsigned Mode) {
714 // Verify that we don't have both "append" and "excl".
715 assert((!(Flags & sys::fs::F_Excl) || !(Flags & sys::fs::F_Append)) &&
716 "Cannot specify both 'excl' and 'append' file creation flags!");
717
Pavel Labathf726dfa2017-01-24 10:57:01 +0000718 int OpenFlags = O_CREAT;
719
720#ifdef O_CLOEXEC
721 OpenFlags |= O_CLOEXEC;
722#endif
Rafael Espindola7a0b6402014-02-24 03:07:41 +0000723
724 if (Flags & F_RW)
725 OpenFlags |= O_RDWR;
726 else
727 OpenFlags |= O_WRONLY;
Rafael Espindola67080ce2013-07-19 15:02:03 +0000728
729 if (Flags & F_Append)
730 OpenFlags |= O_APPEND;
731 else
732 OpenFlags |= O_TRUNC;
733
734 if (Flags & F_Excl)
735 OpenFlags |= O_EXCL;
736
737 SmallString<128> Storage;
738 StringRef P = Name.toNullTerminatedStringRef(Storage);
739 while ((ResultFD = open(P.begin(), OpenFlags, Mode)) < 0) {
740 if (errno != EINTR)
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000741 return std::error_code(errno, std::generic_category());
Rafael Espindola67080ce2013-07-19 15:02:03 +0000742 }
Pavel Labathf726dfa2017-01-24 10:57:01 +0000743#ifndef O_CLOEXEC
744 int r = fcntl(ResultFD, F_SETFD, FD_CLOEXEC);
745 (void)r;
746 assert(r == 0 && "fcntl(F_SETFD, FD_CLOEXEC) failed");
747#endif
Rafael Espindoladb4ed0b2014-06-13 02:24:39 +0000748 return std::error_code();
Rafael Espindola67080ce2013-07-19 15:02:03 +0000749}
750
Taewook Ohd9153272016-06-13 15:54:56 +0000751std::error_code getPathFromOpenFD(int FD, SmallVectorImpl<char> &ResultPath) {
752 if (FD < 0)
753 return make_error_code(errc::bad_file_descriptor);
754
755#if defined(F_GETPATH)
756 // When F_GETPATH is availble, it is the quickest way to get
757 // the path from a file descriptor.
758 ResultPath.reserve(MAXPATHLEN);
759 if (::fcntl(FD, F_GETPATH, ResultPath.begin()) == -1)
760 return std::error_code(errno, std::generic_category());
761
762 ResultPath.set_size(strlen(ResultPath.begin()));
763#else
764 // If we have a /proc filesystem mounted, we can quickly establish the
765 // real name of the file with readlink. Otherwise, we don't know how to
766 // get the filename from a file descriptor. Give up.
767 if (!fs::hasProcSelfFD())
768 return make_error_code(errc::function_not_supported);
769
770 ResultPath.reserve(PATH_MAX);
771 char ProcPath[64];
772 snprintf(ProcPath, sizeof(ProcPath), "/proc/self/fd/%d", FD);
773 ssize_t CharCount = ::readlink(ProcPath, ResultPath.begin(), ResultPath.capacity());
774 if (CharCount < 0)
775 return std::error_code(errno, std::generic_category());
776
777 // Was the filename truncated?
778 if (static_cast<size_t>(CharCount) == ResultPath.capacity()) {
779 // Use lstat to get the size of the filename
780 struct stat sb;
781 if (::lstat(ProcPath, &sb) < 0)
782 return std::error_code(errno, std::generic_category());
783
784 ResultPath.reserve(sb.st_size + 1);
785 CharCount = ::readlink(ProcPath, ResultPath.begin(), ResultPath.capacity());
786 if (CharCount < 0)
787 return std::error_code(errno, std::generic_category());
788
789 // Test for race condition: did the link size change?
790 if (CharCount > sb.st_size)
791 return std::error_code(ENAMETOOLONG, std::generic_category());
792 }
793 ResultPath.set_size(static_cast<size_t>(CharCount));
794#endif
795 return std::error_code();
796}
797
Michael J. Spencer9fc1d9d2010-12-01 19:32:01 +0000798} // end namespace fs
Peter Collingbournef7d41012014-01-31 23:46:06 +0000799
800namespace path {
801
802bool home_directory(SmallVectorImpl<char> &result) {
803 if (char *RequestedDir = getenv("HOME")) {
804 result.clear();
805 result.append(RequestedDir, RequestedDir + strlen(RequestedDir));
806 return true;
807 }
808
809 return false;
810}
811
Duncan P. N. Exon Smith91d3cfe2016-04-05 20:45:04 +0000812static bool getDarwinConfDir(bool TempDir, SmallVectorImpl<char> &Result) {
Pawel Bylica0e97e5c2015-11-02 09:49:17 +0000813 #if defined(_CS_DARWIN_USER_TEMP_DIR) && defined(_CS_DARWIN_USER_CACHE_DIR)
814 // On Darwin, use DARWIN_USER_TEMP_DIR or DARWIN_USER_CACHE_DIR.
815 // macros defined in <unistd.h> on darwin >= 9
816 int ConfName = TempDir ? _CS_DARWIN_USER_TEMP_DIR
817 : _CS_DARWIN_USER_CACHE_DIR;
818 size_t ConfLen = confstr(ConfName, nullptr, 0);
819 if (ConfLen > 0) {
820 do {
821 Result.resize(ConfLen);
822 ConfLen = confstr(ConfName, Result.data(), Result.size());
823 } while (ConfLen > 0 && ConfLen != Result.size());
824
825 if (ConfLen > 0) {
826 assert(Result.back() == 0);
827 Result.pop_back();
828 return true;
829 }
830
831 Result.clear();
832 }
833 #endif
834 return false;
835}
836
Duncan P. N. Exon Smith91d3cfe2016-04-05 20:45:04 +0000837static bool getUserCacheDir(SmallVectorImpl<char> &Result) {
Sean Silvaa915a162016-03-24 22:27:27 +0000838 // First try using XDG_CACHE_HOME env variable,
Pawel Bylica0e97e5c2015-11-02 09:49:17 +0000839 // as specified in XDG Base Directory Specification at
840 // http://standards.freedesktop.org/basedir-spec/basedir-spec-latest.html
Sean Silvaa915a162016-03-24 22:27:27 +0000841 if (const char *XdgCacheDir = std::getenv("XDG_CACHE_HOME")) {
Pawel Bylica0e97e5c2015-11-02 09:49:17 +0000842 Result.clear();
Sean Silvaa915a162016-03-24 22:27:27 +0000843 Result.append(XdgCacheDir, XdgCacheDir + strlen(XdgCacheDir));
Pawel Bylica0e97e5c2015-11-02 09:49:17 +0000844 return true;
845 }
846
847 // Try Darwin configuration query
848 if (getDarwinConfDir(false, Result))
849 return true;
850
851 // Use "$HOME/.cache" if $HOME is available
852 if (home_directory(Result)) {
853 append(Result, ".cache");
854 return true;
855 }
856
857 return false;
858}
Pawel Bylica0e97e5c2015-11-02 09:49:17 +0000859
Duncan P. N. Exon Smith91d3cfe2016-04-05 20:45:04 +0000860static const char *getEnvTempDir() {
Rafael Espindola016a6d52014-08-26 14:47:52 +0000861 // Check whether the temporary directory is specified by an environment
862 // variable.
863 const char *EnvironmentVariables[] = {"TMPDIR", "TMP", "TEMP", "TEMPDIR"};
864 for (const char *Env : EnvironmentVariables) {
865 if (const char *Dir = std::getenv(Env))
866 return Dir;
867 }
868
869 return nullptr;
870}
871
Duncan P. N. Exon Smith91d3cfe2016-04-05 20:45:04 +0000872static const char *getDefaultTempDir(bool ErasedOnReboot) {
Rafael Espindola016a6d52014-08-26 14:47:52 +0000873#ifdef P_tmpdir
Benjamin Kramer870d9512014-08-27 11:47:52 +0000874 if ((bool)P_tmpdir)
Rafael Espindola016a6d52014-08-26 14:47:52 +0000875 return P_tmpdir;
876#endif
877
878 if (ErasedOnReboot)
879 return "/tmp";
880 return "/var/tmp";
881}
882
883void system_temp_directory(bool ErasedOnReboot, SmallVectorImpl<char> &Result) {
884 Result.clear();
885
886 if (ErasedOnReboot) {
887 // There is no env variable for the cache directory.
888 if (const char *RequestedDir = getEnvTempDir()) {
889 Result.append(RequestedDir, RequestedDir + strlen(RequestedDir));
890 return;
891 }
892 }
893
Pawel Bylica0e97e5c2015-11-02 09:49:17 +0000894 if (getDarwinConfDir(ErasedOnReboot, Result))
895 return;
Rafael Espindola016a6d52014-08-26 14:47:52 +0000896
897 const char *RequestedDir = getDefaultTempDir(ErasedOnReboot);
898 Result.append(RequestedDir, RequestedDir + strlen(RequestedDir));
899}
900
Peter Collingbournef7d41012014-01-31 23:46:06 +0000901} // end namespace path
902
Michael J. Spencerebad2f92010-11-29 22:28:51 +0000903} // end namespace sys
904} // end namespace llvm