84 #include <netlink-local.h>
85 #include <netlink/netlink.h>
86 #include <netlink/utils.h>
87 #include <netlink/handlers.h>
88 #include <netlink/msg.h>
89 #include <netlink/attr.h>
114 flags |= SOCK_CLOEXEC;
117 sk->s_fd = socket(AF_NETLINK, SOCK_RAW | flags, protocol);
119 err = -nl_syserr2nlerr(errno);
123 if (!(sk->s_flags & NL_SOCK_BUFSIZE_SET)) {
129 err = bind(sk->s_fd, (
struct sockaddr*) &sk->s_local,
130 sizeof(sk->s_local));
132 err = -nl_syserr2nlerr(errno);
136 addrlen =
sizeof(sk->s_local);
137 err = getsockname(sk->s_fd, (
struct sockaddr *) &sk->s_local,
140 err = -nl_syserr2nlerr(errno);
144 if (addrlen !=
sizeof(sk->s_local)) {
149 if (sk->s_local.nl_family != AF_NETLINK) {
150 err = -NLE_AF_NOSUPPORT;
154 sk->s_proto = protocol;
192 int nl_sendto(
struct nl_sock *sk,
void *buf,
size_t size)
196 ret = sendto(sk->s_fd, buf, size, 0, (
struct sockaddr *)
197 &sk->s_peer,
sizeof(sk->s_peer));
199 return -nl_syserr2nlerr(errno);
211 int nl_sendmsg(
struct nl_sock *sk,
struct nl_msg *msg,
struct msghdr *hdr)
216 nlmsg_set_src(msg, &sk->s_local);
223 ret = sendmsg(sk->s_fd, hdr, 0);
225 return -nl_syserr2nlerr(errno);
227 NL_DBG(4,
"sent %d bytes\n", ret);
241 int nl_send_iovec(
struct nl_sock *sk,
struct nl_msg *msg,
struct iovec *iov,
unsigned iovlen)
243 struct sockaddr_nl *dst;
245 struct msghdr hdr = {
246 .msg_name = (
void *) &sk->s_peer,
247 .msg_namelen =
sizeof(
struct sockaddr_nl),
249 .msg_iovlen = iovlen,
255 dst = nlmsg_get_dst(msg);
256 if (dst->nl_family == AF_NETLINK)
260 creds = nlmsg_get_creds(msg);
262 char buf[CMSG_SPACE(
sizeof(
struct ucred))];
263 struct cmsghdr *cmsg;
265 hdr.msg_control = buf;
266 hdr.msg_controllen =
sizeof(buf);
268 cmsg = CMSG_FIRSTHDR(&hdr);
269 cmsg->cmsg_level = SOL_SOCKET;
270 cmsg->cmsg_type = SCM_CREDENTIALS;
271 cmsg->cmsg_len = CMSG_LEN(
sizeof(
struct ucred));
272 memcpy(CMSG_DATA(cmsg), creds,
sizeof(
struct ucred));
287 int nl_send(
struct nl_sock *sk,
struct nl_msg *msg)
297 void nl_complete_msg(
struct nl_sock *sk,
struct nl_msg *msg)
299 struct nlmsghdr *nlh;
302 if (nlh->nlmsg_pid == 0)
303 nlh->nlmsg_pid = sk->s_local.nl_pid;
305 if (nlh->nlmsg_seq == 0)
306 nlh->nlmsg_seq = sk->s_seq_next++;
308 if (msg->nm_protocol == -1)
309 msg->nm_protocol = sk->s_proto;
311 nlh->nlmsg_flags |= NLM_F_REQUEST;
313 if (!(sk->s_flags & NL_NO_AUTO_ACK))
314 nlh->nlmsg_flags |= NLM_F_ACK;
317 void nl_auto_complete(
struct nl_sock *sk,
struct nl_msg *msg)
319 nl_complete_msg(sk, msg);
339 struct nl_cb *cb = sk->s_cb;
341 nl_complete_msg(sk, msg);
344 return cb->cb_send_ow(sk, msg);
349 int nl_send_auto_complete(
struct nl_sock *sk,
struct nl_msg *msg)
382 return wait_for_ack(sk);
416 err = nl_send_auto_complete(sk, msg);
449 int nl_recv(
struct nl_sock *sk,
struct sockaddr_nl *nla,
450 unsigned char **buf,
struct ucred **creds)
454 static int page_size = 0;
456 struct msghdr msg = {
457 .msg_name = (
void *) nla,
458 .msg_namelen =
sizeof(
struct sockaddr_nl),
465 struct cmsghdr *cmsg;
467 memset(nla, 0,
sizeof(*nla));
469 if (sk->s_flags & NL_MSG_PEEK)
473 page_size = getpagesize();
475 iov.iov_len = page_size;
476 iov.iov_base = *buf = malloc(iov.iov_len);
478 if (sk->s_flags & NL_SOCK_PASSCRED) {
479 msg.msg_controllen = CMSG_SPACE(
sizeof(
struct ucred));
480 msg.msg_control = calloc(1, msg.msg_controllen);
484 n = recvmsg(sk->s_fd, &msg, flags);
488 if (errno == EINTR) {
489 NL_DBG(3,
"recvmsg() returned EINTR, retrying\n");
491 }
else if (errno == EAGAIN) {
492 NL_DBG(3,
"recvmsg() returned EAGAIN, aborting\n");
495 free(msg.msg_control);
497 return -nl_syserr2nlerr(errno);
501 if (iov.iov_len < n ||
502 msg.msg_flags & MSG_TRUNC) {
506 iov.iov_base = *buf = realloc(*buf, iov.iov_len);
508 }
else if (msg.msg_flags & MSG_CTRUNC) {
509 msg.msg_controllen *= 2;
510 msg.msg_control = realloc(msg.msg_control, msg.msg_controllen);
512 }
else if (flags != 0) {
518 if (msg.msg_namelen !=
sizeof(
struct sockaddr_nl)) {
519 free(msg.msg_control);
524 for (cmsg = CMSG_FIRSTHDR(&msg); cmsg; cmsg = CMSG_NXTHDR(&msg, cmsg)) {
525 if (cmsg->cmsg_level == SOL_SOCKET &&
526 cmsg->cmsg_type == SCM_CREDENTIALS) {
528 *creds = calloc(1,
sizeof(
struct ucred));
529 memcpy(*creds, CMSG_DATA(cmsg),
sizeof(
struct ucred));
535 free(msg.msg_control);
539 free(msg.msg_control);
544 #define NL_CB_CALL(cb, type, msg) \
546 err = nl_cb_call(cb, type, msg); \
560 static int recvmsgs(
struct nl_sock *sk,
struct nl_cb *cb)
562 int n, err = 0, multipart = 0, interrupted = 0;
563 unsigned char *buf = NULL;
564 struct nlmsghdr *hdr;
565 struct sockaddr_nl nla = {0};
566 struct nl_msg *msg = NULL;
567 struct ucred *creds = NULL;
570 NL_DBG(3,
"Attempting to read from %p\n", sk);
572 n = cb->cb_recv_ow(sk, &nla, &buf, &creds);
574 n =
nl_recv(sk, &nla, &buf, &creds);
579 NL_DBG(3,
"recvmsgs(%p): Read %d bytes\n", sk, n);
581 hdr = (
struct nlmsghdr *) buf;
583 NL_DBG(3,
"recvmsgs(%p): Processing valid message...\n", sk);
592 nlmsg_set_proto(msg, sk->s_proto);
593 nlmsg_set_src(msg, &nla);
595 nlmsg_set_creds(msg, creds);
609 }
else if (!(sk->s_flags & NL_NO_AUTO_ACK)) {
610 if (hdr->nlmsg_seq != sk->s_seq_expect) {
614 err = -NLE_SEQ_MISMATCH;
620 if (hdr->nlmsg_type == NLMSG_DONE ||
621 hdr->nlmsg_type == NLMSG_ERROR ||
622 hdr->nlmsg_type == NLMSG_NOOP ||
623 hdr->nlmsg_type == NLMSG_OVERRUN) {
627 NL_DBG(3,
"recvmsgs(%p): Increased expected " \
628 "sequence number to %d\n",
629 sk, sk->s_seq_expect);
632 if (hdr->nlmsg_flags & NLM_F_MULTI)
635 if (hdr->nlmsg_flags & NLM_F_DUMP_INTR) {
649 if (hdr->nlmsg_flags & NLM_F_ACK) {
661 if (hdr->nlmsg_type == NLMSG_DONE) {
671 else if (hdr->nlmsg_type == NLMSG_NOOP) {
681 else if (hdr->nlmsg_type == NLMSG_OVERRUN) {
685 err = -NLE_MSG_OVERFLOW;
691 else if (hdr->nlmsg_type == NLMSG_ERROR) {
694 if (hdr->nlmsg_len <
nlmsg_size(
sizeof(*e))) {
702 err = -NLE_MSG_TRUNC;
705 }
else if (e->error) {
708 err = cb->cb_err(&nla, e,
715 err = -nl_syserr2nlerr(e->error);
719 err = -nl_syserr2nlerr(e->error);
745 goto continue_reading;
755 err = -NLE_DUMP_INTR;
777 if (cb->cb_recvmsgs_ow)
778 return cb->cb_recvmsgs_ow(sk, cb);
780 return recvmsgs(sk, cb);
795 static int ack_wait_handler(
struct nl_msg *msg,
void *arg)
827 int (*parser)(
struct nl_cache_ops *,
struct sockaddr_nl *,
834 struct pickup_param *pp = p->
pp_arg;
845 static int __pickup_answer(
struct nl_msg *msg,
void *arg)
847 struct pickup_param *pp = arg;
849 .
pp_cb = __store_answer,
853 return pp->parser(NULL, &msg->nm_src, msg->nm_nlh, &parse_arg);
867 int (*parser)(
struct nl_cache_ops *,
struct sockaddr_nl *,
873 struct pickup_param pp = {