1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
|
#include <fs/gpt.h>
#include <ahci.h>
#include <mem.h>
#include <memory.h>
#include <vga.h>
struct gpt_header *read_gpt_disk(struct disk *disk) {
uint16_t *gpt = kmalloc(512);
if (!read(disk->port, 1, 1, gpt))
return NULL;
struct gpt_header *part = (void*) gpt;
if (memcmp(part->sig, "EFI PART", 8) != 0) {
vga_puts("not gpt");
return NULL;
}
vga_puts("found gpt table on disk 0\n");
vga_puts("number of partitions: ");
vga_putn(part->number_partitions);
vga_putc('\n');
vga_puts("starting lba: ");
vga_putn(part->first_lba);
vga_putc('\n');
vga_puts("size per entry: ");
vga_putn(part->size_partitions);
vga_putc('\n');
return part;
}
struct gpt_entry *get_gpt_entry(struct disk *disk, struct gpt_header *header, size_t num) {
if (num >= header->number_partitions)
return NULL;
uint16_t *data = kmalloc(512);
if (!read(disk->port, header->first_lba + num, 1, data))
return NULL;
struct gpt_entry *entry = (void *) data;
/*
vga_puts("unique guid for first partition: ");
vga_putx(entry->guid);
vga_putc('\n');
*/
vga_puts("starting partition lba: ");
vga_putn(entry->lba.start);
vga_putc('\n');
vga_puts("ending partition lba: ");
vga_putn(entry->lba.end);
vga_putc('\n');
return entry;
}
char *read_gpt_entry(struct disk *disk, struct gpt_entry *entry) {
uint16_t *data = kmalloc(512 * (entry->lba.end - entry->lba.start));
uint16_t *ptr = data;
for (size_t lba = entry->lba.start; lba < entry->lba.end; lba += 1024) {
if (!read(disk->port, lba, entry->lba.end - lba < 1024 ? entry->lba.end - lba : 1024, ptr)) {
vga_puts("failed to read disk");
kfree(data);
return NULL;
}
ptr += 1024 * 256;
}
return (char *)data;
}
|