Skip to content

Commit

Permalink
MEDIUM: sessions: Introduce session flags.
Browse files Browse the repository at this point in the history
Add session flags, and add a new flag, SESS_FL_PREFER_LAST, to be set when
we use NTLM authentication, and we should reuse the last connection. This
should fix using NTLM with HTX. This totally replaces TX_PREFER_LAST.

This should be backported to 1.9.

(cherry picked from commit 250031e)
Signed-off-by: Willy Tarreau <w@1wt.eu>
  • Loading branch information
Olivier Houchard authored and wtarreau committed Jun 7, 2019
1 parent 218bf79 commit 107a732
Show file tree
Hide file tree
Showing 7 changed files with 15 additions and 6 deletions.
1 change: 0 additions & 1 deletion contrib/debug/flags.c
Expand Up @@ -275,7 +275,6 @@ void show_txn_flags(unsigned int f)
SHOW_FLAG(f, TX_HDR_CONN_PRS);
SHOW_FLAG(f, TX_WAIT_NEXT_RQ);
SHOW_FLAG(f, TX_HDR_CONN_UPG);
SHOW_FLAG(f, TX_PREFER_LAST);
SHOW_FLAG(f, TX_CON_KAL_SET);
SHOW_FLAG(f, TX_CON_CLO_SET);

Expand Down
2 changes: 1 addition & 1 deletion include/types/proto_http.h
Expand Up @@ -94,7 +94,7 @@
#define TX_CON_CLO_SET 0x00400000 /* "connection: close" is now set */
#define TX_CON_KAL_SET 0x00800000 /* "connection: keep-alive" is now set */

#define TX_PREFER_LAST 0x01000000 /* try to stay on same server if possible (eg: after 401) */
/* unused: 0x01000000 */

#define TX_HDR_CONN_UPG 0x02000000 /* The "Upgrade" token was found in the "Connection" header */
#define TX_WAIT_NEXT_RQ 0x04000000 /* waiting for the second request to start, use keep-alive timeout */
Expand Down
7 changes: 7 additions & 0 deletions include/types/session.h
Expand Up @@ -45,6 +45,12 @@ struct sess_srv_list {

#define MAX_SRV_LIST 5

/* session flags */
enum {
SESS_FL_NONE = 0x00000000, /* nothing */
SESS_FL_PREFER_LAST = 0x00000001, /* NTML authent, we should reuse last conn */
};

struct session {
struct proxy *fe; /* the proxy this session depends on for the client side */
struct listener *listener; /* the listener by which the request arrived */
Expand All @@ -57,6 +63,7 @@ struct session {
long t_handshake; /* handshake duration, -1 = not completed */
int idle_conns; /* Number of connections we're currently responsible for that we are not using */
struct list srv_list; /* List of servers and the connections the session is currently responsible for */
unsigned int flags; /* session flags, SESS_FL_* */
};

#endif /* _TYPES_SESSION_H */
Expand Down
4 changes: 2 additions & 2 deletions src/backend.c
Expand Up @@ -633,14 +633,14 @@ int assign_server(struct stream *s)
s->target = NULL;

if ((s->be->lbprm.algo & BE_LB_KIND) != BE_LB_KIND_HI &&
((s->txn && s->txn->flags & TX_PREFER_LAST) ||
((s->sess->flags & SESS_FL_PREFER_LAST) ||
(s->be->options & PR_O_PREF_LAST))) {
struct sess_srv_list *srv_list;
list_for_each_entry(srv_list, &s->sess->srv_list, srv_list) {
struct server *tmpsrv = objt_server(srv_list->target);

if (tmpsrv && tmpsrv->proxy == s->be &&
((s->txn && s->txn->flags & TX_PREFER_LAST) ||
((s->sess->flags & SESS_FL_PREFER_LAST) ||
(!s->be->max_ka_queue ||
server_has_room(tmpsrv) || (
tmpsrv->nbpend + 1 < s->be->max_ka_queue))) &&
Expand Down
2 changes: 1 addition & 1 deletion src/proto_http.c
Expand Up @@ -3539,7 +3539,7 @@ void http_end_txn_clean_session(struct stream *s)
* it's better to do it (at least it helps with debugging), at
* least for non-deterministic load balancing algorithms.
*/
s->txn->flags |= TX_PREFER_LAST;
s->sess->flags |= SESS_FL_PREFER_LAST;
}

/* Never ever allow to reuse a connection from a non-reuse backend */
Expand Down
4 changes: 3 additions & 1 deletion src/proto_htx.c
Expand Up @@ -1735,8 +1735,10 @@ int htx_wait_for_response(struct stream *s, struct channel *rep, int an_bit)
ctx.blk = NULL;
while (http_find_header(htx, hdr, &ctx, 0)) {
if ((ctx.value.len >= 9 && word_match(ctx.value.ptr, ctx.value.len, "Negotiate", 9)) ||
(ctx.value.len >= 4 && word_match(ctx.value.ptr, ctx.value.len, "NTLM", 4)))
(ctx.value.len >= 4 && word_match(ctx.value.ptr, ctx.value.len, "NTLM", 4))) {
sess->flags |= SESS_FL_PREFER_LAST;
srv_conn->flags |= CO_FL_PRIVATE;
}
}
}

Expand Down
1 change: 1 addition & 0 deletions src/session.c
Expand Up @@ -59,6 +59,7 @@ struct session *session_new(struct proxy *fe, struct listener *li, enum obj_type
HA_ATOMIC_ADD(&jobs, 1);
LIST_INIT(&sess->srv_list);
sess->idle_conns = 0;
sess->flags = SESS_FL_NONE;
}
return sess;
}
Expand Down

0 comments on commit 107a732

Please sign in to comment.