summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorAnotherTest <ali.mpfard@gmail.com>2020-05-16 23:35:13 +0430
committerAndreas Kling <kling@serenityos.org>2020-05-17 11:58:08 +0200
commita398898c1282b21de887537b3c472c5789c8fd6d (patch)
tree9def62ce7e648fd82b9c8677c4eba979e8a2e2ce
parent172df68666dbfb9770cff6ab472b7fe1eea80a6b (diff)
downloadserenity-a398898c1282b21de887537b3c472c5789c8fd6d.zip
Shell: Switch to using Core::EventLoop
This commit refactors Shell to a Core::Object and switches its looping to Core::EventLoop.
-rw-r--r--Shell/CMakeLists.txt1
-rw-r--r--Shell/Execution.h (renamed from Shell/GlobalState.h)33
-rw-r--r--Shell/Job.h36
-rw-r--r--Shell/Shell.cpp1701
-rw-r--r--Shell/Shell.h185
-rw-r--r--Shell/main.cpp1780
6 files changed, 1975 insertions, 1761 deletions
diff --git a/Shell/CMakeLists.txt b/Shell/CMakeLists.txt
index 77de643c33..bf35dab3a9 100644
--- a/Shell/CMakeLists.txt
+++ b/Shell/CMakeLists.txt
@@ -1,6 +1,7 @@
set(SOURCES
main.cpp
Parser.cpp
+ Shell.cpp
)
serenity_bin(Shell)
diff --git a/Shell/GlobalState.h b/Shell/Execution.h
index 63f7a56cb6..bf45c76418 100644
--- a/Shell/GlobalState.h
+++ b/Shell/Execution.h
@@ -26,28 +26,19 @@
#pragma once
-#include "Job.h"
-#include <AK/CircularQueue.h>
-#include <AK/HashMap.h>
+#include <AK/Forward.h>
#include <AK/String.h>
#include <AK/Vector.h>
-#include <termios.h>
+#include <LibCore/ElapsedTimer.h>
-struct GlobalState {
- String cwd;
- String username;
- String home;
- char ttyname[32];
- char hostname[64];
- uid_t uid;
- struct termios termios;
- struct termios default_termios;
- bool was_interrupted { false };
- bool was_resized { false };
- int last_return_code { 0 };
- Vector<String> directory_stack;
- CircularQueue<String, 8> cd_history; // FIXME: have a configurable cd history length
- HashMap<u64, Job> jobs;
-};
+class FileDescriptionCollector {
+public:
+ FileDescriptionCollector() { }
+ ~FileDescriptionCollector();
+
+ void collect();
+ void add(int fd);
-extern GlobalState g;
+private:
+ Vector<int, 32> m_fds;
+};
diff --git a/Shell/Job.h b/Shell/Job.h
index 073741608a..341084931a 100644
--- a/Shell/Job.h
+++ b/Shell/Job.h
@@ -26,7 +26,13 @@
#pragma once
+#include "Execution.h"
+#include <AK/JsonObject.h>
+#include <AK/JsonValue.h>
+#include <AK/OwnPtr.h>
#include <AK/String.h>
+#include <LibCore/ElapsedTimer.h>
+#include <LibCore/Object.h>
class Job {
public:
@@ -34,22 +40,52 @@ public:
{
}
+ ~Job()
+ {
+ auto elapsed = m_command_timer.elapsed();
+ dbg() << "Command \"" << m_cmd << "\" finished in " << elapsed << " ms";
+ }
+
Job(pid_t pid, unsigned pgid, String cmd, u64 job_id)
: m_pgid(pgid)
, m_pid(pid)
, m_job_id(job_id)
, m_cmd(move(cmd))
{
+ set_running_in_background(false);
+ m_command_timer.start();
}
unsigned pgid() const { return m_pgid; }
pid_t pid() const { return m_pid; }
const String& cmd() const { return m_cmd; }
u64 job_id() const { return m_job_id; }
+ bool exited() const { return m_exited; }
+ int exit_code() const { return m_exit_code; }
+ bool is_running_in_background() const { return m_running_in_background; }
+
+ Core::ElapsedTimer& timer() { return m_command_timer; }
+
+ void set_has_exit(int exit_code)
+ {
+ if (m_exited)
+ return;
+ m_exit_code = exit_code;
+ m_exited = true;
+ }
+
+ void set_running_in_background(bool running_in_background)
+ {
+ m_running_in_background = running_in_background;
+ }
private:
unsigned m_pgid { 0 };
pid_t m_pid { 0 };
u64 m_job_id { 0 };
String m_cmd;
+ bool m_exited { false };
+ bool m_running_in_background { false };
+ int m_exit_code { -1 };
+ Core::ElapsedTimer m_command_timer;
};
diff --git a/Shell/Shell.cpp b/Shell/Shell.cpp
new file mode 100644
index 0000000000..d56f2b83e7
--- /dev/null
+++ b/Shell/Shell.cpp
@@ -0,0 +1,1701 @@
+/*
+ * Copyright (c) 2020, The SerenityOS developers.
+ * All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions are met:
+ *
+ * 1. Redistributions of source code must retain the above copyright notice, this
+ * list of conditions and the following disclaimer.
+ *
+ * 2. Redistributions in binary form must reproduce the above copyright notice,
+ * this list of conditions and the following disclaimer in the documentation
+ * and/or other materials provided with the distribution.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
+ * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+ * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+ * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
+ * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+ * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
+ * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
+ * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
+ * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+ * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ */
+
+#include "Shell.h"
+#include "Execution.h"
+#include <AK/FileSystemPath.h>
+#include <AK/Function.h>
+#include <AK/ScopeGuard.h>
+#include <AK/StringBuilder.h>
+#include <LibCore/ArgsParser.h>
+#include <LibCore/DirIterator.h>
+#include <LibCore/ElapsedTimer.h>
+#include <LibCore/Event.h>
+#include <LibCore/EventLoop.h>
+#include <LibCore/File.h>
+#include <LibLine/Editor.h>
+#include <errno.h>
+#include <fcntl.h>
+#include <pwd.h>
+#include <signal.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <sys/mman.h>
+#include <sys/stat.h>
+#include <sys/utsname.h>
+#include <sys/wait.h>
+#include <termios.h>
+#include <unistd.h>
+
+// FIXME: We do not expand variables inside strings
+// if we want to be more sh-like, we should do that some day
+static constexpr bool HighlightVariablesInsideStrings = false;
+static bool s_disable_hyperlinks = false;
+extern Line::Editor editor;
+
+//#define SH_DEBUG
+
+void Shell::print_path(const String& path)
+{
+ if (s_disable_hyperlinks) {
+ printf("%s", path.characters());
+ return;
+ }
+ printf("\033]8;;file://%s%s\033\\%s\033]8;;\033\\", hostname, path.characters(), path.characters());
+}
+
+String Shell::prompt() const
+{
+ auto build_prompt = [&]() -> String {
+ auto* ps1 = getenv("PROMPT");
+ if (!ps1) {
+ if (uid == 0)
+ return "# ";
+
+ StringBuilder builder;
+ builder.appendf("\033]0;%s@%s:%s\007", username.characters(), hostname, cwd.characters());
+ builder.appendf("\033[31;1m%s\033[0m@\033[37;1m%s\033[0m:\033[32;1m%s\033[0m$> ", username.characters(), hostname, cwd.characters());
+ return builder.to_string();
+ }
+
+ StringBuilder builder;
+ for (char* ptr = ps1; *ptr; ++ptr) {
+ if (*ptr == '\\') {
+ ++ptr;
+ if (!*ptr)
+ break;
+ switch (*ptr) {
+ case 'X':
+ builder.append("\033]0;");
+ break;
+ case 'a':
+ builder.append(0x07);
+ break;
+ case 'e':
+ builder.append(0x1b);
+ break;
+ case 'u':
+ builder.append(username);
+ break;
+ case 'h':
+ builder.append(hostname);
+ break;
+ case 'w': {
+ String home_path = getenv("HOME");
+ if (cwd.starts_with(home_path)) {
+ builder.append('~');
+ builder.append(cwd.substring_view(home_path.length(), cwd.length() - home_path.length()));
+ } else {
+ builder.append(cwd);
+ }
+ break;
+ }
+ case 'p':
+ builder.append(uid == 0 ? '#' : '$');
+ break;
+ }
+ continue;
+ }
+ builder.append(*ptr);
+ }
+ return builder.to_string();
+ };
+
+ auto the_prompt = build_prompt();
+ auto prompt_length = editor.actual_rendered_string_length(the_prompt);
+
+ if (m_should_continue != ExitCodeOrContinuationRequest::Nothing) {
+ const auto format_string = "\033[34m%.*-s\033[m";
+ switch (m_should_continue) {
+ case ExitCodeOrContinuationRequest::Pipe:
+ return String::format(format_string, prompt_length, "pipe> ");
+ case ExitCodeOrContinuationRequest::DoubleQuotedString:
+ return String::format(format_string, prompt_length, "dquote> ");
+ case ExitCodeOrContinuationRequest::SingleQuotedString:
+ return String::format(format_string, prompt_length, "squote> ");
+ default:
+ break;
+ }
+ }
+ return the_prompt;
+}
+
+int Shell::builtin_bg(int argc, const char** argv)
+{
+ int job_id = -1;
+
+ Core::ArgsParser parser;
+ parser.add_positional_argument(job_id, "Job id to run in background", "job_id", Core::ArgsParser::Required::No);
+
+ if (!parser.parse(argc, const_cast<char**>(argv), false))
+ return 1;
+
+ if (job_id == -1)
+ job_id = jobs.size() - 1;
+
+ Job* job = nullptr;
+
+ for (auto& entry : jobs) {
+ if (entry.value->job_id() == (u64)job_id) {
+ job = entry.value;
+ break;
+ }
+ }
+ if (!job) {
+ if (job_id == -1) {
+ printf("bg: no current job\n");
+ } else {
+ printf("bg: job with id %d not found\n", job_id);
+ }
+ return 1;
+ }
+
+ job->set_running_in_background(true);
+
+ dbg() << "Resuming " << job->pid() << " (" << job->cmd() << ")";
+ printf("Resuming job %llu - %s\n", job->job_id(), job->cmd().characters());
+
+ if (killpg(job->pgid(), SIGCONT) < 0) {
+ perror("killpg");
+ return 1;
+ }
+
+ return 0;
+}
+
+int Shell::builtin_cd(int argc, const char** argv)
+{
+ const char* arg_path = nullptr;
+
+ Core::ArgsParser parser;
+ parser.add_positional_argument(arg_path, "Path to change to", "path", Core::ArgsParser::Required::No);
+
+ if (!parser.parse(argc, const_cast<char**>(argv), false))
+ return 1;
+
+ String new_path;
+
+ if (!arg_path) {
+ new_path = home;
+ if (cd_history.is_empty() || cd_history.last() != home)
+ cd_history.enqueue(home);
+ } else {
+ if (cd_history.is_empty() || cd_history.last() != arg_path)
+ cd_history.enqueue(arg_path);
+ if (strcmp(argv[1], "-") == 0) {
+ char* oldpwd = getenv("OLDPWD");
+ if (oldpwd == nullptr)
+ return 1;
+ new_path = oldpwd;
+ } else if (arg_path[0] == '/') {
+ new_path = argv[1];
+ } else {
+ StringBuilder builder;
+ builder.append(cwd);
+ builder.append('/');
+ builder.append(arg_path);
+ new_path = builder.to_string();
+ }
+ }
+
+ FileSystemPath canonical_path(new_path);
+ if (!canonical_path.is_valid()) {
+ printf("FileSystemPath failed to canonicalize '%s'\n", new_path.characters());
+ return 1;
+ }
+ const char* path = canonical_path.string().characters();
+
+ struct stat st;
+ int rc = stat(path, &st);
+ if (rc < 0) {
+ printf("stat(%s) failed: %s\n", path, strerror(errno));
+ return 1;
+ }
+ if (!S_ISDIR(st.st_mode)) {
+ printf("Not a directory: %s\n", path);
+ return 1;
+ }
+ rc = chdir(path);
+ if (rc < 0) {
+ printf("chdir(%s) failed: %s\n", path, strerror(errno));
+ return 1;
+ }
+ setenv("OLDPWD", cwd.characters(), 1);
+ cwd = canonical_path.string();
+ setenv("PWD", cwd.characters(), 1);
+ return 0;
+}
+
+int Shell::builtin_cdh(int argc, const char** argv)
+{
+ int index = -1;
+
+ Core::ArgsParser parser;
+ parser.add_positional_argument(index, "Index of the cd history entry (leave out for a list)", "index", Core::ArgsParser::Required::No);
+
+ if (!parser.parse(argc, const_cast<char**>(argv), false))
+ return 1;
+
+ if (index == -1) {
+ if (cd_history.size() == 0) {
+ printf("cdh: no history available\n");
+ return 0;
+ }
+
+ for (int i = cd_history.size() - 1; i >= 0; --i)
+ printf("%lu: %s\n", cd_history.size() - i, cd_history.at(i).characters());
+ return 0;
+ }
+
+ if (index < 1 || (size_t)index > cd_history.size()) {
+ fprintf(stderr, "cdh: history index out of bounds: %d not in (0, %zu)\n", index, cd_history.size());
+ return 1;
+ }
+
+ const char* path = cd_history.at(cd_history.size() - index).characters();
+ const char* cd_args[] = { "cd", path };
+ return Shell::builtin_cd(2, cd_args);
+}
+
+int Shell::builtin_dirs(int argc, const char** argv)
+{
+ // The first directory in the stack is ALWAYS the current directory
+ directory_stack.at(0) = cwd.characters();
+
+ if (argc == 1) {
+ for (auto& directory : directory_stack) {
+ print_path(directory);
+ fputc(' ', stdout);
+ }
+
+ printf("\n");
+ return 0;
+ }
+
+ bool clear = false;
+ bool print = false;
+ bool number_when_printing = false;
+
+ Vector<const char*> paths;
+
+ Core::ArgsParser parser;
+ parser.add_option(clear, "Clear the directory stack", "clear", 'c');
+ parser.add_option(print, "Print directory entries", "print", 'p');
+ parser.add_option(number_when_printing, "Number the directories in the stack when printing", "number", 'v');
+ parser.add_positional_argument(paths, "Extra paths to put on the stack", "paths", Core::ArgsParser::Required::No);
+
+ if (!parser.parse(argc, const_cast<char**>(argv), false))
+ return 1;
+
+ // -v implies -p
+ print = print || number_when_printing;
+
+ if (clear) {
+ for (size_t i = 1; i < directory_stack.size(); i++)
+ directory_stack.remove(i);
+ }
+
+ for (auto& path : paths)
+ directory_stack.append(path);
+
+ if (print) {
+ auto idx = 0;
+ for (auto& directory : directory_stack) {
+ if (number_when_printing)
+ printf("%d ", idx++);
+ print_path(directory);
+ fputc('\n', stdout);
+ }
+ }
+
+ return 0;
+}
+
+int Shell::builtin_exit(int, const char**)
+{
+ printf("Good-bye!\n");
+ exit(0);
+ return 0;
+}
+
+int Shell::builtin_export(int argc, const char** argv)
+{
+ Vector<const char*> vars;
+
+ Core::ArgsParser parser;
+ parser.add_positional_argument(vars, "List of variable[=value]'s", "values", Core::ArgsParser::Required::No);
+
+ if (!parser.parse(argc, const_cast<char**>(argv), false))
+ return 1;
+
+ if (vars.size() == 0) {
+ for (int i = 0; environ[i]; ++i)
+ puts(environ[i]);
+ return 0;
+ }
+
+ int return_value = 0;
+
+ for (auto& value : vars) {
+ auto parts = String { value }.split_limit('=', 2);
+
+ if (parts.size() == 1) {
+ parts.append("");
+ }
+
+ int setenv_return = setenv(parts[0].characters(), parts[1].characters(), 1);
+
+ if (setenv_return != 0) {
+ perror("setenv");
+ return_value = 1;
+ break;
+ }
+
+ if (parts[0] == "PATH")
+ cache_path();
+ }
+
+ return return_value;
+}
+
+int Shell::builtin_fg(int argc, const char** argv)
+{
+ int job_id = -1;
+
+ Core::ArgsParser parser;
+ parser.add_positional_argument(job_id, "Job id to bring to foreground", "job_id", Core::ArgsParser::Required::No);
+
+ if (!parser.parse(argc, const_cast<char**>(argv), false))
+ return 1;
+
+ if (job_id == -1)
+ job_id = jobs.size() - 1;
+
+ Job* job = nullptr;
+
+ for (auto& entry : jobs) {
+ if (entry.value->job_id() == (u64)job_id) {
+ job = entry.value;
+ break;
+ }
+ }
+ if (!job) {
+ if (job_id == -1) {
+ printf("fg: no current job\n");
+ } else {
+ printf("fg: job with id %d not found\n", job_id);
+ }
+ return 1;
+ }
+
+ job->set_running_in_background(false);
+
+ dbg() << "Resuming " << job->pid() << " (" << job->cmd() << ")";
+ printf("Resuming job %llu - %s\n", job->job_id(), job->cmd().characters());
+
+ if (killpg(job->pgid(), SIGCONT) < 0) {
+ perror("killpg");
+ return 1;
+ }
+
+ int return_value = 0;
+
+ auto current_pid = getpid();
+ auto current_pgid = getpgid(current_pid);
+
+ setpgid(job->pid(), job->pgid());
+ tcsetpgrp(0, job->pgid());
+ SpawnedProcess process { job->cmd(), job->pid() };
+
+ do {
+ if (wait_for_pid(process, true, return_value) == IterationDecision::Break)
+ break;
+ } while (errno == EINTR);
+
+ setpgid(current_pid, current_pgid);
+ tcsetpgrp(0, current_pgid);
+
+ return return_value;
+}
+
+int Shell::builtin_history(int, const char**)
+{
+ for (size_t i = 0; i < editor.history().size(); ++i) {
+ printf("%6zu %s\n", i, editor.history()[i].characters());
+ }
+ return 0;
+}
+
+int Shell::builtin_jobs(int argc, const char** argv)
+{
+ bool list = false, show_pid = false;
+
+ Core::ArgsParser parser;
+ parser.add_option(list, "List all information about jobs", "list", 'l');
+ parser.add_option(show_pid, "Display the PID of the jobs", "pid", 'p');
+
+ if (!parser.parse(argc, const_cast<char**>(argv), false))
+ return 1;
+
+ enum {
+ Basic,
+ OnlyPID,
+ ListAll,
+ } mode { Basic };
+
+ if (show_pid)
+ mode = OnlyPID;
+
+ if (list)
+ mode = ListAll;
+
+ for (auto& job : jobs) {
+ auto pid = job.value->pid();
+ int wstatus;
+ auto rc = waitpid(pid, &wstatus, WNOHANG);
+ if (rc == -1) {
+ perror("waitpid");
+ return 1;
+ }
+ auto status = "running";
+
+ if (rc != 0) {
+ if (WIFEXITED(wstatus))
+ status = "exited";
+
+ if (WIFSTOPPED(wstatus))
+ status = "stopped";
+
+ if (WIFSIGNALED(wstatus))
+ status = "signaled";
+ }
+
+ char background_indicator = '-';
+
+ if (job.value->is_running_in_background())
+ background_indicator = '+';
+
+ switch (mode) {
+ case Basic:
+ printf("[%llu] %c %s %s\n", job.value->job_id(), background_indicator, status, job.value->cmd().characters());
+ break;
+ case OnlyPID:
+ printf("[%llu] %c %d %s %s\n", job.value->job_id(), background_indicator, pid, status, job.value->cmd().characters());
+ break;
+ case ListAll:
+ printf("[%llu] %c %d %d %s %s\n", job.value->job_id(), background_indicator, pid, job.value->pgid(), status, job.value->cmd().characters());
+ break;
+ }
+ }
+
+ return 0;
+}
+
+int Shell::builtin_popd(int argc, const char** argv)
+{
+ if (directory_stack.size() <= 1) {
+ fprintf(stderr, "Shell: popd: directory stack empty\n");
+ return 1;
+ }
+
+ bool should_not_switch = false;
+ String path = directory_stack.take_last();
+
+ Core::ArgsParser parser;
+ parser.add_option(should_not_switch, "Do not switch dirs", "no-switch", 'n');
+
+ if (!parser.parse(argc, const_cast<char**>(argv), false))
+ return 1;
+
+ bool should_switch = !should_not_switch;
+
+ // When no arguments are given, popd removes the top directory from the stack and performs a cd to the new top directory.
+ if (argc == 1) {
+ int rc = chdir(path.characters());
+ if (rc < 0) {
+ fprintf(stderr, "chdir(%s) failed: %s\n", path.characters(), strerror(errno));
+ return 1;
+ }
+
+ cwd = path;
+ return 0;
+ }
+
+ FileSystemPath canonical_path(path.characters());
+ if (!canonical_path.is_valid()) {
+ fprintf(stderr, "FileSystemPath failed to canonicalize '%s'\n", path.characters());
+ return 1;
+ }
+
+ const char* real_path = canonical_path.string().characters();
+
+ struct stat st;
+ int rc = stat(real_path, &st);
+ if (rc < 0) {
+ fprintf(stderr, "stat(%s) failed: %s\n", real_path, strerror(errno));
+ return 1;
+ }
+
+ if (!S_ISDIR(st.st_mode)) {
+ fprintf(stderr, "Not a directory: %s\n", real_path);
+ return 1;
+ }
+
+ if (should_switch) {
+ int rc = chdir(real_path);
+ if (rc < 0) {
+ fprintf(stderr, "chdir(%s) failed: %s\n", real_path, strerror(errno));
+ return 1;
+ }
+
+ cwd = canonical_path.string();
+ }
+
+ return 0;
+}
+
+int Shell::builtin_pushd(int argc, const char** argv)
+{
+ StringBuilder path_builder;
+ bool should_switch = true;
+
+ // From the BASH reference manual: https://www.gnu.org/software/bash/manual/html_node/Directory-Stack-Builtins.html
+ // With no arguments, pushd exchanges the top two directories and makes the new top the current directory.
+ if (argc == 1) {
+ if (directory_stack.size() < 2) {
+ fprintf(stderr, "pushd: no other directory\n");
+ return 1;
+ }
+
+ String dir1 = directory_stack.take_first();
+ String dir2 = directory_stack.take_first();
+ directory_stack.insert(0, dir2);
+ directory_stack.insert(1, dir1);
+
+ int rc = chdir(dir2.characters());
+ if (rc < 0) {
+ fprintf(stderr, "chdir(%s) failed: %s\n", dir2.characters(), strerror(errno));
+ return 1;
+ }
+
+ cwd = dir2;
+
+ return 0;
+ }
+
+ // Let's assume the user's typed in 'pushd <dir>'
+ if (argc == 2) {
+ directory_stack.append(cwd.characters());
+ if (argv[1][0] == '/') {
+ path_builder.append(argv[1]);
+ } else {
+ path_builder.appendf("%s/%s", cwd.characters(), argv[1]);
+ }
+ } else if (argc == 3) {
+ directory_stack.append(cwd.characters());
+ for (int i = 1; i < argc; i++) {
+ const char* arg = argv[i];
+
+ if (arg[0] != '-') {
+ if (arg[0] == '/') {
+ path_builder.append(arg);
+ } else
+ path_builder.appendf("%s/%s", cwd.characters(), arg);
+ }
+
+ if (!strcmp(arg, "-n"))
+ should_switch = false;
+ }
+ }
+
+ FileSystemPath canonical_path(path_builder.to_string());
+ if (!canonical_path.is_valid()) {
+ fprintf(stderr, "FileSystemPath failed to canonicalize '%s'\n", path_builder.to_string().characters());
+ return 1;
+ }
+
+ const char* real_path = canonical_path.string().characters();
+
+ struct stat st;
+ int rc = stat(real_path, &st);
+ if (rc < 0) {
+ fprintf(stderr, "stat(%s) failed: %s\n", real_path, strerror(errno));
+ return 1;
+ }
+
+ if (!S_ISDIR(st.st_mode)) {
+ fprintf(stderr, "Not a directory: %s\n", real_path);
+ return 1;
+ }
+
+ if (should_switch) {
+ int rc = chdir(real_path);
+ if (rc < 0) {
+ fprintf(stderr, "chdir(%s) failed: %s\n", real_path, strerror(errno));
+ return 1;
+ }
+
+ cwd = canonical_path.string();
+ }
+
+ return 0;
+}
+
+int Shell::builtin_pwd(int, const char**)
+{
+ print_path(cwd);
+ fputc('\n', stdout);
+ return 0;
+}
+
+int Shell::builtin_time(int argc, const char** argv)
+{
+ Vector<const char*> args;
+
+ Core::ArgsParser parser;
+ parser.add_positional_argument(args, "Command to execute with arguments", "command", Core::ArgsParser::Required::Yes);
+
+ if (!parser.parse(argc, const_cast<char**>(argv), false))
+ return 1;
+
+ StringBuilder builder;
+ builder.join(' ', args);
+
+ Core::ElapsedTimer timer;
+ timer.start();
+ auto exit_code = run_command(builder.string_view());
+ if (!exit_code.has_value()) {
+ printf("Shell: Incomplete command: %s\n", builder.to_string().characters());
+ exit_code = 1;
+ }
+ printf("Time: %d ms\n", timer.elapsed());
+ return exit_code.value();
+}
+
+int Shell::builtin_umask(int argc, const char** argv)
+{
+ const char* mask_text = nullptr;
+
+ Core::ArgsParser parser;
+ parser.add_positional_argument(mask_text, "New mask (omit to get current mask)", "octal-mask", Core::ArgsParser::Required::No);
+
+ if (!parser.parse(argc, const_cast<char**>(argv), false))
+ return 1;
+
+ if (!mask_text) {
+ mode_t old_mask = umask(0);
+ printf("%#o\n", old_mask);
+ umask(old_mask);
+ return 0;
+ }
+
+ unsigned mask;
+ int matches = sscanf(mask_text, "%o", &mask);
+ if (matches == 1) {
+ umask(mask);
+ return 0;
+ }
+
+ return 0;
+}
+
+int Shell::builtin_unset(int argc, const char** argv)
+{
+ Vector<const char*> vars;
+
+ Core::ArgsParser parser;
+ parser.add_positional_argument(vars, "List of variables", "variables", Core::ArgsParser::Required::Yes);
+
+ if (!parser.parse(argc, const_cast<char**>(argv), false))
+ return 1;
+
+ for (auto& value : vars)
+ unsetenv(value);
+
+ return 0;
+}
+
+bool Shell::run_builtin(int argc, const char** argv, int& retval)
+{
+ if (argc == 0)
+ return false;
+
+ StringView name { argv[0] };
+
+#define __ENUMERATE_SHELL_BUILTIN(builtin) \
+ if (name == #builtin) { \
+ retval = builtin_##builtin(argc, argv); \
+ return true; \
+ }
+
+ ENUMERATE_SHELL_BUILTINS();
+
+#undef __ENUMERATE_SHELL_BUILTIN
+
+ return false;
+}
+
+String Shell::expand_tilde(const String& expression)
+{
+ ASSERT(expression.starts_with('~'));
+
+ StringBuilder login_name;
+ size_t first_slash_index = expression.length();
+ for (size_t i = 1; i < expression.length(); ++i) {
+ if (expression[i] == '/') {
+ first_slash_index = i;
+ break;
+ }
+ login_name.append(expression[i]);
+ }
+
+ StringBuilder path;
+ for (size_t i = first_slash_index; i < expression.length(); ++i)
+ path.append(expression[i]);
+
+ if (login_name.is_empty()) {
+ const char* home = getenv("HOME");
+ if (!home) {
+ auto passwd = getpwuid(getuid());
+ ASSERT(passwd && passwd->pw_dir);
+ return String::format("%s/%s", passwd->pw_dir, path.to_string().characters());
+ }
+ return String::format("%s/%s", home, path.to_string().characters());
+ }
+
+ auto passwd = getpwnam(login_name.to_string().characters());
+ if (!passwd)
+ return expression;
+ ASSERT(passwd->pw_dir);
+
+ return String::format("%s/%s", passwd->pw_dir, path.to_string().characters());
+}
+
+bool Shell::is_glob(const StringView& s)
+{
+ for (size_t i = 0; i < s.length(); i++) {
+ char c = s.characters_without_null_termination()[i];
+ if (c == '*' || c == '?')
+ return true;
+ }
+ return false;
+}
+
+Vector<StringView> Shell::split_path(const StringView& path)
+{
+ Vector<StringView> parts;
+
+ size_t substart = 0;
+ for (size_t i = 0; i < path.length(); i++) {
+ char ch = path.characters_without_null_termination()[i];
+ if (ch != '/')
+ continue;
+ size_t sublen = i - substart;
+ if (sublen != 0)
+ parts.append(path.substring_view(substart, sublen));
+ parts.append(path.substring_view(i, 1));
+ substart = i + 1;
+ }
+
+ size_t taillen = path.length() - substart;
+ if (taillen != 0)
+ parts.append(path.substring_view(substart, taillen));
+
+ return parts;
+}
+
+Vector<String> Shell::expand_globs(const StringView& path, const StringView& base)
+{
+ auto parts = split_path(path);
+
+ StringBuilder builder;
+ builder.append(base);
+ Vector<String> res;
+
+ for (size_t i = 0; i < parts.size(); ++i) {
+ auto& part = parts[i];
+ if (!is_glob(part)) {
+ builder.append(part);
+ continue;
+ }
+
+ // Found a glob.
+ String new_base = builder.to_string();
+ StringView new_base_v = new_base;
+ if (new_base_v.is_empty())
+ new_base_v = ".";
+ Core::DirIterator di(new_base_v, Core::DirIterator::SkipParentAndBaseDir);
+
+ if (di.has_error()) {
+ return res;
+ }
+
+ while (di.has_next()) {
+ String name = di.next_path();
+
+ // Dotfiles have to be explicitly requested
+ if (name[0] == '.' && part[0] != '.')
+ continue;
+
+ if (name.matches(part, CaseSensitivity::CaseSensitive)) {
+
+ StringBuilder nested_base;
+ nested_base.append(new_base);
+ nested_base.append(name);
+
+ StringView remaining_path = path.substring_view_starting_after_substring(part);
+ Vector<String> nested_res = expand_globs(remaining_path, nested_base.to_string());
+ for (auto& s : nested_res)
+ res.append(s);
+ }
+ }
+ return res;
+ }
+
+ // Found no globs.
+ String new_path = builder.to_string();
+ if (access(new_path.characters(), F_OK) == 0)
+ res.append(new_path);
+ return res;
+}
+
+Vector<String> Shell::expand_parameters(const StringView& param) const
+{
+ if (!param.starts_with('$'))
+ return { param };
+
+ String variable_name = String(param.substring_view(1, param.length() - 1));
+ if (variable_name == "?")
+ return { String::number(last_return_code) };
+ else if (variable_name == "$")
+ return { String::number(getpid()) };
+
+ char* env_value = getenv(variable_name.characters());
+ if (env_value == nullptr)
+ return { "" };
+
+ Vector<String> res;
+ String str_env_value = String(env_value);
+ const auto& split_text = str_env_value.split_view(' ');
+ for (auto& part : split_text)
+ res.append(part);
+ return res;
+}
+
+Vector<String> Shell::process_arguments(const Vector<Token>& args)
+{
+ Vector<String> argv_string;
+ for (auto& arg : args) {
+ if (arg.type == Token::Comment)
+ continue;
+
+ // This will return the text passed in if it wasn't a variable
+ // This lets us just loop over its values
+ auto expanded_parameters = expand_parameters(arg.text);
+
+ for (auto& exp_arg : expanded_parameters) {
+ if (exp_arg.starts_with('~'))
+ exp_arg = expand_tilde(exp_arg);
+
+ auto expanded_globs = expand_globs(exp_arg, "");
+ for (auto& path : expanded_globs)
+ argv_string.append(path);
+
+ if (expanded_globs.is_empty())
+ argv_string.append(exp_arg);
+ }
+ }
+
+ return argv_string;
+}
+
+ContinuationRequest Shell::is_complete(const Vector<Command>& commands)
+{
+ // check if the last command ends with a pipe, or an unterminated string
+ auto& last_command = commands.last();
+ auto& subcommands = last_command.subcommands;
+ if (subcommands.size() == 0)
+ return ContinuationRequest::Nothing;
+
+ auto& last_subcommand = subcommands.last();
+
+ if (!last_subcommand.redirections.find([](auto& redirection) { return redirection.type == Redirection::Pipe; }).is_end())
+ return ContinuationRequest::Pipe;
+
+ if (!last_subcommand.args.find([](auto& token) { return token.type == Token::UnterminatedSingleQuoted; }).is_end())
+ return ContinuationRequest::SingleQuotedString;
+
+ if (!last_subcommand.args.find([](auto& token) { return token.type == Token::UnterminatedDoubleQuoted; }).is_end())
+ return ContinuationRequest::DoubleQuotedString;
+
+ return ContinuationRequest::Nothing;
+}
+
+IterationDecision Shell::wait_for_pid(const Shell::SpawnedProcess& process, bool is_first_command_in_chain, int& return_value)
+{
+ // Disable signal handler for the first command, as we actively wait for it
+ sighandler_t chld_handler = nullptr;
+ if (is_first_command_in_chain) {
+ chld_handler = signal(SIGCHLD, nullptr);
+ dbg() << "Waiting for " << process.name;
+ }
+
+ int wstatus = 0;
+ int rc = waitpid(process.pid, &wstatus, WSTOPPED);
+ auto errno_save = errno;
+
+ if (is_first_command_in_chain) {
+ signal(SIGCHLD, chld_handler);
+ dbg() << process.name << " is probably dead now (" << rc << ", " << strerror(errno_save) << ") -> exited " << WIFEXITED(wstatus) << " stopped " << WIFSTOPPED(wstatus);
+ }
+
+ errno = errno_save;
+ if (rc < 0 && errno != EINTR) {
+ if (errno != ECHILD)
+ perror("waitpid");
+ return IterationDecision::Break;
+ }
+
+ const Job* job = nullptr;
+ u64 job_id = 0;
+ auto maybe_job = jobs.get(process.pid);
+ if (maybe_job.has_value()) {
+ job = maybe_job.value();
+ job_id = job->job_id();
+ }
+
+ if (WIFEXITED(wstatus)) {
+ if (WEXITSTATUS(wstatus) != 0)
+ dbg() << "Shell: " << process.name << ":" << process.pid << " exited with status " << WEXITSTATUS(wstatus);
+
+ if (is_first_command_in_chain)
+ return_value = WEXITSTATUS(wstatus);
+
+ if (job) {
+ auto* mutable_job = const_cast<Job*>(job);
+ mutable_job->set_has_exit(return_value);
+ Core::EventLoop::current().post_event(*this, make<Core::CustomEvent>(ChildExited, mutable_job));
+ }
+ return IterationDecision::Break;
+ }
+
+ if (WIFSTOPPED(wstatus)) {
+ fprintf(stderr, "Shell: [%llu] %s(%d) %s\n", job_id, process.name.characters(), process.pid, strsignal(WSTOPSIG(wstatus)));
+ return IterationDecision::Continue;
+ }
+
+ if (WIFSIGNALED(wstatus)) {
+ printf("Shell: [%llu] %s(%d) exited due to signal '%s'\n", job_id, process.name.characters(), process.pid, strsignal(WTERMSIG(wstatus)));
+ } else {
+ printf("Shell: [%llu] %s(%d) exited abnormally\n", job_id, process.name.characters(), process.pid);
+ }
+ if (job) {
+ auto* mutable_job = const_cast<Job*>(job);
+ mutable_job->set_has_exit(-1);
+ Core::EventLoop::current().post_event(*this, make<Core::CustomEvent>(ChildExited, mutable_job));
+ }
+ return IterationDecision::Break;
+}
+
+ExitCodeOrContinuationRequest Shell::run_command(const StringView& cmd)
+{
+ if (cmd.is_empty())
+ return 0;
+
+ if (cmd.starts_with("#"))
+ return 0;
+
+ auto commands = Parser(cmd).parse();
+
+ if (!commands.size())
+ return 1;
+
+ auto needs_more = is_complete(commands);
+ if (needs_more != ExitCodeOrContinuationRequest::Nothing)
+ return needs_more;
+
+#ifdef SH_DEBUG
+ for (auto& command : commands) {
+ for (size_t i = 0; i < command.subcommands.size(); ++i) {
+ for (size_t j = 0; j < i; ++j)
+ dbgprintf(" ");
+ for (auto& arg : command.subcommands[i].args) {
+ switch (arg.type) {
+ case Token::Bare:
+ dbgprintf("<%s> ", arg.text.characters());
+ break;
+ case Token::SingleQuoted:
+ dbgprintf("'<%s>' ", arg.text.characters());
+ break;
+ case Token::DoubleQuoted:
+ dbgprintf("\"<%s>\" ", arg.text.characters());
+ break;
+ case Token::UnterminatedSingleQuoted:
+ dbgprintf("\'<%s> ", arg.text.characters());
+ break;
+ case Token::UnterminatedDoubleQuoted:
+ dbgprintf("\"<%s> ", arg.text.characters());
+ break;
+ case Token::Special:
+ dbgprintf("<%s> ", arg.text.characters());
+ break;
+ case Token::Comment:
+ dbgprintf("<%s> ", arg.text.characters());
+ break;
+ }
+ }
+ dbgprintf("\n");
+ for (auto& redirecton : command.subcommands[i].redirections) {
+ for (size_t j = 0; j < i; ++j)
+ dbgprintf(" ");
+ dbgprintf(" ");
+ switch (redirecton.type) {
+ case Redirection::Pipe:
+ dbgprintf("Pipe\n");
+ break;
+ case Redirection::FileRead:
+ dbgprintf("fd:%d = FileRead: %s\n", redirecton.fd, redirecton.path.characters());
+ break;
+ case Redirection::FileWrite:
+ dbgprintf("fd:%d = FileWrite: %s\n", redirecton.fd, redirecton.path.characters());
+ break;
+ case Redirection::FileWriteAppend:
+ dbgprintf("fd:%d = FileWriteAppend: %s\n", redirecton.fd, redirecton.path.characters());
+ break;
+ default:
+ break;
+ }
+ }
+ }
+ dbgprintf("\n");
+ }
+#endif
+
+ struct termios trm;
+ tcgetattr(0, &trm);
+
+ int return_value = 0;
+
+ for (auto& command : commands) {
+ if (command.subcommands.is_empty())
+ continue;
+
+ FileDescriptionCollector fds;
+
+ for (size_t i = 0; i < command.subcommands.size(); ++i) {
+ auto& subcommand = command.subcommands[i];
+ for (auto& redirection : subcommand.redirections) {
+ switch (redirection.type) {
+ case Redirection::Pipe: {
+ int pipefd[2];
+ int rc = pipe(pipefd);
+ if (rc < 0) {
+ perror("pipe");
+ return 1;
+ }
+ subcommand.rewirings.append({ STDOUT_FILENO, pipefd[1] });
+ auto& next_command = command.subcommands[i + 1];
+ next_command.rewirings.append({ STDIN_FILENO, pipefd[0] });
+ fds.add(pipefd[0]);
+ fds.add(pipefd[1]);
+ break;
+ }
+ case Redirection::FileWriteAppend: {
+ int fd = open(redirection.path.characters(), O_WRONLY | O_CREAT | O_APPEND, 0666);
+ if (fd < 0) {
+ perror("open");
+ return 1;
+ }
+ subcommand.rewirings.append({ redirection.fd, fd });
+ fds.add(fd);
+ break;
+ }
+ case Redirection::FileWrite: {
+ int fd = open(redirection.path.characters(), O_WRONLY | O_CREAT | O_TRUNC, 0666);
+ if (fd < 0) {
+ perror("open");
+ return 1;
+ }
+ subcommand.rewirings.append({ redirection.fd, fd });
+ fds.add(fd);
+ break;
+ }
+ case Redirection::FileRead: {
+ int fd = open(redirection.path.characters(), O_RDONLY);
+ if (fd < 0) {
+ perror("open");
+ return 1;
+ }
+ subcommand.rewirings.append({ redirection.fd, fd });
+ fds.add(fd);
+ break;
+ }
+ }
+ }
+ }
+
+ Vector<SpawnedProcess> children;
+
+ for (size_t i = 0; i < command.subcommands.size(); ++i) {
+ auto& subcommand = command.subcommands[i];
+ Vector<String> argv_string = process_arguments(subcommand.args);
+ Vector<const char*> argv;
+ argv.ensure_capacity(argv_string.size());
+ for (const auto& s : argv_string) {
+ argv.append(s.characters());
+ }
+ argv.append(nullptr);
+
+#ifdef SH_DEBUG
+ for (auto& arg : argv) {
+ dbgprintf("<%s> ", arg);
+ }
+ dbgprintf("\n");
+#endif
+
+ int retval = 0;
+ if (run_builtin(argv.size() - 1, argv.data(), retval))
+ return retval;
+
+ pid_t child = fork();
+ if (!child) {
+ setpgid(0, 0);
+ tcsetpgrp(0, getpid());
+ tcsetattr(0, TCSANOW, &default_termios);
+ for (auto& rewiring : subcommand.rewirings) {
+#ifdef SH_DEBUG
+ dbgprintf("in %s<%d>, dup2(%d, %d)\n", argv[0], getpid(), rewiring.rewire_fd, rewiring.fd);
+#endif
+ int rc = dup2(rewiring.rewire_fd, rewiring.fd);
+ if (rc < 0) {
+ perror("dup2");
+ return 1;
+ }
+ }
+
+ fds.collect();
+
+ int rc = execvp(argv[0], const_cast<char* const*>(argv.data()));
+ if (rc < 0) {
+ if (errno == ENOENT) {
+ int shebang_fd = open(argv[0], O_RDONLY);
+ auto close_argv = ScopeGuard([shebang_fd]() { if (shebang_fd >= 0) close(shebang_fd); });
+ char shebang[256] {};
+ ssize_t num_read = -1;
+ if ((shebang_fd >= 0) && ((num_read = read(shebang_fd, shebang, sizeof(shebang))) >= 2) && (StringView(shebang).starts_with("#!"))) {
+ StringView shebang_path_view(&shebang[2], num_read - 2);
+ Optional<size_t> newline_pos = shebang_path_view.find_first_of("\n\r");
+ shebang[newline_pos.has_value() ? newline_pos.value() : num_read] = '\0';
+ fprintf(stderr, "%s: Invalid interpreter \"%s\": %s\n", argv[0], &shebang[2], strerror(ENOENT));
+ } else
+ fprintf(stderr, "%s: Command not found.\n", argv[0]);
+ } else {
+ int saved_errno = errno;
+ struct stat st;
+ if (stat(argv[0], &st) == 0 && S_ISDIR(st.st_mode)) {
+ fprintf(stderr, "Shell: %s: Is a directory\n", argv[0]);
+ _exit(126);
+ }
+ fprintf(stderr, "execvp(%s): %s\n", argv[0], strerror(saved_errno));
+ }
+ _exit(126);
+ }
+ ASSERT_NOT_REACHED();
+ }
+ children.append({ argv[0], child });
+
+ StringBuilder cmd;
+ cmd.join(" ", argv_string);
+
+ auto job = make<Job>(child, (unsigned)child, cmd.build(), jobs.size());
+ jobs.set((u64)child, move(job));
+ }
+
+#ifdef SH_DEBUG
+ dbgprintf("Closing fds in shell process:\n");
+#endif
+ fds.collect();
+
+#ifdef SH_DEBUG
+ dbgprintf("Now we gotta wait on children:\n");
+ for (auto& child : children)
+ dbgprintf(" %d (%s)\n", child.pid, child.name.characters());
+#endif
+
+ for (size_t i = 0; i < children.size(); ++i) {
+ auto& child = children[i];
+ dbg() << "Now waiting for " << child.name << " (" << child.pid << ")";
+ do {
+ if (wait_for_pid(child, i != children.size() - 1, return_value) == IterationDecision::Break)
+ break;
+ } while (errno == EINTR);
+ }
+ }
+
+ last_return_code = return_value;
+
+ // FIXME: Should I really have to tcsetpgrp() after my child has exited?
+ // Is the terminal controlling pgrp really still the PGID of the dead process?
+ tcsetpgrp(0, getpid());
+ tcsetattr(0, TCSANOW, &trm);
+ return return_value;
+}
+
+String Shell::get_history_path()
+{
+ StringBuilder builder;
+ builder.append(home);
+ builder.append("/.history");
+ return builder.to_string();
+}
+
+void Shell::load_history()
+{
+ auto history_file = Core::File::construct(get_history_path());
+ if (!history_file->open(Core::IODevice::ReadOnly))
+ return;
+ while (history_file->can_read_line()) {
+ auto b = history_file->read_line(1024);
+ // skip the newline and terminating bytes
+ editor.add_to_history(String(reinterpret_cast<const char*>(b.data()), b.size() - 2));
+ }
+}
+
+void Shell::save_history()
+{
+ auto file_or_error = Core::File::open(get_history_path(), Core::IODevice::WriteOnly, 0600);
+ if (file_or_error.is_error())
+ return;
+ auto& file = *file_or_error.value();
+ for (const auto& line : editor.history()) {
+ file.write(line);
+ file.write("\n");
+ }
+}
+
+String Shell::escape_token(const String& token)
+{
+ StringBuilder builder;
+
+ for (auto c : token) {
+ switch (c) {
+ case '\'':
+ case '"':
+ case '$':
+ case '|':
+ case '>':
+ case '<':
+ case '&':
+ case '\\':
+ case ' ':
+ builder.append('\\');
+ break;
+ default:
+ break;
+ }
+ builder.append(c);
+ }
+
+ return builder.build();
+}
+
+String Shell::unescape_token(const String& token)
+{
+ StringBuilder builder;
+
+ enum {
+ Free,
+ Escaped
+ } state { Free };
+
+ for (auto c : token) {
+ switch (state) {
+ case Escaped:
+ builder.append(c);
+ state = Free;
+ break;
+ case Free:
+ if (c == '\\')
+ state = Escaped;
+ else
+ builder.append(c);
+ break;
+ }
+ }
+
+ if (state == Escaped)
+ builder.append('\\');
+
+ return builder.build();
+}
+
+void Shell::cache_path()
+{
+ if (!cached_path.is_empty())
+ cached_path.clear_with_capacity();
+
+ String path = getenv("PATH");
+ if (path.is_empty())
+ return;
+
+ auto directories = path.split(':');
+ for (const auto& directory : directories) {
+ Core::DirIterator programs(directory.characters(), Core::DirIterator::SkipDots);
+ while (programs.has_next()) {
+ auto program = programs.next_path();
+ String program_path = String::format("%s/%s", directory.characters(), program.characters());
+ if (access(program_path.characters(), X_OK) == 0)
+ cached_path.append(escape_token(program.characters()));
+ }
+ }
+
+ quick_sort(cached_path);
+}
+
+void Shell::highlight(Line::Editor&) const
+{
+ StringBuilder builder;
+ if (m_should_continue == ExitCodeOrContinuationRequest::DoubleQuotedString) {
+ builder.append('"');
+ }
+ if (m_should_continue == ExitCodeOrContinuationRequest::SingleQuotedString) {
+ builder.append('\'');
+ }
+ builder.append(StringView { editor.buffer().data(), editor.buffer().size() });
+ auto commands = Parser { builder.string_view() }.parse();
+ auto first_command { true };
+ for (auto& command : commands) {
+ for (auto& subcommand : command.subcommands) {
+ auto first { true };
+ for (auto& arg : subcommand.args) {
+ auto start = arg.end - arg.length;
+
+ if (arg.type == Token::Comment) {
+ editor.stylize({ start, arg.end }, { Line::Style::Foreground(150, 150, 150) }); // light gray
+ continue;
+ }
+
+ if (m_should_continue == ExitCodeOrContinuationRequest::DoubleQuotedString || m_should_continue == ExitCodeOrContinuationRequest::SingleQuotedString) {
+ if (!first_command)
+ --start;
+ --arg.end;
+ }
+ if (first) {
+ first = false;
+ // only treat this as a command name if we're not continuing strings
+ if (!first_command || (m_should_continue == ExitCodeOrContinuationRequest::Nothing || m_should_continue == ExitCodeOrContinuationRequest::Pipe)) {
+ editor.stylize({ start, arg.end }, { Line::Style::Bold });
+ first_command = false;
+ continue;
+ }
+ first_command = false;
+ }
+
+ if (arg.type == Token::SingleQuoted || arg.type == Token::UnterminatedSingleQuoted) {
+ editor.stylize({ start - 1, arg.end + (arg.type != Token::UnterminatedSingleQuoted) }, { Line::Style::Foreground(Line::Style::XtermColor::Yellow) });
+ continue;
+ }
+
+ if (arg.type == Token::DoubleQuoted || arg.type == Token::UnterminatedDoubleQuoted) {
+ editor.stylize({ start - 1, arg.end + (arg.type != Token::UnterminatedDoubleQuoted) }, { Line::Style::Foreground(Line::Style::XtermColor::Yellow) });
+ if constexpr (HighlightVariablesInsideStrings)
+ goto highlight_variables;
+ else
+ continue;
+ }
+
+ if (is_glob(arg.text)) {
+ editor.stylize({ start, arg.end }, { Line::Style::Foreground(59, 142, 234) }); // bright-ish blue
+ continue;
+ }
+
+ if (arg.text.starts_with("--")) {
+ if (arg.length == 2)
+ editor.stylize({ start, arg.end }, { Line::Style::Foreground(Line::Style::XtermColor::Green) });
+ else
+ editor.stylize({ start, arg.end }, { Line::Style::Foreground(Line::Style::XtermColor::Cyan) });
+ } else if (arg.text.starts_with("-") && arg.length > 1) {
+ editor.stylize({ start, arg.end }, { Line::Style::Foreground(Line::Style::XtermColor::Cyan) });
+ }
+
+ highlight_variables:;
+
+ size_t slice_index = 0;
+ Optional<size_t> maybe_index;
+ while (slice_index < arg.length) {
+ maybe_index = arg.text.substring_view(slice_index, arg.length - slice_index).find_first_of('$');
+ if (!maybe_index.has_value())
+ break;
+ auto index = maybe_index.value() + 1;
+ auto end_index = index;
+ if (index >= arg.length)
+ break;
+ for (; end_index < arg.length; ++end_index) {
+ if (!is_word_character(arg.text[end_index]))
+ break;
+ }
+ editor.stylize({ index + start - 1, end_index + start }, { Line::Style::Foreground(214, 112, 214) });
+ slice_index = end_index + 1;
+ }
+ }
+ }
+ }
+}
+
+Vector<Line::CompletionSuggestion> Shell::complete_first(const String& token_to_complete)
+{
+ auto token = unescape_token(token_to_complete);
+
+ auto match = binary_search(cached_path.data(), cached_path.size(), token, [](const String& token, const String& program) -> int {
+ return strncmp(token.characters(), program.characters(), token.length());
+ });
+
+ if (!match) {
+ // There is no executable in the $PATH starting with $token
+ // Suggest local executables and directories
+ String path;
+ Vector<Line::CompletionSuggestion> local_suggestions;
+ bool suggest_executables = true;
+
+ ssize_t last_slash = token.length() - 1;
+ while (last_slash >= 0 && token[last_slash] != '/')
+ --last_slash;
+
+ if (last_slash >= 0) {
+ // Split on the last slash. We'll use the first part as the directory
+ // to search and the second part as the token to complete.
+ path = token.substring(0, last_slash + 1);
+ if (path[0] != '/')
+ path = String::format("%s/%s", cwd.characters(), path.characters());
+ path = canonicalized_path(path);
+ token = token.substring(last_slash + 1, token.length() - last_slash - 1);
+ } else {
+ // We have no slashes, so the directory to search is the current
+ // directory and the token to complete is just the original token.
+ // In this case, do not suggest executables but directories only.
+ path = cwd;
+ suggest_executables = false;
+ }
+
+ // the invariant part of the token is actually just the last segment
+ // e. in `cd /foo/bar', 'bar' is the invariant
+ // since we are not suggesting anything starting with
+ // `/foo/', but rather just `bar...'
+ editor.suggest(escape_token(token).length(), 0);
+
+ // only suggest dot-files if path starts with a dot
+ Core::DirIterator files(path,
+ token.starts_with('.') ? Core::DirIterator::SkipParentAndBaseDir : Core::DirIterator::SkipDots);
+
+ while (files.has_next()) {
+ auto file = files.next_path();
+ auto trivia = " ";
+ if (file.starts_with(token)) {
+ String file_path = String::format("%s/%s", path.characters(), file.characters());
+ struct stat program_status;
+ int stat_error = stat(file_path.characters(), &program_status);
+ if (stat_error)
+ continue;
+ if (access(file_path.characters(), X_OK) != 0)
+ continue;
+ if (S_ISDIR(program_status.st_mode)) {
+ if (!suggest_executables)
+ continue;
+ else
+ trivia = "/";
+ }
+
+ local_suggestions.append({ escape_token(file), trivia });
+ }
+ }
+
+ return local_suggestions;
+ }
+
+ String completion = *match;
+ Vector<Line::CompletionSuggestion> suggestions;
+
+ // Now that we have a program name starting with our token, we look at
+ // other program names starting with our token and cut off any mismatching
+ // characters.
+
+ int index = match - cached_path.data();
+ for (int i = index - 1; i >= 0 && cached_path[i].starts_with(token); --i) {
+ suggestions.append({ cached_path[i], " " });
+ }
+ for (size_t i = index + 1; i < cached_path.size() && cached_path[i].starts_with(token); ++i) {
+ suggestions.append({ cached_path[i], " " });
+ }
+ suggestions.append({ cached_path[index], " " });
+
+ editor.suggest(escape_token(token).length(), 0);
+
+ return suggestions;
+}
+
+Vector<Line::CompletionSuggestion> Shell::complete_other(const String& token_to_complete)
+{
+ auto token = unescape_token(token_to_complete);
+ String path;
+ Vector<Line::CompletionSuggestion> suggestions;
+
+ ssize_t last_slash = token.length() - 1;
+ while (last_slash >= 0 && token[last_slash] != '/')
+ --last_slash;
+
+ if (last_slash >= 0) {
+ // Split on the last slash. We'll use the first part as the directory
+ // to search and the second part as the token to complete.
+ path = token.substring(0, last_slash + 1);
+ if (path[0] != '/')
+ path = String::format("%s/%s", cwd.characters(), path.characters());
+ path = canonicalized_path(path);
+ token = token.substring(last_slash + 1, token.length() - last_slash - 1);
+ } else {
+ // We have no slashes, so the directory to search is the current
+ // directory and the token to complete is just the original token.
+ path = cwd;
+ }
+
+ // the invariant part of the token is actually just the last segment
+ // e. in `cd /foo/bar', 'bar' is the invariant
+ // since we are not suggesting anything starting with
+ // `/foo/', but rather just `bar...'
+ editor.suggest(escape_token(token).length(), 0);
+
+ // only suggest dot-files if path starts with a dot
+ Core::DirIterator files(path,
+ token.starts_with('.') ? Core::DirIterator::SkipParentAndBaseDir : Core::DirIterator::SkipDots);
+
+ while (files.has_next()) {
+ auto file = files.next_path();
+ if (file.starts_with(token)) {
+ struct stat program_status;
+ String file_path = String::format("%s/%s", path.characters(), file.characters());
+ int stat_error = stat(file_path.characters(), &program_status);
+ if (!stat_error) {
+ if (S_ISDIR(program_status.st_mode))
+ suggestions.append({ escape_token(file), "/" });
+ else
+ suggestions.append({ escape_token(file), " " });
+ }
+ }
+ }
+
+ return suggestions;
+}
+
+void Shell::read_single_line()
+{
+ auto line = editor.get_line(prompt());
+
+ if (m_should_break_current_command) {
+ m_complete_line_builder.clear();
+ m_should_continue = ContinuationRequest::Nothing;
+ m_should_break_current_command = false;
+ return;
+ }
+
+ if (line.is_empty())
+ return;
+
+ // FIXME: This might be a bit counter-intuitive, since we put nothing
+ // between the two lines, even though the user has pressed enter
+ // but since the LineEditor cannot yet handle literal newlines
+ // inside the text, we opt to do this the wrong way (for the time being)
+ m_complete_line_builder.append(line);
+
+ auto complete_or_exit_code = run_command(m_complete_line_builder.string_view());
+ m_should_continue = complete_or_exit_code.continuation;
+
+ if (!complete_or_exit_code.has_value())
+ return;
+
+ editor.add_to_history(m_complete_line_builder.build());
+ m_complete_line_builder.clear();
+}
+
+void Shell::custom_event(Core::CustomEvent& event)
+{
+ if (event.custom_type() == ReadLine) {
+ read_single_line();
+ Core::EventLoop::current().post_event(*this, make<Core::CustomEvent>(ShellEventType::ReadLine));
+ return;
+ }
+
+ if (event.custom_type() == ChildExited) {
+ auto* job_ptr = event.data();
+ if (job_ptr) {
+ auto& job = *(Job*)job_ptr;
+ if (job.is_running_in_background())
+ fprintf(stderr, "Shell: Job %d(%s) exited\n", job.pid(), job.cmd().characters());
+ jobs.remove(job.pid());
+ }
+ return;
+ }
+
+ event.ignore();
+}
+
+Shell::Shell()
+{
+ uid = getuid();
+ tcsetpgrp(0, getpgrp());
+
+ int rc = gethostname(hostname, Shell::HostNameSize);
+ if (rc < 0)
+ perror("gethostname");
+ rc = ttyname_r(0, ttyname, Shell::TTYNameSize);
+ if (rc < 0)
+ perror("ttyname_r");
+
+ {
+ auto* cwd = getcwd(nullptr, 0);
+ this->cwd = cwd;
+ setenv("PWD", cwd, 1);
+ free(cwd);
+ }
+
+ {
+ auto* pw = getpwuid(getuid());
+ if (pw) {
+ username = pw->pw_name;
+ home = pw->pw_dir;
+ setenv("HOME", pw->pw_dir, 1);
+ }
+ endpwent();
+ }
+
+ directory_stack.append(cwd);
+ load_history();
+ cache_path();
+}
+
+Shell::~Shell()
+{
+ save_history();
+}
diff --git a/Shell/Shell.h b/Shell/Shell.h
new file mode 100644
index 0000000000..ea9ff440cf
--- /dev/null
+++ b/Shell/Shell.h
@@ -0,0 +1,185 @@
+/*
+ * Copyright (c) 2020, The SerenityOS developers.
+ * All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions are met:
+ *
+ * 1. Redistributions of source code must retain the above copyright notice, this
+ * list of conditions and the following disclaimer.
+ *
+ * 2. Redistributions in binary form must reproduce the above copyright notice,
+ * this list of conditions and the following disclaimer in the documentation
+ * and/or other materials provided with the distribution.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
+ * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+ * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+ * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
+ * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+ * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
+ * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
+ * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
+ * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+ * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ */
+
+#pragma once
+
+#include "Job.h"
+#include "Parser.h"
+#include <AK/CircularQueue.h>
+#include <AK/HashMap.h>
+#include <AK/String.h>
+#include <AK/StringBuilder.h>
+#include <AK/Types.h>
+#include <AK/Vector.h>
+#include <LibCore/Notifier.h>
+#include <LibCore/Object.h>
+#include <LibLine/Editor.h>
+#include <termios.h>
+
+struct ExitCodeOrContinuationRequest {
+ enum ContinuationRequest {
+ Nothing,
+ Pipe,
+ DoubleQuotedString,
+ SingleQuotedString,
+ };
+
+ ExitCodeOrContinuationRequest(ContinuationRequest continuation)
+ : continuation(continuation)
+ {
+ }
+
+ ExitCodeOrContinuationRequest(int exit)
+ : exit_code(exit)
+ {
+ }
+
+ bool has_value() const { return exit_code.has_value(); }
+ int value() const
+ {
+ ASSERT(has_value());
+ return exit_code.value();
+ }
+
+ Optional<int> exit_code;
+ ContinuationRequest continuation { Nothing };
+};
+
+using ContinuationRequest = ExitCodeOrContinuationRequest::ContinuationRequest;
+
+#define ENUMERATE_SHELL_BUILTINS() \
+ __ENUMERATE_SHELL_BUILTIN(cd) \
+ __ENUMERATE_SHELL_BUILTIN(cdh) \
+ __ENUMERATE_SHELL_BUILTIN(pwd) \
+ __ENUMERATE_SHELL_BUILTIN(exit) \
+ __ENUMERATE_SHELL_BUILTIN(export) \
+ __ENUMERATE_SHELL_BUILTIN(unset) \
+ __ENUMERATE_SHELL_BUILTIN(history) \
+ __ENUMERATE_SHELL_BUILTIN(umask) \
+ __ENUMERATE_SHELL_BUILTIN(dirs) \
+ __ENUMERATE_SHELL_BUILTIN(pushd) \
+ __ENUMERATE_SHELL_BUILTIN(popd) \
+ __ENUMERATE_SHELL_BUILTIN(time) \
+ __ENUMERATE_SHELL_BUILTIN(jobs) \
+ __ENUMERATE_SHELL_BUILTIN(fg) \
+ __ENUMERATE_SHELL_BUILTIN(bg)
+
+class Shell;
+
+class Shell : public Core::Object {
+ C_OBJECT(Shell);
+
+public:
+ ExitCodeOrContinuationRequest run_command(const StringView&);
+ bool run_builtin(int argc, const char** argv, int& retval);
+ String prompt() const;
+
+ static String expand_tilde(const String&);
+ static Vector<String> expand_globs(const StringView& path, const StringView& base);
+ Vector<String> expand_parameters(const StringView&) const;
+
+ static String escape_token(const String& token);
+ static String unescape_token(const String& token);
+
+ static bool is_glob(const StringView&);
+ static Vector<StringView> split_path(const StringView&);
+
+ Vector<String> process_arguments(const Vector<Token>&);
+
+ static ContinuationRequest is_complete(const Vector<Command>&);
+
+ void highlight(Line::Editor&) const;
+ Vector<Line::CompletionSuggestion> complete_first(const String&);
+ Vector<Line::CompletionSuggestion> complete_other(const String&);
+
+ String get_history_path();
+ void load_history();
+ void save_history();
+ void print_path(const String& path);
+
+ bool should_read_more() const { return m_should_continue != ContinuationRequest::Nothing; }
+ void finish_command() { m_should_break_current_command = true; }
+
+ void read_single_line();
+
+ struct termios termios;
+ struct termios default_termios;
+ bool was_interrupted { false };
+ bool was_resized { false };
+
+ String cwd;
+ String username;
+ String home;
+
+ constexpr static auto TTYNameSize = 32;
+ constexpr static auto HostNameSize = 64;
+
+ char ttyname[TTYNameSize];
+ char hostname[HostNameSize];
+
+ uid_t uid;
+ int last_return_code { 0 };
+ Vector<String> directory_stack;
+ CircularQueue<String, 8> cd_history; // FIXME: have a configurable cd history length
+ HashMap<u64, OwnPtr<Job>> jobs;
+ Vector<String, 256> cached_path;
+
+ enum ShellEventType {
+ ReadLine,
+ ChildExited,
+ };
+
+private:
+ Shell();
+ virtual ~Shell() override;
+
+ struct SpawnedProcess {
+ String name;
+ pid_t pid;
+ };
+
+ void cache_path();
+
+ IterationDecision wait_for_pid(const SpawnedProcess&, bool is_first_command_in_chain, int& return_value);
+
+ virtual void custom_event(Core::CustomEvent&) override;
+
+#define __ENUMERATE_SHELL_BUILTIN(builtin) \
+ int builtin_##builtin(int argc, const char** argv);
+
+ ENUMERATE_SHELL_BUILTINS();
+
+#undef __ENUMERATE_SHELL_BUILTIN
+
+ ExitCodeOrContinuationRequest::ContinuationRequest m_should_continue { ExitCodeOrContinuationRequest::Nothing };
+ StringBuilder m_complete_line_builder;
+ bool m_should_break_current_command { false };
+};
+
+static constexpr bool is_word_character(char c)
+{
+ return c == '_' || (c <= 'Z' && c >= 'A') || (c <= 'z' && c >= 'a');
+}
diff --git a/Shell/main.cpp b/Shell/main.cpp
index 38cf360a8e..11244eb4b1 100644
--- a/Shell/main.cpp
+++ b/Shell/main.cpp
@@ -24,1754 +24,92 @@
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*/
-#include "GlobalState.h"
-#include "Parser.h"
-#include <AK/FileSystemPath.h>
-#include <AK/Function.h>
-#include <AK/ScopeGuard.h>
-#include <AK/StringBuilder.h>
-#include <LibCore/ArgsParser.h>
-#include <LibCore/DirIterator.h>
-#include <LibCore/ElapsedTimer.h>
+#include "Execution.h"
+#include "Shell.h"
+#include <LibCore/Event.h>
+#include <LibCore/EventLoop.h>
#include <LibCore/File.h>
-#include <LibLine/Editor.h>
-#include <errno.h>
-#include <fcntl.h>
-#include <pwd.h>
#include <signal.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
-#include <sys/mman.h>
-#include <sys/stat.h>
-#include <sys/utsname.h>
#include <sys/wait.h>
-#include <termios.h>
-#include <unistd.h>
-//#define SH_DEBUG
+Line::Editor editor { Line::Configuration { Line::Configuration::UnescapedSpaces } };
+Shell* s_shell;
-GlobalState g;
-static Line::Editor editor { Line::Configuration { Line::Configuration::UnescapedSpaces } };
-// FIXME: We do not expand variables inside strings
-// if we want to be more sh-like, we should do that some day
-static constexpr bool HighlightVariablesInsideStrings = false;
-static bool s_disable_hyperlinks = false;
-
-static IterationDecision wait_for_pid(pid_t pid, const String& name, bool is_first_command_in_chain, int& return_value);
-
-static void print_path(const String& path)
-{
- if (s_disable_hyperlinks) {
- printf("%s", path.characters());
- return;
- }
- printf("\033]8;;file://%s%s\033\\%s\033]8;;\033\\", g.hostname, path.characters(), path.characters());
-}
-
-struct ExitCodeOrContinuationRequest {
- enum ContinuationRequest {
- Nothing,
- Pipe,
- DoubleQuotedString,
- SingleQuotedString,
- };
-
- ExitCodeOrContinuationRequest(ContinuationRequest continuation)
- : continuation(continuation)
- {
- }
-
- ExitCodeOrContinuationRequest(int exit)
- : exit_code(exit)
- {
- }
-
- bool has_value() const { return exit_code.has_value(); }
- int value() const
- {
- ASSERT(has_value());
- return exit_code.value();
- }
-
- Optional<int> exit_code;
- ContinuationRequest continuation { Nothing };
-};
-
-static ExitCodeOrContinuationRequest run_command(const StringView&);
-void cache_path();
-static ExitCodeOrContinuationRequest::ContinuationRequest s_should_continue { ExitCodeOrContinuationRequest::Nothing };
-
-static String prompt()
-{
- auto build_prompt = []() -> String {
- auto* ps1 = getenv("PROMPT");
- if (!ps1) {
- if (g.uid == 0)
- return "# ";
-
- StringBuilder builder;
- builder.appendf("\033]0;%s@%s:%s\007", g.username.characters(), g.hostname, g.cwd.characters());
- builder.appendf("\033[31;1m%s\033[0m@\033[37;1m%s\033[0m:\033[32;1m%s\033[0m$> ", g.username.characters(), g.hostname, g.cwd.characters());
- return builder.to_string();
- }
-
- StringBuilder builder;
- for (char* ptr = ps1; *ptr; ++ptr) {
- if (*ptr == '\\') {
- ++ptr;
- if (!*ptr)
- break;
- switch (*ptr) {
- case 'X':
- builder.append("\033]0;");
- break;
- case 'a':
- builder.append(0x07);
- break;
- case 'e':
- builder.append(0x1b);
- break;
- case 'u':
- builder.append(g.username);
- break;
- case 'h':
- builder.append(g.hostname);
- break;
- case 'w': {
- String home_path = getenv("HOME");
- if (g.cwd.starts_with(home_path)) {
- builder.append('~');
- builder.append(g.cwd.substring_view(home_path.length(), g.cwd.length() - home_path.length()));
- } else {
- builder.append(g.cwd);
- }
- break;
- }
- case 'p':
- builder.append(g.uid == 0 ? '#' : '$');
- break;
- }
- continue;
- }
- builder.append(*ptr);
- }
- return builder.to_string();
- };
-
- auto the_prompt = build_prompt();
- auto prompt_length = editor.actual_rendered_string_length(the_prompt);
-
- if (s_should_continue != ExitCodeOrContinuationRequest::Nothing) {
- const auto format_string = "\033[34m%.*-s\033[m";
- switch (s_should_continue) {
- case ExitCodeOrContinuationRequest::Pipe:
- return String::format(format_string, prompt_length, "pipe> ");
- case ExitCodeOrContinuationRequest::DoubleQuotedString:
- return String::format(format_string, prompt_length, "dquote> ");
- case ExitCodeOrContinuationRequest::SingleQuotedString:
- return String::format(format_string, prompt_length, "squote> ");
- default:
- break;
- }
- }
- return the_prompt;
-}
-
-static int sh_pwd(int, const char**)
-{
- print_path(g.cwd);
- fputc('\n', stdout);
- return 0;
-}
-
-static int sh_exit(int, const char**)
-{
- printf("Good-bye!\n");
- exit(0);
- return 0;
-}
-
-static int sh_export(int argc, const char** argv)
-{
- Vector<const char*> vars;
-
- Core::ArgsParser parser;
- parser.add_positional_argument(vars, "List of variable[=value]'s", "values", Core::ArgsParser::Required::No);
-
- if (!parser.parse(argc, const_cast<char**>(argv), false))
- return 1;
-
- if (vars.size() == 0) {
- for (int i = 0; environ[i]; ++i)
- puts(environ[i]);
- return 0;
- }
-
- int return_value = 0;
-
- for (auto& value : vars) {
- auto parts = String { value }.split_limit('=', 2);
-
- if (parts.size() == 1) {
- parts.append("");
- }
-
- int setenv_return = setenv(parts[0].characters(), parts[1].characters(), 1);
-
- if (setenv_return != 0) {
- perror("setenv");
- return_value = 1;
- break;
- }
-
- if (parts[0] == "PATH")
- cache_path();
- }
-
- return return_value;
-}
-
-static int sh_unset(int argc, const char** argv)
-{
- Vector<const char*> vars;
-
- Core::ArgsParser parser;
- parser.add_positional_argument(vars, "List of variables", "variables", Core::ArgsParser::Required::Yes);
-
- if (!parser.parse(argc, const_cast<char**>(argv), false))
- return 1;
-
- for (auto& value : vars)
- unsetenv(value);
-
- return 0;
-}
-
-static String expand_tilde(const String& expression)
-{
- ASSERT(expression.starts_with('~'));
-
- StringBuilder login_name;
- size_t first_slash_index = expression.length();
- for (size_t i = 1; i < expression.length(); ++i) {
- if (expression[i] == '/') {
- first_slash_index = i;
- break;
- }
- login_name.append(expression[i]);
- }
-
- StringBuilder path;
- for (size_t i = first_slash_index; i < expression.length(); ++i)
- path.append(expression[i]);
-
- if (login_name.is_empty()) {
- const char* home = getenv("HOME");
- if (!home) {
- auto passwd = getpwuid(getuid());
- ASSERT(passwd && passwd->pw_dir);
- return String::format("%s/%s", passwd->pw_dir, path.to_string().characters());
- }
- return String::format("%s/%s", home, path.to_string().characters());
- }
-
- auto passwd = getpwnam(login_name.to_string().characters());
- if (!passwd)
- return expression;
- ASSERT(passwd->pw_dir);
-
- return String::format("%s/%s", passwd->pw_dir, path.to_string().characters());
-}
-
-static int sh_cd(int argc, const char** argv)
+void FileDescriptionCollector::collect()
{
- const char* arg_path = nullptr;
-
- Core::ArgsParser parser;
- parser.add_positional_argument(arg_path, "Path to change to", "path", Core::ArgsParser::Required::No);
-
- if (!parser.parse(argc, const_cast<char**>(argv), false))
- return 1;
-
- String new_path;
-
- if (!arg_path) {
- new_path = g.home;
- if (g.cd_history.is_empty() || g.cd_history.last() != g.home)
- g.cd_history.enqueue(g.home);
- } else {
- if (g.cd_history.is_empty() || g.cd_history.last() != arg_path)
- g.cd_history.enqueue(arg_path);
- if (strcmp(argv[1], "-") == 0) {
- char* oldpwd = getenv("OLDPWD");
- if (oldpwd == nullptr)
- return 1;
- new_path = oldpwd;
- } else if (arg_path[0] == '/') {
- new_path = argv[1];
- } else {
- StringBuilder builder;
- builder.append(g.cwd);
- builder.append('/');
- builder.append(arg_path);
- new_path = builder.to_string();
- }
- }
-
- FileSystemPath canonical_path(new_path);
- if (!canonical_path.is_valid()) {
- printf("FileSystemPath failed to canonicalize '%s'\n", new_path.characters());
- return 1;
- }
- const char* path = canonical_path.string().characters();
-
- struct stat st;
- int rc = stat(path, &st);
- if (rc < 0) {
- printf("stat(%s) failed: %s\n", path, strerror(errno));
- return 1;
- }
- if (!S_ISDIR(st.st_mode)) {
- printf("Not a directory: %s\n", path);
- return 1;
- }
- rc = chdir(path);
- if (rc < 0) {
- printf("chdir(%s) failed: %s\n", path, strerror(errno));
- return 1;
- }
- setenv("OLDPWD", g.cwd.characters(), 1);
- g.cwd = canonical_path.string();
- setenv("PWD", g.cwd.characters(), 1);
- return 0;
+ for (auto fd : m_fds)
+ close(fd);
+ m_fds.clear();
}
-static int sh_cdh(int argc, const char** argv)
+FileDescriptionCollector::~FileDescriptionCollector()
{
- int index = -1;
-
- Core::ArgsParser parser;
- parser.add_positional_argument(index, "Index of the cd history entry (leave out for a list)", "index", Core::ArgsParser::Required::No);
-
- if (!parser.parse(argc, const_cast<char**>(argv), false))
- return 1;
-
- if (index == -1) {
- if (g.cd_history.size() == 0) {
- printf("cdh: no history available\n");
- return 0;
- }
-
- for (int i = g.cd_history.size() - 1; i >= 0; --i)
- printf("%lu: %s\n", g.cd_history.size() - i, g.cd_history.at(i).characters());
- return 0;
- }
-
- if (index < 1 || (size_t)index > g.cd_history.size()) {
- fprintf(stderr, "cdh: history index out of bounds: %d not in (0, %zu)\n", index, g.cd_history.size());
- return 1;
- }
-
- const char* path = g.cd_history.at(g.cd_history.size() - index).characters();
- const char* cd_args[] = { "cd", path };
- return sh_cd(2, cd_args);
+ collect();
}
-static int sh_history(int, const char**)
+void FileDescriptionCollector::add(int fd)
{
- for (size_t i = 0; i < editor.history().size(); ++i) {
- printf("%6zu %s\n", i, editor.history()[i].characters());
- }
- return 0;
-}
-
-static int sh_time(int argc, const char** argv)
-{
- Vector<const char*> args;
-
- Core::ArgsParser parser;
- parser.add_positional_argument(args, "Command to execute with arguments", "command", Core::ArgsParser::Required::Yes);
-
- if (!parser.parse(argc, const_cast<char**>(argv), false))
- return 1;
-
- StringBuilder builder;
- builder.join(' ', args);
-
- Core::ElapsedTimer timer;
- timer.start();
- auto exit_code = run_command(builder.string_view());
- if (!exit_code.has_value()) {
- printf("Shell: Incomplete command: %s\n", builder.to_string().characters());
- exit_code = 1;
- }
- printf("Time: %d ms\n", timer.elapsed());
- return exit_code.value();
-}
-
-static int sh_jobs(int argc, const char** argv)
-{
- bool list = false, show_pid = false;
-
- Core::ArgsParser parser;
- parser.add_option(list, "List all information about jobs", "list", 'l');
- parser.add_option(show_pid, "Display the PID of the jobs", "pid", 'p');
-
- if (!parser.parse(argc, const_cast<char**>(argv), false))
- return 1;
-
- enum {
- Basic,
- OnlyPID,
- ListAll,
- } mode { Basic };
-
- if (show_pid)
- mode = OnlyPID;
-
- if (list)
- mode = ListAll;
-
- for (auto& job : g.jobs) {
- auto pid = job.value.pid();
- int wstatus;
- auto rc = waitpid(pid, &wstatus, WNOHANG);
- if (rc == -1) {
- perror("waitpid");
- return 1;
- }
- auto status = "running";
-
- if (rc != 0) {
- if (WIFEXITED(wstatus))
- status = "exited";
-
- if (WIFSTOPPED(wstatus))
- status = "stopped";
-
- if (WIFSIGNALED(wstatus))
- status = "signaled";
- }
-
- switch (mode) {
- case Basic:
- printf("[%llu] %s %s\n", job.value.job_id(), status, job.value.cmd().characters());
- break;
- case OnlyPID:
- printf("[%llu] %d %s %s\n", job.value.job_id(), pid, status, job.value.cmd().characters());
- break;
- case ListAll:
- printf("[%llu] %d %d %s %s\n", job.value.job_id(), pid, job.value.pgid(), status, job.value.cmd().characters());
- break;
- }
- }
-
- return 0;
-}
-
-static int sh_fg(int argc, const char** argv)
-{
- int job_id = -1;
-
- Core::ArgsParser parser;
- parser.add_positional_argument(job_id, "Job id to bring to foreground", "job_id", Core::ArgsParser::Required::No);
-
- if (!parser.parse(argc, const_cast<char**>(argv), false))
- return 1;
-
- if (job_id == -1)
- job_id = g.jobs.size() - 1;
-
- Job* job = nullptr;
-
- for (auto& entry : g.jobs) {
- if (entry.value.job_id() == (u64)job_id) {
- job = &entry.value;
- break;
- }
- }
- if (!job) {
- if (job_id == -1) {
- printf("fg: no current job\n");
- } else {
- printf("fg: job with id %d not found\n", job_id);
- }
- return 1;
- }
-
- dbg() << "Resuming " << job->pid() << " (" << job->cmd() << ")";
- printf("Resuming job %llu - %s\n", job->job_id(), job->cmd().characters());
-
- if (killpg(job->pgid(), SIGCONT) < 0) {
- perror("killpg");
- return 1;
- }
-
- int return_value = 0;
-
- auto current_pid = getpid();
- auto current_pgid = getpgid(current_pid);
-
- setpgid(job->pid(), job->pgid());
- tcsetpgrp(0, job->pgid());
-
- do {
- if (wait_for_pid(job->pid(), job->cmd(), true, return_value) == IterationDecision::Break)
- break;
- } while (errno == EINTR);
-
- setpgid(current_pid, current_pgid);
- tcsetpgrp(0, current_pgid);
-
- return return_value;
-}
-
-static int sh_bg(int argc, const char** argv)
-{
- int job_id = -1;
-
- Core::ArgsParser parser;
- parser.add_positional_argument(job_id, "Job id to run in background", "job_id", Core::ArgsParser::Required::No);
-
- if (!parser.parse(argc, const_cast<char**>(argv), false))
- return 1;
-
- if (job_id == -1)
- job_id = g.jobs.size() - 1;
-
- Job* job = nullptr;
-
- for (auto& entry : g.jobs) {
- if (entry.value.job_id() == (u64)job_id) {
- job = &entry.value;
- break;
- }
- }
- if (!job) {
- if (job_id == -1) {
- printf("bg: no current job\n");
- } else {
- printf("bg: job with id %d not found\n", job_id);
- }
- return 1;
- }
-
- dbg() << "Resuming " << job->pid() << " (" << job->cmd() << ")";
- printf("Resuming job %llu - %s\n", job->job_id(), job->cmd().characters());
-
- if (killpg(job->pgid(), SIGCONT) < 0) {
- perror("killpg");
- return 1;
- }
-
- return 0;
-}
-
-static int sh_umask(int argc, const char** argv)
-{
- const char* mask_text = nullptr;
-
- Core::ArgsParser parser;
- parser.add_positional_argument(mask_text, "New mask (omit to get current mask)", "octal-mask", Core::ArgsParser::Required::No);
-
- if (!parser.parse(argc, const_cast<char**>(argv), false))
- return 1;
-
- if (!mask_text) {
- mode_t old_mask = umask(0);
- printf("%#o\n", old_mask);
- umask(old_mask);
- return 0;
- }
-
- unsigned mask;
- int matches = sscanf(mask_text, "%o", &mask);
- if (matches == 1) {
- umask(mask);
- return 0;
- }
-
- return 0;
-}
-
-static int sh_popd(int argc, const char** argv)
-{
- if (g.directory_stack.size() <= 1) {
- fprintf(stderr, "Shell: popd: directory stack empty\n");
- return 1;
- }
-
- bool should_not_switch = false;
- String path = g.directory_stack.take_last();
-
- Core::ArgsParser parser;
- parser.add_option(should_not_switch, "Do not switch dirs", "no-switch", 'n');
-
- if (!parser.parse(argc, const_cast<char**>(argv), false))
- return 1;
-
- bool should_switch = !should_not_switch;
-
- // When no arguments are given, popd removes the top directory from the stack and performs a cd to the new top directory.
- if (argc == 1) {
- int rc = chdir(path.characters());
- if (rc < 0) {
- fprintf(stderr, "chdir(%s) failed: %s\n", path.characters(), strerror(errno));
- return 1;
- }
-
- g.cwd = path;
- return 0;
- }
-
- FileSystemPath canonical_path(path.characters());
- if (!canonical_path.is_valid()) {
- fprintf(stderr, "FileSystemPath failed to canonicalize '%s'\n", path.characters());
- return 1;
- }
-
- const char* real_path = canonical_path.string().characters();
-
- struct stat st;
- int rc = stat(real_path, &st);
- if (rc < 0) {
- fprintf(stderr, "stat(%s) failed: %s\n", real_path, strerror(errno));
- return 1;
- }
-
- if (!S_ISDIR(st.st_mode)) {
- fprintf(stderr, "Not a directory: %s\n", real_path);
- return 1;
- }
-
- if (should_switch) {
- int rc = chdir(real_path);
- if (rc < 0) {
- fprintf(stderr, "chdir(%s) failed: %s\n", real_path, strerror(errno));
- return 1;
- }
-
- g.cwd = canonical_path.string();
- }
-
- return 0;
-}
-
-static int sh_pushd(int argc, const char** argv)
-{
- StringBuilder path_builder;
- bool should_switch = true;
-
- // From the BASH reference manual: https://www.gnu.org/software/bash/manual/html_node/Directory-Stack-Builtins.html
- // With no arguments, pushd exchanges the top two directories and makes the new top the current directory.
- if (argc == 1) {
- if (g.directory_stack.size() < 2) {
- fprintf(stderr, "pushd: no other directory\n");
- return 1;
- }
-
- String dir1 = g.directory_stack.take_first();
- String dir2 = g.directory_stack.take_first();
- g.directory_stack.insert(0, dir2);
- g.directory_stack.insert(1, dir1);
-
- int rc = chdir(dir2.characters());
- if (rc < 0) {
- fprintf(stderr, "chdir(%s) failed: %s\n", dir2.characters(), strerror(errno));
- return 1;
- }
-
- g.cwd = dir2;
-
- return 0;
- }
-
- // Let's assume the user's typed in 'pushd <dir>'
- if (argc == 2) {
- g.directory_stack.append(g.cwd.characters());
- if (argv[1][0] == '/') {
- path_builder.append(argv[1]);
- } else {
- path_builder.appendf("%s/%s", g.cwd.characters(), argv[1]);
- }
- } else if (argc == 3) {
- g.directory_stack.append(g.cwd.characters());
- for (int i = 1; i < argc; i++) {
- const char* arg = argv[i];
-
- if (arg[0] != '-') {
- if (arg[0] == '/') {
- path_builder.append(arg);
- } else
- path_builder.appendf("%s/%s", g.cwd.characters(), arg);
- }
-
- if (!strcmp(arg, "-n"))
- should_switch = false;
- }
- }
-
- FileSystemPath canonical_path(path_builder.to_string());
- if (!canonical_path.is_valid()) {
- fprintf(stderr, "FileSystemPath failed to canonicalize '%s'\n", path_builder.to_string().characters());
- return 1;
- }
-
- const char* real_path = canonical_path.string().characters();
-
- struct stat st;
- int rc = stat(real_path, &st);
- if (rc < 0) {
- fprintf(stderr, "stat(%s) failed: %s\n", real_path, strerror(errno));
- return 1;
- }
-
- if (!S_ISDIR(st.st_mode)) {
- fprintf(stderr, "Not a directory: %s\n", real_path);
- return 1;
- }
-
- if (should_switch) {
- int rc = chdir(real_path);
- if (rc < 0) {
- fprintf(stderr, "chdir(%s) failed: %s\n", real_path, strerror(errno));
- return 1;
- }
-
- g.cwd = canonical_path.string();
- }
-
- return 0;
-}
-
-static int sh_dirs(int argc, const char** argv)
-{
- // The first directory in the stack is ALWAYS the current directory
- g.directory_stack.at(0) = g.cwd.characters();
-
- if (argc == 1) {
- for (auto& directory : g.directory_stack) {
- print_path(directory);
- fputc(' ', stdout);
- }
-
- printf("\n");
- return 0;
- }
-
- bool clear = false;
- bool print = false;
- bool number_when_printing = false;
-
- Vector<const char*> paths;
-
- Core::ArgsParser parser;
- parser.add_option(clear, "Clear the directory stack", "clear", 'c');
- parser.add_option(print, "Print directory entries", "print", 'p');
- parser.add_option(number_when_printing, "Number the directories in the stack when printing", "number", 'v');
- parser.add_positional_argument(paths, "Extra paths to put on the stack", "paths", Core::ArgsParser::Required::No);
-
- if (!parser.parse(argc, const_cast<char**>(argv), false))
- return 1;
-
- // -v implies -p
- print = print || number_when_printing;
-
- if (clear) {
- for (size_t i = 1; i < g.directory_stack.size(); i++)
- g.directory_stack.remove(i);
- }
-
- for (auto& path : paths)
- g.directory_stack.append(path);
-
- if (print) {
- auto idx = 0;
- for (auto& directory : g.directory_stack) {
- if (number_when_printing)
- printf("%d ", idx++);
- print_path(directory);
- fputc('\n', stdout);
- }
- }
-
- return 0;
-}
-
-static bool handle_builtin(int argc, const char** argv, int& retval)
-{
- if (argc == 0)
- return false;
- if (!strcmp(argv[0], "cd")) {
- retval = sh_cd(argc, argv);
- return true;
- }
- if (!strcmp(argv[0], "cdh")) {
- retval = sh_cdh(argc, argv);
- return true;
- }
- if (!strcmp(argv[0], "pwd")) {
- retval = sh_pwd(argc, argv);
- return true;
- }
- if (!strcmp(argv[0], "exit")) {
- retval = sh_exit(argc, argv);
- return true;
- }
- if (!strcmp(argv[0], "export")) {
- retval = sh_export(argc, argv);
- return true;
- }
- if (!strcmp(argv[0], "unset")) {
- retval = sh_unset(argc, argv);
- return true;
- }
- if (!strcmp(argv[0], "history")) {
- retval = sh_history(argc, argv);
- return true;
- }
- if (!strcmp(argv[0], "umask")) {
- retval = sh_umask(argc, argv);
- return true;
- }
- if (!strcmp(argv[0], "dirs")) {
- retval = sh_dirs(argc, argv);
- return true;
- }
- if (!strcmp(argv[0], "pushd")) {
- retval = sh_pushd(argc, argv);
- return true;
- }
- if (!strcmp(argv[0], "popd")) {
- retval = sh_popd(argc, argv);
- return true;
- }
- if (!strcmp(argv[0], "time")) {
- retval = sh_time(argc, argv);
- return true;
- }
- if (!strcmp(argv[0], "jobs")) {
- retval = sh_jobs(argc, argv);
- return true;
- }
- if (!strcmp(argv[0], "fg")) {
- retval = sh_fg(argc, argv);
- return true;
- }
- if (!strcmp(argv[0], "bg")) {
- retval = sh_bg(argc, argv);
- return true;
- }
- return false;
-}
-
-class FileDescriptionCollector {
-public:
- FileDescriptionCollector() { }
- ~FileDescriptionCollector() { collect(); }
-
- void collect()
- {
- for (auto fd : m_fds)
- close(fd);
- m_fds.clear();
- }
- void add(int fd) { m_fds.append(fd); }
-
-private:
- Vector<int, 32> m_fds;
-};
-
-class CommandTimer {
-public:
- explicit CommandTimer(const String& command)
- : m_command(command)
- {
- m_timer.start();
- }
- ~CommandTimer()
- {
- dbg() << "Command \"" << m_command << "\" finished in " << m_timer.elapsed() << " ms";
- }
-
-private:
- Core::ElapsedTimer m_timer;
- String m_command;
-};
-
-static bool is_glob(const StringView& s)
-{
- for (size_t i = 0; i < s.length(); i++) {
- char c = s.characters_without_null_termination()[i];
- if (c == '*' || c == '?')
- return true;
- }
- return false;
-}
-
-static Vector<StringView> split_path(const StringView& path)
-{
- Vector<StringView> parts;
-
- size_t substart = 0;
- for (size_t i = 0; i < path.length(); i++) {
- char ch = path.characters_without_null_termination()[i];
- if (ch != '/')
- continue;
- size_t sublen = i - substart;
- if (sublen != 0)
- parts.append(path.substring_view(substart, sublen));
- parts.append(path.substring_view(i, 1));
- substart = i + 1;
- }
-
- size_t taillen = path.length() - substart;
- if (taillen != 0)
- parts.append(path.substring_view(substart, taillen));
-
- return parts;
-}
-
-static Vector<String> expand_globs(const StringView& path, const StringView& base)
-{
- auto parts = split_path(path);
-
- StringBuilder builder;
- builder.append(base);
- Vector<String> res;
-
- for (size_t i = 0; i < parts.size(); ++i) {
- auto& part = parts[i];
- if (!is_glob(part)) {
- builder.append(part);
- continue;
- }
-
- // Found a glob.
- String new_base = builder.to_string();
- StringView new_base_v = new_base;
- if (new_base_v.is_empty())
- new_base_v = ".";
- Core::DirIterator di(new_base_v, Core::DirIterator::SkipParentAndBaseDir);
-
- if (di.has_error()) {
- return res;
- }
-
- while (di.has_next()) {
- String name = di.next_path();
-
- // Dotfiles have to be explicitly requested
- if (name[0] == '.' && part[0] != '.')
- continue;
-
- if (name.matches(part, CaseSensitivity::CaseSensitive)) {
-
- StringBuilder nested_base;
- nested_base.append(new_base);
- nested_base.append(name);
-
- StringView remaining_path = path.substring_view_starting_after_substring(part);
- Vector<String> nested_res = expand_globs(remaining_path, nested_base.to_string());
- for (auto& s : nested_res)
- res.append(s);
- }
- }
- return res;
- }
-
- // Found no globs.
- String new_path = builder.to_string();
- if (access(new_path.characters(), F_OK) == 0)
- res.append(new_path);
- return res;
-}
-
-static Vector<String> expand_parameters(const StringView& param)
-{
- if (!param.starts_with('$'))
- return { param };
-
- String variable_name = String(param.substring_view(1, param.length() - 1));
- if (variable_name == "?")
- return { String::number(g.last_return_code) };
- else if (variable_name == "$")
- return { String::number(getpid()) };
-
- char* env_value = getenv(variable_name.characters());
- if (env_value == nullptr)
- return { "" };
-
- Vector<String> res;
- String str_env_value = String(env_value);
- const auto& split_text = str_env_value.split_view(' ');
- for (auto& part : split_text)
- res.append(part);
- return res;
-}
-
-static Vector<String> process_arguments(const Vector<Token>& args)
-{
- Vector<String> argv_string;
- for (auto& arg : args) {
- if (arg.type == Token::Comment)
- continue;
-
- // This will return the text passed in if it wasn't a variable
- // This lets us just loop over its values
- auto expanded_parameters = expand_parameters(arg.text);
-
- for (auto& exp_arg : expanded_parameters) {
- if (exp_arg.starts_with('~'))
- exp_arg = expand_tilde(exp_arg);
-
- auto expanded_globs = expand_globs(exp_arg, "");
- for (auto& path : expanded_globs)
- argv_string.append(path);
-
- if (expanded_globs.is_empty())
- argv_string.append(exp_arg);
- }
- }
-
- return argv_string;
-}
-
-static ExitCodeOrContinuationRequest::ContinuationRequest is_complete(const Vector<Command>& commands)
-{
- // check if the last command ends with a pipe, or an unterminated string
- auto& last_command = commands.last();
- auto& subcommands = last_command.subcommands;
- if (subcommands.size() == 0)
- return ExitCodeOrContinuationRequest::Nothing;
-
- auto& last_subcommand = subcommands.last();
-
- if (!last_subcommand.redirections.find([](auto& redirection) { return redirection.type == Redirection::Pipe; }).is_end())
- return ExitCodeOrContinuationRequest::Pipe;
-
- if (!last_subcommand.args.find([](auto& token) { return token.type == Token::UnterminatedSingleQuoted; }).is_end())
- return ExitCodeOrContinuationRequest::SingleQuotedString;
-
- if (!last_subcommand.args.find([](auto& token) { return token.type == Token::UnterminatedDoubleQuoted; }).is_end())
- return ExitCodeOrContinuationRequest::DoubleQuotedString;
-
- return ExitCodeOrContinuationRequest::Nothing;
-}
-
-static IterationDecision wait_for_pid(pid_t pid, const String& name, bool is_first_command_in_chain, int& return_value)
-{
- // disable the child signal handler
- auto* sigchld_handler = signal(SIGCHLD, nullptr);
-
- int wstatus = 0;
- int rc = waitpid(pid, &wstatus, WSTOPPED);
- auto errno_save = errno;
-
- // reenable the signal handler
- signal(SIGCHLD, sigchld_handler);
-
- errno = errno_save;
- if (rc < 0 && errno != EINTR) {
- if (errno != ECHILD)
- perror("waitpid");
- return IterationDecision::Break;
- }
-
- auto job_id = g.jobs.get(pid).value_or(Job {}).job_id();
-
- if (WIFEXITED(wstatus)) {
- if (WEXITSTATUS(wstatus) != 0)
- dbg() << "Shell: " << name << ":" << pid << " exited with status " << WEXITSTATUS(wstatus);
-
- if (is_first_command_in_chain)
- return_value = WEXITSTATUS(wstatus);
-
- g.jobs.remove(pid);
- return IterationDecision::Break;
- }
-
- if (WIFSTOPPED(wstatus)) {
- fprintf(stderr, "Shell: [%llu] %s(%d) %s\n", job_id, name.characters(), pid, strsignal(WSTOPSIG(wstatus)));
- return IterationDecision::Continue;
- }
-
- if (WIFSIGNALED(wstatus)) {
- printf("Shell: [%llu] %s(%d) exited due to signal '%s'\n", job_id, name.characters(), pid, strsignal(WTERMSIG(wstatus)));
- } else {
- printf("Shell: [%llu] %s(%d) exited abnormally\n", job_id, name.characters(), pid);
- }
-
- g.jobs.remove(pid);
- return IterationDecision::Break;
-}
-
-static ExitCodeOrContinuationRequest run_command(const StringView& cmd)
-{
- if (cmd.is_empty())
- return 0;
-
- if (cmd.starts_with("#"))
- return 0;
-
- auto commands = Parser(cmd).parse();
-
- if (!commands.size())
- return 1;
-
- auto needs_more = is_complete(commands);
- if (needs_more != ExitCodeOrContinuationRequest::Nothing)
- return needs_more;
-
-#ifdef SH_DEBUG
- for (auto& command : commands) {
- for (size_t i = 0; i < command.subcommands.size(); ++i) {
- for (size_t j = 0; j < i; ++j)
- dbgprintf(" ");
- for (auto& arg : command.subcommands[i].args) {
- switch (arg.type) {
- case Token::Bare:
- dbgprintf("<%s> ", arg.text.characters());
- break;
- case Token::SingleQuoted:
- dbgprintf("'<%s>' ", arg.text.characters());
- break;
- case Token::DoubleQuoted:
- dbgprintf("\"<%s>\" ", arg.text.characters());
- break;
- case Token::UnterminatedSingleQuoted:
- dbgprintf("\'<%s> ", arg.text.characters());
- break;
- case Token::UnterminatedDoubleQuoted:
- dbgprintf("\"<%s> ", arg.text.characters());
- break;
- case Token::Special:
- dbgprintf("<%s> ", arg.text.characters());
- break;
- case Token::Comment:
- dbgprintf("<%s> ", arg.text.characters());
- break;
- }
- }
- dbgprintf("\n");
- for (auto& redirecton : command.subcommands[i].redirections) {
- for (size_t j = 0; j < i; ++j)
- dbgprintf(" ");
- dbgprintf(" ");
- switch (redirecton.type) {
- case Redirection::Pipe:
- dbgprintf("Pipe\n");
- break;
- case Redirection::FileRead:
- dbgprintf("fd:%d = FileRead: %s\n", redirecton.fd, redirecton.path.characters());
- break;
- case Redirection::FileWrite:
- dbgprintf("fd:%d = FileWrite: %s\n", redirecton.fd, redirecton.path.characters());
- break;
- case Redirection::FileWriteAppend:
- dbgprintf("fd:%d = FileWriteAppend: %s\n", redirecton.fd, redirecton.path.characters());
- break;
- default:
- break;
- }
- }
- }
- dbgprintf("\n");
- }
-#endif
-
- struct termios trm;
- tcgetattr(0, &trm);
-
- struct SpawnedProcess {
- String name;
- pid_t pid;
- };
-
- int return_value = 0;
-
- for (auto& command : commands) {
- if (command.subcommands.is_empty())
- continue;
-
- FileDescriptionCollector fds;
-
- for (size_t i = 0; i < command.subcommands.size(); ++i) {
- auto& subcommand = command.subcommands[i];
- for (auto& redirection : subcommand.redirections) {
- switch (redirection.type) {
- case Redirection::Pipe: {
- int pipefd[2];
- int rc = pipe(pipefd);
- if (rc < 0) {
- perror("pipe");
- return 1;
- }
- subcommand.rewirings.append({ STDOUT_FILENO, pipefd[1] });
- auto& next_command = command.subcommands[i + 1];
- next_command.rewirings.append({ STDIN_FILENO, pipefd[0] });
- fds.add(pipefd[0]);
- fds.add(pipefd[1]);
- break;
- }
- case Redirection::FileWriteAppend: {
- int fd = open(redirection.path.characters(), O_WRONLY | O_CREAT | O_APPEND, 0666);
- if (fd < 0) {
- perror("open");
- return 1;
- }
- subcommand.rewirings.append({ redirection.fd, fd });
- fds.add(fd);
- break;
- }
- case Redirection::FileWrite: {
- int fd = open(redirection.path.characters(), O_WRONLY | O_CREAT | O_TRUNC, 0666);
- if (fd < 0) {
- perror("open");
- return 1;
- }
- subcommand.rewirings.append({ redirection.fd, fd });
- fds.add(fd);
- break;
- }
- case Redirection::FileRead: {
- int fd = open(redirection.path.characters(), O_RDONLY);
- if (fd < 0) {
- perror("open");
- return 1;
- }
- subcommand.rewirings.append({ redirection.fd, fd });
- fds.add(fd);
- break;
- }
- }
- }
- }
-
- Vector<SpawnedProcess> children;
-
- CommandTimer timer(cmd);
-
- for (size_t i = 0; i < command.subcommands.size(); ++i) {
- auto& subcommand = command.subcommands[i];
- Vector<String> argv_string = process_arguments(subcommand.args);
- Vector<const char*> argv;
- argv.ensure_capacity(argv_string.size());
- for (const auto& s : argv_string) {
- argv.append(s.characters());
- }
- argv.append(nullptr);
-
-#ifdef SH_DEBUG
- for (auto& arg : argv) {
- dbgprintf("<%s> ", arg);
- }
- dbgprintf("\n");
-#endif
-
- int retval = 0;
- if (handle_builtin(argv.size() - 1, argv.data(), retval))
- return retval;
-
- pid_t child = fork();
- if (!child) {
- setpgid(0, 0);
- tcsetpgrp(0, getpid());
- tcsetattr(0, TCSANOW, &g.default_termios);
- for (auto& rewiring : subcommand.rewirings) {
-#ifdef SH_DEBUG
- dbgprintf("in %s<%d>, dup2(%d, %d)\n", argv[0], getpid(), rewiring.rewire_fd, rewiring.fd);
-#endif
- int rc = dup2(rewiring.rewire_fd, rewiring.fd);
- if (rc < 0) {
- perror("dup2");
- return 1;
- }
- }
-
- fds.collect();
-
- int rc = execvp(argv[0], const_cast<char* const*>(argv.data()));
- if (rc < 0) {
- if (errno == ENOENT) {
- int shebang_fd = open(argv[0], O_RDONLY);
- auto close_argv = ScopeGuard([shebang_fd]() { if (shebang_fd >= 0) close(shebang_fd); });
- char shebang[256] {};
- ssize_t num_read = -1;
- if ((shebang_fd >= 0) && ((num_read = read(shebang_fd, shebang, sizeof(shebang))) >= 2) && (StringView(shebang).starts_with("#!"))) {
- StringView shebang_path_view(&shebang[2], num_read - 2);
- Optional<size_t> newline_pos = shebang_path_view.find_first_of("\n\r");
- shebang[newline_pos.has_value() ? newline_pos.value() : num_read] = '\0';
- fprintf(stderr, "%s: Invalid interpreter \"%s\": %s\n", argv[0], &shebang[2], strerror(ENOENT));
- } else
- fprintf(stderr, "%s: Command not found.\n", argv[0]);
- } else {
- int saved_errno = errno;
- struct stat st;
- if (stat(argv[0], &st) == 0 && S_ISDIR(st.st_mode)) {
- fprintf(stderr, "Shell: %s: Is a directory\n", argv[0]);
- _exit(126);
- }
- fprintf(stderr, "execvp(%s): %s\n", argv[0], strerror(saved_errno));
- }
- _exit(126);
- }
- ASSERT_NOT_REACHED();
- }
- children.append({ argv[0], child });
-
- StringBuilder cmd;
- cmd.join(" ", argv_string);
-
- g.jobs.set((u64)child, { child, (unsigned)child, cmd.build(), g.jobs.size() });
- }
-
-#ifdef SH_DEBUG
- dbgprintf("Closing fds in shell process:\n");
-#endif
- fds.collect();
-
-#ifdef SH_DEBUG
- dbgprintf("Now we gotta wait on children:\n");
- for (auto& child : children)
- dbgprintf(" %d (%s)\n", child.pid, child.name.characters());
-#endif
-
- for (size_t i = 0; i < children.size(); ++i) {
- auto& child = children[i];
- do {
- if (wait_for_pid(child.pid, child.name, i == 0, return_value) == IterationDecision::Break)
- break;
- } while (errno == EINTR);
- }
- }
-
- g.last_return_code = return_value;
-
- // FIXME: Should I really have to tcsetpgrp() after my child has exited?
- // Is the terminal controlling pgrp really still the PGID of the dead process?
- tcsetpgrp(0, getpid());
- tcsetattr(0, TCSANOW, &trm);
- return return_value;
-}
-
-static String get_history_path()
-{
- StringBuilder builder;
- builder.append(g.home);
- builder.append("/.history");
- return builder.to_string();
-}
-
-void load_history()
-{
- auto history_file = Core::File::construct(get_history_path());
- if (!history_file->open(Core::IODevice::ReadOnly))
- return;
- while (history_file->can_read_line()) {
- auto b = history_file->read_line(1024);
- // skip the newline and terminating bytes
- editor.add_to_history(String(reinterpret_cast<const char*>(b.data()), b.size() - 2));
- }
-}
-
-void save_history()
-{
- auto file_or_error = Core::File::open(get_history_path(), Core::IODevice::WriteOnly, 0600);
- if (file_or_error.is_error())
- return;
- auto& file = *file_or_error.value();
- for (const auto& line : editor.history()) {
- file.write(line);
- file.write("\n");
- }
-}
-
-String escape_token(const String& token)
-{
- StringBuilder builder;
-
- for (auto c : token) {
- switch (c) {
- case '\'':
- case '"':
- case '$':
- case '|':
- case '>':
- case '<':
- case '&':
- case '\\':
- case ' ':
- builder.append('\\');
- break;
- default:
- break;
- }
- builder.append(c);
- }
-
- return builder.build();
-}
-
-String unescape_token(const String& token)
-{
- StringBuilder builder;
-
- enum {
- Free,
- Escaped
- } state { Free };
-
- for (auto c : token) {
- switch (state) {
- case Escaped:
- builder.append(c);
- state = Free;
- break;
- case Free:
- if (c == '\\')
- state = Escaped;
- else
- builder.append(c);
- break;
- }
- }
-
- if (state == Escaped)
- builder.append('\\');
-
- return builder.build();
-}
-
-Vector<String, 256> cached_path;
-void cache_path()
-{
- if (!cached_path.is_empty())
- cached_path.clear_with_capacity();
-
- String path = getenv("PATH");
- if (path.is_empty())
- return;
-
- auto directories = path.split(':');
- for (const auto& directory : directories) {
- Core::DirIterator programs(directory.characters(), Core::DirIterator::SkipDots);
- while (programs.has_next()) {
- auto program = programs.next_path();
- String program_path = String::format("%s/%s", directory.characters(), program.characters());
- if (access(program_path.characters(), X_OK) == 0)
- cached_path.append(escape_token(program.characters()));
- }
- }
-
- quick_sort(cached_path);
-}
-
-static bool is_word_character(char c)
-{
- return c == '_' || (c <= 'Z' && c >= 'A') || (c <= 'z' && c >= 'a');
+ m_fds.append(fd);
}
int main(int argc, char** argv)
{
- if (pledge("stdio rpath wpath cpath proc exec tty", nullptr) < 0) {
+ Core::EventLoop loop;
+
+ if (pledge("stdio rpath wpath cpath proc exec tty accept", nullptr) < 0) {
perror("pledge");
return 1;
}
- g.uid = getuid();
- tcsetpgrp(0, getpgrp());
+ auto shell = Shell::construct();
+ s_shell = shell.ptr();
editor.initialize();
- g.termios = editor.termios();
- g.default_termios = editor.default_termios();
- editor.on_display_refresh = [&](Line::Editor& editor) {
- editor.strip_styles();
- StringBuilder builder;
- if (s_should_continue == ExitCodeOrContinuationRequest::DoubleQuotedString) {
- builder.append('"');
- }
- if (s_should_continue == ExitCodeOrContinuationRequest::SingleQuotedString) {
- builder.append('\'');
- }
- builder.append(StringView { editor.buffer().data(), editor.buffer().size() });
- auto commands = Parser { builder.string_view() }.parse();
- auto first_command { true };
- for (auto& command : commands) {
- for (auto& subcommand : command.subcommands) {
- auto first { true };
- for (auto& arg : subcommand.args) {
- auto start = arg.end - arg.length;
-
- if (arg.type == Token::Comment) {
- editor.stylize({ start, arg.end }, { Line::Style::Foreground(150, 150, 150) }); // light gray
- continue;
- }
-
- if (s_should_continue == ExitCodeOrContinuationRequest::DoubleQuotedString || s_should_continue == ExitCodeOrContinuationRequest::SingleQuotedString) {
- if (!first_command)
- --start;
- --arg.end;
- }
- if (first) {
- first = false;
- // only treat this as a command name if we're not continuing strings
- if (!first_command || (s_should_continue == ExitCodeOrContinuationRequest::Nothing || s_should_continue == ExitCodeOrContinuationRequest::Pipe)) {
- editor.stylize({ start, arg.end }, { Line::Style::Bold });
- first_command = false;
- continue;
- }
- first_command = false;
- }
-
- if (arg.type == Token::SingleQuoted || arg.type == Token::UnterminatedSingleQuoted) {
- editor.stylize({ start - 1, arg.end + (arg.type != Token::UnterminatedSingleQuoted) }, { Line::Style::Foreground(Line::Style::XtermColor::Yellow) });
- continue;
- }
+ shell->termios = editor.termios();
+ shell->default_termios = editor.default_termios();
- if (arg.type == Token::DoubleQuoted || arg.type == Token::UnterminatedDoubleQuoted) {
- editor.stylize({ start - 1, arg.end + (arg.type != Token::UnterminatedDoubleQuoted) }, { Line::Style::Foreground(Line::Style::XtermColor::Yellow) });
- if constexpr (HighlightVariablesInsideStrings)
- goto highlight_variables;
- else
- continue;
- }
-
- if (is_glob(arg.text)) {
- editor.stylize({ start, arg.end }, { Line::Style::Foreground(59, 142, 234) }); // bright-ish blue
- continue;
- }
-
- if (arg.text.starts_with("--")) {
- if (arg.length == 2)
- editor.stylize({ start, arg.end }, { Line::Style::Foreground(Line::Style::XtermColor::Green) });
- else
- editor.stylize({ start, arg.end }, { Line::Style::Foreground(Line::Style::XtermColor::Cyan) });
- } else if (arg.text.starts_with("-") && arg.length > 1) {
- editor.stylize({ start, arg.end }, { Line::Style::Foreground(Line::Style::XtermColor::Cyan) });
- }
-
- highlight_variables:;
-
- size_t slice_index = 0;
- Optional<size_t> maybe_index;
- while (slice_index < arg.length) {
- maybe_index = arg.text.substring_view(slice_index, arg.length - slice_index).find_first_of('$');
- if (!maybe_index.has_value())
- break;
- auto index = maybe_index.value() + 1;
- auto end_index = index;
- if (index >= arg.length)
- break;
- for (; end_index < arg.length; ++end_index) {
- if (!is_word_character(arg.text[end_index]))
- break;
- }
- editor.stylize({ index + start - 1, end_index + start }, { Line::Style::Foreground(214, 112, 214) });
- slice_index = end_index + 1;
- }
- }
- }
- }
+ editor.on_display_refresh = [&](auto& editor) {
+ editor.strip_styles();
+ shell->highlight(editor);
};
editor.on_tab_complete_first_token = [&](const String& token_to_complete) -> Vector<Line::CompletionSuggestion> {
- auto token = unescape_token(token_to_complete);
-
- auto match = binary_search(cached_path.data(), cached_path.size(), token, [](const String& token, const String& program) -> int {
- return strncmp(token.characters(), program.characters(), token.length());
- });
-
- if (!match) {
- // There is no executable in the $PATH starting with $token
- // Suggest local executables and directories
- String path;
- Vector<Line::CompletionSuggestion> local_suggestions;
- bool suggest_executables = true;
-
- ssize_t last_slash = token.length() - 1;
- while (last_slash >= 0 && token[last_slash] != '/')
- --last_slash;
-
- if (last_slash >= 0) {
- // Split on the last slash. We'll use the first part as the directory
- // to search and the second part as the token to complete.
- path = token.substring(0, last_slash + 1);
- if (path[0] != '/')
- path = String::format("%s/%s", g.cwd.characters(), path.characters());
- path = canonicalized_path(path);
- token = token.substring(last_slash + 1, token.length() - last_slash - 1);
- } else {
- // We have no slashes, so the directory to search is the current
- // directory and the token to complete is just the original token.
- // In this case, do not suggest executables but directories only.
- path = g.cwd;
- suggest_executables = false;
- }
-
- // the invariant part of the token is actually just the last segment
- // e.g. in `cd /foo/bar', 'bar' is the invariant
- // since we are not suggesting anything starting with
- // `/foo/', but rather just `bar...'
- editor.suggest(escape_token(token).length(), 0);
-
- // only suggest dot-files if path starts with a dot
- Core::DirIterator files(path,
- token.starts_with('.') ? Core::DirIterator::SkipParentAndBaseDir : Core::DirIterator::SkipDots);
-
- while (files.has_next()) {
- auto file = files.next_path();
- auto trivia = " ";
- if (file.starts_with(token)) {
- String file_path = String::format("%s/%s", path.characters(), file.characters());
- struct stat program_status;
- int stat_error = stat(file_path.characters(), &program_status);
- if (stat_error)
- continue;
- if (access(file_path.characters(), X_OK) != 0)
- continue;
- if (S_ISDIR(program_status.st_mode)) {
- if (!suggest_executables)
- continue;
- else
- trivia = "/";
- }
-
- local_suggestions.append({ escape_token(file), trivia });
- }
- }
-
- return local_suggestions;
- }
-
- String completion = *match;
- Vector<Line::CompletionSuggestion> suggestions;
-
- // Now that we have a program name starting with our token, we look at
- // other program names starting with our token and cut off any mismatching
- // characters.
-
- int index = match - cached_path.data();
- for (int i = index - 1; i >= 0 && cached_path[i].starts_with(token); --i) {
- suggestions.append({ cached_path[i], " " });
- }
- for (size_t i = index + 1; i < cached_path.size() && cached_path[i].starts_with(token); ++i) {
- suggestions.append({ cached_path[i], " " });
- }
- suggestions.append({ cached_path[index], " " });
-
- editor.suggest(escape_token(token).length(), 0);
-
- return suggestions;
+ return shell->complete_first(token_to_complete);
};
editor.on_tab_complete_other_token = [&](const String& token_to_complete) -> Vector<Line::CompletionSuggestion> {
- auto token = unescape_token(token_to_complete);
- String path;
- Vector<Line::CompletionSuggestion> suggestions;
-
- ssize_t last_slash = token.length() - 1;
- while (last_slash >= 0 && token[last_slash] != '/')
- --last_slash;
-
- if (last_slash >= 0) {
- // Split on the last slash. We'll use the first part as the directory
- // to search and the second part as the token to complete.
- path = token.substring(0, last_slash + 1);
- if (path[0] != '/')
- path = String::format("%s/%s", g.cwd.characters(), path.characters());
- path = canonicalized_path(path);
- token = token.substring(last_slash + 1, token.length() - last_slash - 1);
- } else {
- // We have no slashes, so the directory to search is the current
- // directory and the token to complete is just the original token.
- path = g.cwd;
- }
-
- // the invariant part of the token is actually just the last segment
- // e.g. in `cd /foo/bar', 'bar' is the invariant
- // since we are not suggesting anything starting with
- // `/foo/', but rather just `bar...'
- editor.suggest(escape_token(token).length(), 0);
-
- // only suggest dot-files if path starts with a dot
- Core::DirIterator files(path,
- token.starts_with('.') ? Core::DirIterator::SkipParentAndBaseDir : Core::DirIterator::SkipDots);
-
- while (files.has_next()) {
- auto file = files.next_path();
- if (file.starts_with(token)) {
- struct stat program_status;
- String file_path = String::format("%s/%s", path.characters(), file.characters());
- int stat_error = stat(file_path.characters(), &program_status);
- if (!stat_error) {
- if (S_ISDIR(program_status.st_mode))
- suggestions.append({ escape_token(file), "/" });
- else
- suggestions.append({ escape_token(file), " " });
- }
- }
- }
-
- return suggestions;
+ return shell->complete_other(token_to_complete);
};
signal(SIGINT, [](int) {
- g.was_interrupted = true;
editor.interrupted();
});
signal(SIGWINCH, [](int) {
- g.was_resized = true;
editor.resized();
});
signal(SIGHUP, [](int) {
- save_history();
+ s_shell->save_history();
});
signal(SIGCHLD, [](int) {
- int wstatus = 0;
- auto child_pid = waitpid(-1, &wstatus, WNOHANG);
- dbg() << "SIGCHLD " << child_pid << " - " << WIFEXITED(wstatus) << " " << WIFSTOPPED(wstatus);
- if (child_pid != -1) {
- if (WIFEXITED(wstatus) || WIFSIGNALED(wstatus)) {
- // FIXME: We should switch to using Core::EventLoop and defer this stuff
- auto entry = g.jobs.find(child_pid);
- if (entry != g.jobs.end()) {
- fprintf(stderr, "Shell: Job %d(%s) exited\n", entry->value.pid(), entry->value.cmd().characters());
- g.jobs.remove(entry);
+ auto& jobs = s_shell->jobs;
+ for (auto& job : jobs) {
+ int wstatus = 0;
+ auto child_pid = waitpid(job.value->pid(), &wstatus, WNOHANG);
+ if (child_pid == job.value->pid()) {
+ if (WIFEXITED(wstatus) || WIFSIGNALED(wstatus)) {
+ Core::EventLoop::current().post_event(*s_shell, make<Core::CustomEvent>(Shell::ShellEventType::ChildExited, job.value));
}
}
}
});
- int rc = gethostname(g.hostname, sizeof(g.hostname));
- if (rc < 0)
- perror("gethostname");
- rc = ttyname_r(0, g.ttyname, sizeof(g.ttyname));
- if (rc < 0)
- perror("ttyname_r");
-
- {
- auto* cwd = getcwd(nullptr, 0);
- g.cwd = cwd;
- setenv("PWD", cwd, 1);
- free(cwd);
- }
-
- {
- auto* pw = getpwuid(getuid());
- if (pw) {
- g.username = pw->pw_name;
- g.home = pw->pw_dir;
- setenv("HOME", pw->pw_dir, 1);
- }
- endpwent();
- }
-
if (argc > 2 && !strcmp(argv[1], "-c")) {
dbgprintf("sh -c '%s'\n", argv[2]);
- run_command(argv[2]);
+ shell->run_command(argv[2]);
return 0;
}
@@ -1785,57 +123,19 @@ int main(int argc, char** argv)
auto line = file->read_line(4096);
if (line.is_null())
break;
- run_command(String::copy(line, Chomp));
+ shell->run_command(String::copy(line, Chomp));
}
return 0;
}
- g.directory_stack.append(g.cwd);
-
- load_history();
- atexit(save_history);
-
- cache_path();
-
- StringBuilder complete_line_builder;
-
- bool should_break_current_command { false };
-
editor.on_interrupt_handled = [&] {
- if (s_should_continue != ExitCodeOrContinuationRequest::ContinuationRequest::Nothing) {
- should_break_current_command = true;
+ if (!shell->should_read_more()) {
+ shell->finish_command();
editor.finish();
}
};
- for (;;) {
- auto line = editor.get_line(prompt());
-
- if (should_break_current_command) {
- complete_line_builder.clear();
- s_should_continue = ExitCodeOrContinuationRequest::ContinuationRequest::Nothing;
- should_break_current_command = false;
- continue;
- }
-
- if (line.is_empty())
- continue;
-
- // FIXME: This might be a bit counter-intuitive, since we put nothing
- // between the two lines, even though the user has pressed enter
- // but since the LineEditor cannot yet handle literal newlines
- // inside the text, we opt to do this the wrong way (for the time being)
- complete_line_builder.append(line);
-
- auto complete_or_exit_code = run_command(complete_line_builder.string_view());
- s_should_continue = complete_or_exit_code.continuation;
-
- if (!complete_or_exit_code.has_value())
- continue;
-
- editor.add_to_history(complete_line_builder.build());
- complete_line_builder.clear();
- }
+ Core::EventLoop::current().post_event(*shell, make<Core::CustomEvent>(Shell::ShellEventType::ReadLine));
- return 0;
+ return loop.exec();
}