One significant benefit of using the netlink interface is that it avoids opening NBD device files since openers can hang in the kernel in some cases (it's hard to resolve with traditional ioctl interfaces, but I don't want to explain more kernel internals here.)
In fact, using the NBD netlink interface is now always recommended. Signed-off-by: Gao Xiang <[email protected]> --- lib/backends/nbd.c | 37 +++++++++++++++++++++++++++++++++++++ lib/liberofs_nbd.h | 1 + mount/main.c | 18 +++++++++++++++--- 3 files changed, 53 insertions(+), 3 deletions(-) diff --git a/lib/backends/nbd.c b/lib/backends/nbd.c index 2e54814af6ca..77c4f609d5de 100644 --- a/lib/backends/nbd.c +++ b/lib/backends/nbd.c @@ -515,6 +515,39 @@ err_nls_free: nl_socket_free(socket); return err; } + +int erofs_nbd_nl_disconnect(int index) +{ + struct nl_sock *socket; + struct nl_msg *msg; + int driver_id, err; + + socket= erofs_nbd_get_nl_sock(&driver_id); + if (IS_ERR(socket)) + return PTR_ERR(socket); + msg = nlmsg_alloc(); + if (!msg) { + erofs_err("Couldn't allocate netlink message"); + err = -ENOMEM; + goto err_nls_free; + } + + err = -EINVAL; + genlmsg_put(msg, NL_AUTO_PORT, NL_AUTO_SEQ, driver_id, 0, 0, + NBD_CMD_DISCONNECT, 0); + NLA_PUT_U32(msg, NBD_ATTR_INDEX, index); + err = nl_send_sync(socket, msg); + if (err < 0) + erofs_err("Failed to disconnect device %d, check dmesg", err); + nl_socket_free(socket); + return err; +nla_put_failure: + erofs_err("Failed to create netlink message"); + nlmsg_free(msg); +err_nls_free: + nl_socket_free(socket); + return err; +} #else int erofs_nbd_nl_connect(int *index, int blkbits, u64 blocks, const char *identifier) @@ -532,6 +565,10 @@ int erofs_nbd_nl_reconfigure(int index, const char *identifier, { return -EOPNOTSUPP; } +int erofs_nbd_nl_disconnect(int index) +{ + return -EOPNOTSUPP; +} #endif int erofs_nbd_do_it(int nbdfd) diff --git a/lib/liberofs_nbd.h b/lib/liberofs_nbd.h index 260605ae201a..78c8af511bec 100644 --- a/lib/liberofs_nbd.h +++ b/lib/liberofs_nbd.h @@ -49,4 +49,5 @@ int erofs_nbd_nl_connect(int *index, int blkbits, u64 blocks, int erofs_nbd_nl_reconnect(int index, const char *identifier); int erofs_nbd_nl_reconfigure(int index, const char *identifier, bool autoclear); +int erofs_nbd_nl_disconnect(int index); #endif diff --git a/mount/main.c b/mount/main.c index ed6bcdcfe26d..b3b2e0fc33e0 100644 --- a/mount/main.c +++ b/mount/main.c @@ -1306,9 +1306,9 @@ out_err: int erofsmount_umount(char *target) { char *device = NULL, *mountpoint = NULL; + int err, fd, nbdnum; struct stat st; FILE *mounts; - int err, fd; size_t n; char *s; bool isblk; @@ -1376,6 +1376,14 @@ int erofsmount_umount(char *target) goto err_out; } + if (isblk && !mountpoint && + S_ISBLK(st.st_mode) && major(st.st_rdev) == EROFS_NBD_MAJOR) { + nbdnum = erofs_nbd_get_index_from_minor(minor(st.st_rdev)); + err = erofs_nbd_nl_disconnect(nbdnum); + if (err != -EOPNOTSUPP) + return err; + } + /* Avoid TOCTOU issue with NBD_CFLAG_DISCONNECT_ON_CLOSE */ fd = open(isblk ? target : device, O_RDWR); if (fd < 0) { @@ -1393,8 +1401,12 @@ int erofsmount_umount(char *target) err = fstat(fd, &st); if (err < 0) err = -errno; - else if (S_ISBLK(st.st_mode) && major(st.st_rdev) == EROFS_NBD_MAJOR) - err = erofs_nbd_disconnect(fd); + else if (S_ISBLK(st.st_mode) && major(st.st_rdev) == EROFS_NBD_MAJOR) { + nbdnum = erofs_nbd_get_index_from_minor(minor(st.st_rdev)); + err = erofs_nbd_nl_disconnect(nbdnum); + if (err == -EOPNOTSUPP) + err = erofs_nbd_disconnect(fd); + } close(fd); err_out: free(device); -- 2.43.5
