diff options
-rw-r--r-- | udev/lib/exported_symbols | 1 | ||||
-rw-r--r-- | udev/lib/libudev-device.c | 4 | ||||
-rw-r--r-- | udev/lib/libudev-monitor.c | 89 | ||||
-rw-r--r-- | udev/lib/libudev.h | 1 |
4 files changed, 70 insertions, 25 deletions
diff --git a/udev/lib/exported_symbols b/udev/lib/exported_symbols index 620b8964a2..72a2937229 100644 --- a/udev/lib/exported_symbols +++ b/udev/lib/exported_symbols @@ -23,6 +23,7 @@ udev_device_get_devnum udev_device_get_seqnum udev_devices_enumerate udev_monitor_new_from_socket +udev_monitor_new_from_netlink udev_monitor_enable_receiving udev_monitor_ref udev_monitor_unref diff --git a/udev/lib/libudev-device.c b/udev/lib/libudev-device.c index fd7c962d3a..00b9440386 100644 --- a/udev/lib/libudev-device.c +++ b/udev/lib/libudev-device.c @@ -194,6 +194,10 @@ void udev_device_unref(struct udev_device *udev_device) free(udev_device->subsystem); name_list_cleanup(udev_device->udev, &udev_device->link_list); name_list_cleanup(udev_device->udev, &udev_device->env_list); + free(udev_device->action); + free(udev_device->driver); + free(udev_device->devpath_old); + free(udev_device->physdevpath); info(udev_device->udev, "udev_device: %p released\n", udev_device); free(udev_device); } diff --git a/udev/lib/libudev-monitor.c b/udev/lib/libudev-monitor.c index 454a646046..3e1731e382 100644 --- a/udev/lib/libudev-monitor.c +++ b/udev/lib/libudev-monitor.c @@ -29,6 +29,7 @@ #include <sys/stat.h> #include <sys/socket.h> #include <sys/un.h> +#include <linux/netlink.h> #include "libudev.h" #include "libudev-private.h" @@ -38,7 +39,8 @@ struct udev_monitor { struct udev *udev; int refcount; int sock; - struct sockaddr_un saddr; + struct sockaddr_nl snl; + struct sockaddr_un sun; socklen_t addrlen; }; @@ -74,13 +76,13 @@ struct udev_monitor *udev_monitor_new_from_socket(struct udev *udev, const char udev_monitor->refcount = 1; udev_monitor->udev = udev; - udev_monitor->saddr.sun_family = AF_LOCAL; - strcpy(udev_monitor->saddr.sun_path, socket_path); - udev_monitor->addrlen = offsetof(struct sockaddr_un, sun_path) + strlen(udev_monitor->saddr.sun_path); + udev_monitor->sun.sun_family = AF_LOCAL; + strcpy(udev_monitor->sun.sun_path, socket_path); + udev_monitor->addrlen = offsetof(struct sockaddr_un, sun_path) + strlen(udev_monitor->sun.sun_path); /* translate leading '@' to abstract namespace */ - if (udev_monitor->saddr.sun_path[0] == '@') - udev_monitor->saddr.sun_path[0] = '\0'; + if (udev_monitor->sun.sun_path[0] == '@') + udev_monitor->sun.sun_path[0] = '\0'; udev_monitor->sock = socket(AF_LOCAL, SOCK_DGRAM, 0); if (udev_monitor->sock == -1) { @@ -92,20 +94,56 @@ struct udev_monitor *udev_monitor_new_from_socket(struct udev *udev, const char return udev_monitor; } +struct udev_monitor *udev_monitor_new_from_netlink(struct udev *udev) +{ + struct udev_monitor *udev_monitor; + + if (udev == NULL) + return NULL; + udev_monitor = malloc(sizeof(struct udev_monitor)); + if (udev_monitor == NULL) + return NULL; + memset(udev_monitor, 0x00, sizeof(struct udev_monitor)); + udev_monitor->refcount = 1; + udev_monitor->udev = udev; + + udev_monitor->sock = socket(PF_NETLINK, SOCK_DGRAM, NETLINK_KOBJECT_UEVENT); + if (udev_monitor->sock == -1) { + err(udev, "error getting socket: %s\n", strerror(errno)); + free(udev_monitor); + return NULL; + } + + udev_monitor->snl.nl_family = AF_NETLINK; + udev_monitor->snl.nl_pid = getpid(); + udev_monitor->snl.nl_groups = 1; + + info(udev, "monitor %p created with NETLINK_KOBJECT_UEVENT\n", udev_monitor); + return udev_monitor; +} + int udev_monitor_enable_receiving(struct udev_monitor *udev_monitor) { int err; const int on = 1; - err = bind(udev_monitor->sock, (struct sockaddr *)&udev_monitor->saddr, udev_monitor->addrlen); - if (err < 0) { - err(udev_monitor->udev, "bind failed: %s\n", strerror(errno)); - return err; + if (udev_monitor->snl.nl_family != 0) { + err = bind(udev_monitor->sock, (struct sockaddr *)&udev_monitor->snl, sizeof(struct sockaddr_nl)); + if (err < 0) { + err(udev_monitor->udev, "bind failed: %s\n", strerror(errno)); + return err; + } + info(udev_monitor->udev, "monitor %p listening on netlink\n", udev_monitor); + } else if (udev_monitor->sun.sun_family != 0) { + err = bind(udev_monitor->sock, (struct sockaddr *)&udev_monitor->sun, udev_monitor->addrlen); + if (err < 0) { + err(udev_monitor->udev, "bind failed: %s\n", strerror(errno)); + return err; + } + /* enable receiving of the sender credentials */ + setsockopt(udev_monitor->sock, SOL_SOCKET, SO_PASSCRED, &on, sizeof(on)); + info(udev_monitor->udev, "monitor %p listening on socket\n", udev_monitor); } - - /* enable receiving of the sender credentials */ - setsockopt(udev_monitor->sock, SOL_SOCKET, SO_PASSCRED, &on, sizeof(on)); - info(udev_monitor->udev, "monitor %p listening\n", udev_monitor); return 0; } @@ -197,9 +235,7 @@ struct udev_device *udev_monitor_receive_device(struct udev_monitor *udev_monito { struct udev_device *udev_device; struct msghdr smsg; - struct cmsghdr *cmsg; struct iovec iov; - struct ucred *cred; char cred_msg[CMSG_SPACE(sizeof(struct ucred))]; char buf[4096]; size_t bufpos; @@ -222,17 +258,20 @@ struct udev_device *udev_monitor_receive_device(struct udev_monitor *udev_monito info(udev_monitor->udev, "unable to receive message"); return NULL; } - cmsg = CMSG_FIRSTHDR(&smsg); - cred = (struct ucred *)CMSG_DATA (cmsg); - if (cmsg == NULL || cmsg->cmsg_type != SCM_CREDENTIALS) { - info(udev_monitor->udev, "no sender credentials received, message ignored"); - return NULL; - } + if (udev_monitor->sun.sun_family != 0) { + struct cmsghdr *cmsg = CMSG_FIRSTHDR(&smsg); + struct ucred *cred = (struct ucred *)CMSG_DATA (cmsg); - if (cred->uid != 0) { - info(udev_monitor->udev, "sender uid=%d, message ignored", cred->uid); - return NULL; + if (cmsg == NULL || cmsg->cmsg_type != SCM_CREDENTIALS) { + info(udev_monitor->udev, "no sender credentials received, message ignored"); + return NULL; + } + + if (cred->uid != 0) { + info(udev_monitor->udev, "sender uid=%d, message ignored", cred->uid); + return NULL; + } } /* skip header */ diff --git a/udev/lib/libudev.h b/udev/lib/libudev.h index 58c4ea97bb..2267c9ea97 100644 --- a/udev/lib/libudev.h +++ b/udev/lib/libudev.h @@ -70,6 +70,7 @@ extern int udev_devices_enumerate(struct udev *udev, const char *subsystem, struct udev_monitor; extern struct udev_monitor *udev_monitor_new_from_socket(struct udev *udev, const char *socket_path); +extern struct udev_monitor *udev_monitor_new_from_netlink(struct udev *udev); extern int udev_monitor_enable_receiving(struct udev_monitor *udev_monitor); extern struct udev_monitor *udev_monitor_ref(struct udev_monitor *udev_monitor); extern void udev_monitor_unref(struct udev_monitor *udev_monitor); |