13 #include <linux/limits.h>
14 #include <linux/auto_fs4.h>
16 #include <libubox/uloop.h>
17 #include <libubox/vlist.h>
18 #include <libubox/ulog.h>
19 #include <libubox/avl-cmp.h>
22 #include "libfstools/libfstools.h"
24 #define AUTOFS_MOUNT_PATH "/tmp/run/blockd/"
25 #define AUTOFS_TIMEOUT 30
26 #define AUTOFS_EXPIRE_TIMER (5 * 1000)
29 struct vlist_node node;
30 struct blob_attr *msg;
37 static struct uloop_fd fd_autofs_read;
38 static int fd_autofs_write = 0;
39 static struct ubus_auto_conn conn;
40 struct blob_buf bb = { 0 };
55 static const struct blobmsg_policy mount_policy[__MOUNT_MAX] = {
56 [MOUNT_UUID] = { .name = "uuid", .type = BLOBMSG_TYPE_STRING },
57 [MOUNT_LABEL] = { .name = "label", .type = BLOBMSG_TYPE_STRING },
58 [MOUNT_DEVICE] = { .name = "device", .type = BLOBMSG_TYPE_STRING },
59 [MOUNT_TARGET] = { .name = "target", .type = BLOBMSG_TYPE_STRING },
60 [MOUNT_OPTIONS] = { .name = "options", .type = BLOBMSG_TYPE_STRING },
61 [MOUNT_ENABLE] = { .name = "enabled", .type = BLOBMSG_TYPE_INT32 },
62 [MOUNT_AUTOFS] = { .name = "autofs", .type = BLOBMSG_TYPE_INT32 },
63 [MOUNT_ANON] = { .name = "anon", .type = BLOBMSG_TYPE_INT32 },
64 [MOUNT_REMOVE] = { .name = "remove", .type = BLOBMSG_TYPE_INT32 },
68 _find_mount_point(char *device)
72 snprintf(dev, sizeof(dev), "/dev/%s", device);
74 return find_mount_point(dev, 0);
78 block(char *cmd, char *action, char *device)
83 char *argv[5] = { 0 };
88 ULOG_ERR("failed to fork block process\n");
92 argv[a++] = "/sbin/block";
96 execvp(argv[0], argv);
97 ULOG_ERR("failed to spawn %s %s %s\n", *argv, action, device);
101 waitpid(pid, &status, 0);
102 ret = WEXITSTATUS(status);
104 ULOG_ERR("failed to run block. %s/%s\n", action, device);
112 device_free(struct device *device)
114 struct blob_attr *data[__MOUNT_MAX];
116 char *path = NULL, _path[64], *mp;
118 blobmsg_parse(mount_policy, __MOUNT_MAX, data,
119 blob_data(device->msg), blob_len(device->msg));
121 if (data[MOUNT_AUTOFS]) {
122 target = device->target;
123 snprintf(_path, sizeof(_path), "/tmp/run/blockd/%s",
124 blobmsg_get_string(data[MOUNT_DEVICE]));
127 path = target = device->target;
130 mp = _find_mount_point(device->name);
132 if (umount2(path, MNT_DETACH))
133 ULOG_ERR("failed to unmount %s\n", path);
139 if (data[MOUNT_AUTOFS])
146 device_add(struct device *device)
148 struct blob_attr *data[__MOUNT_MAX];
151 blobmsg_parse(mount_policy, __MOUNT_MAX, data,
152 blob_data(device->msg), blob_len(device->msg));
154 if (!data[MOUNT_AUTOFS])
157 snprintf(path, sizeof(path), "/tmp/run/blockd/%s",
158 blobmsg_get_string(data[MOUNT_DEVICE]));
159 if (symlink(path, device->target))
160 ULOG_ERR("failed to symlink %s->%s\n", device->target, path);
164 device_move(struct device *device_o, struct device *device_n)
168 if (device_o->autofs != device_n->autofs)
171 if (device_o->anon || device_n->anon)
174 if (device_o->autofs) {
175 unlink(device_o->target);
176 snprintf(path, sizeof(path), "/tmp/run/blockd/%s", device_n->name);
177 if (symlink(path, device_n->target))
178 ULOG_ERR("failed to symlink %s->%s\n", device_n->target, path);
180 mkdir(device_n->target, 0755);
181 if (mount(device_o->target, device_n->target, NULL, MS_MOVE, NULL))
182 rmdir(device_n->target);
184 rmdir(device_o->target);
191 devices_update_cb(struct vlist_tree *tree, struct vlist_node *node_new,
192 struct vlist_node *node_old)
194 struct device *device_o = NULL, *device_n = NULL;
197 device_o = container_of(node_old, struct device, node);
200 device_n = container_of(node_new, struct device, node);
202 if (device_o && device_n) {
203 if (device_move(device_o, device_n)) {
204 device_free(device_o);
205 device_add(device_n);
206 if (!device_n->autofs)
207 block("mount", NULL, NULL);
209 } else if (device_n) {
210 device_add(device_n);
212 device_free(device_o);
219 VLIST_TREE(devices, avl_strcmp, devices_update_cb, false, false);
222 block_hotplug(struct ubus_context *ctx, struct ubus_object *obj,
223 struct ubus_request_data *req, const char *method,
224 struct blob_attr *msg)
226 struct blob_attr *data[__MOUNT_MAX];
227 struct device *device;
228 struct blob_attr *_msg;
229 char *devname, *_name;
230 char *target = NULL, *__target;
233 blobmsg_parse(mount_policy, __MOUNT_MAX, data, blob_data(msg), blob_len(msg));
235 if (!data[MOUNT_DEVICE])
236 return UBUS_STATUS_INVALID_ARGUMENT;
238 devname = blobmsg_get_string(data[MOUNT_DEVICE]);
240 if (data[MOUNT_TARGET]) {
241 target = blobmsg_get_string(data[MOUNT_TARGET]);
243 snprintf(_target, sizeof(_target), "/mnt/%s",
244 blobmsg_get_string(data[MOUNT_DEVICE]));
248 if (data[MOUNT_REMOVE])
249 device = vlist_find(&devices, devname, device, node);
251 device = calloc_a(sizeof(*device), &_msg, blob_raw_len(msg),
252 &_name, strlen(devname) + 1, &__target, strlen(target) + 1);
255 return UBUS_STATUS_UNKNOWN_ERROR;
257 vlist_update(&devices);
258 if (data[MOUNT_REMOVE]) {
259 vlist_delete(&devices, &device->node);
261 if (data[MOUNT_AUTOFS])
262 device->autofs = blobmsg_get_u32(data[MOUNT_AUTOFS]);
265 if (data[MOUNT_ANON])
266 device->anon = blobmsg_get_u32(data[MOUNT_ANON]);
270 memcpy(_msg, msg, blob_raw_len(msg));
271 device->name = _name;
272 strcpy(_name, devname);
273 device->target = __target;
274 strcpy(__target, target);
275 vlist_add(&devices, &device->node, blobmsg_get_string(data[MOUNT_DEVICE]));
277 vlist_flush(&devices);
283 block_info(struct ubus_context *ctx, struct ubus_object *obj,
284 struct ubus_request_data *req, const char *method,
285 struct blob_attr *msg)
287 struct device *device;
290 blob_buf_init(&bb, 0);
291 a = blobmsg_open_array(&bb, "devices");
292 vlist_for_each_element(&devices, device, node) {
293 void *t = blobmsg_open_table(&bb, "");
298 blob_for_each_attr(v, device->msg, rem)
299 blobmsg_add_blob(&bb, v);
301 mp = _find_mount_point(device->name);
303 blobmsg_add_string(&bb, "mount", mp);
306 blobmsg_close_table(&bb, t);
308 blobmsg_close_array(&bb, a);
309 ubus_send_reply(ctx, req, bb.head);
314 static const struct ubus_method block_methods[] = {
315 UBUS_METHOD("hotplug", block_hotplug, mount_policy),
316 UBUS_METHOD_NOARG("info", block_info),
319 static struct ubus_object_type block_object_type =
320 UBUS_OBJECT_TYPE("block", block_methods);
322 static struct ubus_object block_object = {
324 .type = &block_object_type,
325 .methods = block_methods,
326 .n_methods = ARRAY_SIZE(block_methods),
330 ubus_connect_handler(struct ubus_context *ctx)
334 ret = ubus_add_object(ctx, &block_object);
336 fprintf(stderr, "Failed to add object: %s\n", ubus_strerror(ret));
339 static int autofs_umount(void)
341 umount2(AUTOFS_MOUNT_PATH, MNT_DETACH);
345 static void autofs_read_handler(struct uloop_fd *u, unsigned int events)
347 union autofs_v5_packet_union pktu;
348 const struct autofs_v5_packet *pkt;
349 int cmd = AUTOFS_IOC_READY;
352 while (read(u->fd, &pktu, sizeof(pktu)) == -1) {
358 if (pktu.hdr.type != autofs_ptype_missing_indirect) {
359 ULOG_ERR("unknown packet type %d\n", pktu.hdr.type);
363 pkt = &pktu.missing_indirect;
364 ULOG_ERR("kernel is requesting a mount -> %s\n", pkt->name);
365 if (lstat(pkt->name, &st) == -1)
366 if (block("autofs", "add", (char *)pkt->name))
367 cmd = AUTOFS_IOC_FAIL;
369 if (ioctl(fd_autofs_write, cmd, pkt->wait_queue_token) < 0)
370 ULOG_ERR("failed to report back to kernel\n");
373 static void autofs_expire(struct uloop_timeout *t)
375 struct autofs_packet_expire pkt;
377 while (ioctl(fd_autofs_write, AUTOFS_IOC_EXPIRE, &pkt) == 0)
378 block("autofs", "remove", pkt.name);
380 uloop_timeout_set(t, AUTOFS_EXPIRE_TIMER);
383 struct uloop_timeout autofs_expire_timer = {
387 static int autofs_mount(void)
389 int autofs_timeout = AUTOFS_TIMEOUT;
395 if (pipe(pipefd) < 0) {
396 ULOG_ERR("failed to get kernel pipe\n");
400 snprintf(source, sizeof(source), "mountd(pid%u)", getpid());
401 snprintf(opts, sizeof(opts), "fd=%d,pgrp=%u,minproto=5,maxproto=5", pipefd[1], (unsigned) getpgrp());
402 mkdir(AUTOFS_MOUNT_PATH, 0555);
403 if (mount(source, AUTOFS_MOUNT_PATH, "autofs", 0, opts)) {
404 ULOG_ERR("unable to mount autofs on %s\n", AUTOFS_MOUNT_PATH);
410 fd_autofs_read.fd = pipefd[0];
411 fd_autofs_read.cb = autofs_read_handler;
412 uloop_fd_add(&fd_autofs_read, ULOOP_READ);
414 fd_autofs_write = open(AUTOFS_MOUNT_PATH, O_RDONLY);
415 if(fd_autofs_write < 0) {
417 ULOG_ERR("failed to open direcory\n");
421 ioctl(fd_autofs_write, AUTOFS_IOC_PROTOVER, &kproto_version);
422 if (kproto_version != 5) {
423 ULOG_ERR("only kernel protocol version 5 is tested. You have %d.\n",
427 if (ioctl(fd_autofs_write, AUTOFS_IOC_SETTIMEOUT, &autofs_timeout))
428 ULOG_ERR("failed to set autofs timeout\n");
430 uloop_timeout_set(&autofs_expire_timer, AUTOFS_EXPIRE_TIMER);
432 fcntl(fd_autofs_write, F_SETFD, fcntl(fd_autofs_write, F_GETFD) | FD_CLOEXEC);
433 fcntl(fd_autofs_read.fd, F_SETFD, fcntl(fd_autofs_read.fd, F_GETFD) | FD_CLOEXEC);
438 static void blockd_startup(struct uloop_timeout *t)
440 block("autofs", "start", NULL);
443 struct uloop_timeout startup = {
444 .cb = blockd_startup,
447 int main(int argc, char **argv)
449 ulog_open(ULOG_SYSLOG | ULOG_STDIO, LOG_DAEMON, "blockd");
454 conn.cb = ubus_connect_handler;
455 ubus_auto_connect(&conn);
457 uloop_timeout_set(&startup, 1000);
464 vlist_flush_all(&devices);