t/zbd: avoid test case 31 failure with small devices
[fio.git] / engines / net.c
index 1df8d0635cf1b746ae7d597d1b4b4398f0019d2d..29150bb348ac8b2342f2441c1198ce21e5499fa1 100644 (file)
@@ -9,26 +9,40 @@
 #include <unistd.h>
 #include <signal.h>
 #include <errno.h>
-#include <assert.h>
 #include <netinet/in.h>
 #include <netinet/tcp.h>
 #include <arpa/inet.h>
 #include <netdb.h>
-#include <sys/poll.h>
-#include <sys/types.h>
+#include <poll.h>
 #include <sys/stat.h>
 #include <sys/socket.h>
 #include <sys/un.h>
 
+#ifdef CONFIG_VSOCK
+#include <linux/vm_sockets.h>
+#else
+struct sockaddr_vm {
+};
+#ifndef AF_VSOCK
+#define AF_VSOCK       -1
+#endif
+#endif
+
 #include "../fio.h"
+#include "../verify.h"
+#include "../optgroup.h"
 
 struct netio_data {
        int listenfd;
        int use_splice;
+       int seq_off;
        int pipes[2];
        struct sockaddr_in addr;
        struct sockaddr_in6 addr6;
        struct sockaddr_un addr_un;
+       struct sockaddr_vm addr_vm;
+       uint64_t udp_send_seq;
+       uint64_t udp_recv_seq;
 };
 
 struct netio_options {
@@ -39,6 +53,8 @@ struct netio_options {
        unsigned int pingpong;
        unsigned int nodelay;
        unsigned int ttl;
+       unsigned int window_size;
+       unsigned int mss;
        char *intfc;
 };
 
@@ -47,16 +63,24 @@ struct udp_close_msg {
        uint32_t cmd;
 };
 
+struct udp_seq {
+       uint64_t magic;
+       uint64_t seq;
+       uint64_t bs;
+};
+
 enum {
        FIO_LINK_CLOSE = 0x89,
        FIO_LINK_OPEN_CLOSE_MAGIC = 0x6c696e6b,
        FIO_LINK_OPEN = 0x98,
+       FIO_UDP_SEQ_MAGIC = 0x657375716e556563ULL,
 
        FIO_TYPE_TCP    = 1,
        FIO_TYPE_UDP    = 2,
        FIO_TYPE_UNIX   = 3,
        FIO_TYPE_TCP_V6 = 4,
        FIO_TYPE_UDP_V6 = 5,
+       FIO_TYPE_VSOCK_STREAM   = 6,
 };
 
 static int str_hostname_cb(void *data, const char *input);
@@ -114,6 +138,10 @@ static struct fio_option options[] = {
                            .oval = FIO_TYPE_UNIX,
                            .help = "UNIX domain socket",
                          },
+                         { .ival = "vsock",
+                           .oval = FIO_TYPE_VSOCK_STREAM,
+                           .help = "Virtual socket",
+                         },
                },
                .category = FIO_OPT_C_ENGINE,
                .group  = FIO_OPT_G_NETIO,
