lunaix-os/kernel/block/blkpart_gpt.c

   1 #include <klibc/string.h>
   2 #include <lunaix/blkpart_gpt.h>
   3 #include <lunaix/block.h>
   4 #include <lunaix/mm/valloc.h>
   5 #include <lunaix/status.h>
   6 #include <lunaix/syslog.h>
   7
   8 #include <lib/crc.h>
   9
  10 #define GPT_BLKSIZE 512
  11 #define LBA2OFF(lba) ((lba)*GPT_BLKSIZE)
  12 #define ENT_PER_BLK (GPT_BLKSIZE / sizeof(struct gpt_entry))
  13
  14 #define GPTSIG_LO 0x20494645UL
  15 #define GPTSIG_HI 0x54524150UL
  16
  17 static u8_t NULL_GUID[16] = { 0 };
  18
  19 LOG_MODULE("GPT")
  20
  21 int
  22 blkpart_parse(struct device* master, struct gpt_header* header)
  23 {
  24     struct block_dev* bdev = (struct block_dev*)master->underlay;
  25     if (!bdev)
  26         return ENODEV;
  27
  28     int errno;
  29     u32_t ent_lba = (u32_t)header->ents_lba;
  30     struct gpt_entry* ents_parial = (struct gpt_entry*)valloc(GPT_BLKSIZE);
  31
  32     for (size_t i = 0; i < header->ents_len; i++) {
  33         if (!(i % ENT_PER_BLK)) {
  34             errno = master->read(
  35               master, ents_parial, LBA2OFF(ent_lba++), GPT_BLKSIZE);
  36             if (errno < 0) {
  37                 goto done;
  38             }
  39         }
  40
  41         struct gpt_entry* ent = &ents_parial[i % ENT_PER_BLK];
  42
  43         // assuming the entries are not scattered around
  44         if (!memcmp(ent->pguid, NULL_GUID, 16)) {
  45             break;
  46         }
  47
  48         // Convert UEFI's 512B LB representation into local LBA range.
  49         u64_t slba_local = (ent->start_lba * GPT_BLKSIZE) / bdev->blk_size;
  50         u64_t elba_local = (ent->end_lba * GPT_BLKSIZE) / (u64_t)bdev->blk_size;
  51
  52         kprintf("%s: guid part#%d: %d..%d\n",
  53                 bdev->bdev_id,
  54                 i,
  55                 (u32_t)slba_local,
  56                 (u32_t)elba_local);
  57         // we ignore the partition name, as it rarely used.
  58         blk_mount_part(bdev, NULL, i, slba_local, elba_local);
  59     }
  60
  61 done:
  62     vfree(ents_parial);
  63     return errno;
  64 }
  65
  66 int
  67 blkpart_probegpt(struct device* master)
  68 {
  69     int errno;
  70     struct gpt_header* gpt_hdr = (struct gpt_header*)valloc(GPT_BLKSIZE);
  71
  72     if ((errno = master->read(master, gpt_hdr, LBA2OFF(1), LBA2OFF(1))) < 0) {
  73         goto done;
  74     }
  75
  76     if (*(u32_t*)&gpt_hdr->signature != GPTSIG_LO ||
  77         *(u32_t*)&gpt_hdr->signature[4] != GPTSIG_HI) {
  78         return 0;
  79     }
  80
  81     u32_t crc = gpt_hdr->hdr_cksum;
  82     gpt_hdr->hdr_cksum = 0;
  83     if (crc32b((void*)gpt_hdr, sizeof(*gpt_hdr)) != crc) {
  84         kprintf(KWARN "checksum failed\n");
  85         // FUTURE check the backup header
  86         return EINVAL;
  87     }
  88
  89     errno = blkpart_parse(master, gpt_hdr);
  90 done:
  91     vfree(gpt_hdr);
  92     return errno;
  93 }