Refactor tamper_with_syscall_exiting()

[strace] / io.c

diff --git a/io.c b/io.c
index 28fe74ee94bf06ed5bc546b2fdc48973959015d2..cd803057b32131931eab75decbcb7ae9809f2a61 100644 (file)
--- a/io.c
+++ b/io.c
@@ -3,6 +3,7 @@
  * Copyright (c) 1993 Branko Lankester <branko@hacktic.nl>
  * Copyright (c) 1993, 1994, 1995, 1996 Rick Sladkey <jrs@world.std.com>
  * Copyright (c) 1996-1999 Wichert Akkerman <wichert@cistron.nl>
+ * Copyright (c) 1999-2018 The strace developers.
  * All rights reserved.
  *
  * Redistribution and use in source and binary forms, with or without
@@ -42,7 +43,7 @@ SYS_FUNC(read)
                        printaddr(tcp->u_arg[1]);
                else
                        printstrn(tcp, tcp->u_arg[1], tcp->u_rval);
-               tprintf(", %lu", tcp->u_arg[2]);
+               tprintf(", %" PRI_klu, tcp->u_arg[2]);
        }
        return 0;
 }
@@ -52,24 +53,24 @@ SYS_FUNC(write)
        printfd(tcp, tcp->u_arg[0]);
        tprints(", ");
        printstrn(tcp, tcp->u_arg[1], tcp->u_arg[2]);
-       tprintf(", %lu", tcp->u_arg[2]);
+       tprintf(", %" PRI_klu, tcp->u_arg[2]);
 
        return RVAL_DECODED;
 }
 
 struct print_iovec_config {
        enum iov_decode decode_iov;
-       kernel_ureg_t data_size;
+       kernel_ulong_t data_size;
 };
 
 static bool
 print_iovec(struct tcb *tcp, void *elem_buf, size_t elem_size, void *data)
 {
-       const kernel_ureg_t *iov;
-       kernel_ureg_t iov_buf[2], len;
+       const kernel_ulong_t *iov;
+       kernel_ulong_t iov_buf[2], len;
        struct print_iovec_config *c = data;
 
-        if (elem_size < sizeof(iov_buf)) {
+       if (elem_size < sizeof(iov_buf)) {
                iov_buf[0] = ((unsigned int *) elem_buf)[0];
                iov_buf[1] = ((unsigned int *) elem_buf)[1];
                iov = iov_buf;
@@ -85,23 +86,24 @@ print_iovec(struct tcb *tcp, void *elem_buf, size_t elem_size, void *data)
                case IOV_DECODE_STR:
                        if (len > c->data_size)
                                len = c->data_size;
-                       if (c->data_size != (kernel_ureg_t) -1)
+                       if (c->data_size != (kernel_ulong_t) -1)
                                c->data_size -= len;
                        printstrn(tcp, iov[0], len);
                        break;
                case IOV_DECODE_NETLINK:
                        if (len > c->data_size)
                                len = c->data_size;
-                       if (c->data_size != (kernel_ureg_t) -1)
+                       if (c->data_size != (kernel_ulong_t) -1)
                                c->data_size -= len;
-                       decode_netlink(tcp, iov[0], iov[1]);
+                       /* assume that the descriptor is 1st syscall argument */
+                       decode_netlink(tcp, tcp->u_arg[0], iov[0], len);
                        break;
                default:
                        printaddr(iov[0]);
                        break;
        }
 
-       tprintf(", iov_len=%lu}", iov[1]);
+       tprintf(", iov_len=%" PRI_klu "}", iov[1]);
 
        return true;
 }
@@ -111,16 +113,17 @@ print_iovec(struct tcb *tcp, void *elem_buf, size_t elem_size, void *data)
  * Example: recvmsg returing a short read.
  */
 void
-tprint_iov_upto(struct tcb *const tcp, const kernel_ureg_t len,
-               const kernel_ureg_t addr, const enum iov_decode decode_iov,
-               const kernel_ureg_t data_size)
+tprint_iov_upto(struct tcb *const tcp, const kernel_ulong_t len,
+               const kernel_ulong_t addr, const enum iov_decode decode_iov,
+               const kernel_ulong_t data_size)
 {
-       kernel_ureg_t iov[2];
-       struct print_iovec_config config =
-               { .decode_iov = decode_iov, .data_size = data_size };
+       kernel_ulong_t iov[2];
+       struct print_iovec_config config = {
+               .decode_iov = decode_iov, .data_size = data_size
+       };
 
        print_array(tcp, addr, len, iov, current_wordsize * 2,
-                   umoven_or_printaddr_ignore_syserror, print_iovec, &config);
+                   tfetch_mem_ignore_syserror, print_iovec, &config);
 }
 
 SYS_FUNC(readv)
