Willy Tarreau | 56e9c5e | 2012-07-06 09:47:57 +0200 | [diff] [blame] | 1 | /* |
| 2 | * include/types/connection.h |
| 3 | * This file describes the connection struct and associated constants. |
| 4 | * |
| 5 | * Copyright (C) 2000-2012 Willy Tarreau - w@1wt.eu |
| 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 | */ |
| 21 | |
| 22 | #ifndef _TYPES_CONNECTION_H |
| 23 | #define _TYPES_CONNECTION_H |
| 24 | |
| 25 | #include <stdlib.h> |
| 26 | #include <sys/socket.h> |
| 27 | |
| 28 | #include <common/config.h> |
| 29 | |
| 30 | /* referenced below */ |
Willy Tarreau | 56e9c5e | 2012-07-06 09:47:57 +0200 | [diff] [blame] | 31 | struct protocol; |
Willy Tarreau | c578891 | 2012-08-24 18:12:41 +0200 | [diff] [blame] | 32 | struct connection; |
| 33 | struct buffer; |
| 34 | struct pipe; |
Willy Tarreau | 56e9c5e | 2012-07-06 09:47:57 +0200 | [diff] [blame] | 35 | |
Willy Tarreau | b5e2cbd | 2012-08-17 11:55:04 +0200 | [diff] [blame] | 36 | /* Polling flags that are manipulated by I/O callbacks and handshake callbacks |
| 37 | * indicate what they expect from a file descriptor at each layer. For each |
| 38 | * direction, we have 2 bits, one stating whether any suspected activity on the |
| 39 | * FD induce a call to the iocb, and another one indicating that the FD has |
| 40 | * already returned EAGAIN and that polling on it is essential before calling |
| 41 | * the iocb again : |
| 42 | * POL ENA state |
| 43 | * 0 0 STOPPED : any activity on this FD is ignored |
| 44 | * 0 1 ENABLED : any (suspected) activity may call the iocb |
| 45 | * 1 0 STOPPED : as above |
| 46 | * 1 1 POLLED : the FD is being polled for activity |
| 47 | * |
| 48 | * - Enabling an I/O event consists in ORing with 1. |
| 49 | * - Stopping an I/O event consists in ANDing with ~1. |
| 50 | * - Polling for an I/O event consists in ORing with ~3. |
| 51 | * |
| 52 | * The last computed state is remembered in CO_FL_CURR_* so that differential |
| 53 | * changes can be applied. For pollers that do not support speculative I/O, |
| 54 | * POLLED is the same as ENABLED and the POL flag can safely be ignored. |
| 55 | */ |
| 56 | |
Willy Tarreau | 900bc93 | 2012-07-06 09:52:14 +0200 | [diff] [blame] | 57 | /* flags for use in connection->flags */ |
| 58 | enum { |
| 59 | CO_FL_NONE = 0x00000000, |
| 60 | CO_FL_ERROR = 0x00000001, /* a fatal error was reported */ |
Willy Tarreau | c76ae33 | 2012-07-12 15:32:13 +0200 | [diff] [blame] | 61 | CO_FL_CONNECTED = 0x00000002, /* the connection is now established */ |
| 62 | CO_FL_WAIT_L4_CONN = 0x00000004, /* waiting for L4 to be connected */ |
| 63 | CO_FL_WAIT_L6_CONN = 0x00000008, /* waiting for L6 to be connected (eg: SSL) */ |
| 64 | |
| 65 | CO_FL_NOTIFY_SI = 0x00000010, /* notify stream interface about changes */ |
| 66 | |
Willy Tarreau | 2c6be84 | 2012-07-06 17:12:34 +0200 | [diff] [blame] | 67 | /* flags below are used for connection handshakes */ |
Willy Tarreau | c76ae33 | 2012-07-12 15:32:13 +0200 | [diff] [blame] | 68 | CO_FL_SI_SEND_PROXY = 0x00000020, /* send a valid PROXY protocol header */ |
| 69 | |
| 70 | /* below we have all handshake flags grouped into one */ |
| 71 | CO_FL_HANDSHAKE = CO_FL_SI_SEND_PROXY, |
Willy Tarreau | b5e2cbd | 2012-08-17 11:55:04 +0200 | [diff] [blame] | 72 | |
| 73 | /* when any of these flags is set, polling is defined by socket-layer |
| 74 | * operations, as opposed to data-layer. |
| 75 | */ |
| 76 | CO_FL_POLL_SOCK = CO_FL_HANDSHAKE | CO_FL_WAIT_L4_CONN | CO_FL_WAIT_L6_CONN, |
| 77 | |
Willy Tarreau | 2ba4465 | 2012-08-20 17:30:32 +0200 | [diff] [blame] | 78 | /* These flags are used by data layers to indicate to their iterators |
| 79 | * whether they had to stop due to missing data or missing room. Their |
| 80 | * callers must reset them before calling the data layer handlers. |
| 81 | */ |
| 82 | CO_FL_WAIT_DATA = 0x00004000, /* data source is empty */ |
| 83 | CO_FL_WAIT_ROOM = 0x00008000, /* data sink is full */ |
| 84 | |
Willy Tarreau | b5e2cbd | 2012-08-17 11:55:04 +0200 | [diff] [blame] | 85 | /* flags used to remember what shutdown have been performed/reported */ |
| 86 | CO_FL_DATA_RD_SH = 0x00010000, /* DATA layer was notified about shutr/read0 */ |
| 87 | CO_FL_DATA_WR_SH = 0x00020000, /* DATA layer asked for shutw */ |
| 88 | CO_FL_SOCK_RD_SH = 0x00040000, /* SOCK layer was notified about shutr/read0 */ |
| 89 | CO_FL_SOCK_WR_SH = 0x00080000, /* SOCK layer asked for shutw */ |
| 90 | |
| 91 | /****** NOTE: do not change the values of the flags below ******/ |
| 92 | CO_FL_RD_ENA = 1, CO_FL_RD_POL = 2, CO_FL_WR_ENA = 4, CO_FL_WR_POL = 8, |
| 93 | |
| 94 | /* flags describing the DATA layer expectations regarding polling */ |
| 95 | CO_FL_DATA_RD_ENA = CO_FL_RD_ENA << 20, /* receiving is allowed */ |
| 96 | CO_FL_DATA_RD_POL = CO_FL_RD_POL << 20, /* receiving needs to poll first */ |
| 97 | CO_FL_DATA_WR_ENA = CO_FL_WR_ENA << 20, /* sending is desired */ |
| 98 | CO_FL_DATA_WR_POL = CO_FL_WR_POL << 20, /* sending needs to poll first */ |
| 99 | |
| 100 | /* flags describing the SOCK layer expectations regarding polling */ |
| 101 | CO_FL_SOCK_RD_ENA = CO_FL_RD_ENA << 24, /* receiving is allowed */ |
| 102 | CO_FL_SOCK_RD_POL = CO_FL_RD_POL << 24, /* receiving needs to poll first */ |
| 103 | CO_FL_SOCK_WR_ENA = CO_FL_WR_ENA << 24, /* sending is desired */ |
| 104 | CO_FL_SOCK_WR_POL = CO_FL_WR_POL << 24, /* sending needs to poll first */ |
| 105 | |
| 106 | /* flags storing the current polling state */ |
| 107 | CO_FL_CURR_RD_ENA = CO_FL_RD_ENA << 28, /* receiving is allowed */ |
| 108 | CO_FL_CURR_RD_POL = CO_FL_RD_POL << 28, /* receiving needs to poll first */ |
| 109 | CO_FL_CURR_WR_ENA = CO_FL_WR_ENA << 28, /* sending is desired */ |
| 110 | CO_FL_CURR_WR_POL = CO_FL_WR_POL << 28, /* sending needs to poll first */ |
Willy Tarreau | 900bc93 | 2012-07-06 09:52:14 +0200 | [diff] [blame] | 111 | }; |
| 112 | |
Willy Tarreau | c578891 | 2012-08-24 18:12:41 +0200 | [diff] [blame] | 113 | |
| 114 | /* data_ops describes data-layer operations for a connection. They generally |
| 115 | * run over a socket-based control layer, but not always. |
| 116 | */ |
| 117 | struct data_ops { |
| 118 | int (*rcv_buf)(struct connection *conn, struct buffer *buf, int count); /* recv callback */ |
| 119 | int (*snd_buf)(struct connection *conn, struct buffer *buf, int flags); /* send callback */ |
| 120 | int (*rcv_pipe)(struct connection *conn, struct pipe *pipe, unsigned int count); /* recv-to-pipe callback */ |
| 121 | int (*snd_pipe)(struct connection *conn, struct pipe *pipe); /* send-to-pipe callback */ |
| 122 | void (*shutr)(struct connection *, int); /* shutr function */ |
| 123 | void (*shutw)(struct connection *, int); /* shutw function */ |
| 124 | void (*close)(struct connection *); /* close the data channel on the connection */ |
| 125 | }; |
| 126 | |
| 127 | /* app_cb describes read and write callbacks which are called upon detected I/O |
| 128 | * activity at the data layer. These callbacks are supposed to make use of the |
| 129 | * data_ops above to exchange data from/to buffers and pipes. |
| 130 | */ |
| 131 | struct app_cb { |
| 132 | void (*recv)(struct connection *conn); /* application-layer recv callback */ |
| 133 | void (*send)(struct connection *conn); /* application-layer send callback */ |
| 134 | }; |
| 135 | |
Willy Tarreau | 56e9c5e | 2012-07-06 09:47:57 +0200 | [diff] [blame] | 136 | /* This structure describes a connection with its methods and data. |
| 137 | * A connection may be performed to proxy or server via a local or remote |
| 138 | * socket, and can also be made to an internal applet. It can support |
| 139 | * several data schemes (applet, raw, ssl, ...). It can support several |
| 140 | * connection control schemes, generally a protocol for socket-oriented |
| 141 | * connections, but other methods for applets. |
| 142 | */ |
| 143 | struct connection { |
Willy Tarreau | c578891 | 2012-08-24 18:12:41 +0200 | [diff] [blame] | 144 | const struct data_ops *data; /* operations at the data layer */ |
| 145 | const struct protocol *ctrl; /* operations at the socket layer */ |
| 146 | const struct app_cb *app_cb; /* application layer callbacks */ |
Willy Tarreau | 56e9c5e | 2012-07-06 09:47:57 +0200 | [diff] [blame] | 147 | union { /* definitions which depend on connection type */ |
| 148 | struct { /*** information used by socket-based connections ***/ |
| 149 | int fd; /* file descriptor for a stream driver when known */ |
| 150 | } sock; |
| 151 | } t; |
Willy Tarreau | 900bc93 | 2012-07-06 09:52:14 +0200 | [diff] [blame] | 152 | unsigned int flags; /* CO_F_* */ |
Willy Tarreau | 56e9c5e | 2012-07-06 09:47:57 +0200 | [diff] [blame] | 153 | int data_st; /* data layer state, initialized to zero */ |
| 154 | void *data_ctx; /* general purpose pointer, initialized to NULL */ |
| 155 | struct sockaddr *peeraddr; /* pointer to peer's network address, or NULL if unset */ |
| 156 | socklen_t peerlen; /* peer's address length, or 0 if unset */ |
| 157 | }; |
| 158 | |
| 159 | #endif /* _TYPES_CONNECTION_H */ |
| 160 | |
| 161 | /* |
| 162 | * Local variables: |
| 163 | * c-indent-level: 8 |
| 164 | * c-basic-offset: 8 |
| 165 | * End: |
| 166 | */ |