blob: 688d64cf9f9f3a8f6c6d3cc2af140cabe2857993 [file] [log] [blame]
Emeric Brun2b920a12010-09-23 18:30:22 +02001/*
Willy Tarreaubd55e312010-11-11 10:55:09 +01002 * Stick table synchro management.
Emeric Brun2b920a12010-09-23 18:30:22 +02003 *
4 * Copyright 2010 EXCELIANCE, Emeric Brun <ebrun@exceliance.fr>
5 *
6 * This program is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU General Public License
8 * as published by the Free Software Foundation; either version
9 * 2 of the License, or (at your option) any later version.
10 *
11 */
12
13#include <errno.h>
14#include <fcntl.h>
15#include <stdio.h>
16#include <stdlib.h>
17#include <string.h>
18
19#include <sys/socket.h>
20#include <sys/stat.h>
21#include <sys/types.h>
22
23#include <common/compat.h>
24#include <common/config.h>
25#include <common/time.h>
26
27#include <types/global.h>
Willy Tarreau3fdb3662012-11-12 00:42:33 +010028#include <types/listener.h>
29#include <types/obj_type.h>
Emeric Brun2b920a12010-09-23 18:30:22 +020030#include <types/peers.h>
31
32#include <proto/acl.h>
Willy Tarreauc7e42382012-08-24 19:22:53 +020033#include <proto/channel.h>
Emeric Brun2b920a12010-09-23 18:30:22 +020034#include <proto/fd.h>
35#include <proto/log.h>
36#include <proto/hdr_idx.h>
Emeric Brun2b920a12010-09-23 18:30:22 +020037#include <proto/proto_tcp.h>
38#include <proto/proto_http.h>
39#include <proto/proxy.h>
40#include <proto/session.h>
41#include <proto/stream_interface.h>
Emeric Brun2b920a12010-09-23 18:30:22 +020042#include <proto/task.h>
43#include <proto/stick_table.h>
44#include <proto/signal.h>
45
46
47/*******************************/
48/* Current peer learning state */
49/*******************************/
50
51/******************************/
52/* Current table resync state */
53/******************************/
54#define SHTABLE_F_RESYNC_LOCAL 0x00000001 /* Learn from local finished or no more needed */
55#define SHTABLE_F_RESYNC_REMOTE 0x00000002 /* Learn from remote finished or no more needed */
56#define SHTABLE_F_RESYNC_ASSIGN 0x00000004 /* A peer was assigned to learn our lesson */
57#define SHTABLE_F_RESYNC_PROCESS 0x00000008 /* The assigned peer was requested for resync */
58#define SHTABLE_F_DONOTSTOP 0x00010000 /* Main table sync task block process during soft stop
59 to push data to new process */
60
61#define SHTABLE_RESYNC_STATEMASK (SHTABLE_F_RESYNC_LOCAL|SHTABLE_F_RESYNC_REMOTE)
62#define SHTABLE_RESYNC_FROMLOCAL 0x00000000
63#define SHTABLE_RESYNC_FROMREMOTE SHTABLE_F_RESYNC_LOCAL
64#define SHTABLE_RESYNC_FINISHED (SHTABLE_F_RESYNC_LOCAL|SHTABLE_F_RESYNC_REMOTE)
65
66/******************************/
67/* Remote peer teaching state */
68/******************************/
69#define PEER_F_TEACH_PROCESS 0x00000001 /* Teach a lesson to current peer */
70#define PEER_F_TEACH_STAGE1 0x00000002 /* Teach state 1 complete */
71#define PEER_F_TEACH_STAGE2 0x00000004 /* Teach stage 2 complete */
72#define PEER_F_TEACH_FINISHED 0x00000008 /* Teach conclude, (wait for confirm) */
73#define PEER_F_TEACH_COMPLETE 0x00000010 /* All that we know already taught to current peer, used only for a local peer */
74#define PEER_F_LEARN_ASSIGN 0x00000100 /* Current peer was assigned for a lesson */
75#define PEER_F_LEARN_NOTUP2DATE 0x00000200 /* Learn from peer finished but peer is not up to date */
76
77#define PEER_TEACH_RESET ~(PEER_F_TEACH_PROCESS|PEER_F_TEACH_STAGE1|PEER_F_TEACH_STAGE2|PEER_F_TEACH_FINISHED) /* PEER_F_TEACH_COMPLETE should never be reset */
78#define PEER_LEARN_RESET ~(PEER_F_LEARN_ASSIGN|PEER_F_LEARN_NOTUP2DATE)
79
80
81/**********************************/
82/* Peer Session IO handler states */
83/**********************************/
84
85#define PEER_SESSION_ACCEPT 1000 /* Initial state for session create by an accept */
86#define PEER_SESSION_GETVERSION 1001 /* Validate supported protocol version*/
87#define PEER_SESSION_GETHOST 1002 /* Validate host ID correspond to local host id */
88#define PEER_SESSION_GETPEER 1003 /* Validate peer ID correspond to a known remote peer id */
89#define PEER_SESSION_GETTABLE 1004 /* Search into registered table for a table with same id and
90 validate type and size */
91#define PEER_SESSION_SENDSUCCESS 1005 /* Send ret code 200 (success) and wait for message */
92/* next state is WAITMSG */
93
94#define PEER_SESSION_CONNECT 2000 /* Initial state for session create on a connect,
95 push presentation into buffer */
96#define PEER_SESSION_GETSTATUS 2001 /* Wait for the welcome message */
97#define PEER_SESSION_WAITMSG 2002 /* Wait for datamessages*/
98/* loop on WAITMSG */
99
100#define PEER_SESSION_EXIT 10000 /* Exit with status code */
101#define PEER_SESSION_END 10001 /* Killed session */
102/* session ended */
103
104
105/**********************************/
106/* Peer Session status code */
107/**********************************/
108
109#define PEER_SESSION_CONNECTCODE 100 /* connect in progress */
110#define PEER_SESSION_CONNECTEDCODE 110 /* tcp connect success */
111
112#define PEER_SESSION_SUCCESSCODE 200 /* accept or connect successful */
113
114#define PEER_SESSION_TRYAGAIN 300 /* try again later */
115
116#define PEER_SESSION_ERRPROTO 501 /* error protocol */
117#define PEER_SESSION_ERRVERSION 502 /* unknown protocol version */
118#define PEER_SESSION_ERRHOST 503 /* bad host name */
119#define PEER_SESSION_ERRPEER 504 /* unknown peer */
120#define PEER_SESSION_ERRTYPE 505 /* table key type mismatch */
121#define PEER_SESSION_ERRSIZE 506 /* table key size mismatch */
122#define PEER_SESSION_ERRTABLE 507 /* unknown table */
123
124#define PEER_SESSION_PROTO_NAME "HAProxyS"
125
126struct peers *peers = NULL;
Simon Horman96553772011-06-08 09:18:51 +0900127static void peer_session_forceshutdown(struct session * session);
Emeric Brun2b920a12010-09-23 18:30:22 +0200128
129
130/*
131 * This prepare the data update message of the stick session <ts>, <ps> is the the peer session
132 * where the data going to be pushed, <msg> is a buffer of <size> to recieve data message content
133 */
Simon Horman96553772011-06-08 09:18:51 +0900134static int peer_prepare_datamsg(struct stksess *ts, struct peer_session *ps, char *msg, size_t size)
Emeric Brun2b920a12010-09-23 18:30:22 +0200135{
136 uint32_t netinteger;
137 int len;
138 /* construct message */
139 if (ps->lastpush && ts->upd.key > ps->lastpush && (ts->upd.key - ps->lastpush) <= 127) {
140 msg[0] = 0x80 + ts->upd.key - ps->lastpush;
141 len = sizeof(char);
142 }
143 else {
144 msg[0] = 'D';
145 netinteger = htonl(ts->upd.key);
146 memcpy(&msg[sizeof(char)], &netinteger, sizeof(netinteger));
147 len = sizeof(char) + sizeof(netinteger);
148 }
149
150 if (ps->table->table->type == STKTABLE_TYPE_STRING) {
151 int stlen = strlen((char *)ts->key.key);
152
153 netinteger = htonl(strlen((char *)ts->key.key));
154 memcpy(&msg[len], &netinteger, sizeof(netinteger));
155 memcpy(&msg[len+sizeof(netinteger)], ts->key.key, stlen);
156 len += sizeof(netinteger) + stlen;
157
158 }
159 else if (ps->table->table->type == STKTABLE_TYPE_INTEGER) {
160 netinteger = htonl(*((uint32_t *)ts->key.key));
161 memcpy(&msg[len], &netinteger, sizeof(netinteger));
162 len += sizeof(netinteger);
163 }
164 else {
165 memcpy(&msg[len], ts->key.key, ps->table->table->key_size);
166 len += ps->table->table->key_size;
167 }
168
169 if (stktable_data_ptr(ps->table->table, ts, STKTABLE_DT_SERVER_ID))
170 netinteger = htonl(stktable_data_cast(stktable_data_ptr(ps->table->table, ts, STKTABLE_DT_SERVER_ID), server_id));
171 else
172 netinteger = 0;
173
174 memcpy(&msg[len], &netinteger , sizeof(netinteger));
175 len += sizeof(netinteger);
176
177 return len;
178}
179
180
181/*
182 * Callback to release a session with a peer
183 */
Simon Horman96553772011-06-08 09:18:51 +0900184static void peer_session_release(struct stream_interface *si)
Emeric Brun2b920a12010-09-23 18:30:22 +0200185{
Aman Guptad94991d2012-04-06 17:39:26 -0700186 struct task *t = (struct task *)si->owner;
Emeric Brun2b920a12010-09-23 18:30:22 +0200187 struct session *s = (struct session *)t->context;
Willy Tarreauf2943dc2012-10-26 20:10:28 +0200188 struct peer_session *ps = (struct peer_session *)si->conn->xprt_ctx;
Emeric Brun2b920a12010-09-23 18:30:22 +0200189
Willy Tarreauf2943dc2012-10-26 20:10:28 +0200190 /* si->conn->xprt_ctx is not a peer session */
Willy Tarreaubc4af052011-02-13 13:25:14 +0100191 if (si->applet.st0 < PEER_SESSION_SENDSUCCESS)
Emeric Brun2b920a12010-09-23 18:30:22 +0200192 return;
193
194 /* peer session identified */
195 if (ps) {
196 if (ps->session == s) {
197 ps->session = NULL;
198 if (ps->flags & PEER_F_LEARN_ASSIGN) {
199 /* unassign current peer for learning */
200 ps->flags &= ~(PEER_F_LEARN_ASSIGN);
201 ps->table->flags &= ~(SHTABLE_F_RESYNC_ASSIGN|SHTABLE_F_RESYNC_PROCESS);
202
203 /* reschedule a resync */
204 ps->table->resync_timeout = tick_add(now_ms, MS_TO_TICKS(5000));
205 }
206 /* reset teaching and learning flags to 0 */
207 ps->flags &= PEER_TEACH_RESET;
208 ps->flags &= PEER_LEARN_RESET;
209 }
210 task_wakeup(ps->table->sync_task, TASK_WOKEN_MSG);
211 }
212}
213
214
215/*
216 * IO Handler to handle message exchance with a peer
217 */
Willy Tarreaub24281b2011-02-13 13:16:36 +0100218static void peer_io_handler(struct stream_interface *si)
Emeric Brun2b920a12010-09-23 18:30:22 +0200219{
220 struct task *t= (struct task *)si->owner;
221 struct session *s = (struct session *)t->context;
222 struct peers *curpeers = (struct peers *)s->fe->parent;
223 int reql = 0;
224 int repl = 0;
225
226 while (1) {
227switchstate:
Willy Tarreaubc4af052011-02-13 13:25:14 +0100228 switch(si->applet.st0) {
Emeric Brun2b920a12010-09-23 18:30:22 +0200229 case PEER_SESSION_ACCEPT:
Willy Tarreauf2943dc2012-10-26 20:10:28 +0200230 si->conn->xprt_ctx = NULL;
Willy Tarreaubc4af052011-02-13 13:25:14 +0100231 si->applet.st0 = PEER_SESSION_GETVERSION;
Emeric Brun2b920a12010-09-23 18:30:22 +0200232 /* fall through */
233 case PEER_SESSION_GETVERSION:
Willy Tarreau19d14ef2012-10-29 16:51:55 +0100234 reql = bo_getline(si->ob, trash.str, trash.size);
Emeric Brun2b920a12010-09-23 18:30:22 +0200235 if (reql <= 0) { /* closed or EOL not found */
236 if (reql == 0)
237 goto out;
Willy Tarreaubc4af052011-02-13 13:25:14 +0100238 si->applet.st0 = PEER_SESSION_END;
Emeric Brun2b920a12010-09-23 18:30:22 +0200239 goto switchstate;
240 }
Willy Tarreau19d14ef2012-10-29 16:51:55 +0100241 if (trash.str[reql-1] != '\n') {
Willy Tarreaubc4af052011-02-13 13:25:14 +0100242 si->applet.st0 = PEER_SESSION_END;
Emeric Brun2b920a12010-09-23 18:30:22 +0200243 goto switchstate;
244 }
Willy Tarreau19d14ef2012-10-29 16:51:55 +0100245 else if (reql > 1 && (trash.str[reql-2] == '\r'))
246 trash.str[reql-2] = 0;
Emeric Brun2b920a12010-09-23 18:30:22 +0200247 else
Willy Tarreau19d14ef2012-10-29 16:51:55 +0100248 trash.str[reql-1] = 0;
Emeric Brun2b920a12010-09-23 18:30:22 +0200249
Willy Tarreau9dab5fc2012-05-07 11:56:55 +0200250 bo_skip(si->ob, reql);
Emeric Brun2b920a12010-09-23 18:30:22 +0200251
252 /* test version */
Willy Tarreau19d14ef2012-10-29 16:51:55 +0100253 if (strcmp(PEER_SESSION_PROTO_NAME " 1.0", trash.str) != 0) {
Willy Tarreaubc4af052011-02-13 13:25:14 +0100254 si->applet.st0 = PEER_SESSION_EXIT;
255 si->applet.st1 = PEER_SESSION_ERRVERSION;
Emeric Brun2b920a12010-09-23 18:30:22 +0200256 /* test protocol */
Willy Tarreau19d14ef2012-10-29 16:51:55 +0100257 if (strncmp(PEER_SESSION_PROTO_NAME " ", trash.str, strlen(PEER_SESSION_PROTO_NAME)+1) != 0)
Willy Tarreaubc4af052011-02-13 13:25:14 +0100258 si->applet.st1 = PEER_SESSION_ERRPROTO;
Emeric Brun2b920a12010-09-23 18:30:22 +0200259 goto switchstate;
260 }
261
Willy Tarreaubc4af052011-02-13 13:25:14 +0100262 si->applet.st0 = PEER_SESSION_GETHOST;
Emeric Brun2b920a12010-09-23 18:30:22 +0200263 /* fall through */
264 case PEER_SESSION_GETHOST:
Willy Tarreau19d14ef2012-10-29 16:51:55 +0100265 reql = bo_getline(si->ob, trash.str, trash.size);
Emeric Brun2b920a12010-09-23 18:30:22 +0200266 if (reql <= 0) { /* closed or EOL not found */
267 if (reql == 0)
268 goto out;
Willy Tarreaubc4af052011-02-13 13:25:14 +0100269 si->applet.st0 = PEER_SESSION_END;
Emeric Brun2b920a12010-09-23 18:30:22 +0200270 goto switchstate;
271 }
Willy Tarreau19d14ef2012-10-29 16:51:55 +0100272 if (trash.str[reql-1] != '\n') {
Willy Tarreaubc4af052011-02-13 13:25:14 +0100273 si->applet.st0 = PEER_SESSION_END;
Emeric Brun2b920a12010-09-23 18:30:22 +0200274 goto switchstate;
275 }
Willy Tarreau19d14ef2012-10-29 16:51:55 +0100276 else if (reql > 1 && (trash.str[reql-2] == '\r'))
277 trash.str[reql-2] = 0;
Emeric Brun2b920a12010-09-23 18:30:22 +0200278 else
Willy Tarreau19d14ef2012-10-29 16:51:55 +0100279 trash.str[reql-1] = 0;
Emeric Brun2b920a12010-09-23 18:30:22 +0200280
Willy Tarreau9dab5fc2012-05-07 11:56:55 +0200281 bo_skip(si->ob, reql);
Emeric Brun2b920a12010-09-23 18:30:22 +0200282
283 /* test hostname match */
Willy Tarreau19d14ef2012-10-29 16:51:55 +0100284 if (strcmp(localpeer, trash.str) != 0) {
Willy Tarreaubc4af052011-02-13 13:25:14 +0100285 si->applet.st0 = PEER_SESSION_EXIT;
286 si->applet.st1 = PEER_SESSION_ERRHOST;
Emeric Brun2b920a12010-09-23 18:30:22 +0200287 goto switchstate;
288 }
289
Willy Tarreaubc4af052011-02-13 13:25:14 +0100290 si->applet.st0 = PEER_SESSION_GETPEER;
Emeric Brun2b920a12010-09-23 18:30:22 +0200291 /* fall through */
292 case PEER_SESSION_GETPEER: {
293 struct peer *curpeer;
294 char *p;
Willy Tarreau19d14ef2012-10-29 16:51:55 +0100295 reql = bo_getline(si->ob, trash.str, trash.size);
Emeric Brun2b920a12010-09-23 18:30:22 +0200296 if (reql <= 0) { /* closed or EOL not found */
297 if (reql == 0)
298 goto out;
Willy Tarreaubc4af052011-02-13 13:25:14 +0100299 si->applet.st0 = PEER_SESSION_END;
Emeric Brun2b920a12010-09-23 18:30:22 +0200300 goto switchstate;
301 }
Willy Tarreau19d14ef2012-10-29 16:51:55 +0100302 if (trash.str[reql-1] != '\n') {
Emeric Brun2b920a12010-09-23 18:30:22 +0200303 /* Incomplete line, we quit */
Willy Tarreaubc4af052011-02-13 13:25:14 +0100304 si->applet.st0 = PEER_SESSION_END;
Emeric Brun2b920a12010-09-23 18:30:22 +0200305 goto switchstate;
306 }
Willy Tarreau19d14ef2012-10-29 16:51:55 +0100307 else if (reql > 1 && (trash.str[reql-2] == '\r'))
308 trash.str[reql-2] = 0;
Emeric Brun2b920a12010-09-23 18:30:22 +0200309 else
Willy Tarreau19d14ef2012-10-29 16:51:55 +0100310 trash.str[reql-1] = 0;
Emeric Brun2b920a12010-09-23 18:30:22 +0200311
Willy Tarreau9dab5fc2012-05-07 11:56:55 +0200312 bo_skip(si->ob, reql);
Emeric Brun2b920a12010-09-23 18:30:22 +0200313
314 /* parse line "<peer name> <pid>" */
Willy Tarreau19d14ef2012-10-29 16:51:55 +0100315 p = strchr(trash.str, ' ');
Emeric Brun2b920a12010-09-23 18:30:22 +0200316 if (!p) {
Willy Tarreaubc4af052011-02-13 13:25:14 +0100317 si->applet.st0 = PEER_SESSION_EXIT;
318 si->applet.st1 = PEER_SESSION_ERRPROTO;
Emeric Brun2b920a12010-09-23 18:30:22 +0200319 goto switchstate;
320 }
321 *p = 0;
322
323 /* lookup known peer */
324 for (curpeer = curpeers->remote; curpeer; curpeer = curpeer->next) {
Willy Tarreau19d14ef2012-10-29 16:51:55 +0100325 if (strcmp(curpeer->id, trash.str) == 0)
Emeric Brun2b920a12010-09-23 18:30:22 +0200326 break;
327 }
328
329 /* if unknown peer */
330 if (!curpeer) {
Willy Tarreaubc4af052011-02-13 13:25:14 +0100331 si->applet.st0 = PEER_SESSION_EXIT;
332 si->applet.st1 = PEER_SESSION_ERRPEER;
Emeric Brun2b920a12010-09-23 18:30:22 +0200333 goto switchstate;
334 }
335
Willy Tarreauf2943dc2012-10-26 20:10:28 +0200336 si->conn->xprt_ctx = curpeer;
Willy Tarreaubc4af052011-02-13 13:25:14 +0100337 si->applet.st0 = PEER_SESSION_GETTABLE;
Emeric Brun2b920a12010-09-23 18:30:22 +0200338 /* fall through */
339 }
340 case PEER_SESSION_GETTABLE: {
Willy Tarreauf2943dc2012-10-26 20:10:28 +0200341 struct peer *curpeer = (struct peer *)si->conn->xprt_ctx;
Emeric Brun2b920a12010-09-23 18:30:22 +0200342 struct shared_table *st;
343 struct peer_session *ps = NULL;
344 unsigned long key_type;
345 size_t key_size;
346 char *p;
347
Willy Tarreau19d14ef2012-10-29 16:51:55 +0100348 reql = bo_getline(si->ob, trash.str, trash.size);
Emeric Brun2b920a12010-09-23 18:30:22 +0200349 if (reql <= 0) { /* closed or EOL not found */
350 if (reql == 0)
351 goto out;
Willy Tarreauf2943dc2012-10-26 20:10:28 +0200352 si->conn->xprt_ctx = NULL;
Willy Tarreaubc4af052011-02-13 13:25:14 +0100353 si->applet.st0 = PEER_SESSION_END;
Emeric Brun2b920a12010-09-23 18:30:22 +0200354 goto switchstate;
355 }
Willy Tarreauf2943dc2012-10-26 20:10:28 +0200356 /* Re init si->conn->xprt_ctx to null, to handle correctly a release case */
357 si->conn->xprt_ctx = NULL;
Emeric Brun2b920a12010-09-23 18:30:22 +0200358
Willy Tarreau19d14ef2012-10-29 16:51:55 +0100359 if (trash.str[reql-1] != '\n') {
Emeric Brun2b920a12010-09-23 18:30:22 +0200360 /* Incomplete line, we quit */
Willy Tarreaubc4af052011-02-13 13:25:14 +0100361 si->applet.st0 = PEER_SESSION_END;
Emeric Brun2b920a12010-09-23 18:30:22 +0200362 goto switchstate;
363 }
Willy Tarreau19d14ef2012-10-29 16:51:55 +0100364 else if (reql > 1 && (trash.str[reql-2] == '\r'))
365 trash.str[reql-2] = 0;
Emeric Brun2b920a12010-09-23 18:30:22 +0200366 else
Willy Tarreau19d14ef2012-10-29 16:51:55 +0100367 trash.str[reql-1] = 0;
Emeric Brun2b920a12010-09-23 18:30:22 +0200368
Willy Tarreau9dab5fc2012-05-07 11:56:55 +0200369 bo_skip(si->ob, reql);
Emeric Brun2b920a12010-09-23 18:30:22 +0200370
371 /* Parse line "<table name> <type> <size>" */
Willy Tarreau19d14ef2012-10-29 16:51:55 +0100372 p = strchr(trash.str, ' ');
Emeric Brun2b920a12010-09-23 18:30:22 +0200373 if (!p) {
Willy Tarreaubc4af052011-02-13 13:25:14 +0100374 si->applet.st0 = PEER_SESSION_EXIT;
375 si->applet.st1 = PEER_SESSION_ERRPROTO;
Emeric Brun2b920a12010-09-23 18:30:22 +0200376 goto switchstate;
377 }
378 *p = 0;
379 key_type = (unsigned long)atol(p+1);
380
381 p = strchr(p+1, ' ');
382 if (!p) {
Willy Tarreauf2943dc2012-10-26 20:10:28 +0200383 si->conn->xprt_ctx = NULL;
Willy Tarreaubc4af052011-02-13 13:25:14 +0100384 si->applet.st0 = PEER_SESSION_EXIT;
385 si->applet.st1 = PEER_SESSION_ERRPROTO;
Emeric Brun2b920a12010-09-23 18:30:22 +0200386 goto switchstate;
387 }
388
389 key_size = (size_t)atoi(p);
390 for (st = curpeers->tables; st; st = st->next) {
391 /* If table name matches */
Willy Tarreau19d14ef2012-10-29 16:51:55 +0100392 if (strcmp(st->table->id, trash.str) == 0) {
Emeric Brun2b920a12010-09-23 18:30:22 +0200393 /* If key size mismatches */
394 if (key_size != st->table->key_size) {
Willy Tarreaubc4af052011-02-13 13:25:14 +0100395 si->applet.st0 = PEER_SESSION_EXIT;
396 si->applet.st1 = PEER_SESSION_ERRSIZE;
Emeric Brun2b920a12010-09-23 18:30:22 +0200397 goto switchstate;
398 }
399
400 /* If key type mismatches */
401 if (key_type != st->table->type) {
Willy Tarreaubc4af052011-02-13 13:25:14 +0100402 si->applet.st0 = PEER_SESSION_EXIT;
403 si->applet.st1 = PEER_SESSION_ERRTYPE;
Emeric Brun2b920a12010-09-23 18:30:22 +0200404 goto switchstate;
405 }
406
407 /* lookup peer session of current peer */
408 for (ps = st->sessions; ps; ps = ps->next) {
409 if (ps->peer == curpeer) {
410 /* If session already active, replaced by new one */
411 if (ps->session && ps->session != s) {
412 if (ps->peer->local) {
413 /* Local connection, reply a retry */
Willy Tarreaubc4af052011-02-13 13:25:14 +0100414 si->applet.st0 = PEER_SESSION_EXIT;
415 si->applet.st1 = PEER_SESSION_TRYAGAIN;
Emeric Brun2b920a12010-09-23 18:30:22 +0200416 goto switchstate;
417 }
418 peer_session_forceshutdown(ps->session);
419 }
420 ps->session = s;
421 break;
422 }
423 }
424 break;
425 }
426 }
427
428 /* If table not found */
429 if (!st){
Willy Tarreaubc4af052011-02-13 13:25:14 +0100430 si->applet.st0 = PEER_SESSION_EXIT;
431 si->applet.st1 = PEER_SESSION_ERRTABLE;
Emeric Brun2b920a12010-09-23 18:30:22 +0200432 goto switchstate;
433 }
434
435 /* If no peer session for current peer */
436 if (!ps) {
Willy Tarreaubc4af052011-02-13 13:25:14 +0100437 si->applet.st0 = PEER_SESSION_EXIT;
438 si->applet.st1 = PEER_SESSION_ERRPEER;
Emeric Brun2b920a12010-09-23 18:30:22 +0200439 goto switchstate;
440 }
441
Willy Tarreauf2943dc2012-10-26 20:10:28 +0200442 si->conn->xprt_ctx = ps;
Willy Tarreaubc4af052011-02-13 13:25:14 +0100443 si->applet.st0 = PEER_SESSION_SENDSUCCESS;
Emeric Brun2b920a12010-09-23 18:30:22 +0200444 /* fall through */
445 }
446 case PEER_SESSION_SENDSUCCESS:{
Willy Tarreauf2943dc2012-10-26 20:10:28 +0200447 struct peer_session *ps = (struct peer_session *)si->conn->xprt_ctx;
Emeric Brun2b920a12010-09-23 18:30:22 +0200448
Willy Tarreau19d14ef2012-10-29 16:51:55 +0100449 repl = snprintf(trash.str, trash.size, "%d\n", PEER_SESSION_SUCCESSCODE);
450 repl = bi_putblk(si->ib, trash.str, repl);
Emeric Brun2b920a12010-09-23 18:30:22 +0200451 if (repl <= 0) {
452 if (repl == -1)
453 goto out;
Willy Tarreaubc4af052011-02-13 13:25:14 +0100454 si->applet.st0 = PEER_SESSION_END;
Emeric Brun2b920a12010-09-23 18:30:22 +0200455 goto switchstate;
456 }
457
458 /* Register status code */
459 ps->statuscode = PEER_SESSION_SUCCESSCODE;
460
461 /* Awake main task */
462 task_wakeup(ps->table->sync_task, TASK_WOKEN_MSG);
463
464 /* Init cursors */
465 ps->teaching_origin =ps->lastpush = ps->lastack = ps->pushack = 0;
466 ps->pushed = ps->update;
467
468 /* Init confirm counter */
469 ps->confirm = 0;
470
471 /* reset teaching and learning flags to 0 */
472 ps->flags &= PEER_TEACH_RESET;
473 ps->flags &= PEER_LEARN_RESET;
474
475 /* if current peer is local */
476 if (ps->peer->local) {
477 /* if table need resyncfrom local and no process assined */
478 if ((ps->table->flags & SHTABLE_RESYNC_STATEMASK) == SHTABLE_RESYNC_FROMLOCAL &&
479 !(ps->table->flags & SHTABLE_F_RESYNC_ASSIGN)) {
480 /* assign local peer for a lesson, consider lesson already requested */
481 ps->flags |= PEER_F_LEARN_ASSIGN;
482 ps->table->flags |= (SHTABLE_F_RESYNC_ASSIGN|SHTABLE_F_RESYNC_PROCESS);
483 }
484
485 }
486 else if ((ps->table->flags & SHTABLE_RESYNC_STATEMASK) == SHTABLE_RESYNC_FROMREMOTE &&
487 !(ps->table->flags & SHTABLE_F_RESYNC_ASSIGN)) {
488 /* assign peer for a lesson */
489 ps->flags |= PEER_F_LEARN_ASSIGN;
490 ps->table->flags |= SHTABLE_F_RESYNC_ASSIGN;
491 }
492 /* switch to waiting message state */
Willy Tarreaubc4af052011-02-13 13:25:14 +0100493 si->applet.st0 = PEER_SESSION_WAITMSG;
Emeric Brun2b920a12010-09-23 18:30:22 +0200494 goto switchstate;
495 }
496 case PEER_SESSION_CONNECT: {
Willy Tarreauf2943dc2012-10-26 20:10:28 +0200497 struct peer_session *ps = (struct peer_session *)si->conn->xprt_ctx;
Emeric Brun2b920a12010-09-23 18:30:22 +0200498
499 /* Send headers */
Willy Tarreau19d14ef2012-10-29 16:51:55 +0100500 repl = snprintf(trash.str, trash.size,
Emeric Brun2b920a12010-09-23 18:30:22 +0200501 PEER_SESSION_PROTO_NAME " 1.0\n%s\n%s %d\n%s %lu %d\n",
502 ps->peer->id,
503 localpeer,
Willy Tarreau7b77c9f2012-01-07 22:52:12 +0100504 (int)getpid(),
Emeric Brun2b920a12010-09-23 18:30:22 +0200505 ps->table->table->id,
506 ps->table->table->type,
Willy Tarreaubd55e312010-11-11 10:55:09 +0100507 (int)ps->table->table->key_size);
Emeric Brun2b920a12010-09-23 18:30:22 +0200508
Willy Tarreau19d14ef2012-10-29 16:51:55 +0100509 if (repl >= trash.size) {
Willy Tarreaubc4af052011-02-13 13:25:14 +0100510 si->applet.st0 = PEER_SESSION_END;
Emeric Brun2b920a12010-09-23 18:30:22 +0200511 goto switchstate;
512 }
513
Willy Tarreau19d14ef2012-10-29 16:51:55 +0100514 repl = bi_putblk(si->ib, trash.str, repl);
Emeric Brun2b920a12010-09-23 18:30:22 +0200515 if (repl <= 0) {
516 if (repl == -1)
517 goto out;
Willy Tarreaubc4af052011-02-13 13:25:14 +0100518 si->applet.st0 = PEER_SESSION_END;
Emeric Brun2b920a12010-09-23 18:30:22 +0200519 goto switchstate;
520 }
521
522 /* switch to the waiting statuscode state */
Willy Tarreaubc4af052011-02-13 13:25:14 +0100523 si->applet.st0 = PEER_SESSION_GETSTATUS;
Emeric Brun2b920a12010-09-23 18:30:22 +0200524 /* fall through */
525 }
526 case PEER_SESSION_GETSTATUS: {
Willy Tarreauf2943dc2012-10-26 20:10:28 +0200527 struct peer_session *ps = (struct peer_session *)si->conn->xprt_ctx;
Emeric Brun2b920a12010-09-23 18:30:22 +0200528
Willy Tarreau03cdb7c2012-08-27 23:14:58 +0200529 if (si->ib->flags & CF_WRITE_PARTIAL)
Emeric Brun2b920a12010-09-23 18:30:22 +0200530 ps->statuscode = PEER_SESSION_CONNECTEDCODE;
531
Willy Tarreau19d14ef2012-10-29 16:51:55 +0100532 reql = bo_getline(si->ob, trash.str, trash.size);
Emeric Brun2b920a12010-09-23 18:30:22 +0200533 if (reql <= 0) { /* closed or EOL not found */
534 if (reql == 0)
535 goto out;
Willy Tarreaubc4af052011-02-13 13:25:14 +0100536 si->applet.st0 = PEER_SESSION_END;
Emeric Brun2b920a12010-09-23 18:30:22 +0200537 goto switchstate;
538 }
Willy Tarreau19d14ef2012-10-29 16:51:55 +0100539 if (trash.str[reql-1] != '\n') {
Emeric Brun2b920a12010-09-23 18:30:22 +0200540 /* Incomplete line, we quit */
Willy Tarreaubc4af052011-02-13 13:25:14 +0100541 si->applet.st0 = PEER_SESSION_END;
Emeric Brun2b920a12010-09-23 18:30:22 +0200542 goto switchstate;
543 }
Willy Tarreau19d14ef2012-10-29 16:51:55 +0100544 else if (reql > 1 && (trash.str[reql-2] == '\r'))
545 trash.str[reql-2] = 0;
Emeric Brun2b920a12010-09-23 18:30:22 +0200546 else
Willy Tarreau19d14ef2012-10-29 16:51:55 +0100547 trash.str[reql-1] = 0;
Emeric Brun2b920a12010-09-23 18:30:22 +0200548
Willy Tarreau9dab5fc2012-05-07 11:56:55 +0200549 bo_skip(si->ob, reql);
Emeric Brun2b920a12010-09-23 18:30:22 +0200550
551 /* Register status code */
Willy Tarreau19d14ef2012-10-29 16:51:55 +0100552 ps->statuscode = atoi(trash.str);
Emeric Brun2b920a12010-09-23 18:30:22 +0200553
554 /* Awake main task */
555 task_wakeup(ps->table->sync_task, TASK_WOKEN_MSG);
556
557 /* If status code is success */
558 if (ps->statuscode == PEER_SESSION_SUCCESSCODE) {
559 /* Init cursors */
560 ps->teaching_origin = ps->lastpush = ps->lastack = ps->pushack = 0;
561 ps->pushed = ps->update;
562
563 /* Init confirm counter */
564 ps->confirm = 0;
565
566 /* reset teaching and learning flags to 0 */
567 ps->flags &= PEER_TEACH_RESET;
568 ps->flags &= PEER_LEARN_RESET;
569
570 /* If current peer is local */
571 if (ps->peer->local) {
572 /* Init cursors to push a resync */
573 ps->teaching_origin = ps->pushed = ps->table->table->update;
574 /* flag to start to teach lesson */
575 ps->flags |= PEER_F_TEACH_PROCESS;
576
577 }
578 else if ((ps->table->flags & SHTABLE_RESYNC_STATEMASK) == SHTABLE_RESYNC_FROMREMOTE &&
579 !(ps->table->flags & SHTABLE_F_RESYNC_ASSIGN)) {
580 /* If peer is remote and resync from remote is needed,
581 and no peer currently assigned */
582
583 /* assign peer for a lesson */
584 ps->flags |= PEER_F_LEARN_ASSIGN;
585 ps->table->flags |= SHTABLE_F_RESYNC_ASSIGN;
586 }
587
588 }
589 else {
590 /* Status code is not success, abort */
Willy Tarreaubc4af052011-02-13 13:25:14 +0100591 si->applet.st0 = PEER_SESSION_END;
Emeric Brun2b920a12010-09-23 18:30:22 +0200592 goto switchstate;
593 }
Willy Tarreaubc4af052011-02-13 13:25:14 +0100594 si->applet.st0 = PEER_SESSION_WAITMSG;
Emeric Brun2b920a12010-09-23 18:30:22 +0200595 /* fall through */
596 }
597 case PEER_SESSION_WAITMSG: {
Willy Tarreauf2943dc2012-10-26 20:10:28 +0200598 struct peer_session *ps = (struct peer_session *)si->conn->xprt_ctx;
Emeric Brun2b920a12010-09-23 18:30:22 +0200599 char c;
600 int totl = 0;
601
Willy Tarreau9dab5fc2012-05-07 11:56:55 +0200602 reql = bo_getblk(si->ob, (char *)&c, sizeof(c), totl);
Emeric Brun2b920a12010-09-23 18:30:22 +0200603 if (reql <= 0) { /* closed or EOL not found */
604 if (reql == 0) {
605 /* nothing to read */
606 goto incomplete;
607 }
Willy Tarreaubc4af052011-02-13 13:25:14 +0100608 si->applet.st0 = PEER_SESSION_END;
Emeric Brun2b920a12010-09-23 18:30:22 +0200609 goto switchstate;
610 }
611 totl += reql;
612
613 if ((c & 0x80) || (c == 'D')) {
614 /* Here we have data message */
615 unsigned int pushack;
616 struct stksess *ts;
617 struct stksess *newts;
618 struct stktable_key stkey;
619 int srvid;
620 uint32_t netinteger;
621
622 /* Compute update remote version */
623 if (c & 0x80) {
624 pushack = ps->pushack + (unsigned int)(c & 0x7F);
625 }
626 else {
Willy Tarreau9dab5fc2012-05-07 11:56:55 +0200627 reql = bo_getblk(si->ob, (char *)&netinteger, sizeof(netinteger), totl);
Emeric Brun2b920a12010-09-23 18:30:22 +0200628 if (reql <= 0) { /* closed or EOL not found */
629 if (reql == 0) {
630 goto incomplete;
631 }
Willy Tarreaubc4af052011-02-13 13:25:14 +0100632 si->applet.st0 = PEER_SESSION_END;
Emeric Brun2b920a12010-09-23 18:30:22 +0200633 goto switchstate;
634 }
635 totl += reql;
636 pushack = ntohl(netinteger);
637 }
638
639 /* read key */
640 if (ps->table->table->type == STKTABLE_TYPE_STRING) {
641 /* type string */
642 stkey.key = stkey.data.buf;
643
Willy Tarreau9dab5fc2012-05-07 11:56:55 +0200644 reql = bo_getblk(si->ob, (char *)&netinteger, sizeof(netinteger), totl);
Emeric Brun2b920a12010-09-23 18:30:22 +0200645 if (reql <= 0) { /* closed or EOL not found */
646 if (reql == 0) {
647 goto incomplete;
648 }
Willy Tarreaubc4af052011-02-13 13:25:14 +0100649 si->applet.st0 = PEER_SESSION_END;
Emeric Brun2b920a12010-09-23 18:30:22 +0200650 goto switchstate;
651 }
652 totl += reql;
653 stkey.key_len = ntohl(netinteger);
654
Willy Tarreau9dab5fc2012-05-07 11:56:55 +0200655 reql = bo_getblk(si->ob, stkey.key, stkey.key_len, totl);
Emeric Brun2b920a12010-09-23 18:30:22 +0200656 if (reql <= 0) { /* closed or EOL not found */
657 if (reql == 0) {
658 goto incomplete;
659 }
Willy Tarreaubc4af052011-02-13 13:25:14 +0100660 si->applet.st0 = PEER_SESSION_END;
Emeric Brun2b920a12010-09-23 18:30:22 +0200661 goto switchstate;
662 }
663 totl += reql;
664 }
665 else if (ps->table->table->type == STKTABLE_TYPE_INTEGER) {
666 /* type integer */
667 stkey.key_len = (size_t)-1;
668 stkey.key = &stkey.data.integer;
669
Willy Tarreau9dab5fc2012-05-07 11:56:55 +0200670 reql = bo_getblk(si->ob, (char *)&netinteger, sizeof(netinteger), totl);
Emeric Brun2b920a12010-09-23 18:30:22 +0200671 if (reql <= 0) { /* closed or EOL not found */
672 if (reql == 0) {
673 goto incomplete;
674 }
Willy Tarreaubc4af052011-02-13 13:25:14 +0100675 si->applet.st0 = PEER_SESSION_END;
Emeric Brun2b920a12010-09-23 18:30:22 +0200676 goto switchstate;
677 }
678 totl += reql;
679 stkey.data.integer = ntohl(netinteger);
680 }
681 else {
682 /* type ip */
683 stkey.key_len = (size_t)-1;
684 stkey.key = stkey.data.buf;
685
Willy Tarreau9dab5fc2012-05-07 11:56:55 +0200686 reql = bo_getblk(si->ob, (char *)&stkey.data.buf, ps->table->table->key_size, totl);
Emeric Brun2b920a12010-09-23 18:30:22 +0200687 if (reql <= 0) { /* closed or EOL not found */
688 if (reql == 0) {
689 goto incomplete;
690 }
Willy Tarreaubc4af052011-02-13 13:25:14 +0100691 si->applet.st0 = PEER_SESSION_END;
Emeric Brun2b920a12010-09-23 18:30:22 +0200692 goto switchstate;
693 }
694 totl += reql;
695
696 }
697
698 /* read server id */
Willy Tarreau9dab5fc2012-05-07 11:56:55 +0200699 reql = bo_getblk(si->ob, (char *)&netinteger, sizeof(netinteger), totl);
Emeric Brun2b920a12010-09-23 18:30:22 +0200700 if (reql <= 0) { /* closed or EOL not found */
701 if (reql == 0) {
702 goto incomplete;
703 }
Willy Tarreaubc4af052011-02-13 13:25:14 +0100704 si->applet.st0 = PEER_SESSION_END;
Emeric Brun2b920a12010-09-23 18:30:22 +0200705 goto switchstate;
706 }
707 totl += reql;
708 srvid = ntohl(netinteger);
709
710 /* update entry */
711 newts = stksess_new(ps->table->table, &stkey);
712 if (newts) {
713 /* lookup for existing entry */
714 ts = stktable_lookup(ps->table->table, newts);
715 if (ts) {
716 /* the entry already exist, we can free ours */
717 stktable_touch(ps->table->table, ts, 0);
718 stksess_free(ps->table->table, newts);
719 }
720 else {
721 struct eb32_node *eb;
722
723 /* create new entry */
724 ts = stktable_store(ps->table->table, newts, 0);
725 ts->upd.key= (++ps->table->table->update)+(2^31);
726 eb = eb32_insert(&ps->table->table->updates, &ts->upd);
727 if (eb != &ts->upd) {
728 eb32_delete(eb);
729 eb32_insert(&ps->table->table->updates, &ts->upd);
730 }
731 }
732
733 /* update entry */
734 if (srvid && stktable_data_ptr(ps->table->table, ts, STKTABLE_DT_SERVER_ID))
735 stktable_data_cast(stktable_data_ptr(ps->table->table, ts, STKTABLE_DT_SERVER_ID), server_id) = srvid;
736 ps->pushack = pushack;
737 }
738
739 }
740 else if (c == 'R') {
741 /* Reset message: remote need resync */
742
743 /* reinit counters for a resync */
744 ps->lastpush = 0;
745 ps->teaching_origin = ps->pushed = ps->table->table->update;
746
747 /* reset teaching flags to 0 */
748 ps->flags &= PEER_TEACH_RESET;
749
750 /* flag to start to teach lesson */
751 ps->flags |= PEER_F_TEACH_PROCESS;
752 }
753 else if (c == 'F') {
754 /* Finish message, all known updates have been pushed by remote */
755 /* and remote is up to date */
756
757 /* If resync is in progress with remote peer */
758 if (ps->flags & PEER_F_LEARN_ASSIGN) {
759
760 /* unassign current peer for learning */
761 ps->flags &= ~PEER_F_LEARN_ASSIGN;
762 ps->table->flags &= ~(SHTABLE_F_RESYNC_ASSIGN|SHTABLE_F_RESYNC_PROCESS);
763
764 /* Consider table is now up2date, resync resync no more needed from local neither remote */
765 ps->table->flags |= (SHTABLE_F_RESYNC_LOCAL|SHTABLE_F_RESYNC_REMOTE);
766 }
767 /* Increase confirm counter to launch a confirm message */
768 ps->confirm++;
769 }
770 else if (c == 'c') {
771 /* confirm message, remote peer is now up to date with us */
772
773 /* If stopping state */
774 if (stopping) {
775 /* Close session, push resync no more needed */
776 ps->flags |= PEER_F_TEACH_COMPLETE;
Willy Tarreaubc4af052011-02-13 13:25:14 +0100777 si->applet.st0 = PEER_SESSION_END;
Emeric Brun2b920a12010-09-23 18:30:22 +0200778 goto switchstate;
779 }
780
781 /* reset teaching flags to 0 */
782 ps->flags &= PEER_TEACH_RESET;
783 }
784 else if (c == 'C') {
785 /* Continue message, all known updates have been pushed by remote */
786 /* but remote is not up to date */
787
788 /* If resync is in progress with current peer */
789 if (ps->flags & PEER_F_LEARN_ASSIGN) {
790
791 /* unassign current peer */
792 ps->flags &= ~PEER_F_LEARN_ASSIGN;
793 ps->table->flags &= ~(SHTABLE_F_RESYNC_ASSIGN|SHTABLE_F_RESYNC_PROCESS);
794
795 /* flag current peer is not up 2 date to try from an other */
796 ps->flags |= PEER_F_LEARN_NOTUP2DATE;
797
798 /* reschedule a resync */
799 ps->table->resync_timeout = tick_add(now_ms, MS_TO_TICKS(5000));
800 task_wakeup(ps->table->sync_task, TASK_WOKEN_MSG);
801 }
802 ps->confirm++;
803 }
804 else if (c == 'A') {
805 /* ack message */
806 uint32_t netinteger;
807
Willy Tarreau9dab5fc2012-05-07 11:56:55 +0200808 reql = bo_getblk(si->ob, (char *)&netinteger, sizeof(netinteger), totl);
Emeric Brun2b920a12010-09-23 18:30:22 +0200809 if (reql <= 0) { /* closed or EOL not found */
810 if (reql == 0) {
811 goto incomplete;
812 }
Willy Tarreaubc4af052011-02-13 13:25:14 +0100813 si->applet.st0 = PEER_SESSION_END;
Emeric Brun2b920a12010-09-23 18:30:22 +0200814 goto switchstate;
815 }
816 totl += reql;
817
818 /* Consider remote is up to date with "acked" version */
819 ps->update = ntohl(netinteger);
820 }
821 else {
822 /* Unknown message */
Willy Tarreaubc4af052011-02-13 13:25:14 +0100823 si->applet.st0 = PEER_SESSION_END;
Emeric Brun2b920a12010-09-23 18:30:22 +0200824 goto switchstate;
825 }
826
827 /* skip consumed message */
Willy Tarreau9dab5fc2012-05-07 11:56:55 +0200828 bo_skip(si->ob, totl);
Emeric Brun2b920a12010-09-23 18:30:22 +0200829
830 /* loop on that state to peek next message */
831 continue;
832incomplete:
833 /* Nothing to read, now we start to write */
834
835 /* Confirm finished or partial messages */
836 while (ps->confirm) {
837 /* There is a confirm messages to send */
Willy Tarreau9dab5fc2012-05-07 11:56:55 +0200838 repl = bi_putchr(si->ib, 'c');
Emeric Brun2b920a12010-09-23 18:30:22 +0200839 if (repl <= 0) {
840 /* no more write possible */
841 if (repl == -1)
842 goto out;
Willy Tarreaubc4af052011-02-13 13:25:14 +0100843 si->applet.st0 = PEER_SESSION_END;
Emeric Brun2b920a12010-09-23 18:30:22 +0200844 goto switchstate;
845 }
846 ps->confirm--;
847 }
848
849 /* Need to request a resync */
850 if ((ps->flags & PEER_F_LEARN_ASSIGN) &&
851 (ps->table->flags & SHTABLE_F_RESYNC_ASSIGN) &&
852 !(ps->table->flags & SHTABLE_F_RESYNC_PROCESS)) {
853 /* Current peer was elected to request a resync */
854
Willy Tarreau9dab5fc2012-05-07 11:56:55 +0200855 repl = bi_putchr(si->ib, 'R');
Emeric Brun2b920a12010-09-23 18:30:22 +0200856 if (repl <= 0) {
857 /* no more write possible */
858 if (repl == -1)
859 goto out;
Willy Tarreaubc4af052011-02-13 13:25:14 +0100860 si->applet.st0 = PEER_SESSION_END;
Emeric Brun2b920a12010-09-23 18:30:22 +0200861 goto switchstate;
862 }
863 ps->table->flags |= SHTABLE_F_RESYNC_PROCESS;
864 }
865
866 /* It remains some updates to ack */
867 if (ps->pushack != ps->lastack) {
868 uint32_t netinteger;
869
Willy Tarreau19d14ef2012-10-29 16:51:55 +0100870 trash.str[0] = 'A';
Emeric Brun2b920a12010-09-23 18:30:22 +0200871 netinteger = htonl(ps->pushack);
Willy Tarreau19d14ef2012-10-29 16:51:55 +0100872 memcpy(&trash.str[1], &netinteger, sizeof(netinteger));
Emeric Brun2b920a12010-09-23 18:30:22 +0200873
Willy Tarreau19d14ef2012-10-29 16:51:55 +0100874 repl = bi_putblk(si->ib, trash.str, 1+sizeof(netinteger));
Emeric Brun2b920a12010-09-23 18:30:22 +0200875 if (repl <= 0) {
876 /* no more write possible */
877 if (repl == -1)
878 goto out;
Willy Tarreaubc4af052011-02-13 13:25:14 +0100879 si->applet.st0 = PEER_SESSION_END;
Emeric Brun2b920a12010-09-23 18:30:22 +0200880 goto switchstate;
881 }
882 ps->lastack = ps->pushack;
883 }
884
885 if (ps->flags & PEER_F_TEACH_PROCESS) {
886 /* current peer was requested for a lesson */
887
888 if (!(ps->flags & PEER_F_TEACH_STAGE1)) {
889 /* lesson stage 1 not complete */
890 struct eb32_node *eb;
891
892 eb = eb32_lookup_ge(&ps->table->table->updates, ps->pushed+1);
893 while (1) {
894 int msglen;
895 struct stksess *ts;
896
897 if (!eb) {
898 /* flag lesson stage1 complete */
899 ps->flags |= PEER_F_TEACH_STAGE1;
900 eb = eb32_first(&ps->table->table->updates);
901 if (eb)
902 ps->pushed = eb->key - 1;
903 break;
904 }
905
906 ts = eb32_entry(eb, struct stksess, upd);
Willy Tarreau19d14ef2012-10-29 16:51:55 +0100907 msglen = peer_prepare_datamsg(ts, ps, trash.str, trash.size);
Emeric Brun2b920a12010-09-23 18:30:22 +0200908 if (msglen) {
909 /* message to buffer */
Willy Tarreau19d14ef2012-10-29 16:51:55 +0100910 repl = bi_putblk(si->ib, trash.str, msglen);
Emeric Brun2b920a12010-09-23 18:30:22 +0200911 if (repl <= 0) {
912 /* no more write possible */
913 if (repl == -1)
914 goto out;
Willy Tarreaubc4af052011-02-13 13:25:14 +0100915 si->applet.st0 = PEER_SESSION_END;
Emeric Brun2b920a12010-09-23 18:30:22 +0200916 goto switchstate;
917 }
918 ps->lastpush = ps->pushed = ts->upd.key;
919 }
920 eb = eb32_next(eb);
921 }
922 } /* !TEACH_STAGE1 */
923
924 if (!(ps->flags & PEER_F_TEACH_STAGE2)) {
925 /* lesson stage 2 not complete */
926 struct eb32_node *eb;
927
928 eb = eb32_lookup_ge(&ps->table->table->updates, ps->pushed+1);
929 while (1) {
930 int msglen;
931 struct stksess *ts;
932
933 if (!eb || eb->key > ps->teaching_origin) {
934 /* flag lesson stage1 complete */
935 ps->flags |= PEER_F_TEACH_STAGE2;
936 ps->pushed = ps->teaching_origin;
937 break;
938 }
939
940 ts = eb32_entry(eb, struct stksess, upd);
Willy Tarreau19d14ef2012-10-29 16:51:55 +0100941 msglen = peer_prepare_datamsg(ts, ps, trash.str, trash.size);
Emeric Brun2b920a12010-09-23 18:30:22 +0200942 if (msglen) {
943 /* message to buffer */
Willy Tarreau19d14ef2012-10-29 16:51:55 +0100944 repl = bi_putblk(si->ib, trash.str, msglen);
Emeric Brun2b920a12010-09-23 18:30:22 +0200945 if (repl <= 0) {
946 /* no more write possible */
947 if (repl == -1)
948 goto out;
Willy Tarreaubc4af052011-02-13 13:25:14 +0100949 si->applet.st0 = PEER_SESSION_END;
Emeric Brun2b920a12010-09-23 18:30:22 +0200950 goto switchstate;
951 }
952 ps->lastpush = ps->pushed = ts->upd.key;
953 }
954 eb = eb32_next(eb);
955 }
956 } /* !TEACH_STAGE2 */
957
958 if (!(ps->flags & PEER_F_TEACH_FINISHED)) {
959 /* process final lesson message */
Willy Tarreau9dab5fc2012-05-07 11:56:55 +0200960 repl = bi_putchr(si->ib, ((ps->table->flags & SHTABLE_RESYNC_STATEMASK) == SHTABLE_RESYNC_FINISHED) ? 'F' : 'C');
Emeric Brun2b920a12010-09-23 18:30:22 +0200961 if (repl <= 0) {
962 /* no more write possible */
963 if (repl == -1)
964 goto out;
Willy Tarreaubc4af052011-02-13 13:25:14 +0100965 si->applet.st0 = PEER_SESSION_END;
Emeric Brun2b920a12010-09-23 18:30:22 +0200966 goto switchstate;
967 }
968
969 /* flag finished message sent */
970 ps->flags |= PEER_F_TEACH_FINISHED;
971 } /* !TEACH_FINISHED */
972 } /* TEACH_PROCESS */
973
974 if (!(ps->flags & PEER_F_LEARN_ASSIGN) &&
975 (int)(ps->pushed - ps->table->table->localupdate) < 0) {
976 /* Push local updates, only if no learning in progress (to avoid ping-pong effects) */
977 struct eb32_node *eb;
978
979 eb = eb32_lookup_ge(&ps->table->table->updates, ps->pushed+1);
980 while (1) {
981 int msglen;
982 struct stksess *ts;
983
984 /* push local updates */
985 if (!eb) {
986 eb = eb32_first(&ps->table->table->updates);
987 if (!eb || ((int)(eb->key - ps->pushed) <= 0)) {
988 ps->pushed = ps->table->table->localupdate;
989 break;
990 }
991 }
992
993 if ((int)(eb->key - ps->table->table->localupdate) > 0) {
994 ps->pushed = ps->table->table->localupdate;
995 break;
996 }
997
998 ts = eb32_entry(eb, struct stksess, upd);
Willy Tarreau19d14ef2012-10-29 16:51:55 +0100999 msglen = peer_prepare_datamsg(ts, ps, trash.str, trash.size);
Emeric Brun2b920a12010-09-23 18:30:22 +02001000 if (msglen) {
1001 /* message to buffer */
Willy Tarreau19d14ef2012-10-29 16:51:55 +01001002 repl = bi_putblk(si->ib, trash.str, msglen);
Emeric Brun2b920a12010-09-23 18:30:22 +02001003 if (repl <= 0) {
1004 /* no more write possible */
1005 if (repl == -1)
1006 goto out;
Willy Tarreaubc4af052011-02-13 13:25:14 +01001007 si->applet.st0 = PEER_SESSION_END;
Emeric Brun2b920a12010-09-23 18:30:22 +02001008 goto switchstate;
1009 }
1010 ps->lastpush = ps->pushed = ts->upd.key;
1011 }
1012 eb = eb32_next(eb);
1013 }
1014 } /* ! LEARN_ASSIGN */
1015 /* noting more to do */
1016 goto out;
1017 }
1018 case PEER_SESSION_EXIT:
Willy Tarreau19d14ef2012-10-29 16:51:55 +01001019 repl = snprintf(trash.str, trash.size, "%d\n", si->applet.st1);
Emeric Brun2b920a12010-09-23 18:30:22 +02001020
Willy Tarreau19d14ef2012-10-29 16:51:55 +01001021 if (bi_putblk(si->ib, trash.str, repl) == -1)
Emeric Brun2b920a12010-09-23 18:30:22 +02001022 goto out;
Willy Tarreaubc4af052011-02-13 13:25:14 +01001023 si->applet.st0 = PEER_SESSION_END;
Emeric Brun2b920a12010-09-23 18:30:22 +02001024 /* fall through */
1025 case PEER_SESSION_END: {
Willy Tarreau73b013b2012-05-21 16:31:45 +02001026 si_shutw(si);
1027 si_shutr(si);
Willy Tarreau03cdb7c2012-08-27 23:14:58 +02001028 si->ib->flags |= CF_READ_NULL;
Emeric Brun2b920a12010-09-23 18:30:22 +02001029 goto quit;
1030 }
1031 }
1032 }
1033out:
Willy Tarreau73b013b2012-05-21 16:31:45 +02001034 si_update(si);
Willy Tarreau03cdb7c2012-08-27 23:14:58 +02001035 si->ob->flags |= CF_READ_DONTWAIT;
Emeric Brun2b920a12010-09-23 18:30:22 +02001036 /* we don't want to expire timeouts while we're processing requests */
1037 si->ib->rex = TICK_ETERNITY;
1038 si->ob->wex = TICK_ETERNITY;
1039quit:
1040 return;
1041}
1042
Willy Tarreaub24281b2011-02-13 13:16:36 +01001043static struct si_applet peer_applet = {
Willy Tarreau3fdb3662012-11-12 00:42:33 +01001044 .obj_type = OBJ_TYPE_APPLET,
Willy Tarreaub24281b2011-02-13 13:16:36 +01001045 .name = "<PEER>", /* used for logging */
1046 .fct = peer_io_handler,
Aman Gupta9a13e842012-04-02 18:57:53 -07001047 .release = peer_session_release,
Willy Tarreaub24281b2011-02-13 13:16:36 +01001048};
Emeric Brun2b920a12010-09-23 18:30:22 +02001049
1050/*
1051 * Use this function to force a close of a peer session
1052 */
Simon Horman96553772011-06-08 09:18:51 +09001053static void peer_session_forceshutdown(struct session * session)
Emeric Brun2b920a12010-09-23 18:30:22 +02001054{
1055 struct stream_interface *oldsi;
1056
Willy Tarreau3fdb3662012-11-12 00:42:33 +01001057 if (objt_applet(session->si[0].conn->target) == &peer_applet) {
Emeric Brun2b920a12010-09-23 18:30:22 +02001058 oldsi = &session->si[0];
1059 }
1060 else {
1061 oldsi = &session->si[1];
1062 }
1063
1064 /* call release to reinit resync states if needed */
1065 peer_session_release(oldsi);
Willy Tarreaubc4af052011-02-13 13:25:14 +01001066 oldsi->applet.st0 = PEER_SESSION_END;
Willy Tarreauf2943dc2012-10-26 20:10:28 +02001067 oldsi->conn->xprt_ctx = NULL;
Emeric Brun2b920a12010-09-23 18:30:22 +02001068 task_wakeup(session->task, TASK_WOKEN_MSG);
1069}
1070
1071/*
1072 * this function is called on a read event from a listen socket, corresponding
1073 * to an accept. It tries to accept as many connections as possible.
Willy Tarreaubd55e312010-11-11 10:55:09 +01001074 * It returns a positive value upon success, 0 if the connection needs to be
1075 * closed and ignored, or a negative value upon critical failure.
Emeric Brun2b920a12010-09-23 18:30:22 +02001076 */
1077int peer_accept(struct session *s)
1078{
1079 /* we have a dedicated I/O handler for the stats */
Willy Tarreaub24281b2011-02-13 13:16:36 +01001080 stream_int_register_handler(&s->si[1], &peer_applet);
Willy Tarreau3fdb3662012-11-12 00:42:33 +01001081 s->target = s->si[1].conn->target; // for logging only
Willy Tarreauf2943dc2012-10-26 20:10:28 +02001082 s->si[1].conn->xprt_ctx = s;
Willy Tarreaubc4af052011-02-13 13:25:14 +01001083 s->si[1].applet.st0 = PEER_SESSION_ACCEPT;
Emeric Brun2b920a12010-09-23 18:30:22 +02001084
1085 tv_zero(&s->logs.tv_request);
1086 s->logs.t_queue = 0;
1087 s->logs.t_connect = 0;
1088 s->logs.t_data = 0;
1089 s->logs.t_close = 0;
1090 s->logs.bytes_in = s->logs.bytes_out = 0;
1091 s->logs.prx_queue_size = 0;/* we get the number of pending conns before us */
1092 s->logs.srv_queue_size = 0; /* we will get this number soon */
1093
Willy Tarreau03cdb7c2012-08-27 23:14:58 +02001094 s->req->flags |= CF_READ_DONTWAIT; /* we plan to read small requests */
Emeric Brun2b920a12010-09-23 18:30:22 +02001095
1096 if (s->listener->timeout) {
1097 s->req->rto = *s->listener->timeout;
1098 s->rep->wto = *s->listener->timeout;
1099 }
1100 return 1;
1101}
1102
1103/*
Willy Tarreaubd55e312010-11-11 10:55:09 +01001104 * Create a new peer session in assigned state (connect will start automatically)
Emeric Brun2b920a12010-09-23 18:30:22 +02001105 */
Simon Horman96553772011-06-08 09:18:51 +09001106static struct session *peer_session_create(struct peer *peer, struct peer_session *ps)
Emeric Brun2b920a12010-09-23 18:30:22 +02001107{
Willy Tarreau4348fad2012-09-20 16:48:07 +02001108 struct listener *l = LIST_NEXT(&peer->peers->peers_fe->conf.listeners, struct listener *, by_fe);
Emeric Brun2b920a12010-09-23 18:30:22 +02001109 struct proxy *p = (struct proxy *)l->frontend; /* attached frontend */
1110 struct session *s;
1111 struct http_txn *txn;
1112 struct task *t;
1113
1114 if ((s = pool_alloc2(pool2_session)) == NULL) { /* disable this proxy for a while */
1115 Alert("out of memory in event_accept().\n");
Emeric Brun2b920a12010-09-23 18:30:22 +02001116 goto out_close;
1117 }
1118
Willy Tarreauf2943dc2012-10-26 20:10:28 +02001119 if (unlikely((s->si[0].conn = pool_alloc2(pool2_connection)) == NULL))
1120 goto out_fail_conn0;
1121
1122 if (unlikely((s->si[1].conn = pool_alloc2(pool2_connection)) == NULL))
1123 goto out_fail_conn1;
1124
Emeric Brun2b920a12010-09-23 18:30:22 +02001125 LIST_ADDQ(&sessions, &s->list);
1126 LIST_INIT(&s->back_refs);
1127
1128 s->flags = SN_ASSIGNED|SN_ADDR_SET;
Emeric Brun2b920a12010-09-23 18:30:22 +02001129
1130 /* if this session comes from a known monitoring system, we want to ignore
1131 * it as soon as possible, which means closing it immediately for TCP.
1132 */
1133 if ((t = task_new()) == NULL) { /* disable this proxy for a while */
1134 Alert("out of memory in event_accept().\n");
Emeric Brun2b920a12010-09-23 18:30:22 +02001135 goto out_free_session;
1136 }
1137
1138 ps->reconnect = tick_add(now_ms, MS_TO_TICKS(5000));
1139 ps->statuscode = PEER_SESSION_CONNECTCODE;
1140
1141 t->process = l->handler;
1142 t->context = s;
1143 t->nice = l->nice;
1144
Willy Tarreauf2943dc2012-10-26 20:10:28 +02001145 memcpy(&s->si[1].conn->addr.to, &peer->addr, sizeof(s->si[1].conn->addr.to));
Emeric Brun2b920a12010-09-23 18:30:22 +02001146 s->task = t;
1147 s->listener = l;
1148
1149 /* Note: initially, the session's backend points to the frontend.
1150 * This changes later when switching rules are executed or
1151 * when the default backend is assigned.
1152 */
1153 s->be = s->fe = p;
1154
1155 s->req = s->rep = NULL; /* will be allocated later */
1156
Willy Tarreauf2943dc2012-10-26 20:10:28 +02001157 s->si[0].conn->t.sock.fd = -1;
1158 s->si[0].conn->flags = CO_FL_NONE;
Willy Tarreau14cba4b2012-11-30 17:33:05 +01001159 s->si[0].conn->err_code = CO_ER_NONE;
Emeric Brun2b920a12010-09-23 18:30:22 +02001160 s->si[0].owner = t;
1161 s->si[0].state = s->si[0].prev_state = SI_ST_EST;
1162 s->si[0].err_type = SI_ET_NONE;
1163 s->si[0].err_loc = NULL;
Willy Tarreau26d8c592012-05-07 18:12:14 +02001164 s->si[0].release = NULL;
Willy Tarreau63e7fe32012-05-08 15:20:43 +02001165 s->si[0].send_proxy_ofs = 0;
Willy Tarreau3fdb3662012-11-12 00:42:33 +01001166 s->si[0].conn->target = &l->obj_type;
Emeric Brun2b920a12010-09-23 18:30:22 +02001167 s->si[0].exp = TICK_ETERNITY;
1168 s->si[0].flags = SI_FL_NONE;
1169 if (s->fe->options2 & PR_O2_INDEPSTR)
1170 s->si[0].flags |= SI_FL_INDEP_STR;
Emeric Brun2b920a12010-09-23 18:30:22 +02001171
Willy Tarreaub24281b2011-02-13 13:16:36 +01001172 stream_int_register_handler(&s->si[0], &peer_applet);
Willy Tarreaufa6bac62012-05-31 14:16:59 +02001173 s->si[0].applet.st0 = PEER_SESSION_CONNECT;
Willy Tarreauf2943dc2012-10-26 20:10:28 +02001174 s->si[0].conn->xprt_ctx = (void *)ps;
Emeric Brun2b920a12010-09-23 18:30:22 +02001175
Willy Tarreauf2943dc2012-10-26 20:10:28 +02001176 s->si[1].conn->t.sock.fd = -1; /* just to help with debugging */
1177 s->si[1].conn->flags = CO_FL_NONE;
Willy Tarreau14cba4b2012-11-30 17:33:05 +01001178 s->si[1].conn->err_code = CO_ER_NONE;
Emeric Brun2b920a12010-09-23 18:30:22 +02001179 s->si[1].owner = t;
1180 s->si[1].state = s->si[1].prev_state = SI_ST_ASS;
1181 s->si[1].conn_retries = p->conn_retries;
1182 s->si[1].err_type = SI_ET_NONE;
1183 s->si[1].err_loc = NULL;
Willy Tarreau26d8c592012-05-07 18:12:14 +02001184 s->si[1].release = NULL;
Willy Tarreau63e7fe32012-05-08 15:20:43 +02001185 s->si[1].send_proxy_ofs = 0;
Willy Tarreau3fdb3662012-11-12 00:42:33 +01001186 s->si[1].conn->target = &s->be->obj_type;
Willy Tarreauf7bc57c2012-10-03 00:19:48 +02001187 si_prepare_conn(&s->si[1], peer->proto, peer->xprt);
Emeric Brun2b920a12010-09-23 18:30:22 +02001188 s->si[1].exp = TICK_ETERNITY;
1189 s->si[1].flags = SI_FL_NONE;
1190 if (s->be->options2 & PR_O2_INDEPSTR)
1191 s->si[1].flags |= SI_FL_INDEP_STR;
1192
Willy Tarreau9bd0d742011-07-20 00:17:39 +02001193 session_init_srv_conn(s);
Willy Tarreau3fdb3662012-11-12 00:42:33 +01001194 s->target = &s->be->obj_type;
Emeric Brun2b920a12010-09-23 18:30:22 +02001195 s->pend_pos = NULL;
1196
1197 /* init store persistence */
1198 s->store_count = 0;
1199 s->stkctr1_entry = NULL;
1200 s->stkctr2_entry = NULL;
1201
1202 /* FIXME: the logs are horribly complicated now, because they are
1203 * defined in <p>, <p>, and later <be> and <be>.
1204 */
1205
1206 s->logs.logwait = 0;
1207 s->do_log = NULL;
1208
1209 /* default error reporting function, may be changed by analysers */
1210 s->srv_error = default_srv_error;
1211
Emeric Brun2b920a12010-09-23 18:30:22 +02001212 s->uniq_id = 0;
Willy Tarreaubd833142012-05-08 15:51:44 +02001213 s->unique_id = NULL;
Emeric Brun2b920a12010-09-23 18:30:22 +02001214
1215 txn = &s->txn;
1216 /* Those variables will be checked and freed if non-NULL in
1217 * session.c:session_free(). It is important that they are
1218 * properly initialized.
1219 */
1220 txn->sessid = NULL;
1221 txn->srv_cookie = NULL;
1222 txn->cli_cookie = NULL;
1223 txn->uri = NULL;
1224 txn->req.cap = NULL;
1225 txn->rsp.cap = NULL;
1226 txn->hdr_idx.v = NULL;
1227 txn->hdr_idx.size = txn->hdr_idx.used = 0;
1228
Willy Tarreau8263d2b2012-08-28 00:06:31 +02001229 if ((s->req = pool_alloc2(pool2_channel)) == NULL)
Emeric Brun2b920a12010-09-23 18:30:22 +02001230 goto out_fail_req; /* no memory */
1231
Willy Tarreau9b28e032012-10-12 23:49:43 +02001232 if ((s->req->buf = pool_alloc2(pool2_buffer)) == NULL)
1233 goto out_fail_req_buf; /* no memory */
1234
Willy Tarreau19d14ef2012-10-29 16:51:55 +01001235 s->req->buf->size = trash.size;
Willy Tarreau8263d2b2012-08-28 00:06:31 +02001236 channel_init(s->req);
Emeric Brun2b920a12010-09-23 18:30:22 +02001237 s->req->prod = &s->si[0];
1238 s->req->cons = &s->si[1];
1239 s->si[0].ib = s->si[1].ob = s->req;
1240
Willy Tarreau03cdb7c2012-08-27 23:14:58 +02001241 s->req->flags |= CF_READ_ATTACHED; /* the producer is already connected */
Emeric Brun2b920a12010-09-23 18:30:22 +02001242
1243 /* activate default analysers enabled for this listener */
1244 s->req->analysers = l->analysers;
1245
1246 /* note: this should not happen anymore since there's always at least the switching rules */
1247 if (!s->req->analysers) {
Willy Tarreau8263d2b2012-08-28 00:06:31 +02001248 channel_auto_connect(s->req);/* don't wait to establish connection */
1249 channel_auto_close(s->req);/* let the producer forward close requests */
Emeric Brun2b920a12010-09-23 18:30:22 +02001250 }
1251
1252 s->req->rto = s->fe->timeout.client;
1253 s->req->wto = s->be->timeout.server;
1254
Willy Tarreau8263d2b2012-08-28 00:06:31 +02001255 if ((s->rep = pool_alloc2(pool2_channel)) == NULL)
Emeric Brun2b920a12010-09-23 18:30:22 +02001256 goto out_fail_rep; /* no memory */
1257
Willy Tarreau9b28e032012-10-12 23:49:43 +02001258 if ((s->rep->buf = pool_alloc2(pool2_buffer)) == NULL)
1259 goto out_fail_rep_buf; /* no memory */
1260
Willy Tarreau19d14ef2012-10-29 16:51:55 +01001261 s->rep->buf->size = trash.size;
Willy Tarreau8263d2b2012-08-28 00:06:31 +02001262 channel_init(s->rep);
Emeric Brun2b920a12010-09-23 18:30:22 +02001263 s->rep->prod = &s->si[1];
1264 s->rep->cons = &s->si[0];
1265 s->si[0].ob = s->si[1].ib = s->rep;
1266
1267 s->rep->rto = s->be->timeout.server;
1268 s->rep->wto = s->fe->timeout.client;
1269
1270 s->req->rex = TICK_ETERNITY;
1271 s->req->wex = TICK_ETERNITY;
1272 s->req->analyse_exp = TICK_ETERNITY;
1273 s->rep->rex = TICK_ETERNITY;
1274 s->rep->wex = TICK_ETERNITY;
1275 s->rep->analyse_exp = TICK_ETERNITY;
1276 t->expire = TICK_ETERNITY;
1277
Willy Tarreau03cdb7c2012-08-27 23:14:58 +02001278 s->rep->flags |= CF_READ_DONTWAIT;
Emeric Brun2b920a12010-09-23 18:30:22 +02001279 /* it is important not to call the wakeup function directly but to
1280 * pass through task_wakeup(), because this one knows how to apply
1281 * priorities to tasks.
1282 */
1283 task_wakeup(t, TASK_WOKEN_INIT);
1284
1285 l->nbconn++; /* warning! right now, it's up to the handler to decrease this */
1286 p->feconn++;/* beconn will be increased later */
1287 jobs++;
Willy Tarreau3c63fd82011-09-07 18:00:47 +02001288 if (!(s->listener->options & LI_O_UNLIMITED))
1289 actconn++;
Emeric Brun2b920a12010-09-23 18:30:22 +02001290 totalconn++;
1291
1292 return s;
1293
1294 /* Error unrolling */
Willy Tarreau9b28e032012-10-12 23:49:43 +02001295 out_fail_rep_buf:
1296 pool_free2(pool2_channel, s->rep);
Emeric Brun2b920a12010-09-23 18:30:22 +02001297 out_fail_rep:
Willy Tarreau9b28e032012-10-12 23:49:43 +02001298 pool_free2(pool2_buffer, s->req->buf);
1299 out_fail_req_buf:
Willy Tarreau8263d2b2012-08-28 00:06:31 +02001300 pool_free2(pool2_channel, s->req);
Emeric Brun2b920a12010-09-23 18:30:22 +02001301 out_fail_req:
1302 task_free(t);
1303 out_free_session:
1304 LIST_DEL(&s->list);
Willy Tarreauf2943dc2012-10-26 20:10:28 +02001305 pool_free2(pool2_connection, s->si[1].conn);
1306 out_fail_conn1:
1307 pool_free2(pool2_connection, s->si[0].conn);
1308 out_fail_conn0:
Emeric Brun2b920a12010-09-23 18:30:22 +02001309 pool_free2(pool2_session, s);
1310 out_close:
1311 return s;
1312}
1313
1314/*
1315 * Task processing function to manage re-connect and peer session
1316 * tasks wakeup on local update.
1317 */
Simon Horman96553772011-06-08 09:18:51 +09001318static struct task *process_peer_sync(struct task * task)
Emeric Brun2b920a12010-09-23 18:30:22 +02001319{
1320 struct shared_table *st = (struct shared_table *)task->context;
1321 struct peer_session *ps;
1322
1323 task->expire = TICK_ETERNITY;
1324
1325 if (!stopping) {
1326 /* Normal case (not soft stop)*/
1327 if (((st->flags & SHTABLE_RESYNC_STATEMASK) == SHTABLE_RESYNC_FROMLOCAL) &&
1328 (!nb_oldpids || tick_is_expired(st->resync_timeout, now_ms)) &&
1329 !(st->flags & SHTABLE_F_RESYNC_ASSIGN)) {
1330 /* Resync from local peer needed
1331 no peer was assigned for the lesson
1332 and no old local peer found
1333 or resync timeout expire */
1334
1335 /* flag no more resync from local, to try resync from remotes */
1336 st->flags |= SHTABLE_F_RESYNC_LOCAL;
1337
1338 /* reschedule a resync */
1339 st->resync_timeout = tick_add(now_ms, MS_TO_TICKS(5000));
1340 }
1341
1342 /* For each session */
1343 for (ps = st->sessions; ps; ps = ps->next) {
1344 /* For each remote peers */
1345 if (!ps->peer->local) {
1346 if (!ps->session) {
1347 /* no active session */
1348 if (ps->statuscode == 0 ||
1349 ps->statuscode == PEER_SESSION_SUCCESSCODE ||
1350 ((ps->statuscode == PEER_SESSION_CONNECTCODE ||
1351 ps->statuscode == PEER_SESSION_CONNECTEDCODE) &&
1352 tick_is_expired(ps->reconnect, now_ms))) {
1353 /* connection never tried
1354 * or previous session established with success
1355 * or previous session failed during connection
1356 * and reconnection timer is expired */
1357
1358 /* retry a connect */
1359 ps->session = peer_session_create(ps->peer, ps);
1360 }
1361 else if (ps->statuscode == PEER_SESSION_CONNECTCODE ||
1362 ps->statuscode == PEER_SESSION_CONNECTEDCODE) {
1363 /* If previous session failed during connection
1364 * but reconnection timer is not expired */
1365
1366 /* reschedule task for reconnect */
1367 task->expire = tick_first(task->expire, ps->reconnect);
1368 }
1369 /* else do nothing */
1370 } /* !ps->session */
1371 else if (ps->statuscode == PEER_SESSION_SUCCESSCODE) {
1372 /* current session is active and established */
1373 if (((st->flags & SHTABLE_RESYNC_STATEMASK) == SHTABLE_RESYNC_FROMREMOTE) &&
1374 !(st->flags & SHTABLE_F_RESYNC_ASSIGN) &&
1375 !(ps->flags & PEER_F_LEARN_NOTUP2DATE)) {
1376 /* Resync from a remote is needed
1377 * and no peer was assigned for lesson
1378 * and current peer may be up2date */
1379
1380 /* assign peer for the lesson */
1381 ps->flags |= PEER_F_LEARN_ASSIGN;
1382 st->flags |= SHTABLE_F_RESYNC_ASSIGN;
1383
1384 /* awake peer session task to handle a request of resync */
1385 task_wakeup(ps->session->task, TASK_WOKEN_MSG);
1386 }
1387 else if ((int)(ps->pushed - ps->table->table->localupdate) < 0) {
1388 /* awake peer session task to push local updates */
1389 task_wakeup(ps->session->task, TASK_WOKEN_MSG);
1390 }
1391 /* else do nothing */
1392 } /* SUCCESSCODE */
1393 } /* !ps->peer->local */
1394 } /* for */
1395
1396 /* Resync from remotes expired: consider resync is finished */
1397 if (((st->flags & SHTABLE_RESYNC_STATEMASK) == SHTABLE_RESYNC_FROMREMOTE) &&
1398 !(st->flags & SHTABLE_F_RESYNC_ASSIGN) &&
1399 tick_is_expired(st->resync_timeout, now_ms)) {
1400 /* Resync from remote peer needed
1401 * no peer was assigned for the lesson
1402 * and resync timeout expire */
1403
1404 /* flag no more resync from remote, consider resync is finished */
1405 st->flags |= SHTABLE_F_RESYNC_REMOTE;
1406 }
1407
1408 if ((st->flags & SHTABLE_RESYNC_STATEMASK) != SHTABLE_RESYNC_FINISHED) {
1409 /* Resync not finished*/
1410 /* reschedule task to resync timeout, to ended resync if needed */
1411 task->expire = tick_first(task->expire, st->resync_timeout);
1412 }
1413 } /* !stopping */
1414 else {
1415 /* soft stop case */
1416 if (task->state & TASK_WOKEN_SIGNAL) {
1417 /* We've just recieved the signal */
1418 if (!(st->flags & SHTABLE_F_DONOTSTOP)) {
1419 /* add DO NOT STOP flag if not present */
1420 jobs++;
1421 st->flags |= SHTABLE_F_DONOTSTOP;
1422 }
1423
1424 /* disconnect all connected peers */
1425 for (ps = st->sessions; ps; ps = ps->next) {
1426 if (ps->session) {
1427 peer_session_forceshutdown(ps->session);
1428 ps->session = NULL;
1429 }
1430 }
1431 }
1432 ps = st->local_session;
1433
1434 if (ps->flags & PEER_F_TEACH_COMPLETE) {
1435 if (st->flags & SHTABLE_F_DONOTSTOP) {
1436 /* resync of new process was complete, current process can die now */
1437 jobs--;
1438 st->flags &= ~SHTABLE_F_DONOTSTOP;
1439 }
1440 }
1441 else if (!ps->session) {
1442 /* If session is not active */
1443 if (ps->statuscode == 0 ||
1444 ps->statuscode == PEER_SESSION_SUCCESSCODE ||
1445 ps->statuscode == PEER_SESSION_CONNECTEDCODE ||
1446 ps->statuscode == PEER_SESSION_TRYAGAIN) {
1447 /* connection never tried
1448 * or previous session was successfully established
1449 * or previous session tcp connect success but init state incomplete
1450 * or during previous connect, peer replies a try again statuscode */
1451
1452 /* connect to the peer */
1453 ps->session = peer_session_create(ps->peer, ps);
1454 }
1455 else {
1456 /* Other error cases */
1457 if (st->flags & SHTABLE_F_DONOTSTOP) {
1458 /* unable to resync new process, current process can die now */
1459 jobs--;
1460 st->flags &= ~SHTABLE_F_DONOTSTOP;
1461 }
1462 }
1463 }
1464 else if (ps->statuscode == PEER_SESSION_SUCCESSCODE &&
1465 (int)(ps->pushed - ps->table->table->localupdate) < 0) {
1466 /* current session active and established
1467 awake session to push remaining local updates */
1468 task_wakeup(ps->session->task, TASK_WOKEN_MSG);
1469 }
1470 } /* stopping */
1471 /* Wakeup for re-connect */
1472 return task;
1473}
1474
1475/*
1476 * Function used to register a table for sync on a group of peers
1477 *
1478 */
1479void peers_register_table(struct peers *peers, struct stktable *table)
1480{
1481 struct shared_table *st;
1482 struct peer * curpeer;
1483 struct peer_session *ps;
Willy Tarreau4348fad2012-09-20 16:48:07 +02001484 struct listener *listener;
Emeric Brun2b920a12010-09-23 18:30:22 +02001485
1486 st = (struct shared_table *)calloc(1,sizeof(struct shared_table));
1487 st->table = table;
1488 st->next = peers->tables;
1489 st->resync_timeout = tick_add(now_ms, MS_TO_TICKS(5000));
1490 peers->tables = st;
1491
1492 for (curpeer = peers->remote; curpeer; curpeer = curpeer->next) {
1493 ps = (struct peer_session *)calloc(1,sizeof(struct peer_session));
1494 ps->table = st;
1495 ps->peer = curpeer;
1496 if (curpeer->local)
1497 st->local_session = ps;
1498 ps->next = st->sessions;
1499 ps->reconnect = now_ms;
1500 st->sessions = ps;
1501 peers->peers_fe->maxconn += 3;
1502 }
1503
Willy Tarreau4348fad2012-09-20 16:48:07 +02001504 list_for_each_entry(listener, &peers->peers_fe->conf.listeners, by_fe)
1505 listener->maxconn = peers->peers_fe->maxconn;
Emeric Brun2b920a12010-09-23 18:30:22 +02001506 st->sync_task = task_new();
1507 st->sync_task->process = process_peer_sync;
1508 st->sync_task->expire = TICK_ETERNITY;
1509 st->sync_task->context = (void *)st;
1510 table->sync_task =st->sync_task;
1511 signal_register_task(0, table->sync_task, 0);
1512 task_wakeup(st->sync_task, TASK_WOKEN_INIT);
1513}
1514