blob: 1eada2d67044fe474f696f47cf0b0f80e15cee71 [file] [log] [blame]
Willy Tarreau59f98392012-07-06 14:13:49 +02001/*
2 * Connection management functions
3 *
4 * Copyright 2000-2012 Willy Tarreau <w@1wt.eu>
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 <common/compat.h>
14#include <common/config.h>
15
16#include <types/connection.h>
Willy Tarreau2c6be842012-07-06 17:12:34 +020017
Willy Tarreau2da156f2012-07-23 15:07:23 +020018#include <proto/proto_tcp.h>
Willy Tarreau2c6be842012-07-06 17:12:34 +020019#include <proto/stream_interface.h>
Willy Tarreau59f98392012-07-06 14:13:49 +020020
21/* I/O callback for fd-based connections. It calls the read/write handlers
Willy Tarreauafad0e02012-08-09 14:45:22 +020022 * provided by the connection's sock_ops, which must be valid. It returns 0.
Willy Tarreau59f98392012-07-06 14:13:49 +020023 */
24int conn_fd_handler(int fd)
25{
Willy Tarreau80184712012-07-06 14:54:49 +020026 struct connection *conn = fdtab[fd].owner;
Willy Tarreau59f98392012-07-06 14:13:49 +020027
Willy Tarreauc76ae332012-07-12 15:32:13 +020028 if (unlikely(!conn))
Willy Tarreau2c6be842012-07-06 17:12:34 +020029 goto leave;
Willy Tarreau59f98392012-07-06 14:13:49 +020030
Willy Tarreauc76ae332012-07-12 15:32:13 +020031 process_handshake:
Willy Tarreauf9dabec2012-08-17 17:33:53 +020032 /* The handshake callbacks are called in sequence. If either of them is
33 * missing something, it must enable the required polling at the socket
34 * layer of the connection. Polling state is not guaranteed when entering
35 * these handlers, so any handshake handler which does not complete its
36 * work must explicitly disable events it's not interested in.
37 */
Willy Tarreauc76ae332012-07-12 15:32:13 +020038 while (unlikely(conn->flags & CO_FL_HANDSHAKE)) {
39 if (unlikely(conn->flags & CO_FL_ERROR))
Willy Tarreau2c6be842012-07-06 17:12:34 +020040 goto leave;
Willy Tarreau59f98392012-07-06 14:13:49 +020041
Willy Tarreauc76ae332012-07-12 15:32:13 +020042 if (conn->flags & CO_FL_SI_SEND_PROXY)
Willy Tarreauafad0e02012-08-09 14:45:22 +020043 if (!conn_si_send_proxy(conn, CO_FL_SI_SEND_PROXY))
Willy Tarreauc76ae332012-07-12 15:32:13 +020044 goto leave;
45 }
46
Willy Tarreauf9dabec2012-08-17 17:33:53 +020047 /* Once we're purely in the data phase, we disable handshake polling */
48 if (!(conn->flags & CO_FL_POLL_SOCK))
49 __conn_sock_stop_both(conn);
Willy Tarreauc76ae332012-07-12 15:32:13 +020050
Willy Tarreau59f98392012-07-06 14:13:49 +020051 if (fdtab[fd].ev & (FD_POLL_IN | FD_POLL_HUP | FD_POLL_ERR))
Willy Tarreauafad0e02012-08-09 14:45:22 +020052 conn->data->read(conn);
Willy Tarreau59f98392012-07-06 14:13:49 +020053
Willy Tarreauc76ae332012-07-12 15:32:13 +020054 if (unlikely(conn->flags & CO_FL_ERROR))
Willy Tarreau2c6be842012-07-06 17:12:34 +020055 goto leave;
Willy Tarreau59f98392012-07-06 14:13:49 +020056
Willy Tarreauc76ae332012-07-12 15:32:13 +020057 /* It may happen during the data phase that a handshake is
58 * enabled again (eg: SSL)
59 */
60 if (unlikely(conn->flags & CO_FL_HANDSHAKE))
61 goto process_handshake;
62
Willy Tarreau59f98392012-07-06 14:13:49 +020063 if (fdtab[fd].ev & (FD_POLL_OUT | FD_POLL_ERR))
Willy Tarreauafad0e02012-08-09 14:45:22 +020064 conn->data->write(conn);
Willy Tarreau2da156f2012-07-23 15:07:23 +020065
Willy Tarreauc76ae332012-07-12 15:32:13 +020066 if (unlikely(conn->flags & CO_FL_ERROR))
Willy Tarreau2da156f2012-07-23 15:07:23 +020067 goto leave;
68
Willy Tarreauc76ae332012-07-12 15:32:13 +020069 /* It may happen during the data phase that a handshake is
70 * enabled again (eg: SSL)
71 */
72 if (unlikely(conn->flags & CO_FL_HANDSHAKE))
73 goto process_handshake;
74
75 if (unlikely(conn->flags & CO_FL_WAIT_L4_CONN)) {
Willy Tarreau2da156f2012-07-23 15:07:23 +020076 /* still waiting for a connection to establish and no data to
77 * send in order to probe it ? Then let's retry the connect().
78 */
Willy Tarreau239d7182012-07-23 18:53:03 +020079 if (!tcp_connect_probe(conn))
Willy Tarreauafad0e02012-08-09 14:45:22 +020080 goto leave;
Willy Tarreau2da156f2012-07-23 15:07:23 +020081 }
82
Willy Tarreau2c6be842012-07-06 17:12:34 +020083 leave:
Willy Tarreaufd31e532012-07-23 18:24:25 +020084 if (conn->flags & CO_FL_NOTIFY_SI)
Willy Tarreau100c4672012-08-20 12:06:26 +020085 conn_notify_si(conn);
Willy Tarreaufd31e532012-07-23 18:24:25 +020086
Willy Tarreau8f8c92f2012-07-23 19:45:44 +020087 /* Last check, verify if the connection just established */
Willy Tarreauc76ae332012-07-12 15:32:13 +020088 if (unlikely(!(conn->flags & (CO_FL_WAIT_L4_CONN | CO_FL_WAIT_L6_CONN | CO_FL_CONNECTED))))
Willy Tarreau8f8c92f2012-07-23 19:45:44 +020089 conn->flags |= CO_FL_CONNECTED;
90
Willy Tarreau61ace1b2012-07-23 12:14:26 +020091 /* remove the events before leaving */
92 fdtab[fd].ev &= ~(FD_POLL_IN | FD_POLL_OUT | FD_POLL_HUP | FD_POLL_ERR);
Willy Tarreauf9dabec2012-08-17 17:33:53 +020093
94 /* commit polling changes */
95 conn_cond_update_polling(conn);
Willy Tarreauafad0e02012-08-09 14:45:22 +020096 return 0;
Willy Tarreau59f98392012-07-06 14:13:49 +020097}
Willy Tarreaub5e2cbd2012-08-17 11:55:04 +020098
99/* set polling depending on the change between the CURR part of the
100 * flags and the new flags in connection C. The connection flags are
101 * updated with the new flags at the end of the operation. Only the bits
102 * relevant to CO_FL_CURR_* from <flags> are considered.
103 */
104void conn_set_polling(struct connection *c, unsigned int new)
105{
106 unsigned int old = c->flags; /* for CO_FL_CURR_* */
107
108 /* update read status if needed */
109 if ((old & (CO_FL_CURR_RD_ENA|CO_FL_CURR_RD_POL)) != (CO_FL_CURR_RD_ENA|CO_FL_CURR_RD_POL) &&
110 (new & (CO_FL_CURR_RD_ENA|CO_FL_CURR_RD_POL)) == (CO_FL_CURR_RD_ENA|CO_FL_CURR_RD_POL))
111 fd_poll_recv(c->t.sock.fd);
112 else if (!(old & CO_FL_CURR_RD_ENA) && (new & CO_FL_CURR_RD_ENA))
113 fd_want_recv(c->t.sock.fd);
114 else if ((old & CO_FL_CURR_RD_ENA) && !(new & CO_FL_CURR_RD_ENA))
115 fd_stop_recv(c->t.sock.fd);
116
117 /* update write status if needed */
118 if ((old & (CO_FL_CURR_WR_ENA|CO_FL_CURR_WR_POL)) != (CO_FL_CURR_WR_ENA|CO_FL_CURR_WR_POL) &&
119 (new & (CO_FL_CURR_WR_ENA|CO_FL_CURR_WR_POL)) == (CO_FL_CURR_WR_ENA|CO_FL_CURR_WR_POL))
120 fd_poll_send(c->t.sock.fd);
121 else if (!(old & CO_FL_CURR_WR_ENA) && (new & CO_FL_CURR_WR_ENA))
122 fd_want_send(c->t.sock.fd);
123 else if ((old & CO_FL_CURR_WR_ENA) && !(new & CO_FL_CURR_WR_ENA))
124 fd_stop_send(c->t.sock.fd);
125
126 c->flags &= ~(CO_FL_CURR_WR_POL|CO_FL_CURR_WR_ENA|CO_FL_CURR_RD_POL|CO_FL_CURR_RD_ENA);
127 c->flags |= new & (CO_FL_CURR_WR_POL|CO_FL_CURR_WR_ENA|CO_FL_CURR_RD_POL|CO_FL_CURR_RD_ENA);
128}