Loading files

This commit is contained in:
Justin C. Miller
2020-02-25 22:22:12 -08:00
parent 4accfd136e
commit d94907ae79
5 changed files with 82 additions and 15 deletions

View File

@@ -1,25 +1,43 @@
#include <uefi/types.h>
#include <uefi/protos/file.h>
#include <uefi/protos/file_info.h>
#include <uefi/protos/loaded_image.h>
#include <uefi/protos/simple_file_system.h>
#include "fs.h"
#include "console.h"
#include "error.h"
#include "memory.h"
namespace boot {
namespace fs {
file::file(uefi::protos::file *f) :
m_file(f)
file::file(uefi::protos::file *f, uefi::boot_services *bs) :
m_file(f),
m_bs(bs)
{
}
file::file(file &&o) :
m_file(o.m_file)
file::file(file &o) :
m_file(o.m_file),
m_bs(o.m_bs)
{
o.m_file = nullptr;
}
file::file(file &&o) :
m_file(o.m_file),
m_bs(o.m_bs)
{
o.m_file = nullptr;
}
file::~file()
{
if (m_file)
m_file->close();
}
file
file::open(const wchar_t *path)
{
@@ -29,7 +47,38 @@ file::open(const wchar_t *path)
m_file->open(&fh, path, uefi::file_mode::read, uefi::file_attr::none),
L"Could not open relative path to file");
return file(fh);
return file(fh, m_bs);
}
void *
file::load(size_t *out_size)
{
uint8_t buffer[sizeof(uefi::protos::file_info) + 100];
size_t size = sizeof(buffer);
uefi::guid info_guid = uefi::protos::file_info::guid;
try_or_raise(
m_file->get_info(&info_guid, &size, &buffer),
L"Could not get file info");
uefi::protos::file_info *info =
reinterpret_cast<uefi::protos::file_info*>(&buffer);
size_t pages = memory::bytes_to_pages(info->file_size);
void *data = nullptr;
try_or_raise(
m_bs->allocate_pages(
uefi::allocate_type::any_pages,
uefi::memory_type::loader_data,
pages, &data),
L"Could not allocate pages to load file");
try_or_raise(
m_file->read(&size, data),
L"Could not read from file");
*out_size = size;
return data;
}
file
@@ -57,7 +106,7 @@ get_boot_volume(uefi::handle image, uefi::boot_services *bs)
fs->open_volume(&f),
L"Could not open the boot volume");
return file(f);
return file(f, bs);
}
} // namespace fs