Builtin.cpp 58 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799
  1. /*
  2. * Copyright (c) 2020-2021, the SerenityOS developers.
  3. *
  4. * SPDX-License-Identifier: BSD-2-Clause
  5. */
  6. #include "AST.h"
  7. #include "Formatter.h"
  8. #include "PosixParser.h"
  9. #include "Shell.h"
  10. #include <AK/DeprecatedString.h>
  11. #include <AK/LexicalPath.h>
  12. #include <AK/ScopeGuard.h>
  13. #include <AK/Statistics.h>
  14. #include <LibCore/ArgsParser.h>
  15. #include <LibCore/DeprecatedFile.h>
  16. #include <LibCore/EventLoop.h>
  17. #include <errno.h>
  18. #include <inttypes.h>
  19. #include <limits.h>
  20. #include <signal.h>
  21. #include <sys/wait.h>
  22. #include <unistd.h>
  23. extern char** environ;
  24. namespace Shell {
  25. ErrorOr<int> Shell::builtin_noop(Main::Arguments)
  26. {
  27. return 0;
  28. }
  29. ErrorOr<int> Shell::builtin_dump(Main::Arguments arguments)
  30. {
  31. bool posix = false;
  32. StringView source;
  33. Core::ArgsParser parser;
  34. parser.add_positional_argument(source, "Shell code to parse and dump", "source");
  35. parser.add_option(posix, "Use the POSIX parser", "posix", 'p');
  36. if (!parser.parse(arguments, Core::ArgsParser::FailureBehavior::PrintUsage))
  37. return 1;
  38. TRY((posix ? Posix::Parser { source }.parse() : Parser { source }.parse())->dump(0));
  39. return 0;
  40. }
  41. enum FollowSymlinks {
  42. Yes,
  43. No
  44. };
  45. static Vector<DeprecatedString> find_matching_executables_in_path(StringView filename, FollowSymlinks follow_symlinks = FollowSymlinks::No)
  46. {
  47. // Edge cases in which there are guaranteed no solutions
  48. if (filename.is_empty() || filename.contains('/'))
  49. return {};
  50. char const* path_str = getenv("PATH");
  51. auto path = DEFAULT_PATH_SV;
  52. if (path_str != nullptr) // maybe && *path_str
  53. path = { path_str, strlen(path_str) };
  54. Vector<DeprecatedString> executables;
  55. auto directories = path.split_view(':');
  56. for (auto directory : directories) {
  57. auto file = DeprecatedString::formatted("{}/{}", directory, filename);
  58. if (follow_symlinks == FollowSymlinks::Yes) {
  59. auto path_or_error = Core::DeprecatedFile::read_link(file);
  60. if (!path_or_error.is_error())
  61. file = path_or_error.release_value();
  62. }
  63. if (access(file.characters(), X_OK) == 0)
  64. executables.append(move(file));
  65. }
  66. return executables;
  67. }
  68. ErrorOr<int> Shell::builtin_where(Main::Arguments arguments)
  69. {
  70. Vector<StringView> values_to_lookup;
  71. bool do_only_path_search { false };
  72. bool do_follow_symlinks { false };
  73. bool do_print_only_type { false };
  74. Core::ArgsParser parser;
  75. parser.add_positional_argument(values_to_lookup, "List of shell builtins, aliases or executables", "arguments");
  76. parser.add_option(do_only_path_search, "Search only for executables in the PATH environment variable", "path-only", 'p');
  77. parser.add_option(do_follow_symlinks, "Follow symlinks and print the symlink free path", "follow-symlink", 's');
  78. parser.add_option(do_print_only_type, "Print the argument type instead of a human readable description", "type", 'w');
  79. if (!parser.parse(arguments, Core::ArgsParser::FailureBehavior::PrintUsage))
  80. return 1;
  81. auto const lookup_alias = [do_only_path_search, &m_aliases = this->m_aliases](StringView alias) -> Optional<DeprecatedString> {
  82. if (do_only_path_search)
  83. return {};
  84. return m_aliases.get(alias);
  85. };
  86. auto const lookup_builtin = [do_only_path_search](StringView builtin) -> Optional<DeprecatedString> {
  87. if (do_only_path_search)
  88. return {};
  89. for (auto const& _builtin : builtin_names) {
  90. if (_builtin == builtin) {
  91. return builtin;
  92. }
  93. }
  94. return {};
  95. };
  96. bool at_least_one_succeded { false };
  97. for (auto const& argument : values_to_lookup) {
  98. auto const alias = lookup_alias(argument);
  99. if (alias.has_value()) {
  100. if (do_print_only_type)
  101. outln("{}: alias", argument);
  102. else
  103. outln("{}: aliased to {}", argument, alias.value());
  104. at_least_one_succeded = true;
  105. }
  106. auto const builtin = lookup_builtin(argument);
  107. if (builtin.has_value()) {
  108. if (do_print_only_type)
  109. outln("{}: builtin", builtin.value());
  110. else
  111. outln("{}: shell built-in command", builtin.value());
  112. at_least_one_succeded = true;
  113. }
  114. auto const executables = find_matching_executables_in_path(argument, do_follow_symlinks ? FollowSymlinks::Yes : FollowSymlinks::No);
  115. for (auto const& path : executables) {
  116. if (do_print_only_type)
  117. outln("{}: command", argument);
  118. else
  119. outln(path);
  120. at_least_one_succeded = true;
  121. }
  122. if (!at_least_one_succeded)
  123. warnln("{} not found", argument);
  124. }
  125. return at_least_one_succeded ? 0 : 1;
  126. }
  127. ErrorOr<int> Shell::builtin_alias(Main::Arguments arguments)
  128. {
  129. Vector<DeprecatedString> aliases;
  130. Core::ArgsParser parser;
  131. parser.add_positional_argument(aliases, "List of name[=values]'s", "name[=value]", Core::ArgsParser::Required::No);
  132. if (!parser.parse(arguments, Core::ArgsParser::FailureBehavior::PrintUsage))
  133. return 1;
  134. if (aliases.is_empty()) {
  135. for (auto& alias : m_aliases)
  136. printf("%s=%s\n", escape_token(alias.key).characters(), escape_token(alias.value).characters());
  137. return 0;
  138. }
  139. bool fail = false;
  140. for (auto& argument : aliases) {
  141. auto parts = argument.split_limit('=', 2, SplitBehavior::KeepEmpty);
  142. if (parts.size() == 1) {
  143. auto alias = m_aliases.get(parts[0]);
  144. if (alias.has_value()) {
  145. printf("%s=%s\n", escape_token(parts[0]).characters(), escape_token(alias.value()).characters());
  146. } else {
  147. fail = true;
  148. }
  149. } else {
  150. m_aliases.set(parts[0], parts[1]);
  151. add_entry_to_cache({ RunnablePath::Kind::Alias, parts[0] });
  152. }
  153. }
  154. return fail ? 1 : 0;
  155. }
  156. ErrorOr<int> Shell::builtin_unalias(Main::Arguments arguments)
  157. {
  158. bool remove_all { false };
  159. Vector<DeprecatedString> aliases;
  160. Core::ArgsParser parser;
  161. parser.set_general_help("Remove alias from the list of aliases");
  162. parser.add_option(remove_all, "Remove all aliases", nullptr, 'a');
  163. parser.add_positional_argument(aliases, "List of aliases to remove", "alias", Core::ArgsParser::Required::Yes);
  164. if (!parser.parse(arguments, Core::ArgsParser::FailureBehavior::PrintUsage))
  165. return 1;
  166. if (remove_all) {
  167. m_aliases.clear();
  168. cache_path();
  169. return 0;
  170. }
  171. bool failed { false };
  172. for (auto& argument : aliases) {
  173. if (!m_aliases.contains(argument)) {
  174. warnln("unalias: {}: alias not found", argument);
  175. failed = true;
  176. continue;
  177. }
  178. m_aliases.remove(argument);
  179. remove_entry_from_cache(argument);
  180. }
  181. return failed ? 1 : 0;
  182. }
  183. ErrorOr<int> Shell::builtin_break(Main::Arguments arguments)
  184. {
  185. if (!m_in_posix_mode) {
  186. raise_error(ShellError::EvaluatedSyntaxError, "break: Invalid use of builtin break in non-POSIX mode");
  187. return 1;
  188. }
  189. unsigned count = 1;
  190. Core::ArgsParser parser;
  191. parser.add_positional_argument(count, "Number of loops to 'break' out of", "count", Core::ArgsParser::Required::No);
  192. if (!parser.parse(arguments, Core::ArgsParser::FailureBehavior::PrintUsage))
  193. return 1;
  194. if (count != 1) {
  195. raise_error(ShellError::EvaluatedSyntaxError, "break: count must be equal to 1 (NYI)");
  196. return 1;
  197. }
  198. raise_error(ShellError::InternalControlFlowBreak, "POSIX break");
  199. return 0;
  200. }
  201. ErrorOr<int> Shell::builtin_continue(Main::Arguments arguments)
  202. {
  203. if (!m_in_posix_mode) {
  204. raise_error(ShellError::EvaluatedSyntaxError, "break: Invalid use of builtin continue in non-POSIX mode");
  205. return 1;
  206. }
  207. unsigned count = 1;
  208. Core::ArgsParser parser;
  209. parser.add_positional_argument(count, "Number of loops to 'continue' out of", "count", Core::ArgsParser::Required::No);
  210. if (!parser.parse(arguments, Core::ArgsParser::FailureBehavior::PrintUsage))
  211. return 1;
  212. if (count != 0) {
  213. raise_error(ShellError::EvaluatedSyntaxError, "continue: count must be equal to 1 (NYI)");
  214. return 1;
  215. }
  216. raise_error(ShellError::InternalControlFlowContinue, "POSIX continue");
  217. return 0;
  218. }
  219. ErrorOr<int> Shell::builtin_bg(Main::Arguments arguments)
  220. {
  221. int job_id = -1;
  222. bool is_pid = false;
  223. Core::ArgsParser parser;
  224. parser.add_positional_argument(Core::ArgsParser::Arg {
  225. .help_string = "Job ID or Jobspec to run in background",
  226. .name = "job-id",
  227. .min_values = 0,
  228. .max_values = 1,
  229. .accept_value = [&](StringView value) -> bool {
  230. // Check if it's a pid (i.e. literal integer)
  231. if (auto number = value.to_uint(); number.has_value()) {
  232. job_id = number.value();
  233. is_pid = true;
  234. return true;
  235. }
  236. // Check if it's a jobspec
  237. if (auto id = resolve_job_spec(value); id.has_value()) {
  238. job_id = id.value();
  239. is_pid = false;
  240. return true;
  241. }
  242. return false;
  243. } });
  244. if (!parser.parse(arguments, Core::ArgsParser::FailureBehavior::PrintUsage))
  245. return 1;
  246. if (job_id == -1 && !jobs.is_empty())
  247. job_id = find_last_job_id();
  248. auto* job = const_cast<Job*>(find_job(job_id, is_pid));
  249. if (!job) {
  250. if (job_id == -1) {
  251. warnln("bg: No current job");
  252. } else {
  253. warnln("bg: Job with id/pid {} not found", job_id);
  254. }
  255. return 1;
  256. }
  257. job->set_running_in_background(true);
  258. job->set_should_announce_exit(true);
  259. job->set_shell_did_continue(true);
  260. dbgln("Resuming {} ({})", job->pid(), job->cmd());
  261. warnln("Resuming job {} - {}", job->job_id(), job->cmd());
  262. // Try using the PGID, but if that fails, just use the PID.
  263. if (killpg(job->pgid(), SIGCONT) < 0) {
  264. if (kill(job->pid(), SIGCONT) < 0) {
  265. perror("kill");
  266. return 1;
  267. }
  268. }
  269. return 0;
  270. }
  271. ErrorOr<int> Shell::builtin_type(Main::Arguments arguments)
  272. {
  273. Vector<DeprecatedString> commands;
  274. bool dont_show_function_source = false;
  275. Core::ArgsParser parser;
  276. parser.set_general_help("Display information about commands.");
  277. parser.add_positional_argument(commands, "Command(s) to list info about", "command");
  278. parser.add_option(dont_show_function_source, "Do not show functions source.", "no-fn-source", 'f');
  279. if (!parser.parse(arguments, Core::ArgsParser::FailureBehavior::PrintUsage))
  280. return 1;
  281. bool something_not_found = false;
  282. for (auto& command : commands) {
  283. // check if it is an alias
  284. if (auto alias = m_aliases.get(command); alias.has_value()) {
  285. printf("%s is aliased to `%s`\n", escape_token(command).characters(), escape_token(alias.value()).characters());
  286. continue;
  287. }
  288. // check if it is a function
  289. if (auto function = m_functions.get(command); function.has_value()) {
  290. auto fn = function.value();
  291. printf("%s is a function\n", command.characters());
  292. if (!dont_show_function_source) {
  293. StringBuilder builder;
  294. builder.append(fn.name);
  295. builder.append('(');
  296. for (size_t i = 0; i < fn.arguments.size(); i++) {
  297. builder.append(fn.arguments[i]);
  298. if (!(i == fn.arguments.size() - 1))
  299. builder.append(' ');
  300. }
  301. builder.append(") {\n"sv);
  302. if (fn.body) {
  303. auto formatter = Formatter(*fn.body);
  304. builder.append(formatter.format());
  305. printf("%s\n}\n", builder.to_deprecated_string().characters());
  306. } else {
  307. printf("%s\n}\n", builder.to_deprecated_string().characters());
  308. }
  309. }
  310. continue;
  311. }
  312. // check if its a builtin
  313. if (has_builtin(command)) {
  314. printf("%s is a shell builtin\n", command.characters());
  315. continue;
  316. }
  317. // check if its an executable in PATH
  318. auto fullpath = Core::DeprecatedFile::resolve_executable_from_environment(command);
  319. if (fullpath.has_value()) {
  320. printf("%s is %s\n", command.characters(), escape_token(fullpath.release_value()).characters());
  321. continue;
  322. }
  323. something_not_found = true;
  324. printf("type: %s not found\n", command.characters());
  325. }
  326. if (something_not_found)
  327. return 1;
  328. else
  329. return 0;
  330. }
  331. ErrorOr<int> Shell::builtin_cd(Main::Arguments arguments)
  332. {
  333. StringView arg_path;
  334. Core::ArgsParser parser;
  335. parser.add_positional_argument(arg_path, "Path to change to", "path", Core::ArgsParser::Required::No);
  336. if (!parser.parse(arguments, Core::ArgsParser::FailureBehavior::PrintUsage))
  337. return 1;
  338. DeprecatedString new_path;
  339. if (arg_path.is_empty()) {
  340. new_path = home;
  341. } else {
  342. if (arg_path == "-"sv) {
  343. char* oldpwd = getenv("OLDPWD");
  344. if (oldpwd == nullptr)
  345. return 1;
  346. new_path = oldpwd;
  347. } else {
  348. new_path = arg_path;
  349. }
  350. }
  351. auto real_path = Core::DeprecatedFile::real_path_for(new_path);
  352. if (real_path.is_empty()) {
  353. warnln("Invalid path '{}'", new_path);
  354. return 1;
  355. }
  356. if (cd_history.is_empty() || cd_history.last() != real_path)
  357. cd_history.enqueue(real_path);
  358. auto path_relative_to_current_directory = LexicalPath::relative_path(real_path, cwd);
  359. if (path_relative_to_current_directory.is_empty())
  360. path_relative_to_current_directory = real_path;
  361. char const* path = path_relative_to_current_directory.characters();
  362. int rc = chdir(path);
  363. if (rc < 0) {
  364. if (errno == ENOTDIR) {
  365. warnln("Not a directory: {}", path);
  366. } else {
  367. warnln("chdir({}) failed: {}", path, strerror(errno));
  368. }
  369. return 1;
  370. }
  371. setenv("OLDPWD", cwd.characters(), 1);
  372. cwd = move(real_path);
  373. setenv("PWD", cwd.characters(), 1);
  374. return 0;
  375. }
  376. ErrorOr<int> Shell::builtin_cdh(Main::Arguments arguments)
  377. {
  378. int index = -1;
  379. Core::ArgsParser parser;
  380. parser.add_positional_argument(index, "Index of the cd history entry (leave out for a list)", "index", Core::ArgsParser::Required::No);
  381. if (!parser.parse(arguments, Core::ArgsParser::FailureBehavior::PrintUsage))
  382. return 1;
  383. if (index == -1) {
  384. if (cd_history.is_empty()) {
  385. warnln("cdh: no history available");
  386. return 0;
  387. }
  388. for (ssize_t i = cd_history.size() - 1; i >= 0; --i)
  389. printf("%zu: %s\n", cd_history.size() - i, cd_history.at(i).characters());
  390. return 0;
  391. }
  392. if (index < 1 || (size_t)index > cd_history.size()) {
  393. warnln("cdh: history index out of bounds: {} not in (0, {})", index, cd_history.size());
  394. return 1;
  395. }
  396. StringView path = cd_history.at(cd_history.size() - index);
  397. StringView cd_args[] = { "cd"sv, path };
  398. return Shell::builtin_cd({ .argc = 0, .argv = 0, .strings = cd_args });
  399. }
  400. ErrorOr<int> Shell::builtin_dirs(Main::Arguments arguments)
  401. {
  402. // The first directory in the stack is ALWAYS the current directory
  403. directory_stack.at(0) = cwd.characters();
  404. bool clear = false;
  405. bool print = false;
  406. bool number_when_printing = false;
  407. char separator = ' ';
  408. Vector<DeprecatedString> paths;
  409. Core::ArgsParser parser;
  410. parser.add_option(clear, "Clear the directory stack", "clear", 'c');
  411. parser.add_option(print, "Print directory entries one per line", "print", 'p');
  412. parser.add_option(number_when_printing, "Number the directories in the stack when printing", "number", 'v');
  413. parser.add_positional_argument(paths, "Extra paths to put on the stack", "path", Core::ArgsParser::Required::No);
  414. if (!parser.parse(arguments, Core::ArgsParser::FailureBehavior::PrintUsage))
  415. return 1;
  416. // -v implies -p
  417. print = print || number_when_printing;
  418. if (print) {
  419. if (!paths.is_empty()) {
  420. warnln("dirs: 'print' and 'number' are not allowed when any path is specified");
  421. return 1;
  422. }
  423. separator = '\n';
  424. }
  425. if (clear) {
  426. for (size_t i = 1; i < directory_stack.size(); i++)
  427. directory_stack.remove(i);
  428. }
  429. for (auto& path : paths)
  430. directory_stack.append(path);
  431. if (print || (!clear && paths.is_empty())) {
  432. int index = 0;
  433. for (auto& directory : directory_stack) {
  434. if (number_when_printing)
  435. printf("%d ", index++);
  436. print_path(directory);
  437. fputc(separator, stdout);
  438. }
  439. }
  440. return 0;
  441. }
  442. ErrorOr<int> Shell::builtin_exec(Main::Arguments arguments)
  443. {
  444. if (arguments.strings.size() < 2) {
  445. warnln("Shell: No command given to exec");
  446. return 1;
  447. }
  448. TRY(execute_process(arguments.strings.slice(1)));
  449. // NOTE: Won't get here.
  450. return 0;
  451. }
  452. ErrorOr<int> Shell::builtin_exit(Main::Arguments arguments)
  453. {
  454. int exit_code = 0;
  455. Core::ArgsParser parser;
  456. parser.add_positional_argument(exit_code, "Exit code", "code", Core::ArgsParser::Required::No);
  457. if (!parser.parse(arguments, Core::ArgsParser::FailureBehavior::PrintUsage))
  458. return 1;
  459. if (m_is_interactive) {
  460. if (!jobs.is_empty()) {
  461. if (!m_should_ignore_jobs_on_next_exit) {
  462. warnln("Shell: You have {} active job{}, run 'exit' again to really exit.", jobs.size(), jobs.size() > 1 ? "s" : "");
  463. m_should_ignore_jobs_on_next_exit = true;
  464. return 1;
  465. }
  466. }
  467. }
  468. stop_all_jobs();
  469. if (m_is_interactive) {
  470. m_editor->save_history(get_history_path());
  471. printf("Good-bye!\n");
  472. }
  473. exit(exit_code);
  474. }
  475. ErrorOr<int> Shell::builtin_export(Main::Arguments arguments)
  476. {
  477. Vector<DeprecatedString> vars;
  478. Core::ArgsParser parser;
  479. parser.add_positional_argument(vars, "List of variable[=value]'s", "values", Core::ArgsParser::Required::No);
  480. if (!parser.parse(arguments, Core::ArgsParser::FailureBehavior::PrintUsage))
  481. return 1;
  482. if (vars.is_empty()) {
  483. for (size_t i = 0; environ[i]; ++i)
  484. puts(environ[i]);
  485. return 0;
  486. }
  487. for (auto& value : vars) {
  488. auto parts = value.split_limit('=', 2);
  489. if (parts.is_empty()) {
  490. warnln("Shell: Invalid export spec '{}', expected `variable=value' or `variable'", value);
  491. return 1;
  492. }
  493. if (parts.size() == 1) {
  494. auto value = TRY(lookup_local_variable(parts[0]));
  495. if (value) {
  496. auto values = TRY(const_cast<AST::Value&>(*value).resolve_as_list(*this));
  497. StringBuilder builder;
  498. builder.join(' ', values);
  499. parts.append(builder.to_deprecated_string());
  500. } else {
  501. // Ignore the export.
  502. continue;
  503. }
  504. }
  505. int setenv_return = setenv(parts[0].characters(), parts[1].characters(), 1);
  506. if (setenv_return != 0) {
  507. perror("setenv");
  508. return 1;
  509. }
  510. if (parts[0] == "PATH")
  511. cache_path();
  512. }
  513. return 0;
  514. }
  515. ErrorOr<int> Shell::builtin_glob(Main::Arguments arguments)
  516. {
  517. Vector<DeprecatedString> globs;
  518. Core::ArgsParser parser;
  519. parser.add_positional_argument(globs, "Globs to resolve", "glob");
  520. if (!parser.parse(arguments, Core::ArgsParser::FailureBehavior::PrintUsage))
  521. return 1;
  522. for (auto& glob : globs) {
  523. for (auto& expanded : expand_globs(glob, cwd))
  524. outln("{}", expanded);
  525. }
  526. return 0;
  527. }
  528. ErrorOr<int> Shell::builtin_fg(Main::Arguments arguments)
  529. {
  530. int job_id = -1;
  531. bool is_pid = false;
  532. Core::ArgsParser parser;
  533. parser.add_positional_argument(Core::ArgsParser::Arg {
  534. .help_string = "Job ID or Jobspec to bring to foreground",
  535. .name = "job-id",
  536. .min_values = 0,
  537. .max_values = 1,
  538. .accept_value = [&](StringView value) -> bool {
  539. // Check if it's a pid (i.e. literal integer)
  540. if (auto number = value.to_uint(); number.has_value()) {
  541. job_id = number.value();
  542. is_pid = true;
  543. return true;
  544. }
  545. // Check if it's a jobspec
  546. if (auto id = resolve_job_spec(value); id.has_value()) {
  547. job_id = id.value();
  548. is_pid = false;
  549. return true;
  550. }
  551. return false;
  552. } });
  553. if (!parser.parse(arguments, Core::ArgsParser::FailureBehavior::PrintUsage))
  554. return 1;
  555. if (job_id == -1 && !jobs.is_empty())
  556. job_id = find_last_job_id();
  557. RefPtr<Job> job = find_job(job_id, is_pid);
  558. if (!job) {
  559. if (job_id == -1) {
  560. warnln("fg: No current job");
  561. } else {
  562. warnln("fg: Job with id/pid {} not found", job_id);
  563. }
  564. return 1;
  565. }
  566. job->set_running_in_background(false);
  567. job->set_shell_did_continue(true);
  568. dbgln("Resuming {} ({})", job->pid(), job->cmd());
  569. warnln("Resuming job {} - {}", job->job_id(), job->cmd());
  570. tcsetpgrp(STDOUT_FILENO, job->pgid());
  571. tcsetpgrp(STDIN_FILENO, job->pgid());
  572. // Try using the PGID, but if that fails, just use the PID.
  573. if (killpg(job->pgid(), SIGCONT) < 0) {
  574. if (kill(job->pid(), SIGCONT) < 0) {
  575. perror("kill");
  576. return 1;
  577. }
  578. }
  579. block_on_job(job);
  580. if (job->exited())
  581. return job->exit_code();
  582. else
  583. return 0;
  584. }
  585. ErrorOr<int> Shell::builtin_disown(Main::Arguments arguments)
  586. {
  587. Vector<int> job_ids;
  588. Vector<bool> id_is_pid;
  589. Core::ArgsParser parser;
  590. parser.add_positional_argument(Core::ArgsParser::Arg {
  591. .help_string = "Job IDs or Jobspecs to disown",
  592. .name = "job-id",
  593. .min_values = 0,
  594. .max_values = INT_MAX,
  595. .accept_value = [&](StringView value) -> bool {
  596. // Check if it's a pid (i.e. literal integer)
  597. if (auto number = value.to_uint(); number.has_value()) {
  598. job_ids.append(number.value());
  599. id_is_pid.append(true);
  600. return true;
  601. }
  602. // Check if it's a jobspec
  603. if (auto id = resolve_job_spec(value); id.has_value()) {
  604. job_ids.append(id.value());
  605. id_is_pid.append(false);
  606. return true;
  607. }
  608. return false;
  609. } });
  610. if (!parser.parse(arguments, Core::ArgsParser::FailureBehavior::PrintUsage))
  611. return 1;
  612. if (job_ids.is_empty()) {
  613. job_ids.append(find_last_job_id());
  614. id_is_pid.append(false);
  615. }
  616. Vector<Job const*> jobs_to_disown;
  617. for (size_t i = 0; i < job_ids.size(); ++i) {
  618. auto id = job_ids[i];
  619. auto is_pid = id_is_pid[i];
  620. auto job = find_job(id, is_pid);
  621. if (!job)
  622. warnln("disown: Job with id/pid {} not found", id);
  623. else
  624. jobs_to_disown.append(job);
  625. }
  626. if (jobs_to_disown.is_empty()) {
  627. if (job_ids.is_empty())
  628. warnln("disown: No current job");
  629. // An error message has already been printed about the nonexistence of each listed job.
  630. return 1;
  631. }
  632. for (auto job : jobs_to_disown) {
  633. job->deactivate();
  634. if (!job->is_running_in_background())
  635. warnln("disown warning: Job {} is currently not running, 'kill -{} {}' to make it continue", job->job_id(), SIGCONT, job->pid());
  636. jobs.remove(job->pid());
  637. }
  638. return 0;
  639. }
  640. ErrorOr<int> Shell::builtin_history(Main::Arguments)
  641. {
  642. for (size_t i = 0; i < m_editor->history().size(); ++i) {
  643. printf("%6zu %s\n", i + 1, m_editor->history()[i].entry.characters());
  644. }
  645. return 0;
  646. }
  647. ErrorOr<int> Shell::builtin_jobs(Main::Arguments arguments)
  648. {
  649. bool list = false, show_pid = false;
  650. Core::ArgsParser parser;
  651. parser.add_option(list, "List all information about jobs", "list", 'l');
  652. parser.add_option(show_pid, "Display the PID of the jobs", "pid", 'p');
  653. if (!parser.parse(arguments, Core::ArgsParser::FailureBehavior::PrintUsage))
  654. return 1;
  655. Job::PrintStatusMode mode = Job::PrintStatusMode::Basic;
  656. if (show_pid)
  657. mode = Job::PrintStatusMode::OnlyPID;
  658. if (list)
  659. mode = Job::PrintStatusMode::ListAll;
  660. for (auto& it : jobs) {
  661. if (!it.value->print_status(mode))
  662. return 1;
  663. }
  664. return 0;
  665. }
  666. ErrorOr<int> Shell::builtin_popd(Main::Arguments arguments)
  667. {
  668. if (directory_stack.size() <= 1) {
  669. warnln("Shell: popd: directory stack empty");
  670. return 1;
  671. }
  672. bool should_not_switch = false;
  673. Core::ArgsParser parser;
  674. parser.add_option(should_not_switch, "Do not switch dirs", "no-switch", 'n');
  675. if (!parser.parse(arguments, Core::ArgsParser::FailureBehavior::PrintUsage))
  676. return 1;
  677. auto popped_path = directory_stack.take_last();
  678. if (should_not_switch)
  679. return 0;
  680. auto new_path = LexicalPath::canonicalized_path(popped_path);
  681. if (chdir(new_path.characters()) < 0) {
  682. warnln("chdir({}) failed: {}", new_path, strerror(errno));
  683. return 1;
  684. }
  685. cwd = new_path;
  686. return 0;
  687. }
  688. ErrorOr<int> Shell::builtin_pushd(Main::Arguments arguments)
  689. {
  690. StringBuilder path_builder;
  691. bool should_switch = true;
  692. // From the BASH reference manual: https://www.gnu.org/software/bash/manual/html_node/Directory-Stack-Builtins.html
  693. // With no arguments, pushd exchanges the top two directories and makes the new top the current directory.
  694. if (arguments.strings.size() == 1) {
  695. if (directory_stack.size() < 2) {
  696. warnln("pushd: no other directory");
  697. return 1;
  698. }
  699. DeprecatedString dir1 = directory_stack.take_first();
  700. DeprecatedString dir2 = directory_stack.take_first();
  701. directory_stack.insert(0, dir2);
  702. directory_stack.insert(1, dir1);
  703. int rc = chdir(dir2.characters());
  704. if (rc < 0) {
  705. warnln("chdir({}) failed: {}", dir2, strerror(errno));
  706. return 1;
  707. }
  708. cwd = dir2;
  709. return 0;
  710. }
  711. // Let's assume the user's typed in 'pushd <dir>'
  712. if (arguments.strings.size() == 2) {
  713. directory_stack.append(cwd.characters());
  714. if (arguments.strings[1].starts_with('/')) {
  715. path_builder.append(arguments.strings[1]);
  716. } else {
  717. path_builder.appendff("{}/{}", cwd, arguments.strings[1]);
  718. }
  719. } else if (arguments.strings.size() == 3) {
  720. directory_stack.append(cwd.characters());
  721. for (size_t i = 1; i < arguments.strings.size(); i++) {
  722. auto arg = arguments.strings[i];
  723. if (arg.starts_with('-')) {
  724. if (arg.starts_with('/')) {
  725. path_builder.append(arg);
  726. } else {
  727. path_builder.appendff("{}/{}", cwd, arg);
  728. }
  729. }
  730. if (arg == "-n"sv)
  731. should_switch = false;
  732. }
  733. }
  734. auto real_path = LexicalPath::canonicalized_path(path_builder.to_deprecated_string());
  735. struct stat st;
  736. int rc = stat(real_path.characters(), &st);
  737. if (rc < 0) {
  738. warnln("stat({}) failed: {}", real_path, strerror(errno));
  739. return 1;
  740. }
  741. if (!S_ISDIR(st.st_mode)) {
  742. warnln("Not a directory: {}", real_path);
  743. return 1;
  744. }
  745. if (should_switch) {
  746. int rc = chdir(real_path.characters());
  747. if (rc < 0) {
  748. warnln("chdir({}) failed: {}", real_path, strerror(errno));
  749. return 1;
  750. }
  751. cwd = real_path;
  752. }
  753. return 0;
  754. }
  755. ErrorOr<int> Shell::builtin_pwd(Main::Arguments)
  756. {
  757. print_path(cwd);
  758. fputc('\n', stdout);
  759. return 0;
  760. }
  761. ErrorOr<int> Shell::builtin_setopt(Main::Arguments arguments)
  762. {
  763. if (arguments.strings.size() == 1) {
  764. #define __ENUMERATE_SHELL_OPTION(name, default_, description) \
  765. if (options.name) \
  766. warnln("{}", #name);
  767. ENUMERATE_SHELL_OPTIONS();
  768. #undef __ENUMERATE_SHELL_OPTION
  769. }
  770. Core::ArgsParser parser;
  771. #define __ENUMERATE_SHELL_OPTION(name, default_, description) \
  772. bool name = false; \
  773. bool not_##name = false; \
  774. parser.add_option(name, "Enable: " description, #name, '\0'); \
  775. parser.add_option(not_##name, "Disable: " description, "no_" #name, '\0');
  776. ENUMERATE_SHELL_OPTIONS();
  777. #undef __ENUMERATE_SHELL_OPTION
  778. if (!parser.parse(arguments, Core::ArgsParser::FailureBehavior::PrintUsage))
  779. return 1;
  780. #define __ENUMERATE_SHELL_OPTION(name, default_, description) \
  781. if (name) \
  782. options.name = true; \
  783. if (not_##name) \
  784. options.name = false;
  785. ENUMERATE_SHELL_OPTIONS();
  786. #undef __ENUMERATE_SHELL_OPTION
  787. return 0;
  788. }
  789. ErrorOr<int> Shell::builtin_shift(Main::Arguments arguments)
  790. {
  791. int count = 1;
  792. Core::ArgsParser parser;
  793. parser.add_positional_argument(count, "Shift count", "count", Core::ArgsParser::Required::No);
  794. if (!parser.parse(arguments, Core::ArgsParser::FailureBehavior::PrintUsage))
  795. return 1;
  796. if (count < 1)
  797. return 0;
  798. auto argv_ = TRY(lookup_local_variable("ARGV"sv));
  799. if (!argv_) {
  800. warnln("shift: ARGV is unset");
  801. return 1;
  802. }
  803. if (!argv_->is_list())
  804. argv_ = adopt_ref(*new AST::ListValue({ const_cast<AST::Value&>(*argv_) }));
  805. auto& values = const_cast<AST::ListValue*>(static_cast<AST::ListValue const*>(argv_.ptr()))->values();
  806. if ((size_t)count > values.size()) {
  807. warnln("shift: shift count must not be greater than {}", values.size());
  808. return 1;
  809. }
  810. for (auto i = 0; i < count; ++i)
  811. (void)values.take_first();
  812. return 0;
  813. }
  814. ErrorOr<int> Shell::builtin_source(Main::Arguments arguments)
  815. {
  816. StringView file_to_source;
  817. Vector<StringView> args;
  818. Core::ArgsParser parser;
  819. parser.add_positional_argument(file_to_source, "File to read commands from", "path");
  820. parser.add_positional_argument(args, "ARGV for the sourced file", "args", Core::ArgsParser::Required::No);
  821. if (!parser.parse(arguments))
  822. return 1;
  823. auto previous_argv = TRY(lookup_local_variable("ARGV"sv));
  824. ScopeGuard guard { [&] {
  825. if (!args.is_empty())
  826. set_local_variable("ARGV", const_cast<AST::Value&>(*previous_argv));
  827. } };
  828. if (!args.is_empty()) {
  829. Vector<String> arguments;
  830. arguments.ensure_capacity(args.size());
  831. for (auto& arg : args)
  832. arguments.append(TRY(String::from_utf8(arg)));
  833. set_local_variable("ARGV", AST::make_ref_counted<AST::ListValue>(move(arguments)));
  834. }
  835. if (!run_file(file_to_source, true))
  836. return 126;
  837. return 0;
  838. }
  839. ErrorOr<int> Shell::builtin_time(Main::Arguments arguments)
  840. {
  841. Vector<StringView> args;
  842. int number_of_iterations = 1;
  843. Core::ArgsParser parser;
  844. parser.add_option(number_of_iterations, "Number of iterations", "iterations", 'n', "iterations");
  845. parser.set_stop_on_first_non_option(true);
  846. parser.add_positional_argument(args, "Command to execute with arguments", "command", Core::ArgsParser::Required::Yes);
  847. if (!parser.parse(arguments, Core::ArgsParser::FailureBehavior::PrintUsage))
  848. return 1;
  849. if (number_of_iterations < 1)
  850. return 1;
  851. AST::Command command;
  852. TRY(command.argv.try_ensure_capacity(args.size()));
  853. for (auto& arg : args)
  854. command.argv.append(TRY(String::from_utf8(arg)));
  855. auto commands = TRY(expand_aliases({ move(command) }));
  856. AK::Statistics iteration_times;
  857. int exit_code = 1;
  858. for (int i = 0; i < number_of_iterations; ++i) {
  859. auto timer = Core::ElapsedTimer::start_new();
  860. for (auto& job : run_commands(commands)) {
  861. block_on_job(job);
  862. exit_code = job->exit_code();
  863. }
  864. iteration_times.add(static_cast<float>(timer.elapsed()));
  865. }
  866. if (number_of_iterations == 1) {
  867. warnln("Time: {} ms", iteration_times.values().first());
  868. } else {
  869. AK::Statistics iteration_times_excluding_first;
  870. for (size_t i = 1; i < iteration_times.size(); i++)
  871. iteration_times_excluding_first.add(iteration_times.values()[i]);
  872. warnln("Timing report: {} ms", iteration_times.sum());
  873. warnln("==============");
  874. warnln("Command: {}", DeprecatedString::join(' ', arguments.strings));
  875. warnln("Average time: {:.2} ms (median: {}, stddev: {:.2}, min: {}, max:{})",
  876. iteration_times.average(), iteration_times.median(),
  877. iteration_times.standard_deviation(),
  878. iteration_times.min(), iteration_times.max());
  879. warnln("Excluding first: {:.2} ms (median: {}, stddev: {:.2}, min: {}, max:{})",
  880. iteration_times_excluding_first.average(), iteration_times_excluding_first.median(),
  881. iteration_times_excluding_first.standard_deviation(),
  882. iteration_times_excluding_first.min(), iteration_times_excluding_first.max());
  883. }
  884. return exit_code;
  885. }
  886. ErrorOr<int> Shell::builtin_umask(Main::Arguments arguments)
  887. {
  888. StringView mask_text;
  889. Core::ArgsParser parser;
  890. parser.add_positional_argument(mask_text, "New mask (omit to get current mask)", "octal-mask", Core::ArgsParser::Required::No);
  891. if (!parser.parse(arguments, Core::ArgsParser::FailureBehavior::PrintUsage))
  892. return 1;
  893. if (mask_text.is_empty()) {
  894. mode_t old_mask = umask(0);
  895. printf("%#o\n", old_mask);
  896. umask(old_mask);
  897. return 0;
  898. }
  899. unsigned mask = 0;
  900. auto matches = true;
  901. // FIXME: There's currently no way to parse an StringView as an octal integer,
  902. // when that becomes available, replace this code.
  903. for (auto byte : mask_text.bytes()) {
  904. if (isspace(byte))
  905. continue;
  906. if (!is_ascii_octal_digit(byte)) {
  907. matches = false;
  908. break;
  909. }
  910. mask = (mask << 3) + (byte - '0');
  911. }
  912. if (matches) {
  913. umask(mask);
  914. return 0;
  915. }
  916. warnln("umask: Invalid mask '{}'", mask_text);
  917. return 1;
  918. }
  919. ErrorOr<int> Shell::builtin_wait(Main::Arguments arguments)
  920. {
  921. Vector<int> job_ids;
  922. Vector<bool> id_is_pid;
  923. Core::ArgsParser parser;
  924. parser.add_positional_argument(Core::ArgsParser::Arg {
  925. .help_string = "Job IDs or Jobspecs to wait for",
  926. .name = "job-id",
  927. .min_values = 0,
  928. .max_values = INT_MAX,
  929. .accept_value = [&](StringView value) -> bool {
  930. // Check if it's a pid (i.e. literal integer)
  931. if (auto number = value.to_uint(); number.has_value()) {
  932. job_ids.append(number.value());
  933. id_is_pid.append(true);
  934. return true;
  935. }
  936. // Check if it's a jobspec
  937. if (auto id = resolve_job_spec(value); id.has_value()) {
  938. job_ids.append(id.value());
  939. id_is_pid.append(false);
  940. return true;
  941. }
  942. return false;
  943. } });
  944. if (!parser.parse(arguments, Core::ArgsParser::FailureBehavior::PrintUsage))
  945. return 1;
  946. Vector<NonnullRefPtr<Job>> jobs_to_wait_for;
  947. for (size_t i = 0; i < job_ids.size(); ++i) {
  948. auto id = job_ids[i];
  949. auto is_pid = id_is_pid[i];
  950. auto job = find_job(id, is_pid);
  951. if (!job)
  952. warnln("wait: Job with id/pid {} not found", id);
  953. else
  954. jobs_to_wait_for.append(*job);
  955. }
  956. if (job_ids.is_empty()) {
  957. for (auto const& it : jobs)
  958. jobs_to_wait_for.append(it.value);
  959. }
  960. for (auto& job : jobs_to_wait_for) {
  961. job->set_running_in_background(false);
  962. block_on_job(job);
  963. }
  964. return 0;
  965. }
  966. ErrorOr<int> Shell::builtin_unset(Main::Arguments arguments)
  967. {
  968. Vector<DeprecatedString> vars;
  969. Core::ArgsParser parser;
  970. parser.add_positional_argument(vars, "List of variables", "variables", Core::ArgsParser::Required::Yes);
  971. if (!parser.parse(arguments, Core::ArgsParser::FailureBehavior::PrintUsage))
  972. return 1;
  973. bool did_touch_path = false;
  974. for (auto& value : vars) {
  975. if (!did_touch_path && value == "PATH"sv)
  976. did_touch_path = true;
  977. if (TRY(lookup_local_variable(value)) != nullptr) {
  978. unset_local_variable(value);
  979. } else {
  980. unsetenv(value.characters());
  981. }
  982. }
  983. if (did_touch_path)
  984. cache_path();
  985. return 0;
  986. }
  987. ErrorOr<int> Shell::builtin_not(Main::Arguments arguments)
  988. {
  989. Vector<StringView> args;
  990. Core::ArgsParser parser;
  991. parser.set_stop_on_first_non_option(true);
  992. parser.add_positional_argument(args, "Command to run followed by its arguments", "string");
  993. if (!parser.parse(arguments, Core::ArgsParser::FailureBehavior::Ignore))
  994. return 1;
  995. AST::Command command;
  996. TRY(command.argv.try_ensure_capacity(args.size()));
  997. for (auto& arg : args)
  998. command.argv.unchecked_append(TRY(String::from_utf8(arg)));
  999. auto commands = TRY(expand_aliases({ move(command) }));
  1000. int exit_code = 1;
  1001. auto found_a_job = false;
  1002. for (auto& job : run_commands(commands)) {
  1003. found_a_job = true;
  1004. block_on_job(job);
  1005. exit_code = job->exit_code();
  1006. }
  1007. // In case it was a function.
  1008. if (!found_a_job)
  1009. exit_code = last_return_code.value_or(0);
  1010. return exit_code == 0 ? 1 : 0;
  1011. }
  1012. ErrorOr<int> Shell::builtin_kill(Main::Arguments arguments)
  1013. {
  1014. // Simply translate the arguments and pass them to `kill'
  1015. Vector<String> replaced_values;
  1016. auto kill_path = Core::DeprecatedFile::resolve_executable_from_environment("kill"sv);
  1017. if (!kill_path.has_value()) {
  1018. warnln("kill: `kill' not found in PATH");
  1019. return 126;
  1020. }
  1021. replaced_values.append(TRY(String::from_deprecated_string(kill_path.release_value())));
  1022. for (size_t i = 1; i < arguments.strings.size(); ++i) {
  1023. if (auto job_id = resolve_job_spec(arguments.strings[i]); job_id.has_value()) {
  1024. auto job = find_job(job_id.value());
  1025. if (job) {
  1026. replaced_values.append(TRY(String::number(job->pid())));
  1027. } else {
  1028. warnln("kill: Job with pid {} not found", job_id.value());
  1029. return 1;
  1030. }
  1031. } else {
  1032. replaced_values.append(TRY(String::from_utf8(arguments.strings[i])));
  1033. }
  1034. }
  1035. // Now just run `kill'
  1036. AST::Command command;
  1037. command.argv = move(replaced_values);
  1038. command.position = m_source_position.has_value() ? m_source_position->position : Optional<AST::Position> {};
  1039. auto exit_code = 1;
  1040. auto job_result = run_command(command);
  1041. if (job_result.is_error()) {
  1042. warnln("kill: Failed to run {}: {}", command.argv.first(), job_result.error());
  1043. return exit_code;
  1044. }
  1045. if (auto job = job_result.release_value()) {
  1046. block_on_job(job);
  1047. exit_code = job->exit_code();
  1048. }
  1049. return exit_code;
  1050. }
  1051. ErrorOr<bool> Shell::run_builtin(const AST::Command& command, Vector<NonnullRefPtr<AST::Rewiring>> const& rewirings, int& retval)
  1052. {
  1053. if (command.argv.is_empty())
  1054. return false;
  1055. if (!has_builtin(command.argv.first()))
  1056. return false;
  1057. Vector<StringView> arguments;
  1058. TRY(arguments.try_ensure_capacity(command.argv.size()));
  1059. for (auto& arg : command.argv)
  1060. arguments.unchecked_append(arg);
  1061. Main::Arguments arguments_object {
  1062. .argc = 0,
  1063. .argv = nullptr,
  1064. .strings = arguments
  1065. };
  1066. StringView name = command.argv.first();
  1067. SavedFileDescriptors fds { rewirings };
  1068. for (auto& rewiring : rewirings) {
  1069. int rc = dup2(rewiring->old_fd, rewiring->new_fd);
  1070. if (rc < 0) {
  1071. perror("dup2(run)");
  1072. return false;
  1073. }
  1074. }
  1075. Core::EventLoop loop;
  1076. setup_signals();
  1077. if (name == ":"sv)
  1078. name = "noop"sv;
  1079. #define __ENUMERATE_SHELL_BUILTIN(builtin) \
  1080. if (name == #builtin) { \
  1081. retval = TRY(builtin_##builtin(arguments_object)); \
  1082. if (!has_error(ShellError::None)) \
  1083. raise_error(m_error, m_error_description, command.position); \
  1084. fflush(stdout); \
  1085. fflush(stderr); \
  1086. return true; \
  1087. }
  1088. ENUMERATE_SHELL_BUILTINS();
  1089. #undef __ENUMERATE_SHELL_BUILTIN
  1090. return false;
  1091. }
  1092. ErrorOr<int> Shell::builtin_argsparser_parse(Main::Arguments arguments)
  1093. {
  1094. // argsparser_parse
  1095. // --add-option variable [--type (bool | string | i32 | u32 | double | size)] --help-string "" --long-name "" --short-name "" [--value-name "" <if not --type bool>] --list
  1096. // --add-positional-argument variable [--type (bool | string | i32 | u32 | double | size)] ([--min n] [--max n] | [--required]) --help-string "" --value-name ""
  1097. // [--general-help ""]
  1098. // [--stop-on-first-non-option]
  1099. // --
  1100. // $args_to_parse
  1101. Core::ArgsParser parser;
  1102. Core::ArgsParser user_parser;
  1103. Vector<StringView> descriptors;
  1104. Variant<Core::ArgsParser::Option, Core::ArgsParser::Arg, Empty> current;
  1105. DeprecatedString help_string_storage;
  1106. DeprecatedString long_name_storage;
  1107. DeprecatedString value_name_storage;
  1108. DeprecatedString name_storage;
  1109. DeprecatedString current_variable;
  1110. // if max > 1 or min < 1, or explicit `--list`.
  1111. bool treat_arg_as_list = false;
  1112. enum class Type {
  1113. Bool,
  1114. String,
  1115. I32,
  1116. U32,
  1117. Double,
  1118. Size,
  1119. };
  1120. auto type = Type::String;
  1121. auto try_convert = [](StringView value, Type type) -> ErrorOr<Optional<RefPtr<AST::Value>>> {
  1122. switch (type) {
  1123. case Type::Bool:
  1124. return AST::make_ref_counted<AST::StringValue>(TRY("true"_string));
  1125. case Type::String:
  1126. return AST::make_ref_counted<AST::StringValue>(TRY(String::from_utf8(value)));
  1127. case Type::I32:
  1128. if (auto number = value.to_int(); number.has_value())
  1129. return AST::make_ref_counted<AST::StringValue>(TRY(String::number(*number)));
  1130. warnln("Invalid value for type i32: {}", value);
  1131. return OptionalNone {};
  1132. case Type::U32:
  1133. case Type::Size:
  1134. if (auto number = value.to_uint(); number.has_value())
  1135. return AST::make_ref_counted<AST::StringValue>(TRY(String::number(*number)));
  1136. warnln("Invalid value for type u32|size: {}", value);
  1137. return OptionalNone {};
  1138. case Type::Double: {
  1139. DeprecatedString string = value;
  1140. char* endptr = nullptr;
  1141. auto number = strtod(string.characters(), &endptr);
  1142. if (endptr != string.characters() + string.length()) {
  1143. warnln("Invalid value for type double: {}", value);
  1144. return OptionalNone {};
  1145. }
  1146. return AST::make_ref_counted<AST::StringValue>(TRY(String::number(number)));
  1147. }
  1148. default:
  1149. VERIFY_NOT_REACHED();
  1150. }
  1151. };
  1152. auto enlist = [&](auto name, auto value) -> ErrorOr<NonnullRefPtr<AST::Value>> {
  1153. auto variable = TRY(lookup_local_variable(name));
  1154. if (variable) {
  1155. auto list = TRY(const_cast<AST::Value&>(*variable).resolve_as_list(*this));
  1156. auto new_value = TRY(value->resolve_as_string(*this));
  1157. list.append(move(new_value));
  1158. return try_make_ref_counted<AST::ListValue>(move(list));
  1159. }
  1160. return *value;
  1161. };
  1162. // FIXME: We cannot return ErrorOr<T> from Core::ArgsParser::Option::accept_value(), fix the MUST's here whenever that function is ErrorOr-aware.
  1163. auto commit = [&] {
  1164. return current.visit(
  1165. [&](Core::ArgsParser::Option& option) {
  1166. if (!option.long_name && !option.short_name) {
  1167. warnln("Defined option must have at least one of --long-name or --short-name");
  1168. return false;
  1169. }
  1170. option.accept_value = [&, current_variable, treat_arg_as_list, type](StringView value) {
  1171. auto result = MUST(try_convert(value, type));
  1172. if (result.has_value()) {
  1173. auto value = result.release_value();
  1174. if (treat_arg_as_list)
  1175. value = MUST(enlist(current_variable, move(value)));
  1176. this->set_local_variable(current_variable, move(value), true);
  1177. return true;
  1178. }
  1179. return false;
  1180. };
  1181. user_parser.add_option(move(option));
  1182. type = Type::String;
  1183. treat_arg_as_list = false;
  1184. return true;
  1185. },
  1186. [&](Core::ArgsParser::Arg& arg) {
  1187. if (!arg.name) {
  1188. warnln("Defined positional argument must have a name");
  1189. return false;
  1190. }
  1191. arg.accept_value = [&, current_variable, treat_arg_as_list, type](StringView value) {
  1192. auto result = MUST(try_convert(value, type));
  1193. if (result.has_value()) {
  1194. auto value = result.release_value();
  1195. if (treat_arg_as_list)
  1196. value = MUST(enlist(current_variable, move(value)));
  1197. this->set_local_variable(current_variable, move(value), true);
  1198. return true;
  1199. }
  1200. return false;
  1201. };
  1202. user_parser.add_positional_argument(move(arg));
  1203. type = Type::String;
  1204. treat_arg_as_list = false;
  1205. return true;
  1206. },
  1207. [&](Empty) {
  1208. return true;
  1209. });
  1210. };
  1211. parser.add_option(Core::ArgsParser::Option {
  1212. .argument_mode = Core::ArgsParser::OptionArgumentMode::None,
  1213. .help_string = "Stop processing descriptors after a non-argument parameter is seen",
  1214. .long_name = "stop-on-first-non-option",
  1215. .accept_value = [&](auto) {
  1216. user_parser.set_stop_on_first_non_option(true);
  1217. return true;
  1218. },
  1219. });
  1220. parser.add_option(Core::ArgsParser::Option {
  1221. .argument_mode = Core::ArgsParser::OptionArgumentMode::Required,
  1222. .help_string = "Set the general help string for the parser",
  1223. .long_name = "general-help",
  1224. .value_name = "string",
  1225. .accept_value = [&](StringView value) {
  1226. VERIFY(strlen(value.characters_without_null_termination()) == value.length());
  1227. user_parser.set_general_help(value.characters_without_null_termination());
  1228. return true;
  1229. },
  1230. });
  1231. parser.add_option(Core::ArgsParser::Option {
  1232. .argument_mode = Core::ArgsParser::OptionArgumentMode::Required,
  1233. .help_string = "Start describing an option",
  1234. .long_name = "add-option",
  1235. .value_name = "variable-name",
  1236. .accept_value = [&](auto name) {
  1237. if (!commit())
  1238. return false;
  1239. current = Core::ArgsParser::Option {};
  1240. current_variable = name;
  1241. if (current_variable.is_empty() || !all_of(current_variable, [](auto ch) { return ch == '_' || isalnum(ch); })) {
  1242. warnln("Option variable name must be a valid identifier");
  1243. return false;
  1244. }
  1245. return true;
  1246. },
  1247. });
  1248. parser.add_option(Core::ArgsParser::Option {
  1249. .argument_mode = Core::ArgsParser::OptionArgumentMode::None,
  1250. .help_string = "Accept multiple of the current option being given",
  1251. .long_name = "list",
  1252. .accept_value = [&](auto) {
  1253. if (!current.has<Core::ArgsParser::Option>()) {
  1254. warnln("Must be defining an option to use --list");
  1255. return false;
  1256. }
  1257. treat_arg_as_list = true;
  1258. return true;
  1259. },
  1260. });
  1261. parser.add_option(Core::ArgsParser::Option {
  1262. .argument_mode = Core::ArgsParser::OptionArgumentMode::Required,
  1263. .help_string = "Define the type of the option or argument being described",
  1264. .long_name = "type",
  1265. .value_name = "type",
  1266. .accept_value = [&](StringView ty) {
  1267. if (current.has<Empty>()) {
  1268. warnln("Must be defining an argument or option to use --type");
  1269. return false;
  1270. }
  1271. if (ty == "bool") {
  1272. if (auto option = current.get_pointer<Core::ArgsParser::Option>()) {
  1273. if (option->value_name != nullptr) {
  1274. warnln("Type 'bool' does not apply to options with a value (value name is set to {})", option->value_name);
  1275. return false;
  1276. }
  1277. }
  1278. type = Type::Bool;
  1279. } else if (ty == "string") {
  1280. type = Type::String;
  1281. } else if (ty == "i32") {
  1282. type = Type::I32;
  1283. } else if (ty == "u32") {
  1284. type = Type::U32;
  1285. } else if (ty == "double") {
  1286. type = Type::Double;
  1287. } else if (ty == "size") {
  1288. type = Type::Size;
  1289. } else {
  1290. warnln("Invalid type '{}', expected one of bool | string | i32 | u32 | double | size", ty);
  1291. return false;
  1292. }
  1293. if (type == Type::Bool) {
  1294. set_local_variable(
  1295. current_variable,
  1296. make_ref_counted<AST::StringValue>(MUST("false"_string)),
  1297. true);
  1298. }
  1299. return true;
  1300. },
  1301. });
  1302. parser.add_option(Core::ArgsParser::Option {
  1303. .argument_mode = Core::ArgsParser::OptionArgumentMode::Required,
  1304. .help_string = "Set the help string of the option or argument being defined",
  1305. .long_name = "help-string",
  1306. .value_name = "string",
  1307. .accept_value = [&](StringView value) {
  1308. return current.visit(
  1309. [](Empty) {
  1310. warnln("Must be defining an option or argument to use --help-string");
  1311. return false;
  1312. },
  1313. [&](auto& option) {
  1314. help_string_storage = value;
  1315. option.help_string = help_string_storage.characters();
  1316. return true;
  1317. });
  1318. },
  1319. });
  1320. parser.add_option(Core::ArgsParser::Option {
  1321. .argument_mode = Core::ArgsParser::OptionArgumentMode::Required,
  1322. .help_string = "Set the long name of the option being defined",
  1323. .long_name = "long-name",
  1324. .value_name = "name",
  1325. .accept_value = [&](StringView value) {
  1326. auto option = current.get_pointer<Core::ArgsParser::Option>();
  1327. if (!option) {
  1328. warnln("Must be defining an option to use --long-name");
  1329. return false;
  1330. }
  1331. if (option->long_name) {
  1332. warnln("Repeated application of --long-name is not allowed, current option has long name set to \"{}\"", option->long_name);
  1333. return false;
  1334. }
  1335. long_name_storage = value;
  1336. option->long_name = long_name_storage.characters();
  1337. return true;
  1338. },
  1339. });
  1340. parser.add_option(Core::ArgsParser::Option {
  1341. .argument_mode = Core::ArgsParser::OptionArgumentMode::Required,
  1342. .help_string = "Set the short name of the option being defined",
  1343. .long_name = "short-name",
  1344. .value_name = "char",
  1345. .accept_value = [&](StringView value) {
  1346. auto option = current.get_pointer<Core::ArgsParser::Option>();
  1347. if (!option) {
  1348. warnln("Must be defining an option to use --short-name");
  1349. return false;
  1350. }
  1351. if (value.length() != 1) {
  1352. warnln("Option short name ('{}') must be exactly one character long", value);
  1353. return false;
  1354. }
  1355. if (option->short_name) {
  1356. warnln("Repeated application of --short-name is not allowed, current option has short name set to '{}'", option->short_name);
  1357. return false;
  1358. }
  1359. option->short_name = value[0];
  1360. return true;
  1361. },
  1362. });
  1363. parser.add_option(Core::ArgsParser::Option {
  1364. .argument_mode = Core::ArgsParser::OptionArgumentMode::Required,
  1365. .help_string = "Set the value name of the option being defined",
  1366. .long_name = "value-name",
  1367. .value_name = "string",
  1368. .accept_value = [&](StringView value) {
  1369. return current.visit(
  1370. [](Empty) {
  1371. warnln("Must be defining an option or a positional argument to use --value-name");
  1372. return false;
  1373. },
  1374. [&](Core::ArgsParser::Option& option) {
  1375. if (option.value_name) {
  1376. warnln("Repeated application of --value-name is not allowed, current option has value name set to \"{}\"", option.value_name);
  1377. return false;
  1378. }
  1379. if (type == Type::Bool) {
  1380. warnln("Options of type bool cannot have a value name");
  1381. return false;
  1382. }
  1383. value_name_storage = value;
  1384. option.value_name = value_name_storage.characters();
  1385. return true;
  1386. },
  1387. [&](Core::ArgsParser::Arg& arg) {
  1388. if (arg.name) {
  1389. warnln("Repeated application of --value-name is not allowed, current argument has value name set to \"{}\"", arg.name);
  1390. return false;
  1391. }
  1392. name_storage = value;
  1393. arg.name = name_storage.characters();
  1394. return true;
  1395. });
  1396. },
  1397. });
  1398. parser.add_option(Core::ArgsParser::Option {
  1399. .argument_mode = Core::ArgsParser::OptionArgumentMode::Required,
  1400. .help_string = "Start describing a positional argument",
  1401. .long_name = "add-positional-argument",
  1402. .value_name = "variable",
  1403. .accept_value = [&](auto value) {
  1404. if (!commit())
  1405. return false;
  1406. current = Core::ArgsParser::Arg {};
  1407. current_variable = value;
  1408. if (current_variable.is_empty() || !all_of(current_variable, [](auto ch) { return ch == '_' || isalnum(ch); })) {
  1409. warnln("Argument variable name must be a valid identifier");
  1410. return false;
  1411. }
  1412. return true;
  1413. },
  1414. });
  1415. parser.add_option(Core::ArgsParser::Option {
  1416. .argument_mode = Core::ArgsParser::OptionArgumentMode::Required,
  1417. .help_string = "Set the minimum required number of positional descriptors for the argument being described",
  1418. .long_name = "min",
  1419. .value_name = "n",
  1420. .accept_value = [&](StringView value) {
  1421. auto arg = current.get_pointer<Core::ArgsParser::Arg>();
  1422. if (!arg) {
  1423. warnln("Must be describing a positional argument to use --min");
  1424. return false;
  1425. }
  1426. auto number = value.to_uint();
  1427. if (!number.has_value()) {
  1428. warnln("Invalid value for --min: '{}', expected a non-negative number", value);
  1429. return false;
  1430. }
  1431. if (static_cast<unsigned>(arg->max_values) < *number) {
  1432. warnln("Invalid value for --min: {}, min must not be larger than max ({})", *number, arg->max_values);
  1433. return false;
  1434. }
  1435. arg->min_values = *number;
  1436. treat_arg_as_list = arg->max_values > 1 || arg->min_values < 1;
  1437. return true;
  1438. },
  1439. });
  1440. parser.add_option(Core::ArgsParser::Option {
  1441. .argument_mode = Core::ArgsParser::OptionArgumentMode::Required,
  1442. .help_string = "Set the maximum required number of positional descriptors for the argument being described",
  1443. .long_name = "max",
  1444. .value_name = "n",
  1445. .accept_value = [&](StringView value) {
  1446. auto arg = current.get_pointer<Core::ArgsParser::Arg>();
  1447. if (!arg) {
  1448. warnln("Must be describing a positional argument to use --max");
  1449. return false;
  1450. }
  1451. auto number = value.to_uint();
  1452. if (!number.has_value()) {
  1453. warnln("Invalid value for --max: '{}', expected a non-negative number", value);
  1454. return false;
  1455. }
  1456. if (static_cast<unsigned>(arg->min_values) > *number) {
  1457. warnln("Invalid value for --max: {}, max must not be smaller than min ({})", *number, arg->min_values);
  1458. return false;
  1459. }
  1460. arg->max_values = *number;
  1461. treat_arg_as_list = arg->max_values > 1 || arg->min_values < 1;
  1462. return true;
  1463. },
  1464. });
  1465. parser.add_option(Core::ArgsParser::Option {
  1466. .argument_mode = Core::ArgsParser::OptionArgumentMode::None,
  1467. .help_string = "Mark the positional argument being described as required (shorthand for --min 1)",
  1468. .long_name = "required",
  1469. .accept_value = [&](auto) {
  1470. auto arg = current.get_pointer<Core::ArgsParser::Arg>();
  1471. if (!arg) {
  1472. warnln("Must be describing a positional argument to use --required");
  1473. return false;
  1474. }
  1475. arg->min_values = 1;
  1476. if (arg->max_values < arg->min_values)
  1477. arg->max_values = 1;
  1478. treat_arg_as_list = arg->max_values > 1 || arg->min_values < 1;
  1479. return true;
  1480. },
  1481. });
  1482. parser.add_positional_argument(descriptors, "Arguments to parse via the described ArgsParser configuration", "arg", Core::ArgsParser::Required::No);
  1483. if (!parser.parse(arguments, Core::ArgsParser::FailureBehavior::Ignore))
  1484. return 2;
  1485. if (!commit())
  1486. return 2;
  1487. if (!user_parser.parse(descriptors, Core::ArgsParser::FailureBehavior::Ignore))
  1488. return 1;
  1489. return 0;
  1490. }
  1491. bool Shell::has_builtin(StringView name) const
  1492. {
  1493. if (name == ":"sv)
  1494. return true;
  1495. #define __ENUMERATE_SHELL_BUILTIN(builtin) \
  1496. if (name == #builtin) { \
  1497. return true; \
  1498. }
  1499. ENUMERATE_SHELL_BUILTINS();
  1500. #undef __ENUMERATE_SHELL_BUILTIN
  1501. return false;
  1502. }
  1503. }