new helper: msg_data_left()
convert open-coded instances Signed-off-by: Al Viro <viro@zeniv.linux.org.uk>
This commit is contained in:
		
					parent
					
						
							
								a2dd3793a1
							
						
					
				
			
			
				commit
				
					
						01e97e6517
					
				
			
		
					 8 changed files with 27 additions and 23 deletions
				
			
		| 
						 | 
				
			
			@ -56,8 +56,8 @@ static int hash_sendmsg(struct socket *sock, struct msghdr *msg,
 | 
			
		|||
 | 
			
		||||
	ctx->more = 0;
 | 
			
		||||
 | 
			
		||||
	while (iov_iter_count(&msg->msg_iter)) {
 | 
			
		||||
		int len = iov_iter_count(&msg->msg_iter);
 | 
			
		||||
	while (msg_data_left(msg)) {
 | 
			
		||||
		int len = msg_data_left(msg);
 | 
			
		||||
 | 
			
		||||
		if (len > limit)
 | 
			
		||||
			len = limit;
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
| 
						 | 
				
			
			@ -641,7 +641,7 @@ static int skcipher_recvmsg_sync(struct socket *sock, struct msghdr *msg,
 | 
			
		|||
	long copied = 0;
 | 
			
		||||
 | 
			
		||||
	lock_sock(sk);
 | 
			
		||||
	while (iov_iter_count(&msg->msg_iter)) {
 | 
			
		||||
	while (msg_data_left(msg)) {
 | 
			
		||||
		sgl = list_first_entry(&ctx->tsgl,
 | 
			
		||||
				       struct skcipher_sg_list, list);
 | 
			
		||||
		sg = sgl->sg;
 | 
			
		||||
| 
						 | 
				
			
			@ -655,7 +655,7 @@ static int skcipher_recvmsg_sync(struct socket *sock, struct msghdr *msg,
 | 
			
		|||
				goto unlock;
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		used = min_t(unsigned long, ctx->used, iov_iter_count(&msg->msg_iter));
 | 
			
		||||
		used = min_t(unsigned long, ctx->used, msg_data_left(msg));
 | 
			
		||||
 | 
			
		||||
		used = af_alg_make_sg(&ctx->rsgl, &msg->msg_iter, used);
 | 
			
		||||
		err = used;
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
| 
						 | 
				
			
			@ -357,13 +357,13 @@ static void handle_tx(struct vhost_net *net)
 | 
			
		|||
		iov_iter_init(&msg.msg_iter, WRITE, vq->iov, out, len);
 | 
			
		||||
		iov_iter_advance(&msg.msg_iter, hdr_size);
 | 
			
		||||
		/* Sanity check */
 | 
			
		||||
		if (!iov_iter_count(&msg.msg_iter)) {
 | 
			
		||||
		if (!msg_data_left(&msg)) {
 | 
			
		||||
			vq_err(vq, "Unexpected header len for TX: "
 | 
			
		||||
			       "%zd expected %zd\n",
 | 
			
		||||
			       len, hdr_size);
 | 
			
		||||
			break;
 | 
			
		||||
		}
 | 
			
		||||
		len = iov_iter_count(&msg.msg_iter);
 | 
			
		||||
		len = msg_data_left(&msg);
 | 
			
		||||
 | 
			
		||||
		zcopy_used = zcopy && len >= VHOST_GOODCOPY_LEN
 | 
			
		||||
				   && (nvq->upend_idx + 1) % UIO_MAXIOV !=
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
| 
						 | 
				
			
			@ -139,6 +139,11 @@ static inline struct cmsghdr * cmsg_nxthdr (struct msghdr *__msg, struct cmsghdr
 | 
			
		|||
	return __cmsg_nxthdr(__msg->msg_control, __msg->msg_controllen, __cmsg);
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
static inline size_t msg_data_left(struct msghdr *msg)
 | 
			
		||||
{
 | 
			
		||||
	return iov_iter_count(&msg->msg_iter);
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
/* "Socket"-level control message types: */
 | 
			
		||||
 | 
			
		||||
#define	SCM_RIGHTS	0x01		/* rw: access rights (array of int) */
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
| 
						 | 
				
			
			@ -673,7 +673,7 @@ int skb_copy_and_csum_datagram_msg(struct sk_buff *skb,
 | 
			
		|||
	if (!chunk)
 | 
			
		||||
		return 0;
 | 
			
		||||
 | 
			
		||||
	if (iov_iter_count(&msg->msg_iter) < chunk) {
 | 
			
		||||
	if (msg_data_left(msg) < chunk) {
 | 
			
		||||
		if (__skb_checksum_complete(skb))
 | 
			
		||||
			goto csum_error;
 | 
			
		||||
		if (skb_copy_datagram_msg(skb, hlen, msg, chunk))
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
| 
						 | 
				
			
			@ -1119,7 +1119,7 @@ int tcp_sendmsg(struct sock *sk, struct msghdr *msg, size_t size)
 | 
			
		|||
 | 
			
		||||
	sg = !!(sk->sk_route_caps & NETIF_F_SG);
 | 
			
		||||
 | 
			
		||||
	while (iov_iter_count(&msg->msg_iter)) {
 | 
			
		||||
	while (msg_data_left(msg)) {
 | 
			
		||||
		int copy = 0;
 | 
			
		||||
		int max = size_goal;
 | 
			
		||||
 | 
			
		||||
| 
						 | 
				
			
			@ -1163,8 +1163,8 @@ new_segment:
 | 
			
		|||
		}
 | 
			
		||||
 | 
			
		||||
		/* Try to append data to the end of skb. */
 | 
			
		||||
		if (copy > iov_iter_count(&msg->msg_iter))
 | 
			
		||||
			copy = iov_iter_count(&msg->msg_iter);
 | 
			
		||||
		if (copy > msg_data_left(msg))
 | 
			
		||||
			copy = msg_data_left(msg);
 | 
			
		||||
 | 
			
		||||
		/* Where to copy to? */
 | 
			
		||||
		if (skb_availroom(skb) > 0) {
 | 
			
		||||
| 
						 | 
				
			
			@ -1221,7 +1221,7 @@ new_segment:
 | 
			
		|||
		tcp_skb_pcount_set(skb, 0);
 | 
			
		||||
 | 
			
		||||
		copied += copy;
 | 
			
		||||
		if (!iov_iter_count(&msg->msg_iter)) {
 | 
			
		||||
		if (!msg_data_left(msg)) {
 | 
			
		||||
			tcp_tx_timestamp(sk, skb);
 | 
			
		||||
			goto out;
 | 
			
		||||
		}
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
| 
						 | 
				
			
			@ -564,8 +564,8 @@ static int rxrpc_send_data(struct rxrpc_sock *rx,
 | 
			
		|||
			max &= ~(call->conn->size_align - 1UL);
 | 
			
		||||
 | 
			
		||||
			chunk = max;
 | 
			
		||||
			if (chunk > iov_iter_count(&msg->msg_iter) && !more)
 | 
			
		||||
				chunk = iov_iter_count(&msg->msg_iter);
 | 
			
		||||
			if (chunk > msg_data_left(msg) && !more)
 | 
			
		||||
				chunk = msg_data_left(msg);
 | 
			
		||||
 | 
			
		||||
			space = chunk + call->conn->size_align;
 | 
			
		||||
			space &= ~(call->conn->size_align - 1UL);
 | 
			
		||||
| 
						 | 
				
			
			@ -608,11 +608,11 @@ static int rxrpc_send_data(struct rxrpc_sock *rx,
 | 
			
		|||
		sp = rxrpc_skb(skb);
 | 
			
		||||
 | 
			
		||||
		/* append next segment of data to the current buffer */
 | 
			
		||||
		if (iov_iter_count(&msg->msg_iter) > 0) {
 | 
			
		||||
		if (msg_data_left(msg) > 0) {
 | 
			
		||||
			int copy = skb_tailroom(skb);
 | 
			
		||||
			ASSERTCMP(copy, >, 0);
 | 
			
		||||
			if (copy > iov_iter_count(&msg->msg_iter))
 | 
			
		||||
				copy = iov_iter_count(&msg->msg_iter);
 | 
			
		||||
			if (copy > msg_data_left(msg))
 | 
			
		||||
				copy = msg_data_left(msg);
 | 
			
		||||
			if (copy > sp->remain)
 | 
			
		||||
				copy = sp->remain;
 | 
			
		||||
 | 
			
		||||
| 
						 | 
				
			
			@ -633,7 +633,7 @@ static int rxrpc_send_data(struct rxrpc_sock *rx,
 | 
			
		|||
 | 
			
		||||
		/* add the packet to the send queue if it's now full */
 | 
			
		||||
		if (sp->remain <= 0 ||
 | 
			
		||||
		    (iov_iter_count(&msg->msg_iter) == 0 && !more)) {
 | 
			
		||||
		    (msg_data_left(msg) == 0 && !more)) {
 | 
			
		||||
			struct rxrpc_connection *conn = call->conn;
 | 
			
		||||
			uint32_t seq;
 | 
			
		||||
			size_t pad;
 | 
			
		||||
| 
						 | 
				
			
			@ -663,7 +663,7 @@ static int rxrpc_send_data(struct rxrpc_sock *rx,
 | 
			
		|||
			sp->hdr.serviceId = conn->service_id;
 | 
			
		||||
 | 
			
		||||
			sp->hdr.flags = conn->out_clientflag;
 | 
			
		||||
			if (iov_iter_count(&msg->msg_iter) == 0 && !more)
 | 
			
		||||
			if (msg_data_left(msg) == 0 && !more)
 | 
			
		||||
				sp->hdr.flags |= RXRPC_LAST_PACKET;
 | 
			
		||||
			else if (CIRC_SPACE(call->acks_head, call->acks_tail,
 | 
			
		||||
					    call->acks_winsz) > 1)
 | 
			
		||||
| 
						 | 
				
			
			@ -679,11 +679,10 @@ static int rxrpc_send_data(struct rxrpc_sock *rx,
 | 
			
		|||
 | 
			
		||||
			memcpy(skb->head, &sp->hdr,
 | 
			
		||||
			       sizeof(struct rxrpc_header));
 | 
			
		||||
			rxrpc_queue_packet(call, skb,
 | 
			
		||||
					   iov_iter_count(&msg->msg_iter) == 0 && !more);
 | 
			
		||||
			rxrpc_queue_packet(call, skb, !msg_data_left(msg) && !more);
 | 
			
		||||
			skb = NULL;
 | 
			
		||||
		}
 | 
			
		||||
	} while (iov_iter_count(&msg->msg_iter) > 0);
 | 
			
		||||
	} while (msg_data_left(msg) > 0);
 | 
			
		||||
 | 
			
		||||
success:
 | 
			
		||||
	ret = copied;
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
| 
						 | 
				
			
			@ -612,7 +612,7 @@ EXPORT_SYMBOL(__sock_tx_timestamp);
 | 
			
		|||
 | 
			
		||||
static inline int sock_sendmsg_nosec(struct socket *sock, struct msghdr *msg)
 | 
			
		||||
{
 | 
			
		||||
	int ret = sock->ops->sendmsg(sock, msg, iov_iter_count(&msg->msg_iter));
 | 
			
		||||
	int ret = sock->ops->sendmsg(sock, msg, msg_data_left(msg));
 | 
			
		||||
	BUG_ON(ret == -EIOCBQUEUED);
 | 
			
		||||
	return ret;
 | 
			
		||||
}
 | 
			
		||||
| 
						 | 
				
			
			@ -620,7 +620,7 @@ static inline int sock_sendmsg_nosec(struct socket *sock, struct msghdr *msg)
 | 
			
		|||
int sock_sendmsg(struct socket *sock, struct msghdr *msg)
 | 
			
		||||
{
 | 
			
		||||
	int err = security_socket_sendmsg(sock, msg,
 | 
			
		||||
					  iov_iter_count(&msg->msg_iter));
 | 
			
		||||
					  msg_data_left(msg));
 | 
			
		||||
 | 
			
		||||
	return err ?: sock_sendmsg_nosec(sock, msg);
 | 
			
		||||
}
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
		Loading…
	
	Add table
		Add a link
		
	
		Reference in a new issue