blob: b3afe189af61880464ef6562c568016509957293 [file] [log] [blame]
Linus Torvalds1da177e2005-04-16 15:20:36 -07001/*
2 * IPVS An implementation of the IP virtual server support for the
3 * LINUX operating system. IPVS is now implemented as a module
4 * over the NetFilter framework. IPVS can be used to build a
5 * high-performance and highly available server based on a
6 * cluster of servers.
7 *
Linus Torvalds1da177e2005-04-16 15:20:36 -07008 * Authors: Wensong Zhang <wensong@linuxvirtualserver.org>
9 * Peter Kese <peter.kese@ijs.si>
10 * Julian Anastasov <ja@ssi.bg>
11 *
12 * This program is free software; you can redistribute it and/or
13 * modify it under the terms of the GNU General Public License
14 * as published by the Free Software Foundation; either version
15 * 2 of the License, or (at your option) any later version.
16 *
17 * Changes:
18 *
19 */
20
Hannes Eder9aada7a2009-07-30 14:29:44 -070021#define KMSG_COMPONENT "IPVS"
22#define pr_fmt(fmt) KMSG_COMPONENT ": " fmt
23
Linus Torvalds1da177e2005-04-16 15:20:36 -070024#include <linux/module.h>
25#include <linux/init.h>
26#include <linux/types.h>
Randy Dunlap4fc268d2006-01-11 12:17:47 -080027#include <linux/capability.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070028#include <linux/fs.h>
29#include <linux/sysctl.h>
30#include <linux/proc_fs.h>
31#include <linux/workqueue.h>
32#include <linux/swap.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070033#include <linux/seq_file.h>
Tejun Heo5a0e3ad2010-03-24 17:04:11 +090034#include <linux/slab.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070035
36#include <linux/netfilter.h>
37#include <linux/netfilter_ipv4.h>
Ingo Molnar14cc3e22006-03-26 01:37:14 -080038#include <linux/mutex.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070039
Eric W. Biederman457c4cb2007-09-12 12:01:34 +020040#include <net/net_namespace.h>
Hans Schillstrom93304192011-01-03 14:44:51 +010041#include <linux/nsproxy.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070042#include <net/ip.h>
Vince Busam09571c72008-09-02 15:55:52 +020043#ifdef CONFIG_IP_VS_IPV6
44#include <net/ipv6.h>
45#include <net/ip6_route.h>
46#endif
Arnaldo Carvalho de Melo14c85022005-12-27 02:43:12 -020047#include <net/route.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070048#include <net/sock.h>
Julius Volz9a812192008-08-14 14:08:44 +020049#include <net/genetlink.h>
Linus Torvalds1da177e2005-04-16 15:20:36 -070050
51#include <asm/uaccess.h>
52
53#include <net/ip_vs.h>
54
55/* semaphore for IPVS sockopts. And, [gs]etsockopt may sleep. */
Ingo Molnar14cc3e22006-03-26 01:37:14 -080056static DEFINE_MUTEX(__ip_vs_mutex);
Linus Torvalds1da177e2005-04-16 15:20:36 -070057
58/* lock for service table */
59static DEFINE_RWLOCK(__ip_vs_svc_lock);
60
Linus Torvalds1da177e2005-04-16 15:20:36 -070061/* sysctl variables */
Linus Torvalds1da177e2005-04-16 15:20:36 -070062
63#ifdef CONFIG_IP_VS_DEBUG
64static int sysctl_ip_vs_debug_level = 0;
65
66int ip_vs_get_debug_level(void)
67{
68 return sysctl_ip_vs_debug_level;
69}
70#endif
71
Hans Schillstrom7a4f0762011-05-03 22:09:31 +020072
73/* Protos */
74static void __ip_vs_del_service(struct ip_vs_service *svc);
75
76
Vince Busam09571c72008-09-02 15:55:52 +020077#ifdef CONFIG_IP_VS_IPV6
78/* Taken from rt6_fill_node() in net/ipv6/route.c, is there a better way? */
Hans Schillstrom4a984802011-01-03 14:45:02 +010079static int __ip_vs_addr_is_local_v6(struct net *net,
80 const struct in6_addr *addr)
Vince Busam09571c72008-09-02 15:55:52 +020081{
82 struct rt6_info *rt;
David S. Miller4c9483b2011-03-12 16:22:43 -050083 struct flowi6 fl6 = {
84 .daddr = *addr,
Vince Busam09571c72008-09-02 15:55:52 +020085 };
86
David S. Miller4c9483b2011-03-12 16:22:43 -050087 rt = (struct rt6_info *)ip6_route_output(net, NULL, &fl6);
David S. Millerd1918542011-12-28 20:19:20 -050088 if (rt && rt->dst.dev && (rt->dst.dev->flags & IFF_LOOPBACK))
David S. Miller4c9483b2011-03-12 16:22:43 -050089 return 1;
Vince Busam09571c72008-09-02 15:55:52 +020090
91 return 0;
92}
93#endif
Simon Horman14e40542011-02-04 18:33:02 +090094
95#ifdef CONFIG_SYSCTL
Linus Torvalds1da177e2005-04-16 15:20:36 -070096/*
Julian Anastasovaf9debd2005-07-11 20:59:57 -070097 * update_defense_level is called from keventd and from sysctl,
98 * so it needs to protect itself from softirqs
Linus Torvalds1da177e2005-04-16 15:20:36 -070099 */
Hans Schillstrom93304192011-01-03 14:44:51 +0100100static void update_defense_level(struct netns_ipvs *ipvs)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700101{
102 struct sysinfo i;
103 static int old_secure_tcp = 0;
104 int availmem;
105 int nomem;
106 int to_change = -1;
107
108 /* we only count free and buffered memory (in pages) */
109 si_meminfo(&i);
110 availmem = i.freeram + i.bufferram;
111 /* however in linux 2.5 the i.bufferram is total page cache size,
112 we need adjust it */
113 /* si_swapinfo(&i); */
114 /* availmem = availmem - (i.totalswap - i.freeswap); */
115
Hans Schillstroma0840e22011-01-03 14:44:58 +0100116 nomem = (availmem < ipvs->sysctl_amemthresh);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700117
Julian Anastasovaf9debd2005-07-11 20:59:57 -0700118 local_bh_disable();
119
Linus Torvalds1da177e2005-04-16 15:20:36 -0700120 /* drop_entry */
Hans Schillstroma0840e22011-01-03 14:44:58 +0100121 spin_lock(&ipvs->dropentry_lock);
122 switch (ipvs->sysctl_drop_entry) {
Linus Torvalds1da177e2005-04-16 15:20:36 -0700123 case 0:
Hans Schillstroma0840e22011-01-03 14:44:58 +0100124 atomic_set(&ipvs->dropentry, 0);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700125 break;
126 case 1:
127 if (nomem) {
Hans Schillstroma0840e22011-01-03 14:44:58 +0100128 atomic_set(&ipvs->dropentry, 1);
129 ipvs->sysctl_drop_entry = 2;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700130 } else {
Hans Schillstroma0840e22011-01-03 14:44:58 +0100131 atomic_set(&ipvs->dropentry, 0);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700132 }
133 break;
134 case 2:
135 if (nomem) {
Hans Schillstroma0840e22011-01-03 14:44:58 +0100136 atomic_set(&ipvs->dropentry, 1);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700137 } else {
Hans Schillstroma0840e22011-01-03 14:44:58 +0100138 atomic_set(&ipvs->dropentry, 0);
139 ipvs->sysctl_drop_entry = 1;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700140 };
141 break;
142 case 3:
Hans Schillstroma0840e22011-01-03 14:44:58 +0100143 atomic_set(&ipvs->dropentry, 1);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700144 break;
145 }
Hans Schillstroma0840e22011-01-03 14:44:58 +0100146 spin_unlock(&ipvs->dropentry_lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700147
148 /* drop_packet */
Hans Schillstroma0840e22011-01-03 14:44:58 +0100149 spin_lock(&ipvs->droppacket_lock);
150 switch (ipvs->sysctl_drop_packet) {
Linus Torvalds1da177e2005-04-16 15:20:36 -0700151 case 0:
Hans Schillstroma0840e22011-01-03 14:44:58 +0100152 ipvs->drop_rate = 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700153 break;
154 case 1:
155 if (nomem) {
Hans Schillstroma0840e22011-01-03 14:44:58 +0100156 ipvs->drop_rate = ipvs->drop_counter
157 = ipvs->sysctl_amemthresh /
158 (ipvs->sysctl_amemthresh-availmem);
159 ipvs->sysctl_drop_packet = 2;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700160 } else {
Hans Schillstroma0840e22011-01-03 14:44:58 +0100161 ipvs->drop_rate = 0;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700162 }
163 break;
164 case 2:
165 if (nomem) {
Hans Schillstroma0840e22011-01-03 14:44:58 +0100166 ipvs->drop_rate = ipvs->drop_counter
167 = ipvs->sysctl_amemthresh /
168 (ipvs->sysctl_amemthresh-availmem);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700169 } else {
Hans Schillstroma0840e22011-01-03 14:44:58 +0100170 ipvs->drop_rate = 0;
171 ipvs->sysctl_drop_packet = 1;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700172 }
173 break;
174 case 3:
Hans Schillstroma0840e22011-01-03 14:44:58 +0100175 ipvs->drop_rate = ipvs->sysctl_am_droprate;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700176 break;
177 }
Hans Schillstroma0840e22011-01-03 14:44:58 +0100178 spin_unlock(&ipvs->droppacket_lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700179
180 /* secure_tcp */
Hans Schillstroma0840e22011-01-03 14:44:58 +0100181 spin_lock(&ipvs->securetcp_lock);
182 switch (ipvs->sysctl_secure_tcp) {
Linus Torvalds1da177e2005-04-16 15:20:36 -0700183 case 0:
184 if (old_secure_tcp >= 2)
185 to_change = 0;
186 break;
187 case 1:
188 if (nomem) {
189 if (old_secure_tcp < 2)
190 to_change = 1;
Hans Schillstroma0840e22011-01-03 14:44:58 +0100191 ipvs->sysctl_secure_tcp = 2;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700192 } else {
193 if (old_secure_tcp >= 2)
194 to_change = 0;
195 }
196 break;
197 case 2:
198 if (nomem) {
199 if (old_secure_tcp < 2)
200 to_change = 1;
201 } else {
202 if (old_secure_tcp >= 2)
203 to_change = 0;
Hans Schillstroma0840e22011-01-03 14:44:58 +0100204 ipvs->sysctl_secure_tcp = 1;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700205 }
206 break;
207 case 3:
208 if (old_secure_tcp < 2)
209 to_change = 1;
210 break;
211 }
Hans Schillstroma0840e22011-01-03 14:44:58 +0100212 old_secure_tcp = ipvs->sysctl_secure_tcp;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700213 if (to_change >= 0)
Hans Schillstrom93304192011-01-03 14:44:51 +0100214 ip_vs_protocol_timeout_change(ipvs,
Hans Schillstroma0840e22011-01-03 14:44:58 +0100215 ipvs->sysctl_secure_tcp > 1);
216 spin_unlock(&ipvs->securetcp_lock);
Julian Anastasovaf9debd2005-07-11 20:59:57 -0700217
218 local_bh_enable();
Linus Torvalds1da177e2005-04-16 15:20:36 -0700219}
220
221
222/*
223 * Timer for checking the defense
224 */
225#define DEFENSE_TIMER_PERIOD 1*HZ
Linus Torvalds1da177e2005-04-16 15:20:36 -0700226
David Howellsc4028952006-11-22 14:57:56 +0000227static void defense_work_handler(struct work_struct *work)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700228{
Hans Schillstromf6340ee2011-01-03 14:44:59 +0100229 struct netns_ipvs *ipvs =
230 container_of(work, struct netns_ipvs, defense_work.work);
Hans Schillstrom93304192011-01-03 14:44:51 +0100231
232 update_defense_level(ipvs);
Hans Schillstroma0840e22011-01-03 14:44:58 +0100233 if (atomic_read(&ipvs->dropentry))
Hans Schillstromf6340ee2011-01-03 14:44:59 +0100234 ip_vs_random_dropentry(ipvs->net);
235 schedule_delayed_work(&ipvs->defense_work, DEFENSE_TIMER_PERIOD);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700236}
Simon Horman14e40542011-02-04 18:33:02 +0900237#endif
Linus Torvalds1da177e2005-04-16 15:20:36 -0700238
239int
240ip_vs_use_count_inc(void)
241{
242 return try_module_get(THIS_MODULE);
243}
244
245void
246ip_vs_use_count_dec(void)
247{
248 module_put(THIS_MODULE);
249}
250
251
252/*
253 * Hash table: for virtual service lookups
254 */
255#define IP_VS_SVC_TAB_BITS 8
256#define IP_VS_SVC_TAB_SIZE (1 << IP_VS_SVC_TAB_BITS)
257#define IP_VS_SVC_TAB_MASK (IP_VS_SVC_TAB_SIZE - 1)
258
259/* the service table hashed by <protocol, addr, port> */
260static struct list_head ip_vs_svc_table[IP_VS_SVC_TAB_SIZE];
261/* the service table hashed by fwmark */
262static struct list_head ip_vs_svc_fwm_table[IP_VS_SVC_TAB_SIZE];
263
Linus Torvalds1da177e2005-04-16 15:20:36 -0700264
265/*
266 * Returns hash value for virtual service
267 */
Hans Schillstromfc723252011-01-03 14:44:43 +0100268static inline unsigned
269ip_vs_svc_hashkey(struct net *net, int af, unsigned proto,
270 const union nf_inet_addr *addr, __be16 port)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700271{
272 register unsigned porth = ntohs(port);
Julius Volzb18610d2008-09-02 15:55:37 +0200273 __be32 addr_fold = addr->ip;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700274
Julius Volzb18610d2008-09-02 15:55:37 +0200275#ifdef CONFIG_IP_VS_IPV6
276 if (af == AF_INET6)
277 addr_fold = addr->ip6[0]^addr->ip6[1]^
278 addr->ip6[2]^addr->ip6[3];
279#endif
Hans Schillstromfc723252011-01-03 14:44:43 +0100280 addr_fold ^= ((size_t)net>>8);
Julius Volzb18610d2008-09-02 15:55:37 +0200281
282 return (proto^ntohl(addr_fold)^(porth>>IP_VS_SVC_TAB_BITS)^porth)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700283 & IP_VS_SVC_TAB_MASK;
284}
285
286/*
287 * Returns hash value of fwmark for virtual service lookup
288 */
Hans Schillstromfc723252011-01-03 14:44:43 +0100289static inline unsigned ip_vs_svc_fwm_hashkey(struct net *net, __u32 fwmark)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700290{
Hans Schillstromfc723252011-01-03 14:44:43 +0100291 return (((size_t)net>>8) ^ fwmark) & IP_VS_SVC_TAB_MASK;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700292}
293
294/*
Hans Schillstromfc723252011-01-03 14:44:43 +0100295 * Hashes a service in the ip_vs_svc_table by <netns,proto,addr,port>
Linus Torvalds1da177e2005-04-16 15:20:36 -0700296 * or in the ip_vs_svc_fwm_table by fwmark.
297 * Should be called with locked tables.
298 */
299static int ip_vs_svc_hash(struct ip_vs_service *svc)
300{
301 unsigned hash;
302
303 if (svc->flags & IP_VS_SVC_F_HASHED) {
Hannes Eder1e3e2382009-08-02 11:05:41 +0000304 pr_err("%s(): request for already hashed, called from %pF\n",
305 __func__, __builtin_return_address(0));
Linus Torvalds1da177e2005-04-16 15:20:36 -0700306 return 0;
307 }
308
309 if (svc->fwmark == 0) {
310 /*
Hans Schillstromfc723252011-01-03 14:44:43 +0100311 * Hash it by <netns,protocol,addr,port> in ip_vs_svc_table
Linus Torvalds1da177e2005-04-16 15:20:36 -0700312 */
Hans Schillstromfc723252011-01-03 14:44:43 +0100313 hash = ip_vs_svc_hashkey(svc->net, svc->af, svc->protocol,
314 &svc->addr, svc->port);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700315 list_add(&svc->s_list, &ip_vs_svc_table[hash]);
316 } else {
317 /*
Hans Schillstromfc723252011-01-03 14:44:43 +0100318 * Hash it by fwmark in svc_fwm_table
Linus Torvalds1da177e2005-04-16 15:20:36 -0700319 */
Hans Schillstromfc723252011-01-03 14:44:43 +0100320 hash = ip_vs_svc_fwm_hashkey(svc->net, svc->fwmark);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700321 list_add(&svc->f_list, &ip_vs_svc_fwm_table[hash]);
322 }
323
324 svc->flags |= IP_VS_SVC_F_HASHED;
325 /* increase its refcnt because it is referenced by the svc table */
326 atomic_inc(&svc->refcnt);
327 return 1;
328}
329
330
331/*
Hans Schillstromfc723252011-01-03 14:44:43 +0100332 * Unhashes a service from svc_table / svc_fwm_table.
Linus Torvalds1da177e2005-04-16 15:20:36 -0700333 * Should be called with locked tables.
334 */
335static int ip_vs_svc_unhash(struct ip_vs_service *svc)
336{
337 if (!(svc->flags & IP_VS_SVC_F_HASHED)) {
Hannes Eder1e3e2382009-08-02 11:05:41 +0000338 pr_err("%s(): request for unhash flagged, called from %pF\n",
339 __func__, __builtin_return_address(0));
Linus Torvalds1da177e2005-04-16 15:20:36 -0700340 return 0;
341 }
342
343 if (svc->fwmark == 0) {
Hans Schillstromfc723252011-01-03 14:44:43 +0100344 /* Remove it from the svc_table table */
Linus Torvalds1da177e2005-04-16 15:20:36 -0700345 list_del(&svc->s_list);
346 } else {
Hans Schillstromfc723252011-01-03 14:44:43 +0100347 /* Remove it from the svc_fwm_table table */
Linus Torvalds1da177e2005-04-16 15:20:36 -0700348 list_del(&svc->f_list);
349 }
350
351 svc->flags &= ~IP_VS_SVC_F_HASHED;
352 atomic_dec(&svc->refcnt);
353 return 1;
354}
355
356
357/*
Hans Schillstromfc723252011-01-03 14:44:43 +0100358 * Get service by {netns, proto,addr,port} in the service table.
Linus Torvalds1da177e2005-04-16 15:20:36 -0700359 */
Julius Volzb18610d2008-09-02 15:55:37 +0200360static inline struct ip_vs_service *
Hans Schillstromfc723252011-01-03 14:44:43 +0100361__ip_vs_service_find(struct net *net, int af, __u16 protocol,
362 const union nf_inet_addr *vaddr, __be16 vport)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700363{
364 unsigned hash;
365 struct ip_vs_service *svc;
366
367 /* Check for "full" addressed entries */
Hans Schillstromfc723252011-01-03 14:44:43 +0100368 hash = ip_vs_svc_hashkey(net, af, protocol, vaddr, vport);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700369
370 list_for_each_entry(svc, &ip_vs_svc_table[hash], s_list){
Julius Volzb18610d2008-09-02 15:55:37 +0200371 if ((svc->af == af)
372 && ip_vs_addr_equal(af, &svc->addr, vaddr)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700373 && (svc->port == vport)
Hans Schillstromfc723252011-01-03 14:44:43 +0100374 && (svc->protocol == protocol)
375 && net_eq(svc->net, net)) {
Linus Torvalds1da177e2005-04-16 15:20:36 -0700376 /* HIT */
Linus Torvalds1da177e2005-04-16 15:20:36 -0700377 return svc;
378 }
379 }
380
381 return NULL;
382}
383
384
385/*
386 * Get service by {fwmark} in the service table.
387 */
Julius Volzb18610d2008-09-02 15:55:37 +0200388static inline struct ip_vs_service *
Hans Schillstromfc723252011-01-03 14:44:43 +0100389__ip_vs_svc_fwm_find(struct net *net, int af, __u32 fwmark)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700390{
391 unsigned hash;
392 struct ip_vs_service *svc;
393
394 /* Check for fwmark addressed entries */
Hans Schillstromfc723252011-01-03 14:44:43 +0100395 hash = ip_vs_svc_fwm_hashkey(net, fwmark);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700396
397 list_for_each_entry(svc, &ip_vs_svc_fwm_table[hash], f_list) {
Hans Schillstromfc723252011-01-03 14:44:43 +0100398 if (svc->fwmark == fwmark && svc->af == af
399 && net_eq(svc->net, net)) {
Linus Torvalds1da177e2005-04-16 15:20:36 -0700400 /* HIT */
Linus Torvalds1da177e2005-04-16 15:20:36 -0700401 return svc;
402 }
403 }
404
405 return NULL;
406}
407
408struct ip_vs_service *
Hans Schillstromfc723252011-01-03 14:44:43 +0100409ip_vs_service_get(struct net *net, int af, __u32 fwmark, __u16 protocol,
Julius Volz3c2e0502008-09-02 15:55:38 +0200410 const union nf_inet_addr *vaddr, __be16 vport)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700411{
412 struct ip_vs_service *svc;
Hans Schillstrom763f8d02011-01-03 14:45:01 +0100413 struct netns_ipvs *ipvs = net_ipvs(net);
Julius Volz3c2e0502008-09-02 15:55:38 +0200414
Linus Torvalds1da177e2005-04-16 15:20:36 -0700415 read_lock(&__ip_vs_svc_lock);
416
417 /*
418 * Check the table hashed by fwmark first
419 */
Julian Anastasov097fc762011-03-04 12:26:17 +0200420 if (fwmark) {
421 svc = __ip_vs_svc_fwm_find(net, af, fwmark);
422 if (svc)
423 goto out;
424 }
Linus Torvalds1da177e2005-04-16 15:20:36 -0700425
426 /*
427 * Check the table hashed by <protocol,addr,port>
428 * for "full" addressed entries
429 */
Hans Schillstromfc723252011-01-03 14:44:43 +0100430 svc = __ip_vs_service_find(net, af, protocol, vaddr, vport);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700431
432 if (svc == NULL
433 && protocol == IPPROTO_TCP
Hans Schillstrom763f8d02011-01-03 14:45:01 +0100434 && atomic_read(&ipvs->ftpsvc_counter)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700435 && (vport == FTPDATA || ntohs(vport) >= PROT_SOCK)) {
436 /*
437 * Check if ftp service entry exists, the packet
438 * might belong to FTP data connections.
439 */
Hans Schillstromfc723252011-01-03 14:44:43 +0100440 svc = __ip_vs_service_find(net, af, protocol, vaddr, FTPPORT);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700441 }
442
443 if (svc == NULL
Hans Schillstrom763f8d02011-01-03 14:45:01 +0100444 && atomic_read(&ipvs->nullsvc_counter)) {
Linus Torvalds1da177e2005-04-16 15:20:36 -0700445 /*
446 * Check if the catch-all port (port zero) exists
447 */
Hans Schillstromfc723252011-01-03 14:44:43 +0100448 svc = __ip_vs_service_find(net, af, protocol, vaddr, 0);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700449 }
450
451 out:
Julian Anastasov26c15cf2010-09-21 18:12:30 +0200452 if (svc)
453 atomic_inc(&svc->usecnt);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700454 read_unlock(&__ip_vs_svc_lock);
455
Julius Volz3c2e0502008-09-02 15:55:38 +0200456 IP_VS_DBG_BUF(9, "lookup service: fwm %u %s %s:%u %s\n",
457 fwmark, ip_vs_proto_name(protocol),
458 IP_VS_DBG_ADDR(af, vaddr), ntohs(vport),
459 svc ? "hit" : "not hit");
Linus Torvalds1da177e2005-04-16 15:20:36 -0700460
461 return svc;
462}
463
464
465static inline void
466__ip_vs_bind_svc(struct ip_vs_dest *dest, struct ip_vs_service *svc)
467{
468 atomic_inc(&svc->refcnt);
469 dest->svc = svc;
470}
471
Julian Anastasov26c15cf2010-09-21 18:12:30 +0200472static void
Linus Torvalds1da177e2005-04-16 15:20:36 -0700473__ip_vs_unbind_svc(struct ip_vs_dest *dest)
474{
475 struct ip_vs_service *svc = dest->svc;
476
477 dest->svc = NULL;
Julian Anastasov26c15cf2010-09-21 18:12:30 +0200478 if (atomic_dec_and_test(&svc->refcnt)) {
479 IP_VS_DBG_BUF(3, "Removing service %u/%s:%u usecnt=%d\n",
480 svc->fwmark,
481 IP_VS_DBG_ADDR(svc->af, &svc->addr),
482 ntohs(svc->port), atomic_read(&svc->usecnt));
Hans Schillstromb17fc992011-01-03 14:44:56 +0100483 free_percpu(svc->stats.cpustats);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700484 kfree(svc);
Julian Anastasov26c15cf2010-09-21 18:12:30 +0200485 }
Linus Torvalds1da177e2005-04-16 15:20:36 -0700486}
487
488
489/*
490 * Returns hash value for real service
491 */
Julius Volz7937df12008-09-02 15:55:48 +0200492static inline unsigned ip_vs_rs_hashkey(int af,
493 const union nf_inet_addr *addr,
494 __be16 port)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700495{
496 register unsigned porth = ntohs(port);
Julius Volz7937df12008-09-02 15:55:48 +0200497 __be32 addr_fold = addr->ip;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700498
Julius Volz7937df12008-09-02 15:55:48 +0200499#ifdef CONFIG_IP_VS_IPV6
500 if (af == AF_INET6)
501 addr_fold = addr->ip6[0]^addr->ip6[1]^
502 addr->ip6[2]^addr->ip6[3];
503#endif
504
505 return (ntohl(addr_fold)^(porth>>IP_VS_RTAB_BITS)^porth)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700506 & IP_VS_RTAB_MASK;
507}
508
509/*
Hans Schillstromfc723252011-01-03 14:44:43 +0100510 * Hashes ip_vs_dest in rs_table by <proto,addr,port>.
Linus Torvalds1da177e2005-04-16 15:20:36 -0700511 * should be called with locked tables.
512 */
Hans Schillstromfc723252011-01-03 14:44:43 +0100513static int ip_vs_rs_hash(struct netns_ipvs *ipvs, struct ip_vs_dest *dest)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700514{
515 unsigned hash;
516
517 if (!list_empty(&dest->d_list)) {
518 return 0;
519 }
520
521 /*
522 * Hash by proto,addr,port,
523 * which are the parameters of the real service.
524 */
Julius Volz7937df12008-09-02 15:55:48 +0200525 hash = ip_vs_rs_hashkey(dest->af, &dest->addr, dest->port);
526
Hans Schillstromfc723252011-01-03 14:44:43 +0100527 list_add(&dest->d_list, &ipvs->rs_table[hash]);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700528
529 return 1;
530}
531
532/*
Hans Schillstromfc723252011-01-03 14:44:43 +0100533 * UNhashes ip_vs_dest from rs_table.
Linus Torvalds1da177e2005-04-16 15:20:36 -0700534 * should be called with locked tables.
535 */
536static int ip_vs_rs_unhash(struct ip_vs_dest *dest)
537{
538 /*
Hans Schillstromfc723252011-01-03 14:44:43 +0100539 * Remove it from the rs_table table.
Linus Torvalds1da177e2005-04-16 15:20:36 -0700540 */
541 if (!list_empty(&dest->d_list)) {
542 list_del(&dest->d_list);
543 INIT_LIST_HEAD(&dest->d_list);
544 }
545
546 return 1;
547}
548
549/*
550 * Lookup real service by <proto,addr,port> in the real service table.
551 */
552struct ip_vs_dest *
Hans Schillstromfc723252011-01-03 14:44:43 +0100553ip_vs_lookup_real_service(struct net *net, int af, __u16 protocol,
Julius Volz7937df12008-09-02 15:55:48 +0200554 const union nf_inet_addr *daddr,
555 __be16 dport)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700556{
Hans Schillstromfc723252011-01-03 14:44:43 +0100557 struct netns_ipvs *ipvs = net_ipvs(net);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700558 unsigned hash;
559 struct ip_vs_dest *dest;
560
561 /*
562 * Check for "full" addressed entries
563 * Return the first found entry
564 */
Julius Volz7937df12008-09-02 15:55:48 +0200565 hash = ip_vs_rs_hashkey(af, daddr, dport);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700566
Hans Schillstroma0840e22011-01-03 14:44:58 +0100567 read_lock(&ipvs->rs_lock);
Hans Schillstromfc723252011-01-03 14:44:43 +0100568 list_for_each_entry(dest, &ipvs->rs_table[hash], d_list) {
Julius Volz7937df12008-09-02 15:55:48 +0200569 if ((dest->af == af)
570 && ip_vs_addr_equal(af, &dest->addr, daddr)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700571 && (dest->port == dport)
572 && ((dest->protocol == protocol) ||
573 dest->vfwmark)) {
574 /* HIT */
Hans Schillstroma0840e22011-01-03 14:44:58 +0100575 read_unlock(&ipvs->rs_lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700576 return dest;
577 }
578 }
Hans Schillstroma0840e22011-01-03 14:44:58 +0100579 read_unlock(&ipvs->rs_lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700580
581 return NULL;
582}
583
584/*
585 * Lookup destination by {addr,port} in the given service
586 */
587static struct ip_vs_dest *
Julius Volz7937df12008-09-02 15:55:48 +0200588ip_vs_lookup_dest(struct ip_vs_service *svc, const union nf_inet_addr *daddr,
589 __be16 dport)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700590{
591 struct ip_vs_dest *dest;
592
593 /*
594 * Find the destination for the given service
595 */
596 list_for_each_entry(dest, &svc->destinations, n_list) {
Julius Volz7937df12008-09-02 15:55:48 +0200597 if ((dest->af == svc->af)
598 && ip_vs_addr_equal(svc->af, &dest->addr, daddr)
599 && (dest->port == dport)) {
Linus Torvalds1da177e2005-04-16 15:20:36 -0700600 /* HIT */
601 return dest;
602 }
603 }
604
605 return NULL;
606}
607
Rumen G. Bogdanovski1e356f92007-11-07 02:35:54 -0800608/*
609 * Find destination by {daddr,dport,vaddr,protocol}
610 * Cretaed to be used in ip_vs_process_message() in
611 * the backup synchronization daemon. It finds the
612 * destination to be bound to the received connection
613 * on the backup.
614 *
615 * ip_vs_lookup_real_service() looked promissing, but
616 * seems not working as expected.
617 */
Hans Schillstromfc723252011-01-03 14:44:43 +0100618struct ip_vs_dest *ip_vs_find_dest(struct net *net, int af,
619 const union nf_inet_addr *daddr,
Julius Volz7937df12008-09-02 15:55:48 +0200620 __be16 dport,
621 const union nf_inet_addr *vaddr,
Julian Anastasov52793db2011-12-30 14:19:02 +0900622 __be16 vport, __u16 protocol, __u32 fwmark,
623 __u32 flags)
Rumen G. Bogdanovski1e356f92007-11-07 02:35:54 -0800624{
625 struct ip_vs_dest *dest;
626 struct ip_vs_service *svc;
Julian Anastasov52793db2011-12-30 14:19:02 +0900627 __be16 port = dport;
Rumen G. Bogdanovski1e356f92007-11-07 02:35:54 -0800628
Hans Schillstromfc723252011-01-03 14:44:43 +0100629 svc = ip_vs_service_get(net, af, fwmark, protocol, vaddr, vport);
Rumen G. Bogdanovski1e356f92007-11-07 02:35:54 -0800630 if (!svc)
631 return NULL;
Julian Anastasov52793db2011-12-30 14:19:02 +0900632 if (fwmark && (flags & IP_VS_CONN_F_FWD_MASK) != IP_VS_CONN_F_MASQ)
633 port = 0;
634 dest = ip_vs_lookup_dest(svc, daddr, port);
635 if (!dest)
636 dest = ip_vs_lookup_dest(svc, daddr, port ^ dport);
Rumen G. Bogdanovski1e356f92007-11-07 02:35:54 -0800637 if (dest)
638 atomic_inc(&dest->refcnt);
639 ip_vs_service_put(svc);
640 return dest;
641}
Linus Torvalds1da177e2005-04-16 15:20:36 -0700642
643/*
644 * Lookup dest by {svc,addr,port} in the destination trash.
645 * The destination trash is used to hold the destinations that are removed
646 * from the service table but are still referenced by some conn entries.
647 * The reason to add the destination trash is when the dest is temporary
648 * down (either by administrator or by monitor program), the dest can be
649 * picked back from the trash, the remaining connections to the dest can
650 * continue, and the counting information of the dest is also useful for
651 * scheduling.
652 */
653static struct ip_vs_dest *
Julius Volz7937df12008-09-02 15:55:48 +0200654ip_vs_trash_get_dest(struct ip_vs_service *svc, const union nf_inet_addr *daddr,
655 __be16 dport)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700656{
657 struct ip_vs_dest *dest, *nxt;
Hans Schillstromf2431e62011-01-03 14:45:00 +0100658 struct netns_ipvs *ipvs = net_ipvs(svc->net);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700659
660 /*
661 * Find the destination in trash
662 */
Hans Schillstromf2431e62011-01-03 14:45:00 +0100663 list_for_each_entry_safe(dest, nxt, &ipvs->dest_trash, n_list) {
Julius Volz7937df12008-09-02 15:55:48 +0200664 IP_VS_DBG_BUF(3, "Destination %u/%s:%u still in trash, "
665 "dest->refcnt=%d\n",
666 dest->vfwmark,
667 IP_VS_DBG_ADDR(svc->af, &dest->addr),
668 ntohs(dest->port),
669 atomic_read(&dest->refcnt));
670 if (dest->af == svc->af &&
671 ip_vs_addr_equal(svc->af, &dest->addr, daddr) &&
Linus Torvalds1da177e2005-04-16 15:20:36 -0700672 dest->port == dport &&
673 dest->vfwmark == svc->fwmark &&
674 dest->protocol == svc->protocol &&
675 (svc->fwmark ||
Julius Volz7937df12008-09-02 15:55:48 +0200676 (ip_vs_addr_equal(svc->af, &dest->vaddr, &svc->addr) &&
Linus Torvalds1da177e2005-04-16 15:20:36 -0700677 dest->vport == svc->port))) {
678 /* HIT */
679 return dest;
680 }
681
682 /*
683 * Try to purge the destination from trash if not referenced
684 */
685 if (atomic_read(&dest->refcnt) == 1) {
Julius Volz7937df12008-09-02 15:55:48 +0200686 IP_VS_DBG_BUF(3, "Removing destination %u/%s:%u "
687 "from trash\n",
688 dest->vfwmark,
689 IP_VS_DBG_ADDR(svc->af, &dest->addr),
690 ntohs(dest->port));
Linus Torvalds1da177e2005-04-16 15:20:36 -0700691 list_del(&dest->n_list);
692 ip_vs_dst_reset(dest);
693 __ip_vs_unbind_svc(dest);
Hans Schillstromb17fc992011-01-03 14:44:56 +0100694 free_percpu(dest->stats.cpustats);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700695 kfree(dest);
696 }
697 }
698
699 return NULL;
700}
701
702
703/*
704 * Clean up all the destinations in the trash
705 * Called by the ip_vs_control_cleanup()
706 *
707 * When the ip_vs_control_clearup is activated by ipvs module exit,
708 * the service tables must have been flushed and all the connections
709 * are expired, and the refcnt of each destination in the trash must
710 * be 1, so we simply release them here.
711 */
Hans Schillstromf2431e62011-01-03 14:45:00 +0100712static void ip_vs_trash_cleanup(struct net *net)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700713{
714 struct ip_vs_dest *dest, *nxt;
Hans Schillstromf2431e62011-01-03 14:45:00 +0100715 struct netns_ipvs *ipvs = net_ipvs(net);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700716
Hans Schillstromf2431e62011-01-03 14:45:00 +0100717 list_for_each_entry_safe(dest, nxt, &ipvs->dest_trash, n_list) {
Linus Torvalds1da177e2005-04-16 15:20:36 -0700718 list_del(&dest->n_list);
719 ip_vs_dst_reset(dest);
720 __ip_vs_unbind_svc(dest);
Hans Schillstromb17fc992011-01-03 14:44:56 +0100721 free_percpu(dest->stats.cpustats);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700722 kfree(dest);
723 }
724}
725
Julian Anastasov55a3d4e2011-03-14 01:37:49 +0200726static void
727ip_vs_copy_stats(struct ip_vs_stats_user *dst, struct ip_vs_stats *src)
728{
729#define IP_VS_SHOW_STATS_COUNTER(c) dst->c = src->ustats.c - src->ustats0.c
Julian Anastasov55a3d4e2011-03-14 01:37:49 +0200730
731 spin_lock_bh(&src->lock);
732
733 IP_VS_SHOW_STATS_COUNTER(conns);
734 IP_VS_SHOW_STATS_COUNTER(inpkts);
735 IP_VS_SHOW_STATS_COUNTER(outpkts);
736 IP_VS_SHOW_STATS_COUNTER(inbytes);
737 IP_VS_SHOW_STATS_COUNTER(outbytes);
738
Julian Anastasovea9f22c2011-03-14 01:41:54 +0200739 ip_vs_read_estimator(dst, src);
Julian Anastasov55a3d4e2011-03-14 01:37:49 +0200740
741 spin_unlock_bh(&src->lock);
742}
Linus Torvalds1da177e2005-04-16 15:20:36 -0700743
744static void
745ip_vs_zero_stats(struct ip_vs_stats *stats)
746{
747 spin_lock_bh(&stats->lock);
Simon Hormane93615d2008-08-11 17:19:14 +1000748
Julian Anastasov55a3d4e2011-03-14 01:37:49 +0200749 /* get current counters as zero point, rates are zeroed */
750
751#define IP_VS_ZERO_STATS_COUNTER(c) stats->ustats0.c = stats->ustats.c
Julian Anastasov55a3d4e2011-03-14 01:37:49 +0200752
753 IP_VS_ZERO_STATS_COUNTER(conns);
754 IP_VS_ZERO_STATS_COUNTER(inpkts);
755 IP_VS_ZERO_STATS_COUNTER(outpkts);
756 IP_VS_ZERO_STATS_COUNTER(inbytes);
757 IP_VS_ZERO_STATS_COUNTER(outbytes);
758
Linus Torvalds1da177e2005-04-16 15:20:36 -0700759 ip_vs_zero_estimator(stats);
Simon Hormane93615d2008-08-11 17:19:14 +1000760
Sven Wegener3a14a3132008-08-10 18:24:41 +0000761 spin_unlock_bh(&stats->lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700762}
763
764/*
765 * Update a destination in the given service
766 */
767static void
Julian Anastasov26c15cf2010-09-21 18:12:30 +0200768__ip_vs_update_dest(struct ip_vs_service *svc, struct ip_vs_dest *dest,
769 struct ip_vs_dest_user_kern *udest, int add)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700770{
Hans Schillstromfc723252011-01-03 14:44:43 +0100771 struct netns_ipvs *ipvs = net_ipvs(svc->net);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700772 int conn_flags;
773
774 /* set the weight and the flags */
775 atomic_set(&dest->weight, udest->weight);
Julian Anastasov35757922010-09-17 14:18:16 +0200776 conn_flags = udest->conn_flags & IP_VS_CONN_F_DEST_MASK;
777 conn_flags |= IP_VS_CONN_F_INACTIVE;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700778
Linus Torvalds1da177e2005-04-16 15:20:36 -0700779 /* set the IP_VS_CONN_F_NOOUTPUT flag if not masquerading/NAT */
Julian Anastasov35757922010-09-17 14:18:16 +0200780 if ((conn_flags & IP_VS_CONN_F_FWD_MASK) != IP_VS_CONN_F_MASQ) {
Linus Torvalds1da177e2005-04-16 15:20:36 -0700781 conn_flags |= IP_VS_CONN_F_NOOUTPUT;
782 } else {
783 /*
Hans Schillstromfc723252011-01-03 14:44:43 +0100784 * Put the real service in rs_table if not present.
Linus Torvalds1da177e2005-04-16 15:20:36 -0700785 * For now only for NAT!
786 */
Hans Schillstroma0840e22011-01-03 14:44:58 +0100787 write_lock_bh(&ipvs->rs_lock);
Hans Schillstromfc723252011-01-03 14:44:43 +0100788 ip_vs_rs_hash(ipvs, dest);
Hans Schillstroma0840e22011-01-03 14:44:58 +0100789 write_unlock_bh(&ipvs->rs_lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700790 }
791 atomic_set(&dest->conn_flags, conn_flags);
792
793 /* bind the service */
794 if (!dest->svc) {
795 __ip_vs_bind_svc(dest, svc);
796 } else {
797 if (dest->svc != svc) {
798 __ip_vs_unbind_svc(dest);
799 ip_vs_zero_stats(&dest->stats);
800 __ip_vs_bind_svc(dest, svc);
801 }
802 }
803
804 /* set the dest status flags */
805 dest->flags |= IP_VS_DEST_F_AVAILABLE;
806
807 if (udest->u_threshold == 0 || udest->u_threshold > dest->u_threshold)
808 dest->flags &= ~IP_VS_DEST_F_OVERLOAD;
809 dest->u_threshold = udest->u_threshold;
810 dest->l_threshold = udest->l_threshold;
Julian Anastasov26c15cf2010-09-21 18:12:30 +0200811
Julian Anastasovff75f402011-02-22 10:40:25 +0200812 spin_lock_bh(&dest->dst_lock);
Julian Anastasovfc604762010-10-17 16:38:15 +0300813 ip_vs_dst_reset(dest);
Julian Anastasovff75f402011-02-22 10:40:25 +0200814 spin_unlock_bh(&dest->dst_lock);
Julian Anastasovfc604762010-10-17 16:38:15 +0300815
Julian Anastasov26c15cf2010-09-21 18:12:30 +0200816 if (add)
Julian Anastasov6ef757f2011-03-14 01:44:28 +0200817 ip_vs_start_estimator(svc->net, &dest->stats);
Julian Anastasov26c15cf2010-09-21 18:12:30 +0200818
819 write_lock_bh(&__ip_vs_svc_lock);
820
821 /* Wait until all other svc users go away */
822 IP_VS_WAIT_WHILE(atomic_read(&svc->usecnt) > 0);
823
824 if (add) {
825 list_add(&dest->n_list, &svc->destinations);
826 svc->num_dests++;
827 }
828
829 /* call the update_service, because server weight may be changed */
830 if (svc->scheduler->update_service)
831 svc->scheduler->update_service(svc);
832
833 write_unlock_bh(&__ip_vs_svc_lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700834}
835
836
837/*
838 * Create a destination for the given service
839 */
840static int
Julius Volzc860c6b2008-09-02 15:55:36 +0200841ip_vs_new_dest(struct ip_vs_service *svc, struct ip_vs_dest_user_kern *udest,
Linus Torvalds1da177e2005-04-16 15:20:36 -0700842 struct ip_vs_dest **dest_p)
843{
844 struct ip_vs_dest *dest;
845 unsigned atype;
846
847 EnterFunction(2);
848
Vince Busam09571c72008-09-02 15:55:52 +0200849#ifdef CONFIG_IP_VS_IPV6
850 if (svc->af == AF_INET6) {
851 atype = ipv6_addr_type(&udest->addr.in6);
Sven Wegener3bfb92f2008-09-05 16:53:49 +0200852 if ((!(atype & IPV6_ADDR_UNICAST) ||
853 atype & IPV6_ADDR_LINKLOCAL) &&
Hans Schillstrom4a984802011-01-03 14:45:02 +0100854 !__ip_vs_addr_is_local_v6(svc->net, &udest->addr.in6))
Vince Busam09571c72008-09-02 15:55:52 +0200855 return -EINVAL;
856 } else
857#endif
858 {
Hans Schillstrom4a984802011-01-03 14:45:02 +0100859 atype = inet_addr_type(svc->net, udest->addr.ip);
Vince Busam09571c72008-09-02 15:55:52 +0200860 if (atype != RTN_LOCAL && atype != RTN_UNICAST)
861 return -EINVAL;
862 }
Linus Torvalds1da177e2005-04-16 15:20:36 -0700863
Simon Hormandee06e42010-08-26 02:54:31 +0000864 dest = kzalloc(sizeof(struct ip_vs_dest), GFP_KERNEL);
Joe Perches0a9ee812011-08-29 14:17:25 -0700865 if (dest == NULL)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700866 return -ENOMEM;
Joe Perches0a9ee812011-08-29 14:17:25 -0700867
Hans Schillstromb17fc992011-01-03 14:44:56 +0100868 dest->stats.cpustats = alloc_percpu(struct ip_vs_cpu_stats);
Joe Perches0a9ee812011-08-29 14:17:25 -0700869 if (!dest->stats.cpustats)
Hans Schillstromb17fc992011-01-03 14:44:56 +0100870 goto err_alloc;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700871
Julius Volzc860c6b2008-09-02 15:55:36 +0200872 dest->af = svc->af;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700873 dest->protocol = svc->protocol;
Julius Volzc860c6b2008-09-02 15:55:36 +0200874 dest->vaddr = svc->addr;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700875 dest->vport = svc->port;
876 dest->vfwmark = svc->fwmark;
Julius Volzc860c6b2008-09-02 15:55:36 +0200877 ip_vs_addr_copy(svc->af, &dest->addr, &udest->addr);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700878 dest->port = udest->port;
879
880 atomic_set(&dest->activeconns, 0);
881 atomic_set(&dest->inactconns, 0);
882 atomic_set(&dest->persistconns, 0);
Julian Anastasov26c15cf2010-09-21 18:12:30 +0200883 atomic_set(&dest->refcnt, 1);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700884
885 INIT_LIST_HEAD(&dest->d_list);
886 spin_lock_init(&dest->dst_lock);
887 spin_lock_init(&dest->stats.lock);
Julian Anastasov26c15cf2010-09-21 18:12:30 +0200888 __ip_vs_update_dest(svc, dest, udest, 1);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700889
890 *dest_p = dest;
891
892 LeaveFunction(2);
893 return 0;
Hans Schillstromb17fc992011-01-03 14:44:56 +0100894
895err_alloc:
896 kfree(dest);
897 return -ENOMEM;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700898}
899
900
901/*
902 * Add a destination into an existing service
903 */
904static int
Julius Volzc860c6b2008-09-02 15:55:36 +0200905ip_vs_add_dest(struct ip_vs_service *svc, struct ip_vs_dest_user_kern *udest)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700906{
907 struct ip_vs_dest *dest;
Julius Volzc860c6b2008-09-02 15:55:36 +0200908 union nf_inet_addr daddr;
Al Viro014d7302006-09-28 14:29:52 -0700909 __be16 dport = udest->port;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700910 int ret;
911
912 EnterFunction(2);
913
914 if (udest->weight < 0) {
Hannes Eder1e3e2382009-08-02 11:05:41 +0000915 pr_err("%s(): server weight less than zero\n", __func__);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700916 return -ERANGE;
917 }
918
919 if (udest->l_threshold > udest->u_threshold) {
Hannes Eder1e3e2382009-08-02 11:05:41 +0000920 pr_err("%s(): lower threshold is higher than upper threshold\n",
921 __func__);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700922 return -ERANGE;
923 }
924
Julius Volzc860c6b2008-09-02 15:55:36 +0200925 ip_vs_addr_copy(svc->af, &daddr, &udest->addr);
926
Linus Torvalds1da177e2005-04-16 15:20:36 -0700927 /*
928 * Check if the dest already exists in the list
929 */
Julius Volz7937df12008-09-02 15:55:48 +0200930 dest = ip_vs_lookup_dest(svc, &daddr, dport);
931
Linus Torvalds1da177e2005-04-16 15:20:36 -0700932 if (dest != NULL) {
Hannes Eder1e3e2382009-08-02 11:05:41 +0000933 IP_VS_DBG(1, "%s(): dest already exists\n", __func__);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700934 return -EEXIST;
935 }
936
937 /*
938 * Check if the dest already exists in the trash and
939 * is from the same service
940 */
Julius Volz7937df12008-09-02 15:55:48 +0200941 dest = ip_vs_trash_get_dest(svc, &daddr, dport);
942
Linus Torvalds1da177e2005-04-16 15:20:36 -0700943 if (dest != NULL) {
Julius Volzcfc78c52008-09-02 15:55:53 +0200944 IP_VS_DBG_BUF(3, "Get destination %s:%u from trash, "
945 "dest->refcnt=%d, service %u/%s:%u\n",
946 IP_VS_DBG_ADDR(svc->af, &daddr), ntohs(dport),
947 atomic_read(&dest->refcnt),
948 dest->vfwmark,
949 IP_VS_DBG_ADDR(svc->af, &dest->vaddr),
950 ntohs(dest->vport));
951
Linus Torvalds1da177e2005-04-16 15:20:36 -0700952 /*
953 * Get the destination from the trash
954 */
955 list_del(&dest->n_list);
956
Julian Anastasov26c15cf2010-09-21 18:12:30 +0200957 __ip_vs_update_dest(svc, dest, udest, 1);
958 ret = 0;
959 } else {
Linus Torvalds1da177e2005-04-16 15:20:36 -0700960 /*
Julian Anastasov26c15cf2010-09-21 18:12:30 +0200961 * Allocate and initialize the dest structure
Linus Torvalds1da177e2005-04-16 15:20:36 -0700962 */
Julian Anastasov26c15cf2010-09-21 18:12:30 +0200963 ret = ip_vs_new_dest(svc, udest, &dest);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700964 }
Linus Torvalds1da177e2005-04-16 15:20:36 -0700965 LeaveFunction(2);
966
Julian Anastasov26c15cf2010-09-21 18:12:30 +0200967 return ret;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700968}
969
970
971/*
972 * Edit a destination in the given service
973 */
974static int
Julius Volzc860c6b2008-09-02 15:55:36 +0200975ip_vs_edit_dest(struct ip_vs_service *svc, struct ip_vs_dest_user_kern *udest)
Linus Torvalds1da177e2005-04-16 15:20:36 -0700976{
977 struct ip_vs_dest *dest;
Julius Volzc860c6b2008-09-02 15:55:36 +0200978 union nf_inet_addr daddr;
Al Viro014d7302006-09-28 14:29:52 -0700979 __be16 dport = udest->port;
Linus Torvalds1da177e2005-04-16 15:20:36 -0700980
981 EnterFunction(2);
982
983 if (udest->weight < 0) {
Hannes Eder1e3e2382009-08-02 11:05:41 +0000984 pr_err("%s(): server weight less than zero\n", __func__);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700985 return -ERANGE;
986 }
987
988 if (udest->l_threshold > udest->u_threshold) {
Hannes Eder1e3e2382009-08-02 11:05:41 +0000989 pr_err("%s(): lower threshold is higher than upper threshold\n",
990 __func__);
Linus Torvalds1da177e2005-04-16 15:20:36 -0700991 return -ERANGE;
992 }
993
Julius Volzc860c6b2008-09-02 15:55:36 +0200994 ip_vs_addr_copy(svc->af, &daddr, &udest->addr);
995
Linus Torvalds1da177e2005-04-16 15:20:36 -0700996 /*
997 * Lookup the destination list
998 */
Julius Volz7937df12008-09-02 15:55:48 +0200999 dest = ip_vs_lookup_dest(svc, &daddr, dport);
1000
Linus Torvalds1da177e2005-04-16 15:20:36 -07001001 if (dest == NULL) {
Hannes Eder1e3e2382009-08-02 11:05:41 +00001002 IP_VS_DBG(1, "%s(): dest doesn't exist\n", __func__);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001003 return -ENOENT;
1004 }
1005
Julian Anastasov26c15cf2010-09-21 18:12:30 +02001006 __ip_vs_update_dest(svc, dest, udest, 0);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001007 LeaveFunction(2);
1008
1009 return 0;
1010}
1011
1012
1013/*
1014 * Delete a destination (must be already unlinked from the service)
1015 */
Hans Schillstrom29c20262011-01-03 14:44:54 +01001016static void __ip_vs_del_dest(struct net *net, struct ip_vs_dest *dest)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001017{
Hans Schillstroma0840e22011-01-03 14:44:58 +01001018 struct netns_ipvs *ipvs = net_ipvs(net);
1019
Julian Anastasov6ef757f2011-03-14 01:44:28 +02001020 ip_vs_stop_estimator(net, &dest->stats);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001021
1022 /*
1023 * Remove it from the d-linked list with the real services.
1024 */
Hans Schillstroma0840e22011-01-03 14:44:58 +01001025 write_lock_bh(&ipvs->rs_lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001026 ip_vs_rs_unhash(dest);
Hans Schillstroma0840e22011-01-03 14:44:58 +01001027 write_unlock_bh(&ipvs->rs_lock);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001028
1029 /*
1030 * Decrease the refcnt of the dest, and free the dest
1031 * if nobody refers to it (refcnt=0). Otherwise, throw
1032 * the destination into the trash.
1033 */
1034 if (atomic_dec_and_test(&dest->refcnt)) {
Julian Anastasov26c15cf2010-09-21 18:12:30 +02001035 IP_VS_DBG_BUF(3, "Removing destination %u/%s:%u\n",
1036 dest->vfwmark,
1037 IP_VS_DBG_ADDR(dest->af, &dest->addr),
1038 ntohs(dest->port));
Linus Torvalds1da177e2005-04-16 15:20:36 -07001039 ip_vs_dst_reset(dest);
1040 /* simply decrease svc->refcnt here, let the caller check
1041 and release the service if nobody refers to it.
1042 Only user context can release destination and service,
1043 and only one user context can update virtual service at a
1044 time, so the operation here is OK */
1045 atomic_dec(&dest->svc->refcnt);
Hans Schillstromb17fc992011-01-03 14:44:56 +01001046 free_percpu(dest->stats.cpustats);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001047 kfree(dest);
1048 } else {
Julius Volzcfc78c52008-09-02 15:55:53 +02001049 IP_VS_DBG_BUF(3, "Moving dest %s:%u into trash, "
1050 "dest->refcnt=%d\n",
1051 IP_VS_DBG_ADDR(dest->af, &dest->addr),
1052 ntohs(dest->port),
1053 atomic_read(&dest->refcnt));
Hans Schillstromf2431e62011-01-03 14:45:00 +01001054 list_add(&dest->n_list, &ipvs->dest_trash);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001055 atomic_inc(&dest->refcnt);
1056 }
1057}
1058
1059
1060/*
1061 * Unlink a destination from the given service
1062 */
1063static void __ip_vs_unlink_dest(struct ip_vs_service *svc,
1064 struct ip_vs_dest *dest,
1065 int svcupd)
1066{
1067 dest->flags &= ~IP_VS_DEST_F_AVAILABLE;
1068
1069 /*
1070 * Remove it from the d-linked destination list.
1071 */
1072 list_del(&dest->n_list);
1073 svc->num_dests--;
Sven Wegener82dfb6f2008-08-11 19:36:06 +00001074
1075 /*
1076 * Call the update_service function of its scheduler
1077 */
1078 if (svcupd && svc->scheduler->update_service)
1079 svc->scheduler->update_service(svc);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001080}
1081
1082
1083/*
1084 * Delete a destination server in the given service
1085 */
1086static int
Julius Volzc860c6b2008-09-02 15:55:36 +02001087ip_vs_del_dest(struct ip_vs_service *svc, struct ip_vs_dest_user_kern *udest)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001088{
1089 struct ip_vs_dest *dest;
Al Viro014d7302006-09-28 14:29:52 -07001090 __be16 dport = udest->port;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001091
1092 EnterFunction(2);
1093
Julius Volz7937df12008-09-02 15:55:48 +02001094 dest = ip_vs_lookup_dest(svc, &udest->addr, dport);
Julius Volzc860c6b2008-09-02 15:55:36 +02001095
Linus Torvalds1da177e2005-04-16 15:20:36 -07001096 if (dest == NULL) {
Hannes Eder1e3e2382009-08-02 11:05:41 +00001097 IP_VS_DBG(1, "%s(): destination not found!\n", __func__);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001098 return -ENOENT;
1099 }
1100
1101 write_lock_bh(&__ip_vs_svc_lock);
1102
1103 /*
1104 * Wait until all other svc users go away.
1105 */
Julian Anastasov26c15cf2010-09-21 18:12:30 +02001106 IP_VS_WAIT_WHILE(atomic_read(&svc->usecnt) > 0);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001107
1108 /*
1109 * Unlink dest from the service
1110 */
1111 __ip_vs_unlink_dest(svc, dest, 1);
1112
1113 write_unlock_bh(&__ip_vs_svc_lock);
1114
1115 /*
1116 * Delete the destination
1117 */
Hans Schillstroma0840e22011-01-03 14:44:58 +01001118 __ip_vs_del_dest(svc->net, dest);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001119
1120 LeaveFunction(2);
1121
1122 return 0;
1123}
1124
1125
1126/*
1127 * Add a service into the service hash table
1128 */
1129static int
Hans Schillstromfc723252011-01-03 14:44:43 +01001130ip_vs_add_service(struct net *net, struct ip_vs_service_user_kern *u,
Julius Volzc860c6b2008-09-02 15:55:36 +02001131 struct ip_vs_service **svc_p)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001132{
1133 int ret = 0;
1134 struct ip_vs_scheduler *sched = NULL;
Simon Horman0d1e71b2010-08-22 21:37:54 +09001135 struct ip_vs_pe *pe = NULL;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001136 struct ip_vs_service *svc = NULL;
Hans Schillstroma0840e22011-01-03 14:44:58 +01001137 struct netns_ipvs *ipvs = net_ipvs(net);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001138
1139 /* increase the module use count */
1140 ip_vs_use_count_inc();
1141
1142 /* Lookup the scheduler by 'u->sched_name' */
1143 sched = ip_vs_scheduler_get(u->sched_name);
1144 if (sched == NULL) {
Hannes Eder1e3e2382009-08-02 11:05:41 +00001145 pr_info("Scheduler module ip_vs_%s not found\n", u->sched_name);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001146 ret = -ENOENT;
Simon Horman6e08bfb2010-08-22 21:37:52 +09001147 goto out_err;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001148 }
1149
Simon Horman0d1e71b2010-08-22 21:37:54 +09001150 if (u->pe_name && *u->pe_name) {
Simon Hormane9e5eee2010-11-08 20:05:57 +09001151 pe = ip_vs_pe_getbyname(u->pe_name);
Simon Horman0d1e71b2010-08-22 21:37:54 +09001152 if (pe == NULL) {
1153 pr_info("persistence engine module ip_vs_pe_%s "
1154 "not found\n", u->pe_name);
1155 ret = -ENOENT;
1156 goto out_err;
1157 }
1158 }
1159
Julius Volzf94fd042008-09-02 15:55:55 +02001160#ifdef CONFIG_IP_VS_IPV6
Julius Volz48148932008-11-03 17:08:56 -08001161 if (u->af == AF_INET6 && (u->netmask < 1 || u->netmask > 128)) {
1162 ret = -EINVAL;
1163 goto out_err;
Julius Volzf94fd042008-09-02 15:55:55 +02001164 }
1165#endif
1166
Simon Hormandee06e42010-08-26 02:54:31 +00001167 svc = kzalloc(sizeof(struct ip_vs_service), GFP_KERNEL);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001168 if (svc == NULL) {
Hannes Eder1e3e2382009-08-02 11:05:41 +00001169 IP_VS_DBG(1, "%s(): no memory\n", __func__);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001170 ret = -ENOMEM;
1171 goto out_err;
1172 }
Hans Schillstromb17fc992011-01-03 14:44:56 +01001173 svc->stats.cpustats = alloc_percpu(struct ip_vs_cpu_stats);
Joe Perches0a9ee812011-08-29 14:17:25 -07001174 if (!svc->stats.cpustats)
Hans Schillstromb17fc992011-01-03 14:44:56 +01001175 goto out_err;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001176
1177 /* I'm the first user of the service */
Julian Anastasov26c15cf2010-09-21 18:12:30 +02001178 atomic_set(&svc->usecnt, 0);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001179 atomic_set(&svc->refcnt, 0);
1180
Julius Volzc860c6b2008-09-02 15:55:36 +02001181 svc->af = u->af;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001182 svc->protocol = u->protocol;
Julius Volzc860c6b2008-09-02 15:55:36 +02001183 ip_vs_addr_copy(svc->af, &svc->addr, &u->addr);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001184 svc->port = u->port;
1185 svc->fwmark = u->fwmark;
1186 svc->flags = u->flags;
1187 svc->timeout = u->timeout * HZ;
1188 svc->netmask = u->netmask;
Hans Schillstromfc723252011-01-03 14:44:43 +01001189 svc->net = net;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001190
1191 INIT_LIST_HEAD(&svc->destinations);
1192 rwlock_init(&svc->sched_lock);
1193 spin_lock_init(&svc->stats.lock);
1194
1195 /* Bind the scheduler */
1196 ret = ip_vs_bind_scheduler(svc, sched);
1197 if (ret)
1198 goto out_err;
1199 sched = NULL;
1200
Simon Horman0d1e71b2010-08-22 21:37:54 +09001201 /* Bind the ct retriever */
1202 ip_vs_bind_pe(svc, pe);
1203 pe = NULL;
1204
Linus Torvalds1da177e2005-04-16 15:20:36 -07001205 /* Update the virtual service counters */
1206 if (svc->port == FTPPORT)
Hans Schillstrom763f8d02011-01-03 14:45:01 +01001207 atomic_inc(&ipvs->ftpsvc_counter);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001208 else if (svc->port == 0)
Hans Schillstrom763f8d02011-01-03 14:45:01 +01001209 atomic_inc(&ipvs->nullsvc_counter);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001210
Julian Anastasov6ef757f2011-03-14 01:44:28 +02001211 ip_vs_start_estimator(net, &svc->stats);
Julius Volzf94fd042008-09-02 15:55:55 +02001212
1213 /* Count only IPv4 services for old get/setsockopt interface */
1214 if (svc->af == AF_INET)
Hans Schillstroma0840e22011-01-03 14:44:58 +01001215 ipvs->num_services++;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001216
1217 /* Hash the service into the service table */
1218 write_lock_bh(&__ip_vs_svc_lock);
1219 ip_vs_svc_hash(svc);
1220 write_unlock_bh(&__ip_vs_svc_lock);
1221
1222 *svc_p = svc;
Hans Schillstrom7a4f0762011-05-03 22:09:31 +02001223 /* Now there is a service - full throttle */
1224 ipvs->enable = 1;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001225 return 0;
1226
Hans Schillstromb17fc992011-01-03 14:44:56 +01001227
Simon Horman6e08bfb2010-08-22 21:37:52 +09001228 out_err:
Linus Torvalds1da177e2005-04-16 15:20:36 -07001229 if (svc != NULL) {
Simon Horman2fabf352010-08-22 21:37:52 +09001230 ip_vs_unbind_scheduler(svc);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001231 if (svc->inc) {
1232 local_bh_disable();
1233 ip_vs_app_inc_put(svc->inc);
1234 local_bh_enable();
1235 }
Hans Schillstromb17fc992011-01-03 14:44:56 +01001236 if (svc->stats.cpustats)
1237 free_percpu(svc->stats.cpustats);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001238 kfree(svc);
1239 }
1240 ip_vs_scheduler_put(sched);
Simon Horman0d1e71b2010-08-22 21:37:54 +09001241 ip_vs_pe_put(pe);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001242
Linus Torvalds1da177e2005-04-16 15:20:36 -07001243 /* decrease the module use count */
1244 ip_vs_use_count_dec();
1245
1246 return ret;
1247}
1248
1249
1250/*
1251 * Edit a service and bind it with a new scheduler
1252 */
1253static int
Julius Volzc860c6b2008-09-02 15:55:36 +02001254ip_vs_edit_service(struct ip_vs_service *svc, struct ip_vs_service_user_kern *u)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001255{
1256 struct ip_vs_scheduler *sched, *old_sched;
Simon Horman0d1e71b2010-08-22 21:37:54 +09001257 struct ip_vs_pe *pe = NULL, *old_pe = NULL;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001258 int ret = 0;
1259
1260 /*
1261 * Lookup the scheduler, by 'u->sched_name'
1262 */
1263 sched = ip_vs_scheduler_get(u->sched_name);
1264 if (sched == NULL) {
Hannes Eder1e3e2382009-08-02 11:05:41 +00001265 pr_info("Scheduler module ip_vs_%s not found\n", u->sched_name);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001266 return -ENOENT;
1267 }
1268 old_sched = sched;
1269
Simon Horman0d1e71b2010-08-22 21:37:54 +09001270 if (u->pe_name && *u->pe_name) {
Simon Hormane9e5eee2010-11-08 20:05:57 +09001271 pe = ip_vs_pe_getbyname(u->pe_name);
Simon Horman0d1e71b2010-08-22 21:37:54 +09001272 if (pe == NULL) {
1273 pr_info("persistence engine module ip_vs_pe_%s "
1274 "not found\n", u->pe_name);
1275 ret = -ENOENT;
1276 goto out;
1277 }
1278 old_pe = pe;
1279 }
1280
Julius Volzf94fd042008-09-02 15:55:55 +02001281#ifdef CONFIG_IP_VS_IPV6
Julius Volz48148932008-11-03 17:08:56 -08001282 if (u->af == AF_INET6 && (u->netmask < 1 || u->netmask > 128)) {
1283 ret = -EINVAL;
1284 goto out;
Julius Volzf94fd042008-09-02 15:55:55 +02001285 }
1286#endif
1287
Linus Torvalds1da177e2005-04-16 15:20:36 -07001288 write_lock_bh(&__ip_vs_svc_lock);
1289
1290 /*
1291 * Wait until all other svc users go away.
1292 */
Julian Anastasov26c15cf2010-09-21 18:12:30 +02001293 IP_VS_WAIT_WHILE(atomic_read(&svc->usecnt) > 0);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001294
1295 /*
1296 * Set the flags and timeout value
1297 */
1298 svc->flags = u->flags | IP_VS_SVC_F_HASHED;
1299 svc->timeout = u->timeout * HZ;
1300 svc->netmask = u->netmask;
1301
1302 old_sched = svc->scheduler;
1303 if (sched != old_sched) {
1304 /*
1305 * Unbind the old scheduler
1306 */
1307 if ((ret = ip_vs_unbind_scheduler(svc))) {
1308 old_sched = sched;
Simon Horman9e691ed2008-09-17 10:10:41 +10001309 goto out_unlock;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001310 }
1311
1312 /*
1313 * Bind the new scheduler
1314 */
1315 if ((ret = ip_vs_bind_scheduler(svc, sched))) {
1316 /*
1317 * If ip_vs_bind_scheduler fails, restore the old
1318 * scheduler.
1319 * The main reason of failure is out of memory.
1320 *
1321 * The question is if the old scheduler can be
1322 * restored all the time. TODO: if it cannot be
1323 * restored some time, we must delete the service,
1324 * otherwise the system may crash.
1325 */
1326 ip_vs_bind_scheduler(svc, old_sched);
1327 old_sched = sched;
Simon Horman9e691ed2008-09-17 10:10:41 +10001328 goto out_unlock;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001329 }
1330 }
1331
Simon Horman0d1e71b2010-08-22 21:37:54 +09001332 old_pe = svc->pe;
1333 if (pe != old_pe) {
1334 ip_vs_unbind_pe(svc);
1335 ip_vs_bind_pe(svc, pe);
1336 }
1337
Hans Schillstrom552ad652011-06-13 12:19:26 +02001338out_unlock:
Linus Torvalds1da177e2005-04-16 15:20:36 -07001339 write_unlock_bh(&__ip_vs_svc_lock);
Hans Schillstrom552ad652011-06-13 12:19:26 +02001340out:
Simon Horman6e08bfb2010-08-22 21:37:52 +09001341 ip_vs_scheduler_put(old_sched);
Simon Horman0d1e71b2010-08-22 21:37:54 +09001342 ip_vs_pe_put(old_pe);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001343 return ret;
1344}
1345
1346
1347/*
1348 * Delete a service from the service list
1349 * - The service must be unlinked, unlocked and not referenced!
1350 * - We are called under _bh lock
1351 */
1352static void __ip_vs_del_service(struct ip_vs_service *svc)
1353{
1354 struct ip_vs_dest *dest, *nxt;
1355 struct ip_vs_scheduler *old_sched;
Simon Horman0d1e71b2010-08-22 21:37:54 +09001356 struct ip_vs_pe *old_pe;
Hans Schillstroma0840e22011-01-03 14:44:58 +01001357 struct netns_ipvs *ipvs = net_ipvs(svc->net);
Simon Horman0d1e71b2010-08-22 21:37:54 +09001358
1359 pr_info("%s: enter\n", __func__);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001360
Julius Volzf94fd042008-09-02 15:55:55 +02001361 /* Count only IPv4 services for old get/setsockopt interface */
1362 if (svc->af == AF_INET)
Hans Schillstroma0840e22011-01-03 14:44:58 +01001363 ipvs->num_services--;
Julius Volzf94fd042008-09-02 15:55:55 +02001364
Julian Anastasov6ef757f2011-03-14 01:44:28 +02001365 ip_vs_stop_estimator(svc->net, &svc->stats);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001366
1367 /* Unbind scheduler */
1368 old_sched = svc->scheduler;
1369 ip_vs_unbind_scheduler(svc);
Simon Horman6e08bfb2010-08-22 21:37:52 +09001370 ip_vs_scheduler_put(old_sched);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001371
Simon Horman0d1e71b2010-08-22 21:37:54 +09001372 /* Unbind persistence engine */
1373 old_pe = svc->pe;
1374 ip_vs_unbind_pe(svc);
1375 ip_vs_pe_put(old_pe);
1376
Linus Torvalds1da177e2005-04-16 15:20:36 -07001377 /* Unbind app inc */
1378 if (svc->inc) {
1379 ip_vs_app_inc_put(svc->inc);
1380 svc->inc = NULL;
1381 }
1382
1383 /*
1384 * Unlink the whole destination list
1385 */
1386 list_for_each_entry_safe(dest, nxt, &svc->destinations, n_list) {
1387 __ip_vs_unlink_dest(svc, dest, 0);
Hans Schillstrom29c20262011-01-03 14:44:54 +01001388 __ip_vs_del_dest(svc->net, dest);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001389 }
1390
1391 /*
1392 * Update the virtual service counters
1393 */
1394 if (svc->port == FTPPORT)
Hans Schillstrom763f8d02011-01-03 14:45:01 +01001395 atomic_dec(&ipvs->ftpsvc_counter);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001396 else if (svc->port == 0)
Hans Schillstrom763f8d02011-01-03 14:45:01 +01001397 atomic_dec(&ipvs->nullsvc_counter);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001398
1399 /*
1400 * Free the service if nobody refers to it
1401 */
Julian Anastasov26c15cf2010-09-21 18:12:30 +02001402 if (atomic_read(&svc->refcnt) == 0) {
1403 IP_VS_DBG_BUF(3, "Removing service %u/%s:%u usecnt=%d\n",
1404 svc->fwmark,
1405 IP_VS_DBG_ADDR(svc->af, &svc->addr),
1406 ntohs(svc->port), atomic_read(&svc->usecnt));
Hans Schillstromb17fc992011-01-03 14:44:56 +01001407 free_percpu(svc->stats.cpustats);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001408 kfree(svc);
Julian Anastasov26c15cf2010-09-21 18:12:30 +02001409 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07001410
1411 /* decrease the module use count */
1412 ip_vs_use_count_dec();
1413}
1414
1415/*
Julian Anastasov26c15cf2010-09-21 18:12:30 +02001416 * Unlink a service from list and try to delete it if its refcnt reached 0
Linus Torvalds1da177e2005-04-16 15:20:36 -07001417 */
Julian Anastasov26c15cf2010-09-21 18:12:30 +02001418static void ip_vs_unlink_service(struct ip_vs_service *svc)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001419{
Linus Torvalds1da177e2005-04-16 15:20:36 -07001420 /*
1421 * Unhash it from the service table
1422 */
1423 write_lock_bh(&__ip_vs_svc_lock);
1424
1425 ip_vs_svc_unhash(svc);
1426
1427 /*
1428 * Wait until all the svc users go away.
1429 */
Julian Anastasov26c15cf2010-09-21 18:12:30 +02001430 IP_VS_WAIT_WHILE(atomic_read(&svc->usecnt) > 0);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001431
1432 __ip_vs_del_service(svc);
1433
1434 write_unlock_bh(&__ip_vs_svc_lock);
Julian Anastasov26c15cf2010-09-21 18:12:30 +02001435}
1436
1437/*
1438 * Delete a service from the service list
1439 */
1440static int ip_vs_del_service(struct ip_vs_service *svc)
1441{
1442 if (svc == NULL)
1443 return -EEXIST;
1444 ip_vs_unlink_service(svc);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001445
1446 return 0;
1447}
1448
1449
1450/*
1451 * Flush all the virtual services
1452 */
Hans Schillstromfc723252011-01-03 14:44:43 +01001453static int ip_vs_flush(struct net *net)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001454{
1455 int idx;
1456 struct ip_vs_service *svc, *nxt;
1457
1458 /*
Hans Schillstromfc723252011-01-03 14:44:43 +01001459 * Flush the service table hashed by <netns,protocol,addr,port>
Linus Torvalds1da177e2005-04-16 15:20:36 -07001460 */
1461 for(idx = 0; idx < IP_VS_SVC_TAB_SIZE; idx++) {
Hans Schillstromfc723252011-01-03 14:44:43 +01001462 list_for_each_entry_safe(svc, nxt, &ip_vs_svc_table[idx],
1463 s_list) {
1464 if (net_eq(svc->net, net))
1465 ip_vs_unlink_service(svc);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001466 }
1467 }
1468
1469 /*
1470 * Flush the service table hashed by fwmark
1471 */
1472 for(idx = 0; idx < IP_VS_SVC_TAB_SIZE; idx++) {
1473 list_for_each_entry_safe(svc, nxt,
1474 &ip_vs_svc_fwm_table[idx], f_list) {
Hans Schillstromfc723252011-01-03 14:44:43 +01001475 if (net_eq(svc->net, net))
1476 ip_vs_unlink_service(svc);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001477 }
1478 }
1479
1480 return 0;
1481}
1482
Hans Schillstrom7a4f0762011-05-03 22:09:31 +02001483/*
1484 * Delete service by {netns} in the service table.
1485 * Called by __ip_vs_cleanup()
1486 */
Hans Schillstrom503cf152011-05-01 18:50:16 +02001487void ip_vs_service_net_cleanup(struct net *net)
Hans Schillstrom7a4f0762011-05-03 22:09:31 +02001488{
1489 EnterFunction(2);
1490 /* Check for "full" addressed entries */
1491 mutex_lock(&__ip_vs_mutex);
1492 ip_vs_flush(net);
1493 mutex_unlock(&__ip_vs_mutex);
1494 LeaveFunction(2);
1495}
1496/*
1497 * Release dst hold by dst_cache
1498 */
1499static inline void
1500__ip_vs_dev_reset(struct ip_vs_dest *dest, struct net_device *dev)
1501{
1502 spin_lock_bh(&dest->dst_lock);
1503 if (dest->dst_cache && dest->dst_cache->dev == dev) {
1504 IP_VS_DBG_BUF(3, "Reset dev:%s dest %s:%u ,dest->refcnt=%d\n",
1505 dev->name,
1506 IP_VS_DBG_ADDR(dest->af, &dest->addr),
1507 ntohs(dest->port),
1508 atomic_read(&dest->refcnt));
1509 ip_vs_dst_reset(dest);
1510 }
1511 spin_unlock_bh(&dest->dst_lock);
1512
1513}
1514/*
1515 * Netdev event receiver
1516 * Currently only NETDEV_UNREGISTER is handled, i.e. if we hold a reference to
1517 * a device that is "unregister" it must be released.
1518 */
1519static int ip_vs_dst_event(struct notifier_block *this, unsigned long event,
1520 void *ptr)
1521{
1522 struct net_device *dev = ptr;
1523 struct net *net = dev_net(dev);
1524 struct ip_vs_service *svc;
1525 struct ip_vs_dest *dest;
1526 unsigned int idx;
1527
1528 if (event != NETDEV_UNREGISTER)
1529 return NOTIFY_DONE;
1530 IP_VS_DBG(3, "%s() dev=%s\n", __func__, dev->name);
1531 EnterFunction(2);
1532 mutex_lock(&__ip_vs_mutex);
1533 for (idx = 0; idx < IP_VS_SVC_TAB_SIZE; idx++) {
1534 list_for_each_entry(svc, &ip_vs_svc_table[idx], s_list) {
1535 if (net_eq(svc->net, net)) {
1536 list_for_each_entry(dest, &svc->destinations,
1537 n_list) {
1538 __ip_vs_dev_reset(dest, dev);
1539 }
1540 }
1541 }
1542
1543 list_for_each_entry(svc, &ip_vs_svc_fwm_table[idx], f_list) {
1544 if (net_eq(svc->net, net)) {
1545 list_for_each_entry(dest, &svc->destinations,
1546 n_list) {
1547 __ip_vs_dev_reset(dest, dev);
1548 }
1549 }
1550
1551 }
1552 }
1553
1554 list_for_each_entry(dest, &net_ipvs(net)->dest_trash, n_list) {
1555 __ip_vs_dev_reset(dest, dev);
1556 }
1557 mutex_unlock(&__ip_vs_mutex);
1558 LeaveFunction(2);
1559 return NOTIFY_DONE;
1560}
Linus Torvalds1da177e2005-04-16 15:20:36 -07001561
1562/*
1563 * Zero counters in a service or all services
1564 */
1565static int ip_vs_zero_service(struct ip_vs_service *svc)
1566{
1567 struct ip_vs_dest *dest;
1568
1569 write_lock_bh(&__ip_vs_svc_lock);
1570 list_for_each_entry(dest, &svc->destinations, n_list) {
1571 ip_vs_zero_stats(&dest->stats);
1572 }
1573 ip_vs_zero_stats(&svc->stats);
1574 write_unlock_bh(&__ip_vs_svc_lock);
1575 return 0;
1576}
1577
Hans Schillstromfc723252011-01-03 14:44:43 +01001578static int ip_vs_zero_all(struct net *net)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001579{
1580 int idx;
1581 struct ip_vs_service *svc;
1582
1583 for(idx = 0; idx < IP_VS_SVC_TAB_SIZE; idx++) {
1584 list_for_each_entry(svc, &ip_vs_svc_table[idx], s_list) {
Hans Schillstromfc723252011-01-03 14:44:43 +01001585 if (net_eq(svc->net, net))
1586 ip_vs_zero_service(svc);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001587 }
1588 }
1589
1590 for(idx = 0; idx < IP_VS_SVC_TAB_SIZE; idx++) {
1591 list_for_each_entry(svc, &ip_vs_svc_fwm_table[idx], f_list) {
Hans Schillstromfc723252011-01-03 14:44:43 +01001592 if (net_eq(svc->net, net))
1593 ip_vs_zero_service(svc);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001594 }
1595 }
1596
Julian Anastasov2a0751a2011-03-04 12:20:35 +02001597 ip_vs_zero_stats(&net_ipvs(net)->tot_stats);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001598 return 0;
1599}
1600
Simon Horman14e40542011-02-04 18:33:02 +09001601#ifdef CONFIG_SYSCTL
Linus Torvalds1da177e2005-04-16 15:20:36 -07001602static int
Alexey Dobriyan8d65af72009-09-23 15:57:19 -07001603proc_do_defense_mode(ctl_table *table, int write,
Linus Torvalds1da177e2005-04-16 15:20:36 -07001604 void __user *buffer, size_t *lenp, loff_t *ppos)
1605{
Hans Schillstrom93304192011-01-03 14:44:51 +01001606 struct net *net = current->nsproxy->net_ns;
Linus Torvalds1da177e2005-04-16 15:20:36 -07001607 int *valp = table->data;
1608 int val = *valp;
1609 int rc;
1610
Alexey Dobriyan8d65af72009-09-23 15:57:19 -07001611 rc = proc_dointvec(table, write, buffer, lenp, ppos);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001612 if (write && (*valp != val)) {
1613 if ((*valp < 0) || (*valp > 3)) {
1614 /* Restore the correct value */
1615 *valp = val;
1616 } else {
Hans Schillstrom93304192011-01-03 14:44:51 +01001617 update_defense_level(net_ipvs(net));
Linus Torvalds1da177e2005-04-16 15:20:36 -07001618 }
1619 }
1620 return rc;
1621}
1622
Linus Torvalds1da177e2005-04-16 15:20:36 -07001623static int
Alexey Dobriyan8d65af72009-09-23 15:57:19 -07001624proc_do_sync_threshold(ctl_table *table, int write,
Linus Torvalds1da177e2005-04-16 15:20:36 -07001625 void __user *buffer, size_t *lenp, loff_t *ppos)
1626{
1627 int *valp = table->data;
1628 int val[2];
1629 int rc;
1630
1631 /* backup the value first */
1632 memcpy(val, valp, sizeof(val));
1633
Alexey Dobriyan8d65af72009-09-23 15:57:19 -07001634 rc = proc_dointvec(table, write, buffer, lenp, ppos);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001635 if (write && (valp[0] < 0 || valp[1] < 0 || valp[0] >= valp[1])) {
1636 /* Restore the correct value */
1637 memcpy(valp, val, sizeof(val));
1638 }
1639 return rc;
1640}
1641
Hans Schillstromb880c1f2010-11-19 14:25:14 +01001642static int
1643proc_do_sync_mode(ctl_table *table, int write,
1644 void __user *buffer, size_t *lenp, loff_t *ppos)
1645{
1646 int *valp = table->data;
1647 int val = *valp;
1648 int rc;
1649
1650 rc = proc_dointvec(table, write, buffer, lenp, ppos);
1651 if (write && (*valp != val)) {
1652 if ((*valp < 0) || (*valp > 1)) {
1653 /* Restore the correct value */
1654 *valp = val;
1655 } else {
Hans Schillstromf1313152011-01-03 14:44:55 +01001656 struct net *net = current->nsproxy->net_ns;
1657 ip_vs_sync_switch_mode(net, val);
Hans Schillstromb880c1f2010-11-19 14:25:14 +01001658 }
1659 }
1660 return rc;
1661}
Linus Torvalds1da177e2005-04-16 15:20:36 -07001662
1663/*
1664 * IPVS sysctl table (under the /proc/sys/net/ipv4/vs/)
Hans Schillstroma0840e22011-01-03 14:44:58 +01001665 * Do not change order or insert new entries without
Hans Schillstrom503cf152011-05-01 18:50:16 +02001666 * align with netns init in ip_vs_control_net_init()
Linus Torvalds1da177e2005-04-16 15:20:36 -07001667 */
1668
1669static struct ctl_table vs_vars[] = {
1670 {
Linus Torvalds1da177e2005-04-16 15:20:36 -07001671 .procname = "amemthresh",
Hans Schillstroma0840e22011-01-03 14:44:58 +01001672 .maxlen = sizeof(int),
1673 .mode = 0644,
1674 .proc_handler = proc_dointvec,
1675 },
1676 {
1677 .procname = "am_droprate",
1678 .maxlen = sizeof(int),
1679 .mode = 0644,
1680 .proc_handler = proc_dointvec,
1681 },
1682 {
1683 .procname = "drop_entry",
1684 .maxlen = sizeof(int),
1685 .mode = 0644,
1686 .proc_handler = proc_do_defense_mode,
1687 },
1688 {
1689 .procname = "drop_packet",
1690 .maxlen = sizeof(int),
1691 .mode = 0644,
1692 .proc_handler = proc_do_defense_mode,
1693 },
1694#ifdef CONFIG_IP_VS_NFCT
1695 {
1696 .procname = "conntrack",
1697 .maxlen = sizeof(int),
1698 .mode = 0644,
1699 .proc_handler = &proc_dointvec,
1700 },
1701#endif
1702 {
1703 .procname = "secure_tcp",
1704 .maxlen = sizeof(int),
1705 .mode = 0644,
1706 .proc_handler = proc_do_defense_mode,
1707 },
1708 {
1709 .procname = "snat_reroute",
1710 .maxlen = sizeof(int),
1711 .mode = 0644,
1712 .proc_handler = &proc_dointvec,
1713 },
1714 {
1715 .procname = "sync_version",
1716 .maxlen = sizeof(int),
1717 .mode = 0644,
1718 .proc_handler = &proc_do_sync_mode,
1719 },
1720 {
1721 .procname = "cache_bypass",
1722 .maxlen = sizeof(int),
1723 .mode = 0644,
1724 .proc_handler = proc_dointvec,
1725 },
1726 {
1727 .procname = "expire_nodest_conn",
1728 .maxlen = sizeof(int),
1729 .mode = 0644,
1730 .proc_handler = proc_dointvec,
1731 },
1732 {
1733 .procname = "expire_quiescent_template",
1734 .maxlen = sizeof(int),
1735 .mode = 0644,
1736 .proc_handler = proc_dointvec,
1737 },
1738 {
1739 .procname = "sync_threshold",
1740 .maxlen =
1741 sizeof(((struct netns_ipvs *)0)->sysctl_sync_threshold),
1742 .mode = 0644,
1743 .proc_handler = proc_do_sync_threshold,
1744 },
1745 {
1746 .procname = "nat_icmp_send",
Linus Torvalds1da177e2005-04-16 15:20:36 -07001747 .maxlen = sizeof(int),
1748 .mode = 0644,
Alexey Dobriyan6d9f2392008-11-03 18:21:05 -08001749 .proc_handler = proc_dointvec,
Linus Torvalds1da177e2005-04-16 15:20:36 -07001750 },
1751#ifdef CONFIG_IP_VS_DEBUG
1752 {
Linus Torvalds1da177e2005-04-16 15:20:36 -07001753 .procname = "debug_level",
1754 .data = &sysctl_ip_vs_debug_level,
1755 .maxlen = sizeof(int),
1756 .mode = 0644,
Alexey Dobriyan6d9f2392008-11-03 18:21:05 -08001757 .proc_handler = proc_dointvec,
Linus Torvalds1da177e2005-04-16 15:20:36 -07001758 },
1759#endif
Linus Torvalds1da177e2005-04-16 15:20:36 -07001760#if 0
1761 {
Linus Torvalds1da177e2005-04-16 15:20:36 -07001762 .procname = "timeout_established",
1763 .data = &vs_timeout_table_dos.timeout[IP_VS_S_ESTABLISHED],
1764 .maxlen = sizeof(int),
1765 .mode = 0644,
Alexey Dobriyan6d9f2392008-11-03 18:21:05 -08001766 .proc_handler = proc_dointvec_jiffies,
Linus Torvalds1da177e2005-04-16 15:20:36 -07001767 },
1768 {
Linus Torvalds1da177e2005-04-16 15:20:36 -07001769 .procname = "timeout_synsent",
1770 .data = &vs_timeout_table_dos.timeout[IP_VS_S_SYN_SENT],
1771 .maxlen = sizeof(int),
1772 .mode = 0644,
Alexey Dobriyan6d9f2392008-11-03 18:21:05 -08001773 .proc_handler = proc_dointvec_jiffies,
Linus Torvalds1da177e2005-04-16 15:20:36 -07001774 },
1775 {
Linus Torvalds1da177e2005-04-16 15:20:36 -07001776 .procname = "timeout_synrecv",
1777 .data = &vs_timeout_table_dos.timeout[IP_VS_S_SYN_RECV],
1778 .maxlen = sizeof(int),
1779 .mode = 0644,
Alexey Dobriyan6d9f2392008-11-03 18:21:05 -08001780 .proc_handler = proc_dointvec_jiffies,
Linus Torvalds1da177e2005-04-16 15:20:36 -07001781 },
1782 {
Linus Torvalds1da177e2005-04-16 15:20:36 -07001783 .procname = "timeout_finwait",
1784 .data = &vs_timeout_table_dos.timeout[IP_VS_S_FIN_WAIT],
1785 .maxlen = sizeof(int),
1786 .mode = 0644,
Alexey Dobriyan6d9f2392008-11-03 18:21:05 -08001787 .proc_handler = proc_dointvec_jiffies,
Linus Torvalds1da177e2005-04-16 15:20:36 -07001788 },
1789 {
Linus Torvalds1da177e2005-04-16 15:20:36 -07001790 .procname = "timeout_timewait",
1791 .data = &vs_timeout_table_dos.timeout[IP_VS_S_TIME_WAIT],
1792 .maxlen = sizeof(int),
1793 .mode = 0644,
Alexey Dobriyan6d9f2392008-11-03 18:21:05 -08001794 .proc_handler = proc_dointvec_jiffies,
Linus Torvalds1da177e2005-04-16 15:20:36 -07001795 },
1796 {
Linus Torvalds1da177e2005-04-16 15:20:36 -07001797 .procname = "timeout_close",
1798 .data = &vs_timeout_table_dos.timeout[IP_VS_S_CLOSE],
1799 .maxlen = sizeof(int),
1800 .mode = 0644,
Alexey Dobriyan6d9f2392008-11-03 18:21:05 -08001801 .proc_handler = proc_dointvec_jiffies,
Linus Torvalds1da177e2005-04-16 15:20:36 -07001802 },
1803 {
Linus Torvalds1da177e2005-04-16 15:20:36 -07001804 .procname = "timeout_closewait",
1805 .data = &vs_timeout_table_dos.timeout[IP_VS_S_CLOSE_WAIT],
1806 .maxlen = sizeof(int),
1807 .mode = 0644,
Alexey Dobriyan6d9f2392008-11-03 18:21:05 -08001808 .proc_handler = proc_dointvec_jiffies,
Linus Torvalds1da177e2005-04-16 15:20:36 -07001809 },
1810 {
Linus Torvalds1da177e2005-04-16 15:20:36 -07001811 .procname = "timeout_lastack",
1812 .data = &vs_timeout_table_dos.timeout[IP_VS_S_LAST_ACK],
1813 .maxlen = sizeof(int),
1814 .mode = 0644,
Alexey Dobriyan6d9f2392008-11-03 18:21:05 -08001815 .proc_handler = proc_dointvec_jiffies,
Linus Torvalds1da177e2005-04-16 15:20:36 -07001816 },
1817 {
Linus Torvalds1da177e2005-04-16 15:20:36 -07001818 .procname = "timeout_listen",
1819 .data = &vs_timeout_table_dos.timeout[IP_VS_S_LISTEN],
1820 .maxlen = sizeof(int),
1821 .mode = 0644,
Alexey Dobriyan6d9f2392008-11-03 18:21:05 -08001822 .proc_handler = proc_dointvec_jiffies,
Linus Torvalds1da177e2005-04-16 15:20:36 -07001823 },
1824 {
Linus Torvalds1da177e2005-04-16 15:20:36 -07001825 .procname = "timeout_synack",
1826 .data = &vs_timeout_table_dos.timeout[IP_VS_S_SYNACK],
1827 .maxlen = sizeof(int),
1828 .mode = 0644,
Alexey Dobriyan6d9f2392008-11-03 18:21:05 -08001829 .proc_handler = proc_dointvec_jiffies,
Linus Torvalds1da177e2005-04-16 15:20:36 -07001830 },
1831 {
Linus Torvalds1da177e2005-04-16 15:20:36 -07001832 .procname = "timeout_udp",
1833 .data = &vs_timeout_table_dos.timeout[IP_VS_S_UDP],
1834 .maxlen = sizeof(int),
1835 .mode = 0644,
Alexey Dobriyan6d9f2392008-11-03 18:21:05 -08001836 .proc_handler = proc_dointvec_jiffies,
Linus Torvalds1da177e2005-04-16 15:20:36 -07001837 },
1838 {
Linus Torvalds1da177e2005-04-16 15:20:36 -07001839 .procname = "timeout_icmp",
1840 .data = &vs_timeout_table_dos.timeout[IP_VS_S_ICMP],
1841 .maxlen = sizeof(int),
1842 .mode = 0644,
Alexey Dobriyan6d9f2392008-11-03 18:21:05 -08001843 .proc_handler = proc_dointvec_jiffies,
Linus Torvalds1da177e2005-04-16 15:20:36 -07001844 },
1845#endif
Eric W. Biedermanf8572d82009-11-05 13:32:03 -08001846 { }
Linus Torvalds1da177e2005-04-16 15:20:36 -07001847};
1848
Sven Wegener5587da52008-08-10 18:24:40 +00001849const struct ctl_path net_vs_ctl_path[] = {
Eric W. Biedermanf8572d82009-11-05 13:32:03 -08001850 { .procname = "net", },
1851 { .procname = "ipv4", },
Pavel Emelyanov90754f82008-01-12 02:33:50 -08001852 { .procname = "vs", },
1853 { }
Linus Torvalds1da177e2005-04-16 15:20:36 -07001854};
Pavel Emelyanov90754f82008-01-12 02:33:50 -08001855EXPORT_SYMBOL_GPL(net_vs_ctl_path);
Simon Horman14e40542011-02-04 18:33:02 +09001856#endif
Linus Torvalds1da177e2005-04-16 15:20:36 -07001857
Linus Torvalds1da177e2005-04-16 15:20:36 -07001858#ifdef CONFIG_PROC_FS
1859
1860struct ip_vs_iter {
Hans Schillstromfc723252011-01-03 14:44:43 +01001861 struct seq_net_private p; /* Do not move this, netns depends upon it*/
Linus Torvalds1da177e2005-04-16 15:20:36 -07001862 struct list_head *table;
1863 int bucket;
1864};
1865
1866/*
1867 * Write the contents of the VS rule table to a PROCfs file.
1868 * (It is kept just for backward compatibility)
1869 */
1870static inline const char *ip_vs_fwd_name(unsigned flags)
1871{
1872 switch (flags & IP_VS_CONN_F_FWD_MASK) {
1873 case IP_VS_CONN_F_LOCALNODE:
1874 return "Local";
1875 case IP_VS_CONN_F_TUNNEL:
1876 return "Tunnel";
1877 case IP_VS_CONN_F_DROUTE:
1878 return "Route";
1879 default:
1880 return "Masq";
1881 }
1882}
1883
1884
1885/* Get the Nth entry in the two lists */
1886static struct ip_vs_service *ip_vs_info_array(struct seq_file *seq, loff_t pos)
1887{
Hans Schillstromfc723252011-01-03 14:44:43 +01001888 struct net *net = seq_file_net(seq);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001889 struct ip_vs_iter *iter = seq->private;
1890 int idx;
1891 struct ip_vs_service *svc;
1892
1893 /* look in hash by protocol */
1894 for (idx = 0; idx < IP_VS_SVC_TAB_SIZE; idx++) {
1895 list_for_each_entry(svc, &ip_vs_svc_table[idx], s_list) {
Hans Schillstromfc723252011-01-03 14:44:43 +01001896 if (net_eq(svc->net, net) && pos-- == 0) {
Linus Torvalds1da177e2005-04-16 15:20:36 -07001897 iter->table = ip_vs_svc_table;
1898 iter->bucket = idx;
1899 return svc;
1900 }
1901 }
1902 }
1903
1904 /* keep looking in fwmark */
1905 for (idx = 0; idx < IP_VS_SVC_TAB_SIZE; idx++) {
1906 list_for_each_entry(svc, &ip_vs_svc_fwm_table[idx], f_list) {
Hans Schillstromfc723252011-01-03 14:44:43 +01001907 if (net_eq(svc->net, net) && pos-- == 0) {
Linus Torvalds1da177e2005-04-16 15:20:36 -07001908 iter->table = ip_vs_svc_fwm_table;
1909 iter->bucket = idx;
1910 return svc;
1911 }
1912 }
1913 }
1914
1915 return NULL;
1916}
1917
1918static void *ip_vs_info_seq_start(struct seq_file *seq, loff_t *pos)
Simon Horman563e94f2008-09-17 10:10:42 +10001919__acquires(__ip_vs_svc_lock)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001920{
1921
1922 read_lock_bh(&__ip_vs_svc_lock);
1923 return *pos ? ip_vs_info_array(seq, *pos - 1) : SEQ_START_TOKEN;
1924}
1925
1926
1927static void *ip_vs_info_seq_next(struct seq_file *seq, void *v, loff_t *pos)
1928{
1929 struct list_head *e;
1930 struct ip_vs_iter *iter;
1931 struct ip_vs_service *svc;
1932
1933 ++*pos;
1934 if (v == SEQ_START_TOKEN)
1935 return ip_vs_info_array(seq,0);
1936
1937 svc = v;
1938 iter = seq->private;
1939
1940 if (iter->table == ip_vs_svc_table) {
1941 /* next service in table hashed by protocol */
1942 if ((e = svc->s_list.next) != &ip_vs_svc_table[iter->bucket])
1943 return list_entry(e, struct ip_vs_service, s_list);
1944
1945
1946 while (++iter->bucket < IP_VS_SVC_TAB_SIZE) {
1947 list_for_each_entry(svc,&ip_vs_svc_table[iter->bucket],
1948 s_list) {
1949 return svc;
1950 }
1951 }
1952
1953 iter->table = ip_vs_svc_fwm_table;
1954 iter->bucket = -1;
1955 goto scan_fwmark;
1956 }
1957
1958 /* next service in hashed by fwmark */
1959 if ((e = svc->f_list.next) != &ip_vs_svc_fwm_table[iter->bucket])
1960 return list_entry(e, struct ip_vs_service, f_list);
1961
1962 scan_fwmark:
1963 while (++iter->bucket < IP_VS_SVC_TAB_SIZE) {
1964 list_for_each_entry(svc, &ip_vs_svc_fwm_table[iter->bucket],
1965 f_list)
1966 return svc;
1967 }
1968
1969 return NULL;
1970}
1971
1972static void ip_vs_info_seq_stop(struct seq_file *seq, void *v)
Simon Horman563e94f2008-09-17 10:10:42 +10001973__releases(__ip_vs_svc_lock)
Linus Torvalds1da177e2005-04-16 15:20:36 -07001974{
1975 read_unlock_bh(&__ip_vs_svc_lock);
1976}
1977
1978
1979static int ip_vs_info_seq_show(struct seq_file *seq, void *v)
1980{
1981 if (v == SEQ_START_TOKEN) {
1982 seq_printf(seq,
1983 "IP Virtual Server version %d.%d.%d (size=%d)\n",
Catalin(ux) M. BOIE6f7edb42010-01-05 05:50:24 +01001984 NVERSION(IP_VS_VERSION_CODE), ip_vs_conn_tab_size);
Linus Torvalds1da177e2005-04-16 15:20:36 -07001985 seq_puts(seq,
1986 "Prot LocalAddress:Port Scheduler Flags\n");
1987 seq_puts(seq,
1988 " -> RemoteAddress:Port Forward Weight ActiveConn InActConn\n");
1989 } else {
1990 const struct ip_vs_service *svc = v;
1991 const struct ip_vs_iter *iter = seq->private;
1992 const struct ip_vs_dest *dest;
1993
Vince Busam667a5f12008-09-02 15:55:49 +02001994 if (iter->table == ip_vs_svc_table) {
1995#ifdef CONFIG_IP_VS_IPV6
1996 if (svc->af == AF_INET6)
Harvey Harrison5b095d9892008-10-29 12:52:50 -07001997 seq_printf(seq, "%s [%pI6]:%04X %s ",
Vince Busam667a5f12008-09-02 15:55:49 +02001998 ip_vs_proto_name(svc->protocol),
Harvey Harrison38ff4fa2008-10-28 16:08:13 -07001999 &svc->addr.in6,
Vince Busam667a5f12008-09-02 15:55:49 +02002000 ntohs(svc->port),
2001 svc->scheduler->name);
2002 else
2003#endif
Nick Chalk26ec0372010-06-22 08:07:01 +02002004 seq_printf(seq, "%s %08X:%04X %s %s ",
Vince Busam667a5f12008-09-02 15:55:49 +02002005 ip_vs_proto_name(svc->protocol),
2006 ntohl(svc->addr.ip),
2007 ntohs(svc->port),
Nick Chalk26ec0372010-06-22 08:07:01 +02002008 svc->scheduler->name,
2009 (svc->flags & IP_VS_SVC_F_ONEPACKET)?"ops ":"");
Vince Busam667a5f12008-09-02 15:55:49 +02002010 } else {
Nick Chalk26ec0372010-06-22 08:07:01 +02002011 seq_printf(seq, "FWM %08X %s %s",
2012 svc->fwmark, svc->scheduler->name,
2013 (svc->flags & IP_VS_SVC_F_ONEPACKET)?"ops ":"");
Vince Busam667a5f12008-09-02 15:55:49 +02002014 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07002015
2016 if (svc->flags & IP_VS_SVC_F_PERSISTENT)
2017 seq_printf(seq, "persistent %d %08X\n",
2018 svc->timeout,
2019 ntohl(svc->netmask));
2020 else
2021 seq_putc(seq, '\n');
2022
2023 list_for_each_entry(dest, &svc->destinations, n_list) {
Vince Busam667a5f12008-09-02 15:55:49 +02002024#ifdef CONFIG_IP_VS_IPV6
2025 if (dest->af == AF_INET6)
2026 seq_printf(seq,
Harvey Harrison5b095d9892008-10-29 12:52:50 -07002027 " -> [%pI6]:%04X"
Vince Busam667a5f12008-09-02 15:55:49 +02002028 " %-7s %-6d %-10d %-10d\n",
Harvey Harrison38ff4fa2008-10-28 16:08:13 -07002029 &dest->addr.in6,
Vince Busam667a5f12008-09-02 15:55:49 +02002030 ntohs(dest->port),
2031 ip_vs_fwd_name(atomic_read(&dest->conn_flags)),
2032 atomic_read(&dest->weight),
2033 atomic_read(&dest->activeconns),
2034 atomic_read(&dest->inactconns));
2035 else
2036#endif
2037 seq_printf(seq,
2038 " -> %08X:%04X "
2039 "%-7s %-6d %-10d %-10d\n",
2040 ntohl(dest->addr.ip),
2041 ntohs(dest->port),
2042 ip_vs_fwd_name(atomic_read(&dest->conn_flags)),
2043 atomic_read(&dest->weight),
2044 atomic_read(&dest->activeconns),
2045 atomic_read(&dest->inactconns));
2046
Linus Torvalds1da177e2005-04-16 15:20:36 -07002047 }
2048 }
2049 return 0;
2050}
2051
Philippe De Muyter56b3d972007-07-10 23:07:31 -07002052static const struct seq_operations ip_vs_info_seq_ops = {
Linus Torvalds1da177e2005-04-16 15:20:36 -07002053 .start = ip_vs_info_seq_start,
2054 .next = ip_vs_info_seq_next,
2055 .stop = ip_vs_info_seq_stop,
2056 .show = ip_vs_info_seq_show,
2057};
2058
2059static int ip_vs_info_open(struct inode *inode, struct file *file)
2060{
Hans Schillstromfc723252011-01-03 14:44:43 +01002061 return seq_open_net(inode, file, &ip_vs_info_seq_ops,
Pavel Emelyanovcf7732e2007-10-10 02:29:29 -07002062 sizeof(struct ip_vs_iter));
Linus Torvalds1da177e2005-04-16 15:20:36 -07002063}
2064
Arjan van de Ven9a321442007-02-12 00:55:35 -08002065static const struct file_operations ip_vs_info_fops = {
Linus Torvalds1da177e2005-04-16 15:20:36 -07002066 .owner = THIS_MODULE,
2067 .open = ip_vs_info_open,
2068 .read = seq_read,
2069 .llseek = seq_lseek,
Hans Schillstrom0f081902011-05-15 17:20:29 +02002070 .release = seq_release_net,
Linus Torvalds1da177e2005-04-16 15:20:36 -07002071};
2072
Linus Torvalds1da177e2005-04-16 15:20:36 -07002073static int ip_vs_stats_show(struct seq_file *seq, void *v)
2074{
Hans Schillstromb17fc992011-01-03 14:44:56 +01002075 struct net *net = seq_file_single_net(seq);
Julian Anastasov55a3d4e2011-03-14 01:37:49 +02002076 struct ip_vs_stats_user show;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002077
2078/* 01234567 01234567 01234567 0123456701234567 0123456701234567 */
2079 seq_puts(seq,
2080 " Total Incoming Outgoing Incoming Outgoing\n");
2081 seq_printf(seq,
2082 " Conns Packets Packets Bytes Bytes\n");
2083
Julian Anastasov55a3d4e2011-03-14 01:37:49 +02002084 ip_vs_copy_stats(&show, &net_ipvs(net)->tot_stats);
2085 seq_printf(seq, "%8X %8X %8X %16LX %16LX\n\n", show.conns,
2086 show.inpkts, show.outpkts,
2087 (unsigned long long) show.inbytes,
2088 (unsigned long long) show.outbytes);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002089
2090/* 01234567 01234567 01234567 0123456701234567 0123456701234567 */
2091 seq_puts(seq,
2092 " Conns/s Pkts/s Pkts/s Bytes/s Bytes/s\n");
Julian Anastasov55a3d4e2011-03-14 01:37:49 +02002093 seq_printf(seq, "%8X %8X %8X %16X %16X\n",
2094 show.cps, show.inpps, show.outpps,
2095 show.inbps, show.outbps);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002096
2097 return 0;
2098}
2099
2100static int ip_vs_stats_seq_open(struct inode *inode, struct file *file)
2101{
Hans Schillstromfc723252011-01-03 14:44:43 +01002102 return single_open_net(inode, file, ip_vs_stats_show);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002103}
2104
Arjan van de Ven9a321442007-02-12 00:55:35 -08002105static const struct file_operations ip_vs_stats_fops = {
Linus Torvalds1da177e2005-04-16 15:20:36 -07002106 .owner = THIS_MODULE,
2107 .open = ip_vs_stats_seq_open,
2108 .read = seq_read,
2109 .llseek = seq_lseek,
Hans Schillstrom0f081902011-05-15 17:20:29 +02002110 .release = single_release_net,
Linus Torvalds1da177e2005-04-16 15:20:36 -07002111};
2112
Hans Schillstromb17fc992011-01-03 14:44:56 +01002113static int ip_vs_stats_percpu_show(struct seq_file *seq, void *v)
2114{
2115 struct net *net = seq_file_single_net(seq);
Julian Anastasov2a0751a2011-03-04 12:20:35 +02002116 struct ip_vs_stats *tot_stats = &net_ipvs(net)->tot_stats;
2117 struct ip_vs_cpu_stats *cpustats = tot_stats->cpustats;
Julian Anastasovea9f22c2011-03-14 01:41:54 +02002118 struct ip_vs_stats_user rates;
Hans Schillstromb17fc992011-01-03 14:44:56 +01002119 int i;
2120
2121/* 01234567 01234567 01234567 0123456701234567 0123456701234567 */
2122 seq_puts(seq,
2123 " Total Incoming Outgoing Incoming Outgoing\n");
2124 seq_printf(seq,
2125 "CPU Conns Packets Packets Bytes Bytes\n");
2126
2127 for_each_possible_cpu(i) {
Julian Anastasov2a0751a2011-03-04 12:20:35 +02002128 struct ip_vs_cpu_stats *u = per_cpu_ptr(cpustats, i);
2129 unsigned int start;
2130 __u64 inbytes, outbytes;
2131
2132 do {
2133 start = u64_stats_fetch_begin_bh(&u->syncp);
2134 inbytes = u->ustats.inbytes;
2135 outbytes = u->ustats.outbytes;
2136 } while (u64_stats_fetch_retry_bh(&u->syncp, start));
2137
Hans Schillstromb17fc992011-01-03 14:44:56 +01002138 seq_printf(seq, "%3X %8X %8X %8X %16LX %16LX\n",
Julian Anastasov2a0751a2011-03-04 12:20:35 +02002139 i, u->ustats.conns, u->ustats.inpkts,
2140 u->ustats.outpkts, (__u64)inbytes,
2141 (__u64)outbytes);
Hans Schillstromb17fc992011-01-03 14:44:56 +01002142 }
2143
2144 spin_lock_bh(&tot_stats->lock);
Julian Anastasovea9f22c2011-03-14 01:41:54 +02002145
Hans Schillstromb17fc992011-01-03 14:44:56 +01002146 seq_printf(seq, " ~ %8X %8X %8X %16LX %16LX\n\n",
2147 tot_stats->ustats.conns, tot_stats->ustats.inpkts,
2148 tot_stats->ustats.outpkts,
2149 (unsigned long long) tot_stats->ustats.inbytes,
2150 (unsigned long long) tot_stats->ustats.outbytes);
2151
Julian Anastasovea9f22c2011-03-14 01:41:54 +02002152 ip_vs_read_estimator(&rates, tot_stats);
2153
2154 spin_unlock_bh(&tot_stats->lock);
2155
Hans Schillstromb17fc992011-01-03 14:44:56 +01002156/* 01234567 01234567 01234567 0123456701234567 0123456701234567 */
2157 seq_puts(seq,
2158 " Conns/s Pkts/s Pkts/s Bytes/s Bytes/s\n");
2159 seq_printf(seq, " %8X %8X %8X %16X %16X\n",
Julian Anastasovea9f22c2011-03-14 01:41:54 +02002160 rates.cps,
2161 rates.inpps,
2162 rates.outpps,
2163 rates.inbps,
2164 rates.outbps);
Hans Schillstromb17fc992011-01-03 14:44:56 +01002165
2166 return 0;
2167}
2168
2169static int ip_vs_stats_percpu_seq_open(struct inode *inode, struct file *file)
2170{
2171 return single_open_net(inode, file, ip_vs_stats_percpu_show);
2172}
2173
2174static const struct file_operations ip_vs_stats_percpu_fops = {
2175 .owner = THIS_MODULE,
2176 .open = ip_vs_stats_percpu_seq_open,
2177 .read = seq_read,
2178 .llseek = seq_lseek,
Hans Schillstrom0f081902011-05-15 17:20:29 +02002179 .release = single_release_net,
Hans Schillstromb17fc992011-01-03 14:44:56 +01002180};
Linus Torvalds1da177e2005-04-16 15:20:36 -07002181#endif
2182
2183/*
2184 * Set timeout values for tcp tcpfin udp in the timeout_table.
2185 */
Hans Schillstrom93304192011-01-03 14:44:51 +01002186static int ip_vs_set_timeout(struct net *net, struct ip_vs_timeout_user *u)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002187{
Changli Gao091bb342011-01-21 18:02:13 +08002188#if defined(CONFIG_IP_VS_PROTO_TCP) || defined(CONFIG_IP_VS_PROTO_UDP)
Hans Schillstrom93304192011-01-03 14:44:51 +01002189 struct ip_vs_proto_data *pd;
Changli Gao091bb342011-01-21 18:02:13 +08002190#endif
Hans Schillstrom93304192011-01-03 14:44:51 +01002191
Linus Torvalds1da177e2005-04-16 15:20:36 -07002192 IP_VS_DBG(2, "Setting timeout tcp:%d tcpfin:%d udp:%d\n",
2193 u->tcp_timeout,
2194 u->tcp_fin_timeout,
2195 u->udp_timeout);
2196
2197#ifdef CONFIG_IP_VS_PROTO_TCP
2198 if (u->tcp_timeout) {
Hans Schillstrom93304192011-01-03 14:44:51 +01002199 pd = ip_vs_proto_data_get(net, IPPROTO_TCP);
2200 pd->timeout_table[IP_VS_TCP_S_ESTABLISHED]
Linus Torvalds1da177e2005-04-16 15:20:36 -07002201 = u->tcp_timeout * HZ;
2202 }
2203
2204 if (u->tcp_fin_timeout) {
Hans Schillstrom93304192011-01-03 14:44:51 +01002205 pd = ip_vs_proto_data_get(net, IPPROTO_TCP);
2206 pd->timeout_table[IP_VS_TCP_S_FIN_WAIT]
Linus Torvalds1da177e2005-04-16 15:20:36 -07002207 = u->tcp_fin_timeout * HZ;
2208 }
2209#endif
2210
2211#ifdef CONFIG_IP_VS_PROTO_UDP
2212 if (u->udp_timeout) {
Hans Schillstrom93304192011-01-03 14:44:51 +01002213 pd = ip_vs_proto_data_get(net, IPPROTO_UDP);
2214 pd->timeout_table[IP_VS_UDP_S_NORMAL]
Linus Torvalds1da177e2005-04-16 15:20:36 -07002215 = u->udp_timeout * HZ;
2216 }
2217#endif
2218 return 0;
2219}
2220
2221
2222#define SET_CMDID(cmd) (cmd - IP_VS_BASE_CTL)
2223#define SERVICE_ARG_LEN (sizeof(struct ip_vs_service_user))
2224#define SVCDEST_ARG_LEN (sizeof(struct ip_vs_service_user) + \
2225 sizeof(struct ip_vs_dest_user))
2226#define TIMEOUT_ARG_LEN (sizeof(struct ip_vs_timeout_user))
2227#define DAEMON_ARG_LEN (sizeof(struct ip_vs_daemon_user))
2228#define MAX_ARG_LEN SVCDEST_ARG_LEN
2229
Arjan van de Ven9b5b5cf2005-11-29 16:21:38 -08002230static const unsigned char set_arglen[SET_CMDID(IP_VS_SO_SET_MAX)+1] = {
Linus Torvalds1da177e2005-04-16 15:20:36 -07002231 [SET_CMDID(IP_VS_SO_SET_ADD)] = SERVICE_ARG_LEN,
2232 [SET_CMDID(IP_VS_SO_SET_EDIT)] = SERVICE_ARG_LEN,
2233 [SET_CMDID(IP_VS_SO_SET_DEL)] = SERVICE_ARG_LEN,
2234 [SET_CMDID(IP_VS_SO_SET_FLUSH)] = 0,
2235 [SET_CMDID(IP_VS_SO_SET_ADDDEST)] = SVCDEST_ARG_LEN,
2236 [SET_CMDID(IP_VS_SO_SET_DELDEST)] = SVCDEST_ARG_LEN,
2237 [SET_CMDID(IP_VS_SO_SET_EDITDEST)] = SVCDEST_ARG_LEN,
2238 [SET_CMDID(IP_VS_SO_SET_TIMEOUT)] = TIMEOUT_ARG_LEN,
2239 [SET_CMDID(IP_VS_SO_SET_STARTDAEMON)] = DAEMON_ARG_LEN,
2240 [SET_CMDID(IP_VS_SO_SET_STOPDAEMON)] = DAEMON_ARG_LEN,
2241 [SET_CMDID(IP_VS_SO_SET_ZERO)] = SERVICE_ARG_LEN,
2242};
2243
Julius Volzc860c6b2008-09-02 15:55:36 +02002244static void ip_vs_copy_usvc_compat(struct ip_vs_service_user_kern *usvc,
2245 struct ip_vs_service_user *usvc_compat)
2246{
Simon Horman0d1e71b2010-08-22 21:37:54 +09002247 memset(usvc, 0, sizeof(*usvc));
2248
Julius Volzc860c6b2008-09-02 15:55:36 +02002249 usvc->af = AF_INET;
2250 usvc->protocol = usvc_compat->protocol;
2251 usvc->addr.ip = usvc_compat->addr;
2252 usvc->port = usvc_compat->port;
2253 usvc->fwmark = usvc_compat->fwmark;
2254
2255 /* Deep copy of sched_name is not needed here */
2256 usvc->sched_name = usvc_compat->sched_name;
2257
2258 usvc->flags = usvc_compat->flags;
2259 usvc->timeout = usvc_compat->timeout;
2260 usvc->netmask = usvc_compat->netmask;
2261}
2262
2263static void ip_vs_copy_udest_compat(struct ip_vs_dest_user_kern *udest,
2264 struct ip_vs_dest_user *udest_compat)
2265{
Simon Horman0d1e71b2010-08-22 21:37:54 +09002266 memset(udest, 0, sizeof(*udest));
2267
Julius Volzc860c6b2008-09-02 15:55:36 +02002268 udest->addr.ip = udest_compat->addr;
2269 udest->port = udest_compat->port;
2270 udest->conn_flags = udest_compat->conn_flags;
2271 udest->weight = udest_compat->weight;
2272 udest->u_threshold = udest_compat->u_threshold;
2273 udest->l_threshold = udest_compat->l_threshold;
2274}
2275
Linus Torvalds1da177e2005-04-16 15:20:36 -07002276static int
2277do_ip_vs_set_ctl(struct sock *sk, int cmd, void __user *user, unsigned int len)
2278{
Hans Schillstromfc723252011-01-03 14:44:43 +01002279 struct net *net = sock_net(sk);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002280 int ret;
2281 unsigned char arg[MAX_ARG_LEN];
Julius Volzc860c6b2008-09-02 15:55:36 +02002282 struct ip_vs_service_user *usvc_compat;
2283 struct ip_vs_service_user_kern usvc;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002284 struct ip_vs_service *svc;
Julius Volzc860c6b2008-09-02 15:55:36 +02002285 struct ip_vs_dest_user *udest_compat;
2286 struct ip_vs_dest_user_kern udest;
Hans Schillstromae1d48b2011-10-11 10:54:35 +09002287 struct netns_ipvs *ipvs = net_ipvs(net);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002288
2289 if (!capable(CAP_NET_ADMIN))
2290 return -EPERM;
2291
Arjan van de Ven04bcef22010-01-04 16:37:12 +01002292 if (cmd < IP_VS_BASE_CTL || cmd > IP_VS_SO_SET_MAX)
2293 return -EINVAL;
2294 if (len < 0 || len > MAX_ARG_LEN)
2295 return -EINVAL;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002296 if (len != set_arglen[SET_CMDID(cmd)]) {
Hannes Eder1e3e2382009-08-02 11:05:41 +00002297 pr_err("set_ctl: len %u != %u\n",
2298 len, set_arglen[SET_CMDID(cmd)]);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002299 return -EINVAL;
2300 }
2301
2302 if (copy_from_user(arg, user, len) != 0)
2303 return -EFAULT;
2304
2305 /* increase the module use count */
2306 ip_vs_use_count_inc();
2307
Hans Schillstromae1d48b2011-10-11 10:54:35 +09002308 /* Handle daemons since they have another lock */
2309 if (cmd == IP_VS_SO_SET_STARTDAEMON ||
2310 cmd == IP_VS_SO_SET_STOPDAEMON) {
2311 struct ip_vs_daemon_user *dm = (struct ip_vs_daemon_user *)arg;
2312
2313 if (mutex_lock_interruptible(&ipvs->sync_mutex)) {
2314 ret = -ERESTARTSYS;
2315 goto out_dec;
2316 }
2317 if (cmd == IP_VS_SO_SET_STARTDAEMON)
2318 ret = start_sync_thread(net, dm->state, dm->mcast_ifn,
2319 dm->syncid);
2320 else
2321 ret = stop_sync_thread(net, dm->state);
2322 mutex_unlock(&ipvs->sync_mutex);
2323 goto out_dec;
2324 }
2325
Ingo Molnar14cc3e22006-03-26 01:37:14 -08002326 if (mutex_lock_interruptible(&__ip_vs_mutex)) {
Linus Torvalds1da177e2005-04-16 15:20:36 -07002327 ret = -ERESTARTSYS;
2328 goto out_dec;
2329 }
2330
2331 if (cmd == IP_VS_SO_SET_FLUSH) {
2332 /* Flush the virtual service */
Hans Schillstromfc723252011-01-03 14:44:43 +01002333 ret = ip_vs_flush(net);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002334 goto out_unlock;
2335 } else if (cmd == IP_VS_SO_SET_TIMEOUT) {
2336 /* Set timeout values for (tcp tcpfin udp) */
Hans Schillstrom93304192011-01-03 14:44:51 +01002337 ret = ip_vs_set_timeout(net, (struct ip_vs_timeout_user *)arg);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002338 goto out_unlock;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002339 }
2340
Julius Volzc860c6b2008-09-02 15:55:36 +02002341 usvc_compat = (struct ip_vs_service_user *)arg;
2342 udest_compat = (struct ip_vs_dest_user *)(usvc_compat + 1);
2343
2344 /* We only use the new structs internally, so copy userspace compat
2345 * structs to extended internal versions */
2346 ip_vs_copy_usvc_compat(&usvc, usvc_compat);
2347 ip_vs_copy_udest_compat(&udest, udest_compat);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002348
2349 if (cmd == IP_VS_SO_SET_ZERO) {
2350 /* if no service address is set, zero counters in all */
Julius Volzc860c6b2008-09-02 15:55:36 +02002351 if (!usvc.fwmark && !usvc.addr.ip && !usvc.port) {
Hans Schillstromfc723252011-01-03 14:44:43 +01002352 ret = ip_vs_zero_all(net);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002353 goto out_unlock;
2354 }
2355 }
2356
Venkata Mohan Reddy2906f662010-02-18 12:31:05 +01002357 /* Check for valid protocol: TCP or UDP or SCTP, even for fwmark!=0 */
2358 if (usvc.protocol != IPPROTO_TCP && usvc.protocol != IPPROTO_UDP &&
2359 usvc.protocol != IPPROTO_SCTP) {
Hannes Eder1e3e2382009-08-02 11:05:41 +00002360 pr_err("set_ctl: invalid protocol: %d %pI4:%d %s\n",
2361 usvc.protocol, &usvc.addr.ip,
2362 ntohs(usvc.port), usvc.sched_name);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002363 ret = -EFAULT;
2364 goto out_unlock;
2365 }
2366
2367 /* Lookup the exact service by <protocol, addr, port> or fwmark */
Julius Volzc860c6b2008-09-02 15:55:36 +02002368 if (usvc.fwmark == 0)
Hans Schillstromfc723252011-01-03 14:44:43 +01002369 svc = __ip_vs_service_find(net, usvc.af, usvc.protocol,
Julian Anastasov26c15cf2010-09-21 18:12:30 +02002370 &usvc.addr, usvc.port);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002371 else
Hans Schillstromfc723252011-01-03 14:44:43 +01002372 svc = __ip_vs_svc_fwm_find(net, usvc.af, usvc.fwmark);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002373
2374 if (cmd != IP_VS_SO_SET_ADD
Julius Volzc860c6b2008-09-02 15:55:36 +02002375 && (svc == NULL || svc->protocol != usvc.protocol)) {
Linus Torvalds1da177e2005-04-16 15:20:36 -07002376 ret = -ESRCH;
Julian Anastasov26c15cf2010-09-21 18:12:30 +02002377 goto out_unlock;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002378 }
2379
2380 switch (cmd) {
2381 case IP_VS_SO_SET_ADD:
2382 if (svc != NULL)
2383 ret = -EEXIST;
2384 else
Hans Schillstromfc723252011-01-03 14:44:43 +01002385 ret = ip_vs_add_service(net, &usvc, &svc);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002386 break;
2387 case IP_VS_SO_SET_EDIT:
Julius Volzc860c6b2008-09-02 15:55:36 +02002388 ret = ip_vs_edit_service(svc, &usvc);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002389 break;
2390 case IP_VS_SO_SET_DEL:
2391 ret = ip_vs_del_service(svc);
2392 if (!ret)
2393 goto out_unlock;
2394 break;
2395 case IP_VS_SO_SET_ZERO:
2396 ret = ip_vs_zero_service(svc);
2397 break;
2398 case IP_VS_SO_SET_ADDDEST:
Julius Volzc860c6b2008-09-02 15:55:36 +02002399 ret = ip_vs_add_dest(svc, &udest);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002400 break;
2401 case IP_VS_SO_SET_EDITDEST:
Julius Volzc860c6b2008-09-02 15:55:36 +02002402 ret = ip_vs_edit_dest(svc, &udest);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002403 break;
2404 case IP_VS_SO_SET_DELDEST:
Julius Volzc860c6b2008-09-02 15:55:36 +02002405 ret = ip_vs_del_dest(svc, &udest);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002406 break;
2407 default:
2408 ret = -EINVAL;
2409 }
2410
Linus Torvalds1da177e2005-04-16 15:20:36 -07002411 out_unlock:
Ingo Molnar14cc3e22006-03-26 01:37:14 -08002412 mutex_unlock(&__ip_vs_mutex);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002413 out_dec:
2414 /* decrease the module use count */
2415 ip_vs_use_count_dec();
2416
2417 return ret;
2418}
2419
2420
2421static void
Linus Torvalds1da177e2005-04-16 15:20:36 -07002422ip_vs_copy_service(struct ip_vs_service_entry *dst, struct ip_vs_service *src)
2423{
2424 dst->protocol = src->protocol;
Julius Volze7ade462008-09-02 15:55:33 +02002425 dst->addr = src->addr.ip;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002426 dst->port = src->port;
2427 dst->fwmark = src->fwmark;
pageexec4da62fc2005-06-26 16:00:19 -07002428 strlcpy(dst->sched_name, src->scheduler->name, sizeof(dst->sched_name));
Linus Torvalds1da177e2005-04-16 15:20:36 -07002429 dst->flags = src->flags;
2430 dst->timeout = src->timeout / HZ;
2431 dst->netmask = src->netmask;
2432 dst->num_dests = src->num_dests;
2433 ip_vs_copy_stats(&dst->stats, &src->stats);
2434}
2435
2436static inline int
Hans Schillstromfc723252011-01-03 14:44:43 +01002437__ip_vs_get_service_entries(struct net *net,
2438 const struct ip_vs_get_services *get,
Linus Torvalds1da177e2005-04-16 15:20:36 -07002439 struct ip_vs_get_services __user *uptr)
2440{
2441 int idx, count=0;
2442 struct ip_vs_service *svc;
2443 struct ip_vs_service_entry entry;
2444 int ret = 0;
2445
2446 for (idx = 0; idx < IP_VS_SVC_TAB_SIZE; idx++) {
2447 list_for_each_entry(svc, &ip_vs_svc_table[idx], s_list) {
Julius Volzf94fd042008-09-02 15:55:55 +02002448 /* Only expose IPv4 entries to old interface */
Hans Schillstromfc723252011-01-03 14:44:43 +01002449 if (svc->af != AF_INET || !net_eq(svc->net, net))
Julius Volzf94fd042008-09-02 15:55:55 +02002450 continue;
2451
Linus Torvalds1da177e2005-04-16 15:20:36 -07002452 if (count >= get->num_services)
2453 goto out;
pageexec4da62fc2005-06-26 16:00:19 -07002454 memset(&entry, 0, sizeof(entry));
Linus Torvalds1da177e2005-04-16 15:20:36 -07002455 ip_vs_copy_service(&entry, svc);
2456 if (copy_to_user(&uptr->entrytable[count],
2457 &entry, sizeof(entry))) {
2458 ret = -EFAULT;
2459 goto out;
2460 }
2461 count++;
2462 }
2463 }
2464
2465 for (idx = 0; idx < IP_VS_SVC_TAB_SIZE; idx++) {
2466 list_for_each_entry(svc, &ip_vs_svc_fwm_table[idx], f_list) {
Julius Volzf94fd042008-09-02 15:55:55 +02002467 /* Only expose IPv4 entries to old interface */
Hans Schillstromfc723252011-01-03 14:44:43 +01002468 if (svc->af != AF_INET || !net_eq(svc->net, net))
Julius Volzf94fd042008-09-02 15:55:55 +02002469 continue;
2470
Linus Torvalds1da177e2005-04-16 15:20:36 -07002471 if (count >= get->num_services)
2472 goto out;
pageexec4da62fc2005-06-26 16:00:19 -07002473 memset(&entry, 0, sizeof(entry));
Linus Torvalds1da177e2005-04-16 15:20:36 -07002474 ip_vs_copy_service(&entry, svc);
2475 if (copy_to_user(&uptr->entrytable[count],
2476 &entry, sizeof(entry))) {
2477 ret = -EFAULT;
2478 goto out;
2479 }
2480 count++;
2481 }
2482 }
Hans Schillstrom552ad652011-06-13 12:19:26 +02002483out:
Linus Torvalds1da177e2005-04-16 15:20:36 -07002484 return ret;
2485}
2486
2487static inline int
Hans Schillstromfc723252011-01-03 14:44:43 +01002488__ip_vs_get_dest_entries(struct net *net, const struct ip_vs_get_dests *get,
Linus Torvalds1da177e2005-04-16 15:20:36 -07002489 struct ip_vs_get_dests __user *uptr)
2490{
2491 struct ip_vs_service *svc;
Julius Volzb18610d2008-09-02 15:55:37 +02002492 union nf_inet_addr addr = { .ip = get->addr };
Linus Torvalds1da177e2005-04-16 15:20:36 -07002493 int ret = 0;
2494
2495 if (get->fwmark)
Hans Schillstromfc723252011-01-03 14:44:43 +01002496 svc = __ip_vs_svc_fwm_find(net, AF_INET, get->fwmark);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002497 else
Hans Schillstromfc723252011-01-03 14:44:43 +01002498 svc = __ip_vs_service_find(net, AF_INET, get->protocol, &addr,
Julian Anastasov26c15cf2010-09-21 18:12:30 +02002499 get->port);
Julius Volzb18610d2008-09-02 15:55:37 +02002500
Linus Torvalds1da177e2005-04-16 15:20:36 -07002501 if (svc) {
2502 int count = 0;
2503 struct ip_vs_dest *dest;
2504 struct ip_vs_dest_entry entry;
2505
2506 list_for_each_entry(dest, &svc->destinations, n_list) {
2507 if (count >= get->num_dests)
2508 break;
2509
Julius Volze7ade462008-09-02 15:55:33 +02002510 entry.addr = dest->addr.ip;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002511 entry.port = dest->port;
2512 entry.conn_flags = atomic_read(&dest->conn_flags);
2513 entry.weight = atomic_read(&dest->weight);
2514 entry.u_threshold = dest->u_threshold;
2515 entry.l_threshold = dest->l_threshold;
2516 entry.activeconns = atomic_read(&dest->activeconns);
2517 entry.inactconns = atomic_read(&dest->inactconns);
2518 entry.persistconns = atomic_read(&dest->persistconns);
2519 ip_vs_copy_stats(&entry.stats, &dest->stats);
2520 if (copy_to_user(&uptr->entrytable[count],
2521 &entry, sizeof(entry))) {
2522 ret = -EFAULT;
2523 break;
2524 }
2525 count++;
2526 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07002527 } else
2528 ret = -ESRCH;
2529 return ret;
2530}
2531
2532static inline void
Hans Schillstrom93304192011-01-03 14:44:51 +01002533__ip_vs_get_timeouts(struct net *net, struct ip_vs_timeout_user *u)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002534{
Changli Gao091bb342011-01-21 18:02:13 +08002535#if defined(CONFIG_IP_VS_PROTO_TCP) || defined(CONFIG_IP_VS_PROTO_UDP)
Hans Schillstrom93304192011-01-03 14:44:51 +01002536 struct ip_vs_proto_data *pd;
Changli Gao091bb342011-01-21 18:02:13 +08002537#endif
Hans Schillstrom93304192011-01-03 14:44:51 +01002538
Linus Torvalds1da177e2005-04-16 15:20:36 -07002539#ifdef CONFIG_IP_VS_PROTO_TCP
Hans Schillstrom93304192011-01-03 14:44:51 +01002540 pd = ip_vs_proto_data_get(net, IPPROTO_TCP);
2541 u->tcp_timeout = pd->timeout_table[IP_VS_TCP_S_ESTABLISHED] / HZ;
2542 u->tcp_fin_timeout = pd->timeout_table[IP_VS_TCP_S_FIN_WAIT] / HZ;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002543#endif
2544#ifdef CONFIG_IP_VS_PROTO_UDP
Hans Schillstrom93304192011-01-03 14:44:51 +01002545 pd = ip_vs_proto_data_get(net, IPPROTO_UDP);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002546 u->udp_timeout =
Hans Schillstrom93304192011-01-03 14:44:51 +01002547 pd->timeout_table[IP_VS_UDP_S_NORMAL] / HZ;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002548#endif
2549}
2550
2551
2552#define GET_CMDID(cmd) (cmd - IP_VS_BASE_CTL)
2553#define GET_INFO_ARG_LEN (sizeof(struct ip_vs_getinfo))
2554#define GET_SERVICES_ARG_LEN (sizeof(struct ip_vs_get_services))
2555#define GET_SERVICE_ARG_LEN (sizeof(struct ip_vs_service_entry))
2556#define GET_DESTS_ARG_LEN (sizeof(struct ip_vs_get_dests))
2557#define GET_TIMEOUT_ARG_LEN (sizeof(struct ip_vs_timeout_user))
2558#define GET_DAEMON_ARG_LEN (sizeof(struct ip_vs_daemon_user) * 2)
2559
Arjan van de Ven9b5b5cf2005-11-29 16:21:38 -08002560static const unsigned char get_arglen[GET_CMDID(IP_VS_SO_GET_MAX)+1] = {
Linus Torvalds1da177e2005-04-16 15:20:36 -07002561 [GET_CMDID(IP_VS_SO_GET_VERSION)] = 64,
2562 [GET_CMDID(IP_VS_SO_GET_INFO)] = GET_INFO_ARG_LEN,
2563 [GET_CMDID(IP_VS_SO_GET_SERVICES)] = GET_SERVICES_ARG_LEN,
2564 [GET_CMDID(IP_VS_SO_GET_SERVICE)] = GET_SERVICE_ARG_LEN,
2565 [GET_CMDID(IP_VS_SO_GET_DESTS)] = GET_DESTS_ARG_LEN,
2566 [GET_CMDID(IP_VS_SO_GET_TIMEOUT)] = GET_TIMEOUT_ARG_LEN,
2567 [GET_CMDID(IP_VS_SO_GET_DAEMON)] = GET_DAEMON_ARG_LEN,
2568};
2569
2570static int
2571do_ip_vs_get_ctl(struct sock *sk, int cmd, void __user *user, int *len)
2572{
2573 unsigned char arg[128];
2574 int ret = 0;
Arjan van de Ven04bcef22010-01-04 16:37:12 +01002575 unsigned int copylen;
Hans Schillstromfc723252011-01-03 14:44:43 +01002576 struct net *net = sock_net(sk);
Hans Schillstromf1313152011-01-03 14:44:55 +01002577 struct netns_ipvs *ipvs = net_ipvs(net);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002578
Hans Schillstromfc723252011-01-03 14:44:43 +01002579 BUG_ON(!net);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002580 if (!capable(CAP_NET_ADMIN))
2581 return -EPERM;
2582
Arjan van de Ven04bcef22010-01-04 16:37:12 +01002583 if (cmd < IP_VS_BASE_CTL || cmd > IP_VS_SO_GET_MAX)
2584 return -EINVAL;
2585
Linus Torvalds1da177e2005-04-16 15:20:36 -07002586 if (*len < get_arglen[GET_CMDID(cmd)]) {
Hannes Eder1e3e2382009-08-02 11:05:41 +00002587 pr_err("get_ctl: len %u < %u\n",
2588 *len, get_arglen[GET_CMDID(cmd)]);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002589 return -EINVAL;
2590 }
2591
Arjan van de Ven04bcef22010-01-04 16:37:12 +01002592 copylen = get_arglen[GET_CMDID(cmd)];
2593 if (copylen > 128)
2594 return -EINVAL;
2595
2596 if (copy_from_user(arg, user, copylen) != 0)
Linus Torvalds1da177e2005-04-16 15:20:36 -07002597 return -EFAULT;
Hans Schillstromae1d48b2011-10-11 10:54:35 +09002598 /*
2599 * Handle daemons first since it has its own locking
2600 */
2601 if (cmd == IP_VS_SO_GET_DAEMON) {
2602 struct ip_vs_daemon_user d[2];
2603
2604 memset(&d, 0, sizeof(d));
2605 if (mutex_lock_interruptible(&ipvs->sync_mutex))
2606 return -ERESTARTSYS;
2607
2608 if (ipvs->sync_state & IP_VS_STATE_MASTER) {
2609 d[0].state = IP_VS_STATE_MASTER;
2610 strlcpy(d[0].mcast_ifn, ipvs->master_mcast_ifn,
2611 sizeof(d[0].mcast_ifn));
2612 d[0].syncid = ipvs->master_syncid;
2613 }
2614 if (ipvs->sync_state & IP_VS_STATE_BACKUP) {
2615 d[1].state = IP_VS_STATE_BACKUP;
2616 strlcpy(d[1].mcast_ifn, ipvs->backup_mcast_ifn,
2617 sizeof(d[1].mcast_ifn));
2618 d[1].syncid = ipvs->backup_syncid;
2619 }
2620 if (copy_to_user(user, &d, sizeof(d)) != 0)
2621 ret = -EFAULT;
2622 mutex_unlock(&ipvs->sync_mutex);
2623 return ret;
2624 }
Linus Torvalds1da177e2005-04-16 15:20:36 -07002625
Ingo Molnar14cc3e22006-03-26 01:37:14 -08002626 if (mutex_lock_interruptible(&__ip_vs_mutex))
Linus Torvalds1da177e2005-04-16 15:20:36 -07002627 return -ERESTARTSYS;
2628
2629 switch (cmd) {
2630 case IP_VS_SO_GET_VERSION:
2631 {
2632 char buf[64];
2633
2634 sprintf(buf, "IP Virtual Server version %d.%d.%d (size=%d)",
Catalin(ux) M. BOIE6f7edb42010-01-05 05:50:24 +01002635 NVERSION(IP_VS_VERSION_CODE), ip_vs_conn_tab_size);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002636 if (copy_to_user(user, buf, strlen(buf)+1) != 0) {
2637 ret = -EFAULT;
2638 goto out;
2639 }
2640 *len = strlen(buf)+1;
2641 }
2642 break;
2643
2644 case IP_VS_SO_GET_INFO:
2645 {
2646 struct ip_vs_getinfo info;
2647 info.version = IP_VS_VERSION_CODE;
Catalin(ux) M. BOIE6f7edb42010-01-05 05:50:24 +01002648 info.size = ip_vs_conn_tab_size;
Hans Schillstroma0840e22011-01-03 14:44:58 +01002649 info.num_services = ipvs->num_services;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002650 if (copy_to_user(user, &info, sizeof(info)) != 0)
2651 ret = -EFAULT;
2652 }
2653 break;
2654
2655 case IP_VS_SO_GET_SERVICES:
2656 {
2657 struct ip_vs_get_services *get;
2658 int size;
2659
2660 get = (struct ip_vs_get_services *)arg;
2661 size = sizeof(*get) +
2662 sizeof(struct ip_vs_service_entry) * get->num_services;
2663 if (*len != size) {
Hannes Eder1e3e2382009-08-02 11:05:41 +00002664 pr_err("length: %u != %u\n", *len, size);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002665 ret = -EINVAL;
2666 goto out;
2667 }
Hans Schillstromfc723252011-01-03 14:44:43 +01002668 ret = __ip_vs_get_service_entries(net, get, user);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002669 }
2670 break;
2671
2672 case IP_VS_SO_GET_SERVICE:
2673 {
2674 struct ip_vs_service_entry *entry;
2675 struct ip_vs_service *svc;
Julius Volzb18610d2008-09-02 15:55:37 +02002676 union nf_inet_addr addr;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002677
2678 entry = (struct ip_vs_service_entry *)arg;
Julius Volzb18610d2008-09-02 15:55:37 +02002679 addr.ip = entry->addr;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002680 if (entry->fwmark)
Hans Schillstromfc723252011-01-03 14:44:43 +01002681 svc = __ip_vs_svc_fwm_find(net, AF_INET, entry->fwmark);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002682 else
Hans Schillstromfc723252011-01-03 14:44:43 +01002683 svc = __ip_vs_service_find(net, AF_INET,
2684 entry->protocol, &addr,
2685 entry->port);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002686 if (svc) {
2687 ip_vs_copy_service(entry, svc);
2688 if (copy_to_user(user, entry, sizeof(*entry)) != 0)
2689 ret = -EFAULT;
Linus Torvalds1da177e2005-04-16 15:20:36 -07002690 } else
2691 ret = -ESRCH;
2692 }
2693 break;
2694
2695 case IP_VS_SO_GET_DESTS:
2696 {
2697 struct ip_vs_get_dests *get;
2698 int size;
2699
2700 get = (struct ip_vs_get_dests *)arg;
2701 size = sizeof(*get) +
2702 sizeof(struct ip_vs_dest_entry) * get->num_dests;
2703 if (*len != size) {
Hannes Eder1e3e2382009-08-02 11:05:41 +00002704 pr_err("length: %u != %u\n", *len, size);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002705 ret = -EINVAL;
2706 goto out;
2707 }
Hans Schillstromfc723252011-01-03 14:44:43 +01002708 ret = __ip_vs_get_dest_entries(net, get, user);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002709 }
2710 break;
2711
2712 case IP_VS_SO_GET_TIMEOUT:
2713 {
2714 struct ip_vs_timeout_user t;
2715
Hans Schillstrom93304192011-01-03 14:44:51 +01002716 __ip_vs_get_timeouts(net, &t);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002717 if (copy_to_user(user, &t, sizeof(t)) != 0)
2718 ret = -EFAULT;
2719 }
2720 break;
2721
Linus Torvalds1da177e2005-04-16 15:20:36 -07002722 default:
2723 ret = -EINVAL;
2724 }
2725
Hans Schillstrom552ad652011-06-13 12:19:26 +02002726out:
Ingo Molnar14cc3e22006-03-26 01:37:14 -08002727 mutex_unlock(&__ip_vs_mutex);
Linus Torvalds1da177e2005-04-16 15:20:36 -07002728 return ret;
2729}
2730
2731
2732static struct nf_sockopt_ops ip_vs_sockopts = {
2733 .pf = PF_INET,
2734 .set_optmin = IP_VS_BASE_CTL,
2735 .set_optmax = IP_VS_SO_SET_MAX+1,
2736 .set = do_ip_vs_set_ctl,
2737 .get_optmin = IP_VS_BASE_CTL,
2738 .get_optmax = IP_VS_SO_GET_MAX+1,
2739 .get = do_ip_vs_get_ctl,
Neil Horman16fcec32007-09-11 11:28:26 +02002740 .owner = THIS_MODULE,
Linus Torvalds1da177e2005-04-16 15:20:36 -07002741};
2742
Julius Volz9a812192008-08-14 14:08:44 +02002743/*
2744 * Generic Netlink interface
2745 */
2746
2747/* IPVS genetlink family */
2748static struct genl_family ip_vs_genl_family = {
2749 .id = GENL_ID_GENERATE,
2750 .hdrsize = 0,
2751 .name = IPVS_GENL_NAME,
2752 .version = IPVS_GENL_VERSION,
2753 .maxattr = IPVS_CMD_MAX,
Hans Schillstromc6d2d442011-01-03 14:45:03 +01002754 .netnsok = true, /* Make ipvsadm to work on netns */
Julius Volz9a812192008-08-14 14:08:44 +02002755};
2756
2757/* Policy used for first-level command attributes */
2758static const struct nla_policy ip_vs_cmd_policy[IPVS_CMD_ATTR_MAX + 1] = {
2759 [IPVS_CMD_ATTR_SERVICE] = { .type = NLA_NESTED },
2760 [IPVS_CMD_ATTR_DEST] = { .type = NLA_NESTED },
2761 [IPVS_CMD_ATTR_DAEMON] = { .type = NLA_NESTED },
2762 [IPVS_CMD_ATTR_TIMEOUT_TCP] = { .type = NLA_U32 },
2763 [IPVS_CMD_ATTR_TIMEOUT_TCP_FIN] = { .type = NLA_U32 },
2764 [IPVS_CMD_ATTR_TIMEOUT_UDP] = { .type = NLA_U32 },
2765};
2766
2767/* Policy used for attributes in nested attribute IPVS_CMD_ATTR_DAEMON */
2768static const struct nla_policy ip_vs_daemon_policy[IPVS_DAEMON_ATTR_MAX + 1] = {
2769 [IPVS_DAEMON_ATTR_STATE] = { .type = NLA_U32 },
2770 [IPVS_DAEMON_ATTR_MCAST_IFN] = { .type = NLA_NUL_STRING,
2771 .len = IP_VS_IFNAME_MAXLEN },
2772 [IPVS_DAEMON_ATTR_SYNC_ID] = { .type = NLA_U32 },
2773};
2774
2775/* Policy used for attributes in nested attribute IPVS_CMD_ATTR_SERVICE */
2776static const struct nla_policy ip_vs_svc_policy[IPVS_SVC_ATTR_MAX + 1] = {
2777 [IPVS_SVC_ATTR_AF] = { .type = NLA_U16 },
2778 [IPVS_SVC_ATTR_PROTOCOL] = { .type = NLA_U16 },
2779 [IPVS_SVC_ATTR_ADDR] = { .type = NLA_BINARY,
2780 .len = sizeof(union nf_inet_addr) },
2781 [IPVS_SVC_ATTR_PORT] = { .type = NLA_U16 },
2782 [IPVS_SVC_ATTR_FWMARK] = { .type = NLA_U32 },
2783 [IPVS_SVC_ATTR_SCHED_NAME] = { .type = NLA_NUL_STRING,
2784 .len = IP_VS_SCHEDNAME_MAXLEN },
Simon Horman0d1e71b2010-08-22 21:37:54 +09002785 [IPVS_SVC_ATTR_PE_NAME] = { .type = NLA_NUL_STRING,
2786 .len = IP_VS_PENAME_MAXLEN },
Julius Volz9a812192008-08-14 14:08:44 +02002787 [IPVS_SVC_ATTR_FLAGS] = { .type = NLA_BINARY,
2788 .len = sizeof(struct ip_vs_flags) },
2789 [IPVS_SVC_ATTR_TIMEOUT] = { .type = NLA_U32 },
2790 [IPVS_SVC_ATTR_NETMASK] = { .type = NLA_U32 },
2791 [IPVS_SVC_ATTR_STATS] = { .type = NLA_NESTED },
2792};
2793
2794/* Policy used for attributes in nested attribute IPVS_CMD_ATTR_DEST */
2795static const struct nla_policy ip_vs_dest_policy[IPVS_DEST_ATTR_MAX + 1] = {
2796 [IPVS_DEST_ATTR_ADDR] = { .type = NLA_BINARY,
2797 .len = sizeof(union nf_inet_addr) },
2798 [IPVS_DEST_ATTR_PORT] = { .type = NLA_U16 },
2799 [IPVS_DEST_ATTR_FWD_METHOD] = { .type = NLA_U32 },
2800 [IPVS_DEST_ATTR_WEIGHT] = { .type = NLA_U32 },
2801 [IPVS_DEST_ATTR_U_THRESH] = { .type = NLA_U32 },
2802 [IPVS_DEST_ATTR_L_THRESH] = { .type = NLA_U32 },
2803 [IPVS_DEST_ATTR_ACTIVE_CONNS] = { .type = NLA_U32 },
2804 [IPVS_DEST_ATTR_INACT_CONNS] = { .type = NLA_U32 },
2805 [IPVS_DEST_ATTR_PERSIST_CONNS] = { .type = NLA_U32 },
2806 [IPVS_DEST_ATTR_STATS] = { .type = NLA_NESTED },
2807};
2808
2809static int ip_vs_genl_fill_stats(struct sk_buff *skb, int container_type,
2810 struct ip_vs_stats *stats)
2811{
Julian Anastasov55a3d4e2011-03-14 01:37:49 +02002812 struct ip_vs_stats_user ustats;
Julius Volz9a812192008-08-14 14:08:44 +02002813 struct nlattr *nl_stats = nla_nest_start(skb, container_type);
2814 if (!nl_stats)
2815 return -EMSGSIZE;
2816
Julian Anastasov55a3d4e2011-03-14 01:37:49 +02002817 ip_vs_copy_stats(&ustats, stats);
Julius Volz9a812192008-08-14 14:08:44 +02002818
Julian Anastasov55a3d4e2011-03-14 01:37:49 +02002819 NLA_PUT_U32(skb, IPVS_STATS_ATTR_CONNS, ustats.conns);
2820 NLA_PUT_U32(skb, IPVS_STATS_ATTR_INPKTS, ustats.inpkts);
2821 NLA_PUT_U32(skb, IPVS_STATS_ATTR_OUTPKTS, ustats.outpkts);
2822 NLA_PUT_U64(skb, IPVS_STATS_ATTR_INBYTES, ustats.inbytes);
2823 NLA_PUT_U64(skb, IPVS_STATS_ATTR_OUTBYTES, ustats.outbytes);
2824 NLA_PUT_U32(skb, IPVS_STATS_ATTR_CPS, ustats.cps);
2825 NLA_PUT_U32(skb, IPVS_STATS_ATTR_INPPS, ustats.inpps);
2826 NLA_PUT_U32(skb, IPVS_STATS_ATTR_OUTPPS, ustats.outpps);
2827 NLA_PUT_U32(skb, IPVS_STATS_ATTR_INBPS, ustats.inbps);
2828 NLA_PUT_U32(skb, IPVS_STATS_ATTR_OUTBPS, ustats.outbps);
Julius Volz9a812192008-08-14 14:08:44 +02002829
2830 nla_nest_end(skb, nl_stats);
2831
2832 return 0;
2833
2834nla_put_failure:
Julius Volz9a812192008-08-14 14:08:44 +02002835 nla_nest_cancel(skb, nl_stats);
2836 return -EMSGSIZE;
2837}
2838
2839static int ip_vs_genl_fill_service(struct sk_buff *skb,
2840 struct ip_vs_service *svc)
2841{
2842 struct nlattr *nl_service;
2843 struct ip_vs_flags flags = { .flags = svc->flags,
2844 .mask = ~0 };
2845
2846 nl_service = nla_nest_start(skb, IPVS_CMD_ATTR_SERVICE);
2847 if (!nl_service)
2848 return -EMSGSIZE;
2849
Julius Volzf94fd042008-09-02 15:55:55 +02002850 NLA_PUT_U16(skb, IPVS_SVC_ATTR_AF, svc->af);
Julius Volz9a812192008-08-14 14:08:44 +02002851
2852 if (svc->fwmark) {
2853 NLA_PUT_U32(skb, IPVS_SVC_ATTR_FWMARK, svc->fwmark);
2854 } else {
2855 NLA_PUT_U16(skb, IPVS_SVC_ATTR_PROTOCOL, svc->protocol);
2856 NLA_PUT(skb, IPVS_SVC_ATTR_ADDR, sizeof(svc->addr), &svc->addr);
2857 NLA_PUT_U16(skb, IPVS_SVC_ATTR_PORT, svc->port);
2858 }
2859
2860 NLA_PUT_STRING(skb, IPVS_SVC_ATTR_SCHED_NAME, svc->scheduler->name);
Simon Horman0d1e71b2010-08-22 21:37:54 +09002861 if (svc->pe)
2862 NLA_PUT_STRING(skb, IPVS_SVC_ATTR_PE_NAME, svc->pe->name);
Julius Volz9a812192008-08-14 14:08:44 +02002863 NLA_PUT(skb, IPVS_SVC_ATTR_FLAGS, sizeof(flags), &flags);
2864 NLA_PUT_U32(skb, IPVS_SVC_ATTR_TIMEOUT, svc->timeout / HZ);
2865 NLA_PUT_U32(skb, IPVS_SVC_ATTR_NETMASK, svc->netmask);
2866
2867 if (ip_vs_genl_fill_stats(skb, IPVS_SVC_ATTR_STATS, &svc->stats))
2868 goto nla_put_failure;
2869
2870 nla_nest_end(skb, nl_service);
2871
2872 return 0;
2873
2874nla_put_failure:
2875 nla_nest_cancel(skb, nl_service);
2876 return -EMSGSIZE;
2877}
2878
2879static int ip_vs_genl_dump_service(struct sk_buff *skb,
2880 struct ip_vs_service *svc,
2881 struct netlink_callback *cb)
2882{
2883 void *hdr;
2884
2885 hdr = genlmsg_put(skb, NETLINK_CB(cb->skb).pid, cb->nlh->nlmsg_seq,
2886 &ip_vs_genl_family, NLM_F_MULTI,
2887 IPVS_CMD_NEW_SERVICE);
2888 if (!hdr)
2889 return -EMSGSIZE;
2890
2891 if (ip_vs_genl_fill_service(skb, svc) < 0)
2892 goto nla_put_failure;
2893
2894 return genlmsg_end(skb, hdr);
2895
2896nla_put_failure:
2897 genlmsg_cancel(skb, hdr);
2898 return -EMSGSIZE;
2899}
2900
2901static int ip_vs_genl_dump_services(struct sk_buff *skb,
2902 struct netlink_callback *cb)
2903{
2904 int idx = 0, i;
2905 int start = cb->args[0];
2906 struct ip_vs_service *svc;
Hans Schillstromfc723252011-01-03 14:44:43 +01002907 struct net *net = skb_sknet(skb);
Julius Volz9a812192008-08-14 14:08:44 +02002908
2909 mutex_lock(&__ip_vs_mutex);
2910 for (i = 0; i < IP_VS_SVC_TAB_SIZE; i++) {
2911 list_for_each_entry(svc, &ip_vs_svc_table[i], s_list) {
Hans Schillstromfc723252011-01-03 14:44:43 +01002912 if (++idx <= start || !net_eq(svc->net, net))
Julius Volz9a812192008-08-14 14:08:44 +02002913 continue;
2914 if (ip_vs_genl_dump_service(skb, svc, cb) < 0) {
2915 idx--;
2916 goto nla_put_failure;
2917 }
2918 }
2919 }
2920
2921 for (i = 0; i < IP_VS_SVC_TAB_SIZE; i++) {
2922 list_for_each_entry(svc, &ip_vs_svc_fwm_table[i], f_list) {
Hans Schillstromfc723252011-01-03 14:44:43 +01002923 if (++idx <= start || !net_eq(svc->net, net))
Julius Volz9a812192008-08-14 14:08:44 +02002924 continue;
2925 if (ip_vs_genl_dump_service(skb, svc, cb) < 0) {
2926 idx--;
2927 goto nla_put_failure;
2928 }
2929 }
2930 }
2931
2932nla_put_failure:
2933 mutex_unlock(&__ip_vs_mutex);
2934 cb->args[0] = idx;
2935
2936 return skb->len;
2937}
2938
Hans Schillstromfc723252011-01-03 14:44:43 +01002939static int ip_vs_genl_parse_service(struct net *net,
2940 struct ip_vs_service_user_kern *usvc,
Julian Anastasov26c15cf2010-09-21 18:12:30 +02002941 struct nlattr *nla, int full_entry,
2942 struct ip_vs_service **ret_svc)
Julius Volz9a812192008-08-14 14:08:44 +02002943{
2944 struct nlattr *attrs[IPVS_SVC_ATTR_MAX + 1];
2945 struct nlattr *nla_af, *nla_port, *nla_fwmark, *nla_protocol, *nla_addr;
Julian Anastasov26c15cf2010-09-21 18:12:30 +02002946 struct ip_vs_service *svc;
Julius Volz9a812192008-08-14 14:08:44 +02002947
2948 /* Parse mandatory identifying service fields first */
2949 if (nla == NULL ||
2950 nla_parse_nested(attrs, IPVS_SVC_ATTR_MAX, nla, ip_vs_svc_policy))
2951 return -EINVAL;
2952
2953 nla_af = attrs[IPVS_SVC_ATTR_AF];
2954 nla_protocol = attrs[IPVS_SVC_ATTR_PROTOCOL];
2955 nla_addr = attrs[IPVS_SVC_ATTR_ADDR];
2956 nla_port = attrs[IPVS_SVC_ATTR_PORT];
2957 nla_fwmark = attrs[IPVS_SVC_ATTR_FWMARK];
2958
2959 if (!(nla_af && (nla_fwmark || (nla_port && nla_protocol && nla_addr))))
2960 return -EINVAL;
2961
Simon Horman258c8892009-12-15 17:01:25 +01002962 memset(usvc, 0, sizeof(*usvc));
2963
Julius Volzc860c6b2008-09-02 15:55:36 +02002964 usvc->af = nla_get_u16(nla_af);
Julius Volzf94fd042008-09-02 15:55:55 +02002965#ifdef CONFIG_IP_VS_IPV6
2966 if (usvc->af != AF_INET && usvc->af != AF_INET6)
2967#else
2968 if (usvc->af != AF_INET)
2969#endif
Julius Volz9a812192008-08-14 14:08:44 +02002970 return -EAFNOSUPPORT;
2971
2972 if (nla_fwmark) {
2973 usvc->protocol = IPPROTO_TCP;
2974 usvc->fwmark = nla_get_u32(nla_fwmark);
2975 } else {
2976 usvc->protocol = nla_get_u16(nla_protocol);
2977 nla_memcpy(&usvc->addr, nla_addr, sizeof(usvc->addr));
2978 usvc->port = nla_get_u16(nla_port);
2979 usvc->fwmark = 0;
2980 }
2981
Julian Anastasov26c15cf2010-09-21 18:12:30 +02002982 if (usvc->fwmark)
Hans Schillstromfc723252011-01-03 14:44:43 +01002983 svc = __ip_vs_svc_fwm_find(net, usvc->af, usvc->fwmark);
Julian Anastasov26c15cf2010-09-21 18:12:30 +02002984 else
Hans Schillstromfc723252011-01-03 14:44:43 +01002985 svc = __ip_vs_service_find(net, usvc->af, usvc->protocol,
Julian Anastasov26c15cf2010-09-21 18:12:30 +02002986 &usvc->addr, usvc->port);
2987 *ret_svc = svc;
2988
Julius Volz9a812192008-08-14 14:08:44 +02002989 /* If a full entry was requested, check for the additional fields */
2990 if (full_entry) {
Simon Horman0d1e71b2010-08-22 21:37:54 +09002991 struct nlattr *nla_sched, *nla_flags, *nla_pe, *nla_timeout,
Julius Volz9a812192008-08-14 14:08:44 +02002992 *nla_netmask;
2993 struct ip_vs_flags flags;
Julius Volz9a812192008-08-14 14:08:44 +02002994
2995 nla_sched = attrs[IPVS_SVC_ATTR_SCHED_NAME];
Simon Horman0d1e71b2010-08-22 21:37:54 +09002996 nla_pe = attrs[IPVS_SVC_ATTR_PE_NAME];
Julius Volz9a812192008-08-14 14:08:44 +02002997 nla_flags = attrs[IPVS_SVC_ATTR_FLAGS];
2998 nla_timeout = attrs[IPVS_SVC_ATTR_TIMEOUT];
2999 nla_netmask = attrs[IPVS_SVC_ATTR_NETMASK];
3000
3001 if (!(nla_sched && nla_flags && nla_timeout && nla_netmask))
3002 return -EINVAL;
3003
3004 nla_memcpy(&flags, nla_flags, sizeof(flags));
3005
3006 /* prefill flags from service if it already exists */
Julian Anastasov26c15cf2010-09-21 18:12:30 +02003007 if (svc)
Julius Volz9a812192008-08-14 14:08:44 +02003008 usvc->flags = svc->flags;
Julius Volz9a812192008-08-14 14:08:44 +02003009
3010 /* set new flags from userland */
3011 usvc->flags = (usvc->flags & ~flags.mask) |
3012 (flags.flags & flags.mask);
Julius Volzc860c6b2008-09-02 15:55:36 +02003013 usvc->sched_name = nla_data(nla_sched);
Simon Horman0d1e71b2010-08-22 21:37:54 +09003014 usvc->pe_name = nla_pe ? nla_data(nla_pe) : NULL;
Julius Volz9a812192008-08-14 14:08:44 +02003015 usvc->timeout = nla_get_u32(nla_timeout);
3016 usvc->netmask = nla_get_u32(nla_netmask);
3017 }
3018
3019 return 0;
3020}
3021
Hans Schillstromfc723252011-01-03 14:44:43 +01003022static struct ip_vs_service *ip_vs_genl_find_service(struct net *net,
3023 struct nlattr *nla)
Julius Volz9a812192008-08-14 14:08:44 +02003024{
Julius Volzc860c6b2008-09-02 15:55:36 +02003025 struct ip_vs_service_user_kern usvc;
Julian Anastasov26c15cf2010-09-21 18:12:30 +02003026 struct ip_vs_service *svc;
Julius Volz9a812192008-08-14 14:08:44 +02003027 int ret;
3028
Hans Schillstromfc723252011-01-03 14:44:43 +01003029 ret = ip_vs_genl_parse_service(net, &usvc, nla, 0, &svc);
Julian Anastasov26c15cf2010-09-21 18:12:30 +02003030 return ret ? ERR_PTR(ret) : svc;
Julius Volz9a812192008-08-14 14:08:44 +02003031}
3032
3033static int ip_vs_genl_fill_dest(struct sk_buff *skb, struct ip_vs_dest *dest)
3034{
3035 struct nlattr *nl_dest;
3036
3037 nl_dest = nla_nest_start(skb, IPVS_CMD_ATTR_DEST);
3038 if (!nl_dest)
3039 return -EMSGSIZE;
3040
3041 NLA_PUT(skb, IPVS_DEST_ATTR_ADDR, sizeof(dest->addr), &dest->addr);
3042 NLA_PUT_U16(skb, IPVS_DEST_ATTR_PORT, dest->port);
3043
3044 NLA_PUT_U32(skb, IPVS_DEST_ATTR_FWD_METHOD,
3045 atomic_read(&dest->conn_flags) & IP_VS_CONN_F_FWD_MASK);
3046 NLA_PUT_U32(skb, IPVS_DEST_ATTR_WEIGHT, atomic_read(&dest->weight));
3047 NLA_PUT_U32(skb, IPVS_DEST_ATTR_U_THRESH, dest->u_threshold);
3048 NLA_PUT_U32(skb, IPVS_DEST_ATTR_L_THRESH, dest->l_threshold);
3049 NLA_PUT_U32(skb, IPVS_DEST_ATTR_ACTIVE_CONNS,
3050 atomic_read(&dest->activeconns));
3051 NLA_PUT_U32(skb, IPVS_DEST_ATTR_INACT_CONNS,
3052 atomic_read(&dest->inactconns));
3053 NLA_PUT_U32(skb, IPVS_DEST_ATTR_PERSIST_CONNS,
3054 atomic_read(&dest->persistconns));
3055
3056 if (ip_vs_genl_fill_stats(skb, IPVS_DEST_ATTR_STATS, &dest->stats))
3057 goto nla_put_failure;
3058
3059 nla_nest_end(skb, nl_dest);
3060
3061 return 0;
3062
3063nla_put_failure:
3064 nla_nest_cancel(skb, nl_dest);
3065 return -EMSGSIZE;
3066}
3067
3068static int ip_vs_genl_dump_dest(struct sk_buff *skb, struct ip_vs_dest *dest,
3069 struct netlink_callback *cb)
3070{
3071 void *hdr;
3072
3073 hdr = genlmsg_put(skb, NETLINK_CB(cb->skb).pid, cb->nlh->nlmsg_seq,
3074 &ip_vs_genl_family, NLM_F_MULTI,
3075 IPVS_CMD_NEW_DEST);
3076 if (!hdr)
3077 return -EMSGSIZE;
3078
3079 if (ip_vs_genl_fill_dest(skb, dest) < 0)
3080 goto nla_put_failure;
3081
3082 return genlmsg_end(skb, hdr);
3083
3084nla_put_failure:
3085 genlmsg_cancel(skb, hdr);
3086 return -EMSGSIZE;
3087}
3088
3089static int ip_vs_genl_dump_dests(struct sk_buff *skb,
3090 struct netlink_callback *cb)
3091{
3092 int idx = 0;
3093 int start = cb->args[0];
3094 struct ip_vs_service *svc;
3095 struct ip_vs_dest *dest;
3096 struct nlattr *attrs[IPVS_CMD_ATTR_MAX + 1];
Hans Schillstroma0840e22011-01-03 14:44:58 +01003097 struct net *net = skb_sknet(skb);
Julius Volz9a812192008-08-14 14:08:44 +02003098
3099 mutex_lock(&__ip_vs_mutex);
3100
3101 /* Try to find the service for which to dump destinations */
3102 if (nlmsg_parse(cb->nlh, GENL_HDRLEN, attrs,
3103 IPVS_CMD_ATTR_MAX, ip_vs_cmd_policy))
3104 goto out_err;
3105
Hans Schillstroma0840e22011-01-03 14:44:58 +01003106
Hans Schillstromfc723252011-01-03 14:44:43 +01003107 svc = ip_vs_genl_find_service(net, attrs[IPVS_CMD_ATTR_SERVICE]);
Julius Volz9a812192008-08-14 14:08:44 +02003108 if (IS_ERR(svc) || svc == NULL)
3109 goto out_err;
3110
3111 /* Dump the destinations */
3112 list_for_each_entry(dest, &svc->destinations, n_list) {
3113 if (++idx <= start)
3114 continue;
3115 if (ip_vs_genl_dump_dest(skb, dest, cb) < 0) {
3116 idx--;
3117 goto nla_put_failure;
3118 }
3119 }
3120
3121nla_put_failure:
3122 cb->args[0] = idx;
Julius Volz9a812192008-08-14 14:08:44 +02003123
3124out_err:
3125 mutex_unlock(&__ip_vs_mutex);
3126
3127 return skb->len;
3128}
3129
Julius Volzc860c6b2008-09-02 15:55:36 +02003130static int ip_vs_genl_parse_dest(struct ip_vs_dest_user_kern *udest,
Julius Volz9a812192008-08-14 14:08:44 +02003131 struct nlattr *nla, int full_entry)
3132{
3133 struct nlattr *attrs[IPVS_DEST_ATTR_MAX + 1];
3134 struct nlattr *nla_addr, *nla_port;
3135
3136 /* Parse mandatory identifying destination fields first */
3137 if (nla == NULL ||
3138 nla_parse_nested(attrs, IPVS_DEST_ATTR_MAX, nla, ip_vs_dest_policy))
3139 return -EINVAL;
3140
3141 nla_addr = attrs[IPVS_DEST_ATTR_ADDR];
3142 nla_port = attrs[IPVS_DEST_ATTR_PORT];
3143
3144 if (!(nla_addr && nla_port))
3145 return -EINVAL;
3146
Simon Horman258c8892009-12-15 17:01:25 +01003147 memset(udest, 0, sizeof(*udest));
3148
Julius Volz9a812192008-08-14 14:08:44 +02003149 nla_memcpy(&udest->addr, nla_addr, sizeof(udest->addr));
3150 udest->port = nla_get_u16(nla_port);
3151
3152 /* If a full entry was requested, check for the additional fields */
3153 if (full_entry) {
3154 struct nlattr *nla_fwd, *nla_weight, *nla_u_thresh,
3155 *nla_l_thresh;
3156
3157 nla_fwd = attrs[IPVS_DEST_ATTR_FWD_METHOD];
3158 nla_weight = attrs[IPVS_DEST_ATTR_WEIGHT];
3159 nla_u_thresh = attrs[IPVS_DEST_ATTR_U_THRESH];
3160 nla_l_thresh = attrs[IPVS_DEST_ATTR_L_THRESH];
3161
3162 if (!(nla_fwd && nla_weight && nla_u_thresh && nla_l_thresh))
3163 return -EINVAL;
3164
3165 udest->conn_flags = nla_get_u32(nla_fwd)
3166 & IP_VS_CONN_F_FWD_MASK;
3167 udest->weight = nla_get_u32(nla_weight);
3168 udest->u_threshold = nla_get_u32(nla_u_thresh);
3169 udest->l_threshold = nla_get_u32(nla_l_thresh);
3170 }
3171
3172 return 0;
3173}
3174
3175static int ip_vs_genl_fill_daemon(struct sk_buff *skb, __be32 state,
3176 const char *mcast_ifn, __be32 syncid)
3177{
3178 struct nlattr *nl_daemon;
3179
3180 nl_daemon = nla_nest_start(skb, IPVS_CMD_ATTR_DAEMON);
3181 if (!nl_daemon)
3182 return -EMSGSIZE;
3183
3184 NLA_PUT_U32(skb, IPVS_DAEMON_ATTR_STATE, state);
3185 NLA_PUT_STRING(skb, IPVS_DAEMON_ATTR_MCAST_IFN, mcast_ifn);
3186 NLA_PUT_U32(skb, IPVS_DAEMON_ATTR_SYNC_ID, syncid);
3187
3188 nla_nest_end(skb, nl_daemon);
3189
3190 return 0;
3191
3192nla_put_failure:
3193 nla_nest_cancel(skb, nl_daemon);
3194 return -EMSGSIZE;
3195}
3196
3197static int ip_vs_genl_dump_daemon(struct sk_buff *skb, __be32 state,
3198 const char *mcast_ifn, __be32 syncid,
3199 struct netlink_callback *cb)
3200{
3201 void *hdr;
3202 hdr = genlmsg_put(skb, NETLINK_CB(cb->skb).pid, cb->nlh->nlmsg_seq,
3203 &ip_vs_genl_family, NLM_F_MULTI,
3204 IPVS_CMD_NEW_DAEMON);
3205 if (!hdr)
3206 return -EMSGSIZE;
3207
3208 if (ip_vs_genl_fill_daemon(skb, state, mcast_ifn, syncid))
3209 goto nla_put_failure;
3210
3211 return genlmsg_end(skb, hdr);
3212
3213nla_put_failure:
3214 genlmsg_cancel(skb, hdr);
3215 return -EMSGSIZE;
3216}
3217
3218static int ip_vs_genl_dump_daemons(struct sk_buff *skb,
3219 struct netlink_callback *cb)
3220{
Hans Schillstroma09d1972011-04-04 15:25:18 +02003221 struct net *net = skb_sknet(skb);
Hans Schillstromf1313152011-01-03 14:44:55 +01003222 struct netns_ipvs *ipvs = net_ipvs(net);
3223
Hans Schillstromae1d48b2011-10-11 10:54:35 +09003224 mutex_lock(&ipvs->sync_mutex);
Hans Schillstromf1313152011-01-03 14:44:55 +01003225 if ((ipvs->sync_state & IP_VS_STATE_MASTER) && !cb->args[0]) {
Julius Volz9a812192008-08-14 14:08:44 +02003226 if (ip_vs_genl_dump_daemon(skb, IP_VS_STATE_MASTER,
Hans Schillstromf1313152011-01-03 14:44:55 +01003227 ipvs->master_mcast_ifn,
3228 ipvs->master_syncid, cb) < 0)
Julius Volz9a812192008-08-14 14:08:44 +02003229 goto nla_put_failure;
3230
3231 cb->args[0] = 1;
3232 }
3233
Hans Schillstromf1313152011-01-03 14:44:55 +01003234 if ((ipvs->sync_state & IP_VS_STATE_BACKUP) && !cb->args[1]) {
Julius Volz9a812192008-08-14 14:08:44 +02003235 if (ip_vs_genl_dump_daemon(skb, IP_VS_STATE_BACKUP,
Hans Schillstromf1313152011-01-03 14:44:55 +01003236 ipvs->backup_mcast_ifn,
3237 ipvs->backup_syncid, cb) < 0)
Julius Volz9a812192008-08-14 14:08:44 +02003238 goto nla_put_failure;
3239
3240 cb->args[1] = 1;
3241 }
3242
3243nla_put_failure:
Hans Schillstromae1d48b2011-10-11 10:54:35 +09003244 mutex_unlock(&ipvs->sync_mutex);
Julius Volz9a812192008-08-14 14:08:44 +02003245
3246 return skb->len;
3247}
3248
Hans Schillstromf1313152011-01-03 14:44:55 +01003249static int ip_vs_genl_new_daemon(struct net *net, struct nlattr **attrs)
Julius Volz9a812192008-08-14 14:08:44 +02003250{
3251 if (!(attrs[IPVS_DAEMON_ATTR_STATE] &&
3252 attrs[IPVS_DAEMON_ATTR_MCAST_IFN] &&
3253 attrs[IPVS_DAEMON_ATTR_SYNC_ID]))
3254 return -EINVAL;
3255
Hans Schillstromf1313152011-01-03 14:44:55 +01003256 return start_sync_thread(net,
3257 nla_get_u32(attrs[IPVS_DAEMON_ATTR_STATE]),
Julius Volz9a812192008-08-14 14:08:44 +02003258 nla_data(attrs[IPVS_DAEMON_ATTR_MCAST_IFN]),
3259 nla_get_u32(attrs[IPVS_DAEMON_ATTR_SYNC_ID]));
3260}
3261
Hans Schillstromf1313152011-01-03 14:44:55 +01003262static int ip_vs_genl_del_daemon(struct net *net, struct nlattr **attrs)
Julius Volz9a812192008-08-14 14:08:44 +02003263{
3264 if (!attrs[IPVS_DAEMON_ATTR_STATE])
3265 return -EINVAL;
3266
Hans Schillstromf1313152011-01-03 14:44:55 +01003267 return stop_sync_thread(net,
3268 nla_get_u32(attrs[IPVS_DAEMON_ATTR_STATE]));
Julius Volz9a812192008-08-14 14:08:44 +02003269}
3270
Hans Schillstrom93304192011-01-03 14:44:51 +01003271static int ip_vs_genl_set_config(struct net *net, struct nlattr **attrs)
Julius Volz9a812192008-08-14 14:08:44 +02003272{
3273 struct ip_vs_timeout_user t;
3274
Hans Schillstrom93304192011-01-03 14:44:51 +01003275 __ip_vs_get_timeouts(net, &t);
Julius Volz9a812192008-08-14 14:08:44 +02003276
3277 if (attrs[IPVS_CMD_ATTR_TIMEOUT_TCP])
3278 t.tcp_timeout = nla_get_u32(attrs[IPVS_CMD_ATTR_TIMEOUT_TCP]);
3279
3280 if (attrs[IPVS_CMD_ATTR_TIMEOUT_TCP_FIN])
3281 t.tcp_fin_timeout =
3282 nla_get_u32(attrs[IPVS_CMD_ATTR_TIMEOUT_TCP_FIN]);
3283
3284 if (attrs[IPVS_CMD_ATTR_TIMEOUT_UDP])
3285 t.udp_timeout = nla_get_u32(attrs[IPVS_CMD_ATTR_TIMEOUT_UDP]);
3286
Hans Schillstrom93304192011-01-03 14:44:51 +01003287 return ip_vs_set_timeout(net, &t);
Julius Volz9a812192008-08-14 14:08:44 +02003288}
3289
Hans Schillstromae1d48b2011-10-11 10:54:35 +09003290static int ip_vs_genl_set_daemon(struct sk_buff *skb, struct genl_info *info)
3291{
3292 int ret = 0, cmd;
3293 struct net *net;
3294 struct netns_ipvs *ipvs;
3295
3296 net = skb_sknet(skb);
3297 ipvs = net_ipvs(net);
3298 cmd = info->genlhdr->cmd;
3299
3300 if (cmd == IPVS_CMD_NEW_DAEMON || cmd == IPVS_CMD_DEL_DAEMON) {
3301 struct nlattr *daemon_attrs[IPVS_DAEMON_ATTR_MAX + 1];
3302
3303 mutex_lock(&ipvs->sync_mutex);
3304 if (!info->attrs[IPVS_CMD_ATTR_DAEMON] ||
3305 nla_parse_nested(daemon_attrs, IPVS_DAEMON_ATTR_MAX,
3306 info->attrs[IPVS_CMD_ATTR_DAEMON],
3307 ip_vs_daemon_policy)) {
3308 ret = -EINVAL;
3309 goto out;
3310 }
3311
3312 if (cmd == IPVS_CMD_NEW_DAEMON)
3313 ret = ip_vs_genl_new_daemon(net, daemon_attrs);
3314 else
3315 ret = ip_vs_genl_del_daemon(net, daemon_attrs);
3316out:
3317 mutex_unlock(&ipvs->sync_mutex);
3318 }
3319 return ret;
3320}
3321
Julius Volz9a812192008-08-14 14:08:44 +02003322static int ip_vs_genl_set_cmd(struct sk_buff *skb, struct genl_info *info)
3323{
3324 struct ip_vs_service *svc = NULL;
Julius Volzc860c6b2008-09-02 15:55:36 +02003325 struct ip_vs_service_user_kern usvc;
3326 struct ip_vs_dest_user_kern udest;
Julius Volz9a812192008-08-14 14:08:44 +02003327 int ret = 0, cmd;
3328 int need_full_svc = 0, need_full_dest = 0;
Hans Schillstromfc723252011-01-03 14:44:43 +01003329 struct net *net;
Julius Volz9a812192008-08-14 14:08:44 +02003330
Hans Schillstromfc723252011-01-03 14:44:43 +01003331 net = skb_sknet(skb);
Julius Volz9a812192008-08-14 14:08:44 +02003332 cmd = info->genlhdr->cmd;
3333
3334 mutex_lock(&__ip_vs_mutex);
3335
3336 if (cmd == IPVS_CMD_FLUSH) {
Hans Schillstromfc723252011-01-03 14:44:43 +01003337 ret = ip_vs_flush(net);
Julius Volz9a812192008-08-14 14:08:44 +02003338 goto out;
3339 } else if (cmd == IPVS_CMD_SET_CONFIG) {
Hans Schillstrom93304192011-01-03 14:44:51 +01003340 ret = ip_vs_genl_set_config(net, info->attrs);
Julius Volz9a812192008-08-14 14:08:44 +02003341 goto out;
Julius Volz9a812192008-08-14 14:08:44 +02003342 } else if (cmd == IPVS_CMD_ZERO &&
3343 !info->attrs[IPVS_CMD_ATTR_SERVICE]) {
Hans Schillstromfc723252011-01-03 14:44:43 +01003344 ret = ip_vs_zero_all(net);
Julius Volz9a812192008-08-14 14:08:44 +02003345 goto out;
3346 }
3347
3348 /* All following commands require a service argument, so check if we
3349 * received a valid one. We need a full service specification when
3350 * adding / editing a service. Only identifying members otherwise. */
3351 if (cmd == IPVS_CMD_NEW_SERVICE || cmd == IPVS_CMD_SET_SERVICE)
3352 need_full_svc = 1;
3353
Hans Schillstromfc723252011-01-03 14:44:43 +01003354 ret = ip_vs_genl_parse_service(net, &usvc,
Julius Volz9a812192008-08-14 14:08:44 +02003355 info->attrs[IPVS_CMD_ATTR_SERVICE],
Julian Anastasov26c15cf2010-09-21 18:12:30 +02003356 need_full_svc, &svc);
Julius Volz9a812192008-08-14 14:08:44 +02003357 if (ret)
3358 goto out;
3359
Julius Volz9a812192008-08-14 14:08:44 +02003360 /* Unless we're adding a new service, the service must already exist */
3361 if ((cmd != IPVS_CMD_NEW_SERVICE) && (svc == NULL)) {
3362 ret = -ESRCH;
3363 goto out;
3364 }
3365
3366 /* Destination commands require a valid destination argument. For
3367 * adding / editing a destination, we need a full destination
3368 * specification. */
3369 if (cmd == IPVS_CMD_NEW_DEST || cmd == IPVS_CMD_SET_DEST ||
3370 cmd == IPVS_CMD_DEL_DEST) {
3371 if (cmd != IPVS_CMD_DEL_DEST)
3372 need_full_dest = 1;
3373
3374 ret = ip_vs_genl_parse_dest(&udest,
3375 info->attrs[IPVS_CMD_ATTR_DEST],
3376 need_full_dest);
3377 if (ret)
3378 goto out;
3379 }
3380
3381 switch (cmd) {
3382 case IPVS_CMD_NEW_SERVICE:
3383 if (svc == NULL)
Hans Schillstromfc723252011-01-03 14:44:43 +01003384 ret = ip_vs_add_service(net, &usvc, &svc);
Julius Volz9a812192008-08-14 14:08:44 +02003385 else
3386 ret = -EEXIST;
3387 break;
3388 case IPVS_CMD_SET_SERVICE:
3389 ret = ip_vs_edit_service(svc, &usvc);
3390 break;
3391 case IPVS_CMD_DEL_SERVICE:
3392 ret = ip_vs_del_service(svc);
Julian Anastasov26c15cf2010-09-21 18:12:30 +02003393 /* do not use svc, it can be freed */
Julius Volz9a812192008-08-14 14:08:44 +02003394 break;
3395 case IPVS_CMD_NEW_DEST:
3396 ret = ip_vs_add_dest(svc, &udest);
3397 break;
3398 case IPVS_CMD_SET_DEST:
3399 ret = ip_vs_edit_dest(svc, &udest);
3400 break;
3401 case IPVS_CMD_DEL_DEST:
3402 ret = ip_vs_del_dest(svc, &udest);
3403 break;
3404 case IPVS_CMD_ZERO:
3405 ret = ip_vs_zero_service(svc);
3406 break;
3407 default:
3408 ret = -EINVAL;
3409 }
3410
3411out:
Julius Volz9a812192008-08-14 14:08:44 +02003412 mutex_unlock(&__ip_vs_mutex);
3413
3414 return ret;
3415}
3416
3417static int ip_vs_genl_get_cmd(struct sk_buff *skb, struct genl_info *info)
3418{
3419 struct sk_buff *msg;
3420 void *reply;
3421 int ret, cmd, reply_cmd;
Hans Schillstromfc723252011-01-03 14:44:43 +01003422 struct net *net;
Julius Volz9a812192008-08-14 14:08:44 +02003423
Hans Schillstromfc723252011-01-03 14:44:43 +01003424 net = skb_sknet(skb);
Julius Volz9a812192008-08-14 14:08:44 +02003425 cmd = info->genlhdr->cmd;
3426
3427 if (cmd == IPVS_CMD_GET_SERVICE)
3428 reply_cmd = IPVS_CMD_NEW_SERVICE;
3429 else if (cmd == IPVS_CMD_GET_INFO)
3430 reply_cmd = IPVS_CMD_SET_INFO;
3431 else if (cmd == IPVS_CMD_GET_CONFIG)
3432 reply_cmd = IPVS_CMD_SET_CONFIG;
3433 else {
Hannes Eder1e3e2382009-08-02 11:05:41 +00003434 pr_err("unknown Generic Netlink command\n");
Julius Volz9a812192008-08-14 14:08:44 +02003435 return -EINVAL;
3436 }
3437
3438 msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
3439 if (!msg)
3440 return -ENOMEM;
3441
3442 mutex_lock(&__ip_vs_mutex);
3443
3444 reply = genlmsg_put_reply(msg, info, &ip_vs_genl_family, 0, reply_cmd);
3445 if (reply == NULL)
3446 goto nla_put_failure;
3447
3448 switch (cmd) {
3449 case IPVS_CMD_GET_SERVICE:
3450 {
3451 struct ip_vs_service *svc;
3452
Hans Schillstromfc723252011-01-03 14:44:43 +01003453 svc = ip_vs_genl_find_service(net,
3454 info->attrs[IPVS_CMD_ATTR_SERVICE]);
Julius Volz9a812192008-08-14 14:08:44 +02003455 if (IS_ERR(svc)) {
3456 ret = PTR_ERR(svc);
3457 goto out_err;
3458 } else if (svc) {
3459 ret = ip_vs_genl_fill_service(msg, svc);
Julius Volz9a812192008-08-14 14:08:44 +02003460 if (ret)
3461 goto nla_put_failure;
3462 } else {
3463 ret = -ESRCH;
3464 goto out_err;
3465 }
3466
3467 break;
3468 }
3469
3470 case IPVS_CMD_GET_CONFIG:
3471 {
3472 struct ip_vs_timeout_user t;
3473
Hans Schillstrom93304192011-01-03 14:44:51 +01003474 __ip_vs_get_timeouts(net, &t);
Julius Volz9a812192008-08-14 14:08:44 +02003475#ifdef CONFIG_IP_VS_PROTO_TCP
3476 NLA_PUT_U32(msg, IPVS_CMD_ATTR_TIMEOUT_TCP, t.tcp_timeout);
3477 NLA_PUT_U32(msg, IPVS_CMD_ATTR_TIMEOUT_TCP_FIN,
3478 t.tcp_fin_timeout);
3479#endif
3480#ifdef CONFIG_IP_VS_PROTO_UDP
3481 NLA_PUT_U32(msg, IPVS_CMD_ATTR_TIMEOUT_UDP, t.udp_timeout);
3482#endif
3483
3484 break;
3485 }
3486
3487 case IPVS_CMD_GET_INFO:
3488 NLA_PUT_U32(msg, IPVS_INFO_ATTR_VERSION, IP_VS_VERSION_CODE);
3489 NLA_PUT_U32(msg, IPVS_INFO_ATTR_CONN_TAB_SIZE,
Catalin(ux) M. BOIE6f7edb42010-01-05 05:50:24 +01003490 ip_vs_conn_tab_size);
Julius Volz9a812192008-08-14 14:08:44 +02003491 break;
3492 }
3493
3494 genlmsg_end(msg, reply);
Johannes Berg134e6372009-07-10 09:51:34 +00003495 ret = genlmsg_reply(msg, info);
Julius Volz9a812192008-08-14 14:08:44 +02003496 goto out;
3497
3498nla_put_failure:
Hannes Eder1e3e2382009-08-02 11:05:41 +00003499 pr_err("not enough space in Netlink message\n");
Julius Volz9a812192008-08-14 14:08:44 +02003500 ret = -EMSGSIZE;
3501
3502out_err:
3503 nlmsg_free(msg);
3504out:
3505 mutex_unlock(&__ip_vs_mutex);
3506
3507 return ret;
3508}
3509
3510
3511static struct genl_ops ip_vs_genl_ops[] __read_mostly = {
3512 {
3513 .cmd = IPVS_CMD_NEW_SERVICE,
3514 .flags = GENL_ADMIN_PERM,
3515 .policy = ip_vs_cmd_policy,
3516 .doit = ip_vs_genl_set_cmd,
3517 },
3518 {
3519 .cmd = IPVS_CMD_SET_SERVICE,
3520 .flags = GENL_ADMIN_PERM,
3521 .policy = ip_vs_cmd_policy,
3522 .doit = ip_vs_genl_set_cmd,
3523 },
3524 {
3525 .cmd = IPVS_CMD_DEL_SERVICE,
3526 .flags = GENL_ADMIN_PERM,
3527 .policy = ip_vs_cmd_policy,
3528 .doit = ip_vs_genl_set_cmd,
3529 },
3530 {
3531 .cmd = IPVS_CMD_GET_SERVICE,
3532 .flags = GENL_ADMIN_PERM,
3533 .doit = ip_vs_genl_get_cmd,
3534 .dumpit = ip_vs_genl_dump_services,
3535 .policy = ip_vs_cmd_policy,
3536 },
3537 {
3538 .cmd = IPVS_CMD_NEW_DEST,
3539 .flags = GENL_ADMIN_PERM,
3540 .policy = ip_vs_cmd_policy,
3541 .doit = ip_vs_genl_set_cmd,
3542 },
3543 {
3544 .cmd = IPVS_CMD_SET_DEST,
3545 .flags = GENL_ADMIN_PERM,
3546 .policy = ip_vs_cmd_policy,
3547 .doit = ip_vs_genl_set_cmd,
3548 },
3549 {
3550 .cmd = IPVS_CMD_DEL_DEST,
3551 .flags = GENL_ADMIN_PERM,
3552 .policy = ip_vs_cmd_policy,
3553 .doit = ip_vs_genl_set_cmd,
3554 },
3555 {
3556 .cmd = IPVS_CMD_GET_DEST,
3557 .flags = GENL_ADMIN_PERM,
3558 .policy = ip_vs_cmd_policy,
3559 .dumpit = ip_vs_genl_dump_dests,
3560 },
3561 {
3562 .cmd = IPVS_CMD_NEW_DAEMON,
3563 .flags = GENL_ADMIN_PERM,
3564 .policy = ip_vs_cmd_policy,
Hans Schillstromae1d48b2011-10-11 10:54:35 +09003565 .doit = ip_vs_genl_set_daemon,
Julius Volz9a812192008-08-14 14:08:44 +02003566 },
3567 {
3568 .cmd = IPVS_CMD_DEL_DAEMON,
3569 .flags = GENL_ADMIN_PERM,
3570 .policy = ip_vs_cmd_policy,
Hans Schillstromae1d48b2011-10-11 10:54:35 +09003571 .doit = ip_vs_genl_set_daemon,
Julius Volz9a812192008-08-14 14:08:44 +02003572 },
3573 {
3574 .cmd = IPVS_CMD_GET_DAEMON,
3575 .flags = GENL_ADMIN_PERM,
3576 .dumpit = ip_vs_genl_dump_daemons,
3577 },
3578 {
3579 .cmd = IPVS_CMD_SET_CONFIG,
3580 .flags = GENL_ADMIN_PERM,
3581 .policy = ip_vs_cmd_policy,
3582 .doit = ip_vs_genl_set_cmd,
3583 },
3584 {
3585 .cmd = IPVS_CMD_GET_CONFIG,
3586 .flags = GENL_ADMIN_PERM,
3587 .doit = ip_vs_genl_get_cmd,
3588 },
3589 {
3590 .cmd = IPVS_CMD_GET_INFO,
3591 .flags = GENL_ADMIN_PERM,
3592 .doit = ip_vs_genl_get_cmd,
3593 },
3594 {
3595 .cmd = IPVS_CMD_ZERO,
3596 .flags = GENL_ADMIN_PERM,
3597 .policy = ip_vs_cmd_policy,
3598 .doit = ip_vs_genl_set_cmd,
3599 },
3600 {
3601 .cmd = IPVS_CMD_FLUSH,
3602 .flags = GENL_ADMIN_PERM,
3603 .doit = ip_vs_genl_set_cmd,
3604 },
3605};
3606
3607static int __init ip_vs_genl_register(void)
3608{
Michał Mirosław8f698d52009-05-21 10:34:05 +00003609 return genl_register_family_with_ops(&ip_vs_genl_family,
3610 ip_vs_genl_ops, ARRAY_SIZE(ip_vs_genl_ops));
Julius Volz9a812192008-08-14 14:08:44 +02003611}
3612
3613static void ip_vs_genl_unregister(void)
3614{
3615 genl_unregister_family(&ip_vs_genl_family);
3616}
3617
3618/* End of Generic Netlink interface definitions */
3619
Hans Schillstrom61b1ab42011-01-03 14:44:42 +01003620/*
3621 * per netns intit/exit func.
3622 */
Simon Horman14e40542011-02-04 18:33:02 +09003623#ifdef CONFIG_SYSCTL
Hans Schillstrom503cf152011-05-01 18:50:16 +02003624int __net_init ip_vs_control_net_init_sysctl(struct net *net)
Hans Schillstrom61b1ab42011-01-03 14:44:42 +01003625{
Hans Schillstromfc723252011-01-03 14:44:43 +01003626 int idx;
3627 struct netns_ipvs *ipvs = net_ipvs(net);
Hans Schillstroma0840e22011-01-03 14:44:58 +01003628 struct ctl_table *tbl;
Hans Schillstromfc723252011-01-03 14:44:43 +01003629
Hans Schillstroma0840e22011-01-03 14:44:58 +01003630 atomic_set(&ipvs->dropentry, 0);
3631 spin_lock_init(&ipvs->dropentry_lock);
3632 spin_lock_init(&ipvs->droppacket_lock);
3633 spin_lock_init(&ipvs->securetcp_lock);
Hans Schillstroma0840e22011-01-03 14:44:58 +01003634
3635 if (!net_eq(net, &init_net)) {
3636 tbl = kmemdup(vs_vars, sizeof(vs_vars), GFP_KERNEL);
3637 if (tbl == NULL)
Simon Horman14e40542011-02-04 18:33:02 +09003638 return -ENOMEM;
Hans Schillstroma0840e22011-01-03 14:44:58 +01003639 } else
3640 tbl = vs_vars;
3641 /* Initialize sysctl defaults */
3642 idx = 0;
3643 ipvs->sysctl_amemthresh = 1024;
3644 tbl[idx++].data = &ipvs->sysctl_amemthresh;
3645 ipvs->sysctl_am_droprate = 10;
3646 tbl[idx++].data = &ipvs->sysctl_am_droprate;
3647 tbl[idx++].data = &ipvs->sysctl_drop_entry;
3648 tbl[idx++].data = &ipvs->sysctl_drop_packet;
3649#ifdef CONFIG_IP_VS_NFCT
3650 tbl[idx++].data = &ipvs->sysctl_conntrack;
3651#endif
3652 tbl[idx++].data = &ipvs->sysctl_secure_tcp;
3653 ipvs->sysctl_snat_reroute = 1;
3654 tbl[idx++].data = &ipvs->sysctl_snat_reroute;
3655 ipvs->sysctl_sync_ver = 1;
3656 tbl[idx++].data = &ipvs->sysctl_sync_ver;
3657 tbl[idx++].data = &ipvs->sysctl_cache_bypass;
3658 tbl[idx++].data = &ipvs->sysctl_expire_nodest_conn;
3659 tbl[idx++].data = &ipvs->sysctl_expire_quiescent_template;
Simon Horman59e03502011-02-04 18:33:01 +09003660 ipvs->sysctl_sync_threshold[0] = DEFAULT_SYNC_THRESHOLD;
3661 ipvs->sysctl_sync_threshold[1] = DEFAULT_SYNC_PERIOD;
Hans Schillstroma0840e22011-01-03 14:44:58 +01003662 tbl[idx].data = &ipvs->sysctl_sync_threshold;
3663 tbl[idx++].maxlen = sizeof(ipvs->sysctl_sync_threshold);
3664 tbl[idx++].data = &ipvs->sysctl_nat_icmp_send;
3665
3666
3667 ipvs->sysctl_hdr = register_net_sysctl_table(net, net_vs_ctl_path,
Hans Schillstrom07924702011-01-24 15:14:41 +01003668 tbl);
Simon Horman04439292011-02-01 18:29:04 +01003669 if (ipvs->sysctl_hdr == NULL) {
3670 if (!net_eq(net, &init_net))
3671 kfree(tbl);
Simon Horman14e40542011-02-04 18:33:02 +09003672 return -ENOMEM;
Simon Horman04439292011-02-01 18:29:04 +01003673 }
Julian Anastasov6ef757f2011-03-14 01:44:28 +02003674 ip_vs_start_estimator(net, &ipvs->tot_stats);
Hans Schillstroma0840e22011-01-03 14:44:58 +01003675 ipvs->sysctl_tbl = tbl;
Hans Schillstromf6340ee2011-01-03 14:44:59 +01003676 /* Schedule defense work */
3677 INIT_DELAYED_WORK(&ipvs->defense_work, defense_work_handler);
3678 schedule_delayed_work(&ipvs->defense_work, DEFENSE_TIMER_PERIOD);
Simon Horman14e40542011-02-04 18:33:02 +09003679
3680 return 0;
3681}
3682
Hans Schillstrom503cf152011-05-01 18:50:16 +02003683void __net_init ip_vs_control_net_cleanup_sysctl(struct net *net)
Simon Horman14e40542011-02-04 18:33:02 +09003684{
3685 struct netns_ipvs *ipvs = net_ipvs(net);
3686
3687 cancel_delayed_work_sync(&ipvs->defense_work);
3688 cancel_work_sync(&ipvs->defense_work.work);
3689 unregister_net_sysctl_table(ipvs->sysctl_hdr);
3690}
3691
3692#else
3693
Hans Schillstrom503cf152011-05-01 18:50:16 +02003694int __net_init ip_vs_control_net_init_sysctl(struct net *net) { return 0; }
3695void __net_init ip_vs_control_net_cleanup_sysctl(struct net *net) { }
Simon Horman14e40542011-02-04 18:33:02 +09003696
3697#endif
3698
Hans Schillstrom7a4f0762011-05-03 22:09:31 +02003699static struct notifier_block ip_vs_dst_notifier = {
3700 .notifier_call = ip_vs_dst_event,
3701};
3702
Hans Schillstrom503cf152011-05-01 18:50:16 +02003703int __net_init ip_vs_control_net_init(struct net *net)
Simon Horman14e40542011-02-04 18:33:02 +09003704{
3705 int idx;
3706 struct netns_ipvs *ipvs = net_ipvs(net);
3707
Thomas Gleixner3458e212011-10-05 03:24:43 +00003708 rwlock_init(&ipvs->rs_lock);
Simon Horman14e40542011-02-04 18:33:02 +09003709
3710 /* Initialize rs_table */
3711 for (idx = 0; idx < IP_VS_RTAB_SIZE; idx++)
3712 INIT_LIST_HEAD(&ipvs->rs_table[idx]);
3713
3714 INIT_LIST_HEAD(&ipvs->dest_trash);
3715 atomic_set(&ipvs->ftpsvc_counter, 0);
3716 atomic_set(&ipvs->nullsvc_counter, 0);
3717
3718 /* procfs stats */
3719 ipvs->tot_stats.cpustats = alloc_percpu(struct ip_vs_cpu_stats);
Joe Perches0a9ee812011-08-29 14:17:25 -07003720 if (!ipvs->tot_stats.cpustats)
Simon Horman14e40542011-02-04 18:33:02 +09003721 return -ENOMEM;
Joe Perches0a9ee812011-08-29 14:17:25 -07003722
Simon Horman14e40542011-02-04 18:33:02 +09003723 spin_lock_init(&ipvs->tot_stats.lock);
3724
3725 proc_net_fops_create(net, "ip_vs", 0, &ip_vs_info_fops);
3726 proc_net_fops_create(net, "ip_vs_stats", 0, &ip_vs_stats_fops);
3727 proc_net_fops_create(net, "ip_vs_stats_percpu", 0,
3728 &ip_vs_stats_percpu_fops);
3729
Hans Schillstrom503cf152011-05-01 18:50:16 +02003730 if (ip_vs_control_net_init_sysctl(net))
Simon Horman14e40542011-02-04 18:33:02 +09003731 goto err;
3732
Hans Schillstrom61b1ab42011-01-03 14:44:42 +01003733 return 0;
3734
Simon Horman14e40542011-02-04 18:33:02 +09003735err:
Julian Anastasov2a0751a2011-03-04 12:20:35 +02003736 free_percpu(ipvs->tot_stats.cpustats);
Hans Schillstrom61b1ab42011-01-03 14:44:42 +01003737 return -ENOMEM;
3738}
3739
Hans Schillstrom503cf152011-05-01 18:50:16 +02003740void __net_exit ip_vs_control_net_cleanup(struct net *net)
Hans Schillstrom61b1ab42011-01-03 14:44:42 +01003741{
Hans Schillstromb17fc992011-01-03 14:44:56 +01003742 struct netns_ipvs *ipvs = net_ipvs(net);
3743
Hans Schillstromf2431e62011-01-03 14:45:00 +01003744 ip_vs_trash_cleanup(net);
Julian Anastasov6ef757f2011-03-14 01:44:28 +02003745 ip_vs_stop_estimator(net, &ipvs->tot_stats);
Hans Schillstrom503cf152011-05-01 18:50:16 +02003746 ip_vs_control_net_cleanup_sysctl(net);
Hans Schillstromb17fc992011-01-03 14:44:56 +01003747 proc_net_remove(net, "ip_vs_stats_percpu");
Hans Schillstrom61b1ab42011-01-03 14:44:42 +01003748 proc_net_remove(net, "ip_vs_stats");
3749 proc_net_remove(net, "ip_vs");
Julian Anastasov2a0751a2011-03-04 12:20:35 +02003750 free_percpu(ipvs->tot_stats.cpustats);
Hans Schillstrom61b1ab42011-01-03 14:44:42 +01003751}
3752
Sven Wegener048cf482008-08-10 18:24:35 +00003753int __init ip_vs_control_init(void)
Linus Torvalds1da177e2005-04-16 15:20:36 -07003754{
Linus Torvalds1da177e2005-04-16 15:20:36 -07003755 int idx;
Hans Schillstromfc723252011-01-03 14:44:43 +01003756 int ret;
Linus Torvalds1da177e2005-04-16 15:20:36 -07003757
3758 EnterFunction(2);
3759
Hans Schillstromfc723252011-01-03 14:44:43 +01003760 /* Initialize svc_table, ip_vs_svc_fwm_table, rs_table */
Eduardo Blancod86bef72010-10-19 10:26:47 +01003761 for(idx = 0; idx < IP_VS_SVC_TAB_SIZE; idx++) {
3762 INIT_LIST_HEAD(&ip_vs_svc_table[idx]);
3763 INIT_LIST_HEAD(&ip_vs_svc_fwm_table[idx]);
3764 }
Hans Schillstromfc723252011-01-03 14:44:43 +01003765
Hans Schillstromfc723252011-01-03 14:44:43 +01003766 smp_wmb(); /* Do we really need it now ? */
Eduardo Blancod86bef72010-10-19 10:26:47 +01003767
Linus Torvalds1da177e2005-04-16 15:20:36 -07003768 ret = nf_register_sockopt(&ip_vs_sockopts);
3769 if (ret) {
Hannes Eder1e3e2382009-08-02 11:05:41 +00003770 pr_err("cannot register sockopt.\n");
Hans Schillstrom7a4f0762011-05-03 22:09:31 +02003771 goto err_sock;
Linus Torvalds1da177e2005-04-16 15:20:36 -07003772 }
3773
Julius Volz9a812192008-08-14 14:08:44 +02003774 ret = ip_vs_genl_register();
3775 if (ret) {
Hannes Eder1e3e2382009-08-02 11:05:41 +00003776 pr_err("cannot register Generic Netlink interface.\n");
Hans Schillstrom7a4f0762011-05-03 22:09:31 +02003777 goto err_genl;
Julius Volz9a812192008-08-14 14:08:44 +02003778 }
3779
Hans Schillstrom7a4f0762011-05-03 22:09:31 +02003780 ret = register_netdevice_notifier(&ip_vs_dst_notifier);
3781 if (ret < 0)
3782 goto err_notf;
3783
Linus Torvalds1da177e2005-04-16 15:20:36 -07003784 LeaveFunction(2);
3785 return 0;
Hans Schillstromfc723252011-01-03 14:44:43 +01003786
Hans Schillstrom7a4f0762011-05-03 22:09:31 +02003787err_notf:
3788 ip_vs_genl_unregister();
3789err_genl:
3790 nf_unregister_sockopt(&ip_vs_sockopts);
3791err_sock:
Hans Schillstromfc723252011-01-03 14:44:43 +01003792 return ret;
Linus Torvalds1da177e2005-04-16 15:20:36 -07003793}
3794
3795
3796void ip_vs_control_cleanup(void)
3797{
3798 EnterFunction(2);
Simon Horman7676e342011-05-19 21:32:57 +09003799 unregister_netdevice_notifier(&ip_vs_dst_notifier);
Julius Volz9a812192008-08-14 14:08:44 +02003800 ip_vs_genl_unregister();
Linus Torvalds1da177e2005-04-16 15:20:36 -07003801 nf_unregister_sockopt(&ip_vs_sockopts);
3802 LeaveFunction(2);
3803}