payloads/libpayload/drivers/storage/storage.c

   1 /*
   2  *
   3  * Copyright (C) 2012 secunet Security Networks AG
   4  *
   5  * Redistribution and use in source and binary forms, with or without
   6  * modification, are permitted provided that the following conditions
   7  * are met:
   8  * 1. Redistributions of source code must retain the above copyright
   9  *    notice, this list of conditions and the following disclaimer.
  10  * 2. Redistributions in binary form must reproduce the above copyright
  11  *    notice, this list of conditions and the following disclaimer in the
  12  *    documentation and/or other materials provided with the distribution.
  13  * 3. The name of the author may not be used to endorse or promote products
  14  *    derived from this software without specific prior written permission.
  15  *
  16  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
  17  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  18  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  19  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
  20  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  21  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  22  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  23  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  24  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  25  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  26  * SUCH DAMAGE.
  27  */
  28
  29 #include <libpayload.h>
  30 #include <pci/pci.h>
  31 #include <storage/ahci.h>
  32 #include <storage/nvme.h>
  33 #include <storage/storage.h>
  34
  35 static storage_dev_t **devices = NULL;
  36 static size_t devices_length = 0;
  37 static size_t dev_count = 0;
  38
  39 int storage_attach_device(storage_dev_t *const dev)
  40 {
  41         if (dev_count == devices_length) {
  42                 const size_t new_len =
  43                         (0 == devices_length) ? 4 : devices_length << 1;
  44                 storage_dev_t **const new_devices =
  45                         realloc(devices, new_len * sizeof(storage_dev_t *));
  46                 if (!new_devices)
  47                         return -1;
  48                 devices = new_devices;
  49                 memset(devices + devices_length, '\0',
  50                         (new_len - devices_length) * sizeof(storage_dev_t *));
  51                 devices_length = new_len;
  52         }
  53         devices[dev_count++] = dev;
  54
  55         return 0;
  56 }
  57
  58 int storage_device_count(void)
  59 {
  60         return dev_count;
  61 }
  62
  63 /**
  64  * Probe for drive with given number
  65  *
  66  * Looks for a drive with number dev_num and polls for a medium
  67  * in the drive if appropriate.
  68  *
  69  * @dev_num device number counted from 0
  70  */
  71 storage_poll_t storage_probe(const size_t dev_num)
  72 {
  73         if (dev_num >= dev_count)
  74                 return POLL_NO_DEVICE;
  75         else if (devices[dev_num]->poll)
  76                 return devices[dev_num]->poll(devices[dev_num]);
  77         else
  78                 return POLL_MEDIUM_PRESENT;
  79 }
  80
  81 /**
  82  * Read 512-byte blocks
  83  *
  84  * Reads count blocks of 512 bytes from block start of drive dev_num
  85  * into buf.
  86  *
  87  * @dev_num device number counted from 0
  88  * @start number of first block to read from
  89  * @count number of blocks to read
  90  * @buf buffer where the read data should be written
  91  */
  92 ssize_t storage_read_blocks512(const size_t dev_num,
  93                                const lba_t start, const size_t count,
  94                                unsigned char *const buf)
  95 {
  96         if ((dev_num < dev_count) && devices[dev_num]->read_blocks512)
  97                 return devices[dev_num]->read_blocks512(
  98                                 devices[dev_num], start, count, buf);
  99         else
 100                 return -1;
 101 }
 102
 103 /**
 104  * Initializes storage controllers
 105  *
 106  * This function should be called once at startup to bring up supported
 107  * storage controllers.
 108  */
 109 void storage_initialize(void)
 110 {
 111 #if CONFIG(LP_PCI)
 112         struct pci_dev *dev;
 113         for (dev = lib_sysinfo.pacc.devices; dev; dev = dev->next) {
 114                 switch (dev->device_class) {
 115 #if CONFIG(LP_STORAGE_AHCI)
 116                 case PCI_CLASS_STORAGE_AHCI:
 117                         ahci_initialize(dev);
 118                         break;
 119 #endif
 120 #if CONFIG(LP_STORAGE_NVME)
 121                 case PCI_CLASS_STORAGE_NVME:
 122                         nvme_initialize(dev);
 123                         break;
 124 #endif
 125                 default:
 126                         break;
 127                 }
 128         }
 129 #endif
 130 }