Bananymous
/
banan-os
1
1
Fork 1
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Kernel: Add more functionality to Shell 

commands 'time', 'cpuid', 'random'
This commit is contained in:
Bananymous 2022-12-15 17:29:35 +02:00
parent a2d38af37a
commit 930661619b
2 changed files with 78 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
kernel/include/kernel/Shell.h
View File

@ -18,7 +18,7 @@ namespace Kernel
private:
Shell();
void PrintPrompt();
void ProcessCommand(BAN::StringView);
void ProcessCommand(const BAN::Vector<BAN::StringView>&);
void KeyEventCallback(Keyboard::KeyEvent);
private:

79
kernel/kernel/Shell.cpp
View File

@ -1,7 +1,9 @@
#include <BAN/StringView.h>
#include <BAN/Vector.h>
#include <kernel/CPUID.h>
#include <kernel/IO.h>
#include <kernel/Keyboard.h>
#include <kernel/PIT.h>
#include <kernel/RTC.h>
#include <kernel/Shell.h>
#include <kernel/tty.h>
@ -39,9 +41,8 @@ namespace Kernel
}
}
void Shell::ProcessCommand(BAN::StringView command)
void Shell::ProcessCommand(const BAN::Vector<BAN::StringView>& arguments)
{
auto arguments = command.Split(' ');
if (arguments.Empty())
return;
@ -81,6 +82,78 @@ namespace Kernel
return;
}
if (arguments.Front() == "time")
{
auto new_args = arguments;
new_args.Remove(0);
auto start = PIT::ms_since_boot();
ProcessCommand(new_args);
auto duration = PIT::ms_since_boot() - start;
kprintln("took {} ms", duration);
return;
}
if (arguments.Front() == "cpuinfo")
{
if (arguments.Size() != 1)
{
kprintln("'cpuinfo' does not support command line arguments");
return;
}
if (!CPUID::IsAvailable())
{
kprintln("'cpuid' instruction not available");
return;
}
uint32_t ecx, edx;
auto vendor = CPUID::GetVendor();
CPUID::GetFeatures(ecx, edx);
kprintln("Vendor: '{}'", vendor);
bool first = true;
for (int i = 0; i < 32; i++)
if (ecx & ((uint32_t)1 << i))
kprint("{}{}", first ? (first = false, "") : ", ", CPUID::FeatStringECX((uint32_t)1 << i));
for (int i = 0; i < 32; i++)
if (edx & ((uint32_t)1 << i))
kprint("{}{}", first ? (first = false, "") : ", ", CPUID::FeatStringEDX((uint32_t)1 << i));
if (!first)
kprintln();
return;
}
if (arguments.Front() == "random")
{
if (arguments.Size() != 1)
{
kprintln("'random' does not support command line arguments");
return;
}
if (!CPUID::IsAvailable())
{
kprintln("'cpuid' instruction not available");
return;
}
uint32_t ecx, edx;
CPUID::GetFeatures(ecx, edx);
if (!(ecx & CPUID::Features::ECX_RDRND))
{
kprintln("cpu does not support RDRAND instruction");
return;
}
for (int i = 0; i < 10; i++)
{
uint32_t random;
asm volatile("rdrand %0" : "=r"(random));
kprintln(" 0x{8H}", random);
}
return;
}
if (arguments.Front() == "reboot")
{
if (arguments.Size() != 1)
@ -120,7 +193,7 @@ namespace Kernel
case Keyboard::Key::NumpadEnter:
{
kprint("\n");
ProcessCommand(m_buffer);
ProcessCommand(m_buffer.SV().Split(' '));
m_buffer.Clear();
PrintPrompt();
break;