]> git.proxmox.com Git - mirror_ubuntu-jammy-kernel.git/blob - net/netfilter/ipvs/ip_vs_sync.c
projects / mirror_ubuntu-jammy-kernel.git / blob
? search:


3e7961e85e9caaf63d92615a3ffab970775583c4
[mirror_ubuntu-jammy-kernel.git] / net / netfilter / ipvs / ip_vs_sync.c
1 /*
2 * IPVS An implementation of the IP virtual server support for the
3 * LINUX operating system. IPVS is now implemented as a module
4 * over the NetFilter framework. IPVS can be used to build a
5 * high-performance and highly available server based on a
6 * cluster of servers.
7 *
8 * Version 1, is capable of handling both version 0 and 1 messages.
9 * Version 0 is the plain old format.
10 * Note Version 0 receivers will just drop Ver 1 messages.
11 * Version 1 is capable of handle IPv6, Persistence data,
12 * time-outs, and firewall marks.
13 * In ver.1 "ip_vs_sync_conn_options" will be sent in netw. order.
14 * Ver. 0 can be turned on by sysctl -w net.ipv4.vs.sync_version=0
15 *
16 * Definitions Message: is a complete datagram
17 * Sync_conn: is a part of a Message
18 * Param Data is an option to a Sync_conn.
19 *
20 * Authors: Wensong Zhang <wensong@linuxvirtualserver.org>
21 *
22 * ip_vs_sync: sync connection info from master load balancer to backups
23 * through multicast
24 *
25 * Changes:
26 * Alexandre Cassen : Added master & backup support at a time.
27 * Alexandre Cassen : Added SyncID support for incoming sync
28 * messages filtering.
29 * Justin Ossevoort : Fix endian problem on sync message size.
30 * Hans Schillstrom : Added Version 1: i.e. IPv6,
31 * Persistence support, fwmark and time-out.
32 */
33
34 #define KMSG_COMPONENT "IPVS"
35 #define pr_fmt(fmt) KMSG_COMPONENT ": " fmt
36
37 #include <linux/module.h>
38 #include <linux/slab.h>
39 #include <linux/inetdevice.h>
40 #include <linux/net.h>
41 #include <linux/completion.h>
42 #include <linux/delay.h>
43 #include <linux/skbuff.h>
44 #include <linux/in.h>
45 #include <linux/igmp.h> /* for ip_mc_join_group */
46 #include <linux/udp.h>
47 #include <linux/err.h>
48 #include <linux/kthread.h>
49 #include <linux/wait.h>
50 #include <linux/kernel.h>
51
52 #include <asm/unaligned.h> /* Used for ntoh_seq and hton_seq */
53
54 #include <net/ip.h>
55 #include <net/sock.h>
56
57 #include <net/ip_vs.h>
58
59 #define IP_VS_SYNC_GROUP 0xe0000051 /* multicast addr - 224.0.0.81 */
60 #define IP_VS_SYNC_PORT 8848 /* multicast port */
61
62 #define SYNC_PROTO_VER 1 /* Protocol version in header */
63
64 /*
65 * IPVS sync connection entry
66 * Version 0, i.e. original version.
67 */
68 struct ip_vs_sync_conn_v0 {
69 __u8 reserved;
70
71 /* Protocol, addresses and port numbers */
72 __u8 protocol; /* Which protocol (TCP/UDP) */
73 __be16 cport;
74 __be16 vport;
75 __be16 dport;
76 __be32 caddr; /* client address */
77 __be32 vaddr; /* virtual address */
78 __be32 daddr; /* destination address */
79
80 /* Flags and state transition */
81 __be16 flags; /* status flags */
82 __be16 state; /* state info */
83
84 /* The sequence options start here */
85 };
86
87 struct ip_vs_sync_conn_options {
88 struct ip_vs_seq in_seq; /* incoming seq. struct */
89 struct ip_vs_seq out_seq; /* outgoing seq. struct */
90 };
91
92 /*
93 Sync Connection format (sync_conn)
94
95 0 1 2 3
96 0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1
97 +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
98 | Type | Protocol | Ver. | Size |
99 +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
100 | Flags |
101 +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
102 | State | cport |
103 +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
104 | vport | dport |
105 +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
106 | fwmark |
107 +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
108 | timeout (in sec.) |
109 +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
110 | ... |
111 | IP-Addresses (v4 or v6) |
112 | ... |
113 +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
114 Optional Parameters.
115 +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
116 | Param. Type | Param. Length | Param. data |
117 +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+ |
118 | ... |
119 | +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
120 | | Param Type | Param. Length |
121 +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
122 | Param data |
123 | Last Param data should be padded for 32 bit alignment |
124 +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
125 */
126
127 /*
128 * Type 0, IPv4 sync connection format
129 */
130 struct ip_vs_sync_v4 {
131 __u8 type;
132 __u8 protocol; /* Which protocol (TCP/UDP) */
133 __be16 ver_size; /* Version msb 4 bits */
134 /* Flags and state transition */
135 __be32 flags; /* status flags */
136 __be16 state; /* state info */
137 /* Protocol, addresses and port numbers */
138 __be16 cport;
139 __be16 vport;
140 __be16 dport;
141 __be32 fwmark; /* Firewall mark from skb */
142 __be32 timeout; /* cp timeout */
143 __be32 caddr; /* client address */
144 __be32 vaddr; /* virtual address */
145 __be32 daddr; /* destination address */
146 /* The sequence options start here */
147 /* PE data padded to 32bit alignment after seq. options */
148 };
149 /*
150 * Type 2 messages IPv6
151 */
152 struct ip_vs_sync_v6 {
153 __u8 type;
154 __u8 protocol; /* Which protocol (TCP/UDP) */
155 __be16 ver_size; /* Version msb 4 bits */
156 /* Flags and state transition */
157 __be32 flags; /* status flags */
158 __be16 state; /* state info */
159 /* Protocol, addresses and port numbers */
160 __be16 cport;
161 __be16 vport;
162 __be16 dport;
163 __be32 fwmark; /* Firewall mark from skb */
164 __be32 timeout; /* cp timeout */
165 struct in6_addr caddr; /* client address */
166 struct in6_addr vaddr; /* virtual address */
167 struct in6_addr daddr; /* destination address */
168 /* The sequence options start here */
169 /* PE data padded to 32bit alignment after seq. options */
170 };
171
172 union ip_vs_sync_conn {
173 struct ip_vs_sync_v4 v4;
174 struct ip_vs_sync_v6 v6;
175 };
176
177 /* Bits in Type field in above */
178 #define STYPE_INET6 0
179 #define STYPE_F_INET6 (1 << STYPE_INET6)
180
181 #define SVER_SHIFT 12 /* Shift to get version */
182 #define SVER_MASK 0x0fff /* Mask to strip version */
183
184 #define IPVS_OPT_SEQ_DATA 1
185 #define IPVS_OPT_PE_DATA 2
186 #define IPVS_OPT_PE_NAME 3
187 #define IPVS_OPT_PARAM 7
188
189 #define IPVS_OPT_F_SEQ_DATA (1 << (IPVS_OPT_SEQ_DATA-1))
190 #define IPVS_OPT_F_PE_DATA (1 << (IPVS_OPT_PE_DATA-1))
191 #define IPVS_OPT_F_PE_NAME (1 << (IPVS_OPT_PE_NAME-1))
192 #define IPVS_OPT_F_PARAM (1 << (IPVS_OPT_PARAM-1))
193
194 struct ip_vs_sync_thread_data {
195 struct net *net;
196 struct socket *sock;
197 char *buf;
198 };
199
200 /* Version 0 definition of packet sizes */
201 #define SIMPLE_CONN_SIZE (sizeof(struct ip_vs_sync_conn_v0))
202 #define FULL_CONN_SIZE \
203 (sizeof(struct ip_vs_sync_conn_v0) + sizeof(struct ip_vs_sync_conn_options))
204
205
206 /*
207 The master mulitcasts messages (Datagrams) to the backup load balancers
208 in the following format.
209
210 Version 1:
211 Note, first byte should be Zero, so ver 0 receivers will drop the packet.
212
213 0 1 2 3
214 0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1
215 +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
216 | 0 | SyncID | Size |
217 +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
218 | Count Conns | Version | Reserved, set to Zero |
219 +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
220 | |
221 | IPVS Sync Connection (1) |
222 +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
223 | . |
224 ~ . ~
225 | . |
226 +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
227 | |
228 | IPVS Sync Connection (n) |
229 +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
230
231 Version 0 Header
232 0 1 2 3
233 0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1
234 +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
235 | Count Conns | SyncID | Size |
236 +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
237 | IPVS Sync Connection (1) |
238 */
239
240 #define SYNC_MESG_HEADER_LEN 4
241 #define MAX_CONNS_PER_SYNCBUFF 255 /* nr_conns in ip_vs_sync_mesg is 8 bit */
242
243 /* Version 0 header */
244 struct ip_vs_sync_mesg_v0 {
245 __u8 nr_conns;
246 __u8 syncid;
247 __u16 size;
248
249 /* ip_vs_sync_conn entries start here */
250 };
251
252 /* Version 1 header */
253 struct ip_vs_sync_mesg {
254 __u8 reserved; /* must be zero */
255 __u8 syncid;
256 __u16 size;
257 __u8 nr_conns;
258 __s8 version; /* SYNC_PROTO_VER */
259 __u16 spare;
260 /* ip_vs_sync_conn entries start here */
261 };
262
263 struct ip_vs_sync_buff {
264 struct list_head list;
265 unsigned long firstuse;
266
267 /* pointers for the message data */
268 struct ip_vs_sync_mesg *mesg;
269 unsigned char *head;
270 unsigned char *end;
271 };
272
273 /* multicast addr */
274 static struct sockaddr_in mcast_addr = {
275 .sin_family = AF_INET,
276 .sin_port = cpu_to_be16(IP_VS_SYNC_PORT),
277 .sin_addr.s_addr = cpu_to_be32(IP_VS_SYNC_GROUP),
278 };
279
280 /*
281 * Copy of struct ip_vs_seq
282 * From unaligned network order to aligned host order
283 */
284 static void ntoh_seq(struct ip_vs_seq *no, struct ip_vs_seq *ho)
285 {
286 ho->init_seq = get_unaligned_be32(&no->init_seq);
287 ho->delta = get_unaligned_be32(&no->delta);
288 ho->previous_delta = get_unaligned_be32(&no->previous_delta);
289 }
290
291 /*
292 * Copy of struct ip_vs_seq
293 * From Aligned host order to unaligned network order
294 */
295 static void hton_seq(struct ip_vs_seq *ho, struct ip_vs_seq *no)
296 {
297 put_unaligned_be32(ho->init_seq, &no->init_seq);
298 put_unaligned_be32(ho->delta, &no->delta);
299 put_unaligned_be32(ho->previous_delta, &no->previous_delta);
300 }
301
302 static inline struct ip_vs_sync_buff *sb_dequeue(struct netns_ipvs *ipvs)
303 {
304 struct ip_vs_sync_buff *sb;
305
306 spin_lock_bh(&ipvs->sync_lock);
307 if (list_empty(&ipvs->sync_queue)) {
308 sb = NULL;
309 } else {
310 sb = list_entry(ipvs->sync_queue.next,
311 struct ip_vs_sync_buff,
312 list);
313 list_del(&sb->list);
314 }
315 spin_unlock_bh(&ipvs->sync_lock);
316
317 return sb;
318 }
319
320 /*
321 * Create a new sync buffer for Version 1 proto.
322 */
323 static inline struct ip_vs_sync_buff *
324 ip_vs_sync_buff_create(struct netns_ipvs *ipvs)
325 {
326 struct ip_vs_sync_buff *sb;
327
328 if (!(sb=kmalloc(sizeof(struct ip_vs_sync_buff), GFP_ATOMIC)))
329 return NULL;
330
331 sb->mesg = kmalloc(ipvs->send_mesg_maxlen, GFP_ATOMIC);
332 if (!sb->mesg) {
333 kfree(sb);
334 return NULL;
335 }
336 sb->mesg->reserved = 0; /* old nr_conns i.e. must be zeo now */
337 sb->mesg->version = SYNC_PROTO_VER;
338 sb->mesg->syncid = ipvs->master_syncid;
339 sb->mesg->size = sizeof(struct ip_vs_sync_mesg);
340 sb->mesg->nr_conns = 0;
341 sb->mesg->spare = 0;
342 sb->head = (unsigned char *)sb->mesg + sizeof(struct ip_vs_sync_mesg);
343 sb->end = (unsigned char *)sb->mesg + ipvs->send_mesg_maxlen;
344
345 sb->firstuse = jiffies;
346 return sb;
347 }
348
349 static inline void ip_vs_sync_buff_release(struct ip_vs_sync_buff *sb)
350 {
351 kfree(sb->mesg);
352 kfree(sb);
353 }
354
355 static inline void sb_queue_tail(struct netns_ipvs *ipvs)
356 {
357 struct ip_vs_sync_buff *sb = ipvs->sync_buff;
358
359 spin_lock(&ipvs->sync_lock);
360 if (ipvs->sync_state & IP_VS_STATE_MASTER)
361 list_add_tail(&sb->list, &ipvs->sync_queue);
362 else
363 ip_vs_sync_buff_release(sb);
364 spin_unlock(&ipvs->sync_lock);
365 }
366
367 /*
368 * Get the current sync buffer if it has been created for more
369 * than the specified time or the specified time is zero.
370 */
371 static inline struct ip_vs_sync_buff *
372 get_curr_sync_buff(struct netns_ipvs *ipvs, unsigned long time)
373 {
374 struct ip_vs_sync_buff *sb;
375
376 spin_lock_bh(&ipvs->sync_buff_lock);
377 if (ipvs->sync_buff &&
378 time_after_eq(jiffies - ipvs->sync_buff->firstuse, time)) {
379 sb = ipvs->sync_buff;
380 ipvs->sync_buff = NULL;
381 } else
382 sb = NULL;
383 spin_unlock_bh(&ipvs->sync_buff_lock);
384 return sb;
385 }
386
387 /*
388 * Switch mode from sending version 0 or 1
389 * - must handle sync_buf
390 */
391 void ip_vs_sync_switch_mode(struct net *net, int mode)
392 {
393 struct netns_ipvs *ipvs = net_ipvs(net);
394
395 if (!(ipvs->sync_state & IP_VS_STATE_MASTER))
396 return;
397 if (mode == sysctl_sync_ver(ipvs) || !ipvs->sync_buff)
398 return;
399
400 spin_lock_bh(&ipvs->sync_buff_lock);
401 /* Buffer empty ? then let buf_create do the job */
402 if (ipvs->sync_buff->mesg->size <= sizeof(struct ip_vs_sync_mesg)) {
403 kfree(ipvs->sync_buff);
404 ipvs->sync_buff = NULL;
405 } else {
406 spin_lock_bh(&ipvs->sync_lock);
407 if (ipvs->sync_state & IP_VS_STATE_MASTER)
408 list_add_tail(&ipvs->sync_buff->list,
409 &ipvs->sync_queue);
410 else
411 ip_vs_sync_buff_release(ipvs->sync_buff);
412 spin_unlock_bh(&ipvs->sync_lock);
413 }
414 spin_unlock_bh(&ipvs->sync_buff_lock);
415 }
416
417 /*
418 * Create a new sync buffer for Version 0 proto.
419 */
420 static inline struct ip_vs_sync_buff *
421 ip_vs_sync_buff_create_v0(struct netns_ipvs *ipvs)
422 {
423 struct ip_vs_sync_buff *sb;
424 struct ip_vs_sync_mesg_v0 *mesg;
425
426 if (!(sb=kmalloc(sizeof(struct ip_vs_sync_buff), GFP_ATOMIC)))
427 return NULL;
428
429 sb->mesg = kmalloc(ipvs->send_mesg_maxlen, GFP_ATOMIC);
430 if (!sb->mesg) {
431 kfree(sb);
432 return NULL;
433 }
434 mesg = (struct ip_vs_sync_mesg_v0 *)sb->mesg;
435 mesg->nr_conns = 0;
436 mesg->syncid = ipvs->master_syncid;
437 mesg->size = sizeof(struct ip_vs_sync_mesg_v0);
438 sb->head = (unsigned char *)mesg + sizeof(struct ip_vs_sync_mesg_v0);
439 sb->end = (unsigned char *)mesg + ipvs->send_mesg_maxlen;
440 sb->firstuse = jiffies;
441 return sb;
442 }
443
444 /*
445 * Version 0 , could be switched in by sys_ctl.
446 * Add an ip_vs_conn information into the current sync_buff.
447 */
448 void ip_vs_sync_conn_v0(struct net *net, struct ip_vs_conn *cp)
449 {
450 struct netns_ipvs *ipvs = net_ipvs(net);
451 struct ip_vs_sync_mesg_v0 *m;
452 struct ip_vs_sync_conn_v0 *s;
453 int len;
454
455 if (unlikely(cp->af != AF_INET))
456 return;
457 /* Do not sync ONE PACKET */
458 if (cp->flags & IP_VS_CONN_F_ONE_PACKET)
459 return;
460
461 spin_lock(&ipvs->sync_buff_lock);
462 if (!ipvs->sync_buff) {
463 ipvs->sync_buff =
464 ip_vs_sync_buff_create_v0(ipvs);
465 if (!ipvs->sync_buff) {
466 spin_unlock(&ipvs->sync_buff_lock);
467 pr_err("ip_vs_sync_buff_create failed.\n");
468 return;
469 }
470 }
471
472 len = (cp->flags & IP_VS_CONN_F_SEQ_MASK) ? FULL_CONN_SIZE :
473 SIMPLE_CONN_SIZE;
474 m = (struct ip_vs_sync_mesg_v0 *)ipvs->sync_buff->mesg;
475 s = (struct ip_vs_sync_conn_v0 *)ipvs->sync_buff->head;
476
477 /* copy members */
478 s->reserved = 0;
479 s->protocol = cp->protocol;
480 s->cport = cp->cport;
481 s->vport = cp->vport;
482 s->dport = cp->dport;
483 s->caddr = cp->caddr.ip;
484 s->vaddr = cp->vaddr.ip;
485 s->daddr = cp->daddr.ip;
486 s->flags = htons(cp->flags & ~IP_VS_CONN_F_HASHED);
487 s->state = htons(cp->state);
488 if (cp->flags & IP_VS_CONN_F_SEQ_MASK) {
489 struct ip_vs_sync_conn_options *opt =
490 (struct ip_vs_sync_conn_options *)&s[1];
491 memcpy(opt, &cp->in_seq, sizeof(*opt));
492 }
493
494 m->nr_conns++;
495 m->size += len;
496 ipvs->sync_buff->head += len;
497
498 /* check if there is a space for next one */
499 if (ipvs->sync_buff->head + FULL_CONN_SIZE > ipvs->sync_buff->end) {
500 sb_queue_tail(ipvs);
501 ipvs->sync_buff = NULL;
502 }
503 spin_unlock(&ipvs->sync_buff_lock);
504
505 /* synchronize its controller if it has */
506 if (cp->control)
507 ip_vs_sync_conn(net, cp->control);
508 }
509
510 /*
511 * Add an ip_vs_conn information into the current sync_buff.
512 * Called by ip_vs_in.
513 * Sending Version 1 messages
514 */
515 void ip_vs_sync_conn(struct net *net, struct ip_vs_conn *cp)
516 {
517 struct netns_ipvs *ipvs = net_ipvs(net);
518 struct ip_vs_sync_mesg *m;
519 union ip_vs_sync_conn *s;
520 __u8 *p;
521 unsigned int len, pe_name_len, pad;
522
523 /* Handle old version of the protocol */
524 if (sysctl_sync_ver(ipvs) == 0) {
525 ip_vs_sync_conn_v0(net, cp);
526 return;
527 }
528 /* Do not sync ONE PACKET */
529 if (cp->flags & IP_VS_CONN_F_ONE_PACKET)
530 goto control;
531 sloop:
532 /* Sanity checks */
533 pe_name_len = 0;
534 if (cp->pe_data_len) {
535 if (!cp->pe_data || !cp->dest) {
536 IP_VS_ERR_RL("SYNC, connection pe_data invalid\n");
537 return;
538 }
539 pe_name_len = strnlen(cp->pe->name, IP_VS_PENAME_MAXLEN);
540 }
541
542 spin_lock(&ipvs->sync_buff_lock);
543
544 #ifdef CONFIG_IP_VS_IPV6
545 if (cp->af == AF_INET6)
546 len = sizeof(struct ip_vs_sync_v6);
547 else
548 #endif
549 len = sizeof(struct ip_vs_sync_v4);
550
551 if (cp->flags & IP_VS_CONN_F_SEQ_MASK)
552 len += sizeof(struct ip_vs_sync_conn_options) + 2;
553
554 if (cp->pe_data_len)
555 len += cp->pe_data_len + 2; /* + Param hdr field */
556 if (pe_name_len)
557 len += pe_name_len + 2;
558
559 /* check if there is a space for this one */
560 pad = 0;
561 if (ipvs->sync_buff) {
562 pad = (4 - (size_t)ipvs->sync_buff->head) & 3;
563 if (ipvs->sync_buff->head + len + pad > ipvs->sync_buff->end) {
564 sb_queue_tail(ipvs);
565 ipvs->sync_buff = NULL;
566 pad = 0;
567 }
568 }
569
570 if (!ipvs->sync_buff) {
571 ipvs->sync_buff = ip_vs_sync_buff_create(ipvs);
572 if (!ipvs->sync_buff) {
573 spin_unlock(&ipvs->sync_buff_lock);
574 pr_err("ip_vs_sync_buff_create failed.\n");
575 return;
576 }
577 }
578
579 m = ipvs->sync_buff->mesg;
580 p = ipvs->sync_buff->head;
581 ipvs->sync_buff->head += pad + len;
582 m->size += pad + len;
583 /* Add ev. padding from prev. sync_conn */
584 while (pad--)
585 *(p++) = 0;
586
587 s = (union ip_vs_sync_conn *)p;
588
589 /* Set message type & copy members */
590 s->v4.type = (cp->af == AF_INET6 ? STYPE_F_INET6 : 0);
591 s->v4.ver_size = htons(len & SVER_MASK); /* Version 0 */
592 s->v4.flags = htonl(cp->flags & ~IP_VS_CONN_F_HASHED);
593 s->v4.state = htons(cp->state);
594 s->v4.protocol = cp->protocol;
595 s->v4.cport = cp->cport;
596 s->v4.vport = cp->vport;
597 s->v4.dport = cp->dport;
598 s->v4.fwmark = htonl(cp->fwmark);
599 s->v4.timeout = htonl(cp->timeout / HZ);
600 m->nr_conns++;
601
602 #ifdef CONFIG_IP_VS_IPV6
603 if (cp->af == AF_INET6) {
604 p += sizeof(struct ip_vs_sync_v6);
605 ipv6_addr_copy(&s->v6.caddr, &cp->caddr.in6);
606 ipv6_addr_copy(&s->v6.vaddr, &cp->vaddr.in6);
607 ipv6_addr_copy(&s->v6.daddr, &cp->daddr.in6);
608 } else
609 #endif
610 {
611 p += sizeof(struct ip_vs_sync_v4); /* options ptr */
612 s->v4.caddr = cp->caddr.ip;
613 s->v4.vaddr = cp->vaddr.ip;
614 s->v4.daddr = cp->daddr.ip;
615 }
616 if (cp->flags & IP_VS_CONN_F_SEQ_MASK) {
617 *(p++) = IPVS_OPT_SEQ_DATA;
618 *(p++) = sizeof(struct ip_vs_sync_conn_options);
619 hton_seq((struct ip_vs_seq *)p, &cp->in_seq);
620 p += sizeof(struct ip_vs_seq);
621 hton_seq((struct ip_vs_seq *)p, &cp->out_seq);
622 p += sizeof(struct ip_vs_seq);
623 }
624 /* Handle pe data */
625 if (cp->pe_data_len && cp->pe_data) {
626 *(p++) = IPVS_OPT_PE_DATA;
627 *(p++) = cp->pe_data_len;
628 memcpy(p, cp->pe_data, cp->pe_data_len);
629 p += cp->pe_data_len;
630 if (pe_name_len) {
631 /* Add PE_NAME */
632 *(p++) = IPVS_OPT_PE_NAME;
633 *(p++) = pe_name_len;
634 memcpy(p, cp->pe->name, pe_name_len);
635 p += pe_name_len;
636 }
637 }
638
639 spin_unlock(&ipvs->sync_buff_lock);
640
641 control:
642 /* synchronize its controller if it has */
643 cp = cp->control;
644 if (!cp)
645 return;
646 /*
647 * Reduce sync rate for templates
648 * i.e only increment in_pkts for Templates.
649 */
650 if (cp->flags & IP_VS_CONN_F_TEMPLATE) {
651 int pkts = atomic_add_return(1, &cp->in_pkts);
652
653 if (pkts % sysctl_sync_period(ipvs) != 1)
654 return;
655 }
656 goto sloop;
657 }
658
659 /*
660 * fill_param used by version 1
661 */
662 static inline int
663 ip_vs_conn_fill_param_sync(struct net *net, int af, union ip_vs_sync_conn *sc,
664 struct ip_vs_conn_param *p,
665 __u8 *pe_data, unsigned int pe_data_len,
666 __u8 *pe_name, unsigned int pe_name_len)
667 {
668 #ifdef CONFIG_IP_VS_IPV6
669 if (af == AF_INET6)
670 ip_vs_conn_fill_param(net, af, sc->v6.protocol,
671 (const union nf_inet_addr *)&sc->v6.caddr,
672 sc->v6.cport,
673 (const union nf_inet_addr *)&sc->v6.vaddr,
674 sc->v6.vport, p);
675 else
676 #endif
677 ip_vs_conn_fill_param(net, af, sc->v4.protocol,
678 (const union nf_inet_addr *)&sc->v4.caddr,
679 sc->v4.cport,
680 (const union nf_inet_addr *)&sc->v4.vaddr,
681 sc->v4.vport, p);
682 /* Handle pe data */
683 if (pe_data_len) {
684 if (pe_name_len) {
685 char buff[IP_VS_PENAME_MAXLEN+1];
686
687 memcpy(buff, pe_name, pe_name_len);
688 buff[pe_name_len]=0;
689 p->pe = __ip_vs_pe_getbyname(buff);
690 if (!p->pe) {
691 IP_VS_DBG(3, "BACKUP, no %s engine found/loaded\n",
692 buff);
693 return 1;
694 }
695 } else {
696 IP_VS_ERR_RL("BACKUP, Invalid PE parameters\n");
697 return 1;
698 }
699
700 p->pe_data = kmemdup(pe_data, pe_data_len, GFP_ATOMIC);
701 if (!p->pe_data) {
702 if (p->pe->module)
703 module_put(p->pe->module);
704 return -ENOMEM;
705 }
706 p->pe_data_len = pe_data_len;
707 }
708 return 0;
709 }
710
711 /*
712 * Connection Add / Update.
713 * Common for version 0 and 1 reception of backup sync_conns.
714 * Param: ...
715 * timeout is in sec.
716 */
717 static void ip_vs_proc_conn(struct net *net, struct ip_vs_conn_param *param,
718 unsigned int flags, unsigned int state,
719 unsigned int protocol, unsigned int type,
720 const union nf_inet_addr *daddr, __be16 dport,
721 unsigned long timeout, __u32 fwmark,
722 struct ip_vs_sync_conn_options *opt)
723 {
724 struct ip_vs_dest *dest;
725 struct ip_vs_conn *cp;
726 struct netns_ipvs *ipvs = net_ipvs(net);
727
728 if (!(flags & IP_VS_CONN_F_TEMPLATE))
729 cp = ip_vs_conn_in_get(param);
730 else
731 cp = ip_vs_ct_in_get(param);
732
733 if (cp && param->pe_data) /* Free pe_data */
734 kfree(param->pe_data);
735 if (!cp) {
736 /*
737 * Find the appropriate destination for the connection.
738 * If it is not found the connection will remain unbound
739 * but still handled.
740 */
741 dest = ip_vs_find_dest(net, type, daddr, dport, param->vaddr,
742 param->vport, protocol, fwmark);
743
744 /* Set the approprite ativity flag */
745 if (protocol == IPPROTO_TCP) {
746 if (state != IP_VS_TCP_S_ESTABLISHED)
747 flags |= IP_VS_CONN_F_INACTIVE;
748 else
749 flags &= ~IP_VS_CONN_F_INACTIVE;
750 } else if (protocol == IPPROTO_SCTP) {
751 if (state != IP_VS_SCTP_S_ESTABLISHED)
752 flags |= IP_VS_CONN_F_INACTIVE;
753 else
754 flags &= ~IP_VS_CONN_F_INACTIVE;
755 }
756 cp = ip_vs_conn_new(param, daddr, dport, flags, dest, fwmark);
757 if (dest)
758 atomic_dec(&dest->refcnt);
759 if (!cp) {
760 if (param->pe_data)
761 kfree(param->pe_data);
762 IP_VS_DBG(2, "BACKUP, add new conn. failed\n");
763 return;
764 }
765 } else if (!cp->dest) {
766 dest = ip_vs_try_bind_dest(cp);
767 if (dest)
768 atomic_dec(&dest->refcnt);
769 } else if ((cp->dest) && (cp->protocol == IPPROTO_TCP) &&
770 (cp->state != state)) {
771 /* update active/inactive flag for the connection */
772 dest = cp->dest;
773 if (!(cp->flags & IP_VS_CONN_F_INACTIVE) &&
774 (state != IP_VS_TCP_S_ESTABLISHED)) {
775 atomic_dec(&dest->activeconns);
776 atomic_inc(&dest->inactconns);
777 cp->flags |= IP_VS_CONN_F_INACTIVE;
778 } else if ((cp->flags & IP_VS_CONN_F_INACTIVE) &&
779 (state == IP_VS_TCP_S_ESTABLISHED)) {
780 atomic_inc(&dest->activeconns);
781 atomic_dec(&dest->inactconns);
782 cp->flags &= ~IP_VS_CONN_F_INACTIVE;
783 }
784 } else if ((cp->dest) && (cp->protocol == IPPROTO_SCTP) &&
785 (cp->state != state)) {
786 dest = cp->dest;
787 if (!(cp->flags & IP_VS_CONN_F_INACTIVE) &&
788 (state != IP_VS_SCTP_S_ESTABLISHED)) {
789 atomic_dec(&dest->activeconns);
790 atomic_inc(&dest->inactconns);
791 cp->flags &= ~IP_VS_CONN_F_INACTIVE;
792 }
793 }
794
795 if (opt)
796 memcpy(&cp->in_seq, opt, sizeof(*opt));
797 atomic_set(&cp->in_pkts, sysctl_sync_threshold(ipvs));
798 cp->state = state;
799 cp->old_state = cp->state;
800 /*
801 * For Ver 0 messages style
802 * - Not possible to recover the right timeout for templates
803 * - can not find the right fwmark
804 * virtual service. If needed, we can do it for
805 * non-fwmark persistent services.
806 * Ver 1 messages style.
807 * - No problem.
808 */
809 if (timeout) {
810 if (timeout > MAX_SCHEDULE_TIMEOUT / HZ)
811 timeout = MAX_SCHEDULE_TIMEOUT / HZ;
812 cp->timeout = timeout*HZ;
813 } else {
814 struct ip_vs_proto_data *pd;
815
816 pd = ip_vs_proto_data_get(net, protocol);
817 if (!(flags & IP_VS_CONN_F_TEMPLATE) && pd && pd->timeout_table)
818 cp->timeout = pd->timeout_table[state];
819 else
820 cp->timeout = (3*60*HZ);
821 }
822 ip_vs_conn_put(cp);
823 }
824
825 /*
826 * Process received multicast message for Version 0
827 */
828 static void ip_vs_process_message_v0(struct net *net, const char *buffer,
829 const size_t buflen)
830 {
831 struct ip_vs_sync_mesg_v0 *m = (struct ip_vs_sync_mesg_v0 *)buffer;
832 struct ip_vs_sync_conn_v0 *s;
833 struct ip_vs_sync_conn_options *opt;
834 struct ip_vs_protocol *pp;
835 struct ip_vs_conn_param param;
836 char *p;
837 int i;
838
839 p = (char *)buffer + sizeof(struct ip_vs_sync_mesg_v0);
840 for (i=0; i<m->nr_conns; i++) {
841 unsigned flags, state;
842
843 if (p + SIMPLE_CONN_SIZE > buffer+buflen) {
844 IP_VS_ERR_RL("BACKUP v0, bogus conn\n");
845 return;
846 }
847 s = (struct ip_vs_sync_conn_v0 *) p;
848 flags = ntohs(s->flags) | IP_VS_CONN_F_SYNC;
849 flags &= ~IP_VS_CONN_F_HASHED;
850 if (flags & IP_VS_CONN_F_SEQ_MASK) {
851 opt = (struct ip_vs_sync_conn_options *)&s[1];
852 p += FULL_CONN_SIZE;
853 if (p > buffer+buflen) {
854 IP_VS_ERR_RL("BACKUP v0, Dropping buffer bogus conn options\n");
855 return;
856 }
857 } else {
858 opt = NULL;
859 p += SIMPLE_CONN_SIZE;
860 }
861
862 state = ntohs(s->state);
863 if (!(flags & IP_VS_CONN_F_TEMPLATE)) {
864 pp = ip_vs_proto_get(s->protocol);
865 if (!pp) {
866 IP_VS_DBG(2, "BACKUP v0, Unsupported protocol %u\n",
867 s->protocol);
868 continue;
869 }
870 if (state >= pp->num_states) {
871 IP_VS_DBG(2, "BACKUP v0, Invalid %s state %u\n",
872 pp->name, state);
873 continue;
874 }
875 } else {
876 /* protocol in templates is not used for state/timeout */
877 if (state > 0) {
878 IP_VS_DBG(2, "BACKUP v0, Invalid template state %u\n",
879 state);
880 state = 0;
881 }
882 }
883
884 ip_vs_conn_fill_param(net, AF_INET, s->protocol,
885 (const union nf_inet_addr *)&s->caddr,
886 s->cport,
887 (const union nf_inet_addr *)&s->vaddr,
888 s->vport, &param);
889
890 /* Send timeout as Zero */
891 ip_vs_proc_conn(net, &param, flags, state, s->protocol, AF_INET,
892 (union nf_inet_addr *)&s->daddr, s->dport,
893 0, 0, opt);
894 }
895 }
896
897 /*
898 * Handle options
899 */
900 static inline int ip_vs_proc_seqopt(__u8 *p, unsigned int plen,
901 __u32 *opt_flags,
902 struct ip_vs_sync_conn_options *opt)
903 {
904 struct ip_vs_sync_conn_options *topt;
905
906 topt = (struct ip_vs_sync_conn_options *)p;
907
908 if (plen != sizeof(struct ip_vs_sync_conn_options)) {
909 IP_VS_DBG(2, "BACKUP, bogus conn options length\n");
910 return -EINVAL;
911 }
912 if (*opt_flags & IPVS_OPT_F_SEQ_DATA) {
913 IP_VS_DBG(2, "BACKUP, conn options found twice\n");
914 return -EINVAL;
915 }
916 ntoh_seq(&topt->in_seq, &opt->in_seq);
917 ntoh_seq(&topt->out_seq, &opt->out_seq);
918 *opt_flags |= IPVS_OPT_F_SEQ_DATA;
919 return 0;
920 }
921
922 static int ip_vs_proc_str(__u8 *p, unsigned int plen, unsigned int *data_len,
923 __u8 **data, unsigned int maxlen,
924 __u32 *opt_flags, __u32 flag)
925 {
926 if (plen > maxlen) {
927 IP_VS_DBG(2, "BACKUP, bogus par.data len > %d\n", maxlen);
928 return -EINVAL;
929 }
930 if (*opt_flags & flag) {
931 IP_VS_DBG(2, "BACKUP, Par.data found twice 0x%x\n", flag);
932 return -EINVAL;
933 }
934 *data_len = plen;
935 *data = p;
936 *opt_flags |= flag;
937 return 0;
938 }
939 /*
940 * Process a Version 1 sync. connection
941 */
942 static inline int ip_vs_proc_sync_conn(struct net *net, __u8 *p, __u8 *msg_end)
943 {
944 struct ip_vs_sync_conn_options opt;
945 union ip_vs_sync_conn *s;
946 struct ip_vs_protocol *pp;
947 struct ip_vs_conn_param param;
948 __u32 flags;
949 unsigned int af, state, pe_data_len=0, pe_name_len=0;
950 __u8 *pe_data=NULL, *pe_name=NULL;
951 __u32 opt_flags=0;
952 int retc=0;
953
954 s = (union ip_vs_sync_conn *) p;
955
956 if (s->v6.type & STYPE_F_INET6) {
957 #ifdef CONFIG_IP_VS_IPV6
958 af = AF_INET6;
959 p += sizeof(struct ip_vs_sync_v6);
960 #else
961 IP_VS_DBG(3,"BACKUP, IPv6 msg received, and IPVS is not compiled for IPv6\n");
962 retc = 10;
963 goto out;
964 #endif
965 } else if (!s->v4.type) {
966 af = AF_INET;
967 p += sizeof(struct ip_vs_sync_v4);
968 } else {
969 return -10;
970 }
971 if (p > msg_end)
972 return -20;
973
974 /* Process optional params check Type & Len. */
975 while (p < msg_end) {
976 int ptype;
977 int plen;
978
979 if (p+2 > msg_end)
980 return -30;
981 ptype = *(p++);
982 plen = *(p++);
983
984 if (!plen || ((p + plen) > msg_end))
985 return -40;
986 /* Handle seq option p = param data */
987 switch (ptype & ~IPVS_OPT_F_PARAM) {
988 case IPVS_OPT_SEQ_DATA:
989 if (ip_vs_proc_seqopt(p, plen, &opt_flags, &opt))
990 return -50;
991 break;
992
993 case IPVS_OPT_PE_DATA:
994 if (ip_vs_proc_str(p, plen, &pe_data_len, &pe_data,
995 IP_VS_PEDATA_MAXLEN, &opt_flags,
996 IPVS_OPT_F_PE_DATA))
997 return -60;
998 break;
999
1000 case IPVS_OPT_PE_NAME:
1001 if (ip_vs_proc_str(p, plen,&pe_name_len, &pe_name,
1002 IP_VS_PENAME_MAXLEN, &opt_flags,
1003 IPVS_OPT_F_PE_NAME))
1004 return -70;
1005 break;
1006
1007 default:
1008 /* Param data mandatory ? */
1009 if (!(ptype & IPVS_OPT_F_PARAM)) {
1010 IP_VS_DBG(3, "BACKUP, Unknown mandatory param %d found\n",
1011 ptype & ~IPVS_OPT_F_PARAM);
1012 retc = 20;
1013 goto out;
1014 }
1015 }
1016 p += plen; /* Next option */
1017 }
1018
1019 /* Get flags and Mask off unsupported */
1020 flags = ntohl(s->v4.flags) & IP_VS_CONN_F_BACKUP_MASK;
1021 flags |= IP_VS_CONN_F_SYNC;
1022 state = ntohs(s->v4.state);
1023
1024 if (!(flags & IP_VS_CONN_F_TEMPLATE)) {
1025 pp = ip_vs_proto_get(s->v4.protocol);
1026 if (!pp) {
1027 IP_VS_DBG(3,"BACKUP, Unsupported protocol %u\n",
1028 s->v4.protocol);
1029 retc = 30;
1030 goto out;
1031 }
1032 if (state >= pp->num_states) {
1033 IP_VS_DBG(3, "BACKUP, Invalid %s state %u\n",
1034 pp->name, state);
1035 retc = 40;
1036 goto out;
1037 }
1038 } else {
1039 /* protocol in templates is not used for state/timeout */
1040 if (state > 0) {
1041 IP_VS_DBG(3, "BACKUP, Invalid template state %u\n",
1042 state);
1043 state = 0;
1044 }
1045 }
1046 if (ip_vs_conn_fill_param_sync(net, af, s, &param, pe_data,
1047 pe_data_len, pe_name, pe_name_len)) {
1048 retc = 50;
1049 goto out;
1050 }
1051 /* If only IPv4, just silent skip IPv6 */
1052 if (af == AF_INET)
1053 ip_vs_proc_conn(net, &param, flags, state, s->v4.protocol, af,
1054 (union nf_inet_addr *)&s->v4.daddr, s->v4.dport,
1055 ntohl(s->v4.timeout), ntohl(s->v4.fwmark),
1056 (opt_flags & IPVS_OPT_F_SEQ_DATA ? &opt : NULL)
1057 );
1058 #ifdef CONFIG_IP_VS_IPV6
1059 else
1060 ip_vs_proc_conn(net, &param, flags, state, s->v6.protocol, af,
1061 (union nf_inet_addr *)&s->v6.daddr, s->v6.dport,
1062 ntohl(s->v6.timeout), ntohl(s->v6.fwmark),
1063 (opt_flags & IPVS_OPT_F_SEQ_DATA ? &opt : NULL)
1064 );
1065 #endif
1066 return 0;
1067 /* Error exit */
1068 out:
1069 IP_VS_DBG(2, "BACKUP, Single msg dropped err:%d\n", retc);
1070 return retc;
1071
1072 }
1073 /*
1074 * Process received multicast message and create the corresponding
1075 * ip_vs_conn entries.
1076 * Handles Version 0 & 1
1077 */
1078 static void ip_vs_process_message(struct net *net, __u8 *buffer,
1079 const size_t buflen)
1080 {
1081 struct netns_ipvs *ipvs = net_ipvs(net);
1082 struct ip_vs_sync_mesg *m2 = (struct ip_vs_sync_mesg *)buffer;
1083 __u8 *p, *msg_end;
1084 int i, nr_conns;
1085
1086 if (buflen < sizeof(struct ip_vs_sync_mesg_v0)) {
1087 IP_VS_DBG(2, "BACKUP, message header too short\n");
1088 return;
1089 }
1090 /* Convert size back to host byte order */
1091 m2->size = ntohs(m2->size);
1092
1093 if (buflen != m2->size) {
1094 IP_VS_DBG(2, "BACKUP, bogus message size\n");
1095 return;
1096 }
1097 /* SyncID sanity check */
1098 if (ipvs->backup_syncid != 0 && m2->syncid != ipvs->backup_syncid) {
1099 IP_VS_DBG(7, "BACKUP, Ignoring syncid = %d\n", m2->syncid);
1100 return;
1101 }
1102 /* Handle version 1 message */
1103 if ((m2->version == SYNC_PROTO_VER) && (m2->reserved == 0)
1104 && (m2->spare == 0)) {
1105
1106 msg_end = buffer + sizeof(struct ip_vs_sync_mesg);
1107 nr_conns = m2->nr_conns;
1108
1109 for (i=0; i<nr_conns; i++) {
1110 union ip_vs_sync_conn *s;
1111 unsigned size;
1112 int retc;
1113
1114 p = msg_end;
1115 if (p + sizeof(s->v4) > buffer+buflen) {
1116 IP_VS_ERR_RL("BACKUP, Dropping buffer, to small\n");
1117 return;
1118 }
1119 s = (union ip_vs_sync_conn *)p;
1120 size = ntohs(s->v4.ver_size) & SVER_MASK;
1121 msg_end = p + size;
1122 /* Basic sanity checks */
1123 if (msg_end > buffer+buflen) {
1124 IP_VS_ERR_RL("BACKUP, Dropping buffer, msg > buffer\n");
1125 return;
1126 }
1127 if (ntohs(s->v4.ver_size) >> SVER_SHIFT) {
1128 IP_VS_ERR_RL("BACKUP, Dropping buffer, Unknown version %d\n",
1129 ntohs(s->v4.ver_size) >> SVER_SHIFT);
1130 return;
1131 }
1132 /* Process a single sync_conn */
1133 retc = ip_vs_proc_sync_conn(net, p, msg_end);
1134 if (retc < 0) {
1135 IP_VS_ERR_RL("BACKUP, Dropping buffer, Err: %d in decoding\n",
1136 retc);
1137 return;
1138 }
1139 /* Make sure we have 32 bit alignment */
1140 msg_end = p + ((size + 3) & ~3);
1141 }
1142 } else {
1143 /* Old type of message */
1144 ip_vs_process_message_v0(net, buffer, buflen);
1145 return;
1146 }
1147 }
1148
1149
1150 /*
1151 * Setup loopback of outgoing multicasts on a sending socket
1152 */
1153 static void set_mcast_loop(struct sock *sk, u_char loop)
1154 {
1155 struct inet_sock *inet = inet_sk(sk);
1156
1157 /* setsockopt(sock, SOL_IP, IP_MULTICAST_LOOP, &loop, sizeof(loop)); */
1158 lock_sock(sk);
1159 inet->mc_loop = loop ? 1 : 0;
1160 release_sock(sk);
1161 }
1162
1163 /*
1164 * Specify TTL for outgoing multicasts on a sending socket
1165 */
1166 static void set_mcast_ttl(struct sock *sk, u_char ttl)
1167 {
1168 struct inet_sock *inet = inet_sk(sk);
1169
1170 /* setsockopt(sock, SOL_IP, IP_MULTICAST_TTL, &ttl, sizeof(ttl)); */
1171 lock_sock(sk);
1172 inet->mc_ttl = ttl;
1173 release_sock(sk);
1174 }
1175
1176 /*
1177 * Specifiy default interface for outgoing multicasts
1178 */
1179 static int set_mcast_if(struct sock *sk, char *ifname)
1180 {
1181 struct net_device *dev;
1182 struct inet_sock *inet = inet_sk(sk);
1183 struct net *net = sock_net(sk);
1184
1185 dev = __dev_get_by_name(net, ifname);
1186 if (!dev)
1187 return -ENODEV;
1188
1189 if (sk->sk_bound_dev_if && dev->ifindex != sk->sk_bound_dev_if)
1190 return -EINVAL;
1191
1192 lock_sock(sk);
1193 inet->mc_index = dev->ifindex;
1194 /* inet->mc_addr = 0; */
1195 release_sock(sk);
1196
1197 return 0;
1198 }
1199
1200
1201 /*
1202 * Set the maximum length of sync message according to the
1203 * specified interface's MTU.
1204 */
1205 static int set_sync_mesg_maxlen(struct net *net, int sync_state)
1206 {
1207 struct netns_ipvs *ipvs = net_ipvs(net);
1208 struct net_device *dev;
1209 int num;
1210
1211 if (sync_state == IP_VS_STATE_MASTER) {
1212 dev = __dev_get_by_name(net, ipvs->master_mcast_ifn);
1213 if (!dev)
1214 return -ENODEV;
1215
1216 num = (dev->mtu - sizeof(struct iphdr) -
1217 sizeof(struct udphdr) -
1218 SYNC_MESG_HEADER_LEN - 20) / SIMPLE_CONN_SIZE;
1219 ipvs->send_mesg_maxlen = SYNC_MESG_HEADER_LEN +
1220 SIMPLE_CONN_SIZE * min(num, MAX_CONNS_PER_SYNCBUFF);
1221 IP_VS_DBG(7, "setting the maximum length of sync sending "
1222 "message %d.\n", ipvs->send_mesg_maxlen);
1223 } else if (sync_state == IP_VS_STATE_BACKUP) {
1224 dev = __dev_get_by_name(net, ipvs->backup_mcast_ifn);
1225 if (!dev)
1226 return -ENODEV;
1227
1228 ipvs->recv_mesg_maxlen = dev->mtu -
1229 sizeof(struct iphdr) - sizeof(struct udphdr);
1230 IP_VS_DBG(7, "setting the maximum length of sync receiving "
1231 "message %d.\n", ipvs->recv_mesg_maxlen);
1232 }
1233
1234 return 0;
1235 }
1236
1237
1238 /*
1239 * Join a multicast group.
1240 * the group is specified by a class D multicast address 224.0.0.0/8
1241 * in the in_addr structure passed in as a parameter.
1242 */
1243 static int
1244 join_mcast_group(struct sock *sk, struct in_addr *addr, char *ifname)
1245 {
1246 struct net *net = sock_net(sk);
1247 struct ip_mreqn mreq;
1248 struct net_device *dev;
1249 int ret;
1250
1251 memset(&mreq, 0, sizeof(mreq));
1252 memcpy(&mreq.imr_multiaddr, addr, sizeof(struct in_addr));
1253
1254 dev = __dev_get_by_name(net, ifname);
1255 if (!dev)
1256 return -ENODEV;
1257 if (sk->sk_bound_dev_if && dev->ifindex != sk->sk_bound_dev_if)
1258 return -EINVAL;
1259
1260 mreq.imr_ifindex = dev->ifindex;
1261
1262 lock_sock(sk);
1263 ret = ip_mc_join_group(sk, &mreq);
1264 release_sock(sk);
1265
1266 return ret;
1267 }
1268
1269
1270 static int bind_mcastif_addr(struct socket *sock, char *ifname)
1271 {
1272 struct net *net = sock_net(sock->sk);
1273 struct net_device *dev;
1274 __be32 addr;
1275 struct sockaddr_in sin;
1276
1277 dev = __dev_get_by_name(net, ifname);
1278 if (!dev)
1279 return -ENODEV;
1280
1281 addr = inet_select_addr(dev, 0, RT_SCOPE_UNIVERSE);
1282 if (!addr)
1283 pr_err("You probably need to specify IP address on "
1284 "multicast interface.\n");
1285
1286 IP_VS_DBG(7, "binding socket with (%s) %pI4\n",
1287 ifname, &addr);
1288
1289 /* Now bind the socket with the address of multicast interface */
1290 sin.sin_family = AF_INET;
1291 sin.sin_addr.s_addr = addr;
1292 sin.sin_port = 0;
1293
1294 return sock->ops->bind(sock, (struct sockaddr*)&sin, sizeof(sin));
1295 }
1296
1297 /*
1298 * Set up sending multicast socket over UDP
1299 */
1300 static struct socket *make_send_sock(struct net *net)
1301 {
1302 struct netns_ipvs *ipvs = net_ipvs(net);
1303 struct socket *sock;
1304 int result;
1305
1306 /* First create a socket */
1307 result = __sock_create(net, PF_INET, SOCK_DGRAM, IPPROTO_UDP, &sock, 1);
1308 if (result < 0) {
1309 pr_err("Error during creation of socket; terminating\n");
1310 return ERR_PTR(result);
1311 }
1312
1313 result = set_mcast_if(sock->sk, ipvs->master_mcast_ifn);
1314 if (result < 0) {
1315 pr_err("Error setting outbound mcast interface\n");
1316 goto error;
1317 }
1318
1319 set_mcast_loop(sock->sk, 0);
1320 set_mcast_ttl(sock->sk, 1);
1321
1322 result = bind_mcastif_addr(sock, ipvs->master_mcast_ifn);
1323 if (result < 0) {
1324 pr_err("Error binding address of the mcast interface\n");
1325 goto error;
1326 }
1327
1328 result = sock->ops->connect(sock, (struct sockaddr *) &mcast_addr,
1329 sizeof(struct sockaddr), 0);
1330 if (result < 0) {
1331 pr_err("Error connecting to the multicast addr\n");
1332 goto error;
1333 }
1334
1335 return sock;
1336
1337 error:
1338 sock_release(sock);
1339 return ERR_PTR(result);
1340 }
1341
1342
1343 /*
1344 * Set up receiving multicast socket over UDP
1345 */
1346 static struct socket *make_receive_sock(struct net *net)
1347 {
1348 struct netns_ipvs *ipvs = net_ipvs(net);
1349 struct socket *sock;
1350 int result;
1351
1352 /* First create a socket */
1353 result = __sock_create(net, PF_INET, SOCK_DGRAM, IPPROTO_UDP, &sock, 1);
1354 if (result < 0) {
1355 pr_err("Error during creation of socket; terminating\n");
1356 return ERR_PTR(result);
1357 }
1358
1359 /* it is equivalent to the REUSEADDR option in user-space */
1360 sock->sk->sk_reuse = 1;
1361
1362 result = sock->ops->bind(sock, (struct sockaddr *) &mcast_addr,
1363 sizeof(struct sockaddr));
1364 if (result < 0) {
1365 pr_err("Error binding to the multicast addr\n");
1366 goto error;
1367 }
1368
1369 /* join the multicast group */
1370 result = join_mcast_group(sock->sk,
1371 (struct in_addr *) &mcast_addr.sin_addr,
1372 ipvs->backup_mcast_ifn);
1373 if (result < 0) {
1374 pr_err("Error joining to the multicast group\n");
1375 goto error;
1376 }
1377
1378 return sock;
1379
1380 error:
1381 sock_release(sock);
1382 return ERR_PTR(result);
1383 }
1384
1385
1386 static int
1387 ip_vs_send_async(struct socket *sock, const char *buffer, const size_t length)
1388 {
1389 struct msghdr msg = {.msg_flags = MSG_DONTWAIT|MSG_NOSIGNAL};
1390 struct kvec iov;
1391 int len;
1392
1393 EnterFunction(7);
1394 iov.iov_base = (void *)buffer;
1395 iov.iov_len = length;
1396
1397 len = kernel_sendmsg(sock, &msg, &iov, 1, (size_t)(length));
1398
1399 LeaveFunction(7);
1400 return len;
1401 }
1402
1403 static void
1404 ip_vs_send_sync_msg(struct socket *sock, struct ip_vs_sync_mesg *msg)
1405 {
1406 int msize;
1407
1408 msize = msg->size;
1409
1410 /* Put size in network byte order */
1411 msg->size = htons(msg->size);
1412
1413 if (ip_vs_send_async(sock, (char *)msg, msize) != msize)
1414 pr_err("ip_vs_send_async error\n");
1415 }
1416
1417 static int
1418 ip_vs_receive(struct socket *sock, char *buffer, const size_t buflen)
1419 {
1420 struct msghdr msg = {NULL,};
1421 struct kvec iov;
1422 int len;
1423
1424 EnterFunction(7);
1425
1426 /* Receive a packet */
1427 iov.iov_base = buffer;
1428 iov.iov_len = (size_t)buflen;
1429
1430 len = kernel_recvmsg(sock, &msg, &iov, 1, buflen, 0);
1431
1432 if (len < 0)
1433 return -1;
1434
1435 LeaveFunction(7);
1436 return len;
1437 }
1438
1439
1440 static int sync_thread_master(void *data)
1441 {
1442 struct ip_vs_sync_thread_data *tinfo = data;
1443 struct netns_ipvs *ipvs = net_ipvs(tinfo->net);
1444 struct ip_vs_sync_buff *sb;
1445
1446 pr_info("sync thread started: state = MASTER, mcast_ifn = %s, "
1447 "syncid = %d\n",
1448 ipvs->master_mcast_ifn, ipvs->master_syncid);
1449
1450 while (!kthread_should_stop()) {
1451 while ((sb = sb_dequeue(ipvs))) {
1452 ip_vs_send_sync_msg(tinfo->sock, sb->mesg);
1453 ip_vs_sync_buff_release(sb);
1454 }
1455
1456 /* check if entries stay in ipvs->sync_buff for 2 seconds */
1457 sb = get_curr_sync_buff(ipvs, 2 * HZ);
1458 if (sb) {
1459 ip_vs_send_sync_msg(tinfo->sock, sb->mesg);
1460 ip_vs_sync_buff_release(sb);
1461 }
1462
1463 schedule_timeout_interruptible(HZ);
1464 }
1465
1466 /* clean up the sync_buff queue */
1467 while ((sb = sb_dequeue(ipvs)))
1468 ip_vs_sync_buff_release(sb);
1469
1470 /* clean up the current sync_buff */
1471 sb = get_curr_sync_buff(ipvs, 0);
1472 if (sb)
1473 ip_vs_sync_buff_release(sb);
1474
1475 /* release the sending multicast socket */
1476 sock_release(tinfo->sock);
1477 kfree(tinfo);
1478
1479 return 0;
1480 }
1481
1482
1483 static int sync_thread_backup(void *data)
1484 {
1485 struct ip_vs_sync_thread_data *tinfo = data;
1486 struct netns_ipvs *ipvs = net_ipvs(tinfo->net);
1487 int len;
1488
1489 pr_info("sync thread started: state = BACKUP, mcast_ifn = %s, "
1490 "syncid = %d\n",
1491 ipvs->backup_mcast_ifn, ipvs->backup_syncid);
1492
1493 while (!kthread_should_stop()) {
1494 wait_event_interruptible(*sk_sleep(tinfo->sock->sk),
1495 !skb_queue_empty(&tinfo->sock->sk->sk_receive_queue)
1496 || kthread_should_stop());
1497
1498 /* do we have data now? */
1499 while (!skb_queue_empty(&(tinfo->sock->sk->sk_receive_queue))) {
1500 len = ip_vs_receive(tinfo->sock, tinfo->buf,
1501 ipvs->recv_mesg_maxlen);
1502 if (len <= 0) {
1503 pr_err("receiving message error\n");
1504 break;
1505 }
1506
1507 /* disable bottom half, because it accesses the data
1508 shared by softirq while getting/creating conns */
1509 local_bh_disable();
1510 ip_vs_process_message(tinfo->net, tinfo->buf, len);
1511 local_bh_enable();
1512 }
1513 }
1514
1515 /* release the sending multicast socket */
1516 sock_release(tinfo->sock);
1517 kfree(tinfo->buf);
1518 kfree(tinfo);
1519
1520 return 0;
1521 }
1522
1523
1524 int start_sync_thread(struct net *net, int state, char *mcast_ifn, __u8 syncid)
1525 {
1526 struct ip_vs_sync_thread_data *tinfo;
1527 struct task_struct **realtask, *task;
1528 struct socket *sock;
1529 struct netns_ipvs *ipvs = net_ipvs(net);
1530 char *name, *buf = NULL;
1531 int (*threadfn)(void *data);
1532 int result = -ENOMEM;
1533
1534 IP_VS_DBG(7, "%s(): pid %d\n", __func__, task_pid_nr(current));
1535 IP_VS_DBG(7, "Each ip_vs_sync_conn entry needs %Zd bytes\n",
1536 sizeof(struct ip_vs_sync_conn_v0));
1537
1538 if (state == IP_VS_STATE_MASTER) {
1539 if (ipvs->master_thread)
1540 return -EEXIST;
1541
1542 strlcpy(ipvs->master_mcast_ifn, mcast_ifn,
1543 sizeof(ipvs->master_mcast_ifn));
1544 ipvs->master_syncid = syncid;
1545 realtask = &ipvs->master_thread;
1546 name = "ipvs_master:%d";
1547 threadfn = sync_thread_master;
1548 sock = make_send_sock(net);
1549 } else if (state == IP_VS_STATE_BACKUP) {
1550 if (ipvs->backup_thread)
1551 return -EEXIST;
1552
1553 strlcpy(ipvs->backup_mcast_ifn, mcast_ifn,
1554 sizeof(ipvs->backup_mcast_ifn));
1555 ipvs->backup_syncid = syncid;
1556 realtask = &ipvs->backup_thread;
1557 name = "ipvs_backup:%d";
1558 threadfn = sync_thread_backup;
1559 sock = make_receive_sock(net);
1560 } else {
1561 return -EINVAL;
1562 }
1563
1564 if (IS_ERR(sock)) {
1565 result = PTR_ERR(sock);
1566 goto out;
1567 }
1568
1569 set_sync_mesg_maxlen(net, state);
1570 if (state == IP_VS_STATE_BACKUP) {
1571 buf = kmalloc(ipvs->recv_mesg_maxlen, GFP_KERNEL);
1572 if (!buf)
1573 goto outsocket;
1574 }
1575
1576 tinfo = kmalloc(sizeof(*tinfo), GFP_KERNEL);
1577 if (!tinfo)
1578 goto outbuf;
1579
1580 tinfo->net = net;
1581 tinfo->sock = sock;
1582 tinfo->buf = buf;
1583
1584 task = kthread_run(threadfn, tinfo, name, ipvs->gen);
1585 if (IS_ERR(task)) {
1586 result = PTR_ERR(task);
1587 goto outtinfo;
1588 }
1589
1590 /* mark as active */
1591 *realtask = task;
1592 ipvs->sync_state |= state;
1593
1594 /* increase the module use count */
1595 ip_vs_use_count_inc();
1596
1597 return 0;
1598
1599 outtinfo:
1600 kfree(tinfo);
1601 outbuf:
1602 kfree(buf);
1603 outsocket:
1604 sock_release(sock);
1605 out:
1606 return result;
1607 }
1608
1609
1610 int stop_sync_thread(struct net *net, int state)
1611 {
1612 struct netns_ipvs *ipvs = net_ipvs(net);
1613
1614 IP_VS_DBG(7, "%s(): pid %d\n", __func__, task_pid_nr(current));
1615
1616 if (state == IP_VS_STATE_MASTER) {
1617 if (!ipvs->master_thread)
1618 return -ESRCH;
1619
1620 pr_info("stopping master sync thread %d ...\n",
1621 task_pid_nr(ipvs->master_thread));
1622
1623 /*
1624 * The lock synchronizes with sb_queue_tail(), so that we don't
1625 * add sync buffers to the queue, when we are already in
1626 * progress of stopping the master sync daemon.
1627 */
1628
1629 spin_lock_bh(&ipvs->sync_lock);
1630 ipvs->sync_state &= ~IP_VS_STATE_MASTER;
1631 spin_unlock_bh(&ipvs->sync_lock);
1632 kthread_stop(ipvs->master_thread);
1633 ipvs->master_thread = NULL;
1634 } else if (state == IP_VS_STATE_BACKUP) {
1635 if (!ipvs->backup_thread)
1636 return -ESRCH;
1637
1638 pr_info("stopping backup sync thread %d ...\n",
1639 task_pid_nr(ipvs->backup_thread));
1640
1641 ipvs->sync_state &= ~IP_VS_STATE_BACKUP;
1642 kthread_stop(ipvs->backup_thread);
1643 ipvs->backup_thread = NULL;
1644 } else {
1645 return -EINVAL;
1646 }
1647
1648 /* decrease the module use count */
1649 ip_vs_use_count_dec();
1650
1651 return 0;
1652 }
1653
1654 /*
1655 * Initialize data struct for each netns
1656 */
1657 static int __net_init __ip_vs_sync_init(struct net *net)
1658 {
1659 struct netns_ipvs *ipvs = net_ipvs(net);
1660
1661 INIT_LIST_HEAD(&ipvs->sync_queue);
1662 spin_lock_init(&ipvs->sync_lock);
1663 spin_lock_init(&ipvs->sync_buff_lock);
1664
1665 ipvs->sync_mcast_addr.sin_family = AF_INET;
1666 ipvs->sync_mcast_addr.sin_port = cpu_to_be16(IP_VS_SYNC_PORT);
1667 ipvs->sync_mcast_addr.sin_addr.s_addr = cpu_to_be32(IP_VS_SYNC_GROUP);
1668 return 0;
1669 }
1670
1671 static void __ip_vs_sync_cleanup(struct net *net)
1672 {
1673 stop_sync_thread(net, IP_VS_STATE_MASTER);
1674 stop_sync_thread(net, IP_VS_STATE_BACKUP);
1675 }
1676
1677 static struct pernet_operations ipvs_sync_ops = {
1678 .init = __ip_vs_sync_init,
1679 .exit = __ip_vs_sync_cleanup,
1680 };
1681
1682
1683 int __init ip_vs_sync_init(void)
1684 {
1685 return register_pernet_subsys(&ipvs_sync_ops);
1686 }
1687
1688 void ip_vs_sync_cleanup(void)
1689 {
1690 unregister_pernet_subsys(&ipvs_sync_ops);
1691 }
Ubuntu Jammy Linux kernel mirror