Harald Welte | f9e815b | 2005-08-09 19:30:24 -0700 | [diff] [blame] | 1 | /* Netfilter messages via netlink socket. Allows for user space |
| 2 | * protocol helpers and general trouble making from userspace. |
| 3 | * |
| 4 | * (C) 2001 by Jay Schulist <jschlst@samba.org>, |
| 5 | * (C) 2002-2005 by Harald Welte <laforge@gnumonks.org> |
Pablo Neira Ayuso | 67ca396 | 2007-03-14 16:40:38 -0700 | [diff] [blame] | 6 | * (C) 2005,2007 by Pablo Neira Ayuso <pablo@netfilter.org> |
Harald Welte | f9e815b | 2005-08-09 19:30:24 -0700 | [diff] [blame] | 7 | * |
| 8 | * Initial netfilter messages via netlink development funded and |
| 9 | * generally made possible by Network Robots, Inc. (www.networkrobots.com) |
| 10 | * |
| 11 | * Further development of this code funded by Astaro AG (http://www.astaro.com) |
| 12 | * |
| 13 | * This software may be used and distributed according to the terms |
| 14 | * of the GNU General Public License, incorporated herein by reference. |
| 15 | */ |
| 16 | |
Harald Welte | f9e815b | 2005-08-09 19:30:24 -0700 | [diff] [blame] | 17 | #include <linux/module.h> |
| 18 | #include <linux/types.h> |
| 19 | #include <linux/socket.h> |
| 20 | #include <linux/kernel.h> |
| 21 | #include <linux/major.h> |
Harald Welte | f9e815b | 2005-08-09 19:30:24 -0700 | [diff] [blame] | 22 | #include <linux/timer.h> |
| 23 | #include <linux/string.h> |
| 24 | #include <linux/sockios.h> |
| 25 | #include <linux/net.h> |
| 26 | #include <linux/fcntl.h> |
| 27 | #include <linux/skbuff.h> |
| 28 | #include <asm/uaccess.h> |
| 29 | #include <asm/system.h> |
| 30 | #include <net/sock.h> |
Patrick McHardy | a3c5029 | 2007-03-14 16:39:25 -0700 | [diff] [blame] | 31 | #include <net/netlink.h> |
Harald Welte | f9e815b | 2005-08-09 19:30:24 -0700 | [diff] [blame] | 32 | #include <linux/init.h> |
Harald Welte | f9e815b | 2005-08-09 19:30:24 -0700 | [diff] [blame] | 33 | |
Harald Welte | f9e815b | 2005-08-09 19:30:24 -0700 | [diff] [blame] | 34 | #include <linux/netlink.h> |
| 35 | #include <linux/netfilter/nfnetlink.h> |
| 36 | |
| 37 | MODULE_LICENSE("GPL"); |
Harald Welte | 4fdb3bb | 2005-08-09 19:40:55 -0700 | [diff] [blame] | 38 | MODULE_AUTHOR("Harald Welte <laforge@netfilter.org>"); |
| 39 | MODULE_ALIAS_NET_PF_PROTO(PF_NETLINK, NETLINK_NETFILTER); |
Harald Welte | f9e815b | 2005-08-09 19:30:24 -0700 | [diff] [blame] | 40 | |
| 41 | static char __initdata nfversion[] = "0.30"; |
| 42 | |
Harald Welte | f9e815b | 2005-08-09 19:30:24 -0700 | [diff] [blame] | 43 | static struct sock *nfnl = NULL; |
Patrick McHardy | 7c8d4cb | 2007-09-28 14:15:45 -0700 | [diff] [blame] | 44 | static const struct nfnetlink_subsystem *subsys_table[NFNL_SUBSYS_COUNT]; |
Patrick McHardy | a3c5029 | 2007-03-14 16:39:25 -0700 | [diff] [blame] | 45 | static DEFINE_MUTEX(nfnl_mutex); |
Harald Welte | f9e815b | 2005-08-09 19:30:24 -0700 | [diff] [blame] | 46 | |
Pablo Neira Ayuso | e6a7d3c | 2008-10-14 11:58:31 -0700 | [diff] [blame] | 47 | void nfnl_lock(void) |
Harald Welte | f9e815b | 2005-08-09 19:30:24 -0700 | [diff] [blame] | 48 | { |
Patrick McHardy | a3c5029 | 2007-03-14 16:39:25 -0700 | [diff] [blame] | 49 | mutex_lock(&nfnl_mutex); |
Harald Welte | f9e815b | 2005-08-09 19:30:24 -0700 | [diff] [blame] | 50 | } |
Pablo Neira Ayuso | e6a7d3c | 2008-10-14 11:58:31 -0700 | [diff] [blame] | 51 | EXPORT_SYMBOL_GPL(nfnl_lock); |
Harald Welte | f9e815b | 2005-08-09 19:30:24 -0700 | [diff] [blame] | 52 | |
Pablo Neira Ayuso | e6a7d3c | 2008-10-14 11:58:31 -0700 | [diff] [blame] | 53 | void nfnl_unlock(void) |
Patrick McHardy | a3c5029 | 2007-03-14 16:39:25 -0700 | [diff] [blame] | 54 | { |
| 55 | mutex_unlock(&nfnl_mutex); |
| 56 | } |
Pablo Neira Ayuso | e6a7d3c | 2008-10-14 11:58:31 -0700 | [diff] [blame] | 57 | EXPORT_SYMBOL_GPL(nfnl_unlock); |
Patrick McHardy | a3c5029 | 2007-03-14 16:39:25 -0700 | [diff] [blame] | 58 | |
Patrick McHardy | 7c8d4cb | 2007-09-28 14:15:45 -0700 | [diff] [blame] | 59 | int nfnetlink_subsys_register(const struct nfnetlink_subsystem *n) |
Harald Welte | f9e815b | 2005-08-09 19:30:24 -0700 | [diff] [blame] | 60 | { |
Harald Welte | f9e815b | 2005-08-09 19:30:24 -0700 | [diff] [blame] | 61 | nfnl_lock(); |
Harald Welte | 0ab43f8 | 2005-08-09 19:43:44 -0700 | [diff] [blame] | 62 | if (subsys_table[n->subsys_id]) { |
| 63 | nfnl_unlock(); |
| 64 | return -EBUSY; |
| 65 | } |
Harald Welte | f9e815b | 2005-08-09 19:30:24 -0700 | [diff] [blame] | 66 | subsys_table[n->subsys_id] = n; |
| 67 | nfnl_unlock(); |
| 68 | |
| 69 | return 0; |
| 70 | } |
Pablo Neira Ayuso | f4bc177 | 2007-03-14 16:42:11 -0700 | [diff] [blame] | 71 | EXPORT_SYMBOL_GPL(nfnetlink_subsys_register); |
Harald Welte | f9e815b | 2005-08-09 19:30:24 -0700 | [diff] [blame] | 72 | |
Patrick McHardy | 7c8d4cb | 2007-09-28 14:15:45 -0700 | [diff] [blame] | 73 | int nfnetlink_subsys_unregister(const struct nfnetlink_subsystem *n) |
Harald Welte | f9e815b | 2005-08-09 19:30:24 -0700 | [diff] [blame] | 74 | { |
Harald Welte | f9e815b | 2005-08-09 19:30:24 -0700 | [diff] [blame] | 75 | nfnl_lock(); |
| 76 | subsys_table[n->subsys_id] = NULL; |
| 77 | nfnl_unlock(); |
| 78 | |
| 79 | return 0; |
| 80 | } |
Pablo Neira Ayuso | f4bc177 | 2007-03-14 16:42:11 -0700 | [diff] [blame] | 81 | EXPORT_SYMBOL_GPL(nfnetlink_subsys_unregister); |
Harald Welte | f9e815b | 2005-08-09 19:30:24 -0700 | [diff] [blame] | 82 | |
Patrick McHardy | 7c8d4cb | 2007-09-28 14:15:45 -0700 | [diff] [blame] | 83 | static inline const struct nfnetlink_subsystem *nfnetlink_get_subsys(u_int16_t type) |
Harald Welte | f9e815b | 2005-08-09 19:30:24 -0700 | [diff] [blame] | 84 | { |
| 85 | u_int8_t subsys_id = NFNL_SUBSYS_ID(type); |
| 86 | |
Pablo Neira Ayuso | ac0f1d9 | 2007-03-14 16:41:28 -0700 | [diff] [blame] | 87 | if (subsys_id >= NFNL_SUBSYS_COUNT) |
Harald Welte | f9e815b | 2005-08-09 19:30:24 -0700 | [diff] [blame] | 88 | return NULL; |
| 89 | |
| 90 | return subsys_table[subsys_id]; |
| 91 | } |
| 92 | |
Patrick McHardy | 7c8d4cb | 2007-09-28 14:15:45 -0700 | [diff] [blame] | 93 | static inline const struct nfnl_callback * |
| 94 | nfnetlink_find_client(u_int16_t type, const struct nfnetlink_subsystem *ss) |
Harald Welte | f9e815b | 2005-08-09 19:30:24 -0700 | [diff] [blame] | 95 | { |
| 96 | u_int8_t cb_id = NFNL_MSG_TYPE(type); |
YOSHIFUJI Hideaki | 601e68e | 2007-02-12 11:15:49 -0800 | [diff] [blame] | 97 | |
Pablo Neira Ayuso | 67ca396 | 2007-03-14 16:40:38 -0700 | [diff] [blame] | 98 | if (cb_id >= ss->cb_count) |
Harald Welte | f9e815b | 2005-08-09 19:30:24 -0700 | [diff] [blame] | 99 | return NULL; |
Harald Welte | f9e815b | 2005-08-09 19:30:24 -0700 | [diff] [blame] | 100 | |
| 101 | return &ss->cb[cb_id]; |
| 102 | } |
| 103 | |
Patrick McHardy | a242769 | 2006-03-20 18:03:59 -0800 | [diff] [blame] | 104 | int nfnetlink_has_listeners(unsigned int group) |
| 105 | { |
| 106 | return netlink_has_listeners(nfnl, group); |
| 107 | } |
| 108 | EXPORT_SYMBOL_GPL(nfnetlink_has_listeners); |
| 109 | |
Harald Welte | f9e815b | 2005-08-09 19:30:24 -0700 | [diff] [blame] | 110 | int nfnetlink_send(struct sk_buff *skb, u32 pid, unsigned group, int echo) |
| 111 | { |
Patrick McHardy | dd82185 | 2007-09-28 14:38:07 -0700 | [diff] [blame] | 112 | return nlmsg_notify(nfnl, skb, pid, group, echo, gfp_any()); |
Harald Welte | f9e815b | 2005-08-09 19:30:24 -0700 | [diff] [blame] | 113 | } |
Pablo Neira Ayuso | f4bc177 | 2007-03-14 16:42:11 -0700 | [diff] [blame] | 114 | EXPORT_SYMBOL_GPL(nfnetlink_send); |
Harald Welte | f9e815b | 2005-08-09 19:30:24 -0700 | [diff] [blame] | 115 | |
| 116 | int nfnetlink_unicast(struct sk_buff *skb, u_int32_t pid, int flags) |
| 117 | { |
| 118 | return netlink_unicast(nfnl, skb, pid, flags); |
| 119 | } |
Pablo Neira Ayuso | f4bc177 | 2007-03-14 16:42:11 -0700 | [diff] [blame] | 120 | EXPORT_SYMBOL_GPL(nfnetlink_unicast); |
Harald Welte | f9e815b | 2005-08-09 19:30:24 -0700 | [diff] [blame] | 121 | |
| 122 | /* Process one complete nfnetlink message. */ |
Thomas Graf | 1d00a4e | 2007-03-22 23:30:12 -0700 | [diff] [blame] | 123 | static int nfnetlink_rcv_msg(struct sk_buff *skb, struct nlmsghdr *nlh) |
Harald Welte | f9e815b | 2005-08-09 19:30:24 -0700 | [diff] [blame] | 124 | { |
Patrick McHardy | 7c8d4cb | 2007-09-28 14:15:45 -0700 | [diff] [blame] | 125 | const struct nfnl_callback *nc; |
| 126 | const struct nfnetlink_subsystem *ss; |
Thomas Graf | 1d00a4e | 2007-03-22 23:30:12 -0700 | [diff] [blame] | 127 | int type, err; |
Harald Welte | f9e815b | 2005-08-09 19:30:24 -0700 | [diff] [blame] | 128 | |
Thomas Graf | 1d00a4e | 2007-03-22 23:30:12 -0700 | [diff] [blame] | 129 | if (security_netlink_recv(skb, CAP_NET_ADMIN)) |
| 130 | return -EPERM; |
Harald Welte | 37d2e7a | 2005-11-14 15:24:59 -0800 | [diff] [blame] | 131 | |
Harald Welte | f9e815b | 2005-08-09 19:30:24 -0700 | [diff] [blame] | 132 | /* All the messages must at least contain nfgenmsg */ |
Pablo Neira Ayuso | 67ca396 | 2007-03-14 16:40:38 -0700 | [diff] [blame] | 133 | if (nlh->nlmsg_len < NLMSG_SPACE(sizeof(struct nfgenmsg))) |
Harald Welte | f9e815b | 2005-08-09 19:30:24 -0700 | [diff] [blame] | 134 | return 0; |
Harald Welte | f9e815b | 2005-08-09 19:30:24 -0700 | [diff] [blame] | 135 | |
| 136 | type = nlh->nlmsg_type; |
Pablo Neira Ayuso | e6a7d3c | 2008-10-14 11:58:31 -0700 | [diff] [blame] | 137 | replay: |
Harald Welte | f9e815b | 2005-08-09 19:30:24 -0700 | [diff] [blame] | 138 | ss = nfnetlink_get_subsys(type); |
Harald Welte | 0ab43f8 | 2005-08-09 19:43:44 -0700 | [diff] [blame] | 139 | if (!ss) { |
Johannes Berg | 95a5afc | 2008-10-16 15:24:51 -0700 | [diff] [blame] | 140 | #ifdef CONFIG_MODULES |
Denis V. Lunev | 3b71535 | 2007-10-10 21:13:32 -0700 | [diff] [blame] | 141 | nfnl_unlock(); |
Harald Welte | 37d2e7a | 2005-11-14 15:24:59 -0800 | [diff] [blame] | 142 | request_module("nfnetlink-subsys-%d", NFNL_SUBSYS_ID(type)); |
Patrick McHardy | a3c5029 | 2007-03-14 16:39:25 -0700 | [diff] [blame] | 143 | nfnl_lock(); |
Harald Welte | 37d2e7a | 2005-11-14 15:24:59 -0800 | [diff] [blame] | 144 | ss = nfnetlink_get_subsys(type); |
Harald Welte | 0ab43f8 | 2005-08-09 19:43:44 -0700 | [diff] [blame] | 145 | if (!ss) |
| 146 | #endif |
Thomas Graf | 1d00a4e | 2007-03-22 23:30:12 -0700 | [diff] [blame] | 147 | return -EINVAL; |
Harald Welte | 0ab43f8 | 2005-08-09 19:43:44 -0700 | [diff] [blame] | 148 | } |
Harald Welte | f9e815b | 2005-08-09 19:30:24 -0700 | [diff] [blame] | 149 | |
| 150 | nc = nfnetlink_find_client(type, ss); |
Pablo Neira Ayuso | 67ca396 | 2007-03-14 16:40:38 -0700 | [diff] [blame] | 151 | if (!nc) |
Thomas Graf | 1d00a4e | 2007-03-22 23:30:12 -0700 | [diff] [blame] | 152 | return -EINVAL; |
Harald Welte | f9e815b | 2005-08-09 19:30:24 -0700 | [diff] [blame] | 153 | |
Harald Welte | f9e815b | 2005-08-09 19:30:24 -0700 | [diff] [blame] | 154 | { |
Patrick McHardy | e373057 | 2007-09-28 14:38:52 -0700 | [diff] [blame] | 155 | int min_len = NLMSG_SPACE(sizeof(struct nfgenmsg)); |
| 156 | u_int8_t cb_id = NFNL_MSG_TYPE(nlh->nlmsg_type); |
| 157 | u_int16_t attr_count = ss->cb[cb_id].attr_count; |
Patrick McHardy | df6fb86 | 2007-09-28 14:37:03 -0700 | [diff] [blame] | 158 | struct nlattr *cda[attr_count+1]; |
Harald Welte | f9e815b | 2005-08-09 19:30:24 -0700 | [diff] [blame] | 159 | |
Patrick McHardy | e373057 | 2007-09-28 14:38:52 -0700 | [diff] [blame] | 160 | if (likely(nlh->nlmsg_len >= min_len)) { |
| 161 | struct nlattr *attr = (void *)nlh + NLMSG_ALIGN(min_len); |
| 162 | int attrlen = nlh->nlmsg_len - NLMSG_ALIGN(min_len); |
YOSHIFUJI Hideaki | 601e68e | 2007-02-12 11:15:49 -0800 | [diff] [blame] | 163 | |
Patrick McHardy | e373057 | 2007-09-28 14:38:52 -0700 | [diff] [blame] | 164 | err = nla_parse(cda, attr_count, attr, attrlen, |
| 165 | ss->cb[cb_id].policy); |
| 166 | if (err < 0) |
| 167 | return err; |
| 168 | } else |
| 169 | return -EINVAL; |
| 170 | |
Pablo Neira Ayuso | e6a7d3c | 2008-10-14 11:58:31 -0700 | [diff] [blame] | 171 | err = nc->call(nfnl, skb, nlh, cda); |
| 172 | if (err == -EAGAIN) |
| 173 | goto replay; |
| 174 | return err; |
Harald Welte | f9e815b | 2005-08-09 19:30:24 -0700 | [diff] [blame] | 175 | } |
Harald Welte | f9e815b | 2005-08-09 19:30:24 -0700 | [diff] [blame] | 176 | } |
| 177 | |
Denis V. Lunev | cd40b7d | 2007-10-10 21:15:29 -0700 | [diff] [blame] | 178 | static void nfnetlink_rcv(struct sk_buff *skb) |
Harald Welte | f9e815b | 2005-08-09 19:30:24 -0700 | [diff] [blame] | 179 | { |
Denis V. Lunev | cd40b7d | 2007-10-10 21:15:29 -0700 | [diff] [blame] | 180 | nfnl_lock(); |
| 181 | netlink_rcv_skb(skb, &nfnetlink_rcv_msg); |
| 182 | nfnl_unlock(); |
Harald Welte | f9e815b | 2005-08-09 19:30:24 -0700 | [diff] [blame] | 183 | } |
| 184 | |
Adrian Bunk | 395dde2 | 2005-09-05 18:06:45 -0700 | [diff] [blame] | 185 | static void __exit nfnetlink_exit(void) |
Harald Welte | f9e815b | 2005-08-09 19:30:24 -0700 | [diff] [blame] | 186 | { |
| 187 | printk("Removing netfilter NETLINK layer.\n"); |
Denis V. Lunev | b7c6ba6 | 2008-01-28 14:41:19 -0800 | [diff] [blame] | 188 | netlink_kernel_release(nfnl); |
Harald Welte | f9e815b | 2005-08-09 19:30:24 -0700 | [diff] [blame] | 189 | return; |
| 190 | } |
| 191 | |
Adrian Bunk | 395dde2 | 2005-09-05 18:06:45 -0700 | [diff] [blame] | 192 | static int __init nfnetlink_init(void) |
Harald Welte | f9e815b | 2005-08-09 19:30:24 -0700 | [diff] [blame] | 193 | { |
| 194 | printk("Netfilter messages via NETLINK v%s.\n", nfversion); |
| 195 | |
Eric W. Biederman | b4b5102 | 2007-09-12 13:05:38 +0200 | [diff] [blame] | 196 | nfnl = netlink_kernel_create(&init_net, NETLINK_NETFILTER, NFNLGRP_MAX, |
Patrick McHardy | af65bdf | 2007-04-20 14:14:21 -0700 | [diff] [blame] | 197 | nfnetlink_rcv, NULL, THIS_MODULE); |
Harald Welte | f9e815b | 2005-08-09 19:30:24 -0700 | [diff] [blame] | 198 | if (!nfnl) { |
| 199 | printk(KERN_ERR "cannot initialize nfnetlink!\n"); |
| 200 | return -1; |
| 201 | } |
| 202 | |
| 203 | return 0; |
| 204 | } |
| 205 | |
| 206 | module_init(nfnetlink_init); |
| 207 | module_exit(nfnetlink_exit); |