Add counters for MPTP buffers and syscalls.

[swifty.git] / src / kernel / mptp.c

diff --git a/src/kernel/mptp.c b/src/kernel/mptp.c
index a979736..c46461d 100644 (file)
--- a/src/kernel/mptp.c
+++ b/src/kernel/mptp.c
@@ -341,8 +341,11 @@ static int mptp_sendmsg(struct kiocb *iocb, struct socket *sock, struct msghdr *
         if (likely(!err)) {
             log_debug("Sent %u bytes on wire\n", len);
                        ret += len;
-               } else
+                       dest->bytes = len;
+               } else {
             log_error("ip_queue_xmit failed\n");
+                       dest->bytes = -1;
+               }
     }
 
        return ret;
@@ -388,18 +391,23 @@ static int mptp_recvmsg(struct kiocb *iocb, struct socket *sock, struct msghdr *
                        log_error("skb_copy_datagram_iovec\n");
                        goto out_free;
                }
-               msg->msg_iov[i].iov_len = copied;
+               log_debug("Received %d bytes\n", copied);
 
                sock_recv_ts_and_drops(msg, sk, skb);
 
-               if (ret_addr)
+               if (ret_addr) {
                        memcpy(&ret_addr->dests[i], &mptp_addr->dests[0], sizeof(ret_addr->dests[i]));
+                       ret_addr->dests[i].bytes = copied;
+               }
 
                err = copied;
 
 out_free:
            skb_free_datagram(sk, skb);
 
+               if (i == msg->msg_iovlen - 1)
+                       break;
+
                skb = skb_recv_datagram(sk, flags, 1, &err);
                if (likely(err == -EAGAIN)) {
                        log_debug("No more skbs in the queue, returning...\n");