1 // SPDX-License-Identifier: GPL-2.0-or-later
2 /*
3  *  UDPLITEv6   An implementation of the UDP-Lite protocol over IPv6.
4  *              See also net/ipv4/udplite.c
5  *
6  *  Authors:    Gerrit Renker       <gerrit@erg.abdn.ac.uk>
7  *
8  *  Changes:
9  *  Fixes:
10  */
11 #include <linux/export.h>
12 #include <linux/proc_fs.h>
13 #include "udp_impl.h"
14 
udplitev6_sk_init(struct sock * sk)15 static int udplitev6_sk_init(struct sock *sk)
16 {
17 	udpv6_init_sock(sk);
18 	udp_sk(sk)->pcflag = UDPLITE_BIT;
19 	return 0;
20 }
21 
udplitev6_rcv(struct sk_buff * skb)22 static int udplitev6_rcv(struct sk_buff *skb)
23 {
24 	return __udp6_lib_rcv(skb, &udplite_table, IPPROTO_UDPLITE);
25 }
26 
udplitev6_err(struct sk_buff * skb,struct inet6_skb_parm * opt,u8 type,u8 code,int offset,__be32 info)27 static int udplitev6_err(struct sk_buff *skb,
28 			  struct inet6_skb_parm *opt,
29 			  u8 type, u8 code, int offset, __be32 info)
30 {
31 	return __udp6_lib_err(skb, opt, type, code, offset, info,
32 			      &udplite_table);
33 }
34 
35 static const struct inet6_protocol udplitev6_protocol = {
36 	.handler	=	udplitev6_rcv,
37 	.err_handler	=	udplitev6_err,
38 	.flags		=	INET6_PROTO_NOPOLICY|INET6_PROTO_FINAL,
39 };
40 
41 struct proto udplitev6_prot = {
42 	.name		   = "UDPLITEv6",
43 	.owner		   = THIS_MODULE,
44 	.close		   = udp_lib_close,
45 	.connect	   = ip6_datagram_connect,
46 	.disconnect	   = udp_disconnect,
47 	.ioctl		   = udp_ioctl,
48 	.init		   = udplitev6_sk_init,
49 	.destroy	   = udpv6_destroy_sock,
50 	.setsockopt	   = udpv6_setsockopt,
51 	.getsockopt	   = udpv6_getsockopt,
52 	.sendmsg	   = udpv6_sendmsg,
53 	.recvmsg	   = udpv6_recvmsg,
54 	.hash		   = udp_lib_hash,
55 	.unhash		   = udp_lib_unhash,
56 	.rehash		   = udp_v6_rehash,
57 	.get_port	   = udp_v6_get_port,
58 
59 	.memory_allocated  = &udp_memory_allocated,
60 	.per_cpu_fw_alloc  = &udp_memory_per_cpu_fw_alloc,
61 
62 	.sysctl_mem	   = sysctl_udp_mem,
63 	.obj_size	   = sizeof(struct udp6_sock),
64 	.h.udp_table	   = &udplite_table,
65 };
66 
67 static struct inet_protosw udplite6_protosw = {
68 	.type		= SOCK_DGRAM,
69 	.protocol	= IPPROTO_UDPLITE,
70 	.prot		= &udplitev6_prot,
71 	.ops		= &inet6_dgram_ops,
72 	.flags		= INET_PROTOSW_PERMANENT,
73 };
74 
udplitev6_init(void)75 int __init udplitev6_init(void)
76 {
77 	int ret;
78 
79 	ret = inet6_add_protocol(&udplitev6_protocol, IPPROTO_UDPLITE);
80 	if (ret)
81 		goto out;
82 
83 	ret = inet6_register_protosw(&udplite6_protosw);
84 	if (ret)
85 		goto out_udplitev6_protocol;
86 out:
87 	return ret;
88 
89 out_udplitev6_protocol:
90 	inet6_del_protocol(&udplitev6_protocol, IPPROTO_UDPLITE);
91 	goto out;
92 }
93 
udplitev6_exit(void)94 void udplitev6_exit(void)
95 {
96 	inet6_unregister_protosw(&udplite6_protosw);
97 	inet6_del_protocol(&udplitev6_protocol, IPPROTO_UDPLITE);
98 }
99 
100 #ifdef CONFIG_PROC_FS
101 static struct udp_seq_afinfo udplite6_seq_afinfo = {
102 	.family		= AF_INET6,
103 	.udp_table	= &udplite_table,
104 };
105 
udplite6_proc_init_net(struct net * net)106 static int __net_init udplite6_proc_init_net(struct net *net)
107 {
108 	if (!proc_create_net_data("udplite6", 0444, net->proc_net,
109 			&udp6_seq_ops, sizeof(struct udp_iter_state),
110 			&udplite6_seq_afinfo))
111 		return -ENOMEM;
112 	return 0;
113 }
114 
udplite6_proc_exit_net(struct net * net)115 static void __net_exit udplite6_proc_exit_net(struct net *net)
116 {
117 	remove_proc_entry("udplite6", net->proc_net);
118 }
119 
120 static struct pernet_operations udplite6_net_ops = {
121 	.init = udplite6_proc_init_net,
122 	.exit = udplite6_proc_exit_net,
123 };
124 
udplite6_proc_init(void)125 int __init udplite6_proc_init(void)
126 {
127 	return register_pernet_subsys(&udplite6_net_ops);
128 }
129 
udplite6_proc_exit(void)130 void udplite6_proc_exit(void)
131 {
132 	unregister_pernet_subsys(&udplite6_net_ops);
133 }
134 #endif
135