This fixes issues where it did not fit into the first few MiB, or that GRUB loaded it someplace weird. The kernel heap is now also protected against growing into the ramdisk and the kernel stack.
132 lines
3.7 KiB
C++
132 lines
3.7 KiB
C++
/******************************************************************************
|
|
|
|
COPYRIGHT(C) JONAS 'SORTIE' TERMANSEN 2011.
|
|
|
|
This file is part of Sortix.
|
|
|
|
Sortix is free software: you can redistribute it and/or modify it under the
|
|
terms of the GNU General Public License as published by the Free Software
|
|
Foundation, either version 3 of the License, or (at your option) any later
|
|
version.
|
|
|
|
Sortix is distributed in the hope that it will be useful, but WITHOUT ANY
|
|
WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
|
|
FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
|
|
details.
|
|
|
|
You should have received a copy of the GNU General Public License along
|
|
with Sortix. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
initrd.cpp
|
|
Declares the structure of the Sortix ramdisk.
|
|
|
|
******************************************************************************/
|
|
|
|
#include "platform.h"
|
|
#include "initrd.h"
|
|
#include <libmaxsi/memory.h>
|
|
#include <libmaxsi/string.h>
|
|
#include "syscall.h"
|
|
#include "memorymanagement.h"
|
|
|
|
#include "log.h" // DEBUG
|
|
|
|
using namespace Maxsi;
|
|
|
|
namespace Sortix
|
|
{
|
|
namespace InitRD
|
|
{
|
|
byte* initrd;
|
|
size_t initrdsize;
|
|
#ifdef JSSORTIX
|
|
// JSVM never tells JSSortix how big the initrd is!
|
|
const bool CHECK_CHECKSUM = false;
|
|
#else
|
|
const bool CHECK_CHECKSUM = true;
|
|
#endif
|
|
|
|
size_t GetNumFiles()
|
|
{
|
|
Header* header = (Header*) initrd;
|
|
return header->numfiles;
|
|
}
|
|
const char* GetFilename(size_t index)
|
|
{
|
|
Header* header = (Header*) initrd;
|
|
if ( index >= header->numfiles ) { return NULL; }
|
|
FileHeader* fhtbl = (FileHeader*) (initrd + sizeof(Header));
|
|
FileHeader* fileheader = &(fhtbl[index]);
|
|
return fileheader->name;
|
|
}
|
|
|
|
uint8_t ContinueChecksum(uint8_t checksum, const void* p, size_t size)
|
|
{
|
|
const uint8_t* buffer = (const uint8_t*) p;
|
|
while ( size-- )
|
|
{
|
|
checksum += *buffer++;
|
|
}
|
|
return checksum;
|
|
}
|
|
|
|
void CheckSum()
|
|
{
|
|
Trailer* trailer = (Trailer*) (initrd + initrdsize - sizeof(Trailer));
|
|
uint8_t checksum = ContinueChecksum(0, initrd, initrdsize - sizeof(Trailer));
|
|
if ( trailer->sum != checksum )
|
|
{
|
|
PanicF("InitRD Checksum failed: the ramdisk may have been "
|
|
"corrupted by the bootloader: Got %u instead of %u "
|
|
"when checking the ramdisk at 0x%p + 0x%zx bytes\n",
|
|
checksum, trailer->sum, initrd, initrdsize);
|
|
}
|
|
}
|
|
|
|
void Init(addr_t phys, size_t size)
|
|
{
|
|
// First up, map the initrd onto the kernel's address space.
|
|
addr_t virt = Memory::GetInitRD();
|
|
size_t amount = 0;
|
|
while ( amount < size )
|
|
{
|
|
if ( !Memory::MapKernel(phys + amount, virt + amount) )
|
|
{
|
|
Panic("Unable to map the init ramdisk into virtual memory");
|
|
}
|
|
amount += 0x1000UL;
|
|
}
|
|
|
|
Memory::Flush();
|
|
|
|
initrd = (byte*) virt;
|
|
initrdsize = size;
|
|
if ( size < sizeof(Header) ) { PanicF("initrd.cpp: initrd is too small"); }
|
|
Header* header = (Header*) initrd;
|
|
if ( String::Compare(header->magic, "sortix-initrd-1") != 0 ) { PanicF("initrd.cpp: invalid magic value in the initrd"); }
|
|
size_t sizeneeded = sizeof(Header) + header->numfiles * sizeof(FileHeader);
|
|
if ( size < sizeneeded ) { PanicF("initrd.cpp: initrd is too small"); }
|
|
// TODO: We need to do more validation here!
|
|
|
|
if ( CHECK_CHECKSUM ) { CheckSum(); }
|
|
}
|
|
|
|
byte* Open(const char* filepath, size_t* size)
|
|
{
|
|
Header* header = (Header*) initrd;
|
|
FileHeader* fhtbl = (FileHeader*) (initrd + sizeof(Header));
|
|
for ( uint32_t i = 0; i < header->numfiles; i++ )
|
|
{
|
|
FileHeader* fileheader = &(fhtbl[i]);
|
|
|
|
if ( String::Compare(filepath, fileheader->name) != 0 ) { continue; }
|
|
|
|
*size = fileheader->size;
|
|
return initrd + fileheader->offset;
|
|
}
|
|
|
|
return NULL;
|
|
}
|
|
}
|
|
}
|