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
|
// MIT License, Copyright (c) 2021 Marvin Borner
#include <def.h>
#include <fs.h>
#include <drivers/ide.h>
#include <drivers/mbr.h>
#include <mem.h>
#include <print.h>
#include <str.h>
struct mbr_data {
struct mbr_entry mbr;
struct vfs_dev *dev;
};
static res mbr_read(void *buf, u32 lba, u32 sector_count, struct vfs_dev *part)
{
struct mbr_data *data = part->data;
if (!data || !data->dev || !data->dev->read)
return -EINVAL;
return data->dev->read(buf, data->mbr.start + lba, sector_count, data->dev);
}
static res mbr_write(const void *buf, u32 lba, u32 sector_count, struct vfs_dev *part)
{
struct mbr_data *data = part->data;
if (!data || !data->dev || !data->dev->write)
return -EINVAL;
return data->dev->write(buf, data->mbr.start + lba, sector_count, data->dev);
}
CLEAR u8 mbr_load(struct vfs_dev *dev)
{
struct mbr mbr = { 0 };
dev->read(&mbr, 0, 1, dev); // Read first sector
if (mbr.magic != 0xaa55)
return 0;
for (u8 i = 0; i < 4; i++) {
struct mbr_entry *entry = &mbr.entries[i];
if (!entry->type || !entry->size)
continue;
struct mbr_data *data = zalloc(sizeof(*data));
memcpy(&data->mbr, entry, sizeof(*entry));
data->dev = dev;
struct vfs_dev *part = zalloc(sizeof(*part));
part->data = data;
strlcpy(part->name, dev->name, 4);
part->name[3] = '0' + i;
part->type = DEV_BLOCK;
part->read = mbr_read;
part->write = mbr_write;
/* printf("%s [start: %d; size: %dMiB; type: %d]\n", part->name, entry->start, */
/* (entry->size * SECTOR_SIZE) >> 20, entry->type); */
vfs_add_dev(part);
vfs_load(part);
}
return 1;
}
|
