git.meshlink.io Git - catta/blob - avahi-core/netlink.c

   1 /* $Id$ */
   2
   3 /***
   4   This file is part of avahi.
   5
   6   avahi is free software; you can redistribute it and/or modify it
   7   under the terms of the GNU Lesser General Public License as
   8   published by the Free Software Foundation; either version 2.1 of the
   9   License, or (at your option) any later version.
  10
  11   avahi is distributed in the hope that it will be useful, but WITHOUT
  12   ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
  13   or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General
  14   Public License for more details.
  15
  16   You should have received a copy of the GNU Lesser General Public
  17   License along with avahi; if not, write to the Free Software
  18   Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
  19   USA.
  20 ***/
  21
  22 #ifdef HAVE_CONFIG_H
  23 #include <config.h>
  24 #endif
  25
  26 #include <unistd.h>
  27 #include <errno.h>
  28 #include <string.h>
  29 #include <sys/ioctl.h>
  30 #include <assert.h>
  31
  32 #include <avahi-common/malloc.h>
  33 #include "netlink.h"
  34 #include "log.h"
  35
  36 struct AvahiNetlink {
  37     int fd;
  38     unsigned seq;
  39     AvahiNetlinkCallback callback;
  40     void* userdata;
  41     uint8_t* buffer;
  42     size_t buffer_length;
  43
  44     const AvahiPoll *poll_api;
  45     AvahiWatch *watch;
  46 };
  47
  48 int avahi_netlink_work(AvahiNetlink *nl, int block) {
  49     ssize_t bytes;
  50     struct msghdr smsg;
  51     struct cmsghdr *cmsg;
  52     struct ucred *cred;
  53     struct iovec iov;
  54     struct nlmsghdr *p;
  55     char cred_msg[CMSG_SPACE(sizeof(struct ucred))];
  56
  57     assert(nl);
  58
  59     iov.iov_base = nl->buffer;
  60     iov.iov_len = nl->buffer_length;
  61
  62     smsg.msg_name = (void*) NULL;
  63     smsg.msg_namelen = 0;
  64     smsg.msg_iov = &iov;
  65     smsg.msg_iovlen = 1;
  66     smsg.msg_control = cred_msg;
  67     smsg.msg_controllen = sizeof(cred_msg);
  68     smsg.msg_flags = (block ? 0 : MSG_DONTWAIT);
  69
  70     if ((bytes = recvmsg(nl->fd, &smsg, 0)) < 0) {
  71         if (errno == EAGAIN || errno == EINTR)
  72             return 0;
  73
  74         avahi_log_error(__FILE__": recvmsg() failed: %s", strerror(errno));
  75         return -1;
  76     }
  77
  78     cmsg = CMSG_FIRSTHDR(&smsg);
  79     cred = (struct ucred *) CMSG_DATA (cmsg);
  80
  81     if (cmsg == NULL || cmsg->cmsg_type != SCM_CREDENTIALS) {
  82         avahi_log_error("No sender credentials received, ignoring data.");
  83         return -1;
  84     }
  85
  86     if (cred->uid != 0) {
  87         avahi_log_warn("Netlink message received from cred->uid != 0 (%d)", cred->uid);
  88         return -1;
  89     }
  90
  91     p = (struct nlmsghdr *) nl->buffer;
  92
  93     assert(nl->callback);
  94
  95     for (; bytes > 0; p = NLMSG_NEXT(p, bytes)) {
  96         if (!NLMSG_OK(p, (size_t) bytes)) {
  97             avahi_log_warn(__FILE__": packet truncated");
  98             return -1;
  99         }
 100
 101         nl->callback(nl, p, nl->userdata);
 102     }
 103
 104     return 0;
 105 }
 106
 107 static void socket_event(AvahiWatch *w, int fd, AVAHI_GCC_UNUSED AvahiWatchEvent event, void *userdata) {
 108     AvahiNetlink *nl = userdata;
 109
 110     assert(w);
 111     assert(nl);
 112     assert(fd == nl->fd);
 113
 114     avahi_netlink_work(nl, 0);
 115 }
 116
 117 AvahiNetlink *avahi_netlink_new(const AvahiPoll *poll_api, uint32_t groups, void (*cb) (AvahiNetlink *nl, struct nlmsghdr *n, void* userdata), void* userdata) {
 118     int fd = -1;
 119     const int on = 1;
 120     struct sockaddr_nl addr;
 121     AvahiNetlink *nl = NULL;
 122
 123     assert(poll_api);
 124     assert(cb);
 125
 126     if ((fd = socket(PF_NETLINK, SOCK_DGRAM, NETLINK_ROUTE)) < 0) {
 127         avahi_log_error(__FILE__": socket(PF_NETLINK): %s", strerror(errno));
 128         return NULL;
 129     }
 130
 131     memset(&addr, 0, sizeof(addr));
 132     addr.nl_family = AF_NETLINK;
 133     addr.nl_groups = groups;
 134     addr.nl_pid = getpid();
 135
 136     if (bind(fd, (struct sockaddr *) &addr, sizeof(addr)) < 0) {
 137         avahi_log_error(__FILE__": bind(): %s", strerror(errno));
 138         goto fail;
 139     }
 140
 141     if (setsockopt(fd, SOL_SOCKET, SO_PASSCRED, &on, sizeof(on)) < 0) {
 142         avahi_log_error(__FILE__": bind(): %s", strerror(errno));
 143         goto fail;
 144     }
 145
 146     if (!(nl = avahi_new(AvahiNetlink, 1))) {
 147         avahi_log_error(__FILE__": avahi_new() failed.");
 148         goto fail;
 149     }
 150
 151     nl->poll_api = poll_api;
 152     nl->fd = fd;
 153     nl->seq = 0;
 154     nl->callback = cb;
 155     nl->userdata = userdata;
 156
 157     if (!(nl->buffer = avahi_new(uint8_t, nl->buffer_length = 64*1024))) {
 158         avahi_log_error(__FILE__": avahi_new() failed.");
 159         goto fail;
 160     }
 161
 162     if (!(nl->watch = poll_api->watch_new(poll_api, fd, AVAHI_WATCH_IN, socket_event, nl))) {
 163         avahi_log_error(__FILE__": Failed to create watch.");
 164         goto fail;
 165     }
 166
 167     return nl;
 168
 169 fail:
 170
 171     if (fd >= 0)
 172         close(fd);
 173
 174     if (nl) {
 175         avahi_free(nl->buffer);
 176         avahi_free(nl);
 177     }
 178
 179     return NULL;
 180 }
 181
 182 void avahi_netlink_free(AvahiNetlink *nl) {
 183     assert(nl);
 184
 185     if (nl->watch)
 186         nl->poll_api->watch_free(nl->watch);
 187
 188     if (nl->fd >= 0)
 189         close(nl->fd);
 190
 191     avahi_free(nl->buffer);
 192     avahi_free(nl);
 193 }
 194
 195 int avahi_netlink_send(AvahiNetlink *nl, struct nlmsghdr *m, unsigned *ret_seq) {
 196     assert(nl);
 197     assert(m);
 198
 199     m->nlmsg_seq = nl->seq++;
 200     m->nlmsg_flags |= NLM_F_ACK;
 201
 202     if (send(nl->fd, m, m->nlmsg_len, 0) < 0) {
 203         avahi_log_error(__FILE__": send(): %s", strerror(errno));
 204         return -1;
 205     }
 206
 207     if (ret_seq)
 208         *ret_seq = m->nlmsg_seq;
 209
 210     return 0;
 211 }