mirrors/haproxy
1
0
Fork 0
mirror of https://git.haproxy.org/git/haproxy.git/ synced 2025-09-22 22:31:28 +02:00
Code Issues Projects Releases Wiki Activity

MEDIUM: checks: Use a non-comment rule iterator to get next rule

Browse Source 
This kind of iteration is used several times with various degrees of
clarity. Make a proper function for this use.
This commit is contained in:
Gaetan Rivet 2020-02-14 17:47:08 +01:00 committed by Christopher Faulet
parent 04578dbf37
commit 1d22d7ec0e
1 changed files with 36 additions and 46 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

82
src/checks.c
View File

@ -1561,6 +1561,25 @@ static struct tcpcheck_rule *get_first_tcpcheck_rule(struct list *list)
return NULL; return NULL;
} }
/* returns the NON-COMMENT tcp-check rule from list <list> following <start> or
* NULL if non was found. If <start> is NULL, it relies on
* get_first_tcpcheck_rule().
*/
static struct tcpcheck_rule *get_next_tcpcheck_rule(struct list *list, struct tcpcheck_rule *start)
{
struct tcpcheck_rule *r;
if (!start)
return get_first_tcpcheck_rule(list);
r = LIST_NEXT(&start->list, typeof(r), list);
list_for_each_entry_from(r, list, list) {
if (r->action != TCPCHK_ACT_COMMENT)
return r;
}
return NULL;
}
/* /*
* establish a server health-check that makes use of a connection. * establish a server health-check that makes use of a connection.
* *
@ -1594,7 +1613,7 @@ static int connect_conn_chk(struct task *t)
/* tcpcheck send/expect initialisation */ /* tcpcheck send/expect initialisation */
if (check->type == PR_O2_TCPCHK_CHK) { if (check->type == PR_O2_TCPCHK_CHK) {
check->current_step = NULL; check->current_step = check->last_started_step = NULL;
tcp_rule = get_first_tcpcheck_rule(check->tcpcheck_rules); tcp_rule = get_first_tcpcheck_rule(check->tcpcheck_rules);
} }
@ -2714,7 +2733,7 @@ static int tcpcheck_get_step_id(struct check *check)
int i = 0; int i = 0;
/* not even started anything yet => step 0 = initial connect */ /* not even started anything yet => step 0 = initial connect */
if (!check->current_step) if (!check->current_step && !check->last_started_step)
return 0; return 0;
/* no last started step => first step */ /* no last started step => first step */
@ -2808,11 +2827,9 @@ static int tcpcheck_main(struct check *check)
*/ */
/* find first rule and skip comments */ /* find first rule and skip comments */
next = LIST_NEXT(head, struct tcpcheck_rule *, list); next = get_first_tcpcheck_rule(head);
while (&next->list != head && next->action == TCPCHK_ACT_COMMENT)
next = LIST_NEXT(&next->list, struct tcpcheck_rule *, list);
if ((check->current_step || &next->list == head) && if ((check->current_step || next == NULL) &&
(conn->flags & CO_FL_WAIT_XPRT)) { (conn->flags & CO_FL_WAIT_XPRT)) {
/* we allow up to min(inter, timeout.connect) for a connection /* we allow up to min(inter, timeout.connect) for a connection
* to establish but only when timeout.check is set * to establish but only when timeout.check is set
@ -2831,14 +2848,13 @@ static int tcpcheck_main(struct check *check)
} }
/* special case: option tcp-check with no rule, a connect is enough */ /* special case: option tcp-check with no rule, a connect is enough */
if (&next->list == head) { if (next == NULL) {
set_server_check_status(check, HCHK_STATUS_L4OK, NULL); set_server_check_status(check, HCHK_STATUS_L4OK, NULL);
goto out_end_tcpcheck; goto out_end_tcpcheck;
} }
/* no step means first step initialisation */ /* no step means first step initialisation */
if (check->current_step == NULL) { if (check->current_step == NULL && check->last_started_step == NULL) {
check->last_started_step = NULL;
b_reset(&check->bo); b_reset(&check->bo);
b_reset(&check->bi); b_reset(&check->bi);
check->current_step = next; check->current_step = next;
@ -2854,7 +2870,7 @@ static int tcpcheck_main(struct check *check)
* loop after this control. If we have data, conn is valid. * loop after this control. If we have data, conn is valid.
*/ */
if (b_data(&check->bo) && if (b_data(&check->bo) &&
(&check->current_step->list == head || (check->current_step == NULL ||
check->current_step->action != TCPCHK_ACT_SEND || check->current_step->action != TCPCHK_ACT_SEND ||
check->current_step->string_len >= b_room(&check->bo))) { check->current_step->string_len >= b_room(&check->bo))) {
int ret; int ret;
@ -2875,21 +2891,13 @@ static int tcpcheck_main(struct check *check)
} }
} }
if (&check->current_step->list == head) if (check->current_step == NULL)
break; break;
/* have 'next' point to the next rule or NULL if we're on the /* have 'next' point to the next rule or NULL if we're on the
* last one, connect() needs this. * last one, connect() needs this.
*/ */
next = LIST_NEXT(&check->current_step->list, struct tcpcheck_rule *, list); next = get_next_tcpcheck_rule(head, check->current_step);
/* bypass all comment rules */
while (&next->list != head && next->action == TCPCHK_ACT_COMMENT)
next = LIST_NEXT(&next->list, struct tcpcheck_rule *, list);
/* NULL if we're on the last rule */
if (&next->list == head)
next = NULL;
if (check->current_step->action == TCPCHK_ACT_CONNECT) { if (check->current_step->action == TCPCHK_ACT_CONNECT) {
struct protocol *proto; struct protocol *proto;
@ -3062,13 +3070,8 @@ static int tcpcheck_main(struct check *check)
} }
/* allow next rule */ /* allow next rule */
check->current_step = LIST_NEXT(&check->current_step->list, struct tcpcheck_rule *, list); check->current_step = get_next_tcpcheck_rule(head, check->current_step);
if (check->current_step == NULL)
/* bypass all comment rules */
while (&check->current_step->list != head && check->current_step->action == TCPCHK_ACT_COMMENT)
check->current_step = LIST_NEXT(&check->current_step->list, struct tcpcheck_rule *, list);
if (&check->current_step->list == head)
break; break;
/* don't do anything until the connection is established */ /* don't do anything until the connection is established */
@ -3098,14 +3101,7 @@ static int tcpcheck_main(struct check *check)
b_putblk(&check->bo, check->current_step->string, check->current_step->string_len); b_putblk(&check->bo, check->current_step->string, check->current_step->string_len);
/* go to next rule and try to send */ check->current_step = get_next_tcpcheck_rule(head, check->current_step);
check->current_step = LIST_NEXT(&check->current_step->list, struct tcpcheck_rule *, list);
/* bypass all comment rules */
while (&check->current_step->list != head &&
check->current_step->action == TCPCHK_ACT_COMMENT)
check->current_step = LIST_NEXT(&check->current_step->list, struct tcpcheck_rule *, list);
} /* end 'send' */ } /* end 'send' */
else if (check->current_step->action == TCPCHK_ACT_EXPECT) { else if (check->current_step->action == TCPCHK_ACT_EXPECT) {
struct tcpcheck_expect *expect = &check->current_step->expect; struct tcpcheck_expect *expect = &check->current_step->expect;
@ -3217,14 +3213,8 @@ static int tcpcheck_main(struct check *check)
if (match ^ expect->inverse) { if (match ^ expect->inverse) {
/* Result as expected, next rule. */ /* Result as expected, next rule. */
check->current_step = LIST_NEXT(&check->current_step->list, struct tcpcheck_rule *, list); check->current_step = get_next_tcpcheck_rule(head, check->current_step);
if (check->current_step == NULL)
/* bypass all comment rules */
while (&check->current_step->list != head &&
check->current_step->action == TCPCHK_ACT_COMMENT)
check->current_step = LIST_NEXT(&check->current_step->list,
struct tcpcheck_rule *, list);
if (&check->current_step->list == head)
break; break;
if (check->current_step->action == TCPCHK_ACT_EXPECT) { if (check->current_step->action == TCPCHK_ACT_EXPECT) {
@ -3313,12 +3303,12 @@ static int tcpcheck_main(struct check *check)
/* We're waiting for some I/O to complete, we've reached the end of the /* We're waiting for some I/O to complete, we've reached the end of the
* rules, or both. Do what we have to do, otherwise we're done. * rules, or both. Do what we have to do, otherwise we're done.
*/ */
if (&check->current_step->list == head && !b_data(&check->bo)) { if (check->current_step == NULL && !b_data(&check->bo)) {
set_server_check_status(check, HCHK_STATUS_L7OKD, "(tcp-check)"); set_server_check_status(check, HCHK_STATUS_L7OKD, "(tcp-check)");
goto out_end_tcpcheck; goto out_end_tcpcheck;
} }
if (&check->current_step->list != head && if (check->current_step != NULL &&
check->current_step->action == TCPCHK_ACT_EXPECT) check->current_step->action == TCPCHK_ACT_EXPECT)
__event_srv_chk_r(cs); __event_srv_chk_r(cs);
goto out; goto out;
@ -3329,7 +3319,7 @@ static int tcpcheck_main(struct check *check)
chk_report_conn_err(check, 0, 0); chk_report_conn_err(check, 0, 0);
/* cleanup before leaving */ /* cleanup before leaving */
check->current_step = NULL; check->current_step = check->last_started_step = NULL;
if (check->result == CHK_RES_FAILED) if (check->result == CHK_RES_FAILED)
conn->flags |= CO_FL_ERROR; conn->flags |= CO_FL_ERROR;