Willy Tarreau | b1ec8c4 | 2015-04-03 13:53:24 +0200 | [diff] [blame] | 1 | /* |
| 2 | * include/proto/session.h |
| 3 | * This file defines everything related to sessions. |
| 4 | * |
| 5 | * Copyright (C) 2000-2015 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 _PROTO_SESSION_H |
| 23 | #define _PROTO_SESSION_H |
| 24 | |
| 25 | #include <common/config.h> |
| 26 | #include <common/buffer.h> |
| 27 | #include <common/debug.h> |
| 28 | #include <common/memory.h> |
| 29 | |
| 30 | #include <types/global.h> |
| 31 | #include <types/session.h> |
| 32 | |
Willy Tarreau | bb2ef12 | 2015-04-04 16:31:16 +0200 | [diff] [blame] | 33 | #include <proto/stick_table.h> |
| 34 | |
Willy Tarreau | bafbe01 | 2017-11-24 17:34:44 +0100 | [diff] [blame] | 35 | extern struct pool_head *pool_head_session; |
Willy Tarreau | c38f71c | 2015-04-05 00:38:48 +0200 | [diff] [blame] | 36 | struct session *session_new(struct proxy *fe, struct listener *li, enum obj_type *origin); |
Willy Tarreau | 11c3624 | 2015-04-04 15:54:03 +0200 | [diff] [blame] | 37 | void session_free(struct session *sess); |
Willy Tarreau | 9903f0e | 2015-04-04 18:50:31 +0200 | [diff] [blame] | 38 | int session_accept_fd(struct listener *l, int cfd, struct sockaddr_storage *addr); |
Willy Tarreau | b1ec8c4 | 2015-04-03 13:53:24 +0200 | [diff] [blame] | 39 | |
Willy Tarreau | bb2ef12 | 2015-04-04 16:31:16 +0200 | [diff] [blame] | 40 | /* Remove the refcount from the session to the tracked counters, and clear the |
| 41 | * pointer to ensure this is only performed once. The caller is responsible for |
| 42 | * ensuring that the pointer is valid first. |
| 43 | */ |
| 44 | static inline void session_store_counters(struct session *sess) |
| 45 | { |
| 46 | void *ptr; |
| 47 | int i; |
Emeric Brun | 819fc6f | 2017-06-13 19:37:32 +0200 | [diff] [blame] | 48 | struct stksess *ts; |
Willy Tarreau | bb2ef12 | 2015-04-04 16:31:16 +0200 | [diff] [blame] | 49 | |
| 50 | for (i = 0; i < MAX_SESS_STKCTR; i++) { |
| 51 | struct stkctr *stkctr = &sess->stkctr[i]; |
| 52 | |
Emeric Brun | 819fc6f | 2017-06-13 19:37:32 +0200 | [diff] [blame] | 53 | ts = stkctr_entry(stkctr); |
| 54 | if (!ts) |
Willy Tarreau | bb2ef12 | 2015-04-04 16:31:16 +0200 | [diff] [blame] | 55 | continue; |
| 56 | |
Emeric Brun | 819fc6f | 2017-06-13 19:37:32 +0200 | [diff] [blame] | 57 | ptr = stktable_data_ptr(stkctr->table, ts, STKTABLE_DT_CONN_CUR); |
| 58 | if (ptr) { |
Christopher Faulet | 2a944ee | 2017-11-07 10:42:54 +0100 | [diff] [blame] | 59 | HA_RWLOCK_WRLOCK(STK_SESS_LOCK, &ts->lock); |
Emeric Brun | 819fc6f | 2017-06-13 19:37:32 +0200 | [diff] [blame] | 60 | |
Willy Tarreau | bb2ef12 | 2015-04-04 16:31:16 +0200 | [diff] [blame] | 61 | stktable_data_cast(ptr, conn_cur)--; |
Emeric Brun | 819fc6f | 2017-06-13 19:37:32 +0200 | [diff] [blame] | 62 | |
Christopher Faulet | 2a944ee | 2017-11-07 10:42:54 +0100 | [diff] [blame] | 63 | HA_RWLOCK_WRUNLOCK(STK_SESS_LOCK, &ts->lock); |
Emeric Brun | 0fed0b0 | 2017-11-29 16:15:07 +0100 | [diff] [blame] | 64 | |
| 65 | /* If data was modified, we need to touch to re-schedule sync */ |
| 66 | stktable_touch_local(stkctr->table, ts, 0); |
Emeric Brun | 819fc6f | 2017-06-13 19:37:32 +0200 | [diff] [blame] | 67 | } |
| 68 | |
Willy Tarreau | bb2ef12 | 2015-04-04 16:31:16 +0200 | [diff] [blame] | 69 | stkctr_set_entry(stkctr, NULL); |
Emeric Brun | 819fc6f | 2017-06-13 19:37:32 +0200 | [diff] [blame] | 70 | stksess_kill_if_expired(stkctr->table, ts, 1); |
Willy Tarreau | bb2ef12 | 2015-04-04 16:31:16 +0200 | [diff] [blame] | 71 | } |
| 72 | } |
| 73 | |
Olivier Houchard | 00cf70f | 2018-11-30 17:24:55 +0100 | [diff] [blame] | 74 | static inline void session_add_conn(struct session *sess, struct connection *conn, void *target) |
| 75 | { |
| 76 | int avail = -1; |
| 77 | int i; |
| 78 | |
| 79 | for (i = 0; i < MAX_SRV_LIST; i++) { |
| 80 | if (sess->srv_list[i].target == target) { |
| 81 | avail = i; |
| 82 | break; |
| 83 | } |
| 84 | if (LIST_ISEMPTY(&sess->srv_list[i].list) && avail == -1) |
| 85 | avail = i; |
| 86 | } |
| 87 | if (avail == -1) { |
| 88 | struct connection *conn, *conn_back; |
| 89 | int count = 0; |
| 90 | /* We have no slot free, let's free the one with the fewer connections */ |
| 91 | for (i = 0; i < MAX_SRV_LIST; i++) { |
| 92 | int count_list = 0; |
| 93 | list_for_each_entry(conn, &sess->srv_list[i].list, session_list) |
| 94 | count_list++; |
| 95 | if (count == 0 || count_list < count) { |
| 96 | count = count_list; |
| 97 | avail = i; |
| 98 | } |
| 99 | } |
| 100 | /* Now unown all the connections */ |
| 101 | list_for_each_entry_safe(conn, conn_back, &sess->srv_list[avail].list, session_list) { |
| 102 | conn->owner = NULL; |
| 103 | LIST_DEL(&conn->session_list); |
| 104 | LIST_INIT(&conn->session_list); |
| 105 | if (conn->mux) |
| 106 | conn->mux->destroy(conn); |
| 107 | } |
| 108 | |
| 109 | } |
| 110 | sess->srv_list[avail].target = target; |
| 111 | LIST_ADDQ(&sess->srv_list[avail].list, &conn->session_list); |
| 112 | } |
| 113 | |
Willy Tarreau | bb2ef12 | 2015-04-04 16:31:16 +0200 | [diff] [blame] | 114 | |
Willy Tarreau | b1ec8c4 | 2015-04-03 13:53:24 +0200 | [diff] [blame] | 115 | #endif /* _PROTO_SESSION_H */ |
| 116 | |
| 117 | /* |
| 118 | * Local variables: |
| 119 | * c-indent-level: 8 |
| 120 | * c-basic-offset: 8 |
| 121 | * End: |
| 122 | */ |