1
2
3
4
5
6
7
8
9
10
11#define pr_fmt(fmt) "UDPLite: " fmt
12
13#include <linux/export.h>
14#include <linux/proc_fs.h>
15#include "udp_impl.h"
16
17struct udp_table udplite_table __read_mostly;
18EXPORT_SYMBOL(udplite_table);
19
20static int udplite_rcv(struct sk_buff *skb)
21{
22 return __udp4_lib_rcv(skb, &udplite_table, IPPROTO_UDPLITE);
23}
24
25static int udplite_err(struct sk_buff *skb, u32 info)
26{
27 return __udp4_lib_err(skb, info, &udplite_table);
28}
29
30static const struct net_protocol udplite_protocol = {
31 .handler = udplite_rcv,
32 .err_handler = udplite_err,
33 .no_policy = 1,
34};
35
36struct proto udplite_prot = {
37 .name = "UDP-Lite",
38 .owner = THIS_MODULE,
39 .close = udp_lib_close,
40 .connect = ip4_datagram_connect,
41 .disconnect = udp_disconnect,
42 .ioctl = udp_ioctl,
43 .init = udplite_sk_init,
44 .destroy = udp_destroy_sock,
45 .setsockopt = udp_setsockopt,
46 .getsockopt = udp_getsockopt,
47 .sendmsg = udp_sendmsg,
48 .recvmsg = udp_recvmsg,
49 .sendpage = udp_sendpage,
50 .hash = udp_lib_hash,
51 .unhash = udp_lib_unhash,
52 .rehash = udp_v4_rehash,
53 .get_port = udp_v4_get_port,
54 .memory_allocated = &udp_memory_allocated,
55 .sysctl_mem = sysctl_udp_mem,
56 .obj_size = sizeof(struct udp_sock),
57 .h.udp_table = &udplite_table,
58};
59EXPORT_SYMBOL(udplite_prot);
60
61static struct inet_protosw udplite4_protosw = {
62 .type = SOCK_DGRAM,
63 .protocol = IPPROTO_UDPLITE,
64 .prot = &udplite_prot,
65 .ops = &inet_dgram_ops,
66 .flags = INET_PROTOSW_PERMANENT,
67};
68
69#ifdef CONFIG_PROC_FS
70static struct udp_seq_afinfo udplite4_seq_afinfo = {
71 .family = AF_INET,
72 .udp_table = &udplite_table,
73};
74
75static int __net_init udplite4_proc_init_net(struct net *net)
76{
77 if (!proc_create_net_data("udplite", 0444, net->proc_net, &udp_seq_ops,
78 sizeof(struct udp_iter_state), &udplite4_seq_afinfo))
79 return -ENOMEM;
80 return 0;
81}
82
83static void __net_exit udplite4_proc_exit_net(struct net *net)
84{
85 remove_proc_entry("udplite", net->proc_net);
86}
87
88static struct pernet_operations udplite4_net_ops = {
89 .init = udplite4_proc_init_net,
90 .exit = udplite4_proc_exit_net,
91};
92
93static __init int udplite4_proc_init(void)
94{
95 return register_pernet_subsys(&udplite4_net_ops);
96}
97#else
98static inline int udplite4_proc_init(void)
99{
100 return 0;
101}
102#endif
103
104void __init udplite4_register(void)
105{
106 udp_table_init(&udplite_table, "UDP-Lite");
107 if (proto_register(&udplite_prot, 1))
108 goto out_register_err;
109
110 if (inet_add_protocol(&udplite_protocol, IPPROTO_UDPLITE) < 0)
111 goto out_unregister_proto;
112
113 inet_register_protosw(&udplite4_protosw);
114
115 if (udplite4_proc_init())
116 pr_err("%s: Cannot register /proc!\n", __func__);
117 return;
118
119out_unregister_proto:
120 proto_unregister(&udplite_prot);
121out_register_err:
122 pr_crit("%s: Cannot add UDP-Lite protocol\n", __func__);
123}
124
