xref: /freebsd/contrib/llvm-project/libcxx/src/filesystem/operations.cpp (revision 700637cbb5e582861067a11aaca4d053546871d2)
1 //===----------------------------------------------------------------------===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8 
9 #include <__algorithm/copy.h>
10 #include <__assert>
11 #include <__config>
12 #include <__utility/unreachable.h>
13 #include <array>
14 #include <climits>
15 #include <cstdlib>
16 #include <filesystem>
17 #include <iterator>
18 #include <string_view>
19 #include <system_error>
20 #include <type_traits>
21 #include <vector>
22 
23 #include "error.h"
24 #include "file_descriptor.h"
25 #include "path_parser.h"
26 #include "posix_compat.h"
27 #include "time_utils.h"
28 
29 #if defined(_LIBCPP_WIN32API)
30 #  define WIN32_LEAN_AND_MEAN
31 #  define NOMINMAX
32 #  include <windows.h>
33 #else
34 #  include <dirent.h>
35 #  include <sys/stat.h>
36 #  include <sys/statvfs.h>
37 #  include <sys/types.h>
38 #  include <unistd.h>
39 #endif
40 #include <fcntl.h> /* values for fchmodat */
41 #include <time.h>
42 
43 // since Linux 4.5 and FreeBSD 13, but the Linux libc wrapper is only provided by glibc >= 2.27 and musl
44 #if defined(__linux__)
45 #  if defined(_LIBCPP_GLIBC_PREREQ)
46 #    if _LIBCPP_GLIBC_PREREQ(2, 27)
47 #      define _LIBCPP_FILESYSTEM_USE_COPY_FILE_RANGE
48 #    endif
49 #  elif _LIBCPP_HAS_MUSL_LIBC
50 #    define _LIBCPP_FILESYSTEM_USE_COPY_FILE_RANGE
51 #  endif
52 #elif defined(__FreeBSD__)
53 #  define _LIBCPP_FILESYSTEM_USE_COPY_FILE_RANGE
54 #endif
55 #if __has_include(<sys/sendfile.h>)
56 #  include <sys/sendfile.h>
57 #  define _LIBCPP_FILESYSTEM_USE_SENDFILE
58 #elif defined(__APPLE__) || __has_include(<copyfile.h>)
59 #  include <copyfile.h>
60 #  define _LIBCPP_FILESYSTEM_USE_COPYFILE
61 #else
62 #  define _LIBCPP_FILESYSTEM_USE_FSTREAM
63 #endif
64 
65 // sendfile and copy_file_range need to fall back
66 // to the fstream implementation for special files
67 #if (defined(_LIBCPP_FILESYSTEM_USE_SENDFILE) || defined(_LIBCPP_FILESYSTEM_USE_COPY_FILE_RANGE) ||                    \
68      defined(_LIBCPP_FILESYSTEM_USE_FSTREAM)) &&                                                                       \
69     _LIBCPP_HAS_LOCALIZATION
70 #  include <fstream>
71 #  define _LIBCPP_FILESYSTEM_NEED_FSTREAM
72 #endif
73 
74 #if defined(__ELF__) && defined(_LIBCPP_LINK_RT_LIB)
75 #  pragma comment(lib, "rt")
76 #endif
77 
78 _LIBCPP_BEGIN_NAMESPACE_FILESYSTEM
79 
80 using detail::capture_errno;
81 using detail::ErrorHandler;
82 using detail::StatT;
83 using detail::TimeSpec;
84 using parser::createView;
85 using parser::PathParser;
86 using parser::string_view_t;
87 
__do_absolute(const path & p,path * cwd,error_code * ec)88 static path __do_absolute(const path& p, path* cwd, error_code* ec) {
89   if (ec)
90     ec->clear();
91   if (p.is_absolute())
92     return p;
93   *cwd = __current_path(ec);
94   if (ec && *ec)
95     return {};
96   return (*cwd) / p;
97 }
98 
__absolute(const path & p,error_code * ec)99 path __absolute(const path& p, error_code* ec) {
100   path cwd;
101   return __do_absolute(p, &cwd, ec);
102 }
103 
__canonical(path const & orig_p,error_code * ec)104 path __canonical(path const& orig_p, error_code* ec) {
105   path cwd;
106   ErrorHandler<path> err("canonical", ec, &orig_p, &cwd);
107 
108   path p = __do_absolute(orig_p, &cwd, ec);
109 #if (defined(_POSIX_VERSION) && _POSIX_VERSION >= 200112) || defined(_LIBCPP_WIN32API)
110   std::unique_ptr<path::value_type, decltype(&::free)> hold(detail::realpath(p.c_str(), nullptr), &::free);
111   if (hold.get() == nullptr)
112     return err.report(detail::get_last_error());
113   return {hold.get()};
114 #else
115 #  if defined(__MVS__) && !defined(PATH_MAX)
116   path::value_type buff[_XOPEN_PATH_MAX + 1];
117 #  else
118   path::value_type buff[PATH_MAX + 1];
119 #  endif
120   path::value_type* ret;
121   if ((ret = detail::realpath(p.c_str(), buff)) == nullptr)
122     return err.report(detail::get_last_error());
123   return {ret};
124 #endif
125 }
126 
__copy(const path & from,const path & to,copy_options options,error_code * ec)127 void __copy(const path& from, const path& to, copy_options options, error_code* ec) {
128   ErrorHandler<void> err("copy", ec, &from, &to);
129 
130   const bool sym_status = bool(options & (copy_options::create_symlinks | copy_options::skip_symlinks));
131 
132   const bool sym_status2 = bool(options & copy_options::copy_symlinks);
133 
134   error_code m_ec1;
135   StatT f_st;
136   const file_status f =
137       sym_status || sym_status2 ? detail::posix_lstat(from, f_st, &m_ec1) : detail::posix_stat(from, f_st, &m_ec1);
138   if (m_ec1)
139     return err.report(m_ec1);
140 
141   StatT t_st;
142   const file_status t = sym_status ? detail::posix_lstat(to, t_st, &m_ec1) : detail::posix_stat(to, t_st, &m_ec1);
143 
144   if (not status_known(t))
145     return err.report(m_ec1);
146 
147   if (!exists(f) || is_other(f) || is_other(t) || (is_directory(f) && is_regular_file(t)) ||
148       (exists(t) && detail::stat_equivalent(f_st, t_st))) {
149     return err.report(errc::function_not_supported);
150   }
151 
152   if (is_symlink(f)) {
153     if (bool(copy_options::skip_symlinks & options)) {
154       // do nothing
155     } else if (not exists(t)) {
156       __copy_symlink(from, to, ec);
157     } else {
158       return err.report(errc::file_exists);
159     }
160     return;
161   } else if (is_regular_file(f)) {
162     if (bool(copy_options::directories_only & options)) {
163       // do nothing
164     } else if (bool(copy_options::create_symlinks & options)) {
165       __create_symlink(from, to, ec);
166     } else if (bool(copy_options::create_hard_links & options)) {
167       __create_hard_link(from, to, ec);
168     } else if (is_directory(t)) {
169       __copy_file(from, to / from.filename(), options, ec);
170     } else {
171       __copy_file(from, to, options, ec);
172     }
173     return;
174   } else if (is_directory(f) && bool(copy_options::create_symlinks & options)) {
175     return err.report(errc::is_a_directory);
176   } else if (is_directory(f) && (bool(copy_options::recursive & options) || copy_options::none == options)) {
177     if (!exists(t)) {
178       // create directory to with attributes from 'from'.
179       __create_directory(to, from, ec);
180       if (ec && *ec) {
181         return;
182       }
183     }
184     directory_iterator it = ec ? directory_iterator(from, *ec) : directory_iterator(from);
185     if (ec && *ec) {
186       return;
187     }
188     error_code m_ec2;
189     for (; !m_ec2 && it != directory_iterator(); it.increment(m_ec2)) {
190       __copy(it->path(), to / it->path().filename(), options | copy_options::__in_recursive_copy, ec);
191       if (ec && *ec) {
192         return;
193       }
194     }
195     if (m_ec2) {
196       return err.report(m_ec2);
197     }
198   }
199 }
200 
201 namespace detail {
202 namespace {
203 
204 #if defined(_LIBCPP_FILESYSTEM_NEED_FSTREAM)
copy_file_impl_fstream(FileDescriptor & read_fd,FileDescriptor & write_fd,error_code & ec)205 bool copy_file_impl_fstream(FileDescriptor& read_fd, FileDescriptor& write_fd, error_code& ec) {
206   ifstream in;
207   in.__open(read_fd.fd, ios::binary);
208   if (!in.is_open()) {
209     // This assumes that __open didn't reset the error code.
210     ec = capture_errno();
211     return false;
212   }
213   read_fd.fd = -1;
214   ofstream out;
215   out.__open(write_fd.fd, ios::binary);
216   if (!out.is_open()) {
217     ec = capture_errno();
218     return false;
219   }
220   write_fd.fd = -1;
221 
222   if (in.good() && out.good()) {
223     using InIt  = istreambuf_iterator<char>;
224     using OutIt = ostreambuf_iterator<char>;
225     InIt bin(in);
226     InIt ein;
227     OutIt bout(out);
228     copy(bin, ein, bout);
229   }
230   if (out.fail() || in.fail()) {
231     ec = make_error_code(errc::io_error);
232     return false;
233   }
234 
235   ec.clear();
236   return true;
237 }
238 #endif
239 
240 #if defined(_LIBCPP_FILESYSTEM_USE_COPY_FILE_RANGE)
copy_file_impl_copy_file_range(FileDescriptor & read_fd,FileDescriptor & write_fd,error_code & ec)241 bool copy_file_impl_copy_file_range(FileDescriptor& read_fd, FileDescriptor& write_fd, error_code& ec) {
242   size_t count = read_fd.get_stat().st_size;
243   // a zero-length file is either empty, or not copyable by this syscall
244   // return early to avoid the syscall cost
245   if (count == 0) {
246     ec = {EINVAL, generic_category()};
247     return false;
248   }
249   // do not modify the fd positions as copy_file_impl_sendfile may be called after a partial copy
250 #  if defined(__linux__)
251   loff_t off_in  = 0;
252   loff_t off_out = 0;
253 #  else
254   off_t off_in  = 0;
255   off_t off_out = 0;
256 #  endif
257 
258   do {
259     ssize_t res;
260 
261     if ((res = ::copy_file_range(read_fd.fd, &off_in, write_fd.fd, &off_out, count, 0)) == -1) {
262       ec = capture_errno();
263       return false;
264     }
265     count -= res;
266   } while (count > 0);
267 
268   ec.clear();
269 
270   return true;
271 }
272 #endif
273 
274 #if defined(_LIBCPP_FILESYSTEM_USE_SENDFILE)
copy_file_impl_sendfile(FileDescriptor & read_fd,FileDescriptor & write_fd,error_code & ec)275 bool copy_file_impl_sendfile(FileDescriptor& read_fd, FileDescriptor& write_fd, error_code& ec) {
276   size_t count = read_fd.get_stat().st_size;
277   // a zero-length file is either empty, or not copyable by this syscall
278   // return early to avoid the syscall cost
279   // however, we can't afford this luxury in the no-locale build,
280   // as we can't utilize the fstream impl to copy empty files
281 #  if _LIBCPP_HAS_LOCALIZATION
282   if (count == 0) {
283     ec = {EINVAL, generic_category()};
284     return false;
285   }
286 #  endif
287   do {
288     ssize_t res;
289     if ((res = ::sendfile(write_fd.fd, read_fd.fd, nullptr, count)) == -1) {
290       ec = capture_errno();
291       return false;
292     }
293     count -= res;
294   } while (count > 0);
295 
296   ec.clear();
297 
298   return true;
299 }
300 #endif
301 
302 #if defined(_LIBCPP_FILESYSTEM_USE_COPY_FILE_RANGE) || defined(_LIBCPP_FILESYSTEM_USE_SENDFILE)
303 // If we have copy_file_range or sendfile, try both in succession (if available).
304 // If both fail, fall back to using fstream.
copy_file_impl(FileDescriptor & read_fd,FileDescriptor & write_fd,error_code & ec)305 bool copy_file_impl(FileDescriptor& read_fd, FileDescriptor& write_fd, error_code& ec) {
306 #  if defined(_LIBCPP_FILESYSTEM_USE_COPY_FILE_RANGE)
307   if (copy_file_impl_copy_file_range(read_fd, write_fd, ec)) {
308     return true;
309   }
310   // EINVAL: src and dst are the same file (this is not cheaply
311   // detectable from userspace)
312   // EINVAL: copy_file_range is unsupported for this file type by the
313   // underlying filesystem
314   // ENOTSUP: undocumented, can arise with old kernels and NFS
315   // EOPNOTSUPP: filesystem does not implement copy_file_range
316   // ETXTBSY: src or dst is an active swapfile (nonsensical, but allowed
317   // with normal copying)
318   // EXDEV: src and dst are on different filesystems that do not support
319   // cross-fs copy_file_range
320   // ENOENT: undocumented, can arise with CIFS
321   // ENOSYS: unsupported by kernel or blocked by seccomp
322   if (ec.value() != EINVAL && ec.value() != ENOTSUP && ec.value() != EOPNOTSUPP && ec.value() != ETXTBSY &&
323       ec.value() != EXDEV && ec.value() != ENOENT && ec.value() != ENOSYS) {
324     return false;
325   }
326   ec.clear();
327 #  endif
328 
329 #  if defined(_LIBCPP_FILESYSTEM_USE_SENDFILE)
330   if (copy_file_impl_sendfile(read_fd, write_fd, ec)) {
331     return true;
332   }
333   // EINVAL: unsupported file type
334   if (ec.value() != EINVAL) {
335     return false;
336   }
337   ec.clear();
338 #  endif
339 
340 #  if defined(_LIBCPP_FILESYSTEM_NEED_FSTREAM)
341   return copy_file_impl_fstream(read_fd, write_fd, ec);
342 #  else
343   // since iostreams are unavailable in the no-locale build, just fail after a failed sendfile
344   ec.assign(EINVAL, std::system_category());
345   return false;
346 #  endif
347 }
348 #elif defined(_LIBCPP_FILESYSTEM_USE_COPYFILE)
copy_file_impl(FileDescriptor & read_fd,FileDescriptor & write_fd,error_code & ec)349 bool copy_file_impl(FileDescriptor& read_fd, FileDescriptor& write_fd, error_code& ec) {
350   struct CopyFileState {
351     copyfile_state_t state;
352     CopyFileState() { state = copyfile_state_alloc(); }
353     ~CopyFileState() { copyfile_state_free(state); }
354 
355   private:
356     CopyFileState(CopyFileState const&)            = delete;
357     CopyFileState& operator=(CopyFileState const&) = delete;
358   };
359 
360   CopyFileState cfs;
361   if (fcopyfile(read_fd.fd, write_fd.fd, cfs.state, COPYFILE_DATA) < 0) {
362     ec = capture_errno();
363     return false;
364   }
365 
366   ec.clear();
367   return true;
368 }
369 #elif defined(_LIBCPP_FILESYSTEM_USE_FSTREAM)
copy_file_impl(FileDescriptor & read_fd,FileDescriptor & write_fd,error_code & ec)370 bool copy_file_impl(FileDescriptor& read_fd, FileDescriptor& write_fd, error_code& ec) {
371   return copy_file_impl_fstream(read_fd, write_fd, ec);
372 }
373 #else
374 #  error "Unknown implementation for copy_file_impl"
375 #endif // copy_file_impl implementation
376 
377 } // end anonymous namespace
378 } // namespace detail
379 
__copy_file(const path & from,const path & to,copy_options options,error_code * ec)380 bool __copy_file(const path& from, const path& to, copy_options options, error_code* ec) {
381   using detail::FileDescriptor;
382   ErrorHandler<bool> err("copy_file", ec, &to, &from);
383 
384   error_code m_ec;
385   FileDescriptor from_fd = FileDescriptor::create_with_status(&from, m_ec, O_RDONLY | O_NONBLOCK | O_BINARY);
386   if (m_ec)
387     return err.report(m_ec);
388 
389   auto from_st           = from_fd.get_status();
390   StatT const& from_stat = from_fd.get_stat();
391   if (!is_regular_file(from_st)) {
392     if (not m_ec)
393       m_ec = make_error_code(errc::not_supported);
394     return err.report(m_ec);
395   }
396 
397   const bool skip_existing      = bool(copy_options::skip_existing & options);
398   const bool update_existing    = bool(copy_options::update_existing & options);
399   const bool overwrite_existing = bool(copy_options::overwrite_existing & options);
400 
401   StatT to_stat_path;
402   file_status to_st = detail::posix_stat(to, to_stat_path, &m_ec);
403   if (!status_known(to_st))
404     return err.report(m_ec);
405 
406   const bool to_exists = exists(to_st);
407   if (to_exists && !is_regular_file(to_st))
408     return err.report(errc::not_supported);
409 
410   if (to_exists && detail::stat_equivalent(from_stat, to_stat_path))
411     return err.report(errc::file_exists);
412 
413   if (to_exists && skip_existing)
414     return false;
415 
416   bool ShouldCopy = [&]() {
417     if (to_exists && update_existing) {
418       auto from_time = detail::extract_mtime(from_stat);
419       auto to_time   = detail::extract_mtime(to_stat_path);
420       if (from_time.tv_sec < to_time.tv_sec)
421         return false;
422       if (from_time.tv_sec == to_time.tv_sec && from_time.tv_nsec <= to_time.tv_nsec)
423         return false;
424       return true;
425     }
426     if (!to_exists || overwrite_existing)
427       return true;
428     return err.report(errc::file_exists);
429   }();
430   if (!ShouldCopy)
431     return false;
432 
433   // Don't truncate right away. We may not be opening the file we originally
434   // looked at; we'll check this later.
435   int to_open_flags = O_WRONLY | O_BINARY;
436   if (!to_exists)
437     to_open_flags |= O_CREAT;
438   FileDescriptor to_fd = FileDescriptor::create_with_status(&to, m_ec, to_open_flags, from_stat.st_mode);
439   if (m_ec)
440     return err.report(m_ec);
441 
442   if (to_exists) {
443     // Check that the file we initially stat'ed is equivalent to the one
444     // we opened.
445     // FIXME: report this better.
446     if (!detail::stat_equivalent(to_stat_path, to_fd.get_stat()))
447       return err.report(errc::bad_file_descriptor);
448 
449     // Set the permissions and truncate the file we opened.
450     if (detail::posix_fchmod(to_fd, from_stat, m_ec))
451       return err.report(m_ec);
452     if (detail::posix_ftruncate(to_fd, 0, m_ec))
453       return err.report(m_ec);
454   }
455 
456   if (!detail::copy_file_impl(from_fd, to_fd, m_ec)) {
457     // FIXME: Remove the dest file if we failed, and it didn't exist previously.
458     return err.report(m_ec);
459   }
460 
461   return true;
462 }
463 
__copy_symlink(const path & existing_symlink,const path & new_symlink,error_code * ec)464 void __copy_symlink(const path& existing_symlink, const path& new_symlink, error_code* ec) {
465   const path real_path(__read_symlink(existing_symlink, ec));
466   if (ec && *ec) {
467     return;
468   }
469 #if defined(_LIBCPP_WIN32API)
470   error_code local_ec;
471   if (is_directory(real_path, local_ec))
472     __create_directory_symlink(real_path, new_symlink, ec);
473   else
474 #endif
475     __create_symlink(real_path, new_symlink, ec);
476 }
477 
__create_directories(const path & p,error_code * ec)478 bool __create_directories(const path& p, error_code* ec) {
479   ErrorHandler<bool> err("create_directories", ec, &p);
480 
481   error_code m_ec;
482   auto const st = detail::posix_stat(p, &m_ec);
483   if (!status_known(st))
484     return err.report(m_ec);
485   else if (is_directory(st))
486     return false;
487   else if (exists(st))
488     return err.report(errc::file_exists);
489 
490   const path parent = p.parent_path();
491   if (!parent.empty()) {
492     const file_status parent_st = status(parent, m_ec);
493     if (not status_known(parent_st))
494       return err.report(m_ec);
495     if (not exists(parent_st)) {
496       if (parent == p)
497         return err.report(errc::invalid_argument);
498       __create_directories(parent, ec);
499       if (ec && *ec) {
500         return false;
501       }
502     } else if (not is_directory(parent_st))
503       return err.report(errc::not_a_directory);
504   }
505   bool ret = __create_directory(p, &m_ec);
506   if (m_ec)
507     return err.report(m_ec);
508   return ret;
509 }
510 
__create_directory(const path & p,error_code * ec)511 bool __create_directory(const path& p, error_code* ec) {
512   ErrorHandler<bool> err("create_directory", ec, &p);
513 
514   if (detail::mkdir(p.c_str(), static_cast<int>(perms::all)) == 0)
515     return true;
516 
517   error_code mec = detail::get_last_error();
518   if (mec != errc::file_exists)
519     return err.report(mec);
520   error_code ignored_ec;
521   const file_status st = status(p, ignored_ec);
522   if (!is_directory(st))
523     return err.report(mec);
524   return false;
525 }
526 
__create_directory(path const & p,path const & attributes,error_code * ec)527 bool __create_directory(path const& p, path const& attributes, error_code* ec) {
528   ErrorHandler<bool> err("create_directory", ec, &p, &attributes);
529 
530   StatT attr_stat;
531   error_code mec;
532   file_status st = detail::posix_stat(attributes, attr_stat, &mec);
533   if (!status_known(st))
534     return err.report(mec);
535   if (!is_directory(st))
536     return err.report(errc::not_a_directory, "the specified attribute path is invalid");
537 
538   if (detail::mkdir(p.c_str(), attr_stat.st_mode) == 0)
539     return true;
540 
541   mec = detail::get_last_error();
542   if (mec != errc::file_exists)
543     return err.report(mec);
544 
545   error_code ignored_ec;
546   st = status(p, ignored_ec);
547   if (!is_directory(st))
548     return err.report(mec);
549   return false;
550 }
551 
__create_directory_symlink(path const & from,path const & to,error_code * ec)552 void __create_directory_symlink(path const& from, path const& to, error_code* ec) {
553   ErrorHandler<void> err("create_directory_symlink", ec, &from, &to);
554   if (detail::symlink_dir(from.c_str(), to.c_str()) == -1)
555     return err.report(detail::get_last_error());
556 }
557 
__create_hard_link(const path & from,const path & to,error_code * ec)558 void __create_hard_link(const path& from, const path& to, error_code* ec) {
559   ErrorHandler<void> err("create_hard_link", ec, &from, &to);
560   if (detail::link(from.c_str(), to.c_str()) == -1)
561     return err.report(detail::get_last_error());
562 }
563 
__create_symlink(path const & from,path const & to,error_code * ec)564 void __create_symlink(path const& from, path const& to, error_code* ec) {
565   ErrorHandler<void> err("create_symlink", ec, &from, &to);
566   if (detail::symlink_file(from.c_str(), to.c_str()) == -1)
567     return err.report(detail::get_last_error());
568 }
569 
__current_path(error_code * ec)570 path __current_path(error_code* ec) {
571   ErrorHandler<path> err("current_path", ec);
572 
573 #if defined(_LIBCPP_WIN32API) || defined(__GLIBC__) || defined(__APPLE__)
574   // Common extension outside of POSIX getcwd() spec, without needing to
575   // preallocate a buffer. Also supported by a number of other POSIX libcs.
576   int size              = 0;
577   path::value_type* ptr = nullptr;
578   typedef decltype(&::free) Deleter;
579   Deleter deleter = &::free;
580 #else
581   errno     = 0; // Note: POSIX mandates that modifying `errno` is thread-safe.
582   auto size = ::pathconf(".", _PC_PATH_MAX);
583   if (size == -1) {
584     if (errno != 0) {
585       return err.report(capture_errno(), "call to pathconf failed");
586 
587       // `pathconf` returns `-1` without an error to indicate no limit.
588     } else {
589 #  if defined(__MVS__) && !defined(PATH_MAX)
590       size = _XOPEN_PATH_MAX + 1;
591 #  else
592       size = PATH_MAX + 1;
593 #  endif
594     }
595   }
596 
597   auto buff             = unique_ptr<path::value_type[]>(new path::value_type[size + 1]);
598   path::value_type* ptr = buff.get();
599 
600   // Preallocated buffer, don't free the buffer in the second unique_ptr
601   // below.
602   struct Deleter {
603     void operator()(void*) const {}
604   };
605   Deleter deleter;
606 #endif
607 
608   unique_ptr<path::value_type, Deleter> hold(detail::getcwd(ptr, size), deleter);
609   if (hold.get() == nullptr)
610     return err.report(detail::get_last_error(), "call to getcwd failed");
611 
612   return {hold.get()};
613 }
614 
__current_path(const path & p,error_code * ec)615 void __current_path(const path& p, error_code* ec) {
616   ErrorHandler<void> err("current_path", ec, &p);
617   if (detail::chdir(p.c_str()) == -1)
618     err.report(detail::get_last_error());
619 }
620 
__equivalent(const path & p1,const path & p2,error_code * ec)621 bool __equivalent(const path& p1, const path& p2, error_code* ec) {
622   ErrorHandler<bool> err("equivalent", ec, &p1, &p2);
623 
624   error_code ec1, ec2;
625   StatT st1 = {}, st2 = {};
626   auto s1 = detail::posix_stat(p1.native(), st1, &ec1);
627   if (!exists(s1))
628     return err.report(errc::not_supported);
629   auto s2 = detail::posix_stat(p2.native(), st2, &ec2);
630   if (!exists(s2))
631     return err.report(errc::not_supported);
632 
633   return detail::stat_equivalent(st1, st2);
634 }
635 
__file_size(const path & p,error_code * ec)636 uintmax_t __file_size(const path& p, error_code* ec) {
637   ErrorHandler<uintmax_t> err("file_size", ec, &p);
638 
639   error_code m_ec;
640   StatT st;
641   file_status fst = detail::posix_stat(p, st, &m_ec);
642   if (!exists(fst) || !is_regular_file(fst)) {
643     errc error_kind = is_directory(fst) ? errc::is_a_directory : errc::not_supported;
644     if (!m_ec)
645       m_ec = make_error_code(error_kind);
646     return err.report(m_ec);
647   }
648   // is_regular_file(p) == true
649   return static_cast<uintmax_t>(st.st_size);
650 }
651 
__hard_link_count(const path & p,error_code * ec)652 uintmax_t __hard_link_count(const path& p, error_code* ec) {
653   ErrorHandler<uintmax_t> err("hard_link_count", ec, &p);
654 
655   error_code m_ec;
656   StatT st;
657   detail::posix_stat(p, st, &m_ec);
658   if (m_ec)
659     return err.report(m_ec);
660   return static_cast<uintmax_t>(st.st_nlink);
661 }
662 
__fs_is_empty(const path & p,error_code * ec)663 bool __fs_is_empty(const path& p, error_code* ec) {
664   ErrorHandler<bool> err("is_empty", ec, &p);
665 
666   error_code m_ec;
667   StatT pst;
668   auto st = detail::posix_stat(p, pst, &m_ec);
669   if (m_ec)
670     return err.report(m_ec);
671   else if (!is_directory(st) && !is_regular_file(st))
672     return err.report(errc::not_supported);
673   else if (is_directory(st)) {
674     auto it = ec ? directory_iterator(p, *ec) : directory_iterator(p);
675     if (ec && *ec)
676       return false;
677     return it == directory_iterator{};
678   } else if (is_regular_file(st))
679     return static_cast<uintmax_t>(pst.st_size) == 0;
680 
681   __libcpp_unreachable();
682 }
683 
__last_write_time(const path & p,error_code * ec)684 file_time_type __last_write_time(const path& p, error_code* ec) {
685   using namespace chrono;
686   ErrorHandler<file_time_type> err("last_write_time", ec, &p);
687 
688   error_code m_ec;
689   StatT st;
690   detail::posix_stat(p, st, &m_ec);
691   if (m_ec)
692     return err.report(m_ec);
693   return detail::__extract_last_write_time(p, st, ec);
694 }
695 
__last_write_time(const path & p,file_time_type new_time,error_code * ec)696 void __last_write_time(const path& p, file_time_type new_time, error_code* ec) {
697   using detail::fs_time;
698   ErrorHandler<void> err("last_write_time", ec, &p);
699 
700 #if defined(_LIBCPP_WIN32API)
701   TimeSpec ts;
702   if (!fs_time::convert_to_timespec(ts, new_time))
703     return err.report(errc::value_too_large);
704   detail::WinHandle h(p.c_str(), FILE_WRITE_ATTRIBUTES, 0);
705   if (!h)
706     return err.report(detail::get_last_error());
707   FILETIME last_write = timespec_to_filetime(ts);
708   if (!SetFileTime(h, nullptr, nullptr, &last_write))
709     return err.report(detail::get_last_error());
710 #else
711   error_code m_ec;
712   array<TimeSpec, 2> tbuf;
713 #  if !defined(_LIBCPP_USE_UTIMENSAT)
714   // This implementation has a race condition between determining the
715   // last access time and attempting to set it to the same value using
716   // ::utimes
717   StatT st;
718   file_status fst = detail::posix_stat(p, st, &m_ec);
719   if (m_ec)
720     return err.report(m_ec);
721   tbuf[0] = detail::extract_atime(st);
722 #  else
723   tbuf[0].tv_sec  = 0;
724   tbuf[0].tv_nsec = UTIME_OMIT;
725 #  endif
726   if (!fs_time::convert_to_timespec(tbuf[1], new_time))
727     return err.report(errc::value_too_large);
728 
729   detail::set_file_times(p, tbuf, m_ec);
730   if (m_ec)
731     return err.report(m_ec);
732 #endif
733 }
734 
__permissions(const path & p,perms prms,perm_options opts,error_code * ec)735 void __permissions(const path& p, perms prms, perm_options opts, error_code* ec) {
736   ErrorHandler<void> err("permissions", ec, &p);
737 
738   auto has_opt                = [&](perm_options o) { return bool(o & opts); };
739   const bool resolve_symlinks = !has_opt(perm_options::nofollow);
740   const bool add_perms        = has_opt(perm_options::add);
741   const bool remove_perms     = has_opt(perm_options::remove);
742   _LIBCPP_ASSERT_ARGUMENT_WITHIN_DOMAIN(
743       (add_perms + remove_perms + has_opt(perm_options::replace)) == 1,
744       "One and only one of the perm_options constants 'replace', 'add', or 'remove' must be present in opts");
745 
746   bool set_sym_perms = false;
747   prms &= perms::mask;
748   if (!resolve_symlinks || (add_perms || remove_perms)) {
749     error_code m_ec;
750     file_status st = resolve_symlinks ? detail::posix_stat(p, &m_ec) : detail::posix_lstat(p, &m_ec);
751     set_sym_perms  = is_symlink(st);
752     if (m_ec)
753       return err.report(m_ec);
754     // TODO(hardening): double-check this assertion -- it might be a valid (if rare) case when the permissions are
755     // unknown.
756     _LIBCPP_ASSERT_VALID_EXTERNAL_API_CALL(st.permissions() != perms::unknown, "Permissions unexpectedly unknown");
757     if (add_perms)
758       prms |= st.permissions();
759     else if (remove_perms)
760       prms = st.permissions() & ~prms;
761   }
762   const auto real_perms = static_cast<detail::ModeT>(prms & perms::mask);
763 
764 #if defined(AT_SYMLINK_NOFOLLOW) && defined(AT_FDCWD)
765   const int flags = set_sym_perms ? AT_SYMLINK_NOFOLLOW : 0;
766   if (detail::fchmodat(AT_FDCWD, p.c_str(), real_perms, flags) == -1) {
767     return err.report(detail::get_last_error());
768   }
769 #else
770   if (set_sym_perms)
771     return err.report(errc::operation_not_supported);
772   if (::chmod(p.c_str(), real_perms) == -1) {
773     return err.report(capture_errno());
774   }
775 #endif
776 }
777 
__read_symlink(const path & p,error_code * ec)778 path __read_symlink(const path& p, error_code* ec) {
779   ErrorHandler<path> err("read_symlink", ec, &p);
780 
781 #if defined(PATH_MAX) || defined(MAX_SYMLINK_SIZE)
782   struct NullDeleter {
783     void operator()(void*) const {}
784   };
785 #  ifdef MAX_SYMLINK_SIZE
786   const size_t size = MAX_SYMLINK_SIZE + 1;
787 #  else
788   const size_t size = PATH_MAX + 1;
789 #  endif
790   path::value_type stack_buff[size];
791   auto buff = std::unique_ptr<path::value_type[], NullDeleter>(stack_buff);
792 #else
793   StatT sb;
794   if (detail::lstat(p.c_str(), &sb) == -1) {
795     return err.report(detail::get_last_error());
796   }
797   const size_t size = sb.st_size + 1;
798   auto buff         = unique_ptr<path::value_type[]>(new path::value_type[size]);
799 #endif
800   detail::SSizeT ret;
801   if ((ret = detail::readlink(p.c_str(), buff.get(), size)) == -1)
802     return err.report(detail::get_last_error());
803   // Note that `ret` returning `0` would work, resulting in a valid empty string being returned.
804   if (static_cast<size_t>(ret) >= size)
805     return err.report(errc::value_too_large);
806   buff[ret] = 0;
807   return {buff.get()};
808 }
809 
__remove(const path & p,error_code * ec)810 bool __remove(const path& p, error_code* ec) {
811   ErrorHandler<bool> err("remove", ec, &p);
812   if (detail::remove(p.c_str()) == -1) {
813     error_code mec = detail::get_last_error();
814     if (mec != errc::no_such_file_or_directory)
815       err.report(mec);
816     return false;
817   }
818   return true;
819 }
820 
821 // We currently have two implementations of `__remove_all`. The first one is general and
822 // used on platforms where we don't have access to the `openat()` family of POSIX functions.
823 // That implementation uses `directory_iterator`, however it is vulnerable to some race
824 // conditions, see https://reviews.llvm.org/D118134 for details.
825 //
826 // The second implementation is used on platforms where `openat()` & friends are available,
827 // and it threads file descriptors through recursive calls to avoid such race conditions.
828 #if defined(_LIBCPP_WIN32API) || defined(__MVS__)
829 #  define REMOVE_ALL_USE_DIRECTORY_ITERATOR
830 #endif
831 
832 #if defined(REMOVE_ALL_USE_DIRECTORY_ITERATOR)
833 
834 namespace {
835 
remove_all_impl(path const & p,error_code & ec)836 uintmax_t remove_all_impl(path const& p, error_code& ec) {
837   const auto npos      = static_cast<uintmax_t>(-1);
838   const file_status st = __symlink_status(p, &ec);
839   if (ec)
840     return npos;
841   uintmax_t count = 1;
842   if (is_directory(st)) {
843     for (directory_iterator it(p, ec); !ec && it != directory_iterator(); it.increment(ec)) {
844       auto other_count = remove_all_impl(it->path(), ec);
845       if (ec)
846         return npos;
847       count += other_count;
848     }
849     if (ec)
850       return npos;
851   }
852   if (!__remove(p, &ec))
853     return npos;
854   return count;
855 }
856 
857 } // namespace
858 
__remove_all(const path & p,error_code * ec)859 uintmax_t __remove_all(const path& p, error_code* ec) {
860   ErrorHandler<uintmax_t> err("remove_all", ec, &p);
861 
862   error_code mec;
863   auto count = remove_all_impl(p, mec);
864   if (mec) {
865     if (mec == errc::no_such_file_or_directory)
866       return 0;
867     return err.report(mec);
868   }
869   return count;
870 }
871 
872 #else // !REMOVE_ALL_USE_DIRECTORY_ITERATOR
873 
874 namespace {
875 
876 template <class Cleanup>
877 struct scope_exit {
scope_exit__anon9ba8c6480511::scope_exit878   explicit scope_exit(Cleanup const& cleanup) : cleanup_(cleanup) {}
879 
~scope_exit__anon9ba8c6480511::scope_exit880   ~scope_exit() { cleanup_(); }
881 
882 private:
883   Cleanup cleanup_;
884 };
885 _LIBCPP_CTAD_SUPPORTED_FOR_TYPE(scope_exit);
886 
remove_all_impl(int parent_directory,const path & p,error_code & ec)887 uintmax_t remove_all_impl(int parent_directory, const path& p, error_code& ec) {
888   // First, try to open the path as a directory.
889   const int options = O_CLOEXEC | O_RDONLY | O_DIRECTORY | O_NOFOLLOW;
890   int fd            = ::openat(parent_directory, p.c_str(), options);
891   if (fd != -1) {
892     // If that worked, iterate over the contents of the directory and
893     // remove everything in it, recursively.
894     DIR* stream = ::fdopendir(fd);
895     if (stream == nullptr) {
896       ::close(fd);
897       ec = detail::capture_errno();
898       return 0;
899     }
900     // Note: `::closedir` will also close the associated file descriptor, so
901     // there should be no call to `close(fd)`.
902     scope_exit close_stream([=] { ::closedir(stream); });
903 
904     uintmax_t count = 0;
905     while (true) {
906       auto [str, type] = detail::posix_readdir(stream, ec);
907       static_assert(std::is_same_v<decltype(str), std::string_view>);
908       if (str == "." || str == "..") {
909         continue;
910       } else if (ec || str.empty()) {
911         break; // we're done iterating through the directory
912       } else {
913         count += remove_all_impl(fd, str, ec);
914       }
915     }
916 
917     // Then, remove the now-empty directory itself.
918     if (::unlinkat(parent_directory, p.c_str(), AT_REMOVEDIR) == -1) {
919       ec = detail::capture_errno();
920       return count;
921     }
922 
923     return count + 1; // the contents of the directory + the directory itself
924   }
925 
926   ec = detail::capture_errno();
927 
928   // If we failed to open `p` because it didn't exist, it's not an
929   // error -- it might have moved or have been deleted already.
930   if (ec == errc::no_such_file_or_directory) {
931     ec.clear();
932     return 0;
933   }
934 
935   // If opening `p` failed because it wasn't a directory, remove it as
936   // a normal file instead. Note that `openat()` can return either ENOTDIR
937   // or ELOOP depending on the exact reason of the failure. On FreeBSD it
938   // may return EMLINK instead of ELOOP, contradicting POSIX.
939   if (ec == errc::not_a_directory || ec == errc::too_many_symbolic_link_levels || ec == errc::too_many_links) {
940     ec.clear();
941     if (::unlinkat(parent_directory, p.c_str(), /* flags = */ 0) == -1) {
942       ec = detail::capture_errno();
943       return 0;
944     }
945     return 1;
946   }
947 
948   // Otherwise, it's a real error -- we don't remove anything.
949   return 0;
950 }
951 
952 } // namespace
953 
__remove_all(const path & p,error_code * ec)954 uintmax_t __remove_all(const path& p, error_code* ec) {
955   ErrorHandler<uintmax_t> err("remove_all", ec, &p);
956   error_code mec;
957   uintmax_t count = remove_all_impl(AT_FDCWD, p, mec);
958   if (mec)
959     return err.report(mec);
960   return count;
961 }
962 
963 #endif // REMOVE_ALL_USE_DIRECTORY_ITERATOR
964 
__rename(const path & from,const path & to,error_code * ec)965 void __rename(const path& from, const path& to, error_code* ec) {
966   ErrorHandler<void> err("rename", ec, &from, &to);
967   if (detail::rename(from.c_str(), to.c_str()) == -1)
968     err.report(detail::get_last_error());
969 }
970 
__resize_file(const path & p,uintmax_t size,error_code * ec)971 void __resize_file(const path& p, uintmax_t size, error_code* ec) {
972   ErrorHandler<void> err("resize_file", ec, &p);
973   if (detail::truncate(p.c_str(), static_cast< ::off_t>(size)) == -1)
974     return err.report(detail::get_last_error());
975 }
976 
__space(const path & p,error_code * ec)977 space_info __space(const path& p, error_code* ec) {
978   ErrorHandler<void> err("space", ec, &p);
979   space_info si;
980   detail::StatVFS m_svfs = {};
981   if (detail::statvfs(p.c_str(), &m_svfs) == -1) {
982     err.report(detail::get_last_error());
983     si.capacity = si.free = si.available = static_cast<uintmax_t>(-1);
984     return si;
985   }
986   // Multiply with overflow checking.
987   auto do_mult = [&](uintmax_t& out, uintmax_t other) {
988     out = other * m_svfs.f_frsize;
989     if (other == 0 || out / other != m_svfs.f_frsize)
990       out = static_cast<uintmax_t>(-1);
991   };
992   do_mult(si.capacity, m_svfs.f_blocks);
993   do_mult(si.free, m_svfs.f_bfree);
994   do_mult(si.available, m_svfs.f_bavail);
995   return si;
996 }
997 
__status(const path & p,error_code * ec)998 file_status __status(const path& p, error_code* ec) { return detail::posix_stat(p, ec); }
999 
__symlink_status(const path & p,error_code * ec)1000 file_status __symlink_status(const path& p, error_code* ec) { return detail::posix_lstat(p, ec); }
1001 
__temp_directory_path(error_code * ec)1002 path __temp_directory_path(error_code* ec) {
1003   ErrorHandler<path> err("temp_directory_path", ec);
1004 
1005 #if defined(_LIBCPP_WIN32API)
1006   wchar_t buf[MAX_PATH];
1007   DWORD retval = GetTempPathW(MAX_PATH, buf);
1008   if (!retval)
1009     return err.report(detail::get_last_error());
1010   if (retval > MAX_PATH)
1011     return err.report(errc::filename_too_long);
1012   // GetTempPathW returns a path with a trailing slash, which we
1013   // shouldn't include for consistency.
1014   if (buf[retval - 1] == L'\\')
1015     buf[retval - 1] = L'\0';
1016   path p(buf);
1017 #else
1018   const char* env_paths[] = {"TMPDIR", "TMP", "TEMP", "TEMPDIR"};
1019   const char* ret         = nullptr;
1020 
1021   for (auto& ep : env_paths)
1022     if ((ret = getenv(ep)))
1023       break;
1024   if (ret == nullptr) {
1025 #  if defined(__ANDROID__)
1026     ret = "/data/local/tmp";
1027 #  else
1028     ret = "/tmp";
1029 #  endif
1030   }
1031 
1032   path p(ret);
1033 #endif
1034   error_code m_ec;
1035   file_status st = detail::posix_stat(p, &m_ec);
1036   if (!status_known(st))
1037     return err.report(m_ec, "cannot access path " PATH_CSTR_FMT, p.c_str());
1038 
1039   if (!exists(st) || !is_directory(st))
1040     return err.report(errc::not_a_directory, "path " PATH_CSTR_FMT " is not a directory", p.c_str());
1041 
1042   return p;
1043 }
1044 
__weakly_canonical(const path & p,error_code * ec)1045 path __weakly_canonical(const path& p, error_code* ec) {
1046   ErrorHandler<path> err("weakly_canonical", ec, &p);
1047 
1048   if (p.empty())
1049     return __canonical("", ec);
1050 
1051   path result;
1052   path tmp;
1053   tmp.__reserve(p.native().size());
1054   auto PP = PathParser::CreateEnd(p.native());
1055   --PP;
1056   vector<string_view_t> DNEParts;
1057 
1058   error_code m_ec;
1059   while (PP.State_ != PathParser::PS_BeforeBegin) {
1060     tmp.assign(createView(p.native().data(), &PP.RawEntry.back()));
1061     file_status st = __status(tmp, &m_ec);
1062     if (!status_known(st)) {
1063       return err.report(m_ec);
1064     } else if (exists(st)) {
1065       result = __canonical(tmp, &m_ec);
1066       if (m_ec) {
1067         return err.report(m_ec);
1068       }
1069       break;
1070     }
1071     DNEParts.push_back(*PP);
1072     --PP;
1073   }
1074   if (PP.State_ == PathParser::PS_BeforeBegin) {
1075     result = __canonical("", &m_ec);
1076     if (m_ec) {
1077       return err.report(m_ec);
1078     }
1079   }
1080   if (DNEParts.empty())
1081     return result;
1082   for (auto It = DNEParts.rbegin(); It != DNEParts.rend(); ++It)
1083     result /= *It;
1084   return result.lexically_normal();
1085 }
1086 
1087 _LIBCPP_END_NAMESPACE_FILESYSTEM
1088