libtirpc-recverr.patch

   1 diff --git a/src/clnt_dg.c b/src/clnt_dg.c
   2 index 151b449..0e35742 100644
   3 --- a/src/clnt_dg.c
   4 +++ b/src/clnt_dg.c
   5 @@ -55,6 +55,13 @@
   6  #include <err.h>
   7  #include "rpc_com.h"
   8
   9 +#ifdef IP_RECVERR
  10 +#include <asm/types.h>
  11 +#include <linux/errqueue.h>
  12 +#include <sys/uio.h>
  13 +#endif
  14 +
  15 +
  16  #define MAX_DEFAULT_FDS                 20000
  17
  18  static struct clnt_ops *clnt_dg_ops(void);
  19 @@ -246,6 +253,12 @@ clnt_dg_create(fd, svcaddr, program, version, sendsz, recvsz)
  20  #if 0
  21         (void)bindresvport_sa(fd, (struct sockaddr *)svcaddr->buf);
  22  #endif
  23 +#ifdef IP_RECVERR
  24 +       {
  25 +       int on = 1;
  26 +       setsockopt(fd, SOL_IP, IP_RECVERR, &on, sizeof(on));
  27 +       }
  28 +#endif
  29         ioctl(fd, FIONBIO, (char *)(void *)&one);
  30         /*
  31          * By default, closeit is always FALSE. It is users responsibility
  32 @@ -352,7 +365,7 @@ call_again:
  33         xid++;
  34         *(u_int32_t *)(void *)(cu->cu_outbuf) = htonl(xid);
  35
  36 -       if ((! XDR_PUTINT32(xdrs, &proc)) ||
  37 +       if ((! XDR_PUTINT32(xdrs, (int32_t *)&proc)) ||
  38             (! AUTH_MARSHALL(cl->cl_auth, xdrs)) ||
  39             (! (*xargs)(xdrs, argsp))) {
  40                 cu->cu_error.re_status = RPC_CANTENCODEARGS;
  41 @@ -404,6 +417,48 @@ get_reply:
  42                  }
  43                  break;
  44          }
  45 +#ifdef IP_RECVERR
  46 +      if (fd.revents & POLLERR)
  47 +       {
  48 +         struct msghdr msg;
  49 +         struct cmsghdr *cmsg;
  50 +         struct sock_extended_err *e;
  51 +         struct sockaddr_in err_addr;
  52 +         struct sockaddr_in *sin = (struct sockaddr_in *)&cu->cu_raddr;
  53 +         struct iovec iov;
  54 +         char *cbuf = (char *) alloca (outlen + 256);
  55 +         int ret;
  56 +
  57 +         iov.iov_base = cbuf + 256;
  58 +         iov.iov_len = outlen;
  59 +         msg.msg_name = (void *) &err_addr;
  60 +         msg.msg_namelen = sizeof (err_addr);
  61 +         msg.msg_iov = &iov;
  62 +         msg.msg_iovlen = 1;
  63 +         msg.msg_flags = 0;
  64 +         msg.msg_control = cbuf;
  65 +         msg.msg_controllen = 256;
  66 +         ret = recvmsg (cu->cu_fd, &msg, MSG_ERRQUEUE);
  67 +         if (ret >= 0
  68 +             && memcmp (cbuf + 256, cu->cu_outbuf, ret) == 0
  69 +             && (msg.msg_flags & MSG_ERRQUEUE)
  70 +             && ((msg.msg_namelen == 0
  71 +                  && ret >= 12)
  72 +                 || (msg.msg_namelen == sizeof (err_addr)
  73 +                     && err_addr.sin_family == AF_INET
  74 +                     && memcmp (&err_addr.sin_addr, &sin->sin_addr,
  75 +                                sizeof (err_addr.sin_addr)) == 0
  76 +                     && err_addr.sin_port == sin->sin_port)))
  77 +           for (cmsg = CMSG_FIRSTHDR (&msg); cmsg;
  78 +                cmsg = CMSG_NXTHDR (&msg, cmsg))
  79 +             if (cmsg->cmsg_level == SOL_IP && cmsg->cmsg_type == IP_RECVERR)
  80 +               {
  81 +                 e = (struct sock_extended_err *) CMSG_DATA(cmsg);
  82 +                 cu->cu_error.re_errno = e->ee_errno;
  83 +                 return (cu->cu_error.re_status = RPC_CANTRECV);
  84 +               }
  85 +       }
  86 +#endif
  87
  88         /* We have some data now */
  89         do {