forked from ish-app/ish
-
Notifications
You must be signed in to change notification settings - Fork 0
/
proc.c
270 lines (237 loc) · 7.67 KB
/
proc.c
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
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
#include <string.h>
#include <sys/stat.h>
#include "kernel/calls.h"
#include "kernel/fs.h"
#include "fs/proc.h"
#include "fs/path.h"
static int proc_lookup(const char *path, struct proc_entry *entry) {
entry->meta = &proc_root;
char component[MAX_NAME + 1];
int err = 0;
while (path_next_component(&path, component, &err)) {
if (!S_ISDIR(proc_entry_mode(entry))) {
err = _ENOTDIR;
break;
}
unsigned long index = 0;
struct proc_entry next_entry = {0};
char entry_name[MAX_NAME];
while (proc_dir_read(entry, &index, &next_entry)) {
// tack on some dynamically generated attributes
if (next_entry.meta->parent == NULL)
next_entry.meta->parent = entry->meta;
else
// this asserts that an entry has a unique parent
assert(next_entry.meta->parent == entry->meta);
if (next_entry.meta->inode == 0)
next_entry.meta->inode = index + 1;
proc_entry_getname(&next_entry, entry_name);
if (strcmp(entry_name, component) == 0)
goto found;
proc_entry_cleanup(&next_entry);
}
err = _ENOENT;
break;
found:
proc_entry_cleanup(entry);
*entry = next_entry;
}
if (err < 0)
proc_entry_cleanup(entry);
return err;
}
extern const struct fd_ops procfs_fdops;
static struct fd *proc_open(struct mount *UNUSED(mount), const char *path, int UNUSED(flags), int UNUSED(mode)) {
struct proc_entry entry = {0};
int err = proc_lookup(path, &entry);
if (err < 0)
return ERR_PTR(err);
struct fd *fd = fd_create(&procfs_fdops);
fd->proc.entry = entry;
fd->proc.data.data = NULL;
return fd;
}
static int proc_getpath(struct fd *fd, char *buf) {
char *p = buf + MAX_PATH - 1;
size_t n = 0;
p[0] = '\0';
struct proc_entry entry = fd->proc.entry;
while (entry.meta != &proc_root) {
char component[MAX_NAME];
proc_entry_getname(&entry, component);
size_t component_len = strlen(component) + 1; // plus one for the slash
p -= component_len;
n += component_len;
*p = '/';
memcpy(p + 1, component, component_len - 1);
entry.meta = entry.meta->parent;
}
memmove(buf, p, n + 1); // plus one for the null
return 0;
}
static int proc_stat(struct mount *UNUSED(mount), const char *path, struct statbuf *stat) {
struct proc_entry entry = {0};
int err = proc_lookup(path, &entry);
if (err < 0)
return err;
int ret = proc_entry_stat(&entry, stat);
proc_entry_cleanup(&entry);
return ret;
}
static int proc_fstat(struct fd *fd, struct statbuf *stat) {
return proc_entry_stat(&fd->proc.entry, stat);
}
static int proc_refresh_data(struct fd *fd) {
mode_t_ mode = proc_entry_mode(&fd->proc.entry);
if (S_ISDIR(mode))
return _EISDIR;
assert(S_ISREG(mode));
if (fd->proc.data.data == NULL) {
fd->proc.data.capacity = 4096;
fd->proc.data.data = malloc(fd->proc.data.capacity); // default size
}
fd->proc.data.size = 0;
struct proc_entry *entry = &fd->proc.entry;
int err = entry->meta->show(entry, &fd->proc.data);
if (err < 0)
return err;
return 0;
}
static off_t_ proc_seek(struct fd *fd, off_t_ off, int whence) {
int err = proc_refresh_data(fd);
if (err < 0)
return err;
err = generic_seek(fd, off, whence, fd->proc.data.size);
if (err < 0)
return err;
return fd->offset;
}
static ssize_t proc_pread(struct fd *fd, void *buf, size_t bufsize, off_t off) {
if (fd->proc.entry.meta->pread) {
struct proc_data data = {buf, bufsize, bufsize};
return fd->proc.entry.meta->pread(&fd->proc.entry, &data, off);
}
int err = proc_refresh_data(fd);
if (err < 0)
return err;
const char *data = fd->proc.data.data;
assert(data != NULL);
size_t remaining = fd->proc.data.size - off;
if ((size_t) off > fd->proc.data.size)
remaining = 0;
size_t n = bufsize;
if (n > remaining)
n = remaining;
memcpy(buf, data + off, n);
return n;
}
static void proc_buf_write(struct proc_data *buf, const void *data, size_t size, size_t off) {
assert(off <= buf->size);
size_t tidemark = off + size;
if (tidemark > buf->capacity) {
size_t new_capacity = buf->capacity + 1;
while (tidemark > new_capacity)
new_capacity *= 2;
char *new_data = realloc(buf->data, new_capacity);
if (new_data == NULL) {
// just give up, error reporting is a pain
return;
}
buf->data = new_data;
buf->capacity = new_capacity;
}
assert(tidemark <= buf->capacity);
memcpy(buf->data + off, data, size);
if (tidemark > buf->size) {
buf->size = tidemark;
}
}
static ssize_t proc_pwrite(struct fd *fd, const void *buf, size_t bufsize, off_t off) {
mode_t_ mode = proc_entry_mode(&fd->proc.entry);
if (S_ISDIR(mode))
return _EISDIR;
assert(S_ISREG(mode));
if (fd->proc.entry.meta->pwrite) {
struct proc_data data = {(char *) buf, bufsize, bufsize};
return fd->proc.entry.meta->pwrite(&fd->proc.entry, &data, off);
}
if (!fd->proc.entry.meta->update) {
return _EPERM;
}
struct proc_data data = {(char *)buf, bufsize, bufsize};
fd->proc.entry.meta->update(&fd->proc.entry, &data);
return bufsize;
}
static int proc_readdir(struct fd *fd, struct dir_entry *entry) {
struct proc_entry proc_entry = {0};
bool any_left = proc_dir_read(&fd->proc.entry, &fd->offset, &proc_entry);
if (!any_left)
return 0;
proc_entry_getname(&proc_entry, entry->name);
entry->inode = proc_entry.meta->inode;
proc_entry_cleanup(&proc_entry);
return 1;
}
static int proc_close(struct fd *fd) {
if (fd->proc.data.data != NULL)
free(fd->proc.data.data);
proc_entry_cleanup(&fd->proc.entry);
return 0;
}
const struct fd_ops procfs_fdops = {
.pread = proc_pread,
.pwrite = proc_pwrite,
.lseek = proc_seek,
.readdir = proc_readdir,
.close = proc_close,
};
static ssize_t proc_readlink(struct mount *UNUSED(mount), const char *path, char *buf, size_t bufsize) {
struct proc_entry entry = {0};
int err = proc_lookup(path, &entry);
if (err < 0)
return err;
if (!S_ISLNK(proc_entry_mode(&entry))) {
proc_entry_cleanup(&entry);
return _EINVAL;
}
char target[MAX_PATH + 1];
err = entry.meta->readlink(&entry, target);
proc_entry_cleanup(&entry);
if (err < 0)
return err;
if (bufsize > strlen(target))
bufsize = strlen(target);
memcpy(buf, target, bufsize);
return bufsize;
}
static int proc_unlink(struct mount *UNUSED(mount), const char *path) {
struct proc_entry entry = {0};
int err = proc_lookup(path, &entry);
if (err < 0)
return err;
err = _EPERM;
if (entry.meta->unlink)
err = entry.meta->unlink(&entry);
proc_entry_cleanup(&entry);
return err;
}
void proc_buf_append(struct proc_data *buf, const void *data, size_t size) {
proc_buf_write(buf, data, size, buf->size);
}
void proc_printf(struct proc_data *buf, const char *format, ...) {
char data[4096];
va_list args;
va_start(args, format);
size_t size = vsnprintf(data, sizeof(data), format, args);
va_end(args);
proc_buf_append(buf, data, size);
}
const struct fs_ops procfs = {
.name = "proc", .magic = 0x9fa0,
.open = proc_open,
.getpath = proc_getpath,
.stat = proc_stat,
.fstat = proc_fstat,
.readlink = proc_readlink,
.unlink = proc_unlink,
};