xref: /linux/net/netfilter/ipvs/ip_vs_ctl.c (revision a1ff5a7d78a036d6c2178ee5acd6ba4946243800)
1 // SPDX-License-Identifier: GPL-2.0-or-later
2 /*
3  * IPVS         An implementation of the IP virtual server support for the
4  *              LINUX operating system.  IPVS is now implemented as a module
5  *              over the NetFilter framework. IPVS can be used to build a
6  *              high-performance and highly available server based on a
7  *              cluster of servers.
8  *
9  * Authors:     Wensong Zhang <wensong@linuxvirtualserver.org>
10  *              Peter Kese <peter.kese@ijs.si>
11  *              Julian Anastasov <ja@ssi.bg>
12  *
13  * Changes:
14  */
15 
16 #define KMSG_COMPONENT "IPVS"
17 #define pr_fmt(fmt) KMSG_COMPONENT ": " fmt
18 
19 #include <linux/module.h>
20 #include <linux/init.h>
21 #include <linux/types.h>
22 #include <linux/capability.h>
23 #include <linux/fs.h>
24 #include <linux/sysctl.h>
25 #include <linux/proc_fs.h>
26 #include <linux/workqueue.h>
27 #include <linux/seq_file.h>
28 #include <linux/slab.h>
29 
30 #include <linux/netfilter.h>
31 #include <linux/netfilter_ipv4.h>
32 #include <linux/mutex.h>
33 
34 #include <net/net_namespace.h>
35 #include <linux/nsproxy.h>
36 #include <net/ip.h>
37 #ifdef CONFIG_IP_VS_IPV6
38 #include <net/ipv6.h>
39 #include <net/ip6_route.h>
40 #include <net/netfilter/ipv6/nf_defrag_ipv6.h>
41 #endif
42 #include <net/route.h>
43 #include <net/sock.h>
44 #include <net/genetlink.h>
45 
46 #include <linux/uaccess.h>
47 
48 #include <net/ip_vs.h>
49 
50 MODULE_ALIAS_GENL_FAMILY(IPVS_GENL_NAME);
51 
52 DEFINE_MUTEX(__ip_vs_mutex); /* Serialize configuration with sockopt/netlink */
53 
54 /* sysctl variables */
55 
56 #ifdef CONFIG_IP_VS_DEBUG
57 static int sysctl_ip_vs_debug_level = 0;
58 
ip_vs_get_debug_level(void)59 int ip_vs_get_debug_level(void)
60 {
61 	return sysctl_ip_vs_debug_level;
62 }
63 #endif
64 
65 
66 /*  Protos */
67 static void __ip_vs_del_service(struct ip_vs_service *svc, bool cleanup);
68 
69 
70 #ifdef CONFIG_IP_VS_IPV6
71 /* Taken from rt6_fill_node() in net/ipv6/route.c, is there a better way? */
__ip_vs_addr_is_local_v6(struct net * net,const struct in6_addr * addr)72 static bool __ip_vs_addr_is_local_v6(struct net *net,
73 				     const struct in6_addr *addr)
74 {
75 	struct flowi6 fl6 = {
76 		.daddr = *addr,
77 	};
78 	struct dst_entry *dst = ip6_route_output(net, NULL, &fl6);
79 	bool is_local;
80 
81 	is_local = !dst->error && dst->dev && (dst->dev->flags & IFF_LOOPBACK);
82 
83 	dst_release(dst);
84 	return is_local;
85 }
86 #endif
87 
88 #ifdef CONFIG_SYSCTL
89 /*
90  *	update_defense_level is called from keventd and from sysctl,
91  *	so it needs to protect itself from softirqs
92  */
update_defense_level(struct netns_ipvs * ipvs)93 static void update_defense_level(struct netns_ipvs *ipvs)
94 {
95 	struct sysinfo i;
96 	int availmem;
97 	int amemthresh;
98 	int nomem;
99 	int to_change = -1;
100 
101 	/* we only count free and buffered memory (in pages) */
102 	si_meminfo(&i);
103 	availmem = i.freeram + i.bufferram;
104 	/* however in linux 2.5 the i.bufferram is total page cache size,
105 	   we need adjust it */
106 	/* si_swapinfo(&i); */
107 	/* availmem = availmem - (i.totalswap - i.freeswap); */
108 
109 	amemthresh = max(READ_ONCE(ipvs->sysctl_amemthresh), 0);
110 	nomem = (availmem < amemthresh);
111 
112 	local_bh_disable();
113 
114 	/* drop_entry */
115 	spin_lock(&ipvs->dropentry_lock);
116 	switch (ipvs->sysctl_drop_entry) {
117 	case 0:
118 		atomic_set(&ipvs->dropentry, 0);
119 		break;
120 	case 1:
121 		if (nomem) {
122 			atomic_set(&ipvs->dropentry, 1);
123 			ipvs->sysctl_drop_entry = 2;
124 		} else {
125 			atomic_set(&ipvs->dropentry, 0);
126 		}
127 		break;
128 	case 2:
129 		if (nomem) {
130 			atomic_set(&ipvs->dropentry, 1);
131 		} else {
132 			atomic_set(&ipvs->dropentry, 0);
133 			ipvs->sysctl_drop_entry = 1;
134 		}
135 		break;
136 	case 3:
137 		atomic_set(&ipvs->dropentry, 1);
138 		break;
139 	}
140 	spin_unlock(&ipvs->dropentry_lock);
141 
142 	/* drop_packet */
143 	spin_lock(&ipvs->droppacket_lock);
144 	switch (ipvs->sysctl_drop_packet) {
145 	case 0:
146 		ipvs->drop_rate = 0;
147 		break;
148 	case 1:
149 		if (nomem) {
150 			ipvs->drop_counter = amemthresh / (amemthresh - availmem);
151 			ipvs->drop_rate = ipvs->drop_counter;
152 			ipvs->sysctl_drop_packet = 2;
153 		} else {
154 			ipvs->drop_rate = 0;
155 		}
156 		break;
157 	case 2:
158 		if (nomem) {
159 			ipvs->drop_counter = amemthresh / (amemthresh - availmem);
160 			ipvs->drop_rate = ipvs->drop_counter;
161 		} else {
162 			ipvs->drop_rate = 0;
163 			ipvs->sysctl_drop_packet = 1;
164 		}
165 		break;
166 	case 3:
167 		ipvs->drop_rate = ipvs->sysctl_am_droprate;
168 		break;
169 	}
170 	spin_unlock(&ipvs->droppacket_lock);
171 
172 	/* secure_tcp */
173 	spin_lock(&ipvs->securetcp_lock);
174 	switch (ipvs->sysctl_secure_tcp) {
175 	case 0:
176 		if (ipvs->old_secure_tcp >= 2)
177 			to_change = 0;
178 		break;
179 	case 1:
180 		if (nomem) {
181 			if (ipvs->old_secure_tcp < 2)
182 				to_change = 1;
183 			ipvs->sysctl_secure_tcp = 2;
184 		} else {
185 			if (ipvs->old_secure_tcp >= 2)
186 				to_change = 0;
187 		}
188 		break;
189 	case 2:
190 		if (nomem) {
191 			if (ipvs->old_secure_tcp < 2)
192 				to_change = 1;
193 		} else {
194 			if (ipvs->old_secure_tcp >= 2)
195 				to_change = 0;
196 			ipvs->sysctl_secure_tcp = 1;
197 		}
198 		break;
199 	case 3:
200 		if (ipvs->old_secure_tcp < 2)
201 			to_change = 1;
202 		break;
203 	}
204 	ipvs->old_secure_tcp = ipvs->sysctl_secure_tcp;
205 	if (to_change >= 0)
206 		ip_vs_protocol_timeout_change(ipvs,
207 					      ipvs->sysctl_secure_tcp > 1);
208 	spin_unlock(&ipvs->securetcp_lock);
209 
210 	local_bh_enable();
211 }
212 
213 /* Handler for delayed work for expiring no
214  * destination connections
215  */
expire_nodest_conn_handler(struct work_struct * work)216 static void expire_nodest_conn_handler(struct work_struct *work)
217 {
218 	struct netns_ipvs *ipvs;
219 
220 	ipvs = container_of(work, struct netns_ipvs,
221 			    expire_nodest_conn_work.work);
222 	ip_vs_expire_nodest_conn_flush(ipvs);
223 }
224 
225 /*
226  *	Timer for checking the defense
227  */
228 #define DEFENSE_TIMER_PERIOD	1*HZ
229 
defense_work_handler(struct work_struct * work)230 static void defense_work_handler(struct work_struct *work)
231 {
232 	struct netns_ipvs *ipvs =
233 		container_of(work, struct netns_ipvs, defense_work.work);
234 
235 	update_defense_level(ipvs);
236 	if (atomic_read(&ipvs->dropentry))
237 		ip_vs_random_dropentry(ipvs);
238 	queue_delayed_work(system_long_wq, &ipvs->defense_work,
239 			   DEFENSE_TIMER_PERIOD);
240 }
241 #endif
242 
est_reload_work_handler(struct work_struct * work)243 static void est_reload_work_handler(struct work_struct *work)
244 {
245 	struct netns_ipvs *ipvs =
246 		container_of(work, struct netns_ipvs, est_reload_work.work);
247 	int genid_done = atomic_read(&ipvs->est_genid_done);
248 	unsigned long delay = HZ / 10;	/* repeat startups after failure */
249 	bool repeat = false;
250 	int genid;
251 	int id;
252 
253 	mutex_lock(&ipvs->est_mutex);
254 	genid = atomic_read(&ipvs->est_genid);
255 	for (id = 0; id < ipvs->est_kt_count; id++) {
256 		struct ip_vs_est_kt_data *kd = ipvs->est_kt_arr[id];
257 
258 		/* netns clean up started, abort delayed work */
259 		if (!ipvs->enable)
260 			goto unlock;
261 		if (!kd)
262 			continue;
263 		/* New config ? Stop kthread tasks */
264 		if (genid != genid_done)
265 			ip_vs_est_kthread_stop(kd);
266 		if (!kd->task && !ip_vs_est_stopped(ipvs)) {
267 			/* Do not start kthreads above 0 in calc phase */
268 			if ((!id || !ipvs->est_calc_phase) &&
269 			    ip_vs_est_kthread_start(ipvs, kd) < 0)
270 				repeat = true;
271 		}
272 	}
273 
274 	atomic_set(&ipvs->est_genid_done, genid);
275 
276 	if (repeat)
277 		queue_delayed_work(system_long_wq, &ipvs->est_reload_work,
278 				   delay);
279 
280 unlock:
281 	mutex_unlock(&ipvs->est_mutex);
282 }
283 
284 int
ip_vs_use_count_inc(void)285 ip_vs_use_count_inc(void)
286 {
287 	return try_module_get(THIS_MODULE);
288 }
289 
290 void
ip_vs_use_count_dec(void)291 ip_vs_use_count_dec(void)
292 {
293 	module_put(THIS_MODULE);
294 }
295 
296 
297 /*
298  *	Hash table: for virtual service lookups
299  */
300 #define IP_VS_SVC_TAB_BITS 8
301 #define IP_VS_SVC_TAB_SIZE (1 << IP_VS_SVC_TAB_BITS)
302 #define IP_VS_SVC_TAB_MASK (IP_VS_SVC_TAB_SIZE - 1)
303 
304 /* the service table hashed by <protocol, addr, port> */
305 static struct hlist_head ip_vs_svc_table[IP_VS_SVC_TAB_SIZE];
306 /* the service table hashed by fwmark */
307 static struct hlist_head ip_vs_svc_fwm_table[IP_VS_SVC_TAB_SIZE];
308 
309 
310 /*
311  *	Returns hash value for virtual service
312  */
313 static inline unsigned int
ip_vs_svc_hashkey(struct netns_ipvs * ipvs,int af,unsigned int proto,const union nf_inet_addr * addr,__be16 port)314 ip_vs_svc_hashkey(struct netns_ipvs *ipvs, int af, unsigned int proto,
315 		  const union nf_inet_addr *addr, __be16 port)
316 {
317 	unsigned int porth = ntohs(port);
318 	__be32 addr_fold = addr->ip;
319 	__u32 ahash;
320 
321 #ifdef CONFIG_IP_VS_IPV6
322 	if (af == AF_INET6)
323 		addr_fold = addr->ip6[0]^addr->ip6[1]^
324 			    addr->ip6[2]^addr->ip6[3];
325 #endif
326 	ahash = ntohl(addr_fold);
327 	ahash ^= ((size_t) ipvs >> 8);
328 
329 	return (proto ^ ahash ^ (porth >> IP_VS_SVC_TAB_BITS) ^ porth) &
330 	       IP_VS_SVC_TAB_MASK;
331 }
332 
333 /*
334  *	Returns hash value of fwmark for virtual service lookup
335  */
ip_vs_svc_fwm_hashkey(struct netns_ipvs * ipvs,__u32 fwmark)336 static inline unsigned int ip_vs_svc_fwm_hashkey(struct netns_ipvs *ipvs, __u32 fwmark)
337 {
338 	return (((size_t)ipvs>>8) ^ fwmark) & IP_VS_SVC_TAB_MASK;
339 }
340 
341 /*
342  *	Hashes a service in the ip_vs_svc_table by <netns,proto,addr,port>
343  *	or in the ip_vs_svc_fwm_table by fwmark.
344  *	Should be called with locked tables.
345  */
ip_vs_svc_hash(struct ip_vs_service * svc)346 static int ip_vs_svc_hash(struct ip_vs_service *svc)
347 {
348 	unsigned int hash;
349 
350 	if (svc->flags & IP_VS_SVC_F_HASHED) {
351 		pr_err("%s(): request for already hashed, called from %pS\n",
352 		       __func__, __builtin_return_address(0));
353 		return 0;
354 	}
355 
356 	if (svc->fwmark == 0) {
357 		/*
358 		 *  Hash it by <netns,protocol,addr,port> in ip_vs_svc_table
359 		 */
360 		hash = ip_vs_svc_hashkey(svc->ipvs, svc->af, svc->protocol,
361 					 &svc->addr, svc->port);
362 		hlist_add_head_rcu(&svc->s_list, &ip_vs_svc_table[hash]);
363 	} else {
364 		/*
365 		 *  Hash it by fwmark in svc_fwm_table
366 		 */
367 		hash = ip_vs_svc_fwm_hashkey(svc->ipvs, svc->fwmark);
368 		hlist_add_head_rcu(&svc->f_list, &ip_vs_svc_fwm_table[hash]);
369 	}
370 
371 	svc->flags |= IP_VS_SVC_F_HASHED;
372 	/* increase its refcnt because it is referenced by the svc table */
373 	atomic_inc(&svc->refcnt);
374 	return 1;
375 }
376 
377 
378 /*
379  *	Unhashes a service from svc_table / svc_fwm_table.
380  *	Should be called with locked tables.
381  */
ip_vs_svc_unhash(struct ip_vs_service * svc)382 static int ip_vs_svc_unhash(struct ip_vs_service *svc)
383 {
384 	if (!(svc->flags & IP_VS_SVC_F_HASHED)) {
385 		pr_err("%s(): request for unhash flagged, called from %pS\n",
386 		       __func__, __builtin_return_address(0));
387 		return 0;
388 	}
389 
390 	if (svc->fwmark == 0) {
391 		/* Remove it from the svc_table table */
392 		hlist_del_rcu(&svc->s_list);
393 	} else {
394 		/* Remove it from the svc_fwm_table table */
395 		hlist_del_rcu(&svc->f_list);
396 	}
397 
398 	svc->flags &= ~IP_VS_SVC_F_HASHED;
399 	atomic_dec(&svc->refcnt);
400 	return 1;
401 }
402 
403 
404 /*
405  *	Get service by {netns, proto,addr,port} in the service table.
406  */
407 static inline struct ip_vs_service *
__ip_vs_service_find(struct netns_ipvs * ipvs,int af,__u16 protocol,const union nf_inet_addr * vaddr,__be16 vport)408 __ip_vs_service_find(struct netns_ipvs *ipvs, int af, __u16 protocol,
409 		     const union nf_inet_addr *vaddr, __be16 vport)
410 {
411 	unsigned int hash;
412 	struct ip_vs_service *svc;
413 
414 	/* Check for "full" addressed entries */
415 	hash = ip_vs_svc_hashkey(ipvs, af, protocol, vaddr, vport);
416 
417 	hlist_for_each_entry_rcu(svc, &ip_vs_svc_table[hash], s_list) {
418 		if ((svc->af == af)
419 		    && ip_vs_addr_equal(af, &svc->addr, vaddr)
420 		    && (svc->port == vport)
421 		    && (svc->protocol == protocol)
422 		    && (svc->ipvs == ipvs)) {
423 			/* HIT */
424 			return svc;
425 		}
426 	}
427 
428 	return NULL;
429 }
430 
431 
432 /*
433  *	Get service by {fwmark} in the service table.
434  */
435 static inline struct ip_vs_service *
__ip_vs_svc_fwm_find(struct netns_ipvs * ipvs,int af,__u32 fwmark)436 __ip_vs_svc_fwm_find(struct netns_ipvs *ipvs, int af, __u32 fwmark)
437 {
438 	unsigned int hash;
439 	struct ip_vs_service *svc;
440 
441 	/* Check for fwmark addressed entries */
442 	hash = ip_vs_svc_fwm_hashkey(ipvs, fwmark);
443 
444 	hlist_for_each_entry_rcu(svc, &ip_vs_svc_fwm_table[hash], f_list) {
445 		if (svc->fwmark == fwmark && svc->af == af
446 		    && (svc->ipvs == ipvs)) {
447 			/* HIT */
448 			return svc;
449 		}
450 	}
451 
452 	return NULL;
453 }
454 
455 /* Find service, called under RCU lock */
456 struct ip_vs_service *
ip_vs_service_find(struct netns_ipvs * ipvs,int af,__u32 fwmark,__u16 protocol,const union nf_inet_addr * vaddr,__be16 vport)457 ip_vs_service_find(struct netns_ipvs *ipvs, int af, __u32 fwmark, __u16 protocol,
458 		   const union nf_inet_addr *vaddr, __be16 vport)
459 {
460 	struct ip_vs_service *svc;
461 
462 	/*
463 	 *	Check the table hashed by fwmark first
464 	 */
465 	if (fwmark) {
466 		svc = __ip_vs_svc_fwm_find(ipvs, af, fwmark);
467 		if (svc)
468 			goto out;
469 	}
470 
471 	/*
472 	 *	Check the table hashed by <protocol,addr,port>
473 	 *	for "full" addressed entries
474 	 */
475 	svc = __ip_vs_service_find(ipvs, af, protocol, vaddr, vport);
476 
477 	if (!svc && protocol == IPPROTO_TCP &&
478 	    atomic_read(&ipvs->ftpsvc_counter) &&
479 	    (vport == FTPDATA || !inet_port_requires_bind_service(ipvs->net, ntohs(vport)))) {
480 		/*
481 		 * Check if ftp service entry exists, the packet
482 		 * might belong to FTP data connections.
483 		 */
484 		svc = __ip_vs_service_find(ipvs, af, protocol, vaddr, FTPPORT);
485 	}
486 
487 	if (svc == NULL
488 	    && atomic_read(&ipvs->nullsvc_counter)) {
489 		/*
490 		 * Check if the catch-all port (port zero) exists
491 		 */
492 		svc = __ip_vs_service_find(ipvs, af, protocol, vaddr, 0);
493 	}
494 
495   out:
496 	IP_VS_DBG_BUF(9, "lookup service: fwm %u %s %s:%u %s\n",
497 		      fwmark, ip_vs_proto_name(protocol),
498 		      IP_VS_DBG_ADDR(af, vaddr), ntohs(vport),
499 		      svc ? "hit" : "not hit");
500 
501 	return svc;
502 }
503 
504 
505 static inline void
__ip_vs_bind_svc(struct ip_vs_dest * dest,struct ip_vs_service * svc)506 __ip_vs_bind_svc(struct ip_vs_dest *dest, struct ip_vs_service *svc)
507 {
508 	atomic_inc(&svc->refcnt);
509 	rcu_assign_pointer(dest->svc, svc);
510 }
511 
ip_vs_service_free(struct ip_vs_service * svc)512 static void ip_vs_service_free(struct ip_vs_service *svc)
513 {
514 	ip_vs_stats_release(&svc->stats);
515 	kfree(svc);
516 }
517 
ip_vs_service_rcu_free(struct rcu_head * head)518 static void ip_vs_service_rcu_free(struct rcu_head *head)
519 {
520 	struct ip_vs_service *svc;
521 
522 	svc = container_of(head, struct ip_vs_service, rcu_head);
523 	ip_vs_service_free(svc);
524 }
525 
__ip_vs_svc_put(struct ip_vs_service * svc)526 static void __ip_vs_svc_put(struct ip_vs_service *svc)
527 {
528 	if (atomic_dec_and_test(&svc->refcnt)) {
529 		IP_VS_DBG_BUF(3, "Removing service %u/%s:%u\n",
530 			      svc->fwmark,
531 			      IP_VS_DBG_ADDR(svc->af, &svc->addr),
532 			      ntohs(svc->port));
533 		call_rcu(&svc->rcu_head, ip_vs_service_rcu_free);
534 	}
535 }
536 
537 
538 /*
539  *	Returns hash value for real service
540  */
ip_vs_rs_hashkey(int af,const union nf_inet_addr * addr,__be16 port)541 static inline unsigned int ip_vs_rs_hashkey(int af,
542 					    const union nf_inet_addr *addr,
543 					    __be16 port)
544 {
545 	unsigned int porth = ntohs(port);
546 	__be32 addr_fold = addr->ip;
547 
548 #ifdef CONFIG_IP_VS_IPV6
549 	if (af == AF_INET6)
550 		addr_fold = addr->ip6[0]^addr->ip6[1]^
551 			    addr->ip6[2]^addr->ip6[3];
552 #endif
553 
554 	return (ntohl(addr_fold)^(porth>>IP_VS_RTAB_BITS)^porth)
555 		& IP_VS_RTAB_MASK;
556 }
557 
558 /* Hash ip_vs_dest in rs_table by <proto,addr,port>. */
ip_vs_rs_hash(struct netns_ipvs * ipvs,struct ip_vs_dest * dest)559 static void ip_vs_rs_hash(struct netns_ipvs *ipvs, struct ip_vs_dest *dest)
560 {
561 	unsigned int hash;
562 	__be16 port;
563 
564 	if (dest->in_rs_table)
565 		return;
566 
567 	switch (IP_VS_DFWD_METHOD(dest)) {
568 	case IP_VS_CONN_F_MASQ:
569 		port = dest->port;
570 		break;
571 	case IP_VS_CONN_F_TUNNEL:
572 		switch (dest->tun_type) {
573 		case IP_VS_CONN_F_TUNNEL_TYPE_GUE:
574 			port = dest->tun_port;
575 			break;
576 		case IP_VS_CONN_F_TUNNEL_TYPE_IPIP:
577 		case IP_VS_CONN_F_TUNNEL_TYPE_GRE:
578 			port = 0;
579 			break;
580 		default:
581 			return;
582 		}
583 		break;
584 	default:
585 		return;
586 	}
587 
588 	/*
589 	 *	Hash by proto,addr,port,
590 	 *	which are the parameters of the real service.
591 	 */
592 	hash = ip_vs_rs_hashkey(dest->af, &dest->addr, port);
593 
594 	hlist_add_head_rcu(&dest->d_list, &ipvs->rs_table[hash]);
595 	dest->in_rs_table = 1;
596 }
597 
598 /* Unhash ip_vs_dest from rs_table. */
ip_vs_rs_unhash(struct ip_vs_dest * dest)599 static void ip_vs_rs_unhash(struct ip_vs_dest *dest)
600 {
601 	/*
602 	 * Remove it from the rs_table table.
603 	 */
604 	if (dest->in_rs_table) {
605 		hlist_del_rcu(&dest->d_list);
606 		dest->in_rs_table = 0;
607 	}
608 }
609 
610 /* Check if real service by <proto,addr,port> is present */
ip_vs_has_real_service(struct netns_ipvs * ipvs,int af,__u16 protocol,const union nf_inet_addr * daddr,__be16 dport)611 bool ip_vs_has_real_service(struct netns_ipvs *ipvs, int af, __u16 protocol,
612 			    const union nf_inet_addr *daddr, __be16 dport)
613 {
614 	unsigned int hash;
615 	struct ip_vs_dest *dest;
616 
617 	/* Check for "full" addressed entries */
618 	hash = ip_vs_rs_hashkey(af, daddr, dport);
619 
620 	hlist_for_each_entry_rcu(dest, &ipvs->rs_table[hash], d_list) {
621 		if (dest->port == dport &&
622 		    dest->af == af &&
623 		    ip_vs_addr_equal(af, &dest->addr, daddr) &&
624 		    (dest->protocol == protocol || dest->vfwmark) &&
625 		    IP_VS_DFWD_METHOD(dest) == IP_VS_CONN_F_MASQ) {
626 			/* HIT */
627 			return true;
628 		}
629 	}
630 
631 	return false;
632 }
633 
634 /* Find real service record by <proto,addr,port>.
635  * In case of multiple records with the same <proto,addr,port>, only
636  * the first found record is returned.
637  *
638  * To be called under RCU lock.
639  */
ip_vs_find_real_service(struct netns_ipvs * ipvs,int af,__u16 protocol,const union nf_inet_addr * daddr,__be16 dport)640 struct ip_vs_dest *ip_vs_find_real_service(struct netns_ipvs *ipvs, int af,
641 					   __u16 protocol,
642 					   const union nf_inet_addr *daddr,
643 					   __be16 dport)
644 {
645 	unsigned int hash;
646 	struct ip_vs_dest *dest;
647 
648 	/* Check for "full" addressed entries */
649 	hash = ip_vs_rs_hashkey(af, daddr, dport);
650 
651 	hlist_for_each_entry_rcu(dest, &ipvs->rs_table[hash], d_list) {
652 		if (dest->port == dport &&
653 		    dest->af == af &&
654 		    ip_vs_addr_equal(af, &dest->addr, daddr) &&
655 		    (dest->protocol == protocol || dest->vfwmark) &&
656 		    IP_VS_DFWD_METHOD(dest) == IP_VS_CONN_F_MASQ) {
657 			/* HIT */
658 			return dest;
659 		}
660 	}
661 
662 	return NULL;
663 }
664 
665 /* Find real service record by <af,addr,tun_port>.
666  * In case of multiple records with the same <af,addr,tun_port>, only
667  * the first found record is returned.
668  *
669  * To be called under RCU lock.
670  */
ip_vs_find_tunnel(struct netns_ipvs * ipvs,int af,const union nf_inet_addr * daddr,__be16 tun_port)671 struct ip_vs_dest *ip_vs_find_tunnel(struct netns_ipvs *ipvs, int af,
672 				     const union nf_inet_addr *daddr,
673 				     __be16 tun_port)
674 {
675 	struct ip_vs_dest *dest;
676 	unsigned int hash;
677 
678 	/* Check for "full" addressed entries */
679 	hash = ip_vs_rs_hashkey(af, daddr, tun_port);
680 
681 	hlist_for_each_entry_rcu(dest, &ipvs->rs_table[hash], d_list) {
682 		if (dest->tun_port == tun_port &&
683 		    dest->af == af &&
684 		    ip_vs_addr_equal(af, &dest->addr, daddr) &&
685 		    IP_VS_DFWD_METHOD(dest) == IP_VS_CONN_F_TUNNEL) {
686 			/* HIT */
687 			return dest;
688 		}
689 	}
690 
691 	return NULL;
692 }
693 
694 /* Lookup destination by {addr,port} in the given service
695  * Called under RCU lock.
696  */
697 static struct ip_vs_dest *
ip_vs_lookup_dest(struct ip_vs_service * svc,int dest_af,const union nf_inet_addr * daddr,__be16 dport)698 ip_vs_lookup_dest(struct ip_vs_service *svc, int dest_af,
699 		  const union nf_inet_addr *daddr, __be16 dport)
700 {
701 	struct ip_vs_dest *dest;
702 
703 	/*
704 	 * Find the destination for the given service
705 	 */
706 	list_for_each_entry_rcu(dest, &svc->destinations, n_list) {
707 		if ((dest->af == dest_af) &&
708 		    ip_vs_addr_equal(dest_af, &dest->addr, daddr) &&
709 		    (dest->port == dport)) {
710 			/* HIT */
711 			return dest;
712 		}
713 	}
714 
715 	return NULL;
716 }
717 
718 /*
719  * Find destination by {daddr,dport,vaddr,protocol}
720  * Created to be used in ip_vs_process_message() in
721  * the backup synchronization daemon. It finds the
722  * destination to be bound to the received connection
723  * on the backup.
724  * Called under RCU lock, no refcnt is returned.
725  */
ip_vs_find_dest(struct netns_ipvs * ipvs,int svc_af,int dest_af,const union nf_inet_addr * daddr,__be16 dport,const union nf_inet_addr * vaddr,__be16 vport,__u16 protocol,__u32 fwmark,__u32 flags)726 struct ip_vs_dest *ip_vs_find_dest(struct netns_ipvs *ipvs, int svc_af, int dest_af,
727 				   const union nf_inet_addr *daddr,
728 				   __be16 dport,
729 				   const union nf_inet_addr *vaddr,
730 				   __be16 vport, __u16 protocol, __u32 fwmark,
731 				   __u32 flags)
732 {
733 	struct ip_vs_dest *dest;
734 	struct ip_vs_service *svc;
735 	__be16 port = dport;
736 
737 	svc = ip_vs_service_find(ipvs, svc_af, fwmark, protocol, vaddr, vport);
738 	if (!svc)
739 		return NULL;
740 	if (fwmark && (flags & IP_VS_CONN_F_FWD_MASK) != IP_VS_CONN_F_MASQ)
741 		port = 0;
742 	dest = ip_vs_lookup_dest(svc, dest_af, daddr, port);
743 	if (!dest)
744 		dest = ip_vs_lookup_dest(svc, dest_af, daddr, port ^ dport);
745 	return dest;
746 }
747 
ip_vs_dest_dst_rcu_free(struct rcu_head * head)748 void ip_vs_dest_dst_rcu_free(struct rcu_head *head)
749 {
750 	struct ip_vs_dest_dst *dest_dst = container_of(head,
751 						       struct ip_vs_dest_dst,
752 						       rcu_head);
753 
754 	dst_release(dest_dst->dst_cache);
755 	kfree(dest_dst);
756 }
757 
758 /* Release dest_dst and dst_cache for dest in user context */
__ip_vs_dst_cache_reset(struct ip_vs_dest * dest)759 static void __ip_vs_dst_cache_reset(struct ip_vs_dest *dest)
760 {
761 	struct ip_vs_dest_dst *old;
762 
763 	old = rcu_dereference_protected(dest->dest_dst, 1);
764 	if (old) {
765 		RCU_INIT_POINTER(dest->dest_dst, NULL);
766 		call_rcu(&old->rcu_head, ip_vs_dest_dst_rcu_free);
767 	}
768 }
769 
770 /*
771  *  Lookup dest by {svc,addr,port} in the destination trash.
772  *  The destination trash is used to hold the destinations that are removed
773  *  from the service table but are still referenced by some conn entries.
774  *  The reason to add the destination trash is when the dest is temporary
775  *  down (either by administrator or by monitor program), the dest can be
776  *  picked back from the trash, the remaining connections to the dest can
777  *  continue, and the counting information of the dest is also useful for
778  *  scheduling.
779  */
780 static struct ip_vs_dest *
ip_vs_trash_get_dest(struct ip_vs_service * svc,int dest_af,const union nf_inet_addr * daddr,__be16 dport)781 ip_vs_trash_get_dest(struct ip_vs_service *svc, int dest_af,
782 		     const union nf_inet_addr *daddr, __be16 dport)
783 {
784 	struct ip_vs_dest *dest;
785 	struct netns_ipvs *ipvs = svc->ipvs;
786 
787 	/*
788 	 * Find the destination in trash
789 	 */
790 	spin_lock_bh(&ipvs->dest_trash_lock);
791 	list_for_each_entry(dest, &ipvs->dest_trash, t_list) {
792 		IP_VS_DBG_BUF(3, "Destination %u/%s:%u still in trash, "
793 			      "dest->refcnt=%d\n",
794 			      dest->vfwmark,
795 			      IP_VS_DBG_ADDR(dest->af, &dest->addr),
796 			      ntohs(dest->port),
797 			      refcount_read(&dest->refcnt));
798 		if (dest->af == dest_af &&
799 		    ip_vs_addr_equal(dest_af, &dest->addr, daddr) &&
800 		    dest->port == dport &&
801 		    dest->vfwmark == svc->fwmark &&
802 		    dest->protocol == svc->protocol &&
803 		    (svc->fwmark ||
804 		     (ip_vs_addr_equal(svc->af, &dest->vaddr, &svc->addr) &&
805 		      dest->vport == svc->port))) {
806 			/* HIT */
807 			list_del(&dest->t_list);
808 			goto out;
809 		}
810 	}
811 
812 	dest = NULL;
813 
814 out:
815 	spin_unlock_bh(&ipvs->dest_trash_lock);
816 
817 	return dest;
818 }
819 
ip_vs_dest_rcu_free(struct rcu_head * head)820 static void ip_vs_dest_rcu_free(struct rcu_head *head)
821 {
822 	struct ip_vs_dest *dest;
823 
824 	dest = container_of(head, struct ip_vs_dest, rcu_head);
825 	ip_vs_stats_release(&dest->stats);
826 	ip_vs_dest_put_and_free(dest);
827 }
828 
ip_vs_dest_free(struct ip_vs_dest * dest)829 static void ip_vs_dest_free(struct ip_vs_dest *dest)
830 {
831 	struct ip_vs_service *svc = rcu_dereference_protected(dest->svc, 1);
832 
833 	__ip_vs_dst_cache_reset(dest);
834 	__ip_vs_svc_put(svc);
835 	call_rcu(&dest->rcu_head, ip_vs_dest_rcu_free);
836 }
837 
838 /*
839  *  Clean up all the destinations in the trash
840  *  Called by the ip_vs_control_cleanup()
841  *
842  *  When the ip_vs_control_clearup is activated by ipvs module exit,
843  *  the service tables must have been flushed and all the connections
844  *  are expired, and the refcnt of each destination in the trash must
845  *  be 1, so we simply release them here.
846  */
ip_vs_trash_cleanup(struct netns_ipvs * ipvs)847 static void ip_vs_trash_cleanup(struct netns_ipvs *ipvs)
848 {
849 	struct ip_vs_dest *dest, *nxt;
850 
851 	del_timer_sync(&ipvs->dest_trash_timer);
852 	/* No need to use dest_trash_lock */
853 	list_for_each_entry_safe(dest, nxt, &ipvs->dest_trash, t_list) {
854 		list_del(&dest->t_list);
855 		ip_vs_dest_free(dest);
856 	}
857 }
858 
ip_vs_stats_rcu_free(struct rcu_head * head)859 static void ip_vs_stats_rcu_free(struct rcu_head *head)
860 {
861 	struct ip_vs_stats_rcu *rs = container_of(head,
862 						  struct ip_vs_stats_rcu,
863 						  rcu_head);
864 
865 	ip_vs_stats_release(&rs->s);
866 	kfree(rs);
867 }
868 
869 static void
ip_vs_copy_stats(struct ip_vs_kstats * dst,struct ip_vs_stats * src)870 ip_vs_copy_stats(struct ip_vs_kstats *dst, struct ip_vs_stats *src)
871 {
872 #define IP_VS_SHOW_STATS_COUNTER(c) dst->c = src->kstats.c - src->kstats0.c
873 
874 	spin_lock(&src->lock);
875 
876 	IP_VS_SHOW_STATS_COUNTER(conns);
877 	IP_VS_SHOW_STATS_COUNTER(inpkts);
878 	IP_VS_SHOW_STATS_COUNTER(outpkts);
879 	IP_VS_SHOW_STATS_COUNTER(inbytes);
880 	IP_VS_SHOW_STATS_COUNTER(outbytes);
881 
882 	ip_vs_read_estimator(dst, src);
883 
884 	spin_unlock(&src->lock);
885 }
886 
887 static void
ip_vs_export_stats_user(struct ip_vs_stats_user * dst,struct ip_vs_kstats * src)888 ip_vs_export_stats_user(struct ip_vs_stats_user *dst, struct ip_vs_kstats *src)
889 {
890 	dst->conns = (u32)src->conns;
891 	dst->inpkts = (u32)src->inpkts;
892 	dst->outpkts = (u32)src->outpkts;
893 	dst->inbytes = src->inbytes;
894 	dst->outbytes = src->outbytes;
895 	dst->cps = (u32)src->cps;
896 	dst->inpps = (u32)src->inpps;
897 	dst->outpps = (u32)src->outpps;
898 	dst->inbps = (u32)src->inbps;
899 	dst->outbps = (u32)src->outbps;
900 }
901 
902 static void
ip_vs_zero_stats(struct ip_vs_stats * stats)903 ip_vs_zero_stats(struct ip_vs_stats *stats)
904 {
905 	spin_lock(&stats->lock);
906 
907 	/* get current counters as zero point, rates are zeroed */
908 
909 #define IP_VS_ZERO_STATS_COUNTER(c) stats->kstats0.c = stats->kstats.c
910 
911 	IP_VS_ZERO_STATS_COUNTER(conns);
912 	IP_VS_ZERO_STATS_COUNTER(inpkts);
913 	IP_VS_ZERO_STATS_COUNTER(outpkts);
914 	IP_VS_ZERO_STATS_COUNTER(inbytes);
915 	IP_VS_ZERO_STATS_COUNTER(outbytes);
916 
917 	ip_vs_zero_estimator(stats);
918 
919 	spin_unlock(&stats->lock);
920 }
921 
922 /* Allocate fields after kzalloc */
ip_vs_stats_init_alloc(struct ip_vs_stats * s)923 int ip_vs_stats_init_alloc(struct ip_vs_stats *s)
924 {
925 	int i;
926 
927 	spin_lock_init(&s->lock);
928 	s->cpustats = alloc_percpu(struct ip_vs_cpu_stats);
929 	if (!s->cpustats)
930 		return -ENOMEM;
931 
932 	for_each_possible_cpu(i) {
933 		struct ip_vs_cpu_stats *cs = per_cpu_ptr(s->cpustats, i);
934 
935 		u64_stats_init(&cs->syncp);
936 	}
937 	return 0;
938 }
939 
ip_vs_stats_alloc(void)940 struct ip_vs_stats *ip_vs_stats_alloc(void)
941 {
942 	struct ip_vs_stats *s = kzalloc(sizeof(*s), GFP_KERNEL);
943 
944 	if (s && ip_vs_stats_init_alloc(s) >= 0)
945 		return s;
946 	kfree(s);
947 	return NULL;
948 }
949 
ip_vs_stats_release(struct ip_vs_stats * stats)950 void ip_vs_stats_release(struct ip_vs_stats *stats)
951 {
952 	free_percpu(stats->cpustats);
953 }
954 
ip_vs_stats_free(struct ip_vs_stats * stats)955 void ip_vs_stats_free(struct ip_vs_stats *stats)
956 {
957 	if (stats) {
958 		ip_vs_stats_release(stats);
959 		kfree(stats);
960 	}
961 }
962 
963 /*
964  *	Update a destination in the given service
965  */
966 static void
__ip_vs_update_dest(struct ip_vs_service * svc,struct ip_vs_dest * dest,struct ip_vs_dest_user_kern * udest,int add)967 __ip_vs_update_dest(struct ip_vs_service *svc, struct ip_vs_dest *dest,
968 		    struct ip_vs_dest_user_kern *udest, int add)
969 {
970 	struct netns_ipvs *ipvs = svc->ipvs;
971 	struct ip_vs_service *old_svc;
972 	struct ip_vs_scheduler *sched;
973 	int conn_flags;
974 
975 	/* We cannot modify an address and change the address family */
976 	BUG_ON(!add && udest->af != dest->af);
977 
978 	if (add && udest->af != svc->af)
979 		ipvs->mixed_address_family_dests++;
980 
981 	/* keep the last_weight with latest non-0 weight */
982 	if (add || udest->weight != 0)
983 		atomic_set(&dest->last_weight, udest->weight);
984 
985 	/* set the weight and the flags */
986 	atomic_set(&dest->weight, udest->weight);
987 	conn_flags = udest->conn_flags & IP_VS_CONN_F_DEST_MASK;
988 	conn_flags |= IP_VS_CONN_F_INACTIVE;
989 
990 	/* Need to rehash? */
991 	if ((udest->conn_flags & IP_VS_CONN_F_FWD_MASK) !=
992 	    IP_VS_DFWD_METHOD(dest) ||
993 	    udest->tun_type != dest->tun_type ||
994 	    udest->tun_port != dest->tun_port)
995 		ip_vs_rs_unhash(dest);
996 
997 	/* set the tunnel info */
998 	dest->tun_type = udest->tun_type;
999 	dest->tun_port = udest->tun_port;
1000 	dest->tun_flags = udest->tun_flags;
1001 
1002 	/* set the IP_VS_CONN_F_NOOUTPUT flag if not masquerading/NAT */
1003 	if ((conn_flags & IP_VS_CONN_F_FWD_MASK) != IP_VS_CONN_F_MASQ) {
1004 		conn_flags |= IP_VS_CONN_F_NOOUTPUT;
1005 	} else {
1006 		/* FTP-NAT requires conntrack for mangling */
1007 		if (svc->port == FTPPORT)
1008 			ip_vs_register_conntrack(svc);
1009 	}
1010 	atomic_set(&dest->conn_flags, conn_flags);
1011 	/* Put the real service in rs_table if not present. */
1012 	ip_vs_rs_hash(ipvs, dest);
1013 
1014 	/* bind the service */
1015 	old_svc = rcu_dereference_protected(dest->svc, 1);
1016 	if (!old_svc) {
1017 		__ip_vs_bind_svc(dest, svc);
1018 	} else {
1019 		if (old_svc != svc) {
1020 			ip_vs_zero_stats(&dest->stats);
1021 			__ip_vs_bind_svc(dest, svc);
1022 			__ip_vs_svc_put(old_svc);
1023 		}
1024 	}
1025 
1026 	/* set the dest status flags */
1027 	dest->flags |= IP_VS_DEST_F_AVAILABLE;
1028 
1029 	if (udest->u_threshold == 0 || udest->u_threshold > dest->u_threshold)
1030 		dest->flags &= ~IP_VS_DEST_F_OVERLOAD;
1031 	dest->u_threshold = udest->u_threshold;
1032 	dest->l_threshold = udest->l_threshold;
1033 
1034 	dest->af = udest->af;
1035 
1036 	spin_lock_bh(&dest->dst_lock);
1037 	__ip_vs_dst_cache_reset(dest);
1038 	spin_unlock_bh(&dest->dst_lock);
1039 
1040 	if (add) {
1041 		list_add_rcu(&dest->n_list, &svc->destinations);
1042 		svc->num_dests++;
1043 		sched = rcu_dereference_protected(svc->scheduler, 1);
1044 		if (sched && sched->add_dest)
1045 			sched->add_dest(svc, dest);
1046 	} else {
1047 		sched = rcu_dereference_protected(svc->scheduler, 1);
1048 		if (sched && sched->upd_dest)
1049 			sched->upd_dest(svc, dest);
1050 	}
1051 }
1052 
1053 
1054 /*
1055  *	Create a destination for the given service
1056  */
1057 static int
ip_vs_new_dest(struct ip_vs_service * svc,struct ip_vs_dest_user_kern * udest)1058 ip_vs_new_dest(struct ip_vs_service *svc, struct ip_vs_dest_user_kern *udest)
1059 {
1060 	struct ip_vs_dest *dest;
1061 	unsigned int atype;
1062 	int ret;
1063 
1064 #ifdef CONFIG_IP_VS_IPV6
1065 	if (udest->af == AF_INET6) {
1066 		atype = ipv6_addr_type(&udest->addr.in6);
1067 		if ((!(atype & IPV6_ADDR_UNICAST) ||
1068 			atype & IPV6_ADDR_LINKLOCAL) &&
1069 			!__ip_vs_addr_is_local_v6(svc->ipvs->net, &udest->addr.in6))
1070 			return -EINVAL;
1071 
1072 		ret = nf_defrag_ipv6_enable(svc->ipvs->net);
1073 		if (ret)
1074 			return ret;
1075 	} else
1076 #endif
1077 	{
1078 		atype = inet_addr_type(svc->ipvs->net, udest->addr.ip);
1079 		if (atype != RTN_LOCAL && atype != RTN_UNICAST)
1080 			return -EINVAL;
1081 	}
1082 
1083 	dest = kzalloc(sizeof(struct ip_vs_dest), GFP_KERNEL);
1084 	if (dest == NULL)
1085 		return -ENOMEM;
1086 
1087 	ret = ip_vs_stats_init_alloc(&dest->stats);
1088 	if (ret < 0)
1089 		goto err_alloc;
1090 
1091 	ret = ip_vs_start_estimator(svc->ipvs, &dest->stats);
1092 	if (ret < 0)
1093 		goto err_stats;
1094 
1095 	dest->af = udest->af;
1096 	dest->protocol = svc->protocol;
1097 	dest->vaddr = svc->addr;
1098 	dest->vport = svc->port;
1099 	dest->vfwmark = svc->fwmark;
1100 	ip_vs_addr_copy(udest->af, &dest->addr, &udest->addr);
1101 	dest->port = udest->port;
1102 
1103 	atomic_set(&dest->activeconns, 0);
1104 	atomic_set(&dest->inactconns, 0);
1105 	atomic_set(&dest->persistconns, 0);
1106 	refcount_set(&dest->refcnt, 1);
1107 
1108 	INIT_HLIST_NODE(&dest->d_list);
1109 	spin_lock_init(&dest->dst_lock);
1110 	__ip_vs_update_dest(svc, dest, udest, 1);
1111 
1112 	return 0;
1113 
1114 err_stats:
1115 	ip_vs_stats_release(&dest->stats);
1116 
1117 err_alloc:
1118 	kfree(dest);
1119 	return ret;
1120 }
1121 
1122 
1123 /*
1124  *	Add a destination into an existing service
1125  */
1126 static int
ip_vs_add_dest(struct ip_vs_service * svc,struct ip_vs_dest_user_kern * udest)1127 ip_vs_add_dest(struct ip_vs_service *svc, struct ip_vs_dest_user_kern *udest)
1128 {
1129 	struct ip_vs_dest *dest;
1130 	union nf_inet_addr daddr;
1131 	__be16 dport = udest->port;
1132 	int ret;
1133 
1134 	if (udest->weight < 0) {
1135 		pr_err("%s(): server weight less than zero\n", __func__);
1136 		return -ERANGE;
1137 	}
1138 
1139 	if (udest->l_threshold > udest->u_threshold) {
1140 		pr_err("%s(): lower threshold is higher than upper threshold\n",
1141 			__func__);
1142 		return -ERANGE;
1143 	}
1144 
1145 	if (udest->tun_type == IP_VS_CONN_F_TUNNEL_TYPE_GUE) {
1146 		if (udest->tun_port == 0) {
1147 			pr_err("%s(): tunnel port is zero\n", __func__);
1148 			return -EINVAL;
1149 		}
1150 	}
1151 
1152 	ip_vs_addr_copy(udest->af, &daddr, &udest->addr);
1153 
1154 	/* We use function that requires RCU lock */
1155 	rcu_read_lock();
1156 	dest = ip_vs_lookup_dest(svc, udest->af, &daddr, dport);
1157 	rcu_read_unlock();
1158 
1159 	if (dest != NULL) {
1160 		IP_VS_DBG(1, "%s(): dest already exists\n", __func__);
1161 		return -EEXIST;
1162 	}
1163 
1164 	/*
1165 	 * Check if the dest already exists in the trash and
1166 	 * is from the same service
1167 	 */
1168 	dest = ip_vs_trash_get_dest(svc, udest->af, &daddr, dport);
1169 
1170 	if (dest != NULL) {
1171 		IP_VS_DBG_BUF(3, "Get destination %s:%u from trash, "
1172 			      "dest->refcnt=%d, service %u/%s:%u\n",
1173 			      IP_VS_DBG_ADDR(udest->af, &daddr), ntohs(dport),
1174 			      refcount_read(&dest->refcnt),
1175 			      dest->vfwmark,
1176 			      IP_VS_DBG_ADDR(svc->af, &dest->vaddr),
1177 			      ntohs(dest->vport));
1178 
1179 		ret = ip_vs_start_estimator(svc->ipvs, &dest->stats);
1180 		if (ret < 0)
1181 			return ret;
1182 		__ip_vs_update_dest(svc, dest, udest, 1);
1183 	} else {
1184 		/*
1185 		 * Allocate and initialize the dest structure
1186 		 */
1187 		ret = ip_vs_new_dest(svc, udest);
1188 	}
1189 
1190 	return ret;
1191 }
1192 
1193 
1194 /*
1195  *	Edit a destination in the given service
1196  */
1197 static int
ip_vs_edit_dest(struct ip_vs_service * svc,struct ip_vs_dest_user_kern * udest)1198 ip_vs_edit_dest(struct ip_vs_service *svc, struct ip_vs_dest_user_kern *udest)
1199 {
1200 	struct ip_vs_dest *dest;
1201 	union nf_inet_addr daddr;
1202 	__be16 dport = udest->port;
1203 
1204 	if (udest->weight < 0) {
1205 		pr_err("%s(): server weight less than zero\n", __func__);
1206 		return -ERANGE;
1207 	}
1208 
1209 	if (udest->l_threshold > udest->u_threshold) {
1210 		pr_err("%s(): lower threshold is higher than upper threshold\n",
1211 			__func__);
1212 		return -ERANGE;
1213 	}
1214 
1215 	if (udest->tun_type == IP_VS_CONN_F_TUNNEL_TYPE_GUE) {
1216 		if (udest->tun_port == 0) {
1217 			pr_err("%s(): tunnel port is zero\n", __func__);
1218 			return -EINVAL;
1219 		}
1220 	}
1221 
1222 	ip_vs_addr_copy(udest->af, &daddr, &udest->addr);
1223 
1224 	/* We use function that requires RCU lock */
1225 	rcu_read_lock();
1226 	dest = ip_vs_lookup_dest(svc, udest->af, &daddr, dport);
1227 	rcu_read_unlock();
1228 
1229 	if (dest == NULL) {
1230 		IP_VS_DBG(1, "%s(): dest doesn't exist\n", __func__);
1231 		return -ENOENT;
1232 	}
1233 
1234 	__ip_vs_update_dest(svc, dest, udest, 0);
1235 
1236 	return 0;
1237 }
1238 
1239 /*
1240  *	Delete a destination (must be already unlinked from the service)
1241  */
__ip_vs_del_dest(struct netns_ipvs * ipvs,struct ip_vs_dest * dest,bool cleanup)1242 static void __ip_vs_del_dest(struct netns_ipvs *ipvs, struct ip_vs_dest *dest,
1243 			     bool cleanup)
1244 {
1245 	ip_vs_stop_estimator(ipvs, &dest->stats);
1246 
1247 	/*
1248 	 *  Remove it from the d-linked list with the real services.
1249 	 */
1250 	ip_vs_rs_unhash(dest);
1251 
1252 	spin_lock_bh(&ipvs->dest_trash_lock);
1253 	IP_VS_DBG_BUF(3, "Moving dest %s:%u into trash, dest->refcnt=%d\n",
1254 		      IP_VS_DBG_ADDR(dest->af, &dest->addr), ntohs(dest->port),
1255 		      refcount_read(&dest->refcnt));
1256 	if (list_empty(&ipvs->dest_trash) && !cleanup)
1257 		mod_timer(&ipvs->dest_trash_timer,
1258 			  jiffies + (IP_VS_DEST_TRASH_PERIOD >> 1));
1259 	/* dest lives in trash with reference */
1260 	list_add(&dest->t_list, &ipvs->dest_trash);
1261 	dest->idle_start = 0;
1262 	spin_unlock_bh(&ipvs->dest_trash_lock);
1263 
1264 	/* Queue up delayed work to expire all no destination connections.
1265 	 * No-op when CONFIG_SYSCTL is disabled.
1266 	 */
1267 	if (!cleanup)
1268 		ip_vs_enqueue_expire_nodest_conns(ipvs);
1269 }
1270 
1271 
1272 /*
1273  *	Unlink a destination from the given service
1274  */
__ip_vs_unlink_dest(struct ip_vs_service * svc,struct ip_vs_dest * dest,int svcupd)1275 static void __ip_vs_unlink_dest(struct ip_vs_service *svc,
1276 				struct ip_vs_dest *dest,
1277 				int svcupd)
1278 {
1279 	dest->flags &= ~IP_VS_DEST_F_AVAILABLE;
1280 
1281 	/*
1282 	 *  Remove it from the d-linked destination list.
1283 	 */
1284 	list_del_rcu(&dest->n_list);
1285 	svc->num_dests--;
1286 
1287 	if (dest->af != svc->af)
1288 		svc->ipvs->mixed_address_family_dests--;
1289 
1290 	if (svcupd) {
1291 		struct ip_vs_scheduler *sched;
1292 
1293 		sched = rcu_dereference_protected(svc->scheduler, 1);
1294 		if (sched && sched->del_dest)
1295 			sched->del_dest(svc, dest);
1296 	}
1297 }
1298 
1299 
1300 /*
1301  *	Delete a destination server in the given service
1302  */
1303 static int
ip_vs_del_dest(struct ip_vs_service * svc,struct ip_vs_dest_user_kern * udest)1304 ip_vs_del_dest(struct ip_vs_service *svc, struct ip_vs_dest_user_kern *udest)
1305 {
1306 	struct ip_vs_dest *dest;
1307 	__be16 dport = udest->port;
1308 
1309 	/* We use function that requires RCU lock */
1310 	rcu_read_lock();
1311 	dest = ip_vs_lookup_dest(svc, udest->af, &udest->addr, dport);
1312 	rcu_read_unlock();
1313 
1314 	if (dest == NULL) {
1315 		IP_VS_DBG(1, "%s(): destination not found!\n", __func__);
1316 		return -ENOENT;
1317 	}
1318 
1319 	/*
1320 	 *	Unlink dest from the service
1321 	 */
1322 	__ip_vs_unlink_dest(svc, dest, 1);
1323 
1324 	/*
1325 	 *	Delete the destination
1326 	 */
1327 	__ip_vs_del_dest(svc->ipvs, dest, false);
1328 
1329 	return 0;
1330 }
1331 
ip_vs_dest_trash_expire(struct timer_list * t)1332 static void ip_vs_dest_trash_expire(struct timer_list *t)
1333 {
1334 	struct netns_ipvs *ipvs = from_timer(ipvs, t, dest_trash_timer);
1335 	struct ip_vs_dest *dest, *next;
1336 	unsigned long now = jiffies;
1337 
1338 	spin_lock(&ipvs->dest_trash_lock);
1339 	list_for_each_entry_safe(dest, next, &ipvs->dest_trash, t_list) {
1340 		if (refcount_read(&dest->refcnt) > 1)
1341 			continue;
1342 		if (dest->idle_start) {
1343 			if (time_before(now, dest->idle_start +
1344 					     IP_VS_DEST_TRASH_PERIOD))
1345 				continue;
1346 		} else {
1347 			dest->idle_start = max(1UL, now);
1348 			continue;
1349 		}
1350 		IP_VS_DBG_BUF(3, "Removing destination %u/%s:%u from trash\n",
1351 			      dest->vfwmark,
1352 			      IP_VS_DBG_ADDR(dest->af, &dest->addr),
1353 			      ntohs(dest->port));
1354 		list_del(&dest->t_list);
1355 		ip_vs_dest_free(dest);
1356 	}
1357 	if (!list_empty(&ipvs->dest_trash))
1358 		mod_timer(&ipvs->dest_trash_timer,
1359 			  jiffies + (IP_VS_DEST_TRASH_PERIOD >> 1));
1360 	spin_unlock(&ipvs->dest_trash_lock);
1361 }
1362 
1363 /*
1364  *	Add a service into the service hash table
1365  */
1366 static int
ip_vs_add_service(struct netns_ipvs * ipvs,struct ip_vs_service_user_kern * u,struct ip_vs_service ** svc_p)1367 ip_vs_add_service(struct netns_ipvs *ipvs, struct ip_vs_service_user_kern *u,
1368 		  struct ip_vs_service **svc_p)
1369 {
1370 	int ret = 0;
1371 	struct ip_vs_scheduler *sched = NULL;
1372 	struct ip_vs_pe *pe = NULL;
1373 	struct ip_vs_service *svc = NULL;
1374 	int ret_hooks = -1;
1375 
1376 	/* increase the module use count */
1377 	if (!ip_vs_use_count_inc())
1378 		return -ENOPROTOOPT;
1379 
1380 	/* Lookup the scheduler by 'u->sched_name' */
1381 	if (strcmp(u->sched_name, "none")) {
1382 		sched = ip_vs_scheduler_get(u->sched_name);
1383 		if (!sched) {
1384 			pr_info("Scheduler module ip_vs_%s not found\n",
1385 				u->sched_name);
1386 			ret = -ENOENT;
1387 			goto out_err;
1388 		}
1389 	}
1390 
1391 	if (u->pe_name && *u->pe_name) {
1392 		pe = ip_vs_pe_getbyname(u->pe_name);
1393 		if (pe == NULL) {
1394 			pr_info("persistence engine module ip_vs_pe_%s "
1395 				"not found\n", u->pe_name);
1396 			ret = -ENOENT;
1397 			goto out_err;
1398 		}
1399 	}
1400 
1401 #ifdef CONFIG_IP_VS_IPV6
1402 	if (u->af == AF_INET6) {
1403 		__u32 plen = (__force __u32) u->netmask;
1404 
1405 		if (plen < 1 || plen > 128) {
1406 			ret = -EINVAL;
1407 			goto out_err;
1408 		}
1409 
1410 		ret = nf_defrag_ipv6_enable(ipvs->net);
1411 		if (ret)
1412 			goto out_err;
1413 	}
1414 #endif
1415 
1416 	if ((u->af == AF_INET && !ipvs->num_services) ||
1417 	    (u->af == AF_INET6 && !ipvs->num_services6)) {
1418 		ret = ip_vs_register_hooks(ipvs, u->af);
1419 		if (ret < 0)
1420 			goto out_err;
1421 		ret_hooks = ret;
1422 	}
1423 
1424 	svc = kzalloc(sizeof(struct ip_vs_service), GFP_KERNEL);
1425 	if (svc == NULL) {
1426 		IP_VS_DBG(1, "%s(): no memory\n", __func__);
1427 		ret = -ENOMEM;
1428 		goto out_err;
1429 	}
1430 	ret = ip_vs_stats_init_alloc(&svc->stats);
1431 	if (ret < 0)
1432 		goto out_err;
1433 
1434 	/* I'm the first user of the service */
1435 	atomic_set(&svc->refcnt, 0);
1436 
1437 	svc->af = u->af;
1438 	svc->protocol = u->protocol;
1439 	ip_vs_addr_copy(svc->af, &svc->addr, &u->addr);
1440 	svc->port = u->port;
1441 	svc->fwmark = u->fwmark;
1442 	svc->flags = u->flags & ~IP_VS_SVC_F_HASHED;
1443 	svc->timeout = u->timeout * HZ;
1444 	svc->netmask = u->netmask;
1445 	svc->ipvs = ipvs;
1446 
1447 	INIT_LIST_HEAD(&svc->destinations);
1448 	spin_lock_init(&svc->sched_lock);
1449 
1450 	/* Bind the scheduler */
1451 	if (sched) {
1452 		ret = ip_vs_bind_scheduler(svc, sched);
1453 		if (ret)
1454 			goto out_err;
1455 		sched = NULL;
1456 	}
1457 
1458 	ret = ip_vs_start_estimator(ipvs, &svc->stats);
1459 	if (ret < 0)
1460 		goto out_err;
1461 
1462 	/* Update the virtual service counters */
1463 	if (svc->port == FTPPORT)
1464 		atomic_inc(&ipvs->ftpsvc_counter);
1465 	else if (svc->port == 0)
1466 		atomic_inc(&ipvs->nullsvc_counter);
1467 	if (pe && pe->conn_out)
1468 		atomic_inc(&ipvs->conn_out_counter);
1469 
1470 	/* Bind the ct retriever */
1471 	RCU_INIT_POINTER(svc->pe, pe);
1472 	pe = NULL;
1473 
1474 	/* Count only IPv4 services for old get/setsockopt interface */
1475 	if (svc->af == AF_INET)
1476 		ipvs->num_services++;
1477 	else if (svc->af == AF_INET6)
1478 		ipvs->num_services6++;
1479 
1480 	/* Hash the service into the service table */
1481 	ip_vs_svc_hash(svc);
1482 
1483 	*svc_p = svc;
1484 
1485 	if (!ipvs->enable) {
1486 		/* Now there is a service - full throttle */
1487 		ipvs->enable = 1;
1488 
1489 		/* Start estimation for first time */
1490 		ip_vs_est_reload_start(ipvs);
1491 	}
1492 
1493 	return 0;
1494 
1495 
1496  out_err:
1497 	if (ret_hooks >= 0)
1498 		ip_vs_unregister_hooks(ipvs, u->af);
1499 	if (svc != NULL) {
1500 		ip_vs_unbind_scheduler(svc, sched);
1501 		ip_vs_service_free(svc);
1502 	}
1503 	ip_vs_scheduler_put(sched);
1504 	ip_vs_pe_put(pe);
1505 
1506 	/* decrease the module use count */
1507 	ip_vs_use_count_dec();
1508 
1509 	return ret;
1510 }
1511 
1512 
1513 /*
1514  *	Edit a service and bind it with a new scheduler
1515  */
1516 static int
ip_vs_edit_service(struct ip_vs_service * svc,struct ip_vs_service_user_kern * u)1517 ip_vs_edit_service(struct ip_vs_service *svc, struct ip_vs_service_user_kern *u)
1518 {
1519 	struct ip_vs_scheduler *sched = NULL, *old_sched;
1520 	struct ip_vs_pe *pe = NULL, *old_pe = NULL;
1521 	int ret = 0;
1522 	bool new_pe_conn_out, old_pe_conn_out;
1523 
1524 	/*
1525 	 * Lookup the scheduler, by 'u->sched_name'
1526 	 */
1527 	if (strcmp(u->sched_name, "none")) {
1528 		sched = ip_vs_scheduler_get(u->sched_name);
1529 		if (!sched) {
1530 			pr_info("Scheduler module ip_vs_%s not found\n",
1531 				u->sched_name);
1532 			return -ENOENT;
1533 		}
1534 	}
1535 	old_sched = sched;
1536 
1537 	if (u->pe_name && *u->pe_name) {
1538 		pe = ip_vs_pe_getbyname(u->pe_name);
1539 		if (pe == NULL) {
1540 			pr_info("persistence engine module ip_vs_pe_%s "
1541 				"not found\n", u->pe_name);
1542 			ret = -ENOENT;
1543 			goto out;
1544 		}
1545 		old_pe = pe;
1546 	}
1547 
1548 #ifdef CONFIG_IP_VS_IPV6
1549 	if (u->af == AF_INET6) {
1550 		__u32 plen = (__force __u32) u->netmask;
1551 
1552 		if (plen < 1 || plen > 128) {
1553 			ret = -EINVAL;
1554 			goto out;
1555 		}
1556 	}
1557 #endif
1558 
1559 	old_sched = rcu_dereference_protected(svc->scheduler, 1);
1560 	if (sched != old_sched) {
1561 		if (old_sched) {
1562 			ip_vs_unbind_scheduler(svc, old_sched);
1563 			RCU_INIT_POINTER(svc->scheduler, NULL);
1564 			/* Wait all svc->sched_data users */
1565 			synchronize_rcu();
1566 		}
1567 		/* Bind the new scheduler */
1568 		if (sched) {
1569 			ret = ip_vs_bind_scheduler(svc, sched);
1570 			if (ret) {
1571 				ip_vs_scheduler_put(sched);
1572 				goto out;
1573 			}
1574 		}
1575 	}
1576 
1577 	/*
1578 	 * Set the flags and timeout value
1579 	 */
1580 	svc->flags = u->flags | IP_VS_SVC_F_HASHED;
1581 	svc->timeout = u->timeout * HZ;
1582 	svc->netmask = u->netmask;
1583 
1584 	old_pe = rcu_dereference_protected(svc->pe, 1);
1585 	if (pe != old_pe) {
1586 		rcu_assign_pointer(svc->pe, pe);
1587 		/* check for optional methods in new pe */
1588 		new_pe_conn_out = (pe && pe->conn_out) ? true : false;
1589 		old_pe_conn_out = (old_pe && old_pe->conn_out) ? true : false;
1590 		if (new_pe_conn_out && !old_pe_conn_out)
1591 			atomic_inc(&svc->ipvs->conn_out_counter);
1592 		if (old_pe_conn_out && !new_pe_conn_out)
1593 			atomic_dec(&svc->ipvs->conn_out_counter);
1594 	}
1595 
1596 out:
1597 	ip_vs_scheduler_put(old_sched);
1598 	ip_vs_pe_put(old_pe);
1599 	return ret;
1600 }
1601 
1602 /*
1603  *	Delete a service from the service list
1604  *	- The service must be unlinked, unlocked and not referenced!
1605  *	- We are called under _bh lock
1606  */
__ip_vs_del_service(struct ip_vs_service * svc,bool cleanup)1607 static void __ip_vs_del_service(struct ip_vs_service *svc, bool cleanup)
1608 {
1609 	struct ip_vs_dest *dest, *nxt;
1610 	struct ip_vs_scheduler *old_sched;
1611 	struct ip_vs_pe *old_pe;
1612 	struct netns_ipvs *ipvs = svc->ipvs;
1613 
1614 	if (svc->af == AF_INET) {
1615 		ipvs->num_services--;
1616 		if (!ipvs->num_services)
1617 			ip_vs_unregister_hooks(ipvs, svc->af);
1618 	} else if (svc->af == AF_INET6) {
1619 		ipvs->num_services6--;
1620 		if (!ipvs->num_services6)
1621 			ip_vs_unregister_hooks(ipvs, svc->af);
1622 	}
1623 
1624 	ip_vs_stop_estimator(svc->ipvs, &svc->stats);
1625 
1626 	/* Unbind scheduler */
1627 	old_sched = rcu_dereference_protected(svc->scheduler, 1);
1628 	ip_vs_unbind_scheduler(svc, old_sched);
1629 	ip_vs_scheduler_put(old_sched);
1630 
1631 	/* Unbind persistence engine, keep svc->pe */
1632 	old_pe = rcu_dereference_protected(svc->pe, 1);
1633 	if (old_pe && old_pe->conn_out)
1634 		atomic_dec(&ipvs->conn_out_counter);
1635 	ip_vs_pe_put(old_pe);
1636 
1637 	/*
1638 	 *    Unlink the whole destination list
1639 	 */
1640 	list_for_each_entry_safe(dest, nxt, &svc->destinations, n_list) {
1641 		__ip_vs_unlink_dest(svc, dest, 0);
1642 		__ip_vs_del_dest(svc->ipvs, dest, cleanup);
1643 	}
1644 
1645 	/*
1646 	 *    Update the virtual service counters
1647 	 */
1648 	if (svc->port == FTPPORT)
1649 		atomic_dec(&ipvs->ftpsvc_counter);
1650 	else if (svc->port == 0)
1651 		atomic_dec(&ipvs->nullsvc_counter);
1652 
1653 	/*
1654 	 *    Free the service if nobody refers to it
1655 	 */
1656 	__ip_vs_svc_put(svc);
1657 
1658 	/* decrease the module use count */
1659 	ip_vs_use_count_dec();
1660 }
1661 
1662 /*
1663  * Unlink a service from list and try to delete it if its refcnt reached 0
1664  */
ip_vs_unlink_service(struct ip_vs_service * svc,bool cleanup)1665 static void ip_vs_unlink_service(struct ip_vs_service *svc, bool cleanup)
1666 {
1667 	ip_vs_unregister_conntrack(svc);
1668 	/* Hold svc to avoid double release from dest_trash */
1669 	atomic_inc(&svc->refcnt);
1670 	/*
1671 	 * Unhash it from the service table
1672 	 */
1673 	ip_vs_svc_unhash(svc);
1674 
1675 	__ip_vs_del_service(svc, cleanup);
1676 }
1677 
1678 /*
1679  *	Delete a service from the service list
1680  */
ip_vs_del_service(struct ip_vs_service * svc)1681 static int ip_vs_del_service(struct ip_vs_service *svc)
1682 {
1683 	if (svc == NULL)
1684 		return -EEXIST;
1685 	ip_vs_unlink_service(svc, false);
1686 
1687 	return 0;
1688 }
1689 
1690 
1691 /*
1692  *	Flush all the virtual services
1693  */
ip_vs_flush(struct netns_ipvs * ipvs,bool cleanup)1694 static int ip_vs_flush(struct netns_ipvs *ipvs, bool cleanup)
1695 {
1696 	int idx;
1697 	struct ip_vs_service *svc;
1698 	struct hlist_node *n;
1699 
1700 	/*
1701 	 * Flush the service table hashed by <netns,protocol,addr,port>
1702 	 */
1703 	for(idx = 0; idx < IP_VS_SVC_TAB_SIZE; idx++) {
1704 		hlist_for_each_entry_safe(svc, n, &ip_vs_svc_table[idx],
1705 					  s_list) {
1706 			if (svc->ipvs == ipvs)
1707 				ip_vs_unlink_service(svc, cleanup);
1708 		}
1709 	}
1710 
1711 	/*
1712 	 * Flush the service table hashed by fwmark
1713 	 */
1714 	for(idx = 0; idx < IP_VS_SVC_TAB_SIZE; idx++) {
1715 		hlist_for_each_entry_safe(svc, n, &ip_vs_svc_fwm_table[idx],
1716 					  f_list) {
1717 			if (svc->ipvs == ipvs)
1718 				ip_vs_unlink_service(svc, cleanup);
1719 		}
1720 	}
1721 
1722 	return 0;
1723 }
1724 
1725 /*
1726  *	Delete service by {netns} in the service table.
1727  *	Called by __ip_vs_batch_cleanup()
1728  */
ip_vs_service_nets_cleanup(struct list_head * net_list)1729 void ip_vs_service_nets_cleanup(struct list_head *net_list)
1730 {
1731 	struct netns_ipvs *ipvs;
1732 	struct net *net;
1733 
1734 	/* Check for "full" addressed entries */
1735 	mutex_lock(&__ip_vs_mutex);
1736 	list_for_each_entry(net, net_list, exit_list) {
1737 		ipvs = net_ipvs(net);
1738 		ip_vs_flush(ipvs, true);
1739 	}
1740 	mutex_unlock(&__ip_vs_mutex);
1741 }
1742 
1743 /* Put all references for device (dst_cache) */
1744 static inline void
ip_vs_forget_dev(struct ip_vs_dest * dest,struct net_device * dev)1745 ip_vs_forget_dev(struct ip_vs_dest *dest, struct net_device *dev)
1746 {
1747 	struct ip_vs_dest_dst *dest_dst;
1748 
1749 	spin_lock_bh(&dest->dst_lock);
1750 	dest_dst = rcu_dereference_protected(dest->dest_dst, 1);
1751 	if (dest_dst && dest_dst->dst_cache->dev == dev) {
1752 		IP_VS_DBG_BUF(3, "Reset dev:%s dest %s:%u ,dest->refcnt=%d\n",
1753 			      dev->name,
1754 			      IP_VS_DBG_ADDR(dest->af, &dest->addr),
1755 			      ntohs(dest->port),
1756 			      refcount_read(&dest->refcnt));
1757 		__ip_vs_dst_cache_reset(dest);
1758 	}
1759 	spin_unlock_bh(&dest->dst_lock);
1760 
1761 }
1762 /* Netdev event receiver
1763  * Currently only NETDEV_DOWN is handled to release refs to cached dsts
1764  */
ip_vs_dst_event(struct notifier_block * this,unsigned long event,void * ptr)1765 static int ip_vs_dst_event(struct notifier_block *this, unsigned long event,
1766 			   void *ptr)
1767 {
1768 	struct net_device *dev = netdev_notifier_info_to_dev(ptr);
1769 	struct net *net = dev_net(dev);
1770 	struct netns_ipvs *ipvs = net_ipvs(net);
1771 	struct ip_vs_service *svc;
1772 	struct ip_vs_dest *dest;
1773 	unsigned int idx;
1774 
1775 	if (event != NETDEV_DOWN || !ipvs)
1776 		return NOTIFY_DONE;
1777 	IP_VS_DBG(3, "%s() dev=%s\n", __func__, dev->name);
1778 	mutex_lock(&__ip_vs_mutex);
1779 	for (idx = 0; idx < IP_VS_SVC_TAB_SIZE; idx++) {
1780 		hlist_for_each_entry(svc, &ip_vs_svc_table[idx], s_list) {
1781 			if (svc->ipvs == ipvs) {
1782 				list_for_each_entry(dest, &svc->destinations,
1783 						    n_list) {
1784 					ip_vs_forget_dev(dest, dev);
1785 				}
1786 			}
1787 		}
1788 
1789 		hlist_for_each_entry(svc, &ip_vs_svc_fwm_table[idx], f_list) {
1790 			if (svc->ipvs == ipvs) {
1791 				list_for_each_entry(dest, &svc->destinations,
1792 						    n_list) {
1793 					ip_vs_forget_dev(dest, dev);
1794 				}
1795 			}
1796 
1797 		}
1798 	}
1799 
1800 	spin_lock_bh(&ipvs->dest_trash_lock);
1801 	list_for_each_entry(dest, &ipvs->dest_trash, t_list) {
1802 		ip_vs_forget_dev(dest, dev);
1803 	}
1804 	spin_unlock_bh(&ipvs->dest_trash_lock);
1805 	mutex_unlock(&__ip_vs_mutex);
1806 	return NOTIFY_DONE;
1807 }
1808 
1809 /*
1810  *	Zero counters in a service or all services
1811  */
ip_vs_zero_service(struct ip_vs_service * svc)1812 static int ip_vs_zero_service(struct ip_vs_service *svc)
1813 {
1814 	struct ip_vs_dest *dest;
1815 
1816 	list_for_each_entry(dest, &svc->destinations, n_list) {
1817 		ip_vs_zero_stats(&dest->stats);
1818 	}
1819 	ip_vs_zero_stats(&svc->stats);
1820 	return 0;
1821 }
1822 
ip_vs_zero_all(struct netns_ipvs * ipvs)1823 static int ip_vs_zero_all(struct netns_ipvs *ipvs)
1824 {
1825 	int idx;
1826 	struct ip_vs_service *svc;
1827 
1828 	for(idx = 0; idx < IP_VS_SVC_TAB_SIZE; idx++) {
1829 		hlist_for_each_entry(svc, &ip_vs_svc_table[idx], s_list) {
1830 			if (svc->ipvs == ipvs)
1831 				ip_vs_zero_service(svc);
1832 		}
1833 	}
1834 
1835 	for(idx = 0; idx < IP_VS_SVC_TAB_SIZE; idx++) {
1836 		hlist_for_each_entry(svc, &ip_vs_svc_fwm_table[idx], f_list) {
1837 			if (svc->ipvs == ipvs)
1838 				ip_vs_zero_service(svc);
1839 		}
1840 	}
1841 
1842 	ip_vs_zero_stats(&ipvs->tot_stats->s);
1843 	return 0;
1844 }
1845 
1846 #ifdef CONFIG_SYSCTL
1847 
1848 static int
proc_do_defense_mode(const struct ctl_table * table,int write,void * buffer,size_t * lenp,loff_t * ppos)1849 proc_do_defense_mode(const struct ctl_table *table, int write,
1850 		     void *buffer, size_t *lenp, loff_t *ppos)
1851 {
1852 	struct netns_ipvs *ipvs = table->extra2;
1853 	int *valp = table->data;
1854 	int val = *valp;
1855 	int rc;
1856 
1857 	struct ctl_table tmp = {
1858 		.data = &val,
1859 		.maxlen = sizeof(int),
1860 		.mode = table->mode,
1861 	};
1862 
1863 	rc = proc_dointvec(&tmp, write, buffer, lenp, ppos);
1864 	if (write && (*valp != val)) {
1865 		if (val < 0 || val > 3) {
1866 			rc = -EINVAL;
1867 		} else {
1868 			*valp = val;
1869 			update_defense_level(ipvs);
1870 		}
1871 	}
1872 	return rc;
1873 }
1874 
1875 static int
proc_do_sync_threshold(const struct ctl_table * table,int write,void * buffer,size_t * lenp,loff_t * ppos)1876 proc_do_sync_threshold(const struct ctl_table *table, int write,
1877 		       void *buffer, size_t *lenp, loff_t *ppos)
1878 {
1879 	struct netns_ipvs *ipvs = table->extra2;
1880 	int *valp = table->data;
1881 	int val[2];
1882 	int rc;
1883 	struct ctl_table tmp = {
1884 		.data = &val,
1885 		.maxlen = table->maxlen,
1886 		.mode = table->mode,
1887 	};
1888 
1889 	mutex_lock(&ipvs->sync_mutex);
1890 	memcpy(val, valp, sizeof(val));
1891 	rc = proc_dointvec(&tmp, write, buffer, lenp, ppos);
1892 	if (write) {
1893 		if (val[0] < 0 || val[1] < 0 ||
1894 		    (val[0] >= val[1] && val[1]))
1895 			rc = -EINVAL;
1896 		else
1897 			memcpy(valp, val, sizeof(val));
1898 	}
1899 	mutex_unlock(&ipvs->sync_mutex);
1900 	return rc;
1901 }
1902 
1903 static int
proc_do_sync_ports(const struct ctl_table * table,int write,void * buffer,size_t * lenp,loff_t * ppos)1904 proc_do_sync_ports(const struct ctl_table *table, int write,
1905 		   void *buffer, size_t *lenp, loff_t *ppos)
1906 {
1907 	int *valp = table->data;
1908 	int val = *valp;
1909 	int rc;
1910 
1911 	struct ctl_table tmp = {
1912 		.data = &val,
1913 		.maxlen = sizeof(int),
1914 		.mode = table->mode,
1915 	};
1916 
1917 	rc = proc_dointvec(&tmp, write, buffer, lenp, ppos);
1918 	if (write && (*valp != val)) {
1919 		if (val < 1 || !is_power_of_2(val))
1920 			rc = -EINVAL;
1921 		else
1922 			*valp = val;
1923 	}
1924 	return rc;
1925 }
1926 
ipvs_proc_est_cpumask_set(const struct ctl_table * table,void * buffer)1927 static int ipvs_proc_est_cpumask_set(const struct ctl_table *table,
1928 				     void *buffer)
1929 {
1930 	struct netns_ipvs *ipvs = table->extra2;
1931 	cpumask_var_t *valp = table->data;
1932 	cpumask_var_t newmask;
1933 	int ret;
1934 
1935 	if (!zalloc_cpumask_var(&newmask, GFP_KERNEL))
1936 		return -ENOMEM;
1937 
1938 	ret = cpulist_parse(buffer, newmask);
1939 	if (ret)
1940 		goto out;
1941 
1942 	mutex_lock(&ipvs->est_mutex);
1943 
1944 	if (!ipvs->est_cpulist_valid) {
1945 		if (!zalloc_cpumask_var(valp, GFP_KERNEL)) {
1946 			ret = -ENOMEM;
1947 			goto unlock;
1948 		}
1949 		ipvs->est_cpulist_valid = 1;
1950 	}
1951 	cpumask_and(newmask, newmask, &current->cpus_mask);
1952 	cpumask_copy(*valp, newmask);
1953 	/* est_max_threads may depend on cpulist size */
1954 	ipvs->est_max_threads = ip_vs_est_max_threads(ipvs);
1955 	ipvs->est_calc_phase = 1;
1956 	ip_vs_est_reload_start(ipvs);
1957 
1958 unlock:
1959 	mutex_unlock(&ipvs->est_mutex);
1960 
1961 out:
1962 	free_cpumask_var(newmask);
1963 	return ret;
1964 }
1965 
ipvs_proc_est_cpumask_get(const struct ctl_table * table,void * buffer,size_t size)1966 static int ipvs_proc_est_cpumask_get(const struct ctl_table *table,
1967 				     void *buffer, size_t size)
1968 {
1969 	struct netns_ipvs *ipvs = table->extra2;
1970 	cpumask_var_t *valp = table->data;
1971 	struct cpumask *mask;
1972 	int ret;
1973 
1974 	mutex_lock(&ipvs->est_mutex);
1975 
1976 	if (ipvs->est_cpulist_valid)
1977 		mask = *valp;
1978 	else
1979 		mask = (struct cpumask *)housekeeping_cpumask(HK_TYPE_KTHREAD);
1980 	ret = scnprintf(buffer, size, "%*pbl\n", cpumask_pr_args(mask));
1981 
1982 	mutex_unlock(&ipvs->est_mutex);
1983 
1984 	return ret;
1985 }
1986 
ipvs_proc_est_cpulist(const struct ctl_table * table,int write,void * buffer,size_t * lenp,loff_t * ppos)1987 static int ipvs_proc_est_cpulist(const struct ctl_table *table, int write,
1988 				 void *buffer, size_t *lenp, loff_t *ppos)
1989 {
1990 	int ret;
1991 
1992 	/* Ignore both read and write(append) if *ppos not 0 */
1993 	if (*ppos || !*lenp) {
1994 		*lenp = 0;
1995 		return 0;
1996 	}
1997 	if (write) {
1998 		/* proc_sys_call_handler() appends terminator */
1999 		ret = ipvs_proc_est_cpumask_set(table, buffer);
2000 		if (ret >= 0)
2001 			*ppos += *lenp;
2002 	} else {
2003 		/* proc_sys_call_handler() allocates 1 byte for terminator */
2004 		ret = ipvs_proc_est_cpumask_get(table, buffer, *lenp + 1);
2005 		if (ret >= 0) {
2006 			*lenp = ret;
2007 			*ppos += *lenp;
2008 			ret = 0;
2009 		}
2010 	}
2011 	return ret;
2012 }
2013 
ipvs_proc_est_nice(const struct ctl_table * table,int write,void * buffer,size_t * lenp,loff_t * ppos)2014 static int ipvs_proc_est_nice(const struct ctl_table *table, int write,
2015 			      void *buffer, size_t *lenp, loff_t *ppos)
2016 {
2017 	struct netns_ipvs *ipvs = table->extra2;
2018 	int *valp = table->data;
2019 	int val = *valp;
2020 	int ret;
2021 
2022 	struct ctl_table tmp_table = {
2023 		.data = &val,
2024 		.maxlen = sizeof(int),
2025 		.mode = table->mode,
2026 	};
2027 
2028 	ret = proc_dointvec(&tmp_table, write, buffer, lenp, ppos);
2029 	if (write && ret >= 0) {
2030 		if (val < MIN_NICE || val > MAX_NICE) {
2031 			ret = -EINVAL;
2032 		} else {
2033 			mutex_lock(&ipvs->est_mutex);
2034 			if (*valp != val) {
2035 				*valp = val;
2036 				ip_vs_est_reload_start(ipvs);
2037 			}
2038 			mutex_unlock(&ipvs->est_mutex);
2039 		}
2040 	}
2041 	return ret;
2042 }
2043 
ipvs_proc_run_estimation(const struct ctl_table * table,int write,void * buffer,size_t * lenp,loff_t * ppos)2044 static int ipvs_proc_run_estimation(const struct ctl_table *table, int write,
2045 				    void *buffer, size_t *lenp, loff_t *ppos)
2046 {
2047 	struct netns_ipvs *ipvs = table->extra2;
2048 	int *valp = table->data;
2049 	int val = *valp;
2050 	int ret;
2051 
2052 	struct ctl_table tmp_table = {
2053 		.data = &val,
2054 		.maxlen = sizeof(int),
2055 		.mode = table->mode,
2056 	};
2057 
2058 	ret = proc_dointvec(&tmp_table, write, buffer, lenp, ppos);
2059 	if (write && ret >= 0) {
2060 		mutex_lock(&ipvs->est_mutex);
2061 		if (*valp != val) {
2062 			*valp = val;
2063 			ip_vs_est_reload_start(ipvs);
2064 		}
2065 		mutex_unlock(&ipvs->est_mutex);
2066 	}
2067 	return ret;
2068 }
2069 
2070 /*
2071  *	IPVS sysctl table (under the /proc/sys/net/ipv4/vs/)
2072  *	Do not change order or insert new entries without
2073  *	align with netns init in ip_vs_control_net_init()
2074  */
2075 
2076 static struct ctl_table vs_vars[] = {
2077 	{
2078 		.procname	= "amemthresh",
2079 		.maxlen		= sizeof(int),
2080 		.mode		= 0644,
2081 		.proc_handler	= proc_dointvec,
2082 	},
2083 	{
2084 		.procname	= "am_droprate",
2085 		.maxlen		= sizeof(int),
2086 		.mode		= 0644,
2087 		.proc_handler	= proc_dointvec,
2088 	},
2089 	{
2090 		.procname	= "drop_entry",
2091 		.maxlen		= sizeof(int),
2092 		.mode		= 0644,
2093 		.proc_handler	= proc_do_defense_mode,
2094 	},
2095 	{
2096 		.procname	= "drop_packet",
2097 		.maxlen		= sizeof(int),
2098 		.mode		= 0644,
2099 		.proc_handler	= proc_do_defense_mode,
2100 	},
2101 #ifdef CONFIG_IP_VS_NFCT
2102 	{
2103 		.procname	= "conntrack",
2104 		.maxlen		= sizeof(int),
2105 		.mode		= 0644,
2106 		.proc_handler	= &proc_dointvec,
2107 	},
2108 #endif
2109 	{
2110 		.procname	= "secure_tcp",
2111 		.maxlen		= sizeof(int),
2112 		.mode		= 0644,
2113 		.proc_handler	= proc_do_defense_mode,
2114 	},
2115 	{
2116 		.procname	= "snat_reroute",
2117 		.maxlen		= sizeof(int),
2118 		.mode		= 0644,
2119 		.proc_handler	= &proc_dointvec,
2120 	},
2121 	{
2122 		.procname	= "sync_version",
2123 		.maxlen		= sizeof(int),
2124 		.mode		= 0644,
2125 		.proc_handler	= proc_dointvec_minmax,
2126 		.extra1		= SYSCTL_ZERO,
2127 		.extra2		= SYSCTL_ONE,
2128 	},
2129 	{
2130 		.procname	= "sync_ports",
2131 		.maxlen		= sizeof(int),
2132 		.mode		= 0644,
2133 		.proc_handler	= proc_do_sync_ports,
2134 	},
2135 	{
2136 		.procname	= "sync_persist_mode",
2137 		.maxlen		= sizeof(int),
2138 		.mode		= 0644,
2139 		.proc_handler	= proc_dointvec,
2140 	},
2141 	{
2142 		.procname	= "sync_qlen_max",
2143 		.maxlen		= sizeof(unsigned long),
2144 		.mode		= 0644,
2145 		.proc_handler	= proc_doulongvec_minmax,
2146 	},
2147 	{
2148 		.procname	= "sync_sock_size",
2149 		.maxlen		= sizeof(int),
2150 		.mode		= 0644,
2151 		.proc_handler	= proc_dointvec,
2152 	},
2153 	{
2154 		.procname	= "cache_bypass",
2155 		.maxlen		= sizeof(int),
2156 		.mode		= 0644,
2157 		.proc_handler	= proc_dointvec,
2158 	},
2159 	{
2160 		.procname	= "expire_nodest_conn",
2161 		.maxlen		= sizeof(int),
2162 		.mode		= 0644,
2163 		.proc_handler	= proc_dointvec,
2164 	},
2165 	{
2166 		.procname	= "sloppy_tcp",
2167 		.maxlen		= sizeof(int),
2168 		.mode		= 0644,
2169 		.proc_handler	= proc_dointvec,
2170 	},
2171 	{
2172 		.procname	= "sloppy_sctp",
2173 		.maxlen		= sizeof(int),
2174 		.mode		= 0644,
2175 		.proc_handler	= proc_dointvec,
2176 	},
2177 	{
2178 		.procname	= "expire_quiescent_template",
2179 		.maxlen		= sizeof(int),
2180 		.mode		= 0644,
2181 		.proc_handler	= proc_dointvec,
2182 	},
2183 	{
2184 		.procname	= "sync_threshold",
2185 		.maxlen		=
2186 			sizeof(((struct netns_ipvs *)0)->sysctl_sync_threshold),
2187 		.mode		= 0644,
2188 		.proc_handler	= proc_do_sync_threshold,
2189 	},
2190 	{
2191 		.procname	= "sync_refresh_period",
2192 		.maxlen		= sizeof(int),
2193 		.mode		= 0644,
2194 		.proc_handler	= proc_dointvec_jiffies,
2195 	},
2196 	{
2197 		.procname	= "sync_retries",
2198 		.maxlen		= sizeof(int),
2199 		.mode		= 0644,
2200 		.proc_handler	= proc_dointvec_minmax,
2201 		.extra1		= SYSCTL_ZERO,
2202 		.extra2		= SYSCTL_THREE,
2203 	},
2204 	{
2205 		.procname	= "nat_icmp_send",
2206 		.maxlen		= sizeof(int),
2207 		.mode		= 0644,
2208 		.proc_handler	= proc_dointvec,
2209 	},
2210 	{
2211 		.procname	= "pmtu_disc",
2212 		.maxlen		= sizeof(int),
2213 		.mode		= 0644,
2214 		.proc_handler	= proc_dointvec,
2215 	},
2216 	{
2217 		.procname	= "backup_only",
2218 		.maxlen		= sizeof(int),
2219 		.mode		= 0644,
2220 		.proc_handler	= proc_dointvec,
2221 	},
2222 	{
2223 		.procname	= "conn_reuse_mode",
2224 		.maxlen		= sizeof(int),
2225 		.mode		= 0644,
2226 		.proc_handler	= proc_dointvec,
2227 	},
2228 	{
2229 		.procname	= "schedule_icmp",
2230 		.maxlen		= sizeof(int),
2231 		.mode		= 0644,
2232 		.proc_handler	= proc_dointvec,
2233 	},
2234 	{
2235 		.procname	= "ignore_tunneled",
2236 		.maxlen		= sizeof(int),
2237 		.mode		= 0644,
2238 		.proc_handler	= proc_dointvec,
2239 	},
2240 	{
2241 		.procname	= "run_estimation",
2242 		.maxlen		= sizeof(int),
2243 		.mode		= 0644,
2244 		.proc_handler	= ipvs_proc_run_estimation,
2245 	},
2246 	{
2247 		.procname	= "est_cpulist",
2248 		.maxlen		= NR_CPUS,	/* unused */
2249 		.mode		= 0644,
2250 		.proc_handler	= ipvs_proc_est_cpulist,
2251 	},
2252 	{
2253 		.procname	= "est_nice",
2254 		.maxlen		= sizeof(int),
2255 		.mode		= 0644,
2256 		.proc_handler	= ipvs_proc_est_nice,
2257 	},
2258 #ifdef CONFIG_IP_VS_DEBUG
2259 	{
2260 		.procname	= "debug_level",
2261 		.data		= &sysctl_ip_vs_debug_level,
2262 		.maxlen		= sizeof(int),
2263 		.mode		= 0644,
2264 		.proc_handler	= proc_dointvec,
2265 	},
2266 #endif
2267 };
2268 
2269 #endif
2270 
2271 #ifdef CONFIG_PROC_FS
2272 
2273 struct ip_vs_iter {
2274 	struct seq_net_private p;  /* Do not move this, netns depends upon it*/
2275 	struct hlist_head *table;
2276 	int bucket;
2277 };
2278 
2279 /*
2280  *	Write the contents of the VS rule table to a PROCfs file.
2281  *	(It is kept just for backward compatibility)
2282  */
ip_vs_fwd_name(unsigned int flags)2283 static inline const char *ip_vs_fwd_name(unsigned int flags)
2284 {
2285 	switch (flags & IP_VS_CONN_F_FWD_MASK) {
2286 	case IP_VS_CONN_F_LOCALNODE:
2287 		return "Local";
2288 	case IP_VS_CONN_F_TUNNEL:
2289 		return "Tunnel";
2290 	case IP_VS_CONN_F_DROUTE:
2291 		return "Route";
2292 	default:
2293 		return "Masq";
2294 	}
2295 }
2296 
2297 
2298 /* Get the Nth entry in the two lists */
ip_vs_info_array(struct seq_file * seq,loff_t pos)2299 static struct ip_vs_service *ip_vs_info_array(struct seq_file *seq, loff_t pos)
2300 {
2301 	struct net *net = seq_file_net(seq);
2302 	struct netns_ipvs *ipvs = net_ipvs(net);
2303 	struct ip_vs_iter *iter = seq->private;
2304 	int idx;
2305 	struct ip_vs_service *svc;
2306 
2307 	/* look in hash by protocol */
2308 	for (idx = 0; idx < IP_VS_SVC_TAB_SIZE; idx++) {
2309 		hlist_for_each_entry_rcu(svc, &ip_vs_svc_table[idx], s_list) {
2310 			if ((svc->ipvs == ipvs) && pos-- == 0) {
2311 				iter->table = ip_vs_svc_table;
2312 				iter->bucket = idx;
2313 				return svc;
2314 			}
2315 		}
2316 	}
2317 
2318 	/* keep looking in fwmark */
2319 	for (idx = 0; idx < IP_VS_SVC_TAB_SIZE; idx++) {
2320 		hlist_for_each_entry_rcu(svc, &ip_vs_svc_fwm_table[idx],
2321 					 f_list) {
2322 			if ((svc->ipvs == ipvs) && pos-- == 0) {
2323 				iter->table = ip_vs_svc_fwm_table;
2324 				iter->bucket = idx;
2325 				return svc;
2326 			}
2327 		}
2328 	}
2329 
2330 	return NULL;
2331 }
2332 
ip_vs_info_seq_start(struct seq_file * seq,loff_t * pos)2333 static void *ip_vs_info_seq_start(struct seq_file *seq, loff_t *pos)
2334 	__acquires(RCU)
2335 {
2336 	rcu_read_lock();
2337 	return *pos ? ip_vs_info_array(seq, *pos - 1) : SEQ_START_TOKEN;
2338 }
2339 
2340 
ip_vs_info_seq_next(struct seq_file * seq,void * v,loff_t * pos)2341 static void *ip_vs_info_seq_next(struct seq_file *seq, void *v, loff_t *pos)
2342 {
2343 	struct hlist_node *e;
2344 	struct ip_vs_iter *iter;
2345 	struct ip_vs_service *svc;
2346 
2347 	++*pos;
2348 	if (v == SEQ_START_TOKEN)
2349 		return ip_vs_info_array(seq,0);
2350 
2351 	svc = v;
2352 	iter = seq->private;
2353 
2354 	if (iter->table == ip_vs_svc_table) {
2355 		/* next service in table hashed by protocol */
2356 		e = rcu_dereference(hlist_next_rcu(&svc->s_list));
2357 		if (e)
2358 			return hlist_entry(e, struct ip_vs_service, s_list);
2359 
2360 		while (++iter->bucket < IP_VS_SVC_TAB_SIZE) {
2361 			hlist_for_each_entry_rcu(svc,
2362 						 &ip_vs_svc_table[iter->bucket],
2363 						 s_list) {
2364 				return svc;
2365 			}
2366 		}
2367 
2368 		iter->table = ip_vs_svc_fwm_table;
2369 		iter->bucket = -1;
2370 		goto scan_fwmark;
2371 	}
2372 
2373 	/* next service in hashed by fwmark */
2374 	e = rcu_dereference(hlist_next_rcu(&svc->f_list));
2375 	if (e)
2376 		return hlist_entry(e, struct ip_vs_service, f_list);
2377 
2378  scan_fwmark:
2379 	while (++iter->bucket < IP_VS_SVC_TAB_SIZE) {
2380 		hlist_for_each_entry_rcu(svc,
2381 					 &ip_vs_svc_fwm_table[iter->bucket],
2382 					 f_list)
2383 			return svc;
2384 	}
2385 
2386 	return NULL;
2387 }
2388 
ip_vs_info_seq_stop(struct seq_file * seq,void * v)2389 static void ip_vs_info_seq_stop(struct seq_file *seq, void *v)
2390 	__releases(RCU)
2391 {
2392 	rcu_read_unlock();
2393 }
2394 
2395 
ip_vs_info_seq_show(struct seq_file * seq,void * v)2396 static int ip_vs_info_seq_show(struct seq_file *seq, void *v)
2397 {
2398 	if (v == SEQ_START_TOKEN) {
2399 		seq_printf(seq,
2400 			"IP Virtual Server version %d.%d.%d (size=%d)\n",
2401 			NVERSION(IP_VS_VERSION_CODE), ip_vs_conn_tab_size);
2402 		seq_puts(seq,
2403 			 "Prot LocalAddress:Port Scheduler Flags\n");
2404 		seq_puts(seq,
2405 			 "  -> RemoteAddress:Port Forward Weight ActiveConn InActConn\n");
2406 	} else {
2407 		struct net *net = seq_file_net(seq);
2408 		struct netns_ipvs *ipvs = net_ipvs(net);
2409 		const struct ip_vs_service *svc = v;
2410 		const struct ip_vs_iter *iter = seq->private;
2411 		const struct ip_vs_dest *dest;
2412 		struct ip_vs_scheduler *sched = rcu_dereference(svc->scheduler);
2413 		char *sched_name = sched ? sched->name : "none";
2414 
2415 		if (svc->ipvs != ipvs)
2416 			return 0;
2417 		if (iter->table == ip_vs_svc_table) {
2418 #ifdef CONFIG_IP_VS_IPV6
2419 			if (svc->af == AF_INET6)
2420 				seq_printf(seq, "%s  [%pI6]:%04X %s ",
2421 					   ip_vs_proto_name(svc->protocol),
2422 					   &svc->addr.in6,
2423 					   ntohs(svc->port),
2424 					   sched_name);
2425 			else
2426 #endif
2427 				seq_printf(seq, "%s  %08X:%04X %s %s ",
2428 					   ip_vs_proto_name(svc->protocol),
2429 					   ntohl(svc->addr.ip),
2430 					   ntohs(svc->port),
2431 					   sched_name,
2432 					   (svc->flags & IP_VS_SVC_F_ONEPACKET)?"ops ":"");
2433 		} else {
2434 			seq_printf(seq, "FWM  %08X %s %s",
2435 				   svc->fwmark, sched_name,
2436 				   (svc->flags & IP_VS_SVC_F_ONEPACKET)?"ops ":"");
2437 		}
2438 
2439 		if (svc->flags & IP_VS_SVC_F_PERSISTENT)
2440 			seq_printf(seq, "persistent %d %08X\n",
2441 				svc->timeout,
2442 				ntohl(svc->netmask));
2443 		else
2444 			seq_putc(seq, '\n');
2445 
2446 		list_for_each_entry_rcu(dest, &svc->destinations, n_list) {
2447 #ifdef CONFIG_IP_VS_IPV6
2448 			if (dest->af == AF_INET6)
2449 				seq_printf(seq,
2450 					   "  -> [%pI6]:%04X"
2451 					   "      %-7s %-6d %-10d %-10d\n",
2452 					   &dest->addr.in6,
2453 					   ntohs(dest->port),
2454 					   ip_vs_fwd_name(atomic_read(&dest->conn_flags)),
2455 					   atomic_read(&dest->weight),
2456 					   atomic_read(&dest->activeconns),
2457 					   atomic_read(&dest->inactconns));
2458 			else
2459 #endif
2460 				seq_printf(seq,
2461 					   "  -> %08X:%04X      "
2462 					   "%-7s %-6d %-10d %-10d\n",
2463 					   ntohl(dest->addr.ip),
2464 					   ntohs(dest->port),
2465 					   ip_vs_fwd_name(atomic_read(&dest->conn_flags)),
2466 					   atomic_read(&dest->weight),
2467 					   atomic_read(&dest->activeconns),
2468 					   atomic_read(&dest->inactconns));
2469 
2470 		}
2471 	}
2472 	return 0;
2473 }
2474 
2475 static const struct seq_operations ip_vs_info_seq_ops = {
2476 	.start = ip_vs_info_seq_start,
2477 	.next  = ip_vs_info_seq_next,
2478 	.stop  = ip_vs_info_seq_stop,
2479 	.show  = ip_vs_info_seq_show,
2480 };
2481 
ip_vs_stats_show(struct seq_file * seq,void * v)2482 static int ip_vs_stats_show(struct seq_file *seq, void *v)
2483 {
2484 	struct net *net = seq_file_single_net(seq);
2485 	struct ip_vs_kstats show;
2486 
2487 /*               01234567 01234567 01234567 0123456701234567 0123456701234567 */
2488 	seq_puts(seq,
2489 		 "   Total Incoming Outgoing         Incoming         Outgoing\n");
2490 	seq_puts(seq,
2491 		 "   Conns  Packets  Packets            Bytes            Bytes\n");
2492 
2493 	ip_vs_copy_stats(&show, &net_ipvs(net)->tot_stats->s);
2494 	seq_printf(seq, "%8LX %8LX %8LX %16LX %16LX\n\n",
2495 		   (unsigned long long)show.conns,
2496 		   (unsigned long long)show.inpkts,
2497 		   (unsigned long long)show.outpkts,
2498 		   (unsigned long long)show.inbytes,
2499 		   (unsigned long long)show.outbytes);
2500 
2501 /*                01234567 01234567 01234567 0123456701234567 0123456701234567*/
2502 	seq_puts(seq,
2503 		 " Conns/s   Pkts/s   Pkts/s          Bytes/s          Bytes/s\n");
2504 	seq_printf(seq, "%8LX %8LX %8LX %16LX %16LX\n",
2505 		   (unsigned long long)show.cps,
2506 		   (unsigned long long)show.inpps,
2507 		   (unsigned long long)show.outpps,
2508 		   (unsigned long long)show.inbps,
2509 		   (unsigned long long)show.outbps);
2510 
2511 	return 0;
2512 }
2513 
ip_vs_stats_percpu_show(struct seq_file * seq,void * v)2514 static int ip_vs_stats_percpu_show(struct seq_file *seq, void *v)
2515 {
2516 	struct net *net = seq_file_single_net(seq);
2517 	struct ip_vs_stats *tot_stats = &net_ipvs(net)->tot_stats->s;
2518 	struct ip_vs_cpu_stats __percpu *cpustats = tot_stats->cpustats;
2519 	struct ip_vs_kstats kstats;
2520 	int i;
2521 
2522 /*               01234567 01234567 01234567 0123456701234567 0123456701234567 */
2523 	seq_puts(seq,
2524 		 "       Total Incoming Outgoing         Incoming         Outgoing\n");
2525 	seq_puts(seq,
2526 		 "CPU    Conns  Packets  Packets            Bytes            Bytes\n");
2527 
2528 	for_each_possible_cpu(i) {
2529 		struct ip_vs_cpu_stats *u = per_cpu_ptr(cpustats, i);
2530 		unsigned int start;
2531 		u64 conns, inpkts, outpkts, inbytes, outbytes;
2532 
2533 		do {
2534 			start = u64_stats_fetch_begin(&u->syncp);
2535 			conns = u64_stats_read(&u->cnt.conns);
2536 			inpkts = u64_stats_read(&u->cnt.inpkts);
2537 			outpkts = u64_stats_read(&u->cnt.outpkts);
2538 			inbytes = u64_stats_read(&u->cnt.inbytes);
2539 			outbytes = u64_stats_read(&u->cnt.outbytes);
2540 		} while (u64_stats_fetch_retry(&u->syncp, start));
2541 
2542 		seq_printf(seq, "%3X %8LX %8LX %8LX %16LX %16LX\n",
2543 			   i, (u64)conns, (u64)inpkts,
2544 			   (u64)outpkts, (u64)inbytes,
2545 			   (u64)outbytes);
2546 	}
2547 
2548 	ip_vs_copy_stats(&kstats, tot_stats);
2549 
2550 	seq_printf(seq, "  ~ %8LX %8LX %8LX %16LX %16LX\n\n",
2551 		   (unsigned long long)kstats.conns,
2552 		   (unsigned long long)kstats.inpkts,
2553 		   (unsigned long long)kstats.outpkts,
2554 		   (unsigned long long)kstats.inbytes,
2555 		   (unsigned long long)kstats.outbytes);
2556 
2557 /*                ... 01234567 01234567 01234567 0123456701234567 0123456701234567 */
2558 	seq_puts(seq,
2559 		 "     Conns/s   Pkts/s   Pkts/s          Bytes/s          Bytes/s\n");
2560 	seq_printf(seq, "    %8LX %8LX %8LX %16LX %16LX\n",
2561 		   kstats.cps,
2562 		   kstats.inpps,
2563 		   kstats.outpps,
2564 		   kstats.inbps,
2565 		   kstats.outbps);
2566 
2567 	return 0;
2568 }
2569 #endif
2570 
2571 /*
2572  *	Set timeout values for tcp tcpfin udp in the timeout_table.
2573  */
ip_vs_set_timeout(struct netns_ipvs * ipvs,struct ip_vs_timeout_user * u)2574 static int ip_vs_set_timeout(struct netns_ipvs *ipvs, struct ip_vs_timeout_user *u)
2575 {
2576 #if defined(CONFIG_IP_VS_PROTO_TCP) || defined(CONFIG_IP_VS_PROTO_UDP)
2577 	struct ip_vs_proto_data *pd;
2578 #endif
2579 
2580 	IP_VS_DBG(2, "Setting timeout tcp:%d tcpfin:%d udp:%d\n",
2581 		  u->tcp_timeout,
2582 		  u->tcp_fin_timeout,
2583 		  u->udp_timeout);
2584 
2585 #ifdef CONFIG_IP_VS_PROTO_TCP
2586 	if (u->tcp_timeout < 0 || u->tcp_timeout > (INT_MAX / HZ) ||
2587 	    u->tcp_fin_timeout < 0 || u->tcp_fin_timeout > (INT_MAX / HZ)) {
2588 		return -EINVAL;
2589 	}
2590 #endif
2591 
2592 #ifdef CONFIG_IP_VS_PROTO_UDP
2593 	if (u->udp_timeout < 0 || u->udp_timeout > (INT_MAX / HZ))
2594 		return -EINVAL;
2595 #endif
2596 
2597 #ifdef CONFIG_IP_VS_PROTO_TCP
2598 	if (u->tcp_timeout) {
2599 		pd = ip_vs_proto_data_get(ipvs, IPPROTO_TCP);
2600 		pd->timeout_table[IP_VS_TCP_S_ESTABLISHED]
2601 			= u->tcp_timeout * HZ;
2602 	}
2603 
2604 	if (u->tcp_fin_timeout) {
2605 		pd = ip_vs_proto_data_get(ipvs, IPPROTO_TCP);
2606 		pd->timeout_table[IP_VS_TCP_S_FIN_WAIT]
2607 			= u->tcp_fin_timeout * HZ;
2608 	}
2609 #endif
2610 
2611 #ifdef CONFIG_IP_VS_PROTO_UDP
2612 	if (u->udp_timeout) {
2613 		pd = ip_vs_proto_data_get(ipvs, IPPROTO_UDP);
2614 		pd->timeout_table[IP_VS_UDP_S_NORMAL]
2615 			= u->udp_timeout * HZ;
2616 	}
2617 #endif
2618 	return 0;
2619 }
2620 
2621 #define CMDID(cmd)		(cmd - IP_VS_BASE_CTL)
2622 
2623 struct ip_vs_svcdest_user {
2624 	struct ip_vs_service_user	s;
2625 	struct ip_vs_dest_user		d;
2626 };
2627 
2628 static const unsigned char set_arglen[CMDID(IP_VS_SO_SET_MAX) + 1] = {
2629 	[CMDID(IP_VS_SO_SET_ADD)]         = sizeof(struct ip_vs_service_user),
2630 	[CMDID(IP_VS_SO_SET_EDIT)]        = sizeof(struct ip_vs_service_user),
2631 	[CMDID(IP_VS_SO_SET_DEL)]         = sizeof(struct ip_vs_service_user),
2632 	[CMDID(IP_VS_SO_SET_ADDDEST)]     = sizeof(struct ip_vs_svcdest_user),
2633 	[CMDID(IP_VS_SO_SET_DELDEST)]     = sizeof(struct ip_vs_svcdest_user),
2634 	[CMDID(IP_VS_SO_SET_EDITDEST)]    = sizeof(struct ip_vs_svcdest_user),
2635 	[CMDID(IP_VS_SO_SET_TIMEOUT)]     = sizeof(struct ip_vs_timeout_user),
2636 	[CMDID(IP_VS_SO_SET_STARTDAEMON)] = sizeof(struct ip_vs_daemon_user),
2637 	[CMDID(IP_VS_SO_SET_STOPDAEMON)]  = sizeof(struct ip_vs_daemon_user),
2638 	[CMDID(IP_VS_SO_SET_ZERO)]        = sizeof(struct ip_vs_service_user),
2639 };
2640 
2641 union ip_vs_set_arglen {
2642 	struct ip_vs_service_user	field_IP_VS_SO_SET_ADD;
2643 	struct ip_vs_service_user	field_IP_VS_SO_SET_EDIT;
2644 	struct ip_vs_service_user	field_IP_VS_SO_SET_DEL;
2645 	struct ip_vs_svcdest_user	field_IP_VS_SO_SET_ADDDEST;
2646 	struct ip_vs_svcdest_user	field_IP_VS_SO_SET_DELDEST;
2647 	struct ip_vs_svcdest_user	field_IP_VS_SO_SET_EDITDEST;
2648 	struct ip_vs_timeout_user	field_IP_VS_SO_SET_TIMEOUT;
2649 	struct ip_vs_daemon_user	field_IP_VS_SO_SET_STARTDAEMON;
2650 	struct ip_vs_daemon_user	field_IP_VS_SO_SET_STOPDAEMON;
2651 	struct ip_vs_service_user	field_IP_VS_SO_SET_ZERO;
2652 };
2653 
2654 #define MAX_SET_ARGLEN	sizeof(union ip_vs_set_arglen)
2655 
ip_vs_copy_usvc_compat(struct ip_vs_service_user_kern * usvc,struct ip_vs_service_user * usvc_compat)2656 static void ip_vs_copy_usvc_compat(struct ip_vs_service_user_kern *usvc,
2657 				  struct ip_vs_service_user *usvc_compat)
2658 {
2659 	memset(usvc, 0, sizeof(*usvc));
2660 
2661 	usvc->af		= AF_INET;
2662 	usvc->protocol		= usvc_compat->protocol;
2663 	usvc->addr.ip		= usvc_compat->addr;
2664 	usvc->port		= usvc_compat->port;
2665 	usvc->fwmark		= usvc_compat->fwmark;
2666 
2667 	/* Deep copy of sched_name is not needed here */
2668 	usvc->sched_name	= usvc_compat->sched_name;
2669 
2670 	usvc->flags		= usvc_compat->flags;
2671 	usvc->timeout		= usvc_compat->timeout;
2672 	usvc->netmask		= usvc_compat->netmask;
2673 }
2674 
ip_vs_copy_udest_compat(struct ip_vs_dest_user_kern * udest,struct ip_vs_dest_user * udest_compat)2675 static void ip_vs_copy_udest_compat(struct ip_vs_dest_user_kern *udest,
2676 				   struct ip_vs_dest_user *udest_compat)
2677 {
2678 	memset(udest, 0, sizeof(*udest));
2679 
2680 	udest->addr.ip		= udest_compat->addr;
2681 	udest->port		= udest_compat->port;
2682 	udest->conn_flags	= udest_compat->conn_flags;
2683 	udest->weight		= udest_compat->weight;
2684 	udest->u_threshold	= udest_compat->u_threshold;
2685 	udest->l_threshold	= udest_compat->l_threshold;
2686 	udest->af		= AF_INET;
2687 	udest->tun_type		= IP_VS_CONN_F_TUNNEL_TYPE_IPIP;
2688 }
2689 
2690 static int
do_ip_vs_set_ctl(struct sock * sk,int cmd,sockptr_t ptr,unsigned int len)2691 do_ip_vs_set_ctl(struct sock *sk, int cmd, sockptr_t ptr, unsigned int len)
2692 {
2693 	struct net *net = sock_net(sk);
2694 	int ret;
2695 	unsigned char arg[MAX_SET_ARGLEN];
2696 	struct ip_vs_service_user *usvc_compat;
2697 	struct ip_vs_service_user_kern usvc;
2698 	struct ip_vs_service *svc;
2699 	struct ip_vs_dest_user *udest_compat;
2700 	struct ip_vs_dest_user_kern udest;
2701 	struct netns_ipvs *ipvs = net_ipvs(net);
2702 
2703 	BUILD_BUG_ON(sizeof(arg) > 255);
2704 	if (!ns_capable(sock_net(sk)->user_ns, CAP_NET_ADMIN))
2705 		return -EPERM;
2706 
2707 	if (cmd < IP_VS_BASE_CTL || cmd > IP_VS_SO_SET_MAX)
2708 		return -EINVAL;
2709 	if (len != set_arglen[CMDID(cmd)]) {
2710 		IP_VS_DBG(1, "set_ctl: len %u != %u\n",
2711 			  len, set_arglen[CMDID(cmd)]);
2712 		return -EINVAL;
2713 	}
2714 
2715 	if (copy_from_sockptr(arg, ptr, len) != 0)
2716 		return -EFAULT;
2717 
2718 	/* Handle daemons since they have another lock */
2719 	if (cmd == IP_VS_SO_SET_STARTDAEMON ||
2720 	    cmd == IP_VS_SO_SET_STOPDAEMON) {
2721 		struct ip_vs_daemon_user *dm = (struct ip_vs_daemon_user *)arg;
2722 
2723 		if (cmd == IP_VS_SO_SET_STARTDAEMON) {
2724 			struct ipvs_sync_daemon_cfg cfg;
2725 
2726 			memset(&cfg, 0, sizeof(cfg));
2727 			ret = -EINVAL;
2728 			if (strscpy(cfg.mcast_ifn, dm->mcast_ifn,
2729 				    sizeof(cfg.mcast_ifn)) <= 0)
2730 				return ret;
2731 			cfg.syncid = dm->syncid;
2732 			ret = start_sync_thread(ipvs, &cfg, dm->state);
2733 		} else {
2734 			ret = stop_sync_thread(ipvs, dm->state);
2735 		}
2736 		return ret;
2737 	}
2738 
2739 	mutex_lock(&__ip_vs_mutex);
2740 	if (cmd == IP_VS_SO_SET_FLUSH) {
2741 		/* Flush the virtual service */
2742 		ret = ip_vs_flush(ipvs, false);
2743 		goto out_unlock;
2744 	} else if (cmd == IP_VS_SO_SET_TIMEOUT) {
2745 		/* Set timeout values for (tcp tcpfin udp) */
2746 		ret = ip_vs_set_timeout(ipvs, (struct ip_vs_timeout_user *)arg);
2747 		goto out_unlock;
2748 	} else if (!len) {
2749 		/* No more commands with len == 0 below */
2750 		ret = -EINVAL;
2751 		goto out_unlock;
2752 	}
2753 
2754 	usvc_compat = (struct ip_vs_service_user *)arg;
2755 	udest_compat = (struct ip_vs_dest_user *)(usvc_compat + 1);
2756 
2757 	/* We only use the new structs internally, so copy userspace compat
2758 	 * structs to extended internal versions */
2759 	ip_vs_copy_usvc_compat(&usvc, usvc_compat);
2760 	ip_vs_copy_udest_compat(&udest, udest_compat);
2761 
2762 	if (cmd == IP_VS_SO_SET_ZERO) {
2763 		/* if no service address is set, zero counters in all */
2764 		if (!usvc.fwmark && !usvc.addr.ip && !usvc.port) {
2765 			ret = ip_vs_zero_all(ipvs);
2766 			goto out_unlock;
2767 		}
2768 	}
2769 
2770 	if ((cmd == IP_VS_SO_SET_ADD || cmd == IP_VS_SO_SET_EDIT) &&
2771 	    strnlen(usvc.sched_name, IP_VS_SCHEDNAME_MAXLEN) ==
2772 	    IP_VS_SCHEDNAME_MAXLEN) {
2773 		ret = -EINVAL;
2774 		goto out_unlock;
2775 	}
2776 
2777 	/* Check for valid protocol: TCP or UDP or SCTP, even for fwmark!=0 */
2778 	if (usvc.protocol != IPPROTO_TCP && usvc.protocol != IPPROTO_UDP &&
2779 	    usvc.protocol != IPPROTO_SCTP) {
2780 		pr_err("set_ctl: invalid protocol: %d %pI4:%d\n",
2781 		       usvc.protocol, &usvc.addr.ip,
2782 		       ntohs(usvc.port));
2783 		ret = -EFAULT;
2784 		goto out_unlock;
2785 	}
2786 
2787 	/* Lookup the exact service by <protocol, addr, port> or fwmark */
2788 	rcu_read_lock();
2789 	if (usvc.fwmark == 0)
2790 		svc = __ip_vs_service_find(ipvs, usvc.af, usvc.protocol,
2791 					   &usvc.addr, usvc.port);
2792 	else
2793 		svc = __ip_vs_svc_fwm_find(ipvs, usvc.af, usvc.fwmark);
2794 	rcu_read_unlock();
2795 
2796 	if (cmd != IP_VS_SO_SET_ADD
2797 	    && (svc == NULL || svc->protocol != usvc.protocol)) {
2798 		ret = -ESRCH;
2799 		goto out_unlock;
2800 	}
2801 
2802 	switch (cmd) {
2803 	case IP_VS_SO_SET_ADD:
2804 		if (svc != NULL)
2805 			ret = -EEXIST;
2806 		else
2807 			ret = ip_vs_add_service(ipvs, &usvc, &svc);
2808 		break;
2809 	case IP_VS_SO_SET_EDIT:
2810 		ret = ip_vs_edit_service(svc, &usvc);
2811 		break;
2812 	case IP_VS_SO_SET_DEL:
2813 		ret = ip_vs_del_service(svc);
2814 		if (!ret)
2815 			goto out_unlock;
2816 		break;
2817 	case IP_VS_SO_SET_ZERO:
2818 		ret = ip_vs_zero_service(svc);
2819 		break;
2820 	case IP_VS_SO_SET_ADDDEST:
2821 		ret = ip_vs_add_dest(svc, &udest);
2822 		break;
2823 	case IP_VS_SO_SET_EDITDEST:
2824 		ret = ip_vs_edit_dest(svc, &udest);
2825 		break;
2826 	case IP_VS_SO_SET_DELDEST:
2827 		ret = ip_vs_del_dest(svc, &udest);
2828 		break;
2829 	default:
2830 		WARN_ON_ONCE(1);
2831 		ret = -EINVAL;
2832 		break;
2833 	}
2834 
2835   out_unlock:
2836 	mutex_unlock(&__ip_vs_mutex);
2837 	return ret;
2838 }
2839 
2840 
2841 static void
ip_vs_copy_service(struct ip_vs_service_entry * dst,struct ip_vs_service * src)2842 ip_vs_copy_service(struct ip_vs_service_entry *dst, struct ip_vs_service *src)
2843 {
2844 	struct ip_vs_scheduler *sched;
2845 	struct ip_vs_kstats kstats;
2846 	char *sched_name;
2847 
2848 	sched = rcu_dereference_protected(src->scheduler, 1);
2849 	sched_name = sched ? sched->name : "none";
2850 	dst->protocol = src->protocol;
2851 	dst->addr = src->addr.ip;
2852 	dst->port = src->port;
2853 	dst->fwmark = src->fwmark;
2854 	strscpy(dst->sched_name, sched_name, sizeof(dst->sched_name));
2855 	dst->flags = src->flags;
2856 	dst->timeout = src->timeout / HZ;
2857 	dst->netmask = src->netmask;
2858 	dst->num_dests = src->num_dests;
2859 	ip_vs_copy_stats(&kstats, &src->stats);
2860 	ip_vs_export_stats_user(&dst->stats, &kstats);
2861 }
2862 
2863 static inline int
__ip_vs_get_service_entries(struct netns_ipvs * ipvs,const struct ip_vs_get_services * get,struct ip_vs_get_services __user * uptr)2864 __ip_vs_get_service_entries(struct netns_ipvs *ipvs,
2865 			    const struct ip_vs_get_services *get,
2866 			    struct ip_vs_get_services __user *uptr)
2867 {
2868 	int idx, count=0;
2869 	struct ip_vs_service *svc;
2870 	struct ip_vs_service_entry entry;
2871 	int ret = 0;
2872 
2873 	for (idx = 0; idx < IP_VS_SVC_TAB_SIZE; idx++) {
2874 		hlist_for_each_entry(svc, &ip_vs_svc_table[idx], s_list) {
2875 			/* Only expose IPv4 entries to old interface */
2876 			if (svc->af != AF_INET || (svc->ipvs != ipvs))
2877 				continue;
2878 
2879 			if (count >= get->num_services)
2880 				goto out;
2881 			memset(&entry, 0, sizeof(entry));
2882 			ip_vs_copy_service(&entry, svc);
2883 			if (copy_to_user(&uptr->entrytable[count],
2884 					 &entry, sizeof(entry))) {
2885 				ret = -EFAULT;
2886 				goto out;
2887 			}
2888 			count++;
2889 		}
2890 	}
2891 
2892 	for (idx = 0; idx < IP_VS_SVC_TAB_SIZE; idx++) {
2893 		hlist_for_each_entry(svc, &ip_vs_svc_fwm_table[idx], f_list) {
2894 			/* Only expose IPv4 entries to old interface */
2895 			if (svc->af != AF_INET || (svc->ipvs != ipvs))
2896 				continue;
2897 
2898 			if (count >= get->num_services)
2899 				goto out;
2900 			memset(&entry, 0, sizeof(entry));
2901 			ip_vs_copy_service(&entry, svc);
2902 			if (copy_to_user(&uptr->entrytable[count],
2903 					 &entry, sizeof(entry))) {
2904 				ret = -EFAULT;
2905 				goto out;
2906 			}
2907 			count++;
2908 		}
2909 	}
2910 out:
2911 	return ret;
2912 }
2913 
2914 static inline int
__ip_vs_get_dest_entries(struct netns_ipvs * ipvs,const struct ip_vs_get_dests * get,struct ip_vs_get_dests __user * uptr)2915 __ip_vs_get_dest_entries(struct netns_ipvs *ipvs, const struct ip_vs_get_dests *get,
2916 			 struct ip_vs_get_dests __user *uptr)
2917 {
2918 	struct ip_vs_service *svc;
2919 	union nf_inet_addr addr = { .ip = get->addr };
2920 	int ret = 0;
2921 
2922 	rcu_read_lock();
2923 	if (get->fwmark)
2924 		svc = __ip_vs_svc_fwm_find(ipvs, AF_INET, get->fwmark);
2925 	else
2926 		svc = __ip_vs_service_find(ipvs, AF_INET, get->protocol, &addr,
2927 					   get->port);
2928 	rcu_read_unlock();
2929 
2930 	if (svc) {
2931 		int count = 0;
2932 		struct ip_vs_dest *dest;
2933 		struct ip_vs_dest_entry entry;
2934 		struct ip_vs_kstats kstats;
2935 
2936 		memset(&entry, 0, sizeof(entry));
2937 		list_for_each_entry(dest, &svc->destinations, n_list) {
2938 			if (count >= get->num_dests)
2939 				break;
2940 
2941 			/* Cannot expose heterogeneous members via sockopt
2942 			 * interface
2943 			 */
2944 			if (dest->af != svc->af)
2945 				continue;
2946 
2947 			entry.addr = dest->addr.ip;
2948 			entry.port = dest->port;
2949 			entry.conn_flags = atomic_read(&dest->conn_flags);
2950 			entry.weight = atomic_read(&dest->weight);
2951 			entry.u_threshold = dest->u_threshold;
2952 			entry.l_threshold = dest->l_threshold;
2953 			entry.activeconns = atomic_read(&dest->activeconns);
2954 			entry.inactconns = atomic_read(&dest->inactconns);
2955 			entry.persistconns = atomic_read(&dest->persistconns);
2956 			ip_vs_copy_stats(&kstats, &dest->stats);
2957 			ip_vs_export_stats_user(&entry.stats, &kstats);
2958 			if (copy_to_user(&uptr->entrytable[count],
2959 					 &entry, sizeof(entry))) {
2960 				ret = -EFAULT;
2961 				break;
2962 			}
2963 			count++;
2964 		}
2965 	} else
2966 		ret = -ESRCH;
2967 	return ret;
2968 }
2969 
2970 static inline void
__ip_vs_get_timeouts(struct netns_ipvs * ipvs,struct ip_vs_timeout_user * u)2971 __ip_vs_get_timeouts(struct netns_ipvs *ipvs, struct ip_vs_timeout_user *u)
2972 {
2973 #if defined(CONFIG_IP_VS_PROTO_TCP) || defined(CONFIG_IP_VS_PROTO_UDP)
2974 	struct ip_vs_proto_data *pd;
2975 #endif
2976 
2977 	memset(u, 0, sizeof (*u));
2978 
2979 #ifdef CONFIG_IP_VS_PROTO_TCP
2980 	pd = ip_vs_proto_data_get(ipvs, IPPROTO_TCP);
2981 	u->tcp_timeout = pd->timeout_table[IP_VS_TCP_S_ESTABLISHED] / HZ;
2982 	u->tcp_fin_timeout = pd->timeout_table[IP_VS_TCP_S_FIN_WAIT] / HZ;
2983 #endif
2984 #ifdef CONFIG_IP_VS_PROTO_UDP
2985 	pd = ip_vs_proto_data_get(ipvs, IPPROTO_UDP);
2986 	u->udp_timeout =
2987 			pd->timeout_table[IP_VS_UDP_S_NORMAL] / HZ;
2988 #endif
2989 }
2990 
2991 static const unsigned char get_arglen[CMDID(IP_VS_SO_GET_MAX) + 1] = {
2992 	[CMDID(IP_VS_SO_GET_VERSION)]  = 64,
2993 	[CMDID(IP_VS_SO_GET_INFO)]     = sizeof(struct ip_vs_getinfo),
2994 	[CMDID(IP_VS_SO_GET_SERVICES)] = sizeof(struct ip_vs_get_services),
2995 	[CMDID(IP_VS_SO_GET_SERVICE)]  = sizeof(struct ip_vs_service_entry),
2996 	[CMDID(IP_VS_SO_GET_DESTS)]    = sizeof(struct ip_vs_get_dests),
2997 	[CMDID(IP_VS_SO_GET_TIMEOUT)]  = sizeof(struct ip_vs_timeout_user),
2998 	[CMDID(IP_VS_SO_GET_DAEMON)]   = 2 * sizeof(struct ip_vs_daemon_user),
2999 };
3000 
3001 union ip_vs_get_arglen {
3002 	char				field_IP_VS_SO_GET_VERSION[64];
3003 	struct ip_vs_getinfo		field_IP_VS_SO_GET_INFO;
3004 	struct ip_vs_get_services	field_IP_VS_SO_GET_SERVICES;
3005 	struct ip_vs_service_entry	field_IP_VS_SO_GET_SERVICE;
3006 	struct ip_vs_get_dests		field_IP_VS_SO_GET_DESTS;
3007 	struct ip_vs_timeout_user	field_IP_VS_SO_GET_TIMEOUT;
3008 	struct ip_vs_daemon_user	field_IP_VS_SO_GET_DAEMON[2];
3009 };
3010 
3011 #define MAX_GET_ARGLEN	sizeof(union ip_vs_get_arglen)
3012 
3013 static int
do_ip_vs_get_ctl(struct sock * sk,int cmd,void __user * user,int * len)3014 do_ip_vs_get_ctl(struct sock *sk, int cmd, void __user *user, int *len)
3015 {
3016 	unsigned char arg[MAX_GET_ARGLEN];
3017 	int ret = 0;
3018 	unsigned int copylen;
3019 	struct net *net = sock_net(sk);
3020 	struct netns_ipvs *ipvs = net_ipvs(net);
3021 
3022 	BUG_ON(!net);
3023 	BUILD_BUG_ON(sizeof(arg) > 255);
3024 	if (!ns_capable(sock_net(sk)->user_ns, CAP_NET_ADMIN))
3025 		return -EPERM;
3026 
3027 	if (cmd < IP_VS_BASE_CTL || cmd > IP_VS_SO_GET_MAX)
3028 		return -EINVAL;
3029 
3030 	copylen = get_arglen[CMDID(cmd)];
3031 	if (*len < (int) copylen) {
3032 		IP_VS_DBG(1, "get_ctl: len %d < %u\n", *len, copylen);
3033 		return -EINVAL;
3034 	}
3035 
3036 	if (copy_from_user(arg, user, copylen) != 0)
3037 		return -EFAULT;
3038 	/*
3039 	 * Handle daemons first since it has its own locking
3040 	 */
3041 	if (cmd == IP_VS_SO_GET_DAEMON) {
3042 		struct ip_vs_daemon_user d[2];
3043 
3044 		memset(&d, 0, sizeof(d));
3045 		mutex_lock(&ipvs->sync_mutex);
3046 		if (ipvs->sync_state & IP_VS_STATE_MASTER) {
3047 			d[0].state = IP_VS_STATE_MASTER;
3048 			strscpy(d[0].mcast_ifn, ipvs->mcfg.mcast_ifn,
3049 				sizeof(d[0].mcast_ifn));
3050 			d[0].syncid = ipvs->mcfg.syncid;
3051 		}
3052 		if (ipvs->sync_state & IP_VS_STATE_BACKUP) {
3053 			d[1].state = IP_VS_STATE_BACKUP;
3054 			strscpy(d[1].mcast_ifn, ipvs->bcfg.mcast_ifn,
3055 				sizeof(d[1].mcast_ifn));
3056 			d[1].syncid = ipvs->bcfg.syncid;
3057 		}
3058 		if (copy_to_user(user, &d, sizeof(d)) != 0)
3059 			ret = -EFAULT;
3060 		mutex_unlock(&ipvs->sync_mutex);
3061 		return ret;
3062 	}
3063 
3064 	mutex_lock(&__ip_vs_mutex);
3065 	switch (cmd) {
3066 	case IP_VS_SO_GET_VERSION:
3067 	{
3068 		char buf[64];
3069 
3070 		sprintf(buf, "IP Virtual Server version %d.%d.%d (size=%d)",
3071 			NVERSION(IP_VS_VERSION_CODE), ip_vs_conn_tab_size);
3072 		if (copy_to_user(user, buf, strlen(buf)+1) != 0) {
3073 			ret = -EFAULT;
3074 			goto out;
3075 		}
3076 		*len = strlen(buf)+1;
3077 	}
3078 	break;
3079 
3080 	case IP_VS_SO_GET_INFO:
3081 	{
3082 		struct ip_vs_getinfo info;
3083 		info.version = IP_VS_VERSION_CODE;
3084 		info.size = ip_vs_conn_tab_size;
3085 		info.num_services = ipvs->num_services;
3086 		if (copy_to_user(user, &info, sizeof(info)) != 0)
3087 			ret = -EFAULT;
3088 	}
3089 	break;
3090 
3091 	case IP_VS_SO_GET_SERVICES:
3092 	{
3093 		struct ip_vs_get_services *get;
3094 		int size;
3095 
3096 		get = (struct ip_vs_get_services *)arg;
3097 		size = struct_size(get, entrytable, get->num_services);
3098 		if (*len != size) {
3099 			pr_err("length: %u != %u\n", *len, size);
3100 			ret = -EINVAL;
3101 			goto out;
3102 		}
3103 		ret = __ip_vs_get_service_entries(ipvs, get, user);
3104 	}
3105 	break;
3106 
3107 	case IP_VS_SO_GET_SERVICE:
3108 	{
3109 		struct ip_vs_service_entry *entry;
3110 		struct ip_vs_service *svc;
3111 		union nf_inet_addr addr;
3112 
3113 		entry = (struct ip_vs_service_entry *)arg;
3114 		addr.ip = entry->addr;
3115 		rcu_read_lock();
3116 		if (entry->fwmark)
3117 			svc = __ip_vs_svc_fwm_find(ipvs, AF_INET, entry->fwmark);
3118 		else
3119 			svc = __ip_vs_service_find(ipvs, AF_INET,
3120 						   entry->protocol, &addr,
3121 						   entry->port);
3122 		rcu_read_unlock();
3123 		if (svc) {
3124 			ip_vs_copy_service(entry, svc);
3125 			if (copy_to_user(user, entry, sizeof(*entry)) != 0)
3126 				ret = -EFAULT;
3127 		} else
3128 			ret = -ESRCH;
3129 	}
3130 	break;
3131 
3132 	case IP_VS_SO_GET_DESTS:
3133 	{
3134 		struct ip_vs_get_dests *get;
3135 		int size;
3136 
3137 		get = (struct ip_vs_get_dests *)arg;
3138 		size = struct_size(get, entrytable, get->num_dests);
3139 		if (*len != size) {
3140 			pr_err("length: %u != %u\n", *len, size);
3141 			ret = -EINVAL;
3142 			goto out;
3143 		}
3144 		ret = __ip_vs_get_dest_entries(ipvs, get, user);
3145 	}
3146 	break;
3147 
3148 	case IP_VS_SO_GET_TIMEOUT:
3149 	{
3150 		struct ip_vs_timeout_user t;
3151 
3152 		__ip_vs_get_timeouts(ipvs, &t);
3153 		if (copy_to_user(user, &t, sizeof(t)) != 0)
3154 			ret = -EFAULT;
3155 	}
3156 	break;
3157 
3158 	default:
3159 		ret = -EINVAL;
3160 	}
3161 
3162 out:
3163 	mutex_unlock(&__ip_vs_mutex);
3164 	return ret;
3165 }
3166 
3167 
3168 static struct nf_sockopt_ops ip_vs_sockopts = {
3169 	.pf		= PF_INET,
3170 	.set_optmin	= IP_VS_BASE_CTL,
3171 	.set_optmax	= IP_VS_SO_SET_MAX+1,
3172 	.set		= do_ip_vs_set_ctl,
3173 	.get_optmin	= IP_VS_BASE_CTL,
3174 	.get_optmax	= IP_VS_SO_GET_MAX+1,
3175 	.get		= do_ip_vs_get_ctl,
3176 	.owner		= THIS_MODULE,
3177 };
3178 
3179 /*
3180  * Generic Netlink interface
3181  */
3182 
3183 /* IPVS genetlink family */
3184 static struct genl_family ip_vs_genl_family;
3185 
3186 /* Policy used for first-level command attributes */
3187 static const struct nla_policy ip_vs_cmd_policy[IPVS_CMD_ATTR_MAX + 1] = {
3188 	[IPVS_CMD_ATTR_SERVICE]		= { .type = NLA_NESTED },
3189 	[IPVS_CMD_ATTR_DEST]		= { .type = NLA_NESTED },
3190 	[IPVS_CMD_ATTR_DAEMON]		= { .type = NLA_NESTED },
3191 	[IPVS_CMD_ATTR_TIMEOUT_TCP]	= { .type = NLA_U32 },
3192 	[IPVS_CMD_ATTR_TIMEOUT_TCP_FIN]	= { .type = NLA_U32 },
3193 	[IPVS_CMD_ATTR_TIMEOUT_UDP]	= { .type = NLA_U32 },
3194 };
3195 
3196 /* Policy used for attributes in nested attribute IPVS_CMD_ATTR_DAEMON */
3197 static const struct nla_policy ip_vs_daemon_policy[IPVS_DAEMON_ATTR_MAX + 1] = {
3198 	[IPVS_DAEMON_ATTR_STATE]	= { .type = NLA_U32 },
3199 	[IPVS_DAEMON_ATTR_MCAST_IFN]	= { .type = NLA_NUL_STRING,
3200 					    .len = IP_VS_IFNAME_MAXLEN - 1 },
3201 	[IPVS_DAEMON_ATTR_SYNC_ID]	= { .type = NLA_U32 },
3202 	[IPVS_DAEMON_ATTR_SYNC_MAXLEN]	= { .type = NLA_U16 },
3203 	[IPVS_DAEMON_ATTR_MCAST_GROUP]	= { .type = NLA_U32 },
3204 	[IPVS_DAEMON_ATTR_MCAST_GROUP6]	= { .len = sizeof(struct in6_addr) },
3205 	[IPVS_DAEMON_ATTR_MCAST_PORT]	= { .type = NLA_U16 },
3206 	[IPVS_DAEMON_ATTR_MCAST_TTL]	= { .type = NLA_U8 },
3207 };
3208 
3209 /* Policy used for attributes in nested attribute IPVS_CMD_ATTR_SERVICE */
3210 static const struct nla_policy ip_vs_svc_policy[IPVS_SVC_ATTR_MAX + 1] = {
3211 	[IPVS_SVC_ATTR_AF]		= { .type = NLA_U16 },
3212 	[IPVS_SVC_ATTR_PROTOCOL]	= { .type = NLA_U16 },
3213 	[IPVS_SVC_ATTR_ADDR]		= { .type = NLA_BINARY,
3214 					    .len = sizeof(union nf_inet_addr) },
3215 	[IPVS_SVC_ATTR_PORT]		= { .type = NLA_U16 },
3216 	[IPVS_SVC_ATTR_FWMARK]		= { .type = NLA_U32 },
3217 	[IPVS_SVC_ATTR_SCHED_NAME]	= { .type = NLA_NUL_STRING,
3218 					    .len = IP_VS_SCHEDNAME_MAXLEN - 1 },
3219 	[IPVS_SVC_ATTR_PE_NAME]		= { .type = NLA_NUL_STRING,
3220 					    .len = IP_VS_PENAME_MAXLEN },
3221 	[IPVS_SVC_ATTR_FLAGS]		= { .type = NLA_BINARY,
3222 					    .len = sizeof(struct ip_vs_flags) },
3223 	[IPVS_SVC_ATTR_TIMEOUT]		= { .type = NLA_U32 },
3224 	[IPVS_SVC_ATTR_NETMASK]		= { .type = NLA_U32 },
3225 	[IPVS_SVC_ATTR_STATS]		= { .type = NLA_NESTED },
3226 };
3227 
3228 /* Policy used for attributes in nested attribute IPVS_CMD_ATTR_DEST */
3229 static const struct nla_policy ip_vs_dest_policy[IPVS_DEST_ATTR_MAX + 1] = {
3230 	[IPVS_DEST_ATTR_ADDR]		= { .type = NLA_BINARY,
3231 					    .len = sizeof(union nf_inet_addr) },
3232 	[IPVS_DEST_ATTR_PORT]		= { .type = NLA_U16 },
3233 	[IPVS_DEST_ATTR_FWD_METHOD]	= { .type = NLA_U32 },
3234 	[IPVS_DEST_ATTR_WEIGHT]		= { .type = NLA_U32 },
3235 	[IPVS_DEST_ATTR_U_THRESH]	= { .type = NLA_U32 },
3236 	[IPVS_DEST_ATTR_L_THRESH]	= { .type = NLA_U32 },
3237 	[IPVS_DEST_ATTR_ACTIVE_CONNS]	= { .type = NLA_U32 },
3238 	[IPVS_DEST_ATTR_INACT_CONNS]	= { .type = NLA_U32 },
3239 	[IPVS_DEST_ATTR_PERSIST_CONNS]	= { .type = NLA_U32 },
3240 	[IPVS_DEST_ATTR_STATS]		= { .type = NLA_NESTED },
3241 	[IPVS_DEST_ATTR_ADDR_FAMILY]	= { .type = NLA_U16 },
3242 	[IPVS_DEST_ATTR_TUN_TYPE]	= { .type = NLA_U8 },
3243 	[IPVS_DEST_ATTR_TUN_PORT]	= { .type = NLA_U16 },
3244 	[IPVS_DEST_ATTR_TUN_FLAGS]	= { .type = NLA_U16 },
3245 };
3246 
ip_vs_genl_fill_stats(struct sk_buff * skb,int container_type,struct ip_vs_kstats * kstats)3247 static int ip_vs_genl_fill_stats(struct sk_buff *skb, int container_type,
3248 				 struct ip_vs_kstats *kstats)
3249 {
3250 	struct nlattr *nl_stats = nla_nest_start_noflag(skb, container_type);
3251 
3252 	if (!nl_stats)
3253 		return -EMSGSIZE;
3254 
3255 	if (nla_put_u32(skb, IPVS_STATS_ATTR_CONNS, (u32)kstats->conns) ||
3256 	    nla_put_u32(skb, IPVS_STATS_ATTR_INPKTS, (u32)kstats->inpkts) ||
3257 	    nla_put_u32(skb, IPVS_STATS_ATTR_OUTPKTS, (u32)kstats->outpkts) ||
3258 	    nla_put_u64_64bit(skb, IPVS_STATS_ATTR_INBYTES, kstats->inbytes,
3259 			      IPVS_STATS_ATTR_PAD) ||
3260 	    nla_put_u64_64bit(skb, IPVS_STATS_ATTR_OUTBYTES, kstats->outbytes,
3261 			      IPVS_STATS_ATTR_PAD) ||
3262 	    nla_put_u32(skb, IPVS_STATS_ATTR_CPS, (u32)kstats->cps) ||
3263 	    nla_put_u32(skb, IPVS_STATS_ATTR_INPPS, (u32)kstats->inpps) ||
3264 	    nla_put_u32(skb, IPVS_STATS_ATTR_OUTPPS, (u32)kstats->outpps) ||
3265 	    nla_put_u32(skb, IPVS_STATS_ATTR_INBPS, (u32)kstats->inbps) ||
3266 	    nla_put_u32(skb, IPVS_STATS_ATTR_OUTBPS, (u32)kstats->outbps))
3267 		goto nla_put_failure;
3268 	nla_nest_end(skb, nl_stats);
3269 
3270 	return 0;
3271 
3272 nla_put_failure:
3273 	nla_nest_cancel(skb, nl_stats);
3274 	return -EMSGSIZE;
3275 }
3276 
ip_vs_genl_fill_stats64(struct sk_buff * skb,int container_type,struct ip_vs_kstats * kstats)3277 static int ip_vs_genl_fill_stats64(struct sk_buff *skb, int container_type,
3278 				   struct ip_vs_kstats *kstats)
3279 {
3280 	struct nlattr *nl_stats = nla_nest_start_noflag(skb, container_type);
3281 
3282 	if (!nl_stats)
3283 		return -EMSGSIZE;
3284 
3285 	if (nla_put_u64_64bit(skb, IPVS_STATS_ATTR_CONNS, kstats->conns,
3286 			      IPVS_STATS_ATTR_PAD) ||
3287 	    nla_put_u64_64bit(skb, IPVS_STATS_ATTR_INPKTS, kstats->inpkts,
3288 			      IPVS_STATS_ATTR_PAD) ||
3289 	    nla_put_u64_64bit(skb, IPVS_STATS_ATTR_OUTPKTS, kstats->outpkts,
3290 			      IPVS_STATS_ATTR_PAD) ||
3291 	    nla_put_u64_64bit(skb, IPVS_STATS_ATTR_INBYTES, kstats->inbytes,
3292 			      IPVS_STATS_ATTR_PAD) ||
3293 	    nla_put_u64_64bit(skb, IPVS_STATS_ATTR_OUTBYTES, kstats->outbytes,
3294 			      IPVS_STATS_ATTR_PAD) ||
3295 	    nla_put_u64_64bit(skb, IPVS_STATS_ATTR_CPS, kstats->cps,
3296 			      IPVS_STATS_ATTR_PAD) ||
3297 	    nla_put_u64_64bit(skb, IPVS_STATS_ATTR_INPPS, kstats->inpps,
3298 			      IPVS_STATS_ATTR_PAD) ||
3299 	    nla_put_u64_64bit(skb, IPVS_STATS_ATTR_OUTPPS, kstats->outpps,
3300 			      IPVS_STATS_ATTR_PAD) ||
3301 	    nla_put_u64_64bit(skb, IPVS_STATS_ATTR_INBPS, kstats->inbps,
3302 			      IPVS_STATS_ATTR_PAD) ||
3303 	    nla_put_u64_64bit(skb, IPVS_STATS_ATTR_OUTBPS, kstats->outbps,
3304 			      IPVS_STATS_ATTR_PAD))
3305 		goto nla_put_failure;
3306 	nla_nest_end(skb, nl_stats);
3307 
3308 	return 0;
3309 
3310 nla_put_failure:
3311 	nla_nest_cancel(skb, nl_stats);
3312 	return -EMSGSIZE;
3313 }
3314 
ip_vs_genl_fill_service(struct sk_buff * skb,struct ip_vs_service * svc)3315 static int ip_vs_genl_fill_service(struct sk_buff *skb,
3316 				   struct ip_vs_service *svc)
3317 {
3318 	struct ip_vs_scheduler *sched;
3319 	struct ip_vs_pe *pe;
3320 	struct nlattr *nl_service;
3321 	struct ip_vs_flags flags = { .flags = svc->flags,
3322 				     .mask = ~0 };
3323 	struct ip_vs_kstats kstats;
3324 	char *sched_name;
3325 
3326 	nl_service = nla_nest_start_noflag(skb, IPVS_CMD_ATTR_SERVICE);
3327 	if (!nl_service)
3328 		return -EMSGSIZE;
3329 
3330 	if (nla_put_u16(skb, IPVS_SVC_ATTR_AF, svc->af))
3331 		goto nla_put_failure;
3332 	if (svc->fwmark) {
3333 		if (nla_put_u32(skb, IPVS_SVC_ATTR_FWMARK, svc->fwmark))
3334 			goto nla_put_failure;
3335 	} else {
3336 		if (nla_put_u16(skb, IPVS_SVC_ATTR_PROTOCOL, svc->protocol) ||
3337 		    nla_put(skb, IPVS_SVC_ATTR_ADDR, sizeof(svc->addr), &svc->addr) ||
3338 		    nla_put_be16(skb, IPVS_SVC_ATTR_PORT, svc->port))
3339 			goto nla_put_failure;
3340 	}
3341 
3342 	sched = rcu_dereference_protected(svc->scheduler, 1);
3343 	sched_name = sched ? sched->name : "none";
3344 	pe = rcu_dereference_protected(svc->pe, 1);
3345 	if (nla_put_string(skb, IPVS_SVC_ATTR_SCHED_NAME, sched_name) ||
3346 	    (pe && nla_put_string(skb, IPVS_SVC_ATTR_PE_NAME, pe->name)) ||
3347 	    nla_put(skb, IPVS_SVC_ATTR_FLAGS, sizeof(flags), &flags) ||
3348 	    nla_put_u32(skb, IPVS_SVC_ATTR_TIMEOUT, svc->timeout / HZ) ||
3349 	    nla_put_be32(skb, IPVS_SVC_ATTR_NETMASK, svc->netmask))
3350 		goto nla_put_failure;
3351 	ip_vs_copy_stats(&kstats, &svc->stats);
3352 	if (ip_vs_genl_fill_stats(skb, IPVS_SVC_ATTR_STATS, &kstats))
3353 		goto nla_put_failure;
3354 	if (ip_vs_genl_fill_stats64(skb, IPVS_SVC_ATTR_STATS64, &kstats))
3355 		goto nla_put_failure;
3356 
3357 	nla_nest_end(skb, nl_service);
3358 
3359 	return 0;
3360 
3361 nla_put_failure:
3362 	nla_nest_cancel(skb, nl_service);
3363 	return -EMSGSIZE;
3364 }
3365 
ip_vs_genl_dump_service(struct sk_buff * skb,struct ip_vs_service * svc,struct netlink_callback * cb)3366 static int ip_vs_genl_dump_service(struct sk_buff *skb,
3367 				   struct ip_vs_service *svc,
3368 				   struct netlink_callback *cb)
3369 {
3370 	void *hdr;
3371 
3372 	hdr = genlmsg_put(skb, NETLINK_CB(cb->skb).portid, cb->nlh->nlmsg_seq,
3373 			  &ip_vs_genl_family, NLM_F_MULTI,
3374 			  IPVS_CMD_NEW_SERVICE);
3375 	if (!hdr)
3376 		return -EMSGSIZE;
3377 
3378 	if (ip_vs_genl_fill_service(skb, svc) < 0)
3379 		goto nla_put_failure;
3380 
3381 	genlmsg_end(skb, hdr);
3382 	return 0;
3383 
3384 nla_put_failure:
3385 	genlmsg_cancel(skb, hdr);
3386 	return -EMSGSIZE;
3387 }
3388 
ip_vs_genl_dump_services(struct sk_buff * skb,struct netlink_callback * cb)3389 static int ip_vs_genl_dump_services(struct sk_buff *skb,
3390 				    struct netlink_callback *cb)
3391 {
3392 	int idx = 0, i;
3393 	int start = cb->args[0];
3394 	struct ip_vs_service *svc;
3395 	struct net *net = sock_net(skb->sk);
3396 	struct netns_ipvs *ipvs = net_ipvs(net);
3397 
3398 	mutex_lock(&__ip_vs_mutex);
3399 	for (i = 0; i < IP_VS_SVC_TAB_SIZE; i++) {
3400 		hlist_for_each_entry(svc, &ip_vs_svc_table[i], s_list) {
3401 			if (++idx <= start || (svc->ipvs != ipvs))
3402 				continue;
3403 			if (ip_vs_genl_dump_service(skb, svc, cb) < 0) {
3404 				idx--;
3405 				goto nla_put_failure;
3406 			}
3407 		}
3408 	}
3409 
3410 	for (i = 0; i < IP_VS_SVC_TAB_SIZE; i++) {
3411 		hlist_for_each_entry(svc, &ip_vs_svc_fwm_table[i], f_list) {
3412 			if (++idx <= start || (svc->ipvs != ipvs))
3413 				continue;
3414 			if (ip_vs_genl_dump_service(skb, svc, cb) < 0) {
3415 				idx--;
3416 				goto nla_put_failure;
3417 			}
3418 		}
3419 	}
3420 
3421 nla_put_failure:
3422 	mutex_unlock(&__ip_vs_mutex);
3423 	cb->args[0] = idx;
3424 
3425 	return skb->len;
3426 }
3427 
ip_vs_is_af_valid(int af)3428 static bool ip_vs_is_af_valid(int af)
3429 {
3430 	if (af == AF_INET)
3431 		return true;
3432 #ifdef CONFIG_IP_VS_IPV6
3433 	if (af == AF_INET6 && ipv6_mod_enabled())
3434 		return true;
3435 #endif
3436 	return false;
3437 }
3438 
ip_vs_genl_parse_service(struct netns_ipvs * ipvs,struct ip_vs_service_user_kern * usvc,struct nlattr * nla,bool full_entry,struct ip_vs_service ** ret_svc)3439 static int ip_vs_genl_parse_service(struct netns_ipvs *ipvs,
3440 				    struct ip_vs_service_user_kern *usvc,
3441 				    struct nlattr *nla, bool full_entry,
3442 				    struct ip_vs_service **ret_svc)
3443 {
3444 	struct nlattr *attrs[IPVS_SVC_ATTR_MAX + 1];
3445 	struct nlattr *nla_af, *nla_port, *nla_fwmark, *nla_protocol, *nla_addr;
3446 	struct ip_vs_service *svc;
3447 
3448 	/* Parse mandatory identifying service fields first */
3449 	if (nla == NULL ||
3450 	    nla_parse_nested_deprecated(attrs, IPVS_SVC_ATTR_MAX, nla, ip_vs_svc_policy, NULL))
3451 		return -EINVAL;
3452 
3453 	nla_af		= attrs[IPVS_SVC_ATTR_AF];
3454 	nla_protocol	= attrs[IPVS_SVC_ATTR_PROTOCOL];
3455 	nla_addr	= attrs[IPVS_SVC_ATTR_ADDR];
3456 	nla_port	= attrs[IPVS_SVC_ATTR_PORT];
3457 	nla_fwmark	= attrs[IPVS_SVC_ATTR_FWMARK];
3458 
3459 	if (!(nla_af && (nla_fwmark || (nla_port && nla_protocol && nla_addr))))
3460 		return -EINVAL;
3461 
3462 	memset(usvc, 0, sizeof(*usvc));
3463 
3464 	usvc->af = nla_get_u16(nla_af);
3465 	if (!ip_vs_is_af_valid(usvc->af))
3466 		return -EAFNOSUPPORT;
3467 
3468 	if (nla_fwmark) {
3469 		usvc->protocol = IPPROTO_TCP;
3470 		usvc->fwmark = nla_get_u32(nla_fwmark);
3471 	} else {
3472 		usvc->protocol = nla_get_u16(nla_protocol);
3473 		nla_memcpy(&usvc->addr, nla_addr, sizeof(usvc->addr));
3474 		usvc->port = nla_get_be16(nla_port);
3475 		usvc->fwmark = 0;
3476 	}
3477 
3478 	rcu_read_lock();
3479 	if (usvc->fwmark)
3480 		svc = __ip_vs_svc_fwm_find(ipvs, usvc->af, usvc->fwmark);
3481 	else
3482 		svc = __ip_vs_service_find(ipvs, usvc->af, usvc->protocol,
3483 					   &usvc->addr, usvc->port);
3484 	rcu_read_unlock();
3485 	*ret_svc = svc;
3486 
3487 	/* If a full entry was requested, check for the additional fields */
3488 	if (full_entry) {
3489 		struct nlattr *nla_sched, *nla_flags, *nla_pe, *nla_timeout,
3490 			      *nla_netmask;
3491 		struct ip_vs_flags flags;
3492 
3493 		nla_sched = attrs[IPVS_SVC_ATTR_SCHED_NAME];
3494 		nla_pe = attrs[IPVS_SVC_ATTR_PE_NAME];
3495 		nla_flags = attrs[IPVS_SVC_ATTR_FLAGS];
3496 		nla_timeout = attrs[IPVS_SVC_ATTR_TIMEOUT];
3497 		nla_netmask = attrs[IPVS_SVC_ATTR_NETMASK];
3498 
3499 		if (!(nla_sched && nla_flags && nla_timeout && nla_netmask))
3500 			return -EINVAL;
3501 
3502 		nla_memcpy(&flags, nla_flags, sizeof(flags));
3503 
3504 		/* prefill flags from service if it already exists */
3505 		if (svc)
3506 			usvc->flags = svc->flags;
3507 
3508 		/* set new flags from userland */
3509 		usvc->flags = (usvc->flags & ~flags.mask) |
3510 			      (flags.flags & flags.mask);
3511 		usvc->sched_name = nla_data(nla_sched);
3512 		usvc->pe_name = nla_pe ? nla_data(nla_pe) : NULL;
3513 		usvc->timeout = nla_get_u32(nla_timeout);
3514 		usvc->netmask = nla_get_be32(nla_netmask);
3515 	}
3516 
3517 	return 0;
3518 }
3519 
ip_vs_genl_find_service(struct netns_ipvs * ipvs,struct nlattr * nla)3520 static struct ip_vs_service *ip_vs_genl_find_service(struct netns_ipvs *ipvs,
3521 						     struct nlattr *nla)
3522 {
3523 	struct ip_vs_service_user_kern usvc;
3524 	struct ip_vs_service *svc;
3525 	int ret;
3526 
3527 	ret = ip_vs_genl_parse_service(ipvs, &usvc, nla, false, &svc);
3528 	return ret ? ERR_PTR(ret) : svc;
3529 }
3530 
ip_vs_genl_fill_dest(struct sk_buff * skb,struct ip_vs_dest * dest)3531 static int ip_vs_genl_fill_dest(struct sk_buff *skb, struct ip_vs_dest *dest)
3532 {
3533 	struct nlattr *nl_dest;
3534 	struct ip_vs_kstats kstats;
3535 
3536 	nl_dest = nla_nest_start_noflag(skb, IPVS_CMD_ATTR_DEST);
3537 	if (!nl_dest)
3538 		return -EMSGSIZE;
3539 
3540 	if (nla_put(skb, IPVS_DEST_ATTR_ADDR, sizeof(dest->addr), &dest->addr) ||
3541 	    nla_put_be16(skb, IPVS_DEST_ATTR_PORT, dest->port) ||
3542 	    nla_put_u32(skb, IPVS_DEST_ATTR_FWD_METHOD,
3543 			(atomic_read(&dest->conn_flags) &
3544 			 IP_VS_CONN_F_FWD_MASK)) ||
3545 	    nla_put_u32(skb, IPVS_DEST_ATTR_WEIGHT,
3546 			atomic_read(&dest->weight)) ||
3547 	    nla_put_u8(skb, IPVS_DEST_ATTR_TUN_TYPE,
3548 		       dest->tun_type) ||
3549 	    nla_put_be16(skb, IPVS_DEST_ATTR_TUN_PORT,
3550 			 dest->tun_port) ||
3551 	    nla_put_u16(skb, IPVS_DEST_ATTR_TUN_FLAGS,
3552 			dest->tun_flags) ||
3553 	    nla_put_u32(skb, IPVS_DEST_ATTR_U_THRESH, dest->u_threshold) ||
3554 	    nla_put_u32(skb, IPVS_DEST_ATTR_L_THRESH, dest->l_threshold) ||
3555 	    nla_put_u32(skb, IPVS_DEST_ATTR_ACTIVE_CONNS,
3556 			atomic_read(&dest->activeconns)) ||
3557 	    nla_put_u32(skb, IPVS_DEST_ATTR_INACT_CONNS,
3558 			atomic_read(&dest->inactconns)) ||
3559 	    nla_put_u32(skb, IPVS_DEST_ATTR_PERSIST_CONNS,
3560 			atomic_read(&dest->persistconns)) ||
3561 	    nla_put_u16(skb, IPVS_DEST_ATTR_ADDR_FAMILY, dest->af))
3562 		goto nla_put_failure;
3563 	ip_vs_copy_stats(&kstats, &dest->stats);
3564 	if (ip_vs_genl_fill_stats(skb, IPVS_DEST_ATTR_STATS, &kstats))
3565 		goto nla_put_failure;
3566 	if (ip_vs_genl_fill_stats64(skb, IPVS_DEST_ATTR_STATS64, &kstats))
3567 		goto nla_put_failure;
3568 
3569 	nla_nest_end(skb, nl_dest);
3570 
3571 	return 0;
3572 
3573 nla_put_failure:
3574 	nla_nest_cancel(skb, nl_dest);
3575 	return -EMSGSIZE;
3576 }
3577 
ip_vs_genl_dump_dest(struct sk_buff * skb,struct ip_vs_dest * dest,struct netlink_callback * cb)3578 static int ip_vs_genl_dump_dest(struct sk_buff *skb, struct ip_vs_dest *dest,
3579 				struct netlink_callback *cb)
3580 {
3581 	void *hdr;
3582 
3583 	hdr = genlmsg_put(skb, NETLINK_CB(cb->skb).portid, cb->nlh->nlmsg_seq,
3584 			  &ip_vs_genl_family, NLM_F_MULTI,
3585 			  IPVS_CMD_NEW_DEST);
3586 	if (!hdr)
3587 		return -EMSGSIZE;
3588 
3589 	if (ip_vs_genl_fill_dest(skb, dest) < 0)
3590 		goto nla_put_failure;
3591 
3592 	genlmsg_end(skb, hdr);
3593 	return 0;
3594 
3595 nla_put_failure:
3596 	genlmsg_cancel(skb, hdr);
3597 	return -EMSGSIZE;
3598 }
3599 
ip_vs_genl_dump_dests(struct sk_buff * skb,struct netlink_callback * cb)3600 static int ip_vs_genl_dump_dests(struct sk_buff *skb,
3601 				 struct netlink_callback *cb)
3602 {
3603 	int idx = 0;
3604 	int start = cb->args[0];
3605 	struct ip_vs_service *svc;
3606 	struct ip_vs_dest *dest;
3607 	struct nlattr *attrs[IPVS_CMD_ATTR_MAX + 1];
3608 	struct net *net = sock_net(skb->sk);
3609 	struct netns_ipvs *ipvs = net_ipvs(net);
3610 
3611 	mutex_lock(&__ip_vs_mutex);
3612 
3613 	/* Try to find the service for which to dump destinations */
3614 	if (nlmsg_parse_deprecated(cb->nlh, GENL_HDRLEN, attrs, IPVS_CMD_ATTR_MAX, ip_vs_cmd_policy, cb->extack))
3615 		goto out_err;
3616 
3617 
3618 	svc = ip_vs_genl_find_service(ipvs, attrs[IPVS_CMD_ATTR_SERVICE]);
3619 	if (IS_ERR_OR_NULL(svc))
3620 		goto out_err;
3621 
3622 	/* Dump the destinations */
3623 	list_for_each_entry(dest, &svc->destinations, n_list) {
3624 		if (++idx <= start)
3625 			continue;
3626 		if (ip_vs_genl_dump_dest(skb, dest, cb) < 0) {
3627 			idx--;
3628 			goto nla_put_failure;
3629 		}
3630 	}
3631 
3632 nla_put_failure:
3633 	cb->args[0] = idx;
3634 
3635 out_err:
3636 	mutex_unlock(&__ip_vs_mutex);
3637 
3638 	return skb->len;
3639 }
3640 
ip_vs_genl_parse_dest(struct ip_vs_dest_user_kern * udest,struct nlattr * nla,bool full_entry)3641 static int ip_vs_genl_parse_dest(struct ip_vs_dest_user_kern *udest,
3642 				 struct nlattr *nla, bool full_entry)
3643 {
3644 	struct nlattr *attrs[IPVS_DEST_ATTR_MAX + 1];
3645 	struct nlattr *nla_addr, *nla_port;
3646 	struct nlattr *nla_addr_family;
3647 
3648 	/* Parse mandatory identifying destination fields first */
3649 	if (nla == NULL ||
3650 	    nla_parse_nested_deprecated(attrs, IPVS_DEST_ATTR_MAX, nla, ip_vs_dest_policy, NULL))
3651 		return -EINVAL;
3652 
3653 	nla_addr	= attrs[IPVS_DEST_ATTR_ADDR];
3654 	nla_port	= attrs[IPVS_DEST_ATTR_PORT];
3655 	nla_addr_family	= attrs[IPVS_DEST_ATTR_ADDR_FAMILY];
3656 
3657 	if (!(nla_addr && nla_port))
3658 		return -EINVAL;
3659 
3660 	memset(udest, 0, sizeof(*udest));
3661 
3662 	nla_memcpy(&udest->addr, nla_addr, sizeof(udest->addr));
3663 	udest->port = nla_get_be16(nla_port);
3664 
3665 	if (nla_addr_family)
3666 		udest->af = nla_get_u16(nla_addr_family);
3667 	else
3668 		udest->af = 0;
3669 
3670 	/* If a full entry was requested, check for the additional fields */
3671 	if (full_entry) {
3672 		struct nlattr *nla_fwd, *nla_weight, *nla_u_thresh,
3673 			      *nla_l_thresh, *nla_tun_type, *nla_tun_port,
3674 			      *nla_tun_flags;
3675 
3676 		nla_fwd		= attrs[IPVS_DEST_ATTR_FWD_METHOD];
3677 		nla_weight	= attrs[IPVS_DEST_ATTR_WEIGHT];
3678 		nla_u_thresh	= attrs[IPVS_DEST_ATTR_U_THRESH];
3679 		nla_l_thresh	= attrs[IPVS_DEST_ATTR_L_THRESH];
3680 		nla_tun_type	= attrs[IPVS_DEST_ATTR_TUN_TYPE];
3681 		nla_tun_port	= attrs[IPVS_DEST_ATTR_TUN_PORT];
3682 		nla_tun_flags	= attrs[IPVS_DEST_ATTR_TUN_FLAGS];
3683 
3684 		if (!(nla_fwd && nla_weight && nla_u_thresh && nla_l_thresh))
3685 			return -EINVAL;
3686 
3687 		udest->conn_flags = nla_get_u32(nla_fwd)
3688 				    & IP_VS_CONN_F_FWD_MASK;
3689 		udest->weight = nla_get_u32(nla_weight);
3690 		udest->u_threshold = nla_get_u32(nla_u_thresh);
3691 		udest->l_threshold = nla_get_u32(nla_l_thresh);
3692 
3693 		if (nla_tun_type)
3694 			udest->tun_type = nla_get_u8(nla_tun_type);
3695 
3696 		if (nla_tun_port)
3697 			udest->tun_port = nla_get_be16(nla_tun_port);
3698 
3699 		if (nla_tun_flags)
3700 			udest->tun_flags = nla_get_u16(nla_tun_flags);
3701 	}
3702 
3703 	return 0;
3704 }
3705 
ip_vs_genl_fill_daemon(struct sk_buff * skb,__u32 state,struct ipvs_sync_daemon_cfg * c)3706 static int ip_vs_genl_fill_daemon(struct sk_buff *skb, __u32 state,
3707 				  struct ipvs_sync_daemon_cfg *c)
3708 {
3709 	struct nlattr *nl_daemon;
3710 
3711 	nl_daemon = nla_nest_start_noflag(skb, IPVS_CMD_ATTR_DAEMON);
3712 	if (!nl_daemon)
3713 		return -EMSGSIZE;
3714 
3715 	if (nla_put_u32(skb, IPVS_DAEMON_ATTR_STATE, state) ||
3716 	    nla_put_string(skb, IPVS_DAEMON_ATTR_MCAST_IFN, c->mcast_ifn) ||
3717 	    nla_put_u32(skb, IPVS_DAEMON_ATTR_SYNC_ID, c->syncid) ||
3718 	    nla_put_u16(skb, IPVS_DAEMON_ATTR_SYNC_MAXLEN, c->sync_maxlen) ||
3719 	    nla_put_u16(skb, IPVS_DAEMON_ATTR_MCAST_PORT, c->mcast_port) ||
3720 	    nla_put_u8(skb, IPVS_DAEMON_ATTR_MCAST_TTL, c->mcast_ttl))
3721 		goto nla_put_failure;
3722 #ifdef CONFIG_IP_VS_IPV6
3723 	if (c->mcast_af == AF_INET6) {
3724 		if (nla_put_in6_addr(skb, IPVS_DAEMON_ATTR_MCAST_GROUP6,
3725 				     &c->mcast_group.in6))
3726 			goto nla_put_failure;
3727 	} else
3728 #endif
3729 		if (c->mcast_af == AF_INET &&
3730 		    nla_put_in_addr(skb, IPVS_DAEMON_ATTR_MCAST_GROUP,
3731 				    c->mcast_group.ip))
3732 			goto nla_put_failure;
3733 	nla_nest_end(skb, nl_daemon);
3734 
3735 	return 0;
3736 
3737 nla_put_failure:
3738 	nla_nest_cancel(skb, nl_daemon);
3739 	return -EMSGSIZE;
3740 }
3741 
ip_vs_genl_dump_daemon(struct sk_buff * skb,__u32 state,struct ipvs_sync_daemon_cfg * c,struct netlink_callback * cb)3742 static int ip_vs_genl_dump_daemon(struct sk_buff *skb, __u32 state,
3743 				  struct ipvs_sync_daemon_cfg *c,
3744 				  struct netlink_callback *cb)
3745 {
3746 	void *hdr;
3747 	hdr = genlmsg_put(skb, NETLINK_CB(cb->skb).portid, cb->nlh->nlmsg_seq,
3748 			  &ip_vs_genl_family, NLM_F_MULTI,
3749 			  IPVS_CMD_NEW_DAEMON);
3750 	if (!hdr)
3751 		return -EMSGSIZE;
3752 
3753 	if (ip_vs_genl_fill_daemon(skb, state, c))
3754 		goto nla_put_failure;
3755 
3756 	genlmsg_end(skb, hdr);
3757 	return 0;
3758 
3759 nla_put_failure:
3760 	genlmsg_cancel(skb, hdr);
3761 	return -EMSGSIZE;
3762 }
3763 
ip_vs_genl_dump_daemons(struct sk_buff * skb,struct netlink_callback * cb)3764 static int ip_vs_genl_dump_daemons(struct sk_buff *skb,
3765 				   struct netlink_callback *cb)
3766 {
3767 	struct net *net = sock_net(skb->sk);
3768 	struct netns_ipvs *ipvs = net_ipvs(net);
3769 
3770 	mutex_lock(&ipvs->sync_mutex);
3771 	if ((ipvs->sync_state & IP_VS_STATE_MASTER) && !cb->args[0]) {
3772 		if (ip_vs_genl_dump_daemon(skb, IP_VS_STATE_MASTER,
3773 					   &ipvs->mcfg, cb) < 0)
3774 			goto nla_put_failure;
3775 
3776 		cb->args[0] = 1;
3777 	}
3778 
3779 	if ((ipvs->sync_state & IP_VS_STATE_BACKUP) && !cb->args[1]) {
3780 		if (ip_vs_genl_dump_daemon(skb, IP_VS_STATE_BACKUP,
3781 					   &ipvs->bcfg, cb) < 0)
3782 			goto nla_put_failure;
3783 
3784 		cb->args[1] = 1;
3785 	}
3786 
3787 nla_put_failure:
3788 	mutex_unlock(&ipvs->sync_mutex);
3789 
3790 	return skb->len;
3791 }
3792 
ip_vs_genl_new_daemon(struct netns_ipvs * ipvs,struct nlattr ** attrs)3793 static int ip_vs_genl_new_daemon(struct netns_ipvs *ipvs, struct nlattr **attrs)
3794 {
3795 	struct ipvs_sync_daemon_cfg c;
3796 	struct nlattr *a;
3797 	int ret;
3798 
3799 	memset(&c, 0, sizeof(c));
3800 	if (!(attrs[IPVS_DAEMON_ATTR_STATE] &&
3801 	      attrs[IPVS_DAEMON_ATTR_MCAST_IFN] &&
3802 	      attrs[IPVS_DAEMON_ATTR_SYNC_ID]))
3803 		return -EINVAL;
3804 	strscpy(c.mcast_ifn, nla_data(attrs[IPVS_DAEMON_ATTR_MCAST_IFN]),
3805 		sizeof(c.mcast_ifn));
3806 	c.syncid = nla_get_u32(attrs[IPVS_DAEMON_ATTR_SYNC_ID]);
3807 
3808 	a = attrs[IPVS_DAEMON_ATTR_SYNC_MAXLEN];
3809 	if (a)
3810 		c.sync_maxlen = nla_get_u16(a);
3811 
3812 	a = attrs[IPVS_DAEMON_ATTR_MCAST_GROUP];
3813 	if (a) {
3814 		c.mcast_af = AF_INET;
3815 		c.mcast_group.ip = nla_get_in_addr(a);
3816 		if (!ipv4_is_multicast(c.mcast_group.ip))
3817 			return -EINVAL;
3818 	} else {
3819 		a = attrs[IPVS_DAEMON_ATTR_MCAST_GROUP6];
3820 		if (a) {
3821 #ifdef CONFIG_IP_VS_IPV6
3822 			int addr_type;
3823 
3824 			c.mcast_af = AF_INET6;
3825 			c.mcast_group.in6 = nla_get_in6_addr(a);
3826 			addr_type = ipv6_addr_type(&c.mcast_group.in6);
3827 			if (!(addr_type & IPV6_ADDR_MULTICAST))
3828 				return -EINVAL;
3829 #else
3830 			return -EAFNOSUPPORT;
3831 #endif
3832 		}
3833 	}
3834 
3835 	a = attrs[IPVS_DAEMON_ATTR_MCAST_PORT];
3836 	if (a)
3837 		c.mcast_port = nla_get_u16(a);
3838 
3839 	a = attrs[IPVS_DAEMON_ATTR_MCAST_TTL];
3840 	if (a)
3841 		c.mcast_ttl = nla_get_u8(a);
3842 
3843 	/* The synchronization protocol is incompatible with mixed family
3844 	 * services
3845 	 */
3846 	if (ipvs->mixed_address_family_dests > 0)
3847 		return -EINVAL;
3848 
3849 	ret = start_sync_thread(ipvs, &c,
3850 				nla_get_u32(attrs[IPVS_DAEMON_ATTR_STATE]));
3851 	return ret;
3852 }
3853 
ip_vs_genl_del_daemon(struct netns_ipvs * ipvs,struct nlattr ** attrs)3854 static int ip_vs_genl_del_daemon(struct netns_ipvs *ipvs, struct nlattr **attrs)
3855 {
3856 	int ret;
3857 
3858 	if (!attrs[IPVS_DAEMON_ATTR_STATE])
3859 		return -EINVAL;
3860 
3861 	ret = stop_sync_thread(ipvs,
3862 			       nla_get_u32(attrs[IPVS_DAEMON_ATTR_STATE]));
3863 	return ret;
3864 }
3865 
ip_vs_genl_set_config(struct netns_ipvs * ipvs,struct nlattr ** attrs)3866 static int ip_vs_genl_set_config(struct netns_ipvs *ipvs, struct nlattr **attrs)
3867 {
3868 	struct ip_vs_timeout_user t;
3869 
3870 	__ip_vs_get_timeouts(ipvs, &t);
3871 
3872 	if (attrs[IPVS_CMD_ATTR_TIMEOUT_TCP])
3873 		t.tcp_timeout = nla_get_u32(attrs[IPVS_CMD_ATTR_TIMEOUT_TCP]);
3874 
3875 	if (attrs[IPVS_CMD_ATTR_TIMEOUT_TCP_FIN])
3876 		t.tcp_fin_timeout =
3877 			nla_get_u32(attrs[IPVS_CMD_ATTR_TIMEOUT_TCP_FIN]);
3878 
3879 	if (attrs[IPVS_CMD_ATTR_TIMEOUT_UDP])
3880 		t.udp_timeout = nla_get_u32(attrs[IPVS_CMD_ATTR_TIMEOUT_UDP]);
3881 
3882 	return ip_vs_set_timeout(ipvs, &t);
3883 }
3884 
ip_vs_genl_set_daemon(struct sk_buff * skb,struct genl_info * info)3885 static int ip_vs_genl_set_daemon(struct sk_buff *skb, struct genl_info *info)
3886 {
3887 	int ret = -EINVAL, cmd;
3888 	struct net *net = sock_net(skb->sk);
3889 	struct netns_ipvs *ipvs = net_ipvs(net);
3890 
3891 	cmd = info->genlhdr->cmd;
3892 
3893 	if (cmd == IPVS_CMD_NEW_DAEMON || cmd == IPVS_CMD_DEL_DAEMON) {
3894 		struct nlattr *daemon_attrs[IPVS_DAEMON_ATTR_MAX + 1];
3895 
3896 		if (!info->attrs[IPVS_CMD_ATTR_DAEMON] ||
3897 		    nla_parse_nested_deprecated(daemon_attrs, IPVS_DAEMON_ATTR_MAX, info->attrs[IPVS_CMD_ATTR_DAEMON], ip_vs_daemon_policy, info->extack))
3898 			goto out;
3899 
3900 		if (cmd == IPVS_CMD_NEW_DAEMON)
3901 			ret = ip_vs_genl_new_daemon(ipvs, daemon_attrs);
3902 		else
3903 			ret = ip_vs_genl_del_daemon(ipvs, daemon_attrs);
3904 	}
3905 
3906 out:
3907 	return ret;
3908 }
3909 
ip_vs_genl_set_cmd(struct sk_buff * skb,struct genl_info * info)3910 static int ip_vs_genl_set_cmd(struct sk_buff *skb, struct genl_info *info)
3911 {
3912 	bool need_full_svc = false, need_full_dest = false;
3913 	struct ip_vs_service *svc = NULL;
3914 	struct ip_vs_service_user_kern usvc;
3915 	struct ip_vs_dest_user_kern udest;
3916 	int ret = 0, cmd;
3917 	struct net *net = sock_net(skb->sk);
3918 	struct netns_ipvs *ipvs = net_ipvs(net);
3919 
3920 	cmd = info->genlhdr->cmd;
3921 
3922 	mutex_lock(&__ip_vs_mutex);
3923 
3924 	if (cmd == IPVS_CMD_FLUSH) {
3925 		ret = ip_vs_flush(ipvs, false);
3926 		goto out;
3927 	} else if (cmd == IPVS_CMD_SET_CONFIG) {
3928 		ret = ip_vs_genl_set_config(ipvs, info->attrs);
3929 		goto out;
3930 	} else if (cmd == IPVS_CMD_ZERO &&
3931 		   !info->attrs[IPVS_CMD_ATTR_SERVICE]) {
3932 		ret = ip_vs_zero_all(ipvs);
3933 		goto out;
3934 	}
3935 
3936 	/* All following commands require a service argument, so check if we
3937 	 * received a valid one. We need a full service specification when
3938 	 * adding / editing a service. Only identifying members otherwise. */
3939 	if (cmd == IPVS_CMD_NEW_SERVICE || cmd == IPVS_CMD_SET_SERVICE)
3940 		need_full_svc = true;
3941 
3942 	ret = ip_vs_genl_parse_service(ipvs, &usvc,
3943 				       info->attrs[IPVS_CMD_ATTR_SERVICE],
3944 				       need_full_svc, &svc);
3945 	if (ret)
3946 		goto out;
3947 
3948 	/* Unless we're adding a new service, the service must already exist */
3949 	if ((cmd != IPVS_CMD_NEW_SERVICE) && (svc == NULL)) {
3950 		ret = -ESRCH;
3951 		goto out;
3952 	}
3953 
3954 	/* Destination commands require a valid destination argument. For
3955 	 * adding / editing a destination, we need a full destination
3956 	 * specification. */
3957 	if (cmd == IPVS_CMD_NEW_DEST || cmd == IPVS_CMD_SET_DEST ||
3958 	    cmd == IPVS_CMD_DEL_DEST) {
3959 		if (cmd != IPVS_CMD_DEL_DEST)
3960 			need_full_dest = true;
3961 
3962 		ret = ip_vs_genl_parse_dest(&udest,
3963 					    info->attrs[IPVS_CMD_ATTR_DEST],
3964 					    need_full_dest);
3965 		if (ret)
3966 			goto out;
3967 
3968 		/* Old protocols did not allow the user to specify address
3969 		 * family, so we set it to zero instead.  We also didn't
3970 		 * allow heterogeneous pools in the old code, so it's safe
3971 		 * to assume that this will have the same address family as
3972 		 * the service.
3973 		 */
3974 		if (udest.af == 0)
3975 			udest.af = svc->af;
3976 
3977 		if (!ip_vs_is_af_valid(udest.af)) {
3978 			ret = -EAFNOSUPPORT;
3979 			goto out;
3980 		}
3981 
3982 		if (udest.af != svc->af && cmd != IPVS_CMD_DEL_DEST) {
3983 			/* The synchronization protocol is incompatible
3984 			 * with mixed family services
3985 			 */
3986 			if (ipvs->sync_state) {
3987 				ret = -EINVAL;
3988 				goto out;
3989 			}
3990 
3991 			/* Which connection types do we support? */
3992 			switch (udest.conn_flags) {
3993 			case IP_VS_CONN_F_TUNNEL:
3994 				/* We are able to forward this */
3995 				break;
3996 			default:
3997 				ret = -EINVAL;
3998 				goto out;
3999 			}
4000 		}
4001 	}
4002 
4003 	switch (cmd) {
4004 	case IPVS_CMD_NEW_SERVICE:
4005 		if (svc == NULL)
4006 			ret = ip_vs_add_service(ipvs, &usvc, &svc);
4007 		else
4008 			ret = -EEXIST;
4009 		break;
4010 	case IPVS_CMD_SET_SERVICE:
4011 		ret = ip_vs_edit_service(svc, &usvc);
4012 		break;
4013 	case IPVS_CMD_DEL_SERVICE:
4014 		ret = ip_vs_del_service(svc);
4015 		/* do not use svc, it can be freed */
4016 		break;
4017 	case IPVS_CMD_NEW_DEST:
4018 		ret = ip_vs_add_dest(svc, &udest);
4019 		break;
4020 	case IPVS_CMD_SET_DEST:
4021 		ret = ip_vs_edit_dest(svc, &udest);
4022 		break;
4023 	case IPVS_CMD_DEL_DEST:
4024 		ret = ip_vs_del_dest(svc, &udest);
4025 		break;
4026 	case IPVS_CMD_ZERO:
4027 		ret = ip_vs_zero_service(svc);
4028 		break;
4029 	default:
4030 		ret = -EINVAL;
4031 	}
4032 
4033 out:
4034 	mutex_unlock(&__ip_vs_mutex);
4035 
4036 	return ret;
4037 }
4038 
ip_vs_genl_get_cmd(struct sk_buff * skb,struct genl_info * info)4039 static int ip_vs_genl_get_cmd(struct sk_buff *skb, struct genl_info *info)
4040 {
4041 	struct sk_buff *msg;
4042 	void *reply;
4043 	int ret, cmd, reply_cmd;
4044 	struct net *net = sock_net(skb->sk);
4045 	struct netns_ipvs *ipvs = net_ipvs(net);
4046 
4047 	cmd = info->genlhdr->cmd;
4048 
4049 	if (cmd == IPVS_CMD_GET_SERVICE)
4050 		reply_cmd = IPVS_CMD_NEW_SERVICE;
4051 	else if (cmd == IPVS_CMD_GET_INFO)
4052 		reply_cmd = IPVS_CMD_SET_INFO;
4053 	else if (cmd == IPVS_CMD_GET_CONFIG)
4054 		reply_cmd = IPVS_CMD_SET_CONFIG;
4055 	else {
4056 		pr_err("unknown Generic Netlink command\n");
4057 		return -EINVAL;
4058 	}
4059 
4060 	msg = nlmsg_new(NLMSG_DEFAULT_SIZE, GFP_KERNEL);
4061 	if (!msg)
4062 		return -ENOMEM;
4063 
4064 	mutex_lock(&__ip_vs_mutex);
4065 
4066 	reply = genlmsg_put_reply(msg, info, &ip_vs_genl_family, 0, reply_cmd);
4067 	if (reply == NULL)
4068 		goto nla_put_failure;
4069 
4070 	switch (cmd) {
4071 	case IPVS_CMD_GET_SERVICE:
4072 	{
4073 		struct ip_vs_service *svc;
4074 
4075 		svc = ip_vs_genl_find_service(ipvs,
4076 					      info->attrs[IPVS_CMD_ATTR_SERVICE]);
4077 		if (IS_ERR(svc)) {
4078 			ret = PTR_ERR(svc);
4079 			goto out_err;
4080 		} else if (svc) {
4081 			ret = ip_vs_genl_fill_service(msg, svc);
4082 			if (ret)
4083 				goto nla_put_failure;
4084 		} else {
4085 			ret = -ESRCH;
4086 			goto out_err;
4087 		}
4088 
4089 		break;
4090 	}
4091 
4092 	case IPVS_CMD_GET_CONFIG:
4093 	{
4094 		struct ip_vs_timeout_user t;
4095 
4096 		__ip_vs_get_timeouts(ipvs, &t);
4097 #ifdef CONFIG_IP_VS_PROTO_TCP
4098 		if (nla_put_u32(msg, IPVS_CMD_ATTR_TIMEOUT_TCP,
4099 				t.tcp_timeout) ||
4100 		    nla_put_u32(msg, IPVS_CMD_ATTR_TIMEOUT_TCP_FIN,
4101 				t.tcp_fin_timeout))
4102 			goto nla_put_failure;
4103 #endif
4104 #ifdef CONFIG_IP_VS_PROTO_UDP
4105 		if (nla_put_u32(msg, IPVS_CMD_ATTR_TIMEOUT_UDP, t.udp_timeout))
4106 			goto nla_put_failure;
4107 #endif
4108 
4109 		break;
4110 	}
4111 
4112 	case IPVS_CMD_GET_INFO:
4113 		if (nla_put_u32(msg, IPVS_INFO_ATTR_VERSION,
4114 				IP_VS_VERSION_CODE) ||
4115 		    nla_put_u32(msg, IPVS_INFO_ATTR_CONN_TAB_SIZE,
4116 				ip_vs_conn_tab_size))
4117 			goto nla_put_failure;
4118 		break;
4119 	}
4120 
4121 	genlmsg_end(msg, reply);
4122 	ret = genlmsg_reply(msg, info);
4123 	goto out;
4124 
4125 nla_put_failure:
4126 	pr_err("not enough space in Netlink message\n");
4127 	ret = -EMSGSIZE;
4128 
4129 out_err:
4130 	nlmsg_free(msg);
4131 out:
4132 	mutex_unlock(&__ip_vs_mutex);
4133 
4134 	return ret;
4135 }
4136 
4137 
4138 static const struct genl_small_ops ip_vs_genl_ops[] = {
4139 	{
4140 		.cmd	= IPVS_CMD_NEW_SERVICE,
4141 		.validate = GENL_DONT_VALIDATE_STRICT | GENL_DONT_VALIDATE_DUMP,
4142 		.flags	= GENL_ADMIN_PERM,
4143 		.doit	= ip_vs_genl_set_cmd,
4144 	},
4145 	{
4146 		.cmd	= IPVS_CMD_SET_SERVICE,
4147 		.validate = GENL_DONT_VALIDATE_STRICT | GENL_DONT_VALIDATE_DUMP,
4148 		.flags	= GENL_ADMIN_PERM,
4149 		.doit	= ip_vs_genl_set_cmd,
4150 	},
4151 	{
4152 		.cmd	= IPVS_CMD_DEL_SERVICE,
4153 		.validate = GENL_DONT_VALIDATE_STRICT | GENL_DONT_VALIDATE_DUMP,
4154 		.flags	= GENL_ADMIN_PERM,
4155 		.doit	= ip_vs_genl_set_cmd,
4156 	},
4157 	{
4158 		.cmd	= IPVS_CMD_GET_SERVICE,
4159 		.validate = GENL_DONT_VALIDATE_STRICT | GENL_DONT_VALIDATE_DUMP,
4160 		.flags	= GENL_ADMIN_PERM,
4161 		.doit	= ip_vs_genl_get_cmd,
4162 		.dumpit	= ip_vs_genl_dump_services,
4163 	},
4164 	{
4165 		.cmd	= IPVS_CMD_NEW_DEST,
4166 		.validate = GENL_DONT_VALIDATE_STRICT | GENL_DONT_VALIDATE_DUMP,
4167 		.flags	= GENL_ADMIN_PERM,
4168 		.doit	= ip_vs_genl_set_cmd,
4169 	},
4170 	{
4171 		.cmd	= IPVS_CMD_SET_DEST,
4172 		.validate = GENL_DONT_VALIDATE_STRICT | GENL_DONT_VALIDATE_DUMP,
4173 		.flags	= GENL_ADMIN_PERM,
4174 		.doit	= ip_vs_genl_set_cmd,
4175 	},
4176 	{
4177 		.cmd	= IPVS_CMD_DEL_DEST,
4178 		.validate = GENL_DONT_VALIDATE_STRICT | GENL_DONT_VALIDATE_DUMP,
4179 		.flags	= GENL_ADMIN_PERM,
4180 		.doit	= ip_vs_genl_set_cmd,
4181 	},
4182 	{
4183 		.cmd	= IPVS_CMD_GET_DEST,
4184 		.validate = GENL_DONT_VALIDATE_STRICT | GENL_DONT_VALIDATE_DUMP,
4185 		.flags	= GENL_ADMIN_PERM,
4186 		.dumpit	= ip_vs_genl_dump_dests,
4187 	},
4188 	{
4189 		.cmd	= IPVS_CMD_NEW_DAEMON,
4190 		.validate = GENL_DONT_VALIDATE_STRICT | GENL_DONT_VALIDATE_DUMP,
4191 		.flags	= GENL_ADMIN_PERM,
4192 		.doit	= ip_vs_genl_set_daemon,
4193 	},
4194 	{
4195 		.cmd	= IPVS_CMD_DEL_DAEMON,
4196 		.validate = GENL_DONT_VALIDATE_STRICT | GENL_DONT_VALIDATE_DUMP,
4197 		.flags	= GENL_ADMIN_PERM,
4198 		.doit	= ip_vs_genl_set_daemon,
4199 	},
4200 	{
4201 		.cmd	= IPVS_CMD_GET_DAEMON,
4202 		.validate = GENL_DONT_VALIDATE_STRICT | GENL_DONT_VALIDATE_DUMP,
4203 		.flags	= GENL_ADMIN_PERM,
4204 		.dumpit	= ip_vs_genl_dump_daemons,
4205 	},
4206 	{
4207 		.cmd	= IPVS_CMD_SET_CONFIG,
4208 		.validate = GENL_DONT_VALIDATE_STRICT | GENL_DONT_VALIDATE_DUMP,
4209 		.flags	= GENL_ADMIN_PERM,
4210 		.doit	= ip_vs_genl_set_cmd,
4211 	},
4212 	{
4213 		.cmd	= IPVS_CMD_GET_CONFIG,
4214 		.validate = GENL_DONT_VALIDATE_STRICT | GENL_DONT_VALIDATE_DUMP,
4215 		.flags	= GENL_ADMIN_PERM,
4216 		.doit	= ip_vs_genl_get_cmd,
4217 	},
4218 	{
4219 		.cmd	= IPVS_CMD_GET_INFO,
4220 		.validate = GENL_DONT_VALIDATE_STRICT | GENL_DONT_VALIDATE_DUMP,
4221 		.flags	= GENL_ADMIN_PERM,
4222 		.doit	= ip_vs_genl_get_cmd,
4223 	},
4224 	{
4225 		.cmd	= IPVS_CMD_ZERO,
4226 		.validate = GENL_DONT_VALIDATE_STRICT | GENL_DONT_VALIDATE_DUMP,
4227 		.flags	= GENL_ADMIN_PERM,
4228 		.doit	= ip_vs_genl_set_cmd,
4229 	},
4230 	{
4231 		.cmd	= IPVS_CMD_FLUSH,
4232 		.validate = GENL_DONT_VALIDATE_STRICT | GENL_DONT_VALIDATE_DUMP,
4233 		.flags	= GENL_ADMIN_PERM,
4234 		.doit	= ip_vs_genl_set_cmd,
4235 	},
4236 };
4237 
4238 static struct genl_family ip_vs_genl_family __ro_after_init = {
4239 	.hdrsize	= 0,
4240 	.name		= IPVS_GENL_NAME,
4241 	.version	= IPVS_GENL_VERSION,
4242 	.maxattr	= IPVS_CMD_ATTR_MAX,
4243 	.policy = ip_vs_cmd_policy,
4244 	.netnsok        = true,         /* Make ipvsadm to work on netns */
4245 	.module		= THIS_MODULE,
4246 	.small_ops	= ip_vs_genl_ops,
4247 	.n_small_ops	= ARRAY_SIZE(ip_vs_genl_ops),
4248 	.resv_start_op	= IPVS_CMD_FLUSH + 1,
4249 };
4250 
ip_vs_genl_register(void)4251 static int __init ip_vs_genl_register(void)
4252 {
4253 	return genl_register_family(&ip_vs_genl_family);
4254 }
4255 
ip_vs_genl_unregister(void)4256 static void ip_vs_genl_unregister(void)
4257 {
4258 	genl_unregister_family(&ip_vs_genl_family);
4259 }
4260 
4261 /* End of Generic Netlink interface definitions */
4262 
4263 /*
4264  * per netns intit/exit func.
4265  */
4266 #ifdef CONFIG_SYSCTL
ip_vs_control_net_init_sysctl(struct netns_ipvs * ipvs)4267 static int __net_init ip_vs_control_net_init_sysctl(struct netns_ipvs *ipvs)
4268 {
4269 	struct net *net = ipvs->net;
4270 	struct ctl_table *tbl;
4271 	int idx, ret;
4272 	size_t ctl_table_size = ARRAY_SIZE(vs_vars);
4273 	bool unpriv = net->user_ns != &init_user_ns;
4274 
4275 	atomic_set(&ipvs->dropentry, 0);
4276 	spin_lock_init(&ipvs->dropentry_lock);
4277 	spin_lock_init(&ipvs->droppacket_lock);
4278 	spin_lock_init(&ipvs->securetcp_lock);
4279 	INIT_DELAYED_WORK(&ipvs->defense_work, defense_work_handler);
4280 	INIT_DELAYED_WORK(&ipvs->expire_nodest_conn_work,
4281 			  expire_nodest_conn_handler);
4282 	ipvs->est_stopped = 0;
4283 
4284 	if (!net_eq(net, &init_net)) {
4285 		tbl = kmemdup(vs_vars, sizeof(vs_vars), GFP_KERNEL);
4286 		if (tbl == NULL)
4287 			return -ENOMEM;
4288 	} else
4289 		tbl = vs_vars;
4290 	/* Initialize sysctl defaults */
4291 	for (idx = 0; idx < ARRAY_SIZE(vs_vars); idx++) {
4292 		if (tbl[idx].proc_handler == proc_do_defense_mode)
4293 			tbl[idx].extra2 = ipvs;
4294 	}
4295 	idx = 0;
4296 	ipvs->sysctl_amemthresh = 1024;
4297 	tbl[idx++].data = &ipvs->sysctl_amemthresh;
4298 	ipvs->sysctl_am_droprate = 10;
4299 	tbl[idx++].data = &ipvs->sysctl_am_droprate;
4300 	tbl[idx++].data = &ipvs->sysctl_drop_entry;
4301 	tbl[idx++].data = &ipvs->sysctl_drop_packet;
4302 #ifdef CONFIG_IP_VS_NFCT
4303 	tbl[idx++].data = &ipvs->sysctl_conntrack;
4304 #endif
4305 	tbl[idx++].data = &ipvs->sysctl_secure_tcp;
4306 	ipvs->sysctl_snat_reroute = 1;
4307 	tbl[idx++].data = &ipvs->sysctl_snat_reroute;
4308 	ipvs->sysctl_sync_ver = 1;
4309 	tbl[idx++].data = &ipvs->sysctl_sync_ver;
4310 	ipvs->sysctl_sync_ports = 1;
4311 	tbl[idx++].data = &ipvs->sysctl_sync_ports;
4312 	tbl[idx++].data = &ipvs->sysctl_sync_persist_mode;
4313 
4314 	ipvs->sysctl_sync_qlen_max = nr_free_buffer_pages() / 32;
4315 	if (unpriv)
4316 		tbl[idx].mode = 0444;
4317 	tbl[idx++].data = &ipvs->sysctl_sync_qlen_max;
4318 
4319 	ipvs->sysctl_sync_sock_size = 0;
4320 	if (unpriv)
4321 		tbl[idx].mode = 0444;
4322 	tbl[idx++].data = &ipvs->sysctl_sync_sock_size;
4323 
4324 	tbl[idx++].data = &ipvs->sysctl_cache_bypass;
4325 	tbl[idx++].data = &ipvs->sysctl_expire_nodest_conn;
4326 	tbl[idx++].data = &ipvs->sysctl_sloppy_tcp;
4327 	tbl[idx++].data = &ipvs->sysctl_sloppy_sctp;
4328 	tbl[idx++].data = &ipvs->sysctl_expire_quiescent_template;
4329 	ipvs->sysctl_sync_threshold[0] = DEFAULT_SYNC_THRESHOLD;
4330 	ipvs->sysctl_sync_threshold[1] = DEFAULT_SYNC_PERIOD;
4331 	tbl[idx].data = &ipvs->sysctl_sync_threshold;
4332 	tbl[idx].extra2 = ipvs;
4333 	tbl[idx++].maxlen = sizeof(ipvs->sysctl_sync_threshold);
4334 	ipvs->sysctl_sync_refresh_period = DEFAULT_SYNC_REFRESH_PERIOD;
4335 	tbl[idx++].data = &ipvs->sysctl_sync_refresh_period;
4336 	ipvs->sysctl_sync_retries = clamp_t(int, DEFAULT_SYNC_RETRIES, 0, 3);
4337 	tbl[idx++].data = &ipvs->sysctl_sync_retries;
4338 	tbl[idx++].data = &ipvs->sysctl_nat_icmp_send;
4339 	ipvs->sysctl_pmtu_disc = 1;
4340 	tbl[idx++].data = &ipvs->sysctl_pmtu_disc;
4341 	tbl[idx++].data = &ipvs->sysctl_backup_only;
4342 	ipvs->sysctl_conn_reuse_mode = 1;
4343 	tbl[idx++].data = &ipvs->sysctl_conn_reuse_mode;
4344 	tbl[idx++].data = &ipvs->sysctl_schedule_icmp;
4345 	tbl[idx++].data = &ipvs->sysctl_ignore_tunneled;
4346 
4347 	ipvs->sysctl_run_estimation = 1;
4348 	if (unpriv)
4349 		tbl[idx].mode = 0444;
4350 	tbl[idx].extra2 = ipvs;
4351 	tbl[idx++].data = &ipvs->sysctl_run_estimation;
4352 
4353 	ipvs->est_cpulist_valid = 0;
4354 	if (unpriv)
4355 		tbl[idx].mode = 0444;
4356 	tbl[idx].extra2 = ipvs;
4357 	tbl[idx++].data = &ipvs->sysctl_est_cpulist;
4358 
4359 	ipvs->sysctl_est_nice = IPVS_EST_NICE;
4360 	if (unpriv)
4361 		tbl[idx].mode = 0444;
4362 	tbl[idx].extra2 = ipvs;
4363 	tbl[idx++].data = &ipvs->sysctl_est_nice;
4364 
4365 #ifdef CONFIG_IP_VS_DEBUG
4366 	/* Global sysctls must be ro in non-init netns */
4367 	if (!net_eq(net, &init_net))
4368 		tbl[idx++].mode = 0444;
4369 #endif
4370 
4371 	ret = -ENOMEM;
4372 	ipvs->sysctl_hdr = register_net_sysctl_sz(net, "net/ipv4/vs", tbl,
4373 						  ctl_table_size);
4374 	if (!ipvs->sysctl_hdr)
4375 		goto err;
4376 	ipvs->sysctl_tbl = tbl;
4377 
4378 	ret = ip_vs_start_estimator(ipvs, &ipvs->tot_stats->s);
4379 	if (ret < 0)
4380 		goto err;
4381 
4382 	/* Schedule defense work */
4383 	queue_delayed_work(system_long_wq, &ipvs->defense_work,
4384 			   DEFENSE_TIMER_PERIOD);
4385 
4386 	return 0;
4387 
4388 err:
4389 	unregister_net_sysctl_table(ipvs->sysctl_hdr);
4390 	if (!net_eq(net, &init_net))
4391 		kfree(tbl);
4392 	return ret;
4393 }
4394 
ip_vs_control_net_cleanup_sysctl(struct netns_ipvs * ipvs)4395 static void __net_exit ip_vs_control_net_cleanup_sysctl(struct netns_ipvs *ipvs)
4396 {
4397 	struct net *net = ipvs->net;
4398 
4399 	cancel_delayed_work_sync(&ipvs->expire_nodest_conn_work);
4400 	cancel_delayed_work_sync(&ipvs->defense_work);
4401 	cancel_work_sync(&ipvs->defense_work.work);
4402 	unregister_net_sysctl_table(ipvs->sysctl_hdr);
4403 	ip_vs_stop_estimator(ipvs, &ipvs->tot_stats->s);
4404 
4405 	if (ipvs->est_cpulist_valid)
4406 		free_cpumask_var(ipvs->sysctl_est_cpulist);
4407 
4408 	if (!net_eq(net, &init_net))
4409 		kfree(ipvs->sysctl_tbl);
4410 }
4411 
4412 #else
4413 
ip_vs_control_net_init_sysctl(struct netns_ipvs * ipvs)4414 static int __net_init ip_vs_control_net_init_sysctl(struct netns_ipvs *ipvs) { return 0; }
ip_vs_control_net_cleanup_sysctl(struct netns_ipvs * ipvs)4415 static void __net_exit ip_vs_control_net_cleanup_sysctl(struct netns_ipvs *ipvs) { }
4416 
4417 #endif
4418 
4419 static struct notifier_block ip_vs_dst_notifier = {
4420 	.notifier_call = ip_vs_dst_event,
4421 #ifdef CONFIG_IP_VS_IPV6
4422 	.priority = ADDRCONF_NOTIFY_PRIORITY + 5,
4423 #endif
4424 };
4425 
ip_vs_control_net_init(struct netns_ipvs * ipvs)4426 int __net_init ip_vs_control_net_init(struct netns_ipvs *ipvs)
4427 {
4428 	int ret = -ENOMEM;
4429 	int idx;
4430 
4431 	/* Initialize rs_table */
4432 	for (idx = 0; idx < IP_VS_RTAB_SIZE; idx++)
4433 		INIT_HLIST_HEAD(&ipvs->rs_table[idx]);
4434 
4435 	INIT_LIST_HEAD(&ipvs->dest_trash);
4436 	spin_lock_init(&ipvs->dest_trash_lock);
4437 	timer_setup(&ipvs->dest_trash_timer, ip_vs_dest_trash_expire, 0);
4438 	atomic_set(&ipvs->ftpsvc_counter, 0);
4439 	atomic_set(&ipvs->nullsvc_counter, 0);
4440 	atomic_set(&ipvs->conn_out_counter, 0);
4441 
4442 	INIT_DELAYED_WORK(&ipvs->est_reload_work, est_reload_work_handler);
4443 
4444 	/* procfs stats */
4445 	ipvs->tot_stats = kzalloc(sizeof(*ipvs->tot_stats), GFP_KERNEL);
4446 	if (!ipvs->tot_stats)
4447 		goto out;
4448 	if (ip_vs_stats_init_alloc(&ipvs->tot_stats->s) < 0)
4449 		goto err_tot_stats;
4450 
4451 #ifdef CONFIG_PROC_FS
4452 	if (!proc_create_net("ip_vs", 0, ipvs->net->proc_net,
4453 			     &ip_vs_info_seq_ops, sizeof(struct ip_vs_iter)))
4454 		goto err_vs;
4455 	if (!proc_create_net_single("ip_vs_stats", 0, ipvs->net->proc_net,
4456 				    ip_vs_stats_show, NULL))
4457 		goto err_stats;
4458 	if (!proc_create_net_single("ip_vs_stats_percpu", 0,
4459 				    ipvs->net->proc_net,
4460 				    ip_vs_stats_percpu_show, NULL))
4461 		goto err_percpu;
4462 #endif
4463 
4464 	ret = ip_vs_control_net_init_sysctl(ipvs);
4465 	if (ret < 0)
4466 		goto err;
4467 
4468 	return 0;
4469 
4470 err:
4471 #ifdef CONFIG_PROC_FS
4472 	remove_proc_entry("ip_vs_stats_percpu", ipvs->net->proc_net);
4473 
4474 err_percpu:
4475 	remove_proc_entry("ip_vs_stats", ipvs->net->proc_net);
4476 
4477 err_stats:
4478 	remove_proc_entry("ip_vs", ipvs->net->proc_net);
4479 
4480 err_vs:
4481 #endif
4482 	ip_vs_stats_release(&ipvs->tot_stats->s);
4483 
4484 err_tot_stats:
4485 	kfree(ipvs->tot_stats);
4486 
4487 out:
4488 	return ret;
4489 }
4490 
ip_vs_control_net_cleanup(struct netns_ipvs * ipvs)4491 void __net_exit ip_vs_control_net_cleanup(struct netns_ipvs *ipvs)
4492 {
4493 	ip_vs_trash_cleanup(ipvs);
4494 	ip_vs_control_net_cleanup_sysctl(ipvs);
4495 	cancel_delayed_work_sync(&ipvs->est_reload_work);
4496 #ifdef CONFIG_PROC_FS
4497 	remove_proc_entry("ip_vs_stats_percpu", ipvs->net->proc_net);
4498 	remove_proc_entry("ip_vs_stats", ipvs->net->proc_net);
4499 	remove_proc_entry("ip_vs", ipvs->net->proc_net);
4500 #endif
4501 	call_rcu(&ipvs->tot_stats->rcu_head, ip_vs_stats_rcu_free);
4502 }
4503 
ip_vs_register_nl_ioctl(void)4504 int __init ip_vs_register_nl_ioctl(void)
4505 {
4506 	int ret;
4507 
4508 	ret = nf_register_sockopt(&ip_vs_sockopts);
4509 	if (ret) {
4510 		pr_err("cannot register sockopt.\n");
4511 		goto err_sock;
4512 	}
4513 
4514 	ret = ip_vs_genl_register();
4515 	if (ret) {
4516 		pr_err("cannot register Generic Netlink interface.\n");
4517 		goto err_genl;
4518 	}
4519 	return 0;
4520 
4521 err_genl:
4522 	nf_unregister_sockopt(&ip_vs_sockopts);
4523 err_sock:
4524 	return ret;
4525 }
4526 
ip_vs_unregister_nl_ioctl(void)4527 void ip_vs_unregister_nl_ioctl(void)
4528 {
4529 	ip_vs_genl_unregister();
4530 	nf_unregister_sockopt(&ip_vs_sockopts);
4531 }
4532 
ip_vs_control_init(void)4533 int __init ip_vs_control_init(void)
4534 {
4535 	int idx;
4536 	int ret;
4537 
4538 	/* Initialize svc_table, ip_vs_svc_fwm_table */
4539 	for (idx = 0; idx < IP_VS_SVC_TAB_SIZE; idx++) {
4540 		INIT_HLIST_HEAD(&ip_vs_svc_table[idx]);
4541 		INIT_HLIST_HEAD(&ip_vs_svc_fwm_table[idx]);
4542 	}
4543 
4544 	smp_wmb();	/* Do we really need it now ? */
4545 
4546 	ret = register_netdevice_notifier(&ip_vs_dst_notifier);
4547 	if (ret < 0)
4548 		return ret;
4549 
4550 	return 0;
4551 }
4552 
4553 
ip_vs_control_cleanup(void)4554 void ip_vs_control_cleanup(void)
4555 {
4556 	unregister_netdevice_notifier(&ip_vs_dst_notifier);
4557 	/* relying on common rcu_barrier() in ip_vs_cleanup() */
4558 }
4559