#include #include #include #include #include #include #include #include #include #include #include #include #include #include #include namespace Kernel { template static void TTY_PRINT(Args&&... args) { BAN::String message = BAN::String::formatted(BAN::forward(args)...); MUST(Process::current().write(STDOUT_FILENO, message.data(), message.size())); } template static void TTY_PRINTLN(Args&&... args) { TTY_PRINT(BAN::forward(args)...); MUST(Process::current().write(STDOUT_FILENO, "\n", 1)); } static auto s_default_prompt = "\\[\e[32m\\]user\\[\e[m\\]:\\[\e[34m\\]\\w\\[\e[m\\]# "sv; static const char* mode_string(Inode::Mode mode) { static char buffer[11] {}; buffer[0] = mode.ifdir() ? 'd' : mode.ifblk() ? 'b' : mode.ifchr() ? 'c' : mode.iflnk() ? 'l' : '-'; buffer[1] = (mode.mode & Inode::Mode::IRUSR) ? 'r' : '-'; buffer[2] = (mode.mode & Inode::Mode::IWUSR) ? 'w' : '-'; buffer[3] = (mode.mode & Inode::Mode::IXUSR) ? 'x' : '-'; buffer[4] = (mode.mode & Inode::Mode::IRGRP) ? 'r' : '-'; buffer[5] = (mode.mode & Inode::Mode::IWGRP) ? 'w' : '-'; buffer[6] = (mode.mode & Inode::Mode::IXGRP) ? 'x' : '-'; buffer[7] = (mode.mode & Inode::Mode::IROTH) ? 'r' : '-'; buffer[8] = (mode.mode & Inode::Mode::IWOTH) ? 'w' : '-'; buffer[9] = (mode.mode & Inode::Mode::IXOTH) ? 'x' : '-'; return (const char*)buffer; }; Shell::Shell() { MUST(Process::current().open("/dev/tty1", O_RDONLY)); MUST(Process::current().open("/dev/tty1", O_WRONLY)); MUST(Process::current().open("/dev/tty1", O_WRONLY)); MUST(set_prompt(s_default_prompt)); MUST(m_buffer.push_back(""sv)); MUST(Process::current().set_termios(termios { .canonical = false, .echo = false })); } BAN::ErrorOr Shell::set_prompt(BAN::StringView prompt) { m_prompt_string = prompt; TRY(update_prompt()); return {}; } BAN::ErrorOr Shell::update_prompt() { m_prompt_length = 0; m_prompt.clear(); bool skipping = false; for (size_t i = 0; i < m_prompt_string.size(); i++) { if (i < m_prompt_string.size() - 1 && m_prompt_string[i] == '\\') { switch (m_prompt_string[i + 1]) { case '[': skipping = true; break; case ']': skipping = false; break; case 'w': { auto working_directory = TRY(Process::current().working_directory()); TRY(m_prompt.append(working_directory)); m_prompt_length += working_directory.size(); break; } default: dprintln("unknown escape character '{}' in shell prompt", m_prompt_string[i + 1]); break; } i++; continue; } TRY(m_prompt.push_back(m_prompt_string[i])); if (!skipping) m_prompt_length++; } return {}; } void Shell::run() { auto getch = [this] { uint8_t ch; MUST(Process::current().read(STDIN_FILENO, &ch, 1)); return ch; }; MUST(m_buffer.push_back(""sv)); TTY_PRINT("{}", m_prompt); for (;;) { BAN::String& current = m_buffer[m_cursor_pos.line]; uint8_t ch = getch(); if (ch == '\b') { if (!current.empty()) { while ((current.back() & 0xC0) == 0x80) current.pop_back(); current.pop_back(); MUST(Process::current().write(STDOUT_FILENO, "\b \b", 3)); } continue; } if (ch == '\e') { bool handled = false; if (getch() == '[') { handled = true; switch (getch()) { case 'A': // Up if (m_cursor_pos.line > 0) { m_cursor_pos.line--; TTY_PRINT("\e[G{}{}\e[K", m_prompt, m_buffer[m_cursor_pos.line]); } break; case 'B': // Down if (m_cursor_pos.line < m_buffer.size() - 1) { m_cursor_pos.line++; TTY_PRINT("\e[G{}{}\e[K", m_prompt, m_buffer[m_cursor_pos.line]); } break; case 'C': // Right break; case 'D': // Left break; default: handled = false; } } if (!handled) while (!isalpha(ch)) ch = getch(); continue; } MUST(Process::current().write(STDOUT_FILENO, &ch, 1)); if (ch != '\n') { MUST(current.push_back(ch)); continue; } if (auto res = process_command(parse_arguments(current)); res.is_error()) TTY_PRINTLN("{}", res.error()); TTY_PRINT("{}", m_prompt); if (!current.empty()) { MUST(m_old_buffer.push_back(current)); m_buffer = m_old_buffer; MUST(m_buffer.push_back(""sv)); m_cursor_pos.line = m_buffer.size() - 1; } } } BAN::Vector Shell::parse_arguments(BAN::StringView command) const { BAN::Vector result; while (!command.empty()) { while (!command.empty() && isspace(command.front())) command = command.substring(1); if (command.empty()) break; MUST(result.push_back(""sv)); char quoted = '\0'; bool escape = false; while (!command.empty()) { char ch = command.front(); switch (ch) { case '"': case '\'': if (!quoted) quoted = ch; else if (ch == quoted) quoted = '\0'; else goto default_case; break; case '\\': if (escape) goto default_case; escape = true; break; default: default_case: if (isspace(ch) && !quoted && !escape) goto argument_done; if (quoted && escape) { switch (ch) { case 'f': MUST(result.back().push_back('\f')); break; case 'n': MUST(result.back().push_back('\n')); break; case 'r': MUST(result.back().push_back('\r')); break; case 't': MUST(result.back().push_back('\t')); break; case 'v': MUST(result.back().push_back('\v')); break; case '"': MUST(result.back().push_back('"')); break; case '\'': MUST(result.back().push_back('\'')); break; case '\\': MUST(result.back().push_back('\\')); break; default: char buffer[3] { '\\', ch, '\0' }; MUST(result.back().append(buffer)); break; } } else { MUST(result.back().push_back(ch)); } escape = false; break; } command = command.substring(1); } argument_done: continue; } return result; } static uint32_t const crc32_table[256] = { 0x00000000, 0x04c11db7, 0x09823b6e, 0x0d4326d9, 0x130476dc, 0x17c56b6b, 0x1a864db2, 0x1e475005, 0x2608edb8, 0x22c9f00f, 0x2f8ad6d6, 0x2b4bcb61, 0x350c9b64, 0x31cd86d3, 0x3c8ea00a, 0x384fbdbd, 0x4c11db70, 0x48d0c6c7, 0x4593e01e, 0x4152fda9, 0x5f15adac, 0x5bd4b01b, 0x569796c2, 0x52568b75, 0x6a1936c8, 0x6ed82b7f, 0x639b0da6, 0x675a1011, 0x791d4014, 0x7ddc5da3, 0x709f7b7a, 0x745e66cd, 0x9823b6e0, 0x9ce2ab57, 0x91a18d8e, 0x95609039, 0x8b27c03c, 0x8fe6dd8b, 0x82a5fb52, 0x8664e6e5, 0xbe2b5b58, 0xbaea46ef, 0xb7a96036, 0xb3687d81, 0xad2f2d84, 0xa9ee3033, 0xa4ad16ea, 0xa06c0b5d, 0xd4326d90, 0xd0f37027, 0xddb056fe, 0xd9714b49, 0xc7361b4c, 0xc3f706fb, 0xceb42022, 0xca753d95, 0xf23a8028, 0xf6fb9d9f, 0xfbb8bb46, 0xff79a6f1, 0xe13ef6f4, 0xe5ffeb43, 0xe8bccd9a, 0xec7dd02d, 0x34867077, 0x30476dc0, 0x3d044b19, 0x39c556ae, 0x278206ab, 0x23431b1c, 0x2e003dc5, 0x2ac12072, 0x128e9dcf, 0x164f8078, 0x1b0ca6a1, 0x1fcdbb16, 0x018aeb13, 0x054bf6a4, 0x0808d07d, 0x0cc9cdca, 0x7897ab07, 0x7c56b6b0, 0x71159069, 0x75d48dde, 0x6b93dddb, 0x6f52c06c, 0x6211e6b5, 0x66d0fb02, 0x5e9f46bf, 0x5a5e5b08, 0x571d7dd1, 0x53dc6066, 0x4d9b3063, 0x495a2dd4, 0x44190b0d, 0x40d816ba, 0xaca5c697, 0xa864db20, 0xa527fdf9, 0xa1e6e04e, 0xbfa1b04b, 0xbb60adfc, 0xb6238b25, 0xb2e29692, 0x8aad2b2f, 0x8e6c3698, 0x832f1041, 0x87ee0df6, 0x99a95df3, 0x9d684044, 0x902b669d, 0x94ea7b2a, 0xe0b41de7, 0xe4750050, 0xe9362689, 0xedf73b3e, 0xf3b06b3b, 0xf771768c, 0xfa325055, 0xfef34de2, 0xc6bcf05f, 0xc27dede8, 0xcf3ecb31, 0xcbffd686, 0xd5b88683, 0xd1799b34, 0xdc3abded, 0xd8fba05a, 0x690ce0ee, 0x6dcdfd59, 0x608edb80, 0x644fc637, 0x7a089632, 0x7ec98b85, 0x738aad5c, 0x774bb0eb, 0x4f040d56, 0x4bc510e1, 0x46863638, 0x42472b8f, 0x5c007b8a, 0x58c1663d, 0x558240e4, 0x51435d53, 0x251d3b9e, 0x21dc2629, 0x2c9f00f0, 0x285e1d47, 0x36194d42, 0x32d850f5, 0x3f9b762c, 0x3b5a6b9b, 0x0315d626, 0x07d4cb91, 0x0a97ed48, 0x0e56f0ff, 0x1011a0fa, 0x14d0bd4d, 0x19939b94, 0x1d528623, 0xf12f560e, 0xf5ee4bb9, 0xf8ad6d60, 0xfc6c70d7, 0xe22b20d2, 0xe6ea3d65, 0xeba91bbc, 0xef68060b, 0xd727bbb6, 0xd3e6a601, 0xdea580d8, 0xda649d6f, 0xc423cd6a, 0xc0e2d0dd, 0xcda1f604, 0xc960ebb3, 0xbd3e8d7e, 0xb9ff90c9, 0xb4bcb610, 0xb07daba7, 0xae3afba2, 0xaafbe615, 0xa7b8c0cc, 0xa379dd7b, 0x9b3660c6, 0x9ff77d71, 0x92b45ba8, 0x9675461f, 0x8832161a, 0x8cf30bad, 0x81b02d74, 0x857130c3, 0x5d8a9099, 0x594b8d2e, 0x5408abf7, 0x50c9b640, 0x4e8ee645, 0x4a4ffbf2, 0x470cdd2b, 0x43cdc09c, 0x7b827d21, 0x7f436096, 0x7200464f, 0x76c15bf8, 0x68860bfd, 0x6c47164a, 0x61043093, 0x65c52d24, 0x119b4be9, 0x155a565e, 0x18197087, 0x1cd86d30, 0x029f3d35, 0x065e2082, 0x0b1d065b, 0x0fdc1bec, 0x3793a651, 0x3352bbe6, 0x3e119d3f, 0x3ad08088, 0x2497d08d, 0x2056cd3a, 0x2d15ebe3, 0x29d4f654, 0xc5a92679, 0xc1683bce, 0xcc2b1d17, 0xc8ea00a0, 0xd6ad50a5, 0xd26c4d12, 0xdf2f6bcb, 0xdbee767c, 0xe3a1cbc1, 0xe760d676, 0xea23f0af, 0xeee2ed18, 0xf0a5bd1d, 0xf464a0aa, 0xf9278673, 0xfde69bc4, 0x89b8fd09, 0x8d79e0be, 0x803ac667, 0x84fbdbd0, 0x9abc8bd5, 0x9e7d9662, 0x933eb0bb, 0x97ffad0c, 0xafb010b1, 0xab710d06, 0xa6322bdf, 0xa2f33668, 0xbcb4666d, 0xb8757bda, 0xb5365d03, 0xb1f740b4 }; BAN::ErrorOr Shell::process_command(const BAN::Vector& arguments) { if (arguments.empty()) { } else if (arguments.front() == "date") { if (arguments.size() != 1) { TTY_PRINTLN("'date' does not support command line arguments"); return {}; } auto time = RTC::get_current_time(); TTY_PRINTLN("{}", time); } else if (arguments.front() == "echo") { if (arguments.size() > 1) { TTY_PRINT("{}", arguments[1]); for (size_t i = 2; i < arguments.size(); i++) TTY_PRINT(" {}", arguments[i]); } TTY_PRINTLN(""); } else if (arguments.front() == "clear") { if (arguments.size() != 1) { TTY_PRINTLN("'clear' does not support command line arguments"); return {}; } TTY_PRINT("\e[2J\e[H"); } else if (arguments.front() == "exit") { Process::current().exit(); } else if (arguments.front() == "time") { auto new_args = arguments; new_args.remove(0); auto start = PIT::ms_since_boot(); auto ret = process_command(new_args); auto duration = PIT::ms_since_boot() - start; TTY_PRINTLN("took {} ms", duration); if (ret.is_error()) return ret.error(); } else if (arguments.front() == "thread") { struct thread_data_t { Shell* shell; SpinLock& lock; const BAN::Vector& arguments; }; auto function = [](void* data) { thread_data_t* thread_data = (thread_data_t*)data; Shell* shell = thread_data->shell; auto args = thread_data->arguments; thread_data->lock.unlock(); args.remove(0); PIT::sleep(5000); if (auto res = shell->process_command(args); res.is_error()) TTY_PRINT("{}", res.error()); }; SpinLock spinlock; thread_data_t thread_data = { this, spinlock, arguments }; spinlock.lock(); auto* thread = TRY(Thread::create_kernel(function, &thread_data, &Process::current())); Process::current().add_thread(thread); while (spinlock.is_locked()); } else if (arguments.front() == "memory") { if (arguments.size() != 1) { TTY_PRINTLN("'memory' does not support command line arguments"); return {}; } kmalloc_dump_info(); TTY_PRINTLN("heap:"); TTY_PRINTLN(" used: 0x{8H}", Heap::get().used_pages() * PAGE_SIZE); TTY_PRINTLN(" free: 0x{8H}", Heap::get().free_pages() * PAGE_SIZE); } else if (arguments.front() == "sleep") { if (arguments.size() != 1) { TTY_PRINTLN("'sleep' does not support command line arguments"); return {}; } PIT::sleep(5000); } else if (arguments.front() == "cpuinfo") { if (arguments.size() != 1) { TTY_PRINTLN("'cpuinfo' does not support command line arguments"); return {}; } uint32_t ecx, edx; auto vendor = CPUID::get_vendor(); CPUID::get_features(ecx, edx); TTY_PRINTLN("Vendor: '{}'", vendor); TTY_PRINTLN("64-bit: {}", CPUID::is_64_bit()); bool first = true; for (int i = 0; i < 32; i++) if (ecx & ((uint32_t)1 << i)) TTY_PRINT("{}{}", first ? (first = false, "") : ", ", CPUID::feature_string_ecx((uint32_t)1 << i)); for (int i = 0; i < 32; i++) if (edx & ((uint32_t)1 << i)) TTY_PRINT("{}{}", first ? (first = false, "") : ", ", CPUID::feature_string_edx((uint32_t)1 << i)); if (!first) TTY_PRINTLN(""); } else if (arguments.front() == "random") { if (arguments.size() != 1) { TTY_PRINTLN("'random' does not support command line arguments"); return {}; } uint32_t ecx, edx; CPUID::get_features(ecx, edx); if (!(ecx & CPUID::Features::ECX_RDRND)) { TTY_PRINTLN("cpu does not support RDRAND instruction"); return {}; } for (int i = 0; i < 10; i++) { uint32_t random; asm volatile("rdrand %0" : "=r"(random)); TTY_PRINTLN(" 0x{8H}", random); } } else if (arguments.front() == "reboot") { if (arguments.size() != 1) { TTY_PRINTLN("'reboot' does not support command line arguments"); return {}; } uint8_t good = 0x02; while (good & 0x02) good = IO::inb(0x64); IO::outb(0x64, 0xFE); asm volatile("cli; hlt"); } else if (arguments.front() == "lspci") { if (arguments.size() != 1) { TTY_PRINTLN("'lspci' does not support command line arguments"); return {}; } for (auto& device : PCI::get().devices()) TTY_PRINTLN("{2H}:{2H}.{2H} {2H}", device.bus(), device.dev(), device.func(), device.class_code()); } else if (arguments.front() == "ls") { if (arguments.size() > 2) { TTY_PRINTLN("usage: 'ls [path]'"); return {}; } BAN::String path; if (arguments.size() == 2) TRY(path.append(arguments[1])); else TRY(path.append(TRY(Process::current().working_directory()))); int fd = TRY(Process::current().open(path, O_RDONLY)); BAN::ScopeGuard _([fd] { MUST(Process::current().close(fd)); }); BAN::Vector all_entries; BAN::Vector entries; while (!(entries = TRY(Process::current().read_directory_entries(fd))).empty()) { TRY(all_entries.reserve(all_entries.size() + entries.size())); for (auto& entry : entries) TRY(all_entries.push_back(entry)); } BAN::String entry_prefix; TRY(entry_prefix.append(path)); TRY(entry_prefix.push_back('/')); for (const auto& entry : all_entries) { struct stat st; BAN::String entry_path; TRY(entry_path.append(entry_prefix)); TRY(entry_path.append(entry)); TRY(Process::current().stat(entry_path, &st, O_RDONLY | O_NOFOLLOW)); Inode::Mode mode { st.st_mode }; const char* color = mode.ifdir() ? "34" : mode.ifchr() || mode.ifblk() ? "33" : mode.iflnk() ? "36" : (mode.mode & Inode::Mode::IXUSR) ? "32" : ""; TTY_PRINTLN(" {} {7} \e[{}m{}\e[m", mode_string(mode), st.st_size, color, entry); } } else if (arguments.front() == "cat") { if (arguments.size() != 2) { TTY_PRINTLN("usage: 'cat path'"); return {}; } int fd = TRY(Process::current().open(arguments[1], O_RDONLY)); BAN::ScopeGuard _([fd] { MUST(Process::current().close(fd)); }); constexpr size_t buffer_size = 1024; char* buffer = new char[buffer_size]; BAN::ScopeGuard buffer_guard([buffer] { delete[] buffer; }); ASSERT(buffer); while (size_t n_read = TRY(Process::current().read(fd, buffer, buffer_size))) TTY_PRINT("{}", BAN::StringView(buffer, n_read)); TTY_PRINTLN(""); } else if (arguments.front() == "stat") { if (arguments.size() != 2) { TTY_PRINTLN("usage: 'stat path'"); return {}; } struct stat st; TRY(Process::current().stat(arguments[1], &st, O_RDONLY | O_NOFOLLOW)); Inode::Mode mode { st.st_mode }; const char* type = mode.ifreg() ? "regular file" : mode.ifdir() ? "directory" : mode.ifchr() ? "character device" : mode.ifblk() ? "block device" : mode.iflnk() ? "symbolic link" : "other"; TTY_PRINTLN(" File: {}", arguments[1]); TTY_PRINTLN(" Size: {}\tBlocks: {}\tIO Block: {}\t {}", st.st_size, st.st_blocks, st.st_blksize, type); TTY_PRINT("Device: {},{}\tInode: {}\tLinks: {}", major(st.st_dev), minor(st.st_dev), st.st_ino, st.st_nlink); if (st.st_rdev) TTY_PRINT("\tDevice type: {},{}", major(st.st_rdev), minor(st.st_rdev)); TTY_PRINTLN(""); TTY_PRINTLN("Access: ({4O}/{})\tUid: {}\tGid: {}", mode.mode & 0777, mode_string(mode), st.st_uid, st.st_gid); TTY_PRINTLN("Access: {}", BAN::from_unix_time(st.st_atime)); TTY_PRINTLN("Modify: {}", BAN::from_unix_time(st.st_mtime)); TTY_PRINTLN("Change: {}", BAN::from_unix_time(st.st_ctime)); } else if (arguments.front() == "cd") { if (arguments.size() > 2) { TTY_PRINTLN("usage 'cd path'"); return {}; } BAN::StringView path = arguments.size() == 2 ? arguments[1].sv() : "/"sv; TRY(Process::current().set_working_directory(path)); TRY(update_prompt()); } else if (arguments.front() == "touch") { if (arguments.size() != 2) { TTY_PRINTLN("usage 'touch path'"); return {}; } TRY(Process::current().creat(arguments[1], 0)); } else if (arguments.front() == "cksum") { if (arguments.size() < 2) { TTY_PRINTLN("usage 'cksum paths...'"); return {}; } constexpr size_t buffer_size = 1024; uint8_t* buffer = new uint8_t[buffer_size]; BAN::ScopeGuard buffer_guard([buffer] { delete[] buffer; }); ASSERT(buffer); for (size_t i = 1; i < arguments.size(); i++) { int fd = TRY(Process::current().open(arguments[i], O_RDONLY)); BAN::ScopeGuard _([fd] { MUST(Process::current().close(fd)); }); uint32_t crc32 = 0; uint32_t total_read = 0; while (true) { size_t n_read = TRY(Process::current().read(fd, buffer, buffer_size)); if (n_read == 0) break; for (size_t j = 0; j < n_read; j++) crc32 = (crc32 << 8) ^ crc32_table[((crc32 >> 24) ^ buffer[j]) & 0xFF]; total_read += n_read; } for (uint32_t length = total_read; length; length >>= 8) crc32 = (crc32 << 8) ^ crc32_table[((crc32 >> 24) ^ length) & 0xFF]; crc32 = ~crc32 & 0xFFFFFFFF; TTY_PRINTLN("{} {} {}", crc32, total_read, arguments[i]); } } else if (arguments.front() == "mount") { if (arguments.size() != 3) { TTY_PRINTLN("usage: 'mount partition directory'"); return {}; } TRY(Process::current().mount(arguments[1], arguments[2])); } else if (arguments.front() == "loadfont") { if (arguments.size() != 2) { TTY_PRINTLN("usage: 'loadfont font_path'"); return {}; } auto font = TRY(Font::load(arguments[1])); //m_tty->set_font(font); } else { TTY_PRINTLN("unrecognized command '{}'", arguments.front()); return {}; } return {}; } void Shell::rerender_buffer() const { TTY_PRINT("\e[{}G{}\e[K", m_prompt_length + 1, m_buffer[m_cursor_pos.line]); } }