Author: jhb
Date: Fri Jan  6 23:41:45 2017
New Revision: 311568
URL: https://svnweb.freebsd.org/changeset/base/311568

Log:
  Set MORETOCOME for AIO write requests on a socket.
  
  Add a MSG_MOREOTOCOME message flag. When this flag is set, sosend*
  set PRUS_MOREOTOCOME when invoking the protocol send method. The aio
  worker tasks for sending on a socket set this flag when there are
  additional write jobs waiting on the socket buffer.
  
  Reviewed by:  adrian
  MFC after:    1 month
  Sponsored by: Chelsio Communications
  Differential Revision:        https://reviews.freebsd.org/D8955

Modified:
  head/sys/kern/sys_socket.c
  head/sys/kern/uipc_socket.c
  head/sys/sys/socket.h

Modified: head/sys/kern/sys_socket.c
==============================================================================
--- head/sys/kern/sys_socket.c  Fri Jan  6 23:30:54 2017        (r311567)
+++ head/sys/kern/sys_socket.c  Fri Jan  6 23:41:45 2017        (r311568)
@@ -604,6 +604,8 @@ retry:
                if (td->td_ru.ru_msgrcv != ru_before)
                        job->msgrcv = 1;
        } else {
+               if (!TAILQ_EMPTY(&sb->sb_aiojobq))
+                       flags |= MSG_MORETOCOME;
                uio.uio_rw = UIO_WRITE;
                ru_before = td->td_ru.ru_msgsnd;
 #ifdef MAC

Modified: head/sys/kern/uipc_socket.c
==============================================================================
--- head/sys/kern/uipc_socket.c Fri Jan  6 23:30:54 2017        (r311567)
+++ head/sys/kern/uipc_socket.c Fri Jan  6 23:41:45 2017        (r311568)
@@ -1182,6 +1182,7 @@ sosend_dgram(struct socket *so, struct s
             (resid <= 0)) ?
                PRUS_EOF :
                /* If there is more to send set PRUS_MORETOCOME */
+               (flags & MSG_MORETOCOME) ||
                (resid > 0 && space > 0) ? PRUS_MORETOCOME : 0,
                top, addr, control, td);
        if (dontroute) {
@@ -1368,6 +1369,7 @@ restart:
                             (resid <= 0)) ?
                                PRUS_EOF :
                        /* If there is more to send set PRUS_MORETOCOME. */
+                           (flags & MSG_MORETOCOME) ||
                            (resid > 0 && space > 0) ? PRUS_MORETOCOME : 0,
                            top, addr, control, td);
                        if (dontroute) {

Modified: head/sys/sys/socket.h
==============================================================================
--- head/sys/sys/socket.h       Fri Jan  6 23:30:54 2017        (r311567)
+++ head/sys/sys/socket.h       Fri Jan  6 23:41:45 2017        (r311568)
@@ -435,6 +435,7 @@ struct msghdr {
 #endif
 #ifdef _KERNEL
 #define        MSG_SOCALLBCK   0x10000         /* for use by socket callbacks 
- soreceive (TCP) */
+#define        MSG_MORETOCOME  0x20000         /* additional data pending */
 #endif
 
 /*
_______________________________________________
[email protected] mailing list
https://lists.freebsd.org/mailman/listinfo/svn-src-head
To unsubscribe, send any mail to "[email protected]"

Reply via email to