projects / deliverable / linux.git / blame
? search:
summary | shortlog | log | commit | commitdiff | tree
blob | blame (incremental) | history | HEAD
ipv4: fix checkpatch errors
[deliverable/linux.git] / net / netfilter / ipvs / ip_vs_ctl.c
CommitLineData
1da177e4
LT		 1/*
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 *
1da177e4
LT		 8 * 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
9aada7ac
HE		 21#define KMSG_COMPONENT "IPVS"
22#define pr_fmt(fmt) KMSG_COMPONENT ": " fmt
23
1da177e4
LT		 24#include <linux/module.h>
25#include <linux/init.h>
26#include <linux/types.h>
4fc268d2 27#include <linux/capability.h>
1da177e4
LT		 28#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>
1da177e4 33#include <linux/seq_file.h>
5a0e3ad6 34#include <linux/slab.h>
1da177e4
LT		 35
36#include <linux/netfilter.h>
37#include <linux/netfilter_ipv4.h>
14cc3e2b 38#include <linux/mutex.h>
1da177e4 39
457c4cbc 40#include <net/net_namespace.h>
9330419d 41#include <linux/nsproxy.h>
1da177e4 42#include <net/ip.h>
09571c7a
VB		 43#ifdef CONFIG_IP_VS_IPV6
44#include <net/ipv6.h>
45#include <net/ip6_route.h>
46#endif
14c85021 47#include <net/route.h>
1da177e4 48#include <net/sock.h>
9a812198 49#include <net/genetlink.h>
1da177e4
LT		 50
51#include <asm/uaccess.h>
52
53#include <net/ip_vs.h>
54
55/* semaphore for IPVS sockopts. And, [gs]etsockopt may sleep. */
14cc3e2b 56static DEFINE_MUTEX(__ip_vs_mutex);
1da177e4
LT		 57
58/* lock for service table */
59static DEFINE_RWLOCK(__ip_vs_svc_lock);
60
1da177e4 61/* sysctl variables */
1da177e4
LT		 62
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
7a4f0761
HS		 72
73/* Protos */
74static void __ip_vs_del_service(struct ip_vs_service *svc);
75
76
09571c7a
VB		 77#ifdef CONFIG_IP_VS_IPV6
78/* Taken from rt6_fill_node() in net/ipv6/route.c, is there a better way? */
4a98480b
HS		 79static int __ip_vs_addr_is_local_v6(struct net *net,
80 const struct in6_addr *addr)
09571c7a
VB		 81{
82 struct rt6_info *rt;
4c9483b2
DM		 83 struct flowi6 fl6 = {
84 .daddr = *addr,
09571c7a
VB		 85 };
86
4c9483b2 87 rt = (struct rt6_info *)ip6_route_output(net, NULL, &fl6);
d1918542 88 if (rt && rt->dst.dev && (rt->dst.dev->flags & IFF_LOOPBACK))
4c9483b2 89 return 1;
09571c7a
VB		 90
91 return 0;
92}
93#endif
14e40546
SH		 94
95#ifdef CONFIG_SYSCTL
1da177e4 96/*
af9debd4
JA		 97 * update_defense_level is called from keventd and from sysctl,
98 * so it needs to protect itself from softirqs
1da177e4 99 */
9330419d 100static void update_defense_level(struct netns_ipvs *ipvs)
1da177e4
LT		 101{
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
a0840e2e 116 nomem = (availmem < ipvs->sysctl_amemthresh);
1da177e4 117
af9debd4
JA		 118 local_bh_disable();
119
1da177e4 120 /* drop_entry */
a0840e2e
HS		 121 spin_lock(&ipvs->dropentry_lock);
122 switch (ipvs->sysctl_drop_entry) {
1da177e4 123 case 0:
a0840e2e 124 atomic_set(&ipvs->dropentry, 0);
1da177e4
LT		 125 break;
126 case 1:
127 if (nomem) {
a0840e2e
HS		 128 atomic_set(&ipvs->dropentry, 1);
129 ipvs->sysctl_drop_entry = 2;
1da177e4 130 } else {
a0840e2e 131 atomic_set(&ipvs->dropentry, 0);
1da177e4
LT		 132 }
133 break;
134 case 2:
135 if (nomem) {
a0840e2e 136 atomic_set(&ipvs->dropentry, 1);
1da177e4 137 } else {
a0840e2e
HS		 138 atomic_set(&ipvs->dropentry, 0);
139 ipvs->sysctl_drop_entry = 1;
1da177e4
LT		 140 };
141 break;
142 case 3:
a0840e2e 143 atomic_set(&ipvs->dropentry, 1);
1da177e4
LT		 144 break;
145 }
a0840e2e 146 spin_unlock(&ipvs->dropentry_lock);
1da177e4
LT		 147
148 /* drop_packet */
a0840e2e
HS		 149 spin_lock(&ipvs->droppacket_lock);
150 switch (ipvs->sysctl_drop_packet) {
1da177e4 151 case 0:
a0840e2e 152 ipvs->drop_rate = 0;
1da177e4
LT		 153 break;
154 case 1:
155 if (nomem) {
a0840e2e
HS		 156 ipvs->drop_rate = ipvs->drop_counter
157 = ipvs->sysctl_amemthresh /
158 (ipvs->sysctl_amemthresh-availmem);
159 ipvs->sysctl_drop_packet = 2;
1da177e4 160 } else {
a0840e2e 161 ipvs->drop_rate = 0;
1da177e4
LT		 162 }
163 break;
164 case 2:
165 if (nomem) {
a0840e2e
HS		 166 ipvs->drop_rate = ipvs->drop_counter
167 = ipvs->sysctl_amemthresh /
168 (ipvs->sysctl_amemthresh-availmem);
1da177e4 169 } else {
a0840e2e
HS		 170 ipvs->drop_rate = 0;
171 ipvs->sysctl_drop_packet = 1;
1da177e4
LT		 172 }
173 break;
174 case 3:
a0840e2e 175 ipvs->drop_rate = ipvs->sysctl_am_droprate;
1da177e4
LT		 176 break;
177 }
a0840e2e 178 spin_unlock(&ipvs->droppacket_lock);
1da177e4
LT		 179
180 /* secure_tcp */
a0840e2e
HS		 181 spin_lock(&ipvs->securetcp_lock);
182 switch (ipvs->sysctl_secure_tcp) {
1da177e4
LT		 183 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;
a0840e2e 191 ipvs->sysctl_secure_tcp = 2;
1da177e4
LT		 192 } 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;
a0840e2e 204 ipvs->sysctl_secure_tcp = 1;
1da177e4
LT		 205 }
206 break;
207 case 3:
208 if (old_secure_tcp < 2)
209 to_change = 1;
210 break;
211 }
a0840e2e 212 old_secure_tcp = ipvs->sysctl_secure_tcp;
1da177e4 213 if (to_change >= 0)
9330419d 214 ip_vs_protocol_timeout_change(ipvs,
a0840e2e
HS		 215 ipvs->sysctl_secure_tcp > 1);
216 spin_unlock(&ipvs->securetcp_lock);
af9debd4
JA		 217
218 local_bh_enable();
1da177e4
LT		 219}
220
221
222/*
223 * Timer for checking the defense
224 */
225#define DEFENSE_TIMER_PERIOD 1*HZ
1da177e4 226
c4028958 227static void defense_work_handler(struct work_struct *work)
1da177e4 228{
f6340ee0
HS		 229 struct netns_ipvs *ipvs =
230 container_of(work, struct netns_ipvs, defense_work.work);
9330419d
HS		 231
232 update_defense_level(ipvs);
a0840e2e 233 if (atomic_read(&ipvs->dropentry))
f6340ee0
HS		 234 ip_vs_random_dropentry(ipvs->net);
235 schedule_delayed_work(&ipvs->defense_work, DEFENSE_TIMER_PERIOD);
1da177e4 236}
14e40546 237#endif
1da177e4
LT		 238
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
1da177e4
LT		 264
265/*
266 * Returns hash value for virtual service
267 */
fc723250
HS		 268static inline unsigned
269ip_vs_svc_hashkey(struct net *net, int af, unsigned proto,
270 const union nf_inet_addr *addr, __be16 port)
1da177e4
LT		 271{
272 register unsigned porth = ntohs(port);
b18610de 273 __be32 addr_fold = addr->ip;
1da177e4 274
b18610de
JV		 275#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
fc723250 280 addr_fold ^= ((size_t)net>>8);
b18610de
JV		 281
282 return (proto^ntohl(addr_fold)^(porth>>IP_VS_SVC_TAB_BITS)^porth)
1da177e4
LT		 283 & IP_VS_SVC_TAB_MASK;
284}
285
286/*
287 * Returns hash value of fwmark for virtual service lookup
288 */
fc723250 289static inline unsigned ip_vs_svc_fwm_hashkey(struct net *net, __u32 fwmark)
1da177e4 290{
fc723250 291 return (((size_t)net>>8) ^ fwmark) & IP_VS_SVC_TAB_MASK;
1da177e4
LT		 292}
293
294/*
fc723250 295 * Hashes a service in the ip_vs_svc_table by <netns,proto,addr,port>
1da177e4
LT		 296 * 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) {
1e3e238e
HE		 304 pr_err("%s(): request for already hashed, called from %pF\n",
305 __func__, __builtin_return_address(0));
1da177e4
LT		 306 return 0;
307 }
308
309 if (svc->fwmark == 0) {
310 /*
fc723250 311 * Hash it by <netns,protocol,addr,port> in ip_vs_svc_table
1da177e4 312 */
fc723250
HS		 313 hash = ip_vs_svc_hashkey(svc->net, svc->af, svc->protocol,
314 &svc->addr, svc->port);
1da177e4
LT		 315 list_add(&svc->s_list, &ip_vs_svc_table[hash]);
316 } else {
317 /*
fc723250 318 * Hash it by fwmark in svc_fwm_table
1da177e4 319 */
fc723250 320 hash = ip_vs_svc_fwm_hashkey(svc->net, svc->fwmark);
1da177e4
LT		 321 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/*
fc723250 332 * Unhashes a service from svc_table / svc_fwm_table.
1da177e4
LT		 333 * 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)) {
1e3e238e
HE		 338 pr_err("%s(): request for unhash flagged, called from %pF\n",
339 __func__, __builtin_return_address(0));
1da177e4
LT		 340 return 0;
341 }
342
343 if (svc->fwmark == 0) {
fc723250 344 /* Remove it from the svc_table table */
1da177e4
LT		 345 list_del(&svc->s_list);
346 } else {
fc723250 347 /* Remove it from the svc_fwm_table table */
1da177e4
LT		 348 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/*
fc723250 358 * Get service by {netns, proto,addr,port} in the service table.
1da177e4 359 */
b18610de 360static inline struct ip_vs_service *
fc723250
HS		 361__ip_vs_service_find(struct net *net, int af, __u16 protocol,
362 const union nf_inet_addr *vaddr, __be16 vport)
1da177e4
LT		 363{
364 unsigned hash;
365 struct ip_vs_service *svc;
366
367 /* Check for "full" addressed entries */
fc723250 368 hash = ip_vs_svc_hashkey(net, af, protocol, vaddr, vport);
1da177e4
LT		 369
370 list_for_each_entry(svc, &ip_vs_svc_table[hash], s_list){
b18610de
JV		 371 if ((svc->af == af)
372 && ip_vs_addr_equal(af, &svc->addr, vaddr)
1da177e4 373 && (svc->port == vport)
fc723250
HS		 374 && (svc->protocol == protocol)
375 && net_eq(svc->net, net)) {
1da177e4 376 /* HIT */
1da177e4
LT		 377 return svc;
378 }
379 }
380
381 return NULL;
382}
383
384
385/*
386 * Get service by {fwmark} in the service table.
387 */
b18610de 388static inline struct ip_vs_service *
fc723250 389__ip_vs_svc_fwm_find(struct net *net, int af, __u32 fwmark)
1da177e4
LT		 390{
391 unsigned hash;
392 struct ip_vs_service *svc;
393
394 /* Check for fwmark addressed entries */
fc723250 395 hash = ip_vs_svc_fwm_hashkey(net, fwmark);
1da177e4
LT		 396
397 list_for_each_entry(svc, &ip_vs_svc_fwm_table[hash], f_list) {
fc723250
HS		 398 if (svc->fwmark == fwmark && svc->af == af
399 && net_eq(svc->net, net)) {
1da177e4 400 /* HIT */
1da177e4
LT		 401 return svc;
402 }
403 }
404
405 return NULL;
406}
407
408struct ip_vs_service *
fc723250 409ip_vs_service_get(struct net *net, int af, __u32 fwmark, __u16 protocol,
3c2e0505 410 const union nf_inet_addr *vaddr, __be16 vport)
1da177e4
LT		 411{
412 struct ip_vs_service *svc;
763f8d0e 413 struct netns_ipvs *ipvs = net_ipvs(net);
3c2e0505 414
1da177e4
LT		 415 read_lock(&__ip_vs_svc_lock);
416
417 /*
418 * Check the table hashed by fwmark first
419 */
097fc76a
JA		 420 if (fwmark) {
421 svc = __ip_vs_svc_fwm_find(net, af, fwmark);
422 if (svc)
423 goto out;
424 }
1da177e4
LT		 425
426 /*
427 * Check the table hashed by <protocol,addr,port>
428 * for "full" addressed entries
429 */
fc723250 430 svc = __ip_vs_service_find(net, af, protocol, vaddr, vport);
1da177e4
LT		 431
432 if (svc == NULL
433 && protocol == IPPROTO_TCP
763f8d0e 434 && atomic_read(&ipvs->ftpsvc_counter)
1da177e4
LT		 435 && (vport == FTPDATA || ntohs(vport) >= PROT_SOCK)) {
436 /*
437 * Check if ftp service entry exists, the packet
438 * might belong to FTP data connections.
439 */
fc723250 440 svc = __ip_vs_service_find(net, af, protocol, vaddr, FTPPORT);
1da177e4
LT		 441 }
442
443 if (svc == NULL
763f8d0e 444 && atomic_read(&ipvs->nullsvc_counter)) {
1da177e4
LT		 445 /*
446 * Check if the catch-all port (port zero) exists
447 */
fc723250 448 svc = __ip_vs_service_find(net, af, protocol, vaddr, 0);
1da177e4
LT		 449 }
450
451 out:
26c15cfd
JA		 452 if (svc)
453 atomic_inc(&svc->usecnt);
1da177e4
LT		 454 read_unlock(&__ip_vs_svc_lock);
455
3c2e0505
JV		 456 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");
1da177e4
LT		 460
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
26c15cfd 472static void
1da177e4
LT		 473__ip_vs_unbind_svc(struct ip_vs_dest *dest)
474{
475 struct ip_vs_service *svc = dest->svc;
476
477 dest->svc = NULL;
26c15cfd
JA		 478 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));
b17fc996 483 free_percpu(svc->stats.cpustats);
1da177e4 484 kfree(svc);
26c15cfd 485 }
1da177e4
LT		 486}
487
488
489/*
490 * Returns hash value for real service
491 */
7937df15
JV		 492static inline unsigned ip_vs_rs_hashkey(int af,
493 const union nf_inet_addr *addr,
494 __be16 port)
1da177e4
LT		 495{
496 register unsigned porth = ntohs(port);
7937df15
JV		 497 __be32 addr_fold = addr->ip;
498
499#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
1da177e4 504
7937df15 505 return (ntohl(addr_fold)^(porth>>IP_VS_RTAB_BITS)^porth)
1da177e4
LT		 506 & IP_VS_RTAB_MASK;
507}
508
509/*
fc723250 510 * Hashes ip_vs_dest in rs_table by <proto,addr,port>.
1da177e4
LT		 511 * should be called with locked tables.
512 */
fc723250 513static int ip_vs_rs_hash(struct netns_ipvs *ipvs, struct ip_vs_dest *dest)
1da177e4
LT		 514{
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 */
7937df15
JV		 525 hash = ip_vs_rs_hashkey(dest->af, &dest->addr, dest->port);
526
fc723250 527 list_add(&dest->d_list, &ipvs->rs_table[hash]);
1da177e4
LT		 528
529 return 1;
530}
531
532/*
fc723250 533 * UNhashes ip_vs_dest from rs_table.
1da177e4
LT		 534 * should be called with locked tables.
535 */
536static int ip_vs_rs_unhash(struct ip_vs_dest *dest)
537{
538 /*
fc723250 539 * Remove it from the rs_table table.
1da177e4
LT		 540 */
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 *
fc723250 553ip_vs_lookup_real_service(struct net *net, int af, __u16 protocol,
7937df15
JV		 554 const union nf_inet_addr *daddr,
555 __be16 dport)
1da177e4 556{
fc723250 557 struct netns_ipvs *ipvs = net_ipvs(net);
1da177e4
LT		 558 unsigned hash;
559 struct ip_vs_dest *dest;
560
561 /*
562 * Check for "full" addressed entries
563 * Return the first found entry
564 */
7937df15 565 hash = ip_vs_rs_hashkey(af, daddr, dport);
1da177e4 566
a0840e2e 567 read_lock(&ipvs->rs_lock);
fc723250 568 list_for_each_entry(dest, &ipvs->rs_table[hash], d_list) {
7937df15
JV		 569 if ((dest->af == af)
570 && ip_vs_addr_equal(af, &dest->addr, daddr)
1da177e4
LT		 571 && (dest->port == dport)
572 && ((dest->protocol == protocol) ||
573 dest->vfwmark)) {
574 /* HIT */
a0840e2e 575 read_unlock(&ipvs->rs_lock);
1da177e4
LT		 576 return dest;
577 }
578 }
a0840e2e 579 read_unlock(&ipvs->rs_lock);
1da177e4
LT		 580
581 return NULL;
582}
583
584/*
585 * Lookup destination by {addr,port} in the given service
586 */
587static struct ip_vs_dest *
7937df15
JV		 588ip_vs_lookup_dest(struct ip_vs_service *svc, const union nf_inet_addr *daddr,
589 __be16 dport)
1da177e4
LT		 590{
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) {
7937df15
JV		 597 if ((dest->af == svc->af)
598 && ip_vs_addr_equal(svc->af, &dest->addr, daddr)
599 && (dest->port == dport)) {
1da177e4
LT		 600 /* HIT */
601 return dest;
602 }
603 }
604
605 return NULL;
606}
607
1e356f9c
RB		 608/*
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 */
fc723250
HS		 618struct ip_vs_dest *ip_vs_find_dest(struct net *net, int af,
619 const union nf_inet_addr *daddr,
7937df15
JV		 620 __be16 dport,
621 const union nf_inet_addr *vaddr,
52793dbe
JA		 622 __be16 vport, __u16 protocol, __u32 fwmark,
623 __u32 flags)
1e356f9c
RB		 624{
625 struct ip_vs_dest *dest;
626 struct ip_vs_service *svc;
52793dbe 627 __be16 port = dport;
1e356f9c 628
fc723250 629 svc = ip_vs_service_get(net, af, fwmark, protocol, vaddr, vport);
1e356f9c
RB		 630 if (!svc)
631 return NULL;
52793dbe
JA		 632 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);
1e356f9c
RB		 637 if (dest)
638 atomic_inc(&dest->refcnt);
639 ip_vs_service_put(svc);
640 return dest;
641}
1da177e4
LT		 642
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 *
7937df15
JV		 654ip_vs_trash_get_dest(struct ip_vs_service *svc, const union nf_inet_addr *daddr,
655 __be16 dport)
1da177e4
LT		 656{
657 struct ip_vs_dest *dest, *nxt;
f2431e6e 658 struct netns_ipvs *ipvs = net_ipvs(svc->net);
1da177e4
LT		 659
660 /*
661 * Find the destination in trash
662 */
f2431e6e 663 list_for_each_entry_safe(dest, nxt, &ipvs->dest_trash, n_list) {
7937df15
JV		 664 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) &&
1da177e4
LT		 672 dest->port == dport &&
673 dest->vfwmark == svc->fwmark &&
674 dest->protocol == svc->protocol &&
675 (svc->fwmark ||
7937df15 676 (ip_vs_addr_equal(svc->af, &dest->vaddr, &svc->addr) &&
1da177e4
LT		 677 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) {
7937df15
JV		 686 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));
1da177e4
LT		 691 list_del(&dest->n_list);
692 ip_vs_dst_reset(dest);
693 __ip_vs_unbind_svc(dest);
b17fc996 694 free_percpu(dest->stats.cpustats);
1da177e4
LT		 695 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 */
f2431e6e 712static void ip_vs_trash_cleanup(struct net *net)
1da177e4
LT		 713{
714 struct ip_vs_dest *dest, *nxt;
f2431e6e 715 struct netns_ipvs *ipvs = net_ipvs(net);
1da177e4 716
f2431e6e 717 list_for_each_entry_safe(dest, nxt, &ipvs->dest_trash, n_list) {
1da177e4
LT		 718 list_del(&dest->n_list);
719 ip_vs_dst_reset(dest);
720 __ip_vs_unbind_svc(dest);
b17fc996 721 free_percpu(dest->stats.cpustats);
1da177e4
LT		 722 kfree(dest);
723 }
724}
725
55a3d4e1
JA		 726static 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
55a3d4e1
JA		 730
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
ea9f22cc 739 ip_vs_read_estimator(dst, src);
55a3d4e1
JA		 740
741 spin_unlock_bh(&src->lock);
742}
1da177e4
LT		 743
744static void
745ip_vs_zero_stats(struct ip_vs_stats *stats)
746{
747 spin_lock_bh(&stats->lock);
e93615d0 748
55a3d4e1
JA		 749 /* get current counters as zero point, rates are zeroed */
750
751#define IP_VS_ZERO_STATS_COUNTER(c) stats->ustats0.c = stats->ustats.c
55a3d4e1
JA		 752
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
1da177e4 759 ip_vs_zero_estimator(stats);
e93615d0 760
3a14a313 761 spin_unlock_bh(&stats->lock);
1da177e4
LT		 762}
763
764/*
765 * Update a destination in the given service
766 */
767static void
26c15cfd
JA		 768__ip_vs_update_dest(struct ip_vs_service *svc, struct ip_vs_dest *dest,
769 struct ip_vs_dest_user_kern *udest, int add)
1da177e4 770{
fc723250 771 struct netns_ipvs *ipvs = net_ipvs(svc->net);
1da177e4
LT		 772 int conn_flags;
773
774 /* set the weight and the flags */
775 atomic_set(&dest->weight, udest->weight);
3575792e
JA		 776 conn_flags = udest->conn_flags & IP_VS_CONN_F_DEST_MASK;
777 conn_flags |= IP_VS_CONN_F_INACTIVE;
1da177e4 778
1da177e4 779 /* set the IP_VS_CONN_F_NOOUTPUT flag if not masquerading/NAT */
3575792e 780 if ((conn_flags & IP_VS_CONN_F_FWD_MASK) != IP_VS_CONN_F_MASQ) {
1da177e4
LT		 781 conn_flags |= IP_VS_CONN_F_NOOUTPUT;
782 } else {
783 /*
fc723250 784 * Put the real service in rs_table if not present.
1da177e4
LT		 785 * For now only for NAT!
786 */
a0840e2e 787 write_lock_bh(&ipvs->rs_lock);
fc723250 788 ip_vs_rs_hash(ipvs, dest);
a0840e2e 789 write_unlock_bh(&ipvs->rs_lock);
1da177e4
LT		 790 }
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;
26c15cfd 811
ff75f40f 812 spin_lock_bh(&dest->dst_lock);
fc604767 813 ip_vs_dst_reset(dest);
ff75f40f 814 spin_unlock_bh(&dest->dst_lock);
fc604767 815
26c15cfd 816 if (add)
6ef757f9 817 ip_vs_start_estimator(svc->net, &dest->stats);
26c15cfd
JA		 818
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);
1da177e4
LT		 834}
835
836
837/*
838 * Create a destination for the given service
839 */
840static int
c860c6b1 841ip_vs_new_dest(struct ip_vs_service *svc, struct ip_vs_dest_user_kern *udest,
1da177e4
LT		 842 struct ip_vs_dest **dest_p)
843{
844 struct ip_vs_dest *dest;
845 unsigned atype;
846
847 EnterFunction(2);
848
09571c7a
VB		 849#ifdef CONFIG_IP_VS_IPV6
850 if (svc->af == AF_INET6) {
851 atype = ipv6_addr_type(&udest->addr.in6);
3bfb92f4
SW		 852 if ((!(atype & IPV6_ADDR_UNICAST) ||
853 atype & IPV6_ADDR_LINKLOCAL) &&
4a98480b 854 !__ip_vs_addr_is_local_v6(svc->net, &udest->addr.in6))
09571c7a
VB		 855 return -EINVAL;
856 } else
857#endif
858 {
4a98480b 859 atype = inet_addr_type(svc->net, udest->addr.ip);
09571c7a
VB		 860 if (atype != RTN_LOCAL && atype != RTN_UNICAST)
861 return -EINVAL;
862 }
1da177e4 863
dee06e47 864 dest = kzalloc(sizeof(struct ip_vs_dest), GFP_KERNEL);
0a9ee813 865 if (dest == NULL)
1da177e4 866 return -ENOMEM;
0a9ee813 867
b17fc996 868 dest->stats.cpustats = alloc_percpu(struct ip_vs_cpu_stats);
0a9ee813 869 if (!dest->stats.cpustats)
b17fc996 870 goto err_alloc;
1da177e4 871
c860c6b1 872 dest->af = svc->af;
1da177e4 873 dest->protocol = svc->protocol;
c860c6b1 874 dest->vaddr = svc->addr;
1da177e4
LT		 875 dest->vport = svc->port;
876 dest->vfwmark = svc->fwmark;
c860c6b1 877 ip_vs_addr_copy(svc->af, &dest->addr, &udest->addr);
1da177e4
LT		 878 dest->port = udest->port;
879
880 atomic_set(&dest->activeconns, 0);
881 atomic_set(&dest->inactconns, 0);
882 atomic_set(&dest->persistconns, 0);
26c15cfd 883 atomic_set(&dest->refcnt, 1);
1da177e4
LT		 884
885 INIT_LIST_HEAD(&dest->d_list);
886 spin_lock_init(&dest->dst_lock);
887 spin_lock_init(&dest->stats.lock);
26c15cfd 888 __ip_vs_update_dest(svc, dest, udest, 1);
1da177e4
LT		 889
890 *dest_p = dest;
891
892 LeaveFunction(2);
893 return 0;
b17fc996
HS		 894
895err_alloc:
896 kfree(dest);
897 return -ENOMEM;
1da177e4
LT		 898}
899
900
901/*
902 * Add a destination into an existing service
903 */
904static int
c860c6b1 905ip_vs_add_dest(struct ip_vs_service *svc, struct ip_vs_dest_user_kern *udest)
1da177e4
LT		 906{
907 struct ip_vs_dest *dest;
c860c6b1 908 union nf_inet_addr daddr;
014d730d 909 __be16 dport = udest->port;
1da177e4
LT		 910 int ret;
911
912 EnterFunction(2);
913
914 if (udest->weight < 0) {
1e3e238e 915 pr_err("%s(): server weight less than zero\n", __func__);
1da177e4
LT		 916 return -ERANGE;
917 }
918
919 if (udest->l_threshold > udest->u_threshold) {
1e3e238e
HE		 920 pr_err("%s(): lower threshold is higher than upper threshold\n",
921 __func__);
1da177e4
LT		 922 return -ERANGE;
923 }
924
c860c6b1
JV		 925 ip_vs_addr_copy(svc->af, &daddr, &udest->addr);
926
1da177e4
LT		 927 /*
928 * Check if the dest already exists in the list
929 */
7937df15
JV		 930 dest = ip_vs_lookup_dest(svc, &daddr, dport);
931
1da177e4 932 if (dest != NULL) {
1e3e238e 933 IP_VS_DBG(1, "%s(): dest already exists\n", __func__);
1da177e4
LT		 934 return -EEXIST;
935 }
936
937 /*
938 * Check if the dest already exists in the trash and
939 * is from the same service
940 */
7937df15
JV		 941 dest = ip_vs_trash_get_dest(svc, &daddr, dport);
942
1da177e4 943 if (dest != NULL) {
cfc78c5a
JV		 944 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
1da177e4
LT		 952 /*
953 * Get the destination from the trash
954 */
955 list_del(&dest->n_list);
956
26c15cfd
JA		 957 __ip_vs_update_dest(svc, dest, udest, 1);
958 ret = 0;
959 } else {
1da177e4 960 /*
26c15cfd 961 * Allocate and initialize the dest structure
1da177e4 962 */
26c15cfd 963 ret = ip_vs_new_dest(svc, udest, &dest);
1da177e4 964 }
1da177e4
LT		 965 LeaveFunction(2);
966
26c15cfd 967 return ret;
1da177e4
LT		 968}
969
970
971/*
972 * Edit a destination in the given service
973 */
974static int
c860c6b1 975ip_vs_edit_dest(struct ip_vs_service *svc, struct ip_vs_dest_user_kern *udest)
1da177e4
LT		 976{
977 struct ip_vs_dest *dest;
c860c6b1 978 union nf_inet_addr daddr;
014d730d 979 __be16 dport = udest->port;
1da177e4
LT		 980
981 EnterFunction(2);
982
983 if (udest->weight < 0) {
1e3e238e 984 pr_err("%s(): server weight less than zero\n", __func__);
1da177e4
LT		 985 return -ERANGE;
986 }
987
988 if (udest->l_threshold > udest->u_threshold) {
1e3e238e
HE		 989 pr_err("%s(): lower threshold is higher than upper threshold\n",
990 __func__);
1da177e4
LT		 991 return -ERANGE;
992 }
993
c860c6b1
JV		 994 ip_vs_addr_copy(svc->af, &daddr, &udest->addr);
995
1da177e4
LT		 996 /*
997 * Lookup the destination list
998 */
7937df15
JV		 999 dest = ip_vs_lookup_dest(svc, &daddr, dport);
1000
1da177e4 1001 if (dest == NULL) {
1e3e238e 1002 IP_VS_DBG(1, "%s(): dest doesn't exist\n", __func__);
1da177e4
LT		 1003 return -ENOENT;
1004 }
1005
26c15cfd 1006 __ip_vs_update_dest(svc, dest, udest, 0);
1da177e4
LT		 1007 LeaveFunction(2);
1008
1009 return 0;
1010}
1011
1012
1013/*
1014 * Delete a destination (must be already unlinked from the service)
1015 */
29c2026f 1016static void __ip_vs_del_dest(struct net *net, struct ip_vs_dest *dest)
1da177e4 1017{
a0840e2e
HS		 1018 struct netns_ipvs *ipvs = net_ipvs(net);
1019
6ef757f9 1020 ip_vs_stop_estimator(net, &dest->stats);
1da177e4
LT		 1021
1022 /*
1023 * Remove it from the d-linked list with the real services.
1024 */
a0840e2e 1025 write_lock_bh(&ipvs->rs_lock);
1da177e4 1026 ip_vs_rs_unhash(dest);
a0840e2e 1027 write_unlock_bh(&ipvs->rs_lock);
1da177e4
LT		 1028
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)) {
26c15cfd
JA		 1035 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));
1da177e4
LT		 1039 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);
b17fc996 1046 free_percpu(dest->stats.cpustats);
1da177e4
LT		 1047 kfree(dest);
1048 } else {
cfc78c5a
JV		 1049 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));
f2431e6e 1054 list_add(&dest->n_list, &ipvs->dest_trash);
1da177e4
LT		 1055 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--;
82dfb6f3
SW		 1074
1075 /*
1076 * Call the update_service function of its scheduler
1077 */
1078 if (svcupd && svc->scheduler->update_service)
1079 svc->scheduler->update_service(svc);
1da177e4
LT		 1080}
1081
1082
1083/*
1084 * Delete a destination server in the given service
1085 */
1086static int
c860c6b1 1087ip_vs_del_dest(struct ip_vs_service *svc, struct ip_vs_dest_user_kern *udest)
1da177e4
LT		 1088{
1089 struct ip_vs_dest *dest;
014d730d 1090 __be16 dport = udest->port;
1da177e4
LT		 1091
1092 EnterFunction(2);
1093
7937df15 1094 dest = ip_vs_lookup_dest(svc, &udest->addr, dport);
c860c6b1 1095
1da177e4 1096 if (dest == NULL) {
1e3e238e 1097 IP_VS_DBG(1, "%s(): destination not found!\n", __func__);
1da177e4
LT		 1098 return -ENOENT;
1099 }
1100
1101 write_lock_bh(&__ip_vs_svc_lock);
1102
1103 /*
1104 * Wait until all other svc users go away.
1105 */
26c15cfd 1106 IP_VS_WAIT_WHILE(atomic_read(&svc->usecnt) > 0);
1da177e4
LT		 1107
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 */
a0840e2e 1118 __ip_vs_del_dest(svc->net, dest);
1da177e4
LT		 1119
1120 LeaveFunction(2);
1121
1122 return 0;
1123}
1124
1125
1126/*
1127 * Add a service into the service hash table
1128 */
1129static int
fc723250 1130ip_vs_add_service(struct net *net, struct ip_vs_service_user_kern *u,
c860c6b1 1131 struct ip_vs_service **svc_p)
1da177e4
LT		 1132{
1133 int ret = 0;
1134 struct ip_vs_scheduler *sched = NULL;
0d1e71b0 1135 struct ip_vs_pe *pe = NULL;
1da177e4 1136 struct ip_vs_service *svc = NULL;
a0840e2e 1137 struct netns_ipvs *ipvs = net_ipvs(net);
1da177e4
LT		 1138
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) {
1e3e238e 1145 pr_info("Scheduler module ip_vs_%s not found\n", u->sched_name);
1da177e4 1146 ret = -ENOENT;
6e08bfb8 1147 goto out_err;
1da177e4
LT		 1148 }
1149
0d1e71b0 1150 if (u->pe_name && *u->pe_name) {
e9e5eee8 1151 pe = ip_vs_pe_getbyname(u->pe_name);
0d1e71b0
SH		 1152 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
f94fd041 1160#ifdef CONFIG_IP_VS_IPV6
48148938
JV		 1161 if (u->af == AF_INET6 && (u->netmask < 1 || u->netmask > 128)) {
1162 ret = -EINVAL;
1163 goto out_err;
f94fd041
JV		 1164 }
1165#endif
1166
dee06e47 1167 svc = kzalloc(sizeof(struct ip_vs_service), GFP_KERNEL);
1da177e4 1168 if (svc == NULL) {
1e3e238e 1169 IP_VS_DBG(1, "%s(): no memory\n", __func__);
1da177e4
LT		 1170 ret = -ENOMEM;
1171 goto out_err;
1172 }
b17fc996 1173 svc->stats.cpustats = alloc_percpu(struct ip_vs_cpu_stats);
0a9ee813 1174 if (!svc->stats.cpustats)
b17fc996 1175 goto out_err;
1da177e4
LT		 1176
1177 /* I'm the first user of the service */
26c15cfd 1178 atomic_set(&svc->usecnt, 0);
1da177e4
LT		 1179 atomic_set(&svc->refcnt, 0);
1180
c860c6b1 1181 svc->af = u->af;
1da177e4 1182 svc->protocol = u->protocol;
c860c6b1 1183 ip_vs_addr_copy(svc->af, &svc->addr, &u->addr);
1da177e4
LT		 1184 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;
fc723250 1189 svc->net = net;
1da177e4
LT		 1190
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
0d1e71b0
SH		 1201 /* Bind the ct retriever */
1202 ip_vs_bind_pe(svc, pe);
1203 pe = NULL;
1204
1da177e4
LT		 1205 /* Update the virtual service counters */
1206 if (svc->port == FTPPORT)
763f8d0e 1207 atomic_inc(&ipvs->ftpsvc_counter);
1da177e4 1208 else if (svc->port == 0)
763f8d0e 1209 atomic_inc(&ipvs->nullsvc_counter);
1da177e4 1210
6ef757f9 1211 ip_vs_start_estimator(net, &svc->stats);
f94fd041
JV		 1212
1213 /* Count only IPv4 services for old get/setsockopt interface */
1214 if (svc->af == AF_INET)
a0840e2e 1215 ipvs->num_services++;
1da177e4
LT		 1216
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;
7a4f0761
HS		 1223 /* Now there is a service - full throttle */
1224 ipvs->enable = 1;
1da177e4
LT		 1225 return 0;
1226
b17fc996 1227
6e08bfb8 1228 out_err:
1da177e4 1229 if (svc != NULL) {
2fabf35b 1230 ip_vs_unbind_scheduler(svc);
1da177e4
LT		 1231 if (svc->inc) {
1232 local_bh_disable();
1233 ip_vs_app_inc_put(svc->inc);
1234 local_bh_enable();
1235 }
b17fc996
HS		 1236 if (svc->stats.cpustats)
1237 free_percpu(svc->stats.cpustats);
1da177e4
LT		 1238 kfree(svc);
1239 }
1240 ip_vs_scheduler_put(sched);
0d1e71b0 1241 ip_vs_pe_put(pe);
1da177e4 1242
1da177e4
LT		 1243 /* 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
c860c6b1 1254ip_vs_edit_service(struct ip_vs_service *svc, struct ip_vs_service_user_kern *u)
1da177e4
LT		 1255{
1256 struct ip_vs_scheduler *sched, *old_sched;
0d1e71b0 1257 struct ip_vs_pe *pe = NULL, *old_pe = NULL;
1da177e4
LT		 1258 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) {
1e3e238e 1265 pr_info("Scheduler module ip_vs_%s not found\n", u->sched_name);
1da177e4
LT		 1266 return -ENOENT;
1267 }
1268 old_sched = sched;
1269
0d1e71b0 1270 if (u->pe_name && *u->pe_name) {
e9e5eee8 1271 pe = ip_vs_pe_getbyname(u->pe_name);
0d1e71b0
SH		 1272 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
f94fd041 1281#ifdef CONFIG_IP_VS_IPV6
48148938
JV		 1282 if (u->af == AF_INET6 && (u->netmask < 1 || u->netmask > 128)) {
1283 ret = -EINVAL;
1284 goto out;
f94fd041
JV		 1285 }
1286#endif
1287
1da177e4
LT		 1288 write_lock_bh(&__ip_vs_svc_lock);
1289
1290 /*
1291 * Wait until all other svc users go away.
1292 */
26c15cfd 1293 IP_VS_WAIT_WHILE(atomic_read(&svc->usecnt) > 0);
1da177e4
LT		 1294
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;
9e691ed6 1309 goto out_unlock;
1da177e4
LT		 1310 }
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;
9e691ed6 1328 goto out_unlock;
1da177e4
LT		 1329 }
1330 }
1331
0d1e71b0
SH		 1332 old_pe = svc->pe;
1333 if (pe != old_pe) {
1334 ip_vs_unbind_pe(svc);
1335 ip_vs_bind_pe(svc, pe);
1336 }
1337
552ad65a 1338out_unlock:
1da177e4 1339 write_unlock_bh(&__ip_vs_svc_lock);
552ad65a 1340out:
6e08bfb8 1341 ip_vs_scheduler_put(old_sched);
0d1e71b0 1342 ip_vs_pe_put(old_pe);
1da177e4
LT		 1343 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;
0d1e71b0 1356 struct ip_vs_pe *old_pe;
a0840e2e 1357 struct netns_ipvs *ipvs = net_ipvs(svc->net);
0d1e71b0
SH		 1358
1359 pr_info("%s: enter\n", __func__);
1da177e4 1360
f94fd041
JV		 1361 /* Count only IPv4 services for old get/setsockopt interface */
1362 if (svc->af == AF_INET)
a0840e2e 1363 ipvs->num_services--;
f94fd041 1364
6ef757f9 1365 ip_vs_stop_estimator(svc->net, &svc->stats);
1da177e4
LT		 1366
1367 /* Unbind scheduler */
1368 old_sched = svc->scheduler;
1369 ip_vs_unbind_scheduler(svc);
6e08bfb8 1370 ip_vs_scheduler_put(old_sched);
1da177e4 1371
0d1e71b0
SH		 1372 /* Unbind persistence engine */
1373 old_pe = svc->pe;
1374 ip_vs_unbind_pe(svc);
1375 ip_vs_pe_put(old_pe);
1376
1da177e4
LT		 1377 /* 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);
29c2026f 1388 __ip_vs_del_dest(svc->net, dest);
1da177e4
LT		 1389 }
1390
1391 /*
1392 * Update the virtual service counters
1393 */
1394 if (svc->port == FTPPORT)
763f8d0e 1395 atomic_dec(&ipvs->ftpsvc_counter);
1da177e4 1396 else if (svc->port == 0)
763f8d0e 1397 atomic_dec(&ipvs->nullsvc_counter);
1da177e4
LT		 1398
1399 /*
1400 * Free the service if nobody refers to it
1401 */
26c15cfd
JA		 1402 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));
b17fc996 1407 free_percpu(svc->stats.cpustats);
1da177e4 1408 kfree(svc);
26c15cfd 1409 }
1da177e4
LT		 1410
1411 /* decrease the module use count */
1412 ip_vs_use_count_dec();
1413}
1414
1415/*
26c15cfd 1416 * Unlink a service from list and try to delete it if its refcnt reached 0
1da177e4 1417 */
26c15cfd 1418static void ip_vs_unlink_service(struct ip_vs_service *svc)
1da177e4 1419{
1da177e4
LT		 1420 /*
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 */
26c15cfd 1430 IP_VS_WAIT_WHILE(atomic_read(&svc->usecnt) > 0);
1da177e4
LT		 1431
1432 __ip_vs_del_service(svc);
1433
1434 write_unlock_bh(&__ip_vs_svc_lock);
26c15cfd
JA		 1435}
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);
1da177e4
LT		 1445
1446 return 0;
1447}
1448
1449
1450/*
1451 * Flush all the virtual services
1452 */
fc723250 1453static int ip_vs_flush(struct net *net)
1da177e4
LT		 1454{
1455 int idx;
1456 struct ip_vs_service *svc, *nxt;
1457
1458 /*
fc723250 1459 * Flush the service table hashed by <netns,protocol,addr,port>
1da177e4
LT		 1460 */
1461 for(idx = 0; idx < IP_VS_SVC_TAB_SIZE; idx++) {
fc723250
HS		 1462 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);
1da177e4
LT		 1466 }
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) {
fc723250
HS		 1475 if (net_eq(svc->net, net))
1476 ip_vs_unlink_service(svc);
1da177e4
LT		 1477 }
1478 }
1479
1480 return 0;
1481}
1482
7a4f0761
HS		 1483/*
1484 * Delete service by {netns} in the service table.
1485 * Called by __ip_vs_cleanup()
1486 */
503cf15a 1487void ip_vs_service_net_cleanup(struct net *net)
7a4f0761
HS		 1488{
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}
1da177e4
LT		 1561
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
fc723250 1578static int ip_vs_zero_all(struct net *net)
1da177e4
LT		 1579{
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) {
fc723250
HS		 1585 if (net_eq(svc->net, net))
1586 ip_vs_zero_service(svc);
1da177e4
LT		 1587 }
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) {
fc723250
HS		 1592 if (net_eq(svc->net, net))
1593 ip_vs_zero_service(svc);
1da177e4
LT		 1594 }
1595 }
1596
2a0751af 1597 ip_vs_zero_stats(&net_ipvs(net)->tot_stats);
1da177e4
LT		 1598 return 0;
1599}
1600
14e40546 1601#ifdef CONFIG_SYSCTL
1da177e4 1602static int
8d65af78 1603proc_do_defense_mode(ctl_table *table, int write,
1da177e4
LT		 1604 void __user *buffer, size_t *lenp, loff_t *ppos)
1605{
9330419d 1606 struct net *net = current->nsproxy->net_ns;
1da177e4
LT		 1607 int *valp = table->data;
1608 int val = *valp;
1609 int rc;
1610
8d65af78 1611 rc = proc_dointvec(table, write, buffer, lenp, ppos);
1da177e4
LT		 1612 if (write && (*valp != val)) {
1613 if ((*valp < 0) || (*valp > 3)) {
1614 /* Restore the correct value */
1615 *valp = val;
1616 } else {
9330419d 1617 update_defense_level(net_ipvs(net));
1da177e4
LT		 1618 }
1619 }
1620 return rc;
1621}
1622
1da177e4 1623static int
8d65af78 1624proc_do_sync_threshold(ctl_table *table, int write,
1da177e4
LT		 1625 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
8d65af78 1634 rc = proc_dointvec(table, write, buffer, lenp, ppos);
1da177e4
LT		 1635 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
b880c1f0
HS		 1642static 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 {
f131315f
HS		 1656 struct net *net = current->nsproxy->net_ns;
1657 ip_vs_sync_switch_mode(net, val);
b880c1f0
HS		 1658 }
1659 }
1660 return rc;
1661}
1da177e4
LT		 1662
1663/*
1664 * IPVS sysctl table (under the /proc/sys/net/ipv4/vs/)
a0840e2e 1665 * Do not change order or insert new entries without
503cf15a 1666 * align with netns init in ip_vs_control_net_init()
1da177e4
LT		 1667 */
1668
1669static struct ctl_table vs_vars[] = {
1670 {
1da177e4 1671 .procname = "amemthresh",
1da177e4
LT		 1672 .maxlen = sizeof(int),
1673 .mode = 0644,
6d9f239a 1674 .proc_handler = proc_dointvec,
1da177e4 1675 },
1da177e4 1676 {
1da177e4 1677 .procname = "am_droprate",
1da177e4
LT		 1678 .maxlen = sizeof(int),
1679 .mode = 0644,
6d9f239a 1680 .proc_handler = proc_dointvec,
1da177e4
LT		 1681 },
1682 {
1da177e4 1683 .procname = "drop_entry",
1da177e4
LT		 1684 .maxlen = sizeof(int),
1685 .mode = 0644,
6d9f239a 1686 .proc_handler = proc_do_defense_mode,
1da177e4
LT		 1687 },
1688 {
1da177e4 1689 .procname = "drop_packet",
1da177e4
LT		 1690 .maxlen = sizeof(int),
1691 .mode = 0644,
6d9f239a 1692 .proc_handler = proc_do_defense_mode,
1da177e4 1693 },
f4bc17cd
JA		 1694#ifdef CONFIG_IP_VS_NFCT
1695 {
1696 .procname = "conntrack",
f4bc17cd
JA		 1697 .maxlen = sizeof(int),
1698 .mode = 0644,
1699 .proc_handler = &proc_dointvec,
1700 },
1701#endif
1da177e4 1702 {
1da177e4 1703 .procname = "secure_tcp",
1da177e4
LT		 1704 .maxlen = sizeof(int),
1705 .mode = 0644,
6d9f239a 1706 .proc_handler = proc_do_defense_mode,
1da177e4 1707 },
8a803040
JA		 1708 {
1709 .procname = "snat_reroute",
8a803040
JA		 1710 .maxlen = sizeof(int),
1711 .mode = 0644,
1712 .proc_handler = &proc_dointvec,
1713 },
b880c1f0
HS		 1714 {
1715 .procname = "sync_version",
b880c1f0
HS		 1716 .maxlen = sizeof(int),
1717 .mode = 0644,
1718 .proc_handler = &proc_do_sync_mode,
1719 },
a0840e2e
HS		 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",
1747 .maxlen = sizeof(int),
1748 .mode = 0644,
1749 .proc_handler = proc_dointvec,
1750 },
1751#ifdef CONFIG_IP_VS_DEBUG
1752 {
1753 .procname = "debug_level",
1754 .data = &sysctl_ip_vs_debug_level,
1755 .maxlen = sizeof(int),
1756 .mode = 0644,
1757 .proc_handler = proc_dointvec,
1758 },
1759#endif
1da177e4
LT		 1760#if 0
1761 {
1da177e4
LT		 1762 .procname = "timeout_established",
1763 .data = &vs_timeout_table_dos.timeout[IP_VS_S_ESTABLISHED],
1764 .maxlen = sizeof(int),
1765 .mode = 0644,
6d9f239a 1766 .proc_handler = proc_dointvec_jiffies,
1da177e4
LT		 1767 },
1768 {
1da177e4
LT		 1769 .procname = "timeout_synsent",
1770 .data = &vs_timeout_table_dos.timeout[IP_VS_S_SYN_SENT],
1771 .maxlen = sizeof(int),
1772 .mode = 0644,
6d9f239a 1773 .proc_handler = proc_dointvec_jiffies,
1da177e4
LT		 1774 },
1775 {
1da177e4
LT		 1776 .procname = "timeout_synrecv",
1777 .data = &vs_timeout_table_dos.timeout[IP_VS_S_SYN_RECV],
1778 .maxlen = sizeof(int),
1779 .mode = 0644,
6d9f239a 1780 .proc_handler = proc_dointvec_jiffies,
1da177e4
LT		 1781 },
1782 {
1da177e4
LT		 1783 .procname = "timeout_finwait",
1784 .data = &vs_timeout_table_dos.timeout[IP_VS_S_FIN_WAIT],
1785 .maxlen = sizeof(int),
1786 .mode = 0644,
6d9f239a 1787 .proc_handler = proc_dointvec_jiffies,
1da177e4
LT		 1788 },
1789 {
1da177e4
LT		 1790 .procname = "timeout_timewait",
1791 .data = &vs_timeout_table_dos.timeout[IP_VS_S_TIME_WAIT],
1792 .maxlen = sizeof(int),
1793 .mode = 0644,
6d9f239a 1794 .proc_handler = proc_dointvec_jiffies,
1da177e4
LT		 1795 },
1796 {
1da177e4
LT		 1797 .procname = "timeout_close",
1798 .data = &vs_timeout_table_dos.timeout[IP_VS_S_CLOSE],
1799 .maxlen = sizeof(int),
1800 .mode = 0644,
6d9f239a 1801 .proc_handler = proc_dointvec_jiffies,
1da177e4
LT		 1802 },
1803 {
1da177e4
LT		 1804 .procname = "timeout_closewait",
1805 .data = &vs_timeout_table_dos.timeout[IP_VS_S_CLOSE_WAIT],
1806 .maxlen = sizeof(int),
1807 .mode = 0644,
6d9f239a 1808 .proc_handler = proc_dointvec_jiffies,
1da177e4
LT		 1809 },
1810 {
1da177e4
LT		 1811 .procname = "timeout_lastack",
1812 .data = &vs_timeout_table_dos.timeout[IP_VS_S_LAST_ACK],
1813 .maxlen = sizeof(int),
1814 .mode = 0644,
6d9f239a 1815 .proc_handler = proc_dointvec_jiffies,
1da177e4
LT		 1816 },
1817 {
1da177e4
LT		 1818 .procname = "timeout_listen",
1819 .data = &vs_timeout_table_dos.timeout[IP_VS_S_LISTEN],
1820 .maxlen = sizeof(int),
1821 .mode = 0644,
6d9f239a 1822 .proc_handler = proc_dointvec_jiffies,
1da177e4
LT		 1823 },
1824 {
1da177e4
LT		 1825 .procname = "timeout_synack",
1826 .data = &vs_timeout_table_dos.timeout[IP_VS_S_SYNACK],
1827 .maxlen = sizeof(int),
1828 .mode = 0644,
6d9f239a 1829 .proc_handler = proc_dointvec_jiffies,
1da177e4
LT		 1830 },
1831 {
1da177e4
LT		 1832 .procname = "timeout_udp",
1833 .data = &vs_timeout_table_dos.timeout[IP_VS_S_UDP],
1834 .maxlen = sizeof(int),
1835 .mode = 0644,
6d9f239a 1836 .proc_handler = proc_dointvec_jiffies,
1da177e4
LT		 1837 },
1838 {
1da177e4
LT		 1839 .procname = "timeout_icmp",
1840 .data = &vs_timeout_table_dos.timeout[IP_VS_S_ICMP],
1841 .maxlen = sizeof(int),
1842 .mode = 0644,
6d9f239a 1843 .proc_handler = proc_dointvec_jiffies,
1da177e4
LT		 1844 },
1845#endif
f8572d8f 1846 { }
1da177e4
LT		 1847};
1848
5587da55 1849const struct ctl_path net_vs_ctl_path[] = {
f8572d8f
EB		 1850 { .procname = "net", },
1851 { .procname = "ipv4", },
90754f8e
PE		 1852 { .procname = "vs", },
1853 { }
1da177e4 1854};
90754f8e 1855EXPORT_SYMBOL_GPL(net_vs_ctl_path);
14e40546 1856#endif
1da177e4 1857
1da177e4
LT		 1858#ifdef CONFIG_PROC_FS
1859
1860struct ip_vs_iter {
fc723250 1861 struct seq_net_private p; /* Do not move this, netns depends upon it*/
1da177e4
LT		 1862 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{
fc723250 1888 struct net *net = seq_file_net(seq);
1da177e4
LT		 1889 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) {
fc723250 1896 if (net_eq(svc->net, net) && pos-- == 0) {
1da177e4
LT		 1897 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) {
fc723250 1907 if (net_eq(svc->net, net) && pos-- == 0) {
1da177e4
LT		 1908 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)
563e94f0 1919__acquires(__ip_vs_svc_lock)
1da177e4
LT		 1920{
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)
563e94f0 1973__releases(__ip_vs_svc_lock)
1da177e4
LT		 1974{
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",
6f7edb48 1984 NVERSION(IP_VS_VERSION_CODE), ip_vs_conn_tab_size);
1da177e4
LT		 1985 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
667a5f18
VB		 1994 if (iter->table == ip_vs_svc_table) {
1995#ifdef CONFIG_IP_VS_IPV6
1996 if (svc->af == AF_INET6)
5b095d98 1997 seq_printf(seq, "%s [%pI6]:%04X %s ",
667a5f18 1998 ip_vs_proto_name(svc->protocol),
38ff4fa4 1999 &svc->addr.in6,
667a5f18
VB		 2000 ntohs(svc->port),
2001 svc->scheduler->name);
2002 else
2003#endif
26ec037f 2004 seq_printf(seq, "%s %08X:%04X %s %s ",
667a5f18
VB		 2005 ip_vs_proto_name(svc->protocol),
2006 ntohl(svc->addr.ip),
2007 ntohs(svc->port),
26ec037f
NC		 2008 svc->scheduler->name,
2009 (svc->flags & IP_VS_SVC_F_ONEPACKET)?"ops ":"");
667a5f18 2010 } else {
26ec037f
NC		 2011 seq_printf(seq, "FWM %08X %s %s",
2012 svc->fwmark, svc->scheduler->name,
2013 (svc->flags & IP_VS_SVC_F_ONEPACKET)?"ops ":"");
667a5f18 2014 }
1da177e4
LT		 2015
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) {
667a5f18
VB		 2024#ifdef CONFIG_IP_VS_IPV6
2025 if (dest->af == AF_INET6)
2026 seq_printf(seq,
5b095d98 2027 " -> [%pI6]:%04X"
667a5f18 2028 " %-7s %-6d %-10d %-10d\n",
38ff4fa4 2029 &dest->addr.in6,
667a5f18
VB		 2030 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
1da177e4
LT		 2047 }
2048 }
2049 return 0;
2050}
2051
56b3d975 2052static const struct seq_operations ip_vs_info_seq_ops = {
1da177e4
LT		 2053 .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{
fc723250 2061 return seq_open_net(inode, file, &ip_vs_info_seq_ops,
cf7732e4 2062 sizeof(struct ip_vs_iter));
1da177e4
LT		 2063}
2064
9a32144e 2065static const struct file_operations ip_vs_info_fops = {
1da177e4
LT		 2066 .owner = THIS_MODULE,
2067 .open = ip_vs_info_open,
2068 .read = seq_read,
2069 .llseek = seq_lseek,
0f08190f 2070 .release = seq_release_net,
1da177e4
LT		 2071};
2072
1da177e4
LT		 2073static int ip_vs_stats_show(struct seq_file *seq, void *v)
2074{
b17fc996 2075 struct net *net = seq_file_single_net(seq);
55a3d4e1 2076 struct ip_vs_stats_user show;
1da177e4
LT		 2077
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
55a3d4e1
JA		 2084 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);
1da177e4
LT		 2089
2090/* 01234567 01234567 01234567 0123456701234567 0123456701234567 */
2091 seq_puts(seq,
2092 " Conns/s Pkts/s Pkts/s Bytes/s Bytes/s\n");
55a3d4e1
JA		 2093 seq_printf(seq, "%8X %8X %8X %16X %16X\n",
2094 show.cps, show.inpps, show.outpps,
2095 show.inbps, show.outbps);
1da177e4
LT		 2096
2097 return 0;
2098}
2099
2100static int ip_vs_stats_seq_open(struct inode *inode, struct file *file)
2101{
fc723250 2102 return single_open_net(inode, file, ip_vs_stats_show);
1da177e4
LT		 2103}
2104
9a32144e 2105static const struct file_operations ip_vs_stats_fops = {
1da177e4
LT		 2106 .owner = THIS_MODULE,
2107 .open = ip_vs_stats_seq_open,
2108 .read = seq_read,
2109 .llseek = seq_lseek,
0f08190f 2110 .release = single_release_net,
1da177e4
LT		 2111};
2112
b17fc996
HS		 2113static int ip_vs_stats_percpu_show(struct seq_file *seq, void *v)
2114{
2115 struct net *net = seq_file_single_net(seq);
2a0751af
JA		 2116 struct ip_vs_stats *tot_stats = &net_ipvs(net)->tot_stats;
2117 struct ip_vs_cpu_stats *cpustats = tot_stats->cpustats;
ea9f22cc 2118 struct ip_vs_stats_user rates;
b17fc996
HS		 2119 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) {
2a0751af
JA		 2128 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
b17fc996 2138 seq_printf(seq, "%3X %8X %8X %8X %16LX %16LX\n",
2a0751af
JA		 2139 i, u->ustats.conns, u->ustats.inpkts,
2140 u->ustats.outpkts, (__u64)inbytes,
2141 (__u64)outbytes);
b17fc996
HS		 2142 }
2143
2144 spin_lock_bh(&tot_stats->lock);
ea9f22cc 2145
b17fc996
HS		 2146 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
ea9f22cc
JA		 2152 ip_vs_read_estimator(&rates, tot_stats);
2153
2154 spin_unlock_bh(&tot_stats->lock);
2155
b17fc996
HS		 2156/* 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",
ea9f22cc
JA		 2160 rates.cps,
2161 rates.inpps,
2162 rates.outpps,
2163 rates.inbps,
2164 rates.outbps);
b17fc996
HS		 2165
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,
0f08190f 2179 .release = single_release_net,
b17fc996 2180};
1da177e4
LT		 2181#endif
2182
2183/*
2184 * Set timeout values for tcp tcpfin udp in the timeout_table.
2185 */
9330419d 2186static int ip_vs_set_timeout(struct net *net, struct ip_vs_timeout_user *u)
1da177e4 2187{
091bb34c 2188#if defined(CONFIG_IP_VS_PROTO_TCP) || defined(CONFIG_IP_VS_PROTO_UDP)
9330419d 2189 struct ip_vs_proto_data *pd;
091bb34c 2190#endif
9330419d 2191
1da177e4
LT		 2192 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) {
9330419d
HS		 2199 pd = ip_vs_proto_data_get(net, IPPROTO_TCP);
2200 pd->timeout_table[IP_VS_TCP_S_ESTABLISHED]
1da177e4
LT		 2201 = u->tcp_timeout * HZ;
2202 }
2203
2204 if (u->tcp_fin_timeout) {
9330419d
HS		 2205 pd = ip_vs_proto_data_get(net, IPPROTO_TCP);
2206 pd->timeout_table[IP_VS_TCP_S_FIN_WAIT]
1da177e4
LT		 2207 = u->tcp_fin_timeout * HZ;
2208 }
2209#endif
2210
2211#ifdef CONFIG_IP_VS_PROTO_UDP
2212 if (u->udp_timeout) {
9330419d
HS		 2213 pd = ip_vs_proto_data_get(net, IPPROTO_UDP);
2214 pd->timeout_table[IP_VS_UDP_S_NORMAL]
1da177e4
LT		 2215 = 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
9b5b5cff 2230static const unsigned char set_arglen[SET_CMDID(IP_VS_SO_SET_MAX)+1] = {
1da177e4
LT		 2231 [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
c860c6b1
JV		 2244static void ip_vs_copy_usvc_compat(struct ip_vs_service_user_kern *usvc,
2245 struct ip_vs_service_user *usvc_compat)
2246{
0d1e71b0
SH		 2247 memset(usvc, 0, sizeof(*usvc));
2248
c860c6b1
JV		 2249 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{
0d1e71b0
SH		 2266 memset(udest, 0, sizeof(*udest));
2267
c860c6b1
JV		 2268 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
1da177e4
LT		 2276static int
2277do_ip_vs_set_ctl(struct sock *sk, int cmd, void __user *user, unsigned int len)
2278{
fc723250 2279 struct net *net = sock_net(sk);
1da177e4
LT		 2280 int ret;
2281 unsigned char arg[MAX_ARG_LEN];
c860c6b1
JV		 2282 struct ip_vs_service_user *usvc_compat;
2283 struct ip_vs_service_user_kern usvc;
1da177e4 2284 struct ip_vs_service *svc;
c860c6b1
JV		 2285 struct ip_vs_dest_user *udest_compat;
2286 struct ip_vs_dest_user_kern udest;
ae1d48b2 2287 struct netns_ipvs *ipvs = net_ipvs(net);
1da177e4
LT		 2288
2289 if (!capable(CAP_NET_ADMIN))
2290 return -EPERM;
2291
04bcef2a
AV		 2292 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;
1da177e4 2296 if (len != set_arglen[SET_CMDID(cmd)]) {
1e3e238e
HE		 2297 pr_err("set_ctl: len %u != %u\n",
2298 len, set_arglen[SET_CMDID(cmd)]);
1da177e4
LT		 2299 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
ae1d48b2
HS		 2308 /* 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
14cc3e2b 2326 if (mutex_lock_interruptible(&__ip_vs_mutex)) {
1da177e4
LT		 2327 ret = -ERESTARTSYS;
2328 goto out_dec;
2329 }
2330
2331 if (cmd == IP_VS_SO_SET_FLUSH) {
2332 /* Flush the virtual service */
fc723250 2333 ret = ip_vs_flush(net);
1da177e4
LT		 2334 goto out_unlock;
2335 } else if (cmd == IP_VS_SO_SET_TIMEOUT) {
2336 /* Set timeout values for (tcp tcpfin udp) */
9330419d 2337 ret = ip_vs_set_timeout(net, (struct ip_vs_timeout_user *)arg);
1da177e4 2338 goto out_unlock;
1da177e4
LT		 2339 }
2340
c860c6b1
JV		 2341 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);
1da177e4
LT		 2348
2349 if (cmd == IP_VS_SO_SET_ZERO) {
2350 /* if no service address is set, zero counters in all */
c860c6b1 2351 if (!usvc.fwmark && !usvc.addr.ip && !usvc.port) {
fc723250 2352 ret = ip_vs_zero_all(net);
1da177e4
LT		 2353 goto out_unlock;
2354 }
2355 }
2356
2906f66a
VMR		 2357 /* 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) {
1e3e238e
HE		 2360 pr_err("set_ctl: invalid protocol: %d %pI4:%d %s\n",
2361 usvc.protocol, &usvc.addr.ip,
2362 ntohs(usvc.port), usvc.sched_name);
1da177e4
LT		 2363 ret = -EFAULT;
2364 goto out_unlock;
2365 }
2366
2367 /* Lookup the exact service by <protocol, addr, port> or fwmark */
c860c6b1 2368 if (usvc.fwmark == 0)
fc723250 2369 svc = __ip_vs_service_find(net, usvc.af, usvc.protocol,
26c15cfd 2370 &usvc.addr, usvc.port);
1da177e4 2371 else
fc723250 2372 svc = __ip_vs_svc_fwm_find(net, usvc.af, usvc.fwmark);
1da177e4
LT		 2373
2374 if (cmd != IP_VS_SO_SET_ADD
c860c6b1 2375 && (svc == NULL || svc->protocol != usvc.protocol)) {
1da177e4 2376 ret = -ESRCH;
26c15cfd 2377 goto out_unlock;
1da177e4
LT		 2378 }
2379
2380 switch (cmd) {
2381 case IP_VS_SO_SET_ADD:
2382 if (svc != NULL)
2383 ret = -EEXIST;
2384 else
fc723250 2385 ret = ip_vs_add_service(net, &usvc, &svc);
1da177e4
LT		 2386 break;
2387 case IP_VS_SO_SET_EDIT:
c860c6b1 2388 ret = ip_vs_edit_service(svc, &usvc);
1da177e4
LT		 2389 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:
c860c6b1 2399 ret = ip_vs_add_dest(svc, &udest);
1da177e4
LT		 2400 break;
2401 case IP_VS_SO_SET_EDITDEST:
c860c6b1 2402 ret = ip_vs_edit_dest(svc, &udest);
1da177e4
LT		 2403 break;
2404 case IP_VS_SO_SET_DELDEST:
c860c6b1 2405 ret = ip_vs_del_dest(svc, &udest);
1da177e4
LT		 2406 break;
2407 default:
2408 ret = -EINVAL;
2409 }
2410
1da177e4 2411 out_unlock:
14cc3e2b 2412 mutex_unlock(&__ip_vs_mutex);
1da177e4
LT		 2413 out_dec:
2414 /* decrease the module use count */
2415 ip_vs_use_count_dec();
2416
2417 return ret;
2418}
2419
2420
1da177e4
LT		 2421static void
2422ip_vs_copy_service(struct ip_vs_service_entry *dst, struct ip_vs_service *src)
2423{
2424 dst->protocol = src->protocol;
e7ade46a 2425 dst->addr = src->addr.ip;
1da177e4
LT		 2426 dst->port = src->port;
2427 dst->fwmark = src->fwmark;
4da62fc7 2428 strlcpy(dst->sched_name, src->scheduler->name, sizeof(dst->sched_name));
1da177e4
LT		 2429 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
fc723250
HS		 2437__ip_vs_get_service_entries(struct net *net,
2438 const struct ip_vs_get_services *get,
1da177e4
LT		 2439 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) {
f94fd041 2448 /* Only expose IPv4 entries to old interface */
fc723250 2449 if (svc->af != AF_INET || !net_eq(svc->net, net))
f94fd041
JV		 2450 continue;
2451
1da177e4
LT		 2452 if (count >= get->num_services)
2453 goto out;
4da62fc7 2454 memset(&entry, 0, sizeof(entry));
1da177e4
LT		 2455 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) {
f94fd041 2467 /* Only expose IPv4 entries to old interface */
fc723250 2468 if (svc->af != AF_INET || !net_eq(svc->net, net))
f94fd041
JV		 2469 continue;
2470
1da177e4
LT		 2471 if (count >= get->num_services)
2472 goto out;
4da62fc7 2473 memset(&entry, 0, sizeof(entry));
1da177e4
LT		 2474 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 }
552ad65a 2483out:
1da177e4
LT		 2484 return ret;
2485}
2486
2487static inline int
fc723250 2488__ip_vs_get_dest_entries(struct net *net, const struct ip_vs_get_dests *get,
1da177e4
LT		 2489 struct ip_vs_get_dests __user *uptr)
2490{
2491 struct ip_vs_service *svc;
b18610de 2492 union nf_inet_addr addr = { .ip = get->addr };
1da177e4
LT		 2493 int ret = 0;
2494
2495 if (get->fwmark)
fc723250 2496 svc = __ip_vs_svc_fwm_find(net, AF_INET, get->fwmark);
1da177e4 2497 else
fc723250 2498 svc = __ip_vs_service_find(net, AF_INET, get->protocol, &addr,
26c15cfd 2499 get->port);
b18610de 2500
1da177e4
LT		 2501 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
e7ade46a 2510 entry.addr = dest->addr.ip;
1da177e4
LT		 2511 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 }
1da177e4
LT		 2527 } else
2528 ret = -ESRCH;
2529 return ret;
2530}
2531
2532static inline void
9330419d 2533__ip_vs_get_timeouts(struct net *net, struct ip_vs_timeout_user *u)
1da177e4 2534{
091bb34c 2535#if defined(CONFIG_IP_VS_PROTO_TCP) || defined(CONFIG_IP_VS_PROTO_UDP)
9330419d 2536 struct ip_vs_proto_data *pd;
091bb34c 2537#endif
9330419d 2538
1da177e4 2539#ifdef CONFIG_IP_VS_PROTO_TCP
9330419d
HS		 2540 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;
1da177e4
LT		 2543#endif
2544#ifdef CONFIG_IP_VS_PROTO_UDP
9330419d 2545 pd = ip_vs_proto_data_get(net, IPPROTO_UDP);
1da177e4 2546 u->udp_timeout =
9330419d 2547 pd->timeout_table[IP_VS_UDP_S_NORMAL] / HZ;
1da177e4
LT		 2548#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
9b5b5cff 2560static const unsigned char get_arglen[GET_CMDID(IP_VS_SO_GET_MAX)+1] = {
1da177e4
LT		 2561 [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;
04bcef2a 2575 unsigned int copylen;
fc723250 2576 struct net *net = sock_net(sk);
f131315f 2577 struct netns_ipvs *ipvs = net_ipvs(net);
1da177e4 2578
fc723250 2579 BUG_ON(!net);
1da177e4
LT		 2580 if (!capable(CAP_NET_ADMIN))
2581 return -EPERM;
2582
04bcef2a
AV		 2583 if (cmd < IP_VS_BASE_CTL || cmd > IP_VS_SO_GET_MAX)
2584 return -EINVAL;
2585
1da177e4 2586 if (*len < get_arglen[GET_CMDID(cmd)]) {
1e3e238e
HE		 2587 pr_err("get_ctl: len %u < %u\n",
2588 *len, get_arglen[GET_CMDID(cmd)]);
1da177e4
LT		 2589 return -EINVAL;
2590 }
2591
04bcef2a
AV		 2592 copylen = get_arglen[GET_CMDID(cmd)];
2593 if (copylen > 128)
2594 return -EINVAL;
2595
2596 if (copy_from_user(arg, user, copylen) != 0)
1da177e4 2597 return -EFAULT;
ae1d48b2
HS		 2598 /*
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 }
1da177e4 2625
14cc3e2b 2626 if (mutex_lock_interruptible(&__ip_vs_mutex))
1da177e4
LT		 2627 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)",
6f7edb48 2635 NVERSION(IP_VS_VERSION_CODE), ip_vs_conn_tab_size);
1da177e4
LT		 2636 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;
6f7edb48 2648 info.size = ip_vs_conn_tab_size;
a0840e2e 2649 info.num_services = ipvs->num_services;
1da177e4
LT		 2650 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) {
1e3e238e 2664 pr_err("length: %u != %u\n", *len, size);
1da177e4
LT		 2665 ret = -EINVAL;
2666 goto out;
2667 }
fc723250 2668 ret = __ip_vs_get_service_entries(net, get, user);
1da177e4
LT		 2669 }
2670 break;
2671
2672 case IP_VS_SO_GET_SERVICE:
2673 {
2674 struct ip_vs_service_entry *entry;
2675 struct ip_vs_service *svc;
b18610de 2676 union nf_inet_addr addr;
1da177e4
LT		 2677
2678 entry = (struct ip_vs_service_entry *)arg;
b18610de 2679 addr.ip = entry->addr;
1da177e4 2680 if (entry->fwmark)
fc723250 2681 svc = __ip_vs_svc_fwm_find(net, AF_INET, entry->fwmark);
1da177e4 2682 else
fc723250
HS		 2683 svc = __ip_vs_service_find(net, AF_INET,
2684 entry->protocol, &addr,
2685 entry->port);
1da177e4
LT		 2686 if (svc) {
2687 ip_vs_copy_service(entry, svc);
2688 if (copy_to_user(user, entry, sizeof(*entry)) != 0)
2689 ret = -EFAULT;
1da177e4
LT		 2690 } 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) {
1e3e238e 2704 pr_err("length: %u != %u\n", *len, size);
1da177e4
LT		 2705 ret = -EINVAL;
2706 goto out;
2707 }
fc723250 2708 ret = __ip_vs_get_dest_entries(net, get, user);
1da177e4
LT		 2709 }
2710 break;
2711
2712 case IP_VS_SO_GET_TIMEOUT:
2713 {
2714 struct ip_vs_timeout_user t;
2715
9330419d 2716 __ip_vs_get_timeouts(net, &t);
1da177e4
LT		 2717 if (copy_to_user(user, &t, sizeof(t)) != 0)
2718 ret = -EFAULT;
2719 }
2720 break;
2721
1da177e4
LT		 2722 default:
2723 ret = -EINVAL;
2724 }
2725
552ad65a 2726out:
14cc3e2b 2727 mutex_unlock(&__ip_vs_mutex);
1da177e4
LT		 2728 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,
16fcec35 2740 .owner = THIS_MODULE,
1da177e4
LT		 2741};
2742
9a812198
JV		 2743/*
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,
c6d2d445 2754 .netnsok = true, /* Make ipvsadm to work on netns */
9a812198
JV		 2755};
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 },
0d1e71b0
SH		 2785 [IPVS_SVC_ATTR_PE_NAME] = { .type = NLA_NUL_STRING,
2786 .len = IP_VS_PENAME_MAXLEN },
9a812198
JV		 2787 [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{
55a3d4e1 2812 struct ip_vs_stats_user ustats;
9a812198
JV		 2813 struct nlattr *nl_stats = nla_nest_start(skb, container_type);
2814 if (!nl_stats)
2815 return -EMSGSIZE;
2816
55a3d4e1 2817 ip_vs_copy_stats(&ustats, stats);
9a812198 2818
969e8e25
DM		 2819 if (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))
2829 goto nla_put_failure;
9a812198
JV		 2830 nla_nest_end(skb, nl_stats);
2831
2832 return 0;
2833
2834nla_put_failure:
9a812198
JV		 2835 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
969e8e25
DM		 2850 if (nla_put_u16(skb, IPVS_SVC_ATTR_AF, svc->af))
2851 goto nla_put_failure;
9a812198 2852 if (svc->fwmark) {
969e8e25
DM		 2853 if (nla_put_u32(skb, IPVS_SVC_ATTR_FWMARK, svc->fwmark))
2854 goto nla_put_failure;
9a812198 2855 } else {
969e8e25
DM		 2856 if (nla_put_u16(skb, IPVS_SVC_ATTR_PROTOCOL, svc->protocol) ||
2857 nla_put(skb, IPVS_SVC_ATTR_ADDR, sizeof(svc->addr), &svc->addr) ||
2858 nla_put_u16(skb, IPVS_SVC_ATTR_PORT, svc->port))
2859 goto nla_put_failure;
9a812198
JV		 2860 }
2861
969e8e25
DM		 2862 if (nla_put_string(skb, IPVS_SVC_ATTR_SCHED_NAME, svc->scheduler->name) ||
2863 (svc->pe &&
2864 nla_put_string(skb, IPVS_SVC_ATTR_PE_NAME, svc->pe->name)) ||
2865 nla_put(skb, IPVS_SVC_ATTR_FLAGS, sizeof(flags), &flags) ||
2866 nla_put_u32(skb, IPVS_SVC_ATTR_TIMEOUT, svc->timeout / HZ) ||
2867 nla_put_u32(skb, IPVS_SVC_ATTR_NETMASK, svc->netmask))
2868 goto nla_put_failure;
9a812198
JV		 2869 if (ip_vs_genl_fill_stats(skb, IPVS_SVC_ATTR_STATS, &svc->stats))
2870 goto nla_put_failure;
2871
2872 nla_nest_end(skb, nl_service);
2873
2874 return 0;
2875
2876nla_put_failure:
2877 nla_nest_cancel(skb, nl_service);
2878 return -EMSGSIZE;
2879}
2880
2881static int ip_vs_genl_dump_service(struct sk_buff *skb,
2882 struct ip_vs_service *svc,
2883 struct netlink_callback *cb)
2884{
2885 void *hdr;
2886
2887 hdr = genlmsg_put(skb, NETLINK_CB(cb->skb).pid, cb->nlh->nlmsg_seq,
2888 &ip_vs_genl_family, NLM_F_MULTI,
2889 IPVS_CMD_NEW_SERVICE);
2890 if (!hdr)
2891 return -EMSGSIZE;
2892
2893 if (ip_vs_genl_fill_service(skb, svc) < 0)
2894 goto nla_put_failure;
2895
2896 return genlmsg_end(skb, hdr);
2897
2898nla_put_failure:
2899 genlmsg_cancel(skb, hdr);
2900 return -EMSGSIZE;
2901}
2902
2903static int ip_vs_genl_dump_services(struct sk_buff *skb,
2904 struct netlink_callback *cb)
2905{
2906 int idx = 0, i;
2907 int start = cb->args[0];
2908 struct ip_vs_service *svc;
fc723250 2909 struct net *net = skb_sknet(skb);
9a812198
JV		 2910
2911 mutex_lock(&__ip_vs_mutex);
2912 for (i = 0; i < IP_VS_SVC_TAB_SIZE; i++) {
2913 list_for_each_entry(svc, &ip_vs_svc_table[i], s_list) {
fc723250 2914 if (++idx <= start || !net_eq(svc->net, net))
9a812198
JV		 2915 continue;
2916 if (ip_vs_genl_dump_service(skb, svc, cb) < 0) {
2917 idx--;
2918 goto nla_put_failure;
2919 }
2920 }
2921 }
2922
2923 for (i = 0; i < IP_VS_SVC_TAB_SIZE; i++) {
2924 list_for_each_entry(svc, &ip_vs_svc_fwm_table[i], f_list) {
fc723250 2925 if (++idx <= start || !net_eq(svc->net, net))
9a812198
JV		 2926 continue;
2927 if (ip_vs_genl_dump_service(skb, svc, cb) < 0) {
2928 idx--;
2929 goto nla_put_failure;
2930 }
2931 }
2932 }
2933
2934nla_put_failure:
2935 mutex_unlock(&__ip_vs_mutex);
2936 cb->args[0] = idx;
2937
2938 return skb->len;
2939}
2940
fc723250
HS		 2941static int ip_vs_genl_parse_service(struct net *net,
2942 struct ip_vs_service_user_kern *usvc,
26c15cfd
JA		 2943 struct nlattr *nla, int full_entry,
2944 struct ip_vs_service **ret_svc)
9a812198
JV		 2945{
2946 struct nlattr *attrs[IPVS_SVC_ATTR_MAX + 1];
2947 struct nlattr *nla_af, *nla_port, *nla_fwmark, *nla_protocol, *nla_addr;
26c15cfd 2948 struct ip_vs_service *svc;
9a812198
JV		 2949
2950 /* Parse mandatory identifying service fields first */
2951 if (nla == NULL ||
2952 nla_parse_nested(attrs, IPVS_SVC_ATTR_MAX, nla, ip_vs_svc_policy))
2953 return -EINVAL;
2954
2955 nla_af = attrs[IPVS_SVC_ATTR_AF];
2956 nla_protocol = attrs[IPVS_SVC_ATTR_PROTOCOL];
2957 nla_addr = attrs[IPVS_SVC_ATTR_ADDR];
2958 nla_port = attrs[IPVS_SVC_ATTR_PORT];
2959 nla_fwmark = attrs[IPVS_SVC_ATTR_FWMARK];
2960
2961 if (!(nla_af && (nla_fwmark || (nla_port && nla_protocol && nla_addr))))
2962 return -EINVAL;
2963
258c8893
SH		 2964 memset(usvc, 0, sizeof(*usvc));
2965
c860c6b1 2966 usvc->af = nla_get_u16(nla_af);
f94fd041
JV		 2967#ifdef CONFIG_IP_VS_IPV6
2968 if (usvc->af != AF_INET && usvc->af != AF_INET6)
2969#else
2970 if (usvc->af != AF_INET)
2971#endif
9a812198
JV		 2972 return -EAFNOSUPPORT;
2973
2974 if (nla_fwmark) {
2975 usvc->protocol = IPPROTO_TCP;
2976 usvc->fwmark = nla_get_u32(nla_fwmark);
2977 } else {
2978 usvc->protocol = nla_get_u16(nla_protocol);
2979 nla_memcpy(&usvc->addr, nla_addr, sizeof(usvc->addr));
2980 usvc->port = nla_get_u16(nla_port);
2981 usvc->fwmark = 0;
2982 }
2983
26c15cfd 2984 if (usvc->fwmark)
fc723250 2985 svc = __ip_vs_svc_fwm_find(net, usvc->af, usvc->fwmark);
26c15cfd 2986 else
fc723250 2987 svc = __ip_vs_service_find(net, usvc->af, usvc->protocol,
26c15cfd
JA		 2988 &usvc->addr, usvc->port);
2989 *ret_svc = svc;
2990
9a812198
JV		 2991 /* If a full entry was requested, check for the additional fields */
2992 if (full_entry) {
0d1e71b0 2993 struct nlattr *nla_sched, *nla_flags, *nla_pe, *nla_timeout,
9a812198
JV		 2994 *nla_netmask;
2995 struct ip_vs_flags flags;
9a812198
JV		 2996
2997 nla_sched = attrs[IPVS_SVC_ATTR_SCHED_NAME];
0d1e71b0 2998 nla_pe = attrs[IPVS_SVC_ATTR_PE_NAME];
9a812198
JV		 2999 nla_flags = attrs[IPVS_SVC_ATTR_FLAGS];
3000 nla_timeout = attrs[IPVS_SVC_ATTR_TIMEOUT];
3001 nla_netmask = attrs[IPVS_SVC_ATTR_NETMASK];
3002
3003 if (!(nla_sched && nla_flags && nla_timeout && nla_netmask))
3004 return -EINVAL;
3005
3006 nla_memcpy(&flags, nla_flags, sizeof(flags));
3007
3008 /* prefill flags from service if it already exists */
26c15cfd 3009 if (svc)
9a812198 3010 usvc->flags = svc->flags;
9a812198
JV		 3011
3012 /* set new flags from userland */
3013 usvc->flags = (usvc->flags & ~flags.mask) |
3014 (flags.flags & flags.mask);
c860c6b1 3015 usvc->sched_name = nla_data(nla_sched);
0d1e71b0 3016 usvc->pe_name = nla_pe ? nla_data(nla_pe) : NULL;
9a812198
JV		 3017 usvc->timeout = nla_get_u32(nla_timeout);
3018 usvc->netmask = nla_get_u32(nla_netmask);
3019 }
3020
3021 return 0;
3022}
3023
fc723250
HS		 3024static struct ip_vs_service *ip_vs_genl_find_service(struct net *net,
3025 struct nlattr *nla)
9a812198 3026{
c860c6b1 3027 struct ip_vs_service_user_kern usvc;
26c15cfd 3028 struct ip_vs_service *svc;
9a812198
JV		 3029 int ret;
3030
fc723250 3031 ret = ip_vs_genl_parse_service(net, &usvc, nla, 0, &svc);
26c15cfd 3032 return ret ? ERR_PTR(ret) : svc;
9a812198
JV		 3033}
3034
3035static int ip_vs_genl_fill_dest(struct sk_buff *skb, struct ip_vs_dest *dest)
3036{
3037 struct nlattr *nl_dest;
3038
3039 nl_dest = nla_nest_start(skb, IPVS_CMD_ATTR_DEST);
3040 if (!nl_dest)
3041 return -EMSGSIZE;
3042
969e8e25
DM		 3043 if (nla_put(skb, IPVS_DEST_ATTR_ADDR, sizeof(dest->addr), &dest->addr) ||
3044 nla_put_u16(skb, IPVS_DEST_ATTR_PORT, dest->port) ||
3045 nla_put_u32(skb, IPVS_DEST_ATTR_FWD_METHOD,
3046 (atomic_read(&dest->conn_flags) &
3047 IP_VS_CONN_F_FWD_MASK)) ||
3048 nla_put_u32(skb, IPVS_DEST_ATTR_WEIGHT,
3049 atomic_read(&dest->weight)) ||
3050 nla_put_u32(skb, IPVS_DEST_ATTR_U_THRESH, dest->u_threshold) ||
3051 nla_put_u32(skb, IPVS_DEST_ATTR_L_THRESH, dest->l_threshold) ||
3052 nla_put_u32(skb, IPVS_DEST_ATTR_ACTIVE_CONNS,
3053 atomic_read(&dest->activeconns)) ||
3054 nla_put_u32(skb, IPVS_DEST_ATTR_INACT_CONNS,
3055 atomic_read(&dest->inactconns)) ||
3056 nla_put_u32(skb, IPVS_DEST_ATTR_PERSIST_CONNS,
3057 atomic_read(&dest->persistconns)))
3058 goto nla_put_failure;
9a812198
JV		 3059 if (ip_vs_genl_fill_stats(skb, IPVS_DEST_ATTR_STATS, &dest->stats))
3060 goto nla_put_failure;
3061
3062 nla_nest_end(skb, nl_dest);
3063
3064 return 0;
3065
3066nla_put_failure:
3067 nla_nest_cancel(skb, nl_dest);
3068 return -EMSGSIZE;
3069}
3070
3071static int ip_vs_genl_dump_dest(struct sk_buff *skb, struct ip_vs_dest *dest,
3072 struct netlink_callback *cb)
3073{
3074 void *hdr;
3075
3076 hdr = genlmsg_put(skb, NETLINK_CB(cb->skb).pid, cb->nlh->nlmsg_seq,
3077 &ip_vs_genl_family, NLM_F_MULTI,
3078 IPVS_CMD_NEW_DEST);
3079 if (!hdr)
3080 return -EMSGSIZE;
3081
3082 if (ip_vs_genl_fill_dest(skb, dest) < 0)
3083 goto nla_put_failure;
3084
3085 return genlmsg_end(skb, hdr);
3086
3087nla_put_failure:
3088 genlmsg_cancel(skb, hdr);
3089 return -EMSGSIZE;
3090}
3091
3092static int ip_vs_genl_dump_dests(struct sk_buff *skb,
3093 struct netlink_callback *cb)
3094{
3095 int idx = 0;
3096 int start = cb->args[0];
3097 struct ip_vs_service *svc;
3098 struct ip_vs_dest *dest;
3099 struct nlattr *attrs[IPVS_CMD_ATTR_MAX + 1];
a0840e2e 3100 struct net *net = skb_sknet(skb);
9a812198
JV		 3101
3102 mutex_lock(&__ip_vs_mutex);
3103
3104 /* Try to find the service for which to dump destinations */
3105 if (nlmsg_parse(cb->nlh, GENL_HDRLEN, attrs,
3106 IPVS_CMD_ATTR_MAX, ip_vs_cmd_policy))
3107 goto out_err;
3108
a0840e2e 3109
fc723250 3110 svc = ip_vs_genl_find_service(net, attrs[IPVS_CMD_ATTR_SERVICE]);
9a812198
JV		 3111 if (IS_ERR(svc) || svc == NULL)
3112 goto out_err;
3113
3114 /* Dump the destinations */
3115 list_for_each_entry(dest, &svc->destinations, n_list) {
3116 if (++idx <= start)
3117 continue;
3118 if (ip_vs_genl_dump_dest(skb, dest, cb) < 0) {
3119 idx--;
3120 goto nla_put_failure;
3121 }
3122 }
3123
3124nla_put_failure:
3125 cb->args[0] = idx;
9a812198
JV		 3126
3127out_err:
3128 mutex_unlock(&__ip_vs_mutex);
3129
3130 return skb->len;
3131}
3132
c860c6b1 3133static int ip_vs_genl_parse_dest(struct ip_vs_dest_user_kern *udest,
9a812198
JV		 3134 struct nlattr *nla, int full_entry)
3135{
3136 struct nlattr *attrs[IPVS_DEST_ATTR_MAX + 1];
3137 struct nlattr *nla_addr, *nla_port;
3138
3139 /* Parse mandatory identifying destination fields first */
3140 if (nla == NULL ||
3141 nla_parse_nested(attrs, IPVS_DEST_ATTR_MAX, nla, ip_vs_dest_policy))
3142 return -EINVAL;
3143
3144 nla_addr = attrs[IPVS_DEST_ATTR_ADDR];
3145 nla_port = attrs[IPVS_DEST_ATTR_PORT];
3146
3147 if (!(nla_addr && nla_port))
3148 return -EINVAL;
3149
258c8893
SH		 3150 memset(udest, 0, sizeof(*udest));
3151
9a812198
JV		 3152 nla_memcpy(&udest->addr, nla_addr, sizeof(udest->addr));
3153 udest->port = nla_get_u16(nla_port);
3154
3155 /* If a full entry was requested, check for the additional fields */
3156 if (full_entry) {
3157 struct nlattr *nla_fwd, *nla_weight, *nla_u_thresh,
3158 *nla_l_thresh;
3159
3160 nla_fwd = attrs[IPVS_DEST_ATTR_FWD_METHOD];
3161 nla_weight = attrs[IPVS_DEST_ATTR_WEIGHT];
3162 nla_u_thresh = attrs[IPVS_DEST_ATTR_U_THRESH];
3163 nla_l_thresh = attrs[IPVS_DEST_ATTR_L_THRESH];
3164
3165 if (!(nla_fwd && nla_weight && nla_u_thresh && nla_l_thresh))
3166 return -EINVAL;
3167
3168 udest->conn_flags = nla_get_u32(nla_fwd)
3169 & IP_VS_CONN_F_FWD_MASK;
3170 udest->weight = nla_get_u32(nla_weight);
3171 udest->u_threshold = nla_get_u32(nla_u_thresh);
3172 udest->l_threshold = nla_get_u32(nla_l_thresh);
3173 }
3174
3175 return 0;
3176}
3177
3178static int ip_vs_genl_fill_daemon(struct sk_buff *skb, __be32 state,
3179 const char *mcast_ifn, __be32 syncid)
3180{
3181 struct nlattr *nl_daemon;
3182
3183 nl_daemon = nla_nest_start(skb, IPVS_CMD_ATTR_DAEMON);
3184 if (!nl_daemon)
3185 return -EMSGSIZE;
3186
969e8e25
DM		 3187 if (nla_put_u32(skb, IPVS_DAEMON_ATTR_STATE, state) ||
3188 nla_put_string(skb, IPVS_DAEMON_ATTR_MCAST_IFN, mcast_ifn) ||
3189 nla_put_u32(skb, IPVS_DAEMON_ATTR_SYNC_ID, syncid))
3190 goto nla_put_failure;
9a812198
JV		 3191 nla_nest_end(skb, nl_daemon);
3192
3193 return 0;
3194
3195nla_put_failure:
3196 nla_nest_cancel(skb, nl_daemon);
3197 return -EMSGSIZE;
3198}
3199
3200static int ip_vs_genl_dump_daemon(struct sk_buff *skb, __be32 state,
3201 const char *mcast_ifn, __be32 syncid,
3202 struct netlink_callback *cb)
3203{
3204 void *hdr;
3205 hdr = genlmsg_put(skb, NETLINK_CB(cb->skb).pid, cb->nlh->nlmsg_seq,
3206 &ip_vs_genl_family, NLM_F_MULTI,
3207 IPVS_CMD_NEW_DAEMON);
3208 if (!hdr)
3209 return -EMSGSIZE;
3210
3211 if (ip_vs_genl_fill_daemon(skb, state, mcast_ifn, syncid))
3212 goto nla_put_failure;
3213
3214 return genlmsg_end(skb, hdr);
3215
3216nla_put_failure:
3217 genlmsg_cancel(skb, hdr);
3218 return -EMSGSIZE;
3219}
3220
3221static int ip_vs_genl_dump_daemons(struct sk_buff *skb,
3222 struct netlink_callback *cb)
3223{
a09d1977 3224 struct net *net = skb_sknet(skb);
f131315f
HS		 3225 struct netns_ipvs *ipvs = net_ipvs(net);
3226
ae1d48b2 3227 mutex_lock(&ipvs->sync_mutex);
f131315f 3228 if ((ipvs->sync_state & IP_VS_STATE_MASTER) && !cb->args[0]) {
9a812198 3229 if (ip_vs_genl_dump_daemon(skb, IP_VS_STATE_MASTER,
f131315f
HS		 3230 ipvs->master_mcast_ifn,
3231 ipvs->master_syncid, cb) < 0)
9a812198
JV		 3232 goto nla_put_failure;
3233
3234 cb->args[0] = 1;
3235 }
3236
f131315f 3237 if ((ipvs->sync_state & IP_VS_STATE_BACKUP) && !cb->args[1]) {
9a812198 3238 if (ip_vs_genl_dump_daemon(skb, IP_VS_STATE_BACKUP,
f131315f
HS		 3239 ipvs->backup_mcast_ifn,
3240 ipvs->backup_syncid, cb) < 0)
9a812198
JV		 3241 goto nla_put_failure;
3242
3243 cb->args[1] = 1;
3244 }
3245
3246nla_put_failure:
ae1d48b2 3247 mutex_unlock(&ipvs->sync_mutex);
9a812198
JV		 3248
3249 return skb->len;
3250}
3251
f131315f 3252static int ip_vs_genl_new_daemon(struct net *net, struct nlattr **attrs)
9a812198
JV		 3253{
3254 if (!(attrs[IPVS_DAEMON_ATTR_STATE] &&
3255 attrs[IPVS_DAEMON_ATTR_MCAST_IFN] &&
3256 attrs[IPVS_DAEMON_ATTR_SYNC_ID]))
3257 return -EINVAL;
3258
f131315f
HS		 3259 return start_sync_thread(net,
3260 nla_get_u32(attrs[IPVS_DAEMON_ATTR_STATE]),
9a812198
JV		 3261 nla_data(attrs[IPVS_DAEMON_ATTR_MCAST_IFN]),
3262 nla_get_u32(attrs[IPVS_DAEMON_ATTR_SYNC_ID]));
3263}
3264
f131315f 3265static int ip_vs_genl_del_daemon(struct net *net, struct nlattr **attrs)
9a812198
JV		 3266{
3267 if (!attrs[IPVS_DAEMON_ATTR_STATE])
3268 return -EINVAL;
3269
f131315f
HS		 3270 return stop_sync_thread(net,
3271 nla_get_u32(attrs[IPVS_DAEMON_ATTR_STATE]));
9a812198
JV		 3272}
3273
9330419d 3274static int ip_vs_genl_set_config(struct net *net, struct nlattr **attrs)
9a812198
JV		 3275{
3276 struct ip_vs_timeout_user t;
3277
9330419d 3278 __ip_vs_get_timeouts(net, &t);
9a812198
JV		 3279
3280 if (attrs[IPVS_CMD_ATTR_TIMEOUT_TCP])
3281 t.tcp_timeout = nla_get_u32(attrs[IPVS_CMD_ATTR_TIMEOUT_TCP]);
3282
3283 if (attrs[IPVS_CMD_ATTR_TIMEOUT_TCP_FIN])
3284 t.tcp_fin_timeout =
3285 nla_get_u32(attrs[IPVS_CMD_ATTR_TIMEOUT_TCP_FIN]);
3286
3287 if (attrs[IPVS_CMD_ATTR_TIMEOUT_UDP])
3288 t.udp_timeout = nla_get_u32(attrs[IPVS_CMD_ATTR_TIMEOUT_UDP]);
3289
9330419d 3290 return ip_vs_set_timeout(net, &t);
9a812198
JV		 3291}
3292
ae1d48b2 3293static int ip_vs_genl_set_daemon(struct sk_buff *skb, struct genl_info *info)
9a812198 3294{
9a812198 3295 int ret = 0, cmd;
fc723250 3296 struct net *net;
a0840e2e 3297 struct netns_ipvs *ipvs;
9a812198 3298
fc723250 3299 net = skb_sknet(skb);
a0840e2e 3300 ipvs = net_ipvs(net);
9a812198
JV		 3301 cmd = info->genlhdr->cmd;
3302
ae1d48b2 3303 if (cmd == IPVS_CMD_NEW_DAEMON || cmd == IPVS_CMD_DEL_DAEMON) {
9a812198
JV		 3304 struct nlattr *daemon_attrs[IPVS_DAEMON_ATTR_MAX + 1];
3305
ae1d48b2 3306 mutex_lock(&ipvs->sync_mutex);
9a812198
JV		 3307 if (!info->attrs[IPVS_CMD_ATTR_DAEMON] ||
3308 nla_parse_nested(daemon_attrs, IPVS_DAEMON_ATTR_MAX,
3309 info->attrs[IPVS_CMD_ATTR_DAEMON],
3310 ip_vs_daemon_policy)) {
3311 ret = -EINVAL;
3312 goto out;
3313 }
3314
3315 if (cmd == IPVS_CMD_NEW_DAEMON)
f131315f 3316 ret = ip_vs_genl_new_daemon(net, daemon_attrs);
9a812198 3317 else
f131315f 3318 ret = ip_vs_genl_del_daemon(net, daemon_attrs);
ae1d48b2
HS		 3319out:
3320 mutex_unlock(&ipvs->sync_mutex);
3321 }
3322 return ret;
3323}
3324
3325static int ip_vs_genl_set_cmd(struct sk_buff *skb, struct genl_info *info)
3326{
3327 struct ip_vs_service *svc = NULL;
3328 struct ip_vs_service_user_kern usvc;
3329 struct ip_vs_dest_user_kern udest;
3330 int ret = 0, cmd;
3331 int need_full_svc = 0, need_full_dest = 0;
3332 struct net *net;
ae1d48b2
HS		 3333
3334 net = skb_sknet(skb);
ae1d48b2
HS		 3335 cmd = info->genlhdr->cmd;
3336
3337 mutex_lock(&__ip_vs_mutex);
3338
3339 if (cmd == IPVS_CMD_FLUSH) {
3340 ret = ip_vs_flush(net);
3341 goto out;
3342 } else if (cmd == IPVS_CMD_SET_CONFIG) {
3343 ret = ip_vs_genl_set_config(net, info->attrs);
9a812198
JV		 3344 goto out;
3345 } else if (cmd == IPVS_CMD_ZERO &&
3346 !info->attrs[IPVS_CMD_ATTR_SERVICE]) {
fc723250 3347 ret = ip_vs_zero_all(net);
9a812198
JV		 3348 goto out;
3349 }
3350
3351 /* All following commands require a service argument, so check if we
3352 * received a valid one. We need a full service specification when
3353 * adding / editing a service. Only identifying members otherwise. */
3354 if (cmd == IPVS_CMD_NEW_SERVICE || cmd == IPVS_CMD_SET_SERVICE)
3355 need_full_svc = 1;
3356
fc723250 3357 ret = ip_vs_genl_parse_service(net, &usvc,
9a812198 3358 info->attrs[IPVS_CMD_ATTR_SERVICE],
26c15cfd 3359 need_full_svc, &svc);
9a812198
JV		 3360 if (ret)
3361 goto out;
3362
9a812198
JV		 3363 /* Unless we're adding a new service, the service must already exist */
3364 if ((cmd != IPVS_CMD_NEW_SERVICE) && (svc == NULL)) {
3365 ret = -ESRCH;
3366 goto out;
3367 }
3368
3369 /* Destination commands require a valid destination argument. For
3370 * adding / editing a destination, we need a full destination
3371 * specification. */
3372 if (cmd == IPVS_CMD_NEW_DEST || cmd == IPVS_CMD_SET_DEST ||
3373 cmd == IPVS_CMD_DEL_DEST) {
3374 if (cmd != IPVS_CMD_DEL_DEST)
3375 need_full_dest = 1;
3376
3377 ret = ip_vs_genl_parse_dest(&udest,
3378 info->attrs[IPVS_CMD_ATTR_DEST],
3379 need_full_dest);
3380 if (ret)
3381 goto out;
3382 }
3383
3384 switch (cmd) {
3385 case IPVS_CMD_NEW_SERVICE:
3386 if (svc == NULL)
fc723250 3387 ret = ip_vs_add_service(net, &usvc, &svc);
9a812198
JV		 3388 else
3389 ret = -EEXIST;
3390 break;
3391 case IPVS_CMD_SET_SERVICE:
3392 ret = ip_vs_edit_service(svc, &usvc);
3393 break;
3394 case IPVS_CMD_DEL_SERVICE:
3395 ret = ip_vs_del_service(svc);
26c15cfd 3396 /* do not use svc, it can be freed */
9a812198
JV		 3397 break;
3398 case IPVS_CMD_NEW_DEST:
3399 ret = ip_vs_add_dest(svc, &udest);
3400 break;
3401 case IPVS_CMD_SET_DEST:
3402 ret = ip_vs_edit_dest(svc, &udest);
3403 break;
3404 case IPVS_CMD_DEL_DEST:
3405 ret = ip_vs_del_dest(svc, &udest);
3406 break;
3407 case IPVS_CMD_ZERO:
3408 ret = ip_vs_zero_service(svc);
3409 break;
3410 default:
3411 ret = -EINVAL;
3412 }
3413
3414out:
9a812198
JV		 3415 mutex_unlock(&__ip_vs_mutex);
3416
3417 return ret;
3418}
3419
3420static int ip_vs_genl_get_cmd(struct sk_buff *skb, struct genl_info *info)
3421{
3422 struct sk_buff *msg;
3423 void *reply;
3424 int ret, cmd, reply_cmd;
fc723250 3425 struct net *net;
9a812198 3426
fc723250 3427 net = skb_sknet(skb);
9a812198
JV		 3428 cmd = info->genlhdr->cmd;
3429
3430 if (cmd == IPVS_CMD_GET_SERVICE)
3431 reply_cmd = IPVS_CMD_NEW_SERVICE;
3432 else if (cmd == IPVS_CMD_GET_INFO)
3433 reply_cmd = IPVS_CMD_SET_INFO;
3434 else if (cmd == IPVS_CMD_GET_CONFIG)
3435 reply_cmd = IPVS_CMD_SET_CONFIG;
3436 else {
1e3e238e 3437 pr_err("unknown Generic Netlink command\n");
9a812198
JV		 3438 return -EINVAL;
3439 }
3440
3441 msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
3442 if (!msg)
3443 return -ENOMEM;
3444
3445 mutex_lock(&__ip_vs_mutex);
3446
3447 reply = genlmsg_put_reply(msg, info, &ip_vs_genl_family, 0, reply_cmd);
3448 if (reply == NULL)
3449 goto nla_put_failure;
3450
3451 switch (cmd) {
3452 case IPVS_CMD_GET_SERVICE:
3453 {
3454 struct ip_vs_service *svc;
3455
fc723250
HS		 3456 svc = ip_vs_genl_find_service(net,
3457 info->attrs[IPVS_CMD_ATTR_SERVICE]);
9a812198
JV		 3458 if (IS_ERR(svc)) {
3459 ret = PTR_ERR(svc);
3460 goto out_err;
3461 } else if (svc) {
3462 ret = ip_vs_genl_fill_service(msg, svc);
9a812198
JV		 3463 if (ret)
3464 goto nla_put_failure;
3465 } else {
3466 ret = -ESRCH;
3467 goto out_err;
3468 }
3469
3470 break;
3471 }
3472
3473 case IPVS_CMD_GET_CONFIG:
3474 {
3475 struct ip_vs_timeout_user t;
3476
9330419d 3477 __ip_vs_get_timeouts(net, &t);
9a812198 3478#ifdef CONFIG_IP_VS_PROTO_TCP
969e8e25
DM		 3479 if (nla_put_u32(msg, IPVS_CMD_ATTR_TIMEOUT_TCP,
3480 t.tcp_timeout) ||
3481 nla_put_u32(msg, IPVS_CMD_ATTR_TIMEOUT_TCP_FIN,
3482 t.tcp_fin_timeout))
3483 goto nla_put_failure;
9a812198
JV		 3484#endif
3485#ifdef CONFIG_IP_VS_PROTO_UDP
969e8e25
DM		 3486 if (nla_put_u32(msg, IPVS_CMD_ATTR_TIMEOUT_UDP, t.udp_timeout))
3487 goto nla_put_failure;
9a812198
JV		 3488#endif
3489
3490 break;
3491 }
3492
3493 case IPVS_CMD_GET_INFO:
969e8e25
DM		 3494 if (nla_put_u32(msg, IPVS_INFO_ATTR_VERSION,
3495 IP_VS_VERSION_CODE) ||
3496 nla_put_u32(msg, IPVS_INFO_ATTR_CONN_TAB_SIZE,
3497 ip_vs_conn_tab_size))
3498 goto nla_put_failure;
9a812198
JV		 3499 break;
3500 }
3501
3502 genlmsg_end(msg, reply);
134e6375 3503 ret = genlmsg_reply(msg, info);
9a812198
JV		 3504 goto out;
3505
3506nla_put_failure:
1e3e238e 3507 pr_err("not enough space in Netlink message\n");
9a812198
JV		 3508 ret = -EMSGSIZE;
3509
3510out_err:
3511 nlmsg_free(msg);
3512out:
3513 mutex_unlock(&__ip_vs_mutex);
3514
3515 return ret;
3516}
3517
3518
3519static struct genl_ops ip_vs_genl_ops[] __read_mostly = {
3520 {
3521 .cmd = IPVS_CMD_NEW_SERVICE,
3522 .flags = GENL_ADMIN_PERM,
3523 .policy = ip_vs_cmd_policy,
3524 .doit = ip_vs_genl_set_cmd,
3525 },
3526 {
3527 .cmd = IPVS_CMD_SET_SERVICE,
3528 .flags = GENL_ADMIN_PERM,
3529 .policy = ip_vs_cmd_policy,
3530 .doit = ip_vs_genl_set_cmd,
3531 },
3532 {
3533 .cmd = IPVS_CMD_DEL_SERVICE,
3534 .flags = GENL_ADMIN_PERM,
3535 .policy = ip_vs_cmd_policy,
3536 .doit = ip_vs_genl_set_cmd,
3537 },
3538 {
3539 .cmd = IPVS_CMD_GET_SERVICE,
3540 .flags = GENL_ADMIN_PERM,
3541 .doit = ip_vs_genl_get_cmd,
3542 .dumpit = ip_vs_genl_dump_services,
3543 .policy = ip_vs_cmd_policy,
3544 },
3545 {
3546 .cmd = IPVS_CMD_NEW_DEST,
3547 .flags = GENL_ADMIN_PERM,
3548 .policy = ip_vs_cmd_policy,
3549 .doit = ip_vs_genl_set_cmd,
3550 },
3551 {
3552 .cmd = IPVS_CMD_SET_DEST,
3553 .flags = GENL_ADMIN_PERM,
3554 .policy = ip_vs_cmd_policy,
3555 .doit = ip_vs_genl_set_cmd,
3556 },
3557 {
3558 .cmd = IPVS_CMD_DEL_DEST,
3559 .flags = GENL_ADMIN_PERM,
3560 .policy = ip_vs_cmd_policy,
3561 .doit = ip_vs_genl_set_cmd,
3562 },
3563 {
3564 .cmd = IPVS_CMD_GET_DEST,
3565 .flags = GENL_ADMIN_PERM,
3566 .policy = ip_vs_cmd_policy,
3567 .dumpit = ip_vs_genl_dump_dests,
3568 },
3569 {
3570 .cmd = IPVS_CMD_NEW_DAEMON,
3571 .flags = GENL_ADMIN_PERM,
3572 .policy = ip_vs_cmd_policy,
ae1d48b2 3573 .doit = ip_vs_genl_set_daemon,
9a812198
JV		 3574 },
3575 {
3576 .cmd = IPVS_CMD_DEL_DAEMON,
3577 .flags = GENL_ADMIN_PERM,
3578 .policy = ip_vs_cmd_policy,
ae1d48b2 3579 .doit = ip_vs_genl_set_daemon,
9a812198
JV		 3580 },
3581 {
3582 .cmd = IPVS_CMD_GET_DAEMON,
3583 .flags = GENL_ADMIN_PERM,
3584 .dumpit = ip_vs_genl_dump_daemons,
3585 },
3586 {
3587 .cmd = IPVS_CMD_SET_CONFIG,
3588 .flags = GENL_ADMIN_PERM,
3589 .policy = ip_vs_cmd_policy,
3590 .doit = ip_vs_genl_set_cmd,
3591 },
3592 {
3593 .cmd = IPVS_CMD_GET_CONFIG,
3594 .flags = GENL_ADMIN_PERM,
3595 .doit = ip_vs_genl_get_cmd,
3596 },
3597 {
3598 .cmd = IPVS_CMD_GET_INFO,
3599 .flags = GENL_ADMIN_PERM,
3600 .doit = ip_vs_genl_get_cmd,
3601 },
3602 {
3603 .cmd = IPVS_CMD_ZERO,
3604 .flags = GENL_ADMIN_PERM,
3605 .policy = ip_vs_cmd_policy,
3606 .doit = ip_vs_genl_set_cmd,
3607 },
3608 {
3609 .cmd = IPVS_CMD_FLUSH,
3610 .flags = GENL_ADMIN_PERM,
3611 .doit = ip_vs_genl_set_cmd,
3612 },
3613};
3614
3615static int __init ip_vs_genl_register(void)
3616{
8f698d54
MM		 3617 return genl_register_family_with_ops(&ip_vs_genl_family,
3618 ip_vs_genl_ops, ARRAY_SIZE(ip_vs_genl_ops));
9a812198
JV		 3619}
3620
3621static void ip_vs_genl_unregister(void)
3622{
3623 genl_unregister_family(&ip_vs_genl_family);
3624}
3625
3626/* End of Generic Netlink interface definitions */
3627
61b1ab45
HS		 3628/*
3629 * per netns intit/exit func.
3630 */
14e40546 3631#ifdef CONFIG_SYSCTL
503cf15a 3632int __net_init ip_vs_control_net_init_sysctl(struct net *net)
61b1ab45 3633{
fc723250
HS		 3634 int idx;
3635 struct netns_ipvs *ipvs = net_ipvs(net);
a0840e2e 3636 struct ctl_table *tbl;
fc723250 3637
a0840e2e
HS		 3638 atomic_set(&ipvs->dropentry, 0);
3639 spin_lock_init(&ipvs->dropentry_lock);
3640 spin_lock_init(&ipvs->droppacket_lock);
3641 spin_lock_init(&ipvs->securetcp_lock);
a0840e2e
HS		 3642
3643 if (!net_eq(net, &init_net)) {
3644 tbl = kmemdup(vs_vars, sizeof(vs_vars), GFP_KERNEL);
3645 if (tbl == NULL)
14e40546 3646 return -ENOMEM;
a0840e2e
HS		 3647 } else
3648 tbl = vs_vars;
3649 /* Initialize sysctl defaults */
3650 idx = 0;
3651 ipvs->sysctl_amemthresh = 1024;
3652 tbl[idx++].data = &ipvs->sysctl_amemthresh;
3653 ipvs->sysctl_am_droprate = 10;
3654 tbl[idx++].data = &ipvs->sysctl_am_droprate;
3655 tbl[idx++].data = &ipvs->sysctl_drop_entry;
3656 tbl[idx++].data = &ipvs->sysctl_drop_packet;
3657#ifdef CONFIG_IP_VS_NFCT
3658 tbl[idx++].data = &ipvs->sysctl_conntrack;
3659#endif
3660 tbl[idx++].data = &ipvs->sysctl_secure_tcp;
3661 ipvs->sysctl_snat_reroute = 1;
3662 tbl[idx++].data = &ipvs->sysctl_snat_reroute;
3663 ipvs->sysctl_sync_ver = 1;
3664 tbl[idx++].data = &ipvs->sysctl_sync_ver;
3665 tbl[idx++].data = &ipvs->sysctl_cache_bypass;
3666 tbl[idx++].data = &ipvs->sysctl_expire_nodest_conn;
3667 tbl[idx++].data = &ipvs->sysctl_expire_quiescent_template;
59e0350e
SH		 3668 ipvs->sysctl_sync_threshold[0] = DEFAULT_SYNC_THRESHOLD;
3669 ipvs->sysctl_sync_threshold[1] = DEFAULT_SYNC_PERIOD;
a0840e2e
HS		 3670 tbl[idx].data = &ipvs->sysctl_sync_threshold;
3671 tbl[idx++].maxlen = sizeof(ipvs->sysctl_sync_threshold);
3672 tbl[idx++].data = &ipvs->sysctl_nat_icmp_send;
3673
3674
3675 ipvs->sysctl_hdr = register_net_sysctl_table(net, net_vs_ctl_path,
07924709 3676 tbl);
0443929f
SH		 3677 if (ipvs->sysctl_hdr == NULL) {
3678 if (!net_eq(net, &init_net))
3679 kfree(tbl);
14e40546 3680 return -ENOMEM;
0443929f 3681 }
6ef757f9 3682 ip_vs_start_estimator(net, &ipvs->tot_stats);
a0840e2e 3683 ipvs->sysctl_tbl = tbl;
f6340ee0
HS		 3684 /* Schedule defense work */
3685 INIT_DELAYED_WORK(&ipvs->defense_work, defense_work_handler);
3686 schedule_delayed_work(&ipvs->defense_work, DEFENSE_TIMER_PERIOD);
61b1ab45 3687
61b1ab45 3688 return 0;
61b1ab45
HS		 3689}
3690
503cf15a 3691void __net_init ip_vs_control_net_cleanup_sysctl(struct net *net)
61b1ab45 3692{
b17fc996
HS		 3693 struct netns_ipvs *ipvs = net_ipvs(net);
3694
f2431e6e
HS		 3695 cancel_delayed_work_sync(&ipvs->defense_work);
3696 cancel_work_sync(&ipvs->defense_work.work);
a0840e2e 3697 unregister_net_sysctl_table(ipvs->sysctl_hdr);
14e40546
SH		 3698}
3699
3700#else
3701
503cf15a
HS		 3702int __net_init ip_vs_control_net_init_sysctl(struct net *net) { return 0; }
3703void __net_init ip_vs_control_net_cleanup_sysctl(struct net *net) { }
14e40546 3704
0443929f 3705#endif
14e40546 3706
7a4f0761
HS		 3707static struct notifier_block ip_vs_dst_notifier = {
3708 .notifier_call = ip_vs_dst_event,
3709};
3710
503cf15a 3711int __net_init ip_vs_control_net_init(struct net *net)
14e40546
SH		 3712{
3713 int idx;
3714 struct netns_ipvs *ipvs = net_ipvs(net);
3715
3458e21c 3716 rwlock_init(&ipvs->rs_lock);
14e40546
SH		 3717
3718 /* Initialize rs_table */
3719 for (idx = 0; idx < IP_VS_RTAB_SIZE; idx++)
3720 INIT_LIST_HEAD(&ipvs->rs_table[idx]);
3721
3722 INIT_LIST_HEAD(&ipvs->dest_trash);
3723 atomic_set(&ipvs->ftpsvc_counter, 0);
3724 atomic_set(&ipvs->nullsvc_counter, 0);
3725
3726 /* procfs stats */
3727 ipvs->tot_stats.cpustats = alloc_percpu(struct ip_vs_cpu_stats);
0a9ee813 3728 if (!ipvs->tot_stats.cpustats)
14e40546 3729 return -ENOMEM;
0a9ee813 3730
14e40546
SH		 3731 spin_lock_init(&ipvs->tot_stats.lock);
3732
3733 proc_net_fops_create(net, "ip_vs", 0, &ip_vs_info_fops);
3734 proc_net_fops_create(net, "ip_vs_stats", 0, &ip_vs_stats_fops);
3735 proc_net_fops_create(net, "ip_vs_stats_percpu", 0,
3736 &ip_vs_stats_percpu_fops);
3737
503cf15a 3738 if (ip_vs_control_net_init_sysctl(net))
14e40546
SH		 3739 goto err;
3740
3741 return 0;
3742
3743err:
2a0751af 3744 free_percpu(ipvs->tot_stats.cpustats);
61b1ab45
HS		 3745 return -ENOMEM;
3746}
3747
503cf15a 3748void __net_exit ip_vs_control_net_cleanup(struct net *net)
61b1ab45 3749{
b17fc996
HS		 3750 struct netns_ipvs *ipvs = net_ipvs(net);
3751
f2431e6e 3752 ip_vs_trash_cleanup(net);
6ef757f9 3753 ip_vs_stop_estimator(net, &ipvs->tot_stats);
503cf15a 3754 ip_vs_control_net_cleanup_sysctl(net);
b17fc996 3755 proc_net_remove(net, "ip_vs_stats_percpu");
61b1ab45
HS		 3756 proc_net_remove(net, "ip_vs_stats");
3757 proc_net_remove(net, "ip_vs");
2a0751af 3758 free_percpu(ipvs->tot_stats.cpustats);
61b1ab45
HS		 3759}
3760
048cf48b 3761int __init ip_vs_control_init(void)
1da177e4 3762{
1da177e4 3763 int idx;
fc723250 3764 int ret;
1da177e4
LT		 3765
3766 EnterFunction(2);
3767
fc723250 3768 /* Initialize svc_table, ip_vs_svc_fwm_table, rs_table */
d86bef73
EB		 3769 for(idx = 0; idx < IP_VS_SVC_TAB_SIZE; idx++) {
3770 INIT_LIST_HEAD(&ip_vs_svc_table[idx]);
3771 INIT_LIST_HEAD(&ip_vs_svc_fwm_table[idx]);
3772 }
fc723250 3773
fc723250 3774 smp_wmb(); /* Do we really need it now ? */
d86bef73 3775
1da177e4
LT		 3776 ret = nf_register_sockopt(&ip_vs_sockopts);
3777 if (ret) {
1e3e238e 3778 pr_err("cannot register sockopt.\n");
7a4f0761 3779 goto err_sock;
1da177e4
LT		 3780 }
3781
9a812198
JV		 3782 ret = ip_vs_genl_register();
3783 if (ret) {
1e3e238e 3784 pr_err("cannot register Generic Netlink interface.\n");
7a4f0761 3785 goto err_genl;
9a812198
JV		 3786 }
3787
7a4f0761
HS		 3788 ret = register_netdevice_notifier(&ip_vs_dst_notifier);
3789 if (ret < 0)
3790 goto err_notf;
3791
1da177e4
LT		 3792 LeaveFunction(2);
3793 return 0;
fc723250 3794
7a4f0761
HS		 3795err_notf:
3796 ip_vs_genl_unregister();
3797err_genl:
3798 nf_unregister_sockopt(&ip_vs_sockopts);
3799err_sock:
fc723250 3800 return ret;
1da177e4
LT		 3801}
3802
3803
3804void ip_vs_control_cleanup(void)
3805{
3806 EnterFunction(2);
7676e345 3807 unregister_netdevice_notifier(&ip_vs_dst_notifier);
9a812198 3808 ip_vs_genl_unregister();
1da177e4
LT		 3809 nf_unregister_sockopt(&ip_vs_sockopts);
3810 LeaveFunction(2);
3811}
Linux kernel - Deliverables
This page took 0.85903 seconds and 5 git commands to generate.