chiark / gitweb /
fix signed/unsigned warning
[elogind.git] / udev / lib / libudev-monitor.c
index 2540e85c340a427149ff81d6aef0a2493dd2e338..7a0cb478526baaf625e4a89ad531c81bce725940 100644 (file)
@@ -1,7 +1,7 @@
 /*
  * libudev - interface to udev device information
  *
- * Copyright (C) 2008 Kay Sievers <kay.sievers@vrfy.org>
+ * Copyright (C) 2008-2009 Kay Sievers <kay.sievers@vrfy.org>
  *
  * This library is free software; you can redistribute it and/or
  * modify it under the terms of the GNU Lesser General Public
@@ -32,15 +32,17 @@ struct udev_monitor {
        int refcount;
        int sock;
        struct sockaddr_nl snl;
-       struct sockaddr_nl snl_peer;
+       struct sockaddr_nl snl_trusted_sender;
+       struct sockaddr_nl snl_destination;
        struct sockaddr_un sun;
        socklen_t addrlen;
        struct udev_list_node filter_subsystem_list;
 };
 
 enum udev_monitor_netlink_group {
-       UDEV_MONITOR_KERNEL     = 1,
-       UDEV_MONITOR_UDEV       = 2,
+       UDEV_MONITOR_NONE,
+       UDEV_MONITOR_KERNEL,
+       UDEV_MONITOR_UDEV,
 };
 
 #define UDEV_MONITOR_MAGIC             0xcafe1dea
@@ -113,16 +115,16 @@ struct udev_monitor *udev_monitor_new_from_socket(struct udev *udev, const char
        udev_monitor->sun.sun_family = AF_LOCAL;
        if (socket_path[0] == '@') {
                /* translate leading '@' to abstract namespace */
-               util_strlcpy(udev_monitor->sun.sun_path, socket_path, sizeof(udev_monitor->sun.sun_path));
+               util_strscpy(udev_monitor->sun.sun_path, sizeof(udev_monitor->sun.sun_path), socket_path);
                udev_monitor->sun.sun_path[0] = '\0';
                udev_monitor->addrlen = offsetof(struct sockaddr_un, sun_path) + strlen(socket_path);
        } else if (stat(socket_path, &statbuf) == 0 && S_ISSOCK(statbuf.st_mode)) {
                /* existing socket file */
-               util_strlcpy(udev_monitor->sun.sun_path, socket_path, sizeof(udev_monitor->sun.sun_path));
+               util_strscpy(udev_monitor->sun.sun_path, sizeof(udev_monitor->sun.sun_path), socket_path);
                udev_monitor->addrlen = offsetof(struct sockaddr_un, sun_path) + strlen(socket_path);
        } else {
                /* no socket file, assume abstract namespace socket */
-               util_strlcpy(&udev_monitor->sun.sun_path[1], socket_path, sizeof(udev_monitor->sun.sun_path)-1);
+               util_strscpy(&udev_monitor->sun.sun_path[1], sizeof(udev_monitor->sun.sun_path)-1, socket_path);
                udev_monitor->addrlen = offsetof(struct sockaddr_un, sun_path) + strlen(socket_path)+1;
        }
        udev_monitor->sock = socket(AF_LOCAL, SOCK_DGRAM, 0);
