net/ipv4/udplite.c

   1 /*
   2  *  UDPLITE     An implementation of the UDP-Lite protocol (RFC 3828).
   3  *
   4  *  Authors:    Gerrit Renker       <gerrit@erg.abdn.ac.uk>
   5  *
   6  *  Changes:
   7  *  Fixes:
   8  *              This program is free software; you can redistribute it and/or
   9  *              modify it under the terms of the GNU General Public License
  10  *              as published by the Free Software Foundation; either version
  11  *              2 of the License, or (at your option) any later version.
  12  */
  13
  14 #define pr_fmt(fmt) "UDPLite: " fmt
  15
  16 #include <linux/export.h>
  17 #include "udp_impl.h"
  18
  19 struct udp_table        udplite_table __read_mostly;
  20 EXPORT_SYMBOL(udplite_table);
  21
  22 /* Designate sk as UDP-Lite socket */
  23 static int udplite_sk_init(struct sock *sk)
  24 {
  25         udp_init_sock(sk);
  26         udp_sk(sk)->pcflag = UDPLITE_BIT;
  27         return 0;
  28 }
  29
  30 static int udplite_rcv(struct sk_buff *skb)
  31 {
  32         return __udp4_lib_rcv(skb, &udplite_table, IPPROTO_UDPLITE);
  33 }
  34
  35 static void udplite_err(struct sk_buff *skb, u32 info)
  36 {
  37         __udp4_lib_err(skb, info, &udplite_table);
  38 }
  39
  40 static const struct net_protocol udplite_protocol = {
  41         .handler        = udplite_rcv,
  42         .err_handler    = udplite_err,
  43         .no_policy      = 1,
  44         .netns_ok       = 1,
  45 };
  46
  47 struct proto    udplite_prot = {
  48         .name              = "UDP-Lite",
  49         .owner             = THIS_MODULE,
  50         .close             = udp_lib_close,
  51         .connect           = ip4_datagram_connect,
  52         .disconnect        = udp_disconnect,
  53         .ioctl             = udp_ioctl,
  54         .init              = udplite_sk_init,
  55         .destroy           = udp_destroy_sock,
  56         .setsockopt        = udp_setsockopt,
  57         .getsockopt        = udp_getsockopt,
  58         .sendmsg           = udp_sendmsg,
  59         .recvmsg           = udp_recvmsg,
  60         .sendpage          = udp_sendpage,
  61         .hash              = udp_lib_hash,
  62         .unhash            = udp_lib_unhash,
  63         .get_port          = udp_v4_get_port,
  64         .memory_allocated  = &udp_memory_allocated,
  65         .sysctl_mem        = sysctl_udp_mem,
  66         .obj_size          = sizeof(struct udp_sock),
  67         .h.udp_table       = &udplite_table,
  68 #ifdef CONFIG_COMPAT
  69         .compat_setsockopt = compat_udp_setsockopt,
  70         .compat_getsockopt = compat_udp_getsockopt,
  71 #endif
  72 };
  73 EXPORT_SYMBOL(udplite_prot);
  74
  75 static struct inet_protosw udplite4_protosw = {
  76         .type           =  SOCK_DGRAM,
  77         .protocol       =  IPPROTO_UDPLITE,
  78         .prot           =  &udplite_prot,
  79         .ops            =  &inet_dgram_ops,
  80         .flags          =  INET_PROTOSW_PERMANENT,
  81 };
  82
  83 #ifdef CONFIG_PROC_FS
  84
  85 static const struct file_operations udplite_afinfo_seq_fops = {
  86         .owner    = THIS_MODULE,
  87         .open     = udp_seq_open,
  88         .read     = seq_read,
  89         .llseek   = seq_lseek,
  90         .release  = seq_release_net
  91 };
  92
  93 static struct udp_seq_afinfo udplite4_seq_afinfo = {
  94         .name           = "udplite",
  95         .family         = AF_INET,
  96         .udp_table      = &udplite_table,
  97         .seq_fops       = &udplite_afinfo_seq_fops,
  98         .seq_ops        = {
  99                 .show           = udp4_seq_show,
 100         },
 101 };
 102
 103 static int __net_init udplite4_proc_init_net(struct net *net)
 104 {
 105         return udp_proc_register(net, &udplite4_seq_afinfo);
 106 }
 107
 108 static void __net_exit udplite4_proc_exit_net(struct net *net)
 109 {
 110         udp_proc_unregister(net, &udplite4_seq_afinfo);
 111 }
 112
 113 static struct pernet_operations udplite4_net_ops = {
 114         .init = udplite4_proc_init_net,
 115         .exit = udplite4_proc_exit_net,
 116 };
 117
 118 static __init int udplite4_proc_init(void)
 119 {
 120         return register_pernet_subsys(&udplite4_net_ops);
 121 }
 122 #else
 123 static inline int udplite4_proc_init(void)
 124 {
 125         return 0;
 126 }
 127 #endif
 128
 129 void __init udplite4_register(void)
 130 {
 131         udp_table_init(&udplite_table, "UDP-Lite");
 132         if (proto_register(&udplite_prot, 1))
 133                 goto out_register_err;
 134
 135         if (inet_add_protocol(&udplite_protocol, IPPROTO_UDPLITE) < 0)
 136                 goto out_unregister_proto;
 137
 138         inet_register_protosw(&udplite4_protosw);
 139
 140         if (udplite4_proc_init())
 141                 pr_err("%s: Cannot register /proc!\n", __func__);
 142         return;
 143
 144 out_unregister_proto:
 145         proto_unregister(&udplite_prot);
 146 out_register_err:
 147         pr_crit("%s: Cannot add UDP-Lite protocol\n", __func__);
 148 }