Blame view
net/core/sock_diag.c
4.7 KB
8ef874bfc sock_diag: Move t... |
1 2 3 4 5 6 |
#include <linux/mutex.h> #include <linux/socket.h> #include <linux/skbuff.h> #include <net/netlink.h> #include <net/net_namespace.h> #include <linux/module.h> |
5d2e5f274 sock_diag: Introd... |
7 8 |
#include <linux/rtnetlink.h> #include <net/sock.h> |
8ef874bfc sock_diag: Move t... |
9 10 11 12 13 14 15 |
#include <linux/inet_diag.h> #include <linux/sock_diag.h> static struct sock_diag_handler *sock_diag_handlers[AF_MAX]; static int (*inet_rcv_compat)(struct sk_buff *skb, struct nlmsghdr *nlh); static DEFINE_MUTEX(sock_diag_table_mutex); |
f65c1b534 sock_diag: Genera... |
16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 |
int sock_diag_check_cookie(void *sk, __u32 *cookie) { if ((cookie[0] != INET_DIAG_NOCOOKIE || cookie[1] != INET_DIAG_NOCOOKIE) && ((u32)(unsigned long)sk != cookie[0] || (u32)((((unsigned long)sk) >> 31) >> 1) != cookie[1])) return -ESTALE; else return 0; } EXPORT_SYMBOL_GPL(sock_diag_check_cookie); void sock_diag_save_cookie(void *sk, __u32 *cookie) { cookie[0] = (u32)(unsigned long)sk; cookie[1] = (u32)(((unsigned long)sk >> 31) >> 1); } EXPORT_SYMBOL_GPL(sock_diag_save_cookie); |
5d2e5f274 sock_diag: Introd... |
34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 |
int sock_diag_put_meminfo(struct sock *sk, struct sk_buff *skb, int attrtype) { __u32 *mem; mem = RTA_DATA(__RTA_PUT(skb, attrtype, SK_MEMINFO_VARS * sizeof(__u32))); mem[SK_MEMINFO_RMEM_ALLOC] = sk_rmem_alloc_get(sk); mem[SK_MEMINFO_RCVBUF] = sk->sk_rcvbuf; mem[SK_MEMINFO_WMEM_ALLOC] = sk_wmem_alloc_get(sk); mem[SK_MEMINFO_SNDBUF] = sk->sk_sndbuf; mem[SK_MEMINFO_FWD_ALLOC] = sk->sk_forward_alloc; mem[SK_MEMINFO_WMEM_QUEUED] = sk->sk_wmem_queued; mem[SK_MEMINFO_OPTMEM] = atomic_read(&sk->sk_omem_alloc); return 0; rtattr_failure: return -EMSGSIZE; } EXPORT_SYMBOL_GPL(sock_diag_put_meminfo); |
8ef874bfc sock_diag: Move t... |
54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 |
void sock_diag_register_inet_compat(int (*fn)(struct sk_buff *skb, struct nlmsghdr *nlh)) { mutex_lock(&sock_diag_table_mutex); inet_rcv_compat = fn; mutex_unlock(&sock_diag_table_mutex); } EXPORT_SYMBOL_GPL(sock_diag_register_inet_compat); void sock_diag_unregister_inet_compat(int (*fn)(struct sk_buff *skb, struct nlmsghdr *nlh)) { mutex_lock(&sock_diag_table_mutex); inet_rcv_compat = NULL; mutex_unlock(&sock_diag_table_mutex); } EXPORT_SYMBOL_GPL(sock_diag_unregister_inet_compat); int sock_diag_register(struct sock_diag_handler *hndl) { int err = 0; |
6f8e4ad0e sock_diag: off by... |
73 |
if (hndl->family >= AF_MAX) |
8ef874bfc sock_diag: Move t... |
74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 |
return -EINVAL; mutex_lock(&sock_diag_table_mutex); if (sock_diag_handlers[hndl->family]) err = -EBUSY; else sock_diag_handlers[hndl->family] = hndl; mutex_unlock(&sock_diag_table_mutex); return err; } EXPORT_SYMBOL_GPL(sock_diag_register); void sock_diag_unregister(struct sock_diag_handler *hnld) { int family = hnld->family; |
6f8e4ad0e sock_diag: off by... |
90 |
if (family >= AF_MAX) |
8ef874bfc sock_diag: Move t... |
91 92 93 94 95 96 97 98 99 100 101 102 103 |
return; mutex_lock(&sock_diag_table_mutex); BUG_ON(sock_diag_handlers[family] != hnld); sock_diag_handlers[family] = NULL; mutex_unlock(&sock_diag_table_mutex); } EXPORT_SYMBOL_GPL(sock_diag_unregister); static inline struct sock_diag_handler *sock_diag_lock_handler(int family) { if (sock_diag_handlers[family] == NULL) request_module("net-pf-%d-proto-%d-type-%d", PF_NETLINK, |
aec8dc62f sock_diag: Fix mo... |
104 |
NETLINK_SOCK_DIAG, family); |
8ef874bfc sock_diag: Move t... |
105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 |
mutex_lock(&sock_diag_table_mutex); return sock_diag_handlers[family]; } static inline void sock_diag_unlock_handler(struct sock_diag_handler *h) { mutex_unlock(&sock_diag_table_mutex); } static int __sock_diag_rcv_msg(struct sk_buff *skb, struct nlmsghdr *nlh) { int err; struct sock_diag_req *req = NLMSG_DATA(nlh); struct sock_diag_handler *hndl; if (nlmsg_len(nlh) < sizeof(*req)) return -EINVAL; hndl = sock_diag_lock_handler(req->sdiag_family); if (hndl == NULL) err = -ENOENT; else err = hndl->dump(skb, nlh); sock_diag_unlock_handler(hndl); return err; } static int sock_diag_rcv_msg(struct sk_buff *skb, struct nlmsghdr *nlh) { int ret; switch (nlh->nlmsg_type) { case TCPDIAG_GETSOCK: case DCCPDIAG_GETSOCK: if (inet_rcv_compat == NULL) request_module("net-pf-%d-proto-%d-type-%d", PF_NETLINK, |
aec8dc62f sock_diag: Fix mo... |
143 |
NETLINK_SOCK_DIAG, AF_INET); |
8ef874bfc sock_diag: Move t... |
144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 |
mutex_lock(&sock_diag_table_mutex); if (inet_rcv_compat != NULL) ret = inet_rcv_compat(skb, nlh); else ret = -EOPNOTSUPP; mutex_unlock(&sock_diag_table_mutex); return ret; case SOCK_DIAG_BY_FAMILY: return __sock_diag_rcv_msg(skb, nlh); default: return -EINVAL; } } static DEFINE_MUTEX(sock_diag_mutex); static void sock_diag_rcv(struct sk_buff *skb) { mutex_lock(&sock_diag_mutex); netlink_rcv_skb(skb, &sock_diag_rcv_msg); mutex_unlock(&sock_diag_mutex); } struct sock *sock_diag_nlsk; EXPORT_SYMBOL_GPL(sock_diag_nlsk); static int __init sock_diag_init(void) { sock_diag_nlsk = netlink_kernel_create(&init_net, NETLINK_SOCK_DIAG, 0, sock_diag_rcv, NULL, THIS_MODULE); return sock_diag_nlsk == NULL ? -ENOMEM : 0; } static void __exit sock_diag_exit(void) { netlink_kernel_release(sock_diag_nlsk); } module_init(sock_diag_init); module_exit(sock_diag_exit); MODULE_LICENSE("GPL"); MODULE_ALIAS_NET_PF_PROTO(PF_NETLINK, NETLINK_SOCK_DIAG); |