efi/diskio.c

   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 }