]> granicus.if.org Git - strace/blobdiff - msghdr.c
Simplify print_lld_from_low_high_val ifdefery
[strace] / msghdr.c
index c4d8aef38ce7311d719dab54ce95a79a0ad3b190..f9264a60013e2dee6f117d6b673d0f3070809cb4 100644 (file)
--- a/msghdr.c
+++ b/msghdr.c
@@ -31,6 +31,7 @@
 
 #include "defs.h"
 #include "msghdr.h"
+#include <limits.h>
 #include <arpa/inet.h>
 #include <netinet/in.h>
 
@@ -55,17 +56,22 @@ typedef union {
 } union_cmsghdr;
 
 static void
-print_scm_rights(struct tcb *tcp, const void *cmsg_data, const size_t data_len)
+print_scm_rights(struct tcb *tcp, const void *cmsg_data,
+                const unsigned int data_len)
 {
        const int *fds = cmsg_data;
-       const size_t nfds = data_len / sizeof(*fds);
-       size_t i;
+       const unsigned int nfds = data_len / sizeof(*fds);
+       unsigned int i;
 
        tprints("[");
 
        for (i = 0; i < nfds; ++i) {
                if (i)
                        tprints(", ");
+               if (abbrev(tcp) && i >= max_strlen) {
+                       tprints("...");
+                       break;
+               }
                printfd(tcp, fds[i]);
        }
 
@@ -73,7 +79,8 @@ print_scm_rights(struct tcb *tcp, const void *cmsg_data, const size_t data_len)
 }
 
 static void
