264 lines
8.0 KiB
C
264 lines
8.0 KiB
C
#include "../include/shell.h"
|
|
#include "../include/queue.h"
|
|
#include "../include/graphics.h"
|
|
#include "../include/common.h"
|
|
#include "../include/task.h"
|
|
#include "../include/cmos.h"
|
|
#include "../include/vdisk.h"
|
|
|
|
extern Queue *key_char_queue;
|
|
extern vdisk vdisk_ctl[10];
|
|
|
|
char getc() {
|
|
while (key_char_queue->size == 0x00) {
|
|
printf("");
|
|
}
|
|
return queue_pop(key_char_queue);
|
|
}
|
|
|
|
int gets(char *buf, int buf_size) {
|
|
int index = 0;
|
|
char c;
|
|
while ((c = getc()) != '\n') {
|
|
if (c == '\b') {
|
|
if (index > 0) {
|
|
index--;
|
|
vga_writestring("\b \b");
|
|
}
|
|
} else {
|
|
buf[index++] = c;
|
|
vga_putchar(c);
|
|
}
|
|
}
|
|
buf[index] = '\0';
|
|
vga_putchar(c);
|
|
return index;
|
|
}
|
|
|
|
int cmd_parse(char *cmd_str, char **argv, char token) {
|
|
int arg_idx = 0;
|
|
while (arg_idx < MAX_ARG_NR) {
|
|
argv[arg_idx] = NULL;
|
|
arg_idx++;
|
|
}
|
|
char *next = cmd_str;
|
|
int argc = 0;
|
|
|
|
while (*next) {
|
|
while (*next == token) *next++;
|
|
if (*next == 0) break;
|
|
argv[argc] = next;
|
|
while (*next && *next != token) *next++;
|
|
if (*next) *next++ = 0;
|
|
if (argc > MAX_ARG_NR) return -1;
|
|
argc++;
|
|
}
|
|
|
|
return argc;
|
|
}
|
|
|
|
void cmd_echo(int argc, char **argv) {
|
|
for (int i = 1; i < argc; i++) {
|
|
if (i == 1) vga_writestring("");
|
|
else vga_writestring(" ");
|
|
vga_writestring(argv[i]);
|
|
}
|
|
vga_putchar('\n');
|
|
}
|
|
|
|
void cmd_proc(int argc, char **argv){
|
|
if (argc <= 1) {
|
|
printf("\033[Shell-PROC]: If there are too few parameters.\036\n");
|
|
return;
|
|
}
|
|
|
|
if(!strcmp("list",argv[1])){
|
|
print_proc();
|
|
} else if(!strcmp("kill",argv[1])){
|
|
if (argc <= 2) {
|
|
printf("\033[Shell-PROC-kill]: If there are too few parameters.\036\n");
|
|
return;
|
|
}
|
|
int pid = strtol(argv[2],NULL,10);
|
|
task_kill(pid);
|
|
} else{
|
|
printf("\033[Shell-[PROC]]: Unknown parameter\036\n");
|
|
return;
|
|
}
|
|
}
|
|
|
|
void cmd_date(){
|
|
printf("System Time: %s\n",get_date_time());
|
|
printf("Memory Usage: [%dKB] | All Size: [%dMB]\n",memory_usage()/1024,(KHEAP_START+KHEAP_INITIAL_SIZE)/1024/1024);
|
|
print_cpu_id();
|
|
vga_writestring("\n");
|
|
}
|
|
|
|
void cmd_ls() {
|
|
|
|
}
|
|
|
|
void cmd_mkdir(int argc, char **argv) {
|
|
if (argc == 1) {
|
|
printf("\033[Shell-MKDIR]: If there are too few parameters, please specify the directory name.\036\n");
|
|
return;
|
|
}
|
|
printf("Create directory: %s\n",argv[1]);
|
|
|
|
}
|
|
|
|
void cmd_del(int argc, char **argv) {
|
|
if (argc == 1) {
|
|
vga_writestring("\033[Shell-DEL]: If there are too few parameters, please specify the folder name.\036\n");
|
|
return;
|
|
}
|
|
|
|
}
|
|
|
|
void cmd_reset(){
|
|
reset_kernel();
|
|
}
|
|
|
|
void cmd_shutdown(){
|
|
shutdown_kernel();
|
|
}
|
|
|
|
void cmd_debug(){
|
|
vga_clear();
|
|
printf("%s for x86 [Version %s] \n",OS_NAME, OS_VERSION);
|
|
printf("\032Copyright 2024 XIAOYI12 (Build by GCC i686-elf-tools)\036\n");
|
|
extern int acpi_enable_flag;
|
|
extern uint8_t *rsdp_address;
|
|
printf("ACPI: Enable: %s | RSDP Address: 0x%08x\n",acpi_enable_flag?"\035ENABLE\036":"\033DISABLE\036",rsdp_address);
|
|
int index = 0;
|
|
print_proc_t(&index,get_current(),get_current()->next,0);
|
|
printf("Process Runnable: %d\n",index);
|
|
cmd_date();
|
|
for (int i = 0; i < 10; i++) {
|
|
if (vdisk_ctl[i].flag) {
|
|
vdisk vd = vdisk_ctl[i];
|
|
char id = i + ('A');
|
|
printf("[DISK-%c]: Size: %dMB | Name: %s\n",id,vd.size,vd.DriveName);
|
|
}
|
|
}
|
|
printf(" > > > > ====[Registers Info]==== < < < <\n");
|
|
register uint32_t eax asm("eax"),
|
|
ecx asm("ecx"),
|
|
esp asm("esp"),
|
|
ebp asm("ebp"),
|
|
ebx asm("ebx"),
|
|
esi asm("esi"),
|
|
edi asm("edi");
|
|
printf("EAX: 0x%08x | EBX 0x%08x | ECX 0x%08x | ESP 0x%08x \n",eax,ebx,ecx,esp);
|
|
printf("ESI: 0x%08x | EDI 0x%08x | EBP 0x%08x | EFLAGS 0x%08x\n",esi,edi,ebp,get_current()->context.eflags);
|
|
}
|
|
|
|
void cmd_disk(int argc,char **argv){
|
|
if(argc > 1){
|
|
if(!strcmp("list",argv[1])){
|
|
printf("[Disk]: Loaded disk - ");
|
|
for (int i = 0; i < 10; i++) {
|
|
if (vdisk_ctl[i].flag) {
|
|
vdisk vd = vdisk_ctl[i];
|
|
char id = i + ('A');
|
|
printf("(%c) ",id,vd.size,vd.DriveName);
|
|
}
|
|
}
|
|
printf("\n");
|
|
return;
|
|
}
|
|
|
|
if(strlen(argv[1]) > 1){
|
|
printf("\033[DISK]: Cannot found disk.\036\n");
|
|
return;
|
|
}
|
|
|
|
for (int i = 0; i < 10; i++) {
|
|
if (vdisk_ctl[i].flag) {
|
|
vdisk vd = vdisk_ctl[i];
|
|
char id = i + ('A');
|
|
if(id == (argv[1][0])){
|
|
printf("[Disk(%c)]: \n"
|
|
" Size: %dMB\n"
|
|
" Name: %s\n"
|
|
" WriteAddress: 0x%08x\n"
|
|
" ReadAddress: 0x%08x\n",id,vd.size,vd.DriveName,vd.Write,vd.Read);
|
|
return;
|
|
}
|
|
}
|
|
}
|
|
printf("\033[DISK]: Cannot found disk.\036\n");
|
|
return;
|
|
}
|
|
printf("[Disk]: Loaded disk - ");
|
|
for (int i = 0; i < 10; i++) {
|
|
if (vdisk_ctl[i].flag) {
|
|
vdisk vd = vdisk_ctl[i];
|
|
char id = i + ('A');
|
|
printf("(%c) ",id,vd.size,vd.DriveName);
|
|
}
|
|
}
|
|
printf("\n");
|
|
}
|
|
|
|
void setup_shell(){
|
|
vga_clear();
|
|
printf("%s for x86 [Version %s] \n",OS_NAME, OS_VERSION);
|
|
printf("\032Copyright 2024 XIAOYI12 (Build by GCC i686-elf-tools)\036\n");
|
|
|
|
char com[MAX_COMMAND_LEN];
|
|
char *argv[MAX_ARG_NR];
|
|
int argc = -1;
|
|
|
|
while (1) {
|
|
printf("\035CPOS/>\036 ");
|
|
if (gets(com, MAX_COMMAND_LEN) <= 0) continue;
|
|
argc = cmd_parse(com, argv, ' ');
|
|
|
|
if (argc == -1) {
|
|
vga_writestring("[Shell]: Error: out of arguments buffer\n");
|
|
continue;
|
|
}
|
|
|
|
if (!strcmp("version", argv[0]))
|
|
printf("%s for x86 [%s]\n",OS_NAME, OS_VERSION);
|
|
else if (!strcmp("echo", argv[0]))
|
|
cmd_echo(argc, argv);
|
|
else if (!strcmp("clear", argv[0]))
|
|
vga_clear();
|
|
else if (!strcmp("proc", argv[0]))
|
|
cmd_proc(argc, argv);
|
|
else if (!strcmp("sysinfo", argv[0]))
|
|
cmd_date();
|
|
else if (!strcmp("ls", argv[0]))
|
|
cmd_ls();
|
|
else if (!strcmp("mkdir", argv[0]))
|
|
cmd_mkdir(argc, argv);
|
|
else if (!strcmp("del", argv[0]) || !strcmp("rm", argv[0]))
|
|
cmd_del(argc, argv);
|
|
else if (!strcmp("reset", argv[0]))
|
|
cmd_reset();
|
|
else if (!strcmp("shutdown", argv[0])||!strcmp("exit", argv[0]))
|
|
cmd_shutdown();
|
|
else if (!strcmp("debug", argv[0]))
|
|
cmd_debug();
|
|
else if (!strcmp("disk", argv[0]))
|
|
cmd_disk(argc,argv);
|
|
else if (!strcmp("help", argv[0]) || !strcmp("?", argv[0]) || !strcmp("h", argv[0])) {
|
|
vga_writestring("-=[\037CoolPotShell Helper\036]=-\n");
|
|
vga_writestring("help ? h \032Print shell help info.\036\n");
|
|
vga_writestring("version \032Print os version.\036\n");
|
|
vga_writestring("echo <msg> \032Print message.\036\n");
|
|
vga_writestring("ls \032List all files.\036\n");
|
|
vga_writestring("mkdir <name> \032Make a directory.\036\n");
|
|
vga_writestring("del rm <name> \032Delete a file.\036\n");
|
|
vga_writestring("sysinfo \032Print system info.\036\n");
|
|
vga_writestring("proc [kill<pid>|list] \032Lists all running processes.\036\n");
|
|
vga_writestring("reset \032Reset OS.\036\n");
|
|
vga_writestring("shutdown exit \032Shutdown OS.\036\n");
|
|
vga_writestring("debug \032Print os debug info.\036\n");
|
|
vga_writestring("disk [list|<ID>] \032List or view disks.\036\n");
|
|
} else printf("\033[Shell]: Unknown command '%s'.\036\n", argv[0]);
|
|
}
|
|
}
|