Willy Tarreau | cff6411 | 2008-11-03 06:26:53 +0100 | [diff] [blame] | 1 | /* |
Willy Tarreau | 8e89b84 | 2009-10-18 23:56:35 +0200 | [diff] [blame] | 2 | * include/proto/stream_interface.h |
| 3 | * This file contains stream_interface function prototypes |
| 4 | * |
Willy Tarreau | 2bb4a96 | 2014-11-28 11:11:05 +0100 | [diff] [blame] | 5 | * Copyright (C) 2000-2014 Willy Tarreau - w@1wt.eu |
Willy Tarreau | 8e89b84 | 2009-10-18 23:56:35 +0200 | [diff] [blame] | 6 | * |
| 7 | * This library is free software; you can redistribute it and/or |
| 8 | * modify it under the terms of the GNU Lesser General Public |
| 9 | * License as published by the Free Software Foundation, version 2.1 |
| 10 | * exclusively. |
| 11 | * |
| 12 | * This library is distributed in the hope that it will be useful, |
| 13 | * but WITHOUT ANY WARRANTY; without even the implied warranty of |
| 14 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
| 15 | * Lesser General Public License for more details. |
| 16 | * |
| 17 | * You should have received a copy of the GNU Lesser General Public |
| 18 | * License along with this library; if not, write to the Free Software |
| 19 | * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
| 20 | */ |
Willy Tarreau | cff6411 | 2008-11-03 06:26:53 +0100 | [diff] [blame] | 21 | |
| 22 | #ifndef _PROTO_STREAM_INTERFACE_H |
| 23 | #define _PROTO_STREAM_INTERFACE_H |
| 24 | |
| 25 | #include <stdlib.h> |
| 26 | |
| 27 | #include <common/config.h> |
Willy Tarreau | 2c6be84 | 2012-07-06 17:12:34 +0200 | [diff] [blame] | 28 | #include <types/session.h> |
Willy Tarreau | cff6411 | 2008-11-03 06:26:53 +0100 | [diff] [blame] | 29 | #include <types/stream_interface.h> |
Willy Tarreau | 14f8e86 | 2012-08-30 22:23:13 +0200 | [diff] [blame] | 30 | #include <proto/channel.h> |
Willy Tarreau | f9dabec | 2012-08-17 17:33:53 +0200 | [diff] [blame] | 31 | #include <proto/connection.h> |
Willy Tarreau | cff6411 | 2008-11-03 06:26:53 +0100 | [diff] [blame] | 32 | |
| 33 | |
| 34 | /* main event functions used to move data between sockets and buffers */ |
Willy Tarreau | 269358d | 2009-09-20 20:14:49 +0200 | [diff] [blame] | 35 | int stream_int_check_timeouts(struct stream_interface *si); |
Willy Tarreau | cff6411 | 2008-11-03 06:26:53 +0100 | [diff] [blame] | 36 | void stream_int_report_error(struct stream_interface *si); |
Willy Tarreau | dded32d | 2008-11-30 19:48:07 +0100 | [diff] [blame] | 37 | void stream_int_retnclose(struct stream_interface *si, const struct chunk *msg); |
Willy Tarreau | 2c6be84 | 2012-07-06 17:12:34 +0200 | [diff] [blame] | 38 | int conn_si_send_proxy(struct connection *conn, unsigned int flag); |
Willy Tarreau | 9bf9c14 | 2012-08-20 15:38:41 +0200 | [diff] [blame] | 39 | void stream_sock_read0(struct stream_interface *si); |
Willy Tarreau | cff6411 | 2008-11-03 06:26:53 +0100 | [diff] [blame] | 40 | |
Willy Tarreau | c578891 | 2012-08-24 18:12:41 +0200 | [diff] [blame] | 41 | extern struct si_ops si_embedded_ops; |
Willy Tarreau | c578891 | 2012-08-24 18:12:41 +0200 | [diff] [blame] | 42 | extern struct si_ops si_conn_ops; |
Willy Tarreau | 74beec3 | 2012-10-03 00:41:04 +0200 | [diff] [blame] | 43 | extern struct data_cb si_conn_cb; |
Willy Tarreau | 2737562 | 2013-12-17 00:00:28 +0100 | [diff] [blame] | 44 | extern struct data_cb si_idle_conn_cb; |
Willy Tarreau | 5c979a9 | 2012-05-07 17:15:39 +0200 | [diff] [blame] | 45 | |
Willy Tarreau | 1fbe1c9 | 2013-12-01 09:35:41 +0100 | [diff] [blame] | 46 | struct appctx *stream_int_register_handler(struct stream_interface *si, struct si_applet *app); |
Willy Tarreau | fb90d94 | 2009-09-05 20:57:35 +0200 | [diff] [blame] | 47 | void stream_int_unregister_handler(struct stream_interface *si); |
| 48 | |
Willy Tarreau | 2bb4a96 | 2014-11-28 11:11:05 +0100 | [diff] [blame] | 49 | /* returns the channel which receives data from this stream interface (input channel) */ |
| 50 | static inline struct channel *si_ic(struct stream_interface *si) |
| 51 | { |
Willy Tarreau | 0b2fb7f | 2014-11-28 11:50:38 +0100 | [diff] [blame] | 52 | if (si->flags & SI_FL_ISBACK) |
| 53 | return &LIST_ELEM(si, struct session *, si[1])->res; |
| 54 | else |
| 55 | return &LIST_ELEM(si, struct session *, si[0])->req; |
Willy Tarreau | 2bb4a96 | 2014-11-28 11:11:05 +0100 | [diff] [blame] | 56 | } |
| 57 | |
| 58 | /* returns the channel which feeds data to this stream interface (output channel) */ |
| 59 | static inline struct channel *si_oc(struct stream_interface *si) |
| 60 | { |
Willy Tarreau | 0b2fb7f | 2014-11-28 11:50:38 +0100 | [diff] [blame] | 61 | if (si->flags & SI_FL_ISBACK) |
| 62 | return &LIST_ELEM(si, struct session *, si[1])->req; |
| 63 | else |
| 64 | return &LIST_ELEM(si, struct session *, si[0])->res; |
Willy Tarreau | 2bb4a96 | 2014-11-28 11:11:05 +0100 | [diff] [blame] | 65 | } |
| 66 | |
Willy Tarreau | aefd790 | 2014-11-28 12:03:32 +0100 | [diff] [blame] | 67 | /* returns the session associated to a stream interface */ |
| 68 | static inline struct session *si_sess(struct stream_interface *si) |
| 69 | { |
| 70 | if (si->flags & SI_FL_ISBACK) |
| 71 | return LIST_ELEM(si, struct session *, si[1]); |
| 72 | else |
| 73 | return LIST_ELEM(si, struct session *, si[0]); |
| 74 | } |
| 75 | |
Willy Tarreau | 07373b8 | 2014-11-28 12:08:47 +0100 | [diff] [blame] | 76 | /* returns the task associated to this stream interface */ |
| 77 | static inline struct task *si_task(struct stream_interface *si) |
| 78 | { |
| 79 | if (si->flags & SI_FL_ISBACK) |
| 80 | return LIST_ELEM(si, struct session *, si[1])->task; |
| 81 | else |
| 82 | return LIST_ELEM(si, struct session *, si[0])->task; |
| 83 | } |
| 84 | |
Willy Tarreau | 0a23bcb | 2013-12-01 11:31:38 +0100 | [diff] [blame] | 85 | /* Initializes all required fields for a new appctx. Note that it does the |
| 86 | * minimum acceptable initialization for an appctx. This means only the |
| 87 | * 3 integer states st0, st1, st2 are zeroed. |
| 88 | */ |
| 89 | static inline void appctx_init(struct appctx *appctx) |
| 90 | { |
| 91 | appctx->st0 = appctx->st1 = appctx->st2 = 0; |
| 92 | } |
| 93 | |
| 94 | /* sets <appctx>'s applet to point to <applet> */ |
| 95 | static inline void appctx_set_applet(struct appctx *appctx, struct si_applet *applet) |
| 96 | { |
| 97 | appctx->applet = applet; |
| 98 | } |
| 99 | |
| 100 | /* Tries to allocate a new appctx and initialize its main fields. The |
| 101 | * appctx is returned on success, NULL on failure. The appctx must be |
| 102 | * released using pool_free2(connection) or appctx_free(), since it's |
| 103 | * allocated from the connection pool. |
| 104 | */ |
| 105 | static inline struct appctx *appctx_new() |
| 106 | { |
| 107 | struct appctx *appctx; |
| 108 | |
| 109 | appctx = pool_alloc2(pool2_connection); |
| 110 | if (likely(appctx != NULL)) { |
| 111 | appctx->obj_type = OBJ_TYPE_APPCTX; |
| 112 | appctx->applet = NULL; |
| 113 | appctx_init(appctx); |
| 114 | } |
| 115 | return appctx; |
| 116 | } |
| 117 | |
| 118 | /* Releases an appctx previously allocated by appctx_new(). Note that |
| 119 | * we share the connection pool. |
| 120 | */ |
| 121 | static inline void appctx_free(struct appctx *appctx) |
| 122 | { |
| 123 | pool_free2(pool2_connection, appctx); |
| 124 | } |
| 125 | |
Willy Tarreau | 3ed35ef | 2013-10-24 11:51:38 +0200 | [diff] [blame] | 126 | /* initializes a stream interface in the SI_ST_INI state. It's detached from |
Willy Tarreau | 819d332 | 2014-11-28 12:12:34 +0100 | [diff] [blame^] | 127 | * any endpoint and only keeps its side which is expected to have already been |
| 128 | * set. |
Willy Tarreau | 3ed35ef | 2013-10-24 11:51:38 +0200 | [diff] [blame] | 129 | */ |
Willy Tarreau | 819d332 | 2014-11-28 12:12:34 +0100 | [diff] [blame^] | 130 | static inline void si_reset(struct stream_interface *si) |
Willy Tarreau | 3ed35ef | 2013-10-24 11:51:38 +0200 | [diff] [blame] | 131 | { |
Willy Tarreau | 3ed35ef | 2013-10-24 11:51:38 +0200 | [diff] [blame] | 132 | si->err_type = SI_ET_NONE; |
| 133 | si->conn_retries = 0; /* used for logging too */ |
Willy Tarreau | 3ed35ef | 2013-10-24 11:51:38 +0200 | [diff] [blame] | 134 | si->exp = TICK_ETERNITY; |
Willy Tarreau | a5f5d8d | 2014-11-28 11:26:07 +0100 | [diff] [blame] | 135 | si->flags &= SI_FL_ISBACK; |
Willy Tarreau | 3ed35ef | 2013-10-24 11:51:38 +0200 | [diff] [blame] | 136 | si->end = NULL; |
| 137 | si->state = si->prev_state = SI_ST_INI; |
| 138 | } |
| 139 | |
| 140 | /* sets the current and previous state of a stream interface to <state>. This |
| 141 | * is mainly used to create one in the established state on incoming |
| 142 | * conncetions. |
| 143 | */ |
| 144 | static inline void si_set_state(struct stream_interface *si, int state) |
| 145 | { |
| 146 | si->state = si->prev_state = state; |
| 147 | } |
| 148 | |
Willy Tarreau | 0a23bcb | 2013-12-01 11:31:38 +0100 | [diff] [blame] | 149 | /* Release the endpoint if it's a connection or an applet, then nullify it. |
| 150 | * Note: released connections are closed then freed. |
| 151 | */ |
Willy Tarreau | 32e3c6a | 2013-10-11 19:34:20 +0200 | [diff] [blame] | 152 | static inline void si_release_endpoint(struct stream_interface *si) |
Willy Tarreau | 372d670 | 2013-09-29 17:19:56 +0200 | [diff] [blame] | 153 | { |
Willy Tarreau | 32e3c6a | 2013-10-11 19:34:20 +0200 | [diff] [blame] | 154 | struct connection *conn; |
Willy Tarreau | 0a23bcb | 2013-12-01 11:31:38 +0100 | [diff] [blame] | 155 | struct appctx *appctx; |
Willy Tarreau | 32e3c6a | 2013-10-11 19:34:20 +0200 | [diff] [blame] | 156 | |
Willy Tarreau | 0a23bcb | 2013-12-01 11:31:38 +0100 | [diff] [blame] | 157 | if (!si->end) |
| 158 | return; |
| 159 | |
| 160 | if ((conn = objt_conn(si->end))) { |
| 161 | conn_force_close(conn); |
| 162 | conn_free(conn); |
| 163 | } |
| 164 | else if ((appctx = objt_appctx(si->end))) { |
| 165 | if (appctx->applet->release) |
| 166 | appctx->applet->release(si); |
| 167 | appctx_free(appctx); /* we share the connection pool */ |
| 168 | } |
Willy Tarreau | 7d67d7b | 2013-09-29 16:05:22 +0200 | [diff] [blame] | 169 | si->end = NULL; |
Willy Tarreau | 372d670 | 2013-09-29 17:19:56 +0200 | [diff] [blame] | 170 | } |
| 171 | |
Willy Tarreau | 32e3c6a | 2013-10-11 19:34:20 +0200 | [diff] [blame] | 172 | static inline void si_detach(struct stream_interface *si) |
| 173 | { |
| 174 | si_release_endpoint(si); |
| 175 | si->ops = &si_embedded_ops; |
| 176 | } |
| 177 | |
Willy Tarreau | 2737562 | 2013-12-17 00:00:28 +0100 | [diff] [blame] | 178 | /* Turn a possibly existing connection endpoint of stream interface <si> to |
| 179 | * idle mode, which means that the connection will be polled for incoming events |
| 180 | * and might be killed by the underlying I/O handler. |
| 181 | */ |
| 182 | static inline void si_idle_conn(struct stream_interface *si) |
| 183 | { |
| 184 | struct connection *conn = objt_conn(si->end); |
| 185 | |
| 186 | if (!conn) |
| 187 | return; |
| 188 | |
| 189 | conn_attach(conn, si, &si_idle_conn_cb); |
| 190 | conn_data_want_recv(conn); |
| 191 | } |
| 192 | |
Willy Tarreau | 2a6e880 | 2013-10-24 15:50:53 +0200 | [diff] [blame] | 193 | /* Attach connection <conn> to the stream interface <si>. The stream interface |
| 194 | * is configured to work with a connection and the connection it configured |
| 195 | * with a stream interface data layer. |
Willy Tarreau | b363a1f | 2013-10-01 10:45:07 +0200 | [diff] [blame] | 196 | */ |
Willy Tarreau | 2a6e880 | 2013-10-24 15:50:53 +0200 | [diff] [blame] | 197 | static inline void si_attach_conn(struct stream_interface *si, struct connection *conn) |
Willy Tarreau | bd99aab | 2012-10-02 20:57:19 +0200 | [diff] [blame] | 198 | { |
| 199 | si->ops = &si_conn_ops; |
Willy Tarreau | b363a1f | 2013-10-01 10:45:07 +0200 | [diff] [blame] | 200 | si->end = &conn->obj_type; |
Willy Tarreau | 7abddb5 | 2013-10-24 15:31:04 +0200 | [diff] [blame] | 201 | conn_attach(conn, si, &si_conn_cb); |
Willy Tarreau | bd99aab | 2012-10-02 20:57:19 +0200 | [diff] [blame] | 202 | } |
| 203 | |
Willy Tarreau | d02cdd2 | 2013-12-15 10:23:20 +0100 | [diff] [blame] | 204 | /* Returns true if a connection is attached to the stream interface <si> and |
| 205 | * if this connection is ready. |
| 206 | */ |
| 207 | static inline int si_conn_ready(struct stream_interface *si) |
| 208 | { |
| 209 | struct connection *conn = objt_conn(si->end); |
| 210 | |
| 211 | return conn && conn_ctrl_ready(conn) && conn_xprt_ready(conn); |
| 212 | } |
| 213 | |
Willy Tarreau | 0a23bcb | 2013-12-01 11:31:38 +0100 | [diff] [blame] | 214 | /* Attach appctx <appctx> to the stream interface <si>. The stream interface |
| 215 | * is configured to work with an applet context. It is left to the caller to |
| 216 | * call appctx_set_applet() to assign an applet to this context. |
| 217 | */ |
| 218 | static inline void si_attach_appctx(struct stream_interface *si, struct appctx *appctx) |
Willy Tarreau | c578891 | 2012-08-24 18:12:41 +0200 | [diff] [blame] | 219 | { |
| 220 | si->ops = &si_embedded_ops; |
Willy Tarreau | 0a23bcb | 2013-12-01 11:31:38 +0100 | [diff] [blame] | 221 | appctx->obj_type = OBJ_TYPE_APPCTX; |
| 222 | si->end = &appctx->obj_type; |
Willy Tarreau | c578891 | 2012-08-24 18:12:41 +0200 | [diff] [blame] | 223 | } |
| 224 | |
Willy Tarreau | 7b4b499 | 2013-12-01 09:15:12 +0100 | [diff] [blame] | 225 | /* returns a pointer to the appctx being run in the SI or NULL if none */ |
| 226 | static inline struct appctx *si_appctx(struct stream_interface *si) |
| 227 | { |
| 228 | return objt_appctx(si->end); |
| 229 | } |
| 230 | |
Willy Tarreau | 4a59f2f | 2013-10-24 20:10:45 +0200 | [diff] [blame] | 231 | /* returns a pointer to the applet being run in the SI or NULL if none */ |
| 232 | static inline const struct si_applet *si_applet(struct stream_interface *si) |
| 233 | { |
Willy Tarreau | 7b4b499 | 2013-12-01 09:15:12 +0100 | [diff] [blame] | 234 | const struct appctx *appctx; |
| 235 | |
| 236 | appctx = si_appctx(si); |
| 237 | if (appctx) |
| 238 | return appctx->applet; |
Willy Tarreau | cf644ed | 2013-09-29 17:19:56 +0200 | [diff] [blame] | 239 | return NULL; |
| 240 | } |
| 241 | |
| 242 | /* Call the applet's main function when an appctx is attached to the stream |
| 243 | * interface. Returns zero if no call was made, or non-zero if a call was made. |
| 244 | */ |
| 245 | static inline int si_applet_call(struct stream_interface *si) |
| 246 | { |
| 247 | const struct si_applet *applet; |
| 248 | |
| 249 | applet = si_applet(si); |
| 250 | if (applet) { |
| 251 | applet->fct(si); |
| 252 | return 1; |
| 253 | } |
| 254 | return 0; |
Willy Tarreau | 4a59f2f | 2013-10-24 20:10:45 +0200 | [diff] [blame] | 255 | } |
| 256 | |
| 257 | /* call the applet's release function if any. Needs to be called upon close() */ |
| 258 | static inline void si_applet_release(struct stream_interface *si) |
| 259 | { |
| 260 | const struct si_applet *applet; |
| 261 | |
| 262 | applet = si_applet(si); |
| 263 | if (applet && applet->release) |
| 264 | applet->release(si); |
| 265 | } |
| 266 | |
Willy Tarreau | 9471b8c | 2013-12-15 13:31:35 +0100 | [diff] [blame] | 267 | /* Try to allocate a new connection and assign it to the interface. If |
| 268 | * a connection was previously allocated and the <reuse> flag is set, |
| 269 | * it is returned unmodified. Otherwise it is reset. |
| 270 | */ |
Willy Tarreau | 32e3c6a | 2013-10-11 19:34:20 +0200 | [diff] [blame] | 271 | /* Returns the stream interface's existing connection if one such already |
| 272 | * exists, or tries to allocate and initialize a new one which is then |
| 273 | * assigned to the stream interface. |
| 274 | */ |
Willy Tarreau | 9471b8c | 2013-12-15 13:31:35 +0100 | [diff] [blame] | 275 | static inline struct connection *si_alloc_conn(struct stream_interface *si, int reuse) |
Willy Tarreau | 32e3c6a | 2013-10-11 19:34:20 +0200 | [diff] [blame] | 276 | { |
| 277 | struct connection *conn; |
| 278 | |
Willy Tarreau | 0a23bcb | 2013-12-01 11:31:38 +0100 | [diff] [blame] | 279 | /* If we find a connection, we return it, otherwise it's an applet |
| 280 | * and we start by releasing it. |
Willy Tarreau | 32e3c6a | 2013-10-11 19:34:20 +0200 | [diff] [blame] | 281 | */ |
Willy Tarreau | 0a23bcb | 2013-12-01 11:31:38 +0100 | [diff] [blame] | 282 | if (si->end) { |
| 283 | conn = objt_conn(si->end); |
Willy Tarreau | 9471b8c | 2013-12-15 13:31:35 +0100 | [diff] [blame] | 284 | if (conn) { |
| 285 | if (!reuse) { |
| 286 | conn_force_close(conn); |
| 287 | conn_init(conn); |
| 288 | } |
Willy Tarreau | 0a23bcb | 2013-12-01 11:31:38 +0100 | [diff] [blame] | 289 | return conn; |
Willy Tarreau | 9471b8c | 2013-12-15 13:31:35 +0100 | [diff] [blame] | 290 | } |
Willy Tarreau | 0a23bcb | 2013-12-01 11:31:38 +0100 | [diff] [blame] | 291 | /* it was an applet then */ |
| 292 | si_release_endpoint(si); |
| 293 | } |
Willy Tarreau | 32e3c6a | 2013-10-11 19:34:20 +0200 | [diff] [blame] | 294 | |
| 295 | conn = conn_new(); |
| 296 | if (conn) |
| 297 | si_attach_conn(si, conn); |
| 298 | |
| 299 | return conn; |
| 300 | } |
| 301 | |
Willy Tarreau | 0a23bcb | 2013-12-01 11:31:38 +0100 | [diff] [blame] | 302 | /* Release the interface's existing endpoint (connection or appctx) and |
| 303 | * allocate then initialize a new appctx which is assigned to the interface |
| 304 | * and returned. NULL may be returned upon memory shortage. It is left to the |
| 305 | * caller to call appctx_set_applet() to assign an applet to this context. |
| 306 | */ |
| 307 | static inline struct appctx *si_alloc_appctx(struct stream_interface *si) |
| 308 | { |
| 309 | struct appctx *appctx; |
| 310 | |
| 311 | si_release_endpoint(si); |
| 312 | appctx = appctx_new(); |
| 313 | if (appctx) |
| 314 | si_attach_appctx(si, appctx); |
| 315 | |
| 316 | return appctx; |
| 317 | } |
| 318 | |
Willy Tarreau | 73b013b | 2012-05-21 16:31:45 +0200 | [diff] [blame] | 319 | /* Sends a shutr to the connection using the data layer */ |
| 320 | static inline void si_shutr(struct stream_interface *si) |
| 321 | { |
Willy Tarreau | 6fe1541 | 2013-09-29 15:16:03 +0200 | [diff] [blame] | 322 | si->ops->shutr(si); |
Willy Tarreau | 73b013b | 2012-05-21 16:31:45 +0200 | [diff] [blame] | 323 | } |
| 324 | |
| 325 | /* Sends a shutw to the connection using the data layer */ |
| 326 | static inline void si_shutw(struct stream_interface *si) |
| 327 | { |
Willy Tarreau | 6fe1541 | 2013-09-29 15:16:03 +0200 | [diff] [blame] | 328 | si->ops->shutw(si); |
Willy Tarreau | 73b013b | 2012-05-21 16:31:45 +0200 | [diff] [blame] | 329 | } |
| 330 | |
| 331 | /* Calls the data state update on the stream interfaace */ |
| 332 | static inline void si_update(struct stream_interface *si) |
| 333 | { |
Willy Tarreau | c578891 | 2012-08-24 18:12:41 +0200 | [diff] [blame] | 334 | si->ops->update(si); |
Willy Tarreau | 73b013b | 2012-05-21 16:31:45 +0200 | [diff] [blame] | 335 | } |
| 336 | |
| 337 | /* Calls chk_rcv on the connection using the data layer */ |
| 338 | static inline void si_chk_rcv(struct stream_interface *si) |
| 339 | { |
Willy Tarreau | c578891 | 2012-08-24 18:12:41 +0200 | [diff] [blame] | 340 | si->ops->chk_rcv(si); |
Willy Tarreau | 73b013b | 2012-05-21 16:31:45 +0200 | [diff] [blame] | 341 | } |
| 342 | |
| 343 | /* Calls chk_snd on the connection using the data layer */ |
| 344 | static inline void si_chk_snd(struct stream_interface *si) |
| 345 | { |
Willy Tarreau | c578891 | 2012-08-24 18:12:41 +0200 | [diff] [blame] | 346 | si->ops->chk_snd(si); |
Willy Tarreau | 73b013b | 2012-05-21 16:31:45 +0200 | [diff] [blame] | 347 | } |
| 348 | |
| 349 | /* Calls chk_snd on the connection using the ctrl layer */ |
| 350 | static inline int si_connect(struct stream_interface *si) |
| 351 | { |
Willy Tarreau | b363a1f | 2013-10-01 10:45:07 +0200 | [diff] [blame] | 352 | struct connection *conn = objt_conn(si->end); |
Willy Tarreau | b490b4e | 2013-12-15 16:20:50 +0100 | [diff] [blame] | 353 | int ret = SN_ERR_NONE; |
Willy Tarreau | 14f8e86 | 2012-08-30 22:23:13 +0200 | [diff] [blame] | 354 | |
Willy Tarreau | b363a1f | 2013-10-01 10:45:07 +0200 | [diff] [blame] | 355 | if (unlikely(!conn || !conn->ctrl || !conn->ctrl->connect)) |
Willy Tarreau | 73b013b | 2012-05-21 16:31:45 +0200 | [diff] [blame] | 356 | return SN_ERR_INTERNAL; |
Willy Tarreau | 14f8e86 | 2012-08-30 22:23:13 +0200 | [diff] [blame] | 357 | |
Willy Tarreau | b490b4e | 2013-12-15 16:20:50 +0100 | [diff] [blame] | 358 | if (!conn_ctrl_ready(conn) || !conn_xprt_ready(conn)) { |
Willy Tarreau | 2bb4a96 | 2014-11-28 11:11:05 +0100 | [diff] [blame] | 359 | ret = conn->ctrl->connect(conn, !channel_is_empty(si_oc(si)), 0); |
Willy Tarreau | b490b4e | 2013-12-15 16:20:50 +0100 | [diff] [blame] | 360 | if (ret != SN_ERR_NONE) |
| 361 | return ret; |
Willy Tarreau | 9e5a3aa | 2013-12-31 23:32:12 +0100 | [diff] [blame] | 362 | |
| 363 | /* we need to be notified about connection establishment */ |
| 364 | conn->flags |= CO_FL_WAKE_DATA; |
| 365 | |
| 366 | /* we're in the process of establishing a connection */ |
| 367 | si->state = SI_ST_CON; |
Willy Tarreau | b490b4e | 2013-12-15 16:20:50 +0100 | [diff] [blame] | 368 | } |
Willy Tarreau | 2bb4a96 | 2014-11-28 11:11:05 +0100 | [diff] [blame] | 369 | else if (!channel_is_empty(si_oc(si))) { |
Willy Tarreau | b490b4e | 2013-12-15 16:20:50 +0100 | [diff] [blame] | 370 | /* reuse the existing connection, we'll have to send a |
| 371 | * request there. |
| 372 | */ |
| 373 | conn_data_want_send(conn); |
Willy Tarreau | 9e5a3aa | 2013-12-31 23:32:12 +0100 | [diff] [blame] | 374 | |
| 375 | /* the connection is established */ |
| 376 | si->state = SI_ST_EST; |
Willy Tarreau | b490b4e | 2013-12-15 16:20:50 +0100 | [diff] [blame] | 377 | } |
Willy Tarreau | 14f8e86 | 2012-08-30 22:23:13 +0200 | [diff] [blame] | 378 | |
| 379 | /* needs src ip/port for logging */ |
| 380 | if (si->flags & SI_FL_SRC_ADDR) |
Willy Tarreau | b363a1f | 2013-10-01 10:45:07 +0200 | [diff] [blame] | 381 | conn_get_from_addr(conn); |
Willy Tarreau | 14f8e86 | 2012-08-30 22:23:13 +0200 | [diff] [blame] | 382 | |
Willy Tarreau | 14f8e86 | 2012-08-30 22:23:13 +0200 | [diff] [blame] | 383 | return ret; |
Willy Tarreau | 73b013b | 2012-05-21 16:31:45 +0200 | [diff] [blame] | 384 | } |
Willy Tarreau | 59b9479 | 2012-05-11 16:16:40 +0200 | [diff] [blame] | 385 | |
Willy Tarreau | 55e4ecd | 2012-12-08 17:48:47 +0100 | [diff] [blame] | 386 | /* for debugging, reports the stream interface state name */ |
| 387 | static inline const char *si_state_str(int state) |
| 388 | { |
| 389 | switch (state) { |
| 390 | case SI_ST_INI: return "INI"; |
| 391 | case SI_ST_REQ: return "REQ"; |
| 392 | case SI_ST_QUE: return "QUE"; |
| 393 | case SI_ST_TAR: return "TAR"; |
| 394 | case SI_ST_ASS: return "ASS"; |
| 395 | case SI_ST_CON: return "CON"; |
| 396 | case SI_ST_CER: return "CER"; |
| 397 | case SI_ST_EST: return "EST"; |
| 398 | case SI_ST_DIS: return "DIS"; |
| 399 | case SI_ST_CLO: return "CLO"; |
| 400 | default: return "???"; |
| 401 | } |
| 402 | } |
| 403 | |
Willy Tarreau | cff6411 | 2008-11-03 06:26:53 +0100 | [diff] [blame] | 404 | #endif /* _PROTO_STREAM_INTERFACE_H */ |
| 405 | |
| 406 | /* |
| 407 | * Local variables: |
| 408 | * c-indent-level: 8 |
| 409 | * c-basic-offset: 8 |
| 410 | * End: |
| 411 | */ |