@@ -121,6 +149,7 @@ static struct fio_option options[] = {
 #ifdef CONFIG_TCP_NODELAY
        {
                .name   = "nodelay",
+               .lname  = "No Delay",
                .type   = FIO_OPT_BOOL,
                .off1   = offsetof(struct netio_options, nodelay),
                .help   = "Use TCP_NODELAY on TCP connections",
@@ -139,6 +168,7 @@ static struct fio_option options[] = {
        },
        {
                .name   = "pingpong",
+               .lname  = "Ping Pong",
                .type   = FIO_OPT_STR_SET,
                .off1   = offsetof(struct netio_options, pingpong),
                .help   = "Ping-pong IO requests",
@@ -165,6 +195,30 @@ static struct fio_option options[] = {
                .category = FIO_OPT_C_ENGINE,
                .group  = FIO_OPT_G_NETIO,
        },
+#ifdef CONFIG_NET_WINDOWSIZE
+       {
+               .name   = "window_size",
+               .lname  = "Window Size",
+               .type   = FIO_OPT_INT,
+               .off1   = offsetof(struct netio_options, window_size),
+               .minval = 0,
+               .help   = "Set socket buffer window size",
+               .category = FIO_OPT_C_ENGINE,
+               .group  = FIO_OPT_G_NETIO,
+       },
+#endif
+#ifdef CONFIG_NET_MSS
+       {
+               .name   = "mss",
+               .lname  = "Maximum segment size",
+               .type   = FIO_OPT_INT,
+               .off1   = offsetof(struct netio_options, mss),
+               .minval = 0,
+               .help   = "Set TCP maximum segment size",
+               .category = FIO_OPT_C_ENGINE,
+               .group  = FIO_OPT_G_NETIO,
+       },
+#endif
        {
                .name   = NULL,
        },
@@ -185,6 +239,70 @@ static inline int is_ipv6(struct netio_options *o)
        return o->proto == FIO_TYPE_UDP_V6 || o->proto == FIO_TYPE_TCP_V6;
 }
 
+static inline int is_vsock(struct netio_options *o)
+{
+       return o->proto == FIO_TYPE_VSOCK_STREAM;
+}
+
+static int set_window_size(struct thread_data *td, int fd)
+{
+#ifdef CONFIG_NET_WINDOWSIZE
+       struct netio_options *o = td->eo;
+       unsigned int wss;
+       int snd, rcv, ret;
+
+       if (!o->window_size)
+               return 0;
+
+       rcv = o->listen || o->pingpong;
+       snd = !o->listen || o->pingpong;
+       wss = o->window_size;
+       ret = 0;
+
+       if (rcv) {
+               ret = setsockopt(fd, SOL_SOCKET, SO_RCVBUF, (void *) &wss,
+                                       sizeof(wss));
+               if (ret < 0)
+                       td_verror(td, errno, "rcvbuf window size");
+       }
+       if (snd && !ret) {
+               ret = setsockopt(fd, SOL_SOCKET, SO_SNDBUF, (void *) &wss,
+                                       sizeof(wss));
+               if (ret < 0)
+                       td_verror(td, errno, "sndbuf window size");
+       }
+
+       return ret;
+#else
+       td_verror(td, -EINVAL, "setsockopt window size");
+       return -1;
+#endif
+}
+
+static int set_mss(struct thread_data *td, int fd)
+{
+#ifdef CONFIG_NET_MSS
+       struct netio_options *o = td->eo;
+       unsigned int mss;
+       int ret;
+
+       if (!o->mss || !is_tcp(o))
+               return 0;
+
+       mss = o->mss;
+       ret = setsockopt(fd, IPPROTO_TCP, TCP_MAXSEG, (void *) &mss,
+                               sizeof(mss));
+       if (ret < 0)
+               td_verror(td, errno, "setsockopt TCP_MAXSEG");
+
+       return ret;
+#else
+       td_verror(td, -EINVAL, "setsockopt TCP_MAXSEG");
+       return -1;
+#endif
+}
+
+
 /*
  * Return -1 for error and 'nr events' for a positive number
  * of events
@@ -277,7 +395,7 @@ static int splice_io_u(int fdin, int fdout, unsigned int len)
  */
 static int splice_in(struct thread_data *td, struct io_u *io_u)
 {
-       struct netio_data *nd = td->io_ops->data;
+       struct netio_data *nd = td->io_ops_data;
 
        return splice_io_u(io_u->file->fd, nd->pipes[1], io_u->xfer_buflen);
 }
@@ -288,7 +406,7 @@ static int splice_in(struct thread_data *td, struct io_u *io_u)
 static int splice_out(struct thread_data *td, struct io_u *io_u,
                      unsigned int len)
 {
-       struct netio_data *nd = td->io_ops->data;
+       struct netio_data *nd = td->io_ops_data;
 
        return splice_io_u(nd->pipes[0], io_u->file->fd, len);
 }
@@ -326,7 +444,7 @@ static int vmsplice_io_u(struct io_u *io_u, int fd, unsigned int len)
 static int vmsplice_io_u_out(struct thread_data *td, struct io_u *io_u,
                             unsigned int len)
 {
-       struct netio_data *nd = td->io_ops->data;
+       struct netio_data *nd = td->io_ops_data;
 
        return vmsplice_io_u(io_u, nd->pipes[0], len);
 }
@@ -336,7 +454,7 @@ static int vmsplice_io_u_out(struct thread_data *td, struct io_u *io_u,
  */
 static int vmsplice_io_u_in(struct thread_data *td, struct io_u *io_u)
 {
-       struct netio_data *nd = td->io_ops->data;
+       struct netio_data *nd = td->io_ops_data;
 
        return vmsplice_io_u(io_u, nd->pipes[1], io_u->xfer_buflen);
 }
@@ -384,15 +502,56 @@ static int fio_netio_splice_out(struct thread_data *td, struct io_u *io_u)
 }
 #endif
 
+static void store_udp_seq(struct netio_data *nd, struct io_u *io_u)
+{
+       struct udp_seq *us;
+
+       if (io_u->xfer_buflen < sizeof(*us))
+               return;
+
+       us = io_u->xfer_buf + io_u->xfer_buflen - sizeof(*us);
+       us->magic = cpu_to_le64((uint64_t) FIO_UDP_SEQ_MAGIC);
+       us->bs = cpu_to_le64((uint64_t) io_u->xfer_buflen);
+       us->seq = cpu_to_le64(nd->udp_send_seq++);
+}
+
+static void verify_udp_seq(struct thread_data *td, struct netio_data *nd,
+                          struct io_u *io_u)
+{
+       struct udp_seq *us;
+       uint64_t seq;
+
+       if (io_u->xfer_buflen < sizeof(*us))
+               return;
+
+       if (nd->seq_off)
+               return;
+
+       us = io_u->xfer_buf + io_u->xfer_buflen - sizeof(*us);
+       if (le64_to_cpu(us->magic) != FIO_UDP_SEQ_MAGIC)
+               return;
+       if (le64_to_cpu(us->bs) != io_u->xfer_buflen) {
+               nd->seq_off = 1;
+               return;
+       }
+
+       seq = le64_to_cpu(us->seq);
+
+       if (seq != nd->udp_recv_seq)
+               td->ts.drop_io_u[io_u->ddir] += seq - nd->udp_recv_seq;
+
+       nd->udp_recv_seq = seq + 1;
+}
+
 static int fio_netio_send(struct thread_data *td, struct io_u *io_u)
 {
-       struct netio_data *nd = td->io_ops->data;
+       struct netio_data *nd = td->io_ops_data;
        struct netio_options *o = td->eo;
        int ret, flags = 0;
 
        do {
                if (is_udp(o)) {
-                       struct sockaddr *to;
+                       const struct sockaddr *to;
                        socklen_t len;
 
                        if (is_ipv6(o)) {
@@ -403,6 +562,9 @@ static int fio_netio_send(struct thread_data *td, struct io_u *io_u)
                                len = sizeof(nd->addr);
                        }
 
+                       if (td->o.verify == VERIFY_NONE)
+                               store_udp_seq(nd, io_u);
+
                        ret = sendto(io_u->file->fd, io_u->xfer_buf,
                                        io_u->xfer_buflen, flags, to, len);
                } else {
@@ -428,7 +590,7 @@ static int fio_netio_send(struct thread_data *td, struct io_u *io_u)
        return ret;
 }
 
-static int is_udp_close(struct io_u *io_u, int len)
+static int is_close_msg(struct io_u *io_u, int len)
 {
        struct udp_close_msg *msg;
 
@@ -436,9 +598,9 @@ static int is_udp_close(struct io_u *io_u, int len)
                return 0;
 
        msg = io_u->xfer_buf;
-       if (ntohl(msg->magic) != FIO_LINK_OPEN_CLOSE_MAGIC)
+       if (le32_to_cpu(msg->magic) != FIO_LINK_OPEN_CLOSE_MAGIC)
                return 0;
-       if (ntohl(msg->cmd) != FIO_LINK_CLOSE)
+       if (le32_to_cpu(msg->cmd) != FIO_LINK_CLOSE)
                return 0;
 
        return 1;
@@ -446,7 +608,7 @@ static int is_udp_close(struct io_u *io_u, int len)
 
 static int fio_netio_recv(struct thread_data *td, struct io_u *io_u)
 {
-       struct netio_data *nd = td->io_ops->data;
+       struct netio_data *nd = td->io_ops_data;
        struct netio_options *o = td->eo;
        int ret, flags = 0;
 
@@ -470,13 +632,19 @@ static int fio_netio_recv(struct thread_data *td, struct io_u *io_u)
 
                        ret = recvfrom(io_u->file->fd, io_u->xfer_buf,
                                        io_u->xfer_buflen, flags, from, len);
-                       if (is_udp_close(io_u, ret)) {
+
+                       if (is_close_msg(io_u, ret)) {
                                td->done = 1;
                                return 0;
                        }
                } else {
                        ret = recv(io_u->file->fd, io_u->xfer_buf,
                                        io_u->xfer_buflen, flags);
+
+                       if (is_close_msg(io_u, ret)) {
+                               td->done = 1;
+                               return 0;
+                       }
                }
                if (ret > 0)
                        break;
@@ -489,13 +657,17 @@ static int fio_netio_recv(struct thread_data *td, struct io_u *io_u)
                flags |= MSG_WAITALL;
        } while (1);
 
+       if (is_udp(o) && td->o.verify == VERIFY_NONE)
+               verify_udp_seq(td, nd, io_u);
+
        return ret;
 }
 
-static int __fio_netio_queue(struct thread_data *td, struct io_u *io_u,
-                            enum fio_ddir ddir)
+static enum fio_q_status __fio_netio_queue(struct thread_data *td,
+                                          struct io_u *io_u,
+                                          enum fio_ddir ddir)
 {
-       struct netio_data *nd = td->io_ops->data;
+       struct netio_data *nd = td->io_ops_data;
        struct netio_options *o = td->eo;
        int ret;
 
@@ -515,11 +687,13 @@ static int __fio_netio_queue(struct thread_data *td, struct io_u *io_u,
                ret = 0;        /* must be a SYNC */
 
        if (ret != (int) io_u->xfer_buflen) {
-               if (ret >= 0) {
+               if (ret > 0) {
                        io_u->resid = io_u->xfer_buflen - ret;
                        io_u->error = 0;
                        return FIO_Q_COMPLETED;
-               } else {
+               } else if (!ret)
+                       return FIO_Q_BUSY;
+               else {
                        int err = errno;
 
                        if (ddir == DDIR_WRITE && err == EMSGSIZE)
@@ -535,7 +709,8 @@ static int __fio_netio_queue(struct thread_data *td, struct io_u *io_u,
        return FIO_Q_COMPLETED;
 }
 
-static int fio_netio_queue(struct thread_data *td, struct io_u *io_u)
+static enum fio_q_status fio_netio_queue(struct thread_data *td,
+                                        struct io_u *io_u)
 {
        struct netio_options *o = td->eo;
        int ret;
@@ -559,7 +734,7 @@ static int fio_netio_queue(struct thread_data *td, struct io_u *io_u)
 
 static int fio_netio_connect(struct thread_data *td, struct fio_file *f)
 {
-       struct netio_data *nd = td->io_ops->data;
+       struct netio_data *nd = td->io_ops_data;
        struct netio_options *o = td->eo;
        int type, domain;
 
@@ -578,6 +753,9 @@ static int fio_netio_connect(struct thread_data *td, struct fio_file *f)
        } else if (o->proto == FIO_TYPE_UNIX) {
                domain = AF_UNIX;
                type = SOCK_STREAM;
+       } else if (is_vsock(o)) {
+               domain = AF_VSOCK;
+               type = SOCK_STREAM;
        } else {
                log_err("fio: bad network type %d\n", o->proto);
                f->fd = -1;
@@ -601,6 +779,15 @@ static int fio_netio_connect(struct thread_data *td, struct fio_file *f)
        }
 #endif
 
+       if (set_window_size(td, f->fd)) {
+               close(f->fd);
+               return 1;
+       }
+       if (set_mss(td, f->fd)) {
+               close(f->fd);
+               return 1;
+       }
+
        if (is_udp(o)) {
                if (!fio_netio_is_multicast(td->o.filename))
                        return 0;
@@ -646,7 +833,14 @@ static int fio_netio_connect(struct thread_data *td, struct fio_file *f)
                        close(f->fd);
                        return 1;
                }
+       } else if (is_vsock(o)) {
+               socklen_t len = sizeof(nd->addr_vm);
 
+               if (connect(f->fd, (struct sockaddr *) &nd->addr_vm, len) < 0) {
+                       td_verror(td, errno, "connect");
+                       close(f->fd);
+                       return 1;
+               }
        } else {
                struct sockaddr_un *addr = &nd->addr_un;
                socklen_t len;
@@ -665,7 +859,7 @@ static int fio_netio_connect(struct thread_data *td, struct fio_file *f)
 
 static int fio_netio_accept(struct thread_data *td, struct fio_file *f)
 {
-       struct netio_data *nd = td->io_ops->data;
+       struct netio_data *nd = td->io_ops_data;
        struct netio_options *o = td->eo;
        socklen_t socklen;
        int state;
@@ -686,6 +880,9 @@ static int fio_netio_accept(struct thread_data *td, struct fio_file *f)
        if (o->proto == FIO_TYPE_TCP) {
                socklen = sizeof(nd->addr);
                f->fd = accept(nd->listenfd, (struct sockaddr *) &nd->addr, &socklen);
+       } else if (is_vsock(o)) {
+               socklen = sizeof(nd->addr_vm);
+               f->fd = accept(nd->listenfd, (struct sockaddr *) &nd->addr_vm, &socklen);
        } else {
                socklen = sizeof(nd->addr6);
                f->fd = accept(nd->listenfd, (struct sockaddr *) &nd->addr6, &socklen);
@@ -715,9 +912,9 @@ err:
        return 1;
 }
 
-static void fio_netio_udp_close(struct thread_data *td, struct fio_file *f)
+static void fio_netio_send_close(struct thread_data *td, struct fio_file *f)
 {
-       struct netio_data *nd = td->io_ops->data;
+       struct netio_data *nd = td->io_ops_data;
        struct netio_options *o = td->eo;
        struct udp_close_msg msg;
        struct sockaddr *to;
@@ -727,13 +924,16 @@ static void fio_netio_udp_close(struct thread_data *td, struct fio_file *f)
        if (is_ipv6(o)) {
                to = (struct sockaddr *) &nd->addr6;
                len = sizeof(nd->addr6);
+       } else if (is_vsock(o)) {
+               to = NULL;
+               len = 0;
        } else {
                to = (struct sockaddr *) &nd->addr;
                len = sizeof(nd->addr);
        }
 
-       msg.magic = htonl(FIO_LINK_OPEN_CLOSE_MAGIC);
-       msg.cmd = htonl(FIO_LINK_CLOSE);
+       msg.magic = cpu_to_le32((uint32_t) FIO_LINK_OPEN_CLOSE_MAGIC);
+       msg.cmd = cpu_to_le32((uint32_t) FIO_LINK_CLOSE);
 
        ret = sendto(f->fd, (void *) &msg, sizeof(msg), MSG_WAITALL, to, len);
        if (ret < 0)
@@ -742,21 +942,17 @@ static void fio_netio_udp_close(struct thread_data *td, struct fio_file *f)
 
 static int fio_netio_close_file(struct thread_data *td, struct fio_file *f)
 {
-       struct netio_options *o = td->eo;
-
        /*
-        * If this is an UDP connection, notify the receiver that we are
-        * closing down the link
+        * Notify the receiver that we are closing down the link
         */
-       if (is_udp(o))
-               fio_netio_udp_close(td, f);
+       fio_netio_send_close(td, f);
 
        return generic_close_file(td, f);
 }
 
 static int fio_netio_udp_recv_open(struct thread_data *td, struct fio_file *f)
 {
-       struct netio_data *nd = td->io_ops->data;
+       struct netio_data *nd = td->io_ops_data;
        struct netio_options *o = td->eo;
        struct udp_close_msg msg;
        struct sockaddr *to;
@@ -779,17 +975,19 @@ static int fio_netio_udp_recv_open(struct thread_data *td, struct fio_file *f)
 
        if (ntohl(msg.magic) != FIO_LINK_OPEN_CLOSE_MAGIC ||
            ntohl(msg.cmd) != FIO_LINK_OPEN) {
-               log_err("fio: bad udp open magic %x/%x\n", ntohl(msg.magic),
-                                                               ntohl(msg.cmd));
+               log_err("fio: bad udp open magic %x/%x\n",
+                       (unsigned int) ntohl(msg.magic),
+                       (unsigned int) ntohl(msg.cmd));
                return -1;
        }
 
+       fio_gettime(&td->start, NULL);
        return 0;
 }
 
-static int fio_netio_udp_send_open(struct thread_data *td, struct fio_file *f)
+static int fio_netio_send_open(struct thread_data *td, struct fio_file *f)
 {
-       struct netio_data *nd = td->io_ops->data;
+       struct netio_data *nd = td->io_ops_data;
        struct netio_options *o = td->eo;
        struct udp_close_msg msg;
        struct sockaddr *to;
@@ -799,6 +997,9 @@ static int fio_netio_udp_send_open(struct thread_data *td, struct fio_file *f)
        if (is_ipv6(o)) {
                len = sizeof(nd->addr6);
                to = (struct sockaddr *) &nd->addr6;
+       } else if (is_vsock(o)) {
+               len = sizeof(nd->addr_vm);
+               to = (struct sockaddr *) &nd->addr_vm;
        } else {
                len = sizeof(nd->addr);
                to = (struct sockaddr *) &nd->addr;
@@ -833,7 +1034,7 @@ static int fio_netio_open_file(struct thread_data *td, struct fio_file *f)
 
        if (is_udp(o)) {
                if (td_write(td))
-                       ret = fio_netio_udp_send_open(td, f);
+                       ret = fio_netio_send_open(td, f);
                else {
                        int state;
 
@@ -862,13 +1063,17 @@ static int fio_fill_addr(struct thread_data *td, const char *host, int af,
 
        memset(&hints, 0, sizeof(hints));
 
-       if (is_tcp(o))
+       if (is_tcp(o) || is_vsock(o))
                hints.ai_socktype = SOCK_STREAM;
        else
                hints.ai_socktype = SOCK_DGRAM;
 
        if (is_ipv6(o))
                hints.ai_family = AF_INET6;
+#ifdef CONFIG_VSOCK
+       else if (is_vsock(o))
+               hints.ai_family = AF_VSOCK;
+#endif
        else
                hints.ai_family = AF_INET;
 
@@ -891,7 +1096,7 @@ static int fio_fill_addr(struct thread_data *td, const char *host, int af,
 static int fio_netio_setup_connect_inet(struct thread_data *td,
                                        const char *host, unsigned short port)
 {
-       struct netio_data *nd = td->io_ops->data;
+       struct netio_data *nd = td->io_ops_data;
        struct netio_options *o = td->eo;
        struct addrinfo *res = NULL;
        void *dst, *src;
@@ -941,12 +1146,48 @@ static int fio_netio_setup_connect_inet(struct thread_data *td,
 static int fio_netio_setup_connect_unix(struct thread_data *td,
                                        const char *path)
 {
-       struct netio_data *nd = td->io_ops->data;
+       struct netio_data *nd = td->io_ops_data;
        struct sockaddr_un *soun = &nd->addr_un;
 
        soun->sun_family = AF_UNIX;
-       strcpy(soun->sun_path, path);
+       snprintf(soun->sun_path, sizeof(soun->sun_path), "%s", path);
+       return 0;
+}
+
+static int fio_netio_setup_connect_vsock(struct thread_data *td,
+                                       const char *host, unsigned short port)
+{
+#ifdef CONFIG_VSOCK
+       struct netio_data *nd = td->io_ops_data;
+       struct sockaddr_vm *addr = &nd->addr_vm;
+       int cid;
+
+       if (!host) {
+               log_err("fio: connect with no host to connect to.\n");
+               if (td_read(td))
+                       log_err("fio: did you forget to set 'listen'?\n");
+
+               td_verror(td, EINVAL, "no hostname= set");
+               return 1;
+       }
+
+       addr->svm_family = AF_VSOCK;
+       addr->svm_port = port;
+
+       if (host) {
+               cid = atoi(host);
+               if (cid < 0 || cid > UINT32_MAX) {
+                       log_err("fio: invalid CID %d\n", cid);
+                       return 1;
+               }
+               addr->svm_cid = cid;
+       }
+
        return 0;
+#else
+       td_verror(td, -EINVAL, "vsock not supported");
+       return 1;
+#endif
 }
 
 static int fio_netio_setup_connect(struct thread_data *td)
@@ -955,13 +1196,15 @@ static int fio_netio_setup_connect(struct thread_data *td)
 
        if (is_udp(o) || is_tcp(o))
                return fio_netio_setup_connect_inet(td, td->o.filename,o->port);
+       else if (is_vsock(o))
+               return fio_netio_setup_connect_vsock(td, td->o.filename, o->port);
        else
                return fio_netio_setup_connect_unix(td, td->o.filename);
 }
 
 static int fio_netio_setup_listen_unix(struct thread_data *td, const char *path)
 {
-       struct netio_data *nd = td->io_ops->data;
+       struct netio_data *nd = td->io_ops_data;
        struct sockaddr_un *addr = &nd->addr_un;
        mode_t mode;
        int len, fd;
@@ -974,9 +1217,8 @@ static int fio_netio_setup_listen_unix(struct thread_data *td, const char *path)
 
        mode = umask(000);
 
-       memset(addr, 0, sizeof(*addr));
        addr->sun_family = AF_UNIX;
-       strcpy(addr->sun_path, path);
+       snprintf(addr->sun_path, sizeof(addr->sun_path), "%s", path);
        unlink(path);
 
        len = sizeof(addr->sun_family) + strlen(path) + 1;
@@ -994,17 +1236,15 @@ static int fio_netio_setup_listen_unix(struct thread_data *td, const char *path)
 
 static int fio_netio_setup_listen_inet(struct thread_data *td, short port)
 {
-       struct netio_data *nd = td->io_ops->data;
+       struct netio_data *nd = td->io_ops_data;
        struct netio_options *o = td->eo;
        struct ip_mreq mr;
        struct sockaddr_in sin;
-       struct sockaddr_in6 sin6;
        struct sockaddr *saddr;
        int fd, opt, type, domain;
        socklen_t len;
 
        memset(&sin, 0, sizeof(sin));
-       memset(&sin6, 0, sizeof(sin6));
 
        if (o->proto == FIO_TYPE_TCP) {
                type = SOCK_STREAM;
@@ -1043,6 +1283,15 @@ static int fio_netio_setup_listen_inet(struct thread_data *td, short port)
        }
 #endif
 
+       if (set_window_size(td, fd)) {
+               close(fd);
+               return 1;
+       }
+       if (set_mss(td, fd)) {
+               close(fd);
+               return 1;
+       }
+
        if (td->o.filename) {
                if (!is_udp(o) || !fio_netio_is_multicast(td->o.filename)) {
                        log_err("fio: hostname not valid for non-multicast inbound network IO\n");
@@ -1050,7 +1299,7 @@ static int fio_netio_setup_listen_inet(struct thread_data *td, short port)
                        return 1;
                }
                if (is_ipv6(o)) {
-                       log_err("fio: IPv6 not supported for multicast network IO");
+                       log_err("fio: IPv6 not supported for multicast network IO\n");
                        close(fd);
                        return 1;
                }
@@ -1087,27 +1336,71 @@ static int fio_netio_setup_listen_inet(struct thread_data *td, short port)
                len = sizeof(nd->addr6);
 
                nd->addr6.sin6_family = AF_INET6;
-               nd->addr6.sin6_addr = sin6.sin6_addr.s6_addr ? sin6.sin6_addr : in6addr_any;
+               nd->addr6.sin6_addr = in6addr_any;
                nd->addr6.sin6_port = htons(port);
        }
 
        if (bind(fd, saddr, len) < 0) {
+               close(fd);
+               td_verror(td, errno, "bind");
+               return 1;
+       }
+
+       nd->listenfd = fd;
+       return 0;
+}
+
+static int fio_netio_setup_listen_vsock(struct thread_data *td, short port, int type)
+{
+#ifdef CONFIG_VSOCK
+       struct netio_data *nd = td->io_ops_data;
+       struct sockaddr_vm *addr = &nd->addr_vm;
+       int fd, opt;
+       socklen_t len;
+
+       fd = socket(AF_VSOCK, type, 0);
+       if (fd < 0) {
+               td_verror(td, errno, "socket");
+               return 1;
+       }
+
+       opt = 1;
+       if (setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, (void *) &opt, sizeof(opt)) < 0) {
+               td_verror(td, errno, "setsockopt");
+               close(fd);
+               return 1;
+       }
+
+       len = sizeof(*addr);
+
+       nd->addr_vm.svm_family = AF_VSOCK;
+       nd->addr_vm.svm_cid = VMADDR_CID_ANY;
+       nd->addr_vm.svm_port = port;
+
+       if (bind(fd, (struct sockaddr *) addr, len) < 0) {
                td_verror(td, errno, "bind");
+               close(fd);
                return 1;
        }
 
        nd->listenfd = fd;
        return 0;
+#else
+       td_verror(td, -EINVAL, "vsock not supported");
+       return -1;
+#endif
 }
 
 static int fio_netio_setup_listen(struct thread_data *td)
 {
-       struct netio_data *nd = td->io_ops->data;
+       struct netio_data *nd = td->io_ops_data;
        struct netio_options *o = td->eo;
        int ret;
 
        if (is_udp(o) || is_tcp(o))
                ret = fio_netio_setup_listen_inet(td, o->port);
+       else if (is_vsock(o))
+               ret = fio_netio_setup_listen_vsock(td, o->port, SOCK_STREAM);
        else
                ret = fio_netio_setup_listen_unix(td, td->o.filename);
 
@@ -1143,12 +1436,17 @@ static int fio_netio_init(struct thread_data *td)
        if (o->proto == FIO_TYPE_UNIX && o->port) {
                log_err("fio: network IO port not valid with unix socket\n");
                return 1;
+       } else if (is_vsock(o) && !o->port) {
+               log_err("fio: network IO requires port for vsock\n");
+               return 1;
        } else if (o->proto != FIO_TYPE_UNIX && !o->port) {
                log_err("fio: network IO requires port for tcp or udp\n");
                return 1;
        }
 
-       if (!is_tcp(o)) {
+       o->port += td->subjob_number;
+
+       if (!is_tcp(o) && !is_vsock(o)) {
                if (o->listen) {
                        log_err("fio: listen only valid for TCP proto IO\n");
                        return 1;
@@ -1175,7 +1473,7 @@ static int fio_netio_init(struct thread_data *td)
 
 static void fio_netio_cleanup(struct thread_data *td)
 {
-       struct netio_data *nd = td->io_ops->data;
+       struct netio_data *nd = td->io_ops_data;
 
        if (nd) {
                if (nd->listenfd != -1)
@@ -1196,15 +1494,14 @@ static int fio_netio_setup(struct thread_data *td)
        if (!td->files_index) {
                add_file(td, td->o.filename ?: "net", 0, 0);
                td->o.nr_files = td->o.nr_files ?: 1;
+               td->o.open_files++;
        }
 
-       if (!td->io_ops->data) {
-               nd = malloc(sizeof(*nd));;
-
-               memset(nd, 0, sizeof(*nd));
+       if (!td->io_ops_data) {
+               nd = calloc(1, sizeof(*nd));
                nd->listenfd = -1;
                nd->pipes[0] = nd->pipes[1] = -1;
-               td->io_ops->data = nd;
+               td->io_ops_data = nd;
        }
 
        return 0;
@@ -1212,7 +1509,7 @@ static int fio_netio_setup(struct thread_data *td)
 
 static void fio_netio_terminate(struct thread_data *td)
 {
-       kill(td->pid, SIGUSR2);
+       kill(td->pid, SIGTERM);
 }
 
 #ifdef CONFIG_LINUX_SPLICE
@@ -1222,7 +1519,7 @@ static int fio_netio_setup_splice(struct thread_data *td)
 
        fio_netio_setup(td);
 
-       nd = td->io_ops->data;
+       nd = td->io_ops_data;
        if (nd) {
                if (pipe(nd->pipes) < 0)
                        return 1;