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
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
|
#include <lib.h>
#include <comus/fs.h>
#include <comus/mboot.h>
struct disk fs_disks[N_DISKS];
struct file_system fs_loaded_file_systems[N_DISKS];
void fs_init(void)
{
size_t idx = 0;
// zero structures
memsetv(fs_disks, 0, sizeof(fs_disks));
memsetv(fs_loaded_file_systems, 0, sizeof(fs_loaded_file_systems));
// check for initrd
size_t rd_len;
void *rd = mboot_get_initrd(&rd_len);
if (rd != NULL) {
assert(idx < N_DISKS, "Too many disks, limit is: %d\n", N_DISKS);
fs_disks[idx] = (struct disk) {
.present = 1,
.id = idx,
.type = DISK_TYPE_RAMDISK,
.rd.start = rd,
.rd.len = rd_len,
};
idx++;
}
// check for ide/ata devices
struct ide_devicelist ide_list = ide_devices_enumerate();
for (size_t i = 0; i < ide_list.num_devices; i++) {
assert(idx < N_DISKS, "Too many disks, limit is: %d\n", N_DISKS);
fs_disks[idx] = (struct disk) {
.present = 1,
.id = idx,
.type = DISK_TYPE_ATA,
.ide = ide_list.devices[i],
};
idx++;
}
INFO("loaded %zu disks\n", idx);
// TODO: load filesystems on disks
}
struct disk *fs_get_root_disk(void)
{
// NOTE: currently im just getting the first disk
// found, is this fine?
for (int i = 0; i < N_DISKS; i++) {
struct disk *disk = &fs_disks[i];
if (disk->present)
return disk;
}
return NULL;
}
struct file_system *fs_get_root_file_system(void)
{
// NOTE: currently im just getting the first file system
// found, is this fine?
for (int i = 0; i < N_DISKS; i++) {
struct file_system *fs = &fs_loaded_file_systems[i];
if (fs->present)
return fs;
}
return NULL;
}
int fs_find_file_abs(struct file_system *fs, char *abs_path, struct file *res)
{
(void)fs;
(void)abs_path;
(void)res;
panic("fs_find_file_abs NOT YET IMPLEMENTED");
}
int fs_find_file_rel(struct file *rel, char *rel_path, struct file *res)
{
(void)rel;
(void)rel_path;
(void)res;
panic("fs_find_file_rel NOT YET IMPLEMENTED");
}
static int disk_read_rd(struct disk *disk, size_t offset, size_t len, uint8_t *buffer)
{
if (offset + len >= disk->rd.len) {
WARN("attempted to read past length of ramdisk");
return 1;
}
memcpy(buffer, disk->rd.start + offset, len);
return 0;
}
static int disk_read_ata(struct disk *disk, size_t offset, size_t len, uint8_t *buffer)
{
static size_t atabuf_len = 0;
static uint16_t *atabuf = NULL;
uint32_t numsects = (len + ATA_SECT_SIZE - 1) / ATA_SECT_SIZE;
uint32_t err = offset % ATA_SECT_SIZE;
int ret = 0;
if (atabuf == NULL || atabuf_len < numsects*ATA_SECT_SIZE) {
if ((atabuf = krealloc(atabuf, numsects*ATA_SECT_SIZE)) == NULL)
return 1;
atabuf_len = numsects*ATA_SECT_SIZE;
}
// read sectors
if ((ret = ide_device_read_sectors(disk->ide, numsects, offset / ATA_SECT_SIZE, atabuf)))
return 1;
// copy over to buffer
memcpy(buffer, atabuf + err, len);
return ret;
}
int disk_read(struct disk *disk, size_t offset, size_t len, uint8_t *buffer)
{
int ret = 0;
switch(disk->type) {
case DISK_TYPE_RAMDISK:
ret = disk_read_rd(disk, offset, len, buffer);
break;
case DISK_TYPE_ATA:
ret = disk_read_ata(disk, offset, len, buffer);
break;
default:
ERROR("attempted to read from disk with invalid type: %d\n", disk->type);
ret = 1;
}
return ret;
}
static int disk_write_rd(struct disk *disk, size_t offset, size_t len, uint8_t *buffer)
{
if (offset + len >= disk->rd.len) {
WARN("attempted to write past length of ramdisk");
return 1;
}
memcpy(disk->rd.start + offset, buffer, len);
return 0;
}
static int disk_write_ata(struct disk *disk, size_t offset, size_t len, uint8_t *buffer)
{
static size_t atabuf_len = 0;
static uint16_t *atabuf = NULL;
uint32_t numsects = (len + ATA_SECT_SIZE - 1) / ATA_SECT_SIZE;
uint32_t err = offset % ATA_SECT_SIZE;
int ret = 0;
if (atabuf == NULL || atabuf_len < numsects*ATA_SECT_SIZE) {
if ((atabuf = krealloc(atabuf, numsects*ATA_SECT_SIZE)) == NULL)
return 1;
atabuf_len = numsects*ATA_SECT_SIZE;
}
// read sectors what will be overwritten
if ((ret = ide_device_read_sectors(disk->ide, numsects, offset / ATA_SECT_SIZE, atabuf)))
return 1;
// copy custom data over
memcpy(atabuf + err, buffer, len);
// write back sectors
if ((ret = ide_device_write_sectors(disk->ide, numsects, offset / ATA_SECT_SIZE, atabuf)))
return 1;
return ret;
}
int disk_write(struct disk *disk, size_t offset, size_t len, uint8_t *buffer)
{
int ret = 0;
switch(disk->type) {
case DISK_TYPE_RAMDISK:
ret = disk_write_rd(disk, offset, len, buffer);
break;
case DISK_TYPE_ATA:
ret = disk_write_ata(disk, offset, len, buffer);
break;
default:
ERROR("attempted to write to disk with invalid type: %d\n", disk->type);
ret = 1;
}
return ret;
}
|
