This uncovers a bug where the bootloader may corrupt the initrd on x64. This is probably related to our kernel64 hack..
149 lines
4.2 KiB
C++
149 lines
4.2 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 "log.h" // DEBUG
|
|
|
|
using namespace Maxsi;
|
|
|
|
namespace Sortix
|
|
{
|
|
namespace InitRD
|
|
{
|
|
byte* initrd;
|
|
size_t initrdsize;
|
|
|
|
void SysPrintPathFiles()
|
|
{
|
|
Header* header = (Header*) initrd;
|
|
FileHeader* fhtbl = (FileHeader*) (initrd + sizeof(Header));
|
|
for ( uint32_t i = 0; i < header->numfiles; i++ )
|
|
{
|
|
FileHeader* fileheader = &(fhtbl[i]);
|
|
Log::PrintF("%s\n", fileheader->name);
|
|
}
|
|
}
|
|
|
|
size_t GetNumFiles()
|
|
{
|
|
Header* header = (Header*) initrd;
|
|
return header->numfiles;
|
|
}
|
|
|
|
size_t SysGetNumFiles()
|
|
{
|
|
return GetNumFiles();
|
|
}
|
|
|
|
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;
|
|
}
|
|
|
|
struct FileInfo
|
|
{
|
|
mode_t permissions;
|
|
char name[128];
|
|
};
|
|
|
|
int SysGetFileInfo(size_t index, FileInfo* fileinfo)
|
|
{
|
|
Header* header = (Header*) initrd;
|
|
if ( index >= header->numfiles ) { return -1; }
|
|
FileHeader* fhtbl = (FileHeader*) (initrd + sizeof(Header));
|
|
FileHeader* fileheader = &(fhtbl[index]);
|
|
|
|
// TODO: Check that fileinfo is a userspace writable buffer.
|
|
|
|
STATIC_ASSERT(sizeof(fileheader->name) == sizeof(fileinfo->name));
|
|
|
|
fileinfo->permissions = fileheader->permissions;
|
|
Maxsi::Memory::Copy(fileinfo->name, fileheader->name, sizeof(fileheader->name));
|
|
|
|
return 0;
|
|
}
|
|
|
|
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 Init(byte* theinitrd, size_t size)
|
|
{
|
|
initrd = theinitrd;
|
|
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!
|
|
|
|
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\n",
|
|
checksum, trailer->sum);
|
|
}
|
|
|
|
Syscall::Register(SYSCALL_PRINT_PATH_FILES, (void*) SysPrintPathFiles);
|
|
Syscall::Register(SYSCALL_GET_FILEINFO, (void*) SysGetFileInfo);
|
|
Syscall::Register(SYSCALL_GET_NUM_FILES, (void*) SysGetNumFiles);
|
|
}
|
|
|
|
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;
|
|
}
|
|
}
|
|
}
|