@@ -132,7 +135,7 @@ SYS_FUNC(readv)
                tprint_iov_upto(tcp, tcp->u_arg[2], tcp->u_arg[1],
                                syserror(tcp) ? IOV_DECODE_ADDR :
                                IOV_DECODE_STR, tcp->u_rval);
-               tprintf(", %lu", tcp->u_arg[2]);
+               tprintf(", %" PRI_klu, tcp->u_arg[2]);
        }
        return 0;
 }
@@ -142,7 +145,7 @@ SYS_FUNC(writev)
        printfd(tcp, tcp->u_arg[0]);
        tprints(", ");
        tprint_iov(tcp, tcp->u_arg[2], tcp->u_arg[1], IOV_DECODE_STR);
-       tprintf(", %lu", tcp->u_arg[2]);
+       tprintf(", %" PRI_klu, tcp->u_arg[2]);
 
        return RVAL_DECODED;
 }
@@ -157,7 +160,7 @@ SYS_FUNC(pread)
                        printaddr(tcp->u_arg[1]);
                else
                        printstrn(tcp, tcp->u_arg[1], tcp->u_rval);
-               tprintf(", %lu, ", tcp->u_arg[2]);
+               tprintf(", %" PRI_klu ", ", tcp->u_arg[2]);
                printllval(tcp, "%lld", 3);
        }
        return 0;
@@ -168,7 +171,7 @@ SYS_FUNC(pwrite)
        printfd(tcp, tcp->u_arg[0]);
        tprints(", ");
        printstrn(tcp, tcp->u_arg[1], tcp->u_arg[2]);
-       tprintf(", %lu, ", tcp->u_arg[2]);
+       tprintf(", %" PRI_klu ", ", tcp->u_arg[2]);
        printllval(tcp, "%lld", 3);
 
        return RVAL_DECODED;
