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
|
#include "kernel/filesystem/device_inode.hpp"
#include "kapi/system.hpp"
#include "kernel/devices/block_device.hpp"
#include "kernel/devices/device.hpp"
#include "kernel/filesystem/inode.hpp"
#include <kstd/cstring>
#include <kstd/memory>
#include <kstd/vector>
#include <algorithm>
#include <cstddef>
namespace filesystem
{
device_inode::device_inode(kstd::shared_ptr<devices::device> const & device)
: inode(inode_kind::device)
, m_device(device)
{
if (!device)
{
kapi::system::panic("[FILESYSTEM] device_inode constructed with null device.");
}
}
auto device_inode::read(void * buffer, size_t offset, size_t size) const -> size_t
{
if (m_device->is_block_device())
{
return process_blocks(offset, size, buffer,
[](size_t idx, size_t off, size_t len, size_t done, devices::block_device * device,
std::byte * scratch, void * buffer) {
auto * out = static_cast<std::byte *>(buffer);
if (off == 0 && len == device->block_size())
{
device->read_block(idx, out + done);
}
else
{
device->read_block(idx, scratch);
kstd::libc::memcpy(out + done, scratch + off, len);
}
});
}
else
{
kapi::system::panic("[FILESYSTEM] device_file::read called on non-block device.");
}
}
auto device_inode::write(void const * buffer, size_t offset, size_t size) -> size_t
{
if (m_device->is_block_device())
{
return process_blocks(offset, size, const_cast<void *>(buffer),
[](size_t idx, size_t off, size_t len, size_t done, devices::block_device * device,
std::byte * scratch, void * buffer) {
auto const * in = static_cast<std::byte const *>(buffer);
if (off == 0 && len == device->block_size())
{
device->write_block(idx, in + done);
}
else
{
device->read_block(idx, scratch);
kstd::libc::memcpy(scratch + off, in + done, len);
device->write_block(idx, scratch);
}
});
}
else
{
kapi::system::panic("[FILESYSTEM] device_file::write called on non-block device.");
}
}
auto device_inode::device() const -> kstd::shared_ptr<devices::device> const &
{
return m_device;
}
auto device_inode::process_blocks(size_t offset, size_t size, void * buffer, block_op op) const -> size_t
{
if (buffer == nullptr)
{
kapi::system::panic("[FILESYSTEM] device_file::process_blocks called with null buffer.");
}
if (size == 0)
{
return 0;
}
auto * block_dev = static_cast<devices::block_device *>(m_device.get());
if (block_dev == nullptr)
{
kapi::system::panic("[FILESYSTEM] device_file: expected block_device.");
}
size_t const block_size = block_dev->block_size();
size_t const capacity = block_dev->capacity();
if (offset >= capacity)
return 0;
size_t const total_to_process = std::min(size, capacity - offset);
kstd::vector<std::byte> scratch_buffer{block_size};
auto processed = 0uz;
while (processed < total_to_process)
{
size_t const absolute_offset = offset + processed;
size_t const block_index = absolute_offset / block_size;
size_t const in_block_offset = absolute_offset % block_size;
size_t const chunk_size = std::min(total_to_process - processed, block_size - in_block_offset);
op(block_index, in_block_offset, chunk_size, processed, block_dev, scratch_buffer.data(), buffer);
processed += chunk_size;
}
return processed;
}
} // namespace filesystem
|
