udplite.h 3.83 KB
Newer Older
1
/* SPDX-License-Identifier: GPL-2.0 */
2 3 4 5 6 7
/*
 *	Definitions for the UDP-Lite (RFC 3828) code.
 */
#ifndef _UDPLITE_H
#define _UDPLITE_H

8 9
#include <net/ip6_checksum.h>

10 11 12 13 14
/* UDP-Lite socket options */
#define UDPLITE_SEND_CSCOV   10 /* sender partial coverage (as sent)      */
#define UDPLITE_RECV_CSCOV   11 /* receiver partial coverage (threshold ) */

extern struct proto 		udplite_prot;
15
extern struct udp_table		udplite_table;
16 17 18 19 20 21 22

/*
 *	Checksum computation is all in software, hence simpler getfrag.
 */
static __inline__ int udplite_getfrag(void *from, char *to, int  offset,
				      int len, int odd, struct sk_buff *skb)
{
23
	struct msghdr *msg = from;
24
	return copy_from_iter_full(to, len, &msg->msg_iter) ? 0 : -EFAULT;
25 26 27 28 29
}

/* Designate sk as UDP-Lite socket */
static inline int udplite_sk_init(struct sock *sk)
{
30
	udp_init_sock(sk);
31 32 33 34 35 36 37 38 39 40 41 42 43 44 45
	udp_sk(sk)->pcflag = UDPLITE_BIT;
	return 0;
}

/*
 * 	Checksumming routines
 */
static inline int udplite_checksum_init(struct sk_buff *skb, struct udphdr *uh)
{
	u16 cscov;

        /* In UDPv4 a zero checksum means that the transmitter generated no
         * checksum. UDP-Lite (like IPv6) mandates checksums, hence packets
         * with a zero checksum field are illegal.                            */
	if (uh->check == 0) {
46
		net_dbg_ratelimited("UDPLite: zeroed checksum field\n");
47 48 49 50 51 52
		return 1;
	}

	cscov = ntohs(uh->len);

	if (cscov == 0)		 /* Indicates that full coverage is required. */
53
		;
54 55 56 57
	else if (cscov < 8  || cscov > skb->len) {
		/*
		 * Coverage length violates RFC 3828: log and discard silently.
		 */
58 59
		net_dbg_ratelimited("UDPLite: bad csum coverage %d/%d\n",
				    cscov, skb->len);
60 61
		return 1;

62
	} else if (cscov < skb->len) {
63
        	UDP_SKB_CB(skb)->partial_cov = 1;
64 65 66
		UDP_SKB_CB(skb)->cscov = cscov;
		if (skb->ip_summed == CHECKSUM_COMPLETE)
			skb->ip_summed = CHECKSUM_NONE;
67
		skb->csum_valid = 0;
68
        }
69 70 71 72

	return 0;
}

73 74
/* Slow-path computation of checksum. Socket is locked. */
static inline __wsum udplite_csum_outgoing(struct sock *sk, struct sk_buff *skb)
75
{
76
	const struct udp_sock *up = udp_sk(skb->sk);
77
	int cscov = up->len;
78
	__wsum csum = 0;
79

80 81 82 83 84
	if (up->pcflag & UDPLITE_SEND_CC) {
		/*
		 * Sender has set `partial coverage' option on UDP-Lite socket.
		 * The special case "up->pcslen == 0" signifies full coverage.
		 */
85
		if (up->pcslen < up->len) {
86 87 88
			if (0 < up->pcslen)
				cscov = up->pcslen;
			udp_hdr(skb)->len = htons(up->pcslen);
89
		}
90 91 92 93 94 95 96 97 98 99
		/*
		 * NOTE: Causes for the error case  `up->pcslen > up->len':
		 *        (i)  Application error (will not be penalized).
		 *       (ii)  Payload too big for send buffer: data is split
		 *             into several packets, each with its own header.
		 *             In this case (e.g. last segment), coverage may
		 *             exceed packet length.
		 *       Since packets with coverage length > packet length are
		 *       illegal, we fall back to the defaults here.
		 */
100 101 102 103 104
	}

	skb->ip_summed = CHECKSUM_NONE;     /* no HW support for checksumming */

	skb_queue_walk(&sk->sk_write_queue, skb) {
105 106
		const int off = skb_transport_offset(skb);
		const int len = skb->len - off;
107 108 109 110 111 112 113 114 115

		csum = skb_checksum(skb, off, (cscov > len)? len : cscov, csum);

		if ((cscov -= len) <= 0)
			break;
	}
	return csum;
}

116
/* Fast-path computation of checksum. Socket may not be locked. */
Herbert Xu's avatar
Herbert Xu committed
117 118
static inline __wsum udplite_csum(struct sk_buff *skb)
{
119
	const struct udp_sock *up = udp_sk(skb->sk);
Herbert Xu's avatar
Herbert Xu committed
120
	const int off = skb_transport_offset(skb);
121
	int len = skb->len - off;
Herbert Xu's avatar
Herbert Xu committed
122

123 124 125 126 127
	if ((up->pcflag & UDPLITE_SEND_CC) && up->pcslen < len) {
		if (0 < up->pcslen)
			len = up->pcslen;
		udp_hdr(skb)->len = htons(up->pcslen);
	}
Herbert Xu's avatar
Herbert Xu committed
128 129
	skb->ip_summed = CHECKSUM_NONE;     /* no HW support for checksumming */

130
	return skb_checksum(skb, off, len, 0);
Herbert Xu's avatar
Herbert Xu committed
131 132
}

133 134 135
void udplite4_register(void);
int udplite_get_port(struct sock *sk, unsigned short snum,
		     int (*scmp)(const struct sock *, const struct sock *));
136
#endif	/* _UDPLITE_H */