@@ -177,33 +180,20 @@ SYS_FUNC(pwrite)
 static void
 print_lld_from_low_high_val(struct tcb *tcp, int arg)
 {
-#if SIZEOF_LONG > 4 && SIZEOF_LONG == SIZEOF_LONG_LONG
+#if SIZEOF_KERNEL_LONG_T > 4
 # ifndef current_klongsize
-       if (current_klongsize < SIZEOF_LONG) {
-               tprintf("%ld", (tcp->u_arg[arg + 1] << current_wordsize * 8)
+       if (current_klongsize < SIZEOF_KERNEL_LONG_T) {
+               tprintf("%" PRI_kld, (tcp->u_arg[arg + 1] << 32)
                               | tcp->u_arg[arg]);
        } else
 # endif /* !current_klongsize */
        {
-               tprintf("%ld", tcp->u_arg[arg]);
+               tprintf("%" PRI_kld, tcp->u_arg[arg]);
        }
-#elif SIZEOF_LONG > 4
-# error Unsupported configuration: SIZEOF_LONG > 4 && SIZEOF_LONG_LONG > SIZEOF_LONG
-#elif HAVE_STRUCT_TCB_EXT_ARG
-# ifndef current_klongsize
-       if (current_klongsize < SIZEOF_LONG_LONG) {
-               tprintf("%lld",
-                       (zero_extend_signed_to_ull(tcp->u_arg[arg + 1]) << sizeof(long) * 8)
-                       | zero_extend_signed_to_ull(tcp->u_arg[arg]));
-       } else
-# endif /* !current_klongsize */
-       {
-               tprintf("%lld", tcp->ext_arg[arg]);
-       }
-#else /* SIZEOF_LONG_LONG > SIZEOF_LONG && !HAVE_STRUCT_TCB_EXT_ARG */
+#else /* SIZEOF_KERNEL_LONG_T == 4 */
        tprintf("%lld",
-               (zero_extend_signed_to_ull(tcp->u_arg[arg + 1]) << sizeof(long) * 8)
-               | zero_extend_signed_to_ull(tcp->u_arg[arg]));
+                 ((long long) tcp->u_arg[arg + 1] << 32)
+               | ((long long) tcp->u_arg[arg]));
 #endif
 }
 
@@ -216,12 +206,13 @@ do_preadv(struct tcb *tcp, const int flags_arg)
                printfd(tcp, tcp->u_arg[0]);
                tprints(", ");
        } else {
-               unsigned long len = widen_to_ulong(tcp->u_arg[2]);
+               kernel_ulong_t len =
+                       truncate_kulong_to_current_wordsize(tcp->u_arg[2]);
 
                tprint_iov_upto(tcp, len, tcp->u_arg[1],
                                syserror(tcp) ? IOV_DECODE_ADDR :
                                IOV_DECODE_STR, tcp->u_rval);
-               tprintf(", %lu, ", len);
+               tprintf(", %" PRI_klu ", ", len);
                print_lld_from_low_high_val(tcp, 3);
                if (flags_arg >= 0) {
                        tprints(", ");
@@ -236,20 +227,16 @@ SYS_FUNC(preadv)
        return do_preadv(tcp, -1);
 }
 
-SYS_FUNC(preadv2)
-{
-       return do_preadv(tcp, 5);
-}
-
 static int
 do_pwritev(struct tcb *tcp, const int flags_arg)
 {
-       unsigned long len = widen_to_ulong(tcp->u_arg[2]);
+       kernel_ulong_t len =
+               truncate_kulong_to_current_wordsize(tcp->u_arg[2]);
 
        printfd(tcp, tcp->u_arg[0]);
        tprints(", ");
        tprint_iov(tcp, len, tcp->u_arg[1], IOV_DECODE_STR);
-       tprintf(", %lu, ", len);
+       tprintf(", %" PRI_klu ", ", len);
        print_lld_from_low_high_val(tcp, 3);
        if (flags_arg >= 0) {
                tprints(", ");
@@ -264,9 +251,29 @@ SYS_FUNC(pwritev)
        return do_pwritev(tcp, -1);
 }
 
+/*
+ * x32 is the only architecture where preadv2 takes 5 arguments
+ * instead of 6, see preadv64v2 in kernel sources.
+ * Likewise, x32 is the only architecture where pwritev2 takes 5 arguments
+ * instead of 6, see pwritev64v2 in kernel sources.
+ */
+
+#if defined X86_64
+# define PREADV2_PWRITEV2_FLAGS_ARG_NO (current_personality == 2 ? 4 : 5)
+#elif defined X32
+# define PREADV2_PWRITEV2_FLAGS_ARG_NO (current_personality == 0 ? 4 : 5)
+#else
+# define PREADV2_PWRITEV2_FLAGS_ARG_NO 5
+#endif
+
+SYS_FUNC(preadv2)
+{
+       return do_preadv(tcp, PREADV2_PWRITEV2_FLAGS_ARG_NO);
+}
+
 SYS_FUNC(pwritev2)
 {
-       return do_pwritev(tcp, 5);
+       return do_pwritev(tcp, PREADV2_PWRITEV2_FLAGS_ARG_NO);
 }
 
 #include "xlat/splice_flags.h"
@@ -280,7 +287,7 @@ SYS_FUNC(tee)
        printfd(tcp, tcp->u_arg[1]);
        tprints(", ");
        /* size_t len */
-       tprintf("%lu, ", tcp->u_arg[2]);
+       tprintf("%" PRI_klu ", ", tcp->u_arg[2]);
        /* unsigned int flags */
        printflags(splice_flags, tcp->u_arg[3], "SPLICE_F_???");
 
@@ -302,7 +309,7 @@ SYS_FUNC(splice)
        printnum_int64(tcp, tcp->u_arg[3], "%" PRId64);
        tprints(", ");
        /* size_t len */
-       tprintf("%lu, ", tcp->u_arg[4]);
+       tprintf("%" PRI_klu ", ", tcp->u_arg[4]);
        /* unsigned int flags */
        printflags(splice_flags, tcp->u_arg[5], "SPLICE_F_???");
 
@@ -316,7 +323,7 @@ SYS_FUNC(vmsplice)
        tprints(", ");
        /* const struct iovec *iov, unsigned long nr_segs */
        tprint_iov(tcp, tcp->u_arg[2], tcp->u_arg[1], IOV_DECODE_STR);
-       tprintf(", %lu, ", tcp->u_arg[2]);
+       tprintf(", %" PRI_klu ", ", tcp->u_arg[2]);
        /* unsigned int flags */
        printflags(splice_flags, tcp->u_arg[3], "SPLICE_F_???");