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