X-Git-Url: http://git.meshlink.io/?a=blobdiff_plain;f=avahi-core%2Fnetlink.c;h=b2a6684c60445fd93356a6ec789c47863990cbdf;hb=f8086b037dec60a82aa6890f19dbc3f73af46c98;hp=ccb4d836ef8599c7a3f973648557e712c4ca71fe;hpb=6b3876eb3740666e9f3e036d49efc7b3b3ee5b45;p=catta diff --git a/avahi-core/netlink.c b/avahi-core/netlink.c index ccb4d83..b2a6684 100644 --- a/avahi-core/netlink.c +++ b/avahi-core/netlink.c @@ -46,44 +46,37 @@ struct AvahiNetlink { }; int avahi_netlink_work(AvahiNetlink *nl, int block) { + ssize_t bytes; + struct nlmsghdr *p; + assert(nl); + + if ((bytes = recv(nl->fd, nl->buffer, nl->buffer_length, block ? 0 : MSG_DONTWAIT)) < 0) { + + if (errno == EAGAIN || errno == EINTR) + return 0; + + avahi_log_error(__FILE__": recv() failed: %s", strerror(errno)); + return -1; + } - for (;;) { - ssize_t bytes; - struct nlmsghdr *p; - - for (;;) { - if ((bytes = recv(nl->fd, nl->buffer, nl->buffer_length, block ? 0 : MSG_DONTWAIT)) < 0) { - - if (errno == EAGAIN || errno == EINTR) - return 1; - - avahi_log_error(__FILE__": recv() failed: %s", strerror(errno)); - return 0; - } - - break; + p = (struct nlmsghdr *) nl->buffer; + + assert(nl->callback); + + for (; bytes > 0; p = NLMSG_NEXT(p, bytes)) { + if (!NLMSG_OK(p, (size_t) bytes)) { + avahi_log_warn(__FILE__": packet truncated"); + return -1; } - - p = (struct nlmsghdr *) nl->buffer; - if (nl->callback) { - for (; bytes > 0; p = NLMSG_NEXT(p, bytes)) { - if (!NLMSG_OK(p, (size_t) bytes)) { - avahi_log_warn(__FILE__": packet truncated"); - return 0; - } - - nl->callback(nl, p, nl->userdata); - } - } - - if (block) - return 1; + nl->callback(nl, p, nl->userdata); } + + return 0; } -static void socket_event(AvahiWatch *w, int fd, AvahiWatchEvent event, void *userdata) { +static void socket_event(AvahiWatch *w, int fd, AVAHI_GCC_UNUSED AvahiWatchEvent event, void *userdata) { AvahiNetlink *nl = userdata; assert(w); @@ -145,9 +138,7 @@ fail: close(fd); if (nl) { - if (nl->buffer) - avahi_free(nl); - + avahi_free(nl->buffer); avahi_free(nl); }