platform-intel.c

   1 /*
   2  * Intel(R) Matrix Storage Manager hardware and firmware support routines
   3  *
   4  * Copyright (C) 2008 Intel Corporation
   5  *
   6  * This program is free software; you can redistribute it and/or modify it
   7  * under the terms and conditions of the GNU General Public License,
   8  * version 2, as published by the Free Software Foundation.
   9  *
  10  * This program is distributed in the hope it will be useful, but WITHOUT
  11  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  12  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
  13  * more details.
  14  *
  15  * You should have received a copy of the GNU General Public License along with
  16  * this program; if not, write to the Free Software Foundation, Inc.,
  17  * 51 Franklin St - Fifth Floor, Boston, MA 02110-1301 USA.
  18  */
  19 #include "mdadm.h"
  20 #include "platform-intel.h"
  21 #include "probe_roms.h"
  22 #include <stdio.h>
  23 #include <stdlib.h>
  24 #include <string.h>
  25 #include <unistd.h>
  26 #include <dirent.h>
  27 #include <fcntl.h>
  28 #include <sys/mman.h>
  29 #include <sys/types.h>
  30 #include <sys/stat.h>
  31 #include <limits.h>
  32
  33 static int devpath_to_ll(const char *dev_path, const char *entry,
  34                          unsigned long long *val);
  35
  36 static void free_sys_dev(struct sys_dev **list)
  37 {
  38         while (*list) {
  39                 struct sys_dev *next = (*list)->next;
  40
  41                 if ((*list)->path)
  42                         free((*list)->path);
  43                 free(*list);
  44                 *list = next;
  45         }
  46 }
  47
  48 struct sys_dev *find_driver_devices(const char *bus, const char *driver)
  49 {
  50         /* search sysfs for devices driven by 'driver' */
  51         char path[PATH_MAX];
  52         char link[PATH_MAX];
  53         char *c, *p;
  54         DIR *driver_dir;
  55         struct dirent *de;
  56         struct sys_dev *head = NULL;
  57         struct sys_dev *list = NULL;
  58         struct sys_dev *vmd = NULL;
  59         enum sys_dev_type type;
  60         unsigned long long dev_id;
  61         unsigned long long class;
  62
  63         if (strcmp(driver, "isci") == 0)
  64                 type = SYS_DEV_SAS;
  65         else if (strcmp(driver, "ahci") == 0)
  66                 type = SYS_DEV_SATA;
  67         else if (strcmp(driver, "nvme") == 0) {
  68                 /* if looking for nvme devs, first look for vmd */
  69                 vmd = find_driver_devices("pci", "vmd");
  70                 type = SYS_DEV_NVME;
  71         } else if (strcmp(driver, "vmd") == 0)
  72                 type = SYS_DEV_VMD;
  73         else
  74                 type = SYS_DEV_UNKNOWN;
  75
  76         sprintf(path, "/sys/bus/%s/drivers/%s", bus, driver);
  77         driver_dir = opendir(path);
  78         if (!driver_dir) {
  79                 if (vmd)
  80                         free_sys_dev(&vmd);
  81                 return NULL;
  82         }
  83         for (de = readdir(driver_dir); de; de = readdir(driver_dir)) {
  84                 int n;
  85                 int skip = 0;
  86
  87                 /* is 'de' a device? check that the 'subsystem' link exists and
  88                  * that its target matches 'bus'
  89                  */
  90                 sprintf(path, "/sys/bus/%s/drivers/%s/%s/subsystem",
  91                         bus, driver, de->d_name);
  92                 n = readlink(path, link, sizeof(link));
  93                 if (n < 0 || n >= (int)sizeof(link))
  94                         continue;
  95                 link[n] = '\0';
  96                 c = strrchr(link, '/');
  97                 if (!c)
  98                         continue;
  99                 if (strncmp(bus, c+1, strlen(bus)) != 0)
 100                         continue;
 101
 102                 sprintf(path, "/sys/bus/%s/drivers/%s/%s",
 103                         bus, driver, de->d_name);
 104
 105                 /* if searching for nvme - skip vmd connected one */
 106                 if (type == SYS_DEV_NVME) {
 107                         struct sys_dev *dev;
 108                         char *rp = realpath(path, NULL);
 109                         for (dev = vmd; dev; dev = dev->next) {
 110                                 if ((strncmp(dev->path, rp, strlen(dev->path)) == 0))
 111                                         skip = 1;
 112                         }
 113                         free(rp);
 114                 }
 115
 116                 /* if it's not Intel device or mark as VMD connected - skip it. */
 117                 if (devpath_to_vendor(path) != 0x8086 || skip == 1)
 118                         continue;
 119
 120                 if (devpath_to_ll(path, "device", &dev_id) != 0)
 121                         continue;
 122
 123                 if (devpath_to_ll(path, "class", &class) != 0)
 124                         continue;
 125
 126                 /*
 127                  * Each VMD device (domain) adds separate PCI bus, it is better
 128                  * to store path as a path to that bus (easier further
 129                  * determination which NVMe dev is connected to this particular
 130                  * VMD domain).
 131                  */
 132                 if (type == SYS_DEV_VMD) {
 133                         sprintf(path, "/sys/bus/%s/drivers/%s/%s/domain/device",
 134                                 bus, driver, de->d_name);
 135                 }
 136                 p = realpath(path, NULL);
 137                 if (p == NULL) {
 138                         pr_err("Unable to get real path for '%s'\n", path);
 139                         continue;
 140                 }
 141
 142                 /* start / add list entry */
 143                 if (!head) {
 144                         head = xmalloc(sizeof(*head));
 145                         list = head;
 146                 } else {
 147                         list->next = xmalloc(sizeof(*head));
 148                         list = list->next;
 149                 }
 150
 151                 if (!list) {
 152                         free_sys_dev(&head);
 153                         break;
 154                 }
 155
 156                 list->dev_id = (__u16) dev_id;
 157                 list->class = (__u32) class;
 158                 list->type = type;
 159                 list->next = NULL;
 160                 list->path = p;
 161
 162                 if ((list->pci_id = strrchr(list->path, '/')) != NULL)
 163                         list->pci_id++;
 164         }
 165         closedir(driver_dir);
 166
 167         if (vmd) {
 168                 if (list)
 169                         list->next = vmd;
 170                 else
 171                         head = vmd;
 172         }
 173
 174         return head;
 175 }
 176
 177 static struct sys_dev *intel_devices=NULL;
 178 static time_t valid_time = 0;
 179
 180 struct sys_dev *device_by_id(__u16 device_id)
 181 {
 182         struct sys_dev *iter;
 183
 184         for (iter = intel_devices; iter != NULL; iter = iter->next)
 185                 if (iter->dev_id == device_id)
 186                         return iter;
 187         return NULL;
 188 }
 189
 190 static int devpath_to_ll(const char *dev_path, const char *entry, unsigned long long *val)
 191 {
 192         char path[strlen(dev_path) + strlen(entry) + 2];
 193         int fd;
 194         int n;
 195
 196         sprintf(path, "%s/%s", dev_path, entry);
 197
 198         fd = open(path, O_RDONLY);
 199         if (fd < 0)
 200                 return -1;
 201         n = sysfs_fd_get_ll(fd, val);
 202         close(fd);
 203         return n;
 204 }
 205
 206 __u16 devpath_to_vendor(const char *dev_path)
 207 {
 208         char path[strlen(dev_path) + strlen("/vendor") + 1];
 209         char vendor[7];
 210         int fd;
 211         __u16 id = 0xffff;
 212         int n;
 213
 214         sprintf(path, "%s/vendor", dev_path);
 215
 216         fd = open(path, O_RDONLY);
 217         if (fd < 0)
 218                 return 0xffff;
 219
 220         n = read(fd, vendor, sizeof(vendor));
 221         if (n == sizeof(vendor)) {
 222                 vendor[n - 1] = '\0';
 223                 id = strtoul(vendor, NULL, 16);
 224         }
 225         close(fd);
 226
 227         return id;
 228 }
 229
 230 struct sys_dev *find_intel_devices(void)
 231 {
 232         struct sys_dev *ahci, *isci, *nvme;
 233
 234         if (valid_time > time(0) - 10)
 235                 return intel_devices;
 236
 237         if (intel_devices)
 238                 free_sys_dev(&intel_devices);
 239
 240         isci = find_driver_devices("pci", "isci");
 241         ahci = find_driver_devices("pci", "ahci");
 242         /* Searching for NVMe will return list of NVMe and VMD controllers */
 243         nvme = find_driver_devices("pci", "nvme");
 244
 245         if (!isci && !ahci) {
 246                 ahci = nvme;
 247         } else if (!ahci) {
 248                 ahci = isci;
 249                 struct sys_dev *elem = ahci;
 250                 while (elem->next)
 251                         elem = elem->next;
 252                 elem->next = nvme;
 253         } else {
 254                 struct sys_dev *elem = ahci;
 255                 while (elem->next)
 256                         elem = elem->next;
 257                 elem->next = isci;
 258                 while (elem->next)
 259                         elem = elem->next;
 260                 elem->next = nvme;
 261         }
 262         intel_devices = ahci;
 263         valid_time = time(0);
 264         return intel_devices;
 265 }
 266
 267 /*
 268  * PCI Expansion ROM Data Structure Format */
 269 struct pciExpDataStructFormat {
 270         __u8  ver[4];
 271         __u16 vendorID;
 272         __u16 deviceID;
 273         __u16 devListOffset;
 274         __u16 pciDataStructLen;
 275         __u8 pciDataStructRev;
 276 } __attribute__ ((packed));
 277
 278 struct orom_entry *orom_entries;
 279
 280 const struct orom_entry *get_orom_entry_by_device_id(__u16 dev_id)
 281 {
 282         struct orom_entry *entry;
 283         struct devid_list *devid;
 284
 285         for (entry = orom_entries; entry; entry = entry->next) {
 286                 for (devid = entry->devid_list; devid; devid = devid->next) {
 287                         if (devid->devid == dev_id)
 288                                 return entry;
 289                 }
 290         }
 291
 292         return NULL;
 293 }
 294
 295 const struct imsm_orom *get_orom_by_device_id(__u16 dev_id)
 296 {
 297         const struct orom_entry *entry = get_orom_entry_by_device_id(dev_id);
 298
 299         if (entry)
 300                 return &entry->orom;
 301
 302         return NULL;
 303 }
 304
 305 static struct orom_entry *add_orom(const struct imsm_orom *orom)
 306 {
 307         struct orom_entry *list;
 308         struct orom_entry *prev = NULL;
 309
 310         for (list = orom_entries; list; prev = list, list = list->next)
 311                 ;
 312
 313         list = xmalloc(sizeof(struct orom_entry));
 314         list->orom = *orom;
 315         list->devid_list = NULL;
 316         list->next = NULL;
 317
 318         if (prev == NULL)
 319                 orom_entries = list;
 320         else
 321                 prev->next = list;
 322
 323         return list;
 324 }
 325
 326 static void add_orom_device_id(struct orom_entry *entry, __u16 dev_id)
 327 {
 328         struct devid_list *list;
 329         struct devid_list *prev = NULL;
 330
 331         for (list = entry->devid_list; list; prev = list, list = list->next) {
 332                 if (list->devid == dev_id)
 333                         return;
 334         }
 335         list = xmalloc(sizeof(struct devid_list));
 336         list->devid = dev_id;
 337         list->next = NULL;
 338
 339         if (prev == NULL)
 340                 entry->devid_list = list;
 341         else
 342                 prev->next = list;
 343 }
 344
 345 static int scan(const void *start, const void *end, const void *data)
 346 {
 347         int offset;
 348         const struct imsm_orom *imsm_mem = NULL;
 349         int len = (end - start);
 350         struct pciExpDataStructFormat *ptr= (struct pciExpDataStructFormat *)data;
 351
 352         if (data + 0x18 > end) {
 353                 dprintf("cannot find pciExpDataStruct \n");
 354                 return 0;
 355         }
 356
 357         dprintf("ptr->vendorID: %lx __le16_to_cpu(ptr->deviceID): %lx \n",
 358                 (ulong) __le16_to_cpu(ptr->vendorID),
 359                 (ulong) __le16_to_cpu(ptr->deviceID));
 360
 361         if (__le16_to_cpu(ptr->vendorID) != 0x8086)
 362                 return 0;
 363
 364         for (offset = 0; offset < len; offset += 4) {
 365                 const void *mem = start + offset;
 366
 367                 if ((memcmp(mem, IMSM_OROM_SIGNATURE, 4) == 0)) {
 368                         imsm_mem = mem;
 369                         break;
 370                 }
 371         }
 372
 373         if (!imsm_mem)
 374                 return 0;
 375
 376         struct orom_entry *orom = add_orom(imsm_mem);
 377
 378         /* only PciDataStructure with revision 3 and above supports devices list. */
 379         if (ptr->pciDataStructRev >= 3 && ptr->devListOffset) {
 380                 const __u16 *dev_list = (void *)ptr + ptr->devListOffset;
 381                 int i;
 382
 383                 for (i = 0; dev_list[i] != 0; i++)
 384                         add_orom_device_id(orom, dev_list[i]);
 385         } else {
 386                 add_orom_device_id(orom, __le16_to_cpu(ptr->deviceID));
 387         }
 388
 389         return 0;
 390 }
 391
 392 const struct imsm_orom *imsm_platform_test(struct sys_dev *hba)
 393 {
 394         struct imsm_orom orom = {
 395                 .signature = IMSM_OROM_SIGNATURE,
 396                 .rlc = IMSM_OROM_RLC_RAID0 | IMSM_OROM_RLC_RAID1 |
 397                                         IMSM_OROM_RLC_RAID10 | IMSM_OROM_RLC_RAID5,
 398                 .sss = IMSM_OROM_SSS_4kB | IMSM_OROM_SSS_8kB |
 399                                         IMSM_OROM_SSS_16kB | IMSM_OROM_SSS_32kB |
 400                                         IMSM_OROM_SSS_64kB | IMSM_OROM_SSS_128kB |
 401                                         IMSM_OROM_SSS_256kB | IMSM_OROM_SSS_512kB |
 402                                         IMSM_OROM_SSS_1MB | IMSM_OROM_SSS_2MB,
 403                 .dpa = IMSM_OROM_DISKS_PER_ARRAY,
 404                 .tds = IMSM_OROM_TOTAL_DISKS,
 405                 .vpa = IMSM_OROM_VOLUMES_PER_ARRAY,
 406                 .vphba = IMSM_OROM_VOLUMES_PER_HBA
 407         };
 408         orom.attr = orom.rlc | IMSM_OROM_ATTR_ChecksumVerify;
 409
 410         if (check_env("IMSM_TEST_OROM_NORAID5")) {
 411                 orom.rlc = IMSM_OROM_RLC_RAID0 | IMSM_OROM_RLC_RAID1 |
 412                                 IMSM_OROM_RLC_RAID10;
 413         }
 414         if (check_env("IMSM_TEST_AHCI_EFI_NORAID5") && (hba->type == SYS_DEV_SAS)) {
 415                 orom.rlc = IMSM_OROM_RLC_RAID0 | IMSM_OROM_RLC_RAID1 |
 416                                 IMSM_OROM_RLC_RAID10;
 417         }
 418         if (check_env("IMSM_TEST_SCU_EFI_NORAID5") && (hba->type == SYS_DEV_SATA)) {
 419                 orom.rlc = IMSM_OROM_RLC_RAID0 | IMSM_OROM_RLC_RAID1 |
 420                                 IMSM_OROM_RLC_RAID10;
 421         }
 422
 423         struct orom_entry *ret = add_orom(&orom);
 424
 425         add_orom_device_id(ret, hba->dev_id);
 426
 427         return &ret->orom;
 428 }
 429
 430 static const struct imsm_orom *find_imsm_hba_orom(struct sys_dev *hba)
 431 {
 432         unsigned long align;
 433
 434         if (check_env("IMSM_TEST_OROM"))
 435                 return imsm_platform_test(hba);
 436
 437         /* return empty OROM capabilities in EFI test mode */
 438         if (check_env("IMSM_TEST_AHCI_EFI") || check_env("IMSM_TEST_SCU_EFI"))
 439                 return NULL;
 440
 441         find_intel_devices();
 442
 443         if (intel_devices == NULL)
 444                 return NULL;
 445
 446         /* scan option-rom memory looking for an imsm signature */
 447         if (check_env("IMSM_SAFE_OROM_SCAN"))
 448                 align = 2048;
 449         else
 450                 align = 512;
 451         if (probe_roms_init(align) != 0)
 452                 return NULL;
 453         probe_roms();
 454         /* ignore return value - True is returned if both adapater roms are found */
 455         scan_adapter_roms(scan);
 456         probe_roms_exit();
 457
 458         return get_orom_by_device_id(hba->dev_id);
 459 }
 460
 461 #define GUID_STR_MAX    37  /* according to GUID format:
 462                              * xxxxxxxx-xxxx-xxxx-xxxx-xxxxxxxxxxxx" */
 463
 464 #define EFI_GUID(a, b, c, d0, d1, d2, d3, d4, d5, d6, d7) \
 465 ((struct efi_guid) \
 466 {{ (a) & 0xff, ((a) >> 8) & 0xff, ((a) >> 16) & 0xff, ((a) >> 24) & 0xff, \
 467   (b) & 0xff, ((b) >> 8) & 0xff, \
 468   (c) & 0xff, ((c) >> 8) & 0xff, \
 469   (d0), (d1), (d2), (d3), (d4), (d5), (d6), (d7) }})
 470
 471 #define SYS_EFI_VAR_PATH "/sys/firmware/efi/vars"
 472 #define SYS_EFIVARS_PATH "/sys/firmware/efi/efivars"
 473 #define SCU_PROP "RstScuV"
 474 #define AHCI_PROP "RstSataV"
 475 #define AHCI_SSATA_PROP "RstsSatV"
 476 #define AHCI_CSATA_PROP "RstCSatV"
 477 #define VMD_PROP "RstUefiV"
 478
 479 #define VENDOR_GUID \
 480         EFI_GUID(0x193dfefa, 0xa445, 0x4302, 0x99, 0xd8, 0xef, 0x3a, 0xad, 0x1a, 0x04, 0xc6)
 481
 482 #define PCI_CLASS_RAID_CNTRL 0x010400
 483
 484 static int read_efi_var(void *buffer, ssize_t buf_size, char *variable_name, struct efi_guid guid)
 485 {
 486         char path[PATH_MAX];
 487         char buf[GUID_STR_MAX];
 488         int fd;
 489         ssize_t n;
 490
 491         snprintf(path, PATH_MAX, "%s/%s-%s", SYS_EFIVARS_PATH, variable_name, guid_str(buf, guid));
 492
 493         fd = open(path, O_RDONLY);
 494         if (fd < 0)
 495                 return 1;
 496
 497         /* read the variable attributes and ignore it */
 498         n = read(fd, buf, sizeof(__u32));
 499         if (n < 0) {
 500                 close(fd);
 501                 return 1;
 502         }
 503
 504         /* read the variable data */
 505         n = read(fd, buffer, buf_size);
 506         close(fd);
 507         if (n < buf_size)
 508                 return 1;
 509
 510         return 0;
 511 }
 512
 513 static int read_efi_variable(void *buffer, ssize_t buf_size, char *variable_name, struct efi_guid guid)
 514 {
 515         char path[PATH_MAX];
 516         char buf[GUID_STR_MAX];
 517         int dfd;
 518         ssize_t n, var_data_len;
 519
 520         /* Try to read the variable using the new efivarfs interface first.
 521          * If that fails, fall back to the old sysfs-efivars interface. */
 522         if (!read_efi_var(buffer, buf_size, variable_name, guid))
 523                 return 0;
 524
 525         snprintf(path, PATH_MAX, "%s/%s-%s/size", SYS_EFI_VAR_PATH, variable_name, guid_str(buf, guid));
 526
 527         dprintf("EFI VAR: path=%s\n", path);
 528         /* get size of variable data */
 529         dfd = open(path, O_RDONLY);
 530         if (dfd < 0)
 531                 return 1;
 532
 533         n = read(dfd, &buf, sizeof(buf));
 534         close(dfd);
 535         if (n < 0)
 536                 return 1;
 537         buf[n] = '\0';
 538
 539         errno = 0;
 540         var_data_len = strtoul(buf, NULL, 16);
 541         if ((errno == ERANGE && (var_data_len == LONG_MAX))
 542             || (errno != 0 && var_data_len == 0))
 543                 return 1;
 544
 545         /* get data */
 546         snprintf(path, PATH_MAX, "%s/%s-%s/data", SYS_EFI_VAR_PATH, variable_name, guid_str(buf, guid));
 547
 548         dprintf("EFI VAR: path=%s\n", path);
 549         dfd = open(path, O_RDONLY);
 550         if (dfd < 0)
 551                 return 1;
 552
 553         n = read(dfd, buffer, buf_size);
 554         close(dfd);
 555         if (n != var_data_len || n < buf_size) {
 556                 return 1;
 557         }
 558
 559         return 0;
 560 }
 561
 562 const struct imsm_orom *find_imsm_efi(struct sys_dev *hba)
 563 {
 564         struct imsm_orom orom;
 565         struct orom_entry *ret;
 566         int err;
 567
 568         if (check_env("IMSM_TEST_AHCI_EFI") || check_env("IMSM_TEST_SCU_EFI"))
 569                 return imsm_platform_test(hba);
 570
 571         /* OROM test is set, return that there is no EFI capabilities */
 572         if (check_env("IMSM_TEST_OROM"))
 573                 return NULL;
 574
 575         if (hba->type == SYS_DEV_SATA && hba->class != PCI_CLASS_RAID_CNTRL)
 576                 return NULL;
 577
 578         err = read_efi_variable(&orom, sizeof(orom), hba->type == SYS_DEV_SAS ? SCU_PROP : AHCI_PROP, VENDOR_GUID);
 579
 580         /* try to read variable for second AHCI controller */
 581         if (err && hba->type == SYS_DEV_SATA)
 582                 err = read_efi_variable(&orom, sizeof(orom), AHCI_SSATA_PROP, VENDOR_GUID);
 583
 584         /* try to read variable for combined AHCI controllers */
 585         if (err && hba->type == SYS_DEV_SATA) {
 586                 static struct orom_entry *csata;
 587
 588                 err = read_efi_variable(&orom, sizeof(orom), AHCI_CSATA_PROP, VENDOR_GUID);
 589                 if (!err) {
 590                         if (!csata)
 591                                 csata = add_orom(&orom);
 592                         add_orom_device_id(csata, hba->dev_id);
 593                         csata->type = hba->type;
 594                         return &csata->orom;
 595                 }
 596         }
 597
 598         if (hba->type == SYS_DEV_VMD) {
 599                 err = read_efi_variable(&orom, sizeof(orom), VMD_PROP, VENDOR_GUID);
 600         }
 601
 602         if (err)
 603                 return NULL;
 604
 605         ret = add_orom(&orom);
 606         add_orom_device_id(ret, hba->dev_id);
 607         ret->type = hba->type;
 608
 609         return &ret->orom;
 610 }
 611
 612 const struct imsm_orom *find_imsm_nvme(struct sys_dev *hba)
 613 {
 614         static struct orom_entry *nvme_orom;
 615
 616         if (hba->type != SYS_DEV_NVME)
 617                 return NULL;
 618
 619         if (!nvme_orom) {
 620                 struct imsm_orom nvme_orom_compat = {
 621                         .signature = IMSM_NVME_OROM_COMPAT_SIGNATURE,
 622                         .rlc = IMSM_OROM_RLC_RAID0 | IMSM_OROM_RLC_RAID1 |
 623                                                 IMSM_OROM_RLC_RAID10 | IMSM_OROM_RLC_RAID5,
 624                         .sss = IMSM_OROM_SSS_4kB | IMSM_OROM_SSS_8kB |
 625                                                 IMSM_OROM_SSS_16kB | IMSM_OROM_SSS_32kB |
 626                                                 IMSM_OROM_SSS_64kB | IMSM_OROM_SSS_128kB,
 627                         .dpa = IMSM_OROM_DISKS_PER_ARRAY_NVME,
 628                         .tds = IMSM_OROM_TOTAL_DISKS_NVME,
 629                         .vpa = IMSM_OROM_VOLUMES_PER_ARRAY,
 630                         .vphba = IMSM_OROM_TOTAL_DISKS_NVME / 2 * IMSM_OROM_VOLUMES_PER_ARRAY,
 631                         .attr = IMSM_OROM_ATTR_2TB | IMSM_OROM_ATTR_2TB_DISK,
 632                         .driver_features = IMSM_OROM_CAPABILITIES_EnterpriseSystem
 633                 };
 634                 nvme_orom = add_orom(&nvme_orom_compat);
 635         }
 636         add_orom_device_id(nvme_orom, hba->dev_id);
 637         nvme_orom->type = SYS_DEV_NVME;
 638         return &nvme_orom->orom;
 639 }
 640
 641 const struct imsm_orom *find_imsm_capability(struct sys_dev *hba)
 642 {
 643         const struct imsm_orom *cap = get_orom_by_device_id(hba->dev_id);
 644
 645         if (cap)
 646                 return cap;
 647
 648         if (hba->type == SYS_DEV_NVME)
 649                 return find_imsm_nvme(hba);
 650         if ((cap = find_imsm_efi(hba)) != NULL)
 651                 return cap;
 652         if ((cap = find_imsm_hba_orom(hba)) != NULL)
 653                 return cap;
 654
 655         return NULL;
 656 }
 657
 658 char *devt_to_devpath(dev_t dev)
 659 {
 660         char device[46];
 661
 662         sprintf(device, "/sys/dev/block/%d:%d/device", major(dev), minor(dev));
 663         return realpath(device, NULL);
 664 }
 665
 666 char *diskfd_to_devpath(int fd)
 667 {
 668         /* return the device path for a disk, return NULL on error or fd
 669          * refers to a partition
 670          */
 671         struct stat st;
 672
 673         if (fstat(fd, &st) != 0)
 674                 return NULL;
 675         if (!S_ISBLK(st.st_mode))
 676                 return NULL;
 677
 678         return devt_to_devpath(st.st_rdev);
 679 }
 680
 681 int path_attached_to_hba(const char *disk_path, const char *hba_path)
 682 {
 683         int rc;
 684
 685         if (check_env("IMSM_TEST_AHCI_DEV") ||
 686             check_env("IMSM_TEST_SCU_DEV")) {
 687                 return 1;
 688         }
 689
 690         if (!disk_path || !hba_path)
 691                 return 0;
 692         dprintf("hba: %s - disk: %s\n", hba_path, disk_path);
 693         if (strncmp(disk_path, hba_path, strlen(hba_path)) == 0)
 694                 rc = 1;
 695         else
 696                 rc = 0;
 697
 698         return rc;
 699 }
 700
 701 int devt_attached_to_hba(dev_t dev, const char *hba_path)
 702 {
 703         char *disk_path = devt_to_devpath(dev);
 704         int rc = path_attached_to_hba(disk_path, hba_path);
 705
 706         if (disk_path)
 707                 free(disk_path);
 708
 709         return rc;
 710 }
 711
 712 int disk_attached_to_hba(int fd, const char *hba_path)
 713 {
 714         char *disk_path = diskfd_to_devpath(fd);
 715         int rc = path_attached_to_hba(disk_path, hba_path);
 716
 717         if (disk_path)
 718                 free(disk_path);
 719
 720         return rc;
 721 }
 722
 723 char *vmd_domain_to_controller(struct sys_dev *hba, char *buf)
 724 {
 725         struct dirent *ent;
 726         DIR *dir;
 727         char path[PATH_MAX];
 728
 729         if (!hba)
 730                 return NULL;
 731
 732         if (hba->type != SYS_DEV_VMD)
 733                 return NULL;
 734
 735         dir = opendir("/sys/bus/pci/drivers/vmd");
 736         if (!dir)
 737                 return NULL;
 738
 739         for (ent = readdir(dir); ent; ent = readdir(dir)) {
 740                 sprintf(path, "/sys/bus/pci/drivers/vmd/%s/domain/device",
 741                         ent->d_name);
 742
 743                 if (!realpath(path, buf))
 744                         continue;
 745
 746                 if (strncmp(buf, hba->path, strlen(buf)) == 0) {
 747                         sprintf(path, "/sys/bus/pci/drivers/vmd/%s", ent->d_name);
 748                         closedir(dir);
 749                         return realpath(path, buf);
 750                 }
 751         }
 752
 753         closedir(dir);
 754         return NULL;
 755 }