#include "bootx64.h"


EFI_STATUS
open(fs_t *fs, EFI_FILE_HANDLE *fd, CHAR16 *name)
{
    EFI_STATUS status;

    if (name == NULL || fd == NULL) return EFI_INVALID_PARAMETER;

    status = uefi_call_wrapper(fs->volume->Open, 5, fs->volume, &fd, name, EFI_FILE_MODE_READ, (UINT64)0);

    return status;
}

EFI_STATUS
close(fs_t *fs, EFI_FILE_HANDLE *fd)
{
    if (fs == NULL || fd == NULL) return EFI_INVALID_PARAMETER;

    return uefi_call_wrapper(fs->volume->Close, 1, fd);
}

EFI_STATUS
read(fs_t *fs, EFI_FILE_HANDLE *fd, VOID *buf, UINTN *size)
{
    if (buf == NULL || fd == NULL || size == NULL) return EFI_INVALID_PARAMETER;
    return uefi_call_wrapper(fs->volume->Read, 3, fd, size, buf);
}

EFI_STATUS
seek(fs_t *fs, EFI_FILE_HANDLE *fd, UINTN newpos)
{
    if (newpos <= 0 || fd == NULL || fs == NULL) return EFI_INVALID_PARAMETER; 
    return uefi_call_wrapper(fs->volume->SetPosition, 2, *fd, newpos);
}


EFI_STATUS
load(fs_t *fs, EFI_FILE_HANDLE *fd, CHAR16 *kname)
{
    VOID *buf = NULL;
    UINTN size;
    UINTN newpos = 0;
    
    
    
    
    read(fs ,fd, buf, &size);
    seek(fs ,fd, newpos);

    return EFI_SUCCESS;
}

EFI_STATUS
config_fs_one(EFI_HANDLE dev, VOID *fs)
{
    EFI_STATUS status;
    EFI_FILE_IO_INTERFACE *volume;
    EFI_FILE_HANDLE volume_fh;
    EFI_GUID LocalFsProtocol = LOCALFS_PROTOCOL;

    status = uefi_call_wrapper(BS->HandleProtocol, 3, dev, &FileSystemProtocol, (VOID **)&volume);
    if (EFI_ERROR(status)) return EFI_INVALID_PARAMETER;

    /* alloc */

    status = uefi_call_wrapper(volume->OpenVolume, 2, volume, &volume_fh);
    if (EFI_ERROR(status)) {
        Print(L"Can not open volume.\n");
        return status;
    }
    Print(L"Open volume.\n");
    
    fs_t *fs_tmp = (fs_t *)fs;

    SetMem(fs, sizeof(fs_t), 0);

    fs_tmp->dev = dev;
    fs_tmp->volume = volume_fh;

    status = LibInstallProtocolInterfaces(&dev, &LocalFsProtocol, fs, NULL);
    if (EFI_ERROR(status)) return status; 
    /* free */

    return EFI_SUCCESS;
}

EFI_STATUS
config_fs(fs_t *fs, EFI_HANDLE boot_handle, dev_tab_t *boot_dev)
{
    UINTN i;
    EFI_GUID *proto = NULL;

    Print(L"configure filesystems for all volume. \n");

    UINTN size = 0;
    uefi_call_wrapper(BS->LocateHandle, 5, ByProtocol, &FileSystemProtocol, NULL, &size, NULL);
    if (size == 0) return EFI_UNSUPPORTED;

    Print(L"find device, size=%d\n", size);

    dev_tab_t *dev_tab = NULL;

    /* alloc */
    EFI_STATUS status = uefi_call_wrapper(BS->AllocatePool, 3, EfiLoaderData, size, (VOID **)dev_tab);
    if (EFI_ERROR(status)) { 
        Print(L"can not allocate.\n");
        return status;
    }

    UINTN ndev = size / sizeof(EFI_HANDLE);

    for (i = 0; i < ndev; i++) {
        VOID *fs = NULL;
        config_fs_one(dev_tab[i].dev, fs);
        dev_tab[i].fs = fs;
    }

    status = uefi_call_wrapper(BS->LocateHandle, 5, ByProtocol, proto, NULL, &size, NULL);
    if (EFI_ERROR(status)) {
        Print(L"No useable filesystem found.\n");
        return status;
    }
    
    SetMem(proto, sizeof(EFI_HANDLE), 0);

    uefi_call_wrapper(BS->LocateHandle, 5, ByProtocol, &proto, NULL, &size, NULL);

    EFI_HANDLE *tab = NULL;

    SetMem(tab, size, 0);

    status = uefi_call_wrapper(BS->LocateHandle, 5, ByProtocol, proto, NULL, &size, tab);
    if (status != EFI_SUCCESS) {
        Print(L"faild to get handles\n");
    }

    size /= sizeof(EFI_HANDLE);

    UINTN idx = 0;

    for (i=0; i<size; i++) { 
        dev_tab[idx].dev = tab[i];
        if (tab[i] == boot_handle) boot_dev = dev_tab + idx;
        idx++;
    }

    return EFI_SUCCESS;
}