// file : build2/test/script/builtin.cxx -*- C++ -*- // copyright : Copyright (c) 2014-2018 Code Synthesis Ltd // license : MIT; see accompanying LICENSE file #include #include #include #include #include #include // use default operator<< implementation #include // fdopen_mode, fdstream_mode #include #include // Strictly speaking a builtin which reads/writes from/to standard streams // must be asynchronous so that the caller can communicate with it through // pipes without being blocked on I/O operations. However, as an optimization, // we allow builtins that only print diagnostics to STDERR to be synchronous // assuming that their output will always fit the pipe buffer. Synchronous // builtins must not read from STDIN and write to STDOUT. Later we may relax // this rule to allow a "short" output for such builtins. // using namespace std; using namespace butl; namespace build2 { namespace test { namespace script { using builtin_impl = uint8_t (scope&, const strings& args, auto_fd in, auto_fd out, auto_fd err); // Operation failed, diagnostics has already been issued. // struct failed {}; // Accumulate an error message, print it atomically in dtor to the // provided stream and throw failed afterwards if requested. Prefixes // the message with the builtin name. // // Move constructible-only, not assignable (based to diag_record). // class error_record { public: template friend const error_record& operator<< (const error_record& r, const T& x) { r.ss_ << x; return r; } error_record (ostream& o, bool fail, const char* name) : os_ (o), fail_ (fail), empty_ (false) { ss_ << name << ": "; } // Older versions of libstdc++ don't have the ostringstream move // support. Luckily, GCC doesn't seem to be actually needing move due // to copy/move elision. // #ifdef __GLIBCXX__ error_record (error_record&&); #else error_record (error_record&& r) : os_ (r.os_), ss_ (move (r.ss_)), fail_ (r.fail_), empty_ (r.empty_) { r.empty_ = true; } #endif ~error_record () noexcept (false) { if (!empty_) { // The output stream can be in a bad state (for example as a // result of unsuccessful attempt to report a previous error), so // we check it. // if (os_.good ()) { ss_.put ('\n'); os_ << ss_.str (); os_.flush (); } if (fail_) throw failed (); } } private: ostream& os_; mutable ostringstream ss_; bool fail_; bool empty_; }; // Parse and normalize a path. Also, unless it is already absolute, make // the path absolute using the specified directory. Throw invalid_path // if the path is empty, and on parsing and normalization failures. // static path parse_path (string s, const dir_path& d) { path p (move (s)); if (p.empty ()) throw invalid_path (""); if (p.relative ()) p = d / move (p); p.normalize (); return p; } // Builtin commands functions. // // cat ... // // Note that POSIX doesn't specify if after I/O operation failure the // command should proceed with the rest of the arguments. The current // implementation exits immediatelly in such a case. // // @@ Shouldn't we check that we don't print a nonempty regular file to // itself, as that would merely exhaust the output device? POSIX // allows (but not requires) such a check and some implementations do // this. That would require to fstat() file descriptors and complicate // the code a bit. Was able to reproduce on a big file (should be // bigger than the stream buffer size) with the test // 'cat file >+file'. // // Note: must be executed asynchronously. // static uint8_t cat (scope& sp, const strings& args, auto_fd in, auto_fd out, auto_fd err) noexcept try { uint8_t r (1); ofdstream cerr (move (err)); auto error = [&cerr] (bool fail = true) { return error_record (cerr, fail, "cat"); }; try { ifdstream cin (move (in), fdstream_mode::binary); ofdstream cout (move (out), fdstream_mode::binary); // Copy input stream to STDOUT. // auto copy = [&cout] (istream& is) { if (is.peek () != ifdstream::traits_type::eof ()) cout << is.rdbuf (); is.clear (istream::eofbit); // Sets eofbit. }; // Path of a file being printed to STDOUT. An empty path represents // STDIN. Used in diagnostics. // path p; try { // Print STDIN. // if (args.empty ()) copy (cin); // Print files. // for (auto i (args.begin ()); i != args.end (); ++i) { if (*i == "-") { if (!cin.eof ()) { p.clear (); copy (cin); } continue; } p = parse_path (*i, sp.wd_path); ifdstream is (p, ifdstream::binary); copy (is); is.close (); } } catch (const io_error& e) { error_record d (error ()); d << "unable to print "; if (p.empty ()) d << "stdin"; else d << "'" << p << "'"; d << ": " << e; } cin.close (); cout.close (); r = 0; } catch (const invalid_path& e) { error (false) << "invalid path '" << e.path << "'"; } // Can be thrown while creating/closing cin, cout or writing to cerr. // catch (const io_error& e) { error (false) << e; } catch (const failed&) { // Diagnostics has already been issued. } cerr.close (); return r; } catch (const std::exception&) { return 1; } // Make a copy of a file at the specified path, preserving permissions, // and registering a cleanup for a newly created file. The file paths // must be absolute. Fail if an exception is thrown by the underlying // copy operation. // static void cpfile (scope& sp, const path& from, const path& to, bool overwrite, bool attrs, bool cleanup, const function& fail) { try { bool exists (file_exists (to)); cpflags f ( overwrite ? cpflags::overwrite_permissions | cpflags::overwrite_content : cpflags::none); if (attrs) f |= cpflags::overwrite_permissions | cpflags::copy_timestamps; cpfile (from, to, f); if (!exists && cleanup) sp.clean ({cleanup_type::always, to}, true); } catch (const system_error& e) { fail () << "unable to copy file '" << from << "' to '" << to << "': " << e; } } // Make a copy of a directory at the specified path, registering a // cleanup for the created directory. The directory paths must be // absolute. Fail if the destination directory already exists or // an exception is thrown by the underlying copy operation. // static void cpdir (scope& sp, const dir_path& from, const dir_path& to, bool attrs, bool cleanup, const function& fail) { try { if (try_mkdir (to) == mkdir_status::already_exists) throw_generic_error (EEXIST); if (cleanup) sp.clean ({cleanup_type::always, to}, true); for (const auto& de: dir_iterator (from, false /* ignore_dangling */)) { path f (from / de.path ()); path t (to / de.path ()); if (de.type () == entry_type::directory) cpdir (sp, path_cast (move (f)), path_cast (move (t)), attrs, cleanup, fail); else cpfile (sp, f, t, false /* overwrite */, attrs, cleanup, fail); } // Note that it is essential to copy timestamps and permissions after // the directory content is copied. // if (attrs) { path_permissions (to, path_permissions (from)); dir_time (to, dir_time (from)); } } catch (const system_error& e) { fail () << "unable to copy directory '" << from << "' to '" << to << "': " << e; } } // cp [-p] [--no-cleanup] // cp [-p] [--no-cleanup] -R|-r // cp [-p] [--no-cleanup] ... / // cp [-p] [--no-cleanup] -R|-r ... / // // Note: can be executed synchronously. // static uint8_t cp (scope& sp, const strings& args, auto_fd in, auto_fd out, auto_fd err) noexcept try { uint8_t r (1); ofdstream cerr (move (err)); auto error = [&cerr] (bool fail = true) { return error_record (cerr, fail, "cp"); }; try { in.close (); out.close (); auto i (args.begin ()); auto e (args.end ()); // Process options. // bool recursive (false); bool attrs (false); bool cleanup (true); for (; i != e; ++i) { const string& o (*i); if (o == "-R" || o == "-r") recursive = true; else if (o == "-p") attrs = true; else if (o == "--no-cleanup") cleanup = false; else { if (o == "--") ++i; break; } } // Copy files or directories. // if (i == e) error () << "missing arguments"; const dir_path& wd (sp.wd_path); auto j (args.rbegin ()); path dst (parse_path (*j++, wd)); e = j.base (); if (i == e) error () << "missing source path"; auto fail = [&error] () {return error (true);}; // If destination is not a directory path (no trailing separator) // then make a copy of the filesystem entry at the specified path // (the only source path is allowed in such a case). Otherwise copy // the source filesystem entries into the destination directory. // if (!dst.to_directory ()) { path src (parse_path (*i++, wd)); // If there are multiple sources but no trailing separator for the // destination, then, most likelly, it is missing. // if (i != e) error () << "multiple source paths without trailing separator " << "for destination directory"; if (!recursive) // Synopsis 1: make a file copy at the specified path. // cpfile (sp, src, dst, true /* overwrite */, attrs, cleanup, fail); else // Synopsis 2: make a directory copy at the specified path. // cpdir (sp, path_cast (src), path_cast (dst), attrs, cleanup, fail); } else { for (; i != e; ++i) { path src (parse_path (*i, wd)); if (recursive && dir_exists (src)) // Synopsis 4: copy a filesystem entry into the specified // directory. Note that we handle only source directories here. // Source files are handled below. // cpdir (sp, path_cast (src), path_cast (dst / src.leaf ()), attrs, cleanup, fail); else // Synopsis 3: copy a file into the specified directory. Also, // here we cover synopsis 4 for the source path being a file. // cpfile (sp, src, dst / src.leaf (), true /* overwrite */, attrs, cleanup, fail); } } r = 0; } catch (const invalid_path& e) { error (false) << "invalid path '" << e.path << "'"; } // Can be thrown while closing in, out or writing to cerr. // catch (const io_error& e) { error (false) << e; } catch (const failed&) { // Diagnostics has already been issued. } cerr.close (); return r; } catch (const std::exception&) { return 1; } // echo ... // // Note: must be executed asynchronously. // static uint8_t echo (scope&, const strings& args, auto_fd in, auto_fd out, auto_fd err) noexcept try { uint8_t r (1); ofdstream cerr (move (err)); try { in.close (); ofdstream cout (move (out)); for (auto b (args.begin ()), i (b), e (args.end ()); i != e; ++i) cout << (i != b ? " " : "") << *i; cout << '\n'; cout.close (); r = 0; } catch (const std::exception& e) { cerr << "echo: " << e << endl; } cerr.close (); return r; } catch (const std::exception&) { return 1; } // false // // Failure to close the file descriptors is silently ignored. // // Note: can be executed synchronously. // static builtin false_ (scope&, uint8_t& r, const strings&, auto_fd, auto_fd, auto_fd) { return builtin (r = 1); } // true // // Failure to close the file descriptors is silently ignored. // // Note: can be executed synchronously. // static builtin true_ (scope&, uint8_t& r, const strings&, auto_fd, auto_fd, auto_fd) { return builtin (r = 0); } // Create a symlink to a file or directory at the specified path. The // paths must be absolute. Fall back to creating a hardlink, if symlink // creation is not supported for the link path. If hardlink creation is // not supported either, then fall back to copies. If requested, created // filesystem entries are registered for cleanup. Fail if the target // filesystem entry doesn't exist or an exception is thrown by the // underlying filesystem operation (specifically for an already existing // filesystem entry at the link path). // // Note that supporting optional removal of an existing filesystem entry // at the link path (the -f option) tends to get hairy. As soon as an // existing and the resulting filesystem entries could be of different // types, we would end up with canceling an old cleanup and registering // the new one. Also removing non-empty directories doesn't look very // natural, but would be required if we want the behavior on POSIX and // Windows to be consistent. // static void mksymlink (scope& sp, const path& target, const path& link, bool cleanup, const function& fail) { // Determine the target type, fail if the target doesn't exist. // bool dir (false); try { pair pe (path_entry (target)); if (!pe.first) fail () << "unable to create symlink to '" << target << "': " << "no such file or directory"; dir = pe.second.type == entry_type::directory; } catch (const system_error& e) { fail () << "unable to stat '" << target << "': " << e; } // First we try to create a symlink. If that fails (e.g., "Windows // happens"), then we resort to hard links. If that doesn't work out // either (e.g., not on the same filesystem), then we fall back to // copies. So things are going to get a bit nested. // try { mksymlink (target, link, dir); if (cleanup) sp.clean ({cleanup_type::always, link}, true); } catch (const system_error& e) { // Note that we are not guaranteed (here and below) that the // system_error exception is of the generic category. // int c (e.code ().value ()); if (!(e.code ().category () == generic_category () && (c == ENOSYS || // Not implemented. c == EPERM))) // Not supported by the filesystem(s). fail () << "unable to create symlink '" << link << "' to '" << target << "': " << e; try { mkhardlink (target, link, dir); if (cleanup) sp.clean ({cleanup_type::always, link}, true); } catch (const system_error& e) { c = e.code ().value (); if (!(e.code ().category () == generic_category () && (c == ENOSYS || // Not implemented. c == EPERM || // Not supported by the filesystem(s). c == EXDEV))) // On different filesystems. fail () << "unable to create hardlink '" << link << "' to '" << target << "': " << e; if (dir) cpdir (sp, path_cast (target), path_cast (link), false, cleanup, fail); else cpfile (sp, target, link, false /* overwrite */, true /* attrs */, cleanup, fail); } } } // ln [--no-cleanup] -s // ln [--no-cleanup] -s ... / // // Note: can be executed synchronously. // static uint8_t ln (scope& sp, const strings& args, auto_fd in, auto_fd out, auto_fd err) noexcept try { uint8_t r (1); ofdstream cerr (move (err)); auto error = [&cerr] (bool fail = true) { return error_record (cerr, fail, "ln"); }; try { in.close (); out.close (); auto i (args.begin ()); auto e (args.end ()); // Process options. // bool cleanup (true); bool symlink (false); for (; i != e; ++i) { const string& o (*i); if (o == "--no-cleanup") cleanup = false; else if (o == "-s") symlink = true; else { if (o == "--") ++i; break; } } if (!symlink) error () << "missing -s option"; // Create file or directory symlinks. // if (i == e) error () << "missing arguments"; const dir_path& wd (sp.wd_path); auto j (args.rbegin ()); path link (parse_path (*j++, wd)); e = j.base (); if (i == e) error () << "missing target path"; auto fail = [&error] () {return error (true);}; // If link is not a directory path (no trailing separator), then // create a symlink to the target path at the specified link path // (the only target path is allowed in such a case). Otherwise create // links to the target paths inside the specified directory. // if (!link.to_directory ()) { path target (parse_path (*i++, wd)); // If there are multiple targets but no trailing separator for the // link, then, most likelly, it is missing. // if (i != e) error () << "multiple target paths with non-directory link path"; // Synopsis 1: create a target path symlink at the specified path. // mksymlink (sp, target, link, cleanup, fail); } else { for (; i != e; ++i) { path target (parse_path (*i, wd)); // Synopsis 2: create a target path symlink in the specified // directory. // mksymlink (sp, target, link / target.leaf (), cleanup, fail); } } r = 0; } catch (const invalid_path& e) { error (false) << "invalid path '" << e.path << "'"; } // Can be thrown while closing in, out or writing to cerr. // catch (const io_error& e) { error (false) << e; } catch (const failed&) { // Diagnostics has already been issued. } cerr.close (); return r; } catch (const std::exception&) { return 1; } // Create a directory if not exist and its parent directories if // necessary. Throw system_error on failure. Register created // directories for cleanup. The directory path must be absolute. // static void mkdir_p (scope& sp, const dir_path& p, bool cleanup) { if (!dir_exists (p)) { if (!p.root ()) mkdir_p (sp, p.directory (), cleanup); try_mkdir (p); // Returns success or throws. if (cleanup) sp.clean ({cleanup_type::always, p}, true); } } // mkdir [--no-cleanup] [-p] ... // // Note that POSIX doesn't specify if after a directory creation failure // the command should proceed with the rest of the arguments. The current // implementation exits immediatelly in such a case. // // Note: can be executed synchronously. // static uint8_t mkdir (scope& sp, const strings& args, auto_fd in, auto_fd out, auto_fd err) noexcept try { uint8_t r (1); ofdstream cerr (move (err)); auto error = [&cerr] (bool fail = true) { return error_record (cerr, fail, "mkdir"); }; try { in.close (); out.close (); auto i (args.begin ()); auto e (args.end ()); // Process options. // bool parent (false); bool cleanup (true); for (; i != e; ++i) { const string& o (*i); if (o == "-p") parent = true; else if (o == "--no-cleanup") cleanup = false; else { if (*i == "--") ++i; break; } } // Create directories. // if (i == e) error () << "missing directory"; for (; i != e; ++i) { dir_path p (path_cast (parse_path (*i, sp.wd_path))); try { if (parent) mkdir_p (sp, p, cleanup); else if (try_mkdir (p) == mkdir_status::success) { if (cleanup) sp.clean ({cleanup_type::always, p}, true); } else // == mkdir_status::already_exists throw_generic_error (EEXIST); } catch (const system_error& e) { error () << "unable to create directory '" << p << "': " << e; } } r = 0; } catch (const invalid_path& e) { error (false) << "invalid path '" << e.path << "'"; } // Can be thrown while closing in, out or writing to cerr. // catch (const io_error& e) { error (false) << e; } catch (const failed&) { // Diagnostics has already been issued. } cerr.close (); return r; } catch (const std::exception&) { return 1; } // mv [--no-cleanup] [-f] // mv [--no-cleanup] [-f] ... / // // Note: can be executed synchronously. // static uint8_t mv (scope& sp, const strings& args, auto_fd in, auto_fd out, auto_fd err) noexcept try { uint8_t r (1); ofdstream cerr (move (err)); auto error = [&cerr] (bool fail = true) { return error_record (cerr, fail, "mv"); }; try { in.close (); out.close (); auto i (args.begin ()); auto e (args.end ()); // Process options. // bool no_cleanup (false); bool force (false); for (; i != e; ++i) { const string& o (*i); if (o == "--no-cleanup") no_cleanup = true; else if (*i == "-f") force = true; else { if (o == "--") ++i; break; } } // Move filesystem entries. // if (i == e) error () << "missing arguments"; const dir_path& wd (sp.wd_path); auto j (args.rbegin ()); path dst (parse_path (*j++, wd)); e = j.base (); if (i == e) error () << "missing source path"; auto mv = [no_cleanup, force, &wd, &sp, &error] (const path& from, const path& to) { const dir_path& rwd (sp.root->wd_path); if (!from.sub (rwd) && !force) error () << "'" << from << "' is out of working directory '" << rwd << "'"; try { auto check_wd = [&wd, &error] (const path& p) { if (wd.sub (path_cast (p))) error () << "'" << p << "' contains test working directory '" << wd << "'"; }; check_wd (from); check_wd (to); bool exists (entry_exists (to)); // Fail if the source and destination paths are the same. // // Note that for mventry() function (that is based on the POSIX // rename() function) this is a noop. // if (exists && to == from) error () << "unable to move entity '" << from << "' to itself"; // Rename/move the filesystem entry, replacing an existing one. // mventry (from, to, cpflags::overwrite_permissions | cpflags::overwrite_content); // Unless suppressed, adjust the cleanups that are sub-paths of // the source path. // if (!no_cleanup) { // "Move" the matching cleanup if the destination path doesn't // exist and is a sub-path of the working directory. Otherwise // just remove it. // // Note that it's not enough to just change the cleanup paths. // We also need to make sure that these cleanups happen before // the destination directory (or any of its parents) cleanup, // that is potentially registered. To achieve that we can just // relocate these cleanup entries to the end of the list, // preserving their mutual order. Remember that cleanups in // the list are executed in the reversed order. // bool mv_cleanups (!exists && to.sub (rwd)); cleanups cs; // Remove the source path sub-path cleanups from the list, // adjusting/caching them if required (see above). // for (auto i (sp.cleanups.begin ()); i != sp.cleanups.end (); ) { cleanup& c (*i); path& p (c.path); if (p.sub (from)) { if (mv_cleanups) { // Note that we need to preserve the cleanup path // trailing separator which indicates the removal // method. Also note that leaf(), in particular, does // that. // p = p != from ? to / p.leaf (path_cast (from)) : p.to_directory () ? path_cast (to) : to; cs.push_back (move (c)); } i = sp.cleanups.erase (i); } else ++i; } // Re-insert the adjusted cleanups at the end of the list. // sp.cleanups.insert (sp.cleanups.end (), make_move_iterator (cs.begin ()), make_move_iterator (cs.end ())); } } catch (const system_error& e) { error () << "unable to move entity '" << from << "' to '" << to << "': " << e; } }; // If destination is not a directory path (no trailing separator) // then move the filesystem entry to the specified path (the only // source path is allowed in such a case). Otherwise move the source // filesystem entries into the destination directory. // if (!dst.to_directory ()) { path src (parse_path (*i++, wd)); // If there are multiple sources but no trailing separator for the // destination, then, most likelly, it is missing. // if (i != e) error () << "multiple source paths without trailing separator " << "for destination directory"; // Synopsis 1: move an entity to the specified path. // mv (src, dst); } else { // Synopsis 2: move entities into the specified directory. // for (; i != e; ++i) { path src (parse_path (*i, wd)); mv (src, dst / src.leaf ()); } } r = 0; } catch (const invalid_path& e) { error (false) << "invalid path '" << e.path << "'"; } // Can be thrown while closing in, out or writing to cerr. // catch (const io_error& e) { error (false) << e; } catch (const failed&) { // Diagnostics has already been issued. } cerr.close (); return r; } catch (const std::exception&) { return 1; } // rm [-r] [-f] ... // // The implementation deviates from POSIX in a number of ways. It doesn't // interact with a user and fails immediatelly if unable to process an // argument. It doesn't check for dots containment in the path, and // doesn't consider files and directory permissions in any way just // trying to remove a filesystem entry. Always fails if empty path is // specified. // // Note: can be executed synchronously. // static uint8_t rm (scope& sp, const strings& args, auto_fd in, auto_fd out, auto_fd err) noexcept try { uint8_t r (1); ofdstream cerr (move (err)); auto error = [&cerr] (bool fail = true) { return error_record (cerr, fail, "rm"); }; try { in.close (); out.close (); auto i (args.begin ()); auto e (args.end ()); // Process options. // bool dir (false); bool force (false); for (; i != e; ++i) { if (*i == "-r") dir = true; else if (*i == "-f") force = true; else { if (*i == "--") ++i; break; } } // Remove entries. // if (i == e && !force) error () << "missing file"; const dir_path& wd (sp.wd_path); const dir_path& rwd (sp.root->wd_path); for (; i != e; ++i) { path p (parse_path (*i, wd)); if (!p.sub (rwd) && !force) error () << "'" << p << "' is out of working directory '" << rwd << "'"; try { dir_path d (path_cast (p)); if (dir_exists (d)) { if (!dir) error () << "'" << p << "' is a directory"; if (wd.sub (d)) error () << "'" << p << "' contains test working directory '" << wd << "'"; // The call can result in rmdir_status::not_exist. That's not // very likelly but there is also nothing bad about it. // try_rmdir_r (d); } else if (try_rmfile (p) == rmfile_status::not_exist && !force) throw_generic_error (ENOENT); } catch (const system_error& e) { error () << "unable to remove '" << p << "': " << e; } } r = 0; } catch (const invalid_path& e) { error (false) << "invalid path '" << e.path << "'"; } // Can be thrown while closing in, out or writing to cerr. // catch (const io_error& e) { error (false) << e; } catch (const failed&) { // Diagnostics has already been issued. } cerr.close (); return r; } catch (const std::exception&) { return 1; } // rmdir [-f] ... // // Note: can be executed synchronously. // static uint8_t rmdir (scope& sp, const strings& args, auto_fd in, auto_fd out, auto_fd err) noexcept try { uint8_t r (1); ofdstream cerr (move (err)); auto error = [&cerr] (bool fail = true) { return error_record (cerr, fail, "rmdir"); }; try { in.close (); out.close (); auto i (args.begin ()); auto e (args.end ()); // Process options. // bool force (false); for (; i != e; ++i) { if (*i == "-f") force = true; else { if (*i == "--") ++i; break; } } // Remove directories. // if (i == e && !force) error () << "missing directory"; const dir_path& wd (sp.wd_path); const dir_path& rwd (sp.root->wd_path); for (; i != e; ++i) { dir_path p (path_cast (parse_path (*i, wd))); if (wd.sub (p)) error () << "'" << p << "' contains test working directory '" << wd << "'"; if (!p.sub (rwd) && !force) error () << "'" << p << "' is out of working directory '" << rwd << "'"; try { rmdir_status s (try_rmdir (p)); if (s == rmdir_status::not_empty) throw_generic_error (ENOTEMPTY); else if (s == rmdir_status::not_exist && !force) throw_generic_error (ENOENT); } catch (const system_error& e) { error () << "unable to remove '" << p << "': " << e; } } r = 0; } catch (const invalid_path& e) { error (false) << "invalid path '" << e.path << "'"; } // Can be thrown while closing in, out or writing to cerr. // catch (const io_error& e) { error (false) << e; } catch (const failed&) { // Diagnostics has already been issued. } cerr.close (); return r; } catch (const std::exception&) { return 1; } // sed [-n] [-i] -e