| 1 | /* |
|---|
| 2 | * Copyright 2011-2014 Intel Corporation - All Rights Reserved |
|---|
| 3 | */ |
|---|
| 4 | |
|---|
| 5 | #include <fs.h> |
|---|
| 6 | #include <ilog2.h> |
|---|
| 7 | #include <disk.h> |
|---|
| 8 | #include <dprintf.h> |
|---|
| 9 | #include "efi.h" |
|---|
| 10 | |
|---|
| 11 | static inline EFI_STATUS read_blocks(EFI_BLOCK_IO *bio, uint32_t id, |
|---|
| 12 | sector_t lba, UINTN bytes, void *buf) |
|---|
| 13 | { |
|---|
| 14 | return uefi_call_wrapper(bio->ReadBlocks, 5, bio, id, lba, bytes, buf); |
|---|
| 15 | } |
|---|
| 16 | |
|---|
| 17 | static inline EFI_STATUS write_blocks(EFI_BLOCK_IO *bio, uint32_t id, |
|---|
| 18 | sector_t lba, UINTN bytes, void *buf) |
|---|
| 19 | { |
|---|
| 20 | return uefi_call_wrapper(bio->WriteBlocks, 5, bio, id, lba, bytes, buf); |
|---|
| 21 | } |
|---|
| 22 | |
|---|
| 23 | static int efi_rdwr_sectors(struct disk *disk, void *buf, |
|---|
| 24 | sector_t lba, size_t count, bool is_write) |
|---|
| 25 | { |
|---|
| 26 | struct efi_disk_private *priv = (struct efi_disk_private *)disk->private; |
|---|
| 27 | EFI_BLOCK_IO *bio = priv->bio; |
|---|
| 28 | EFI_STATUS status; |
|---|
| 29 | UINTN bytes = count * disk->sector_size; |
|---|
| 30 | |
|---|
| 31 | if (is_write) |
|---|
| 32 | status = write_blocks(bio, disk->disk_number, lba, bytes, buf); |
|---|
| 33 | else |
|---|
| 34 | status = read_blocks(bio, disk->disk_number, lba, bytes, buf); |
|---|
| 35 | |
|---|
| 36 | if (status != EFI_SUCCESS) |
|---|
| 37 | Print(L"Failed to %s blocks: 0x%x\n", |
|---|
| 38 | is_write ? L"write" : L"read", |
|---|
| 39 | status); |
|---|
| 40 | |
|---|
| 41 | return count << disk->sector_shift; |
|---|
| 42 | } |
|---|
| 43 | |
|---|
| 44 | struct disk *efi_disk_init(void *private) |
|---|
| 45 | { |
|---|
| 46 | static struct disk disk; |
|---|
| 47 | struct efi_disk_private *priv = (struct efi_disk_private *)private; |
|---|
| 48 | EFI_HANDLE handle = priv->dev_handle; |
|---|
| 49 | EFI_BLOCK_IO *bio; |
|---|
| 50 | EFI_DISK_IO *dio; |
|---|
| 51 | EFI_STATUS status; |
|---|
| 52 | |
|---|
| 53 | status = uefi_call_wrapper(BS->HandleProtocol, 3, handle, |
|---|
| 54 | &DiskIoProtocol, (void **)&dio); |
|---|
| 55 | if (status != EFI_SUCCESS) |
|---|
| 56 | return NULL; |
|---|
| 57 | |
|---|
| 58 | status = uefi_call_wrapper(BS->HandleProtocol, 3, handle, |
|---|
| 59 | &BlockIoProtocol, (void **)&bio); |
|---|
| 60 | if (status != EFI_SUCCESS) |
|---|
| 61 | return NULL; |
|---|
| 62 | |
|---|
| 63 | /* |
|---|
| 64 | * XXX Do we need to map this to a BIOS disk number? |
|---|
| 65 | */ |
|---|
| 66 | disk.disk_number = bio->Media->MediaId; |
|---|
| 67 | |
|---|
| 68 | disk.sector_size = bio->Media->BlockSize; |
|---|
| 69 | disk.rdwr_sectors = efi_rdwr_sectors; |
|---|
| 70 | disk.sector_shift = ilog2(disk.sector_size); |
|---|
| 71 | |
|---|
| 72 | dprintf("sector_size=%d, disk_number=%d\n", disk.sector_size, |
|---|
| 73 | disk.disk_number); |
|---|
| 74 | |
|---|
| 75 | priv->bio = bio; |
|---|
| 76 | priv->dio = dio; |
|---|
| 77 | disk.private = private; |
|---|
| 78 | #if 0 |
|---|
| 79 | |
|---|
| 80 | disk.part_start = part_start; |
|---|
| 81 | disk.secpercyl = disk.h * disk.s; |
|---|
| 82 | |
|---|
| 83 | |
|---|
| 84 | disk.maxtransfer = MaxTransfer; |
|---|
| 85 | |
|---|
| 86 | dprintf("disk %02x cdrom %d type %d sector %u/%u offset %llu limit %u\n", |
|---|
| 87 | media_id, cdrom, ebios, sector_size, disk.sector_shift, |
|---|
| 88 | part_start, disk.maxtransfer); |
|---|
| 89 | #endif |
|---|
| 90 | |
|---|
| 91 | return &disk; |
|---|
| 92 | } |
|---|
