limine/common/fs/file.s2.c

126 lines
3.0 KiB
C

#include <stddef.h>
#include <stdint.h>
#include <fs/file.h>
#include <fs/ext2.h>
#include <fs/fat32.h>
#include <fs/iso9660.h>
#include <lib/print.h>
#include <lib/misc.h>
#include <mm/pmm.h>
#include <lib/part.h>
#include <lib/libc.h>
#include <pxe/tftp.h>
char *fs_get_label(struct volume *part) {
char *ret;
if ((ret = fat32_get_label(part)) != NULL) {
return ret;
}
if ((ret = ext2_get_label(part)) != NULL) {
return ret;
}
return NULL;
}
bool fs_get_guid(struct guid *guid, struct volume *part) {
if (ext2_get_guid(guid, part) == true) {
return true;
}
return false;
}
bool case_insensitive_fopen = false;
struct file_handle *fopen(struct volume *part, const char *filename) {
size_t filename_new_len = strlen(filename) + 2;
char *filename_new = ext_mem_alloc(filename_new_len);
if (filename[0] != '/') {
filename_new[0] = '/';
strcpy(&filename_new[1], filename);
} else {
strcpy(filename_new, filename);
}
filename = filename_new;
struct file_handle *ret;
if (part->pxe) {
if ((ret = tftp_open(part, "", filename)) == NULL) {
return NULL;
}
return ret;
}
if ((ret = ext2_open(part, filename)) != NULL) {
goto success;
}
if ((ret = iso9660_open(part, filename)) != NULL) {
goto success;
}
if ((ret = fat32_open(part, filename)) != NULL) {
goto success;
}
return NULL;
success:
ret->path = (char *)filename;
ret->path_len = filename_new_len;
return ret;
}
void fclose(struct file_handle *fd) {
if (fd->is_memfile) {
if (fd->readall == false) {
pmm_free(fd->fd, fd->size);
}
} else {
fd->close(fd);
}
pmm_free(fd->path, fd->path_len);
pmm_free(fd, sizeof(struct file_handle));
}
void fread(struct file_handle *fd, void *buf, uint64_t loc, uint64_t count) {
if (fd->is_memfile) {
memcpy(buf, fd->fd + loc, count);
} else {
fd->read(fd, buf, loc, count);
}
}
void *freadall(struct file_handle *fd, uint32_t type) {
#if defined (__x86_64__) || defined (__aarch64__) || defined (__riscv64)
return freadall_mode(fd, type, false);
}
void *freadall_mode(struct file_handle *fd, uint32_t type, bool allow_high_allocs) {
#endif
if (fd->is_memfile) {
if (fd->readall) {
return fd->fd;
}
memmap_alloc_range((uint64_t)(size_t)fd->fd, ALIGN_UP(fd->size, 4096), type, 0, true, false, false);
fd->readall = true;
return fd->fd;
} else {
#if defined (__x86_64__) || defined (__aarch64__) || defined (__riscv64)
void *ret = ext_mem_alloc_type_aligned_mode(fd->size, type, 4096, allow_high_allocs);
#else
void *ret = ext_mem_alloc_type_aligned(fd->size, type, 4096);
#endif
fd->read(fd, ret, 0, fd->size);
fd->close(fd);
fd->fd = ret;
fd->readall = true;
fd->is_memfile = true;
return ret;
}
}