mirror of
https://github.com/Fennix-Project/Kernel.git
synced 2025-05-25 22:14:37 +00:00
246 lines
12 KiB
C++
246 lines
12 KiB
C++
#include <exec.hpp>
|
|
|
|
#include <memory.hpp>
|
|
#include <lock.hpp>
|
|
#include <msexec.h>
|
|
#include <cwalk.h>
|
|
#include <elf.h>
|
|
#include <abi.h>
|
|
|
|
#include "../kernel.h"
|
|
#include "../Fex.hpp"
|
|
|
|
using namespace Tasking;
|
|
|
|
namespace Execute
|
|
{
|
|
SpawnData Spawn(char *Path, const char **argv, const char **envp)
|
|
{
|
|
SpawnData ret = {.Status = ExStatus::Unknown,
|
|
.Process = nullptr,
|
|
.Thread = nullptr};
|
|
FileSystem::FILE *ExFile = vfs->Open(Path);
|
|
if (ExFile->Status == FileSystem::FileStatus::OK)
|
|
{
|
|
if (ExFile->Node->Flags == FileSystem::NodeFlags::FS_FILE)
|
|
{
|
|
BinaryType Type = GetBinaryType(Path);
|
|
switch (Type)
|
|
{
|
|
case BinaryType::BinTypeFex:
|
|
{
|
|
#if defined(__amd64__)
|
|
|
|
Fex *FexHdr = (Fex *)ExFile->Node->Address;
|
|
if (FexHdr->Type == FexFormatType::FexFormatType_Executable)
|
|
{
|
|
const char *BaseName;
|
|
cwk_path_get_basename(Path, &BaseName, nullptr);
|
|
PCB *Process = TaskManager->CreateProcess(TaskManager->GetCurrentProcess(), BaseName, TaskTrustLevel::User);
|
|
|
|
void *BaseImage = KernelAllocator.RequestPages(TO_PAGES(ExFile->Node->Length));
|
|
memcpy(BaseImage, (void *)ExFile->Node->Address, ExFile->Node->Length);
|
|
|
|
Memory::Virtual pva = Memory::Virtual(Process->PageTable);
|
|
for (uint64_t i = 0; i < TO_PAGES(ExFile->Node->Length); i++)
|
|
pva.Map((void *)((uint64_t)BaseImage + (i * PAGE_SIZE)), (void *)((uint64_t)BaseImage + (i * PAGE_SIZE)), Memory::PTFlag::RW | Memory::PTFlag::US);
|
|
|
|
Vector<AuxiliaryVector> auxv; // TODO!
|
|
|
|
TCB *Thread = TaskManager->CreateThread(Process,
|
|
(IP)FexHdr->Pointer,
|
|
argv, envp, auxv,
|
|
(IPOffset)BaseImage,
|
|
TaskArchitecture::x64,
|
|
TaskCompatibility::Native);
|
|
ret.Process = Process;
|
|
ret.Thread = Thread;
|
|
ret.Status = ExStatus::OK;
|
|
#elif defined(__i386__)
|
|
if (1)
|
|
{
|
|
#elif defined(__aarch64__)
|
|
if (1)
|
|
{
|
|
#endif
|
|
goto Exit;
|
|
}
|
|
ret.Status = ExStatus::InvalidFileHeader;
|
|
goto Exit;
|
|
}
|
|
case BinaryType::BinTypeELF:
|
|
{
|
|
#if defined(__amd64__)
|
|
const char *BaseName;
|
|
cwk_path_get_basename(Path, &BaseName, nullptr);
|
|
PCB *Process = TaskManager->CreateProcess(TaskManager->GetCurrentProcess(), BaseName, TaskTrustLevel::User);
|
|
|
|
void *BaseImage = KernelAllocator.RequestPages(TO_PAGES(ExFile->Node->Length));
|
|
memcpy(BaseImage, (void *)ExFile->Node->Address, ExFile->Node->Length);
|
|
debug("Image Size: %#lx - %#lx (length: %ld)", BaseImage, (uint64_t)BaseImage + ExFile->Node->Length, ExFile->Node->Length);
|
|
|
|
Memory::Virtual pva = Memory::Virtual(Process->PageTable);
|
|
for (uint64_t i = 0; i < TO_PAGES(ExFile->Node->Length); i++)
|
|
pva.Remap((void *)((uint64_t)BaseImage + (i * PAGE_SIZE)), (void *)((uint64_t)BaseImage + (i * PAGE_SIZE)), Memory::PTFlag::RW | Memory::PTFlag::US);
|
|
|
|
Elf64_Ehdr *ELFHeader = (Elf64_Ehdr *)BaseImage;
|
|
|
|
TaskArchitecture Arch = TaskArchitecture::x64;
|
|
TaskCompatibility Comp = TaskCompatibility::Native;
|
|
if (ELFHeader->e_machine == EM_386)
|
|
Arch = TaskArchitecture::x32;
|
|
else if (ELFHeader->e_machine == EM_AMD64)
|
|
Arch = TaskArchitecture::x64;
|
|
else if (ELFHeader->e_machine == EM_AARCH64)
|
|
Arch = TaskArchitecture::ARM64;
|
|
else
|
|
Arch = TaskArchitecture::UnknownArchitecture;
|
|
|
|
// TODO: Should I care about this?
|
|
if (ELFHeader->e_ident[EI_CLASS] == ELFCLASS32)
|
|
{
|
|
if (ELFHeader->e_ident[EI_DATA] == ELFDATA2LSB)
|
|
fixme("ELF32 LSB");
|
|
else if (ELFHeader->e_ident[EI_DATA] == ELFDATA2MSB)
|
|
fixme("ELF32 MSB");
|
|
else
|
|
fixme("ELF32 Unknown");
|
|
}
|
|
else if (ELFHeader->e_ident[EI_CLASS] == ELFCLASS64)
|
|
{
|
|
if (ELFHeader->e_ident[EI_DATA] == ELFDATA2LSB)
|
|
fixme("ELF64 LSB");
|
|
else if (ELFHeader->e_ident[EI_DATA] == ELFDATA2MSB)
|
|
fixme("ELF64 MSB");
|
|
else
|
|
fixme("ELF64 Unknown");
|
|
}
|
|
else
|
|
fixme("Unknown ELF");
|
|
|
|
if (ELFHeader->e_type == ET_EXEC)
|
|
{
|
|
trace("Executable");
|
|
Elf64_Phdr *pheader = (Elf64_Phdr *)(((char *)BaseImage) + ELFHeader->e_phoff);
|
|
debug("p_paddr: %#lx | p_vaddr: %#lx | p_filesz: %#lx | p_memsz: %#lx | p_offset: %#lx", pheader->p_paddr, pheader->p_vaddr, pheader->p_filesz, pheader->p_memsz, pheader->p_offset);
|
|
|
|
void *Address = nullptr;
|
|
for (int i = 0; i < ELFHeader->e_phnum; i++, pheader++)
|
|
{
|
|
if (pheader->p_type != PT_LOAD)
|
|
continue;
|
|
Address = (void *)((uint64_t)pheader->p_vaddr + pheader->p_memsz);
|
|
}
|
|
void *Offset = KernelAllocator.RequestPages(TO_PAGES((uint64_t)Address));
|
|
|
|
pheader = (Elf64_Phdr *)(((char *)BaseImage) + ELFHeader->e_phoff);
|
|
for (uint64_t i = 0; i < TO_PAGES((uint64_t)Address); i++)
|
|
{
|
|
pva.Remap((void *)((uint64_t)pheader->p_vaddr + (i * PAGE_SIZE)), (void *)((uint64_t)Offset + (i * PAGE_SIZE)), Memory::PTFlag::RW | Memory::PTFlag::US);
|
|
// debug("Mapping: %#lx -> %#lx", (uint64_t)pheader->p_vaddr + (i * PAGE_SIZE), (uint64_t)Offset + (i * PAGE_SIZE));
|
|
}
|
|
|
|
pheader = (Elf64_Phdr *)(((char *)BaseImage) + ELFHeader->e_phoff);
|
|
for (int i = 0; i < ELFHeader->e_phnum; i++, pheader++)
|
|
{
|
|
if (pheader->p_type != PT_LOAD)
|
|
continue;
|
|
void *dst = (void *)((uint64_t)pheader->p_vaddr + (uint64_t)Offset);
|
|
memset(dst, 0, pheader->p_memsz);
|
|
memcpy(dst, ((char *)BaseImage) + pheader->p_offset, pheader->p_filesz);
|
|
}
|
|
|
|
debug("Entry Point: %#lx", ELFHeader->e_entry);
|
|
|
|
Vector<AuxiliaryVector> auxv;
|
|
|
|
pheader = (Elf64_Phdr *)(((char *)BaseImage) + ELFHeader->e_phoff);
|
|
auxv.push_back({.archaux = {.a_type = AT_PHDR, .a_un = {.a_val = (uint64_t)ELFHeader->e_phoff}}});
|
|
auxv.push_back({.archaux = {.a_type = AT_PHENT, .a_un = {.a_val = (uint64_t)ELFHeader->e_phentsize}}});
|
|
auxv.push_back({.archaux = {.a_type = AT_PHNUM, .a_un = {.a_val = (uint64_t)ELFHeader->e_phnum}}});
|
|
auxv.push_back({.archaux = {.a_type = AT_PAGESZ, .a_un = {.a_val = (uint64_t)PAGE_SIZE}}});
|
|
auxv.push_back({.archaux = {.a_type = AT_BASE, .a_un = {.a_val = (uint64_t)Offset}}});
|
|
auxv.push_back({.archaux = {.a_type = AT_ENTRY, .a_un = {.a_val = (uint64_t)ELFHeader->e_entry + (uint64_t)pheader->p_offset}}});
|
|
auxv.push_back({.archaux = {.a_type = AT_PLATFORM, .a_un = {.a_val = (uint64_t) "x86_64"}}});
|
|
auxv.push_back({.archaux = {.a_type = AT_EXECFN, .a_un = {.a_val = (uint64_t)Path}}});
|
|
|
|
TCB *Thread = TaskManager->CreateThread(Process,
|
|
(IP)ELFHeader->e_entry,
|
|
argv, envp, auxv,
|
|
(IPOffset)pheader->p_offset,
|
|
Arch,
|
|
Comp);
|
|
ret.Process = Process;
|
|
ret.Thread = Thread;
|
|
ret.Status = ExStatus::OK;
|
|
goto Exit;
|
|
}
|
|
else if (ELFHeader->e_type == ET_DYN)
|
|
{
|
|
fixme("Shared Object");
|
|
}
|
|
else if (ELFHeader->e_type == ET_REL)
|
|
{
|
|
trace("Relocatable");
|
|
void *EP = ELFLoadRel(ELFHeader);
|
|
if (EP == (void *)0xdeadbeef || EP == 0x0)
|
|
{
|
|
ret.Status = ExStatus::InvalidFileEntryPoint;
|
|
goto Exit;
|
|
}
|
|
|
|
Vector<AuxiliaryVector> auxv;
|
|
fixme("auxv");
|
|
|
|
TCB *Thread = TaskManager->CreateThread(Process,
|
|
(IP)EP,
|
|
argv, envp, auxv,
|
|
(IPOffset)BaseImage,
|
|
Arch,
|
|
Comp);
|
|
ret.Process = Process;
|
|
ret.Thread = Thread;
|
|
ret.Status = ExStatus::OK;
|
|
goto Exit;
|
|
}
|
|
else if (ELFHeader->e_type == ET_CORE)
|
|
{
|
|
fixme("Core");
|
|
}
|
|
else
|
|
{
|
|
fixme("Unknown");
|
|
}
|
|
ret.Status = ExStatus::InvalidFileHeader;
|
|
#elif defined(__i386__)
|
|
#elif defined(__aarch64__)
|
|
#endif
|
|
goto Exit;
|
|
}
|
|
default:
|
|
ret.Status = ExStatus::Unsupported;
|
|
goto Exit;
|
|
}
|
|
goto Exit;
|
|
}
|
|
}
|
|
else if (ExFile->Status == FileSystem::FileStatus::NOT_FOUND)
|
|
{
|
|
ret.Status = ExStatus::InvalidFilePath;
|
|
goto Exit;
|
|
}
|
|
else
|
|
{
|
|
ret.Status = ExStatus::InvalidFile;
|
|
goto Exit;
|
|
}
|
|
|
|
Exit:
|
|
if (ret.Status != ExStatus::OK)
|
|
if (ret.Process)
|
|
ret.Process->Status = TaskStatus::Terminated;
|
|
vfs->Close(ExFile);
|
|
return ret;
|
|
}
|
|
}
|