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)
28 struct hotplug_context
{
29 struct uloop_process process
;
34 struct vlist_node node
;
35 struct blob_attr
*msg
;
42 static struct uloop_fd fd_autofs_read
;
43 static int fd_autofs_write
= 0;
44 static struct ubus_auto_conn conn
;
45 struct blob_buf bb
= { 0 };
60 static const struct blobmsg_policy mount_policy
[__MOUNT_MAX
] = {
61 [MOUNT_UUID
] = { .name
= "uuid", .type
= BLOBMSG_TYPE_STRING
},
62 [MOUNT_LABEL
] = { .name
= "label", .type
= BLOBMSG_TYPE_STRING
},
63 [MOUNT_DEVICE
] = { .name
= "device", .type
= BLOBMSG_TYPE_STRING
},
64 [MOUNT_TARGET
] = { .name
= "target", .type
= BLOBMSG_TYPE_STRING
},
65 [MOUNT_OPTIONS
] = { .name
= "options", .type
= BLOBMSG_TYPE_STRING
},
66 [MOUNT_ENABLE
] = { .name
= "enabled", .type
= BLOBMSG_TYPE_INT32
},
67 [MOUNT_AUTOFS
] = { .name
= "autofs", .type
= BLOBMSG_TYPE_INT32
},
68 [MOUNT_ANON
] = { .name
= "anon", .type
= BLOBMSG_TYPE_INT32
},
69 [MOUNT_REMOVE
] = { .name
= "remove", .type
= BLOBMSG_TYPE_INT32
},
77 static const struct blobmsg_policy info_policy
[__INFO_MAX
] = {
78 [INFO_DEVICE
] = { .name
= "device", .type
= BLOBMSG_TYPE_STRING
},
82 _find_mount_point(char *device
)
86 if (asprintf(&dev
, "/dev/%s", device
) == -1)
89 mp
= find_mount_point(dev
, 0);
96 block(char *cmd
, char *action
, char *device
, int sync
, struct uloop_process
*process
)
101 char *argv
[5] = { 0 };
106 ULOG_ERR("failed to fork block process\n");
112 argv
[a
++] = "/sbin/block";
116 execvp(argv
[0], argv
);
117 ULOG_ERR("failed to spawn %s %s %s\n", *argv
, action
, device
);
121 if (!sync
&& process
) {
123 uloop_process_add(process
);
125 waitpid(pid
, &status
, 0);
126 ret
= WEXITSTATUS(status
);
128 ULOG_ERR("failed to run block. %s/%s\n", action
, device
);
136 static int send_block_notification(struct ubus_context
*ctx
, const char *action
,
137 const char *devname
, const char *target
);
138 static int hotplug_call_mount(struct ubus_context
*ctx
, const char *action
,
139 const char *devname
, uloop_process_handler cb
, void *priv
)
141 char * const argv
[] = { "hotplug-call", "mount", NULL
};
142 struct hotplug_context
*c
= NULL
;
147 c
= calloc(1, sizeof(*c
));
159 ULOG_ERR("fork() failed\n");
164 setenv("ACTION", action
, 1);
165 setenv("DEVICE", devname
, 1);
167 execv("/sbin/hotplug-call", argv
);
172 c
->process
.pid
= pid
;
175 uloop_process_add(&c
->process
);
183 static void device_mount_remove_hotplug_cb(struct uloop_process
*p
, int stat
)
185 struct hotplug_context
*hctx
= container_of(p
, struct hotplug_context
, process
);
186 struct device
*device
= hctx
->priv
;
190 unlink(device
->target
);
192 mp
= _find_mount_point(device
->name
);
194 block("autofs", "remove", device
->name
, 0, NULL
);
202 static void device_mount_remove(struct ubus_context
*ctx
, struct device
*device
)
204 static const char *action
= "remove";
206 hotplug_call_mount(ctx
, action
, device
->name
,
207 device_mount_remove_hotplug_cb
, device
);
209 send_block_notification(ctx
, action
, device
->name
, device
->target
);
212 static void device_mount_add(struct ubus_context
*ctx
, struct device
*device
)
217 if (asprintf(&path
, "/tmp/run/blockd/%s", device
->name
) == -1)
220 if (!lstat(device
->target
, &st
)) {
221 if (S_ISLNK(st
.st_mode
))
222 unlink(device
->target
);
223 else if (S_ISDIR(st
.st_mode
))
224 rmdir(device
->target
);
227 tmp
= strrchr(device
->target
, '/');
228 if (tmp
&& tmp
!= device
->target
&& tmp
!= &device
->target
[strlen(path
)-1]) {
230 mkdir_p(device
->target
, 0755);
234 if (symlink(path
, device
->target
)) {
235 ULOG_ERR("failed to symlink %s->%s (%d) - %m\n", device
->target
, path
, errno
);
237 static const char *action
= "add";
238 hotplug_call_mount(ctx
, action
, device
->name
, NULL
, NULL
);
239 send_block_notification(ctx
, action
, device
->name
, device
->target
);
245 device_move(struct device
*device_o
, struct device
*device_n
)
249 if (device_o
->autofs
!= device_n
->autofs
)
252 if (device_o
->anon
|| device_n
->anon
)
255 if (device_o
->autofs
) {
256 unlink(device_o
->target
);
257 if (asprintf(&path
, "/tmp/run/blockd/%s", device_n
->name
) == -1)
260 if (symlink(path
, device_n
->target
))
261 ULOG_ERR("failed to symlink %s->%s (%d) - %m\n", device_n
->target
, path
, errno
);
265 mkdir(device_n
->target
, 0755);
266 if (mount(device_o
->target
, device_n
->target
, NULL
, MS_MOVE
, NULL
))
267 rmdir(device_n
->target
);
269 rmdir(device_o
->target
);
275 static void vlist_nop_update(struct vlist_tree
*tree
,
276 struct vlist_node
*node_new
,
277 struct vlist_node
*node_old
)
281 VLIST_TREE(devices
, avl_strcmp
, vlist_nop_update
, false, false);
284 block_hotplug(struct ubus_context
*ctx
, struct ubus_object
*obj
,
285 struct ubus_request_data
*req
, const char *method
,
286 struct blob_attr
*msg
)
288 struct blob_attr
*data
[__MOUNT_MAX
];
289 struct device
*device
;
290 struct blob_attr
*_msg
;
291 char *devname
, *_name
;
292 char *target
= NULL
, *__target
;
293 char *_target
= NULL
;
295 blobmsg_parse(mount_policy
, __MOUNT_MAX
, data
, blob_data(msg
), blob_len(msg
));
297 if (!data
[MOUNT_DEVICE
])
298 return UBUS_STATUS_INVALID_ARGUMENT
;
300 devname
= blobmsg_get_string(data
[MOUNT_DEVICE
]);
302 if (data
[MOUNT_TARGET
]) {
303 target
= blobmsg_get_string(data
[MOUNT_TARGET
]);
305 if (asprintf(&_target
, "/mnt/%s",
306 blobmsg_get_string(data
[MOUNT_DEVICE
])) == -1)
312 if (data
[MOUNT_REMOVE
])
313 device
= vlist_find(&devices
, devname
, device
, node
);
315 device
= calloc_a(sizeof(*device
), &_msg
, blob_raw_len(msg
),
316 &_name
, strlen(devname
) + 1, &__target
, strlen(target
) + 1);
322 return UBUS_STATUS_UNKNOWN_ERROR
;
325 if (data
[MOUNT_REMOVE
]) {
326 vlist_delete(&devices
, &device
->node
);
329 device_mount_remove(ctx
, device
);
336 struct device
*old
= vlist_find(&devices
, devname
, device
, node
);
338 device
->autofs
= data
[MOUNT_AUTOFS
] ? blobmsg_get_u32(data
[MOUNT_AUTOFS
]) : 0;
339 device
->anon
= data
[MOUNT_ANON
] ? blobmsg_get_u32(data
[MOUNT_ANON
]) : 0;
341 memcpy(_msg
, msg
, blob_raw_len(msg
));
342 device
->name
= _name
;
343 strcpy(_name
, devname
);
344 device
->target
= __target
;
345 strcpy(__target
, target
);
349 vlist_add(&devices
, &device
->node
, device
->name
);
351 if (old
&& !device_move(old
, device
)) {
352 if (device
->autofs
) {
353 device_mount_remove(ctx
, old
);
354 device_mount_add(ctx
, device
);
356 block("mount", NULL
, NULL
, 0, NULL
);
358 } else if (device
->autofs
) {
359 device_mount_add(ctx
, device
);
366 static int blockd_mount(struct ubus_context
*ctx
, struct ubus_object
*obj
,
367 struct ubus_request_data
*req
, const char *method
,
368 struct blob_attr
*msg
)
370 static const char *action
= "add";
371 struct blob_attr
*data
[__MOUNT_MAX
];
372 struct device
*device
;
375 blobmsg_parse(mount_policy
, __MOUNT_MAX
, data
, blob_data(msg
), blob_len(msg
));
377 if (!data
[MOUNT_DEVICE
])
378 return UBUS_STATUS_INVALID_ARGUMENT
;
380 devname
= blobmsg_get_string(data
[MOUNT_DEVICE
]);
382 device
= vlist_find(&devices
, devname
, device
, node
);
384 return UBUS_STATUS_UNKNOWN_ERROR
;
386 hotplug_call_mount(ctx
, action
, device
->name
, NULL
, NULL
);
387 send_block_notification(ctx
, action
, device
->name
, device
->target
);
392 struct blockd_umount_context
{
393 struct ubus_context
*ctx
;
394 struct ubus_request_data req
;
397 static void blockd_umount_hotplug_cb(struct uloop_process
*p
, int stat
)
399 struct hotplug_context
*hctx
= container_of(p
, struct hotplug_context
, process
);
400 struct blockd_umount_context
*c
= hctx
->priv
;
402 ubus_complete_deferred_request(c
->ctx
, &c
->req
, 0);
408 static int blockd_umount(struct ubus_context
*ctx
, struct ubus_object
*obj
,
409 struct ubus_request_data
*req
, const char *method
,
410 struct blob_attr
*msg
)
412 struct blob_attr
*data
[__MOUNT_MAX
];
413 struct blockd_umount_context
*c
;
414 static const char *action
= "remove";
416 static char oldtarget
[PATH_MAX
];
417 struct device
*device
;
420 blobmsg_parse(mount_policy
, __MOUNT_MAX
, data
, blob_data(msg
), blob_len(msg
));
422 if (!data
[MOUNT_DEVICE
])
423 return UBUS_STATUS_INVALID_ARGUMENT
;
425 devname
= blobmsg_get_string(data
[MOUNT_DEVICE
]);
426 device
= vlist_find(&devices
, devname
, device
, node
);
428 strncpy(oldtarget
, device
->target
, sizeof(oldtarget
)-1);
429 oldtarget
[PATH_MAX
- 1] = '\0';
432 c
= calloc(1, sizeof(*c
));
434 return UBUS_STATUS_UNKNOWN_ERROR
;
437 ubus_defer_request(ctx
, req
, &c
->req
);
439 err
= hotplug_call_mount(ctx
, action
, devname
, blockd_umount_hotplug_cb
, c
);
442 return UBUS_STATUS_UNKNOWN_ERROR
;
445 send_block_notification(ctx
, action
, devname
, oldtarget
);
450 static void block_info_dump(struct blob_buf
*b
, struct device
*device
)
456 blob_for_each_attr(v
, device
->msg
, rem
)
457 blobmsg_add_blob(b
, v
);
459 mp
= _find_mount_point(device
->name
);
461 blobmsg_add_string(b
, "mount", mp
);
463 } else if (device
->autofs
&& device
->target
) {
464 blobmsg_add_string(b
, "mount", device
->target
);
469 block_info(struct ubus_context
*ctx
, struct ubus_object
*obj
,
470 struct ubus_request_data
*req
, const char *method
,
471 struct blob_attr
*msg
)
473 struct blob_attr
*data
[__INFO_MAX
];
474 struct device
*device
= NULL
;
476 blobmsg_parse(info_policy
, __INFO_MAX
, data
, blob_data(msg
), blob_len(msg
));
478 if (data
[INFO_DEVICE
]) {
479 device
= vlist_find(&devices
, blobmsg_get_string(data
[INFO_DEVICE
]), device
, node
);
481 return UBUS_STATUS_INVALID_ARGUMENT
;
484 blob_buf_init(&bb
, 0);
486 block_info_dump(&bb
, device
);
490 a
= blobmsg_open_array(&bb
, "devices");
491 vlist_for_each_element(&devices
, device
, node
) {
494 t
= blobmsg_open_table(&bb
, "");
495 block_info_dump(&bb
, device
);
496 blobmsg_close_table(&bb
, t
);
498 blobmsg_close_array(&bb
, a
);
500 ubus_send_reply(ctx
, req
, bb
.head
);
505 static const struct ubus_method block_methods
[] = {
506 UBUS_METHOD("hotplug", block_hotplug
, mount_policy
),
507 UBUS_METHOD("mount", blockd_mount
, mount_policy
),
508 UBUS_METHOD("umount", blockd_umount
, mount_policy
),
509 UBUS_METHOD("info", block_info
, info_policy
),
512 static struct ubus_object_type block_object_type
=
513 UBUS_OBJECT_TYPE("block", block_methods
);
515 static struct ubus_object block_object
= {
517 .type
= &block_object_type
,
518 .methods
= block_methods
,
519 .n_methods
= ARRAY_SIZE(block_methods
),
522 /* send ubus event for successful mounts, useful for procd triggers */
523 static int send_block_notification(struct ubus_context
*ctx
, const char *action
,
524 const char *devname
, const char *target
)
526 struct blob_buf buf
= { 0 };
527 char evname
[16] = "mount.";
533 strncat(evname
, action
, sizeof(evname
) - 1);
535 blob_buf_init(&buf
, 0);
538 blobmsg_add_string(&buf
, "device", devname
);
541 blobmsg_add_string(&buf
, "target", target
);
543 err
= ubus_notify(ctx
, &block_object
, evname
, buf
.head
, -1);
549 ubus_connect_handler(struct ubus_context
*ctx
)
553 ret
= ubus_add_object(ctx
, &block_object
);
555 fprintf(stderr
, "Failed to add object: %s\n", ubus_strerror(ret
));
558 static int autofs_umount(void)
560 umount2(AUTOFS_MOUNT_PATH
, MNT_DETACH
);
564 static void autofs_read_handler(struct uloop_fd
*u
, unsigned int events
)
566 union autofs_v5_packet_union pktu
;
567 const struct autofs_v5_packet
*pkt
;
568 int cmd
= AUTOFS_IOC_READY
;
571 while (read(u
->fd
, &pktu
, sizeof(pktu
)) == -1) {
577 if (pktu
.hdr
.type
!= autofs_ptype_missing_indirect
) {
578 ULOG_ERR("unknown packet type %d\n", pktu
.hdr
.type
);
582 pkt
= &pktu
.missing_indirect
;
583 ULOG_ERR("kernel is requesting a mount -> %s\n", pkt
->name
);
584 if (lstat(pkt
->name
, &st
) == -1)
585 if (block("autofs", "add", (char *)pkt
->name
, 1, NULL
))
586 cmd
= AUTOFS_IOC_FAIL
;
588 if (ioctl(fd_autofs_write
, cmd
, pkt
->wait_queue_token
) < 0)
589 ULOG_ERR("failed to report back to kernel\n");
592 static void autofs_expire(struct uloop_timeout
*t
)
594 struct autofs_packet_expire pkt
;
596 while (ioctl(fd_autofs_write
, AUTOFS_IOC_EXPIRE
, &pkt
) == 0)
597 block("autofs", "remove", pkt
.name
, 1, NULL
);
599 uloop_timeout_set(t
, AUTOFS_EXPIRE_TIMER
);
602 struct uloop_timeout autofs_expire_timer
= {
606 static int autofs_mount(void)
608 unsigned long autofs_timeout
= AUTOFS_TIMEOUT
;
614 if (pipe(pipefd
) < 0) {
615 ULOG_ERR("failed to get kernel pipe\n");
619 snprintf(source
, sizeof(source
), "mountd(pid%u)", getpid());
620 snprintf(opts
, sizeof(opts
), "fd=%d,pgrp=%u,minproto=5,maxproto=5", pipefd
[1], (unsigned) getpgrp());
621 mkdir(AUTOFS_MOUNT_PATH
, 0555);
622 if (mount(source
, AUTOFS_MOUNT_PATH
, "autofs", 0, opts
)) {
623 ULOG_ERR("unable to mount autofs on %s\n", AUTOFS_MOUNT_PATH
);
629 fd_autofs_read
.fd
= pipefd
[0];
630 fd_autofs_read
.cb
= autofs_read_handler
;
631 uloop_fd_add(&fd_autofs_read
, ULOOP_READ
);
633 fd_autofs_write
= open(AUTOFS_MOUNT_PATH
, O_RDONLY
);
634 if(fd_autofs_write
< 0) {
636 ULOG_ERR("failed to open direcory\n");
640 ioctl(fd_autofs_write
, AUTOFS_IOC_PROTOVER
, &kproto_version
);
641 if (kproto_version
!= 5) {
642 ULOG_ERR("only kernel protocol version 5 is tested. You have %d.\n",
646 if (ioctl(fd_autofs_write
, AUTOFS_IOC_SETTIMEOUT
, &autofs_timeout
))
647 ULOG_ERR("failed to set autofs timeout\n");
649 uloop_timeout_set(&autofs_expire_timer
, AUTOFS_EXPIRE_TIMER
);
651 fcntl(fd_autofs_write
, F_SETFD
, fcntl(fd_autofs_write
, F_GETFD
) | FD_CLOEXEC
);
652 fcntl(fd_autofs_read
.fd
, F_SETFD
, fcntl(fd_autofs_read
.fd
, F_GETFD
) | FD_CLOEXEC
);
657 static void blockd_startup_cb(struct uloop_process
*p
, int stat
)
659 send_block_notification(&conn
.ctx
, "ready", NULL
, NULL
);
662 static struct uloop_process startup_process
= {
663 .cb
= blockd_startup_cb
,
666 static void blockd_startup(struct uloop_timeout
*t
)
668 block("autofs", "start", NULL
, 0, &startup_process
);
671 struct uloop_timeout startup
= {
672 .cb
= blockd_startup
,
675 int main(int argc
, char **argv
)
677 /* make sure blockd is in it's own POSIX process group */
680 ulog_open(ULOG_SYSLOG
| ULOG_STDIO
, LOG_DAEMON
, "blockd");
685 conn
.cb
= ubus_connect_handler
;
686 ubus_auto_connect(&conn
);
688 uloop_timeout_set(&startup
, 1000);
695 vlist_flush_all(&devices
);