187 lines
4.5 KiB
C
187 lines
4.5 KiB
C
#include <string.h>
|
|
#include "vga.h"
|
|
#include <grub/text_fb_info.h>
|
|
#include <ipc/vfs.h>
|
|
#include <elf.h>
|
|
#include <mailboxes.h>
|
|
#include <memory.h>
|
|
#include <tasking.h>
|
|
#include <stdlib.h>
|
|
#include <stdio.h>
|
|
|
|
typedef struct {
|
|
char filename[100];
|
|
char mode[8];
|
|
char uid[8];
|
|
char gid[8];
|
|
char size[12];
|
|
char mtime[12];
|
|
char chksum[8];
|
|
char typeflag[1];
|
|
} tar_header;
|
|
|
|
uint32_t getsize(const char *in) {
|
|
uint32_t size=0;
|
|
uint32_t j;
|
|
uint32_t count=1;
|
|
for (j=11;j>0;j--,count*=8) {
|
|
size+=((in[j-1]-'0')*count);
|
|
}
|
|
return size;
|
|
}
|
|
|
|
void display_msg(vfs_message* vfs_msg) {
|
|
vga_write_string("Message of type ");
|
|
char str[256];
|
|
str[0]='\0';
|
|
int_to_ascii(vfs_msg->type,str);
|
|
vga_write_string(str);
|
|
vga_write_string("\n");
|
|
vga_write_string("ID ");
|
|
str[0]='\0';
|
|
int_to_ascii(vfs_msg->id,str);
|
|
vga_write_string(str);
|
|
vga_write_string("\n");
|
|
vga_write_string("Mode ");
|
|
vga_write_string(&vfs_msg->mode[0]);
|
|
vga_write_string("\n");
|
|
vga_write_string("FD ");
|
|
str[0]='\0';
|
|
int_to_ascii(vfs_msg->fd,str);
|
|
vga_write_string(str);
|
|
vga_write_string("\n");
|
|
vga_write_string("Path ");
|
|
vga_write_string(&vfs_msg->path[0]);
|
|
vga_write_string("\n");
|
|
vga_write_string("Flags ");
|
|
str[0]='\0';
|
|
hex_to_ascii(vfs_msg->flags,str);
|
|
vga_write_string(str);
|
|
vga_write_string("\n");
|
|
}
|
|
|
|
uint32_t find_loc(char* name,char* initrd) {
|
|
uint32_t pos=0;
|
|
tar_header tar_hdr;
|
|
for (int i=0;;i++) {
|
|
char* tar_hdr_ptr=(char*)&tar_hdr;
|
|
for (size_t i=0;i<sizeof(tar_hdr);i++) {
|
|
tar_hdr_ptr[i]=initrd[pos+i];
|
|
}
|
|
if (tar_hdr.filename[0]=='\0') break;
|
|
uint32_t size=getsize(tar_hdr.size);
|
|
pos+=512;
|
|
if (strcmp(tar_hdr.filename,name)==0) {
|
|
return pos;
|
|
break;
|
|
}
|
|
pos+=size;
|
|
if (pos%512!=0) {
|
|
pos+=512-(pos%512);
|
|
}
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
char load_task(uint32_t datapos,char* initrd) {
|
|
int pos=0;
|
|
elf_header header;
|
|
pos=datapos;
|
|
char* hdr_ptr=(char*)&header;
|
|
for (size_t i=0;i<sizeof(elf_header);i++) {
|
|
hdr_ptr[i]=initrd[pos];
|
|
pos++;
|
|
}
|
|
if (header.magic!=ELF_MAGIC) {
|
|
vga_write_string("[INFO] Invalid magic number\n");
|
|
return 0;
|
|
} else {
|
|
void* cr3=new_address_space();
|
|
for (int i=0;i<header.pheader_ent_nm;i++) {
|
|
elf_pheader pheader;
|
|
pos=(header.prog_hdr)+(header.pheader_ent_sz*i)+datapos;
|
|
char* phdr_ptr=(char*)&pheader;
|
|
for (size_t i=0;i<sizeof(elf_pheader);i++) {
|
|
phdr_ptr[i]=initrd[pos];
|
|
pos++;
|
|
}
|
|
char* ptr=alloc_memory(((pheader.memsz)/4096)+1);
|
|
memset(ptr,0,pheader.memsz);
|
|
if (pheader.filesz>0) {
|
|
pos=pheader.offset+datapos;
|
|
for (size_t i=0;i<pheader.filesz;i++) {
|
|
ptr[i]=initrd[pos];
|
|
pos++;
|
|
}
|
|
}
|
|
copy_data(cr3,ptr,pheader.memsz,(void*)pheader.vaddr);
|
|
}
|
|
createTaskCr3((void*)header.entry,cr3);
|
|
}
|
|
return 1;
|
|
}
|
|
|
|
vfs_message* make_msg(vfs_message_type type, char* mode, char* path) {
|
|
static uint32_t id=0;
|
|
vfs_message* msg_data=malloc(sizeof(vfs_message));
|
|
msg_data->type=type;
|
|
msg_data->id=id;
|
|
id++;
|
|
strcpy(&msg_data->mode[0],mode);
|
|
strcpy(&msg_data->path[0],path);
|
|
return msg_data;
|
|
}
|
|
|
|
void test_vfs(char* path,uint32_t box,uint32_t fs_box) {
|
|
vga_write_string("Sending test message\n");
|
|
vfs_message* msg_data=make_msg(VFS_OPEN,"r",path);
|
|
Message msg;
|
|
msg.from=box;
|
|
msg.to=1;
|
|
msg.msg=msg_data;
|
|
msg.size=sizeof(vfs_message);
|
|
mailbox_send_msg(&msg);
|
|
free(msg.msg);
|
|
yield();
|
|
vga_write_string("Getting fs_box message\n");
|
|
yield();
|
|
vga_write_string("Getting message\n");
|
|
msg.msg=malloc(sizeof(vfs_message));
|
|
mailbox_get_msg(box,&msg,sizeof(vfs_message));
|
|
if (msg.from==0) {
|
|
vga_write_string("No message\n");
|
|
} else {
|
|
vfs_message* vfs_msg=(vfs_message*)msg.msg;
|
|
display_msg(vfs_msg);
|
|
}
|
|
free(msg.msg);
|
|
}
|
|
|
|
int main(char* initrd, uint32_t initrd_sz) {
|
|
text_fb_info info;
|
|
info.address=map_phys((void*)0xB8000,10);
|
|
info.width=80;
|
|
info.height=25;
|
|
vga_init(info);
|
|
vga_write_string("INIT\n");
|
|
uint32_t datapos=find_loc("vfs",initrd);
|
|
load_task(datapos,initrd);
|
|
yield();
|
|
datapos=find_loc("fsdrv",initrd);
|
|
load_task(datapos,initrd);
|
|
yield();
|
|
vga_write_string("CALLING FOPEN\n");
|
|
FILE* file=fopen("/dev/sda","w");
|
|
vga_write_string("FOPEN RETURNED\n");
|
|
vga_write_string("CALLING FPUTC\n");
|
|
fputc('a',file);
|
|
vga_write_string("FPUTC RETURNED\n");
|
|
// yield();
|
|
// // uint32_t fs_box=mailbox_new(16);
|
|
// // test_vfs("/dev/sda",box,fs_box);
|
|
// // test_vfs("/dev/sdb",box,fs_box);
|
|
for(;;) {
|
|
yield();
|
|
}
|
|
}
|