-print_scm_creds(struct tcb *tcp, const void *cmsg_data, const size_t data_len)
+print_scm_creds(struct tcb *tcp, const void *cmsg_data,
+               const unsigned int data_len)
 {
        const struct ucred *uc = cmsg_data;
 
@@ -83,14 +90,14 @@ print_scm_creds(struct tcb *tcp, const void *cmsg_data, const size_t data_len)
 
 static void
 print_scm_security(struct tcb *tcp, const void *cmsg_data,
-                  const size_t data_len)
+                  const unsigned int data_len)
 {
        print_quoted_string(cmsg_data, data_len, 0);
 }
 
 static void
 print_cmsg_ip_pktinfo(struct tcb *tcp, const void *cmsg_data,
-                     const size_t data_len)
+                     const unsigned int data_len)
 {
        const struct in_pktinfo *info = cmsg_data;
 
@@ -103,7 +110,8 @@ print_cmsg_ip_pktinfo(struct tcb *tcp, const void *cmsg_data,
 }
 
 static void
-print_cmsg_uint(struct tcb *tcp, const void *cmsg_data, const size_t data_len)
+print_cmsg_uint(struct tcb *tcp, const void *cmsg_data,
+               const unsigned int data_len)
 {
        const unsigned int *p = cmsg_data;
 
@@ -112,7 +120,7 @@ print_cmsg_uint(struct tcb *tcp, const void *cmsg_data, const size_t data_len)
 
 static void
 print_cmsg_uint8_t(struct tcb *tcp, const void *cmsg_data,
-                  const size_t data_len)
+                  const unsigned int data_len)
 {
        const uint8_t *p = cmsg_data;
 
@@ -121,15 +129,19 @@ print_cmsg_uint8_t(struct tcb *tcp, const void *cmsg_data,
 
 static void
 print_cmsg_ip_opts(struct tcb *tcp, const void *cmsg_data,
-                  const size_t data_len)
+                  const unsigned int data_len)
 {
        const unsigned char *opts = cmsg_data;
-       size_t i;
+       unsigned int i;
 
        tprints("[");
        for (i = 0; i < data_len; ++i) {
                if (i)
                        tprints(", ");
+               if (abbrev(tcp) && i >= max_strlen) {
+                       tprints("...");
+                       break;
+               }
                tprintf("0x%02x", opts[i]);
        }
        tprints("]");
@@ -148,7 +160,7 @@ struct sock_ee {
 
 static void
 print_cmsg_ip_recverr(struct tcb *tcp, const void *cmsg_data,
-                     const size_t data_len)
+                     const unsigned int data_len)
 {
        const struct sock_ee *const err = cmsg_data;
 
@@ -162,20 +174,20 @@ print_cmsg_ip_recverr(struct tcb *tcp, const void *cmsg_data,
 
 static void
 print_cmsg_ip_origdstaddr(struct tcb *tcp, const void *cmsg_data,
-                         const size_t data_len)
+                         const unsigned int data_len)
 {
-       const int addr_len =
+       const unsigned int addr_len =
                data_len > sizeof(struct sockaddr_storage)
                ? sizeof(struct sockaddr_storage) : data_len;
 
        print_sockaddr(tcp, cmsg_data, addr_len);
 }
 
-typedef void (* const cmsg_printer)(struct tcb *, const void *, size_t);
+typedef void (* const cmsg_printer)(struct tcb *, const void *, unsigned int);
 
 static const struct {
        const cmsg_printer printer;
-       const size_t min_len;
+       const unsigned int min_len;
 } cmsg_socket_printers[] = {
        [SCM_RIGHTS] = { print_scm_rights, sizeof(int) },
        [SCM_CREDENTIALS] = { print_scm_creds, sizeof(struct ucred) },
@@ -194,7 +206,7 @@ static const struct {
 
 static void
 print_cmsg_type_data(struct tcb *tcp, const int cmsg_level, const int cmsg_type,
-                    const void *cmsg_data, const size_t data_len)
+                    const void *cmsg_data, const unsigned int data_len)
 {
        const unsigned int utype = cmsg_type;
        switch (cmsg_level) {
@@ -221,20 +233,41 @@ print_cmsg_type_data(struct tcb *tcp, const int cmsg_level, const int cmsg_type,
        }
 }
 
+static unsigned int
+get_optmem_max(void)
+{
+       static int optmem_max;
+
+       if (!optmem_max) {
+               if (read_int_from_file("/proc/sys/net/core/optmem_max",
+                                      &optmem_max) || optmem_max <= 0) {
+                       optmem_max = sizeof(long long) * (2 * IOV_MAX + 512);
+               } else {
+                       optmem_max = (optmem_max + sizeof(long long) - 1)
+                                    & ~(sizeof(long long) - 1);
+               }
+       }
+
+       return optmem_max;
+}
+
 static void
-decode_msg_control(struct tcb *tcp, unsigned long addr, const size_t control_len)
+decode_msg_control(struct tcb *const tcp, const kernel_ulong_t addr,
+                  const kernel_ulong_t in_control_len)
 {
-       if (!control_len)
+       if (!in_control_len)
                return;
        tprints(", msg_control=");
 
-       const size_t cmsg_size =
+       const unsigned int cmsg_size =
 #if SUPPORTED_PERSONALITIES > 1 && SIZEOF_LONG > 4
                (current_wordsize < sizeof(long)) ? sizeof(struct cmsghdr32) :
 #endif
                        sizeof(struct cmsghdr);
 
-       size_t buf_len = control_len;
+       unsigned int control_len = in_control_len > get_optmem_max()
+                                  ? get_optmem_max() : in_control_len;
+       unsigned int buf_len = control_len;
        char *buf = buf_len < cmsg_size ? NULL : malloc(buf_len);
        if (!buf || umoven(tcp, addr, buf_len, buf) < 0) {
                printaddr(addr);
@@ -246,7 +279,7 @@ decode_msg_control(struct tcb *tcp, unsigned long addr, const size_t control_len
 
        tprints("[");
        while (buf_len >= cmsg_size) {
-               const size_t cmsg_len =
+               const kernel_ulong_t cmsg_len =
 #if SUPPORTED_PERSONALITIES > 1 && SIZEOF_LONG > 4
                        (current_wordsize < sizeof(long)) ? u.cmsg32->cmsg_len :
 #endif
@@ -264,11 +297,11 @@ decode_msg_control(struct tcb *tcp, unsigned long addr, const size_t control_len
 
                if (u.ptr != buf)
                        tprints(", ");
-               tprintf("{cmsg_len=%lu, cmsg_level=", (unsigned long) cmsg_len);
+               tprintf("{cmsg_len=%" PRI_klu ", cmsg_level=", cmsg_len);
                printxval(socketlayers, cmsg_level, "SOL_???");
                tprints(", cmsg_type=");
 
-               size_t len = cmsg_len > buf_len ? buf_len : cmsg_len;
+               kernel_ulong_t len = cmsg_len > buf_len ? buf_len : cmsg_len;
 
                print_cmsg_type_data(tcp, cmsg_level, cmsg_type,
                                     (const void *) (u.ptr + cmsg_size),
@@ -280,7 +313,7 @@ decode_msg_control(struct tcb *tcp, unsigned long addr, const size_t control_len
                        break;
                }
                len = (cmsg_len + current_wordsize - 1) &
-                       (size_t) ~(current_wordsize - 1);
+                       ~((kernel_ulong_t) current_wordsize - 1);
                if (len >= buf_len) {
                        buf_len = 0;
                        break;
@@ -291,104 +324,123 @@ decode_msg_control(struct tcb *tcp, unsigned long addr, const size_t control_len
        if (buf_len) {
                tprints(", ");
                printaddr(addr + (control_len - buf_len));
+       } else if (control_len < in_control_len) {
+               tprints(", ...");
        }
        tprints("]");
        free(buf);
 }
 
-static void
-print_msghdr(struct tcb *tcp, struct msghdr *msg, unsigned long data_size)
+void
+print_struct_msghdr(struct tcb *tcp, const struct msghdr *msg,
+                   const int *const p_user_msg_namelen,
+                   const kernel_ulong_t data_size)
 {
+       const int msg_namelen =
+               p_user_msg_namelen && (int) msg->msg_namelen > *p_user_msg_namelen
+               ? *p_user_msg_namelen : (int) msg->msg_namelen;
+
        tprints("{msg_name=");
-       decode_sockaddr(tcp, (long)msg->msg_name, msg->msg_namelen);
-       tprintf(", msg_namelen=%d", msg->msg_namelen);
+       const int family =
+               decode_sockaddr(tcp, ptr_to_kulong(msg->msg_name), msg_namelen);
+       const enum iov_decode decode =
+               (family == AF_NETLINK) ? IOV_DECODE_NETLINK : IOV_DECODE_STR;
+
+       tprints(", msg_namelen=");
+       if (p_user_msg_namelen && *p_user_msg_namelen != (int) msg->msg_namelen)
+               tprintf("%d->", *p_user_msg_namelen);
+       tprintf("%d", msg->msg_namelen);
 
        tprints(", msg_iov=");
-       tprint_iov_upto(tcp, (unsigned long) msg->msg_iovlen,
-                       (unsigned long) msg->msg_iov, IOV_DECODE_STR, data_size);
-       tprintf(", msg_iovlen=%lu", (unsigned long) msg->msg_iovlen);
 
-       decode_msg_control(tcp, (unsigned long) msg->msg_control,
+       tprint_iov_upto(tcp, msg->msg_iovlen,
+                       ptr_to_kulong(msg->msg_iov), decode, data_size);
+       tprintf(", msg_iovlen=%" PRI_klu, (kernel_ulong_t) msg->msg_iovlen);
+
+       decode_msg_control(tcp, ptr_to_kulong(msg->msg_control),
                           msg->msg_controllen);
-       tprintf(", msg_controllen=%lu", (unsigned long) msg->msg_controllen);
+       tprintf(", msg_controllen=%" PRI_klu, (kernel_ulong_t) msg->msg_controllen);
 
        tprints(", msg_flags=");
        printflags(msg_flags, msg->msg_flags, "MSG_???");
        tprints("}");
 }
 
-void
-decode_msghdr(struct tcb *tcp, long addr, unsigned long data_size)
+static bool
+fetch_msghdr_namelen(struct tcb *const tcp, const kernel_ulong_t addr,
+                    int *const p_msg_namelen)
+{
+       struct msghdr msg;
+
+       if (addr && verbose(tcp) && fetch_struct_msghdr(tcp, addr, &msg)) {
+               *p_msg_namelen = msg.msg_namelen;
+               return true;
+       } else {
+               return false;
+       }
+}
+
+static void
+decode_msghdr(struct tcb *const tcp, const int *const p_user_msg_namelen,
+             const kernel_ulong_t addr, const kernel_ulong_t data_size)
 {
        struct msghdr msg;
 
        if (addr && verbose(tcp) && fetch_struct_msghdr(tcp, addr, &msg))
-               print_msghdr(tcp, &msg, data_size);
+               print_struct_msghdr(tcp, &msg, p_user_msg_namelen, data_size);
        else
                printaddr(addr);
 }
 
 void
-dumpiov_in_msghdr(struct tcb *tcp, long addr, unsigned long data_size)
+dumpiov_in_msghdr(struct tcb *const tcp, const kernel_ulong_t addr,
+                 const kernel_ulong_t data_size)
 {
        struct msghdr msg;
 
-       if (fetch_struct_msghdr(tcp, addr, &msg))
-               dumpiov_upto(tcp, msg.msg_iovlen, (long)msg.msg_iov, data_size);
+       if (fetch_struct_msghdr(tcp, addr, &msg)) {
+               dumpiov_upto(tcp, msg.msg_iovlen,
+                            ptr_to_kulong(msg.msg_iov), data_size);
+       }
 }
 
-static int
-decode_mmsghdr(struct tcb *tcp, long addr, bool use_msg_len)
+SYS_FUNC(sendmsg)
 {
-       struct mmsghdr mmsg;
-       int fetched = fetch_struct_mmsghdr(tcp, addr, &mmsg);
-
-       if (fetched) {
-               tprints("{msg_hdr=");
-               print_msghdr(tcp, &mmsg.msg_hdr, use_msg_len ? mmsg.msg_len : -1UL);
-               tprintf(", msg_len=%u}", mmsg.msg_len);
-       } else {
-               printaddr(addr);
-       }
-
-       return fetched;
+       printfd(tcp, tcp->u_arg[0]);
+       tprints(", ");
+       decode_msghdr(tcp, 0, tcp->u_arg[1], -1);
+       /* flags */
+       tprints(", ");
+       printflags(msg_flags, tcp->u_arg[2], "MSG_???");
+
+       return RVAL_DECODED;
 }
 
-void
-decode_mmsgvec(struct tcb *tcp, unsigned long addr, unsigned int len,
-              bool use_msg_len)
+SYS_FUNC(recvmsg)
 {
-       if (syserror(tcp)) {
-               printaddr(addr);
-       } else {
-               unsigned int i, fetched;
-
-               tprints("[");
-               for (i = 0; i < len; ++i, addr += fetched) {
-                       if (i)
-                               tprints(", ");
-                       fetched = decode_mmsghdr(tcp, addr, use_msg_len);
-                       if (!fetched)
-                               break;
+       int msg_namelen;
+
+       if (entering(tcp)) {
+               printfd(tcp, tcp->u_arg[0]);
+               tprints(", ");
+               if (fetch_msghdr_namelen(tcp, tcp->u_arg[1], &msg_namelen)) {
+                       set_tcb_priv_ulong(tcp, msg_namelen);
+                       return 0;
                }
-               tprints("]");
+               printaddr(tcp->u_arg[1]);
+       } else {
+               msg_namelen = get_tcb_priv_ulong(tcp);
+
+               if (syserror(tcp))
+                       tprintf("{msg_namelen=%d}", msg_namelen);
+               else
+                       decode_msghdr(tcp, &msg_namelen, tcp->u_arg[1],
+                                     tcp->u_rval);
        }
-}
 
-void
-dumpiov_in_mmsghdr(struct tcb *tcp, long addr)
-{
-       unsigned int len = tcp->u_rval;
-       unsigned int i, fetched;
-       struct mmsghdr mmsg;
+       /* flags */
+       tprints(", ");
+       printflags(msg_flags, tcp->u_arg[2], "MSG_???");
 
-       for (i = 0; i < len; ++i, addr += fetched) {
-               fetched = fetch_struct_mmsghdr(tcp, addr, &mmsg);
-               if (!fetched)
-                       break;
-               tprintf(" = %lu buffers in vector %u\n",
-                       (unsigned long)mmsg.msg_hdr.msg_iovlen, i);
-               dumpiov_upto(tcp, mmsg.msg_hdr.msg_iovlen,
-                       (long)mmsg.msg_hdr.msg_iov, mmsg.msg_len);
-       }
+       return RVAL_DECODED;
 }