hollalinux
/
linux-sg2042
mirror of https://github.com/sophgo/linux-riscv.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

tun: Use iovec iterators 

This patch removes the use of skb_copy_datagram_const_iovec in
favour of the iovec iterator-based skb_copy_datagram_iter.

Signed-off-by: Herbert Xu <herbert@gondor.apana.org.au>
Signed-off-by: David S. Miller <davem@davemloft.net>
This commit is contained in:
Herbert Xu 2014-11-07 21:22:23 +08:00 committed by David S. Miller
parent a8f820aa40
commit e0b46d0ee9
1 changed files with 30 additions and 35 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

65
drivers/net/tun.c
View File

@ -71,6 +71,7 @@
#include <net/rtnetlink.h> #include <net/rtnetlink.h>
#include <net/sock.h> #include <net/sock.h>
#include <linux/seq_file.h> #include <linux/seq_file.h>
#include <linux/uio.h>
#include <asm/uaccess.h> #include <asm/uaccess.h>
@ -1230,11 +1231,11 @@ static ssize_t tun_chr_aio_write(struct kiocb *iocb, const struct iovec *iv,
static ssize_t tun_put_user(struct tun_struct *tun, static ssize_t tun_put_user(struct tun_struct *tun,
struct tun_file *tfile, struct tun_file *tfile,
struct sk_buff *skb, struct sk_buff *skb,
const struct iovec *iv, int len) struct iov_iter *iter)
{ {
struct tun_pi pi = { 0, skb->protocol }; struct tun_pi pi = { 0, skb->protocol };
ssize_t total = 0; ssize_t total;
int vlan_offset = 0, copied; int vlan_offset;
int vlan_hlen = 0; int vlan_hlen = 0;
int vnet_hdr_sz = 0; int vnet_hdr_sz = 0;
@ -1244,23 +1245,25 @@ static ssize_t tun_put_user(struct tun_struct *tun,
if (tun->flags & TUN_VNET_HDR) if (tun->flags & TUN_VNET_HDR)
vnet_hdr_sz = tun->vnet_hdr_sz; vnet_hdr_sz = tun->vnet_hdr_sz;
total = skb->len + vlan_hlen + vnet_hdr_sz;
if (!(tun->flags & TUN_NO_PI)) { if (!(tun->flags & TUN_NO_PI)) {
if ((len -= sizeof(pi)) < 0) if (iov_iter_count(iter) < sizeof(pi))
return -EINVAL; return -EINVAL;
if (len < skb->len + vlan_hlen + vnet_hdr_sz) { total += sizeof(pi);
if (iov_iter_count(iter) < total) {
/* Packet will be striped */ /* Packet will be striped */
pi.flags |= TUN_PKT_STRIP; pi.flags |= TUN_PKT_STRIP;
} }
if (memcpy_toiovecend(iv, (void *) &pi, 0, sizeof(pi))) if (copy_to_iter(&pi, sizeof(pi), iter) != sizeof(pi))
return -EFAULT; return -EFAULT;
total += sizeof(pi);
} }
if (vnet_hdr_sz) { if (vnet_hdr_sz) {
struct virtio_net_hdr gso = { 0 }; /* no info leak */ struct virtio_net_hdr gso = { 0 }; /* no info leak */
if ((len -= vnet_hdr_sz) < 0) if (iov_iter_count(iter) < vnet_hdr_sz)
return -EINVAL; return -EINVAL;
if (skb_is_gso(skb)) { if (skb_is_gso(skb)) {
@ -1299,17 +1302,12 @@ static ssize_t tun_put_user(struct tun_struct *tun,
gso.flags = VIRTIO_NET_HDR_F_DATA_VALID; gso.flags = VIRTIO_NET_HDR_F_DATA_VALID;
} /* else everything is zero */ } /* else everything is zero */
if (unlikely(memcpy_toiovecend(iv, (void *)&gso, total, if (copy_to_iter(&gso, sizeof(gso), iter) != sizeof(gso))
sizeof(gso))))
return -EFAULT; return -EFAULT;
total += vnet_hdr_sz;
} }
copied = total;
len = min_t(int, skb->len + vlan_hlen, len);
total += skb->len + vlan_hlen;
if (vlan_hlen) { if (vlan_hlen) {
int copy, ret; int ret;
struct { struct {
__be16 h_vlan_proto; __be16 h_vlan_proto;
__be16 h_vlan_TCI; __be16 h_vlan_TCI;
@ -1320,36 +1318,32 @@ static ssize_t tun_put_user(struct tun_struct *tun,
vlan_offset = offsetof(struct vlan_ethhdr, h_vlan_proto); vlan_offset = offsetof(struct vlan_ethhdr, h_vlan_proto);
copy = min_t(int, vlan_offset, len); ret = skb_copy_datagram_iter(skb, 0, iter, vlan_offset);
ret = skb_copy_datagram_const_iovec(skb, 0, iv, copied, copy); if (ret || !iov_iter_count(iter))
len -= copy;
copied += copy;
if (ret || !len)
goto done; goto done;
copy = min_t(int, sizeof(veth), len); ret = copy_to_iter(&veth, sizeof(veth), iter);
ret = memcpy_toiovecend(iv, (void *)&veth, copied, copy); if (ret != sizeof(veth) || !iov_iter_count(iter))
len -= copy;
copied += copy;
if (ret || !len)
goto done; goto done;
} }
skb_copy_datagram_const_iovec(skb, vlan_offset, iv, copied, len); skb_copy_datagram_iter(skb, vlan_offset, iter, skb->len - vlan_offset);
done: done:
tun->dev->stats.tx_packets++; tun->dev->stats.tx_packets++;
tun->dev->stats.tx_bytes += len; tun->dev->stats.tx_bytes += skb->len + vlan_hlen;
return total; return total;
} }
static ssize_t tun_do_read(struct tun_struct *tun, struct tun_file *tfile, static ssize_t tun_do_read(struct tun_struct *tun, struct tun_file *tfile,
const struct iovec *iv, ssize_t len, int noblock) const struct iovec *iv, unsigned long segs,
ssize_t len, int noblock)
{ {
struct sk_buff *skb; struct sk_buff *skb;
ssize_t ret = 0; ssize_t ret = 0;
int peeked, err, off = 0; int peeked, err, off = 0;
struct iov_iter iter;
tun_debug(KERN_INFO, tun, "tun_do_read\n"); tun_debug(KERN_INFO, tun, "tun_do_read\n");
@ -1362,11 +1356,12 @@ static ssize_t tun_do_read(struct tun_struct *tun, struct tun_file *tfile,
/* Read frames from queue */ /* Read frames from queue */
skb = __skb_recv_datagram(tfile->socket.sk, noblock ? MSG_DONTWAIT : 0, skb = __skb_recv_datagram(tfile->socket.sk, noblock ? MSG_DONTWAIT : 0,
&peeked, &off, &err); &peeked, &off, &err);
if (skb) { if (!skb)
ret = tun_put_user(tun, tfile, skb, iv, len); return ret;
kfree_skb(skb);
} else iov_iter_init(&iter, READ, iv, segs, len);
ret = err; ret = tun_put_user(tun, tfile, skb, &iter);
kfree_skb(skb);
return ret; return ret;
} }
@ -1387,7 +1382,7 @@ static ssize_t tun_chr_aio_read(struct kiocb *iocb, const struct iovec *iv,
goto out; goto out;
} }
ret = tun_do_read(tun, tfile, iv, len, ret = tun_do_read(tun, tfile, iv, count, len,
file->f_flags & O_NONBLOCK); file->f_flags & O_NONBLOCK);
ret = min_t(ssize_t, ret, len); ret = min_t(ssize_t, ret, len);
if (ret > 0) if (ret > 0)
@ -1488,7 +1483,7 @@ static int tun_recvmsg(struct kiocb *iocb, struct socket *sock,
SOL_PACKET, TUN_TX_TIMESTAMP); SOL_PACKET, TUN_TX_TIMESTAMP);
goto out; goto out;
} }
ret = tun_do_read(tun, tfile, m->msg_iov, total_len, ret = tun_do_read(tun, tfile, m->msg_iov, m->msg_iovlen, total_len,
flags & MSG_DONTWAIT); flags & MSG_DONTWAIT);
if (ret > total_len) { if (ret > total_len) {
m->msg_flags |= MSG_TRUNC; m->msg_flags |= MSG_TRUNC;