@@ -171,11 +173,11 @@ struct udev_monitor *udev_monitor_new_from_netlink(struct udev *udev, const char
                return NULL;
 
        if (name == NULL)
-               return NULL;
-       if (strcmp(name, "kernel") == 0)
-               group = UDEV_MONITOR_KERNEL;
+               group = UDEV_MONITOR_NONE;
        else if (strcmp(name, "udev") == 0)
                group = UDEV_MONITOR_UDEV;
+       else if (strcmp(name, "kernel") == 0)
+               group = UDEV_MONITOR_KERNEL;
        else
                return NULL;
 
@@ -193,8 +195,10 @@ struct udev_monitor *udev_monitor_new_from_netlink(struct udev *udev, const char
 
        udev_monitor->snl.nl_family = AF_NETLINK;
        udev_monitor->snl.nl_groups = group;
-       udev_monitor->snl_peer.nl_family = AF_NETLINK;
-       udev_monitor->snl_peer.nl_groups = UDEV_MONITOR_UDEV;
+
+       /* default destination for sending */
+       udev_monitor->snl_destination.nl_family = AF_NETLINK;
+       udev_monitor->snl_destination.nl_groups = UDEV_MONITOR_UDEV;
 
        dbg(udev, "monitor %p created with NETLINK_KOBJECT_UEVENT (%u)\n", udev_monitor, group);
        return udev_monitor;
@@ -281,6 +285,12 @@ int udev_monitor_filter_update(struct udev_monitor *udev_monitor)
        return err;
 }
 
+int udev_monitor_allow_unicast_sender(struct udev_monitor *udev_monitor, struct udev_monitor *sender)
+{
+       udev_monitor->snl_trusted_sender.nl_pid = sender->snl.nl_pid;
+       return 0;
+}
+
 int udev_monitor_enable_receiving(struct udev_monitor *udev_monitor)
 {
        int err;
@@ -293,6 +303,19 @@ int udev_monitor_enable_receiving(struct udev_monitor *udev_monitor)
                udev_monitor_filter_update(udev_monitor);
                err = bind(udev_monitor->sock,
                           (struct sockaddr *)&udev_monitor->snl, sizeof(struct sockaddr_nl));
+               if (err == 0) {
+                       struct sockaddr_nl snl;
+                       socklen_t addrlen;
+
+                       /*
+                        * get the address the kernel has assigned us
+                        * it is usually, but not neccessarily the pid
+                        */
+                       addrlen = sizeof(struct sockaddr_nl);
+                       err = getsockname(udev_monitor->sock, (struct sockaddr *)&snl, &addrlen);
+                       if (err == 0)
+                               udev_monitor->snl.nl_pid = snl.nl_pid;
+               }
        } else {
                return -EINVAL;
        }
@@ -314,6 +337,15 @@ int udev_monitor_set_receive_buffer_size(struct udev_monitor *udev_monitor, int
        return setsockopt(udev_monitor->sock, SOL_SOCKET, SO_RCVBUFFORCE, &size, sizeof(size));
 }
 
+int udev_monitor_disconnect(struct udev_monitor *udev_monitor)
+{
+       int err;
+
+       err = close(udev_monitor->sock);
+       udev_monitor->sock = -1;
+       return err;
+}
+
 /**
  * udev_monitor_ref:
  * @udev_monitor: udev monitor
@@ -478,10 +510,13 @@ retry:
 
        if (udev_monitor->snl.nl_family != 0) {
                if (snl.nl_groups == 0) {
-                       info(udev_monitor->udev, "unicast netlink message ignored\n");
-                       return NULL;
-               }
-               if (snl.nl_groups == UDEV_MONITOR_KERNEL) {
+                       /* unicast message, check if we trust the sender */
+                       if (udev_monitor->snl_trusted_sender.nl_pid == 0 ||
+                           snl.nl_pid != udev_monitor->snl_trusted_sender.nl_pid) {
+                               info(udev_monitor->udev, "unicast netlink message ignored\n");
+                               return NULL;
+                       }
+               } else if (snl.nl_groups == UDEV_MONITOR_KERNEL) {
                        if (snl.nl_pid > 0) {
                                info(udev_monitor->udev, "multicast kernel netlink message from pid %d ignored\n", snl.nl_pid);
                                return NULL;
@@ -509,7 +544,7 @@ retry:
                        return NULL;
                if (nlh->properties_off < sizeof(struct udev_monitor_netlink_header))
                        return NULL;
-               if (nlh->properties_off+32U > buflen)
+               if (nlh->properties_off+32 > buflen)
                        return NULL;
                bufpos = nlh->properties_off;
        } else {
@@ -545,8 +580,7 @@ retry:
                if (strncmp(key, "DEVPATH=", 8) == 0) {
                        char path[UTIL_PATH_SIZE];
 
-                       util_strlcpy(path, udev_get_sys_path(udev_monitor->udev), sizeof(path));
-                       util_strlcat(path, &key[8], sizeof(path));
+                       util_strscpyl(path, sizeof(path), udev_get_sys_path(udev_monitor->udev), &key[8], NULL);
                        udev_device_set_syspath(udev_device, path);
                        devpath_set = 1;
                } else if (strncmp(key, "SUBSYSTEM=", 10) == 0) {
@@ -564,7 +598,7 @@ retry:
                        char *slink;
                        char *next;
 
-                       util_strlcpy(devlinks, &key[9], sizeof(devlinks));
+                       util_strscpy(devlinks, sizeof(devlinks), &key[9]);
                        slink = devlinks;
                        next = strchr(slink, ' ');
                        while (next != NULL) {
@@ -622,7 +656,8 @@ retry:
        return udev_device;
 }
 
-int udev_monitor_send_device(struct udev_monitor *udev_monitor, struct udev_device *udev_device)
+int udev_monitor_send_device(struct udev_monitor *udev_monitor,
+                            struct udev_monitor *destination, struct udev_device *udev_device)
 {
        struct msghdr smsg;
        struct iovec iov[2];
@@ -637,19 +672,17 @@ int udev_monitor_send_device(struct udev_monitor *udev_monitor, struct udev_devi
        if (udev_monitor->sun.sun_family != 0) {
                const char *action;
                char header[2048];
-               size_t hlen;
+               char *s;
 
                /* header <action>@<devpath> */
                action = udev_device_get_action(udev_device);
                if (action == NULL)
                        return -EINVAL;
-               util_strlcpy(header, action, sizeof(header));
-               util_strlcat(header, "@", sizeof(header));
-               hlen = util_strlcat(header, udev_device_get_devpath(udev_device), sizeof(header))+1;
-               if (hlen >= sizeof(header))
+               s = header;
+               if (util_strpcpyl(&s, sizeof(header), action, "@", udev_device_get_devpath(udev_device), NULL) == 0)
                        return -EINVAL;
                iov[0].iov_base = header;
-               iov[0].iov_len = hlen;
+               iov[0].iov_len = (s - header)+1;
 
                /* add properties list */
                iov[1].iov_base = (char *)buf;
@@ -667,7 +700,7 @@ int udev_monitor_send_device(struct udev_monitor *udev_monitor, struct udev_devi
 
                /* add versioned header */
                memset(&nlh, 0x00, sizeof(struct udev_monitor_netlink_header));
-               util_strlcpy(nlh.version, "udev-" VERSION, sizeof(nlh.version));
+               util_strscpy(nlh.version, sizeof(nlh.version), "udev-" VERSION);
                nlh.magic = htonl(UDEV_MONITOR_MAGIC);
                val = udev_device_get_subsystem(udev_device);
                nlh.filter_subsystem = htonl(util_string_hash32(val));
@@ -686,8 +719,16 @@ int udev_monitor_send_device(struct udev_monitor *udev_monitor, struct udev_devi
                memset(&smsg, 0x00, sizeof(struct msghdr));
                smsg.msg_iov = iov;
                smsg.msg_iovlen = 2;
-               /* no destination besides the muticast group, we will always get ECONNREFUSED */
-               smsg.msg_name = &udev_monitor->snl_peer;
+               /*
+                * Use custom address for target, or the default one.
+                *
+                * If we send to a muticast group, we will get
+                * ECONNREFUSED, which is expected.
+                */
+               if (destination != NULL)
+                       smsg.msg_name = &destination->snl;
+               else
+                       smsg.msg_name = &udev_monitor->snl_destination;
                smsg.msg_namelen = sizeof(struct sockaddr_nl);
        } else {
                return -1;